]> git.wh0rd.org - dump.git/blobdiff - dump/main.c
Andreas Dilger fixes (CPP, ext2 features, const char etc).
[dump.git] / dump / main.c
index 61afed803398fa5e486422f35a03e80949013399..60f2bb4b7a4f9d145ac70f89c2addfb14fe6598f 100644 (file)
@@ -2,7 +2,8 @@
  *     Ported to Linux's Second Extended File System as part of the
  *     dump and restore backup suit
  *     Remy Card <card@Linux.EU.Org>, 1994-1997
- *     Stelian Pop <pop@cybercable.fr>, 1999-2000
+ *     Stelian Pop <pop@noos.fr>, 1999-2000
+ *     Stelian Pop <pop@noos.fr> - AlcĂ´ve <www.alcove.fr>, 2000
  */
 
 /*-
 
 #ifndef lint
 static const char rcsid[] =
-       "$Id: main.c,v 1.27 2000/11/10 13:22:10 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>
@@ -59,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"
@@ -96,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;
@@ -126,9 +127,9 @@ main(int argc, char *argv[])
        char pathname[MAXPATHLEN];
 #endif
        time_t tnow;
-       char labelstr[LBLSIZE];
        char *diskparam;
 
+       spcl.c_label[0] = '\0';
        spcl.c_date = 0;
 #ifdef __linux__
        (void)time4(&spcl.c_date);
@@ -148,7 +149,6 @@ main(int argc, char *argv[])
        if ((tapeprefix = getenv("TAPE")) == NULL)
                tapeprefix = _PATH_DEFTAPE;
        dumpdates = _PATH_DUMPDATES;
-       strcpy(labelstr, "none");       /* XXX safe strcpy. */
        if (TP_BSIZE / DEV_BSIZE == 0 || TP_BSIZE % DEV_BSIZE != 0)
                quit("TP_BSIZE must be a multiple of DEV_BSIZE\n");
        level = '0';
@@ -157,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
-       while ((ch = getopt(argc, argv, optstring)) != -1)
+                           "Mns:ST:uWw"
+#ifdef HAVE_ZLIB
+                           "z"
+#endif
+                           )) != -1)
 #undef optstring
                switch (ch) {
                /* dump level */
@@ -212,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",
@@ -243,14 +248,14 @@ main(int argc, char *argv[])
                         * the last must be '\0', so the limit on strlen()
                         * is really LBLSIZE-1.
                         */
-                       strncpy(labelstr, optarg, LBLSIZE);
-                       labelstr[LBLSIZE-1] = '\0';
+                       strncpy(spcl.c_label, optarg, LBLSIZE);
+                       spcl.c_label[LBLSIZE-1] = '\0';
                        if (strlen(optarg) > LBLSIZE-1) {
                                msg(
                "WARNING Label `%s' is larger than limit of %d characters.\n",
                                    optarg, LBLSIZE-1);
                                msg("WARNING: Using truncated label `%s'.\n",
-                                   labelstr);
+                                   spcl.c_label);
                        }
                        break;
 
@@ -290,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();
@@ -303,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--;
@@ -378,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
@@ -385,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,
@@ -401,8 +414,8 @@ main(int argc, char *argv[])
                disk = rawname(dt->fs_spec);
                (void)strncpy(spcl.c_dev, dt->fs_spec, NAMELEN);
                (void)strncpy(spcl.c_filesys, dt->fs_file, NAMELEN);
-#ifdef __linux__
        } else {
+#ifdef __linux__
 #ifdef HAVE_REALPATH
                if (realpath(disk, pathname) == NULL)
 #endif
@@ -434,14 +447,12 @@ main(int argc, char *argv[])
                                    NAMELEN);
                        }
                }
-       }
 #else
-       } else {
                (void)strncpy(spcl.c_dev, disk, NAMELEN);
                (void)strncpy(spcl.c_filesys, "an unlisted file system",
                    NAMELEN);
-       }
 #endif
+       }
 
        if (directory[0] != 0) {
                if (level != '0') {
@@ -453,10 +464,10 @@ main(int argc, char *argv[])
                        exit(X_STARTUP);
                }
        }
-       spcl.c_dev[NAMELEN-1]='\0';
-       spcl.c_filesys[NAMELEN-1]='\0';
-       (void)strncpy(spcl.c_label, labelstr, sizeof(spcl.c_label) - 1);
+       spcl.c_dev[NAMELEN-1] = '\0';
+       spcl.c_filesys[NAMELEN-1] = '\0';
        (void)gethostname(spcl.c_host, NAMELEN);
+       spcl.c_host[NAMELEN-1] = '\0';
        spcl.c_level = level - '0';
        spcl.c_type = TS_TAPE;
        if (!Tflag)
@@ -494,8 +505,6 @@ main(int argc, char *argv[])
                        msgtail("to %s on host %s\n", tape, host);
                else
                        msgtail("to %s\n", tape);
-               msg("Label: %s\n", labelstr);
-
        } /* end of size estimate */
 
 #ifdef __linux__
@@ -515,6 +524,16 @@ main(int argc, char *argv[])
                msg("Cannot open %s\n", disk);
                exit(X_STARTUP);
        }
+       /* if no user label specified, use ext2 filesystem label if available */
+       if (spcl.c_label[0] == '\0') {
+               const char *lbl;
+               if ( (lbl = get_device_label(disk)) != NULL) {
+                       strncpy(spcl.c_label, lbl, LBLSIZE);
+                       spcl.c_label[LBLSIZE-1] = '\0';
+               }
+               else
+                       strcpy(spcl.c_label, "none");   /* safe strcpy. */
+       }
        sync();
        dev_bsize = DEV_BSIZE;
        dev_bshift = ffs(dev_bsize) - 1;
@@ -554,10 +573,14 @@ main(int argc, char *argv[])
        usedinomap = (char *)calloc((unsigned) mapsize, sizeof(char));
        dumpdirmap = (char *)calloc((unsigned) mapsize, sizeof(char));
        dumpinomap = (char *)calloc((unsigned) mapsize, sizeof(char));
+       if (usedinomap == NULL || dumpdirmap == NULL || dumpinomap == NULL)
+               quit("out of memory allocating inode maps\n");
        tapesize = 2 * (howmany(mapsize * sizeof(char), TP_BSIZE) + 1);
 
        nonodump = spcl.c_level < honorlevel;
 
+       msg("Label: %s\n", spcl.c_label);
+
 #if defined(SIGINFO)
        (void)signal(SIGINFO, statussig);
 #endif
@@ -734,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");
@@ -760,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);
@@ -810,8 +844,8 @@ sig(int signo)
        }
 }
 
-char *
-rawname(char *cp)
+const char *
+rawname(const char *cp)
 {
 #ifdef __linux__
        return cp;
@@ -821,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);