e2fsprogs/fsck.c

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