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