]> CyberLeo.Net >> Repos - FreeBSD/releng/7.2.git/blob - sbin/dump/main.c
Create releng/7.2 from stable/7 in preparation for 7.2-RELEASE.
[FreeBSD/releng/7.2.git] / sbin / dump / main.c
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 }