advapi32: Added support for SERVICE_CONFIG_PRESHUTDOWN_INFO.
[wine] / dlls / ws2_32 / socket.c
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) 2005 Marcus Meissner
6  * Copyright (C) 2006-2008 Kai Blin
7  *
8  * This library is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * This library is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with this library; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21  *
22  * NOTE: If you make any changes to fix a particular app, make sure
23  * they don't break something else like Netscape or telnet and ftp
24  * clients and servers (www.winsite.com got a lot of those).
25  */
26
27 #include "config.h"
28 #include "wine/port.h"
29
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <string.h>
33 #include <sys/types.h>
34 #ifdef HAVE_SYS_IPC_H
35 # include <sys/ipc.h>
36 #endif
37 #ifdef HAVE_SYS_IOCTL_H
38 # include <sys/ioctl.h>
39 #endif
40 #ifdef HAVE_SYS_FILIO_H
41 # include <sys/filio.h>
42 #endif
43 #ifdef HAVE_SYS_SOCKIO_H
44 # include <sys/sockio.h>
45 #endif
46
47 #if defined(__EMX__)
48 # include <sys/so_ioctl.h>
49 #endif
50
51 #ifdef HAVE_SYS_PARAM_H
52 # include <sys/param.h>
53 #endif
54
55 #ifdef HAVE_SYS_MSG_H
56 # include <sys/msg.h>
57 #endif
58 #ifdef HAVE_SYS_WAIT_H
59 # include <sys/wait.h>
60 #endif
61 #ifdef HAVE_SYS_UIO_H
62 # include <sys/uio.h>
63 #endif
64 #ifdef HAVE_SYS_SOCKET_H
65 #include <sys/socket.h>
66 #endif
67 #ifdef HAVE_NETINET_IN_H
68 # include <netinet/in.h>
69 #endif
70 #ifdef HAVE_NETINET_TCP_H
71 # include <netinet/tcp.h>
72 #endif
73 #ifdef HAVE_ARPA_INET_H
74 # include <arpa/inet.h>
75 #endif
76 #include <ctype.h>
77 #include <fcntl.h>
78 #include <errno.h>
79 #ifdef HAVE_SYS_ERRNO_H
80 #include <sys/errno.h>
81 #endif
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
99 #ifdef HAVE_NETIPX_IPX_H
100 # include <netipx/ipx.h>
101 # define HAVE_IPX
102 #elif defined(HAVE_LINUX_IPX_H)
103 # ifdef HAVE_ASM_TYPES_H
104 #  include <asm/types.h>
105 # endif
106 # ifdef HAVE_LINUX_TYPES_H
107 #  include <linux/types.h>
108 # endif
109 # include <linux/ipx.h>
110 # define HAVE_IPX
111 #endif
112
113 #ifdef HAVE_LINUX_IRDA_H
114 # ifdef HAVE_LINUX_TYPES_H
115 #  include <linux/types.h>
116 # endif
117 # include <linux/irda.h>
118 # define HAVE_IRDA
119 #endif
120
121 #ifdef HAVE_POLL_H
122 #include <poll.h>
123 #endif
124 #ifdef HAVE_SYS_POLL_H
125 # include <sys/poll.h>
126 #endif
127 #ifdef HAVE_SYS_TIME_H
128 # include <sys/time.h>
129 #endif
130
131 #define NONAMELESSUNION
132 #define NONAMELESSSTRUCT
133 #include "ntstatus.h"
134 #define WIN32_NO_STATUS
135 #include "windef.h"
136 #include "winbase.h"
137 #include "wingdi.h"
138 #include "winuser.h"
139 #include "winerror.h"
140 #include "winnls.h"
141 #include "winsock2.h"
142 #include "mswsock.h"
143 #include "ws2tcpip.h"
144 #include "ws2spi.h"
145 #include "wsipx.h"
146 #include "mstcpip.h"
147 #include "af_irda.h"
148 #include "winnt.h"
149 #include "iphlpapi.h"
150 #include "wine/server.h"
151 #include "wine/debug.h"
152 #include "wine/exception.h"
153 #include "wine/unicode.h"
154
155 #ifdef HAVE_IPX
156 # include "wsnwlink.h"
157 #endif
158
159
160 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)  || defined(__DragonFly__)
161 # define sipx_network    sipx_addr.x_net
162 # define sipx_node       sipx_addr.x_host.c_host
163 #endif  /* __FreeBSD__ */
164
165 #ifndef INADDR_NONE
166 #define INADDR_NONE ~0UL
167 #endif
168
169 WINE_DEFAULT_DEBUG_CHANNEL(winsock);
170 WINE_DECLARE_DEBUG_CHANNEL(winediag);
171
172
173 /*
174  * The actual definition of WSASendTo, wrapped in a different function name
175  * so that internal calls from ws2_32 itself will not trigger programs like
176  * Garena, which hooks WSASendTo/WSARecvFrom calls.
177  */
178 static int WS2_sendto( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
179                        LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
180                        const struct WS_sockaddr *to, int tolen,
181                        LPWSAOVERLAPPED lpOverlapped,
182                        LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine );
183
184 /*
185  * Internal fundamental receive function, essentially WSARecvFrom with an
186  * additional parameter to support message control headers.
187  */
188 static int WS2_recv_base( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
189                           LPDWORD lpNumberOfBytesRecvd, LPDWORD lpFlags,
190                           struct WS_sockaddr *lpFrom,
191                           LPINT lpFromlen, LPWSAOVERLAPPED lpOverlapped,
192                           LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine,
193                           LPWSABUF lpControlBuffer );
194
195 /* critical section to protect some non-reentrant net function */
196 static CRITICAL_SECTION csWSgetXXXbyYYY;
197 static CRITICAL_SECTION_DEBUG critsect_debug =
198 {
199     0, 0, &csWSgetXXXbyYYY,
200     { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
201       0, 0, { (DWORD_PTR)(__FILE__ ": csWSgetXXXbyYYY") }
202 };
203 static CRITICAL_SECTION csWSgetXXXbyYYY = { &critsect_debug, -1, 0, 0, 0, 0 };
204
205 union generic_unix_sockaddr
206 {
207     struct sockaddr addr;
208     char data[128];  /* should be big enough for all families */
209 };
210
211 static inline const char *debugstr_sockaddr( const struct WS_sockaddr *a )
212 {
213     if (!a) return "(nil)";
214     switch (a->sa_family)
215     {
216     case WS_AF_INET:
217         return wine_dbg_sprintf("{ family AF_INET, address %s, port %d }",
218                                 inet_ntoa(((const struct sockaddr_in *)a)->sin_addr),
219                                 ntohs(((const struct sockaddr_in *)a)->sin_port));
220     case WS_AF_INET6:
221     {
222         char buf[46];
223         const char *p;
224         struct WS_sockaddr_in6 *sin = (struct WS_sockaddr_in6 *)a;
225
226         p = WS_inet_ntop( WS_AF_INET6, &sin->sin6_addr, buf, sizeof(buf) );
227         if (!p)
228             p = "(unknown IPv6 address)";
229         return wine_dbg_sprintf("{ family AF_INET6, address %s, port %d }",
230                                 p, ntohs(sin->sin6_port));
231     }
232     case WS_AF_IRDA:
233     {
234         DWORD addr;
235
236         memcpy( &addr, ((const SOCKADDR_IRDA *)a)->irdaDeviceID, sizeof(addr) );
237         addr = ntohl( addr );
238         return wine_dbg_sprintf("{ family AF_IRDA, addr %08x, name %s }",
239                                 addr,
240                                 ((const SOCKADDR_IRDA *)a)->irdaServiceName);
241     }
242     default:
243         return wine_dbg_sprintf("{ family %d }", a->sa_family);
244     }
245 }
246
247 /* HANDLE<->SOCKET conversion (SOCKET is UINT_PTR). */
248 #define SOCKET2HANDLE(s) ((HANDLE)(s))
249 #define HANDLE2SOCKET(h) ((SOCKET)(h))
250
251 /****************************************************************
252  * Async IO declarations
253  ****************************************************************/
254
255 typedef struct ws2_async
256 {
257     HANDLE                              hSocket;
258     int                                 type;
259     LPWSAOVERLAPPED                     user_overlapped;
260     LPWSAOVERLAPPED_COMPLETION_ROUTINE  completion_func;
261     IO_STATUS_BLOCK                     local_iosb;
262     struct WS_sockaddr                  *addr;
263     union
264     {
265         int val;     /* for send operations */
266         int *ptr;    /* for recv operations */
267     }                                   addrlen;
268     DWORD                               flags;
269     DWORD                              *lpFlags;
270     WSABUF                             *control;
271     unsigned int                        n_iovecs;
272     unsigned int                        first_iovec;
273     struct iovec                        iovec[1];
274 } ws2_async;
275
276 typedef struct ws2_accept_async
277 {
278     HANDLE              listen_socket;
279     HANDLE              accept_socket;
280     LPOVERLAPPED        user_overlapped;
281     ULONG_PTR           cvalue;
282     PVOID               buf;      /* buffer to write data to */
283     int                 data_len;
284     int                 local_len;
285     int                 remote_len;
286     struct ws2_async    *read;
287 } ws2_accept_async;
288
289 /****************************************************************/
290
291 /* ----------------------------------- internal data */
292
293 /* ws_... struct conversion flags */
294
295 typedef struct          /* WSAAsyncSelect() control struct */
296 {
297   HANDLE      service, event, sock;
298   HWND        hWnd;
299   UINT        uMsg;
300   LONG        lEvent;
301 } ws_select_info;
302
303 #define WS_MAX_SOCKETS_PER_PROCESS      128     /* reasonable guess */
304 #define WS_MAX_UDP_DATAGRAM             1024
305 static INT WINAPI WSA_DefaultBlockingHook( FARPROC x );
306
307 /* hostent's, servent's and protent's are stored in one buffer per thread,
308  * as documented on MSDN for the functions that return any of the buffers */
309 struct per_thread_data
310 {
311     int opentype;
312     struct WS_hostent *he_buffer;
313     struct WS_servent *se_buffer;
314     struct WS_protoent *pe_buffer;
315     int he_len;
316     int se_len;
317     int pe_len;
318 };
319
320 /* internal: routing description information */
321 struct route {
322     struct in_addr addr;
323     DWORD interface;
324     DWORD metric;
325 };
326
327 static INT num_startup;          /* reference counter */
328 static FARPROC blocking_hook = (FARPROC)WSA_DefaultBlockingHook;
329
330 /* function prototypes */
331 static struct WS_hostent *WS_create_he(char *name, int aliases, int addresses, int fill_addresses);
332 static struct WS_hostent *WS_dup_he(const struct hostent* p_he);
333 static struct WS_protoent *WS_dup_pe(const struct protoent* p_pe);
334 static struct WS_servent *WS_dup_se(const struct servent* p_se);
335
336 int WSAIOCTL_GetInterfaceCount(void);
337 int WSAIOCTL_GetInterfaceName(int intNumber, char *intName);
338
339 static void WS_AddCompletion( SOCKET sock, ULONG_PTR CompletionValue, NTSTATUS CompletionStatus, ULONG Information );
340
341 #define MAP_OPTION(opt) { WS_##opt, opt }
342
343 static const int ws_sock_map[][2] =
344 {
345     MAP_OPTION( SO_DEBUG ),
346     MAP_OPTION( SO_ACCEPTCONN ),
347     MAP_OPTION( SO_REUSEADDR ),
348     MAP_OPTION( SO_KEEPALIVE ),
349     MAP_OPTION( SO_DONTROUTE ),
350     MAP_OPTION( SO_BROADCAST ),
351     MAP_OPTION( SO_LINGER ),
352     MAP_OPTION( SO_OOBINLINE ),
353     MAP_OPTION( SO_SNDBUF ),
354     MAP_OPTION( SO_RCVBUF ),
355     MAP_OPTION( SO_ERROR ),
356     MAP_OPTION( SO_TYPE ),
357 #ifdef SO_RCVTIMEO
358     MAP_OPTION( SO_RCVTIMEO ),
359 #endif
360 #ifdef SO_SNDTIMEO
361     MAP_OPTION( SO_SNDTIMEO ),
362 #endif
363 };
364
365 static const int ws_tcp_map[][2] =
366 {
367 #ifdef TCP_NODELAY
368     MAP_OPTION( TCP_NODELAY ),
369 #endif
370 };
371
372 static const int ws_ip_map[][2] =
373 {
374     MAP_OPTION( IP_MULTICAST_IF ),
375     MAP_OPTION( IP_MULTICAST_TTL ),
376     MAP_OPTION( IP_MULTICAST_LOOP ),
377     MAP_OPTION( IP_ADD_MEMBERSHIP ),
378     MAP_OPTION( IP_DROP_MEMBERSHIP ),
379     MAP_OPTION( IP_OPTIONS ),
380 #ifdef IP_HDRINCL
381     MAP_OPTION( IP_HDRINCL ),
382 #endif
383     MAP_OPTION( IP_TOS ),
384     MAP_OPTION( IP_TTL ),
385 #ifdef IP_PKTINFO
386     MAP_OPTION( IP_PKTINFO ),
387 #endif
388 };
389
390 static const int ws_ipv6_map[][2] =
391 {
392 #ifdef IPV6_ADD_MEMBERSHIP
393     MAP_OPTION( IPV6_ADD_MEMBERSHIP ),
394 #endif
395 #ifdef IPV6_DROP_MEMBERSHIP
396     MAP_OPTION( IPV6_DROP_MEMBERSHIP ),
397 #endif
398     MAP_OPTION( IPV6_MULTICAST_IF ),
399     MAP_OPTION( IPV6_MULTICAST_HOPS ),
400     MAP_OPTION( IPV6_MULTICAST_LOOP ),
401     MAP_OPTION( IPV6_UNICAST_HOPS ),
402     MAP_OPTION( IPV6_V6ONLY ),
403 };
404
405 static const int ws_af_map[][2] =
406 {
407     MAP_OPTION( AF_UNSPEC ),
408     MAP_OPTION( AF_INET ),
409     MAP_OPTION( AF_INET6 ),
410 #ifdef HAVE_IPX
411     MAP_OPTION( AF_IPX ),
412 #endif
413 #ifdef AF_IRDA
414     MAP_OPTION( AF_IRDA ),
415 #endif
416     {FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO},
417 };
418
419 static const int ws_socktype_map[][2] =
420 {
421     MAP_OPTION( SOCK_DGRAM ),
422     MAP_OPTION( SOCK_STREAM ),
423     MAP_OPTION( SOCK_RAW ),
424     {FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO},
425 };
426
427 static const int ws_proto_map[][2] =
428 {
429     MAP_OPTION( IPPROTO_IP ),
430     MAP_OPTION( IPPROTO_TCP ),
431     MAP_OPTION( IPPROTO_UDP ),
432     MAP_OPTION( IPPROTO_ICMP ),
433     MAP_OPTION( IPPROTO_IGMP ),
434     MAP_OPTION( IPPROTO_RAW ),
435     {FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO},
436 };
437
438 static const int ws_aiflag_map[][2] =
439 {
440     MAP_OPTION( AI_PASSIVE ),
441     MAP_OPTION( AI_CANONNAME ),
442     MAP_OPTION( AI_NUMERICHOST ),
443     MAP_OPTION( AI_ADDRCONFIG ),
444 };
445
446 static const int ws_niflag_map[][2] =
447 {
448     MAP_OPTION( NI_NOFQDN ),
449     MAP_OPTION( NI_NUMERICHOST ),
450     MAP_OPTION( NI_NAMEREQD ),
451     MAP_OPTION( NI_NUMERICSERV ),
452     MAP_OPTION( NI_DGRAM ),
453 };
454
455 static const int ws_eai_map[][2] =
456 {
457     MAP_OPTION( EAI_AGAIN ),
458     MAP_OPTION( EAI_BADFLAGS ),
459     MAP_OPTION( EAI_FAIL ),
460     MAP_OPTION( EAI_FAMILY ),
461     MAP_OPTION( EAI_MEMORY ),
462 /* Note: EAI_NODATA is deprecated, but still 
463  * used by Windows and Linux... We map the newer
464  * EAI_NONAME to EAI_NODATA for now until Windows
465  * changes too.
466  */
467 #ifdef EAI_NODATA
468     MAP_OPTION( EAI_NODATA ),
469 #endif
470 #ifdef EAI_NONAME
471     { WS_EAI_NODATA, EAI_NONAME },
472 #endif
473
474     MAP_OPTION( EAI_SERVICE ),
475     MAP_OPTION( EAI_SOCKTYPE ),
476     { 0, 0 }
477 };
478
479 static const char magic_loopback_addr[] = {127, 12, 34, 56};
480
481 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
482 static inline WSACMSGHDR *fill_control_message(int level, int type, WSACMSGHDR *current, ULONG *maxsize, void *data, int len)
483 {
484     ULONG msgsize = sizeof(WSACMSGHDR) + WSA_CMSG_ALIGN(len);
485     char *ptr = (char *) current + sizeof(WSACMSGHDR);
486
487     /* Make sure there is at least enough room for this entry */
488     if (msgsize > *maxsize)
489         return NULL;
490     *maxsize -= msgsize;
491     /* Fill in the entry */
492     current->cmsg_len = sizeof(WSACMSGHDR) + len;
493     current->cmsg_level = level;
494     current->cmsg_type = type;
495     memcpy(ptr, data, len);
496     /* Return the pointer to where next entry should go */
497     return (WSACMSGHDR *) (ptr + WSA_CMSG_ALIGN(len));
498 }
499
500 static inline int convert_control_headers(struct msghdr *hdr, WSABUF *control)
501 {
502 #ifdef IP_PKTINFO
503     WSACMSGHDR *cmsg_win = (WSACMSGHDR *) control->buf, *ptr;
504     ULONG ctlsize = control->len;
505     struct cmsghdr *cmsg_unix;
506
507     ptr = cmsg_win;
508     /* Loop over all the headers, converting as appropriate */
509     for (cmsg_unix = CMSG_FIRSTHDR(hdr); cmsg_unix != NULL; cmsg_unix = CMSG_NXTHDR(hdr, cmsg_unix))
510     {
511         switch(cmsg_unix->cmsg_level)
512         {
513             case IPPROTO_IP:
514                 switch(cmsg_unix->cmsg_type)
515                 {
516                     case IP_PKTINFO:
517                     {
518                         /* Convert the Unix IP_PKTINFO structure to the Windows version */
519                         struct in_pktinfo *data_unix = (struct in_pktinfo *) CMSG_DATA(cmsg_unix);
520                         struct WS_in_pktinfo data_win;
521
522                         memcpy(&data_win.ipi_addr,&data_unix->ipi_addr.s_addr,4); /* 4 bytes = 32 address bits */
523                         data_win.ipi_ifindex = data_unix->ipi_ifindex;
524                         ptr = fill_control_message(WS_IPPROTO_IP, WS_IP_PKTINFO, ptr, &ctlsize,
525                                                    (void*)&data_win, sizeof(data_win));
526                         if (!ptr) goto error;
527                     }   break;
528                     default:
529                         FIXME("Unhandled IPPROTO_IP message header type %d\n", cmsg_unix->cmsg_type);
530                         break;
531                 }
532                 break;
533             default:
534                 FIXME("Unhandled message header level %d\n", cmsg_unix->cmsg_level);
535                 break;
536         }
537     }
538
539 error:
540     /* Set the length of the returned control headers */
541     control->len = (ptr == NULL ? 0 : (char*)ptr - (char*)cmsg_win);
542     return (ptr != NULL);
543 #else /* IP_PKTINFO */
544     control->len = 0;
545     return 1;
546 #endif /* IP_PKTINFO */
547 }
548 #endif /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
549
550 /* ----------------------------------- error handling */
551
552 static NTSTATUS sock_get_ntstatus( int err )
553 {
554     switch ( err )
555     {
556         case EBADF:             return STATUS_INVALID_HANDLE;
557         case EBUSY:             return STATUS_DEVICE_BUSY;
558         case EPERM:
559         case EACCES:            return STATUS_ACCESS_DENIED;
560         case EFAULT:            return STATUS_NO_MEMORY;
561         case EINVAL:            return STATUS_INVALID_PARAMETER;
562         case ENFILE:
563         case EMFILE:            return STATUS_TOO_MANY_OPENED_FILES;
564         case EWOULDBLOCK:       return STATUS_CANT_WAIT;
565         case EINPROGRESS:       return STATUS_PENDING;
566         case EALREADY:          return STATUS_NETWORK_BUSY;
567         case ENOTSOCK:          return STATUS_OBJECT_TYPE_MISMATCH;
568         case EDESTADDRREQ:      return STATUS_INVALID_PARAMETER;
569         case EMSGSIZE:          return STATUS_BUFFER_OVERFLOW;
570         case EPROTONOSUPPORT:
571         case ESOCKTNOSUPPORT:
572         case EPFNOSUPPORT:
573         case EAFNOSUPPORT:
574         case EPROTOTYPE:        return STATUS_NOT_SUPPORTED;
575         case ENOPROTOOPT:       return STATUS_INVALID_PARAMETER;
576         case EOPNOTSUPP:        return STATUS_NOT_SUPPORTED;
577         case EADDRINUSE:        return STATUS_ADDRESS_ALREADY_ASSOCIATED;
578         case EADDRNOTAVAIL:     return STATUS_INVALID_PARAMETER;
579         case ECONNREFUSED:      return STATUS_CONNECTION_REFUSED;
580         case ESHUTDOWN:         return STATUS_PIPE_DISCONNECTED;
581         case ENOTCONN:          return STATUS_CONNECTION_DISCONNECTED;
582         case ETIMEDOUT:         return STATUS_IO_TIMEOUT;
583         case ENETUNREACH:       return STATUS_NETWORK_UNREACHABLE;
584         case ENETDOWN:          return STATUS_NETWORK_BUSY;
585         case EPIPE:
586         case ECONNRESET:        return STATUS_CONNECTION_RESET;
587         case ECONNABORTED:      return STATUS_CONNECTION_ABORTED;
588
589         case 0:                 return STATUS_SUCCESS;
590         default:
591             WARN("Unknown errno %d!\n", err);
592             return STATUS_UNSUCCESSFUL;
593     }
594 }
595
596 static UINT sock_get_error( int err )
597 {
598         switch(err)
599     {
600         case EINTR:             return WSAEINTR;
601         case EBADF:             return WSAEBADF;
602         case EPERM:
603         case EACCES:            return WSAEACCES;
604         case EFAULT:            return WSAEFAULT;
605         case EINVAL:            return WSAEINVAL;
606         case EMFILE:            return WSAEMFILE;
607         case EWOULDBLOCK:       return WSAEWOULDBLOCK;
608         case EINPROGRESS:       return WSAEINPROGRESS;
609         case EALREADY:          return WSAEALREADY;
610         case ENOTSOCK:          return WSAENOTSOCK;
611         case EDESTADDRREQ:      return WSAEDESTADDRREQ;
612         case EMSGSIZE:          return WSAEMSGSIZE;
613         case EPROTOTYPE:        return WSAEPROTOTYPE;
614         case ENOPROTOOPT:       return WSAENOPROTOOPT;
615         case EPROTONOSUPPORT:   return WSAEPROTONOSUPPORT;
616         case ESOCKTNOSUPPORT:   return WSAESOCKTNOSUPPORT;
617         case EOPNOTSUPP:        return WSAEOPNOTSUPP;
618         case EPFNOSUPPORT:      return WSAEPFNOSUPPORT;
619         case EAFNOSUPPORT:      return WSAEAFNOSUPPORT;
620         case EADDRINUSE:        return WSAEADDRINUSE;
621         case EADDRNOTAVAIL:     return WSAEADDRNOTAVAIL;
622         case ENETDOWN:          return WSAENETDOWN;
623         case ENETUNREACH:       return WSAENETUNREACH;
624         case ENETRESET:         return WSAENETRESET;
625         case ECONNABORTED:      return WSAECONNABORTED;
626         case EPIPE:
627         case ECONNRESET:        return WSAECONNRESET;
628         case ENOBUFS:           return WSAENOBUFS;
629         case EISCONN:           return WSAEISCONN;
630         case ENOTCONN:          return WSAENOTCONN;
631         case ESHUTDOWN:         return WSAESHUTDOWN;
632         case ETOOMANYREFS:      return WSAETOOMANYREFS;
633         case ETIMEDOUT:         return WSAETIMEDOUT;
634         case ECONNREFUSED:      return WSAECONNREFUSED;
635         case ELOOP:             return WSAELOOP;
636         case ENAMETOOLONG:      return WSAENAMETOOLONG;
637         case EHOSTDOWN:         return WSAEHOSTDOWN;
638         case EHOSTUNREACH:      return WSAEHOSTUNREACH;
639         case ENOTEMPTY:         return WSAENOTEMPTY;
640 #ifdef EPROCLIM
641         case EPROCLIM:          return WSAEPROCLIM;
642 #endif
643 #ifdef EUSERS
644         case EUSERS:            return WSAEUSERS;
645 #endif
646 #ifdef EDQUOT
647         case EDQUOT:            return WSAEDQUOT;
648 #endif
649 #ifdef ESTALE
650         case ESTALE:            return WSAESTALE;
651 #endif
652 #ifdef EREMOTE
653         case EREMOTE:           return WSAEREMOTE;
654 #endif
655
656         /* just in case we ever get here and there are no problems */
657         case 0:                 return 0;
658         default:
659                 WARN("Unknown errno %d!\n", err);
660                 return WSAEOPNOTSUPP;
661     }
662 }
663
664 static UINT wsaErrno(void)
665 {
666     int loc_errno = errno;
667     WARN("errno %d, (%s).\n", loc_errno, strerror(loc_errno));
668
669     return sock_get_error( loc_errno );
670 }
671
672 /* most ws2 overlapped functions return an ntstatus-based error code */
673 static NTSTATUS wsaErrStatus(void)
674 {
675     int loc_errno = errno;
676     WARN("errno %d, (%s).\n", loc_errno, strerror(loc_errno));
677
678     return sock_get_ntstatus(loc_errno);
679 }
680
681 static UINT wsaHerrno(int loc_errno)
682 {
683     WARN("h_errno %d.\n", loc_errno);
684
685     switch(loc_errno)
686     {
687         case HOST_NOT_FOUND:    return WSAHOST_NOT_FOUND;
688         case TRY_AGAIN:         return WSATRY_AGAIN;
689         case NO_RECOVERY:       return WSANO_RECOVERY;
690         case NO_DATA:           return WSANO_DATA;
691         case ENOBUFS:           return WSAENOBUFS;
692
693         case 0:                 return 0;
694         default:
695                 WARN("Unknown h_errno %d!\n", loc_errno);
696                 return WSAEOPNOTSUPP;
697     }
698 }
699
700 static inline DWORD NtStatusToWSAError( const DWORD status )
701 {
702     /* We only need to cover the status codes set by server async request handling */
703     DWORD wserr;
704     switch ( status )
705     {
706     case STATUS_SUCCESS:                    wserr = 0;                     break;
707     case STATUS_PENDING:                    wserr = WSA_IO_PENDING;        break;
708     case STATUS_OBJECT_TYPE_MISMATCH:       wserr = WSAENOTSOCK;           break;
709     case STATUS_INVALID_HANDLE:             wserr = WSAEBADF;              break;
710     case STATUS_INVALID_PARAMETER:          wserr = WSAEINVAL;             break;
711     case STATUS_PIPE_DISCONNECTED:          wserr = WSAESHUTDOWN;          break;
712     case STATUS_NETWORK_BUSY:               wserr = WSAEALREADY;           break;
713     case STATUS_NETWORK_UNREACHABLE:        wserr = WSAENETUNREACH;        break;
714     case STATUS_CONNECTION_REFUSED:         wserr = WSAECONNREFUSED;       break;
715     case STATUS_CONNECTION_DISCONNECTED:    wserr = WSAENOTCONN;           break;
716     case STATUS_CONNECTION_RESET:           wserr = WSAECONNRESET;         break;
717     case STATUS_CONNECTION_ABORTED:         wserr = WSAECONNABORTED;       break;
718     case STATUS_CANCELLED:                  wserr = WSA_OPERATION_ABORTED; break;
719     case STATUS_ADDRESS_ALREADY_ASSOCIATED: wserr = WSAEADDRINUSE;         break;
720     case STATUS_IO_TIMEOUT:
721     case STATUS_TIMEOUT:                    wserr = WSAETIMEDOUT;          break;
722     case STATUS_NO_MEMORY:                  wserr = WSAEFAULT;             break;
723     case STATUS_ACCESS_DENIED:              wserr = WSAEACCES;             break;
724     case STATUS_TOO_MANY_OPENED_FILES:      wserr = WSAEMFILE;             break;
725     case STATUS_CANT_WAIT:                  wserr = WSAEWOULDBLOCK;        break;
726     case STATUS_BUFFER_OVERFLOW:            wserr = WSAEMSGSIZE;           break;
727     case STATUS_NOT_SUPPORTED:              wserr = WSAEOPNOTSUPP;         break;
728     case STATUS_HOST_UNREACHABLE:           wserr = WSAEHOSTUNREACH;       break;
729
730     default:
731         wserr = RtlNtStatusToDosError( status );
732         FIXME( "Status code %08x converted to DOS error code %x\n", status, wserr );
733     }
734     return wserr;
735 }
736
737 /* set last error code from NT status without mapping WSA errors */
738 static inline unsigned int set_error( unsigned int err )
739 {
740     if (err)
741     {
742         err = NtStatusToWSAError( err );
743         SetLastError( err );
744     }
745     return err;
746 }
747
748 static inline int get_sock_fd( SOCKET s, DWORD access, unsigned int *options )
749 {
750     int fd;
751     if (set_error( wine_server_handle_to_fd( SOCKET2HANDLE(s), access, &fd, options ) ))
752         return -1;
753     return fd;
754 }
755
756 static inline void release_sock_fd( SOCKET s, int fd )
757 {
758     wine_server_release_fd( SOCKET2HANDLE(s), fd );
759 }
760
761 static void _enable_event( HANDLE s, unsigned int event,
762                            unsigned int sstate, unsigned int cstate )
763 {
764     SERVER_START_REQ( enable_socket_event )
765     {
766         req->handle = wine_server_obj_handle( s );
767         req->mask   = event;
768         req->sstate = sstate;
769         req->cstate = cstate;
770         wine_server_call( req );
771     }
772     SERVER_END_REQ;
773 }
774
775 static int _is_blocking(SOCKET s)
776 {
777     int ret;
778     SERVER_START_REQ( get_socket_event )
779     {
780         req->handle  = wine_server_obj_handle( SOCKET2HANDLE(s) );
781         req->service = FALSE;
782         req->c_event = 0;
783         wine_server_call( req );
784         ret = (reply->state & FD_WINE_NONBLOCKING) == 0;
785     }
786     SERVER_END_REQ;
787     return ret;
788 }
789
790 static unsigned int _get_sock_mask(SOCKET s)
791 {
792     unsigned int ret;
793     SERVER_START_REQ( get_socket_event )
794     {
795         req->handle  = wine_server_obj_handle( SOCKET2HANDLE(s) );
796         req->service = FALSE;
797         req->c_event = 0;
798         wine_server_call( req );
799         ret = reply->mask;
800     }
801     SERVER_END_REQ;
802     return ret;
803 }
804
805 static void _sync_sock_state(SOCKET s)
806 {
807     /* do a dummy wineserver request in order to let
808        the wineserver run through its select loop once */
809     (void)_is_blocking(s);
810 }
811
812 static int _get_sock_error(SOCKET s, unsigned int bit)
813 {
814     int events[FD_MAX_EVENTS];
815
816     SERVER_START_REQ( get_socket_event )
817     {
818         req->handle  = wine_server_obj_handle( SOCKET2HANDLE(s) );
819         req->service = FALSE;
820         req->c_event = 0;
821         wine_server_set_reply( req, events, sizeof(events) );
822         wine_server_call( req );
823     }
824     SERVER_END_REQ;
825     return events[bit];
826 }
827
828 static struct per_thread_data *get_per_thread_data(void)
829 {
830     struct per_thread_data * ptb = NtCurrentTeb()->WinSockData;
831     /* lazy initialization */
832     if (!ptb)
833     {
834         ptb = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*ptb) );
835         NtCurrentTeb()->WinSockData = ptb;
836     }
837     return ptb;
838 }
839
840 static void free_per_thread_data(void)
841 {
842     struct per_thread_data * ptb = NtCurrentTeb()->WinSockData;
843
844     if (!ptb) return;
845
846     /* delete scratch buffers */
847     HeapFree( GetProcessHeap(), 0, ptb->he_buffer );
848     HeapFree( GetProcessHeap(), 0, ptb->se_buffer );
849     HeapFree( GetProcessHeap(), 0, ptb->pe_buffer );
850     ptb->he_buffer = NULL;
851     ptb->se_buffer = NULL;
852     ptb->pe_buffer = NULL;
853
854     HeapFree( GetProcessHeap(), 0, ptb );
855     NtCurrentTeb()->WinSockData = NULL;
856 }
857
858 /***********************************************************************
859  *              DllMain (WS2_32.init)
860  */
861 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID fImpLoad)
862 {
863     TRACE("%p 0x%x %p\n", hInstDLL, fdwReason, fImpLoad);
864     switch (fdwReason) {
865     case DLL_PROCESS_ATTACH:
866         break;
867     case DLL_PROCESS_DETACH:
868         free_per_thread_data();
869         DeleteCriticalSection(&csWSgetXXXbyYYY);
870         num_startup = 0;
871         break;
872     case DLL_THREAD_DETACH:
873         free_per_thread_data();
874         break;
875     }
876     return TRUE;
877 }
878
879 /***********************************************************************
880  *          convert_sockopt()
881  *
882  * Converts socket flags from Windows format.
883  * Return 1 if converted, 0 if not (error).
884  */
885 static int convert_sockopt(INT *level, INT *optname)
886 {
887   unsigned int i;
888   switch (*level)
889   {
890      case WS_SOL_SOCKET:
891         *level = SOL_SOCKET;
892         for(i=0; i<sizeof(ws_sock_map)/sizeof(ws_sock_map[0]); i++) {
893             if( ws_sock_map[i][0] == *optname )
894             {
895                 *optname = ws_sock_map[i][1];
896                 return 1;
897             }
898         }
899         FIXME("Unknown SOL_SOCKET optname 0x%x\n", *optname);
900         break;
901      case WS_IPPROTO_TCP:
902         *level = IPPROTO_TCP;
903         for(i=0; i<sizeof(ws_tcp_map)/sizeof(ws_tcp_map[0]); i++) {
904             if ( ws_tcp_map[i][0] == *optname )
905             {
906                 *optname = ws_tcp_map[i][1];
907                 return 1;
908             }
909         }
910         FIXME("Unknown IPPROTO_TCP optname 0x%x\n", *optname);
911         break;
912      case WS_IPPROTO_IP:
913         *level = IPPROTO_IP;
914         for(i=0; i<sizeof(ws_ip_map)/sizeof(ws_ip_map[0]); i++) {
915             if (ws_ip_map[i][0] == *optname )
916             {
917                 *optname = ws_ip_map[i][1];
918                 return 1;
919             }
920         }
921         FIXME("Unknown IPPROTO_IP optname 0x%x\n", *optname);
922         break;
923      case WS_IPPROTO_IPV6:
924         *level = IPPROTO_IPV6;
925         for(i=0; i<sizeof(ws_ipv6_map)/sizeof(ws_ipv6_map[0]); i++) {
926             if (ws_ipv6_map[i][0] == *optname )
927             {
928                 *optname = ws_ipv6_map[i][1];
929                 return 1;
930             }
931         }
932         FIXME("Unknown IPPROTO_IPV6 optname 0x%x\n", *optname);
933         break;
934      default: FIXME("Unimplemented or unknown socket level\n");
935   }
936   return 0;
937 }
938
939 /* ----------------------------------- Per-thread info (or per-process?) */
940
941 static char *strdup_lower(const char *str)
942 {
943     int i;
944     char *ret = HeapAlloc( GetProcessHeap(), 0, strlen(str) + 1 );
945
946     if (ret)
947     {
948         for (i = 0; str[i]; i++) ret[i] = tolower(str[i]);
949         ret[i] = 0;
950     }
951     else SetLastError(WSAENOBUFS);
952     return ret;
953 }
954
955 static inline int sock_error_p(int s)
956 {
957     unsigned int optval, optlen;
958
959     optlen = sizeof(optval);
960     getsockopt(s, SOL_SOCKET, SO_ERROR, (void *) &optval, &optlen);
961     if (optval) WARN("\t[%i] error: %d\n", s, optval);
962     return optval != 0;
963 }
964
965 /* Utility: get the SO_RCVTIMEO or SO_SNDTIMEO socket option
966  * from an fd and return the value converted to milli seconds
967  * or -1 if there is an infinite time out */
968 static inline int get_rcvsnd_timeo( int fd, int optname)
969 {
970   struct timeval tv;
971   unsigned int len = sizeof(tv);
972   int ret = getsockopt(fd, SOL_SOCKET, optname, &tv, &len);
973   if( ret >= 0)
974       ret = tv.tv_sec * 1000 + tv.tv_usec / 1000;
975   if( ret <= 0 ) /* tv == {0,0} means infinite time out */
976       return -1;
977   return ret;
978 }
979
980 /* macro wrappers for portability */
981 #ifdef SO_RCVTIMEO
982 #define GET_RCVTIMEO(fd) get_rcvsnd_timeo( (fd), SO_RCVTIMEO)
983 #else
984 #define GET_RCVTIMEO(fd) (-1)
985 #endif
986
987 #ifdef SO_SNDTIMEO
988 #define GET_SNDTIMEO(fd) get_rcvsnd_timeo( (fd), SO_SNDTIMEO)
989 #else
990 #define GET_SNDTIMEO(fd) (-1)
991 #endif
992
993 /* utility: given an fd, will block until one of the events occurs */
994 static inline int do_block( int fd, int events, int timeout )
995 {
996   struct pollfd pfd;
997   int ret;
998
999   pfd.fd = fd;
1000   pfd.events = events;
1001
1002   while ((ret = poll(&pfd, 1, timeout)) < 0)
1003   {
1004       if (errno != EINTR)
1005           return -1;
1006   }
1007   if( ret == 0 )
1008       return 0;
1009   return pfd.revents;
1010 }
1011
1012 static int
1013 convert_af_w2u(int windowsaf) {
1014     unsigned int i;
1015
1016     for (i=0;i<sizeof(ws_af_map)/sizeof(ws_af_map[0]);i++)
1017         if (ws_af_map[i][0] == windowsaf)
1018             return ws_af_map[i][1];
1019     FIXME("unhandled Windows address family %d\n", windowsaf);
1020     return -1;
1021 }
1022
1023 static int
1024 convert_af_u2w(int unixaf) {
1025     unsigned int i;
1026
1027     for (i=0;i<sizeof(ws_af_map)/sizeof(ws_af_map[0]);i++)
1028         if (ws_af_map[i][1] == unixaf)
1029             return ws_af_map[i][0];
1030     FIXME("unhandled UNIX address family %d\n", unixaf);
1031     return -1;
1032 }
1033
1034 static int
1035 convert_proto_w2u(int windowsproto) {
1036     unsigned int i;
1037
1038     for (i=0;i<sizeof(ws_proto_map)/sizeof(ws_proto_map[0]);i++)
1039         if (ws_proto_map[i][0] == windowsproto)
1040             return ws_proto_map[i][1];
1041     FIXME("unhandled Windows socket protocol %d\n", windowsproto);
1042     return -1;
1043 }
1044
1045 static int
1046 convert_proto_u2w(int unixproto) {
1047     unsigned int i;
1048
1049     for (i=0;i<sizeof(ws_proto_map)/sizeof(ws_proto_map[0]);i++)
1050         if (ws_proto_map[i][1] == unixproto)
1051             return ws_proto_map[i][0];
1052     FIXME("unhandled UNIX socket protocol %d\n", unixproto);
1053     return -1;
1054 }
1055
1056 static int
1057 convert_socktype_w2u(int windowssocktype) {
1058     unsigned int i;
1059
1060     for (i=0;i<sizeof(ws_socktype_map)/sizeof(ws_socktype_map[0]);i++)
1061         if (ws_socktype_map[i][0] == windowssocktype)
1062             return ws_socktype_map[i][1];
1063     FIXME("unhandled Windows socket type %d\n", windowssocktype);
1064     return -1;
1065 }
1066
1067 static int
1068 convert_socktype_u2w(int unixsocktype) {
1069     unsigned int i;
1070
1071     for (i=0;i<sizeof(ws_socktype_map)/sizeof(ws_socktype_map[0]);i++)
1072         if (ws_socktype_map[i][1] == unixsocktype)
1073             return ws_socktype_map[i][0];
1074     FIXME("unhandled UNIX socket type %d\n", unixsocktype);
1075     return -1;
1076 }
1077
1078 /* ----------------------------------- API -----
1079  *
1080  * Init / cleanup / error checking.
1081  */
1082
1083 /***********************************************************************
1084  *      WSAStartup              (WS2_32.115)
1085  */
1086 int WINAPI WSAStartup(WORD wVersionRequested, LPWSADATA lpWSAData)
1087 {
1088     TRACE("verReq=%x\n", wVersionRequested);
1089
1090     if (LOBYTE(wVersionRequested) < 1)
1091         return WSAVERNOTSUPPORTED;
1092
1093     if (!lpWSAData) return WSAEINVAL;
1094
1095     num_startup++;
1096
1097     /* that's the whole of the negotiation for now */
1098     lpWSAData->wVersion = wVersionRequested;
1099     /* return winsock information */
1100     lpWSAData->wHighVersion = 0x0202;
1101     strcpy(lpWSAData->szDescription, "WinSock 2.0" );
1102     strcpy(lpWSAData->szSystemStatus, "Running" );
1103     lpWSAData->iMaxSockets = WS_MAX_SOCKETS_PER_PROCESS;
1104     lpWSAData->iMaxUdpDg = WS_MAX_UDP_DATAGRAM;
1105     /* don't do anything with lpWSAData->lpVendorInfo */
1106     /* (some apps don't allocate the space for this field) */
1107
1108     TRACE("succeeded\n");
1109     return 0;
1110 }
1111
1112
1113 /***********************************************************************
1114  *      WSACleanup                      (WS2_32.116)
1115  */
1116 INT WINAPI WSACleanup(void)
1117 {
1118     if (num_startup) {
1119         num_startup--;
1120         return 0;
1121     }
1122     SetLastError(WSANOTINITIALISED);
1123     return SOCKET_ERROR;
1124 }
1125
1126
1127 /***********************************************************************
1128  *      WSAGetLastError         (WS2_32.111)
1129  */
1130 INT WINAPI WSAGetLastError(void)
1131 {
1132         return GetLastError();
1133 }
1134
1135 /***********************************************************************
1136  *      WSASetLastError         (WS2_32.112)
1137  */
1138 void WINAPI WSASetLastError(INT iError) {
1139     SetLastError(iError);
1140 }
1141
1142 static struct WS_hostent *check_buffer_he(int size)
1143 {
1144     struct per_thread_data * ptb = get_per_thread_data();
1145     if (ptb->he_buffer)
1146     {
1147         if (ptb->he_len >= size ) return ptb->he_buffer;
1148         HeapFree( GetProcessHeap(), 0, ptb->he_buffer );
1149     }
1150     ptb->he_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->he_len = size) );
1151     if (!ptb->he_buffer) SetLastError(WSAENOBUFS);
1152     return ptb->he_buffer;
1153 }
1154
1155 static struct WS_servent *check_buffer_se(int size)
1156 {
1157     struct per_thread_data * ptb = get_per_thread_data();
1158     if (ptb->se_buffer)
1159     {
1160         if (ptb->se_len >= size ) return ptb->se_buffer;
1161         HeapFree( GetProcessHeap(), 0, ptb->se_buffer );
1162     }
1163     ptb->se_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->se_len = size) );
1164     if (!ptb->se_buffer) SetLastError(WSAENOBUFS);
1165     return ptb->se_buffer;
1166 }
1167
1168 static struct WS_protoent *check_buffer_pe(int size)
1169 {
1170     struct per_thread_data * ptb = get_per_thread_data();
1171     if (ptb->pe_buffer)
1172     {
1173         if (ptb->pe_len >= size ) return ptb->pe_buffer;
1174         HeapFree( GetProcessHeap(), 0, ptb->pe_buffer );
1175     }
1176     ptb->pe_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->pe_len = size) );
1177     if (!ptb->pe_buffer) SetLastError(WSAENOBUFS);
1178     return ptb->pe_buffer;
1179 }
1180
1181 /* ----------------------------------- i/o APIs */
1182
1183 static inline BOOL supported_pf(int pf)
1184 {
1185     switch (pf)
1186     {
1187     case WS_AF_INET:
1188     case WS_AF_INET6:
1189         return TRUE;
1190 #ifdef HAVE_IPX
1191     case WS_AF_IPX:
1192         return TRUE;
1193 #endif
1194 #ifdef HAVE_IRDA
1195     case WS_AF_IRDA:
1196         return TRUE;
1197 #endif
1198     default:
1199         return FALSE;
1200     }
1201 }
1202
1203
1204 /**********************************************************************/
1205
1206 /* Returns the length of the converted address if successful, 0 if it was too small to
1207  * start with.
1208  */
1209 static unsigned int ws_sockaddr_ws2u(const struct WS_sockaddr* wsaddr, int wsaddrlen,
1210                                      union generic_unix_sockaddr *uaddr)
1211 {
1212     unsigned int uaddrlen = 0;
1213
1214     switch (wsaddr->sa_family)
1215     {
1216 #ifdef HAVE_IPX
1217     case WS_AF_IPX:
1218         {
1219             const struct WS_sockaddr_ipx* wsipx=(const struct WS_sockaddr_ipx*)wsaddr;
1220             struct sockaddr_ipx* uipx = (struct sockaddr_ipx *)uaddr;
1221
1222             if (wsaddrlen<sizeof(struct WS_sockaddr_ipx))
1223                 return 0;
1224
1225             uaddrlen = sizeof(struct sockaddr_ipx);
1226             memset( uaddr, 0, uaddrlen );
1227             uipx->sipx_family=AF_IPX;
1228             uipx->sipx_port=wsipx->sa_socket;
1229             /* copy sa_netnum and sa_nodenum to sipx_network and sipx_node
1230              * in one go
1231              */
1232             memcpy(&uipx->sipx_network,wsipx->sa_netnum,sizeof(uipx->sipx_network)+sizeof(uipx->sipx_node));
1233 #ifdef IPX_FRAME_NONE
1234             uipx->sipx_type=IPX_FRAME_NONE;
1235 #endif
1236             break;
1237         }
1238 #endif
1239     case WS_AF_INET6: {
1240         struct sockaddr_in6* uin6 = (struct sockaddr_in6 *)uaddr;
1241         const struct WS_sockaddr_in6* win6 = (const struct WS_sockaddr_in6*)wsaddr;
1242
1243         /* Note: Windows has 2 versions of the sockaddr_in6 struct, one with
1244          * scope_id, one without.
1245          */
1246         if (wsaddrlen >= sizeof(struct WS_sockaddr_in6_old)) {
1247             uaddrlen = sizeof(struct sockaddr_in6);
1248             memset( uaddr, 0, uaddrlen );
1249             uin6->sin6_family   = AF_INET6;
1250             uin6->sin6_port     = win6->sin6_port;
1251             uin6->sin6_flowinfo = win6->sin6_flowinfo;
1252 #ifdef HAVE_STRUCT_SOCKADDR_IN6_SIN6_SCOPE_ID
1253             if (wsaddrlen >= sizeof(struct WS_sockaddr_in6)) uin6->sin6_scope_id = win6->sin6_scope_id;
1254 #endif
1255             memcpy(&uin6->sin6_addr,&win6->sin6_addr,16); /* 16 bytes = 128 address bits */
1256             break;
1257         }
1258         FIXME("bad size %d for WS_sockaddr_in6\n",wsaddrlen);
1259         return 0;
1260     }
1261     case WS_AF_INET: {
1262         struct sockaddr_in* uin = (struct sockaddr_in *)uaddr;
1263         const struct WS_sockaddr_in* win = (const struct WS_sockaddr_in*)wsaddr;
1264
1265         if (wsaddrlen<sizeof(struct WS_sockaddr_in))
1266             return 0;
1267         uaddrlen = sizeof(struct sockaddr_in);
1268         memset( uaddr, 0, uaddrlen );
1269         uin->sin_family = AF_INET;
1270         uin->sin_port   = win->sin_port;
1271         memcpy(&uin->sin_addr,&win->sin_addr,4); /* 4 bytes = 32 address bits */
1272         break;
1273     }
1274 #ifdef HAVE_IRDA
1275     case WS_AF_IRDA: {
1276         struct sockaddr_irda *uin = (struct sockaddr_irda *)uaddr;
1277         const SOCKADDR_IRDA *win = (const SOCKADDR_IRDA *)wsaddr;
1278
1279         if (wsaddrlen < sizeof(SOCKADDR_IRDA))
1280             return 0;
1281         uaddrlen = sizeof(struct sockaddr_irda);
1282         memset( uaddr, 0, uaddrlen );
1283         uin->sir_family = AF_IRDA;
1284         if (!strncmp( win->irdaServiceName, "LSAP-SEL", strlen( "LSAP-SEL" ) ))
1285         {
1286             unsigned int lsap_sel = 0;
1287
1288             sscanf( win->irdaServiceName, "LSAP-SEL%u", &lsap_sel );
1289             uin->sir_lsap_sel = lsap_sel;
1290         }
1291         else
1292         {
1293             uin->sir_lsap_sel = LSAP_ANY;
1294             memcpy( uin->sir_name, win->irdaServiceName, 25 );
1295         }
1296         memcpy( &uin->sir_addr, win->irdaDeviceID, sizeof(uin->sir_addr) );
1297         break;
1298     }
1299 #endif
1300     case WS_AF_UNSPEC: {
1301         /* Try to determine the needed space by the passed windows sockaddr space */
1302         switch (wsaddrlen) {
1303         default: /* likely a ipv4 address */
1304         case sizeof(struct WS_sockaddr_in):
1305             uaddrlen = sizeof(struct sockaddr_in);
1306             break;
1307 #ifdef HAVE_IPX
1308         case sizeof(struct WS_sockaddr_ipx):
1309             uaddrlen = sizeof(struct sockaddr_ipx);
1310             break;
1311 #endif
1312 #ifdef HAVE_IRDA
1313         case sizeof(SOCKADDR_IRDA):
1314             uaddrlen = sizeof(struct sockaddr_irda);
1315             break;
1316 #endif
1317         case sizeof(struct WS_sockaddr_in6):
1318         case sizeof(struct WS_sockaddr_in6_old):
1319             uaddrlen = sizeof(struct sockaddr_in6);
1320             break;
1321         }
1322         memset( uaddr, 0, uaddrlen );
1323         break;
1324     }
1325     default:
1326         FIXME("Unknown address family %d, return NULL.\n", wsaddr->sa_family);
1327         return 0;
1328     }
1329     return uaddrlen;
1330 }
1331
1332 static BOOL is_sockaddr_bound(const struct sockaddr *uaddr, int uaddrlen)
1333 {
1334     switch (uaddr->sa_family)
1335     {
1336 #ifdef HAVE_IPX
1337         case AF_IPX:
1338             FIXME("don't know how to tell if IPX socket is bound, assuming it is!\n");
1339             return TRUE;
1340 #endif
1341         case AF_INET6:
1342         {
1343             static const struct sockaddr_in6 emptyAddr;
1344             const struct sockaddr_in6 *in6 = (const struct sockaddr_in6*) uaddr;
1345             return in6->sin6_port || memcmp(&in6->sin6_addr, &emptyAddr.sin6_addr, sizeof(struct in6_addr));
1346         }
1347         case AF_INET:
1348         {
1349             static const struct sockaddr_in emptyAddr;
1350             const struct sockaddr_in *in = (const struct sockaddr_in*) uaddr;
1351             return in->sin_port || memcmp(&in->sin_addr, &emptyAddr.sin_addr, sizeof(struct in_addr));
1352         }
1353         case AF_UNSPEC:
1354             return FALSE;
1355         default:
1356             FIXME("unknown address family %d\n", uaddr->sa_family);
1357             return TRUE;
1358     }
1359 }
1360
1361 /* Returns 0 if successful, -1 if the buffer is too small */
1362 static int ws_sockaddr_u2ws(const struct sockaddr* uaddr, struct WS_sockaddr* wsaddr, int* wsaddrlen)
1363 {
1364     int res;
1365
1366     switch(uaddr->sa_family)
1367     {
1368 #ifdef HAVE_IPX
1369     case AF_IPX:
1370         {
1371             const struct sockaddr_ipx* uipx=(const struct sockaddr_ipx*)uaddr;
1372             struct WS_sockaddr_ipx* wsipx=(struct WS_sockaddr_ipx*)wsaddr;
1373
1374             res=-1;
1375             switch (*wsaddrlen) /* how much can we copy? */
1376             {
1377             default:
1378                 res=0; /* enough */
1379                 *wsaddrlen = sizeof(*wsipx);
1380                 wsipx->sa_socket=uipx->sipx_port;
1381                 /* fall through */
1382             case 13:
1383             case 12:
1384                 memcpy(wsipx->sa_nodenum,uipx->sipx_node,sizeof(wsipx->sa_nodenum));
1385                 /* fall through */
1386             case 11:
1387             case 10:
1388             case 9:
1389             case 8:
1390             case 7:
1391             case 6:
1392                 memcpy(wsipx->sa_netnum,&uipx->sipx_network,sizeof(wsipx->sa_netnum));
1393                 /* fall through */
1394             case 5:
1395             case 4:
1396             case 3:
1397             case 2:
1398                 wsipx->sa_family=WS_AF_IPX;
1399                 /* fall through */
1400             case 1:
1401             case 0:
1402                 /* way too small */
1403                 break;
1404             }
1405         }
1406         break;
1407 #endif
1408 #ifdef HAVE_IRDA
1409     case AF_IRDA: {
1410         const struct sockaddr_irda *uin = (const struct sockaddr_irda *)uaddr;
1411         SOCKADDR_IRDA *win = (SOCKADDR_IRDA *)wsaddr;
1412
1413         if (*wsaddrlen < sizeof(SOCKADDR_IRDA))
1414             return -1;
1415         win->irdaAddressFamily = WS_AF_IRDA;
1416         memcpy( win->irdaDeviceID, &uin->sir_addr, sizeof(win->irdaDeviceID) );
1417         if (uin->sir_lsap_sel != LSAP_ANY)
1418             sprintf( win->irdaServiceName, "LSAP-SEL%u", uin->sir_lsap_sel );
1419         else
1420             memcpy( win->irdaServiceName, uin->sir_name,
1421                     sizeof(win->irdaServiceName) );
1422         return 0;
1423     }
1424 #endif
1425     case AF_INET6: {
1426         const struct sockaddr_in6* uin6 = (const struct sockaddr_in6*)uaddr;
1427         struct WS_sockaddr_in6_old* win6old = (struct WS_sockaddr_in6_old*)wsaddr;
1428
1429         if (*wsaddrlen < sizeof(struct WS_sockaddr_in6_old))
1430             return -1;
1431         win6old->sin6_family   = WS_AF_INET6;
1432         win6old->sin6_port     = uin6->sin6_port;
1433         win6old->sin6_flowinfo = uin6->sin6_flowinfo;
1434         memcpy(&win6old->sin6_addr,&uin6->sin6_addr,16); /* 16 bytes = 128 address bits */
1435 #ifdef HAVE_STRUCT_SOCKADDR_IN6_SIN6_SCOPE_ID
1436         if (*wsaddrlen >= sizeof(struct WS_sockaddr_in6)) {
1437             struct WS_sockaddr_in6* win6 = (struct WS_sockaddr_in6*)wsaddr;
1438             win6->sin6_scope_id = uin6->sin6_scope_id;
1439             *wsaddrlen = sizeof(struct WS_sockaddr_in6);
1440         }
1441         else
1442             *wsaddrlen = sizeof(struct WS_sockaddr_in6_old);
1443 #else
1444         *wsaddrlen = sizeof(struct WS_sockaddr_in6_old);
1445 #endif
1446         return 0;
1447     }
1448     case AF_INET: {
1449         const struct sockaddr_in* uin = (const struct sockaddr_in*)uaddr;
1450         struct WS_sockaddr_in* win = (struct WS_sockaddr_in*)wsaddr;
1451
1452         if (*wsaddrlen < sizeof(struct WS_sockaddr_in))
1453             return -1;
1454         win->sin_family = WS_AF_INET;
1455         win->sin_port   = uin->sin_port;
1456         memcpy(&win->sin_addr,&uin->sin_addr,4); /* 4 bytes = 32 address bits */
1457         memset(win->sin_zero, 0, 8); /* Make sure the null padding is null */
1458         *wsaddrlen = sizeof(struct WS_sockaddr_in);
1459         return 0;
1460     }
1461     case AF_UNSPEC: {
1462         memset(wsaddr,0,*wsaddrlen);
1463         return 0;
1464     }
1465     default:
1466         FIXME("Unknown address family %d\n", uaddr->sa_family);
1467         return -1;
1468     }
1469     return res;
1470 }
1471
1472 /**************************************************************************
1473  * Functions for handling overlapped I/O
1474  **************************************************************************/
1475
1476 /* user APC called upon async completion */
1477 static void WINAPI ws2_async_apc( void *arg, IO_STATUS_BLOCK *iosb, ULONG reserved )
1478 {
1479     ws2_async *wsa = arg;
1480
1481     if (wsa->completion_func) wsa->completion_func( NtStatusToWSAError(iosb->u.Status),
1482                                                     iosb->Information, wsa->user_overlapped,
1483                                                     wsa->flags );
1484     HeapFree( GetProcessHeap(), 0, wsa );
1485 }
1486
1487 /***********************************************************************
1488  *              WS2_recv                (INTERNAL)
1489  *
1490  * Workhorse for both synchronous and asynchronous recv() operations.
1491  */
1492 static int WS2_recv( int fd, struct ws2_async *wsa )
1493 {
1494 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
1495     char pktbuf[512];
1496 #endif
1497     struct msghdr hdr;
1498     union generic_unix_sockaddr unix_sockaddr;
1499     int n;
1500
1501     hdr.msg_name = NULL;
1502
1503     if (wsa->addr)
1504     {
1505         hdr.msg_namelen = sizeof(unix_sockaddr);
1506         hdr.msg_name = &unix_sockaddr;
1507     }
1508     else
1509         hdr.msg_namelen = 0;
1510
1511     hdr.msg_iov = wsa->iovec + wsa->first_iovec;
1512     hdr.msg_iovlen = wsa->n_iovecs - wsa->first_iovec;
1513 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
1514     hdr.msg_accrights = NULL;
1515     hdr.msg_accrightslen = 0;
1516 #else
1517     hdr.msg_control = pktbuf;
1518     hdr.msg_controllen = sizeof(pktbuf);
1519     hdr.msg_flags = 0;
1520 #endif
1521
1522     if ( (n = recvmsg(fd, &hdr, wsa->flags)) == -1 )
1523         return -1;
1524
1525 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
1526     if (wsa->control)
1527     {
1528         ERR("Message control headers cannot be properly supported on this system.\n");
1529         wsa->control->len = 0;
1530     }
1531 #else
1532     if (wsa->control && !convert_control_headers(&hdr, wsa->control))
1533     {
1534         WARN("Application passed insufficient room for control headers.\n");
1535         *wsa->lpFlags |= WS_MSG_CTRUNC;
1536         errno = EMSGSIZE;
1537         return -1;
1538     }
1539 #endif
1540
1541     /* if this socket is connected and lpFrom is not NULL, Linux doesn't give us
1542      * msg_name and msg_namelen from recvmsg, but it does set msg_namelen to zero.
1543      *
1544      * quoting linux 2.6 net/ipv4/tcp.c:
1545      *  "According to UNIX98, msg_name/msg_namelen are ignored
1546      *  on connected socket. I was just happy when found this 8) --ANK"
1547      *
1548      * likewise MSDN says that lpFrom and lpFromlen are ignored for
1549      * connection-oriented sockets, so don't try to update lpFrom.
1550      */
1551     if (wsa->addr && hdr.msg_namelen)
1552         ws_sockaddr_u2ws( &unix_sockaddr.addr, wsa->addr, wsa->addrlen.ptr );
1553
1554     return n;
1555 }
1556
1557 /***********************************************************************
1558  *              WS2_async_recv          (INTERNAL)
1559  *
1560  * Handler for overlapped recv() operations.
1561  */
1562 static NTSTATUS WS2_async_recv( void* user, IO_STATUS_BLOCK* iosb, NTSTATUS status, void **apc)
1563 {
1564     ws2_async* wsa = user;
1565     int result = 0, fd;
1566
1567     switch (status)
1568     {
1569     case STATUS_ALERTED:
1570         if ((status = wine_server_handle_to_fd( wsa->hSocket, FILE_READ_DATA, &fd, NULL ) ))
1571             break;
1572
1573         result = WS2_recv( fd, wsa );
1574         wine_server_release_fd( wsa->hSocket, fd );
1575         if (result >= 0)
1576         {
1577             status = STATUS_SUCCESS;
1578             _enable_event( wsa->hSocket, FD_READ, 0, 0 );
1579         }
1580         else
1581         {
1582             if (errno == EINTR || errno == EAGAIN)
1583             {
1584                 status = STATUS_PENDING;
1585                 _enable_event( wsa->hSocket, FD_READ, 0, 0 );
1586             }
1587             else
1588             {
1589                 result = 0;
1590                 status = wsaErrStatus();
1591             }
1592         }
1593         break;
1594     }
1595     if (status != STATUS_PENDING)
1596     {
1597         iosb->u.Status = status;
1598         iosb->Information = result;
1599         *apc = ws2_async_apc;
1600     }
1601     return status;
1602 }
1603
1604 /* user APC called upon async accept completion */
1605 static void WINAPI ws2_async_accept_apc( void *arg, IO_STATUS_BLOCK *iosb, ULONG reserved )
1606 {
1607     struct ws2_accept_async *wsa = arg;
1608
1609     HeapFree( GetProcessHeap(), 0, wsa->read );
1610     HeapFree( GetProcessHeap(), 0, wsa );
1611 }
1612
1613 /***********************************************************************
1614  *              WS2_async_accept_recv            (INTERNAL)
1615  *
1616  * This function is used to finish the read part of an accept request. It is
1617  * needed to place the completion on the correct socket (listener).
1618  */
1619 static NTSTATUS WS2_async_accept_recv( void *arg, IO_STATUS_BLOCK *iosb, NTSTATUS status, void **apc )
1620 {
1621     void *junk;
1622     struct ws2_accept_async *wsa = arg;
1623
1624     status = WS2_async_recv( wsa->read, iosb, status, &junk );
1625     if (status == STATUS_PENDING)
1626         return status;
1627
1628     if (wsa->user_overlapped->hEvent)
1629         SetEvent(wsa->user_overlapped->hEvent);
1630     if (wsa->cvalue)
1631         WS_AddCompletion( HANDLE2SOCKET(wsa->listen_socket), wsa->cvalue, iosb->u.Status, iosb->Information );
1632
1633     *apc = ws2_async_accept_apc;
1634     return status;
1635 }
1636
1637 /***********************************************************************
1638  *              WS2_async_accept                (INTERNAL)
1639  *
1640  * This is the function called to satisfy the AcceptEx callback
1641  */
1642 static NTSTATUS WS2_async_accept( void *arg, IO_STATUS_BLOCK *iosb, NTSTATUS status, void **apc )
1643 {
1644     struct ws2_accept_async *wsa = arg;
1645     int len;
1646     char *addr;
1647
1648     TRACE("status: 0x%x listen: %p, accept: %p\n", status, wsa->listen_socket, wsa->accept_socket);
1649
1650     if (status == STATUS_ALERTED)
1651     {
1652         SERVER_START_REQ( accept_into_socket )
1653         {
1654             req->lhandle = wine_server_obj_handle( wsa->listen_socket );
1655             req->ahandle = wine_server_obj_handle( wsa->accept_socket );
1656             status = wine_server_call( req );
1657         }
1658         SERVER_END_REQ;
1659
1660         if (status == STATUS_CANT_WAIT)
1661             return STATUS_PENDING;
1662
1663         if (status == STATUS_INVALID_HANDLE)
1664         {
1665             FIXME("AcceptEx accepting socket closed but request was not cancelled\n");
1666             status = STATUS_CANCELLED;
1667         }
1668     }
1669     else if (status == STATUS_HANDLES_CLOSED)
1670         status = STATUS_CANCELLED;  /* strange windows behavior */
1671
1672     if (status != STATUS_SUCCESS)
1673         goto finish;
1674
1675     /* WS2 Spec says size param is extra 16 bytes long...what do we put in it? */
1676     addr = ((char *)wsa->buf) + wsa->data_len;
1677     len = wsa->local_len - sizeof(int);
1678     WS_getsockname(HANDLE2SOCKET(wsa->accept_socket),
1679                    (struct WS_sockaddr *)(addr + sizeof(int)), &len);
1680     *(int *)addr = len;
1681
1682     addr += wsa->local_len;
1683     len = wsa->remote_len - sizeof(int);
1684     WS_getpeername(HANDLE2SOCKET(wsa->accept_socket),
1685                    (struct WS_sockaddr *)(addr + sizeof(int)), &len);
1686     *(int *)addr = len;
1687
1688     if (!wsa->read)
1689         goto finish;
1690
1691     SERVER_START_REQ( register_async )
1692     {
1693         req->type           = ASYNC_TYPE_READ;
1694         req->async.handle   = wine_server_obj_handle( wsa->accept_socket );
1695         req->async.callback = wine_server_client_ptr( WS2_async_accept_recv );
1696         req->async.iosb     = wine_server_client_ptr( iosb );
1697         req->async.arg      = wine_server_client_ptr( wsa );
1698         status = wine_server_call( req );
1699     }
1700     SERVER_END_REQ;
1701
1702     if (status != STATUS_PENDING)
1703         goto finish;
1704
1705     return STATUS_SUCCESS;
1706
1707 finish:
1708     iosb->u.Status = status;
1709     iosb->Information = 0;
1710
1711     if (wsa->user_overlapped->hEvent)
1712         SetEvent(wsa->user_overlapped->hEvent);
1713     if (wsa->cvalue)
1714         WS_AddCompletion( HANDLE2SOCKET(wsa->listen_socket), wsa->cvalue, iosb->u.Status, iosb->Information );
1715
1716     *apc = ws2_async_accept_apc;
1717     return status;
1718 }
1719
1720 /***********************************************************************
1721  *              WS2_send                (INTERNAL)
1722  *
1723  * Workhorse for both synchronous and asynchronous send() operations.
1724  */
1725 static int WS2_send( int fd, struct ws2_async *wsa )
1726 {
1727     struct msghdr hdr;
1728     union generic_unix_sockaddr unix_addr;
1729     int n, ret;
1730
1731     hdr.msg_name = NULL;
1732     hdr.msg_namelen = 0;
1733
1734     if (wsa->addr)
1735     {
1736         hdr.msg_name = &unix_addr;
1737         hdr.msg_namelen = ws_sockaddr_ws2u( wsa->addr, wsa->addrlen.val, &unix_addr );
1738         if ( !hdr.msg_namelen )
1739         {
1740             errno = EFAULT;
1741             return -1;
1742         }
1743
1744 #if defined(HAVE_IPX) && defined(SOL_IPX)
1745         if(wsa->addr->sa_family == WS_AF_IPX)
1746         {
1747             struct sockaddr_ipx* uipx = (struct sockaddr_ipx*)hdr.msg_name;
1748             int val=0;
1749             unsigned int len=sizeof(int);
1750
1751             /* The packet type is stored at the ipx socket level; At least the linux kernel seems
1752              *  to do something with it in case hdr.msg_name is NULL. Nonetheless can we use it to store
1753              *  the packet type and then we can retrieve it using getsockopt. After that we can set the
1754              *  ipx type in the sockaddr_opx structure with the stored value.
1755              */
1756             if(getsockopt(fd, SOL_IPX, IPX_TYPE, &val, &len) != -1)
1757                 uipx->sipx_type = val;
1758         }
1759 #endif
1760     }
1761
1762     hdr.msg_iov = wsa->iovec + wsa->first_iovec;
1763     hdr.msg_iovlen = wsa->n_iovecs - wsa->first_iovec;
1764 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
1765     hdr.msg_accrights = NULL;
1766     hdr.msg_accrightslen = 0;
1767 #else
1768     hdr.msg_control = NULL;
1769     hdr.msg_controllen = 0;
1770     hdr.msg_flags = 0;
1771 #endif
1772
1773     ret = sendmsg(fd, &hdr, wsa->flags);
1774     if (ret >= 0)
1775     {
1776         n = ret;
1777         while (wsa->first_iovec < wsa->n_iovecs && wsa->iovec[wsa->first_iovec].iov_len <= n)
1778             n -= wsa->iovec[wsa->first_iovec++].iov_len;
1779         if (wsa->first_iovec < wsa->n_iovecs)
1780         {
1781             wsa->iovec[wsa->first_iovec].iov_base = (char*)wsa->iovec[wsa->first_iovec].iov_base + n;
1782             wsa->iovec[wsa->first_iovec].iov_len -= n;
1783         }
1784     }
1785     return ret;
1786 }
1787
1788 /***********************************************************************
1789  *              WS2_async_send          (INTERNAL)
1790  *
1791  * Handler for overlapped send() operations.
1792  */
1793 static NTSTATUS WS2_async_send(void* user, IO_STATUS_BLOCK* iosb, NTSTATUS status, void **apc)
1794 {
1795     ws2_async* wsa = user;
1796     int result = 0, fd;
1797
1798     switch (status)
1799     {
1800     case STATUS_ALERTED:
1801         if ( wsa->n_iovecs <= wsa->first_iovec )
1802         {
1803             /* Nothing to do */
1804             status = STATUS_SUCCESS;
1805             break;
1806         }
1807         if ((status = wine_server_handle_to_fd( wsa->hSocket, FILE_WRITE_DATA, &fd, NULL ) ))
1808             break;
1809
1810         /* check to see if the data is ready (non-blocking) */
1811         result = WS2_send( fd, wsa );
1812         wine_server_release_fd( wsa->hSocket, fd );
1813
1814         if (result >= 0)
1815         {
1816             if (wsa->first_iovec < wsa->n_iovecs)
1817                 status = STATUS_PENDING;
1818             else
1819                 status = STATUS_SUCCESS;
1820
1821             iosb->Information += result;
1822         }
1823         else if (errno == EINTR || errno == EAGAIN)
1824         {
1825             status = STATUS_PENDING;
1826         }
1827         else
1828         {
1829             status = wsaErrStatus();
1830         }
1831         break;
1832     }
1833     if (status != STATUS_PENDING)
1834     {
1835         iosb->u.Status = status;
1836         *apc = ws2_async_apc;
1837     }
1838     return status;
1839 }
1840
1841 /***********************************************************************
1842  *              WS2_async_shutdown      (INTERNAL)
1843  *
1844  * Handler for shutdown() operations on overlapped sockets.
1845  */
1846 static NTSTATUS WS2_async_shutdown( void* user, PIO_STATUS_BLOCK iosb, NTSTATUS status, void **apc )
1847 {
1848     ws2_async* wsa = user;
1849     int fd, err = 1;
1850
1851     switch (status)
1852     {
1853     case STATUS_ALERTED:
1854         if ((status = wine_server_handle_to_fd( wsa->hSocket, 0, &fd, NULL ) ))
1855             break;
1856
1857         switch ( wsa->type )
1858         {
1859         case ASYNC_TYPE_READ:   err = shutdown( fd, 0 );  break;
1860         case ASYNC_TYPE_WRITE:  err = shutdown( fd, 1 );  break;
1861         }
1862         status = err ? wsaErrStatus() : STATUS_SUCCESS;
1863         wine_server_release_fd( wsa->hSocket, fd );
1864         break;
1865     }
1866     iosb->u.Status = status;
1867     iosb->Information = 0;
1868     *apc = ws2_async_apc;
1869     return status;
1870 }
1871
1872 /***********************************************************************
1873  *  WS2_register_async_shutdown         (INTERNAL)
1874  *
1875  * Helper function for WS_shutdown() on overlapped sockets.
1876  */
1877 static int WS2_register_async_shutdown( SOCKET s, int type )
1878 {
1879     struct ws2_async *wsa;
1880     NTSTATUS status;
1881
1882     TRACE("s %ld type %d\n", s, type);
1883
1884     wsa = HeapAlloc( GetProcessHeap(), 0, sizeof(*wsa) );
1885     if ( !wsa )
1886         return WSAEFAULT;
1887
1888     wsa->hSocket         = SOCKET2HANDLE(s);
1889     wsa->type            = type;
1890     wsa->completion_func = NULL;
1891
1892     SERVER_START_REQ( register_async )
1893     {
1894         req->type   = type;
1895         req->async.handle   = wine_server_obj_handle( wsa->hSocket );
1896         req->async.callback = wine_server_client_ptr( WS2_async_shutdown );
1897         req->async.iosb     = wine_server_client_ptr( &wsa->local_iosb );
1898         req->async.arg      = wine_server_client_ptr( wsa );
1899         req->async.cvalue   = 0;
1900         status = wine_server_call( req );
1901     }
1902     SERVER_END_REQ;
1903
1904     if (status != STATUS_PENDING)
1905     {
1906         HeapFree( GetProcessHeap(), 0, wsa );
1907         return NtStatusToWSAError( status );
1908     }
1909     return 0;
1910 }
1911
1912 /***********************************************************************
1913  *              accept          (WS2_32.1)
1914  */
1915 SOCKET WINAPI WS_accept(SOCKET s, struct WS_sockaddr *addr,
1916                                  int *addrlen32)
1917 {
1918     NTSTATUS status;
1919     SOCKET as;
1920     BOOL is_blocking;
1921
1922     TRACE("socket %04lx\n", s );
1923     is_blocking = _is_blocking(s);
1924
1925     do {
1926         /* try accepting first (if there is a deferred connection) */
1927         SERVER_START_REQ( accept_socket )
1928         {
1929             req->lhandle    = wine_server_obj_handle( SOCKET2HANDLE(s) );
1930             req->access     = GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE;
1931             req->attributes = OBJ_INHERIT;
1932             status = wine_server_call( req );
1933             as = HANDLE2SOCKET( wine_server_ptr_handle( reply->handle ));
1934         }
1935         SERVER_END_REQ;
1936         if (!status)
1937         {
1938             if (addr) WS_getpeername(as, addr, addrlen32);
1939             return as;
1940         }
1941         if (is_blocking && status == STATUS_CANT_WAIT)
1942         {
1943             int fd = get_sock_fd( s, FILE_READ_DATA, NULL );
1944             /* block here */
1945             do_block(fd, POLLIN, -1);
1946             _sync_sock_state(s); /* let wineserver notice connection */
1947             release_sock_fd( s, fd );
1948         }
1949     } while (is_blocking && status == STATUS_CANT_WAIT);
1950
1951     set_error(status);
1952     return INVALID_SOCKET;
1953 }
1954
1955 /***********************************************************************
1956  *     AcceptEx
1957  */
1958 static BOOL WINAPI WS2_AcceptEx(SOCKET listener, SOCKET acceptor, PVOID dest, DWORD dest_len,
1959                          DWORD local_addr_len, DWORD rem_addr_len, LPDWORD received,
1960                          LPOVERLAPPED overlapped)
1961 {
1962     DWORD status;
1963     struct ws2_accept_async *wsa;
1964     int fd;
1965     ULONG_PTR cvalue = (overlapped && ((ULONG_PTR)overlapped->hEvent & 1) == 0) ? (ULONG_PTR)overlapped : 0;
1966
1967     TRACE("(%lx, %lx, %p, %d, %d, %d, %p, %p)\n", listener, acceptor, dest, dest_len, local_addr_len,
1968                                                   rem_addr_len, received, overlapped);
1969
1970     if (!dest)
1971     {
1972         SetLastError(WSAEINVAL);
1973         return FALSE;
1974     }
1975
1976     if (!overlapped)
1977     {
1978         SetLastError(WSA_INVALID_PARAMETER);
1979         return FALSE;
1980     }
1981
1982     fd = get_sock_fd( listener, FILE_READ_DATA, NULL );
1983     if (fd == -1)
1984     {
1985         SetLastError(WSAENOTSOCK);
1986         return FALSE;
1987     }
1988     release_sock_fd( listener, fd );
1989
1990     fd = get_sock_fd( acceptor, FILE_READ_DATA, NULL );
1991     if (fd == -1)
1992     {
1993         SetLastError(WSAEINVAL);
1994         return FALSE;
1995     }
1996     release_sock_fd( acceptor, fd );
1997
1998     wsa = HeapAlloc( GetProcessHeap(), 0, sizeof(*wsa) );
1999     if(!wsa)
2000     {
2001         SetLastError(WSAEFAULT);
2002         return FALSE;
2003     }
2004
2005     wsa->listen_socket   = SOCKET2HANDLE(listener);
2006     wsa->accept_socket   = SOCKET2HANDLE(acceptor);
2007     wsa->user_overlapped = overlapped;
2008     wsa->cvalue          = cvalue;
2009     wsa->buf             = dest;
2010     wsa->data_len        = dest_len;
2011     wsa->local_len       = local_addr_len;
2012     wsa->remote_len      = rem_addr_len;
2013     wsa->read            = NULL;
2014
2015     if (wsa->data_len)
2016     {
2017         /* set up a read request if we need it */
2018         wsa->read = HeapAlloc( GetProcessHeap(), 0, FIELD_OFFSET(struct ws2_async, iovec[1]) );
2019         if (!wsa->read)
2020         {
2021             HeapFree( GetProcessHeap(), 0, wsa );
2022             SetLastError(WSAEFAULT);
2023             return FALSE;
2024         }
2025
2026         wsa->read->hSocket     = wsa->accept_socket;
2027         wsa->read->flags       = 0;
2028         wsa->read->lpFlags     = &wsa->read->flags;
2029         wsa->read->addr        = NULL;
2030         wsa->read->addrlen.ptr = NULL;
2031         wsa->read->control     = NULL;
2032         wsa->read->n_iovecs    = 1;
2033         wsa->read->first_iovec = 0;
2034         wsa->read->iovec[0].iov_base = wsa->buf;
2035         wsa->read->iovec[0].iov_len  = wsa->data_len;
2036     }
2037
2038     SERVER_START_REQ( register_async )
2039     {
2040         req->type           = ASYNC_TYPE_READ;
2041         req->async.handle   = wine_server_obj_handle( SOCKET2HANDLE(listener) );
2042         req->async.callback = wine_server_client_ptr( WS2_async_accept );
2043         req->async.iosb     = wine_server_client_ptr( overlapped );
2044         req->async.arg      = wine_server_client_ptr( wsa );
2045         /* We don't set event or completion since we may also have to read */
2046         status = wine_server_call( req );
2047     }
2048     SERVER_END_REQ;
2049
2050     if(status != STATUS_PENDING)
2051     {
2052         HeapFree( GetProcessHeap(), 0, wsa->read );
2053         HeapFree( GetProcessHeap(), 0, wsa );
2054     }
2055
2056     SetLastError( NtStatusToWSAError(status) );
2057     return FALSE;
2058 }
2059
2060 /***********************************************************************
2061  *     GetAcceptExSockaddrs
2062  */
2063 static void WINAPI WS2_GetAcceptExSockaddrs(PVOID buffer, DWORD data_size, DWORD local_size, DWORD remote_size,
2064                                      struct WS_sockaddr **local_addr, LPINT local_addr_len,
2065                                      struct WS_sockaddr **remote_addr, LPINT remote_addr_len)
2066 {
2067     char *cbuf = buffer;
2068     TRACE("(%p, %d, %d, %d, %p, %p, %p, %p)\n", buffer, data_size, local_size, remote_size, local_addr,
2069                                                 local_addr_len, remote_addr, remote_addr_len );
2070     cbuf += data_size;
2071
2072     *local_addr_len = *(int *) cbuf;
2073     *local_addr = (struct WS_sockaddr *)(cbuf + sizeof(int));
2074
2075     cbuf += local_size;
2076
2077     *remote_addr_len = *(int *) cbuf;
2078     *remote_addr = (struct WS_sockaddr *)(cbuf + sizeof(int));
2079 }
2080
2081 /***********************************************************************
2082  *     WSARecvMsg
2083  *
2084  * Perform a receive operation that is capable of returning message
2085  * control headers.  It is important to note that the WSAMSG parameter
2086  * must remain valid throughout the operation, even when an overlapped
2087  * receive is performed.
2088  */
2089 static int WINAPI WS2_WSARecvMsg( SOCKET s, LPWSAMSG msg, LPDWORD lpNumberOfBytesRecvd,
2090                                   LPWSAOVERLAPPED lpOverlapped,
2091                                   LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
2092 {
2093     if (!msg)
2094     {
2095         SetLastError( WSAEFAULT );
2096         return SOCKET_ERROR;
2097     }
2098
2099     return WS2_recv_base( s, msg->lpBuffers, msg->dwBufferCount, lpNumberOfBytesRecvd,
2100                           &msg->dwFlags, msg->name, &msg->namelen,
2101                           lpOverlapped, lpCompletionRoutine, &msg->Control );
2102 }
2103
2104 /***********************************************************************
2105  *              bind                    (WS2_32.2)
2106  */
2107 int WINAPI WS_bind(SOCKET s, const struct WS_sockaddr* name, int namelen)
2108 {
2109     int fd = get_sock_fd( s, 0, NULL );
2110     int res = SOCKET_ERROR;
2111
2112     TRACE("socket %04lx, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen);
2113
2114     if (fd != -1)
2115     {
2116         if (!name || (name->sa_family && !supported_pf(name->sa_family)))
2117         {
2118             SetLastError(WSAEAFNOSUPPORT);
2119         }
2120         else
2121         {
2122             union generic_unix_sockaddr uaddr;
2123             unsigned int uaddrlen = ws_sockaddr_ws2u(name, namelen, &uaddr);
2124             if (!uaddrlen)
2125             {
2126                 SetLastError(WSAEFAULT);
2127             }
2128             else
2129             {
2130 #ifdef IPV6_V6ONLY
2131                 const struct sockaddr_in6 *in6 = (const struct sockaddr_in6*) &uaddr;
2132                 if (name->sa_family == WS_AF_INET6 &&
2133                     !memcmp(&in6->sin6_addr, &in6addr_any, sizeof(struct in6_addr)))
2134                 {
2135                     int enable = 1;
2136                     if (setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, &enable, sizeof(enable)) == -1)
2137                     {
2138                         release_sock_fd( s, fd );
2139                         SetLastError(WSAEAFNOSUPPORT);
2140                         return SOCKET_ERROR;
2141                     }
2142                 }
2143 #endif
2144                 if (name->sa_family == WS_AF_INET)
2145                 {
2146                     struct sockaddr_in *in4 = (struct sockaddr_in*) &uaddr;
2147                     if (memcmp(&in4->sin_addr, magic_loopback_addr, 4) == 0)
2148                     {
2149                         /* Trying to bind to the default host interface, using
2150                          * INADDR_ANY instead*/
2151                         WARN("Trying to bind to magic IP address, using "
2152                              "INADDR_ANY instead.\n");
2153                         in4->sin_addr.s_addr = htonl(WS_INADDR_ANY);
2154                     }
2155                 }
2156                 if (bind(fd, &uaddr.addr, uaddrlen) < 0)
2157                 {
2158                     int loc_errno = errno;
2159                     WARN("\tfailure - errno = %i\n", errno);
2160                     errno = loc_errno;
2161                     switch (errno)
2162                     {
2163                     case EBADF:
2164                         SetLastError(WSAENOTSOCK);
2165                         break;
2166                     case EADDRNOTAVAIL:
2167                         SetLastError(WSAEINVAL);
2168                         break;
2169                     default:
2170                         SetLastError(wsaErrno());
2171                         break;
2172                     }
2173                 }
2174                 else
2175                 {
2176                     res=0; /* success */
2177                 }
2178             }
2179         }
2180         release_sock_fd( s, fd );
2181     }
2182     return res;
2183 }
2184
2185 /***********************************************************************
2186  *              closesocket             (WS2_32.3)
2187  */
2188 int WINAPI WS_closesocket(SOCKET s)
2189 {
2190     TRACE("socket %04lx\n", s);
2191     if (CloseHandle(SOCKET2HANDLE(s))) return 0;
2192     return SOCKET_ERROR;
2193 }
2194
2195 static int do_connect(int fd, const struct WS_sockaddr* name, int namelen)
2196 {
2197     union generic_unix_sockaddr uaddr;
2198     unsigned int uaddrlen = ws_sockaddr_ws2u(name, namelen, &uaddr);
2199
2200     if (!uaddrlen)
2201         return WSAEFAULT;
2202
2203     if (name->sa_family == WS_AF_INET)
2204     {
2205         struct sockaddr_in *in4 = (struct sockaddr_in*) &uaddr;
2206         if (memcmp(&in4->sin_addr, magic_loopback_addr, 4) == 0)
2207         {
2208             /* Trying to connect to magic replace-loopback address,
2209                 * assuming we really want to connect to localhost */
2210             TRACE("Trying to connect to magic IP address, using "
2211                     "INADDR_LOOPBACK instead.\n");
2212             in4->sin_addr.s_addr = htonl(WS_INADDR_LOOPBACK);
2213         }
2214     }
2215
2216     if (connect(fd, &uaddr.addr, uaddrlen) == 0)
2217         return 0;
2218
2219     return wsaErrno();
2220 }
2221
2222 /***********************************************************************
2223  *              connect         (WS2_32.4)
2224  */
2225 int WINAPI WS_connect(SOCKET s, const struct WS_sockaddr* name, int namelen)
2226 {
2227     int fd = get_sock_fd( s, FILE_READ_DATA, NULL );
2228
2229     TRACE("socket %04lx, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen);
2230
2231     if (fd != -1)
2232     {
2233         int ret = do_connect(fd, name, namelen);
2234         if (ret == 0)
2235             goto connect_success;
2236
2237         if (ret == WSAEINPROGRESS)
2238         {
2239             /* tell wineserver that a connection is in progress */
2240             _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
2241                           FD_CONNECT,
2242                           FD_WINE_CONNECTED|FD_WINE_LISTENING);
2243             if (_is_blocking(s))
2244             {
2245                 int result;
2246                 /* block here */
2247                 do_block(fd, POLLIN | POLLOUT, -1);
2248                 _sync_sock_state(s); /* let wineserver notice connection */
2249                 /* retrieve any error codes from it */
2250                 result = _get_sock_error(s, FD_CONNECT_BIT);
2251                 if (result)
2252                     SetLastError(NtStatusToWSAError(result));
2253                 else
2254                 {
2255                     goto connect_success;
2256                 }
2257             }
2258             else
2259             {
2260                 SetLastError(WSAEWOULDBLOCK);
2261             }
2262         }
2263         else
2264         {
2265             SetLastError(ret);
2266         }
2267         release_sock_fd( s, fd );
2268     }
2269     return SOCKET_ERROR;
2270
2271 connect_success:
2272     release_sock_fd( s, fd );
2273     _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
2274                   FD_WINE_CONNECTED|FD_READ|FD_WRITE,
2275                   FD_CONNECT|FD_WINE_LISTENING);
2276     return 0;
2277 }
2278
2279 /***********************************************************************
2280  *              WSAConnect             (WS2_32.30)
2281  */
2282 int WINAPI WSAConnect( SOCKET s, const struct WS_sockaddr* name, int namelen,
2283                        LPWSABUF lpCallerData, LPWSABUF lpCalleeData,
2284                        LPQOS lpSQOS, LPQOS lpGQOS )
2285 {
2286     if ( lpCallerData || lpCalleeData || lpSQOS || lpGQOS )
2287         FIXME("unsupported parameters!\n");
2288     return WS_connect( s, name, namelen );
2289 }
2290
2291 /***********************************************************************
2292  *             ConnectEx
2293  */
2294 static BOOL WINAPI WS2_ConnectEx(SOCKET s, const struct WS_sockaddr* name, int namelen,
2295                           PVOID sendBuf, DWORD sendBufLen, LPDWORD sent, LPOVERLAPPED ov)
2296 {
2297     int fd, ret, status;
2298
2299     if (!ov)
2300     {
2301         SetLastError( ERROR_INVALID_PARAMETER );
2302         return FALSE;
2303     }
2304
2305     fd = get_sock_fd( s, FILE_READ_DATA, NULL );
2306     if (fd == -1)
2307     {
2308         SetLastError( WSAENOTSOCK );
2309         return FALSE;
2310     }
2311
2312     TRACE("socket %04lx, ptr %p %s, length %d, sendptr %p, len %d, ov %p\n",
2313           s, name, debugstr_sockaddr(name), namelen, sendBuf, sendBufLen, ov);
2314
2315     /* FIXME: technically the socket has to be bound */
2316     ret = do_connect(fd, name, namelen);
2317     if (ret == 0)
2318     {
2319         WSABUF wsabuf;
2320
2321         _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
2322                             FD_WINE_CONNECTED|FD_READ|FD_WRITE,
2323                             FD_CONNECT|FD_WINE_LISTENING);
2324
2325         wsabuf.len = sendBufLen;
2326         wsabuf.buf = (char*) sendBuf;
2327
2328         /* WSASend takes care of completion if need be */
2329         if (WSASend(s, &wsabuf, sendBuf ? 1 : 0, sent, 0, ov, NULL) != SOCKET_ERROR)
2330             goto connection_success;
2331     }
2332     else if (ret == WSAEINPROGRESS)
2333     {
2334         struct ws2_async *wsa;
2335         ULONG_PTR cvalue = (((ULONG_PTR)ov->hEvent & 1) == 0) ? (ULONG_PTR)ov : 0;
2336
2337         _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
2338                       FD_CONNECT,
2339                       FD_WINE_CONNECTED|FD_WINE_LISTENING);
2340
2341         /* Indirectly call WSASend */
2342         if (!(wsa = HeapAlloc( GetProcessHeap(), 0, sizeof(*wsa) )))
2343         {
2344             SetLastError(WSAEFAULT);
2345         }
2346         else
2347         {
2348             IO_STATUS_BLOCK *iosb = (IO_STATUS_BLOCK *)ov;
2349             iosb->u.Status = STATUS_PENDING;
2350             iosb->Information = 0;
2351
2352             wsa->hSocket     = SOCKET2HANDLE(s);
2353             wsa->addr        = NULL;
2354             wsa->addrlen.val = 0;
2355             wsa->flags       = 0;
2356             wsa->lpFlags     = &wsa->flags;
2357             wsa->control     = NULL;
2358             wsa->n_iovecs    = sendBuf ? 1 : 0;
2359             wsa->first_iovec = 0;
2360             wsa->completion_func = NULL;
2361             wsa->iovec[0].iov_base = sendBuf;
2362             wsa->iovec[0].iov_len  = sendBufLen;
2363
2364             SERVER_START_REQ( register_async )
2365             {
2366                 req->type           = ASYNC_TYPE_WRITE;
2367                 req->async.handle   = wine_server_obj_handle( wsa->hSocket );
2368                 req->async.callback = wine_server_client_ptr( WS2_async_send );
2369                 req->async.iosb     = wine_server_client_ptr( iosb );
2370                 req->async.arg      = wine_server_client_ptr( wsa );
2371                 req->async.event    = wine_server_obj_handle( ov->hEvent );
2372                 req->async.cvalue   = cvalue;
2373                 status = wine_server_call( req );
2374             }
2375             SERVER_END_REQ;
2376
2377             if (status != STATUS_PENDING) HeapFree(GetProcessHeap(), 0, wsa);
2378
2379             /* If the connect already failed */
2380             if (status == STATUS_PIPE_DISCONNECTED)
2381                 status = _get_sock_error(s, FD_CONNECT_BIT);
2382             SetLastError( NtStatusToWSAError(status) );
2383         }
2384     }
2385     else
2386     {
2387         SetLastError(ret);
2388     }
2389
2390     release_sock_fd( s, fd );
2391     return FALSE;
2392
2393 connection_success:
2394     release_sock_fd( s, fd );
2395     return TRUE;
2396 }
2397
2398
2399 /***********************************************************************
2400  *              getpeername             (WS2_32.5)
2401  */
2402 int WINAPI WS_getpeername(SOCKET s, struct WS_sockaddr *name, int *namelen)
2403 {
2404     int fd;
2405     int res;
2406
2407     TRACE("socket: %04lx, ptr %p, len %08x\n", s, name, namelen?*namelen:0);
2408
2409     fd = get_sock_fd( s, 0, NULL );
2410     res = SOCKET_ERROR;
2411
2412     if (fd != -1)
2413     {
2414         union generic_unix_sockaddr uaddr;
2415         unsigned int uaddrlen = sizeof(uaddr);
2416
2417         if (getpeername(fd, &uaddr.addr, &uaddrlen) == 0)
2418         {
2419             if (!name || !namelen)
2420                 SetLastError(WSAEFAULT);
2421             else if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0)
2422                 /* The buffer was too small */
2423                 SetLastError(WSAEFAULT);
2424             else
2425                 res = 0;
2426         }
2427         else
2428             SetLastError(wsaErrno());
2429         release_sock_fd( s, fd );
2430     }
2431     return res;
2432 }
2433
2434 /***********************************************************************
2435  *              getsockname             (WS2_32.6)
2436  */
2437 int WINAPI WS_getsockname(SOCKET s, struct WS_sockaddr *name, int *namelen)
2438 {
2439     int fd;
2440     int res;
2441
2442     TRACE("socket: %04lx, ptr %p, len %8x\n", s, name, *namelen);
2443
2444     /* Check if what we've received is valid. Should we use IsBadReadPtr? */
2445     if( (name == NULL) || (namelen == NULL) )
2446     {
2447         SetLastError( WSAEFAULT );
2448         return SOCKET_ERROR;
2449     }
2450
2451     fd = get_sock_fd( s, 0, NULL );
2452     res = SOCKET_ERROR;
2453
2454     if (fd != -1)
2455     {
2456         union generic_unix_sockaddr uaddr;
2457         unsigned int uaddrlen = sizeof(uaddr);
2458
2459         if (getsockname(fd, &uaddr.addr, &uaddrlen) != 0)
2460         {
2461             SetLastError(wsaErrno());
2462         }
2463         else if (!is_sockaddr_bound(&uaddr.addr, uaddrlen))
2464         {
2465             SetLastError(WSAEINVAL);
2466         }
2467         else if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0)
2468         {
2469             /* The buffer was too small */
2470             SetLastError(WSAEFAULT);
2471         }
2472         else
2473         {
2474             res=0;
2475         }
2476         release_sock_fd( s, fd );
2477     }
2478     return res;
2479 }
2480
2481 /***********************************************************************
2482  *              getsockopt              (WS2_32.7)
2483  */
2484 INT WINAPI WS_getsockopt(SOCKET s, INT level,
2485                                   INT optname, char *optval, INT *optlen)
2486 {
2487     int fd;
2488     INT ret = 0;
2489
2490     TRACE("socket: %04lx, level 0x%x, name 0x%x, ptr %p, len %d\n",
2491           s, level, optname, optval, *optlen);
2492
2493     switch(level)
2494     {
2495     case WS_SOL_SOCKET:
2496     {
2497         switch(optname)
2498         {
2499         /* Handle common cases. The special cases are below, sorted
2500          * alphabetically */
2501         case WS_SO_ACCEPTCONN:
2502         case WS_SO_BROADCAST:
2503         case WS_SO_DEBUG:
2504         case WS_SO_ERROR:
2505         case WS_SO_KEEPALIVE:
2506         case WS_SO_OOBINLINE:
2507         case WS_SO_RCVBUF:
2508         case WS_SO_REUSEADDR:
2509         case WS_SO_SNDBUF:
2510         case WS_SO_TYPE:
2511             if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2512                 return SOCKET_ERROR;
2513             convert_sockopt(&level, &optname);
2514             if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
2515             {
2516                 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2517                 ret = SOCKET_ERROR;
2518             }
2519             release_sock_fd( s, fd );
2520             return ret;
2521
2522         case WS_SO_DONTLINGER:
2523         {
2524             struct linger lingval;
2525             unsigned int len = sizeof(struct linger);
2526
2527             if (!optlen || *optlen < sizeof(BOOL)|| !optval)
2528             {
2529                 SetLastError(WSAEFAULT);
2530                 return SOCKET_ERROR;
2531             }
2532             if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2533                 return SOCKET_ERROR;
2534
2535             if (getsockopt(fd, SOL_SOCKET, SO_LINGER, &lingval, &len) != 0 )
2536             {
2537                 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2538                 ret = SOCKET_ERROR;
2539             }
2540             else
2541             {
2542                 *(BOOL *)optval = (lingval.l_onoff) ? FALSE : TRUE;
2543                 *optlen = sizeof(BOOL);
2544             }
2545
2546             release_sock_fd( s, fd );
2547             return ret;
2548         }
2549
2550         case WS_SO_CONNECT_TIME:
2551         {
2552             static int pretendtime = 0;
2553
2554             if (!pretendtime) FIXME("WS_SO_CONNECT_TIME - faking results\n");
2555             if (!optlen || *optlen < sizeof(DWORD) || !optval)
2556             {
2557                 SetLastError(WSAEFAULT);
2558                 return SOCKET_ERROR;
2559             }
2560             *(DWORD*)optval = pretendtime++;
2561             *optlen = sizeof(DWORD);
2562             return ret;
2563         }
2564         /* As mentioned in setsockopt, Windows ignores this, so we
2565          * always return true here */
2566         case WS_SO_DONTROUTE:
2567             if (!optlen || *optlen < sizeof(BOOL) || !optval)
2568             {
2569                 SetLastError(WSAEFAULT);
2570                 return SOCKET_ERROR;
2571             }
2572             *(BOOL *)optval = TRUE;
2573             *optlen = sizeof(BOOL);
2574             return 0;
2575
2576         case WS_SO_LINGER:
2577         {
2578             struct linger lingval;
2579             int so_type;
2580             unsigned int len = sizeof(struct linger), slen = sizeof(int);
2581
2582             /* struct linger and LINGER have different sizes */
2583             if (!optlen || *optlen < sizeof(LINGER) || !optval)
2584             {
2585                 SetLastError(WSAEFAULT);
2586                 return SOCKET_ERROR;
2587             }
2588             if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2589                 return SOCKET_ERROR;
2590
2591             if ((getsockopt(fd, SOL_SOCKET, SO_TYPE, &so_type, &slen) == 0 && so_type == SOCK_DGRAM))
2592             {
2593                 SetLastError(WSAENOPROTOOPT);
2594                 ret = SOCKET_ERROR;
2595             }
2596             else if (getsockopt(fd, SOL_SOCKET, SO_LINGER, &lingval, &len) != 0)
2597             {
2598                 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2599                 ret = SOCKET_ERROR;
2600             }
2601             else
2602             {
2603                 ((LINGER *)optval)->l_onoff = lingval.l_onoff;
2604                 ((LINGER *)optval)->l_linger = lingval.l_linger;
2605                 *optlen = sizeof(struct linger);
2606             }
2607
2608             release_sock_fd( s, fd );
2609             return ret;
2610         }
2611
2612         case WS_SO_MAX_MSG_SIZE:
2613             if (!optlen || *optlen < sizeof(int) || !optval)
2614             {
2615                 SetLastError(WSAEFAULT);
2616                 return SOCKET_ERROR;
2617             }
2618             TRACE("getting global SO_MAX_MSG_SIZE = 65507\n");
2619             *(int *)optval = 65507;
2620             *optlen = sizeof(int);
2621             return 0;
2622
2623         /* SO_OPENTYPE does not require a valid socket handle. */
2624         case WS_SO_OPENTYPE:
2625             if (!optlen || *optlen < sizeof(int) || !optval)
2626             {
2627                 SetLastError(WSAEFAULT);
2628                 return SOCKET_ERROR;
2629             }
2630             *(int *)optval = get_per_thread_data()->opentype;
2631             *optlen = sizeof(int);
2632             TRACE("getting global SO_OPENTYPE = 0x%x\n", *((int*)optval) );
2633             return 0;
2634
2635 #ifdef SO_RCVTIMEO
2636         case WS_SO_RCVTIMEO:
2637 #endif
2638 #ifdef SO_SNDTIMEO
2639         case WS_SO_SNDTIMEO:
2640 #endif
2641 #if defined(SO_RCVTIMEO) || defined(SO_SNDTIMEO)
2642         {
2643             struct timeval tv;
2644             unsigned int len = sizeof(struct timeval);
2645
2646             if (!optlen || *optlen < sizeof(int)|| !optval)
2647             {
2648                 SetLastError(WSAEFAULT);
2649                 return SOCKET_ERROR;
2650             }
2651             if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2652                 return SOCKET_ERROR;
2653
2654             convert_sockopt(&level, &optname);
2655             if (getsockopt(fd, level, optname, &tv, &len) != 0 )
2656             {
2657                 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2658                 ret = SOCKET_ERROR;
2659             }
2660             else
2661             {
2662                 *(int *)optval = tv.tv_sec * 1000 + tv.tv_usec / 1000;
2663                 *optlen = sizeof(int);
2664             }
2665
2666             release_sock_fd( s, fd );
2667             return ret;
2668         }
2669 #endif
2670         default:
2671             TRACE("Unknown SOL_SOCKET optname: 0x%08x\n", optname);
2672             SetLastError(WSAENOPROTOOPT);
2673             return SOCKET_ERROR;
2674         } /* end switch(optname) */
2675     }/* end case WS_SOL_SOCKET */
2676 #ifdef HAVE_IPX
2677     case NSPROTO_IPX:
2678     {
2679         struct WS_sockaddr_ipx addr;
2680         IPX_ADDRESS_DATA *data;
2681         int namelen;
2682         switch(optname)
2683         {
2684         case IPX_PTYPE:
2685             if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR;
2686 #ifdef SOL_IPX
2687             if(getsockopt(fd, SOL_IPX, IPX_TYPE, optval, (unsigned int*)optlen) == -1)
2688             {
2689                 ret = SOCKET_ERROR;
2690             }
2691 #else
2692             {
2693                 struct ipx val;
2694                 socklen_t len=sizeof(struct ipx);
2695                 if(getsockopt(fd, 0, SO_DEFAULT_HEADERS, &val, &len) == -1 )
2696                     ret = SOCKET_ERROR;
2697                 else
2698                     *optval = (int)val.ipx_pt;
2699             }
2700 #endif
2701             TRACE("ptype: %d (fd: %d)\n", *(int*)optval, fd);
2702             release_sock_fd( s, fd );
2703             return ret;
2704
2705         case IPX_ADDRESS:
2706             /*
2707             *  On a Win2000 system with one network card there are usually
2708             *  three ipx devices one with a speed of 28.8kbps, 10Mbps and 100Mbps.
2709             *  Using this call you can then retrieve info about this all.
2710             *  In case of Linux it is a bit different. Usually you have
2711             *  only "one" device active and further it is not possible to
2712             *  query things like the linkspeed.
2713             */
2714             FIXME("IPX_ADDRESS\n");
2715             namelen = sizeof(struct WS_sockaddr_ipx);
2716             memset(&addr, 0, sizeof(struct WS_sockaddr_ipx));
2717             WS_getsockname(s, (struct WS_sockaddr*)&addr, &namelen);
2718
2719             data = (IPX_ADDRESS_DATA*)optval;
2720                     memcpy(data->nodenum,addr.sa_nodenum,sizeof(data->nodenum));
2721                     memcpy(data->netnum,addr.sa_netnum,sizeof(data->netnum));
2722             data->adapternum = 0;
2723             data->wan = FALSE; /* We are not on a wan for now .. */
2724             data->status = FALSE; /* Since we are not on a wan, the wan link isn't up */
2725             data->maxpkt = 1467; /* This value is the default one, at least on Win2k/WinXP */
2726             data->linkspeed = 100000; /* Set the line speed in 100bit/s to 10 Mbit;
2727                                        * note 1MB = 1000kB in this case */
2728             return 0;
2729
2730         case IPX_MAX_ADAPTER_NUM:
2731             FIXME("IPX_MAX_ADAPTER_NUM\n");
2732             *(int*)optval = 1; /* As noted under IPX_ADDRESS we have just one card. */
2733             return 0;
2734
2735         default:
2736             FIXME("IPX optname:%x\n", optname);
2737             return SOCKET_ERROR;
2738         }/* end switch(optname) */
2739     } /* end case NSPROTO_IPX */
2740 #endif
2741
2742 #ifdef HAVE_IRDA
2743     case WS_SOL_IRLMP:
2744         switch(optname)
2745         {
2746         case WS_IRLMP_ENUMDEVICES:
2747         {
2748             static const int MAX_IRDA_DEVICES = 10;
2749             char buf[sizeof(struct irda_device_list) +
2750                      (MAX_IRDA_DEVICES - 1) * sizeof(struct irda_device_info)];
2751             int fd, res;
2752             socklen_t len = sizeof(buf);
2753
2754             if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2755                 return SOCKET_ERROR;
2756             res = getsockopt( fd, SOL_IRLMP, IRLMP_ENUMDEVICES, buf, &len );
2757             if (res < 0)
2758             {
2759                 SetLastError(wsaErrno());
2760                 return SOCKET_ERROR;
2761             }
2762             else
2763             {
2764                 struct irda_device_list *src = (struct irda_device_list *)buf;
2765                 DEVICELIST *dst = (DEVICELIST *)optval;
2766                 INT needed = sizeof(DEVICELIST), i;
2767
2768                 if (src->len > 0)
2769                     needed += (src->len - 1) * sizeof(IRDA_DEVICE_INFO);
2770                 if (*optlen < needed)
2771                 {
2772                     SetLastError(WSAEFAULT);
2773                     return SOCKET_ERROR;
2774                 }
2775                 *optlen = needed;
2776                 TRACE("IRLMP_ENUMDEVICES: %d devices found:\n", src->len);
2777                 dst->numDevice = src->len;
2778                 for (i = 0; i < src->len; i++)
2779                 {
2780                     TRACE("saddr = %08x, daddr = %08x, info = %s, hints = %02x%02x\n",
2781                           src->dev[i].saddr, src->dev[i].daddr,
2782                           src->dev[i].info, src->dev[i].hints[0],
2783                           src->dev[i].hints[1]);
2784                     memcpy( dst->Device[i].irdaDeviceID,
2785                             &src->dev[i].daddr,
2786                             sizeof(dst->Device[i].irdaDeviceID) ) ;
2787                     memcpy( dst->Device[i].irdaDeviceName,
2788                             src->dev[i].info,
2789                             sizeof(dst->Device[i].irdaDeviceName) ) ;
2790                     memcpy( &dst->Device[i].irdaDeviceHints1,
2791                             &src->dev[i].hints[0],
2792                             sizeof(dst->Device[i].irdaDeviceHints1) ) ;
2793                     memcpy( &dst->Device[i].irdaDeviceHints2,
2794                             &src->dev[i].hints[1],
2795                             sizeof(dst->Device[i].irdaDeviceHints2) ) ;
2796                     dst->Device[i].irdaCharSet = src->dev[i].charset;
2797                 }
2798                 return 0;
2799             }
2800         }
2801         default:
2802             FIXME("IrDA optname:0x%x\n", optname);
2803             return SOCKET_ERROR;
2804         }
2805         break; /* case WS_SOL_IRLMP */
2806 #endif
2807
2808     /* Levels WS_IPPROTO_TCP and WS_IPPROTO_IP convert directly */
2809     case WS_IPPROTO_TCP:
2810         switch(optname)
2811         {
2812         case WS_TCP_NODELAY:
2813             if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2814                 return SOCKET_ERROR;
2815             convert_sockopt(&level, &optname);
2816             if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
2817             {
2818                 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2819                 ret = SOCKET_ERROR;
2820             }
2821             release_sock_fd( s, fd );
2822             return ret;
2823         }
2824         FIXME("Unknown IPPROTO_TCP optname 0x%08x\n", optname);
2825         return SOCKET_ERROR;
2826
2827     case WS_IPPROTO_IP:
2828         switch(optname)
2829         {
2830         case WS_IP_ADD_MEMBERSHIP:
2831         case WS_IP_DROP_MEMBERSHIP:
2832 #ifdef IP_HDRINCL
2833         case WS_IP_HDRINCL:
2834 #endif
2835         case WS_IP_MULTICAST_IF:
2836         case WS_IP_MULTICAST_LOOP:
2837         case WS_IP_MULTICAST_TTL:
2838         case WS_IP_OPTIONS:
2839 #ifdef IP_PKTINFO
2840         case WS_IP_PKTINFO:
2841 #endif
2842         case WS_IP_TOS:
2843         case WS_IP_TTL:
2844             if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2845                 return SOCKET_ERROR;
2846             convert_sockopt(&level, &optname);
2847             if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
2848             {
2849                 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2850                 ret = SOCKET_ERROR;
2851             }
2852             release_sock_fd( s, fd );
2853             return ret;
2854         case WS_IP_DONTFRAGMENT:
2855             FIXME("WS_IP_DONTFRAGMENT is always false!\n");
2856             *(BOOL*)optval = FALSE;
2857             return 0;
2858         }
2859         FIXME("Unknown IPPROTO_IP optname 0x%08x\n", optname);
2860         return SOCKET_ERROR;
2861
2862     case WS_IPPROTO_IPV6:
2863         switch(optname)
2864         {
2865 #ifdef IPV6_ADD_MEMBERSHIP
2866         case WS_IPV6_ADD_MEMBERSHIP:
2867 #endif
2868 #ifdef IPV6_DROP_MEMBERSHIP
2869         case WS_IPV6_DROP_MEMBERSHIP:
2870 #endif
2871         case WS_IPV6_MULTICAST_IF:
2872         case WS_IPV6_MULTICAST_HOPS:
2873         case WS_IPV6_MULTICAST_LOOP:
2874         case WS_IPV6_UNICAST_HOPS:
2875         case WS_IPV6_V6ONLY:
2876             if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2877                 return SOCKET_ERROR;
2878             convert_sockopt(&level, &optname);
2879             if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
2880             {
2881                 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2882                 ret = SOCKET_ERROR;
2883             }
2884             release_sock_fd( s, fd );
2885             return ret;
2886         case WS_IPV6_DONTFRAG:
2887             FIXME("WS_IPV6_DONTFRAG is always false!\n");
2888             *(BOOL*)optval = FALSE;
2889             return 0;
2890         }
2891         FIXME("Unknown IPPROTO_IPV6 optname 0x%08x\n", optname);
2892         return SOCKET_ERROR;
2893
2894     default:
2895         WARN("Unknown level: 0x%08x\n", level);
2896         SetLastError(WSAEINVAL);
2897         return SOCKET_ERROR;
2898     } /* end switch(level) */
2899 }
2900
2901 /***********************************************************************
2902  *              htonl                   (WS2_32.8)
2903  */
2904 WS_u_long WINAPI WS_htonl(WS_u_long hostlong)
2905 {
2906     return htonl(hostlong);
2907 }
2908
2909
2910 /***********************************************************************
2911  *              htons                   (WS2_32.9)
2912  */
2913 WS_u_short WINAPI WS_htons(WS_u_short hostshort)
2914 {
2915     return htons(hostshort);
2916 }
2917
2918 /***********************************************************************
2919  *              WSAHtonl                (WS2_32.46)
2920  *  From MSDN description of error codes, this function should also
2921  *  check if WinSock has been initialized and the socket is a valid
2922  *  socket. But why? This function only translates a host byte order
2923  *  u_long into a network byte order u_long...
2924  */
2925 int WINAPI WSAHtonl(SOCKET s, WS_u_long hostlong, WS_u_long *lpnetlong)
2926 {
2927     if (lpnetlong)
2928     {
2929         *lpnetlong = htonl(hostlong);
2930         return 0;
2931     }
2932     WSASetLastError(WSAEFAULT);
2933     return SOCKET_ERROR;
2934 }
2935
2936 /***********************************************************************
2937  *              WSAHtons                (WS2_32.47)
2938  *  From MSDN description of error codes, this function should also
2939  *  check if WinSock has been initialized and the socket is a valid
2940  *  socket. But why? This function only translates a host byte order
2941  *  u_short into a network byte order u_short...
2942  */
2943 int WINAPI WSAHtons(SOCKET s, WS_u_short hostshort, WS_u_short *lpnetshort)
2944 {
2945
2946     if (lpnetshort)
2947     {
2948         *lpnetshort = htons(hostshort);
2949         return 0;
2950     }
2951     WSASetLastError(WSAEFAULT);
2952     return SOCKET_ERROR;
2953 }
2954
2955
2956 /***********************************************************************
2957  *              inet_addr               (WS2_32.11)
2958  */
2959 WS_u_long WINAPI WS_inet_addr(const char *cp)
2960 {
2961     if (!cp) return INADDR_NONE;
2962     return inet_addr(cp);
2963 }
2964
2965
2966 /***********************************************************************
2967  *              ntohl                   (WS2_32.14)
2968  */
2969 WS_u_long WINAPI WS_ntohl(WS_u_long netlong)
2970 {
2971     return ntohl(netlong);
2972 }
2973
2974
2975 /***********************************************************************
2976  *              ntohs                   (WS2_32.15)
2977  */
2978 WS_u_short WINAPI WS_ntohs(WS_u_short netshort)
2979 {
2980     return ntohs(netshort);
2981 }
2982
2983
2984 /***********************************************************************
2985  *              inet_ntoa               (WS2_32.12)
2986  */
2987 char* WINAPI WS_inet_ntoa(struct WS_in_addr in)
2988 {
2989   /* use "buffer for dummies" here because some applications have a
2990    * propensity to decode addresses in ws_hostent structure without
2991    * saving them first...
2992    */
2993     static char dbuffer[16]; /* Yes, 16: 4*3 digits + 3 '.' + 1 '\0' */
2994
2995     char* s = inet_ntoa(*((struct in_addr*)&in));
2996     if( s )
2997     {
2998         strcpy(dbuffer, s);
2999         return dbuffer;
3000     }
3001     SetLastError(wsaErrno());
3002     return NULL;
3003 }
3004
3005 static const char *debugstr_wsaioctl(DWORD ioctl)
3006 {
3007     const char *buf_type, *family;
3008
3009     switch(ioctl & 0x18000000)
3010     {
3011     case WS_IOC_WS2:
3012         family = "IOC_WS2";
3013         break;
3014     case WS_IOC_PROTOCOL:
3015         family = "IOC_PROTOCOL";
3016         break;
3017     case WS_IOC_VENDOR:
3018         family = "IOC_VENDOR";
3019         break;
3020     default: /* WS_IOC_UNIX */
3021     {
3022         BYTE size = (ioctl >> 16) & WS_IOCPARM_MASK;
3023         char x = (ioctl & 0xff00) >> 8;
3024         BYTE y = ioctl & 0xff;
3025         char args[14];
3026
3027         switch (ioctl & (WS_IOC_VOID|WS_IOC_INOUT))
3028         {
3029             case WS_IOC_VOID:
3030                 buf_type = "_IO";
3031                 sprintf(args, "%d, %d", x, y);
3032                 break;
3033             case WS_IOC_IN:
3034                 buf_type = "_IOW";
3035                 sprintf(args, "'%c', %d, %d", x, y, size);
3036                 break;
3037             case WS_IOC_OUT:
3038                 buf_type = "_IOR";
3039                 sprintf(args, "'%c', %d, %d", x, y, size);
3040                 break;
3041             default:
3042                 buf_type = "?";
3043                 sprintf(args, "'%c', %d, %d", x, y, size);
3044                 break;
3045         }
3046         return wine_dbg_sprintf("%s(%s)", buf_type, args);
3047     }
3048     }
3049
3050     /* We are different from WS_IOC_UNIX. */
3051     switch (ioctl & (WS_IOC_VOID|WS_IOC_INOUT))
3052     {
3053         case WS_IOC_VOID:
3054             buf_type = "_WSAIO";
3055             break;
3056         case WS_IOC_INOUT:
3057             buf_type = "_WSAIORW";
3058             break;
3059         case WS_IOC_IN:
3060             buf_type = "_WSAIOW";
3061             break;
3062         case WS_IOC_OUT:
3063             buf_type = "_WSAIOR";
3064             break;
3065         default:
3066             buf_type = "?";
3067             break;
3068     }
3069
3070     return wine_dbg_sprintf("%s(%s, %d)", buf_type, family,
3071                             (USHORT)(ioctl & 0xffff));
3072 }
3073
3074 /**********************************************************************
3075  *              WSAIoctl                (WS2_32.50)
3076  *
3077  */
3078 INT WINAPI WSAIoctl(SOCKET s, DWORD code, LPVOID in_buff, DWORD in_size, LPVOID out_buff,
3079                     DWORD out_size, LPDWORD ret_size, LPWSAOVERLAPPED overlapped,
3080                     LPWSAOVERLAPPED_COMPLETION_ROUTINE completion )
3081 {
3082     int fd;
3083     DWORD status = 0, total = 0;
3084
3085     TRACE("%ld, 0x%08x, %p, %d, %p, %d, %p, %p, %p\n",
3086           s, code, in_buff, in_size, out_buff, out_size, ret_size, overlapped, completion);
3087
3088     switch (code)
3089     {
3090     case WS_FIONBIO:
3091         if (in_size != sizeof(WS_u_long) || IS_INTRESOURCE(in_buff))
3092         {
3093             WSASetLastError(WSAEFAULT);
3094             return SOCKET_ERROR;
3095         }
3096         if (_get_sock_mask(s))
3097         {
3098             /* AsyncSelect()'ed sockets are always nonblocking */
3099             if (!*(WS_u_long *)in_buff) status = WSAEINVAL;
3100             break;
3101         }
3102         if (*(WS_u_long *)in_buff)
3103             _enable_event(SOCKET2HANDLE(s), 0, FD_WINE_NONBLOCKING, 0);
3104         else
3105             _enable_event(SOCKET2HANDLE(s), 0, 0, FD_WINE_NONBLOCKING);
3106         break;
3107
3108     case WS_FIONREAD:
3109     {
3110         if (out_size != sizeof(WS_u_long) || IS_INTRESOURCE(out_buff))
3111         {
3112             WSASetLastError(WSAEFAULT);
3113             return SOCKET_ERROR;
3114         }
3115         if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR;
3116         if (ioctl(fd, FIONREAD, out_buff ) == -1)
3117             status = (errno == EBADF) ? WSAENOTSOCK : wsaErrno();
3118         release_sock_fd( s, fd );
3119         break;
3120     }
3121
3122     case WS_SIOCATMARK:
3123     {
3124         unsigned int oob = 0, oobsize = sizeof(int), atmark = 0;
3125         if (out_size != sizeof(WS_u_long) || IS_INTRESOURCE(out_buff))
3126         {
3127             WSASetLastError(WSAEFAULT);
3128             return SOCKET_ERROR;
3129         }
3130         if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR;
3131         /* SO_OOBINLINE sockets must always return TRUE to SIOCATMARK */
3132         if ((getsockopt(fd, SOL_SOCKET, SO_OOBINLINE, &oob, &oobsize ) == -1)
3133            || (!oob && ioctl(fd, SIOCATMARK, &atmark ) == -1))
3134             status = (errno == EBADF) ? WSAENOTSOCK : wsaErrno();
3135         else
3136         {
3137             /* The SIOCATMARK value read from ioctl() is reversed
3138              * because BSD returns TRUE if it's in the OOB mark
3139              * while Windows returns TRUE if there are NO OOB bytes.
3140              */
3141             (*(WS_u_long *) out_buff) = oob | !atmark;
3142         }
3143
3144         release_sock_fd( s, fd );
3145         break;
3146     }
3147
3148     case WS_FIOASYNC:
3149         WARN("Warning: WS1.1 shouldn't be using async I/O\n");
3150         SetLastError(WSAEINVAL);
3151         return SOCKET_ERROR;
3152
3153    case WS_SIO_GET_INTERFACE_LIST:
3154        {
3155            INTERFACE_INFO* intArray = out_buff;
3156            DWORD size, numInt = 0, apiReturn;
3157
3158            TRACE("-> SIO_GET_INTERFACE_LIST request\n");
3159
3160            if (!out_buff || !ret_size)
3161            {
3162                WSASetLastError(WSAEFAULT);
3163                return SOCKET_ERROR;
3164            }
3165
3166            fd = get_sock_fd( s, 0, NULL );
3167            if (fd == -1) return SOCKET_ERROR;
3168
3169            apiReturn = GetAdaptersInfo(NULL, &size);
3170            if (apiReturn == ERROR_BUFFER_OVERFLOW)
3171            {
3172                PIP_ADAPTER_INFO table = HeapAlloc(GetProcessHeap(),0,size);
3173
3174                if (table)
3175                {
3176                   if (GetAdaptersInfo(table, &size) == NO_ERROR)
3177                   {
3178                      PIP_ADAPTER_INFO ptr;
3179
3180                      if (size*sizeof(INTERFACE_INFO)/sizeof(IP_ADAPTER_INFO) > out_size)
3181                      {
3182                         WARN("Buffer too small = %u, out_size = %u\n", size, out_size);
3183                         HeapFree(GetProcessHeap(),0,table);
3184                         release_sock_fd( s, fd );
3185                         status = WSAEFAULT;
3186                         break;
3187                      }
3188                      for (ptr = table, numInt = 0; ptr;
3189                       ptr = ptr->Next, intArray++, numInt++)
3190                      {
3191                         unsigned int addr, mask, bcast;
3192                         struct ifreq ifInfo;
3193
3194                         /* Socket Status Flags */
3195                         lstrcpynA(ifInfo.ifr_name, ptr->AdapterName, IFNAMSIZ);
3196                         if (ioctl(fd, SIOCGIFFLAGS, &ifInfo) < 0)
3197                         {
3198                            ERR("Error obtaining status flags for socket!\n");
3199                            HeapFree(GetProcessHeap(),0,table);
3200                            release_sock_fd( s, fd );
3201                            status = WSAEINVAL;
3202                            break;
3203                         }
3204                         else
3205                         {
3206                            /* set flags; the values of IFF_* are not the same
3207                               under Linux and Windows, therefore must generate
3208                               new flags */
3209                            intArray->iiFlags = 0;
3210                            if (ifInfo.ifr_flags & IFF_BROADCAST)
3211                               intArray->iiFlags |= WS_IFF_BROADCAST;
3212 #ifdef IFF_POINTOPOINT
3213                            if (ifInfo.ifr_flags & IFF_POINTOPOINT)
3214                               intArray->iiFlags |= WS_IFF_POINTTOPOINT;
3215 #endif
3216                            if (ifInfo.ifr_flags & IFF_LOOPBACK)
3217                               intArray->iiFlags |= WS_IFF_LOOPBACK;
3218                            if (ifInfo.ifr_flags & IFF_UP)
3219                               intArray->iiFlags |= WS_IFF_UP;
3220                            if (ifInfo.ifr_flags & IFF_MULTICAST)
3221                               intArray->iiFlags |= WS_IFF_MULTICAST;
3222                         }
3223
3224                         addr = inet_addr(ptr->IpAddressList.IpAddress.String);
3225                         mask = inet_addr(ptr->IpAddressList.IpMask.String);
3226                         bcast = addr | ~mask;
3227                         intArray->iiAddress.AddressIn.sin_family = AF_INET;
3228                         intArray->iiAddress.AddressIn.sin_port = 0;
3229                         intArray->iiAddress.AddressIn.sin_addr.WS_s_addr =
3230                          addr;
3231                         intArray->iiNetmask.AddressIn.sin_family = AF_INET;
3232                         intArray->iiNetmask.AddressIn.sin_port = 0;
3233                         intArray->iiNetmask.AddressIn.sin_addr.WS_s_addr =
3234                          mask;
3235                         intArray->iiBroadcastAddress.AddressIn.sin_family =
3236                          AF_INET;
3237                         intArray->iiBroadcastAddress.AddressIn.sin_port = 0;
3238                         intArray->iiBroadcastAddress.AddressIn.sin_addr.
3239                          WS_s_addr = bcast;
3240                      }
3241                   }
3242                   else
3243                   {
3244                      ERR("Unable to get interface table!\n");
3245                      status = WSAEINVAL;
3246                   }
3247                   HeapFree(GetProcessHeap(),0,table);
3248                }
3249                else status = WSAEINVAL;
3250            }
3251            else if (apiReturn != ERROR_NO_DATA)
3252            {
3253                ERR("Unable to get interface table!\n");
3254                status = WSAEINVAL;
3255            }
3256            /* Calculate the size of the array being returned */
3257            total = sizeof(INTERFACE_INFO) * numInt;
3258            release_sock_fd( s, fd );
3259            break;
3260        }
3261
3262    case WS_SIO_ADDRESS_LIST_CHANGE:
3263        FIXME("-> SIO_ADDRESS_LIST_CHANGE request: stub\n");
3264        /* FIXME: error and return code depend on whether socket was created
3265         * with WSA_FLAG_OVERLAPPED, but there is no easy way to get this */
3266        break;
3267
3268    case WS_SIO_ADDRESS_LIST_QUERY:
3269    {
3270         DWORD size;
3271
3272         TRACE("-> SIO_ADDRESS_LIST_QUERY request\n");
3273
3274         if (!ret_size)
3275         {
3276             WSASetLastError(WSAEFAULT);
3277             return SOCKET_ERROR;
3278         }
3279
3280         if (GetAdaptersInfo(NULL, &size) == ERROR_BUFFER_OVERFLOW)
3281         {
3282             IP_ADAPTER_INFO *p, *table = HeapAlloc(GetProcessHeap(), 0, size);
3283             DWORD num;
3284
3285             if (!table || GetAdaptersInfo(table, &size))
3286             {
3287                 HeapFree(GetProcessHeap(), 0, table);
3288                 status = WSAEINVAL;
3289                 break;
3290             }
3291
3292             for (p = table, num = 0; p; p = p->Next)
3293                 if (p->IpAddressList.IpAddress.String[0]) num++;
3294
3295             total = sizeof(SOCKET_ADDRESS_LIST) + sizeof(SOCKET_ADDRESS) * (num - 1);
3296             total += sizeof(SOCKADDR) * num;
3297
3298             if (total > out_size)
3299             {
3300                 HeapFree(GetProcessHeap(), 0, table);
3301                 status = WSAEFAULT;
3302                 break;
3303             }
3304
3305             if (out_buff)
3306             {
3307                 unsigned int i;
3308                 SOCKET_ADDRESS *sa;
3309                 SOCKET_ADDRESS_LIST *sa_list = out_buff;
3310                 SOCKADDR_IN *sockaddr;
3311
3312                 sa = sa_list->Address;
3313                 sockaddr = (SOCKADDR_IN *)((char *)sa + num * sizeof(SOCKET_ADDRESS));
3314                 sa_list->iAddressCount = num;
3315
3316                 for (p = table, i = 0; p; p = p->Next)
3317                 {
3318                     if (!p->IpAddressList.IpAddress.String[0]) continue;
3319
3320                     sa[i].lpSockaddr = (SOCKADDR *)&sockaddr[i];
3321                     sa[i].iSockaddrLength = sizeof(SOCKADDR);
3322
3323                     sockaddr[i].sin_family = AF_INET;
3324                     sockaddr[i].sin_port = 0;
3325                     sockaddr[i].sin_addr.WS_s_addr = inet_addr(p->IpAddressList.IpAddress.String);
3326                     i++;
3327                 }
3328             }
3329
3330             HeapFree(GetProcessHeap(), 0, table);
3331         }
3332         else
3333         {
3334             WARN("unable to get IP address list\n");
3335             status = WSAEINVAL;
3336         }
3337         break;
3338    }
3339
3340    case WS_SIO_FLUSH:
3341         FIXME("SIO_FLUSH: stub.\n");
3342         break;
3343
3344    case WS_SIO_GET_EXTENSION_FUNCTION_POINTER:
3345    {
3346         static const GUID connectex_guid = WSAID_CONNECTEX;
3347         static const GUID disconnectex_guid = WSAID_DISCONNECTEX;
3348         static const GUID acceptex_guid = WSAID_ACCEPTEX;
3349         static const GUID getaccepexsockaddrs_guid = WSAID_GETACCEPTEXSOCKADDRS;
3350         static const GUID transmitfile_guid = WSAID_TRANSMITFILE;
3351         static const GUID transmitpackets_guid = WSAID_TRANSMITPACKETS;
3352         static const GUID wsarecvmsg_guid = WSAID_WSARECVMSG;
3353         static const GUID wsasendmsg_guid = WSAID_WSASENDMSG;
3354
3355         if ( IsEqualGUID(&connectex_guid, in_buff) )
3356         {
3357             *(LPFN_CONNECTEX *)out_buff = WS2_ConnectEx;
3358             break;
3359         }
3360         else if ( IsEqualGUID(&disconnectex_guid, in_buff) )
3361         {
3362             FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER: unimplemented DisconnectEx\n");
3363         }
3364         else if ( IsEqualGUID(&acceptex_guid, in_buff) )
3365         {
3366             *(LPFN_ACCEPTEX *)out_buff = WS2_AcceptEx;
3367             break;
3368         }
3369         else if ( IsEqualGUID(&getaccepexsockaddrs_guid, in_buff) )
3370         {
3371             *(LPFN_GETACCEPTEXSOCKADDRS *)out_buff = WS2_GetAcceptExSockaddrs;
3372             break;
3373         }
3374         else if ( IsEqualGUID(&transmitfile_guid, in_buff) )
3375         {
3376             FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER: unimplemented TransmitFile\n");
3377         }
3378         else if ( IsEqualGUID(&transmitpackets_guid, in_buff) )
3379         {
3380             FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER: unimplemented TransmitPackets\n");
3381         }
3382         else if ( IsEqualGUID(&wsarecvmsg_guid, in_buff) )
3383         {
3384             *(LPFN_WSARECVMSG *)out_buff = WS2_WSARecvMsg;
3385             break;
3386         }
3387         else if ( IsEqualGUID(&wsasendmsg_guid, in_buff) )
3388         {
3389             FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER: unimplemented WSASendMsg\n");
3390         }
3391         else
3392             FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER %s: stub\n", debugstr_guid(in_buff));
3393
3394         status = WSAEOPNOTSUPP;
3395         break;
3396    }
3397    case WS_SIO_KEEPALIVE_VALS:
3398    {
3399         struct tcp_keepalive *k;
3400         int keepalive, keepidle, keepintvl;
3401
3402         if (!in_buff || in_size < sizeof(struct tcp_keepalive))
3403         {
3404             WSASetLastError(WSAEFAULT);
3405             return SOCKET_ERROR;
3406         }
3407
3408         k = in_buff;
3409         keepalive = k->onoff ? 1 : 0;
3410         keepidle = max( 1, (k->keepalivetime + 500) / 1000 );
3411         keepintvl = max( 1, (k->keepaliveinterval + 500) / 1000 );
3412
3413         TRACE("onoff: %d, keepalivetime: %d, keepaliveinterval: %d\n", keepalive, keepidle, keepintvl);
3414
3415         fd = get_sock_fd(s, 0, NULL);
3416         if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, (void *)&keepalive, sizeof(int)) == -1)
3417             status = WSAEINVAL;
3418 #if defined(TCP_KEEPIDLE) && defined(TCP_KEEPINTVL)
3419         /* these values need to be set only if SO_KEEPALIVE is enabled */
3420         else if(keepalive)
3421         {
3422             if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPIDLE, (void *)&keepidle, sizeof(int)) == -1)
3423                 status = WSAEINVAL;
3424             else if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPINTVL, (void *)&keepintvl, sizeof(int)) == -1)
3425                 status = WSAEINVAL;
3426         }
3427 #else
3428         else
3429             FIXME("ignoring keepalive interval and timeout\n");
3430 #endif
3431         release_sock_fd(s, fd);
3432         break;
3433    }
3434    case WS_SIO_ROUTING_INTERFACE_QUERY:
3435    {
3436        struct WS_sockaddr *daddr = (struct WS_sockaddr *)in_buff;
3437        struct WS_sockaddr_in *daddr_in = (struct WS_sockaddr_in *)daddr;
3438        struct WS_sockaddr_in *saddr_in = out_buff;
3439        MIB_IPFORWARDROW row;
3440        PMIB_IPADDRTABLE ipAddrTable = NULL;
3441        DWORD size, i, found_index;
3442
3443        TRACE("-> WS_SIO_ROUTING_INTERFACE_QUERY request\n");
3444
3445        if (!in_buff || in_size < sizeof(struct WS_sockaddr) ||
3446            !out_buff || out_size < sizeof(struct WS_sockaddr_in) || !ret_size)
3447        {
3448            WSASetLastError(WSAEFAULT);
3449            return SOCKET_ERROR;
3450        }
3451        if (daddr->sa_family != AF_INET)
3452        {
3453            FIXME("unsupported address family %d\n", daddr->sa_family);
3454            status = WSAEAFNOSUPPORT;
3455            break;
3456        }
3457        if (GetBestRoute(daddr_in->sin_addr.S_un.S_addr, 0, &row) != NOERROR ||
3458            GetIpAddrTable(NULL, &size, FALSE) != ERROR_INSUFFICIENT_BUFFER)
3459        {
3460            status = WSAEFAULT;
3461            break;
3462        }
3463        ipAddrTable = HeapAlloc(GetProcessHeap(), 0, size);
3464        if (GetIpAddrTable(ipAddrTable, &size, FALSE))
3465        {
3466            HeapFree(GetProcessHeap(), 0, ipAddrTable);
3467            status = WSAEFAULT;
3468            break;
3469        }
3470        for (i = 0, found_index = ipAddrTable->dwNumEntries;
3471             i < ipAddrTable->dwNumEntries; i++)
3472        {
3473            if (ipAddrTable->table[i].dwIndex == row.dwForwardIfIndex)
3474                found_index = i;
3475        }
3476        if (found_index == ipAddrTable->dwNumEntries)
3477        {
3478            ERR("no matching IP address for interface %d\n",
3479                row.dwForwardIfIndex);
3480            HeapFree(GetProcessHeap(), 0, ipAddrTable);
3481            status = WSAEFAULT;
3482            break;
3483        }
3484        saddr_in->sin_family = AF_INET;
3485        saddr_in->sin_addr.S_un.S_addr = ipAddrTable->table[found_index].dwAddr;
3486        saddr_in->sin_port = 0;
3487        total = sizeof(struct WS_sockaddr_in);
3488        HeapFree(GetProcessHeap(), 0, ipAddrTable);
3489        break;
3490    }
3491    case WS_SIO_SET_COMPATIBILITY_MODE:
3492        TRACE("WS_SIO_SET_COMPATIBILITY_MODE ignored\n");
3493        status = WSAEOPNOTSUPP;
3494        break;
3495    case WS_SIO_UDP_CONNRESET:
3496        FIXME("WS_SIO_UDP_CONNRESET stub\n");
3497        break;
3498     case 0x667e: /* Netscape tries hard to use bogus ioctl 0x667e */
3499         WSASetLastError(WSAEOPNOTSUPP);
3500         return SOCKET_ERROR;
3501     default:
3502         FIXME("unsupported WS_IOCTL cmd (%s)\n", debugstr_wsaioctl(code));
3503         status = WSAEOPNOTSUPP;
3504         break;
3505     }
3506
3507     if (completion)
3508     {
3509         FIXME( "completion routine %p not supported\n", completion );
3510     }
3511     else if (overlapped)
3512     {
3513         ULONG_PTR cvalue = (overlapped && ((ULONG_PTR)overlapped->hEvent & 1) == 0) ? (ULONG_PTR)overlapped : 0;
3514         overlapped->Internal = status;
3515         overlapped->InternalHigh = total;
3516         if (overlapped->hEvent) NtSetEvent( overlapped->hEvent, NULL );
3517         if (cvalue) WS_AddCompletion( HANDLE2SOCKET(s), cvalue, status, total );
3518     }
3519
3520     if (!status)
3521     {
3522         if (ret_size) *ret_size = total;
3523         return 0;
3524     }
3525     SetLastError( status );
3526     return SOCKET_ERROR;
3527 }
3528
3529
3530 /***********************************************************************
3531  *              ioctlsocket             (WS2_32.10)
3532  */
3533 int WINAPI WS_ioctlsocket(SOCKET s, LONG cmd, WS_u_long *argp)
3534 {
3535     DWORD ret_size;
3536     return WSAIoctl( s, cmd, argp, sizeof(WS_u_long), argp, sizeof(WS_u_long), &ret_size, NULL, NULL );
3537 }
3538
3539 /***********************************************************************
3540  *              listen          (WS2_32.13)
3541  */
3542 int WINAPI WS_listen(SOCKET s, int backlog)
3543 {
3544     int fd = get_sock_fd( s, FILE_READ_DATA, NULL );
3545
3546     TRACE("socket %04lx, backlog %d\n", s, backlog);
3547     if (fd != -1)
3548     {
3549         if (listen(fd, backlog) == 0)
3550         {
3551             release_sock_fd( s, fd );
3552             _enable_event(SOCKET2HANDLE(s), FD_ACCEPT,
3553                           FD_WINE_LISTENING,
3554                           FD_CONNECT|FD_WINE_CONNECTED);
3555             return 0;
3556         }
3557         SetLastError(wsaErrno());
3558         release_sock_fd( s, fd );
3559     }
3560     return SOCKET_ERROR;
3561 }
3562
3563 /***********************************************************************
3564  *              recv                    (WS2_32.16)
3565  */
3566 int WINAPI WS_recv(SOCKET s, char *buf, int len, int flags)
3567 {
3568     DWORD n, dwFlags = flags;
3569     WSABUF wsabuf;
3570
3571     wsabuf.len = len;
3572     wsabuf.buf = buf;
3573
3574     if ( WS2_recv_base(s, &wsabuf, 1, &n, &dwFlags, NULL, NULL, NULL, NULL, NULL) == SOCKET_ERROR )
3575         return SOCKET_ERROR;
3576     else
3577         return n;
3578 }
3579
3580 /***********************************************************************
3581  *              recvfrom                (WS2_32.17)
3582  */
3583 int WINAPI WS_recvfrom(SOCKET s, char *buf, INT len, int flags,
3584                        struct WS_sockaddr *from, int *fromlen)
3585 {
3586     DWORD n, dwFlags = flags;
3587     WSABUF wsabuf;
3588
3589     wsabuf.len = len;
3590     wsabuf.buf = buf;
3591
3592     if ( WS2_recv_base(s, &wsabuf, 1, &n, &dwFlags, from, fromlen, NULL, NULL, NULL) == SOCKET_ERROR )
3593         return SOCKET_ERROR;
3594     else
3595         return n;
3596 }
3597
3598 /* allocate a poll array for the corresponding fd sets */
3599 static struct pollfd *fd_sets_to_poll( const WS_fd_set *readfds, const WS_fd_set *writefds,
3600                                        const WS_fd_set *exceptfds, int *count_ptr )
3601 {
3602     unsigned int i, j = 0, count = 0;
3603     struct pollfd *fds;
3604
3605     if (readfds) count += readfds->fd_count;
3606     if (writefds) count += writefds->fd_count;
3607     if (exceptfds) count += exceptfds->fd_count;
3608     *count_ptr = count;
3609     if (!count)
3610     {
3611         SetLastError(WSAEINVAL);
3612         return NULL;
3613     }
3614     if (!(fds = HeapAlloc( GetProcessHeap(), 0, count * sizeof(fds[0]))))
3615     {
3616         SetLastError( ERROR_NOT_ENOUGH_MEMORY );
3617         return NULL;
3618     }
3619     if (readfds)
3620         for (i = 0; i < readfds->fd_count; i++, j++)
3621         {
3622             fds[j].fd = get_sock_fd( readfds->fd_array[i], FILE_READ_DATA, NULL );
3623             if (fds[j].fd == -1) goto failed;
3624             fds[j].events = POLLIN;
3625             fds[j].revents = 0;
3626         }
3627     if (writefds)
3628         for (i = 0; i < writefds->fd_count; i++, j++)
3629         {
3630             fds[j].fd = get_sock_fd( writefds->fd_array[i], FILE_WRITE_DATA, NULL );
3631             if (fds[j].fd == -1) goto failed;
3632             fds[j].events = POLLOUT;
3633             fds[j].revents = 0;
3634         }
3635     if (exceptfds)
3636         for (i = 0; i < exceptfds->fd_count; i++, j++)
3637         {
3638             fds[j].fd = get_sock_fd( exceptfds->fd_array[i], 0, NULL );
3639             if (fds[j].fd == -1) goto failed;
3640             fds[j].events = POLLHUP;
3641             fds[j].revents = 0;
3642         }
3643     return fds;
3644
3645 failed:
3646     count = j;
3647     j = 0;
3648     if (readfds)
3649         for (i = 0; i < readfds->fd_count && j < count; i++, j++)
3650             release_sock_fd( readfds->fd_array[i], fds[j].fd );
3651     if (writefds)
3652         for (i = 0; i < writefds->fd_count && j < count; i++, j++)
3653             release_sock_fd( writefds->fd_array[i], fds[j].fd );
3654     if (exceptfds)
3655         for (i = 0; i < exceptfds->fd_count && j < count; i++, j++)
3656             release_sock_fd( exceptfds->fd_array[i], fds[j].fd );
3657     HeapFree( GetProcessHeap(), 0, fds );
3658     return NULL;
3659 }
3660
3661 /* release the file descriptor obtained in fd_sets_to_poll */
3662 /* must be called with the original fd_set arrays, before calling get_poll_results */
3663 static void release_poll_fds( const WS_fd_set *readfds, const WS_fd_set *writefds,
3664                               const WS_fd_set *exceptfds, struct pollfd *fds )
3665 {
3666     unsigned int i, j = 0;
3667
3668     if (readfds)
3669     {
3670         for (i = 0; i < readfds->fd_count; i++, j++)
3671             if (fds[j].fd != -1) release_sock_fd( readfds->fd_array[i], fds[j].fd );
3672     }
3673     if (writefds)
3674     {
3675         for (i = 0; i < writefds->fd_count; i++, j++)
3676             if (fds[j].fd != -1) release_sock_fd( writefds->fd_array[i], fds[j].fd );
3677     }
3678     if (exceptfds)
3679     {
3680         for (i = 0; i < exceptfds->fd_count; i++, j++)
3681             if (fds[j].fd != -1)
3682             {
3683                 /* make sure we have a real error before releasing the fd */
3684                 if (!sock_error_p( fds[j].fd )) fds[j].revents = 0;
3685                 release_sock_fd( exceptfds->fd_array[i], fds[j].fd );
3686             }
3687     }
3688 }
3689
3690 /* map the poll results back into the Windows fd sets */
3691 static int get_poll_results( WS_fd_set *readfds, WS_fd_set *writefds, WS_fd_set *exceptfds,
3692                              const struct pollfd *fds )
3693 {
3694     unsigned int i, j = 0, k, total = 0;
3695
3696     if (readfds)
3697     {
3698         for (i = k = 0; i < readfds->fd_count; i++, j++)
3699             if (fds[j].revents) readfds->fd_array[k++] = readfds->fd_array[i];
3700         readfds->fd_count = k;
3701         total += k;
3702     }
3703     if (writefds)
3704     {
3705         for (i = k = 0; i < writefds->fd_count; i++, j++)
3706             if ((fds[j].revents & POLLOUT) && !(fds[j].revents & POLLHUP))
3707                 writefds->fd_array[k++] = writefds->fd_array[i];
3708         writefds->fd_count = k;
3709         total += k;
3710     }
3711     if (exceptfds)
3712     {
3713         for (i = k = 0; i < exceptfds->fd_count; i++, j++)
3714             if (fds[j].revents) exceptfds->fd_array[k++] = exceptfds->fd_array[i];
3715         exceptfds->fd_count = k;
3716         total += k;
3717     }
3718     return total;
3719 }
3720
3721
3722 /***********************************************************************
3723  *              select                  (WS2_32.18)
3724  */
3725 int WINAPI WS_select(int nfds, WS_fd_set *ws_readfds,
3726                      WS_fd_set *ws_writefds, WS_fd_set *ws_exceptfds,
3727                      const struct WS_timeval* ws_timeout)
3728 {
3729     struct pollfd *pollfds;
3730     struct timeval tv1, tv2;
3731     int torig = 0;
3732     int count, ret, timeout = -1;
3733
3734     TRACE("read %p, write %p, excp %p timeout %p\n",
3735           ws_readfds, ws_writefds, ws_exceptfds, ws_timeout);
3736
3737     if (!(pollfds = fd_sets_to_poll( ws_readfds, ws_writefds, ws_exceptfds, &count )))
3738         return SOCKET_ERROR;
3739
3740     if (ws_timeout)
3741     {
3742         torig = (ws_timeout->tv_sec * 1000) + (ws_timeout->tv_usec + 999) / 1000;
3743         timeout = torig;
3744         gettimeofday( &tv1, 0 );
3745     }
3746
3747     while ((ret = poll( pollfds, count, timeout )) < 0)
3748     {
3749         if (errno == EINTR)
3750         {
3751             if (!ws_timeout) continue;
3752             gettimeofday( &tv2, 0 );
3753
3754             tv2.tv_sec  -= tv1.tv_sec;
3755             tv2.tv_usec -= tv1.tv_usec;
3756             if (tv2.tv_usec < 0)
3757             {
3758                 tv2.tv_usec += 1000000;
3759                 tv2.tv_sec  -= 1;
3760             }
3761
3762             timeout = torig - (tv2.tv_sec * 1000) - (tv2.tv_usec + 999) / 1000;
3763             if (timeout <= 0) break;
3764         } else break;
3765     }
3766     release_poll_fds( ws_readfds, ws_writefds, ws_exceptfds, pollfds );
3767
3768     if (ret == -1) SetLastError(wsaErrno());
3769     else ret = get_poll_results( ws_readfds, ws_writefds, ws_exceptfds, pollfds );
3770     HeapFree( GetProcessHeap(), 0, pollfds );
3771     return ret;
3772 }
3773
3774 /* helper to send completion messages for client-only i/o operation case */
3775 static void WS_AddCompletion( SOCKET sock, ULONG_PTR CompletionValue, NTSTATUS CompletionStatus,
3776                               ULONG Information )
3777 {
3778     SERVER_START_REQ( add_fd_completion )
3779     {
3780         req->handle      = wine_server_obj_handle( SOCKET2HANDLE(sock) );
3781         req->cvalue      = CompletionValue;
3782         req->status      = CompletionStatus;
3783         req->information = Information;
3784         wine_server_call( req );
3785     }
3786     SERVER_END_REQ;
3787 }
3788
3789
3790 /***********************************************************************
3791  *              send                    (WS2_32.19)
3792  */
3793 int WINAPI WS_send(SOCKET s, const char *buf, int len, int flags)
3794 {
3795     DWORD n;
3796     WSABUF wsabuf;
3797
3798     wsabuf.len = len;
3799     wsabuf.buf = (char*) buf;
3800
3801     if ( WS2_sendto( s, &wsabuf, 1, &n, flags, NULL, 0, NULL, NULL) == SOCKET_ERROR )
3802         return SOCKET_ERROR;
3803     else
3804         return n;
3805 }
3806
3807 /***********************************************************************
3808  *              WSASend                 (WS2_32.72)
3809  */
3810 INT WINAPI WSASend( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
3811                     LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
3812                     LPWSAOVERLAPPED lpOverlapped,
3813                     LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
3814 {
3815     return WS2_sendto( s, lpBuffers, dwBufferCount, lpNumberOfBytesSent, dwFlags,
3816                       NULL, 0, lpOverlapped, lpCompletionRoutine );
3817 }
3818
3819 /***********************************************************************
3820  *              WSASendDisconnect       (WS2_32.73)
3821  */
3822 INT WINAPI WSASendDisconnect( SOCKET s, LPWSABUF lpBuffers )
3823 {
3824     return WS_shutdown( s, SD_SEND );
3825 }
3826
3827
3828 static int WS2_sendto( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
3829                        LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
3830                        const struct WS_sockaddr *to, int tolen,
3831                        LPWSAOVERLAPPED lpOverlapped,
3832                        LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
3833 {
3834     unsigned int i, options;
3835     int n, fd, err;
3836     struct ws2_async *wsa = NULL;
3837     int totalLength = 0;
3838     ULONG_PTR cvalue = (lpOverlapped && ((ULONG_PTR)lpOverlapped->hEvent & 1) == 0) ? (ULONG_PTR)lpOverlapped : 0;
3839     DWORD bytes_sent;
3840
3841     TRACE("socket %04lx, wsabuf %p, nbufs %d, flags %d, to %p, tolen %d, ovl %p, func %p\n",
3842           s, lpBuffers, dwBufferCount, dwFlags,
3843           to, tolen, lpOverlapped, lpCompletionRoutine);
3844
3845     fd = get_sock_fd( s, FILE_WRITE_DATA, &options );
3846     TRACE( "fd=%d, options=%x\n", fd, options );
3847
3848     if ( fd == -1 ) return SOCKET_ERROR;
3849
3850     if (!lpOverlapped && !lpNumberOfBytesSent)
3851     {
3852         err = WSAEFAULT;
3853         goto error;
3854     }
3855     if (!(wsa = HeapAlloc( GetProcessHeap(), 0, FIELD_OFFSET(struct ws2_async, iovec[dwBufferCount]) )))
3856     {
3857         err = WSAEFAULT;
3858         goto error;
3859     }
3860
3861     wsa->hSocket     = SOCKET2HANDLE(s);
3862     wsa->addr        = (struct WS_sockaddr *)to;
3863     wsa->addrlen.val = tolen;
3864     wsa->flags       = dwFlags;
3865     wsa->lpFlags     = &wsa->flags;
3866     wsa->control     = NULL;
3867     wsa->n_iovecs    = dwBufferCount;
3868     wsa->first_iovec = 0;
3869     for ( i = 0; i < dwBufferCount; i++ )
3870     {
3871         wsa->iovec[i].iov_base = lpBuffers[i].buf;
3872         wsa->iovec[i].iov_len  = lpBuffers[i].len;
3873         totalLength += lpBuffers[i].len;
3874     }
3875
3876     for (;;)
3877     {
3878         n = WS2_send( fd, wsa );
3879         if (n != -1 || errno != EINTR) break;
3880     }
3881     if (n == -1 && errno != EAGAIN)
3882     {
3883         int loc_errno = errno;
3884         err = wsaErrno();
3885         if (cvalue) WS_AddCompletion( s, cvalue, sock_get_ntstatus(loc_errno), 0 );
3886         goto error;
3887     }
3888
3889     if ((lpOverlapped || lpCompletionRoutine) &&
3890         !(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)))
3891     {
3892         IO_STATUS_BLOCK *iosb = lpOverlapped ? (IO_STATUS_BLOCK *)lpOverlapped : &wsa->local_iosb;
3893
3894         wsa->user_overlapped = lpOverlapped;
3895         wsa->completion_func = lpCompletionRoutine;
3896         release_sock_fd( s, fd );
3897
3898         if (n == -1 || n < totalLength)
3899         {
3900             iosb->u.Status = STATUS_PENDING;
3901             iosb->Information = n == -1 ? 0 : n;
3902
3903             SERVER_START_REQ( register_async )
3904             {
3905                 req->type           = ASYNC_TYPE_WRITE;
3906                 req->async.handle   = wine_server_obj_handle( wsa->hSocket );
3907                 req->async.callback = wine_server_client_ptr( WS2_async_send );
3908                 req->async.iosb     = wine_server_client_ptr( iosb );
3909                 req->async.arg      = wine_server_client_ptr( wsa );
3910                 req->async.event    = wine_server_obj_handle( lpCompletionRoutine ? 0 : lpOverlapped->hEvent );
3911                 req->async.cvalue   = cvalue;
3912                 err = wine_server_call( req );
3913             }
3914             SERVER_END_REQ;
3915
3916             /* Enable the event only after starting the async. The server will deliver it as soon as
3917                the async is done. */
3918             _enable_event(SOCKET2HANDLE(s), FD_WRITE, 0, 0);
3919
3920             if (err != STATUS_PENDING) HeapFree( GetProcessHeap(), 0, wsa );
3921             WSASetLastError( NtStatusToWSAError( err ));
3922             return SOCKET_ERROR;
3923         }
3924
3925         iosb->u.Status = STATUS_SUCCESS;
3926         iosb->Information = n;
3927         if (lpNumberOfBytesSent) *lpNumberOfBytesSent = n;
3928         if (!wsa->completion_func)
3929         {
3930             if (cvalue) WS_AddCompletion( s, cvalue, STATUS_SUCCESS, n );
3931             if (lpOverlapped->hEvent) SetEvent( lpOverlapped->hEvent );
3932             HeapFree( GetProcessHeap(), 0, wsa );
3933         }
3934         else NtQueueApcThread( GetCurrentThread(), (PNTAPCFUNC)ws2_async_apc,
3935                                (ULONG_PTR)wsa, (ULONG_PTR)iosb, 0 );
3936         WSASetLastError(0);
3937         return 0;
3938     }
3939
3940     if ( _is_blocking(s) )
3941     {
3942         /* On a blocking non-overlapped stream socket,
3943          * sending blocks until the entire buffer is sent. */
3944         DWORD timeout_start = GetTickCount();
3945
3946         bytes_sent = n == -1 ? 0 : n;
3947
3948         while (wsa->first_iovec < wsa->n_iovecs)
3949         {
3950             struct pollfd pfd;
3951             int timeout = GET_SNDTIMEO(fd);
3952
3953             if (timeout != -1)
3954             {
3955                 timeout -= GetTickCount() - timeout_start;
3956                 if (timeout < 0) timeout = 0;
3957             }
3958
3959             pfd.fd = fd;
3960             pfd.events = POLLOUT;
3961
3962             if (!timeout || !poll( &pfd, 1, timeout ))
3963             {
3964                 err = WSAETIMEDOUT;
3965                 goto error; /* msdn says a timeout in send is fatal */
3966             }
3967
3968             n = WS2_send( fd, wsa );
3969             if (n == -1 && errno != EAGAIN && errno != EINTR)
3970             {
3971                 err = wsaErrno();
3972                 goto error;
3973             }
3974
3975             if (n >= 0)
3976                 bytes_sent += n;
3977         }
3978     }
3979     else  /* non-blocking */
3980     {
3981         if (n < totalLength)
3982             _enable_event(SOCKET2HANDLE(s), FD_WRITE, 0, 0);
3983         if (n == -1)
3984         {
3985             err = WSAEWOULDBLOCK;
3986             goto error;
3987         }
3988         bytes_sent = n;
3989     }
3990
3991     TRACE(" -> %i bytes\n", bytes_sent);
3992
3993     if (lpNumberOfBytesSent) *lpNumberOfBytesSent = bytes_sent;
3994     HeapFree( GetProcessHeap(), 0, wsa );
3995     release_sock_fd( s, fd );
3996     WSASetLastError(0);
3997     return 0;
3998
3999 error:
4000     HeapFree( GetProcessHeap(), 0, wsa );
4001     release_sock_fd( s, fd );
4002     WARN(" -> ERROR %d\n", err);
4003     WSASetLastError(err);
4004     return SOCKET_ERROR;
4005 }
4006
4007 /***********************************************************************
4008  *              WSASendTo               (WS2_32.74)
4009  */
4010 INT WINAPI WSASendTo( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
4011                       LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
4012                       const struct WS_sockaddr *to, int tolen,
4013                       LPWSAOVERLAPPED lpOverlapped,
4014                       LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
4015 {
4016     return WS2_sendto( s, lpBuffers, dwBufferCount,
4017                 lpNumberOfBytesSent, dwFlags,
4018                 to, tolen,
4019                 lpOverlapped, lpCompletionRoutine );
4020 }
4021
4022 /***********************************************************************
4023  *              sendto          (WS2_32.20)
4024  */
4025 int WINAPI WS_sendto(SOCKET s, const char *buf, int len, int flags,
4026                               const struct WS_sockaddr *to, int tolen)
4027 {
4028     DWORD n;
4029     WSABUF wsabuf;
4030
4031     wsabuf.len = len;
4032     wsabuf.buf = (char*) buf;
4033
4034     if ( WS2_sendto(s, &wsabuf, 1, &n, flags, to, tolen, NULL, NULL) == SOCKET_ERROR )
4035         return SOCKET_ERROR;
4036     else
4037         return n;
4038 }
4039
4040 /***********************************************************************
4041  *              setsockopt              (WS2_32.21)
4042  */
4043 int WINAPI WS_setsockopt(SOCKET s, int level, int optname,
4044                          const char *optval, int optlen)
4045 {
4046     int fd;
4047     int woptval;
4048     struct linger linger;
4049     struct timeval tval;
4050
4051     TRACE("socket: %04lx, level 0x%x, name 0x%x, ptr %p, len %d\n",
4052           s, level, optname, optval, optlen);
4053
4054     /* some broken apps pass the value directly instead of a pointer to it */
4055     if(optlen && IS_INTRESOURCE(optval))
4056     {
4057         SetLastError(WSAEFAULT);
4058         return SOCKET_ERROR;
4059     }
4060
4061     switch(level)
4062     {
4063     case WS_SOL_SOCKET:
4064         switch(optname)
4065         {
4066         /* Some options need some conversion before they can be sent to
4067          * setsockopt. The conversions are done here, then they will fall though
4068          * to the general case. Special options that are not passed to
4069          * setsockopt follow below that.*/
4070
4071         case WS_SO_DONTLINGER:
4072             if (!optval)
4073             {
4074                 SetLastError(WSAEFAULT);
4075                 return SOCKET_ERROR;
4076             }
4077             linger.l_onoff  = *((const int*)optval) ? 0: 1;
4078             linger.l_linger = 0;
4079             level = SOL_SOCKET;
4080             optname = SO_LINGER;
4081             optval = (char*)&linger;
4082             optlen = sizeof(struct linger);
4083             break;
4084
4085         case WS_SO_LINGER:
4086             if (!optval)
4087             {
4088                 SetLastError(WSAEFAULT);
4089                 return SOCKET_ERROR;
4090             }
4091             linger.l_onoff  = ((LINGER*)optval)->l_onoff;
4092             linger.l_linger  = ((LINGER*)optval)->l_linger;
4093             level = SOL_SOCKET;
4094             optname = SO_LINGER;
4095             optval = (char*)&linger;
4096             optlen = sizeof(struct linger);
4097             break;
4098
4099         case WS_SO_RCVBUF:
4100             if (*(const int*)optval < 2048)
4101             {
4102                 WARN("SO_RCVBF for %d bytes is too small: ignored\n", *(const int*)optval );
4103                 return 0;
4104             }
4105             /* Fall through */
4106
4107         /* The options listed here don't need any special handling. Thanks to
4108          * the conversion happening above, options from there will fall through
4109          * to this, too.*/
4110         case WS_SO_ACCEPTCONN:
4111         case WS_SO_BROADCAST:
4112         case WS_SO_ERROR:
4113         case WS_SO_KEEPALIVE:
4114         case WS_SO_OOBINLINE:
4115         /* BSD socket SO_REUSEADDR is not 100% compatible to winsock semantics.
4116          * however, using it the BSD way fixes bug 8513 and seems to be what
4117          * most programmers assume, anyway */
4118         case WS_SO_REUSEADDR:
4119         case WS_SO_SNDBUF:
4120         case WS_SO_TYPE:
4121             convert_sockopt(&level, &optname);
4122             break;
4123
4124         /* SO_DEBUG is a privileged operation, ignore it. */
4125         case WS_SO_DEBUG:
4126             TRACE("Ignoring SO_DEBUG\n");
4127             return 0;
4128
4129         /* For some reason the game GrandPrixLegends does set SO_DONTROUTE on its
4130          * socket. According to MSDN, this option is silently ignored.*/
4131         case WS_SO_DONTROUTE:
4132             TRACE("Ignoring SO_DONTROUTE\n");
4133             return 0;
4134
4135         /* Stops two sockets from being bound to the same port. Always happens
4136          * on unix systems, so just drop it. */
4137         case WS_SO_EXCLUSIVEADDRUSE:
4138             TRACE("Ignoring SO_EXCLUSIVEADDRUSE, is always set.\n");
4139             return 0;
4140
4141         /* After a ConnectEx call succeeds, the socket can't be used with half of the
4142          * normal winsock functions on windows. We don't have that problem. */
4143         case WS_SO_UPDATE_CONNECT_CONTEXT:
4144             TRACE("Ignoring SO_UPDATE_CONNECT_CONTEXT, since our sockets are normal\n");
4145             return 0;
4146
4147         /* After a AcceptEx call succeeds, the socket can't be used with half of the
4148          * normal winsock functions on windows. We don't have that problem. */
4149         case WS_SO_UPDATE_ACCEPT_CONTEXT:
4150             TRACE("Ignoring SO_UPDATE_ACCEPT_CONTEXT, since our sockets are normal\n");
4151             return 0;
4152
4153         /* SO_OPENTYPE does not require a valid socket handle. */
4154         case WS_SO_OPENTYPE:
4155             if (!optlen || optlen < sizeof(int) || !optval)
4156             {
4157                 SetLastError(WSAEFAULT);
4158                 return SOCKET_ERROR;
4159             }
4160             get_per_thread_data()->opentype = *(const int *)optval;
4161             TRACE("setting global SO_OPENTYPE = 0x%x\n", *((const int*)optval) );
4162             return 0;
4163
4164 #ifdef SO_RCVTIMEO
4165         case WS_SO_RCVTIMEO:
4166 #endif
4167 #ifdef SO_SNDTIMEO
4168         case WS_SO_SNDTIMEO:
4169 #endif
4170 #if defined(SO_RCVTIMEO) || defined(SO_SNDTIMEO)
4171             if (optval && optlen == sizeof(UINT32)) {
4172                 /* WinSock passes milliseconds instead of struct timeval */
4173                 tval.tv_usec = (*(const UINT32*)optval % 1000) * 1000;
4174                 tval.tv_sec = *(const UINT32*)optval / 1000;
4175                 /* min of 500 milliseconds */
4176                 if (tval.tv_sec == 0 && tval.tv_usec && tval.tv_usec < 500000)
4177                     tval.tv_usec = 500000;
4178                 optlen = sizeof(struct timeval);
4179                 optval = (char*)&tval;
4180             } else if (optlen == sizeof(struct timeval)) {
4181                 WARN("SO_SND/RCVTIMEO for %d bytes: assuming unixism\n", optlen);
4182             } else {
4183                 WARN("SO_SND/RCVTIMEO for %d bytes is weird: ignored\n", optlen);
4184                 return 0;
4185             }
4186             convert_sockopt(&level, &optname);
4187             break;
4188 #endif
4189
4190         default:
4191             TRACE("Unknown SOL_SOCKET optname: 0x%08x\n", optname);
4192             SetLastError(WSAENOPROTOOPT);
4193             return SOCKET_ERROR;
4194         }
4195         break; /* case WS_SOL_SOCKET */
4196
4197 #ifdef HAVE_IPX
4198     case NSPROTO_IPX:
4199         switch(optname)
4200         {
4201         case IPX_PTYPE:
4202             fd = get_sock_fd( s, 0, NULL );
4203             TRACE("trying to set IPX_PTYPE: %d (fd: %d)\n", *(const int*)optval, fd);
4204
4205             /* We try to set the ipx type on ipx socket level. */
4206 #ifdef SOL_IPX
4207             if(setsockopt(fd, SOL_IPX, IPX_TYPE, optval, optlen) == -1)
4208             {
4209                 ERR("IPX: could not set ipx option type; expect weird behaviour\n");
4210                 release_sock_fd( s, fd );
4211                 return SOCKET_ERROR;
4212             }
4213 #else
4214             {
4215                 struct ipx val;
4216                 /* Should we retrieve val using a getsockopt call and then
4217                  * set the modified one? */
4218                 val.ipx_pt = *optval;
4219                 setsockopt(fd, 0, SO_DEFAULT_HEADERS, &val, sizeof(struct ipx));
4220             }
4221 #endif
4222             release_sock_fd( s, fd );
4223             return 0;
4224
4225         case IPX_FILTERPTYPE:
4226             /* Sets the receive filter packet type, at the moment we don't support it */
4227             FIXME("IPX_FILTERPTYPE: %x\n", *optval);
4228             /* Returning 0 is better for now than returning a SOCKET_ERROR */
4229             return 0;
4230
4231         default:
4232             FIXME("opt_name:%x\n", optname);
4233             return SOCKET_ERROR;
4234         }
4235         break; /* case NSPROTO_IPX */
4236 #endif
4237
4238     /* Levels WS_IPPROTO_TCP and WS_IPPROTO_IP convert directly */
4239     case WS_IPPROTO_TCP:
4240         switch(optname)
4241         {
4242         case WS_TCP_NODELAY:
4243             convert_sockopt(&level, &optname);
4244             break;
4245         default:
4246             FIXME("Unknown IPPROTO_TCP optname 0x%08x\n", optname);
4247             return SOCKET_ERROR;
4248         }
4249         break;
4250
4251     case WS_IPPROTO_IP:
4252         switch(optname)
4253         {
4254         case WS_IP_ADD_MEMBERSHIP:
4255         case WS_IP_DROP_MEMBERSHIP:
4256 #ifdef IP_HDRINCL
4257         case WS_IP_HDRINCL:
4258 #endif
4259         case WS_IP_MULTICAST_IF:
4260         case WS_IP_MULTICAST_LOOP:
4261         case WS_IP_MULTICAST_TTL:
4262         case WS_IP_OPTIONS:
4263 #ifdef IP_PKTINFO
4264         case WS_IP_PKTINFO:
4265 #endif
4266         case WS_IP_TOS:
4267         case WS_IP_TTL:
4268             convert_sockopt(&level, &optname);
4269             break;
4270         case WS_IP_DONTFRAGMENT:
4271             FIXME("IP_DONTFRAGMENT is silently ignored!\n");
4272             return 0;
4273         default:
4274             FIXME("Unknown IPPROTO_IP optname 0x%08x\n", optname);
4275             return SOCKET_ERROR;
4276         }
4277         break;
4278
4279     case WS_IPPROTO_IPV6:
4280         switch(optname)
4281         {
4282 #ifdef IPV6_ADD_MEMBERSHIP
4283         case WS_IPV6_ADD_MEMBERSHIP:
4284 #endif
4285 #ifdef IPV6_DROP_MEMBERSHIP
4286         case WS_IPV6_DROP_MEMBERSHIP:
4287 #endif
4288         case WS_IPV6_MULTICAST_IF:
4289         case WS_IPV6_MULTICAST_HOPS:
4290         case WS_IPV6_MULTICAST_LOOP:
4291         case WS_IPV6_UNICAST_HOPS:
4292         case WS_IPV6_V6ONLY:
4293             convert_sockopt(&level, &optname);
4294             break;
4295         case WS_IPV6_DONTFRAG:
4296             FIXME("IPV6_DONTFRAG is silently ignored!\n");
4297             return 0;
4298         default:
4299             FIXME("Unknown IPPROTO_IPV6 optname 0x%08x\n", optname);
4300             return SOCKET_ERROR;
4301         }
4302         break;
4303
4304     default:
4305         WARN("Unknown level: 0x%08x\n", level);
4306         SetLastError(WSAEINVAL);
4307         return SOCKET_ERROR;
4308     } /* end switch(level) */
4309
4310     /* avoid endianness issues if argument is a 16-bit int */
4311     if (optval && optlen < sizeof(int))
4312     {
4313         woptval= *((const INT16 *) optval);
4314         optval= (char*) &woptval;
4315         optlen=sizeof(int);
4316     }
4317     fd = get_sock_fd( s, 0, NULL );
4318     if (fd == -1) return SOCKET_ERROR;
4319
4320     if (setsockopt(fd, level, optname, optval, optlen) == 0)
4321     {
4322         release_sock_fd( s, fd );
4323         return 0;
4324     }
4325     TRACE("Setting socket error, %d\n", wsaErrno());
4326     SetLastError(wsaErrno());
4327     release_sock_fd( s, fd );
4328
4329     return SOCKET_ERROR;
4330 }
4331
4332 /***********************************************************************
4333  *              shutdown                (WS2_32.22)
4334  */
4335 int WINAPI WS_shutdown(SOCKET s, int how)
4336 {
4337     int fd, err = WSAENOTSOCK;
4338     unsigned int options, clear_flags = 0;
4339
4340     fd = get_sock_fd( s, 0, &options );
4341     TRACE("socket %04lx, how %i %x\n", s, how, options );
4342
4343     if (fd == -1)
4344         return SOCKET_ERROR;
4345
4346     switch( how )
4347     {
4348     case 0: /* drop receives */
4349         clear_flags |= FD_READ;
4350         break;
4351     case 1: /* drop sends */
4352         clear_flags |= FD_WRITE;
4353         break;
4354     case 2: /* drop all */
4355         clear_flags |= FD_READ|FD_WRITE;
4356         /*fall through */
4357     default:
4358         clear_flags |= FD_WINE_LISTENING;
4359     }
4360
4361     if (!(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)))
4362     {
4363         switch ( how )
4364         {
4365         case SD_RECEIVE:
4366             err = WS2_register_async_shutdown( s, ASYNC_TYPE_READ );
4367             break;
4368         case SD_SEND:
4369             err = WS2_register_async_shutdown( s, ASYNC_TYPE_WRITE );
4370             break;
4371         case SD_BOTH:
4372         default:
4373             err = WS2_register_async_shutdown( s, ASYNC_TYPE_READ );
4374             if (!err) err = WS2_register_async_shutdown( s, ASYNC_TYPE_WRITE );
4375             break;
4376         }
4377         if (err) goto error;
4378     }
4379     else /* non-overlapped mode */
4380     {
4381         if ( shutdown( fd, how ) )
4382         {
4383             err = wsaErrno();
4384             goto error;
4385         }
4386     }
4387
4388     release_sock_fd( s, fd );
4389     _enable_event( SOCKET2HANDLE(s), 0, 0, clear_flags );
4390     if ( how > 1) WSAAsyncSelect( s, 0, 0, 0 );
4391     return 0;
4392
4393 error:
4394     release_sock_fd( s, fd );
4395     _enable_event( SOCKET2HANDLE(s), 0, 0, clear_flags );
4396     WSASetLastError( err );
4397     return SOCKET_ERROR;
4398 }
4399
4400 /***********************************************************************
4401  *              socket          (WS2_32.23)
4402  */
4403 SOCKET WINAPI WS_socket(int af, int type, int protocol)
4404 {
4405     TRACE("af=%d type=%d protocol=%d\n", af, type, protocol);
4406
4407     return WSASocketA( af, type, protocol, NULL, 0,
4408                        get_per_thread_data()->opentype ? 0 : WSA_FLAG_OVERLAPPED );
4409 }
4410
4411
4412 /***********************************************************************
4413  *              gethostbyaddr           (WS2_32.51)
4414  */
4415 struct WS_hostent* WINAPI WS_gethostbyaddr(const char *addr, int len, int type)
4416 {
4417     struct WS_hostent *retval = NULL;
4418     struct hostent* host;
4419
4420 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
4421     char *extrabuf;
4422     int ebufsize=1024;
4423     struct hostent hostentry;
4424     int locerr=ENOBUFS;
4425     host = NULL;
4426     extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ;
4427     while(extrabuf) {
4428         int res = gethostbyaddr_r(addr, len, type,
4429                                   &hostentry, extrabuf, ebufsize, &host, &locerr);
4430         if( res != ERANGE) break;
4431         ebufsize *=2;
4432         extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ;
4433     }
4434     if (!host) SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr));
4435 #else
4436     EnterCriticalSection( &csWSgetXXXbyYYY );
4437     host = gethostbyaddr(addr, len, type);
4438     if (!host) SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno));
4439 #endif
4440     if( host != NULL ) retval = WS_dup_he(host);
4441 #ifdef  HAVE_LINUX_GETHOSTBYNAME_R_6
4442     HeapFree(GetProcessHeap(),0,extrabuf);
4443 #else
4444     LeaveCriticalSection( &csWSgetXXXbyYYY );
4445 #endif
4446     TRACE("ptr %p, len %d, type %d ret %p\n", addr, len, type, retval);
4447     return retval;
4448 }
4449
4450 /***********************************************************************
4451  *              WS_get_local_ips                (INTERNAL)
4452  *
4453  * Returns the list of local IP addresses by going through the network
4454  * adapters and using the local routing table to sort the addresses
4455  * from highest routing priority to lowest routing priority. This
4456  * functionality is inferred from the description for obtaining local
4457  * IP addresses given in the Knowledge Base Article Q160215.
4458  *
4459  * Please note that the returned hostent is only freed when the thread
4460  * closes and is replaced if another hostent is requested.
4461  */
4462 static struct WS_hostent* WS_get_local_ips( char *hostname )
4463 {
4464     int last_metric, numroutes = 0, i, j;
4465     PIP_ADAPTER_INFO adapters = NULL, k;
4466     struct WS_hostent *hostlist = NULL;
4467     PMIB_IPFORWARDTABLE routes = NULL;
4468     struct route *route_addrs = NULL;
4469     DWORD adap_size, route_size;
4470
4471     /* Obtain the size of the adapter list and routing table, also allocate memory */
4472     if (GetAdaptersInfo(NULL, &adap_size) != ERROR_BUFFER_OVERFLOW)
4473         return NULL;
4474     if (GetIpForwardTable(NULL, &route_size, FALSE) != ERROR_INSUFFICIENT_BUFFER)
4475         return NULL;
4476     adapters = HeapAlloc(GetProcessHeap(), 0, adap_size);
4477     routes = HeapAlloc(GetProcessHeap(), 0, route_size);
4478     route_addrs = HeapAlloc(GetProcessHeap(), 0, 0); /* HeapReAlloc doesn't work on NULL */
4479     if (adapters == NULL || routes == NULL || route_addrs == NULL)
4480         goto cleanup;
4481     /* Obtain the adapter list and the full routing table */
4482     if (GetAdaptersInfo(adapters, &adap_size) != NO_ERROR)
4483         goto cleanup;
4484     if (GetIpForwardTable(routes, &route_size, FALSE) != NO_ERROR)
4485         goto cleanup;
4486     /* Store the interface associated with each route */
4487     for (i = 0; i < routes->dwNumEntries; i++)
4488     {
4489         DWORD ifindex, ifmetric, exists = FALSE;
4490
4491         if (routes->table[i].dwForwardType != MIB_IPROUTE_TYPE_DIRECT)
4492             continue;
4493         ifindex = routes->table[i].dwForwardIfIndex;
4494         ifmetric = routes->table[i].dwForwardMetric1;
4495         /* Only store the lowest valued metric for an interface */
4496         for (j = 0; j < numroutes; j++)
4497         {
4498             if (route_addrs[j].interface == ifindex)
4499             {
4500                 if (route_addrs[j].metric > ifmetric)
4501                     route_addrs[j].metric = ifmetric;
4502                 exists = TRUE;
4503             }
4504         }
4505         if (exists)
4506             continue;
4507         route_addrs = HeapReAlloc(GetProcessHeap(), 0, route_addrs, (numroutes+1)*sizeof(struct route));
4508         if (route_addrs == NULL)
4509             goto cleanup; /* Memory allocation error, fail gracefully */
4510         route_addrs[numroutes].interface = ifindex;
4511         route_addrs[numroutes].metric = ifmetric;
4512         /* If no IP is found in the next step (for whatever reason)
4513          * then fall back to the magic loopback address.
4514          */
4515         memcpy(&(route_addrs[numroutes].addr.s_addr), magic_loopback_addr, 4);
4516         numroutes++;
4517     }
4518    if (numroutes == 0)
4519        goto cleanup; /* No routes, fall back to the Magic IP */
4520     /* Find the IP address associated with each found interface */
4521     for (i = 0; i < numroutes; i++)
4522     {
4523         for (k = adapters; k != NULL; k = k->Next)
4524         {
4525             char *ip = k->IpAddressList.IpAddress.String;
4526
4527             if (route_addrs[i].interface == k->Index)
4528                 route_addrs[i].addr.s_addr = (in_addr_t) inet_addr(ip);
4529         }
4530     }
4531     /* Allocate a hostent and enough memory for all the IPs,
4532      * including the NULL at the end of the list.
4533      */
4534     hostlist = WS_create_he(hostname, 1, numroutes+1, TRUE);
4535     if (hostlist == NULL)
4536         goto cleanup; /* Failed to allocate a hostent for the list of IPs */
4537     hostlist->h_addr_list[numroutes] = NULL; /* NULL-terminate the address list */
4538     hostlist->h_aliases[0] = NULL; /* NULL-terminate the alias list */
4539     hostlist->h_addrtype = AF_INET;
4540     hostlist->h_length = sizeof(struct in_addr); /* = 4 */
4541     /* Reorder the entries when placing them in the host list, Windows expects
4542      * the IP list in order from highest priority to lowest (the critical thing
4543      * is that most applications expect the first IP to be the default route).
4544      */
4545     last_metric = -1;
4546     for (i = 0; i < numroutes; i++)
4547     {
4548        struct in_addr addr;
4549        int metric = 0xFFFF;
4550
4551        memcpy(&addr, magic_loopback_addr, 4);
4552        for (j = 0; j < numroutes; j++)
4553        {
4554            int this_metric = route_addrs[j].metric;
4555
4556            if (this_metric > last_metric && this_metric < metric)
4557            {
4558                addr = route_addrs[j].addr;
4559                metric = this_metric;
4560            }
4561        }
4562        last_metric = metric;
4563        (*(struct in_addr *) hostlist->h_addr_list[i]) = addr;
4564     }
4565
4566     /* Cleanup all allocated memory except the address list,
4567      * the address list is used by the calling app.
4568      */
4569 cleanup:
4570     HeapFree(GetProcessHeap(), 0, route_addrs);
4571     HeapFree(GetProcessHeap(), 0, adapters);
4572     HeapFree(GetProcessHeap(), 0, routes);
4573     return hostlist;
4574 }
4575
4576 /***********************************************************************
4577  *              gethostbyname           (WS2_32.52)
4578  */
4579 struct WS_hostent* WINAPI WS_gethostbyname(const char* name)
4580 {
4581     struct WS_hostent *retval = NULL;
4582     struct hostent*     host;
4583 #ifdef  HAVE_LINUX_GETHOSTBYNAME_R_6
4584     char *extrabuf;
4585     int ebufsize=1024;
4586     struct hostent hostentry;
4587     int locerr = ENOBUFS;
4588 #endif
4589     char hostname[100];
4590     if(!num_startup) {
4591         SetLastError(WSANOTINITIALISED);
4592         return NULL;
4593     }
4594     if( gethostname( hostname, 100) == -1) {
4595         SetLastError( WSAENOBUFS); /* appropriate ? */
4596         return retval;
4597     }
4598     if( !name || !name[0]) {
4599         name = hostname;
4600     }
4601     /* If the hostname of the local machine is requested then return the
4602      * complete list of local IP addresses */
4603     if(strcmp(name, hostname) == 0)
4604         retval = WS_get_local_ips(hostname);
4605     /* If any other hostname was requested (or the routing table lookup failed)
4606      * then return the IP found by the host OS */
4607     if(retval == NULL)
4608     {
4609 #ifdef  HAVE_LINUX_GETHOSTBYNAME_R_6
4610         host = NULL;
4611         extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ;
4612         while(extrabuf) {
4613             int res = gethostbyname_r(name, &hostentry, extrabuf, ebufsize, &host, &locerr);
4614             if( res != ERANGE) break;
4615             ebufsize *=2;
4616             extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ;
4617         }
4618         if (!host) SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr));
4619 #else
4620         EnterCriticalSection( &csWSgetXXXbyYYY );
4621         host = gethostbyname(name);
4622         if (!host) SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno));
4623 #endif
4624         if (host) retval = WS_dup_he(host);
4625 #ifdef  HAVE_LINUX_GETHOSTBYNAME_R_6
4626         HeapFree(GetProcessHeap(),0,extrabuf);
4627 #else
4628         LeaveCriticalSection( &csWSgetXXXbyYYY );
4629 #endif
4630     }
4631     if (retval && retval->h_addr_list[0][0] == 127 &&
4632         strcmp(name, "localhost") != 0)
4633     {
4634         /* hostname != "localhost" but has loopback address. replace by our
4635          * special address.*/
4636         memcpy(retval->h_addr_list[0], magic_loopback_addr, 4);
4637     }
4638     TRACE( "%s ret %p\n", debugstr_a(name), retval );
4639     return retval;
4640 }
4641
4642
4643 /***********************************************************************
4644  *              getprotobyname          (WS2_32.53)
4645  */
4646 struct WS_protoent* WINAPI WS_getprotobyname(const char* name)
4647 {
4648     struct WS_protoent* retval = NULL;
4649 #ifdef HAVE_GETPROTOBYNAME
4650     struct protoent*     proto;
4651     EnterCriticalSection( &csWSgetXXXbyYYY );
4652     if( (proto = getprotobyname(name)) != NULL )
4653     {
4654         retval = WS_dup_pe(proto);
4655     }
4656     else {
4657         MESSAGE("protocol %s not found; You might want to add "
4658                 "this to /etc/protocols\n", debugstr_a(name) );
4659         SetLastError(WSANO_DATA);
4660     }
4661     LeaveCriticalSection( &csWSgetXXXbyYYY );
4662 #endif
4663     TRACE( "%s ret %p\n", debugstr_a(name), retval );
4664     return retval;
4665 }
4666
4667
4668 /***********************************************************************
4669  *              getprotobynumber        (WS2_32.54)
4670  */
4671 struct WS_protoent* WINAPI WS_getprotobynumber(int number)
4672 {
4673     struct WS_protoent* retval = NULL;
4674 #ifdef HAVE_GETPROTOBYNUMBER
4675     struct protoent*     proto;
4676     EnterCriticalSection( &csWSgetXXXbyYYY );
4677     if( (proto = getprotobynumber(number)) != NULL )
4678     {
4679         retval = WS_dup_pe(proto);
4680     }
4681     else {
4682         MESSAGE("protocol number %d not found; You might want to add "
4683                 "this to /etc/protocols\n", number );
4684         SetLastError(WSANO_DATA);
4685     }
4686     LeaveCriticalSection( &csWSgetXXXbyYYY );
4687 #endif
4688     TRACE("%i ret %p\n", number, retval);
4689     return retval;
4690 }
4691
4692
4693 /***********************************************************************
4694  *              getservbyname           (WS2_32.55)
4695  */
4696 struct WS_servent* WINAPI WS_getservbyname(const char *name, const char *proto)
4697 {
4698     struct WS_servent* retval = NULL;
4699     struct servent*     serv;
4700     char *name_str;
4701     char *proto_str = NULL;
4702
4703     if (!(name_str = strdup_lower(name))) return NULL;
4704
4705     if (proto && *proto)
4706     {
4707         if (!(proto_str = strdup_lower(proto)))
4708         {
4709             HeapFree( GetProcessHeap(), 0, name_str );
4710             return NULL;
4711         }
4712     }
4713
4714     EnterCriticalSection( &csWSgetXXXbyYYY );
4715     serv = getservbyname(name_str, proto_str);
4716     if( serv != NULL )
4717     {
4718         retval = WS_dup_se(serv);
4719     }
4720     else SetLastError(WSANO_DATA);
4721     LeaveCriticalSection( &csWSgetXXXbyYYY );
4722     HeapFree( GetProcessHeap(), 0, proto_str );
4723     HeapFree( GetProcessHeap(), 0, name_str );
4724     TRACE( "%s, %s ret %p\n", debugstr_a(name), debugstr_a(proto), retval );
4725     return retval;
4726 }
4727
4728 /***********************************************************************
4729  *              freeaddrinfo            (WS2_32.@)
4730  */
4731 void WINAPI WS_freeaddrinfo(struct WS_addrinfo *res)
4732 {
4733     while (res) {
4734         struct WS_addrinfo *next;
4735
4736         HeapFree(GetProcessHeap(),0,res->ai_canonname);
4737         HeapFree(GetProcessHeap(),0,res->ai_addr);
4738         next = res->ai_next;
4739         HeapFree(GetProcessHeap(),0,res);
4740         res = next;
4741     }
4742 }
4743
4744 /* helper functions for getaddrinfo()/getnameinfo() */
4745 static int convert_aiflag_w2u(int winflags) {
4746     unsigned int i;
4747     int unixflags = 0;
4748
4749     for (i=0;i<sizeof(ws_aiflag_map)/sizeof(ws_aiflag_map[0]);i++)
4750         if (ws_aiflag_map[i][0] & winflags) {
4751             unixflags |= ws_aiflag_map[i][1];
4752             winflags &= ~ws_aiflag_map[i][0];
4753         }
4754     if (winflags)
4755         FIXME("Unhandled windows AI_xxx flags %x\n", winflags);
4756     return unixflags;
4757 }
4758
4759 static int convert_niflag_w2u(int winflags) {
4760     unsigned int i;
4761     int unixflags = 0;
4762
4763     for (i=0;i<sizeof(ws_niflag_map)/sizeof(ws_niflag_map[0]);i++)
4764         if (ws_niflag_map[i][0] & winflags) {
4765             unixflags |= ws_niflag_map[i][1];
4766             winflags &= ~ws_niflag_map[i][0];
4767         }
4768     if (winflags)
4769         FIXME("Unhandled windows NI_xxx flags %x\n", winflags);
4770     return unixflags;
4771 }
4772
4773 static int convert_aiflag_u2w(int unixflags) {
4774     unsigned int i;
4775     int winflags = 0;
4776
4777     for (i=0;i<sizeof(ws_aiflag_map)/sizeof(ws_aiflag_map[0]);i++)
4778         if (ws_aiflag_map[i][1] & unixflags) {
4779             winflags |= ws_aiflag_map[i][0];
4780             unixflags &= ~ws_aiflag_map[i][1];
4781         }
4782     if (unixflags) /* will warn usually */
4783         WARN("Unhandled UNIX AI_xxx flags %x\n", unixflags);
4784     return winflags;
4785 }
4786
4787 static int convert_eai_u2w(int unixret) {
4788     int i;
4789
4790     for (i=0;ws_eai_map[i][0];i++)
4791         if (ws_eai_map[i][1] == unixret)
4792             return ws_eai_map[i][0];
4793     return unixret;
4794 }
4795
4796 static char *get_hostname(void)
4797 {
4798     char *ret;
4799     DWORD size = 0;
4800
4801     GetComputerNameExA( ComputerNamePhysicalDnsHostname, NULL, &size );
4802     if (GetLastError() != ERROR_MORE_DATA) return NULL;
4803     if (!(ret = HeapAlloc( GetProcessHeap(), 0, size ))) return NULL;
4804     if (!GetComputerNameExA( ComputerNamePhysicalDnsHostname, ret, &size ))
4805     {
4806         HeapFree( GetProcessHeap(), 0, ret );
4807         return NULL;
4808     }
4809     return ret;
4810 }
4811
4812 /***********************************************************************
4813  *              getaddrinfo             (WS2_32.@)
4814  */
4815 int WINAPI WS_getaddrinfo(LPCSTR nodename, LPCSTR servname, const struct WS_addrinfo *hints, struct WS_addrinfo **res)
4816 {
4817 #ifdef HAVE_GETADDRINFO
4818     struct addrinfo *unixaires = NULL;
4819     int   result;
4820     struct addrinfo unixhints, *punixhints = NULL;
4821     char *hostname = NULL;
4822     const char *node;
4823
4824     if (!nodename && !servname) return WSAHOST_NOT_FOUND;
4825
4826     if (!nodename)
4827         node = NULL;
4828     else if (!nodename[0])
4829     {
4830         node = hostname = get_hostname();
4831         if (!node) return WSA_NOT_ENOUGH_MEMORY;
4832     }
4833     else
4834         node = nodename;
4835
4836     if (hints) {
4837         punixhints = &unixhints;
4838
4839         memset(&unixhints, 0, sizeof(unixhints));
4840         punixhints->ai_flags    = convert_aiflag_w2u(hints->ai_flags);
4841         if (hints->ai_family == 0) /* wildcard, specific to getaddrinfo() */
4842             punixhints->ai_family = 0;
4843         else
4844             punixhints->ai_family = convert_af_w2u(hints->ai_family);
4845         if (hints->ai_socktype == 0) /* wildcard, specific to getaddrinfo() */
4846             punixhints->ai_socktype = 0;
4847         else
4848             punixhints->ai_socktype = convert_socktype_w2u(hints->ai_socktype);
4849         if (hints->ai_protocol == 0) /* wildcard, specific to getaddrinfo() */
4850             punixhints->ai_protocol = 0;
4851         else
4852             punixhints->ai_protocol = convert_proto_w2u(hints->ai_protocol);
4853     }
4854
4855     /* getaddrinfo(3) is thread safe, no need to wrap in CS */
4856     result = getaddrinfo(node, servname, punixhints, &unixaires);
4857
4858     TRACE("%s, %s %p -> %p %d\n", debugstr_a(nodename), debugstr_a(servname), hints, res, result);
4859     HeapFree(GetProcessHeap(), 0, hostname);
4860
4861     if (!result) {
4862         struct addrinfo *xuai = unixaires;
4863         struct WS_addrinfo **xai = res;
4864
4865         *xai = NULL;
4866         while (xuai) {
4867             struct WS_addrinfo *ai = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY, sizeof(struct WS_addrinfo));
4868             int len;
4869
4870             if (!ai)
4871                 goto outofmem;
4872
4873             *xai = ai;xai = &ai->ai_next;
4874             ai->ai_flags    = convert_aiflag_u2w(xuai->ai_flags);
4875             ai->ai_family   = convert_af_u2w(xuai->ai_family);
4876             ai->ai_socktype = convert_socktype_u2w(xuai->ai_socktype);
4877             ai->ai_protocol = convert_proto_u2w(xuai->ai_protocol);
4878             if (xuai->ai_canonname) {
4879                 TRACE("canon name - %s\n",debugstr_a(xuai->ai_canonname));
4880                 ai->ai_canonname = HeapAlloc(GetProcessHeap(),0,strlen(xuai->ai_canonname)+1);
4881                 if (!ai->ai_canonname)
4882                     goto outofmem;
4883                 strcpy(ai->ai_canonname,xuai->ai_canonname);
4884             }
4885             len = xuai->ai_addrlen;
4886             ai->ai_addr = HeapAlloc(GetProcessHeap(),0,len);
4887             if (!ai->ai_addr)
4888                 goto outofmem;
4889             ai->ai_addrlen = len;
4890             do {
4891                 int winlen = ai->ai_addrlen;
4892
4893                 if (!ws_sockaddr_u2ws(xuai->ai_addr, ai->ai_addr, &winlen)) {
4894                     ai->ai_addrlen = winlen;
4895                     break;
4896                 }
4897                 len = 2*len;
4898                 ai->ai_addr = HeapReAlloc(GetProcessHeap(),0,ai->ai_addr,len);
4899                 if (!ai->ai_addr)
4900                     goto outofmem;
4901                 ai->ai_addrlen = len;
4902             } while (1);
4903             xuai = xuai->ai_next;
4904         }
4905         freeaddrinfo(unixaires);
4906     } else {
4907         result = convert_eai_u2w(result);
4908         *res = NULL;
4909     }
4910     return result;
4911
4912 outofmem:
4913     if (*res) WS_freeaddrinfo(*res);
4914     if (unixaires) freeaddrinfo(unixaires);
4915     *res = NULL;
4916     return WSA_NOT_ENOUGH_MEMORY;
4917 #else
4918     FIXME("getaddrinfo() failed, not found during buildtime.\n");
4919     return EAI_FAIL;
4920 #endif
4921 }
4922
4923 static struct WS_addrinfoW *addrinfo_AtoW(const struct WS_addrinfo *ai)
4924 {
4925     struct WS_addrinfoW *ret;
4926
4927     if (!(ret = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_addrinfoW)))) return NULL;
4928     ret->ai_flags     = ai->ai_flags;
4929     ret->ai_family    = ai->ai_family;
4930     ret->ai_socktype  = ai->ai_socktype;
4931     ret->ai_protocol  = ai->ai_protocol;
4932     ret->ai_addrlen   = ai->ai_addrlen;
4933     ret->ai_canonname = NULL;
4934     ret->ai_addr      = NULL;
4935     ret->ai_next      = NULL;
4936     if (ai->ai_canonname)
4937     {
4938         int len = MultiByteToWideChar(CP_ACP, 0, ai->ai_canonname, -1, NULL, 0);
4939         if (!(ret->ai_canonname = HeapAlloc(GetProcessHeap(), 0, len)))
4940         {
4941             HeapFree(GetProcessHeap(), 0, ret);
4942             return NULL;
4943         }
4944         MultiByteToWideChar(CP_ACP, 0, ai->ai_canonname, -1, ret->ai_canonname, len);
4945     }
4946     if (ai->ai_addr)
4947     {
4948         if (!(ret->ai_addr = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_sockaddr))))
4949         {
4950             HeapFree(GetProcessHeap(), 0, ret->ai_canonname);
4951             HeapFree(GetProcessHeap(), 0, ret);
4952             return NULL;
4953         }
4954         memcpy(ret->ai_addr, ai->ai_addr, sizeof(struct WS_sockaddr));
4955     }
4956     return ret;
4957 }
4958
4959 static struct WS_addrinfoW *addrinfo_list_AtoW(const struct WS_addrinfo *info)
4960 {
4961     struct WS_addrinfoW *ret, *infoW;
4962
4963     if (!(ret = infoW = addrinfo_AtoW(info))) return NULL;
4964     while (info->ai_next)
4965     {
4966         if (!(infoW->ai_next = addrinfo_AtoW(info->ai_next)))
4967         {
4968             FreeAddrInfoW(ret);
4969             return NULL;
4970         }
4971         infoW = infoW->ai_next;
4972         info = info->ai_next;
4973     }
4974     return ret;
4975 }
4976
4977 static struct WS_addrinfo *addrinfo_WtoA(const struct WS_addrinfoW *ai)
4978 {
4979     struct WS_addrinfo *ret;
4980
4981     if (!(ret = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_addrinfo)))) return NULL;
4982     ret->ai_flags     = ai->ai_flags;
4983     ret->ai_family    = ai->ai_family;
4984     ret->ai_socktype  = ai->ai_socktype;
4985     ret->ai_protocol  = ai->ai_protocol;
4986     ret->ai_addrlen   = ai->ai_addrlen;
4987     ret->ai_canonname = NULL;
4988     ret->ai_addr      = NULL;
4989     ret->ai_next      = NULL;
4990     if (ai->ai_canonname)
4991     {
4992         int len = WideCharToMultiByte(CP_ACP, 0, ai->ai_canonname, -1, NULL, 0, NULL, NULL);
4993         if (!(ret->ai_canonname = HeapAlloc(GetProcessHeap(), 0, len)))
4994         {
4995             HeapFree(GetProcessHeap(), 0, ret);
4996             return NULL;
4997         }
4998         WideCharToMultiByte(CP_ACP, 0, ai->ai_canonname, -1, ret->ai_canonname, len, NULL, NULL);
4999     }
5000     if (ai->ai_addr)
5001     {
5002         if (!(ret->ai_addr = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_sockaddr))))
5003         {
5004             HeapFree(GetProcessHeap(), 0, ret->ai_canonname);
5005             HeapFree(GetProcessHeap(), 0, ret);
5006             return NULL;
5007         }
5008         memcpy(ret->ai_addr, ai->ai_addr, sizeof(struct WS_sockaddr));
5009     }
5010     return ret;
5011 }
5012
5013 /***********************************************************************
5014  *              GetAddrInfoW            (WS2_32.@)
5015  */
5016 int WINAPI GetAddrInfoW(LPCWSTR nodename, LPCWSTR servname, const ADDRINFOW *hints, PADDRINFOW *res)
5017 {
5018     int ret, len;
5019     char *nodenameA = NULL, *servnameA = NULL;
5020     struct WS_addrinfo *resA, *hintsA = NULL;
5021
5022     if (nodename)
5023     {
5024         len = WideCharToMultiByte(CP_ACP, 0, nodename, -1, NULL, 0, NULL, NULL);
5025         if (!(nodenameA = HeapAlloc(GetProcessHeap(), 0, len))) return EAI_MEMORY;
5026         WideCharToMultiByte(CP_ACP, 0, nodename, -1, nodenameA, len, NULL, NULL);
5027     }
5028     if (servname)
5029     {
5030         len = WideCharToMultiByte(CP_ACP, 0, servname, -1, NULL, 0, NULL, NULL);
5031         if (!(servnameA = HeapAlloc(GetProcessHeap(), 0, len)))
5032         {
5033             HeapFree(GetProcessHeap(), 0, nodenameA);
5034             return EAI_MEMORY;
5035         }
5036         WideCharToMultiByte(CP_ACP, 0, servname, -1, servnameA, len, NULL, NULL);
5037     }
5038
5039     if (hints) hintsA = addrinfo_WtoA(hints);
5040     ret = WS_getaddrinfo(nodenameA, servnameA, hintsA, &resA);
5041     WS_freeaddrinfo(hintsA);
5042
5043     if (!ret)
5044     {
5045         *res = addrinfo_list_AtoW(resA);
5046         WS_freeaddrinfo(resA);
5047     }
5048
5049     HeapFree(GetProcessHeap(), 0, nodenameA);
5050     HeapFree(GetProcessHeap(), 0, servnameA);
5051     return ret;
5052 }
5053
5054 /***********************************************************************
5055  *      FreeAddrInfoW        (WS2_32.@)
5056  */
5057 void WINAPI FreeAddrInfoW(PADDRINFOW ai)
5058 {
5059     while (ai)
5060     {
5061         ADDRINFOW *next;
5062         HeapFree(GetProcessHeap(), 0, ai->ai_canonname);
5063         HeapFree(GetProcessHeap(), 0, ai->ai_addr);
5064         next = ai->ai_next;
5065         HeapFree(GetProcessHeap(), 0, ai);
5066         ai = next;
5067     }
5068 }
5069
5070 int WINAPI WS_getnameinfo(const SOCKADDR *sa, WS_socklen_t salen, PCHAR host,
5071                           DWORD hostlen, PCHAR serv, DWORD servlen, INT flags)
5072 {
5073 #ifdef HAVE_GETNAMEINFO
5074     int ret;
5075     union generic_unix_sockaddr sa_u;
5076     unsigned int size;
5077
5078     TRACE("%s %d %p %d %p %d %d\n", debugstr_sockaddr(sa), salen, host, hostlen,
5079           serv, servlen, flags);
5080
5081     size = ws_sockaddr_ws2u(sa, salen, &sa_u);
5082     if (!size)
5083     {
5084         WSASetLastError(WSAEFAULT);
5085         return WSA_NOT_ENOUGH_MEMORY;
5086     }
5087     ret = getnameinfo(&sa_u.addr, size, host, hostlen, serv, servlen, convert_niflag_w2u(flags));
5088     return convert_eai_u2w(ret);
5089 #else
5090     FIXME("getnameinfo() failed, not found during buildtime.\n");
5091     return EAI_FAIL;
5092 #endif
5093 }
5094
5095 int WINAPI GetNameInfoW(const SOCKADDR *sa, WS_socklen_t salen, PWCHAR host,
5096                         DWORD hostlen, PWCHAR serv, DWORD servlen, INT flags)
5097 {
5098     int ret;
5099     char *hostA = NULL, *servA = NULL;
5100
5101     if (host && (!(hostA = HeapAlloc(GetProcessHeap(), 0, hostlen)))) return EAI_MEMORY;
5102     if (serv && (!(servA = HeapAlloc(GetProcessHeap(), 0, servlen))))
5103     {
5104         HeapFree(GetProcessHeap(), 0, hostA);
5105         return EAI_MEMORY;
5106     }
5107
5108     ret = WS_getnameinfo(sa, salen, hostA, hostlen, servA, servlen, flags);
5109     if (!ret)
5110     {
5111         if (host) MultiByteToWideChar(CP_ACP, 0, hostA, -1, host, hostlen);
5112         if (serv) MultiByteToWideChar(CP_ACP, 0, servA, -1, serv, servlen);
5113     }
5114
5115     HeapFree(GetProcessHeap(), 0, hostA);
5116     HeapFree(GetProcessHeap(), 0, servA);
5117     return ret;
5118 }
5119
5120 /***********************************************************************
5121  *              getservbyport           (WS2_32.56)
5122  */
5123 struct WS_servent* WINAPI WS_getservbyport(int port, const char *proto)
5124 {
5125     struct WS_servent* retval = NULL;
5126 #ifdef HAVE_GETSERVBYPORT
5127     struct servent*     serv;
5128     char *proto_str = NULL;
5129
5130     if (proto && *proto)
5131     {
5132         if (!(proto_str = strdup_lower(proto))) return NULL;
5133     }
5134     EnterCriticalSection( &csWSgetXXXbyYYY );
5135     if( (serv = getservbyport(port, proto_str)) != NULL ) {
5136         retval = WS_dup_se(serv);
5137     }
5138     else SetLastError(WSANO_DATA);
5139     LeaveCriticalSection( &csWSgetXXXbyYYY );
5140     HeapFree( GetProcessHeap(), 0, proto_str );
5141 #endif
5142     TRACE("%d (i.e. port %d), %s ret %p\n", port, (int)ntohl(port), debugstr_a(proto), retval);
5143     return retval;
5144 }
5145
5146
5147 /***********************************************************************
5148  *              gethostname           (WS2_32.57)
5149  */
5150 int WINAPI WS_gethostname(char *name, int namelen)
5151 {
5152     TRACE("name %p, len %d\n", name, namelen);
5153
5154     if (gethostname(name, namelen) == 0)
5155     {
5156         TRACE("<- '%s'\n", name);
5157         return 0;
5158     }
5159     SetLastError((errno == EINVAL) ? WSAEFAULT : wsaErrno());
5160     TRACE("<- ERROR !\n");
5161     return SOCKET_ERROR;
5162 }
5163
5164
5165 /* ------------------------------------- Windows sockets extensions -- *
5166  *                                                                     *
5167  * ------------------------------------------------------------------- */
5168
5169 /***********************************************************************
5170  *              WSAEnumNetworkEvents (WS2_32.36)
5171  */
5172 int WINAPI WSAEnumNetworkEvents(SOCKET s, WSAEVENT hEvent, LPWSANETWORKEVENTS lpEvent)
5173 {
5174     int ret;
5175     int i;
5176     int errors[FD_MAX_EVENTS];
5177
5178     TRACE("%08lx, hEvent %p, lpEvent %p\n", s, hEvent, lpEvent );
5179
5180     SERVER_START_REQ( get_socket_event )
5181     {
5182         req->handle  = wine_server_obj_handle( SOCKET2HANDLE(s) );
5183         req->service = TRUE;
5184         req->c_event = wine_server_obj_handle( hEvent );
5185         wine_server_set_reply( req, errors, sizeof(errors) );
5186         if (!(ret = wine_server_call(req))) lpEvent->lNetworkEvents = reply->pmask & reply->mask;
5187     }
5188     SERVER_END_REQ;
5189     if (!ret)
5190     {
5191         for (i = 0; i < FD_MAX_EVENTS; i++)
5192             lpEvent->iErrorCode[i] = NtStatusToWSAError(errors[i]);
5193         return 0;
5194     }
5195     SetLastError(WSAEINVAL);
5196     return SOCKET_ERROR;
5197 }
5198
5199 /***********************************************************************
5200  *              WSAEventSelect (WS2_32.39)
5201  */
5202 int WINAPI WSAEventSelect(SOCKET s, WSAEVENT hEvent, LONG lEvent)
5203 {
5204     int ret;
5205
5206     TRACE("%08lx, hEvent %p, event %08x\n", s, hEvent, lEvent);
5207
5208     SERVER_START_REQ( set_socket_event )
5209     {
5210         req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
5211         req->mask   = lEvent;
5212         req->event  = wine_server_obj_handle( hEvent );
5213         req->window = 0;
5214         req->msg    = 0;
5215         ret = wine_server_call( req );
5216     }
5217     SERVER_END_REQ;
5218     if (!ret) return 0;
5219     SetLastError(WSAEINVAL);
5220     return SOCKET_ERROR;
5221 }
5222
5223 /**********************************************************************
5224  *      WSAGetOverlappedResult (WS2_32.40)
5225  */
5226 BOOL WINAPI WSAGetOverlappedResult( SOCKET s, LPWSAOVERLAPPED lpOverlapped,
5227                                     LPDWORD lpcbTransfer, BOOL fWait,
5228                                     LPDWORD lpdwFlags )
5229 {
5230     NTSTATUS status;
5231
5232     TRACE( "socket %04lx ovl %p trans %p, wait %d flags %p\n",
5233            s, lpOverlapped, lpcbTransfer, fWait, lpdwFlags );
5234
5235     if ( lpOverlapped == NULL )
5236     {
5237         ERR( "Invalid pointer\n" );
5238         WSASetLastError(WSA_INVALID_PARAMETER);
5239         return FALSE;
5240     }
5241
5242     status = lpOverlapped->Internal;
5243     if (status == STATUS_PENDING)
5244     {
5245         if (!fWait)
5246         {
5247             SetLastError( WSA_IO_INCOMPLETE );
5248             return FALSE;
5249         }
5250
5251         if (WaitForSingleObject( lpOverlapped->hEvent ? lpOverlapped->hEvent : SOCKET2HANDLE(s),
5252                                  INFINITE ) == WAIT_FAILED)
5253             return FALSE;
5254         status = lpOverlapped->Internal;
5255     }
5256
5257     if ( lpcbTransfer )
5258         *lpcbTransfer = lpOverlapped->InternalHigh;
5259
5260     if ( lpdwFlags )
5261         *lpdwFlags = lpOverlapped->u.s.Offset;
5262
5263     if (status) SetLastError( RtlNtStatusToDosError(status) );
5264     return !status;
5265 }
5266
5267
5268 /***********************************************************************
5269  *      WSAAsyncSelect                  (WS2_32.101)
5270  */
5271 INT WINAPI WSAAsyncSelect(SOCKET s, HWND hWnd, UINT uMsg, LONG lEvent)
5272 {
5273     int ret;
5274
5275     TRACE("%lx, hWnd %p, uMsg %08x, event %08x\n", s, hWnd, uMsg, lEvent);
5276
5277     SERVER_START_REQ( set_socket_event )
5278     {
5279         req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
5280         req->mask   = lEvent;
5281         req->event  = 0;
5282         req->window = wine_server_user_handle( hWnd );
5283         req->msg    = uMsg;
5284         ret = wine_server_call( req );
5285     }
5286     SERVER_END_REQ;
5287     if (!ret) return 0;
5288     SetLastError(WSAEINVAL);
5289     return SOCKET_ERROR;
5290 }
5291
5292 /***********************************************************************
5293  *      WSACreateEvent          (WS2_32.31)
5294  *
5295  */
5296 WSAEVENT WINAPI WSACreateEvent(void)
5297 {
5298     /* Create a manual-reset event, with initial state: unsignaled */
5299     TRACE("\n");
5300
5301     return CreateEventW(NULL, TRUE, FALSE, NULL);
5302 }
5303
5304 /***********************************************************************
5305  *      WSACloseEvent          (WS2_32.29)
5306  *
5307  */
5308 BOOL WINAPI WSACloseEvent(WSAEVENT event)
5309 {
5310     TRACE ("event=%p\n", event);
5311
5312     return CloseHandle(event);
5313 }
5314
5315 /***********************************************************************
5316  *      WSASocketA          (WS2_32.78)
5317  *
5318  */
5319 SOCKET WINAPI WSASocketA(int af, int type, int protocol,
5320                          LPWSAPROTOCOL_INFOA lpProtocolInfo,
5321                          GROUP g, DWORD dwFlags)
5322 {
5323     INT len;
5324     WSAPROTOCOL_INFOW info;
5325
5326     TRACE("af=%d type=%d protocol=%d protocol_info=%p group=%d flags=0x%x\n",
5327           af, type, protocol, lpProtocolInfo, g, dwFlags);
5328
5329     if (!lpProtocolInfo) return WSASocketW(af, type, protocol, NULL, g, dwFlags);
5330
5331     memcpy(&info, lpProtocolInfo, FIELD_OFFSET(WSAPROTOCOL_INFOW, szProtocol));
5332     len = MultiByteToWideChar(CP_ACP, 0, lpProtocolInfo->szProtocol, -1,
5333                               info.szProtocol, WSAPROTOCOL_LEN + 1);
5334
5335     if (!len)
5336     {
5337         WSASetLastError( WSAEINVAL);
5338         return SOCKET_ERROR;
5339     }
5340
5341     return WSASocketW(af, type, protocol, &info, g, dwFlags);
5342 }
5343
5344 /***********************************************************************
5345  *      WSASocketW          (WS2_32.79)
5346  *
5347  */
5348 SOCKET WINAPI WSASocketW(int af, int type, int protocol,
5349                          LPWSAPROTOCOL_INFOW lpProtocolInfo,
5350                          GROUP g, DWORD dwFlags)
5351 {
5352     SOCKET ret;
5353
5354    /*
5355       FIXME: The "advanced" parameters of WSASocketW (lpProtocolInfo,
5356       g, dwFlags except WSA_FLAG_OVERLAPPED) are ignored.
5357    */
5358
5359    TRACE("af=%d type=%d protocol=%d protocol_info=%p group=%d flags=0x%x\n",
5360          af, type, protocol, lpProtocolInfo, g, dwFlags );
5361
5362     /* hack for WSADuplicateSocket */
5363     if (lpProtocolInfo && lpProtocolInfo->dwServiceFlags4 == 0xff00ff00) {
5364       ret = lpProtocolInfo->dwCatalogEntryId;
5365       TRACE("\tgot duplicate %04lx\n", ret);
5366       return ret;
5367     }
5368
5369     /* convert the socket family and type */
5370     af = convert_af_w2u(af);
5371     type = convert_socktype_w2u(type);
5372
5373     if (lpProtocolInfo)
5374     {
5375         if (af == FROM_PROTOCOL_INFO)
5376             af = lpProtocolInfo->iAddressFamily;
5377         if (type == FROM_PROTOCOL_INFO)
5378             type = lpProtocolInfo->iSocketType;
5379         if (protocol == FROM_PROTOCOL_INFO)
5380             protocol = lpProtocolInfo->iProtocol;
5381     }
5382
5383     if ( af == AF_UNSPEC)  /* did they not specify the address family? */
5384     {
5385         if ((protocol == IPPROTO_TCP && type == SOCK_STREAM) ||
5386             (protocol == IPPROTO_UDP && type == SOCK_DGRAM))
5387         {
5388             af = AF_INET;
5389         }
5390         else
5391         {
5392             SetLastError(WSAEPROTOTYPE);
5393             return INVALID_SOCKET;
5394         }
5395     }
5396
5397     SERVER_START_REQ( create_socket )
5398     {
5399         req->family     = af;
5400         req->type       = type;
5401         req->protocol   = protocol;
5402         req->access     = GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE;
5403         req->attributes = OBJ_INHERIT;
5404         req->flags      = dwFlags;
5405         set_error( wine_server_call( req ) );
5406         ret = HANDLE2SOCKET( wine_server_ptr_handle( reply->handle ));
5407     }
5408     SERVER_END_REQ;
5409     if (ret)
5410     {
5411         TRACE("\tcreated %04lx\n", ret );
5412         return ret;
5413     }
5414
5415     if (GetLastError() == WSAEACCES) /* raw socket denied */
5416     {
5417         if (type == SOCK_RAW)
5418             ERR_(winediag)("Failed to create a socket of type SOCK_RAW, this requires special permissions.\n");
5419         else
5420             ERR_(winediag)("Failed to create socket, this requires special permissions.\n");
5421         SetLastError(WSAESOCKTNOSUPPORT);
5422     }
5423
5424     WARN("\t\tfailed!\n");
5425     return INVALID_SOCKET;
5426 }
5427
5428 /***********************************************************************
5429  *      WSAJoinLeaf          (WS2_32.58)
5430  *
5431  */
5432 SOCKET WINAPI WSAJoinLeaf(
5433         SOCKET s,
5434         const struct WS_sockaddr *addr,
5435         int addrlen,
5436         LPWSABUF lpCallerData,
5437         LPWSABUF lpCalleeData,
5438         LPQOS lpSQOS,
5439         LPQOS lpGQOS,
5440         DWORD dwFlags)
5441 {
5442     FIXME("stub.\n");
5443     return INVALID_SOCKET;
5444 }
5445
5446 /***********************************************************************
5447  *      __WSAFDIsSet                    (WS2_32.151)
5448  */
5449 int WINAPI __WSAFDIsSet(SOCKET s, WS_fd_set *set)
5450 {
5451   int i = set->fd_count;
5452
5453   TRACE("(%ld,%p(%i))\n", s, set, i);
5454
5455   while (i--)
5456       if (set->fd_array[i] == s) return 1;
5457   return 0;
5458 }
5459
5460 /***********************************************************************
5461  *      WSAIsBlocking                   (WS2_32.114)
5462  */
5463 BOOL WINAPI WSAIsBlocking(void)
5464 {
5465   /* By default WinSock should set all its sockets to non-blocking mode
5466    * and poll in PeekMessage loop when processing "blocking" ones. This
5467    * function is supposed to tell if the program is in this loop. Our
5468    * blocking calls are truly blocking so we always return FALSE.
5469    *
5470    * Note: It is allowed to call this function without prior WSAStartup().
5471    */
5472
5473   TRACE("\n");
5474   return FALSE;
5475 }
5476
5477 /***********************************************************************
5478  *      WSACancelBlockingCall           (WS2_32.113)
5479  */
5480 INT WINAPI WSACancelBlockingCall(void)
5481 {
5482     TRACE("\n");
5483     return 0;
5484 }
5485
5486 static INT WINAPI WSA_DefaultBlockingHook( FARPROC x )
5487 {
5488     FIXME("How was this called?\n");
5489     return x();
5490 }
5491
5492
5493 /***********************************************************************
5494  *      WSASetBlockingHook (WS2_32.109)
5495  */
5496 FARPROC WINAPI WSASetBlockingHook(FARPROC lpBlockFunc)
5497 {
5498   FARPROC prev = blocking_hook;
5499   blocking_hook = lpBlockFunc;
5500   TRACE("hook %p\n", lpBlockFunc);
5501   return prev;
5502 }
5503
5504
5505 /***********************************************************************
5506  *      WSAUnhookBlockingHook (WS2_32.110)
5507  */
5508 INT WINAPI WSAUnhookBlockingHook(void)
5509 {
5510     blocking_hook = (FARPROC)WSA_DefaultBlockingHook;
5511     return 0;
5512 }
5513
5514
5515 /* ----------------------------------- end of API stuff */
5516
5517 /* ----------------------------------- helper functions -
5518  *
5519  * TODO: Merge WS_dup_..() stuff into one function that
5520  * would operate with a generic structure containing internal
5521  * pointers (via a template of some kind).
5522  */
5523
5524 static int list_size(char** l, int item_size)
5525 {
5526   int i,j = 0;
5527   if(l)
5528   { for(i=0;l[i];i++)
5529         j += (item_size) ? item_size : strlen(l[i]) + 1;
5530     j += (i + 1) * sizeof(char*); }
5531   return j;
5532 }
5533
5534 static int list_dup(char** l_src, char** l_to, int item_size)
5535 {
5536    char *p;
5537    int i;
5538
5539    for (i = 0; l_src[i]; i++) ;
5540    p = (char *)(l_to + i + 1);
5541    for (i = 0; l_src[i]; i++)
5542    {
5543        int count = ( item_size ) ? item_size : strlen(l_src[i]) + 1;
5544        memcpy(p, l_src[i], count);
5545        l_to[i] = p;
5546        p += count;
5547    }
5548    l_to[i] = NULL;
5549    return p - (char *)l_to;
5550 }
5551
5552 /* ----- hostent */
5553
5554 /* create a hostent entry
5555  *
5556  * Creates the entry with enough memory for the name, aliases
5557  * addresses, and the address pointers.  Also copies the name
5558  * and sets up all the pointers.  If "fill_addresses" is set then
5559  * sufficient memory for the addresses is also allocated and the
5560  * address pointers are set to this memory.
5561  *
5562  * NOTE: The alias and address lists must be allocated with room
5563  * for the NULL item terminating the list.  This is true even if
5564  * the list has no items ("aliases" and "addresses" must be
5565  * at least "1", a truly empty list is invalid).
5566  */
5567 static struct WS_hostent *WS_create_he(char *name, int aliases, int addresses, int fill_addresses)
5568 {
5569     struct WS_hostent *p_to;
5570     char *p;
5571
5572     int size = (sizeof(struct WS_hostent) +
5573                 strlen(name) + 1 +
5574                 sizeof(char *)*aliases +
5575                 sizeof(char *)*addresses);
5576
5577     /* Allocate enough memory for the addresses */
5578     if (fill_addresses)
5579         size += sizeof(struct in_addr)*addresses;
5580
5581     if (!(p_to = check_buffer_he(size))) return NULL;
5582     memset(p_to, 0, size);
5583
5584     p = (char *)(p_to + 1);
5585     p_to->h_name = p;
5586     strcpy(p, name);
5587     p += strlen(p) + 1;
5588
5589     p_to->h_aliases = (char **)p;
5590     p += sizeof(char *)*aliases;
5591     p_to->h_addr_list = (char **)p;
5592     p += sizeof(char *)*addresses;
5593     if (fill_addresses)
5594     {
5595         int i;
5596
5597         /* NOTE: h_aliases must be filled in manually, leave these
5598          * pointers NULL (already set to NULL by memset earlier).
5599          */
5600
5601         /* Fill in the list of address pointers */
5602         for (i = 0; i < addresses; i++)
5603             p_to->h_addr_list[i] = (p += sizeof(struct in_addr));
5604         p += sizeof(struct in_addr);
5605     }
5606     return p_to;
5607 }
5608
5609 /* duplicate hostent entry
5610  * and handle all Win16/Win32 dependent things (struct size, ...) *correctly*.
5611  * Ditto for protoent and servent.
5612  */
5613 static struct WS_hostent *WS_dup_he(const struct hostent* p_he)
5614 {
5615     int addresses = list_size(p_he->h_addr_list, p_he->h_length);
5616     int aliases = list_size(p_he->h_aliases, 0);
5617     struct WS_hostent *p_to;
5618
5619     p_to = WS_create_he(p_he->h_name, aliases, addresses, FALSE);
5620
5621     if (!p_to) return NULL;
5622     p_to->h_addrtype = p_he->h_addrtype;
5623     p_to->h_length = p_he->h_length;
5624
5625     list_dup(p_he->h_aliases, p_to->h_aliases, 0);
5626     list_dup(p_he->h_addr_list, p_to->h_addr_list, p_he->h_length);
5627     return p_to;
5628 }
5629
5630 /* ----- protoent */
5631
5632 static struct WS_protoent *WS_dup_pe(const struct protoent* p_pe)
5633 {
5634     char *p;
5635     struct WS_protoent *p_to;
5636
5637     int size = (sizeof(*p_pe) +
5638                 strlen(p_pe->p_name) + 1 +
5639                 list_size(p_pe->p_aliases, 0));
5640
5641     if (!(p_to = check_buffer_pe(size))) return NULL;
5642     p_to->p_proto = p_pe->p_proto;
5643
5644     p = (char *)(p_to + 1);
5645     p_to->p_name = p;
5646     strcpy(p, p_pe->p_name);
5647     p += strlen(p) + 1;
5648
5649     p_to->p_aliases = (char **)p;
5650     list_dup(p_pe->p_aliases, p_to->p_aliases, 0);
5651     return p_to;
5652 }
5653
5654 /* ----- servent */
5655
5656 static struct WS_servent *WS_dup_se(const struct servent* p_se)
5657 {
5658     char *p;
5659     struct WS_servent *p_to;
5660
5661     int size = (sizeof(*p_se) +
5662                 strlen(p_se->s_proto) + 1 +
5663                 strlen(p_se->s_name) + 1 +
5664                 list_size(p_se->s_aliases, 0));
5665
5666     if (!(p_to = check_buffer_se(size))) return NULL;
5667     p_to->s_port = p_se->s_port;
5668
5669     p = (char *)(p_to + 1);
5670     p_to->s_name = p;
5671     strcpy(p, p_se->s_name);
5672     p += strlen(p) + 1;
5673
5674     p_to->s_proto = p;
5675     strcpy(p, p_se->s_proto);
5676     p += strlen(p) + 1;
5677
5678     p_to->s_aliases = (char **)p;
5679     list_dup(p_se->s_aliases, p_to->s_aliases, 0);
5680     return p_to;
5681 }
5682
5683
5684 /***********************************************************************
5685  *              WSARecv                 (WS2_32.67)
5686  */
5687 int WINAPI WSARecv(SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
5688                    LPDWORD NumberOfBytesReceived, LPDWORD lpFlags,
5689                    LPWSAOVERLAPPED lpOverlapped,
5690                    LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine)
5691 {
5692     return WS2_recv_base(s, lpBuffers, dwBufferCount, NumberOfBytesReceived, lpFlags,
5693                        NULL, NULL, lpOverlapped, lpCompletionRoutine, NULL);
5694 }
5695
5696 static int WS2_recv_base( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
5697                           LPDWORD lpNumberOfBytesRecvd, LPDWORD lpFlags,
5698                           struct WS_sockaddr *lpFrom,
5699                           LPINT lpFromlen, LPWSAOVERLAPPED lpOverlapped,
5700                           LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine,
5701                           LPWSABUF lpControlBuffer )
5702 {
5703     unsigned int i, options;
5704     int n, fd, err;
5705     struct ws2_async *wsa;
5706     DWORD timeout_start = GetTickCount();
5707     ULONG_PTR cvalue = (lpOverlapped && ((ULONG_PTR)lpOverlapped->hEvent & 1) == 0) ? (ULONG_PTR)lpOverlapped : 0;
5708
5709     TRACE("socket %04lx, wsabuf %p, nbufs %d, flags %d, from %p, fromlen %d, ovl %p, func %p\n",
5710           s, lpBuffers, dwBufferCount, *lpFlags, lpFrom,
5711           (lpFromlen ? *lpFromlen : -1),
5712           lpOverlapped, lpCompletionRoutine);
5713
5714     fd = get_sock_fd( s, FILE_READ_DATA, &options );
5715     TRACE( "fd=%d, options=%x\n", fd, options );
5716
5717     if (fd == -1) return SOCKET_ERROR;
5718
5719     if (!(wsa = HeapAlloc( GetProcessHeap(), 0, FIELD_OFFSET(struct ws2_async, iovec[dwBufferCount]) )))
5720     {
5721         err = WSAEFAULT;
5722         goto error;
5723     }
5724
5725     wsa->hSocket     = SOCKET2HANDLE(s);
5726     wsa->flags       = *lpFlags;
5727     wsa->lpFlags     = lpFlags;
5728     wsa->addr        = lpFrom;
5729     wsa->addrlen.ptr = lpFromlen;
5730     wsa->control     = lpControlBuffer;
5731     wsa->n_iovecs    = dwBufferCount;
5732     wsa->first_iovec = 0;
5733     for (i = 0; i < dwBufferCount; i++)
5734     {
5735         /* check buffer first to trigger write watches */
5736         if (IsBadWritePtr( lpBuffers[i].buf, lpBuffers[i].len ))
5737         {
5738             err = WSAEFAULT;
5739             goto error;
5740         }
5741         wsa->iovec[i].iov_base = lpBuffers[i].buf;
5742         wsa->iovec[i].iov_len  = lpBuffers[i].len;
5743     }
5744
5745     for (;;)
5746     {
5747         n = WS2_recv( fd, wsa );
5748         if (n == -1)
5749         {
5750             if (errno == EINTR) continue;
5751             if (errno != EAGAIN)
5752             {
5753                 int loc_errno = errno;
5754                 err = wsaErrno();
5755                 if (cvalue) WS_AddCompletion( s, cvalue, sock_get_ntstatus(loc_errno), 0 );
5756                 goto error;
5757             }
5758         }
5759         else if (lpNumberOfBytesRecvd) *lpNumberOfBytesRecvd = n;
5760
5761         if ((lpOverlapped || lpCompletionRoutine) &&
5762              !(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)))
5763         {
5764             IO_STATUS_BLOCK *iosb = lpOverlapped ? (IO_STATUS_BLOCK *)lpOverlapped : &wsa->local_iosb;
5765
5766             wsa->user_overlapped = lpOverlapped;
5767             wsa->completion_func = lpCompletionRoutine;
5768             release_sock_fd( s, fd );
5769
5770             if (n == -1)
5771             {
5772                 iosb->u.Status = STATUS_PENDING;
5773                 iosb->Information = 0;
5774
5775                 SERVER_START_REQ( register_async )
5776                 {
5777                     req->type           = ASYNC_TYPE_READ;
5778                     req->async.handle   = wine_server_obj_handle( wsa->hSocket );
5779                     req->async.callback = wine_server_client_ptr( WS2_async_recv );
5780                     req->async.iosb     = wine_server_client_ptr( iosb );
5781                     req->async.arg      = wine_server_client_ptr( wsa );
5782                     req->async.event    = wine_server_obj_handle( lpCompletionRoutine ? 0 : lpOverlapped->hEvent );
5783                     req->async.cvalue   = cvalue;
5784                     err = wine_server_call( req );
5785                 }
5786                 SERVER_END_REQ;
5787
5788                 if (err != STATUS_PENDING) HeapFree( GetProcessHeap(), 0, wsa );
5789                 WSASetLastError( NtStatusToWSAError( err ));
5790                 return SOCKET_ERROR;
5791             }
5792
5793             iosb->u.Status = STATUS_SUCCESS;
5794             iosb->Information = n;
5795             if (!wsa->completion_func)
5796             {
5797                 if (cvalue) WS_AddCompletion( s, cvalue, STATUS_SUCCESS, n );
5798                 if (lpOverlapped->hEvent) SetEvent( lpOverlapped->hEvent );
5799                 HeapFree( GetProcessHeap(), 0, wsa );
5800             }
5801             else NtQueueApcThread( GetCurrentThread(), (PNTAPCFUNC)ws2_async_apc,
5802                                    (ULONG_PTR)wsa, (ULONG_PTR)iosb, 0 );
5803             _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
5804             return 0;
5805         }
5806
5807         if (n != -1) break;
5808
5809         if ( _is_blocking(s) )
5810         {
5811             struct pollfd pfd;
5812             int timeout = GET_RCVTIMEO(fd);
5813             if (timeout != -1)
5814             {
5815                 timeout -= GetTickCount() - timeout_start;
5816                 if (timeout < 0) timeout = 0;
5817             }
5818
5819             pfd.fd = fd;
5820             pfd.events = POLLIN;
5821             if (*lpFlags & WS_MSG_OOB) pfd.events |= POLLPRI;
5822
5823             if (!timeout || !poll( &pfd, 1, timeout ))
5824             {
5825                 err = WSAETIMEDOUT;
5826                 /* a timeout is not fatal */
5827                 _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
5828                 goto error;
5829             }
5830         }
5831         else
5832         {
5833             _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
5834             err = WSAEWOULDBLOCK;
5835             goto error;
5836         }
5837     }
5838
5839     TRACE(" -> %i bytes\n", n);
5840     HeapFree( GetProcessHeap(), 0, wsa );
5841     release_sock_fd( s, fd );
5842     _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
5843
5844     return 0;
5845
5846 error:
5847     HeapFree( GetProcessHeap(), 0, wsa );
5848     release_sock_fd( s, fd );
5849     WARN(" -> ERROR %d\n", err);
5850     WSASetLastError( err );
5851     return SOCKET_ERROR;
5852 }
5853
5854 /***********************************************************************
5855  *              WSARecvFrom             (WS2_32.69)
5856  */
5857 INT WINAPI WSARecvFrom( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
5858                         LPDWORD lpNumberOfBytesRecvd, LPDWORD lpFlags, struct WS_sockaddr *lpFrom,
5859                         LPINT lpFromlen, LPWSAOVERLAPPED lpOverlapped,
5860                         LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
5861
5862 {
5863     return WS2_recv_base( s, lpBuffers, dwBufferCount,
5864                 lpNumberOfBytesRecvd, lpFlags,
5865                 lpFrom, lpFromlen,
5866                 lpOverlapped, lpCompletionRoutine, NULL );
5867 }
5868
5869 /***********************************************************************
5870  *              WSCInstallProvider             (WS2_32.88)
5871  */
5872 INT WINAPI WSCInstallProvider( const LPGUID lpProviderId,
5873                                LPCWSTR lpszProviderDllPath,
5874                                const LPWSAPROTOCOL_INFOW lpProtocolInfoList,
5875                                DWORD dwNumberOfEntries,
5876                                LPINT lpErrno )
5877 {
5878     FIXME("(%s, %s, %p, %d, %p): stub !\n", debugstr_guid(lpProviderId),
5879           debugstr_w(lpszProviderDllPath), lpProtocolInfoList,
5880           dwNumberOfEntries, lpErrno);
5881     *lpErrno = 0;
5882     return 0;
5883 }
5884
5885
5886 /***********************************************************************
5887  *              WSCDeinstallProvider             (WS2_32.83)
5888  */
5889 INT WINAPI WSCDeinstallProvider(LPGUID lpProviderId, LPINT lpErrno)
5890 {
5891     FIXME("(%s, %p): stub !\n", debugstr_guid(lpProviderId), lpErrno);
5892     *lpErrno = 0;
5893     return 0;
5894 }
5895
5896
5897 /***********************************************************************
5898  *              WSAAccept                        (WS2_32.26)
5899  */
5900 SOCKET WINAPI WSAAccept( SOCKET s, struct WS_sockaddr *addr, LPINT addrlen,
5901                LPCONDITIONPROC lpfnCondition, DWORD_PTR dwCallbackData)
5902 {
5903
5904        int ret = 0, size = 0;
5905        WSABUF CallerId, CallerData, CalleeId, CalleeData;
5906        /*        QOS SQOS, GQOS; */
5907        GROUP g;
5908        SOCKET cs;
5909        SOCKADDR src_addr, dst_addr;
5910
5911        TRACE("Socket %04lx, sockaddr %p, addrlen %p, fnCondition %p, dwCallbackData %ld\n",
5912                s, addr, addrlen, lpfnCondition, dwCallbackData);
5913
5914
5915        size = sizeof(src_addr);
5916        cs = WS_accept(s, &src_addr, &size);
5917
5918        if (cs == SOCKET_ERROR) return SOCKET_ERROR;
5919
5920        if (!lpfnCondition) return cs;
5921
5922        CallerId.buf = (char *)&src_addr;
5923        CallerId.len = sizeof(src_addr);
5924
5925        CallerData.buf = NULL;
5926        CallerData.len = 0;
5927
5928        WS_getsockname(cs, &dst_addr, &size);
5929
5930        CalleeId.buf = (char *)&dst_addr;
5931        CalleeId.len = sizeof(dst_addr);
5932
5933
5934        ret = (*lpfnCondition)(&CallerId, &CallerData, NULL, NULL,
5935                        &CalleeId, &CalleeData, &g, dwCallbackData);
5936
5937        switch (ret)
5938        {
5939                case CF_ACCEPT:
5940                        if (addr && addrlen)
5941                            memcpy(addr, &src_addr, (*addrlen > size) ?  size : *addrlen );
5942                        return cs;
5943                case CF_DEFER:
5944                        SERVER_START_REQ( set_socket_deferred )
5945                        {
5946                            req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
5947                            req->deferred = wine_server_obj_handle( SOCKET2HANDLE(cs) );
5948                            if ( !wine_server_call_err ( req ) )
5949                            {
5950                                SetLastError( WSATRY_AGAIN );
5951                                WS_closesocket( cs );
5952                            }
5953                        }
5954                        SERVER_END_REQ;
5955                        return SOCKET_ERROR;
5956                case CF_REJECT:
5957                        WS_closesocket(cs);
5958                        SetLastError(WSAECONNREFUSED);
5959                        return SOCKET_ERROR;
5960                default:
5961                        FIXME("Unknown return type from Condition function\n");
5962                        SetLastError(WSAENOTSOCK);
5963                        return SOCKET_ERROR;
5964        }
5965 }
5966
5967 /***********************************************************************
5968  *              WSADuplicateSocketA                      (WS2_32.32)
5969  */
5970 int WINAPI WSADuplicateSocketA( SOCKET s, DWORD dwProcessId, LPWSAPROTOCOL_INFOA lpProtocolInfo )
5971 {
5972    HANDLE hProcess;
5973
5974    TRACE("(%ld,%x,%p)\n", s, dwProcessId, lpProtocolInfo);
5975    memset(lpProtocolInfo, 0, sizeof(*lpProtocolInfo));
5976    /* FIXME: WS_getsockopt(s, WS_SOL_SOCKET, SO_PROTOCOL_INFO, lpProtocolInfo, sizeof(*lpProtocolInfo)); */
5977    /* I don't know what the real Windoze does next, this is a hack */
5978    /* ...we could duplicate and then use ConvertToGlobalHandle on the duplicate, then let
5979     * the target use the global duplicate, or we could copy a reference to us to the structure
5980     * and let the target duplicate it from us, but let's do it as simple as possible */
5981    hProcess = OpenProcess(PROCESS_DUP_HANDLE, FALSE, dwProcessId);
5982    DuplicateHandle(GetCurrentProcess(), SOCKET2HANDLE(s),
5983                    hProcess, (LPHANDLE)&lpProtocolInfo->dwCatalogEntryId,
5984                    0, FALSE, DUPLICATE_SAME_ACCESS);
5985    CloseHandle(hProcess);
5986    lpProtocolInfo->dwServiceFlags4 = 0xff00ff00; /* magic */
5987    return 0;
5988 }
5989
5990 /***********************************************************************
5991  *              WSADuplicateSocketW                      (WS2_32.33)
5992  */
5993 int WINAPI WSADuplicateSocketW( SOCKET s, DWORD dwProcessId, LPWSAPROTOCOL_INFOW lpProtocolInfo )
5994 {
5995    HANDLE hProcess;
5996
5997    TRACE("(%ld,%x,%p)\n", s, dwProcessId, lpProtocolInfo);
5998
5999    memset(lpProtocolInfo, 0, sizeof(*lpProtocolInfo));
6000    hProcess = OpenProcess(PROCESS_DUP_HANDLE, FALSE, dwProcessId);
6001    DuplicateHandle(GetCurrentProcess(), SOCKET2HANDLE(s),
6002                    hProcess, (LPHANDLE)&lpProtocolInfo->dwCatalogEntryId,
6003                    0, FALSE, DUPLICATE_SAME_ACCESS);
6004    CloseHandle(hProcess);
6005    lpProtocolInfo->dwServiceFlags4 = 0xff00ff00; /* magic */
6006    return 0;
6007 }
6008
6009 /***********************************************************************
6010  *              WSAInstallServiceClassA                  (WS2_32.48)
6011  */
6012 int WINAPI WSAInstallServiceClassA(LPWSASERVICECLASSINFOA info)
6013 {
6014     FIXME("Request to install service %s\n",debugstr_a(info->lpszServiceClassName));
6015     WSASetLastError(WSAEACCES);
6016     return SOCKET_ERROR;
6017 }
6018
6019 /***********************************************************************
6020  *              WSAInstallServiceClassW                  (WS2_32.49)
6021  */
6022 int WINAPI WSAInstallServiceClassW(LPWSASERVICECLASSINFOW info)
6023 {
6024     FIXME("Request to install service %s\n",debugstr_w(info->lpszServiceClassName));
6025     WSASetLastError(WSAEACCES);
6026     return SOCKET_ERROR;
6027 }
6028
6029 /***********************************************************************
6030  *              WSARemoveServiceClass                    (WS2_32.70)
6031  */
6032 int WINAPI WSARemoveServiceClass(LPGUID info)
6033 {
6034     FIXME("Request to remove service %p\n",info);
6035     WSASetLastError(WSATYPE_NOT_FOUND);
6036     return SOCKET_ERROR;
6037 }
6038
6039 /***********************************************************************
6040  *              inet_ntop                      (WS2_32.@)
6041  */
6042 PCSTR WINAPI WS_inet_ntop( INT family, PVOID addr, PSTR buffer, SIZE_T len )
6043 {
6044 #ifdef HAVE_INET_NTOP
6045     struct WS_in6_addr *in6;
6046     struct WS_in_addr  *in;
6047     PCSTR pdst;
6048
6049     TRACE("family %d, addr (%p), buffer (%p), len %ld\n", family, addr, buffer, len);
6050     if (!buffer)
6051     {
6052         WSASetLastError( STATUS_INVALID_PARAMETER );
6053         return NULL;
6054     }
6055
6056     switch (family)
6057     {
6058     case WS_AF_INET:
6059     {
6060         in = addr;
6061         pdst = inet_ntop( AF_INET, &in->WS_s_addr, buffer, len );
6062         break;
6063     }
6064     case WS_AF_INET6:
6065     {
6066         in6 = addr;
6067         pdst = inet_ntop( AF_INET6, in6->WS_s6_addr, buffer, len );
6068         break;
6069     }
6070     default:
6071         WSASetLastError( WSAEAFNOSUPPORT );
6072         return NULL;
6073     }
6074
6075     if (!pdst) WSASetLastError( STATUS_INVALID_PARAMETER );
6076     return pdst;
6077 #else
6078     FIXME( "not supported on this platform\n" );
6079     WSASetLastError( WSAEAFNOSUPPORT );
6080     return NULL;
6081 #endif
6082 }
6083
6084 /***********************************************************************
6085  *              WSAStringToAddressA                      (WS2_32.80)
6086  */
6087 INT WINAPI WSAStringToAddressA(LPSTR AddressString,
6088                                INT AddressFamily,
6089                                LPWSAPROTOCOL_INFOA lpProtocolInfo,
6090                                LPSOCKADDR lpAddress,
6091                                LPINT lpAddressLength)
6092 {
6093     INT res=0;
6094     LPSTR workBuffer=NULL,ptrPort;
6095
6096     TRACE( "(%s, %x, %p, %p, %p)\n", debugstr_a(AddressString), AddressFamily,
6097            lpProtocolInfo, lpAddress, lpAddressLength );
6098
6099     if (!lpAddressLength || !lpAddress) return SOCKET_ERROR;
6100
6101     if (!AddressString)
6102     {
6103         WSASetLastError(WSAEINVAL);
6104         return SOCKET_ERROR;
6105     }
6106
6107     if (lpProtocolInfo)
6108         FIXME("ProtocolInfo not implemented.\n");
6109
6110     workBuffer = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
6111                             strlen(AddressString) + 1);
6112     if (!workBuffer)
6113     {
6114         WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6115         return SOCKET_ERROR;
6116     }
6117
6118     strcpy(workBuffer, AddressString);
6119
6120     switch(AddressFamily)
6121     {
6122     case WS_AF_INET:
6123     {
6124         struct in_addr inetaddr;
6125
6126         /* If lpAddressLength is too small, tell caller the size we need */
6127         if (*lpAddressLength < sizeof(SOCKADDR_IN))
6128         {
6129             *lpAddressLength = sizeof(SOCKADDR_IN);
6130             res = WSAEFAULT;
6131             break;
6132         }
6133         *lpAddressLength = sizeof(SOCKADDR_IN);
6134         memset(lpAddress, 0, sizeof(SOCKADDR_IN));
6135
6136         ((LPSOCKADDR_IN)lpAddress)->sin_family = AF_INET;
6137
6138         ptrPort = strchr(workBuffer, ':');
6139         if(ptrPort)
6140         {
6141             ((LPSOCKADDR_IN)lpAddress)->sin_port = htons(atoi(ptrPort+1));
6142             *ptrPort = '\0';
6143         }
6144         else
6145         {
6146             ((LPSOCKADDR_IN)lpAddress)->sin_port = 0;
6147         }
6148
6149         if(inet_aton(workBuffer, &inetaddr) > 0)
6150         {
6151             ((LPSOCKADDR_IN)lpAddress)->sin_addr.WS_s_addr = inetaddr.s_addr;
6152             res = 0;
6153         }
6154         else
6155             res = WSAEINVAL;
6156
6157         break;
6158
6159     }
6160     case WS_AF_INET6:
6161     {
6162         struct in6_addr inetaddr;
6163         /* If lpAddressLength is too small, tell caller the size we need */
6164         if (*lpAddressLength < sizeof(SOCKADDR_IN6))
6165         {
6166             *lpAddressLength = sizeof(SOCKADDR_IN6);
6167             res = WSAEFAULT;
6168             break;
6169         }
6170 #ifdef HAVE_INET_PTON
6171         *lpAddressLength = sizeof(SOCKADDR_IN6);
6172         memset(lpAddress, 0, sizeof(SOCKADDR_IN6));
6173
6174         ((LPSOCKADDR_IN6)lpAddress)->sin6_family = WS_AF_INET6;
6175
6176         /* This one is a bit tricky. An IPv6 address contains colons, so the
6177          * check from IPv4 doesn't work like that. However, IPv6 addresses that
6178          * contain a port are written with braces like [fd12:3456:7890::1]:12345
6179          * so what we will do is to look for ']', check if the next char is a
6180          * colon, and if it is, parse the port as in IPv4. */
6181
6182         ptrPort = strchr(workBuffer, ']');
6183         if(ptrPort && *(++ptrPort) == ':')
6184         {
6185             ((LPSOCKADDR_IN6)lpAddress)->sin6_port = htons(atoi(ptrPort+1));
6186             *ptrPort = '\0';
6187         }
6188         else
6189         {
6190             ((LPSOCKADDR_IN6)lpAddress)->sin6_port = 0;
6191         }
6192
6193         if(inet_pton(AF_INET6, workBuffer, &inetaddr) > 0)
6194         {
6195             memcpy(&((LPSOCKADDR_IN6)lpAddress)->sin6_addr, &inetaddr,
6196                     sizeof(struct in6_addr));
6197             res = 0;
6198         }
6199         else
6200 #endif /* HAVE_INET_PTON */
6201             res = WSAEINVAL;
6202
6203         break;
6204     }
6205     default:
6206         /* According to MSDN, only AF_INET and AF_INET6 are supported. */
6207         TRACE("Unsupported address family specified: %d.\n", AddressFamily);
6208         res = WSAEINVAL;
6209     }
6210
6211     HeapFree(GetProcessHeap(), 0, workBuffer);
6212
6213     if (!res) return 0;
6214     WSASetLastError(res);
6215     return SOCKET_ERROR;
6216 }
6217
6218 /***********************************************************************
6219  *              WSAStringToAddressW                      (WS2_32.81)
6220  *
6221  * Does anybody know if this function allows to use Hebrew/Arabic/Chinese... digits?
6222  * If this should be the case, it would be required to map these digits
6223  * to Unicode digits (0-9) using FoldString first.
6224  */
6225 INT WINAPI WSAStringToAddressW(LPWSTR AddressString,
6226                                INT AddressFamily,
6227                                LPWSAPROTOCOL_INFOW lpProtocolInfo,
6228                                LPSOCKADDR lpAddress,
6229                                LPINT lpAddressLength)
6230 {
6231     INT sBuffer,res=0;
6232     LPSTR workBuffer=NULL;
6233     WSAPROTOCOL_INFOA infoA;
6234     LPWSAPROTOCOL_INFOA lpProtoInfoA = NULL;
6235
6236     TRACE( "(%s, %x, %p, %p, %p)\n", debugstr_w(AddressString), AddressFamily, lpProtocolInfo,
6237            lpAddress, lpAddressLength );
6238
6239     if (!lpAddressLength || !lpAddress) return SOCKET_ERROR;
6240
6241     /* if ProtocolInfo is available - convert to ANSI variant */
6242     if (lpProtocolInfo)
6243     {
6244         lpProtoInfoA = &infoA;
6245         memcpy( lpProtoInfoA, lpProtocolInfo, FIELD_OFFSET( WSAPROTOCOL_INFOA, szProtocol ) );
6246
6247         if (!WideCharToMultiByte( CP_ACP, 0, lpProtocolInfo->szProtocol, -1,
6248                                   lpProtoInfoA->szProtocol, WSAPROTOCOL_LEN+1, NULL, NULL ))
6249         {
6250             WSASetLastError( WSAEINVAL);
6251             return SOCKET_ERROR;
6252         }
6253     }
6254
6255     if (AddressString)
6256     {
6257         /* Translate AddressString to ANSI code page - assumes that only
6258            standard digits 0-9 are used with this API call */
6259         sBuffer = WideCharToMultiByte( CP_ACP, 0, AddressString, -1, NULL, 0, NULL, NULL );
6260         workBuffer = HeapAlloc( GetProcessHeap(), 0, sBuffer );
6261
6262         if (workBuffer)
6263         {
6264             WideCharToMultiByte( CP_ACP, 0, AddressString, -1, workBuffer, sBuffer, NULL, NULL );
6265             res = WSAStringToAddressA(workBuffer,AddressFamily,lpProtoInfoA,
6266                                       lpAddress,lpAddressLength);
6267             HeapFree( GetProcessHeap(), 0, workBuffer );
6268             return res;
6269         }
6270         else
6271             res = WSA_NOT_ENOUGH_MEMORY;
6272     }
6273     else
6274         res = WSAEINVAL;
6275
6276     WSASetLastError(res);
6277     return SOCKET_ERROR;
6278 }
6279
6280 /***********************************************************************
6281  *              WSAAddressToStringA                      (WS2_32.27)
6282  *
6283  *  See WSAAddressToStringW
6284  */
6285 INT WINAPI WSAAddressToStringA( LPSOCKADDR sockaddr, DWORD len,
6286                                 LPWSAPROTOCOL_INFOA info, LPSTR string,
6287                                 LPDWORD lenstr )
6288 {
6289     DWORD size;
6290     CHAR buffer[54]; /* 32 digits + 7':' + '[' + '%" + 5 digits + ']:' + 5 digits + '\0' */
6291     CHAR *p;
6292
6293     TRACE( "(%p, %d, %p, %p, %p)\n", sockaddr, len, info, string, lenstr );
6294
6295     if (!sockaddr) return SOCKET_ERROR;
6296     if (!string || !lenstr) return SOCKET_ERROR;
6297
6298     switch(sockaddr->sa_family)
6299     {
6300     case WS_AF_INET:
6301         if (len < sizeof(SOCKADDR_IN)) return SOCKET_ERROR;
6302         sprintf( buffer, "%u.%u.%u.%u:%u",
6303                (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) >> 24 & 0xff),
6304                (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) >> 16 & 0xff),
6305                (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) >> 8 & 0xff),
6306                (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) & 0xff),
6307                ntohs( ((SOCKADDR_IN *)sockaddr)->sin_port ) );
6308
6309         p = strchr( buffer, ':' );
6310         if (!((SOCKADDR_IN *)sockaddr)->sin_port) *p = 0;
6311         break;
6312
6313     case WS_AF_INET6:
6314     {
6315         struct WS_sockaddr_in6 *sockaddr6 = (LPSOCKADDR_IN6) sockaddr;
6316
6317         buffer[0] = 0;
6318         if (len < sizeof(SOCKADDR_IN6)) return SOCKET_ERROR;
6319         if ((sockaddr6->sin6_port))
6320             strcpy(buffer, "[");
6321         if (!WS_inet_ntop(WS_AF_INET6, &sockaddr6->sin6_addr, buffer+strlen(buffer), sizeof(buffer)))
6322         {
6323             WSASetLastError(WSAEINVAL);
6324             return SOCKET_ERROR;
6325         }
6326         if ((sockaddr6->sin6_scope_id))
6327             sprintf(buffer+strlen(buffer), "%%%u", sockaddr6->sin6_scope_id);
6328         if ((sockaddr6->sin6_port))
6329             sprintf(buffer+strlen(buffer), "]:%u", ntohs(sockaddr6->sin6_port));
6330         break;
6331     }
6332
6333     default:
6334         WSASetLastError(WSAEINVAL);
6335         return SOCKET_ERROR;
6336     }
6337
6338     size = strlen( buffer ) + 1;
6339
6340     if (*lenstr <  size)
6341     {
6342         *lenstr = size;
6343         WSASetLastError(WSAEFAULT);
6344         return SOCKET_ERROR;
6345     }
6346
6347     *lenstr = size;
6348     strcpy( string, buffer );
6349     return 0;
6350 }
6351
6352 /***********************************************************************
6353  *              WSAAddressToStringW                      (WS2_32.28)
6354  *
6355  * Convert a sockaddr address into a readable address string. 
6356  *
6357  * PARAMS
6358  *  sockaddr [I]    Pointer to a sockaddr structure.
6359  *  len      [I]    Size of the sockaddr structure.
6360  *  info     [I]    Pointer to a WSAPROTOCOL_INFOW structure (optional).
6361  *  string   [I/O]  Pointer to a buffer to receive the address string.
6362  *  lenstr   [I/O]  Size of the receive buffer in WCHARs.
6363  *
6364  * RETURNS
6365  *  Success: 0
6366  *  Failure: SOCKET_ERROR
6367  *
6368  * NOTES
6369  *  The 'info' parameter is ignored.
6370  */
6371 INT WINAPI WSAAddressToStringW( LPSOCKADDR sockaddr, DWORD len,
6372                                 LPWSAPROTOCOL_INFOW info, LPWSTR string,
6373                                 LPDWORD lenstr )
6374 {
6375     INT   ret;
6376     DWORD size;
6377     WCHAR buffer[54]; /* 32 digits + 7':' + '[' + '%" + 5 digits + ']:' + 5 digits + '\0' */
6378     CHAR bufAddr[54];
6379
6380     TRACE( "(%p, %d, %p, %p, %p)\n", sockaddr, len, info, string, lenstr );
6381
6382     size = *lenstr;
6383     ret = WSAAddressToStringA(sockaddr, len, NULL, bufAddr, &size);
6384
6385     if (ret) return ret;
6386
6387     MultiByteToWideChar( CP_ACP, 0, bufAddr, size, buffer, sizeof( buffer )/sizeof(WCHAR));
6388
6389     if (*lenstr <  size)
6390     {
6391         *lenstr = size;
6392         WSASetLastError(WSAEFAULT);
6393         return SOCKET_ERROR;
6394     }
6395
6396     *lenstr = size;
6397     lstrcpyW( string, buffer );
6398     return 0;
6399 }
6400
6401 /***********************************************************************
6402  *              WSAEnumNameSpaceProvidersA                  (WS2_32.34)
6403  */
6404 INT WINAPI WSAEnumNameSpaceProvidersA( LPDWORD len, LPWSANAMESPACE_INFOA buffer )
6405 {
6406     FIXME( "(%p %p) Stub!\n", len, buffer );
6407     return 0;
6408 }
6409
6410 /***********************************************************************
6411  *              WSAEnumNameSpaceProvidersW                  (WS2_32.35)
6412  */
6413 INT WINAPI WSAEnumNameSpaceProvidersW( LPDWORD len, LPWSANAMESPACE_INFOW buffer )
6414 {
6415     FIXME( "(%p %p) Stub!\n", len, buffer );
6416     return 0;
6417 }
6418
6419 /***********************************************************************
6420  *              WSAGetQOSByName                             (WS2_32.41)
6421  */
6422 BOOL WINAPI WSAGetQOSByName( SOCKET s, LPWSABUF lpQOSName, LPQOS lpQOS )
6423 {
6424     FIXME( "(0x%04lx %p %p) Stub!\n", s, lpQOSName, lpQOS );
6425     return FALSE;
6426 }
6427
6428 /***********************************************************************
6429  *              WSAGetServiceClassInfoA                     (WS2_32.42)
6430  */
6431 INT WINAPI WSAGetServiceClassInfoA( LPGUID provider, LPGUID service, LPDWORD len,
6432                                     LPWSASERVICECLASSINFOA info )
6433 {
6434     FIXME( "(%s %s %p %p) Stub!\n", debugstr_guid(provider), debugstr_guid(service),
6435            len, info );
6436     WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6437     return SOCKET_ERROR; 
6438 }
6439
6440 /***********************************************************************
6441  *              WSAGetServiceClassInfoW                     (WS2_32.43)
6442  */
6443 INT WINAPI WSAGetServiceClassInfoW( LPGUID provider, LPGUID service, LPDWORD len,
6444                                     LPWSASERVICECLASSINFOW info )
6445 {
6446     FIXME( "(%s %s %p %p) Stub!\n", debugstr_guid(provider), debugstr_guid(service),
6447            len, info );
6448     WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6449     return SOCKET_ERROR;
6450 }
6451
6452 /***********************************************************************
6453  *              WSAGetServiceClassNameByClassIdA            (WS2_32.44)
6454  */
6455 INT WINAPI WSAGetServiceClassNameByClassIdA( LPGUID class, LPSTR service, LPDWORD len )
6456 {
6457     FIXME( "(%s %p %p) Stub!\n", debugstr_guid(class), service, len );
6458     WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6459     return SOCKET_ERROR;
6460 }
6461
6462 /***********************************************************************
6463  *              WSAGetServiceClassNameByClassIdW            (WS2_32.45)
6464  */
6465 INT WINAPI WSAGetServiceClassNameByClassIdW( LPGUID class, LPWSTR service, LPDWORD len )
6466 {
6467     FIXME( "(%s %p %p) Stub!\n", debugstr_guid(class), service, len );
6468     WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6469     return SOCKET_ERROR;
6470 }
6471
6472 /***********************************************************************
6473  *              WSALookupServiceBeginA                       (WS2_32.59)
6474  */
6475 INT WINAPI WSALookupServiceBeginA( LPWSAQUERYSETA lpqsRestrictions,
6476                                    DWORD dwControlFlags,
6477                                    LPHANDLE lphLookup)
6478 {
6479     FIXME("(%p 0x%08x %p) Stub!\n", lpqsRestrictions, dwControlFlags,
6480             lphLookup);
6481     WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6482     return SOCKET_ERROR;
6483 }
6484
6485 /***********************************************************************
6486  *              WSALookupServiceBeginW                       (WS2_32.60)
6487  */
6488 INT WINAPI WSALookupServiceBeginW( LPWSAQUERYSETW lpqsRestrictions,
6489                                    DWORD dwControlFlags,
6490                                    LPHANDLE lphLookup)
6491 {
6492     FIXME("(%p 0x%08x %p) Stub!\n", lpqsRestrictions, dwControlFlags,
6493             lphLookup);
6494     WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6495     return SOCKET_ERROR;
6496 }
6497
6498 /***********************************************************************
6499  *              WSALookupServiceBeginW                       (WS2_32.61)
6500  */
6501 INT WINAPI WSALookupServiceEnd( HANDLE lookup )
6502 {
6503     FIXME("(%p) Stub!\n", lookup );
6504     return 0;
6505 }
6506
6507 /***********************************************************************
6508  *              WSALookupServiceNextA                       (WS2_32.62)
6509  */
6510 INT WINAPI WSALookupServiceNextA( HANDLE lookup, DWORD flags, LPDWORD len, LPWSAQUERYSETA results )
6511 {
6512     FIXME( "(%p 0x%08x %p %p) Stub!\n", lookup, flags, len, results );
6513     WSASetLastError(WSA_E_NO_MORE);
6514     return SOCKET_ERROR;
6515 }
6516
6517 /***********************************************************************
6518  *              WSALookupServiceNextW                       (WS2_32.63)
6519  */
6520 INT WINAPI WSALookupServiceNextW( HANDLE lookup, DWORD flags, LPDWORD len, LPWSAQUERYSETW results )
6521 {
6522     FIXME( "(%p 0x%08x %p %p) Stub!\n", lookup, flags, len, results );
6523     WSASetLastError(WSA_E_NO_MORE);
6524     return SOCKET_ERROR;
6525 }
6526
6527 /***********************************************************************
6528  *              WSANtohl                                   (WS2_32.64)
6529  */
6530 INT WINAPI WSANtohl( SOCKET s, WS_u_long netlong, WS_u_long* lphostlong )
6531 {
6532     TRACE( "(0x%04lx 0x%08x %p)\n", s, netlong, lphostlong );
6533
6534     if (!lphostlong) return WSAEFAULT;
6535
6536     *lphostlong = ntohl( netlong );
6537     return 0;
6538 }
6539
6540 /***********************************************************************
6541  *              WSANtohs                                   (WS2_32.65)
6542  */
6543 INT WINAPI WSANtohs( SOCKET s, WS_u_short netshort, WS_u_short* lphostshort )
6544 {
6545     TRACE( "(0x%04lx 0x%08x %p)\n", s, netshort, lphostshort );
6546
6547     if (!lphostshort) return WSAEFAULT;
6548
6549     *lphostshort = ntohs( netshort );
6550     return 0;
6551 }
6552
6553 /***********************************************************************
6554  *              WSAProviderConfigChange                     (WS2_32.66)
6555  */
6556 INT WINAPI WSAProviderConfigChange( LPHANDLE handle, LPWSAOVERLAPPED overlapped,
6557                                     LPWSAOVERLAPPED_COMPLETION_ROUTINE completion )
6558 {
6559     FIXME( "(%p %p %p) Stub!\n", handle, overlapped, completion );
6560     return SOCKET_ERROR;
6561 }
6562
6563 /***********************************************************************
6564  *              WSARecvDisconnect                           (WS2_32.68)
6565  */
6566 INT WINAPI WSARecvDisconnect( SOCKET s, LPWSABUF disconnectdata )
6567 {
6568     TRACE( "(0x%04lx %p)\n", s, disconnectdata );
6569
6570     return WS_shutdown( s, 0 );
6571 }
6572
6573 /***********************************************************************
6574  *              WSASetServiceA                              (WS2_32.76)
6575  */
6576 INT WINAPI WSASetServiceA( LPWSAQUERYSETA query, WSAESETSERVICEOP operation, DWORD flags )
6577 {
6578     FIXME( "(%p 0x%08x 0x%08x) Stub!\n", query, operation, flags );
6579     return 0;
6580 }
6581
6582 /***********************************************************************
6583  *              WSASetServiceW                              (WS2_32.77)
6584  */
6585 INT WINAPI WSASetServiceW( LPWSAQUERYSETW query, WSAESETSERVICEOP operation, DWORD flags )
6586 {
6587     FIXME( "(%p 0x%08x 0x%08x) Stub!\n", query, operation, flags );
6588     return 0;
6589 }
6590
6591 /***********************************************************************
6592  *              WSCEnableNSProvider                         (WS2_32.84)
6593  */
6594 INT WINAPI WSCEnableNSProvider( LPGUID provider, BOOL enable )
6595 {
6596     FIXME( "(%s 0x%08x) Stub!\n", debugstr_guid(provider), enable );
6597     return 0;
6598 }
6599
6600 /***********************************************************************
6601  *              WSCGetProviderPath                          (WS2_32.86)
6602  */
6603 INT WINAPI WSCGetProviderPath( LPGUID provider, LPWSTR path, LPINT len, LPINT errcode )
6604 {
6605     FIXME( "(%s %p %p %p) Stub!\n", debugstr_guid(provider), path, len, errcode );
6606
6607     if (!errcode || !provider || !len) return WSAEFAULT;
6608
6609     *errcode = WSAEINVAL;
6610     return SOCKET_ERROR;
6611 }
6612
6613 /***********************************************************************
6614  *              WSCInstallNameSpace                         (WS2_32.87)
6615  */
6616 INT WINAPI WSCInstallNameSpace( LPWSTR identifier, LPWSTR path, DWORD namespace,
6617                                 DWORD version, LPGUID provider )
6618 {
6619     FIXME( "(%s %s 0x%08x 0x%08x %s) Stub!\n", debugstr_w(identifier), debugstr_w(path),
6620            namespace, version, debugstr_guid(provider) );
6621     return 0;
6622 }
6623
6624 /***********************************************************************
6625  *              WSCUnInstallNameSpace                       (WS2_32.89)
6626  */
6627 INT WINAPI WSCUnInstallNameSpace( LPGUID lpProviderId )
6628 {
6629     FIXME("(%p) Stub!\n", lpProviderId);
6630     return NO_ERROR;
6631 }
6632
6633 /***********************************************************************
6634  *              WSCWriteProviderOrder                       (WS2_32.91)
6635  */
6636 INT WINAPI WSCWriteProviderOrder( LPDWORD entry, DWORD number )
6637 {
6638     FIXME("(%p 0x%08x) Stub!\n", entry, number);
6639     return 0;
6640 }
6641
6642 /***********************************************************************
6643  *              WSANSPIoctl                       (WS2_32.91)
6644  */
6645 INT WINAPI WSANSPIoctl( HANDLE hLookup, DWORD dwControlCode, LPVOID lpvInBuffer,
6646                         DWORD cbInBuffer, LPVOID lpvOutBuffer, DWORD cbOutBuffer,
6647                         LPDWORD lpcbBytesReturned, LPWSACOMPLETION lpCompletion )
6648 {
6649     FIXME("(%p, 0x%08x, %p, 0x%08x, %p, 0x%08x, %p, %p) Stub!\n", hLookup, dwControlCode,
6650     lpvInBuffer, cbInBuffer, lpvOutBuffer, cbOutBuffer, lpcbBytesReturned, lpCompletion);
6651     WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6652     return SOCKET_ERROR;
6653 };