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