1 /*
2 * based on Windows Sockets 1.1 specs
3 *
4 * Copyright (C) 1993,1994,1996,1997 John Brezak, Erik Bos, Alex Korobka.
5 * Copyright (C) 2005 Marcus Meissner
6 * Copyright (C) 2006-2008 Kai Blin
7 *
8 * This library is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
12 *
13 * This library is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with this library; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 *
22 * NOTE: If you make any changes to fix a particular app, make sure
23 * they don't break something else like Netscape or telnet and ftp
24 * clients and servers (www.winsite.com got a lot of those).
25 */
26
27 #include "config.h"
28 #include "wine/port.h"
29
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <string.h>
33 #include <sys/types.h>
34 #ifdef HAVE_SYS_IPC_H
35 # include <sys/ipc.h>
36 #endif
37 #ifdef HAVE_SYS_IOCTL_H
38 # include <sys/ioctl.h>
39 #endif
40 #ifdef HAVE_SYS_FILIO_H
41 # include <sys/filio.h>
42 #endif
43 #ifdef HAVE_SYS_SOCKIO_H
44 # include <sys/sockio.h>
45 #endif
46
47 #if defined(__EMX__)
48 # include <sys/so_ioctl.h>
49 #endif
50
51 #ifdef HAVE_SYS_PARAM_H
52 # include <sys/param.h>
53 #endif
54
55 #ifdef HAVE_SYS_MSG_H
56 # include <sys/msg.h>
57 #endif
58 #ifdef HAVE_SYS_WAIT_H
59 # include <sys/wait.h>
60 #endif
61 #ifdef HAVE_SYS_UIO_H
62 # include <sys/uio.h>
63 #endif
64 #ifdef HAVE_SYS_SOCKET_H
65 #include <sys/socket.h>
66 #endif
67 #ifdef HAVE_NETINET_IN_H
68 # include <netinet/in.h>
69 #endif
70 #ifdef HAVE_NETINET_TCP_H
71 # include <netinet/tcp.h>
72 #endif
73 #ifdef HAVE_ARPA_INET_H
74 # include <arpa/inet.h>
75 #endif
76 #include <ctype.h>
77 #include <fcntl.h>
78 #include <errno.h>
79 #ifdef HAVE_SYS_ERRNO_H
80 #include <sys/errno.h>
81 #endif
82 #ifdef HAVE_NETDB_H
83 #include <netdb.h>
84 #endif
85 #ifdef HAVE_UNISTD_H
86 # include <unistd.h>
87 #endif
88 #include <stdlib.h>
89 #ifdef HAVE_ARPA_NAMESER_H
90 # include <arpa/nameser.h>
91 #endif
92 #ifdef HAVE_RESOLV_H
93 # include <resolv.h>
94 #endif
95 #ifdef HAVE_NET_IF_H
96 # include <net/if.h>
97 #endif
98
99 #ifdef HAVE_NETIPX_IPX_H
100 # include <netipx/ipx.h>
101 # define HAVE_IPX
102 #elif defined(HAVE_LINUX_IPX_H)
103 # ifdef HAVE_ASM_TYPES_H
104 # include <asm/types.h>
105 # endif
106 # ifdef HAVE_LINUX_TYPES_H
107 # include <linux/types.h>
108 # endif
109 # include <linux/ipx.h>
110 # define HAVE_IPX
111 #endif
112
113 #ifdef HAVE_POLL_H
114 #include <poll.h>
115 #endif
116 #ifdef HAVE_SYS_POLL_H
117 # include <sys/poll.h>
118 #endif
119 #ifdef HAVE_SYS_TIME_H
120 # include <sys/time.h>
121 #endif
122
123 #define NONAMELESSUNION
124 #define NONAMELESSSTRUCT
125 #include "ntstatus.h"
126 #define WIN32_NO_STATUS
127 #include "windef.h"
128 #include "winbase.h"
129 #include "wingdi.h"
130 #include "winuser.h"
131 #include "winerror.h"
132 #include "winnls.h"
133 #include "winsock2.h"
134 #include "mswsock.h"
135 #include "ws2tcpip.h"
136 #include "ws2spi.h"
137 #include "wsipx.h"
138 #include "mstcpip.h"
139 #include "winnt.h"
140 #include "iphlpapi.h"
141 #include "wine/server.h"
142 #include "wine/debug.h"
143 #include "wine/unicode.h"
144
145 #ifdef HAVE_IPX
146 # include "wsnwlink.h"
147 #endif
148
149
150 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
151 # define sipx_network sipx_addr.x_net
152 # define sipx_node sipx_addr.x_host.c_host
153 #endif /* __FreeBSD__ */
154
155 #ifndef INADDR_NONE
156 #define INADDR_NONE ~0UL
157 #endif
158
159 WINE_DEFAULT_DEBUG_CHANNEL(winsock);
160
161 /* critical section to protect some non-reentrant net function */
162 extern CRITICAL_SECTION csWSgetXXXbyYYY;
163
164 union generic_unix_sockaddr
165 {
166 struct sockaddr addr;
167 char data[128]; /* should be big enough for all families */
168 };
169
170 static inline const char *debugstr_sockaddr( const struct WS_sockaddr *a )
171 {
172 if (!a) return "(nil)";
173 switch (a->sa_family)
174 {
175 case WS_AF_INET:
176 return wine_dbg_sprintf("{ family AF_INET, address %s, port %d }",
177 inet_ntoa(((const struct sockaddr_in *)a)->sin_addr),
178 ntohs(((const struct sockaddr_in *)a)->sin_port));
179 case WS_AF_INET6:
180 {
181 char buf[46];
182 const char *p;
183 struct WS_sockaddr_in6 *sin = (struct WS_sockaddr_in6 *)a;
184
185 p = WS_inet_ntop( WS_AF_INET6, &sin->sin6_addr, buf, sizeof(buf) );
186 if (!p)
187 p = "(unknown IPv6 address)";
188 return wine_dbg_sprintf("{ family AF_INET6, address %s, port %d }",
189 p, ntohs(sin->sin6_port));
190 }
191 default:
192 return wine_dbg_sprintf("{ family %d }", a->sa_family);
193 }
194 }
195
196 /* HANDLE<->SOCKET conversion (SOCKET is UINT_PTR). */
197 #define SOCKET2HANDLE(s) ((HANDLE)(s))
198 #define HANDLE2SOCKET(h) ((SOCKET)(h))
199
200 /****************************************************************
201 * Async IO declarations
202 ****************************************************************/
203
204 typedef struct ws2_async
205 {
206 HANDLE hSocket;
207 int type;
208 LPWSAOVERLAPPED user_overlapped;
209 LPWSAOVERLAPPED_COMPLETION_ROUTINE completion_func;
210 IO_STATUS_BLOCK local_iosb;
211 struct WS_sockaddr *addr;
212 union
213 {
214 int val; /* for send operations */
215 int *ptr; /* for recv operations */
216 } addrlen;
217 DWORD flags;
218 unsigned int n_iovecs;
219 unsigned int first_iovec;
220 struct iovec iovec[1];
221 } ws2_async;
222
223 /****************************************************************/
224
225 /* ----------------------------------- internal data */
226
227 /* ws_... struct conversion flags */
228
229 typedef struct /* WSAAsyncSelect() control struct */
230 {
231 HANDLE service, event, sock;
232 HWND hWnd;
233 UINT uMsg;
234 LONG lEvent;
235 } ws_select_info;
236
237 #define WS_MAX_SOCKETS_PER_PROCESS 128 /* reasonable guess */
238 #define WS_MAX_UDP_DATAGRAM 1024
239 static INT WINAPI WSA_DefaultBlockingHook( FARPROC x );
240
241 /* hostent's, servent's and protent's are stored in one buffer per thread,
242 * as documented on MSDN for the functions that return any of the buffers */
243 struct per_thread_data
244 {
245 int opentype;
246 struct WS_hostent *he_buffer;
247 struct WS_servent *se_buffer;
248 struct WS_protoent *pe_buffer;
249 int he_len;
250 int se_len;
251 int pe_len;
252 };
253
254 static INT num_startup; /* reference counter */
255 static FARPROC blocking_hook = (FARPROC)WSA_DefaultBlockingHook;
256
257 /* function prototypes */
258 static struct WS_hostent *WS_dup_he(const struct hostent* p_he);
259 static struct WS_protoent *WS_dup_pe(const struct protoent* p_pe);
260 static struct WS_servent *WS_dup_se(const struct servent* p_se);
261
262 int WSAIOCTL_GetInterfaceCount(void);
263 int WSAIOCTL_GetInterfaceName(int intNumber, char *intName);
264
265 UINT wsaErrno(void);
266 UINT wsaHerrno(int errnr);
267
268 #define MAP_OPTION(opt) { WS_##opt, opt }
269
270 static const int ws_sock_map[][2] =
271 {
272 MAP_OPTION( SO_DEBUG ),
273 MAP_OPTION( SO_ACCEPTCONN ),
274 MAP_OPTION( SO_REUSEADDR ),
275 MAP_OPTION( SO_KEEPALIVE ),
276 MAP_OPTION( SO_DONTROUTE ),
277 MAP_OPTION( SO_BROADCAST ),
278 MAP_OPTION( SO_LINGER ),
279 MAP_OPTION( SO_OOBINLINE ),
280 MAP_OPTION( SO_SNDBUF ),
281 MAP_OPTION( SO_RCVBUF ),
282 MAP_OPTION( SO_ERROR ),
283 MAP_OPTION( SO_TYPE ),
284 #ifdef SO_RCVTIMEO
285 MAP_OPTION( SO_RCVTIMEO ),
286 #endif
287 #ifdef SO_SNDTIMEO
288 MAP_OPTION( SO_SNDTIMEO ),
289 #endif
290 };
291
292 static const int ws_tcp_map[][2] =
293 {
294 #ifdef TCP_NODELAY
295 MAP_OPTION( TCP_NODELAY ),
296 #endif
297 };
298
299 static const int ws_ip_map[][2] =
300 {
301 MAP_OPTION( IP_MULTICAST_IF ),
302 MAP_OPTION( IP_MULTICAST_TTL ),
303 MAP_OPTION( IP_MULTICAST_LOOP ),
304 MAP_OPTION( IP_ADD_MEMBERSHIP ),
305 MAP_OPTION( IP_DROP_MEMBERSHIP ),
306 MAP_OPTION( IP_OPTIONS ),
307 #ifdef IP_HDRINCL
308 MAP_OPTION( IP_HDRINCL ),
309 #endif
310 MAP_OPTION( IP_TOS ),
311 MAP_OPTION( IP_TTL ),
312 };
313
314 static const int ws_af_map[][2] =
315 {
316 MAP_OPTION( AF_UNSPEC ),
317 MAP_OPTION( AF_INET ),
318 MAP_OPTION( AF_INET6 ),
319 #ifdef HAVE_IPX
320 MAP_OPTION( AF_IPX ),
321 #endif
322 {FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO},
323 };
324
325 static const int ws_socktype_map[][2] =
326 {
327 MAP_OPTION( SOCK_DGRAM ),
328 MAP_OPTION( SOCK_STREAM ),
329 MAP_OPTION( SOCK_RAW ),
330 {FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO},
331 };
332
333 static const int ws_proto_map[][2] =
334 {
335 MAP_OPTION( IPPROTO_IP ),
336 MAP_OPTION( IPPROTO_TCP ),
337 MAP_OPTION( IPPROTO_UDP ),
338 MAP_OPTION( IPPROTO_ICMP ),
339 MAP_OPTION( IPPROTO_IGMP ),
340 MAP_OPTION( IPPROTO_RAW ),
341 {FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO},
342 };
343
344 static const int ws_aiflag_map[][2] =
345 {
346 MAP_OPTION( AI_PASSIVE ),
347 MAP_OPTION( AI_CANONNAME ),
348 MAP_OPTION( AI_NUMERICHOST ),
349 /* Linux/UNIX knows a lot more. But Windows only
350 * has 3 as far as I could see. -Marcus
351 */
352 };
353
354 static const int ws_niflag_map[][2] =
355 {
356 MAP_OPTION( NI_NOFQDN ),
357 MAP_OPTION( NI_NUMERICHOST ),
358 MAP_OPTION( NI_NAMEREQD ),
359 MAP_OPTION( NI_NUMERICSERV ),
360 MAP_OPTION( NI_DGRAM ),
361 };
362
363 static const int ws_eai_map[][2] =
364 {
365 MAP_OPTION( EAI_AGAIN ),
366 MAP_OPTION( EAI_BADFLAGS ),
367 MAP_OPTION( EAI_FAIL ),
368 MAP_OPTION( EAI_FAMILY ),
369 MAP_OPTION( EAI_MEMORY ),
370 /* Note: EAI_NODATA is deprecated, but still
371 * used by Windows and Linux... We map the newer
372 * EAI_NONAME to EAI_NODATA for now until Windows
373 * changes too.
374 */
375 #ifdef EAI_NODATA
376 MAP_OPTION( EAI_NODATA ),
377 #endif
378 #ifdef EAI_NONAME
379 { WS_EAI_NODATA, EAI_NONAME },
380 #endif
381
382 MAP_OPTION( EAI_SERVICE ),
383 MAP_OPTION( EAI_SOCKTYPE ),
384 { 0, 0 }
385 };
386
387 static const char magic_loopback_addr[] = {127, 12, 34, 56};
388
389 static inline DWORD NtStatusToWSAError( const DWORD status )
390 {
391 /* We only need to cover the status codes set by server async request handling */
392 DWORD wserr;
393 switch ( status )
394 {
395 case STATUS_SUCCESS: wserr = 0; break;
396 case STATUS_PENDING: wserr = WSA_IO_PENDING; break;
397 case STATUS_OBJECT_TYPE_MISMATCH: wserr = WSAENOTSOCK; break;
398 case STATUS_INVALID_HANDLE: wserr = WSAEBADF; break;
399 case STATUS_INVALID_PARAMETER: wserr = WSAEINVAL; break;
400 case STATUS_PIPE_DISCONNECTED: wserr = WSAESHUTDOWN; break;
401 case STATUS_CANCELLED: wserr = WSA_OPERATION_ABORTED; break;
402 case STATUS_TIMEOUT: wserr = WSAETIMEDOUT; break;
403 case STATUS_NO_MEMORY: wserr = WSAEFAULT; break;
404 default:
405 if ( status >= WSABASEERR && status <= WSABASEERR+1004 )
406 /* It is not an NT status code but a winsock error */
407 wserr = status;
408 else
409 {
410 wserr = RtlNtStatusToDosError( status );
411 FIXME( "Status code %08x converted to DOS error code %x\n", status, wserr );
412 }
413 }
414 return wserr;
415 }
416
417 /* set last error code from NT status without mapping WSA errors */
418 static inline unsigned int set_error( unsigned int err )
419 {
420 if (err)
421 {
422 err = NtStatusToWSAError( err );
423 SetLastError( err );
424 }
425 return err;
426 }
427
428 static inline int get_sock_fd( SOCKET s, DWORD access, unsigned int *options )
429 {
430 int fd;
431 if (set_error( wine_server_handle_to_fd( SOCKET2HANDLE(s), access, &fd, options ) ))
432 return -1;
433 return fd;
434 }
435
436 static inline void release_sock_fd( SOCKET s, int fd )
437 {
438 wine_server_release_fd( SOCKET2HANDLE(s), fd );
439 }
440
441 static void _enable_event( HANDLE s, unsigned int event,
442 unsigned int sstate, unsigned int cstate )
443 {
444 SERVER_START_REQ( enable_socket_event )
445 {
446 req->handle = wine_server_obj_handle( s );
447 req->mask = event;
448 req->sstate = sstate;
449 req->cstate = cstate;
450 wine_server_call( req );
451 }
452 SERVER_END_REQ;
453 }
454
455 static int _is_blocking(SOCKET s)
456 {
457 int ret;
458 SERVER_START_REQ( get_socket_event )
459 {
460 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
461 req->service = FALSE;
462 req->c_event = 0;
463 wine_server_call( req );
464 ret = (reply->state & FD_WINE_NONBLOCKING) == 0;
465 }
466 SERVER_END_REQ;
467 return ret;
468 }
469
470 static unsigned int _get_sock_mask(SOCKET s)
471 {
472 unsigned int ret;
473 SERVER_START_REQ( get_socket_event )
474 {
475 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
476 req->service = FALSE;
477 req->c_event = 0;
478 wine_server_call( req );
479 ret = reply->mask;
480 }
481 SERVER_END_REQ;
482 return ret;
483 }
484
485 static void _sync_sock_state(SOCKET s)
486 {
487 /* do a dummy wineserver request in order to let
488 the wineserver run through its select loop once */
489 (void)_is_blocking(s);
490 }
491
492 static int _get_sock_error(SOCKET s, unsigned int bit)
493 {
494 int events[FD_MAX_EVENTS];
495
496 SERVER_START_REQ( get_socket_event )
497 {
498 req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) );
499 req->service = FALSE;
500 req->c_event = 0;
501 wine_server_set_reply( req, events, sizeof(events) );
502 wine_server_call( req );
503 }
504 SERVER_END_REQ;
505 return events[bit];
506 }
507
508 static struct per_thread_data *get_per_thread_data(void)
509 {
510 struct per_thread_data * ptb = NtCurrentTeb()->WinSockData;
511 /* lazy initialization */
512 if (!ptb)
513 {
514 ptb = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*ptb) );
515 NtCurrentTeb()->WinSockData = ptb;
516 }
517 return ptb;
518 }
519
520 static void free_per_thread_data(void)
521 {
522 struct per_thread_data * ptb = NtCurrentTeb()->WinSockData;
523
524 if (!ptb) return;
525
526 /* delete scratch buffers */
527 HeapFree( GetProcessHeap(), 0, ptb->he_buffer );
528 HeapFree( GetProcessHeap(), 0, ptb->se_buffer );
529 HeapFree( GetProcessHeap(), 0, ptb->pe_buffer );
530 ptb->he_buffer = NULL;
531 ptb->se_buffer = NULL;
532 ptb->pe_buffer = NULL;
533
534 HeapFree( GetProcessHeap(), 0, ptb );
535 NtCurrentTeb()->WinSockData = NULL;
536 }
537
538 /***********************************************************************
539 * DllMain (WS2_32.init)
540 */
541 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID fImpLoad)
542 {
543 TRACE("%p 0x%x %p\n", hInstDLL, fdwReason, fImpLoad);
544 switch (fdwReason) {
545 case DLL_PROCESS_ATTACH:
546 break;
547 case DLL_PROCESS_DETACH:
548 free_per_thread_data();
549 num_startup = 0;
550 break;
551 case DLL_THREAD_DETACH:
552 free_per_thread_data();
553 break;
554 }
555 return TRUE;
556 }
557
558 /***********************************************************************
559 * convert_sockopt()
560 *
561 * Converts socket flags from Windows format.
562 * Return 1 if converted, 0 if not (error).
563 */
564 static int convert_sockopt(INT *level, INT *optname)
565 {
566 unsigned int i;
567 switch (*level)
568 {
569 case WS_SOL_SOCKET:
570 *level = SOL_SOCKET;
571 for(i=0; i<sizeof(ws_sock_map)/sizeof(ws_sock_map[0]); i++) {
572 if( ws_sock_map[i][0] == *optname )
573 {
574 *optname = ws_sock_map[i][1];
575 return 1;
576 }
577 }
578 FIXME("Unknown SOL_SOCKET optname 0x%x\n", *optname);
579 break;
580 case WS_IPPROTO_TCP:
581 *level = IPPROTO_TCP;
582 for(i=0; i<sizeof(ws_tcp_map)/sizeof(ws_tcp_map[0]); i++) {
583 if ( ws_tcp_map[i][0] == *optname )
584 {
585 *optname = ws_tcp_map[i][1];
586 return 1;
587 }
588 }
589 FIXME("Unknown IPPROTO_TCP optname 0x%x\n", *optname);
590 break;
591 case WS_IPPROTO_IP:
592 *level = IPPROTO_IP;
593 for(i=0; i<sizeof(ws_ip_map)/sizeof(ws_ip_map[0]); i++) {
594 if (ws_ip_map[i][0] == *optname )
595 {
596 *optname = ws_ip_map[i][1];
597 return 1;
598 }
599 }
600 FIXME("Unknown IPPROTO_IP optname 0x%x\n", *optname);
601 break;
602 default: FIXME("Unimplemented or unknown socket level\n");
603 }
604 return 0;
605 }
606
607 /* ----------------------------------- Per-thread info (or per-process?) */
608
609 static char *strdup_lower(const char *str)
610 {
611 int i;
612 char *ret = HeapAlloc( GetProcessHeap(), 0, strlen(str) + 1 );
613
614 if (ret)
615 {
616 for (i = 0; str[i]; i++) ret[i] = tolower(str[i]);
617 ret[i] = 0;
618 }
619 else SetLastError(WSAENOBUFS);
620 return ret;
621 }
622
623 static inline int sock_error_p(int s)
624 {
625 unsigned int optval, optlen;
626
627 optlen = sizeof(optval);
628 getsockopt(s, SOL_SOCKET, SO_ERROR, (void *) &optval, &optlen);
629 if (optval) WARN("\t[%i] error: %d\n", s, optval);
630 return optval != 0;
631 }
632
633 /* Utility: get the SO_RCVTIMEO or SO_SNDTIMEO socket option
634 * from an fd and return the value converted to milli seconds
635 * or -1 if there is an infinite time out */
636 static inline int get_rcvsnd_timeo( int fd, int optname)
637 {
638 struct timeval tv;
639 unsigned int len = sizeof(tv);
640 int ret = getsockopt(fd, SOL_SOCKET, optname, &tv, &len);
641 if( ret >= 0)
642 ret = tv.tv_sec * 1000 + tv.tv_usec / 1000;
643 if( ret <= 0 ) /* tv == {0,0} means infinite time out */
644 return -1;
645 return ret;
646 }
647
648 /* macro wrappers for portability */
649 #ifdef SO_RCVTIMEO
650 #define GET_RCVTIMEO(fd) get_rcvsnd_timeo( (fd), SO_RCVTIMEO)
651 #else
652 #define GET_RCVTIMEO(fd) (-1)
653 #endif
654
655 #ifdef SO_SNDTIMEO
656 #define GET_SNDTIMEO(fd) get_rcvsnd_timeo( (fd), SO_SNDTIMEO)
657 #else
658 #define GET_SNDTIMEO(fd) (-1)
659 #endif
660
661 /* utility: given an fd, will block until one of the events occurs */
662 static inline int do_block( int fd, int events, int timeout )
663 {
664 struct pollfd pfd;
665 int ret;
666
667 pfd.fd = fd;
668 pfd.events = events;
669
670 while ((ret = poll(&pfd, 1, timeout)) < 0)
671 {
672 if (errno != EINTR)
673 return -1;
674 }
675 if( ret == 0 )
676 return 0;
677 return pfd.revents;
678 }
679
680 static int
681 convert_af_w2u(int windowsaf) {
682 unsigned int i;
683
684 for (i=0;i<sizeof(ws_af_map)/sizeof(ws_af_map[0]);i++)
685 if (ws_af_map[i][0] == windowsaf)
686 return ws_af_map[i][1];
687 FIXME("unhandled Windows address family %d\n", windowsaf);
688 return -1;
689 }
690
691 static int
692 convert_af_u2w(int unixaf) {
693 unsigned int i;
694
695 for (i=0;i<sizeof(ws_af_map)/sizeof(ws_af_map[0]);i++)
696 if (ws_af_map[i][1] == unixaf)
697 return ws_af_map[i][0];
698 FIXME("unhandled UNIX address family %d\n", unixaf);
699 return -1;
700 }
701
702 static int
703 convert_proto_w2u(int windowsproto) {
704 unsigned int i;
705
706 for (i=0;i<sizeof(ws_proto_map)/sizeof(ws_proto_map[0]);i++)
707 if (ws_proto_map[i][0] == windowsproto)
708 return ws_proto_map[i][1];
709 FIXME("unhandled Windows socket protocol %d\n", windowsproto);
710 return -1;
711 }
712
713 static int
714 convert_proto_u2w(int unixproto) {
715 unsigned int i;
716
717 for (i=0;i<sizeof(ws_proto_map)/sizeof(ws_proto_map[0]);i++)
718 if (ws_proto_map[i][1] == unixproto)
719 return ws_proto_map[i][0];
720 FIXME("unhandled UNIX socket protocol %d\n", unixproto);
721 return -1;
722 }
723
724 static int
725 convert_socktype_w2u(int windowssocktype) {
726 unsigned int i;
727
728 for (i=0;i<sizeof(ws_socktype_map)/sizeof(ws_socktype_map[0]);i++)
729 if (ws_socktype_map[i][0] == windowssocktype)
730 return ws_socktype_map[i][1];
731 FIXME("unhandled Windows socket type %d\n", windowssocktype);
732 return -1;
733 }
734
735 static int
736 convert_socktype_u2w(int unixsocktype) {
737 unsigned int i;
738
739 for (i=0;i<sizeof(ws_socktype_map)/sizeof(ws_socktype_map[0]);i++)
740 if (ws_socktype_map[i][1] == unixsocktype)
741 return ws_socktype_map[i][0];
742 FIXME("unhandled UNIX socket type %d\n", unixsocktype);
743 return -1;
744 }
745
746 /* ----------------------------------- API -----
747 *
748 * Init / cleanup / error checking.
749 */
750
751 /***********************************************************************
752 * WSAStartup (WS2_32.115)
753 */
754 int WINAPI WSAStartup(WORD wVersionRequested, LPWSADATA lpWSAData)
755 {
756 TRACE("verReq=%x\n", wVersionRequested);
757
758 if (LOBYTE(wVersionRequested) < 1)
759 return WSAVERNOTSUPPORTED;
760
761 if (!lpWSAData) return WSAEINVAL;
762
763 num_startup++;
764
765 /* that's the whole of the negotiation for now */
766 lpWSAData->wVersion = wVersionRequested;
767 /* return winsock information */
768 lpWSAData->wHighVersion = 0x0202;
769 strcpy(lpWSAData->szDescription, "WinSock 2.0" );
770 strcpy(lpWSAData->szSystemStatus, "Running" );
771 lpWSAData->iMaxSockets = WS_MAX_SOCKETS_PER_PROCESS;
772 lpWSAData->iMaxUdpDg = WS_MAX_UDP_DATAGRAM;
773 /* don't do anything with lpWSAData->lpVendorInfo */
774 /* (some apps don't allocate the space for this field) */
775
776 TRACE("succeeded\n");
777 return 0;
778 }
779
780
781 /***********************************************************************
782 * WSACleanup (WS2_32.116)
783 */
784 INT WINAPI WSACleanup(void)
785 {
786 if (num_startup) {
787 num_startup--;
788 return 0;
789 }
790 SetLastError(WSANOTINITIALISED);
791 return SOCKET_ERROR;
792 }
793
794
795 /***********************************************************************
796 * WSAGetLastError (WINSOCK.111)
797 * WSAGetLastError (WS2_32.111)
798 */
799 INT WINAPI WSAGetLastError(void)
800 {
801 return GetLastError();
802 }
803
804 /***********************************************************************
805 * WSASetLastError (WS2_32.112)
806 */
807 void WINAPI WSASetLastError(INT iError) {
808 SetLastError(iError);
809 }
810
811 static struct WS_hostent *check_buffer_he(int size)
812 {
813 struct per_thread_data * ptb = get_per_thread_data();
814 if (ptb->he_buffer)
815 {
816 if (ptb->he_len >= size ) return ptb->he_buffer;
817 HeapFree( GetProcessHeap(), 0, ptb->he_buffer );
818 }
819 ptb->he_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->he_len = size) );
820 if (!ptb->he_buffer) SetLastError(WSAENOBUFS);
821 return ptb->he_buffer;
822 }
823
824 static struct WS_servent *check_buffer_se(int size)
825 {
826 struct per_thread_data * ptb = get_per_thread_data();
827 if (ptb->se_buffer)
828 {
829 if (ptb->se_len >= size ) return ptb->se_buffer;
830 HeapFree( GetProcessHeap(), 0, ptb->se_buffer );
831 }
832 ptb->se_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->se_len = size) );
833 if (!ptb->se_buffer) SetLastError(WSAENOBUFS);
834 return ptb->se_buffer;
835 }
836
837 static struct WS_protoent *check_buffer_pe(int size)
838 {
839 struct per_thread_data * ptb = get_per_thread_data();
840 if (ptb->pe_buffer)
841 {
842 if (ptb->pe_len >= size ) return ptb->pe_buffer;
843 HeapFree( GetProcessHeap(), 0, ptb->pe_buffer );
844 }
845 ptb->pe_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->pe_len = size) );
846 if (!ptb->pe_buffer) SetLastError(WSAENOBUFS);
847 return ptb->pe_buffer;
848 }
849
850 /* ----------------------------------- i/o APIs */
851
852 #ifdef HAVE_IPX
853 #define SUPPORTED_PF(pf) ((pf)==WS_AF_INET || (pf)== WS_AF_IPX || (pf) == WS_AF_INET6)
854 #else
855 #define SUPPORTED_PF(pf) ((pf)==WS_AF_INET || (pf) == WS_AF_INET6)
856 #endif
857
858
859 /**********************************************************************/
860
861 /* Returns the length of the converted address if successful, 0 if it was too small to
862 * start with.
863 */
864 static unsigned int ws_sockaddr_ws2u(const struct WS_sockaddr* wsaddr, int wsaddrlen,
865 union generic_unix_sockaddr *uaddr)
866 {
867 unsigned int uaddrlen = 0;
868
869 switch (wsaddr->sa_family)
870 {
871 #ifdef HAVE_IPX
872 case WS_AF_IPX:
873 {
874 const struct WS_sockaddr_ipx* wsipx=(const struct WS_sockaddr_ipx*)wsaddr;
875 struct sockaddr_ipx* uipx = (struct sockaddr_ipx *)uaddr;
876
877 if (wsaddrlen<sizeof(struct WS_sockaddr_ipx))
878 return 0;
879
880 uaddrlen = sizeof(struct sockaddr_ipx);
881 memset( uaddr, 0, uaddrlen );
882 uipx->sipx_family=AF_IPX;
883 uipx->sipx_port=wsipx->sa_socket;
884 /* copy sa_netnum and sa_nodenum to sipx_network and sipx_node
885 * in one go
886 */
887 memcpy(&uipx->sipx_network,wsipx->sa_netnum,sizeof(uipx->sipx_network)+sizeof(uipx->sipx_node));
888 #ifdef IPX_FRAME_NONE
889 uipx->sipx_type=IPX_FRAME_NONE;
890 #endif
891 break;
892 }
893 #endif
894 case WS_AF_INET6: {
895 struct sockaddr_in6* uin6 = (struct sockaddr_in6 *)uaddr;
896 const struct WS_sockaddr_in6* win6 = (const struct WS_sockaddr_in6*)wsaddr;
897
898 /* Note: Windows has 2 versions of the sockaddr_in6 struct, one with
899 * scope_id, one without.
900 */
901 if (wsaddrlen >= sizeof(struct WS_sockaddr_in6_old)) {
902 uaddrlen = sizeof(struct sockaddr_in6);
903 memset( uaddr, 0, uaddrlen );
904 uin6->sin6_family = AF_INET6;
905 uin6->sin6_port = win6->sin6_port;
906 uin6->sin6_flowinfo = win6->sin6_flowinfo;
907 #ifdef HAVE_STRUCT_SOCKADDR_IN6_SIN6_SCOPE_ID
908 if (wsaddrlen >= sizeof(struct WS_sockaddr_in6)) uin6->sin6_scope_id = win6->sin6_scope_id;
909 #endif
910 memcpy(&uin6->sin6_addr,&win6->sin6_addr,16); /* 16 bytes = 128 address bits */
911 break;
912 }
913 FIXME("bad size %d for WS_sockaddr_in6\n",wsaddrlen);
914 return 0;
915 }
916 case WS_AF_INET: {
917 struct sockaddr_in* uin = (struct sockaddr_in *)uaddr;
918 const struct WS_sockaddr_in* win = (const struct WS_sockaddr_in*)wsaddr;
919
920 if (wsaddrlen<sizeof(struct WS_sockaddr_in))
921 return 0;
922 uaddrlen = sizeof(struct sockaddr_in);
923 memset( uaddr, 0, uaddrlen );
924 uin->sin_family = AF_INET;
925 uin->sin_port = win->sin_port;
926 memcpy(&uin->sin_addr,&win->sin_addr,4); /* 4 bytes = 32 address bits */
927 break;
928 }
929 case WS_AF_UNSPEC: {
930 /* Try to determine the needed space by the passed windows sockaddr space */
931 switch (wsaddrlen) {
932 default: /* likely a ipv4 address */
933 case sizeof(struct WS_sockaddr_in):
934 uaddrlen = sizeof(struct sockaddr_in);
935 break;
936 #ifdef HAVE_IPX
937 case sizeof(struct WS_sockaddr_ipx):
938 uaddrlen = sizeof(struct sockaddr_ipx);
939 break;
940 #endif
941 case sizeof(struct WS_sockaddr_in6):
942 case sizeof(struct WS_sockaddr_in6_old):
943 uaddrlen = sizeof(struct sockaddr_in6);
944 break;
945 }
946 memset( uaddr, 0, uaddrlen );
947 break;
948 }
949 default:
950 FIXME("Unknown address family %d, return NULL.\n", wsaddr->sa_family);
951 return 0;
952 }
953 return uaddrlen;
954 }
955
956 static BOOL is_sockaddr_bound(const struct sockaddr *uaddr, int uaddrlen)
957 {
958 switch (uaddr->sa_family)
959 {
960 #ifdef HAVE_IPX
961 case AF_IPX:
962 FIXME("don't know how to tell if IPX socket is bound, assuming it is!\n");
963 return TRUE;
964 #endif
965 case AF_INET6:
966 {
967 static const struct sockaddr_in6 emptyAddr;
968 const struct sockaddr_in6 *in6 = (const struct sockaddr_in6*) uaddr;
969 return in6->sin6_port || memcmp(&in6->sin6_addr, &emptyAddr.sin6_addr, sizeof(struct in6_addr));
970 }
971 case AF_INET:
972 {
973 static const struct sockaddr_in emptyAddr;
974 const struct sockaddr_in *in = (const struct sockaddr_in*) uaddr;
975 return in->sin_port || memcmp(&in->sin_addr, &emptyAddr.sin_addr, sizeof(struct in_addr));
976 }
977 case AF_UNSPEC:
978 return FALSE;
979 default:
980 FIXME("unknown address family %d\n", uaddr->sa_family);
981 return TRUE;
982 }
983 }
984
985 /* Returns 0 if successful, -1 if the buffer is too small */
986 static int ws_sockaddr_u2ws(const struct sockaddr* uaddr, struct WS_sockaddr* wsaddr, int* wsaddrlen)
987 {
988 int res;
989
990 switch(uaddr->sa_family)
991 {
992 #ifdef HAVE_IPX
993 case AF_IPX:
994 {
995 const struct sockaddr_ipx* uipx=(const struct sockaddr_ipx*)uaddr;
996 struct WS_sockaddr_ipx* wsipx=(struct WS_sockaddr_ipx*)wsaddr;
997
998 res=-1;
999 switch (*wsaddrlen) /* how much can we copy? */
1000 {
1001 default:
1002 res=0; /* enough */
1003 *wsaddrlen = sizeof(*wsipx);
1004 wsipx->sa_socket=uipx->sipx_port;
1005 /* fall through */
1006 case 13:
1007 case 12:
1008 memcpy(wsipx->sa_nodenum,uipx->sipx_node,sizeof(wsipx->sa_nodenum));
1009 /* fall through */
1010 case 11:
1011 case 10:
1012 case 9:
1013 case 8:
1014 case 7:
1015 case 6:
1016 memcpy(wsipx->sa_netnum,&uipx->sipx_network,sizeof(wsipx->sa_netnum));
1017 /* fall through */
1018 case 5:
1019 case 4:
1020 case 3:
1021 case 2:
1022 wsipx->sa_family=WS_AF_IPX;
1023 /* fall through */
1024 case 1:
1025 case 0:
1026 /* way too small */
1027 break;
1028 }
1029 }
1030 break;
1031 #endif
1032 case AF_INET6: {
1033 const struct sockaddr_in6* uin6 = (const struct sockaddr_in6*)uaddr;
1034 struct WS_sockaddr_in6_old* win6old = (struct WS_sockaddr_in6_old*)wsaddr;
1035
1036 if (*wsaddrlen < sizeof(struct WS_sockaddr_in6_old))
1037 return -1;
1038 win6old->sin6_family = WS_AF_INET6;
1039 win6old->sin6_port = uin6->sin6_port;
1040 win6old->sin6_flowinfo = uin6->sin6_flowinfo;
1041 memcpy(&win6old->sin6_addr,&uin6->sin6_addr,16); /* 16 bytes = 128 address bits */
1042 *wsaddrlen = sizeof(struct WS_sockaddr_in6_old);
1043 #ifdef HAVE_STRUCT_SOCKADDR_IN6_SIN6_SCOPE_ID
1044 if (*wsaddrlen >= sizeof(struct WS_sockaddr_in6)) {
1045 struct WS_sockaddr_in6* win6 = (struct WS_sockaddr_in6*)wsaddr;
1046 win6->sin6_scope_id = uin6->sin6_scope_id;
1047 *wsaddrlen = sizeof(struct WS_sockaddr_in6);
1048 }
1049 #endif
1050 return 0;
1051 }
1052 case AF_INET: {
1053 const struct sockaddr_in* uin = (const struct sockaddr_in*)uaddr;
1054 struct WS_sockaddr_in* win = (struct WS_sockaddr_in*)wsaddr;
1055
1056 if (*wsaddrlen < sizeof(struct WS_sockaddr_in))
1057 return -1;
1058 win->sin_family = WS_AF_INET;
1059 win->sin_port = uin->sin_port;
1060 memcpy(&win->sin_addr,&uin->sin_addr,4); /* 4 bytes = 32 address bits */
1061 memset(win->sin_zero, 0, 8); /* Make sure the null padding is null */
1062 *wsaddrlen = sizeof(struct WS_sockaddr_in);
1063 return 0;
1064 }
1065 case AF_UNSPEC: {
1066 memset(wsaddr,0,*wsaddrlen);
1067 return 0;
1068 }
1069 default:
1070 FIXME("Unknown address family %d\n", uaddr->sa_family);
1071 return -1;
1072 }
1073 return res;
1074 }
1075
1076 /**************************************************************************
1077 * Functions for handling overlapped I/O
1078 **************************************************************************/
1079
1080 /* user APC called upon async completion */
1081 static void WINAPI ws2_async_apc( void *arg, IO_STATUS_BLOCK *iosb, ULONG reserved )
1082 {
1083 ws2_async *wsa = arg;
1084
1085 if (wsa->completion_func) wsa->completion_func( NtStatusToWSAError(iosb->u.Status),
1086 iosb->Information, wsa->user_overlapped,
1087 wsa->flags );
1088 HeapFree( GetProcessHeap(), 0, wsa );
1089 }
1090
1091 /***********************************************************************
1092 * WS2_recv (INTERNAL)
1093 *
1094 * Workhorse for both synchronous and asynchronous recv() operations.
1095 */
1096 static int WS2_recv( int fd, struct ws2_async *wsa )
1097 {
1098 struct msghdr hdr;
1099 union generic_unix_sockaddr unix_sockaddr;
1100 int n;
1101
1102 hdr.msg_name = NULL;
1103
1104 if (wsa->addr)
1105 {
1106 hdr.msg_namelen = sizeof(unix_sockaddr);
1107 hdr.msg_name = &unix_sockaddr;
1108 }
1109 else
1110 hdr.msg_namelen = 0;
1111
1112 hdr.msg_iov = wsa->iovec + wsa->first_iovec;
1113 hdr.msg_iovlen = wsa->n_iovecs - wsa->first_iovec;
1114 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
1115 hdr.msg_accrights = NULL;
1116 hdr.msg_accrightslen = 0;
1117 #else
1118 hdr.msg_control = NULL;
1119 hdr.msg_controllen = 0;
1120 hdr.msg_flags = 0;
1121 #endif
1122
1123 if ( (n = recvmsg(fd, &hdr, wsa->flags)) == -1 )
1124 return -1;
1125
1126 /* if this socket is connected and lpFrom is not NULL, Linux doesn't give us
1127 * msg_name and msg_namelen from recvmsg, but it does set msg_namelen to zero.
1128 *
1129 * quoting linux 2.6 net/ipv4/tcp.c:
1130 * "According to UNIX98, msg_name/msg_namelen are ignored
1131 * on connected socket. I was just happy when found this 8) --ANK"
1132 *
1133 * likewise MSDN says that lpFrom and lpFromlen are ignored for
1134 * connection-oriented sockets, so don't try to update lpFrom.
1135 */
1136 if (wsa->addr && hdr.msg_namelen)
1137 ws_sockaddr_u2ws( &unix_sockaddr.addr, wsa->addr, wsa->addrlen.ptr );
1138
1139 return n;
1140 }
1141
1142 /***********************************************************************
1143 * WS2_async_recv (INTERNAL)
1144 *
1145 * Handler for overlapped recv() operations.
1146 */
1147 static NTSTATUS WS2_async_recv( void* user, IO_STATUS_BLOCK* iosb, NTSTATUS status, void **apc)
1148 {
1149 ws2_async* wsa = user;
1150 int result = 0, fd;
1151
1152 switch (status)
1153 {
1154 case STATUS_ALERTED:
1155 if ((status = wine_server_handle_to_fd( wsa->hSocket, FILE_READ_DATA, &fd, NULL ) ))
1156 break;
1157
1158 result = WS2_recv( fd, wsa );
1159 wine_server_release_fd( wsa->hSocket, fd );
1160 if (result >= 0)
1161 {
1162 status = STATUS_SUCCESS;
1163 _enable_event( wsa->hSocket, FD_READ, 0, 0 );
1164 }
1165 else
1166 {
1167 if (errno == EINTR || errno == EAGAIN)
1168 {
1169 status = STATUS_PENDING;
1170 _enable_event( wsa->hSocket, FD_READ, 0, 0 );
1171 }
1172 else
1173 {
1174 result = 0;
1175 status = wsaErrno(); /* FIXME: is this correct ???? */
1176 }
1177 }
1178 break;
1179 }
1180 if (status != STATUS_PENDING)
1181 {
1182 iosb->u.Status = status;
1183 iosb->Information = result;
1184 *apc = ws2_async_apc;
1185 }
1186 return status;
1187 }
1188
1189 /***********************************************************************
1190 * WS2_send (INTERNAL)
1191 *
1192 * Workhorse for both synchronous and asynchronous send() operations.
1193 */
1194 static int WS2_send( int fd, struct ws2_async *wsa )
1195 {
1196 struct msghdr hdr;
1197 union generic_unix_sockaddr unix_addr;
1198
1199 hdr.msg_name = NULL;
1200 hdr.msg_namelen = 0;
1201
1202 if (wsa->addr)
1203 {
1204 hdr.msg_name = &unix_addr;
1205 hdr.msg_namelen = ws_sockaddr_ws2u( wsa->addr, wsa->addrlen.val, &unix_addr );
1206 if ( !hdr.msg_namelen )
1207 {
1208 errno = EFAULT;
1209 return -1;
1210 }
1211
1212 #if defined(HAVE_IPX) && defined(SOL_IPX)
1213 if(wsa->addr->sa_family == WS_AF_IPX)
1214 {
1215 struct sockaddr_ipx* uipx = (struct sockaddr_ipx*)hdr.msg_name;
1216 int val=0;
1217 unsigned int len=sizeof(int);
1218
1219 /* The packet type is stored at the ipx socket level; At least the linux kernel seems
1220 * to do something with it in case hdr.msg_name is NULL. Nonetheless can we use it to store
1221 * the packet type and then we can retrieve it using getsockopt. After that we can set the
1222 * ipx type in the sockaddr_opx structure with the stored value.
1223 */
1224 if(getsockopt(fd, SOL_IPX, IPX_TYPE, &val, &len) != -1)
1225 uipx->sipx_type = val;
1226 }
1227 #endif
1228 }
1229
1230 hdr.msg_iov = wsa->iovec + wsa->first_iovec;
1231 hdr.msg_iovlen = wsa->n_iovecs - wsa->first_iovec;
1232 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
1233 hdr.msg_accrights = NULL;
1234 hdr.msg_accrightslen = 0;
1235 #else
1236 hdr.msg_control = NULL;
1237 hdr.msg_controllen = 0;
1238 hdr.msg_flags = 0;
1239 #endif
1240
1241 return sendmsg(fd, &hdr, wsa->flags);
1242 }
1243
1244 /***********************************************************************
1245 * WS2_async_send (INTERNAL)
1246 *
1247 * Handler for overlapped send() operations.
1248 */
1249 static NTSTATUS WS2_async_send(void* user, IO_STATUS_BLOCK* iosb, NTSTATUS status, void **apc)
1250 {
1251 ws2_async* wsa = user;
1252 int result = 0, fd;
1253
1254 switch (status)
1255 {
1256 case STATUS_ALERTED:
1257 if ((status = wine_server_handle_to_fd( wsa->hSocket, FILE_WRITE_DATA, &fd, NULL ) ))
1258 break;
1259
1260 /* check to see if the data is ready (non-blocking) */
1261 result = WS2_send( fd, wsa );
1262 wine_server_release_fd( wsa->hSocket, fd );
1263
1264 if (result >= 0)
1265 {
1266 int totalLength = 0;
1267 unsigned int i;
1268 status = STATUS_SUCCESS;
1269 for (i = 0; i < wsa->n_iovecs; i++)
1270 totalLength += wsa->iovec[i].iov_len;
1271 if (result < totalLength)
1272 _enable_event( wsa->hSocket, FD_WRITE, 0, 0 );
1273 }
1274 else
1275 {
1276 if (errno == EINTR || errno == EAGAIN)
1277 {
1278 status = STATUS_PENDING;
1279 _enable_event( wsa->hSocket, FD_WRITE, 0, 0 );
1280 }
1281 else
1282 {
1283 /* We set the status to a winsock error code and check for that
1284 later in NtStatusToWSAError () */
1285 status = wsaErrno();
1286 result = 0;
1287 }
1288 }
1289 break;
1290 }
1291 if (status != STATUS_PENDING)
1292 {
1293 iosb->u.Status = status;
1294 iosb->Information = result;
1295 *apc = ws2_async_apc;
1296 }
1297 return status;
1298 }
1299
1300 /***********************************************************************
1301 * WS2_async_shutdown (INTERNAL)
1302 *
1303 * Handler for shutdown() operations on overlapped sockets.
1304 */
1305 static NTSTATUS WS2_async_shutdown( void* user, PIO_STATUS_BLOCK iosb, NTSTATUS status, void **apc )
1306 {
1307 ws2_async* wsa = user;
1308 int fd, err = 1;
1309
1310 switch (status)
1311 {
1312 case STATUS_ALERTED:
1313 if ((status = wine_server_handle_to_fd( wsa->hSocket, 0, &fd, NULL ) ))
1314 break;
1315
1316 switch ( wsa->type )
1317 {
1318 case ASYNC_TYPE_READ: err = shutdown( fd, 0 ); break;
1319 case ASYNC_TYPE_WRITE: err = shutdown( fd, 1 ); break;
1320 }
1321 wine_server_release_fd( wsa->hSocket, fd );
1322 status = err ? wsaErrno() : STATUS_SUCCESS;
1323 break;
1324 }
1325 iosb->u.Status = status;
1326 *apc = ws2_async_apc;
1327 return status;
1328 }
1329
1330 /***********************************************************************
1331 * WS2_register_async_shutdown (INTERNAL)
1332 *
1333 * Helper function for WS_shutdown() on overlapped sockets.
1334 */
1335 static int WS2_register_async_shutdown( SOCKET s, int type )
1336 {
1337 struct ws2_async *wsa;
1338 NTSTATUS status;
1339
1340 TRACE("s %ld type %d\n", s, type);
1341
1342 wsa = HeapAlloc( GetProcessHeap(), 0, sizeof(*wsa) );
1343 if ( !wsa )
1344 return WSAEFAULT;
1345
1346 wsa->hSocket = SOCKET2HANDLE(s);
1347 wsa->type = type;
1348 wsa->completion_func = NULL;
1349
1350 SERVER_START_REQ( register_async )
1351 {
1352 req->type = type;
1353 req->async.handle = wine_server_obj_handle( wsa->hSocket );
1354 req->async.callback = wine_server_client_ptr( WS2_async_shutdown );
1355 req->async.iosb = wine_server_client_ptr( &wsa->local_iosb );
1356 req->async.arg = wine_server_client_ptr( wsa );
1357 req->async.cvalue = 0;
1358 status = wine_server_call( req );
1359 }
1360 SERVER_END_REQ;
1361
1362 if (status != STATUS_PENDING)
1363 {
1364 HeapFree( GetProcessHeap(), 0, wsa );
1365 return NtStatusToWSAError( status );
1366 }
1367 return 0;
1368 }
1369
1370 /***********************************************************************
1371 * accept (WS2_32.1)
1372 */
1373 SOCKET WINAPI WS_accept(SOCKET s, struct WS_sockaddr *addr,
1374 int *addrlen32)
1375 {
1376 SOCKET as;
1377 BOOL is_blocking;
1378
1379 TRACE("socket %04lx\n", s );
1380 is_blocking = _is_blocking(s);
1381
1382 do {
1383 if (is_blocking)
1384 {
1385 int fd = get_sock_fd( s, FILE_READ_DATA, NULL );
1386 if (fd == -1) return INVALID_SOCKET;
1387 /* block here */
1388 do_block(fd, POLLIN, -1);
1389 _sync_sock_state(s); /* let wineserver notice connection */
1390 release_sock_fd( s, fd );
1391 /* retrieve any error codes from it */
1392 SetLastError(_get_sock_error(s, FD_ACCEPT_BIT));
1393 /* FIXME: care about the error? */
1394 }
1395 SERVER_START_REQ( accept_socket )
1396 {
1397 req->lhandle = wine_server_obj_handle( SOCKET2HANDLE(s) );
1398 req->access = GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE;
1399 req->attributes = OBJ_INHERIT;
1400 set_error( wine_server_call( req ) );
1401 as = HANDLE2SOCKET( wine_server_ptr_handle( reply->handle ));
1402 }
1403 SERVER_END_REQ;
1404 if (as)
1405 {
1406 if (addr) WS_getpeername(as, addr, addrlen32);
1407 return as;
1408 }
1409 } while (is_blocking);
1410 return INVALID_SOCKET;
1411 }
1412
1413 /***********************************************************************
1414 * bind (WS2_32.2)
1415 */
1416 int WINAPI WS_bind(SOCKET s, const struct WS_sockaddr* name, int namelen)
1417 {
1418 int fd = get_sock_fd( s, 0, NULL );
1419 int res = SOCKET_ERROR;
1420
1421 TRACE("socket %04lx, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen);
1422
1423 if (fd != -1)
1424 {
1425 if (!name || (name->sa_family && !SUPPORTED_PF(name->sa_family)))
1426 {
1427 SetLastError(WSAEAFNOSUPPORT);
1428 }
1429 else
1430 {
1431 union generic_unix_sockaddr uaddr;
1432 unsigned int uaddrlen = ws_sockaddr_ws2u(name, namelen, &uaddr);
1433 if (!uaddrlen)
1434 {
1435 SetLastError(WSAEFAULT);
1436 }
1437 else
1438 {
1439 #ifdef IPV6_V6ONLY
1440 const struct sockaddr_in6 *in6 = (const struct sockaddr_in6*) &uaddr;
1441 if (name->sa_family == WS_AF_INET6 &&
1442 !memcmp(&in6->sin6_addr, &in6addr_any, sizeof(struct in6_addr)))
1443 {
1444 int enable = 1;
1445 if (setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, &enable, sizeof(enable)) == -1)
1446 {
1447 release_sock_fd( s, fd );
1448 SetLastError(WSAEAFNOSUPPORT);
1449 return SOCKET_ERROR;
1450 }
1451 }
1452 #endif
1453 if (name->sa_family == WS_AF_INET)
1454 {
1455 struct sockaddr_in *in4 = (struct sockaddr_in*) &uaddr;
1456 if (memcmp(&in4->sin_addr, magic_loopback_addr, 4) == 0)
1457 {
1458 /* Trying to bind to the default host interface, using
1459 * INADDR_ANY instead*/
1460 WARN("Trying to bind to magic IP address, using "
1461 "INADDR_ANY instead.\n");
1462 in4->sin_addr.s_addr = htonl(WS_INADDR_ANY);
1463 }
1464 }
1465 if (bind(fd, &uaddr.addr, uaddrlen) < 0)
1466 {
1467 int loc_errno = errno;
1468 WARN("\tfailure - errno = %i\n", errno);
1469 errno = loc_errno;
1470 switch (errno)
1471 {
1472 case EBADF:
1473 SetLastError(WSAENOTSOCK);
1474 break;
1475 case EADDRNOTAVAIL:
1476 SetLastError(WSAEINVAL);
1477 break;
1478 default:
1479 SetLastError(wsaErrno());
1480 break;
1481 }
1482 }
1483 else
1484 {
1485 res=0; /* success */
1486 }
1487 }
1488 }
1489 release_sock_fd( s, fd );
1490 }
1491 return res;
1492 }
1493
1494 /***********************************************************************
1495 * closesocket (WS2_32.3)
1496 */
1497 int WINAPI WS_closesocket(SOCKET s)
1498 {
1499 TRACE("socket %04lx\n", s);
1500 if (CloseHandle(SOCKET2HANDLE(s))) return 0;
1501 return SOCKET_ERROR;
1502 }
1503
1504 /***********************************************************************
1505 * connect (WS2_32.4)
1506 */
1507 int WINAPI WS_connect(SOCKET s, const struct WS_sockaddr* name, int namelen)
1508 {
1509 int fd = get_sock_fd( s, FILE_READ_DATA, NULL );
1510
1511 TRACE("socket %04lx, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen);
1512
1513 if (fd != -1)
1514 {
1515 union generic_unix_sockaddr uaddr;
1516 unsigned int uaddrlen = ws_sockaddr_ws2u(name, namelen, &uaddr);
1517
1518 if (!uaddrlen)
1519 {
1520 SetLastError(WSAEFAULT);
1521 }
1522 else
1523 {
1524 if (name->sa_family == WS_AF_INET)
1525 {
1526 struct sockaddr_in *in4 = (struct sockaddr_in*) &uaddr;
1527 if (memcmp(&in4->sin_addr, magic_loopback_addr, 4) == 0)
1528 {
1529 /* Trying to connect to magic replace-loopback address,
1530 * assuming we really want to connect to localhost */
1531 TRACE("Trying to connect to magic IP address, using "
1532 "INADDR_LOOPBACK instead.\n");
1533 in4->sin_addr.s_addr = htonl(WS_INADDR_LOOPBACK);
1534 }
1535 }
1536
1537 if (connect(fd, &uaddr.addr, uaddrlen) == 0)
1538 goto connect_success;
1539 }
1540
1541 if (errno == EINPROGRESS)
1542 {
1543 /* tell wineserver that a connection is in progress */
1544 _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
1545 FD_CONNECT|FD_READ|FD_WRITE,
1546 FD_WINE_CONNECTED|FD_WINE_LISTENING);
1547 if (_is_blocking(s))
1548 {
1549 int result;
1550 /* block here */
1551 do_block(fd, POLLIN | POLLOUT, -1);
1552 _sync_sock_state(s); /* let wineserver notice connection */
1553 /* retrieve any error codes from it */
1554 result = _get_sock_error(s, FD_CONNECT_BIT);
1555 if (result)
1556 SetLastError(result);
1557 else
1558 {
1559 goto connect_success;
1560 }
1561 }
1562 else
1563 {
1564 SetLastError(WSAEWOULDBLOCK);
1565 }
1566 }
1567 else
1568 {
1569 SetLastError(wsaErrno());
1570 }
1571 release_sock_fd( s, fd );
1572 }
1573 return SOCKET_ERROR;
1574
1575 connect_success:
1576 release_sock_fd( s, fd );
1577 _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
1578 FD_WINE_CONNECTED|FD_READ|FD_WRITE,
1579 FD_CONNECT|FD_WINE_LISTENING);
1580 return 0;
1581 }
1582
1583 /***********************************************************************
1584 * WSAConnect (WS2_32.30)
1585 */
1586 int WINAPI WSAConnect( SOCKET s, const struct WS_sockaddr* name, int namelen,
1587 LPWSABUF lpCallerData, LPWSABUF lpCalleeData,
1588 LPQOS lpSQOS, LPQOS lpGQOS )
1589 {
1590 if ( lpCallerData || lpCalleeData || lpSQOS || lpGQOS )
1591 FIXME("unsupported parameters!\n");
1592 return WS_connect( s, name, namelen );
1593 }
1594
1595
1596 /***********************************************************************
1597 * getpeername (WS2_32.5)
1598 */
1599 int WINAPI WS_getpeername(SOCKET s, struct WS_sockaddr *name, int *namelen)
1600 {
1601 int fd;
1602 int res;
1603
1604 TRACE("socket: %04lx, ptr %p, len %08x\n", s, name, *namelen);
1605
1606 /* Check if what we've received is valid. Should we use IsBadReadPtr? */
1607 if( (name == NULL) || (namelen == NULL) )
1608 {
1609 SetLastError( WSAEFAULT );
1610 return SOCKET_ERROR;
1611 }
1612
1613 fd = get_sock_fd( s, 0, NULL );
1614 res = SOCKET_ERROR;
1615
1616 if (fd != -1)
1617 {
1618 union generic_unix_sockaddr uaddr;
1619 unsigned int uaddrlen = sizeof(uaddr);
1620
1621 if (getpeername(fd, &uaddr.addr, &uaddrlen) != 0)
1622 {
1623 SetLastError(wsaErrno());
1624 }
1625 else if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0)
1626 {
1627 /* The buffer was too small */
1628 SetLastError(WSAEFAULT);
1629 }
1630 else
1631 {
1632 res=0;
1633 }
1634 release_sock_fd( s, fd );
1635 }
1636 return res;
1637 }
1638
1639 /***********************************************************************
1640 * getsockname (WS2_32.6)
1641 */
1642 int WINAPI WS_getsockname(SOCKET s, struct WS_sockaddr *name, int *namelen)
1643 {
1644 int fd;
1645 int res;
1646
1647 TRACE("socket: %04lx, ptr %p, len %8x\n", s, name, *namelen);
1648
1649 /* Check if what we've received is valid. Should we use IsBadReadPtr? */
1650 if( (name == NULL) || (namelen == NULL) )
1651 {
1652 SetLastError( WSAEFAULT );
1653 return SOCKET_ERROR;
1654 }
1655
1656 fd = get_sock_fd( s, 0, NULL );
1657 res = SOCKET_ERROR;
1658
1659 if (fd != -1)
1660 {
1661 union generic_unix_sockaddr uaddr;
1662 unsigned int uaddrlen = sizeof(uaddr);
1663
1664 if (getsockname(fd, &uaddr.addr, &uaddrlen) != 0)
1665 {
1666 SetLastError(wsaErrno());
1667 }
1668 else if (!is_sockaddr_bound(&uaddr.addr, uaddrlen))
1669 {
1670 SetLastError(WSAEINVAL);
1671 }
1672 else if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0)
1673 {
1674 /* The buffer was too small */
1675 SetLastError(WSAEFAULT);
1676 }
1677 else
1678 {
1679 res=0;
1680 }
1681 release_sock_fd( s, fd );
1682 }
1683 return res;
1684 }
1685
1686 /***********************************************************************
1687 * getsockopt (WS2_32.7)
1688 */
1689 INT WINAPI WS_getsockopt(SOCKET s, INT level,
1690 INT optname, char *optval, INT *optlen)
1691 {
1692 int fd;
1693 INT ret = 0;
1694
1695 TRACE("socket: %04lx, level 0x%x, name 0x%x, ptr %p, len %d\n",
1696 s, level, optname, optval, *optlen);
1697
1698 switch(level)
1699 {
1700 case WS_SOL_SOCKET:
1701 {
1702 switch(optname)
1703 {
1704 /* Handle common cases. The special cases are below, sorted
1705 * alphabetically */
1706 case WS_SO_ACCEPTCONN:
1707 case WS_SO_BROADCAST:
1708 case WS_SO_DEBUG:
1709 case WS_SO_ERROR:
1710 case WS_SO_KEEPALIVE:
1711 case WS_SO_OOBINLINE:
1712 case WS_SO_RCVBUF:
1713 case WS_SO_REUSEADDR:
1714 case WS_SO_SNDBUF:
1715 case WS_SO_TYPE:
1716 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
1717 return SOCKET_ERROR;
1718 convert_sockopt(&level, &optname);
1719 if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
1720 {
1721 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
1722 ret = SOCKET_ERROR;
1723 }
1724 release_sock_fd( s, fd );
1725 return ret;
1726
1727 case WS_SO_DONTLINGER:
1728 {
1729 struct linger lingval;
1730 unsigned int len = sizeof(struct linger);
1731
1732 if (!optlen || *optlen < sizeof(BOOL)|| !optval)
1733 {
1734 SetLastError(WSAEFAULT);
1735 return SOCKET_ERROR;
1736 }
1737 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
1738 return SOCKET_ERROR;
1739
1740 if (getsockopt(fd, SOL_SOCKET, SO_LINGER, &lingval, &len) != 0 )
1741 {
1742 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
1743 ret = SOCKET_ERROR;
1744 }
1745 else
1746 {
1747 *(BOOL *)optval = (lingval.l_onoff) ? FALSE : TRUE;
1748 *optlen = sizeof(BOOL);
1749 }
1750
1751 release_sock_fd( s, fd );
1752 return ret;
1753 }
1754
1755 /* As mentioned in setsockopt, Windows ignores this, so we
1756 * always return true here */
1757 case WS_SO_DONTROUTE:
1758 if (!optlen || *optlen < sizeof(BOOL) || !optval)
1759 {
1760 SetLastError(WSAEFAULT);
1761 return SOCKET_ERROR;
1762 }
1763 *(BOOL *)optval = TRUE;
1764 *optlen = sizeof(BOOL);
1765 return 0;
1766
1767 case WS_SO_LINGER:
1768 {
1769 struct linger lingval;
1770 unsigned int len = sizeof(struct linger);
1771
1772 /* struct linger and LINGER have different sizes */
1773 if (!optlen || *optlen < sizeof(LINGER) || !optval)
1774 {
1775 SetLastError(WSAEFAULT);
1776 return SOCKET_ERROR;
1777 }
1778 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
1779 return SOCKET_ERROR;
1780
1781 if (getsockopt(fd, SOL_SOCKET, SO_LINGER, &lingval, &len) != 0 )
1782 {
1783 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
1784 ret = SOCKET_ERROR;
1785 }
1786 else
1787 {
1788 ((LINGER *)optval)->l_onoff = lingval.l_onoff;
1789 ((LINGER *)optval)->l_linger = lingval.l_linger;
1790 *optlen = sizeof(struct linger);
1791 }
1792
1793 release_sock_fd( s, fd );
1794 return ret;
1795 }
1796
1797 case WS_SO_MAX_MSG_SIZE:
1798 if (!optlen || *optlen < sizeof(int) || !optval)
1799 {
1800 SetLastError(WSAEFAULT);
1801 return SOCKET_ERROR;
1802 }
1803 TRACE("getting global SO_MAX_MSG_SIZE = 65507\n");
1804 *(int *)optval = 65507;
1805 *optlen = sizeof(int);
1806 return 0;
1807
1808 /* SO_OPENTYPE does not require a valid socket handle. */
1809 case WS_SO_OPENTYPE:
1810 if (!optlen || *optlen < sizeof(int) || !optval)
1811 {
1812 SetLastError(WSAEFAULT);
1813 return SOCKET_ERROR;
1814 }
1815 *(int *)optval = get_per_thread_data()->opentype;
1816 *optlen = sizeof(int);
1817 TRACE("getting global SO_OPENTYPE = 0x%x\n", *((int*)optval) );
1818 return 0;
1819
1820 #ifdef SO_RCVTIMEO
1821 case WS_SO_RCVTIMEO:
1822 #endif
1823 #ifdef SO_SNDTIMEO
1824 case WS_SO_SNDTIMEO:
1825 #endif
1826 #if defined(SO_RCVTIMEO) || defined(SO_SNDTIMEO)
1827 {
1828 struct timeval tv;
1829 unsigned int len = sizeof(struct timeval);
1830
1831 if (!optlen || *optlen < sizeof(int)|| !optval)
1832 {
1833 SetLastError(WSAEFAULT);
1834 return SOCKET_ERROR;
1835 }
1836 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
1837 return SOCKET_ERROR;
1838
1839 convert_sockopt(&level, &optname);
1840 if (getsockopt(fd, level, optname, &tv, &len) != 0 )
1841 {
1842 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
1843 ret = SOCKET_ERROR;
1844 }
1845 else
1846 {
1847 *(int *)optval = tv.tv_sec * 1000 + tv.tv_usec / 1000;
1848 *optlen = sizeof(int);
1849 }
1850
1851 release_sock_fd( s, fd );
1852 return ret;
1853 }
1854 #endif
1855 default:
1856 TRACE("Unknown SOL_SOCKET optname: 0x%08x\n", optname);
1857 SetLastError(WSAENOPROTOOPT);
1858 return SOCKET_ERROR;
1859 } /* end switch(optname) */
1860 }/* end case WS_SOL_SOCKET */
1861 #ifdef HAVE_IPX
1862 case NSPROTO_IPX:
1863 {
1864 struct WS_sockaddr_ipx addr;
1865 IPX_ADDRESS_DATA *data;
1866 int namelen;
1867 switch(optname)
1868 {
1869 case IPX_PTYPE:
1870 if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR;
1871 #ifdef SOL_IPX
1872 if(getsockopt(fd, SOL_IPX, IPX_TYPE, optval, (unsigned int*)optlen) == -1)
1873 {
1874 ret = SOCKET_ERROR;
1875 }
1876 #else
1877 {
1878 struct ipx val;
1879 socklen_t len=sizeof(struct ipx);
1880 if(getsockopt(fd, 0, SO_DEFAULT_HEADERS, &val, &len) == -1 )
1881 ret = SOCKET_ERROR;
1882 else
1883 *optval = (int)val.ipx_pt;
1884 }
1885 #endif
1886 TRACE("ptype: %d (fd: %d)\n", *(int*)optval, fd);
1887 release_sock_fd( s, fd );
1888 return ret;
1889
1890 case IPX_ADDRESS:
1891 /*
1892 * On a Win2000 system with one network card there are usually
1893 * three ipx devices one with a speed of 28.8kbps, 10Mbps and 100Mbps.
1894 * Using this call you can then retrieve info about this all.
1895 * In case of Linux it is a bit different. Usually you have
1896 * only "one" device active and further it is not possible to
1897 * query things like the linkspeed.
1898 */
1899 FIXME("IPX_ADDRESS\n");
1900 namelen = sizeof(struct WS_sockaddr_ipx);
1901 memset(&addr, 0, sizeof(struct WS_sockaddr_ipx));
1902 WS_getsockname(s, (struct WS_sockaddr*)&addr, &namelen);
1903
1904 data = (IPX_ADDRESS_DATA*)optval;
1905 memcpy(data->nodenum,addr.sa_nodenum,sizeof(data->nodenum));
1906 memcpy(data->netnum,addr.sa_netnum,sizeof(data->netnum));
1907 data->adapternum = 0;
1908 data->wan = FALSE; /* We are not on a wan for now .. */
1909 data->status = FALSE; /* Since we are not on a wan, the wan link isn't up */
1910 data->maxpkt = 1467; /* This value is the default one, at least on Win2k/WinXP */
1911 data->linkspeed = 100000; /* Set the line speed in 100bit/s to 10 Mbit;
1912 * note 1MB = 1000kB in this case */
1913 return 0;
1914
1915 case IPX_MAX_ADAPTER_NUM:
1916 FIXME("IPX_MAX_ADAPTER_NUM\n");
1917 *(int*)optval = 1; /* As noted under IPX_ADDRESS we have just one card. */
1918 return 0;
1919
1920 default:
1921 FIXME("IPX optname:%x\n", optname);
1922 return SOCKET_ERROR;
1923 }/* end switch(optname) */
1924 } /* end case NSPROTO_IPX */
1925 #endif
1926 /* Levels WS_IPPROTO_TCP and WS_IPPROTO_IP convert directly */
1927 case WS_IPPROTO_TCP:
1928 switch(optname)
1929 {
1930 case WS_TCP_NODELAY:
1931 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
1932 return SOCKET_ERROR;
1933 convert_sockopt(&level, &optname);
1934 if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
1935 {
1936 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
1937 ret = SOCKET_ERROR;
1938 }
1939 release_sock_fd( s, fd );
1940 return ret;
1941 }
1942 FIXME("Unknown IPPROTO_TCP optname 0x%08x\n", optname);
1943 return SOCKET_ERROR;
1944
1945 case WS_IPPROTO_IP:
1946 switch(optname)
1947 {
1948 case WS_IP_ADD_MEMBERSHIP:
1949 case WS_IP_DROP_MEMBERSHIP:
1950 #ifdef IP_HDRINCL
1951 case WS_IP_HDRINCL:
1952 #endif
1953 case WS_IP_MULTICAST_IF:
1954 case WS_IP_MULTICAST_LOOP:
1955 case WS_IP_MULTICAST_TTL:
1956 case WS_IP_OPTIONS:
1957 case WS_IP_TOS:
1958 case WS_IP_TTL:
1959 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
1960 return SOCKET_ERROR;
1961 convert_sockopt(&level, &optname);
1962 if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
1963 {
1964 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
1965 ret = SOCKET_ERROR;
1966 }
1967 release_sock_fd( s, fd );
1968 return ret;
1969 case WS_IP_DONTFRAGMENT:
1970 FIXME("WS_IP_DONTFRAGMENT is always false!\n");
1971 *(BOOL*)optval = FALSE;
1972 return 0;
1973 }
1974 FIXME("Unknown IPPROTO_IP optname 0x%08x\n", optname);
1975 return SOCKET_ERROR;
1976
1977 default:
1978 FIXME("Unknown level: 0x%08x\n", level);
1979 return SOCKET_ERROR;
1980 } /* end switch(level) */
1981 }
1982
1983 /***********************************************************************
1984 * htonl (WINSOCK.8)
1985 * htonl (WS2_32.8)
1986 */
1987 WS_u_long WINAPI WS_htonl(WS_u_long hostlong)
1988 {
1989 return htonl(hostlong);
1990 }
1991
1992
1993 /***********************************************************************
1994 * htons (WINSOCK.9)
1995 * htons (WS2_32.9)
1996 */
1997 WS_u_short WINAPI WS_htons(WS_u_short hostshort)
1998 {
1999 return htons(hostshort);
2000 }
2001
2002 /***********************************************************************
2003 * WSAHtonl (WS2_32.46)
2004 * From MSDN description of error codes, this function should also
2005 * check if WinSock has been initialized and the socket is a valid
2006 * socket. But why? This function only translates a host byte order
2007 * u_long into a network byte order u_long...
2008 */
2009 int WINAPI WSAHtonl(SOCKET s, WS_u_long hostlong, WS_u_long *lpnetlong)
2010 {
2011 if (lpnetlong)
2012 {
2013 *lpnetlong = htonl(hostlong);
2014 return 0;
2015 }
2016 WSASetLastError(WSAEFAULT);
2017 return SOCKET_ERROR;
2018 }
2019
2020 /***********************************************************************
2021 * WSAHtons (WS2_32.47)
2022 * From MSDN description of error codes, this function should also
2023 * check if WinSock has been initialized and the socket is a valid
2024 * socket. But why? This function only translates a host byte order
2025 * u_short into a network byte order u_short...
2026 */
2027 int WINAPI WSAHtons(SOCKET s, WS_u_short hostshort, WS_u_short *lpnetshort)
2028 {
2029
2030 if (lpnetshort)
2031 {
2032 *lpnetshort = htons(hostshort);
2033 return 0;
2034 }
2035 WSASetLastError(WSAEFAULT);
2036 return SOCKET_ERROR;
2037 }
2038
2039
2040 /***********************************************************************
2041 * inet_addr (WINSOCK.10)
2042 * inet_addr (WS2_32.11)
2043 */
2044 WS_u_long WINAPI WS_inet_addr(const char *cp)
2045 {
2046 if (!cp) return INADDR_NONE;
2047 return inet_addr(cp);
2048 }
2049
2050
2051 /***********************************************************************
2052 * ntohl (WINSOCK.14)
2053 * ntohl (WS2_32.14)
2054 */
2055 WS_u_long WINAPI WS_ntohl(WS_u_long netlong)
2056 {
2057 return ntohl(netlong);
2058 }
2059
2060
2061 /***********************************************************************
2062 * ntohs (WINSOCK.15)
2063 * ntohs (WS2_32.15)
2064 */
2065 WS_u_short WINAPI WS_ntohs(WS_u_short netshort)
2066 {
2067 return ntohs(netshort);
2068 }
2069
2070
2071 /***********************************************************************
2072 * inet_ntoa (WS2_32.12)
2073 */
2074 char* WINAPI WS_inet_ntoa(struct WS_in_addr in)
2075 {
2076 /* use "buffer for dummies" here because some applications have a
2077 * propensity to decode addresses in ws_hostent structure without
2078 * saving them first...
2079 */
2080 static char dbuffer[16]; /* Yes, 16: 4*3 digits + 3 '.' + 1 '\0' */
2081
2082 char* s = inet_ntoa(*((struct in_addr*)&in));
2083 if( s )
2084 {
2085 strcpy(dbuffer, s);
2086 return dbuffer;
2087 }
2088 SetLastError(wsaErrno());
2089 return NULL;
2090 }
2091
2092 /**********************************************************************
2093 * WSAIoctl (WS2_32.50)
2094 *
2095 */
2096 INT WINAPI WSAIoctl(SOCKET s,
2097 DWORD dwIoControlCode,
2098 LPVOID lpvInBuffer,
2099 DWORD cbInBuffer,
2100 LPVOID lpbOutBuffer,
2101 DWORD cbOutBuffer,
2102 LPDWORD lpcbBytesReturned,
2103 LPWSAOVERLAPPED lpOverlapped,
2104 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine)
2105 {
2106 TRACE("%ld, 0x%08x, %p, %d, %p, %d, %p, %p, %p\n",
2107 s, dwIoControlCode, lpvInBuffer, cbInBuffer, lpbOutBuffer,
2108 cbOutBuffer, lpcbBytesReturned, lpOverlapped, lpCompletionRoutine);
2109
2110 switch( dwIoControlCode )
2111 {
2112 case WS_FIONBIO:
2113 if (cbInBuffer != sizeof(WS_u_long)) {
2114 WSASetLastError(WSAEFAULT);
2115 return SOCKET_ERROR;
2116 }
2117 return WS_ioctlsocket( s, WS_FIONBIO, lpvInBuffer);
2118
2119 case WS_FIONREAD:
2120 if (cbOutBuffer != sizeof(WS_u_long)) {
2121 WSASetLastError(WSAEFAULT);
2122 return SOCKET_ERROR;
2123 }
2124 return WS_ioctlsocket( s, WS_FIONREAD, lpbOutBuffer);
2125
2126 case WS_SIO_GET_INTERFACE_LIST:
2127 {
2128 INTERFACE_INFO* intArray = (INTERFACE_INFO*)lpbOutBuffer;
2129 DWORD size, numInt, apiReturn;
2130 int fd;
2131
2132 TRACE("-> SIO_GET_INTERFACE_LIST request\n");
2133
2134 if (!lpbOutBuffer)
2135 {
2136 WSASetLastError(WSAEFAULT);
2137 return SOCKET_ERROR;
2138 }
2139 if (!lpcbBytesReturned)
2140 {
2141 WSASetLastError(WSAEFAULT);
2142 return SOCKET_ERROR;
2143 }
2144
2145 fd = get_sock_fd( s, 0, NULL );
2146 if (fd == -1) return SOCKET_ERROR;
2147
2148 apiReturn = GetAdaptersInfo(NULL, &size);
2149 if (apiReturn == ERROR_NO_DATA)
2150 {
2151 numInt = 0;
2152 }
2153 else if (apiReturn == ERROR_BUFFER_OVERFLOW)
2154 {
2155 PIP_ADAPTER_INFO table = HeapAlloc(GetProcessHeap(),0,size);
2156
2157 if (table)
2158 {
2159 if (GetAdaptersInfo(table, &size) == NO_ERROR)
2160 {
2161 PIP_ADAPTER_INFO ptr;
2162
2163 if (size*sizeof(INTERFACE_INFO)/sizeof(IP_ADAPTER_INFO) > cbOutBuffer)
2164 {
2165 WARN("Buffer too small = %u, cbOutBuffer = %u\n", size, cbOutBuffer);
2166 HeapFree(GetProcessHeap(),0,table);
2167 release_sock_fd( s, fd );
2168 WSASetLastError(WSAEFAULT);
2169 return SOCKET_ERROR;
2170 }
2171 for (ptr = table, numInt = 0; ptr;
2172 ptr = ptr->Next, intArray++, numInt++)
2173 {
2174 unsigned int addr, mask, bcast;
2175 struct ifreq ifInfo;
2176
2177 /* Socket Status Flags */
2178 lstrcpynA(ifInfo.ifr_name, ptr->AdapterName, IFNAMSIZ);
2179 if (ioctl(fd, SIOCGIFFLAGS, &ifInfo) < 0)
2180 {
2181 ERR("Error obtaining status flags for socket!\n");
2182 HeapFree(GetProcessHeap(),0,table);
2183 release_sock_fd( s, fd );
2184 WSASetLastError(WSAEINVAL);
2185 return SOCKET_ERROR;
2186 }
2187 else
2188 {
2189 /* set flags; the values of IFF_* are not the same
2190 under Linux and Windows, therefore must generate
2191 new flags */
2192 intArray->iiFlags = 0;
2193 if (ifInfo.ifr_flags & IFF_BROADCAST)
2194 intArray->iiFlags |= WS_IFF_BROADCAST;
2195 #ifdef IFF_POINTOPOINT
2196 if (ifInfo.ifr_flags & IFF_POINTOPOINT)
2197 intArray->iiFlags |= WS_IFF_POINTTOPOINT;
2198 #endif
2199 if (ifInfo.ifr_flags & IFF_LOOPBACK)
2200 intArray->iiFlags |= WS_IFF_LOOPBACK;
2201 if (ifInfo.ifr_flags & IFF_UP)
2202 intArray->iiFlags |= WS_IFF_UP;
2203 if (ifInfo.ifr_flags & IFF_MULTICAST)
2204 intArray->iiFlags |= WS_IFF_MULTICAST;
2205 }
2206
2207 addr = inet_addr(ptr->IpAddressList.IpAddress.String);
2208 mask = inet_addr(ptr->IpAddressList.IpMask.String);
2209 bcast = addr | ~mask;
2210 intArray->iiAddress.AddressIn.sin_family = AF_INET;
2211 intArray->iiAddress.AddressIn.sin_port = 0;
2212 intArray->iiAddress.AddressIn.sin_addr.WS_s_addr =
2213 addr;
2214 intArray->iiNetmask.AddressIn.sin_family = AF_INET;
2215 intArray->iiNetmask.AddressIn.sin_port = 0;
2216 intArray->iiNetmask.AddressIn.sin_addr.WS_s_addr =
2217 mask;
2218 intArray->iiBroadcastAddress.AddressIn.sin_family =
2219 AF_INET;
2220 intArray->iiBroadcastAddress.AddressIn.sin_port = 0;
2221 intArray->iiBroadcastAddress.AddressIn.sin_addr.
2222 WS_s_addr = bcast;
2223 }
2224 }
2225 else
2226 {
2227 ERR("Unable to get interface table!\n");
2228 release_sock_fd( s, fd );
2229 HeapFree(GetProcessHeap(),0,table);
2230 WSASetLastError(WSAEINVAL);
2231 return SOCKET_ERROR;
2232 }
2233 HeapFree(GetProcessHeap(),0,table);
2234 }
2235 else
2236 {
2237 release_sock_fd( s, fd );
2238 WSASetLastError(WSAEINVAL);
2239 return SOCKET_ERROR;
2240 }
2241 }
2242 else
2243 {
2244 ERR("Unable to get interface table!\n");
2245 release_sock_fd( s, fd );
2246 WSASetLastError(WSAEINVAL);
2247 return SOCKET_ERROR;
2248 }
2249 /* Calculate the size of the array being returned */
2250 *lpcbBytesReturned = sizeof(INTERFACE_INFO) * numInt;
2251 release_sock_fd( s, fd );
2252 break;
2253 }
2254
2255 case WS_SIO_ADDRESS_LIST_CHANGE:
2256 FIXME("-> SIO_ADDRESS_LIST_CHANGE request: stub\n");
2257 /* FIXME: error and return code depend on whether socket was created
2258 * with WSA_FLAG_OVERLAPPED, but there is no easy way to get this */
2259 break;
2260
2261 case WS_SIO_ADDRESS_LIST_QUERY:
2262 {
2263 DWORD size;
2264
2265 TRACE("-> SIO_ADDRESS_LIST_QUERY request\n");
2266
2267 if (!lpcbBytesReturned)
2268 {
2269 WSASetLastError(WSAEFAULT);
2270 return SOCKET_ERROR;
2271 }
2272
2273 if (GetAdaptersInfo(NULL, &size) == ERROR_BUFFER_OVERFLOW)
2274 {
2275 IP_ADAPTER_INFO *p, *table = HeapAlloc(GetProcessHeap(), 0, size);
2276 DWORD need, num;
2277
2278 if (!table || GetAdaptersInfo(table, &size))
2279 {
2280 HeapFree(GetProcessHeap(), 0, table);
2281 WSASetLastError(WSAEINVAL);
2282 return SOCKET_ERROR;
2283 }
2284
2285 for (p = table, num = 0; p; p = p->Next)
2286 if (p->IpAddressList.IpAddress.String[0]) num++;
2287
2288 need = sizeof(SOCKET_ADDRESS_LIST) + sizeof(SOCKET_ADDRESS) * (num - 1);
2289 need += sizeof(SOCKADDR) * num;
2290 *lpcbBytesReturned = need;
2291
2292 if (need > cbOutBuffer)
2293 {
2294 HeapFree(GetProcessHeap(), 0, table);
2295 WSASetLastError(WSAEFAULT);
2296 return SOCKET_ERROR;
2297 }
2298
2299 if (lpbOutBuffer)
2300 {
2301 unsigned int i;
2302 SOCKET_ADDRESS *sa;
2303 SOCKET_ADDRESS_LIST *sa_list = (SOCKET_ADDRESS_LIST *)lpbOutBuffer;
2304 SOCKADDR_IN *sockaddr;
2305
2306 sa = sa_list->Address;
2307 sockaddr = (SOCKADDR_IN *)((char *)sa + num * sizeof(SOCKET_ADDRESS));
2308 sa_list->iAddressCount = num;
2309
2310 for (p = table, i = 0; p; p = p->Next)
2311 {
2312 if (!p->IpAddressList.IpAddress.String[0]) continue;
2313
2314 sa[i].lpSockaddr = (SOCKADDR *)&sockaddr[i];
2315 sa[i].iSockaddrLength = sizeof(SOCKADDR);
2316
2317 sockaddr[i].sin_family = AF_INET;
2318 sockaddr[i].sin_port = 0;
2319 sockaddr[i].sin_addr.WS_s_addr = inet_addr(p->IpAddressList.IpAddress.String);
2320 i++;
2321 }
2322 }
2323
2324 HeapFree(GetProcessHeap(), 0, table);
2325 return 0;
2326 }
2327 else
2328 {
2329 WARN("unable to get IP address list\n");
2330 WSASetLastError(WSAEINVAL);
2331 return SOCKET_ERROR;
2332 }
2333 }
2334 case WS_SIO_FLUSH:
2335 FIXME("SIO_FLUSH: stub.\n");
2336 break;
2337
2338 case WS_SIO_GET_EXTENSION_FUNCTION_POINTER:
2339 FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER %s: stub\n", debugstr_guid(lpvInBuffer));
2340 WSASetLastError(WSAEOPNOTSUPP);
2341 return SOCKET_ERROR;
2342
2343 case WS_SIO_KEEPALIVE_VALS:
2344 {
2345 int fd;
2346 struct tcp_keepalive *k = lpvInBuffer;
2347 int keepalive = k->onoff ? 1 : 0;
2348 int keepidle = k->keepalivetime / 1000;
2349 int keepintvl = k->keepaliveinterval / 1000;
2350
2351 if (!lpvInBuffer)
2352 {
2353 WSASetLastError(WSAEINVAL);
2354 return SOCKET_ERROR;
2355 }
2356
2357 TRACE("onoff: %d, keepalivetime: %d, keepaliveinterval: %d\n", keepalive, keepidle, keepintvl);
2358
2359 fd = get_sock_fd(s, 0, NULL);
2360 if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, (void *)&keepalive, sizeof(int)) == -1)
2361 {
2362 release_sock_fd(s, fd);
2363 WSASetLastError(WSAEINVAL);
2364 return SOCKET_ERROR;
2365 }
2366 #if defined(TCP_KEEPIDLE) && defined(TCP_KEEPINTVL)
2367 if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPIDLE, (void *)&keepidle, sizeof(int)) == -1)
2368 {
2369 release_sock_fd(s, fd);
2370 WSASetLastError(WSAEINVAL);
2371 return SOCKET_ERROR;
2372 }
2373 if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPINTVL, (void *)&keepintvl, sizeof(int)) == -1)
2374 {
2375 release_sock_fd(s, fd);
2376 WSASetLastError(WSAEINVAL);
2377 return SOCKET_ERROR;
2378 }
2379 #else
2380 FIXME("ignoring keepalive interval and timeout\n");
2381 #endif
2382
2383 release_sock_fd(s, fd);
2384 break;
2385 }
2386 default:
2387 FIXME("unsupported WS_IOCTL cmd (%08x)\n", dwIoControlCode);
2388 WSASetLastError(WSAEOPNOTSUPP);
2389 return SOCKET_ERROR;
2390 }
2391
2392 return 0;
2393 }
2394
2395
2396 /***********************************************************************
2397 * ioctlsocket (WS2_32.10)
2398 */
2399 int WINAPI WS_ioctlsocket(SOCKET s, LONG cmd, WS_u_long *argp)
2400 {
2401 int fd;
2402 LONG newcmd = cmd;
2403
2404 TRACE("socket %04lx, cmd %08x, ptr %p\n", s, cmd, argp);
2405 /* broken apps like defcon pass the argp value directly instead of a pointer to it */
2406 if(IS_INTRESOURCE(argp))
2407 {
2408 SetLastError(WSAEFAULT);
2409 return SOCKET_ERROR;
2410 }
2411
2412 switch( cmd )
2413 {
2414 case WS_FIONREAD:
2415 newcmd=FIONREAD;
2416 break;
2417
2418 case WS_FIONBIO:
2419 if( _get_sock_mask(s) )
2420 {
2421 /* AsyncSelect()'ed sockets are always nonblocking */
2422 if (*argp) return 0;
2423 SetLastError(WSAEINVAL);
2424 return SOCKET_ERROR;
2425 }
2426 if (*argp)
2427 _enable_event(SOCKET2HANDLE(s), 0, FD_WINE_NONBLOCKING, 0);
2428 else
2429 _enable_event(SOCKET2HANDLE(s), 0, 0, FD_WINE_NONBLOCKING);
2430 return 0;
2431
2432 case WS_SIOCATMARK:
2433 newcmd=SIOCATMARK;
2434 break;
2435
2436 case WS_FIOASYNC:
2437 WARN("Warning: WS1.1 shouldn't be using async I/O\n");
2438 SetLastError(WSAEINVAL);
2439 return SOCKET_ERROR;
2440
2441 case SIOCGIFBRDADDR:
2442 case SIOCGIFNETMASK:
2443 case SIOCGIFADDR:
2444 /* These don't need any special handling. They are used by
2445 WsControl, and are here to suppress an unnecessary warning. */
2446 break;
2447
2448 default:
2449 /* Netscape tries hard to use bogus ioctl 0x667e */
2450 /* FIXME: 0x667e above is ('f' << 8) | 126, and is a low word of
2451 * FIONBIO (_IOW('f', 126, u_long)), how that should be handled?
2452 */
2453 WARN("\tunknown WS_IOCTL cmd (%08x)\n", cmd);
2454 break;
2455 }
2456
2457 fd = get_sock_fd( s, 0, NULL );
2458 if (fd != -1)
2459 {
2460 if( ioctl(fd, newcmd, (char*)argp ) == 0 )
2461 {
2462 release_sock_fd( s, fd );
2463 return 0;
2464 }
2465 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2466 release_sock_fd( s, fd );
2467 }
2468 return SOCKET_ERROR;
2469 }
2470
2471 /***********************************************************************
2472 * listen (WS2_32.13)
2473 */
2474 int WINAPI WS_listen(SOCKET s, int backlog)
2475 {
2476 int fd = get_sock_fd( s, FILE_READ_DATA, NULL );
2477
2478 TRACE("socket %04lx, backlog %d\n", s, backlog);
2479 if (fd != -1)
2480 {
2481 if (listen(fd, backlog) == 0)
2482 {
2483 release_sock_fd( s, fd );
2484 _enable_event(SOCKET2HANDLE(s), FD_ACCEPT,
2485 FD_WINE_LISTENING,
2486 FD_CONNECT|FD_WINE_CONNECTED);
2487 return 0;
2488 }
2489 SetLastError(wsaErrno());
2490 release_sock_fd( s, fd );
2491 }
2492 return SOCKET_ERROR;
2493 }
2494
2495 /***********************************************************************
2496 * recv (WS2_32.16)
2497 */
2498 int WINAPI WS_recv(SOCKET s, char *buf, int len, int flags)
2499 {
2500 DWORD n, dwFlags = flags;
2501 WSABUF wsabuf;
2502
2503 wsabuf.len = len;
2504 wsabuf.buf = buf;
2505
2506 if ( WSARecvFrom(s, &wsabuf, 1, &n, &dwFlags, NULL, NULL, NULL, NULL) == SOCKET_ERROR )
2507 return SOCKET_ERROR;
2508 else
2509 return n;
2510 }
2511
2512 /***********************************************************************
2513 * recvfrom (WS2_32.17)
2514 */
2515 int WINAPI WS_recvfrom(SOCKET s, char *buf, INT len, int flags,
2516 struct WS_sockaddr *from, int *fromlen)
2517 {
2518 DWORD n, dwFlags = flags;
2519 WSABUF wsabuf;
2520
2521 wsabuf.len = len;
2522 wsabuf.buf = buf;
2523
2524 if ( WSARecvFrom(s, &wsabuf, 1, &n, &dwFlags, from, fromlen, NULL, NULL) == SOCKET_ERROR )
2525 return SOCKET_ERROR;
2526 else
2527 return n;
2528 }
2529
2530 /* allocate a poll array for the corresponding fd sets */
2531 static struct pollfd *fd_sets_to_poll( const WS_fd_set *readfds, const WS_fd_set *writefds,
2532 const WS_fd_set *exceptfds, int *count_ptr )
2533 {
2534 unsigned int i, j = 0, count = 0;
2535 struct pollfd *fds;
2536
2537 if (readfds) count += readfds->fd_count;
2538 if (writefds) count += writefds->fd_count;
2539 if (exceptfds) count += exceptfds->fd_count;
2540 *count_ptr = count;
2541 if (!count) return NULL;
2542 if (!(fds = HeapAlloc( GetProcessHeap(), 0, count * sizeof(fds[0])))) return NULL;
2543 if (readfds)
2544 for (i = 0; i < readfds->fd_count; i++, j++)
2545 {
2546 fds[j].fd = get_sock_fd( readfds->fd_array[i], FILE_READ_DATA, NULL );
2547 fds[j].events = POLLIN;
2548 fds[j].revents = 0;
2549 }
2550 if (writefds)
2551 for (i = 0; i < writefds->fd_count; i++, j++)
2552 {
2553 fds[j].fd = get_sock_fd( writefds->fd_array[i], FILE_WRITE_DATA, NULL );
2554 fds[j].events = POLLOUT;
2555 fds[j].revents = 0;
2556 }
2557 if (exceptfds)
2558 for (i = 0; i < exceptfds->fd_count; i++, j++)
2559 {
2560 fds[j].fd = get_sock_fd( exceptfds->fd_array[i], 0, NULL );
2561 fds[j].events = POLLHUP;
2562 fds[j].revents = 0;
2563 }
2564 return fds;
2565 }
2566
2567 /* release the file descriptor obtained in fd_sets_to_poll */
2568 /* must be called with the original fd_set arrays, before calling get_poll_results */
2569 static void release_poll_fds( const WS_fd_set *readfds, const WS_fd_set *writefds,
2570 const WS_fd_set *exceptfds, struct pollfd *fds )
2571 {
2572 unsigned int i, j = 0;
2573
2574 if (readfds)
2575 {
2576 for (i = 0; i < readfds->fd_count; i++, j++)
2577 if (fds[j].fd != -1) release_sock_fd( readfds->fd_array[i], fds[j].fd );
2578 }
2579 if (writefds)
2580 {
2581 for (i = 0; i < writefds->fd_count; i++, j++)
2582 if (fds[j].fd != -1) release_sock_fd( writefds->fd_array[i], fds[j].fd );
2583 }
2584 if (exceptfds)
2585 {
2586 for (i = 0; i < exceptfds->fd_count; i++, j++)
2587 if (fds[j].fd != -1)
2588 {
2589 /* make sure we have a real error before releasing the fd */
2590 if (!sock_error_p( fds[j].fd )) fds[j].revents = 0;
2591 release_sock_fd( exceptfds->fd_array[i], fds[j].fd );
2592 }
2593 }
2594 }
2595
2596 /* map the poll results back into the Windows fd sets */
2597 static int get_poll_results( WS_fd_set *readfds, WS_fd_set *writefds, WS_fd_set *exceptfds,
2598 const struct pollfd *fds )
2599 {
2600 unsigned int i, j = 0, k, total = 0;
2601
2602 if (readfds)
2603 {
2604 for (i = k = 0; i < readfds->fd_count; i++, j++)
2605 if (fds[j].revents) readfds->fd_array[k++] = readfds->fd_array[i];
2606 readfds->fd_count = k;
2607 total += k;
2608 }
2609 if (writefds)
2610 {
2611 for (i = k = 0; i < writefds->fd_count; i++, j++)
2612 if (fds[j].revents) writefds->fd_array[k++] = writefds->fd_array[i];
2613 writefds->fd_count = k;
2614 total += k;
2615 }
2616 if (exceptfds)
2617 {
2618 for (i = k = 0; i < exceptfds->fd_count; i++, j++)
2619 if (fds[j].revents) exceptfds->fd_array[k++] = exceptfds->fd_array[i];
2620 exceptfds->fd_count = k;
2621 total += k;
2622 }
2623 return total;
2624 }
2625
2626
2627 /***********************************************************************
2628 * select (WS2_32.18)
2629 */
2630 int WINAPI WS_select(int nfds, WS_fd_set *ws_readfds,
2631 WS_fd_set *ws_writefds, WS_fd_set *ws_exceptfds,
2632 const struct WS_timeval* ws_timeout)
2633 {
2634 struct pollfd *pollfds;
2635 int count, ret, timeout = -1;
2636
2637 TRACE("read %p, write %p, excp %p timeout %p\n",
2638 ws_readfds, ws_writefds, ws_exceptfds, ws_timeout);
2639
2640 if (!(pollfds = fd_sets_to_poll( ws_readfds, ws_writefds, ws_exceptfds, &count )) && count)
2641 {
2642 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
2643 return SOCKET_ERROR;
2644 }
2645
2646 if (ws_timeout) timeout = (ws_timeout->tv_sec * 1000) + (ws_timeout->tv_usec + 999) / 1000;
2647
2648 ret = poll( pollfds, count, timeout );
2649 release_poll_fds( ws_readfds, ws_writefds, ws_exceptfds, pollfds );
2650
2651 if (ret == -1) SetLastError(wsaErrno());
2652 else ret = get_poll_results( ws_readfds, ws_writefds, ws_exceptfds, pollfds );
2653 HeapFree( GetProcessHeap(), 0, pollfds );
2654 return ret;
2655 }
2656
2657 /* helper to send completion messages for client-only i/o operation case */
2658 static void WS_AddCompletion( SOCKET sock, ULONG_PTR CompletionValue, NTSTATUS CompletionStatus,
2659 ULONG Information )
2660 {
2661 NTSTATUS status;
2662
2663 SERVER_START_REQ( add_fd_completion )
2664 {
2665 req->handle = wine_server_obj_handle( SOCKET2HANDLE(sock) );
2666 req->cvalue = CompletionValue;
2667 req->status = CompletionStatus;
2668 req->information = Information;
2669 status = wine_server_call( req );
2670 }
2671 SERVER_END_REQ;
2672 }
2673
2674
2675 /***********************************************************************
2676 * send (WS2_32.19)
2677 */
2678 int WINAPI WS_send(SOCKET s, const char *buf, int len, int flags)
2679 {
2680 DWORD n;
2681 WSABUF wsabuf;
2682
2683 wsabuf.len = len;
2684 wsabuf.buf = (char*) buf;
2685
2686 if ( WSASendTo( s, &wsabuf, 1, &n, flags, NULL, 0, NULL, NULL) == SOCKET_ERROR )
2687 return SOCKET_ERROR;
2688 else
2689 return n;
2690 }
2691
2692 /***********************************************************************
2693 * WSASend (WS2_32.72)
2694 */
2695 INT WINAPI WSASend( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
2696 LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
2697 LPWSAOVERLAPPED lpOverlapped,
2698 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
2699 {
2700 return WSASendTo( s, lpBuffers, dwBufferCount, lpNumberOfBytesSent, dwFlags,
2701 NULL, 0, lpOverlapped, lpCompletionRoutine );
2702 }
2703
2704 /***********************************************************************
2705 * WSASendDisconnect (WS2_32.73)
2706 */
2707 INT WINAPI WSASendDisconnect( SOCKET s, LPWSABUF lpBuffers )
2708 {
2709 return WS_shutdown( s, SD_SEND );
2710 }
2711
2712
2713 /***********************************************************************
2714 * WSASendTo (WS2_32.74)
2715 */
2716 INT WINAPI WSASendTo( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
2717 LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
2718 const struct WS_sockaddr *to, int tolen,
2719 LPWSAOVERLAPPED lpOverlapped,
2720 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
2721 {
2722 unsigned int i, options;
2723 int n, fd, err;
2724 struct ws2_async *wsa;
2725 int totalLength = 0;
2726 ULONG_PTR cvalue = (lpOverlapped && ((ULONG_PTR)lpOverlapped->hEvent & 1) == 0) ? (ULONG_PTR)lpOverlapped : 0;
2727
2728 TRACE("socket %04lx, wsabuf %p, nbufs %d, flags %d, to %p, tolen %d, ovl %p, func %p\n",
2729 s, lpBuffers, dwBufferCount, dwFlags,
2730 to, tolen, lpOverlapped, lpCompletionRoutine);
2731
2732 fd = get_sock_fd( s, FILE_WRITE_DATA, &options );
2733 TRACE( "fd=%d, options=%x\n", fd, options );
2734
2735 if ( fd == -1 ) return SOCKET_ERROR;
2736
2737 if (!(wsa = HeapAlloc( GetProcessHeap(), 0, FIELD_OFFSET(struct ws2_async, iovec[dwBufferCount]) )))
2738 {
2739 err = WSAEFAULT;
2740 goto error;
2741 }
2742
2743 wsa->hSocket = SOCKET2HANDLE(s);
2744 wsa->addr = (struct WS_sockaddr *)to;
2745 wsa->addrlen.val = tolen;
2746 wsa->flags = dwFlags;
2747 wsa->n_iovecs = dwBufferCount;
2748 wsa->first_iovec = 0;
2749 for ( i = 0; i < dwBufferCount; i++ )
2750 {
2751 wsa->iovec[i].iov_base = lpBuffers[i].buf;
2752 wsa->iovec[i].iov_len = lpBuffers[i].len;
2753 totalLength += lpBuffers[i].len;
2754 }
2755
2756 if (!lpNumberOfBytesSent)
2757 {
2758 err = WSAEFAULT;
2759 goto error;
2760 }
2761
2762 for (;;)
2763 {
2764 n = WS2_send( fd, wsa );
2765 if (n != -1 || errno != EINTR) break;
2766 }
2767 if (n == -1 && errno != EAGAIN)
2768 {
2769 err = wsaErrno();
2770 if (cvalue) WS_AddCompletion( s, cvalue, err, 0 );
2771 goto error;
2772 }
2773
2774 if ((lpOverlapped || lpCompletionRoutine) &&
2775 !(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)))
2776 {
2777 IO_STATUS_BLOCK *iosb = lpOverlapped ? (IO_STATUS_BLOCK *)lpOverlapped : &wsa->local_iosb;
2778
2779 wsa->user_overlapped = lpOverlapped;
2780 wsa->completion_func = lpCompletionRoutine;
2781 release_sock_fd( s, fd );
2782
2783 if (n == -1)
2784 {
2785 iosb->u.Status = STATUS_PENDING;
2786 iosb->Information = 0;
2787
2788 SERVER_START_REQ( register_async )
2789 {
2790 req->type = ASYNC_TYPE_WRITE;
2791 req->async.handle = wine_server_obj_handle( wsa->hSocket );
2792 req->async.callback = wine_server_client_ptr( WS2_async_send );
2793 req->async.iosb = wine_server_client_ptr( iosb );
2794 req->async.arg = wine_server_client_ptr( wsa );
2795 req->async.event = wine_server_obj_handle( lpCompletionRoutine ? 0 : lpOverlapped->hEvent );
2796 req->async.cvalue = cvalue;
2797 err = wine_server_call( req );
2798 }
2799 SERVER_END_REQ;
2800
2801 if (err != STATUS_PENDING) HeapFree( GetProcessHeap(), 0, wsa );
2802 WSASetLastError( NtStatusToWSAError( err ));
2803 return SOCKET_ERROR;
2804 }
2805
2806 iosb->u.Status = STATUS_SUCCESS;
2807 iosb->Information = n;
2808 *lpNumberOfBytesSent = n;
2809 if (!wsa->completion_func)
2810 {
2811 if (cvalue) WS_AddCompletion( s, cvalue, STATUS_SUCCESS, n );
2812 if (lpOverlapped->hEvent) SetEvent( lpOverlapped->hEvent );
2813 HeapFree( GetProcessHeap(), 0, wsa );
2814 }
2815 else NtQueueApcThread( GetCurrentThread(), (PNTAPCFUNC)ws2_async_apc,
2816 (ULONG_PTR)wsa, (ULONG_PTR)iosb, 0 );
2817 WSASetLastError(0);
2818 return 0;
2819 }
2820
2821 if ( _is_blocking(s) )
2822 {
2823 /* On a blocking non-overlapped stream socket,
2824 * sending blocks until the entire buffer is sent. */
2825 DWORD timeout_start = GetTickCount();
2826
2827 *lpNumberOfBytesSent = 0;
2828
2829 while (wsa->first_iovec < dwBufferCount)
2830 {
2831 struct pollfd pfd;
2832 int timeout = GET_SNDTIMEO(fd);
2833
2834 if (n >= 0)
2835 {
2836 *lpNumberOfBytesSent += n;
2837 while (wsa->first_iovec < dwBufferCount && wsa->iovec[wsa->first_iovec].iov_len <= n)
2838 n -= wsa->iovec[wsa->first_iovec++].iov_len;
2839 if (wsa->first_iovec >= dwBufferCount) break;
2840 wsa->iovec[wsa->first_iovec].iov_base = (char*)wsa->iovec[wsa->first_iovec].iov_base + n;
2841 wsa->iovec[wsa->first_iovec].iov_len -= n;
2842 }
2843
2844 if (timeout != -1)
2845 {
2846 timeout -= GetTickCount() - timeout_start;
2847 if (timeout < 0) timeout = 0;
2848 }
2849
2850 pfd.fd = fd;
2851 pfd.events = POLLOUT;
2852
2853 if (!timeout || !poll( &pfd, 1, timeout ))
2854 {
2855 err = WSAETIMEDOUT;
2856 goto error; /* msdn says a timeout in send is fatal */
2857 }
2858
2859 n = WS2_send( fd, wsa );
2860 if (n == -1 && errno != EAGAIN && errno != EINTR)
2861 {
2862 err = wsaErrno();
2863 goto error;
2864 }
2865 }
2866 }
2867 else /* non-blocking */
2868 {
2869 if (n < totalLength)
2870 _enable_event(SOCKET2HANDLE(s), FD_WRITE, 0, 0);
2871 if (n == -1)
2872 {
2873 err = WSAEWOULDBLOCK;
2874 goto error;
2875 }
2876 *lpNumberOfBytesSent = n;
2877 }
2878
2879 TRACE(" -> %i bytes\n", *lpNumberOfBytesSent);
2880
2881 HeapFree( GetProcessHeap(), 0, wsa );
2882 release_sock_fd( s, fd );
2883 WSASetLastError(0);
2884 return 0;
2885
2886 error:
2887 HeapFree( GetProcessHeap(), 0, wsa );
2888 release_sock_fd( s, fd );
2889 WARN(" -> ERROR %d\n", err);
2890 WSASetLastError(err);
2891 return SOCKET_ERROR;
2892 }
2893
2894 /***********************************************************************
2895 * sendto (WS2_32.20)
2896 */
2897 int WINAPI WS_sendto(SOCKET s, const char *buf, int len, int flags,
2898 const struct WS_sockaddr *to, int tolen)
2899 {
2900 DWORD n;
2901 WSABUF wsabuf;
2902
2903 wsabuf.len = len;
2904 wsabuf.buf = (char*) buf;
2905
2906 if ( WSASendTo(s, &wsabuf, 1, &n, flags, to, tolen, NULL, NULL) == SOCKET_ERROR )
2907 return SOCKET_ERROR;
2908 else
2909 return n;
2910 }
2911
2912 /***********************************************************************
2913 * setsockopt (WS2_32.21)
2914 */
2915 int WINAPI WS_setsockopt(SOCKET s, int level, int optname,
2916 const char *optval, int optlen)
2917 {
2918 int fd;
2919 int woptval;
2920 struct linger linger;
2921 struct timeval tval;
2922
2923 TRACE("socket: %04lx, level 0x%x, name 0x%x, ptr %p, len %d\n",
2924 s, level, optname, optval, optlen);
2925
2926 /* some broken apps pass the value directly instead of a pointer to it */
2927 if(IS_INTRESOURCE(optval))
2928 {
2929 SetLastError(WSAEFAULT);
2930 return SOCKET_ERROR;
2931 }
2932
2933 switch(level)
2934 {
2935 case WS_SOL_SOCKET:
2936 switch(optname)
2937 {
2938 /* Some options need some conversion before they can be sent to
2939 * setsockopt. The conversions are done here, then they will fall though
2940 * to the general case. Special options that are not passed to
2941 * setsockopt follow below that.*/
2942
2943 case WS_SO_DONTLINGER:
2944 linger.l_onoff = *((const int*)optval) ? 0: 1;
2945 linger.l_linger = 0;
2946 level = SOL_SOCKET;
2947 optname = SO_LINGER;
2948 optval = (char*)&linger;
2949 optlen = sizeof(struct linger);
2950 break;
2951
2952 case WS_SO_LINGER:
2953 linger.l_onoff = ((LINGER*)optval)->l_onoff;
2954 linger.l_linger = ((LINGER*)optval)->l_linger;
2955 /* FIXME: what is documented behavior if SO_LINGER optval
2956 is null?? */
2957 level = SOL_SOCKET;
2958 optname = SO_LINGER;
2959 optval = (char*)&linger;
2960 optlen = sizeof(struct linger);
2961 break;
2962
2963 case WS_SO_RCVBUF:
2964 if (*(const int*)optval < 2048)
2965 {
2966 WARN("SO_RCVBF for %d bytes is too small: ignored\n", *(const int*)optval );
2967 return 0;
2968 }
2969 /* Fall through */
2970
2971 /* The options listed here don't need any special handling. Thanks to
2972 * the conversion happening above, options from there will fall through
2973 * to this, too.*/
2974 case WS_SO_ACCEPTCONN:
2975 case WS_SO_BROADCAST:
2976 case WS_SO_ERROR:
2977 case WS_SO_KEEPALIVE:
2978 case WS_SO_OOBINLINE:
2979 /* BSD socket SO_REUSEADDR is not 100% compatible to winsock semantics.
2980 * however, using it the BSD way fixes bug 8513 and seems to be what
2981 * most programmers assume, anyway */
2982 case WS_SO_REUSEADDR:
2983 case WS_SO_SNDBUF:
2984 case WS_SO_TYPE:
2985 convert_sockopt(&level, &optname);
2986 break;
2987
2988 /* SO_DEBUG is a privileged operation, ignore it. */
2989 case WS_SO_DEBUG:
2990 TRACE("Ignoring SO_DEBUG\n");
2991 return 0;
2992
2993 /* For some reason the game GrandPrixLegends does set SO_DONTROUTE on its
2994 * socket. According to MSDN, this option is silently ignored.*/
2995 case WS_SO_DONTROUTE:
2996 TRACE("Ignoring SO_DONTROUTE\n");
2997 return 0;
2998
2999 /* Stops two sockets from being bound to the same port. Always happens
3000 * on unix systems, so just drop it. */
3001 case WS_SO_EXCLUSIVEADDRUSE:
3002 TRACE("Ignoring SO_EXCLUSIVEADDRUSE, is always set.\n");
3003 return 0;
3004
3005 /* SO_OPENTYPE does not require a valid socket handle. */
3006 case WS_SO_OPENTYPE:
3007 if (!optlen || optlen < sizeof(int) || !optval)
3008 {
3009 SetLastError(WSAEFAULT);
3010 return SOCKET_ERROR;
3011 }
3012 get_per_thread_data()->opentype = *(const int *)optval;
3013 TRACE("setting global SO_OPENTYPE = 0x%x\n", *((int*)optval) );
3014 return 0;
3015
3016 #ifdef SO_RCVTIMEO
3017 case WS_SO_RCVTIMEO:
3018 #endif
3019 #ifdef SO_SNDTIMEO
3020 case WS_SO_SNDTIMEO:
3021 #endif
3022 #if defined(SO_RCVTIMEO) || defined(SO_SNDTIMEO)
3023 if (optval && optlen == sizeof(UINT32)) {
3024 /* WinSock passes milliseconds instead of struct timeval */
3025 tval.tv_usec = (*(const UINT32*)optval % 1000) * 1000;
3026 tval.tv_sec = *(const UINT32*)optval / 1000;
3027 /* min of 500 milliseconds */
3028 if (tval.tv_sec == 0 && tval.tv_usec < 500000)
3029 tval.tv_usec = 500000;
3030 optlen = sizeof(struct timeval);
3031 optval = (char*)&tval;
3032 } else if (optlen == sizeof(struct timeval)) {
3033 WARN("SO_SND/RCVTIMEO for %d bytes: assuming unixism\n", optlen);
3034 } else {
3035 WARN("SO_SND/RCVTIMEO for %d bytes is weird: ignored\n", optlen);
3036 return 0;
3037 }
3038 convert_sockopt(&level, &optname);
3039 break;
3040 #endif
3041
3042 default:
3043 TRACE("Unknown SOL_SOCKET optname: 0x%08x\n", optname);
3044 SetLastError(WSAENOPROTOOPT);
3045 return SOCKET_ERROR;
3046 }
3047 break; /* case WS_SOL_SOCKET */
3048
3049 #ifdef HAVE_IPX
3050 case NSPROTO_IPX:
3051 switch(optname)
3052 {
3053 case IPX_PTYPE:
3054 fd = get_sock_fd( s, 0, NULL );
3055 TRACE("trying to set IPX_PTYPE: %d (fd: %d)\n", *(const int*)optval, fd);
3056
3057 /* We try to set the ipx type on ipx socket level. */
3058 #ifdef SOL_IPX
3059 if(setsockopt(fd, SOL_IPX, IPX_TYPE, optval, optlen) == -1)
3060 {
3061 ERR("IPX: could not set ipx option type; expect weird behaviour\n");
3062 release_sock_fd( s, fd );
3063 return SOCKET_ERROR;
3064 }
3065 #else
3066 {
3067 struct ipx val;
3068 /* Should we retrieve val using a getsockopt call and then
3069 * set the modified one? */
3070 val.ipx_pt = *optval;
3071 setsockopt(fd, 0, SO_DEFAULT_HEADERS, &val, sizeof(struct ipx));
3072 }
3073 #endif
3074 release_sock_fd( s, fd );
3075 return 0;
3076
3077 case IPX_FILTERPTYPE:
3078 /* Sets the receive filter packet type, at the moment we don't support it */
3079 FIXME("IPX_FILTERPTYPE: %x\n", *optval);
3080 /* Returning 0 is better for now than returning a SOCKET_ERROR */
3081 return 0;
3082
3083 default:
3084 FIXME("opt_name:%x\n", optname);
3085 return SOCKET_ERROR;
3086 }
3087 break; /* case NSPROTO_IPX */
3088 #endif
3089
3090 /* Levels WS_IPPROTO_TCP and WS_IPPROTO_IP convert directly */
3091 case WS_IPPROTO_TCP:
3092 switch(optname)
3093 {
3094 case WS_TCP_NODELAY:
3095 convert_sockopt(&level, &optname);
3096 break;
3097 default:
3098 FIXME("Unknown IPPROTO_TCP optname 0x%08x\n", optname);
3099 return SOCKET_ERROR;
3100 }
3101 break;
3102
3103 case WS_IPPROTO_IP:
3104 switch(optname)
3105 {
3106 case WS_IP_ADD_MEMBERSHIP:
3107 case WS_IP_DROP_MEMBERSHIP:
3108 #ifdef IP_HDRINCL
3109 case WS_IP_HDRINCL:
3110 #endif
3111 case WS_IP_MULTICAST_IF:
3112 case WS_IP_MULTICAST_LOOP:
3113 case WS_IP_MULTICAST_TTL:
3114 case WS_IP_OPTIONS:
3115 case WS_IP_TOS:
3116 case WS_IP_TTL:
3117 convert_sockopt(&level, &optname);
3118 break;
3119 case WS_IP_DONTFRAGMENT:
3120 FIXME("IP_DONTFRAGMENT is silently ignored!\n");
3121 return 0;
3122 default:
3123 FIXME("Unknown IPPROTO_IP optname 0x%08x\n", optname);
3124 return SOCKET_ERROR;
3125 }
3126 break;
3127
3128 default:
3129 FIXME("Unknown level: 0x%08x\n", level);
3130 return SOCKET_ERROR;
3131 } /* end switch(level) */
3132
3133 /* avoid endianness issues if argument is a 16-bit int */
3134 if (optval && optlen < sizeof(int))
3135 {
3136 woptval= *((const INT16 *) optval);
3137 optval= (char*) &woptval;
3138 optlen=sizeof(int);
3139 }
3140 fd = get_sock_fd( s, 0, NULL );
3141 if (fd == -1) return SOCKET_ERROR;
3142
3143 if (setsockopt(fd, level, optname, optval, optlen) == 0)
3144 {
3145 release_sock_fd( s, fd );
3146 return 0;
3147 }
3148 TRACE("Setting socket error, %d\n", wsaErrno());
3149 SetLastError(wsaErrno());
3150 release_sock_fd( s, fd );
3151
3152 return SOCKET_ERROR;
3153 }
3154
3155 /***********************************************************************
3156 * shutdown (WS2_32.22)
3157 */
3158 int WINAPI WS_shutdown(SOCKET s, int how)
3159 {
3160 int fd, err = WSAENOTSOCK;
3161 unsigned int options, clear_flags = 0;
3162
3163 fd = get_sock_fd( s, 0, &options );
3164 TRACE("socket %04lx, how %i %x\n", s, how, options );
3165
3166 if (fd == -1)
3167 return SOCKET_ERROR;
3168
3169 switch( how )
3170 {
3171 case 0: /* drop receives */
3172 clear_flags |= FD_READ;
3173 break;
3174 case 1: /* drop sends */
3175 clear_flags |= FD_WRITE;
3176 break;
3177 case 2: /* drop all */
3178 clear_flags |= FD_READ|FD_WRITE;
3179 default:
3180 clear_flags |= FD_WINE_LISTENING;
3181 }
3182
3183 if (!(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)))
3184 {
3185 switch ( how )
3186 {
3187 case SD_RECEIVE:
3188 err = WS2_register_async_shutdown( s, ASYNC_TYPE_READ );
3189 break;
3190 case SD_SEND:
3191 err = WS2_register_async_shutdown( s, ASYNC_TYPE_WRITE );
3192 break;
3193 case SD_BOTH:
3194 default:
3195 err = WS2_register_async_shutdown( s, ASYNC_TYPE_READ );
3196 if (!err) err = WS2_register_async_shutdown( s, ASYNC_TYPE_WRITE );
3197 break;
3198 }
3199 if (err) goto error;
3200 }
3201 else /* non-overlapped mode */
3202 {
3203 if ( shutdown( fd, how ) )
3204 {
3205 err = wsaErrno();
3206 goto error;
3207 }
3208 }
3209
3210 release_sock_fd( s, fd );
3211 _enable_event( SOCKET2HANDLE(s), 0, 0, clear_flags );
3212 if ( how > 1) WSAAsyncSelect( s, 0, 0, 0 );
3213 return 0;
3214
3215 error:
3216 release_sock_fd( s, fd );
3217 _enable_event( SOCKET2HANDLE(s), 0, 0, clear_flags );
3218 WSASetLastError( err );
3219 return SOCKET_ERROR;
3220 }
3221
3222 /***********************************************************************
3223 * socket (WS2_32.23)
3224 */
3225 SOCKET WINAPI WS_socket(int af, int type, int protocol)
3226 {
3227 TRACE("af=%d type=%d protocol=%d\n", af, type, protocol);
3228
3229 return WSASocketA( af, type, protocol, NULL, 0,
3230 get_per_thread_data()->opentype ? 0 : WSA_FLAG_OVERLAPPED );
3231 }
3232
3233
3234 /***********************************************************************
3235 * gethostbyaddr (WS2_32.51)
3236 */
3237 struct WS_hostent* WINAPI WS_gethostbyaddr(const char *addr, int len, int type)
3238 {
3239 struct WS_hostent *retval = NULL;
3240 struct hostent* host;
3241
3242 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
3243 char *extrabuf;
3244 int ebufsize=1024;
3245 struct hostent hostentry;
3246 int locerr=ENOBUFS;
3247 host = NULL;
3248 extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ;
3249 while(extrabuf) {
3250 int res = gethostbyaddr_r(addr, len, type,
3251 &hostentry, extrabuf, ebufsize, &host, &locerr);
3252 if( res != ERANGE) break;
3253 ebufsize *=2;
3254 extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ;
3255 }
3256 if (!host) SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr));
3257 #else
3258 EnterCriticalSection( &csWSgetXXXbyYYY );
3259 host = gethostbyaddr(addr, len, type);
3260 if (!host) SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno));
3261 #endif
3262 if( host != NULL ) retval = WS_dup_he(host);
3263 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
3264 HeapFree(GetProcessHeap(),0,extrabuf);
3265 #else
3266 LeaveCriticalSection( &csWSgetXXXbyYYY );
3267 #endif
3268 TRACE("ptr %p, len %d, type %d ret %p\n", addr, len, type, retval);
3269 return retval;
3270 }
3271
3272 /***********************************************************************
3273 * gethostbyname (WS2_32.52)
3274 */
3275 struct WS_hostent* WINAPI WS_gethostbyname(const char* name)
3276 {
3277 struct WS_hostent *retval = NULL;
3278 struct hostent* host;
3279 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
3280 char *extrabuf;
3281 int ebufsize=1024;
3282 struct hostent hostentry;
3283 int locerr = ENOBUFS;
3284 #endif
3285 char buf[100];
3286 if( !name || !name[0]) {
3287 name = buf;
3288 if( gethostname( buf, 100) == -1) {
3289 SetLastError( WSAENOBUFS); /* appropriate ? */
3290 return retval;
3291 }
3292 }
3293 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
3294 host = NULL;
3295 extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ;
3296 while(extrabuf) {
3297 int res = gethostbyname_r(name, &hostentry, extrabuf, ebufsize, &host, &locerr);
3298 if( res != ERANGE) break;
3299 ebufsize *=2;
3300 extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ;
3301 }
3302 if (!host) SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr));
3303 #else
3304 EnterCriticalSection( &csWSgetXXXbyYYY );
3305 host = gethostbyname(name);
3306 if (!host) SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno));
3307 #endif
3308 if (host) retval = WS_dup_he(host);
3309 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
3310 HeapFree(GetProcessHeap(),0,extrabuf);
3311 #else
3312 LeaveCriticalSection( &csWSgetXXXbyYYY );
3313 #endif
3314 if (retval && retval->h_addr_list[0][0] == 127 &&
3315 strcmp(name, "localhost") != 0)
3316 {
3317 /* hostname != "localhost" but has loopback address. replace by our
3318 * special address.*/
3319 memcpy(retval->h_addr_list[0], magic_loopback_addr, 4);
3320 }
3321 TRACE( "%s ret %p\n", debugstr_a(name), retval );
3322 return retval;
3323 }
3324
3325
3326 /***********************************************************************
3327 * getprotobyname (WS2_32.53)
3328 */
3329 struct WS_protoent* WINAPI WS_getprotobyname(const char* name)
3330 {
3331 struct WS_protoent* retval = NULL;
3332 #ifdef HAVE_GETPROTOBYNAME
3333 struct protoent* proto;
3334 EnterCriticalSection( &csWSgetXXXbyYYY );
3335 if( (proto = getprotobyname(name)) != NULL )
3336 {
3337 retval = WS_dup_pe(proto);
3338 }
3339 else {
3340 MESSAGE("protocol %s not found; You might want to add "
3341 "this to /etc/protocols\n", debugstr_a(name) );
3342 SetLastError(WSANO_DATA);
3343 }
3344 LeaveCriticalSection( &csWSgetXXXbyYYY );
3345 #endif
3346 TRACE( "%s ret %p\n", debugstr_a(name), retval );
3347 return retval;
3348 }
3349
3350
3351 /***********************************************************************
3352 * getprotobynumber (WS2_32.54)
3353 */
3354 struct WS_protoent* WINAPI WS_getprotobynumber(int number)
3355 {
3356 struct WS_protoent* retval = NULL;
3357 #ifdef HAVE_GETPROTOBYNUMBER
3358 struct protoent* proto;
3359 EnterCriticalSection( &csWSgetXXXbyYYY );
3360 if( (proto = getprotobynumber(number)) != NULL )
3361 {
3362 retval = WS_dup_pe(proto);
3363 }
3364 else {
3365 MESSAGE("protocol number %d not found; You might want to add "
3366 "this to /etc/protocols\n", number );
3367 SetLastError(WSANO_DATA);
3368 }
3369 LeaveCriticalSection( &csWSgetXXXbyYYY );
3370 #endif
3371 TRACE("%i ret %p\n", number, retval);
3372 return retval;
3373 }
3374
3375
3376 /***********************************************************************
3377 * getservbyname (WS2_32.55)
3378 */
3379 struct WS_servent* WINAPI WS_getservbyname(const char *name, const char *proto)
3380 {
3381 struct WS_servent* retval = NULL;
3382 struct servent* serv;
3383 char *name_str;
3384 char *proto_str = NULL;
3385
3386 if (!(name_str = strdup_lower(name))) return NULL;
3387
3388 if (proto && *proto)
3389 {
3390 if (!(proto_str = strdup_lower(proto)))
3391 {
3392 HeapFree( GetProcessHeap(), 0, name_str );
3393 return NULL;
3394 }
3395 }
3396
3397 EnterCriticalSection( &csWSgetXXXbyYYY );
3398 serv = getservbyname(name_str, proto_str);
3399 if( serv != NULL )
3400 {
3401 retval = WS_dup_se(serv);
3402 }
3403 else SetLastError(WSANO_DATA);
3404 LeaveCriticalSection( &csWSgetXXXbyYYY );
3405 HeapFree( GetProcessHeap(), 0, proto_str );
3406 HeapFree( GetProcessHeap(), 0, name_str );
3407 TRACE( "%s, %s ret %p\n", debugstr_a(name), debugstr_a(proto), retval );
3408 return retval;
3409 }
3410
3411 /***********************************************************************
3412 * freeaddrinfo (WS2_32.@)
3413 */
3414 void WINAPI WS_freeaddrinfo(struct WS_addrinfo *res)
3415 {
3416 while (res) {
3417 struct WS_addrinfo *next;
3418
3419 HeapFree(GetProcessHeap(),0,res->ai_canonname);
3420 HeapFree(GetProcessHeap(),0,res->ai_addr);
3421 next = res->ai_next;
3422 HeapFree(GetProcessHeap(),0,res);
3423 res = next;
3424 }
3425 }
3426
3427 /* helper functions for getaddrinfo()/getnameinfo() */
3428 static int convert_aiflag_w2u(int winflags) {
3429 unsigned int i;
3430 int unixflags = 0;
3431
3432 for (i=0;i<sizeof(ws_aiflag_map)/sizeof(ws_aiflag_map[0]);i++)
3433 if (ws_aiflag_map[i][0] & winflags) {
3434 unixflags |= ws_aiflag_map[i][1];
3435 winflags &= ~ws_aiflag_map[i][0];
3436 }
3437 if (winflags)
3438 FIXME("Unhandled windows AI_xxx flags %x\n", winflags);
3439 return unixflags;
3440 }
3441
3442 static int convert_niflag_w2u(int winflags) {
3443 unsigned int i;
3444 int unixflags = 0;
3445
3446 for (i=0;i<sizeof(ws_niflag_map)/sizeof(ws_niflag_map[0]);i++)
3447 if (ws_niflag_map[i][0] & winflags) {
3448 unixflags |= ws_niflag_map[i][1];
3449 winflags &= ~ws_niflag_map[i][0];
3450 }
3451 if (winflags)
3452 FIXME("Unhandled windows NI_xxx flags %x\n", winflags);
3453 return unixflags;
3454 }
3455
3456 static int convert_aiflag_u2w(int unixflags) {
3457 unsigned int i;
3458 int winflags = 0;
3459
3460 for (i=0;i<sizeof(ws_aiflag_map)/sizeof(ws_aiflag_map[0]);i++)
3461 if (ws_aiflag_map[i][1] & unixflags) {
3462 winflags |= ws_aiflag_map[i][0];
3463 unixflags &= ~ws_aiflag_map[i][1];
3464 }
3465 if (unixflags) /* will warn usually */
3466 WARN("Unhandled UNIX AI_xxx flags %x\n", unixflags);
3467 return winflags;
3468 }
3469
3470 static int convert_eai_u2w(int unixret) {
3471 int i;
3472
3473 for (i=0;ws_eai_map[i][0];i++)
3474 if (ws_eai_map[i][1] == unixret)
3475 return ws_eai_map[i][0];
3476 return unixret;
3477 }
3478
3479 /***********************************************************************
3480 * getaddrinfo (WS2_32.@)
3481 */
3482 int WINAPI WS_getaddrinfo(LPCSTR nodename, LPCSTR servname, const struct WS_addrinfo *hints, struct WS_addrinfo **res)
3483 {
3484 #ifdef HAVE_GETADDRINFO
3485 struct addrinfo *unixaires = NULL;
3486 int result;
3487 struct addrinfo unixhints, *punixhints = NULL;
3488 CHAR *node = NULL, *serv = NULL;
3489
3490 if (nodename)
3491 if (!(node = strdup_lower(nodename))) return WSA_NOT_ENOUGH_MEMORY;
3492
3493 if (servname) {
3494 if (!(serv = strdup_lower(servname))) {
3495 HeapFree(GetProcessHeap(), 0, node);
3496 return WSA_NOT_ENOUGH_MEMORY;
3497 }
3498 }
3499
3500 if (hints) {
3501 punixhints = &unixhints;
3502
3503 memset(&unixhints, 0, sizeof(unixhints));
3504 punixhints->ai_flags = convert_aiflag_w2u(hints->ai_flags);
3505 if (hints->ai_family == 0) /* wildcard, specific to getaddrinfo() */
3506 punixhints->ai_family = 0;
3507 else
3508 punixhints->ai_family = convert_af_w2u(hints->ai_family);
3509 if (hints->ai_socktype == 0) /* wildcard, specific to getaddrinfo() */
3510 punixhints->ai_socktype = 0;
3511 else
3512 punixhints->ai_socktype = convert_socktype_w2u(hints->ai_socktype);
3513 if (hints->ai_protocol == 0) /* wildcard, specific to getaddrinfo() */
3514 punixhints->ai_protocol = 0;
3515 else
3516 punixhints->ai_protocol = convert_proto_w2u(hints->ai_protocol);
3517 }
3518
3519 /* getaddrinfo(3) is thread safe, no need to wrap in CS */
3520 result = getaddrinfo(nodename, servname, punixhints, &unixaires);
3521
3522 TRACE("%s, %s %p -> %p %d\n", debugstr_a(nodename), debugstr_a(servname), hints, res, result);
3523
3524 HeapFree(GetProcessHeap(), 0, node);
3525 HeapFree(GetProcessHeap(), 0, serv);
3526
3527 if (!result) {
3528 struct addrinfo *xuai = unixaires;
3529 struct WS_addrinfo **xai = res;
3530
3531 *xai = NULL;
3532 while (xuai) {
3533 struct WS_addrinfo *ai = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY, sizeof(struct WS_addrinfo));
3534 int len;
3535
3536 if (!ai)
3537 goto outofmem;
3538
3539 *xai = ai;xai = &ai->ai_next;
3540 ai->ai_flags = convert_aiflag_u2w(xuai->ai_flags);
3541 ai->ai_family = convert_af_u2w(xuai->ai_family);
3542 ai->ai_socktype = convert_socktype_u2w(xuai->ai_socktype);
3543 ai->ai_protocol = convert_proto_u2w(xuai->ai_protocol);
3544 if (xuai->ai_canonname) {
3545 TRACE("canon name - %s\n",debugstr_a(xuai->ai_canonname));
3546 ai->ai_canonname = HeapAlloc(GetProcessHeap(),0,strlen(xuai->ai_canonname)+1);
3547 if (!ai->ai_canonname)
3548 goto outofmem;
3549 strcpy(ai->ai_canonname,xuai->ai_canonname);
3550 }
3551 len = xuai->ai_addrlen;
3552 ai->ai_addr = HeapAlloc(GetProcessHeap(),0,len);
3553 if (!ai->ai_addr)
3554 goto outofmem;
3555 ai->ai_addrlen = len;
3556 do {
3557 int winlen = ai->ai_addrlen;
3558
3559 if (!ws_sockaddr_u2ws(xuai->ai_addr, ai->ai_addr, &winlen)) {
3560 ai->ai_addrlen = winlen;
3561 break;
3562 }
3563 len = 2*len;
3564 ai->ai_addr = HeapReAlloc(GetProcessHeap(),0,ai->ai_addr,len);
3565 if (!ai->ai_addr)
3566 goto outofmem;
3567 ai->ai_addrlen = len;
3568 } while (1);
3569 xuai = xuai->ai_next;
3570 }
3571 freeaddrinfo(unixaires);
3572 } else {
3573 result = convert_eai_u2w(result);
3574 *res = NULL;
3575 }
3576 return result;
3577
3578 outofmem:
3579 if (*res) WS_freeaddrinfo(*res);
3580 if (unixaires) freeaddrinfo(unixaires);
3581 *res = NULL;
3582 return WSA_NOT_ENOUGH_MEMORY;
3583 #else
3584 FIXME("getaddrinfo() failed, not found during buildtime.\n");
3585 return EAI_FAIL;
3586 #endif
3587 }
3588
3589 static struct WS_addrinfoW *addrinfo_AtoW(const struct WS_addrinfo *ai)
3590 {
3591 struct WS_addrinfoW *ret;
3592
3593 if (!(ret = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_addrinfoW)))) return NULL;
3594 ret->ai_flags = ai->ai_flags;
3595 ret->ai_family = ai->ai_family;
3596 ret->ai_socktype = ai->ai_socktype;
3597 ret->ai_protocol = ai->ai_protocol;
3598 ret->ai_addrlen = ai->ai_addrlen;
3599 ret->ai_canonname = NULL;
3600 ret->ai_addr = NULL;
3601 ret->ai_next = NULL;
3602 if (ai->ai_canonname)
3603 {
3604 int len = MultiByteToWideChar(CP_ACP, 0, ai->ai_canonname, -1, NULL, 0);
3605 if (!(ret->ai_canonname = HeapAlloc(GetProcessHeap(), 0, len)))
3606 {
3607 HeapFree(GetProcessHeap(), 0, ret);
3608 return NULL;
3609 }
3610 MultiByteToWideChar(CP_ACP, 0, ai->ai_canonname, -1, ret->ai_canonname, len);
3611 }
3612 if (ai->ai_addr)
3613 {
3614 if (!(ret->ai_addr = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_sockaddr))))
3615 {
3616 HeapFree(GetProcessHeap(), 0, ret->ai_canonname);
3617 HeapFree(GetProcessHeap(), 0, ret);
3618 return NULL;
3619 }
3620 memcpy(ret->ai_addr, ai->ai_addr, sizeof(struct WS_sockaddr));
3621 }
3622 return ret;
3623 }
3624
3625 static struct WS_addrinfoW *addrinfo_list_AtoW(const struct WS_addrinfo *info)
3626 {
3627 struct WS_addrinfoW *ret, *infoW;
3628
3629 if (!(ret = infoW = addrinfo_AtoW(info))) return NULL;
3630 while (info->ai_next)
3631 {
3632 if (!(infoW->ai_next = addrinfo_AtoW(info->ai_next)))
3633 {
3634 FreeAddrInfoW(ret);
3635 return NULL;