lustre/tests/createmany-mpi.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  * GPL HEADER START
   5  *
   6  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License version 2 only,
  10  * as published by the Free Software Foundation.
  11  *
  12  * This program is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * General Public License version 2 for more details (a copy is included
  16  * in the LICENSE file that accompanied this code).
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * version 2 along with this program; If not, see
  20  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  21  *
  22  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  23  * CA 95054 USA or visit www.sun.com if you need additional information or
  24  * have any questions.
  25  *
  26  * GPL HEADER END
  27  */
  28 /*
  29  * Copyright  2008 Sun Microsystems, Inc. All rights reserved
  30  * Use is subject to license terms.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  */
  36
  37 #include <stdio.h>
  38 #include <sys/types.h>
  39 #include <sys/stat.h>
  40 #include <time.h>
  41 #include <errno.h>
  42 #include <string.h>
  43 #include <fcntl.h>
  44 #include <unistd.h>
  45 #include <stdlib.h>
  46 #include <stdarg.h>
  47
  48 #include "mpi.h"
  49
  50 void usage(char *prog)
  51 {
  52         printf("usage: %s {-o|-m|-l<tgt>} filenamefmt count\n", prog);
  53         printf("       %s {-o|-m|-l<tgt>} filenamefmt -seconds\n", prog);
  54         printf("       %s {-o|-m|-l<tgt>} filenamefmt start count\n", prog);
  55 }
  56
  57 /* Print process rank, loop count, message, and exit (i.e. a fatal error) */
  58 int rprintf(int rank, int loop, const char *fmt, ...)
  59 {
  60         va_list       ap;
  61
  62         printf("rank %d, loop %d: ", rank, loop);
  63
  64         va_start(ap, fmt);
  65
  66         printf(fmt, ap);
  67
  68         MPI_Finalize();
  69         exit(1);
  70 }
  71
  72 int main(int argc, char ** argv)
  73 {
  74         int i, rc = 0, do_open = 0, do_link = 0, rank;
  75         char format[4096], *fmt, *tgt = NULL;
  76         char filename[4096];
  77         long start, last, end;
  78         long begin = 0, count;
  79
  80         rc = MPI_Init(&argc, &argv);
  81         if (rc != MPI_SUCCESS)
  82                 rprintf(-1, -1, "MPI_Init failed: %d\n", rc);
  83
  84         if (argc < 4 || argc > 5) {
  85                 usage(argv[0]);
  86                 return 1;
  87         }
  88
  89         if (strcmp(argv[1], "-o") == 0) {
  90                 do_open = 1;
  91                 tgt = NULL;
  92         } else if (strncmp(argv[1], "-l", 2) == 0 && argv[1][2]) {
  93                 tgt = argv[1] + 2;
  94                 do_link = 1;
  95         } else if (strcmp(argv[1], "-m") != 0) {
  96                 usage(argv[0]);
  97                 return 1;
  98         }
  99
 100         if (strlen(argv[2]) > 4080) {
 101                 printf("name too long\n");
 102                 return 1;
 103         }
 104
 105         rc = MPI_Comm_rank(MPI_COMM_WORLD, &rank);
 106         if (rc != MPI_SUCCESS)
 107                 rprintf(-1, -1, "MPI_Comm_rank failed: %d\n", rc);
 108
 109         rc = MPI_Barrier(MPI_COMM_WORLD);
 110         if (rc != MPI_SUCCESS)
 111                 rprintf(rank, -1, "prep MPI_Barrier failed: %d\n", rc);
 112
 113         start = last = time(0);
 114
 115         if (argc == 4) {
 116                 end = strtol(argv[3], NULL, 0);
 117         } else {
 118                 begin = strtol(argv[3], NULL, 0);
 119                 end = strtol(argv[4], NULL, 0);
 120         }
 121         if (end > 0) {
 122                 count = end;
 123                 end = -1UL >> 1;
 124         } else {
 125                 end = start - end;
 126                 count = -1UL >> 1;
 127         }
 128
 129         if (strchr(argv[2], '%'))
 130                 fmt = argv[2];
 131         else {
 132                 sprintf(format, "%s%%d", argv[2]);
 133                 fmt = format;
 134         }
 135         printf("starting at %s", ctime(&start));
 136         for (i = 0; i < count && time(0) < end; i++, begin++) {
 137                 sprintf(filename, fmt, begin);
 138                 if (do_open) {
 139                         int fd = open(filename, O_CREAT|O_RDWR, 0644);
 140                         if (fd < 0) {
 141                                 printf("open(%s) error: %s\n", filename,
 142                                        strerror(errno));
 143                                 rc = errno;
 144                                 break;
 145                         }
 146                         close(fd);
 147                 } else if (do_link) {
 148                         rc = link(tgt, filename);
 149                         if (rc) {
 150                                 printf("link(%s, %s) error: %s\n",
 151                                        tgt, filename, strerror(errno));
 152                                 rc = errno;
 153                                 break;
 154                         }
 155                 } else {
 156                         rc = mknod(filename, S_IFREG| 0444, 0);
 157                         if (rc) {
 158                                 printf("mknod(%s) error: %s\n",
 159                                        filename, strerror(errno));
 160                                 rc = errno;
 161                                 break;
 162                         }
 163                 }
 164                 if ((i % 10000) == 0) {
 165                         printf(" - created %d (time %ld total %ld last %ld)\n",
 166                                i, time(0), time(0) - start, time(0) - last);
 167                         last = time(0);
 168                 }
 169         }
 170         printf("total: %d creates in %ld seconds: %f creates/second\n", i,
 171                time(0) - start, ((float)i / (time(0) - start)));
 172         start = time(0);
 173         printf("finish at %s", ctime(&start));
 174
 175         return rc;
 176 }