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