- tagging RC_CURRENT

[fs/lustre-release.git] / lustre / tests / createmany-mpi.c

diff --git a/lustre/tests/createmany-mpi.c b/lustre/tests/createmany-mpi.c
index 1474b7b..289e071 100644 (file)
--- a/lustre/tests/createmany-mpi.c
+++ b/lustre/tests/createmany-mpi.c
@@ -1,3 +1,7 @@
+/* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
+ * vim:expandtab:shiftwidth=8:tabstop=8:
+ */
+
 #include <stdio.h>
 #include <sys/types.h>
 #include <sys/stat.h>
@@ -13,9 +17,9 @@
 
 void usage(char *prog)
 {
-       printf("usage: %s {-o|-m} filenamefmt count\n", prog);
-       printf("       %s {-o|-m} filenamefmt -seconds\n", prog);
-       printf("       %s {-o|-m} filenamefmt start count\n", prog);
+        printf("usage: %s {-o|-m|-l<tgt>} filenamefmt count\n", prog);
+        printf("       %s {-o|-m|-l<tgt>} filenamefmt -seconds\n", prog);
+        printf("       %s {-o|-m|-l<tgt>} filenamefmt start count\n", prog);
 }
 
 /* Print process rank, loop count, message, and exit (i.e. a fatal error) */
@@ -35,27 +39,28 @@ int rprintf(int rank, int loop, const char *fmt, ...)
 
 int main(int argc, char ** argv)
 {
-        int i, rc = 0, do_open, rank;
-        char format[4096], *fmt;
+        int i, rc = 0, do_open = 0, do_link = 0, rank;
+        char format[4096], *fmt, *tgt;
         char filename[4096];
         long start, last, end;
-       long begin = 0, count;
+        long begin = 0, count;
 
         rc = MPI_Init(&argc, &argv);
         if (rc != MPI_SUCCESS)
                 rprintf(-1, -1, "MPI_Init failed: %d\n", rc);
 
         if (argc < 4 || argc > 5) {
-               usage(argv[0]);
+                usage(argv[0]);
                 return 1;
         }
 
         if (strcmp(argv[1], "-o") == 0) {
                 do_open = 1;
-        } else if (strcmp(argv[1], "-m") == 0) {
-                do_open = 0;
-        } else {
-               usage(argv[0]);
+        } else if (strncmp(argv[1], "-l", 2) == 0 && argv[1][2]) {
+                tgt = argv[1] + 2;
+                do_link = 1;
+        } else if (strcmp(argv[1], "-m") != 0) {
+                usage(argv[0]);
                 return 1;
         }
 
@@ -74,28 +79,27 @@ int main(int argc, char ** argv)
 
         start = last = time(0);
 
-       if (argc == 4) {
-               end = strtol(argv[3], NULL, 0);
-               if (end > 0) {
-                       count = end;
-                       end = -1UL >> 1;
-               } else {
-                       end = start - end;
-                       count = -1UL >> 1;
-               }
-       } else {
-               end = -1UL >> 1;
-               begin = strtol(argv[3], NULL, 0);
-               count = strtol(argv[4], NULL, 0);
-       }
-
-       if (strchr(argv[2], '%'))
-               fmt = argv[2];
-       else {
-               sprintf(format, "%s%%d", argv[2]);
-               fmt = format;
-       }
-       printf("starting at %s", ctime(&start));
+        if (argc == 4) {
+                end = strtol(argv[3], NULL, 0);
+        } else {
+                begin = strtol(argv[3], NULL, 0);
+                end = strtol(argv[4], NULL, 0);
+        }
+        if (end > 0) {
+                count = end;
+                end = -1UL >> 1;
+        } else {
+                end = start - end;
+                count = -1UL >> 1;
+        }
+
+        if (strchr(argv[2], '%'))
+                fmt = argv[2];
+        else {
+                sprintf(format, "%s%%d", argv[2]);
+                fmt = format;
+        }
+        printf("starting at %s", ctime(&start));
         for (i = 0; i < count && time(0) < end; i++, begin++) {
                 sprintf(filename, fmt, begin);
                 if (do_open) {
@@ -107,6 +111,14 @@ int main(int argc, char ** argv)
                                 break;
                         }
                         close(fd);
+                } else if (do_link) {
+                        rc = link(tgt, filename);
+                        if (rc) {
+                                printf("link(%s, %s) error: %s\n",
+                                       tgt, filename, strerror(errno));
+                                rc = errno;
+                                break;
+                        }
                 } else {
                         rc = mknod(filename, S_IFREG| 0444, 0);
                         if (rc) {
@@ -117,15 +129,15 @@ int main(int argc, char ** argv)
                         }
                 }
                 if ((i % 10000) == 0) {
-                        printf(" - created %d (time %ld ; total %ld ; last %ld)\n",
+                        printf(" - created %d (time %ld total %ld last %ld)\n",
                                i, time(0), time(0) - start, time(0) - last);
                         last = time(0);
                 }
         }
         printf("total: %d creates in %ld seconds: %f creates/second\n", i,
                time(0) - start, ((float)i / (time(0) - start)));
-       start = time(0);
-       printf("finish at %s", ctime(&start));
+        start = time(0);
+        printf("finish at %s", ctime(&start));
 
         return rc;
 }