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

Diff for /wine4/wine/dlls/winsock/tests/sock.c between version 1.2 and 1.15

version 1.2, 2005/03/03 03:26:04 version 1.15, 2005/03/10 02:51:33
Line 2 
Line 2 
  * Unit test suite for winsock functions  * Unit test suite for winsock functions
  *  *
  * Copyright 2002 Martin Wilck  * Copyright 2002 Martin Wilck
    * select() server Copyright 2005 Thomas Kho
  *  *
  * This library is free software; you can redistribute it and/or  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public  * modify it under the terms of the GNU Lesser General Public
Line 60 
Line 61 
     struct sockaddr_in     addr;     struct sockaddr_in     addr;
     struct sockaddr_in     peer;     struct sockaddr_in     peer;
     char                  *buf;     char                  *buf;
     int                    nread;      int                    n_recvd;
       int                    n_sent;
 } sock_info; } sock_info;
  
 /* Test parameters for both server & client */ /* Test parameters for both server & client */
Line 200 
Line 202 
     return p - buf;     return p - buf;
 } }
  
 static int do_nonblocking_send ( int id, SOCKET s, char *buf, int buflen, int sendlen )  
 {  
     char *last = buf + buflen, *p[MAX_CLIENTS] = NULL;  
     int n;  
   
     if ( p[id] == NULL ) {  
         p[id] = buf;  
     }  
   
     n = send ( s, p[id], min ( sendlen, last - p[id] ), 0 );  
     p[id] += n;  
   
     //wsa_ok ( n, 0 <=, "do_synchronous_send (%lx): error %d\n" );  
     return p[id] - buf;  
 }  
   
 static int do_nonblocking_recv ( int id, SOCKET s, char *buf, int buflen, int recvlen )  
 {  
     char *last = buf + buflen, *p[MAX_CLIENTS] = NULL;  
     int n;  
   
     if ( p[id] == NULL ) {  
         p[id] = buf;  
     }  
   
     n = recv ( s, p[id], min ( recvlen, last - p[id] ), 0 );  
     p[id] += n;  
   
     //wsa_ok ( n, 0 <=, "do_synchronous_recv (%lx): error %d:\n" );  
     return p[id] - buf;  
 }  
   
 /* /*
  *  Call this routine right after thread startup.  *  Call this routine right after thread startup.
  *  SO_OPENTYPE must by 0, regardless what the server did.  *  SO_OPENTYPE must by 0, regardless what the server did.
Line 284 
Line 254 
     {     {
         mem->sock[i].s = INVALID_SOCKET;         mem->sock[i].s = INVALID_SOCKET;
         mem->sock[i].buf = (LPVOID) LocalAlloc ( LPTR, gen->n_chunks * gen->chunk_size );         mem->sock[i].buf = (LPVOID) LocalAlloc ( LPTR, gen->n_chunks * gen->chunk_size );
         mem->sock[i].nread = 0;          mem->sock[i].n_recvd = 0;
           mem->sock[i].n_sent = 0;
     }     }
  
     if ( gen->sock_type == SOCK_STREAM )     if ( gen->sock_type == SOCK_STREAM )
Line 411 
Line 382 
 { {
     test_params *gen = par->general;     test_params *gen = par->general;
     server_memory *mem;     server_memory *mem;
     int n_recvd, n_sent, n_expected = gen->n_chunks * gen->chunk_size, tmp, i,      int n_expected = gen->n_chunks * gen->chunk_size, tmp, i,
         id = GetCurrentThreadId(), n_connections = 0;          id = GetCurrentThreadId(), n_connections = 0, n_sent, n_recvd,
           n_set, delta, n_ready;
     char *p;     char *p;
     struct timeval zerotime = {0,0};      struct timeval timeout = {0,10}; /* wait for 10 milliseconds */
     fd_set fds_read, fds_write, fds_open;      fd_set fds_recv, fds_send, fds_openrecv, fds_opensend;
  
     trace ( "select_server (%x) starting\n", id );     trace ( "select_server (%x) starting\n", id );
  
Line 429 
Line 401 
     trace ( "select_server (%x) ready\n", id );     trace ( "select_server (%x) ready\n", id );
     SetEvent ( server_ready ); /* notify clients */     SetEvent ( server_ready ); /* notify clients */
  
     FD_ZERO ( &fds_openread );      FD_ZERO ( &fds_openrecv );
     FD_ZERO ( &fds_read );      FD_ZERO ( &fds_recv );
     FD_ZERO ( &fds_write );      FD_ZERO ( &fds_send );
       FD_ZERO ( &fds_opensend );
  
     FD_SET ( mem->s, &fds_open );      FD_SET ( mem->s, &fds_openrecv );
  
     while(1)     while(1)
     {     {
         fds_read = fds_write = fds_open;          fds_recv = fds_openrecv;
           fds_send = fds_opensend;
  
         select ( 0, &fds_read, &fds_write, NULL, &zerotime );          n_set = 0;
   
           wsa_ok ( ( n_ready = select ( 0, &fds_recv, &fds_send, NULL, &timeout ) ), SOCKET_ERROR !=,
               "select_server (%lx): select() failed: %d\n" );
  
         /* check for incoming requests */         /* check for incoming requests */
         if ( FD_ISSET ( mem->s, &fds_read ) ) {          if ( FD_ISSET ( mem->s, &fds_recv ) ) {
               n_set += 1;
   
             trace ( "select_server (%x): accepting client connection\n", id );             trace ( "select_server (%x): accepting client connection\n", id );
  
             /* accept a single connection */             /* accept a single connection */
             tmp = sizeof ( mem->sock[i].peer );              tmp = sizeof ( mem->sock[n_connections].peer );
             mem->sock[i].s = accept ( mem->s, (struct sockaddr*) &mem->sock[i].peer, &tmp );              mem->sock[n_connections].s = accept ( mem->s, (struct sockaddr*) &mem->sock[n_connections].peer, &tmp );
             wsa_ok ( mem->sock[i].s, INVALID_SOCKET !=, "select_server (%lx): accept failed: %d\n" );              wsa_ok ( mem->sock[n_connections].s, INVALID_SOCKET !=, "select_server (%lx): accept() failed: %d\n" );
  
             ok ( mem->sock[i].peer.sin_addr.s_addr == inet_addr ( gen->inet_addr ),              ok ( mem->sock[n_connections].peer.sin_addr.s_addr == inet_addr ( gen->inet_addr ),
                 "select_server (%x): strange peer address\n", id );                 "select_server (%x): strange peer address\n", id );
  
             /* add to list of open connections */             /* add to list of open connections */
             FD_SET ( mem->sock[i].s, &fds_open );              FD_SET ( mem->sock[n_connections].s, &fds_openrecv );
               FD_SET ( mem->sock[n_connections].s, &fds_opensend );
  
             n_connections += 1;              n_connections++;
         }         }
  
         /* handle open requests */         /* handle open requests */
  
         for ( i = 0; i < n_connections; i++ )         for ( i = 0; i < n_connections; i++ )
         {         {
             if ( (mem->sock[i].nread < n_expected)              if ( FD_ISSET( mem->sock[i].s, &fds_recv ) ) {
                 && FD_ISSET( mem->sock[i].s, &fds_read ) ) {                  n_set += 1;
                 /* more data to receive */  
  
                   if ( mem->sock[i].n_recvd < n_expected ) {
                 /* Receive data & check it */                 /* Receive data & check it */
                 n_recvd = do_nonblocking_recv ( i, mem->sock[0].s, mem->sock[0].buf, n_expected, par->buflen );                      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 );
                 //ok ( n_recvd == n_expected,                      ok ( n_recvd != SOCKET_ERROR, "select_server (%x): error in recv(): %d\n", id, WSAGetLastError() );
                 //    "select_server (%x): received less data than expected: %d of %d\n", id, n_recvd, n_expected );                      mem->sock[i].n_recvd += n_recvd;
  
                 if ( n_recvd == n_expected ) {                      if ( mem->sock[i].n_recvd == n_expected ) {
                     p = test_buffer ( mem->sock[0].buf, gen->chunk_size, gen->n_chunks );                          p = test_buffer ( mem->sock[i].buf, gen->chunk_size, gen->n_chunks );
                     ok ( p == NULL, "select_server (%x): test pattern error: %d\n", id, p - mem->sock[0].buf);                          ok ( p == NULL, "select_server (%x): test pattern error: %d\n", id, p - mem->sock[i].buf );
                           FD_CLR ( mem->sock[i].s, &fds_openrecv );
                 }                 }
  
             } else if ( mem->sock[i].nread == n_expected                      ok ( mem->sock[i].n_recvd <= n_expected, "select_server (%x): received too many bytes: %d\n", id, mem->sock[i].n_recvd );
                 && FD_ISSET ( mem->sock[i].s, &fds_write ) ) {                  }
                 /* send data */              }
  
                 /* Echo data back */              /* only echo back what we've received */
                 n_sent = do_nonblocking_send ( i, mem->sock[0].s, mem->sock[0].buf, n_expected, par->buflen );              delta = mem->sock[i].n_recvd - mem->sock[i].n_sent;
                 ok ( n_sent == n_expected,  
                     "select_server (%x): sent less data than expected: %d of %d\n", id, n_sent, n_expected );  
  
               if ( FD_ISSET ( mem->sock[i].s, &fds_send ) ) {
                   n_set += 1;
  
                   if ( ( delta > 0 ) && ( mem->sock[i].n_sent < n_expected ) ) {
                       /* Echo data back */
                       n_sent = send ( mem->sock[i].s, mem->sock[i].buf + mem->sock[i].n_sent, min ( delta, par->buflen ), 0 );
                       ok ( n_sent != SOCKET_ERROR, "select_server (%x): error in send(): %d\n", id, WSAGetLastError() );
                       mem->sock[i].n_sent += n_sent;
   
                       if ( mem->sock[i].n_sent == n_expected ) {
                           FD_CLR ( mem->sock[i].s, &fds_opensend );
                       }
  
             } else {                      ok ( mem->sock[i].n_sent <= n_expected, "select_server (%x): sent too many bytes: %d\n", id, mem->sock[i].n_sent );
                 ok ( 0, "select_server (%x): too many bytes read\n", id );  
             }             }
         }         }
     }     }
  
           /* check that select returned the correct number of ready sockets */
           ok ( ( n_set == n_ready ), "select_server (%x): select() returns wrong number of ready sockets\n", id );
   
           /* check if all clients are done */
           if ( ( fds_opensend.fd_count == 0 )
               && ( fds_openrecv.fd_count == 1 ) /* initial socket that accepts clients */
               && ( n_connections  == min ( gen->n_clients, MAX_CLIENTS ) ) ) {
               break;
           }
       }
   
     for ( i = 0; i < min ( gen->n_clients, MAX_CLIENTS ); i++ )     for ( i = 0; i < min ( gen->n_clients, MAX_CLIENTS ); i++ )
     {     {
         /* cleanup */         /* cleanup */
Line 909 
Line 909 
             128             128
         }         }
     },     },
     /* Test 2: event-driven client, non-blocking server via select() */      /* Test 2: synchronous client, non-blocking server via select() */
     {     {
         {         {
             STD_STREAM_SOCKET,             STD_STREAM_SOCKET,
Line 923 
Line 923 
             0,             0,
             64             64
         },         },
         event_client,          simple_client,
         {         {
             NULL,             NULL,
             WSA_FLAG_OVERLAPPED,              0,
             128             128
         }         }
     }     }

Legend:
Removed from v.1.2  
changed lines
  Added in v.1.15
Rizwan Kassim
 Powered by
ViewCVS 0.9.2