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