tests/zfs-tests/cmd/mktree/mktree.c

   1 /*
   2  * CDDL HEADER START
   3  *
   4  * The contents of this file are subject to the terms of the
   5  * Common Development and Distribution License (the "License").
   6  * You may not use this file except in compliance with the License.
   7  *
   8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
   9  * or http://www.opensolaris.org/os/licensing.
  10  * See the License for the specific language governing permissions
  11  * and limitations under the License.
  12  *
  13  * When distributing Covered Code, include this CDDL HEADER in each
  14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  15  * If applicable, add the following below this CDDL HEADER, with the
  16  * fields enclosed by brackets "[]" replaced with your own identifying
  17  * information: Portions Copyright [yyyy] [name of copyright owner]
  18  *
  19  * CDDL HEADER END
  20  */
  21
  22 /*
  23  * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
  24  * Use is subject to license terms.
  25  */
  26
  27 #include <errno.h>
  28 #include <stdio.h>
  29 #include <stdlib.h>
  30 #include <unistd.h>
  31 #include <string.h>
  32 #include <fcntl.h>
  33 #ifdef __linux__
  34 #include <sys/xattr.h>
  35 #endif
  36 #include <sys/stat.h>
  37 #include <sys/types.h>
  38 #include <sys/param.h>
  39
  40 #define TYPE_D 'D'
  41 #define TYPE_F 'F'
  42
  43 static char fdname[MAXPATHLEN] = {0};
  44 static char *pbasedir = NULL;
  45 static int nlevel = 2;
  46 static int ndir = 2;
  47 static int nfile = 2;
  48
  49 static void  usage(char *this);
  50 static void  crtfile(char *pname);
  51 static char *getfdname(char *pdir, char type, int level, int dir, int file);
  52 static int   mktree(char *pbasedir, int level);
  53
  54 int
  55 main(int argc, char *argv[])
  56 {
  57         int c, ret;
  58
  59         while ((c = getopt(argc, argv, "b:l:d:f:")) != -1) {
  60                 switch (c) {
  61                 case 'b':
  62                         pbasedir = optarg;
  63                         break;
  64                 case 'l':
  65                         nlevel = atoi(optarg);
  66                         break;
  67                 case 'd':
  68                         ndir = atoi(optarg);
  69                         break;
  70                 case 'f':
  71                         nfile = atoi(optarg);
  72                         break;
  73                 case '?':
  74                         usage(argv[0]);
  75                 }
  76         }
  77         if (nlevel < 0 || ndir < 0 || nfile < 0 || pbasedir == NULL) {
  78                 usage(argv[0]);
  79         }
  80
  81         ret = mktree(pbasedir, 1);
  82
  83         return (ret);
  84 }
  85
  86 static void
  87 usage(char *this)
  88 {
  89         (void) fprintf(stderr,
  90             "\tUsage: %s -b <base_dir> -l [nlevel] -d [ndir] -f [nfile]\n",
  91             this);
  92         exit(1);
  93 }
  94
  95 static int
  96 mktree(char *pdir, int level)
  97 {
  98         int d, f;
  99         char dname[MAXPATHLEN] = {0};
 100         char fname[MAXPATHLEN] = {0};
 101
 102         if (level > nlevel) {
 103                 return (1);
 104         }
 105
 106         for (d = 0; d < ndir; d++) {
 107                 (void) memset(dname, '\0', sizeof (dname));
 108                 (void) strcpy(dname, getfdname(pdir, TYPE_D, level, d, 0));
 109
 110                 if (mkdir(dname, 0777) != 0) {
 111                         (void) fprintf(stderr, "mkdir(%s) failed."
 112                             "\n[%d]: %s.\n",
 113                             dname, errno, strerror(errno));
 114                         exit(errno);
 115                 }
 116
 117                 /*
 118                  * No sub-directory need be created, only create files in it.
 119                  */
 120                 if (mktree(dname, level+1) != 0) {
 121                         for (f = 0; f < nfile; f++) {
 122                                 (void) memset(fname, '\0', sizeof (fname));
 123                                 (void) strcpy(fname,
 124                                     getfdname(dname, TYPE_F, level+1, d, f));
 125                                 crtfile(fname);
 126                         }
 127                 }
 128         }
 129
 130         for (f = 0; f < nfile; f++) {
 131                 (void) memset(fname, '\0', sizeof (fname));
 132                 (void) strcpy(fname, getfdname(pdir, TYPE_F, level, d, f));
 133                 crtfile(fname);
 134         }
 135
 136         return (0);
 137 }
 138
 139 static char *
 140 getfdname(char *pdir, char type, int level, int dir, int file)
 141 {
 142         size_t size = sizeof (fdname);
 143         if (snprintf(fdname, size, "%s/%c-l%dd%df%d", pdir, type, level, dir,
 144             file) >= size) {
 145                 (void) fprintf(stderr, "fdname truncated\n");
 146                 exit(EINVAL);
 147         }
 148         return (fdname);
 149 }
 150
 151 static void
 152 crtfile(char *pname)
 153 {
 154         int fd = -1;
 155         int i, size;
 156         char *context = "0123456789ABCDF";
 157         char *pbuf;
 158
 159         if (pname == NULL) {
 160                 exit(1);
 161         }
 162
 163         size = sizeof (char) * 1024;
 164         pbuf = (char *)valloc(size);
 165         for (i = 0; i < size / strlen(context); i++) {
 166                 int offset = i * strlen(context);
 167                 (void) snprintf(pbuf+offset, size-offset, "%s", context);
 168         }
 169
 170         if ((fd = open(pname, O_CREAT|O_RDWR, 0777)) < 0) {
 171                 (void) fprintf(stderr, "open(%s, O_CREAT|O_RDWR, 0777) failed."
 172                     "\n[%d]: %s.\n", pname, errno, strerror(errno));
 173                 exit(errno);
 174         }
 175         if (write(fd, pbuf, 1024) < 1024) {
 176                 (void) fprintf(stderr, "write(fd, pbuf, 1024) failed."
 177                     "\n[%d]: %s.\n", errno, strerror(errno));
 178                 exit(errno);
 179         }
 180
 181 #ifdef __linux__
 182         if (fsetxattr(fd, "user.xattr", pbuf, 1024, 0) < 0) {
 183                 (void) fprintf(stderr, "fsetxattr(fd, \"xattr\", pbuf, "
 184                     "1024, 0) failed.\n[%d]: %s.\n", errno, strerror(errno));
 185                 exit(errno);
 186         }
 187 #endif
 188
 189         (void) close(fd);
 190         free(pbuf);
 191 }