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 } 192