git.oblomov.eu Git - wine/blob - dlls/ws2_32/socket.c

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