12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115 |
- #include "libbb.h"
- #include "common_bufsiz.h"
- #define DO_PROGRESS_INDICATOR 0
- #define EXIT_OK 0
- #define EXIT_NONDESTRUCT 1
- #define EXIT_DESTRUCT 2
- #define EXIT_UNCORRECTED 4
- #define EXIT_ERROR 8
- #define EXIT_USAGE 16
- #define FSCK_CANCELED 32
- struct fs_info {
- struct fs_info *next;
- char *device;
- char *mountpt;
- char *type;
- char *opts;
- int passno;
- int flags;
- };
- #define FLAG_DONE 1
- #define FLAG_PROGRESS 2
- struct fsck_instance {
- struct fsck_instance *next;
- int pid;
- int flags;
- #if DO_PROGRESS_INDICATOR
- time_t start_time;
- #endif
- char *prog;
- char *device;
- char *base_device;
- };
- static const char ignored_types[] ALIGN1 =
- "ignore\0"
- "iso9660\0"
- "nfs\0"
- "proc\0"
- "sw\0"
- "swap\0"
- "tmpfs\0"
- "devpts\0";
- #if 0
- static const char really_wanted[] ALIGN1 =
- "minix\0"
- "ext2\0"
- "ext3\0"
- "jfs\0"
- "reiserfs\0"
- "xiafs\0"
- "xfs\0";
- #endif
- #define BASE_MD "/dev/md"
- struct globals {
- char **args;
- int num_args;
- int verbose;
- #define FS_TYPE_FLAG_NORMAL 0
- #define FS_TYPE_FLAG_OPT 1
- #define FS_TYPE_FLAG_NEGOPT 2
- char **fs_type_list;
- uint8_t *fs_type_flag;
- smallint fs_type_negated;
- smallint noexecute;
- smallint serialize;
- smallint skip_root;
-
- smallint parallel_root;
- smallint force_all_parallel;
- smallint kill_sent;
- #if DO_PROGRESS_INDICATOR
- smallint progress;
- int progress_fd;
- #endif
- int num_running;
- int max_running;
- char *fstype;
- struct fs_info *filesys_info;
- struct fs_info *filesys_last;
- struct fsck_instance *instance_list;
- } FIX_ALIASING;
- #define G (*(struct globals*)bb_common_bufsiz1)
- #define INIT_G() do { \
- setup_common_bufsiz(); \
- BUILD_BUG_ON(sizeof(G) > COMMON_BUFSIZE); \
- } while (0)
- #if ENABLE_FEATURE_DEVFS
- static const char *const devfs_hier[] = {
- "host", "bus", "target", "lun", NULL
- };
- #endif
- static char *base_device(const char *device)
- {
- char *str, *cp;
- #if ENABLE_FEATURE_DEVFS
- const char *const *hier;
- const char *disk;
- int len;
- #endif
- str = xstrdup(device);
-
- cp = skip_dev_pfx(str);
- if (cp == str)
- goto errout;
-
- if (cp[0] == 'm' && cp[1] == 'd') {
- cp[2] = 0;
- return str;
- }
-
- if (is_prefixed_with(cp, "rd/")) {
- cp += 3;
- if (cp[0] != 'c' || !isdigit(cp[1])
- || cp[2] != 'd' || !isdigit(cp[3]))
- goto errout;
- cp[4] = 0;
- return str;
- }
-
- if ((cp[0] == 'h' || cp[0] == 's') && cp[1] == 'd') {
- cp += 2;
-
- if (isdigit(*cp))
- cp++;
-
- if (!isalpha(*cp))
- goto errout;
- cp[1] = 0;
- return str;
- }
- #if ENABLE_FEATURE_DEVFS
-
- len = 0;
- if (is_prefixed_with(cp, "ide/"))
- len = 4;
- if (is_prefixed_with(cp, "scsi/"))
- len = 5;
- if (len) {
- cp += len;
-
- for (hier = devfs_hier; *hier; hier++) {
- cp = is_prefixed_with(cp, *hier);
- if (!cp)
- goto errout;
- while (*cp != '/' && *cp != '\0') {
- if (!isdigit(*cp))
- goto errout;
- cp++;
- }
- cp++;
- }
- cp[-1] = '\0';
- return str;
- }
-
- disk = NULL;
- if (is_prefixed_with(cp, "discs/"))
- disk = "disc";
- else if (is_prefixed_with(cp, "disks/"))
- disk = "disk";
- if (disk) {
- cp += 6;
- cp = is_prefixed_with(cp, disk);
- if (!cp)
- goto errout;
- while (*cp != '/' && *cp != '\0') {
- if (!isdigit(*cp))
- goto errout;
- cp++;
- }
- *cp = '\0';
- return str;
- }
- #endif
- errout:
- free(str);
- return NULL;
- }
- static void free_instance(struct fsck_instance *p)
- {
- free(p->prog);
- free(p->device);
- free(p->base_device);
- free(p);
- }
- static struct fs_info *create_fs_device(const char *device, const char *mntpnt,
- const char *type, const char *opts,
- int passno)
- {
- struct fs_info *fs;
- fs = xzalloc(sizeof(*fs));
- fs->device = xstrdup(device);
- fs->mountpt = xstrdup(mntpnt);
- if (strchr(type, ','))
- type = (char *)"auto";
- fs->type = xstrdup(type);
- fs->opts = xstrdup(opts ? opts : "");
- fs->passno = passno < 0 ? 1 : passno;
-
-
- if (!G.filesys_info)
- G.filesys_info = fs;
- else
- G.filesys_last->next = fs;
- G.filesys_last = fs;
- return fs;
- }
- static void load_fs_info(const char *filename)
- {
- FILE *fstab;
- struct mntent mte;
- char buf[1024];
- fstab = setmntent(filename, "r");
- if (!fstab) {
- bb_perror_msg("can't read '%s'", filename);
- return;
- }
-
- while (getmntent_r(fstab, &mte, buf, sizeof(buf))) {
-
-
-
- create_fs_device(mte.mnt_fsname, mte.mnt_dir,
- mte.mnt_type, mte.mnt_opts,
- mte.mnt_passno);
- }
- endmntent(fstab);
- }
- static struct fs_info *lookup(char *filesys)
- {
- struct fs_info *fs;
- for (fs = G.filesys_info; fs; fs = fs->next) {
- if (strcmp(filesys, fs->device) == 0
- || (fs->mountpt && strcmp(filesys, fs->mountpt) == 0)
- )
- break;
- }
- return fs;
- }
- #if DO_PROGRESS_INDICATOR
- static int progress_active(void)
- {
- struct fsck_instance *inst;
- for (inst = G.instance_list; inst; inst = inst->next) {
- if (inst->flags & FLAG_DONE)
- continue;
- if (inst->flags & FLAG_PROGRESS)
- return 1;
- }
- return 0;
- }
- #endif
- static void kill_all_if_got_signal(void)
- {
- struct fsck_instance *inst;
- if (!bb_got_signal || G.kill_sent)
- return;
- for (inst = G.instance_list; inst; inst = inst->next) {
- if (inst->flags & FLAG_DONE)
- continue;
- kill(inst->pid, SIGTERM);
- }
- G.kill_sent = 1;
- }
- static int wait_one(int flags)
- {
- int status;
- int exitcode;
- struct fsck_instance *inst, *prev;
- pid_t pid;
- if (!G.instance_list)
- return -1;
-
- while (1) {
- pid = waitpid(-1, &status, flags);
- kill_all_if_got_signal();
- if (pid == 0)
- return -1;
- if (pid < 0) {
- if (errno == EINTR)
- continue;
- if (errno == ECHILD) {
- bb_error_msg("wait: no more children");
- return -1;
- }
- bb_perror_msg("wait");
- continue;
- }
- prev = NULL;
- inst = G.instance_list;
- do {
- if (inst->pid == pid)
- goto child_died;
- prev = inst;
- inst = inst->next;
- } while (inst);
- }
- child_died:
- exitcode = WEXITSTATUS(status);
- if (WIFSIGNALED(status)) {
- unsigned sig;
- sig = WTERMSIG(status);
- exitcode = EXIT_UNCORRECTED;
- if (sig != SIGINT) {
- printf("Warning: %s %s terminated "
- "by signal %u\n",
- inst->prog, inst->device, sig);
- exitcode = EXIT_ERROR;
- }
- }
- #if DO_PROGRESS_INDICATOR
- if (progress && (inst->flags & FLAG_PROGRESS) && !progress_active()) {
- struct fsck_instance *inst2;
- for (inst2 = G.instance_list; inst2; inst2 = inst2->next) {
- if (inst2->flags & FLAG_DONE)
- continue;
- if (strcmp(inst2->type, "ext2") != 0
- && strcmp(inst2->type, "ext3") != 0
- ) {
- continue;
- }
-
- if (inst2->start_time >= time(NULL) - 1)
- sleep(1);
- kill(inst2->pid, SIGUSR1);
- inst2->flags |= FLAG_PROGRESS;
- break;
- }
- }
- #endif
- if (prev)
- prev->next = inst->next;
- else
- G.instance_list = inst->next;
- if (G.verbose > 1)
- printf("Finished with %s (exit status %u)\n",
- inst->device, exitcode);
- G.num_running--;
- free_instance(inst);
- return exitcode;
- }
- #define FLAG_WAIT_ALL 0
- #define FLAG_WAIT_ATLEAST_ONE WNOHANG
- static int wait_many(int flags)
- {
- int exit_status;
- int global_status = 0;
- int wait_flags = 0;
- while ((exit_status = wait_one(wait_flags)) != -1) {
- global_status |= exit_status;
- wait_flags |= flags;
- }
- return global_status;
- }
- static void execute(const char *type, const char *device,
- const char *mntpt )
- {
- int i;
- struct fsck_instance *inst;
- pid_t pid;
- G.args[0] = xasprintf("fsck.%s", type);
- #if DO_PROGRESS_INDICATOR
- if (progress && !progress_active()) {
- if (strcmp(type, "ext2") == 0
- || strcmp(type, "ext3") == 0
- ) {
- G.args[XXX] = xasprintf("-C%d", progress_fd);
- inst->flags |= FLAG_PROGRESS;
- }
- }
- #endif
- G.args[G.num_args - 2] = (char*)device;
-
- if (G.verbose || G.noexecute) {
- printf("[%s (%d) -- %s]", G.args[0], G.num_running,
- mntpt ? mntpt : device);
- for (i = 0; G.args[i]; i++)
- printf(" %s", G.args[i]);
- bb_putchar('\n');
- }
-
- pid = -1;
- if (!G.noexecute) {
- pid = spawn(G.args);
- if (pid < 0)
- bb_simple_perror_msg(G.args[0]);
- }
- #if DO_PROGRESS_INDICATOR
- free(G.args[XXX]);
- #endif
-
- if (pid <= 0) {
- free(G.args[0]);
- return;
- }
- inst = xzalloc(sizeof(*inst));
- inst->pid = pid;
- inst->prog = G.args[0];
- inst->device = xstrdup(device);
- inst->base_device = base_device(device);
- #if DO_PROGRESS_INDICATOR
- inst->start_time = time(NULL);
- #endif
-
- inst->next = G.instance_list;
- G.instance_list = inst;
- }
- static void fsck_device(struct fs_info *fs )
- {
- const char *type;
- if (strcmp(fs->type, "auto") != 0) {
- type = fs->type;
- if (G.verbose > 2)
- printf("using filesystem type '%s' %s\n",
- type, "from fstab");
- } else if (G.fstype
- && (G.fstype[0] != 'n' || G.fstype[1] != 'o')
- && !is_prefixed_with(G.fstype, "opts=")
- && !is_prefixed_with(G.fstype, "loop")
- && !strchr(G.fstype, ',')
- ) {
- type = G.fstype;
- if (G.verbose > 2)
- printf("using filesystem type '%s' %s\n",
- type, "from -t");
- } else {
- type = "auto";
- if (G.verbose > 2)
- printf("using filesystem type '%s' %s\n",
- type, "(default)");
- }
- G.num_running++;
- execute(type, fs->device, fs->mountpt );
- }
- static int device_already_active(char *device)
- {
- struct fsck_instance *inst;
- char *base;
- if (G.force_all_parallel)
- return 0;
- #ifdef BASE_MD
-
- if (G.instance_list
- && (is_prefixed_with(G.instance_list->device, BASE_MD)
- || is_prefixed_with(device, BASE_MD))
- ) {
- return 1;
- }
- #endif
- base = base_device(device);
-
- if (!base)
- return (G.instance_list != NULL);
- for (inst = G.instance_list; inst; inst = inst->next) {
- if (!inst->base_device || strcmp(base, inst->base_device) == 0) {
- free(base);
- return 1;
- }
- }
- free(base);
- return 0;
- }
- static int opt_in_list(char *opt, char *optlist)
- {
- char *s;
- int len;
- if (!optlist)
- return 0;
- len = strlen(opt);
- s = optlist - 1;
- while (1) {
- s = strstr(s + 1, opt);
- if (!s)
- return 0;
-
- if (s != optlist && s[-1] != ',')
- continue;
-
- if (s[len] != '\0' && s[len] != ',')
- continue;
- return 1;
- }
- }
- static int fs_match(struct fs_info *fs)
- {
- int n, ret, checked_type;
- char *cp;
- if (!G.fs_type_list)
- return 1;
- ret = 0;
- checked_type = 0;
- n = 0;
- while (1) {
- cp = G.fs_type_list[n];
- if (!cp)
- break;
- switch (G.fs_type_flag[n]) {
- case FS_TYPE_FLAG_NORMAL:
- checked_type++;
- if (strcmp(cp, fs->type) == 0)
- ret = 1;
- break;
- case FS_TYPE_FLAG_NEGOPT:
- if (opt_in_list(cp, fs->opts))
- return 0;
- break;
- case FS_TYPE_FLAG_OPT:
- if (!opt_in_list(cp, fs->opts))
- return 0;
- break;
- }
- n++;
- }
- if (checked_type == 0)
- return 1;
- return (G.fs_type_negated ? !ret : ret);
- }
- static int ignore(struct fs_info *fs)
- {
-
- if (fs->passno == 0)
- return 1;
-
- if (!fs_match(fs))
- return 1;
-
- if (index_in_strings(ignored_types, fs->type) >= 0)
- return 1;
-
- return 0;
- }
- static int check_all(void)
- {
- struct fs_info *fs;
- int status = EXIT_OK;
- smallint not_done_yet;
- smallint pass_done;
- int passno;
- if (G.verbose)
- puts("Checking all filesystems");
-
- for (fs = G.filesys_info; fs; fs = fs->next)
- if (ignore(fs))
- fs->flags |= FLAG_DONE;
-
- if (!G.parallel_root) {
- for (fs = G.filesys_info; fs; fs = fs->next) {
- if (LONE_CHAR(fs->mountpt, '/')) {
- if (!G.skip_root && !ignore(fs)) {
- fsck_device(fs );
- status |= wait_many(FLAG_WAIT_ALL);
- if (status > EXIT_NONDESTRUCT)
- return status;
- }
- fs->flags |= FLAG_DONE;
- break;
- }
- }
- }
-
- if (G.skip_root)
- for (fs = G.filesys_info; fs; fs = fs->next)
- if (LONE_CHAR(fs->mountpt, '/'))
- fs->flags |= FLAG_DONE;
- not_done_yet = 1;
- passno = 1;
- while (not_done_yet) {
- not_done_yet = 0;
- pass_done = 1;
- for (fs = G.filesys_info; fs; fs = fs->next) {
- if (bb_got_signal)
- break;
- if (fs->flags & FLAG_DONE)
- continue;
-
- if (fs->passno > passno) {
- not_done_yet = 1;
- continue;
- }
-
- if (device_already_active(fs->device)) {
- pass_done = 0;
- continue;
- }
-
- fsck_device(fs );
- fs->flags |= FLAG_DONE;
-
- if (G.serialize
- || (G.num_running >= G.max_running)
- ) {
- pass_done = 0;
- break;
- }
- }
- if (bb_got_signal)
- break;
- if (G.verbose > 1)
- printf("--waiting-- (pass %d)\n", passno);
- status |= wait_many(pass_done ? FLAG_WAIT_ALL :
- FLAG_WAIT_ATLEAST_ONE);
- if (pass_done) {
- if (G.verbose > 1)
- puts("----------------------------------");
- passno++;
- } else
- not_done_yet = 1;
- }
- kill_all_if_got_signal();
- status |= wait_many(FLAG_WAIT_ATLEAST_ONE);
- return status;
- }
- static void compile_fs_type(char *fs_type)
- {
- char *s;
- int num = 2;
- smallint negate;
- s = fs_type;
- while ((s = strchr(s, ','))) {
- num++;
- s++;
- }
- G.fs_type_list = xzalloc(num * sizeof(G.fs_type_list[0]));
- G.fs_type_flag = xzalloc(num * sizeof(G.fs_type_flag[0]));
- G.fs_type_negated = -1;
- num = 0;
- s = fs_type;
- while (1) {
- char *comma;
- negate = 0;
- if (s[0] == 'n' && s[1] == 'o') {
- s += 2;
- negate = 1;
- } else if (s[0] == '!') {
- s++;
- negate = 1;
- }
- if (strcmp(s, "loop") == 0)
-
- goto loop_special_case;
- if (is_prefixed_with(s, "opts=")) {
- s += 5;
- loop_special_case:
- G.fs_type_flag[num] = negate ? FS_TYPE_FLAG_NEGOPT : FS_TYPE_FLAG_OPT;
- } else {
- if (G.fs_type_negated == -1)
- G.fs_type_negated = negate;
- if (G.fs_type_negated != negate)
- bb_error_msg_and_die(
- "either all or none of the filesystem types passed to -t must be prefixed "
- "with 'no' or '!'");
- }
- comma = strchrnul(s, ',');
- G.fs_type_list[num++] = xstrndup(s, comma-s);
- if (*comma == '\0')
- break;
- s = comma + 1;
- }
- }
- static char **new_args(void)
- {
- G.args = xrealloc_vector(G.args, 2, G.num_args);
- return &G.args[G.num_args++];
- }
- int fsck_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
- int fsck_main(int argc UNUSED_PARAM, char **argv)
- {
- int i, status;
-
- struct fs_info *fs;
- const char *fstab;
- char *tmp;
- char **devices;
- int num_devices;
- smallint opts_for_fsck;
- smallint doall;
- smallint notitle;
- INIT_G();
-
- signal_no_SA_RESTART_empty_mask(SIGINT, record_signo);
- signal_no_SA_RESTART_empty_mask(SIGTERM, record_signo);
- setbuf(stdout, NULL);
- opts_for_fsck = doall = notitle = 0;
- devices = NULL;
- num_devices = 0;
- new_args();
-
- while (*++argv) {
- int j;
- int optpos;
- char *options;
- char *arg = *argv;
-
- if ((arg[0] == '/' && !opts_for_fsck) || strchr(arg, '=')) {
- devices = xrealloc_vector(devices, 2, num_devices);
- devices[num_devices++] = arg;
- continue;
- }
- if (arg[0] != '-' || opts_for_fsck) {
- *new_args() = arg;
- continue;
- }
- if (LONE_CHAR(arg + 1, '-')) {
- opts_for_fsck = 1;
- continue;
- }
- optpos = 0;
- options = NULL;
- for (j = 1; arg[j]; j++) {
- switch (arg[j]) {
- case 'A':
- doall = 1;
- break;
- #if DO_PROGRESS_INDICATOR
- case 'C':
- progress = 1;
- if (arg[++j]) {
- progress_fd = xatoi_positive(&arg[j]);
- goto next_arg;
- }
-
- if (!*++argv)
- bb_show_usage();
- progress_fd = xatoi_positive(*argv);
- goto next_arg;
- #endif
- case 'V':
- G.verbose++;
- break;
- case 'N':
- G.noexecute = 1;
- break;
- case 'R':
- G.skip_root = 1;
- break;
- case 'T':
- notitle = 1;
- break;
- case 'P':
- G.parallel_root = 1;
- break;
- case 's':
- G.serialize = 1;
- break;
- case 't':
- if (G.fstype)
- bb_show_usage();
- if (arg[++j])
- tmp = &arg[j];
- else if (*++argv)
- tmp = *argv;
- else
- bb_show_usage();
- G.fstype = xstrdup(tmp);
- compile_fs_type(G.fstype);
- goto next_arg;
- case '?':
- bb_show_usage();
- break;
- default:
- optpos++;
-
- options = xrealloc(options, optpos + 2);
- options[optpos] = arg[j];
- break;
- }
- }
- next_arg:
- if (optpos) {
- options[0] = '-';
- options[optpos + 1] = '\0';
- *new_args() = options;
- }
- }
- if (getenv("FSCK_FORCE_ALL_PARALLEL"))
- G.force_all_parallel = 1;
- tmp = getenv("FSCK_MAX_INST");
- G.max_running = INT_MAX;
- if (tmp)
- G.max_running = xatoi(tmp);
- new_args();
- new_args();
- if (!notitle)
- puts("fsck (busybox "BB_VER")");
-
- fstab = getenv("FSTAB_FILE");
- if (!fstab)
- fstab = "/etc/fstab";
- load_fs_info(fstab);
-
- if (num_devices == 0)
- G.serialize = doall = 1;
- if (doall)
- return check_all();
- status = 0;
- for (i = 0; i < num_devices; i++) {
- if (bb_got_signal) {
- kill_all_if_got_signal();
- break;
- }
- fs = lookup(devices[i]);
- if (!fs)
- fs = create_fs_device(devices[i], "", "auto", NULL, -1);
- fsck_device(fs );
- if (G.serialize
- || (G.num_running >= G.max_running)
- ) {
- int exit_status = wait_one(0);
- if (exit_status >= 0)
- status |= exit_status;
- if (G.verbose > 1)
- puts("----------------------------------");
- }
- }
- status |= wait_many(FLAG_WAIT_ALL);
- return status;
- }
|