makefs.c revision 1.13 1 /* $NetBSD: makefs.c,v 1.13 2002/01/31 22:44:02 tv 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.13 2002/01/31 22:44:02 tv 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 #include "strsuftoull.h"
55
56 /*
57 * list of supported file systems and dispatch functions
58 */
59 typedef struct {
60 const char *type;
61 int (*parse_options)(const char *, fsinfo_t *);
62 void (*make_fs)(const char *, const char *, fsnode *,
63 fsinfo_t *);
64 } fstype_t;
65
66 static fstype_t fstypes[] = {
67 { "ffs", ffs_parse_opts, ffs_makefs },
68 { NULL },
69 };
70
71 uint debug;
72 struct timespec start_time;
73
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:")) != -1) {
123 switch (ch) {
124
125 case 'B':
126 if (strcmp(optarg, "be") == 0 ||
127 strcmp(optarg, "big") == 0) {
128 #if BYTE_ORDER == LITTLE_ENDIAN
129 fsoptions.needswap = 1;
130 #endif
131 } else if (strcmp(optarg, "le") == 0 ||
132 strcmp(optarg, "little") == 0) {
133 #if BYTE_ORDER == BIG_ENDIAN
134 fsoptions.needswap = 1;
135 #endif
136 } else {
137 warnx("Invalid endian `%s'.", optarg);
138 usage();
139 }
140 break;
141
142 case 'b':
143 len = strlen(optarg) - 1;
144 if (optarg[len] == '%') {
145 optarg[len] = '\0';
146 fsoptions.freeblockpc =
147 strsuftoull("free block percentage",
148 optarg, 0, 99);
149 } else {
150 fsoptions.freeblocks =
151 strsuftoull("free blocks",
152 optarg, 0, LLONG_MAX);
153 }
154 break;
155
156 case 'd':
157 debug =
158 (int)strsuftoull("debug mask", optarg, 0, UINT_MAX);
159 break;
160
161 case 'f':
162 len = strlen(optarg) - 1;
163 if (optarg[len] == '%') {
164 optarg[len] = '\0';
165 fsoptions.freefilepc =
166 strsuftoull("free file percentage",
167 optarg, 0, 99);
168 } else {
169 fsoptions.freefiles =
170 strsuftoull("free files",
171 optarg, 0, LLONG_MAX);
172 }
173 break;
174
175 case 'F':
176 specfile = optarg;
177 break;
178
179 case 'M':
180 fsoptions.minsize =
181 strsuftoull("minimum size", optarg, 1LL, LLONG_MAX);
182 break;
183
184 case 'N':
185 if (! setup_getid(optarg))
186 errx(1,
187 "Unable to use user and group databases in `%s'",
188 optarg);
189 break;
190
191 case 'm':
192 fsoptions.maxsize =
193 strsuftoull("maximum size", optarg, 1LL, LLONG_MAX);
194 break;
195
196 case 'o':
197 {
198 char *p;
199
200 while ((p = strsep(&optarg, ",")) != NULL) {
201 if (*p == '\0')
202 errx(1, "Empty option");
203 if (! fstype->parse_options(p, &fsoptions))
204 usage();
205 }
206 break;
207 }
208
209 case 's':
210 fsoptions.minsize = fsoptions.maxsize =
211 strsuftoull("size", optarg, 1LL, LLONG_MAX);
212 break;
213
214 case 'S':
215 fsoptions.sectorsize =
216 (int)strsuftoull("sector size", optarg,
217 1LL, INT_MAX);
218 break;
219
220 case 't':
221 if ((fstype = get_fstype(optarg)) == NULL)
222 errx(1, "Unknown fs type `%s'.", optarg);
223 break;
224
225 case '?':
226 default:
227 usage();
228 /* NOTREACHED */
229
230 }
231 }
232 if (debug) {
233 printf("debug mask: 0x%08x\n", debug);
234 printf("start time: %ld.%ld, %s",
235 (long)start_time.tv_sec, (long)start_time.tv_nsec,
236 ctime(&start_time.tv_sec));
237 }
238 argc -= optind;
239 argv += optind;
240
241 if (argc != 2)
242 usage();
243
244 /* walk the tree */
245 TIMER_START(start);
246 root = walk_dir(argv[1], NULL);
247 TIMER_RESULTS(start, "walk_dir");
248
249 if (specfile) { /* apply a specfile */
250 TIMER_START(start);
251 apply_specfile(specfile, argv[1], root);
252 TIMER_RESULTS(start, "apply_specfile");
253 }
254
255 if (debug & DEBUG_DUMP_FSNODES) {
256 printf("\nparent: %s\n", argv[1]);
257 dump_fsnodes(".", root);
258 putchar('\n');
259 }
260
261 /* build the file system */
262 TIMER_START(start);
263 fstype->make_fs(argv[0], argv[1], root, &fsoptions);
264 TIMER_RESULTS(start, "make_fs");
265
266 exit(0);
267 /* NOTREACHED */
268 }
269
270
271 int
272 set_option(option_t *options, const char *var, const char *val)
273 {
274 int i;
275
276 for (i = 0; options[i].name != NULL; i++) {
277 if (strcmp(options[i].name, var) != 0)
278 continue;
279 *options[i].value = (int)strsuftoull(options[i].desc, val,
280 options[i].minimum, options[i].maximum);
281 return (1);
282 }
283 warnx("Unknown option `%s'", var);
284 return (0);
285 }
286
287
288 static fstype_t *
289 get_fstype(const char *type)
290 {
291 int i;
292
293 for (i = 0; fstypes[i].type != NULL; i++)
294 if (strcmp(fstypes[i].type, type) == 0)
295 return (&fstypes[i]);
296 return (NULL);
297 }
298
299 static void
300 usage(void)
301 {
302 const char *prog;
303
304 prog = getprogname();
305 fprintf(stderr,
306 "Usage: %s [-t fs-type] [-o fs-options] [-d debug-mask] [-B endian]\n"
307 "\t[-S sector-size] [-M minimum-size] [-m maximum-size] [-s image-size]\n"
308 "\t[-b free-blocks] [-f free-files] [-F mtree-specfile] [-N userdb-dir]\n"
309 "\timage-file directory\n",
310 prog);
311 exit(1);
312 }
313