]> git.saurik.com Git - apple/file_cmds.git/blobdiff - mtree/create.c
file_cmds-116.10.tar.gz
[apple/file_cmds.git] / mtree / create.c
index 7487de1d9ad2fe9fd1e2d3ea21e27ad14cdc01ef..f755accadcd99af91df135ea5afb2991da0cccf6 100644 (file)
@@ -1,5 +1,3 @@
-/*     $NetBSD: create.c,v 1.16 1998/08/30 03:20:09 nathanw Exp $      */
-
 /*-
  * Copyright (c) 1989, 1993
  *     The Regents of the University of California.  All rights reserved.
  * SUCH DAMAGE.
  */
 
-#include <sys/cdefs.h>
 #ifndef lint
 #if 0
 static char sccsid[] = "@(#)create.c   8.1 (Berkeley) 6/6/93";
-#else
-__RCSID("$NetBSD: create.c,v 1.16 1998/08/30 03:20:09 nathanw Exp $");
 #endif
+static const char rcsid[] =
+  "$FreeBSD: src/usr.sbin/mtree/create.c,v 1.18.2.3 2001/01/12 19:17:18 phk Exp $";
 #endif /* not lint */
 
 #include <sys/param.h>
 #include <sys/stat.h>
 #include <dirent.h>
+#include <err.h>
 #include <errno.h>
 #include <fcntl.h>
 #include <fts.h>
 #include <grp.h>
+#ifdef MD5
+#include <md5.h>
+#endif
+#ifdef SHA1
+#include <sha.h>
+#endif
+#ifdef RMD160
+#include <ripemd.h>
+#endif
 #include <pwd.h>
 #include <stdio.h>
-#include <string.h>
 #include <time.h>
 #include <unistd.h>
+#include <vis.h>
 #include "mtree.h"
 #include "extern.h"
 
 #define        INDENTNAMELEN   15
 #define        MAXLINELEN      80
 
-extern int crc_total, ftsoptions;
-extern int dflag, sflag;
-extern u_short keys;
+extern long int crc_total;
+extern int ftsoptions;
+extern int dflag, iflag, nflag, sflag;
+extern u_int keys;
 extern char fullpath[MAXPATHLEN];
+extern int lineno;
 
 static gid_t gid;
 static uid_t uid;
 static mode_t mode;
+static u_long flags = 0xffffffff;
 
 static int     dsort __P((const FTSENT **, const FTSENT **));
-static void    output __P((int *, const char *, ...));
-static int     statd __P((FTS *, FTSENT *, uid_t *, gid_t *, mode_t *));
-static void    statf __P((FTSENT *));
+static void    output __P((int, int *, const char *, ...));
+static int     statd __P((FTS *, FTSENT *, uid_t *, gid_t *, mode_t *,
+                          u_long *));
+static void    statf __P((int, FTSENT *));
 
 void
 cwalk()
 {
-       FTS *t;
-       FTSENT *p;
+       register FTS *t;
+       register FTSENT *p;
        time_t clock;
-       char *argv[2], host[MAXHOSTNAMELEN + 1];
+       char *argv[2], host[MAXHOSTNAMELEN];
+       int indent = 0;
 
        (void)time(&clock);
        (void)gethostname(host, sizeof(host));
-       host[sizeof(host) - 1] = '\0';
        (void)printf(
            "#\t   user: %s\n#\tmachine: %s\n#\t   tree: %s\n#\t   date: %s",
            getlogin(), host, fullpath, ctime(&clock));
@@ -92,171 +103,294 @@ cwalk()
        argv[0] = ".";
        argv[1] = NULL;
        if ((t = fts_open(argv, ftsoptions, dsort)) == NULL)
-               mtree_err("fts_open: %s", strerror(errno));
-       while ((p = fts_read(t)) != NULL)
+               err(1, "line %d: fts_open", lineno);
+       while ((p = fts_read(t))) {
+               if (iflag)
+                       indent = p->fts_level * 4;
+               if (check_excludes(p->fts_name, p->fts_path)) {
+                       fts_set(t, p, FTS_SKIP);
+                       continue;
+               }
                switch(p->fts_info) {
                case FTS_D:
-                       (void)printf("\n# %s\n", p->fts_path);
-                       statd(t, p, &uid, &gid, &mode);
-                       statf(p);
+                       if (!dflag)
+                               (void)printf("\n");
+                       if (!nflag)
+                               (void)printf("# %s\n", p->fts_path);
+                       statd(t, p, &uid, &gid, &mode, &flags);
+                       statf(indent, p);
                        break;
                case FTS_DP:
-                       if (p->fts_level > 0)
-                               (void)printf("# %s\n..\n\n", p->fts_path);
+                       if (!nflag && (p->fts_level > 0))
+                               (void)printf("%*s# %s\n", indent, "", p->fts_path);
+                       (void)printf("%*s..\n", indent, "");
+                       if (!dflag)
+                               (void)printf("\n");
                        break;
                case FTS_DNR:
                case FTS_ERR:
                case FTS_NS:
-                       (void)fprintf(stderr,
-                           "mtree: %s: %s\n", p->fts_path, strerror(errno));
+                       warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
                        break;
                default:
                        if (!dflag)
-                               statf(p);
+                               statf(indent, p);
                        break;
-                       
+
                }
+       }
        (void)fts_close(t);
        if (sflag && keys & F_CKSUM)
-               (void)fprintf(stderr,
-                   "mtree: %s checksum: %u\n", fullpath, crc_total);
+               warnx("%s checksum: %lu", fullpath, crc_total);
 }
 
 static void
-statf(p)
+statf(indent, p)
+       int indent;
        FTSENT *p;
 {
        struct group *gr;
        struct passwd *pw;
-       u_int32_t len, val;
-       int fd, indent;
+       u_long len, val;
+       int fd, offset;
+       char *fflags;
+       char *escaped_name;
+
+       escaped_name = calloc(1, p->fts_namelen * 4  +  1);
+       if (escaped_name == NULL)
+               errx(1, "statf(): calloc() failed");
+       strvis(escaped_name, p->fts_name, VIS_WHITE | VIS_OCTAL);
 
-       if (S_ISDIR(p->fts_statp->st_mode))
-               indent = printf("%s", p->fts_name); 
+       if (iflag || S_ISDIR(p->fts_statp->st_mode))
+               offset = printf("%*s%s", indent, "", escaped_name);
        else
-               indent = printf("    %s", p->fts_name);
+               offset = printf("%*s    %s", indent, "", escaped_name);
+       
+       free(escaped_name);
 
-       if (indent > INDENTNAMELEN)
-               indent = MAXLINELEN;
+       if (offset > (INDENTNAMELEN + indent))
+               offset = MAXLINELEN;
        else
-               indent += printf("%*s", INDENTNAMELEN - indent, "");
+               offset += printf("%*s", (INDENTNAMELEN + indent) - offset, "");
 
-       if (!S_ISREG(p->fts_statp->st_mode))
-               output(&indent, "type=%s", inotype(p->fts_statp->st_mode));
-       if (keys & (F_UID | F_UNAME) && p->fts_statp->st_uid != uid) {
-               if (keys & F_UNAME && (pw = getpwuid(p->fts_statp->st_uid)))
-                       output(&indent, "uname=%s", pw->pw_name);
-               else /* if (keys & F_UID) */
-                       output(&indent, "uid=%u", p->fts_statp->st_uid);
+       if (!S_ISREG(p->fts_statp->st_mode) && !dflag)
+               output(indent, &offset, "type=%s", inotype(p->fts_statp->st_mode));
+       if (p->fts_statp->st_uid != uid) {
+               if (keys & F_UNAME) {
+                       if ((pw = getpwuid(p->fts_statp->st_uid)) != NULL) {
+                               output(indent, &offset, "uname=%s", pw->pw_name);
+                       } else {
+                               errx(1,
+                               "line %d: could not get uname for uid=%u",
+                               lineno, p->fts_statp->st_uid);
+                       }
+               }
+               if (keys & F_UID)
+                       output(indent, &offset, "uid=%u", p->fts_statp->st_uid);
        }
-       if (keys & (F_GID | F_GNAME) && p->fts_statp->st_gid != gid) {
-               if (keys & F_GNAME && (gr = getgrgid(p->fts_statp->st_gid)))
-                       output(&indent, "gname=%s", gr->gr_name);
-               else /* if (keys & F_GID) */
-                       output(&indent, "gid=%u", p->fts_statp->st_gid);
+       if (p->fts_statp->st_gid != gid) {
+               if (keys & F_GNAME) {
+                       if ((gr = getgrgid(p->fts_statp->st_gid)) != NULL) {
+                               output(indent, &offset, "gname=%s", gr->gr_name);
+                       } else {
+                               errx(1,
+                               "line %d: could not get gname for gid=%u",
+                               lineno, p->fts_statp->st_gid);
+                       }
+               }
+               if (keys & F_GID)
+                       output(indent, &offset, "gid=%u", p->fts_statp->st_gid);
        }
        if (keys & F_MODE && (p->fts_statp->st_mode & MBITS) != mode)
-               output(&indent, "mode=%#o", p->fts_statp->st_mode & MBITS);
+               output(indent, &offset, "mode=%#o", p->fts_statp->st_mode & MBITS);
        if (keys & F_NLINK && p->fts_statp->st_nlink != 1)
-               output(&indent, "nlink=%u", p->fts_statp->st_nlink);
-       if (keys & F_SIZE && S_ISREG(p->fts_statp->st_mode))
-               output(&indent, "size=%qd", p->fts_statp->st_size);
+               output(indent, &offset, "nlink=%u", p->fts_statp->st_nlink);
+       if (keys & F_SIZE)
+               output(indent, &offset, "size=%qd", p->fts_statp->st_size);
        if (keys & F_TIME)
-               output(&indent, "time=%ld.%ld",
+               output(indent, &offset, "time=%ld.%ld",
                    p->fts_statp->st_mtimespec.tv_sec,
                    p->fts_statp->st_mtimespec.tv_nsec);
        if (keys & F_CKSUM && S_ISREG(p->fts_statp->st_mode)) {
                if ((fd = open(p->fts_accpath, O_RDONLY, 0)) < 0 ||
                    crc(fd, &val, &len))
-                       mtree_err("%s: %s", p->fts_accpath, strerror(errno));
+                       err(1, "line %d: %s", lineno, p->fts_accpath);
                (void)close(fd);
-               output(&indent, "cksum=%lu", val);
+               output(indent, &offset, "cksum=%lu", val);
+       }
+#ifdef MD5
+       if (keys & F_MD5 && S_ISREG(p->fts_statp->st_mode)) {
+               char *digest, buf[33];
+
+               digest = MD5File(p->fts_accpath, buf);
+               if (!digest) {
+                       err(1, "line %d: %s", lineno, p->fts_accpath);
+               } else {
+                       output(indent, &offset, "md5digest=%s", digest);
+               }
        }
+#endif /* MD5 */
+#ifdef SHA1
+       if (keys & F_SHA1 && S_ISREG(p->fts_statp->st_mode)) {
+               char *digest, buf[41];
+
+               digest = SHA1_File(p->fts_accpath, buf);
+               if (!digest) {
+                       err(1, "line %d: %s", lineno, p->fts_accpath);
+               } else {
+                       output(indent, &offset, "sha1digest=%s", digest);
+               }
+       }
+#endif /* SHA1 */
+#ifdef RMD160
+       if (keys & F_RMD160 && S_ISREG(p->fts_statp->st_mode)) {
+               char *digest, buf[41];
+
+               digest = RIPEMD160_File(p->fts_accpath, buf);
+               if (!digest) {
+                       err(1, "line %d: %s", lineno, p->fts_accpath);
+               } else {
+                       output(indent, &offset, "ripemd160digest=%s", digest);
+               }
+       }
+#endif /* RMD160 */
        if (keys & F_SLINK &&
            (p->fts_info == FTS_SL || p->fts_info == FTS_SLNONE))
-               output(&indent, "link=%s", rlink(p->fts_accpath));
+               output(indent, &offset, "link=%s", rlink(p->fts_accpath));
+       if (keys & F_FLAGS && p->fts_statp->st_flags != flags) {
+               fflags = flags_to_string(p->fts_statp->st_flags);
+               output(indent, &offset, "flags=%s", fflags);
+               free(fflags);
+       }
        (void)putchar('\n');
 }
 
 #define        MAXGID  5000
 #define        MAXUID  5000
 #define        MAXMODE MBITS + 1
+#define        MAXFLAGS 256
+#define        MAXS 16
 
 static int
-statd(t, parent, puid, pgid, pmode)
+statd(t, parent, puid, pgid, pmode, pflags)
        FTS *t;
        FTSENT *parent;
        uid_t *puid;
        gid_t *pgid;
        mode_t *pmode;
+       u_long *pflags;
 {
-       FTSENT *p;
-       gid_t sgid;
-       uid_t suid;
-       mode_t smode;
+       register FTSENT *p;
+       register gid_t sgid;
+       register uid_t suid;
+       register mode_t smode;
+       register u_long sflags;
        struct group *gr;
        struct passwd *pw;
-       gid_t savegid;
-       uid_t saveuid;
-       mode_t savemode;
-       u_short maxgid, maxuid, maxmode, g[MAXGID], u[MAXUID], m[MAXMODE];
+       gid_t savegid = *pgid;
+       uid_t saveuid = *puid;
+       mode_t savemode = *pmode;
+       u_long saveflags = *pflags;
+       u_short maxgid, maxuid, maxmode, maxflags;
+       u_short g[MAXGID], u[MAXUID], m[MAXMODE], f[MAXFLAGS];
+       char *fflags;
+       static int first = 1;
 
-       savegid = 0;
-       saveuid = 0;
-       savemode = 0;
        if ((p = fts_children(t, 0)) == NULL) {
                if (errno)
-                       mtree_err("%s: %s", RP(parent), strerror(errno));
+                       err(1, "line %d: %s", lineno, RP(parent));
                return (1);
        }
 
-       memset(g, 0, sizeof(g));
-       memset(u, 0, sizeof(u));
-       memset(m, 0, sizeof(m));
+       bzero(g, sizeof(g));
+       bzero(u, sizeof(u));
+       bzero(m, sizeof(m));
+       bzero(f, sizeof(f));
 
-       maxuid = maxgid = maxmode = 0;
+       maxuid = maxgid = maxmode = maxflags = 0;
        for (; p; p = p->fts_link) {
-               smode = p->fts_statp->st_mode & MBITS;
-               if (smode < MAXMODE && ++m[smode] > maxmode) {
-                       savemode = smode;
-                       maxmode = m[smode];
-               }
-               sgid = p->fts_statp->st_gid;
-               if (sgid < MAXGID && ++g[sgid] > maxgid) {
-                       savegid = sgid;
-                       maxgid = g[sgid];
-               }
-               suid = p->fts_statp->st_uid;
-               if (suid < MAXUID && ++u[suid] > maxuid) {
-                       saveuid = suid;
-                       maxuid = u[suid];
+               if (!dflag || (dflag && S_ISDIR(p->fts_statp->st_mode))) {
+                       smode = p->fts_statp->st_mode & MBITS;
+                       if (smode < MAXMODE && ++m[smode] > maxmode) {
+                               savemode = smode;
+                               maxmode = m[smode];
+                       }
+                       sgid = p->fts_statp->st_gid;
+                       if (sgid < MAXGID && ++g[sgid] > maxgid) {
+                               savegid = sgid;
+                               maxgid = g[sgid];
+                       }
+                       suid = p->fts_statp->st_uid;
+                       if (suid < MAXUID && ++u[suid] > maxuid) {
+                               saveuid = suid;
+                               maxuid = u[suid];
+                       }
+
+                       /*
+                        * XXX
+                        * note that the below will break when file flags
+                        * are extended beyond the first 4 bytes of each
+                        * half word of the flags
+                        */
+#define FLAGS2IDX(f) ((f & 0xf) | ((f >> 12) & 0xf0))
+                       sflags = p->fts_statp->st_flags;
+                       if (FLAGS2IDX(sflags) < MAXFLAGS &&
+                           ++f[FLAGS2IDX(sflags)] > maxflags) {
+                               saveflags = sflags;
+                               maxflags = f[FLAGS2IDX(sflags)];
+                       }
                }
        }
-       (void)printf("/set type=file");
-       if (keys & F_GID)
-               (void)printf(" gid=%u", savegid);
-       if (keys & F_GNAME) {
-               if ((gr = getgrgid(savegid)) != NULL)
-                       (void)printf(" gname=%s", gr->gr_name);
-               else
-                       (void)printf(" gid=%u", savegid);
-       }
-       if (keys & F_UNAME) {
-               if ((pw = getpwuid(saveuid)) != NULL)
-                       (void)printf(" uname=%s", pw->pw_name);
+       /*
+        * If the /set record is the same as the last one we do not need to output
+        * a new one.  So first we check to see if anything changed.  Note that we
+        * always output a /set record for the first directory.
+        */
+       if ((((keys & F_UNAME) | (keys & F_UID)) && (*puid != saveuid)) ||
+           (((keys & F_GNAME) | (keys & F_GID)) && (*pgid != savegid)) ||
+           ((keys & F_MODE) && (*pmode != savemode)) || 
+           ((keys & F_FLAGS) && (*pflags != saveflags)) ||
+           (first)) {
+               first = 0;
+               if (dflag)
+                       (void)printf("/set type=dir");
                else
-                       (void)printf(" uid=%u", saveuid);
+                       (void)printf("/set type=file");
+               if (keys & F_UNAME) {
+                       if ((pw = getpwuid(saveuid)) != NULL)
+                               (void)printf(" uname=%s", pw->pw_name);
+                       else
+                               errx(1,
+                               "line %d: could not get uname for uid=%u",
+                               lineno, saveuid);
+               }
+               if (keys & F_UID)
+                       (void)printf(" uid=%lu", (u_long)saveuid);
+               if (keys & F_GNAME) {
+                       if ((gr = getgrgid(savegid)) != NULL)
+                               (void)printf(" gname=%s", gr->gr_name);
+                       else
+                               errx(1,
+                               "line %d: could not get gname for gid=%u",
+                               lineno, savegid);
+               }
+               if (keys & F_GID)
+                       (void)printf(" gid=%lu", (u_long)savegid);
+               if (keys & F_MODE)
+                       (void)printf(" mode=%#o", savemode);
+               if (keys & F_NLINK)
+                       (void)printf(" nlink=1");
+               if (keys & F_FLAGS) {
+                       fflags = flags_to_string(saveflags);
+                       (void)printf(" flags=%s", fflags);
+                       free(fflags);
+               }
+               (void)printf("\n");
+               *puid = saveuid;
+               *pgid = savegid;
+               *pmode = savemode;
+               *pflags = saveflags;
        }
-       if (keys & F_UID)
-               (void)printf(" uid=%u", saveuid);
-       if (keys & F_MODE)
-               (void)printf(" mode=%#o", savemode);
-       if (keys & F_NLINK)
-               (void)printf(" nlink=1");
-       (void)printf("\n");
-       *puid = saveuid;
-       *pgid = savegid;
-       *pmode = savemode;
        return (0);
 }
 
@@ -280,9 +414,10 @@ dsort(a, b)
 
 void
 #if __STDC__
-output(int *offset, const char *fmt, ...)
+output(int indent, int *offset, const char *fmt, ...)
 #else
-output(offset, fmt, va_alist)
+output(indent, offset, fmt, va_alist)
+       int indent;
        int *offset;
        char *fmt;
         va_dcl
@@ -299,8 +434,8 @@ output(offset, fmt, va_alist)
        va_end(ap);
 
        if (*offset + strlen(buf) > MAXLINELEN - 3) {
-               (void)printf(" \\\n%*s", INDENTNAMELEN, "");
-               *offset = INDENTNAMELEN;
+               (void)printf(" \\\n%*s", INDENTNAMELEN + indent, "");
+               *offset = INDENTNAMELEN + indent;
        }
        *offset += printf(" %s", buf) + 1;
 }