xref: /netbsd-src/sbin/restore/tape.c (revision 6e3d02bbcd6ad76dfee4cfb9687a4fd87c0e86d6)
1 /*	$NetBSD: tape.c,v 1.75 2024/02/05 22:08:04 andvar Exp $	*/
2 
3 /*
4  * Copyright (c) 1983, 1993
5  *	The Regents of the University of California.  All rights reserved.
6  * (c) UNIX System Laboratories, Inc.
7  * All or some portions of this file are derived from material licensed
8  * to the University of California by American Telephone and Telegraph
9  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
10  * the permission of UNIX System Laboratories, Inc.
11  *
12  * Redistribution and use in source and binary forms, with or without
13  * modification, are permitted provided that the following conditions
14  * are met:
15  * 1. Redistributions of source code must retain the above copyright
16  *    notice, this list of conditions and the following disclaimer.
17  * 2. Redistributions in binary form must reproduce the above copyright
18  *    notice, this list of conditions and the following disclaimer in the
19  *    documentation and/or other materials provided with the distribution.
20  * 3. Neither the name of the University nor the names of its contributors
21  *    may be used to endorse or promote products derived from this software
22  *    without specific prior written permission.
23  *
24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34  * SUCH DAMAGE.
35  */
36 
37 #include <sys/cdefs.h>
38 #ifndef lint
39 #if 0
40 static char sccsid[] = "@(#)tape.c	8.9 (Berkeley) 5/1/95";
41 #else
42 __RCSID("$NetBSD: tape.c,v 1.75 2024/02/05 22:08:04 andvar Exp $");
43 #endif
44 #endif /* not lint */
45 
46 #include <sys/param.h>
47 #include <sys/file.h>
48 #include <sys/ioctl.h>
49 #include <sys/mtio.h>
50 #include <sys/stat.h>
51 #include <sys/extattr.h>
52 #define _ACL_PRIVATE
53 #include <sys/acl.h>
54 
55 #include <ufs/ufs/extattr.h>
56 #include <ufs/ufs/dinode.h>
57 #include <protocols/dumprestore.h>
58 
59 #include <err.h>
60 #include <errno.h>
61 #include <paths.h>
62 #include <setjmp.h>
63 #include <stdio.h>
64 #include <stdlib.h>
65 #include <string.h>
66 #include <time.h>
67 #include <unistd.h>
68 
69 #include <md5.h>
70 #include <rmd160.h>
71 #include <sha1.h>
72 
73 #include "restore.h"
74 #include "extern.h"
75 
76 static u_int32_t fssize = MAXBSIZE;
77 static int	mt = -1;
78 static int	pipein = 0;
79 static char	magtape[BUFSIZ];
80 static int	blkcnt;
81 static int	numtrec;
82 static char	*tapebuf;
83 static union	u_spcl endoftapemark;
84 static int	blksread;		/* blocks read since last header */
85 static int	tpblksread = 0;		/* TP_BSIZE blocks read */
86 static int	tapesread;
87 static jmp_buf	restart;
88 static int	gettingfile = 0;	/* restart has a valid frame */
89 #ifdef RRESTORE
90 static const char *host = NULL;
91 #endif
92 
93 static int	ofile;
94 static char	lnkbuf[MAXPATHLEN + 1];
95 static int	pathlen;
96 
97 int		oldinofmt;	/* old inode format conversion required */
98 int		Bcvt;		/* Swap Bytes (for CCI or sun) */
99 
100 const struct digest_desc *ddesc;
101 
102 static union digest_context {
103 	MD5_CTX dc_MD5;
104 	SHA1_CTX dc_SHA1;
105 	RMD160_CTX dc_RMD160;
106 } dcontext;
107 
108 /*
109  * 32 for md5; 40 for sha1 and rmd160
110  * plus a null terminator.
111  */
112 #define DIGEST_BUFFER_SIZE (40 + 1)
113 
114 #define	FLUSHTAPEBUF()	blkcnt = ntrec + 1
115 
116 const char *namespace_names[] = EXTATTR_NAMESPACE_NAMES;
117 
118 
119 union u_ospcl {
120 	char dummy[TP_BSIZE];
121 	struct	s_ospcl {
122 		int32_t   c_type;
123 		int32_t   c_date;
124 		int32_t   c_ddate;
125 		int32_t   c_volume;
126 		int32_t   c_tapea;
127 		u_int16_t c_inumber;
128 		int32_t   c_magic;
129 		int32_t   c_checksum;
130 		struct odinode {
131 			unsigned short odi_mode;
132 			u_int16_t odi_nlink;
133 			u_int16_t odi_uid;
134 			u_int16_t odi_gid;
135 			int32_t   odi_size;
136 			int32_t   odi_rdev;
137 			char      odi_addr[36];
138 			int32_t   odi_atime;
139 			int32_t   odi_mtime;
140 			int32_t   odi_ctime;
141 		} c_odinode;
142 		int32_t c_count;
143 		char    c_addr[256];
144 	} s_ospcl;
145 };
146 
147 static void	 accthdr(struct s_spcl *);
148 static int	 checksum(int *);
149 static void	 findinode(struct s_spcl *);
150 static void	 findtapeblksize(void);
151 static char	*setupextattr(size_t);
152 static void	 xtrattr(char *, size_t);
153 static void	 skiphole(void (*)(char *, size_t), volatile size_t *);
154 static void	 getbitmap(char **);
155 static int	 gethead(struct s_spcl *);
156 static void	 readtape(char *);
157 static void	 setdumpnum(void);
158 static void	 terminateinput(void);
159 static void	 xtrfile(char *, size_t);
160 static void	 xtrlnkfile(char *, size_t);
161 __dead static void	 xtrlnkskip(char *, size_t);
162 static void	 xtrskip(char *, size_t);
163 static void	 swap_header(struct s_spcl *);
164 static void	 swap_old_header(struct s_ospcl *);
165 
166 ////////////////////////////////////////////////////////////
167 // thunks for type correctness
168 
169 #define WRAP(alg) \
170 	static void							\
171 	do_##alg##Init(void *ctx)					\
172 	{								\
173 		alg##Init(ctx);						\
174 	}								\
175 									\
176 	static void							\
177 	do_##alg##Update(union digest_context *ctx,			\
178 		const void *buf, unsigned len)				\
179 	{								\
180 		alg##Update(&ctx->dc_##alg, buf, len);			\
181 	}								\
182 									\
183 	static char *							\
184 	do_##alg##End(void *ctx, char *str)				\
185 	{								\
186 		return alg##End(ctx, str);				\
187 	}
188 
189 WRAP(MD5);
190 WRAP(SHA1);
191 WRAP(RMD160);
192 
193 static const struct digest_desc digest_descs[] = {
194 	{ "MD5",
195 	  do_MD5Init,
196 	  do_MD5Update,
197 	  do_MD5End, },
198 	{ "SHA1",
199 	  do_SHA1Init,
200 	  do_SHA1Update,
201 	  do_SHA1End, },
202 	{ "RMD160",
203 	  do_RMD160Init,
204 	  do_RMD160Update,
205 	  do_RMD160End, },
206 	{ .dd_name = NULL },
207 };
208 
209 ////////////////////////////////////////////////////////////
210 
211 const struct digest_desc *
digest_lookup(const char * name)212 digest_lookup(const char *name)
213 {
214 	const struct digest_desc *dd;
215 
216 	for (dd = digest_descs; dd->dd_name != NULL; dd++)
217 		if (strcasecmp(dd->dd_name, name) == 0)
218 			return (dd);
219 
220 	return (NULL);
221 }
222 
223 /*
224  * Set up an input source
225  */
226 void
setinput(const char * source)227 setinput(const char *source)
228 {
229 	char *cp;
230 	FLUSHTAPEBUF();
231 	if (bflag)
232 		newtapebuf(ntrec);
233 	else
234 		newtapebuf(NTREC > HIGHDENSITYTREC ? NTREC : HIGHDENSITYTREC);
235 	terminal = stdin;
236 
237 #ifdef RRESTORE
238 	if ((cp = strchr(source, ':')) != NULL) {
239 		host = source;
240 		/* Ok, because const strings don't have : */
241 		*cp++ = '\0';
242 		source = cp;
243 		if (rmthost(host) == 0)
244 			exit(1);
245 	} else
246 #endif
247 	if (strcmp(source, "-") == 0) {
248 		/*
249 		 * Since input is coming from a pipe we must establish
250 		 * our own connection to the terminal.
251 		 */
252 		terminal = fopen(_PATH_TTY, "r");
253 		if (terminal == NULL) {
254 			(void)fprintf(stderr, "cannot open %s: %s\n",
255 			    _PATH_TTY, strerror(errno));
256 			terminal = fopen(_PATH_DEVNULL, "r");
257 			if (terminal == NULL) {
258 				(void)fprintf(stderr, "cannot open %s: %s\n",
259 				    _PATH_DEVNULL, strerror(errno));
260 				exit(1);
261 			}
262 		}
263 		pipein++;
264 	}
265 	(void) strcpy(magtape, source);
266 }
267 
268 void
newtapebuf(long size)269 newtapebuf(long size)
270 {
271 	static int tapebufsize = -1;
272 
273 	ntrec = size;
274 	if (size <= tapebufsize)
275 		return;
276 	if (tapebuf != NULL)
277 		free(tapebuf);
278 	tapebuf = malloc(size * TP_BSIZE);
279 	if (tapebuf == NULL) {
280 		fprintf(stderr, "Cannot allocate space for tape buffer\n");
281 		exit(1);
282 	}
283 	tapebufsize = size;
284 }
285 
286 /*
287  * Verify that the tape drive can be accessed and
288  * that it actually is a dump tape.
289  */
290 void
setup(void)291 setup(void)
292 {
293 	int i, j, *ip;
294 	struct stat stbuf;
295 
296 	vprintf(stdout, "Verify tape and initialize maps\n");
297 #ifdef RRESTORE
298 	if (host)
299 		mt = rmtopen(magtape, 0, 0);
300 	else
301 #endif
302 	if (pipein)
303 		mt = 0;
304 	else
305 		mt = open(magtape, O_RDONLY, 0);
306 	if (mt < 0) {
307 		fprintf(stderr, "%s: %s\n", magtape, strerror(errno));
308 		exit(1);
309 	}
310 	volno = 1;
311 	setdumpnum();
312 	FLUSHTAPEBUF();
313 	if (!pipein && !bflag)
314 		findtapeblksize();
315 	if (gethead(&spcl) == FAIL) {
316 		blkcnt--; /* push back this block */
317 		blksread--;
318 		tpblksread--;
319 		cvtflag++;
320 		if (gethead(&spcl) == FAIL) {
321 			fprintf(stderr, "Tape is not a dump tape\n");
322 			exit(1);
323 		}
324 		fprintf(stderr, "Converting to new file system format.\n");
325 	}
326 	if (pipein) {
327 		endoftapemark.s_spcl.c_magic = cvtflag ? OFS_MAGIC :
328 		    FS_UFS2_MAGIC;
329 		endoftapemark.s_spcl.c_type = TS_END;
330 		ip = (int *)&endoftapemark;
331 		j = sizeof(union u_spcl) / sizeof(int);
332 		i = 0;
333 		do
334 			i += *ip++;
335 		while (--j);
336 		endoftapemark.s_spcl.c_checksum = CHECKSUM - i;
337 	}
338 	if (vflag || command == 't')
339 		printdumpinfo();
340 	dumptime = spcl.c_ddate;
341 	dumpdate = spcl.c_date;
342 	if (stat(".", &stbuf) < 0) {
343 		fprintf(stderr, "cannot stat .: %s\n", strerror(errno));
344 		exit(1);
345 	}
346 	if (stbuf.st_blksize >= TP_BSIZE && stbuf.st_blksize <= MAXBSIZE)
347 		fssize = stbuf.st_blksize;
348 	if (((fssize - 1) & fssize) != 0) {
349 		fprintf(stderr, "bad block size %d\n", fssize);
350 		exit(1);
351 	}
352 	if (spcl.c_volume != 1) {
353 		fprintf(stderr, "Tape is not volume 1 of the dump\n");
354 		exit(1);
355 	}
356 	if (gethead(&spcl) == FAIL) {
357 		dprintf(stdout, "header read failed at %d blocks\n", blksread);
358 		panic("no header after volume mark!\n");
359 	}
360 	findinode(&spcl);
361 	if (spcl.c_type != TS_CLRI) {
362 		fprintf(stderr, "Cannot find file removal list\n");
363 		exit(1);
364 	}
365 	getbitmap(&usedinomap);
366 	getbitmap(&dumpmap);
367 	/*
368 	 * If there may be whiteout entries on the tape, pretend that the
369 	 * whiteout inode exists, so that the whiteout entries can be
370 	 * extracted.
371 	 */
372 	if (oldinofmt == 0)
373 		SETINO(UFS_WINO, dumpmap);
374 }
375 
376 /*
377  * Prompt user to load a new dump volume.
378  * "Nextvol" is the next suggested volume to use.
379  * This suggested volume is enforced when doing full
380  * or incremental restores, but can be overridden by
381  * the user when only extracting a subset of the files.
382  */
383 void
getvol(int nextvol)384 getvol(int nextvol)
385 {
386 	int newvol, savecnt, wantnext, i;
387 	union u_spcl tmpspcl;
388 #	define tmpbuf tmpspcl.s_spcl
389 	char buf[TP_BSIZE];
390 
391 	newvol = savecnt = wantnext = 0;
392 	if (nextvol == 1) {
393 		tapesread = 0;
394 		gettingfile = 0;
395 	}
396 	if (pipein) {
397 		if (nextvol != 1)
398 			panic("Changing volumes on pipe input?\n");
399 		if (volno == 1)
400 			return;
401 		goto gethdr;
402 	}
403 	savecnt = blksread;
404 again:
405 	if (pipein)
406 		exit(1); /* pipes do not get a second chance */
407 	if (command == 'R' || command == 'r' || curfile.action != SKIP) {
408 		newvol = nextvol;
409 		wantnext = 1;
410 	} else {
411 		newvol = 0;
412 		wantnext = 0;
413 	}
414 	while (newvol <= 0) {
415 		if (tapesread == 0) {
416 			fprintf(stderr, "%s%s%s%s%s",
417 			    "You have not read any tapes yet.\n",
418 			    "Unless you know which volume your",
419 			    " file(s) are on you should start\n",
420 			    "with the last volume and work",
421 			    " towards the first.\n");
422 			fprintf(stderr,
423 			    "(Use 1 for the first volume/tape, etc.)\n");
424 		} else {
425 			fprintf(stderr, "You have read volumes");
426 			strcpy(buf, ": ");
427 			for (i = 1; i < 32; i++)
428 				if (tapesread & (1 << i)) {
429 					fprintf(stderr, "%s%d", buf, i);
430 					strcpy(buf, ", ");
431 				}
432 			fprintf(stderr, "\n");
433 		}
434 		do	{
435 			fprintf(stderr, "Specify next volume #: ");
436 			(void) fflush(stderr);
437 			(void) fgets(buf, BUFSIZ, terminal);
438 		} while (!feof(terminal) && buf[0] == '\n');
439 		if (feof(terminal))
440 			exit(1);
441 		newvol = atoi(buf);
442 		if (newvol <= 0) {
443 			fprintf(stderr,
444 			    "Volume numbers are positive numerics\n");
445 		}
446 	}
447 	if (newvol == volno) {
448 		tapesread |= 1 << volno;
449 		return;
450 	}
451 	closemt();
452 	fprintf(stderr, "Mount tape volume %d\n", newvol);
453 	fprintf(stderr, "Enter ``none'' if there are no more tapes\n");
454 	fprintf(stderr, "otherwise enter tape name (default: %s) ", magtape);
455 	(void) fflush(stderr);
456 	(void) fgets(buf, BUFSIZ, terminal);
457 	if (feof(terminal))
458 		exit(1);
459 	if (!strcmp(buf, "none\n")) {
460 		terminateinput();
461 		return;
462 	}
463 	if (buf[0] != '\n') {
464 		(void) strcpy(magtape, buf);
465 		magtape[strlen(magtape) - 1] = '\0';
466 	}
467 #ifdef RRESTORE
468 	if (host)
469 		mt = rmtopen(magtape, 0, 0);
470 	else
471 #endif
472 		mt = open(magtape, O_RDONLY, 0);
473 
474 	if (mt == -1) {
475 		fprintf(stderr, "Cannot open %s\n", magtape);
476 		volno = -1;
477 		goto again;
478 	}
479 gethdr:
480 	volno = newvol;
481 	setdumpnum();
482 	FLUSHTAPEBUF();
483 	if (gethead(&tmpbuf) == FAIL) {
484 		dprintf(stdout, "header read failed at %d blocks\n", blksread);
485 		fprintf(stderr, "tape is not dump tape\n");
486 		volno = 0;
487 		goto again;
488 	}
489 	if (tmpbuf.c_volume != volno) {
490 	  	fprintf(stderr,
491 		"Volume mismatch: expecting %d, tape header claims it is %d\n",
492 		    volno, tmpbuf.c_volume);
493 		volno = 0;
494 		goto again;
495 	}
496 	if (tmpbuf.c_date != dumpdate || tmpbuf.c_ddate != dumptime) {
497 		time_t ttime = tmpbuf.c_date;
498 		fprintf(stderr, "Wrong dump date\n\tgot: %s",
499 			ctime(&ttime));
500 		fprintf(stderr, "\twanted: %s", ctime(&dumpdate));
501 		volno = 0;
502 		goto again;
503 	}
504 	tapesread |= 1 << volno;
505 	blksread = savecnt;
506  	/*
507  	 * If continuing from the previous volume, skip over any
508  	 * blocks read already at the end of the previous volume.
509  	 *
510  	 * If coming to this volume at random, skip to the beginning
511  	 * of the next record.
512  	 */
513 	dprintf(stdout, "read %ld recs, tape starts with %ld\n",
514 		(long)tpblksread, (long)tmpbuf.c_firstrec);
515  	if (tmpbuf.c_type == TS_TAPE && (tmpbuf.c_flags & DR_NEWHEADER)) {
516  		if (!wantnext) {
517  			tpblksread = tmpbuf.c_firstrec;
518  			for (i = tmpbuf.c_count; i > 0; i--)
519  				readtape(buf);
520  		} else if (tmpbuf.c_firstrec > 0 &&
521 			   tmpbuf.c_firstrec < tpblksread - 1) {
522 			/*
523 			 * -1 since we've read the volume header
524 			 */
525  			i = tpblksread - tmpbuf.c_firstrec - 1;
526 			dprintf(stderr, "Skipping %d duplicate record%s.\n",
527 				i, i > 1 ? "s" : "");
528  			while (--i >= 0)
529  				readtape(buf);
530  		}
531  	}
532 	if (curfile.action == USING) {
533 		if (volno == 1)
534 			panic("active file into volume 1\n");
535 		return;
536 	}
537 	/*
538 	 * Skip up to the beginning of the next record
539 	 */
540 	if (tmpbuf.c_type == TS_TAPE && (tmpbuf.c_flags & DR_NEWHEADER))
541 		for (i = tmpbuf.c_count; i > 0; i--)
542 			readtape(buf);
543 	(void) gethead(&spcl);
544 	findinode(&spcl);
545 	if (gettingfile) {
546 		gettingfile = 0;
547 		longjmp(restart, 1);
548 	}
549 }
550 
551 /*
552  * Handle unexpected EOF.
553  */
554 static void
terminateinput(void)555 terminateinput(void)
556 {
557 
558 	if (gettingfile && curfile.action == USING) {
559 		printf("Warning: %s %s\n",
560 		    "End-of-input encountered while extracting", curfile.name);
561 	}
562 	curfile.name = "<name unknown>";
563 	curfile.action = UNKNOWN;
564 	curfile.mode = 0;
565 	curfile.ino = maxino;
566 	if (gettingfile) {
567 		gettingfile = 0;
568 		longjmp(restart, 1);
569 	}
570 }
571 
572 /*
573  * handle multiple dumps per tape by skipping forward to the
574  * appropriate one.
575  */
576 static void
setdumpnum(void)577 setdumpnum(void)
578 {
579 	struct mtop tcom;
580 
581 	if (dumpnum == 1 || volno != 1)
582 		return;
583 	if (pipein) {
584 		fprintf(stderr, "Cannot have multiple dumps on pipe input\n");
585 		exit(1);
586 	}
587 	tcom.mt_op = MTFSF;
588 	tcom.mt_count = dumpnum - 1;
589 #ifdef RRESTORE
590 	if (host)
591 		rmtioctl(MTFSF, dumpnum - 1);
592 	else
593 #endif
594 		if (ioctl(mt, (int)MTIOCTOP, (char *)&tcom) < 0)
595 			fprintf(stderr, "ioctl MTFSF: %s\n", strerror(errno));
596 }
597 
598 void
printdumpinfo(void)599 printdumpinfo(void)
600 {
601 	time_t ttime;
602 
603 	ttime = spcl.c_date;
604 	fprintf(stdout, "Dump   date: %s", ctime(&ttime));
605 	ttime = spcl.c_ddate;
606 	fprintf(stdout, "Dumped from: %s",
607 	    (spcl.c_ddate == 0) ? "the epoch\n" : ctime(&ttime));
608 	fprintf(stderr, "Level %d dump of %s on %s:%s\n",
609 		spcl.c_level, spcl.c_filesys,
610 		*spcl.c_host? spcl.c_host: "[unknown]", spcl.c_dev);
611 	fprintf(stderr, "Label: %s\n", spcl.c_label);
612 
613 	if (Mtreefile) {
614 		ttime = spcl.c_date;
615 		fprintf(Mtreefile, "#Dump   date: %s", ctime(&ttime));
616 		ttime = spcl.c_ddate;
617 		fprintf(Mtreefile, "#Dumped from: %s",
618 		    (spcl.c_ddate == 0) ? "the epoch\n" : ctime(&ttime));
619 		fprintf(Mtreefile, "#Level %d dump of %s on %s:%s\n",
620 			spcl.c_level, spcl.c_filesys,
621 			*spcl.c_host? spcl.c_host: "[unknown]", spcl.c_dev);
622 		fprintf(Mtreefile, "#Label: %s\n", spcl.c_label);
623 		fprintf(Mtreefile, "/set uname=root gname=wheel\n");
624 		if (ferror(Mtreefile))
625 			err(1, "error writing to mtree file");
626 	}
627 }
628 
629 int
extractfile(char * name)630 extractfile(char *name)
631 {
632 	char dbuffer[DIGEST_BUFFER_SIZE];
633 	u_int flags;
634 	uid_t uid;
635 	gid_t gid;
636 	mode_t mode;
637 	int extsize;
638 	struct timespec mtimep[2], ctimep[2];
639 	struct entry *ep;
640 	int setbirth;
641 	char *buf;
642 
643 	curfile.name = name;
644 	curfile.action = USING;
645 	mtimep[0].tv_sec = curfile.atime_sec;
646 	mtimep[0].tv_nsec = curfile.atime_nsec;
647 	mtimep[1].tv_sec = curfile.mtime_sec;
648 	mtimep[1].tv_nsec = curfile.mtime_nsec;
649 
650 	setbirth = curfile.birthtime_sec != 0;
651 
652 	if (setbirth) {
653 		ctimep[0].tv_sec = curfile.atime_sec;
654 		ctimep[0].tv_nsec = curfile.atime_nsec;
655 		ctimep[1].tv_sec = curfile.birthtime_sec;
656 		ctimep[1].tv_nsec = curfile.birthtime_nsec;
657 	}
658 	extsize = curfile.extsize;
659 	uid = getuid();
660 	if (uid == 0)
661 		uid = curfile.uid;
662 	gid = curfile.gid;
663 	mode = curfile.mode;
664 	flags = curfile.file_flags;
665 	switch (mode & IFMT) {
666 
667 	default:
668 		fprintf(stderr, "%s: unknown file mode 0%o\n", name, mode);
669 		skipfile();
670 		return (FAIL);
671 
672 	case IFSOCK:
673 		vprintf(stdout, "skipped socket %s\n", name);
674 		skipfile();
675 		return (GOOD);
676 
677 	case IFDIR:
678 		if (mflag) {
679 			ep = lookupname(name);
680 			if (ep == NULL || ep->e_flags & EXTRACT)
681 				panic("unextracted directory %s\n", name);
682 			skipfile();
683 			return (GOOD);
684 		}
685 		vprintf(stdout, "extract file %s\n", name);
686 		return (genliteraldir(name, curfile.ino));
687 
688 	case IFLNK:
689 		lnkbuf[0] = '\0';
690 		pathlen = 0;
691 		buf = setupextattr(extsize);
692 		getfile(xtrlnkfile, xtrattr, xtrlnkskip);
693 		if (pathlen == 0) {
694 			vprintf(stdout,
695 			    "%s: zero length symbolic link (ignored)\n", name);
696 			return (GOOD);
697 		}
698 		if (uflag)
699 			(void) unlink(name);
700 		if (linkit(lnkbuf, name, SYMLINK) == GOOD) {
701 			if (setbirth)
702 				(void) lutimens(name, ctimep);
703 			(void) lutimens(name, mtimep);
704 			(void) lchown(name, uid, gid);
705 			(void) lchmod(name, mode);
706 			if (extsize > 0)
707 				set_extattr(-1, name, buf, extsize, SXA_LINK);
708 			if (Mtreefile) {
709 				writemtree(name, "link",
710 				    uid, gid, mode, flags);
711 			} else
712 				(void) lchflags(name, flags);
713 			return (GOOD);
714 		}
715 		return (FAIL);
716 
717 	case IFCHR:
718 	case IFBLK:
719 		vprintf(stdout, "extract special file %s\n", name);
720 		if (Nflag) {
721 			skipfile();
722 			return (GOOD);
723 		}
724 		if (uflag)
725 			(void) unlink(name);
726 		if (mknod(name, (mode & (IFCHR | IFBLK)) | 0600,
727 		    (int)curfile.rdev) < 0) {
728 			fprintf(stderr, "%s: cannot create special file: %s\n",
729 			    name, strerror(errno));
730 			skipfile();
731 			return (FAIL);
732 		}
733 		(void) chown(name, uid, gid);
734 		(void) chmod(name, mode);
735 		if (extsize == 0) {
736 			skipfile();
737 		} else {
738 			buf = setupextattr(extsize);
739 			getfile(xtrnull, xtrattr, xtrnull);
740 			set_extattr(-1, name, buf, extsize, SXA_FILE);
741 		}
742 		if (setbirth)
743 			(void) utimens(name, ctimep);
744 		(void) utimens(name, mtimep);
745 		if (Mtreefile) {
746 			writemtree(name,
747 			    ((mode & (S_IFBLK | IFCHR)) == IFBLK) ?
748 			    "block" : "char",
749 			    uid, gid, mode, flags);
750 		} else
751 			(void) chflags(name, flags);
752 		return (GOOD);
753 
754 	case IFIFO:
755 		vprintf(stdout, "extract fifo %s\n", name);
756 		if (Nflag) {
757 			skipfile();
758 			return (GOOD);
759 		}
760 		if (uflag)
761 			(void) unlink(name);
762 		if (mkfifo(name, 0600) < 0) {
763 			fprintf(stderr, "%s: cannot create fifo: %s\n",
764 			    name, strerror(errno));
765 			skipfile();
766 			return (FAIL);
767 		}
768 		(void) chown(name, uid, gid);
769 		(void) chmod(name, mode);
770 		if (extsize == 0) {
771 			skipfile();
772 		} else {
773 			buf = setupextattr(extsize);
774 			getfile(xtrnull, xtrattr, xtrnull);
775 			set_extattr(-1, name, buf, extsize, SXA_FILE);
776 		}
777 		if (setbirth)
778 			(void) utimens(name, ctimep);
779 		(void) utimens(name, mtimep);
780 		if (Mtreefile) {
781 			writemtree(name, "fifo",
782 			    uid, gid, mode, flags);
783 		} else
784 			(void) chflags(name, flags);
785 		return (GOOD);
786 
787 	case IFREG:
788 		vprintf(stdout, "extract file %s\n", name);
789 		if (uflag)
790 			(void) unlink(name);
791 		if (!Nflag && (ofile = open(name, O_WRONLY | O_CREAT | O_TRUNC,
792 		    0600)) < 0) {
793 			fprintf(stderr, "%s: cannot create file: %s\n",
794 			    name, strerror(errno));
795 			skipfile();
796 			return (FAIL);
797 		}
798 		if (Dflag)
799 			(*ddesc->dd_init)(&dcontext);
800 		(void) fchown(ofile, uid, gid);
801 		(void) fchmod(ofile, mode);
802 		buf = setupextattr(extsize);
803 		getfile(xtrfile, xtrattr, xtrskip);
804 		if (extsize > 0)
805 			set_extattr(ofile, name, buf, extsize, SXA_FD);
806 		if (Dflag) {
807 			(*ddesc->dd_end)(&dcontext, dbuffer);
808 			for (ep = lookupname(name); ep != NULL;
809 			    ep = ep->e_links)
810 				fprintf(stdout, "%s (%s) = %s\n",
811 				    ddesc->dd_name, myname(ep),
812 				    dbuffer);
813 		}
814 		if (Nflag)
815 			return (GOOD);
816 		if (setbirth)
817 			(void) futimens(ofile, ctimep);
818 		(void) futimens(ofile, mtimep);
819 		if (Mtreefile) {
820 			writemtree(name, "file",
821 			    uid, gid, mode, flags);
822 		} else
823 			(void) fchflags(ofile, flags);
824 		(void) close(ofile);
825 		return (GOOD);
826 	}
827 	/* NOTREACHED */
828 }
829 
830 /*
831  * Set attributes on a file descriptor, link, or file.
832  */
833 void
set_extattr(int fd,char * name,void * buf,int size,enum set_extattr_mode mode)834 set_extattr(int fd, char *name, void *buf, int size, enum set_extattr_mode mode)
835 {
836 	struct extattr *eap, *eaend;
837 	const char *method;
838 	ssize_t res;
839 	int error;
840 	char eaname[EXTATTR_MAXNAMELEN + 1];
841 
842 	vprintf(stdout, "Set attributes for %s:", name);
843 	eaend = (void *)((char *)buf + size);
844 	for (eap = buf; eap < eaend; eap = EXTATTR_NEXT(eap)) {
845 		/*
846 		 * Make sure this entry is complete.
847 		 */
848 		if (EXTATTR_NEXT(eap) > eaend || eap->ea_length <= 0) {
849 			dprintf(stdout, "\n\t%scorrupted",
850 				eap == buf ? "" : "remainder ");
851 			break;
852 		}
853 		if (eap->ea_namespace == EXTATTR_NAMESPACE_EMPTY)
854 			continue;
855 		snprintf(eaname, sizeof(eaname), "%.*s",
856 		    (int)eap->ea_namelength, eap->ea_name);
857 		vprintf(stdout, "\n\t%s, (%d bytes), %s",
858 			namespace_names[eap->ea_namespace], eap->ea_length,
859 			eaname);
860 		/*
861 		 * First we try the general attribute setting interface.
862 		 * However, some attributes can only be set by root or
863 		 * by using special interfaces (for example, ACLs).
864 		 */
865 		switch (mode) {
866 		case SXA_FD:
867 			res = extattr_set_fd(fd, eap->ea_namespace,
868 			    eaname, EXTATTR_CONTENT(eap),
869 			    EXTATTR_CONTENT_SIZE(eap));
870 			method = "extattr_set_fd";
871 			break;
872 		case SXA_LINK:
873 			res = extattr_set_link(name, eap->ea_namespace,
874 			    eaname, EXTATTR_CONTENT(eap),
875 			    EXTATTR_CONTENT_SIZE(eap));
876 			method = "extattr_set_link";
877 			break;
878 		case SXA_FILE:
879 			res = extattr_set_file(name, eap->ea_namespace,
880 			    eaname, EXTATTR_CONTENT(eap),
881 			    EXTATTR_CONTENT_SIZE(eap));
882 			method = "extattr_set_file";
883 			break;
884 		default:
885 			abort();
886 		}
887 		if (res != -1) {
888 			dprintf(stdout, " (set using %s)", method);
889 			continue;
890 		}
891 		/*
892 		 * If the general interface refuses to set the attribute,
893 		 * then we try all the specialized interfaces that we
894 		 * know about.
895 		 */
896 		if (eap->ea_namespace == EXTATTR_NAMESPACE_SYSTEM &&
897 		    strcmp(eaname, POSIX1E_ACL_ACCESS_EXTATTR_NAME) == 0) {
898 			switch (mode) {
899 			case SXA_FD:
900 				error = acl_set_fd(fd, EXTATTR_CONTENT(eap));
901 				method = "acl_set_fd";
902 				break;
903 			case SXA_LINK:
904 				error = acl_set_link_np(name, ACL_TYPE_ACCESS,
905 				    EXTATTR_CONTENT(eap));
906 				method = "acl_set_link_np";
907 				break;
908 			case SXA_FILE:
909 				error = acl_set_file(name, ACL_TYPE_ACCESS,
910 				    EXTATTR_CONTENT(eap));
911 				method = "acl_set_file";
912 				break;
913 			default:
914 				abort();
915 			}
916 			if (error != -1) {
917 				dprintf(stdout, " (set using %s)", method);
918 				continue;
919 			}
920 		}
921 		if (eap->ea_namespace == EXTATTR_NAMESPACE_SYSTEM &&
922 		    strcmp(eaname, POSIX1E_ACL_DEFAULT_EXTATTR_NAME) == 0) {
923 			switch (mode) {
924 			case SXA_FD:
925 				error = acl_set_fd(fd, EXTATTR_CONTENT(eap));
926 				method = "acl_set_fd";
927 				break;
928 			case SXA_LINK:
929 				error = acl_set_link_np(name, ACL_TYPE_DEFAULT,
930 				    EXTATTR_CONTENT(eap));
931 				method = "acl_set_link_np";
932 				break;
933 			case SXA_FILE:
934 				error = acl_set_file(name, ACL_TYPE_DEFAULT,
935 				    EXTATTR_CONTENT(eap));
936 				method = "acl_set_file";
937 				break;
938 			default:
939 				abort();
940 			}
941 			if (error != -1) {
942 				dprintf(stdout, " (set using %s)", method);
943 				continue;
944 			}
945 		}
946 		vprintf(stdout, " (unable to set)");
947 	}
948 	vprintf(stdout, "\n");
949 }
950 
951 /*
952  * skip over bit maps on the tape
953  */
954 void
skipmaps(void)955 skipmaps(void)
956 {
957 
958 	while (spcl.c_type == TS_BITS || spcl.c_type == TS_CLRI)
959 		skipfile();
960 }
961 
962 /*
963  * skip over a file on the tape
964  */
965 void
skipfile(void)966 skipfile(void)
967 {
968 
969 	curfile.action = SKIP;
970 	getfile(xtrnull, xtrnull, xtrnull);
971 }
972 
973 /*
974  * Skip a hole in an output file
975  */
976 static void
skiphole(void (* skip)(char *,size_t),volatile size_t * seekpos)977 skiphole(void (*skip)(char *, size_t), volatile size_t *seekpos)
978 {
979 	char buf[MAXBSIZE];
980 	size_t s = *seekpos;
981 
982 	if (s > 0) {
983 		(*skip)(buf, s);
984 		*seekpos = 0;
985 	}
986 }
987 
988 /*
989  * Extract a bitmap from the tape.
990  * The first bitmap sets maxino;
991  * other bitmaps must be of same size.
992  */
993 void
getbitmap(char ** map)994 getbitmap(char **map)
995 {
996 	int i;
997 	size_t volatile size = spcl.c_size;
998 	size_t volatile mapsize = size;
999 	char *mapptr;
1000 
1001 	curfile.action = USING;
1002 	if (spcl.c_type == TS_END)
1003 		panic("ran off end of tape\n");
1004 	if (spcl.c_magic != FS_UFS2_MAGIC)
1005 		panic("not at beginning of a file\n");
1006 	if (!gettingfile && setjmp(restart) != 0)
1007 		return;
1008 	gettingfile++;
1009 	mapptr = *map = malloc(size);
1010 loop:
1011 	if (*map == NULL)
1012 		panic("no memory for %s\n", curfile.name);
1013 	for (i = 0; i < spcl.c_count && size >= TP_BSIZE; i++) {
1014 		readtape(mapptr);
1015 		mapptr += TP_BSIZE;
1016 		size -= TP_BSIZE;
1017 	}
1018 	if (size != 0 || i != spcl.c_count)
1019 		panic("%s: inconsistent map size\n", curfile.name);
1020 	if (gethead(&spcl) == GOOD && spcl.c_type == TS_ADDR) {
1021 		size = spcl.c_count * TP_BSIZE;
1022 		*map = realloc(*map, mapsize + size);
1023 		mapptr = *map + mapsize;
1024 		mapsize += size;
1025 		goto loop;
1026 	}
1027 	if (maxino == 0)
1028 		maxino = mapsize * NBBY + 1;
1029 	else if (maxino != mapsize * NBBY + 1)
1030 		panic("%s: map size changed\n", curfile.name);
1031 	findinode(&spcl);
1032 	gettingfile = 0;
1033 }
1034 
1035 /*
1036  * Extract a file from the tape.
1037  * When an allocated block is found it is passed to the fill function;
1038  * when an unallocated block (hole) is found, a zeroed buffer is passed
1039  * to the skip function.
1040  */
1041 void
getfile(void (* datafill)(char *,size_t),void (* attrfill)(char *,size_t),void (* skip)(char *,size_t))1042 getfile(void (*datafill)(char *, size_t), void (*attrfill)(char *, size_t),
1043     void (*skip)(char *, size_t))
1044 {
1045 	int i;
1046 	volatile off_t size;
1047 	volatile size_t seekpos;
1048 	volatile int curblk, attrsize;
1049 	void (*fillit)(char *, size_t);
1050 	char buf[MAXBSIZE / TP_BSIZE][TP_BSIZE];
1051 	char junk[TP_BSIZE];
1052 
1053 	curblk = 0;
1054 	size = spcl.c_size;
1055 	seekpos = 0;
1056 	attrsize = spcl.c_extsize;
1057 
1058 	if (spcl.c_type == TS_END)
1059 		panic("ran off end of tape\n");
1060 	if (spcl.c_magic != FS_UFS2_MAGIC)
1061 		panic("not at beginning of a file\n");
1062 	if (!gettingfile && setjmp(restart) != 0)
1063 		return;
1064 	gettingfile++;
1065 	fillit = datafill;
1066 	if (size == 0 && attrsize > 0) {
1067 		fillit = attrfill;
1068 		size = attrsize;
1069 		attrsize = 0;
1070 	}
1071 loop:
1072 	for (i = 0; i < spcl.c_count; i++) {
1073 		if (spcl.c_addr[i]) {
1074 			readtape(&buf[curblk++][0]);
1075 			if ((uint32_t)curblk == fssize / TP_BSIZE) {
1076 				skiphole(skip, &seekpos);
1077 				(*fillit)((char *)buf, (long)(size > TP_BSIZE ?
1078 				     fssize : (curblk - 1) * TP_BSIZE + size));
1079 				curblk = 0;
1080 			}
1081 		} else {
1082 			if (curblk > 0) {
1083 				skiphole(skip, &seekpos);
1084 				(*fillit)((char *)buf, (long)(size > TP_BSIZE ?
1085 				     curblk * TP_BSIZE :
1086 				     (curblk - 1) * TP_BSIZE + size));
1087 				curblk = 0;
1088 			}
1089 			/*
1090 			 * We have a block of a hole. Don't skip it
1091 			 * now, because there may be next adjacent
1092 			 * block of the hole in the file. Postpone the
1093 			 * seek until next file write.
1094 			 */
1095 			seekpos += (long)MIN(TP_BSIZE, size);
1096 		}
1097 		if ((size -= TP_BSIZE) <= 0) {
1098 			if (size > -TP_BSIZE && curblk > 0) {
1099 				skiphole(skip, &seekpos);
1100 				(*fillit)((char *)buf,
1101 					(long)((curblk * TP_BSIZE) + size));
1102 				curblk = 0;
1103 			}
1104 			if (attrsize > 0) {
1105 				fillit = attrfill;
1106 				size = attrsize;
1107 				attrsize = 0;
1108 				continue;
1109 			}
1110 			if (spcl.c_count - i > 1)
1111 				dprintf(stdout, "skipping %d junk block(s)\n",
1112 					spcl.c_count - i - 1);
1113 			for (i++; i < spcl.c_count; i++)
1114 				if (spcl.c_addr[i])
1115 					readtape(junk);
1116 			break;
1117 		}
1118 	}
1119 	if (gethead(&spcl) == GOOD && size > 0) {
1120 		if (spcl.c_type == TS_ADDR)
1121 			goto loop;
1122 		dprintf(stdout,
1123 			"Missing address (header) block for %s at %d blocks\n",
1124 			curfile.name, blksread);
1125 	}
1126 	if (curblk > 0)
1127 		panic("getfile: lost data: %s\n", curfile.name);
1128 	/* Skip over Linux extended attributes. */
1129 	if (spcl.c_type == TS_INODE && (spcl.c_flags & DR_EXTATTRIBUTES)) {
1130 		for (i = 0; i < spcl.c_count; i++)
1131 			readtape(junk);
1132 		(void)gethead(&spcl);
1133 	}
1134 	findinode(&spcl);
1135 	gettingfile = 0;
1136 }
1137 
1138 /*
1139  * These variables are shared between the next two functions.
1140  */
1141 static size_t extbufsize = 0;
1142 static char *extbuf;
1143 static size_t extloc;
1144 
1145 /*
1146  * Allocate a buffer into which to extract extended attributes.
1147  */
1148 static char *
setupextattr(size_t extsize)1149 setupextattr(size_t extsize)
1150 {
1151 
1152 	extloc = 0;
1153 	if (extsize <= extbufsize)
1154 		return (extbuf);
1155 	if (extbufsize > 0)
1156 		free(extbuf);
1157 	if ((extbuf = malloc(extsize)) != NULL) {
1158 		extbufsize = extsize;
1159 		return (extbuf);
1160 	}
1161 	extbufsize = 0;
1162 	extbuf = NULL;
1163 	fprintf(stderr, "Cannot extract %zu bytes %s for inode %ju, name %s\n",
1164 	    extsize, "of extended attributes", (uintmax_t)curfile.ino,
1165 	    curfile.name);
1166 	return (NULL);
1167 }
1168 
1169 /*
1170  * Extract the next block of extended attributes.
1171  */
1172 static void
xtrattr(char * buf,size_t size)1173 xtrattr(char *buf, size_t size)
1174 {
1175 
1176 	if (extloc + size > extbufsize)
1177 		panic("overrun attribute buffer\n");
1178 	memmove(&extbuf[extloc], buf, size);
1179 	extloc += size;
1180 }
1181 
1182 /*
1183  * Write out the next block of a file.
1184  */
1185 static void
xtrfile(char * buf,size_t size)1186 xtrfile(char *buf, size_t size)
1187 {
1188 
1189 	if (Dflag)
1190 		(*ddesc->dd_update)(&dcontext, buf, size);
1191 	if (Nflag)
1192 		return;
1193 	if (write(ofile, buf, (int) size) == -1) {
1194 		fprintf(stderr,
1195 		    "write error extracting inode %ju, name %s\nwrite: %s\n",
1196 			(uintmax_t)curfile.ino, curfile.name,
1197 			strerror(errno));
1198 		exit(1);
1199 	}
1200 }
1201 
1202 /*
1203  * Skip over a hole in a file.
1204  */
1205 /* ARGSUSED */
1206 static void
xtrskip(char * buf,size_t size)1207 xtrskip(char *buf, size_t size)
1208 {
1209 
1210 	if (Dflag)
1211 		(*ddesc->dd_update)(&dcontext, buf, size);
1212 	if (Nflag)
1213 		return;
1214 	if (lseek(ofile, size, SEEK_CUR) == -1) {
1215 		fprintf(stderr,
1216 		    "seek error extracting inode %ju, name %s\nlseek: %s\n",
1217 			(uintmax_t)curfile.ino, curfile.name,
1218 			strerror(errno));
1219 		exit(1);
1220 	}
1221 }
1222 
1223 /*
1224  * Collect the next block of a symbolic link.
1225  */
1226 static void
xtrlnkfile(char * buf,size_t size)1227 xtrlnkfile(char *buf, size_t size)
1228 {
1229 
1230 	pathlen += size;
1231 	if (pathlen > MAXPATHLEN) {
1232 		fprintf(stderr, "symbolic link name: %s->%s%s; too long %d\n",
1233 		    curfile.name, lnkbuf, buf, pathlen);
1234 		exit(1);
1235 	}
1236 	(void) strcat(lnkbuf, buf);
1237 }
1238 
1239 /*
1240  * Skip over a hole in a symbolic link (should never happen).
1241  */
1242 /* ARGSUSED */
1243 static void
xtrlnkskip(char * buf __unused,size_t size __unused)1244 xtrlnkskip(char *buf __unused, size_t size __unused)
1245 {
1246 
1247 	fprintf(stderr, "unallocated block in symbolic link %s\n",
1248 		curfile.name);
1249 	exit(1);
1250 }
1251 
1252 /*
1253  * Noop, when an extraction function is not needed.
1254  */
1255 /* ARGSUSED */
1256 void
xtrnull(char * buf __unused,size_t size __unused)1257 xtrnull(char *buf __unused, size_t size __unused)
1258 {
1259 
1260 	return;
1261 }
1262 
1263 /*
1264  * Read TP_BSIZE blocks from the input.
1265  * Handle read errors, and end of media.
1266  */
1267 static void
readtape(char * buf)1268 readtape(char *buf)
1269 {
1270 	int rd, newvol, i;
1271 	int cnt, seek_failed;
1272 
1273 	if (blkcnt < numtrec) {
1274 		memmove(buf, &tapebuf[(blkcnt++ * TP_BSIZE)], (long)TP_BSIZE);
1275 		blksread++;
1276 		tpblksread++;
1277 		return;
1278 	}
1279 	for (i = 0; i < ntrec; i++)
1280 		((struct s_spcl *)&tapebuf[i * TP_BSIZE])->c_magic = 0;
1281 	if (numtrec == 0)
1282 		numtrec = ntrec;
1283 	cnt = ntrec * TP_BSIZE;
1284 	rd = 0;
1285 getmore:
1286 #ifdef RRESTORE
1287 	if (host)
1288 		i = rmtread(&tapebuf[rd], cnt);
1289 	else
1290 #endif
1291 		i = read(mt, &tapebuf[rd], cnt);
1292 	/*
1293 	 * Check for mid-tape short read error.
1294 	 * If found, skip rest of buffer and start with the next.
1295 	 */
1296 	if (!pipein && numtrec < ntrec && i > 0) {
1297 		dprintf(stdout, "mid-media short read error.\n");
1298 		numtrec = ntrec;
1299 	}
1300 	/*
1301 	 * Handle partial block read.
1302 	 */
1303 	if (pipein && i == 0 && rd > 0)
1304 		i = rd;
1305 	else if (i > 0 && i != ntrec * TP_BSIZE) {
1306 		if (pipein) {
1307 			rd += i;
1308 			cnt -= i;
1309 			if (cnt > 0)
1310 				goto getmore;
1311 			i = rd;
1312 		} else {
1313 			/*
1314 			 * Short read. Process the blocks read.
1315 			 */
1316 			if (i % TP_BSIZE != 0)
1317 				vprintf(stdout,
1318 				    "partial block read: %d should be %d\n",
1319 				    i, ntrec * TP_BSIZE);
1320 			numtrec = i / TP_BSIZE;
1321 		}
1322 	}
1323 	/*
1324 	 * Handle read error.
1325 	 */
1326 	if (i < 0) {
1327 		fprintf(stderr, "Tape read error while ");
1328 		switch (curfile.action) {
1329 		default:
1330 			fprintf(stderr, "trying to set up tape\n");
1331 			break;
1332 		case UNKNOWN:
1333 			fprintf(stderr, "trying to resynchronize\n");
1334 			break;
1335 		case USING:
1336 			fprintf(stderr, "restoring %s\n", curfile.name);
1337 			break;
1338 		case SKIP:
1339 			fprintf(stderr, "skipping over inode %ju\n",
1340 			    (uintmax_t)curfile.ino);
1341 			break;
1342 		}
1343 		if (!yflag && !reply("continue"))
1344 			exit(1);
1345 		i = ntrec * TP_BSIZE;
1346 		memset(tapebuf, 0, i);
1347 #ifdef RRESTORE
1348 		if (host)
1349 			seek_failed = (rmtseek(i, 1) < 0);
1350 		else
1351 #endif
1352 			seek_failed = (lseek(mt, i, SEEK_CUR) == (off_t)-1);
1353 
1354 		if (seek_failed) {
1355 			fprintf(stderr,
1356 			    "continuation failed: %s\n", strerror(errno));
1357 			exit(1);
1358 		}
1359 	}
1360 	/*
1361 	 * Handle end of tape.
1362 	 */
1363 	if (i == 0) {
1364 		vprintf(stdout, "End-of-tape encountered\n");
1365 		if (!pipein) {
1366 			newvol = volno + 1;
1367 			volno = 0;
1368 			numtrec = 0;
1369 			getvol(newvol);
1370 			readtape(buf);
1371 			return;
1372 		}
1373 		if (rd % TP_BSIZE != 0)
1374 			panic("partial block read: %d should be %d\n",
1375 				rd, ntrec * TP_BSIZE);
1376 		terminateinput();
1377 		memmove(&tapebuf[rd], &endoftapemark, (long)TP_BSIZE);
1378 	}
1379 	blkcnt = 0;
1380 	memmove(buf, &tapebuf[(blkcnt++ * TP_BSIZE)], (long)TP_BSIZE);
1381 	blksread++;
1382 	tpblksread++;
1383 }
1384 
1385 static void
findtapeblksize(void)1386 findtapeblksize(void)
1387 {
1388 	long i;
1389 
1390 	for (i = 0; i < ntrec; i++)
1391 		((struct s_spcl *)&tapebuf[i * TP_BSIZE])->c_magic = 0;
1392 	blkcnt = 0;
1393 #ifdef RRESTORE
1394 	if (host)
1395 		i = rmtread(tapebuf, ntrec * TP_BSIZE);
1396 	else
1397 #endif
1398 		i = read(mt, tapebuf, ntrec * TP_BSIZE);
1399 
1400 	if (i <= 0) {
1401 		fprintf(stderr, "tape read error: %s\n", strerror(errno));
1402 		exit(1);
1403 	}
1404 	if (i % TP_BSIZE != 0) {
1405 		fprintf(stderr, "Tape block size (%ld) %s (%ld)\n",
1406 			(long)i, "is not a multiple of dump block size",
1407 			(long)TP_BSIZE);
1408 		exit(1);
1409 	}
1410 	ntrec = i / TP_BSIZE;
1411 	numtrec = ntrec;
1412 	vprintf(stdout, "Tape block size is %d\n", ntrec);
1413 }
1414 
1415 void
closemt(void)1416 closemt(void)
1417 {
1418 
1419 	if (mt < 0)
1420 		return;
1421 #ifdef RRESTORE
1422 	if (host)
1423 		rmtclose();
1424 	else
1425 #endif
1426 		(void) close(mt);
1427 }
1428 
1429 /*
1430  * Read the next block from the tape.
1431  * Check to see if it is one of several vintage headers.
1432  * If it is an old style header, convert it to a new style header.
1433  * If it is not any valid header, return an error.
1434  */
1435 static int
gethead(struct s_spcl * buf)1436 gethead(struct s_spcl *buf)
1437 {
1438 	union u_ospcl u_ospcl;
1439 
1440 	if (!cvtflag) {
1441 		readtape((char *)buf);
1442 		if (buf->c_magic != NFS_MAGIC &&
1443 		    buf->c_magic != FS_UFS2_MAGIC) {
1444 			if (bswap32(buf->c_magic) != NFS_MAGIC &&
1445 			    bswap32(buf->c_magic) != FS_UFS2_MAGIC)
1446 				return (FAIL);
1447 			if (!Bcvt) {
1448 				vprintf(stdout, "Note: Doing Byte swapping\n");
1449 				Bcvt = 1;
1450 			}
1451 		}
1452 		if (checksum((int *)buf) == FAIL)
1453 			return (FAIL);
1454 		if (Bcvt)
1455 			swap_header(buf);
1456 		goto good;
1457 	}
1458 
1459 	readtape((char *)(&u_ospcl.s_ospcl));
1460 	if (checksum((int *)(&u_ospcl.s_ospcl)) == FAIL)
1461 		return (FAIL);
1462 	if (u_ospcl.s_ospcl.c_magic != OFS_MAGIC) {
1463 		if (bswap32(u_ospcl.s_ospcl.c_magic) != OFS_MAGIC)
1464 			return (FAIL);
1465 		if (!Bcvt) {
1466 			vprintf(stdout, "Note: Doing Byte swapping\n");
1467 			Bcvt = 1;
1468 		}
1469 		swap_old_header(&u_ospcl.s_ospcl);
1470 	}
1471 
1472 	memset(buf, 0, TP_BSIZE);
1473 	buf->c_type = u_ospcl.s_ospcl.c_type;
1474 	buf->c_date = u_ospcl.s_ospcl.c_date;
1475 	buf->c_ddate = u_ospcl.s_ospcl.c_ddate;
1476 	buf->c_volume = u_ospcl.s_ospcl.c_volume;
1477 	buf->c_tapea = u_ospcl.s_ospcl.c_tapea;
1478 	buf->c_inumber = u_ospcl.s_ospcl.c_inumber;
1479 	buf->c_checksum = u_ospcl.s_ospcl.c_checksum;
1480 	buf->c_mode = u_ospcl.s_ospcl.c_odinode.odi_mode;
1481 	buf->c_uid = u_ospcl.s_ospcl.c_odinode.odi_uid;
1482 	buf->c_gid = u_ospcl.s_ospcl.c_odinode.odi_gid;
1483 	buf->c_size = u_ospcl.s_ospcl.c_odinode.odi_size;
1484 	buf->c_rdev = u_ospcl.s_ospcl.c_odinode.odi_rdev;
1485 	buf->c_atime = u_ospcl.s_ospcl.c_odinode.odi_atime;
1486 	buf->c_mtime = u_ospcl.s_ospcl.c_odinode.odi_mtime;
1487 	buf->c_count = u_ospcl.s_ospcl.c_count;
1488 	memmove(buf->c_addr, u_ospcl.s_ospcl.c_addr, (long)256);
1489 	buf->c_magic = FS_UFS2_MAGIC;
1490 good:
1491 	switch (buf->c_type) {
1492 
1493 	case TS_CLRI:
1494 	case TS_BITS:
1495 		/*
1496 		 * Have to patch up missing information in bit map headers
1497 		 */
1498 		buf->c_inumber = 0;
1499 		buf->c_size = buf->c_count * TP_BSIZE;
1500 		break;
1501 
1502 	case TS_TAPE:
1503 		if ((buf->c_flags & DR_NEWINODEFMT) == 0)
1504 			oldinofmt = 1;
1505 		/* fall through */
1506 	case TS_END:
1507 		buf->c_inumber = 0;
1508 		break;
1509 
1510 	case TS_INODE:
1511 		if (buf->c_magic == NFS_MAGIC) {
1512 			buf->c_tapea = buf->c_old_tapea;
1513 			buf->c_firstrec = buf->c_old_firstrec;
1514 			buf->c_date = buf->c_old_date;
1515 			buf->c_ddate = buf->c_old_ddate;
1516 			buf->c_atime = buf->c_old_atime;
1517 			buf->c_mtime = buf->c_old_mtime;
1518 			buf->c_birthtime = 0;
1519 			buf->c_birthtimensec = 0;
1520 			buf->c_atimensec = buf->c_mtimensec = 0;
1521 			buf->c_extsize = 0;
1522 		}
1523 
1524 	case TS_ADDR:
1525 		break;
1526 
1527 	default:
1528 		panic("gethead: unknown inode type %d\n", buf->c_type);
1529 		break;
1530 	}
1531 
1532 	buf->c_magic = FS_UFS2_MAGIC;
1533 
1534 	/*
1535 	 * If we are restoring a filesystem with old format inodes,
1536 	 * copy the uid/gid to the new location.
1537 	 */
1538 	if (oldinofmt) {
1539 		buf->c_uid = buf->c_spare1[1];
1540 		buf->c_gid = buf->c_spare1[2];
1541 	}
1542 	if (dflag)
1543 		accthdr(buf);
1544 	return(GOOD);
1545 }
1546 
1547 /*
1548  * Check that a header is where it belongs and predict the next header
1549  */
1550 static void
accthdr(struct s_spcl * header)1551 accthdr(struct s_spcl *header)
1552 {
1553 	static ino_t previno = 0x7fffffff;
1554 	static int prevtype;
1555 	static long predict;
1556 	long blks, i;
1557 
1558 	if (header->c_type == TS_TAPE) {
1559 		fprintf(stderr, "Volume header (%s inode format) ",
1560 		    oldinofmt ? "old" : "new");
1561  		if (header->c_firstrec)
1562  			fprintf(stderr, "begins with record %lld",
1563  				(long long)header->c_firstrec);
1564  		fprintf(stderr, "\n");
1565 		previno = 0x7fffffff;
1566 		return;
1567 	}
1568 	if (previno == 0x7fffffff)
1569 		goto newcalc;
1570 	switch (prevtype) {
1571 	case TS_BITS:
1572 		fprintf(stderr, "Dumped inodes map header");
1573 		break;
1574 	case TS_CLRI:
1575 		fprintf(stderr, "Used inodes map header");
1576 		break;
1577 	case TS_INODE:
1578 		fprintf(stderr, "File header, ino %ju",
1579 		    (uintmax_t)previno);
1580 		break;
1581 	case TS_ADDR:
1582 		fprintf(stderr, "File continuation header, ino %ju",
1583 		    (uintmax_t)previno);
1584 		break;
1585 	case TS_END:
1586 		fprintf(stderr, "End of tape header");
1587 		break;
1588 	}
1589 	if (predict != blksread - 1)
1590 		fprintf(stderr, "; predicted %ld blocks, got %ld blocks",
1591 			(long)predict, (long)(blksread - 1));
1592 	fprintf(stderr, "\n");
1593 newcalc:
1594 	blks = 0;
1595 	switch (header->c_type) {
1596 	case TS_END:
1597 		break;
1598 	case TS_CLRI:
1599 	case TS_BITS:
1600 		blks = header->c_count;
1601 		break;
1602 	default:
1603 		for (i = 0; i < header->c_count; i++)
1604 			if (header->c_addr[i] != 0)
1605 				blks++;
1606 		break;
1607 	}
1608 	predict = blks;
1609 	blksread = 0;
1610 	prevtype = header->c_type;
1611 	previno = header->c_inumber;
1612 }
1613 
1614 /*
1615  * Find an inode header.
1616  * Complain if had to skip, and complain is set.
1617  */
1618 static void
findinode(struct s_spcl * header)1619 findinode(struct s_spcl *header)
1620 {
1621 	static long skipcnt = 0;
1622 	long i;
1623 	char buf[TP_BSIZE];
1624 
1625 	curfile.name = "<name unknown>";
1626 	curfile.action = UNKNOWN;
1627 	curfile.mode = 0;
1628 	curfile.ino = 0;
1629     top:
1630 	do {
1631 		if (header->c_magic != FS_UFS2_MAGIC) {
1632 skip:
1633 			skipcnt++;
1634 			while (gethead(header) == FAIL ||
1635 			    header->c_date != dumpdate)
1636 				skipcnt++;
1637 		}
1638 		switch (header->c_type) {
1639 
1640 		case TS_ADDR:
1641 			/*
1642 			 * Skip up to the beginning of the next record
1643 			 */
1644 			for (i = 0; i < header->c_count; i++)
1645 				if (header->c_addr[i])
1646 					readtape(buf);
1647 			while (gethead(header) == FAIL ||
1648 			    header->c_date != dumpdate)
1649 				skipcnt++;
1650 			/* We've read a header; don't drop it. */
1651 			goto top;
1652 
1653 		case TS_INODE:
1654 			curfile.mode = header->c_mode;
1655 			curfile.uid = header->c_uid;
1656 			curfile.gid = header->c_gid;
1657 			curfile.file_flags = header->c_file_flags;
1658 			curfile.rdev = header->c_rdev;
1659 			curfile.atime_sec = header->c_atime;
1660 			curfile.atime_nsec = header->c_atimensec;
1661 			curfile.mtime_sec = header->c_mtime;
1662 			curfile.mtime_nsec = header->c_mtimensec;
1663 			curfile.birthtime_sec = header->c_birthtime;
1664 			curfile.birthtime_nsec = header->c_birthtimensec;
1665 			curfile.extsize = header->c_extsize;
1666 			curfile.size = header->c_size;
1667 			curfile.ino = header->c_inumber;
1668 			break;
1669 
1670 		case TS_END:
1671 			curfile.ino = maxino;
1672 			break;
1673 
1674 		case TS_CLRI:
1675 			curfile.name = "<file removal list>";
1676 			break;
1677 
1678 		case TS_BITS:
1679 			curfile.name = "<file dump list>";
1680 			break;
1681 
1682 		case TS_TAPE:
1683 			panic("unexpected tape header\n");
1684 			break;
1685 
1686 		default:
1687 			panic("unknown tape header type %d\n", spcl.c_type);
1688 			fprintf(stderr, "skipping to next header\n");
1689 			goto skip;
1690 
1691 		}
1692 	} while (header->c_type == TS_ADDR);
1693 	if (skipcnt > 0)
1694 		fprintf(stderr, "resync restore, skipped %ld blocks\n",
1695 		    (long)skipcnt);
1696 	skipcnt = 0;
1697 }
1698 
1699 static int
checksum(int * buf)1700 checksum(int *buf)
1701 {
1702 	int i, j;
1703 
1704 	j = sizeof(union u_spcl) / sizeof(int);
1705 	i = 0;
1706 	if(!Bcvt) {
1707 		do
1708 			i += *buf++;
1709 		while (--j);
1710 	} else {
1711 		do
1712 			i += bswap32(*buf++);
1713 		while (--j);
1714 	}
1715 
1716 	if (i != CHECKSUM) {
1717 		fprintf(stderr, "Checksum error %o, inode %ju file %s\n", i,
1718 		    (uintmax_t)curfile.ino, curfile.name);
1719 		return(FAIL);
1720 	}
1721 	return(GOOD);
1722 }
1723 
1724 #ifdef RRESTORE
1725 #include <stdarg.h>
1726 
1727 void
msg(const char * fmt,...)1728 msg(const char *fmt, ...)
1729 {
1730 	va_list ap;
1731 
1732 	va_start(ap, fmt);
1733 	(void)vfprintf(stderr, fmt, ap);
1734 	va_end(ap);
1735 }
1736 #endif /* RRESTORE */
1737 
1738 static void
swap_header(struct s_spcl * s)1739 swap_header(struct s_spcl *s)
1740 {
1741 	s->c_type = bswap32(s->c_type);
1742 	s->c_old_date = bswap32(s->c_old_date);
1743 	s->c_old_ddate = bswap32(s->c_old_ddate);
1744 	s->c_volume = bswap32(s->c_volume);
1745 	s->c_old_tapea = bswap32(s->c_old_tapea);
1746 	s->c_inumber = bswap32(s->c_inumber);
1747 	s->c_magic = bswap32(s->c_magic);
1748 	s->c_checksum = bswap32(s->c_checksum);
1749 
1750 	s->c_mode = bswap16(s->c_mode);
1751 	s->c_extsize = bswap64(s->c_extsize);
1752 	s->c_size = bswap64(s->c_size);
1753 	s->c_old_atime = bswap32(s->c_old_atime);
1754 	s->c_atimensec = bswap32(s->c_atimensec);
1755 	s->c_old_mtime = bswap32(s->c_old_mtime);
1756 	s->c_mtimensec = bswap32(s->c_mtimensec);
1757 	s->c_rdev = bswap32(s->c_rdev);
1758 	s->c_birthtimensec = bswap32(s->c_birthtimensec);
1759 	s->c_birthtime = bswap64(s->c_birthtime);
1760 	s->c_atime = bswap64(s->c_atime);
1761 	s->c_mtime = bswap64(s->c_mtime);
1762 	s->c_file_flags = bswap32(s->c_file_flags);
1763 	s->c_uid = bswap32(s->c_uid);
1764 	s->c_gid = bswap32(s->c_gid);
1765 
1766 	s->c_count = bswap32(s->c_count);
1767 	s->c_level = bswap32(s->c_level);
1768 	s->c_flags = bswap32(s->c_flags);
1769 	s->c_old_firstrec = bswap32(s->c_old_firstrec);
1770 
1771 	s->c_date = bswap64(s->c_date);
1772 	s->c_ddate = bswap64(s->c_ddate);
1773 	s->c_tapea = bswap64(s->c_tapea);
1774 	s->c_firstrec = bswap64(s->c_firstrec);
1775 
1776 	/*
1777 	 * These are ouid and ogid.
1778 	 */
1779 	s->c_spare1[1] = bswap16(s->c_spare1[1]);
1780 	s->c_spare1[2] = bswap16(s->c_spare1[2]);
1781 }
1782 
1783 static void
swap_old_header(struct s_ospcl * os)1784 swap_old_header(struct s_ospcl *os)
1785 {
1786 	os->c_type = bswap32(os->c_type);
1787 	os->c_date = bswap32(os->c_date);
1788 	os->c_ddate = bswap32(os->c_ddate);
1789 	os->c_volume = bswap32(os->c_volume);
1790 	os->c_tapea = bswap32(os->c_tapea);
1791 	os->c_inumber = bswap16(os->c_inumber);
1792 	os->c_magic = bswap32(os->c_magic);
1793 	os->c_checksum = bswap32(os->c_checksum);
1794 
1795 	os->c_odinode.odi_mode = bswap16(os->c_odinode.odi_mode);
1796 	os->c_odinode.odi_nlink = bswap16(os->c_odinode.odi_nlink);
1797 	os->c_odinode.odi_uid = bswap16(os->c_odinode.odi_uid);
1798 	os->c_odinode.odi_gid = bswap16(os->c_odinode.odi_gid);
1799 
1800 	os->c_odinode.odi_size = bswap32(os->c_odinode.odi_size);
1801 	os->c_odinode.odi_rdev = bswap32(os->c_odinode.odi_rdev);
1802 	os->c_odinode.odi_atime = bswap32(os->c_odinode.odi_atime);
1803 	os->c_odinode.odi_mtime = bswap32(os->c_odinode.odi_mtime);
1804 	os->c_odinode.odi_ctime = bswap32(os->c_odinode.odi_ctime);
1805 
1806 	os->c_count = bswap32(os->c_count);
1807 }
1808