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