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