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