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.gnu.org/licenses/gpl-2.0.html
  19  *
  20  * GPL HEADER END
  21  */
  22 /*
  23  * Copyright (c) 2003, 2010, Oracle and/or its affiliates. All rights reserved.
  24  * Use is subject to license terms.
  25  */
  26 /*
  27  * This file is part of Lustre, http://www.lustre.org/
  28  * Lustre is a trademark of Sun Microsystems, Inc.
  29  */
  30
  31 #include <stdio.h>
  32 #include <sys/types.h>
  33 #include <sys/stat.h>
  34 #include <time.h>
  35 #include <errno.h>
  36 #include <string.h>
  37 #include <fcntl.h>
  38 #include <unistd.h>
  39 #include <stdlib.h>
  40
  41 void usage(char *prog)
  42 {
  43         printf("usage: %s [option] filenamefmt count\n", prog);
  44         printf("       %s [option] filenamefmt start count\n", prog);
  45         printf("Options are:\n");
  46         printf("       -d : remove directory\n");
  47         printf("Examples:\n");
  48         printf("unlinkmany $DIR/dir 100 # Remove file dir0..dir99\n");
  49         printf("unlinkmany -d $DIR/dir 100 # Remove directory dir0..dir99\n");
  50 }
  51
  52 int main(int argc, char **argv)
  53 {
  54         int i, rc = 0, do_rmdir = 0;
  55         char format[4096], *fmt;
  56         char filename[4096];
  57         long start, last;
  58         long begin = 0, count;
  59
  60         if (argc < 3 || argc > 5) {
  61                 usage(argv[0]);
  62                 return 1;
  63         }
  64
  65         if (strcmp(argv[1], "-d") == 0) {
  66                 do_rmdir = 1;
  67                 argv++;
  68                 argc--;
  69         }
  70
  71         if (strlen(argv[1]) > 4080) {
  72                 printf("name too long\n");
  73                 return 1;
  74         }
  75
  76         start = time(0);
  77         last = start;
  78
  79         if (argc == 3) {
  80                 count = strtol(argv[2], NULL, 0);
  81                 if (count < 1) {
  82                         printf("count must be at least one\n");
  83                         return 1;
  84                 }
  85         } else {
  86                 begin = strtol(argv[2], NULL, 0);
  87                 count = strtol(argv[3], NULL, 0);
  88         }
  89
  90         if (strchr(argv[1], '%')) {
  91                 fmt = argv[1];
  92         } else {
  93                 sprintf(format, "%s%%d", argv[1]);
  94                 fmt = format;
  95         }
  96
  97         for (i = 0; i < count; i++, begin++) {
  98                 sprintf(filename, fmt, begin);
  99                 if (do_rmdir)
 100                         rc = rmdir(filename);
 101                 else
 102                         rc = unlink(filename);
 103                 if (rc) {
 104                         printf("%s(%s) error: %s\n",
 105                                do_rmdir ? "rmdir" : "unlink",
 106                                filename, strerror(errno));
 107                         rc = errno;
 108                         break;
 109                 }
 110                 if ((i % 10000) == 0) {
 111                         printf(" - unlinked %d (time %ld ; total %ld ; last %ld)\n",
 112                                i, time(0), time(0) - start, time(0) - last);
 113                         last = time(0);
 114                 }
 115         }
 116         printf("total: %d unlinks in %ld seconds: %f unlinks/second\n", i,
 117                time(0) - start, ((float)i / (time(0) - start)));
 118
 119         return rc;
 120 }