?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
server: Return WSA error codes in socket events.
[wine.git] / dlls / ws2_32 / socket.c
blob2ea426943888e0a21fe13cdf9ee7714bd26ddcbc
1 /*
2 * based on Windows Sockets 1.1 specs
3 *
4 * Copyright (C) 1993,1994,1996,1997 John Brezak, Erik Bos, Alex Korobka.
5 * Copyright (C) 2001 Stefan Leichter
6 * Copyright (C) 2004 Hans Leidekker
7 * Copyright (C) 2005 Marcus Meissner
8 * Copyright (C) 2006-2008 Kai Blin
9 *
10 * This library is free software; you can redistribute it and/or
11 * modify it under the terms of the GNU Lesser General Public
12 * License as published by the Free Software Foundation; either
13 * version 2.1 of the License, or (at your option) any later version.
14 *
15 * This library is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 * Lesser General Public License for more details.
19 *
20 * You should have received a copy of the GNU Lesser General Public
21 * License along with this library; if not, write to the Free Software
22 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
23 *
24 * NOTE: If you make any changes to fix a particular app, make sure
25 * they don't break something else like Netscape or telnet and ftp
26 * clients and servers (www.winsite.com got a lot of those).
27 */
28
29 #include "config.h"
30 #include "wine/port.h"
31
32 #include <stdarg.h>
33 #include <stdio.h>
34 #include <string.h>
35 #include <sys/types.h>
36 #include <limits.h>
37 #ifdef HAVE_SYS_IPC_H
38 # include <sys/ipc.h>
39 #endif
40 #ifdef HAVE_SYS_IOCTL_H
41 # include <sys/ioctl.h>
42 #endif
43 #ifdef HAVE_SYS_FILIO_H
44 # include <sys/filio.h>
45 #endif
46 #ifdef HAVE_SYS_SOCKIO_H
47 # include <sys/sockio.h>
48 #endif
49
50 #if defined(__EMX__)
51 # include <sys/so_ioctl.h>
52 #endif
53
54 #ifdef HAVE_SYS_PARAM_H
55 # include <sys/param.h>
56 #endif
57
58 #ifdef HAVE_SYS_MSG_H
59 # include <sys/msg.h>
60 #endif
61 #ifdef HAVE_SYS_WAIT_H
62 # include <sys/wait.h>
63 #endif
64 #ifdef HAVE_SYS_UIO_H
65 # include <sys/uio.h>
66 #endif
67 #ifdef HAVE_SYS_SOCKET_H
68 #include <sys/socket.h>
69 #endif
70 #ifdef HAVE_NETINET_IN_H
71 # include <netinet/in.h>
72 #endif
73 #ifdef HAVE_NETINET_TCP_H
74 # include <netinet/tcp.h>
75 #endif
76 #ifdef HAVE_ARPA_INET_H
77 # include <arpa/inet.h>
78 #endif
79 #include <ctype.h>
80 #include <fcntl.h>
81 #include <errno.h>
82 #ifdef HAVE_NETDB_H
83 #include <netdb.h>
84 #endif
85 #ifdef HAVE_UNISTD_H
86 # include <unistd.h>
87 #endif
88 #include <stdlib.h>
89 #ifdef HAVE_ARPA_NAMESER_H
90 # include <arpa/nameser.h>
91 #endif
92 #ifdef HAVE_RESOLV_H
93 # include <resolv.h>
94 #endif
95 #ifdef HAVE_NET_IF_H
96 # include <net/if.h>
97 #endif
98 #ifdef HAVE_LINUX_FILTER_H
99 # include <linux/filter.h>
100 #endif
101
102 #ifdef HAVE_NETIPX_IPX_H
103 # include <netipx/ipx.h>
104 #elif defined(HAVE_LINUX_IPX_H)
105 # ifdef HAVE_ASM_TYPES_H
106 # include <asm/types.h>
107 # endif
108 # ifdef HAVE_LINUX_TYPES_H
109 # include <linux/types.h>
110 # endif
111 # include <linux/ipx.h>
112 #endif
113 #if defined(SOL_IPX) || defined(SO_DEFAULT_HEADERS)
114 # define HAS_IPX
115 #endif
116
117 #ifdef HAVE_LINUX_IRDA_H
118 # ifdef HAVE_LINUX_TYPES_H
119 # include <linux/types.h>
120 # endif
121 # include <linux/irda.h>
122 # define HAS_IRDA
123 #endif
124
125 #ifdef HAVE_POLL_H
126 #include <poll.h>
127 #endif
128 #ifdef HAVE_SYS_POLL_H
129 # include <sys/poll.h>
130 #endif
131 #ifdef HAVE_SYS_TIME_H
132 # include <sys/time.h>
133 #endif
134
135 #define NONAMELESSUNION
136 #define NONAMELESSSTRUCT
137 #include "ntstatus.h"
138 #define WIN32_NO_STATUS
139 #include "windef.h"
140 #include "winbase.h"
141 #include "wingdi.h"
142 #include "winuser.h"
143 #include "winerror.h"
144 #include "winnls.h"
145 #include "winsock2.h"
146 #include "mswsock.h"
147 #include "ws2tcpip.h"
148 #include "ws2spi.h"
149 #include "wsipx.h"
150 #include "wsnwlink.h"
151 #include "wshisotp.h"
152 #include "mstcpip.h"
153 #include "af_irda.h"
154 #include "winnt.h"
155 #define USE_WC_PREFIX /* For CMSG_DATA */
156 #include "iphlpapi.h"
157 #include "wine/server.h"
158 #include "wine/debug.h"
159 #include "wine/exception.h"
160 #include "wine/unicode.h"
161
162 #if defined(linux) && !defined(IP_UNICAST_IF)
163 #define IP_UNICAST_IF 50
164 #endif
165
166 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__) || defined(__DragonFly__)
167 # define sipx_network sipx_addr.x_net
168 # define sipx_node sipx_addr.x_host.c_host
169 #endif /* __FreeBSD__ */
170
171 #ifndef INADDR_NONE
172 #define INADDR_NONE ~0UL
173 #endif
174
175 #if !defined(TCP_KEEPIDLE) && defined(TCP_KEEPALIVE)
176 /* TCP_KEEPALIVE is the Mac OS name for TCP_KEEPIDLE */
177 #define TCP_KEEPIDLE TCP_KEEPALIVE
178 #endif
179
180 #define FILE_USE_FILE_POINTER_POSITION ((LONGLONG)-2)
181
182 WINE_DEFAULT_DEBUG_CHANNEL(winsock);
183 WINE_DECLARE_DEBUG_CHANNEL(winediag);
184
185 /* names of the protocols */
186 static const WCHAR NameIpxW[] = {'I', 'P', 'X', '\0'};
187 static const WCHAR NameSpxW[] = {'S', 'P', 'X', '\0'};
188 static const WCHAR NameSpxIIW[] = {'S', 'P', 'X', ' ', 'I', 'I', '\0'};
189 static const WCHAR NameTcpW[] = {'T', 'C', 'P', '/', 'I', 'P', '\0'};
190 static const WCHAR NameUdpW[] = {'U', 'D', 'P', '/', 'I', 'P', '\0'};
191
192 /* Taken from Win2k */
193 static const GUID ProviderIdIP = { 0xe70f1aa0, 0xab8b, 0x11cf,
194 { 0x8c, 0xa3, 0x00, 0x80, 0x5f, 0x48, 0xa1, 0x92 } };
195 static const GUID ProviderIdIPX = { 0x11058240, 0xbe47, 0x11cf,
196 { 0x95, 0xc8, 0x00, 0x80, 0x5f, 0x48, 0xa1, 0x92 } };
197 static const GUID ProviderIdSPX = { 0x11058241, 0xbe47, 0x11cf,
198 { 0x95, 0xc8, 0x00, 0x80, 0x5f, 0x48, 0xa1, 0x92 } };
199
200 static const INT valid_protocols[] =
201 {
202 WS_IPPROTO_TCP,
203 WS_IPPROTO_UDP,
204 WS_NSPROTO_IPX,
205 WS_NSPROTO_SPX,
206 WS_NSPROTO_SPXII,
207 0
208 };
209
210 #define IS_IPX_PROTO(X) ((X) >= WS_NSPROTO_IPX && (X) <= WS_NSPROTO_IPX + 255)
211
212 #if defined(IP_UNICAST_IF) && defined(SO_ATTACH_FILTER)
213 # define LINUX_BOUND_IF
214 struct interface_filter {
215 struct sock_filter iface_memaddr;
216 struct sock_filter iface_rule;
217 struct sock_filter ip_memaddr;
218 struct sock_filter ip_rule;
219 struct sock_filter return_keep;
220 struct sock_filter return_dump;
221 };
222 # define FILTER_JUMP_DUMP(here) (u_char)(offsetof(struct interface_filter, return_dump) \
223 -offsetof(struct interface_filter, here)-sizeof(struct sock_filter)) \
224 /sizeof(struct sock_filter)
225 # define FILTER_JUMP_KEEP(here) (u_char)(offsetof(struct interface_filter, return_keep) \
226 -offsetof(struct interface_filter, here)-sizeof(struct sock_filter)) \
227 /sizeof(struct sock_filter)
228 # define FILTER_JUMP_NEXT() (u_char)(0)
229 # define SKF_NET_DESTIP 16 /* offset in the network header to the destination IP */
230 static struct interface_filter generic_interface_filter = {
231 /* This filter rule allows incoming packets on the specified interface, which works for all
232 * remotely generated packets and for locally generated broadcast packets. */
233 BPF_STMT(BPF_LD+BPF_W+BPF_ABS, SKF_AD_OFF+SKF_AD_IFINDEX),
234 BPF_JUMP(BPF_JMP+BPF_JEQ+BPF_K, 0xdeadbeef, FILTER_JUMP_KEEP(iface_rule), FILTER_JUMP_NEXT()),
235 /* This rule allows locally generated packets targeted at the specific IP address of the chosen
236 * adapter (local packets not destined for the broadcast address do not have IFINDEX set) */
237 BPF_STMT(BPF_LD+BPF_W+BPF_ABS, SKF_NET_OFF+SKF_NET_DESTIP),
238 BPF_JUMP(BPF_JMP+BPF_JEQ+BPF_K, 0xdeadbeef, FILTER_JUMP_KEEP(ip_rule), FILTER_JUMP_DUMP(ip_rule)),
239 BPF_STMT(BPF_RET+BPF_K, (u_int)-1), /* keep packet */
240 BPF_STMT(BPF_RET+BPF_K, 0) /* dump packet */
241 };
242 #endif /* LINUX_BOUND_IF */
243
244 extern ssize_t CDECL __wine_locked_recvmsg( int fd, struct msghdr *hdr, int flags );
245
246 /*
247 * The actual definition of WSASendTo, wrapped in a different function name
248 * so that internal calls from ws2_32 itself will not trigger programs like
249 * Garena, which hooks WSASendTo/WSARecvFrom calls.
250 */
251 static int WS2_sendto( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
252 LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
253 const struct WS_sockaddr *to, int tolen,
254 LPWSAOVERLAPPED lpOverlapped,
255 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine );
256
257 /*
258 * Internal fundamental receive function, essentially WSARecvFrom with an
259 * additional parameter to support message control headers.
260 */
261 static int WS2_recv_base( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
262 LPDWORD lpNumberOfBytesRecvd, LPDWORD lpFlags,
263 struct WS_sockaddr *lpFrom,
264 LPINT lpFromlen, LPWSAOVERLAPPED lpOverlapped,
265 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine,
266 LPWSABUF lpControlBuffer );
267
268 /* critical section to protect some non-reentrant net function */
269 static CRITICAL_SECTION csWSgetXXXbyYYY;
270 static CRITICAL_SECTION_DEBUG critsect_debug =
271 {
272 0, 0, &csWSgetXXXbyYYY,
273 { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
274 0, 0, { (DWORD_PTR)(__FILE__ ": csWSgetXXXbyYYY") }
275 };
276 static CRITICAL_SECTION csWSgetXXXbyYYY = { &critsect_debug, -1, 0, 0, 0, 0 };
277
278 union generic_unix_sockaddr
279 {
280 struct sockaddr addr;
281 char data[128]; /* should be big enough for all families */
282 };
283
284 static inline const char *debugstr_sockaddr( const struct WS_sockaddr *a )
285 {
286 if (!a) return "(nil)";
287 switch (a->sa_family)
288 {
289 case WS_AF_INET:
290 {
291 char buf[16];
292 const char *p;
293 struct WS_sockaddr_in *sin = (struct WS_sockaddr_in *)a;
294
295 p = WS_inet_ntop( WS_AF_INET, &sin->sin_addr, buf, sizeof(buf) );
296 if (!p)
297 p = "(unknown IPv4 address)";
298
299 return wine_dbg_sprintf("{ family AF_INET, address %s, port %d }",
300 p, ntohs(sin->sin_port));
301 }
302 case WS_AF_INET6:
303 {
304 char buf[46];
305 const char *p;
306 struct WS_sockaddr_in6 *sin = (struct WS_sockaddr_in6 *)a;
307
308 p = WS_inet_ntop( WS_AF_INET6, &sin->sin6_addr, buf, sizeof(buf) );
309 if (!p)
310 p = "(unknown IPv6 address)";
311 return wine_dbg_sprintf("{ family AF_INET6, address %s, port %d }",
312 p, ntohs(sin->sin6_port));
313 }
314 case WS_AF_IPX:
315 {
316 int i;
317 char netnum[16], nodenum[16];
318 struct WS_sockaddr_ipx *sin = (struct WS_sockaddr_ipx *)a;
319
320 for (i = 0;i < 4; i++) sprintf(netnum + i * 2, "%02X", (unsigned char) sin->sa_netnum[i]);
321 for (i = 0;i < 6; i++) sprintf(nodenum + i * 2, "%02X", (unsigned char) sin->sa_nodenum[i]);
322
323 return wine_dbg_sprintf("{ family AF_IPX, address %s.%s, ipx socket %d }",
324 netnum, nodenum, sin->sa_socket);
325 }
326 case WS_AF_IRDA:
327 {
328 DWORD addr;
329
330 memcpy( &addr, ((const SOCKADDR_IRDA *)a)->irdaDeviceID, sizeof(addr) );
331 addr = ntohl( addr );
332 return wine_dbg_sprintf("{ family AF_IRDA, addr %08x, name %s }",
333 addr,
334 ((const SOCKADDR_IRDA *)a)->irdaServiceName);
335 }
336 default:
337 return wine_dbg_sprintf("{ family %d }", a->sa_family);
338 }
339 }
340
341 static inline const char *debugstr_sockopt(int level, int optname)
342 {
343 const char *stropt = NULL, *strlevel = NULL;
344
345 #define DEBUG_SOCKLEVEL(x) case (x): strlevel = #x
346 #define DEBUG_SOCKOPT(x) case (x): stropt = #x; break
347
348 switch(level)
349 {
350 DEBUG_SOCKLEVEL(WS_SOL_SOCKET);
351 switch(optname)
352 {
353 DEBUG_SOCKOPT(WS_SO_ACCEPTCONN);
354 DEBUG_SOCKOPT(WS_SO_BROADCAST);
355 DEBUG_SOCKOPT(WS_SO_BSP_STATE);
356 DEBUG_SOCKOPT(WS_SO_CONDITIONAL_ACCEPT);
357 DEBUG_SOCKOPT(WS_SO_CONNECT_TIME);
358 DEBUG_SOCKOPT(WS_SO_DEBUG);
359 DEBUG_SOCKOPT(WS_SO_DONTLINGER);
360 DEBUG_SOCKOPT(WS_SO_DONTROUTE);
361 DEBUG_SOCKOPT(WS_SO_ERROR);
362 DEBUG_SOCKOPT(WS_SO_EXCLUSIVEADDRUSE);
363 DEBUG_SOCKOPT(WS_SO_GROUP_ID);
364 DEBUG_SOCKOPT(WS_SO_GROUP_PRIORITY);
365 DEBUG_SOCKOPT(WS_SO_KEEPALIVE);
366 DEBUG_SOCKOPT(WS_SO_LINGER);
367 DEBUG_SOCKOPT(WS_SO_MAX_MSG_SIZE);
368 DEBUG_SOCKOPT(WS_SO_OOBINLINE);
369 DEBUG_SOCKOPT(WS_SO_OPENTYPE);
370 DEBUG_SOCKOPT(WS_SO_PROTOCOL_INFOA);
371 DEBUG_SOCKOPT(WS_SO_PROTOCOL_INFOW);
372 DEBUG_SOCKOPT(WS_SO_RCVBUF);
373 DEBUG_SOCKOPT(WS_SO_RCVTIMEO);
374 DEBUG_SOCKOPT(WS_SO_REUSEADDR);
375 DEBUG_SOCKOPT(WS_SO_SNDBUF);
376 DEBUG_SOCKOPT(WS_SO_SNDTIMEO);
377 DEBUG_SOCKOPT(WS_SO_TYPE);
378 DEBUG_SOCKOPT(WS_SO_UPDATE_CONNECT_CONTEXT);
379 }
380 break;
381
382 DEBUG_SOCKLEVEL(WS_NSPROTO_IPX);
383 switch(optname)
384 {
385 DEBUG_SOCKOPT(WS_IPX_PTYPE);
386 DEBUG_SOCKOPT(WS_IPX_FILTERPTYPE);
387 DEBUG_SOCKOPT(WS_IPX_DSTYPE);
388 DEBUG_SOCKOPT(WS_IPX_RECVHDR);
389 DEBUG_SOCKOPT(WS_IPX_MAXSIZE);
390 DEBUG_SOCKOPT(WS_IPX_ADDRESS);
391 DEBUG_SOCKOPT(WS_IPX_MAX_ADAPTER_NUM);
392 }
393 break;
394
395 DEBUG_SOCKLEVEL(WS_SOL_IRLMP);
396 switch(optname)
397 {
398 DEBUG_SOCKOPT(WS_IRLMP_ENUMDEVICES);
399 }
400 break;
401
402 DEBUG_SOCKLEVEL(WS_IPPROTO_TCP);
403 switch(optname)
404 {
405 DEBUG_SOCKOPT(WS_TCP_BSDURGENT);
406 DEBUG_SOCKOPT(WS_TCP_EXPEDITED_1122);
407 DEBUG_SOCKOPT(WS_TCP_NODELAY);
408 }
409 break;
410
411 DEBUG_SOCKLEVEL(WS_IPPROTO_IP);
412 switch(optname)
413 {
414 DEBUG_SOCKOPT(WS_IP_ADD_MEMBERSHIP);
415 DEBUG_SOCKOPT(WS_IP_DONTFRAGMENT);
416 DEBUG_SOCKOPT(WS_IP_DROP_MEMBERSHIP);
417 DEBUG_SOCKOPT(WS_IP_HDRINCL);
418 DEBUG_SOCKOPT(WS_IP_MULTICAST_IF);
419 DEBUG_SOCKOPT(WS_IP_MULTICAST_LOOP);
420 DEBUG_SOCKOPT(WS_IP_MULTICAST_TTL);
421 DEBUG_SOCKOPT(WS_IP_OPTIONS);
422 DEBUG_SOCKOPT(WS_IP_PKTINFO);
423 DEBUG_SOCKOPT(WS_IP_RECEIVE_BROADCAST);
424 DEBUG_SOCKOPT(WS_IP_TOS);
425 DEBUG_SOCKOPT(WS_IP_TTL);
426 DEBUG_SOCKOPT(WS_IP_UNICAST_IF);
427 }
428 break;
429
430 DEBUG_SOCKLEVEL(WS_IPPROTO_IPV6);
431 switch(optname)
432 {
433 DEBUG_SOCKOPT(WS_IPV6_ADD_MEMBERSHIP);
434 DEBUG_SOCKOPT(WS_IPV6_DROP_MEMBERSHIP);
435 DEBUG_SOCKOPT(WS_IPV6_MULTICAST_IF);
436 DEBUG_SOCKOPT(WS_IPV6_MULTICAST_HOPS);
437 DEBUG_SOCKOPT(WS_IPV6_MULTICAST_LOOP);
438 DEBUG_SOCKOPT(WS_IPV6_UNICAST_HOPS);
439 DEBUG_SOCKOPT(WS_IPV6_V6ONLY);
440 DEBUG_SOCKOPT(WS_IPV6_UNICAST_IF);
441 DEBUG_SOCKOPT(WS_IPV6_DONTFRAG);
442 }
443 break;
444 }
445 #undef DEBUG_SOCKLEVEL
446 #undef DEBUG_SOCKOPT
447
448 if (!strlevel)
449 strlevel = wine_dbg_sprintf("WS_0x%x", level);
450 if (!stropt)
451 stropt = wine_dbg_sprintf("WS_0x%x", optname);
452
453 return wine_dbg_sprintf("level %s, name %s", strlevel + 3, stropt + 3);
454 }
455
456 static inline const char *debugstr_optval(const char *optval, int optlenval)
457 {
458 if (optval && !IS_INTRESOURCE(optval) && optlenval >= 1 && optlenval <= sizeof(DWORD))
459 {
460 DWORD value = 0;
461 memcpy(&value, optval, optlenval);
462 return wine_dbg_sprintf("%p (%u)", optval, value);
463 }
464 return wine_dbg_sprintf("%p", optval);
465 }
466
467 /* HANDLE<->SOCKET conversion (SOCKET is UINT_PTR). */
468 #define SOCKET2HANDLE(s) ((HANDLE)(s))
469 #define HANDLE2SOCKET(h) ((SOCKET)(h))
470
471 /****************************************************************
472 * Async IO declarations
473 ****************************************************************/
474
475 typedef NTSTATUS async_callback_t( void *user, IO_STATUS_BLOCK *io, NTSTATUS status );
476
477 struct ws2_async_io
478 {
479 async_callback_t *callback; /* must be the first field */
480 struct ws2_async_io *next;
481 };
482
483 struct ws2_async_shutdown
484 {
485 struct ws2_async_io io;
486 HANDLE hSocket;
487 IO_STATUS_BLOCK iosb;
488 int type;
489 };
490
491 struct ws2_async
492 {
493 struct ws2_async_io io;
494 HANDLE hSocket;
495 LPWSAOVERLAPPED user_overlapped;
496 LPWSAOVERLAPPED_COMPLETION_ROUTINE completion_func;
497 IO_STATUS_BLOCK local_iosb;
498 struct WS_sockaddr *addr;
499 union
500 {
501 int val; /* for send operations */
502 int *ptr; /* for recv operations */
503 } addrlen;
504 DWORD flags;
505 DWORD *lpFlags;
506 WSABUF *control;
507 unsigned int n_iovecs;
508 unsigned int first_iovec;
509 struct iovec iovec[1];
510 };
511
512 struct ws2_accept_async
513 {
514 struct ws2_async_io io;
515 HANDLE listen_socket;
516 HANDLE accept_socket;
517 LPOVERLAPPED user_overlapped;
518 ULONG_PTR cvalue;
519 PVOID buf; /* buffer to write data to */
520 int data_len;
521 int local_len;
522 int remote_len;
523 struct ws2_async *read;
524 };
525
526 struct ws2_transmitfile_async
527 {
528 struct ws2_async_io io;
529 char *buffer;
530 HANDLE file;
531 DWORD file_read;
532 DWORD file_bytes;
533 DWORD bytes_per_send;
534 TRANSMIT_FILE_BUFFERS buffers;
535 DWORD flags;
536 LARGE_INTEGER offset;
537 struct ws2_async write;
538 };
539
540 static struct ws2_async_io *async_io_freelist;
541
542 static void release_async_io( struct ws2_async_io *io )
543 {
544 for (;;)
545 {
546 struct ws2_async_io *next = async_io_freelist;
547 io->next = next;
548 if (InterlockedCompareExchangePointer( (void **)&async_io_freelist, io, next ) == next) return;
549 }
550 }
551
552 static struct ws2_async_io *alloc_async_io( DWORD size, async_callback_t callback )
553 {
554 /* first free remaining previous fileinfos */
555
556 struct ws2_async_io *io = InterlockedExchangePointer( (void **)&async_io_freelist, NULL );
557
558 while (io)
559 {
560 struct ws2_async_io *next = io->next;
561 HeapFree( GetProcessHeap(), 0, io );
562 io = next;
563 }
564
565 io = HeapAlloc( GetProcessHeap(), 0, size );
566 if (io) io->callback = callback;
567 return io;
568 }
569
570 static NTSTATUS register_async( int type, HANDLE handle, struct ws2_async_io *async, HANDLE event,
571 PIO_APC_ROUTINE apc, void *apc_context, IO_STATUS_BLOCK *io )
572 {
573 NTSTATUS status;
574
575 SERVER_START_REQ( register_async )
576 {
577 req->type = type;
578 req->async.handle = wine_server_obj_handle( handle );
579 req->async.user = wine_server_client_ptr( async );
580 req->async.iosb = wine_server_client_ptr( io );
581 req->async.event = wine_server_obj_handle( event );
582 req->async.apc = wine_server_client_ptr( apc );
583 req->async.apc_context = wine_server_client_ptr( apc_context );
584 status = wine_server_call( req );
585 }
586 SERVER_END_REQ;
587
588 return status;
589 }
590
591 /****************************************************************/
592
593 /* ----------------------------------- internal data */
594
595 /* ws_... struct conversion flags */
596
597 typedef struct /* WSAAsyncSelect() control struct */
598 {
599 HANDLE service, event, sock;
600 HWND hWnd;
601 UINT uMsg;
602 LONG lEvent;
603 } ws_select_info;
604
605 #define WS_MAX_SOCKETS_PER_PROCESS 128 /* reasonable guess */
606 #define WS_MAX_UDP_DATAGRAM 1024
607 static INT WINAPI WSA_DefaultBlockingHook( FARPROC x );
608
609 /* hostent's, servent's and protent's are stored in one buffer per thread,
610 * as documented on MSDN for the functions that return any of the buffers */
611 struct per_thread_data
612 {
613 int opentype;
614 struct WS_hostent *he_buffer;
615 struct WS_servent *se_buffer;
616 struct WS_protoent *pe_buffer;
617 struct pollfd *fd_cache;
618 unsigned int fd_count;
619 int he_len;
620 int se_len;
621 int pe_len;
622 char ntoa_buffer[16]; /* 4*3 digits + 3 '.' + 1 '\0' */
623 };
624
625 /* internal: routing description information */
626 struct route {
627 struct in_addr addr;
628 IF_INDEX interface;
629 DWORD metric, default_route;
630 };
631
632 static INT num_startup; /* reference counter */
633 static FARPROC blocking_hook = (FARPROC)WSA_DefaultBlockingHook;
634
635 /* function prototypes */
636 static struct WS_hostent *WS_create_he(char *name, int aliases, int aliases_size, int addresses, int address_length);
637 static struct WS_hostent *WS_dup_he(const struct hostent* p_he);
638 static struct WS_protoent *WS_create_pe( const char *name, char **aliases, int prot );
639 static struct WS_servent *WS_dup_se(const struct servent* p_se);
640 static int ws_protocol_info(SOCKET s, int unicode, WSAPROTOCOL_INFOW *buffer, int *size);
641
642 int WSAIOCTL_GetInterfaceCount(void);
643 int WSAIOCTL_GetInterfaceName(int intNumber, char *intName);
644
645 static void WS_AddCompletion( SOCKET sock, ULONG_PTR CompletionValue, NTSTATUS CompletionStatus, ULONG Information, BOOL force );
646
647 #define MAP_OPTION(opt) { WS_##opt, opt }
648
649 static const int ws_flags_map[][2] =
650 {
651 MAP_OPTION( MSG_OOB ),
652 MAP_OPTION( MSG_PEEK ),
653 MAP_OPTION( MSG_DONTROUTE ),
654 MAP_OPTION( MSG_WAITALL ),
655 { WS_MSG_PARTIAL, 0 },
656 };
657
658 static const int ws_sock_map[][2] =
659 {
660 MAP_OPTION( SO_DEBUG ),
661 MAP_OPTION( SO_ACCEPTCONN ),
662 MAP_OPTION( SO_REUSEADDR ),
663 MAP_OPTION( SO_KEEPALIVE ),
664 MAP_OPTION( SO_DONTROUTE ),
665 MAP_OPTION( SO_BROADCAST ),
666 MAP_OPTION( SO_LINGER ),
667 MAP_OPTION( SO_OOBINLINE ),
668 MAP_OPTION( SO_SNDBUF ),
669 MAP_OPTION( SO_RCVBUF ),
670 MAP_OPTION( SO_ERROR ),
671 MAP_OPTION( SO_TYPE ),
672 #ifdef SO_RCVTIMEO
673 MAP_OPTION( SO_RCVTIMEO ),
674 #endif
675 #ifdef SO_SNDTIMEO
676 MAP_OPTION( SO_SNDTIMEO ),
677 #endif
678 };
679
680 static const int ws_tcp_map[][2] =
681 {
682 #ifdef TCP_NODELAY
683 MAP_OPTION( TCP_NODELAY ),
684 #endif
685 };
686
687 static const int ws_ip_map[][2] =
688 {
689 MAP_OPTION( IP_MULTICAST_IF ),
690 MAP_OPTION( IP_MULTICAST_TTL ),
691 MAP_OPTION( IP_MULTICAST_LOOP ),
692 MAP_OPTION( IP_ADD_MEMBERSHIP ),
693 MAP_OPTION( IP_DROP_MEMBERSHIP ),
694 MAP_OPTION( IP_OPTIONS ),
695 #ifdef IP_HDRINCL
696 MAP_OPTION( IP_HDRINCL ),
697 #endif
698 MAP_OPTION( IP_TOS ),
699 MAP_OPTION( IP_TTL ),
700 #ifdef IP_PKTINFO
701 MAP_OPTION( IP_PKTINFO ),
702 #endif
703 #ifdef IP_UNICAST_IF
704 MAP_OPTION( IP_UNICAST_IF ),
705 #endif
706 };
707
708 static const int ws_ipv6_map[][2] =
709 {
710 #ifdef IPV6_ADD_MEMBERSHIP
711 MAP_OPTION( IPV6_ADD_MEMBERSHIP ),
712 #endif
713 #ifdef IPV6_DROP_MEMBERSHIP
714 MAP_OPTION( IPV6_DROP_MEMBERSHIP ),
715 #endif
716 MAP_OPTION( IPV6_MULTICAST_IF ),
717 MAP_OPTION( IPV6_MULTICAST_HOPS ),
718 MAP_OPTION( IPV6_MULTICAST_LOOP ),
719 MAP_OPTION( IPV6_UNICAST_HOPS ),
720 MAP_OPTION( IPV6_V6ONLY ),
721 #ifdef IPV6_UNICAST_IF
722 MAP_OPTION( IPV6_UNICAST_IF ),
723 #endif
724 };
725
726 static const int ws_af_map[][2] =
727 {
728 MAP_OPTION( AF_UNSPEC ),
729 MAP_OPTION( AF_INET ),
730 MAP_OPTION( AF_INET6 ),
731 #ifdef HAS_IPX
732 MAP_OPTION( AF_IPX ),
733 #endif
734 #ifdef AF_IRDA
735 MAP_OPTION( AF_IRDA ),
736 #endif
737 {FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO},
738 };
739
740 static const int ws_socktype_map[][2] =
741 {
742 MAP_OPTION( SOCK_DGRAM ),
743 MAP_OPTION( SOCK_STREAM ),
744 MAP_OPTION( SOCK_RAW ),
745 {FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO},
746 };
747
748 static const int ws_proto_map[][2] =
749 {
750 MAP_OPTION( IPPROTO_IP ),
751 MAP_OPTION( IPPROTO_TCP ),
752 MAP_OPTION( IPPROTO_UDP ),
753 MAP_OPTION( IPPROTO_IPV6 ),
754 MAP_OPTION( IPPROTO_ICMP ),
755 MAP_OPTION( IPPROTO_IGMP ),
756 MAP_OPTION( IPPROTO_RAW ),
757 {FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO},
758 };
759
760 static const int ws_aiflag_map[][2] =
761 {
762 MAP_OPTION( AI_PASSIVE ),
763 MAP_OPTION( AI_CANONNAME ),
764 MAP_OPTION( AI_NUMERICHOST ),
765 #ifdef AI_NUMERICSERV
766 MAP_OPTION( AI_NUMERICSERV ),
767 #endif
768 #ifdef AI_V4MAPPED
769 MAP_OPTION( AI_V4MAPPED ),
770 #endif
771 MAP_OPTION( AI_ADDRCONFIG ),
772 };
773
774 static const int ws_niflag_map[][2] =
775 {
776 MAP_OPTION( NI_NOFQDN ),
777 MAP_OPTION( NI_NUMERICHOST ),
778 MAP_OPTION( NI_NAMEREQD ),
779 MAP_OPTION( NI_NUMERICSERV ),
780 MAP_OPTION( NI_DGRAM ),
781 };
782
783 static const int ws_eai_map[][2] =
784 {
785 MAP_OPTION( EAI_AGAIN ),
786 MAP_OPTION( EAI_BADFLAGS ),
787 MAP_OPTION( EAI_FAIL ),
788 MAP_OPTION( EAI_FAMILY ),
789 MAP_OPTION( EAI_MEMORY ),
790 /* Note: EAI_NODATA is deprecated, but still
791 * used by Windows and Linux... We map the newer
792 * EAI_NONAME to EAI_NODATA for now until Windows
793 * changes too.
794 */
795 #ifdef EAI_NODATA
796 MAP_OPTION( EAI_NODATA ),
797 #endif
798 #ifdef EAI_NONAME
799 { WS_EAI_NODATA, EAI_NONAME },
800 #endif
801
802 MAP_OPTION( EAI_SERVICE ),
803 MAP_OPTION( EAI_SOCKTYPE ),
804 { 0, 0 }
805 };
806
807 static const int ws_poll_map[][2] =
808 {
809 MAP_OPTION( POLLERR ),
810 MAP_OPTION( POLLHUP ),
811 MAP_OPTION( POLLNVAL ),
812 MAP_OPTION( POLLWRNORM ),
813 MAP_OPTION( POLLWRBAND ),
814 MAP_OPTION( POLLRDNORM ),
815 { WS_POLLRDBAND, POLLPRI }
816 };
817
818 static const char magic_loopback_addr[] = {127, 12, 34, 56};
819
820 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
821 #ifdef IP_PKTINFO
822 static inline WSACMSGHDR *fill_control_message(int level, int type, WSACMSGHDR *current, ULONG *maxsize, void *data, int len)
823 {
824 ULONG msgsize = sizeof(WSACMSGHDR) + WSA_CMSG_ALIGN(len);
825 char *ptr = (char *) current + sizeof(WSACMSGHDR);
826
827 /* Make sure there is at least enough room for this entry */
828 if (msgsize > *maxsize)
829 return NULL;
830 *maxsize -= msgsize;
831 /* Fill in the entry */
832 current->cmsg_len = sizeof(WSACMSGHDR) + len;
833 current->cmsg_level = level;
834 current->cmsg_type = type;
835 memcpy(ptr, data, len);
836 /* Return the pointer to where next entry should go */
837 return (WSACMSGHDR *) (ptr + WSA_CMSG_ALIGN(len));
838 }
839 #endif /* IP_PKTINFO */
840
841 static inline int convert_control_headers(struct msghdr *hdr, WSABUF *control)
842 {
843 #ifdef IP_PKTINFO
844 WSACMSGHDR *cmsg_win = (WSACMSGHDR *) control->buf, *ptr;
845 ULONG ctlsize = control->len;
846 struct cmsghdr *cmsg_unix;
847
848 ptr = cmsg_win;
849 /* Loop over all the headers, converting as appropriate */
850 for (cmsg_unix = CMSG_FIRSTHDR(hdr); cmsg_unix != NULL; cmsg_unix = CMSG_NXTHDR(hdr, cmsg_unix))
851 {
852 switch(cmsg_unix->cmsg_level)
853 {
854 case IPPROTO_IP:
855 switch(cmsg_unix->cmsg_type)
856 {
857 case IP_PKTINFO:
858 {
859 /* Convert the Unix IP_PKTINFO structure to the Windows version */
860 struct in_pktinfo *data_unix = (struct in_pktinfo *) CMSG_DATA(cmsg_unix);
861 struct WS_in_pktinfo data_win;
862
863 memcpy(&data_win.ipi_addr,&data_unix->ipi_addr.s_addr,4); /* 4 bytes = 32 address bits */
864 data_win.ipi_ifindex = data_unix->ipi_ifindex;
865 ptr = fill_control_message(WS_IPPROTO_IP, WS_IP_PKTINFO, ptr, &ctlsize,
866 (void*)&data_win, sizeof(data_win));
867 if (!ptr) goto error;
868 } break;
869 default:
870 FIXME("Unhandled IPPROTO_IP message header type %d\n", cmsg_unix->cmsg_type);
871 break;
872 }
873 break;
874 default:
875 FIXME("Unhandled message header level %d\n", cmsg_unix->cmsg_level);
876 break;
877 }
878 }
879
880 error:
881 /* Set the length of the returned control headers */
882 control->len = (ptr == NULL ? 0 : (char*)ptr - (char*)cmsg_win);
883 return (ptr != NULL);
884 #else /* IP_PKTINFO */
885 control->len = 0;
886 return 1;
887 #endif /* IP_PKTINFO */
888 }
889 #endif /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
890
891 /* ----------------------------------- error handling */
892
893 static NTSTATUS sock_get_ntstatus( int err )
894 {
895 switch ( err )
896 {
897 case EBADF: return STATUS_INVALID_HANDLE;
898 case EBUSY: return STATUS_DEVICE_BUSY;
899 case EPERM:
900 case EACCES: return STATUS_ACCESS_DENIED;
901 case EFAULT: return STATUS_NO_MEMORY;
902 case EINVAL: return STATUS_INVALID_PARAMETER;
903 case ENFILE:
904 case EMFILE: return STATUS_TOO_MANY_OPENED_FILES;
905 case EWOULDBLOCK: return STATUS_CANT_WAIT;
906 case EINPROGRESS: return STATUS_PENDING;
907 case EALREADY: return STATUS_NETWORK_BUSY;
908 case ENOTSOCK: return STATUS_OBJECT_TYPE_MISMATCH;
909 case EDESTADDRREQ: return STATUS_INVALID_PARAMETER;
910 case EMSGSIZE: return STATUS_BUFFER_OVERFLOW;
911 case EPROTONOSUPPORT:
912 case ESOCKTNOSUPPORT:
913 case EPFNOSUPPORT:
914 case EAFNOSUPPORT:
915 case EPROTOTYPE: return STATUS_NOT_SUPPORTED;
916 case ENOPROTOOPT: return STATUS_INVALID_PARAMETER;
917 case EOPNOTSUPP: return STATUS_NOT_SUPPORTED;
918 case EADDRINUSE: return STATUS_ADDRESS_ALREADY_ASSOCIATED;
919 case EADDRNOTAVAIL: return STATUS_INVALID_PARAMETER;
920 case ECONNREFUSED: return STATUS_CONNECTION_REFUSED;
921 case ESHUTDOWN: return STATUS_PIPE_DISCONNECTED;
922 case ENOTCONN: return STATUS_CONNECTION_DISCONNECTED;
923 case ETIMEDOUT: return STATUS_IO_TIMEOUT;
924 case ENETUNREACH: return STATUS_NETWORK_UNREACHABLE;
925 case ENETDOWN: return STATUS_NETWORK_BUSY;
926 case EPIPE:
927 case ECONNRESET: return STATUS_CONNECTION_RESET;
928 case ECONNABORTED: return STATUS_CONNECTION_ABORTED;
929
930 case 0: return STATUS_SUCCESS;
931 default:
932 WARN("Unknown errno %d!\n", err);
933 return STATUS_UNSUCCESSFUL;
934 }
935 }
936
937 static UINT sock_get_error( int err )
938 {
939 switch(err)
940 {
941 case EINTR: return WSAEINTR;
942 case EPERM:
943 case EACCES: return WSAEACCES;
944 case EFAULT: return WSAEFAULT;
945 case EINVAL: return WSAEINVAL;
946 case EMFILE: return WSAEMFILE;
947 case EWOULDBLOCK: return WSAEWOULDBLOCK;
948 case EINPROGRESS: return WSAEINPROGRESS;
949 case EALREADY: return WSAEALREADY;
950 case EBADF:
951 case ENOTSOCK: return WSAENOTSOCK;
952 case EDESTADDRREQ: return WSAEDESTADDRREQ;
953 case EMSGSIZE: return WSAEMSGSIZE;
954 case EPROTOTYPE: return WSAEPROTOTYPE;
955 case ENOPROTOOPT: return WSAENOPROTOOPT;
956 case EPROTONOSUPPORT: return WSAEPROTONOSUPPORT;
957 case ESOCKTNOSUPPORT: return WSAESOCKTNOSUPPORT;
958 case EOPNOTSUPP: return WSAEOPNOTSUPP;
959 case EPFNOSUPPORT: return WSAEPFNOSUPPORT;
960 case EAFNOSUPPORT: return WSAEAFNOSUPPORT;
961 case EADDRINUSE: return WSAEADDRINUSE;
962 case EADDRNOTAVAIL: return WSAEADDRNOTAVAIL;
963 case ENETDOWN: return WSAENETDOWN;
964 case ENETUNREACH: return WSAENETUNREACH;
965 case ENETRESET: return WSAENETRESET;
966 case ECONNABORTED: return WSAECONNABORTED;
967 case EPIPE:
968 case ECONNRESET: return WSAECONNRESET;
969 case ENOBUFS: return WSAENOBUFS;
970 case EISCONN: return WSAEISCONN;
971 case ENOTCONN: return WSAENOTCONN;
972 case ESHUTDOWN: return WSAESHUTDOWN;
973 case ETOOMANYREFS: return WSAETOOMANYREFS;
974 case ETIMEDOUT: return WSAETIMEDOUT;
975 case ECONNREFUSED: return WSAECONNREFUSED;
976 case ELOOP: return WSAELOOP;
977 case ENAMETOOLONG: return WSAENAMETOOLONG;
978 case EHOSTDOWN: return WSAEHOSTDOWN;
979 case EHOSTUNREACH: return WSAEHOSTUNREACH;
980 case ENOTEMPTY: return WSAENOTEMPTY;
981 #ifdef EPROCLIM
982 case EPROCLIM: return WSAEPROCLIM;
983 #endif
984 #ifdef EUSERS
985 case EUSERS: return WSAEUSERS;
986 #endif
987 #ifdef EDQUOT
988 case EDQUOT: return WSAEDQUOT;
989 #endif
990 #ifdef ESTALE
991 case ESTALE: return WSAESTALE;
992 #endif
993 #ifdef EREMOTE
994 case EREMOTE: return WSAEREMOTE;
995 #endif
996
997 /* just in case we ever get here and there are no problems */
998 case 0: return 0;
999 default:
1000 WARN("Unknown errno %d!\n", err);
1001 return WSAEOPNOTSUPP;
1002 }
1003 }
1004
1005 static UINT wsaErrno(void)
1006 {
1007 int loc_errno = errno;
1008 WARN("errno %d, (%s).\n", loc_errno, strerror(loc_errno));
1009
1010 return sock_get_error( loc_errno );
1011 }
1012
1013 /* most ws2 overlapped functions return an ntstatus-based error code */
1014 static NTSTATUS wsaErrStatus(void)
1015 {
1016 int loc_errno = errno;
1017 WARN("errno %d, (%s).\n", loc_errno, strerror(loc_errno));
1018
1019 return sock_get_ntstatus(loc_errno);
1020 }
1021
1022 static UINT wsaHerrno(int loc_errno)
1023 {
1024 WARN("h_errno %d.\n", loc_errno);
1025
1026 switch(loc_errno)
1027 {
1028 case HOST_NOT_FOUND: return WSAHOST_NOT_FOUND;
1029 case TRY_AGAIN: return WSATRY_AGAIN;
1030 case NO_RECOVERY: return WSANO_RECOVERY;
1031 case NO_DATA: return WSANO_DATA;
1032 case ENOBUFS: return WSAENOBUFS;
1033
1034 case 0: return 0;
1035 default:
1036 WARN("Unknown h_errno %d!\n", loc_errno);
1037 return WSAEOPNOTSUPP;
1038 }
1039 }
1040
1041 static NTSTATUS sock_error_to_ntstatus( DWORD err )
1042 {
1043 switch (err)
1044 {
1045 case 0: return STATUS_SUCCESS;
1046 case WSAEBADF: return STATUS_INVALID_HANDLE;
1047 case WSAEACCES: return STATUS_ACCESS_DENIED;
1048 case WSAEFAULT: return STATUS_NO_MEMORY;
1049 case WSAEINVAL: return STATUS_INVALID_PARAMETER;
1050 case WSAEMFILE: return STATUS_TOO_MANY_OPENED_FILES;
1051 case WSAEWOULDBLOCK: return STATUS_CANT_WAIT;
1052 case WSAEINPROGRESS: return STATUS_PENDING;
1053 case WSAEALREADY: return STATUS_NETWORK_BUSY;
1054 case WSAENOTSOCK: return STATUS_OBJECT_TYPE_MISMATCH;
1055 case WSAEDESTADDRREQ: return STATUS_INVALID_PARAMETER;
1056 case WSAEMSGSIZE: return STATUS_BUFFER_OVERFLOW;
1057 case WSAEPROTONOSUPPORT:
1058 case WSAESOCKTNOSUPPORT:
1059 case WSAEPFNOSUPPORT:
1060 case WSAEAFNOSUPPORT:
1061 case WSAEPROTOTYPE: return STATUS_NOT_SUPPORTED;
1062 case WSAENOPROTOOPT: return STATUS_INVALID_PARAMETER;
1063 case WSAEOPNOTSUPP: return STATUS_NOT_SUPPORTED;
1064 case WSAEADDRINUSE: return STATUS_ADDRESS_ALREADY_ASSOCIATED;
1065 case WSAEADDRNOTAVAIL: return STATUS_INVALID_PARAMETER;
1066 case WSAECONNREFUSED: return STATUS_CONNECTION_REFUSED;
1067 case WSAESHUTDOWN: return STATUS_PIPE_DISCONNECTED;
1068 case WSAENOTCONN: return STATUS_CONNECTION_DISCONNECTED;
1069 case WSAETIMEDOUT: return STATUS_IO_TIMEOUT;
1070 case WSAENETUNREACH: return STATUS_NETWORK_UNREACHABLE;
1071 case WSAENETDOWN: return STATUS_NETWORK_BUSY;
1072 case WSAECONNRESET: return STATUS_CONNECTION_RESET;
1073 case WSAECONNABORTED: return STATUS_CONNECTION_ABORTED;
1074 default:
1075 FIXME("unmapped error %u\n", err);
1076 return STATUS_UNSUCCESSFUL;
1077 }
1078 }
1079
1080 static DWORD NtStatusToWSAError( DWORD status )
1081 {
1082 switch ( status )
1083 {
1084 case STATUS_SUCCESS: return 0;
1085 case STATUS_PENDING: return WSA_IO_PENDING;
1086 case STATUS_OBJECT_TYPE_MISMATCH: return WSAENOTSOCK;
1087 case STATUS_INVALID_HANDLE: return WSAEBADF;
1088 case STATUS_INVALID_PARAMETER: return WSAEINVAL;
1089 case STATUS_PIPE_DISCONNECTED: return WSAESHUTDOWN;
1090 case STATUS_NETWORK_BUSY: return WSAEALREADY;
1091 case STATUS_NETWORK_UNREACHABLE: return WSAENETUNREACH;
1092 case STATUS_CONNECTION_REFUSED: return WSAECONNREFUSED;
1093 case STATUS_CONNECTION_DISCONNECTED: return WSAENOTCONN;
1094 case STATUS_CONNECTION_RESET: return WSAECONNRESET;
1095 case STATUS_CONNECTION_ABORTED: return WSAECONNABORTED;
1096 case STATUS_CANCELLED: return WSA_OPERATION_ABORTED;
1097 case STATUS_ADDRESS_ALREADY_ASSOCIATED: return WSAEADDRINUSE;
1098 case STATUS_IO_TIMEOUT:
1099 case STATUS_TIMEOUT: return WSAETIMEDOUT;
1100 case STATUS_NO_MEMORY: return WSAEFAULT;
1101 case STATUS_ACCESS_DENIED: return WSAEACCES;
1102 case STATUS_TOO_MANY_OPENED_FILES: return WSAEMFILE;
1103 case STATUS_CANT_WAIT: return WSAEWOULDBLOCK;
1104 case STATUS_BUFFER_OVERFLOW: return WSAEMSGSIZE;
1105 case STATUS_NOT_SUPPORTED: return WSAEOPNOTSUPP;
1106 case STATUS_HOST_UNREACHABLE: return WSAEHOSTUNREACH;
1107 default: return RtlNtStatusToDosError( status );
1108 }
1109 }
1110
1111 /* set last error code from NT status without mapping WSA errors */
1112 static inline unsigned int set_error( unsigned int err )
1113 {
1114 if (err)
1115 {
1116 err = NtStatusToWSAError( err );
1117 SetLastError( err );
1118 }
1119 return err;
1120 }
1121
1122 static inline int get_sock_fd( SOCKET s, DWORD access, unsigned int *options )
1123 {
1124 int fd;
1125 if (set_error( wine_server_handle_to_fd( SOCKET2HANDLE(s), access, &fd, options ) ))
1126 return -1;
1127 return fd;
1128 }
1129
1130 static inline void release_sock_fd( SOCKET s, int fd )
1131 {
1132 wine_server_release_fd( SOCKET2HANDLE(s), fd );
1133 }
1134
1135 static void _enable_event( HANDLE s, unsigned int event,
1136 unsigned int sstate, unsigned int cstate )
1137 {
1138 SERVER_START_REQ( enable_socket_event )
1139 {
1140 req->handle = wine_server_obj_handle( s );
1141 req->mask = event;
1142 req->sstate = sstate;
1143 req->cstate = cstate;
1144 wine_server_call( req );
1145 }
1146 SERVER_END_REQ;
1147 }
1148
1149 static NTSTATUS _is_blocking(SOCKET s, BOOL *ret)
1150 {
1151 NTSTATUS status;
1152 SERVER_START_REQ( get_socket_event )
1153 {
1154 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
1155 req->service = FALSE;
1156 req->c_event = 0;
1157 status = wine_server_call( req );
1158 *ret = (reply->state & FD_WINE_NONBLOCKING) == 0;
1159 }
1160 SERVER_END_REQ;
1161 return status;
1162 }
1163
1164 static unsigned int _get_sock_mask(SOCKET s)
1165 {
1166 unsigned int ret;
1167 SERVER_START_REQ( get_socket_event )
1168 {
1169 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
1170 req->service = FALSE;
1171 req->c_event = 0;
1172 wine_server_call( req );
1173 ret = reply->mask;
1174 }
1175 SERVER_END_REQ;
1176 return ret;
1177 }
1178
1179 static void _sync_sock_state(SOCKET s)
1180 {
1181 BOOL dummy;
1182 /* do a dummy wineserver request in order to let
1183 the wineserver run through its select loop once */
1184 (void)_is_blocking(s, &dummy);
1185 }
1186
1187 static void _get_sock_errors(SOCKET s, int *events)
1188 {
1189 SERVER_START_REQ( get_socket_event )
1190 {
1191 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
1192 req->service = FALSE;
1193 req->c_event = 0;
1194 wine_server_set_reply( req, events, sizeof(int) * FD_MAX_EVENTS );
1195 wine_server_call( req );
1196 }
1197 SERVER_END_REQ;
1198 }
1199
1200 static int get_sock_error(SOCKET s, unsigned int bit)
1201 {
1202 int events[FD_MAX_EVENTS];
1203 _get_sock_errors(s, events);
1204 return events[bit];
1205 }
1206
1207 static int _get_fd_type(int fd)
1208 {
1209 int sock_type = -1;
1210 socklen_t optlen = sizeof(sock_type);
1211 getsockopt(fd, SOL_SOCKET, SO_TYPE, (char*) &sock_type, &optlen);
1212 return sock_type;
1213 }
1214
1215 static BOOL set_dont_fragment(SOCKET s, int level, BOOL value)
1216 {
1217 int fd, optname;
1218
1219 if (level == IPPROTO_IP)
1220 {
1221 #ifdef IP_DONTFRAG
1222 optname = IP_DONTFRAG;
1223 #elif defined(IP_MTU_DISCOVER) && defined(IP_PMTUDISC_DO) && defined(IP_PMTUDISC_DONT)
1224 optname = IP_MTU_DISCOVER;
1225 value = value ? IP_PMTUDISC_DO : IP_PMTUDISC_DONT;
1226 #else
1227 static int once;
1228 if (!once++)
1229 FIXME("IP_DONTFRAGMENT for IPv4 not supported in this platform\n");
1230 return TRUE; /* fake success */
1231 #endif
1232 }
1233 else
1234 {
1235 #ifdef IPV6_DONTFRAG
1236 optname = IPV6_DONTFRAG;
1237 #elif defined(IPV6_MTU_DISCOVER) && defined(IPV6_PMTUDISC_DO) && defined(IPV6_PMTUDISC_DONT)
1238 optname = IPV6_MTU_DISCOVER;
1239 value = value ? IPV6_PMTUDISC_DO : IPV6_PMTUDISC_DONT;
1240 #else
1241 static int once;
1242 if (!once++)
1243 FIXME("IP_DONTFRAGMENT for IPv6 not supported in this platform\n");
1244 return TRUE; /* fake success */
1245 #endif
1246 }
1247
1248 fd = get_sock_fd(s, 0, NULL);
1249 if (fd == -1) return FALSE;
1250
1251 if (!setsockopt(fd, level, optname, &value, sizeof(value)))
1252 value = TRUE;
1253 else
1254 {
1255 WSASetLastError(wsaErrno());
1256 value = FALSE;
1257 }
1258
1259 release_sock_fd(s, fd);
1260 return value;
1261 }
1262
1263 static BOOL get_dont_fragment(SOCKET s, int level, BOOL *out)
1264 {
1265 int fd, optname, value, not_expected;
1266 socklen_t optlen = sizeof(value);
1267
1268 if (level == IPPROTO_IP)
1269 {
1270 #ifdef IP_DONTFRAG
1271 optname = IP_DONTFRAG;
1272 not_expected = 0;
1273 #elif defined(IP_MTU_DISCOVER) && defined(IP_PMTUDISC_DONT)
1274 optname = IP_MTU_DISCOVER;
1275 not_expected = IP_PMTUDISC_DONT;
1276 #else
1277 static int once;
1278 if (!once++)
1279 FIXME("IP_DONTFRAGMENT for IPv4 not supported in this platform\n");
1280 return TRUE; /* fake success */
1281 #endif
1282 }
1283 else
1284 {
1285 #ifdef IPV6_DONTFRAG
1286 optname = IPV6_DONTFRAG;
1287 not_expected = 0;
1288 #elif defined(IPV6_MTU_DISCOVER) && defined(IPV6_PMTUDISC_DONT)
1289 optname = IPV6_MTU_DISCOVER;
1290 not_expected = IPV6_PMTUDISC_DONT;
1291 #else
1292 static int once;
1293 if (!once++)
1294 FIXME("IP_DONTFRAGMENT for IPv6 not supported in this platform\n");
1295 return TRUE; /* fake success */
1296 #endif
1297 }
1298
1299 fd = get_sock_fd(s, 0, NULL);
1300 if (fd == -1) return FALSE;
1301
1302 if (!getsockopt(fd, level, optname, &value, &optlen))
1303 {
1304 *out = value != not_expected;
1305 value = TRUE;
1306 }
1307 else
1308 {
1309 WSASetLastError(wsaErrno());
1310 value = FALSE;
1311 }
1312
1313 release_sock_fd(s, fd);
1314 return value;
1315 }
1316
1317 static struct per_thread_data *get_per_thread_data(void)
1318 {
1319 struct per_thread_data * ptb = NtCurrentTeb()->WinSockData;
1320 /* lazy initialization */
1321 if (!ptb)
1322 {
1323 ptb = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*ptb) );
1324 NtCurrentTeb()->WinSockData = ptb;
1325 }
1326 return ptb;
1327 }
1328
1329 static void free_per_thread_data(void)
1330 {
1331 struct per_thread_data * ptb = NtCurrentTeb()->WinSockData;
1332
1333 if (!ptb) return;
1334
1335 /* delete scratch buffers */
1336 HeapFree( GetProcessHeap(), 0, ptb->he_buffer );
1337 HeapFree( GetProcessHeap(), 0, ptb->se_buffer );
1338 HeapFree( GetProcessHeap(), 0, ptb->pe_buffer );
1339 HeapFree( GetProcessHeap(), 0, ptb->fd_cache );
1340
1341 HeapFree( GetProcessHeap(), 0, ptb );
1342 NtCurrentTeb()->WinSockData = NULL;
1343 }
1344
1345 /***********************************************************************
1346 * DllMain (WS2_32.init)
1347 */
1348 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID fImpLoad)
1349 {
1350 TRACE("%p 0x%x %p\n", hInstDLL, fdwReason, fImpLoad);
1351 switch (fdwReason) {
1352 case DLL_PROCESS_ATTACH:
1353 break;
1354 case DLL_PROCESS_DETACH:
1355 if (fImpLoad) break;
1356 free_per_thread_data();
1357 DeleteCriticalSection(&csWSgetXXXbyYYY);
1358 break;
1359 case DLL_THREAD_DETACH:
1360 free_per_thread_data();
1361 break;
1362 }
1363 return TRUE;
1364 }
1365
1366 /***********************************************************************
1367 * convert_flags()
1368 *
1369 * Converts send/recv flags from Windows format.
1370 * Return the converted flag bits, unsupported flags remain unchanged.
1371 */
1372 static int convert_flags(int flags)
1373 {
1374 int i, out;
1375 if (!flags) return 0;
1376
1377 for (out = i = 0; flags && i < ARRAY_SIZE(ws_flags_map); i++)
1378 {
1379 if (ws_flags_map[i][0] & flags)
1380 {
1381 out |= ws_flags_map[i][1];
1382 flags &= ~ws_flags_map[i][0];
1383 }
1384 }
1385 if (flags)
1386 {
1387 FIXME("Unknown send/recv flags 0x%x, using anyway...\n", flags);
1388 out |= flags;
1389 }
1390 return out;
1391 }
1392
1393 /***********************************************************************
1394 * convert_sockopt()
1395 *
1396 * Converts socket flags from Windows format.
1397 * Return 1 if converted, 0 if not (error).
1398 */
1399 static int convert_sockopt(INT *level, INT *optname)
1400 {
1401 unsigned int i;
1402 switch (*level)
1403 {
1404 case WS_SOL_SOCKET:
1405 *level = SOL_SOCKET;
1406 for(i = 0; i < ARRAY_SIZE(ws_sock_map); i++) {
1407 if( ws_sock_map[i][0] == *optname )
1408 {
1409 *optname = ws_sock_map[i][1];
1410 return 1;
1411 }
1412 }
1413 FIXME("Unknown SOL_SOCKET optname 0x%x\n", *optname);
1414 break;
1415 case WS_IPPROTO_TCP:
1416 *level = IPPROTO_TCP;
1417 for(i = 0; i < ARRAY_SIZE(ws_tcp_map); i++) {
1418 if ( ws_tcp_map[i][0] == *optname )
1419 {
1420 *optname = ws_tcp_map[i][1];
1421 return 1;
1422 }
1423 }
1424 FIXME("Unknown IPPROTO_TCP optname 0x%x\n", *optname);
1425 break;
1426 case WS_IPPROTO_IP:
1427 *level = IPPROTO_IP;
1428 for(i = 0; i < ARRAY_SIZE(ws_ip_map); i++) {
1429 if (ws_ip_map[i][0] == *optname )
1430 {
1431 *optname = ws_ip_map[i][1];
1432 return 1;
1433 }
1434 }
1435 FIXME("Unknown IPPROTO_IP optname 0x%x\n", *optname);
1436 break;
1437 case WS_IPPROTO_IPV6:
1438 *level = IPPROTO_IPV6;
1439 for(i = 0; i < ARRAY_SIZE(ws_ipv6_map); i++) {
1440 if (ws_ipv6_map[i][0] == *optname )
1441 {
1442 *optname = ws_ipv6_map[i][1];
1443 return 1;
1444 }
1445 }
1446 FIXME("Unknown IPPROTO_IPV6 optname 0x%x\n", *optname);
1447 break;
1448 default: FIXME("Unimplemented or unknown socket level\n");
1449 }
1450 return 0;
1451 }
1452
1453 /* ----------------------------------- Per-thread info (or per-process?) */
1454
1455 static char *strdup_lower(const char *str)
1456 {
1457 int i;
1458 char *ret = HeapAlloc( GetProcessHeap(), 0, strlen(str) + 1 );
1459
1460 if (ret)
1461 {
1462 for (i = 0; str[i]; i++) ret[i] = tolower(str[i]);
1463 ret[i] = 0;
1464 }
1465 else SetLastError(WSAENOBUFS);
1466 return ret;
1467 }
1468
1469 /* Utility: get the SO_RCVTIMEO or SO_SNDTIMEO socket option
1470 * from an fd and return the value converted to milli seconds
1471 * or 0 if there is an infinite time out */
1472 static inline INT64 get_rcvsnd_timeo( int fd, BOOL is_recv)
1473 {
1474 struct timeval tv;
1475 socklen_t len = sizeof(tv);
1476 int optname, res;
1477
1478 if (is_recv)
1479 #ifdef SO_RCVTIMEO
1480 optname = SO_RCVTIMEO;
1481 #else
1482 return 0;
1483 #endif
1484 else
1485 #ifdef SO_SNDTIMEO
1486 optname = SO_SNDTIMEO;
1487 #else
1488 return 0;
1489 #endif
1490
1491 res = getsockopt(fd, SOL_SOCKET, optname, &tv, &len);
1492 if (res < 0)
1493 return 0;
1494 return (UINT64)tv.tv_sec * 1000 + tv.tv_usec / 1000;
1495 }
1496
1497 /* utility: given an fd, will block until one of the events occurs */
1498 static inline int do_block( int fd, int events, int timeout )
1499 {
1500 struct pollfd pfd;
1501 int ret;
1502
1503 pfd.fd = fd;
1504 pfd.events = events;
1505
1506 while ((ret = poll(&pfd, 1, timeout)) < 0)
1507 {
1508 if (errno != EINTR)
1509 return -1;
1510 }
1511 if( ret == 0 )
1512 return 0;
1513 return pfd.revents;
1514 }
1515
1516 static int
1517 convert_af_w2u(int windowsaf) {
1518 unsigned int i;
1519
1520 for (i = 0; i < ARRAY_SIZE(ws_af_map); i++)
1521 if (ws_af_map[i][0] == windowsaf)
1522 return ws_af_map[i][1];
1523 FIXME("unhandled Windows address family %d\n", windowsaf);
1524 return -1;
1525 }
1526
1527 static int
1528 convert_af_u2w(int unixaf) {
1529 unsigned int i;
1530
1531 for (i = 0; i < ARRAY_SIZE(ws_af_map); i++)
1532 if (ws_af_map[i][1] == unixaf)
1533 return ws_af_map[i][0];
1534 FIXME("unhandled UNIX address family %d\n", unixaf);
1535 return -1;
1536 }
1537
1538 static int
1539 convert_proto_w2u(int windowsproto) {
1540 unsigned int i;
1541
1542 for (i = 0; i < ARRAY_SIZE(ws_proto_map); i++)
1543 if (ws_proto_map[i][0] == windowsproto)
1544 return ws_proto_map[i][1];
1545
1546 /* check for extended IPX */
1547 if (IS_IPX_PROTO(windowsproto))
1548 return windowsproto;
1549
1550 FIXME("unhandled Windows socket protocol %d\n", windowsproto);
1551 return -1;
1552 }
1553
1554 static int
1555 convert_proto_u2w(int unixproto) {
1556 unsigned int i;
1557
1558 for (i = 0; i < ARRAY_SIZE(ws_proto_map); i++)
1559 if (ws_proto_map[i][1] == unixproto)
1560 return ws_proto_map[i][0];
1561
1562 /* if value is inside IPX range just return it - the kernel simply
1563 * echoes the value used in the socket() function */
1564 if (IS_IPX_PROTO(unixproto))
1565 return unixproto;
1566
1567 FIXME("unhandled UNIX socket protocol %d\n", unixproto);
1568 return -1;
1569 }
1570
1571 static int
1572 convert_socktype_w2u(int windowssocktype) {
1573 unsigned int i;
1574
1575 for (i = 0; i < ARRAY_SIZE(ws_socktype_map); i++)
1576 if (ws_socktype_map[i][0] == windowssocktype)
1577 return ws_socktype_map[i][1];
1578 FIXME("unhandled Windows socket type %d\n", windowssocktype);
1579 return -1;
1580 }
1581
1582 static int
1583 convert_socktype_u2w(int unixsocktype) {
1584 unsigned int i;
1585
1586 for (i = 0; i < ARRAY_SIZE(ws_socktype_map); i++)
1587 if (ws_socktype_map[i][1] == unixsocktype)
1588 return ws_socktype_map[i][0];
1589 FIXME("unhandled UNIX socket type %d\n", unixsocktype);
1590 return -1;
1591 }
1592
1593 static int convert_poll_w2u(int events)
1594 {
1595 int i, ret;
1596 for (i = ret = 0; events && i < ARRAY_SIZE(ws_poll_map); i++)
1597 {
1598 if (ws_poll_map[i][0] & events)
1599 {
1600 ret |= ws_poll_map[i][1];
1601 events &= ~ws_poll_map[i][0];
1602 }
1603 }
1604
1605 if (events)
1606 FIXME("Unsupported WSAPoll() flags 0x%x\n", events);
1607 return ret;
1608 }
1609
1610 static int convert_poll_u2w(int events)
1611 {
1612 int i, ret;
1613 for (i = ret = 0; events && i < ARRAY_SIZE(ws_poll_map); i++)
1614 {
1615 if (ws_poll_map[i][1] & events)
1616 {
1617 ret |= ws_poll_map[i][0];
1618 events &= ~ws_poll_map[i][1];
1619 }
1620 }
1621
1622 if (events)
1623 FIXME("Unsupported poll() flags 0x%x\n", events);
1624 return ret;
1625 }
1626
1627 static int set_ipx_packettype(int sock, int ptype)
1628 {
1629 #ifdef HAS_IPX
1630 int fd = get_sock_fd( sock, 0, NULL ), ret = 0;
1631 TRACE("trying to set IPX_PTYPE: %d (fd: %d)\n", ptype, fd);
1632
1633 if (fd == -1) return SOCKET_ERROR;
1634
1635 /* We try to set the ipx type on ipx socket level. */
1636 #ifdef SOL_IPX
1637 if(setsockopt(fd, SOL_IPX, IPX_TYPE, &ptype, sizeof(ptype)) == -1)
1638 {
1639 ERR("IPX: could not set ipx option type; expect weird behaviour\n");
1640 ret = SOCKET_ERROR;
1641 }
1642 #else
1643 {
1644 struct ipx val;
1645 /* Should we retrieve val using a getsockopt call and then
1646 * set the modified one? */
1647 val.ipx_pt = ptype;
1648 setsockopt(fd, 0, SO_DEFAULT_HEADERS, &val, sizeof(struct ipx));
1649 }
1650 #endif
1651 release_sock_fd( sock, fd );
1652 return ret;
1653 #else
1654 WARN("IPX support is not enabled, can't set packet type\n");
1655 return SOCKET_ERROR;
1656 #endif
1657 }
1658
1659 /* ----------------------------------- API -----
1660 *
1661 * Init / cleanup / error checking.
1662 */
1663
1664 /***********************************************************************
1665 * WSAStartup (WS2_32.115)
1666 */
1667 int WINAPI WSAStartup(WORD wVersionRequested, LPWSADATA lpWSAData)
1668 {
1669 TRACE("verReq=%x\n", wVersionRequested);
1670
1671 if (LOBYTE(wVersionRequested) < 1)
1672 return WSAVERNOTSUPPORTED;
1673
1674 if (!lpWSAData) return WSAEINVAL;
1675
1676 num_startup++;
1677
1678 /* that's the whole of the negotiation for now */
1679 lpWSAData->wVersion = wVersionRequested;
1680 /* return winsock information */
1681 lpWSAData->wHighVersion = 0x0202;
1682 strcpy(lpWSAData->szDescription, "WinSock 2.0" );
1683 strcpy(lpWSAData->szSystemStatus, "Running" );
1684 lpWSAData->iMaxSockets = WS_MAX_SOCKETS_PER_PROCESS;
1685 lpWSAData->iMaxUdpDg = WS_MAX_UDP_DATAGRAM;
1686 /* don't do anything with lpWSAData->lpVendorInfo */
1687 /* (some apps don't allocate the space for this field) */
1688
1689 TRACE("succeeded starts: %d\n", num_startup);
1690 return 0;
1691 }
1692
1693
1694 /***********************************************************************
1695 * WSACleanup (WS2_32.116)
1696 */
1697 INT WINAPI WSACleanup(void)
1698 {
1699 if (num_startup) {
1700 num_startup--;
1701 TRACE("pending cleanups: %d\n", num_startup);
1702 return 0;
1703 }
1704 SetLastError(WSANOTINITIALISED);
1705 return SOCKET_ERROR;
1706 }
1707
1708
1709 /***********************************************************************
1710 * WSAGetLastError (WS2_32.111)
1711 */
1712 INT WINAPI WSAGetLastError(void)
1713 {
1714 return GetLastError();
1715 }
1716
1717 /***********************************************************************
1718 * WSASetLastError (WS2_32.112)
1719 */
1720 void WINAPI WSASetLastError(INT iError) {
1721 SetLastError(iError);
1722 }
1723
1724 static struct WS_hostent *check_buffer_he(int size)
1725 {
1726 struct per_thread_data * ptb = get_per_thread_data();
1727 if (ptb->he_buffer)
1728 {
1729 if (ptb->he_len >= size ) return ptb->he_buffer;
1730 HeapFree( GetProcessHeap(), 0, ptb->he_buffer );
1731 }
1732 ptb->he_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->he_len = size) );
1733 if (!ptb->he_buffer) SetLastError(WSAENOBUFS);
1734 return ptb->he_buffer;
1735 }
1736
1737 static struct WS_servent *check_buffer_se(int size)
1738 {
1739 struct per_thread_data * ptb = get_per_thread_data();
1740 if (ptb->se_buffer)
1741 {
1742 if (ptb->se_len >= size ) return ptb->se_buffer;
1743 HeapFree( GetProcessHeap(), 0, ptb->se_buffer );
1744 }
1745 ptb->se_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->se_len = size) );
1746 if (!ptb->se_buffer) SetLastError(WSAENOBUFS);
1747 return ptb->se_buffer;
1748 }
1749
1750 static struct WS_protoent *check_buffer_pe(int size)
1751 {
1752 struct per_thread_data * ptb = get_per_thread_data();
1753 if (ptb->pe_buffer)
1754 {
1755 if (ptb->pe_len >= size ) return ptb->pe_buffer;
1756 HeapFree( GetProcessHeap(), 0, ptb->pe_buffer );
1757 }
1758 ptb->pe_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->pe_len = size) );
1759 if (!ptb->pe_buffer) SetLastError(WSAENOBUFS);
1760 return ptb->pe_buffer;
1761 }
1762
1763 /* ----------------------------------- i/o APIs */
1764
1765 static inline BOOL supported_pf(int pf)
1766 {
1767 switch (pf)
1768 {
1769 case WS_AF_INET:
1770 case WS_AF_INET6:
1771 return TRUE;
1772 #ifdef HAS_IPX
1773 case WS_AF_IPX:
1774 return TRUE;
1775 #endif
1776 #ifdef HAS_IRDA
1777 case WS_AF_IRDA:
1778 return TRUE;
1779 #endif
1780 default:
1781 return FALSE;
1782 }
1783 }
1784
1785 static inline BOOL supported_protocol(int protocol)
1786 {
1787 int i;
1788 for (i = 0; i < ARRAY_SIZE(valid_protocols); i++)
1789 if (protocol == valid_protocols[i])
1790 return TRUE;
1791 return FALSE;
1792 }
1793
1794 /**********************************************************************/
1795
1796 /* Returns the length of the converted address if successful, 0 if it was too
1797 * small to start with or unknown family or invalid address buffer.
1798 */
1799 static unsigned int ws_sockaddr_ws2u(const struct WS_sockaddr* wsaddr, int wsaddrlen,
1800 union generic_unix_sockaddr *uaddr)
1801 {
1802 unsigned int uaddrlen = 0;
1803
1804 if (!wsaddr)
1805 return 0;
1806
1807 switch (wsaddr->sa_family)
1808 {
1809 #ifdef HAS_IPX
1810 case WS_AF_IPX:
1811 {
1812 const struct WS_sockaddr_ipx* wsipx=(const struct WS_sockaddr_ipx*)wsaddr;
1813 struct sockaddr_ipx* uipx = (struct sockaddr_ipx *)uaddr;
1814
1815 if (wsaddrlen<sizeof(struct WS_sockaddr_ipx))
1816 return 0;
1817
1818 uaddrlen = sizeof(struct sockaddr_ipx);
1819 memset( uaddr, 0, uaddrlen );
1820 uipx->sipx_family=AF_IPX;
1821 uipx->sipx_port=wsipx->sa_socket;
1822 /* copy sa_netnum and sa_nodenum to sipx_network and sipx_node
1823 * in one go
1824 */
1825 memcpy(&uipx->sipx_network,wsipx->sa_netnum,sizeof(uipx->sipx_network)+sizeof(uipx->sipx_node));
1826 #ifdef IPX_FRAME_NONE
1827 uipx->sipx_type=IPX_FRAME_NONE;
1828 #endif
1829 break;
1830 }
1831 #endif
1832 case WS_AF_INET6: {
1833 struct sockaddr_in6* uin6 = (struct sockaddr_in6 *)uaddr;
1834 const struct WS_sockaddr_in6* win6 = (const struct WS_sockaddr_in6*)wsaddr;
1835
1836 /* Note: Windows has 2 versions of the sockaddr_in6 struct, one with
1837 * scope_id, one without.
1838 */
1839 if (wsaddrlen >= sizeof(struct WS_sockaddr_in6_old)) {
1840 uaddrlen = sizeof(struct sockaddr_in6);
1841 memset( uaddr, 0, uaddrlen );
1842 uin6->sin6_family = AF_INET6;
1843 uin6->sin6_port = win6->sin6_port;
1844 uin6->sin6_flowinfo = win6->sin6_flowinfo;
1845 #ifdef HAVE_STRUCT_SOCKADDR_IN6_SIN6_SCOPE_ID
1846 if (wsaddrlen >= sizeof(struct WS_sockaddr_in6)) uin6->sin6_scope_id = win6->sin6_scope_id;
1847 #endif
1848 memcpy(&uin6->sin6_addr,&win6->sin6_addr,16); /* 16 bytes = 128 address bits */
1849 break;
1850 }
1851 FIXME("bad size %d for WS_sockaddr_in6\n",wsaddrlen);
1852 return 0;
1853 }
1854 case WS_AF_INET: {
1855 struct sockaddr_in* uin = (struct sockaddr_in *)uaddr;
1856 const struct WS_sockaddr_in* win = (const struct WS_sockaddr_in*)wsaddr;
1857
1858 if (wsaddrlen<sizeof(struct WS_sockaddr_in))
1859 return 0;
1860 uaddrlen = sizeof(struct sockaddr_in);
1861 memset( uaddr, 0, uaddrlen );
1862 uin->sin_family = AF_INET;
1863 uin->sin_port = win->sin_port;
1864 memcpy(&uin->sin_addr,&win->sin_addr,4); /* 4 bytes = 32 address bits */
1865 break;
1866 }
1867 #ifdef HAS_IRDA
1868 case WS_AF_IRDA: {
1869 struct sockaddr_irda *uin = (struct sockaddr_irda *)uaddr;
1870 const SOCKADDR_IRDA *win = (const SOCKADDR_IRDA *)wsaddr;
1871
1872 if (wsaddrlen < sizeof(SOCKADDR_IRDA))
1873 return 0;
1874 uaddrlen = sizeof(struct sockaddr_irda);
1875 memset( uaddr, 0, uaddrlen );
1876 uin->sir_family = AF_IRDA;
1877 if (!strncmp( win->irdaServiceName, "LSAP-SEL", strlen( "LSAP-SEL" ) ))
1878 {
1879 unsigned int lsap_sel = 0;
1880
1881 sscanf( win->irdaServiceName, "LSAP-SEL%u", &lsap_sel );
1882 uin->sir_lsap_sel = lsap_sel;
1883 }
1884 else
1885 {
1886 uin->sir_lsap_sel = LSAP_ANY;
1887 memcpy( uin->sir_name, win->irdaServiceName, 25 );
1888 }
1889 memcpy( &uin->sir_addr, win->irdaDeviceID, sizeof(uin->sir_addr) );
1890 break;
1891 }
1892 #endif
1893 case WS_AF_UNSPEC: {
1894 /* Try to determine the needed space by the passed windows sockaddr space */
1895 switch (wsaddrlen) {
1896 default: /* likely an ipv4 address */
1897 case sizeof(struct WS_sockaddr_in):
1898 uaddrlen = sizeof(struct sockaddr_in);
1899 break;
1900 #ifdef HAS_IPX
1901 case sizeof(struct WS_sockaddr_ipx):
1902 uaddrlen = sizeof(struct sockaddr_ipx);
1903 break;
1904 #endif
1905 #ifdef HAS_IRDA
1906 case sizeof(SOCKADDR_IRDA):
1907 uaddrlen = sizeof(struct sockaddr_irda);
1908 break;
1909 #endif
1910 case sizeof(struct WS_sockaddr_in6):
1911 case sizeof(struct WS_sockaddr_in6_old):
1912 uaddrlen = sizeof(struct sockaddr_in6);
1913 break;
1914 }
1915 memset( uaddr, 0, uaddrlen );
1916 break;
1917 }
1918 default:
1919 FIXME("Unknown address family %d, return NULL.\n", wsaddr->sa_family);
1920 return 0;
1921 }
1922 return uaddrlen;
1923 }
1924
1925 static BOOL is_sockaddr_bound(const struct sockaddr *uaddr, int uaddrlen)
1926 {
1927 switch (uaddr->sa_family)
1928 {
1929 #ifdef HAS_IPX
1930 case AF_IPX:
1931 {
1932 static const struct sockaddr_ipx emptyAddr;
1933 struct sockaddr_ipx *ipx = (struct sockaddr_ipx*) uaddr;
1934 return ipx->sipx_port
1935 || memcmp(&ipx->sipx_network, &emptyAddr.sipx_network, sizeof(emptyAddr.sipx_network))
1936 || memcmp(&ipx->sipx_node, &emptyAddr.sipx_node, sizeof(emptyAddr.sipx_node));
1937 }
1938 #endif
1939 case AF_INET6:
1940 {
1941 static const struct sockaddr_in6 emptyAddr;
1942 const struct sockaddr_in6 *in6 = (const struct sockaddr_in6*) uaddr;
1943 return in6->sin6_port || memcmp(&in6->sin6_addr, &emptyAddr.sin6_addr, sizeof(struct in6_addr));
1944 }
1945 case AF_INET:
1946 {
1947 static const struct sockaddr_in emptyAddr;
1948 const struct sockaddr_in *in = (const struct sockaddr_in*) uaddr;
1949 return in->sin_port || memcmp(&in->sin_addr, &emptyAddr.sin_addr, sizeof(struct in_addr));
1950 }
1951 case AF_UNSPEC:
1952 return FALSE;
1953 default:
1954 FIXME("unknown address family %d\n", uaddr->sa_family);
1955 return TRUE;
1956 }
1957 }
1958
1959 /* Returns -1 if getsockname fails, 0 if not bound, 1 otherwise */
1960 static int is_fd_bound(int fd, union generic_unix_sockaddr *uaddr, socklen_t *uaddrlen)
1961 {
1962 union generic_unix_sockaddr inaddr;
1963 socklen_t inlen;
1964 int res;
1965
1966 if (!uaddr) uaddr = &inaddr;
1967 if (!uaddrlen) uaddrlen = &inlen;
1968
1969 *uaddrlen = sizeof(inaddr);
1970 res = getsockname(fd, &uaddr->addr, uaddrlen);
1971 if (!res) res = is_sockaddr_bound(&uaddr->addr, *uaddrlen);
1972 return res;
1973 }
1974
1975 /* Returns 0 if successful, -1 if the buffer is too small */
1976 static int ws_sockaddr_u2ws(const struct sockaddr* uaddr, struct WS_sockaddr* wsaddr, int* wsaddrlen)
1977 {
1978 int res;
1979
1980 switch(uaddr->sa_family)
1981 {
1982 #ifdef HAS_IPX
1983 case AF_IPX:
1984 {
1985 const struct sockaddr_ipx* uipx=(const struct sockaddr_ipx*)uaddr;
1986 struct WS_sockaddr_ipx* wsipx=(struct WS_sockaddr_ipx*)wsaddr;
1987
1988 res=-1;
1989 switch (*wsaddrlen) /* how much can we copy? */
1990 {
1991 default:
1992 res=0; /* enough */
1993 *wsaddrlen = sizeof(*wsipx);
1994 wsipx->sa_socket=uipx->sipx_port;
1995 /* fall through */
1996 case 13:
1997 case 12:
1998 memcpy(wsipx->sa_nodenum,uipx->sipx_node,sizeof(wsipx->sa_nodenum));
1999 /* fall through */
2000 case 11:
2001 case 10:
2002 case 9:
2003 case 8:
2004 case 7:
2005 case 6:
2006 memcpy(wsipx->sa_netnum,&uipx->sipx_network,sizeof(wsipx->sa_netnum));
2007 /* fall through */
2008 case 5:
2009 case 4:
2010 case 3:
2011 case 2:
2012 wsipx->sa_family=WS_AF_IPX;
2013 /* fall through */
2014 case 1:
2015 case 0:
2016 /* way too small */
2017 break;
2018 }
2019 }
2020 break;
2021 #endif
2022 #ifdef HAS_IRDA
2023 case AF_IRDA: {
2024 const struct sockaddr_irda *uin = (const struct sockaddr_irda *)uaddr;
2025 SOCKADDR_IRDA *win = (SOCKADDR_IRDA *)wsaddr;
2026
2027 if (*wsaddrlen < sizeof(SOCKADDR_IRDA))
2028 return -1;
2029 win->irdaAddressFamily = WS_AF_IRDA;
2030 memcpy( win->irdaDeviceID, &uin->sir_addr, sizeof(win->irdaDeviceID) );
2031 if (uin->sir_lsap_sel != LSAP_ANY)
2032 sprintf( win->irdaServiceName, "LSAP-SEL%u", uin->sir_lsap_sel );
2033 else
2034 memcpy( win->irdaServiceName, uin->sir_name,
2035 sizeof(win->irdaServiceName) );
2036 return 0;
2037 }
2038 #endif
2039 case AF_INET6: {
2040 const struct sockaddr_in6* uin6 = (const struct sockaddr_in6*)uaddr;
2041 struct WS_sockaddr_in6_old* win6old = (struct WS_sockaddr_in6_old*)wsaddr;
2042
2043 if (*wsaddrlen < sizeof(struct WS_sockaddr_in6_old))
2044 return -1;
2045 win6old->sin6_family = WS_AF_INET6;
2046 win6old->sin6_port = uin6->sin6_port;
2047 win6old->sin6_flowinfo = uin6->sin6_flowinfo;
2048 memcpy(&win6old->sin6_addr,&uin6->sin6_addr,16); /* 16 bytes = 128 address bits */
2049 #ifdef HAVE_STRUCT_SOCKADDR_IN6_SIN6_SCOPE_ID
2050 if (*wsaddrlen >= sizeof(struct WS_sockaddr_in6)) {
2051 struct WS_sockaddr_in6* win6 = (struct WS_sockaddr_in6*)wsaddr;
2052 win6->sin6_scope_id = uin6->sin6_scope_id;
2053 *wsaddrlen = sizeof(struct WS_sockaddr_in6);
2054 }
2055 else
2056 *wsaddrlen = sizeof(struct WS_sockaddr_in6_old);
2057 #else
2058 *wsaddrlen = sizeof(struct WS_sockaddr_in6_old);
2059 #endif
2060 return 0;
2061 }
2062 case AF_INET: {
2063 const struct sockaddr_in* uin = (const struct sockaddr_in*)uaddr;
2064 struct WS_sockaddr_in* win = (struct WS_sockaddr_in*)wsaddr;
2065
2066 if (*wsaddrlen < sizeof(struct WS_sockaddr_in))
2067 return -1;
2068 win->sin_family = WS_AF_INET;
2069 win->sin_port = uin->sin_port;
2070 memcpy(&win->sin_addr,&uin->sin_addr,4); /* 4 bytes = 32 address bits */
2071 memset(win->sin_zero, 0, 8); /* Make sure the null padding is null */
2072 *wsaddrlen = sizeof(struct WS_sockaddr_in);
2073 return 0;
2074 }
2075 case AF_UNSPEC: {
2076 memset(wsaddr,0,*wsaddrlen);
2077 return 0;
2078 }
2079 default:
2080 FIXME("Unknown address family %d\n", uaddr->sa_family);
2081 return -1;
2082 }
2083 return res;
2084 }
2085
2086 static INT WS_DuplicateSocket(BOOL unicode, SOCKET s, DWORD dwProcessId, LPWSAPROTOCOL_INFOW lpProtocolInfo)
2087 {
2088 HANDLE hProcess;
2089 int size;
2090 WSAPROTOCOL_INFOW infow;
2091
2092 TRACE("(unicode %d, socket %04lx, processid %x, buffer %p)\n",
2093 unicode, s, dwProcessId, lpProtocolInfo);
2094
2095 if (!ws_protocol_info(s, unicode, &infow, &size))
2096 return SOCKET_ERROR;
2097
2098 if (!(hProcess = OpenProcess(PROCESS_DUP_HANDLE, FALSE, dwProcessId)))
2099 {
2100 SetLastError(WSAEINVAL);
2101 return SOCKET_ERROR;
2102 }
2103
2104 if (!lpProtocolInfo)
2105 {
2106 CloseHandle(hProcess);
2107 SetLastError(WSAEFAULT);
2108 return SOCKET_ERROR;
2109 }
2110
2111 /* I don't know what the real Windoze does next, this is a hack */
2112 /* ...we could duplicate and then use ConvertToGlobalHandle on the duplicate, then let
2113 * the target use the global duplicate, or we could copy a reference to us to the structure
2114 * and let the target duplicate it from us, but let's do it as simple as possible */
2115 memcpy(lpProtocolInfo, &infow, size);
2116 DuplicateHandle(GetCurrentProcess(), SOCKET2HANDLE(s),
2117 hProcess, (LPHANDLE)&lpProtocolInfo->dwServiceFlags3,
2118 0, FALSE, DUPLICATE_SAME_ACCESS);
2119 CloseHandle(hProcess);
2120 lpProtocolInfo->dwServiceFlags4 = 0xff00ff00; /* magic */
2121 return 0;
2122 }
2123
2124 /*****************************************************************************
2125 * WS_EnterSingleProtocolW [internal]
2126 *
2127 * enters the protocol information of one given protocol into the given
2128 * buffer.
2129 *
2130 * RETURNS
2131 * TRUE if a protocol was entered into the buffer.
2132 *
2133 * BUGS
2134 * - only implemented for IPX, SPX, SPXII, TCP, UDP
2135 * - there is no check that the operating system supports the returned
2136 * protocols
2137 */
2138 static BOOL WS_EnterSingleProtocolW( INT protocol, WSAPROTOCOL_INFOW* info )
2139 {
2140 memset( info, 0, sizeof(WSAPROTOCOL_INFOW) );
2141 info->iProtocol = protocol;
2142
2143 switch (protocol)
2144 {
2145 case WS_IPPROTO_TCP:
2146 info->dwServiceFlags1 = XP1_IFS_HANDLES | XP1_EXPEDITED_DATA |
2147 XP1_GRACEFUL_CLOSE | XP1_GUARANTEED_ORDER |
2148 XP1_GUARANTEED_DELIVERY;
2149 info->ProviderId = ProviderIdIP;
2150 info->dwCatalogEntryId = 0x3e9;
2151 info->dwProviderFlags = PFL_MATCHES_PROTOCOL_ZERO;
2152 info->ProtocolChain.ChainLen = 1;
2153 info->iVersion = 2;
2154 info->iAddressFamily = WS_AF_INET;
2155 info->iMaxSockAddr = 0x10;
2156 info->iMinSockAddr = 0x10;
2157 info->iSocketType = WS_SOCK_STREAM;
2158 strcpyW( info->szProtocol, NameTcpW );
2159 break;
2160
2161 case WS_IPPROTO_UDP:
2162 info->dwServiceFlags1 = XP1_IFS_HANDLES | XP1_SUPPORT_BROADCAST |
2163 XP1_SUPPORT_MULTIPOINT | XP1_MESSAGE_ORIENTED |
2164 XP1_CONNECTIONLESS;
2165 info->ProviderId = ProviderIdIP;
2166 info->dwCatalogEntryId = 0x3ea;
2167 info->dwProviderFlags = PFL_MATCHES_PROTOCOL_ZERO;
2168 info->ProtocolChain.ChainLen = 1;
2169 info->iVersion = 2;
2170 info->iAddressFamily = WS_AF_INET;
2171 info->iMaxSockAddr = 0x10;
2172 info->iMinSockAddr = 0x10;
2173 info->iSocketType = WS_SOCK_DGRAM;
2174 info->dwMessageSize = 0xffbb;
2175 strcpyW( info->szProtocol, NameUdpW );
2176 break;
2177
2178 case WS_NSPROTO_IPX:
2179 info->dwServiceFlags1 = XP1_PARTIAL_MESSAGE | XP1_SUPPORT_BROADCAST |
2180 XP1_SUPPORT_MULTIPOINT | XP1_MESSAGE_ORIENTED |
2181 XP1_CONNECTIONLESS;
2182 info->ProviderId = ProviderIdIPX;
2183 info->dwCatalogEntryId = 0x406;
2184 info->dwProviderFlags = PFL_MATCHES_PROTOCOL_ZERO;
2185 info->ProtocolChain.ChainLen = 1;
2186 info->iVersion = 2;
2187 info->iAddressFamily = WS_AF_IPX;
2188 info->iMaxSockAddr = 0x10;
2189 info->iMinSockAddr = 0x0e;
2190 info->iSocketType = WS_SOCK_DGRAM;
2191 info->iProtocolMaxOffset = 0xff;
2192 info->dwMessageSize = 0x240;
2193 strcpyW( info->szProtocol, NameIpxW );
2194 break;
2195
2196 case WS_NSPROTO_SPX:
2197 info->dwServiceFlags1 = XP1_IFS_HANDLES | XP1_PSEUDO_STREAM |
2198 XP1_MESSAGE_ORIENTED | XP1_GUARANTEED_ORDER |
2199 XP1_GUARANTEED_DELIVERY;
2200 info->ProviderId = ProviderIdSPX;
2201 info->dwCatalogEntryId = 0x407;
2202 info->dwProviderFlags = PFL_MATCHES_PROTOCOL_ZERO;
2203 info->ProtocolChain.ChainLen = 1;
2204 info->iVersion = 2;
2205 info->iAddressFamily = WS_AF_IPX;
2206 info->iMaxSockAddr = 0x10;
2207 info->iMinSockAddr = 0x0e;
2208 info->iSocketType = WS_SOCK_SEQPACKET;
2209 info->dwMessageSize = 0xffffffff;
2210 strcpyW( info->szProtocol, NameSpxW );
2211 break;
2212
2213 case WS_NSPROTO_SPXII:
2214 info->dwServiceFlags1 = XP1_IFS_HANDLES | XP1_GRACEFUL_CLOSE |
2215 XP1_PSEUDO_STREAM | XP1_MESSAGE_ORIENTED |
2216 XP1_GUARANTEED_ORDER | XP1_GUARANTEED_DELIVERY;
2217 info->ProviderId = ProviderIdSPX;
2218 info->dwCatalogEntryId = 0x409;
2219 info->dwProviderFlags = PFL_MATCHES_PROTOCOL_ZERO;
2220 info->ProtocolChain.ChainLen = 1;
2221 info->iVersion = 2;
2222 info->iAddressFamily = WS_AF_IPX;
2223 info->iMaxSockAddr = 0x10;
2224 info->iMinSockAddr = 0x0e;
2225 info->iSocketType = WS_SOCK_SEQPACKET;
2226 info->dwMessageSize = 0xffffffff;
2227 strcpyW( info->szProtocol, NameSpxIIW );
2228 break;
2229
2230 default:
2231 FIXME("unknown Protocol <0x%08x>\n", protocol);
2232 return FALSE;
2233 }
2234 return TRUE;
2235 }
2236
2237 /*****************************************************************************
2238 * WS_EnterSingleProtocolA [internal]
2239 *
2240 * see function WS_EnterSingleProtocolW
2241 *
2242 */
2243 static BOOL WS_EnterSingleProtocolA( INT protocol, WSAPROTOCOL_INFOA* info )
2244 {
2245 WSAPROTOCOL_INFOW infow;
2246 INT ret;
2247 memset( info, 0, sizeof(WSAPROTOCOL_INFOA) );
2248
2249 ret = WS_EnterSingleProtocolW( protocol, &infow );
2250 if (ret)
2251 {
2252 /* convert the structure from W to A */
2253 memcpy( info, &infow, FIELD_OFFSET( WSAPROTOCOL_INFOA, szProtocol ) );
2254 WideCharToMultiByte( CP_ACP, 0, infow.szProtocol, -1,
2255 info->szProtocol, WSAPROTOCOL_LEN+1, NULL, NULL );
2256 }
2257
2258 return ret;
2259 }
2260
2261 static INT WS_EnumProtocols( BOOL unicode, const INT *protocols, LPWSAPROTOCOL_INFOW buffer, LPDWORD len )
2262 {
2263 INT i = 0, items = 0;
2264 DWORD size = 0;
2265 union _info
2266 {
2267 LPWSAPROTOCOL_INFOA a;
2268 LPWSAPROTOCOL_INFOW w;
2269 } info;
2270 info.w = buffer;
2271
2272 if (!protocols) protocols = valid_protocols;
2273
2274 while (protocols[i])
2275 {
2276 if(supported_protocol(protocols[i++]))
2277 items++;
2278 }
2279
2280 size = items * (unicode ? sizeof(WSAPROTOCOL_INFOW) : sizeof(WSAPROTOCOL_INFOA));
2281
2282 TRACE("unicode %d, protocols %p, buffer %p, length %p %d, items %d, required %d\n",
2283 unicode, protocols, buffer, len, len ? *len : 0, items, size);
2284
2285 if (*len < size || !buffer)
2286 {
2287 *len = size;
2288 SetLastError(WSAENOBUFS);
2289 return SOCKET_ERROR;
2290 }
2291
2292 for (i = items = 0; protocols[i]; i++)
2293 {
2294 if (!supported_protocol(protocols[i])) continue;
2295 if (unicode)
2296 {
2297 if (WS_EnterSingleProtocolW( protocols[i], &info.w[items] ))
2298 items++;
2299 }
2300 else
2301 {
2302 if (WS_EnterSingleProtocolA( protocols[i], &info.a[items] ))
2303 items++;
2304 }
2305 }
2306 return items;
2307 }
2308
2309 static BOOL ws_protocol_info(SOCKET s, int unicode, WSAPROTOCOL_INFOW *buffer, int *size)
2310 {
2311 NTSTATUS status;
2312
2313 *size = unicode ? sizeof(WSAPROTOCOL_INFOW) : sizeof(WSAPROTOCOL_INFOA);
2314 memset(buffer, 0, *size);
2315
2316 SERVER_START_REQ( get_socket_info )
2317 {
2318 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
2319 status = wine_server_call( req );
2320 if (!status)
2321 {
2322 buffer->iAddressFamily = convert_af_u2w(reply->family);
2323 buffer->iSocketType = convert_socktype_u2w(reply->type);
2324 buffer->iProtocol = convert_proto_u2w(reply->protocol);
2325 }
2326 }
2327 SERVER_END_REQ;
2328
2329 if (status)
2330 {
2331 unsigned int err = NtStatusToWSAError( status );
2332 SetLastError( err == WSAEBADF ? WSAENOTSOCK : err );
2333 return FALSE;
2334 }
2335
2336 if (unicode)
2337 WS_EnterSingleProtocolW( buffer->iProtocol, buffer);
2338 else
2339 WS_EnterSingleProtocolA( buffer->iProtocol, (WSAPROTOCOL_INFOA *)buffer);
2340
2341 return TRUE;
2342 }
2343
2344 /**************************************************************************
2345 * Functions for handling overlapped I/O
2346 **************************************************************************/
2347
2348 /* user APC called upon async completion */
2349 static void WINAPI ws2_async_apc( void *arg, IO_STATUS_BLOCK *iosb, ULONG reserved )
2350 {
2351 struct ws2_async *wsa = arg;
2352
2353 if (wsa->completion_func) wsa->completion_func( NtStatusToWSAError(iosb->u.Status),
2354 iosb->Information, wsa->user_overlapped,
2355 wsa->flags );
2356 release_async_io( &wsa->io );
2357 }
2358
2359 /***********************************************************************
2360 * WS2_recv (INTERNAL)
2361 *
2362 * Workhorse for both synchronous and asynchronous recv() operations.
2363 */
2364 static int WS2_recv( int fd, struct ws2_async *wsa, int flags )
2365 {
2366 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
2367 char pktbuf[512];
2368 #endif
2369 struct msghdr hdr;
2370 union generic_unix_sockaddr unix_sockaddr;
2371 int n;
2372
2373 hdr.msg_name = NULL;
2374
2375 if (wsa->addr)
2376 {
2377 hdr.msg_namelen = sizeof(unix_sockaddr);
2378 hdr.msg_name = &unix_sockaddr;
2379 }
2380 else
2381 hdr.msg_namelen = 0;
2382
2383 hdr.msg_iov = wsa->iovec + wsa->first_iovec;
2384 hdr.msg_iovlen = wsa->n_iovecs - wsa->first_iovec;
2385 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
2386 hdr.msg_accrights = NULL;
2387 hdr.msg_accrightslen = 0;
2388 #else
2389 hdr.msg_control = pktbuf;
2390 hdr.msg_controllen = sizeof(pktbuf);
2391 hdr.msg_flags = 0;
2392 #endif
2393
2394 while ((n = __wine_locked_recvmsg( fd, &hdr, flags )) == -1)
2395 {
2396 if (errno != EINTR)
2397 return -1;
2398 }
2399
2400 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
2401 if (wsa->control)
2402 {
2403 ERR("Message control headers cannot be properly supported on this system.\n");
2404 wsa->control->len = 0;
2405 }
2406 #else
2407 if (wsa->control && !convert_control_headers(&hdr, wsa->control))
2408 {
2409 WARN("Application passed insufficient room for control headers.\n");
2410 *wsa->lpFlags |= WS_MSG_CTRUNC;
2411 errno = EMSGSIZE;
2412 return -1;
2413 }
2414 #endif
2415
2416 /* if this socket is connected and lpFrom is not NULL, Linux doesn't give us
2417 * msg_name and msg_namelen from recvmsg, but it does set msg_namelen to zero.
2418 *
2419 * quoting linux 2.6 net/ipv4/tcp.c:
2420 * "According to UNIX98, msg_name/msg_namelen are ignored
2421 * on connected socket. I was just happy when found this 8) --ANK"
2422 *
2423 * likewise MSDN says that lpFrom and lpFromlen are ignored for
2424 * connection-oriented sockets, so don't try to update lpFrom.
2425 */
2426 if (wsa->addr && hdr.msg_namelen)
2427 ws_sockaddr_u2ws( &unix_sockaddr.addr, wsa->addr, wsa->addrlen.ptr );
2428
2429 return n;
2430 }
2431
2432 /***********************************************************************
2433 * WS2_async_recv (INTERNAL)
2434 *
2435 * Handler for overlapped recv() operations.
2436 */
2437 static NTSTATUS WS2_async_recv( void *user, IO_STATUS_BLOCK *iosb, NTSTATUS status )
2438 {
2439 struct ws2_async *wsa = user;
2440 int result = 0, fd;
2441
2442 switch (status)
2443 {
2444 case STATUS_ALERTED:
2445 if ((status = wine_server_handle_to_fd( wsa->hSocket, FILE_READ_DATA, &fd, NULL ) ))
2446 break;
2447
2448 result = WS2_recv( fd, wsa, convert_flags(wsa->flags) );
2449 wine_server_release_fd( wsa->hSocket, fd );
2450 if (result >= 0)
2451 {
2452 status = STATUS_SUCCESS;
2453 _enable_event( wsa->hSocket, FD_READ, 0, 0 );
2454 }
2455 else
2456 {
2457 if (errno == EAGAIN)
2458 {
2459 status = STATUS_PENDING;
2460 _enable_event( wsa->hSocket, FD_READ, 0, 0 );
2461 }
2462 else
2463 {
2464 result = 0;
2465 status = wsaErrStatus();
2466 }
2467 }
2468 break;
2469 }
2470 if (status != STATUS_PENDING)
2471 {
2472 iosb->u.Status = status;
2473 iosb->Information = result;
2474 if (!wsa->completion_func)
2475 release_async_io( &wsa->io );
2476 }
2477 return status;
2478 }
2479
2480 /***********************************************************************
2481 * WS2_async_accept_recv (INTERNAL)
2482 *
2483 * This function is used to finish the read part of an accept request. It is
2484 * needed to place the completion on the correct socket (listener).
2485 */
2486 static NTSTATUS WS2_async_accept_recv( void *user, IO_STATUS_BLOCK *iosb, NTSTATUS status )
2487 {
2488 struct ws2_accept_async *wsa = user;
2489
2490 status = WS2_async_recv( wsa->read, iosb, status );
2491 if (status == STATUS_PENDING)
2492 return status;
2493
2494 if (wsa->cvalue)
2495 WS_AddCompletion( HANDLE2SOCKET(wsa->listen_socket), wsa->cvalue, iosb->u.Status, iosb->Information, TRUE );
2496
2497 release_async_io( &wsa->io );
2498 return status;
2499 }
2500
2501 /***********************************************************************
2502 * WS2_async_accept (INTERNAL)
2503 *
2504 * This is the function called to satisfy the AcceptEx callback
2505 */
2506 static NTSTATUS WS2_async_accept( void *user, IO_STATUS_BLOCK *iosb, NTSTATUS status )
2507 {
2508 struct ws2_accept_async *wsa = user;
2509 int len;
2510 char *addr;
2511
2512 TRACE("status: 0x%x listen: %p, accept: %p\n", status, wsa->listen_socket, wsa->accept_socket);
2513
2514 if (status == STATUS_ALERTED)
2515 {
2516 SERVER_START_REQ( accept_into_socket )
2517 {
2518 req->lhandle = wine_server_obj_handle( wsa->listen_socket );
2519 req->ahandle = wine_server_obj_handle( wsa->accept_socket );
2520 status = wine_server_call( req );
2521 }
2522 SERVER_END_REQ;
2523
2524 if (status == STATUS_CANT_WAIT)
2525 return STATUS_PENDING;
2526
2527 if (status == STATUS_INVALID_HANDLE)
2528 {
2529 FIXME("AcceptEx accepting socket closed but request was not cancelled\n");
2530 status = STATUS_CANCELLED;
2531 }
2532 }
2533 else if (status == STATUS_HANDLES_CLOSED)
2534 status = STATUS_CANCELLED; /* strange windows behavior */
2535
2536 if (status != STATUS_SUCCESS)
2537 goto finish;
2538
2539 /* WS2 Spec says size param is extra 16 bytes long...what do we put in it? */
2540 addr = ((char *)wsa->buf) + wsa->data_len;
2541 len = wsa->local_len - sizeof(int);
2542 WS_getsockname(HANDLE2SOCKET(wsa->accept_socket),
2543 (struct WS_sockaddr *)(addr + sizeof(int)), &len);
2544 *(int *)addr = len;
2545
2546 addr += wsa->local_len;
2547 len = wsa->remote_len - sizeof(int);
2548 WS_getpeername(HANDLE2SOCKET(wsa->accept_socket),
2549 (struct WS_sockaddr *)(addr + sizeof(int)), &len);
2550 *(int *)addr = len;
2551
2552 if (!wsa->read)
2553 goto finish;
2554
2555 wsa->io.callback = WS2_async_accept_recv;
2556 status = register_async( ASYNC_TYPE_READ, wsa->accept_socket, &wsa->io,
2557 wsa->user_overlapped->hEvent, NULL, NULL, iosb);
2558
2559 if (status != STATUS_PENDING)
2560 goto finish;
2561
2562 /* The APC has finished but no completion should be sent for the operation yet, additional processing
2563 * needs to be performed by WS2_async_accept_recv() first. */
2564 return STATUS_MORE_PROCESSING_REQUIRED;
2565
2566 finish:
2567 iosb->u.Status = status;
2568 iosb->Information = 0;
2569
2570 if (wsa->read) release_async_io( &wsa->read->io );
2571 release_async_io( &wsa->io );
2572 return status;
2573 }
2574
2575 /***********************************************************************
2576 * WS2_send (INTERNAL)
2577 *
2578 * Workhorse for both synchronous and asynchronous send() operations.
2579 */
2580 static int WS2_send( int fd, struct ws2_async *wsa, int flags )
2581 {
2582 struct msghdr hdr;
2583 union generic_unix_sockaddr unix_addr;
2584 int n, ret;
2585
2586 hdr.msg_name = NULL;
2587 hdr.msg_namelen = 0;
2588
2589 if (wsa->addr)
2590 {
2591 hdr.msg_name = &unix_addr;
2592 hdr.msg_namelen = ws_sockaddr_ws2u( wsa->addr, wsa->addrlen.val, &unix_addr );
2593 if ( !hdr.msg_namelen )
2594 {
2595 errno = EFAULT;
2596 return -1;
2597 }
2598
2599 #if defined(HAS_IPX) && defined(SOL_IPX)
2600 if(wsa->addr->sa_family == WS_AF_IPX)
2601 {
2602 struct sockaddr_ipx* uipx = (struct sockaddr_ipx*)hdr.msg_name;
2603 int val=0;
2604 socklen_t len = sizeof(int);
2605
2606 /* The packet type is stored at the ipx socket level; At least the linux kernel seems
2607 * to do something with it in case hdr.msg_name is NULL. Nonetheless can we use it to store
2608 * the packet type and then we can retrieve it using getsockopt. After that we can set the
2609 * ipx type in the sockaddr_opx structure with the stored value.
2610 */
2611 if(getsockopt(fd, SOL_IPX, IPX_TYPE, &val, &len) != -1)
2612 uipx->sipx_type = val;
2613 }
2614 #endif
2615 }
2616
2617 hdr.msg_iov = wsa->iovec + wsa->first_iovec;
2618 hdr.msg_iovlen = wsa->n_iovecs - wsa->first_iovec;
2619 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
2620 hdr.msg_accrights = NULL;
2621 hdr.msg_accrightslen = 0;
2622 #else
2623 hdr.msg_control = NULL;
2624 hdr.msg_controllen = 0;
2625 hdr.msg_flags = 0;
2626 #endif
2627
2628 while ((ret = sendmsg(fd, &hdr, flags)) == -1)
2629 {
2630 if (errno != EINTR)
2631 return -1;
2632 }
2633
2634 n = ret;
2635 while (wsa->first_iovec < wsa->n_iovecs && wsa->iovec[wsa->first_iovec].iov_len <= n)
2636 n -= wsa->iovec[wsa->first_iovec++].iov_len;
2637 if (wsa->first_iovec < wsa->n_iovecs)
2638 {
2639 wsa->iovec[wsa->first_iovec].iov_base = (char*)wsa->iovec[wsa->first_iovec].iov_base + n;
2640 wsa->iovec[wsa->first_iovec].iov_len -= n;
2641 }
2642 return ret;
2643 }
2644
2645 /***********************************************************************
2646 * WS2_async_send (INTERNAL)
2647 *
2648 * Handler for overlapped send() operations.
2649 */
2650 static NTSTATUS WS2_async_send( void *user, IO_STATUS_BLOCK *iosb, NTSTATUS status )
2651 {
2652 struct ws2_async *wsa = user;
2653 int result = 0, fd;
2654
2655 switch (status)
2656 {
2657 case STATUS_ALERTED:
2658 if ( wsa->n_iovecs <= wsa->first_iovec )
2659 {
2660 /* Nothing to do */
2661 status = STATUS_SUCCESS;
2662 break;
2663 }
2664 if ((status = wine_server_handle_to_fd( wsa->hSocket, FILE_WRITE_DATA, &fd, NULL ) ))
2665 break;
2666
2667 /* check to see if the data is ready (non-blocking) */
2668 result = WS2_send( fd, wsa, convert_flags(wsa->flags) );
2669 wine_server_release_fd( wsa->hSocket, fd );
2670
2671 if (result >= 0)
2672 {
2673 if (wsa->first_iovec < wsa->n_iovecs)
2674 status = STATUS_PENDING;
2675 else
2676 status = STATUS_SUCCESS;
2677
2678 iosb->Information += result;
2679 }
2680 else if (errno == EAGAIN)
2681 {
2682 status = STATUS_PENDING;
2683 }
2684 else
2685 {
2686 status = wsaErrStatus();
2687 }
2688 break;
2689 }
2690 if (status != STATUS_PENDING)
2691 {
2692 iosb->u.Status = status;
2693 if (!wsa->completion_func)
2694 release_async_io( &wsa->io );
2695 }
2696 return status;
2697 }
2698
2699 /***********************************************************************
2700 * WS2_async_shutdown (INTERNAL)
2701 *
2702 * Handler for shutdown() operations on overlapped sockets.
2703 */
2704 static NTSTATUS WS2_async_shutdown( void *user, IO_STATUS_BLOCK *iosb, NTSTATUS status )
2705 {
2706 struct ws2_async_shutdown *wsa = user;
2707 int fd, err = 1;
2708
2709 switch (status)
2710 {
2711 case STATUS_ALERTED:
2712 if ((status = wine_server_handle_to_fd( wsa->hSocket, 0, &fd, NULL ) ))
2713 break;
2714
2715 switch ( wsa->type )
2716 {
2717 case ASYNC_TYPE_READ: err = shutdown( fd, 0 ); break;
2718 case ASYNC_TYPE_WRITE: err = shutdown( fd, 1 ); break;
2719 }
2720 status = err ? wsaErrStatus() : STATUS_SUCCESS;
2721 wine_server_release_fd( wsa->hSocket, fd );
2722 break;
2723 }
2724 iosb->u.Status = status;
2725 iosb->Information = 0;
2726 release_async_io( &wsa->io );
2727 return status;
2728 }
2729
2730 /***********************************************************************
2731 * WS2_register_async_shutdown (INTERNAL)
2732 *
2733 * Helper function for WS_shutdown() on overlapped sockets.
2734 */
2735 static int WS2_register_async_shutdown( SOCKET s, int type )
2736 {
2737 struct ws2_async_shutdown *wsa;
2738 NTSTATUS status;
2739
2740 TRACE("socket %04lx type %d\n", s, type);
2741
2742 wsa = (struct ws2_async_shutdown *)alloc_async_io( sizeof(*wsa), WS2_async_shutdown );
2743 if ( !wsa )
2744 return WSAEFAULT;
2745
2746 wsa->hSocket = SOCKET2HANDLE(s);
2747 wsa->type = type;
2748
2749 status = register_async( type, wsa->hSocket, &wsa->io, 0, NULL, NULL, &wsa->iosb );
2750 if (status != STATUS_PENDING)
2751 {
2752 HeapFree( GetProcessHeap(), 0, wsa );
2753 return NtStatusToWSAError( status );
2754 }
2755 return 0;
2756 }
2757
2758 /***********************************************************************
2759 * accept (WS2_32.1)
2760 */
2761 SOCKET WINAPI WS_accept(SOCKET s, struct WS_sockaddr *addr, int *addrlen32)
2762 {
2763 NTSTATUS status;
2764 SOCKET as;
2765 BOOL is_blocking;
2766
2767 TRACE("socket %04lx\n", s );
2768 status = _is_blocking(s, &is_blocking);
2769 if (status)
2770 goto error;
2771
2772 do {
2773 /* try accepting first (if there is a deferred connection) */
2774 SERVER_START_REQ( accept_socket )
2775 {
2776 req->lhandle = wine_server_obj_handle( SOCKET2HANDLE(s) );
2777 req->access = GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE;
2778 req->attributes = OBJ_INHERIT;
2779 status = wine_server_call( req );
2780 as = HANDLE2SOCKET( wine_server_ptr_handle( reply->handle ));
2781 }
2782 SERVER_END_REQ;
2783 if (!status)
2784 {
2785 if (addr && addrlen32 && WS_getpeername(as, addr, addrlen32))
2786 {
2787 WS_closesocket(as);
2788 return SOCKET_ERROR;
2789 }
2790 TRACE("\taccepted %04lx\n", as);
2791 return as;
2792 }
2793 if (is_blocking && status == STATUS_CANT_WAIT)
2794 {
2795 int fd = get_sock_fd( s, FILE_READ_DATA, NULL );
2796 /* block here */
2797 do_block(fd, POLLIN, -1);
2798 _sync_sock_state(s); /* let wineserver notice connection */
2799 release_sock_fd( s, fd );
2800 }
2801 } while (is_blocking && status == STATUS_CANT_WAIT);
2802
2803 error:
2804 set_error(status);
2805 WARN(" -> ERROR %d\n", GetLastError());
2806 return INVALID_SOCKET;
2807 }
2808
2809 /***********************************************************************
2810 * AcceptEx
2811 */
2812 static BOOL WINAPI WS2_AcceptEx(SOCKET listener, SOCKET acceptor, PVOID dest, DWORD dest_len,
2813 DWORD local_addr_len, DWORD rem_addr_len, LPDWORD received,
2814 LPOVERLAPPED overlapped)
2815 {
2816 DWORD status;
2817 struct ws2_accept_async *wsa;
2818 int fd;
2819
2820 TRACE("(%04lx, %04lx, %p, %d, %d, %d, %p, %p)\n", listener, acceptor, dest, dest_len, local_addr_len,
2821 rem_addr_len, received, overlapped);
2822
2823 if (!dest)
2824 {
2825 SetLastError(WSAEINVAL);
2826 return FALSE;
2827 }
2828
2829 if (!overlapped)
2830 {
2831 SetLastError(WSA_INVALID_PARAMETER);
2832 return FALSE;
2833 }
2834
2835 if (!rem_addr_len)
2836 {
2837 SetLastError(WSAEFAULT);
2838 return FALSE;
2839 }
2840
2841 fd = get_sock_fd( listener, FILE_READ_DATA, NULL );
2842 if (fd == -1)
2843 {
2844 SetLastError(WSAENOTSOCK);
2845 return FALSE;
2846 }
2847 release_sock_fd( listener, fd );
2848
2849 fd = get_sock_fd( acceptor, FILE_READ_DATA, NULL );
2850 if (fd == -1)
2851 {
2852 SetLastError(WSAENOTSOCK);
2853 return FALSE;
2854 }
2855 release_sock_fd( acceptor, fd );
2856
2857 wsa = (struct ws2_accept_async *)alloc_async_io( sizeof(*wsa), WS2_async_accept );
2858 if(!wsa)
2859 {
2860 SetLastError(WSAEFAULT);
2861 return FALSE;
2862 }
2863
2864 wsa->listen_socket = SOCKET2HANDLE(listener);
2865 wsa->accept_socket = SOCKET2HANDLE(acceptor);
2866 wsa->user_overlapped = overlapped;
2867 wsa->cvalue = !((ULONG_PTR)overlapped->hEvent & 1) ? (ULONG_PTR)overlapped : 0;
2868 wsa->buf = dest;
2869 wsa->data_len = dest_len;
2870 wsa->local_len = local_addr_len;
2871 wsa->remote_len = rem_addr_len;
2872 wsa->read = NULL;
2873
2874 if (wsa->data_len)
2875 {
2876 /* set up a read request if we need it */
2877 wsa->read = (struct ws2_async *)alloc_async_io( offsetof(struct ws2_async, iovec[1]), WS2_async_accept_recv );
2878 if (!wsa->read)
2879 {
2880 HeapFree( GetProcessHeap(), 0, wsa );
2881 SetLastError(WSAEFAULT);
2882 return FALSE;
2883 }
2884
2885 wsa->read->hSocket = wsa->accept_socket;
2886 wsa->read->flags = 0;
2887 wsa->read->lpFlags = &wsa->read->flags;
2888 wsa->read->addr = NULL;
2889 wsa->read->addrlen.ptr = NULL;
2890 wsa->read->control = NULL;
2891 wsa->read->n_iovecs = 1;
2892 wsa->read->first_iovec = 0;
2893 wsa->read->completion_func = NULL;
2894 wsa->read->iovec[0].iov_base = wsa->buf;
2895 wsa->read->iovec[0].iov_len = wsa->data_len;
2896 }
2897
2898 status = register_async( ASYNC_TYPE_READ, SOCKET2HANDLE(listener), &wsa->io,
2899 overlapped->hEvent, NULL, (void *)wsa->cvalue, (IO_STATUS_BLOCK *)overlapped );
2900
2901 if(status != STATUS_PENDING)
2902 {
2903 HeapFree( GetProcessHeap(), 0, wsa->read );
2904 HeapFree( GetProcessHeap(), 0, wsa );
2905 }
2906
2907 SetLastError( NtStatusToWSAError(status) );
2908 return FALSE;
2909 }
2910
2911 /***********************************************************************
2912 * WS2_ReadFile (INTERNAL)
2913 *
2914 * Perform an APC-safe ReadFile operation
2915 */
2916 static NTSTATUS WS2_ReadFile(HANDLE hFile, PIO_STATUS_BLOCK io_status, char* buffer, ULONG length,
2917 PLARGE_INTEGER offset)
2918 {
2919 int result = -1, unix_handle;
2920 unsigned int options;
2921 NTSTATUS status;
2922
2923 TRACE( "(%p,%p,0x%08x)\n", hFile, buffer,length );
2924
2925 status = wine_server_handle_to_fd( hFile, FILE_READ_DATA, &unix_handle, &options );
2926 if (status) return status;
2927
2928 while (result == -1)
2929 {
2930 if (offset->QuadPart != FILE_USE_FILE_POINTER_POSITION)
2931 result = pread( unix_handle, buffer, length, offset->QuadPart );
2932 else
2933 result = read( unix_handle, buffer, length );
2934 if (errno != EINTR)
2935 break;
2936 }
2937
2938 if (!result)
2939 status = (length ? STATUS_END_OF_FILE : STATUS_SUCCESS);
2940 else if (result != -1)
2941 status = STATUS_SUCCESS;
2942 else if (errno != EAGAIN)
2943 status = wsaErrStatus();
2944 else
2945 status = STATUS_PENDING;
2946
2947 wine_server_release_fd( hFile, unix_handle );
2948 TRACE("= 0x%08x (%d)\n", status, result);
2949 if (status == STATUS_SUCCESS || status == STATUS_END_OF_FILE)
2950 {
2951 io_status->u.Status = status;
2952 io_status->Information = result;
2953 }
2954
2955 return status;
2956 }
2957
2958 /***********************************************************************
2959 * WS2_transmitfile_getbuffer (INTERNAL)
2960 *
2961 * Pick the appropriate buffer for a TransmitFile send operation.
2962 */
2963 static NTSTATUS WS2_transmitfile_getbuffer( int fd, struct ws2_transmitfile_async *wsa )
2964 {
2965 /* send any incomplete writes from a previous iteration */
2966 if (wsa->write.first_iovec < wsa->write.n_iovecs)
2967 return STATUS_PENDING;
2968
2969 /* process the header (if applicable) */
2970 if (wsa->buffers.Head)
2971 {
2972 wsa->write.first_iovec = 0;
2973 wsa->write.n_iovecs = 1;
2974 wsa->write.iovec[0].iov_base = wsa->buffers.Head;
2975 wsa->write.iovec[0].iov_len = wsa->buffers.HeadLength;
2976 wsa->buffers.Head = NULL;
2977 return STATUS_PENDING;
2978 }
2979
2980 /* process the main file */
2981 if (wsa->file)
2982 {
2983 DWORD bytes_per_send = wsa->bytes_per_send;
2984 IO_STATUS_BLOCK iosb;
2985 NTSTATUS status;
2986
2987 iosb.Information = 0;
2988 /* when the size of the transfer is limited ensure that we don't go past that limit */
2989 if (wsa->file_bytes != 0)
2990 bytes_per_send = min(bytes_per_send, wsa->file_bytes - wsa->file_read);
2991 status = WS2_ReadFile( wsa->file, &iosb, wsa->buffer, bytes_per_send, &wsa->offset );
2992 if (wsa->offset.QuadPart != FILE_USE_FILE_POINTER_POSITION)
2993 wsa->offset.QuadPart += iosb.Information;
2994 if (status == STATUS_END_OF_FILE)
2995 wsa->file = NULL; /* continue on to the footer */
2996 else if (status != STATUS_SUCCESS)
2997 return status;
2998 else
2999 {
3000 if (iosb.Information)
3001 {
3002 wsa->write.first_iovec = 0;
3003 wsa->write.n_iovecs = 1;
3004 wsa->write.iovec[0].iov_base = wsa->buffer;
3005 wsa->write.iovec[0].iov_len = iosb.Information;
3006 wsa->file_read += iosb.Information;
3007 }
3008
3009 if (wsa->file_bytes != 0 && wsa->file_read >= wsa->file_bytes)
3010 wsa->file = NULL;
3011
3012 return STATUS_PENDING;
3013 }
3014 }
3015
3016 /* send the footer (if applicable) */
3017 if (wsa->buffers.Tail)
3018 {
3019 wsa->write.first_iovec = 0;
3020 wsa->write.n_iovecs = 1;
3021 wsa->write.iovec[0].iov_base = wsa->buffers.Tail;
3022 wsa->write.iovec[0].iov_len = wsa->buffers.TailLength;
3023 wsa->buffers.Tail = NULL;
3024 return STATUS_PENDING;
3025 }
3026
3027 return STATUS_SUCCESS;
3028 }
3029
3030 /***********************************************************************
3031 * WS2_transmitfile_base (INTERNAL)
3032 *
3033 * Shared implementation for both synchronous and asynchronous TransmitFile.
3034 */
3035 static NTSTATUS WS2_transmitfile_base( int fd, struct ws2_transmitfile_async *wsa )
3036 {
3037 NTSTATUS status;
3038
3039 status = WS2_transmitfile_getbuffer( fd, wsa );
3040 if (status == STATUS_PENDING)
3041 {
3042 IO_STATUS_BLOCK *iosb = (IO_STATUS_BLOCK *)wsa->write.user_overlapped;
3043 int n;
3044
3045 n = WS2_send( fd, &wsa->write, convert_flags(wsa->write.flags) );
3046 if (n >= 0)
3047 {
3048 if (iosb) iosb->Information += n;
3049 }
3050 else if (errno != EAGAIN)
3051 return wsaErrStatus();
3052 }
3053
3054 return status;
3055 }
3056
3057 /***********************************************************************
3058 * WS2_async_transmitfile (INTERNAL)
3059 *
3060 * Asynchronous callback for overlapped TransmitFile operations.
3061 */
3062 static NTSTATUS WS2_async_transmitfile( void *user, IO_STATUS_BLOCK *iosb, NTSTATUS status )
3063 {
3064 struct ws2_transmitfile_async *wsa = user;
3065 int fd;
3066
3067 if (status == STATUS_ALERTED)
3068 {
3069 if (!(status = wine_server_handle_to_fd( wsa->write.hSocket, FILE_WRITE_DATA, &fd, NULL )))
3070 {
3071 status = WS2_transmitfile_base( fd, wsa );
3072 wine_server_release_fd( wsa->write.hSocket, fd );
3073 }
3074 if (status == STATUS_PENDING)
3075 return status;
3076 }
3077
3078 iosb->u.Status = status;
3079 release_async_io( &wsa->io );
3080 return status;
3081 }
3082
3083 /***********************************************************************
3084 * TransmitFile
3085 */
3086 static BOOL WINAPI WS2_TransmitFile( SOCKET s, HANDLE h, DWORD file_bytes, DWORD bytes_per_send,
3087 LPOVERLAPPED overlapped, LPTRANSMIT_FILE_BUFFERS buffers,
3088 DWORD flags )
3089 {
3090 union generic_unix_sockaddr uaddr;
3091 socklen_t uaddrlen = sizeof(uaddr);
3092 struct ws2_transmitfile_async *wsa;
3093 NTSTATUS status;
3094 int fd;
3095
3096 TRACE("(%lx, %p, %d, %d, %p, %p, %d)\n", s, h, file_bytes, bytes_per_send, overlapped,
3097 buffers, flags );
3098
3099 fd = get_sock_fd( s, FILE_WRITE_DATA, NULL );
3100 if (fd == -1)
3101 {
3102 WSASetLastError( WSAENOTSOCK );
3103 return FALSE;
3104 }
3105 if (getpeername( fd, &uaddr.addr, &uaddrlen ) != 0)
3106 {
3107 release_sock_fd( s, fd );
3108 WSASetLastError( WSAENOTCONN );
3109 return FALSE;
3110 }
3111 if (flags)
3112 FIXME("Flags are not currently supported (0x%x).\n", flags);
3113
3114 if (h && GetFileType( h ) != FILE_TYPE_DISK)
3115 {
3116 FIXME("Non-disk file handles are not currently supported.\n");
3117 release_sock_fd( s, fd );
3118 WSASetLastError( WSAEOPNOTSUPP );
3119 return FALSE;
3120 }
3121
3122 /* set reasonable defaults when requested */
3123 if (!bytes_per_send)
3124 bytes_per_send = (1 << 16); /* Depends on OS version: PAGE_SIZE, 2*PAGE_SIZE, or 2^16 */
3125
3126 if (!(wsa = (struct ws2_transmitfile_async *)alloc_async_io( sizeof(*wsa) + bytes_per_send,
3127 WS2_async_transmitfile )))
3128 {
3129 release_sock_fd( s, fd );
3130 WSASetLastError( WSAEFAULT );
3131 return FALSE;
3132 }
3133 if (buffers)
3134 wsa->buffers = *buffers;
3135 else
3136 memset(&wsa->buffers, 0x0, sizeof(wsa->buffers));
3137 wsa->buffer = (char *)(wsa + 1);
3138 wsa->file = h;
3139 wsa->file_read = 0;
3140 wsa->file_bytes = file_bytes;
3141 wsa->bytes_per_send = bytes_per_send;
3142 wsa->flags = flags;
3143 wsa->offset.QuadPart = FILE_USE_FILE_POINTER_POSITION;
3144 wsa->write.hSocket = SOCKET2HANDLE(s);
3145 wsa->write.addr = NULL;
3146 wsa->write.addrlen.val = 0;
3147 wsa->write.flags = 0;
3148 wsa->write.lpFlags = &wsa->flags;
3149 wsa->write.control = NULL;
3150 wsa->write.n_iovecs = 0;
3151 wsa->write.first_iovec = 0;
3152 wsa->write.user_overlapped = overlapped;
3153 if (overlapped)
3154 {
3155 IO_STATUS_BLOCK *iosb = (IO_STATUS_BLOCK *)overlapped;
3156 int status;
3157
3158 wsa->offset.u.LowPart = overlapped->u.s.Offset;
3159 wsa->offset.u.HighPart = overlapped->u.s.OffsetHigh;
3160 iosb->u.Status = STATUS_PENDING;
3161 iosb->Information = 0;
3162 status = register_async( ASYNC_TYPE_WRITE, SOCKET2HANDLE(s), &wsa->io,
3163 overlapped->hEvent, NULL, NULL, iosb );
3164 if(status != STATUS_PENDING) HeapFree( GetProcessHeap(), 0, wsa );
3165 release_sock_fd( s, fd );
3166 WSASetLastError( NtStatusToWSAError(status) );
3167 return FALSE;
3168 }
3169
3170 do
3171 {
3172 status = WS2_transmitfile_base( fd, wsa );
3173 if (status == STATUS_PENDING)
3174 {
3175 /* block here */
3176 do_block(fd, POLLOUT, -1);
3177 _sync_sock_state(s); /* let wineserver notice connection */
3178 }
3179 }
3180 while (status == STATUS_PENDING);
3181 release_sock_fd( s, fd );
3182
3183 if (status != STATUS_SUCCESS)
3184 WSASetLastError( NtStatusToWSAError(status) );
3185 HeapFree( GetProcessHeap(), 0, wsa );
3186 return (status == STATUS_SUCCESS);
3187 }
3188
3189 /***********************************************************************
3190 * GetAcceptExSockaddrs
3191 */
3192 static void WINAPI WS2_GetAcceptExSockaddrs(PVOID buffer, DWORD data_size, DWORD local_size, DWORD remote_size,
3193 struct WS_sockaddr **local_addr, LPINT local_addr_len,
3194 struct WS_sockaddr **remote_addr, LPINT remote_addr_len)
3195 {
3196 char *cbuf = buffer;
3197 TRACE("(%p, %d, %d, %d, %p, %p, %p, %p)\n", buffer, data_size, local_size, remote_size, local_addr,
3198 local_addr_len, remote_addr, remote_addr_len );
3199 cbuf += data_size;
3200
3201 *local_addr_len = *(int *) cbuf;
3202 *local_addr = (struct WS_sockaddr *)(cbuf + sizeof(int));
3203
3204 cbuf += local_size;
3205
3206 *remote_addr_len = *(int *) cbuf;
3207 *remote_addr = (struct WS_sockaddr *)(cbuf + sizeof(int));
3208 }
3209
3210 /***********************************************************************
3211 * WSASendMsg
3212 */
3213 int WINAPI WSASendMsg( SOCKET s, LPWSAMSG msg, DWORD dwFlags, LPDWORD lpNumberOfBytesSent,
3214 LPWSAOVERLAPPED lpOverlapped,
3215 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine)
3216 {
3217 if (!msg)
3218 {
3219 SetLastError( WSAEFAULT );
3220 return SOCKET_ERROR;
3221 }
3222
3223 return WS2_sendto( s, msg->lpBuffers, msg->dwBufferCount, lpNumberOfBytesSent,
3224 dwFlags, msg->name, msg->namelen,
3225 lpOverlapped, lpCompletionRoutine );
3226 }
3227
3228 /***********************************************************************
3229 * WSARecvMsg
3230 *
3231 * Perform a receive operation that is capable of returning message
3232 * control headers. It is important to note that the WSAMSG parameter
3233 * must remain valid throughout the operation, even when an overlapped
3234 * receive is performed.
3235 */
3236 static int WINAPI WS2_WSARecvMsg( SOCKET s, LPWSAMSG msg, LPDWORD lpNumberOfBytesRecvd,
3237 LPWSAOVERLAPPED lpOverlapped,
3238 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
3239 {
3240 if (!msg)
3241 {
3242 SetLastError( WSAEFAULT );
3243 return SOCKET_ERROR;
3244 }
3245
3246 return WS2_recv_base( s, msg->lpBuffers, msg->dwBufferCount, lpNumberOfBytesRecvd,
3247 &msg->dwFlags, msg->name, &msg->namelen,
3248 lpOverlapped, lpCompletionRoutine, &msg->Control );
3249 }
3250
3251 /***********************************************************************
3252 * interface_bind (INTERNAL)
3253 *
3254 * Take bind() calls on any name corresponding to a local network adapter and restrict the given socket to
3255 * operating only on the specified interface. This restriction consists of two components:
3256 * 1) An outgoing packet restriction suggesting the egress interface for all packets.
3257 * 2) An incoming packet restriction dropping packets not meant for the interface.
3258 * If the function succeeds in placing these restrictions (returns TRUE) then the name for the bind() may
3259 * safely be changed to INADDR_ANY, permitting the transmission and receipt of broadcast packets on the
3260 * socket. This behavior is only relevant to UDP sockets and is needed for applications that expect to be able
3261 * to receive broadcast packets on a socket that is bound to a specific network interface.
3262 */
3263 static BOOL interface_bind( SOCKET s, int fd, struct sockaddr *addr )
3264 {
3265 struct sockaddr_in *in_sock = (struct sockaddr_in *) addr;
3266 in_addr_t bind_addr = in_sock->sin_addr.s_addr;
3267 PIP_ADAPTER_INFO adapters = NULL, adapter;
3268 BOOL ret = FALSE;
3269 DWORD adap_size;
3270 int enable = 1;
3271
3272 if (bind_addr == htonl(INADDR_ANY) || bind_addr == htonl(INADDR_LOOPBACK))
3273 return FALSE; /* Not binding to a network adapter, special interface binding unnecessary. */
3274 if (_get_fd_type(fd) != SOCK_DGRAM)
3275 return FALSE; /* Special interface binding is only necessary for UDP datagrams. */
3276 if (GetAdaptersInfo(NULL, &adap_size) != ERROR_BUFFER_OVERFLOW)
3277 goto cleanup;
3278 adapters = HeapAlloc(GetProcessHeap(), 0, adap_size);
3279 if (adapters == NULL || GetAdaptersInfo(adapters, &adap_size) != NO_ERROR)
3280 goto cleanup;
3281 /* Search the IPv4 adapter list for the appropriate binding interface */
3282 for (adapter = adapters; adapter != NULL; adapter = adapter->Next)
3283 {
3284 in_addr_t adapter_addr = (in_addr_t) inet_addr(adapter->IpAddressList.IpAddress.String);
3285
3286 if (bind_addr == adapter_addr)
3287 {
3288 #if defined(IP_BOUND_IF)
3289 /* IP_BOUND_IF sets both the incoming and outgoing restriction at once */
3290 if (setsockopt(fd, IPPROTO_IP, IP_BOUND_IF, &adapter->Index, sizeof(adapter->Index)) != 0)
3291 goto cleanup;
3292 ret = TRUE;
3293 #elif defined(LINUX_BOUND_IF)
3294 in_addr_t ifindex = (in_addr_t) htonl(adapter->Index);
3295 struct interface_filter specific_interface_filter;
3296 struct sock_fprog filter_prog;
3297
3298 if (setsockopt(fd, IPPROTO_IP, IP_UNICAST_IF, &ifindex, sizeof(ifindex)) != 0)
3299 goto cleanup; /* Failed to suggest egress interface */
3300 specific_interface_filter = generic_interface_filter;
3301 specific_interface_filter.iface_rule.k = adapter->Index;
3302 specific_interface_filter.ip_rule.k = htonl(adapter_addr);
3303 filter_prog.len = sizeof(generic_interface_filter)/sizeof(struct sock_filter);
3304 filter_prog.filter = (struct sock_filter *) &specific_interface_filter;
3305 if (setsockopt(fd, SOL_SOCKET, SO_ATTACH_FILTER, &filter_prog, sizeof(filter_prog)) != 0)
3306 goto cleanup; /* Failed to specify incoming packet filter */
3307 ret = TRUE;
3308 #else
3309 FIXME("Broadcast packets on interface-bound sockets are not currently supported on this platform, "
3310 "receiving broadcast packets will not work on socket %04lx.\n", s);
3311 #endif
3312 break;
3313 }
3314 }
3315 /* Will soon be switching to INADDR_ANY: permit address reuse */
3316 if (ret && setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &enable, sizeof(enable)) == 0)
3317 TRACE("Socket %04lx bound to interface index %d\n", s, adapter->Index);
3318 else
3319 ret = FALSE;
3320
3321 cleanup:
3322 if(!ret)
3323 ERR("Failed to bind to interface, receiving broadcast packets will not work on socket %04lx.\n", s);
3324 HeapFree(GetProcessHeap(), 0, adapters);
3325 return ret;
3326 }
3327
3328 /***********************************************************************
3329 * bind (WS2_32.2)
3330 */
3331 int WINAPI WS_bind(SOCKET s, const struct WS_sockaddr* name, int namelen)
3332 {
3333 int fd = get_sock_fd( s, 0, NULL );
3334 int res = SOCKET_ERROR;
3335
3336 TRACE("socket %04lx, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen);
3337
3338 if (fd != -1)
3339 {
3340 if (!name || (name->sa_family && !supported_pf(name->sa_family)))
3341 {
3342 SetLastError(WSAEAFNOSUPPORT);
3343 }
3344 else
3345 {
3346 union generic_unix_sockaddr uaddr;
3347 unsigned int uaddrlen = ws_sockaddr_ws2u(name, namelen, &uaddr);
3348 if (!uaddrlen)
3349 {
3350 SetLastError(WSAEFAULT);
3351 }
3352 else
3353 {
3354 if (name->sa_family == WS_AF_INET)
3355 {
3356 struct sockaddr_in *in4 = (struct sockaddr_in*) &uaddr;
3357 if (memcmp(&in4->sin_addr, magic_loopback_addr, 4) == 0)
3358 {
3359 /* Trying to bind to the default host interface, using
3360 * INADDR_ANY instead*/
3361 WARN("Trying to bind to magic IP address, using "
3362 "INADDR_ANY instead.\n");
3363 in4->sin_addr.s_addr = htonl(INADDR_ANY);
3364 }
3365 else if (interface_bind(s, fd, &uaddr.addr))
3366 in4->sin_addr.s_addr = htonl(INADDR_ANY);
3367 }
3368 if (bind(fd, &uaddr.addr, uaddrlen) < 0)
3369 {
3370 int loc_errno = errno;
3371 WARN("\tfailure - errno = %i\n", errno);
3372 errno = loc_errno;
3373 switch (errno)
3374 {
3375 case EADDRNOTAVAIL:
3376 SetLastError(WSAEINVAL);
3377 break;
3378 case EADDRINUSE:
3379 {
3380 int optval = 0;
3381 socklen_t optlen = sizeof(optval);
3382 /* Windows >= 2003 will return different results depending on
3383 * SO_REUSEADDR, WSAEACCES may be returned representing that
3384 * the socket hijacking protection prevented the bind */
3385 if (!getsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char *)&optval, &optlen) && optval)
3386 {
3387 SetLastError(WSAEACCES);
3388 break;
3389 }
3390 /* fall through */
3391 }
3392 default:
3393 SetLastError(wsaErrno());
3394 break;
3395 }
3396 }
3397 else
3398 {
3399 res=0; /* success */
3400 }
3401 }
3402 }
3403 release_sock_fd( s, fd );
3404 }
3405 return res;
3406 }
3407
3408 /***********************************************************************
3409 * closesocket (WS2_32.3)
3410 */
3411 int WINAPI WS_closesocket(SOCKET s)
3412 {
3413 int res = SOCKET_ERROR, fd;
3414 if (num_startup)
3415 {
3416 fd = get_sock_fd(s, FILE_READ_DATA, NULL);
3417 if (fd >= 0)
3418 {
3419 release_sock_fd(s, fd);
3420 if (CloseHandle(SOCKET2HANDLE(s)))
3421 res = 0;
3422 }
3423 else
3424 SetLastError(WSAENOTSOCK);
3425 }
3426 else
3427 SetLastError(WSANOTINITIALISED);
3428 TRACE("(socket %04lx) -> %d\n", s, res);
3429 return res;
3430 }
3431
3432 static int do_connect(int fd, const struct WS_sockaddr* name, int namelen)
3433 {
3434 union generic_unix_sockaddr uaddr;
3435 unsigned int uaddrlen = ws_sockaddr_ws2u(name, namelen, &uaddr);
3436
3437 if (!uaddrlen)
3438 return WSAEFAULT;
3439
3440 if (name->sa_family == WS_AF_INET)
3441 {
3442 struct sockaddr_in *in4 = (struct sockaddr_in*) &uaddr;
3443 if (memcmp(&in4->sin_addr, magic_loopback_addr, 4) == 0)
3444 {
3445 /* Trying to connect to magic replace-loopback address,
3446 * assuming we really want to connect to localhost */
3447 TRACE("Trying to connect to magic IP address, using "
3448 "INADDR_LOOPBACK instead.\n");
3449 in4->sin_addr.s_addr = htonl(INADDR_LOOPBACK);
3450 }
3451 }
3452
3453 if (connect(fd, &uaddr.addr, uaddrlen) == 0)
3454 return 0;
3455
3456 return wsaErrno();
3457 }
3458
3459 /***********************************************************************
3460 * connect (WS2_32.4)
3461 */
3462 int WINAPI WS_connect(SOCKET s, const struct WS_sockaddr* name, int namelen)
3463 {
3464 int fd = get_sock_fd( s, FILE_READ_DATA, NULL );
3465
3466 TRACE("socket %04lx, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen);
3467
3468 if (fd != -1)
3469 {
3470 NTSTATUS status;
3471 BOOL is_blocking;
3472 int ret = do_connect(fd, name, namelen);
3473 if (ret == 0)
3474 goto connect_success;
3475
3476 if (ret == WSAEINPROGRESS)
3477 {
3478 /* tell wineserver that a connection is in progress */
3479 _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
3480 FD_CONNECT,
3481 FD_WINE_CONNECTED|FD_WINE_LISTENING);
3482 status = _is_blocking( s, &is_blocking );
3483 if (status)
3484 {
3485 release_sock_fd( s, fd );
3486 set_error( status );
3487 return SOCKET_ERROR;
3488 }
3489 if (is_blocking)
3490 {
3491 int result;
3492 /* block here */
3493 do_block(fd, POLLIN | POLLOUT, -1);
3494 _sync_sock_state(s); /* let wineserver notice connection */
3495 /* retrieve any error codes from it */
3496 if (!(result = get_sock_error(s, FD_CONNECT_BIT))) goto connect_success;
3497 SetLastError(result);
3498 }
3499 else
3500 {
3501 SetLastError(WSAEWOULDBLOCK);
3502 }
3503 }
3504 else
3505 {
3506 SetLastError(ret);
3507 }
3508 release_sock_fd( s, fd );
3509 }
3510 return SOCKET_ERROR;
3511
3512 connect_success:
3513 release_sock_fd( s, fd );
3514 _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
3515 FD_WINE_CONNECTED|FD_READ|FD_WRITE,
3516 FD_CONNECT|FD_WINE_LISTENING);
3517 TRACE("\tconnected %04lx\n", s);
3518 return 0;
3519 }
3520
3521 /***********************************************************************
3522 * WSAConnect (WS2_32.30)
3523 */
3524 int WINAPI WSAConnect( SOCKET s, const struct WS_sockaddr* name, int namelen,
3525 LPWSABUF lpCallerData, LPWSABUF lpCalleeData,
3526 LPQOS lpSQOS, LPQOS lpGQOS )
3527 {
3528 if ( lpCallerData || lpCalleeData || lpSQOS || lpGQOS )
3529 FIXME("unsupported parameters!\n");
3530 return WS_connect( s, name, namelen );
3531 }
3532
3533 /***********************************************************************
3534 * ConnectEx
3535 */
3536 static BOOL WINAPI WS2_ConnectEx(SOCKET s, const struct WS_sockaddr* name, int namelen,
3537 PVOID sendBuf, DWORD sendBufLen, LPDWORD sent, LPOVERLAPPED ov)
3538 {
3539 int fd, ret, status;
3540
3541 if (!ov)
3542 {
3543 SetLastError( ERROR_INVALID_PARAMETER );
3544 return FALSE;
3545 }
3546
3547 fd = get_sock_fd( s, FILE_READ_DATA, NULL );
3548 if (fd == -1)
3549 {
3550 SetLastError( WSAENOTSOCK );
3551 return FALSE;
3552 }
3553
3554 TRACE("socket %04lx, ptr %p %s, length %d, sendptr %p, len %d, ov %p\n",
3555 s, name, debugstr_sockaddr(name), namelen, sendBuf, sendBufLen, ov);
3556
3557 ret = is_fd_bound(fd, NULL, NULL);
3558 if (ret <= 0)
3559 {
3560 SetLastError(ret == -1 ? wsaErrno() : WSAEINVAL);
3561 release_sock_fd( s, fd );
3562 return FALSE;
3563 }
3564
3565 ret = do_connect(fd, name, namelen);
3566 if (ret == 0)
3567 {
3568 WSABUF wsabuf;
3569
3570 _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
3571 FD_WINE_CONNECTED|FD_READ|FD_WRITE,
3572 FD_CONNECT|FD_WINE_LISTENING);
3573
3574 wsabuf.len = sendBufLen;
3575 wsabuf.buf = (char*) sendBuf;
3576
3577 /* WSASend takes care of completion if need be */
3578 if (WSASend(s, &wsabuf, sendBuf ? 1 : 0, sent, 0, ov, NULL) != SOCKET_ERROR)
3579 goto connection_success;
3580 }
3581 else if (ret == WSAEINPROGRESS)
3582 {
3583 struct ws2_async *wsa;
3584 ULONG_PTR cvalue = (((ULONG_PTR)ov->hEvent & 1) == 0) ? (ULONG_PTR)ov : 0;
3585
3586 _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
3587 FD_CONNECT,
3588 FD_WINE_CONNECTED|FD_WINE_LISTENING);
3589
3590 /* Indirectly call WSASend */
3591 if (!(wsa = (struct ws2_async *)alloc_async_io( offsetof( struct ws2_async, iovec[1] ), WS2_async_send )))
3592 {
3593 SetLastError(WSAEFAULT);
3594 }
3595 else
3596 {
3597 IO_STATUS_BLOCK *iosb = (IO_STATUS_BLOCK *)ov;
3598 iosb->u.Status = STATUS_PENDING;
3599 iosb->Information = 0;
3600
3601 wsa->hSocket = SOCKET2HANDLE(s);
3602 wsa->addr = NULL;
3603 wsa->addrlen.val = 0;
3604 wsa->flags = 0;
3605 wsa->lpFlags = &wsa->flags;
3606 wsa->control = NULL;
3607 wsa->n_iovecs = sendBuf ? 1 : 0;
3608 wsa->first_iovec = 0;
3609 wsa->completion_func = NULL;
3610 wsa->iovec[0].iov_base = sendBuf;
3611 wsa->iovec[0].iov_len = sendBufLen;
3612
3613 status = register_async( ASYNC_TYPE_WRITE, wsa->hSocket, &wsa->io, ov->hEvent,
3614 NULL, (void *)cvalue, iosb );
3615 if (status != STATUS_PENDING) HeapFree(GetProcessHeap(), 0, wsa);
3616
3617 /* If the connect already failed */
3618 if (status == STATUS_PIPE_DISCONNECTED)
3619 {
3620 ov->Internal = sock_error_to_ntstatus( get_sock_error( s, FD_CONNECT_BIT ));
3621 ov->InternalHigh = 0;
3622 if (cvalue) WS_AddCompletion( s, cvalue, ov->Internal, ov->InternalHigh, FALSE );
3623 if (ov->hEvent) NtSetEvent( ov->hEvent, NULL );
3624 status = STATUS_PENDING;
3625 }
3626 SetLastError( NtStatusToWSAError(status) );
3627 }
3628 }
3629 else
3630 {
3631 SetLastError(ret);
3632 }
3633
3634 release_sock_fd( s, fd );
3635 return FALSE;
3636
3637 connection_success:
3638 release_sock_fd( s, fd );
3639 return TRUE;
3640 }
3641
3642 /***********************************************************************
3643 * DisconnectEx
3644 */
3645 static BOOL WINAPI WS2_DisconnectEx( SOCKET s, LPOVERLAPPED ov, DWORD flags, DWORD reserved )
3646 {
3647 TRACE( "socket %04lx, ov %p, flags 0x%x, reserved 0x%x\n", s, ov, flags, reserved );
3648
3649 if (flags & TF_REUSE_SOCKET)
3650 FIXME( "Reusing socket not supported yet\n" );
3651
3652 if (ov)
3653 {
3654 ov->Internal = STATUS_PENDING;
3655 ov->InternalHigh = 0;
3656 }
3657
3658 return !WS_shutdown( s, SD_BOTH );
3659 }
3660
3661 /***********************************************************************
3662 * getpeername (WS2_32.5)
3663 */
3664 int WINAPI WS_getpeername(SOCKET s, struct WS_sockaddr *name, int *namelen)
3665 {
3666 int fd;
3667 int res;
3668
3669 TRACE("socket %04lx, ptr %p, len %08x\n", s, name, namelen ? *namelen : 0);
3670
3671 fd = get_sock_fd( s, 0, NULL );
3672 res = SOCKET_ERROR;
3673
3674 if (fd != -1)
3675 {
3676 union generic_unix_sockaddr uaddr;
3677 socklen_t uaddrlen = sizeof(uaddr);
3678
3679 if (getpeername(fd, &uaddr.addr, &uaddrlen) == 0)
3680 {
3681 if (!name || !namelen)
3682 SetLastError(WSAEFAULT);
3683 else if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0)
3684 /* The buffer was too small */
3685 SetLastError(WSAEFAULT);
3686 else
3687 {
3688 res = 0;
3689 TRACE("=> %s\n", debugstr_sockaddr(name));
3690 }
3691 }
3692 else
3693 SetLastError(wsaErrno());
3694 release_sock_fd( s, fd );
3695 }
3696 return res;
3697 }
3698
3699 /* When binding to an UDP address with filter support the getsockname call on the socket
3700 * will always return 0.0.0.0 instead of the filtered interface address. This function
3701 * checks if the socket is interface-bound on UDP and return the correct address.
3702 * This is required because applications often do a bind() with port zero followed by a
3703 * getsockname() to retrieve the port and address acquired.
3704 */
3705 static void interface_bind_check(int fd, struct sockaddr_in *addr)
3706 {
3707 #if !defined(IP_BOUND_IF) && !defined(LINUX_BOUND_IF)
3708 return;
3709 #else
3710 unsigned int ifindex;
3711 int ret;
3712 socklen_t len;
3713
3714 /* Check for IPv4, address 0.0.0.0 and UDP socket */
3715 if (addr->sin_family != AF_INET || addr->sin_addr.s_addr != 0)
3716 return;
3717 if (_get_fd_type(fd) != SOCK_DGRAM)
3718 return;
3719
3720 len = sizeof(ifindex);
3721 #if defined(IP_BOUND_IF)
3722 ret = getsockopt(fd, IPPROTO_IP, IP_BOUND_IF, &ifindex, &len);
3723 #elif defined(LINUX_BOUND_IF)
3724 ret = getsockopt(fd, IPPROTO_IP, IP_UNICAST_IF, &ifindex, &len);
3725 if (!ret) ifindex = ntohl(ifindex);
3726 #endif
3727 if (!ret)
3728 {
3729 PIP_ADAPTER_INFO adapters, adapter;
3730 DWORD adap_size;
3731
3732 if (GetAdaptersInfo(NULL, &adap_size) != ERROR_BUFFER_OVERFLOW)
3733 return;
3734 adapters = HeapAlloc(GetProcessHeap(), 0, adap_size);
3735 if (adapters && GetAdaptersInfo(adapters, &adap_size) == NO_ERROR)
3736 {
3737 /* Search the IPv4 adapter list for the appropriate bound interface */
3738 for (adapter = adapters; adapter != NULL; adapter = adapter->Next)
3739 {
3740 in_addr_t adapter_addr;
3741 if (adapter->Index != ifindex) continue;
3742
3743 adapter_addr = inet_addr(adapter->IpAddressList.IpAddress.String);
3744 addr->sin_addr.s_addr = adapter_addr;
3745 TRACE("reporting interface address from adapter %d\n", ifindex);
3746 break;
3747 }
3748 }
3749 HeapFree(GetProcessHeap(), 0, adapters);
3750 }
3751 #endif
3752 }
3753
3754 /***********************************************************************
3755 * getsockname (WS2_32.6)
3756 */
3757 int WINAPI WS_getsockname(SOCKET s, struct WS_sockaddr *name, int *namelen)
3758 {
3759 int fd;
3760 int res;
3761
3762 TRACE("socket %04lx, ptr %p, len %08x\n", s, name, namelen ? *namelen : 0);
3763
3764 /* Check if what we've received is valid. Should we use IsBadReadPtr? */
3765 if( (name == NULL) || (namelen == NULL) )
3766 {
3767 SetLastError( WSAEFAULT );
3768 return SOCKET_ERROR;
3769 }
3770
3771 fd = get_sock_fd( s, 0, NULL );
3772 res = SOCKET_ERROR;
3773
3774 if (fd != -1)
3775 {
3776 union generic_unix_sockaddr uaddr;
3777 socklen_t uaddrlen;
3778 int bound = is_fd_bound(fd, &uaddr, &uaddrlen);
3779
3780 if (bound <= 0)
3781 {
3782 SetLastError(bound == -1 ? wsaErrno() : WSAEINVAL);
3783 }
3784 else if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0)
3785 {
3786 /* The buffer was too small */
3787 SetLastError(WSAEFAULT);
3788 }
3789 else
3790 {
3791 interface_bind_check(fd, (struct sockaddr_in*) &uaddr);
3792 if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0)
3793 {
3794 /* The buffer was too small */
3795 SetLastError(WSAEFAULT);
3796 }
3797 else
3798 {
3799 res = 0;
3800 TRACE("=> %s\n", debugstr_sockaddr(name));
3801 }
3802 }
3803 release_sock_fd( s, fd );
3804 }
3805 return res;
3806 }
3807
3808 /***********************************************************************
3809 * getsockopt (WS2_32.7)
3810 */
3811 INT WINAPI WS_getsockopt(SOCKET s, INT level,
3812 INT optname, char *optval, INT *optlen)
3813 {
3814 int fd;
3815 INT ret = 0;
3816
3817 TRACE("(socket %04lx, %s, optval %s, optlen %p (%d))\n", s,
3818 debugstr_sockopt(level, optname), debugstr_optval(optval, 0),
3819 optlen, optlen ? *optlen : 0);
3820
3821 switch(level)
3822 {
3823 case WS_SOL_SOCKET:
3824 {
3825 switch(optname)
3826 {
3827 /* Handle common cases. The special cases are below, sorted
3828 * alphabetically */
3829 case WS_SO_BROADCAST:
3830 case WS_SO_DEBUG:
3831 case WS_SO_KEEPALIVE:
3832 case WS_SO_OOBINLINE:
3833 case WS_SO_RCVBUF:
3834 case WS_SO_REUSEADDR:
3835 case WS_SO_SNDBUF:
3836 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
3837 return SOCKET_ERROR;
3838 convert_sockopt(&level, &optname);
3839 if (getsockopt(fd, level, optname, optval, (socklen_t *)optlen) != 0 )
3840 {
3841 SetLastError(wsaErrno());
3842 ret = SOCKET_ERROR;
3843 }
3844 release_sock_fd( s, fd );
3845 return ret;
3846 case WS_SO_ACCEPTCONN:
3847 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
3848 return SOCKET_ERROR;
3849 if (getsockopt(fd, SOL_SOCKET, SO_ACCEPTCONN, optval, (socklen_t *)optlen) != 0 )
3850 {
3851 SetLastError(wsaErrno());
3852 ret = SOCKET_ERROR;
3853 }
3854 else
3855 {
3856 /* BSD returns != 0 while Windows return exact == 1 */
3857 if (*(int *)optval) *(int *)optval = 1;
3858 }
3859 release_sock_fd( s, fd );
3860 return ret;
3861 case WS_SO_BSP_STATE:
3862 {
3863 int req_size, addr_size;
3864 WSAPROTOCOL_INFOW infow;
3865 CSADDR_INFO *csinfo;
3866
3867 ret = ws_protocol_info(s, TRUE, &infow, &addr_size);
3868 if (ret)
3869 {
3870 if (infow.iAddressFamily == WS_AF_INET)
3871 addr_size = sizeof(struct sockaddr_in);
3872 else if (infow.iAddressFamily == WS_AF_INET6)
3873 addr_size = sizeof(struct sockaddr_in6);
3874 else
3875 {
3876 FIXME("Family %d is unsupported for SO_BSP_STATE\n", infow.iAddressFamily);
3877 SetLastError(WSAEAFNOSUPPORT);
3878 return SOCKET_ERROR;
3879 }
3880
3881 req_size = sizeof(CSADDR_INFO) + addr_size * 2;
3882 if (*optlen < req_size)
3883 {
3884 ret = 0;
3885 SetLastError(WSAEFAULT);
3886 }
3887 else
3888 {
3889 union generic_unix_sockaddr uaddr;
3890 socklen_t uaddrlen;
3891
3892 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
3893 return SOCKET_ERROR;
3894
3895 csinfo = (CSADDR_INFO*) optval;
3896
3897 /* Check if the sock is bound */
3898 if (is_fd_bound(fd, &uaddr, &uaddrlen) == 1)
3899 {
3900 csinfo->LocalAddr.lpSockaddr =
3901 (LPSOCKADDR) (optval + sizeof(CSADDR_INFO));
3902 ws_sockaddr_u2ws(&uaddr.addr, csinfo->LocalAddr.lpSockaddr, &addr_size);
3903 csinfo->LocalAddr.iSockaddrLength = addr_size;
3904 }
3905 else
3906 {
3907 csinfo->LocalAddr.lpSockaddr = NULL;
3908 csinfo->LocalAddr.iSockaddrLength = 0;
3909 }
3910
3911 /* Check if the sock is connected */
3912 if (!getpeername(fd, &uaddr.addr, &uaddrlen) &&
3913 is_sockaddr_bound(&uaddr.addr, uaddrlen))
3914 {
3915 csinfo->RemoteAddr.lpSockaddr =
3916 (LPSOCKADDR) (optval + sizeof(CSADDR_INFO) + addr_size);
3917 ws_sockaddr_u2ws(&uaddr.addr, csinfo->RemoteAddr.lpSockaddr, &addr_size);
3918 csinfo->RemoteAddr.iSockaddrLength = addr_size;
3919 }
3920 else
3921 {
3922 csinfo->RemoteAddr.lpSockaddr = NULL;
3923 csinfo->RemoteAddr.iSockaddrLength = 0;
3924 }
3925
3926 csinfo->iSocketType = infow.iSocketType;
3927 csinfo->iProtocol = infow.iProtocol;
3928 release_sock_fd( s, fd );
3929 }
3930 }
3931 return ret ? 0 : SOCKET_ERROR;
3932 }
3933 case WS_SO_DONTLINGER:
3934 {
3935 struct linger lingval;
3936 socklen_t len = sizeof(struct linger);
3937
3938 if (!optlen || *optlen < sizeof(BOOL)|| !optval)
3939 {
3940 SetLastError(WSAEFAULT);
3941 return SOCKET_ERROR;
3942 }
3943 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
3944 return SOCKET_ERROR;
3945
3946 if (getsockopt(fd, SOL_SOCKET, SO_LINGER, &lingval, &len) != 0 )
3947 {
3948 SetLastError(wsaErrno());
3949 ret = SOCKET_ERROR;
3950 }
3951 else
3952 {
3953 *(BOOL *)optval = !lingval.l_onoff;
3954 *optlen = sizeof(BOOL);
3955 }
3956
3957 release_sock_fd( s, fd );
3958 return ret;
3959 }
3960
3961 case WS_SO_CONNECT_TIME:
3962 {
3963 static int pretendtime = 0;
3964 struct WS_sockaddr addr;
3965 int len = sizeof(addr);
3966
3967 if (!optlen || *optlen < sizeof(DWORD) || !optval)
3968 {
3969 SetLastError(WSAEFAULT);
3970 return SOCKET_ERROR;
3971 }
3972 if (WS_getpeername(s, &addr, &len) == SOCKET_ERROR)
3973 *(DWORD *)optval = ~0u;
3974 else
3975 {
3976 if (!pretendtime) FIXME("WS_SO_CONNECT_TIME - faking results\n");
3977 *(DWORD *)optval = pretendtime++;
3978 }
3979 *optlen = sizeof(DWORD);
3980 return ret;
3981 }
3982 /* As mentioned in setsockopt, Windows ignores this, so we
3983 * always return true here */
3984 case WS_SO_DONTROUTE:
3985 if (!optlen || *optlen < sizeof(BOOL) || !optval)
3986 {
3987 SetLastError(WSAEFAULT);
3988 return SOCKET_ERROR;
3989 }
3990 *(BOOL *)optval = TRUE;
3991 *optlen = sizeof(BOOL);
3992 return 0;
3993
3994 case WS_SO_ERROR:
3995 {
3996 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
3997 return SOCKET_ERROR;
3998 if (getsockopt(fd, SOL_SOCKET, SO_ERROR, optval, (socklen_t *)optlen) != 0 )
3999 {
4000 SetLastError(wsaErrno());
4001 ret = SOCKET_ERROR;
4002 }
4003 release_sock_fd( s, fd );
4004
4005 /* The wineserver may have swallowed the error before us */
4006 if (!ret && *(int*) optval == 0)
4007 {
4008 int i, events[FD_MAX_EVENTS];
4009 _get_sock_errors(s, events);
4010 for (i = 0; i < FD_MAX_EVENTS; i++)
4011 {
4012 if(events[i])
4013 {
4014 TRACE("returning SO_ERROR %d from wine server\n", events[i]);
4015 *(int*) optval = events[i];
4016 break;
4017 }
4018 }
4019 }
4020 return ret;
4021 }
4022
4023 case WS_SO_LINGER:
4024 {
4025 struct linger lingval;
4026 socklen_t len = sizeof(struct linger);
4027
4028 /* struct linger and LINGER have different sizes */
4029 if (!optlen || *optlen < sizeof(LINGER) || !optval)
4030 {
4031 SetLastError(WSAEFAULT);
4032 return SOCKET_ERROR;
4033 }
4034 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
4035 return SOCKET_ERROR;
4036
4037 if (_get_fd_type(fd) == SOCK_DGRAM)
4038 {
4039 SetLastError(WSAENOPROTOOPT);
4040 ret = SOCKET_ERROR;
4041 }
4042 else if (getsockopt(fd, SOL_SOCKET, SO_LINGER, &lingval, &len) != 0)
4043 {
4044 SetLastError(wsaErrno());
4045 ret = SOCKET_ERROR;
4046 }
4047 else
4048 {
4049 ((LINGER *)optval)->l_onoff = lingval.l_onoff;
4050 ((LINGER *)optval)->l_linger = lingval.l_linger;
4051 *optlen = sizeof(struct linger);
4052 }
4053
4054 release_sock_fd( s, fd );
4055 return ret;
4056 }
4057
4058 case WS_SO_MAX_MSG_SIZE:
4059 if (!optlen || *optlen < sizeof(int) || !optval)
4060 {
4061 SetLastError(WSAEFAULT);
4062 return SOCKET_ERROR;
4063 }
4064 TRACE("getting global SO_MAX_MSG_SIZE = 65507\n");
4065 *(int *)optval = 65507;
4066 *optlen = sizeof(int);
4067 return 0;
4068
4069 /* SO_OPENTYPE does not require a valid socket handle. */
4070 case WS_SO_OPENTYPE:
4071 if (!optlen || *optlen < sizeof(int) || !optval)
4072 {
4073 SetLastError(WSAEFAULT);
4074 return SOCKET_ERROR;
4075 }
4076 *(int *)optval = get_per_thread_data()->opentype;
4077 *optlen = sizeof(int);
4078 TRACE("getting global SO_OPENTYPE = 0x%x\n", *((int*)optval) );
4079 return 0;
4080 case WS_SO_PROTOCOL_INFOA:
4081 case WS_SO_PROTOCOL_INFOW:
4082 {
4083 int size;
4084 WSAPROTOCOL_INFOW infow;
4085
4086 ret = ws_protocol_info(s, optname == WS_SO_PROTOCOL_INFOW, &infow, &size);
4087 if (ret)
4088 {
4089 if (!optlen || !optval || *optlen < size)
4090 {
4091 if(optlen) *optlen = size;
4092 ret = 0;
4093 SetLastError(WSAEFAULT);
4094 }
4095 else
4096 memcpy(optval, &infow, size);
4097 }
4098 return ret ? 0 : SOCKET_ERROR;
4099 }
4100 case WS_SO_RCVTIMEO:
4101 case WS_SO_SNDTIMEO:
4102 {
4103 INT64 timeout;
4104
4105 if (!optlen || *optlen < sizeof(int)|| !optval)
4106 {
4107 SetLastError(WSAEFAULT);
4108 return SOCKET_ERROR;
4109 }
4110 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
4111 return SOCKET_ERROR;
4112
4113 timeout = get_rcvsnd_timeo(fd, optname == WS_SO_RCVTIMEO);
4114 *(int *)optval = timeout <= UINT_MAX ? timeout : UINT_MAX;
4115
4116 release_sock_fd( s, fd );
4117 return ret;
4118 }
4119 case WS_SO_TYPE:
4120 {
4121 int sock_type;
4122 if (!optlen || *optlen < sizeof(int) || !optval)
4123 {
4124 SetLastError(WSAEFAULT);
4125 return SOCKET_ERROR;
4126 }
4127 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
4128 return SOCKET_ERROR;
4129
4130 sock_type = _get_fd_type(fd);
4131 if (sock_type == -1)
4132 {
4133 SetLastError(wsaErrno());
4134 ret = SOCKET_ERROR;
4135 }
4136 else
4137 (*(int *)optval) = convert_socktype_u2w(sock_type);
4138
4139 release_sock_fd( s, fd );
4140 return ret;
4141 }
4142 default:
4143 TRACE("Unknown SOL_SOCKET optname: 0x%08x\n", optname);
4144 SetLastError(WSAENOPROTOOPT);
4145 return SOCKET_ERROR;
4146 } /* end switch(optname) */
4147 }/* end case WS_SOL_SOCKET */
4148 #ifdef HAS_IPX
4149 case WS_NSPROTO_IPX:
4150 {
4151 struct WS_sockaddr_ipx addr;
4152 IPX_ADDRESS_DATA *data;
4153 int namelen;
4154 switch(optname)
4155 {
4156 case WS_IPX_PTYPE:
4157 if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR;
4158 #ifdef SOL_IPX
4159 if(getsockopt(fd, SOL_IPX, IPX_TYPE, optval, (socklen_t *)optlen) == -1)
4160 {
4161 ret = SOCKET_ERROR;
4162 }
4163 #else
4164 {
4165 struct ipx val;
4166 socklen_t len=sizeof(struct ipx);
4167 if(getsockopt(fd, 0, SO_DEFAULT_HEADERS, &val, &len) == -1 )
4168 ret = SOCKET_ERROR;
4169 else
4170 *optval = (int)val.ipx_pt;
4171 }
4172 #endif
4173 TRACE("ptype: %d (fd: %d)\n", *(int*)optval, fd);
4174 release_sock_fd( s, fd );
4175 return ret;
4176
4177 case WS_IPX_ADDRESS:
4178 /*
4179 * On a Win2000 system with one network card there are usually
4180 * three ipx devices one with a speed of 28.8kbps, 10Mbps and 100Mbps.
4181 * Using this call you can then retrieve info about this all.
4182 * In case of Linux it is a bit different. Usually you have
4183 * only "one" device active and further it is not possible to
4184 * query things like the linkspeed.
4185 */
4186 FIXME("IPX_ADDRESS\n");
4187 namelen = sizeof(struct WS_sockaddr_ipx);
4188 memset(&addr, 0, sizeof(struct WS_sockaddr_ipx));
4189 WS_getsockname(s, (struct WS_sockaddr*)&addr, &namelen);
4190
4191 data = (IPX_ADDRESS_DATA*)optval;
4192 memcpy(data->nodenum,addr.sa_nodenum,sizeof(data->nodenum));
4193 memcpy(data->netnum,addr.sa_netnum,sizeof(data->netnum));
4194 data->adapternum = 0;
4195 data->wan = FALSE; /* We are not on a wan for now .. */
4196 data->status = FALSE; /* Since we are not on a wan, the wan link isn't up */
4197 data->maxpkt = 1467; /* This value is the default one, at least on Win2k/WinXP */
4198 data->linkspeed = 100000; /* Set the line speed in 100bit/s to 10 Mbit;
4199 * note 1MB = 1000kB in this case */
4200 return 0;
4201
4202 case WS_IPX_MAX_ADAPTER_NUM:
4203 FIXME("IPX_MAX_ADAPTER_NUM\n");
4204 *(int*)optval = 1; /* As noted under IPX_ADDRESS we have just one card. */
4205 return 0;
4206
4207 default:
4208 FIXME("IPX optname:%x\n", optname);
4209 return SOCKET_ERROR;
4210 }/* end switch(optname) */
4211 } /* end case WS_NSPROTO_IPX */
4212 #endif
4213
4214 #ifdef HAS_IRDA
4215 #define MAX_IRDA_DEVICES 10
4216
4217 case WS_SOL_IRLMP:
4218 switch(optname)
4219 {
4220 case WS_IRLMP_ENUMDEVICES:
4221 {
4222 char buf[sizeof(struct irda_device_list) +
4223 (MAX_IRDA_DEVICES - 1) * sizeof(struct irda_device_info)];
4224 int res;
4225 socklen_t len = sizeof(buf);
4226
4227 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
4228 return SOCKET_ERROR;
4229 res = getsockopt( fd, SOL_IRLMP, IRLMP_ENUMDEVICES, buf, &len );
4230 release_sock_fd( s, fd );
4231 if (res < 0)
4232 {
4233 SetLastError(wsaErrno());
4234 return SOCKET_ERROR;
4235 }
4236 else
4237 {
4238 struct irda_device_list *src = (struct irda_device_list *)buf;
4239 DEVICELIST *dst = (DEVICELIST *)optval;
4240 INT needed = sizeof(DEVICELIST);
4241 unsigned int i;
4242
4243 if (src->len > 0)
4244 needed += (src->len - 1) * sizeof(IRDA_DEVICE_INFO);
4245 if (*optlen < needed)
4246 {
4247 SetLastError(WSAEFAULT);
4248 return SOCKET_ERROR;
4249 }
4250 *optlen = needed;
4251 TRACE("IRLMP_ENUMDEVICES: %d devices found:\n", src->len);
4252 dst->numDevice = src->len;
4253 for (i = 0; i < src->len; i++)
4254 {
4255 TRACE("saddr = %08x, daddr = %08x, info = %s, hints = %02x%02x\n",
4256 src->dev[i].saddr, src->dev[i].daddr,
4257 src->dev[i].info, src->dev[i].hints[0],
4258 src->dev[i].hints[1]);
4259 memcpy( dst->Device[i].irdaDeviceID,
4260 &src->dev[i].daddr,
4261 sizeof(dst->Device[i].irdaDeviceID) ) ;
4262 memcpy( dst->Device[i].irdaDeviceName,
4263 src->dev[i].info,
4264 sizeof(dst->Device[i].irdaDeviceName) ) ;
4265 memcpy( &dst->Device[i].irdaDeviceHints1,
4266 &src->dev[i].hints[0],
4267 sizeof(dst->Device[i].irdaDeviceHints1) ) ;
4268 memcpy( &dst->Device[i].irdaDeviceHints2,
4269 &src->dev[i].hints[1],
4270 sizeof(dst->Device[i].irdaDeviceHints2) ) ;
4271 dst->Device[i].irdaCharSet = src->dev[i].charset;
4272 }
4273 return 0;
4274 }
4275 }
4276 default:
4277 FIXME("IrDA optname:0x%x\n", optname);
4278 return SOCKET_ERROR;
4279 }
4280 break; /* case WS_SOL_IRLMP */
4281 #undef MAX_IRDA_DEVICES
4282 #endif
4283
4284 /* Levels WS_IPPROTO_TCP and WS_IPPROTO_IP convert directly */
4285 case WS_IPPROTO_TCP:
4286 switch(optname)
4287 {
4288 case WS_TCP_NODELAY:
4289 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
4290 return SOCKET_ERROR;
4291 convert_sockopt(&level, &optname);
4292 if (getsockopt(fd, level, optname, optval, (socklen_t *)optlen) != 0 )
4293 {
4294 SetLastError(wsaErrno());
4295 ret = SOCKET_ERROR;
4296 }
4297 release_sock_fd( s, fd );
4298 return ret;
4299 }
4300 FIXME("Unknown IPPROTO_TCP optname 0x%08x\n", optname);
4301 return SOCKET_ERROR;
4302
4303 case WS_IPPROTO_IP:
4304 switch(optname)
4305 {
4306 case WS_IP_ADD_MEMBERSHIP:
4307 case WS_IP_DROP_MEMBERSHIP:
4308 #ifdef IP_HDRINCL
4309 case WS_IP_HDRINCL:
4310 #endif
4311 case WS_IP_MULTICAST_IF:
4312 case WS_IP_MULTICAST_LOOP:
4313 case WS_IP_MULTICAST_TTL:
4314 case WS_IP_OPTIONS:
4315 #ifdef IP_PKTINFO
4316 case WS_IP_PKTINFO:
4317 #endif
4318 case WS_IP_TOS:
4319 case WS_IP_TTL:
4320 #ifdef IP_UNICAST_IF
4321 case WS_IP_UNICAST_IF:
4322 #endif
4323 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
4324 return SOCKET_ERROR;
4325 convert_sockopt(&level, &optname);
4326 if (getsockopt(fd, level, optname, optval, (socklen_t *)optlen) != 0 )
4327 {
4328 SetLastError(wsaErrno());
4329 ret = SOCKET_ERROR;
4330 }
4331 release_sock_fd( s, fd );
4332 return ret;
4333 case WS_IP_DONTFRAGMENT:
4334 return get_dont_fragment(s, IPPROTO_IP, (BOOL *)optval) ? 0 : SOCKET_ERROR;
4335 }
4336 FIXME("Unknown IPPROTO_IP optname 0x%08x\n", optname);
4337 return SOCKET_ERROR;
4338
4339 case WS_IPPROTO_IPV6:
4340 switch(optname)
4341 {
4342 #ifdef IPV6_ADD_MEMBERSHIP
4343 case WS_IPV6_ADD_MEMBERSHIP:
4344 #endif
4345 #ifdef IPV6_DROP_MEMBERSHIP
4346 case WS_IPV6_DROP_MEMBERSHIP:
4347 #endif
4348 case WS_IPV6_MULTICAST_IF:
4349 case WS_IPV6_MULTICAST_HOPS:
4350 case WS_IPV6_MULTICAST_LOOP:
4351 case WS_IPV6_UNICAST_HOPS:
4352 case WS_IPV6_V6ONLY:
4353 #ifdef IPV6_UNICAST_IF
4354 case WS_IPV6_UNICAST_IF:
4355 #endif
4356 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
4357 return SOCKET_ERROR;
4358 convert_sockopt(&level, &optname);
4359 if (getsockopt(fd, level, optname, optval, (socklen_t *)optlen) != 0 )
4360 {
4361 SetLastError(wsaErrno());
4362 ret = SOCKET_ERROR;
4363 }
4364 release_sock_fd( s, fd );
4365 return ret;
4366 case WS_IPV6_DONTFRAG:
4367 return get_dont_fragment(s, IPPROTO_IPV6, (BOOL *)optval) ? 0 : SOCKET_ERROR;
4368 }
4369 FIXME("Unknown IPPROTO_IPV6 optname 0x%08x\n", optname);
4370 return SOCKET_ERROR;
4371
4372 default:
4373 WARN("Unknown level: 0x%08x\n", level);
4374 SetLastError(WSAEINVAL);
4375 return SOCKET_ERROR;
4376 } /* end switch(level) */
4377 }
4378
4379 /***********************************************************************
4380 * htonl (WS2_32.8)
4381 */
4382 WS_u_long WINAPI WS_htonl(WS_u_long hostlong)
4383 {
4384 return htonl(hostlong);
4385 }
4386
4387
4388 /***********************************************************************
4389 * htons (WS2_32.9)
4390 */
4391 WS_u_short WINAPI WS_htons(WS_u_short hostshort)
4392 {
4393 return htons(hostshort);
4394 }
4395
4396 /***********************************************************************
4397 * WSAHtonl (WS2_32.46)
4398 * From MSDN description of error codes, this function should also
4399 * check if WinSock has been initialized and the socket is a valid
4400 * socket. But why? This function only translates a host byte order
4401 * u_long into a network byte order u_long...
4402 */
4403 int WINAPI WSAHtonl(SOCKET s, WS_u_long hostlong, WS_u_long *lpnetlong)
4404 {
4405 if (lpnetlong)
4406 {
4407 *lpnetlong = htonl(hostlong);
4408 return 0;
4409 }
4410 SetLastError(WSAEFAULT);
4411 return SOCKET_ERROR;
4412 }
4413
4414 /***********************************************************************
4415 * WSAHtons (WS2_32.47)
4416 * From MSDN description of error codes, this function should also
4417 * check if WinSock has been initialized and the socket is a valid
4418 * socket. But why? This function only translates a host byte order
4419 * u_short into a network byte order u_short...
4420 */
4421 int WINAPI WSAHtons(SOCKET s, WS_u_short hostshort, WS_u_short *lpnetshort)
4422 {
4423
4424 if (lpnetshort)
4425 {
4426 *lpnetshort = htons(hostshort);
4427 return 0;
4428 }
4429 SetLastError(WSAEFAULT);
4430 return SOCKET_ERROR;
4431 }
4432
4433
4434 /***********************************************************************
4435 * inet_addr (WS2_32.11)
4436 */
4437 WS_u_long WINAPI WS_inet_addr(const char *cp)
4438 {
4439 if (!cp) return INADDR_NONE;
4440 return inet_addr(cp);
4441 }
4442
4443
4444 /***********************************************************************
4445 * ntohl (WS2_32.14)
4446 */
4447 WS_u_long WINAPI WS_ntohl(WS_u_long netlong)
4448 {
4449 return ntohl(netlong);
4450 }
4451
4452
4453 /***********************************************************************
4454 * ntohs (WS2_32.15)
4455 */
4456 WS_u_short WINAPI WS_ntohs(WS_u_short netshort)
4457 {
4458 return ntohs(netshort);
4459 }
4460
4461
4462 /***********************************************************************
4463 * inet_ntoa (WS2_32.12)
4464 */
4465 char* WINAPI WS_inet_ntoa(struct WS_in_addr in)
4466 {
4467 unsigned int long_ip = ntohl(in.WS_s_addr);
4468 struct per_thread_data *data = get_per_thread_data();
4469
4470 sprintf( data->ntoa_buffer, "%u.%u.%u.%u",
4471 (long_ip >> 24) & 0xff,
4472 (long_ip >> 16) & 0xff,
4473 (long_ip >> 8) & 0xff,
4474 long_ip & 0xff);
4475
4476 return data->ntoa_buffer;
4477 }
4478
4479 static const char *debugstr_wsaioctl(DWORD code)
4480 {
4481 const char *name = NULL, *buf_type, *family;
4482
4483 #define IOCTL_NAME(x) case x: name = #x; break
4484 switch (code)
4485 {
4486 IOCTL_NAME(WS_FIONBIO);
4487 IOCTL_NAME(WS_FIONREAD);
4488 IOCTL_NAME(WS_SIOCATMARK);
4489 /* IOCTL_NAME(WS_SIO_ACQUIRE_PORT_RESERVATION); */
4490 IOCTL_NAME(WS_SIO_ADDRESS_LIST_CHANGE);
4491 IOCTL_NAME(WS_SIO_ADDRESS_LIST_QUERY);
4492 IOCTL_NAME(WS_SIO_ASSOCIATE_HANDLE);
4493 /* IOCTL_NAME(WS_SIO_ASSOCIATE_PORT_RESERVATION);
4494 IOCTL_NAME(WS_SIO_BASE_HANDLE);
4495 IOCTL_NAME(WS_SIO_BSP_HANDLE);
4496 IOCTL_NAME(WS_SIO_BSP_HANDLE_SELECT);
4497 IOCTL_NAME(WS_SIO_BSP_HANDLE_POLL);
4498 IOCTL_NAME(WS_SIO_CHK_QOS); */
4499 IOCTL_NAME(WS_SIO_ENABLE_CIRCULAR_QUEUEING);
4500 IOCTL_NAME(WS_SIO_FIND_ROUTE);
4501 IOCTL_NAME(WS_SIO_FLUSH);
4502 IOCTL_NAME(WS_SIO_GET_BROADCAST_ADDRESS);
4503 IOCTL_NAME(WS_SIO_GET_EXTENSION_FUNCTION_POINTER);
4504 IOCTL_NAME(WS_SIO_GET_GROUP_QOS);
4505 IOCTL_NAME(WS_SIO_GET_INTERFACE_LIST);
4506 /* IOCTL_NAME(WS_SIO_GET_INTERFACE_LIST_EX); */
4507 IOCTL_NAME(WS_SIO_GET_QOS);
4508 /* IOCTL_NAME(WS_SIO_IDEAL_SEND_BACKLOG_CHANGE);
4509 IOCTL_NAME(WS_SIO_IDEAL_SEND_BACKLOG_QUERY); */
4510 IOCTL_NAME(WS_SIO_KEEPALIVE_VALS);
4511 IOCTL_NAME(WS_SIO_MULTIPOINT_LOOPBACK);
4512 IOCTL_NAME(WS_SIO_MULTICAST_SCOPE);
4513 /* IOCTL_NAME(WS_SIO_QUERY_RSS_SCALABILITY_INFO);
4514 IOCTL_NAME(WS_SIO_QUERY_WFP_ALE_ENDPOINT_HANDLE); */
4515 IOCTL_NAME(WS_SIO_RCVALL);
4516 IOCTL_NAME(WS_SIO_RCVALL_IGMPMCAST);
4517 IOCTL_NAME(WS_SIO_RCVALL_MCAST);
4518 /* IOCTL_NAME(WS_SIO_RELEASE_PORT_RESERVATION); */
4519 IOCTL_NAME(WS_SIO_ROUTING_INTERFACE_CHANGE);
4520 IOCTL_NAME(WS_SIO_ROUTING_INTERFACE_QUERY);
4521 IOCTL_NAME(WS_SIO_SET_COMPATIBILITY_MODE);
4522 IOCTL_NAME(WS_SIO_SET_GROUP_QOS);
4523 IOCTL_NAME(WS_SIO_SET_QOS);
4524 IOCTL_NAME(WS_SIO_TRANSLATE_HANDLE);
4525 IOCTL_NAME(WS_SIO_UDP_CONNRESET);
4526 }
4527 #undef IOCTL_NAME
4528
4529 if (name)
4530 return name + 3;
4531
4532 /* If this is not a known code split its bits */
4533 switch(code & 0x18000000)
4534 {
4535 case WS_IOC_WS2:
4536 family = "IOC_WS2";
4537 break;
4538 case WS_IOC_PROTOCOL:
4539 family = "IOC_PROTOCOL";
4540 break;
4541 case WS_IOC_VENDOR:
4542 family = "IOC_VENDOR";
4543 break;
4544 default: /* WS_IOC_UNIX */
4545 {
4546 BYTE size = (code >> 16) & WS_IOCPARM_MASK;
4547 char x = (code & 0xff00) >> 8;
4548 BYTE y = code & 0xff;
4549 char args[14];
4550
4551 switch (code & (WS_IOC_VOID|WS_IOC_INOUT))
4552 {
4553 case WS_IOC_VOID:
4554 buf_type = "_IO";
4555 sprintf(args, "%d, %d", x, y);
4556 break;
4557 case WS_IOC_IN:
4558 buf_type = "_IOW";
4559 sprintf(args, "'%c', %d, %d", x, y, size);
4560 break;
4561 case WS_IOC_OUT:
4562 buf_type = "_IOR";
4563 sprintf(args, "'%c', %d, %d", x, y, size);
4564 break;
4565 default:
4566 buf_type = "?";
4567 sprintf(args, "'%c', %d, %d", x, y, size);
4568 break;
4569 }
4570 return wine_dbg_sprintf("%s(%s)", buf_type, args);
4571 }
4572 }
4573
4574 /* We are different from WS_IOC_UNIX. */
4575 switch (code & (WS_IOC_VOID|WS_IOC_INOUT))
4576 {
4577 case WS_IOC_VOID:
4578 buf_type = "_WSAIO";
4579 break;
4580 case WS_IOC_INOUT:
4581 buf_type = "_WSAIORW";
4582 break;
4583 case WS_IOC_IN:
4584 buf_type = "_WSAIOW";
4585 break;
4586 case WS_IOC_OUT:
4587 buf_type = "_WSAIOR";
4588 break;
4589 default:
4590 buf_type = "?";
4591 break;
4592 }
4593
4594 return wine_dbg_sprintf("%s(%s, %d)", buf_type, family,
4595 (USHORT)(code & 0xffff));
4596 }
4597
4598 /* do an ioctl call through the server */
4599 static DWORD server_ioctl_sock( SOCKET s, DWORD code, LPVOID in_buff, DWORD in_size,
4600 LPVOID out_buff, DWORD out_size, LPDWORD ret_size,
4601 LPWSAOVERLAPPED overlapped,
4602 LPWSAOVERLAPPED_COMPLETION_ROUTINE completion )
4603 {
4604 HANDLE event = overlapped ? overlapped->hEvent : 0;
4605 HANDLE handle = SOCKET2HANDLE( s );
4606 struct ws2_async *wsa = NULL;
4607 IO_STATUS_BLOCK *io = (PIO_STATUS_BLOCK)overlapped, iosb;
4608 void *cvalue = NULL;
4609 NTSTATUS status;
4610
4611 if (completion)
4612 {
4613 if (!(wsa = (struct ws2_async *)alloc_async_io( sizeof(*wsa), NULL )))
4614 return WSA_NOT_ENOUGH_MEMORY;
4615 wsa->hSocket = handle;
4616 wsa->user_overlapped = overlapped;
4617 wsa->completion_func = completion;
4618 if (!io) io = &wsa->local_iosb;
4619 cvalue = wsa;
4620 }
4621 else if (!io)
4622 io = &iosb;
4623 else if (!((ULONG_PTR)overlapped->hEvent & 1))
4624 cvalue = overlapped;
4625
4626 status = NtDeviceIoControlFile( handle, event, wsa ? ws2_async_apc : NULL, cvalue, io, code,
4627 in_buff, in_size, out_buff, out_size );
4628 if (status == STATUS_NOT_SUPPORTED)
4629 {
4630 FIXME("Unsupported ioctl %x (device=%x access=%x func=%x method=%x)\n",
4631 code, code >> 16, (code >> 14) & 3, (code >> 2) & 0xfff, code & 3);
4632 }
4633 else if (status == STATUS_SUCCESS)
4634 *ret_size = io->Information; /* "Information" is the size written to the output buffer */
4635
4636 if (status != STATUS_PENDING) RtlFreeHeap( GetProcessHeap(), 0, wsa );
4637
4638 return NtStatusToWSAError( status );
4639 }
4640
4641 /**********************************************************************
4642 * WSAIoctl (WS2_32.50)
4643 *
4644 */
4645 INT WINAPI WSAIoctl(SOCKET s, DWORD code, LPVOID in_buff, DWORD in_size, LPVOID out_buff,
4646 DWORD out_size, LPDWORD ret_size, LPWSAOVERLAPPED overlapped,
4647 LPWSAOVERLAPPED_COMPLETION_ROUTINE completion )
4648 {
4649 int fd;
4650 DWORD status = 0, total = 0;
4651
4652 TRACE("%04lx, %s, %p, %d, %p, %d, %p, %p, %p\n",
4653 s, debugstr_wsaioctl(code), in_buff, in_size, out_buff, out_size, ret_size, overlapped, completion);
4654
4655 switch (code)
4656 {
4657 case WS_FIONBIO:
4658 if (in_size != sizeof(WS_u_long) || IS_INTRESOURCE(in_buff))
4659 {
4660 SetLastError(WSAEFAULT);
4661 return SOCKET_ERROR;
4662 }
4663 TRACE("-> FIONBIO (%x)\n", *(WS_u_long*)in_buff);
4664 if (_get_sock_mask(s))
4665 {
4666 /* AsyncSelect()'ed sockets are always nonblocking */
4667 if (!*(WS_u_long *)in_buff) status = WSAEINVAL;
4668 break;
4669 }
4670 if (*(WS_u_long *)in_buff)
4671 _enable_event(SOCKET2HANDLE(s), 0, FD_WINE_NONBLOCKING, 0);
4672 else
4673 _enable_event(SOCKET2HANDLE(s), 0, 0, FD_WINE_NONBLOCKING);
4674 break;
4675
4676 case WS_FIONREAD:
4677 {
4678 #if defined(linux)
4679 int listening = 0;
4680 socklen_t len = sizeof(listening);
4681 #endif
4682 if (out_size != sizeof(WS_u_long) || IS_INTRESOURCE(out_buff))
4683 {
4684 SetLastError(WSAEFAULT);
4685 return SOCKET_ERROR;
4686 }
4687 if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR;
4688
4689 #if defined(linux)
4690 /* On Linux, FIONREAD on listening socket always fails (see tcp(7)).
4691 However, it succeeds on native. */
4692 if (!getsockopt( fd, SOL_SOCKET, SO_ACCEPTCONN, &listening, &len ) && listening)
4693 (*(WS_u_long *) out_buff) = 0;
4694 else
4695 #endif
4696 if (ioctl(fd, FIONREAD, out_buff ) == -1)
4697 status = wsaErrno();
4698 release_sock_fd( s, fd );
4699 break;
4700 }
4701
4702 case WS_SIOCATMARK:
4703 {
4704 unsigned int oob = 0, atmark = 0;
4705 socklen_t oobsize = sizeof(int);
4706 if (out_size != sizeof(WS_u_long) || IS_INTRESOURCE(out_buff))
4707 {
4708 SetLastError(WSAEFAULT);
4709 return SOCKET_ERROR;
4710 }
4711 if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR;
4712 /* SO_OOBINLINE sockets must always return TRUE to SIOCATMARK */
4713 if ((getsockopt(fd, SOL_SOCKET, SO_OOBINLINE, &oob, &oobsize ) == -1)
4714 || (!oob && ioctl(fd, SIOCATMARK, &atmark ) == -1))
4715 status = wsaErrno();
4716 else
4717 {
4718 /* The SIOCATMARK value read from ioctl() is reversed
4719 * because BSD returns TRUE if it's in the OOB mark
4720 * while Windows returns TRUE if there are NO OOB bytes.
4721 */
4722 (*(WS_u_long *) out_buff) = oob || !atmark;
4723 }
4724
4725 release_sock_fd( s, fd );
4726 break;
4727 }
4728
4729 case WS_FIOASYNC:
4730 WARN("Warning: WS1.1 shouldn't be using async I/O\n");
4731 SetLastError(WSAEINVAL);
4732 return SOCKET_ERROR;
4733
4734 case WS_SIO_GET_INTERFACE_LIST:
4735 {
4736 INTERFACE_INFO* intArray = out_buff;
4737 DWORD size, numInt = 0, apiReturn;
4738
4739 TRACE("-> SIO_GET_INTERFACE_LIST request\n");
4740
4741 if (!out_buff || !ret_size)
4742 {
4743 SetLastError(WSAEFAULT);
4744 return SOCKET_ERROR;
4745 }
4746
4747 fd = get_sock_fd( s, 0, NULL );
4748 if (fd == -1) return SOCKET_ERROR;
4749
4750 apiReturn = GetAdaptersInfo(NULL, &size);
4751 if (apiReturn == ERROR_BUFFER_OVERFLOW)
4752 {
4753 PIP_ADAPTER_INFO table = HeapAlloc(GetProcessHeap(),0,size);
4754
4755 if (table)
4756 {
4757 if (GetAdaptersInfo(table, &size) == NO_ERROR)
4758 {
4759 PIP_ADAPTER_INFO ptr;
4760
4761 for (ptr = table, numInt = 0; ptr; ptr = ptr->Next)
4762 {
4763 unsigned int addr, mask, bcast;
4764 struct ifreq ifInfo;
4765
4766 /* Skip interfaces without an IPv4 address. */
4767 if (ptr->IpAddressList.IpAddress.String[0] == '\0')
4768 continue;
4769
4770 if ((numInt + 1)*sizeof(INTERFACE_INFO)/sizeof(IP_ADAPTER_INFO) > out_size)
4771 {
4772 WARN("Buffer too small = %u, out_size = %u\n", numInt + 1, out_size);
4773 status = WSAEFAULT;
4774 break;
4775 }
4776
4777 /* Socket Status Flags */
4778 lstrcpynA(ifInfo.ifr_name, ptr->AdapterName, IFNAMSIZ);
4779 if (ioctl(fd, SIOCGIFFLAGS, &ifInfo) < 0)
4780 {
4781 ERR("Error obtaining status flags for socket!\n");
4782 status = WSAEINVAL;
4783 break;
4784 }
4785 else
4786 {
4787 /* set flags; the values of IFF_* are not the same
4788 under Linux and Windows, therefore must generate
4789 new flags */
4790 intArray->iiFlags = 0;
4791 if (ifInfo.ifr_flags & IFF_BROADCAST)
4792 intArray->iiFlags |= WS_IFF_BROADCAST;
4793 #ifdef IFF_POINTOPOINT
4794 if (ifInfo.ifr_flags & IFF_POINTOPOINT)
4795 intArray->iiFlags |= WS_IFF_POINTTOPOINT;
4796 #endif
4797 if (ifInfo.ifr_flags & IFF_LOOPBACK)
4798 intArray->iiFlags |= WS_IFF_LOOPBACK;
4799 if (ifInfo.ifr_flags & IFF_UP)
4800 intArray->iiFlags |= WS_IFF_UP;
4801 if (ifInfo.ifr_flags & IFF_MULTICAST)
4802 intArray->iiFlags |= WS_IFF_MULTICAST;
4803 }
4804
4805 addr = inet_addr(ptr->IpAddressList.IpAddress.String);
4806 mask = inet_addr(ptr->IpAddressList.IpMask.String);
4807 bcast = addr | ~mask;
4808 intArray->iiAddress.AddressIn.sin_family = WS_AF_INET;
4809 intArray->iiAddress.AddressIn.sin_port = 0;
4810 intArray->iiAddress.AddressIn.sin_addr.WS_s_addr = addr;
4811
4812 intArray->iiNetmask.AddressIn.sin_family = WS_AF_INET;
4813 intArray->iiNetmask.AddressIn.sin_port = 0;
4814 intArray->iiNetmask.AddressIn.sin_addr.WS_s_addr = mask;
4815
4816 intArray->iiBroadcastAddress.AddressIn.sin_family = WS_AF_INET;
4817 intArray->iiBroadcastAddress.AddressIn.sin_port = 0;
4818 intArray->iiBroadcastAddress.AddressIn.sin_addr.WS_s_addr = bcast;
4819 intArray++;
4820 numInt++;
4821 }
4822 }
4823 else
4824 {
4825 ERR("Unable to get interface table!\n");
4826 status = WSAEINVAL;
4827 }
4828 HeapFree(GetProcessHeap(),0,table);
4829 }
4830 else status = WSAEINVAL;
4831 }
4832 else if (apiReturn != ERROR_NO_DATA)
4833 {
4834 ERR("Unable to get interface table!\n");
4835 status = WSAEINVAL;
4836 }
4837 /* Calculate the size of the array being returned */
4838 total = sizeof(INTERFACE_INFO) * numInt;
4839 release_sock_fd( s, fd );
4840 break;
4841 }
4842
4843 case WS_SIO_ADDRESS_LIST_QUERY:
4844 {
4845 DWORD size;
4846
4847 TRACE("-> SIO_ADDRESS_LIST_QUERY request\n");
4848
4849 if (!ret_size)
4850 {
4851 SetLastError(WSAEFAULT);
4852 return SOCKET_ERROR;
4853 }
4854
4855 if (out_size && out_size < FIELD_OFFSET(SOCKET_ADDRESS_LIST, Address[0]))
4856 {
4857 *ret_size = 0;
4858 SetLastError(WSAEINVAL);
4859 return SOCKET_ERROR;
4860 }
4861
4862 if (GetAdaptersInfo(NULL, &size) == ERROR_BUFFER_OVERFLOW)
4863 {
4864 IP_ADAPTER_INFO *p, *table = HeapAlloc(GetProcessHeap(), 0, size);
4865 SOCKET_ADDRESS_LIST *sa_list;
4866 SOCKADDR_IN *sockaddr;
4867 SOCKET_ADDRESS *sa;
4868 unsigned int i;
4869 DWORD num;
4870
4871 if (!table || GetAdaptersInfo(table, &size))
4872 {
4873 HeapFree(GetProcessHeap(), 0, table);
4874 status = WSAEINVAL;
4875 break;
4876 }
4877
4878 for (p = table, num = 0; p; p = p->Next)
4879 if (p->IpAddressList.IpAddress.String[0]) num++;
4880
4881 total = FIELD_OFFSET(SOCKET_ADDRESS_LIST, Address[num]) + num * sizeof(*sockaddr);
4882 if (total > out_size || !out_buff)
4883 {
4884 *ret_size = total;
4885 HeapFree(GetProcessHeap(), 0, table);
4886 status = WSAEFAULT;
4887 break;
4888 }
4889
4890 sa_list = out_buff;
4891 sa = sa_list->Address;
4892 sockaddr = (SOCKADDR_IN *)&sa[num];
4893 sa_list->iAddressCount = num;
4894
4895 for (p = table, i = 0; p; p = p->Next)
4896 {
4897 if (!p->IpAddressList.IpAddress.String[0]) continue;
4898
4899 sa[i].lpSockaddr = (SOCKADDR *)&sockaddr[i];
4900 sa[i].iSockaddrLength = sizeof(SOCKADDR);
4901
4902 sockaddr[i].sin_family = WS_AF_INET;
4903 sockaddr[i].sin_port = 0;
4904 sockaddr[i].sin_addr.WS_s_addr = inet_addr(p->IpAddressList.IpAddress.String);
4905 i++;
4906 }
4907
4908 HeapFree(GetProcessHeap(), 0, table);
4909 }
4910 else
4911 {
4912 WARN("unable to get IP address list\n");
4913 status = WSAEINVAL;
4914 }
4915 break;
4916 }
4917
4918 case WS_SIO_FLUSH:
4919 FIXME("SIO_FLUSH: stub.\n");
4920 break;
4921
4922 case WS_SIO_GET_EXTENSION_FUNCTION_POINTER:
4923 {
4924 #define EXTENSION_FUNCTION(x, y) { x, y, #y },
4925 static const struct
4926 {
4927 GUID guid;
4928 void *func_ptr;
4929 const char *name;
4930 } guid_funcs[] = {
4931 EXTENSION_FUNCTION(WSAID_CONNECTEX, WS2_ConnectEx)
4932 EXTENSION_FUNCTION(WSAID_DISCONNECTEX, WS2_DisconnectEx)
4933 EXTENSION_FUNCTION(WSAID_ACCEPTEX, WS2_AcceptEx)
4934 EXTENSION_FUNCTION(WSAID_GETACCEPTEXSOCKADDRS, WS2_GetAcceptExSockaddrs)
4935 EXTENSION_FUNCTION(WSAID_TRANSMITFILE, WS2_TransmitFile)
4936 /* EXTENSION_FUNCTION(WSAID_TRANSMITPACKETS, WS2_TransmitPackets) */
4937 EXTENSION_FUNCTION(WSAID_WSARECVMSG, WS2_WSARecvMsg)
4938 EXTENSION_FUNCTION(WSAID_WSASENDMSG, WSASendMsg)
4939 };
4940 #undef EXTENSION_FUNCTION
4941 BOOL found = FALSE;
4942 unsigned int i;
4943
4944 for (i = 0; i < ARRAY_SIZE(guid_funcs); i++)
4945 {
4946 if (IsEqualGUID(&guid_funcs[i].guid, in_buff))
4947 {
4948 found = TRUE;
4949 break;
4950 }
4951 }
4952
4953 if (found)
4954 {
4955 TRACE("-> got %s\n", guid_funcs[i].name);
4956 *(void **)out_buff = guid_funcs[i].func_ptr;
4957 total = sizeof(void *);
4958 break;
4959 }
4960
4961 FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER %s: stub\n", debugstr_guid(in_buff));
4962 status = WSAEOPNOTSUPP;
4963 break;
4964 }
4965 case WS_SIO_KEEPALIVE_VALS:
4966 {
4967 struct tcp_keepalive *k;
4968 int keepalive, keepidle, keepintvl;
4969
4970 if (!in_buff || in_size < sizeof(struct tcp_keepalive))
4971 {
4972 SetLastError(WSAEFAULT);
4973 return SOCKET_ERROR;
4974 }
4975
4976 k = in_buff;
4977 keepalive = k->onoff ? 1 : 0;
4978 keepidle = max( 1, (k->keepalivetime + 500) / 1000 );
4979 keepintvl = max( 1, (k->keepaliveinterval + 500) / 1000 );
4980
4981 TRACE("onoff: %d, keepalivetime: %d, keepaliveinterval: %d\n", keepalive, keepidle, keepintvl);
4982
4983 fd = get_sock_fd(s, 0, NULL);
4984 if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, (void *)&keepalive, sizeof(int)) == -1)
4985 status = WSAEINVAL;
4986 #if defined(TCP_KEEPIDLE) || defined(TCP_KEEPINTVL)
4987 /* these values need to be set only if SO_KEEPALIVE is enabled */
4988 else if(keepalive)
4989 {
4990 #ifndef TCP_KEEPIDLE
4991 FIXME("ignoring keepalive timeout\n");
4992 #else
4993 if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPIDLE, (void *)&keepidle, sizeof(int)) == -1)
4994 status = WSAEINVAL;
4995 else
4996 #endif
4997 #ifdef TCP_KEEPINTVL
4998 if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPINTVL, (void *)&keepintvl, sizeof(int)) == -1)
4999 status = WSAEINVAL;
5000 #else
5001 FIXME("ignoring keepalive interval\n");
5002 #endif
5003 }
5004 #else
5005 else
5006 FIXME("ignoring keepalive interval and timeout\n");
5007 #endif
5008 release_sock_fd(s, fd);
5009 break;
5010 }
5011 case WS_SIO_ROUTING_INTERFACE_QUERY:
5012 {
5013 struct WS_sockaddr *daddr = (struct WS_sockaddr *)in_buff;
5014 struct WS_sockaddr_in *daddr_in = (struct WS_sockaddr_in *)daddr;
5015 struct WS_sockaddr_in *saddr_in = out_buff;
5016 MIB_IPFORWARDROW row;
5017 PMIB_IPADDRTABLE ipAddrTable = NULL;
5018 DWORD size, i, found_index;
5019
5020 TRACE("-> WS_SIO_ROUTING_INTERFACE_QUERY request\n");
5021
5022 if (!in_buff || in_size < sizeof(struct WS_sockaddr) ||
5023 !out_buff || out_size < sizeof(struct WS_sockaddr_in) || !ret_size)
5024 {
5025 SetLastError(WSAEFAULT);
5026 return SOCKET_ERROR;
5027 }
5028 if (daddr->sa_family != WS_AF_INET)
5029 {
5030 FIXME("unsupported address family %d\n", daddr->sa_family);
5031 status = WSAEAFNOSUPPORT;
5032 break;
5033 }
5034 if (GetBestRoute(daddr_in->sin_addr.S_un.S_addr, 0, &row) != NOERROR ||
5035 GetIpAddrTable(NULL, &size, FALSE) != ERROR_INSUFFICIENT_BUFFER)
5036 {
5037 status = WSAEFAULT;
5038 break;
5039 }
5040 ipAddrTable = HeapAlloc(GetProcessHeap(), 0, size);
5041 if (GetIpAddrTable(ipAddrTable, &size, FALSE))
5042 {
5043 HeapFree(GetProcessHeap(), 0, ipAddrTable);
5044 status = WSAEFAULT;
5045 break;
5046 }
5047 for (i = 0, found_index = ipAddrTable->dwNumEntries;
5048 i < ipAddrTable->dwNumEntries; i++)
5049 {
5050 if (ipAddrTable->table[i].dwIndex == row.dwForwardIfIndex)
5051 found_index = i;
5052 }
5053 if (found_index == ipAddrTable->dwNumEntries)
5054 {
5055 ERR("no matching IP address for interface %d\n",
5056 row.dwForwardIfIndex);
5057 HeapFree(GetProcessHeap(), 0, ipAddrTable);
5058 status = WSAEFAULT;
5059 break;
5060 }
5061 saddr_in->sin_family = WS_AF_INET;
5062 saddr_in->sin_addr.S_un.S_addr = ipAddrTable->table[found_index].dwAddr;
5063 saddr_in->sin_port = 0;
5064 total = sizeof(struct WS_sockaddr_in);
5065 HeapFree(GetProcessHeap(), 0, ipAddrTable);
5066 break;
5067 }
5068 case WS_SIO_SET_COMPATIBILITY_MODE:
5069 TRACE("WS_SIO_SET_COMPATIBILITY_MODE ignored\n");
5070 status = WSAEOPNOTSUPP;
5071 break;
5072 case WS_SIO_UDP_CONNRESET:
5073 FIXME("WS_SIO_UDP_CONNRESET stub\n");
5074 break;
5075 case 0x667e: /* Netscape tries hard to use bogus ioctl 0x667e */
5076 SetLastError(WSAEOPNOTSUPP);
5077 return SOCKET_ERROR;
5078 default:
5079 status = WSAEOPNOTSUPP;
5080 break;
5081 }
5082
5083 if (status == WSAEOPNOTSUPP)
5084 {
5085 status = server_ioctl_sock(s, code, in_buff, in_size, out_buff, out_size, &total,
5086 overlapped, completion);
5087 if (status != WSAEOPNOTSUPP)
5088 {
5089 if (status == 0 || status == WSA_IO_PENDING)
5090 TRACE("-> %s request\n", debugstr_wsaioctl(code));
5091 else
5092 ERR("-> %s request failed with status 0x%x\n", debugstr_wsaioctl(code), status);
5093
5094 /* overlapped and completion operations will be handled by the server */
5095 completion = NULL;
5096 overlapped = NULL;
5097 }
5098 else
5099 FIXME("unsupported WS_IOCTL cmd (%s)\n", debugstr_wsaioctl(code));
5100 }
5101
5102 if (completion)
5103 {
5104 FIXME( "completion routine %p not supported\n", completion );
5105 }
5106 else if (overlapped)
5107 {
5108 ULONG_PTR cvalue = (overlapped && ((ULONG_PTR)overlapped->hEvent & 1) == 0) ? (ULONG_PTR)overlapped : 0;
5109 overlapped->Internal = sock_error_to_ntstatus( status );
5110 overlapped->InternalHigh = total;
5111 if (cvalue) WS_AddCompletion( HANDLE2SOCKET(s), cvalue, overlapped->Internal, total, FALSE );
5112 if (overlapped->hEvent) NtSetEvent( overlapped->hEvent, NULL );
5113 }
5114
5115 if (!status)
5116 {
5117 if (ret_size) *ret_size = total;
5118 return 0;
5119 }
5120 SetLastError( status );
5121 return SOCKET_ERROR;
5122 }
5123
5124
5125 /***********************************************************************
5126 * ioctlsocket (WS2_32.10)
5127 */
5128 int WINAPI WS_ioctlsocket(SOCKET s, LONG cmd, WS_u_long *argp)
5129 {
5130 DWORD ret_size;
5131 return WSAIoctl( s, cmd, argp, sizeof(WS_u_long), argp, sizeof(WS_u_long), &ret_size, NULL, NULL );
5132 }
5133
5134 /***********************************************************************
5135 * listen (WS2_32.13)
5136 */
5137 int WINAPI WS_listen(SOCKET s, int backlog)
5138 {
5139 int fd = get_sock_fd( s, FILE_READ_DATA, NULL ), ret = SOCKET_ERROR;
5140
5141 TRACE("socket %04lx, backlog %d\n", s, backlog);
5142 if (fd != -1)
5143 {
5144 int bound = is_fd_bound(fd, NULL, NULL);
5145
5146 if (bound <= 0)
5147 {
5148 SetLastError(bound == -1 ? wsaErrno() : WSAEINVAL);
5149 }
5150 else if (listen(fd, backlog) == 0)
5151 {
5152 _enable_event(SOCKET2HANDLE(s), FD_ACCEPT,
5153 FD_WINE_LISTENING,
5154 FD_CONNECT|FD_WINE_CONNECTED);
5155 ret = 0;
5156 }
5157 else
5158 SetLastError(wsaErrno());
5159 release_sock_fd( s, fd );
5160 }
5161 else
5162 SetLastError(WSAENOTSOCK);
5163 return ret;
5164 }
5165
5166 /***********************************************************************
5167 * recv (WS2_32.16)
5168 */
5169 int WINAPI WS_recv(SOCKET s, char *buf, int len, int flags)
5170 {
5171 DWORD n, dwFlags = flags;
5172 WSABUF wsabuf;
5173
5174 wsabuf.len = len;
5175 wsabuf.buf = buf;
5176
5177 if ( WS2_recv_base(s, &wsabuf, 1, &n, &dwFlags, NULL, NULL, NULL, NULL, NULL) == SOCKET_ERROR )
5178 return SOCKET_ERROR;
5179 else
5180 return n;
5181 }
5182
5183 /***********************************************************************
5184 * recvfrom (WS2_32.17)
5185 */
5186 int WINAPI WS_recvfrom(SOCKET s, char *buf, INT len, int flags,
5187 struct WS_sockaddr *from, int *fromlen)
5188 {
5189 DWORD n, dwFlags = flags;
5190 WSABUF wsabuf;
5191
5192 wsabuf.len = len;
5193 wsabuf.buf = buf;
5194
5195 if ( WS2_recv_base(s, &wsabuf, 1, &n, &dwFlags, from, fromlen, NULL, NULL, NULL) == SOCKET_ERROR )
5196 return SOCKET_ERROR;
5197 else
5198 return n;
5199 }
5200
5201 /* allocate a poll array for the corresponding fd sets */
5202 static struct pollfd *fd_sets_to_poll( const WS_fd_set *readfds, const WS_fd_set *writefds,
5203 const WS_fd_set *exceptfds, int *count_ptr )
5204 {
5205 unsigned int i, j = 0, count = 0;
5206 struct pollfd *fds;
5207 struct per_thread_data *ptb = get_per_thread_data();
5208
5209 if (readfds) count += readfds->fd_count;
5210 if (writefds) count += writefds->fd_count;
5211 if (exceptfds) count += exceptfds->fd_count;
5212 *count_ptr = count;
5213 if (!count)
5214 {
5215 SetLastError(WSAEINVAL);
5216 return NULL;
5217 }
5218
5219 /* check if the cache can hold all descriptors, if not do the resizing */
5220 if (ptb->fd_count < count)
5221 {
5222 if (!(fds = HeapAlloc(GetProcessHeap(), 0, count * sizeof(fds[0]))))
5223 {
5224 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
5225 return NULL;
5226 }
5227 HeapFree(GetProcessHeap(), 0, ptb->fd_cache);
5228 ptb->fd_cache = fds;
5229 ptb->fd_count = count;
5230 }
5231 else
5232 fds = ptb->fd_cache;
5233
5234 if (readfds)
5235 for (i = 0; i < readfds->fd_count; i++, j++)
5236 {
5237 fds[j].fd = get_sock_fd( readfds->fd_array[i], FILE_READ_DATA, NULL );
5238 if (fds[j].fd == -1) goto failed;
5239 fds[j].revents = 0;
5240 if (is_fd_bound(fds[j].fd, NULL, NULL) == 1)
5241 {
5242 fds[j].events = POLLIN;
5243 }
5244 else
5245 {
5246 release_sock_fd( readfds->fd_array[i], fds[j].fd );
5247 fds[j].fd = -1;
5248 fds[j].events = 0;
5249 }
5250 }
5251 if (writefds)
5252 for (i = 0; i < writefds->fd_count; i++, j++)
5253 {
5254 fds[j].fd = get_sock_fd( writefds->fd_array[i], FILE_WRITE_DATA, NULL );
5255 if (fds[j].fd == -1) goto failed;
5256 fds[j].revents = 0;
5257 if (is_fd_bound(fds[j].fd, NULL, NULL) == 1 ||
5258 _get_fd_type(fds[j].fd) == SOCK_DGRAM)
5259 {
5260 fds[j].events = POLLOUT;
5261 }
5262 else
5263 {
5264 release_sock_fd( writefds->fd_array[i], fds[j].fd );
5265 fds[j].fd = -1;
5266 fds[j].events = 0;
5267 }
5268 }
5269 if (exceptfds)
5270 for (i = 0; i < exceptfds->fd_count; i++, j++)
5271 {
5272 fds[j].fd = get_sock_fd( exceptfds->fd_array[i], 0, NULL );
5273 if (fds[j].fd == -1) goto failed;
5274 fds[j].revents = 0;
5275 if (is_fd_bound(fds[j].fd, NULL, NULL) == 1)
5276 {
5277 int oob_inlined = 0;
5278 socklen_t olen = sizeof(oob_inlined);
5279
5280 fds[j].events = POLLHUP;
5281
5282 /* Check if we need to test for urgent data or not */
5283 getsockopt(fds[j].fd, SOL_SOCKET, SO_OOBINLINE, (char*) &oob_inlined, &olen);
5284 if (!oob_inlined)
5285 fds[j].events |= POLLPRI;
5286 }
5287 else
5288 {
5289 release_sock_fd( exceptfds->fd_array[i], fds[j].fd );
5290 fds[j].fd = -1;
5291 fds[j].events = 0;
5292 }
5293 }
5294 return fds;
5295
5296 failed:
5297 count = j;
5298 j = 0;
5299 if (readfds)
5300 for (i = 0; i < readfds->fd_count && j < count; i++, j++)
5301 if (fds[j].fd != -1) release_sock_fd( readfds->fd_array[i], fds[j].fd );
5302 if (writefds)
5303 for (i = 0; i < writefds->fd_count && j < count; i++, j++)
5304 if (fds[j].fd != -1) release_sock_fd( writefds->fd_array[i], fds[j].fd );
5305 if (exceptfds)
5306 for (i = 0; i < exceptfds->fd_count && j < count; i++, j++)
5307 if (fds[j].fd != -1) release_sock_fd( exceptfds->fd_array[i], fds[j].fd );
5308 return NULL;
5309 }
5310
5311 /* release the file descriptor obtained in fd_sets_to_poll */
5312 /* must be called with the original fd_set arrays, before calling get_poll_results */
5313 static void release_poll_fds( const WS_fd_set *readfds, const WS_fd_set *writefds,
5314 const WS_fd_set *exceptfds, struct pollfd *fds )
5315 {
5316 unsigned int i, j = 0;
5317
5318 if (readfds)
5319 {
5320 for (i = 0; i < readfds->fd_count; i++, j++)
5321 if (fds[j].fd != -1) release_sock_fd( readfds->fd_array[i], fds[j].fd );
5322 }
5323 if (writefds)
5324 {
5325 for (i = 0; i < writefds->fd_count; i++, j++)
5326 if (fds[j].fd != -1) release_sock_fd( writefds->fd_array[i], fds[j].fd );
5327 }
5328 if (exceptfds)
5329 {
5330 for (i = 0; i < exceptfds->fd_count; i++, j++)
5331 {
5332 if (fds[j].fd == -1) continue;
5333 release_sock_fd( exceptfds->fd_array[i], fds[j].fd );
5334 if (fds[j].revents & POLLHUP)
5335 {
5336 int fd = get_sock_fd( exceptfds->fd_array[i], 0, NULL );
5337 if (fd != -1)
5338 release_sock_fd( exceptfds->fd_array[i], fd );
5339 else
5340 fds[j].revents = 0;
5341 }
5342 }
5343 }
5344 }
5345
5346 static int do_poll(struct pollfd *pollfds, int count, int timeout)
5347 {
5348 struct timeval tv1, tv2;
5349 int ret, torig = timeout;
5350
5351 if (timeout > 0) gettimeofday( &tv1, 0 );
5352
5353 while ((ret = poll( pollfds, count, timeout )) < 0)
5354 {
5355 if (errno != EINTR) break;
5356 if (timeout < 0) continue;
5357 if (timeout == 0) return 0;
5358
5359 gettimeofday( &tv2, 0 );
5360
5361 tv2.tv_sec -= tv1.tv_sec;
5362 tv2.tv_usec -= tv1.tv_usec;
5363 if (tv2.tv_usec < 0)
5364 {
5365 tv2.tv_usec += 1000000;
5366 tv2.tv_sec -= 1;
5367 }
5368
5369 timeout = torig - (tv2.tv_sec * 1000) - (tv2.tv_usec + 999) / 1000;
5370 if (timeout <= 0) return 0;
5371 }
5372 return ret;
5373 }
5374
5375 /* map the poll results back into the Windows fd sets */
5376 static int get_poll_results( WS_fd_set *readfds, WS_fd_set *writefds, WS_fd_set *exceptfds,
5377 const struct pollfd *fds )
5378 {
5379 const struct pollfd *poll_writefds = fds + (readfds ? readfds->fd_count : 0);
5380 const struct pollfd *poll_exceptfds = poll_writefds + (writefds ? writefds->fd_count : 0);
5381 unsigned int i, k, total = 0;
5382
5383 if (readfds)
5384 {
5385 for (i = k = 0; i < readfds->fd_count; i++)
5386 {
5387 if (fds[i].revents ||
5388 (readfds == writefds && (poll_writefds[i].revents & POLLOUT) && !(poll_writefds[i].revents & POLLHUP)) ||
5389 (readfds == exceptfds && poll_exceptfds[i].revents))
5390 readfds->fd_array[k++] = readfds->fd_array[i];
5391 }
5392 readfds->fd_count = k;
5393 total += k;
5394 }
5395 if (writefds && writefds != readfds)
5396 {
5397 for (i = k = 0; i < writefds->fd_count; i++)
5398 {
5399 if (((poll_writefds[i].revents & POLLOUT) && !(poll_writefds[i].revents & POLLHUP)) ||
5400 (writefds == exceptfds && poll_exceptfds[i].revents))
5401 writefds->fd_array[k++] = writefds->fd_array[i];
5402 }
5403 writefds->fd_count = k;
5404 total += k;
5405 }
5406 if (exceptfds && exceptfds != readfds && exceptfds != writefds)
5407 {
5408 for (i = k = 0; i < exceptfds->fd_count; i++)
5409 if (poll_exceptfds[i].revents) exceptfds->fd_array[k++] = exceptfds->fd_array[i];
5410 exceptfds->fd_count = k;
5411 total += k;
5412 }
5413 return total;
5414 }
5415
5416 /***********************************************************************
5417 * select (WS2_32.18)
5418 */
5419 int WINAPI WS_select(int nfds, WS_fd_set *ws_readfds,
5420 WS_fd_set *ws_writefds, WS_fd_set *ws_exceptfds,
5421 const struct WS_timeval* ws_timeout)
5422 {
5423 struct pollfd *pollfds;
5424 int count, ret, timeout = -1;
5425
5426 TRACE("read %p, write %p, excp %p timeout %p\n",
5427 ws_readfds, ws_writefds, ws_exceptfds, ws_timeout);
5428
5429 if (!(pollfds = fd_sets_to_poll( ws_readfds, ws_writefds, ws_exceptfds, &count )))
5430 return SOCKET_ERROR;
5431
5432 if (ws_timeout)
5433 timeout = (ws_timeout->tv_sec * 1000) + (ws_timeout->tv_usec + 999) / 1000;
5434
5435 ret = do_poll(pollfds, count, timeout);
5436 release_poll_fds( ws_readfds, ws_writefds, ws_exceptfds, pollfds );
5437
5438 if (ret == -1) SetLastError(wsaErrno());
5439 else ret = get_poll_results( ws_readfds, ws_writefds, ws_exceptfds, pollfds );
5440 return ret;
5441 }
5442
5443 /***********************************************************************
5444 * WSAPoll
5445 */
5446 int WINAPI WSAPoll(WSAPOLLFD *wfds, ULONG count, int timeout)
5447 {
5448 int i, ret;
5449 struct pollfd *ufds;
5450
5451 if (!count)
5452 {
5453 SetLastError(WSAEINVAL);
5454 return SOCKET_ERROR;
5455 }
5456 if (!wfds)
5457 {
5458 SetLastError(WSAEFAULT);
5459 return SOCKET_ERROR;
5460 }
5461
5462 if (!(ufds = HeapAlloc(GetProcessHeap(), 0, count * sizeof(ufds[0]))))
5463 {
5464 SetLastError(WSAENOBUFS);
5465 return SOCKET_ERROR;
5466 }
5467
5468 for (i = 0; i < count; i++)
5469 {
5470 ufds[i].fd = get_sock_fd(wfds[i].fd, 0, NULL);
5471 ufds[i].events = convert_poll_w2u(wfds[i].events);
5472 ufds[i].revents = 0;
5473 }
5474
5475 ret = do_poll(ufds, count, timeout);
5476
5477 for (i = 0; i < count; i++)
5478 {
5479 if (ufds[i].fd != -1)
5480 {
5481 release_sock_fd(wfds[i].fd, ufds[i].fd);
5482 if (ufds[i].revents & POLLHUP)
5483 {
5484 /* Check if the socket still exists */
5485 int fd = get_sock_fd(wfds[i].fd, 0, NULL);
5486 if (fd != -1)
5487 {
5488 wfds[i].revents = WS_POLLHUP;
5489 release_sock_fd(wfds[i].fd, fd);
5490 }
5491 else
5492 wfds[i].revents = WS_POLLNVAL;
5493 }
5494 else
5495 wfds[i].revents = convert_poll_u2w(ufds[i].revents);
5496 }
5497 else
5498 wfds[i].revents = WS_POLLNVAL;
5499 }
5500
5501 HeapFree(GetProcessHeap(), 0, ufds);
5502 return ret;
5503 }
5504
5505 /* helper to send completion messages for client-only i/o operation case */
5506 static void WS_AddCompletion( SOCKET sock, ULONG_PTR CompletionValue, NTSTATUS CompletionStatus,
5507 ULONG Information, BOOL async )
5508 {
5509 SERVER_START_REQ( add_fd_completion )
5510 {
5511 req->handle = wine_server_obj_handle( SOCKET2HANDLE(sock) );
5512 req->cvalue = CompletionValue;
5513 req->status = CompletionStatus;
5514 req->information = Information;
5515 req->async = async;
5516 wine_server_call( req );
5517 }
5518 SERVER_END_REQ;
5519 }
5520
5521
5522 /***********************************************************************
5523 * send (WS2_32.19)
5524 */
5525 int WINAPI WS_send(SOCKET s, const char *buf, int len, int flags)
5526 {
5527 DWORD n;
5528 WSABUF wsabuf;
5529
5530 wsabuf.len = len;
5531 wsabuf.buf = (char*) buf;
5532
5533 if ( WS2_sendto( s, &wsabuf, 1, &n, flags, NULL, 0, NULL, NULL) == SOCKET_ERROR )
5534 return SOCKET_ERROR;
5535 else
5536 return n;
5537 }
5538
5539 /***********************************************************************
5540 * WSASend (WS2_32.72)
5541 */
5542 INT WINAPI WSASend( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
5543 LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
5544 LPWSAOVERLAPPED lpOverlapped,
5545 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
5546 {
5547 return WS2_sendto( s, lpBuffers, dwBufferCount, lpNumberOfBytesSent, dwFlags,
5548 NULL, 0, lpOverlapped, lpCompletionRoutine );
5549 }
5550
5551 /***********************************************************************
5552 * WSASendDisconnect (WS2_32.73)
5553 */
5554 INT WINAPI WSASendDisconnect( SOCKET s, LPWSABUF lpBuffers )
5555 {
5556 return WS_shutdown( s, SD_SEND );
5557 }
5558
5559
5560 static int WS2_sendto( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
5561 LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
5562 const struct WS_sockaddr *to, int tolen,
5563 LPWSAOVERLAPPED lpOverlapped,
5564 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
5565 {
5566 unsigned int i, options;
5567 int n, fd, err, overlapped, flags;
5568 struct ws2_async *wsa = NULL, localwsa;
5569 int totalLength = 0;
5570 DWORD bytes_sent;
5571 BOOL is_blocking;
5572
5573 TRACE("socket %04lx, wsabuf %p, nbufs %d, flags %d, to %p, tolen %d, ovl %p, func %p\n",
5574 s, lpBuffers, dwBufferCount, dwFlags,
5575 to, tolen, lpOverlapped, lpCompletionRoutine);
5576
5577 fd = get_sock_fd( s, FILE_WRITE_DATA, &options );
5578 TRACE( "fd=%d, options=%x\n", fd, options );
5579
5580 if ( fd == -1 ) return SOCKET_ERROR;
5581
5582 if (!lpOverlapped && !lpNumberOfBytesSent)
5583 {
5584 err = WSAEFAULT;
5585 goto error;
5586 }
5587
5588 overlapped = (lpOverlapped || lpCompletionRoutine) &&
5589 !(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT));
5590 if (overlapped || dwBufferCount > 1)
5591 {
5592 if (!(wsa = (struct ws2_async *)alloc_async_io( offsetof(struct ws2_async, iovec[dwBufferCount]),
5593 WS2_async_send )))
5594 {
5595 err = WSAEFAULT;
5596 goto error;
5597 }
5598 }
5599 else
5600 wsa = &localwsa;
5601
5602 wsa->hSocket = SOCKET2HANDLE(s);
5603 wsa->addr = (struct WS_sockaddr *)to;
5604 wsa->addrlen.val = tolen;
5605 wsa->flags = dwFlags;
5606 wsa->lpFlags = &wsa->flags;
5607 wsa->control = NULL;
5608 wsa->n_iovecs = dwBufferCount;
5609 wsa->first_iovec = 0;
5610 for ( i = 0; i < dwBufferCount; i++ )
5611 {
5612 wsa->iovec[i].iov_base = lpBuffers[i].buf;
5613 wsa->iovec[i].iov_len = lpBuffers[i].len;
5614 totalLength += lpBuffers[i].len;
5615 }
5616
5617 flags = convert_flags(dwFlags);
5618 n = WS2_send( fd, wsa, flags );
5619 if (n == -1 && errno != EAGAIN)
5620 {
5621 err = wsaErrno();
5622 goto error;
5623 }
5624
5625 if (overlapped)
5626 {
5627 IO_STATUS_BLOCK *iosb = lpOverlapped ? (IO_STATUS_BLOCK *)lpOverlapped : &wsa->local_iosb;
5628 ULONG_PTR cvalue = (lpOverlapped && ((ULONG_PTR)lpOverlapped->hEvent & 1) == 0) ? (ULONG_PTR)lpOverlapped : 0;
5629
5630 wsa->user_overlapped = lpOverlapped;
5631 wsa->completion_func = lpCompletionRoutine;
5632 release_sock_fd( s, fd );
5633
5634 if (n == -1 || n < totalLength)
5635 {
5636 iosb->u.Status = STATUS_PENDING;
5637 iosb->Information = n == -1 ? 0 : n;
5638
5639 if (wsa->completion_func)
5640 err = register_async( ASYNC_TYPE_WRITE, wsa->hSocket, &wsa->io, NULL,
5641 ws2_async_apc, wsa, iosb );
5642 else
5643 err = register_async( ASYNC_TYPE_WRITE, wsa->hSocket, &wsa->io, lpOverlapped->hEvent,
5644 NULL, (void *)cvalue, iosb );
5645
5646 /* Enable the event only after starting the async. The server will deliver it as soon as
5647 the async is done. */
5648 _enable_event(SOCKET2HANDLE(s), FD_WRITE, 0, 0);
5649
5650 if (err != STATUS_PENDING) HeapFree( GetProcessHeap(), 0, wsa );
5651 SetLastError(NtStatusToWSAError( err ));
5652 return SOCKET_ERROR;
5653 }
5654
5655 iosb->u.Status = STATUS_SUCCESS;
5656 iosb->Information = n;
5657 if (lpNumberOfBytesSent) *lpNumberOfBytesSent = n;
5658 if (!wsa->completion_func)
5659 {
5660 if (cvalue) WS_AddCompletion( s, cvalue, STATUS_SUCCESS, n, FALSE );
5661 if (lpOverlapped->hEvent) SetEvent( lpOverlapped->hEvent );
5662 HeapFree( GetProcessHeap(), 0, wsa );
5663 }
5664 else NtQueueApcThread( GetCurrentThread(), (PNTAPCFUNC)ws2_async_apc,
5665 (ULONG_PTR)wsa, (ULONG_PTR)iosb, 0 );
5666 SetLastError(ERROR_SUCCESS);
5667 return 0;
5668 }
5669
5670 if ((err = _is_blocking( s, &is_blocking )))
5671 {
5672 err = NtStatusToWSAError( err );
5673 goto error;
5674 }
5675
5676 if ( is_blocking )
5677 {
5678 /* On a blocking non-overlapped stream socket,
5679 * sending blocks until the entire buffer is sent. */
5680 DWORD timeout_start = GetTickCount();
5681
5682 bytes_sent = n == -1 ? 0 : n;
5683
5684 while (wsa->first_iovec < wsa->n_iovecs)
5685 {
5686 struct pollfd pfd;
5687 int poll_timeout = -1;
5688 INT64 timeout = get_rcvsnd_timeo(fd, FALSE);
5689
5690 if (timeout)
5691 {
5692 timeout -= GetTickCount() - timeout_start;
5693 if (timeout < 0) poll_timeout = 0;
5694 else poll_timeout = timeout <= INT_MAX ? timeout : INT_MAX;
5695 }
5696
5697 pfd.fd = fd;
5698 pfd.events = POLLOUT;
5699
5700 if (!poll_timeout || !poll( &pfd, 1, poll_timeout ))
5701 {
5702 err = WSAETIMEDOUT;
5703 goto error; /* msdn says a timeout in send is fatal */
5704 }
5705
5706 n = WS2_send( fd, wsa, flags );
5707 if (n == -1 && errno != EAGAIN)
5708 {
5709 err = wsaErrno();
5710 goto error;
5711 }
5712
5713 if (n >= 0)
5714 bytes_sent += n;
5715 }
5716 }
5717 else /* non-blocking */
5718 {
5719 if (n < totalLength)
5720 _enable_event(SOCKET2HANDLE(s), FD_WRITE, 0, 0);
5721 if (n == -1)
5722 {
5723 err = WSAEWOULDBLOCK;
5724 goto error;
5725 }
5726 bytes_sent = n;
5727 }
5728
5729 TRACE(" -> %i bytes\n", bytes_sent);
5730
5731 if (lpNumberOfBytesSent) *lpNumberOfBytesSent = bytes_sent;
5732 if (wsa != &localwsa) HeapFree( GetProcessHeap(), 0, wsa );
5733 release_sock_fd( s, fd );
5734 SetLastError(ERROR_SUCCESS);
5735 return 0;
5736
5737 error:
5738 if (wsa != &localwsa) HeapFree( GetProcessHeap(), 0, wsa );
5739 release_sock_fd( s, fd );
5740 WARN(" -> ERROR %d\n", err);
5741 SetLastError(err);
5742 return SOCKET_ERROR;
5743 }
5744
5745 /***********************************************************************
5746 * WSASendTo (WS2_32.74)
5747 */
5748 INT WINAPI WSASendTo( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
5749 LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
5750 const struct WS_sockaddr *to, int tolen,
5751 LPWSAOVERLAPPED lpOverlapped,
5752 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
5753 {
5754 return WS2_sendto( s, lpBuffers, dwBufferCount,
5755 lpNumberOfBytesSent, dwFlags,
5756 to, tolen,
5757 lpOverlapped, lpCompletionRoutine );
5758 }
5759
5760 /***********************************************************************
5761 * sendto (WS2_32.20)
5762 */
5763 int WINAPI WS_sendto(SOCKET s, const char *buf, int len, int flags,
5764 const struct WS_sockaddr *to, int tolen)
5765 {
5766 DWORD n;
5767 WSABUF wsabuf;
5768
5769 wsabuf.len = len;
5770 wsabuf.buf = (char*) buf;
5771
5772 if ( WS2_sendto(s, &wsabuf, 1, &n, flags, to, tolen, NULL, NULL) == SOCKET_ERROR )
5773 return SOCKET_ERROR;
5774 else
5775 return n;
5776 }
5777
5778 /***********************************************************************
5779 * setsockopt (WS2_32.21)
5780 */
5781 int WINAPI WS_setsockopt(SOCKET s, int level, int optname,
5782 const char *optval, int optlen)
5783 {
5784 int fd;
5785 int woptval;
5786 struct linger linger;
5787 struct timeval tval;
5788
5789 TRACE("(socket %04lx, %s, optval %s, optlen %d)\n", s,
5790 debugstr_sockopt(level, optname), debugstr_optval(optval, optlen),
5791 optlen);
5792
5793 /* some broken apps pass the value directly instead of a pointer to it */
5794 if(optlen && IS_INTRESOURCE(optval))
5795 {
5796 SetLastError(WSAEFAULT);
5797 return SOCKET_ERROR;
5798 }
5799
5800 switch(level)
5801 {
5802 case WS_SOL_SOCKET:
5803 switch(optname)
5804 {
5805 /* Some options need some conversion before they can be sent to
5806 * setsockopt. The conversions are done here, then they will fall through
5807 * to the general case. Special options that are not passed to
5808 * setsockopt follow below that.*/
5809
5810 case WS_SO_DONTLINGER:
5811 if (!optval)
5812 {
5813 SetLastError(WSAEFAULT);
5814 return SOCKET_ERROR;
5815 }
5816 linger.l_onoff = *(const int*)optval == 0;
5817 linger.l_linger = 0;
5818 level = SOL_SOCKET;
5819 optname = SO_LINGER;
5820 optval = (char*)&linger;
5821 optlen = sizeof(struct linger);
5822 break;
5823
5824 case WS_SO_LINGER:
5825 if (!optval)
5826 {
5827 SetLastError(WSAEFAULT);
5828 return SOCKET_ERROR;
5829 }
5830 linger.l_onoff = ((LINGER*)optval)->l_onoff;
5831 linger.l_linger = ((LINGER*)optval)->l_linger;
5832 level = SOL_SOCKET;
5833 optname = SO_LINGER;
5834 optval = (char*)&linger;
5835 optlen = sizeof(struct linger);
5836 break;
5837
5838 case WS_SO_SNDBUF:
5839 if (!*(const int *)optval)
5840 {
5841 FIXME("SO_SNDBUF ignoring request to disable send buffering\n");
5842 #ifdef __APPLE__
5843 return 0;
5844 #endif
5845 }
5846 convert_sockopt(&level, &optname);
5847 break;
5848
5849 case WS_SO_RCVBUF:
5850 if (*(const int*)optval < 2048)
5851 {
5852 WARN("SO_RCVBF for %d bytes is too small: ignored\n", *(const int*)optval );
5853 return 0;
5854 }
5855 /* Fall through */
5856
5857 /* The options listed here don't need any special handling. Thanks to
5858 * the conversion happening above, options from there will fall through
5859 * to this, too.*/
5860 case WS_SO_ACCEPTCONN:
5861 case WS_SO_BROADCAST:
5862 case WS_SO_ERROR:
5863 case WS_SO_KEEPALIVE:
5864 case WS_SO_OOBINLINE:
5865 /* BSD socket SO_REUSEADDR is not 100% compatible to winsock semantics.
5866 * however, using it the BSD way fixes bug 8513 and seems to be what
5867 * most programmers assume, anyway */
5868 case WS_SO_REUSEADDR:
5869 case WS_SO_TYPE:
5870 convert_sockopt(&level, &optname);
5871 break;
5872
5873 /* SO_DEBUG is a privileged operation, ignore it. */
5874 case WS_SO_DEBUG:
5875 TRACE("Ignoring SO_DEBUG\n");
5876 return 0;
5877
5878 /* For some reason the game GrandPrixLegends does set SO_DONTROUTE on its
5879 * socket. According to MSDN, this option is silently ignored.*/
5880 case WS_SO_DONTROUTE:
5881 TRACE("Ignoring SO_DONTROUTE\n");
5882 return 0;
5883
5884 /* Stops two sockets from being bound to the same port. Always happens
5885 * on unix systems, so just drop it. */
5886 case WS_SO_EXCLUSIVEADDRUSE:
5887 TRACE("Ignoring SO_EXCLUSIVEADDRUSE, is always set.\n");
5888 return 0;
5889
5890 /* After a ConnectEx call succeeds, the socket can't be used with half of the
5891 * normal winsock functions on windows. We don't have that problem. */
5892 case WS_SO_UPDATE_CONNECT_CONTEXT:
5893 TRACE("Ignoring SO_UPDATE_CONNECT_CONTEXT, since our sockets are normal\n");
5894 return 0;
5895
5896 /* After a AcceptEx call succeeds, the socket can't be used with half of the
5897 * normal winsock functions on windows. We don't have that problem. */
5898 case WS_SO_UPDATE_ACCEPT_CONTEXT:
5899 TRACE("Ignoring SO_UPDATE_ACCEPT_CONTEXT, since our sockets are normal\n");
5900 return 0;
5901
5902 /* SO_OPENTYPE does not require a valid socket handle. */
5903 case WS_SO_OPENTYPE:
5904 if (!optlen || optlen < sizeof(int) || !optval)
5905 {
5906 SetLastError(WSAEFAULT);
5907 return SOCKET_ERROR;
5908 }
5909 get_per_thread_data()->opentype = *(const int *)optval;
5910 TRACE("setting global SO_OPENTYPE = 0x%x\n", *((const int*)optval) );
5911 return 0;
5912
5913 #ifdef SO_RCVTIMEO
5914 case WS_SO_RCVTIMEO:
5915 #endif
5916 #ifdef SO_SNDTIMEO
5917 case WS_SO_SNDTIMEO:
5918 #endif
5919 #if defined(SO_RCVTIMEO) || defined(SO_SNDTIMEO)
5920 if (optval && optlen == sizeof(UINT32)) {
5921 /* WinSock passes milliseconds instead of struct timeval */
5922 tval.tv_usec = (*(const UINT32*)optval % 1000) * 1000;
5923 tval.tv_sec = *(const UINT32*)optval / 1000;
5924 /* min of 500 milliseconds */
5925 if (tval.tv_sec == 0 && tval.tv_usec && tval.tv_usec < 500000)
5926 tval.tv_usec = 500000;
5927 optlen = sizeof(struct timeval);
5928 optval = (char*)&tval;
5929 } else if (optlen == sizeof(struct timeval)) {
5930 WARN("SO_SND/RCVTIMEO for %d bytes: assuming unixism\n", optlen);
5931 } else {
5932 WARN("SO_SND/RCVTIMEO for %d bytes is weird: ignored\n", optlen);
5933 return 0;
5934 }
5935 convert_sockopt(&level, &optname);
5936 break;
5937 #endif
5938
5939 case WS_SO_RANDOMIZE_PORT:
5940 FIXME("Ignoring WS_SO_RANDOMIZE_PORT\n");
5941 return 0;
5942
5943 case WS_SO_PORT_SCALABILITY:
5944 FIXME("Ignoring WS_SO_PORT_SCALABILITY\n");
5945 return 0;
5946
5947 case WS_SO_REUSE_UNICASTPORT:
5948 FIXME("Ignoring WS_SO_REUSE_UNICASTPORT\n");
5949 return 0;
5950
5951 case WS_SO_REUSE_MULTICASTPORT:
5952 FIXME("Ignoring WS_SO_REUSE_MULTICASTPORT\n");
5953 return 0;
5954
5955 default:
5956 TRACE("Unknown SOL_SOCKET optname: 0x%08x\n", optname);
5957 SetLastError(WSAENOPROTOOPT);
5958 return SOCKET_ERROR;
5959 }
5960 break; /* case WS_SOL_SOCKET */
5961
5962 #ifdef HAS_IPX
5963 case WS_NSPROTO_IPX:
5964 switch(optname)
5965 {
5966 case WS_IPX_PTYPE:
5967 return set_ipx_packettype(s, *(int*)optval);
5968
5969 case WS_IPX_FILTERPTYPE:
5970 /* Sets the receive filter packet type, at the moment we don't support it */
5971 FIXME("IPX_FILTERPTYPE: %x\n", *optval);
5972 /* Returning 0 is better for now than returning a SOCKET_ERROR */
5973 return 0;
5974
5975 default:
5976 FIXME("opt_name:%x\n", optname);
5977 return SOCKET_ERROR;
5978 }
5979 break; /* case WS_NSPROTO_IPX */
5980 #endif
5981
5982 /* Levels WS_IPPROTO_TCP and WS_IPPROTO_IP convert directly */
5983 case WS_IPPROTO_TCP:
5984 switch(optname)
5985 {
5986 case WS_TCP_NODELAY:
5987 convert_sockopt(&level, &optname);
5988 break;
5989 default:
5990 FIXME("Unknown IPPROTO_TCP optname 0x%08x\n", optname);
5991 return SOCKET_ERROR;
5992 }
5993 break;
5994
5995 case WS_IPPROTO_IP:
5996 switch(optname)
5997 {
5998 case WS_IP_ADD_MEMBERSHIP:
5999 case WS_IP_DROP_MEMBERSHIP:
6000 #ifdef IP_HDRINCL
6001 case WS_IP_HDRINCL:
6002 #endif
6003 case WS_IP_MULTICAST_IF:
6004 case WS_IP_MULTICAST_LOOP:
6005 case WS_IP_MULTICAST_TTL:
6006 case WS_IP_OPTIONS:
6007 #ifdef IP_PKTINFO
6008 case WS_IP_PKTINFO:
6009 #endif
6010 case WS_IP_TOS:
6011 case WS_IP_TTL:
6012 #ifdef IP_UNICAST_IF
6013 case WS_IP_UNICAST_IF:
6014 #endif
6015 convert_sockopt(&level, &optname);
6016 break;
6017 case WS_IP_DONTFRAGMENT:
6018 return set_dont_fragment(s, IPPROTO_IP, *(BOOL *)optval) ? 0 : SOCKET_ERROR;
6019 default:
6020 FIXME("Unknown IPPROTO_IP optname 0x%08x\n", optname);
6021 return SOCKET_ERROR;
6022 }
6023 break;
6024
6025 case WS_IPPROTO_IPV6:
6026 switch(optname)
6027 {
6028 #ifdef IPV6_ADD_MEMBERSHIP
6029 case WS_IPV6_ADD_MEMBERSHIP:
6030 #endif
6031 #ifdef IPV6_DROP_MEMBERSHIP
6032 case WS_IPV6_DROP_MEMBERSHIP:
6033 #endif
6034 case WS_IPV6_MULTICAST_IF:
6035 case WS_IPV6_MULTICAST_HOPS:
6036 case WS_IPV6_MULTICAST_LOOP:
6037 case WS_IPV6_UNICAST_HOPS:
6038 #ifdef IPV6_UNICAST_IF
6039 case WS_IPV6_UNICAST_IF:
6040 #endif
6041 convert_sockopt(&level, &optname);
6042 break;
6043 case WS_IPV6_DONTFRAG:
6044 return set_dont_fragment(s, IPPROTO_IPV6, *(BOOL *)optval) ? 0 : SOCKET_ERROR;
6045 case WS_IPV6_PROTECTION_LEVEL:
6046 FIXME("IPV6_PROTECTION_LEVEL is ignored!\n");
6047 return 0;
6048 case WS_IPV6_V6ONLY:
6049 {
6050 union generic_unix_sockaddr uaddr;
6051 socklen_t uaddrlen;
6052 int bound;
6053
6054 fd = get_sock_fd( s, 0, NULL );
6055 if (fd == -1) return SOCKET_ERROR;
6056
6057 bound = is_fd_bound(fd, &uaddr, &uaddrlen);
6058 release_sock_fd( s, fd );
6059 if (bound == 0 && uaddr.addr.sa_family == AF_INET)
6060 {
6061 /* Changing IPV6_V6ONLY succeeds on AF_INET (IPv4) socket
6062 * on Windows (with IPv6 support) if the socket is unbound.
6063 * It is essentially a noop, though Windows does store the value
6064 */
6065 WARN("Silently ignoring IPPROTO_IPV6+IPV6_V6ONLY on AF_INET socket\n");
6066 return 0;
6067 }
6068 level = IPPROTO_IPV6;
6069 optname = IPV6_V6ONLY;
6070 break;
6071 }
6072 default:
6073 FIXME("Unknown IPPROTO_IPV6 optname 0x%08x\n", optname);
6074 return SOCKET_ERROR;
6075 }
6076 break;
6077
6078 default:
6079 WARN("Unknown level: 0x%08x\n", level);
6080 SetLastError(WSAEINVAL);
6081 return SOCKET_ERROR;
6082 } /* end switch(level) */
6083
6084 /* avoid endianness issues if argument is a 16-bit int */
6085 if (optval && optlen < sizeof(int))
6086 {
6087 woptval= *((const INT16 *) optval);
6088 optval= (char*) &woptval;
6089 woptval&= (1 << optlen * 8) - 1;
6090 optlen=sizeof(int);
6091 }
6092 fd = get_sock_fd( s, 0, NULL );
6093 if (fd == -1) return SOCKET_ERROR;
6094
6095 if (setsockopt(fd, level, optname, optval, optlen) == 0)
6096 {
6097 #ifdef __APPLE__
6098 if (level == SOL_SOCKET && optname == SO_REUSEADDR &&
6099 setsockopt(fd, level, SO_REUSEPORT, optval, optlen) != 0)
6100 {
6101 SetLastError(wsaErrno());
6102 release_sock_fd( s, fd );
6103 return SOCKET_ERROR;
6104 }
6105 #endif
6106 release_sock_fd( s, fd );
6107 return 0;
6108 }
6109 TRACE("Setting socket error, %d\n", wsaErrno());
6110 SetLastError(wsaErrno());
6111 release_sock_fd( s, fd );
6112
6113 return SOCKET_ERROR;
6114 }
6115
6116 /***********************************************************************
6117 * shutdown (WS2_32.22)
6118 */
6119 int WINAPI WS_shutdown(SOCKET s, int how)
6120 {
6121 int fd, err = WSAENOTSOCK;
6122 unsigned int options = 0, clear_flags = 0;
6123
6124 fd = get_sock_fd( s, 0, &options );
6125 TRACE("socket %04lx, how 0x%x, options 0x%x\n", s, how, options );
6126
6127 if (fd == -1)
6128 return SOCKET_ERROR;
6129
6130 switch( how )
6131 {
6132 case SD_RECEIVE: /* drop receives */
6133 clear_flags |= FD_READ;
6134 break;
6135 case SD_SEND: /* drop sends */
6136 clear_flags |= FD_WRITE;
6137 break;
6138 case SD_BOTH: /* drop all */
6139 clear_flags |= FD_READ|FD_WRITE;
6140 /*fall through */
6141 default:
6142 clear_flags |= FD_WINE_LISTENING;
6143 }
6144
6145 if (!(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)))
6146 {
6147 switch ( how )
6148 {
6149 case SD_RECEIVE:
6150 err = WS2_register_async_shutdown( s, ASYNC_TYPE_READ );
6151 break;
6152 case SD_SEND:
6153 err = WS2_register_async_shutdown( s, ASYNC_TYPE_WRITE );
6154 break;
6155 case SD_BOTH:
6156 default:
6157 err = WS2_register_async_shutdown( s, ASYNC_TYPE_READ );
6158 if (!err) err = WS2_register_async_shutdown( s, ASYNC_TYPE_WRITE );
6159 break;
6160 }
6161 if (err) goto error;
6162 }
6163 else /* non-overlapped mode */
6164 {
6165 if ( shutdown( fd, how ) )
6166 {
6167 err = wsaErrno();
6168 goto error;
6169 }
6170 }
6171
6172 release_sock_fd( s, fd );
6173 _enable_event( SOCKET2HANDLE(s), 0, 0, clear_flags );
6174 if ( how > 1) WSAAsyncSelect( s, 0, 0, 0 );
6175 return 0;
6176
6177 error:
6178 release_sock_fd( s, fd );
6179 _enable_event( SOCKET2HANDLE(s), 0, 0, clear_flags );
6180 SetLastError( err );
6181 return SOCKET_ERROR;
6182 }
6183
6184 /***********************************************************************
6185 * socket (WS2_32.23)
6186 */
6187 SOCKET WINAPI WS_socket(int af, int type, int protocol)
6188 {
6189 TRACE("af=%d type=%d protocol=%d\n", af, type, protocol);
6190
6191 return WSASocketW( af, type, protocol, NULL, 0,
6192 get_per_thread_data()->opentype ? 0 : WSA_FLAG_OVERLAPPED );
6193 }
6194
6195
6196 /***********************************************************************
6197 * gethostbyaddr (WS2_32.51)
6198 */
6199 struct WS_hostent* WINAPI WS_gethostbyaddr(const char *addr, int len, int type)
6200 {
6201 struct WS_hostent *retval = NULL;
6202 struct hostent* host;
6203 int unixtype = convert_af_w2u(type);
6204 const char *paddr = addr;
6205 unsigned long loopback;
6206 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
6207 char *extrabuf;
6208 int ebufsize = 1024;
6209 struct hostent hostentry;
6210 int locerr = ENOBUFS;
6211 #endif
6212
6213 /* convert back the magic loopback address if necessary */
6214 if (unixtype == AF_INET && len == 4 && !memcmp(addr, magic_loopback_addr, 4))
6215 {
6216 loopback = htonl(INADDR_LOOPBACK);
6217 paddr = (char*) &loopback;
6218 }
6219
6220 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
6221 host = NULL;
6222 extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ;
6223 while(extrabuf) {
6224 int res = gethostbyaddr_r(paddr, len, unixtype,
6225 &hostentry, extrabuf, ebufsize, &host, &locerr);
6226 if (res != ERANGE) break;
6227 ebufsize *=2;
6228 extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ;
6229 }
6230 if (host) retval = WS_dup_he(host);
6231 else SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr));
6232 HeapFree(GetProcessHeap(),0,extrabuf);
6233 #else
6234 EnterCriticalSection( &csWSgetXXXbyYYY );
6235 host = gethostbyaddr(paddr, len, unixtype);
6236 if (host) retval = WS_dup_he(host);
6237 else SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno));
6238 LeaveCriticalSection( &csWSgetXXXbyYYY );
6239 #endif
6240 TRACE("ptr %p, len %d, type %d ret %p\n", addr, len, type, retval);
6241 return retval;
6242 }
6243
6244 /***********************************************************************
6245 * WS_compare_routes_by_metric_asc (INTERNAL)
6246 *
6247 * Comparison function for qsort(), for sorting two routes (struct route)
6248 * by metric in ascending order.
6249 */
6250 static int WS_compare_routes_by_metric_asc(const void *left, const void *right)
6251 {
6252 const struct route *a = left, *b = right;
6253 if (a->default_route && b->default_route)
6254 return a->default_route - b->default_route;
6255 if (a->default_route && !b->default_route)
6256 return -1;
6257 if (b->default_route && !a->default_route)
6258 return 1;
6259 return a->metric - b->metric;
6260 }
6261
6262 /***********************************************************************
6263 * WS_get_local_ips (INTERNAL)
6264 *
6265 * Returns the list of local IP addresses by going through the network
6266 * adapters and using the local routing table to sort the addresses
6267 * from highest routing priority to lowest routing priority. This
6268 * functionality is inferred from the description for obtaining local
6269 * IP addresses given in the Knowledge Base Article Q160215.
6270 *
6271 * Please note that the returned hostent is only freed when the thread
6272 * closes and is replaced if another hostent is requested.
6273 */
6274 static struct WS_hostent* WS_get_local_ips( char *hostname )
6275 {
6276 int numroutes = 0, i, j, default_routes = 0;
6277 DWORD n;
6278 PIP_ADAPTER_INFO adapters = NULL, k;
6279 struct WS_hostent *hostlist = NULL;
6280 PMIB_IPFORWARDTABLE routes = NULL;
6281 struct route *route_addrs = NULL;
6282 DWORD adap_size, route_size;
6283
6284 /* Obtain the size of the adapter list and routing table, also allocate memory */
6285 if (GetAdaptersInfo(NULL, &adap_size) != ERROR_BUFFER_OVERFLOW)
6286 return NULL;
6287 if (GetIpForwardTable(NULL, &route_size, FALSE) != ERROR_INSUFFICIENT_BUFFER)
6288 return NULL;
6289 adapters = HeapAlloc(GetProcessHeap(), 0, adap_size);
6290 routes = HeapAlloc(GetProcessHeap(), 0, route_size);
6291 route_addrs = HeapAlloc(GetProcessHeap(), 0, 0); /* HeapReAlloc doesn't work on NULL */
6292 if (adapters == NULL || routes == NULL || route_addrs == NULL)
6293 goto cleanup;
6294 /* Obtain the adapter list and the full routing table */
6295 if (GetAdaptersInfo(adapters, &adap_size) != NO_ERROR)
6296 goto cleanup;
6297 if (GetIpForwardTable(routes, &route_size, FALSE) != NO_ERROR)
6298 goto cleanup;
6299 /* Store the interface associated with each route */
6300 for (n = 0; n < routes->dwNumEntries; n++)
6301 {
6302 IF_INDEX ifindex;
6303 DWORD ifmetric, ifdefault = 0;
6304 BOOL exists = FALSE;
6305
6306 /* Check if this is a default route (there may be more than one) */
6307 if (!routes->table[n].dwForwardDest)
6308 ifdefault = ++default_routes;
6309 else if (routes->table[n].u1.ForwardType != MIB_IPROUTE_TYPE_DIRECT)
6310 continue;
6311 ifindex = routes->table[n].dwForwardIfIndex;
6312 ifmetric = routes->table[n].dwForwardMetric1;
6313 /* Only store the lowest valued metric for an interface */
6314 for (j = 0; j < numroutes; j++)
6315 {
6316 if (route_addrs[j].interface == ifindex)
6317 {
6318 if (route_addrs[j].metric > ifmetric)
6319 route_addrs[j].metric = ifmetric;
6320 exists = TRUE;
6321 }
6322 }
6323 if (exists)
6324 continue;
6325 route_addrs = HeapReAlloc(GetProcessHeap(), 0, route_addrs, (numroutes+1)*sizeof(struct route));
6326 if (route_addrs == NULL)
6327 goto cleanup; /* Memory allocation error, fail gracefully */
6328 route_addrs[numroutes].interface = ifindex;
6329 route_addrs[numroutes].metric = ifmetric;
6330 route_addrs[numroutes].default_route = ifdefault;
6331 /* If no IP is found in the next step (for whatever reason)
6332 * then fall back to the magic loopback address.
6333 */
6334 memcpy(&(route_addrs[numroutes].addr.s_addr), magic_loopback_addr, 4);
6335 numroutes++;
6336 }
6337 if (numroutes == 0)
6338 goto cleanup; /* No routes, fall back to the Magic IP */
6339 /* Find the IP address associated with each found interface */
6340 for (i = 0; i < numroutes; i++)
6341 {
6342 for (k = adapters; k != NULL; k = k->Next)
6343 {
6344 char *ip = k->IpAddressList.IpAddress.String;
6345
6346 if (route_addrs[i].interface == k->Index)
6347 route_addrs[i].addr.s_addr = (in_addr_t) inet_addr(ip);
6348 }
6349 }
6350 /* Allocate a hostent and enough memory for all the IPs,
6351 * including the NULL at the end of the list.
6352 */
6353 hostlist = WS_create_he(hostname, 1, 0, numroutes+1, sizeof(struct in_addr));
6354 if (hostlist == NULL)
6355 goto cleanup; /* Failed to allocate a hostent for the list of IPs */
6356 hostlist->h_addr_list[numroutes] = NULL; /* NULL-terminate the address list */
6357 hostlist->h_aliases[0] = NULL; /* NULL-terminate the alias list */
6358 hostlist->h_addrtype = AF_INET;
6359 hostlist->h_length = sizeof(struct in_addr); /* = 4 */
6360 /* Reorder the entries before placing them in the host list. Windows expects
6361 * the IP list in order from highest priority to lowest (the critical thing
6362 * is that most applications expect the first IP to be the default route).
6363 */
6364 if (numroutes > 1)
6365 qsort(route_addrs, numroutes, sizeof(struct route), WS_compare_routes_by_metric_asc);
6366
6367 for (i = 0; i < numroutes; i++)
6368 (*(struct in_addr *) hostlist->h_addr_list[i]) = route_addrs[i].addr;
6369
6370 /* Cleanup all allocated memory except the address list,
6371 * the address list is used by the calling app.
6372 */
6373 cleanup:
6374 HeapFree(GetProcessHeap(), 0, route_addrs);
6375 HeapFree(GetProcessHeap(), 0, adapters);
6376 HeapFree(GetProcessHeap(), 0, routes);
6377 return hostlist;
6378 }
6379
6380 /***********************************************************************
6381 * gethostbyname (WS2_32.52)
6382 */
6383 struct WS_hostent* WINAPI WS_gethostbyname(const char* name)
6384 {
6385 struct WS_hostent *retval = NULL;
6386 struct hostent* host;
6387 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
6388 char *extrabuf;
6389 int ebufsize=1024;
6390 struct hostent hostentry;
6391 int locerr = ENOBUFS;
6392 #endif
6393 char hostname[100];
6394 if(!num_startup) {
6395 SetLastError(WSANOTINITIALISED);
6396 return NULL;
6397 }
6398 if( gethostname( hostname, 100) == -1) {
6399 SetLastError(WSAENOBUFS); /* appropriate ? */
6400 return retval;
6401 }
6402 if( !name || !name[0]) {
6403 name = hostname;
6404 }
6405 /* If the hostname of the local machine is requested then return the
6406 * complete list of local IP addresses */
6407 if(strcmp(name, hostname) == 0)
6408 retval = WS_get_local_ips(hostname);
6409 /* If any other hostname was requested (or the routing table lookup failed)
6410 * then return the IP found by the host OS */
6411 if(retval == NULL)
6412 {
6413 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
6414 host = NULL;
6415 extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ;
6416 while(extrabuf) {
6417 int res = gethostbyname_r(name, &hostentry, extrabuf, ebufsize, &host, &locerr);
6418 if( res != ERANGE) break;
6419 ebufsize *=2;
6420 extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ;
6421 }
6422 if (!host) SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr));
6423 #else
6424 EnterCriticalSection( &csWSgetXXXbyYYY );
6425 host = gethostbyname(name);
6426 if (!host) SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno));
6427 #endif
6428 if (host) retval = WS_dup_he(host);
6429 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
6430 HeapFree(GetProcessHeap(),0,extrabuf);
6431 #else
6432 LeaveCriticalSection( &csWSgetXXXbyYYY );
6433 #endif
6434 }
6435 if (retval && retval->h_addr_list[0][0] == 127 &&
6436 strcmp(name, "localhost") != 0)
6437 {
6438 /* hostname != "localhost" but has loopback address. replace by our
6439 * special address.*/
6440 memcpy(retval->h_addr_list[0], magic_loopback_addr, 4);
6441 }
6442 TRACE( "%s ret %p\n", debugstr_a(name), retval );
6443 return retval;
6444 }
6445
6446
6447 static const struct { int prot; const char *names[3]; } protocols[] =
6448 {
6449 { 0, { "ip", "IP" }},
6450 { 1, { "icmp", "ICMP" }},
6451 { 2, { "igmp", "IGMP" }},
6452 { 3, { "ggp", "GGP" }},
6453 { 6, { "tcp", "TCP" }},
6454 { 8, { "egp", "EGP" }},
6455 { 9, { "igp", "IGP" }},
6456 { 12, { "pup", "PUP" }},
6457 { 17, { "udp", "UDP" }},
6458 { 20, { "hmp", "HMP" }},
6459 { 22, { "xns-idp", "XNS-IDP" }},
6460 { 27, { "rdp", "RDP" }},
6461 { 29, { "iso-tp4", "ISO-TP4" }},
6462 { 33, { "dccp", "DCCP" }},
6463 { 36, { "xtp", "XTP" }},
6464 { 37, { "ddp", "DDP" }},
6465 { 38, { "idpr-cmtp", "IDPR-CMTP" }},
6466 { 41, { "ipv6", "IPv6" }},
6467 { 43, { "ipv6-route", "IPv6-Route" }},
6468 { 44, { "ipv6-frag", "IPv6-Frag" }},
6469 { 45, { "idrp", "IDRP" }},
6470 { 46, { "rsvp", "RSVP" }},
6471 { 47, { "gre", "GRE" }},
6472 { 50, { "esp", "ESP" }},
6473 { 51, { "ah", "AH" }},
6474 { 57, { "skip", "SKIP" }},
6475 { 58, { "ipv6-icmp", "IPv6-ICMP" }},
6476 { 59, { "ipv6-nonxt", "IPv6-NoNxt" }},
6477 { 60, { "ipv6-opts", "IPv6-Opts" }},
6478 { 66, { "rvd", "RVD" }},
6479 { 73, { "rspf", "RSPF" }},
6480 { 81, { "vmtp", "VMTP" }},
6481 { 88, { "eigrp", "EIGRP" }},
6482 { 89, { "ospf", "OSPFIGP" }},
6483 { 93, { "ax.25", "AX.25" }},
6484 { 94, { "ipip", "IPIP" }},
6485 { 97, { "etherip", "ETHERIP" }},
6486 { 98, { "encap", "ENCAP" }},
6487 { 103, { "pim", "PIM" }},
6488 { 108, { "ipcomp", "IPCOMP" }},
6489 { 112, { "vrrp", "VRRP" }},
6490 { 115, { "l2tp", "L2TP" }},
6491 { 124, { "isis", "ISIS" }},
6492 { 132, { "sctp", "SCTP" }},
6493 { 133, { "fc", "FC" }},
6494 { 135, { "mobility-header", "Mobility-Header" }},
6495 { 136, { "udplite", "UDPLite" }},
6496 { 137, { "mpls-in-ip", "MPLS-in-IP" }},
6497 { 139, { "hip", "HIP" }},
6498 { 140, { "shim6", "Shim6" }},
6499 { 141, { "wesp", "WESP" }},
6500 { 142, { "rohc", "ROHC" }},
6501 };
6502
6503 /***********************************************************************
6504 * getprotobyname (WS2_32.53)
6505 */
6506 struct WS_protoent* WINAPI WS_getprotobyname(const char* name)
6507 {
6508 struct WS_protoent* retval = NULL;
6509 #ifdef HAVE_GETPROTOBYNAME
6510 struct protoent* proto;
6511 EnterCriticalSection( &csWSgetXXXbyYYY );
6512 if( (proto = getprotobyname(name)) != NULL )
6513 retval = WS_create_pe( proto->p_name, proto->p_aliases, proto->p_proto );
6514 LeaveCriticalSection( &csWSgetXXXbyYYY );
6515 #endif
6516 if (!retval)
6517 {
6518 unsigned int i;
6519 for (i = 0; i < ARRAY_SIZE(protocols); i++)
6520 {
6521 if (strcasecmp( protocols[i].names[0], name )) continue;
6522 retval = WS_create_pe( protocols[i].names[0], (char **)protocols[i].names + 1,
6523 protocols[i].prot );
6524 break;
6525 }
6526 }
6527 if (!retval)
6528 {
6529 WARN( "protocol %s not found\n", debugstr_a(name) );
6530 SetLastError(WSANO_DATA);
6531 }
6532 TRACE( "%s ret %p\n", debugstr_a(name), retval );
6533 return retval;
6534 }
6535
6536
6537 /***********************************************************************
6538 * getprotobynumber (WS2_32.54)
6539 */
6540 struct WS_protoent* WINAPI WS_getprotobynumber(int number)
6541 {
6542 struct WS_protoent* retval = NULL;
6543 #ifdef HAVE_GETPROTOBYNUMBER
6544 struct protoent* proto;
6545 EnterCriticalSection( &csWSgetXXXbyYYY );
6546 if( (proto = getprotobynumber(number)) != NULL )
6547 retval = WS_create_pe( proto->p_name, proto->p_aliases, proto->p_proto );
6548 LeaveCriticalSection( &csWSgetXXXbyYYY );
6549 #endif
6550 if (!retval)
6551 {
6552 unsigned int i;
6553 for (i = 0; i < ARRAY_SIZE(protocols); i++)
6554 {
6555 if (protocols[i].prot != number) continue;
6556 retval = WS_create_pe( protocols[i].names[0], (char **)protocols[i].names + 1,
6557 protocols[i].prot );
6558 break;
6559 }
6560 }
6561 if (!retval)
6562 {
6563 WARN( "protocol %d not found\n", number );
6564 SetLastError(WSANO_DATA);
6565 }
6566 TRACE("%i ret %p\n", number, retval);
6567 return retval;
6568 }
6569
6570
6571 /***********************************************************************
6572 * getservbyname (WS2_32.55)
6573 */
6574 struct WS_servent* WINAPI WS_getservbyname(const char *name, const char *proto)
6575 {
6576 struct WS_servent* retval = NULL;
6577 struct servent* serv;
6578 char *name_str;
6579 char *proto_str = NULL;
6580
6581 if (!(name_str = strdup_lower(name))) return NULL;
6582
6583 if (proto && *proto)
6584 {
6585 if (!(proto_str = strdup_lower(proto)))
6586 {
6587 HeapFree( GetProcessHeap(), 0, name_str );
6588 return NULL;
6589 }
6590 }
6591
6592 EnterCriticalSection( &csWSgetXXXbyYYY );
6593 serv = getservbyname(name_str, proto_str);
6594 if( serv != NULL )
6595 {
6596 retval = WS_dup_se(serv);
6597 }
6598 else SetLastError(WSANO_DATA);
6599 LeaveCriticalSection( &csWSgetXXXbyYYY );
6600 HeapFree( GetProcessHeap(), 0, proto_str );
6601 HeapFree( GetProcessHeap(), 0, name_str );
6602 TRACE( "%s, %s ret %p\n", debugstr_a(name), debugstr_a(proto), retval );
6603 return retval;
6604 }
6605
6606 /***********************************************************************
6607 * freeaddrinfo (WS2_32.@)
6608 */
6609 void WINAPI WS_freeaddrinfo(struct WS_addrinfo *res)
6610 {
6611 while (res) {
6612 struct WS_addrinfo *next;
6613
6614 HeapFree(GetProcessHeap(),0,res->ai_canonname);
6615 HeapFree(GetProcessHeap(),0,res->ai_addr);
6616 next = res->ai_next;
6617 HeapFree(GetProcessHeap(),0,res);
6618 res = next;
6619 }
6620 }
6621
6622 /* helper functions for getaddrinfo()/getnameinfo() */
6623 static int convert_aiflag_w2u(int winflags) {
6624 unsigned int i;
6625 int unixflags = 0;
6626
6627 for (i = 0; i < ARRAY_SIZE(ws_aiflag_map); i++)
6628 if (ws_aiflag_map[i][0] & winflags) {
6629 unixflags |= ws_aiflag_map[i][1];
6630 winflags &= ~ws_aiflag_map[i][0];
6631 }
6632 if (winflags)
6633 FIXME("Unhandled windows AI_xxx flags 0x%x\n", winflags);
6634 return unixflags;
6635 }
6636
6637 static int convert_niflag_w2u(int winflags) {
6638 unsigned int i;
6639 int unixflags = 0;
6640
6641 for (i = 0; i < ARRAY_SIZE(ws_niflag_map); i++)
6642 if (ws_niflag_map[i][0] & winflags) {
6643 unixflags |= ws_niflag_map[i][1];
6644 winflags &= ~ws_niflag_map[i][0];
6645 }
6646 if (winflags)
6647 FIXME("Unhandled windows NI_xxx flags 0x%x\n", winflags);
6648 return unixflags;
6649 }
6650
6651 static int convert_aiflag_u2w(int unixflags) {
6652 unsigned int i;
6653 int winflags = 0;
6654
6655 for (i = 0; i < ARRAY_SIZE(ws_aiflag_map); i++)
6656 if (ws_aiflag_map[i][1] & unixflags) {
6657 winflags |= ws_aiflag_map[i][0];
6658 unixflags &= ~ws_aiflag_map[i][1];
6659 }
6660 if (unixflags)
6661 WARN("Unhandled UNIX AI_xxx flags 0x%x\n", unixflags);
6662 return winflags;
6663 }
6664
6665 static int convert_eai_u2w(int unixret) {
6666 int i;
6667
6668 if (!unixret) return 0;
6669
6670 for (i=0;ws_eai_map[i][0];i++)
6671 if (ws_eai_map[i][1] == unixret)
6672 return ws_eai_map[i][0];
6673
6674 if (unixret == EAI_SYSTEM)
6675 /* There are broken versions of glibc which return EAI_SYSTEM
6676 * and set errno to 0 instead of returning EAI_NONAME.
6677 */
6678 return errno ? sock_get_error( errno ) : WS_EAI_NONAME;
6679
6680 FIXME("Unhandled unix EAI_xxx ret %d\n", unixret);
6681 return unixret;
6682 }
6683
6684 static char *get_fqdn(void)
6685 {
6686 char *ret;
6687 DWORD size = 0;
6688
6689 GetComputerNameExA( ComputerNamePhysicalDnsFullyQualified, NULL, &size );
6690 if (GetLastError() != ERROR_MORE_DATA) return NULL;
6691 if (!(ret = HeapAlloc( GetProcessHeap(), 0, size ))) return NULL;
6692 if (!GetComputerNameExA( ComputerNamePhysicalDnsFullyQualified, ret, &size ))
6693 {
6694 HeapFree( GetProcessHeap(), 0, ret );
6695 return NULL;
6696 }
6697 return ret;
6698 }
6699
6700 /***********************************************************************
6701 * getaddrinfo (WS2_32.@)
6702 */
6703 int WINAPI WS_getaddrinfo(LPCSTR nodename, LPCSTR servname, const struct WS_addrinfo *hints, struct WS_addrinfo **res)
6704 {
6705 #ifdef HAVE_GETADDRINFO
6706 struct addrinfo *unixaires = NULL;
6707 int result;
6708 struct addrinfo unixhints, *punixhints = NULL;
6709 char *dot, *nodeV6 = NULL, *fqdn;
6710 const char *node;
6711 size_t hostname_len = 0;
6712
6713 *res = NULL;
6714 if (!nodename && !servname)
6715 {
6716 SetLastError(WSAHOST_NOT_FOUND);
6717 return WSAHOST_NOT_FOUND;
6718 }
6719
6720 fqdn = get_fqdn();
6721 if (!fqdn) return WSA_NOT_ENOUGH_MEMORY;
6722 dot = strchr(fqdn, '.');
6723 if (dot)
6724 hostname_len = dot - fqdn;
6725
6726 if (!nodename)
6727 node = NULL;
6728 else if (!nodename[0])
6729 node = fqdn;
6730 else
6731 {
6732 node = nodename;
6733
6734 /* Check for [ipv6] or [ipv6]:portnumber, which are supported by Windows */
6735 if (!hints || hints->ai_family == WS_AF_UNSPEC || hints->ai_family == WS_AF_INET6)
6736 {
6737 char *close_bracket;
6738
6739 if (node[0] == '[' && (close_bracket = strchr(node + 1, ']')))
6740 {
6741 nodeV6 = HeapAlloc(GetProcessHeap(), 0, close_bracket - node);
6742 if (!nodeV6)
6743 {
6744 HeapFree(GetProcessHeap(), 0, fqdn);
6745 return WSA_NOT_ENOUGH_MEMORY;
6746 }
6747 lstrcpynA(nodeV6, node + 1, close_bracket - node);
6748 node = nodeV6;
6749 }
6750 }
6751 }
6752
6753 /* servname tweak required by OSX and BSD kernels */
6754 if (servname && !servname[0]) servname = "0";
6755
6756 if (hints) {
6757 punixhints = &unixhints;
6758
6759 memset(&unixhints, 0, sizeof(unixhints));
6760 punixhints->ai_flags = convert_aiflag_w2u(hints->ai_flags);
6761
6762 /* zero is a wildcard, no need to convert */
6763 if (hints->ai_family)
6764 punixhints->ai_family = convert_af_w2u(hints->ai_family);
6765 if (hints->ai_socktype)
6766 punixhints->ai_socktype = convert_socktype_w2u(hints->ai_socktype);
6767 if (hints->ai_protocol)
6768 punixhints->ai_protocol = max(convert_proto_w2u(hints->ai_protocol), 0);
6769
6770 if (punixhints->ai_socktype < 0)
6771 {
6772 SetLastError(WSAESOCKTNOSUPPORT);
6773 HeapFree(GetProcessHeap(), 0, fqdn);
6774 HeapFree(GetProcessHeap(), 0, nodeV6);
6775 return SOCKET_ERROR;
6776 }
6777
6778 /* windows allows invalid combinations of socket type and protocol, unix does not.
6779 * fix the parameters here to make getaddrinfo call always work */
6780 if (punixhints->ai_protocol == IPPROTO_TCP &&
6781 punixhints->ai_socktype != SOCK_STREAM && punixhints->ai_socktype != SOCK_SEQPACKET)
6782 punixhints->ai_socktype = 0;
6783
6784 else if (punixhints->ai_protocol == IPPROTO_UDP && punixhints->ai_socktype != SOCK_DGRAM)
6785 punixhints->ai_socktype = 0;
6786
6787 else if (IS_IPX_PROTO(punixhints->ai_protocol) && punixhints->ai_socktype != SOCK_DGRAM)
6788 punixhints->ai_socktype = 0;
6789
6790 else if (punixhints->ai_protocol == IPPROTO_IPV6)
6791 punixhints->ai_protocol = 0;
6792 }
6793
6794 /* getaddrinfo(3) is thread safe, no need to wrap in CS */
6795 result = getaddrinfo(node, servname, punixhints, &unixaires);
6796
6797 if (result && (!hints || !(hints->ai_flags & WS_AI_NUMERICHOST))
6798 && (!strcmp(fqdn, node) || (!strncmp(fqdn, node, hostname_len) && !node[hostname_len])))
6799 {
6800 /* If it didn't work it means the host name IP is not in /etc/hosts, try again
6801 * by sending a NULL host and avoid sending a NULL servname too because that
6802 * is invalid */
6803 ERR_(winediag)("Failed to resolve your host name IP\n");
6804 result = getaddrinfo(NULL, servname ? servname : "0", punixhints, &unixaires);
6805 }
6806 TRACE("%s, %s %p -> %p %d\n", debugstr_a(nodename), debugstr_a(servname), hints, res, result);
6807 HeapFree(GetProcessHeap(), 0, fqdn);
6808 HeapFree(GetProcessHeap(), 0, nodeV6);
6809
6810 if (!result) {
6811 struct addrinfo *xuai = unixaires;
6812 struct WS_addrinfo **xai = res;
6813
6814 *xai = NULL;
6815 while (xuai) {
6816 struct WS_addrinfo *ai = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY, sizeof(struct WS_addrinfo));
6817 SIZE_T len;
6818
6819 if (!ai)
6820 goto outofmem;
6821
6822 *xai = ai;xai = &ai->ai_next;
6823 ai->ai_flags = convert_aiflag_u2w(xuai->ai_flags);
6824 ai->ai_family = convert_af_u2w(xuai->ai_family);
6825 /* copy whatever was sent in the hints */
6826 if(hints) {
6827 ai->ai_socktype = hints->ai_socktype;
6828 ai->ai_protocol = hints->ai_protocol;
6829 } else {
6830 ai->ai_socktype = convert_socktype_u2w(xuai->ai_socktype);
6831 ai->ai_protocol = convert_proto_u2w(xuai->ai_protocol);
6832 }
6833 if (xuai->ai_canonname) {
6834 TRACE("canon name - %s\n",debugstr_a(xuai->ai_canonname));
6835 ai->ai_canonname = HeapAlloc(GetProcessHeap(),0,strlen(xuai->ai_canonname)+1);
6836 if (!ai->ai_canonname)
6837 goto outofmem;
6838 strcpy(ai->ai_canonname,xuai->ai_canonname);
6839 }
6840 len = xuai->ai_addrlen;
6841 ai->ai_addr = HeapAlloc(GetProcessHeap(),0,len);
6842 if (!ai->ai_addr)
6843 goto outofmem;
6844 ai->ai_addrlen = len;
6845 do {
6846 int winlen = ai->ai_addrlen;
6847
6848 if (!ws_sockaddr_u2ws(xuai->ai_addr, ai->ai_addr, &winlen)) {
6849 ai->ai_addrlen = winlen;
6850 break;
6851 }
6852 len = 2*len;
6853 ai->ai_addr = HeapReAlloc(GetProcessHeap(),0,ai->ai_addr,len);
6854 if (!ai->ai_addr)
6855 goto outofmem;
6856 ai->ai_addrlen = len;
6857 } while (1);
6858 xuai = xuai->ai_next;
6859 }
6860 freeaddrinfo(unixaires);
6861
6862 if (TRACE_ON(winsock))
6863 {
6864 struct WS_addrinfo *ai = *res;
6865 while (ai)
6866 {
6867 TRACE("=> %p, flags %#x, family %d, type %d, protocol %d, len %ld, name %s, addr %s\n",
6868 ai, ai->ai_flags, ai->ai_family, ai->ai_socktype, ai->ai_protocol, ai->ai_addrlen,
6869 ai->ai_canonname, debugstr_sockaddr(ai->ai_addr));
6870 ai = ai->ai_next;
6871 }
6872 }
6873 } else
6874 result = convert_eai_u2w(result);
6875
6876 SetLastError(result);
6877 return result;
6878
6879 outofmem:
6880 if (*res) WS_freeaddrinfo(*res);
6881 if (unixaires) freeaddrinfo(unixaires);
6882 return WSA_NOT_ENOUGH_MEMORY;
6883 #else
6884 FIXME("getaddrinfo() failed, not found during buildtime.\n");
6885 return EAI_FAIL;
6886 #endif
6887 }
6888
6889 static ADDRINFOEXW *addrinfo_AtoW(const struct WS_addrinfo *ai)
6890 {
6891 ADDRINFOEXW *ret;
6892
6893 if (!(ret = HeapAlloc(GetProcessHeap(), 0, sizeof(ADDRINFOEXW)))) return NULL;
6894 ret->ai_flags = ai->ai_flags;
6895 ret->ai_family = ai->ai_family;
6896 ret->ai_socktype = ai->ai_socktype;
6897 ret->ai_protocol = ai->ai_protocol;
6898 ret->ai_addrlen = ai->ai_addrlen;
6899 ret->ai_canonname = NULL;
6900 ret->ai_addr = NULL;
6901 ret->ai_blob = NULL;
6902 ret->ai_bloblen = 0;
6903 ret->ai_provider = NULL;
6904 ret->ai_next = NULL;
6905 if (ai->ai_canonname)
6906 {
6907 int len = MultiByteToWideChar(CP_ACP, 0, ai->ai_canonname, -1, NULL, 0);
6908 if (!(ret->ai_canonname = HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR))))
6909 {
6910 HeapFree(GetProcessHeap(), 0, ret);
6911 return NULL;
6912 }
6913 MultiByteToWideChar(CP_ACP, 0, ai->ai_canonname, -1, ret->ai_canonname, len);
6914 }
6915 if (ai->ai_addr)
6916 {
6917 if (!(ret->ai_addr = HeapAlloc(GetProcessHeap(), 0, ai->ai_addrlen)))
6918 {
6919 HeapFree(GetProcessHeap(), 0, ret->ai_canonname);
6920 HeapFree(GetProcessHeap(), 0, ret);
6921 return NULL;
6922 }
6923 memcpy(ret->ai_addr, ai->ai_addr, ai->ai_addrlen);
6924 }
6925 return ret;
6926 }
6927
6928 static ADDRINFOEXW *addrinfo_list_AtoW(const struct WS_addrinfo *info)
6929 {
6930 ADDRINFOEXW *ret, *infoW;
6931
6932 if (!(ret = infoW = addrinfo_AtoW(info))) return NULL;
6933 while (info->ai_next)
6934 {
6935 if (!(infoW->ai_next = addrinfo_AtoW(info->ai_next)))
6936 {
6937 FreeAddrInfoExW(ret);
6938 return NULL;
6939 }
6940 infoW = infoW->ai_next;
6941 info = info->ai_next;
6942 }
6943 return ret;
6944 }
6945
6946 static struct WS_addrinfo *addrinfo_WtoA(const struct WS_addrinfoW *ai)
6947 {
6948 struct WS_addrinfo *ret;
6949
6950 if (!(ret = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_addrinfo)))) return NULL;
6951 ret->ai_flags = ai->ai_flags;
6952 ret->ai_family = ai->ai_family;
6953 ret->ai_socktype = ai->ai_socktype;
6954 ret->ai_protocol = ai->ai_protocol;
6955 ret->ai_addrlen = ai->ai_addrlen;
6956 ret->ai_canonname = NULL;
6957 ret->ai_addr = NULL;
6958 ret->ai_next = NULL;
6959 if (ai->ai_canonname)
6960 {
6961 int len = WideCharToMultiByte(CP_ACP, 0, ai->ai_canonname, -1, NULL, 0, NULL, NULL);
6962 if (!(ret->ai_canonname = HeapAlloc(GetProcessHeap(), 0, len)))
6963 {
6964 HeapFree(GetProcessHeap(), 0, ret);
6965 return NULL;
6966 }
6967 WideCharToMultiByte(CP_ACP, 0, ai->ai_canonname, -1, ret->ai_canonname, len, NULL, NULL);
6968 }
6969 if (ai->ai_addr)
6970 {
6971 if (!(ret->ai_addr = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_sockaddr))))
6972 {
6973 HeapFree(GetProcessHeap(), 0, ret->ai_canonname);
6974 HeapFree(GetProcessHeap(), 0, ret);
6975 return NULL;
6976 }
6977 memcpy(ret->ai_addr, ai->ai_addr, sizeof(struct WS_sockaddr));
6978 }
6979 return ret;
6980 }
6981
6982 struct getaddrinfo_args
6983 {
6984 OVERLAPPED *overlapped;
6985 ADDRINFOEXW **result;
6986 char *nodename;
6987 char *servname;
6988 };
6989
6990 static void WINAPI getaddrinfo_callback(TP_CALLBACK_INSTANCE *instance, void *context)
6991 {
6992 struct getaddrinfo_args *args = context;
6993 OVERLAPPED *overlapped = args->overlapped;
6994 HANDLE event = overlapped->hEvent;
6995 struct WS_addrinfo *res;
6996 int ret;
6997
6998 ret = WS_getaddrinfo(args->nodename, args->servname, NULL, &res);
6999 if (res)
7000 {
7001 *args->result = addrinfo_list_AtoW(res);
7002 overlapped->u.Pointer = args->result;
7003 WS_freeaddrinfo(res);
7004 }
7005
7006 HeapFree(GetProcessHeap(), 0, args->nodename);
7007 HeapFree(GetProcessHeap(), 0, args->servname);
7008 HeapFree(GetProcessHeap(), 0, args);
7009
7010 overlapped->Internal = ret;
7011 if (event) SetEvent(event);
7012 }
7013
7014 static int WS_getaddrinfoW(const WCHAR *nodename, const WCHAR *servname, const struct WS_addrinfo *hints, ADDRINFOEXW **res, OVERLAPPED *overlapped)
7015 {
7016 int ret = EAI_MEMORY, len, i;
7017 char *nodenameA = NULL, *servnameA = NULL;
7018 struct WS_addrinfo *resA;
7019 WCHAR *local_nodenameW = (WCHAR *)nodename;
7020
7021 *res = NULL;
7022 if (nodename)
7023 {
7024 /* Is this an IDN? Most likely if any char is above the Ascii table, this
7025 * is the simplest validation possible, further validation will be done by
7026 * the native getaddrinfo() */
7027 for (i = 0; nodename[i]; i++)
7028 {
7029 if (nodename[i] > 'z')
7030 break;
7031 }
7032 if (nodename[i])
7033 {
7034 if (hints && (hints->ai_flags & WS_AI_DISABLE_IDN_ENCODING))
7035 {
7036 /* Name requires conversion but it was disabled */
7037 ret = WSAHOST_NOT_FOUND;
7038 WSASetLastError(ret);
7039 goto end;
7040 }
7041
7042 len = IdnToAscii(0, nodename, -1, NULL, 0);
7043 if (!len)
7044 {
7045 ERR("Failed to convert %s to punycode\n", debugstr_w(nodename));
7046 ret = EAI_FAIL;
7047 goto end;
7048 }
7049 if (!(local_nodenameW = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR)))) goto end;
7050 IdnToAscii(0, nodename, -1, local_nodenameW, len);
7051 }
7052 }
7053 if (local_nodenameW)
7054 {
7055 len = WideCharToMultiByte(CP_ACP, 0, local_nodenameW, -1, NULL, 0, NULL, NULL);
7056 if (!(nodenameA = HeapAlloc(GetProcessHeap(), 0, len))) goto end;
7057 WideCharToMultiByte(CP_ACP, 0, local_nodenameW, -1, nodenameA, len, NULL, NULL);
7058 }
7059 if (servname)
7060 {
7061 len = WideCharToMultiByte(CP_ACP, 0, servname, -1, NULL, 0, NULL, NULL);
7062 if (!(servnameA = HeapAlloc(GetProcessHeap(), 0, len))) goto end;
7063 WideCharToMultiByte(CP_ACP, 0, servname, -1, servnameA, len, NULL, NULL);
7064 }
7065
7066 if (overlapped)
7067 {
7068 struct getaddrinfo_args *args;
7069
7070 if (!(args = HeapAlloc(GetProcessHeap(), 0, sizeof(*args)))) goto end;
7071 args->overlapped = overlapped;
7072 args->result = res;
7073 args->nodename = nodenameA;
7074 args->servname = servnameA;
7075
7076 overlapped->Internal = WSAEINPROGRESS;
7077 if (!TrySubmitThreadpoolCallback(getaddrinfo_callback, args, NULL))
7078 {
7079 HeapFree(GetProcessHeap(), 0, args);
7080 ret = GetLastError();
7081 goto end;
7082 }
7083
7084
7085 if (local_nodenameW != nodename)
7086 HeapFree(GetProcessHeap(), 0, local_nodenameW);
7087 WSASetLastError(ERROR_IO_PENDING);
7088 return ERROR_IO_PENDING;
7089 }
7090
7091 ret = WS_getaddrinfo(nodenameA, servnameA, hints, &resA);
7092 if (!ret)
7093 {
7094 *res = addrinfo_list_AtoW(resA);
7095 WS_freeaddrinfo(resA);
7096 }
7097
7098 end:
7099 if (local_nodenameW != nodename)
7100 HeapFree(GetProcessHeap(), 0, local_nodenameW);
7101 HeapFree(GetProcessHeap(), 0, nodenameA);
7102 HeapFree(GetProcessHeap(), 0, servnameA);
7103 return ret;
7104 }
7105
7106 /***********************************************************************
7107 * GetAddrInfoExW (WS2_32.@)
7108 */
7109 int WINAPI GetAddrInfoExW(const WCHAR *name, const WCHAR *servname, DWORD namespace, GUID *namespace_id,
7110 const ADDRINFOEXW *hints, ADDRINFOEXW **result, struct timeval *timeout, OVERLAPPED *overlapped,
7111 LPLOOKUPSERVICE_COMPLETION_ROUTINE completion_routine, HANDLE *handle)
7112 {
7113 int ret;
7114
7115 TRACE("(%s %s %x %s %p %p %p %p %p %p)\n", debugstr_w(name), debugstr_w(servname), namespace,
7116 debugstr_guid(namespace_id), hints, result, timeout, overlapped, completion_routine, handle);
7117
7118 if (namespace != NS_DNS)
7119 FIXME("Unsupported namespace %u\n", namespace);
7120 if (namespace_id)
7121 FIXME("Unsupported naemspace_id %s\n", debugstr_guid(namespace_id));
7122 if (hints)
7123 FIXME("Unsupported hints\n");
7124 if (timeout)
7125 FIXME("Unsupported timeout\n");
7126 if (completion_routine)
7127 FIXME("Unsupported completion_routine\n");
7128 if (handle)
7129 FIXME("Unsupported cancel handle\n");
7130
7131 ret = WS_getaddrinfoW(name, servname, NULL, result, overlapped);
7132 if (ret) return ret;
7133 if (handle) *handle = (HANDLE)0xdeadbeef;
7134 return 0;
7135 }
7136
7137 /***********************************************************************
7138 * GetAddrInfoExOverlappedResult (WS2_32.@)
7139 */
7140 int WINAPI GetAddrInfoExOverlappedResult(OVERLAPPED *overlapped)
7141 {
7142 TRACE("(%p)\n", overlapped);
7143 return overlapped->Internal;
7144 }
7145
7146 /***********************************************************************
7147 * GetAddrInfoExCancel (WS2_32.@)
7148 */
7149 int WINAPI GetAddrInfoExCancel(HANDLE *handle)
7150 {
7151 FIXME("(%p)\n", handle);
7152 return WSA_INVALID_HANDLE;
7153 }
7154
7155 /***********************************************************************
7156 * GetAddrInfoW (WS2_32.@)
7157 */
7158 int WINAPI GetAddrInfoW(LPCWSTR nodename, LPCWSTR servname, const ADDRINFOW *hints, PADDRINFOW *res)
7159 {
7160 struct WS_addrinfo *hintsA = NULL;
7161 ADDRINFOEXW *resex;
7162 int ret = EAI_MEMORY;
7163
7164 TRACE("nodename %s, servname %s, hints %p, result %p\n",
7165 debugstr_w(nodename), debugstr_w(servname), hints, res);
7166
7167 *res = NULL;
7168 if (hints) hintsA = addrinfo_WtoA(hints);
7169 ret = WS_getaddrinfoW(nodename, servname, hintsA, &resex, NULL);
7170 WS_freeaddrinfo(hintsA);
7171 if (ret) return ret;
7172
7173 if (resex)
7174 {
7175 /* ADDRINFOEXW has layout compatible with ADDRINFOW except for ai_next field,
7176 * so we may convert it in place */
7177 *res = (ADDRINFOW*)resex;
7178 do {
7179 ((ADDRINFOW*)resex)->ai_next = (ADDRINFOW*)resex->ai_next;
7180 resex = resex->ai_next;
7181 } while (resex);
7182 }
7183 return 0;
7184 }
7185
7186 /***********************************************************************
7187 * FreeAddrInfoW (WS2_32.@)
7188 */
7189 void WINAPI FreeAddrInfoW(PADDRINFOW ai)
7190 {
7191 while (ai)
7192 {
7193 ADDRINFOW *next;
7194 HeapFree(GetProcessHeap(), 0, ai->ai_canonname);
7195 HeapFree(GetProcessHeap(), 0, ai->ai_addr);
7196 next = ai->ai_next;
7197 HeapFree(GetProcessHeap(), 0, ai);
7198 ai = next;
7199 }
7200 }
7201
7202 /***********************************************************************
7203 * FreeAddrInfoExW (WS2_32.@)
7204 */
7205 void WINAPI FreeAddrInfoExW(ADDRINFOEXW *ai)
7206 {
7207 TRACE("(%p)\n", ai);
7208
7209 while (ai)
7210 {
7211 ADDRINFOEXW *next;
7212 HeapFree(GetProcessHeap(), 0, ai->ai_canonname);
7213 HeapFree(GetProcessHeap(), 0, ai->ai_addr);
7214 next = ai->ai_next;
7215 HeapFree(GetProcessHeap(), 0, ai);
7216 ai = next;
7217 }
7218 }
7219
7220 int WINAPI WS_getnameinfo(const SOCKADDR *sa, WS_socklen_t salen, PCHAR host,
7221 DWORD hostlen, PCHAR serv, DWORD servlen, INT flags)
7222 {
7223 #ifdef HAVE_GETNAMEINFO
7224 int ret;
7225 union generic_unix_sockaddr sa_u;
7226 unsigned int size;
7227
7228 TRACE("%s %d %p %d %p %d %d\n", debugstr_sockaddr(sa), salen, host, hostlen,
7229 serv, servlen, flags);
7230
7231 size = ws_sockaddr_ws2u(sa, salen, &sa_u);
7232 if (!size)
7233 {
7234 SetLastError(WSAEFAULT);
7235 return WSA_NOT_ENOUGH_MEMORY;
7236 }
7237 ret = getnameinfo(&sa_u.addr, size, host, hostlen, serv, servlen, convert_niflag_w2u(flags));
7238 return convert_eai_u2w(ret);
7239 #else
7240 FIXME("getnameinfo() failed, not found during buildtime.\n");
7241 return EAI_FAIL;
7242 #endif
7243 }
7244
7245 int WINAPI GetNameInfoW(const SOCKADDR *sa, WS_socklen_t salen, PWCHAR host,
7246 DWORD hostlen, PWCHAR serv, DWORD servlen, INT flags)
7247 {
7248 int ret;
7249 char *hostA = NULL, *servA = NULL;
7250
7251 if (host && (!(hostA = HeapAlloc(GetProcessHeap(), 0, hostlen)))) return EAI_MEMORY;
7252 if (serv && (!(servA = HeapAlloc(GetProcessHeap(), 0, servlen))))
7253 {
7254 HeapFree(GetProcessHeap(), 0, hostA);
7255 return EAI_MEMORY;
7256 }
7257
7258 ret = WS_getnameinfo(sa, salen, hostA, hostlen, servA, servlen, flags);
7259 if (!ret)
7260 {
7261 if (host) MultiByteToWideChar(CP_ACP, 0, hostA, -1, host, hostlen);
7262 if (serv) MultiByteToWideChar(CP_ACP, 0, servA, -1, serv, servlen);
7263 }
7264
7265 HeapFree(GetProcessHeap(), 0, hostA);
7266 HeapFree(GetProcessHeap(), 0, servA);
7267 return ret;
7268 }
7269
7270 /***********************************************************************
7271 * getservbyport (WS2_32.56)
7272 */
7273 struct WS_servent* WINAPI WS_getservbyport(int port, const char *proto)
7274 {
7275 struct WS_servent* retval = NULL;
7276 #ifdef HAVE_GETSERVBYPORT
7277 struct servent* serv;
7278 char *proto_str = NULL;
7279
7280 if (proto && *proto)
7281 {
7282 if (!(proto_str = strdup_lower(proto))) return NULL;
7283 }
7284 EnterCriticalSection( &csWSgetXXXbyYYY );
7285 if( (serv = getservbyport(port, proto_str)) != NULL ) {
7286 retval = WS_dup_se(serv);
7287 }
7288 else SetLastError(WSANO_DATA);
7289 LeaveCriticalSection( &csWSgetXXXbyYYY );
7290 HeapFree( GetProcessHeap(), 0, proto_str );
7291 #endif
7292 TRACE("%d (i.e. port %d), %s ret %p\n", port, (int)ntohl(port), debugstr_a(proto), retval);
7293 return retval;
7294 }
7295
7296
7297 /***********************************************************************
7298 * gethostname (WS2_32.57)
7299 */
7300 int WINAPI WS_gethostname(char *name, int namelen)
7301 {
7302 char buf[256];
7303 int len;
7304
7305 TRACE("name %p, len %d\n", name, namelen);
7306
7307 if (!name)
7308 {
7309 SetLastError(WSAEFAULT);
7310 return SOCKET_ERROR;
7311 }
7312
7313 if (gethostname(buf, sizeof(buf)) != 0)
7314 {
7315 SetLastError(wsaErrno());
7316 return SOCKET_ERROR;
7317 }
7318
7319 TRACE("<- '%s'\n", buf);
7320 len = strlen(buf);
7321 if (len > 15)
7322 WARN("Windows supports NetBIOS name length up to 15 bytes!\n");
7323 if (namelen <= len)
7324 {
7325 SetLastError(WSAEFAULT);
7326 WARN("<- not enough space for hostname, required %d, got %d!\n", len + 1, namelen);
7327 return SOCKET_ERROR;
7328 }
7329 strcpy(name, buf);
7330 return 0;
7331 }
7332
7333
7334 /* ------------------------------------- Windows sockets extensions -- *
7335 * *
7336 * ------------------------------------------------------------------- */
7337
7338 /***********************************************************************
7339 * WSAEnumNetworkEvents (WS2_32.36)
7340 */
7341 int WINAPI WSAEnumNetworkEvents(SOCKET s, WSAEVENT hEvent, LPWSANETWORKEVENTS lpEvent)
7342 {
7343 int ret;
7344 int i;
7345 int errors[FD_MAX_EVENTS];
7346
7347 TRACE("%04lx, hEvent %p, lpEvent %p\n", s, hEvent, lpEvent );
7348
7349 SERVER_START_REQ( get_socket_event )
7350 {
7351 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
7352 req->service = TRUE;
7353 req->c_event = wine_server_obj_handle( hEvent );
7354 wine_server_set_reply( req, errors, sizeof(errors) );
7355 if (!(ret = wine_server_call(req))) lpEvent->lNetworkEvents = reply->pmask & reply->mask;
7356 }
7357 SERVER_END_REQ;
7358 if (!ret)
7359 {
7360 for (i = 0; i < FD_MAX_EVENTS; i++)
7361 {
7362 if (lpEvent->lNetworkEvents & (1 << i))
7363 lpEvent->iErrorCode[i] = errors[i];
7364 }
7365 return 0;
7366 }
7367 SetLastError(WSAEINVAL);
7368 return SOCKET_ERROR;
7369 }
7370
7371 /***********************************************************************
7372 * WSAEventSelect (WS2_32.39)
7373 */
7374 int WINAPI WSAEventSelect(SOCKET s, WSAEVENT hEvent, LONG lEvent)
7375 {
7376 int ret;
7377
7378 TRACE("%04lx, hEvent %p, event %08x\n", s, hEvent, lEvent);
7379
7380 SERVER_START_REQ( set_socket_event )
7381 {
7382 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
7383 req->mask = lEvent;
7384 req->event = wine_server_obj_handle( hEvent );
7385 req->window = 0;
7386 req->msg = 0;
7387 ret = wine_server_call( req );
7388 }
7389 SERVER_END_REQ;
7390 if (!ret) return 0;
7391 SetLastError(WSAEINVAL);
7392 return SOCKET_ERROR;
7393 }
7394
7395 /**********************************************************************
7396 * WSAGetOverlappedResult (WS2_32.40)
7397 */
7398 BOOL WINAPI WSAGetOverlappedResult( SOCKET s, LPWSAOVERLAPPED lpOverlapped,
7399 LPDWORD lpcbTransfer, BOOL fWait,
7400 LPDWORD lpdwFlags )
7401 {
7402 NTSTATUS status;
7403
7404 TRACE( "socket %04lx ovl %p trans %p, wait %d flags %p\n",
7405 s, lpOverlapped, lpcbTransfer, fWait, lpdwFlags );
7406
7407 if ( lpOverlapped == NULL )
7408 {
7409 ERR( "Invalid pointer\n" );
7410 SetLastError(WSA_INVALID_PARAMETER);
7411 return FALSE;
7412 }
7413
7414 status = lpOverlapped->Internal;
7415 if (status == STATUS_PENDING)
7416 {
7417 if (!fWait)
7418 {
7419 SetLastError( WSA_IO_INCOMPLETE );
7420 return FALSE;
7421 }
7422
7423 if (WaitForSingleObject( lpOverlapped->hEvent ? lpOverlapped->hEvent : SOCKET2HANDLE(s),
7424 INFINITE ) == WAIT_FAILED)
7425 return FALSE;
7426 status = lpOverlapped->Internal;
7427 }
7428
7429 if ( lpcbTransfer )
7430 *lpcbTransfer = lpOverlapped->InternalHigh;
7431
7432 if ( lpdwFlags )
7433 *lpdwFlags = lpOverlapped->u.s.Offset;
7434
7435 if (status) SetLastError( RtlNtStatusToDosError(status) );
7436 return !status;
7437 }
7438
7439
7440 /***********************************************************************
7441 * WSAAsyncSelect (WS2_32.101)
7442 */
7443 INT WINAPI WSAAsyncSelect(SOCKET s, HWND hWnd, UINT uMsg, LONG lEvent)
7444 {
7445 int ret;
7446
7447 TRACE("%04lx, hWnd %p, uMsg %08x, event %08x\n", s, hWnd, uMsg, lEvent);
7448
7449 SERVER_START_REQ( set_socket_event )
7450 {
7451 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
7452 req->mask = lEvent;
7453 req->event = 0;
7454 req->window = wine_server_user_handle( hWnd );
7455 req->msg = uMsg;
7456 ret = wine_server_call( req );
7457 }
7458 SERVER_END_REQ;
7459 if (!ret) return 0;
7460 SetLastError(WSAEINVAL);
7461 return SOCKET_ERROR;
7462 }
7463
7464 /***********************************************************************
7465 * WSACreateEvent (WS2_32.31)
7466 *
7467 */
7468 WSAEVENT WINAPI WSACreateEvent(void)
7469 {
7470 /* Create a manual-reset event, with initial state: unsignaled */
7471 TRACE("\n");
7472
7473 return CreateEventW(NULL, TRUE, FALSE, NULL);
7474 }
7475
7476 /***********************************************************************
7477 * WSACloseEvent (WS2_32.29)
7478 *
7479 */
7480 BOOL WINAPI WSACloseEvent(WSAEVENT event)
7481 {
7482 TRACE ("event=%p\n", event);
7483
7484 return CloseHandle(event);
7485 }
7486
7487 /***********************************************************************
7488 * WSASocketA (WS2_32.78)
7489 *
7490 */
7491 SOCKET WINAPI WSASocketA(int af, int type, int protocol,
7492 LPWSAPROTOCOL_INFOA lpProtocolInfo,
7493 GROUP g, DWORD dwFlags)
7494 {
7495 INT len;
7496 WSAPROTOCOL_INFOW info;
7497
7498 TRACE("af=%d type=%d protocol=%d protocol_info=%p group=%d flags=0x%x\n",
7499 af, type, protocol, lpProtocolInfo, g, dwFlags);
7500
7501 if (!lpProtocolInfo) return WSASocketW(af, type, protocol, NULL, g, dwFlags);
7502
7503 memcpy(&info, lpProtocolInfo, FIELD_OFFSET(WSAPROTOCOL_INFOW, szProtocol));
7504 len = MultiByteToWideChar(CP_ACP, 0, lpProtocolInfo->szProtocol, -1,
7505 info.szProtocol, WSAPROTOCOL_LEN + 1);
7506
7507 if (!len)
7508 {
7509 SetLastError(WSAEINVAL);
7510 return SOCKET_ERROR;
7511 }
7512
7513 return WSASocketW(af, type, protocol, &info, g, dwFlags);
7514 }
7515
7516 /***********************************************************************
7517 * WSASocketW (WS2_32.79)
7518 *
7519 */
7520 SOCKET WINAPI WSASocketW(int af, int type, int protocol,
7521 LPWSAPROTOCOL_INFOW lpProtocolInfo,
7522 GROUP g, DWORD dwFlags)
7523 {
7524 SOCKET ret;
7525 DWORD err;
7526 int unixaf, unixtype, ipxptype = -1;
7527
7528 /*
7529 FIXME: The "advanced" parameters of WSASocketW (lpProtocolInfo,
7530 g, dwFlags except WSA_FLAG_OVERLAPPED) are ignored.
7531 */
7532
7533 TRACE("af=%d type=%d protocol=%d protocol_info=%p group=%d flags=0x%x\n",
7534 af, type, protocol, lpProtocolInfo, g, dwFlags );
7535
7536 if (!num_startup)
7537 {
7538 err = WSANOTINITIALISED;
7539 goto done;
7540 }
7541
7542 /* hack for WSADuplicateSocket */
7543 if (lpProtocolInfo && lpProtocolInfo->dwServiceFlags4 == 0xff00ff00) {
7544 ret = lpProtocolInfo->dwServiceFlags3;
7545 TRACE("\tgot duplicate %04lx\n", ret);
7546 return ret;
7547 }
7548
7549 if (lpProtocolInfo)
7550 {
7551 if (af == FROM_PROTOCOL_INFO || !af)
7552 af = lpProtocolInfo->iAddressFamily;
7553 if (type == FROM_PROTOCOL_INFO || !type)
7554 type = lpProtocolInfo->iSocketType;
7555 if (protocol == FROM_PROTOCOL_INFO || !protocol)
7556 protocol = lpProtocolInfo->iProtocol;
7557 }
7558
7559 if (!type && (af || protocol))
7560 {
7561 int autoproto = protocol;
7562 WSAPROTOCOL_INFOW infow;
7563
7564 /* default to the first valid protocol */
7565 if (!autoproto)
7566 autoproto = valid_protocols[0];
7567 else if(IS_IPX_PROTO(autoproto))
7568 autoproto = WS_NSPROTO_IPX;
7569
7570 if (WS_EnterSingleProtocolW(autoproto, &infow))
7571 {
7572 type = infow.iSocketType;
7573
7574 /* after win2003 it's no longer possible to pass AF_UNSPEC
7575 using the protocol info struct */
7576 if (!lpProtocolInfo && af == WS_AF_UNSPEC)
7577 af = infow.iAddressFamily;
7578 }
7579 }
7580
7581 /*
7582 Windows has an extension to the IPX protocol that allows one to create sockets
7583 and set the IPX packet type at the same time, to do that a caller will use
7584 a protocol like NSPROTO_IPX + <PACKET TYPE>
7585 */
7586 if (IS_IPX_PROTO(protocol))
7587 ipxptype = protocol - WS_NSPROTO_IPX;
7588
7589 /* convert the socket family, type and protocol */
7590 unixaf = convert_af_w2u(af);
7591 unixtype = convert_socktype_w2u(type);
7592 protocol = convert_proto_w2u(protocol);
7593 if (unixaf == AF_UNSPEC) unixaf = -1;
7594
7595 /* filter invalid parameters */
7596 if (protocol < 0)
7597 {
7598 /* the type could not be converted */
7599 if (type && unixtype < 0)
7600 {
7601 err = WSAESOCKTNOSUPPORT;
7602 goto done;
7603 }
7604
7605 err = WSAEPROTONOSUPPORT;
7606 goto done;
7607 }
7608 if (unixaf < 0)
7609 {
7610 /* both family and protocol can't be invalid */
7611 if (protocol <= 0)
7612 {
7613 err = WSAEINVAL;
7614 goto done;
7615 }
7616
7617 /* family could not be converted and neither socket type */
7618 if (unixtype < 0 && af >= 0)
7619 {
7620
7621 err = WSAESOCKTNOSUPPORT;
7622 goto done;
7623 }
7624
7625 err = WSAEAFNOSUPPORT;
7626 goto done;
7627 }
7628
7629 SERVER_START_REQ( create_socket )
7630 {
7631 req->family = unixaf;
7632 req->type = unixtype;
7633 req->protocol = protocol;
7634 req->access = GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE;
7635 req->attributes = (dwFlags & WSA_FLAG_NO_HANDLE_INHERIT) ? 0 : OBJ_INHERIT;
7636 req->flags = dwFlags & ~WSA_FLAG_NO_HANDLE_INHERIT;
7637 set_error( wine_server_call( req ) );
7638 ret = HANDLE2SOCKET( wine_server_ptr_handle( reply->handle ));
7639 }
7640 SERVER_END_REQ;
7641 if (ret)
7642 {
7643 TRACE("\tcreated %04lx\n", ret );
7644 if (ipxptype > 0)
7645 set_ipx_packettype(ret, ipxptype);
7646
7647 if (unixaf == AF_INET || unixaf == AF_INET6)
7648 {
7649 /* ensure IP_DONTFRAGMENT is disabled for SOCK_DGRAM and SOCK_RAW, enabled for SOCK_STREAM */
7650 if (unixtype == SOCK_DGRAM || unixtype == SOCK_RAW) /* in Linux the global default can be enabled */
7651 set_dont_fragment(ret, unixaf == AF_INET6 ? IPPROTO_IPV6 : IPPROTO_IP, FALSE);
7652 else if (unixtype == SOCK_STREAM)
7653 set_dont_fragment(ret, unixaf == AF_INET6 ? IPPROTO_IPV6 : IPPROTO_IP, TRUE);
7654 }
7655
7656 #ifdef IPV6_V6ONLY
7657 if (unixaf == AF_INET6)
7658 {
7659 int fd = get_sock_fd(ret, 0, NULL);
7660 if (fd != -1)
7661 {
7662 /* IPV6_V6ONLY is set by default on Windows */
7663 int enable = 1;
7664 if (setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, &enable, sizeof(enable)))
7665 WARN("\tsetting IPV6_V6ONLY failed - errno = %i\n", errno);
7666 release_sock_fd(ret, fd);
7667 }
7668 }
7669 #endif
7670 return ret;
7671 }
7672
7673 err = GetLastError();
7674 if (err == WSAEACCES) /* raw socket denied */
7675 {
7676 if (type == SOCK_RAW)
7677 ERR_(winediag)("Failed to create a socket of type SOCK_RAW, this requires special permissions.\n");
7678 else
7679 ERR_(winediag)("Failed to create socket, this requires special permissions.\n");
7680 }
7681 else
7682 {
7683 /* invalid combination of valid parameters, like SOCK_STREAM + IPPROTO_UDP */
7684 if (err == WSAEINVAL)
7685 err = WSAESOCKTNOSUPPORT;
7686 else if (err == WSAEOPNOTSUPP)
7687 err = WSAEPROTONOSUPPORT;
7688 }
7689
7690 done:
7691 WARN("\t\tfailed, error %d!\n", err);
7692 SetLastError(err);
7693 return INVALID_SOCKET;
7694 }
7695
7696 /***********************************************************************
7697 * WSAJoinLeaf (WS2_32.58)
7698 *
7699 */
7700 SOCKET WINAPI WSAJoinLeaf(
7701 SOCKET s,
7702 const struct WS_sockaddr *addr,
7703 int addrlen,
7704 LPWSABUF lpCallerData,
7705 LPWSABUF lpCalleeData,
7706 LPQOS lpSQOS,
7707 LPQOS lpGQOS,
7708 DWORD dwFlags)
7709 {
7710 FIXME("stub.\n");
7711 return INVALID_SOCKET;
7712 }
7713
7714 /***********************************************************************
7715 * __WSAFDIsSet (WS2_32.151)
7716 */
7717 int WINAPI __WSAFDIsSet(SOCKET s, WS_fd_set *set)
7718 {
7719 int i = set->fd_count, ret = 0;
7720
7721 while (i--)
7722 if (set->fd_array[i] == s)
7723 {
7724 ret = 1;
7725 break;
7726 }
7727
7728 TRACE("(socket %04lx, fd_set %p, count %i) <- %d\n", s, set, set->fd_count, ret);
7729 return ret;
7730 }
7731
7732 /***********************************************************************
7733 * WSAIsBlocking (WS2_32.114)
7734 */
7735 BOOL WINAPI WSAIsBlocking(void)
7736 {
7737 /* By default WinSock should set all its sockets to non-blocking mode
7738 * and poll in PeekMessage loop when processing "blocking" ones. This
7739 * function is supposed to tell if the program is in this loop. Our
7740 * blocking calls are truly blocking so we always return FALSE.
7741 *
7742 * Note: It is allowed to call this function without prior WSAStartup().
7743 */
7744
7745 TRACE("\n");
7746 return FALSE;
7747 }
7748
7749 /***********************************************************************
7750 * WSACancelBlockingCall (WS2_32.113)
7751 */
7752 INT WINAPI WSACancelBlockingCall(void)
7753 {
7754 TRACE("\n");
7755 return 0;
7756 }
7757
7758 static INT WINAPI WSA_DefaultBlockingHook( FARPROC x )
7759 {
7760 FIXME("How was this called?\n");
7761 return x();
7762 }
7763
7764
7765 /***********************************************************************
7766 * WSASetBlockingHook (WS2_32.109)
7767 */
7768 FARPROC WINAPI WSASetBlockingHook(FARPROC lpBlockFunc)
7769 {
7770 FARPROC prev = blocking_hook;
7771 blocking_hook = lpBlockFunc;
7772 TRACE("hook %p\n", lpBlockFunc);
7773 return prev;
7774 }
7775
7776
7777 /***********************************************************************
7778 * WSAUnhookBlockingHook (WS2_32.110)
7779 */
7780 INT WINAPI WSAUnhookBlockingHook(void)
7781 {
7782 blocking_hook = (FARPROC)WSA_DefaultBlockingHook;
7783 return 0;
7784 }
7785
7786
7787 /* ----------------------------------- end of API stuff */
7788
7789 /* ----------------------------------- helper functions -
7790 *
7791 * TODO: Merge WS_dup_..() stuff into one function that
7792 * would operate with a generic structure containing internal
7793 * pointers (via a template of some kind).
7794 */
7795
7796 static int list_size(char** l, int item_size)
7797 {
7798 int i,j = 0;
7799 if(l)
7800 { for(i=0;l[i];i++)
7801 j += (item_size) ? item_size : strlen(l[i]) + 1;
7802 j += (i + 1) * sizeof(char*); }
7803 return j;
7804 }
7805
7806 static int list_dup(char** l_src, char** l_to, int item_size)
7807 {
7808 char *p;
7809 int i;
7810
7811 for (i = 0; l_src[i]; i++) ;
7812 p = (char *)(l_to + i + 1);
7813 for (i = 0; l_src[i]; i++)
7814 {
7815 int count = ( item_size ) ? item_size : strlen(l_src[i]) + 1;
7816 memcpy(p, l_src[i], count);
7817 l_to[i] = p;
7818 p += count;
7819 }
7820 l_to[i] = NULL;
7821 return p - (char *)l_to;
7822 }
7823
7824 /* ----- hostent */
7825
7826 /* create a hostent entry
7827 *
7828 * Creates the entry with enough memory for the name, aliases
7829 * addresses, and the address pointers. Also copies the name
7830 * and sets up all the pointers.
7831 *
7832 * NOTE: The alias and address lists must be allocated with room
7833 * for the NULL item terminating the list. This is true even if
7834 * the list has no items ("aliases" and "addresses" must be
7835 * at least "1", a truly empty list is invalid).
7836 */
7837 static struct WS_hostent *WS_create_he(char *name, int aliases, int aliases_size, int addresses, int address_length)
7838 {
7839 struct WS_hostent *p_to;
7840 char *p;
7841 int size = (sizeof(struct WS_hostent) +
7842 strlen(name) + 1 +
7843 sizeof(char *) * aliases +
7844 aliases_size +
7845 sizeof(char *) * addresses +
7846 address_length * (addresses - 1)), i;
7847
7848 if (!(p_to = check_buffer_he(size))) return NULL;
7849 memset(p_to, 0, size);
7850
7851 /* Use the memory in the same way winsock does.
7852 * First set the pointer for aliases, second set the pointers for addresses.
7853 * Third fill the addresses indexes, fourth jump aliases names size.
7854 * Fifth fill the hostname.
7855 * NOTE: This method is valid for OS version's >= XP.
7856 */
7857 p = (char *)(p_to + 1);
7858 p_to->h_aliases = (char **)p;
7859 p += sizeof(char *)*aliases;
7860
7861 p_to->h_addr_list = (char **)p;
7862 p += sizeof(char *)*addresses;
7863
7864 for (i = 0, addresses--; i < addresses; i++, p += address_length)
7865 p_to->h_addr_list[i] = p;
7866
7867 /* NOTE: h_aliases must be filled in manually because we don't know each string
7868 * size, leave these pointers NULL (already set to NULL by memset earlier).
7869 */
7870 p += aliases_size;
7871
7872 p_to->h_name = p;
7873 strcpy(p, name);
7874
7875 return p_to;
7876 }
7877
7878 /* duplicate hostent entry
7879 * and handle all Win16/Win32 dependent things (struct size, ...) *correctly*.
7880 * Ditto for protoent and servent.
7881 */
7882 static struct WS_hostent *WS_dup_he(const struct hostent* p_he)
7883 {
7884 int i, addresses = 0, alias_size = 0;
7885 struct WS_hostent *p_to;
7886 char *p;
7887
7888 for( i = 0; p_he->h_aliases[i]; i++) alias_size += strlen(p_he->h_aliases[i]) + 1;
7889 while (p_he->h_addr_list[addresses]) addresses++;
7890
7891 p_to = WS_create_he(p_he->h_name, i + 1, alias_size, addresses + 1, p_he->h_length);
7892
7893 if (!p_to) return NULL;
7894 p_to->h_addrtype = convert_af_u2w(p_he->h_addrtype);
7895 p_to->h_length = p_he->h_length;
7896
7897 for(i = 0, p = p_to->h_addr_list[0]; p_he->h_addr_list[i]; i++, p += p_to->h_length)
7898 memcpy(p, p_he->h_addr_list[i], p_to->h_length);
7899
7900 /* Fill the aliases after the IP data */
7901 for(i = 0; p_he->h_aliases[i]; i++)
7902 {
7903 p_to->h_aliases[i] = p;
7904 strcpy(p, p_he->h_aliases[i]);
7905 p += strlen(p) + 1;
7906 }
7907
7908 return p_to;
7909 }
7910
7911 /* ----- protoent */
7912
7913 static struct WS_protoent *WS_create_pe( const char *name, char **aliases, int prot )
7914 {
7915 struct WS_protoent *ret;
7916 unsigned int size = sizeof(*ret) + strlen(name) + sizeof(char *) + list_size(aliases, 0);
7917
7918 if (!(ret = check_buffer_pe( size ))) return NULL;
7919 ret->p_proto = prot;
7920 ret->p_name = (char *)(ret + 1);
7921 strcpy( ret->p_name, name );
7922 ret->p_aliases = (char **)ret->p_name + strlen(name) / sizeof(char *) + 1;
7923 list_dup( aliases, ret->p_aliases, 0 );
7924 return ret;
7925 }
7926
7927 /* ----- servent */
7928
7929 static struct WS_servent *WS_dup_se(const struct servent* p_se)
7930 {
7931 char *p;
7932 struct WS_servent *p_to;
7933
7934 int size = (sizeof(*p_se) +
7935 strlen(p_se->s_proto) + 1 +
7936 strlen(p_se->s_name) + 1 +
7937 list_size(p_se->s_aliases, 0));
7938
7939 if (!(p_to = check_buffer_se(size))) return NULL;
7940 p_to->s_port = p_se->s_port;
7941
7942 p = (char *)(p_to + 1);
7943 p_to->s_name = p;
7944 strcpy(p, p_se->s_name);
7945 p += strlen(p) + 1;
7946
7947 p_to->s_proto = p;
7948 strcpy(p, p_se->s_proto);
7949 p += strlen(p) + 1;
7950
7951 p_to->s_aliases = (char **)p;
7952 list_dup(p_se->s_aliases, p_to->s_aliases, 0);
7953 return p_to;
7954 }
7955
7956
7957 /***********************************************************************
7958 * WSARecv (WS2_32.67)
7959 */
7960 int WINAPI WSARecv(SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
7961 LPDWORD NumberOfBytesReceived, LPDWORD lpFlags,
7962 LPWSAOVERLAPPED lpOverlapped,
7963 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine)
7964 {
7965 return WS2_recv_base(s, lpBuffers, dwBufferCount, NumberOfBytesReceived, lpFlags,
7966 NULL, NULL, lpOverlapped, lpCompletionRoutine, NULL);
7967 }
7968
7969 static int WS2_recv_base( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
7970 LPDWORD lpNumberOfBytesRecvd, LPDWORD lpFlags,
7971 struct WS_sockaddr *lpFrom,
7972 LPINT lpFromlen, LPWSAOVERLAPPED lpOverlapped,
7973 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine,
7974 LPWSABUF lpControlBuffer )
7975 {
7976 unsigned int i, options;
7977 int n, fd, err, overlapped, flags;
7978 struct ws2_async *wsa = NULL, localwsa;
7979 BOOL is_blocking;
7980 DWORD timeout_start = GetTickCount();
7981 ULONG_PTR cvalue = (lpOverlapped && ((ULONG_PTR)lpOverlapped->hEvent & 1) == 0) ? (ULONG_PTR)lpOverlapped : 0;
7982
7983 TRACE("socket %04lx, wsabuf %p, nbufs %d, flags %d, from %p, fromlen %d, ovl %p, func %p\n",
7984 s, lpBuffers, dwBufferCount, *lpFlags, lpFrom,
7985 (lpFromlen ? *lpFromlen : -1),
7986 lpOverlapped, lpCompletionRoutine);
7987
7988 fd = get_sock_fd( s, FILE_READ_DATA, &options );
7989 TRACE( "fd=%d, options=%x\n", fd, options );
7990
7991 if (fd == -1) return SOCKET_ERROR;
7992
7993 if (*lpFlags & WS_MSG_OOB)
7994 {
7995 /* It's invalid to receive OOB data from an OOBINLINED socket
7996 * as OOB data is turned into normal data. */
7997 socklen_t len = sizeof(n);
7998 if (!getsockopt(fd, SOL_SOCKET, SO_OOBINLINE, (char*) &n, &len) && n)
7999 {
8000 err = WSAEINVAL;
8001 goto error;
8002 }
8003 }
8004
8005 overlapped = (lpOverlapped || lpCompletionRoutine) &&
8006 !(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT));
8007 if (overlapped || dwBufferCount > 1)
8008 {
8009 if (!(wsa = (struct ws2_async *)alloc_async_io( offsetof(struct ws2_async, iovec[dwBufferCount]),
8010 WS2_async_recv )))
8011 {
8012 err = WSAEFAULT;
8013 goto error;
8014 }
8015 }
8016 else
8017 wsa = &localwsa;
8018
8019 wsa->hSocket = SOCKET2HANDLE(s);
8020 wsa->flags = *lpFlags;
8021 wsa->lpFlags = lpFlags;
8022 wsa->addr = lpFrom;
8023 wsa->addrlen.ptr = lpFromlen;
8024 wsa->control = lpControlBuffer;
8025 wsa->n_iovecs = dwBufferCount;
8026 wsa->first_iovec = 0;
8027 for (i = 0; i < dwBufferCount; i++)
8028 {
8029 /* check buffer first to trigger write watches */
8030 if (IsBadWritePtr( lpBuffers[i].buf, lpBuffers[i].len ))
8031 {
8032 err = WSAEFAULT;
8033 goto error;
8034 }
8035 wsa->iovec[i].iov_base = lpBuffers[i].buf;
8036 wsa->iovec[i].iov_len = lpBuffers[i].len;
8037 }
8038
8039 flags = convert_flags(wsa->flags);
8040 for (;;)
8041 {
8042 n = WS2_recv( fd, wsa, flags );
8043 if (n == -1)
8044 {
8045 /* Unix-like systems return EINVAL when attempting to read OOB data from
8046 * an empty socket buffer, convert that to a Windows expected return. */
8047 if ((flags & MSG_OOB) && errno == EINVAL)
8048 errno = EWOULDBLOCK;
8049
8050 if (errno != EAGAIN)
8051 {
8052 err = wsaErrno();
8053 goto error;
8054 }
8055 }
8056 else if (lpNumberOfBytesRecvd) *lpNumberOfBytesRecvd = n;
8057
8058 if (overlapped)
8059 {
8060 IO_STATUS_BLOCK *iosb = lpOverlapped ? (IO_STATUS_BLOCK *)lpOverlapped : &wsa->local_iosb;
8061
8062 wsa->user_overlapped = lpOverlapped;
8063 wsa->completion_func = lpCompletionRoutine;
8064 release_sock_fd( s, fd );
8065
8066 if (n == -1)
8067 {
8068 iosb->u.Status = STATUS_PENDING;
8069 iosb->Information = 0;
8070
8071 if (wsa->completion_func)
8072 err = register_async( ASYNC_TYPE_READ, wsa->hSocket, &wsa->io, NULL,
8073 ws2_async_apc, wsa, iosb );
8074 else
8075 err = register_async( ASYNC_TYPE_READ, wsa->hSocket, &wsa->io, lpOverlapped->hEvent,
8076 NULL, (void *)cvalue, iosb );
8077
8078 if (err != STATUS_PENDING) HeapFree( GetProcessHeap(), 0, wsa );
8079 SetLastError(NtStatusToWSAError( err ));
8080 return SOCKET_ERROR;
8081 }
8082
8083 iosb->u.Status = STATUS_SUCCESS;
8084 iosb->Information = n;
8085 if (!wsa->completion_func)
8086 {
8087 if (cvalue) WS_AddCompletion( s, cvalue, STATUS_SUCCESS, n, FALSE );
8088 if (lpOverlapped->hEvent) SetEvent( lpOverlapped->hEvent );
8089 HeapFree( GetProcessHeap(), 0, wsa );
8090 }
8091 else NtQueueApcThread( GetCurrentThread(), (PNTAPCFUNC)ws2_async_apc,
8092 (ULONG_PTR)wsa, (ULONG_PTR)iosb, 0 );
8093 _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
8094 return 0;
8095 }
8096
8097 if (n != -1) break;
8098
8099 if ((err = _is_blocking( s, &is_blocking )))
8100 {
8101 err = NtStatusToWSAError( err );
8102 goto error;
8103 }
8104
8105 if ( is_blocking )
8106 {
8107 struct pollfd pfd;
8108 int poll_timeout = -1;
8109 INT64 timeout = get_rcvsnd_timeo(fd, TRUE);
8110
8111 if (timeout)
8112 {
8113 timeout -= GetTickCount() - timeout_start;
8114 if (timeout < 0) poll_timeout = 0;
8115 else poll_timeout = timeout <= INT_MAX ? timeout : INT_MAX;
8116 }
8117
8118 pfd.fd = fd;
8119 pfd.events = POLLIN;
8120 if (*lpFlags & WS_MSG_OOB) pfd.events |= POLLPRI;
8121
8122 if (!poll_timeout || !poll( &pfd, 1, poll_timeout ))
8123 {
8124 err = WSAETIMEDOUT;
8125 /* a timeout is not fatal */
8126 _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
8127 goto error;
8128 }
8129 }
8130 else
8131 {
8132 _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
8133 err = WSAEWOULDBLOCK;
8134 goto error;
8135 }
8136 }
8137
8138 TRACE(" -> %i bytes\n", n);
8139 if (wsa != &localwsa) HeapFree( GetProcessHeap(), 0, wsa );
8140 release_sock_fd( s, fd );
8141 _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
8142 SetLastError(ERROR_SUCCESS);
8143
8144 return 0;
8145
8146 error:
8147 if (wsa != &localwsa) HeapFree( GetProcessHeap(), 0, wsa );
8148 release_sock_fd( s, fd );
8149 WARN(" -> ERROR %d\n", err);
8150 SetLastError( err );
8151 return SOCKET_ERROR;
8152 }
8153
8154 /***********************************************************************
8155 * WSARecvFrom (WS2_32.69)
8156 */
8157 INT WINAPI WSARecvFrom( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
8158 LPDWORD lpNumberOfBytesRecvd, LPDWORD lpFlags, struct WS_sockaddr *lpFrom,
8159 LPINT lpFromlen, LPWSAOVERLAPPED lpOverlapped,
8160 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
8161
8162 {
8163 return WS2_recv_base( s, lpBuffers, dwBufferCount,
8164 lpNumberOfBytesRecvd, lpFlags,
8165 lpFrom, lpFromlen,
8166 lpOverlapped, lpCompletionRoutine, NULL );
8167 }
8168
8169 /***********************************************************************
8170 * WSCInstallProvider (WS2_32.88)
8171 */
8172 INT WINAPI WSCInstallProvider( const LPGUID lpProviderId,
8173 LPCWSTR lpszProviderDllPath,
8174 const LPWSAPROTOCOL_INFOW lpProtocolInfoList,
8175 DWORD dwNumberOfEntries,
8176 LPINT lpErrno )
8177 {
8178 FIXME("(%s, %s, %p, %d, %p): stub !\n", debugstr_guid(lpProviderId),
8179 debugstr_w(lpszProviderDllPath), lpProtocolInfoList,
8180 dwNumberOfEntries, lpErrno);
8181 *lpErrno = 0;
8182 return 0;
8183 }
8184
8185
8186 /***********************************************************************
8187 * WSCDeinstallProvider (WS2_32.83)
8188 */
8189 INT WINAPI WSCDeinstallProvider(LPGUID lpProviderId, LPINT lpErrno)
8190 {
8191 FIXME("(%s, %p): stub !\n", debugstr_guid(lpProviderId), lpErrno);
8192 *lpErrno = 0;
8193 return 0;
8194 }
8195
8196
8197 /***********************************************************************
8198 * WSAAccept (WS2_32.26)
8199 */
8200 SOCKET WINAPI WSAAccept( SOCKET s, struct WS_sockaddr *addr, LPINT addrlen,
8201 LPCONDITIONPROC lpfnCondition, DWORD_PTR dwCallbackData)
8202 {
8203
8204 int ret = 0, size;
8205 WSABUF CallerId, CallerData, CalleeId, CalleeData;
8206 /* QOS SQOS, GQOS; */
8207 GROUP g;
8208 SOCKET cs;
8209 SOCKADDR src_addr, dst_addr;
8210
8211 TRACE("socket %04lx, sockaddr %p, addrlen %p, fnCondition %p, dwCallbackData %ld\n",
8212 s, addr, addrlen, lpfnCondition, dwCallbackData);
8213
8214 cs = WS_accept(s, addr, addrlen);
8215 if (cs == SOCKET_ERROR) return SOCKET_ERROR;
8216 if (!lpfnCondition) return cs;
8217
8218 if (addr && addrlen)
8219 {
8220 CallerId.buf = (char *)addr;
8221 CallerId.len = *addrlen;
8222 }
8223 else
8224 {
8225 size = sizeof(src_addr);
8226 WS_getpeername(cs, &src_addr, &size);
8227 CallerId.buf = (char *)&src_addr;
8228 CallerId.len = size;
8229 }
8230 CallerData.buf = NULL;
8231 CallerData.len = 0;
8232
8233 size = sizeof(dst_addr);
8234 WS_getsockname(cs, &dst_addr, &size);
8235
8236 CalleeId.buf = (char *)&dst_addr;
8237 CalleeId.len = sizeof(dst_addr);
8238
8239 ret = (*lpfnCondition)(&CallerId, &CallerData, NULL, NULL,
8240 &CalleeId, &CalleeData, &g, dwCallbackData);
8241
8242 switch (ret)
8243 {
8244 case CF_ACCEPT:
8245 return cs;
8246 case CF_DEFER:
8247 SERVER_START_REQ( set_socket_deferred )
8248 {
8249 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
8250 req->deferred = wine_server_obj_handle( SOCKET2HANDLE(cs) );
8251 if ( !wine_server_call_err ( req ) )
8252 {
8253 SetLastError( WSATRY_AGAIN );
8254 WS_closesocket( cs );
8255 }
8256 }
8257 SERVER_END_REQ;
8258 return SOCKET_ERROR;
8259 case CF_REJECT:
8260 WS_closesocket(cs);
8261 SetLastError(WSAECONNREFUSED);
8262 return SOCKET_ERROR;
8263 default:
8264 FIXME("Unknown return type from Condition function\n");
8265 SetLastError(WSAENOTSOCK);
8266 return SOCKET_ERROR;
8267 }
8268 }
8269
8270 /***********************************************************************
8271 * WSADuplicateSocketA (WS2_32.32)
8272 */
8273 int WINAPI WSADuplicateSocketA( SOCKET s, DWORD dwProcessId, LPWSAPROTOCOL_INFOA lpProtocolInfo )
8274 {
8275 return WS_DuplicateSocket(FALSE, s, dwProcessId, (LPWSAPROTOCOL_INFOW) lpProtocolInfo);
8276 }
8277
8278 /***********************************************************************
8279 * WSADuplicateSocketW (WS2_32.33)
8280 */
8281 int WINAPI WSADuplicateSocketW( SOCKET s, DWORD dwProcessId, LPWSAPROTOCOL_INFOW lpProtocolInfo )
8282 {
8283 return WS_DuplicateSocket(TRUE, s, dwProcessId, lpProtocolInfo);
8284 }
8285
8286 /***********************************************************************
8287 * WSAInstallServiceClassA (WS2_32.48)
8288 */
8289 int WINAPI WSAInstallServiceClassA(LPWSASERVICECLASSINFOA info)
8290 {
8291 FIXME("Request to install service %s\n",debugstr_a(info->lpszServiceClassName));
8292 SetLastError(WSAEACCES);
8293 return SOCKET_ERROR;
8294 }
8295
8296 /***********************************************************************
8297 * WSAInstallServiceClassW (WS2_32.49)
8298 */
8299 int WINAPI WSAInstallServiceClassW(LPWSASERVICECLASSINFOW info)
8300 {
8301 FIXME("Request to install service %s\n",debugstr_w(info->lpszServiceClassName));
8302 SetLastError(WSAEACCES);
8303 return SOCKET_ERROR;
8304 }
8305
8306 /***********************************************************************
8307 * WSARemoveServiceClass (WS2_32.70)
8308 */
8309 int WINAPI WSARemoveServiceClass(LPGUID info)
8310 {
8311 FIXME("Request to remove service %s\n", debugstr_guid(info));
8312 SetLastError(WSATYPE_NOT_FOUND);
8313 return SOCKET_ERROR;
8314 }
8315
8316 /***********************************************************************
8317 * inet_ntop (WS2_32.@)
8318 */
8319 PCSTR WINAPI WS_inet_ntop( INT family, PVOID addr, PSTR buffer, SIZE_T len )
8320 {
8321 #ifdef HAVE_INET_NTOP
8322 struct WS_in6_addr *in6;
8323 struct WS_in_addr *in;
8324 PCSTR pdst;
8325
8326 TRACE("family %d, addr (%p), buffer (%p), len %ld\n", family, addr, buffer, len);
8327 if (!buffer)
8328 {
8329 SetLastError( STATUS_INVALID_PARAMETER );
8330 return NULL;
8331 }
8332
8333 switch (family)
8334 {
8335 case WS_AF_INET:
8336 {
8337 in = addr;
8338 pdst = inet_ntop( AF_INET, &in->WS_s_addr, buffer, len );
8339 break;
8340 }
8341 case WS_AF_INET6:
8342 {
8343 in6 = addr;
8344 pdst = inet_ntop( AF_INET6, in6->WS_s6_addr, buffer, len );
8345 break;
8346 }
8347 default:
8348 SetLastError( WSAEAFNOSUPPORT );
8349 return NULL;
8350 }
8351
8352 if (!pdst) SetLastError( STATUS_INVALID_PARAMETER );
8353 return pdst;
8354 #else
8355 FIXME( "not supported on this platform\n" );
8356 SetLastError( WSAEAFNOSUPPORT );
8357 return NULL;
8358 #endif
8359 }
8360
8361 /***********************************************************************
8362 * inet_pton (WS2_32.@)
8363 */
8364 INT WINAPI WS_inet_pton( INT family, PCSTR addr, PVOID buffer)
8365 {
8366 #ifdef HAVE_INET_PTON
8367 int unixaf, ret;
8368
8369 TRACE("family %d, addr %s, buffer (%p)\n", family, debugstr_a(addr), buffer);
8370
8371 if (!addr || !buffer)
8372 {
8373 SetLastError(WSAEFAULT);
8374 return SOCKET_ERROR;
8375 }
8376
8377 unixaf = convert_af_w2u(family);
8378 if (unixaf != AF_INET && unixaf != AF_INET6)
8379 {
8380 SetLastError(WSAEAFNOSUPPORT);
8381 return SOCKET_ERROR;
8382 }
8383
8384 ret = inet_pton(unixaf, addr, buffer);
8385 if (ret == -1) SetLastError(wsaErrno());
8386 return ret;
8387 #else
8388 FIXME( "not supported on this platform\n" );
8389 SetLastError( WSAEAFNOSUPPORT );
8390 return SOCKET_ERROR;
8391 #endif
8392 }
8393
8394 /***********************************************************************
8395 * InetPtonW (WS2_32.@)
8396 */
8397 INT WINAPI InetPtonW(INT family, PCWSTR addr, PVOID buffer)
8398 {
8399 char *addrA;
8400 int len;
8401 INT ret;
8402
8403 TRACE("family %d, addr %s, buffer (%p)\n", family, debugstr_w(addr), buffer);
8404
8405 if (!addr)
8406 {
8407 SetLastError(WSAEFAULT);
8408 return SOCKET_ERROR;
8409 }
8410
8411 len = WideCharToMultiByte(CP_ACP, 0, addr, -1, NULL, 0, NULL, NULL);
8412 if (!(addrA = HeapAlloc(GetProcessHeap(), 0, len)))
8413 {
8414 SetLastError(WSA_NOT_ENOUGH_MEMORY);
8415 return SOCKET_ERROR;
8416 }
8417 WideCharToMultiByte(CP_ACP, 0, addr, -1, addrA, len, NULL, NULL);
8418
8419 ret = WS_inet_pton(family, addrA, buffer);
8420
8421 HeapFree(GetProcessHeap(), 0, addrA);
8422 return ret;
8423 }
8424
8425 /***********************************************************************
8426 * InetNtopW (WS2_32.@)
8427 */
8428 PCWSTR WINAPI InetNtopW(INT family, PVOID addr, PWSTR buffer, SIZE_T len)
8429 {
8430 char bufferA[WS_INET6_ADDRSTRLEN];
8431 PWSTR ret = NULL;
8432
8433 TRACE("family %d, addr (%p), buffer (%p), len %ld\n", family, addr, buffer, len);
8434
8435 if (WS_inet_ntop(family, addr, bufferA, sizeof(bufferA)))
8436 {
8437 if (MultiByteToWideChar(CP_ACP, 0, bufferA, -1, buffer, len))
8438 ret = buffer;
8439 else
8440 SetLastError(ERROR_INVALID_PARAMETER);
8441 }
8442 return ret;
8443 }
8444
8445 /***********************************************************************
8446 * WSAStringToAddressA (WS2_32.80)
8447 */
8448 INT WINAPI WSAStringToAddressA(LPSTR AddressString,
8449 INT AddressFamily,
8450 LPWSAPROTOCOL_INFOA lpProtocolInfo,
8451 LPSOCKADDR lpAddress,
8452 LPINT lpAddressLength)
8453 {
8454 INT res=0;
8455 LPSTR workBuffer=NULL,ptrPort;
8456
8457 TRACE( "(%s, %x, %p, %p, %p)\n", debugstr_a(AddressString), AddressFamily,
8458 lpProtocolInfo, lpAddress, lpAddressLength );
8459
8460 if (!lpAddressLength || !lpAddress) return SOCKET_ERROR;
8461
8462 if (!AddressString)
8463 {
8464 SetLastError(WSAEINVAL);
8465 return SOCKET_ERROR;
8466 }
8467
8468 if (lpProtocolInfo)
8469 FIXME("ProtocolInfo not implemented.\n");
8470
8471 workBuffer = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
8472 strlen(AddressString) + 1);
8473 if (!workBuffer)
8474 {
8475 SetLastError(WSA_NOT_ENOUGH_MEMORY);
8476 return SOCKET_ERROR;
8477 }
8478
8479 strcpy(workBuffer, AddressString);
8480
8481 switch(AddressFamily)
8482 {
8483 case WS_AF_INET:
8484 {
8485 struct in_addr inetaddr;
8486
8487 /* If lpAddressLength is too small, tell caller the size we need */
8488 if (*lpAddressLength < sizeof(SOCKADDR_IN))
8489 {
8490 *lpAddressLength = sizeof(SOCKADDR_IN);
8491 res = WSAEFAULT;
8492 break;
8493 }
8494 *lpAddressLength = sizeof(SOCKADDR_IN);
8495 memset(lpAddress, 0, sizeof(SOCKADDR_IN));
8496
8497 ((LPSOCKADDR_IN)lpAddress)->sin_family = WS_AF_INET;
8498
8499 ptrPort = strchr(workBuffer, ':');
8500 if(ptrPort)
8501 {
8502 /* User may have entered an IPv6 and asked to parse as IPv4 */
8503 if(strchr(ptrPort + 1, ':'))
8504 {
8505 res = WSAEINVAL;
8506 break;
8507 }
8508 ((LPSOCKADDR_IN)lpAddress)->sin_port = htons(atoi(ptrPort+1));
8509 *ptrPort = '\0';
8510 }
8511
8512 if(inet_aton(workBuffer, &inetaddr) > 0)
8513 {
8514 ((LPSOCKADDR_IN)lpAddress)->sin_addr.WS_s_addr = inetaddr.s_addr;
8515 res = 0;
8516 }
8517 else
8518 res = WSAEINVAL;
8519
8520 break;
8521 }
8522 case WS_AF_INET6:
8523 {
8524 struct in6_addr inetaddr;
8525 char *ptrAddr = workBuffer;
8526
8527 /* If lpAddressLength is too small, tell caller the size we need */
8528 if (*lpAddressLength < sizeof(SOCKADDR_IN6))
8529 {
8530 *lpAddressLength = sizeof(SOCKADDR_IN6);
8531 res = WSAEFAULT;
8532 break;
8533 }
8534 #ifdef HAVE_INET_PTON
8535 *lpAddressLength = sizeof(SOCKADDR_IN6);
8536 memset(lpAddress, 0, sizeof(SOCKADDR_IN6));
8537
8538 ((LPSOCKADDR_IN6)lpAddress)->sin6_family = WS_AF_INET6;
8539
8540 /* Valid IPv6 addresses can also be surrounded by [ ], and in this case
8541 * a port number may follow after like in [fd12:3456:7890::1]:12345
8542 * We need to cut the brackets and find the port if any. */
8543
8544 if(*workBuffer == '[')
8545 {
8546 ptrPort = strchr(workBuffer, ']');
8547 if (!ptrPort)
8548 {
8549 SetLastError(WSAEINVAL);
8550 return SOCKET_ERROR;
8551 }
8552
8553 if (ptrPort[1] == ':')
8554 ((LPSOCKADDR_IN6)lpAddress)->sin6_port = htons(atoi(ptrPort + 2));
8555
8556 *ptrPort = '\0';
8557 ptrAddr = workBuffer + 1;
8558 }
8559
8560 if(inet_pton(AF_INET6, ptrAddr, &inetaddr) > 0)
8561 {
8562 memcpy(&((LPSOCKADDR_IN6)lpAddress)->sin6_addr, &inetaddr,
8563 sizeof(struct in6_addr));
8564 res = 0;
8565 }
8566 else
8567 #endif /* HAVE_INET_PTON */
8568 res = WSAEINVAL;
8569
8570 break;
8571 }
8572 default:
8573 /* According to MSDN, only AF_INET and AF_INET6 are supported. */
8574 TRACE("Unsupported address family specified: %d.\n", AddressFamily);
8575 res = WSAEINVAL;
8576 }
8577
8578 HeapFree(GetProcessHeap(), 0, workBuffer);
8579
8580 if (!res) return 0;
8581 SetLastError(res);
8582 return SOCKET_ERROR;
8583 }
8584
8585 /***********************************************************************
8586 * WSAStringToAddressW (WS2_32.81)
8587 *
8588 * FIXME: Does anybody know if this function allows using Hebrew/Arabic/Chinese... digits?
8589 * If this should be the case, it would be required to map these digits
8590 * to Unicode digits (0-9) using FoldString first.
8591 */
8592 INT WINAPI WSAStringToAddressW(LPWSTR AddressString,
8593 INT AddressFamily,
8594 LPWSAPROTOCOL_INFOW lpProtocolInfo,
8595 LPSOCKADDR lpAddress,
8596 LPINT lpAddressLength)
8597 {
8598 INT sBuffer,res=0;
8599 LPSTR workBuffer=NULL;
8600 WSAPROTOCOL_INFOA infoA;
8601 LPWSAPROTOCOL_INFOA lpProtoInfoA = NULL;
8602
8603 TRACE( "(%s, %x, %p, %p, %p)\n", debugstr_w(AddressString), AddressFamily, lpProtocolInfo,
8604 lpAddress, lpAddressLength );
8605
8606 if (!lpAddressLength || !lpAddress) return SOCKET_ERROR;
8607
8608 /* if ProtocolInfo is available - convert to ANSI variant */
8609 if (lpProtocolInfo)
8610 {
8611 lpProtoInfoA = &infoA;
8612 memcpy( lpProtoInfoA, lpProtocolInfo, FIELD_OFFSET( WSAPROTOCOL_INFOA, szProtocol ) );
8613
8614 if (!WideCharToMultiByte( CP_ACP, 0, lpProtocolInfo->szProtocol, -1,
8615 lpProtoInfoA->szProtocol, WSAPROTOCOL_LEN+1, NULL, NULL ))
8616 {
8617 SetLastError(WSAEINVAL);
8618 return SOCKET_ERROR;
8619 }
8620 }
8621
8622 if (AddressString)
8623 {
8624 /* Translate AddressString to ANSI code page - assumes that only
8625 standard digits 0-9 are used with this API call */
8626 sBuffer = WideCharToMultiByte( CP_ACP, 0, AddressString, -1, NULL, 0, NULL, NULL );
8627 workBuffer = HeapAlloc( GetProcessHeap(), 0, sBuffer );
8628
8629 if (workBuffer)
8630 {
8631 WideCharToMultiByte( CP_ACP, 0, AddressString, -1, workBuffer, sBuffer, NULL, NULL );
8632 res = WSAStringToAddressA(workBuffer,AddressFamily,lpProtoInfoA,
8633 lpAddress,lpAddressLength);
8634 HeapFree( GetProcessHeap(), 0, workBuffer );
8635 return res;
8636 }
8637 else
8638 res = WSA_NOT_ENOUGH_MEMORY;
8639 }
8640 else
8641 res = WSAEINVAL;
8642
8643 SetLastError(res);
8644 return SOCKET_ERROR;
8645 }
8646
8647 /***********************************************************************
8648 * WSAAddressToStringA (WS2_32.27)
8649 *
8650 * See WSAAddressToStringW
8651 */
8652 INT WINAPI WSAAddressToStringA( LPSOCKADDR sockaddr, DWORD len,
8653 LPWSAPROTOCOL_INFOA info, LPSTR string,
8654 LPDWORD lenstr )
8655 {
8656 DWORD size;
8657 CHAR buffer[54]; /* 32 digits + 7':' + '[' + '%" + 5 digits + ']:' + 5 digits + '\0' */
8658 CHAR *p;
8659
8660 TRACE( "(%p, %d, %p, %p, %p)\n", sockaddr, len, info, string, lenstr );
8661
8662 if (!sockaddr) return SOCKET_ERROR;
8663 if (!string || !lenstr) return SOCKET_ERROR;
8664
8665 switch(sockaddr->sa_family)
8666 {
8667 case WS_AF_INET:
8668 {
8669 unsigned int long_ip = ntohl(((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr);
8670 if (len < sizeof(SOCKADDR_IN)) return SOCKET_ERROR;
8671 sprintf( buffer, "%u.%u.%u.%u:%u",
8672 (long_ip >> 24) & 0xff,
8673 (long_ip >> 16) & 0xff,
8674 (long_ip >> 8) & 0xff,
8675 long_ip & 0xff,
8676 ntohs( ((SOCKADDR_IN *)sockaddr)->sin_port ) );
8677
8678 p = strchr( buffer, ':' );
8679 if (!((SOCKADDR_IN *)sockaddr)->sin_port) *p = 0;
8680 break;
8681 }
8682 case WS_AF_INET6:
8683 {
8684 struct WS_sockaddr_in6 *sockaddr6 = (LPSOCKADDR_IN6) sockaddr;
8685 size_t slen;
8686
8687 buffer[0] = 0;
8688 if (len < sizeof(SOCKADDR_IN6)) return SOCKET_ERROR;
8689 if ((sockaddr6->sin6_port))
8690 strcpy(buffer, "[");
8691 slen = strlen(buffer);
8692 if (!WS_inet_ntop(WS_AF_INET6, &sockaddr6->sin6_addr, &buffer[slen], sizeof(buffer) - slen))
8693 {
8694 SetLastError(WSAEINVAL);
8695 return SOCKET_ERROR;
8696 }
8697 if ((sockaddr6->sin6_scope_id))
8698 sprintf(buffer+strlen(buffer), "%%%u", sockaddr6->sin6_scope_id);
8699 if ((sockaddr6->sin6_port))
8700 sprintf(buffer+strlen(buffer), "]:%u", ntohs(sockaddr6->sin6_port));
8701 break;
8702 }
8703
8704 default:
8705 SetLastError(WSAEINVAL);
8706 return SOCKET_ERROR;
8707 }
8708
8709 size = strlen( buffer ) + 1;
8710
8711 if (*lenstr < size)
8712 {
8713 *lenstr = size;
8714 SetLastError(WSAEFAULT);
8715 return SOCKET_ERROR;
8716 }
8717
8718 TRACE("=> %s,%u bytes\n", debugstr_a(buffer), size);
8719 *lenstr = size;
8720 strcpy( string, buffer );
8721 return 0;
8722 }
8723
8724 /***********************************************************************
8725 * WSAAddressToStringW (WS2_32.28)
8726 *
8727 * Convert a sockaddr address into a readable address string.
8728 *
8729 * PARAMS
8730 * sockaddr [I] Pointer to a sockaddr structure.
8731 * len [I] Size of the sockaddr structure.
8732 * info [I] Pointer to a WSAPROTOCOL_INFOW structure (optional).
8733 * string [I/O] Pointer to a buffer to receive the address string.
8734 * lenstr [I/O] Size of the receive buffer in WCHARs.
8735 *
8736 * RETURNS
8737 * Success: 0
8738 * Failure: SOCKET_ERROR
8739 *
8740 * NOTES
8741 * The 'info' parameter is ignored.
8742 */
8743 INT WINAPI WSAAddressToStringW( LPSOCKADDR sockaddr, DWORD len,
8744 LPWSAPROTOCOL_INFOW info, LPWSTR string,
8745 LPDWORD lenstr )
8746 {
8747 INT ret;
8748 DWORD size;
8749 WCHAR buffer[54]; /* 32 digits + 7':' + '[' + '%" + 5 digits + ']:' + 5 digits + '\0' */
8750 CHAR bufAddr[54];
8751
8752 TRACE( "(%p, %d, %p, %p, %p)\n", sockaddr, len, info, string, lenstr );
8753
8754 size = *lenstr;
8755 ret = WSAAddressToStringA(sockaddr, len, NULL, bufAddr, &size);
8756
8757 if (ret) return ret;
8758
8759 MultiByteToWideChar(CP_ACP, 0, bufAddr, size, buffer, ARRAY_SIZE(buffer));
8760
8761 if (*lenstr < size)
8762 {
8763 *lenstr = size;
8764 SetLastError(WSAEFAULT);
8765 return SOCKET_ERROR;
8766 }
8767
8768 TRACE("=> %s,%u bytes\n", debugstr_w(buffer), size);
8769 *lenstr = size;
8770 lstrcpyW( string, buffer );
8771 return 0;
8772 }
8773
8774 /***********************************************************************
8775 * WSAEnumNameSpaceProvidersA (WS2_32.34)
8776 */
8777 INT WINAPI WSAEnumNameSpaceProvidersA( LPDWORD len, LPWSANAMESPACE_INFOA buffer )
8778 {
8779 FIXME( "(%p %p) Stub!\n", len, buffer );
8780 return 0;
8781 }
8782
8783 /***********************************************************************
8784 * WSAEnumNameSpaceProvidersW (WS2_32.35)
8785 */
8786 INT WINAPI WSAEnumNameSpaceProvidersW( LPDWORD len, LPWSANAMESPACE_INFOW buffer )
8787 {
8788 FIXME( "(%p %p) Stub!\n", len, buffer );
8789 return 0;
8790 }
8791
8792 /***********************************************************************
8793 * WSAGetQOSByName (WS2_32.41)
8794 */
8795 BOOL WINAPI WSAGetQOSByName( SOCKET s, LPWSABUF lpQOSName, LPQOS lpQOS )
8796 {
8797 FIXME( "(0x%04lx %p %p) Stub!\n", s, lpQOSName, lpQOS );
8798 return FALSE;
8799 }
8800
8801 /***********************************************************************
8802 * WSAGetServiceClassInfoA (WS2_32.42)
8803 */
8804 INT WINAPI WSAGetServiceClassInfoA( LPGUID provider, LPGUID service, LPDWORD len,
8805 LPWSASERVICECLASSINFOA info )
8806 {
8807 FIXME( "(%s %s %p %p) Stub!\n", debugstr_guid(provider), debugstr_guid(service),
8808 len, info );
8809 SetLastError(WSA_NOT_ENOUGH_MEMORY);
8810 return SOCKET_ERROR;
8811 }
8812
8813 /***********************************************************************
8814 * WSAGetServiceClassInfoW (WS2_32.43)
8815 */
8816 INT WINAPI WSAGetServiceClassInfoW( LPGUID provider, LPGUID service, LPDWORD len,
8817 LPWSASERVICECLASSINFOW info )
8818 {
8819 FIXME( "(%s %s %p %p) Stub!\n", debugstr_guid(provider), debugstr_guid(service),
8820 len, info );
8821 SetLastError(WSA_NOT_ENOUGH_MEMORY);
8822 return SOCKET_ERROR;
8823 }
8824
8825 /***********************************************************************
8826 * WSAGetServiceClassNameByClassIdA (WS2_32.44)
8827 */
8828 INT WINAPI WSAGetServiceClassNameByClassIdA( LPGUID class, LPSTR service, LPDWORD len )
8829 {
8830 FIXME( "(%s %p %p) Stub!\n", debugstr_guid(class), service, len );
8831 SetLastError(WSA_NOT_ENOUGH_MEMORY);
8832 return SOCKET_ERROR;
8833 }
8834
8835 /***********************************************************************
8836 * WSAGetServiceClassNameByClassIdW (WS2_32.45)
8837 */
8838 INT WINAPI WSAGetServiceClassNameByClassIdW( LPGUID class, LPWSTR service, LPDWORD len )
8839 {
8840 FIXME( "(%s %p %p) Stub!\n", debugstr_guid(class), service, len );
8841 SetLastError(WSA_NOT_ENOUGH_MEMORY);
8842 return SOCKET_ERROR;
8843 }
8844
8845 /***********************************************************************
8846 * WSALookupServiceBeginA (WS2_32.59)
8847 */
8848 INT WINAPI WSALookupServiceBeginA( LPWSAQUERYSETA lpqsRestrictions,
8849 DWORD dwControlFlags,
8850 LPHANDLE lphLookup)
8851 {
8852 FIXME("(%p 0x%08x %p) Stub!\n", lpqsRestrictions, dwControlFlags,
8853 lphLookup);
8854 SetLastError(WSA_NOT_ENOUGH_MEMORY);
8855 return SOCKET_ERROR;
8856 }
8857
8858 /***********************************************************************
8859 * WSALookupServiceBeginW (WS2_32.60)
8860 */
8861 INT WINAPI WSALookupServiceBeginW( LPWSAQUERYSETW lpqsRestrictions,
8862 DWORD dwControlFlags,
8863 LPHANDLE lphLookup)
8864 {
8865 FIXME("(%p 0x%08x %p) Stub!\n", lpqsRestrictions, dwControlFlags,
8866 lphLookup);
8867 SetLastError(WSA_NOT_ENOUGH_MEMORY);
8868 return SOCKET_ERROR;
8869 }
8870
8871 /***********************************************************************
8872 * WSALookupServiceEnd (WS2_32.61)
8873 */
8874 INT WINAPI WSALookupServiceEnd( HANDLE lookup )
8875 {
8876 FIXME("(%p) Stub!\n", lookup );
8877 return 0;
8878 }
8879
8880 /***********************************************************************
8881 * WSALookupServiceNextA (WS2_32.62)
8882 */
8883 INT WINAPI WSALookupServiceNextA( HANDLE lookup, DWORD flags, LPDWORD len, LPWSAQUERYSETA results )
8884 {
8885 FIXME( "(%p 0x%08x %p %p) Stub!\n", lookup, flags, len, results );
8886 SetLastError(WSA_E_NO_MORE);
8887 return SOCKET_ERROR;
8888 }
8889
8890 /***********************************************************************
8891 * WSALookupServiceNextW (WS2_32.63)
8892 */
8893 INT WINAPI WSALookupServiceNextW( HANDLE lookup, DWORD flags, LPDWORD len, LPWSAQUERYSETW results )
8894 {
8895 FIXME( "(%p 0x%08x %p %p) Stub!\n", lookup, flags, len, results );
8896 SetLastError(WSA_E_NO_MORE);
8897 return SOCKET_ERROR;
8898 }
8899
8900 /***********************************************************************
8901 * WSANtohl (WS2_32.64)
8902 */
8903 INT WINAPI WSANtohl( SOCKET s, WS_u_long netlong, WS_u_long* lphostlong )
8904 {
8905 TRACE( "(%04lx 0x%08x %p)\n", s, netlong, lphostlong );
8906
8907 if (!lphostlong) return WSAEFAULT;
8908
8909 *lphostlong = ntohl( netlong );
8910 return 0;
8911 }
8912
8913 /***********************************************************************
8914 * WSANtohs (WS2_32.65)
8915 */
8916 INT WINAPI WSANtohs( SOCKET s, WS_u_short netshort, WS_u_short* lphostshort )
8917 {
8918 TRACE( "(%04lx 0x%08x %p)\n", s, netshort, lphostshort );
8919
8920 if (!lphostshort) return WSAEFAULT;
8921
8922 *lphostshort = ntohs( netshort );
8923 return 0;
8924 }
8925
8926 /***********************************************************************
8927 * WSAProviderConfigChange (WS2_32.66)
8928 */
8929 INT WINAPI WSAProviderConfigChange( LPHANDLE handle, LPWSAOVERLAPPED overlapped,
8930 LPWSAOVERLAPPED_COMPLETION_ROUTINE completion )
8931 {
8932 FIXME( "(%p %p %p) Stub!\n", handle, overlapped, completion );
8933 return SOCKET_ERROR;
8934 }
8935
8936 /***********************************************************************
8937 * WSARecvDisconnect (WS2_32.68)
8938 */
8939 INT WINAPI WSARecvDisconnect( SOCKET s, LPWSABUF disconnectdata )
8940 {
8941 TRACE( "(%04lx %p)\n", s, disconnectdata );
8942
8943 return WS_shutdown( s, SD_RECEIVE );
8944 }
8945
8946 /***********************************************************************
8947 * WSASetServiceA (WS2_32.76)
8948 */
8949 INT WINAPI WSASetServiceA( LPWSAQUERYSETA query, WSAESETSERVICEOP operation, DWORD flags )
8950 {
8951 FIXME( "(%p 0x%08x 0x%08x) Stub!\n", query, operation, flags );
8952 return 0;
8953 }
8954
8955 /***********************************************************************
8956 * WSASetServiceW (WS2_32.77)
8957 */
8958 INT WINAPI WSASetServiceW( LPWSAQUERYSETW query, WSAESETSERVICEOP operation, DWORD flags )
8959 {
8960 FIXME( "(%p 0x%08x 0x%08x) Stub!\n", query, operation, flags );
8961 return 0;
8962 }
8963
8964 /***********************************************************************
8965 * WSCEnableNSProvider (WS2_32.84)
8966 */
8967 INT WINAPI WSCEnableNSProvider( LPGUID provider, BOOL enable )
8968 {
8969 FIXME( "(%s 0x%08x) Stub!\n", debugstr_guid(provider), enable );
8970 return 0;
8971 }
8972
8973 /***********************************************************************
8974 * WSCGetProviderPath (WS2_32.86)
8975 */
8976 INT WINAPI WSCGetProviderPath( LPGUID provider, LPWSTR path, LPINT len, LPINT errcode )
8977 {
8978 FIXME( "(%s %p %p %p) Stub!\n", debugstr_guid(provider), path, len, errcode );
8979
8980 if (!errcode || !provider || !len) return WSAEFAULT;
8981
8982 *errcode = WSAEINVAL;
8983 return SOCKET_ERROR;
8984 }
8985
8986 /***********************************************************************
8987 * WSCInstallNameSpace (WS2_32.87)
8988 */
8989 INT WINAPI WSCInstallNameSpace( LPWSTR identifier, LPWSTR path, DWORD namespace,
8990 DWORD version, LPGUID provider )
8991 {
8992 FIXME( "(%s %s 0x%08x 0x%08x %s) Stub!\n", debugstr_w(identifier), debugstr_w(path),
8993 namespace, version, debugstr_guid(provider) );
8994 return 0;
8995 }
8996
8997 /***********************************************************************
8998 * WSCUnInstallNameSpace (WS2_32.89)
8999 */
9000 INT WINAPI WSCUnInstallNameSpace( LPGUID lpProviderId )
9001 {
9002 FIXME("(%s) Stub!\n", debugstr_guid(lpProviderId));
9003 return NO_ERROR;
9004 }
9005
9006 /***********************************************************************
9007 * WSCWriteProviderOrder (WS2_32.91)
9008 */
9009 INT WINAPI WSCWriteProviderOrder( LPDWORD entry, DWORD number )
9010 {
9011 FIXME("(%p 0x%08x) Stub!\n", entry, number);
9012 return 0;
9013 }
9014
9015 /***********************************************************************
9016 * WSANSPIoctl (WS2_32.91)
9017 */
9018 INT WINAPI WSANSPIoctl( HANDLE hLookup, DWORD dwControlCode, LPVOID lpvInBuffer,
9019 DWORD cbInBuffer, LPVOID lpvOutBuffer, DWORD cbOutBuffer,
9020 LPDWORD lpcbBytesReturned, LPWSACOMPLETION lpCompletion )
9021 {
9022 FIXME("(%p, 0x%08x, %p, 0x%08x, %p, 0x%08x, %p, %p) Stub!\n", hLookup, dwControlCode,
9023 lpvInBuffer, cbInBuffer, lpvOutBuffer, cbOutBuffer, lpcbBytesReturned, lpCompletion);
9024 SetLastError(WSA_NOT_ENOUGH_MEMORY);
9025 return SOCKET_ERROR;
9026 }
9027
9028 /*****************************************************************************
9029 * WSAEnumProtocolsA [WS2_32.@]
9030 *
9031 * see function WSAEnumProtocolsW
9032 */
9033 INT WINAPI WSAEnumProtocolsA( LPINT protocols, LPWSAPROTOCOL_INFOA buffer, LPDWORD len )
9034 {
9035 return WS_EnumProtocols( FALSE, protocols, (LPWSAPROTOCOL_INFOW) buffer, len);
9036 }
9037
9038 /*****************************************************************************
9039 * WSAEnumProtocolsW [WS2_32.@]
9040 *
9041 * Retrieves information about specified set of active network protocols.
9042 *
9043 * PARAMS
9044 * protocols [I] Pointer to null-terminated array of protocol id's. NULL
9045 * retrieves information on all available protocols.
9046 * buffer [I] Pointer to a buffer to be filled with WSAPROTOCOL_INFO
9047 * structures.
9048 * len [I/O] Pointer to a variable specifying buffer size. On output
9049 * the variable holds the number of bytes needed when the
9050 * specified size is too small.
9051 *
9052 * RETURNS
9053 * Success: number of WSAPROTOCOL_INFO structures in buffer.
9054 * Failure: SOCKET_ERROR
9055 *
9056 * NOTES
9057 * NT4SP5 does not return SPX if protocols == NULL
9058 *
9059 * BUGS
9060 * - NT4SP5 returns in addition these list of NETBIOS protocols
9061 * (address family 17), each entry two times one for socket type 2 and 5
9062 *
9063 * iProtocol szProtocol
9064 * 0x80000000 \Device\NwlnkNb
9065 * 0xfffffffa \Device\NetBT_CBENT7
9066 * 0xfffffffb \Device\Nbf_CBENT7
9067 * 0xfffffffc \Device\NetBT_NdisWan5
9068 * 0xfffffffd \Device\NetBT_El9202
9069 * 0xfffffffe \Device\Nbf_El9202
9070 * 0xffffffff \Device\Nbf_NdisWan4
9071 *
9072 * - there is no check that the operating system supports the returned
9073 * protocols
9074 */
9075 INT WINAPI WSAEnumProtocolsW( LPINT protocols, LPWSAPROTOCOL_INFOW buffer, LPDWORD len )
9076 {
9077 return WS_EnumProtocols( TRUE, protocols, buffer, len);
9078 }
9079
9080 /*****************************************************************************
9081 * WSCEnumProtocols [WS2_32.@]
9082 *
9083 * PARAMS
9084 * protocols [I] Null-terminated array of iProtocol values.
9085 * buffer [O] Buffer of WSAPROTOCOL_INFOW structures.
9086 * len [I/O] Size of buffer on input/output.
9087 * errno [O] Error code.
9088 *
9089 * RETURNS
9090 * Success: number of protocols to be reported on.
9091 * Failure: SOCKET_ERROR. error is in errno.
9092 *
9093 * BUGS
9094 * Doesn't supply info on layered protocols.
9095 *
9096 */
9097 INT WINAPI WSCEnumProtocols( LPINT protocols, LPWSAPROTOCOL_INFOW buffer, LPDWORD len, LPINT err )
9098 {
9099 INT ret = WSAEnumProtocolsW( protocols, buffer, len );
9100
9101 if (ret == SOCKET_ERROR) *err = WSAENOBUFS;
9102
9103 return ret;
9104 }
Windows API implementation