X-Git-Url: https://git.wh0rd.org/?p=dump.git;a=blobdiff_plain;f=restore%2Ftape.c;h=553abbc8e99fef89f0d75fc27bdbd2a612451a3c;hp=3195ab53b4746893ca50f62ef3e085980bd3db60;hb=206f768c7a7a4bca1c4acc3c686d55578c0cb919;hpb=ba3af39e8b08027d3338b5533bddc86a9c803a74

diff --git a/restore/tape.c b/restore/tape.c
index 3195ab5..553abbc 100644
--- a/restore/tape.c
+++ b/restore/tape.c
@@ -23,11 +23,7 @@
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
- * 3. All advertising materials mentioning features or use of this software
- *    must display the following acknowledgement:
- *	This product includes software developed by the University of
- *	California, Berkeley and its contributors.
- * 4. Neither the name of the University nor the names of its contributors
+ * 3. Neither the name of the University nor the names of its contributors
  *    may be used to endorse or promote products derived from this software
  *    without specific prior written permission.
  *
@@ -46,7 +42,7 @@
 
 #ifndef lint
 static const char rcsid[] =
-	"$Id: tape.c,v 1.64 2002/07/29 12:00:34 stelian Exp $";
+	"$Id: tape.c,v 1.74 2003/03/31 09:42:59 stelian Exp $";
 #endif /* not lint */
 
 #include <config.h>
@@ -88,6 +84,10 @@ static const char rcsid[] =
 #include <bzlib.h>
 #endif /* HAVE_BZLIB */
 
+#ifdef HAVE_LZO
+#include <minilzo.h>
+#endif /* HAVE_LZO */
+
 #include "restore.h"
 #include "extern.h"
 #include "pathnames.h"
@@ -106,7 +106,7 @@ static int	numtrec;
 static char	*tapebuf;		/* input buffer for read */
 static int	bufsize;		/* buffer size without prefix */
 static char	*tbufptr = NULL;	/* active tape buffer */
-#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB)
+#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
 static char	*comprbuf;		/* uncompress work buf */
 static size_t	comprlen;		/* size including prefix */
 #endif
@@ -134,6 +134,8 @@ static int	 checksum __P((int *));
 static void	 findinode __P((struct s_spcl *));
 static void	 findtapeblksize __P((void));
 static int	 gethead __P((struct s_spcl *));
+static int	 converthead __P((struct s_spcl *));
+static void	 converttapebuf __P((struct tapebuf *));
 static void	 readtape __P((char *));
 static void	 setdumpnum __P((void));
 static u_int	 swabi __P((u_int));
@@ -152,7 +154,7 @@ static void	 xtrmapskip __P((char *, size_t));
 static void	 xtrskip __P((char *, size_t));
 static void	 setmagtapein __P((void));
 
-#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB)
+#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
 static void	newcomprbuf __P((int));
 static void	(*readtape_func) __P((char *));
 static void	readtape_set __P((char *));
@@ -241,7 +243,7 @@ newtapebuf(long size)
 	tapebufsize = size;
 }
 
-#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB)
+#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
 static void
 newcomprbuf(int size)
 {
@@ -255,7 +257,7 @@ newcomprbuf(int size)
 	if (comprbuf == NULL)
 		errx(1, "Cannot allocate space for decompress buffer");
 }
-#endif /* HAVE_ZLIB || HAVE_BZLIB */
+#endif /* HAVE_ZLIB || HAVE_BZLIB || HAVE_LZO */
 
 /*
  * Verify that the tape drive can be accessed and
@@ -284,7 +286,7 @@ setup(void)
 		temptape = magtape;
 
 #ifdef RRESTORE
-	if (host)
+	if (!Afile && host)
 		mt = rmtopen(temptape, O_RDONLY);
 	else
 #endif
@@ -293,14 +295,20 @@ setup(void)
 	else
 		mt = OPEN(temptape, O_RDONLY, 0);
 	if (mt < 0)
-		err(1, "%s", magtape);
+		err(1, "%s", temptape);
 	if (!Afile) {
 		volno = 1;
 		setmagtapein();
 		setdumpnum();
 	}
-#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB)
+#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
 	readtape_func = readtape_set;
+#if defined(HAVE_LZO)
+	if (lzo_init() != LZO_E_OK) {
+	  msg("internal error - lzo_init failed \n");
+	  exit(1);
+        }
+#endif
 #endif
 	FLUSHTAPEBUF();
 	findtapeblksize();
@@ -316,7 +324,7 @@ setup(void)
 
 	if (zflag) {
 		fprintf(stderr, "Dump tape is compressed.\n");
-#if !defined(HAVE_ZLIB) && !defined(HAVE_BZLIB)
+#if !defined(HAVE_ZLIB) && !defined(HAVE_BZLIB) && !defined(HAVE_LZO)
 		errx(1,"This restore version doesn't support decompression");
 #endif /* !HAVE_ZLIB && !HAVE_BZLIB */
 	}
@@ -374,15 +382,9 @@ setup(void)
 	getfile(xtrmap, xtrmapskip);
 	while (spcl.c_type == TS_ADDR) {
 		/* Recompute maxino and the map */
-		char *oldmap = usedinomap;
 		dump_ino_t oldmaxino = maxino;
 		maxino += (spcl.c_count * TP_BSIZE * NBBY) + 1;
-		map = calloc((unsigned)1, (unsigned)howmany(maxino, NBBY));
-		if (map == NULL)
-			errx(1, "no memory for active inode map");
-		usedinomap = map;
-		memcpy(usedinomap, oldmap, howmany(oldmaxino, NBBY));
-		free(oldmap);
+		resizemaps(oldmaxino, maxino);
 
 		spcl.c_dinode.di_size = spcl.c_count * TP_BSIZE;
 		getfile(xtrmap, xtrmapskip);
@@ -506,6 +508,13 @@ again:
 		return;
 	}
 	closemt();
+
+	/* 
+	 * if using an archive file, reset its name so readtape()
+	 * could properly use remote access.
+	 */
+	Afile = NULL;
+
 	if (Mflag) {
 		snprintf(magtape, MAXPATHLEN, "%s%03ld", magtapeprefix, newvol);
 		magtape[MAXPATHLEN - 1] = '\0';
@@ -554,7 +563,7 @@ again:
 	}
 gethdr:
 	setmagtapein();
-#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB)
+#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
 	readtape_func = readtape_set;
 #endif
 	volno = newvol;
@@ -598,7 +607,7 @@ gethdr:
  	 */
 	if (zflag) {
 		fprintf(stderr, "Dump tape is compressed.\n");
-#if !defined(HAVE_ZLIB) && !defined(HAVE_BZLIB)
+#if !defined(HAVE_ZLIB) && !defined(HAVE_BZLIB) && !defined(HAVE_LZO)
 		errx(1,"This restore version doesn't support decompression");
 #endif /* !HAVE_ZLIB && !HAVE_BZLIB */
 	}
@@ -1023,7 +1032,7 @@ static void
 xtrskip(UNUSED(char *buf), size_t size)
 {
 
-	if (LSEEK(ofile, (off_t)size, SEEK_CUR) == -1)
+	if (LSEEK(ofile, (OFF_T)size, SEEK_CUR) == -1)
 		err(1, "seek error extracting inode %lu, name %s\nlseek",
 			(unsigned long)curfile.ino, curfile.name);
 }
@@ -1270,6 +1279,16 @@ comparefile(char *name)
 	struct STAT stemp;
 #endif
 
+	curfile.name = name;
+	curfile.action = USING;
+	mode = curfile.dip->di_mode;
+
+	if ((mode & IFMT) == IFSOCK) {
+		Vprintf(stdout, "skipped socket %s\n", name);
+		skipfile();
+		return;
+	}
+
 	if ((r = LSTAT(name, &sb)) != 0) {
 		warn("%s: does not exist (%d)", name, r);
 		do_compare_error;
@@ -1277,10 +1296,6 @@ comparefile(char *name)
 		return;
 	}
 
-	curfile.name = name;
-	curfile.action = USING;
-	mode = curfile.dip->di_mode;
-
 	Vprintf(stdout, "comparing %s (size: %ld, mode: 0%o)\n", name,
 		(long)sb.st_size, mode);
 
@@ -1416,7 +1431,7 @@ comparefile(char *name)
 	/* NOTREACHED */
 }
 
-#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB)
+#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
 static void (*readtape_func)(char *) = readtape_set;
 
 /*
@@ -1449,7 +1464,7 @@ readtape_set(char *buf)
 	readtape(buf);
 }
 
-#endif /* HAVE_ZLIB || HAVE_BZLIB */
+#endif /* HAVE_ZLIB || HAVE_BZLIB || HAVE_LZO */
 
 /*
  * This is the original readtape(), it's used for reading uncompressed input.
@@ -1457,7 +1472,7 @@ readtape_set(char *buf)
  * Handle read errors, and end of media.
  */
 static void
-#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB)
+#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
 readtape_uncompr(char *buf)
 #else
 readtape(char *buf)
@@ -1485,7 +1500,7 @@ readtape(char *buf)
 #endif
 getmore:
 #ifdef RRESTORE
-	if (host)
+	if (!Afile && host)
 		i = rmtread(&tapebuf[rd], cnt);
 	else
 #endif
@@ -1547,11 +1562,11 @@ getmore:
 		i = ntrec * TP_BSIZE;
 		memset(tapebuf, 0, (size_t)i);
 #ifdef RRESTORE
-		if (host)
+		if (!Afile && host)
 			seek_failed = (rmtseek(i, 1) < 0);
 		else
 #endif
-			seek_failed = (LSEEK(mt, i, SEEK_CUR) == (off_t)-1);
+			seek_failed = (LSEEK(mt, i, SEEK_CUR) == (OFF_T)-1);
 
 		if (seek_failed) {
 			warn("continuation failed");
@@ -1585,7 +1600,7 @@ getmore:
 	tpblksread++;
 }
 
-#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB)
+#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
 
 /*
  * Read a compressed format block from a file or pipe and uncompress it.
@@ -1613,6 +1628,7 @@ readtape_comprfile(char *buf)
 
 	/* read the block prefix */
 	ret = read_a_block(mt, tapebuf, PREFIXSIZE, &rl);
+	converttapebuf(tpb);
 
 	if (Vflag && (ret == 0 || rl < (int)PREFIXSIZE  ||  tpb->length == 0))
 		ret = 0;
@@ -1705,6 +1721,7 @@ readtape_comprtape(char *buf)
 	if (ret <= 0)
 		goto readerr;
 
+	converttapebuf(tpb);
 	tbufptr = decompress_tapebuf(tpb, rl);
 	if (tbufptr == NULL) {
 		msg_read_error("Tape decompression error while");
@@ -1852,6 +1869,29 @@ decompress_tapebuf(struct tapebuf *tpbin, int readsize)
 				cresult = 0;
 #endif /* HAVE_BZLIB */
 		}
+		if (tpbin->flags == COMPRESS_LZO) {
+#ifndef HAVE_LZO
+			errx(1,"This restore version doesn't support lzo decompression");
+#else
+			cresult = lzo1x_decompress(tpbin->buf, blocklen,
+                                                   comprbuf, (lzo_uintp) &worklen,NULL);
+			output = comprbuf;
+			switch (cresult) {
+				case LZO_E_OK:
+					break;
+                                case LZO_E_ERROR:
+                                case LZO_E_EOF_NOT_FOUND:
+                                	reason = "data error";
+					break;
+				default:
+					reason = "unknown";
+			}
+			if (cresult == LZO_E_OK)
+				cresult = 1;
+			else
+				cresult = 0;
+#endif /* HAVE_LZO */
+		}
 	}
 	else {
 		output = tpbin->buf;
@@ -1897,7 +1937,7 @@ msg_read_error(char *m)
 			break;
 	}
 }
-#endif /* HAVE_ZLIB || HAVE_BZLIB */
+#endif /* HAVE_ZLIB || HAVE_BZLIB || HAVE_LZO */
 
 /*
  * Read the first block and get the blocksize from it. Test
@@ -1915,7 +1955,7 @@ findtapeblksize(void)
 	long i;
 	size_t len;
 	struct tapebuf *tpb = (struct tapebuf *) tapebuf;
-	struct s_spcl *spclpt = (struct s_spcl *) tapebuf;
+	struct s_spcl spclpt;
 
 	for (i = 0; i < ntrec; i++)
 		((struct s_spcl *)&tapebuf[i * TP_BSIZE])->c_magic = 0;
@@ -1930,6 +1970,17 @@ findtapeblksize(void)
 	if (read_a_block(mt, tapebuf, len, &i) <= 0)
 		errx(1, "Tape read error on first record");
 
+	memcpy(&spclpt, tapebuf, TP_BSIZE);
+	if (converthead(&spclpt) == FAIL) {
+		cvtflag++;
+		if (converthead(&spclpt) == FAIL) {
+			/* Special case for old compressed tapes with prefix */
+			if (magtapein && (i % TP_BSIZE != 0)) 
+				goto oldformat;
+			errx(1, "Tape is not a dump tape");
+		}
+		fprintf(stderr, "Converting to new file system format.\n");
+	}
 	/*
 	 * If the input is from a file or a pipe, we read TP_BSIZE
 	 * bytes looking for a dump header. If the dump is compressed
@@ -1938,14 +1989,14 @@ findtapeblksize(void)
 	 * dump is not compressed and does not have a prefix.
 	 */
 	if (!magtapein) {
-		if (spclpt->c_type == TS_TAPE
-		    && spclpt->c_flags & DR_COMPRESSED) {
+		if (spclpt.c_type == TS_TAPE
+		    && spclpt.c_flags & DR_COMPRESSED) {
 			/* It's a compressed dump file, read in the */
-			/* rest of the block based on spclpt->c_ntrec. */
-			if (spclpt->c_ntrec > ntrec)
+			/* rest of the block based on spclpt.c_ntrec. */
+			if (spclpt.c_ntrec > ntrec)
 				errx(1, "Tape blocksize is too large, use "
-				     "\'-b %d\' ", spclpt->c_ntrec);
-			ntrec = spclpt->c_ntrec;
+				     "\'-b %d\' ", spclpt.c_ntrec);
+			ntrec = spclpt.c_ntrec;
 			len = (ntrec - 1) * TP_BSIZE;
 			zflag = 1;   
 		}
@@ -1968,14 +2019,21 @@ findtapeblksize(void)
 	 * what we asked for; adjust the value of ntrec and test for 
 	 * a compressed dump tape.
 	 */
-
 	if (i % TP_BSIZE != 0) {
+oldformat:
 		/* may be old format compressed dump tape with a prefix */
-		spclpt = (struct s_spcl *) tpb->buf;
+		memcpy(&spclpt, tpb->buf, TP_BSIZE);
+		cvtflag = 0;
+		if (converthead(&spclpt) == FAIL) {
+			cvtflag++;
+			if (converthead(&spclpt) == FAIL)
+				errx(1, "Tape is not a dump tape");
+			fprintf(stderr, "Converting to new file system format.\n");
+		}
 		if (i % TP_BSIZE == PREFIXSIZE
 		    && tpb->compressed == 0
-		    && spclpt->c_type == TS_TAPE
-		    && spclpt->c_flags & DR_COMPRESSED) {
+		    && spclpt.c_type == TS_TAPE
+		    && spclpt.c_flags & DR_COMPRESSED) {
 			zflag = 1;
 			tbufptr = tpb->buf;
 			if (tpb->length > bufsize)
@@ -1987,12 +2045,12 @@ findtapeblksize(void)
 				i, TP_BSIZE);
 	}
 	ntrec = i / TP_BSIZE;
-	if (spclpt->c_type == TS_TAPE) {
-		if (spclpt->c_flags & DR_COMPRESSED)
+	if (spclpt.c_type == TS_TAPE) {
+		if (spclpt.c_flags & DR_COMPRESSED)
 			zflag = 1;
-		if (spclpt->c_ntrec > ntrec)
+		if (spclpt.c_ntrec > ntrec)
 			errx(1, "Tape blocksize is too large, use "
-				"\'-b %d\' ", spclpt->c_ntrec);
+				"\'-b %d\' ", spclpt.c_ntrec);
 	}
 	numtrec = ntrec;
 	Vprintf(stdout, "Tape block size is %ld\n", ntrec);
@@ -2008,7 +2066,7 @@ static int read_a_block(int fd, char *buf, size_t len, long *lengthread)
 	size = len;
 	while (size > 0) {
 #ifdef RRESTORE
-		if (host)
+		if (!Afile && host)
 			i = rmtread(buf, size);
 		else
 #endif
@@ -2032,7 +2090,7 @@ closemt(void)
 	if (mt < 0)
 		return;
 #ifdef RRESTORE
-	if (host)
+	if (!Afile && host)
 		rmtclose();
 	else
 #endif
@@ -2056,7 +2114,8 @@ setmagtapein(void) {
 			magtapein = ioctl(mt, MTIOCGET, (char *)&mt_stat) == 0;
 	}
 
-	Vprintf(stdout,"Input is from %s\n", 
+	Vprintf(stdout,"Input is from a %s %s\n",
+			host ? "remote" : "local",
 			magtapein ? "tape" :
 			Vflag ? "multi-volume (no tape)" : "file/pipe");
 }
@@ -2069,6 +2128,13 @@ setmagtapein(void) {
  */
 static int
 gethead(struct s_spcl *buf)
+{
+	readtape((char *)buf);
+	return converthead(buf);
+}
+
+static int
+converthead(struct s_spcl *buf)
 {
 	int32_t i;
 	union {
@@ -2104,7 +2170,6 @@ gethead(struct s_spcl *buf)
 	} u_ospcl;
 
 	if (!cvtflag) {
-		readtape((char *)buf);
 		if (buf->c_magic != NFS_MAGIC) {
 			if (swabi(buf->c_magic) != NFS_MAGIC)
 				return (FAIL);
@@ -2116,10 +2181,10 @@ gethead(struct s_spcl *buf)
 		if (checksum((int *)buf) == FAIL)
 			return (FAIL);
 		if (Bcvt)
-			swabst((u_char *)"8i4s31i528bi192b2i", (u_char *)buf);
+			swabst((u_char *)"8i4s31i528bi192b3i", (u_char *)buf);
 		goto good;
 	}
-	readtape((char *)(&u_ospcl.s_ospcl));
+	memcpy(&u_ospcl.s_ospcl, buf, TP_BSIZE);
 	memset((char *)buf, 0, (long)TP_BSIZE);
 	buf->c_type = u_ospcl.s_ospcl.c_type;
 	buf->c_date = u_ospcl.s_ospcl.c_date;
@@ -2219,6 +2284,23 @@ good:
 	return(GOOD);
 }
 
+static void
+converttapebuf(struct tapebuf *tpb)
+{
+	if (Bcvt) {
+		struct tb {
+			unsigned int    length:28;
+			unsigned int    flags:3;
+			unsigned int    compressed:1;
+		} tb;
+		swabst((u_char *)"i", (u_char *)tpb);
+		memcpy(&tb, tpb, 4);	
+		tpb->length = tb.length;
+		tpb->flags = tb.flags;
+		tpb->compressed = tb.compressed;
+	}
+}
+
 /*
  * Check that a header is where it belongs and predict the next header
  */
@@ -2550,7 +2632,7 @@ GotoTapePos(long long pos)
 
 #ifdef RDUMP
 	if (host)
-		err = (rmtseek((long)pos, SEEK_SET) < 0);
+		err = (rmtseek(pos, SEEK_SET) < 0);
 	else
 #endif
 	{
@@ -2601,13 +2683,12 @@ ReReadInodeFromTape(dump_ino_t theino)
 	do {
 		cntloop++;
 		gethead(&spcl);
-	} while (!(spcl.c_inumber == theino && spcl.c_type == TS_INODE && spcl.c_date == dumpdate) && (cntloop < 32));
+	} while (!(spcl.c_inumber == theino && spcl.c_type == TS_INODE && spcl.c_date == dumpdate) && (cntloop < ntrec));
 #ifdef DEBUG_QFA
 	fprintf(stderr, "%ld reads\n", cntloop);
-	if (cntloop == 32) {
+	if (cntloop == ntrec) {
 		fprintf(stderr, "DEBUG: bufsize %d\n", bufsize);
 		fprintf(stderr, "DEBUG: ntrec %ld\n", ntrec);
-		fprintf(stderr, "DEBUG: %ld reads\n", cntloop);
 	}
 #endif
 	findinode(&spcl);