_t_c_p_server_8cpp-source.html

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

<a name="l00091"></a>00091                                                               boost::bind(&amp;TCPServer::finishConnection,
<a name="l00092"></a>00092                                                                           <span class="keyword">this</span>, _1)));
<a name="l00093"></a>00093         m_conn_pool.insert(new_connection);
<a name="l00094"></a>00094         
<a name="l00095"></a>00095         <span class="comment">// use the new object to accept a connection</span>
<a name="l00096"></a>00096         <span class="keywordflow">if</span> (m_ssl_flag) {
<a name="l00097"></a>00097 <span class="preprocessor">#ifdef PION_HAVE_SSL</span>
<a name="l00098"></a>00098 <span class="preprocessor"></span>            m_tcp_acceptor.async_accept(new_connection-&gt;getSSLSocket().lowest_layer(),
<a name="l00099"></a>00099                                         boost::bind(&amp;TCPServer::handleAccept,
<a name="l00100"></a>00100                                                     <span class="keyword">this</span>, new_connection,
<a name="l00101"></a>00101                                                     boost::asio::placeholders::error) );
<a name="l00102"></a>00102 <span class="preprocessor">#else</span>
<a name="l00103"></a>00103 <span class="preprocessor"></span>            PION_LOG_ERROR(<a class="code" href="classpion_1_1_t_c_p_server.html#028f2f5c2af69741b003b438ce682029" title="primary logging interface used by this class">m_logger</a>, <span class="stringliteral">"SSL flag set for server, but support is not enabled"</span>);
<a name="l00104"></a>00104             new_connection-&gt;finish();
<a name="l00105"></a>00105 <span class="preprocessor">#endif</span>
<a name="l00106"></a>00106 <span class="preprocessor"></span>        } <span class="keywordflow">else</span> {
<a name="l00107"></a>00107             m_tcp_acceptor.async_accept(new_connection-&gt;getSocket(),
<a name="l00108"></a>00108                                         boost::bind(&amp;TCPServer::handleAccept,
<a name="l00109"></a>00109                                                     <span class="keyword">this</span>, new_connection,
<a name="l00110"></a>00110                                                     boost::asio::placeholders::error) );
<a name="l00111"></a>00111         }
<a name="l00112"></a>00112     }
<a name="l00113"></a>00113 }
<a name="l00114"></a>00114 
<a name="l00115"></a>00115 <span class="keywordtype">void</span> TCPServer::handleAccept(<a class="code" href="namespacepion.html#1858b89656b28c6498ad9efd6b7e9d41" title="data type for a TCPConnection pointer">TCPConnectionPtr</a>&amp; tcp_conn,
<a name="l00116"></a>00116                              <span class="keyword">const</span> boost::system::error_code&amp; accept_error)
<a name="l00117"></a>00117 {
<a name="l00118"></a>00118     <span class="keywordflow">if</span> (accept_error) {
<a name="l00119"></a>00119         <span class="comment">// an error occured while trying to a accept a new connection</span>
<a name="l00120"></a>00120         <span class="comment">// this happens when the server is being shut down</span>
<a name="l00121"></a>00121         <span class="keywordflow">if</span> (m_is_listening) {
<a name="l00122"></a>00122             tcp_conn-&gt;setKeepAlive(<span class="keyword">false</span>);  <span class="comment">// make sure it will get closed</span>
<a name="l00123"></a>00123             finishConnection(tcp_conn);
<a name="l00124"></a>00124         }
<a name="l00125"></a>00125     } <span class="keywordflow">else</span> {
<a name="l00126"></a>00126         <span class="comment">// got a new TCP connection</span>
<a name="l00127"></a>00127         PION_LOG_INFO(<a class="code" href="classpion_1_1_t_c_p_server.html#028f2f5c2af69741b003b438ce682029" title="primary logging interface used by this class">m_logger</a>, <span class="stringliteral">"New"</span> &lt;&lt; (tcp_conn-&gt;getSSLFlag() ? <span class="stringliteral">" SSL "</span> : <span class="stringliteral">" "</span>)
<a name="l00128"></a>00128                       &lt;&lt; <span class="stringliteral">"connection on port "</span> &lt;&lt; <a class="code" href="classpion_1_1_t_c_p_server.html#ae2c1a58569dfd7342fcf6ccac0746a2" title="returns tcp port number server listens for connections on">getPort</a>());
<a name="l00129"></a>00129 
<a name="l00130"></a>00130         <span class="comment">// schedule the acceptance of another new connection</span>
<a name="l00131"></a>00131         <span class="comment">// (this returns immediately since it schedules it as an event)</span>
<a name="l00132"></a>00132         <span class="keywordflow">if</span> (m_is_listening) listen();
<a name="l00133"></a>00133         
<a name="l00134"></a>00134         <span class="comment">// handle the new connection</span>
<a name="l00135"></a>00135         <span class="keywordflow">if</span> (tcp_conn-&gt;getSSLFlag()) {
<a name="l00136"></a>00136 #ifdef PION_HAVE_SSL
<a name="l00137"></a>00137             <span class="comment">// SSL -&gt; perform handshake first</span>
<a name="l00138"></a>00138             tcp_conn-&gt;getSSLSocket().async_handshake(boost::asio::ssl::stream_base::server,
<a name="l00139"></a>00139                                                      boost::bind(&amp;TCPServer::handleSSLHandshake,
<a name="l00140"></a>00140                                                                  <span class="keyword">this</span>, tcp_conn,
<a name="l00141"></a>00141                                                                  boost::asio::placeholders::error) );
<a name="l00142"></a>00142 <span class="preprocessor">#else</span>
<a name="l00143"></a>00143 <span class="preprocessor"></span>            PION_LOG_ERROR(<a class="code" href="classpion_1_1_t_c_p_server.html#028f2f5c2af69741b003b438ce682029" title="primary logging interface used by this class">m_logger</a>, <span class="stringliteral">"SSL flag set for server, but support is not enabled"</span>);
<a name="l00144"></a>00144             tcp_conn-&gt;setKeepAlive(<span class="keyword">false</span>);  <span class="comment">// make sure it will get closed</span>
<a name="l00145"></a>00145             finishConnection(tcp_conn);
<a name="l00146"></a>00146 <span class="preprocessor">#endif</span>
<a name="l00147"></a>00147 <span class="preprocessor"></span>        } <span class="keywordflow">else</span> {
<a name="l00148"></a>00148             <span class="comment">// not SSL -&gt; call the handler immediately</span>
<a name="l00149"></a>00149             <a class="code" href="classpion_1_1_t_c_p_server.html#bdf3bb235cf282e635dcd620fe17f448">handleConnection</a>(tcp_conn);
<a name="l00150"></a>00150         }
<a name="l00151"></a>00151     }
<a name="l00152"></a>00152 }
<a name="l00153"></a>00153 
<a name="l00154"></a>00154 <span class="keywordtype">void</span> TCPServer::handleSSLHandshake(<a class="code" href="namespacepion.html#1858b89656b28c6498ad9efd6b7e9d41" title="data type for a TCPConnection pointer">TCPConnectionPtr</a>&amp; tcp_conn,
<a name="l00155"></a>00155                                    <span class="keyword">const</span> boost::system::error_code&amp; handshake_error)
<a name="l00156"></a>00156 {
<a name="l00157"></a>00157     <span class="keywordflow">if</span> (handshake_error) {
<a name="l00158"></a>00158         <span class="comment">// an error occured while trying to establish the SSL connection</span>
<a name="l00159"></a>00159         PION_LOG_WARN(<a class="code" href="classpion_1_1_t_c_p_server.html#028f2f5c2af69741b003b438ce682029" title="primary logging interface used by this class">m_logger</a>, <span class="stringliteral">"SSL handshake failed on port "</span> &lt;&lt; <a class="code" href="classpion_1_1_t_c_p_server.html#ae2c1a58569dfd7342fcf6ccac0746a2" title="returns tcp port number server listens for connections on">getPort</a>()
<a name="l00160"></a>00160                       &lt;&lt; <span class="stringliteral">" ("</span> &lt;&lt; handshake_error.message() &lt;&lt; <span class="charliteral">')'</span>);
<a name="l00161"></a>00161         tcp_conn-&gt;setKeepAlive(<span class="keyword">false</span>);  <span class="comment">// make sure it will get closed</span>
<a name="l00162"></a>00162         finishConnection(tcp_conn);
<a name="l00163"></a>00163     } <span class="keywordflow">else</span> {
<a name="l00164"></a>00164         <span class="comment">// handle the new connection</span>
<a name="l00165"></a>00165         PION_LOG_DEBUG(m_logger, <span class="stringliteral">"SSL handshake succeeded on port "</span> &lt;&lt; <a class="code" href="classpion_1_1_t_c_p_server.html#ae2c1a58569dfd7342fcf6ccac0746a2" title="returns tcp port number server listens for connections on">getPort</a>());
<a name="l00166"></a>00166         <a class="code" href="classpion_1_1_t_c_p_server.html#bdf3bb235cf282e635dcd620fe17f448">handleConnection</a>(tcp_conn);
<a name="l00167"></a>00167     }
<a name="l00168"></a>00168 }
<a name="l00169"></a>00169 
<a name="l00170"></a>00170 <span class="keywordtype">void</span> TCPServer::finishConnection(<a class="code" href="namespacepion.html#1858b89656b28c6498ad9efd6b7e9d41" title="data type for a TCPConnection pointer">TCPConnectionPtr</a>&amp; tcp_conn)
<a name="l00171"></a>00171 {
<a name="l00172"></a>00172     <span class="keywordflow">if</span> (tcp_conn-&gt;getKeepAlive()) {
<a name="l00173"></a>00173         
<a name="l00174"></a>00174         <span class="comment">// keep the connection alive</span>
<a name="l00175"></a>00175         <a class="code" href="classpion_1_1_t_c_p_server.html#bdf3bb235cf282e635dcd620fe17f448">handleConnection</a>(tcp_conn);
<a name="l00176"></a>00176 
<a name="l00177"></a>00177     } <span class="keywordflow">else</span> {
<a name="l00178"></a>00178         PION_LOG_INFO(m_logger, <span class="stringliteral">"Closing connection on port "</span> &lt;&lt; <a class="code" href="classpion_1_1_t_c_p_server.html#ae2c1a58569dfd7342fcf6ccac0746a2" title="returns tcp port number server listens for connections on">getPort</a>());
<a name="l00179"></a>00179         
<a name="l00180"></a>00180         <span class="comment">// remove the connection from the server's management pool</span>
<a name="l00181"></a>00181         boost::mutex::scoped_lock server_lock(m_mutex);
<a name="l00182"></a>00182         ConnectionPool::iterator conn_itr = m_conn_pool.find(tcp_conn);
<a name="l00183"></a>00183         <span class="keywordflow">if</span> (conn_itr != m_conn_pool.end())
<a name="l00184"></a>00184             m_conn_pool.erase(conn_itr);
<a name="l00185"></a>00185         server_lock.unlock();
<a name="l00186"></a>00186     }
<a name="l00187"></a>00187 }
<a name="l00188"></a>00188 
<a name="l00189"></a>00189 }   <span class="comment">// end namespace pion</span>
</pre></div><hr size="1"><address style="text-align: right;"><small>Generated on Tue Jun 19 13:29:22 2007 for libpion by&nbsp;
<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>