1/*	$NetBSD: makefs.c,v 1.26 2006/10/22 21:11:56 christos Exp $	*/
2
3/*-
4 * SPDX-License-Identifier: BSD-4-Clause
5 *
6 * Copyright (c) 2001-2003 Wasabi Systems, Inc.
7 * All rights reserved.
8 *
9 * Written by Luke Mewburn for Wasabi Systems, Inc.
10 *
11 * Redistribution and use in source and binary forms, with or without
12 * modification, are permitted provided that the following conditions
13 * are met:
14 * 1. Redistributions of source code must retain the above copyright
15 *    notice, this list of conditions and the following disclaimer.
16 * 2. Redistributions in binary form must reproduce the above copyright
17 *    notice, this list of conditions and the following disclaimer in the
18 *    documentation and/or other materials provided with the distribution.
19 * 3. All advertising materials mentioning features or use of this software
20 *    must display the following acknowledgement:
21 *      This product includes software developed for the NetBSD Project by
22 *      Wasabi Systems, Inc.
23 * 4. The name of Wasabi Systems, Inc. may not be used to endorse
24 *    or promote products derived from this software without specific prior
25 *    written permission.
26 *
27 * THIS SOFTWARE IS PROVIDED BY WASABI SYSTEMS, INC. ``AS IS'' AND
28 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
29 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
30 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL WASABI SYSTEMS, INC
31 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
32 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
33 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
34 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
35 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
36 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
37 * POSSIBILITY OF SUCH DAMAGE.
38 */
39
40#include <sys/types.h>
41#include <sys/stat.h>
42#include <assert.h>
43#include <ctype.h>
44#include <errno.h>
45#include <limits.h>
46#include <stdio.h>
47#include <stdlib.h>
48#include <string.h>
49#include <time.h>
50#include <unistd.h>
51#include <stdbool.h>
52#include <util.h>
53
54#include "makefs.h"
55#include "mtree.h"
56
57/*
58 * list of supported file systems and dispatch functions
59 */
60typedef struct {
61	const char	*type;
62	void		(*prepare_options)(fsinfo_t *);
63	int		(*parse_options)(const char *, fsinfo_t *);
64	void		(*cleanup_options)(fsinfo_t *);
65	void		(*make_fs)(const char *, const char *, fsnode *,
66				fsinfo_t *);
67} fstype_t;
68
69static fstype_t fstypes[] = {
70#define ENTRY(name) { \
71	# name, name ## _prep_opts, name ## _parse_opts, \
72	name ## _cleanup_opts, name ## _makefs  \
73}
74	ENTRY(cd9660),
75	ENTRY(ffs),
76	ENTRY(msdos),
77#ifdef HAVE_ZFS
78	ENTRY(zfs),
79#endif
80	{ .type = NULL	},
81};
82
83u_int		debug;
84int		dupsok;
85struct timespec	start_time;
86struct stat stampst;
87
88static	fstype_t *get_fstype(const char *);
89static int get_tstamp(const char *, struct stat *);
90static	void	usage(fstype_t *, fsinfo_t *);
91
92int
93main(int argc, char *argv[])
94{
95	struct stat	 sb;
96	struct timeval	 start;
97	fstype_t	*fstype;
98	fsinfo_t	 fsoptions;
99	fsnode		*root;
100	int		 ch, i, len;
101	const char	*subtree;
102	const char	*specfile;
103
104	setprogname(argv[0]);
105
106	debug = 0;
107	if ((fstype = get_fstype(DEFAULT_FSTYPE)) == NULL)
108		errx(1, "Unknown default fs type `%s'.", DEFAULT_FSTYPE);
109
110		/* set default fsoptions */
111	(void)memset(&fsoptions, 0, sizeof(fsoptions));
112	fsoptions.fd = -1;
113	fsoptions.sectorsize = -1;
114
115	if (fstype->prepare_options)
116		fstype->prepare_options(&fsoptions);
117
118	specfile = NULL;
119#ifdef CLOCK_REALTIME
120	ch = clock_gettime(CLOCK_REALTIME, &start_time);
121#else
122	ch = gettimeofday(&start, NULL);
123	start_time.tv_sec = start.tv_sec;
124	start_time.tv_nsec = start.tv_usec * 1000;
125#endif
126	if (ch == -1)
127		err(1, "Unable to get system time");
128
129
130	while ((ch = getopt(argc, argv, "B:b:Dd:f:F:M:m:N:O:o:pR:s:S:t:T:xZ")) != -1) {
131		switch (ch) {
132
133		case 'B':
134			if (strcmp(optarg, "be") == 0 ||
135			    strcmp(optarg, "4321") == 0 ||
136			    strcmp(optarg, "big") == 0) {
137#if BYTE_ORDER == LITTLE_ENDIAN
138				fsoptions.needswap = 1;
139#endif
140			} else if (strcmp(optarg, "le") == 0 ||
141			    strcmp(optarg, "1234") == 0 ||
142			    strcmp(optarg, "little") == 0) {
143#if BYTE_ORDER == BIG_ENDIAN
144				fsoptions.needswap = 1;
145#endif
146			} else {
147				warnx("Invalid endian `%s'.", optarg);
148				usage(fstype, &fsoptions);
149			}
150			break;
151
152		case 'b':
153			len = strlen(optarg) - 1;
154			if (optarg[len] == '%') {
155				optarg[len] = '\0';
156				fsoptions.freeblockpc =
157				    strsuftoll("free block percentage",
158					optarg, 0, 99);
159			} else {
160				fsoptions.freeblocks =
161				    strsuftoll("free blocks",
162					optarg, 0, LLONG_MAX);
163			}
164			break;
165
166		case 'D':
167			dupsok++;
168			break;
169
170		case 'd':
171			debug = strtoll(optarg, NULL, 0);
172			break;
173
174		case 'f':
175			len = strlen(optarg) - 1;
176			if (optarg[len] == '%') {
177				optarg[len] = '\0';
178				fsoptions.freefilepc =
179				    strsuftoll("free file percentage",
180					optarg, 0, 99);
181			} else {
182				fsoptions.freefiles =
183				    strsuftoll("free files",
184					optarg, 0, LLONG_MAX);
185			}
186			break;
187
188		case 'F':
189			specfile = optarg;
190			break;
191
192		case 'M':
193			fsoptions.minsize =
194			    strsuftoll("minimum size", optarg, 1LL, LLONG_MAX);
195			break;
196
197		case 'N':
198			if (! setup_getid(optarg))
199				errx(1,
200			    "Unable to use user and group databases in `%s'",
201				    optarg);
202			break;
203
204		case 'm':
205			fsoptions.maxsize =
206			    strsuftoll("maximum size", optarg, 1LL, LLONG_MAX);
207			break;
208
209		case 'O':
210			fsoptions.offset =
211			    strsuftoll("offset", optarg, 0LL, LLONG_MAX);
212			break;
213
214		case 'o':
215		{
216			char *p;
217
218			while ((p = strsep(&optarg, ",")) != NULL) {
219				if (*p == '\0')
220					errx(1, "Empty option");
221				if (! fstype->parse_options(p, &fsoptions))
222					usage(fstype, &fsoptions);
223			}
224			break;
225		}
226		case 'p':
227			/* Deprecated in favor of 'Z' */
228			fsoptions.sparse = 1;
229			break;
230
231		case 'R':
232			/* Round image size up to specified block size */
233			fsoptions.roundup =
234			    strsuftoll("roundup-size", optarg, 0, LLONG_MAX);
235			break;
236
237		case 's':
238			fsoptions.minsize = fsoptions.maxsize =
239			    strsuftoll("size", optarg, 1LL, LLONG_MAX);
240			break;
241
242		case 'S':
243			fsoptions.sectorsize =
244			    (int)strsuftoll("sector size", optarg,
245				1LL, INT_MAX);
246			break;
247
248		case 't':
249			/* Check current one and cleanup if necessary. */
250			if (fstype->cleanup_options)
251				fstype->cleanup_options(&fsoptions);
252			fsoptions.fs_specific = NULL;
253			if ((fstype = get_fstype(optarg)) == NULL)
254				errx(1, "Unknown fs type `%s'.", optarg);
255			fstype->prepare_options(&fsoptions);
256			break;
257
258		case 'T':
259			if (get_tstamp(optarg, &stampst) == -1)
260				errx(1, "Cannot get timestamp from `%s'",
261				    optarg);
262			break;
263
264		case 'x':
265			fsoptions.onlyspec = 1;
266			break;
267
268		case 'Z':
269			/* Superscedes 'p' for compatibility with NetBSD makefs(8) */
270			fsoptions.sparse = 1;
271			break;
272
273		default:
274			usage(fstype, &fsoptions);
275			/* NOTREACHED */
276
277		}
278	}
279	if (debug) {
280		printf("debug mask: 0x%08x\n", debug);
281		printf("start time: %ld.%ld, %s",
282		    (long)start_time.tv_sec, (long)start_time.tv_nsec,
283		    ctime(&start_time.tv_sec));
284	}
285	argc -= optind;
286	argv += optind;
287
288	if (argc < 2)
289		usage(fstype, &fsoptions);
290
291	/* -x must be accompanied by -F */
292	if (fsoptions.onlyspec != 0 && specfile == NULL)
293		errx(1, "-x requires -F mtree-specfile.");
294
295	/* Accept '-' as meaning "read from standard input". */
296	if (strcmp(argv[1], "-") == 0)
297		sb.st_mode = S_IFREG;
298	else {
299		if (stat(argv[1], &sb) == -1)
300			err(1, "Can't stat `%s'", argv[1]);
301	}
302
303	switch (sb.st_mode & S_IFMT) {
304	case S_IFDIR:		/* walk the tree */
305		subtree = argv[1];
306		TIMER_START(start);
307		root = walk_dir(subtree, ".", NULL, NULL);
308		TIMER_RESULTS(start, "walk_dir");
309		break;
310	case S_IFREG:		/* read the manifest file */
311		subtree = ".";
312		TIMER_START(start);
313		root = read_mtree(argv[1], NULL);
314		TIMER_RESULTS(start, "manifest");
315		break;
316	default:
317		errx(1, "%s: not a file or directory", argv[1]);
318		/* NOTREACHED */
319	}
320
321	/* append extra directory */
322	for (i = 2; i < argc; i++) {
323		if (stat(argv[i], &sb) == -1)
324			err(1, "Can't stat `%s'", argv[i]);
325		if (!S_ISDIR(sb.st_mode))
326			errx(1, "%s: not a directory", argv[i]);
327		TIMER_START(start);
328		root = walk_dir(argv[i], ".", NULL, root);
329		TIMER_RESULTS(start, "walk_dir2");
330	}
331
332	if (specfile) {		/* apply a specfile */
333		TIMER_START(start);
334		apply_specfile(specfile, subtree, root, fsoptions.onlyspec);
335		TIMER_RESULTS(start, "apply_specfile");
336	}
337
338	if (debug & DEBUG_DUMP_FSNODES) {
339		printf("\nparent: %s\n", subtree);
340		dump_fsnodes(root);
341		putchar('\n');
342	}
343
344				/* build the file system */
345	TIMER_START(start);
346	fstype->make_fs(argv[0], subtree, root, &fsoptions);
347	TIMER_RESULTS(start, "make_fs");
348
349	free_fsnodes(root);
350
351	exit(0);
352	/* NOTREACHED */
353}
354
355int
356set_option(const option_t *options, const char *option, char *buf, size_t len)
357{
358	char *var, *val;
359	int retval;
360
361	assert(option != NULL);
362
363	var = estrdup(option);
364	for (val = var; *val; val++)
365		if (*val == '=') {
366			*val++ = '\0';
367			break;
368		}
369	retval = set_option_var(options, var, val, buf, len);
370	free(var);
371	return retval;
372}
373
374int
375set_option_var(const option_t *options, const char *var, const char *val,
376    char *buf, size_t len)
377{
378	char *s;
379	size_t i;
380
381#define NUM(type) \
382	if (!*val) { \
383		*(type *)options[i].value = 1; \
384		break; \
385	} \
386	*(type *)options[i].value = (type)strsuftoll(options[i].desc, val, \
387	    options[i].minimum, options[i].maximum); break
388
389	for (i = 0; options[i].name != NULL; i++) {
390		if (var[1] == '\0') {
391			if (options[i].letter != var[0])
392				continue;
393		} else if (strcmp(options[i].name, var) != 0)
394			continue;
395		switch (options[i].type) {
396		case OPT_BOOL:
397			*(bool *)options[i].value = 1;
398			break;
399		case OPT_STRARRAY:
400			strlcpy((void *)options[i].value, val, (size_t)
401			    options[i].maximum);
402			break;
403		case OPT_STRPTR:
404			s = estrdup(val);
405			*(char **)options[i].value = s;
406			break;
407		case OPT_STRBUF:
408			if (buf == NULL)
409				abort();
410			strlcpy(buf, val, len);
411			break;
412		case OPT_INT64:
413			NUM(uint64_t);
414		case OPT_INT32:
415			NUM(uint32_t);
416		case OPT_INT16:
417			NUM(uint16_t);
418		case OPT_INT8:
419			NUM(uint8_t);
420		default:
421			warnx("Unknown type %d in option %s", options[i].type,
422			    val);
423			return 0;
424		}
425		return i;
426	}
427	warnx("Unknown option `%s'", var);
428	return -1;
429}
430
431
432static fstype_t *
433get_fstype(const char *type)
434{
435	int i;
436
437	for (i = 0; fstypes[i].type != NULL; i++)
438		if (strcmp(fstypes[i].type, type) == 0)
439			return (&fstypes[i]);
440	return (NULL);
441}
442
443option_t *
444copy_opts(const option_t *o)
445{
446	size_t i;
447
448	for (i = 0; o[i].name; i++)
449		continue;
450	i++;
451	return memcpy(ecalloc(i, sizeof(*o)), o, i * sizeof(*o));
452}
453
454static int
455get_tstamp(const char *b, struct stat *st)
456{
457	time_t when;
458	char *eb;
459	long long l;
460
461	if (stat(b, st) != -1)
462		return 0;
463
464	{
465		errno = 0;
466		l = strtoll(b, &eb, 0);
467		if (b == eb || *eb || errno)
468			return -1;
469		when = (time_t)l;
470	}
471
472	st->st_ino = 1;
473#ifdef HAVE_STRUCT_STAT_BIRTHTIME
474	st->st_birthtime =
475#endif
476	st->st_mtime = st->st_ctime = st->st_atime = when;
477	return 0;
478}
479
480static void
481usage(fstype_t *fstype, fsinfo_t *fsoptions)
482{
483	const char *prog;
484
485	prog = getprogname();
486	fprintf(stderr,
487"Usage: %s [-xZ] [-B endian] [-b free-blocks] [-d debug-mask]\n"
488"\t[-F mtree-specfile] [-f free-files] [-M minimum-size] [-m maximum-size]\n"
489"\t[-N userdb-dir] [-O offset] [-o fs-options] [-R roundup-size]\n"
490"\t[-S sector-size] [-s image-size] [-T <timestamp/file>] [-t fs-type]\n"
491"\timage-file directory | manifest [extra-directory ...]\n",
492	    prog);
493
494	if (fstype) {
495		size_t i;
496		option_t *o = fsoptions->fs_options;
497
498		fprintf(stderr, "\n%s specific options:\n", fstype->type);
499		for (i = 0; o[i].name != NULL; i++)
500			fprintf(stderr, "\t%c%c%20.20s\t%s\n",
501			    o[i].letter ? o[i].letter : ' ',
502			    o[i].letter ? ',' : ' ',
503			    o[i].name, o[i].desc);
504	}
505	exit(1);
506}
507