1 /*- 2 * Copyright (c) 1992 Keith Muller. 3 * Copyright (c) 1992, 1993 4 * The Regents of the University of California. All rights reserved. 5 * 6 * This code is derived from software contributed to Berkeley by 7 * Keith Muller of the University of California, San Diego. 8 * 9 * Redistribution and use in source and binary forms, with or without 10 * modification, are permitted provided that the following conditions 11 * are met: 12 * 1. Redistributions of source code must retain the above copyright 13 * notice, this list of conditions and the following disclaimer. 14 * 2. Redistributions in binary form must reproduce the above copyright 15 * notice, this list of conditions and the following disclaimer in the 16 * documentation and/or other materials provided with the distribution. 17 * 4. Neither the name of the University nor the names of its contributors 18 * may be used to endorse or promote products derived from this software 19 * without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31 * SUCH DAMAGE. 32 */ 33 34 #ifndef lint 35 #if 0 36 static char sccsid[] = "@(#)ar_io.c 8.2 (Berkeley) 4/18/94"; 37 #endif 38 #endif /* not lint */ 39 #include <sys/cdefs.h> 40 __FBSDID("$FreeBSD$"); 41 42 #include <sys/types.h> 43 #include <sys/ioctl.h> 44 #include <sys/mtio.h> 45 #include <sys/stat.h> 46 #include <sys/wait.h> 47 #include <err.h> 48 #include <errno.h> 49 #include <fcntl.h> 50 #include <signal.h> 51 #include <stdint.h> 52 #include <stdio.h> 53 #include <string.h> 54 #include <stdlib.h> 55 #include <unistd.h> 56 #include "pax.h" 57 #include "options.h" 58 #include "extern.h" 59 60 /* 61 * Routines which deal directly with the archive I/O device/file. 62 */ 63 64 #define DMOD 0666 /* default mode of created archives */ 65 #define EXT_MODE O_RDONLY /* open mode for list/extract */ 66 #define AR_MODE (O_WRONLY | O_CREAT | O_TRUNC) /* mode for archive */ 67 #define APP_MODE O_RDWR /* mode for append */ 68 69 static char none[] = "<NONE>"; /* pseudo name for no file */ 70 static char stdo[] = "<STDOUT>"; /* pseudo name for stdout */ 71 static char stdn[] = "<STDIN>"; /* pseudo name for stdin */ 72 static int arfd = -1; /* archive file descriptor */ 73 static int artyp = ISREG; /* archive type: file/FIFO/tape */ 74 static int arvol = 1; /* archive volume number */ 75 static int lstrval = -1; /* return value from last i/o */ 76 static int io_ok; /* i/o worked on volume after resync */ 77 static int did_io; /* did i/o ever occur on volume? */ 78 static int done; /* set via tty termination */ 79 static struct stat arsb; /* stat of archive device at open */ 80 static int invld_rec; /* tape has out of spec record size */ 81 static int wr_trail = 1; /* trailer was rewritten in append */ 82 static int can_unlnk = 0; /* do we unlink null archives? */ 83 const char *arcname; /* printable name of archive */ 84 const char *gzip_program; /* name of gzip program */ 85 static pid_t zpid = -1; /* pid of child process */ 86 87 static int get_phys(void); 88 static void ar_start_gzip(int, const char *, int); 89 90 /* 91 * ar_open() 92 * Opens the next archive volume. Determines the type of the device and 93 * sets up block sizes as required by the archive device and the format. 94 * Note: we may be called with name == NULL on the first open only. 95 * Return: 96 * -1 on failure, 0 otherwise 97 */ 98 99 int 100 ar_open(const char *name) 101 { 102 struct mtget mb; 103 104 if (arfd != -1) 105 (void)close(arfd); 106 arfd = -1; 107 can_unlnk = did_io = io_ok = invld_rec = 0; 108 artyp = ISREG; 109 flcnt = 0; 110 111 /* 112 * open based on overall operation mode 113 */ 114 switch (act) { 115 case LIST: 116 case EXTRACT: 117 if (name == NULL) { 118 arfd = STDIN_FILENO; 119 arcname = stdn; 120 } else if ((arfd = open(name, EXT_MODE, DMOD)) < 0) 121 syswarn(0, errno, "Failed open to read on %s", name); 122 if (arfd != -1 && gzip_program != NULL) 123 ar_start_gzip(arfd, gzip_program, 0); 124 break; 125 case ARCHIVE: 126 if (name == NULL) { 127 arfd = STDOUT_FILENO; 128 arcname = stdo; 129 } else if ((arfd = open(name, AR_MODE, DMOD)) < 0) 130 syswarn(0, errno, "Failed open to write on %s", name); 131 else 132 can_unlnk = 1; 133 if (arfd != -1 && gzip_program != NULL) 134 ar_start_gzip(arfd, gzip_program, 1); 135 break; 136 case APPND: 137 if (name == NULL) { 138 arfd = STDOUT_FILENO; 139 arcname = stdo; 140 } else if ((arfd = open(name, APP_MODE, DMOD)) < 0) 141 syswarn(0, errno, "Failed open to read/write on %s", 142 name); 143 break; 144 case COPY: 145 /* 146 * arfd not used in COPY mode 147 */ 148 arcname = none; 149 lstrval = 1; 150 return(0); 151 } 152 if (arfd < 0) 153 return(-1); 154 155 if (chdname != NULL) 156 if (chdir(chdname) != 0) { 157 syswarn(1, errno, "Failed chdir to %s", chdname); 158 return(-1); 159 } 160 /* 161 * set up is based on device type 162 */ 163 if (fstat(arfd, &arsb) < 0) { 164 syswarn(0, errno, "Failed stat on %s", arcname); 165 (void)close(arfd); 166 arfd = -1; 167 can_unlnk = 0; 168 return(-1); 169 } 170 if (S_ISDIR(arsb.st_mode)) { 171 paxwarn(0, "Cannot write an archive on top of a directory %s", 172 arcname); 173 (void)close(arfd); 174 arfd = -1; 175 can_unlnk = 0; 176 return(-1); 177 } 178 179 if (S_ISCHR(arsb.st_mode)) 180 artyp = ioctl(arfd, MTIOCGET, &mb) ? ISCHR : ISTAPE; 181 else if (S_ISBLK(arsb.st_mode)) 182 artyp = ISBLK; 183 else if ((lseek(arfd, (off_t)0L, SEEK_CUR) == -1) && (errno == ESPIPE)) 184 artyp = ISPIPE; 185 else 186 artyp = ISREG; 187 188 /* 189 * make sure we beyond any doubt that we only can unlink regular files 190 * we created 191 */ 192 if (artyp != ISREG) 193 can_unlnk = 0; 194 /* 195 * if we are writing, we are done 196 */ 197 if (act == ARCHIVE) { 198 blksz = rdblksz = wrblksz; 199 lstrval = 1; 200 return(0); 201 } 202 203 /* 204 * set default blksz on read. APPNDs writes rdblksz on the last volume 205 * On all new archive volumes, we shift to wrblksz (if the user 206 * specified one, otherwise we will continue to use rdblksz). We 207 * must to set blocksize based on what kind of device the archive is 208 * stored. 209 */ 210 switch(artyp) { 211 case ISTAPE: 212 /* 213 * Tape drives come in at least two flavors. Those that support 214 * variable sized records and those that have fixed sized 215 * records. They must be treated differently. For tape drives 216 * that support variable sized records, we must make large 217 * reads to make sure we get the entire record, otherwise we 218 * will just get the first part of the record (up to size we 219 * asked). Tapes with fixed sized records may or may not return 220 * multiple records in a single read. We really do not care 221 * what the physical record size is UNLESS we are going to 222 * append. (We will need the physical block size to rewrite 223 * the trailer). Only when we are appending do we go to the 224 * effort to figure out the true PHYSICAL record size. 225 */ 226 blksz = rdblksz = MAXBLK; 227 break; 228 case ISPIPE: 229 case ISBLK: 230 case ISCHR: 231 /* 232 * Blocksize is not a major issue with these devices (but must 233 * be kept a multiple of 512). If the user specified a write 234 * block size, we use that to read. Under append, we must 235 * always keep blksz == rdblksz. Otherwise we go ahead and use 236 * the device optimal blocksize as (and if) returned by stat 237 * and if it is within pax specs. 238 */ 239 if ((act == APPND) && wrblksz) { 240 blksz = rdblksz = wrblksz; 241 break; 242 } 243 244 if ((arsb.st_blksize > 0) && (arsb.st_blksize < MAXBLK) && 245 ((arsb.st_blksize % BLKMULT) == 0)) 246 rdblksz = arsb.st_blksize; 247 else 248 rdblksz = DEVBLK; 249 /* 250 * For performance go for large reads when we can without harm 251 */ 252 if ((act == APPND) || (artyp == ISCHR)) 253 blksz = rdblksz; 254 else 255 blksz = MAXBLK; 256 break; 257 case ISREG: 258 /* 259 * if the user specified wrblksz works, use it. Under appends 260 * we must always keep blksz == rdblksz 261 */ 262 if ((act == APPND) && wrblksz && ((arsb.st_size%wrblksz)==0)){ 263 blksz = rdblksz = wrblksz; 264 break; 265 } 266 /* 267 * See if we can find the blocking factor from the file size 268 */ 269 for (rdblksz = MAXBLK; rdblksz > 0; rdblksz -= BLKMULT) 270 if ((arsb.st_size % rdblksz) == 0) 271 break; 272 /* 273 * When we cannot find a match, we may have a flawed archive. 274 */ 275 if (rdblksz <= 0) 276 rdblksz = FILEBLK; 277 /* 278 * for performance go for large reads when we can 279 */ 280 if (act == APPND) 281 blksz = rdblksz; 282 else 283 blksz = MAXBLK; 284 break; 285 default: 286 /* 287 * should never happen, worse case, slow... 288 */ 289 blksz = rdblksz = BLKMULT; 290 break; 291 } 292 lstrval = 1; 293 return(0); 294 } 295 296 /* 297 * ar_close() 298 * closes archive device, increments volume number, and prints i/o summary 299 */ 300 void 301 ar_close(void) 302 { 303 int status; 304 305 if (arfd < 0) { 306 did_io = io_ok = flcnt = 0; 307 return; 308 } 309 310 /* 311 * Close archive file. This may take a LONG while on tapes (we may be 312 * forced to wait for the rewind to complete) so tell the user what is 313 * going on (this avoids the user hitting control-c thinking pax is 314 * broken). 315 */ 316 if (vflag && (artyp == ISTAPE)) { 317 if (vfpart) 318 (void)putc('\n', listf); 319 (void)fprintf(listf, 320 "%s: Waiting for tape drive close to complete...", 321 argv0); 322 (void)fflush(listf); 323 } 324 325 /* 326 * if nothing was written to the archive (and we created it), we remove 327 * it 328 */ 329 if (can_unlnk && (fstat(arfd, &arsb) == 0) && (S_ISREG(arsb.st_mode)) && 330 (arsb.st_size == 0)) { 331 (void)unlink(arcname); 332 can_unlnk = 0; 333 } 334 335 /* 336 * for a quick extract/list, pax frequently exits before the child 337 * process is done 338 */ 339 if ((act == LIST || act == EXTRACT) && nflag && zpid > 0) 340 kill(zpid, SIGINT); 341 342 (void)close(arfd); 343 344 /* Do not exit before child to ensure data integrity */ 345 if (zpid > 0) 346 waitpid(zpid, &status, 0); 347 348 if (vflag && (artyp == ISTAPE)) { 349 (void)fputs("done.\n", listf); 350 vfpart = 0; 351 (void)fflush(listf); 352 } 353 arfd = -1; 354 355 if (!io_ok && !did_io) { 356 flcnt = 0; 357 return; 358 } 359 did_io = io_ok = 0; 360 361 /* 362 * The volume number is only increased when the last device has data 363 * and we have already determined the archive format. 364 */ 365 if (frmt != NULL) 366 ++arvol; 367 368 if (!vflag) { 369 flcnt = 0; 370 return; 371 } 372 373 /* 374 * Print out a summary of I/O for this archive volume. 375 */ 376 if (vfpart) { 377 (void)putc('\n', listf); 378 vfpart = 0; 379 } 380 381 /* 382 * If we have not determined the format yet, we just say how many bytes 383 * we have skipped over looking for a header to id. There is no way we 384 * could have written anything yet. 385 */ 386 if (frmt == NULL) { 387 # ifdef NET2_STAT 388 (void)fprintf(listf, "%s: unknown format, %lu bytes skipped.\n", 389 argv0, rdcnt); 390 # else 391 (void)fprintf(listf, "%s: unknown format, %ju bytes skipped.\n", 392 argv0, (uintmax_t)rdcnt); 393 # endif 394 (void)fflush(listf); 395 flcnt = 0; 396 return; 397 } 398 399 if (strcmp(NM_CPIO, argv0) == 0) 400 (void)fprintf(listf, "%llu blocks\n", 401 (unsigned long long)((rdcnt ? rdcnt : wrcnt) / 5120)); 402 else if (strcmp(NM_TAR, argv0) != 0) 403 (void)fprintf(listf, 404 # ifdef NET2_STAT 405 "%s: %s vol %d, %lu files, %lu bytes read, %lu bytes written.\n", 406 argv0, frmt->name, arvol-1, flcnt, rdcnt, wrcnt); 407 # else 408 "%s: %s vol %d, %ju files, %ju bytes read, %ju bytes written.\n", 409 argv0, frmt->name, arvol-1, (uintmax_t)flcnt, 410 (uintmax_t)rdcnt, (uintmax_t)wrcnt); 411 # endif 412 (void)fflush(listf); 413 flcnt = 0; 414 } 415 416 /* 417 * ar_drain() 418 * drain any archive format independent padding from an archive read 419 * from a socket or a pipe. This is to prevent the process on the 420 * other side of the pipe from getting a SIGPIPE (pax will stop 421 * reading an archive once a format dependent trailer is detected). 422 */ 423 void 424 ar_drain(void) 425 { 426 int res; 427 char drbuf[MAXBLK]; 428 429 /* 430 * we only drain from a pipe/socket. Other devices can be closed 431 * without reading up to end of file. We sure hope that pipe is closed 432 * on the other side so we will get an EOF. 433 */ 434 if ((artyp != ISPIPE) || (lstrval <= 0)) 435 return; 436 437 /* 438 * keep reading until pipe is drained 439 */ 440 while ((res = read(arfd, drbuf, sizeof(drbuf))) > 0) 441 ; 442 lstrval = res; 443 } 444 445 /* 446 * ar_set_wr() 447 * Set up device right before switching from read to write in an append. 448 * device dependent code (if required) to do this should be added here. 449 * For all archive devices we are already positioned at the place we want 450 * to start writing when this routine is called. 451 * Return: 452 * 0 if all ready to write, -1 otherwise 453 */ 454 455 int 456 ar_set_wr(void) 457 { 458 off_t cpos; 459 460 /* 461 * we must make sure the trailer is rewritten on append, ar_next() 462 * will stop us if the archive containing the trailer was not written 463 */ 464 wr_trail = 0; 465 466 /* 467 * Add any device dependent code as required here 468 */ 469 if (artyp != ISREG) 470 return(0); 471 /* 472 * Ok we have an archive in a regular file. If we were rewriting a 473 * file, we must get rid of all the stuff after the current offset 474 * (it was not written by pax). 475 */ 476 if (((cpos = lseek(arfd, (off_t)0L, SEEK_CUR)) < 0) || 477 (ftruncate(arfd, cpos) < 0)) { 478 syswarn(1, errno, "Unable to truncate archive file"); 479 return(-1); 480 } 481 return(0); 482 } 483 484 /* 485 * ar_app_ok() 486 * check if the last volume in the archive allows appends. We cannot check 487 * this until we are ready to write since there is no spec that says all 488 * volumes in a single archive have to be of the same type... 489 * Return: 490 * 0 if we can append, -1 otherwise. 491 */ 492 493 int 494 ar_app_ok(void) 495 { 496 if (artyp == ISPIPE) { 497 paxwarn(1, "Cannot append to an archive obtained from a pipe."); 498 return(-1); 499 } 500 501 if (!invld_rec) 502 return(0); 503 paxwarn(1,"Cannot append, device record size %d does not support %s spec", 504 rdblksz, argv0); 505 return(-1); 506 } 507 508 /* 509 * ar_read() 510 * read up to a specified number of bytes from the archive into the 511 * supplied buffer. When dealing with tapes we may not always be able to 512 * read what we want. 513 * Return: 514 * Number of bytes in buffer. 0 for end of file, -1 for a read error. 515 */ 516 517 int 518 ar_read(char *buf, int cnt) 519 { 520 int res = 0; 521 522 /* 523 * if last i/o was in error, no more reads until reset or new volume 524 */ 525 if (lstrval <= 0) 526 return(lstrval); 527 528 /* 529 * how we read must be based on device type 530 */ 531 switch (artyp) { 532 case ISTAPE: 533 if ((res = read(arfd, buf, cnt)) > 0) { 534 /* 535 * CAUTION: tape systems may not always return the same 536 * sized records so we leave blksz == MAXBLK. The 537 * physical record size that a tape drive supports is 538 * very hard to determine in a uniform and portable 539 * manner. 540 */ 541 io_ok = 1; 542 if (res != rdblksz) { 543 /* 544 * Record size changed. If this is happens on 545 * any record after the first, we probably have 546 * a tape drive which has a fixed record size 547 * we are getting multiple records in a single 548 * read). Watch out for record blocking that 549 * violates pax spec (must be a multiple of 550 * BLKMULT). 551 */ 552 rdblksz = res; 553 if (rdblksz % BLKMULT) 554 invld_rec = 1; 555 } 556 return(res); 557 } 558 break; 559 case ISREG: 560 case ISBLK: 561 case ISCHR: 562 case ISPIPE: 563 default: 564 /* 565 * Files are so easy to deal with. These other things cannot 566 * be trusted at all. So when we are dealing with character 567 * devices and pipes we just take what they have ready for us 568 * and return. Trying to do anything else with them runs the 569 * risk of failure. 570 */ 571 if ((res = read(arfd, buf, cnt)) > 0) { 572 io_ok = 1; 573 return(res); 574 } 575 break; 576 } 577 578 /* 579 * We are in trouble at this point, something is broken... 580 */ 581 lstrval = res; 582 if (res < 0) 583 syswarn(1, errno, "Failed read on archive volume %d", arvol); 584 else 585 paxwarn(0, "End of archive volume %d reached", arvol); 586 return(res); 587 } 588 589 /* 590 * ar_write() 591 * Write a specified number of bytes in supplied buffer to the archive 592 * device so it appears as a single "block". Deals with errors and tries 593 * to recover when faced with short writes. 594 * Return: 595 * Number of bytes written. 0 indicates end of volume reached and with no 596 * flaws (as best that can be detected). A -1 indicates an unrecoverable 597 * error in the archive occurred. 598 */ 599 600 int 601 ar_write(char *buf, int bsz) 602 { 603 int res; 604 off_t cpos; 605 606 /* 607 * do not allow pax to create a "bad" archive. Once a write fails on 608 * an archive volume prevent further writes to it. 609 */ 610 if (lstrval <= 0) 611 return(lstrval); 612 613 if ((res = write(arfd, buf, bsz)) == bsz) { 614 wr_trail = 1; 615 io_ok = 1; 616 return(bsz); 617 } 618 /* 619 * write broke, see what we can do with it. We try to send any partial 620 * writes that may violate pax spec to the next archive volume. 621 */ 622 if (res < 0) 623 lstrval = res; 624 else 625 lstrval = 0; 626 627 switch (artyp) { 628 case ISREG: 629 if ((res > 0) && (res % BLKMULT)) { 630 /* 631 * try to fix up partial writes which are not BLKMULT 632 * in size by forcing the runt record to next archive 633 * volume 634 */ 635 if ((cpos = lseek(arfd, (off_t)0L, SEEK_CUR)) < 0) 636 break; 637 cpos -= (off_t)res; 638 if (ftruncate(arfd, cpos) < 0) 639 break; 640 res = lstrval = 0; 641 break; 642 } 643 if (res >= 0) 644 break; 645 /* 646 * if file is out of space, handle it like a return of 0 647 */ 648 if ((errno == ENOSPC) || (errno == EFBIG) || (errno == EDQUOT)) 649 res = lstrval = 0; 650 break; 651 case ISTAPE: 652 case ISCHR: 653 case ISBLK: 654 if (res >= 0) 655 break; 656 if (errno == EACCES) { 657 paxwarn(0, "Write failed, archive is write protected."); 658 res = lstrval = 0; 659 return(0); 660 } 661 /* 662 * see if we reached the end of media, if so force a change to 663 * the next volume 664 */ 665 if ((errno == ENOSPC) || (errno == EIO) || (errno == ENXIO)) 666 res = lstrval = 0; 667 break; 668 case ISPIPE: 669 default: 670 /* 671 * we cannot fix errors to these devices 672 */ 673 break; 674 } 675 676 /* 677 * Better tell the user the bad news... 678 * if this is a block aligned archive format, we may have a bad archive 679 * if the format wants the header to start at a BLKMULT boundary. While 680 * we can deal with the mis-aligned data, it violates spec and other 681 * archive readers will likely fail. If the format is not block 682 * aligned, the user may be lucky (and the archive is ok). 683 */ 684 if (res >= 0) { 685 if (res > 0) 686 wr_trail = 1; 687 io_ok = 1; 688 } 689 690 /* 691 * If we were trying to rewrite the trailer and it didn't work, we 692 * must quit right away. 693 */ 694 if (!wr_trail && (res <= 0)) { 695 paxwarn(1,"Unable to append, trailer re-write failed. Quitting."); 696 return(res); 697 } 698 699 if (res == 0) 700 paxwarn(0, "End of archive volume %d reached", arvol); 701 else if (res < 0) 702 syswarn(1, errno, "Failed write to archive volume: %d", arvol); 703 else if (!frmt->blkalgn || ((res % frmt->blkalgn) == 0)) 704 paxwarn(0,"WARNING: partial archive write. Archive MAY BE FLAWED"); 705 else 706 paxwarn(1,"WARNING: partial archive write. Archive IS FLAWED"); 707 return(res); 708 } 709 710 /* 711 * ar_rdsync() 712 * Try to move past a bad spot on a flawed archive as needed to continue 713 * I/O. Clears error flags to allow I/O to continue. 714 * Return: 715 * 0 when ok to try i/o again, -1 otherwise. 716 */ 717 718 int 719 ar_rdsync(void) 720 { 721 long fsbz; 722 off_t cpos; 723 off_t mpos; 724 struct mtop mb; 725 726 /* 727 * Fail resync attempts at user request (done) or this is going to be 728 * an update/append to an existing archive. If last i/o hit media end, 729 * we need to go to the next volume not try a resync. 730 */ 731 if ((done > 0) || (lstrval == 0)) 732 return(-1); 733 734 if ((act == APPND) || (act == ARCHIVE)) { 735 paxwarn(1, "Cannot allow updates to an archive with flaws."); 736 return(-1); 737 } 738 if (io_ok) 739 did_io = 1; 740 741 switch(artyp) { 742 case ISTAPE: 743 /* 744 * if the last i/o was a successful data transfer, we assume 745 * the fault is just a bad record on the tape that we are now 746 * past. If we did not get any data since the last resync try 747 * to move the tape forward one PHYSICAL record past any 748 * damaged tape section. Some tape drives are stubborn and need 749 * to be pushed. 750 */ 751 if (io_ok) { 752 io_ok = 0; 753 lstrval = 1; 754 break; 755 } 756 mb.mt_op = MTFSR; 757 mb.mt_count = 1; 758 if (ioctl(arfd, MTIOCTOP, &mb) < 0) 759 break; 760 lstrval = 1; 761 break; 762 case ISREG: 763 case ISCHR: 764 case ISBLK: 765 /* 766 * try to step over the bad part of the device. 767 */ 768 io_ok = 0; 769 if (((fsbz = arsb.st_blksize) <= 0) || (artyp != ISREG)) 770 fsbz = BLKMULT; 771 if ((cpos = lseek(arfd, (off_t)0L, SEEK_CUR)) < 0) 772 break; 773 mpos = fsbz - (cpos % (off_t)fsbz); 774 if (lseek(arfd, mpos, SEEK_CUR) < 0) 775 break; 776 lstrval = 1; 777 break; 778 case ISPIPE: 779 default: 780 /* 781 * cannot recover on these archive device types 782 */ 783 io_ok = 0; 784 break; 785 } 786 if (lstrval <= 0) { 787 paxwarn(1, "Unable to recover from an archive read failure."); 788 return(-1); 789 } 790 paxwarn(0, "Attempting to recover from an archive read failure."); 791 return(0); 792 } 793 794 /* 795 * ar_fow() 796 * Move the I/O position within the archive forward the specified number of 797 * bytes as supported by the device. If we cannot move the requested 798 * number of bytes, return the actual number of bytes moved in skipped. 799 * Return: 800 * 0 if moved the requested distance, -1 on complete failure, 1 on 801 * partial move (the amount moved is in skipped) 802 */ 803 804 int 805 ar_fow(off_t sksz, off_t *skipped) 806 { 807 off_t cpos; 808 off_t mpos; 809 810 *skipped = 0; 811 if (sksz <= 0) 812 return(0); 813 814 /* 815 * we cannot move forward at EOF or error 816 */ 817 if (lstrval <= 0) 818 return(lstrval); 819 820 /* 821 * Safer to read forward on devices where it is hard to find the end of 822 * the media without reading to it. With tapes we cannot be sure of the 823 * number of physical blocks to skip (we do not know physical block 824 * size at this point), so we must only read forward on tapes! 825 */ 826 if (artyp != ISREG) 827 return(0); 828 829 /* 830 * figure out where we are in the archive 831 */ 832 if ((cpos = lseek(arfd, (off_t)0L, SEEK_CUR)) >= 0) { 833 /* 834 * we can be asked to move farther than there are bytes in this 835 * volume, if so, just go to file end and let normal buf_fill() 836 * deal with the end of file (it will go to next volume by 837 * itself) 838 */ 839 if ((mpos = cpos + sksz) > arsb.st_size) { 840 *skipped = arsb.st_size - cpos; 841 mpos = arsb.st_size; 842 } else 843 *skipped = sksz; 844 if (lseek(arfd, mpos, SEEK_SET) >= 0) 845 return(0); 846 } 847 syswarn(1, errno, "Forward positioning operation on archive failed"); 848 lstrval = -1; 849 return(-1); 850 } 851 852 /* 853 * ar_rev() 854 * move the i/o position within the archive backwards the specified byte 855 * count as supported by the device. With tapes drives we RESET rdblksz to 856 * the PHYSICAL blocksize. 857 * NOTE: We should only be called to move backwards so we can rewrite the 858 * last records (the trailer) of an archive (APPEND). 859 * Return: 860 * 0 if moved the requested distance, -1 on complete failure 861 */ 862 863 int 864 ar_rev(off_t sksz) 865 { 866 off_t cpos; 867 struct mtop mb; 868 int phyblk; 869 870 /* 871 * make sure we do not have try to reverse on a flawed archive 872 */ 873 if (lstrval < 0) 874 return(lstrval); 875 876 switch(artyp) { 877 case ISPIPE: 878 if (sksz <= 0) 879 break; 880 /* 881 * cannot go backwards on these critters 882 */ 883 paxwarn(1, "Reverse positioning on pipes is not supported."); 884 lstrval = -1; 885 return(-1); 886 case ISREG: 887 case ISBLK: 888 case ISCHR: 889 default: 890 if (sksz <= 0) 891 break; 892 893 /* 894 * For things other than files, backwards movement has a very 895 * high probability of failure as we really do not know the 896 * true attributes of the device we are talking to (the device 897 * may not even have the ability to lseek() in any direction). 898 * First we figure out where we are in the archive. 899 */ 900 if ((cpos = lseek(arfd, (off_t)0L, SEEK_CUR)) < 0) { 901 syswarn(1, errno, 902 "Unable to obtain current archive byte offset"); 903 lstrval = -1; 904 return(-1); 905 } 906 907 /* 908 * we may try to go backwards past the start when the archive 909 * is only a single record. If this happens and we are on a 910 * multi volume archive, we need to go to the end of the 911 * previous volume and continue our movement backwards from 912 * there. 913 */ 914 if ((cpos -= sksz) < (off_t)0L) { 915 if (arvol > 1) { 916 /* 917 * this should never happen 918 */ 919 paxwarn(1,"Reverse position on previous volume."); 920 lstrval = -1; 921 return(-1); 922 } 923 cpos = (off_t)0L; 924 } 925 if (lseek(arfd, cpos, SEEK_SET) < 0) { 926 syswarn(1, errno, "Unable to seek archive backwards"); 927 lstrval = -1; 928 return(-1); 929 } 930 break; 931 case ISTAPE: 932 /* 933 * Calculate and move the proper number of PHYSICAL tape 934 * blocks. If the sksz is not an even multiple of the physical 935 * tape size, we cannot do the move (this should never happen). 936 * (We also cannot handler trailers spread over two vols). 937 * get_phys() also makes sure we are in front of the filemark. 938 */ 939 if ((phyblk = get_phys()) <= 0) { 940 lstrval = -1; 941 return(-1); 942 } 943 944 /* 945 * make sure future tape reads only go by physical tape block 946 * size (set rdblksz to the real size). 947 */ 948 rdblksz = phyblk; 949 950 /* 951 * if no movement is required, just return (we must be after 952 * get_phys() so the physical blocksize is properly set) 953 */ 954 if (sksz <= 0) 955 break; 956 957 /* 958 * ok we have to move. Make sure the tape drive can do it. 959 */ 960 if (sksz % phyblk) { 961 paxwarn(1, 962 "Tape drive unable to backspace requested amount"); 963 lstrval = -1; 964 return(-1); 965 } 966 967 /* 968 * move backwards the requested number of bytes 969 */ 970 mb.mt_op = MTBSR; 971 mb.mt_count = sksz/phyblk; 972 if (ioctl(arfd, MTIOCTOP, &mb) < 0) { 973 syswarn(1,errno, "Unable to backspace tape %d blocks.", 974 mb.mt_count); 975 lstrval = -1; 976 return(-1); 977 } 978 break; 979 } 980 lstrval = 1; 981 return(0); 982 } 983 984 /* 985 * get_phys() 986 * Determine the physical block size on a tape drive. We need the physical 987 * block size so we know how many bytes we skip over when we move with 988 * mtio commands. We also make sure we are BEFORE THE TAPE FILEMARK when 989 * return. 990 * This is one really SLOW routine... 991 * Return: 992 * physical block size if ok (ok > 0), -1 otherwise 993 */ 994 995 static int 996 get_phys(void) 997 { 998 int padsz = 0; 999 int res; 1000 int phyblk; 1001 struct mtop mb; 1002 char scbuf[MAXBLK]; 1003 1004 /* 1005 * move to the file mark, and then back up one record and read it. 1006 * this should tell us the physical record size the tape is using. 1007 */ 1008 if (lstrval == 1) { 1009 /* 1010 * we know we are at file mark when we get back a 0 from 1011 * read() 1012 */ 1013 while ((res = read(arfd, scbuf, sizeof(scbuf))) > 0) 1014 padsz += res; 1015 if (res < 0) { 1016 syswarn(1, errno, "Unable to locate tape filemark."); 1017 return(-1); 1018 } 1019 } 1020 1021 /* 1022 * move backwards over the file mark so we are at the end of the 1023 * last record. 1024 */ 1025 mb.mt_op = MTBSF; 1026 mb.mt_count = 1; 1027 if (ioctl(arfd, MTIOCTOP, &mb) < 0) { 1028 syswarn(1, errno, "Unable to backspace over tape filemark."); 1029 return(-1); 1030 } 1031 1032 /* 1033 * move backwards so we are in front of the last record and read it to 1034 * get physical tape blocksize. 1035 */ 1036 mb.mt_op = MTBSR; 1037 mb.mt_count = 1; 1038 if (ioctl(arfd, MTIOCTOP, &mb) < 0) { 1039 syswarn(1, errno, "Unable to backspace over last tape block."); 1040 return(-1); 1041 } 1042 if ((phyblk = read(arfd, scbuf, sizeof(scbuf))) <= 0) { 1043 syswarn(1, errno, "Cannot determine archive tape blocksize."); 1044 return(-1); 1045 } 1046 1047 /* 1048 * read forward to the file mark, then back up in front of the filemark 1049 * (this is a bit paranoid, but should be safe to do). 1050 */ 1051 while ((res = read(arfd, scbuf, sizeof(scbuf))) > 0) 1052 ; 1053 if (res < 0) { 1054 syswarn(1, errno, "Unable to locate tape filemark."); 1055 return(-1); 1056 } 1057 mb.mt_op = MTBSF; 1058 mb.mt_count = 1; 1059 if (ioctl(arfd, MTIOCTOP, &mb) < 0) { 1060 syswarn(1, errno, "Unable to backspace over tape filemark."); 1061 return(-1); 1062 } 1063 1064 /* 1065 * set lstrval so we know that the filemark has not been seen 1066 */ 1067 lstrval = 1; 1068 1069 /* 1070 * return if there was no padding 1071 */ 1072 if (padsz == 0) 1073 return(phyblk); 1074 1075 /* 1076 * make sure we can move backwards over the padding. (this should 1077 * never fail). 1078 */ 1079 if (padsz % phyblk) { 1080 paxwarn(1, "Tape drive unable to backspace requested amount"); 1081 return(-1); 1082 } 1083 1084 /* 1085 * move backwards over the padding so the head is where it was when 1086 * we were first called (if required). 1087 */ 1088 mb.mt_op = MTBSR; 1089 mb.mt_count = padsz/phyblk; 1090 if (ioctl(arfd, MTIOCTOP, &mb) < 0) { 1091 syswarn(1,errno,"Unable to backspace tape over %d pad blocks", 1092 mb.mt_count); 1093 return(-1); 1094 } 1095 return(phyblk); 1096 } 1097 1098 /* 1099 * ar_next() 1100 * prompts the user for the next volume in this archive. For some devices 1101 * we may allow the media to be changed. Otherwise a new archive is 1102 * prompted for. By pax spec, if there is no controlling tty or an eof is 1103 * read on tty input, we must quit pax. 1104 * Return: 1105 * 0 when ready to continue, -1 when all done 1106 */ 1107 1108 int 1109 ar_next(void) 1110 { 1111 static char *arcbuf; 1112 char buf[PAXPATHLEN+2]; 1113 sigset_t o_mask; 1114 1115 /* 1116 * WE MUST CLOSE THE DEVICE. A lot of devices must see last close, (so 1117 * things like writing EOF etc will be done) (Watch out ar_close() can 1118 * also be called via a signal handler, so we must prevent a race. 1119 */ 1120 if (sigprocmask(SIG_BLOCK, &s_mask, &o_mask) < 0) 1121 syswarn(0, errno, "Unable to set signal mask"); 1122 ar_close(); 1123 if (sigprocmask(SIG_SETMASK, &o_mask, NULL) < 0) 1124 syswarn(0, errno, "Unable to restore signal mask"); 1125 1126 if (done || !wr_trail || Oflag || strcmp(NM_TAR, argv0) == 0) 1127 return(-1); 1128 1129 tty_prnt("\nATTENTION! %s archive volume change required.\n", argv0); 1130 1131 /* 1132 * if i/o is on stdin or stdout, we cannot reopen it (we do not know 1133 * the name), the user will be forced to type it in. 1134 */ 1135 if (strcmp(arcname, stdo) && strcmp(arcname, stdn) && (artyp != ISREG) 1136 && (artyp != ISPIPE)) { 1137 if (artyp == ISTAPE) { 1138 tty_prnt("%s ready for archive tape volume: %d\n", 1139 arcname, arvol); 1140 tty_prnt("Load the NEXT TAPE on the tape drive"); 1141 } else { 1142 tty_prnt("%s ready for archive volume: %d\n", 1143 arcname, arvol); 1144 tty_prnt("Load the NEXT STORAGE MEDIA (if required)"); 1145 } 1146 1147 if ((act == ARCHIVE) || (act == APPND)) 1148 tty_prnt(" and make sure it is WRITE ENABLED.\n"); 1149 else 1150 tty_prnt("\n"); 1151 1152 for(;;) { 1153 tty_prnt("Type \"y\" to continue, \".\" to quit %s,", 1154 argv0); 1155 tty_prnt(" or \"s\" to switch to new device.\nIf you"); 1156 tty_prnt(" cannot change storage media, type \"s\"\n"); 1157 tty_prnt("Is the device ready and online? > "); 1158 1159 if ((tty_read(buf,sizeof(buf))<0) || !strcmp(buf,".")){ 1160 done = 1; 1161 lstrval = -1; 1162 tty_prnt("Quitting %s!\n", argv0); 1163 vfpart = 0; 1164 return(-1); 1165 } 1166 1167 if ((buf[0] == '\0') || (buf[1] != '\0')) { 1168 tty_prnt("%s unknown command, try again\n",buf); 1169 continue; 1170 } 1171 1172 switch (buf[0]) { 1173 case 'y': 1174 case 'Y': 1175 /* 1176 * we are to continue with the same device 1177 */ 1178 if (ar_open(arcname) >= 0) 1179 return(0); 1180 tty_prnt("Cannot re-open %s, try again\n", 1181 arcname); 1182 continue; 1183 case 's': 1184 case 'S': 1185 /* 1186 * user wants to open a different device 1187 */ 1188 tty_prnt("Switching to a different archive\n"); 1189 break; 1190 default: 1191 tty_prnt("%s unknown command, try again\n",buf); 1192 continue; 1193 } 1194 break; 1195 } 1196 } else 1197 tty_prnt("Ready for archive volume: %d\n", arvol); 1198 1199 /* 1200 * have to go to a different archive 1201 */ 1202 for (;;) { 1203 tty_prnt("Input archive name or \".\" to quit %s.\n", argv0); 1204 tty_prnt("Archive name > "); 1205 1206 if ((tty_read(buf, sizeof(buf)) < 0) || !strcmp(buf, ".")) { 1207 done = 1; 1208 lstrval = -1; 1209 tty_prnt("Quitting %s!\n", argv0); 1210 vfpart = 0; 1211 return(-1); 1212 } 1213 if (buf[0] == '\0') { 1214 tty_prnt("Empty file name, try again\n"); 1215 continue; 1216 } 1217 if (!strcmp(buf, "..")) { 1218 tty_prnt("Illegal file name: .. try again\n"); 1219 continue; 1220 } 1221 if (strlen(buf) > PAXPATHLEN) { 1222 tty_prnt("File name too long, try again\n"); 1223 continue; 1224 } 1225 1226 /* 1227 * try to open new archive 1228 */ 1229 if (ar_open(buf) >= 0) { 1230 free(arcbuf); 1231 if ((arcbuf = strdup(buf)) == NULL) { 1232 done = 1; 1233 lstrval = -1; 1234 paxwarn(0, "Cannot save archive name."); 1235 return(-1); 1236 } 1237 arcname = arcbuf; 1238 break; 1239 } 1240 tty_prnt("Cannot open %s, try again\n", buf); 1241 continue; 1242 } 1243 return(0); 1244 } 1245 1246 /* 1247 * ar_start_gzip() 1248 * starts the gzip compression/decompression process as a child, using magic 1249 * to keep the fd the same in the calling function (parent). 1250 */ 1251 void 1252 ar_start_gzip(int fd, const char *gzip_prog, int wr) 1253 { 1254 int fds[2]; 1255 const char *gzip_flags; 1256 1257 if (pipe(fds) < 0) 1258 err(1, "could not pipe"); 1259 zpid = fork(); 1260 if (zpid < 0) 1261 err(1, "could not fork"); 1262 1263 /* parent */ 1264 if (zpid) { 1265 if (wr) 1266 dup2(fds[1], fd); 1267 else 1268 dup2(fds[0], fd); 1269 close(fds[0]); 1270 close(fds[1]); 1271 } else { 1272 if (wr) { 1273 dup2(fds[0], STDIN_FILENO); 1274 dup2(fd, STDOUT_FILENO); 1275 gzip_flags = "-c"; 1276 } else { 1277 dup2(fds[1], STDOUT_FILENO); 1278 dup2(fd, STDIN_FILENO); 1279 gzip_flags = "-dc"; 1280 } 1281 close(fds[0]); 1282 close(fds[1]); 1283 if (execlp(gzip_prog, gzip_prog, gzip_flags, 1284 (char *)NULL) < 0) 1285 err(1, "could not exec"); 1286 /* NOTREACHED */ 1287 } 1288 } 1289