mktree.c

Version: - FREEBSD - FREEBSD-13-STABLE - FREEBSD-13-0 - FREEBSD-12-STABLE - FREEBSD-12-0 - FREEBSD-11-STABLE - FREEBSD-11-0 - FREEBSD-10-STABLE - FREEBSD-10-0 - FREEBSD-9-STABLE - FREEBSD-9-0 - FREEBSD-8-STABLE - FREEBSD-8-0 - FREEBSD-7-STABLE - FREEBSD-7-0 - FREEBSD-6-STABLE - FREEBSD-6-0 - FREEBSD-5-STABLE - FREEBSD-5-0 - FREEBSD-4-STABLE - FREEBSD-3-STABLE - FREEBSD22 - l41 - OPENBSD - linux-2.6 - MK84 - PLAN9 - xnu-8792
SearchContext: - none - 3 - 10

    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 https://opensource.org/licenses/CDDL-1.0.
   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         const 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 }

Cache object: 5ac7fbd3224ba406d9fafe60e081bacf

[ source navigation ] [ diff markup ] [ identifier search ] [ freetext search ] [ file search ] [ list types ] [ track identifier ]

This page is part of the FreeBSD/Linux Linux Kernel Cross-Reference, and was automatically generated using a modified version of the LXR engine.

FreeBSD/Linux Kernel Cross Reference sys/contrib/openzfs/tests/zfs-tests/cmd/mktree.c

FreeBSD/Linux Kernel Cross Reference
sys/contrib/openzfs/tests/zfs-tests/cmd/mktree.c