_h_t_t_p_request_parser_8cpp-source.html

用c++编写http server的源码库,对socket等网络处理的代码可迅速转为己用.

<a name="l00111"></a>00111                 }
<a name="l00112"></a>00112             }
<a name="l00113"></a>00113 
<a name="l00114"></a>00114             <span class="keywordflow">if</span> (content_bytes_to_read == 0) {
<a name="l00115"></a>00115             
<a name="l00116"></a>00116                 <span class="comment">// read all of the content from the last read operation</span>
<a name="l00117"></a>00117                 <a class="code" href="classpion_1_1_h_t_t_p_request_parser.html#20e7739e89be3603f6463801e8bdc489">readContentBytes</a>(read_error, 0);
<a name="l00118"></a>00118 
<a name="l00119"></a>00119             } <span class="keywordflow">else</span> {
<a name="l00120"></a>00120 
<a name="l00121"></a>00121                 <span class="comment">// read the rest of the post content into the buffer</span>
<a name="l00122"></a>00122                 <span class="comment">// and only return after we've finished or an error occurs</span>
<a name="l00123"></a>00123                 <span class="keywordflow">if</span> (m_tcp_conn-&gt;getSSLFlag()) {
<a name="l00124"></a>00124 #ifdef PION_HAVE_SSL
<a name="l00125"></a>00125                     boost::asio::async_read(m_tcp_conn-&gt;getSSLSocket(),
<a name="l00126"></a>00126                                             boost::asio::buffer(post_buffer, content_bytes_to_read),
<a name="l00127"></a>00127                                             boost::asio::transfer_at_least(content_bytes_to_read),
<a name="l00128"></a>00128                                             boost::bind(&amp;<a class="code" href="classpion_1_1_h_t_t_p_request_parser.html#20e7739e89be3603f6463801e8bdc489">HTTPRequestParser::readContentBytes</a>,
<a name="l00129"></a>00129                                                         shared_from_this(),
<a name="l00130"></a>00130                                                         boost::asio::placeholders::error,
<a name="l00131"></a>00131                                                         boost::asio::placeholders::bytes_transferred));
<a name="l00132"></a>00132 <span class="preprocessor">#else</span>
<a name="l00133"></a>00133 <span class="preprocessor"></span>                    PION_LOG_ERROR(m_logger, <span class="stringliteral">"SSL flag set for server, but support is not enabled"</span>);
<a name="l00134"></a>00134                     m_tcp_conn-&gt;finish();
<a name="l00135"></a>00135 <span class="preprocessor">#endif</span>
<a name="l00136"></a>00136 <span class="preprocessor"></span>                } <span class="keywordflow">else</span> {
<a name="l00137"></a>00137                     boost::asio::async_read(m_tcp_conn-&gt;getSocket(),
<a name="l00138"></a>00138                                             boost::asio::buffer(post_buffer, content_bytes_to_read),
<a name="l00139"></a>00139                                             boost::asio::transfer_at_least(content_bytes_to_read),
<a name="l00140"></a>00140                                             boost::bind(&amp;<a class="code" href="classpion_1_1_h_t_t_p_request_parser.html#20e7739e89be3603f6463801e8bdc489">HTTPRequestParser::readContentBytes</a>,
<a name="l00141"></a>00141                                                         shared_from_this(),
<a name="l00142"></a>00142                                                         boost::asio::placeholders::error,
<a name="l00143"></a>00143                                                         boost::asio::placeholders::bytes_transferred));
<a name="l00144"></a>00144                 }
<a name="l00145"></a>00145             }
<a name="l00146"></a>00146         }
<a name="l00147"></a>00147         
<a name="l00148"></a>00148     } <span class="keywordflow">else</span> <span class="keywordflow">if</span> (!result) {
<a name="l00149"></a>00149         <span class="comment">// the request is invalid or an error occured</span>
<a name="l00150"></a>00150 
<a name="l00151"></a>00151 <span class="preprocessor">    #ifndef NDEBUG</span>
<a name="l00152"></a>00152 <span class="preprocessor"></span>        <span class="comment">// display extra error information if debug mode is enabled</span>
<a name="l00153"></a>00153         std::string bad_request;
<a name="l00154"></a>00154         read_ptr = m_read_buffer.data();
<a name="l00155"></a>00155         <span class="keywordflow">while</span> (read_ptr &lt; read_end &amp;&amp; bad_request.size() &lt; 50) {
<a name="l00156"></a>00156             <span class="keywordflow">if</span> (!isprint(*read_ptr) || *read_ptr == <span class="charliteral">'\n'</span> || *read_ptr==<span class="charliteral">'\r'</span>)
<a name="l00157"></a>00157                 bad_request += <span class="charliteral">'.'</span>;
<a name="l00158"></a>00158             <span class="keywordflow">else</span> bad_request += *read_ptr;
<a name="l00159"></a>00159             ++read_ptr;
<a name="l00160"></a>00160         }
<a name="l00161"></a>00161         PION_LOG_ERROR(m_logger, <span class="stringliteral">"Bad request debug: "</span> &lt;&lt; bad_request);
<a name="l00162"></a>00162 <span class="preprocessor">    #endif</span>
<a name="l00163"></a>00163 <span class="preprocessor"></span>
<a name="l00164"></a>00164         m_http_request-&gt;setIsValid(<span class="keyword">false</span>);
<a name="l00165"></a>00165         m_request_handler(m_http_request, m_tcp_conn);
<a name="l00166"></a>00166         
<a name="l00167"></a>00167     } <span class="keywordflow">else</span> {
<a name="l00168"></a>00168         <span class="comment">// not yet finished parsing the request -&gt; read more data</span>
<a name="l00169"></a>00169         
<a name="l00170"></a>00170         <a class="code" href="classpion_1_1_h_t_t_p_request_parser.html#9208a39a2d32349d65ffaee6bca5c849" title="Incrementally reads &amp; parses a new HTTP request.">readRequest</a>();
<a name="l00171"></a>00171     }
<a name="l00172"></a>00172 }
<a name="l00173"></a>00173 
<a name="l00174"></a><a class="code" href="classpion_1_1_h_t_t_p_request_parser.html#20e7739e89be3603f6463801e8bdc489">00174</a> <span class="keywordtype">void</span> <a class="code" href="classpion_1_1_h_t_t_p_request_parser.html#20e7739e89be3603f6463801e8bdc489">HTTPRequestParser::readContentBytes</a>(<span class="keyword">const</span> boost::system::error_code&amp; read_error,
<a name="l00175"></a>00175                                          std::size_t bytes_read)
<a name="l00176"></a>00176 {
<a name="l00177"></a>00177     <span class="keywordflow">if</span> (read_error) {
<a name="l00178"></a>00178         <span class="comment">// a read error occured</span>
<a name="l00179"></a>00179         <a class="code" href="classpion_1_1_h_t_t_p_request_parser.html#af553431fca527b6f1eb41489f7a7b1e">handleReadError</a>(read_error);
<a name="l00180"></a>00180         <span class="keywordflow">return</span>;
<a name="l00181"></a>00181     }
<a name="l00182"></a>00182 
<a name="l00183"></a>00183     <span class="keywordflow">if</span> (bytes_read != 0) {
<a name="l00184"></a>00184         PION_LOG_DEBUG(m_logger, <span class="stringliteral">"Read "</span> &lt;&lt; bytes_read &lt;&lt; <span class="stringliteral">" request content bytes (finished)"</span>);
<a name="l00185"></a>00185     }
<a name="l00186"></a>00186 
<a name="l00187"></a>00187     <span class="comment">// the request is valid</span>
<a name="l00188"></a>00188     m_http_request-&gt;setIsValid(<span class="keyword">true</span>);
<a name="l00189"></a>00189     
<a name="l00190"></a>00190     <span class="comment">// parse query pairs from the URI query string</span>
<a name="l00191"></a>00191     <span class="keywordflow">if</span> (! m_http_request-&gt;getQueryString().empty()) {
<a name="l00192"></a>00192         <span class="keywordflow">if</span> (! <a class="code" href="classpion_1_1_h_t_t_p_request_parser.html#57d7592db90c02bca8d57c9b27846bc8">parseURLEncoded</a>(m_http_request-&gt;getQueryParams(),
<a name="l00193"></a>00193                               m_http_request-&gt;getQueryString().c_str(),
<a name="l00194"></a>00194                               m_http_request-&gt;getQueryString().size())) 
<a name="l00195"></a>00195             PION_LOG_WARN(m_logger, <span class="stringliteral">"Request query string parsing failed (URI)"</span>);
<a name="l00196"></a>00196     }
<a name="l00197"></a>00197     
<a name="l00198"></a>00198     <span class="comment">// parse query pairs from post content (x-www-form-urlencoded)</span>
<a name="l00199"></a>00199     <span class="keywordflow">if</span> (m_http_request-&gt;getHeader(<a class="code" href="structpion_1_1_h_t_t_p_types.html#e8e700a28694fa413ef6f504bfa2428b">HTTPTypes::HEADER_CONTENT_TYPE</a>) ==
<a name="l00200"></a>00200         <a class="code" href="structpion_1_1_h_t_t_p_types.html#fc311852428a44319675fca55c720cef">HTTPTypes::CONTENT_TYPE_URLENCODED</a>)
<a name="l00201"></a>00201     {
<a name="l00202"></a>00202         <span class="keywordflow">if</span> (! <a class="code" href="classpion_1_1_h_t_t_p_request_parser.html#57d7592db90c02bca8d57c9b27846bc8">parseURLEncoded</a>(m_http_request-&gt;getQueryParams(),
<a name="l00203"></a>00203                               m_http_request-&gt;getPostContent(),
<a name="l00204"></a>00204                               m_http_request-&gt;getContentLength())) 
<a name="l00205"></a>00205             PION_LOG_WARN(m_logger, <span class="stringliteral">"Request query string parsing failed (POST content)"</span>);
<a name="l00206"></a>00206     }
<a name="l00207"></a>00207     
<a name="l00208"></a>00208     <span class="comment">// parse "Cookie" headers</span>
<a name="l00209"></a>00209     std::pair&lt;HTTPTypes::Headers::const_iterator, HTTPTypes::Headers::const_iterator&gt;
<a name="l00210"></a>00210         cookie_pair = m_http_request-&gt;getHeaders().equal_range(<a class="code" href="structpion_1_1_h_t_t_p_types.html#af65a3450f333790ee3bc10e511a7243">HTTPTypes::HEADER_COOKIE</a>);
<a name="l00211"></a>00211     <span class="keywordflow">for</span> (HTTPTypes::Headers::const_iterator cookie_iterator = cookie_pair.first;
<a name="l00212"></a>00212          cookie_iterator != m_http_request-&gt;getCookieParams().end()
<a name="l00213"></a>00213          &amp;&amp; cookie_iterator != cookie_pair.second; ++cookie_iterator)
<a name="l00214"></a>00214     {
<a name="l00215"></a>00215         <span class="keywordflow">if</span> (! <a class="code" href="classpion_1_1_h_t_t_p_request_parser.html#0de32e9cf76ad689b6f86588c68e3118">parseCookieHeader</a>(m_http_request-&gt;getCookieParams(),
<a name="l00216"></a>00216                                 cookie_iterator-&gt;second) )
<a name="l00217"></a>00217             PION_LOG_WARN(m_logger, <span class="stringliteral">"Cookie header parsing failed"</span>);
<a name="l00218"></a>00218     }
<a name="l00219"></a>00219     
<a name="l00220"></a>00220     <span class="comment">// call the request handler with the finished request</span>
<a name="l00221"></a>00221     m_request_handler(m_http_request, m_tcp_conn);
<a name="l00222"></a>00222 }
<a name="l00223"></a>00223 
<a name="l00224"></a><a class="code" href="classpion_1_1_h_t_t_p_request_parser.html#af553431fca527b6f1eb41489f7a7b1e">00224</a> <span class="keywordtype">void</span> <a class="code" href="classpion_1_1_h_t_t_p_request_parser.html#af553431fca527b6f1eb41489f7a7b1e">HTTPRequestParser::handleReadError</a>(<span class="keyword">const</span> boost::system::error_code&amp; read_error)
<a name="l00225"></a>00225 {
<a name="l00226"></a>00226     <span class="comment">// only log errors if the parsing has already begun</span>
<a name="l00227"></a>00227     <span class="keywordflow">if</span> (m_parse_state != PARSE_METHOD_START) {
<a name="l00228"></a>00228         <span class="keywordflow">if</span> (read_error == boost::asio::error::operation_aborted) {
<a name="l00229"></a>00229             <span class="comment">// if the operation was aborted, the acceptor was stopped,</span>
<a name="l00230"></a>00230             <span class="comment">// which means another thread is shutting-down the server</span>
<a name="l00231"></a>00231             PION_LOG_INFO(m_logger, <span class="stringliteral">"HTTP request parsing aborted (shutting down)"</span>);
<a name="l00232"></a>00232         } <span class="keywordflow">else</span> {
<a name="l00233"></a>00233             PION_LOG_INFO(m_logger, <span class="stringliteral">"HTTP request parsing aborted ("</span> &lt;&lt; read_error.message() &lt;&lt; <span class="charliteral">')'</span>);
<a name="l00234"></a>00234         }
<a name="l00235"></a>00235     }