?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
ws2_32: Fix UDP LINGER support in getsockopt.
[wine/multimedia.git] / dlls / ws2_32 / socket.c
blob7a1947f68f5bd2f5e46d26a07b456ae301ad4ff3
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)
2050 {
2051 HeapFree( GetProcessHeap(), 0, wsa->read );
2052 HeapFree( GetProcessHeap(), 0, wsa );
2053 }
2054
2055 SetLastError( NtStatusToWSAError(status) );
2056 return FALSE;
2057 }
2058
2059 /***********************************************************************
2060 * GetAcceptExSockaddrs
2061 */
2062 static void WINAPI WS2_GetAcceptExSockaddrs(PVOID buffer, DWORD data_size, DWORD local_size, DWORD remote_size,
2063 struct WS_sockaddr **local_addr, LPINT local_addr_len,
2064 struct WS_sockaddr **remote_addr, LPINT remote_addr_len)
2065 {
2066 char *cbuf = buffer;
2067 TRACE("(%p, %d, %d, %d, %p, %p, %p, %p)\n", buffer, data_size, local_size, remote_size, local_addr,
2068 local_addr_len, remote_addr, remote_addr_len );
2069 cbuf += data_size;
2070
2071 *local_addr_len = *(int *) cbuf;
2072 *local_addr = (struct WS_sockaddr *)(cbuf + sizeof(int));
2073
2074 cbuf += local_size;
2075
2076 *remote_addr_len = *(int *) cbuf;
2077 *remote_addr = (struct WS_sockaddr *)(cbuf + sizeof(int));
2078 }
2079
2080 /***********************************************************************
2081 * WSARecvMsg
2082 *
2083 * Perform a receive operation that is capable of returning message
2084 * control headers. It is important to note that the WSAMSG parameter
2085 * must remain valid throughout the operation, even when an overlapped
2086 * receive is performed.
2087 */
2088 static int WINAPI WS2_WSARecvMsg( SOCKET s, LPWSAMSG msg, LPDWORD lpNumberOfBytesRecvd,
2089 LPWSAOVERLAPPED lpOverlapped,
2090 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
2091 {
2092 if (!msg)
2093 {
2094 SetLastError( WSAEFAULT );
2095 return SOCKET_ERROR;
2096 }
2097
2098 return WS2_recv_base( s, msg->lpBuffers, msg->dwBufferCount, lpNumberOfBytesRecvd,
2099 &msg->dwFlags, msg->name, &msg->namelen,
2100 lpOverlapped, lpCompletionRoutine, &msg->Control );
2101 }
2102
2103 /***********************************************************************
2104 * bind (WS2_32.2)
2105 */
2106 int WINAPI WS_bind(SOCKET s, const struct WS_sockaddr* name, int namelen)
2107 {
2108 int fd = get_sock_fd( s, 0, NULL );
2109 int res = SOCKET_ERROR;
2110
2111 TRACE("socket %04lx, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen);
2112
2113 if (fd != -1)
2114 {
2115 if (!name || (name->sa_family && !supported_pf(name->sa_family)))
2116 {
2117 SetLastError(WSAEAFNOSUPPORT);
2118 }
2119 else
2120 {
2121 union generic_unix_sockaddr uaddr;
2122 unsigned int uaddrlen = ws_sockaddr_ws2u(name, namelen, &uaddr);
2123 if (!uaddrlen)
2124 {
2125 SetLastError(WSAEFAULT);
2126 }
2127 else
2128 {
2129 #ifdef IPV6_V6ONLY
2130 const struct sockaddr_in6 *in6 = (const struct sockaddr_in6*) &uaddr;
2131 if (name->sa_family == WS_AF_INET6 &&
2132 !memcmp(&in6->sin6_addr, &in6addr_any, sizeof(struct in6_addr)))
2133 {
2134 int enable = 1;
2135 if (setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, &enable, sizeof(enable)) == -1)
2136 {
2137 release_sock_fd( s, fd );
2138 SetLastError(WSAEAFNOSUPPORT);
2139 return SOCKET_ERROR;
2140 }
2141 }
2142 #endif
2143 if (name->sa_family == WS_AF_INET)
2144 {
2145 struct sockaddr_in *in4 = (struct sockaddr_in*) &uaddr;
2146 if (memcmp(&in4->sin_addr, magic_loopback_addr, 4) == 0)
2147 {
2148 /* Trying to bind to the default host interface, using
2149 * INADDR_ANY instead*/
2150 WARN("Trying to bind to magic IP address, using "
2151 "INADDR_ANY instead.\n");
2152 in4->sin_addr.s_addr = htonl(WS_INADDR_ANY);
2153 }
2154 }
2155 if (bind(fd, &uaddr.addr, uaddrlen) < 0)
2156 {
2157 int loc_errno = errno;
2158 WARN("\tfailure - errno = %i\n", errno);
2159 errno = loc_errno;
2160 switch (errno)
2161 {
2162 case EBADF:
2163 SetLastError(WSAENOTSOCK);
2164 break;
2165 case EADDRNOTAVAIL:
2166 SetLastError(WSAEINVAL);
2167 break;
2168 default:
2169 SetLastError(wsaErrno());
2170 break;
2171 }
2172 }
2173 else
2174 {
2175 res=0; /* success */
2176 }
2177 }
2178 }
2179 release_sock_fd( s, fd );
2180 }
2181 return res;
2182 }
2183
2184 /***********************************************************************
2185 * closesocket (WS2_32.3)
2186 */
2187 int WINAPI WS_closesocket(SOCKET s)
2188 {
2189 TRACE("socket %04lx\n", s);
2190 if (CloseHandle(SOCKET2HANDLE(s))) return 0;
2191 return SOCKET_ERROR;
2192 }
2193
2194 static int do_connect(int fd, const struct WS_sockaddr* name, int namelen)
2195 {
2196 union generic_unix_sockaddr uaddr;
2197 unsigned int uaddrlen = ws_sockaddr_ws2u(name, namelen, &uaddr);
2198
2199 if (!uaddrlen)
2200 return WSAEFAULT;
2201
2202 if (name->sa_family == WS_AF_INET)
2203 {
2204 struct sockaddr_in *in4 = (struct sockaddr_in*) &uaddr;
2205 if (memcmp(&in4->sin_addr, magic_loopback_addr, 4) == 0)
2206 {
2207 /* Trying to connect to magic replace-loopback address,
2208 * assuming we really want to connect to localhost */
2209 TRACE("Trying to connect to magic IP address, using "
2210 "INADDR_LOOPBACK instead.\n");
2211 in4->sin_addr.s_addr = htonl(WS_INADDR_LOOPBACK);
2212 }
2213 }
2214
2215 if (connect(fd, &uaddr.addr, uaddrlen) == 0)
2216 return 0;
2217
2218 return wsaErrno();
2219 }
2220
2221 /***********************************************************************
2222 * connect (WS2_32.4)
2223 */
2224 int WINAPI WS_connect(SOCKET s, const struct WS_sockaddr* name, int namelen)
2225 {
2226 int fd = get_sock_fd( s, FILE_READ_DATA, NULL );
2227
2228 TRACE("socket %04lx, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen);
2229
2230 if (fd != -1)
2231 {
2232 int ret = do_connect(fd, name, namelen);
2233 if (ret == 0)
2234 goto connect_success;
2235
2236 if (ret == WSAEINPROGRESS)
2237 {
2238 /* tell wineserver that a connection is in progress */
2239 _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
2240 FD_CONNECT,
2241 FD_WINE_CONNECTED|FD_WINE_LISTENING);
2242 if (_is_blocking(s))
2243 {
2244 int result;
2245 /* block here */
2246 do_block(fd, POLLIN | POLLOUT, -1);
2247 _sync_sock_state(s); /* let wineserver notice connection */
2248 /* retrieve any error codes from it */
2249 result = _get_sock_error(s, FD_CONNECT_BIT);
2250 if (result)
2251 SetLastError(NtStatusToWSAError(result));
2252 else
2253 {
2254 goto connect_success;
2255 }
2256 }
2257 else
2258 {
2259 SetLastError(WSAEWOULDBLOCK);
2260 }
2261 }
2262 else
2263 {
2264 SetLastError(ret);
2265 }
2266 release_sock_fd( s, fd );
2267 }
2268 return SOCKET_ERROR;
2269
2270 connect_success:
2271 release_sock_fd( s, fd );
2272 _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
2273 FD_WINE_CONNECTED|FD_READ|FD_WRITE,
2274 FD_CONNECT|FD_WINE_LISTENING);
2275 return 0;
2276 }
2277
2278 /***********************************************************************
2279 * WSAConnect (WS2_32.30)
2280 */
2281 int WINAPI WSAConnect( SOCKET s, const struct WS_sockaddr* name, int namelen,
2282 LPWSABUF lpCallerData, LPWSABUF lpCalleeData,
2283 LPQOS lpSQOS, LPQOS lpGQOS )
2284 {
2285 if ( lpCallerData || lpCalleeData || lpSQOS || lpGQOS )
2286 FIXME("unsupported parameters!\n");
2287 return WS_connect( s, name, namelen );
2288 }
2289
2290 /***********************************************************************
2291 * ConnectEx
2292 */
2293 static BOOL WINAPI WS2_ConnectEx(SOCKET s, const struct WS_sockaddr* name, int namelen,
2294 PVOID sendBuf, DWORD sendBufLen, LPDWORD sent, LPOVERLAPPED ov)
2295 {
2296 int fd, ret, status;
2297
2298 if (!ov)
2299 {
2300 SetLastError( ERROR_INVALID_PARAMETER );
2301 return FALSE;
2302 }
2303
2304 fd = get_sock_fd( s, FILE_READ_DATA, NULL );
2305 if (fd == -1)
2306 {
2307 SetLastError( WSAENOTSOCK );
2308 return FALSE;
2309 }
2310
2311 TRACE("socket %04lx, ptr %p %s, length %d, sendptr %p, len %d, ov %p\n",
2312 s, name, debugstr_sockaddr(name), namelen, sendBuf, sendBufLen, ov);
2313
2314 /* FIXME: technically the socket has to be bound */
2315 ret = do_connect(fd, name, namelen);
2316 if (ret == 0)
2317 {
2318 WSABUF wsabuf;
2319
2320 _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
2321 FD_WINE_CONNECTED|FD_READ|FD_WRITE,
2322 FD_CONNECT|FD_WINE_LISTENING);
2323
2324 wsabuf.len = sendBufLen;
2325 wsabuf.buf = (char*) sendBuf;
2326
2327 /* WSASend takes care of completion if need be */
2328 if (WSASend(s, &wsabuf, sendBuf ? 1 : 0, sent, 0, ov, NULL) != SOCKET_ERROR)
2329 goto connection_success;
2330 }
2331 else if (ret == WSAEINPROGRESS)
2332 {
2333 struct ws2_async *wsa;
2334 ULONG_PTR cvalue = (((ULONG_PTR)ov->hEvent & 1) == 0) ? (ULONG_PTR)ov : 0;
2335
2336 _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
2337 FD_CONNECT,
2338 FD_WINE_CONNECTED|FD_WINE_LISTENING);
2339
2340 /* Indirectly call WSASend */
2341 if (!(wsa = HeapAlloc( GetProcessHeap(), 0, sizeof(*wsa) )))
2342 {
2343 SetLastError(WSAEFAULT);
2344 }
2345 else
2346 {
2347 IO_STATUS_BLOCK *iosb = (IO_STATUS_BLOCK *)ov;
2348 iosb->u.Status = STATUS_PENDING;
2349 iosb->Information = 0;
2350
2351 wsa->hSocket = SOCKET2HANDLE(s);
2352 wsa->addr = NULL;
2353 wsa->addrlen.val = 0;
2354 wsa->flags = 0;
2355 wsa->lpFlags = &wsa->flags;
2356 wsa->control = NULL;
2357 wsa->n_iovecs = sendBuf ? 1 : 0;
2358 wsa->first_iovec = 0;
2359 wsa->completion_func = NULL;
2360 wsa->iovec[0].iov_base = sendBuf;
2361 wsa->iovec[0].iov_len = sendBufLen;
2362
2363 SERVER_START_REQ( register_async )
2364 {
2365 req->type = ASYNC_TYPE_WRITE;
2366 req->async.handle = wine_server_obj_handle( wsa->hSocket );
2367 req->async.callback = wine_server_client_ptr( WS2_async_send );
2368 req->async.iosb = wine_server_client_ptr( iosb );
2369 req->async.arg = wine_server_client_ptr( wsa );
2370 req->async.event = wine_server_obj_handle( ov->hEvent );
2371 req->async.cvalue = cvalue;
2372 status = wine_server_call( req );
2373 }
2374 SERVER_END_REQ;
2375
2376 if (status != STATUS_PENDING) HeapFree(GetProcessHeap(), 0, wsa);
2377
2378 /* If the connect already failed */
2379 if (status == STATUS_PIPE_DISCONNECTED)
2380 status = _get_sock_error(s, FD_CONNECT_BIT);
2381 SetLastError( NtStatusToWSAError(status) );
2382 }
2383 }
2384 else
2385 {
2386 SetLastError(ret);
2387 }
2388
2389 release_sock_fd( s, fd );
2390 return FALSE;
2391
2392 connection_success:
2393 release_sock_fd( s, fd );
2394 return TRUE;
2395 }
2396
2397
2398 /***********************************************************************
2399 * getpeername (WS2_32.5)
2400 */
2401 int WINAPI WS_getpeername(SOCKET s, struct WS_sockaddr *name, int *namelen)
2402 {
2403 int fd;
2404 int res;
2405
2406 TRACE("socket: %04lx, ptr %p, len %08x\n", s, name, namelen?*namelen:0);
2407
2408 fd = get_sock_fd( s, 0, NULL );
2409 res = SOCKET_ERROR;
2410
2411 if (fd != -1)
2412 {
2413 union generic_unix_sockaddr uaddr;
2414 unsigned int uaddrlen = sizeof(uaddr);
2415
2416 if (getpeername(fd, &uaddr.addr, &uaddrlen) == 0)
2417 {
2418 if (!name || !namelen)
2419 SetLastError(WSAEFAULT);
2420 else if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0)
2421 /* The buffer was too small */
2422 SetLastError(WSAEFAULT);
2423 else
2424 res = 0;
2425 }
2426 else
2427 SetLastError(wsaErrno());
2428 release_sock_fd( s, fd );
2429 }
2430 return res;
2431 }
2432
2433 /***********************************************************************
2434 * getsockname (WS2_32.6)
2435 */
2436 int WINAPI WS_getsockname(SOCKET s, struct WS_sockaddr *name, int *namelen)
2437 {
2438 int fd;
2439 int res;
2440
2441 TRACE("socket: %04lx, ptr %p, len %8x\n", s, name, *namelen);
2442
2443 /* Check if what we've received is valid. Should we use IsBadReadPtr? */
2444 if( (name == NULL) || (namelen == NULL) )
2445 {
2446 SetLastError( WSAEFAULT );
2447 return SOCKET_ERROR;
2448 }
2449
2450 fd = get_sock_fd( s, 0, NULL );
2451 res = SOCKET_ERROR;
2452
2453 if (fd != -1)
2454 {
2455 union generic_unix_sockaddr uaddr;
2456 unsigned int uaddrlen = sizeof(uaddr);
2457
2458 if (getsockname(fd, &uaddr.addr, &uaddrlen) != 0)
2459 {
2460 SetLastError(wsaErrno());
2461 }
2462 else if (!is_sockaddr_bound(&uaddr.addr, uaddrlen))
2463 {
2464 SetLastError(WSAEINVAL);
2465 }
2466 else if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0)
2467 {
2468 /* The buffer was too small */
2469 SetLastError(WSAEFAULT);
2470 }
2471 else
2472 {
2473 res=0;
2474 }
2475 release_sock_fd( s, fd );
2476 }
2477 return res;
2478 }
2479
2480 /***********************************************************************
2481 * getsockopt (WS2_32.7)
2482 */
2483 INT WINAPI WS_getsockopt(SOCKET s, INT level,
2484 INT optname, char *optval, INT *optlen)
2485 {
2486 int fd;
2487 INT ret = 0;
2488
2489 TRACE("socket: %04lx, level 0x%x, name 0x%x, ptr %p, len %d\n",
2490 s, level, optname, optval, *optlen);
2491
2492 switch(level)
2493 {
2494 case WS_SOL_SOCKET:
2495 {
2496 switch(optname)
2497 {
2498 /* Handle common cases. The special cases are below, sorted
2499 * alphabetically */
2500 case WS_SO_ACCEPTCONN:
2501 case WS_SO_BROADCAST:
2502 case WS_SO_DEBUG:
2503 case WS_SO_ERROR:
2504 case WS_SO_KEEPALIVE:
2505 case WS_SO_OOBINLINE:
2506 case WS_SO_RCVBUF:
2507 case WS_SO_REUSEADDR:
2508 case WS_SO_SNDBUF:
2509 case WS_SO_TYPE:
2510 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2511 return SOCKET_ERROR;
2512 convert_sockopt(&level, &optname);
2513 if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
2514 {
2515 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2516 ret = SOCKET_ERROR;
2517 }
2518 release_sock_fd( s, fd );
2519 return ret;
2520
2521 case WS_SO_DONTLINGER:
2522 {
2523 struct linger lingval;
2524 unsigned int len = sizeof(struct linger);
2525
2526 if (!optlen || *optlen < sizeof(BOOL)|| !optval)
2527 {
2528 SetLastError(WSAEFAULT);
2529 return SOCKET_ERROR;
2530 }
2531 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2532 return SOCKET_ERROR;
2533
2534 if (getsockopt(fd, SOL_SOCKET, SO_LINGER, &lingval, &len) != 0 )
2535 {
2536 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2537 ret = SOCKET_ERROR;
2538 }
2539 else
2540 {
2541 *(BOOL *)optval = (lingval.l_onoff) ? FALSE : TRUE;
2542 *optlen = sizeof(BOOL);
2543 }
2544
2545 release_sock_fd( s, fd );
2546 return ret;
2547 }
2548
2549 case WS_SO_CONNECT_TIME:
2550 {
2551 static int pretendtime = 0;
2552
2553 if (!pretendtime) FIXME("WS_SO_CONNECT_TIME - faking results\n");
2554 if (!optlen || *optlen < sizeof(DWORD) || !optval)
2555 {
2556 SetLastError(WSAEFAULT);
2557 return SOCKET_ERROR;
2558 }
2559 *(DWORD*)optval = pretendtime++;
2560 *optlen = sizeof(DWORD);
2561 return ret;
2562 }
2563 /* As mentioned in setsockopt, Windows ignores this, so we
2564 * always return true here */
2565 case WS_SO_DONTROUTE:
2566 if (!optlen || *optlen < sizeof(BOOL) || !optval)
2567 {
2568 SetLastError(WSAEFAULT);
2569 return SOCKET_ERROR;
2570 }
2571 *(BOOL *)optval = TRUE;
2572 *optlen = sizeof(BOOL);
2573 return 0;
2574
2575 case WS_SO_LINGER:
2576 {
2577 struct linger lingval;
2578 int so_type;
2579 unsigned int len = sizeof(struct linger), slen = sizeof(int);
2580
2581 /* struct linger and LINGER have different sizes */
2582 if (!optlen || *optlen < sizeof(LINGER) || !optval)
2583 {
2584 SetLastError(WSAEFAULT);
2585 return SOCKET_ERROR;
2586 }
2587 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2588 return SOCKET_ERROR;
2589
2590 if ((getsockopt(fd, SOL_SOCKET, SO_TYPE, &so_type, &slen) == 0 && so_type == SOCK_DGRAM))
2591 {
2592 SetLastError(WSAENOPROTOOPT);
2593 ret = SOCKET_ERROR;
2594 }
2595 else if (getsockopt(fd, SOL_SOCKET, SO_LINGER, &lingval, &len) != 0)
2596 {
2597 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2598 ret = SOCKET_ERROR;
2599 }
2600 else
2601 {
2602 ((LINGER *)optval)->l_onoff = lingval.l_onoff;
2603 ((LINGER *)optval)->l_linger = lingval.l_linger;
2604 *optlen = sizeof(struct linger);
2605 }
2606
2607 release_sock_fd( s, fd );
2608 return ret;
2609 }
2610
2611 case WS_SO_MAX_MSG_SIZE:
2612 if (!optlen || *optlen < sizeof(int) || !optval)
2613 {
2614 SetLastError(WSAEFAULT);
2615 return SOCKET_ERROR;
2616 }
2617 TRACE("getting global SO_MAX_MSG_SIZE = 65507\n");
2618 *(int *)optval = 65507;
2619 *optlen = sizeof(int);
2620 return 0;
2621
2622 /* SO_OPENTYPE does not require a valid socket handle. */
2623 case WS_SO_OPENTYPE:
2624 if (!optlen || *optlen < sizeof(int) || !optval)
2625 {
2626 SetLastError(WSAEFAULT);
2627 return SOCKET_ERROR;
2628 }
2629 *(int *)optval = get_per_thread_data()->opentype;
2630 *optlen = sizeof(int);
2631 TRACE("getting global SO_OPENTYPE = 0x%x\n", *((int*)optval) );
2632 return 0;
2633
2634 #ifdef SO_RCVTIMEO
2635 case WS_SO_RCVTIMEO:
2636 #endif
2637 #ifdef SO_SNDTIMEO
2638 case WS_SO_SNDTIMEO:
2639 #endif
2640 #if defined(SO_RCVTIMEO) || defined(SO_SNDTIMEO)
2641 {
2642 struct timeval tv;
2643 unsigned int len = sizeof(struct timeval);
2644
2645 if (!optlen || *optlen < sizeof(int)|| !optval)
2646 {
2647 SetLastError(WSAEFAULT);
2648 return SOCKET_ERROR;
2649 }
2650 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2651 return SOCKET_ERROR;
2652
2653 convert_sockopt(&level, &optname);
2654 if (getsockopt(fd, level, optname, &tv, &len) != 0 )
2655 {
2656 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2657 ret = SOCKET_ERROR;
2658 }
2659 else
2660 {
2661 *(int *)optval = tv.tv_sec * 1000 + tv.tv_usec / 1000;
2662 *optlen = sizeof(int);
2663 }
2664
2665 release_sock_fd( s, fd );
2666 return ret;
2667 }
2668 #endif
2669 default:
2670 TRACE("Unknown SOL_SOCKET optname: 0x%08x\n", optname);
2671 SetLastError(WSAENOPROTOOPT);
2672 return SOCKET_ERROR;
2673 } /* end switch(optname) */
2674 }/* end case WS_SOL_SOCKET */
2675 #ifdef HAVE_IPX
2676 case NSPROTO_IPX:
2677 {
2678 struct WS_sockaddr_ipx addr;
2679 IPX_ADDRESS_DATA *data;
2680 int namelen;
2681 switch(optname)
2682 {
2683 case IPX_PTYPE:
2684 if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR;
2685 #ifdef SOL_IPX
2686 if(getsockopt(fd, SOL_IPX, IPX_TYPE, optval, (unsigned int*)optlen) == -1)
2687 {
2688 ret = SOCKET_ERROR;
2689 }
2690 #else
2691 {
2692 struct ipx val;
2693 socklen_t len=sizeof(struct ipx);
2694 if(getsockopt(fd, 0, SO_DEFAULT_HEADERS, &val, &len) == -1 )
2695 ret = SOCKET_ERROR;
2696 else
2697 *optval = (int)val.ipx_pt;
2698 }
2699 #endif
2700 TRACE("ptype: %d (fd: %d)\n", *(int*)optval, fd);
2701 release_sock_fd( s, fd );
2702 return ret;
2703
2704 case IPX_ADDRESS:
2705 /*
2706 * On a Win2000 system with one network card there are usually
2707 * three ipx devices one with a speed of 28.8kbps, 10Mbps and 100Mbps.
2708 * Using this call you can then retrieve info about this all.
2709 * In case of Linux it is a bit different. Usually you have
2710 * only "one" device active and further it is not possible to
2711 * query things like the linkspeed.
2712 */
2713 FIXME("IPX_ADDRESS\n");
2714 namelen = sizeof(struct WS_sockaddr_ipx);
2715 memset(&addr, 0, sizeof(struct WS_sockaddr_ipx));
2716 WS_getsockname(s, (struct WS_sockaddr*)&addr, &namelen);
2717
2718 data = (IPX_ADDRESS_DATA*)optval;
2719 memcpy(data->nodenum,addr.sa_nodenum,sizeof(data->nodenum));
2720 memcpy(data->netnum,addr.sa_netnum,sizeof(data->netnum));
2721 data->adapternum = 0;
2722 data->wan = FALSE; /* We are not on a wan for now .. */
2723 data->status = FALSE; /* Since we are not on a wan, the wan link isn't up */
2724 data->maxpkt = 1467; /* This value is the default one, at least on Win2k/WinXP */
2725 data->linkspeed = 100000; /* Set the line speed in 100bit/s to 10 Mbit;
2726 * note 1MB = 1000kB in this case */
2727 return 0;
2728
2729 case IPX_MAX_ADAPTER_NUM:
2730 FIXME("IPX_MAX_ADAPTER_NUM\n");
2731 *(int*)optval = 1; /* As noted under IPX_ADDRESS we have just one card. */
2732 return 0;
2733
2734 default:
2735 FIXME("IPX optname:%x\n", optname);
2736 return SOCKET_ERROR;
2737 }/* end switch(optname) */
2738 } /* end case NSPROTO_IPX */
2739 #endif
2740
2741 #ifdef HAVE_IRDA
2742 case WS_SOL_IRLMP:
2743 switch(optname)
2744 {
2745 case WS_IRLMP_ENUMDEVICES:
2746 {
2747 static const int MAX_IRDA_DEVICES = 10;
2748 char buf[sizeof(struct irda_device_list) +
2749 (MAX_IRDA_DEVICES - 1) * sizeof(struct irda_device_info)];
2750 int fd, res;
2751 socklen_t len = sizeof(buf);
2752
2753 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2754 return SOCKET_ERROR;
2755 res = getsockopt( fd, SOL_IRLMP, IRLMP_ENUMDEVICES, buf, &len );
2756 if (res < 0)
2757 {
2758 SetLastError(wsaErrno());
2759 return SOCKET_ERROR;
2760 }
2761 else
2762 {
2763 struct irda_device_list *src = (struct irda_device_list *)buf;
2764 DEVICELIST *dst = (DEVICELIST *)optval;
2765 INT needed = sizeof(DEVICELIST), i;
2766
2767 if (src->len > 0)
2768 needed += (src->len - 1) * sizeof(IRDA_DEVICE_INFO);
2769 if (*optlen < needed)
2770 {
2771 SetLastError(WSAEFAULT);
2772 return SOCKET_ERROR;
2773 }
2774 *optlen = needed;
2775 TRACE("IRLMP_ENUMDEVICES: %d devices found:\n", src->len);
2776 dst->numDevice = src->len;
2777 for (i = 0; i < src->len; i++)
2778 {
2779 TRACE("saddr = %08x, daddr = %08x, info = %s, hints = %02x%02x\n",
2780 src->dev[i].saddr, src->dev[i].daddr,
2781 src->dev[i].info, src->dev[i].hints[0],
2782 src->dev[i].hints[1]);
2783 memcpy( dst->Device[i].irdaDeviceID,
2784 &src->dev[i].daddr,
2785 sizeof(dst->Device[i].irdaDeviceID) ) ;
2786 memcpy( dst->Device[i].irdaDeviceName,
2787 &src->dev[i].info,
2788 sizeof(dst->Device[i].irdaDeviceName) ) ;
2789 memcpy( &dst->Device[i].irdaDeviceHints1,
2790 &src->dev[i].hints[0],
2791 sizeof(dst->Device[i].irdaDeviceHints1) ) ;
2792 memcpy( &dst->Device[i].irdaDeviceHints2,
2793 &src->dev[i].hints[1],
2794 sizeof(dst->Device[i].irdaDeviceHints2) ) ;
2795 dst->Device[i].irdaCharSet = src->dev[i].charset;
2796 }
2797 return 0;
2798 }
2799 }
2800 default:
2801 FIXME("IrDA optname:0x%x\n", optname);
2802 return SOCKET_ERROR;
2803 }
2804 break; /* case WS_SOL_IRLMP */
2805 #endif
2806
2807 /* Levels WS_IPPROTO_TCP and WS_IPPROTO_IP convert directly */
2808 case WS_IPPROTO_TCP:
2809 switch(optname)
2810 {
2811 case WS_TCP_NODELAY:
2812 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2813 return SOCKET_ERROR;
2814 convert_sockopt(&level, &optname);
2815 if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
2816 {
2817 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2818 ret = SOCKET_ERROR;
2819 }
2820 release_sock_fd( s, fd );
2821 return ret;
2822 }
2823 FIXME("Unknown IPPROTO_TCP optname 0x%08x\n", optname);
2824 return SOCKET_ERROR;
2825
2826 case WS_IPPROTO_IP:
2827 switch(optname)
2828 {
2829 case WS_IP_ADD_MEMBERSHIP:
2830 case WS_IP_DROP_MEMBERSHIP:
2831 #ifdef IP_HDRINCL
2832 case WS_IP_HDRINCL:
2833 #endif
2834 case WS_IP_MULTICAST_IF:
2835 case WS_IP_MULTICAST_LOOP:
2836 case WS_IP_MULTICAST_TTL:
2837 case WS_IP_OPTIONS:
2838 #ifdef IP_PKTINFO
2839 case WS_IP_PKTINFO:
2840 #endif
2841 case WS_IP_TOS:
2842 case WS_IP_TTL:
2843 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2844 return SOCKET_ERROR;
2845 convert_sockopt(&level, &optname);
2846 if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
2847 {
2848 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2849 ret = SOCKET_ERROR;
2850 }
2851 release_sock_fd( s, fd );
2852 return ret;
2853 case WS_IP_DONTFRAGMENT:
2854 FIXME("WS_IP_DONTFRAGMENT is always false!\n");
2855 *(BOOL*)optval = FALSE;
2856 return 0;
2857 }
2858 FIXME("Unknown IPPROTO_IP optname 0x%08x\n", optname);
2859 return SOCKET_ERROR;
2860
2861 case WS_IPPROTO_IPV6:
2862 switch(optname)
2863 {
2864 #ifdef IPV6_ADD_MEMBERSHIP
2865 case WS_IPV6_ADD_MEMBERSHIP:
2866 #endif
2867 #ifdef IPV6_DROP_MEMBERSHIP
2868 case WS_IPV6_DROP_MEMBERSHIP:
2869 #endif
2870 case WS_IPV6_MULTICAST_IF:
2871 case WS_IPV6_MULTICAST_HOPS:
2872 case WS_IPV6_MULTICAST_LOOP:
2873 case WS_IPV6_UNICAST_HOPS:
2874 case WS_IPV6_V6ONLY:
2875 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
2876 return SOCKET_ERROR;
2877 convert_sockopt(&level, &optname);
2878 if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
2879 {
2880 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2881 ret = SOCKET_ERROR;
2882 }
2883 release_sock_fd( s, fd );
2884 return ret;
2885 case WS_IPV6_DONTFRAG:
2886 FIXME("WS_IPV6_DONTFRAG is always false!\n");
2887 *(BOOL*)optval = FALSE;
2888 return 0;
2889 }
2890 FIXME("Unknown IPPROTO_IPV6 optname 0x%08x\n", optname);
2891 return SOCKET_ERROR;
2892
2893 default:
2894 WARN("Unknown level: 0x%08x\n", level);
2895 SetLastError(WSAEINVAL);
2896 return SOCKET_ERROR;
2897 } /* end switch(level) */
2898 }
2899
2900 /***********************************************************************
2901 * htonl (WS2_32.8)
2902 */
2903 WS_u_long WINAPI WS_htonl(WS_u_long hostlong)
2904 {
2905 return htonl(hostlong);
2906 }
2907
2908
2909 /***********************************************************************
2910 * htons (WS2_32.9)
2911 */
2912 WS_u_short WINAPI WS_htons(WS_u_short hostshort)
2913 {
2914 return htons(hostshort);
2915 }
2916
2917 /***********************************************************************
2918 * WSAHtonl (WS2_32.46)
2919 * From MSDN description of error codes, this function should also
2920 * check if WinSock has been initialized and the socket is a valid
2921 * socket. But why? This function only translates a host byte order
2922 * u_long into a network byte order u_long...
2923 */
2924 int WINAPI WSAHtonl(SOCKET s, WS_u_long hostlong, WS_u_long *lpnetlong)
2925 {
2926 if (lpnetlong)
2927 {
2928 *lpnetlong = htonl(hostlong);
2929 return 0;
2930 }
2931 WSASetLastError(WSAEFAULT);
2932 return SOCKET_ERROR;
2933 }
2934
2935 /***********************************************************************
2936 * WSAHtons (WS2_32.47)
2937 * From MSDN description of error codes, this function should also
2938 * check if WinSock has been initialized and the socket is a valid
2939 * socket. But why? This function only translates a host byte order
2940 * u_short into a network byte order u_short...
2941 */
2942 int WINAPI WSAHtons(SOCKET s, WS_u_short hostshort, WS_u_short *lpnetshort)
2943 {
2944
2945 if (lpnetshort)
2946 {
2947 *lpnetshort = htons(hostshort);
2948 return 0;
2949 }
2950 WSASetLastError(WSAEFAULT);
2951 return SOCKET_ERROR;
2952 }
2953
2954
2955 /***********************************************************************
2956 * inet_addr (WS2_32.11)
2957 */
2958 WS_u_long WINAPI WS_inet_addr(const char *cp)
2959 {
2960 if (!cp) return INADDR_NONE;
2961 return inet_addr(cp);
2962 }
2963
2964
2965 /***********************************************************************
2966 * ntohl (WS2_32.14)
2967 */
2968 WS_u_long WINAPI WS_ntohl(WS_u_long netlong)
2969 {
2970 return ntohl(netlong);
2971 }
2972
2973
2974 /***********************************************************************
2975 * ntohs (WS2_32.15)
2976 */
2977 WS_u_short WINAPI WS_ntohs(WS_u_short netshort)
2978 {
2979 return ntohs(netshort);
2980 }
2981
2982
2983 /***********************************************************************
2984 * inet_ntoa (WS2_32.12)
2985 */
2986 char* WINAPI WS_inet_ntoa(struct WS_in_addr in)
2987 {
2988 /* use "buffer for dummies" here because some applications have a
2989 * propensity to decode addresses in ws_hostent structure without
2990 * saving them first...
2991 */
2992 static char dbuffer[16]; /* Yes, 16: 4*3 digits + 3 '.' + 1 '\0' */
2993
2994 char* s = inet_ntoa(*((struct in_addr*)&in));
2995 if( s )
2996 {
2997 strcpy(dbuffer, s);
2998 return dbuffer;
2999 }
3000 SetLastError(wsaErrno());
3001 return NULL;
3002 }
3003
3004 static const char *debugstr_wsaioctl(DWORD ioctl)
3005 {
3006 const char *buf_type, *family;
3007
3008 switch(ioctl & 0x18000000)
3009 {
3010 case WS_IOC_WS2:
3011 family = "IOC_WS2";
3012 break;
3013 case WS_IOC_PROTOCOL:
3014 family = "IOC_PROTOCOL";
3015 break;
3016 case WS_IOC_VENDOR:
3017 family = "IOC_VENDOR";
3018 break;
3019 default: /* WS_IOC_UNIX */
3020 {
3021 BYTE size = (ioctl >> 16) & WS_IOCPARM_MASK;
3022 char x = (ioctl & 0xff00) >> 8;
3023 BYTE y = ioctl & 0xff;
3024 char args[14];
3025
3026 switch (ioctl & (WS_IOC_VOID|WS_IOC_INOUT))
3027 {
3028 case WS_IOC_VOID:
3029 buf_type = "_IO";
3030 sprintf(args, "%d, %d", x, y);
3031 break;
3032 case WS_IOC_IN:
3033 buf_type = "_IOW";
3034 sprintf(args, "'%c', %d, %d", x, y, size);
3035 break;
3036 case WS_IOC_OUT:
3037 buf_type = "_IOR";
3038 sprintf(args, "'%c', %d, %d", x, y, size);
3039 break;
3040 default:
3041 buf_type = "?";
3042 sprintf(args, "'%c', %d, %d", x, y, size);
3043 break;
3044 }
3045 return wine_dbg_sprintf("%s(%s)", buf_type, args);
3046 }
3047 }
3048
3049 /* We are different from WS_IOC_UNIX. */
3050 switch (ioctl & (WS_IOC_VOID|WS_IOC_INOUT))
3051 {
3052 case WS_IOC_VOID:
3053 buf_type = "_WSAIO";
3054 break;
3055 case WS_IOC_INOUT:
3056 buf_type = "_WSAIORW";
3057 break;
3058 case WS_IOC_IN:
3059 buf_type = "_WSAIOW";
3060 break;
3061 case WS_IOC_OUT:
3062 buf_type = "_WSAIOR";
3063 break;
3064 default:
3065 buf_type = "?";
3066 break;
3067 }
3068
3069 return wine_dbg_sprintf("%s(%s, %d)", buf_type, family,
3070 (USHORT)(ioctl & 0xffff));
3071 }
3072
3073 /**********************************************************************
3074 * WSAIoctl (WS2_32.50)
3075 *
3076 */
3077 INT WINAPI WSAIoctl(SOCKET s, DWORD code, LPVOID in_buff, DWORD in_size, LPVOID out_buff,
3078 DWORD out_size, LPDWORD ret_size, LPWSAOVERLAPPED overlapped,
3079 LPWSAOVERLAPPED_COMPLETION_ROUTINE completion )
3080 {
3081 int fd;
3082 DWORD status = 0, total = 0;
3083
3084 TRACE("%ld, 0x%08x, %p, %d, %p, %d, %p, %p, %p\n",
3085 s, code, in_buff, in_size, out_buff, out_size, ret_size, overlapped, completion);
3086
3087 switch (code)
3088 {
3089 case WS_FIONBIO:
3090 if (in_size != sizeof(WS_u_long) || IS_INTRESOURCE(in_buff))
3091 {
3092 WSASetLastError(WSAEFAULT);
3093 return SOCKET_ERROR;
3094 }
3095 if (_get_sock_mask(s))
3096 {
3097 /* AsyncSelect()'ed sockets are always nonblocking */
3098 if (!*(WS_u_long *)in_buff) status = WSAEINVAL;
3099 break;
3100 }
3101 if (*(WS_u_long *)in_buff)
3102 _enable_event(SOCKET2HANDLE(s), 0, FD_WINE_NONBLOCKING, 0);
3103 else
3104 _enable_event(SOCKET2HANDLE(s), 0, 0, FD_WINE_NONBLOCKING);
3105 break;
3106
3107 case WS_FIONREAD:
3108 {
3109 if (out_size != sizeof(WS_u_long) || IS_INTRESOURCE(out_buff))
3110 {
3111 WSASetLastError(WSAEFAULT);
3112 return SOCKET_ERROR;
3113 }
3114 if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR;
3115 if (ioctl(fd, FIONREAD, out_buff ) == -1)
3116 status = (errno == EBADF) ? WSAENOTSOCK : wsaErrno();
3117 release_sock_fd( s, fd );
3118 break;
3119 }
3120
3121 case WS_SIOCATMARK:
3122 {
3123 unsigned int oob = 0, oobsize = sizeof(int), atmark = 0;
3124 if (out_size != sizeof(WS_u_long) || IS_INTRESOURCE(out_buff))
3125 {
3126 WSASetLastError(WSAEFAULT);
3127 return SOCKET_ERROR;
3128 }
3129 if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR;
3130 /* SO_OOBINLINE sockets must always return TRUE to SIOCATMARK */
3131 if ((getsockopt(fd, SOL_SOCKET, SO_OOBINLINE, &oob, &oobsize ) == -1)
3132 || (!oob && ioctl(fd, SIOCATMARK, &atmark ) == -1))
3133 status = (errno == EBADF) ? WSAENOTSOCK : wsaErrno();
3134 else
3135 (*(WS_u_long *) out_buff) = oob | atmark;
3136 release_sock_fd( s, fd );
3137 break;
3138 }
3139
3140 case WS_FIOASYNC:
3141 WARN("Warning: WS1.1 shouldn't be using async I/O\n");
3142 SetLastError(WSAEINVAL);
3143 return SOCKET_ERROR;
3144
3145 case WS_SIO_GET_INTERFACE_LIST:
3146 {
3147 INTERFACE_INFO* intArray = out_buff;
3148 DWORD size, numInt = 0, apiReturn;
3149
3150 TRACE("-> SIO_GET_INTERFACE_LIST request\n");
3151
3152 if (!out_buff || !ret_size)
3153 {
3154 WSASetLastError(WSAEFAULT);
3155 return SOCKET_ERROR;
3156 }
3157
3158 fd = get_sock_fd( s, 0, NULL );
3159 if (fd == -1) return SOCKET_ERROR;
3160
3161 apiReturn = GetAdaptersInfo(NULL, &size);
3162 if (apiReturn == ERROR_BUFFER_OVERFLOW)
3163 {
3164 PIP_ADAPTER_INFO table = HeapAlloc(GetProcessHeap(),0,size);
3165
3166 if (table)
3167 {
3168 if (GetAdaptersInfo(table, &size) == NO_ERROR)
3169 {
3170 PIP_ADAPTER_INFO ptr;
3171
3172 if (size*sizeof(INTERFACE_INFO)/sizeof(IP_ADAPTER_INFO) > out_size)
3173 {
3174 WARN("Buffer too small = %u, out_size = %u\n", size, out_size);
3175 HeapFree(GetProcessHeap(),0,table);
3176 release_sock_fd( s, fd );
3177 status = WSAEFAULT;
3178 break;
3179 }
3180 for (ptr = table, numInt = 0; ptr;
3181 ptr = ptr->Next, intArray++, numInt++)
3182 {
3183 unsigned int addr, mask, bcast;
3184 struct ifreq ifInfo;
3185
3186 /* Socket Status Flags */
3187 lstrcpynA(ifInfo.ifr_name, ptr->AdapterName, IFNAMSIZ);
3188 if (ioctl(fd, SIOCGIFFLAGS, &ifInfo) < 0)
3189 {
3190 ERR("Error obtaining status flags for socket!\n");
3191 HeapFree(GetProcessHeap(),0,table);
3192 release_sock_fd( s, fd );
3193 status = WSAEINVAL;
3194 break;
3195 }
3196 else
3197 {
3198 /* set flags; the values of IFF_* are not the same
3199 under Linux and Windows, therefore must generate
3200 new flags */
3201 intArray->iiFlags = 0;
3202 if (ifInfo.ifr_flags & IFF_BROADCAST)
3203 intArray->iiFlags |= WS_IFF_BROADCAST;
3204 #ifdef IFF_POINTOPOINT
3205 if (ifInfo.ifr_flags & IFF_POINTOPOINT)
3206 intArray->iiFlags |= WS_IFF_POINTTOPOINT;
3207 #endif
3208 if (ifInfo.ifr_flags & IFF_LOOPBACK)
3209 intArray->iiFlags |= WS_IFF_LOOPBACK;
3210 if (ifInfo.ifr_flags & IFF_UP)
3211 intArray->iiFlags |= WS_IFF_UP;
3212 if (ifInfo.ifr_flags & IFF_MULTICAST)
3213 intArray->iiFlags |= WS_IFF_MULTICAST;
3214 }
3215
3216 addr = inet_addr(ptr->IpAddressList.IpAddress.String);
3217 mask = inet_addr(ptr->IpAddressList.IpMask.String);
3218 bcast = addr | ~mask;
3219 intArray->iiAddress.AddressIn.sin_family = AF_INET;
3220 intArray->iiAddress.AddressIn.sin_port = 0;
3221 intArray->iiAddress.AddressIn.sin_addr.WS_s_addr =
3222 addr;
3223 intArray->iiNetmask.AddressIn.sin_family = AF_INET;
3224 intArray->iiNetmask.AddressIn.sin_port = 0;
3225 intArray->iiNetmask.AddressIn.sin_addr.WS_s_addr =
3226 mask;
3227 intArray->iiBroadcastAddress.AddressIn.sin_family =
3228 AF_INET;
3229 intArray->iiBroadcastAddress.AddressIn.sin_port = 0;
3230 intArray->iiBroadcastAddress.AddressIn.sin_addr.
3231 WS_s_addr = bcast;
3232 }
3233 }
3234 else
3235 {
3236 ERR("Unable to get interface table!\n");
3237 status = WSAEINVAL;
3238 }
3239 HeapFree(GetProcessHeap(),0,table);
3240 }
3241 else status = WSAEINVAL;
3242 }
3243 else if (apiReturn != ERROR_NO_DATA)
3244 {
3245 ERR("Unable to get interface table!\n");
3246 status = WSAEINVAL;
3247 }
3248 /* Calculate the size of the array being returned */
3249 total = sizeof(INTERFACE_INFO) * numInt;
3250 release_sock_fd( s, fd );
3251 break;
3252 }
3253
3254 case WS_SIO_ADDRESS_LIST_CHANGE:
3255 FIXME("-> SIO_ADDRESS_LIST_CHANGE request: stub\n");
3256 /* FIXME: error and return code depend on whether socket was created
3257 * with WSA_FLAG_OVERLAPPED, but there is no easy way to get this */
3258 break;
3259
3260 case WS_SIO_ADDRESS_LIST_QUERY:
3261 {
3262 DWORD size;
3263
3264 TRACE("-> SIO_ADDRESS_LIST_QUERY request\n");
3265
3266 if (!ret_size)
3267 {
3268 WSASetLastError(WSAEFAULT);
3269 return SOCKET_ERROR;
3270 }
3271
3272 if (GetAdaptersInfo(NULL, &size) == ERROR_BUFFER_OVERFLOW)
3273 {
3274 IP_ADAPTER_INFO *p, *table = HeapAlloc(GetProcessHeap(), 0, size);
3275 DWORD num;
3276
3277 if (!table || GetAdaptersInfo(table, &size))
3278 {
3279 HeapFree(GetProcessHeap(), 0, table);
3280 status = WSAEINVAL;
3281 break;
3282 }
3283
3284 for (p = table, num = 0; p; p = p->Next)
3285 if (p->IpAddressList.IpAddress.String[0]) num++;
3286
3287 total = sizeof(SOCKET_ADDRESS_LIST) + sizeof(SOCKET_ADDRESS) * (num - 1);
3288 total += sizeof(SOCKADDR) * num;
3289
3290 if (total > out_size)
3291 {
3292 HeapFree(GetProcessHeap(), 0, table);
3293 status = WSAEFAULT;
3294 break;
3295 }
3296
3297 if (out_buff)
3298 {
3299 unsigned int i;
3300 SOCKET_ADDRESS *sa;
3301 SOCKET_ADDRESS_LIST *sa_list = out_buff;
3302 SOCKADDR_IN *sockaddr;
3303
3304 sa = sa_list->Address;
3305 sockaddr = (SOCKADDR_IN *)((char *)sa + num * sizeof(SOCKET_ADDRESS));
3306 sa_list->iAddressCount = num;
3307
3308 for (p = table, i = 0; p; p = p->Next)
3309 {
3310 if (!p->IpAddressList.IpAddress.String[0]) continue;
3311
3312 sa[i].lpSockaddr = (SOCKADDR *)&sockaddr[i];
3313 sa[i].iSockaddrLength = sizeof(SOCKADDR);
3314
3315 sockaddr[i].sin_family = AF_INET;
3316 sockaddr[i].sin_port = 0;
3317 sockaddr[i].sin_addr.WS_s_addr = inet_addr(p->IpAddressList.IpAddress.String);
3318 i++;
3319 }
3320 }
3321
3322 HeapFree(GetProcessHeap(), 0, table);
3323 }
3324 else
3325 {
3326 WARN("unable to get IP address list\n");
3327 status = WSAEINVAL;
3328 }
3329 break;
3330 }
3331
3332 case WS_SIO_FLUSH:
3333 FIXME("SIO_FLUSH: stub.\n");
3334 break;
3335
3336 case WS_SIO_GET_EXTENSION_FUNCTION_POINTER:
3337 {
3338 static const GUID connectex_guid = WSAID_CONNECTEX;
3339 static const GUID disconnectex_guid = WSAID_DISCONNECTEX;
3340 static const GUID acceptex_guid = WSAID_ACCEPTEX;
3341 static const GUID getaccepexsockaddrs_guid = WSAID_GETACCEPTEXSOCKADDRS;
3342 static const GUID transmitfile_guid = WSAID_TRANSMITFILE;
3343 static const GUID transmitpackets_guid = WSAID_TRANSMITPACKETS;
3344 static const GUID wsarecvmsg_guid = WSAID_WSARECVMSG;
3345 static const GUID wsasendmsg_guid = WSAID_WSASENDMSG;
3346
3347 if ( IsEqualGUID(&connectex_guid, in_buff) )
3348 {
3349 *(LPFN_CONNECTEX *)out_buff = WS2_ConnectEx;
3350 break;
3351 }
3352 else if ( IsEqualGUID(&disconnectex_guid, in_buff) )
3353 {
3354 FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER: unimplemented DisconnectEx\n");
3355 }
3356 else if ( IsEqualGUID(&acceptex_guid, in_buff) )
3357 {
3358 *(LPFN_ACCEPTEX *)out_buff = WS2_AcceptEx;
3359 break;
3360 }
3361 else if ( IsEqualGUID(&getaccepexsockaddrs_guid, in_buff) )
3362 {
3363 *(LPFN_GETACCEPTEXSOCKADDRS *)out_buff = WS2_GetAcceptExSockaddrs;
3364 break;
3365 }
3366 else if ( IsEqualGUID(&transmitfile_guid, in_buff) )
3367 {
3368 FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER: unimplemented TransmitFile\n");
3369 }
3370 else if ( IsEqualGUID(&transmitpackets_guid, in_buff) )
3371 {
3372 FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER: unimplemented TransmitPackets\n");
3373 }
3374 else if ( IsEqualGUID(&wsarecvmsg_guid, in_buff) )
3375 {
3376 *(LPFN_WSARECVMSG *)out_buff = WS2_WSARecvMsg;
3377 break;
3378 }
3379 else if ( IsEqualGUID(&wsasendmsg_guid, in_buff) )
3380 {
3381 FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER: unimplemented WSASendMsg\n");
3382 }
3383 else
3384 FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER %s: stub\n", debugstr_guid(in_buff));
3385
3386 status = WSAEOPNOTSUPP;
3387 break;
3388 }
3389 case WS_SIO_KEEPALIVE_VALS:
3390 {
3391 struct tcp_keepalive *k;
3392 int keepalive, keepidle, keepintvl;
3393
3394 if (!in_buff || in_size < sizeof(struct tcp_keepalive))
3395 {
3396 WSASetLastError(WSAEFAULT);
3397 return SOCKET_ERROR;
3398 }
3399
3400 k = in_buff;
3401 keepalive = k->onoff ? 1 : 0;
3402 keepidle = max( 1, (k->keepalivetime + 500) / 1000 );
3403 keepintvl = max( 1, (k->keepaliveinterval + 500) / 1000 );
3404
3405 TRACE("onoff: %d, keepalivetime: %d, keepaliveinterval: %d\n", keepalive, keepidle, keepintvl);
3406
3407 fd = get_sock_fd(s, 0, NULL);
3408 if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, (void *)&keepalive, sizeof(int)) == -1)
3409 status = WSAEINVAL;
3410 #if defined(TCP_KEEPIDLE) && defined(TCP_KEEPINTVL)
3411 /* these values need to be set only if SO_KEEPALIVE is enabled */
3412 else if(keepalive)
3413 {
3414 if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPIDLE, (void *)&keepidle, sizeof(int)) == -1)
3415 status = WSAEINVAL;
3416 else if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPINTVL, (void *)&keepintvl, sizeof(int)) == -1)
3417 status = WSAEINVAL;
3418 }
3419 #else
3420 else
3421 FIXME("ignoring keepalive interval and timeout\n");
3422 #endif
3423 release_sock_fd(s, fd);
3424 break;
3425 }
3426 case WS_SIO_ROUTING_INTERFACE_QUERY:
3427 {
3428 struct WS_sockaddr *daddr = (struct WS_sockaddr *)in_buff;
3429 struct WS_sockaddr_in *daddr_in = (struct WS_sockaddr_in *)daddr;
3430 struct WS_sockaddr_in *saddr_in = out_buff;
3431 MIB_IPFORWARDROW row;
3432 PMIB_IPADDRTABLE ipAddrTable = NULL;
3433 DWORD size, i, found_index;
3434
3435 TRACE("-> WS_SIO_ROUTING_INTERFACE_QUERY request\n");
3436
3437 if (!in_buff || in_size < sizeof(struct WS_sockaddr) ||
3438 !out_buff || out_size < sizeof(struct WS_sockaddr_in) || !ret_size)
3439 {
3440 WSASetLastError(WSAEFAULT);
3441 return SOCKET_ERROR;
3442 }
3443 if (daddr->sa_family != AF_INET)
3444 {
3445 FIXME("unsupported address family %d\n", daddr->sa_family);
3446 status = WSAEAFNOSUPPORT;
3447 break;
3448 }
3449 if (GetBestRoute(daddr_in->sin_addr.S_un.S_addr, 0, &row) != NOERROR ||
3450 GetIpAddrTable(NULL, &size, FALSE) != ERROR_INSUFFICIENT_BUFFER)
3451 {
3452 status = WSAEFAULT;
3453 break;
3454 }
3455 ipAddrTable = HeapAlloc(GetProcessHeap(), 0, size);
3456 if (GetIpAddrTable(ipAddrTable, &size, FALSE))
3457 {
3458 HeapFree(GetProcessHeap(), 0, ipAddrTable);
3459 status = WSAEFAULT;
3460 break;
3461 }
3462 for (i = 0, found_index = ipAddrTable->dwNumEntries;
3463 i < ipAddrTable->dwNumEntries; i++)
3464 {
3465 if (ipAddrTable->table[i].dwIndex == row.dwForwardIfIndex)
3466 found_index = i;
3467 }
3468 if (found_index == ipAddrTable->dwNumEntries)
3469 {
3470 ERR("no matching IP address for interface %d\n",
3471 row.dwForwardIfIndex);
3472 HeapFree(GetProcessHeap(), 0, ipAddrTable);
3473 status = WSAEFAULT;
3474 break;
3475 }
3476 saddr_in->sin_family = AF_INET;
3477 saddr_in->sin_addr.S_un.S_addr = ipAddrTable->table[found_index].dwAddr;
3478 saddr_in->sin_port = 0;
3479 total = sizeof(struct WS_sockaddr_in);
3480 HeapFree(GetProcessHeap(), 0, ipAddrTable);
3481 break;
3482 }
3483 case WS_SIO_SET_COMPATIBILITY_MODE:
3484 TRACE("WS_SIO_SET_COMPATIBILITY_MODE ignored\n");
3485 status = WSAEOPNOTSUPP;
3486 break;
3487 case WS_SIO_UDP_CONNRESET:
3488 FIXME("WS_SIO_UDP_CONNRESET stub\n");
3489 break;
3490 case 0x667e: /* Netscape tries hard to use bogus ioctl 0x667e */
3491 WSASetLastError(WSAEOPNOTSUPP);
3492 return SOCKET_ERROR;
3493 default:
3494 FIXME("unsupported WS_IOCTL cmd (%s)\n", debugstr_wsaioctl(code));
3495 status = WSAEOPNOTSUPP;
3496 break;
3497 }
3498
3499 if (completion)
3500 {
3501 FIXME( "completion routine %p not supported\n", completion );
3502 }
3503 else if (overlapped)
3504 {
3505 ULONG_PTR cvalue = (overlapped && ((ULONG_PTR)overlapped->hEvent & 1) == 0) ? (ULONG_PTR)overlapped : 0;
3506 overlapped->Internal = status;
3507 overlapped->InternalHigh = total;
3508 if (overlapped->hEvent) NtSetEvent( overlapped->hEvent, NULL );
3509 if (cvalue) WS_AddCompletion( HANDLE2SOCKET(s), cvalue, status, total );
3510 }
3511
3512 if (!status)
3513 {
3514 if (ret_size) *ret_size = total;
3515 return 0;
3516 }
3517 SetLastError( status );
3518 return SOCKET_ERROR;
3519 }
3520
3521
3522 /***********************************************************************
3523 * ioctlsocket (WS2_32.10)
3524 */
3525 int WINAPI WS_ioctlsocket(SOCKET s, LONG cmd, WS_u_long *argp)
3526 {
3527 DWORD ret_size;
3528 return WSAIoctl( s, cmd, argp, sizeof(WS_u_long), argp, sizeof(WS_u_long), &ret_size, NULL, NULL );
3529 }
3530
3531 /***********************************************************************
3532 * listen (WS2_32.13)
3533 */
3534 int WINAPI WS_listen(SOCKET s, int backlog)
3535 {
3536 int fd = get_sock_fd( s, FILE_READ_DATA, NULL );
3537
3538 TRACE("socket %04lx, backlog %d\n", s, backlog);
3539 if (fd != -1)
3540 {
3541 if (listen(fd, backlog) == 0)
3542 {
3543 release_sock_fd( s, fd );
3544 _enable_event(SOCKET2HANDLE(s), FD_ACCEPT,
3545 FD_WINE_LISTENING,
3546 FD_CONNECT|FD_WINE_CONNECTED);
3547 return 0;
3548 }
3549 SetLastError(wsaErrno());
3550 release_sock_fd( s, fd );
3551 }
3552 return SOCKET_ERROR;
3553 }
3554
3555 /***********************************************************************
3556 * recv (WS2_32.16)
3557 */
3558 int WINAPI WS_recv(SOCKET s, char *buf, int len, int flags)
3559 {
3560 DWORD n, dwFlags = flags;
3561 WSABUF wsabuf;
3562
3563 wsabuf.len = len;
3564 wsabuf.buf = buf;
3565
3566 if ( WS2_recv_base(s, &wsabuf, 1, &n, &dwFlags, NULL, NULL, NULL, NULL, NULL) == SOCKET_ERROR )
3567 return SOCKET_ERROR;
3568 else
3569 return n;
3570 }
3571
3572 /***********************************************************************
3573 * recvfrom (WS2_32.17)
3574 */
3575 int WINAPI WS_recvfrom(SOCKET s, char *buf, INT len, int flags,
3576 struct WS_sockaddr *from, int *fromlen)
3577 {
3578 DWORD n, dwFlags = flags;
3579 WSABUF wsabuf;
3580
3581 wsabuf.len = len;
3582 wsabuf.buf = buf;
3583
3584 if ( WS2_recv_base(s, &wsabuf, 1, &n, &dwFlags, from, fromlen, NULL, NULL, NULL) == SOCKET_ERROR )
3585 return SOCKET_ERROR;
3586 else
3587 return n;
3588 }
3589
3590 /* allocate a poll array for the corresponding fd sets */
3591 static struct pollfd *fd_sets_to_poll( const WS_fd_set *readfds, const WS_fd_set *writefds,
3592 const WS_fd_set *exceptfds, int *count_ptr )
3593 {
3594 unsigned int i, j = 0, count = 0;
3595 struct pollfd *fds;
3596
3597 if (readfds) count += readfds->fd_count;
3598 if (writefds) count += writefds->fd_count;
3599 if (exceptfds) count += exceptfds->fd_count;
3600 *count_ptr = count;
3601 if (!count)
3602 {
3603 SetLastError(WSAEINVAL);
3604 return NULL;
3605 }
3606 if (!(fds = HeapAlloc( GetProcessHeap(), 0, count * sizeof(fds[0]))))
3607 {
3608 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
3609 return NULL;
3610 }
3611 if (readfds)
3612 for (i = 0; i < readfds->fd_count; i++, j++)
3613 {
3614 fds[j].fd = get_sock_fd( readfds->fd_array[i], FILE_READ_DATA, NULL );
3615 if (fds[j].fd == -1) goto failed;
3616 fds[j].events = POLLIN;
3617 fds[j].revents = 0;
3618 }
3619 if (writefds)
3620 for (i = 0; i < writefds->fd_count; i++, j++)
3621 {
3622 fds[j].fd = get_sock_fd( writefds->fd_array[i], FILE_WRITE_DATA, NULL );
3623 if (fds[j].fd == -1) goto failed;
3624 fds[j].events = POLLOUT;
3625 fds[j].revents = 0;
3626 }
3627 if (exceptfds)
3628 for (i = 0; i < exceptfds->fd_count; i++, j++)
3629 {
3630 fds[j].fd = get_sock_fd( exceptfds->fd_array[i], 0, NULL );
3631 if (fds[j].fd == -1) goto failed;
3632 fds[j].events = POLLHUP;
3633 fds[j].revents = 0;
3634 }
3635 return fds;
3636
3637 failed:
3638 count = j;
3639 j = 0;
3640 if (readfds)
3641 for (i = 0; i < readfds->fd_count && j < count; i++, j++)
3642 release_sock_fd( readfds->fd_array[i], fds[j].fd );
3643 if (writefds)
3644 for (i = 0; i < writefds->fd_count && j < count; i++, j++)
3645 release_sock_fd( writefds->fd_array[i], fds[j].fd );
3646 if (exceptfds)
3647 for (i = 0; i < exceptfds->fd_count && j < count; i++, j++)
3648 release_sock_fd( exceptfds->fd_array[i], fds[j].fd );
3649 HeapFree( GetProcessHeap(), 0, fds );
3650 return NULL;
3651 }
3652
3653 /* release the file descriptor obtained in fd_sets_to_poll */
3654 /* must be called with the original fd_set arrays, before calling get_poll_results */
3655 static void release_poll_fds( const WS_fd_set *readfds, const WS_fd_set *writefds,
3656 const WS_fd_set *exceptfds, struct pollfd *fds )
3657 {
3658 unsigned int i, j = 0;
3659
3660 if (readfds)
3661 {
3662 for (i = 0; i < readfds->fd_count; i++, j++)
3663 if (fds[j].fd != -1) release_sock_fd( readfds->fd_array[i], fds[j].fd );
3664 }
3665 if (writefds)
3666 {
3667 for (i = 0; i < writefds->fd_count; i++, j++)
3668 if (fds[j].fd != -1) release_sock_fd( writefds->fd_array[i], fds[j].fd );
3669 }
3670 if (exceptfds)
3671 {
3672 for (i = 0; i < exceptfds->fd_count; i++, j++)
3673 if (fds[j].fd != -1)
3674 {
3675 /* make sure we have a real error before releasing the fd */
3676 if (!sock_error_p( fds[j].fd )) fds[j].revents = 0;
3677 release_sock_fd( exceptfds->fd_array[i], fds[j].fd );
3678 }
3679 }
3680 }
3681
3682 /* map the poll results back into the Windows fd sets */
3683 static int get_poll_results( WS_fd_set *readfds, WS_fd_set *writefds, WS_fd_set *exceptfds,
3684 const struct pollfd *fds )
3685 {
3686 unsigned int i, j = 0, k, total = 0;
3687
3688 if (readfds)
3689 {
3690 for (i = k = 0; i < readfds->fd_count; i++, j++)
3691 if (fds[j].revents) readfds->fd_array[k++] = readfds->fd_array[i];
3692 readfds->fd_count = k;
3693 total += k;
3694 }
3695 if (writefds)
3696 {
3697 for (i = k = 0; i < writefds->fd_count; i++, j++)
3698 if ((fds[j].revents & POLLOUT) && !(fds[j].revents & POLLHUP))
3699 writefds->fd_array[k++] = writefds->fd_array[i];
3700 writefds->fd_count = k;
3701 total += k;
3702 }
3703 if (exceptfds)
3704 {
3705 for (i = k = 0; i < exceptfds->fd_count; i++, j++)
3706 if (fds[j].revents) exceptfds->fd_array[k++] = exceptfds->fd_array[i];
3707 exceptfds->fd_count = k;
3708 total += k;
3709 }
3710 return total;
3711 }
3712
3713
3714 /***********************************************************************
3715 * select (WS2_32.18)
3716 */
3717 int WINAPI WS_select(int nfds, WS_fd_set *ws_readfds,
3718 WS_fd_set *ws_writefds, WS_fd_set *ws_exceptfds,
3719 const struct WS_timeval* ws_timeout)
3720 {
3721 struct pollfd *pollfds;
3722 struct timeval tv1, tv2;
3723 int torig = 0;
3724 int count, ret, timeout = -1;
3725
3726 TRACE("read %p, write %p, excp %p timeout %p\n",
3727 ws_readfds, ws_writefds, ws_exceptfds, ws_timeout);
3728
3729 if (!(pollfds = fd_sets_to_poll( ws_readfds, ws_writefds, ws_exceptfds, &count )))
3730 return SOCKET_ERROR;
3731
3732 if (ws_timeout)
3733 {
3734 torig = (ws_timeout->tv_sec * 1000) + (ws_timeout->tv_usec + 999) / 1000;
3735 timeout = torig;
3736 gettimeofday( &tv1, 0 );
3737 }
3738
3739 while ((ret = poll( pollfds, count, timeout )) < 0)
3740 {
3741 if (errno == EINTR)
3742 {
3743 if (!ws_timeout) continue;
3744 gettimeofday( &tv2, 0 );
3745
3746 tv2.tv_sec -= tv1.tv_sec;
3747 tv2.tv_usec -= tv1.tv_usec;
3748 if (tv2.tv_usec < 0)
3749 {
3750 tv2.tv_usec += 1000000;
3751 tv2.tv_sec -= 1;
3752 }
3753
3754 timeout = torig - (tv2.tv_sec * 1000) - (tv2.tv_usec + 999) / 1000;
3755 if (timeout <= 0) break;
3756 } else break;
3757 }
3758 release_poll_fds( ws_readfds, ws_writefds, ws_exceptfds, pollfds );
3759
3760 if (ret == -1) SetLastError(wsaErrno());
3761 else ret = get_poll_results( ws_readfds, ws_writefds, ws_exceptfds, pollfds );
3762 HeapFree( GetProcessHeap(), 0, pollfds );
3763 return ret;
3764 }
3765
3766 /* helper to send completion messages for client-only i/o operation case */
3767 static void WS_AddCompletion( SOCKET sock, ULONG_PTR CompletionValue, NTSTATUS CompletionStatus,
3768 ULONG Information )
3769 {
3770 SERVER_START_REQ( add_fd_completion )
3771 {
3772 req->handle = wine_server_obj_handle( SOCKET2HANDLE(sock) );
3773 req->cvalue = CompletionValue;
3774 req->status = CompletionStatus;
3775 req->information = Information;
3776 wine_server_call( req );
3777 }
3778 SERVER_END_REQ;
3779 }
3780
3781
3782 /***********************************************************************
3783 * send (WS2_32.19)
3784 */
3785 int WINAPI WS_send(SOCKET s, const char *buf, int len, int flags)
3786 {
3787 DWORD n;
3788 WSABUF wsabuf;
3789
3790 wsabuf.len = len;
3791 wsabuf.buf = (char*) buf;
3792
3793 if ( WS2_sendto( s, &wsabuf, 1, &n, flags, NULL, 0, NULL, NULL) == SOCKET_ERROR )
3794 return SOCKET_ERROR;
3795 else
3796 return n;
3797 }
3798
3799 /***********************************************************************
3800 * WSASend (WS2_32.72)
3801 */
3802 INT WINAPI WSASend( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
3803 LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
3804 LPWSAOVERLAPPED lpOverlapped,
3805 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
3806 {
3807 return WS2_sendto( s, lpBuffers, dwBufferCount, lpNumberOfBytesSent, dwFlags,
3808 NULL, 0, lpOverlapped, lpCompletionRoutine );
3809 }
3810
3811 /***********************************************************************
3812 * WSASendDisconnect (WS2_32.73)
3813 */
3814 INT WINAPI WSASendDisconnect( SOCKET s, LPWSABUF lpBuffers )
3815 {
3816 return WS_shutdown( s, SD_SEND );
3817 }
3818
3819
3820 static int WS2_sendto( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
3821 LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
3822 const struct WS_sockaddr *to, int tolen,
3823 LPWSAOVERLAPPED lpOverlapped,
3824 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
3825 {
3826 unsigned int i, options;
3827 int n, fd, err;
3828 struct ws2_async *wsa = NULL;
3829 int totalLength = 0;
3830 ULONG_PTR cvalue = (lpOverlapped && ((ULONG_PTR)lpOverlapped->hEvent & 1) == 0) ? (ULONG_PTR)lpOverlapped : 0;
3831 DWORD bytes_sent;
3832
3833 TRACE("socket %04lx, wsabuf %p, nbufs %d, flags %d, to %p, tolen %d, ovl %p, func %p\n",
3834 s, lpBuffers, dwBufferCount, dwFlags,
3835 to, tolen, lpOverlapped, lpCompletionRoutine);
3836
3837 fd = get_sock_fd( s, FILE_WRITE_DATA, &options );
3838 TRACE( "fd=%d, options=%x\n", fd, options );
3839
3840 if ( fd == -1 ) return SOCKET_ERROR;
3841
3842 if (!lpOverlapped && !lpNumberOfBytesSent)
3843 {
3844 err = WSAEFAULT;
3845 goto error;
3846 }
3847 if (!(wsa = HeapAlloc( GetProcessHeap(), 0, FIELD_OFFSET(struct ws2_async, iovec[dwBufferCount]) )))
3848 {
3849 err = WSAEFAULT;
3850 goto error;
3851 }
3852
3853 wsa->hSocket = SOCKET2HANDLE(s);
3854 wsa->addr = (struct WS_sockaddr *)to;
3855 wsa->addrlen.val = tolen;
3856 wsa->flags = dwFlags;
3857 wsa->lpFlags = &wsa->flags;
3858 wsa->control = NULL;
3859 wsa->n_iovecs = dwBufferCount;
3860 wsa->first_iovec = 0;
3861 for ( i = 0; i < dwBufferCount; i++ )
3862 {
3863 wsa->iovec[i].iov_base = lpBuffers[i].buf;
3864 wsa->iovec[i].iov_len = lpBuffers[i].len;
3865 totalLength += lpBuffers[i].len;
3866 }
3867
3868 for (;;)
3869 {
3870 n = WS2_send( fd, wsa );
3871 if (n != -1 || errno != EINTR) break;
3872 }
3873 if (n == -1 && errno != EAGAIN)
3874 {
3875 int loc_errno = errno;
3876 err = wsaErrno();
3877 if (cvalue) WS_AddCompletion( s, cvalue, sock_get_ntstatus(loc_errno), 0 );
3878 goto error;
3879 }
3880
3881 if ((lpOverlapped || lpCompletionRoutine) &&
3882 !(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)))
3883 {
3884 IO_STATUS_BLOCK *iosb = lpOverlapped ? (IO_STATUS_BLOCK *)lpOverlapped : &wsa->local_iosb;
3885
3886 wsa->user_overlapped = lpOverlapped;
3887 wsa->completion_func = lpCompletionRoutine;
3888 release_sock_fd( s, fd );
3889
3890 if (n == -1 || n < totalLength)
3891 {
3892 iosb->u.Status = STATUS_PENDING;
3893 iosb->Information = n == -1 ? 0 : n;
3894
3895 SERVER_START_REQ( register_async )
3896 {
3897 req->type = ASYNC_TYPE_WRITE;
3898 req->async.handle = wine_server_obj_handle( wsa->hSocket );
3899 req->async.callback = wine_server_client_ptr( WS2_async_send );
3900 req->async.iosb = wine_server_client_ptr( iosb );
3901 req->async.arg = wine_server_client_ptr( wsa );
3902 req->async.event = wine_server_obj_handle( lpCompletionRoutine ? 0 : lpOverlapped->hEvent );
3903 req->async.cvalue = cvalue;
3904 err = wine_server_call( req );
3905 }
3906 SERVER_END_REQ;
3907
3908 /* Enable the event only after starting the async. The server will deliver it as soon as
3909 the async is done. */
3910 _enable_event(SOCKET2HANDLE(s), FD_WRITE, 0, 0);
3911
3912 if (err != STATUS_PENDING) HeapFree( GetProcessHeap(), 0, wsa );
3913 WSASetLastError( NtStatusToWSAError( err ));
3914 return SOCKET_ERROR;
3915 }
3916
3917 iosb->u.Status = STATUS_SUCCESS;
3918 iosb->Information = n;
3919 if (lpNumberOfBytesSent) *lpNumberOfBytesSent = n;
3920 if (!wsa->completion_func)
3921 {
3922 if (cvalue) WS_AddCompletion( s, cvalue, STATUS_SUCCESS, n );
3923 if (lpOverlapped->hEvent) SetEvent( lpOverlapped->hEvent );
3924 HeapFree( GetProcessHeap(), 0, wsa );
3925 }
3926 else NtQueueApcThread( GetCurrentThread(), (PNTAPCFUNC)ws2_async_apc,
3927 (ULONG_PTR)wsa, (ULONG_PTR)iosb, 0 );
3928 WSASetLastError(0);
3929 return 0;
3930 }
3931
3932 if ( _is_blocking(s) )
3933 {
3934 /* On a blocking non-overlapped stream socket,
3935 * sending blocks until the entire buffer is sent. */
3936 DWORD timeout_start = GetTickCount();
3937
3938 bytes_sent = n == -1 ? 0 : n;
3939
3940 while (wsa->first_iovec < wsa->n_iovecs)
3941 {
3942 struct pollfd pfd;
3943 int timeout = GET_SNDTIMEO(fd);
3944
3945 if (timeout != -1)
3946 {
3947 timeout -= GetTickCount() - timeout_start;
3948 if (timeout < 0) timeout = 0;
3949 }
3950
3951 pfd.fd = fd;
3952 pfd.events = POLLOUT;
3953
3954 if (!timeout || !poll( &pfd, 1, timeout ))
3955 {
3956 err = WSAETIMEDOUT;
3957 goto error; /* msdn says a timeout in send is fatal */
3958 }
3959
3960 n = WS2_send( fd, wsa );
3961 if (n == -1 && errno != EAGAIN && errno != EINTR)
3962 {
3963 err = wsaErrno();
3964 goto error;
3965 }
3966
3967 if (n >= 0)
3968 bytes_sent += n;
3969 }
3970 }
3971 else /* non-blocking */
3972 {
3973 if (n < totalLength)
3974 _enable_event(SOCKET2HANDLE(s), FD_WRITE, 0, 0);
3975 if (n == -1)
3976 {
3977 err = WSAEWOULDBLOCK;
3978 goto error;
3979 }
3980 bytes_sent = n;
3981 }
3982
3983 TRACE(" -> %i bytes\n", bytes_sent);
3984
3985 if (lpNumberOfBytesSent) *lpNumberOfBytesSent = bytes_sent;
3986 HeapFree( GetProcessHeap(), 0, wsa );
3987 release_sock_fd( s, fd );
3988 WSASetLastError(0);
3989 return 0;
3990
3991 error:
3992 HeapFree( GetProcessHeap(), 0, wsa );
3993 release_sock_fd( s, fd );
3994 WARN(" -> ERROR %d\n", err);
3995 WSASetLastError(err);
3996 return SOCKET_ERROR;
3997 }
3998
3999 /***********************************************************************
4000 * WSASendTo (WS2_32.74)
4001 */
4002 INT WINAPI WSASendTo( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
4003 LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
4004 const struct WS_sockaddr *to, int tolen,
4005 LPWSAOVERLAPPED lpOverlapped,
4006 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
4007 {
4008 return WS2_sendto( s, lpBuffers, dwBufferCount,
4009 lpNumberOfBytesSent, dwFlags,
4010 to, tolen,
4011 lpOverlapped, lpCompletionRoutine );
4012 }
4013
4014 /***********************************************************************
4015 * sendto (WS2_32.20)
4016 */
4017 int WINAPI WS_sendto(SOCKET s, const char *buf, int len, int flags,
4018 const struct WS_sockaddr *to, int tolen)
4019 {
4020 DWORD n;
4021 WSABUF wsabuf;
4022
4023 wsabuf.len = len;
4024 wsabuf.buf = (char*) buf;
4025
4026 if ( WS2_sendto(s, &wsabuf, 1, &n, flags, to, tolen, NULL, NULL) == SOCKET_ERROR )
4027 return SOCKET_ERROR;
4028 else
4029 return n;
4030 }
4031
4032 /***********************************************************************
4033 * setsockopt (WS2_32.21)
4034 */
4035 int WINAPI WS_setsockopt(SOCKET s, int level, int optname,
4036 const char *optval, int optlen)
4037 {
4038 int fd;
4039 int woptval;
4040 struct linger linger;
4041 struct timeval tval;
4042
4043 TRACE("socket: %04lx, level 0x%x, name 0x%x, ptr %p, len %d\n",
4044 s, level, optname, optval, optlen);
4045
4046 /* some broken apps pass the value directly instead of a pointer to it */
4047 if(optlen && IS_INTRESOURCE(optval))
4048 {
4049 SetLastError(WSAEFAULT);
4050 return SOCKET_ERROR;
4051 }
4052
4053 switch(level)
4054 {
4055 case WS_SOL_SOCKET:
4056 switch(optname)
4057 {
4058 /* Some options need some conversion before they can be sent to
4059 * setsockopt. The conversions are done here, then they will fall though
4060 * to the general case. Special options that are not passed to
4061 * setsockopt follow below that.*/
4062
4063 case WS_SO_DONTLINGER:
4064 if (!optval)
4065 {
4066 SetLastError(WSAEFAULT);
4067 return SOCKET_ERROR;
4068 }
4069 linger.l_onoff = *((const int*)optval) ? 0: 1;
4070 linger.l_linger = 0;
4071 level = SOL_SOCKET;
4072 optname = SO_LINGER;
4073 optval = (char*)&linger;
4074 optlen = sizeof(struct linger);
4075 break;
4076
4077 case WS_SO_LINGER:
4078 if (!optval)
4079 {
4080 SetLastError(WSAEFAULT);
4081 return SOCKET_ERROR;
4082 }
4083 linger.l_onoff = ((LINGER*)optval)->l_onoff;
4084 linger.l_linger = ((LINGER*)optval)->l_linger;
4085 level = SOL_SOCKET;
4086 optname = SO_LINGER;
4087 optval = (char*)&linger;
4088 optlen = sizeof(struct linger);
4089 break;
4090
4091 case WS_SO_RCVBUF:
4092 if (*(const int*)optval < 2048)
4093 {
4094 WARN("SO_RCVBF for %d bytes is too small: ignored\n", *(const int*)optval );
4095 return 0;
4096 }
4097 /* Fall through */
4098
4099 /* The options listed here don't need any special handling. Thanks to
4100 * the conversion happening above, options from there will fall through
4101 * to this, too.*/
4102 case WS_SO_ACCEPTCONN:
4103 case WS_SO_BROADCAST:
4104 case WS_SO_ERROR:
4105 case WS_SO_KEEPALIVE:
4106 case WS_SO_OOBINLINE:
4107 /* BSD socket SO_REUSEADDR is not 100% compatible to winsock semantics.
4108 * however, using it the BSD way fixes bug 8513 and seems to be what
4109 * most programmers assume, anyway */
4110 case WS_SO_REUSEADDR:
4111 case WS_SO_SNDBUF:
4112 case WS_SO_TYPE:
4113 convert_sockopt(&level, &optname);
4114 break;
4115
4116 /* SO_DEBUG is a privileged operation, ignore it. */
4117 case WS_SO_DEBUG:
4118 TRACE("Ignoring SO_DEBUG\n");
4119 return 0;
4120
4121 /* For some reason the game GrandPrixLegends does set SO_DONTROUTE on its
4122 * socket. According to MSDN, this option is silently ignored.*/
4123 case WS_SO_DONTROUTE:
4124 TRACE("Ignoring SO_DONTROUTE\n");
4125 return 0;
4126
4127 /* Stops two sockets from being bound to the same port. Always happens
4128 * on unix systems, so just drop it. */
4129 case WS_SO_EXCLUSIVEADDRUSE:
4130 TRACE("Ignoring SO_EXCLUSIVEADDRUSE, is always set.\n");
4131 return 0;
4132
4133 /* After a ConnectEx call succeeds, the socket can't be used with half of the
4134 * normal winsock functions on windows. We don't have that problem. */
4135 case WS_SO_UPDATE_CONNECT_CONTEXT:
4136 TRACE("Ignoring SO_UPDATE_CONNECT_CONTEXT, since our sockets are normal\n");
4137 return 0;
4138
4139 /* After a AcceptEx call succeeds, the socket can't be used with half of the
4140 * normal winsock functions on windows. We don't have that problem. */
4141 case WS_SO_UPDATE_ACCEPT_CONTEXT:
4142 TRACE("Ignoring SO_UPDATE_ACCEPT_CONTEXT, since our sockets are normal\n");
4143 return 0;
4144
4145 /* SO_OPENTYPE does not require a valid socket handle. */
4146 case WS_SO_OPENTYPE:
4147 if (!optlen || optlen < sizeof(int) || !optval)
4148 {
4149 SetLastError(WSAEFAULT);
4150 return SOCKET_ERROR;
4151 }
4152 get_per_thread_data()->opentype = *(const int *)optval;
4153 TRACE("setting global SO_OPENTYPE = 0x%x\n", *((const int*)optval) );
4154 return 0;
4155
4156 #ifdef SO_RCVTIMEO
4157 case WS_SO_RCVTIMEO:
4158 #endif
4159 #ifdef SO_SNDTIMEO
4160 case WS_SO_SNDTIMEO:
4161 #endif
4162 #if defined(SO_RCVTIMEO) || defined(SO_SNDTIMEO)
4163 if (optval && optlen == sizeof(UINT32)) {
4164 /* WinSock passes milliseconds instead of struct timeval */
4165 tval.tv_usec = (*(const UINT32*)optval % 1000) * 1000;
4166 tval.tv_sec = *(const UINT32*)optval / 1000;
4167 /* min of 500 milliseconds */
4168 if (tval.tv_sec == 0 && tval.tv_usec && tval.tv_usec < 500000)
4169 tval.tv_usec = 500000;
4170 optlen = sizeof(struct timeval);
4171 optval = (char*)&tval;
4172 } else if (optlen == sizeof(struct timeval)) {
4173 WARN("SO_SND/RCVTIMEO for %d bytes: assuming unixism\n", optlen);
4174 } else {
4175 WARN("SO_SND/RCVTIMEO for %d bytes is weird: ignored\n", optlen);
4176 return 0;
4177 }
4178 convert_sockopt(&level, &optname);
4179 break;
4180 #endif
4181
4182 default:
4183 TRACE("Unknown SOL_SOCKET optname: 0x%08x\n", optname);
4184 SetLastError(WSAENOPROTOOPT);
4185 return SOCKET_ERROR;
4186 }
4187 break; /* case WS_SOL_SOCKET */
4188
4189 #ifdef HAVE_IPX
4190 case NSPROTO_IPX:
4191 switch(optname)
4192 {
4193 case IPX_PTYPE:
4194 fd = get_sock_fd( s, 0, NULL );
4195 TRACE("trying to set IPX_PTYPE: %d (fd: %d)\n", *(const int*)optval, fd);
4196
4197 /* We try to set the ipx type on ipx socket level. */
4198 #ifdef SOL_IPX
4199 if(setsockopt(fd, SOL_IPX, IPX_TYPE, optval, optlen) == -1)
4200 {
4201 ERR("IPX: could not set ipx option type; expect weird behaviour\n");
4202 release_sock_fd( s, fd );
4203 return SOCKET_ERROR;
4204 }
4205 #else
4206 {
4207 struct ipx val;
4208 /* Should we retrieve val using a getsockopt call and then
4209 * set the modified one? */
4210 val.ipx_pt = *optval;
4211 setsockopt(fd, 0, SO_DEFAULT_HEADERS, &val, sizeof(struct ipx));
4212 }
4213 #endif
4214 release_sock_fd( s, fd );
4215 return 0;
4216
4217 case IPX_FILTERPTYPE:
4218 /* Sets the receive filter packet type, at the moment we don't support it */
4219 FIXME("IPX_FILTERPTYPE: %x\n", *optval);
4220 /* Returning 0 is better for now than returning a SOCKET_ERROR */
4221 return 0;
4222
4223 default:
4224 FIXME("opt_name:%x\n", optname);
4225 return SOCKET_ERROR;
4226 }
4227 break; /* case NSPROTO_IPX */
4228 #endif
4229
4230 /* Levels WS_IPPROTO_TCP and WS_IPPROTO_IP convert directly */
4231 case WS_IPPROTO_TCP:
4232 switch(optname)
4233 {
4234 case WS_TCP_NODELAY:
4235 convert_sockopt(&level, &optname);
4236 break;
4237 default:
4238 FIXME("Unknown IPPROTO_TCP optname 0x%08x\n", optname);
4239 return SOCKET_ERROR;
4240 }
4241 break;
4242
4243 case WS_IPPROTO_IP:
4244 switch(optname)
4245 {
4246 case WS_IP_ADD_MEMBERSHIP:
4247 case WS_IP_DROP_MEMBERSHIP:
4248 #ifdef IP_HDRINCL
4249 case WS_IP_HDRINCL:
4250 #endif
4251 case WS_IP_MULTICAST_IF:
4252 case WS_IP_MULTICAST_LOOP:
4253 case WS_IP_MULTICAST_TTL:
4254 case WS_IP_OPTIONS:
4255 #ifdef IP_PKTINFO
4256 case WS_IP_PKTINFO:
4257 #endif
4258 case WS_IP_TOS:
4259 case WS_IP_TTL:
4260 convert_sockopt(&level, &optname);
4261 break;
4262 case WS_IP_DONTFRAGMENT:
4263 FIXME("IP_DONTFRAGMENT is silently ignored!\n");
4264 return 0;
4265 default:
4266 FIXME("Unknown IPPROTO_IP optname 0x%08x\n", optname);
4267 return SOCKET_ERROR;
4268 }
4269 break;
4270
4271 case WS_IPPROTO_IPV6:
4272 switch(optname)
4273 {
4274 #ifdef IPV6_ADD_MEMBERSHIP
4275 case WS_IPV6_ADD_MEMBERSHIP:
4276 #endif
4277 #ifdef IPV6_DROP_MEMBERSHIP
4278 case WS_IPV6_DROP_MEMBERSHIP:
4279 #endif
4280 case WS_IPV6_MULTICAST_IF:
4281 case WS_IPV6_MULTICAST_HOPS:
4282 case WS_IPV6_MULTICAST_LOOP:
4283 case WS_IPV6_UNICAST_HOPS:
4284 case WS_IPV6_V6ONLY:
4285 convert_sockopt(&level, &optname);
4286 break;
4287 case WS_IPV6_DONTFRAG:
4288 FIXME("IPV6_DONTFRAG is silently ignored!\n");
4289 return 0;
4290 default:
4291 FIXME("Unknown IPPROTO_IPV6 optname 0x%08x\n", optname);
4292 return SOCKET_ERROR;
4293 }
4294 break;
4295
4296 default:
4297 WARN("Unknown level: 0x%08x\n", level);
4298 SetLastError(WSAEINVAL);
4299 return SOCKET_ERROR;
4300 } /* end switch(level) */
4301
4302 /* avoid endianness issues if argument is a 16-bit int */
4303 if (optval && optlen < sizeof(int))
4304 {
4305 woptval= *((const INT16 *) optval);
4306 optval= (char*) &woptval;
4307 optlen=sizeof(int);
4308 }
4309 fd = get_sock_fd( s, 0, NULL );
4310 if (fd == -1) return SOCKET_ERROR;
4311
4312 if (setsockopt(fd, level, optname, optval, optlen) == 0)
4313 {
4314 release_sock_fd( s, fd );
4315 return 0;
4316 }
4317 TRACE("Setting socket error, %d\n", wsaErrno());
4318 SetLastError(wsaErrno());
4319 release_sock_fd( s, fd );
4320
4321 return SOCKET_ERROR;
4322 }
4323
4324 /***********************************************************************
4325 * shutdown (WS2_32.22)
4326 */
4327 int WINAPI WS_shutdown(SOCKET s, int how)
4328 {
4329 int fd, err = WSAENOTSOCK;
4330 unsigned int options, clear_flags = 0;
4331
4332 fd = get_sock_fd( s, 0, &options );
4333 TRACE("socket %04lx, how %i %x\n", s, how, options );
4334
4335 if (fd == -1)
4336 return SOCKET_ERROR;
4337
4338 switch( how )
4339 {
4340 case 0: /* drop receives */
4341 clear_flags |= FD_READ;
4342 break;
4343 case 1: /* drop sends */
4344 clear_flags |= FD_WRITE;
4345 break;
4346 case 2: /* drop all */
4347 clear_flags |= FD_READ|FD_WRITE;
4348 default:
4349 clear_flags |= FD_WINE_LISTENING;
4350 }
4351
4352 if (!(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)))
4353 {
4354 switch ( how )
4355 {
4356 case SD_RECEIVE:
4357 err = WS2_register_async_shutdown( s, ASYNC_TYPE_READ );
4358 break;
4359 case SD_SEND:
4360 err = WS2_register_async_shutdown( s, ASYNC_TYPE_WRITE );
4361 break;
4362 case SD_BOTH:
4363 default:
4364 err = WS2_register_async_shutdown( s, ASYNC_TYPE_READ );
4365 if (!err) err = WS2_register_async_shutdown( s, ASYNC_TYPE_WRITE );
4366 break;
4367 }
4368 if (err) goto error;
4369 }
4370 else /* non-overlapped mode */
4371 {
4372 if ( shutdown( fd, how ) )
4373 {
4374 err = wsaErrno();
4375 goto error;
4376 }
4377 }
4378
4379 release_sock_fd( s, fd );
4380 _enable_event( SOCKET2HANDLE(s), 0, 0, clear_flags );
4381 if ( how > 1) WSAAsyncSelect( s, 0, 0, 0 );
4382 return 0;
4383
4384 error:
4385 release_sock_fd( s, fd );
4386 _enable_event( SOCKET2HANDLE(s), 0, 0, clear_flags );
4387 WSASetLastError( err );
4388 return SOCKET_ERROR;
4389 }
4390
4391 /***********************************************************************
4392 * socket (WS2_32.23)
4393 */
4394 SOCKET WINAPI WS_socket(int af, int type, int protocol)
4395 {
4396 TRACE("af=%d type=%d protocol=%d\n", af, type, protocol);
4397
4398 return WSASocketA( af, type, protocol, NULL, 0,
4399 get_per_thread_data()->opentype ? 0 : WSA_FLAG_OVERLAPPED );
4400 }
4401
4402
4403 /***********************************************************************
4404 * gethostbyaddr (WS2_32.51)
4405 */
4406 struct WS_hostent* WINAPI WS_gethostbyaddr(const char *addr, int len, int type)
4407 {
4408 struct WS_hostent *retval = NULL;
4409 struct hostent* host;
4410
4411 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
4412 char *extrabuf;
4413 int ebufsize=1024;
4414 struct hostent hostentry;
4415 int locerr=ENOBUFS;
4416 host = NULL;
4417 extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ;
4418 while(extrabuf) {
4419 int res = gethostbyaddr_r(addr, len, type,
4420 &hostentry, extrabuf, ebufsize, &host, &locerr);
4421 if( res != ERANGE) break;
4422 ebufsize *=2;
4423 extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ;
4424 }
4425 if (!host) SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr));
4426 #else
4427 EnterCriticalSection( &csWSgetXXXbyYYY );
4428 host = gethostbyaddr(addr, len, type);
4429 if (!host) SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno));
4430 #endif
4431 if( host != NULL ) retval = WS_dup_he(host);
4432 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
4433 HeapFree(GetProcessHeap(),0,extrabuf);
4434 #else
4435 LeaveCriticalSection( &csWSgetXXXbyYYY );
4436 #endif
4437 TRACE("ptr %p, len %d, type %d ret %p\n", addr, len, type, retval);
4438 return retval;
4439 }
4440
4441 /***********************************************************************
4442 * WS_get_local_ips (INTERNAL)
4443 *
4444 * Returns the list of local IP addresses by going through the network
4445 * adapters and using the local routing table to sort the addresses
4446 * from highest routing priority to lowest routing priority. This
4447 * functionality is inferred from the description for obtaining local
4448 * IP addresses given in the Knowledge Base Article Q160215.
4449 *
4450 * Please note that the returned hostent is only freed when the thread
4451 * closes and is replaced if another hostent is requested.
4452 */
4453 static struct WS_hostent* WS_get_local_ips( char *hostname )
4454 {
4455 int last_metric, numroutes = 0, i, j;
4456 PIP_ADAPTER_INFO adapters = NULL, k;
4457 struct WS_hostent *hostlist = NULL;
4458 PMIB_IPFORWARDTABLE routes = NULL;
4459 struct route *route_addrs = NULL;
4460 DWORD adap_size, route_size;
4461
4462 /* Obtain the size of the adapter list and routing table, also allocate memory */
4463 if (GetAdaptersInfo(NULL, &adap_size) != ERROR_BUFFER_OVERFLOW)
4464 return NULL;
4465 if (GetIpForwardTable(NULL, &route_size, FALSE) != ERROR_INSUFFICIENT_BUFFER)
4466 return NULL;
4467 adapters = HeapAlloc(GetProcessHeap(), 0, adap_size);
4468 routes = HeapAlloc(GetProcessHeap(), 0, route_size);
4469 route_addrs = HeapAlloc(GetProcessHeap(), 0, 0); /* HeapReAlloc doesn't work on NULL */
4470 if (adapters == NULL || routes == NULL || route_addrs == NULL)
4471 goto cleanup;
4472 /* Obtain the adapter list and the full routing table */
4473 if (GetAdaptersInfo(adapters, &adap_size) != NO_ERROR)
4474 goto cleanup;
4475 if (GetIpForwardTable(routes, &route_size, FALSE) != NO_ERROR)
4476 goto cleanup;
4477 /* Store the interface associated with each route */
4478 for (i = 0; i < routes->dwNumEntries; i++)
4479 {
4480 DWORD ifindex, ifmetric, exists = FALSE;
4481
4482 if (routes->table[i].dwForwardType != MIB_IPROUTE_TYPE_DIRECT)
4483 continue;
4484 ifindex = routes->table[i].dwForwardIfIndex;
4485 ifmetric = routes->table[i].dwForwardMetric1;
4486 /* Only store the lowest valued metric for an interface */
4487 for (j = 0; j < numroutes; j++)
4488 {
4489 if (route_addrs[j].interface == ifindex)
4490 {
4491 if (route_addrs[j].metric > ifmetric)
4492 route_addrs[j].metric = ifmetric;
4493 exists = TRUE;
4494 }
4495 }
4496 if (exists)
4497 continue;
4498 route_addrs = HeapReAlloc(GetProcessHeap(), 0, route_addrs, (numroutes+1)*sizeof(struct route));
4499 if (route_addrs == NULL)
4500 goto cleanup; /* Memory allocation error, fail gracefully */
4501 route_addrs[numroutes].interface = ifindex;
4502 route_addrs[numroutes].metric = ifmetric;
4503 /* If no IP is found in the next step (for whatever reason)
4504 * then fall back to the magic loopback address.
4505 */
4506 memcpy(&(route_addrs[numroutes].addr.s_addr), magic_loopback_addr, 4);
4507 numroutes++;
4508 }
4509 if (numroutes == 0)
4510 goto cleanup; /* No routes, fall back to the Magic IP */
4511 /* Find the IP address associated with each found interface */
4512 for (i = 0; i < numroutes; i++)
4513 {
4514 for (k = adapters; k != NULL; k = k->Next)
4515 {
4516 char *ip = k->IpAddressList.IpAddress.String;
4517
4518 if (route_addrs[i].interface == k->Index)
4519 route_addrs[i].addr.s_addr = (in_addr_t) inet_addr(ip);
4520 }
4521 }
4522 /* Allocate a hostent and enough memory for all the IPs,
4523 * including the NULL at the end of the list.
4524 */
4525 hostlist = WS_create_he(hostname, 1, numroutes+1, TRUE);
4526 if (hostlist == NULL)
4527 goto cleanup; /* Failed to allocate a hostent for the list of IPs */
4528 hostlist->h_addr_list[numroutes] = NULL; /* NULL-terminate the address list */
4529 hostlist->h_aliases[0] = NULL; /* NULL-terminate the alias list */
4530 hostlist->h_addrtype = AF_INET;
4531 hostlist->h_length = sizeof(struct in_addr); /* = 4 */
4532 /* Reorder the entries when placing them in the host list, Windows expects
4533 * the IP list in order from highest priority to lowest (the critical thing
4534 * is that most applications expect the first IP to be the default route).
4535 */
4536 last_metric = -1;
4537 for (i = 0; i < numroutes; i++)
4538 {
4539 struct in_addr addr;
4540 int metric = 0xFFFF;
4541
4542 memcpy(&addr, magic_loopback_addr, 4);
4543 for (j = 0; j < numroutes; j++)
4544 {
4545 int this_metric = route_addrs[j].metric;
4546
4547 if (this_metric > last_metric && this_metric < metric)
4548 {
4549 addr = route_addrs[j].addr;
4550 metric = this_metric;
4551 }
4552 }
4553 last_metric = metric;
4554 (*(struct in_addr *) hostlist->h_addr_list[i]) = addr;
4555 }
4556
4557 /* Cleanup all allocated memory except the address list,
4558 * the address list is used by the calling app.
4559 */
4560 cleanup:
4561 HeapFree(GetProcessHeap(), 0, route_addrs);
4562 HeapFree(GetProcessHeap(), 0, adapters);
4563 HeapFree(GetProcessHeap(), 0, routes);
4564 return hostlist;
4565 }
4566
4567 /***********************************************************************
4568 * gethostbyname (WS2_32.52)
4569 */
4570 struct WS_hostent* WINAPI WS_gethostbyname(const char* name)
4571 {
4572 struct WS_hostent *retval = NULL;
4573 struct hostent* host;
4574 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
4575 char *extrabuf;
4576 int ebufsize=1024;
4577 struct hostent hostentry;
4578 int locerr = ENOBUFS;
4579 #endif
4580 char hostname[100];
4581 if(!num_startup) {
4582 SetLastError(WSANOTINITIALISED);
4583 return NULL;
4584 }
4585 if( gethostname( hostname, 100) == -1) {
4586 SetLastError( WSAENOBUFS); /* appropriate ? */
4587 return retval;
4588 }
4589 if( !name || !name[0]) {
4590 name = hostname;
4591 }
4592 /* If the hostname of the local machine is requested then return the
4593 * complete list of local IP addresses */
4594 if(strcmp(name, hostname) == 0)
4595 retval = WS_get_local_ips(hostname);
4596 /* If any other hostname was requested (or the routing table lookup failed)
4597 * then return the IP found by the host OS */
4598 if(retval == NULL)
4599 {
4600 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
4601 host = NULL;
4602 extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ;
4603 while(extrabuf) {
4604 int res = gethostbyname_r(name, &hostentry, extrabuf, ebufsize, &host, &locerr);
4605 if( res != ERANGE) break;
4606 ebufsize *=2;
4607 extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ;
4608 }
4609 if (!host) SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr));
4610 #else
4611 EnterCriticalSection( &csWSgetXXXbyYYY );
4612 host = gethostbyname(name);
4613 if (!host) SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno));
4614 #endif
4615 if (host) retval = WS_dup_he(host);
4616 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
4617 HeapFree(GetProcessHeap(),0,extrabuf);
4618 #else
4619 LeaveCriticalSection( &csWSgetXXXbyYYY );
4620 #endif
4621 }
4622 if (retval && retval->h_addr_list[0][0] == 127 &&
4623 strcmp(name, "localhost") != 0)
4624 {
4625 /* hostname != "localhost" but has loopback address. replace by our
4626 * special address.*/
4627 memcpy(retval->h_addr_list[0], magic_loopback_addr, 4);
4628 }
4629 TRACE( "%s ret %p\n", debugstr_a(name), retval );
4630 return retval;
4631 }
4632
4633
4634 /***********************************************************************
4635 * getprotobyname (WS2_32.53)
4636 */
4637 struct WS_protoent* WINAPI WS_getprotobyname(const char* name)
4638 {
4639 struct WS_protoent* retval = NULL;
4640 #ifdef HAVE_GETPROTOBYNAME
4641 struct protoent* proto;
4642 EnterCriticalSection( &csWSgetXXXbyYYY );
4643 if( (proto = getprotobyname(name)) != NULL )
4644 {
4645 retval = WS_dup_pe(proto);
4646 }
4647 else {
4648 MESSAGE("protocol %s not found; You might want to add "
4649 "this to /etc/protocols\n", debugstr_a(name) );
4650 SetLastError(WSANO_DATA);
4651 }
4652 LeaveCriticalSection( &csWSgetXXXbyYYY );
4653 #endif
4654 TRACE( "%s ret %p\n", debugstr_a(name), retval );
4655 return retval;
4656 }
4657
4658
4659 /***********************************************************************
4660 * getprotobynumber (WS2_32.54)
4661 */
4662 struct WS_protoent* WINAPI WS_getprotobynumber(int number)
4663 {
4664 struct WS_protoent* retval = NULL;
4665 #ifdef HAVE_GETPROTOBYNUMBER
4666 struct protoent* proto;
4667 EnterCriticalSection( &csWSgetXXXbyYYY );
4668 if( (proto = getprotobynumber(number)) != NULL )
4669 {
4670 retval = WS_dup_pe(proto);
4671 }
4672 else {
4673 MESSAGE("protocol number %d not found; You might want to add "
4674 "this to /etc/protocols\n", number );
4675 SetLastError(WSANO_DATA);
4676 }
4677 LeaveCriticalSection( &csWSgetXXXbyYYY );
4678 #endif
4679 TRACE("%i ret %p\n", number, retval);
4680 return retval;
4681 }
4682
4683
4684 /***********************************************************************
4685 * getservbyname (WS2_32.55)
4686 */
4687 struct WS_servent* WINAPI WS_getservbyname(const char *name, const char *proto)
4688 {
4689 struct WS_servent* retval = NULL;
4690 struct servent* serv;
4691 char *name_str;
4692 char *proto_str = NULL;
4693
4694 if (!(name_str = strdup_lower(name))) return NULL;
4695
4696 if (proto && *proto)
4697 {
4698 if (!(proto_str = strdup_lower(proto)))
4699 {
4700 HeapFree( GetProcessHeap(), 0, name_str );
4701 return NULL;
4702 }
4703 }
4704
4705 EnterCriticalSection( &csWSgetXXXbyYYY );
4706 serv = getservbyname(name_str, proto_str);
4707 if( serv != NULL )
4708 {
4709 retval = WS_dup_se(serv);
4710 }
4711 else SetLastError(WSANO_DATA);
4712 LeaveCriticalSection( &csWSgetXXXbyYYY );
4713 HeapFree( GetProcessHeap(), 0, proto_str );
4714 HeapFree( GetProcessHeap(), 0, name_str );
4715 TRACE( "%s, %s ret %p\n", debugstr_a(name), debugstr_a(proto), retval );
4716 return retval;
4717 }
4718
4719 /***********************************************************************
4720 * freeaddrinfo (WS2_32.@)
4721 */
4722 void WINAPI WS_freeaddrinfo(struct WS_addrinfo *res)
4723 {
4724 while (res) {
4725 struct WS_addrinfo *next;
4726
4727 HeapFree(GetProcessHeap(),0,res->ai_canonname);
4728 HeapFree(GetProcessHeap(),0,res->ai_addr);
4729 next = res->ai_next;
4730 HeapFree(GetProcessHeap(),0,res);
4731 res = next;
4732 }
4733 }
4734
4735 /* helper functions for getaddrinfo()/getnameinfo() */
4736 static int convert_aiflag_w2u(int winflags) {
4737 unsigned int i;
4738 int unixflags = 0;
4739
4740 for (i=0;i<sizeof(ws_aiflag_map)/sizeof(ws_aiflag_map[0]);i++)
4741 if (ws_aiflag_map[i][0] & winflags) {
4742 unixflags |= ws_aiflag_map[i][1];
4743 winflags &= ~ws_aiflag_map[i][0];
4744 }
4745 if (winflags)
4746 FIXME("Unhandled windows AI_xxx flags %x\n", winflags);
4747 return unixflags;
4748 }
4749
4750 static int convert_niflag_w2u(int winflags) {
4751 unsigned int i;
4752 int unixflags = 0;
4753
4754 for (i=0;i<sizeof(ws_niflag_map)/sizeof(ws_niflag_map[0]);i++)
4755 if (ws_niflag_map[i][0] & winflags) {
4756 unixflags |= ws_niflag_map[i][1];
4757 winflags &= ~ws_niflag_map[i][0];
4758 }
4759 if (winflags)
4760 FIXME("Unhandled windows NI_xxx flags %x\n", winflags);
4761 return unixflags;
4762 }
4763
4764 static int convert_aiflag_u2w(int unixflags) {
4765 unsigned int i;
4766 int winflags = 0;
4767
4768 for (i=0;i<sizeof(ws_aiflag_map)/sizeof(ws_aiflag_map[0]);i++)
4769 if (ws_aiflag_map[i][1] & unixflags) {
4770 winflags |= ws_aiflag_map[i][0];
4771 unixflags &= ~ws_aiflag_map[i][1];
4772 }
4773 if (unixflags) /* will warn usually */
4774 WARN("Unhandled UNIX AI_xxx flags %x\n", unixflags);
4775 return winflags;
4776 }
4777
4778 static int convert_eai_u2w(int unixret) {
4779 int i;
4780
4781 for (i=0;ws_eai_map[i][0];i++)
4782 if (ws_eai_map[i][1] == unixret)
4783 return ws_eai_map[i][0];
4784 return unixret;
4785 }
4786
4787 /***********************************************************************
4788 * getaddrinfo (WS2_32.@)
4789 */
4790 int WINAPI WS_getaddrinfo(LPCSTR nodename, LPCSTR servname, const struct WS_addrinfo *hints, struct WS_addrinfo **res)
4791 {
4792 #ifdef HAVE_GETADDRINFO
4793 struct addrinfo *unixaires = NULL;
4794 int result;
4795 struct addrinfo unixhints, *punixhints = NULL;
4796 CHAR *node = NULL, *serv = NULL;
4797
4798 if (nodename)
4799 if (!(node = strdup_lower(nodename))) return WSA_NOT_ENOUGH_MEMORY;
4800
4801 if (servname) {
4802 if (!(serv = strdup_lower(servname))) {
4803 HeapFree(GetProcessHeap(), 0, node);
4804 return WSA_NOT_ENOUGH_MEMORY;
4805 }
4806 }
4807
4808 if (hints) {
4809 punixhints = &unixhints;
4810
4811 memset(&unixhints, 0, sizeof(unixhints));
4812 punixhints->ai_flags = convert_aiflag_w2u(hints->ai_flags);
4813 if (hints->ai_family == 0) /* wildcard, specific to getaddrinfo() */
4814 punixhints->ai_family = 0;
4815 else
4816 punixhints->ai_family = convert_af_w2u(hints->ai_family);
4817 if (hints->ai_socktype == 0) /* wildcard, specific to getaddrinfo() */
4818 punixhints->ai_socktype = 0;
4819 else
4820 punixhints->ai_socktype = convert_socktype_w2u(hints->ai_socktype);
4821 if (hints->ai_protocol == 0) /* wildcard, specific to getaddrinfo() */
4822 punixhints->ai_protocol = 0;
4823 else
4824 punixhints->ai_protocol = convert_proto_w2u(hints->ai_protocol);
4825 }
4826
4827 /* getaddrinfo(3) is thread safe, no need to wrap in CS */
4828 result = getaddrinfo(nodename, servname, punixhints, &unixaires);
4829
4830 TRACE("%s, %s %p -> %p %d\n", debugstr_a(nodename), debugstr_a(servname), hints, res, result);
4831
4832 HeapFree(GetProcessHeap(), 0, node);
4833 HeapFree(GetProcessHeap(), 0, serv);
4834
4835 if (!result) {
4836 struct addrinfo *xuai = unixaires;
4837 struct WS_addrinfo **xai = res;
4838
4839 *xai = NULL;
4840 while (xuai) {
4841 struct WS_addrinfo *ai = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY, sizeof(struct WS_addrinfo));
4842 int len;
4843
4844 if (!ai)
4845 goto outofmem;
4846
4847 *xai = ai;xai = &ai->ai_next;
4848 ai->ai_flags = convert_aiflag_u2w(xuai->ai_flags);
4849 ai->ai_family = convert_af_u2w(xuai->ai_family);
4850 ai->ai_socktype = convert_socktype_u2w(xuai->ai_socktype);
4851 ai->ai_protocol = convert_proto_u2w(xuai->ai_protocol);
4852 if (xuai->ai_canonname) {
4853 TRACE("canon name - %s\n",debugstr_a(xuai->ai_canonname));
4854 ai->ai_canonname = HeapAlloc(GetProcessHeap(),0,strlen(xuai->ai_canonname)+1);
4855 if (!ai->ai_canonname)
4856 goto outofmem;
4857 strcpy(ai->ai_canonname,xuai->ai_canonname);
4858 }
4859 len = xuai->ai_addrlen;
4860 ai->ai_addr = HeapAlloc(GetProcessHeap(),0,len);
4861 if (!ai->ai_addr)
4862 goto outofmem;
4863 ai->ai_addrlen = len;
4864 do {
4865 int winlen = ai->ai_addrlen;
4866
4867 if (!ws_sockaddr_u2ws(xuai->ai_addr, ai->ai_addr, &winlen)) {
4868 ai->ai_addrlen = winlen;
4869 break;
4870 }
4871 len = 2*len;
4872 ai->ai_addr = HeapReAlloc(GetProcessHeap(),0,ai->ai_addr,len);
4873 if (!ai->ai_addr)
4874 goto outofmem;
4875 ai->ai_addrlen = len;
4876 } while (1);
4877 xuai = xuai->ai_next;
4878 }
4879 freeaddrinfo(unixaires);
4880 } else {
4881 result = convert_eai_u2w(result);
4882 *res = NULL;
4883 }
4884 return result;
4885
4886 outofmem:
4887 if (*res) WS_freeaddrinfo(*res);
4888 if (unixaires) freeaddrinfo(unixaires);
4889 *res = NULL;
4890 return WSA_NOT_ENOUGH_MEMORY;
4891 #else
4892 FIXME("getaddrinfo() failed, not found during buildtime.\n");
4893 return EAI_FAIL;
4894 #endif
4895 }
4896
4897 static struct WS_addrinfoW *addrinfo_AtoW(const struct WS_addrinfo *ai)
4898 {
4899 struct WS_addrinfoW *ret;
4900
4901 if (!(ret = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_addrinfoW)))) return NULL;
4902 ret->ai_flags = ai->ai_flags;
4903 ret->ai_family = ai->ai_family;
4904 ret->ai_socktype = ai->ai_socktype;
4905 ret->ai_protocol = ai->ai_protocol;
4906 ret->ai_addrlen = ai->ai_addrlen;
4907 ret->ai_canonname = NULL;
4908 ret->ai_addr = NULL;
4909 ret->ai_next = NULL;
4910 if (ai->ai_canonname)
4911 {
4912 int len = MultiByteToWideChar(CP_ACP, 0, ai->ai_canonname, -1, NULL, 0);
4913 if (!(ret->ai_canonname = HeapAlloc(GetProcessHeap(), 0, len)))
4914 {
4915 HeapFree(GetProcessHeap(), 0, ret);
4916 return NULL;
4917 }
4918 MultiByteToWideChar(CP_ACP, 0, ai->ai_canonname, -1, ret->ai_canonname, len);
4919 }
4920 if (ai->ai_addr)
4921 {
4922 if (!(ret->ai_addr = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_sockaddr))))
4923 {
4924 HeapFree(GetProcessHeap(), 0, ret->ai_canonname);
4925 HeapFree(GetProcessHeap(), 0, ret);
4926 return NULL;
4927 }
4928 memcpy(ret->ai_addr, ai->ai_addr, sizeof(struct WS_sockaddr));
4929 }
4930 return ret;
4931 }
4932
4933 static struct WS_addrinfoW *addrinfo_list_AtoW(const struct WS_addrinfo *info)
4934 {
4935 struct WS_addrinfoW *ret, *infoW;
4936
4937 if (!(ret = infoW = addrinfo_AtoW(info))) return NULL;
4938 while (info->ai_next)
4939 {
4940 if (!(infoW->ai_next = addrinfo_AtoW(info->ai_next)))
4941 {
4942 FreeAddrInfoW(ret);
4943 return NULL;
4944 }
4945 infoW = infoW->ai_next;
4946 info = info->ai_next;
4947 }
4948 return ret;
4949 }
4950
4951 static struct WS_addrinfo *addrinfo_WtoA(const struct WS_addrinfoW *ai)
4952 {
4953 struct WS_addrinfo *ret;
4954
4955 if (!(ret = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_addrinfo)))) return NULL;
4956 ret->ai_flags = ai->ai_flags;
4957 ret->ai_family = ai->ai_family;
4958 ret->ai_socktype = ai->ai_socktype;
4959 ret->ai_protocol = ai->ai_protocol;
4960 ret->ai_addrlen = ai->ai_addrlen;
4961 ret->ai_canonname = NULL;
4962 ret->ai_addr = NULL;
4963 ret->ai_next = NULL;
4964 if (ai->ai_canonname)
4965 {
4966 int len = WideCharToMultiByte(CP_ACP, 0, ai->ai_canonname, -1, NULL, 0, NULL, NULL);
4967 if (!(ret->ai_canonname = HeapAlloc(GetProcessHeap(), 0, len)))
4968 {
4969 HeapFree(GetProcessHeap(), 0, ret);
4970 return NULL;
4971 }
4972 WideCharToMultiByte(CP_ACP, 0, ai->ai_canonname, -1, ret->ai_canonname, len, NULL, NULL);
4973 }
4974 if (ai->ai_addr)
4975 {
4976 if (!(ret->ai_addr = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_sockaddr))))
4977 {
4978 HeapFree(GetProcessHeap(), 0, ret->ai_canonname);
4979 HeapFree(GetProcessHeap(), 0, ret);
4980 return NULL;
4981 }
4982 memcpy(ret->ai_addr, ai->ai_addr, sizeof(struct WS_sockaddr));
4983 }
4984 return ret;
4985 }
4986
4987 /***********************************************************************
4988 * GetAddrInfoW (WS2_32.@)
4989 */
4990 int WINAPI GetAddrInfoW(LPCWSTR nodename, LPCWSTR servname, const ADDRINFOW *hints, PADDRINFOW *res)
4991 {
4992 int ret, len;
4993 char *nodenameA, *servnameA = NULL;
4994 struct WS_addrinfo *resA, *hintsA = NULL;
4995
4996 if (!nodename) return WSAHOST_NOT_FOUND;
4997
4998 len = WideCharToMultiByte(CP_ACP, 0, nodename, -1, NULL, 0, NULL, NULL);
4999 if (!(nodenameA = HeapAlloc(GetProcessHeap(), 0, len))) return EAI_MEMORY;
5000 WideCharToMultiByte(CP_ACP, 0, nodename, -1, nodenameA, len, NULL, NULL);
5001
5002 if (servname)
5003 {
5004 len = WideCharToMultiByte(CP_ACP, 0, servname, -1, NULL, 0, NULL, NULL);
5005 if (!(servnameA = HeapAlloc(GetProcessHeap(), 0, len)))
5006 {
5007 HeapFree(GetProcessHeap(), 0, nodenameA);
5008 return EAI_MEMORY;
5009 }
5010 WideCharToMultiByte(CP_ACP, 0, servname, -1, servnameA, len, NULL, NULL);
5011 }
5012
5013 if (hints) hintsA = addrinfo_WtoA(hints);
5014 ret = WS_getaddrinfo(nodenameA, servnameA, hintsA, &resA);
5015 WS_freeaddrinfo(hintsA);
5016
5017 if (!ret)
5018 {
5019 *res = addrinfo_list_AtoW(resA);
5020 WS_freeaddrinfo(resA);
5021 }
5022
5023 HeapFree(GetProcessHeap(), 0, nodenameA);
5024 HeapFree(GetProcessHeap(), 0, servnameA);
5025 return ret;
5026 }
5027
5028 /***********************************************************************
5029 * FreeAddrInfoW (WS2_32.@)
5030 */
5031 void WINAPI FreeAddrInfoW(PADDRINFOW ai)
5032 {
5033 while (ai)
5034 {
5035 ADDRINFOW *next;
5036 HeapFree(GetProcessHeap(), 0, ai->ai_canonname);
5037 HeapFree(GetProcessHeap(), 0, ai->ai_addr);
5038 next = ai->ai_next;
5039 HeapFree(GetProcessHeap(), 0, ai);
5040 ai = next;
5041 }
5042 }
5043
5044 int WINAPI WS_getnameinfo(const SOCKADDR *sa, WS_socklen_t salen, PCHAR host,
5045 DWORD hostlen, PCHAR serv, DWORD servlen, INT flags)
5046 {
5047 #ifdef HAVE_GETNAMEINFO
5048 int ret;
5049 union generic_unix_sockaddr sa_u;
5050 unsigned int size;
5051
5052 TRACE("%s %d %p %d %p %d %d\n", debugstr_sockaddr(sa), salen, host, hostlen,
5053 serv, servlen, flags);
5054
5055 size = ws_sockaddr_ws2u(sa, salen, &sa_u);
5056 if (!size)
5057 {
5058 WSASetLastError(WSAEFAULT);
5059 return WSA_NOT_ENOUGH_MEMORY;
5060 }
5061 ret = getnameinfo(&sa_u.addr, size, host, hostlen, serv, servlen, convert_niflag_w2u(flags));
5062 return convert_eai_u2w(ret);
5063 #else
5064 FIXME("getnameinfo() failed, not found during buildtime.\n");
5065 return EAI_FAIL;
5066 #endif
5067 }
5068
5069 int WINAPI GetNameInfoW(const SOCKADDR *sa, WS_socklen_t salen, PWCHAR host,
5070 DWORD hostlen, PWCHAR serv, DWORD servlen, INT flags)
5071 {
5072 int ret;
5073 char *hostA = NULL, *servA = NULL;
5074
5075 if (host && (!(hostA = HeapAlloc(GetProcessHeap(), 0, hostlen)))) return EAI_MEMORY;
5076 if (serv && (!(servA = HeapAlloc(GetProcessHeap(), 0, servlen))))
5077 {
5078 HeapFree(GetProcessHeap(), 0, hostA);
5079 return EAI_MEMORY;
5080 }
5081
5082 ret = WS_getnameinfo(sa, salen, hostA, hostlen, servA, servlen, flags);
5083 if (!ret)
5084 {
5085 if (host) MultiByteToWideChar(CP_ACP, 0, hostA, -1, host, hostlen);
5086 if (serv) MultiByteToWideChar(CP_ACP, 0, servA, -1, serv, servlen);
5087 }
5088
5089 HeapFree(GetProcessHeap(), 0, hostA);
5090 HeapFree(GetProcessHeap(), 0, servA);
5091 return ret;
5092 }
5093
5094 /***********************************************************************
5095 * getservbyport (WS2_32.56)
5096 */
5097 struct WS_servent* WINAPI WS_getservbyport(int port, const char *proto)
5098 {
5099 struct WS_servent* retval = NULL;
5100 #ifdef HAVE_GETSERVBYPORT
5101 struct servent* serv;
5102 char *proto_str = NULL;
5103
5104 if (proto && *proto)
5105 {
5106 if (!(proto_str = strdup_lower(proto))) return NULL;
5107 }
5108 EnterCriticalSection( &csWSgetXXXbyYYY );
5109 if( (serv = getservbyport(port, proto_str)) != NULL ) {
5110 retval = WS_dup_se(serv);
5111 }
5112 else SetLastError(WSANO_DATA);
5113 LeaveCriticalSection( &csWSgetXXXbyYYY );
5114 HeapFree( GetProcessHeap(), 0, proto_str );
5115 #endif
5116 TRACE("%d (i.e. port %d), %s ret %p\n", port, (int)ntohl(port), debugstr_a(proto), retval);
5117 return retval;
5118 }
5119
5120
5121 /***********************************************************************
5122 * gethostname (WS2_32.57)
5123 */
5124 int WINAPI WS_gethostname(char *name, int namelen)
5125 {
5126 TRACE("name %p, len %d\n", name, namelen);
5127
5128 if (gethostname(name, namelen) == 0)
5129 {
5130 TRACE("<- '%s'\n", name);
5131 return 0;
5132 }
5133 SetLastError((errno == EINVAL) ? WSAEFAULT : wsaErrno());
5134 TRACE("<- ERROR !\n");
5135 return SOCKET_ERROR;
5136 }
5137
5138
5139 /* ------------------------------------- Windows sockets extensions -- *
5140 * *
5141 * ------------------------------------------------------------------- */
5142
5143 /***********************************************************************
5144 * WSAEnumNetworkEvents (WS2_32.36)
5145 */
5146 int WINAPI WSAEnumNetworkEvents(SOCKET s, WSAEVENT hEvent, LPWSANETWORKEVENTS lpEvent)
5147 {
5148 int ret;
5149 int i;
5150 int errors[FD_MAX_EVENTS];
5151
5152 TRACE("%08lx, hEvent %p, lpEvent %p\n", s, hEvent, lpEvent );
5153
5154 SERVER_START_REQ( get_socket_event )
5155 {
5156 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
5157 req->service = TRUE;
5158 req->c_event = wine_server_obj_handle( hEvent );
5159 wine_server_set_reply( req, errors, sizeof(errors) );
5160 if (!(ret = wine_server_call(req))) lpEvent->lNetworkEvents = reply->pmask & reply->mask;
5161 }
5162 SERVER_END_REQ;
5163 if (!ret)
5164 {
5165 for (i = 0; i < FD_MAX_EVENTS; i++)
5166 lpEvent->iErrorCode[i] = NtStatusToWSAError(errors[i]);
5167 return 0;
5168 }
5169 SetLastError(WSAEINVAL);
5170 return SOCKET_ERROR;
5171 }
5172
5173 /***********************************************************************
5174 * WSAEventSelect (WS2_32.39)
5175 */
5176 int WINAPI WSAEventSelect(SOCKET s, WSAEVENT hEvent, LONG lEvent)
5177 {
5178 int ret;
5179
5180 TRACE("%08lx, hEvent %p, event %08x\n", s, hEvent, lEvent);
5181
5182 SERVER_START_REQ( set_socket_event )
5183 {
5184 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
5185 req->mask = lEvent;
5186 req->event = wine_server_obj_handle( hEvent );
5187 req->window = 0;
5188 req->msg = 0;
5189 ret = wine_server_call( req );
5190 }
5191 SERVER_END_REQ;
5192 if (!ret) return 0;
5193 SetLastError(WSAEINVAL);
5194 return SOCKET_ERROR;
5195 }
5196
5197 /**********************************************************************
5198 * WSAGetOverlappedResult (WS2_32.40)
5199 */
5200 BOOL WINAPI WSAGetOverlappedResult( SOCKET s, LPWSAOVERLAPPED lpOverlapped,
5201 LPDWORD lpcbTransfer, BOOL fWait,
5202 LPDWORD lpdwFlags )
5203 {
5204 NTSTATUS status;
5205
5206 TRACE( "socket %04lx ovl %p trans %p, wait %d flags %p\n",
5207 s, lpOverlapped, lpcbTransfer, fWait, lpdwFlags );
5208
5209 if ( lpOverlapped == NULL )
5210 {
5211 ERR( "Invalid pointer\n" );
5212 WSASetLastError(WSA_INVALID_PARAMETER);
5213 return FALSE;
5214 }
5215
5216 status = lpOverlapped->Internal;
5217 if (status == STATUS_PENDING)
5218 {
5219 if (!fWait)
5220 {
5221 SetLastError( WSA_IO_INCOMPLETE );
5222 return FALSE;
5223 }
5224
5225 if (WaitForSingleObject( lpOverlapped->hEvent ? lpOverlapped->hEvent : SOCKET2HANDLE(s),
5226 INFINITE ) == WAIT_FAILED)
5227 return FALSE;
5228 status = lpOverlapped->Internal;
5229 }
5230
5231 if ( lpcbTransfer )
5232 *lpcbTransfer = lpOverlapped->InternalHigh;
5233
5234 if ( lpdwFlags )
5235 *lpdwFlags = lpOverlapped->u.s.Offset;
5236
5237 if (status) SetLastError( RtlNtStatusToDosError(status) );
5238 return !status;
5239 }
5240
5241
5242 /***********************************************************************
5243 * WSAAsyncSelect (WS2_32.101)
5244 */
5245 INT WINAPI WSAAsyncSelect(SOCKET s, HWND hWnd, UINT uMsg, LONG lEvent)
5246 {
5247 int ret;
5248
5249 TRACE("%lx, hWnd %p, uMsg %08x, event %08x\n", s, hWnd, uMsg, lEvent);
5250
5251 SERVER_START_REQ( set_socket_event )
5252 {
5253 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
5254 req->mask = lEvent;
5255 req->event = 0;
5256 req->window = wine_server_user_handle( hWnd );
5257 req->msg = uMsg;
5258 ret = wine_server_call( req );
5259 }
5260 SERVER_END_REQ;
5261 if (!ret) return 0;
5262 SetLastError(WSAEINVAL);
5263 return SOCKET_ERROR;
5264 }
5265
5266 /***********************************************************************
5267 * WSACreateEvent (WS2_32.31)
5268 *
5269 */
5270 WSAEVENT WINAPI WSACreateEvent(void)
5271 {
5272 /* Create a manual-reset event, with initial state: unsignaled */
5273 TRACE("\n");
5274
5275 return CreateEventW(NULL, TRUE, FALSE, NULL);
5276 }
5277
5278 /***********************************************************************
5279 * WSACloseEvent (WS2_32.29)
5280 *
5281 */
5282 BOOL WINAPI WSACloseEvent(WSAEVENT event)
5283 {
5284 TRACE ("event=%p\n", event);
5285
5286 return CloseHandle(event);
5287 }
5288
5289 /***********************************************************************
5290 * WSASocketA (WS2_32.78)
5291 *
5292 */
5293 SOCKET WINAPI WSASocketA(int af, int type, int protocol,
5294 LPWSAPROTOCOL_INFOA lpProtocolInfo,
5295 GROUP g, DWORD dwFlags)
5296 {
5297 INT len;
5298 WSAPROTOCOL_INFOW info;
5299
5300 TRACE("af=%d type=%d protocol=%d protocol_info=%p group=%d flags=0x%x\n",
5301 af, type, protocol, lpProtocolInfo, g, dwFlags);
5302
5303 if (!lpProtocolInfo) return WSASocketW(af, type, protocol, NULL, g, dwFlags);
5304
5305 memcpy(&info, lpProtocolInfo, FIELD_OFFSET(WSAPROTOCOL_INFOW, szProtocol));
5306 len = MultiByteToWideChar(CP_ACP, 0, lpProtocolInfo->szProtocol, -1,
5307 info.szProtocol, WSAPROTOCOL_LEN + 1);
5308
5309 if (!len)
5310 {
5311 WSASetLastError( WSAEINVAL);
5312 return SOCKET_ERROR;
5313 }
5314
5315 return WSASocketW(af, type, protocol, &info, g, dwFlags);
5316 }
5317
5318 /***********************************************************************
5319 * WSASocketW (WS2_32.79)
5320 *
5321 */
5322 SOCKET WINAPI WSASocketW(int af, int type, int protocol,
5323 LPWSAPROTOCOL_INFOW lpProtocolInfo,
5324 GROUP g, DWORD dwFlags)
5325 {
5326 SOCKET ret;
5327
5328 /*
5329 FIXME: The "advanced" parameters of WSASocketW (lpProtocolInfo,
5330 g, dwFlags except WSA_FLAG_OVERLAPPED) are ignored.
5331 */
5332
5333 TRACE("af=%d type=%d protocol=%d protocol_info=%p group=%d flags=0x%x\n",
5334 af, type, protocol, lpProtocolInfo, g, dwFlags );
5335
5336 /* hack for WSADuplicateSocket */
5337 if (lpProtocolInfo && lpProtocolInfo->dwServiceFlags4 == 0xff00ff00) {
5338 ret = lpProtocolInfo->dwCatalogEntryId;
5339 TRACE("\tgot duplicate %04lx\n", ret);
5340 return ret;
5341 }
5342
5343 /* convert the socket family and type */
5344 af = convert_af_w2u(af);
5345 type = convert_socktype_w2u(type);
5346
5347 if (lpProtocolInfo)
5348 {
5349 if (af == FROM_PROTOCOL_INFO)
5350 af = lpProtocolInfo->iAddressFamily;
5351 if (type == FROM_PROTOCOL_INFO)
5352 type = lpProtocolInfo->iSocketType;
5353 if (protocol == FROM_PROTOCOL_INFO)
5354 protocol = lpProtocolInfo->iProtocol;
5355 }
5356
5357 if ( af == AF_UNSPEC) /* did they not specify the address family? */
5358 switch(protocol)
5359 {
5360 case IPPROTO_TCP:
5361 if (type == SOCK_STREAM) { af = AF_INET; break; }
5362 case IPPROTO_UDP:
5363 if (type == SOCK_DGRAM) { af = AF_INET; break; }
5364 default: SetLastError(WSAEPROTOTYPE); return INVALID_SOCKET;
5365 }
5366
5367 SERVER_START_REQ( create_socket )
5368 {
5369 req->family = af;
5370 req->type = type;
5371 req->protocol = protocol;
5372 req->access = GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE;
5373 req->attributes = OBJ_INHERIT;
5374 req->flags = dwFlags;
5375 set_error( wine_server_call( req ) );
5376 ret = HANDLE2SOCKET( wine_server_ptr_handle( reply->handle ));
5377 }
5378 SERVER_END_REQ;
5379 if (ret)
5380 {
5381 TRACE("\tcreated %04lx\n", ret );
5382 return ret;
5383 }
5384
5385 if (GetLastError() == WSAEACCES) /* raw socket denied */
5386 {
5387 if (type == SOCK_RAW)
5388 ERR_(winediag)("Failed to create a socket of type SOCK_RAW, this requires special permissions.\n");
5389 else
5390 ERR_(winediag)("Failed to create socket, this requires special permissions.\n");
5391 SetLastError(WSAESOCKTNOSUPPORT);
5392 }
5393
5394 WARN("\t\tfailed!\n");
5395 return INVALID_SOCKET;
5396 }
5397
5398 /***********************************************************************
5399 * WSAJoinLeaf (WS2_32.58)
5400 *
5401 */
5402 SOCKET WINAPI WSAJoinLeaf(
5403 SOCKET s,
5404 const struct WS_sockaddr *addr,
5405 int addrlen,
5406 LPWSABUF lpCallerData,
5407 LPWSABUF lpCalleeData,
5408 LPQOS lpSQOS,
5409 LPQOS lpGQOS,
5410 DWORD dwFlags)
5411 {
5412 FIXME("stub.\n");
5413 return INVALID_SOCKET;
5414 }
5415
5416 /***********************************************************************
5417 * __WSAFDIsSet (WS2_32.151)
5418 */
5419 int WINAPI __WSAFDIsSet(SOCKET s, WS_fd_set *set)
5420 {
5421 int i = set->fd_count;
5422
5423 TRACE("(%ld,%p(%i))\n", s, set, i);
5424
5425 while (i--)
5426 if (set->fd_array[i] == s) return 1;
5427 return 0;
5428 }
5429
5430 /***********************************************************************
5431 * WSAIsBlocking (WS2_32.114)
5432 */
5433 BOOL WINAPI WSAIsBlocking(void)
5434 {
5435 /* By default WinSock should set all its sockets to non-blocking mode
5436 * and poll in PeekMessage loop when processing "blocking" ones. This
5437 * function is supposed to tell if the program is in this loop. Our
5438 * blocking calls are truly blocking so we always return FALSE.
5439 *
5440 * Note: It is allowed to call this function without prior WSAStartup().
5441 */
5442
5443 TRACE("\n");
5444 return FALSE;
5445 }
5446
5447 /***********************************************************************
5448 * WSACancelBlockingCall (WS2_32.113)
5449 */
5450 INT WINAPI WSACancelBlockingCall(void)
5451 {
5452 TRACE("\n");
5453 return 0;
5454 }
5455
5456 static INT WINAPI WSA_DefaultBlockingHook( FARPROC x )
5457 {
5458 FIXME("How was this called?\n");
5459 return x();
5460 }
5461
5462
5463 /***********************************************************************
5464 * WSASetBlockingHook (WS2_32.109)
5465 */
5466 FARPROC WINAPI WSASetBlockingHook(FARPROC lpBlockFunc)
5467 {
5468 FARPROC prev = blocking_hook;
5469 blocking_hook = lpBlockFunc;
5470 TRACE("hook %p\n", lpBlockFunc);
5471 return prev;
5472 }
5473
5474
5475 /***********************************************************************
5476 * WSAUnhookBlockingHook (WS2_32.110)
5477 */
5478 INT WINAPI WSAUnhookBlockingHook(void)
5479 {
5480 blocking_hook = (FARPROC)WSA_DefaultBlockingHook;
5481 return 0;
5482 }
5483
5484
5485 /* ----------------------------------- end of API stuff */
5486
5487 /* ----------------------------------- helper functions -
5488 *
5489 * TODO: Merge WS_dup_..() stuff into one function that
5490 * would operate with a generic structure containing internal
5491 * pointers (via a template of some kind).
5492 */
5493
5494 static int list_size(char** l, int item_size)
5495 {
5496 int i,j = 0;
5497 if(l)
5498 { for(i=0;l[i];i++)
5499 j += (item_size) ? item_size : strlen(l[i]) + 1;
5500 j += (i + 1) * sizeof(char*); }
5501 return j;
5502 }
5503
5504 static int list_dup(char** l_src, char** l_to, int item_size)
5505 {
5506 char *p;
5507 int i;
5508
5509 for (i = 0; l_src[i]; i++) ;
5510 p = (char *)(l_to + i + 1);
5511 for (i = 0; l_src[i]; i++)
5512 {
5513 int count = ( item_size ) ? item_size : strlen(l_src[i]) + 1;
5514 memcpy(p, l_src[i], count);
5515 l_to[i] = p;
5516 p += count;
5517 }
5518 l_to[i] = NULL;
5519 return p - (char *)l_to;
5520 }
5521
5522 /* ----- hostent */
5523
5524 /* create a hostent entry
5525 *
5526 * Creates the entry with enough memory for the name, aliases
5527 * addresses, and the address pointers. Also copies the name
5528 * and sets up all the pointers. If "fill_addresses" is set then
5529 * sufficient memory for the addresses is also allocated and the
5530 * address pointers are set to this memory.
5531 *
5532 * NOTE: The alias and address lists must be allocated with room
5533 * for the NULL item terminating the list. This is true even if
5534 * the list has no items ("aliases" and "addresses" must be
5535 * at least "1", a truly empty list is invalid).
5536 */
5537 static struct WS_hostent *WS_create_he(char *name, int aliases, int addresses, int fill_addresses)
5538 {
5539 struct WS_hostent *p_to;
5540 char *p;
5541
5542 int size = (sizeof(struct WS_hostent) +
5543 strlen(name) + 1 +
5544 sizeof(char *)*aliases +
5545 sizeof(char *)*addresses);
5546
5547 /* Allocate enough memory for the addresses */
5548 if (fill_addresses)
5549 size += sizeof(struct in_addr)*addresses;
5550
5551 if (!(p_to = check_buffer_he(size))) return NULL;
5552 memset(p_to, 0, size);
5553
5554 p = (char *)(p_to + 1);
5555 p_to->h_name = p;
5556 strcpy(p, name);
5557 p += strlen(p) + 1;
5558
5559 p_to->h_aliases = (char **)p;
5560 p += sizeof(char *)*aliases;
5561 p_to->h_addr_list = (char **)p;
5562 p += sizeof(char *)*addresses;
5563 if (fill_addresses)
5564 {
5565 int i;
5566
5567 /* NOTE: h_aliases must be filled in manually, leave these
5568 * pointers NULL (already set to NULL by memset earlier).
5569 */
5570
5571 /* Fill in the list of address pointers */
5572 for (i = 0; i < addresses; i++)
5573 p_to->h_addr_list[i] = (p += sizeof(struct in_addr));
5574 p += sizeof(struct in_addr);
5575 }
5576 return p_to;
5577 }
5578
5579 /* duplicate hostent entry
5580 * and handle all Win16/Win32 dependent things (struct size, ...) *correctly*.
5581 * Ditto for protoent and servent.
5582 */
5583 static struct WS_hostent *WS_dup_he(const struct hostent* p_he)
5584 {
5585 int addresses = list_size(p_he->h_addr_list, p_he->h_length);
5586 int aliases = list_size(p_he->h_aliases, 0);
5587 struct WS_hostent *p_to;
5588
5589 p_to = WS_create_he(p_he->h_name, aliases, addresses, FALSE);
5590
5591 if (!p_to) return NULL;
5592 p_to->h_addrtype = p_he->h_addrtype;
5593 p_to->h_length = p_he->h_length;
5594
5595 list_dup(p_he->h_aliases, p_to->h_aliases, 0);
5596 list_dup(p_he->h_addr_list, p_to->h_addr_list, p_he->h_length);
5597 return p_to;
5598 }
5599
5600 /* ----- protoent */
5601
5602 static struct WS_protoent *WS_dup_pe(const struct protoent* p_pe)
5603 {
5604 char *p;
5605 struct WS_protoent *p_to;
5606
5607 int size = (sizeof(*p_pe) +
5608 strlen(p_pe->p_name) + 1 +
5609 list_size(p_pe->p_aliases, 0));
5610
5611 if (!(p_to = check_buffer_pe(size))) return NULL;
5612 p_to->p_proto = p_pe->p_proto;
5613
5614 p = (char *)(p_to + 1);
5615 p_to->p_name = p;
5616 strcpy(p, p_pe->p_name);
5617 p += strlen(p) + 1;
5618
5619 p_to->p_aliases = (char **)p;
5620 list_dup(p_pe->p_aliases, p_to->p_aliases, 0);
5621 return p_to;
5622 }
5623
5624 /* ----- servent */
5625
5626 static struct WS_servent *WS_dup_se(const struct servent* p_se)
5627 {
5628 char *p;
5629 struct WS_servent *p_to;
5630
5631 int size = (sizeof(*p_se) +
5632 strlen(p_se->s_proto) + 1 +
5633 strlen(p_se->s_name) + 1 +
5634 list_size(p_se->s_aliases, 0));
5635
5636 if (!(p_to = check_buffer_se(size))) return NULL;
5637 p_to->s_port = p_se->s_port;
5638
5639 p = (char *)(p_to + 1);
5640 p_to->s_name = p;
5641 strcpy(p, p_se->s_name);
5642 p += strlen(p) + 1;
5643
5644 p_to->s_proto = p;
5645 strcpy(p, p_se->s_proto);
5646 p += strlen(p) + 1;
5647
5648 p_to->s_aliases = (char **)p;
5649 list_dup(p_se->s_aliases, p_to->s_aliases, 0);
5650 return p_to;
5651 }
5652
5653
5654 /***********************************************************************
5655 * WSARecv (WS2_32.67)
5656 */
5657 int WINAPI WSARecv(SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
5658 LPDWORD NumberOfBytesReceived, LPDWORD lpFlags,
5659 LPWSAOVERLAPPED lpOverlapped,
5660 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine)
5661 {
5662 return WS2_recv_base(s, lpBuffers, dwBufferCount, NumberOfBytesReceived, lpFlags,
5663 NULL, NULL, lpOverlapped, lpCompletionRoutine, NULL);
5664 }
5665
5666 static int WS2_recv_base( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
5667 LPDWORD lpNumberOfBytesRecvd, LPDWORD lpFlags,
5668 struct WS_sockaddr *lpFrom,
5669 LPINT lpFromlen, LPWSAOVERLAPPED lpOverlapped,
5670 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine,
5671 LPWSABUF lpControlBuffer )
5672 {
5673 unsigned int i, options;
5674 int n, fd, err;
5675 struct ws2_async *wsa;
5676 DWORD timeout_start = GetTickCount();
5677 ULONG_PTR cvalue = (lpOverlapped && ((ULONG_PTR)lpOverlapped->hEvent & 1) == 0) ? (ULONG_PTR)lpOverlapped : 0;
5678
5679 TRACE("socket %04lx, wsabuf %p, nbufs %d, flags %d, from %p, fromlen %d, ovl %p, func %p\n",
5680 s, lpBuffers, dwBufferCount, *lpFlags, lpFrom,
5681 (lpFromlen ? *lpFromlen : -1),
5682 lpOverlapped, lpCompletionRoutine);
5683
5684 fd = get_sock_fd( s, FILE_READ_DATA, &options );
5685 TRACE( "fd=%d, options=%x\n", fd, options );
5686
5687 if (fd == -1) return SOCKET_ERROR;
5688
5689 if (!(wsa = HeapAlloc( GetProcessHeap(), 0, FIELD_OFFSET(struct ws2_async, iovec[dwBufferCount]) )))
5690 {
5691 err = WSAEFAULT;
5692 goto error;
5693 }
5694
5695 wsa->hSocket = SOCKET2HANDLE(s);
5696 wsa->flags = *lpFlags;
5697 wsa->lpFlags = lpFlags;
5698 wsa->addr = lpFrom;
5699 wsa->addrlen.ptr = lpFromlen;
5700 wsa->control = lpControlBuffer;
5701 wsa->n_iovecs = dwBufferCount;
5702 wsa->first_iovec = 0;
5703 for (i = 0; i < dwBufferCount; i++)
5704 {
5705 /* check buffer first to trigger write watches */
5706 if (IsBadWritePtr( lpBuffers[i].buf, lpBuffers[i].len ))
5707 {
5708 err = WSAEFAULT;
5709 goto error;
5710 }
5711 wsa->iovec[i].iov_base = lpBuffers[i].buf;
5712 wsa->iovec[i].iov_len = lpBuffers[i].len;
5713 }
5714
5715 for (;;)
5716 {
5717 n = WS2_recv( fd, wsa );
5718 if (n == -1)
5719 {
5720 if (errno == EINTR) continue;
5721 if (errno != EAGAIN)
5722 {
5723 int loc_errno = errno;
5724 err = wsaErrno();
5725 if (cvalue) WS_AddCompletion( s, cvalue, sock_get_ntstatus(loc_errno), 0 );
5726 goto error;
5727 }
5728 }
5729 else if (lpNumberOfBytesRecvd) *lpNumberOfBytesRecvd = n;
5730
5731 if ((lpOverlapped || lpCompletionRoutine) &&
5732 !(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)))
5733 {
5734 IO_STATUS_BLOCK *iosb = lpOverlapped ? (IO_STATUS_BLOCK *)lpOverlapped : &wsa->local_iosb;
5735
5736 wsa->user_overlapped = lpOverlapped;
5737 wsa->completion_func = lpCompletionRoutine;
5738 release_sock_fd( s, fd );
5739
5740 if (n == -1)
5741 {
5742 iosb->u.Status = STATUS_PENDING;
5743 iosb->Information = 0;
5744
5745 SERVER_START_REQ( register_async )
5746 {
5747 req->type = ASYNC_TYPE_READ;
5748 req->async.handle = wine_server_obj_handle( wsa->hSocket );
5749 req->async.callback = wine_server_client_ptr( WS2_async_recv );
5750 req->async.iosb = wine_server_client_ptr( iosb );
5751 req->async.arg = wine_server_client_ptr( wsa );
5752 req->async.event = wine_server_obj_handle( lpCompletionRoutine ? 0 : lpOverlapped->hEvent );
5753 req->async.cvalue = cvalue;
5754 err = wine_server_call( req );
5755 }
5756 SERVER_END_REQ;
5757
5758 if (err != STATUS_PENDING) HeapFree( GetProcessHeap(), 0, wsa );
5759 WSASetLastError( NtStatusToWSAError( err ));
5760 return SOCKET_ERROR;
5761 }
5762
5763 iosb->u.Status = STATUS_SUCCESS;
5764 iosb->Information = n;
5765 if (!wsa->completion_func)
5766 {
5767 if (cvalue) WS_AddCompletion( s, cvalue, STATUS_SUCCESS, n );
5768 if (lpOverlapped->hEvent) SetEvent( lpOverlapped->hEvent );
5769 HeapFree( GetProcessHeap(), 0, wsa );
5770 }
5771 else NtQueueApcThread( GetCurrentThread(), (PNTAPCFUNC)ws2_async_apc,
5772 (ULONG_PTR)wsa, (ULONG_PTR)iosb, 0 );
5773 _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
5774 return 0;
5775 }
5776
5777 if (n != -1) break;
5778
5779 if ( _is_blocking(s) )
5780 {
5781 struct pollfd pfd;
5782 int timeout = GET_RCVTIMEO(fd);
5783 if (timeout != -1)
5784 {
5785 timeout -= GetTickCount() - timeout_start;
5786 if (timeout < 0) timeout = 0;
5787 }
5788
5789 pfd.fd = fd;
5790 pfd.events = POLLIN;
5791 if (*lpFlags & WS_MSG_OOB) pfd.events |= POLLPRI;
5792
5793 if (!timeout || !poll( &pfd, 1, timeout ))
5794 {
5795 err = WSAETIMEDOUT;
5796 /* a timeout is not fatal */
5797 _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
5798 goto error;
5799 }
5800 }
5801 else
5802 {
5803 _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
5804 err = WSAEWOULDBLOCK;
5805 goto error;
5806 }
5807 }
5808
5809 TRACE(" -> %i bytes\n", n);
5810 HeapFree( GetProcessHeap(), 0, wsa );
5811 release_sock_fd( s, fd );
5812 _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
5813
5814 return 0;
5815
5816 error:
5817 HeapFree( GetProcessHeap(), 0, wsa );
5818 release_sock_fd( s, fd );
5819 WARN(" -> ERROR %d\n", err);
5820 WSASetLastError( err );
5821 return SOCKET_ERROR;
5822 }
5823
5824 /***********************************************************************
5825 * WSARecvFrom (WS2_32.69)
5826 */
5827 INT WINAPI WSARecvFrom( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
5828 LPDWORD lpNumberOfBytesRecvd, LPDWORD lpFlags, struct WS_sockaddr *lpFrom,
5829 LPINT lpFromlen, LPWSAOVERLAPPED lpOverlapped,
5830 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
5831
5832 {
5833 return WS2_recv_base( s, lpBuffers, dwBufferCount,
5834 lpNumberOfBytesRecvd, lpFlags,
5835 lpFrom, lpFromlen,
5836 lpOverlapped, lpCompletionRoutine, NULL );
5837 }
5838
5839 /***********************************************************************
5840 * WSCInstallProvider (WS2_32.88)
5841 */
5842 INT WINAPI WSCInstallProvider( const LPGUID lpProviderId,
5843 LPCWSTR lpszProviderDllPath,
5844 const LPWSAPROTOCOL_INFOW lpProtocolInfoList,
5845 DWORD dwNumberOfEntries,
5846 LPINT lpErrno )
5847 {
5848 FIXME("(%s, %s, %p, %d, %p): stub !\n", debugstr_guid(lpProviderId),
5849 debugstr_w(lpszProviderDllPath), lpProtocolInfoList,
5850 dwNumberOfEntries, lpErrno);
5851 *lpErrno = 0;
5852 return 0;
5853 }
5854
5855
5856 /***********************************************************************
5857 * WSCDeinstallProvider (WS2_32.83)
5858 */
5859 INT WINAPI WSCDeinstallProvider(LPGUID lpProviderId, LPINT lpErrno)
5860 {
5861 FIXME("(%s, %p): stub !\n", debugstr_guid(lpProviderId), lpErrno);
5862 *lpErrno = 0;
5863 return 0;
5864 }
5865
5866
5867 /***********************************************************************
5868 * WSAAccept (WS2_32.26)
5869 */
5870 SOCKET WINAPI WSAAccept( SOCKET s, struct WS_sockaddr *addr, LPINT addrlen,
5871 LPCONDITIONPROC lpfnCondition, DWORD_PTR dwCallbackData)
5872 {
5873
5874 int ret = 0, size = 0;
5875 WSABUF CallerId, CallerData, CalleeId, CalleeData;
5876 /* QOS SQOS, GQOS; */
5877 GROUP g;
5878 SOCKET cs;
5879 SOCKADDR src_addr, dst_addr;
5880
5881 TRACE("Socket %04lx, sockaddr %p, addrlen %p, fnCondition %p, dwCallbackData %ld\n",
5882 s, addr, addrlen, lpfnCondition, dwCallbackData);
5883
5884
5885 size = sizeof(src_addr);
5886 cs = WS_accept(s, &src_addr, &size);
5887
5888 if (cs == SOCKET_ERROR) return SOCKET_ERROR;
5889
5890 if (!lpfnCondition) return cs;
5891
5892 CallerId.buf = (char *)&src_addr;
5893 CallerId.len = sizeof(src_addr);
5894
5895 CallerData.buf = NULL;
5896 CallerData.len = 0;
5897
5898 WS_getsockname(cs, &dst_addr, &size);
5899
5900 CalleeId.buf = (char *)&dst_addr;
5901 CalleeId.len = sizeof(dst_addr);
5902
5903
5904 ret = (*lpfnCondition)(&CallerId, &CallerData, NULL, NULL,
5905 &CalleeId, &CalleeData, &g, dwCallbackData);
5906
5907 switch (ret)
5908 {
5909 case CF_ACCEPT:
5910 if (addr && addrlen)
5911 memcpy(addr, &src_addr, (*addrlen > size) ? size : *addrlen );
5912 return cs;
5913 case CF_DEFER:
5914 SERVER_START_REQ( set_socket_deferred )
5915 {
5916 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
5917 req->deferred = wine_server_obj_handle( SOCKET2HANDLE(cs) );
5918 if ( !wine_server_call_err ( req ) )
5919 {
5920 SetLastError( WSATRY_AGAIN );
5921 WS_closesocket( cs );
5922 }
5923 }
5924 SERVER_END_REQ;
5925 return SOCKET_ERROR;
5926 case CF_REJECT:
5927 WS_closesocket(cs);
5928 SetLastError(WSAECONNREFUSED);
5929 return SOCKET_ERROR;
5930 default:
5931 FIXME("Unknown return type from Condition function\n");
5932 SetLastError(WSAENOTSOCK);
5933 return SOCKET_ERROR;
5934 }
5935 }
5936
5937 /***********************************************************************
5938 * WSADuplicateSocketA (WS2_32.32)
5939 */
5940 int WINAPI WSADuplicateSocketA( SOCKET s, DWORD dwProcessId, LPWSAPROTOCOL_INFOA lpProtocolInfo )
5941 {
5942 HANDLE hProcess;
5943
5944 TRACE("(%ld,%x,%p)\n", s, dwProcessId, lpProtocolInfo);
5945 memset(lpProtocolInfo, 0, sizeof(*lpProtocolInfo));
5946 /* FIXME: WS_getsockopt(s, WS_SOL_SOCKET, SO_PROTOCOL_INFO, lpProtocolInfo, sizeof(*lpProtocolInfo)); */
5947 /* I don't know what the real Windoze does next, this is a hack */
5948 /* ...we could duplicate and then use ConvertToGlobalHandle on the duplicate, then let
5949 * the target use the global duplicate, or we could copy a reference to us to the structure
5950 * and let the target duplicate it from us, but let's do it as simple as possible */
5951 hProcess = OpenProcess(PROCESS_DUP_HANDLE, FALSE, dwProcessId);
5952 DuplicateHandle(GetCurrentProcess(), SOCKET2HANDLE(s),
5953 hProcess, (LPHANDLE)&lpProtocolInfo->dwCatalogEntryId,
5954 0, FALSE, DUPLICATE_SAME_ACCESS);
5955 CloseHandle(hProcess);
5956 lpProtocolInfo->dwServiceFlags4 = 0xff00ff00; /* magic */
5957 return 0;
5958 }
5959
5960 /***********************************************************************
5961 * WSADuplicateSocketW (WS2_32.33)
5962 */
5963 int WINAPI WSADuplicateSocketW( SOCKET s, DWORD dwProcessId, LPWSAPROTOCOL_INFOW lpProtocolInfo )
5964 {
5965 HANDLE hProcess;
5966
5967 TRACE("(%ld,%x,%p)\n", s, dwProcessId, lpProtocolInfo);
5968
5969 memset(lpProtocolInfo, 0, sizeof(*lpProtocolInfo));
5970 hProcess = OpenProcess(PROCESS_DUP_HANDLE, FALSE, dwProcessId);
5971 DuplicateHandle(GetCurrentProcess(), SOCKET2HANDLE(s),
5972 hProcess, (LPHANDLE)&lpProtocolInfo->dwCatalogEntryId,
5973 0, FALSE, DUPLICATE_SAME_ACCESS);
5974 CloseHandle(hProcess);
5975 lpProtocolInfo->dwServiceFlags4 = 0xff00ff00; /* magic */
5976 return 0;
5977 }
5978
5979 /***********************************************************************
5980 * WSAInstallServiceClassA (WS2_32.48)
5981 */
5982 int WINAPI WSAInstallServiceClassA(LPWSASERVICECLASSINFOA info)
5983 {
5984 FIXME("Request to install service %s\n",debugstr_a(info->lpszServiceClassName));
5985 WSASetLastError(WSAEACCES);
5986 return SOCKET_ERROR;
5987 }
5988
5989 /***********************************************************************
5990 * WSAInstallServiceClassW (WS2_32.49)
5991 */
5992 int WINAPI WSAInstallServiceClassW(LPWSASERVICECLASSINFOW info)
5993 {
5994 FIXME("Request to install service %s\n",debugstr_w(info->lpszServiceClassName));
5995 WSASetLastError(WSAEACCES);
5996 return SOCKET_ERROR;
5997 }
5998
5999 /***********************************************************************
6000 * WSARemoveServiceClass (WS2_32.70)
6001 */
6002 int WINAPI WSARemoveServiceClass(LPGUID info)
6003 {
6004 FIXME("Request to remove service %p\n",info);
6005 WSASetLastError(WSATYPE_NOT_FOUND);
6006 return SOCKET_ERROR;
6007 }
6008
6009 /***********************************************************************
6010 * inet_ntop (WS2_32.@)
6011 */
6012 PCSTR WINAPI WS_inet_ntop( INT family, PVOID addr, PSTR buffer, SIZE_T len )
6013 {
6014 #ifdef HAVE_INET_NTOP
6015 struct WS_in6_addr *in6;
6016 struct WS_in_addr *in;
6017 PCSTR pdst;
6018
6019 TRACE("family %d, addr (%p), buffer (%p), len %ld\n", family, addr, buffer, len);
6020 if (!buffer)
6021 {
6022 WSASetLastError( STATUS_INVALID_PARAMETER );
6023 return NULL;
6024 }
6025
6026 switch (family)
6027 {
6028 case WS_AF_INET:
6029 {
6030 in = addr;
6031 pdst = inet_ntop( AF_INET, &in->WS_s_addr, buffer, len );
6032 break;
6033 }
6034 case WS_AF_INET6:
6035 {
6036 in6 = addr;
6037 pdst = inet_ntop( AF_INET6, in6->WS_s6_addr, buffer, len );
6038 break;
6039 }
6040 default:
6041 WSASetLastError( WSAEAFNOSUPPORT );
6042 return NULL;
6043 }
6044
6045 if (!pdst) WSASetLastError( STATUS_INVALID_PARAMETER );
6046 return pdst;
6047 #else
6048 FIXME( "not supported on this platform\n" );
6049 WSASetLastError( WSAEAFNOSUPPORT );
6050 return NULL;
6051 #endif
6052 }
6053
6054 /***********************************************************************
6055 * WSAStringToAddressA (WS2_32.80)
6056 */
6057 INT WINAPI WSAStringToAddressA(LPSTR AddressString,
6058 INT AddressFamily,
6059 LPWSAPROTOCOL_INFOA lpProtocolInfo,
6060 LPSOCKADDR lpAddress,
6061 LPINT lpAddressLength)
6062 {
6063 INT res=0;
6064 LPSTR workBuffer=NULL,ptrPort;
6065
6066 TRACE( "(%s, %x, %p, %p, %p)\n", debugstr_a(AddressString), AddressFamily,
6067 lpProtocolInfo, lpAddress, lpAddressLength );
6068
6069 if (!lpAddressLength || !lpAddress) return SOCKET_ERROR;
6070
6071 if (!AddressString)
6072 {
6073 WSASetLastError(WSAEINVAL);
6074 return SOCKET_ERROR;
6075 }
6076
6077 if (lpProtocolInfo)
6078 FIXME("ProtocolInfo not implemented.\n");
6079
6080 workBuffer = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
6081 strlen(AddressString) + 1);
6082 if (!workBuffer)
6083 {
6084 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6085 return SOCKET_ERROR;
6086 }
6087
6088 strcpy(workBuffer, AddressString);
6089
6090 switch(AddressFamily)
6091 {
6092 case WS_AF_INET:
6093 {
6094 struct in_addr inetaddr;
6095
6096 /* If lpAddressLength is too small, tell caller the size we need */
6097 if (*lpAddressLength < sizeof(SOCKADDR_IN))
6098 {
6099 *lpAddressLength = sizeof(SOCKADDR_IN);
6100 res = WSAEFAULT;
6101 break;
6102 }
6103 *lpAddressLength = sizeof(SOCKADDR_IN);
6104 memset(lpAddress, 0, sizeof(SOCKADDR_IN));
6105
6106 ((LPSOCKADDR_IN)lpAddress)->sin_family = AF_INET;
6107
6108 ptrPort = strchr(workBuffer, ':');
6109 if(ptrPort)
6110 {
6111 ((LPSOCKADDR_IN)lpAddress)->sin_port = htons(atoi(ptrPort+1));
6112 *ptrPort = '\0';
6113 }
6114 else
6115 {
6116 ((LPSOCKADDR_IN)lpAddress)->sin_port = 0;
6117 }
6118
6119 if(inet_aton(workBuffer, &inetaddr) > 0)
6120 {
6121 ((LPSOCKADDR_IN)lpAddress)->sin_addr.WS_s_addr = inetaddr.s_addr;
6122 res = 0;
6123 }
6124 else
6125 res = WSAEINVAL;
6126
6127 break;
6128
6129 }
6130 case WS_AF_INET6:
6131 {
6132 struct in6_addr inetaddr;
6133 /* If lpAddressLength is too small, tell caller the size we need */
6134 if (*lpAddressLength < sizeof(SOCKADDR_IN6))
6135 {
6136 *lpAddressLength = sizeof(SOCKADDR_IN6);
6137 res = WSAEFAULT;
6138 break;
6139 }
6140 #ifdef HAVE_INET_PTON
6141 *lpAddressLength = sizeof(SOCKADDR_IN6);
6142 memset(lpAddress, 0, sizeof(SOCKADDR_IN6));
6143
6144 ((LPSOCKADDR_IN6)lpAddress)->sin6_family = WS_AF_INET6;
6145
6146 /* This one is a bit tricky. An IPv6 address contains colons, so the
6147 * check from IPv4 doesn't work like that. However, IPv6 addresses that
6148 * contain a port are written with braces like [fd12:3456:7890::1]:12345
6149 * so what we will do is to look for ']', check if the next char is a
6150 * colon, and if it is, parse the port as in IPv4. */
6151
6152 ptrPort = strchr(workBuffer, ']');
6153 if(ptrPort && *(++ptrPort) == ':')
6154 {
6155 ((LPSOCKADDR_IN6)lpAddress)->sin6_port = htons(atoi(ptrPort+1));
6156 *ptrPort = '\0';
6157 }
6158 else
6159 {
6160 ((LPSOCKADDR_IN6)lpAddress)->sin6_port = 0;
6161 }
6162
6163 if(inet_pton(AF_INET6, workBuffer, &inetaddr) > 0)
6164 {
6165 memcpy(&((LPSOCKADDR_IN6)lpAddress)->sin6_addr, &inetaddr,
6166 sizeof(struct in6_addr));
6167 res = 0;
6168 }
6169 else
6170 #endif /* HAVE_INET_PTON */
6171 res = WSAEINVAL;
6172
6173 break;
6174 }
6175 default:
6176 /* According to MSDN, only AF_INET and AF_INET6 are supported. */
6177 TRACE("Unsupported address family specified: %d.\n", AddressFamily);
6178 res = WSAEINVAL;
6179 }
6180
6181 HeapFree(GetProcessHeap(), 0, workBuffer);
6182
6183 if (!res) return 0;
6184 WSASetLastError(res);
6185 return SOCKET_ERROR;
6186 }
6187
6188 /***********************************************************************
6189 * WSAStringToAddressW (WS2_32.81)
6190 *
6191 * Does anybody know if this function allows to use Hebrew/Arabic/Chinese... digits?
6192 * If this should be the case, it would be required to map these digits
6193 * to Unicode digits (0-9) using FoldString first.
6194 */
6195 INT WINAPI WSAStringToAddressW(LPWSTR AddressString,
6196 INT AddressFamily,
6197 LPWSAPROTOCOL_INFOW lpProtocolInfo,
6198 LPSOCKADDR lpAddress,
6199 LPINT lpAddressLength)
6200 {
6201 INT sBuffer,res=0;
6202 LPSTR workBuffer=NULL;
6203 WSAPROTOCOL_INFOA infoA;
6204 LPWSAPROTOCOL_INFOA lpProtoInfoA = NULL;
6205
6206 TRACE( "(%s, %x, %p, %p, %p)\n", debugstr_w(AddressString), AddressFamily, lpProtocolInfo,
6207 lpAddress, lpAddressLength );
6208
6209 if (!lpAddressLength || !lpAddress) return SOCKET_ERROR;
6210
6211 /* if ProtocolInfo is available - convert to ANSI variant */
6212 if (lpProtocolInfo)
6213 {
6214 lpProtoInfoA = &infoA;
6215 memcpy( lpProtoInfoA, lpProtocolInfo, FIELD_OFFSET( WSAPROTOCOL_INFOA, szProtocol ) );
6216
6217 if (!WideCharToMultiByte( CP_ACP, 0, lpProtocolInfo->szProtocol, -1,
6218 lpProtoInfoA->szProtocol, WSAPROTOCOL_LEN+1, NULL, NULL ))
6219 {
6220 WSASetLastError( WSAEINVAL);
6221 return SOCKET_ERROR;
6222 }
6223 }
6224
6225 if (AddressString)
6226 {
6227 /* Translate AddressString to ANSI code page - assumes that only
6228 standard digits 0-9 are used with this API call */
6229 sBuffer = WideCharToMultiByte( CP_ACP, 0, AddressString, -1, NULL, 0, NULL, NULL );
6230 workBuffer = HeapAlloc( GetProcessHeap(), 0, sBuffer );
6231
6232 if (workBuffer)
6233 {
6234 WideCharToMultiByte( CP_ACP, 0, AddressString, -1, workBuffer, sBuffer, NULL, NULL );
6235 res = WSAStringToAddressA(workBuffer,AddressFamily,lpProtoInfoA,
6236 lpAddress,lpAddressLength);
6237 HeapFree( GetProcessHeap(), 0, workBuffer );
6238 return res;
6239 }
6240 else
6241 res = WSA_NOT_ENOUGH_MEMORY;
6242 }
6243 else
6244 res = WSAEINVAL;
6245
6246 WSASetLastError(res);
6247 return SOCKET_ERROR;
6248 }
6249
6250 /***********************************************************************
6251 * WSAAddressToStringA (WS2_32.27)
6252 *
6253 * See WSAAddressToStringW
6254 */
6255 INT WINAPI WSAAddressToStringA( LPSOCKADDR sockaddr, DWORD len,
6256 LPWSAPROTOCOL_INFOA info, LPSTR string,
6257 LPDWORD lenstr )
6258 {
6259 DWORD size;
6260 CHAR buffer[54]; /* 32 digits + 7':' + '[' + '%" + 5 digits + ']:' + 5 digits + '\0' */
6261 CHAR *p;
6262
6263 TRACE( "(%p, %d, %p, %p, %p)\n", sockaddr, len, info, string, lenstr );
6264
6265 if (!sockaddr) return SOCKET_ERROR;
6266 if (!string || !lenstr) return SOCKET_ERROR;
6267
6268 switch(sockaddr->sa_family)
6269 {
6270 case WS_AF_INET:
6271 if (len < sizeof(SOCKADDR_IN)) return SOCKET_ERROR;
6272 sprintf( buffer, "%u.%u.%u.%u:%u",
6273 (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) >> 24 & 0xff),
6274 (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) >> 16 & 0xff),
6275 (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) >> 8 & 0xff),
6276 (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) & 0xff),
6277 ntohs( ((SOCKADDR_IN *)sockaddr)->sin_port ) );
6278
6279 p = strchr( buffer, ':' );
6280 if (!((SOCKADDR_IN *)sockaddr)->sin_port) *p = 0;
6281 break;
6282
6283 case WS_AF_INET6:
6284 {
6285 struct WS_sockaddr_in6 *sockaddr6 = (LPSOCKADDR_IN6) sockaddr;
6286
6287 buffer[0] = 0;
6288 if (len < sizeof(SOCKADDR_IN6)) return SOCKET_ERROR;
6289 if ((sockaddr6->sin6_port))
6290 strcpy(buffer, "[");
6291 if (!WS_inet_ntop(WS_AF_INET6, &sockaddr6->sin6_addr, buffer+strlen(buffer), sizeof(buffer)))
6292 {
6293 WSASetLastError(WSAEINVAL);
6294 return SOCKET_ERROR;
6295 }
6296 if ((sockaddr6->sin6_scope_id))
6297 sprintf(buffer+strlen(buffer), "%%%u", sockaddr6->sin6_scope_id);
6298 if ((sockaddr6->sin6_port))
6299 sprintf(buffer+strlen(buffer), "]:%u", ntohs(sockaddr6->sin6_port));
6300 break;
6301 }
6302
6303 default:
6304 WSASetLastError(WSAEINVAL);
6305 return SOCKET_ERROR;
6306 }
6307
6308 size = strlen( buffer ) + 1;
6309
6310 if (*lenstr < size)
6311 {
6312 *lenstr = size;
6313 WSASetLastError(WSAEFAULT);
6314 return SOCKET_ERROR;
6315 }
6316
6317 *lenstr = size;
6318 strcpy( string, buffer );
6319 return 0;
6320 }
6321
6322 /***********************************************************************
6323 * WSAAddressToStringW (WS2_32.28)
6324 *
6325 * Convert a sockaddr address into a readable address string.
6326 *
6327 * PARAMS
6328 * sockaddr [I] Pointer to a sockaddr structure.
6329 * len [I] Size of the sockaddr structure.
6330 * info [I] Pointer to a WSAPROTOCOL_INFOW structure (optional).
6331 * string [I/O] Pointer to a buffer to receive the address string.
6332 * lenstr [I/O] Size of the receive buffer in WCHARs.
6333 *
6334 * RETURNS
6335 * Success: 0
6336 * Failure: SOCKET_ERROR
6337 *
6338 * NOTES
6339 * The 'info' parameter is ignored.
6340 */
6341 INT WINAPI WSAAddressToStringW( LPSOCKADDR sockaddr, DWORD len,
6342 LPWSAPROTOCOL_INFOW info, LPWSTR string,
6343 LPDWORD lenstr )
6344 {
6345 INT ret;
6346 DWORD size;
6347 WCHAR buffer[54]; /* 32 digits + 7':' + '[' + '%" + 5 digits + ']:' + 5 digits + '\0' */
6348 CHAR bufAddr[54];
6349
6350 TRACE( "(%p, %d, %p, %p, %p)\n", sockaddr, len, info, string, lenstr );
6351
6352 size = *lenstr;
6353 ret = WSAAddressToStringA(sockaddr, len, NULL, bufAddr, &size);
6354
6355 if (ret) return ret;
6356
6357 MultiByteToWideChar( CP_ACP, 0, bufAddr, size, buffer, sizeof( buffer )/sizeof(WCHAR));
6358
6359 if (*lenstr < size)
6360 {
6361 *lenstr = size;
6362 WSASetLastError(WSAEFAULT);
6363 return SOCKET_ERROR;
6364 }
6365
6366 *lenstr = size;
6367 lstrcpyW( string, buffer );
6368 return 0;
6369 }
6370
6371 /***********************************************************************
6372 * WSAEnumNameSpaceProvidersA (WS2_32.34)
6373 */
6374 INT WINAPI WSAEnumNameSpaceProvidersA( LPDWORD len, LPWSANAMESPACE_INFOA buffer )
6375 {
6376 FIXME( "(%p %p) Stub!\n", len, buffer );
6377 return 0;
6378 }
6379
6380 /***********************************************************************
6381 * WSAEnumNameSpaceProvidersW (WS2_32.35)
6382 */
6383 INT WINAPI WSAEnumNameSpaceProvidersW( LPDWORD len, LPWSANAMESPACE_INFOW buffer )
6384 {
6385 FIXME( "(%p %p) Stub!\n", len, buffer );
6386 return 0;
6387 }
6388
6389 /***********************************************************************
6390 * WSAGetQOSByName (WS2_32.41)
6391 */
6392 BOOL WINAPI WSAGetQOSByName( SOCKET s, LPWSABUF lpQOSName, LPQOS lpQOS )
6393 {
6394 FIXME( "(0x%04lx %p %p) Stub!\n", s, lpQOSName, lpQOS );
6395 return FALSE;
6396 }
6397
6398 /***********************************************************************
6399 * WSAGetServiceClassInfoA (WS2_32.42)
6400 */
6401 INT WINAPI WSAGetServiceClassInfoA( LPGUID provider, LPGUID service, LPDWORD len,
6402 LPWSASERVICECLASSINFOA info )
6403 {
6404 FIXME( "(%s %s %p %p) Stub!\n", debugstr_guid(provider), debugstr_guid(service),
6405 len, info );
6406 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6407 return SOCKET_ERROR;
6408 }
6409
6410 /***********************************************************************
6411 * WSAGetServiceClassInfoW (WS2_32.43)
6412 */
6413 INT WINAPI WSAGetServiceClassInfoW( LPGUID provider, LPGUID service, LPDWORD len,
6414 LPWSASERVICECLASSINFOW info )
6415 {
6416 FIXME( "(%s %s %p %p) Stub!\n", debugstr_guid(provider), debugstr_guid(service),
6417 len, info );
6418 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6419 return SOCKET_ERROR;
6420 }
6421
6422 /***********************************************************************
6423 * WSAGetServiceClassNameByClassIdA (WS2_32.44)
6424 */
6425 INT WINAPI WSAGetServiceClassNameByClassIdA( LPGUID class, LPSTR service, LPDWORD len )
6426 {
6427 FIXME( "(%s %p %p) Stub!\n", debugstr_guid(class), service, len );
6428 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6429 return SOCKET_ERROR;
6430 }
6431
6432 /***********************************************************************
6433 * WSAGetServiceClassNameByClassIdW (WS2_32.45)
6434 */
6435 INT WINAPI WSAGetServiceClassNameByClassIdW( LPGUID class, LPWSTR service, LPDWORD len )
6436 {
6437 FIXME( "(%s %p %p) Stub!\n", debugstr_guid(class), service, len );
6438 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6439 return SOCKET_ERROR;
6440 }
6441
6442 /***********************************************************************
6443 * WSALookupServiceBeginA (WS2_32.59)
6444 */
6445 INT WINAPI WSALookupServiceBeginA( LPWSAQUERYSETA lpqsRestrictions,
6446 DWORD dwControlFlags,
6447 LPHANDLE lphLookup)
6448 {
6449 FIXME("(%p 0x%08x %p) Stub!\n", lpqsRestrictions, dwControlFlags,
6450 lphLookup);
6451 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6452 return SOCKET_ERROR;
6453 }
6454
6455 /***********************************************************************
6456 * WSALookupServiceBeginW (WS2_32.60)
6457 */
6458 INT WINAPI WSALookupServiceBeginW( LPWSAQUERYSETW lpqsRestrictions,
6459 DWORD dwControlFlags,
6460 LPHANDLE lphLookup)
6461 {
6462 FIXME("(%p 0x%08x %p) Stub!\n", lpqsRestrictions, dwControlFlags,
6463 lphLookup);
6464 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6465 return SOCKET_ERROR;
6466 }
6467
6468 /***********************************************************************
6469 * WSALookupServiceBeginW (WS2_32.61)
6470 */
6471 INT WINAPI WSALookupServiceEnd( HANDLE lookup )
6472 {
6473 FIXME("(%p) Stub!\n", lookup );
6474 return 0;
6475 }
6476
6477 /***********************************************************************
6478 * WSALookupServiceNextA (WS2_32.62)
6479 */
6480 INT WINAPI WSALookupServiceNextA( HANDLE lookup, DWORD flags, LPDWORD len, LPWSAQUERYSETA results )
6481 {
6482 FIXME( "(%p 0x%08x %p %p) Stub!\n", lookup, flags, len, results );
6483 WSASetLastError(WSA_E_NO_MORE);
6484 return SOCKET_ERROR;
6485 }
6486
6487 /***********************************************************************
6488 * WSALookupServiceNextW (WS2_32.63)
6489 */
6490 INT WINAPI WSALookupServiceNextW( HANDLE lookup, DWORD flags, LPDWORD len, LPWSAQUERYSETW results )
6491 {
6492 FIXME( "(%p 0x%08x %p %p) Stub!\n", lookup, flags, len, results );
6493 WSASetLastError(WSA_E_NO_MORE);
6494 return SOCKET_ERROR;
6495 }
6496
6497 /***********************************************************************
6498 * WSANtohl (WS2_32.64)
6499 */
6500 INT WINAPI WSANtohl( SOCKET s, WS_u_long netlong, WS_u_long* lphostlong )
6501 {
6502 TRACE( "(0x%04lx 0x%08x %p)\n", s, netlong, lphostlong );
6503
6504 if (!lphostlong) return WSAEFAULT;
6505
6506 *lphostlong = ntohl( netlong );
6507 return 0;
6508 }
6509
6510 /***********************************************************************
6511 * WSANtohs (WS2_32.65)
6512 */
6513 INT WINAPI WSANtohs( SOCKET s, WS_u_short netshort, WS_u_short* lphostshort )
6514 {
6515 TRACE( "(0x%04lx 0x%08x %p)\n", s, netshort, lphostshort );
6516
6517 if (!lphostshort) return WSAEFAULT;
6518
6519 *lphostshort = ntohs( netshort );
6520 return 0;
6521 }
6522
6523 /***********************************************************************
6524 * WSAProviderConfigChange (WS2_32.66)
6525 */
6526 INT WINAPI WSAProviderConfigChange( LPHANDLE handle, LPWSAOVERLAPPED overlapped,
6527 LPWSAOVERLAPPED_COMPLETION_ROUTINE completion )
6528 {
6529 FIXME( "(%p %p %p) Stub!\n", handle, overlapped, completion );
6530 return SOCKET_ERROR;
6531 }
6532
6533 /***********************************************************************
6534 * WSARecvDisconnect (WS2_32.68)
6535 */
6536 INT WINAPI WSARecvDisconnect( SOCKET s, LPWSABUF disconnectdata )
6537 {
6538 TRACE( "(0x%04lx %p)\n", s, disconnectdata );
6539
6540 return WS_shutdown( s, 0 );
6541 }
6542
6543 /***********************************************************************
6544 * WSASetServiceA (WS2_32.76)
6545 */
6546 INT WINAPI WSASetServiceA( LPWSAQUERYSETA query, WSAESETSERVICEOP operation, DWORD flags )
6547 {
6548 FIXME( "(%p 0x%08x 0x%08x) Stub!\n", query, operation, flags );
6549 return 0;
6550 }
6551
6552 /***********************************************************************
6553 * WSASetServiceW (WS2_32.77)
6554 */
6555 INT WINAPI WSASetServiceW( LPWSAQUERYSETW query, WSAESETSERVICEOP operation, DWORD flags )
6556 {
6557 FIXME( "(%p 0x%08x 0x%08x) Stub!\n", query, operation, flags );
6558 return 0;
6559 }
6560
6561 /***********************************************************************
6562 * WSCEnableNSProvider (WS2_32.84)
6563 */
6564 INT WINAPI WSCEnableNSProvider( LPGUID provider, BOOL enable )
6565 {
6566 FIXME( "(%s 0x%08x) Stub!\n", debugstr_guid(provider), enable );
6567 return 0;
6568 }
6569
6570 /***********************************************************************
6571 * WSCGetProviderPath (WS2_32.86)
6572 */
6573 INT WINAPI WSCGetProviderPath( LPGUID provider, LPWSTR path, LPINT len, LPINT errcode )
6574 {
6575 FIXME( "(%s %p %p %p) Stub!\n", debugstr_guid(provider), path, len, errcode );
6576
6577 if (!errcode || !provider || !len) return WSAEFAULT;
6578
6579 *errcode = WSAEINVAL;
6580 return SOCKET_ERROR;
6581 }
6582
6583 /***********************************************************************
6584 * WSCInstallNameSpace (WS2_32.87)
6585 */
6586 INT WINAPI WSCInstallNameSpace( LPWSTR identifier, LPWSTR path, DWORD namespace,
6587 DWORD version, LPGUID provider )
6588 {
6589 FIXME( "(%s %s 0x%08x 0x%08x %s) Stub!\n", debugstr_w(identifier), debugstr_w(path),
6590 namespace, version, debugstr_guid(provider) );
6591 return 0;
6592 }
6593
6594 /***********************************************************************
6595 * WSCUnInstallNameSpace (WS2_32.89)
6596 */
6597 INT WINAPI WSCUnInstallNameSpace( LPGUID lpProviderId )
6598 {
6599 FIXME("(%p) Stub!\n", lpProviderId);
6600 return NO_ERROR;
6601 }
6602
6603 /***********************************************************************
6604 * WSCWriteProviderOrder (WS2_32.91)
6605 */
6606 INT WINAPI WSCWriteProviderOrder( LPDWORD entry, DWORD number )
6607 {
6608 FIXME("(%p 0x%08x) Stub!\n", entry, number);
6609 return 0;
6610 }
6611
6612 /***********************************************************************
6613 * WSANSPIoctl (WS2_32.91)
6614 */
6615 INT WINAPI WSANSPIoctl( HANDLE hLookup, DWORD dwControlCode, LPVOID lpvInBuffer,
6616 DWORD cbInBuffer, LPVOID lpvOutBuffer, DWORD cbOutBuffer,
6617 LPDWORD lpcbBytesReturned, LPWSACOMPLETION lpCompletion )
6618 {
6619 FIXME("(%p, 0x%08x, %p, 0x%08x, %p, 0x%08x, %p, %p) Stub!\n", hLookup, dwControlCode,
6620 lpvInBuffer, cbInBuffer, lpvOutBuffer, cbOutBuffer, lpcbBytesReturned, lpCompletion);
6621 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
6622 return SOCKET_ERROR;
6623 };
Maarten's multimedia branch