src/rx/test/testclient.c

   1 /*
   2  * Copyright 2000, International Business Machines Corporation and others.
   3  * All Rights Reserved.
   4  *
   5  * This software has been released under the terms of the IBM Public
   6  * License.  For details, see the LICENSE file in the top-level source
   7  * directory or online at http://www.openafs.org/dl/license10.html
   8  */
   9
  10 /* Client test program */
  11
  12 #include <afsconfig.h>
  13 #include <afs/param.h>
  14
  15
  16 #include <sys/types.h>
  17 #include <stdio.h>
  18 #ifdef AFS_NT40_ENV
  19 #include <io.h>
  20 #include <winsock2.h>
  21 #include <fcntl.h>
  22 #include <stdlib.h>
  23 #else
  24 #include <netinet/in.h>
  25 #include <netdb.h>
  26 #include <sys/file.h>
  27 #endif
  28 #include <sys/stat.h>
  29 #include <signal.h>
  30 #include "rx_clock.h"
  31 #include "rx.h"
  32 #include "rx_globals.h"
  33 #include "rx_null.h"
  34 #include <errno.h>
  35 #include <afs/afsutil.h>
  36
  37 #ifndef osi_Alloc
  38 #define osi_Alloc(n) (char *) malloc(n)
  39 #endif
  40
  41 int timeReadvs = 0;
  42 int print = 1, eventlog = 0, rxlog = 0;
  43 int fillPackets;
  44 int burst;
  45 struct clock burstTime;
  46 struct clock retryTime;
  47 FILE *debugFile;
  48 int timeout;
  49 struct clock waitTime, computeTime;
  50
  51 void OpenFD();
  52
  53 void
  54 intSignal(int ignore)
  55 {
  56     Quit("Interrupted");
  57 }
  58
  59 void
  60 quitSignal(int ignore)
  61 {
  62     static int quitCount = 0;
  63     if (++quitCount > 1)
  64         Quit("rx_ctest: second quit signal, aborting");
  65     rx_debugFile = debugFile = fopen("rx_ctest.db", "w");
  66     if (debugFile)
  67         rx_PrintStats(debugFile);
  68 }
  69
  70 #if !defined(AFS_NT40_ENV) && !defined(AFS_LINUX20_ENV)
  71 int
  72 test_syscall(a3, a4, a5)
  73      afs_uint32 a3, a4;
  74      void *a5;
  75 {
  76     afs_uint32 rcode;
  77     void (*old) (int);
  78
  79     old = signal(SIGSYS, SIG_IGN);
  80     rcode =
  81         syscall(31 /* AFS_SYSCALL */ , 28 /* AFSCALL_CALL */ , a3, a4, a5);
  82     signal(SIGSYS, old);
  83
  84     return rcode;
  85 }
  86 #endif
  87
  88 main(argc, argv)
  89      char **argv;
  90 {
  91     char *hostname;
  92     struct hostent *hostent;
  93     afs_uint32 host;
  94     int logstdout = 0;
  95     struct rx_connection *conn;
  96     struct rx_call *call;
  97     int err = 0;
  98     int nCalls = 1, nBytes = 1;
  99     int bufferSize = 4000000;
 100     char *buffer;
 101     char *sendFile = 0;
 102     int setFD = 0;
 103     int jumbo = 0;
 104
 105 #if !defined(AFS_NT40_ENV) && !defined(AFS_LINUX20_ENV)
 106     setlinebuf(stdout);
 107     rxi_syscallp = test_syscall;
 108 #endif
 109
 110
 111     argv++;
 112     argc--;
 113     while (argc && **argv == '-') {
 114         if (strcmp(*argv, "-silent") == 0)
 115             print = 0;
 116         if (strcmp(*argv, "-jumbo") == 0)
 117             jumbo = 1;
 118         else if (strcmp(*argv, "-nc") == 0)
 119             nCalls = atoi(*++argv), argc--;
 120         else if (strcmp(*argv, "-nb") == 0)
 121             nBytes = atoi(*++argv), argc--;
 122         else if (strcmp(*argv, "-np") == 0)
 123             rx_nPackets = atoi(*++argv), argc--;
 124         else if (!strcmp(*argv, "-nsf"))
 125             rxi_nSendFrags = atoi(*++argv), argc--;
 126         else if (!strcmp(*argv, "-nrf"))
 127             rxi_nRecvFrags = atoi(*++argv), argc--;
 128         else if (strcmp(*argv, "-twind") == 0)
 129             rx_initSendWindow = atoi(*++argv), argc--;
 130         else if (strcmp(*argv, "-rwind") == 0)
 131             rx_initReceiveWindow = atoi(*++argv), argc--;
 132         else if (strcmp(*argv, "-rxlog") == 0)
 133             rxlog = 1;
 134         else if (strcmp(*argv, "-logstdout") == 0)
 135             logstdout = 1;
 136         else if (strcmp(*argv, "-eventlog") == 0)
 137             eventlog = 1;
 138         else if (strcmp(*argv, "-drop") == 0) {
 139 #ifdef RXDEBUG
 140             rx_intentionallyDroppedPacketsPer100 = atoi(*++argv), argc--;
 141 #else
 142             fprintf(stderr, "ERROR: Compiled without RXDEBUG\n");
 143 #endif
 144         }
 145         else if (strcmp(*argv, "-burst") == 0) {
 146             burst = atoi(*++argv), argc--;
 147             burstTime.sec = atoi(*++argv), argc--;
 148             burstTime.usec = atoi(*++argv), argc--;
 149         } else if (strcmp(*argv, "-retry") == 0) {
 150             retryTime.sec = atoi(*++argv), argc--;
 151             retryTime.usec = atoi(*++argv), argc--;
 152         } else if (strcmp(*argv, "-timeout") == 0)
 153             timeout = atoi(*++argv), argc--;
 154         else if (strcmp(*argv, "-fill") == 0)
 155             fillPackets++;
 156         else if (strcmp(*argv, "-file") == 0)
 157             sendFile = *++argv, argc--;
 158         else if (strcmp(*argv, "-timereadvs") == 0)
 159             timeReadvs = 1;
 160         else if (strcmp(*argv, "-wait") == 0) {
 161             /* Wait time between calls--to test lastack code */
 162             waitTime.sec = atoi(*++argv), argc--;
 163             waitTime.usec = atoi(*++argv), argc--;
 164         } else if (strcmp(*argv, "-compute") == 0) {
 165             /* Simulated "compute" time for each call--to test acknowledgement protocol.  This is simulated by doing an iomgr_select:  imperfect, admittedly. */
 166             computeTime.sec = atoi(*++argv), argc--;
 167             computeTime.usec = atoi(*++argv), argc--;
 168         } else if (strcmp(*argv, "-fd") == 0) {
 169             /* Open at least this many fd's. */
 170             setFD = atoi(*++argv), argc--;
 171         } else {
 172             err = 1;
 173             break;
 174         }
 175         argv++, argc--;
 176     }
 177     if (err || argc != 1)
 178         Quit("usage: rx_ctest [-silent] [-rxlog] [-eventlog] [-nc NCALLS] [-np NPACKETS] hostname");
 179     hostname = *argv++, argc--;
 180
 181     if (rxlog || eventlog) {
 182         if (logstdout)
 183             debugFile = stdout;
 184         else
 185             debugFile = fopen("rx_ctest.db", "w");
 186         if (debugFile == NULL)
 187             Quit("Couldn't open rx_ctest.db");
 188         if (rxlog)
 189             rx_debugFile = debugFile;
 190         if (eventlog)
 191             rxevent_debugFile = debugFile;
 192     }
 193
 194     signal(SIGINT, intSignal);  /*Changed to sigquit since dbx is broken right now */
 195 #ifndef AFS_NT40_ENV
 196     signal(SIGQUIT, quitSignal);
 197 #endif
 198
 199 #ifdef AFS_NT40_ENV
 200     if (afs_winsockInit() < 0) {
 201         printf("Can't initialize winsock.\n");
 202         exit(1);
 203     }
 204     rx_EnableHotThread();
 205 #endif
 206
 207     rx_SetUdpBufSize(256 * 1024);
 208
 209     if (!jumbo)
 210         rx_SetNoJumbo();
 211
 212     hostent = gethostbyname(hostname);
 213     if (!hostent)
 214         Abort("host %s not found", hostname);
 215     if (hostent->h_length != 4)
 216         Abort("host address is disagreeable length (%d)", hostent->h_length);
 217     memcpy((char *)&host, hostent->h_addr, sizeof(host));
 218     if (setFD > 0)
 219         OpenFD(setFD);
 220     if (rx_Init(0) != 0) {
 221         printf("RX failed to initialize, exiting.\n");
 222         exit(2);
 223     }
 224     if (setFD > 0) {
 225         printf("rx_socket=%d\n", rx_socket);
 226     }
 227
 228     printf("Using %d packet buffers\n", rx_nPackets);
 229
 230     conn =
 231         rx_NewConnection(host, htons(2500), 3,
 232                          rxnull_NewClientSecurityObject(), 0);
 233
 234     if (!conn)
 235         Abort("unable to make a new connection");
 236
 237     /* Set initial parameters.  This is (currently) not the approved interface */
 238     if (burst)
 239         conn->peer->burstSize = conn->peer->burst = burst;
 240     if (!clock_IsZero(&burstTime))
 241         conn->peer->burstWait = burstTime;
 242     if (!clock_IsZero(&retryTime))
 243         conn->peer->timeout = retryTime;
 244
 245     if (sendFile)
 246         SendFile(sendFile, conn);
 247     else {
 248         buffer = (char *)osi_Alloc(bufferSize);
 249         while (nCalls--) {
 250             struct clock startTime;
 251             struct timeval t;
 252             int nbytes;
 253             int nSent;
 254             int bytesSent = 0;
 255             int bytesRead = 0;
 256             call = rx_NewCall(conn);
 257             if (!call)
 258                 Abort("unable to make a new call");
 259
 260             clock_GetTime(&startTime);
 261             for (bytesSent = 0; bytesSent < nBytes; bytesSent += nSent) {
 262                 int tryCount;
 263                 tryCount =
 264                     (bufferSize >
 265                      nBytes - bytesSent) ? nBytes - bytesSent : bufferSize;
 266                 nSent = rx_Write(call, buffer, tryCount);
 267                 if (nSent == 0)
 268                     break;
 269
 270             }
 271             for (bytesRead = 0; nbytes = rx_Read(call, buffer, bufferSize);
 272                  bytesRead += nbytes) {
 273             };
 274             if (print)
 275                 printf("Received %d characters in response\n", bytesRead);
 276             err = rx_EndCall(call, 0);
 277             if (err)
 278                 printf("Error %d returned from rpc call\n", err);
 279             else {
 280                 struct clock totalTime;
 281                 float elapsedTime;
 282                 clock_GetTime(&totalTime);
 283                 clock_Sub(&totalTime, &startTime);
 284                 elapsedTime = clock_Float(&totalTime);
 285                 fprintf(stderr,
 286                         "Sent %d bytes in %0.3f seconds:  %0.0f bytes per second\n",
 287                         bytesSent, elapsedTime, bytesSent / elapsedTime);
 288             }
 289             if (!clock_IsZero(&computeTime)) {
 290                 t.tv_sec = computeTime.sec;
 291                 t.tv_usec = computeTime.usec;
 292                 if (select(0, 0, 0, 0, &t) != 0)
 293                     Quit("Select didn't return 0");
 294             }
 295             if (!clock_IsZero(&waitTime)) {
 296                 struct timeval t;
 297                 t.tv_sec = waitTime.sec;
 298                 t.tv_usec = waitTime.usec;
 299 #ifdef AFS_PTHREAD_ENV
 300                 select(0, 0, 0, 0, &t);
 301 #else
 302                 IOMGR_Sleep(t.tv_sec);
 303 #endif
 304             }
 305             if (debugFile)
 306                 rx_PrintPeerStats(debugFile, rx_PeerOf(conn));
 307             rx_PrintPeerStats(stdout, rx_PeerOf(conn));
 308         }
 309     }
 310     Quit("testclient: done!\n");
 311 }
 312
 313 int SendFile(file, conn)
 314      char *file;
 315      struct rx_connection *conn;
 316 {
 317     struct rx_call *call;
 318     int fd;
 319     struct stat status;
 320     int blockSize, bytesLeft;
 321     char *buf;
 322     int nbytes;
 323     int err;
 324     struct clock startTime;
 325     int receivedStore = 0;
 326     struct clock totalReadvDelay;
 327     int nReadvs;
 328     int code;
 329 #ifdef  AFS_AIX_ENV
 330 #include <sys/statfs.h>
 331     struct statfs tstatfs;
 332 #endif
 333
 334     if (timeReadvs) {
 335         nReadvs = 0;
 336         clock_Zero(&totalReadvDelay);
 337     }
 338     fd = open(file, O_RDONLY, 0);
 339     if (fd < 0)
 340         Abort("Couldn't open %s\n", file);
 341     fstat(fd, &status);
 342 #ifdef AFS_NT40_ENV
 343     blockSize = 1024;
 344 #else
 345 #ifdef  AFS_AIX_ENV
 346 /* Unfortunately in AIX valuable fields such as st_blksize are gone from the stat structure!! */
 347     fstatfs(fd, &tstatfs);
 348     blockSize = tstatfs.f_bsize;
 349 #else
 350     blockSize = status.st_blksize;
 351 #endif
 352 #endif
 353     buf = (char *)osi_Alloc(blockSize);
 354     bytesLeft = status.st_size;
 355     clock_GetTime(&startTime);
 356     call = rx_NewCall(conn);
 357     while (bytesLeft) {
 358         if (!receivedStore && rx_GetRemoteStatus(call) == 79) {
 359             receivedStore = 1;
 360             fprintf(stderr,
 361                     "Remote status indicates file accepted (\"received store\")\n");
 362         }
 363         nbytes = (bytesLeft > blockSize ? blockSize : bytesLeft);
 364         errno = 0;
 365         code = read(fd, buf, nbytes);
 366         if (code != nbytes) {
 367             Abort("Only read %d bytes of %d, errno=%d\n", code, nbytes,
 368                   errno);
 369         }
 370         code = rx_Write(call, buf, nbytes);
 371         if (code != nbytes) {
 372             Abort("Only wrote %d bytes of %d\n", code, nbytes);
 373         }
 374         bytesLeft -= nbytes;
 375     }
 376     while ((nbytes = rx_Read(call, buf, sizeof(buf))) > 0) {
 377         char *p = buf;
 378         while (nbytes--) {
 379             putchar(*p);
 380             *p++;
 381         }
 382     }
 383     if ((err = rx_EndCall(call, 0)) != 0) {
 384         fprintf(stderr, "rx_Endcall returned error %d\n", err);
 385     } else {
 386         struct clock totalTime;
 387         float elapsedTime;
 388         clock_GetTime(&totalTime);
 389         clock_Sub(&totalTime, &startTime);
 390         elapsedTime = totalTime.sec + totalTime.usec / 1e6;
 391         fprintf(stderr,
 392                 "Sent %d bytes in %0.3f seconds:  %0.0f bytes per second\n",
 393                 status.st_size, elapsedTime, status.st_size / elapsedTime);
 394         if (timeReadvs) {
 395             float delay = clock_Float(&totalReadvDelay) / nReadvs;
 396             fprintf(stderr, "%d readvs, average delay of %0.4f seconds\n",
 397                     nReadvs, delay);
 398         }
 399     }
 400     close(fd);
 401
 402     return(0);
 403 }
 404
 405 Abort(msg, a, b, c, d, e)
 406 {
 407     printf((char *)msg, a, b, c, d, e);
 408     printf("\n");
 409     if (debugFile) {
 410         rx_PrintStats(debugFile);
 411         fflush(debugFile);
 412     }
 413     afs_abort();
 414     exit(1);
 415 }
 416
 417 Quit(char *msg, int a, int b, int c, int d, int e)
 418 {
 419     printf((char *)msg, a, b, c, d, e);
 420     printf("\n");
 421     if (debugFile) {
 422         rx_PrintStats(debugFile);
 423         fflush(debugFile);
 424     }
 425     rx_PrintStats(stdout);
 426     exit(0);
 427 }
 428
 429 /* OpenFD
 430  *
 431  * Open file descriptors until file descriptor n or higher is returned.
 432  */
 433 #include <sys/stat.h>
 434 void
 435 OpenFD(n)
 436      int n;
 437 {
 438     int i;
 439     struct stat sbuf;
 440     int fd, lfd;
 441
 442     lfd = -1;
 443     for (i = 0; i < n; i++) {
 444         if (fstat(i, &sbuf) == 0)
 445             continue;
 446         if ((fd = open("/dev/null", 0, 0)) < 0) {
 447             if (lfd >= 0) {
 448                 close(lfd);
 449                 return;
 450             }
 451         } else {
 452             if (fd >= n) {
 453                 close(fd);
 454                 return;
 455             } else {
 456                 lfd = fd;
 457             }
 458         }
 459     }
 460 }