(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 cs130_tom 1.11         id = GetCurrentThreadId(), n_connections = 0, n_sent, n_recvd;
 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.10             if ( ( mem->sock[i].n_recvd < n_expected ) && FD_ISSET( mem->sock[i].s, &fds_recv ) ) {
 441 cs130_tom 1.2  
 442                                /* Receive data & check it */
 443 cs130_tom 1.11                 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                                ok ( n_recvd != SOCKET_ERROR, "select_server (%x): error in recv(): %d\n", id, WSAGetLastError() );
 445                                mem->sock[i].n_recvd += n_recvd;
 446 cs130_tom 1.3  
 447 cs130_tom 1.4                  if ( mem->sock[i].n_recvd == n_expected ) {
 448 cs130_tom 1.3                      p = test_buffer ( mem->sock[i].buf, gen->chunk_size, gen->n_chunks );
 449 cs130_tom 1.8                      ok ( p == NULL, "select_server (%x): test pattern error: %d\n", id, p - mem->sock[i].buf );
 450 cs130_tom 1.3                      FD_CLR ( mem->sock[i].s, &fds_openrecv );
 451 cs130_tom 1.2                  }
 452                
 453 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 );
 454                            }
 455                
 456 cs130_tom 1.10             if ( ( mem->sock[i].n_sent < n_expected ) && FD_ISSET ( mem->sock[i].s, &fds_send ) ) {
 457 cs130_tom 1.5  
 458 cs130_tom 1.2                  /* Echo data back */
 459 cs130_tom 1.11                 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 );
 460                                ok ( n_sent != SOCKET_ERROR, "select_server (%x): error in send(): %d\n", id, WSAGetLastError() );
 461                                mem->sock[i].n_sent += n_sent;
 462 cs130_tom 1.2  
 463 cs130_tom 1.4                  if ( mem->sock[i].n_sent == n_expected ) {
 464 cs130_tom 1.3                      FD_CLR ( mem->sock[i].s, &fds_opensend );
 465                                }
 466 cs130_tom 1.8  
 467                                ok ( mem->sock[i].n_sent <= n_expected, "select_server (%x): sent too many bytes: %d\n", id, mem->sock[i].n_sent );
 468 cs130_tom 1.3              }
 469 cs130_tom 1.4          }
 470 cs130_tom 1.2  
 471 cs130_tom 1.4          /* check if all clients are done */
 472                        if ( ( fds_opensend.fd_count == 0 ) 
 473 cs130_tom 1.9              && ( fds_openrecv.fd_count == 1 ) /* initial socket that accepts clients */
 474 cs130_tom 1.4              && ( n_connections  == min ( gen->n_clients, MAX_CLIENTS ) ) ) {
 475 cs130_tom 1.5              break;
 476 cs130_tom 1.2          }
 477                    }
 478                
 479                    for ( i = 0; i < min ( gen->n_clients, MAX_CLIENTS ); i++ )
 480                    {
 481                        /* cleanup */
 482                        read_zero_bytes ( mem->sock[i].s );
 483                        wsa_ok ( closesocket ( mem->sock[i].s ),  0 ==, "select_server (%lx): closesocket error: %d\n" );
 484                        mem->sock[i].s = INVALID_SOCKET;
 485                    }
 486                
 487                    trace ( "select_server (%x) exiting\n", id );
 488                    server_stop ();
 489                }
 490                
 491 cs130_tom 1.1  /**************** Clients ***************/
 492                
 493                /*
 494                 * simple_client: A very basic client doing synchronous IO.
 495                 */
 496                static VOID WINAPI simple_client ( client_params *par )
 497                {
 498                    test_params *gen = par->general;
 499                    client_memory *mem;
 500                    int n_sent, n_recvd, n_expected = gen->n_chunks * gen->chunk_size, id;
 501                    char *p;
 502                
 503                    id = GetCurrentThreadId();
 504                    trace ( "simple_client (%x): starting\n", id );
 505                    /* wait here because we want to call set_so_opentype before creating a socket */
 506                    WaitForSingleObject ( server_ready, INFINITE );
 507                    trace ( "simple_client (%x): server ready\n", id );
 508                
 509                    check_so_opentype ();
 510                    set_so_opentype ( FALSE ); /* non-overlapped */
 511                    client_start ( par );
 512 cs130_tom 1.1      mem = TlsGetValue ( tls );
 513                
 514                    /* Connect */
 515                    wsa_ok ( connect ( mem->s, (struct sockaddr*) &mem->addr, sizeof ( mem->addr ) ),
 516                             0 ==, "simple_client (%lx): connect error: %d\n" );
 517                    ok ( set_blocking ( mem->s, TRUE ) == 0,
 518                         "simple_client (%x): failed to set blocking mode\n", id );
 519                    trace ( "simple_client (%x) connected\n", id );
 520                
 521                    /* send data to server */
 522                    n_sent = do_synchronous_send ( mem->s, mem->send_buf, n_expected, par->buflen );
 523                    ok ( n_sent == n_expected,
 524                         "simple_client (%x): sent less data than expected: %d of %d\n", id, n_sent, n_expected );
 525                
 526                    /* shutdown send direction */
 527                    wsa_ok ( shutdown ( mem->s, SD_SEND ), 0 ==, "simple_client (%lx): shutdown failed: %d\n" );
 528                
 529                    /* Receive data echoed back & check it */
 530                    n_recvd = do_synchronous_recv ( mem->s, mem->recv_buf, n_expected, par->buflen );
 531                    ok ( n_recvd == n_expected,
 532                         "simple_client (%x): received less data than expected: %d of %d\n", id, n_recvd, n_expected );
 533 cs130_tom 1.1  
 534                    /* check data */
 535                    p = test_buffer ( mem->recv_buf, gen->chunk_size, gen->n_chunks );
 536                    ok ( p == NULL, "simple_client (%x): test pattern error: %d\n", id, p - mem->recv_buf);
 537                
 538                    /* cleanup */
 539                    read_zero_bytes ( mem->s );
 540                    trace ( "simple_client (%x) exiting\n", id );
 541                    client_stop ();
 542                }
 543                
 544                /*
 545                 * event_client: An event-driven client
 546                 */
 547                static void WINAPI event_client ( client_params *par )
 548                {
 549                    test_params *gen = par->general;
 550                    client_memory *mem;
 551                    int id = GetCurrentThreadId(), n_expected = gen->n_chunks * gen->chunk_size,
 552                        tmp, err, n;
 553                    HANDLE event;
 554 cs130_tom 1.1      WSANETWORKEVENTS wsa_events;
 555                    char *send_last, *recv_last, *send_p, *recv_p;
 556                    long mask = FD_READ | FD_WRITE | FD_CLOSE;
 557                
 558                    trace ( "event_client (%x): starting\n", id );
 559                    client_start ( par );
 560                    trace ( "event_client (%x): server ready\n", id );
 561                
 562                    mem = TlsGetValue ( tls );
 563                
 564                    /* Prepare event notification for connect, makes socket nonblocking */
 565                    event = WSACreateEvent ();
 566                    WSAEventSelect ( mem->s, event, FD_CONNECT );
 567                    tmp = connect ( mem->s, (struct sockaddr*) &mem->addr, sizeof ( mem->addr ) );
 568                    if ( tmp != 0 && ( err = WSAGetLastError () ) != WSAEWOULDBLOCK )
 569                        ok ( 0, "event_client (%x): connect error: %d\n", id, err );
 570                
 571                    tmp = WaitForSingleObject ( event, INFINITE );
 572                    ok ( tmp == WAIT_OBJECT_0, "event_client (%x): wait for connect event failed: %d\n", id, tmp );
 573                    err = WSAEnumNetworkEvents ( mem->s, event, &wsa_events );
 574                    wsa_ok ( err, 0 ==, "event_client (%lx): WSAEnumNetworkEvents error: %d\n" );
 575 cs130_tom 1.1  
 576                    err = wsa_events.iErrorCode[ FD_CONNECT_BIT ];
 577                    ok ( err == 0, "event_client (%x): connect error: %d\n", id, err );
 578                    if ( err ) goto out;
 579                
 580                    trace ( "event_client (%x) connected\n", id );
 581                
 582                    WSAEventSelect ( mem->s, event, mask );
 583                
 584                    recv_p = mem->recv_buf;
 585                    recv_last = mem->recv_buf + n_expected;
 586                    send_p = mem->send_buf;
 587                    send_last = mem->send_buf + n_expected;
 588                
 589                    while ( TRUE )
 590                    {
 591                        err = WaitForSingleObject ( event, INFINITE );
 592                        ok ( err == WAIT_OBJECT_0, "event_client (%x): wait failed\n", id );
 593                
 594                        err = WSAEnumNetworkEvents ( mem->s, event, &wsa_events );
 595                        wsa_ok ( err, 0 ==, "event_client (%lx): WSAEnumNetworkEvents error: %d\n" );
 596 cs130_tom 1.1  
 597                        if ( wsa_events.lNetworkEvents & FD_WRITE )
 598                        {
 599                            err = wsa_events.iErrorCode[ FD_WRITE_BIT ];
 600                            ok ( err == 0, "event_client (%x): FD_WRITE error code: %d\n", id, err );
 601                
 602                            if ( err== 0 )
 603                                do
 604                                {
 605                                    n = send ( mem->s, send_p, min ( send_last - send_p, par->buflen ), 0 );
 606                                    if ( n < 0 )
 607                                    {
 608                                        err = WSAGetLastError ();
 609                                        ok ( err == WSAEWOULDBLOCK, "event_client (%x): send error: %d\n", id, err );
 610                                    }
 611                                    else
 612                                        send_p += n;
 613                                }
 614                                while ( n >= 0 && send_p < send_last );
 615                
 616                            if ( send_p == send_last )
 617 cs130_tom 1.1              {
 618                                trace ( "event_client (%x): all data sent - shutdown\n", id );
 619                                shutdown ( mem->s, SD_SEND );
 620                                mask &= ~FD_WRITE;
 621                                WSAEventSelect ( mem->s, event, mask );
 622                            }
 623                        }
 624                        if ( wsa_events.lNetworkEvents & FD_READ )
 625                        {
 626                            err = wsa_events.iErrorCode[ FD_READ_BIT ];
 627                            ok ( err == 0, "event_client (%x): FD_READ error code: %d\n", id, err );
 628                            if ( err != 0 ) break;
 629                            
 630                            /* First read must succeed */
 631                            n = recv ( mem->s, recv_p, min ( recv_last - recv_p, par->buflen ), 0 );
 632                            wsa_ok ( n, 0 <=, "event_client (%lx): recv error: %d\n" );
 633                
 634                            while ( n >= 0 ) {
 635                                recv_p += n;
 636                                if ( recv_p == recv_last )
 637                                {
 638 cs130_tom 1.1                      mask &= ~FD_READ;
 639                                    trace ( "event_client (%x): all data received\n", id );
 640                                    WSAEventSelect ( mem->s, event, mask );
 641                                    break;
 642                                }
 643                                n = recv ( mem->s, recv_p, min ( recv_last - recv_p, par->buflen ), 0 );
 644                                if ( n < 0 && ( err = WSAGetLastError()) != WSAEWOULDBLOCK )
 645                                    ok ( 0, "event_client (%x): read error: %d\n", id, err );
 646                                
 647                            }
 648                        }   
 649                        if ( wsa_events.lNetworkEvents & FD_CLOSE )
 650                        {
 651                            trace ( "event_client (%x): close event\n", id );
 652                            err = wsa_events.iErrorCode[ FD_CLOSE_BIT ];
 653                            ok ( err == 0, "event_client (%x): FD_CLOSE error code: %d\n", id, err );
 654                            break;
 655                        }
 656                    }
 657                
 658                    ok ( send_p == send_last,
 659 cs130_tom 1.1           "simple_client (%x): sent less data than expected: %d of %d\n",
 660                         id, send_p - mem->send_buf, n_expected );
 661                    ok ( recv_p == recv_last,
 662                         "simple_client (%x): received less data than expected: %d of %d\n",
 663                         id, recv_p - mem->recv_buf, n_expected );
 664                    recv_p = test_buffer ( mem->recv_buf, gen->chunk_size, gen->n_chunks );
 665                    ok ( recv_p == NULL, "event_client (%x): test pattern error: %d\n", id, recv_p - mem->recv_buf);
 666                
 667                out:
 668                    WSACloseEvent ( event );
 669                    trace ( "event_client (%x) exiting\n", id );
 670                    client_stop ();
 671                }
 672                
 673                /**************** Main program utility functions ***************/
 674                
 675                static void Init (void)
 676                {
 677                    WORD ver = MAKEWORD (2, 2);
 678                    WSADATA data;
 679                
 680 cs130_tom 1.1      ok ( WSAStartup ( ver, &data ) == 0, "WSAStartup failed\n" );
 681                    tls = TlsAlloc();
 682                }
 683                
 684                static void Exit (void)
 685                {
 686                    TlsFree ( tls );
 687                    ok ( WSACleanup() == 0, "WSACleanup failed\n" );
 688                }
 689                
 690                static void StartServer (LPTHREAD_START_ROUTINE routine,
 691                                         test_params *general, server_params *par)
 692                {
 693                    par->general = general;
 694                    thread[0] = CreateThread ( NULL, 0, routine, par, 0, &thread_id[0] );
 695                    ok ( thread[0] != NULL, "Failed to create server thread\n" );
 696                }
 697                
 698                static void StartClients (LPTHREAD_START_ROUTINE routine,
 699                                          test_params *general, client_params *par)
 700                {
 701 cs130_tom 1.1      int i;
 702                    par->general = general;
 703                    for ( i = 1; i <= min ( general->n_clients, MAX_CLIENTS ); i++ )
 704                    {
 705                        client_id = i - 1;
 706                        thread[i] = CreateThread ( NULL, 0, routine, par, 0, &thread_id[i] );
 707                        ok ( thread[i] != NULL, "Failed to create client thread\n" );
 708                        /* Make sure the client is up and running */
 709                        WaitForSingleObject ( client_ready[client_id], INFINITE );
 710                    };
 711                }
 712                
 713                static void do_test( test_setup *test )
 714                {
 715                    DWORD i, n = min (test->general.n_clients, MAX_CLIENTS);
 716                    DWORD wait;
 717                
 718                    server_ready = CreateEventW ( NULL, TRUE, FALSE, NULL );
 719                    for (i = 0; i <= n; i++)
 720                        client_ready[i] = CreateEventW ( NULL, TRUE, FALSE, NULL );
 721                
 722 cs130_tom 1.1      StartServer ( test->srv, &test->general, &test->srv_params );
 723                    StartClients ( test->clt, &test->general, &test->clt_params );
 724                    WaitForSingleObject ( server_ready, INFINITE );
 725                
 726                    wait = WaitForMultipleObjects ( 1 + n, thread, TRUE, 1000 * TEST_TIMEOUT );
 727                    ok ( wait >= WAIT_OBJECT_0 && wait <= WAIT_OBJECT_0 + n , 
 728                         "some threads have not completed: %lx\n", wait );
 729                
 730                    if ( ! ( wait >= WAIT_OBJECT_0 && wait <= WAIT_OBJECT_0 + n ) )
 731                    {
 732                        for (i = 0; i <= n; i++)
 733                        {
 734                            trace ("terminating thread %08lx\n", thread_id[i]);
 735                            if ( WaitForSingleObject ( thread[i], 0 ) != WAIT_OBJECT_0 )
 736                                TerminateThread ( thread [i], 0 );
 737                        }
 738                    }
 739                    CloseHandle ( server_ready );
 740                    for (i = 0; i <= n; i++)
 741                        CloseHandle ( client_ready[i] );
 742                }
 743 cs130_tom 1.1  
 744                /********* some tests for getsockopt(setsockopt(X)) == X ***********/
 745                /* optname = SO_LINGER */
 746                LINGER linger_testvals[] = {
 747                    {0,0},
 748                    {0,73}, 
 749                    {1,0},
 750                    {5,189}
 751                };
 752                
 753                /* optname = SO_RCVTIMEO, SOSNDTIMEO */
 754                #define SOCKTIMEOUT1 63000 /* 63 seconds. Do not test fractional part because of a
 755                                        bug in the linux kernel (fixed in 2.6.8) */ 
 756                #define SOCKTIMEOUT2 997000 /* 997 seconds */
 757                
 758                static void test_set_getsockopt()
 759                {
 760                    SOCKET s;
 761                    int i, err;
 762                    int timeout;
 763                    LINGER lingval;
 764 cs130_tom 1.1      int size;
 765                
 766                    s = socket(AF_INET, SOCK_STREAM, 0);
 767                    ok(s!=INVALID_SOCKET, "socket() failed error: %d\n", WSAGetLastError());
 768                    if( s == INVALID_SOCKET) return;
 769                    /* SO_RCVTIMEO */
 770                    timeout = SOCKTIMEOUT1;
 771                    size = sizeof(timeout);
 772                    err = setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, (char *) &timeout, size); 
 773                    if( !err)
 774                        err = getsockopt(s, SOL_SOCKET, SO_RCVTIMEO, (char *) &timeout, &size); 
 775                    ok( !err, "get/setsockopt(SO_RCVTIMEO) failed error: %d\n", WSAGetLastError());
 776                    ok( timeout == SOCKTIMEOUT1, "getsockopt(SO_RCVTIMEO) returned wrong value %d\n", timeout);
 777                    /* SO_SNDTIMEO */
 778                    timeout = SOCKTIMEOUT2; /* 54 seconds. See remark above */ 
 779                    size = sizeof(timeout);
 780                    err = setsockopt(s, SOL_SOCKET, SO_SNDTIMEO, (char *) &timeout, size); 
 781                    if( !err)
 782                        err = getsockopt(s, SOL_SOCKET, SO_SNDTIMEO, (char *) &timeout, &size); 
 783                    ok( !err, "get/setsockopt(SO_SNDTIMEO) failed error: %d\n", WSAGetLastError());
 784                    ok( timeout == SOCKTIMEOUT2, "getsockopt(SO_SNDTIMEO) returned wrong value %d\n", timeout);
 785 cs130_tom 1.1      /* SO_LINGER */
 786                    for( i = 0; i < sizeof(linger_testvals)/sizeof(LINGER);i++) {
 787                        size =  sizeof(lingval);
 788                        lingval = linger_testvals[i];
 789                        err = setsockopt(s, SOL_SOCKET, SO_LINGER, (char *) &lingval, size); 
 790                        if( !err)
 791                            err = getsockopt(s, SOL_SOCKET, SO_LINGER, (char *) &lingval, &size); 
 792                        ok( !err, "get/setsockopt(SO_LINGER) failed error: %d\n", WSAGetLastError());
 793                        ok( !lingval.l_onoff == !linger_testvals[i].l_onoff &&
 794                                (lingval.l_linger == linger_testvals[i].l_linger ||
 795                                 (!lingval.l_linger && !linger_testvals[i].l_onoff))
 796                                , "getsockopt(SO_LINGER #%d) returned wrong value %d,%d not %d,%d \n", i, 
 797                                 lingval.l_onoff, lingval.l_linger,
 798                                 linger_testvals[i].l_onoff, linger_testvals[i].l_linger);
 799                    }
 800                    closesocket(s);
 801                }
 802                
 803                static void test_so_reuseaddr()
 804                {
 805                    struct sockaddr_in saddr;
 806 cs130_tom 1.1      SOCKET s1,s2;
 807                    unsigned int rc,reuse,size;
 808                
 809                    saddr.sin_family      = AF_INET;
 810                    saddr.sin_port        = htons(9375);
 811                    saddr.sin_addr.s_addr = inet_addr("127.0.0.1");
 812                
 813                    s1=socket(AF_INET, SOCK_STREAM, 0);
 814                    ok(s1!=INVALID_SOCKET, "socket() failed error: %d\n", WSAGetLastError());
 815                    rc = bind(s1, (struct sockaddr*)&saddr, sizeof(saddr));
 816                    ok(rc!=SOCKET_ERROR, "bind(s1) failed error: %d\n", WSAGetLastError());
 817                
 818                    s2=socket(AF_INET, SOCK_STREAM, 0);
 819                    ok(s2!=INVALID_SOCKET, "socket() failed error: %d\n", WSAGetLastError());
 820                
 821                    reuse=0x1234;
 822                    size=sizeof(reuse);
 823                    rc=getsockopt(s2, SOL_SOCKET, SO_REUSEADDR, (char*)&reuse, &size );
 824                    ok(rc==0 && reuse==0,"wrong result in getsockopt(SO_REUSEADDR): rc=%d reuse=%d\n",rc,reuse);
 825                
 826                    rc = bind(s2, (struct sockaddr*)&saddr, sizeof(saddr));
 827 cs130_tom 1.1      ok(rc==SOCKET_ERROR, "bind() succeeded\n");
 828                
 829                    reuse = 1;
 830                    rc = setsockopt(s2, SOL_SOCKET, SO_REUSEADDR, (char*)&reuse, sizeof(reuse));
 831                    ok(rc==0, "setsockopt() failed error: %d\n", WSAGetLastError());
 832                
 833                    todo_wine {
 834                    rc = bind(s2, (struct sockaddr*)&saddr, sizeof(saddr));
 835                    ok(rc==0, "bind() failed error: %d\n", WSAGetLastError());
 836                    }
 837                
 838                    closesocket(s2);
 839                    closesocket(s1);
 840                }
 841                
 842                /************* Array containing the tests to run **********/
 843                
 844                #define STD_STREAM_SOCKET \
 845                            SOCK_STREAM, \
 846                            0, \
 847                            "127.0.0.1", \
 848 cs130_tom 1.1              9374
 849                
 850                static test_setup tests [NUM_TESTS] =
 851                {
 852                    /* Test 0: synchronous client and server */
 853                    {
 854                        {
 855                            STD_STREAM_SOCKET,
 856                            2048,
 857                            16,
 858                            2
 859                        },
 860                        simple_server,
 861                        {
 862                            NULL,
 863                            0,
 864                            64
 865                        },
 866                        simple_client,
 867                        {
 868                            NULL,
 869 cs130_tom 1.1              0,
 870                            128
 871                        }
 872                    },
 873                    /* Test 1: event-driven client, synchronous server */
 874                    {
 875                        {
 876                            STD_STREAM_SOCKET,
 877                            2048,
 878                            16,
 879                            2
 880                        },
 881                        simple_server,
 882 cs130_tom 1.2          {
 883                            NULL,
 884                            0,
 885                            64
 886                        },
 887                        event_client,
 888                        {
 889                            NULL,
 890                            WSA_FLAG_OVERLAPPED,
 891                            128
 892                        }
 893                    },
 894 cs130_tom 1.6      /* Test 2: synchronous client, non-blocking server via select() */
 895 cs130_tom 1.2      {
 896                        {
 897                            STD_STREAM_SOCKET,
 898                            2048,
 899                            16,
 900                            2
 901                        },
 902                        select_server,
 903 cs130_tom 1.1          {
 904                            NULL,
 905                            0,
 906                            64
 907                        },
 908 cs130_tom 1.3          simple_client,
 909 cs130_tom 1.1          {
 910                            NULL,
 911 cs130_tom 1.3              0,
 912 cs130_tom 1.1              128
 913                        }
 914                    }
 915                };
 916                
 917                static void test_WSAAddressToStringA()
 918                {
 919                    INT ret;
 920                    DWORD len;
 921                    SOCKADDR_IN sockaddr;
 922                    CHAR address[22]; /* 12 digits + 3 dots + ':' + 5 digits + '\0' */
 923                
 924                    CHAR expect1[] = "0.0.0.0";
 925                    CHAR expect2[] = "255.255.255.255";
 926                    CHAR expect3[] = "0.0.0.0:65535";
 927                    CHAR expect4[] = "255.255.255.255:65535";
 928                
 929                    len = 0;
 930                
 931                    sockaddr.sin_family = AF_INET;
 932                    sockaddr.sin_port = 0;
 933 cs130_tom 1.1      sockaddr.sin_addr.s_addr = 0;
 934                
 935                    ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
 936                    ok( ret == SOCKET_ERROR, "WSAAddressToStringA() succeeded unexpectedly: %d\n", WSAGetLastError() );
 937                
 938                    len = sizeof(address);
 939                
 940                    sockaddr.sin_family = AF_INET;
 941                    sockaddr.sin_port = 0;
 942                    sockaddr.sin_addr.s_addr = 0;
 943                
 944                    ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
 945                    ok( !ret, "WSAAddressToStringA() failed unexpectedly: %d\n", WSAGetLastError() );
 946                
 947                    ok( !strcmp( address, expect1 ), "Expected: %s, got: %s\n", expect1, address );
 948                
 949                    len = sizeof(address);
 950                
 951                    sockaddr.sin_family = AF_INET;
 952                    sockaddr.sin_port = 0;
 953                    sockaddr.sin_addr.s_addr = 0xffffffff;
 954 cs130_tom 1.1  
 955                    ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
 956                    ok( !ret, "WSAAddressToStringA() failed unexpectedly: %d\n", WSAGetLastError() );
 957                
 958                    ok( !strcmp( address, expect2 ), "Expected: %s, got: %s\n", expect2, address );
 959                
 960                    len = sizeof(address);
 961                
 962                    sockaddr.sin_family = AF_INET;
 963                    sockaddr.sin_port = 0xffff;
 964                    sockaddr.sin_addr.s_addr = 0;
 965                
 966                    ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
 967                    ok( !ret, "WSAAddressToStringA() failed unexpectedly: %d\n", WSAGetLastError() );
 968                
 969                    ok( !strcmp( address, expect3 ), "Expected: %s, got: %s\n", expect3, address );
 970                
 971                    len = sizeof(address);
 972                
 973                    sockaddr.sin_family = AF_INET;
 974                    sockaddr.sin_port = 0xffff;
 975 cs130_tom 1.1      sockaddr.sin_addr.s_addr = 0xffffffff;
 976                
 977                    ret = WSAAddressToStringA( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
 978                    ok( !ret, "WSAAddressToStringA() failed unexpectedly: %d\n", WSAGetLastError() );
 979                
 980                    ok( !strcmp( address, expect4 ), "Expected: %s, got: %s\n", expect4, address );
 981                }
 982                
 983                static void test_WSAAddressToStringW()
 984                {
 985                    INT ret;
 986                    DWORD len;
 987                    SOCKADDR_IN sockaddr;
 988                    WCHAR address[22]; /* 12 digits + 3 dots + ':' + 5 digits + '\0' */
 989                
 990                    WCHAR expect1[] = { '0','.','0','.','0','.','0', 0 };
 991                    WCHAR expect2[] = { '2','5','5','.','2','5','5','.','2','5','5','.','2','5','5', 0 };
 992                    WCHAR expect3[] = { '0','.','0','.','0','.','0', ':', '6', '5', '5', '3', '5', 0 };
 993                    WCHAR expect4[] = { '2','5','5','.','2','5','5','.','2','5','5','.','2','5','5', ':',
 994                                        '6', '5', '5', '3', '5', 0 };
 995                
 996 cs130_tom 1.1      len = 0;
 997                
 998                    sockaddr.sin_family = AF_INET;
 999                    sockaddr.sin_port = 0;
1000                    sockaddr.sin_addr.s_addr = 0;
1001                
1002                    ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1003                    ok( ret == SOCKET_ERROR, "WSAAddressToStringW() succeeded unexpectedly: %x\n", WSAGetLastError() );
1004                
1005                    len = sizeof(address);
1006                
1007                    sockaddr.sin_family = AF_INET;
1008                    sockaddr.sin_port = 0;
1009                    sockaddr.sin_addr.s_addr = 0;
1010                
1011                    ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1012                    ok( !ret, "WSAAddressToStringW() failed unexpectedly: %x\n", WSAGetLastError() );
1013                
1014                    ok( !lstrcmpW( address, expect1 ), "Expected different address string\n" );
1015                
1016                    len = sizeof(address);
1017 cs130_tom 1.1  
1018                    sockaddr.sin_family = AF_INET;
1019                    sockaddr.sin_port = 0;
1020                    sockaddr.sin_addr.s_addr = 0xffffffff;
1021                
1022                    ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1023                    ok( !ret, "WSAAddressToStringW() failed unexpectedly: %x\n", WSAGetLastError() );
1024                
1025                    ok( !lstrcmpW( address, expect2 ), "Expected different address string\n" );
1026                
1027                    len = sizeof(address);
1028                
1029                    sockaddr.sin_family = AF_INET;
1030                    sockaddr.sin_port = 0xffff;
1031                    sockaddr.sin_addr.s_addr = 0;
1032                
1033                    ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1034                    ok( !ret, "WSAAddressToStringW() failed unexpectedly: %x\n", WSAGetLastError() );
1035                
1036                    ok( !lstrcmpW( address, expect3 ), "Expected different address string\n" );
1037                
1038 cs130_tom 1.1      len = sizeof(address);
1039                
1040                    sockaddr.sin_family = AF_INET;
1041                    sockaddr.sin_port = 0xffff;
1042                    sockaddr.sin_addr.s_addr = 0xffffffff;
1043                
1044                    ret = WSAAddressToStringW( (SOCKADDR*)&sockaddr, sizeof(sockaddr), NULL, address, &len );
1045                    ok( !ret, "WSAAddressToStringW() failed unexpectedly: %x\n", WSAGetLastError() );
1046                
1047                    ok( !lstrcmpW( address, expect4 ), "Expected different address string\n" );
1048                }
1049                
1050                static void test_WSAStringToAddressA()
1051                {
1052                    INT ret, len;
1053                    SOCKADDR_IN sockaddr;
1054                
1055                    CHAR address1[] = "0.0.0.0";
1056                    CHAR address2[] = "127.127.127.127";
1057                    CHAR address3[] = "255.255.255.255";
1058                    CHAR address4[] = "127.127.127.127:65535";
1059 cs130_tom 1.1      CHAR address5[] = "255.255.255.255:65535";
1060                
1061                    len = 0;
1062                    sockaddr.sin_family = AF_INET;
1063                
1064                    ret = WSAStringToAddressA( address1, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1065                    ok( ret == SOCKET_ERROR, "WSAStringToAddressA() succeeded unexpectedly: %x\n",
1066                        WSAGetLastError() );
1067                
1068                    len = sizeof(sockaddr);
1069                    sockaddr.sin_port = 0;
1070                    sockaddr.sin_addr.s_addr = 0;
1071                
1072                    ret = WSAStringToAddressA( address1, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1073                    ok( !ret && sockaddr.sin_addr.s_addr == 0,
1074                        "WSAStringToAddressA() failed unexpectedly: %d\n", WSAGetLastError() );
1075                
1076                    len = sizeof(sockaddr);
1077                    sockaddr.sin_port = 0;
1078                    sockaddr.sin_addr.s_addr = 0;
1079                
1080 cs130_tom 1.1      ret = WSAStringToAddressA( address2, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1081                    ok( !ret && sockaddr.sin_addr.s_addr == 0x7f7f7f7f,
1082                        "WSAStringToAddressA() failed unexpectedly: %d\n", WSAGetLastError() );
1083                
1084                    len = sizeof(sockaddr);
1085                
1086                    ret = WSAStringToAddressA( address3, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1087                    ok( ret, "WSAStringToAddressA() succeeded unexpectedly: %d\n", WSAGetLastError() );
1088                
1089                    len = sizeof(sockaddr);
1090                    sockaddr.sin_port = 0;
1091                    sockaddr.sin_addr.s_addr = 0;
1092                
1093                    ret = WSAStringToAddressA( address4, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1094                    ok( !ret && sockaddr.sin_addr.s_addr == 0x7f7f7f7f && sockaddr.sin_port == 0xffff,
1095                        "WSAStringToAddressA() failed unexpectedly: %d\n", WSAGetLastError() );
1096                
1097                    len = sizeof(sockaddr);
1098                
1099                    ret = WSAStringToAddressA( address5, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1100                    ok( ret, "WSAStringToAddressA() succeeded unexpectedly: %d\n", WSAGetLastError() );
1101 cs130_tom 1.1  }
1102                
1103                static void test_WSAStringToAddressW()
1104                {
1105                    INT ret, len;
1106                    SOCKADDR_IN sockaddr;
1107                
1108                    WCHAR address1[] = { '0','.','0','.','0','.','0', 0 };
1109                    WCHAR address2[] = { '1','2','7','.','1','2','7','.','1','2','7','.','1','2','7', 0 };
1110                    WCHAR address3[] = { '2','5','5','.','2','5','5','.','2','5','5','.','2','5','5', 0 };
1111                    WCHAR address4[] = { '1','2','7','.','1','2','7','.','1','2','7','.','1','2','7',
1112                                         ':', '6', '5', '5', '3', '5', 0 };
1113                    WCHAR address5[] = { '2','5','5','.','2','5','5','.','2','5','5','.','2','5','5', ':',
1114                                         '6', '5', '5', '3', '5', 0 };
1115                
1116                    len = 0;
1117                    sockaddr.sin_family = AF_INET;
1118                
1119                    ret = WSAStringToAddressW( address1, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1120                    ok( ret == SOCKET_ERROR, "WSAStringToAddressW() failed unexpectedly: %d\n",
1121                        WSAGetLastError() );
1122 cs130_tom 1.1  
1123                    len = sizeof(sockaddr);
1124                    sockaddr.sin_port = 0;
1125                    sockaddr.sin_addr.s_addr = 0;
1126                
1127                    ret = WSAStringToAddressW( address1, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1128                    ok( !ret && sockaddr.sin_addr.s_addr == 0,
1129                        "WSAStringToAddressW() failed unexpectedly: %d\n", WSAGetLastError() );
1130                
1131                    len = sizeof(sockaddr);
1132                
1133                    sockaddr.sin_port = 0;
1134                    sockaddr.sin_addr.s_addr = 0;
1135                
1136                    ret = WSAStringToAddressW( address2, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1137                    ok( !ret && sockaddr.sin_addr.s_addr == 0x7f7f7f7f,
1138                        "WSAStringToAddressW() failed unexpectedly: %d\n", WSAGetLastError() );
1139                
1140                    len = sizeof(sockaddr);
1141                
1142                    ret = WSAStringToAddressW( address3, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1143 cs130_tom 1.1      ok( ret, "WSAStringToAddressW() failed unexpectedly: %d\n", WSAGetLastError() );
1144                
1145                    len = sizeof(sockaddr);
1146                    sockaddr.sin_port = 0;
1147                    sockaddr.sin_addr.s_addr = 0;
1148                
1149                    ret = WSAStringToAddressW( address4, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1150                    ok( !ret && sockaddr.sin_addr.s_addr == 0x7f7f7f7f && sockaddr.sin_port == 0xffff,
1151                        "WSAStringToAddressW() failed unexpectedly: %d\n", WSAGetLastError() );
1152                
1153                    len = sizeof(sockaddr);
1154                
1155                    ret = WSAStringToAddressW( address5, AF_INET, NULL, (SOCKADDR*)&sockaddr, &len );
1156                    ok( ret, "WSAStringToAddressW() succeeded unexpectedly: %d\n", WSAGetLastError() );
1157                }
1158                
1159                /**************** Main program  ***************/
1160                
1161                START_TEST( sock )
1162                {
1163                    int i;
1164 cs130_tom 1.1      Init();
1165                
1166                    test_set_getsockopt();
1167                    test_so_reuseaddr();
1168                
1169                    for (i = 0; i < NUM_TESTS; i++)
1170                    {
1171                        trace ( " **** STARTING TEST %d **** \n", i );
1172                        do_test (  &tests[i] );
1173                        trace ( " **** TEST %d COMPLETE **** \n", i );
1174                    }
1175                
1176                    test_WSAAddressToStringA();
1177                    test_WSAAddressToStringW();
1178                
1179                    test_WSAStringToAddressA();
1180                    test_WSAStringToAddressW();
1181                
1182                    Exit();
1183                }

Rizwan Kassim
Powered by
ViewCVS 0.9.2