mtree/create.c

   1 /*-
   2  * Copyright (c) 1989, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. Neither the name of the University nor the names of its contributors
  14  *    may be used to endorse or promote products derived from this software
  15  *    without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  27  * SUCH DAMAGE.
  28  */
  29
  30 #if 0
  31 #ifndef lint
  32 static char sccsid[] = "@(#)create.c    8.1 (Berkeley) 6/6/93";
  33 #endif /* not lint */
  34 #endif
  35 #include <sys/cdefs.h>
  36 __FBSDID("$FreeBSD: src/usr.sbin/mtree/create.c,v 1.37 2005/03/29 11:44:17 tobez Exp $");
  37
  38 #include <sys/param.h>
  39 #include <sys/stat.h>
  40 #include <dirent.h>
  41 #include <err.h>
  42 #include <errno.h>
  43 #include <fcntl.h>
  44 #include <fts.h>
  45 #include <grp.h>
  46 #ifdef ENABLE_MD5
  47 #include <md5.h>
  48 #endif
  49 #ifdef ENABLE_SHA1
  50 #include <sha.h>
  51 #endif
  52 #ifdef ENABLE_RMD160
  53 #include <ripemd.h>
  54 #endif
  55 #ifdef ENABLE_SHA256
  56 #include <sha256.h>
  57 #endif
  58 #include <pwd.h>
  59 #include <stdint.h>
  60 #include <stdio.h>
  61 #include <time.h>
  62 #include <unistd.h>
  63 #include <vis.h>
  64 #include "mtree.h"
  65 #include "extern.h"
  66
  67 #define INDENTNAMELEN   15
  68 #define MAXLINELEN      80
  69
  70 static gid_t gid;
  71 static uid_t uid;
  72 static mode_t mode;
  73 static u_long flags = 0xffffffff;
  74
  75 static int      dsort(const FTSENT * const *, const FTSENT * const *);
  76 static void     output(int, int *, const char *, ...) __printflike(3, 4);
  77 static int      statd(FTS *, FTSENT *, uid_t *, gid_t *, mode_t *, u_long *);
  78 static void     statf(int, FTSENT *);
  79
  80 void
  81 cwalk(void)
  82 {
  83         FTS *t;
  84         FTSENT *p;
  85         time_t cl;
  86         char *argv[2], host[MAXHOSTNAMELEN];
  87         char dot[] = ".";
  88         int indent = 0;
  89
  90         if (!nflag) {
  91                 (void)time(&cl);
  92                 (void)gethostname(host, sizeof(host));
  93                 (void)printf(
  94                     "#\t   user: %s\n#\tmachine: %s\n",
  95                     getlogin(), host);
  96                 (void)printf(
  97                     "#\t   tree: %s\n#\t   date: %s",
  98                     fullpath, ctime(&cl));
  99         }
 100
 101         argv[0] = dot;
 102         argv[1] = NULL;
 103         if ((t = fts_open(argv, ftsoptions, dsort)) == NULL)
 104                 err(1, "fts_open()");
 105         while ((p = fts_read(t))) {
 106                 if (iflag)
 107                         indent = p->fts_level * 4;
 108                 if (check_excludes(p->fts_name, p->fts_path)) {
 109                         fts_set(t, p, FTS_SKIP);
 110                         continue;
 111                 }
 112                 switch(p->fts_info) {
 113                 case FTS_D:
 114                         if (!dflag)
 115                                 (void)printf("\n");
 116                         if (!nflag)
 117                                 (void)printf("# %s\n", p->fts_path);
 118                         statd(t, p, &uid, &gid, &mode, &flags);
 119                         statf(indent, p);
 120                         break;
 121                 case FTS_DP:
 122                         if (!nflag && (p->fts_level > 0))
 123                                 (void)printf("%*s# %s\n", indent, "", p->fts_path);
 124                         (void)printf("%*s..\n", indent, "");
 125                         if (!dflag)
 126                                 (void)printf("\n");
 127                         break;
 128                 case FTS_DNR:
 129                 case FTS_ERR:
 130                 case FTS_NS:
 131                         warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
 132                         break;
 133                 default:
 134                         if (!dflag)
 135                                 statf(indent, p);
 136                         break;
 137
 138                 }
 139         }
 140         (void)fts_close(t);
 141         if (sflag && keys & F_CKSUM)
 142                 warnx("%s checksum: %lu", fullpath, (unsigned long)crc_total);
 143 }
 144
 145 static void
 146 statf(int indent, FTSENT *p)
 147 {
 148         struct group *gr;
 149         struct passwd *pw;
 150         uint32_t val;
 151         off_t len;
 152         int fd, offset;
 153         char *fflags;
 154         char *escaped_name;
 155
 156         escaped_name = calloc(1, p->fts_namelen * 4  +  1);
 157         if (escaped_name == NULL)
 158                 errx(1, "statf(): calloc() failed");
 159         strvis(escaped_name, p->fts_name, VIS_WHITE | VIS_OCTAL | VIS_GLOB);
 160
 161         if (iflag || S_ISDIR(p->fts_statp->st_mode))
 162                 offset = printf("%*s%s", indent, "", escaped_name);
 163         else
 164                 offset = printf("%*s    %s", indent, "", escaped_name);
 165
 166         free(escaped_name);
 167
 168         if (offset > (INDENTNAMELEN + indent))
 169                 offset = MAXLINELEN;
 170         else
 171                 offset += printf("%*s", (INDENTNAMELEN + indent) - offset, "");
 172
 173         if (!S_ISREG(p->fts_statp->st_mode) && !dflag)
 174                 output(indent, &offset, "type=%s", inotype(p->fts_statp->st_mode));
 175         if (p->fts_statp->st_uid != uid) {
 176                 if (keys & F_UNAME) {
 177                         pw = getpwuid(p->fts_statp->st_uid);
 178                         if (pw != NULL)
 179                                 output(indent, &offset, "uname=%s", pw->pw_name);
 180                         else if (wflag)
 181                                 warnx("Could not get uname for uid=%u",
 182                                     p->fts_statp->st_uid);
 183                         else
 184                                 errx(1,
 185                                     "Could not get uname for uid=%u",
 186                                     p->fts_statp->st_uid);
 187                 }
 188                 if (keys & F_UID)
 189                         output(indent, &offset, "uid=%u", p->fts_statp->st_uid);
 190         }
 191         if (p->fts_statp->st_gid != gid) {
 192                 if (keys & F_GNAME) {
 193                         gr = getgrgid(p->fts_statp->st_gid);
 194                         if (gr != NULL)
 195                                 output(indent, &offset, "gname=%s", gr->gr_name);
 196                         else if (wflag)
 197                                 warnx("Could not get gname for gid=%u",
 198                                     p->fts_statp->st_gid);
 199                         else
 200                                 errx(1,
 201                                     "Could not get gname for gid=%u",
 202                                     p->fts_statp->st_gid);
 203                 }
 204                 if (keys & F_GID)
 205                         output(indent, &offset, "gid=%u", p->fts_statp->st_gid);
 206         }
 207         if (keys & F_MODE && (p->fts_statp->st_mode & MBITS) != mode)
 208                 output(indent, &offset, "mode=%#o", p->fts_statp->st_mode & MBITS);
 209         if (keys & F_NLINK && p->fts_statp->st_nlink != 1)
 210                 output(indent, &offset, "nlink=%u", p->fts_statp->st_nlink);
 211         if (keys & F_SIZE)
 212                 output(indent, &offset, "size=%jd",
 213                     (intmax_t)p->fts_statp->st_size);
 214         if (keys & F_TIME)
 215                 output(indent, &offset, "time=%ld.%ld",
 216                     (long)p->fts_statp->st_mtimespec.tv_sec,
 217                     p->fts_statp->st_mtimespec.tv_nsec);
 218         if (keys & F_CKSUM && S_ISREG(p->fts_statp->st_mode)) {
 219                 if ((fd = open(p->fts_accpath, O_RDONLY, 0)) < 0 ||
 220                     crc(fd, &val, &len))
 221                         err(1, "%s", p->fts_accpath);
 222                 (void)close(fd);
 223                 output(indent, &offset, "cksum=%lu", (unsigned long)val);
 224         }
 225 #ifdef ENABLE_MD5
 226         if (keys & F_MD5 && S_ISREG(p->fts_statp->st_mode)) {
 227                 char *digest, buf[33];
 228
 229                 digest = MD5File(p->fts_accpath, buf);
 230                 if (!digest)
 231                         err(1, "%s", p->fts_accpath);
 232                 output(indent, &offset, "md5digest=%s", digest);
 233         }
 234 #endif /* ENABLE_MD5 */
 235 #ifdef ENABLE_SHA1
 236         if (keys & F_SHA1 && S_ISREG(p->fts_statp->st_mode)) {
 237                 char *digest, buf[41];
 238
 239                 digest = SHA1_File(p->fts_accpath, buf);
 240                 if (!digest)
 241                         err(1, "%s", p->fts_accpath);
 242                 output(indent, &offset, "sha1digest=%s", digest);
 243         }
 244 #endif /* ENABLE_SHA1 */
 245 #ifdef ENABLE_RMD160
 246         if (keys & F_RMD160 && S_ISREG(p->fts_statp->st_mode)) {
 247                 char *digest, buf[41];
 248
 249                 digest = RIPEMD160_File(p->fts_accpath, buf);
 250                 if (!digest)
 251                         err(1, "%s", p->fts_accpath);
 252                 output(indent, &offset, "ripemd160digest=%s", digest);
 253         }
 254 #endif /* ENABLE_RMD160 */
 255 #ifdef ENABLE_SHA256
 256         if (keys & F_SHA256 && S_ISREG(p->fts_statp->st_mode)) {
 257                 char *digest, buf[65];
 258
 259                 digest = SHA256_File(p->fts_accpath, buf);
 260                 if (!digest)
 261                         err(1, "%s", p->fts_accpath);
 262                 output(indent, &offset, "sha256digest=%s", digest);
 263         }
 264 #endif /* ENABLE_SHA256 */
 265         if (keys & F_SLINK &&
 266             (p->fts_info == FTS_SL || p->fts_info == FTS_SLNONE))
 267                 output(indent, &offset, "link=%s", rlink(p->fts_accpath));
 268         if (keys & F_FLAGS && p->fts_statp->st_flags != flags) {
 269                 fflags = flags_to_string(p->fts_statp->st_flags);
 270                 output(indent, &offset, "flags=%s", fflags);
 271                 free(fflags);
 272         }
 273         (void)putchar('\n');
 274 }
 275
 276 #define MAXGID  5000
 277 #define MAXUID  5000
 278 #define MAXMODE MBITS + 1
 279 #define MAXFLAGS 256
 280 #define MAXS 16
 281
 282 static int
 283 statd(FTS *t, FTSENT *parent, uid_t *puid, gid_t *pgid, mode_t *pmode, u_long *pflags)
 284 {
 285         FTSENT *p;
 286         gid_t sgid;
 287         uid_t suid;
 288         mode_t smode;
 289         u_long sflags;
 290         struct group *gr;
 291         struct passwd *pw;
 292         gid_t savegid = *pgid;
 293         uid_t saveuid = *puid;
 294         mode_t savemode = *pmode;
 295         u_long saveflags = *pflags;
 296         u_short maxgid, maxuid, maxmode, maxflags;
 297         u_short g[MAXGID], u[MAXUID], m[MAXMODE], f[MAXFLAGS];
 298         char *fflags;
 299         static int first = 1;
 300
 301         if ((p = fts_children(t, 0)) == NULL) {
 302                 if (errno)
 303                         err(1, "%s", RP(parent));
 304                 return (1);
 305         }
 306
 307         bzero(g, sizeof(g));
 308         bzero(u, sizeof(u));
 309         bzero(m, sizeof(m));
 310         bzero(f, sizeof(f));
 311
 312         maxuid = maxgid = maxmode = maxflags = 0;
 313         for (; p; p = p->fts_link) {
 314                 if (!dflag || (dflag && S_ISDIR(p->fts_statp->st_mode))) {
 315                         smode = p->fts_statp->st_mode & MBITS;
 316                         if (smode < MAXMODE && ++m[smode] > maxmode) {
 317                                 savemode = smode;
 318                                 maxmode = m[smode];
 319                         }
 320                         sgid = p->fts_statp->st_gid;
 321                         if (sgid < MAXGID && ++g[sgid] > maxgid) {
 322                                 savegid = sgid;
 323                                 maxgid = g[sgid];
 324                         }
 325                         suid = p->fts_statp->st_uid;
 326                         if (suid < MAXUID && ++u[suid] > maxuid) {
 327                                 saveuid = suid;
 328                                 maxuid = u[suid];
 329                         }
 330
 331                         /*
 332                          * XXX
 333                          * note that the below will break when file flags
 334                          * are extended beyond the first 4 bytes of each
 335                          * half word of the flags
 336                          */
 337 #define FLAGS2IDX(f) ((f & 0xf) | ((f >> 12) & 0xf0))
 338                         sflags = p->fts_statp->st_flags;
 339                         if (FLAGS2IDX(sflags) < MAXFLAGS &&
 340                             ++f[FLAGS2IDX(sflags)] > maxflags) {
 341                                 saveflags = sflags;
 342                                 maxflags = f[FLAGS2IDX(sflags)];
 343                         }
 344                 }
 345         }
 346         /*
 347          * If the /set record is the same as the last one we do not need to output
 348          * a new one.  So first we check to see if anything changed.  Note that we
 349          * always output a /set record for the first directory.
 350          */
 351         if ((((keys & F_UNAME) | (keys & F_UID)) && (*puid != saveuid)) ||
 352             (((keys & F_GNAME) | (keys & F_GID)) && (*pgid != savegid)) ||
 353             ((keys & F_MODE) && (*pmode != savemode)) ||
 354             ((keys & F_FLAGS) && (*pflags != saveflags)) ||
 355             (first)) {
 356                 first = 0;
 357                 if (dflag)
 358                         (void)printf("/set type=dir");
 359                 else
 360                         (void)printf("/set type=file");
 361                 if (keys & F_UNAME) {
 362                         pw = getpwuid(saveuid);
 363                         if (pw != NULL)
 364                                 (void)printf(" uname=%s", pw->pw_name);
 365                         else if (wflag)
 366                                 warnx( "Could not get uname for uid=%u", saveuid);
 367                         else
 368                                 errx(1, "Could not get uname for uid=%u", saveuid);
 369                 }
 370                 if (keys & F_UID)
 371                         (void)printf(" uid=%lu", (u_long)saveuid);
 372                 if (keys & F_GNAME) {
 373                         gr = getgrgid(savegid);
 374                         if (gr != NULL)
 375                                 (void)printf(" gname=%s", gr->gr_name);
 376                         else if (wflag)
 377                                 warnx("Could not get gname for gid=%u", savegid);
 378                         else
 379                                 errx(1, "Could not get gname for gid=%u", savegid);
 380                 }
 381                 if (keys & F_GID)
 382                         (void)printf(" gid=%lu", (u_long)savegid);
 383                 if (keys & F_MODE)
 384                         (void)printf(" mode=%#o", savemode);
 385                 if (keys & F_NLINK)
 386                         (void)printf(" nlink=1");
 387                 if (keys & F_FLAGS) {
 388                         fflags = flags_to_string(saveflags);
 389                         (void)printf(" flags=%s", fflags);
 390                         free(fflags);
 391                 }
 392                 (void)printf("\n");
 393                 *puid = saveuid;
 394                 *pgid = savegid;
 395                 *pmode = savemode;
 396                 *pflags = saveflags;
 397         }
 398         return (0);
 399 }
 400
 401 static int
 402 dsort(const FTSENT * const *a, const FTSENT * const *b)
 403 {
 404         if (S_ISDIR((*a)->fts_statp->st_mode)) {
 405                 if (!S_ISDIR((*b)->fts_statp->st_mode))
 406                         return (1);
 407         } else if (S_ISDIR((*b)->fts_statp->st_mode))
 408                 return (-1);
 409         return (strcmp((*a)->fts_name, (*b)->fts_name));
 410 }
 411
 412 #include <stdarg.h>
 413
 414 void
 415 output(int indent, int *offset, const char *fmt, ...)
 416 {
 417         va_list ap;
 418         char buf[1024];
 419         va_start(ap, fmt);
 420         (void)vsnprintf(buf, sizeof(buf), fmt, ap);
 421         va_end(ap);
 422
 423         if (*offset + strlen(buf) > MAXLINELEN - 3) {
 424                 (void)printf(" \\\n%*s", INDENTNAMELEN + indent, "");
 425                 *offset = INDENTNAMELEN + indent;
 426         }
 427         *offset += printf(" %s", buf) + 1;
 428 }