makefs.c revision 1.14 1 /* $NetBSD: makefs.c,v 1.14 2002/11/29 13:06:33 lukem Exp $ */
2
3 /*
4 * Copyright (c) 2001-2002 Wasabi Systems, Inc.
5 * All rights reserved.
6 *
7 * Written by Luke Mewburn for Wasabi Systems, Inc.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in the
16 * documentation and/or other materials provided with the distribution.
17 * 3. All advertising materials mentioning features or use of this software
18 * must display the following acknowledgement:
19 * This product includes software developed for the NetBSD Project by
20 * Wasabi Systems, Inc.
21 * 4. The name of Wasabi Systems, Inc. may not be used to endorse
22 * or promote products derived from this software without specific prior
23 * written permission.
24 *
25 * THIS SOFTWARE IS PROVIDED BY WASABI SYSTEMS, INC. ``AS IS'' AND
26 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
27 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL WASABI SYSTEMS, INC
29 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
30 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
31 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
32 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
33 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
34 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
35 * POSSIBILITY OF SUCH DAMAGE.
36 */
37
38 #include <sys/cdefs.h>
39 #if defined(__RCSID) && !defined(__lint)
40 __RCSID("$NetBSD: makefs.c,v 1.14 2002/11/29 13:06:33 lukem Exp $");
41 #endif /* !__lint */
42
43 #include <assert.h>
44 #include <ctype.h>
45 #include <errno.h>
46 #include <limits.h>
47 #include <stdio.h>
48 #include <stdlib.h>
49 #include <string.h>
50 #include <unistd.h>
51
52 #include "makefs.h"
53 #include "mtree.h"
54
55 /*
56 * list of supported file systems and dispatch functions
57 */
58 typedef struct {
59 const char *type;
60 int (*parse_options)(const char *, fsinfo_t *);
61 void (*make_fs)(const char *, const char *, fsnode *,
62 fsinfo_t *);
63 } fstype_t;
64
65 static fstype_t fstypes[] = {
66 { "ffs", ffs_parse_opts, ffs_makefs },
67 { NULL },
68 };
69
70 uint debug;
71 struct timespec start_time;
72
73
74 static fstype_t *get_fstype(const char *);
75 static void usage(void);
76 int main(int, char *[]);
77
78 int
79 main(int argc, char *argv[])
80 {
81 struct timeval start;
82 fstype_t *fstype;
83 fsinfo_t fsoptions;
84 fsnode *root;
85 int ch, len;
86 char *specfile;
87
88 setprogname(argv[0]);
89
90 debug = 0;
91 if ((fstype = get_fstype(DEFAULT_FSTYPE)) == NULL)
92 errx(1, "Unknown default fs type `%s'.", DEFAULT_FSTYPE);
93
94 /* set default fsoptions */
95 (void)memset(&fsoptions, 0, sizeof(fsoptions));
96 fsoptions.fd = -1;
97 fsoptions.sectorsize = -1;
98 fsoptions.bsize= -1;
99 fsoptions.fsize= -1;
100 fsoptions.cpg= -1;
101 fsoptions.density= -1;
102 fsoptions.ntracks= -1;
103 fsoptions.nsectors= -1;
104 fsoptions.rpm= -1;
105 fsoptions.minfree= -1;
106 fsoptions.optimization= -1;
107 fsoptions.maxcontig= -1;
108 fsoptions.rotdelay= -1;
109 fsoptions.maxbpg= -1;
110 fsoptions.nrpos= -1;
111 fsoptions.avgfilesize= -1;
112 fsoptions.avgfpdir= -1;
113
114 specfile = NULL;
115 if (gettimeofday(&start, NULL) == -1)
116 err(1, "Unable to get system time");
117
118 start_time.tv_sec = start.tv_sec;
119 start_time.tv_nsec = start.tv_usec * 1000;
120
121 while ((ch = getopt(argc, argv, "B:b:d:f:F:M:m:N:o:s:S:t:")) != -1) {
122 switch (ch) {
123
124 case 'B':
125 if (strcmp(optarg, "be") == 0 ||
126 strcmp(optarg, "big") == 0) {
127 #if BYTE_ORDER == LITTLE_ENDIAN
128 fsoptions.needswap = 1;
129 #endif
130 } else if (strcmp(optarg, "le") == 0 ||
131 strcmp(optarg, "little") == 0) {
132 #if BYTE_ORDER == BIG_ENDIAN
133 fsoptions.needswap = 1;
134 #endif
135 } else {
136 warnx("Invalid endian `%s'.", optarg);
137 usage();
138 }
139 break;
140
141 case 'b':
142 len = strlen(optarg) - 1;
143 if (optarg[len] == '%') {
144 optarg[len] = '\0';
145 fsoptions.freeblockpc =
146 strsuftoll("free block percentage",
147 optarg, 0, 99);
148 } else {
149 fsoptions.freeblocks =
150 strsuftoll("free blocks",
151 optarg, 0, LLONG_MAX);
152 }
153 break;
154
155 case 'd':
156 debug =
157 (int)strsuftoll("debug mask", optarg, 0, UINT_MAX);
158 break;
159
160 case 'f':
161 len = strlen(optarg) - 1;
162 if (optarg[len] == '%') {
163 optarg[len] = '\0';
164 fsoptions.freefilepc =
165 strsuftoll("free file percentage",
166 optarg, 0, 99);
167 } else {
168 fsoptions.freefiles =
169 strsuftoll("free files",
170 optarg, 0, LLONG_MAX);
171 }
172 break;
173
174 case 'F':
175 specfile = optarg;
176 break;
177
178 case 'M':
179 fsoptions.minsize =
180 strsuftoll("minimum size", optarg, 1LL, LLONG_MAX);
181 break;
182
183 case 'N':
184 if (! setup_getid(optarg))
185 errx(1,
186 "Unable to use user and group databases in `%s'",
187 optarg);
188 break;
189
190 case 'm':
191 fsoptions.maxsize =
192 strsuftoll("maximum size", optarg, 1LL, LLONG_MAX);
193 break;
194
195 case 'o':
196 {
197 char *p;
198
199 while ((p = strsep(&optarg, ",")) != NULL) {
200 if (*p == '\0')
201 errx(1, "Empty option");
202 if (! fstype->parse_options(p, &fsoptions))
203 usage();
204 }
205 break;
206 }
207
208 case 's':
209 fsoptions.minsize = fsoptions.maxsize =
210 strsuftoll("size", optarg, 1LL, LLONG_MAX);
211 break;
212
213 case 'S':
214 fsoptions.sectorsize =
215 (int)strsuftoll("sector size", optarg,
216 1LL, INT_MAX);
217 break;
218
219 case 't':
220 if ((fstype = get_fstype(optarg)) == NULL)
221 errx(1, "Unknown fs type `%s'.", optarg);
222 break;
223
224 case '?':
225 default:
226 usage();
227 /* NOTREACHED */
228
229 }
230 }
231 if (debug) {
232 printf("debug mask: 0x%08x\n", debug);
233 printf("start time: %ld.%ld, %s",
234 (long)start_time.tv_sec, (long)start_time.tv_nsec,
235 ctime(&start_time.tv_sec));
236 }
237 argc -= optind;
238 argv += optind;
239
240 if (argc != 2)
241 usage();
242
243 /* walk the tree */
244 TIMER_START(start);
245 root = walk_dir(argv[1], NULL);
246 TIMER_RESULTS(start, "walk_dir");
247
248 if (specfile) { /* apply a specfile */
249 TIMER_START(start);
250 apply_specfile(specfile, argv[1], root);
251 TIMER_RESULTS(start, "apply_specfile");
252 }
253
254 if (debug & DEBUG_DUMP_FSNODES) {
255 printf("\nparent: %s\n", argv[1]);
256 dump_fsnodes(".", root);
257 putchar('\n');
258 }
259
260 /* build the file system */
261 TIMER_START(start);
262 fstype->make_fs(argv[0], argv[1], root, &fsoptions);
263 TIMER_RESULTS(start, "make_fs");
264
265 exit(0);
266 /* NOTREACHED */
267 }
268
269
270 int
271 set_option(option_t *options, const char *var, const char *val)
272 {
273 int i;
274
275 for (i = 0; options[i].name != NULL; i++) {
276 if (strcmp(options[i].name, var) != 0)
277 continue;
278 *options[i].value = (int)strsuftoll(options[i].desc, val,
279 options[i].minimum, options[i].maximum);
280 return (1);
281 }
282 warnx("Unknown option `%s'", var);
283 return (0);
284 }
285
286
287 static fstype_t *
288 get_fstype(const char *type)
289 {
290 int i;
291
292 for (i = 0; fstypes[i].type != NULL; i++)
293 if (strcmp(fstypes[i].type, type) == 0)
294 return (&fstypes[i]);
295 return (NULL);
296 }
297
298 static void
299 usage(void)
300 {
301 const char *prog;
302
303 prog = getprogname();
304 fprintf(stderr,
305 "Usage: %s [-t fs-type] [-o fs-options] [-d debug-mask] [-B endian]\n"
306 "\t[-S sector-size] [-M minimum-size] [-m maximum-size] [-s image-size]\n"
307 "\t[-b free-blocks] [-f free-files] [-F mtree-specfile] [-N userdb-dir]\n"
308 "\timage-file directory\n",
309 prog);
310 exit(1);
311 }
312