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