1 /* $NetBSD: makefs.c,v 1.26 2006/10/22 21:11:56 christos 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 __FBSDID("$FreeBSD: stable/10/usr.sbin/makefs/makefs.c 293025 2016-01-01 00:36:59Z ngie $");
40
41 #include <sys/types.h>
42 #include <sys/stat.h>
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 <time.h>
51 #include <unistd.h>
52
53 #include "makefs.h"
54 #include "mtree.h"
55
56 /*
57 * list of supported file systems and dispatch functions
58 */
59 typedef struct {
60 const char *type;
61 void (*prepare_options)(fsinfo_t *);
62 int (*parse_options)(const char *, fsinfo_t *);
63 void (*cleanup_options)(fsinfo_t *);
64 void (*make_fs)(const char *, const char *, fsnode *,
65 fsinfo_t *);
66 } fstype_t;
67
68 static fstype_t fstypes[] = {
69 { "ffs", ffs_prep_opts, ffs_parse_opts, ffs_cleanup_opts, ffs_makefs },
70 { "cd9660", cd9660_prep_opts, cd9660_parse_opts, cd9660_cleanup_opts,
71 cd9660_makefs},
72 { .type = NULL },
73 };
74
75 u_int debug;
76 int dupsok;
77 struct timespec start_time;
78
79 static fstype_t *get_fstype(const char *);
80 static void usage(void);
81 int main(int, char *[]);
82
83 int
main(int argc,char * argv[])84 main(int argc, char *argv[])
85 {
86 struct stat sb;
87 struct timeval start;
88 fstype_t *fstype;
89 fsinfo_t fsoptions;
90 fsnode *root;
91 int ch, i, len;
92 char *subtree;
93 char *specfile;
94
95 setprogname(argv[0]);
96
97 debug = 0;
98 if ((fstype = get_fstype(DEFAULT_FSTYPE)) == NULL)
99 errx(1, "Unknown default fs type `%s'.", DEFAULT_FSTYPE);
100
101 /* set default fsoptions */
102 (void)memset(&fsoptions, 0, sizeof(fsoptions));
103 fsoptions.fd = -1;
104 fsoptions.sectorsize = -1;
105
106 if (fstype->prepare_options)
107 fstype->prepare_options(&fsoptions);
108
109 specfile = NULL;
110 if (gettimeofday(&start, NULL) == -1)
111 err(1, "Unable to get system time");
112
113 start_time.tv_sec = start.tv_sec;
114 start_time.tv_nsec = start.tv_usec * 1000;
115
116 while ((ch = getopt(argc, argv, "B:b:Dd:f:F:M:m:N:o:pR:s:S:t:xZ")) != -1) {
117 switch (ch) {
118
119 case 'B':
120 if (strcmp(optarg, "be") == 0 ||
121 strcmp(optarg, "4321") == 0 ||
122 strcmp(optarg, "big") == 0) {
123 #if BYTE_ORDER == LITTLE_ENDIAN
124 fsoptions.needswap = 1;
125 #endif
126 } else if (strcmp(optarg, "le") == 0 ||
127 strcmp(optarg, "1234") == 0 ||
128 strcmp(optarg, "little") == 0) {
129 #if BYTE_ORDER == BIG_ENDIAN
130 fsoptions.needswap = 1;
131 #endif
132 } else {
133 warnx("Invalid endian `%s'.", optarg);
134 usage();
135 }
136 break;
137
138 case 'b':
139 len = strlen(optarg) - 1;
140 if (optarg[len] == '%') {
141 optarg[len] = '\0';
142 fsoptions.freeblockpc =
143 strsuftoll("free block percentage",
144 optarg, 0, 99);
145 } else {
146 fsoptions.freeblocks =
147 strsuftoll("free blocks",
148 optarg, 0, LLONG_MAX);
149 }
150 break;
151
152 case 'D':
153 dupsok = 1;
154 break;
155
156 case 'd':
157 debug = strtoll(optarg, NULL, 0);
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 case 'p':
208 /* Deprecated in favor of 'Z' */
209 fsoptions.sparse = 1;
210 break;
211
212 case 'R':
213 /* Round image size up to specified block size */
214 fsoptions.roundup =
215 strsuftoll("roundup-size", optarg, 0, LLONG_MAX);
216 break;
217
218 case 's':
219 fsoptions.minsize = fsoptions.maxsize =
220 strsuftoll("size", optarg, 1LL, LLONG_MAX);
221 break;
222
223 case 'S':
224 fsoptions.sectorsize =
225 (int)strsuftoll("sector size", optarg,
226 1LL, INT_MAX);
227 break;
228
229 case 't':
230 /* Check current one and cleanup if necessary. */
231 if (fstype->cleanup_options)
232 fstype->cleanup_options(&fsoptions);
233 fsoptions.fs_specific = NULL;
234 if ((fstype = get_fstype(optarg)) == NULL)
235 errx(1, "Unknown fs type `%s'.", optarg);
236 fstype->prepare_options(&fsoptions);
237 break;
238
239 case 'x':
240 fsoptions.onlyspec = 1;
241 break;
242
243 case 'Z':
244 /* Superscedes 'p' for compatibility with NetBSD makefs(8) */
245 fsoptions.sparse = 1;
246 break;
247
248 case '?':
249 default:
250 usage();
251 /* NOTREACHED */
252
253 }
254 }
255 if (debug) {
256 printf("debug mask: 0x%08x\n", debug);
257 printf("start time: %ld.%ld, %s",
258 (long)start_time.tv_sec, (long)start_time.tv_nsec,
259 ctime(&start_time.tv_sec));
260 }
261 argc -= optind;
262 argv += optind;
263
264 if (argc < 2)
265 usage();
266
267 /* -x must be accompanied by -F */
268 if (fsoptions.onlyspec != 0 && specfile == NULL)
269 errx(1, "-x requires -F mtree-specfile.");
270
271 /* Accept '-' as meaning "read from standard input". */
272 if (strcmp(argv[1], "-") == 0)
273 sb.st_mode = S_IFREG;
274 else {
275 if (stat(argv[1], &sb) == -1)
276 err(1, "Can't stat `%s'", argv[1]);
277 }
278
279 switch (sb.st_mode & S_IFMT) {
280 case S_IFDIR: /* walk the tree */
281 subtree = argv[1];
282 TIMER_START(start);
283 root = walk_dir(subtree, ".", NULL, NULL);
284 TIMER_RESULTS(start, "walk_dir");
285 break;
286 case S_IFREG: /* read the manifest file */
287 subtree = ".";
288 TIMER_START(start);
289 root = read_mtree(argv[1], NULL);
290 TIMER_RESULTS(start, "manifest");
291 break;
292 default:
293 errx(1, "%s: not a file or directory", argv[1]);
294 /* NOTREACHED */
295 }
296
297 /* append extra directory */
298 for (i = 2; i < argc; i++) {
299 if (stat(argv[i], &sb) == -1)
300 err(1, "Can't stat `%s'", argv[i]);
301 if (!S_ISDIR(sb.st_mode))
302 errx(1, "%s: not a directory", argv[i]);
303 TIMER_START(start);
304 root = walk_dir(argv[i], ".", NULL, root);
305 TIMER_RESULTS(start, "walk_dir2");
306 }
307
308 if (specfile) { /* apply a specfile */
309 TIMER_START(start);
310 apply_specfile(specfile, subtree, root, fsoptions.onlyspec);
311 TIMER_RESULTS(start, "apply_specfile");
312 }
313
314 if (debug & DEBUG_DUMP_FSNODES) {
315 printf("\nparent: %s\n", subtree);
316 dump_fsnodes(root);
317 putchar('\n');
318 }
319
320 /* build the file system */
321 TIMER_START(start);
322 fstype->make_fs(argv[0], subtree, root, &fsoptions);
323 TIMER_RESULTS(start, "make_fs");
324
325 free_fsnodes(root);
326
327 exit(0);
328 /* NOTREACHED */
329 }
330
331
332 int
set_option(option_t * options,const char * var,const char * val)333 set_option(option_t *options, const char *var, const char *val)
334 {
335 int i;
336
337 for (i = 0; options[i].name != NULL; i++) {
338 if (strcmp(options[i].name, var) != 0)
339 continue;
340 *options[i].value = (int)strsuftoll(options[i].desc, val,
341 options[i].minimum, options[i].maximum);
342 return (1);
343 }
344 warnx("Unknown option `%s'", var);
345 return (0);
346 }
347
348
349 static fstype_t *
get_fstype(const char * type)350 get_fstype(const char *type)
351 {
352 int i;
353
354 for (i = 0; fstypes[i].type != NULL; i++)
355 if (strcmp(fstypes[i].type, type) == 0)
356 return (&fstypes[i]);
357 return (NULL);
358 }
359
360 static void
usage(void)361 usage(void)
362 {
363 const char *prog;
364
365 prog = getprogname();
366 fprintf(stderr,
367 "usage: %s [-t fs-type] [-o fs-options] [-d debug-mask] [-B endian]\n"
368 "\t[-S sector-size] [-M minimum-size] [-m maximum-size] [-R roundup-size]\n"
369 "\t[-s image-size] [-b free-blocks] [-f free-files] [-F mtree-specfile]\n"
370 "\t[-xZ] [-N userdb-dir] image-file directory | manifest [extra-directory ...]\n",
371 prog);
372 exit(1);
373 }
374