X-Git-Url: https://git.wh0rd.org/?a=blobdiff_plain;f=dump%2Ftape.c;h=45535ea692e1c3adfbf9fd6b0dc3ca9ebb18cd88;hb=3195bc1f1361373549c37145a37381eceb623371;hp=5e79fd2f66a7abac0c70d893a186d21928c1b71c;hpb=b0522d37ac04aeae5ba01a848a8530ec4785b79d;p=dump.git

diff --git a/dump/tape.c b/dump/tape.c
index 5e79fd2..45535ea 100644
--- a/dump/tape.c
+++ b/dump/tape.c
@@ -41,7 +41,7 @@
 
 #ifndef lint
 static const char rcsid[] =
-	"$Id: tape.c,v 1.53 2001/07/20 11:02:45 stelian Exp $";
+	"$Id: tape.c,v 1.57 2001/09/12 10:21:49 stelian Exp $";
 #endif /* not lint */
 
 #include <config.h>
@@ -68,10 +68,12 @@ int    write(), read();
 #endif
 #include <sys/param.h>
 #include <sys/socket.h>
-#include <sys/time.h>
 #include <sys/wait.h>
 #include <sys/mtio.h>
 #ifdef __linux__
+#include <linux/fs.h>
+#undef atomic_read	/* this get wrongly defined in kernel */
+			/* headers and we don't want it */
 #ifdef HAVE_EXT2FS_EXT2_FS_H
 #include <ext2fs/ext2_fs.h>
 #else
@@ -95,6 +97,10 @@ int    write(), read();
 #include <zlib.h>
 #endif /* HAVE_ZLIB */
 
+#ifdef HAVE_BZLIB
+#include <bzlib.h>
+#endif /* HAVE_BZLIB */
+
 #include "dump.h"
 
 int	writesize;		/* size of malloc()ed buffer for tape */
@@ -558,7 +564,7 @@ close_rewind(void)
 void
 rollforward(void)
 {
-	register struct req *p, *q, *prev;
+	register struct req *p, *q = NULL, *prev;
 	register struct slave *tslp;
 	int i, size, savedtapea, got;
 	union u_spcl *ntb, *otb;
@@ -1017,11 +1023,14 @@ doslave(int cmd, int slave_number, int first)
 	int nextslave, size, eot_count, bufsize;
 	volatile int wrote = 0;
 	char *buffer;
-#ifdef HAVE_ZLIB
+#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB)
 	struct tapebuf *comp_buf = NULL;
 	int compresult, do_compress = !first;
 	unsigned long worklen;
-#endif /* HAVE_ZLIB */
+#ifdef HAVE_BZLIB
+	unsigned int worklen2;
+#endif
+#endif /* HAVE_ZLIB || HAVE_BZLIB */
 	struct slave_results returns;
 #ifdef	__linux__
 	errcode_t retval;
@@ -1039,6 +1048,9 @@ doslave(int cmd, int slave_number, int first)
 	if ((diskfd = OPEN(disk, O_RDONLY)) < 0)
 		quit("slave couldn't reopen disk: %s\n", strerror(errno));
 #ifdef	__linux__
+#ifdef BLKFLSBUF
+	(void)ioctl(diskfd, BLKFLSBUF);
+#endif
 	ext2fs_close(fs);
 	retval = dump_fs_open(disk, &fs);
 	if (retval)
@@ -1053,15 +1065,18 @@ doslave(int cmd, int slave_number, int first)
 		quit("master/slave protocol botched - didn't get pid of next slave.\n");
 	}
 
-#ifdef HAVE_ZLIB
+#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB)
 	/* if we're doing a compressed dump, allocate the compress buffer */
 	if (compressed) {
 		comp_buf = malloc(sizeof(struct tapebuf) + TP_BSIZE + writesize);
 		if (comp_buf == NULL)
 			quit("couldn't allocate a compress buffer.\n");
-		comp_buf->flags = 0;
+		if (bzipflag)
+			comp_buf->flags = COMPRESS_BZLIB;
+		else
+			comp_buf->flags = COMPRESS_ZLIB;
 	}
-#endif /* HAVE_ZLIB */
+#endif /* HAVE_ZLIB || HAVE_BZLIB */
 
 	/*
 	 * Get list of blocks to dump, read the blocks into tape buffer
@@ -1090,7 +1105,7 @@ doslave(int cmd, int slave_number, int first)
 		bufsize = writesize;			/* length to write */
 		returns.clen = returns.unclen = bufsize;
 
-#ifdef HAVE_ZLIB
+#if defined(HAVE_ZLIB) || defined(HAVE_BZLIB)
 		/* 
 		 * When writing a compressed dump, each block except
 		 * the first one on each tape is written
@@ -1107,9 +1122,39 @@ doslave(int cmd, int slave_number, int first)
 		if (compressed && do_compress) {
 			comp_buf->length = bufsize;
 			worklen = TP_BSIZE + writesize;
-			compresult = compress2(comp_buf->buf, &worklen,
-				(char *)slp->tblock[0], writesize, compressed);
-			if (compresult == Z_OK && worklen <= (writesize - 16)) {
+			compresult = 1;
+#ifdef HAVE_ZLIB
+			if (!bzipflag) {
+				compresult = compress2(comp_buf->buf, 
+						       &worklen,
+						       (char *)slp->tblock[0],
+						       writesize, 
+						       compressed);
+				if (compresult == Z_OK)
+					compresult = 1;
+				else
+					compresult = 0;
+			}
+#endif /* HAVE_ZLIB */
+#ifdef HAVE_BZLIB
+			if (bzipflag) {
+				worklen2 = worklen;
+				compresult = BZ2_bzBuffToBuffCompress(
+						       comp_buf->buf,
+						       &worklen2,
+						       (char *)slp->tblock[0],
+						       writesize,
+						       compressed,
+						       0, 30);
+				worklen = worklen2;
+				if (compresult == BZ_OK)
+					compresult = 1;
+				else
+					compresult = 0;
+			}
+
+#endif /* HAVE_BZLIB */
+			if (compresult && worklen <= (writesize - 16)) {
 				/* write the compressed buffer */
 				comp_buf->length = worklen;
 				comp_buf->compressed = 1;
@@ -1128,7 +1173,7 @@ doslave(int cmd, int slave_number, int first)
 		}
 		/* compress the remaining blocks if we're compressing */
 		do_compress = compressed;
-#endif /* HAVE_ZLIB */
+#endif /* HAVE_ZLIB  || HAVE_BZLIB */
 
 		if (sigsetjmp(jmpbuf, 1) == 0) {
 			ready = 1;