(file) Return to sock.c CVS log (file) (dir) Up to [RizwankCVS] / wine4 / wine / dlls / winsock / tests

   1 cs130_tom 1.1 /*
   2                * Unit test suite for winsock functions
   3                *
   4                * Copyright 2002 Martin Wilck
   5 cs130_tom 1.3  * select() server Copyright 2005 Thomas Kho
   6 cs130_tom 1.1  *
   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.
  11                *
  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.
  16                *
  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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20                */
  21               
  22               #include <stdarg.h>
  23               
  24               #include <windef.h>
  25               #include <winbase.h>
  26               #include <winsock2.h>
  27 cs130_tom 1.1 #include <mswsock.h>
  28               #include "wine/test.h"
  29               #include <winnt.h>
  30               #include <winerror.h>
  31               
  32               #define MAX_CLIENTS 4      /* Max number of clients */
  33 cs130_tom 1.2 #define NUM_TESTS   3      /* Number of tests performed */
  34 cs130_tom 1.1 #define FIRST_CHAR 'A'     /* First character in transferred pattern */
  35               #define BIND_SLEEP 10      /* seconds to wait between attempts to bind() */
  36               #define BIND_TRIES 6       /* Number of bind() attempts */
  37               #define TEST_TIMEOUT 30    /* seconds to wait before killing child threads
  38                                             after server initialization, if something hangs */
  39               
  40               #define wsa_ok(op, cond, msg) \
  41                  do { \
  42                       int tmp, err = 0; \
  43                       tmp = op; \
  44                       if ( !(cond tmp) ) err = WSAGetLastError(); \
  45                       ok ( cond tmp, msg, GetCurrentThreadId(), err); \
  46                  } while (0);
  47               
  48               
  49               /**************** Structs and typedefs ***************/
  50               
  51               typedef struct thread_info
  52               {
  53                   HANDLE thread;
  54                   DWORD id;
  55 cs130_tom 1.1 } thread_info;
  56               
  57               /* Information in the server about open client connections */
  58               typedef struct sock_info
  59               {
  60                   SOCKET                 s;
  61                   struct sockaddr_in     addr;
  62                   struct sockaddr_in     peer;
  63                   char                  *buf;
  64 cs130_tom 1.4     int                    n_recvd;
  65 cs130_tom 1.7     int                    n_sent;
  66 cs130_tom 1.1 } sock_info;
  67               
  68               /* Test parameters for both server & client */
  69               typedef struct test_params
  70               {
  71                   int          sock_type;
  72                   int          sock_prot;
  73                   const char  *inet_addr;
  74                   short        inet_port;
  75                   int          chunk_size;
  76                   int          n_chunks;
  77                   int          n_clients;
  78               } test_params;
  79               
  80               /* server-specific test parameters */
  81               typedef struct server_params
  82               {
  83                   test_params   *general;
  84                   DWORD          sock_flags;
  85                   int            buflen;
  86               } server_params;
  87 cs130_tom 1.1 
  88               /* client-specific test parameters */
  89               typedef struct client_params
  90               {
  91                   test_params   *general;
  92                   DWORD          sock_flags;
  93                   int            buflen;
  94               } client_params;
  95               
  96               /* This type combines all information for setting up a test scenario */
  97               typedef struct test_setup
  98               {
  99                   test_params              general;
 100                   LPVOID                   srv;
 101                   server_params            srv_params;
 102                   LPVOID                   clt;
 103                   client_params            clt_params;
 104               } test_setup;
 105               
 106               /* Thread local storage for server */
 107               typedef struct server_memory
 108 cs130_tom 1.1 {
 109                   SOCKET                  s;
 110                   struct sockaddr_in      addr;
 111                   sock_info               sock[MAX_CLIENTS];
 112               } server_memory;
 113               
 114               /* Thread local storage for client */
 115               typedef struct client_memory
 116               {
 117                   SOCKET s;
 118                   struct sockaddr_in      addr;
 119                   char                   *send_buf;
 120                   char                   *recv_buf;
 121               } client_memory;
 122               
 123               /**************** Static variables ***************/
 124               
 125               static DWORD      tls;              /* Thread local storage index */
 126               static HANDLE     thread[1+MAX_CLIENTS];
 127               static DWORD      thread_id[1+MAX_CLIENTS];
 128               static HANDLE     server_ready;
 129 cs130_tom 1.1 static HANDLE     client_ready[MAX_CLIENTS];
 130               static int        client_id;
 131               
 132               /**************** General utility functions ***************/
 133               
 134               static void set_so_opentype ( BOOL overlapped )
 135               {
 136                   int optval = !overlapped, newval, len = sizeof (int);
 137               
 138                   ok ( setsockopt ( INVALID_SOCKET, SOL_SOCKET, SO_OPENTYPE,
 139                                     (LPVOID) &optval, sizeof (optval) ) == 0,
 140                        "setting SO_OPENTYPE failed\n" );
 141                   ok ( getsockopt ( INVALID_SOCKET, SOL_SOCKET, SO_OPENTYPE,
 142                                     (LPVOID) &newval, &len ) == 0,
 143                        "getting SO_OPENTYPE failed\n" );
 144                   ok ( optval == newval, "failed to set SO_OPENTYPE\n" );
 145               }
 146               
 147               static int set_blocking ( SOCKET s, BOOL blocking )
 148               {
 149                   u_long val = !blocking;
 150 cs130_tom 1.1     return ioctlsocket ( s, FIONBIO, &val );
 151               }
 152               
 153               static void fill_buffer ( char *buf, int chunk_size, int n_chunks )
 154               {
 155                   char c, *p;
 156                   for ( c = FIRST_CHAR, p = buf; c < FIRST_CHAR + n_chunks; c++, p += chunk_size )
 157                       memset ( p, c, chunk_size );
 158               }
 159               
 160               static char* test_buffer ( char *buf, int chunk_size, int n_chunks )
 161               {
 162                   char c, *p;
 163                   int i;
 164                   for ( c = FIRST_CHAR, p = buf; c < FIRST_CHAR + n_chunks; c++, p += chunk_size )
 165                   {
 166                       for ( i = 0; i < chunk_size; i++ )
 167                           if ( p[i] != c ) return p + i;
 168                   }
 169                   return NULL;
 170               }
 171 cs130_tom 1.1 
 172               /*
 173                * This routine is called when a client / server does not expect any more data,
 174                * but needs to acknowedge the closing of the connection (by reasing 0 bytes).
 175                */
 176               static void read_zero_bytes ( SOCKET s )
 177               {
 178                   char buf[256];
 179                   int tmp, n = 0;
 180                   while ( ( tmp = recv ( s, buf, 256, 0 ) ) > 0 )
 181                       n += tmp;
 182                   ok ( n <= 0, "garbage data received: %d bytes\n", n );
 183               }
 184               
 185               static int do_synchronous_send ( SOCKET s, char *buf, int buflen, int sendlen )
 186               {
 187                   char* last = buf + buflen, *p;
 188                   int n = 1;
 189                   for ( p = buf; n > 0 && p < last; p += n )
 190                       n = send ( s, p, min ( sendlen, last - p ), 0 );
 191                   wsa_ok ( n, 0 <=, "do_synchronous_send (%lx): error %d\n" );
 192 cs130_tom 1.1     return p - buf;
 193               }
 194               
 195               static int do_synchronous_recv ( SOCKET s, char *buf, int buflen, int recvlen )
 196               {
 197                   char* last = buf + buflen, *p;
 198                   int n = 1;
 199                   for ( p = buf; n > 0 && p < last; p += n )
 200                       n = recv ( s, p, min ( recvlen, last - p ), 0 );
 201                   wsa_ok ( n, 0 <=, "do_synchronous_recv (%lx): error %d:\n" );
 202                   return p - buf;
 203               }
 204               
 205               /*
 206                *  Call this routine right after thread startup.
 207                *  SO_OPENTYPE must by 0, regardless what the server did.
 208                */
 209               static void check_so_opentype (void)
 210               {
 211                   int tmp = 1, len;
 212                   len = sizeof (tmp);
 213 cs130_tom 1.1     getsockopt ( INVALID_SOCKET, SOL_SOCKET, SO_OPENTYPE, (LPVOID) &tmp, &len );
 214                   ok ( tmp == 0, "check_so_opentype: wrong startup value of SO_OPENTYPE: %d\n", tmp );
 215               }
 216               
 217               /**************** Server utility functions ***************/
 218               
 219               /*
 220                *  Even if we have closed our server socket cleanly,
 221                *  the OS may mark the address "in use" for some time -
 222                *  this happens with native Linux apps, too.
 223                */
 224               static void do_bind ( SOCKET s, struct sockaddr* addr, int addrlen )
 225               {
 226                   int err, wsaerr = 0, n_try = BIND_TRIES;
 227               
 228                   while ( ( err = bind ( s, addr, addrlen ) ) != 0 &&
 229                           ( wsaerr = WSAGetLastError () ) == WSAEADDRINUSE &&
 230                           n_try-- >= 0)
 231                   {
 232                       trace ( "address in use, waiting ...\n" );
 233                       Sleep ( 1000 * BIND_SLEEP );
 234 cs130_tom 1.1     }
 235                   ok ( err == 0, "failed to bind: %d\n", wsaerr );
 236               }
 237               
 238               static void server_start ( server_params *par )
 239               {
 240                   int i;
 241                   test_params *gen = par->general;
 242                   server_memory *mem = (LPVOID) LocalAlloc ( LPTR, sizeof (server_memory));
 243               
 244                   TlsSetValue ( tls, mem );
 245                   mem->s = WSASocketA ( AF_INET, gen->sock_type, gen->sock_prot,
 246                                         NULL, 0, par->sock_flags );
 247                   ok ( mem->s != INVALID_SOCKET, "Server: WSASocket failed\n" );
 248               
 249                   mem->addr.sin_family = AF_INET;
 250                   mem->addr.sin_addr.s_addr = inet_addr ( gen->inet_addr );
 251                   mem->addr.sin_port = htons ( gen->inet_port );
 252               
 253                   for (i = 0; i < MAX_CLIENTS; i++)
 254                   {
 255 cs130_tom 1.1         mem->sock[i].s = INVALID_SOCKET;
 256                       mem->sock[i].buf = (LPVOID) LocalAlloc ( LPTR, gen->n_chunks * gen->chunk_size );
 257 cs130_tom 1.4         mem->sock[i].n_recvd = 0;
 258 cs130_tom 1.6         mem->sock[i].n_sent = 0;
 259 cs130_tom 1.1     }
 260               
 261                   if ( gen->sock_type == SOCK_STREAM )
 262                       do_bind ( mem->s, (struct sockaddr*) &mem->addr, sizeof (mem->addr) );
 263               }
 264               
 265               static void server_stop (void)
 266               {
 267                   int i;
 268                   server_memory *mem = TlsGetValue ( tls );
 269               
 270                   for (i = 0; i < MAX_CLIENTS; i++ )
 271                   {
 272                       LocalFree ( (HANDLE) mem->sock[i].buf );
 273                       if ( mem->sock[i].s != INVALID_SOCKET )
 274                           closesocket ( mem->sock[i].s );
 275                   }
 276                   ok ( closesocket ( mem->s ) == 0, "closesocket failed\n" );
 277                   LocalFree ( (HANDLE) mem );
 278                   ExitThread ( GetCurrentThreadId () );
 279               }
 280 cs130_tom 1.1 
 281               /**************** Client utilitiy functions ***************/
 282               
 283               static void client_start ( client_params *par )
 284               {
 285                   test_params *gen = par->general;
 286                   client_memory *mem = (LPVOID) LocalAlloc (LPTR, sizeof (client_memory));
 287               
 288                   TlsSetValue ( tls, mem );
 289               
 290                   WaitForSingleObject ( server_ready, INFINITE );
 291               
 292                   mem->s = WSASocketA ( AF_INET, gen->sock_type, gen->sock_prot,
 293                                         NULL, 0, par->sock_flags );
 294               
 295                   mem->addr.sin_family = AF_INET;
 296                   mem->addr.sin_addr.s_addr = inet_addr ( gen->inet_addr );
 297                   mem->addr.sin_port = htons ( gen->inet_port );
 298               
 299                   ok ( mem->s != INVALID_SOCKET, "Client: WSASocket failed\n" );
 300               
 301 cs130_tom 1.1     mem->send_buf = (LPVOID) LocalAlloc ( LPTR, 2 * gen->n_chunks * gen->chunk_size );
 302                   mem->recv_buf = mem->send_buf + gen->n_chunks * gen->chunk_size;
 303                   fill_buffer ( mem->send_buf, gen->chunk_size, gen->n_chunks );
 304               
 305                   SetEvent ( client_ready[client_id] );
 306                   /* Wait for the other clients to come up */
 307                   WaitForMultipleObjects ( min ( gen->n_clients, MAX_CLIENTS ), client_ready, TRUE, INFINITE );
 308               }
 309               
 310               static void client_stop (void)
 311               {
 312                   client_memory *mem = TlsGetValue ( tls );
 313                   wsa_ok ( closesocket ( mem->s ), 0 ==, "closesocket error (%lx): %d\n" );
 314                   LocalFree ( (HANDLE) mem->send_buf );
 315                   LocalFree ( (HANDLE) mem );
 316                   ExitThread(0);
 317               }
 318               
 319               /**************** Servers ***************/
 320               
 321               /*
 322 cs130_tom 1.1  * simple_server: A very basic server doing synchronous IO.
 323                */
 324               static VOID WINAPI simple_server ( server_params *par )
 325               {
 326                   test_params *gen = par->general;
 327                   server_memory *mem;
 328                   int n_recvd, n_sent, n_expected = gen->n_chunks * gen->chunk_size, tmp, i,
 329                       id = GetCurrentThreadId();
 330                   char *p;
 331               
 332 cs130_tom 1.4 		trace ( "simple_server (%x) starting\n", id );
 333 cs130_tom 1.1 
 334                   set_so_opentype ( FALSE ); /* non-overlapped */
 335                   server_start ( par );
 336                   mem = TlsGetValue ( tls );
 337               
 338                   wsa_ok ( set_blocking ( mem->s, TRUE ), 0 ==, "simple_server (%lx): failed to set blocking mode: %d\n");
 339                   wsa_ok ( listen ( mem->s, SOMAXCONN ), 0 ==, "simple_server (%lx): listen failed: %d\n");
 340               
 341 cs130_tom 1.4 		trace ( "simple_server (%x) ready\n", id );
 342 cs130_tom 1.1     SetEvent ( server_ready ); /* notify clients */
 343               
 344                   for ( i = 0; i < min ( gen->n_clients, MAX_CLIENTS ); i++ )
 345                   {
 346 cs130_tom 1.4 				trace ( "simple_server (%x): waiting for client\n", id );
 347 cs130_tom 1.1 
 348                       /* accept a single connection */
 349                       tmp = sizeof ( mem->sock[0].peer );
 350                       mem->sock[0].s = accept ( mem->s, (struct sockaddr*) &mem->sock[0].peer, &tmp );
 351                       wsa_ok ( mem->sock[0].s, INVALID_SOCKET !=, "simple_server (%lx): accept failed: %d\n" );
 352               
 353                       ok ( mem->sock[0].peer.sin_addr.s_addr == inet_addr ( gen->inet_addr ),
 354                            "simple_server (%x): strange peer address\n", id );
 355               
 356                       /* Receive data & check it */
 357                       n_recvd = do_synchronous_recv ( mem->sock[0].s, mem->sock[0].buf, n_expected, par->buflen );
 358                       ok ( n_recvd == n_expected,
 359                            "simple_server (%x): received less data than expected: %d of %d\n", id, n_recvd, n_expected );
 360                       p = test_buffer ( mem->sock[0].buf, gen->chunk_size, gen->n_chunks );
 361                       ok ( p == NULL, "simple_server (%x): test pattern error: %d\n", id, p - mem->sock[0].buf);
 362               
 363                       /* Echo data back */
 364                       n_sent = do_synchronous_send ( mem->sock[0].s, mem->sock[0].buf, n_expected, par->buflen );
 365                       ok ( n_sent == n_expected,
 366                            "simple_server (%x): sent less data than expected: %d of %d\n", id, n_sent, n_expected );
 367               
 368 cs130_tom 1.1         /* cleanup */
 369                       read_zero_bytes ( mem->sock[0].s );
 370                       wsa_ok ( closesocket ( mem->sock[0].s ),  0 ==, "simple_server (%lx): closesocket error: %d\n" );
 371                       mem->sock[0].s = INVALID_SOCKET;
 372                   }
 373               
 374 cs130_tom 1.4 		trace ( "simple_server (%x) exiting\n", id );
 375 cs130_tom 1.1     server_stop ();
 376               }
 377               
 378 cs130_tom 1.2 /*
 379                * select_server: A non-blocking server.
 380                */
 381               static VOID WINAPI select_server ( server_params *par )
 382               {
 383                   test_params *gen = par->general;
 384                   server_memory *mem;
 385 cs130_tom 1.4     int n_expected = gen->n_chunks * gen->chunk_size, tmp, i,
 386                       id = GetCurrentThreadId(), n_connections = 0;
 387 cs130_tom 1.2     char *p;
 388                   struct timeval zerotime = {0,0};
 389 cs130_tom 1.3     fd_set fds_recv, fds_send, fds_openrecv, fds_opensend;
 390 cs130_tom 1.2 
 391 cs130_tom 1.5     trace ( "select_server (%x) starting\n", id );
 392 cs130_tom 1.2 
 393                   set_so_opentype ( FALSE ); /* non-overlapped */
 394                   server_start ( par );
 395                   mem = TlsGetValue ( tls );
 396               
 397                   wsa_ok ( set_blocking ( mem->s, FALSE ), 0 ==, "select_server (%lx): failed to set blocking mode: %d\n");
 398                   wsa_ok ( listen ( mem->s, SOMAXCONN ), 0 ==, "select_server (%lx): listen failed: %d\n");
 399               
 400 cs130_tom 1.5     trace ( "select_server (%x) ready\n", id );
 401 cs130_tom 1.2     SetEvent ( server_ready ); /* notify clients */
 402               
 403 cs130_tom 1.3     FD_ZERO ( &fds_openrecv );
 404                   FD_ZERO ( &fds_recv );
 405                   FD_ZERO ( &fds_send );
 406                   FD_ZERO ( &fds_opensend );
 407 cs130_tom 1.5 
 408 cs130_tom 1.3     FD_SET ( mem->s, &fds_openrecv );
 409 cs130_tom 1.2 
 410                   while(1)
 411                   {
 412 cs130_tom 1.3         fds_recv = fds_openrecv;
 413                       fds_send = fds_opensend;
 414 cs130_tom 1.2 
 415 cs130_tom 1.3         select ( 0, &fds_recv, &fds_send, NULL, &zerotime );
 416 cs130_tom 1.2 
 417                       /* check for incoming requests */
 418 cs130_tom 1.3         if ( FD_ISSET ( mem->s, &fds_recv ) ) {
 419 cs130_tom 1.2             trace ( "select_server (%x): accepting client connection\n", id );
 420               
 421                           /* accept a single connection */
 422                           tmp = sizeof ( mem->sock[i].peer );
 423                           mem->sock[i].s = accept ( mem->s, (struct sockaddr*) &mem->sock[i].peer, &tmp );
 424                           wsa_ok ( mem->sock[i].s, INVALID_SOCKET !=, "select_server (%lx): accept failed: %d\n" );
 425               
 426                           ok ( mem->sock[i].peer.sin_addr.s_addr == inet_addr ( gen->inet_addr ),
 427                               "select_server (%x): strange peer address\n", id );
 428               
 429                           /* add to list of open connections */
 430 cs130_tom 1.3             FD_SET ( mem->sock[i].s, &fds_openrecv );
 431                           FD_SET ( mem->sock[i].s, &fds_opensend );
 432 cs130_tom 1.2 
 433 cs130_tom 1.3             n_connections++;
 434 cs130_tom 1.2         }
 435               
 436                       /* handle open requests */
 437               
 438                       for ( i = 0; i < n_connections; i++ )
 439                       {
 440 cs130_tom 1.8             if ( FD_ISSET( mem->sock[i].s, &fds_recv ) ) {
 441 cs130_tom 1.2 
 442                               /* Receive data & check it */
 443 cs130_tom 1.8                 mem->sock[i].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 );
 444 cs130_tom 1.3 
 445 cs130_tom 1.4                 if ( mem->sock[i].n_recvd == n_expected ) {
 446 cs130_tom 1.3                     p = test_buffer ( mem->sock[i].buf, gen->chunk_size, gen->n_chunks );
 447 cs130_tom 1.8                     ok ( p == NULL, "select_server (%x): test pattern error: %d\n", id, p - mem->sock[i].buf );
 448 cs130_tom 1.3                     FD_CLR ( mem->sock[i].s, &fds_openrecv );
 449 cs130_tom 1.2                 }
 450               
 451 cs130_tom 1.8                 ok ( mem->sock[i].n_recvd <= n_expected, "select_server (%x): received too many bytes: %d\n", id, mem->sock[i].n_recvd );
 452                           }
 453               
 454                           if ( FD_ISSET ( mem->sock[i].s, &fds_send ) ) {
 455 cs130_tom 1.5 
 456 cs130_tom 1.2                 /* Echo data back */
 457 cs130_tom 1.8                 mem->sock[i].n_sent += send ( mem->sock[i].s, mem->sock[i].buf + mem->sock[i].n_sent, min ( n_expected - mem->sock[i].n_sent, par->buflen ), 0 );
 458 cs130_tom 1.2 
 459 cs130_tom 1.4                 if ( mem->sock[i].n_sent == n_expected ) {
 460 cs130_tom 1.3                     FD_CLR ( mem->sock[i].s, &fds_opensend );
 461                               }
 462 cs130_tom 1.8 
 463                               ok ( mem->sock[i].n_sent <= n_expected, "select_server (%x): sent too many bytes: %d\n", id, mem->sock[i].n_sent );
 464 cs130_tom 1.3             }
 465 cs130_tom 1.4         }
 466 cs130_tom 1.2 
 467 cs130_tom 1.4         /* check if all clients are done */
 468                       if ( ( fds_opensend.fd_count == 0 ) 
 469 cs130_tom 1.9             && ( fds_openrecv.fd_count == 1 ) /* initial socket that accepts clients */
 470 cs130_tom 1.4             && ( n_connections  == min ( gen->n_clients, MAX_CLIENTS ) ) ) {
 471 cs130_tom 1.5             break;
 472 cs130_tom 1.2         }
 473                   }
 474               
 475                   for ( i = 0; i < min ( gen->n_clients, MAX_CLIENTS ); i++ )
 476                   {
 477                       /* cleanup */
 478                       read_zero_bytes ( mem->sock[i].s );
 479                       wsa_ok ( closesocket ( mem->sock[i].s ),  0 ==, "select_server (%lx): closesocket error: %d\n" );
 480                       mem->sock[i].s = INVALID_SOCKET;
 481                   }
 482               
 483                   trace ( "select_server (%x) exiting\n", id );
 484                   server_stop ();
 485               }
 486               
 487 cs130_tom 1.1 /**************** Clients ***************/
 488               
 489               /*
 490                * simple_client: A very basic client doing synchronous IO.
 491                */
 492               static VOID WINAPI simple_client ( client_params *par )
 493               {
 494                   test_params *gen = par->general;
 495                   client_memory *mem;
 496                   int n_sent, n_recvd, n_expected = gen->n_chunks * gen->chunk_size, id;
 497                   char *p;
 498               
 499                   id = GetCurrentThreadId();
 500                   trace ( "simple_client (%x): starting\n", id );
 501                   /* wait here because we want to call set_so_opentype before creating a socket */
 502                   WaitForSingleObject ( server_ready, INFINITE );
 503                   trace ( "simple_client (%x): server ready\n", id );
 504               
 505                   check_so_opentype ();
 506                   set_so_opentype ( FALSE ); /* non-overlapped */
 507                   client_start ( par );
 508 cs130_tom 1.1     mem = TlsGetValue ( tls );
 509               
 510                   /* Connect */
 511                   wsa_ok ( connect ( mem->s, (struct sockaddr*) &mem->addr, sizeof ( mem->addr ) ),
 512                            0 ==, "simple_client (%lx): connect error: %d\n" );
 513                   ok ( set_blocking ( mem->s, TRUE ) == 0,
 514                        "simple_client (%x): failed to set blocking mode\n", id );
 515                   trace ( "simple_client (%x) connected\n", id );
 516               
 517                   /* send data to server */
 518                   n_sent = do_synchronous_send ( mem->s, mem->send_buf, n_expected, par->buflen );
 519                   ok ( n_sent == n_expected,
 520                        "simple_client (%x): sent less data than expected: %d of %d\n", id, n_sent, n_expected );
 521               
 522                   /* shutdown send direction */
 523                   wsa_ok ( shutdown ( mem->s, SD_SEND ), 0 ==, "simple_client (%lx): shutdown failed: %d\n" );
 524               
 525                   /* Receive data echoed back & check it */
 526                   n_recvd = do_synchronous_recv ( mem->s, mem->recv_buf, n_expected, par->buflen );
 527                   ok ( n_recvd == n_expected,
 528                        "simple_client (%x): received less data than expected: %d of %d\n", id, n_recvd, n_expected );
 529 cs130_tom 1.1 
 530                   /* check data */
 531                   p = test_buffer ( mem->recv_buf, gen->chunk_size, gen->n_chunks );
 532                   ok ( p == NULL, "simple_client (%x): test pattern error: %d\n", id, p - mem->recv_buf);
 533               
 534                   /* cleanup */
 535                   read_zero_bytes ( mem->s );
 536                   trace ( "simple_client (%x) exiting\n", id );
 537                   client_stop ();
 538               }
 539               
 540               /*
 541                * event_client: An event-driven client
 542                */
 543               static void WINAPI event_client ( client_params *par )
 544               {
 545                   test_params *gen = par->general;
 546                   client_memory *mem;
 547                   int id = GetCurrentThreadId(), n_expected = gen->n_chunks * gen->chunk_size,
 548                       tmp, err, n;
 549                   HANDLE event;
 550 cs130_tom 1.1     WSANETWORKEVENTS wsa_events;
 551                   char *send_last, *recv_last, *send_p, *recv_p;
 552                   long mask = FD_READ | FD_WRITE | FD_CLOSE;
 553               
 554                   trace ( "event_client (%x): starting\n", id );
 555                   client_start ( par );
 556                   trace ( "event_client (%x): server ready\n", id );
 557               
 558                   mem = TlsGetValue ( tls );
 559               
 560                   /* Prepare event notification for connect, makes socket nonblocking */
 561                   event = WSACreateEvent ();
 562                   WSAEventSelect ( mem->s, event, FD_CONNECT );
 563                   tmp = connect ( mem->s, (struct sockaddr*) &mem->addr, sizeof ( mem->addr ) );
 564                   if ( tmp != 0 && ( err = WSAGetLastError () ) != WSAEWOULDBLOCK )
 565                       ok ( 0, "event_client (%x): connect error: %d\n", id, err );
 566               
 567                   tmp = WaitForSingleObject ( event, INFINITE );
 568                   ok ( tmp == WAIT_OBJECT_0, "event_client (%x): wait for connect event failed: %d\n", id, tmp );
 569                   err = WSAEnumNetworkEvents ( mem->s, event, &wsa_events );
 570                   wsa_ok ( err, 0 ==, "event_client (%lx): WSAEnumNetworkEvents error: %d\n" );
 571 cs130_tom 1.1 
 572                   err = wsa_events.iErrorCode[ FD_CONNECT_BIT ];
 573                   ok ( err == 0, "event_client (%x): connect error: %d\n", id, err );
 574                   if ( err ) goto out;
 575               
 576                   trace ( "event_client (%x) connected\n", id );
 577               
 578                   WSAEventSelect ( mem->s, event, mask );
 579               
 580                   recv_p = mem->recv_buf;
 581                   recv_last = mem->recv_buf + n_expected;
 582                   send_p = mem->send_buf;
 583                   send_last = mem->send_buf + n_expected;
 584               
 585                   while ( TRUE )
 586                   {
 587                       err = WaitForSingleObject ( event, INFINITE );
 588                       ok ( err == WAIT_OBJECT_0, "event_client (%x): wait failed\n", id );
 589               
 590                       err = WSAEnumNetworkEvents ( mem->s, event, &wsa_events );
 591                       wsa_ok ( err, 0 ==, "event_client (%lx): WSAEnumNetworkEvents error: %d\n" );
 592 cs130_tom 1.1 
 593                       if ( wsa_events.lNetworkEvents & FD_WRITE )
 594                       {
 595                           err = wsa_events.iErrorCode[ FD_WRITE_BIT ];
 596                           ok ( err == 0, "event_client (%x): FD_WRITE error code: %d\n", id, err );
 597               
 598                           if ( err== 0 )
 599                               do
 600                               {
 601                                   n = send ( mem->s, send_p, min ( send_last - send_p, par->buflen ), 0 );
 602                                   if ( n < 0 )
 603                                   {
 604                                       err = WSAGetLastError ();
 605                                       ok ( err == WSAEWOULDBLOCK, "event_client (%x): send error: %d\n", id, err );
 606                                   }
 607                                   else
 608                                       send_p += n;
 609                               }
 610                               while ( n >= 0 && send_p < send_last );
 611               
 612                           if ( send_p == send_last )
 613 cs130_tom 1.1             {
 614                               trace ( "event_client (%x): all data sent - shutdown\n", id );
 615                               shutdown ( mem->s, SD_SEND );
 616                               mask &= ~FD_WRITE;
 617                               WSAEventSelect ( mem->s, event, mask );
 618                           }
 619                       }
 620                       if ( wsa_events.lNetworkEvents & FD_READ )
 621                       {
 622                           err = wsa_events.iErrorCode[ FD_READ_BIT ];
 623                           ok ( err == 0, "event_client (%x): FD_READ error code: %d\n", id, err );
 624                           if ( err != 0 ) break;
 625                           
 626                           /* First read must succeed */
 627                           n = recv ( mem->s, recv_p, min ( recv_last - recv_p, par->buflen ), 0 );
 628                           wsa_ok ( n, 0 <=, "event_client (%lx): recv error: %d\n" );
 629               
 630                           while ( n >= 0 ) {
 631                               recv_p += n;
 632                               if ( recv_p == recv_last )
 633                               {
 634 cs130_tom 1.1                     mask &= ~FD_READ;
 635                                   trace ( "event_client (%x): all data received\n", id );
 636                                   WSAEventSelect ( mem->s, event, mask );
 637                                   break;
 638                               }
 639                               n = recv ( mem->s, recv_p, min ( recv_last - recv_p, par->buflen ), 0 );
 640                               if ( n < 0 && ( err = WSAGetLastError()) != WSAEWOULDBLOCK )
 641                                   ok ( 0, "event_client (%x): read error: %d\n", id, err );
 642                               
 643                           }
 644                       }   
 645                       if ( wsa_events.lNetworkEvents & FD_CLOSE )
 646                       {
 647                           trace ( "event_client (%x): close event\n", id );
 648                           err = wsa_events.iErrorCode[ FD_CLOSE_BIT ];
 649                           ok ( err == 0, "event_client (%x): FD_CLOSE error code: %d\n", id, err );
 650                           break;
 651                       }
 652                   }
 653               
 654                   ok ( send_p == send_last,
 655 cs130_tom 1.1          "simple_client (%x): sent less data than expected: %d of %d\n",
 656                        id, send_p - mem->send_buf, n_expected );
 657                   ok ( recv_p == recv_last,
 658                        "simple_client (%x): received less data than expected: %d of %d\n",
 659                        id, recv_p - mem->recv_buf, n_expected );
 660                   recv_p = test_buffer ( mem->recv_buf, gen->chunk_size, gen->n_chunks );
 661                   ok ( recv_p == NULL, "event_client (%x): test pattern error: %d\n", id, recv_p - mem->recv_buf);
 662               
 663               out:
 664                   WSACloseEvent ( event );
 665                   trace ( "event_client (%x) exiting\n", id );
 666                   client_stop ();
 667               }
 668               
 669               /**************** Main program utility functions ***************/
 670               
 671               static void Init (void)
 672               {
 673                   WORD ver = MAKEWORD (2, 2);
 674                   WSADATA data;
 675               
 676 cs130_tom 1.1     ok ( WSAStartup ( ver, &data ) == 0, "WSAStartup failed\n" );
 677                   tls = TlsAlloc();
 678               }
 679               
 680               static void Exit (void)
 681               {
 682                   TlsFree ( tls );
 683                   ok ( WSACleanup() == 0, "WSACleanup failed\n" );
 684               }
 685               
 686               static void StartServer (LPTHREAD_START_ROUTINE routine,
 687                                        test_params *general, server_params *par)
 688               {
 689                   par->general = general;
 690                   thread[0] = CreateThread ( NULL, 0, routine, par, 0, &thread_id[0] );
 691                   ok ( thread[0] != NULL, "Failed to create server thread\n" );
 692               }
 693               
 694               static void StartClients (LPTHREAD_START_ROUTINE routine,
 695                                         test_params *general, client_params *par)
 696               {
 697 cs130_tom 1.1     int i;
 698                   par->general = general;
 699                   for ( i = 1; i <= min ( general->n_clients, MAX_CLIENTS ); i++ )
 700                   {
 701                       client_id = i - 1;
 702                       thread[i] = CreateThread ( NULL, 0, routine, par, 0, &thread_id[i] );
 703                       ok ( thread[i] != NULL, "Failed to create client thread\n" );
 704                       /* Make sure the client is up and running */
 705                       WaitForSingleObject ( client_ready[client_id], INFINITE );
 706                   };
 707               }
 708               
 709               static void do_test( test_setup *test )
 710               {
 711                   DWORD i, n = min (test->general.n_clients, MAX_CLIENTS);
 712                   DWORD wait;
 713               
 714                   server_ready = CreateEventW ( NULL, TRUE, FALSE, NULL );
 715                   for (i = 0; i <= n; i++)
 716                       client_ready[i] = CreateEventW ( NULL, TRUE, FALSE, NULL );
 717               
 718 cs130_tom 1.1     StartServer ( test->srv, &test->general, &test->srv_params );
 719                   StartClients ( test->clt, &test->general, &test->clt_params );
 720                   WaitForSingleObject ( server_ready, INFINITE );
 721               
 722                   wait = WaitForMultipleObjects ( 1 + n, thread, TRUE, 1000 * TEST_TIMEOUT );
 723                   ok ( wait >= WAIT_OBJECT_0 && wait <= WAIT_OBJECT_0 + n , 
 724                        "some threads have not completed: %lx\n", wait );
 725               
 726                   if ( ! ( wait >= WAIT_OBJECT_0 && wait <= WAIT_OBJECT_0 + n ) )
 727                   {
 728                       for (i = 0; i <= n; i++)
 729                       {
 730                           trace ("terminating thread %08lx\n", thread_id[i]);
 731                           if ( WaitForSingleObject ( thread[i], 0 ) != WAIT_OBJECT_0 )
 732                               TerminateThread ( thread [i], 0 );
 733                       }
 734                   }
 735                   CloseHandle ( server_ready );
 736                   for (i = 0; i <= n; i++)
 737                       CloseHandle ( client_ready[i] );
 738               }
 739 cs130_tom 1.1 
 740               /********* some tests for getsockopt(setsockopt(X)) == X ***********/
 741               /* optname = SO_LINGER */
 742               LINGER linger_testvals[] = {
 743                   {0,0},
 744                   {0,73}, 
 745                   {1,0},
 746                   {5,189}
 747               };
 748               
 749               /* optname = SO_RCVTIMEO, SOSNDTIMEO */
 750               #define SOCKTIMEOUT1 63000 /* 63 seconds. Do not test fractional part because of a
 751                                       bug in the linux kernel (fixed in 2.6.8) */ 
 752               #define SOCKTIMEOUT2 997000 /* 997 seconds */
 753               
 754               static void test_set_getsockopt()
 755               {
 756                   SOCKET s;
 757                   int i, err;
 758                   int timeout;
 759                   LINGER lingval;
 760 cs130_tom 1.1     int size;
 761               
 762                   s = socket(AF_INET, SOCK_STREAM, 0);
 763                   ok(s!=INVALID_SOCKET, "socket() failed error: %d\n", WSAGetLastError());
 764                   if( s == INVALID_SOCKET) return;
 765                   /* SO_RCVTIMEO */
 766                   timeout = SOCKTIMEOUT1;
 767                   size = sizeof(timeout);
 768                   err = setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, (char *) &timeout, size); 
 769                   if( !err)
 770                       err = getsockopt(s, SOL_SOCKET, SO_RCVTIMEO, (char *) &timeout, &size); 
 771                   ok( !err, "get/setsockopt(SO_RCVTIMEO) failed error: %d\n", WSAGetLastError());
 772                   ok( timeout == SOCKTIMEOUT1, "getsockopt(SO_RCVTIMEO) returned wrong value %d\n", timeout);
 773                   /* SO_SNDTIMEO */
 774                   timeout = SOCKTIMEOUT2; /* 54 seconds. See remark above */ 
 775                   size = sizeof(timeout);
 776                   err = setsockopt(s, SOL_SOCKET, SO_SNDTIMEO, (char *) &timeout, size); 
 777                   if( !err)
 778                       err = getsockopt(s, SOL_SOCKET, SO_SNDTIMEO, (char *) &timeout, &size); 
 779                   ok( !err, "get/setsockopt(SO_SNDTIMEO) failed error: %d\n", WSAGetLastError());
 780                   ok( timeout == SOCKTIMEOUT2, "getsockopt(SO_SNDTIMEO) returned wrong value %d\n", timeout);
 781 cs130_tom 1.1     /* SO_LINGER */
 782                   for( i = 0; i < sizeof(linger_testvals)/sizeof(LINGER);i++) {
 783                       size =  sizeof(lingval);
 784                       lingval = linger_testvals[i];
 785                       err = setsockopt(s, SOL_SOCKET, SO_LINGER, (char *) &lingval, size); 
 786                       if( !err)
 787                           err = getsockopt(s, SOL_SOCKET, SO_LINGER, (char *) &lingval, &size); 
 788                       ok( !err, "get/setsockopt(SO_LINGER) failed error: %d\n", WSAGetLastError());
 789                       ok( !lingval.l_onoff == !linger_testvals[i].l_onoff &&
 790                               (lingval.l_linger == linger_testvals[i].l_linger ||
 791                                (!lingval.l_linger && !linger_testvals[i].l_onoff))
 792                               , "getsockopt(SO_LINGER #%d) returned wrong value %d,%d not %d,%d \n", i, 
 793                                lingval.l_onoff, lingval.l_linger,
 794                                linger_testvals[i].l_onoff, linger_testvals[i].l_linger);
 795                   }
 796                   closesocket(s);
 797               }
 798               
 799               static void test_so_reuseaddr()
 800               {
 801                   struct sockaddr_in saddr;
 802 cs130_tom 1.1     SOCKET s1,s2;
 803                   unsigned int rc,reuse,size;
 804               
 805                   saddr.sin_family      = AF_INET;
 806                   saddr.sin_port        = htons(9375);
 807                   saddr.sin_addr.s_addr = inet_addr("127.0.0.1");
 808               
 809                   s1=socket(AF_INET, SOCK_STREAM, 0);
 810                   ok(s1!=INVALID_SOCKET, "socket() failed error: %d\n", WSAGetLastError());
 811                   rc = bind(s1, (struct sockaddr*)&saddr, sizeof(saddr));
 812                   ok(rc!=SOCKET_ERROR, "bind(s1) failed error: %d\n", WSAGetLastError());
 813               
 814                   s2=socket(AF_INET, SOCK_STREAM, 0);
 815                   ok(s2!=INVALID_SOCKET, "socket() failed error: %d\n", WSAGetLastError());
 816               
 817                   reuse=0x1234;
 818                   size=sizeof(reuse);
 819                   rc=getsockopt(s2, SOL_SOCKET, SO_REUSEADDR, (char*)&reuse, &size );
 820                   ok(rc==0 && reuse==0,"wrong result in getsockopt(SO_REUSEADDR): rc=%d reuse=%d\n",rc,reuse);
 821               
 822                   rc = bind(s2, (struct sockaddr*)&saddr, sizeof(saddr));
 823 cs130_tom 1.1     ok(rc==SOCKET_ERROR, "bind() succeeded\n");
 824               
 825                   reuse = 1;
 826                   rc = setsockopt(s2, SOL_SOCKET, SO_REUSEADDR, (char*)&reuse, sizeof(reuse));
 827                   ok(rc==0, "setsockopt() failed error: %d\n", WSAGetLastError());
 828               
 829                   todo_wine {
 830                   rc = bind(s2, (struct sockaddr*)&saddr, sizeof(saddr));
 831                   ok(rc==0, "bind() failed error: %d\n", WSAGetLastError());
 832                   }
 833               
 834                   closesocket(s2);
 835                   closesocket(s1);
 836               }
 837               
 838               /************* Array containing the tests to run **********/
 839               
 840               #define STD_STREAM_SOCKET \
 841                           SOCK_STREAM, \
 842                           0, \
 843                           "127.0.0.1", \
 844 cs130_tom 1.1             9374
 845               
 846               static test_setup tests [NUM_TESTS] =
 847               {
 848                   /* Test 0: synchronous client and server */
 849                   {
 850                       {
 851                           STD_STREAM_SOCKET,
 852                           2048,
 853                           16,
 854                           2
 855                       },
 856                       simple_server,
 857                       {
 858                           NULL,
 859                           0,
 860                           64
 861                       },
 862                       simple_client,
 863                       {
 864                           NULL,
 865 cs130_tom 1.1             0,
 866                           128
 867                       }
 868                   },
 869                   /* Test 1: event-driven client, synchronous server */
 870                   {
 871                       {
 872                           STD_STREAM_SOCKET,
 873                           2048,
 874                           16,
 875                           2
 876                       },
 877                       simple_server,
 878 cs130_tom 1.2         {
 879                           NULL,
 880                           0,
 881                           64
 882                       },
 883                       event_client,
 884                       {
 885                           NULL,
 886                           WSA_FLAG_OVERLAPPED,
 887                           128
 888                       }
 889                   },
 890 cs130_tom 1.6     /* Test 2: synchronous client, non-blocking server via select() */
 891 cs130_tom 1.2     {
 892                       {
 893                           STD_STREAM_SOCKET,
 894                           2048,
 895                           16,
 896                           2
 897                       },
 898                       select_server,
 899 cs130_tom 1.1         {
 900                           NULL,
 901                           0,
 902                           64
 903                       },
 904 cs130_tom 1.3         simple_client,
 905 cs130_tom 1.1         {
 906                           NULL,
 907 cs130_tom 1.3             0,
 908 cs130_tom 1.1             128
 909                       }
 910                   }
 911               };
 912               
 913               static void test_WSAAddressToStringA()
 914               {
 915                   INT ret;
 916                   DWORD len;
 917                   SOCKADDR_IN sockaddr;
 918                   CHAR address[22]; /* 12 digits + 3 dots + ':' + 5 digits + '\0' */
 919               
 920                   CHAR expect1[] = "0.0.0.0";
 921                   CHAR expect2[] = "255.255.255.255";
 922                   CHAR expect3[] = "0.0.0.0:65535";
 923                   CHAR expect4[] = "255.255.255.255:65535";
 924               
 925                   len = 0;
 926               
 927                   sockaddr.sin_family = AF_INET;
 928                   sockaddr.sin_port = 0;
 929 cs130_tom 1.1     sockaddr.sin_addr.s_addr = 0;
 930               
 931                   ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
 932                   ok( ret == SOCKET_ERROR, "WSAAddressToStringA() succeeded unexpectedly: %d\n", WSAGetLastError() );
 933               
 934                   len = sizeof(address);
 935               
 936                   sockaddr.sin_family = AF_INET;
 937                   sockaddr.sin_port = 0;
 938                   sockaddr.sin_addr.s_addr = 0;
 939               
 940                   ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
 941                   ok( !ret, "WSAAddressToStringA() failed unexpectedly: %d\n", WSAGetLastError() );
 942               
 943                   ok( !strcmp( address, expect1 ), "Expected: %s, got: %s\n", expect1, address );
 944               
 945                   len = sizeof(address);
 946               
 947                   sockaddr.sin_family = AF_INET;
 948                   sockaddr.sin_port = 0;
 949                   sockaddr.sin_addr.s_addr = 0xffffffff;
 950 cs130_tom 1.1 
 951                   ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
 952                   ok( !ret, "WSAAddressToStringA() failed unexpectedly: %d\n", WSAGetLastError() );
 953               
 954                   ok( !strcmp( address, expect2 ), "Expected: %s, got: %s\n", expect2, address );
 955               
 956                   len = sizeof(address);
 957               
 958                   sockaddr.sin_family = AF_INET;
 959                   sockaddr.sin_port = 0xffff;
 960                   sockaddr.sin_addr.s_addr = 0;
 961               
 962                   ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
 963                   ok( !ret, "WSAAddressToStringA() failed unexpectedly: %d\n", WSAGetLastError() );
 964               
 965                   ok( !strcmp( address, expect3 ), "Expected: %s, got: %s\n", expect3, address );
 966               
 967                   len = sizeof(address);
 968               
 969                   sockaddr.sin_family = AF_INET;
 970                   sockaddr.sin_port = 0xffff;
 971 cs130_tom 1.1     sockaddr.sin_addr.s_addr = 0xffffffff;
 972               
 973                   ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
 974                   ok( !ret, "WSAAddressToStringA() failed unexpectedly: %d\n", WSAGetLastError() );
 975               
 976                   ok( !strcmp( address, expect4 ), "Expected: %s, got: %s\n", expect4, address );
 977               }
 978               
 979               static void test_WSAAddressToStringW()
 980               {
 981                   INT ret;
 982                   DWORD len;
 983                   SOCKADDR_IN sockaddr;
 984                   WCHAR address[22]; /* 12 digits + 3 dots + ':' + 5 digits + '\0' */
 985               
 986                   WCHAR expect1[] = { '0','.','0','.','0','.','0', 0 };
 987                   WCHAR expect2[] = { '2','5','5','.','2','5','5','.','2','5','5','.','2','5','5', 0 };
 988                   WCHAR expect3[] = { '0','.','0','.','0','.','0', ':', '6', '5', '5', '3', '5', 0 };
 989                   WCHAR expect4[] = { '2','5','5','.','2','5','5','.','2','5','5','.','2','5','5', ':',
 990                                       '6', '5', '5', '3', '5', 0 };
 991               
 992 cs130_tom 1.1     len = 0;
 993               
 994                   sockaddr.sin_family = AF_INET;
 995                   sockaddr.sin_port = 0;
 996                   sockaddr.sin_addr.s_addr = 0;
 997               
 998                   ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
 999                   ok( ret == SOCKET_ERROR, "WSAAddressToStringW() succeeded unexpectedly: %x\n", WSAGetLastError() );
1000               
1001                   len = sizeof(address);
1002               
1003                   sockaddr.sin_family = AF_INET;
1004                   sockaddr.sin_port = 0;
1005                   sockaddr.sin_addr.s_addr = 0;
1006               
1007                   ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1008                   ok( !ret, "WSAAddressToStringW() failed unexpectedly: %x\n", WSAGetLastError() );
1009               
1010                   ok( !lstrcmpW( address, expect1 ), "Expected different address string\n" );
1011               
1012                   len = sizeof(address);
1013 cs130_tom 1.1 
1014                   sockaddr.sin_family = AF_INET;
1015                   sockaddr.sin_port = 0;
1016                   sockaddr.sin_addr.s_addr = 0xffffffff;
1017               
1018                   ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1019                   ok( !ret, "WSAAddressToStringW() failed unexpectedly: %x\n", WSAGetLastError() );
1020               
1021                   ok( !lstrcmpW( address, expect2 ), "Expected different address string\n" );
1022               
1023                   len = sizeof(address);
1024               
1025                   sockaddr.sin_family = AF_INET;
1026                   sockaddr.sin_port = 0xffff;
1027                   sockaddr.sin_addr.s_addr = 0;
1028               
1029                   ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1030                   ok( !ret, "WSAAddressToStringW() failed unexpectedly: %x\n", WSAGetLastError() );
1031               
1032                   ok( !lstrcmpW( address, expect3 ), "Expected different address string\n" );
1033               
1034 cs130_tom 1.1     len = sizeof(address);
1035               
1036                   sockaddr.sin_family = AF_INET;
1037                   sockaddr.sin_port = 0xffff;
1038                   sockaddr.sin_addr.s_addr = 0xffffffff;
1039               
1040                   ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1041                   ok( !ret, "WSAAddressToStringW() failed unexpectedly: %x\n", WSAGetLastError() );
1042               
1043                   ok( !lstrcmpW( address, expect4 ), "Expected different address string\n" );
1044               }
1045               
1046               static void test_WSAStringToAddressA()
1047               {
1048                   INT ret, len;
1049                   SOCKADDR_IN sockaddr;
1050               
1051                   CHAR address1[] = "0.0.0.0";
1052                   CHAR address2[] = "127.127.127.127";
1053                   CHAR address3[] = "255.255.255.255";
1054                   CHAR address4[] = "127.127.127.127:65535";
1055 cs130_tom 1.1     CHAR address5[] = "255.255.255.255:65535";
1056               
1057                   len = 0;
1058                   sockaddr.sin_family = AF_INET;
1059               
1060                   ret = WSAStringToAddressA( address1, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1061                   ok( ret == SOCKET_ERROR, "WSAStringToAddressA() succeeded unexpectedly: %x\n",
1062                       WSAGetLastError() );
1063               
1064                   len = sizeof(sockaddr);
1065                   sockaddr.sin_port = 0;
1066                   sockaddr.sin_addr.s_addr = 0;
1067               
1068                   ret = WSAStringToAddressA( address1, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1069                   ok( !ret && sockaddr.sin_addr.s_addr == 0,
1070                       "WSAStringToAddressA() failed unexpectedly: %d\n", WSAGetLastError() );
1071               
1072                   len = sizeof(sockaddr);
1073                   sockaddr.sin_port = 0;
1074                   sockaddr.sin_addr.s_addr = 0;
1075               
1076 cs130_tom 1.1     ret = WSAStringToAddressA( address2, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1077                   ok( !ret && sockaddr.sin_addr.s_addr == 0x7f7f7f7f,
1078                       "WSAStringToAddressA() failed unexpectedly: %d\n", WSAGetLastError() );
1079               
1080                   len = sizeof(sockaddr);
1081               
1082                   ret = WSAStringToAddressA( address3, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1083                   ok( ret, "WSAStringToAddressA() succeeded unexpectedly: %d\n", WSAGetLastError() );
1084               
1085                   len = sizeof(sockaddr);
1086                   sockaddr.sin_port = 0;
1087                   sockaddr.sin_addr.s_addr = 0;
1088               
1089                   ret = WSAStringToAddressA( address4, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1090                   ok( !ret && sockaddr.sin_addr.s_addr == 0x7f7f7f7f && sockaddr.sin_port == 0xffff,
1091                       "WSAStringToAddressA() failed unexpectedly: %d\n", WSAGetLastError() );
1092               
1093                   len = sizeof(sockaddr);
1094               
1095                   ret = WSAStringToAddressA( address5, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1096                   ok( ret, "WSAStringToAddressA() succeeded unexpectedly: %d\n", WSAGetLastError() );
1097 cs130_tom 1.1 }
1098               
1099               static void test_WSAStringToAddressW()
1100               {
1101                   INT ret, len;
1102                   SOCKADDR_IN sockaddr;
1103               
1104                   WCHAR address1[] = { '0','.','0','.','0','.','0', 0 };
1105                   WCHAR address2[] = { '1','2','7','.','1','2','7','.','1','2','7','.','1','2','7', 0 };
1106                   WCHAR address3[] = { '2','5','5','.','2','5','5','.','2','5','5','.','2','5','5', 0 };
1107                   WCHAR address4[] = { '1','2','7','.','1','2','7','.','1','2','7','.','1','2','7',
1108                                        ':', '6', '5', '5', '3', '5', 0 };
1109                   WCHAR address5[] = { '2','5','5','.','2','5','5','.','2','5','5','.','2','5','5', ':',
1110                                        '6', '5', '5', '3', '5', 0 };
1111               
1112                   len = 0;
1113                   sockaddr.sin_family = AF_INET;
1114               
1115                   ret = WSAStringToAddressW( address1, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1116                   ok( ret == SOCKET_ERROR, "WSAStringToAddressW() failed unexpectedly: %d\n",
1117                       WSAGetLastError() );
1118 cs130_tom 1.1 
1119                   len = sizeof(sockaddr);
1120                   sockaddr.sin_port = 0;
1121                   sockaddr.sin_addr.s_addr = 0;
1122               
1123                   ret = WSAStringToAddressW( address1, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1124                   ok( !ret && sockaddr.sin_addr.s_addr == 0,
1125                       "WSAStringToAddressW() failed unexpectedly: %d\n", WSAGetLastError() );
1126               
1127                   len = sizeof(sockaddr);
1128               
1129                   sockaddr.sin_port = 0;
1130                   sockaddr.sin_addr.s_addr = 0;
1131               
1132                   ret = WSAStringToAddressW( address2, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1133                   ok( !ret && sockaddr.sin_addr.s_addr == 0x7f7f7f7f,
1134                       "WSAStringToAddressW() failed unexpectedly: %d\n", WSAGetLastError() );
1135               
1136                   len = sizeof(sockaddr);
1137               
1138                   ret = WSAStringToAddressW( address3, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1139 cs130_tom 1.1     ok( ret, "WSAStringToAddressW() failed unexpectedly: %d\n", WSAGetLastError() );
1140               
1141                   len = sizeof(sockaddr);
1142                   sockaddr.sin_port = 0;
1143                   sockaddr.sin_addr.s_addr = 0;
1144               
1145                   ret = WSAStringToAddressW( address4, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1146                   ok( !ret && sockaddr.sin_addr.s_addr == 0x7f7f7f7f && sockaddr.sin_port == 0xffff,
1147                       "WSAStringToAddressW() failed unexpectedly: %d\n", WSAGetLastError() );
1148               
1149                   len = sizeof(sockaddr);
1150               
1151                   ret = WSAStringToAddressW( address5, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1152                   ok( ret, "WSAStringToAddressW() succeeded unexpectedly: %d\n", WSAGetLastError() );
1153               }
1154               
1155               /**************** Main program  ***************/
1156               
1157               START_TEST( sock )
1158               {
1159                   int i;
1160 cs130_tom 1.1     Init();
1161               
1162                   test_set_getsockopt();
1163                   test_so_reuseaddr();
1164               
1165                   for (i = 0; i < NUM_TESTS; i++)
1166                   {
1167                       trace ( " **** STARTING TEST %d **** \n", i );
1168                       do_test (  &tests[i] );
1169                       trace ( " **** TEST %d COMPLETE **** \n", i );
1170                   }
1171               
1172                   test_WSAAddressToStringA();
1173                   test_WSAAddressToStringW();
1174               
1175                   test_WSAStringToAddressA();
1176                   test_WSAStringToAddressW();
1177               
1178                   Exit();
1179               }

Rizwan Kassim
Powered by
ViewCVS 0.9.2