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