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