src/rx/test/testserver.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 /* Server test program */
  11
  12 #include <afsconfig.h>
  13 #include <afs/param.h>
  14 #include <roken.h>
  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 #include <sys/time.h>
  28 #endif
  29 #include <signal.h>
  30 #include <sys/stat.h>
  31 #include <rx/rx_clock.h>
  32 #include <rx/rx.h>
  33 #include <rx/rx_globals.h>
  34 #include <rx/rx_null.h>
  35
  36 int error;                      /* Return this error number on a call */
  37 int print = 0, eventlog = 0, rxlog = 0;
  38 struct clock computeTime, waitTime;
  39 FILE *debugFile;
  40 char *rcvFile;
  41 int logstdout = 0;
  42
  43 void Abort(char *msg, ...);
  44 void Quit(char *msg);
  45 void OpenFD(int n);
  46 int  FileRequest(struct rx_call *call);
  47 int  SimpleRequest(struct rx_call *call);
  48
  49 void
  50 intSignal(int ignore)
  51 {
  52     Quit("Interrupted");
  53 }
  54
  55 void
  56 quitSignal(int ignore)
  57 {
  58     static int quitCount = 0;
  59     if (++quitCount > 1)
  60         Quit("rx_ctest: second quit signal, aborting");
  61     rx_debugFile = debugFile = fopen("rx_stest.db", "w");
  62     if (debugFile)
  63         rx_PrintStats(debugFile);
  64 }
  65
  66 #if !defined(AFS_NT40_ENV) && !defined(AFS_LINUX20_ENV)
  67 int
  68 test_syscall(a3, a4, a5)
  69      afs_uint32 a3, a4;
  70      void *a5;
  71 {
  72     afs_uint32 rcode;
  73     void (*old) (int);
  74
  75     old = signal(SIGSYS, SIG_IGN);
  76     rcode =
  77         syscall(31 /* AFS_SYSCALL */ , 28 /* AFSCALL_CALL */ , a3, a4, a5);
  78     signal(SIGSYS, old);
  79
  80     return rcode;
  81 }
  82 #endif
  83
  84 main(argc, argv)
  85      char **argv;
  86 {
  87     struct rx_service *service;
  88     struct rx_securityClass *(secobjs[1]);
  89     int err = 0;
  90     int setFD = 0;
  91     int jumbo = 0;
  92
  93 #if !defined(AFS_NT40_ENV) && !defined(AFS_LINUX20_ENV)
  94     setlinebuf(stdout);
  95     rxi_syscallp = test_syscall;
  96 #endif
  97     argv++;
  98     argc--;
  99     while (argc && **argv == '-') {
 100         if (strcmp(*argv, "-verbose") == 0)
 101             print = 1;
 102         else if (strcmp(*argv, "-jumbo") == 0)
 103             jumbo = 1;
 104         else if (strcmp(*argv, "-rxlog") == 0)
 105             rxlog = 1;
 106 #if defined(RXDEBUG) && !defined(AFS_NT40_ENV)
 107         else if (strcmp(*argv, "-trace") == 0)
 108             strcpy(rxi_tracename, *(++argv)), argc--;
 109 #endif
 110         else if (strcmp(*argv, "-logstdout") == 0)
 111             logstdout = 1;
 112         else if (strcmp(*argv, "-eventlog") == 0)
 113             eventlog = 1;
 114         else if (strcmp(*argv, "-npb") == 0)
 115             rx_nPackets = atoi(*++argv), argc--;
 116         else if (!strcmp(*argv, "-nsf"))
 117             rxi_nSendFrags = atoi(*++argv), argc--;
 118         else if (!strcmp(*argv, "-nrf"))
 119             rxi_nRecvFrags = atoi(*++argv), argc--;
 120         else if (strcmp(*argv, "-twind") == 0)
 121             rx_initSendWindow = atoi(*++argv), argc--;
 122         else if (strcmp(*argv, "-rwind") == 0)
 123             rx_initReceiveWindow = atoi(*++argv), argc--;
 124         else if (strcmp(*argv, "-file") == 0)
 125             rcvFile = *++argv, argc--;
 126         else if (strcmp(*argv, "-drop") == 0) {
 127 #ifdef RXDEBUG
 128             rx_intentionallyDroppedPacketsPer100 = atoi(*++argv), argc--;
 129 #else
 130             fprintf(stderr, "ERROR: Compiled without RXDEBUG\n");
 131 #endif
 132         }
 133         else if (strcmp(*argv, "-err") == 0)
 134             error = atoi(*++argv), argc--;
 135         else if (strcmp(*argv, "-compute") == 0) {
 136             /* Simulated "compute" time for each call--to test acknowledgement protocol.  This is simulated by doing an iomgr_select:  imperfect, admittedly. */
 137             computeTime.sec = atoi(*++argv), argc--;
 138             computeTime.usec = atoi(*++argv), argc--;
 139         } else if (strcmp(*argv, "-wait") == 0) {
 140             /* Wait time between calls--to test lastack code */
 141             waitTime.sec = atoi(*++argv), argc--;
 142             waitTime.usec = atoi(*++argv), argc--;
 143         } else if (strcmp(*argv, "-fd") == 0) {
 144             /* Open at least this many fd's. */
 145             setFD = atoi(*++argv), argc--;
 146         } else {
 147             err++;
 148             break;
 149         }
 150         argv++, argc--;
 151     }
 152     if (err || argc != 0)
 153         Quit("usage: rx_stest [-silent] [-rxlog] [-eventlog]");
 154
 155     if (rxlog || eventlog) {
 156         if (logstdout)
 157             debugFile = stdout;
 158         else
 159             debugFile = fopen("rx_stest.db", "w");
 160         if (debugFile == NULL)
 161             Quit("Couldn't open rx_stest.db");
 162         if (rxlog)
 163             rx_debugFile = debugFile;
 164         if (eventlog)
 165             rxevent_debugFile = debugFile;
 166     }
 167
 168     signal(SIGINT, intSignal);  /* Changed to SIGQUIT since dbx is broken right now */
 169 #ifndef AFS_NT40_ENV
 170     signal(SIGQUIT, quitSignal);
 171 #endif
 172
 173     if (setFD > 0)
 174         OpenFD(setFD);
 175
 176 #ifdef AFS_NT40_ENV
 177     rx_EnableHotThread();
 178 #endif
 179
 180     if (!jumbo)
 181         rx_SetNoJumbo();
 182
 183     rx_SetUdpBufSize(256 * 1024);
 184
 185     if (rx_Init(htons(2500)) != 0) {
 186         printf("RX initialization failed, exiting.\n");
 187         exit(1);
 188     }
 189     if (setFD > 0) {
 190         printf("rx_socket=%d\n", rx_socket);
 191     }
 192
 193     secobjs[0] = rxnull_NewServerSecurityObject();
 194     service = rx_NewService( /*port */ 0, /*service */ 3, "test", secobjs,      /*nsec */
 195                             1,  /*Execute request */
 196                             rcvFile ? FileRequest : SimpleRequest);
 197     if (!service)
 198         Abort("rx_NewService returned 0!\n");
 199
 200     rx_SetMinProcs(service, 2);
 201     rx_SetMaxProcs(service, 100);
 202     rx_SetCheckReach(service, 1);
 203
 204     printf("Using %d packet buffers\n", rx_nPackets);
 205     rx_StartServer(1);
 206 }
 207
 208 static char buf[2000000];
 209
 210 int SimpleRequest(struct rx_call *call)
 211 {
 212     int n;
 213     int nbytes = sizeof(buf);
 214     while ((n = rx_Read(call, buf, nbytes)) > 0)
 215         if (print)
 216             printf("stest: Received %d bytes\n", n);
 217     if (!rx_Error(call)) {
 218         /* Fake compute time (use select to lock out everything) */
 219 #ifndef AFS_NT40_ENV
 220         struct timeval t;
 221 #endif
 222         if (!clock_IsZero(&computeTime)) {
 223 #ifdef AFS_NT40_ENV
 224             Sleep(computeTime.sec * 1000 + 100 * computeTime.usec);
 225 #else
 226             t.tv_sec = computeTime.sec;
 227             t.tv_usec = computeTime.usec;
 228 #ifdef AFS_PTHREAD_ENV
 229             if (select(0, 0, 0, 0, &t) != 0)
 230                 Quit("Select didn't return 0");
 231 #else
 232             IOMGR_Sleep(t.tv_sec);
 233 #endif
 234 #endif
 235         }
 236         /* Then wait time (use iomgr_select to allow rx to run) */
 237         if (!clock_IsZero(&waitTime)) {
 238 #ifdef AFS_NT40_ENV
 239             Sleep(waitTime.sec * 1000 + 100 * waitTime.usec);
 240 #else
 241             t.tv_sec = waitTime.sec;
 242             t.tv_usec = waitTime.usec;
 243 #ifdef AFS_PTHREAD_ENV
 244             select(0, 0, 0, 0, &t);
 245 #else
 246             IOMGR_Sleep(t.tv_sec);
 247 #endif
 248 #endif
 249         }
 250         rx_Write(call, "So long, and thanks for all the fish!\n",
 251                  strlen("So long, and thanks for all the fish!\n"));
 252     }
 253     if (debugFile)
 254         rx_PrintPeerStats(debugFile, rx_PeerOf(call->conn));
 255     rx_PrintPeerStats(stdout, rx_PeerOf(call->conn));
 256     return 0;
 257 }
 258
 259 int
 260 FileRequest(struct rx_call *call)
 261 {
 262     int fd;
 263     struct stat status;
 264     int nbytes, blockSize;
 265     char *buffer;
 266 #ifdef  AFS_AIX_ENV
 267 #include <sys/statfs.h>
 268     struct statfs tstatfs;
 269 #endif
 270     /* Open the file ahead of time:  the client timing the operation doesn't have to include the open time */
 271     fd = open(rcvFile, O_WRONLY | O_CREAT | O_TRUNC, 0666);
 272     if (fd < 0) {
 273         perror("open");
 274         return -1;
 275     }
 276     fstat(fd, &status);
 277 #ifdef AFS_NT40_ENV
 278     blockSize = 1024;
 279 #else
 280 #ifdef  AFS_AIX_ENV
 281     fstatfs(fd, &tstatfs);
 282     blockSize = tstatfs.f_bsize;
 283 #else
 284     blockSize = status.st_blksize;
 285 #endif
 286 #endif /* AFS_NT40_ENV */
 287     buffer = (char *)malloc(blockSize);
 288
 289     rx_SetLocalStatus(call, 79);        /* Emulation of file server's old "RCallBackReceivedStore" */
 290
 291     while (nbytes = rx_Read(call, buffer, blockSize)) {
 292         if (write(fd, buffer, nbytes) != nbytes) {
 293             perror("writev");
 294             Abort("Write Failed.\n");
 295             break;
 296         }
 297     }
 298     rx_Write(call, "So long, and thanks for all the fish!\n",
 299              strlen("So long, and thanks for all the fish!\n"));
 300     printf("Received file %s\n", rcvFile);
 301     close(fd);
 302     if (debugFile)
 303         rx_PrintPeerStats(debugFile, rx_PeerOf(call->conn));
 304     rx_PrintPeerStats(stdout, rx_PeerOf(call->conn));
 305     return 0;
 306 }
 307
 308 void
 309 Abort(char *msg, ...)
 310 {
 311     va_list args;
 312
 313     va_start(args, msg);
 314     printf((char *)msg, args);
 315     va_end(args);
 316
 317     printf("\n");
 318     if (debugFile) {
 319         rx_PrintStats(debugFile);
 320         fflush(debugFile);
 321     }
 322     afs_abort();
 323     exit(1);
 324 }
 325
 326 void
 327 Quit(char *msg)
 328 {
 329     printf("%s\n", msg);
 330     if (debugFile) {
 331         rx_PrintStats(debugFile);
 332         fflush(debugFile);
 333     }
 334     rx_PrintStats(stdout);
 335     exit(0);
 336 }
 337
 338
 339 /* OpenFD
 340  *
 341  * Open file descriptors until file descriptor n or higher is returned.
 342  */
 343 #include <sys/stat.h>
 344 void
 345 OpenFD(int n)
 346 {
 347     int i;
 348     struct stat sbuf;
 349     int fd, lfd;
 350
 351     lfd = -1;
 352     for (i = 0; i < n; i++) {
 353         if (fstat(i, &sbuf) == 0)
 354             continue;
 355         if ((fd = open("/dev/null", 0, 0)) < 0) {
 356             if (lfd >= 0) {
 357                 close(lfd);
 358                 return;
 359             }
 360         } else {
 361             if (fd >= n) {
 362                 close(fd);
 363                 return;
 364             } else {
 365                 lfd = fd;
 366             }
 367         }
 368     }
 369 }