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