Whamcloud - gitweb
LU-14264 tests: make PARALLEL available to all suites
[fs/lustre-release.git] / lustre / tests / statmany.c
index 8a4f82a..bf58277 100644 (file)
@@ -27,7 +27,6 @@
  */
 /*
  * This file is part of Lustre, http://www.lustre.org/
- * Lustre is a trademark of Sun Microsystems, Inc.
  */
 
 #include <stdio.h>
@@ -50,33 +49,45 @@ struct option longopts[] = {
        { .name = "stat", .has_arg = no_argument, .val = 's' },
        { .name = NULL }
 };
+
 char *shortopts = "hlr:s0123456789";
 
 static int usage(char *prog, FILE *out)
 {
-        fprintf(out,
-               "Usage: %s [-r rand_seed] {-s|-l} filenamebase total_files iterations\n"
-               "-r : random seed\n"
-               "-s : regular stat() calls\n"
-               "-l : lookup ioctl only\n", prog);
-        exit(out == stderr);
+       fprintf(out,
+               "random stat of files within a directory\n"
+               "usage: %s [-r rand_seed] {-s|-l} filenamebase total_files [iterations]\n"
+               "-r : random seed for repeatable sequence\n"
+               "-s : regular stat() calls\n"
+               "-l : llapi_file_lookup() ioctl only\n"
+               "iterations: default = total_files, or negative for seconds\n",
+               prog);
+       exit(out == stderr);
 }
 
 #ifndef LONG_MAX
 #define LONG_MAX (1 << ((8 * sizeof(long)) - 1))
 #endif
 
-int main(int argc, char ** argv)
+int main(int argc, char **argv)
 {
-        long i, count, iter = LONG_MAX, mode = 0, offset;
-        long int start, length = LONG_MAX, last;
-        char parent[4096], *t;
-       char *prog = argv[0], *base;
+       long i, count, iter = 1, mode = 0, offset;
+       long int start, duration = LONG_MAX, last, now;
+       char parent[4096], *t;
+       char *prog, *base;
        int seed = 0, rc;
        int fd = -1;
 
-       while ((rc = getopt_long(argc, argv, shortopts, longopts, NULL)) != -1) {
+       prog = strrchr(argv[0], '/');
+       if (prog)
+               prog++;
+       else
+               prog = argv[0];
+
+       while ((rc = getopt_long(argc, argv, shortopts, longopts,
+                                NULL)) != -1) {
                char *e;
+
                switch (rc) {
                case 'r':
                        seed = strtoul(optarg, &e, 0);
@@ -89,6 +100,10 @@ int main(int argc, char ** argv)
                case 's':
                        mode = rc;
                        break;
+               /* a negative "count" argument (test duration in seconds,
+                * e.g. "-300") is treated as a command-line argument.
+                * Parse all of the digits here back into "duration".
+                */
                case '0':
                case '1':
                case '2':
@@ -99,10 +114,10 @@ int main(int argc, char ** argv)
                case '7':
                case '8':
                case '9':
-                       if (length == LONG_MAX)
-                               length = rc - '0';
+                       if (duration == LONG_MAX)
+                               duration = rc - '0';
                        else
-                               length = length * 10 + (rc - '0');
+                               duration = duration * 10 + (rc - '0');
                        break;
                case 'h':
                        usage(prog, stdout);
@@ -111,16 +126,17 @@ int main(int argc, char ** argv)
                }
        }
 
-       if (optind + 2 + (length == LONG_MAX) != argc) {
-               fprintf(stderr, "missing filenamebase, total_files, or iterations\n");
+       if (argc < optind + 2 || argc > optind + 3) {
+               fprintf(stderr,
+                       "missing filenamebase, total_files, or iterations\n");
                usage(prog, stderr);
        }
 
-        base = argv[optind];
-        if (strlen(base) > 4080) {
-                fprintf(stderr, "filenamebase too long\n");
-                exit(1);
-        }
+       base = argv[optind];
+       if (strlen(base) > 4080) {
+               fprintf(stderr, "filenamebase too long\n");
+               exit(1);
+       }
 
        if (seed == 0) {
                int f = open("/dev/urandom", O_RDONLY);
@@ -134,23 +150,29 @@ int main(int argc, char ** argv)
        printf("using seed %u\n", seed);
        srand(seed);
 
-        count = strtoul(argv[optind + 1], NULL, 0);
-       if (length == LONG_MAX) {
-               iter = strtoul(argv[optind + 2], NULL, 0);
+       count = strtoul(argv[optind + 1], NULL, 0);
+       if (duration == LONG_MAX) {
+               if (argc > optind + 2)
+                       iter = strtoul(argv[optind + 2], NULL, 0);
+               else
+                       iter = count;
                printf("running for %lu iterations\n", iter);
-       } else
-               printf("running for %lu seconds\n", length);
+       } else {
+               iter = LONG_MAX;
+               printf("running for %lu seconds\n", duration);
+       }
 
-        start = last = time(0);
+       start = time(0);
+       last = start;
 
-        t = strrchr(base, '/');
-        if (t == NULL) {
-                strcpy(parent, ".");
-                offset = -1;
-        } else {
-                strncpy(parent, base, t - base);
-                offset = t - base + 1;
-        }
+       t = strrchr(base, '/');
+       if (!t) {
+               strcpy(parent, ".");
+               offset = -1;
+       } else {
+               strncpy(parent, base, t - base);
+               offset = t - base + 1;
+       }
 
        if (mode == 'l') {
                fd = open(parent, O_RDONLY);
@@ -161,22 +183,22 @@ int main(int argc, char ** argv)
                }
        }
 
-        for (i = 0; i < iter && time(0) - start < length; i++) {
-                char filename[4096];
-                int tmp;
+       for (i = 0, now = start; i < iter && now - start < duration; i++) {
+               char filename[4096] = "";
+               int tmp;
 
-                tmp = random() % count;
-                sprintf(filename, "%s%d", base, tmp);
+               tmp = random() % count;
+               sprintf(filename, "%s%d", base, tmp);
 
                if (mode == 's') {
-                        struct stat buf;
-
-                        rc = stat(filename, &buf);
-                        if (rc < 0) {
-                                printf("stat(%s) error: %s\n", filename,
-                                       strerror(errno));
-                                break;
-                        }
+                       struct stat buf;
+
+                       rc = stat(filename, &buf);
+                       if (rc < 0) {
+                               printf("stat(%s) error: %s\n", filename,
+                                      strerror(errno));
+                               break;
+                       }
                } else if (mode == 'l') {
                        char *name = filename;
 
@@ -184,24 +206,25 @@ int main(int argc, char ** argv)
                                name += offset;
 
                        rc = llapi_file_lookup(fd, name);
-                        if (rc < 0) {
+                       if (rc < 0) {
                                printf("llapi_file_lookup for (%s) error: %s\n",
                                       filename, strerror(errno));
-                                break;
-                        }
-                }
-                if ((i % 10000) == 0) {
-                        printf(" - stat %lu (time %ld ; total %ld ; last %ld)\n",
-                               i, time(0), time(0) - start, time(0) - last);
-                        last = time(0);
-                }
-        }
+                               break;
+                       }
+               }
+               now = time(0);
+               if ((i > 0 && (i % 10000) == 0) || now - last > 10) {
+                       printf(" - stat %lu (time %ld ; total %ld ; last %ld)\n",
+                              i, now, now - start, now - last);
+                       last = now;
+               }
+       }
 
        if (mode == 'l')
                close(fd);
 
-        printf("total: %lu stats in %ld seconds: %f stats/second\n", i,
-               time(0) - start, ((float)i / (time(0) - start)));
+       printf("total: %lu stats in %ld seconds: %f stats/second\n",
+              i, now - start, ((float)i / (now - start)));
 
-        exit(rc);
+       return rc;
 }