1/* $NetBSD: main.c,v 1.65 2010/02/26 02:11:40 christos Exp $ */ 2 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#include <sys/cdefs.h> 33#ifndef lint 34__COPYRIGHT("@(#) Copyright (c) 1980, 1991, 1993, 1994\ 35 The Regents of the University of California. All rights reserved."); 36#endif /* not lint */ 37 38#ifndef lint 39#if 0 40static char sccsid[] = "@(#)main.c 8.6 (Berkeley) 5/1/95"; 41#else 42__RCSID("$NetBSD: main.c,v 1.65 2010/02/26 02:11:40 christos Exp $"); 43#endif 44#endif /* not lint */ 45 46#include <sys/param.h> 47#include <sys/time.h> 48#include <sys/stat.h> 49#include <sys/mount.h> 50 51#include <ufs/ufs/dinode.h> 52#include <ufs/ffs/fs.h> 53#include <ufs/ffs/ffs_extern.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 <signal.h> 63#include <stdio.h> 64#include <stdlib.h> 65#include <string.h> 66#include <time.h> 67#include <unistd.h> 68 69#include "dump.h" 70#include "pathnames.h" 71#include "snapshot.h" 72 73int timestamp; /* print message timestamps */ 74int notify; /* notify operator flag */ 75int blockswritten; /* number of blocks written on current tape */ 76int tapeno; /* current tape number */ 77int density; /* density in bytes/0.1" */ 78int ntrec = NTREC; /* # tape blocks in each tape record */ 79int cartridge; /* Assume non-cartridge tape */ 80long dev_bsize = 1; /* recalculated below */ 81long blocksperfile; /* output blocks per file */ 82const char *host; /* remote host (if any) */ 83int readcache = -1; /* read cache size (in readblksize blks) */ 84int readblksize = 32 * 1024; /* read block size */ 85char default_time_string[] = "%T %Z"; /* default timestamp string */ 86char *time_string = default_time_string; /* timestamp string */ 87 88int main(int, char *[]); 89static long numarg(const char *, long, long); 90static void obsolete(int *, char **[]); 91static void usage(void); 92 93int 94main(int argc, char *argv[]) 95{ 96 ino_t ino; 97 int dirty; 98 union dinode *dp; 99 struct fstab *dt; 100 struct statvfs *mntinfo, fsbuf; 101 char *map, *cp; 102 int ch; 103 int i, anydirskipped, bflag = 0, Tflag = 0, Fflag = 0, honorlevel = 1; 104 int snap_internal = 0; 105 ino_t maxino; 106 time_t tnow, date; 107 int dirc; 108 char *mountpoint; 109 int just_estimate = 0; 110 char labelstr[LBLSIZE]; 111 char *new_time_format; 112 char *snap_backup = NULL; 113 114 spcl.c_date = 0; 115 (void)time(&tnow); 116 spcl.c_date = tnow; 117 tzset(); /* set up timezone for strftime */ 118 if ((new_time_format = getenv("TIMEFORMAT")) != NULL) 119 time_string = new_time_format; 120 121 tsize = 0; /* Default later, based on 'c' option for cart tapes */ 122 if ((tape = getenv("TAPE")) == NULL) 123 tape = _PATH_DEFTAPE; 124 dumpdates = _PATH_DUMPDATES; 125 temp = _PATH_DTMP; 126 strcpy(labelstr, "none"); /* XXX safe strcpy. */ 127 if (TP_BSIZE / DEV_BSIZE == 0 || TP_BSIZE % DEV_BSIZE != 0) 128 quit("TP_BSIZE must be a multiple of DEV_BSIZE\n"); 129 level = '0'; 130 timestamp = 0; 131 132 if (argc < 2) 133 usage(); 134 135 obsolete(&argc, &argv); 136 while ((ch = getopt(argc, argv, 137 "0123456789aB:b:cd:eFf:h:ik:l:L:nr:s:StT:uWwx:X")) != -1) 138 switch (ch) { 139 /* dump level */ 140 case '0': case '1': case '2': case '3': case '4': 141 case '5': case '6': case '7': case '8': case '9': 142 level = ch; 143 break; 144 145 case 'a': /* `auto-size', Write to EOM. */ 146 unlimited = 1; 147 break; 148 149 case 'B': /* blocks per output file */ 150 blocksperfile = numarg("blocks per file", 1L, 0L); 151 break; 152 153 case 'b': /* blocks per tape write */ 154 ntrec = numarg("blocks per write", 1L, 1000L); 155 bflag = 1; 156 break; 157 158 case 'c': /* Tape is cart. not 9-track */ 159 cartridge = 1; 160 break; 161 162 case 'd': /* density, in bits per inch */ 163 density = numarg("density", 10L, 327670L) / 10; 164 if (density >= 625 && !bflag) 165 ntrec = HIGHDENSITYTREC; 166 break; 167 168 case 'e': /* eject full tapes */ 169 eflag = 1; 170 break; 171 172 case 'F': /* files-to-dump is an fs image */ 173 Fflag = 1; 174 break; 175 176 case 'f': /* output file */ 177 tape = optarg; 178 break; 179 180 case 'h': 181 honorlevel = numarg("honor level", 0L, 10L); 182 break; 183 184 case 'i': /* "true incremental" regardless level */ 185 level = 'i'; 186 trueinc = 1; 187 break; 188 189 case 'k': 190 readblksize = numarg("read block size", 0, 64) * 1024; 191 break; 192 193 case 'l': /* autoload after eject full tapes */ 194 eflag = 1; 195 lflag = numarg("timeout (in seconds)", 1, 0); 196 break; 197 198 case 'L': 199 /* 200 * Note that although there are LBLSIZE characters, 201 * the last must be '\0', so the limit on strlen() 202 * is really LBLSIZE-1. 203 */ 204 if (strlcpy(labelstr, optarg, sizeof(labelstr)) 205 >= sizeof(labelstr)) { 206 msg( 207 "WARNING Label `%s' is larger than limit of %lu characters.\n", 208 optarg, 209 (unsigned long)sizeof(labelstr) - 1); 210 msg("WARNING: Using truncated label `%s'.\n", 211 labelstr); 212 } 213 break; 214 case 'n': /* notify operators */ 215 notify = 1; 216 break; 217 218 case 'r': /* read cache size */ 219 readcache = numarg("read cache size", 0, 512); 220 break; 221 222 case 's': /* tape size, feet */ 223 tsize = numarg("tape size", 1L, 0L) * 12 * 10; 224 break; 225 226 case 'S': /* exit after estimating # of tapes */ 227 just_estimate = 1; 228 break; 229 230 case 't': 231 timestamp = 1; 232 break; 233 234 case 'T': /* time of last dump */ 235 spcl.c_ddate = unctime(optarg); 236 if (spcl.c_ddate < 0) { 237 (void)fprintf(stderr, "bad time \"%s\"\n", 238 optarg); 239 exit(X_STARTUP); 240 } 241 Tflag = 1; 242 lastlevel = '?'; 243 break; 244 245 case 'u': /* update /etc/dumpdates */ 246 uflag = 1; 247 break; 248 249 case 'W': /* what to do */ 250 case 'w': 251 lastdump(ch); 252 exit(X_FINOK); /* do nothing else */ 253 254 case 'x': 255 snap_backup = optarg; 256 break; 257 258 case 'X': 259 snap_internal = 1; 260 break; 261 262 default: 263 usage(); 264 } 265 argc -= optind; 266 argv += optind; 267 268 if (argc < 1) { 269 (void)fprintf(stderr, 270 "Must specify disk or image, or file list\n"); 271 exit(X_STARTUP); 272 } 273 274 275 /* 276 * determine if disk is a subdirectory, and setup appropriately 277 */ 278 getfstab(); /* /etc/fstab snarfed */ 279 disk = NULL; 280 disk_dev = NULL; 281 mountpoint = NULL; 282 dirc = 0; 283 for (i = 0; i < argc; i++) { 284 struct stat sb; 285 286 if (lstat(argv[i], &sb) == -1) 287 quit("Cannot stat %s: %s\n", argv[i], strerror(errno)); 288 if (Fflag || S_ISCHR(sb.st_mode) || S_ISBLK(sb.st_mode)) { 289 disk = argv[i]; 290 multicheck: 291 if (dirc != 0) 292 quit( 293 "Can't dump a disk or image at the same time as a file list\n"); 294 break; 295 } 296 if ((dt = fstabsearch(argv[i])) != NULL) { 297 disk = dt->fs_spec; 298 mountpoint = xstrdup(dt->fs_file); 299 goto multicheck; 300 } 301 if (statvfs(argv[i], &fsbuf) == -1) 302 quit("Cannot statvfs %s: %s\n", argv[i], 303 strerror(errno)); 304 disk = fsbuf.f_mntfromname; 305 if (strcmp(argv[i], fsbuf.f_mntonname) == 0) 306 goto multicheck; 307 if (mountpoint == NULL) { 308 mountpoint = xstrdup(fsbuf.f_mntonname); 309 if (uflag) { 310 msg("Ignoring u flag for subdir dump\n"); 311 uflag = 0; 312 } 313 if (level > '0') { 314 msg("Subdir dump is done at level 0\n"); 315 level = '0'; 316 } 317 msg("Dumping sub files/directories from %s\n", 318 mountpoint); 319 } else { 320 if (strcmp(mountpoint, fsbuf.f_mntonname) != 0) 321 quit("%s is not on %s\n", argv[i], mountpoint); 322 } 323 msg("Dumping file/directory %s\n", argv[i]); 324 dirc++; 325 } 326 if (mountpoint) 327 free(mountpoint); 328 329 if (dirc == 0) { 330 argv++; 331 if (argc != 1) { 332 (void)fprintf(stderr, "Excess arguments to dump:"); 333 while (--argc) 334 (void)fprintf(stderr, " %s", *argv++); 335 (void)fprintf(stderr, "\n"); 336 exit(X_STARTUP); 337 } 338 } 339 if (Tflag && uflag) { 340 (void)fprintf(stderr, 341 "You cannot use the T and u flags together.\n"); 342 exit(X_STARTUP); 343 } 344 if (strcmp(tape, "-") == 0) { 345 pipeout++; 346 tape = "standard output"; 347 } 348 349 if (blocksperfile) 350 blocksperfile = blocksperfile / ntrec * ntrec; /* round down */ 351 else if (!unlimited) { 352 /* 353 * Determine how to default tape size and density 354 * 355 * density tape size 356 * 9-track 1600 bpi (160 bytes/.1") 2300 ft. 357 * 9-track 6250 bpi (625 bytes/.1") 2300 ft. 358 * cartridge 8000 bpi (100 bytes/.1") 1700 ft. 359 * (450*4 - slop) 360 */ 361 if (density == 0) 362 density = cartridge ? 100 : 160; 363 if (tsize == 0) 364 tsize = cartridge ? 1700L*120L : 2300L*120L; 365 } 366 367 if ((cp = strchr(tape, ':')) != NULL) { 368 host = tape; 369 /* This is fine, because all the const strings don't have : */ 370 *cp++ = '\0'; 371 tape = cp; 372#ifdef RDUMP 373 if (rmthost(host) == 0) 374 exit(X_STARTUP); 375#else 376 (void)fprintf(stderr, "remote dump not enabled\n"); 377 exit(X_STARTUP); 378#endif 379 } 380 381 if (signal(SIGHUP, SIG_IGN) != SIG_IGN) 382 signal(SIGHUP, sig); 383 if (signal(SIGTRAP, SIG_IGN) != SIG_IGN) 384 signal(SIGTRAP, sig); 385 if (signal(SIGFPE, SIG_IGN) != SIG_IGN) 386 signal(SIGFPE, sig); 387 if (signal(SIGBUS, SIG_IGN) != SIG_IGN) 388 signal(SIGBUS, sig); 389#if 0 390 if (signal(SIGSEGV, SIG_IGN) != SIG_IGN) 391 signal(SIGSEGV, sig); 392#endif 393 if (signal(SIGTERM, SIG_IGN) != SIG_IGN) 394 signal(SIGTERM, sig); 395 if (signal(SIGINT, interrupt) == SIG_IGN) 396 signal(SIGINT, SIG_IGN); 397 398 /* 399 * disk can be either the full special file name, or 400 * the file system name. 401 */ 402 mountpoint = NULL; 403 mntinfo = mntinfosearch(disk); 404 if ((dt = fstabsearch(disk)) != NULL) { 405 disk = rawname(dt->fs_spec); 406 mountpoint = dt->fs_file; 407 msg("Found %s on %s in %s\n", disk, mountpoint, _PATH_FSTAB); 408 } else if (mntinfo != NULL) { 409 disk = rawname(mntinfo->f_mntfromname); 410 mountpoint = mntinfo->f_mntonname; 411 msg("Found %s on %s in mount table\n", disk, mountpoint); 412 } 413 if (mountpoint != NULL) { 414 if (dirc != 0) 415 (void)snprintf(spcl.c_filesys, sizeof(spcl.c_filesys), 416 "a subset of %s", mountpoint); 417 else 418 (void)strlcpy(spcl.c_filesys, mountpoint, 419 sizeof(spcl.c_filesys)); 420 } else if (Fflag) { 421 (void)strlcpy(spcl.c_filesys, "a file system image", 422 sizeof(spcl.c_filesys)); 423 } else { 424 (void)strlcpy(spcl.c_filesys, "an unlisted file system", 425 sizeof(spcl.c_filesys)); 426 } 427 (void)strlcpy(spcl.c_dev, disk, sizeof(spcl.c_dev)); 428 (void)strlcpy(spcl.c_label, labelstr, sizeof(spcl.c_label)); 429 (void)gethostname(spcl.c_host, sizeof(spcl.c_host)); 430 spcl.c_host[sizeof(spcl.c_host) - 1] = '\0'; 431 432 if ((snap_backup != NULL || snap_internal) && mntinfo == NULL) { 433 msg("WARNING: Cannot use -x or -X on unmounted file system.\n"); 434 snap_backup = NULL; 435 snap_internal = 0; 436 } 437 438#ifdef DUMP_LFS 439 sync(); 440 if (snap_backup != NULL || snap_internal) { 441 if (lfs_wrap_stop(mountpoint) < 0) { 442 msg("Cannot stop writing on %s\n", mountpoint); 443 exit(X_STARTUP); 444 } 445 } 446 if ((diskfd = open(disk, O_RDONLY)) < 0) { 447 msg("Cannot open %s\n", disk); 448 exit(X_STARTUP); 449 } 450 disk_dev = disk; 451#else /* ! DUMP_LFS */ 452 if (snap_backup != NULL || snap_internal) { 453 diskfd = snap_open(mntinfo->f_mntonname, snap_backup, 454 &tnow, &disk_dev); 455 if (diskfd < 0) { 456 msg("Cannot open snapshot of %s\n", 457 mntinfo->f_mntonname); 458 exit(X_STARTUP); 459 } 460 spcl.c_date = tnow; 461 } else { 462 if ((diskfd = open(disk, O_RDONLY)) < 0) { 463 msg("Cannot open %s\n", disk); 464 exit(X_STARTUP); 465 } 466 disk_dev = disk; 467 } 468 sync(); 469#endif /* ! DUMP_LFS */ 470 471 needswap = fs_read_sblock(sblock_buf); 472 473 /* true incremental is always a level 10 dump */ 474 spcl.c_level = trueinc? iswap32(10): iswap32(level - '0'); 475 spcl.c_type = iswap32(TS_TAPE); 476 spcl.c_date = iswap32(spcl.c_date); 477 spcl.c_ddate = iswap32(spcl.c_ddate); 478 if (!Tflag) 479 getdumptime(); /* /etc/dumpdates snarfed */ 480 481 date = iswap32(spcl.c_date); 482 msg("Date of this level %c dump: %s", level, 483 spcl.c_date == 0 ? "the epoch\n" : ctime(&date)); 484 date = iswap32(spcl.c_ddate); 485 msg("Date of last level %c dump: %s", lastlevel, 486 spcl.c_ddate == 0 ? "the epoch\n" : ctime(&date)); 487 msg("Dumping "); 488 if (snap_backup != NULL || snap_internal) 489 msgtail("a snapshot of "); 490 if (dirc != 0) 491 msgtail("a subset of "); 492 msgtail("%s (%s) ", disk, spcl.c_filesys); 493 if (host) 494 msgtail("to %s on host %s\n", tape, host); 495 else 496 msgtail("to %s\n", tape); 497 msg("Label: %s\n", labelstr); 498 499 ufsib = fs_parametrize(); 500 501 dev_bshift = ffs(dev_bsize) - 1; 502 if (dev_bsize != (1 << dev_bshift)) 503 quit("dev_bsize (%ld) is not a power of 2", dev_bsize); 504 tp_bshift = ffs(TP_BSIZE) - 1; 505 if (TP_BSIZE != (1 << tp_bshift)) 506 quit("TP_BSIZE (%d) is not a power of 2", TP_BSIZE); 507 maxino = fs_maxino(); 508 mapsize = roundup(howmany(maxino, NBBY), TP_BSIZE); 509 usedinomap = (char *)xcalloc((unsigned) mapsize, sizeof(char)); 510 dumpdirmap = (char *)xcalloc((unsigned) mapsize, sizeof(char)); 511 dumpinomap = (char *)xcalloc((unsigned) mapsize, sizeof(char)); 512 tapesize = 3 * (howmany(mapsize * sizeof(char), TP_BSIZE) + 1); 513 514 nonodump = iswap32(spcl.c_level) < honorlevel; 515 516 initcache(readcache, readblksize); 517 518 (void)signal(SIGINFO, statussig); 519 520 msg("mapping (Pass I) [regular files]\n"); 521 anydirskipped = mapfiles(maxino, &tapesize, mountpoint, 522 (dirc ? argv : NULL)); 523 524 msg("mapping (Pass II) [directories]\n"); 525 while (anydirskipped) { 526 anydirskipped = mapdirs(maxino, &tapesize); 527 } 528 529 if (pipeout || unlimited) { 530 tapesize += 10; /* 10 trailer blocks */ 531 msg("estimated %llu tape blocks.\n", 532 (unsigned long long)tapesize); 533 } else { 534 double fetapes; 535 536 if (blocksperfile) 537 fetapes = (double) tapesize / blocksperfile; 538 else if (cartridge) { 539 /* Estimate number of tapes, assuming streaming stops at 540 the end of each block written, and not in mid-block. 541 Assume no erroneous blocks; this can be compensated 542 for with an artificially low tape size. */ 543 fetapes = 544 ( (double) tapesize /* blocks */ 545 * TP_BSIZE /* bytes/block */ 546 * (1.0/density) /* 0.1" / byte */ 547 + 548 (double) tapesize /* blocks */ 549 * (1.0/ntrec) /* streaming-stops per block */ 550 * 15.48 /* 0.1" / streaming-stop */ 551 ) * (1.0 / tsize ); /* tape / 0.1" */ 552 } else { 553 /* Estimate number of tapes, for old fashioned 9-track 554 tape */ 555 int tenthsperirg = (density == 625) ? 3 : 7; 556 fetapes = 557 ( tapesize /* blocks */ 558 * TP_BSIZE /* bytes / block */ 559 * (1.0/density) /* 0.1" / byte */ 560 + 561 tapesize /* blocks */ 562 * (1.0/ntrec) /* IRG's / block */ 563 * tenthsperirg /* 0.1" / IRG */ 564 ) * (1.0 / tsize ); /* tape / 0.1" */ 565 } 566 etapes = fetapes; /* truncating assignment */ 567 etapes++; 568 /* count the dumped inodes map on each additional tape */ 569 tapesize += (etapes - 1) * 570 (howmany(mapsize * sizeof(char), TP_BSIZE) + 1); 571 tapesize += etapes + 10; /* headers + 10 trailer blks */ 572 msg("estimated %llu tape blocks on %3.2f tape(s).\n", 573 (unsigned long long)tapesize, fetapes); 574 } 575 /* 576 * If the user only wants an estimate of the number of 577 * tapes, exit now. 578 */ 579 if (just_estimate) 580 exit(X_FINOK); 581 582 /* 583 * Allocate tape buffer. 584 */ 585 if (!alloctape()) 586 quit("can't allocate tape buffers - try a smaller blocking factor.\n"); 587 588 startnewtape(1); 589 (void)time((time_t *)&(tstart_writing)); 590 xferrate = 0; 591 dumpmap(usedinomap, TS_CLRI, maxino - 1); 592 593 msg("dumping (Pass III) [directories]\n"); 594 dirty = 0; /* XXX just to get gcc to shut up */ 595 for (map = dumpdirmap, ino = 1; ino < maxino; ino++) { 596 if (((ino - 1) % NBBY) == 0) /* map is offset by 1 */ 597 dirty = *map++; 598 else 599 dirty >>= 1; 600 if ((dirty & 1) == 0) 601 continue; 602 /* 603 * Skip directory inodes deleted and maybe reallocated 604 */ 605 dp = getino(ino); 606 if ((DIP(dp, mode) & IFMT) != IFDIR) 607 continue; 608 (void)dumpino(dp, ino); 609 } 610 611 msg("dumping (Pass IV) [regular files]\n"); 612 for (map = dumpinomap, ino = 1; ino < maxino; ino++) { 613 int mode; 614 615 if (((ino - 1) % NBBY) == 0) /* map is offset by 1 */ 616 dirty = *map++; 617 else 618 dirty >>= 1; 619 if ((dirty & 1) == 0) 620 continue; 621 /* 622 * Skip inodes deleted and reallocated as directories. 623 */ 624 dp = getino(ino); 625 mode = DIP(dp, mode) & IFMT; 626 if (mode == IFDIR) 627 continue; 628 (void)dumpino(dp, ino); 629 } 630 631 spcl.c_type = iswap32(TS_END); 632 for (i = 0; i < ntrec; i++) 633 writeheader(maxino - 1); 634 if (pipeout) 635 msg("%d tape blocks\n",iswap32(spcl.c_tapea)); 636 else 637 msg("%d tape blocks on %d volume%s\n", 638 iswap32(spcl.c_tapea), iswap32(spcl.c_volume), 639 (iswap32(spcl.c_volume) == 1) ? "" : "s"); 640 tnow = do_stats(); 641 date = iswap32(spcl.c_date); 642 msg("Date of this level %c dump: %s", level, 643 spcl.c_date == 0 ? "the epoch\n" : ctime(&date)); 644 msg("Date this dump completed: %s", ctime(&tnow)); 645 msg("Average transfer rate: %d KB/s\n", xferrate / tapeno); 646 putdumptime(); 647 trewind(0); 648 broadcast("DUMP IS DONE!\a\a\n"); 649#ifdef DUMP_LFS 650 lfs_wrap_go(); 651#endif /* DUMP_LFS */ 652 msg("DUMP IS DONE\n"); 653 Exit(X_FINOK); 654 /* NOTREACHED */ 655 exit(X_FINOK); /* XXX: to satisfy gcc */ 656} 657 658static void 659usage(void) 660{ 661 const char *prog = getprogname(); 662 663 (void)fprintf(stderr, 664"usage: %s [-0123456789aceFnStuX] [-B records] [-b blocksize]\n" 665" [-d density] [-f file] [-h level] [-k read-blocksize]\n" 666" [-L label] [-l timeout] [-r read-cache] [-s feet]\n" 667" [-T date] [-x snap-backup] files-to-dump\n" 668" %s [-W | -w]\n", prog, prog); 669 exit(X_STARTUP); 670} 671 672/* 673 * Pick up a numeric argument. It must be nonnegative and in the given 674 * range (except that a vmax of 0 means unlimited). 675 */ 676static long 677numarg(const char *meaning, long vmin, long vmax) 678{ 679 char *p; 680 long val; 681 682 val = strtol(optarg, &p, 10); 683 if (*p) 684 errx(X_STARTUP, "illegal %s -- %s", meaning, optarg); 685 if (val < vmin || (vmax && val > vmax)) 686 errx(X_STARTUP, "%s must be between %ld and %ld", 687 meaning, vmin, vmax); 688 return (val); 689} 690 691void 692sig(int signo) 693{ 694 695 switch(signo) { 696 case SIGALRM: 697 case SIGBUS: 698 case SIGFPE: 699 case SIGHUP: 700 case SIGTERM: 701 case SIGTRAP: 702 if (pipeout) 703 quit("Signal on pipe: cannot recover\n"); 704 msg("Rewriting attempted as response to signal %s.\n", sys_siglist[signo]); 705 (void)fflush(stderr); 706 (void)fflush(stdout); 707 close_rewind(); 708 exit(X_REWRITE); 709 /* NOTREACHED */ 710 case SIGSEGV: 711 msg("SIGSEGV: ABORTING!\n"); 712 (void)signal(SIGSEGV, SIG_DFL); 713 (void)kill(0, SIGSEGV); 714 /* NOTREACHED */ 715 } 716} 717 718char * 719rawname(char *cp) 720{ 721 static char rawbuf[MAXPATHLEN]; 722 char *dp = strrchr(cp, '/'); 723 724 if (dp == NULL) 725 return (NULL); 726 *dp = '\0'; 727 (void)snprintf(rawbuf, sizeof rawbuf, "%s/r%s", cp, dp + 1); 728 *dp = '/'; 729 return (rawbuf); 730} 731 732/* 733 * obsolete -- 734 * Change set of key letters and ordered arguments into something 735 * getopt(3) will like. 736 */ 737static void 738obsolete(int *argcp, char **argvp[]) 739{ 740 int argc, flags; 741 char *ap, **argv, *flagsp, **nargv, *p; 742 743 /* Setup. */ 744 argv = *argvp; 745 argc = *argcp; 746 747 /* Return if no arguments or first argument has leading dash. */ 748 ap = argv[1]; 749 if (argc == 1 || *ap == '-') 750 return; 751 752 /* Allocate space for new arguments. */ 753 *argvp = nargv = xmalloc((argc + 1) * sizeof(char *)); 754 p = flagsp = xmalloc(strlen(ap) + 2); 755 756 *nargv++ = *argv; 757 argv += 2; 758 759 for (flags = 0; *ap; ++ap) { 760 switch (*ap) { 761 case 'B': 762 case 'b': 763 case 'd': 764 case 'f': 765 case 'h': 766 case 's': 767 case 'T': 768 case 'x': 769 if (*argv == NULL) { 770 warnx("option requires an argument -- %c", *ap); 771 usage(); 772 } 773 nargv[0] = xmalloc(strlen(*argv) + 2 + 1); 774 nargv[0][0] = '-'; 775 nargv[0][1] = *ap; 776 (void)strcpy(&nargv[0][2], *argv); /* XXX safe strcpy */ 777 ++argv; 778 ++nargv; 779 break; 780 default: 781 if (!flags) { 782 *p++ = '-'; 783 flags = 1; 784 } 785 *p++ = *ap; 786 break; 787 } 788 } 789 790 /* Terminate flags. */ 791 if (flags) { 792 *p = '\0'; 793 *nargv++ = flagsp; 794 } else 795 free(flagsp); 796 797 /* Copy remaining arguments. */ 798 while ((*nargv++ = *argv++) != NULL) 799 ; 800 801 /* Update argument count. */ 802 *argcp = nargv - *argvp - 1; 803} 804 805 806void * 807xcalloc(size_t number, size_t size) 808{ 809 void *p; 810 811 p = calloc(number, size); 812 if (p == NULL) 813 quit("%s\n", strerror(errno)); 814 return (p); 815} 816 817void * 818xmalloc(size_t size) 819{ 820 void *p; 821 822 p = malloc(size); 823 if (p == NULL) 824 quit("%s\n", strerror(errno)); 825 return (p); 826} 827 828char * 829xstrdup(const char *str) 830{ 831 char *p; 832 833 p = strdup(str); 834 if (p == NULL) 835 quit("%s\n", strerror(errno)); 836 return (p); 837} 838