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