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