1 /* $OpenBSD: fsck.c,v 1.41 2021/07/12 15:09:18 beck Exp $ */
2 /* $NetBSD: fsck.c,v 1.7 1996/10/03 20:06:30 christos Exp $ */
3
4 /*
5 * Copyright (c) 1996 Christos Zoulas. All rights reserved.
6 * Copyright (c) 1980, 1989, 1993, 1994
7 * The Regents of the University of California. All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in the
16 * documentation and/or other materials provided with the distribution.
17 * 3. Neither the name of the University nor the names of its contributors
18 * may be used to endorse or promote products derived from this software
19 * without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 *
33 * From: @(#)mount.c 8.19 (Berkeley) 4/19/94
34 * From: NetBSD: mount.c,v 1.24 1995/11/18 03:34:29 cgd Exp
35 *
36 */
37
38 #include <sys/types.h>
39 #include <sys/mount.h>
40 #include <sys/queue.h>
41 #include <sys/resource.h>
42 #include <sys/stat.h>
43 #include <sys/wait.h>
44
45 #include <err.h>
46 #include <errno.h>
47 #include <fstab.h>
48 #include <signal.h>
49 #include <stdio.h>
50 #include <stdlib.h>
51 #include <string.h>
52 #include <unistd.h>
53 #include <limits.h>
54 #include <util.h>
55
56 #include "pathnames.h"
57 #include "fsutil.h"
58
59 static enum { IN_LIST, NOT_IN_LIST } which = NOT_IN_LIST;
60 static enum { NONET_FILTER, NET_FILTER } filter = NONET_FILTER;
61
TAILQ_HEAD(fstypelist,entry)62 TAILQ_HEAD(fstypelist, entry) opthead, selhead;
63
64 struct entry {
65 char *type;
66 char *options;
67 TAILQ_ENTRY(entry) entries;
68 };
69
70 static int maxrun;
71 static char *options;
72 static int flags;
73
74 int main(int, char *[]);
75
76 static int checkfs(const char *, const char *, const char *, void *, pid_t *);
77 static int selected(const char *);
78 static void addoption(char *);
79 static const char *getoptions(const char *);
80 static void addentry(struct fstypelist *, const char *, const char *);
81 static void maketypelist(char *);
82 static char *catopt(char *, const char *, int);
83 static void mangle(char *, int *, const char ***, int *);
84 static void usage(void);
85 static void *isok(struct fstab *);
86 static int hasopt(const char *, const char *);
87
88
89 int
main(int argc,char * argv[])90 main(int argc, char *argv[])
91 {
92 const char *errstr;
93 struct fstab *fs;
94 int i, rval = 0;
95 char *vfstype = NULL;
96 char *p, globopt[3];
97 struct rlimit rl;
98
99 /* Increase our data size to the max */
100 if (getrlimit(RLIMIT_DATA, &rl) == 0) {
101 if (geteuid() == 0)
102 rl.rlim_cur = rl.rlim_max = RLIM_INFINITY;
103 else
104 rl.rlim_cur = rl.rlim_max;
105 if (setrlimit(RLIMIT_DATA, &rl) == -1)
106 warn("Can't set resource limit to max data size");
107 } else
108 warn("Can't get resource limit for data size");
109
110 checkroot();
111
112 if (unveil("/dev", "rw") == -1)
113 err(1, "unveil /dev");
114 if (unveil(_PATH_FSTAB, "r") == -1)
115 err(1, "unveil %s", _PATH_FSTAB);
116 if (unveil("/sbin", "x") == -1)
117 err(1, "unveil /sbin");
118 if (pledge("stdio rpath wpath disklabel proc exec", NULL) == -1)
119 err(1, "pledge");
120
121 globopt[0] = '-';
122 globopt[2] = '\0';
123
124 TAILQ_INIT(&selhead);
125 TAILQ_INIT(&opthead);
126
127 while ((i = getopt(argc, argv, "b:dfl:nNpT:t:vy")) != -1)
128 switch (i) {
129 case 'd':
130 flags |= CHECK_DEBUG;
131 break;
132
133 case 'v':
134 flags |= CHECK_VERBOSE;
135 break;
136
137 case 'p':
138 flags |= CHECK_PREEN;
139 /*FALLTHROUGH*/
140 case 'n':
141 case 'f':
142 case 'y':
143 globopt[1] = i;
144 options = catopt(options, globopt, 1);
145 break;
146
147 case 'b':
148 if (asprintf(&p, "-b %s", optarg) == -1)
149 err(1, "malloc failed");
150 options = catopt(options, p, 1);
151 free(p);
152 break;
153
154 case 'l':
155 maxrun = strtonum(optarg, 0, INT_MAX, &errstr);
156 if (errstr)
157 errx(1, "-l %s: %s", optarg, errstr);
158
159 break;
160
161 case 'T':
162 if (*optarg)
163 addoption(optarg);
164 break;
165
166 case 't':
167 if (!TAILQ_EMPTY(&selhead))
168 errx(1, "only one -t option may be specified.");
169
170 maketypelist(optarg);
171 vfstype = optarg;
172 break;
173
174 case 'N':
175 filter = NET_FILTER;
176 break;
177
178 case '?':
179 default:
180 usage();
181 /* NOTREACHED */
182 }
183
184 argc -= optind;
185 argv += optind;
186
187 if (argc == 0)
188 return checkfstab(flags, maxrun, isok, checkfs);
189
190 #define BADTYPE(type) \
191 (strcmp(type, FSTAB_RO) && \
192 strcmp(type, FSTAB_RW) && strcmp(type, FSTAB_RQ))
193
194
195 for (; argc--; argv++) {
196 char *spec, *type;
197
198 if ((strncmp(*argv, "/dev/", 5) == 0 || isduid(*argv, 0)) &&
199 (type = readlabelfs(*argv, 0))) {
200 spec = *argv;
201 } else if ((fs = getfsfile(*argv)) == NULL &&
202 (fs = getfsspec(*argv)) == NULL) {
203 if (vfstype == NULL)
204 errx(1,
205 "%s: unknown special file or file system.",
206 *argv);
207 spec = *argv;
208 type = vfstype;
209 } else {
210 spec = fs->fs_spec;
211 type = fs->fs_vfstype;
212 if (BADTYPE(fs->fs_type))
213 errx(1, "%s has unknown file system type.",
214 *argv);
215 }
216
217 rval |= checkfs(type, blockcheck(spec), *argv, NULL, NULL);
218 }
219
220 return rval;
221 }
222
223
224 static void *
isok(struct fstab * fs)225 isok(struct fstab *fs)
226 {
227 if (fs->fs_passno == 0)
228 return NULL;
229
230 if (BADTYPE(fs->fs_type))
231 return NULL;
232
233 switch (filter) {
234 case NET_FILTER:
235 if (!hasopt(fs->fs_mntops, "net"))
236 return NULL;
237 break;
238 case NONET_FILTER:
239 if (hasopt(fs->fs_mntops, "net"))
240 return NULL;
241 break;
242 }
243 if (!selected(fs->fs_vfstype))
244 return NULL;
245
246 return fs;
247 }
248
249
250 static int
checkfs(const char * vfstype,const char * spec,const char * mntpt,void * auxarg,pid_t * pidp)251 checkfs(const char *vfstype, const char *spec, const char *mntpt, void *auxarg,
252 pid_t *pidp)
253 {
254 /* List of directories containing fsck_xxx subcommands. */
255 static const char *edirs[] = {
256 _PATH_SBIN,
257 _PATH_USRSBIN,
258 NULL
259 };
260 const char **argv, **edir;
261 pid_t pid;
262 int argc, i, status, maxargc;
263 char *optbuf = NULL, fsname[PATH_MAX], execname[PATH_MAX];
264 const char *extra = getoptions(vfstype);
265
266 if (strcmp(vfstype, "ufs") == 0)
267 vfstype = MOUNT_UFS;
268
269 maxargc = 100;
270 argv = ereallocarray(NULL, maxargc, sizeof(char *));
271
272 argc = 0;
273 (void)snprintf(fsname, sizeof(fsname), "fsck_%s", vfstype);
274 argv[argc++] = fsname;
275
276 if (options) {
277 if (extra != NULL)
278 optbuf = catopt(options, extra, 0);
279 else
280 optbuf = estrdup(options);
281 }
282 else if (extra)
283 optbuf = estrdup(extra);
284
285 if (optbuf)
286 mangle(optbuf, &argc, &argv, &maxargc);
287
288 argv[argc++] = spec;
289 argv[argc] = NULL;
290
291 if (flags & (CHECK_DEBUG|CHECK_VERBOSE)) {
292 (void)printf("start %s %swait %s", mntpt,
293 pidp ? "no" : "", fsname);
294 for (i = 1; i < argc; i++)
295 (void)printf(" %s", argv[i]);
296 (void)printf("\n");
297 }
298
299 switch (pid = fork()) {
300 case -1: /* Error. */
301 warn("fork");
302 free(optbuf);
303 free(argv);
304 return (1);
305
306 case 0: /* Child. */
307 if (flags & CHECK_DEBUG)
308 _exit(0);
309
310 /* Go find an executable. */
311 edir = edirs;
312 do {
313 (void)snprintf(execname,
314 sizeof(execname), "%s/fsck_%s", *edir, vfstype);
315 execv(execname, (char * const *)argv);
316 if (errno != ENOENT) {
317 if (spec)
318 warn("exec %s for %s", execname, spec);
319 else
320 warn("exec %s", execname);
321 }
322 } while (*++edir != NULL);
323
324 if (errno == ENOENT) {
325 if (spec)
326 warn("exec %s for %s", execname, spec);
327 else
328 warn("exec %s", execname);
329 }
330 exit(1);
331 /* NOTREACHED */
332
333 default: /* Parent. */
334 free(optbuf);
335 free(argv);
336
337 if (pidp) {
338 *pidp = pid;
339 return 0;
340 }
341
342 if (waitpid(pid, &status, 0) == -1) {
343 warn("waitpid");
344 return (1);
345 }
346
347 if (WIFEXITED(status)) {
348 if (WEXITSTATUS(status) != 0)
349 return (WEXITSTATUS(status));
350 }
351 else if (WIFSIGNALED(status)) {
352 warnx("%s: %s", spec, strsignal(WTERMSIG(status)));
353 return (1);
354 }
355 break;
356 }
357
358 return (0);
359 }
360
361
362 static int
selected(const char * type)363 selected(const char *type)
364 {
365 struct entry *e;
366
367 /* If no type specified, it's always selected. */
368 TAILQ_FOREACH(e, &selhead, entries)
369 if (!strncmp(e->type, type, MFSNAMELEN))
370 return which == IN_LIST ? 1 : 0;
371
372 return which == IN_LIST ? 0 : 1;
373 }
374
375
376 static const char *
getoptions(const char * type)377 getoptions(const char *type)
378 {
379 struct entry *e;
380
381 TAILQ_FOREACH(e, &opthead, entries)
382 if (!strncmp(e->type, type, MFSNAMELEN))
383 return e->options;
384 return "";
385 }
386
387
388 static void
addoption(char * optstr)389 addoption(char *optstr)
390 {
391 char *newoptions;
392 struct entry *e;
393
394 if ((newoptions = strchr(optstr, ':')) == NULL)
395 errx(1, "Invalid option string");
396
397 *newoptions++ = '\0';
398
399 TAILQ_FOREACH(e, &opthead, entries)
400 if (!strncmp(e->type, optstr, MFSNAMELEN)) {
401 e->options = catopt(e->options, newoptions, 1);
402 return;
403 }
404 addentry(&opthead, optstr, newoptions);
405 }
406
407
408 static void
addentry(struct fstypelist * list,const char * type,const char * opts)409 addentry(struct fstypelist *list, const char *type, const char *opts)
410 {
411 struct entry *e;
412
413 e = emalloc(sizeof(struct entry));
414 e->type = estrdup(type);
415 e->options = estrdup(opts);
416 TAILQ_INSERT_TAIL(list, e, entries);
417 }
418
419
420 static void
maketypelist(char * fslist)421 maketypelist(char *fslist)
422 {
423 char *ptr;
424
425 if ((fslist == NULL) || (fslist[0] == '\0'))
426 errx(1, "empty type list");
427
428 if (fslist[0] == 'n' && fslist[1] == 'o') {
429 fslist += 2;
430 which = NOT_IN_LIST;
431 }
432 else
433 which = IN_LIST;
434
435 while ((ptr = strsep(&fslist, ",")) != NULL)
436 addentry(&selhead, ptr, "");
437
438 }
439
440
441 static char *
catopt(char * s0,const char * s1,int fr)442 catopt(char *s0, const char *s1, int fr)
443 {
444 char *cp;
445
446 if (s0 && *s0) {
447 if (asprintf(&cp, "%s,%s", s0, s1) == -1)
448 err(1, "malloc failed");
449 } else
450 cp = estrdup(s1);
451
452 if (fr)
453 free(s0);
454 return (cp);
455 }
456
457
458 static void
mangle(char * opts,int * argcp,const char *** argvp,int * maxargcp)459 mangle(char *opts, int *argcp, const char ***argvp, int *maxargcp)
460 {
461 char *p, *s;
462 int argc = *argcp, maxargc = *maxargcp;
463 const char **argv = *argvp;
464
465 for (s = opts; (p = strsep(&s, ",")) != NULL;) {
466 /* always leave space for one more argument and the NULL */
467 if (argc >= maxargc - 3) {
468 int newmaxargc = maxargc + 50;
469
470 argv = ereallocarray(argv, newmaxargc, sizeof(char *));
471 maxargc = newmaxargc;
472 }
473 if (*p != '\0') {
474 if (*p == '-') {
475 argv[argc++] = p;
476 p = strchr(p, '=');
477 if (p) {
478 *p = '\0';
479 argv[argc++] = p+1;
480 }
481 }
482 else {
483 argv[argc++] = "-o";
484 argv[argc++] = p;
485 }
486 }
487 }
488
489 *argcp = argc;
490 *argvp = argv;
491 *maxargcp = maxargc;
492 }
493
494 static int
hasopt(const char * mntopts,const char * option)495 hasopt(const char *mntopts, const char *option)
496 {
497 int found;
498 char *opt, *optbuf;
499
500 if (mntopts == NULL)
501 return (0);
502 optbuf = strdup(mntopts);
503 found = 0;
504 for (opt = optbuf; !found && opt != NULL; strsep(&opt, ","))
505 found = !strncmp(opt, option, strlen(option));
506 free(optbuf);
507 return (found);
508 }
509
510
511 static void
usage(void)512 usage(void)
513 {
514 extern char *__progname;
515
516 fprintf(stderr, "usage: %s "
517 "[-dfNnpvy] [-b block#] [-l maxparallel] [-T fstype:fsoptions]\n"
518 " [-t fstype] [special | node ...]\n", __progname);
519 exit(1);
520 }
521