xref: /freebsd/usr.sbin/makefs/makefs.c (revision 195ebc7e9e4b129de810833791a19dfb4349d6a9)
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$");
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  */
56 typedef 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 
63 static fstype_t fstypes[] = {
64 	{ "ffs",	ffs_parse_opts,		ffs_makefs },
65 	{ NULL	},
66 };
67 
68 u_int		debug;
69 struct timespec	start_time;
70 
71 static	fstype_t *get_fstype(const char *);
72 static	void	usage(void);
73 int		main(int, char *[]);
74 
75 int
76 main(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 
273 int
274 set_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 
290 static fstype_t *
291 get_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 
301 static void
302 usage(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