makefs.c revision 1.16 1 /* $NetBSD: makefs.c,v 1.16 2003/03/29 00:12:12 thorpej Exp $ */
2
3 /*
4 * Copyright (c) 2001-2003 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.16 2003/03/29 00:12:12 thorpej 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 int x_flag;
74
75 static fstype_t *get_fstype(const char *);
76 static void usage(void);
77 int main(int, char *[]);
78
79 int
80 main(int argc, char *argv[])
81 {
82 struct timeval start;
83 fstype_t *fstype;
84 fsinfo_t fsoptions;
85 fsnode *root;
86 int ch, len;
87 char *specfile;
88
89 setprogname(argv[0]);
90
91 debug = 0;
92 if ((fstype = get_fstype(DEFAULT_FSTYPE)) == NULL)
93 errx(1, "Unknown default fs type `%s'.", DEFAULT_FSTYPE);
94
95 /* set default fsoptions */
96 (void)memset(&fsoptions, 0, sizeof(fsoptions));
97 fsoptions.fd = -1;
98 fsoptions.sectorsize = -1;
99 fsoptions.bsize= -1;
100 fsoptions.fsize= -1;
101 fsoptions.cpg= -1;
102 fsoptions.density= -1;
103 fsoptions.ntracks= -1;
104 fsoptions.nsectors= -1;
105 fsoptions.rpm= -1;
106 fsoptions.minfree= -1;
107 fsoptions.optimization= -1;
108 fsoptions.maxcontig= -1;
109 fsoptions.rotdelay= -1;
110 fsoptions.maxbpg= -1;
111 fsoptions.nrpos= -1;
112 fsoptions.avgfilesize= -1;
113 fsoptions.avgfpdir= -1;
114
115 specfile = NULL;
116 if (gettimeofday(&start, NULL) == -1)
117 err(1, "Unable to get system time");
118
119 start_time.tv_sec = start.tv_sec;
120 start_time.tv_nsec = start.tv_usec * 1000;
121
122 while ((ch = getopt(argc, argv, "B:b:d:f:F:M:m:N:o:s:S:t:x")) != -1) {
123 switch (ch) {
124
125 case 'B':
126 if (strcmp(optarg, "be") == 0 ||
127 strcmp(optarg, "4321") == 0 ||
128 strcmp(optarg, "big") == 0) {
129 #if BYTE_ORDER == LITTLE_ENDIAN
130 fsoptions.needswap = 1;
131 #endif
132 } else if (strcmp(optarg, "le") == 0 ||
133 strcmp(optarg, "1234") == 0 ||
134 strcmp(optarg, "little") == 0) {
135 #if BYTE_ORDER == BIG_ENDIAN
136 fsoptions.needswap = 1;
137 #endif
138 } else {
139 warnx("Invalid endian `%s'.", optarg);
140 usage();
141 }
142 break;
143
144 case 'b':
145 len = strlen(optarg) - 1;
146 if (optarg[len] == '%') {
147 optarg[len] = '\0';
148 fsoptions.freeblockpc =
149 strsuftoll("free block percentage",
150 optarg, 0, 99);
151 } else {
152 fsoptions.freeblocks =
153 strsuftoll("free blocks",
154 optarg, 0, LLONG_MAX);
155 }
156 break;
157
158 case 'd':
159 debug =
160 (int)strsuftoll("debug mask", optarg, 0, UINT_MAX);
161 break;
162
163 case 'f':
164 len = strlen(optarg) - 1;
165 if (optarg[len] == '%') {
166 optarg[len] = '\0';
167 fsoptions.freefilepc =
168 strsuftoll("free file percentage",
169 optarg, 0, 99);
170 } else {
171 fsoptions.freefiles =
172 strsuftoll("free files",
173 optarg, 0, LLONG_MAX);
174 }
175 break;
176
177 case 'F':
178 specfile = optarg;
179 break;
180
181 case 'M':
182 fsoptions.minsize =
183 strsuftoll("minimum size", optarg, 1LL, LLONG_MAX);
184 break;
185
186 case 'N':
187 if (! setup_getid(optarg))
188 errx(1,
189 "Unable to use user and group databases in `%s'",
190 optarg);
191 break;
192
193 case 'm':
194 fsoptions.maxsize =
195 strsuftoll("maximum size", optarg, 1LL, LLONG_MAX);
196 break;
197
198 case 'o':
199 {
200 char *p;
201
202 while ((p = strsep(&optarg, ",")) != NULL) {
203 if (*p == '\0')
204 errx(1, "Empty option");
205 if (! fstype->parse_options(p, &fsoptions))
206 usage();
207 }
208 break;
209 }
210
211 case 's':
212 fsoptions.minsize = fsoptions.maxsize =
213 strsuftoll("size", optarg, 1LL, LLONG_MAX);
214 break;
215
216 case 'S':
217 fsoptions.sectorsize =
218 (int)strsuftoll("sector size", optarg,
219 1LL, INT_MAX);
220 break;
221
222 case 't':
223 if ((fstype = get_fstype(optarg)) == NULL)
224 errx(1, "Unknown fs type `%s'.", optarg);
225 break;
226
227 case 'x':
228 x_flag = 1;
229 break;
230
231 case '?':
232 default:
233 usage();
234 /* NOTREACHED */
235
236 }
237 }
238 if (debug) {
239 printf("debug mask: 0x%08x\n", debug);
240 printf("start time: %ld.%ld, %s",
241 (long)start_time.tv_sec, (long)start_time.tv_nsec,
242 ctime(&start_time.tv_sec));
243 }
244 argc -= optind;
245 argv += optind;
246
247 if (argc != 2)
248 usage();
249
250 /* -x must be accompanied by -F */
251 if (x_flag != 0 && specfile == NULL)
252 errx(1, "-x requires -F mtree-specfile.");
253
254 /* walk the tree */
255 TIMER_START(start);
256 root = walk_dir(argv[1], NULL);
257 TIMER_RESULTS(start, "walk_dir");
258
259 if (specfile) { /* apply a specfile */
260 TIMER_START(start);
261 apply_specfile(specfile, argv[1], root);
262 TIMER_RESULTS(start, "apply_specfile");
263 }
264
265 if (debug & DEBUG_DUMP_FSNODES) {
266 printf("\nparent: %s\n", argv[1]);
267 dump_fsnodes(".", root);
268 putchar('\n');
269 }
270
271 /* build the file system */
272 TIMER_START(start);
273 fstype->make_fs(argv[0], argv[1], root, &fsoptions);
274 TIMER_RESULTS(start, "make_fs");
275
276 exit(0);
277 /* NOTREACHED */
278 }
279
280
281 int
282 set_option(option_t *options, const char *var, const char *val)
283 {
284 int i;
285
286 for (i = 0; options[i].name != NULL; i++) {
287 if (strcmp(options[i].name, var) != 0)
288 continue;
289 *options[i].value = (int)strsuftoll(options[i].desc, val,
290 options[i].minimum, options[i].maximum);
291 return (1);
292 }
293 warnx("Unknown option `%s'", var);
294 return (0);
295 }
296
297
298 static fstype_t *
299 get_fstype(const char *type)
300 {
301 int i;
302
303 for (i = 0; fstypes[i].type != NULL; i++)
304 if (strcmp(fstypes[i].type, type) == 0)
305 return (&fstypes[i]);
306 return (NULL);
307 }
308
309 static void
310 usage(void)
311 {
312 const char *prog;
313
314 prog = getprogname();
315 fprintf(stderr,
316 "Usage: %s [-t fs-type] [-o fs-options] [-d debug-mask] [-B endian]\n"
317 "\t[-S sector-size] [-M minimum-size] [-m maximum-size] [-s image-size]\n"
318 "\t[-b free-blocks] [-f free-files] [-F mtree-specfile] [-N userdb-dir]\n"
319 "\timage-file directory\n",
320 prog);
321 exit(1);
322 }
323