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
main(int argc,char * argv[])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
usage(char * this)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
mktree(char * pdir,int level)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 *
getfdname(char * pdir,char type,int level,int dir,int file)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
crtfile(char * pname)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