push b5232b2081a0e20e4bf07d6ded424d0101e4a589
[wine/hacks.git] / dlls / ws2_32 / tests / sock.c
blobd04552e7211e24367cc2c93e1d98ab2accb96549
1 /*
2 * Unit test suite for winsock functions
4 * Copyright 2002 Martin Wilck
5 * Copyright 2005 Thomas Kho
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
22 #include <stdarg.h>
24 #include <windef.h>
25 #include <winbase.h>
26 #include <winsock2.h>
27 #include <ws2tcpip.h>
28 #include <mswsock.h>
29 #include "wine/test.h"
30 #include <winnt.h>
31 #include <winerror.h>
33 #define MAX_CLIENTS 4 /* Max number of clients */
34 #define NUM_TESTS 3 /* Number of tests performed */
35 #define FIRST_CHAR 'A' /* First character in transferred pattern */
36 #define BIND_SLEEP 10 /* seconds to wait between attempts to bind() */
37 #define BIND_TRIES 6 /* Number of bind() attempts */
38 #define TEST_TIMEOUT 30 /* seconds to wait before killing child threads
39 after server initialization, if something hangs */
41 #define NUM_UDP_PEERS 3 /* Number of UDP sockets to create and test > 1 */
43 #define NUM_THREADS 3 /* Number of threads to run getservbyname */
44 #define NUM_QUERIES 250 /* Number of getservbyname queries per thread */
46 #define SERVERIP "127.0.0.1" /* IP to bind to */
47 #define SERVERPORT 9374 /* Port number to bind to */
49 #define wsa_ok(op, cond, msg) \
50 do { \
51 int tmp, err = 0; \
52 tmp = op; \
53 if ( !(cond tmp) ) err = WSAGetLastError(); \
54 ok ( cond tmp, msg, GetCurrentThreadId(), err); \
55 } while (0);
58 /**************** Structs and typedefs ***************/
60 typedef struct thread_info
62 HANDLE thread;
63 DWORD id;
64 } thread_info;
66 /* Information in the server about open client connections */
67 typedef struct sock_info
69 SOCKET s;
70 struct sockaddr_in addr;
71 struct sockaddr_in peer;
72 char *buf;
73 int n_recvd;
74 int n_sent;
75 } sock_info;
77 /* Test parameters for both server & client */
78 typedef struct test_params
80 int sock_type;
81 int sock_prot;
82 const char *inet_addr;
83 short inet_port;
84 int chunk_size;
85 int n_chunks;
86 int n_clients;
87 } test_params;
89 /* server-specific test parameters */
90 typedef struct server_params
92 test_params *general;
93 DWORD sock_flags;
94 int buflen;
95 } server_params;
97 /* client-specific test parameters */
98 typedef struct client_params
100 test_params *general;
101 DWORD sock_flags;
102 int buflen;
103 } client_params;
105 /* This type combines all information for setting up a test scenario */
106 typedef struct test_setup
108 test_params general;
109 LPVOID srv;
110 server_params srv_params;
111 LPVOID clt;
112 client_params clt_params;
113 } test_setup;
115 /* Thread local storage for server */
116 typedef struct server_memory
118 SOCKET s;
119 struct sockaddr_in addr;
120 sock_info sock[MAX_CLIENTS];
121 } server_memory;
123 /* Thread local storage for client */
124 typedef struct client_memory
126 SOCKET s;
127 struct sockaddr_in addr;
128 char *send_buf;
129 char *recv_buf;
130 } client_memory;
132 /* SelectReadThread thread parameters */
133 typedef struct select_thread_params
135 SOCKET s;
136 BOOL ReadKilled;
137 } select_thread_params;
139 /**************** Static variables ***************/
141 static DWORD tls; /* Thread local storage index */
142 static HANDLE thread[1+MAX_CLIENTS];
143 static DWORD thread_id[1+MAX_CLIENTS];
144 static HANDLE server_ready;
145 static HANDLE client_ready[MAX_CLIENTS];
146 static int client_id;
148 /**************** General utility functions ***************/
150 static int tcp_socketpair(SOCKET *src, SOCKET *dst)
152 SOCKET server = INVALID_SOCKET;
153 struct sockaddr_in addr;
154 int len;
155 int ret;
157 *src = INVALID_SOCKET;
158 *dst = INVALID_SOCKET;
160 *src = socket(AF_INET, SOCK_STREAM, 0);
161 if (*src == INVALID_SOCKET)
162 goto end;
164 server = socket(AF_INET, SOCK_STREAM, 0);
165 if (server == INVALID_SOCKET)
166 goto end;
168 memset(&addr, 0, sizeof(addr));
169 addr.sin_family = AF_INET;
170 addr.sin_addr.s_addr = inet_addr("127.0.0.1");
171 ret = bind(server, (struct sockaddr*)&addr, sizeof(addr));
172 if (ret != 0)
173 goto end;
175 len = sizeof(addr);
176 ret = getsockname(server, (struct sockaddr*)&addr, &len);
177 if (ret != 0)
178 goto end;
180 ret = listen(server, 1);
181 if (ret != 0)
182 goto end;
184 ret = connect(*src, (struct sockaddr*)&addr, sizeof(addr));
185 if (ret != 0)
186 goto end;
188 len = sizeof(addr);
189 *dst = accept(server, (struct sockaddr*)&addr, &len);
191 end:
192 if (server != INVALID_SOCKET)
193 closesocket(server);
194 if (*src != INVALID_SOCKET && *dst != INVALID_SOCKET)
195 return 0;
196 closesocket(*src);
197 closesocket(*dst);
198 return -1;
201 static void set_so_opentype ( BOOL overlapped )
203 int optval = !overlapped, newval, len = sizeof (int);
205 ok ( setsockopt ( INVALID_SOCKET, SOL_SOCKET, SO_OPENTYPE,
206 (LPVOID) &optval, sizeof (optval) ) == 0,
207 "setting SO_OPENTYPE failed\n" );
208 ok ( getsockopt ( INVALID_SOCKET, SOL_SOCKET, SO_OPENTYPE,
209 (LPVOID) &newval, &len ) == 0,
210 "getting SO_OPENTYPE failed\n" );
211 ok ( optval == newval, "failed to set SO_OPENTYPE\n" );
214 static int set_blocking ( SOCKET s, BOOL blocking )
216 u_long val = !blocking;
217 return ioctlsocket ( s, FIONBIO, &val );
220 static void fill_buffer ( char *buf, int chunk_size, int n_chunks )
222 char c, *p;
223 for ( c = FIRST_CHAR, p = buf; c < FIRST_CHAR + n_chunks; c++, p += chunk_size )
224 memset ( p, c, chunk_size );
227 static char* test_buffer ( char *buf, int chunk_size, int n_chunks )
229 char c, *p;
230 int i;
231 for ( c = FIRST_CHAR, p = buf; c < FIRST_CHAR + n_chunks; c++, p += chunk_size )
233 for ( i = 0; i < chunk_size; i++ )
234 if ( p[i] != c ) return p + i;
236 return NULL;
240 * This routine is called when a client / server does not expect any more data,
241 * but needs to acknowedge the closing of the connection (by reasing 0 bytes).
243 static void read_zero_bytes ( SOCKET s )
245 char buf[256];
246 int tmp, n = 0;
247 while ( ( tmp = recv ( s, buf, 256, 0 ) ) > 0 )
248 n += tmp;
249 ok ( n <= 0, "garbage data received: %d bytes\n", n );
252 static int do_synchronous_send ( SOCKET s, char *buf, int buflen, int sendlen )
254 char* last = buf + buflen, *p;
255 int n = 1;
256 for ( p = buf; n > 0 && p < last; p += n )
257 n = send ( s, p, min ( sendlen, last - p ), 0 );
258 wsa_ok ( n, 0 <=, "do_synchronous_send (%x): error %d\n" );
259 return p - buf;
262 static int do_synchronous_recv ( SOCKET s, char *buf, int buflen, int recvlen )
264 char* last = buf + buflen, *p;
265 int n = 1;
266 for ( p = buf; n > 0 && p < last; p += n )
267 n = recv ( s, p, min ( recvlen, last - p ), 0 );
268 wsa_ok ( n, 0 <=, "do_synchronous_recv (%x): error %d:\n" );
269 return p - buf;
273 * Call this routine right after thread startup.
274 * SO_OPENTYPE must by 0, regardless what the server did.
276 static void check_so_opentype (void)
278 int tmp = 1, len;
279 len = sizeof (tmp);
280 getsockopt ( INVALID_SOCKET, SOL_SOCKET, SO_OPENTYPE, (LPVOID) &tmp, &len );
281 ok ( tmp == 0, "check_so_opentype: wrong startup value of SO_OPENTYPE: %d\n", tmp );
284 /**************** Server utility functions ***************/
287 * Even if we have closed our server socket cleanly,
288 * the OS may mark the address "in use" for some time -
289 * this happens with native Linux apps, too.
291 static void do_bind ( SOCKET s, struct sockaddr* addr, int addrlen )
293 int err, wsaerr = 0, n_try = BIND_TRIES;
295 while ( ( err = bind ( s, addr, addrlen ) ) != 0 &&
296 ( wsaerr = WSAGetLastError () ) == WSAEADDRINUSE &&
297 n_try-- >= 0)
299 trace ( "address in use, waiting ...\n" );
300 Sleep ( 1000 * BIND_SLEEP );
302 ok ( err == 0, "failed to bind: %d\n", wsaerr );
305 static void server_start ( server_params *par )
307 int i;
308 test_params *gen = par->general;
309 server_memory *mem = (LPVOID) LocalAlloc ( LPTR, sizeof (server_memory));
311 TlsSetValue ( tls, mem );
312 mem->s = WSASocketA ( AF_INET, gen->sock_type, gen->sock_prot,
313 NULL, 0, par->sock_flags );
314 ok ( mem->s != INVALID_SOCKET, "Server: WSASocket failed\n" );
316 mem->addr.sin_family = AF_INET;
317 mem->addr.sin_addr.s_addr = inet_addr ( gen->inet_addr );
318 mem->addr.sin_port = htons ( gen->inet_port );
320 for (i = 0; i < MAX_CLIENTS; i++)
322 mem->sock[i].s = INVALID_SOCKET;
323 mem->sock[i].buf = (LPVOID) LocalAlloc ( LPTR, gen->n_chunks * gen->chunk_size );
324 mem->sock[i].n_recvd = 0;
325 mem->sock[i].n_sent = 0;
328 if ( gen->sock_type == SOCK_STREAM )
329 do_bind ( mem->s, (struct sockaddr*) &mem->addr, sizeof (mem->addr) );
332 static void server_stop (void)
334 int i;
335 server_memory *mem = TlsGetValue ( tls );
337 for (i = 0; i < MAX_CLIENTS; i++ )
339 LocalFree ( (HANDLE) mem->sock[i].buf );
340 if ( mem->sock[i].s != INVALID_SOCKET )
341 closesocket ( mem->sock[i].s );
343 ok ( closesocket ( mem->s ) == 0, "closesocket failed\n" );
344 LocalFree ( (HANDLE) mem );
345 ExitThread ( GetCurrentThreadId () );
348 /**************** Client utilitiy functions ***************/
350 static void client_start ( client_params *par )
352 test_params *gen = par->general;
353 client_memory *mem = (LPVOID) LocalAlloc (LPTR, sizeof (client_memory));
355 TlsSetValue ( tls, mem );
357 WaitForSingleObject ( server_ready, INFINITE );
359 mem->s = WSASocketA ( AF_INET, gen->sock_type, gen->sock_prot,
360 NULL, 0, par->sock_flags );
362 mem->addr.sin_family = AF_INET;
363 mem->addr.sin_addr.s_addr = inet_addr ( gen->inet_addr );
364 mem->addr.sin_port = htons ( gen->inet_port );
366 ok ( mem->s != INVALID_SOCKET, "Client: WSASocket failed\n" );
368 mem->send_buf = (LPVOID) LocalAlloc ( LPTR, 2 * gen->n_chunks * gen->chunk_size );
369 mem->recv_buf = mem->send_buf + gen->n_chunks * gen->chunk_size;
370 fill_buffer ( mem->send_buf, gen->chunk_size, gen->n_chunks );
372 SetEvent ( client_ready[client_id] );
373 /* Wait for the other clients to come up */
374 WaitForMultipleObjects ( min ( gen->n_clients, MAX_CLIENTS ), client_ready, TRUE, INFINITE );
377 static void client_stop (void)
379 client_memory *mem = TlsGetValue ( tls );
380 wsa_ok ( closesocket ( mem->s ), 0 ==, "closesocket error (%x): %d\n" );
381 LocalFree ( (HANDLE) mem->send_buf );
382 LocalFree ( (HANDLE) mem );
383 ExitThread(0);
386 /**************** Servers ***************/
389 * simple_server: A very basic server doing synchronous IO.
391 static VOID WINAPI simple_server ( server_params *par )
393 test_params *gen = par->general;
394 server_memory *mem;
395 int n_recvd, n_sent, n_expected = gen->n_chunks * gen->chunk_size, tmp, i,
396 id = GetCurrentThreadId();
397 char *p;
399 trace ( "simple_server (%x) starting\n", id );
401 set_so_opentype ( FALSE ); /* non-overlapped */
402 server_start ( par );
403 mem = TlsGetValue ( tls );
405 wsa_ok ( set_blocking ( mem->s, TRUE ), 0 ==, "simple_server (%x): failed to set blocking mode: %d\n");
406 wsa_ok ( listen ( mem->s, SOMAXCONN ), 0 ==, "simple_server (%x): listen failed: %d\n");
408 trace ( "simple_server (%x) ready\n", id );
409 SetEvent ( server_ready ); /* notify clients */
411 for ( i = 0; i < min ( gen->n_clients, MAX_CLIENTS ); i++ )
413 trace ( "simple_server (%x): waiting for client\n", id );
415 /* accept a single connection */
416 tmp = sizeof ( mem->sock[0].peer );
417 mem->sock[0].s = accept ( mem->s, (struct sockaddr*) &mem->sock[0].peer, &tmp );
418 wsa_ok ( mem->sock[0].s, INVALID_SOCKET !=, "simple_server (%x): accept failed: %d\n" );
420 ok ( mem->sock[0].peer.sin_addr.s_addr == inet_addr ( gen->inet_addr ),
421 "simple_server (%x): strange peer address\n", id );
423 /* Receive data & check it */
424 n_recvd = do_synchronous_recv ( mem->sock[0].s, mem->sock[0].buf, n_expected, par->buflen );
425 ok ( n_recvd == n_expected,
426 "simple_server (%x): received less data than expected: %d of %d\n", id, n_recvd, n_expected );
427 p = test_buffer ( mem->sock[0].buf, gen->chunk_size, gen->n_chunks );
428 ok ( p == NULL, "simple_server (%x): test pattern error: %d\n", id, p - mem->sock[0].buf);
430 /* Echo data back */
431 n_sent = do_synchronous_send ( mem->sock[0].s, mem->sock[0].buf, n_expected, par->buflen );
432 ok ( n_sent == n_expected,
433 "simple_server (%x): sent less data than expected: %d of %d\n", id, n_sent, n_expected );
435 /* cleanup */
436 read_zero_bytes ( mem->sock[0].s );
437 wsa_ok ( closesocket ( mem->sock[0].s ), 0 ==, "simple_server (%x): closesocket error: %d\n" );
438 mem->sock[0].s = INVALID_SOCKET;
441 trace ( "simple_server (%x) exiting\n", id );
442 server_stop ();
446 * select_server: A non-blocking server.
448 static VOID WINAPI select_server ( server_params *par )
450 test_params *gen = par->general;
451 server_memory *mem;
452 int n_expected = gen->n_chunks * gen->chunk_size, tmp, i,
453 id = GetCurrentThreadId(), n_connections = 0, n_sent, n_recvd,
454 n_set, delta, n_ready;
455 char *p;
456 struct timeval timeout = {0,10}; /* wait for 10 milliseconds */
457 fd_set fds_recv, fds_send, fds_openrecv, fds_opensend;
459 trace ( "select_server (%x) starting\n", id );
461 set_so_opentype ( FALSE ); /* non-overlapped */
462 server_start ( par );
463 mem = TlsGetValue ( tls );
465 wsa_ok ( set_blocking ( mem->s, FALSE ), 0 ==, "select_server (%x): failed to set blocking mode: %d\n");
466 wsa_ok ( listen ( mem->s, SOMAXCONN ), 0 ==, "select_server (%x): listen failed: %d\n");
468 trace ( "select_server (%x) ready\n", id );
469 SetEvent ( server_ready ); /* notify clients */
471 FD_ZERO ( &fds_openrecv );
472 FD_ZERO ( &fds_recv );
473 FD_ZERO ( &fds_send );
474 FD_ZERO ( &fds_opensend );
476 FD_SET ( mem->s, &fds_openrecv );
478 while(1)
480 fds_recv = fds_openrecv;
481 fds_send = fds_opensend;
483 n_set = 0;
485 wsa_ok ( ( n_ready = select ( 0, &fds_recv, &fds_send, NULL, &timeout ) ), SOCKET_ERROR !=,
486 "select_server (%x): select() failed: %d\n" );
488 /* check for incoming requests */
489 if ( FD_ISSET ( mem->s, &fds_recv ) ) {
490 n_set += 1;
492 trace ( "select_server (%x): accepting client connection\n", id );
494 /* accept a single connection */
495 tmp = sizeof ( mem->sock[n_connections].peer );
496 mem->sock[n_connections].s = accept ( mem->s, (struct sockaddr*) &mem->sock[n_connections].peer, &tmp );
497 wsa_ok ( mem->sock[n_connections].s, INVALID_SOCKET !=, "select_server (%x): accept() failed: %d\n" );
499 ok ( mem->sock[n_connections].peer.sin_addr.s_addr == inet_addr ( gen->inet_addr ),
500 "select_server (%x): strange peer address\n", id );
502 /* add to list of open connections */
503 FD_SET ( mem->sock[n_connections].s, &fds_openrecv );
504 FD_SET ( mem->sock[n_connections].s, &fds_opensend );
506 n_connections++;
509 /* handle open requests */
511 for ( i = 0; i < n_connections; i++ )
513 if ( FD_ISSET( mem->sock[i].s, &fds_recv ) ) {
514 n_set += 1;
516 if ( mem->sock[i].n_recvd < n_expected ) {
517 /* Receive data & check it */
518 n_recvd = recv ( mem->sock[i].s, mem->sock[i].buf + mem->sock[i].n_recvd, min ( n_expected - mem->sock[i].n_recvd, par->buflen ), 0 );
519 ok ( n_recvd != SOCKET_ERROR, "select_server (%x): error in recv(): %d\n", id, WSAGetLastError() );
520 mem->sock[i].n_recvd += n_recvd;
522 if ( mem->sock[i].n_recvd == n_expected ) {
523 p = test_buffer ( mem->sock[i].buf, gen->chunk_size, gen->n_chunks );
524 ok ( p == NULL, "select_server (%x): test pattern error: %d\n", id, p - mem->sock[i].buf );
525 FD_CLR ( mem->sock[i].s, &fds_openrecv );
528 ok ( mem->sock[i].n_recvd <= n_expected, "select_server (%x): received too many bytes: %d\n", id, mem->sock[i].n_recvd );
532 /* only echo back what we've received */
533 delta = mem->sock[i].n_recvd - mem->sock[i].n_sent;
535 if ( FD_ISSET ( mem->sock[i].s, &fds_send ) ) {
536 n_set += 1;
538 if ( ( delta > 0 ) && ( mem->sock[i].n_sent < n_expected ) ) {
539 /* Echo data back */
540 n_sent = send ( mem->sock[i].s, mem->sock[i].buf + mem->sock[i].n_sent, min ( delta, par->buflen ), 0 );
541 ok ( n_sent != SOCKET_ERROR, "select_server (%x): error in send(): %d\n", id, WSAGetLastError() );
542 mem->sock[i].n_sent += n_sent;
544 if ( mem->sock[i].n_sent == n_expected ) {
545 FD_CLR ( mem->sock[i].s, &fds_opensend );
548 ok ( mem->sock[i].n_sent <= n_expected, "select_server (%x): sent too many bytes: %d\n", id, mem->sock[i].n_sent );
553 /* check that select returned the correct number of ready sockets */
554 ok ( ( n_set == n_ready ), "select_server (%x): select() returns wrong number of ready sockets\n", id );
556 /* check if all clients are done */
557 if ( ( fds_opensend.fd_count == 0 )
558 && ( fds_openrecv.fd_count == 1 ) /* initial socket that accepts clients */
559 && ( n_connections == min ( gen->n_clients, MAX_CLIENTS ) ) ) {
560 break;
564 for ( i = 0; i < min ( gen->n_clients, MAX_CLIENTS ); i++ )
566 /* cleanup */
567 read_zero_bytes ( mem->sock[i].s );
568 wsa_ok ( closesocket ( mem->sock[i].s ), 0 ==, "select_server (%x): closesocket error: %d\n" );
569 mem->sock[i].s = INVALID_SOCKET;
572 trace ( "select_server (%x) exiting\n", id );
573 server_stop ();
576 /**************** Clients ***************/
579 * simple_client: A very basic client doing synchronous IO.
581 static VOID WINAPI simple_client ( client_params *par )
583 test_params *gen = par->general;
584 client_memory *mem;
585 int n_sent, n_recvd, n_expected = gen->n_chunks * gen->chunk_size, id;
586 char *p;
588 id = GetCurrentThreadId();
589 trace ( "simple_client (%x): starting\n", id );
590 /* wait here because we want to call set_so_opentype before creating a socket */
591 WaitForSingleObject ( server_ready, INFINITE );
592 trace ( "simple_client (%x): server ready\n", id );
594 check_so_opentype ();
595 set_so_opentype ( FALSE ); /* non-overlapped */
596 client_start ( par );
597 mem = TlsGetValue ( tls );
599 /* Connect */
600 wsa_ok ( connect ( mem->s, (struct sockaddr*) &mem->addr, sizeof ( mem->addr ) ),
601 0 ==, "simple_client (%x): connect error: %d\n" );
602 ok ( set_blocking ( mem->s, TRUE ) == 0,
603 "simple_client (%x): failed to set blocking mode\n", id );
604 trace ( "simple_client (%x) connected\n", id );
606 /* send data to server */
607 n_sent = do_synchronous_send ( mem->s, mem->send_buf, n_expected, par->buflen );
608 ok ( n_sent == n_expected,
609 "simple_client (%x): sent less data than expected: %d of %d\n", id, n_sent, n_expected );
611 /* shutdown send direction */
612 wsa_ok ( shutdown ( mem->s, SD_SEND ), 0 ==, "simple_client (%x): shutdown failed: %d\n" );
614 /* Receive data echoed back & check it */
615 n_recvd = do_synchronous_recv ( mem->s, mem->recv_buf, n_expected, par->buflen );
616 ok ( n_recvd == n_expected,
617 "simple_client (%x): received less data than expected: %d of %d\n", id, n_recvd, n_expected );
619 /* check data */
620 p = test_buffer ( mem->recv_buf, gen->chunk_size, gen->n_chunks );
621 ok ( p == NULL, "simple_client (%x): test pattern error: %d\n", id, p - mem->recv_buf);
623 /* cleanup */
624 read_zero_bytes ( mem->s );
625 trace ( "simple_client (%x) exiting\n", id );
626 client_stop ();
630 * event_client: An event-driven client
632 static void WINAPI event_client ( client_params *par )
634 test_params *gen = par->general;
635 client_memory *mem;
636 int id = GetCurrentThreadId(), n_expected = gen->n_chunks * gen->chunk_size,
637 tmp, err, n;
638 HANDLE event;
639 WSANETWORKEVENTS wsa_events;
640 char *send_last, *recv_last, *send_p, *recv_p;
641 long mask = FD_READ | FD_WRITE | FD_CLOSE;
643 trace ( "event_client (%x): starting\n", id );
644 client_start ( par );
645 trace ( "event_client (%x): server ready\n", id );
647 mem = TlsGetValue ( tls );
649 /* Prepare event notification for connect, makes socket nonblocking */
650 event = WSACreateEvent ();
651 WSAEventSelect ( mem->s, event, FD_CONNECT );
652 tmp = connect ( mem->s, (struct sockaddr*) &mem->addr, sizeof ( mem->addr ) );
653 if ( tmp != 0 ) {
654 err = WSAGetLastError ();
655 ok ( err == WSAEWOULDBLOCK, "event_client (%x): connect error: %d\n", id, err );
656 tmp = WaitForSingleObject ( event, INFINITE );
657 ok ( tmp == WAIT_OBJECT_0, "event_client (%x): wait for connect event failed: %d\n", id, tmp );
658 err = WSAEnumNetworkEvents ( mem->s, event, &wsa_events );
659 wsa_ok ( err, 0 ==, "event_client (%x): WSAEnumNetworkEvents error: %d\n" );
660 err = wsa_events.iErrorCode[ FD_CONNECT_BIT ];
661 ok ( err == 0, "event_client (%x): connect error: %d\n", id, err );
662 if ( err ) goto out;
665 trace ( "event_client (%x) connected\n", id );
667 WSAEventSelect ( mem->s, event, mask );
669 recv_p = mem->recv_buf;
670 recv_last = mem->recv_buf + n_expected;
671 send_p = mem->send_buf;
672 send_last = mem->send_buf + n_expected;
674 while ( TRUE )
676 err = WaitForSingleObject ( event, INFINITE );
677 ok ( err == WAIT_OBJECT_0, "event_client (%x): wait failed\n", id );
679 err = WSAEnumNetworkEvents ( mem->s, event, &wsa_events );
680 wsa_ok ( err, 0 ==, "event_client (%x): WSAEnumNetworkEvents error: %d\n" );
682 if ( wsa_events.lNetworkEvents & FD_WRITE )
684 err = wsa_events.iErrorCode[ FD_WRITE_BIT ];
685 ok ( err == 0, "event_client (%x): FD_WRITE error code: %d\n", id, err );
687 if ( err== 0 )
690 n = send ( mem->s, send_p, min ( send_last - send_p, par->buflen ), 0 );
691 if ( n < 0 )
693 err = WSAGetLastError ();
694 ok ( err == WSAEWOULDBLOCK, "event_client (%x): send error: %d\n", id, err );
696 else
697 send_p += n;
699 while ( n >= 0 && send_p < send_last );
701 if ( send_p == send_last )
703 trace ( "event_client (%x): all data sent - shutdown\n", id );
704 shutdown ( mem->s, SD_SEND );
705 mask &= ~FD_WRITE;
706 WSAEventSelect ( mem->s, event, mask );
709 if ( wsa_events.lNetworkEvents & FD_READ )
711 err = wsa_events.iErrorCode[ FD_READ_BIT ];
712 ok ( err == 0, "event_client (%x): FD_READ error code: %d\n", id, err );
713 if ( err != 0 ) break;
715 /* First read must succeed */
716 n = recv ( mem->s, recv_p, min ( recv_last - recv_p, par->buflen ), 0 );
717 wsa_ok ( n, 0 <=, "event_client (%x): recv error: %d\n" );
719 while ( n >= 0 ) {
720 recv_p += n;
721 if ( recv_p == recv_last )
723 mask &= ~FD_READ;
724 trace ( "event_client (%x): all data received\n", id );
725 WSAEventSelect ( mem->s, event, mask );
726 break;
728 n = recv ( mem->s, recv_p, min ( recv_last - recv_p, par->buflen ), 0 );
729 if ( n < 0 && ( err = WSAGetLastError()) != WSAEWOULDBLOCK )
730 ok ( 0, "event_client (%x): read error: %d\n", id, err );
734 if ( wsa_events.lNetworkEvents & FD_CLOSE )
736 trace ( "event_client (%x): close event\n", id );
737 err = wsa_events.iErrorCode[ FD_CLOSE_BIT ];
738 ok ( err == 0, "event_client (%x): FD_CLOSE error code: %d\n", id, err );
739 break;
743 ok ( send_p == send_last,
744 "simple_client (%x): sent less data than expected: %d of %d\n",
745 id, send_p - mem->send_buf, n_expected );
746 ok ( recv_p == recv_last,
747 "simple_client (%x): received less data than expected: %d of %d\n",
748 id, recv_p - mem->recv_buf, n_expected );
749 recv_p = test_buffer ( mem->recv_buf, gen->chunk_size, gen->n_chunks );
750 ok ( recv_p == NULL, "event_client (%x): test pattern error: %d\n", id, recv_p - mem->recv_buf);
752 out:
753 WSACloseEvent ( event );
754 trace ( "event_client (%x) exiting\n", id );
755 client_stop ();
758 /**************** Main program utility functions ***************/
760 static void Init (void)
762 WORD ver = MAKEWORD (2, 2);
763 WSADATA data;
765 ok ( WSAStartup ( ver, &data ) == 0, "WSAStartup failed\n" );
766 tls = TlsAlloc();
769 static void Exit (void)
771 INT ret, err;
772 TlsFree ( tls );
773 ret = WSACleanup();
774 err = WSAGetLastError();
775 ok ( ret == 0, "WSACleanup failed ret = %d GetLastError is %d\n", ret, err);
776 ret = WSACleanup();
777 err = WSAGetLastError();
778 ok ( ret == SOCKET_ERROR && err == WSANOTINITIALISED,
779 "WSACleanup returned %d GetLastError is %d\n", ret, err);
782 static void StartServer (LPTHREAD_START_ROUTINE routine,
783 test_params *general, server_params *par)
785 par->general = general;
786 thread[0] = CreateThread ( NULL, 0, routine, par, 0, &thread_id[0] );
787 ok ( thread[0] != NULL, "Failed to create server thread\n" );
790 static void StartClients (LPTHREAD_START_ROUTINE routine,
791 test_params *general, client_params *par)
793 int i;
794 par->general = general;
795 for ( i = 1; i <= min ( general->n_clients, MAX_CLIENTS ); i++ )
797 client_id = i - 1;
798 thread[i] = CreateThread ( NULL, 0, routine, par, 0, &thread_id[i] );
799 ok ( thread[i] != NULL, "Failed to create client thread\n" );
800 /* Make sure the client is up and running */
801 WaitForSingleObject ( client_ready[client_id], INFINITE );
805 static void do_test( test_setup *test )
807 DWORD i, n = min (test->general.n_clients, MAX_CLIENTS);
808 DWORD wait;
810 server_ready = CreateEventW ( NULL, TRUE, FALSE, NULL );
811 for (i = 0; i <= n; i++)
812 client_ready[i] = CreateEventW ( NULL, TRUE, FALSE, NULL );
814 StartServer ( test->srv, &test->general, &test->srv_params );
815 StartClients ( test->clt, &test->general, &test->clt_params );
816 WaitForSingleObject ( server_ready, INFINITE );
818 wait = WaitForMultipleObjects ( 1 + n, thread, TRUE, 1000 * TEST_TIMEOUT );
819 ok ( wait >= WAIT_OBJECT_0 && wait <= WAIT_OBJECT_0 + n ,
820 "some threads have not completed: %x\n", wait );
822 if ( ! ( wait >= WAIT_OBJECT_0 && wait <= WAIT_OBJECT_0 + n ) )
824 for (i = 0; i <= n; i++)
826 trace ("terminating thread %08x\n", thread_id[i]);
827 if ( WaitForSingleObject ( thread[i], 0 ) != WAIT_OBJECT_0 )
828 TerminateThread ( thread [i], 0 );
831 CloseHandle ( server_ready );
832 for (i = 0; i <= n; i++)
833 CloseHandle ( client_ready[i] );
836 /********* some tests for getsockopt(setsockopt(X)) == X ***********/
837 /* optname = SO_LINGER */
838 LINGER linger_testvals[] = {
839 {0,0},
840 {0,73},
841 {1,0},
842 {5,189}
845 /* optname = SO_RCVTIMEO, SOSNDTIMEO */
846 #define SOCKTIMEOUT1 63000 /* 63 seconds. Do not test fractional part because of a
847 bug in the linux kernel (fixed in 2.6.8) */
848 #define SOCKTIMEOUT2 997000 /* 997 seconds */
850 static void test_set_getsockopt(void)
852 SOCKET s;
853 int i, err;
854 int timeout;
855 LINGER lingval;
856 int size;
858 s = socket(AF_INET, SOCK_STREAM, 0);
859 ok(s!=INVALID_SOCKET, "socket() failed error: %d\n", WSAGetLastError());
860 if( s == INVALID_SOCKET) return;
861 /* SO_RCVTIMEO */
862 timeout = SOCKTIMEOUT1;
863 size = sizeof(timeout);
864 err = setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, (char *) &timeout, size);
865 if( !err)
866 err = getsockopt(s, SOL_SOCKET, SO_RCVTIMEO, (char *) &timeout, &size);
867 ok( !err, "get/setsockopt(SO_RCVTIMEO) failed error: %d\n", WSAGetLastError());
868 ok( timeout == SOCKTIMEOUT1, "getsockopt(SO_RCVTIMEO) returned wrong value %d\n", timeout);
869 /* SO_SNDTIMEO */
870 timeout = SOCKTIMEOUT2; /* 54 seconds. See remark above */
871 size = sizeof(timeout);
872 err = setsockopt(s, SOL_SOCKET, SO_SNDTIMEO, (char *) &timeout, size);
873 if( !err)
874 err = getsockopt(s, SOL_SOCKET, SO_SNDTIMEO, (char *) &timeout, &size);
875 ok( !err, "get/setsockopt(SO_SNDTIMEO) failed error: %d\n", WSAGetLastError());
876 ok( timeout == SOCKTIMEOUT2, "getsockopt(SO_SNDTIMEO) returned wrong value %d\n", timeout);
877 /* SO_LINGER */
878 for( i = 0; i < sizeof(linger_testvals)/sizeof(LINGER);i++) {
879 size = sizeof(lingval);
880 lingval = linger_testvals[i];
881 err = setsockopt(s, SOL_SOCKET, SO_LINGER, (char *) &lingval, size);
882 if( !err)
883 err = getsockopt(s, SOL_SOCKET, SO_LINGER, (char *) &lingval, &size);
884 ok( !err, "get/setsockopt(SO_LINGER) failed error: %d\n", WSAGetLastError());
885 ok( !lingval.l_onoff == !linger_testvals[i].l_onoff &&
886 (lingval.l_linger == linger_testvals[i].l_linger ||
887 (!lingval.l_linger && !linger_testvals[i].l_onoff))
888 , "getsockopt(SO_LINGER #%d) returned wrong value %d,%d not %d,%d\n", i,
889 lingval.l_onoff, lingval.l_linger,
890 linger_testvals[i].l_onoff, linger_testvals[i].l_linger);
892 closesocket(s);
895 static void test_so_reuseaddr(void)
897 struct sockaddr_in saddr;
898 SOCKET s1,s2;
899 unsigned int rc,reuse;
900 int size;
902 saddr.sin_family = AF_INET;
903 saddr.sin_port = htons(9375);
904 saddr.sin_addr.s_addr = inet_addr("127.0.0.1");
906 s1=socket(AF_INET, SOCK_STREAM, 0);
907 ok(s1!=INVALID_SOCKET, "socket() failed error: %d\n", WSAGetLastError());
908 rc = bind(s1, (struct sockaddr*)&saddr, sizeof(saddr));
909 ok(rc!=SOCKET_ERROR, "bind(s1) failed error: %d\n", WSAGetLastError());
911 s2=socket(AF_INET, SOCK_STREAM, 0);
912 ok(s2!=INVALID_SOCKET, "socket() failed error: %d\n", WSAGetLastError());
914 reuse=0x1234;
915 size=sizeof(reuse);
916 rc=getsockopt(s2, SOL_SOCKET, SO_REUSEADDR, (char*)&reuse, &size );
917 ok(rc==0 && reuse==0,"wrong result in getsockopt(SO_REUSEADDR): rc=%d reuse=%d\n",rc,reuse);
919 rc = bind(s2, (struct sockaddr*)&saddr, sizeof(saddr));
920 ok(rc==SOCKET_ERROR, "bind() succeeded\n");
922 reuse = 1;
923 rc = setsockopt(s2, SOL_SOCKET, SO_REUSEADDR, (char*)&reuse, sizeof(reuse));
924 ok(rc==0, "setsockopt() failed error: %d\n", WSAGetLastError());
926 todo_wine {
927 rc = bind(s2, (struct sockaddr*)&saddr, sizeof(saddr));
928 ok(rc==0, "bind() failed error: %d\n", WSAGetLastError());
931 closesocket(s2);
932 closesocket(s1);
935 /************* Array containing the tests to run **********/
937 #define STD_STREAM_SOCKET \
938 SOCK_STREAM, \
939 0, \
940 SERVERIP, \
941 SERVERPORT
943 static test_setup tests [NUM_TESTS] =
945 /* Test 0: synchronous client and server */
948 STD_STREAM_SOCKET,
949 2048,
953 simple_server,
955 NULL,
959 simple_client,
961 NULL,
966 /* Test 1: event-driven client, synchronous server */
969 STD_STREAM_SOCKET,
970 2048,
974 simple_server,
976 NULL,
980 event_client,
982 NULL,
983 WSA_FLAG_OVERLAPPED,
987 /* Test 2: synchronous client, non-blocking server via select() */
990 STD_STREAM_SOCKET,
991 2048,
995 select_server,
997 NULL,
1001 simple_client,
1003 NULL,
1010 static void test_UDP(void)
1012 /* This function tests UDP sendto() and recvfrom(). UDP is unreliable, so it is
1013 possible that this test fails due to dropped packets. */
1015 /* peer 0 receives data from all other peers */
1016 struct sock_info peer[NUM_UDP_PEERS];
1017 char buf[16];
1018 int ss, i, n_recv, n_sent;
1020 for ( i = NUM_UDP_PEERS - 1; i >= 0; i-- ) {
1021 ok ( ( peer[i].s = socket ( AF_INET, SOCK_DGRAM, 0 ) ) != INVALID_SOCKET, "UDP: socket failed\n" );
1023 peer[i].addr.sin_family = AF_INET;
1024 peer[i].addr.sin_addr.s_addr = inet_addr ( SERVERIP );
1026 if ( i == 0 ) {
1027 peer[i].addr.sin_port = htons ( SERVERPORT );
1028 } else {
1029 peer[i].addr.sin_port = htons ( 0 );
1032 do_bind ( peer[i].s, (struct sockaddr *) &peer[i].addr, sizeof( peer[i].addr ) );
1034 /* test getsockname() to get peer's port */
1035 ss = sizeof ( peer[i].addr );
1036 ok ( getsockname ( peer[i].s, (struct sockaddr *) &peer[i].addr, &ss ) != SOCKET_ERROR, "UDP: could not getsockname()\n" );
1037 ok ( peer[i].addr.sin_port != htons ( 0 ), "UDP: bind() did not associate port\n" );
1040 /* test getsockname() */
1041 ok ( peer[0].addr.sin_port == htons ( SERVERPORT ), "UDP: getsockname returned incorrect peer port\n" );
1043 for ( i = 1; i < NUM_UDP_PEERS; i++ ) {
1044 /* send client's ip */
1045 memcpy( buf, &peer[i].addr.sin_port, sizeof(peer[i].addr.sin_port) );
1046 n_sent = sendto ( peer[i].s, buf, sizeof(buf), 0, (struct sockaddr*) &peer[0].addr, sizeof(peer[0].addr) );
1047 ok ( n_sent == sizeof(buf), "UDP: sendto() sent wrong amount of data or socket error: %d\n", n_sent );
1050 for ( i = 1; i < NUM_UDP_PEERS; i++ ) {
1051 n_recv = recvfrom ( peer[0].s, buf, sizeof(buf), 0,(struct sockaddr *) &peer[0].peer, &ss );
1052 ok ( n_recv == sizeof(buf), "UDP: recvfrom() received wrong amount of data or socket error: %d\n", n_recv );
1053 ok ( memcmp ( &peer[0].peer.sin_port, buf, sizeof(peer[0].addr.sin_port) ) == 0, "UDP: port numbers do not match\n" );
1057 static void WINAPI do_getservbyname( HANDLE *starttest )
1059 struct {
1060 const char *name;
1061 const char *proto;
1062 int port;
1063 } serv[2] = { {"domain", "udp", 53}, {"telnet", "tcp", 23} };
1065 int i, j;
1066 struct servent *pserv[2];
1068 ok ( WaitForSingleObject ( *starttest, TEST_TIMEOUT * 1000 ) != WAIT_TIMEOUT, "test_getservbyname: timeout waiting for start signal\n");
1070 /* ensure that necessary buffer resizes are completed */
1071 for ( j = 0; j < 2; j++) {
1072 pserv[j] = getservbyname ( serv[j].name, serv[j].proto );
1075 for ( i = 0; i < NUM_QUERIES / 2; i++ ) {
1076 for ( j = 0; j < 2; j++ ) {
1077 pserv[j] = getservbyname ( serv[j].name, serv[j].proto );
1078 ok ( pserv[j] != NULL, "getservbyname could not retrieve information for %s: %d\n", serv[j].name, WSAGetLastError() );
1079 ok ( pserv[j]->s_port == htons(serv[j].port), "getservbyname returned the wrong port for %s: %d\n", serv[j].name, ntohs(pserv[j]->s_port) );
1080 ok ( !strcmp ( pserv[j]->s_proto, serv[j].proto ), "getservbyname returned the wrong protocol for %s: %s\n", serv[j].name, pserv[j]->s_proto );
1081 ok ( !strcmp ( pserv[j]->s_name, serv[j].name ), "getservbyname returned the wrong name for %s: %s\n", serv[j].name, pserv[j]->s_name );
1084 ok ( pserv[0] == pserv[1], "getservbyname: winsock resized servent buffer when not necessary\n" );
1088 static void test_getservbyname(void)
1090 int i;
1091 HANDLE starttest, thread[NUM_THREADS];
1092 DWORD thread_id[NUM_THREADS];
1094 starttest = CreateEvent ( NULL, 1, 0, "test_getservbyname_starttest" );
1096 /* create threads */
1097 for ( i = 0; i < NUM_THREADS; i++ ) {
1098 thread[i] = CreateThread ( NULL, 0, (LPTHREAD_START_ROUTINE) &do_getservbyname, &starttest, 0, &thread_id[i] );
1101 /* signal threads to start */
1102 SetEvent ( starttest );
1104 for ( i = 0; i < NUM_THREADS; i++) {
1105 WaitForSingleObject ( thread[i], TEST_TIMEOUT * 1000 );
1109 static void test_WSAAddressToStringA(void)
1111 INT ret;
1112 DWORD len;
1113 int GLE;
1114 SOCKADDR_IN sockaddr;
1115 CHAR address[22]; /* 12 digits + 3 dots + ':' + 5 digits + '\0' */
1117 CHAR expect1[] = "0.0.0.0";
1118 CHAR expect2[] = "255.255.255.255";
1119 CHAR expect3[] = "0.0.0.0:65535";
1120 CHAR expect4[] = "255.255.255.255:65535";
1122 len = 0;
1124 sockaddr.sin_family = AF_INET;
1125 sockaddr.sin_port = 0;
1126 sockaddr.sin_addr.s_addr = 0;
1128 ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1129 GLE = WSAGetLastError();
1130 ok( (ret == SOCKET_ERROR && GLE == WSAEFAULT) || (ret == 0),
1131 "WSAAddressToStringA() failed unexpectedly: WSAGetLastError()=%d, ret=%d\n",
1132 GLE, ret );
1134 len = sizeof(address);
1136 sockaddr.sin_family = AF_INET;
1137 sockaddr.sin_port = 0;
1138 sockaddr.sin_addr.s_addr = 0;
1140 ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1141 ok( !ret, "WSAAddressToStringA() failed unexpectedly: %d\n", WSAGetLastError() );
1143 ok( !strcmp( address, expect1 ), "Expected: %s, got: %s\n", expect1, address );
1145 len = sizeof(address);
1147 sockaddr.sin_family = AF_INET;
1148 sockaddr.sin_port = 0;
1149 sockaddr.sin_addr.s_addr = 0xffffffff;
1151 ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1152 ok( !ret, "WSAAddressToStringA() failed unexpectedly: %d\n", WSAGetLastError() );
1154 ok( !strcmp( address, expect2 ), "Expected: %s, got: %s\n", expect2, address );
1156 len = sizeof(address);
1158 sockaddr.sin_family = AF_INET;
1159 sockaddr.sin_port = 0xffff;
1160 sockaddr.sin_addr.s_addr = 0;
1162 ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1163 ok( !ret, "WSAAddressToStringA() failed unexpectedly: %d\n", WSAGetLastError() );
1165 ok( !strcmp( address, expect3 ), "Expected: %s, got: %s\n", expect3, address );
1167 len = sizeof(address);
1169 sockaddr.sin_family = AF_INET;
1170 sockaddr.sin_port = 0xffff;
1171 sockaddr.sin_addr.s_addr = 0xffffffff;
1173 ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1174 ok( !ret, "WSAAddressToStringA() failed unexpectedly: %d\n", WSAGetLastError() );
1176 ok( !strcmp( address, expect4 ), "Expected: %s, got: %s\n", expect4, address );
1179 static void test_WSAAddressToStringW(void)
1181 INT ret;
1182 DWORD len;
1183 int GLE;
1184 SOCKADDR_IN sockaddr;
1185 WCHAR address[22]; /* 12 digits + 3 dots + ':' + 5 digits + '\0' */
1187 WCHAR expect1[] = { '0','.','0','.','0','.','0', 0 };
1188 WCHAR expect2[] = { '2','5','5','.','2','5','5','.','2','5','5','.','2','5','5', 0 };
1189 WCHAR expect3[] = { '0','.','0','.','0','.','0', ':', '6', '5', '5', '3', '5', 0 };
1190 WCHAR expect4[] = { '2','5','5','.','2','5','5','.','2','5','5','.','2','5','5', ':',
1191 '6', '5', '5', '3', '5', 0 };
1193 len = 0;
1195 sockaddr.sin_family = AF_INET;
1196 sockaddr.sin_port = 0;
1197 sockaddr.sin_addr.s_addr = 0;
1199 ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1200 GLE = WSAGetLastError();
1201 ok( (ret == SOCKET_ERROR && GLE == WSAEFAULT) || (ret == 0),
1202 "WSAAddressToStringW() failed unexpectedly: WSAGetLastError()=%d, ret=%d\n",
1203 GLE, ret );
1205 len = sizeof(address);
1207 sockaddr.sin_family = AF_INET;
1208 sockaddr.sin_port = 0;
1209 sockaddr.sin_addr.s_addr = 0;
1211 ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1212 ok( !ret, "WSAAddressToStringW() failed unexpectedly: %d\n", WSAGetLastError() );
1214 ok( !lstrcmpW( address, expect1 ), "Expected different address string\n" );
1216 len = sizeof(address);
1218 sockaddr.sin_family = AF_INET;
1219 sockaddr.sin_port = 0;
1220 sockaddr.sin_addr.s_addr = 0xffffffff;
1222 ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1223 ok( !ret, "WSAAddressToStringW() failed unexpectedly: %d\n", WSAGetLastError() );
1225 ok( !lstrcmpW( address, expect2 ), "Expected different address string\n" );
1227 len = sizeof(address);
1229 sockaddr.sin_family = AF_INET;
1230 sockaddr.sin_port = 0xffff;
1231 sockaddr.sin_addr.s_addr = 0;
1233 ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1234 ok( !ret, "WSAAddressToStringW() failed unexpectedly: %d\n", WSAGetLastError() );
1236 ok( !lstrcmpW( address, expect3 ), "Expected different address string\n" );
1238 len = sizeof(address);
1240 sockaddr.sin_family = AF_INET;
1241 sockaddr.sin_port = 0xffff;
1242 sockaddr.sin_addr.s_addr = 0xffffffff;
1244 ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1245 ok( !ret, "WSAAddressToStringW() failed unexpectedly: %d\n", WSAGetLastError() );
1247 ok( !lstrcmpW( address, expect4 ), "Expected different address string\n" );
1250 static void test_WSAStringToAddressA(void)
1252 INT ret, len;
1253 SOCKADDR_IN sockaddr;
1254 SOCKADDR_IN6 sockaddr6;
1255 int GLE;
1257 CHAR address1[] = "0.0.0.0";
1258 CHAR address2[] = "127.127.127.127";
1259 CHAR address3[] = "255.255.255.255";
1260 CHAR address4[] = "127.127.127.127:65535";
1261 CHAR address5[] = "255.255.255.255:65535";
1262 CHAR address6[] = "::1";
1263 CHAR address7[] = "[::1]";
1264 CHAR address8[] = "[::1]:65535";
1266 len = 0;
1267 sockaddr.sin_family = AF_INET;
1269 ret = WSAStringToAddressA( address1, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1270 ok( ret == SOCKET_ERROR, "WSAStringToAddressA() succeeded unexpectedly: %d\n",
1271 WSAGetLastError() );
1273 len = sizeof(sockaddr);
1274 sockaddr.sin_port = 0;
1275 sockaddr.sin_addr.s_addr = 0;
1277 ret = WSAStringToAddressA( address1, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1278 ok( !ret && sockaddr.sin_addr.s_addr == 0,
1279 "WSAStringToAddressA() failed unexpectedly: %d\n", WSAGetLastError() );
1281 len = sizeof(sockaddr);
1282 sockaddr.sin_port = 0;
1283 sockaddr.sin_addr.s_addr = 0;
1285 ret = WSAStringToAddressA( address2, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1286 ok( !ret && sockaddr.sin_addr.s_addr == 0x7f7f7f7f,
1287 "WSAStringToAddressA() failed unexpectedly: %d\n", WSAGetLastError() );
1289 len = sizeof(sockaddr);
1290 sockaddr.sin_port = 0;
1291 sockaddr.sin_addr.s_addr = 0;
1293 ret = WSAStringToAddressA( address3, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1294 GLE = WSAGetLastError();
1295 ok( (ret == 0 && sockaddr.sin_addr.s_addr == 0xffffffff) ||
1296 (ret == SOCKET_ERROR && (GLE == ERROR_INVALID_PARAMETER || GLE == WSAEINVAL)),
1297 "WSAStringToAddressA() failed unexpectedly: %d\n", GLE );
1299 len = sizeof(sockaddr);
1300 sockaddr.sin_port = 0;
1301 sockaddr.sin_addr.s_addr = 0;
1303 ret = WSAStringToAddressA( address4, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1304 ok( !ret && sockaddr.sin_addr.s_addr == 0x7f7f7f7f && sockaddr.sin_port == 0xffff,
1305 "WSAStringToAddressA() failed unexpectedly: %d\n", WSAGetLastError() );
1307 len = sizeof(sockaddr);
1308 sockaddr.sin_port = 0;
1309 sockaddr.sin_addr.s_addr = 0;
1311 ret = WSAStringToAddressA( address5, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1312 GLE = WSAGetLastError();
1313 ok( (ret == 0 && sockaddr.sin_addr.s_addr == 0xffffffff && sockaddr.sin_port == 0xffff) ||
1314 (ret == SOCKET_ERROR && (GLE == ERROR_INVALID_PARAMETER || GLE == WSAEINVAL)),
1315 "WSAStringToAddressA() failed unexpectedly: %d\n", GLE );
1317 len = sizeof(sockaddr6);
1318 memset(&sockaddr6, 0, len);
1319 sockaddr6.sin6_family = AF_INET6;
1321 ret = WSAStringToAddressA( address6, AF_INET6, NULL, (SOCKADDR*)&sockaddr6,
1322 &len );
1323 GLE = WSAGetLastError();
1324 ok( ret == 0 || (ret == SOCKET_ERROR && GLE == WSAEINVAL),
1325 "WSAStringToAddressA() failed for IPv6 address: %d\n", GLE);
1327 len = sizeof(sockaddr6);
1328 memset(&sockaddr6, 0, len);
1329 sockaddr6.sin6_family = AF_INET6;
1331 ret = WSAStringToAddressA( address7, AF_INET6, NULL, (SOCKADDR*)&sockaddr6,
1332 &len );
1333 GLE = WSAGetLastError();
1334 ok( ret == 0 || (ret == SOCKET_ERROR && GLE == WSAEINVAL),
1335 "WSAStringToAddressA() failed for IPv6 address: %d\n", GLE);
1337 len = sizeof(sockaddr6);
1338 memset(&sockaddr6, 0, len);
1339 sockaddr6.sin6_family = AF_INET6;
1341 ret = WSAStringToAddressA( address8, AF_INET6, NULL, (SOCKADDR*)&sockaddr6,
1342 &len );
1343 GLE = WSAGetLastError();
1344 ok( (ret == 0 && sockaddr6.sin6_port == 0xffff) ||
1345 (ret == SOCKET_ERROR && GLE == WSAEINVAL),
1346 "WSAStringToAddressA() failed for IPv6 address: %d\n", GLE);
1350 static void test_WSAStringToAddressW(void)
1352 INT ret, len;
1353 SOCKADDR_IN sockaddr;
1354 SOCKADDR_IN6 sockaddr6;
1355 int GLE;
1357 WCHAR address1[] = { '0','.','0','.','0','.','0', 0 };
1358 WCHAR address2[] = { '1','2','7','.','1','2','7','.','1','2','7','.','1','2','7', 0 };
1359 WCHAR address3[] = { '2','5','5','.','2','5','5','.','2','5','5','.','2','5','5', 0 };
1360 WCHAR address4[] = { '1','2','7','.','1','2','7','.','1','2','7','.','1','2','7',
1361 ':', '6', '5', '5', '3', '5', 0 };
1362 WCHAR address5[] = { '2','5','5','.','2','5','5','.','2','5','5','.','2','5','5', ':',
1363 '6', '5', '5', '3', '5', 0 };
1364 WCHAR address6[] = {':',':','1','\0'};
1365 WCHAR address7[] = {'[',':',':','1',']','\0'};
1366 WCHAR address8[] = {'[',':',':','1',']',':','6','5','5','3','5','\0'};
1368 len = 0;
1369 sockaddr.sin_family = AF_INET;
1371 ret = WSAStringToAddressW( address1, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1372 ok( ret == SOCKET_ERROR, "WSAStringToAddressW() failed unexpectedly: %d\n",
1373 WSAGetLastError() );
1375 len = sizeof(sockaddr);
1376 sockaddr.sin_port = 0;
1377 sockaddr.sin_addr.s_addr = 0;
1379 ret = WSAStringToAddressW( address1, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1380 ok( !ret && sockaddr.sin_addr.s_addr == 0,
1381 "WSAStringToAddressW() failed unexpectedly: %d\n", WSAGetLastError() );
1383 len = sizeof(sockaddr);
1384 sockaddr.sin_port = 0;
1385 sockaddr.sin_addr.s_addr = 0;
1387 ret = WSAStringToAddressW( address2, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1388 ok( !ret && sockaddr.sin_addr.s_addr == 0x7f7f7f7f,
1389 "WSAStringToAddressW() failed unexpectedly: %d\n", WSAGetLastError() );
1391 len = sizeof(sockaddr);
1392 sockaddr.sin_port = 0;
1393 sockaddr.sin_addr.s_addr = 0;
1395 ret = WSAStringToAddressW( address3, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1396 GLE = WSAGetLastError();
1397 ok( (ret == 0 && sockaddr.sin_addr.s_addr == 0xffffffff) ||
1398 (ret == SOCKET_ERROR && (GLE == ERROR_INVALID_PARAMETER || GLE == WSAEINVAL)),
1399 "WSAStringToAddressW() failed unexpectedly: %d\n", GLE );
1401 len = sizeof(sockaddr);
1402 sockaddr.sin_port = 0;
1403 sockaddr.sin_addr.s_addr = 0;
1405 ret = WSAStringToAddressW( address4, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1406 ok( !ret && sockaddr.sin_addr.s_addr == 0x7f7f7f7f && sockaddr.sin_port == 0xffff,
1407 "WSAStringToAddressW() failed unexpectedly: %d\n", WSAGetLastError() );
1409 len = sizeof(sockaddr);
1410 sockaddr.sin_port = 0;
1411 sockaddr.sin_addr.s_addr = 0;
1413 ret = WSAStringToAddressW( address5, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1414 ok( (ret == 0 && sockaddr.sin_addr.s_addr == 0xffffffff && sockaddr.sin_port == 0xffff) ||
1415 (ret == SOCKET_ERROR && (GLE == ERROR_INVALID_PARAMETER || GLE == WSAEINVAL)),
1416 "WSAStringToAddressW() failed unexpectedly: %d\n", GLE );
1418 len = sizeof(sockaddr6);
1419 memset(&sockaddr6, 0, len);
1420 sockaddr6.sin6_family = AF_INET6;
1422 ret = WSAStringToAddressW( address6, AF_INET6, NULL, (SOCKADDR*)&sockaddr6,
1423 &len );
1424 GLE = WSAGetLastError();
1425 ok( ret == 0 || (ret == SOCKET_ERROR && GLE == WSAEINVAL),
1426 "WSAStringToAddressW() failed for IPv6 address: %d\n", GLE);
1428 len = sizeof(sockaddr6);
1429 memset(&sockaddr6, 0, len);
1430 sockaddr6.sin6_family = AF_INET6;
1432 ret = WSAStringToAddressW( address7, AF_INET6, NULL, (SOCKADDR*)&sockaddr6,
1433 &len );
1434 GLE = WSAGetLastError();
1435 ok( ret == 0 || (ret == SOCKET_ERROR && GLE == WSAEINVAL),
1436 "WSAStringToAddressW() failed for IPv6 address: %d\n", GLE);
1438 len = sizeof(sockaddr6);
1439 memset(&sockaddr6, 0, len);
1440 sockaddr6.sin6_family = AF_INET6;
1442 ret = WSAStringToAddressW( address8, AF_INET6, NULL, (SOCKADDR*)&sockaddr6,
1443 &len );
1444 GLE = WSAGetLastError();
1445 ok( (ret == 0 && sockaddr6.sin6_port == 0xffff) ||
1446 (ret == SOCKET_ERROR && GLE == WSAEINVAL),
1447 "WSAStringToAddressW() failed for IPv6 address: %d\n", GLE);
1451 static VOID WINAPI SelectReadThread(select_thread_params *par)
1453 fd_set readfds;
1454 int ret;
1455 struct sockaddr_in addr;
1456 struct timeval select_timeout;
1458 FD_ZERO(&readfds);
1459 FD_SET(par->s, &readfds);
1460 select_timeout.tv_sec=5;
1461 select_timeout.tv_usec=0;
1462 addr.sin_family = AF_INET;
1463 addr.sin_addr.s_addr = inet_addr(SERVERIP);
1464 addr.sin_port = htons(SERVERPORT);
1466 do_bind(par->s, (struct sockaddr *)&addr, sizeof(addr));
1467 wsa_ok(listen(par->s, SOMAXCONN ), 0 ==, "SelectReadThread (%x): listen failed: %d\n");
1469 SetEvent(server_ready);
1470 ret = select(par->s+1, &readfds, NULL, NULL, &select_timeout);
1471 par->ReadKilled = (ret == 1);
1474 static void test_select(void)
1476 SOCKET fdRead, fdWrite;
1477 fd_set readfds, writefds, exceptfds;
1478 unsigned int maxfd;
1479 int ret;
1480 struct timeval select_timeout;
1481 select_thread_params thread_params;
1482 HANDLE thread_handle;
1483 DWORD id;
1485 fdRead = socket(AF_INET, SOCK_STREAM, 0);
1486 ok( (fdRead != INVALID_SOCKET), "socket failed unexpectedly: %d\n", WSAGetLastError() );
1487 fdWrite = socket(AF_INET, SOCK_STREAM, 0);
1488 ok( (fdWrite != INVALID_SOCKET), "socket failed unexpectedly: %d\n", WSAGetLastError() );
1490 FD_ZERO(&readfds);
1491 FD_ZERO(&writefds);
1492 FD_ZERO(&exceptfds);
1493 FD_SET(fdRead, &readfds);
1494 FD_SET(fdWrite, &writefds);
1495 FD_SET(fdRead, &exceptfds);
1496 FD_SET(fdWrite, &exceptfds);
1497 select_timeout.tv_sec=0;
1498 select_timeout.tv_usec=500;
1500 maxfd = fdRead;
1501 if (fdWrite > maxfd)
1502 maxfd = fdWrite;
1504 todo_wine {
1505 ret = select(maxfd+1, &readfds, &writefds, &exceptfds, &select_timeout);
1506 ok ( (ret == 0), "select should not return any socket handles\n");
1507 ok ( !FD_ISSET(fdRead, &readfds), "FD should not be set\n");
1508 ok ( !FD_ISSET(fdWrite, &writefds), "FD should not be set\n");
1511 ok ( !FD_ISSET(fdRead, &exceptfds), "FD should not be set\n");
1512 ok ( !FD_ISSET(fdWrite, &exceptfds), "FD should not be set\n");
1514 todo_wine {
1515 ok ((listen(fdWrite, SOMAXCONN) == SOCKET_ERROR), "listen did not fail\n");
1517 ret = closesocket(fdWrite);
1518 ok ( (ret == 0), "closesocket failed unexpectedly: %d\n", ret);
1520 thread_params.s = fdRead;
1521 thread_params.ReadKilled = FALSE;
1522 server_ready = CreateEventW(NULL, TRUE, FALSE, NULL);
1523 thread_handle = CreateThread (NULL, 0, (LPTHREAD_START_ROUTINE) &SelectReadThread, &thread_params, 0, &id );
1524 ok ( (thread_handle != NULL), "CreateThread failed unexpectedly: %d\n", GetLastError());
1526 WaitForSingleObject (server_ready, INFINITE);
1527 Sleep(200);
1528 ret = closesocket(fdRead);
1529 ok ( (ret == 0), "closesocket failed unexpectedly: %d\n", ret);
1531 WaitForSingleObject (thread_handle, 1000);
1532 ok ( (thread_params.ReadKilled), "closesocket did not wakeup select\n");
1536 static DWORD WINAPI AcceptKillThread(select_thread_params *par)
1538 struct sockaddr_in address;
1539 int len = sizeof(address);
1540 SOCKET client_socket;
1542 SetEvent(server_ready);
1543 client_socket = accept(par->s, (struct sockaddr*) &address, &len);
1544 if (client_socket != INVALID_SOCKET)
1545 closesocket(client_socket);
1546 par->ReadKilled = (client_socket == INVALID_SOCKET);
1547 return 0;
1550 static void test_accept(void)
1552 int ret;
1553 SOCKET server_socket = INVALID_SOCKET;
1554 struct sockaddr_in address;
1555 select_thread_params thread_params;
1556 HANDLE thread_handle = NULL;
1557 DWORD id;
1559 server_socket = socket(AF_INET, SOCK_STREAM, 0);
1560 if (server_socket == INVALID_SOCKET)
1562 trace("error creating server socket: %d\n", WSAGetLastError());
1563 goto done;
1566 memset(&address, 0, sizeof(address));
1567 address.sin_family = AF_INET;
1568 ret = bind(server_socket, (struct sockaddr*) &address, sizeof(address));
1569 if (ret != 0)
1571 trace("error binding server socket: %d\n", WSAGetLastError());
1572 goto done;
1575 ret = listen(server_socket, 1);
1576 if (ret != 0)
1578 trace("error making server socket listen: %d\n", WSAGetLastError());
1579 goto done;
1582 server_ready = CreateEventW(NULL, TRUE, FALSE, NULL);
1583 if (server_ready == INVALID_HANDLE_VALUE)
1585 trace("error creating event: %d\n", GetLastError());
1586 goto done;
1589 thread_params.s = server_socket;
1590 thread_params.ReadKilled = FALSE;
1591 thread_handle = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) AcceptKillThread,
1592 &thread_params, 0, &id);
1593 if (thread_handle == NULL)
1595 trace("error creating thread: %d\n", GetLastError());
1596 goto done;
1599 WaitForSingleObject(server_ready, INFINITE);
1600 Sleep(200);
1601 ret = closesocket(server_socket);
1602 if (ret != 0)
1604 trace("closesocket failed: %d\n", WSAGetLastError());
1605 goto done;
1608 WaitForSingleObject(thread_handle, 1000);
1609 ok(thread_params.ReadKilled, "closesocket did not wakeup accept\n");
1611 done:
1612 if (thread_handle != NULL)
1613 CloseHandle(thread_handle);
1614 if (server_ready != INVALID_HANDLE_VALUE)
1615 CloseHandle(server_ready);
1616 if (server_socket != INVALID_SOCKET)
1617 closesocket(server_socket);
1620 static void test_extendedSocketOptions(void)
1622 WSADATA wsa;
1623 SOCKET sock;
1624 struct sockaddr_in sa;
1625 int sa_len = sizeof(struct sockaddr_in);
1626 int optval, optlen = sizeof(int), ret;
1627 BOOL bool_opt_val;
1628 LINGER linger_val;
1630 if(WSAStartup(MAKEWORD(2,0), &wsa)){
1631 trace("Winsock failed: %d. Aborting test\n", WSAGetLastError());
1632 return;
1635 memset(&sa, 0, sa_len);
1637 sa.sin_family = AF_INET;
1638 sa.sin_port = htons(0);
1639 sa.sin_addr.s_addr = htonl(INADDR_ANY);
1641 if((sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_IP)) < 0){
1642 trace("Creating the socket failed: %d\n", WSAGetLastError());
1643 WSACleanup();
1644 return;
1647 if(bind(sock, (struct sockaddr *) &sa, sa_len) < 0){
1648 trace("Failed to bind socket: %d\n", WSAGetLastError());
1649 closesocket(sock);
1650 WSACleanup();
1651 return;
1654 ret = getsockopt(sock, SOL_SOCKET, SO_MAX_MSG_SIZE, (char *)&optval, &optlen);
1656 ok(ret == 0, "getsockopt failed to query SO_MAX_MSG_SIZE, return value is 0x%08x\n", ret);
1657 ok(optval == 65507, "SO_MAX_MSG_SIZE reported %d, expected 65507\n", optval);
1659 optlen = sizeof(LINGER);
1660 ret = getsockopt(sock, SOL_SOCKET, SO_LINGER, (char *)&linger_val, &optlen);
1661 todo_wine{
1662 ok(ret == SOCKET_ERROR, "getsockopt should fail for UDP sockets but return value is 0x%08x\n", ret);
1665 closesocket(sock);
1667 if((sock = socket(PF_INET, SOCK_STREAM, IPPROTO_IP)) < 0){
1668 trace("Creating the socket failed: %d\n", WSAGetLastError());
1669 WSACleanup();
1670 return;
1673 if(bind(sock, (struct sockaddr *) &sa, sa_len) < 0){
1674 trace("Failed to bind socket: %d\n", WSAGetLastError());
1675 closesocket(sock);
1676 WSACleanup();
1677 return;
1680 ret = getsockopt(sock, SOL_SOCKET, SO_LINGER, (char *)&linger_val, &optlen);
1681 ok(ret == 0, "getsockopt failed to query SO_LINGER, return value is 0x%08x\n", ret);
1683 optlen = sizeof(BOOL);
1684 ret = getsockopt(sock, SOL_SOCKET, SO_DONTLINGER, (char *)&bool_opt_val, &optlen);
1685 ok(ret == 0, "getsockopt failed to query SO_DONTLINGER, return value is 0x%08x\n", ret);
1686 ok((linger_val.l_onoff && !bool_opt_val) || (!linger_val.l_onoff && bool_opt_val),
1687 "Return value of SO_DONTLINGER is %d, but SO_LINGER returned l_onoff == %d.\n",
1688 bool_opt_val, linger_val.l_onoff);
1690 closesocket(sock);
1691 WSACleanup();
1694 static void test_getsockname(void)
1696 WSADATA wsa;
1697 SOCKET sock;
1698 struct sockaddr_in sa_set, sa_get;
1699 int sa_set_len = sizeof(struct sockaddr_in);
1700 int sa_get_len = sa_set_len;
1701 static const unsigned char null_padding[] = {0,0,0,0,0,0,0,0};
1703 if(WSAStartup(MAKEWORD(2,0), &wsa)){
1704 trace("Winsock failed: %d. Aborting test\n", WSAGetLastError());
1705 return;
1708 memset(&sa_set, 0, sa_set_len);
1710 sa_set.sin_family = AF_INET;
1711 sa_set.sin_port = htons(0);
1712 sa_set.sin_addr.s_addr = htonl(INADDR_ANY);
1714 if((sock = socket(PF_INET, SOCK_STREAM, IPPROTO_IP)) < 0){
1715 trace("Creating the socket failed: %d\n", WSAGetLastError());
1716 WSACleanup();
1717 return;
1720 memcpy(&sa_get, &sa_set, sizeof(sa_set));
1721 if (getsockname(sock, (struct sockaddr*) &sa_get, &sa_get_len) == 0)
1722 ok(0, "getsockname on unbound socket should fail\n");
1723 else {
1724 ok(WSAGetLastError() == WSAEINVAL, "getsockname on unbound socket "
1725 "failed with %d, expected %d\n", WSAGetLastError(), WSAEINVAL);
1726 ok(memcmp(&sa_get, &sa_set, sizeof(sa_get)) == 0,
1727 "failed getsockname modified sockaddr when it shouldn't\n");
1730 if(bind(sock, (struct sockaddr *) &sa_set, sa_set_len) < 0){
1731 trace("Failed to bind socket: %d\n", WSAGetLastError());
1732 closesocket(sock);
1733 WSACleanup();
1734 return;
1737 if(getsockname(sock, (struct sockaddr *) &sa_get, &sa_get_len) != 0){
1738 trace("Failed to call getsockname: %d\n", WSAGetLastError());
1739 closesocket(sock);
1740 WSACleanup();
1741 return;
1744 ok(memcmp(sa_get.sin_zero, null_padding, 8) == 0,
1745 "getsockname did not zero the sockaddr_in structure\n");
1747 closesocket(sock);
1748 WSACleanup();
1751 static void test_dns(void)
1753 struct hostent *h;
1755 h = gethostbyname("");
1756 ok(h != NULL, "gethostbyname(\"\") failed with %d\n", h_errno);
1759 static void test_inet_addr(void)
1761 u_long addr;
1763 addr = inet_addr(NULL);
1764 ok(addr == INADDR_NONE, "inet_addr succeeded unexpectedly\n");
1767 static DWORD WINAPI drain_socket_thread(LPVOID arg)
1769 char buffer[1024];
1770 SOCKET sock = *(SOCKET*)arg;
1772 while (recv(sock, buffer, sizeof(buffer), 0) > 0)
1774 return 0;
1777 static void test_send(void)
1779 SOCKET src = INVALID_SOCKET;
1780 SOCKET dst = INVALID_SOCKET;
1781 HANDLE hThread = NULL;
1782 const int buflen = 1024*1024;
1783 char *buffer = NULL;
1784 int ret;
1785 DWORD id;
1787 if (tcp_socketpair(&src, &dst) != 0)
1789 ok(0, "creating socket pair failed, skipping test\n");
1790 return;
1793 hThread = CreateThread(NULL, 0, drain_socket_thread, &dst, 0, &id);
1794 if (hThread == NULL)
1796 ok(0, "CreateThread failed, error %d\n", GetLastError());
1797 goto end;
1800 buffer = HeapAlloc(GetProcessHeap(), 0, buflen);
1801 if (buffer == NULL)
1803 ok(0, "HeapAlloc failed, error %d\n", GetLastError());
1804 goto end;
1807 ret = send(src, buffer, buflen, 0);
1808 if (ret >= 0)
1809 ok(ret == buflen, "send should have sent %d bytes, but it only sent %d\n", buflen, ret);
1810 else
1811 ok(0, "send failed, error %d\n", WSAGetLastError());
1813 end:
1814 if (src != INVALID_SOCKET)
1815 closesocket(src);
1816 if (dst != INVALID_SOCKET)
1817 closesocket(dst);
1818 if (hThread != NULL)
1819 CloseHandle(hThread);
1820 HeapFree(GetProcessHeap(), 0, buffer);
1823 static void test_write_events(void)
1825 SOCKET src = INVALID_SOCKET;
1826 SOCKET dst = INVALID_SOCKET;
1827 HANDLE hThread = NULL;
1828 HANDLE hEvent = INVALID_HANDLE_VALUE;
1829 int len;
1830 u_long one = 1;
1831 int ret;
1832 DWORD id;
1834 if (tcp_socketpair(&src, &dst) != 0)
1836 ok(0, "creating socket pair failed, skipping test\n");
1837 return;
1840 hThread = CreateThread(NULL, 0, drain_socket_thread, &dst, 0, &id);
1841 if (hThread == NULL)
1843 ok(0, "CreateThread failed, error %d\n", GetLastError());
1844 goto end;
1847 hEvent = CreateEventW(NULL, FALSE, TRUE, NULL);
1848 if (hEvent == INVALID_HANDLE_VALUE)
1850 ok(0, "CreateEventW failed, error %d\n", GetLastError());
1851 goto end;
1854 ret = ioctlsocket(src, FIONBIO, &one);
1855 if (ret)
1857 ok(0, "ioctlsocket failed, error %d\n", WSAGetLastError());
1858 goto end;
1861 ret = WSAEventSelect(src, hEvent, FD_WRITE | FD_CLOSE);
1862 if (ret)
1864 ok(0, "WSAEventSelect failed, error %d\n", ret);
1865 goto end;
1868 for (len = 100; len > 0; --len)
1870 WSANETWORKEVENTS netEvents;
1871 DWORD dwRet = WaitForSingleObject(hEvent, 5000);
1872 if (dwRet != WAIT_OBJECT_0)
1874 ok(0, "WaitForSingleObject failed, error %d\n", dwRet);
1875 goto end;
1878 ret = WSAEnumNetworkEvents(src, NULL, &netEvents);
1879 if (ret)
1881 ok(0, "WSAEnumNetworkEvents failed, error %d\n", ret);
1882 goto end;
1885 if (netEvents.lNetworkEvents & FD_WRITE)
1887 ret = send(src, "a", 1, 0);
1888 if (ret < 0 && WSAGetLastError() != WSAEWOULDBLOCK)
1890 ok(0, "send failed, error %d\n", WSAGetLastError());
1891 goto end;
1895 if (netEvents.lNetworkEvents & FD_CLOSE)
1897 ok(0, "unexpected close\n");
1898 goto end;
1902 end:
1903 if (src != INVALID_SOCKET)
1904 closesocket(src);
1905 if (dst != INVALID_SOCKET)
1906 closesocket(dst);
1907 if (hThread != NULL)
1908 CloseHandle(hThread);
1909 CloseHandle(hEvent);
1912 /**************** Main program ***************/
1914 START_TEST( sock )
1916 int i;
1917 Init();
1919 test_set_getsockopt();
1920 test_so_reuseaddr();
1921 test_extendedSocketOptions();
1923 for (i = 0; i < NUM_TESTS; i++)
1925 trace ( " **** STARTING TEST %d ****\n", i );
1926 do_test ( &tests[i] );
1927 trace ( " **** TEST %d COMPLETE ****\n", i );
1930 test_UDP();
1932 test_getservbyname();
1934 test_WSAAddressToStringA();
1935 test_WSAAddressToStringW();
1937 test_WSAStringToAddressA();
1938 test_WSAStringToAddressW();
1940 test_select();
1941 test_accept();
1942 test_getsockname();
1943 test_inet_addr();
1944 test_dns();
1946 test_send();
1947 test_write_events();
1949 Exit();