xref: /freebsd/sbin/fsck/fsck.c (revision 1c05a6ea6b849ff95e539c31adea887c644a6a01)
1 /*	$NetBSD: fsck.c,v 1.30 2003/08/07 10:04:15 agc Exp $	*/
2 
3 /*
4  * Copyright (c) 1996 Christos Zoulas. All rights reserved.
5  * Copyright (c) 1980, 1989, 1993, 1994
6  *	The Regents of the University of California.  All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. Neither the name of the University nor the names of its contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30  * SUCH DAMAGE.
31  *
32  * From: @(#)mount.c	8.19 (Berkeley) 4/19/94
33  * From: $NetBSD: mount.c,v 1.24 1995/11/18 03:34:29 cgd Exp
34  * $NetBSD: fsck.c,v 1.30 2003/08/07 10:04:15 agc Exp $
35  */
36 
37 #include <sys/cdefs.h>
38 __FBSDID("$FreeBSD$");
39 
40 #include <sys/param.h>
41 #include <sys/mount.h>
42 #include <sys/queue.h>
43 #include <sys/wait.h>
44 #include <sys/disk.h>
45 #include <sys/ioctl.h>
46 
47 #include <ctype.h>
48 #include <err.h>
49 #include <errno.h>
50 #include <fstab.h>
51 #include <fcntl.h>
52 #include <paths.h>
53 #include <signal.h>
54 #include <stdio.h>
55 #include <stdlib.h>
56 #include <string.h>
57 #include <unistd.h>
58 
59 #include "fsutil.h"
60 
61 static enum { IN_LIST, NOT_IN_LIST } which = NOT_IN_LIST;
62 
63 static TAILQ_HEAD(fstypelist, entry) opthead, selhead;
64 
65 struct entry {
66 	char *type;
67 	char *options;
68 	TAILQ_ENTRY(entry) entries;
69 };
70 
71 static char *options = NULL;
72 static int flags = 0;
73 static int forceflag = 0;
74 
75 static int checkfs(const char *, const char *, const char *, const char *, pid_t *);
76 static int selected(const char *);
77 static void addoption(char *);
78 static const char *getoptions(const char *);
79 static void addentry(struct fstypelist *, const char *, const char *);
80 static void maketypelist(char *);
81 static void catopt(char **, const char *);
82 static void mangle(char *, int *, const char ** volatile *, int *);
83 static const char *getfstype(const char *);
84 static void usage(void) __dead2;
85 static int isok(struct fstab *);
86 
87 static struct {
88 	const char *ptype;
89 	const char *name;
90 } ptype_map[] = {
91 	{ "ufs",	"ffs" },
92 	{ "ffs",	"ffs" },
93 	{ "fat",	"msdosfs" },
94 	{ "efi",	"msdosfs" },
95 	{ NULL,		NULL },
96 };
97 
98 int
99 main(int argc, char *argv[])
100 {
101 	struct fstab *fs;
102 	int i, rval = 0;
103 	const char *vfstype = NULL;
104 	char globopt[3];
105 	const char *etc_fstab;
106 
107 	globopt[0] = '-';
108 	globopt[2] = '\0';
109 
110 	TAILQ_INIT(&selhead);
111 	TAILQ_INIT(&opthead);
112 
113 	etc_fstab = NULL;
114 	while ((i = getopt(argc, argv, "BCdvpfFnyl:t:T:c:")) != -1)
115 		switch (i) {
116 		case 'B':
117 			if (flags & CHECK_BACKGRD)
118 				errx(1, "Cannot specify -B and -F.");
119 			flags |= DO_BACKGRD;
120 			break;
121 
122 		case 'd':
123 			flags |= CHECK_DEBUG;
124 			break;
125 
126 		case 'v':
127 			flags |= CHECK_VERBOSE;
128 			break;
129 
130 		case 'F':
131 			if (flags & DO_BACKGRD)
132 				errx(1, "Cannot specify -B and -F.");
133 			flags |= CHECK_BACKGRD;
134 			break;
135 
136 		case 'p':
137 			flags |= CHECK_PREEN;
138 			/*FALLTHROUGH*/
139 		case 'C':
140 			flags |= CHECK_CLEAN;
141 			/*FALLTHROUGH*/
142 		case 'n':
143 		case 'y':
144 			globopt[1] = i;
145 			catopt(&options, globopt);
146 			break;
147 
148 		case 'f':
149 			forceflag = 1;
150 			globopt[1] = i;
151 			catopt(&options, globopt);
152 			break;
153 
154 		case 'l':
155 			warnx("Ignoring obsolete -l option\n");
156 			break;
157 
158 		case 'T':
159 			if (*optarg)
160 				addoption(optarg);
161 			break;
162 
163 		case 't':
164 			if (!TAILQ_EMPTY(&selhead))
165 				errx(1, "only one -t option may be specified.");
166 
167 			maketypelist(optarg);
168 			vfstype = optarg;
169 			break;
170 
171 		case 'c':
172 			etc_fstab = optarg;
173 			break;
174 
175 		case '?':
176 		default:
177 			usage();
178 			/* NOTREACHED */
179 		}
180 
181 	argc -= optind;
182 	argv += optind;
183 
184 	if (etc_fstab != NULL)
185 		setfstab(etc_fstab);
186 
187 	if (argc == 0)
188 		return checkfstab(flags, isok, checkfs);
189 
190 #define	BADTYPE(type)							\
191 	(strcmp(type, FSTAB_RO) &&					\
192 	    strcmp(type, FSTAB_RW) && strcmp(type, FSTAB_RQ))
193 
194 
195 	for (; argc--; argv++) {
196 		const char *spec, *mntpt, *type, *cp;
197 		char device[MAXPATHLEN];
198 		struct statfs *mntp;
199 
200 		mntpt = NULL;
201 		spec = *argv;
202 		cp = strrchr(spec, '/');
203 		if (cp == NULL) {
204 			(void)snprintf(device, sizeof(device), "%s%s",
205 				_PATH_DEV, spec);
206 			spec = device;
207 		}
208 		mntp = getmntpt(spec);
209 		if (mntp != NULL) {
210 			spec = mntp->f_mntfromname;
211 			mntpt = mntp->f_mntonname;
212 		}
213 		if ((fs = getfsfile(spec)) == NULL &&
214 		    (fs = getfsspec(spec)) == NULL) {
215 			if (vfstype == NULL)
216 				vfstype = getfstype(spec);
217 			if (vfstype == NULL)
218 				vfstype = "ufs";
219 			type = vfstype;
220 			devcheck(spec);
221 		} else {
222 			spec = fs->fs_spec;
223 			type = fs->fs_vfstype;
224 			mntpt = fs->fs_file;
225 			if (BADTYPE(fs->fs_type))
226 				errx(1, "%s has unknown file system type.",
227 				    spec);
228 		}
229 		if ((flags & CHECK_BACKGRD) &&
230 		    checkfs(type, spec, mntpt, "-F", NULL) == 0) {
231 			printf("%s: DEFER FOR BACKGROUND CHECKING\n", *argv);
232 			continue;
233 		}
234 		if ((flags & DO_BACKGRD) && forceflag == 0 &&
235 		    checkfs(type, spec, mntpt, "-F", NULL) != 0)
236 			continue;
237 
238 		rval |= checkfs(type, spec, mntpt, NULL, NULL);
239 	}
240 
241 	return rval;
242 }
243 
244 
245 static int
246 isok(struct fstab *fs)
247 {
248 	int i;
249 
250 	if (fs->fs_passno == 0)
251 		return (0);
252 	if (BADTYPE(fs->fs_type))
253 		return (0);
254 	if (!selected(fs->fs_vfstype))
255 		return (0);
256 	/*
257 	 * If the -B flag has been given, then process the needed
258 	 * background checks. Background checks cannot be run on
259 	 * file systems that will be mounted read-only or that were
260 	 * not mounted at boot time (typically those marked `noauto').
261 	 * If these basic tests are passed, check with the file system
262 	 * itself to see if it is willing to do background checking
263 	 * by invoking its check program with the -F flag.
264 	 */
265 	if (flags & DO_BACKGRD) {
266 		if (!strcmp(fs->fs_type, FSTAB_RO))
267 			return (0);
268 		if (getmntpt(fs->fs_spec) == NULL)
269 			return (0);
270 		if (checkfs(fs->fs_vfstype, fs->fs_spec, fs->fs_file, "-F", 0))
271 			return (0);
272 		return (1);
273 	}
274 	/*
275 	 * If the -F flag has been given, then consider deferring the
276 	 * check to background. Background checks cannot be run on
277 	 * file systems that will be mounted read-only or that will
278 	 * not be mounted at boot time (e.g., marked `noauto'). If
279 	 * these basic tests are passed, check with the file system
280 	 * itself to see if it is willing to defer to background
281 	 * checking by invoking its check program with the -F flag.
282 	 */
283 	if ((flags & CHECK_BACKGRD) == 0 || !strcmp(fs->fs_type, FSTAB_RO))
284 		return (1);
285 	for (i = strlen(fs->fs_mntops) - 6; i >= 0; i--)
286 		if (!strncmp(&fs->fs_mntops[i], "noauto", 6))
287 			break;
288 	if (i >= 0)
289 		return (1);
290 	if (checkfs(fs->fs_vfstype, fs->fs_spec, fs->fs_file, "-F", NULL) != 0)
291 		return (1);
292 	printf("%s: DEFER FOR BACKGROUND CHECKING\n", fs->fs_spec);
293 	return (0);
294 }
295 
296 
297 static int
298 checkfs(const char *pvfstype, const char *spec, const char *mntpt,
299     const char *auxopt, pid_t *pidp)
300 {
301 	const char ** volatile argv;
302 	pid_t pid;
303 	int argc, i, status, maxargc;
304 	char *optbuf, execbase[MAXPATHLEN];
305 	char *vfstype = NULL;
306 	const char *extra = NULL;
307 
308 #ifdef __GNUC__
309 	/* Avoid vfork clobbering */
310 	(void) &optbuf;
311 	(void) &vfstype;
312 #endif
313 	/*
314 	 * We convert the vfstype to lowercase and any spaces to underscores
315 	 * to not confuse the issue
316 	 *
317 	 * XXX This is a kludge to make automatic filesystem type guessing
318 	 * from the disklabel work for "4.2BSD" filesystems.  It does a
319 	 * very limited subset of transliteration to a normalised form of
320 	 * filesystem name, and we do not seem to enforce a filesystem
321 	 * name character set.
322 	 */
323 	vfstype = strdup(pvfstype);
324 	if (vfstype == NULL)
325 		perr("strdup(pvfstype)");
326 	for (i = 0; i < (int)strlen(vfstype); i++) {
327 		vfstype[i] = tolower(vfstype[i]);
328 		if (vfstype[i] == ' ')
329 			vfstype[i] = '_';
330 	}
331 
332 	extra = getoptions(vfstype);
333 	optbuf = NULL;
334 	if (options)
335 		catopt(&optbuf, options);
336 	if (extra)
337 		catopt(&optbuf, extra);
338 	if (auxopt)
339 		catopt(&optbuf, auxopt);
340 	else if (flags & DO_BACKGRD)
341 		catopt(&optbuf, "-B");
342 
343 	maxargc = 64;
344 	argv = emalloc(sizeof(char *) * maxargc);
345 
346 	(void) snprintf(execbase, sizeof(execbase), "fsck_%s", vfstype);
347 	argc = 0;
348 	argv[argc++] = execbase;
349 	if (optbuf)
350 		mangle(optbuf, &argc, &argv, &maxargc);
351 	argv[argc++] = spec;
352 	argv[argc] = NULL;
353 
354 	if (flags & (CHECK_DEBUG|CHECK_VERBOSE)) {
355 		(void)printf("start %s %swait", mntpt,
356 			pidp ? "no" : "");
357 		for (i = 0; i < argc; i++)
358 			(void)printf(" %s", argv[i]);
359 		(void)printf("\n");
360 	}
361 
362 	switch (pid = vfork()) {
363 	case -1:				/* Error. */
364 		warn("vfork");
365 		if (optbuf)
366 			free(optbuf);
367 		free(vfstype);
368 		return (1);
369 
370 	case 0:					/* Child. */
371 		if ((flags & CHECK_DEBUG) && auxopt == NULL)
372 			_exit(0);
373 
374 		/* Go find an executable. */
375 		execvP(execbase, _PATH_SYSPATH, __DECONST(char * const *, argv));
376 		if (spec)
377 			warn("exec %s for %s in %s", execbase, spec, _PATH_SYSPATH);
378 		else
379 			warn("exec %s in %s", execbase, _PATH_SYSPATH);
380 		_exit(1);
381 		/* NOTREACHED */
382 
383 	default:				/* Parent. */
384 		if (optbuf)
385 			free(optbuf);
386 
387 		free(vfstype);
388 
389 		if (pidp) {
390 			*pidp = pid;
391 			return 0;
392 		}
393 
394 		if (waitpid(pid, &status, 0) < 0) {
395 			warn("waitpid");
396 			return (1);
397 		}
398 
399 		if (WIFEXITED(status)) {
400 			if (WEXITSTATUS(status) != 0)
401 				return (WEXITSTATUS(status));
402 		}
403 		else if (WIFSIGNALED(status)) {
404 			warnx("%s: %s", spec, strsignal(WTERMSIG(status)));
405 			return (1);
406 		}
407 		break;
408 	}
409 
410 	return (0);
411 }
412 
413 
414 static int
415 selected(const char *type)
416 {
417 	struct entry *e;
418 
419 	/* If no type specified, it's always selected. */
420 	TAILQ_FOREACH(e, &selhead, entries)
421 		if (!strncmp(e->type, type, MFSNAMELEN))
422 			return which == IN_LIST ? 1 : 0;
423 
424 	return which == IN_LIST ? 0 : 1;
425 }
426 
427 
428 static const char *
429 getoptions(const char *type)
430 {
431 	struct entry *e;
432 
433 	TAILQ_FOREACH(e, &opthead, entries)
434 		if (!strncmp(e->type, type, MFSNAMELEN))
435 			return e->options;
436 	return "";
437 }
438 
439 
440 static void
441 addoption(char *optstr)
442 {
443 	char *newoptions;
444 	struct entry *e;
445 
446 	if ((newoptions = strchr(optstr, ':')) == NULL)
447 		errx(1, "Invalid option string");
448 
449 	*newoptions++ = '\0';
450 
451 	TAILQ_FOREACH(e, &opthead, entries)
452 		if (!strncmp(e->type, optstr, MFSNAMELEN)) {
453 			catopt(&e->options, newoptions);
454 			return;
455 		}
456 	addentry(&opthead, optstr, newoptions);
457 }
458 
459 
460 static void
461 addentry(struct fstypelist *list, const char *type, const char *opts)
462 {
463 	struct entry *e;
464 
465 	e = emalloc(sizeof(struct entry));
466 	e->type = estrdup(type);
467 	e->options = estrdup(opts);
468 	TAILQ_INSERT_TAIL(list, e, entries);
469 }
470 
471 
472 static void
473 maketypelist(char *fslist)
474 {
475 	char *ptr;
476 
477 	if ((fslist == NULL) || (fslist[0] == '\0'))
478 		errx(1, "empty type list");
479 
480 	if (fslist[0] == 'n' && fslist[1] == 'o') {
481 		fslist += 2;
482 		which = NOT_IN_LIST;
483 	}
484 	else
485 		which = IN_LIST;
486 
487 	while ((ptr = strsep(&fslist, ",")) != NULL)
488 		addentry(&selhead, ptr, "");
489 
490 }
491 
492 
493 static void
494 catopt(char **sp, const char *o)
495 {
496 	char *s;
497 	size_t i, j;
498 
499 	s = *sp;
500 	if (s) {
501 		i = strlen(s);
502 		j = i + 1 + strlen(o) + 1;
503 		s = erealloc(s, j);
504 		(void)snprintf(s + i, j, ",%s", o);
505 	} else
506 		s = estrdup(o);
507 	*sp = s;
508 }
509 
510 
511 static void
512 mangle(char *opts, int *argcp, const char ** volatile *argvp, int *maxargcp)
513 {
514 	char *p, *s;
515 	int argc, maxargc;
516 	const char **argv;
517 
518 	argc = *argcp;
519 	argv = *argvp;
520 	maxargc = *maxargcp;
521 
522 	for (s = opts; (p = strsep(&s, ",")) != NULL;) {
523 		/* Always leave space for one more argument and the NULL. */
524 		if (argc >= maxargc - 3) {
525 			maxargc <<= 1;
526 			argv = erealloc(argv, maxargc * sizeof(char *));
527 		}
528 		if (*p != '\0')  {
529 			if (*p == '-') {
530 				argv[argc++] = p;
531 				p = strchr(p, '=');
532 				if (p) {
533 					*p = '\0';
534 					argv[argc++] = p+1;
535 				}
536 			} else {
537 				argv[argc++] = "-o";
538 				argv[argc++] = p;
539 			}
540 		}
541 	}
542 
543 	*argcp = argc;
544 	*argvp = argv;
545 	*maxargcp = maxargc;
546 }
547 
548 static const char *
549 getfstype(const char *str)
550 {
551 	struct diocgattr_arg attr;
552 	int fd, i;
553 
554 	if ((fd = open(str, O_RDONLY)) == -1)
555 		err(1, "cannot open `%s'", str);
556 
557 	strncpy(attr.name, "PART::type", sizeof(attr.name));
558 	memset(&attr.value, 0, sizeof(attr.value));
559 	attr.len = sizeof(attr.value);
560 	if (ioctl(fd, DIOCGATTR, &attr) == -1) {
561 		(void) close(fd);
562 		return(NULL);
563 	}
564 	(void) close(fd);
565 	for (i = 0; ptype_map[i].ptype != NULL; i++)
566 		if (strstr(attr.value.str, ptype_map[i].ptype) != NULL)
567 			return (ptype_map[i].name);
568 	return (NULL);
569 }
570 
571 
572 static void
573 usage(void)
574 {
575 	static const char common[] =
576 	    "[-Cdfnpvy] [-B | -F] [-T fstype:fsoptions] [-t fstype] [-c fstab]";
577 
578 	(void)fprintf(stderr, "usage: %s %s [special | node] ...\n",
579 	    getprogname(), common);
580 	exit(1);
581 }
582