xref: /freebsd/sbin/dump/main.c (revision 282a3889ebf826db9839be296ff1dd903f6d6d6e)
1 /*-
2  * Copyright (c) 1980, 1991, 1993, 1994
3  *	The Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 4. Neither the name of the University nor the names of its contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  */
29 
30 #ifndef lint
31 static const char copyright[] =
32 "@(#) Copyright (c) 1980, 1991, 1993, 1994\n\
33 	The Regents of the University of California.  All rights reserved.\n";
34 #endif /* not lint */
35 
36 #ifndef lint
37 #if 0
38 static char sccsid[] = "@(#)main.c	8.6 (Berkeley) 5/1/95";
39 #endif
40 static const char rcsid[] =
41   "$FreeBSD$";
42 #endif /* not lint */
43 
44 #include <sys/param.h>
45 #include <sys/stat.h>
46 #include <sys/time.h>
47 #include <sys/mount.h>
48 #include <sys/disklabel.h>
49 
50 #include <ufs/ufs/dinode.h>
51 #include <ufs/ufs/ufsmount.h>
52 #include <ufs/ffs/fs.h>
53 
54 #include <protocols/dumprestore.h>
55 
56 #include <ctype.h>
57 #include <err.h>
58 #include <errno.h>
59 #include <fcntl.h>
60 #include <fstab.h>
61 #include <limits.h>
62 #include <signal.h>
63 #include <stdint.h>
64 #include <stdio.h>
65 #include <stdlib.h>
66 #include <string.h>
67 #include <timeconv.h>
68 #include <unistd.h>
69 
70 #include "dump.h"
71 #include "pathnames.h"
72 
73 int	notify = 0;	/* notify operator flag */
74 int	snapdump = 0;	/* dumping live filesystem, so use snapshot */
75 int	blockswritten = 0;	/* number of blocks written on current tape */
76 int	tapeno = 0;	/* current tape number */
77 int	density = 0;	/* density in bytes/0.1" " <- this is for hilit19 */
78 int	ntrec = NTREC;	/* # tape blocks in each tape record */
79 int	cartridge = 0;	/* Assume non-cartridge tape */
80 int	cachesize = 0;	/* block cache size (in bytes), defaults to 0 */
81 long	dev_bsize = 1;	/* recalculated below */
82 long	blocksperfile;	/* output blocks per file */
83 char	*host = NULL;	/* remote host (if any) */
84 
85 /*
86  * Possible superblock locations ordered from most to least likely.
87  */
88 static int sblock_try[] = SBLOCKSEARCH;
89 
90 static char *getmntpt(char *, int *);
91 static long numarg(const char *, long, long);
92 static void obsolete(int *, char **[]);
93 static void usage(void) __dead2;
94 
95 int
96 main(int argc, char *argv[])
97 {
98 	struct stat sb;
99 	ino_t ino;
100 	int dirty;
101 	union dinode *dp;
102 	struct fstab *dt;
103 	char *map, *mntpt;
104 	int ch, mode, mntflags;
105 	int i, anydirskipped, bflag = 0, Tflag = 0, honorlevel = 1;
106 	int just_estimate = 0;
107 	ino_t maxino;
108 	char *tmsg;
109 
110 	spcl.c_date = _time_to_time64(time(NULL));
111 
112 	tsize = 0;	/* Default later, based on 'c' option for cart tapes */
113 	dumpdates = _PATH_DUMPDATES;
114 	popenout = NULL;
115 	tape = NULL;
116 	temp = _PATH_DTMP;
117 	if (TP_BSIZE / DEV_BSIZE == 0 || TP_BSIZE % DEV_BSIZE != 0)
118 		quit("TP_BSIZE must be a multiple of DEV_BSIZE\n");
119 	level = '0';
120 
121 	if (argc < 2)
122 		usage();
123 
124 	obsolete(&argc, &argv);
125 	while ((ch = getopt(argc, argv,
126 	    "0123456789aB:b:C:cD:d:f:h:LnP:Ss:T:uWw")) != -1)
127 		switch (ch) {
128 		/* dump level */
129 		case '0': case '1': case '2': case '3': case '4':
130 		case '5': case '6': case '7': case '8': case '9':
131 			level = ch;
132 			break;
133 
134 		case 'a':		/* `auto-size', Write to EOM. */
135 			unlimited = 1;
136 			break;
137 
138 		case 'B':		/* blocks per output file */
139 			blocksperfile = numarg("number of blocks per file",
140 			    1L, 0L);
141 			break;
142 
143 		case 'b':		/* blocks per tape write */
144 			ntrec = numarg("number of blocks per write",
145 			    1L, 1000L);
146 			break;
147 
148 		case 'C':
149 			cachesize = numarg("cachesize", 0, 0) * 1024 * 1024;
150 			break;
151 
152 		case 'c':		/* Tape is cart. not 9-track */
153 			cartridge = 1;
154 			break;
155 
156 		case 'D':
157 			dumpdates = optarg;
158 			break;
159 
160 		case 'd':		/* density, in bits per inch */
161 			density = numarg("density", 10L, 327670L) / 10;
162 			if (density >= 625 && !bflag)
163 				ntrec = HIGHDENSITYTREC;
164 			break;
165 
166 		case 'f':		/* output file */
167 			if (popenout != NULL)
168 				errx(X_STARTUP, "You cannot use the P and f "
169 				    "flags together.\n");
170 			tape = optarg;
171 			break;
172 
173 		case 'h':
174 			honorlevel = numarg("honor level", 0L, 10L);
175 			break;
176 
177 		case 'L':
178 			snapdump = 1;
179 			break;
180 
181 		case 'n':		/* notify operators */
182 			notify = 1;
183 			break;
184 
185 		case 'P':
186 			if (tape != NULL)
187 				errx(X_STARTUP, "You cannot use the P and f "
188 				    "flags together.\n");
189 			popenout = optarg;
190 			break;
191 
192 		case 'S':               /* exit after estimating # of tapes */
193 			just_estimate = 1;
194 			break;
195 
196 		case 's':		/* tape size, feet */
197 			tsize = numarg("tape size", 1L, 0L) * 12 * 10;
198 			break;
199 
200 		case 'T':		/* time of last dump */
201 			spcl.c_ddate = unctime(optarg);
202 			if (spcl.c_ddate < 0) {
203 				(void)fprintf(stderr, "bad time \"%s\"\n",
204 				    optarg);
205 				exit(X_STARTUP);
206 			}
207 			Tflag = 1;
208 			lastlevel = '?';
209 			break;
210 
211 		case 'u':		/* update /etc/dumpdates */
212 			uflag = 1;
213 			break;
214 
215 		case 'W':		/* what to do */
216 		case 'w':
217 			lastdump(ch);
218 			exit(X_FINOK);	/* do nothing else */
219 
220 		default:
221 			usage();
222 		}
223 	argc -= optind;
224 	argv += optind;
225 
226 	if (argc < 1) {
227 		(void)fprintf(stderr, "Must specify disk or file system\n");
228 		exit(X_STARTUP);
229 	}
230 	disk = *argv++;
231 	argc--;
232 	if (argc >= 1) {
233 		(void)fprintf(stderr, "Unknown arguments to dump:");
234 		while (argc--)
235 			(void)fprintf(stderr, " %s", *argv++);
236 		(void)fprintf(stderr, "\n");
237 		exit(X_STARTUP);
238 	}
239 	if (Tflag && uflag) {
240 	        (void)fprintf(stderr,
241 		    "You cannot use the T and u flags together.\n");
242 		exit(X_STARTUP);
243 	}
244 	if (popenout) {
245 		tape = "child pipeline process";
246 	} else if (tape == NULL && (tape = getenv("TAPE")) == NULL)
247 		tape = _PATH_DEFTAPE;
248 	if (strcmp(tape, "-") == 0) {
249 		pipeout++;
250 		tape = "standard output";
251 	}
252 
253 	if (blocksperfile)
254 		blocksperfile = blocksperfile / ntrec * ntrec; /* round down */
255 	else if (!unlimited) {
256 		/*
257 		 * Determine how to default tape size and density
258 		 *
259 		 *         	density				tape size
260 		 * 9-track	1600 bpi (160 bytes/.1")	2300 ft.
261 		 * 9-track	6250 bpi (625 bytes/.1")	2300 ft.
262 		 * cartridge	8000 bpi (100 bytes/.1")	1700 ft.
263 		 *						(450*4 - slop)
264 		 * hilit19 hits again: "
265 		 */
266 		if (density == 0)
267 			density = cartridge ? 100 : 160;
268 		if (tsize == 0)
269 			tsize = cartridge ? 1700L*120L : 2300L*120L;
270 	}
271 
272 	if (strchr(tape, ':')) {
273 		host = tape;
274 		tape = strchr(host, ':');
275 		*tape++ = '\0';
276 #ifdef RDUMP
277 		if (index(tape, '\n')) {
278 		    (void)fprintf(stderr, "invalid characters in tape\n");
279 		    exit(X_STARTUP);
280 		}
281 		if (rmthost(host) == 0)
282 			exit(X_STARTUP);
283 #else
284 		(void)fprintf(stderr, "remote dump not enabled\n");
285 		exit(X_STARTUP);
286 #endif
287 	}
288 	(void)setuid(getuid()); /* rmthost() is the only reason to be setuid */
289 
290 	if (signal(SIGHUP, SIG_IGN) != SIG_IGN)
291 		signal(SIGHUP, sig);
292 	if (signal(SIGTRAP, SIG_IGN) != SIG_IGN)
293 		signal(SIGTRAP, sig);
294 	if (signal(SIGFPE, SIG_IGN) != SIG_IGN)
295 		signal(SIGFPE, sig);
296 	if (signal(SIGBUS, SIG_IGN) != SIG_IGN)
297 		signal(SIGBUS, sig);
298 	if (signal(SIGSEGV, SIG_IGN) != SIG_IGN)
299 		signal(SIGSEGV, sig);
300 	if (signal(SIGTERM, SIG_IGN) != SIG_IGN)
301 		signal(SIGTERM, sig);
302 	if (signal(SIGINT, interrupt) == SIG_IGN)
303 		signal(SIGINT, SIG_IGN);
304 
305 	dump_getfstab();	/* /etc/fstab snarfed */
306 	/*
307 	 *	disk can be either the full special file name,
308 	 *	the suffix of the special file name,
309 	 *	the special name missing the leading '/',
310 	 *	the file system name with or without the leading '/'.
311 	 */
312 	dt = fstabsearch(disk);
313 	if (dt != NULL) {
314 		disk = rawname(dt->fs_spec);
315  		if (disk == NULL)
316  			errx(X_STARTUP, "%s: unknown file system", dt->fs_spec);
317 		(void)strncpy(spcl.c_dev, dt->fs_spec, NAMELEN);
318 		(void)strncpy(spcl.c_filesys, dt->fs_file, NAMELEN);
319 	} else {
320 		(void)strncpy(spcl.c_dev, disk, NAMELEN);
321 		(void)strncpy(spcl.c_filesys, "an unlisted file system",
322 		    NAMELEN);
323 	}
324 	spcl.c_dev[NAMELEN-1]='\0';
325 	spcl.c_filesys[NAMELEN-1]='\0';
326 
327 	if ((mntpt = getmntpt(disk, &mntflags)) != 0) {
328 		if (mntflags & MNT_RDONLY) {
329 			if (snapdump != 0) {
330 				msg("WARNING: %s\n",
331 				    "-L ignored for read-only filesystem.");
332 				snapdump = 0;
333 			}
334 		} else if (snapdump == 0) {
335 			msg("WARNING: %s\n",
336 			    "should use -L when dumping live read-write "
337 			    "filesystems!");
338 		} else {
339 			char snapname[BUFSIZ], snapcmd[BUFSIZ];
340 
341 			snprintf(snapname, sizeof snapname, "%s/.snap", mntpt);
342 			if ((stat(snapname, &sb) < 0) || !S_ISDIR(sb.st_mode)) {
343 				msg("WARNING: %s %s\n",
344 				    "-L requested but snapshot location",
345 				    snapname);
346 				msg("         %s: %s\n",
347 				    "is not a directory",
348 				    "dump downgraded, -L ignored");
349 				snapdump = 0;
350 			} else {
351 				snprintf(snapname, sizeof snapname,
352 				    "%s/.snap/dump_snapshot", mntpt);
353 				snprintf(snapcmd, sizeof snapcmd, "%s %s %s",
354 				    _PATH_MKSNAP_FFS, mntpt, snapname);
355 				unlink(snapname);
356 				if (system(snapcmd) != 0)
357 					errx(X_STARTUP, "Cannot create %s: %s\n",
358 					    snapname, strerror(errno));
359 				if ((diskfd = open(snapname, O_RDONLY)) < 0) {
360 					unlink(snapname);
361 					errx(X_STARTUP, "Cannot open %s: %s\n",
362 					    snapname, strerror(errno));
363 				}
364 				unlink(snapname);
365 				if (fstat(diskfd, &sb) != 0)
366 					err(X_STARTUP, "%s: stat", snapname);
367 				spcl.c_date = _time_to_time64(sb.st_mtime);
368 			}
369 		}
370 	} else if (snapdump != 0) {
371 		msg("WARNING: Cannot use -L on an unmounted filesystem.\n");
372 		snapdump = 0;
373 	}
374 	if (snapdump == 0) {
375 		if ((diskfd = open(disk, O_RDONLY)) < 0)
376 			err(X_STARTUP, "Cannot open %s", disk);
377 		if (fstat(diskfd, &sb) != 0)
378 			err(X_STARTUP, "%s: stat", disk);
379 		if (S_ISDIR(sb.st_mode))
380 			errx(X_STARTUP, "%s: unknown file system", disk);
381 	}
382 
383 	(void)strcpy(spcl.c_label, "none");
384 	(void)gethostname(spcl.c_host, NAMELEN);
385 	spcl.c_level = level - '0';
386 	spcl.c_type = TS_TAPE;
387 
388 	if (spcl.c_date == 0) {
389 		tmsg = "the epoch\n";
390 	} else {
391 		time_t t = _time64_to_time(spcl.c_date);
392 		tmsg = ctime(&t);
393 	}
394 	msg("Date of this level %c dump: %s", level, tmsg);
395 
396 	if (!Tflag)
397 	        getdumptime();		/* /etc/dumpdates snarfed */
398 	if (spcl.c_ddate == 0) {
399 		tmsg = "the epoch\n";
400 	} else {
401 		time_t t = _time64_to_time(spcl.c_ddate);
402 		tmsg = ctime(&t);
403 	}
404  	msg("Date of last level %c dump: %s", lastlevel, tmsg);
405 
406 	msg("Dumping %s%s ", snapdump ? "snapshot of ": "", disk);
407 	if (dt != NULL)
408 		msgtail("(%s) ", dt->fs_file);
409 	if (host)
410 		msgtail("to %s on host %s\n", tape, host);
411 	else
412 		msgtail("to %s\n", tape);
413 
414 	sync();
415 	sblock = (struct fs *)sblock_buf;
416 	for (i = 0; sblock_try[i] != -1; i++) {
417 		sblock->fs_fsize = SBLOCKSIZE; /* needed in bread */
418 		bread(sblock_try[i] >> dev_bshift, (char *) sblock, SBLOCKSIZE);
419 		if ((sblock->fs_magic == FS_UFS1_MAGIC ||
420 		     (sblock->fs_magic == FS_UFS2_MAGIC &&
421 		      sblock->fs_sblockloc == sblock_try[i])) &&
422 		    sblock->fs_bsize <= MAXBSIZE &&
423 		    sblock->fs_bsize >= sizeof(struct fs))
424 			break;
425 	}
426 	if (sblock_try[i] == -1)
427 		quit("Cannot find file system superblock\n");
428 	dev_bsize = sblock->fs_fsize / fsbtodb(sblock, 1);
429 	dev_bshift = ffs(dev_bsize) - 1;
430 	if (dev_bsize != (1 << dev_bshift))
431 		quit("dev_bsize (%ld) is not a power of 2", dev_bsize);
432 	tp_bshift = ffs(TP_BSIZE) - 1;
433 	if (TP_BSIZE != (1 << tp_bshift))
434 		quit("TP_BSIZE (%d) is not a power of 2", TP_BSIZE);
435 	maxino = sblock->fs_ipg * sblock->fs_ncg;
436 	mapsize = roundup(howmany(maxino, CHAR_BIT), TP_BSIZE);
437 	usedinomap = (char *)calloc((unsigned) mapsize, sizeof(char));
438 	dumpdirmap = (char *)calloc((unsigned) mapsize, sizeof(char));
439 	dumpinomap = (char *)calloc((unsigned) mapsize, sizeof(char));
440 	tapesize = 3 * (howmany(mapsize * sizeof(char), TP_BSIZE) + 1);
441 
442 	nonodump = spcl.c_level < honorlevel;
443 
444 	passno = 1;
445 	setproctitle("%s: pass 1: regular files", disk);
446 	msg("mapping (Pass I) [regular files]\n");
447 	anydirskipped = mapfiles(maxino, &tapesize);
448 
449 	passno = 2;
450 	setproctitle("%s: pass 2: directories", disk);
451 	msg("mapping (Pass II) [directories]\n");
452 	while (anydirskipped) {
453 		anydirskipped = mapdirs(maxino, &tapesize);
454 	}
455 
456 	if (pipeout || unlimited) {
457 		tapesize += 10;	/* 10 trailer blocks */
458 		msg("estimated %ld tape blocks.\n", tapesize);
459 	} else {
460 		double fetapes;
461 
462 		if (blocksperfile)
463 			fetapes = (double) tapesize / blocksperfile;
464 		else if (cartridge) {
465 			/* Estimate number of tapes, assuming streaming stops at
466 			   the end of each block written, and not in mid-block.
467 			   Assume no erroneous blocks; this can be compensated
468 			   for with an artificially low tape size. */
469 			fetapes =
470 			(	  (double) tapesize	/* blocks */
471 				* TP_BSIZE	/* bytes/block */
472 				* (1.0/density)	/* 0.1" / byte " */
473 			  +
474 				  (double) tapesize	/* blocks */
475 				* (1.0/ntrec)	/* streaming-stops per block */
476 				* 15.48		/* 0.1" / streaming-stop " */
477 			) * (1.0 / tsize );	/* tape / 0.1" " */
478 		} else {
479 			/* Estimate number of tapes, for old fashioned 9-track
480 			   tape */
481 			int tenthsperirg = (density == 625) ? 3 : 7;
482 			fetapes =
483 			(	  (double) tapesize	/* blocks */
484 				* TP_BSIZE	/* bytes / block */
485 				* (1.0/density)	/* 0.1" / byte " */
486 			  +
487 				  (double) tapesize	/* blocks */
488 				* (1.0/ntrec)	/* IRG's / block */
489 				* tenthsperirg	/* 0.1" / IRG " */
490 			) * (1.0 / tsize );	/* tape / 0.1" " */
491 		}
492 		etapes = fetapes;		/* truncating assignment */
493 		etapes++;
494 		/* count the dumped inodes map on each additional tape */
495 		tapesize += (etapes - 1) *
496 			(howmany(mapsize * sizeof(char), TP_BSIZE) + 1);
497 		tapesize += etapes + 10;	/* headers + 10 trailer blks */
498 		msg("estimated %ld tape blocks on %3.2f tape(s).\n",
499 		    tapesize, fetapes);
500 	}
501 
502         /*
503          * If the user only wants an estimate of the number of
504          * tapes, exit now.
505          */
506         if (just_estimate)
507                 exit(0);
508 
509 	/*
510 	 * Allocate tape buffer.
511 	 */
512 	if (!alloctape())
513 		quit(
514 	"can't allocate tape buffers - try a smaller blocking factor.\n");
515 
516 	startnewtape(1);
517 	(void)time((time_t *)&(tstart_writing));
518 	dumpmap(usedinomap, TS_CLRI, maxino - 1);
519 
520 	passno = 3;
521 	setproctitle("%s: pass 3: directories", disk);
522 	msg("dumping (Pass III) [directories]\n");
523 	dirty = 0;		/* XXX just to get gcc to shut up */
524 	for (map = dumpdirmap, ino = 1; ino < maxino; ino++) {
525 		if (((ino - 1) % CHAR_BIT) == 0)	/* map is offset by 1 */
526 			dirty = *map++;
527 		else
528 			dirty >>= 1;
529 		if ((dirty & 1) == 0)
530 			continue;
531 		/*
532 		 * Skip directory inodes deleted and maybe reallocated
533 		 */
534 		dp = getino(ino, &mode);
535 		if (mode != IFDIR)
536 			continue;
537 		(void)dumpino(dp, ino);
538 	}
539 
540 	passno = 4;
541 	setproctitle("%s: pass 4: regular files", disk);
542 	msg("dumping (Pass IV) [regular files]\n");
543 	for (map = dumpinomap, ino = 1; ino < maxino; ino++) {
544 		if (((ino - 1) % CHAR_BIT) == 0)	/* map is offset by 1 */
545 			dirty = *map++;
546 		else
547 			dirty >>= 1;
548 		if ((dirty & 1) == 0)
549 			continue;
550 		/*
551 		 * Skip inodes deleted and reallocated as directories.
552 		 */
553 		dp = getino(ino, &mode);
554 		if (mode == IFDIR)
555 			continue;
556 		(void)dumpino(dp, ino);
557 	}
558 
559 	(void)time((time_t *)&(tend_writing));
560 	spcl.c_type = TS_END;
561 	for (i = 0; i < ntrec; i++)
562 		writeheader(maxino - 1);
563 	if (pipeout)
564 		msg("DUMP: %jd tape blocks\n", (intmax_t)spcl.c_tapea);
565 	else
566 		msg("DUMP: %jd tape blocks on %d volume%s\n",
567 		    (intmax_t)spcl.c_tapea, spcl.c_volume,
568 		    (spcl.c_volume == 1) ? "" : "s");
569 
570 	/* report dump performance, avoid division through zero */
571 	if (tend_writing - tstart_writing == 0)
572 		msg("finished in less than a second\n");
573 	else
574 		msg("finished in %jd seconds, throughput %jd KBytes/sec\n",
575 		    (intmax_t)tend_writing - tstart_writing,
576 		    (intmax_t)(spcl.c_tapea /
577 		    (tend_writing - tstart_writing)));
578 
579 	putdumptime();
580 	trewind();
581 	broadcast("DUMP IS DONE!\a\a\n");
582 	msg("DUMP IS DONE\n");
583 	Exit(X_FINOK);
584 	/* NOTREACHED */
585 }
586 
587 static void
588 usage(void)
589 {
590 	fprintf(stderr,
591 		"usage: dump [-0123456789acLnSu] [-B records] [-b blocksize] [-C cachesize]\n"
592 		"            [-D dumpdates] [-d density] [-f file | -P pipecommand] [-h level]\n"
593 		"            [-s feet] [-T date] filesystem\n"
594 		"       dump -W | -w\n");
595 	exit(X_STARTUP);
596 }
597 
598 /*
599  * Check to see if a disk is currently mounted.
600  */
601 static char *
602 getmntpt(char *name, int *mntflagsp)
603 {
604 	long mntsize, i;
605 	struct statfs *mntbuf;
606 
607 	mntsize = getmntinfo(&mntbuf, MNT_NOWAIT);
608 	for (i = 0; i < mntsize; i++) {
609 		if (!strcmp(mntbuf[i].f_mntfromname, name)) {
610 			*mntflagsp = mntbuf[i].f_flags;
611 			return (mntbuf[i].f_mntonname);
612 		}
613 	}
614 	return (0);
615 }
616 
617 /*
618  * Pick up a numeric argument.  It must be nonnegative and in the given
619  * range (except that a vmax of 0 means unlimited).
620  */
621 static long
622 numarg(const char *meaning, long vmin, long vmax)
623 {
624 	char *p;
625 	long val;
626 
627 	val = strtol(optarg, &p, 10);
628 	if (*p)
629 		errx(1, "illegal %s -- %s", meaning, optarg);
630 	if (val < vmin || (vmax && val > vmax))
631 		errx(1, "%s must be between %ld and %ld", meaning, vmin, vmax);
632 	return (val);
633 }
634 
635 void
636 sig(int signo)
637 {
638 	switch(signo) {
639 	case SIGALRM:
640 	case SIGBUS:
641 	case SIGFPE:
642 	case SIGHUP:
643 	case SIGTERM:
644 	case SIGTRAP:
645 		if (pipeout)
646 			quit("Signal on pipe: cannot recover\n");
647 		msg("Rewriting attempted as response to unknown signal.\n");
648 		(void)fflush(stderr);
649 		(void)fflush(stdout);
650 		close_rewind();
651 		exit(X_REWRITE);
652 		/* NOTREACHED */
653 	case SIGSEGV:
654 		msg("SIGSEGV: ABORTING!\n");
655 		(void)signal(SIGSEGV, SIG_DFL);
656 		(void)kill(0, SIGSEGV);
657 		/* NOTREACHED */
658 	}
659 }
660 
661 char *
662 rawname(char *cp)
663 {
664 	struct stat sb;
665 
666 	/*
667 	 * Ensure that the device passed in is a raw device.
668 	 */
669 	if (stat(cp, &sb) == 0 && (sb.st_mode & S_IFMT) == S_IFCHR)
670 		return (cp);
671 
672 	/*
673 	 * Since there's only one device type now, we can't construct any
674 	 * better name, so we have to return NULL.
675 	 */
676 	return (NULL);
677 }
678 
679 /*
680  * obsolete --
681  *	Change set of key letters and ordered arguments into something
682  *	getopt(3) will like.
683  */
684 static void
685 obsolete(int *argcp, char **argvp[])
686 {
687 	int argc, flags;
688 	char *ap, **argv, *flagsp, **nargv, *p;
689 
690 	/* Setup. */
691 	argv = *argvp;
692 	argc = *argcp;
693 
694 	/*
695 	 * Return if no arguments or first argument has leading
696 	 * dash or slash.
697 	 */
698 	ap = argv[1];
699 	if (argc == 1 || *ap == '-' || *ap == '/')
700 		return;
701 
702 	/* Allocate space for new arguments. */
703 	if ((*argvp = nargv = malloc((argc + 1) * sizeof(char *))) == NULL ||
704 	    (p = flagsp = malloc(strlen(ap) + 2)) == NULL)
705 		err(1, NULL);
706 
707 	*nargv++ = *argv;
708 	argv += 2;
709 
710 	for (flags = 0; *ap; ++ap) {
711 		switch (*ap) {
712 		case 'B':
713 		case 'b':
714 		case 'd':
715 		case 'f':
716 		case 'D':
717 		case 'C':
718 		case 'h':
719 		case 's':
720 		case 'T':
721 			if (*argv == NULL) {
722 				warnx("option requires an argument -- %c", *ap);
723 				usage();
724 			}
725 			if ((nargv[0] = malloc(strlen(*argv) + 2 + 1)) == NULL)
726 				err(1, NULL);
727 			nargv[0][0] = '-';
728 			nargv[0][1] = *ap;
729 			(void)strcpy(&nargv[0][2], *argv);
730 			++argv;
731 			++nargv;
732 			break;
733 		default:
734 			if (!flags) {
735 				*p++ = '-';
736 				flags = 1;
737 			}
738 			*p++ = *ap;
739 			break;
740 		}
741 	}
742 
743 	/* Terminate flags. */
744 	if (flags) {
745 		*p = '\0';
746 		*nargv++ = flagsp;
747 	}
748 
749 	/* Copy remaining arguments. */
750 	while ((*nargv++ = *argv++));
751 
752 	/* Update argument count. */
753 	*argcp = nargv - *argvp - 1;
754 }
755