X-Git-Url: https://git.wh0rd.org/?a=blobdiff_plain;f=dump%2Fmain.c;h=60f2bb4b7a4f9d145ac70f89c2addfb14fe6598f;hb=43460f04f3920ae39f345a2fb39d000a890a4283;hp=7745e46c4e3cd4ad9b83fd1229b866f3ec0bf592;hpb=d6967896bc4e18f9b4359a2ef5be5c0f1bcc456a;p=dump.git

diff --git a/dump/main.c b/dump/main.c
index 7745e46..60f2bb4 100644
--- a/dump/main.c
+++ b/dump/main.c
@@ -41,18 +41,29 @@
 
 #ifndef lint
 static const char rcsid[] =
-	"$Id: main.c,v 1.32 2000/12/21 11:14:54 stelian Exp $";
+	"$Id: main.c,v 1.35 2001/03/19 13:22:48 stelian Exp $";
 #endif /* not lint */
 
 #include <config.h>
+#include <ctype.h>
+#include <compaterr.h>
+#include <fcntl.h>
+#include <fstab.h>
+#include <signal.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
 #include <sys/param.h>
 #include <sys/time.h>
 #ifdef __linux__
 #include <linux/ext2_fs.h>
+#include <ext2fs/ext2fs.h>
+#include <time.h>
 #include <sys/stat.h>
 #include <bsdcompat.h>
-#else
-#ifdef sunos
+#elif defined sunos
 #include <sys/vnode.h>
 
 #include <ufs/inode.h>
@@ -61,24 +72,9 @@ static const char rcsid[] =
 #include <ufs/ufs/dinode.h>
 #include <ufs/ffs/fs.h>
 #endif
-#endif
 
 #include <protocols/dumprestore.h>
 
-#include <ctype.h>
-#include <compaterr.h>
-#include <fcntl.h>
-#include <fstab.h>
-#include <signal.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <unistd.h>
-
-#ifdef __linux__
-#include <ext2fs/ext2fs.h>
-#endif
-
 #include "dump.h"
 #include "pathnames.h"
 #include "bylabel.h"
@@ -98,6 +94,9 @@ long	dev_bsize = 1;	/* recalculated below */
 long	blocksperfile;	/* output blocks per file */
 char	*host = NULL;	/* remote host (if any) */
 int	sizest = 0;	/* return size estimate only */
+int	compressed = 0;	/* use zlib to compress the output */
+long long bytes_written = 0; /* total bytes written */
+long	uncomprblks = 0;/* uncompressed blocks written */
 
 #ifdef	__linux__
 char	*__progname;
@@ -158,12 +157,17 @@ main(int argc, char *argv[])
 		usage();
 
 	obsolete(&argc, &argv);
+
+	while ((ch = getopt(argc, argv,
+			    "0123456789aB:b:cd:e:f:F:h:L:"
 #ifdef KERBEROS
-#define optstring "0123456789aB:b:cd:e:f:F:h:kL:Mns:ST:uWw"
-#else
-#define optstring "0123456789aB:b:cd:e:f:F:h:L:Mns:ST:uWw"
+			    "k"
+#endif
+			    "Mns:ST:uWw"
+#ifdef HAVE_ZLIB
+			    "z"
 #endif
-	while ((ch = getopt(argc, argv, optstring)) != -1)
+			    )) != -1)
 #undef optstring
 		switch (ch) {
 		/* dump level */
@@ -213,7 +217,7 @@ main(int argc, char *argv[])
 			}
 		        iexclude_list[iexclude_num++] = numarg("inode to exclude",0L,0L);
 			if (iexclude_list[iexclude_num-1] <= ROOTINO) {
-				(void)fprintf(stderr, "Cannot exclude inode %ld\n", iexclude_list[iexclude_num-1]);
+				(void)fprintf(stderr, "Cannot exclude inode %ld\n", (long)iexclude_list[iexclude_num-1]);
 				exit(X_STARTUP);
 			}
 			msg("Added %d to exclude list\n",
@@ -291,6 +295,11 @@ main(int argc, char *argv[])
 		case 'w':
 			lastdump(ch);
 			exit(X_FINOK);	/* do nothing else */
+#ifdef HAVE_ZLIB
+		case 'z':
+			compressed = 1;
+			break;
+#endif /* HAVE_ZLIB */
 
 		default:
 			usage();
@@ -304,7 +313,8 @@ main(int argc, char *argv[])
 	}
 	diskparam = *argv++;
 	if (strlen(diskparam) >= MAXPATHLEN) {
-		(void)fprintf(stderr, "Disk or filesystem name too long: %s\n", diskparam);
+		(void)fprintf(stderr, "Disk or filesystem name too long: %s\n", 
+			      diskparam);
 		exit(X_STARTUP);
 	}
 	argc--;
@@ -379,6 +389,14 @@ main(int argc, char *argv[])
 	set_operators();	/* /etc/group snarfed */
 	getfstab();		/* /etc/fstab snarfed */
 
+	/*
+	 *      disk may end in / and this can confuse
+	 *      fstabsearch.
+	 */
+	i = strlen(diskparam) - 1;
+	if (i > 1 && diskparam[i] == '/')
+		diskparam[i] = '\0';
+
 	disk = get_device_name(diskparam);
 	if (!disk) {		/* null means the disk is some form
 				   of LABEL= or UID= but it was not
@@ -386,12 +404,6 @@ main(int argc, char *argv[])
 		msg("Cannot find a disk having %s\n", diskparam);
 		exit(X_STARTUP);
 	}
-	/*
-	 *      disk may end in / and this can confuse
-	 *      fstabsearch.
-	 */
-	if (strlen(disk) > 1 && disk[strlen(disk) - 1] == '/')
-		disk[strlen(disk) - 1] = '\0';
 	/*
 	 *	disk can be either the full special file name,
 	 *	the suffix of the special file name,
@@ -745,6 +757,13 @@ main(int argc, char *argv[])
 	msg("Date this dump completed:  %s", ctime(&tnow));
 
 	msg("Average transfer rate: %ld KB/s\n", xferrate / tapeno);
+	if (compressed) {
+		long tapekb = bytes_written / 1024;
+		double rate = .0005 + (double) spcl.c_tapea / tapekb;
+		msg("Wrote %ldKB uncompressed, %ldKB compressed,"
+			" compression ratio %1.3f\n",
+			spcl.c_tapea, tapekb, rate);
+	}
 
 	broadcast("DUMP IS DONE!\7\7\n");
 	msg("DUMP IS DONE\n");
@@ -771,7 +790,11 @@ usage(void)
 #ifdef KERBEROS
 		"k"
 #endif
-		"MnSu] [-B records] [-b blocksize] [-d density]\n"
+		"MnSu"
+#ifdef HAVE_ZLIB
+		"z"
+#endif
+		"] [-B records] [-b blocksize] [-d density]\n"
 		"\t%s [-e inode#] [-f file] [-h level] [-s feet] [-T date] filesystem\n"
 		"\t%s [-W | -w]\n", __progname, white, __progname);
 	exit(X_STARTUP);
@@ -821,8 +844,8 @@ sig(int signo)
 	}
 }
 
-char *
-rawname(char *cp)
+const char *
+rawname(const char *cp)
 {
 #ifdef	__linux__
 	return cp;
@@ -832,10 +855,8 @@ rawname(char *cp)
 
 	if (dp == NULL)
 		return (NULL);
-	*dp = '\0';
-	(void)strncpy(rawbuf, cp, MAXPATHLEN - 1);
-	rawbuf[MAXPATHLEN-1] = '\0';
-	*dp = '/';
+	(void)strncpy(rawbuf, cp, min(dp-cp, MAXPATHLEN - 1));
+	rawbuf[min(dp-cp, MAXPATHLEN-1)] = '\0';
 	(void)strncat(rawbuf, "/r", MAXPATHLEN - 1 - strlen(rawbuf));
 	(void)strncat(rawbuf, dp + 1, MAXPATHLEN - 1 - strlen(rawbuf));
 	return (rawbuf);