fold INFO and ENTRY structures together into NODE structure

[unix-history] / usr / src / usr.sbin / mtree / create.c

/*-
* Copyright (c) 1989 The Regents of the University of California.
* All rights reserved.
*
* %sccs.include.redist.c%
*/

#ifndef lint
static char sccsid[] = "@(#)create.c    5.9 (Berkeley) %G%";
#endif /* not lint */

#include <sys/param.h>
#include <sys/stat.h>
#include <time.h>
#include <fts.h>
#include <dirent.h>
#include <errno.h>
#include <stdio.h>
#include "mtree.h"

#define LABEL \
       if (label++) \
               (void)putchar(' '); \

int ftsoptions = FTS_PHYSICAL;

cwalk()
{
       extern int dflag;
       register FTS *t;
       register FTSENT *p;
       register int cnt, label, notset;
       time_t clock;
       uid_t uid;
       gid_t gid;
       mode_t mode;
       int tabs, dsort();
       char curp[MAXPATHLEN], *inotype(), *getlogin(), *rlink();

       if (!getwd(curp)) {
               (void)fprintf(stderr, "mtree: %s\n", curp);
               exit(1);
       }
       (void)time(&clock);
       (void)printf("#\n#\t  fs: %s\n#\t  by: %s\n#\tdate: %s#\n",
           curp, getlogin(), ctime(&clock));

       if (!(t = ftsopen(".", ftsoptions, dsort))) {
               (void)fprintf(stderr,
                   "mtree: ftsopen: %s.\n", strerror(errno));
               exit(1);
       }
       while (p = ftsread(t)) {
               switch(p->fts_info) {
               case FTS_D:
                       if (dflag)
                               notset = 1;
                       else
                               notset =
                                   statdir(t, p, &uid, &gid, &mode, &tabs);
                       if (!strcmp(p->fts_name, "."))
                               continue;
                       break;
               case FTS_DC:
                       (void)fprintf(stderr,
                           "mtree: directory cycle: %s.\n", p->fts_path);
                       continue;
               case FTS_DNR:
                       (void)fprintf(stderr,
                           "mtree: %s: unable to read.\n", p->fts_path);
                       continue;
               case FTS_DNX:
                       (void)fprintf(stderr,
                           "mtree: %s: unable to search.\n", p->fts_path);
                       continue;
               case FTS_DP:
                       if (p->fts_level <= 0)
                               continue;
                       for (cnt = p->fts_level - 1; cnt-- > 0; )
                               (void)putchar('\t');
                       (void)printf("..\n");
                       continue;
               case FTS_ERR:
                       (void)fprintf(stderr, "mtree: %s: %s.\n",
                           p->fts_path, strerror(errno));
                       continue;
               case FTS_NS:
                       (void)fprintf(stderr,
                           "mtree: can't stat: %s.\n", p->fts_path);
                       continue;
               default:
                       if (dflag)
                               continue;
               }

               for (cnt = p->fts_level - 1; cnt-- > 0; )
                       (void)putchar('\t');
               (void)printf("%s", p->fts_name);
               if (p->fts_info == FTS_D)
                       (void)putchar('\t');
               else {
                       if (tabs > 1 && p->fts_namelen < 8)
                               (void)putchar('\t');
                       (void)putchar('\t');
               }

               label = 0;
               if (!S_ISREG(p->fts_statb.st_mode) || notset) {
                       LABEL;
                       (void)printf("type=%s", inotype(p->fts_statb.st_mode));
               }
               if (p->fts_statb.st_uid != uid || notset) {
                       LABEL;
                       (void)printf("owner=%u", p->fts_statb.st_uid);
               }
               if (p->fts_statb.st_gid != gid || notset) {
                       LABEL;
                       (void)printf("group=%u", p->fts_statb.st_gid);
               }
               if ((p->fts_statb.st_mode & MBITS) != mode || notset) {
                       LABEL;
                       (void)printf("mode=%#o", p->fts_statb.st_mode & MBITS);
               }
               if (p->fts_statb.st_nlink != 1 || notset) {
                       LABEL;
                       (void)printf("nlink=%u", p->fts_statb.st_nlink);
               }
               LABEL;
               (void)printf("size=%ld", p->fts_statb.st_size);

               if (p->fts_info == FTS_SL || p->fts_info == FTS_SLNONE) {
                       LABEL;
                       (void)printf("link=%s", rlink(p->fts_accpath));
               }
               (void)putchar('\n');
       }
}

#define MAXGID  5000
#define MAXUID  5000
#define MAXMODE MBITS + 1

static
statdir(t, parent, puid, pgid, pmode, tabs)
       FTS *t;
       FTSENT *parent;
       uid_t *puid;
       gid_t *pgid;
       mode_t *pmode;
       int *tabs;
{
       register FTSENT *p;
       register gid_t gid;
       register uid_t uid;
       register mode_t mode;
       gid_t savegid;
       uid_t saveuid;
       mode_t savemode;
       u_short maxgid, maxuid, maxmode, g[MAXGID], u[MAXUID], m[MAXMODE];

       if (!(p = ftschildren(t))) {
               if (errno) {
                       (void)fprintf(stderr, "mtree: %s: %s.\n",
                           RP(parent), strerror(errno));
                       exit(1);
               }
               return(1);
       }

       bzero(g, sizeof(g));
       bzero(u, sizeof(u));
       bzero(m, sizeof(m));

       *tabs = 1;
       maxuid = maxgid = maxmode = 0;
       for (; p; p = p->fts_link) {
               mode = p->fts_statb.st_mode & MBITS;
               if (mode < MAXMODE && ++m[mode] > maxmode) {
                       savemode = mode;
                       maxmode = m[mode];
               }
               gid = p->fts_statb.st_gid;
               if (gid < MAXGID && ++g[gid] > maxgid) {
                       savegid = gid;
                       maxgid = g[gid];
               }
               uid = p->fts_statb.st_uid;
               if (uid < MAXUID && ++u[uid] > maxuid) {
                       saveuid = uid;
                       maxuid = u[uid];
               }
               if (p->fts_namelen > 7)
                       *tabs = 2;
       }
       (void)printf("\n/set group=%u mode=%#o nlink=1 owner=%u type=file\n",
           savegid, savemode, saveuid);
       *puid = saveuid;
       *pgid = savegid;
       *pmode = savemode;
       return(0);
}

static
dsort(p1, p2)
       FTSENT **p1, **p2;
{
       register FTSENT *a, *b;

       a = *p1;
       b = *p2;

       if (S_ISDIR(a->fts_statb.st_mode)) {
               if (!S_ISDIR(b->fts_statb.st_mode))
                       return(1);
       } else if (S_ISDIR(b->fts_statb.st_mode))
               return(-1);
       return(strcmp(a->fts_name, b->fts_name));
}