Initial public busybox upstream commit
[busybox4maemo] / e2fsprogs / fsck.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * fsck --- A generic, parallelizing front-end for the fsck program.
4  * It will automatically try to run fsck programs in parallel if the
5  * devices are on separate spindles.  It is based on the same ideas as
6  * the generic front end for fsck by David Engel and Fred van Kempen,
7  * but it has been completely rewritten from scratch to support
8  * parallel execution.
9  *
10  * Written by Theodore Ts'o, <tytso@mit.edu>
11  *
12  * Miquel van Smoorenburg (miquels@drinkel.ow.org) 20-Oct-1994:
13  *   o Changed -t fstype to behave like with mount when -A (all file
14  *     systems) or -M (like mount) is specified.
15  *   o fsck looks if it can find the fsck.type program to decide
16  *     if it should ignore the fs type. This way more fsck programs
17  *     can be added without changing this front-end.
18  *   o -R flag skip root file system.
19  *
20  * Copyright (C) 1993, 1994, 1995, 1996, 1997, 1998, 1999, 2000,
21  *      2001, 2002, 2003, 2004, 2005 by  Theodore Ts'o.
22  *
23  * %Begin-Header%
24  * This file may be redistributed under the terms of the GNU Public
25  * License.
26  * %End-Header%
27  */
28
29 /* All filesystem specific hooks have been removed.
30  * If filesystem cannot be determined, we will execute
31  * "fsck.auto". Currently this also happens if you specify
32  * UUID=xxx or LABEL=xxx as an object to check.
33  * Detection code for that is also probably has to be in fsck.auto.
34  *
35  * In other words, this is _really_ is just a driver program which
36  * spawns actual fsck.something for each filesystem to check.
37  * It doesn't guess filesystem types from on-disk format.
38  */
39
40 #include "libbb.h"
41
42 /* "progress indicator" code is somewhat buggy and ext[23] specific.
43  * We should be filesystem agnostic. IOW: there should be a well-defined
44  * API for fsck.something, NOT ad-hoc hacks in generic fsck. */
45 #define DO_PROGRESS_INDICATOR 0
46
47 #define EXIT_OK          0
48 #define EXIT_NONDESTRUCT 1
49 #define EXIT_DESTRUCT    2
50 #define EXIT_UNCORRECTED 4
51 #define EXIT_ERROR       8
52 #define EXIT_USAGE       16
53 #define FSCK_CANCELED    32     /* Aborted with a signal or ^C */
54
55 /*
56  * Internal structure for mount table entries.
57  */
58
59 struct fs_info {
60         struct fs_info *next;
61         char    *device;
62         char    *mountpt;
63         char    *type;
64         char    *opts;
65         int     passno;
66         int     flags;
67 };
68
69 #define FLAG_DONE 1
70 #define FLAG_PROGRESS 2
71 /*
72  * Structure to allow exit codes to be stored
73  */
74 struct fsck_instance {
75         struct fsck_instance *next;
76         int     pid;
77         int     flags;
78 #if DO_PROGRESS_INDICATOR
79         time_t  start_time;
80 #endif
81         char    *prog;
82         char    *device;
83         char    *base_device; /* /dev/hda for /dev/hdaN etc */
84 };
85
86 static const char ignored_types[] ALIGN1 =
87         "ignore\0"
88         "iso9660\0"
89         "nfs\0"
90         "proc\0"
91         "sw\0"
92         "swap\0"
93         "tmpfs\0"
94         "devpts\0";
95
96 #if 0
97 static const char really_wanted[] ALIGN1 =
98         "minix\0"
99         "ext2\0"
100         "ext3\0"
101         "jfs\0"
102         "reiserfs\0"
103         "xiafs\0"
104         "xfs\0";
105 #endif
106
107 #define BASE_MD "/dev/md"
108
109 static char **devices;
110 static char **args;
111 static int num_devices;
112 static int num_args;
113 static int verbose;
114
115 #define FS_TYPE_FLAG_NORMAL 0
116 #define FS_TYPE_FLAG_OPT    1
117 #define FS_TYPE_FLAG_NEGOPT 2
118 static char **fs_type_list;
119 static uint8_t *fs_type_flag;
120 static smallint fs_type_negated;
121
122 static volatile smallint cancel_requested;
123 static smallint doall;
124 static smallint noexecute;
125 static smallint serialize;
126 static smallint skip_root;
127 /* static smallint like_mount; */
128 static smallint notitle;
129 static smallint parallel_root;
130 static smallint force_all_parallel;
131
132 #if DO_PROGRESS_INDICATOR
133 static smallint progress;
134 static int progress_fd;
135 #endif
136
137 static int num_running;
138 static int max_running;
139 static char *fstype;
140 static struct fs_info *filesys_info;
141 static struct fs_info *filesys_last;
142 static struct fsck_instance *instance_list;
143
144 /*
145  * Return the "base device" given a particular device; this is used to
146  * assure that we only fsck one partition on a particular drive at any
147  * one time.  Otherwise, the disk heads will be seeking all over the
148  * place.  If the base device cannot be determined, return NULL.
149  *
150  * The base_device() function returns an allocated string which must
151  * be freed.
152  */
153 #if ENABLE_FEATURE_DEVFS
154 /*
155  * Required for the uber-silly devfs /dev/ide/host1/bus2/target3/lun3
156  * pathames.
157  */
158 static const char *const devfs_hier[] = {
159         "host", "bus", "target", "lun", NULL
160 };
161 #endif
162
163 static char *base_device(const char *device)
164 {
165         char *str, *cp;
166 #if ENABLE_FEATURE_DEVFS
167         const char *const *hier;
168         const char *disk;
169         int len;
170 #endif
171         cp = str = xstrdup(device);
172
173         /* Skip over /dev/; if it's not present, give up. */
174         if (strncmp(cp, "/dev/", 5) != 0)
175                 goto errout;
176         cp += 5;
177
178         /*
179          * For md devices, we treat them all as if they were all
180          * on one disk, since we don't know how to parallelize them.
181          */
182         if (cp[0] == 'm' && cp[1] == 'd') {
183                 cp[2] = 0;
184                 return str;
185         }
186
187         /* Handle DAC 960 devices */
188         if (strncmp(cp, "rd/", 3) == 0) {
189                 cp += 3;
190                 if (cp[0] != 'c' || !isdigit(cp[1])
191                  || cp[2] != 'd' || !isdigit(cp[3]))
192                         goto errout;
193                 cp[4] = 0;
194                 return str;
195         }
196
197         /* Now let's handle /dev/hd* and /dev/sd* devices.... */
198         if ((cp[0] == 'h' || cp[0] == 's') && cp[1] == 'd') {
199                 cp += 2;
200                 /* If there's a single number after /dev/hd, skip it */
201                 if (isdigit(*cp))
202                         cp++;
203                 /* What follows must be an alpha char, or give up */
204                 if (!isalpha(*cp))
205                         goto errout;
206                 cp[1] = 0;
207                 return str;
208         }
209
210 #if ENABLE_FEATURE_DEVFS
211         /* Now let's handle devfs (ugh) names */
212         len = 0;
213         if (strncmp(cp, "ide/", 4) == 0)
214                 len = 4;
215         if (strncmp(cp, "scsi/", 5) == 0)
216                 len = 5;
217         if (len) {
218                 cp += len;
219                 /*
220                  * Now we proceed down the expected devfs hierarchy.
221                  * i.e., .../host1/bus2/target3/lun4/...
222                  * If we don't find the expected token, followed by
223                  * some number of digits at each level, abort.
224                  */
225                 for (hier = devfs_hier; *hier; hier++) {
226                         len = strlen(*hier);
227                         if (strncmp(cp, *hier, len) != 0)
228                                 goto errout;
229                         cp += len;
230                         while (*cp != '/' && *cp != 0) {
231                                 if (!isdigit(*cp))
232                                         goto errout;
233                                 cp++;
234                         }
235                         cp++;
236                 }
237                 cp[-1] = 0;
238                 return str;
239         }
240
241         /* Now handle devfs /dev/disc or /dev/disk names */
242         disk = 0;
243         if (strncmp(cp, "discs/", 6) == 0)
244                 disk = "disc";
245         else if (strncmp(cp, "disks/", 6) == 0)
246                 disk = "disk";
247         if (disk) {
248                 cp += 6;
249                 if (strncmp(cp, disk, 4) != 0)
250                         goto errout;
251                 cp += 4;
252                 while (*cp != '/' && *cp != 0) {
253                         if (!isdigit(*cp))
254                                 goto errout;
255                         cp++;
256                 }
257                 *cp = 0;
258                 return str;
259         }
260 #endif
261  errout:
262         free(str);
263         return NULL;
264 }
265
266 static void free_instance(struct fsck_instance *p)
267 {
268         free(p->prog);
269         free(p->device);
270         free(p->base_device);
271         free(p);
272 }
273
274 static struct fs_info *create_fs_device(const char *device, const char *mntpnt,
275                                         const char *type, const char *opts,
276                                         int passno)
277 {
278         struct fs_info *fs;
279
280         fs = xzalloc(sizeof(*fs));
281         fs->device = xstrdup(device);
282         fs->mountpt = xstrdup(mntpnt);
283         fs->type = xstrdup(type);
284         fs->opts = xstrdup(opts ? opts : "");
285         fs->passno = passno;
286         /*fs->flags = 0; */
287         /*fs->next = NULL; */
288
289         if (!filesys_info)
290                 filesys_info = fs;
291         else
292                 filesys_last->next = fs;
293         filesys_last = fs;
294
295         return fs;
296 }
297
298 static void strip_line(char *line)
299 {
300         char *p = line + strlen(line) - 1;
301
302         while (*line) {
303                 if (*p != '\n' && *p != '\r')
304                         break;
305                 *p-- = '\0';
306         }
307 }
308
309 static char *parse_word(char **buf)
310 {
311         char *word, *next;
312
313         word = *buf;
314         if (*word == '\0')
315                 return NULL;
316
317         word = skip_whitespace(word);
318         next = skip_non_whitespace(word);
319         if (*next)
320                 *next++ = '\0';
321         *buf = next;
322         return word;
323 }
324
325 static void parse_escape(char *word)
326 {
327         char *q, c;
328         const char *p;
329
330         if (!word)
331                 return;
332
333         for (p = q = word; *p; q++) {
334                 c = *p++;
335                 if (c != '\\') {
336                         *q = c;
337                 } else {
338                         *q = bb_process_escape_sequence(&p);
339                 }
340         }
341         *q = '\0';
342 }
343
344 static int parse_fstab_line(char *line, struct fs_info **ret_fs)
345 {
346         char *device, *mntpnt, *type, *opts, *passno, *cp;
347         struct fs_info *fs;
348
349         *ret_fs = NULL;
350         strip_line(line);
351         *strchrnul(line, '#') = '\0'; /* Ignore everything after comment */
352         cp = line;
353
354         device = parse_word(&cp);
355         if (!device) return 0; /* Allow blank lines */
356         mntpnt = parse_word(&cp);
357         type = parse_word(&cp);
358         opts = parse_word(&cp);
359         /*freq =*/ parse_word(&cp);
360         passno = parse_word(&cp);
361
362         if (!mntpnt || !type)
363                 return -1;
364
365         parse_escape(device);
366         parse_escape(mntpnt);
367         parse_escape(type);
368         parse_escape(opts);
369         parse_escape(passno);
370
371         if (strchr(type, ','))
372                 type = NULL;
373
374         fs = create_fs_device(device, mntpnt, type ? type : "auto", opts,
375                         (passno ? atoi(passno) : -1));
376         *ret_fs = fs;
377         return 0;
378 }
379
380 /* Load the filesystem database from /etc/fstab */
381 static void load_fs_info(const char *filename)
382 {
383         FILE *f;
384         int lineno = 0;
385         int old_fstab = 1;
386         struct fs_info *fs;
387
388         f = fopen_or_warn(filename, "r");
389         if (f == NULL) {
390                 return;
391         }
392         while (1) {
393                 int r;
394                 char *buf = xmalloc_getline(f);
395                 if (!buf) break;
396                 r = parse_fstab_line(buf, &fs);
397                 free(buf);
398                 lineno++;
399                 if (r < 0) {
400                         bb_error_msg("WARNING: bad format "
401                                 "on line %d of %s", lineno, filename);
402                         continue;
403                 }
404                 if (!fs)
405                         continue;
406                 if (fs->passno < 0)
407                         fs->passno = 0;
408                 else
409                         old_fstab = 0;
410         }
411         fclose(f);
412
413         if (old_fstab) {
414                 fputs("\007"
415 "WARNING: Your /etc/fstab does not contain the fsck passno field.\n"
416 "I will kludge around things for you, but you should fix\n"
417 "your /etc/fstab file as soon as you can.\n\n", stderr);
418                 for (fs = filesys_info; fs; fs = fs->next) {
419                         fs->passno = 1;
420                 }
421         }
422 }
423
424 /* Lookup filesys in /etc/fstab and return the corresponding entry. */
425 static struct fs_info *lookup(char *filesys)
426 {
427         struct fs_info *fs;
428
429         for (fs = filesys_info; fs; fs = fs->next) {
430                 if (strcmp(filesys, fs->device) == 0
431                  || (fs->mountpt && strcmp(filesys, fs->mountpt) == 0)
432                 )
433                         break;
434         }
435
436         return fs;
437 }
438
439 #if DO_PROGRESS_INDICATOR
440 static int progress_active(void)
441 {
442         struct fsck_instance *inst;
443
444         for (inst = instance_list; inst; inst = inst->next) {
445                 if (inst->flags & FLAG_DONE)
446                         continue;
447                 if (inst->flags & FLAG_PROGRESS)
448                         return 1;
449         }
450         return 0;
451 }
452 #endif
453
454
455 /*
456  * Send a signal to all outstanding fsck child processes
457  */
458 static void kill_all_if_cancel_requested(void)
459 {
460         static smallint kill_sent;
461
462         struct fsck_instance *inst;
463
464         if (!cancel_requested || kill_sent)
465                 return;
466
467         for (inst = instance_list; inst; inst = inst->next) {
468                 if (inst->flags & FLAG_DONE)
469                         continue;
470                 kill(inst->pid, SIGTERM);
471         }
472         kill_sent = 1;
473 }
474
475 /*
476  * Wait for one child process to exit; when it does, unlink it from
477  * the list of executing child processes, free, and return its exit status.
478  * If there is no exited child, return -1.
479  */
480 static int wait_one(int flags)
481 {
482         int status;
483         int sig;
484         struct fsck_instance *inst, *prev;
485         pid_t pid;
486
487         if (!instance_list)
488                 return -1;
489         /* if (noexecute) { already returned -1; } */
490
491         while (1) {
492                 pid = waitpid(-1, &status, flags);
493                 kill_all_if_cancel_requested();
494                 if (pid == 0) /* flags == WNOHANG and no children exited */
495                         return -1;
496                 if (pid < 0) {
497                         if (errno == EINTR)
498                                 continue;
499                         if (errno == ECHILD) { /* paranoia */
500                                 bb_error_msg("wait: no more children");
501                                 return -1;
502                         }
503                         bb_perror_msg("wait");
504                         continue;
505                 }
506                 prev = NULL;
507                 inst = instance_list;
508                 do {
509                         if (inst->pid == pid)
510                                 goto child_died;
511                         prev = inst;
512                         inst = inst->next;
513                 } while (inst);
514         }
515  child_died:
516
517         if (WIFEXITED(status))
518                 status = WEXITSTATUS(status);
519         else if (WIFSIGNALED(status)) {
520                 sig = WTERMSIG(status);
521                 status = EXIT_UNCORRECTED;
522                 if (sig != SIGINT) {
523                         printf("Warning: %s %s terminated "
524                                 "by signal %d\n",
525                                 inst->prog, inst->device, sig);
526                         status = EXIT_ERROR;
527                 }
528         } else {
529                 printf("%s %s: status is %x, should never happen\n",
530                         inst->prog, inst->device, status);
531                 status = EXIT_ERROR;
532         }
533
534 #if DO_PROGRESS_INDICATOR
535         if (progress && (inst->flags & FLAG_PROGRESS) && !progress_active()) {
536                 struct fsck_instance *inst2;
537                 for (inst2 = instance_list; inst2; inst2 = inst2->next) {
538                         if (inst2->flags & FLAG_DONE)
539                                 continue;
540                         if (strcmp(inst2->type, "ext2") != 0
541                          && strcmp(inst2->type, "ext3") != 0
542                         ) {
543                                 continue;
544                         }
545                         /* ext[23], we will send USR1
546                          * (request to start displaying progress bar)
547                          *
548                          * If we've just started the fsck, wait a tiny
549                          * bit before sending the kill, to give it
550                          * time to set up the signal handler
551                          */
552                         if (inst2->start_time >= time(NULL) - 1)
553                                 sleep(1);
554                         kill(inst2->pid, SIGUSR1);
555                         inst2->flags |= FLAG_PROGRESS;
556                         break;
557                 }
558         }
559 #endif
560
561         if (prev)
562                 prev->next = inst->next;
563         else
564                 instance_list = inst->next;
565         if (verbose > 1)
566                 printf("Finished with %s (exit status %d)\n",
567                        inst->device, status);
568         num_running--;
569         free_instance(inst);
570
571         return status;
572 }
573
574 /*
575  * Wait until all executing child processes have exited; return the
576  * logical OR of all of their exit code values.
577  */
578 #define FLAG_WAIT_ALL           0
579 #define FLAG_WAIT_ATLEAST_ONE   WNOHANG
580 static int wait_many(int flags)
581 {
582         int exit_status;
583         int global_status = 0;
584         int wait_flags = 0;
585
586         while ((exit_status = wait_one(wait_flags)) != -1) {
587                 global_status |= exit_status;
588                 wait_flags |= flags;
589         }
590         return global_status;
591 }
592
593 /*
594  * Execute a particular fsck program, and link it into the list of
595  * child processes we are waiting for.
596  */
597 static void execute(const char *type, const char *device,
598                 const char *mntpt /*, int interactive */)
599 {
600         char *argv[num_args + 4]; /* see count below: */
601         int argc;
602         int i;
603         struct fsck_instance *inst;
604         pid_t pid;
605
606         argv[0] = xasprintf("fsck.%s", type); /* 1 */
607         for (i = 0; i < num_args; i++)
608                 argv[i+1] = args[i]; /* num_args */
609         argc = num_args + 1;
610
611 #if DO_PROGRESS_INDICATOR
612         if (progress && !progress_active()) {
613                 if (strcmp(type, "ext2") == 0
614                  || strcmp(type, "ext3") == 0
615                 ) {
616                         argv[argc++] = xasprintf("-C%d", progress_fd); /* 1 */
617                         inst->flags |= FLAG_PROGRESS;
618                 }
619         }
620 #endif
621
622         argv[argc++] = (char*)device; /* 1 */
623         argv[argc] = NULL; /* 1 */
624
625         if (verbose || noexecute) {
626                 printf("[%s (%d) -- %s]", argv[0], num_running,
627                                         mntpt ? mntpt : device);
628                 for (i = 0; i < argc; i++)
629                         printf(" %s", argv[i]);
630                 bb_putchar('\n');
631         }
632
633         /* Fork and execute the correct program. */
634         pid = -1;
635         if (!noexecute) {
636                 pid = spawn(argv);
637                 if (pid < 0)
638                         bb_simple_perror_msg(argv[0]);
639         }
640
641 #if DO_PROGRESS_INDICATOR
642         free(argv[num_args + 1]);
643 #endif
644
645         /* No child, so don't record an instance */
646         if (pid <= 0) {
647                 free(argv[0]);
648                 return;
649         }
650
651         inst = xzalloc(sizeof(*inst));
652         inst->pid = pid;
653         inst->prog = argv[0];
654         inst->device = xstrdup(device);
655         inst->base_device = base_device(device);
656 #if DO_PROGRESS_INDICATOR
657         inst->start_time = time(NULL);
658 #endif
659
660         /* Add to the list of running fsck's.
661          * (was adding to the end, but adding to the front is simpler...) */
662         inst->next = instance_list;
663         instance_list = inst;
664 }
665
666 /*
667  * Run the fsck program on a particular device
668  *
669  * If the type is specified using -t, and it isn't prefixed with "no"
670  * (as in "noext2") and only one filesystem type is specified, then
671  * use that type regardless of what is specified in /etc/fstab.
672  *
673  * If the type isn't specified by the user, then use either the type
674  * specified in /etc/fstab, or "auto".
675  */
676 static void fsck_device(struct fs_info *fs /*, int interactive */)
677 {
678         const char *type;
679
680         if (strcmp(fs->type, "auto") != 0) {
681                 type = fs->type;
682                 if (verbose > 2)
683                         bb_info_msg("using filesystem type '%s' %s",
684                                         type, "from fstab");
685         } else if (fstype
686          && (fstype[0] != 'n' || fstype[1] != 'o') /* != "no" */
687          && strncmp(fstype, "opts=", 5) != 0
688          && strncmp(fstype, "loop", 4) != 0
689          && !strchr(fstype, ',')
690         ) {
691                 type = fstype;
692                 if (verbose > 2)
693                         bb_info_msg("using filesystem type '%s' %s",
694                                         type, "from -t");
695         } else {
696                 type = "auto";
697                 if (verbose > 2)
698                         bb_info_msg("using filesystem type '%s' %s",
699                                         type, "(default)");
700         }
701
702         num_running++;
703         execute(type, fs->device, fs->mountpt /*, interactive */);
704 }
705
706 /*
707  * Returns TRUE if a partition on the same disk is already being
708  * checked.
709  */
710 static int device_already_active(char *device)
711 {
712         struct fsck_instance *inst;
713         char *base;
714
715         if (force_all_parallel)
716                 return 0;
717
718 #ifdef BASE_MD
719         /* Don't check a soft raid disk with any other disk */
720         if (instance_list
721          && (!strncmp(instance_list->device, BASE_MD, sizeof(BASE_MD)-1)
722              || !strncmp(device, BASE_MD, sizeof(BASE_MD)-1))
723         ) {
724                 return 1;
725         }
726 #endif
727
728         base = base_device(device);
729         /*
730          * If we don't know the base device, assume that the device is
731          * already active if there are any fsck instances running.
732          */
733         if (!base)
734                 return (instance_list != NULL);
735
736         for (inst = instance_list; inst; inst = inst->next) {
737                 if (!inst->base_device || !strcmp(base, inst->base_device)) {
738                         free(base);
739                         return 1;
740                 }
741         }
742
743         free(base);
744         return 0;
745 }
746
747 /*
748  * This function returns true if a particular option appears in a
749  * comma-delimited options list
750  */
751 static int opt_in_list(char *opt, char *optlist)
752 {
753         char *s;
754         int len;
755
756         if (!optlist)
757                 return 0;
758
759         len = strlen(opt);
760         s = optlist - 1;
761         while (1) {
762                 s = strstr(s + 1, opt);
763                 if (!s)
764                         return 0;
765                 /* neither "opt.." nor "xxx,opt.."? */
766                 if (s != optlist && s[-1] != ',')
767                         continue;
768                 /* neither "..opt" nor "..opt,xxx"? */
769                 if (s[len] != '\0' && s[len] != ',')
770                         continue;
771                 return 1;
772         }
773 }
774
775 /* See if the filesystem matches the criteria given by the -t option */
776 static int fs_match(struct fs_info *fs)
777 {
778         int n, ret, checked_type;
779         char *cp;
780
781         if (!fs_type_list)
782                 return 1;
783
784         ret = 0;
785         checked_type = 0;
786         n = 0;
787         while (1) {
788                 cp = fs_type_list[n];
789                 if (!cp)
790                         break;
791                 switch (fs_type_flag[n]) {
792                 case FS_TYPE_FLAG_NORMAL:
793                         checked_type++;
794                         if (strcmp(cp, fs->type) == 0)
795                                 ret = 1;
796                         break;
797                 case FS_TYPE_FLAG_NEGOPT:
798                         if (opt_in_list(cp, fs->opts))
799                                 return 0;
800                         break;
801                 case FS_TYPE_FLAG_OPT:
802                         if (!opt_in_list(cp, fs->opts))
803                                 return 0;
804                         break;
805                 }
806                 n++;
807         }
808         if (checked_type == 0)
809                 return 1;
810
811         return (fs_type_negated ? !ret : ret);
812 }
813
814 /* Check if we should ignore this filesystem. */
815 static int ignore(struct fs_info *fs)
816 {
817         /*
818          * If the pass number is 0, ignore it.
819          */
820         if (fs->passno == 0)
821                 return 1;
822
823         /*
824          * If a specific fstype is specified, and it doesn't match,
825          * ignore it.
826          */
827         if (!fs_match(fs))
828                 return 1;
829
830         /* Are we ignoring this type? */
831         if (index_in_strings(ignored_types, fs->type) >= 0)
832                 return 1;
833
834         /* We can and want to check this file system type. */
835         return 0;
836 }
837
838 /* Check all file systems, using the /etc/fstab table. */
839 static int check_all(void)
840 {
841         struct fs_info *fs;
842         int status = EXIT_OK;
843         smallint not_done_yet;
844         smallint pass_done;
845         int passno;
846
847         if (verbose)
848                 puts("Checking all filesystems");
849
850         /*
851          * Do an initial scan over the filesystem; mark filesystems
852          * which should be ignored as done, and resolve any "auto"
853          * filesystem types (done as a side-effect of calling ignore()).
854          */
855         for (fs = filesys_info; fs; fs = fs->next)
856                 if (ignore(fs))
857                         fs->flags |= FLAG_DONE;
858
859         /*
860          * Find and check the root filesystem.
861          */
862         if (!parallel_root) {
863                 for (fs = filesys_info; fs; fs = fs->next) {
864                         if (LONE_CHAR(fs->mountpt, '/')) {
865                                 if (!skip_root && !ignore(fs)) {
866                                         fsck_device(fs /*, 1*/);
867                                         status |= wait_many(FLAG_WAIT_ALL);
868                                         if (status > EXIT_NONDESTRUCT)
869                                                 return status;
870                                 }
871                                 fs->flags |= FLAG_DONE;
872                                 break;
873                         }
874                 }
875         }
876         /*
877          * This is for the bone-headed user who has root
878          * filesystem listed twice.
879          * "Skip root" will skip _all_ root entries.
880          */
881         if (skip_root)
882                 for (fs = filesys_info; fs; fs = fs->next)
883                         if (LONE_CHAR(fs->mountpt, '/'))
884                                 fs->flags |= FLAG_DONE;
885
886         not_done_yet = 1;
887         passno = 1;
888         while (not_done_yet) {
889                 not_done_yet = 0;
890                 pass_done = 1;
891
892                 for (fs = filesys_info; fs; fs = fs->next) {
893                         if (cancel_requested)
894                                 break;
895                         if (fs->flags & FLAG_DONE)
896                                 continue;
897                         /*
898                          * If the filesystem's pass number is higher
899                          * than the current pass number, then we didn't
900                          * do it yet.
901                          */
902                         if (fs->passno > passno) {
903                                 not_done_yet = 1;
904                                 continue;
905                         }
906                         /*
907                          * If a filesystem on a particular device has
908                          * already been spawned, then we need to defer
909                          * this to another pass.
910                          */
911                         if (device_already_active(fs->device)) {
912                                 pass_done = 0;
913                                 continue;
914                         }
915                         /*
916                          * Spawn off the fsck process
917                          */
918                         fsck_device(fs /*, serialize*/);
919                         fs->flags |= FLAG_DONE;
920
921                         /*
922                          * Only do one filesystem at a time, or if we
923                          * have a limit on the number of fsck's extant
924                          * at one time, apply that limit.
925                          */
926                         if (serialize
927                          || (max_running && (num_running >= max_running))
928                         ) {
929                                 pass_done = 0;
930                                 break;
931                         }
932                 }
933                 if (cancel_requested)
934                         break;
935                 if (verbose > 1)
936                         printf("--waiting-- (pass %d)\n", passno);
937                 status |= wait_many(pass_done ? FLAG_WAIT_ALL :
938                                     FLAG_WAIT_ATLEAST_ONE);
939                 if (pass_done) {
940                         if (verbose > 1)
941                                 puts("----------------------------------");
942                         passno++;
943                 } else
944                         not_done_yet = 1;
945         }
946         kill_all_if_cancel_requested();
947         status |= wait_many(FLAG_WAIT_ATLEAST_ONE);
948         return status;
949 }
950
951 /*
952  * Deal with the fsck -t argument.
953  * Huh, for mount "-t novfat,nfs" means "neither vfat nor nfs"!
954  * Why here we require "-t novfat,nonfs" ??
955  */
956 static void compile_fs_type(char *fs_type)
957 {
958         char *s;
959         int num = 2;
960         smallint negate;
961
962         s = fs_type;
963         while ((s = strchr(s, ','))) {
964                 num++;
965                 s++;
966         }
967
968         fs_type_list = xzalloc(num * sizeof(fs_type_list[0]));
969         fs_type_flag = xzalloc(num * sizeof(fs_type_flag[0]));
970         fs_type_negated = -1; /* not yet known is it negated or not */
971
972         num = 0;
973         s = fs_type;
974         while (1) {
975                 char *comma;
976
977                 negate = 0;
978                 if (s[0] == 'n' && s[1] == 'o') { /* "no.." */
979                         s += 2;
980                         negate = 1;
981                 } else if (s[0] == '!') {
982                         s++;
983                         negate = 1;
984                 }
985
986                 if (strcmp(s, "loop") == 0)
987                         /* loop is really short-hand for opts=loop */
988                         goto loop_special_case;
989                 if (strncmp(s, "opts=", 5) == 0) {
990                         s += 5;
991  loop_special_case:
992                         fs_type_flag[num] = negate ? FS_TYPE_FLAG_NEGOPT : FS_TYPE_FLAG_OPT;
993                 } else {
994                         if (fs_type_negated == -1)
995                                 fs_type_negated = negate;
996                         if (fs_type_negated != negate)
997                                 bb_error_msg_and_die(
998 "either all or none of the filesystem types passed to -t must be prefixed "
999 "with 'no' or '!'");
1000                 }
1001                 comma = strchr(s, ',');
1002                 fs_type_list[num++] = comma ? xstrndup(s, comma-s) : xstrdup(s);
1003                 if (!comma)
1004                         break;
1005                 s = comma + 1;
1006         }
1007 }
1008
1009 static void parse_args(char **argv)
1010 {
1011         int i, j;
1012         char *arg, *tmp;
1013         char *options;
1014         int optpos;
1015         int opts_for_fsck = 0;
1016
1017         /* in bss, so already zeroed
1018         num_devices = 0;
1019         num_args = 0;
1020         instance_list = NULL;
1021         */
1022
1023         for (i = 1; argv[i]; i++) {
1024                 arg = argv[i];
1025
1026                 /* "/dev/blk" or "/path" or "UUID=xxx" or "LABEL=xxx" */
1027                 if ((arg[0] == '/' && !opts_for_fsck) || strchr(arg, '=')) {
1028 // FIXME: must check that arg is a blkdev, or resolve
1029 // "/path", "UUID=xxx" or "LABEL=xxx" into block device name
1030 // ("UUID=xxx"/"LABEL=xxx" can probably shifted to fsck.auto duties)
1031                         devices = xrealloc(devices, (num_devices+1) * sizeof(devices[0]));
1032                         devices[num_devices++] = xstrdup(arg);
1033                         continue;
1034                 }
1035
1036                 if (arg[0] != '-' || opts_for_fsck) {
1037                         args = xrealloc(args, (num_args+1) * sizeof(args[0]));
1038                         args[num_args++] = xstrdup(arg);
1039                         continue;
1040                 }
1041
1042                 if (LONE_CHAR(arg + 1, '-')) { /* "--" ? */
1043                         opts_for_fsck = 1;
1044                         continue;
1045                 }
1046
1047                 optpos = 0;
1048                 options = NULL;
1049                 for (j = 1; arg[j]; j++) {
1050                         switch (arg[j]) {
1051                         case 'A':
1052                                 doall = 1;
1053                                 break;
1054 #if DO_PROGRESS_INDICATOR
1055                         case 'C':
1056                                 progress = 1;
1057                                 if (arg[++j]) { /* -Cn */
1058                                         progress_fd = xatoi_u(&arg[j]);
1059                                         goto next_arg;
1060                                 }
1061                                 /* -C n */
1062                                 if (!argv[++i]) bb_show_usage();
1063                                 progress_fd = xatoi_u(argv[i]);
1064                                 goto next_arg;
1065 #endif
1066                         case 'V':
1067                                 verbose++;
1068                                 break;
1069                         case 'N':
1070                                 noexecute = 1;
1071                                 break;
1072                         case 'R':
1073                                 skip_root = 1;
1074                                 break;
1075                         case 'T':
1076                                 notitle = 1;
1077                                 break;
1078 /*                      case 'M':
1079                                 like_mount = 1;
1080                                 break; */
1081                         case 'P':
1082                                 parallel_root = 1;
1083                                 break;
1084                         case 's':
1085                                 serialize = 1;
1086                                 break;
1087                         case 't':
1088                                 if (fstype)
1089                                         bb_show_usage();
1090                                 if (arg[++j])
1091                                         tmp = &arg[j];
1092                                 else if (argv[++i])
1093                                         tmp = argv[i];
1094                                 else
1095                                         bb_show_usage();
1096                                 fstype = xstrdup(tmp);
1097                                 compile_fs_type(fstype);
1098                                 goto next_arg;
1099                         case '?':
1100                                 bb_show_usage();
1101                                 break;
1102                         default:
1103                                 optpos++;
1104                                 /* one extra for '\0' */
1105                                 options = xrealloc(options, optpos + 2);
1106                                 options[optpos] = arg[j];
1107                                 break;
1108                         }
1109                 }
1110  next_arg:
1111                 if (optpos) {
1112                         options[0] = '-';
1113                         options[optpos + 1] = '\0';
1114                         args = xrealloc(args, (num_args+1) * sizeof(args[0]));
1115                         args[num_args++] = options;
1116                 }
1117         }
1118         if (getenv("FSCK_FORCE_ALL_PARALLEL"))
1119                 force_all_parallel = 1;
1120         tmp = getenv("FSCK_MAX_INST");
1121         if (tmp)
1122                 max_running = xatoi(tmp);
1123 }
1124
1125 static void signal_cancel(int sig ATTRIBUTE_UNUSED)
1126 {
1127         cancel_requested = 1;
1128 }
1129
1130 int fsck_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
1131 int fsck_main(int argc ATTRIBUTE_UNUSED, char **argv)
1132 {
1133         int i, status;
1134         /*int interactive;*/
1135         const char *fstab;
1136         struct fs_info *fs;
1137
1138         /* we want wait() to be interruptible */
1139         signal_no_SA_RESTART_empty_mask(SIGINT, signal_cancel);
1140         signal_no_SA_RESTART_empty_mask(SIGTERM, signal_cancel);
1141
1142         setbuf(stdout, NULL);
1143
1144         parse_args(argv);
1145
1146         if (!notitle)
1147                 puts("fsck (busybox "BB_VER", "BB_BT")");
1148
1149         /* Even plain "fsck /dev/hda1" needs fstab to get fs type,
1150          * so we are scanning it anyway */
1151         fstab = getenv("FSTAB_FILE");
1152         if (!fstab)
1153                 fstab = "/etc/fstab";
1154         load_fs_info(fstab);
1155
1156         /*interactive = (num_devices == 1) | serialize;*/
1157
1158         if (num_devices == 0)
1159                 /*interactive =*/ serialize = doall = 1;
1160         if (doall)
1161                 return check_all();
1162
1163         status = 0;
1164         for (i = 0; i < num_devices; i++) {
1165                 if (cancel_requested) {
1166                         kill_all_if_cancel_requested();
1167                         break;
1168                 }
1169
1170                 fs = lookup(devices[i]);
1171                 if (!fs)
1172                         fs = create_fs_device(devices[i], "", "auto", NULL, -1);
1173                 fsck_device(fs /*, interactive */);
1174
1175                 if (serialize
1176                  || (max_running && (num_running >= max_running))
1177                 ) {
1178                         int exit_status = wait_one(0);
1179                         if (exit_status >= 0)
1180                                 status |= exit_status;
1181                         if (verbose > 1)
1182                                 puts("----------------------------------");
1183                 }
1184         }
1185         status |= wait_many(FLAG_WAIT_ALL);
1186         return status;
1187 }