socketft_8cpp-source.html

著名的密码库Crypto++的文档 C++语言的杰作。程序员必备。

<a name="l00260"></a>00260 
<a name="l00261"></a>00261         servent *se = getservbyname(name, protocol);
<a name="l00262"></a>00262         <span class="keywordflow">if</span> (!se)
<a name="l00263"></a>00263                 <span class="keywordflow">throw</span> <a class="code" href="class_socket_1_1_err.html" title="exception thrown by Socket class">Err</a>(INVALID_SOCKET, <span class="stringliteral">"getservbyname"</span>, SOCKET_EINVAL);
<a name="l00264"></a>00264         <span class="keywordflow">return</span> ntohs(se-&gt;s_port);
<a name="l00265"></a>00265 }
<a name="l00266"></a>00266 
<a name="l00267"></a><a class="code" href="class_socket.html#653d3bf891dceca75bee878574b98df3">00267</a> <span class="keywordtype">void</span> <a class="code" href="class_socket.html#653d3bf891dceca75bee878574b98df3" title="start Windows Sockets 2">Socket::StartSockets</a>()
<a name="l00268"></a>00268 {
<a name="l00269"></a>00269 <span class="preprocessor">#ifdef USE_WINDOWS_STYLE_SOCKETS</span>
<a name="l00270"></a>00270 <span class="preprocessor"></span>        WSADATA wsd;
<a name="l00271"></a>00271         <span class="keywordtype">int</span> result = WSAStartup(0x0202, &amp;wsd);
<a name="l00272"></a>00272         <span class="keywordflow">if</span> (result != 0)
<a name="l00273"></a>00273                 <span class="keywordflow">throw</span> <a class="code" href="class_socket_1_1_err.html" title="exception thrown by Socket class">Err</a>(INVALID_SOCKET, <span class="stringliteral">"WSAStartup"</span>, result);
<a name="l00274"></a>00274 <span class="preprocessor">#endif</span>
<a name="l00275"></a>00275 <span class="preprocessor"></span>}
<a name="l00276"></a>00276 
<a name="l00277"></a><a class="code" href="class_socket.html#078a98371080e50a230f6a4950e408c0">00277</a> <span class="keywordtype">void</span> <a class="code" href="class_socket.html#078a98371080e50a230f6a4950e408c0" title="calls WSACleanup for Windows Sockets">Socket::ShutdownSockets</a>()
<a name="l00278"></a>00278 {
<a name="l00279"></a>00279 <span class="preprocessor">#ifdef USE_WINDOWS_STYLE_SOCKETS</span>
<a name="l00280"></a>00280 <span class="preprocessor"></span>        <span class="keywordtype">int</span> result = WSACleanup();
<a name="l00281"></a>00281         <span class="keywordflow">if</span> (result != 0)
<a name="l00282"></a>00282                 <span class="keywordflow">throw</span> <a class="code" href="class_socket_1_1_err.html" title="exception thrown by Socket class">Err</a>(INVALID_SOCKET, <span class="stringliteral">"WSACleanup"</span>, result);
<a name="l00283"></a>00283 <span class="preprocessor">#endif</span>
<a name="l00284"></a>00284 <span class="preprocessor"></span>}
<a name="l00285"></a>00285 
<a name="l00286"></a><a class="code" href="class_socket.html#9cf0d25a9da54bbe641dbb89bee452bc">00286</a> <span class="keywordtype">int</span> <a class="code" href="class_socket.html#9cf0d25a9da54bbe641dbb89bee452bc" title="returns errno or WSAGetLastError">Socket::GetLastError</a>()
<a name="l00287"></a>00287 {
<a name="l00288"></a>00288 <span class="preprocessor">#ifdef USE_WINDOWS_STYLE_SOCKETS</span>
<a name="l00289"></a>00289 <span class="preprocessor"></span>        <span class="keywordflow">return</span> WSAGetLastError();
<a name="l00290"></a>00290 <span class="preprocessor">#else</span>
<a name="l00291"></a>00291 <span class="preprocessor"></span>        <span class="keywordflow">return</span> errno;
<a name="l00292"></a>00292 <span class="preprocessor">#endif</span>
<a name="l00293"></a>00293 <span class="preprocessor"></span>}
<a name="l00294"></a>00294 
<a name="l00295"></a><a class="code" href="class_socket.html#8c34f549b69c55bf0ee4c04ea8890bef">00295</a> <span class="keywordtype">void</span> <a class="code" href="class_socket.html#8c34f549b69c55bf0ee4c04ea8890bef" title="sets errno or calls WSASetLastError">Socket::SetLastError</a>(<span class="keywordtype">int</span> errorCode)
<a name="l00296"></a>00296 {
<a name="l00297"></a>00297 <span class="preprocessor">#ifdef USE_WINDOWS_STYLE_SOCKETS</span>
<a name="l00298"></a>00298 <span class="preprocessor"></span>        WSASetLastError(errorCode);
<a name="l00299"></a>00299 <span class="preprocessor">#else</span>
<a name="l00300"></a>00300 <span class="preprocessor"></span>        errno = errorCode;
<a name="l00301"></a>00301 <span class="preprocessor">#endif</span>
<a name="l00302"></a>00302 <span class="preprocessor"></span>}
<a name="l00303"></a>00303 
<a name="l00304"></a><a class="code" href="class_socket.html#a2ef5204ea0d284b6ef003b06c3e6a2d">00304</a> <span class="keywordtype">void</span> <a class="code" href="class_socket.html#a2ef5204ea0d284b6ef003b06c3e6a2d">Socket::HandleError</a>(<span class="keyword">const</span> <span class="keywordtype">char</span> *operation)<span class="keyword"> const</span>
<a name="l00305"></a>00305 <span class="keyword"></span>{
<a name="l00306"></a>00306         <span class="keywordtype">int</span> err = <a class="code" href="class_socket.html#9cf0d25a9da54bbe641dbb89bee452bc" title="returns errno or WSAGetLastError">GetLastError</a>();
<a name="l00307"></a>00307         <span class="keywordflow">throw</span> <a class="code" href="class_socket_1_1_err.html" title="exception thrown by Socket class">Err</a>(<a class="code" href="class_socket.html#2a82c6dd3794c4d45e33e68f2112df35">m_s</a>, operation, err);
<a name="l00308"></a>00308 }
<a name="l00309"></a>00309 
<a name="l00310"></a>00310 <span class="preprocessor">#ifdef USE_WINDOWS_STYLE_SOCKETS</span>
<a name="l00311"></a>00311 <span class="preprocessor"></span>
<a name="l00312"></a><a class="code" href="class_socket_receiver.html#05f59443fbdb49fe8696aa2fae78c9b2">00312</a> <a class="code" href="class_socket_receiver.html#05f59443fbdb49fe8696aa2fae78c9b2">SocketReceiver::SocketReceiver</a>(<a class="code" href="class_socket.html" title="wrapper for Windows or Berkeley Sockets">Socket</a> &amp;s)
<a name="l00313"></a>00313         : <a class="code" href="class_socket.html#2a82c6dd3794c4d45e33e68f2112df35">m_s</a>(s), m_resultPending(false), m_eofReceived(false)
<a name="l00314"></a>00314 {
<a name="l00315"></a>00315         m_event.<a class="code" href="class_windows_handle.html#6f1053e346d9c41fe13df20c59624feb">AttachHandle</a>(CreateEvent(NULL, <span class="keyword">true</span>, <span class="keyword">false</span>, NULL), <span class="keyword">true</span>);
<a name="l00316"></a>00316         m_s.<a class="code" href="class_socket.html#c29e3d5e0acd2d8d6c7fd59a7cc4bd40">CheckAndHandleError</a>(<span class="stringliteral">"CreateEvent"</span>, m_event.<a class="code" href="class_windows_handle.html#8176ee81240cc1add7184528fde31d91">HandleValid</a>());
<a name="l00317"></a>00317         memset(&amp;m_overlapped, 0, <span class="keyword">sizeof</span>(m_overlapped));
<a name="l00318"></a>00318         m_overlapped.hEvent = m_event;
<a name="l00319"></a>00319 }
<a name="l00320"></a>00320 
<a name="l00321"></a><a class="code" href="class_socket_receiver.html#3a58d69fa4b0afbda1184ad3cdfd3b0c">00321</a> <a class="code" href="class_socket_receiver.html#3a58d69fa4b0afbda1184ad3cdfd3b0c">SocketReceiver::~SocketReceiver</a>()
<a name="l00322"></a>00322 {
<a name="l00323"></a>00323 <span class="preprocessor">#ifdef USE_WINDOWS_STYLE_SOCKETS</span>
<a name="l00324"></a>00324 <span class="preprocessor"></span>        CancelIo((HANDLE) m_s.<a class="code" href="class_socket.html#fd9c1f7ef23c16a2c4290e1a0fccb142">GetSocket</a>());
<a name="l00325"></a>00325 <span class="preprocessor">#endif</span>
<a name="l00326"></a>00326 <span class="preprocessor"></span>}
<a name="l00327"></a>00327 
<a name="l00328"></a><a class="code" href="class_socket_receiver.html#31e0ccf504d9c16760051f10d749c034">00328</a> <span class="keywordtype">bool</span> <a class="code" href="class_socket_receiver.html#31e0ccf504d9c16760051f10d749c034" title="receive data from network source, returns whether result is immediately available...">SocketReceiver::Receive</a>(byte* buf, <span class="keywordtype">size_t</span> bufLen)
<a name="l00329"></a>00329 {
<a name="l00330"></a>00330         assert(!m_resultPending &amp;&amp; !m_eofReceived);
<a name="l00331"></a>00331 
<a name="l00332"></a>00332         DWORD flags = 0;
<a name="l00333"></a>00333         <span class="comment">// don't queue too much at once, or we might use up non-paged memory</span>
<a name="l00334"></a>00334         WSABUF wsabuf = {UnsignedMin((u_long)128*1024, bufLen), (<span class="keywordtype">char</span> *)buf};
<a name="l00335"></a>00335         <span class="keywordflow">if</span> (WSARecv(m_s, &amp;wsabuf, 1, &amp;m_lastResult, &amp;flags, &amp;m_overlapped, NULL) == 0)
<a name="l00336"></a>00336         {
<a name="l00337"></a>00337                 <span class="keywordflow">if</span> (m_lastResult == 0)
<a name="l00338"></a>00338                         m_eofReceived = <span class="keyword">true</span>;
<a name="l00339"></a>00339         }
<a name="l00340"></a>00340         <span class="keywordflow">else</span>
<a name="l00341"></a>00341         {
<a name="l00342"></a>00342                 <span class="keywordflow">switch</span> (WSAGetLastError())
<a name="l00343"></a>00343                 {
<a name="l00344"></a>00344                 <span class="keywordflow">default</span>:
<a name="l00345"></a>00345                         m_s.CheckAndHandleError_int(<span class="stringliteral">"WSARecv"</span>, SOCKET_ERROR);
<a name="l00346"></a>00346                 <span class="keywordflow">case</span> WSAEDISCON:
<a name="l00347"></a>00347                         m_lastResult = 0;
<a name="l00348"></a>00348                         m_eofReceived = <span class="keyword">true</span>;
<a name="l00349"></a>00349                         <span class="keywordflow">break</span>;
<a name="l00350"></a>00350                 <span class="keywordflow">case</span> WSA_IO_PENDING:
<a name="l00351"></a>00351                         m_resultPending = <span class="keyword">true</span>;
<a name="l00352"></a>00352                 }
<a name="l00353"></a>00353         }
<a name="l00354"></a>00354         <span class="keywordflow">return</span> !m_resultPending;
<a name="l00355"></a>00355 }
<a name="l00356"></a>00356 
<a name="l00357"></a><a class="code" href="class_socket_receiver.html#a851a6020843d4311036a6f4299f9a6b">00357</a> <span class="keywordtype">void</span> <a class="code" href="class_socket_receiver.html#a851a6020843d4311036a6f4299f9a6b" title="put wait objects into container">SocketReceiver::GetWaitObjects</a>(<a class="code" href="class_wait_object_container.html" title="container of wait objects">WaitObjectContainer</a> &amp;container, <a class="code" href="class_call_stack.html">CallStack</a> <span class="keyword">const</span>&amp; callStack)
<a name="l00358"></a>00358 {
<a name="l00359"></a>00359         <span class="keywordflow">if</span> (m_resultPending)
<a name="l00360"></a>00360                 container.<a class="code" href="class_wait_object_container.html#9ce9ba96948923609861455027da7830">AddHandle</a>(m_event, <a class="code" href="class_call_stack.html">CallStack</a>(<span class="stringliteral">"SocketReceiver::GetWaitObjects() - result pending"</span>, &amp;callStack));
<a name="l00361"></a>00361         <span class="keywordflow">else</span> <span class="keywordflow">if</span> (!m_eofReceived)
<a name="l00362"></a>00362                 container.<a class="code" href="class_wait_object_container.html#ce9c529ebb1956ee2cb9121bdf30402d">SetNoWait</a>(<a class="code" href="class_call_stack.html">CallStack</a>(<span class="stringliteral">"SocketReceiver::GetWaitObjects() - result ready"</span>, &amp;callStack));
<a name="l00363"></a>00363 }
<a name="l00364"></a>00364 
<a name="l00365"></a><a class="code" href="class_socket_receiver.html#428a62118c6e2ca975a3d4560668b3b2">00365</a> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> <a class="code" href="class_socket_receiver.html#428a62118c6e2ca975a3d4560668b3b2">SocketReceiver::GetReceiveResult</a>()
<a name="l00366"></a>00366 {
<a name="l00367"></a>00367         <span class="keywordflow">if</span> (m_resultPending)
<a name="l00368"></a>00368         {
<a name="l00369"></a>00369                 DWORD flags = 0;
<a name="l00370"></a>00370                 <span class="keywordflow">if</span> (WSAGetOverlappedResult(m_s, &amp;m_overlapped, &amp;m_lastResult, <span class="keyword">false</span>, &amp;flags))
<a name="l00371"></a>00371                 {
<a name="l00372"></a>00372                         <span class="keywordflow">if</span> (m_lastResult == 0)
<a name="l00373"></a>00373                                 m_eofReceived = <span class="keyword">true</span>;
<a name="l00374"></a>00374                 }
<a name="l00375"></a>00375                 <span class="keywordflow">else</span>
<a name="l00376"></a>00376                 {
<a name="l00377"></a>00377                         <span class="keywordflow">switch</span> (WSAGetLastError())
<a name="l00378"></a>00378                         {
<a name="l00379"></a>00379                         <span class="keywordflow">default</span>:
<a name="l00380"></a>00380                                 m_s.<a class="code" href="class_socket.html#c29e3d5e0acd2d8d6c7fd59a7cc4bd40">CheckAndHandleError</a>(<span class="stringliteral">"WSAGetOverlappedResult"</span>, FALSE);
<a name="l00381"></a>00381                         <span class="keywordflow">case</span> WSAEDISCON:
<a name="l00382"></a>00382                                 m_lastResult = 0;
<a name="l00383"></a>00383                                 m_eofReceived = <span class="keyword">true</span>;
<a name="l00384"></a>00384                         }
<a name="l00385"></a>00385                 }
<a name="l00386"></a>00386                 m_resultPending = <span class="keyword">false</span>;
<a name="l00387"></a>00387         }
<a name="l00388"></a>00388         <span class="keywordflow">return</span> m_lastResult;
<a name="l00389"></a>00389 }
<a name="l00390"></a>00390 
<a name="l00391"></a>00391 <span class="comment">// *************************************************************</span>
<a name="l00392"></a>00392 
<a name="l00393"></a><a class="code" href="class_socket_sender.html#c85881f31c51b8d7efe88c06c6954421">00393</a> <a class="code" href="class_socket_sender.html#c85881f31c51b8d7efe88c06c6954421">SocketSender::SocketSender</a>(<a class="code" href="class_socket.html" title="wrapper for Windows or Berkeley Sockets">Socket</a> &amp;s)
<a name="l00394"></a>00394         : m_s(s), m_resultPending(false), m_lastResult(0)
<a name="l00395"></a>00395 {
<a name="l00396"></a>00396         m_event.<a class="code" href="class_windows_handle.html#6f1053e346d9c41fe13df20c59624feb">AttachHandle</a>(CreateEvent(NULL, <span class="keyword">true</span>, <span class="keyword">false</span>, NULL), <span class="keyword">true</span>);
<a name="l00397"></a>00397         m_s.<a class="code" href="class_socket.html#c29e3d5e0acd2d8d6c7fd59a7cc4bd40">CheckAndHandleError</a>(<span class="stringliteral">"CreateEvent"</span>, m_event.<a class="code" href="class_windows_handle.html#8176ee81240cc1add7184528fde31d91">HandleValid</a>());
<a name="l00398"></a>00398         memset(&amp;m_overlapped, 0, <span class="keyword">sizeof</span>(m_overlapped));
<a name="l00399"></a>00399         m_overlapped.hEvent = m_event;