BSD 4_3_Net_2 release

[unix-history] / usr / src / sys / tests / nfs / unix-tests / basic / test5.c

/*      @(#)test5.c     1.5 90/01/03 NFS Rev 2 Testsuite
*      1.5 Lachman ONC Test Suite source
*
* Test read and write
*
* Uses the following important system calls against the server:
*
*      chdir()
*      mkdir()         (for initial directory creation if not -m)
*      creat()
*      open()
*      read()
*      write()
*      stat()
*      fstat()
*      unlink()
*/

#include <sys/param.h>
#ifndef major
#include <sys/types.h>
#endif
#ifdef SVR3
#include <sys/fs/nfs/time.h>
#include <fcntl.h>
#else
#include <sys/time.h>
#endif
#include <sys/stat.h>
#include <stdio.h>
#include "tests.h"

#ifndef MIN
#define MIN(a, b)       ((a) < (b) ? (a) : (b))
#endif

#define BUFSZ   8192
#define DSIZE   1048576

int Tflag = 0;          /* print timing */
int Hflag = 0;          /* print help message */
int Fflag = 0;          /* test function only;  set count to 1, negate -t */
int Nflag = 0;          /* Suppress directory operations */
#ifdef SVR3
int Sflag = 0;          /* use synchronous writes */
#endif

usage()
{
#ifdef SVR3
       fprintf(stdout, "usage: %s [-htfns] [size count fname]\n", Myname);
#else
       fprintf(stdout, "usage: %s [-htfn] [size count fname]\n", Myname);
#endif
       fprintf(stdout, "  Flags:  h    Help - print this usage info\n");
       fprintf(stdout, "          t    Print execution time statistics\n");
       fprintf(stdout, "          f    Test function only (negate -t)\n");
       fprintf(stdout, "          n    Suppress test directory create operations\n");
#ifdef SVR3
       fprintf(stdout, "          s    Use synchronous writes\n");
#endif
}

main(argc, argv)
       int argc;
       char *argv[];
{
       int count = DCOUNT;     /* times to do each file */
       int ct;
       int size = DSIZE;
       int si;
       int i;
       int fd;
       int bytes;
#ifdef SVR3
       int oflags;
#endif
       char *bigfile = "bigfile";
       struct timeval time;
       char str[MAXPATHLEN];
       struct stat statb;
       char *opts;
       char buf[BUFSZ];

       umask(0);
       setbuf(stdout, NULL);
       Myname = *argv++;
       argc--;
       while (argc && **argv == '-') {
               for (opts = &argv[0][1]; *opts; opts++) {
                       switch (*opts) {
                               case 'h':       /* help */
                                       usage();
                                       exit(1);

                               case 't':       /* time */
                                       Tflag++;
                                       break;
                               
                               case 'f':       /* funtionality */
                                       Fflag++;
                                       break;
                               
                               case 'n':       /* No Test Directory create */
                                       Nflag++;
                                       break;

#ifdef SVR3
                               case 's':       /* synchronous writes */
                                       Sflag++;
                                       break;
#endif

                               default:
                                       error("unknown option '%c'", *opts);
                                       usage();
                                       exit(1);
                       }
               }
               argc--;
               argv++;
       }

       if (argc) {
               size = getparm(*argv, 1, "size");
               if (size <= 0) {
                       usage();
                       exit(1);
               }
               argv++;
               argc--;
       }
       if (argc) {
               count = getparm(*argv, 1, "count");
               if (count <= 0) {
                       usage();
                       exit(1);
               }
               argv++;
               argc--;
       }
       if (argc) {
               bigfile = *argv;
               argv++;
               argc--;
       }
       if (argc) {
               usage();
               exit(1);
       }
       
       if (Fflag) {
               Tflag = 0;
               count = 1;
       }

#ifdef SVR3
       if (Sflag) {
               oflags = O_WRONLY|O_SYNC|O_CREAT|O_TRUNC;
       } else {
               oflags = O_WRONLY|O_CREAT|O_TRUNC;
       }
#endif

       fprintf(stdout, "%s: read and write\n", Myname);

       if (!Nflag)
               testdir(NULL);
       else
               mtestdir(NULL);

       for (i=0; i < BUFSZ / sizeof(int); i++) {
               ((int *)buf)[i] = i;
       }

       if (Tflag) {
               starttime();
       }

       for (ct = 0; ct < count; ct++) {
#ifdef SVR3
               if ((fd = open(bigfile, oflags, 0666)) < 0) {
#else
               if ((fd = creat(bigfile, 0666)) < 0) {
#endif
                       error("can't create '%s'", bigfile);
                       exit(1);
               }
               if (fstat(fd, &statb) < 0) {
                       error("can't stat '%s'", bigfile);
                       exit(1);
               }
               if (statb.st_size != 0) {
                       error("'%s' has size %d, should be 0",
                           bigfile, statb.st_size);
                       exit(1);
               }
               for (si = size; si > 0; si -= bytes) {
                       bytes = MIN(BUFSZ, si);
                       if (write(fd, buf, bytes) != bytes) {
                               error("'%s' write failed", bigfile);
                               exit(1);
                       }
               }
               close(fd);
               if (stat(bigfile, &statb) < 0) {
                       error("can't stat '%s'", bigfile);
                       exit(1);
               }
               if (statb.st_size != size) {
                       error("'%s' has size %d, should be %d",
                           bigfile, statb.st_size, size);
                       exit(1);
               }
       }

       if (Tflag) {
               endtime(&time);
       }

       if ((fd = open(bigfile, 0)) < 0) {
               error("can't open '%s'", bigfile);
               exit(1);
       }
       for (si = size; si > 0; si -= bytes) {
               bytes = MIN(BUFSZ, si);
               if (read(fd, buf, bytes) != bytes) {
                       error("'%s' read failed", bigfile);
                       exit(1);
               }
               for (i = 0; i < bytes / sizeof(int); i++) {
                       if (((int *)buf)[i] != i) {
                               error("bad data in '%s'", bigfile);
                               exit(1);
                       }
               }
       }
       close(fd);

       fprintf(stdout, "\twrote %d byte file %d times", size, count);
       if (Tflag) {
               fprintf(stdout, " in %d.%-2d seconds (%d bytes/sec)",
                   time.tv_sec, time.tv_usec / 10000, size*count/time.tv_sec);
       }
       fprintf(stdout, "\n");
       if (Tflag) {
               starttime();
       }

       for (ct = 0; ct < count; ct++) {
               if ((fd = open(bigfile, 0)) < 0) {
                       error("can't open '%s'", bigfile);
                       exit(1);
               }
               for (si = size; si > 0; si -= bytes) {
                       bytes = MIN(BUFSZ, si);
                       if (read(fd, buf, bytes) != bytes) {
                               error("'%s' read failed", bigfile);
                               exit(1);
                       }
               }
               close(fd);
       }

       if (Tflag) {
               endtime(&time);
       }
       fprintf(stdout, "\tread %d byte file %d times", size, count);
       if (Tflag) {
               fprintf(stdout, " in %d.%-2d seconds (%d bytes/sec)",
                   time.tv_sec, time.tv_usec / 10000, size*count/time.tv_sec);
       }
       fprintf(stdout, "\n");

       if (unlink(bigfile) < 0) {
               error("can't unlink '%s'", bigfile);
               exit(1);
       }
       complete();
}