<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> <html xmlns="http://www.w3.org/1999/xhtml"> <head> <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> <title>Crypto++: socketft.cpp Source File</title> <link href="tabs.css" rel="stylesheet" type="text/css"/> <link href="doxygen.css" rel="stylesheet" type="text/css"/> </head> <body> <!-- Generated by Doxygen 1.6.1 --> <div class="navigation" id="top"> <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> <div class="tabs"> <ul> <li><a href="files.html"><span>File List</span></a></li> <li><a href="globals.html"><span>File Members</span></a></li> </ul> </div> <h1>socketft.cpp</h1><div class="fragment"><pre class="fragment"><a name="l00001"></a>00001 <span class="comment">// socketft.cpp - written and placed in the public domain by Wei Dai</span> <a name="l00002"></a>00002 <a name="l00003"></a>00003 <span class="preprocessor">#include "pch.h"</span> <a name="l00004"></a>00004 <span class="preprocessor">#include "socketft.h"</span> <a name="l00005"></a>00005 <a name="l00006"></a>00006 <span class="preprocessor">#ifdef SOCKETS_AVAILABLE</span> <a name="l00007"></a>00007 <span class="preprocessor"></span> <a name="l00008"></a>00008 <span class="preprocessor">#include "wait.h"</span> <a name="l00009"></a>00009 <a name="l00010"></a>00010 <span class="preprocessor">#ifdef USE_BERKELEY_STYLE_SOCKETS</span> <a name="l00011"></a>00011 <span class="preprocessor"></span><span class="preprocessor">#include <errno.h></span> <a name="l00012"></a>00012 <span class="preprocessor">#include <netdb.h></span> <a name="l00013"></a>00013 <span class="preprocessor">#include <unistd.h></span> <a name="l00014"></a>00014 <span class="preprocessor">#include <arpa/inet.h></span> <a name="l00015"></a>00015 <span class="preprocessor">#include <netinet/in.h></span> <a name="l00016"></a>00016 <span class="preprocessor">#include <sys/ioctl.h></span> <a name="l00017"></a>00017 <span class="preprocessor">#endif</span> <a name="l00018"></a>00018 <span class="preprocessor"></span> <a name="l00019"></a>00019 NAMESPACE_BEGIN(CryptoPP) <a name="l00020"></a>00020 <a name="l00021"></a>00021 <span class="preprocessor">#ifdef USE_WINDOWS_STYLE_SOCKETS</span> <a name="l00022"></a>00022 <span class="preprocessor"></span><span class="keyword">const</span> <span class="keywordtype">int</span> SOCKET_EINVAL = WSAEINVAL; <a name="l00023"></a>00023 <span class="keyword">const</span> <span class="keywordtype">int</span> SOCKET_EWOULDBLOCK = WSAEWOULDBLOCK; <a name="l00024"></a>00024 <span class="keyword">typedef</span> <span class="keywordtype">int</span> socklen_t; <a name="l00025"></a>00025 <span class="preprocessor">#else</span> <a name="l00026"></a>00026 <span class="preprocessor"></span><span class="keyword">const</span> <span class="keywordtype">int</span> SOCKET_EINVAL = EINVAL; <a name="l00027"></a>00027 <span class="keyword">const</span> <span class="keywordtype">int</span> SOCKET_EWOULDBLOCK = EWOULDBLOCK; <a name="l00028"></a>00028 <span class="preprocessor">#endif</span> <a name="l00029"></a>00029 <span class="preprocessor"></span> <a name="l00030"></a>00030 Socket::Err::Err(socket_t s, <span class="keyword">const</span> std::string& operation, <span class="keywordtype">int</span> error) <a name="l00031"></a>00031 : <a class="code" href="class_o_s___error.html" title="error reported by the operating system">OS_Error</a>(IO_ERROR, <span class="stringliteral">"Socket: "</span> + operation + <span class="stringliteral">" operation failed with error "</span> + IntToString(error), operation, error) <a name="l00032"></a>00032 , m_s(s) <a name="l00033"></a>00033 { <a name="l00034"></a>00034 } <a name="l00035"></a>00035 <a name="l00036"></a>00036 Socket::~Socket() <a name="l00037"></a>00037 { <a name="l00038"></a>00038 <span class="keywordflow">if</span> (m_own) <a name="l00039"></a>00039 { <a name="l00040"></a>00040 <span class="keywordflow">try</span> <a name="l00041"></a>00041 { <a name="l00042"></a>00042 CloseSocket(); <a name="l00043"></a>00043 } <a name="l00044"></a>00044 <span class="keywordflow">catch</span> (...) <a name="l00045"></a>00045 { <a name="l00046"></a>00046 } <a name="l00047"></a>00047 } <a name="l00048"></a>00048 } <a name="l00049"></a>00049 <a name="l00050"></a>00050 <span class="keywordtype">void</span> Socket::AttachSocket(socket_t s, <span class="keywordtype">bool</span> own) <a name="l00051"></a>00051 { <a name="l00052"></a>00052 <span class="keywordflow">if</span> (m_own) <a name="l00053"></a>00053 CloseSocket(); <a name="l00054"></a>00054 <a name="l00055"></a>00055 m_s = s; <a name="l00056"></a>00056 m_own = own; <a name="l00057"></a>00057 SocketChanged(); <a name="l00058"></a>00058 } <a name="l00059"></a>00059 <a name="l00060"></a>00060 socket_t Socket::DetachSocket() <a name="l00061"></a>00061 { <a name="l00062"></a>00062 socket_t s = m_s; <a name="l00063"></a>00063 m_s = INVALID_SOCKET; <a name="l00064"></a>00064 SocketChanged(); <a name="l00065"></a>00065 <span class="keywordflow">return</span> s; <a name="l00066"></a>00066 } <a name="l00067"></a>00067 <a name="l00068"></a>00068 <span class="keywordtype">void</span> Socket::Create(<span class="keywordtype">int</span> nType) <a name="l00069"></a>00069 { <a name="l00070"></a>00070 assert(m_s == INVALID_SOCKET); <a name="l00071"></a>00071 m_s = socket(AF_INET, nType, 0); <a name="l00072"></a>00072 CheckAndHandleError(<span class="stringliteral">"socket"</span>, m_s); <a name="l00073"></a>00073 m_own = <span class="keyword">true</span>; <a name="l00074"></a>00074 SocketChanged(); <a name="l00075"></a>00075 } <a name="l00076"></a>00076 <a name="l00077"></a>00077 <span class="keywordtype">void</span> Socket::CloseSocket() <a name="l00078"></a>00078 { <a name="l00079"></a>00079 <span class="keywordflow">if</span> (m_s != INVALID_SOCKET) <a name="l00080"></a>00080 { <a name="l00081"></a>00081 <span class="preprocessor">#ifdef USE_WINDOWS_STYLE_SOCKETS</span> <a name="l00082"></a>00082 <span class="preprocessor"></span> CancelIo((HANDLE) m_s); <a name="l00083"></a>00083 CheckAndHandleError_int(<span class="stringliteral">"closesocket"</span>, closesocket(m_s)); <a name="l00084"></a>00084 <span class="preprocessor">#else</span> <a name="l00085"></a>00085 <span class="preprocessor"></span> CheckAndHandleError_int(<span class="stringliteral">"close"</span>, close(m_s)); <a name="l00086"></a>00086 <span class="preprocessor">#endif</span> <a name="l00087"></a>00087 <span class="preprocessor"></span> m_s = INVALID_SOCKET; <a name="l00088"></a>00088 SocketChanged(); <a name="l00089"></a>00089 } <a name="l00090"></a>00090 } <a name="l00091"></a>00091 <a name="l00092"></a>00092 <span class="keywordtype">void</span> Socket::Bind(<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> port, <span class="keyword">const</span> <span class="keywordtype">char</span> *addr) <a name="l00093"></a>00093 { <a name="l00094"></a>00094 sockaddr_in sa; <a name="l00095"></a>00095 memset(&sa, 0, <span class="keyword">sizeof</span>(sa)); <a name="l00096"></a>00096 sa.sin_family = AF_INET; <a name="l00097"></a>00097 <a name="l00098"></a>00098 <span class="keywordflow">if</span> (addr == NULL) <a name="l00099"></a>00099 sa.sin_addr.s_addr = htonl(INADDR_ANY); <a name="l00100"></a>00100 <span class="keywordflow">else</span> <a name="l00101"></a>00101 { <a name="l00102"></a>00102 <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> result = inet_addr(addr); <a name="l00103"></a>00103 <span class="keywordflow">if</span> (result == -1) <span class="comment">// Solaris doesn't have INADDR_NONE</span> <a name="l00104"></a>00104 { <a name="l00105"></a>00105 <a class="code" href="class_socket.html#a8c34f549b69c55bf0ee4c04ea8890bef" title="sets errno or calls WSASetLastError">SetLastError</a>(SOCKET_EINVAL); <a name="l00106"></a>00106 CheckAndHandleError_int(<span class="stringliteral">"inet_addr"</span>, SOCKET_ERROR); <a name="l00107"></a>00107 } <a name="l00108"></a>00108 sa.sin_addr.s_addr = result; <a name="l00109"></a>00109 } <a name="l00110"></a>00110 <a name="l00111"></a>00111 sa.sin_port = htons((u_short)port); <a name="l00112"></a>00112 <a name="l00113"></a>00113 Bind((sockaddr *)&sa, <span class="keyword">sizeof</span>(sa)); <a name="l00114"></a>00114 } <a name="l00115"></a>00115 <a name="l00116"></a>00116 <span class="keywordtype">void</span> Socket::Bind(<span class="keyword">const</span> sockaddr *psa, socklen_t saLen) <a name="l00117"></a>00117 { <a name="l00118"></a>00118 assert(m_s != INVALID_SOCKET); <a name="l00119"></a>00119 <span class="comment">// cygwin workaround: needs const_cast</span> <a name="l00120"></a>00120 CheckAndHandleError_int(<span class="stringliteral">"bind"</span>, bind(m_s, const_cast<sockaddr *>(psa), saLen)); <a name="l00121"></a>00121 } <a name="l00122"></a>00122 <a name="l00123"></a>00123 <span class="keywordtype">void</span> Socket::Listen(<span class="keywordtype">int</span> backlog) <a name="l00124"></a>00124 { <a name="l00125"></a>00125 assert(m_s != INVALID_SOCKET); <a name="l00126"></a>00126 CheckAndHandleError_int(<span class="stringliteral">"listen"</span>, listen(m_s, backlog)); <a name="l00127"></a>00127 } <a name="l00128"></a>00128 <a name="l00129"></a>00129 <span class="keywordtype">bool</span> Socket::Connect(<span class="keyword">const</span> <span class="keywordtype">char</span> *addr, <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> port) <a name="l00130"></a>00130 { <a name="l00131"></a>00131 assert(addr != NULL); <a name="l00132"></a>00132 <a name="l00133"></a>00133 sockaddr_in sa; <a name="l00134"></a>00134 memset(&sa, 0, <span class="keyword">sizeof</span>(sa)); <a name="l00135"></a>00135 sa.sin_family = AF_INET; <a name="l00136"></a>00136 sa.sin_addr.s_addr = inet_addr(addr); <a name="l00137"></a>00137 <a name="l00138"></a>00138 <span class="keywordflow">if</span> (sa.sin_addr.s_addr == -1) <span class="comment">// Solaris doesn't have INADDR_NONE</span> <a name="l00139"></a>00139 { <a name="l00140"></a>00140 hostent *lphost = gethostbyname(addr); <a name="l00141"></a>00141 <span class="keywordflow">if</span> (lphost == NULL) <a name="l00142"></a>00142 { <a name="l00143"></a>00143 <a class="code" href="class_socket.html#a8c34f549b69c55bf0ee4c04ea8890bef" title="sets errno or calls WSASetLastError">SetLastError</a>(SOCKET_EINVAL); <a name="l00144"></a>00144 CheckAndHandleError_int(<span class="stringliteral">"gethostbyname"</span>, SOCKET_ERROR); <a name="l00145"></a>00145 } <a name="l00146"></a>00146 <a name="l00147"></a>00147 sa.sin_addr.s_addr = ((in_addr *)lphost->h_addr)->s_addr; <a name="l00148"></a>00148 } <a name="l00149"></a>00149 <a name="l00150"></a>00150 sa.sin_port = htons((u_short)port); <a name="l00151"></a>00151 <a name="l00152"></a>00152 <span class="keywordflow">return</span> Connect((<span class="keyword">const</span> sockaddr *)&sa, <span class="keyword">sizeof</span>(sa)); <a name="l00153"></a>00153 } <a name="l00154"></a>00154 <a name="l00155"></a>00155 <span class="keywordtype">bool</span> Socket::Connect(<span class="keyword">const</span> sockaddr* psa, socklen_t saLen) <a name="l00156"></a>00156 { <a name="l00157"></a>00157 assert(m_s != INVALID_SOCKET); <a name="l00158"></a>00158 <span class="keywordtype">int</span> result = connect(m_s, const_cast<sockaddr*>(psa), saLen); <a name="l00159"></a>00159 <span class="keywordflow">if</span> (result == SOCKET_ERROR && <a class="code" href="class_socket.html#a9cf0d25a9da54bbe641dbb89bee452bc" title="returns errno or WSAGetLastError">GetLastError</a>() == SOCKET_EWOULDBLOCK) <a name="l00160"></a>00160 <span class="keywordflow">return</span> <span class="keyword">false</span>; <a name="l00161"></a>00161 CheckAndHandleError_int(<span class="stringliteral">"connect"</span>, result); <a name="l00162"></a>00162 <span class="keywordflow">return</span> <span class="keyword">true</span>; <a name="l00163"></a>00163 } <a name="l00164"></a>00164 <a name="l00165"></a>00165 <span class="keywordtype">bool</span> Socket::Accept(<a class="code" href="class_socket.html" title="wrapper for Windows or Berkeley Sockets">Socket</a>& target, sockaddr *psa, socklen_t *psaLen) <a name="l00166"></a>00166 { <a name="l00167"></a>00167 assert(m_s != INVALID_SOCKET); <a name="l00168"></a>00168 socket_t s = accept(m_s, psa, psaLen); <a name="l00169"></a>00169 <span class="keywordflow">if</span> (s == INVALID_SOCKET && <a class="code" href="class_socket.html#a9cf0d25a9da54bbe641dbb89bee452bc" title="returns errno or WSAGetLastError">GetLastError</a>() == SOCKET_EWOULDBLOCK) <a name="l00170"></a>00170 <span class="keywordflow">return</span> <span class="keyword">false</span>; <a name="l00171"></a>00171 CheckAndHandleError(<span class="stringliteral">"accept"</span>, s); <a name="l00172"></a>00172 target.AttachSocket(s, <span class="keyword">true</span>); <a name="l00173"></a>00173 <span class="keywordflow">return</span> <span class="keyword">true</span>; <a name="l00174"></a>00174 } <a name="l00175"></a>00175 <a name="l00176"></a>00176 <span class="keywordtype">void</span> Socket::GetSockName(sockaddr *psa, socklen_t *psaLen) <a name="l00177"></a>00177 { <a name="l00178"></a>00178 assert(m_s != INVALID_SOCKET); <a name="l00179"></a>00179 CheckAndHandleError_int(<span class="stringliteral">"getsockname"</span>, getsockname(m_s, psa, psaLen)); <a name="l00180"></a>00180 } <a name="l00181"></a>00181 <a name="l00182"></a>00182 <span class="keywordtype">void</span> Socket::GetPeerName(sockaddr *psa, socklen_t *psaLen) <a name="l00183"></a>00183 { <a name="l00184"></a>00184 assert(m_s != INVALID_SOCKET); <a name="l00185"></a>00185 CheckAndHandleError_int(<span class="stringliteral">"getpeername"</span>, getpeername(m_s, psa, psaLen)); <a name="l00186"></a>00186 } <a name="l00187"></a>00187 <a name="l00188"></a>00188 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> Socket::Send(<span class="keyword">const</span> byte* buf, <span class="keywordtype">size_t</span> bufLen, <span class="keywordtype">int</span> flags) <a name="l00189"></a>00189 { <a name="l00190"></a>00190 assert(m_s != INVALID_SOCKET); <a name="l00191"></a>00191 <span class="keywordtype">int</span> result = send(m_s, (<span class="keyword">const</span> <span class="keywordtype">char</span> *)buf, UnsignedMin(INT_MAX, bufLen), flags); <a name="l00192"></a>00192 CheckAndHandleError_int(<span class="stringliteral">"send"</span>, result); <a name="l00193"></a>00193 <span class="keywordflow">return</span> result; <a name="l00194"></a>00194 } <a name="l00195"></a>00195 <a name="l00196"></a>00196 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> Socket::Receive(byte* buf, <span class="keywordtype">size_t</span> bufLen, <span class="keywordtype">int</span> flags) <a name="l00197"></a>00197 { <a name="l00198"></a>00198 assert(m_s != INVALID_SOCKET); <a name="l00199"></a>00199 <span class="keywordtype">int</span> result = recv(m_s, (<span class="keywordtype">char</span> *)buf, UnsignedMin(INT_MAX, bufLen), flags); <a name="l00200"></a>00200 CheckAndHandleError_int(<span class="stringliteral">"recv"</span>, result); <a name="l00201"></a>00201 <span class="keywordflow">return</span> result; <a name="l00202"></a>00202 } <a name="l00203"></a>00203 <a name="l00204"></a>00204 <span class="keywordtype">void</span> Socket::ShutDown(<span class="keywordtype">int</span> how) <a name="l00205"></a>00205 { <a name="l00206"></a>00206 assert(m_s != INVALID_SOCKET); <a name="l00207"></a>00207 <span class="keywordtype">int</span> result = shutdown(m_s, how); <a name="l00208"></a>00208 CheckAndHandleError_int(<span class="stringliteral">"shutdown"</span>, result); <a name="l00209"></a>00209 } <a name="l00210"></a>00210 <a name="l00211"></a>00211 <span class="keywordtype">void</span> Socket::IOCtl(<span class="keywordtype">long</span> cmd, <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> *argp) <a name="l00212"></a>00212 { <a name="l00213"></a>00213 assert(m_s != INVALID_SOCKET); <a name="l00214"></a>00214 <span class="preprocessor">#ifdef USE_WINDOWS_STYLE_SOCKETS</span> <a name="l00215"></a>00215 <span class="preprocessor"></span> CheckAndHandleError_int(<span class="stringliteral">"ioctlsocket"</span>, ioctlsocket(m_s, cmd, argp)); <a name="l00216"></a>00216 <span class="preprocessor">#else</span> <a name="l00217"></a>00217 <span class="preprocessor"></span> CheckAndHandleError_int(<span class="stringliteral">"ioctl"</span>, ioctl(m_s, cmd, argp)); <a name="l00218"></a>00218 <span class="preprocessor">#endif</span> <a name="l00219"></a>00219 <span class="preprocessor"></span>} <a name="l00220"></a>00220 <a name="l00221"></a>00221 <span class="keywordtype">bool</span> Socket::SendReady(<span class="keyword">const</span> timeval *timeout) <a name="l00222"></a>00222 { <a name="l00223"></a>00223 fd_set fds; <a name="l00224"></a>00224 FD_ZERO(&fds); <a name="l00225"></a>00225 FD_SET(m_s, &fds); <a name="l00226"></a>00226 <span class="keywordtype">int</span> ready; <a name="l00227"></a>00227 <span class="keywordflow">if</span> (timeout == NULL) <a name="l00228"></a>00228 ready = select((<span class="keywordtype">int</span>)m_s+1, NULL, &fds, NULL, NULL); <a name="l00229"></a>00229 <span class="keywordflow">else</span> <a name="l00230"></a>00230 { <a name="l00231"></a>00231 timeval timeoutCopy = *timeout; <span class="comment">// select() modified timeout on Linux</span> <a name="l00232"></a>00232 ready = select((<span class="keywordtype">int</span>)m_s+1, NULL, &fds, NULL, &timeoutCopy); <a name="l00233"></a>00233 } <a name="l00234"></a>00234 CheckAndHandleError_int(<span class="stringliteral">"select"</span>, ready); <a name="l00235"></a>00235 <span class="keywordflow">return</span> ready > 0; <a name="l00236"></a>00236 } <a name="l00237"></a>00237 <a name="l00238"></a>00238 <span class="keywordtype">bool</span> Socket::ReceiveReady(<span class="keyword">const</span> timeval *timeout) <a name="l00239"></a>00239 { <a name="l00240"></a>00240 fd_set fds; <a name="l00241"></a>00241 FD_ZERO(&fds); <a name="l00242"></a>00242 FD_SET(m_s, &fds); <a name="l00243"></a>00243 <span class="keywordtype">int</span> ready; <a name="l00244"></a>00244 <span class="keywordflow">if</span> (timeout == NULL) <a name="l00245"></a>00245 ready = select((<span class="keywordtype">int</span>)m_s+1, &fds, NULL, NULL, NULL); <a name="l00246"></a>00246 <span class="keywordflow">else</span> <a name="l00247"></a>00247 { <a name="l00248"></a>00248 timeval timeoutCopy = *timeout; <span class="comment">// select() modified timeout on Linux</span> <a name="l00249"></a>00249 ready = select((<span class="keywordtype">int</span>)m_s+1, &fds, NULL, NULL, &timeoutCopy); <a name="l00250"></a>00250 } <a name="l00251"></a>00251 CheckAndHandleError_int(<span class="stringliteral">"select"</span>, ready); <a name="l00252"></a>00252 <span class="keywordflow">return</span> ready > 0; <a name="l00253"></a>00253 } <a name="l00254"></a>00254 <a name="l00255"></a><a class="code" href="class_socket.html#a6417cdd4aec086457b67eb34d0964f57">00255</a> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> <a class="code" href="class_socket.html#a6417cdd4aec086457b67eb34d0964f57" title="look up the port number given its name, returns 0 if not found">Socket::PortNameToNumber</a>(<span class="keyword">const</span> <span class="keywordtype">char</span> *name, <span class="keyword">const</span> <span class="keywordtype">char</span> *protocol) <a name="l00256"></a>00256 { <a name="l00257"></a>00257 <span class="keywordtype">int</span> port = atoi(name); <a name="l00258"></a>00258 <span class="keywordflow">if</span> (IntToString(port) == name) <a name="l00259"></a>00259 <span class="keywordflow">return</span> port; <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->s_port); <a name="l00265"></a>00265 } <a name="l00266"></a>00266 <a name="l00267"></a><a class="code" href="class_socket.html#a653d3bf891dceca75bee878574b98df3">00267</a> <span class="keywordtype">void</span> <a class="code" href="class_socket.html#a653d3bf891dceca75bee878574b98df3" 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, &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#a078a98371080e50a230f6a4950e408c0">00277</a> <span class="keywordtype">void</span> <a class="code" href="class_socket.html#a078a98371080e50a230f6a4950e408c0" 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#a9cf0d25a9da54bbe641dbb89bee452bc">00286</a> <span class="keywordtype">int</span> <a class="code" href="class_socket.html#a9cf0d25a9da54bbe641dbb89bee452bc" 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#a8c34f549b69c55bf0ee4c04ea8890bef">00295</a> <span class="keywordtype">void</span> <a class="code" href="class_socket.html#a8c34f549b69c55bf0ee4c04ea8890bef" 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>00304 <span class="keywordtype">void</span> Socket::HandleError(<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#a9cf0d25a9da54bbe641dbb89bee452bc" title="returns errno or WSAGetLastError">GetLastError</a>(); <a name="l00307"></a>00307 <span class="keywordflow">throw</span> Err(m_s, 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>00312 SocketReceiver::SocketReceiver(<a class="code" href="class_socket.html" title="wrapper for Windows or Berkeley Sockets">Socket</a> &s) <a name="l00313"></a>00313 : m_s(s), m_resultPending(false), m_eofReceived(false) <a name="l00314"></a>00314 { <a name="l00315"></a>00315 m_event.AttachHandle(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.CheckAndHandleError(<span class="stringliteral">"CreateEvent"</span>, m_event.HandleValid()); <a name="l00317"></a>00317 memset(&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>00321 SocketReceiver::~SocketReceiver() <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.GetSocket()); <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#a31e0ccf504d9c16760051f10d749c034">00328</a> <span class="keywordtype">bool</span> <a class="code" href="class_socket_receiver.html#a31e0ccf504d9c16760051f10d749c034" 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 && !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, &wsabuf, 1, &m_lastResult, &flags, &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#aa851a6020843d4311036a6f4299f9a6b">00357</a> <span class="keywordtype">void</span> <a class="code" href="class_socket_receiver.html#aa851a6020843d4311036a6f4299f9a6b" 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> &container, <a class="code" href="class_call_stack.html">CallStack</a> <span class="keyword">const</span>& callStack) <a name="l00358"></a>00358 { <a name="l00359"></a>00359 <span class="keywordflow">if</span> (m_resultPending) <a name="l00360"></a>00360 container.AddHandle(m_event, <a class="code" href="class_call_stack.html">CallStack</a>(<span class="stringliteral">"SocketReceiver::GetWaitObjects() - result pending"</span>, &callStack)); <a name="l00361"></a>00361 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (!m_eofReceived) <a name="l00362"></a>00362 container.SetNoWait(<a class="code" href="class_call_stack.html">CallStack</a>(<span class="stringliteral">"SocketReceiver::GetWaitObjects() - result ready"</span>, &callStack)); <a name="l00363"></a>00363 } <a name="l00364"></a>00364 <a name="l00365"></a>00365 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> SocketReceiver::GetReceiveResult() <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, &m_overlapped, &m_lastResult, <span class="keyword">false</span>, &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.CheckAndHandleError(<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>00393 SocketSender::SocketSender(<a class="code" href="class_socket.html" title="wrapper for Windows or Berkeley Sockets">Socket</a> &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.AttachHandle(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.CheckAndHandleError(<span class="stringliteral">"CreateEvent"</span>, m_event.HandleValid()); <a name="l00398"></a>00398 memset(&m_overlapped, 0, <span class="keyword">sizeof</span>(m_overlapped)); <a name="l00399"></a>00399 m_overlapped.hEvent = m_event; <a name="l00400"></a>00400 } <a name="l00401"></a>00401 <a name="l00402"></a>00402 <a name="l00403"></a>00403 SocketSender::~SocketSender() <a name="l00404"></a>00404 { <a name="l00405"></a>00405 <span class="preprocessor">#ifdef USE_WINDOWS_STYLE_SOCKETS</span> <a name="l00406"></a>00406 <span class="preprocessor"></span> CancelIo((HANDLE) m_s.GetSocket()); <a name="l00407"></a>00407 <span class="preprocessor">#endif</span> <a name="l00408"></a>00408 <span class="preprocessor"></span>} <a name="l00409"></a>00409 <a name="l00410"></a>00410 <span class="keywordtype">void</span> SocketSender::Send(<span class="keyword">const</span> byte* buf, <span class="keywordtype">size_t</span> bufLen) <a name="l00411"></a>00411 { <a name="l00412"></a>00412 assert(!m_resultPending); <a name="l00413"></a>00413 DWORD written = 0; <a name="l00414"></a>00414 <span class="comment">// don't queue too much at once, or we might use up non-paged memory</span> <a name="l00415"></a>00415 WSABUF wsabuf = {UnsignedMin((u_long)128*1024, bufLen), (<span class="keywordtype">char</span> *)buf}; <a name="l00416"></a>00416 <span class="keywordflow">if</span> (WSASend(m_s, &wsabuf, 1, &written, 0, &m_overlapped, NULL) == 0) <a name="l00417"></a>00417 { <a name="l00418"></a>00418 m_resultPending = <span class="keyword">false</span>; <a name="l00419"></a>00419 m_lastResult = written; <a name="l00420"></a>00420 } <a name="l00421"></a>00421 <span class="keywordflow">else</span> <a name="l00422"></a>00422 { <a name="l00423"></a>00423 <span class="keywordflow">if</span> (WSAGetLastError() != WSA_IO_PENDING) <a name="l00424"></a>00424 m_s.CheckAndHandleError_int(<span class="stringliteral">"WSASend"</span>, SOCKET_ERROR); <a name="l00425"></a>00425 <a name="l00426"></a>00426 m_resultPending = <span class="keyword">true</span>; <a name="l00427"></a>00427 } <a name="l00428"></a>00428 } <a name="l00429"></a>00429 <a name="l00430"></a>00430 <span class="keywordtype">void</span> SocketSender::SendEof() <a name="l00431"></a>00431 { <a name="l00432"></a>00432 assert(!m_resultPending); <a name="l00433"></a>00433 m_s.ShutDown(SD_SEND); <a name="l00434"></a>00434 m_s.CheckAndHandleError(<span class="stringliteral">"ResetEvent"</span>, ResetEvent(m_event)); <a name="l00435"></a>00435 m_s.CheckAndHandleError_int(<span class="stringliteral">"WSAEventSelect"</span>, WSAEventSelect(m_s, m_event, FD_CLOSE)); <a name="l00436"></a>00436 m_resultPending = <span class="keyword">true</span>; <a name="l00437"></a>00437 } <a name="l00438"></a>00438 <a name="l00439"></a>00439 <span class="keywordtype">bool</span> SocketSender::EofSent() <a name="l00440"></a>00440 { <a name="l00441"></a>00441 <span class="keywordflow">if</span> (m_resultPending) <a name="l00442"></a>00442 { <a name="l00443"></a>00443 WSANETWORKEVENTS events; <a name="l00444"></a>00444 m_s.CheckAndHandleError_int(<span class="stringliteral">"WSAEnumNetworkEvents"</span>, WSAEnumNetworkEvents(m_s, m_event, &events)); <a name="l00445"></a>00445 <span class="keywordflow">if</span> ((events.lNetworkEvents & FD_CLOSE) != FD_CLOSE) <a name="l00446"></a>00446 <span class="keywordflow">throw</span> <a class="code" href="class_socket_1_1_err.html" title="exception thrown by Socket class">Socket::Err</a>(m_s, <span class="stringliteral">"WSAEnumNetworkEvents (FD_CLOSE not present)"</span>, E_FAIL); <a name="l00447"></a>00447 <span class="keywordflow">if</span> (events.iErrorCode[FD_CLOSE_BIT] != 0) <a name="l00448"></a>00448 <span class="keywordflow">throw</span> <a class="code" href="class_socket_1_1_err.html" title="exception thrown by Socket class">Socket::Err</a>(m_s, <span class="stringliteral">"FD_CLOSE (via WSAEnumNetworkEvents)"</span>, events.iErrorCode[FD_CLOSE_BIT]); <a name="l00449"></a>00449 m_resultPending = <span class="keyword">false</span>; <a name="l00450"></a>00450 } <a name="l00451"></a>00451 <span class="keywordflow">return</span> m_lastResult != 0; <a name="l00452"></a>00452 } <a name="l00453"></a>00453 <a name="l00454"></a><a class="code" href="class_socket_sender.html#a9f5d2a6bb5665c5b93d857485c8348e4">00454</a> <span class="keywordtype">void</span> <a class="code" href="class_socket_sender.html#a9f5d2a6bb5665c5b93d857485c8348e4" title="put wait objects into container">SocketSender::GetWaitObjects</a>(<a class="code" href="class_wait_object_container.html" title="container of wait objects">WaitObjectContainer</a> &container, <a class="code" href="class_call_stack.html">CallStack</a> <span class="keyword">const</span>& callStack) <a name="l00455"></a>00455 { <a name="l00456"></a>00456 <span class="keywordflow">if</span> (m_resultPending) <a name="l00457"></a>00457 container.AddHandle(m_event, <a class="code" href="class_call_stack.html">CallStack</a>(<span class="stringliteral">"SocketSender::GetWaitObjects() - result pending"</span>, &callStack)); <a name="l00458"></a>00458 <span class="keywordflow">else</span> <a name="l00459"></a>00459 container.SetNoWait(<a class="code" href="class_call_stack.html">CallStack</a>(<span class="stringliteral">"SocketSender::GetWaitObjects() - result ready"</span>, &callStack)); <a name="l00460"></a>00460 } <a name="l00461"></a>00461 <a name="l00462"></a>00462 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> SocketSender::GetSendResult() <a name="l00463"></a>00463 { <a name="l00464"></a>00464 <span class="keywordflow">if</span> (m_resultPending) <a name="l00465"></a>00465 { <a name="l00466"></a>00466 DWORD flags = 0; <a name="l00467"></a>00467 BOOL result = WSAGetOverlappedResult(m_s, &m_overlapped, &m_lastResult, <span class="keyword">false</span>, &flags); <a name="l00468"></a>00468 m_s.CheckAndHandleError(<span class="stringliteral">"WSAGetOverlappedResult"</span>, result); <a name="l00469"></a>00469 m_resultPending = <span class="keyword">false</span>; <a name="l00470"></a>00470 } <a name="l00471"></a>00471 <span class="keywordflow">return</span> m_lastResult; <a name="l00472"></a>00472 } <a name="l00473"></a>00473 <a name="l00474"></a>00474 <span class="preprocessor">#endif</span> <a name="l00475"></a>00475 <span class="preprocessor"></span> <a name="l00476"></a>00476 <span class="preprocessor">#ifdef USE_BERKELEY_STYLE_SOCKETS</span> <a name="l00477"></a>00477 <span class="preprocessor"></span> <a name="l00478"></a>00478 SocketReceiver::SocketReceiver(<a class="code" href="class_socket.html" title="wrapper for Windows or Berkeley Sockets">Socket</a> &s) <a name="l00479"></a>00479 : m_s(s), m_lastResult(0), m_eofReceived(false) <a name="l00480"></a>00480 { <a name="l00481"></a>00481 } <a name="l00482"></a>00482 <a name="l00483"></a>00483 <span class="keywordtype">void</span> <a class="code" href="class_socket_receiver.html#aa851a6020843d4311036a6f4299f9a6b" 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> &container, <a class="code" href="class_call_stack.html">CallStack</a> <span class="keyword">const</span>& callStack) <a name="l00484"></a>00484 { <a name="l00485"></a>00485 <span class="keywordflow">if</span> (!m_eofReceived) <a name="l00486"></a>00486 container.AddReadFd(m_s, <a class="code" href="class_call_stack.html">CallStack</a>(<span class="stringliteral">"SocketReceiver::GetWaitObjects()"</span>, &callStack)); <a name="l00487"></a>00487 } <a name="l00488"></a>00488 <a name="l00489"></a>00489 <span class="keywordtype">bool</span> <a class="code" href="class_socket_receiver.html#a31e0ccf504d9c16760051f10d749c034" 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="l00490"></a>00490 { <a name="l00491"></a>00491 m_lastResult = m_s.Receive(buf, bufLen); <a name="l00492"></a>00492 <span class="keywordflow">if</span> (bufLen > 0 && m_lastResult == 0) <a name="l00493"></a>00493 m_eofReceived = <span class="keyword">true</span>; <a name="l00494"></a>00494 <span class="keywordflow">return</span> <span class="keyword">true</span>; <a name="l00495"></a>00495 } <a name="l00496"></a>00496 <a name="l00497"></a>00497 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> SocketReceiver::GetReceiveResult() <a name="l00498"></a>00498 { <a name="l00499"></a>00499 <span class="keywordflow">return</span> m_lastResult; <a name="l00500"></a>00500 } <a name="l00501"></a>00501 <a name="l00502"></a>00502 SocketSender::SocketSender(<a class="code" href="class_socket.html" title="wrapper for Windows or Berkeley Sockets">Socket</a> &s) <a name="l00503"></a>00503 : m_s(s), m_lastResult(0) <a name="l00504"></a>00504 { <a name="l00505"></a>00505 } <a name="l00506"></a>00506 <a name="l00507"></a>00507 <span class="keywordtype">void</span> SocketSender::Send(<span class="keyword">const</span> byte* buf, <span class="keywordtype">size_t</span> bufLen) <a name="l00508"></a>00508 { <a name="l00509"></a>00509 m_lastResult = m_s.Send(buf, bufLen); <a name="l00510"></a>00510 } <a name="l00511"></a>00511 <a name="l00512"></a>00512 <span class="keywordtype">void</span> SocketSender::SendEof() <a name="l00513"></a>00513 { <a name="l00514"></a>00514 m_s.ShutDown(SD_SEND); <a name="l00515"></a>00515 } <a name="l00516"></a>00516 <a name="l00517"></a>00517 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> SocketSender::GetSendResult() <a name="l00518"></a>00518 { <a name="l00519"></a>00519 <span class="keywordflow">return</span> m_lastResult; <a name="l00520"></a>00520 } <a name="l00521"></a>00521 <a name="l00522"></a>00522 <span class="keywordtype">void</span> <a class="code" href="class_socket_sender.html#a9f5d2a6bb5665c5b93d857485c8348e4" title="put wait objects into container">SocketSender::GetWaitObjects</a>(<a class="code" href="class_wait_object_container.html" title="container of wait objects">WaitObjectContainer</a> &container, <a class="code" href="class_call_stack.html">CallStack</a> <span class="keyword">const</span>& callStack) <a name="l00523"></a>00523 { <a name="l00524"></a>00524 container.AddWriteFd(m_s, <a class="code" href="class_call_stack.html">CallStack</a>(<span class="stringliteral">"SocketSender::GetWaitObjects()"</span>, &callStack)); <a name="l00525"></a>00525 } <a name="l00526"></a>00526 <a name="l00527"></a>00527 <span class="preprocessor">#endif</span> <a name="l00528"></a>00528 <span class="preprocessor"></span> <a name="l00529"></a>00529 NAMESPACE_END <a name="l00530"></a>00530 <a name="l00531"></a>00531 <span class="preprocessor">#endif // #ifdef SOCKETS_AVAILABLE</span> </pre></div></div> <hr size="1"/><address style="text-align: right;"><small>Generated on 9 Dec 2009 for Crypto++ by <a href="http://www.doxygen.org/index.html"> <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.6.1 </small></address> </body> </html>