regress/sys/fs/lfs/ckckp/cleanalot_async.c

   1 /*      $NetBSD: cleanalot_async.c,v 1.6 2006/07/21 00:29:23 perseant Exp $     */
   2
   3 /*-
   4  * Copyright (c) 2006 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * This code is derived from software contributed to The NetBSD Foundation
   8  * by Konrad E. Schroder <perseant@hhhh.org>.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29  * POSSIBILITY OF SUCH DAMAGE.
  30  */
  31
  32 #include <stdio.h>
  33 #include <stdlib.h>
  34 #include <signal.h>
  35 #include <string.h>
  36 #include <unistd.h>
  37
  38 void dirname(int n, char *s)
  39 {
  40         if (n == 0) {
  41                 strcat(s, "/0");
  42                 mkdir(s);
  43         }
  44         while (n) {
  45                 sprintf(s + strlen(s), "/%x", n & 0xf);
  46                 n >>= 4;
  47                 mkdir(s);
  48         }
  49 }
  50
  51 /*
  52  * Write a file, creating the directory if necessary.
  53  */
  54 int write_file(int gen, int n, int plex, char *buf, int size)
  55 {
  56         FILE *fp;
  57         char s[1024], *t;
  58         int r;
  59
  60         sprintf(s, "dir_%x_%x", plex, gen);
  61         dirname(n, s);
  62         strcat(s, ".file");
  63
  64         /* printf("write file %d.%d.%x: %s\n", gen, plex, n, s); */
  65
  66         fp = fopen(s, "wb");
  67         if (fp == NULL)
  68                 return 0;
  69         if (size)
  70         r = fwrite(buf, size, 1, fp);
  71         else
  72                 r = 1;
  73         fclose(fp);
  74
  75         return r;
  76 }
  77
  78 int write_dirs(int gen, int size, int plex)
  79 {
  80         int i, j, tot;
  81         char s[1024];
  82         char *buf;
  83
  84         /* Create all base dirs */
  85         for (i = 0; i < plex; i++) {
  86                 sprintf(s, "dir_%x_%x", i, gen);
  87                 if (mkdir(s, 0700) != 0)
  88                         return 0;
  89         }
  90
  91         /* Write files */
  92         if (size) {
  93         buf = malloc(size);
  94         if (buf == NULL)
  95                 return 0;
  96         }
  97         tot = 0;
  98         for (i = 0; ; i++) {
  99                 for (j = 0; j < plex; j++) {
 100                         if (write_file(gen, i, j, buf, size) == 0) {
 101                                 if (size)
 102                                 free(buf);
 103                                 return tot;
 104                         }
 105                         ++tot;
 106                 }
 107         }
 108         /* NOTREACHED */
 109 }
 110
 111 int main(int argc, char **argv)
 112 {
 113         int c, i, j;
 114         int bs = 0;
 115         int count = 0;
 116         int plex = 2;
 117         char cmd[1024];
 118
 119         bs = -1;
 120         while((c = getopt(argc, argv, "b:n:p:")) != -1) {
 121                 switch(c) {
 122                     case 'b':
 123                         bs = atoi(optarg);
 124                         break;
 125                     case 'n':
 126                         count = atoi(optarg);
 127                         break;
 128                     case 'p':
 129                         plex = atoi(optarg);
 130                         break;
 131                     default:
 132                         exit(1);
 133                 }
 134         }
 135
 136         /*
 137          * Process old-style, non-flag parameters
 138          */
 139         if (count == 0) {
 140                 if (argv[optind] != NULL)
 141                         count = atoi(argv[optind]);
 142         }
 143         if (bs < 0 && getenv("BS") != NULL)
 144                 bs = atoi(getenv("BS"));
 145         if (bs < 0)
 146                 bs = 16384;
 147         if (plex == 0)
 148                 plex = 2;
 149
 150         for (i = 0; count == 0 || i < count; i++) {
 151                 if (count)
 152                         printf("::: begin iteration %d/%d\n", i, count);
 153                 else
 154                         printf("::: begin iteration %d\n", i);
 155
 156                 for (j = 0; ; j++) {
 157                         if ((c = write_dirs(j, bs, plex)) == 0)
 158                                 break;
 159                         printf("%d: %d files of size %d\n", j, c, bs);
 160                         sprintf(cmd, "rm -rf dir_%x_%x", plex - 1, j);
 161                         system(cmd);
 162                         sync();
 163                 }
 164                 system("df -k .");
 165                 printf("remove files\n");
 166                 system("rm -rf dir_*");
 167                 system("df -k .");
 168         }
 169 }