?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
dmusic: Assign to structs instead of using memcpy.
[wine.git] / dlls / ws2_32 / socket.c
blobfb6a20d1aaa8ec8639be6e9b6f054c2649f1f38e
1 /*
2 * based on Windows Sockets 1.1 specs
3 * (ftp.microsoft.com:/Advsys/winsock/spec11/WINSOCK.TXT)
4 *
5 * Copyright (C) 1993,1994,1996,1997 John Brezak, Erik Bos, Alex Korobka.
6 * Copyright (C) 2005 Marcus Meissner
7 * Copyright (C) 2006 Kai Blin
8 *
9 * This library is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU Lesser General Public
11 * License as published by the Free Software Foundation; either
12 * version 2.1 of the License, or (at your option) any later version.
13 *
14 * This library is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 * Lesser General Public License for more details.
18 *
19 * You should have received a copy of the GNU Lesser General Public
20 * License along with this library; if not, write to the Free Software
21 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
22 *
23 * NOTE: If you make any changes to fix a particular app, make sure
24 * they don't break something else like Netscape or telnet and ftp
25 * clients and servers (www.winsite.com got a lot of those).
26 */
27
28 #include "config.h"
29 #include "wine/port.h"
30
31 #include <stdarg.h>
32 #include <stdio.h>
33 #include <string.h>
34 #include <sys/types.h>
35 #ifdef HAVE_SYS_IPC_H
36 # include <sys/ipc.h>
37 #endif
38 #ifdef HAVE_SYS_IOCTL_H
39 # include <sys/ioctl.h>
40 #endif
41 #ifdef HAVE_SYS_FILIO_H
42 # include <sys/filio.h>
43 #endif
44 #ifdef HAVE_SYS_SOCKIO_H
45 # include <sys/sockio.h>
46 #endif
47
48 #if defined(__EMX__)
49 # include <sys/so_ioctl.h>
50 #endif
51
52 #ifdef HAVE_SYS_PARAM_H
53 # include <sys/param.h>
54 #endif
55
56 #ifdef HAVE_SYS_MSG_H
57 # include <sys/msg.h>
58 #endif
59 #ifdef HAVE_SYS_WAIT_H
60 # include <sys/wait.h>
61 #endif
62 #ifdef HAVE_SYS_UIO_H
63 # include <sys/uio.h>
64 #endif
65 #ifdef HAVE_SYS_SOCKET_H
66 #include <sys/socket.h>
67 #endif
68 #ifdef HAVE_NETINET_IN_H
69 # include <netinet/in.h>
70 #endif
71 #ifdef HAVE_NETINET_TCP_H
72 # include <netinet/tcp.h>
73 #endif
74 #ifdef HAVE_ARPA_INET_H
75 # include <arpa/inet.h>
76 #endif
77 #include <ctype.h>
78 #include <fcntl.h>
79 #include <errno.h>
80 #ifdef HAVE_SYS_ERRNO_H
81 #include <sys/errno.h>
82 #endif
83 #ifdef HAVE_NETDB_H
84 #include <netdb.h>
85 #endif
86 #ifdef HAVE_UNISTD_H
87 # include <unistd.h>
88 #endif
89 #include <stdlib.h>
90 #ifdef HAVE_ARPA_NAMESER_H
91 # include <arpa/nameser.h>
92 #endif
93 #ifdef HAVE_RESOLV_H
94 # include <resolv.h>
95 #endif
96 #ifdef HAVE_NET_IF_H
97 # include <net/if.h>
98 #endif
99
100 #ifdef HAVE_NETIPX_IPX_H
101 # include <netipx/ipx.h>
102 # define HAVE_IPX
103 #elif defined(HAVE_LINUX_IPX_H)
104 # ifdef HAVE_ASM_TYPES_H
105 # include <asm/types.h>
106 # endif
107 # include <linux/ipx.h>
108 # define HAVE_IPX
109 #endif
110
111 #ifdef HAVE_POLL_H
112 #include <poll.h>
113 #endif
114 #ifdef HAVE_SYS_POLL_H
115 # include <sys/poll.h>
116 #endif
117 #ifdef HAVE_SYS_TIME_H
118 # include <sys/time.h>
119 #endif
120
121 #define NONAMELESSUNION
122 #define NONAMELESSSTRUCT
123 #include "ntstatus.h"
124 #define WIN32_NO_STATUS
125 #include "windef.h"
126 #include "winbase.h"
127 #include "wingdi.h"
128 #include "winuser.h"
129 #include "winerror.h"
130 #include "winnls.h"
131 #include "winsock2.h"
132 #include "mswsock.h"
133 #include "ws2tcpip.h"
134 #include "ws2spi.h"
135 #include "wsipx.h"
136 #include "winnt.h"
137 #include "iphlpapi.h"
138 #include "wine/server.h"
139 #include "wine/debug.h"
140 #include "wine/unicode.h"
141
142 #ifdef HAVE_IPX
143 # include "wsnwlink.h"
144 #endif
145
146
147 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
148 # define sipx_network sipx_addr.x_net
149 # define sipx_node sipx_addr.x_host.c_host
150 #endif /* __FreeBSD__ */
151
152 #ifndef INADDR_NONE
153 #define INADDR_NONE ~0UL
154 #endif
155
156 WINE_DEFAULT_DEBUG_CHANNEL(winsock);
157
158 /* critical section to protect some non-reentrant net function */
159 extern CRITICAL_SECTION csWSgetXXXbyYYY;
160
161 union generic_unix_sockaddr
162 {
163 struct sockaddr addr;
164 char data[128]; /* should be big enough for all families */
165 };
166
167 static inline const char *debugstr_sockaddr( const struct WS_sockaddr *a )
168 {
169 if (!a) return "(nil)";
170 return wine_dbg_sprintf("{ family %d, address %s, port %d }",
171 ((const struct sockaddr_in *)a)->sin_family,
172 inet_ntoa(((const struct sockaddr_in *)a)->sin_addr),
173 ntohs(((const struct sockaddr_in *)a)->sin_port));
174 }
175
176 /* HANDLE<->SOCKET conversion (SOCKET is UINT_PTR). */
177 #define SOCKET2HANDLE(s) ((HANDLE)(s))
178 #define HANDLE2SOCKET(h) ((SOCKET)(h))
179
180 /****************************************************************
181 * Async IO declarations
182 ****************************************************************/
183
184 typedef struct ws2_async
185 {
186 HANDLE hSocket;
187 int type;
188 LPWSAOVERLAPPED user_overlapped;
189 LPWSAOVERLAPPED_COMPLETION_ROUTINE completion_func;
190 IO_STATUS_BLOCK local_iosb;
191 struct iovec iovec[WS_MSG_MAXIOVLEN];
192 int n_iovecs;
193 struct WS_sockaddr *addr;
194 union
195 {
196 int val; /* for send operations */
197 int *ptr; /* for recv operations */
198 } addrlen;
199 DWORD flags;
200 } ws2_async;
201
202 /****************************************************************/
203
204 /* ----------------------------------- internal data */
205
206 /* ws_... struct conversion flags */
207
208 typedef struct /* WSAAsyncSelect() control struct */
209 {
210 HANDLE service, event, sock;
211 HWND hWnd;
212 UINT uMsg;
213 LONG lEvent;
214 } ws_select_info;
215
216 #define WS_MAX_SOCKETS_PER_PROCESS 128 /* reasonable guess */
217 #define WS_MAX_UDP_DATAGRAM 1024
218 static INT WINAPI WSA_DefaultBlockingHook( FARPROC x );
219
220 /* hostent's, servent's and protent's are stored in one buffer per thread,
221 * as documented on MSDN for the functions that return any of the buffers */
222 struct per_thread_data
223 {
224 int opentype;
225 struct WS_hostent *he_buffer;
226 struct WS_servent *se_buffer;
227 struct WS_protoent *pe_buffer;
228 int he_len;
229 int se_len;
230 int pe_len;
231 };
232
233 static INT num_startup; /* reference counter */
234 static FARPROC blocking_hook = (FARPROC)WSA_DefaultBlockingHook;
235
236 /* function prototypes */
237 static struct WS_hostent *WS_dup_he(const struct hostent* p_he);
238 static struct WS_protoent *WS_dup_pe(const struct protoent* p_pe);
239 static struct WS_servent *WS_dup_se(const struct servent* p_se);
240
241 int WSAIOCTL_GetInterfaceCount(void);
242 int WSAIOCTL_GetInterfaceName(int intNumber, char *intName);
243
244 UINT wsaErrno(void);
245 UINT wsaHerrno(int errnr);
246
247 #define MAP_OPTION(opt) { WS_##opt, opt }
248
249 static const int ws_sock_map[][2] =
250 {
251 MAP_OPTION( SO_DEBUG ),
252 MAP_OPTION( SO_ACCEPTCONN ),
253 MAP_OPTION( SO_REUSEADDR ),
254 MAP_OPTION( SO_KEEPALIVE ),
255 MAP_OPTION( SO_DONTROUTE ),
256 MAP_OPTION( SO_BROADCAST ),
257 MAP_OPTION( SO_LINGER ),
258 MAP_OPTION( SO_OOBINLINE ),
259 MAP_OPTION( SO_SNDBUF ),
260 MAP_OPTION( SO_RCVBUF ),
261 MAP_OPTION( SO_ERROR ),
262 MAP_OPTION( SO_TYPE ),
263 #ifdef SO_RCVTIMEO
264 MAP_OPTION( SO_RCVTIMEO ),
265 #endif
266 #ifdef SO_SNDTIMEO
267 MAP_OPTION( SO_SNDTIMEO ),
268 #endif
269 };
270
271 static const int ws_tcp_map[][2] =
272 {
273 #ifdef TCP_NODELAY
274 MAP_OPTION( TCP_NODELAY ),
275 #endif
276 };
277
278 static const int ws_ip_map[][2] =
279 {
280 MAP_OPTION( IP_MULTICAST_IF ),
281 MAP_OPTION( IP_MULTICAST_TTL ),
282 MAP_OPTION( IP_MULTICAST_LOOP ),
283 MAP_OPTION( IP_ADD_MEMBERSHIP ),
284 MAP_OPTION( IP_DROP_MEMBERSHIP ),
285 MAP_OPTION( IP_OPTIONS ),
286 #ifdef IP_HDRINCL
287 MAP_OPTION( IP_HDRINCL ),
288 #endif
289 MAP_OPTION( IP_TOS ),
290 MAP_OPTION( IP_TTL ),
291 };
292
293 static const int ws_af_map[][2] =
294 {
295 MAP_OPTION( AF_UNSPEC ),
296 MAP_OPTION( AF_INET ),
297 MAP_OPTION( AF_INET6 ),
298 #ifdef HAVE_IPX
299 MAP_OPTION( AF_IPX ),
300 #endif
301 };
302
303 static const int ws_socktype_map[][2] =
304 {
305 MAP_OPTION( SOCK_DGRAM ),
306 MAP_OPTION( SOCK_STREAM ),
307 MAP_OPTION( SOCK_RAW ),
308 };
309
310 static const int ws_proto_map[][2] =
311 {
312 MAP_OPTION( IPPROTO_IP ),
313 MAP_OPTION( IPPROTO_TCP ),
314 MAP_OPTION( IPPROTO_UDP ),
315 MAP_OPTION( IPPROTO_ICMP ),
316 MAP_OPTION( IPPROTO_IGMP ),
317 MAP_OPTION( IPPROTO_RAW ),
318 };
319
320 static const int ws_aiflag_map[][2] =
321 {
322 MAP_OPTION( AI_PASSIVE ),
323 MAP_OPTION( AI_CANONNAME ),
324 MAP_OPTION( AI_NUMERICHOST ),
325 /* Linux/UNIX knows a lot more. But Windows only
326 * has 3 as far as I could see. -Marcus
327 */
328 };
329
330 static const int ws_niflag_map[][2] =
331 {
332 MAP_OPTION( NI_NOFQDN ),
333 MAP_OPTION( NI_NUMERICHOST ),
334 MAP_OPTION( NI_NAMEREQD ),
335 MAP_OPTION( NI_NUMERICSERV ),
336 MAP_OPTION( NI_DGRAM ),
337 };
338
339 static const int ws_eai_map[][2] =
340 {
341 MAP_OPTION( EAI_AGAIN ),
342 MAP_OPTION( EAI_BADFLAGS ),
343 MAP_OPTION( EAI_FAIL ),
344 MAP_OPTION( EAI_FAMILY ),
345 MAP_OPTION( EAI_MEMORY ),
346 /* Note: EAI_NODATA is deprecated, but still
347 * used by Windows and Linux... We map the newer
348 * EAI_NONAME to EAI_NODATA for now until Windows
349 * changes too.
350 */
351 #ifdef EAI_NODATA
352 MAP_OPTION( EAI_NODATA ),
353 #endif
354 #ifdef EAI_NONAME
355 { WS_EAI_NODATA, EAI_NONAME },
356 #endif
357
358 MAP_OPTION( EAI_SERVICE ),
359 MAP_OPTION( EAI_SOCKTYPE ),
360 { 0, 0 }
361 };
362
363 static inline DWORD NtStatusToWSAError( const DWORD status )
364 {
365 /* We only need to cover the status codes set by server async request handling */
366 DWORD wserr;
367 switch ( status )
368 {
369 case STATUS_SUCCESS: wserr = 0; break;
370 case STATUS_PENDING: wserr = WSA_IO_PENDING; break;
371 case STATUS_OBJECT_TYPE_MISMATCH: wserr = WSAENOTSOCK; break;
372 case STATUS_INVALID_HANDLE: wserr = WSAEBADF; break;
373 case STATUS_INVALID_PARAMETER: wserr = WSAEINVAL; break;
374 case STATUS_PIPE_DISCONNECTED: wserr = WSAESHUTDOWN; break;
375 case STATUS_CANCELLED: wserr = WSA_OPERATION_ABORTED; break;
376 case STATUS_TIMEOUT: wserr = WSAETIMEDOUT; break;
377 case STATUS_NO_MEMORY: wserr = WSAEFAULT; break;
378 default:
379 if ( status >= WSABASEERR && status <= WSABASEERR+1004 )
380 /* It is not an NT status code but a winsock error */
381 wserr = status;
382 else
383 {
384 wserr = RtlNtStatusToDosError( status );
385 FIXME( "Status code %08x converted to DOS error code %x\n", status, wserr );
386 }
387 }
388 return wserr;
389 }
390
391 /* set last error code from NT status without mapping WSA errors */
392 static inline unsigned int set_error( unsigned int err )
393 {
394 if (err)
395 {
396 err = NtStatusToWSAError( err );
397 SetLastError( err );
398 }
399 return err;
400 }
401
402 static inline int get_sock_fd( SOCKET s, DWORD access, unsigned int *options )
403 {
404 int fd;
405 if (set_error( wine_server_handle_to_fd( SOCKET2HANDLE(s), access, &fd, options ) ))
406 return -1;
407 return fd;
408 }
409
410 static inline void release_sock_fd( SOCKET s, int fd )
411 {
412 wine_server_release_fd( SOCKET2HANDLE(s), fd );
413 }
414
415 static void _enable_event( HANDLE s, unsigned int event,
416 unsigned int sstate, unsigned int cstate )
417 {
418 SERVER_START_REQ( enable_socket_event )
419 {
420 req->handle = s;
421 req->mask = event;
422 req->sstate = sstate;
423 req->cstate = cstate;
424 wine_server_call( req );
425 }
426 SERVER_END_REQ;
427 }
428
429 static int _is_blocking(SOCKET s)
430 {
431 int ret;
432 SERVER_START_REQ( get_socket_event )
433 {
434 req->handle = SOCKET2HANDLE(s);
435 req->service = FALSE;
436 req->c_event = 0;
437 wine_server_call( req );
438 ret = (reply->state & FD_WINE_NONBLOCKING) == 0;
439 }
440 SERVER_END_REQ;
441 return ret;
442 }
443
444 static unsigned int _get_sock_mask(SOCKET s)
445 {
446 unsigned int ret;
447 SERVER_START_REQ( get_socket_event )
448 {
449 req->handle = SOCKET2HANDLE(s);
450 req->service = FALSE;
451 req->c_event = 0;
452 wine_server_call( req );
453 ret = reply->mask;
454 }
455 SERVER_END_REQ;
456 return ret;
457 }
458
459 static void _sync_sock_state(SOCKET s)
460 {
461 /* do a dummy wineserver request in order to let
462 the wineserver run through its select loop once */
463 (void)_is_blocking(s);
464 }
465
466 static int _get_sock_error(SOCKET s, unsigned int bit)
467 {
468 int events[FD_MAX_EVENTS];
469
470 SERVER_START_REQ( get_socket_event )
471 {
472 req->handle = SOCKET2HANDLE(s);
473 req->service = FALSE;
474 req->c_event = 0;
475 wine_server_set_reply( req, events, sizeof(events) );
476 wine_server_call( req );
477 }
478 SERVER_END_REQ;
479 return events[bit];
480 }
481
482 static struct per_thread_data *get_per_thread_data(void)
483 {
484 struct per_thread_data * ptb = NtCurrentTeb()->WinSockData;
485 /* lazy initialization */
486 if (!ptb)
487 {
488 ptb = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*ptb) );
489 NtCurrentTeb()->WinSockData = ptb;
490 }
491 return ptb;
492 }
493
494 static void free_per_thread_data(void)
495 {
496 struct per_thread_data * ptb = NtCurrentTeb()->WinSockData;
497
498 if (!ptb) return;
499
500 /* delete scratch buffers */
501 HeapFree( GetProcessHeap(), 0, ptb->he_buffer );
502 HeapFree( GetProcessHeap(), 0, ptb->se_buffer );
503 HeapFree( GetProcessHeap(), 0, ptb->pe_buffer );
504 ptb->he_buffer = NULL;
505 ptb->se_buffer = NULL;
506 ptb->pe_buffer = NULL;
507
508 HeapFree( GetProcessHeap(), 0, ptb );
509 NtCurrentTeb()->WinSockData = NULL;
510 }
511
512 /***********************************************************************
513 * DllMain (WS2_32.init)
514 */
515 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID fImpLoad)
516 {
517 TRACE("%p 0x%x %p\n", hInstDLL, fdwReason, fImpLoad);
518 switch (fdwReason) {
519 case DLL_PROCESS_ATTACH:
520 break;
521 case DLL_PROCESS_DETACH:
522 free_per_thread_data();
523 num_startup = 0;
524 break;
525 case DLL_THREAD_DETACH:
526 free_per_thread_data();
527 break;
528 }
529 return TRUE;
530 }
531
532 /***********************************************************************
533 * convert_sockopt()
534 *
535 * Converts socket flags from Windows format.
536 * Return 1 if converted, 0 if not (error).
537 */
538 static int convert_sockopt(INT *level, INT *optname)
539 {
540 int i;
541 switch (*level)
542 {
543 case WS_SOL_SOCKET:
544 *level = SOL_SOCKET;
545 for(i=0; i<sizeof(ws_sock_map)/sizeof(ws_sock_map[0]); i++) {
546 if( ws_sock_map[i][0] == *optname )
547 {
548 *optname = ws_sock_map[i][1];
549 return 1;
550 }
551 }
552 FIXME("Unknown SOL_SOCKET optname 0x%x\n", *optname);
553 break;
554 case WS_IPPROTO_TCP:
555 *level = IPPROTO_TCP;
556 for(i=0; i<sizeof(ws_tcp_map)/sizeof(ws_tcp_map[0]); i++) {
557 if ( ws_tcp_map[i][0] == *optname )
558 {
559 *optname = ws_tcp_map[i][1];
560 return 1;
561 }
562 }
563 FIXME("Unknown IPPROTO_TCP optname 0x%x\n", *optname);
564 break;
565 case WS_IPPROTO_IP:
566 *level = IPPROTO_IP;
567 for(i=0; i<sizeof(ws_ip_map)/sizeof(ws_ip_map[0]); i++) {
568 if (ws_ip_map[i][0] == *optname )
569 {
570 *optname = ws_ip_map[i][1];
571 return 1;
572 }
573 }
574 FIXME("Unknown IPPROTO_IP optname 0x%x\n", *optname);
575 break;
576 default: FIXME("Unimplemented or unknown socket level\n");
577 }
578 return 0;
579 }
580
581 static inline BOOL is_timeout_option( int optname )
582 {
583 #ifdef SO_RCVTIMEO
584 if (optname == SO_RCVTIMEO) return TRUE;
585 #endif
586 #ifdef SO_SNDTIMEO
587 if (optname == SO_SNDTIMEO) return TRUE;
588 #endif
589 return FALSE;
590 }
591
592 /* ----------------------------------- Per-thread info (or per-process?) */
593
594 static char *strdup_lower(const char *str)
595 {
596 int i;
597 char *ret = HeapAlloc( GetProcessHeap(), 0, strlen(str) + 1 );
598
599 if (ret)
600 {
601 for (i = 0; str[i]; i++) ret[i] = tolower(str[i]);
602 ret[i] = 0;
603 }
604 else SetLastError(WSAENOBUFS);
605 return ret;
606 }
607
608 static inline int sock_error_p(int s)
609 {
610 unsigned int optval, optlen;
611
612 optlen = sizeof(optval);
613 getsockopt(s, SOL_SOCKET, SO_ERROR, (void *) &optval, &optlen);
614 if (optval) WARN("\t[%i] error: %d\n", s, optval);
615 return optval != 0;
616 }
617
618 /* Utility: get the SO_RCVTIMEO or SO_SNDTIMEO socket option
619 * from an fd and return the value converted to milli seconds
620 * or -1 if there is an infinite time out */
621 static inline int get_rcvsnd_timeo( int fd, int optname)
622 {
623 struct timeval tv;
624 unsigned int len = sizeof(tv);
625 int ret = getsockopt(fd, SOL_SOCKET, optname, &tv, &len);
626 if( ret >= 0)
627 ret = tv.tv_sec * 1000 + tv.tv_usec / 1000;
628 if( ret <= 0 ) /* tv == {0,0} means infinite time out */
629 return -1;
630 return ret;
631 }
632
633 /* macro wrappers for portability */
634 #ifdef SO_RCVTIMEO
635 #define GET_RCVTIMEO(fd) get_rcvsnd_timeo( (fd), SO_RCVTIMEO)
636 #else
637 #define GET_RCVTIMEO(fd) (-1)
638 #endif
639
640 #ifdef SO_SNDTIMEO
641 #define GET_SNDTIMEO(fd) get_rcvsnd_timeo( (fd), SO_SNDTIMEO)
642 #else
643 #define GET_SNDTIMEO(fd) (-1)
644 #endif
645
646 /* utility: given an fd, will block until one of the events occurs */
647 static inline int do_block( int fd, int events, int timeout )
648 {
649 struct pollfd pfd;
650 int ret;
651
652 pfd.fd = fd;
653 pfd.events = events;
654
655 while ((ret = poll(&pfd, 1, timeout)) < 0)
656 {
657 if (errno != EINTR)
658 return -1;
659 }
660 if( ret == 0 )
661 return 0;
662 return pfd.revents;
663 }
664
665 static int
666 convert_af_w2u(int windowsaf) {
667 int i;
668
669 for (i=0;i<sizeof(ws_af_map)/sizeof(ws_af_map[0]);i++)
670 if (ws_af_map[i][0] == windowsaf)
671 return ws_af_map[i][1];
672 FIXME("unhandled Windows address family %d\n", windowsaf);
673 return -1;
674 }
675
676 static int
677 convert_af_u2w(int unixaf) {
678 int i;
679
680 for (i=0;i<sizeof(ws_af_map)/sizeof(ws_af_map[0]);i++)
681 if (ws_af_map[i][1] == unixaf)
682 return ws_af_map[i][0];
683 FIXME("unhandled UNIX address family %d\n", unixaf);
684 return -1;
685 }
686
687 static int
688 convert_proto_w2u(int windowsproto) {
689 int i;
690
691 for (i=0;i<sizeof(ws_proto_map)/sizeof(ws_proto_map[0]);i++)
692 if (ws_proto_map[i][0] == windowsproto)
693 return ws_proto_map[i][1];
694 FIXME("unhandled Windows socket protocol %d\n", windowsproto);
695 return -1;
696 }
697
698 static int
699 convert_proto_u2w(int unixproto) {
700 int i;
701
702 for (i=0;i<sizeof(ws_proto_map)/sizeof(ws_proto_map[0]);i++)
703 if (ws_proto_map[i][1] == unixproto)
704 return ws_proto_map[i][0];
705 FIXME("unhandled UNIX socket protocol %d\n", unixproto);
706 return -1;
707 }
708
709 static int
710 convert_socktype_w2u(int windowssocktype) {
711 int i;
712
713 for (i=0;i<sizeof(ws_socktype_map)/sizeof(ws_socktype_map[0]);i++)
714 if (ws_socktype_map[i][0] == windowssocktype)
715 return ws_socktype_map[i][1];
716 FIXME("unhandled Windows socket type %d\n", windowssocktype);
717 return -1;
718 }
719
720 static int
721 convert_socktype_u2w(int unixsocktype) {
722 int i;
723
724 for (i=0;i<sizeof(ws_socktype_map)/sizeof(ws_socktype_map[0]);i++)
725 if (ws_socktype_map[i][1] == unixsocktype)
726 return ws_socktype_map[i][0];
727 FIXME("unhandled UNIX socket type %d\n", unixsocktype);
728 return -1;
729 }
730
731 /* ----------------------------------- API -----
732 *
733 * Init / cleanup / error checking.
734 */
735
736 /***********************************************************************
737 * WSAStartup (WS2_32.115)
738 */
739 int WINAPI WSAStartup(WORD wVersionRequested, LPWSADATA lpWSAData)
740 {
741 TRACE("verReq=%x\n", wVersionRequested);
742
743 if (LOBYTE(wVersionRequested) < 1)
744 return WSAVERNOTSUPPORTED;
745
746 if (!lpWSAData) return WSAEINVAL;
747
748 num_startup++;
749
750 /* that's the whole of the negotiation for now */
751 lpWSAData->wVersion = wVersionRequested;
752 /* return winsock information */
753 lpWSAData->wHighVersion = 0x0202;
754 strcpy(lpWSAData->szDescription, "WinSock 2.0" );
755 strcpy(lpWSAData->szSystemStatus, "Running" );
756 lpWSAData->iMaxSockets = WS_MAX_SOCKETS_PER_PROCESS;
757 lpWSAData->iMaxUdpDg = WS_MAX_UDP_DATAGRAM;
758 /* don't do anything with lpWSAData->lpVendorInfo */
759 /* (some apps don't allocate the space for this field) */
760
761 TRACE("succeeded\n");
762 return 0;
763 }
764
765
766 /***********************************************************************
767 * WSACleanup (WS2_32.116)
768 */
769 INT WINAPI WSACleanup(void)
770 {
771 if (num_startup) {
772 num_startup--;
773 return 0;
774 }
775 SetLastError(WSANOTINITIALISED);
776 return SOCKET_ERROR;
777 }
778
779
780 /***********************************************************************
781 * WSAGetLastError (WINSOCK.111)
782 * WSAGetLastError (WS2_32.111)
783 */
784 INT WINAPI WSAGetLastError(void)
785 {
786 return GetLastError();
787 }
788
789 /***********************************************************************
790 * WSASetLastError (WS2_32.112)
791 */
792 void WINAPI WSASetLastError(INT iError) {
793 SetLastError(iError);
794 }
795
796 static struct WS_hostent *check_buffer_he(int size)
797 {
798 struct per_thread_data * ptb = get_per_thread_data();
799 if (ptb->he_buffer)
800 {
801 if (ptb->he_len >= size ) return ptb->he_buffer;
802 HeapFree( GetProcessHeap(), 0, ptb->he_buffer );
803 }
804 ptb->he_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->he_len = size) );
805 if (!ptb->he_buffer) SetLastError(WSAENOBUFS);
806 return ptb->he_buffer;
807 }
808
809 static struct WS_servent *check_buffer_se(int size)
810 {
811 struct per_thread_data * ptb = get_per_thread_data();
812 if (ptb->se_buffer)
813 {
814 if (ptb->se_len >= size ) return ptb->se_buffer;
815 HeapFree( GetProcessHeap(), 0, ptb->se_buffer );
816 }
817 ptb->se_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->se_len = size) );
818 if (!ptb->se_buffer) SetLastError(WSAENOBUFS);
819 return ptb->se_buffer;
820 }
821
822 static struct WS_protoent *check_buffer_pe(int size)
823 {
824 struct per_thread_data * ptb = get_per_thread_data();
825 if (ptb->pe_buffer)
826 {
827 if (ptb->pe_len >= size ) return ptb->pe_buffer;
828 HeapFree( GetProcessHeap(), 0, ptb->pe_buffer );
829 }
830 ptb->pe_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->pe_len = size) );
831 if (!ptb->pe_buffer) SetLastError(WSAENOBUFS);
832 return ptb->pe_buffer;
833 }
834
835 /* ----------------------------------- i/o APIs */
836
837 #ifdef HAVE_IPX
838 #define SUPPORTED_PF(pf) ((pf)==WS_AF_INET || (pf)== WS_AF_IPX || (pf) == WS_AF_INET6)
839 #else
840 #define SUPPORTED_PF(pf) ((pf)==WS_AF_INET || (pf) == WS_AF_INET6)
841 #endif
842
843
844 /**********************************************************************/
845
846 /* Returns the length of the converted address if successful, 0 if it was too small to
847 * start with.
848 */
849 static unsigned int ws_sockaddr_ws2u(const struct WS_sockaddr* wsaddr, int wsaddrlen,
850 union generic_unix_sockaddr *uaddr)
851 {
852 unsigned int uaddrlen = 0;
853
854 switch (wsaddr->sa_family)
855 {
856 #ifdef HAVE_IPX
857 case WS_AF_IPX:
858 {
859 const struct WS_sockaddr_ipx* wsipx=(const struct WS_sockaddr_ipx*)wsaddr;
860 struct sockaddr_ipx* uipx = (struct sockaddr_ipx *)uaddr;
861
862 if (wsaddrlen<sizeof(struct WS_sockaddr_ipx))
863 return 0;
864
865 uaddrlen = sizeof(struct sockaddr_ipx);
866 memset( uaddr, 0, uaddrlen );
867 uipx->sipx_family=AF_IPX;
868 uipx->sipx_port=wsipx->sa_socket;
869 /* copy sa_netnum and sa_nodenum to sipx_network and sipx_node
870 * in one go
871 */
872 memcpy(&uipx->sipx_network,wsipx->sa_netnum,sizeof(uipx->sipx_network)+sizeof(uipx->sipx_node));
873 #ifdef IPX_FRAME_NONE
874 uipx->sipx_type=IPX_FRAME_NONE;
875 #endif
876 break;
877 }
878 #endif
879 case WS_AF_INET6: {
880 struct sockaddr_in6* uin6 = (struct sockaddr_in6 *)uaddr;
881 const struct WS_sockaddr_in6* win6 = (const struct WS_sockaddr_in6*)wsaddr;
882
883 /* Note: Windows has 2 versions of the sockaddr_in6 struct, one with
884 * scope_id, one without. Check:
885 * http://msdn.microsoft.com/library/en-us/winsock/winsock/sockaddr_2.asp
886 */
887 if (wsaddrlen >= sizeof(struct WS_sockaddr_in6_old)) {
888 uaddrlen = sizeof(struct sockaddr_in6);
889 memset( uaddr, 0, uaddrlen );
890 uin6->sin6_family = AF_INET6;
891 uin6->sin6_port = win6->sin6_port;
892 uin6->sin6_flowinfo = win6->sin6_flowinfo;
893 #ifdef HAVE_STRUCT_SOCKADDR_IN6_SIN6_SCOPE_ID
894 if (wsaddrlen >= sizeof(struct WS_sockaddr_in6)) uin6->sin6_scope_id = win6->sin6_scope_id;
895 #endif
896 memcpy(&uin6->sin6_addr,&win6->sin6_addr,16); /* 16 bytes = 128 address bits */
897 break;
898 }
899 FIXME("bad size %d for WS_sockaddr_in6\n",wsaddrlen);
900 return 0;
901 }
902 case WS_AF_INET: {
903 struct sockaddr_in* uin = (struct sockaddr_in *)uaddr;
904 const struct WS_sockaddr_in* win = (const struct WS_sockaddr_in*)wsaddr;
905
906 if (wsaddrlen<sizeof(struct WS_sockaddr_in))
907 return 0;
908 uaddrlen = sizeof(struct sockaddr_in);
909 memset( uaddr, 0, uaddrlen );
910 uin->sin_family = AF_INET;
911 uin->sin_port = win->sin_port;
912 memcpy(&uin->sin_addr,&win->sin_addr,4); /* 4 bytes = 32 address bits */
913 break;
914 }
915 case WS_AF_UNSPEC: {
916 /* Try to determine the needed space by the passed windows sockaddr space */
917 switch (wsaddrlen) {
918 default: /* likely a ipv4 address */
919 case sizeof(struct WS_sockaddr_in):
920 uaddrlen = sizeof(struct sockaddr_in);
921 break;
922 #ifdef HAVE_IPX
923 case sizeof(struct WS_sockaddr_ipx):
924 uaddrlen = sizeof(struct sockaddr_ipx);
925 break;
926 #endif
927 case sizeof(struct WS_sockaddr_in6):
928 case sizeof(struct WS_sockaddr_in6_old):
929 uaddrlen = sizeof(struct sockaddr_in6);
930 break;
931 }
932 memset( uaddr, 0, uaddrlen );
933 break;
934 }
935 default:
936 FIXME("Unknown address family %d, return NULL.\n", wsaddr->sa_family);
937 return 0;
938 }
939 return uaddrlen;
940 }
941
942 static BOOL is_sockaddr_bound(const struct sockaddr *uaddr, int uaddrlen)
943 {
944 switch (uaddr->sa_family)
945 {
946 #ifdef HAVE_IPX
947 case AF_IPX:
948 FIXME("don't know how to tell if IPX socket is bound, assuming it is!\n");
949 return TRUE;
950 #endif
951 case AF_INET6:
952 {
953 static const struct sockaddr_in6 emptyAddr;
954 const struct sockaddr_in6 *in6 = (const struct sockaddr_in6*) uaddr;
955 return in6->sin6_port || memcmp(&in6->sin6_addr, &emptyAddr.sin6_addr, sizeof(struct in6_addr));
956 }
957 case AF_INET:
958 {
959 static const struct sockaddr_in emptyAddr;
960 const struct sockaddr_in *in = (const struct sockaddr_in*) uaddr;
961 return in->sin_port || memcmp(&in->sin_addr, &emptyAddr.sin_addr, sizeof(struct in_addr));
962 }
963 case AF_UNSPEC:
964 return FALSE;
965 default:
966 FIXME("unknown address family %d\n", uaddr->sa_family);
967 return TRUE;
968 }
969 }
970
971 /* Returns 0 if successful, -1 if the buffer is too small */
972 static int ws_sockaddr_u2ws(const struct sockaddr* uaddr, struct WS_sockaddr* wsaddr, int* wsaddrlen)
973 {
974 int res;
975
976 switch(uaddr->sa_family)
977 {
978 #ifdef HAVE_IPX
979 case AF_IPX:
980 {
981 const struct sockaddr_ipx* uipx=(const struct sockaddr_ipx*)uaddr;
982 struct WS_sockaddr_ipx* wsipx=(struct WS_sockaddr_ipx*)wsaddr;
983
984 res=-1;
985 switch (*wsaddrlen) /* how much can we copy? */
986 {
987 default:
988 res=0; /* enough */
989 *wsaddrlen = sizeof(*wsipx);
990 wsipx->sa_socket=uipx->sipx_port;
991 /* fall through */
992 case 13:
993 case 12:
994 memcpy(wsipx->sa_nodenum,uipx->sipx_node,sizeof(wsipx->sa_nodenum));
995 /* fall through */
996 case 11:
997 case 10:
998 case 9:
999 case 8:
1000 case 7:
1001 case 6:
1002 memcpy(wsipx->sa_netnum,&uipx->sipx_network,sizeof(wsipx->sa_netnum));
1003 /* fall through */
1004 case 5:
1005 case 4:
1006 case 3:
1007 case 2:
1008 wsipx->sa_family=WS_AF_IPX;
1009 /* fall through */
1010 case 1:
1011 case 0:
1012 /* way too small */
1013 break;
1014 }
1015 }
1016 break;
1017 #endif
1018 case AF_INET6: {
1019 const struct sockaddr_in6* uin6 = (const struct sockaddr_in6*)uaddr;
1020 struct WS_sockaddr_in6_old* win6old = (struct WS_sockaddr_in6_old*)wsaddr;
1021
1022 if (*wsaddrlen < sizeof(struct WS_sockaddr_in6_old))
1023 return -1;
1024 win6old->sin6_family = WS_AF_INET6;
1025 win6old->sin6_port = uin6->sin6_port;
1026 win6old->sin6_flowinfo = uin6->sin6_flowinfo;
1027 memcpy(&win6old->sin6_addr,&uin6->sin6_addr,16); /* 16 bytes = 128 address bits */
1028 *wsaddrlen = sizeof(struct WS_sockaddr_in6_old);
1029 #ifdef HAVE_STRUCT_SOCKADDR_IN6_SIN6_SCOPE_ID
1030 if (*wsaddrlen >= sizeof(struct WS_sockaddr_in6)) {
1031 struct WS_sockaddr_in6* win6 = (struct WS_sockaddr_in6*)wsaddr;
1032 win6->sin6_scope_id = uin6->sin6_scope_id;
1033 *wsaddrlen = sizeof(struct WS_sockaddr_in6);
1034 }
1035 #endif
1036 return 0;
1037 }
1038 case AF_INET: {
1039 const struct sockaddr_in* uin = (const struct sockaddr_in*)uaddr;
1040 struct WS_sockaddr_in* win = (struct WS_sockaddr_in*)wsaddr;
1041
1042 if (*wsaddrlen < sizeof(struct WS_sockaddr_in))
1043 return -1;
1044 win->sin_family = WS_AF_INET;
1045 win->sin_port = uin->sin_port;
1046 memcpy(&win->sin_addr,&uin->sin_addr,4); /* 4 bytes = 32 address bits */
1047 memset(&win->sin_zero, 0, 8); /* Make sure the null padding is null */
1048 *wsaddrlen = sizeof(struct WS_sockaddr_in);
1049 return 0;
1050 }
1051 case AF_UNSPEC: {
1052 memset(wsaddr,0,*wsaddrlen);
1053 return 0;
1054 }
1055 default:
1056 FIXME("Unknown address family %d\n", uaddr->sa_family);
1057 return -1;
1058 }
1059 return res;
1060 }
1061
1062 /**************************************************************************
1063 * Functions for handling overlapped I/O
1064 **************************************************************************/
1065
1066 /* user APC called upon async completion */
1067 static void WINAPI ws2_async_apc( void *arg, IO_STATUS_BLOCK *iosb, ULONG reserved )
1068 {
1069 ws2_async *wsa = arg;
1070
1071 if (wsa->completion_func) wsa->completion_func( NtStatusToWSAError(iosb->u.Status),
1072 iosb->Information, wsa->user_overlapped,
1073 wsa->flags );
1074 HeapFree( GetProcessHeap(), 0, wsa );
1075 }
1076
1077 /***********************************************************************
1078 * WS2_recv (INTERNAL)
1079 *
1080 * Workhorse for both synchronous and asynchronous recv() operations.
1081 */
1082 static int WS2_recv( int fd, struct iovec* iov, int count,
1083 struct WS_sockaddr *lpFrom, LPINT lpFromlen,
1084 LPDWORD lpFlags )
1085 {
1086 struct msghdr hdr;
1087 union generic_unix_sockaddr unix_sockaddr;
1088 int n;
1089
1090 hdr.msg_name = NULL;
1091
1092 if ( lpFrom )
1093 {
1094 hdr.msg_namelen = sizeof(unix_sockaddr);
1095 hdr.msg_name = &unix_sockaddr;
1096 }
1097 else
1098 hdr.msg_namelen = 0;
1099
1100 hdr.msg_iov = iov;
1101 hdr.msg_iovlen = count;
1102 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
1103 hdr.msg_accrights = NULL;
1104 hdr.msg_accrightslen = 0;
1105 #else
1106 hdr.msg_control = NULL;
1107 hdr.msg_controllen = 0;
1108 hdr.msg_flags = 0;
1109 #endif
1110
1111 if ( (n = recvmsg(fd, &hdr, *lpFlags)) == -1 )
1112 return -1;
1113
1114 if ( lpFrom &&
1115 ws_sockaddr_u2ws( &unix_sockaddr.addr, lpFrom, lpFromlen ) != 0 )
1116 {
1117 /* The from buffer was too small, but we read the data
1118 * anyway. Is that really bad?
1119 */
1120 }
1121
1122 return n;
1123 }
1124
1125 /***********************************************************************
1126 * WS2_async_recv (INTERNAL)
1127 *
1128 * Handler for overlapped recv() operations.
1129 */
1130 static NTSTATUS WS2_async_recv( void* user, IO_STATUS_BLOCK* iosb, NTSTATUS status, ULONG_PTR *total )
1131 {
1132 ws2_async* wsa = user;
1133 int result = 0, fd;
1134
1135 switch (status)
1136 {
1137 case STATUS_ALERTED:
1138 if ((status = wine_server_handle_to_fd( wsa->hSocket, FILE_READ_DATA, &fd, NULL ) ))
1139 break;
1140
1141 result = WS2_recv( fd, wsa->iovec, wsa->n_iovecs,
1142 wsa->addr, wsa->addrlen.ptr, &wsa->flags );
1143 wine_server_release_fd( wsa->hSocket, fd );
1144 if (result >= 0)
1145 {
1146 status = STATUS_SUCCESS;
1147 _enable_event( wsa->hSocket, FD_READ, 0, 0 );
1148 }
1149 else
1150 {
1151 if (errno == EINTR || errno == EAGAIN)
1152 {
1153 status = STATUS_PENDING;
1154 _enable_event( wsa->hSocket, FD_READ, 0, 0 );
1155 }
1156 else
1157 {
1158 result = 0;
1159 status = wsaErrno(); /* FIXME: is this correct ???? */
1160 }
1161 }
1162 break;
1163 }
1164 if (status != STATUS_PENDING)
1165 {
1166 iosb->u.Status = status;
1167 iosb->Information = *total = result;
1168 }
1169 return status;
1170 }
1171
1172 /***********************************************************************
1173 * WS2_send (INTERNAL)
1174 *
1175 * Workhorse for both synchronous and asynchronous send() operations.
1176 */
1177 static int WS2_send( int fd, struct iovec* iov, int count,
1178 const struct WS_sockaddr *to, INT tolen, DWORD dwFlags )
1179 {
1180 struct msghdr hdr;
1181 union generic_unix_sockaddr unix_addr;
1182
1183
1184 hdr.msg_name = NULL;
1185 hdr.msg_namelen = 0;
1186
1187 if ( to )
1188 {
1189 hdr.msg_name = &unix_addr;
1190 hdr.msg_namelen = ws_sockaddr_ws2u( to, tolen, &unix_addr );
1191 if ( !hdr.msg_namelen )
1192 {
1193 errno = EFAULT;
1194 return -1;
1195 }
1196
1197 #if defined(HAVE_IPX) && defined(SOL_IPX)
1198 if(to->sa_family == WS_AF_IPX)
1199 {
1200 struct sockaddr_ipx* uipx = (struct sockaddr_ipx*)hdr.msg_name;
1201 int val=0;
1202 unsigned int len=sizeof(int);
1203
1204 /* The packet type is stored at the ipx socket level; At least the linux kernel seems
1205 * to do something with it in case hdr.msg_name is NULL. Nonetheless can we use it to store
1206 * the packet type and then we can retrieve it using getsockopt. After that we can set the
1207 * ipx type in the sockaddr_opx structure with the stored value.
1208 */
1209 if(getsockopt(fd, SOL_IPX, IPX_TYPE, &val, &len) != -1)
1210 uipx->sipx_type = val;
1211 }
1212 #endif
1213 }
1214
1215 hdr.msg_iov = iov;
1216 hdr.msg_iovlen = count;
1217 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
1218 hdr.msg_accrights = NULL;
1219 hdr.msg_accrightslen = 0;
1220 #else
1221 hdr.msg_control = NULL;
1222 hdr.msg_controllen = 0;
1223 hdr.msg_flags = 0;
1224 #endif
1225
1226 return sendmsg(fd, &hdr, dwFlags);
1227 }
1228
1229 /***********************************************************************
1230 * WS2_async_send (INTERNAL)
1231 *
1232 * Handler for overlapped send() operations.
1233 */
1234 static NTSTATUS WS2_async_send(void* user, IO_STATUS_BLOCK* iosb, NTSTATUS status, ULONG_PTR *total )
1235 {
1236 ws2_async* wsa = user;
1237 int result = 0, fd;
1238
1239 switch (status)
1240 {
1241 case STATUS_ALERTED:
1242 if ((status = wine_server_handle_to_fd( wsa->hSocket, FILE_WRITE_DATA, &fd, NULL ) ))
1243 break;
1244
1245 /* check to see if the data is ready (non-blocking) */
1246 result = WS2_send( fd, wsa->iovec, wsa->n_iovecs, wsa->addr, wsa->addrlen.val, wsa->flags );
1247 wine_server_release_fd( wsa->hSocket, fd );
1248
1249 if (result >= 0)
1250 {
1251 status = STATUS_SUCCESS;
1252 _enable_event( wsa->hSocket, FD_WRITE, 0, 0 );
1253 }
1254 else
1255 {
1256 if (errno == EINTR || errno == EAGAIN)
1257 {
1258 status = STATUS_PENDING;
1259 _enable_event( wsa->hSocket, FD_WRITE, 0, 0 );
1260 }
1261 else
1262 {
1263 /* We set the status to a winsock error code and check for that
1264 later in NtStatusToWSAError () */
1265 status = wsaErrno();
1266 result = 0;
1267 }
1268 }
1269 break;
1270 }
1271 if (status != STATUS_PENDING)
1272 {
1273 iosb->u.Status = status;
1274 iosb->Information = *total = result;
1275 }
1276 return status;
1277 }
1278
1279 /***********************************************************************
1280 * WS2_async_shutdown (INTERNAL)
1281 *
1282 * Handler for shutdown() operations on overlapped sockets.
1283 */
1284 static NTSTATUS WS2_async_shutdown( void* user, PIO_STATUS_BLOCK iosb, NTSTATUS status )
1285 {
1286 ws2_async* wsa = user;
1287 int fd, err = 1;
1288
1289 switch (status)
1290 {
1291 case STATUS_ALERTED:
1292 if ((status = wine_server_handle_to_fd( wsa->hSocket, 0, &fd, NULL ) ))
1293 break;
1294
1295 switch ( wsa->type )
1296 {
1297 case ASYNC_TYPE_READ: err = shutdown( fd, 0 ); break;
1298 case ASYNC_TYPE_WRITE: err = shutdown( fd, 1 ); break;
1299 }
1300 wine_server_release_fd( wsa->hSocket, fd );
1301 status = err ? wsaErrno() : STATUS_SUCCESS;
1302 break;
1303 }
1304 iosb->u.Status = status;
1305 return status;
1306 }
1307
1308 /***********************************************************************
1309 * WS2_register_async_shutdown (INTERNAL)
1310 *
1311 * Helper function for WS_shutdown() on overlapped sockets.
1312 */
1313 static int WS2_register_async_shutdown( SOCKET s, int type )
1314 {
1315 struct ws2_async *wsa;
1316 NTSTATUS status;
1317
1318 TRACE("s %ld type %d\n", s, type);
1319
1320 wsa = HeapAlloc( GetProcessHeap(), 0, sizeof(*wsa) );
1321 if ( !wsa )
1322 return WSAEFAULT;
1323
1324 wsa->hSocket = SOCKET2HANDLE(s);
1325 wsa->type = type;
1326 wsa->completion_func = NULL;
1327
1328 SERVER_START_REQ( register_async )
1329 {
1330 req->handle = wsa->hSocket;
1331 req->type = type;
1332 req->async.callback = WS2_async_shutdown;
1333 req->async.iosb = &wsa->local_iosb;
1334 req->async.arg = wsa;
1335 req->async.apc = ws2_async_apc;
1336 req->async.cvalue = 0;
1337 status = wine_server_call( req );
1338 }
1339 SERVER_END_REQ;
1340
1341 if (status != STATUS_PENDING)
1342 {
1343 HeapFree( GetProcessHeap(), 0, wsa );
1344 return NtStatusToWSAError( status );
1345 }
1346 return 0;
1347 }
1348
1349 /***********************************************************************
1350 * accept (WS2_32.1)
1351 */
1352 SOCKET WINAPI WS_accept(SOCKET s, struct WS_sockaddr *addr,
1353 int *addrlen32)
1354 {
1355 SOCKET as;
1356 BOOL is_blocking;
1357
1358 TRACE("socket %04lx\n", s );
1359 is_blocking = _is_blocking(s);
1360
1361 do {
1362 if (is_blocking)
1363 {
1364 int fd = get_sock_fd( s, FILE_READ_DATA, NULL );
1365 if (fd == -1) return INVALID_SOCKET;
1366 /* block here */
1367 do_block(fd, POLLIN, -1);
1368 _sync_sock_state(s); /* let wineserver notice connection */
1369 release_sock_fd( s, fd );
1370 /* retrieve any error codes from it */
1371 SetLastError(_get_sock_error(s, FD_ACCEPT_BIT));
1372 /* FIXME: care about the error? */
1373 }
1374 SERVER_START_REQ( accept_socket )
1375 {
1376 req->lhandle = SOCKET2HANDLE(s);
1377 req->access = GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE;
1378 req->attributes = OBJ_INHERIT;
1379 set_error( wine_server_call( req ) );
1380 as = HANDLE2SOCKET( reply->handle );
1381 }
1382 SERVER_END_REQ;
1383 if (as)
1384 {
1385 if (addr) WS_getpeername(as, addr, addrlen32);
1386 return as;
1387 }
1388 } while (is_blocking);
1389 return INVALID_SOCKET;
1390 }
1391
1392 /***********************************************************************
1393 * bind (WS2_32.2)
1394 */
1395 int WINAPI WS_bind(SOCKET s, const struct WS_sockaddr* name, int namelen)
1396 {
1397 int fd = get_sock_fd( s, 0, NULL );
1398 int res = SOCKET_ERROR;
1399
1400 TRACE("socket %04lx, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen);
1401
1402 if (fd != -1)
1403 {
1404 if (!name || (name->sa_family && !SUPPORTED_PF(name->sa_family)))
1405 {
1406 SetLastError(WSAEAFNOSUPPORT);
1407 }
1408 else
1409 {
1410 union generic_unix_sockaddr uaddr;
1411 unsigned int uaddrlen = ws_sockaddr_ws2u(name, namelen, &uaddr);
1412 if (!uaddrlen)
1413 {
1414 SetLastError(WSAEFAULT);
1415 }
1416 else
1417 {
1418 #ifdef IPV6_V6ONLY
1419 const struct sockaddr_in6 *in6 = (const struct sockaddr_in6*) &uaddr;
1420 if (name->sa_family == WS_AF_INET6 &&
1421 !memcmp(&in6->sin6_addr, &in6addr_any, sizeof(struct in6_addr)))
1422 {
1423 int enable = 1;
1424 if (setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, &enable, sizeof(enable)) == -1)
1425 {
1426 release_sock_fd( s, fd );
1427 SetLastError(WSAEAFNOSUPPORT);
1428 return INVALID_SOCKET;
1429 }
1430 }
1431 #endif
1432 if (bind(fd, &uaddr.addr, uaddrlen) < 0)
1433 {
1434 int loc_errno = errno;
1435 WARN("\tfailure - errno = %i\n", errno);
1436 errno = loc_errno;
1437 switch (errno)
1438 {
1439 case EBADF:
1440 SetLastError(WSAENOTSOCK);
1441 break;
1442 case EADDRNOTAVAIL:
1443 SetLastError(WSAEINVAL);
1444 break;
1445 default:
1446 SetLastError(wsaErrno());
1447 break;
1448 }
1449 }
1450 else
1451 {
1452 res=0; /* success */
1453 }
1454 }
1455 }
1456 release_sock_fd( s, fd );
1457 }
1458 return res;
1459 }
1460
1461 /***********************************************************************
1462 * closesocket (WS2_32.3)
1463 */
1464 int WINAPI WS_closesocket(SOCKET s)
1465 {
1466 TRACE("socket %04lx\n", s);
1467 if (CloseHandle(SOCKET2HANDLE(s))) return 0;
1468 return SOCKET_ERROR;
1469 }
1470
1471 /***********************************************************************
1472 * connect (WS2_32.4)
1473 */
1474 int WINAPI WS_connect(SOCKET s, const struct WS_sockaddr* name, int namelen)
1475 {
1476 int fd = get_sock_fd( s, FILE_READ_DATA, NULL );
1477
1478 TRACE("socket %04lx, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen);
1479
1480 if (fd != -1)
1481 {
1482 union generic_unix_sockaddr uaddr;
1483 unsigned int uaddrlen = ws_sockaddr_ws2u(name, namelen, &uaddr);
1484
1485 if (!uaddrlen)
1486 {
1487 SetLastError(WSAEFAULT);
1488 }
1489 else
1490 {
1491 if (connect(fd, &uaddr.addr, uaddrlen) == 0)
1492 goto connect_success;
1493 }
1494
1495 if (errno == EINPROGRESS)
1496 {
1497 /* tell wineserver that a connection is in progress */
1498 _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
1499 FD_CONNECT|FD_READ|FD_WRITE,
1500 FD_WINE_CONNECTED|FD_WINE_LISTENING);
1501 if (_is_blocking(s))
1502 {
1503 int result;
1504 /* block here */
1505 do_block(fd, POLLIN | POLLOUT, -1);
1506 _sync_sock_state(s); /* let wineserver notice connection */
1507 /* retrieve any error codes from it */
1508 result = _get_sock_error(s, FD_CONNECT_BIT);
1509 if (result)
1510 SetLastError(result);
1511 else
1512 {
1513 goto connect_success;
1514 }
1515 }
1516 else
1517 {
1518 SetLastError(WSAEWOULDBLOCK);
1519 }
1520 }
1521 else
1522 {
1523 SetLastError(wsaErrno());
1524 }
1525 release_sock_fd( s, fd );
1526 }
1527 return SOCKET_ERROR;
1528
1529 connect_success:
1530 release_sock_fd( s, fd );
1531 _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE,
1532 FD_WINE_CONNECTED|FD_READ|FD_WRITE,
1533 FD_CONNECT|FD_WINE_LISTENING);
1534 return 0;
1535 }
1536
1537 /***********************************************************************
1538 * WSAConnect (WS2_32.30)
1539 */
1540 int WINAPI WSAConnect( SOCKET s, const struct WS_sockaddr* name, int namelen,
1541 LPWSABUF lpCallerData, LPWSABUF lpCalleeData,
1542 LPQOS lpSQOS, LPQOS lpGQOS )
1543 {
1544 if ( lpCallerData || lpCalleeData || lpSQOS || lpGQOS )
1545 FIXME("unsupported parameters!\n");
1546 return WS_connect( s, name, namelen );
1547 }
1548
1549
1550 /***********************************************************************
1551 * getpeername (WS2_32.5)
1552 */
1553 int WINAPI WS_getpeername(SOCKET s, struct WS_sockaddr *name, int *namelen)
1554 {
1555 int fd;
1556 int res;
1557
1558 TRACE("socket: %04lx, ptr %p, len %08x\n", s, name, *namelen);
1559
1560 /* Check if what we've received is valid. Should we use IsBadReadPtr? */
1561 if( (name == NULL) || (namelen == NULL) )
1562 {
1563 SetLastError( WSAEFAULT );
1564 return SOCKET_ERROR;
1565 }
1566
1567 fd = get_sock_fd( s, 0, NULL );
1568 res = SOCKET_ERROR;
1569
1570 if (fd != -1)
1571 {
1572 union generic_unix_sockaddr uaddr;
1573 unsigned int uaddrlen = sizeof(uaddr);
1574
1575 if (getpeername(fd, &uaddr.addr, &uaddrlen) != 0)
1576 {
1577 SetLastError(wsaErrno());
1578 }
1579 else if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0)
1580 {
1581 /* The buffer was too small */
1582 SetLastError(WSAEFAULT);
1583 }
1584 else
1585 {
1586 res=0;
1587 }
1588 release_sock_fd( s, fd );
1589 }
1590 return res;
1591 }
1592
1593 /***********************************************************************
1594 * getsockname (WS2_32.6)
1595 */
1596 int WINAPI WS_getsockname(SOCKET s, struct WS_sockaddr *name, int *namelen)
1597 {
1598 int fd;
1599 int res;
1600
1601 TRACE("socket: %04lx, ptr %p, len %8x\n", s, name, *namelen);
1602
1603 /* Check if what we've received is valid. Should we use IsBadReadPtr? */
1604 if( (name == NULL) || (namelen == NULL) )
1605 {
1606 SetLastError( WSAEFAULT );
1607 return SOCKET_ERROR;
1608 }
1609
1610 fd = get_sock_fd( s, 0, NULL );
1611 res = SOCKET_ERROR;
1612
1613 if (fd != -1)
1614 {
1615 union generic_unix_sockaddr uaddr;
1616 unsigned int uaddrlen = sizeof(uaddr);
1617
1618 if (getsockname(fd, &uaddr.addr, &uaddrlen) != 0)
1619 {
1620 SetLastError(wsaErrno());
1621 }
1622 else if (!is_sockaddr_bound(&uaddr.addr, uaddrlen))
1623 {
1624 SetLastError(WSAEINVAL);
1625 }
1626 else if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0)
1627 {
1628 /* The buffer was too small */
1629 SetLastError(WSAEFAULT);
1630 }
1631 else
1632 {
1633 res=0;
1634 }
1635 release_sock_fd( s, fd );
1636 }
1637 return res;
1638 }
1639
1640 /***********************************************************************
1641 * getsockopt (WS2_32.7)
1642 */
1643 INT WINAPI WS_getsockopt(SOCKET s, INT level,
1644 INT optname, char *optval, INT *optlen)
1645 {
1646 int fd;
1647 INT ret = 0;
1648
1649 TRACE("socket: %04lx, level 0x%x, name 0x%x, ptr %p, len %d\n",
1650 s, level, optname, optval, *optlen);
1651
1652 switch(level)
1653 {
1654 case WS_SOL_SOCKET:
1655 {
1656 switch(optname)
1657 {
1658 /* Handle common cases. The special cases are below, sorted
1659 * alphabetically */
1660 case WS_SO_ACCEPTCONN:
1661 case WS_SO_BROADCAST:
1662 case WS_SO_DEBUG:
1663 case WS_SO_ERROR:
1664 case WS_SO_KEEPALIVE:
1665 case WS_SO_OOBINLINE:
1666 case WS_SO_RCVBUF:
1667 case WS_SO_REUSEADDR:
1668 case WS_SO_SNDBUF:
1669 case WS_SO_TYPE:
1670 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
1671 return SOCKET_ERROR;
1672 convert_sockopt(&level, &optname);
1673 if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
1674 {
1675 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
1676 ret = SOCKET_ERROR;
1677 }
1678 release_sock_fd( s, fd );
1679 return ret;
1680
1681 case WS_SO_DONTLINGER:
1682 {
1683 struct linger lingval;
1684 unsigned int len = sizeof(struct linger);
1685
1686 if (!optlen || *optlen < sizeof(BOOL)|| !optval)
1687 {
1688 SetLastError(WSAEFAULT);
1689 return SOCKET_ERROR;
1690 }
1691 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
1692 return SOCKET_ERROR;
1693
1694 if (getsockopt(fd, SOL_SOCKET, SO_LINGER, &lingval, &len) != 0 )
1695 {
1696 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
1697 ret = SOCKET_ERROR;
1698 }
1699 else
1700 {
1701 *(BOOL *)optval = (lingval.l_onoff) ? FALSE : TRUE;
1702 *optlen = sizeof(BOOL);
1703 }
1704
1705 release_sock_fd( s, fd );
1706 return ret;
1707 }
1708
1709 /* As mentioned in setsockopt, Windows ignores this, so we
1710 * always return true here */
1711 case WS_SO_DONTROUTE:
1712 if (!optlen || *optlen < sizeof(BOOL) || !optval)
1713 {
1714 SetLastError(WSAEFAULT);
1715 return SOCKET_ERROR;
1716 }
1717 *(BOOL *)optval = TRUE;
1718 *optlen = sizeof(BOOL);
1719 return 0;
1720
1721 case WS_SO_LINGER:
1722 {
1723 struct linger lingval;
1724 unsigned int len = sizeof(struct linger);
1725
1726 /* struct linger and LINGER have different sizes */
1727 if (!optlen || *optlen < sizeof(LINGER) || !optval)
1728 {
1729 SetLastError(WSAEFAULT);
1730 return SOCKET_ERROR;
1731 }
1732 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
1733 return SOCKET_ERROR;
1734
1735 if (getsockopt(fd, SOL_SOCKET, SO_LINGER, &lingval, &len) != 0 )
1736 {
1737 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
1738 ret = SOCKET_ERROR;
1739 }
1740 else
1741 {
1742 ((LINGER *)optval)->l_onoff = lingval.l_onoff;
1743 ((LINGER *)optval)->l_linger = lingval.l_linger;
1744 *optlen = sizeof(struct linger);
1745 }
1746
1747 release_sock_fd( s, fd );
1748 return ret;
1749 }
1750
1751 case WS_SO_MAX_MSG_SIZE:
1752 if (!optlen || *optlen < sizeof(int) || !optval)
1753 {
1754 SetLastError(WSAEFAULT);
1755 return SOCKET_ERROR;
1756 }
1757 TRACE("getting global SO_MAX_MSG_SIZE = 65507\n");
1758 *(int *)optval = 65507;
1759 *optlen = sizeof(int);
1760 return 0;
1761
1762 /* SO_OPENTYPE does not require a valid socket handle. */
1763 case WS_SO_OPENTYPE:
1764 if (!optlen || *optlen < sizeof(int) || !optval)
1765 {
1766 SetLastError(WSAEFAULT);
1767 return SOCKET_ERROR;
1768 }
1769 *(int *)optval = get_per_thread_data()->opentype;
1770 *optlen = sizeof(int);
1771 TRACE("getting global SO_OPENTYPE = 0x%x\n", *((int*)optval) );
1772 return 0;
1773
1774 #ifdef SO_RCVTIMEO
1775 case WS_SO_RCVTIMEO:
1776 #endif
1777 #ifdef SO_SNDTIMEO
1778 case WS_SO_SNDTIMEO:
1779 #endif
1780 #if defined(SO_RCVTIMEO) || defined(SO_SNDTIMEO)
1781 {
1782 struct timeval tv;
1783 unsigned int len = sizeof(struct timeval);
1784
1785 if (!optlen || *optlen < sizeof(int)|| !optval)
1786 {
1787 SetLastError(WSAEFAULT);
1788 return SOCKET_ERROR;
1789 }
1790 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
1791 return SOCKET_ERROR;
1792
1793 convert_sockopt(&level, &optname);
1794 if (getsockopt(fd, level, optname, &tv, &len) != 0 )
1795 {
1796 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
1797 ret = SOCKET_ERROR;
1798 }
1799 else
1800 {
1801 *(int *)optval = tv.tv_sec * 1000 + tv.tv_usec / 1000;
1802 *optlen = sizeof(int);
1803 }
1804
1805 release_sock_fd( s, fd );
1806 return ret;
1807 }
1808 #endif
1809 default:
1810 TRACE("Unknown SOL_SOCKET optname: 0x%08x\n", optname);
1811 SetLastError(WSAENOPROTOOPT);
1812 return SOCKET_ERROR;
1813 } /* end switch(optname) */
1814 }/* end case WS_SOL_SOCKET */
1815 #ifdef HAVE_IPX
1816 case NSPROTO_IPX:
1817 {
1818 struct WS_sockaddr_ipx addr;
1819 IPX_ADDRESS_DATA *data;
1820 int namelen;
1821 switch(optname)
1822 {
1823 case IPX_PTYPE:
1824 if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR;
1825 #ifdef SOL_IPX
1826 if(getsockopt(fd, SOL_IPX, IPX_TYPE, optval, (unsigned int*)optlen) == -1)
1827 {
1828 ret = SOCKET_ERROR;
1829 }
1830 #else
1831 {
1832 struct ipx val;
1833 socklen_t len=sizeof(struct ipx);
1834 if(getsockopt(fd, 0, SO_DEFAULT_HEADERS, &val, &len) == -1 )
1835 ret = SOCKET_ERROR;
1836 else
1837 *optval = (int)val.ipx_pt;
1838 }
1839 #endif
1840 TRACE("ptype: %d (fd: %d)\n", *(int*)optval, fd);
1841 release_sock_fd( s, fd );
1842 return ret;
1843
1844 case IPX_ADDRESS:
1845 /*
1846 * On a Win2000 system with one network card there are usually
1847 * three ipx devices one with a speed of 28.8kbps, 10Mbps and 100Mbps.
1848 * Using this call you can then retrieve info about this all.
1849 * In case of Linux it is a bit different. Usually you have
1850 * only "one" device active and further it is not possible to
1851 * query things like the linkspeed.
1852 */
1853 FIXME("IPX_ADDRESS\n");
1854 namelen = sizeof(struct WS_sockaddr_ipx);
1855 memset(&addr, 0, sizeof(struct WS_sockaddr_ipx));
1856 WS_getsockname(s, (struct WS_sockaddr*)&addr, &namelen);
1857
1858 data = (IPX_ADDRESS_DATA*)optval;
1859 memcpy(data->nodenum,&addr.sa_nodenum,sizeof(data->nodenum));
1860 memcpy(data->netnum,&addr.sa_netnum,sizeof(data->netnum));
1861 data->adapternum = 0;
1862 data->wan = FALSE; /* We are not on a wan for now .. */
1863 data->status = FALSE; /* Since we are not on a wan, the wan link isn't up */
1864 data->maxpkt = 1467; /* This value is the default one, at least on Win2k/WinXP */
1865 data->linkspeed = 100000; /* Set the line speed in 100bit/s to 10 Mbit;
1866 * note 1MB = 1000kB in this case */
1867 return 0;
1868
1869 case IPX_MAX_ADAPTER_NUM:
1870 FIXME("IPX_MAX_ADAPTER_NUM\n");
1871 *(int*)optval = 1; /* As noted under IPX_ADDRESS we have just one card. */
1872 return 0;
1873
1874 default:
1875 FIXME("IPX optname:%x\n", optname);
1876 return SOCKET_ERROR;
1877 }/* end switch(optname) */
1878 } /* end case NSPROTO_IPX */
1879 #endif
1880 /* Levels WS_IPPROTO_TCP and WS_IPPROTO_IP convert directly */
1881 case WS_IPPROTO_TCP:
1882 switch(optname)
1883 {
1884 case WS_TCP_NODELAY:
1885 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
1886 return SOCKET_ERROR;
1887 convert_sockopt(&level, &optname);
1888 if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
1889 {
1890 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
1891 ret = SOCKET_ERROR;
1892 }
1893 release_sock_fd( s, fd );
1894 return ret;
1895 }
1896 FIXME("Unknown IPPROTO_TCP optname 0x%08x\n", optname);
1897 return SOCKET_ERROR;
1898
1899 case WS_IPPROTO_IP:
1900 switch(optname)
1901 {
1902 case WS_IP_ADD_MEMBERSHIP:
1903 case WS_IP_DROP_MEMBERSHIP:
1904 #ifdef IP_HDRINCL
1905 case WS_IP_HDRINCL:
1906 #endif
1907 case WS_IP_MULTICAST_IF:
1908 case WS_IP_MULTICAST_LOOP:
1909 case WS_IP_MULTICAST_TTL:
1910 case WS_IP_OPTIONS:
1911 case WS_IP_TOS:
1912 case WS_IP_TTL:
1913 if ( (fd = get_sock_fd( s, 0, NULL )) == -1)
1914 return SOCKET_ERROR;
1915 convert_sockopt(&level, &optname);
1916 if (getsockopt(fd, level, optname, optval, (unsigned int *)optlen) != 0 )
1917 {
1918 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
1919 ret = SOCKET_ERROR;
1920 }
1921 release_sock_fd( s, fd );
1922 return ret;
1923 case WS_IP_DONTFRAGMENT:
1924 FIXME("WS_IP_DONTFRAGMENT is always false!\n");
1925 *(BOOL*)optval = FALSE;
1926 return 0;
1927 }
1928 FIXME("Unknown IPPROTO_IP optname 0x%08x\n", optname);
1929 return SOCKET_ERROR;
1930
1931 default:
1932 FIXME("Unknown level: 0x%08x\n", level);
1933 return SOCKET_ERROR;
1934 } /* end switch(level) */
1935 }
1936
1937 /***********************************************************************
1938 * htonl (WINSOCK.8)
1939 * htonl (WS2_32.8)
1940 */
1941 WS_u_long WINAPI WS_htonl(WS_u_long hostlong)
1942 {
1943 return htonl(hostlong);
1944 }
1945
1946
1947 /***********************************************************************
1948 * htons (WINSOCK.9)
1949 * htons (WS2_32.9)
1950 */
1951 WS_u_short WINAPI WS_htons(WS_u_short hostshort)
1952 {
1953 return htons(hostshort);
1954 }
1955
1956 /***********************************************************************
1957 * WSAHtonl (WS2_32.46)
1958 * From MSDN description of error codes, this function should also
1959 * check if WinSock has been initialized and the socket is a valid
1960 * socket. But why? This function only translates a host byte order
1961 * u_long into a network byte order u_long...
1962 */
1963 int WINAPI WSAHtonl(SOCKET s, WS_u_long hostlong, WS_u_long *lpnetlong)
1964 {
1965 if (lpnetlong)
1966 {
1967 *lpnetlong = htonl(hostlong);
1968 return 0;
1969 }
1970 WSASetLastError(WSAEFAULT);
1971 return SOCKET_ERROR;
1972 }
1973
1974 /***********************************************************************
1975 * WSAHtons (WS2_32.47)
1976 * From MSDN description of error codes, this function should also
1977 * check if WinSock has been initialized and the socket is a valid
1978 * socket. But why? This function only translates a host byte order
1979 * u_short into a network byte order u_short...
1980 */
1981 int WINAPI WSAHtons(SOCKET s, WS_u_short hostshort, WS_u_short *lpnetshort)
1982 {
1983
1984 if (lpnetshort)
1985 {
1986 *lpnetshort = htons(hostshort);
1987 return 0;
1988 }
1989 WSASetLastError(WSAEFAULT);
1990 return SOCKET_ERROR;
1991 }
1992
1993
1994 /***********************************************************************
1995 * inet_addr (WINSOCK.10)
1996 * inet_addr (WS2_32.11)
1997 */
1998 WS_u_long WINAPI WS_inet_addr(const char *cp)
1999 {
2000 if (!cp) return INADDR_NONE;
2001 return inet_addr(cp);
2002 }
2003
2004
2005 /***********************************************************************
2006 * ntohl (WINSOCK.14)
2007 * ntohl (WS2_32.14)
2008 */
2009 WS_u_long WINAPI WS_ntohl(WS_u_long netlong)
2010 {
2011 return ntohl(netlong);
2012 }
2013
2014
2015 /***********************************************************************
2016 * ntohs (WINSOCK.15)
2017 * ntohs (WS2_32.15)
2018 */
2019 WS_u_short WINAPI WS_ntohs(WS_u_short netshort)
2020 {
2021 return ntohs(netshort);
2022 }
2023
2024
2025 /***********************************************************************
2026 * inet_ntoa (WS2_32.12)
2027 */
2028 char* WINAPI WS_inet_ntoa(struct WS_in_addr in)
2029 {
2030 /* use "buffer for dummies" here because some applications have a
2031 * propensity to decode addresses in ws_hostent structure without
2032 * saving them first...
2033 */
2034 static char dbuffer[16]; /* Yes, 16: 4*3 digits + 3 '.' + 1 '\0' */
2035
2036 char* s = inet_ntoa(*((struct in_addr*)&in));
2037 if( s )
2038 {
2039 strcpy(dbuffer, s);
2040 return dbuffer;
2041 }
2042 SetLastError(wsaErrno());
2043 return NULL;
2044 }
2045
2046 /**********************************************************************
2047 * WSAIoctl (WS2_32.50)
2048 *
2049 */
2050 INT WINAPI WSAIoctl(SOCKET s,
2051 DWORD dwIoControlCode,
2052 LPVOID lpvInBuffer,
2053 DWORD cbInBuffer,
2054 LPVOID lpbOutBuffer,
2055 DWORD cbOutBuffer,
2056 LPDWORD lpcbBytesReturned,
2057 LPWSAOVERLAPPED lpOverlapped,
2058 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine)
2059 {
2060 TRACE("%ld, 0x%08x, %p, %d, %p, %d, %p, %p, %p\n",
2061 s, dwIoControlCode, lpvInBuffer, cbInBuffer, lpbOutBuffer,
2062 cbOutBuffer, lpcbBytesReturned, lpOverlapped, lpCompletionRoutine);
2063
2064 switch( dwIoControlCode )
2065 {
2066 case WS_FIONBIO:
2067 if (cbInBuffer != sizeof(WS_u_long)) {
2068 WSASetLastError(WSAEFAULT);
2069 return SOCKET_ERROR;
2070 }
2071 return WS_ioctlsocket( s, WS_FIONBIO, lpvInBuffer);
2072
2073 case WS_FIONREAD:
2074 if (cbOutBuffer != sizeof(WS_u_long)) {
2075 WSASetLastError(WSAEFAULT);
2076 return SOCKET_ERROR;
2077 }
2078 return WS_ioctlsocket( s, WS_FIONREAD, lpbOutBuffer);
2079
2080 case WS_SIO_GET_INTERFACE_LIST:
2081 {
2082 INTERFACE_INFO* intArray = (INTERFACE_INFO*)lpbOutBuffer;
2083 DWORD size, numInt, apiReturn;
2084 int fd;
2085
2086 TRACE("-> SIO_GET_INTERFACE_LIST request\n");
2087
2088 if (!lpbOutBuffer)
2089 {
2090 WSASetLastError(WSAEFAULT);
2091 return SOCKET_ERROR;
2092 }
2093 if (!lpcbBytesReturned)
2094 {
2095 WSASetLastError(WSAEFAULT);
2096 return SOCKET_ERROR;
2097 }
2098
2099 fd = get_sock_fd( s, 0, NULL );
2100 if (fd == -1) return SOCKET_ERROR;
2101
2102 apiReturn = GetAdaptersInfo(NULL, &size);
2103 if (apiReturn == ERROR_NO_DATA)
2104 {
2105 numInt = 0;
2106 }
2107 else if (apiReturn == ERROR_BUFFER_OVERFLOW)
2108 {
2109 PIP_ADAPTER_INFO table = HeapAlloc(GetProcessHeap(),0,size);
2110
2111 if (table)
2112 {
2113 if (GetAdaptersInfo(table, &size) == NO_ERROR)
2114 {
2115 PIP_ADAPTER_INFO ptr;
2116
2117 if (size*sizeof(INTERFACE_INFO)/sizeof(IP_ADAPTER_INFO) > cbOutBuffer)
2118 {
2119 WARN("Buffer too small = %u, cbOutBuffer = %u\n", size, cbOutBuffer);
2120 HeapFree(GetProcessHeap(),0,table);
2121 release_sock_fd( s, fd );
2122 WSASetLastError(WSAEFAULT);
2123 return SOCKET_ERROR;
2124 }
2125 for (ptr = table, numInt = 0; ptr;
2126 ptr = ptr->Next, intArray++, numInt++)
2127 {
2128 unsigned int addr, mask, bcast;
2129 struct ifreq ifInfo;
2130
2131 /* Socket Status Flags */
2132 lstrcpynA(ifInfo.ifr_name, ptr->AdapterName, IFNAMSIZ);
2133 if (ioctl(fd, SIOCGIFFLAGS, &ifInfo) < 0)
2134 {
2135 ERR("Error obtaining status flags for socket!\n");
2136 HeapFree(GetProcessHeap(),0,table);
2137 release_sock_fd( s, fd );
2138 WSASetLastError(WSAEINVAL);
2139 return SOCKET_ERROR;
2140 }
2141 else
2142 {
2143 /* set flags; the values of IFF_* are not the same
2144 under Linux and Windows, therefore must generate
2145 new flags */
2146 intArray->iiFlags = 0;
2147 if (ifInfo.ifr_flags & IFF_BROADCAST)
2148 intArray->iiFlags |= WS_IFF_BROADCAST;
2149 #ifdef IFF_POINTOPOINT
2150 if (ifInfo.ifr_flags & IFF_POINTOPOINT)
2151 intArray->iiFlags |= WS_IFF_POINTTOPOINT;
2152 #endif
2153 if (ifInfo.ifr_flags & IFF_LOOPBACK)
2154 intArray->iiFlags |= WS_IFF_LOOPBACK;
2155 if (ifInfo.ifr_flags & IFF_UP)
2156 intArray->iiFlags |= WS_IFF_UP;
2157 if (ifInfo.ifr_flags & IFF_MULTICAST)
2158 intArray->iiFlags |= WS_IFF_MULTICAST;
2159 }
2160
2161 addr = inet_addr(ptr->IpAddressList.IpAddress.String);
2162 mask = inet_addr(ptr->IpAddressList.IpMask.String);
2163 bcast = addr | ~mask;
2164 intArray->iiAddress.AddressIn.sin_family = AF_INET;
2165 intArray->iiAddress.AddressIn.sin_port = 0;
2166 intArray->iiAddress.AddressIn.sin_addr.WS_s_addr =
2167 addr;
2168 intArray->iiNetmask.AddressIn.sin_family = AF_INET;
2169 intArray->iiNetmask.AddressIn.sin_port = 0;
2170 intArray->iiNetmask.AddressIn.sin_addr.WS_s_addr =
2171 mask;
2172 intArray->iiBroadcastAddress.AddressIn.sin_family =
2173 AF_INET;
2174 intArray->iiBroadcastAddress.AddressIn.sin_port = 0;
2175 intArray->iiBroadcastAddress.AddressIn.sin_addr.
2176 WS_s_addr = bcast;
2177 }
2178 }
2179 else
2180 {
2181 ERR("Unable to get interface table!\n");
2182 release_sock_fd( s, fd );
2183 HeapFree(GetProcessHeap(),0,table);
2184 WSASetLastError(WSAEINVAL);
2185 return SOCKET_ERROR;
2186 }
2187 HeapFree(GetProcessHeap(),0,table);
2188 }
2189 else
2190 {
2191 release_sock_fd( s, fd );
2192 WSASetLastError(WSAEINVAL);
2193 return SOCKET_ERROR;
2194 }
2195 }
2196 else
2197 {
2198 ERR("Unable to get interface table!\n");
2199 release_sock_fd( s, fd );
2200 WSASetLastError(WSAEINVAL);
2201 return SOCKET_ERROR;
2202 }
2203 /* Calculate the size of the array being returned */
2204 *lpcbBytesReturned = sizeof(INTERFACE_INFO) * numInt;
2205 release_sock_fd( s, fd );
2206 break;
2207 }
2208
2209 case WS_SIO_ADDRESS_LIST_CHANGE:
2210 FIXME("-> SIO_ADDRESS_LIST_CHANGE request: stub\n");
2211 /* FIXME: error and return code depend on whether socket was created
2212 * with WSA_FLAG_OVERLAPPED, but there is no easy way to get this */
2213 break;
2214
2215 case WS_SIO_ADDRESS_LIST_QUERY:
2216 {
2217 DWORD size;
2218
2219 TRACE("-> SIO_ADDRESS_LIST_QUERY request\n");
2220
2221 if (!lpcbBytesReturned)
2222 {
2223 WSASetLastError(WSAEFAULT);
2224 return SOCKET_ERROR;
2225 }
2226
2227 if (GetAdaptersInfo(NULL, &size) == ERROR_BUFFER_OVERFLOW)
2228 {
2229 IP_ADAPTER_INFO *p, *table = HeapAlloc(GetProcessHeap(), 0, size);
2230 DWORD need, num;
2231
2232 if (!table || GetAdaptersInfo(table, &size))
2233 {
2234 HeapFree(GetProcessHeap(), 0, table);
2235 WSASetLastError(WSAEINVAL);
2236 return SOCKET_ERROR;
2237 }
2238
2239 for (p = table, num = 0; p; p = p->Next)
2240 if (p->IpAddressList.IpAddress.String[0]) num++;
2241
2242 need = sizeof(SOCKET_ADDRESS_LIST) + sizeof(SOCKET_ADDRESS) * (num - 1);
2243 need += sizeof(SOCKADDR) * num;
2244 *lpcbBytesReturned = need;
2245
2246 if (need > cbOutBuffer)
2247 {
2248 HeapFree(GetProcessHeap(), 0, table);
2249 WSASetLastError(WSAEFAULT);
2250 return SOCKET_ERROR;
2251 }
2252
2253 if (lpbOutBuffer)
2254 {
2255 unsigned int i;
2256 SOCKET_ADDRESS *sa;
2257 SOCKET_ADDRESS_LIST *sa_list = (SOCKET_ADDRESS_LIST *)lpbOutBuffer;
2258 SOCKADDR_IN *sockaddr;
2259
2260 sa = sa_list->Address;
2261 sockaddr = (SOCKADDR_IN *)((char *)sa + num * sizeof(SOCKET_ADDRESS));
2262 sa_list->iAddressCount = num;
2263
2264 for (p = table, i = 0; p; p = p->Next)
2265 {
2266 if (!p->IpAddressList.IpAddress.String[0]) continue;
2267
2268 sa[i].lpSockaddr = (SOCKADDR *)&sockaddr[i];
2269 sa[i].iSockaddrLength = sizeof(SOCKADDR);
2270
2271 sockaddr[i].sin_family = AF_INET;
2272 sockaddr[i].sin_port = 0;
2273 sockaddr[i].sin_addr.WS_s_addr = inet_addr(p->IpAddressList.IpAddress.String);
2274 i++;
2275 }
2276 }
2277
2278 HeapFree(GetProcessHeap(), 0, table);
2279 return 0;
2280 }
2281 else
2282 {
2283 WARN("unable to get IP address list\n");
2284 WSASetLastError(WSAEINVAL);
2285 return SOCKET_ERROR;
2286 }
2287 }
2288 case WS_SIO_FLUSH:
2289 FIXME("SIO_FLUSH: stub.\n");
2290 break;
2291
2292 case WS_SIO_GET_EXTENSION_FUNCTION_POINTER:
2293 FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER %s: stub\n", debugstr_guid(lpvInBuffer));
2294 WSASetLastError(WSAEOPNOTSUPP);
2295 return SOCKET_ERROR;
2296
2297 default:
2298 FIXME("unsupported WS_IOCTL cmd (%08x)\n", dwIoControlCode);
2299 WSASetLastError(WSAEOPNOTSUPP);
2300 return SOCKET_ERROR;
2301 }
2302
2303 return 0;
2304 }
2305
2306
2307 /***********************************************************************
2308 * ioctlsocket (WS2_32.10)
2309 */
2310 int WINAPI WS_ioctlsocket(SOCKET s, LONG cmd, WS_u_long *argp)
2311 {
2312 int fd;
2313 long newcmd = cmd;
2314
2315 TRACE("socket %04lx, cmd %08x, ptr %p\n", s, cmd, argp);
2316
2317 switch( cmd )
2318 {
2319 case WS_FIONREAD:
2320 newcmd=FIONREAD;
2321 break;
2322
2323 case WS_FIONBIO:
2324 if( _get_sock_mask(s) )
2325 {
2326 /* AsyncSelect()'ed sockets are always nonblocking */
2327 if (*argp) return 0;
2328 SetLastError(WSAEINVAL);
2329 return SOCKET_ERROR;
2330 }
2331 fd = get_sock_fd( s, 0, NULL );
2332 if (fd != -1)
2333 {
2334 int ret;
2335 if (*argp)
2336 {
2337 _enable_event(SOCKET2HANDLE(s), 0, FD_WINE_NONBLOCKING, 0);
2338 ret = fcntl( fd, F_SETFL, O_NONBLOCK );
2339 }
2340 else
2341 {
2342 _enable_event(SOCKET2HANDLE(s), 0, 0, FD_WINE_NONBLOCKING);
2343 ret = fcntl( fd, F_SETFL, 0 );
2344 }
2345 release_sock_fd( s, fd );
2346 if (!ret) return 0;
2347 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2348 }
2349 return SOCKET_ERROR;
2350
2351 case WS_SIOCATMARK:
2352 newcmd=SIOCATMARK;
2353 break;
2354
2355 case WS_FIOASYNC:
2356 WARN("Warning: WS1.1 shouldn't be using async I/O\n");
2357 SetLastError(WSAEINVAL);
2358 return SOCKET_ERROR;
2359
2360 case SIOCGIFBRDADDR:
2361 case SIOCGIFNETMASK:
2362 case SIOCGIFADDR:
2363 /* These don't need any special handling. They are used by
2364 WsControl, and are here to suppress an unnecessary warning. */
2365 break;
2366
2367 default:
2368 /* Netscape tries hard to use bogus ioctl 0x667e */
2369 /* FIXME: 0x667e above is ('f' << 8) | 126, and is a low word of
2370 * FIONBIO (_IOW('f', 126, u_long)), how that should be handled?
2371 */
2372 WARN("\tunknown WS_IOCTL cmd (%08x)\n", cmd);
2373 break;
2374 }
2375
2376 fd = get_sock_fd( s, 0, NULL );
2377 if (fd != -1)
2378 {
2379 if( ioctl(fd, newcmd, (char*)argp ) == 0 )
2380 {
2381 release_sock_fd( s, fd );
2382 return 0;
2383 }
2384 SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno());
2385 release_sock_fd( s, fd );
2386 }
2387 return SOCKET_ERROR;
2388 }
2389
2390 /***********************************************************************
2391 * listen (WS2_32.13)
2392 */
2393 int WINAPI WS_listen(SOCKET s, int backlog)
2394 {
2395 int fd = get_sock_fd( s, FILE_READ_DATA, NULL );
2396
2397 TRACE("socket %04lx, backlog %d\n", s, backlog);
2398 if (fd != -1)
2399 {
2400 if (listen(fd, backlog) == 0)
2401 {
2402 release_sock_fd( s, fd );
2403 _enable_event(SOCKET2HANDLE(s), FD_ACCEPT,
2404 FD_WINE_LISTENING,
2405 FD_CONNECT|FD_WINE_CONNECTED);
2406 return 0;
2407 }
2408 SetLastError(wsaErrno());
2409 release_sock_fd( s, fd );
2410 }
2411 return SOCKET_ERROR;
2412 }
2413
2414 /***********************************************************************
2415 * recv (WS2_32.16)
2416 */
2417 int WINAPI WS_recv(SOCKET s, char *buf, int len, int flags)
2418 {
2419 DWORD n, dwFlags = flags;
2420 WSABUF wsabuf;
2421
2422 wsabuf.len = len;
2423 wsabuf.buf = buf;
2424
2425 if ( WSARecvFrom(s, &wsabuf, 1, &n, &dwFlags, NULL, NULL, NULL, NULL) == SOCKET_ERROR )
2426 return SOCKET_ERROR;
2427 else
2428 return n;
2429 }
2430
2431 /***********************************************************************
2432 * recvfrom (WS2_32.17)
2433 */
2434 int WINAPI WS_recvfrom(SOCKET s, char *buf, INT len, int flags,
2435 struct WS_sockaddr *from, int *fromlen)
2436 {
2437 DWORD n, dwFlags = flags;
2438 WSABUF wsabuf;
2439
2440 wsabuf.len = len;
2441 wsabuf.buf = buf;
2442
2443 if ( WSARecvFrom(s, &wsabuf, 1, &n, &dwFlags, from, fromlen, NULL, NULL) == SOCKET_ERROR )
2444 return SOCKET_ERROR;
2445 else
2446 return n;
2447 }
2448
2449 /* allocate a poll array for the corresponding fd sets */
2450 static struct pollfd *fd_sets_to_poll( const WS_fd_set *readfds, const WS_fd_set *writefds,
2451 const WS_fd_set *exceptfds, int *count_ptr )
2452 {
2453 int i, j = 0, count = 0;
2454 struct pollfd *fds;
2455
2456 if (readfds) count += readfds->fd_count;
2457 if (writefds) count += writefds->fd_count;
2458 if (exceptfds) count += exceptfds->fd_count;
2459 *count_ptr = count;
2460 if (!count) return NULL;
2461 if (!(fds = HeapAlloc( GetProcessHeap(), 0, count * sizeof(fds[0])))) return NULL;
2462 if (readfds)
2463 for (i = 0; i < readfds->fd_count; i++, j++)
2464 {
2465 fds[j].fd = get_sock_fd( readfds->fd_array[i], FILE_READ_DATA, NULL );
2466 fds[j].events = POLLIN;
2467 fds[j].revents = 0;
2468 }
2469 if (writefds)
2470 for (i = 0; i < writefds->fd_count; i++, j++)
2471 {
2472 fds[j].fd = get_sock_fd( writefds->fd_array[i], FILE_WRITE_DATA, NULL );
2473 fds[j].events = POLLOUT;
2474 fds[j].revents = 0;
2475 }
2476 if (exceptfds)
2477 for (i = 0; i < exceptfds->fd_count; i++, j++)
2478 {
2479 fds[j].fd = get_sock_fd( exceptfds->fd_array[i], 0, NULL );
2480 fds[j].events = POLLHUP;
2481 fds[j].revents = 0;
2482 }
2483 return fds;
2484 }
2485
2486 /* release the file descriptor obtained in fd_sets_to_poll */
2487 /* must be called with the original fd_set arrays, before calling get_poll_results */
2488 static void release_poll_fds( const WS_fd_set *readfds, const WS_fd_set *writefds,
2489 const WS_fd_set *exceptfds, struct pollfd *fds )
2490 {
2491 int i, j = 0;
2492
2493 if (readfds)
2494 {
2495 for (i = 0; i < readfds->fd_count; i++, j++)
2496 if (fds[j].fd != -1) release_sock_fd( readfds->fd_array[i], fds[j].fd );
2497 }
2498 if (writefds)
2499 {
2500 for (i = 0; i < writefds->fd_count; i++, j++)
2501 if (fds[j].fd != -1) release_sock_fd( writefds->fd_array[i], fds[j].fd );
2502 }
2503 if (exceptfds)
2504 {
2505 for (i = 0; i < exceptfds->fd_count; i++, j++)
2506 if (fds[j].fd != -1)
2507 {
2508 /* make sure we have a real error before releasing the fd */
2509 if (!sock_error_p( fds[j].fd )) fds[j].revents = 0;
2510 release_sock_fd( exceptfds->fd_array[i], fds[j].fd );
2511 }
2512 }
2513 }
2514
2515 /* map the poll results back into the Windows fd sets */
2516 static int get_poll_results( WS_fd_set *readfds, WS_fd_set *writefds, WS_fd_set *exceptfds,
2517 const struct pollfd *fds )
2518 {
2519 int i, j = 0, k, total = 0;
2520
2521 if (readfds)
2522 {
2523 for (i = k = 0; i < readfds->fd_count; i++, j++)
2524 if (fds[j].revents) readfds->fd_array[k++] = readfds->fd_array[i];
2525 readfds->fd_count = k;
2526 total += k;
2527 }
2528 if (writefds)
2529 {
2530 for (i = k = 0; i < writefds->fd_count; i++, j++)
2531 if (fds[j].revents) writefds->fd_array[k++] = writefds->fd_array[i];
2532 writefds->fd_count = k;
2533 total += k;
2534 }
2535 if (exceptfds)
2536 {
2537 for (i = k = 0; i < exceptfds->fd_count; i++, j++)
2538 if (fds[j].revents) exceptfds->fd_array[k++] = exceptfds->fd_array[i];
2539 exceptfds->fd_count = k;
2540 total += k;
2541 }
2542 return total;
2543 }
2544
2545
2546 /***********************************************************************
2547 * select (WS2_32.18)
2548 */
2549 int WINAPI WS_select(int nfds, WS_fd_set *ws_readfds,
2550 WS_fd_set *ws_writefds, WS_fd_set *ws_exceptfds,
2551 const struct WS_timeval* ws_timeout)
2552 {
2553 struct pollfd *pollfds;
2554 int count, ret, timeout = -1;
2555
2556 TRACE("read %p, write %p, excp %p timeout %p\n",
2557 ws_readfds, ws_writefds, ws_exceptfds, ws_timeout);
2558
2559 if (!(pollfds = fd_sets_to_poll( ws_readfds, ws_writefds, ws_exceptfds, &count )) && count)
2560 {
2561 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
2562 return SOCKET_ERROR;
2563 }
2564
2565 if (ws_timeout) timeout = (ws_timeout->tv_sec * 1000) + (ws_timeout->tv_usec + 999) / 1000;
2566
2567 ret = poll( pollfds, count, timeout );
2568 release_poll_fds( ws_readfds, ws_writefds, ws_exceptfds, pollfds );
2569
2570 if (ret == -1) SetLastError(wsaErrno());
2571 else ret = get_poll_results( ws_readfds, ws_writefds, ws_exceptfds, pollfds );
2572 HeapFree( GetProcessHeap(), 0, pollfds );
2573 return ret;
2574 }
2575
2576 /* helper to send completion messages for client-only i/o operation case */
2577 static void WS_AddCompletion( SOCKET sock, ULONG_PTR CompletionValue, NTSTATUS CompletionStatus, ULONG_PTR Information )
2578 {
2579 NTSTATUS status;
2580
2581 SERVER_START_REQ( add_fd_completion )
2582 {
2583 req->handle = SOCKET2HANDLE(sock);
2584 req->cvalue = CompletionValue;
2585 req->status = CompletionStatus;
2586 req->information = Information;
2587 status = wine_server_call( req );
2588 }
2589 SERVER_END_REQ;
2590 }
2591
2592
2593 /***********************************************************************
2594 * send (WS2_32.19)
2595 */
2596 int WINAPI WS_send(SOCKET s, const char *buf, int len, int flags)
2597 {
2598 DWORD n;
2599 WSABUF wsabuf;
2600
2601 wsabuf.len = len;
2602 wsabuf.buf = (char*) buf;
2603
2604 if ( WSASendTo( s, &wsabuf, 1, &n, flags, NULL, 0, NULL, NULL) == SOCKET_ERROR )
2605 return SOCKET_ERROR;
2606 else
2607 return n;
2608 }
2609
2610 /***********************************************************************
2611 * WSASend (WS2_32.72)
2612 */
2613 INT WINAPI WSASend( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
2614 LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
2615 LPWSAOVERLAPPED lpOverlapped,
2616 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
2617 {
2618 return WSASendTo( s, lpBuffers, dwBufferCount, lpNumberOfBytesSent, dwFlags,
2619 NULL, 0, lpOverlapped, lpCompletionRoutine );
2620 }
2621
2622 /***********************************************************************
2623 * WSASendDisconnect (WS2_32.73)
2624 */
2625 INT WINAPI WSASendDisconnect( SOCKET s, LPWSABUF lpBuffers )
2626 {
2627 return WS_shutdown( s, SD_SEND );
2628 }
2629
2630
2631 /***********************************************************************
2632 * WSASendTo (WS2_32.74)
2633 */
2634 INT WINAPI WSASendTo( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
2635 LPDWORD lpNumberOfBytesSent, DWORD dwFlags,
2636 const struct WS_sockaddr *to, int tolen,
2637 LPWSAOVERLAPPED lpOverlapped,
2638 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
2639 {
2640 unsigned int i, options;
2641 int n, fd, err;
2642 struct iovec iovec[WS_MSG_MAXIOVLEN];
2643 ULONG_PTR cvalue = (lpOverlapped && ((ULONG_PTR)lpOverlapped->hEvent & 1) == 0) ? (ULONG_PTR)lpOverlapped : 0;
2644
2645 TRACE("socket %04lx, wsabuf %p, nbufs %d, flags %d, to %p, tolen %d, ovl %p, func %p\n",
2646 s, lpBuffers, dwBufferCount, dwFlags,
2647 to, tolen, lpOverlapped, lpCompletionRoutine);
2648
2649 if (dwBufferCount > WS_MSG_MAXIOVLEN)
2650 {
2651 WSASetLastError( WSAEINVAL );
2652 return SOCKET_ERROR;
2653 }
2654
2655 fd = get_sock_fd( s, FILE_WRITE_DATA, &options );
2656 TRACE( "fd=%d, options=%x\n", fd, options );
2657
2658 if ( fd == -1 ) return SOCKET_ERROR;
2659
2660 if ( !lpNumberOfBytesSent )
2661 {
2662 err = WSAEFAULT;
2663 goto error;
2664 }
2665
2666 for ( i = 0; i < dwBufferCount; i++ )
2667 {
2668 iovec[i].iov_base = lpBuffers[i].buf;
2669 iovec[i].iov_len = lpBuffers[i].len;
2670 }
2671
2672 for (;;)
2673 {
2674 n = WS2_send( fd, iovec, dwBufferCount, to, tolen, dwFlags );
2675 if (n != -1 || errno != EINTR) break;
2676 }
2677 if (n == -1 && errno != EAGAIN)
2678 {
2679 err = wsaErrno();
2680 if (cvalue) WS_AddCompletion( s, cvalue, err, 0 );
2681 goto error;
2682 }
2683
2684 if ((lpOverlapped || lpCompletionRoutine) &&
2685 !(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)))
2686 {
2687 IO_STATUS_BLOCK *iosb;
2688 struct ws2_async *wsa = HeapAlloc( GetProcessHeap(), 0, sizeof(*wsa) );
2689
2690 if ( !wsa )
2691 {
2692 err = WSAEFAULT;
2693 goto error;
2694 }
2695 release_sock_fd( s, fd );
2696
2697 wsa->hSocket = SOCKET2HANDLE(s);
2698 wsa->addr = (struct WS_sockaddr *)to;
2699 wsa->addrlen.val = tolen;
2700 wsa->flags = 0;
2701 wsa->user_overlapped = lpOverlapped;
2702 wsa->completion_func = lpCompletionRoutine;
2703 wsa->n_iovecs = dwBufferCount;
2704 memcpy( wsa->iovec, iovec, dwBufferCount * sizeof(*iovec) );
2705
2706 iosb = lpOverlapped ? (IO_STATUS_BLOCK *)lpOverlapped : &wsa->local_iosb;
2707 if (n == -1)
2708 {
2709 iosb->u.Status = STATUS_PENDING;
2710 iosb->Information = 0;
2711
2712 SERVER_START_REQ( register_async )
2713 {
2714 req->handle = wsa->hSocket;
2715 req->type = ASYNC_TYPE_WRITE;
2716 req->async.callback = WS2_async_send;
2717 req->async.iosb = iosb;
2718 req->async.arg = wsa;
2719 req->async.apc = ws2_async_apc;
2720 req->async.event = lpCompletionRoutine ? 0 : lpOverlapped->hEvent;
2721 req->async.cvalue = cvalue;
2722 err = wine_server_call( req );
2723 }
2724 SERVER_END_REQ;
2725
2726 if (err != STATUS_PENDING) HeapFree( GetProcessHeap(), 0, wsa );
2727 WSASetLastError( NtStatusToWSAError( err ));
2728 return SOCKET_ERROR;
2729 }
2730
2731 iosb->u.Status = STATUS_SUCCESS;
2732 iosb->Information = n;
2733 *lpNumberOfBytesSent = n;
2734 if (!wsa->completion_func)
2735 {
2736 if (cvalue) WS_AddCompletion( s, cvalue, STATUS_SUCCESS, n );
2737 SetEvent( lpOverlapped->hEvent );
2738 HeapFree( GetProcessHeap(), 0, wsa );
2739 }
2740 else NtQueueApcThread( GetCurrentThread(), (PNTAPCFUNC)ws2_async_apc,
2741 (ULONG_PTR)wsa, (ULONG_PTR)iosb, 0 );
2742 return 0;
2743 }
2744
2745 if ( _is_blocking(s) )
2746 {
2747 /* On a blocking non-overlapped stream socket,
2748 * sending blocks until the entire buffer is sent. */
2749 DWORD timeout_start = GetTickCount();
2750 unsigned int first_buff = 0;
2751
2752 *lpNumberOfBytesSent = 0;
2753
2754 while (first_buff < dwBufferCount)
2755 {
2756 struct pollfd pfd;
2757 int timeout = GET_SNDTIMEO(fd);
2758
2759 if (n >= 0)
2760 {
2761 *lpNumberOfBytesSent += n;
2762 while (first_buff < dwBufferCount && iovec[first_buff].iov_len <= n)
2763 n -= iovec[first_buff++].iov_len;
2764 if (first_buff >= dwBufferCount) break;
2765 iovec[first_buff].iov_base = (char*)iovec[first_buff].iov_base + n;
2766 iovec[first_buff].iov_len -= n;
2767 }
2768
2769 if (timeout != -1)
2770 {
2771 timeout -= GetTickCount() - timeout_start;
2772 if (timeout < 0) timeout = 0;
2773 }
2774
2775 pfd.fd = fd;
2776 pfd.events = POLLOUT;
2777
2778 if (!timeout || !poll( &pfd, 1, timeout ))
2779 {
2780 err = WSAETIMEDOUT;
2781 goto error; /* msdn says a timeout in send is fatal */
2782 }
2783
2784 n = WS2_send( fd, iovec + first_buff, dwBufferCount - first_buff, to, tolen, dwFlags );
2785 if (n == -1 && errno != EAGAIN && errno != EINTR)
2786 {
2787 err = wsaErrno();
2788 goto error;
2789 }
2790 }
2791 }
2792 else /* non-blocking */
2793 {
2794 _enable_event(SOCKET2HANDLE(s), FD_WRITE, 0, 0);
2795 if (n == -1)
2796 {
2797 err = WSAEWOULDBLOCK;
2798 goto error;
2799 }
2800 *lpNumberOfBytesSent = n;
2801 }
2802
2803 TRACE(" -> %i bytes\n", *lpNumberOfBytesSent);
2804
2805 release_sock_fd( s, fd );
2806 return 0;
2807
2808 error:
2809 release_sock_fd( s, fd );
2810 WARN(" -> ERROR %d\n", err);
2811 WSASetLastError(err);
2812 return SOCKET_ERROR;
2813 }
2814
2815 /***********************************************************************
2816 * sendto (WS2_32.20)
2817 */
2818 int WINAPI WS_sendto(SOCKET s, const char *buf, int len, int flags,
2819 const struct WS_sockaddr *to, int tolen)
2820 {
2821 DWORD n;
2822 WSABUF wsabuf;
2823
2824 wsabuf.len = len;
2825 wsabuf.buf = (char*) buf;
2826
2827 if ( WSASendTo(s, &wsabuf, 1, &n, flags, to, tolen, NULL, NULL) == SOCKET_ERROR )
2828 return SOCKET_ERROR;
2829 else
2830 return n;
2831 }
2832
2833 /***********************************************************************
2834 * setsockopt (WS2_32.21)
2835 */
2836 int WINAPI WS_setsockopt(SOCKET s, int level, int optname,
2837 const char *optval, int optlen)
2838 {
2839 int fd;
2840 int woptval;
2841 struct linger linger;
2842 struct timeval tval;
2843
2844 TRACE("socket: %04lx, level 0x%x, name 0x%x, ptr %p, len %d\n",
2845 s, level, optname, optval, optlen);
2846
2847 /* some broken apps pass the value directly instead of a pointer to it */
2848 if(IS_INTRESOURCE(optval))
2849 {
2850 SetLastError(WSAEFAULT);
2851 return SOCKET_ERROR;
2852 }
2853
2854 switch(level)
2855 {
2856 case WS_SOL_SOCKET:
2857 switch(optname)
2858 {
2859 /* Some options need some conversion before they can be sent to
2860 * setsockopt. The conversions are done here, then they will fall though
2861 * to the general case. Special options that are not passed to
2862 * setsockopt follow below that.*/
2863
2864 case WS_SO_DONTLINGER:
2865 linger.l_onoff = *((const int*)optval) ? 0: 1;
2866 linger.l_linger = 0;
2867 level = SOL_SOCKET;
2868 optname = SO_LINGER;
2869 optval = (char*)&linger;
2870 optlen = sizeof(struct linger);
2871 break;
2872
2873 case WS_SO_LINGER:
2874 linger.l_onoff = ((LINGER*)optval)->l_onoff;
2875 linger.l_linger = ((LINGER*)optval)->l_linger;
2876 /* FIXME: what is documented behavior if SO_LINGER optval
2877 is null?? */
2878 level = SOL_SOCKET;
2879 optname = SO_LINGER;
2880 optval = (char*)&linger;
2881 optlen = sizeof(struct linger);
2882 break;
2883
2884 case WS_SO_RCVBUF:
2885 if (*(const int*)optval < 2048)
2886 {
2887 WARN("SO_RCVBF for %d bytes is too small: ignored\n", *(const int*)optval );
2888 return 0;
2889 }
2890 /* Fall through */
2891
2892 /* The options listed here don't need any special handling. Thanks to
2893 * the conversion happening above, options from there will fall through
2894 * to this, too.*/
2895 case WS_SO_ACCEPTCONN:
2896 case WS_SO_BROADCAST:
2897 case WS_SO_ERROR:
2898 case WS_SO_KEEPALIVE:
2899 case WS_SO_OOBINLINE:
2900 /* BSD socket SO_REUSEADDR is not 100% compatible to winsock semantics.
2901 * however, using it the BSD way fixes bug 8513 and seems to be what
2902 * most programmers assume, anyway */
2903 case WS_SO_REUSEADDR:
2904 case WS_SO_SNDBUF:
2905 case WS_SO_TYPE:
2906 convert_sockopt(&level, &optname);
2907 break;
2908
2909 /* SO_DEBUG is a privileged operation, ignore it. */
2910 case WS_SO_DEBUG:
2911 TRACE("Ignoring SO_DEBUG\n");
2912 return 0;
2913
2914 /* For some reason the game GrandPrixLegends does set SO_DONTROUTE on its
2915 * socket. According to MSDN, this option is silently ignored.*/
2916 case WS_SO_DONTROUTE:
2917 TRACE("Ignoring SO_DONTROUTE\n");
2918 return 0;
2919
2920 /* Stops two sockets from being bound to the same port. Always happens
2921 * on unix systems, so just drop it. */
2922 case WS_SO_EXCLUSIVEADDRUSE:
2923 TRACE("Ignoring SO_EXCLUSIVEADDRUSE, is always set.\n");
2924 return 0;
2925
2926 /* SO_OPENTYPE does not require a valid socket handle. */
2927 case WS_SO_OPENTYPE:
2928 if (!optlen || optlen < sizeof(int) || !optval)
2929 {
2930 SetLastError(WSAEFAULT);
2931 return SOCKET_ERROR;
2932 }
2933 get_per_thread_data()->opentype = *(const int *)optval;
2934 TRACE("setting global SO_OPENTYPE = 0x%x\n", *((int*)optval) );
2935 return 0;
2936
2937 #ifdef SO_RCVTIMEO
2938 case WS_SO_RCVTIMEO:
2939 #endif
2940 #ifdef SO_SNDTIMEO
2941 case WS_SO_SNDTIMEO:
2942 #endif
2943 #if defined(SO_RCVTIMEO) || defined(SO_SNDTIMEO)
2944 if (optval && optlen == sizeof(UINT32)) {
2945 /* WinSock passes milliseconds instead of struct timeval */
2946 tval.tv_usec = (*(const UINT32*)optval % 1000) * 1000;
2947 tval.tv_sec = *(const UINT32*)optval / 1000;
2948 /* min of 500 milliseconds */
2949 if (tval.tv_sec == 0 && tval.tv_usec < 500000)
2950 tval.tv_usec = 500000;
2951 optlen = sizeof(struct timeval);
2952 optval = (char*)&tval;
2953 } else if (optlen == sizeof(struct timeval)) {
2954 WARN("SO_SND/RCVTIMEO for %d bytes: assuming unixism\n", optlen);
2955 } else {
2956 WARN("SO_SND/RCVTIMEO for %d bytes is weird: ignored\n", optlen);
2957 return 0;
2958 }
2959 convert_sockopt(&level, &optname);
2960 break;
2961 #endif
2962
2963 default:
2964 TRACE("Unknown SOL_SOCKET optname: 0x%08x\n", optname);
2965 SetLastError(WSAENOPROTOOPT);
2966 return SOCKET_ERROR;
2967 }
2968 break; /* case WS_SOL_SOCKET */
2969
2970 #ifdef HAVE_IPX
2971 case NSPROTO_IPX:
2972 switch(optname)
2973 {
2974 case IPX_PTYPE:
2975 fd = get_sock_fd( s, 0, NULL );
2976 TRACE("trying to set IPX_PTYPE: %d (fd: %d)\n", *(const int*)optval, fd);
2977
2978 /* We try to set the ipx type on ipx socket level. */
2979 #ifdef SOL_IPX
2980 if(setsockopt(fd, SOL_IPX, IPX_TYPE, optval, optlen) == -1)
2981 {
2982 ERR("IPX: could not set ipx option type; expect weird behaviour\n");
2983 release_sock_fd( s, fd );
2984 return SOCKET_ERROR;
2985 }
2986 #else
2987 {
2988 struct ipx val;
2989 /* Should we retrieve val using a getsockopt call and then
2990 * set the modified one? */
2991 val.ipx_pt = *optval;
2992 setsockopt(fd, 0, SO_DEFAULT_HEADERS, &val, sizeof(struct ipx));
2993 }
2994 #endif
2995 release_sock_fd( s, fd );
2996 return 0;
2997
2998 case IPX_FILTERPTYPE:
2999 /* Sets the receive filter packet type, at the moment we don't support it */
3000 FIXME("IPX_FILTERPTYPE: %x\n", *optval);
3001 /* Returning 0 is better for now than returning a SOCKET_ERROR */
3002 return 0;
3003
3004 default:
3005 FIXME("opt_name:%x\n", optname);
3006 return SOCKET_ERROR;
3007 }
3008 break; /* case NSPROTO_IPX */
3009 #endif
3010
3011 /* Levels WS_IPPROTO_TCP and WS_IPPROTO_IP convert directly */
3012 case WS_IPPROTO_TCP:
3013 switch(optname)
3014 {
3015 case WS_TCP_NODELAY:
3016 convert_sockopt(&level, &optname);
3017 break;
3018 default:
3019 FIXME("Unknown IPPROTO_TCP optname 0x%08x\n", optname);
3020 return SOCKET_ERROR;
3021 }
3022 break;
3023
3024 case WS_IPPROTO_IP:
3025 switch(optname)
3026 {
3027 case WS_IP_ADD_MEMBERSHIP:
3028 case WS_IP_DROP_MEMBERSHIP:
3029 #ifdef IP_HDRINCL
3030 case WS_IP_HDRINCL:
3031 #endif
3032 case WS_IP_MULTICAST_IF:
3033 case WS_IP_MULTICAST_LOOP:
3034 case WS_IP_MULTICAST_TTL:
3035 case WS_IP_OPTIONS:
3036 case WS_IP_TOS:
3037 case WS_IP_TTL:
3038 convert_sockopt(&level, &optname);
3039 break;
3040 case WS_IP_DONTFRAGMENT:
3041 FIXME("IP_DONTFRAGMENT is silently ignored!\n");
3042 return 0;
3043 default:
3044 FIXME("Unknown IPPROTO_IP optname 0x%08x\n", optname);
3045 return SOCKET_ERROR;
3046 }
3047 break;
3048
3049 default:
3050 FIXME("Unknown level: 0x%08x\n", level);
3051 return SOCKET_ERROR;
3052 } /* end switch(level) */
3053
3054 /* avoid endianness issues if argument is a 16-bit int */
3055 if (optval && optlen < sizeof(int))
3056 {
3057 woptval= *((const INT16 *) optval);
3058 optval= (char*) &woptval;
3059 optlen=sizeof(int);
3060 }
3061 fd = get_sock_fd( s, 0, NULL );
3062 if (fd == -1) return SOCKET_ERROR;
3063
3064 if (setsockopt(fd, level, optname, optval, optlen) == 0)
3065 {
3066 release_sock_fd( s, fd );
3067 return 0;
3068 }
3069 TRACE("Setting socket error, %d\n", wsaErrno());
3070 SetLastError(wsaErrno());
3071 release_sock_fd( s, fd );
3072
3073 return SOCKET_ERROR;
3074 }
3075
3076 /***********************************************************************
3077 * shutdown (WS2_32.22)
3078 */
3079 int WINAPI WS_shutdown(SOCKET s, int how)
3080 {
3081 int fd, err = WSAENOTSOCK;
3082 unsigned int options, clear_flags = 0;
3083
3084 fd = get_sock_fd( s, 0, &options );
3085 TRACE("socket %04lx, how %i %x\n", s, how, options );
3086
3087 if (fd == -1)
3088 return SOCKET_ERROR;
3089
3090 switch( how )
3091 {
3092 case 0: /* drop receives */
3093 clear_flags |= FD_READ;
3094 break;
3095 case 1: /* drop sends */
3096 clear_flags |= FD_WRITE;
3097 break;
3098 case 2: /* drop all */
3099 clear_flags |= FD_READ|FD_WRITE;
3100 default:
3101 clear_flags |= FD_WINE_LISTENING;
3102 }
3103
3104 if (!(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)))
3105 {
3106 switch ( how )
3107 {
3108 case SD_RECEIVE:
3109 err = WS2_register_async_shutdown( s, ASYNC_TYPE_READ );
3110 break;
3111 case SD_SEND:
3112 err = WS2_register_async_shutdown( s, ASYNC_TYPE_WRITE );
3113 break;
3114 case SD_BOTH:
3115 default:
3116 err = WS2_register_async_shutdown( s, ASYNC_TYPE_READ );
3117 if (!err) err = WS2_register_async_shutdown( s, ASYNC_TYPE_WRITE );
3118 break;
3119 }
3120 if (err) goto error;
3121 }
3122 else /* non-overlapped mode */
3123 {
3124 if ( shutdown( fd, how ) )
3125 {
3126 err = wsaErrno();
3127 goto error;
3128 }
3129 }
3130
3131 release_sock_fd( s, fd );
3132 _enable_event( SOCKET2HANDLE(s), 0, 0, clear_flags );
3133 if ( how > 1) WSAAsyncSelect( s, 0, 0, 0 );
3134 return 0;
3135
3136 error:
3137 release_sock_fd( s, fd );
3138 _enable_event( SOCKET2HANDLE(s), 0, 0, clear_flags );
3139 WSASetLastError( err );
3140 return SOCKET_ERROR;
3141 }
3142
3143 /***********************************************************************
3144 * socket (WS2_32.23)
3145 */
3146 SOCKET WINAPI WS_socket(int af, int type, int protocol)
3147 {
3148 TRACE("af=%d type=%d protocol=%d\n", af, type, protocol);
3149
3150 return WSASocketA( af, type, protocol, NULL, 0,
3151 get_per_thread_data()->opentype ? 0 : WSA_FLAG_OVERLAPPED );
3152 }
3153
3154
3155 /***********************************************************************
3156 * gethostbyaddr (WS2_32.51)
3157 */
3158 struct WS_hostent* WINAPI WS_gethostbyaddr(const char *addr, int len, int type)
3159 {
3160 struct WS_hostent *retval = NULL;
3161 struct hostent* host;
3162
3163 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
3164 char *extrabuf;
3165 int ebufsize=1024;
3166 struct hostent hostentry;
3167 int locerr=ENOBUFS;
3168 host = NULL;
3169 extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ;
3170 while(extrabuf) {
3171 int res = gethostbyaddr_r(addr, len, type,
3172 &hostentry, extrabuf, ebufsize, &host, &locerr);
3173 if( res != ERANGE) break;
3174 ebufsize *=2;
3175 extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ;
3176 }
3177 if (!host) SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr));
3178 #else
3179 EnterCriticalSection( &csWSgetXXXbyYYY );
3180 host = gethostbyaddr(addr, len, type);
3181 if (!host) SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno));
3182 #endif
3183 if( host != NULL ) retval = WS_dup_he(host);
3184 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
3185 HeapFree(GetProcessHeap(),0,extrabuf);
3186 #else
3187 LeaveCriticalSection( &csWSgetXXXbyYYY );
3188 #endif
3189 TRACE("ptr %p, len %d, type %d ret %p\n", addr, len, type, retval);
3190 return retval;
3191 }
3192
3193 /***********************************************************************
3194 * gethostbyname (WS2_32.52)
3195 */
3196 struct WS_hostent* WINAPI WS_gethostbyname(const char* name)
3197 {
3198 struct WS_hostent *retval = NULL;
3199 struct hostent* host;
3200 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
3201 char *extrabuf;
3202 int ebufsize=1024;
3203 struct hostent hostentry;
3204 int locerr = ENOBUFS;
3205 #endif
3206 char buf[100];
3207 if( !name || !name[0]) {
3208 name = buf;
3209 if( gethostname( buf, 100) == -1) {
3210 SetLastError( WSAENOBUFS); /* appropriate ? */
3211 return retval;
3212 }
3213 }
3214 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
3215 host = NULL;
3216 extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ;
3217 while(extrabuf) {
3218 int res = gethostbyname_r(name, &hostentry, extrabuf, ebufsize, &host, &locerr);
3219 if( res != ERANGE) break;
3220 ebufsize *=2;
3221 extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ;
3222 }
3223 if (!host) SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr));
3224 #else
3225 EnterCriticalSection( &csWSgetXXXbyYYY );
3226 host = gethostbyname(name);
3227 if (!host) SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno));
3228 #endif
3229 if (host) retval = WS_dup_he(host);
3230 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
3231 HeapFree(GetProcessHeap(),0,extrabuf);
3232 #else
3233 LeaveCriticalSection( &csWSgetXXXbyYYY );
3234 #endif
3235 TRACE( "%s ret %p\n", debugstr_a(name), retval );
3236 return retval;
3237 }
3238
3239
3240 /***********************************************************************
3241 * getprotobyname (WS2_32.53)
3242 */
3243 struct WS_protoent* WINAPI WS_getprotobyname(const char* name)
3244 {
3245 struct WS_protoent* retval = NULL;
3246 #ifdef HAVE_GETPROTOBYNAME
3247 struct protoent* proto;
3248 EnterCriticalSection( &csWSgetXXXbyYYY );
3249 if( (proto = getprotobyname(name)) != NULL )
3250 {
3251 retval = WS_dup_pe(proto);
3252 }
3253 else {
3254 MESSAGE("protocol %s not found; You might want to add "
3255 "this to /etc/protocols\n", debugstr_a(name) );
3256 SetLastError(WSANO_DATA);
3257 }
3258 LeaveCriticalSection( &csWSgetXXXbyYYY );
3259 #endif
3260 TRACE( "%s ret %p\n", debugstr_a(name), retval );
3261 return retval;
3262 }
3263
3264
3265 /***********************************************************************
3266 * getprotobynumber (WS2_32.54)
3267 */
3268 struct WS_protoent* WINAPI WS_getprotobynumber(int number)
3269 {
3270 struct WS_protoent* retval = NULL;
3271 #ifdef HAVE_GETPROTOBYNUMBER
3272 struct protoent* proto;
3273 EnterCriticalSection( &csWSgetXXXbyYYY );
3274 if( (proto = getprotobynumber(number)) != NULL )
3275 {
3276 retval = WS_dup_pe(proto);
3277 }
3278 else {
3279 MESSAGE("protocol number %d not found; You might want to add "
3280 "this to /etc/protocols\n", number );
3281 SetLastError(WSANO_DATA);
3282 }
3283 LeaveCriticalSection( &csWSgetXXXbyYYY );
3284 #endif
3285 TRACE("%i ret %p\n", number, retval);
3286 return retval;
3287 }
3288
3289
3290 /***********************************************************************
3291 * getservbyname (WS2_32.55)
3292 */
3293 struct WS_servent* WINAPI WS_getservbyname(const char *name, const char *proto)
3294 {
3295 struct WS_servent* retval = NULL;
3296 struct servent* serv;
3297 char *name_str;
3298 char *proto_str = NULL;
3299
3300 if (!(name_str = strdup_lower(name))) return NULL;
3301
3302 if (proto && *proto)
3303 {
3304 if (!(proto_str = strdup_lower(proto)))
3305 {
3306 HeapFree( GetProcessHeap(), 0, name_str );
3307 return NULL;
3308 }
3309 }
3310
3311 EnterCriticalSection( &csWSgetXXXbyYYY );
3312 serv = getservbyname(name_str, proto_str);
3313 if( serv != NULL )
3314 {
3315 retval = WS_dup_se(serv);
3316 }
3317 else SetLastError(WSANO_DATA);
3318 LeaveCriticalSection( &csWSgetXXXbyYYY );
3319 HeapFree( GetProcessHeap(), 0, proto_str );
3320 HeapFree( GetProcessHeap(), 0, name_str );
3321 TRACE( "%s, %s ret %p\n", debugstr_a(name), debugstr_a(proto), retval );
3322 return retval;
3323 }
3324
3325 /***********************************************************************
3326 * freeaddrinfo (WS2_32.@)
3327 */
3328 void WINAPI WS_freeaddrinfo(struct WS_addrinfo *res)
3329 {
3330 while (res) {
3331 struct WS_addrinfo *next;
3332
3333 HeapFree(GetProcessHeap(),0,res->ai_canonname);
3334 HeapFree(GetProcessHeap(),0,res->ai_addr);
3335 next = res->ai_next;
3336 HeapFree(GetProcessHeap(),0,res);
3337 res = next;
3338 }
3339 }
3340
3341 /* helper functions for getaddrinfo()/getnameinfo() */
3342 static int convert_aiflag_w2u(int winflags) {
3343 int i, unixflags = 0;
3344
3345 for (i=0;i<sizeof(ws_aiflag_map)/sizeof(ws_aiflag_map[0]);i++)
3346 if (ws_aiflag_map[i][0] & winflags) {
3347 unixflags |= ws_aiflag_map[i][1];
3348 winflags &= ~ws_aiflag_map[i][0];
3349 }
3350 if (winflags)
3351 FIXME("Unhandled windows AI_xxx flags %x\n", winflags);
3352 return unixflags;
3353 }
3354
3355 static int convert_niflag_w2u(int winflags) {
3356 int i, unixflags = 0;
3357
3358 for (i=0;i<sizeof(ws_niflag_map)/sizeof(ws_niflag_map[0]);i++)
3359 if (ws_niflag_map[i][0] & winflags) {
3360 unixflags |= ws_niflag_map[i][1];
3361 winflags &= ~ws_niflag_map[i][0];
3362 }
3363 if (winflags)
3364 FIXME("Unhandled windows NI_xxx flags %x\n", winflags);
3365 return unixflags;
3366 }
3367
3368 static int convert_aiflag_u2w(int unixflags) {
3369 int i, winflags = 0;
3370
3371 for (i=0;i<sizeof(ws_aiflag_map)/sizeof(ws_aiflag_map[0]);i++)
3372 if (ws_aiflag_map[i][1] & unixflags) {
3373 winflags |= ws_aiflag_map[i][0];
3374 unixflags &= ~ws_aiflag_map[i][1];
3375 }
3376 if (unixflags) /* will warn usually */
3377 WARN("Unhandled UNIX AI_xxx flags %x\n", unixflags);
3378 return winflags;
3379 }
3380
3381 static int convert_eai_u2w(int unixret) {
3382 int i;
3383
3384 for (i=0;ws_eai_map[i][0];i++)
3385 if (ws_eai_map[i][1] == unixret)
3386 return ws_eai_map[i][0];
3387 return unixret;
3388 }
3389
3390 /***********************************************************************
3391 * getaddrinfo (WS2_32.@)
3392 */
3393 int WINAPI WS_getaddrinfo(LPCSTR nodename, LPCSTR servname, const struct WS_addrinfo *hints, struct WS_addrinfo **res)
3394 {
3395 #if HAVE_GETADDRINFO
3396 struct addrinfo *unixaires = NULL;
3397 int result;
3398 struct addrinfo unixhints, *punixhints = NULL;
3399 CHAR *node = NULL, *serv = NULL;
3400
3401 if (nodename)
3402 if (!(node = strdup_lower(nodename))) return WSA_NOT_ENOUGH_MEMORY;
3403
3404 if (servname) {
3405 if (!(serv = strdup_lower(servname))) {
3406 HeapFree(GetProcessHeap(), 0, node);
3407 return WSA_NOT_ENOUGH_MEMORY;
3408 }
3409 }
3410
3411 if (hints) {
3412 punixhints = &unixhints;
3413
3414 memset(&unixhints, 0, sizeof(unixhints));
3415 punixhints->ai_flags = convert_aiflag_w2u(hints->ai_flags);
3416 if (hints->ai_family == 0) /* wildcard, specific to getaddrinfo() */
3417 punixhints->ai_family = 0;
3418 else
3419 punixhints->ai_family = convert_af_w2u(hints->ai_family);
3420 if (hints->ai_socktype == 0) /* wildcard, specific to getaddrinfo() */
3421 punixhints->ai_socktype = 0;
3422 else
3423 punixhints->ai_socktype = convert_socktype_w2u(hints->ai_socktype);
3424 if (hints->ai_protocol == 0) /* wildcard, specific to getaddrinfo() */
3425 punixhints->ai_protocol = 0;
3426 else
3427 punixhints->ai_protocol = convert_proto_w2u(hints->ai_protocol);
3428 }
3429
3430 /* getaddrinfo(3) is thread safe, no need to wrap in CS */
3431 result = getaddrinfo(nodename, servname, punixhints, &unixaires);
3432
3433 TRACE("%s, %s %p -> %p %d\n", nodename, servname, hints, res, result);
3434
3435 HeapFree(GetProcessHeap(), 0, node);
3436 HeapFree(GetProcessHeap(), 0, serv);
3437
3438 if (!result) {
3439 struct addrinfo *xuai = unixaires;
3440 struct WS_addrinfo **xai = res;
3441
3442 *xai = NULL;
3443 while (xuai) {
3444 struct WS_addrinfo *ai = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY, sizeof(struct WS_addrinfo));
3445 int len;
3446
3447 if (!ai)
3448 goto outofmem;
3449
3450 *xai = ai;xai = &ai->ai_next;
3451 ai->ai_flags = convert_aiflag_u2w(xuai->ai_flags);
3452 ai->ai_family = convert_af_u2w(xuai->ai_family);
3453 ai->ai_socktype = convert_socktype_u2w(xuai->ai_socktype);
3454 ai->ai_protocol = convert_proto_u2w(xuai->ai_protocol);
3455 if (xuai->ai_canonname) {
3456 TRACE("canon name - %s\n",debugstr_a(xuai->ai_canonname));
3457 ai->ai_canonname = HeapAlloc(GetProcessHeap(),0,strlen(xuai->ai_canonname)+1);
3458 if (!ai->ai_canonname)
3459 goto outofmem;
3460 strcpy(ai->ai_canonname,xuai->ai_canonname);
3461 }
3462 len = xuai->ai_addrlen;
3463 ai->ai_addr = HeapAlloc(GetProcessHeap(),0,len);
3464 if (!ai->ai_addr)
3465 goto outofmem;
3466 ai->ai_addrlen = len;
3467 do {
3468 int winlen = ai->ai_addrlen;
3469
3470 if (!ws_sockaddr_u2ws(xuai->ai_addr, ai->ai_addr, &winlen)) {
3471 ai->ai_addrlen = winlen;
3472 break;
3473 }
3474 len = 2*len;
3475 ai->ai_addr = HeapReAlloc(GetProcessHeap(),0,ai->ai_addr,len);
3476 if (!ai->ai_addr)
3477 goto outofmem;
3478 ai->ai_addrlen = len;
3479 } while (1);
3480 xuai = xuai->ai_next;
3481 }
3482 freeaddrinfo(unixaires);
3483 } else {
3484 result = convert_eai_u2w(result);
3485 *res = NULL;
3486 }
3487 return result;
3488
3489 outofmem:
3490 if (*res) WS_freeaddrinfo(*res);
3491 if (unixaires) freeaddrinfo(unixaires);
3492 *res = NULL;
3493 return WSA_NOT_ENOUGH_MEMORY;
3494 #else
3495 FIXME("getaddrinfo() failed, not found during buildtime.\n");
3496 return EAI_FAIL;
3497 #endif
3498 }
3499
3500 /***********************************************************************
3501 * GetAddrInfoW (WS2_32.@)
3502 */
3503 int WINAPI GetAddrInfoW(LPCWSTR nodename, LPCWSTR servname, const ADDRINFOW *hints, PADDRINFOW *res)
3504 {
3505 FIXME("empty stub!\n");
3506 return EAI_FAIL;
3507 }
3508
3509 int WINAPI WS_getnameinfo(const SOCKADDR *sa, WS_socklen_t salen, PCHAR host,
3510 DWORD hostlen, PCHAR serv, DWORD servlen, INT flags)
3511 {
3512 #if HAVE_GETNAMEINFO
3513 int ret;
3514 union generic_unix_sockaddr sa_u;
3515 unsigned int size;
3516
3517 TRACE("%s %d %p %d %p %d %d\n", debugstr_sockaddr(sa), salen, host, hostlen,
3518 serv, servlen, flags);
3519
3520 size = ws_sockaddr_ws2u(sa, salen, &sa_u);
3521 if (!size)
3522 {
3523 WSASetLastError(WSAEFAULT);
3524 return WSA_NOT_ENOUGH_MEMORY;
3525 }
3526 ret = getnameinfo(&sa_u.addr, size, host, hostlen, serv, servlen, convert_niflag_w2u(flags));
3527 return convert_eai_u2w(ret);
3528 #else
3529 FIXME("getnameinfo() failed, not found during buildtime.\n");
3530 return EAI_FAIL;
3531 #endif
3532 }
3533
3534 /***********************************************************************
3535 * getservbyport (WS2_32.56)
3536 */
3537 struct WS_servent* WINAPI WS_getservbyport(int port, const char *proto)
3538 {
3539 struct WS_servent* retval = NULL;
3540 #ifdef HAVE_GETSERVBYPORT
3541 struct servent* serv;
3542 char *proto_str = NULL;
3543
3544 if (proto && *proto)
3545 {
3546 if (!(proto_str = strdup_lower(proto))) return NULL;
3547 }
3548 EnterCriticalSection( &csWSgetXXXbyYYY );
3549 if( (serv = getservbyport(port, proto_str)) != NULL ) {
3550 retval = WS_dup_se(serv);
3551 }
3552 else SetLastError(WSANO_DATA);
3553 LeaveCriticalSection( &csWSgetXXXbyYYY );
3554 HeapFree( GetProcessHeap(), 0, proto_str );
3555 #endif
3556 TRACE("%d (i.e. port %d), %s ret %p\n", port, (int)ntohl(port), debugstr_a(proto), retval);
3557 return retval;
3558 }
3559
3560
3561 /***********************************************************************
3562 * gethostname (WS2_32.57)
3563 */
3564 int WINAPI WS_gethostname(char *name, int namelen)
3565 {
3566 TRACE("name %p, len %d\n", name, namelen);
3567
3568 if (gethostname(name, namelen) == 0)
3569 {
3570 TRACE("<- '%s'\n", name);
3571 return 0;
3572 }
3573 SetLastError((errno == EINVAL) ? WSAEFAULT : wsaErrno());
3574 TRACE("<- ERROR !\n");
3575 return SOCKET_ERROR;
3576 }
3577
3578
3579 /* ------------------------------------- Windows sockets extensions -- *
3580 * *
3581 * ------------------------------------------------------------------- */
3582
3583 /***********************************************************************
3584 * WSAEnumNetworkEvents (WS2_32.36)
3585 */
3586 int WINAPI WSAEnumNetworkEvents(SOCKET s, WSAEVENT hEvent, LPWSANETWORKEVENTS lpEvent)
3587 {
3588 int ret;
3589
3590 TRACE("%08lx, hEvent %p, lpEvent %p\n", s, hEvent, lpEvent );
3591
3592 SERVER_START_REQ( get_socket_event )
3593 {
3594 req->handle = SOCKET2HANDLE(s);
3595 req->service = TRUE;
3596 req->c_event = hEvent;
3597 wine_server_set_reply( req, lpEvent->iErrorCode, sizeof(lpEvent->iErrorCode) );
3598 if (!(ret = wine_server_call(req))) lpEvent->lNetworkEvents = reply->pmask & reply->mask;
3599 }
3600 SERVER_END_REQ;
3601 if (!ret) return 0;
3602 SetLastError(WSAEINVAL);
3603 return SOCKET_ERROR;
3604 }
3605
3606 /***********************************************************************
3607 * WSAEventSelect (WS2_32.39)
3608 */
3609 int WINAPI WSAEventSelect(SOCKET s, WSAEVENT hEvent, LONG lEvent)
3610 {
3611 int ret;
3612
3613 TRACE("%08lx, hEvent %p, event %08x\n", s, hEvent, lEvent);
3614
3615 SERVER_START_REQ( set_socket_event )
3616 {
3617 req->handle = SOCKET2HANDLE(s);
3618 req->mask = lEvent;
3619 req->event = hEvent;
3620 req->window = 0;
3621 req->msg = 0;
3622 ret = wine_server_call( req );
3623 }
3624 SERVER_END_REQ;
3625 if (!ret) return 0;
3626 SetLastError(WSAEINVAL);
3627 return SOCKET_ERROR;
3628 }
3629
3630 /**********************************************************************
3631 * WSAGetOverlappedResult (WS2_32.40)
3632 */
3633 BOOL WINAPI WSAGetOverlappedResult( SOCKET s, LPWSAOVERLAPPED lpOverlapped,
3634 LPDWORD lpcbTransfer, BOOL fWait,
3635 LPDWORD lpdwFlags )
3636 {
3637 NTSTATUS status;
3638
3639 TRACE( "socket %04lx ovl %p trans %p, wait %d flags %p\n",
3640 s, lpOverlapped, lpcbTransfer, fWait, lpdwFlags );
3641
3642 if ( lpOverlapped == NULL )
3643 {
3644 ERR( "Invalid pointer\n" );
3645 WSASetLastError(WSA_INVALID_PARAMETER);
3646 return FALSE;
3647 }
3648
3649 status = lpOverlapped->Internal;
3650 if (status == STATUS_PENDING)
3651 {
3652 if (!fWait)
3653 {
3654 SetLastError( WSA_IO_INCOMPLETE );
3655 return FALSE;
3656 }
3657
3658 if (WaitForSingleObject( lpOverlapped->hEvent ? lpOverlapped->hEvent : SOCKET2HANDLE(s),
3659 INFINITE ) == WAIT_FAILED)
3660 return FALSE;
3661 status = lpOverlapped->Internal;
3662 }
3663
3664 if ( lpcbTransfer )
3665 *lpcbTransfer = lpOverlapped->InternalHigh;
3666
3667 if ( lpdwFlags )
3668 *lpdwFlags = lpOverlapped->u.s.Offset;
3669
3670 if (status) SetLastError( RtlNtStatusToDosError(status) );
3671 return !status;
3672 }
3673
3674
3675 /***********************************************************************
3676 * WSAAsyncSelect (WS2_32.101)
3677 */
3678 INT WINAPI WSAAsyncSelect(SOCKET s, HWND hWnd, UINT uMsg, LONG lEvent)
3679 {
3680 int ret;
3681
3682 TRACE("%lx, hWnd %p, uMsg %08x, event %08x\n", s, hWnd, uMsg, lEvent);
3683
3684 SERVER_START_REQ( set_socket_event )
3685 {
3686 req->handle = SOCKET2HANDLE(s);
3687 req->mask = lEvent;
3688 req->event = 0;
3689 req->window = hWnd;
3690 req->msg = uMsg;
3691 ret = wine_server_call( req );
3692 }
3693 SERVER_END_REQ;
3694 if (!ret) return 0;
3695 SetLastError(WSAEINVAL);
3696 return SOCKET_ERROR;
3697 }
3698
3699 /***********************************************************************
3700 * WSACreateEvent (WS2_32.31)
3701 *
3702 */
3703 WSAEVENT WINAPI WSACreateEvent(void)
3704 {
3705 /* Create a manual-reset event, with initial state: unsignaled */
3706 TRACE("\n");
3707
3708 return CreateEventW(NULL, TRUE, FALSE, NULL);
3709 }
3710
3711 /***********************************************************************
3712 * WSACloseEvent (WS2_32.29)
3713 *
3714 */
3715 BOOL WINAPI WSACloseEvent(WSAEVENT event)
3716 {
3717 TRACE ("event=%p\n", event);
3718
3719 return CloseHandle(event);
3720 }
3721
3722 /***********************************************************************
3723 * WSASocketA (WS2_32.78)
3724 *
3725 */
3726 SOCKET WINAPI WSASocketA(int af, int type, int protocol,
3727 LPWSAPROTOCOL_INFOA lpProtocolInfo,
3728 GROUP g, DWORD dwFlags)
3729 {
3730 INT len;
3731 WSAPROTOCOL_INFOW info;
3732
3733 TRACE("af=%d type=%d protocol=%d protocol_info=%p group=%d flags=0x%x\n",
3734 af, type, protocol, lpProtocolInfo, g, dwFlags);
3735
3736 if (!lpProtocolInfo) return WSASocketW(af, type, protocol, NULL, g, dwFlags);
3737
3738 memcpy(&info, lpProtocolInfo, FIELD_OFFSET(WSAPROTOCOL_INFOW, szProtocol));
3739 len = MultiByteToWideChar(CP_ACP, 0, lpProtocolInfo->szProtocol, -1,
3740 info.szProtocol, WSAPROTOCOL_LEN + 1);
3741
3742 if (!len)
3743 {
3744 WSASetLastError( WSAEINVAL);
3745 return SOCKET_ERROR;
3746 }
3747
3748 return WSASocketW(af, type, protocol, &info, g, dwFlags);
3749 }
3750
3751 /***********************************************************************
3752 * WSASocketW (WS2_32.79)
3753 *
3754 */
3755 SOCKET WINAPI WSASocketW(int af, int type, int protocol,
3756 LPWSAPROTOCOL_INFOW lpProtocolInfo,
3757 GROUP g, DWORD dwFlags)
3758 {
3759 SOCKET ret;
3760
3761 /*
3762 FIXME: The "advanced" parameters of WSASocketW (lpProtocolInfo,
3763 g, dwFlags except WSA_FLAG_OVERLAPPED) are ignored.
3764 */
3765
3766 TRACE("af=%d type=%d protocol=%d protocol_info=%p group=%d flags=0x%x\n",
3767 af, type, protocol, lpProtocolInfo, g, dwFlags );
3768
3769 /* hack for WSADuplicateSocket */
3770 if (lpProtocolInfo && lpProtocolInfo->dwServiceFlags4 == 0xff00ff00) {
3771 ret = lpProtocolInfo->dwCatalogEntryId;
3772 TRACE("\tgot duplicate %04lx\n", ret);
3773 return ret;
3774 }
3775
3776 /* check and convert the socket family */
3777 af = convert_af_w2u(af);
3778 if (af == -1)
3779 {
3780 FIXME("Unsupported socket family %d!\n", af);
3781 SetLastError(WSAEAFNOSUPPORT);
3782 return INVALID_SOCKET;
3783 }
3784
3785 /* check the socket type */
3786 type = convert_socktype_w2u(type);
3787 if (type == -1)
3788 {
3789 SetLastError(WSAESOCKTNOSUPPORT);
3790 return INVALID_SOCKET;
3791 }
3792
3793 /* check the protocol type */
3794 if ( protocol < 0 ) /* don't support negative values */
3795 {
3796 SetLastError(WSAEPROTONOSUPPORT);
3797 return INVALID_SOCKET;
3798 }
3799
3800 if ( af == AF_UNSPEC) /* did they not specify the address family? */
3801 switch(protocol)
3802 {
3803 case IPPROTO_TCP:
3804 if (type == SOCK_STREAM) { af = AF_INET; break; }
3805 case IPPROTO_UDP:
3806 if (type == SOCK_DGRAM) { af = AF_INET; break; }
3807 default: SetLastError(WSAEPROTOTYPE); return INVALID_SOCKET;
3808 }
3809
3810 SERVER_START_REQ( create_socket )
3811 {
3812 req->family = af;
3813 req->type = type;
3814 req->protocol = protocol;
3815 req->access = GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE;
3816 req->attributes = OBJ_INHERIT;
3817 req->flags = dwFlags;
3818 set_error( wine_server_call( req ) );
3819 ret = HANDLE2SOCKET( reply->handle );
3820 }
3821 SERVER_END_REQ;
3822 if (ret)
3823 {
3824 TRACE("\tcreated %04lx\n", ret );
3825 return ret;
3826 }
3827
3828 if (GetLastError() == WSAEACCES) /* raw socket denied */
3829 {
3830 if (type == SOCK_RAW)
3831 MESSAGE("WARNING: Trying to create a socket of type SOCK_RAW, this"
3832 " will fail unless you have special permissions.\n");
3833 else
3834 MESSAGE("WS_SOCKET: Failed to create socket, this requires"
3835 " special permissions.\n");
3836 SetLastError(WSAESOCKTNOSUPPORT);
3837 }
3838
3839 WARN("\t\tfailed!\n");
3840 return INVALID_SOCKET;
3841 }
3842
3843 /***********************************************************************
3844 * WSAJoinLeaf (WS2_32.58)
3845 *
3846 */
3847 SOCKET WINAPI WSAJoinLeaf(
3848 SOCKET s,
3849 const struct WS_sockaddr *addr,
3850 int addrlen,
3851 LPWSABUF lpCallerData,
3852 LPWSABUF lpCalleeData,
3853 LPQOS lpSQOS,
3854 LPQOS lpGQOS,
3855 DWORD dwFlags)
3856 {
3857 FIXME("stub.\n");
3858 return INVALID_SOCKET;
3859 }
3860
3861 /***********************************************************************
3862 * __WSAFDIsSet (WS2_32.151)
3863 */
3864 int WINAPI __WSAFDIsSet(SOCKET s, WS_fd_set *set)
3865 {
3866 int i = set->fd_count;
3867
3868 TRACE("(%ld,%p(%i))\n", s, set, i);
3869
3870 while (i--)
3871 if (set->fd_array[i] == s) return 1;
3872 return 0;
3873 }
3874
3875 /***********************************************************************
3876 * WSAIsBlocking (WINSOCK.114)
3877 * WSAIsBlocking (WS2_32.114)
3878 */
3879 BOOL WINAPI WSAIsBlocking(void)
3880 {
3881 /* By default WinSock should set all its sockets to non-blocking mode
3882 * and poll in PeekMessage loop when processing "blocking" ones. This
3883 * function is supposed to tell if the program is in this loop. Our
3884 * blocking calls are truly blocking so we always return FALSE.
3885 *
3886 * Note: It is allowed to call this function without prior WSAStartup().
3887 */
3888
3889 TRACE("\n");
3890 return FALSE;
3891 }
3892
3893 /***********************************************************************
3894 * WSACancelBlockingCall (WINSOCK.113)
3895 * WSACancelBlockingCall (WS2_32.113)
3896 */
3897 INT WINAPI WSACancelBlockingCall(void)
3898 {
3899 TRACE("\n");
3900 return 0;
3901 }
3902
3903 static INT WINAPI WSA_DefaultBlockingHook( FARPROC x )
3904 {
3905 FIXME("How was this called?\n");
3906 return x();
3907 }
3908
3909
3910 /***********************************************************************
3911 * WSASetBlockingHook (WS2_32.109)
3912 */
3913 FARPROC WINAPI WSASetBlockingHook(FARPROC lpBlockFunc)
3914 {
3915 FARPROC prev = blocking_hook;
3916 blocking_hook = lpBlockFunc;
3917 TRACE("hook %p\n", lpBlockFunc);
3918 return prev;
3919 }
3920
3921
3922 /***********************************************************************
3923 * WSAUnhookBlockingHook (WS2_32.110)
3924 */
3925 INT WINAPI WSAUnhookBlockingHook(void)
3926 {
3927 blocking_hook = (FARPROC)WSA_DefaultBlockingHook;
3928 return 0;
3929 }
3930
3931
3932 /* ----------------------------------- end of API stuff */
3933
3934 /* ----------------------------------- helper functions -
3935 *
3936 * TODO: Merge WS_dup_..() stuff into one function that
3937 * would operate with a generic structure containing internal
3938 * pointers (via a template of some kind).
3939 */
3940
3941 static int list_size(char** l, int item_size)
3942 {
3943 int i,j = 0;
3944 if(l)
3945 { for(i=0;l[i];i++)
3946 j += (item_size) ? item_size : strlen(l[i]) + 1;
3947 j += (i + 1) * sizeof(char*); }
3948 return j;
3949 }
3950
3951 static int list_dup(char** l_src, char** l_to, int item_size)
3952 {
3953 char *p;
3954 int i;
3955
3956 for (i = 0; l_src[i]; i++) ;
3957 p = (char *)(l_to + i + 1);
3958 for (i = 0; l_src[i]; i++)
3959 {
3960 int count = ( item_size ) ? item_size : strlen(l_src[i]) + 1;
3961 memcpy(p, l_src[i], count);
3962 l_to[i] = p;
3963 p += count;
3964 }
3965 l_to[i] = NULL;
3966 return p - (char *)l_to;
3967 }
3968
3969 /* ----- hostent */
3970
3971 /* duplicate hostent entry
3972 * and handle all Win16/Win32 dependent things (struct size, ...) *correctly*.
3973 * Ditto for protoent and servent.
3974 */
3975 static struct WS_hostent *WS_dup_he(const struct hostent* p_he)
3976 {
3977 char *p;
3978 struct WS_hostent *p_to;
3979
3980 int size = (sizeof(*p_he) +
3981 strlen(p_he->h_name) + 1 +
3982 list_size(p_he->h_aliases, 0) +
3983 list_size(p_he->h_addr_list, p_he->h_length));
3984
3985 if (!(p_to = check_buffer_he(size))) return NULL;
3986 p_to->h_addrtype = p_he->h_addrtype;
3987 p_to->h_length = p_he->h_length;
3988
3989 p = (char *)(p_to + 1);
3990 p_to->h_name = p;
3991 strcpy(p, p_he->h_name);
3992 p += strlen(p) + 1;
3993
3994 p_to->h_aliases = (char **)p;
3995 p += list_dup(p_he->h_aliases, p_to->h_aliases, 0);
3996
3997 p_to->h_addr_list = (char **)p;
3998 list_dup(p_he->h_addr_list, p_to->h_addr_list, p_he->h_length);
3999 return p_to;
4000 }
4001
4002 /* ----- protoent */
4003
4004 static struct WS_protoent *WS_dup_pe(const struct protoent* p_pe)
4005 {
4006 char *p;
4007 struct WS_protoent *p_to;
4008
4009 int size = (sizeof(*p_pe) +
4010 strlen(p_pe->p_name) + 1 +
4011 list_size(p_pe->p_aliases, 0));
4012
4013 if (!(p_to = check_buffer_pe(size))) return NULL;
4014 p_to->p_proto = p_pe->p_proto;
4015
4016 p = (char *)(p_to + 1);
4017 p_to->p_name = p;
4018 strcpy(p, p_pe->p_name);
4019 p += strlen(p) + 1;
4020
4021 p_to->p_aliases = (char **)p;
4022 list_dup(p_pe->p_aliases, p_to->p_aliases, 0);
4023 return p_to;
4024 }
4025
4026 /* ----- servent */
4027
4028 static struct WS_servent *WS_dup_se(const struct servent* p_se)
4029 {
4030 char *p;
4031 struct WS_servent *p_to;
4032
4033 int size = (sizeof(*p_se) +
4034 strlen(p_se->s_proto) + 1 +
4035 strlen(p_se->s_name) + 1 +
4036 list_size(p_se->s_aliases, 0));
4037
4038 if (!(p_to = check_buffer_se(size))) return NULL;
4039 p_to->s_port = p_se->s_port;
4040
4041 p = (char *)(p_to + 1);
4042 p_to->s_name = p;
4043 strcpy(p, p_se->s_name);
4044 p += strlen(p) + 1;
4045
4046 p_to->s_proto = p;
4047 strcpy(p, p_se->s_proto);
4048 p += strlen(p) + 1;
4049
4050 p_to->s_aliases = (char **)p;
4051 list_dup(p_se->s_aliases, p_to->s_aliases, 0);
4052 return p_to;
4053 }
4054
4055 /* ----------------------------------- error handling */
4056
4057 UINT wsaErrno(void)
4058 {
4059 int loc_errno = errno;
4060 WARN("errno %d, (%s).\n", loc_errno, strerror(loc_errno));
4061
4062 switch(loc_errno)
4063 {
4064 case EINTR: return WSAEINTR;
4065 case EBADF: return WSAEBADF;
4066 case EPERM:
4067 case EACCES: return WSAEACCES;
4068 case EFAULT: return WSAEFAULT;
4069 case EINVAL: return WSAEINVAL;
4070 case EMFILE: return WSAEMFILE;
4071 case EWOULDBLOCK: return WSAEWOULDBLOCK;
4072 case EINPROGRESS: return WSAEINPROGRESS;
4073 case EALREADY: return WSAEALREADY;
4074 case ENOTSOCK: return WSAENOTSOCK;
4075 case EDESTADDRREQ: return WSAEDESTADDRREQ;
4076 case EMSGSIZE: return WSAEMSGSIZE;
4077 case EPROTOTYPE: return WSAEPROTOTYPE;
4078 case ENOPROTOOPT: return WSAENOPROTOOPT;
4079 case EPROTONOSUPPORT: return WSAEPROTONOSUPPORT;
4080 case ESOCKTNOSUPPORT: return WSAESOCKTNOSUPPORT;
4081 case EOPNOTSUPP: return WSAEOPNOTSUPP;
4082 case EPFNOSUPPORT: return WSAEPFNOSUPPORT;
4083 case EAFNOSUPPORT: return WSAEAFNOSUPPORT;
4084 case EADDRINUSE: return WSAEADDRINUSE;
4085 case EADDRNOTAVAIL: return WSAEADDRNOTAVAIL;
4086 case ENETDOWN: return WSAENETDOWN;
4087 case ENETUNREACH: return WSAENETUNREACH;
4088 case ENETRESET: return WSAENETRESET;
4089 case ECONNABORTED: return WSAECONNABORTED;
4090 case EPIPE:
4091 case ECONNRESET: return WSAECONNRESET;
4092 case ENOBUFS: return WSAENOBUFS;
4093 case EISCONN: return WSAEISCONN;
4094 case ENOTCONN: return WSAENOTCONN;
4095 case ESHUTDOWN: return WSAESHUTDOWN;
4096 case ETOOMANYREFS: return WSAETOOMANYREFS;
4097 case ETIMEDOUT: return WSAETIMEDOUT;
4098 case ECONNREFUSED: return WSAECONNREFUSED;
4099 case ELOOP: return WSAELOOP;
4100 case ENAMETOOLONG: return WSAENAMETOOLONG;
4101 case EHOSTDOWN: return WSAEHOSTDOWN;
4102 case EHOSTUNREACH: return WSAEHOSTUNREACH;
4103 case ENOTEMPTY: return WSAENOTEMPTY;
4104 #ifdef EPROCLIM
4105 case EPROCLIM: return WSAEPROCLIM;
4106 #endif
4107 #ifdef EUSERS
4108 case EUSERS: return WSAEUSERS;
4109 #endif
4110 #ifdef EDQUOT
4111 case EDQUOT: return WSAEDQUOT;
4112 #endif
4113 #ifdef ESTALE
4114 case ESTALE: return WSAESTALE;
4115 #endif
4116 #ifdef EREMOTE
4117 case EREMOTE: return WSAEREMOTE;
4118 #endif
4119
4120 /* just in case we ever get here and there are no problems */
4121 case 0: return 0;
4122 default:
4123 WARN("Unknown errno %d!\n", loc_errno);
4124 return WSAEOPNOTSUPP;
4125 }
4126 }
4127
4128 UINT wsaHerrno(int loc_errno)
4129 {
4130
4131 WARN("h_errno %d.\n", loc_errno);
4132
4133 switch(loc_errno)
4134 {
4135 case HOST_NOT_FOUND: return WSAHOST_NOT_FOUND;
4136 case TRY_AGAIN: return WSATRY_AGAIN;
4137 case NO_RECOVERY: return WSANO_RECOVERY;
4138 case NO_DATA: return WSANO_DATA;
4139 case ENOBUFS: return WSAENOBUFS;
4140
4141 case 0: return 0;
4142 default:
4143 WARN("Unknown h_errno %d!\n", loc_errno);
4144 return WSAEOPNOTSUPP;
4145 }
4146 }
4147
4148
4149 /***********************************************************************
4150 * WSARecv (WS2_32.67)
4151 */
4152 int WINAPI WSARecv(SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
4153 LPDWORD NumberOfBytesReceived, LPDWORD lpFlags,
4154 LPWSAOVERLAPPED lpOverlapped,
4155 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine)
4156 {
4157 return WSARecvFrom(s, lpBuffers, dwBufferCount, NumberOfBytesReceived, lpFlags,
4158 NULL, NULL, lpOverlapped, lpCompletionRoutine);
4159 }
4160
4161 /***********************************************************************
4162 * WSARecvFrom (WS2_32.69)
4163 */
4164 INT WINAPI WSARecvFrom( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount,
4165 LPDWORD lpNumberOfBytesRecvd, LPDWORD lpFlags, struct WS_sockaddr *lpFrom,
4166 LPINT lpFromlen, LPWSAOVERLAPPED lpOverlapped,
4167 LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine )
4168
4169 {
4170 unsigned int i, options;
4171 int n, fd, err;
4172 DWORD timeout_start = GetTickCount();
4173 struct iovec iovec[WS_MSG_MAXIOVLEN];
4174 ULONG_PTR cvalue = (lpOverlapped && ((ULONG_PTR)lpOverlapped->hEvent & 1) == 0) ? (ULONG_PTR)lpOverlapped : 0;
4175
4176 TRACE("socket %04lx, wsabuf %p, nbufs %d, flags %d, from %p, fromlen %d, ovl %p, func %p\n",
4177 s, lpBuffers, dwBufferCount, *lpFlags, lpFrom,
4178 (lpFromlen ? *lpFromlen : -1),
4179 lpOverlapped, lpCompletionRoutine);
4180
4181 if (dwBufferCount > WS_MSG_MAXIOVLEN)
4182 {
4183 WSASetLastError( WSAEINVAL );
4184 return SOCKET_ERROR;
4185 }
4186
4187 fd = get_sock_fd( s, FILE_READ_DATA, &options );
4188 TRACE( "fd=%d, options=%x\n", fd, options );
4189
4190 if (fd == -1) return SOCKET_ERROR;
4191
4192 for (i = 0; i < dwBufferCount; i++)
4193 {
4194 iovec[i].iov_base = lpBuffers[i].buf;
4195 iovec[i].iov_len = lpBuffers[i].len;
4196 }
4197
4198 for (;;)
4199 {
4200 n = WS2_recv( fd, iovec, dwBufferCount, lpFrom, lpFromlen, lpFlags );
4201 if (n == -1)
4202 {
4203 if (errno == EINTR) continue;
4204 if (errno != EAGAIN)
4205 {
4206 err = wsaErrno();
4207 if (cvalue) WS_AddCompletion( s, cvalue, err, 0 );
4208 goto error;
4209 }
4210 }
4211 else
4212 *lpNumberOfBytesRecvd = n;
4213
4214 if ((lpOverlapped || lpCompletionRoutine) &&
4215 !(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)))
4216 {
4217 IO_STATUS_BLOCK *iosb;
4218 struct ws2_async *wsa = HeapAlloc( GetProcessHeap(), 0, sizeof(*wsa) );
4219
4220 if ( !wsa )
4221 {
4222 err = WSAEFAULT;
4223 goto error;
4224 }
4225 release_sock_fd( s, fd );
4226
4227 wsa->hSocket = SOCKET2HANDLE(s);
4228 wsa->flags = *lpFlags;
4229 wsa->addr = lpFrom;
4230 wsa->addrlen.ptr = lpFromlen;
4231 wsa->user_overlapped = lpOverlapped;
4232 wsa->completion_func = lpCompletionRoutine;
4233 wsa->n_iovecs = dwBufferCount;
4234 memcpy( wsa->iovec, iovec, dwBufferCount * sizeof(*iovec) );
4235
4236 iosb = lpOverlapped ? (IO_STATUS_BLOCK *)lpOverlapped : &wsa->local_iosb;
4237
4238 if (n == -1)
4239 {
4240 iosb->u.Status = STATUS_PENDING;
4241 iosb->Information = 0;
4242
4243 SERVER_START_REQ( register_async )
4244 {
4245 req->handle = wsa->hSocket;
4246 req->type = ASYNC_TYPE_READ;
4247 req->async.callback = WS2_async_recv;
4248 req->async.iosb = iosb;
4249 req->async.arg = wsa;
4250 req->async.apc = ws2_async_apc;
4251 req->async.event = lpCompletionRoutine ? 0 : lpOverlapped->hEvent;
4252 req->async.cvalue = cvalue;
4253 err = wine_server_call( req );
4254 }
4255 SERVER_END_REQ;
4256
4257 if (err != STATUS_PENDING) HeapFree( GetProcessHeap(), 0, wsa );
4258 WSASetLastError( NtStatusToWSAError( err ));
4259 return SOCKET_ERROR;
4260 }
4261
4262 iosb->u.Status = STATUS_SUCCESS;
4263 iosb->Information = n;
4264 if (!wsa->completion_func)
4265 {
4266 if (cvalue) WS_AddCompletion( s, cvalue, STATUS_SUCCESS, n );
4267 SetEvent( lpOverlapped->hEvent );
4268 HeapFree( GetProcessHeap(), 0, wsa );
4269 }
4270 else NtQueueApcThread( GetCurrentThread(), (PNTAPCFUNC)ws2_async_apc,
4271 (ULONG_PTR)wsa, (ULONG_PTR)iosb, 0 );
4272 _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
4273 return 0;
4274 }
4275
4276 if (n != -1) break;
4277
4278 if ( _is_blocking(s) )
4279 {
4280 struct pollfd pfd;
4281 int timeout = GET_RCVTIMEO(fd);
4282 if (timeout != -1)
4283 {
4284 timeout -= GetTickCount() - timeout_start;
4285 if (timeout < 0) timeout = 0;
4286 }
4287
4288 pfd.fd = fd;
4289 pfd.events = POLLIN;
4290 if (*lpFlags & WS_MSG_OOB) pfd.events |= POLLPRI;
4291
4292 if (!timeout || !poll( &pfd, 1, timeout ))
4293 {
4294 err = WSAETIMEDOUT;
4295 /* a timeout is not fatal */
4296 _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
4297 goto error;
4298 }
4299 }
4300 else
4301 {
4302 _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
4303 err = WSAEWOULDBLOCK;
4304 goto error;
4305 }
4306 }
4307
4308 TRACE(" -> %i bytes\n", n);
4309 release_sock_fd( s, fd );
4310 _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0);
4311
4312 return 0;
4313
4314 error:
4315 release_sock_fd( s, fd );
4316 WARN(" -> ERROR %d\n", err);
4317 WSASetLastError( err );
4318 return SOCKET_ERROR;
4319 }
4320
4321 /***********************************************************************
4322 * WSCInstallProvider (WS2_32.88)
4323 */
4324 INT WINAPI WSCInstallProvider( const LPGUID lpProviderId,
4325 LPCWSTR lpszProviderDllPath,
4326 const LPWSAPROTOCOL_INFOW lpProtocolInfoList,
4327 DWORD dwNumberOfEntries,
4328 LPINT lpErrno )
4329 {
4330 FIXME("(%s, %s, %p, %d, %p): stub !\n", debugstr_guid(lpProviderId),
4331 debugstr_w(lpszProviderDllPath), lpProtocolInfoList,
4332 dwNumberOfEntries, lpErrno);
4333 *lpErrno = 0;
4334 return 0;
4335 }
4336
4337
4338 /***********************************************************************
4339 * WSCDeinstallProvider (WS2_32.83)
4340 */
4341 INT WINAPI WSCDeinstallProvider(LPGUID lpProviderId, LPINT lpErrno)
4342 {
4343 FIXME("(%s, %p): stub !\n", debugstr_guid(lpProviderId), lpErrno);
4344 *lpErrno = 0;
4345 return 0;
4346 }
4347
4348
4349 /***********************************************************************
4350 * WSAAccept (WS2_32.26)
4351 */
4352 SOCKET WINAPI WSAAccept( SOCKET s, struct WS_sockaddr *addr, LPINT addrlen,
4353 LPCONDITIONPROC lpfnCondition, DWORD dwCallbackData)
4354 {
4355
4356 int ret = 0, size = 0;
4357 WSABUF CallerId, CallerData, CalleeId, CalleeData;
4358 /* QOS SQOS, GQOS; */
4359 GROUP g;
4360 SOCKET cs;
4361 SOCKADDR src_addr, dst_addr;
4362
4363 TRACE("Socket %04lx, sockaddr %p, addrlen %p, fnCondition %p, dwCallbackData %d\n",
4364 s, addr, addrlen, lpfnCondition, dwCallbackData);
4365
4366
4367 size = sizeof(src_addr);
4368 cs = WS_accept(s, &src_addr, &size);
4369
4370 if (cs == SOCKET_ERROR) return SOCKET_ERROR;
4371
4372 CallerId.buf = (char *)&src_addr;
4373 CallerId.len = sizeof(src_addr);
4374
4375 CallerData.buf = NULL;
4376 CallerData.len = (ULONG)NULL;
4377
4378 WS_getsockname(cs, &dst_addr, &size);
4379
4380 CalleeId.buf = (char *)&dst_addr;
4381 CalleeId.len = sizeof(dst_addr);
4382
4383
4384 ret = (*lpfnCondition)(&CallerId, &CallerData, NULL, NULL,
4385 &CalleeId, &CalleeData, &g, dwCallbackData);
4386
4387 switch (ret)
4388 {
4389 case CF_ACCEPT:
4390 if (addr && addrlen)
4391 addr = memcpy(addr, &src_addr, (*addrlen > size) ? size : *addrlen );
4392 return cs;
4393 case CF_DEFER:
4394 SERVER_START_REQ( set_socket_deferred )
4395 {
4396 req->handle = SOCKET2HANDLE(s);
4397 req->deferred = SOCKET2HANDLE(cs);
4398 if ( !wine_server_call_err ( req ) )
4399 {
4400 SetLastError( WSATRY_AGAIN );
4401 WS_closesocket( cs );
4402 }
4403 }
4404 SERVER_END_REQ;
4405 return SOCKET_ERROR;
4406 case CF_REJECT:
4407 WS_closesocket(cs);
4408 SetLastError(WSAECONNREFUSED);
4409 return SOCKET_ERROR;
4410 default:
4411 FIXME("Unknown return type from Condition function\n");
4412 SetLastError(WSAENOTSOCK);
4413 return SOCKET_ERROR;
4414 }
4415 }
4416
4417 /***********************************************************************
4418 * WSADuplicateSocketA (WS2_32.32)
4419 */
4420 int WINAPI WSADuplicateSocketA( SOCKET s, DWORD dwProcessId, LPWSAPROTOCOL_INFOA lpProtocolInfo )
4421 {
4422 HANDLE hProcess;
4423
4424 TRACE("(%ld,%x,%p)\n", s, dwProcessId, lpProtocolInfo);
4425 memset(lpProtocolInfo, 0, sizeof(*lpProtocolInfo));
4426 /* FIXME: WS_getsockopt(s, WS_SOL_SOCKET, SO_PROTOCOL_INFO, lpProtocolInfo, sizeof(*lpProtocolInfo)); */
4427 /* I don't know what the real Windoze does next, this is a hack */
4428 /* ...we could duplicate and then use ConvertToGlobalHandle on the duplicate, then let
4429 * the target use the global duplicate, or we could copy a reference to us to the structure
4430 * and let the target duplicate it from us, but let's do it as simple as possible */
4431 hProcess = OpenProcess(PROCESS_DUP_HANDLE, FALSE, dwProcessId);
4432 DuplicateHandle(GetCurrentProcess(), SOCKET2HANDLE(s),
4433 hProcess, (LPHANDLE)&lpProtocolInfo->dwCatalogEntryId,
4434 0, FALSE, DUPLICATE_SAME_ACCESS);
4435 CloseHandle(hProcess);
4436 lpProtocolInfo->dwServiceFlags4 = 0xff00ff00; /* magic */
4437 return 0;
4438 }
4439
4440 /***********************************************************************
4441 * WSADuplicateSocketW (WS2_32.33)
4442 */
4443 int WINAPI WSADuplicateSocketW( SOCKET s, DWORD dwProcessId, LPWSAPROTOCOL_INFOW lpProtocolInfo )
4444 {
4445 HANDLE hProcess;
4446
4447 TRACE("(%ld,%x,%p)\n", s, dwProcessId, lpProtocolInfo);
4448
4449 memset(lpProtocolInfo, 0, sizeof(*lpProtocolInfo));
4450 hProcess = OpenProcess(PROCESS_DUP_HANDLE, FALSE, dwProcessId);
4451 DuplicateHandle(GetCurrentProcess(), SOCKET2HANDLE(s),
4452 hProcess, (LPHANDLE)&lpProtocolInfo->dwCatalogEntryId,
4453 0, FALSE, DUPLICATE_SAME_ACCESS);
4454 CloseHandle(hProcess);
4455 lpProtocolInfo->dwServiceFlags4 = 0xff00ff00; /* magic */
4456 return 0;
4457 }
4458
4459 /***********************************************************************
4460 * WSAInstallServiceClassA (WS2_32.48)
4461 */
4462 int WINAPI WSAInstallServiceClassA(LPWSASERVICECLASSINFOA info)
4463 {
4464 FIXME("Request to install service %s\n",debugstr_a(info->lpszServiceClassName));
4465 WSASetLastError(WSAEACCES);
4466 return SOCKET_ERROR;
4467 }
4468
4469 /***********************************************************************
4470 * WSAInstallServiceClassW (WS2_32.49)
4471 */
4472 int WINAPI WSAInstallServiceClassW(LPWSASERVICECLASSINFOW info)
4473 {
4474 FIXME("Request to install service %s\n",debugstr_w(info->lpszServiceClassName));
4475 WSASetLastError(WSAEACCES);
4476 return SOCKET_ERROR;
4477 }
4478
4479 /***********************************************************************
4480 * WSARemoveServiceClass (WS2_32.70)
4481 */
4482 int WINAPI WSARemoveServiceClass(LPGUID info)
4483 {
4484 FIXME("Request to remove service %p\n",info);
4485 WSASetLastError(WSATYPE_NOT_FOUND);
4486 return SOCKET_ERROR;
4487 }
4488
4489 /***********************************************************************
4490 * WSAStringToAddressA (WS2_32.80)
4491 */
4492 INT WINAPI WSAStringToAddressA(LPSTR AddressString,
4493 INT AddressFamily,
4494 LPWSAPROTOCOL_INFOA lpProtocolInfo,
4495 LPSOCKADDR lpAddress,
4496 LPINT lpAddressLength)
4497 {
4498 INT res=0;
4499 LPSTR workBuffer=NULL,ptrPort;
4500
4501 TRACE( "(%s, %x, %p, %p, %p)\n", AddressString, AddressFamily, lpProtocolInfo,
4502 lpAddress, lpAddressLength );
4503
4504 if (!lpAddressLength || !lpAddress) return SOCKET_ERROR;
4505
4506 if (!AddressString)
4507 {
4508 WSASetLastError(WSAEINVAL);
4509 return SOCKET_ERROR;
4510 }
4511
4512 if (lpProtocolInfo)
4513 FIXME("ProtocolInfo not implemented.\n");
4514
4515 workBuffer = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
4516 strlen(AddressString) + 1);
4517 if (!workBuffer)
4518 {
4519 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
4520 return SOCKET_ERROR;
4521 }
4522
4523 strcpy(workBuffer, AddressString);
4524
4525 switch(AddressFamily)
4526 {
4527 case WS_AF_INET:
4528 {
4529 struct in_addr inetaddr;
4530
4531 /* If lpAddressLength is too small, tell caller the size we need */
4532 if (*lpAddressLength < sizeof(SOCKADDR_IN))
4533 {
4534 *lpAddressLength = sizeof(SOCKADDR_IN);
4535 res = WSAEFAULT;
4536 break;
4537 }
4538 memset(lpAddress, 0, sizeof(SOCKADDR_IN));
4539
4540 ((LPSOCKADDR_IN)lpAddress)->sin_family = AF_INET;
4541
4542 ptrPort = strchr(workBuffer, ':');
4543 if(ptrPort)
4544 {
4545 ((LPSOCKADDR_IN)lpAddress)->sin_port = (WS_u_short)atoi(ptrPort+1);
4546 *ptrPort = '\0';
4547 }
4548 else
4549 {
4550 ((LPSOCKADDR_IN)lpAddress)->sin_port = 0;
4551 }
4552
4553 if(inet_aton(workBuffer, &inetaddr) > 0)
4554 {
4555 ((LPSOCKADDR_IN)lpAddress)->sin_addr.WS_s_addr = inetaddr.s_addr;
4556 res = 0;
4557 }
4558 else
4559 res = WSAEINVAL;
4560
4561 break;
4562
4563 }
4564 case WS_AF_INET6:
4565 {
4566 struct in6_addr inetaddr;
4567 /* If lpAddressLength is too small, tell caller the size we need */
4568 if (*lpAddressLength < sizeof(SOCKADDR_IN6))
4569 {
4570 *lpAddressLength = sizeof(SOCKADDR_IN6);
4571 res = WSAEFAULT;
4572 break;
4573 }
4574 #ifdef HAVE_INET_PTON
4575 memset(lpAddress, 0, sizeof(SOCKADDR_IN6));
4576
4577 ((LPSOCKADDR_IN6)lpAddress)->sin6_family = WS_AF_INET6;
4578
4579 /* This one is a bit tricky. An IPv6 address contains colons, so the
4580 * check from IPv4 doesn't work like that. However, IPv6 addresses that
4581 * contain a port are written with braces like [fd12:3456:7890::1]:12345
4582 * so what we will do is to look for ']', check if the next char is a
4583 * colon, and if it is, parse the port as in IPv4. */
4584
4585 ptrPort = strchr(workBuffer, ']');
4586 if(ptrPort && *(++ptrPort) == ':')
4587 {
4588 ((LPSOCKADDR_IN6)lpAddress)->sin6_port = (WS_u_short)atoi(ptrPort+1);
4589 *ptrPort = '\0';
4590 }
4591 else
4592 {
4593 ((LPSOCKADDR_IN6)lpAddress)->sin6_port = 0;
4594 }
4595
4596 if(inet_pton(AF_INET6, workBuffer, &inetaddr) > 0)
4597 {
4598 memcpy(&((LPSOCKADDR_IN6)lpAddress)->sin6_addr, &inetaddr,
4599 sizeof(struct in6_addr));
4600 res = 0;
4601 }
4602 else
4603 #endif /* HAVE_INET_PTON */
4604 res = WSAEINVAL;
4605
4606 break;
4607 }
4608 default:
4609 /* According to MSDN, only AF_INET and AF_INET6 are supported. */
4610 TRACE("Unsupported address family specified: %d.\n", AddressFamily);
4611 res = WSAEINVAL;
4612 }
4613
4614 HeapFree(GetProcessHeap(), 0, workBuffer);
4615
4616 if (!res) return 0;
4617 WSASetLastError(res);
4618 return SOCKET_ERROR;
4619 }
4620
4621 /***********************************************************************
4622 * WSAStringToAddressW (WS2_32.81)
4623 *
4624 * Does anybody know if this functions allows to use hebrew/arabic/chinese... digits?
4625 * If this should be the case, it would be required to map these digits
4626 * to Unicode digits (0-9) using FoldString first.
4627 */
4628 INT WINAPI WSAStringToAddressW(LPWSTR AddressString,
4629 INT AddressFamily,
4630 LPWSAPROTOCOL_INFOW lpProtocolInfo,
4631 LPSOCKADDR lpAddress,
4632 LPINT lpAddressLength)
4633 {
4634 INT sBuffer,res=0;
4635 LPSTR workBuffer=NULL;
4636 WSAPROTOCOL_INFOA infoA;
4637 LPWSAPROTOCOL_INFOA lpProtoInfoA = NULL;
4638
4639 TRACE( "(%s, %x, %p, %p, %p)\n", debugstr_w(AddressString), AddressFamily, lpProtocolInfo,
4640 lpAddress, lpAddressLength );
4641
4642 if (!lpAddressLength || !lpAddress) return SOCKET_ERROR;
4643
4644 /* if ProtocolInfo is available - convert to ANSI variant */
4645 if (lpProtocolInfo)
4646 {
4647 lpProtoInfoA = &infoA;
4648 memcpy( lpProtoInfoA, lpProtocolInfo, FIELD_OFFSET( WSAPROTOCOL_INFOA, szProtocol ) );
4649
4650 if (!WideCharToMultiByte( CP_ACP, 0, lpProtocolInfo->szProtocol, -1,
4651 lpProtoInfoA->szProtocol, WSAPROTOCOL_LEN+1, NULL, NULL ))
4652 {
4653 WSASetLastError( WSAEINVAL);
4654 return SOCKET_ERROR;
4655 }
4656 }
4657
4658 if (AddressString)
4659 {
4660 /* Translate AddressString to ANSI code page - assumes that only
4661 standard digits 0-9 are used with this API call */
4662 sBuffer = WideCharToMultiByte( CP_ACP, 0, AddressString, -1, NULL, 0, NULL, NULL );
4663 workBuffer = HeapAlloc( GetProcessHeap(), 0, sBuffer );
4664
4665 if (workBuffer)
4666 {
4667 WideCharToMultiByte( CP_ACP, 0, AddressString, -1, workBuffer, sBuffer, NULL, NULL );
4668 res = WSAStringToAddressA(workBuffer,AddressFamily,lpProtoInfoA,
4669 lpAddress,lpAddressLength);
4670 HeapFree( GetProcessHeap(), 0, workBuffer );
4671 return res;
4672 }
4673 else
4674 res = WSA_NOT_ENOUGH_MEMORY;
4675 }
4676 else
4677 res = WSAEINVAL;
4678
4679 WSASetLastError(res);
4680 return SOCKET_ERROR;
4681 }
4682
4683 /***********************************************************************
4684 * WSAAddressToStringA (WS2_32.27)
4685 *
4686 * See WSAAddressToStringW
4687 */
4688 INT WINAPI WSAAddressToStringA( LPSOCKADDR sockaddr, DWORD len,
4689 LPWSAPROTOCOL_INFOA info, LPSTR string,
4690 LPDWORD lenstr )
4691 {
4692 INT size;
4693 CHAR buffer[22]; /* 12 digits + 3 dots + ':' + 5 digits + '\0' */
4694 CHAR *p;
4695
4696 TRACE( "(%p, %d, %p, %p, %p)\n", sockaddr, len, info, string, lenstr );
4697
4698 if (!sockaddr || len < sizeof(SOCKADDR_IN)) return SOCKET_ERROR;
4699 if (!string || !lenstr) return SOCKET_ERROR;
4700
4701 /* sin_family is guaranteed to be the first u_short */
4702 if (((SOCKADDR_IN *)sockaddr)->sin_family != AF_INET) return SOCKET_ERROR;
4703
4704 sprintf( buffer, "%u.%u.%u.%u:%u",
4705 (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) >> 24 & 0xff),
4706 (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) >> 16 & 0xff),
4707 (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) >> 8 & 0xff),
4708 (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) & 0xff),
4709 ntohs( ((SOCKADDR_IN *)sockaddr)->sin_port ) );
4710
4711 p = strchr( buffer, ':' );
4712 if (!((SOCKADDR_IN *)sockaddr)->sin_port) *p = 0;
4713
4714 size = strlen( buffer );
4715
4716 if (*lenstr < size)
4717 {
4718 *lenstr = size;
4719 WSASetLastError(WSAEFAULT);
4720 return SOCKET_ERROR;
4721 }
4722
4723 strcpy( string, buffer );
4724 return 0;
4725 }
4726
4727 /***********************************************************************
4728 * WSAAddressToStringW (WS2_32.28)
4729 *
4730 * Convert a sockaddr address into a readable address string.
4731 *
4732 * PARAMS
4733 * sockaddr [I] Pointer to a sockaddr structure.
4734 * len [I] Size of the sockaddr structure.
4735 * info [I] Pointer to a WSAPROTOCOL_INFOW structure (optional).
4736 * string [I/O] Pointer to a buffer to receive the address string.
4737 * lenstr [I/O] Size of the receive buffer in WCHARs.
4738 *
4739 * RETURNS
4740 * Success: 0
4741 * Failure: SOCKET_ERROR
4742 *
4743 * NOTES
4744 * The 'info' parameter is ignored.
4745 *
4746 * BUGS
4747 * Only supports AF_INET addresses.
4748 */
4749 INT WINAPI WSAAddressToStringW( LPSOCKADDR sockaddr, DWORD len,
4750 LPWSAPROTOCOL_INFOW info, LPWSTR string,
4751 LPDWORD lenstr )
4752 {
4753 INT size;
4754 WCHAR buffer[22]; /* 12 digits + 3 dots + ':' + 5 digits + '\0' */
4755 static const WCHAR format[] = { '%','u','.','%','u','.','%','u','.','%','u',':','%','u',0 };
4756 WCHAR *p;
4757
4758 TRACE( "(%p, %x, %p, %p, %p)\n", sockaddr, len, info, string, lenstr );
4759
4760 if (!sockaddr || len < sizeof(SOCKADDR_IN)) return SOCKET_ERROR;
4761 if (!string || !lenstr) return SOCKET_ERROR;
4762
4763 /* sin_family is guaranteed to be the first u_short */
4764 if (((SOCKADDR_IN *)sockaddr)->sin_family != AF_INET) return SOCKET_ERROR;
4765
4766 sprintfW( buffer, format,
4767 (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) >> 24 & 0xff),
4768 (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) >> 16 & 0xff),
4769 (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) >> 8 & 0xff),
4770 (unsigned int)(ntohl( ((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr ) & 0xff),
4771 ntohs( ((SOCKADDR_IN *)sockaddr)->sin_port ) );
4772
4773 p = strchrW( buffer, ':' );
4774 if (!((SOCKADDR_IN *)sockaddr)->sin_port) *p = 0;
4775
4776 size = lstrlenW( buffer );
4777
4778 if (*lenstr < size)
4779 {
4780 *lenstr = size;
4781 return SOCKET_ERROR;
4782 }
4783
4784 lstrcpyW( string, buffer );
4785 return 0;
4786 }
4787
4788 /***********************************************************************
4789 * WSAEnumNameSpaceProvidersA (WS2_32.34)
4790 */
4791 INT WINAPI WSAEnumNameSpaceProvidersA( LPDWORD len, LPWSANAMESPACE_INFOA buffer )
4792 {
4793 FIXME( "(%p %p) Stub!\n", len, buffer );
4794 return 0;
4795 }
4796
4797 /***********************************************************************
4798 * WSAEnumNameSpaceProvidersW (WS2_32.35)
4799 */
4800 INT WINAPI WSAEnumNameSpaceProvidersW( LPDWORD len, LPWSANAMESPACE_INFOW buffer )
4801 {
4802 FIXME( "(%p %p) Stub!\n", len, buffer );
4803 return 0;
4804 }
4805
4806 /***********************************************************************
4807 * WSAGetQOSByName (WS2_32.41)
4808 */
4809 BOOL WINAPI WSAGetQOSByName( SOCKET s, LPWSABUF lpQOSName, LPQOS lpQOS )
4810 {
4811 FIXME( "(0x%04lx %p %p) Stub!\n", s, lpQOSName, lpQOS );
4812 return FALSE;
4813 }
4814
4815 /***********************************************************************
4816 * WSAGetServiceClassInfoA (WS2_32.42)
4817 */
4818 INT WINAPI WSAGetServiceClassInfoA( LPGUID provider, LPGUID service, LPDWORD len,
4819 LPWSASERVICECLASSINFOA info )
4820 {
4821 FIXME( "(%s %s %p %p) Stub!\n", debugstr_guid(provider), debugstr_guid(service),
4822 len, info );
4823 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
4824 return SOCKET_ERROR;
4825 }
4826
4827 /***********************************************************************
4828 * WSAGetServiceClassInfoW (WS2_32.43)
4829 */
4830 INT WINAPI WSAGetServiceClassInfoW( LPGUID provider, LPGUID service, LPDWORD len,
4831 LPWSASERVICECLASSINFOW info )
4832 {
4833 FIXME( "(%s %s %p %p) Stub!\n", debugstr_guid(provider), debugstr_guid(service),
4834 len, info );
4835 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
4836 return SOCKET_ERROR;
4837 }
4838
4839 /***********************************************************************
4840 * WSAGetServiceClassNameByClassIdA (WS2_32.44)
4841 */
4842 INT WINAPI WSAGetServiceClassNameByClassIdA( LPGUID class, LPSTR service, LPDWORD len )
4843 {
4844 FIXME( "(%s %p %p) Stub!\n", debugstr_guid(class), service, len );
4845 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
4846 return SOCKET_ERROR;
4847 }
4848
4849 /***********************************************************************
4850 * WSAGetServiceClassNameByClassIdW (WS2_32.45)
4851 */
4852 INT WINAPI WSAGetServiceClassNameByClassIdW( LPGUID class, LPWSTR service, LPDWORD len )
4853 {
4854 FIXME( "(%s %p %p) Stub!\n", debugstr_guid(class), service, len );
4855 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
4856 return SOCKET_ERROR;
4857 }
4858
4859 /***********************************************************************
4860 * WSALookupServiceBeginA (WS2_32.59)
4861 */
4862 INT WINAPI WSALookupServiceBeginA( LPWSAQUERYSETA lpqsRestrictions,
4863 DWORD dwControlFlags,
4864 LPHANDLE lphLookup)
4865 {
4866 FIXME("(%p 0x%08x %p) Stub!\n", lpqsRestrictions, dwControlFlags,
4867 lphLookup);
4868 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
4869 return SOCKET_ERROR;
4870 }
4871
4872 /***********************************************************************
4873 * WSALookupServiceBeginW (WS2_32.60)
4874 */
4875 INT WINAPI WSALookupServiceBeginW( LPWSAQUERYSETW lpqsRestrictions,
4876 DWORD dwControlFlags,
4877 LPHANDLE lphLookup)
4878 {
4879 FIXME("(%p 0x%08x %p) Stub!\n", lpqsRestrictions, dwControlFlags,
4880 lphLookup);
4881 WSASetLastError(WSA_NOT_ENOUGH_MEMORY);
4882 return SOCKET_ERROR;
4883 }
4884
4885 /***********************************************************************
4886 * WSALookupServiceBeginW (WS2_32.61)
4887 */
4888 INT WINAPI WSALookupServiceEnd( HANDLE lookup )
4889 {
4890 FIXME("(%p) Stub!\n", lookup );
4891 return 0;
4892 }
4893
4894 /***********************************************************************
4895 * WSALookupServiceNextA (WS2_32.62)
4896 */
4897 INT WINAPI WSALookupServiceNextA( HANDLE lookup, DWORD flags, LPDWORD len, LPWSAQUERYSETA results )
4898 {
4899 FIXME( "(%p 0x%08x %p %p) Stub!\n", lookup, flags, len, results );
4900 return 0;
4901 }
4902
4903 /***********************************************************************
4904 * WSALookupServiceNextW (WS2_32.63)
4905 */
4906 INT WINAPI WSALookupServiceNextW( HANDLE lookup, DWORD flags, LPDWORD len, LPWSAQUERYSETW results )
4907 {
4908 FIXME( "(%p 0x%08x %p %p) Stub!\n", lookup, flags, len, results );
4909 return 0;
4910 }
4911
4912 /***********************************************************************
4913 * WSANtohl (WS2_32.64)
4914 */
4915 INT WINAPI WSANtohl( SOCKET s, WS_u_long netlong, WS_u_long* lphostlong )
4916 {
4917 TRACE( "(0x%04lx 0x%08x %p)\n", s, netlong, lphostlong );
4918
4919 if (!lphostlong) return WSAEFAULT;
4920
4921 *lphostlong = ntohl( netlong );
4922 return 0;
4923 }
4924
4925 /***********************************************************************
4926 * WSANtohs (WS2_32.65)
4927 */
4928 INT WINAPI WSANtohs( SOCKET s, WS_u_short netshort, WS_u_short* lphostshort )
4929 {
4930 TRACE( "(0x%04lx 0x%08x %p)\n", s, netshort, lphostshort );
4931
4932 if (!lphostshort) return WSAEFAULT;
4933
4934 *lphostshort = ntohs( netshort );
4935 return 0;
4936 }
4937
4938 /***********************************************************************
4939 * WSAProviderConfigChange (WS2_32.66)
4940 */
4941 INT WINAPI WSAProviderConfigChange( LPHANDLE handle, LPWSAOVERLAPPED overlapped,
4942 LPWSAOVERLAPPED_COMPLETION_ROUTINE completion )
4943 {
4944 FIXME( "(%p %p %p) Stub!\n", handle, overlapped, completion );
4945 return SOCKET_ERROR;
4946 }
4947
4948 /***********************************************************************
4949 * WSARecvDisconnect (WS2_32.68)
4950 */
4951 INT WINAPI WSARecvDisconnect( SOCKET s, LPWSABUF disconnectdata )
4952 {
4953 TRACE( "(0x%04lx %p)\n", s, disconnectdata );
4954
4955 return WS_shutdown( s, 0 );
4956 }
4957
4958 /***********************************************************************
4959 * WSASetServiceA (WS2_32.76)
4960 */
4961 INT WINAPI WSASetServiceA( LPWSAQUERYSETA query, WSAESETSERVICEOP operation, DWORD flags )
4962 {
4963 FIXME( "(%p 0x%08x 0x%08x) Stub!\n", query, operation, flags );
4964 return 0;
4965 }
4966
4967 /***********************************************************************
4968 * WSASetServiceW (WS2_32.77)
4969 */
4970 INT WINAPI WSASetServiceW( LPWSAQUERYSETW query, WSAESETSERVICEOP operation, DWORD flags )
4971 {
4972 FIXME( "(%p 0x%08x 0x%08x) Stub!\n", query, operation, flags );
4973 return 0;
4974 }
4975
4976 /***********************************************************************
4977 * WSCEnableNSProvider (WS2_32.84)
4978 */
4979 INT WINAPI WSCEnableNSProvider( LPGUID provider, BOOL enable )
4980 {
4981 FIXME( "(%s 0x%08x) Stub!\n", debugstr_guid(provider), enable );
4982 return 0;
4983 }
4984
4985 /***********************************************************************
4986 * WSCGetProviderPath (WS2_32.86)
4987 */
4988 INT WINAPI WSCGetProviderPath( LPGUID provider, LPWSTR path, LPINT len, LPINT errcode )
4989 {
4990 FIXME( "(%s %p %p %p) Stub!\n", debugstr_guid(provider), path, len, errcode );
4991
4992 if (!errcode || !provider || !len) return WSAEFAULT;
4993
4994 *errcode = WSAEINVAL;
4995 return SOCKET_ERROR;
4996 }
4997
4998 /***********************************************************************
4999 * WSCInstallNameSpace (WS2_32.87)
5000 */
5001 INT WINAPI WSCInstallNameSpace( LPWSTR identifier, LPWSTR path, DWORD namespace,
5002 DWORD version, LPGUID provider )
5003 {
5004 FIXME( "(%s %s 0x%08x 0x%08x %s) Stub!\n", debugstr_w(identifier), debugstr_w(path),
5005 namespace, version, debugstr_guid(provider) );
5006 return 0;
5007 }
5008
5009 /***********************************************************************
5010 * WSCUnInstallNameSpace (WS2_32.89)
5011 */
5012 INT WINAPI WSCUnInstallNameSpace( LPGUID lpProviderId )
5013 {
5014 FIXME("(%p) Stub!\n", lpProviderId);
5015 return NO_ERROR;
5016 }
5017
5018 /***********************************************************************
5019 * WSCWriteProviderOrder (WS2_32.91)
5020 */
5021 INT WINAPI WSCWriteProviderOrder( LPDWORD entry, DWORD number )
5022 {
5023 FIXME("(%p 0x%08x) Stub!\n", entry, number);
5024 return 0;
5025 }
Windows API implementation