1 | /* $Id: socket.cpp,v 1.16 2003-02-24 11:14:30 sandervl Exp $ */
|
---|
2 | /*
|
---|
3 | * based on Windows Sockets 1.1 specs
|
---|
4 | * (ftp.microsoft.com:/Advsys/winsock/spec11/WINSOCK.TXT)
|
---|
5 | *
|
---|
6 | * (C) 1993,1994,1996,1997 John Brezak, Erik Bos, Alex Korobka.
|
---|
7 | *
|
---|
8 | * NOTE: If you make any changes to fix a particular app, make sure
|
---|
9 | * they don't break something else like Netscape or telnet and ftp
|
---|
10 | * clients and servers (www.winsite.com got a lot of those).
|
---|
11 | *
|
---|
12 | * NOTE 2: Many winsock structs such as servent, hostent, protoent, ...
|
---|
13 | * are used with 1-byte alignment for Win16 programs and 4-byte alignment
|
---|
14 | * for Win32 programs in winsock.h. winsock2.h uses forced 4-byte alignment.
|
---|
15 | * So we have non-forced (just as MSDN) ws_XXXXent (winsock.h), 4-byte forced
|
---|
16 | * ws_XXXXent32 (winsock2.h) and 1-byte forced ws_XXXXent16 (winsock16.h).
|
---|
17 | */
|
---|
18 |
|
---|
19 |
|
---|
20 | /*****************************************************************************
|
---|
21 | * Includes *
|
---|
22 | *****************************************************************************/
|
---|
23 |
|
---|
24 | #include <odin.h>
|
---|
25 | #include <winsock2.h>
|
---|
26 | #include <debugtools.h>
|
---|
27 | #include <odinwrap.h>
|
---|
28 |
|
---|
29 |
|
---|
30 | ODINDEBUGCHANNEL(WS2_32-SOCKET)
|
---|
31 |
|
---|
32 |
|
---|
33 | /***********************************************************************
|
---|
34 | * WSACreateEvent() (WS2_32.???)
|
---|
35 | *
|
---|
36 | */
|
---|
37 | ODINFUNCTION0(WSAEVENT, WSACreateEvent)
|
---|
38 | {
|
---|
39 | /* Create a manual-reset event, with initial state: unsignealed */
|
---|
40 |
|
---|
41 | return CreateEventA(NULL, TRUE, FALSE, NULL);
|
---|
42 | }
|
---|
43 |
|
---|
44 | ODINFUNCTION1(BOOL, WSASetEvent,
|
---|
45 | WSAEVENT, hEvent)
|
---|
46 | {
|
---|
47 | return SetEvent(hEvent);
|
---|
48 | }
|
---|
49 |
|
---|
50 | /***********************************************************************
|
---|
51 | * WSACloseEvent() (WS2_32.???)
|
---|
52 | *
|
---|
53 | */
|
---|
54 | ODINFUNCTION1(BOOL, WSACloseEvent,
|
---|
55 | WSAEVENT, hEvent)
|
---|
56 | {
|
---|
57 | return CloseHandle(hEvent);
|
---|
58 | }
|
---|
59 |
|
---|
60 | /***********************************************************************
|
---|
61 | * WSASocketA() (WS2_32.???)
|
---|
62 | *
|
---|
63 | */
|
---|
64 | ODINFUNCTION6(SOCKET, WSASocketA,
|
---|
65 | int, af,
|
---|
66 | int, type,
|
---|
67 | int, protocol,
|
---|
68 | LPWSAPROTOCOL_INFOA, lpProtocolInfo,
|
---|
69 | GROUP, g,
|
---|
70 | DWORD, dwFlags)
|
---|
71 | {
|
---|
72 | dprintf(("WSASocketA incorrectly implemented"));
|
---|
73 | /*
|
---|
74 | FIXME: The "advanced" parameters of WSASocketA (lpProtocolInfo,
|
---|
75 | g, dwFlags) are ignored.
|
---|
76 | */
|
---|
77 |
|
---|
78 | TRACE("WSASocketA af=%d type=%d protocol=%d protocol_info=%p group=%d flags=0x%lx\n",
|
---|
79 | af, type, protocol, lpProtocolInfo, g, dwFlags );
|
---|
80 |
|
---|
81 | return ( socket (af, type, protocol) );
|
---|
82 | }
|
---|
83 |
|
---|
84 | /***********************************************************************
|
---|
85 | * WSASocketA() (WS2_32.???)
|
---|
86 | *
|
---|
87 | */
|
---|
88 | ODINFUNCTION6(SOCKET, WSASocketW,
|
---|
89 | int, af,
|
---|
90 | int, type,
|
---|
91 | int, protocol,
|
---|
92 | LPWSAPROTOCOL_INFOW, lpProtocolInfo,
|
---|
93 | GROUP, g,
|
---|
94 | DWORD, dwFlags)
|
---|
95 | {
|
---|
96 | dprintf(("WSASocketW incorrectly implemented"));
|
---|
97 | /*
|
---|
98 | FIXME: The "advanced" parameters of WSASocketA (lpProtocolInfo,
|
---|
99 | g, dwFlags) are ignored.
|
---|
100 | */
|
---|
101 |
|
---|
102 | TRACE("WSASocketW af=%d type=%d protocol=%d protocol_info=%p group=%d flags=0x%lx\n",
|
---|
103 | af, type, protocol, lpProtocolInfo, g, dwFlags );
|
---|
104 |
|
---|
105 | return ( socket (af, type, protocol) );
|
---|
106 | }
|
---|
107 | /***********************************************************************
|
---|
108 | * WSAAccept (WS2_32.26)
|
---|
109 | */
|
---|
110 | SOCKET WINAPI WSAAccept( SOCKET s, struct WS_sockaddr *addr, LPINT addrlen,
|
---|
111 | LPCONDITIONPROC lpfnCondition, DWORD dwCallbackData)
|
---|
112 | {
|
---|
113 |
|
---|
114 | int ret = 0, size = 0;
|
---|
115 | WSABUF CallerId, CallerData, CalleeId, CalleeData;
|
---|
116 | /* QOS SQOS, GQOS; */
|
---|
117 | GROUP g;
|
---|
118 | SOCKET cs;
|
---|
119 | SOCKADDR src_addr, dst_addr;
|
---|
120 |
|
---|
121 | TRACE("Socket %u, sockaddr %p, addrlen %p, fnCondition %p, dwCallbackData %ld\n",
|
---|
122 | s, addr, addrlen, lpfnCondition, dwCallbackData);
|
---|
123 |
|
---|
124 |
|
---|
125 | size = sizeof(src_addr);
|
---|
126 | cs = accept(s, &src_addr, &size);
|
---|
127 |
|
---|
128 | if (cs == SOCKET_ERROR) return SOCKET_ERROR;
|
---|
129 |
|
---|
130 | CallerId.buf = (char *)&src_addr;
|
---|
131 | CallerId.len = sizeof(src_addr);
|
---|
132 |
|
---|
133 | CallerData.buf = NULL;
|
---|
134 | CallerData.len = (ULONG)NULL;
|
---|
135 |
|
---|
136 | getsockname(cs, &dst_addr, &size);
|
---|
137 |
|
---|
138 | CalleeId.buf = (char *)&dst_addr;
|
---|
139 | CalleeId.len = sizeof(dst_addr);
|
---|
140 |
|
---|
141 | ret = (*lpfnCondition)(&CallerId, &CallerData, NULL, NULL,
|
---|
142 | &CalleeId, &CalleeData, &g, dwCallbackData);
|
---|
143 |
|
---|
144 | switch (ret)
|
---|
145 | {
|
---|
146 | case CF_ACCEPT:
|
---|
147 | if (addr && addrlen)
|
---|
148 | addr = (struct WS_sockaddr *)memcpy(addr, &src_addr, (*addrlen > size) ? size : *addrlen );
|
---|
149 | return cs;
|
---|
150 | case CF_DEFER:
|
---|
151 | WSASetLastError(WSATRY_AGAIN);
|
---|
152 | return SOCKET_ERROR;
|
---|
153 | case CF_REJECT:
|
---|
154 | closesocket(cs);
|
---|
155 | WSASetLastError(WSAECONNREFUSED);
|
---|
156 | return SOCKET_ERROR;
|
---|
157 | default:
|
---|
158 | FIXME("Unknown return type from Condition function\n");
|
---|
159 | WSASetLastError(WSAENOTSOCK);
|
---|
160 | return SOCKET_ERROR;
|
---|
161 | }
|
---|
162 |
|
---|
163 | WSASetLastError(WSAENOTSOCK);
|
---|
164 | return SOCKET_ERROR;
|
---|
165 | }
|
---|
166 | //******************************************************************************
|
---|
167 | //******************************************************************************
|
---|
168 | /***********************************************************************
|
---|
169 | * WSASendTo (WS2_32.74)
|
---|
170 | */
|
---|
171 | INT WINAPI WSASendTo( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
|
---|
172 | LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
|
---|
173 | const struct WS_sockaddr *to, int tolen,
|
---|
174 | LPWSAOVERLAPPED lpOverlapped,
|
---|
175 | LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
|
---|
176 | {
|
---|
177 | WSASetLastError(WSAENOTSOCK);
|
---|
178 | return SOCKET_ERROR;
|
---|
179 | }
|
---|
180 | /***********************************************************************
|
---|
181 | * WSASend (WS2_32.72)
|
---|
182 | */
|
---|
183 | INT WINAPI WSASend( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
|
---|
184 | LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
|
---|
185 | LPWSAOVERLAPPED lpOverlapped,
|
---|
186 | LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
|
---|
187 | {
|
---|
188 | DWORD dwBytesSent, tmpret;
|
---|
189 | DWORD ret = NO_ERROR;
|
---|
190 |
|
---|
191 | dprintf(("WSASend %d %x %d %x %x %x %x", s, lpBuffers, dwBufferCount, lpNumberOfBytesSent, dwFlags, lpOverlapped, lpCompletionRoutine));
|
---|
192 |
|
---|
193 | if(lpBuffers == NULL) {
|
---|
194 | WSASetLastError(WSAEINVAL);
|
---|
195 | return SOCKET_ERROR;
|
---|
196 | }
|
---|
197 | if(lpNumberOfBytesSent == NULL) {
|
---|
198 | lpNumberOfBytesSent = &dwBytesSent;
|
---|
199 | }
|
---|
200 | if(!lpOverlapped && !lpCompletionRoutine)
|
---|
201 | {
|
---|
202 | *lpNumberOfBytesSent = 0;
|
---|
203 |
|
---|
204 | for(int i=0;i<dwBufferCount;i++) {
|
---|
205 | tmpret = send(s, lpBuffers[i].buf, lpBuffers[i].len, dwFlags);
|
---|
206 | if(tmpret != SOCKET_ERROR) {
|
---|
207 | *lpNumberOfBytesSent += tmpret;
|
---|
208 | }
|
---|
209 | else {
|
---|
210 | ret = SOCKET_ERROR;
|
---|
211 | break;
|
---|
212 | }
|
---|
213 | }
|
---|
214 | if(*lpNumberOfBytesSent) {
|
---|
215 | WSASetLastError(NO_ERROR);
|
---|
216 | ret = NO_ERROR;
|
---|
217 | }
|
---|
218 | return ret;
|
---|
219 | }
|
---|
220 | return WSASendTo ( s, lpBuffers, dwBufferCount, lpNumberOfBytesSent, dwFlags,
|
---|
221 | NULL, 0, lpOverlapped, lpCompletionRoutine );
|
---|
222 | }
|
---|
223 | /***********************************************************************
|
---|
224 | * WSARecvFrom (WS2_32.69)
|
---|
225 | */
|
---|
226 | INT WINAPI WSARecvFrom( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
|
---|
227 | LPDWORD lpNumberOfBytesRecvd, LPDWORD lpFlags, struct WS_sockaddr *lpFrom,
|
---|
228 | LPINT lpFromlen, LPWSAOVERLAPPED lpOverlapped,
|
---|
229 | LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
|
---|
230 |
|
---|
231 | {
|
---|
232 | WSASetLastError(WSAENOTSOCK);
|
---|
233 | return SOCKET_ERROR;
|
---|
234 | }
|
---|
235 | /***********************************************************************
|
---|
236 | * WSARecv (WS2_32.67)
|
---|
237 | */
|
---|
238 | int WINAPI WSARecv (SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
|
---|
239 | LPDWORD lpNumberOfBytesReceived, LPDWORD lpFlags,
|
---|
240 | LPWSAOVERLAPPED lpOverlapped,
|
---|
241 | LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine)
|
---|
242 | {
|
---|
243 | DWORD dwBytesReceived, tmpret, flags;
|
---|
244 | DWORD ret = NO_ERROR;
|
---|
245 |
|
---|
246 | dprintf(("WSARecv %d %x %d %x %x %x %x", s, lpBuffers, dwBufferCount, lpNumberOfBytesReceived, lpFlags, lpOverlapped, lpCompletionRoutine));
|
---|
247 | if(lpBuffers == NULL) {
|
---|
248 | WSASetLastError(WSAEINVAL);
|
---|
249 | return SOCKET_ERROR;
|
---|
250 | }
|
---|
251 | if(lpNumberOfBytesReceived == NULL) {
|
---|
252 | lpNumberOfBytesReceived = &dwBytesReceived;
|
---|
253 | }
|
---|
254 | if(lpFlags == NULL) {
|
---|
255 | lpFlags = &flags;
|
---|
256 | flags = 0;
|
---|
257 | }
|
---|
258 | if(!lpOverlapped && !lpCompletionRoutine)
|
---|
259 | {
|
---|
260 | *lpNumberOfBytesReceived = 0;
|
---|
261 |
|
---|
262 | for(int i=0;i<dwBufferCount;i++) {
|
---|
263 | tmpret = recv(s, lpBuffers[i].buf, lpBuffers[i].len, *lpFlags);
|
---|
264 | if(tmpret != SOCKET_ERROR) {
|
---|
265 | *lpNumberOfBytesReceived += tmpret;
|
---|
266 | }
|
---|
267 | else {
|
---|
268 | ret = SOCKET_ERROR;
|
---|
269 | break;
|
---|
270 | }
|
---|
271 | }
|
---|
272 | if(*lpNumberOfBytesReceived) {
|
---|
273 | WSASetLastError(NO_ERROR);
|
---|
274 | ret = NO_ERROR;
|
---|
275 | }
|
---|
276 | dprintf(("WSARecv returned %d (read %x)", ret, *lpNumberOfBytesReceived));
|
---|
277 | *lpFlags = 0; //what to do with this?
|
---|
278 | return ret;
|
---|
279 | }
|
---|
280 | return WSARecvFrom (s, lpBuffers, dwBufferCount, lpNumberOfBytesReceived, lpFlags,
|
---|
281 | NULL, NULL, lpOverlapped, lpCompletionRoutine);
|
---|
282 | }
|
---|
283 |
|
---|
284 | /***********************************************************************
|
---|
285 | * WSAAddressToStringA (WS2_32.27)
|
---|
286 | *
|
---|
287 | * See WSAAddressToStringW
|
---|
288 | */
|
---|
289 | INT WINAPI WSAAddressToStringA( LPSOCKADDR sockaddr, DWORD len,
|
---|
290 | LPWSAPROTOCOL_INFOA info, LPSTR string,
|
---|
291 | LPDWORD lenstr )
|
---|
292 | {
|
---|
293 | INT size;
|
---|
294 | CHAR buffer[22]; /* 12 digits + 3 dots + ':' + 5 digits + '\0' */
|
---|
295 | CHAR *p;
|
---|
296 |
|
---|
297 | TRACE( "(%p, %d, %p, %p, %p)\n", sockaddr, len, info, string, lenstr );
|
---|
298 |
|
---|
299 | if (!sockaddr || len < sizeof(SOCKADDR_IN)) return SOCKET_ERROR;
|
---|
300 | if (!string || !lenstr) return SOCKET_ERROR;
|
---|
301 |
|
---|
302 | /* sin_family is guaranteed to be the first u_short */
|
---|
303 | if (((SOCKADDR_IN *)sockaddr)->sin_family != AF_INET) return SOCKET_ERROR;
|
---|
304 |
|
---|
305 | sprintf( buffer, "%u.%u.%u.%u:%u",
|
---|
306 | (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.s_addr ) >> 24 & 0xff),
|
---|
307 | (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.s_addr ) >> 16 & 0xff),
|
---|
308 | (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.s_addr ) >> 8 & 0xff),
|
---|
309 | (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.s_addr ) & 0xff),
|
---|
310 | ntohs( ((SOCKADDR_IN *)sockaddr)->sin_port ) );
|
---|
311 |
|
---|
312 | p = strchr( buffer, ':' );
|
---|
313 | if (!((SOCKADDR_IN *)sockaddr)->sin_port) *p = 0;
|
---|
314 |
|
---|
315 | size = strlen( buffer );
|
---|
316 |
|
---|
317 | if (*lenstr < size)
|
---|
318 | {
|
---|
319 | *lenstr = size;
|
---|
320 | WSASetLastError(WSAEFAULT);
|
---|
321 | return SOCKET_ERROR;
|
---|
322 | }
|
---|
323 |
|
---|
324 | strcpy( string, buffer );
|
---|
325 | return 0;
|
---|
326 | }
|
---|