📄 _cookie_module_8cpp-source.html
字号:
<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN"><html><head><meta http-equiv="Content-Type" content="text/html;charset=UTF-8"><title>libpion: /Users/mdickey/code/libpion-export/modules/CookieModule.cpp Source File</title><link href="doxygen.css" rel="stylesheet" type="text/css"><link href="tabs.css" rel="stylesheet" type="text/css"></head><body><!-- Generated by Doxygen 1.5.2 --><div class="tabs"> <ul> <li><a href="index.html"><span>Main Page</span></a></li> <li><a href="namespaces.html"><span>Namespaces</span></a></li> <li><a href="annotated.html"><span>Classes</span></a></li> <li class="current"><a href="files.html"><span>Files</span></a></li> </ul></div><h1>/Users/mdickey/code/libpion-export/modules/CookieModule.cpp</h1><div class="fragment"><pre class="fragment"><a name="l00001"></a>00001 <span class="comment">// -----------------------------------------------------------------</span><a name="l00002"></a>00002 <span class="comment">// libpion: a C++ framework for building lightweight HTTP interfaces</span><a name="l00003"></a>00003 <span class="comment">// -----------------------------------------------------------------</span><a name="l00004"></a>00004 <span class="comment">// Copyright (C) 2007 Atomic Labs, Inc. (http://www.atomiclabs.com)</span><a name="l00005"></a>00005 <span class="comment">//</span><a name="l00006"></a>00006 <span class="comment">// Distributed under the Boost Software License, Version 1.0.</span><a name="l00007"></a>00007 <span class="comment">// See accompanying file COPYING or copy at http://www.boost.org/LICENSE_1_0.txt</span><a name="l00008"></a>00008 <span class="comment">//</span><a name="l00009"></a>00009 <a name="l00010"></a>00010 <span class="preprocessor">#include "CookieModule.hpp"</span><a name="l00011"></a>00011 <span class="preprocessor">#include <libpion/HTTPResponse.hpp></span><a name="l00012"></a>00012 <a name="l00013"></a>00013 <span class="keyword">using namespace </span>pion;<a name="l00014"></a>00014 <a name="l00015"></a>00015 <a name="l00016"></a>00016 <span class="comment">// CookieModule member functions</span><a name="l00017"></a>00017 <a name="l00019"></a><a class="code" href="class_cookie_module.html#089422fe106cb23a93ae7845215d174c">00019</a> <span class="keywordtype">bool</span> <a class="code" href="class_cookie_module.html#089422fe106cb23a93ae7845215d174c" title="handles requests for CookieModule">CookieModule::handleRequest</a>(HTTPRequestPtr& request, TCPConnectionPtr& tcp_conn)<a name="l00020"></a>00020 {<a name="l00021"></a>00021 <span class="keyword">static</span> <span class="keyword">const</span> std::string HEADER_HTML = <span class="stringliteral">"<html>\n<head>\n<title>Cookie Module</title>\n"</span><a name="l00022"></a>00022 <span class="stringliteral">"</head>\n<body>\n\n<h1>Cookie Module</h1>\n"</span>;<a name="l00023"></a>00023 <span class="keyword">static</span> <span class="keyword">const</span> std::string FOOTER_HTML = <span class="stringliteral">"\n</body>\n</html>\n"</span>;<a name="l00024"></a>00024 <a name="l00025"></a>00025 <span class="comment">// Set Content-type for HTML and write the header</span><a name="l00026"></a>00026 <a class="code" href="namespacepion.html#e27a744c3e8cf19c3620400bfe68fb69" title="data type for a HTTPResponse pointer">HTTPResponsePtr</a> response(HTTPResponse::create());<a name="l00027"></a>00027 response->setContentType(HTTPTypes::CONTENT_TYPE_HTML);<a name="l00028"></a>00028 response->writeNoCopy(HEADER_HTML);<a name="l00029"></a>00029 <a name="l00030"></a>00030 <span class="comment">// Check if we have an action to perform</span><a name="l00031"></a>00031 <span class="keywordflow">if</span> (request->hasQuery(<span class="stringliteral">"action"</span>)) {<a name="l00032"></a>00032 <span class="keywordflow">if</span> (HTTPTypes::url_decode(request->getQuery(<span class="stringliteral">"action"</span>)) == <span class="stringliteral">"Add Cookie"</span>) {<a name="l00033"></a>00033 <span class="comment">// add a new cookie</span><a name="l00034"></a>00034 <span class="keyword">const</span> std::string cookie_name(request->getQuery(<span class="stringliteral">"cookie_name"</span>));<a name="l00035"></a>00035 <span class="keyword">const</span> std::string cookie_value(request->getQuery(<span class="stringliteral">"cookie_value"</span>));<a name="l00036"></a>00036 <span class="keywordflow">if</span> (cookie_name.empty() || cookie_value.empty()) {<a name="l00037"></a>00037 response << <span class="stringliteral">"\n<p>[Error: You must specify a name and value to add a cookie]</p>\n\n"</span>;<a name="l00038"></a>00038 } <span class="keywordflow">else</span> {<a name="l00039"></a>00039 response->setCookie(cookie_name, cookie_value);<a name="l00040"></a>00040 response << <span class="stringliteral">"\n<p>[Added cookie "</span><a name="l00041"></a>00041 << cookie_name << <span class="charliteral">'='</span> << cookie_value << <span class="stringliteral">"]</p>\n\n"</span>;<a name="l00042"></a>00042 }<a name="l00043"></a>00043 } <span class="keywordflow">else</span> <span class="keywordflow">if</span> (request->getQuery(<span class="stringliteral">"action"</span>) == <span class="stringliteral">"delete"</span>) {<a name="l00044"></a>00044 <span class="keyword">const</span> std::string cookie_name(request->getQuery(<span class="stringliteral">"cookie_name"</span>));<a name="l00045"></a>00045 <span class="keywordflow">if</span> (cookie_name.empty()) {<a name="l00046"></a>00046 response << <span class="stringliteral">"\n<p>[Error: You must specify a name to delete a cookie]</p>\n\n"</span>;<a name="l00047"></a>00047 } <span class="keywordflow">else</span> {<a name="l00048"></a>00048 response->deleteCookie(cookie_name);<a name="l00049"></a>00049 response << <span class="stringliteral">"\n<p>[Deleted cookie "</span> << cookie_name << <span class="stringliteral">"]</p>\n\n"</span>;<a name="l00050"></a>00050 }<a name="l00051"></a>00051 } <span class="keywordflow">else</span> {<a name="l00052"></a>00052 response << <span class="stringliteral">"\n<p>[Error: Unrecognized action]</p>\n\n"</span>;<a name="l00053"></a>00053 }<a name="l00054"></a>00054 }<a name="l00055"></a>00055 <a name="l00056"></a>00056 <span class="comment">// display cookie headers in request</span><a name="l00057"></a>00057 <span class="keywordflow">if</span> (request->hasHeader(HTTPTypes::HEADER_COOKIE)) {<a name="l00058"></a>00058 response << <span class="stringliteral">"\n<h2>Cookie Headers</h2>\n<ul>\n"</span>;<a name="l00059"></a>00059 std::pair<HTTPTypes::Headers::const_iterator, HTTPTypes::Headers::const_iterator><a name="l00060"></a>00060 header_pair = request->getHeaders().equal_range(HTTPTypes::HEADER_COOKIE);<a name="l00061"></a>00061 <span class="keywordflow">for</span> (HTTPTypes::Headers::const_iterator header_iterator = header_pair.first;<a name="l00062"></a>00062 header_iterator != request->getCookieParams().end()<a name="l00063"></a>00063 && header_iterator != header_pair.second; ++header_iterator)<a name="l00064"></a>00064 {<a name="l00065"></a>00065 response << <span class="stringliteral">"<li>Cookie: "</span> << header_iterator->second << <span class="stringliteral">"\n"</span>;<a name="l00066"></a>00066 }<a name="l00067"></a>00067 response << <span class="stringliteral">"</ul>\n\n"</span>;<a name="l00068"></a>00068 } <span class="keywordflow">else</span> {<a name="l00069"></a>00069 response << <span class="stringliteral">"\n<h2>No Cookie Headers</h2>\n\n"</span>;<a name="l00070"></a>00070 }<a name="l00071"></a>00071 <a name="l00072"></a>00072 <span class="comment">// display existing cookies</span><a name="l00073"></a>00073 HTTPTypes::CookieParams& cookie_params = request->getCookieParams();<a name="l00074"></a>00074 <span class="keywordflow">if</span> (! cookie_params.empty()) {<a name="l00075"></a>00075 response << <span class="stringliteral">"\n<h2>Cookie Variables</h2>\n<ul>\n"</span>;<a name="l00076"></a>00076 <span class="keywordflow">for</span> (HTTPTypes::CookieParams::const_iterator i = cookie_params.begin();<a name="l00077"></a>00077 i != cookie_params.end(); ++i)<a name="l00078"></a>00078 {<a name="l00079"></a>00079 response << <span class="stringliteral">"<li>"</span> << i->first << <span class="stringliteral">": "</span> << i->second<a name="l00080"></a>00080 << <span class="stringliteral">" <a href=\""</span> << request->getResource()<a name="l00081"></a>00081 << <span class="stringliteral">"?action=delete&cookie_name="</span> << i->first<a name="l00082"></a>00082 << <span class="stringliteral">"\">[Delete]</a>\n"</span>;<a name="l00083"></a>00083 }<a name="l00084"></a>00084 response << <span class="stringliteral">"</ul>\n\n"</span>;<a name="l00085"></a>00085 } <span class="keywordflow">else</span> {<a name="l00086"></a>00086 response << <span class="stringliteral">"\n<h2>No Cookie Variables</h2>\n\n"</span>;<a name="l00087"></a>00087 }<a name="l00088"></a>00088 <a name="l00089"></a>00089 <span class="comment">// display form to add a cookie</span><a name="l00090"></a>00090 response << <span class="stringliteral">"\n<h2>Add Cookie</h2>\n"</span><a name="l00091"></a>00091 <span class="stringliteral">"<p><form action=\""</span> << request->getResource() << <span class="stringliteral">"\" method=\"POST\">\n"</span><a name="l00092"></a>00092 <span class="stringliteral">"Name: <input type=\"text\" name=\"cookie_name\"><br />\n"</span><a name="l00093"></a>00093 <span class="stringliteral">"Value: <input type=\"text\" name=\"cookie_value\"><br />\n"</span><a name="l00094"></a>00094 <span class="stringliteral">"<input type=\"submit\" name=\"action\" value=\"Add Cookie\"></p>\n"</span><a name="l00095"></a>00095 <span class="stringliteral">"</form>\n\n"</span>;<a name="l00096"></a>00096 <a name="l00097"></a>00097 <span class="comment">// write the footer</span><a name="l00098"></a>00098 response->writeNoCopy(FOOTER_HTML);<a name="l00099"></a>00099 <a name="l00100"></a>00100 <span class="comment">// send the response</span><a name="l00101"></a>00101 response->send(tcp_conn);<a name="l00102"></a>00102 <span class="keywordflow">return</span> <span class="keyword">true</span>;<a name="l00103"></a>00103 }<a name="l00104"></a>00104 <a name="l00105"></a>00105 <a name="l00107"></a>00107 <span class="keyword">extern</span> <span class="stringliteral">"C"</span> <a class="code" href="class_cookie_module.html">CookieModule</a> *pion_create_CookieModule(<span class="keywordtype">void</span>)<a name="l00108"></a>00108 {<a name="l00109"></a>00109 <span class="keywordflow">return</span> <span class="keyword">new</span> <a class="code" href="class_cookie_module.html#06194edd43bccd3e100d252c0fb4c2f7">CookieModule</a>();<a name="l00110"></a>00110 }<a name="l00111"></a>00111 <a name="l00112"></a>00112 <a name="l00114"></a>00114 <span class="keyword">extern</span> <span class="stringliteral">"C"</span> <span class="keywordtype">void</span> pion_destroy_CookieModule(<a class="code" href="class_cookie_module.html">CookieModule</a> *module_ptr)<a name="l00115"></a>00115 {<a name="l00116"></a>00116 <span class="keyword">delete</span> module_ptr;<a name="l00117"></a>00117 }</pre></div><hr size="1"><address style="text-align: right;"><small>Generated on Tue Jun 19 13:29:22 2007 for libpion by <a href="http://www.doxygen.org/index.html"><img src="doxygen.png" alt="doxygen" align="middle" border="0"></a> 1.5.2 </small></address></body></html>⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -