1 /*
2  * pfsck --- A generic, parallelizing front-end for the fsck program.
3  * It will automatically try to run fsck programs in parallel if the
4  * devices are on separate spindles.  It is based on the same ideas as
5  * the generic front end for fsck by David Engel and Fred van Kempen,
6  * but it has been completely rewritten from scratch to support
7  * parallel execution.
8  *
9  * Written by Theodore Ts'o, <tytso@mit.edu>
10  *
11  * Miquel van Smoorenburg (miquels@drinkel.ow.org) 20-Oct-1994:
12  *   o Changed -t fstype to behave like with mount when -A (all file
13  *     systems) or -M (like mount) is specified.
14  *   o fsck looks if it can find the fsck.type program to decide
15  *     if it should ignore the fs type. This way more fsck programs
16  *     can be added without changing this front-end.
17  *   o -R flag skip root file system.
18  *
19  * Copyright (C) 1993, 1994, 1995, 1996, 1997, 1998, 1999, 2000,
20  * 	2001, 2002, 2003, 2004, 2005 by  Theodore Ts'o.
21  *
22  * %Begin-Header%
23  * This file may be redistributed under the terms of the GNU Public
24  * License.
25  * %End-Header%
26  */
27 
28 #define _XOPEN_SOURCE 600 /* for inclusion of sa_handler in Solaris */
29 
30 #include "config.h"
31 #include <sys/types.h>
32 #include <sys/wait.h>
33 #include <sys/stat.h>
34 #include <limits.h>
35 #include <stdio.h>
36 #include <ctype.h>
37 #include <string.h>
38 #include <time.h>
39 #if HAVE_STDLIB_H
40 #include <stdlib.h>
41 #endif
42 #if HAVE_ERRNO_H
43 #include <errno.h>
44 #endif
45 #if HAVE_PATHS_H
46 #include <paths.h>
47 #endif
48 #if HAVE_UNISTD_H
49 #include <unistd.h>
50 #endif
51 #if HAVE_ERRNO_H
52 #include <errno.h>
53 #endif
54 #if HAVE_MALLOC_H
55 #include <malloc.h>
56 #endif
57 #ifdef HAVE_SIGNAL_H
58 #include <signal.h>
59 #endif
60 
61 #include "../version.h"
62 #include "support/nls-enable.h"
63 #include "fsck.h"
64 #include "blkid/blkid.h"
65 
66 #ifndef _PATH_MNTTAB
67 #define	_PATH_MNTTAB	"/etc/fstab"
68 #endif
69 
70 static const char *ignored_types[] = {
71 	"ignore",
72 	"iso9660",
73 	"nfs",
74 	"proc",
75 	"sw",
76 	"swap",
77 	"tmpfs",
78 	"devpts",
79 	NULL
80 };
81 
82 static const char *really_wanted[] = {
83 	"minix",
84 	"ext2",
85 	"ext3",
86 	"ext4",
87 	"ext4dev",
88 	"jfs",
89 	"reiserfs",
90 	"xiafs",
91 	"xfs",
92 	NULL
93 };
94 
95 #define BASE_MD "/dev/md"
96 
97 /*
98  * Global variables for options
99  */
100 static char *devices[MAX_DEVICES];
101 static char *args[MAX_ARGS];
102 static int num_devices, num_args;
103 
104 static int verbose = 0;
105 static int doall = 0;
106 static int noexecute = 0;
107 static int serialize = 0;
108 static int skip_root = 0;
109 static int ignore_mounted = 0;
110 static int notitle = 0;
111 static int parallel_root = 0;
112 static int progress = 0;
113 static int progress_fd = 0;
114 static int force_all_parallel = 0;
115 static int num_running = 0;
116 static int max_running = 0;
117 static volatile int cancel_requested = 0;
118 static int kill_sent = 0;
119 static char *progname;
120 static char *fstype = NULL;
121 static struct fs_info *filesys_info = NULL, *filesys_last = NULL;
122 static struct fsck_instance *instance_list;
123 static const char *fsck_prefix_path = "/sbin:/sbin/fs.d:/sbin/fs:/etc/fs:/etc";
124 static char *fsck_path = 0;
125 static blkid_cache cache = NULL;
126 
string_copy(const char * s)127 static char *string_copy(const char *s)
128 {
129 	char	*ret;
130 
131 	if (!s)
132 		return 0;
133 	ret = malloc(strlen(s)+1);
134 	if (ret)
135 		strcpy(ret, s);
136 	return ret;
137 }
138 
string_to_int(const char * s)139 static int string_to_int(const char *s)
140 {
141 	long l;
142 	char *p;
143 
144 	l = strtol(s, &p, 0);
145 	if (*p || l == LONG_MIN || l == LONG_MAX || l < 0 || l > INT_MAX)
146 		return -1;
147 	else
148 		return (int) l;
149 }
150 
151 static int ignore(struct fs_info *);
152 
skip_over_blank(char * cp)153 static char *skip_over_blank(char *cp)
154 {
155 	while (*cp && isspace(*cp))
156 		cp++;
157 	return cp;
158 }
159 
skip_over_word(char * cp)160 static char *skip_over_word(char *cp)
161 {
162 	while (*cp && !isspace(*cp))
163 		cp++;
164 	return cp;
165 }
166 
strip_line(char * line)167 static void strip_line(char *line)
168 {
169 	char	*p;
170 
171 	while (*line) {
172 		p = line + strlen(line) - 1;
173 		if ((*p == '\n') || (*p == '\r'))
174 			*p = 0;
175 		else
176 			break;
177 	}
178 }
179 
parse_word(char ** buf)180 static char *parse_word(char **buf)
181 {
182 	char *word, *next;
183 
184 	word = *buf;
185 	if (*word == 0)
186 		return 0;
187 
188 	word = skip_over_blank(word);
189 	next = skip_over_word(word);
190 	if (*next)
191 		*next++ = 0;
192 	*buf = next;
193 	return word;
194 }
195 
parse_escape(char * word)196 static void parse_escape(char *word)
197 {
198 	char	*p, *q;
199 	int	ac, i;
200 
201 	if (!word)
202 		return;
203 
204 	for (p = word, q = word; *p; p++, q++) {
205 		*q = *p;
206 		if (*p != '\\')
207 			continue;
208 		if (*++p == 0)
209 			break;
210 		if (*p == 't') {
211 			*q = '\t';
212 			continue;
213 		}
214 		if (*p == 'n') {
215 			*q = '\n';
216 			continue;
217 		}
218 		if (!isdigit(*p)) {
219 			*q = *p;
220 			continue;
221 		}
222 		ac = 0;
223 		for (i = 0; i < 3; i++, p++) {
224 			if (!isdigit(*p))
225 				break;
226 			ac = (ac * 8) + (*p - '0');
227 		}
228 		*q = ac;
229 		p--;
230 	}
231 	*q = 0;
232 }
233 
free_instance(struct fsck_instance * i)234 static void free_instance(struct fsck_instance *i)
235 {
236 	free(i->prog);
237 	free(i->device);
238 	free(i->base_device);
239 	free(i);
240 	return;
241 }
242 
create_fs_device(const char * device,const char * mntpnt,const char * type,const char * opts,int freq,int passno)243 static struct fs_info *create_fs_device(const char *device, const char *mntpnt,
244 					const char *type, const char *opts,
245 					int freq, int passno)
246 {
247 	struct fs_info *fs;
248 
249 	if (!(fs = malloc(sizeof(struct fs_info))))
250 		return NULL;
251 
252 	fs->device = string_copy(device);
253 	fs->mountpt = string_copy(mntpnt);
254 	fs->type = string_copy(type);
255 	fs->opts = string_copy(opts ? opts : "");
256 	fs->freq = freq;
257 	fs->passno = passno;
258 	fs->flags = 0;
259 	fs->next = NULL;
260 
261 	if (!filesys_info)
262 		filesys_info = fs;
263 	else
264 		filesys_last->next = fs;
265 	filesys_last = fs;
266 
267 	return fs;
268 }
269 
270 
271 
parse_fstab_line(char * line,struct fs_info ** ret_fs)272 static int parse_fstab_line(char *line, struct fs_info **ret_fs)
273 {
274 	char	*dev, *device, *mntpnt, *type, *opts, *freq, *passno, *cp;
275 	struct fs_info *fs;
276 
277 	*ret_fs = 0;
278 	strip_line(line);
279 	cp = line;
280 
281 	device = parse_word(&cp);
282 	if (!device || *device == '#')
283 		return 0;	/* Ignore blank lines and comments */
284 	mntpnt = parse_word(&cp);
285 	type = parse_word(&cp);
286 	opts = parse_word(&cp);
287 	freq = parse_word(&cp);
288 	passno = parse_word(&cp);
289 
290 	if (!mntpnt || !type)
291 		return -1;
292 
293 	parse_escape(device);
294 	parse_escape(mntpnt);
295 	parse_escape(type);
296 	parse_escape(opts);
297 	parse_escape(freq);
298 	parse_escape(passno);
299 
300 	dev = blkid_get_devname(cache, device, NULL);
301 	if (dev)
302 		device = dev;
303 
304 	if (strchr(type, ','))
305 		type = 0;
306 
307 	fs = create_fs_device(device, mntpnt, type ? type : "auto", opts,
308 			      freq ? atoi(freq) : -1,
309 			      passno ? atoi(passno) : -1);
310 	free(dev);
311 
312 	if (!fs)
313 		return -1;
314 	*ret_fs = fs;
315 	return 0;
316 }
317 
interpret_type(struct fs_info * fs)318 static void interpret_type(struct fs_info *fs)
319 {
320 	char	*t;
321 
322 	if (strcmp(fs->type, "auto") != 0)
323 		return;
324 	t = blkid_get_tag_value(cache, "TYPE", fs->device);
325 	if (t) {
326 		free(fs->type);
327 		fs->type = t;
328 	}
329 }
330 
331 /*
332  * Load the filesystem database from /etc/fstab
333  */
load_fs_info(const char * filename)334 static void load_fs_info(const char *filename)
335 {
336 	FILE	*f;
337 	char	buf[1024];
338 	int	lineno = 0;
339 	int	old_fstab = 1;
340 	struct fs_info *fs;
341 
342 	if ((f = fopen(filename, "r")) == NULL) {
343 		fprintf(stderr, _("WARNING: couldn't open %s: %s\n"),
344 			filename, strerror(errno));
345 		return;
346 	}
347 	while (!feof(f)) {
348 		lineno++;
349 		if (!fgets(buf, sizeof(buf), f))
350 			break;
351 		buf[sizeof(buf)-1] = 0;
352 		if (parse_fstab_line(buf, &fs) < 0) {
353 			fprintf(stderr, _("WARNING: bad format "
354 				"on line %d of %s\n"), lineno, filename);
355 			continue;
356 		}
357 		if (!fs)
358 			continue;
359 		if (fs->passno < 0)
360 			fs->passno = 0;
361 		else
362 			old_fstab = 0;
363 	}
364 
365 	fclose(f);
366 
367 	if (old_fstab && filesys_info) {
368 		fputs("\007\007\007", stderr);
369 		fputs(_(
370 		"WARNING: Your /etc/fstab does not contain the fsck passno\n"
371 		"	field.  I will kludge around things for you, but you\n"
372 		"	should fix your /etc/fstab file as soon as you can.\n\n"), stderr);
373 
374 		for (fs = filesys_info; fs; fs = fs->next) {
375 			fs->passno = 1;
376 		}
377 	}
378 }
379 
380 /* Lookup filesys in /etc/fstab and return the corresponding entry. */
lookup(char * filesys)381 static struct fs_info *lookup(char *filesys)
382 {
383 	struct fs_info *fs;
384 
385 	/* No filesys name given. */
386 	if (filesys == NULL)
387 		return NULL;
388 
389 	for (fs = filesys_info; fs; fs = fs->next) {
390 		if (!strcmp(filesys, fs->device) ||
391 		    (fs->mountpt && !strcmp(filesys, fs->mountpt)))
392 			break;
393 	}
394 
395 	return fs;
396 }
397 
398 /* Find fsck program for a given fs type. */
find_fsck(char * type)399 static char *find_fsck(char *type)
400 {
401   char *s;
402   const char *tpl;
403   static char prog[256];
404   char *p = string_copy(fsck_path);
405   struct stat st;
406 
407   /* Are we looking for a program or just a type? */
408   tpl = (strncmp(type, "fsck.", 5) ? "%s/fsck.%s" : "%s/%s");
409 
410   for(s = strtok(p, ":"); s; s = strtok(NULL, ":")) {
411 	sprintf(prog, tpl, s, type);
412 	if (stat(prog, &st) == 0) break;
413   }
414   free(p);
415   return(s ? prog : NULL);
416 }
417 
progress_active(NOARGS)418 static int progress_active(NOARGS)
419 {
420 	struct fsck_instance *inst;
421 
422 	for (inst = instance_list; inst; inst = inst->next) {
423 		if (inst->flags & FLAG_DONE)
424 			continue;
425 		if (inst->flags & FLAG_PROGRESS)
426 			return 1;
427 	}
428 	return 0;
429 }
430 
431 /*
432  * Execute a particular fsck program, and link it into the list of
433  * child processes we are waiting for.
434  */
execute(const char * type,const char * device,const char * mntpt,int interactive)435 static int execute(const char *type, const char *device, const char *mntpt,
436 		   int interactive)
437 {
438 	char *s, *argv[80], prog[80];
439 	int  argc, i;
440 	struct fsck_instance *inst, *p;
441 	pid_t	pid;
442 
443 	inst = malloc(sizeof(struct fsck_instance));
444 	if (!inst)
445 		return ENOMEM;
446 	memset(inst, 0, sizeof(struct fsck_instance));
447 
448 	sprintf(prog, "fsck.%s", type);
449 	argv[0] = string_copy(prog);
450 	argc = 1;
451 
452 	for (i=0; i <num_args; i++)
453 		argv[argc++] = string_copy(args[i]);
454 
455 	if (progress) {
456 		if ((strcmp(type, "ext2") == 0) ||
457 		    (strcmp(type, "ext3") == 0) ||
458 		    (strcmp(type, "ext4") == 0) ||
459 		    (strcmp(type, "ext4dev") == 0)) {
460 			char tmp[80];
461 
462 			tmp[0] = 0;
463 			if (!progress_active()) {
464 				snprintf(tmp, 80, "-C%d", progress_fd);
465 				inst->flags |= FLAG_PROGRESS;
466 			} else if (progress_fd)
467 				snprintf(tmp, 80, "-C%d", progress_fd * -1);
468 			if (tmp[0])
469 				argv[argc++] = string_copy(tmp);
470 		}
471 	}
472 
473 	argv[argc++] = string_copy(device);
474 	argv[argc] = 0;
475 
476 	s = find_fsck(prog);
477 	if (s == NULL) {
478 		fprintf(stderr, _("fsck: %s: not found\n"), prog);
479 		free(inst);
480 		return ENOENT;
481 	}
482 
483 	if (verbose || noexecute) {
484 		printf("[%s (%d) -- %s] ", s, num_running,
485 		       mntpt ? mntpt : device);
486 		for (i=0; i < argc; i++)
487 			printf("%s ", argv[i]);
488 		printf("\n");
489 	}
490 
491 	/* Fork and execute the correct program. */
492 	if (noexecute)
493 		pid = -1;
494 	else if ((pid = fork()) < 0) {
495 		perror("fork");
496 		free(inst);
497 		return errno;
498 	} else if (pid == 0) {
499 		if (!interactive)
500 			close(0);
501 		(void) execv(s, argv);
502 		perror(argv[0]);
503 		free(inst);
504 		exit(EXIT_ERROR);
505 	}
506 
507 	for (i=0; i < argc; i++)
508 		free(argv[i]);
509 
510 	inst->pid = pid;
511 	inst->prog = string_copy(prog);
512 	inst->type = string_copy(type);
513 	inst->device = string_copy(device);
514 	inst->base_device = base_device(device);
515 	inst->start_time = time(0);
516 	inst->next = NULL;
517 
518 	/*
519 	 * Find the end of the list, so we add the instance on at the end.
520 	 */
521 	for (p = instance_list; p && p->next; p = p->next);
522 
523 	if (p)
524 		p->next = inst;
525 	else
526 		instance_list = inst;
527 
528 	return 0;
529 }
530 
531 /*
532  * Send a signal to all outstanding fsck child processes
533  */
kill_all(int signum)534 static int kill_all(int signum)
535 {
536 	struct fsck_instance *inst;
537 	int	n = 0;
538 
539 	for (inst = instance_list; inst; inst = inst->next) {
540 		if (inst->flags & FLAG_DONE)
541 			continue;
542 		kill(inst->pid, signum);
543 		n++;
544 	}
545 	return n;
546 }
547 
548 /*
549  * Wait for one child process to exit; when it does, unlink it from
550  * the list of executing child processes, and return it.
551  */
wait_one(int flags)552 static struct fsck_instance *wait_one(int flags)
553 {
554 	int	status;
555 	int	sig;
556 	struct fsck_instance *inst, *inst2, *prev;
557 	pid_t	pid;
558 
559 	if (!instance_list)
560 		return NULL;
561 
562 	if (noexecute) {
563 		inst = instance_list;
564 		prev = 0;
565 #ifdef RANDOM_DEBUG
566 		while (inst->next && (random() & 1)) {
567 			prev = inst;
568 			inst = inst->next;
569 		}
570 #endif
571 		inst->exit_status = 0;
572 		goto ret_inst;
573 	}
574 
575 	/*
576 	 * gcc -Wall fails saving throw against stupidity
577 	 * (inst and prev are thought to be uninitialized variables)
578 	 */
579 	inst = prev = NULL;
580 
581 	do {
582 		pid = waitpid(-1, &status, flags);
583 		if (cancel_requested && !kill_sent) {
584 			kill_all(SIGTERM);
585 			kill_sent++;
586 		}
587 		if ((pid == 0) && (flags & WNOHANG))
588 			return NULL;
589 		if (pid < 0) {
590 			if ((errno == EINTR) || (errno == EAGAIN))
591 				continue;
592 			if (errno == ECHILD) {
593 				fprintf(stderr,
594 					_("%s: wait: No more child process?!?\n"),
595 					progname);
596 				return NULL;
597 			}
598 			perror("wait");
599 			continue;
600 		}
601 		for (prev = 0, inst = instance_list;
602 		     inst;
603 		     prev = inst, inst = inst->next) {
604 			if (inst->pid == pid)
605 				break;
606 		}
607 	} while (!inst);
608 
609 	if (WIFEXITED(status))
610 		status = WEXITSTATUS(status);
611 	else if (WIFSIGNALED(status)) {
612 		sig = WTERMSIG(status);
613 		if (sig == SIGINT) {
614 			status = EXIT_UNCORRECTED;
615 		} else {
616 			printf(_("Warning... %s for device %s exited "
617 			       "with signal %d.\n"),
618 			       inst->prog, inst->device, sig);
619 			status = EXIT_ERROR;
620 		}
621 	} else {
622 		printf(_("%s %s: status is %x, should never happen.\n"),
623 		       inst->prog, inst->device, status);
624 		status = EXIT_ERROR;
625 	}
626 	inst->exit_status = status;
627 	inst->flags |= FLAG_DONE;
628 	if (progress && (inst->flags & FLAG_PROGRESS) &&
629 	    !progress_active()) {
630 		for (inst2 = instance_list; inst2; inst2 = inst2->next) {
631 			if (inst2->flags & FLAG_DONE)
632 				continue;
633 			if (strcmp(inst2->type, "ext2") &&
634 			    strcmp(inst2->type, "ext3") &&
635 			    strcmp(inst2->type, "ext4") &&
636 			    strcmp(inst2->type, "ext4dev"))
637 				continue;
638 			/*
639 			 * If we've just started the fsck, wait a tiny
640 			 * bit before sending the kill, to give it
641 			 * time to set up the signal handler
642 			 */
643 			if (inst2->start_time < time(0)+2) {
644 				if (fork() == 0) {
645 					sleep(1);
646 					kill(inst2->pid, SIGUSR1);
647 					exit(0);
648 				}
649 			} else
650 				kill(inst2->pid, SIGUSR1);
651 			inst2->flags |= FLAG_PROGRESS;
652 			break;
653 		}
654 	}
655 ret_inst:
656 	if (prev)
657 		prev->next = inst->next;
658 	else
659 		instance_list = inst->next;
660 	if (verbose > 1)
661 		printf(_("Finished with %s (exit status %d)\n"),
662 		       inst->device, inst->exit_status);
663 	num_running--;
664 	return inst;
665 }
666 
667 #define FLAG_WAIT_ALL		0
668 #define FLAG_WAIT_ATLEAST_ONE	1
669 /*
670  * Wait until all executing child processes have exited; return the
671  * logical OR of all of their exit code values.
672  */
wait_many(int flags)673 static int wait_many(int flags)
674 {
675 	struct fsck_instance *inst;
676 	int	global_status = 0;
677 	int	wait_flags = 0;
678 
679 	while ((inst = wait_one(wait_flags))) {
680 		global_status |= inst->exit_status;
681 		free_instance(inst);
682 #ifdef RANDOM_DEBUG
683 		if (noexecute && (flags & WNOHANG) && !(random() % 3))
684 			break;
685 #endif
686 		if (flags & FLAG_WAIT_ATLEAST_ONE)
687 			wait_flags = WNOHANG;
688 	}
689 	return global_status;
690 }
691 
692 /*
693  * Run the fsck program on a particular device
694  *
695  * If the type is specified using -t, and it isn't prefixed with "no"
696  * (as in "noext2") and only one filesystem type is specified, then
697  * use that type regardless of what is specified in /etc/fstab.
698  *
699  * If the type isn't specified by the user, then use either the type
700  * specified in /etc/fstab, or DEFAULT_FSTYPE.
701  */
fsck_device(struct fs_info * fs,int interactive)702 static void fsck_device(struct fs_info *fs, int interactive)
703 {
704 	const char *type;
705 	int retval;
706 
707 	interpret_type(fs);
708 
709 	if (strcmp(fs->type, "auto") != 0)
710 		type = fs->type;
711 	else if (fstype && strncmp(fstype, "no", 2) &&
712 	    strncmp(fstype, "opts=", 5) && strncmp(fstype, "loop", 4) &&
713 	    !strchr(fstype, ','))
714 		type = fstype;
715 	else
716 		type = DEFAULT_FSTYPE;
717 
718 	num_running++;
719 	retval = execute(type, fs->device, fs->mountpt, interactive);
720 	if (retval) {
721 		fprintf(stderr, _("%s: Error %d while executing fsck.%s "
722 			"for %s\n"), progname, retval, type, fs->device);
723 		num_running--;
724 	}
725 }
726 
727 
728 /*
729  * Deal with the fsck -t argument.
730  */
731 static struct fs_type_compile {
732 	char **list;
733 	int *type;
734 	int  negate;
735 } fs_type_compiled;
736 
737 #define FS_TYPE_NORMAL	0
738 #define FS_TYPE_OPT	1
739 #define FS_TYPE_NEGOPT	2
740 
741 static const char *fs_type_syntax_error =
742 N_("Either all or none of the filesystem types passed to -t must be prefixed\n"
743    "with 'no' or '!'.\n");
744 
compile_fs_type(char * fs_type,struct fs_type_compile * cmp)745 static void compile_fs_type(char *fs_type, struct fs_type_compile *cmp)
746 {
747 	char 	*cp, *list, *s;
748 	int	num = 2;
749 	int	negate, first_negate = 1;
750 
751 	if (fs_type) {
752 		for (cp=fs_type; *cp; cp++) {
753 			if (*cp == ',')
754 				num++;
755 		}
756 	}
757 
758 	cmp->list = malloc(num * sizeof(char *));
759 	cmp->type = malloc(num * sizeof(int));
760 	if (!cmp->list || !cmp->type) {
761 		fputs(_("Couldn't allocate memory for filesystem types\n"),
762 		      stderr);
763 		exit(EXIT_ERROR);
764 	}
765 	memset(cmp->list, 0, num * sizeof(char *));
766 	memset(cmp->type, 0, num * sizeof(int));
767 	cmp->negate = 0;
768 
769 	if (!fs_type)
770 		return;
771 
772 	list = string_copy(fs_type);
773 	num = 0;
774 	s = strtok(list, ",");
775 	while(s) {
776 		negate = 0;
777 		if (strncmp(s, "no", 2) == 0) {
778 			s += 2;
779 			negate = 1;
780 		} else if (*s == '!') {
781 			s++;
782 			negate = 1;
783 		}
784 		if (strcmp(s, "loop") == 0)
785 			/* loop is really short-hand for opts=loop */
786 			goto loop_special_case;
787 		else if (strncmp(s, "opts=", 5) == 0) {
788 			s += 5;
789 		loop_special_case:
790 			cmp->type[num] = negate ? FS_TYPE_NEGOPT : FS_TYPE_OPT;
791 		} else {
792 			if (first_negate) {
793 				cmp->negate = negate;
794 				first_negate = 0;
795 			}
796 			if ((negate && !cmp->negate) ||
797 			    (!negate && cmp->negate)) {
798 				fputs(_(fs_type_syntax_error), stderr);
799 				exit(EXIT_USAGE);
800 			}
801 		}
802 #if 0
803 		printf("Adding %s to list (type %d).\n", s, cmp->type[num]);
804 #endif
805 	        cmp->list[num++] = string_copy(s);
806 		s = strtok(NULL, ",");
807 	}
808 	free(list);
809 }
810 
811 /*
812  * This function returns true if a particular option appears in a
813  * comma-delimited options list
814  */
opt_in_list(const char * opt,char * optlist)815 static int opt_in_list(const char *opt, char *optlist)
816 {
817 	char	*list, *s;
818 
819 	if (!optlist)
820 		return 0;
821 	list = string_copy(optlist);
822 
823 	s = strtok(list, ",");
824 	while(s) {
825 		if (strcmp(s, opt) == 0) {
826 			free(list);
827 			return 1;
828 		}
829 		s = strtok(NULL, ",");
830 	}
831         free(list);
832 	return 0;
833 }
834 
835 /* See if the filesystem matches the criteria given by the -t option */
fs_match(struct fs_info * fs,struct fs_type_compile * cmp)836 static int fs_match(struct fs_info *fs, struct fs_type_compile *cmp)
837 {
838 	int n, ret = 0, checked_type = 0;
839 	char *cp;
840 
841 	if (cmp->list == 0 || cmp->list[0] == 0)
842 		return 1;
843 
844 	for (n=0; (cp = cmp->list[n]); n++) {
845 		switch (cmp->type[n]) {
846 		case FS_TYPE_NORMAL:
847 			checked_type++;
848 			if (strcmp(cp, fs->type) == 0) {
849 				ret = 1;
850 			}
851 			break;
852 		case FS_TYPE_NEGOPT:
853 			if (opt_in_list(cp, fs->opts))
854 				return 0;
855 			break;
856 		case FS_TYPE_OPT:
857 			if (!opt_in_list(cp, fs->opts))
858 				return 0;
859 			break;
860 		}
861 	}
862 	if (checked_type == 0)
863 		return 1;
864 	return (cmp->negate ? !ret : ret);
865 }
866 
867 /* Check if we should ignore this filesystem. */
ignore(struct fs_info * fs)868 static int ignore(struct fs_info *fs)
869 {
870 	const char **ip;
871 	int wanted = 0;
872 
873 	/*
874 	 * If the pass number is 0, ignore it.
875 	 */
876 	if (fs->passno == 0)
877 		return 1;
878 
879 	/*
880 	 * If this is a bind mount, ignore it.
881 	 */
882 	if (opt_in_list("bind", fs->opts)) {
883 		fprintf(stderr,
884 			_("%s: skipping bad line in /etc/fstab: bind mount with nonzero fsck pass number\n"),
885 			fs->mountpt);
886 		return 1;
887 	}
888 
889 	interpret_type(fs);
890 
891 	/*
892 	 * If a specific fstype is specified, and it doesn't match,
893 	 * ignore it.
894 	 */
895 	if (!fs_match(fs, &fs_type_compiled)) return 1;
896 
897 	/* Are we ignoring this type? */
898 	for(ip = ignored_types; *ip; ip++)
899 		if (strcmp(fs->type, *ip) == 0) return 1;
900 
901 	/* Do we really really want to check this fs? */
902 	for(ip = really_wanted; *ip; ip++)
903 		if (strcmp(fs->type, *ip) == 0) {
904 			wanted = 1;
905 			break;
906 		}
907 
908 	/* See if the <fsck.fs> program is available. */
909 	if (find_fsck(fs->type) == NULL) {
910 		if (wanted)
911 			fprintf(stderr, _("fsck: cannot check %s: fsck.%s not found\n"),
912 				fs->device, fs->type);
913 		return 1;
914 	}
915 
916 	/* We can and want to check this file system type. */
917 	return 0;
918 }
919 
920 /*
921  * Returns TRUE if a partition on the same disk is already being
922  * checked.
923  */
device_already_active(char * device)924 static int device_already_active(char *device)
925 {
926 	struct fsck_instance *inst;
927 	char *base;
928 
929 	if (force_all_parallel)
930 		return 0;
931 
932 #ifdef BASE_MD
933 	/* Don't check a soft raid disk with any other disk */
934 	if (instance_list &&
935 	    (!strncmp(instance_list->device, BASE_MD, sizeof(BASE_MD)-1) ||
936 	     !strncmp(device, BASE_MD, sizeof(BASE_MD)-1)))
937 		return 1;
938 #endif
939 
940 	base = base_device(device);
941 	/*
942 	 * If we don't know the base device, assume that the device is
943 	 * already active if there are any fsck instances running.
944 	 */
945 	if (!base)
946 		return (instance_list != 0);
947 	for (inst = instance_list; inst; inst = inst->next) {
948 		if (!inst->base_device || !strcmp(base, inst->base_device)) {
949 			free(base);
950 			return 1;
951 		}
952 	}
953 	free(base);
954 	return 0;
955 }
956 
957 /* Check all file systems, using the /etc/fstab table. */
check_all(NOARGS)958 static int check_all(NOARGS)
959 {
960 	struct fs_info *fs = NULL;
961 	int status = EXIT_OK;
962 	int not_done_yet = 1;
963 	int passno = 1;
964 	int pass_done;
965 
966 	if (verbose)
967 		fputs(_("Checking all file systems.\n"), stdout);
968 
969 	/*
970 	 * Do an initial scan over the filesystem; mark filesystems
971 	 * which should be ignored as done, and resolve any "auto"
972 	 * filesystem types (done as a side-effect of calling ignore()).
973 	 */
974 	for (fs = filesys_info; fs; fs = fs->next) {
975 		if (ignore(fs))
976 			fs->flags |= FLAG_DONE;
977 	}
978 
979 	/*
980 	 * Find and check the root filesystem.
981 	 */
982 	if (!parallel_root) {
983 		for (fs = filesys_info; fs; fs = fs->next) {
984 			if (!strcmp(fs->mountpt, "/"))
985 				break;
986 		}
987 		if (fs) {
988 			if (!skip_root && !ignore(fs) &&
989 			    !(ignore_mounted && is_mounted(fs->device))) {
990 				fsck_device(fs, 1);
991 				status |= wait_many(FLAG_WAIT_ALL);
992 				if (status > EXIT_NONDESTRUCT)
993 					return status;
994 			}
995 			fs->flags |= FLAG_DONE;
996 		}
997 	}
998 	/*
999 	 * This is for the bone-headed user who enters the root
1000 	 * filesystem twice.  Skip root will skep all root entries.
1001 	 */
1002 	if (skip_root)
1003 		for (fs = filesys_info; fs; fs = fs->next)
1004 			if (!strcmp(fs->mountpt, "/"))
1005 				fs->flags |= FLAG_DONE;
1006 
1007 	while (not_done_yet) {
1008 		not_done_yet = 0;
1009 		pass_done = 1;
1010 
1011 		for (fs = filesys_info; fs; fs = fs->next) {
1012 			if (cancel_requested)
1013 				break;
1014 			if (fs->flags & FLAG_DONE)
1015 				continue;
1016 			/*
1017 			 * If the filesystem's pass number is higher
1018 			 * than the current pass number, then we don't
1019 			 * do it yet.
1020 			 */
1021 			if (fs->passno > passno) {
1022 				not_done_yet++;
1023 				continue;
1024 			}
1025 			if (ignore_mounted && is_mounted(fs->device)) {
1026 				fs->flags |= FLAG_DONE;
1027 				continue;
1028 			}
1029 			/*
1030 			 * If a filesystem on a particular device has
1031 			 * already been spawned, then we need to defer
1032 			 * this to another pass.
1033 			 */
1034 			if (device_already_active(fs->device)) {
1035 				pass_done = 0;
1036 				continue;
1037 			}
1038 			/*
1039 			 * Spawn off the fsck process
1040 			 */
1041 			fsck_device(fs, serialize);
1042 			fs->flags |= FLAG_DONE;
1043 
1044 			/*
1045 			 * Only do one filesystem at a time, or if we
1046 			 * have a limit on the number of fsck's extant
1047 			 * at one time, apply that limit.
1048 			 */
1049 			if (serialize ||
1050 			    (max_running && (num_running >= max_running))) {
1051 				pass_done = 0;
1052 				break;
1053 			}
1054 		}
1055 		if (cancel_requested)
1056 			break;
1057 		if (verbose > 1)
1058 			printf(_("--waiting-- (pass %d)\n"), passno);
1059 		status |= wait_many(pass_done ? FLAG_WAIT_ALL :
1060 				    FLAG_WAIT_ATLEAST_ONE);
1061 		if (pass_done) {
1062 			if (verbose > 1)
1063 				printf("----------------------------------\n");
1064 			passno++;
1065 		} else
1066 			not_done_yet++;
1067 	}
1068 	if (cancel_requested && !kill_sent) {
1069 		kill_all(SIGTERM);
1070 		kill_sent++;
1071 	}
1072 	status |= wait_many(FLAG_WAIT_ATLEAST_ONE);
1073 	return status;
1074 }
1075 
usage(NOARGS)1076 static void usage(NOARGS)
1077 {
1078 	fputs(_("Usage: fsck [-AMNPRTV] [ -C [ fd ] ] [-t fstype] [fs-options] [filesys ...]\n"), stderr);
1079 	exit(EXIT_USAGE);
1080 }
1081 
1082 #ifdef HAVE_SIGNAL_H
signal_cancel(int sig FSCK_ATTR ((unused)))1083 static void signal_cancel(int sig FSCK_ATTR((unused)))
1084 {
1085 	cancel_requested++;
1086 }
1087 #endif
1088 
PRS(int argc,char * argv[])1089 static void PRS(int argc, char *argv[])
1090 {
1091 	int	i, j;
1092 	char	*arg, *dev, *tmp = 0;
1093 	char	options[128];
1094 	int	opt = 0;
1095 	int     opts_for_fsck = 0;
1096 #ifdef HAVE_SIGNAL_H
1097 	struct sigaction	sa;
1098 
1099 	/*
1100 	 * Set up signal action
1101 	 */
1102 	memset(&sa, 0, sizeof(struct sigaction));
1103 	sa.sa_handler = signal_cancel;
1104 	sigaction(SIGINT, &sa, 0);
1105 	sigaction(SIGTERM, &sa, 0);
1106 #endif
1107 
1108 	num_devices = 0;
1109 	num_args = 0;
1110 	instance_list = 0;
1111 
1112 	progname = argv[0];
1113 
1114 	for (i=1; i < argc; i++) {
1115 		arg = argv[i];
1116 		if (!arg)
1117 			continue;
1118 		if ((arg[0] == '/' && !opts_for_fsck) || strchr(arg, '=')) {
1119 			if (num_devices >= MAX_DEVICES) {
1120 				fprintf(stderr, _("%s: too many devices\n"),
1121 					progname);
1122 				exit(EXIT_ERROR);
1123 			}
1124 			dev = blkid_get_devname(cache, arg, NULL);
1125 			if (!dev && strchr(arg, '=')) {
1126 				/*
1127 				 * Check to see if we failed because
1128 				 * /proc/partitions isn't found.
1129 				 */
1130 				if (access("/proc/partitions", R_OK) < 0) {
1131 					fprintf(stderr, "Couldn't open /proc/partitions: %s\n",
1132 						strerror(errno));
1133 					fprintf(stderr, "Is /proc mounted?\n");
1134 					exit(EXIT_ERROR);
1135 				}
1136 				/*
1137 				 * Check to see if this is because
1138 				 * we're not running as root
1139 				 */
1140 				if (geteuid())
1141 					fprintf(stderr,
1142 		"Must be root to scan for matching filesystems: %s\n", arg);
1143 				else
1144 					fprintf(stderr,
1145 		"Couldn't find matching filesystem: %s\n", arg);
1146 				exit(EXIT_ERROR);
1147 			}
1148 			devices[num_devices++] = dev ? dev : string_copy(arg);
1149 			continue;
1150 		}
1151 		if (arg[0] != '-' || opts_for_fsck) {
1152 			if (num_args >= MAX_ARGS) {
1153 				fprintf(stderr, _("%s: too many arguments\n"),
1154 					progname);
1155 				exit(EXIT_ERROR);
1156 			}
1157 			args[num_args++] = string_copy(arg);
1158 			continue;
1159 		}
1160 		for (j=1; arg[j]; j++) {
1161 			if (opts_for_fsck) {
1162 				options[++opt] = arg[j];
1163 				continue;
1164 			}
1165 			switch (arg[j]) {
1166 			case 'A':
1167 				doall++;
1168 				break;
1169 			case 'C':
1170 				progress++;
1171 				if (arg[j+1]) {
1172 					progress_fd = string_to_int(arg+j+1);
1173 					if (progress_fd < 0)
1174 						progress_fd = 0;
1175 					else
1176 						goto next_arg;
1177 				} else if (argc > i + 1 &&
1178 					   argv[i + 1][0] != '-') {
1179 					progress_fd = string_to_int(argv[i]);
1180 					if (progress_fd < 0)
1181 						progress_fd = 0;
1182 					else {
1183 						++i;
1184 						goto next_arg;
1185 					}
1186 				}
1187 				break;
1188 			case 'V':
1189 				verbose++;
1190 				break;
1191 			case 'N':
1192 				noexecute++;
1193 				break;
1194 			case 'R':
1195 				skip_root++;
1196 				break;
1197 			case 'T':
1198 				notitle++;
1199 				break;
1200 			case 'M':
1201 				ignore_mounted++;
1202 				break;
1203 			case 'P':
1204 				parallel_root++;
1205 				break;
1206 			case 's':
1207 				serialize++;
1208 				break;
1209 			case 't':
1210 				tmp = 0;
1211 				if (fstype)
1212 					usage();
1213 				if (arg[j+1])
1214 					tmp = arg+j+1;
1215 				else if ((i+1) < argc)
1216 					tmp = argv[++i];
1217 				else
1218 					usage();
1219 				fstype = string_copy(tmp);
1220 				compile_fs_type(fstype, &fs_type_compiled);
1221 				goto next_arg;
1222 			case '-':
1223 				opts_for_fsck++;
1224 				break;
1225 			case '?':
1226 				usage();
1227 				break;
1228 			default:
1229 				options[++opt] = arg[j];
1230 				break;
1231 			}
1232 		}
1233 	next_arg:
1234 		if (opt) {
1235 			options[0] = '-';
1236 			options[++opt] = '\0';
1237 			if (num_args >= MAX_ARGS) {
1238 				fprintf(stderr,
1239 					_("%s: too many arguments\n"),
1240 					progname);
1241 				exit(EXIT_ERROR);
1242 			}
1243 			args[num_args++] = string_copy(options);
1244 			opt = 0;
1245 		}
1246 	}
1247 	if (getenv("FSCK_FORCE_ALL_PARALLEL"))
1248 		force_all_parallel++;
1249 	if ((tmp = getenv("FSCK_MAX_INST")))
1250 	    max_running = atoi(tmp);
1251 }
1252 
main(int argc,char * argv[])1253 int main(int argc, char *argv[])
1254 {
1255 	int i, status = 0;
1256 	int interactive = 0;
1257 	char *oldpath = getenv("PATH");
1258 	const char *fstab;
1259 	struct fs_info *fs;
1260 
1261 	setvbuf(stdout, NULL, _IONBF, BUFSIZ);
1262 	setvbuf(stderr, NULL, _IONBF, BUFSIZ);
1263 
1264 #ifdef ENABLE_NLS
1265 	setlocale(LC_MESSAGES, "");
1266 	setlocale(LC_CTYPE, "");
1267 	bindtextdomain(NLS_CAT_NAME, LOCALEDIR);
1268 	textdomain(NLS_CAT_NAME);
1269 #endif
1270 	blkid_get_cache(&cache, NULL);
1271 	PRS(argc, argv);
1272 
1273 	if (!notitle)
1274 		printf("fsck %s (%s)\n", E2FSPROGS_VERSION, E2FSPROGS_DATE);
1275 
1276 	fstab = getenv("FSTAB_FILE");
1277 	if (!fstab)
1278 		fstab = _PATH_MNTTAB;
1279 	load_fs_info(fstab);
1280 
1281 	/* Update our search path to include uncommon directories. */
1282 	if (oldpath) {
1283 		fsck_path = malloc (strlen (fsck_prefix_path) + 1 +
1284 				    strlen (oldpath) + 1);
1285 		if (!fsck_path) {
1286 			fprintf(stderr, "%s: Unable to allocate memory for fsck_path\n", progname);
1287 			exit(EXIT_ERROR);
1288 		}
1289 		strcpy (fsck_path, fsck_prefix_path);
1290 		strcat (fsck_path, ":");
1291 		strcat (fsck_path, oldpath);
1292 	} else {
1293 		fsck_path = string_copy(fsck_prefix_path);
1294 	}
1295 
1296 	if ((num_devices == 1) || (serialize))
1297 		interactive = 1;
1298 
1299 	/* If -A was specified ("check all"), do that! */
1300 	if (doall)
1301 		return check_all();
1302 
1303 	if (num_devices == 0) {
1304 		serialize++;
1305 		interactive++;
1306 		return check_all();
1307 	}
1308 	for (i = 0 ; i < num_devices; i++) {
1309 		if (cancel_requested) {
1310 			if (!kill_sent) {
1311 				kill_all(SIGTERM);
1312 				kill_sent++;
1313 			}
1314 			break;
1315 		}
1316 		fs = lookup(devices[i]);
1317 		if (!fs) {
1318 			fs = create_fs_device(devices[i], 0, "auto",
1319 					      0, -1, -1);
1320 			if (!fs)
1321 				continue;
1322 		}
1323 		if (ignore_mounted && is_mounted(fs->device))
1324 			continue;
1325 		fsck_device(fs, interactive);
1326 		if (serialize ||
1327 		    (max_running && (num_running >= max_running))) {
1328 			struct fsck_instance *inst;
1329 
1330 			inst = wait_one(0);
1331 			if (inst) {
1332 				status |= inst->exit_status;
1333 				free_instance(inst);
1334 			}
1335 			if (verbose > 1)
1336 				printf("----------------------------------\n");
1337 		}
1338 	}
1339 	status |= wait_many(FLAG_WAIT_ALL);
1340 	free(fsck_path);
1341 	blkid_put_cache(cache);
1342 	return status;
1343 }
1344