lustre/tests/unlinkmany.c

   1 /*
   2  * GPL HEADER START
   3  *
   4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License version 2 only,
   8  * as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License version 2 for more details (a copy is included
  14  * in the LICENSE file that accompanied this code).
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * version 2 along with this program; If not, see
  18  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  19  *
  20  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  21  * CA 95054 USA or visit www.sun.com if you need additional information or
  22  * have any questions.
  23  *
  24  * GPL HEADER END
  25  */
  26 /*
  27  * Copyright (c) 2003, 2010, Oracle and/or its affiliates. All rights reserved.
  28  * Use is subject to license terms.
  29  */
  30 /*
  31  * This file is part of Lustre, http://www.lustre.org/
  32  * Lustre is a trademark of Sun Microsystems, Inc.
  33  */
  34
  35 #include <stdio.h>
  36 #include <sys/types.h>
  37 #include <sys/stat.h>
  38 #include <time.h>
  39 #include <errno.h>
  40 #include <string.h>
  41 #include <fcntl.h>
  42 #include <unistd.h>
  43 #include <stdlib.h>
  44
  45 void usage(char *prog)
  46 {
  47         printf("usage: %s filenamefmt count\n", prog);
  48         printf("       %s filenamefmt start count\n", prog);
  49 }
  50
  51 int main(int argc, char ** argv)
  52 {
  53         int i, rc = 0, do_rmdir = 0;
  54         char format[4096], *fmt;
  55         char filename[4096];
  56         long start, last;
  57         long begin = 0, count;
  58
  59         if (argc < 3 || argc > 5) {
  60                 usage(argv[0]);
  61                 return 1;
  62         }
  63
  64         if (strcmp(argv[1], "-d") == 0) {
  65                 do_rmdir = 1;
  66                 argv++;
  67                 argc--;
  68         }
  69
  70         if (strlen(argv[1]) > 4080) {
  71                 printf("name too long\n");
  72                 return 1;
  73         }
  74
  75         start = last = time(0);
  76
  77         if (argc == 3) {
  78                 count = strtol(argv[2], NULL, 0);
  79                 if (count < 1) {
  80                         printf("count must be at least one\n");
  81                         return 1;
  82                 }
  83         } else {
  84                 begin = strtol(argv[2], NULL, 0);
  85                 count = strtol(argv[3], NULL, 0);
  86         }
  87
  88         if (strchr(argv[1], '%')) {
  89                 fmt = argv[1];
  90         } else {
  91                 sprintf(format, "%s%%d", argv[1]);
  92                 fmt = format;
  93         }
  94         for (i = 0; i < count; i++, begin++) {
  95                 sprintf(filename, fmt, begin);
  96                 if (do_rmdir)
  97                         rc = rmdir(filename);
  98                 else
  99                         rc = unlink(filename);
 100                 if (rc) {
 101                         printf("%s(%s) error: %s\n",
 102                                do_rmdir ? "rmdir" : "unlink",
 103                                filename, strerror(errno));
 104                         rc = errno;
 105                         break;
 106                 }
 107                 if ((i % 10000) == 0) {
 108                         printf(" - unlinked %d (time %ld ; total %ld ; last "
 109                                "%ld)\n", i, time(0), time(0) - start,
 110                                time(0) - last);
 111                         last = time(0);
 112                 }
 113         }
 114         printf("total: %d unlinks in %ld seconds: %f unlinks/second\n", i,
 115                time(0) - start, ((float)i / (time(0) - start)));
 116
 117         return rc;
 118 }