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