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