usr.bin/xinstall/xinstall.c

   1 /*
   2  * Copyright (c) 2012, 2013 SRI International
   3  * Copyright (c) 1987, 1993
   4  *      The Regents of the University of California.  All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  * 4. Neither the name of the University nor the names of its contributors
  15  *    may be used to endorse or promote products derived from this software
  16  *    without specific prior written permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  19  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  22  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  23  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  24  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  25  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  27  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  28  * SUCH DAMAGE.
  29  */
  30
  31 #ifndef lint
  32 static const char copyright[] =
  33 "@(#) Copyright (c) 1987, 1993\n\
  34         The Regents of the University of California.  All rights reserved.\n";
  35 #endif /* not lint */
  36
  37 #if 0
  38 #ifndef lint
  39 static char sccsid[] = "@(#)xinstall.c  8.1 (Berkeley) 7/21/93";
  40 #endif /* not lint */
  41 #endif
  42
  43 #include <sys/cdefs.h>
  44 __FBSDID("$FreeBSD$");
  45
  46 #include <sys/param.h>
  47 #include <sys/mman.h>
  48 #include <sys/mount.h>
  49 #include <sys/stat.h>
  50 #include <sys/time.h>
  51 #include <sys/wait.h>
  52
  53 #include <err.h>
  54 #include <errno.h>
  55 #include <fcntl.h>
  56 #include <grp.h>
  57 #include <libgen.h>
  58 #include <md5.h>
  59 #include <paths.h>
  60 #include <pwd.h>
  61 #include <ripemd.h>
  62 #include <sha.h>
  63 #include <sha256.h>
  64 #include <sha512.h>
  65 #include <spawn.h>
  66 #include <stdint.h>
  67 #include <stdio.h>
  68 #include <stdlib.h>
  69 #include <string.h>
  70 #include <sysexits.h>
  71 #include <unistd.h>
  72 #include <vis.h>
  73
  74 #include "mtree.h"
  75
  76 #define MAX_CMP_SIZE    (16 * 1024 * 1024)
  77
  78 #define LN_ABSOLUTE     0x01
  79 #define LN_RELATIVE     0x02
  80 #define LN_HARD         0x04
  81 #define LN_SYMBOLIC     0x08
  82 #define LN_MIXED        0x10
  83
  84 #define DIRECTORY       0x01            /* Tell install it's a directory. */
  85 #define SETFLAGS        0x02            /* Tell install to set flags. */
  86 #define NOCHANGEBITS    (UF_IMMUTABLE | UF_APPEND | SF_IMMUTABLE | SF_APPEND)
  87 #define BACKUP_SUFFIX   ".old"
  88
  89 typedef union {
  90         MD5_CTX         MD5;
  91         RIPEMD160_CTX   RIPEMD160;
  92         SHA1_CTX        SHA1;
  93         SHA256_CTX      SHA256;
  94         SHA512_CTX      SHA512;
  95 }       DIGEST_CTX;
  96
  97 static enum {
  98         DIGEST_NONE = 0,
  99         DIGEST_MD5,
 100         DIGEST_RIPEMD160,
 101         DIGEST_SHA1,
 102         DIGEST_SHA256,
 103         DIGEST_SHA512,
 104 } digesttype = DIGEST_NONE;
 105
 106 extern char **environ;
 107
 108 static gid_t gid;
 109 static uid_t uid;
 110 static int dobackup, docompare, dodir, dolink, dopreserve, dostrip, dounpriv,
 111     safecopy, verbose;
 112 static int haveopt_f, haveopt_g, haveopt_m, haveopt_o;
 113 static mode_t mode = S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH;
 114 static FILE *metafp;
 115 static const char *group, *owner;
 116 static const char *suffix = BACKUP_SUFFIX;
 117 static char *destdir, *digest, *fflags, *metafile, *tags;
 118
 119 static int      compare(int, const char *, size_t, int, const char *, size_t,
 120                     char **);
 121 static char     *copy(int, const char *, int, const char *, off_t);
 122 static int      create_newfile(const char *, int, struct stat *);
 123 static int      create_tempfile(const char *, char *, size_t);
 124 static char     *quiet_mktemp(char *template);
 125 static char     *digest_file(const char *);
 126 static void     digest_init(DIGEST_CTX *);
 127 static void     digest_update(DIGEST_CTX *, const char *, size_t);
 128 static char     *digest_end(DIGEST_CTX *, char *);
 129 static int      do_link(const char *, const char *, const struct stat *);
 130 static void     do_symlink(const char *, const char *, const struct stat *);
 131 static void     makelink(const char *, const char *, const struct stat *);
 132 static void     install(const char *, const char *, u_long, u_int);
 133 static void     install_dir(char *);
 134 static void     metadata_log(const char *, const char *, struct timeval *,
 135                     const char *, const char *, off_t);
 136 static int      parseid(const char *, id_t *);
 137 static void     strip(const char *);
 138 static int      trymmap(int);
 139 static void     usage(void);
 140
 141 int
 142 main(int argc, char *argv[])
 143 {
 144         struct stat from_sb, to_sb;
 145         mode_t *set;
 146         u_long fset;
 147         int ch, no_target;
 148         u_int iflags;
 149         char *p;
 150         const char *to_name;
 151
 152         iflags = 0;
 153         group = owner = NULL;
 154         while ((ch = getopt(argc, argv, "B:bCcD:df:g:h:l:M:m:N:o:pSsT:Uv")) !=
 155              -1)
 156                 switch((char)ch) {
 157                 case 'B':
 158                         suffix = optarg;
 159                         /* FALLTHROUGH */
 160                 case 'b':
 161                         dobackup = 1;
 162                         break;
 163                 case 'C':
 164                         docompare = 1;
 165                         break;
 166                 case 'c':
 167                         /* For backwards compatibility. */
 168                         break;
 169                 case 'D':
 170                         destdir = optarg;
 171                         break;
 172                 case 'd':
 173                         dodir = 1;
 174                         break;
 175                 case 'f':
 176                         haveopt_f = 1;
 177                         fflags = optarg;
 178                         break;
 179                 case 'g':
 180                         haveopt_g = 1;
 181                         group = optarg;
 182                         break;
 183                 case 'h':
 184                         digest = optarg;
 185                         break;
 186                 case 'l':
 187                         for (p = optarg; *p != '\0'; p++)
 188                                 switch (*p) {
 189                                 case 's':
 190                                         dolink &= ~(LN_HARD|LN_MIXED);
 191                                         dolink |= LN_SYMBOLIC;
 192                                         break;
 193                                 case 'h':
 194                                         dolink &= ~(LN_SYMBOLIC|LN_MIXED);
 195                                         dolink |= LN_HARD;
 196                                         break;
 197                                 case 'm':
 198                                         dolink &= ~(LN_SYMBOLIC|LN_HARD);
 199                                         dolink |= LN_MIXED;
 200                                         break;
 201                                 case 'a':
 202                                         dolink &= ~LN_RELATIVE;
 203                                         dolink |= LN_ABSOLUTE;
 204                                         break;
 205                                 case 'r':
 206                                         dolink &= ~LN_ABSOLUTE;
 207                                         dolink |= LN_RELATIVE;
 208                                         break;
 209                                 default:
 210                                         errx(1, "%c: invalid link type", *p);
 211                                         /* NOTREACHED */
 212                                 }
 213                         break;
 214                 case 'M':
 215                         metafile = optarg;
 216                         break;
 217                 case 'm':
 218                         haveopt_m = 1;
 219                         if (!(set = setmode(optarg)))
 220                                 errx(EX_USAGE, "invalid file mode: %s",
 221                                      optarg);
 222                         mode = getmode(set, 0);
 223                         free(set);
 224                         break;
 225                 case 'N':
 226                         if (!setup_getid(optarg))
 227                                 err(EX_OSERR, "Unable to use user and group "
 228                                     "databases in `%s'", optarg);
 229                         break;
 230                 case 'o':
 231                         haveopt_o = 1;
 232                         owner = optarg;
 233                         break;
 234                 case 'p':
 235                         docompare = dopreserve = 1;
 236                         break;
 237                 case 'S':
 238                         safecopy = 1;
 239                         break;
 240                 case 's':
 241                         dostrip = 1;
 242                         break;
 243                 case 'T':
 244                         tags = optarg;
 245                         break;
 246                 case 'U':
 247                         dounpriv = 1;
 248                         break;
 249                 case 'v':
 250                         verbose = 1;
 251                         break;
 252                 case '?':
 253                 default:
 254                         usage();
 255                 }
 256         argc -= optind;
 257         argv += optind;
 258
 259         /* some options make no sense when creating directories */
 260         if (dostrip && dodir) {
 261                 warnx("-d and -s may not be specified together");
 262                 usage();
 263         }
 264
 265         if (getenv("DONTSTRIP") != NULL) {
 266                 warnx("DONTSTRIP set - will not strip installed binaries");
 267                 dostrip = 0;
 268         }
 269
 270         /* must have at least two arguments, except when creating directories */
 271         if (argc == 0 || (argc == 1 && !dodir))
 272                 usage();
 273
 274         if (digest != NULL) {
 275                 if (strcmp(digest, "none") == 0) {
 276                         digesttype = DIGEST_NONE;
 277                 } else if (strcmp(digest, "md5") == 0) {
 278                        digesttype = DIGEST_MD5;
 279                 } else if (strcmp(digest, "rmd160") == 0) {
 280                         digesttype = DIGEST_RIPEMD160;
 281                 } else if (strcmp(digest, "sha1") == 0) {
 282                         digesttype = DIGEST_SHA1;
 283                 } else if (strcmp(digest, "sha256") == 0) {
 284                         digesttype = DIGEST_SHA256;
 285                 } else if (strcmp(digest, "sha512") == 0) {
 286                         digesttype = DIGEST_SHA512;
 287                 } else {
 288                         warnx("unknown digest `%s'", digest);
 289                         usage();
 290                 }
 291         }
 292
 293         /* need to make a temp copy so we can compare stripped version */
 294         if (docompare && dostrip)
 295                 safecopy = 1;
 296
 297         /* get group and owner id's */
 298         if (group != NULL && !dounpriv) {
 299                 if (gid_from_group(group, &gid) == -1) {
 300                         id_t id;
 301                         if (!parseid(group, &id))
 302                                 errx(1, "unknown group %s", group);
 303                         gid = id;
 304                 }
 305         } else
 306                 gid = (gid_t)-1;
 307
 308         if (owner != NULL && !dounpriv) {
 309                 if (uid_from_user(owner, &uid) == -1) {
 310                         id_t id;
 311                         if (!parseid(owner, &id))
 312                                 errx(1, "unknown user %s", owner);
 313                         uid = id;
 314                 }
 315         } else
 316                 uid = (uid_t)-1;
 317
 318         if (fflags != NULL && !dounpriv) {
 319                 if (strtofflags(&fflags, &fset, NULL))
 320                         errx(EX_USAGE, "%s: invalid flag", fflags);
 321                 iflags |= SETFLAGS;
 322         }
 323
 324         if (metafile != NULL) {
 325                 if ((metafp = fopen(metafile, "a")) == NULL)
 326                         warn("open %s", metafile);
 327         } else
 328                 digesttype = DIGEST_NONE;
 329
 330         if (dodir) {
 331                 for (; *argv != NULL; ++argv)
 332                         install_dir(*argv);
 333                 exit(EX_OK);
 334                 /* NOTREACHED */
 335         }
 336
 337         to_name = argv[argc - 1];
 338         no_target = stat(to_name, &to_sb);
 339         if (!no_target && S_ISDIR(to_sb.st_mode)) {
 340                 if (dolink & LN_SYMBOLIC) {
 341                         if (lstat(to_name, &to_sb) != 0)
 342                                 err(EX_OSERR, "%s vanished", to_name);
 343                         if (S_ISLNK(to_sb.st_mode)) {
 344                                 if (argc != 2) {
 345                                         errno = ENOTDIR;
 346                                         err(EX_USAGE, "%s", to_name);
 347                                 }
 348                                 install(*argv, to_name, fset, iflags);
 349                                 exit(EX_OK);
 350                         }
 351                 }
 352                 for (; *argv != to_name; ++argv)
 353                         install(*argv, to_name, fset, iflags | DIRECTORY);
 354                 exit(EX_OK);
 355                 /* NOTREACHED */
 356         }
 357
 358         /* can't do file1 file2 directory/file */
 359         if (argc != 2) {
 360                 if (no_target)
 361                         warnx("target directory `%s' does not exist",
 362                             argv[argc - 1]);
 363                 else
 364                         warnx("target `%s' is not a directory",
 365                             argv[argc - 1]);
 366                 usage();
 367         }
 368
 369         if (!no_target && !dolink) {
 370                 if (stat(*argv, &from_sb))
 371                         err(EX_OSERR, "%s", *argv);
 372                 if (!S_ISREG(to_sb.st_mode)) {
 373                         errno = EFTYPE;
 374                         err(EX_OSERR, "%s", to_name);
 375                 }
 376                 if (to_sb.st_dev == from_sb.st_dev &&
 377                     to_sb.st_ino == from_sb.st_ino)
 378                         errx(EX_USAGE,
 379                             "%s and %s are the same file", *argv, to_name);
 380         }
 381         install(*argv, to_name, fset, iflags);
 382         exit(EX_OK);
 383         /* NOTREACHED */
 384 }
 385
 386 static char *
 387 digest_file(const char *name)
 388 {
 389
 390         switch (digesttype) {
 391         case DIGEST_MD5:
 392                 return (MD5File(name, NULL));
 393         case DIGEST_RIPEMD160:
 394                 return (RIPEMD160_File(name, NULL));
 395         case DIGEST_SHA1:
 396                 return (SHA1_File(name, NULL));
 397         case DIGEST_SHA256:
 398                 return (SHA256_File(name, NULL));
 399         case DIGEST_SHA512:
 400                 return (SHA512_File(name, NULL));
 401         default:
 402                 return (NULL);
 403         }
 404 }
 405
 406 static void
 407 digest_init(DIGEST_CTX *c)
 408 {
 409
 410         switch (digesttype) {
 411         case DIGEST_NONE:
 412                 break;
 413         case DIGEST_MD5:
 414                 MD5Init(&(c->MD5));
 415                 break;
 416         case DIGEST_RIPEMD160:
 417                 RIPEMD160_Init(&(c->RIPEMD160));
 418                 break;
 419         case DIGEST_SHA1:
 420                 SHA1_Init(&(c->SHA1));
 421                 break;
 422         case DIGEST_SHA256:
 423                 SHA256_Init(&(c->SHA256));
 424                 break;
 425         case DIGEST_SHA512:
 426                 SHA512_Init(&(c->SHA512));
 427                 break;
 428         }
 429 }
 430
 431 static void
 432 digest_update(DIGEST_CTX *c, const char *data, size_t len)
 433 {
 434
 435         switch (digesttype) {
 436         case DIGEST_NONE:
 437                 break;
 438         case DIGEST_MD5:
 439                 MD5Update(&(c->MD5), data, len);
 440                 break;
 441         case DIGEST_RIPEMD160:
 442                 RIPEMD160_Update(&(c->RIPEMD160), data, len);
 443                 break;
 444         case DIGEST_SHA1:
 445                 SHA1_Update(&(c->SHA1), data, len);
 446                 break;
 447         case DIGEST_SHA256:
 448                 SHA256_Update(&(c->SHA256), data, len);
 449                 break;
 450         case DIGEST_SHA512:
 451                 SHA512_Update(&(c->SHA512), data, len);
 452                 break;
 453         }
 454 }
 455
 456 static char *
 457 digest_end(DIGEST_CTX *c, char *buf)
 458 {
 459
 460         switch (digesttype) {
 461         case DIGEST_MD5:
 462                 return (MD5End(&(c->MD5), buf));
 463         case DIGEST_RIPEMD160:
 464                 return (RIPEMD160_End(&(c->RIPEMD160), buf));
 465         case DIGEST_SHA1:
 466                 return (SHA1_End(&(c->SHA1), buf));
 467         case DIGEST_SHA256:
 468                 return (SHA256_End(&(c->SHA256), buf));
 469         case DIGEST_SHA512:
 470                 return (SHA512_End(&(c->SHA512), buf));
 471         default:
 472                 return (NULL);
 473         }
 474 }
 475
 476 /*
 477  * parseid --
 478  *      parse uid or gid from arg into id, returning non-zero if successful
 479  */
 480 static int
 481 parseid(const char *name, id_t *id)
 482 {
 483         char    *ep;
 484         errno = 0;
 485         *id = (id_t)strtoul(name, &ep, 10);
 486         if (errno || *ep != '\0')
 487                 return (0);
 488         return (1);
 489 }
 490
 491 /*
 492  * quiet_mktemp --
 493  *      mktemp implementation used mkstemp to avoid mktemp warnings.  We
 494  *      really do need mktemp semantics here as we will be creating a link.
 495  */
 496 static char *
 497 quiet_mktemp(char *template)
 498 {
 499         int fd;
 500
 501         if ((fd = mkstemp(template)) == -1)
 502                 return (NULL);
 503         close (fd);
 504         if (unlink(template) == -1)
 505                 err(EX_OSERR, "unlink %s", template);
 506         return (template);
 507 }
 508
 509 /*
 510  * do_link --
 511  *      make a hard link, obeying dorename if set
 512  *      return -1 on failure
 513  */
 514 static int
 515 do_link(const char *from_name, const char *to_name,
 516     const struct stat *target_sb)
 517 {
 518         char tmpl[MAXPATHLEN];
 519         int ret;
 520
 521         if (safecopy && target_sb != NULL) {
 522                 (void)snprintf(tmpl, sizeof(tmpl), "%s.inst.XXXXXX", to_name);
 523                 /* This usage is safe. */
 524                 if (quiet_mktemp(tmpl) == NULL)
 525                         err(EX_OSERR, "%s: mktemp", tmpl);
 526                 ret = link(from_name, tmpl);
 527                 if (ret == 0) {
 528                         if (target_sb->st_mode & S_IFDIR && rmdir(to_name) ==
 529                             -1) {
 530                                 unlink(tmpl);
 531                                 err(EX_OSERR, "%s", to_name);
 532                         }
 533                         if (target_sb->st_flags & NOCHANGEBITS)
 534                                 (void)chflags(to_name, target_sb->st_flags &
 535                                      ~NOCHANGEBITS);
 536                         unlink(to_name);
 537                         ret = rename(tmpl, to_name);
 538                         /*
 539                          * If rename has posix semantics, then the temporary
 540                          * file may still exist when from_name and to_name point
 541                          * to the same file, so unlink it unconditionally.
 542                          */
 543                         (void)unlink(tmpl);
 544                 }
 545                 return (ret);
 546         } else
 547                 return (link(from_name, to_name));
 548 }
 549
 550 /*
 551  * do_symlink --
 552  *      Make a symbolic link, obeying dorename if set. Exit on failure.
 553  */
 554 static void
 555 do_symlink(const char *from_name, const char *to_name,
 556     const struct stat *target_sb)
 557 {
 558         char tmpl[MAXPATHLEN];
 559
 560         if (safecopy && target_sb != NULL) {
 561                 (void)snprintf(tmpl, sizeof(tmpl), "%s.inst.XXXXXX", to_name);
 562                 /* This usage is safe. */
 563                 if (quiet_mktemp(tmpl) == NULL)
 564                         err(EX_OSERR, "%s: mktemp", tmpl);
 565
 566                 if (symlink(from_name, tmpl) == -1)
 567                         err(EX_OSERR, "symlink %s -> %s", from_name, tmpl);
 568
 569                 if (target_sb->st_mode & S_IFDIR && rmdir(to_name) == -1) {
 570                         (void)unlink(tmpl);
 571                         err(EX_OSERR, "%s", to_name);
 572                 }
 573                 if (target_sb->st_flags & NOCHANGEBITS)
 574                         (void)chflags(to_name, target_sb->st_flags &
 575                              ~NOCHANGEBITS);
 576                 unlink(to_name);
 577
 578                 if (rename(tmpl, to_name) == -1) {
 579                         /* Remove temporary link before exiting. */
 580                         (void)unlink(tmpl);
 581                         err(EX_OSERR, "%s: rename", to_name);
 582                 }
 583         } else {
 584                 if (symlink(from_name, to_name) == -1)
 585                         err(EX_OSERR, "symlink %s -> %s", from_name, to_name);
 586         }
 587 }
 588
 589 /*
 590  * makelink --
 591  *      make a link from source to destination
 592  */
 593 static void
 594 makelink(const char *from_name, const char *to_name,
 595     const struct stat *target_sb)
 596 {
 597         char    src[MAXPATHLEN], dst[MAXPATHLEN], lnk[MAXPATHLEN];
 598         struct stat     to_sb;
 599
 600         /* Try hard links first. */
 601         if (dolink & (LN_HARD|LN_MIXED)) {
 602                 if (do_link(from_name, to_name, target_sb) == -1) {
 603                         if ((dolink & LN_HARD) || errno != EXDEV)
 604                                 err(EX_OSERR, "link %s -> %s", from_name, to_name);
 605                 } else {
 606                         if (stat(to_name, &to_sb))
 607                                 err(EX_OSERR, "%s: stat", to_name);
 608                         if (S_ISREG(to_sb.st_mode)) {
 609                                 /*
 610                                  * XXX: hard links to anything other than
 611                                  * plain files are not metalogged
 612                                  */
 613                                 int omode;
 614                                 const char *oowner, *ogroup;
 615                                 char *offlags;
 616                                 char *dres;
 617
 618                                 /*
 619                                  * XXX: use underlying perms, unless
 620                                  * overridden on command line.
 621                                  */
 622                                 omode = mode;
 623                                 if (!haveopt_m)
 624                                         mode = (to_sb.st_mode & 0777);
 625                                 oowner = owner;
 626                                 if (!haveopt_o)
 627                                         owner = NULL;
 628                                 ogroup = group;
 629                                 if (!haveopt_g)
 630                                         group = NULL;
 631                                 offlags = fflags;
 632                                 if (!haveopt_f)
 633                                         fflags = NULL;
 634                                 dres = digest_file(from_name);
 635                                 metadata_log(to_name, "file", NULL, NULL,
 636                                     dres, to_sb.st_size);
 637                                 free(dres);
 638                                 mode = omode;
 639                                 owner = oowner;
 640                                 group = ogroup;
 641                                 fflags = offlags;
 642                         }
 643                         return;
 644                 }
 645         }
 646
 647         /* Symbolic links. */
 648         if (dolink & LN_ABSOLUTE) {
 649                 /* Convert source path to absolute. */
 650                 if (realpath(from_name, src) == NULL)
 651                         err(EX_OSERR, "%s: realpath", from_name);
 652                 do_symlink(src, to_name, target_sb);
 653                 /* XXX: src may point outside of destdir */
 654                 metadata_log(to_name, "link", NULL, src, NULL, 0);
 655                 return;
 656         }
 657
 658         if (dolink & LN_RELATIVE) {
 659                 char *cp, *d, *s;
 660
 661                 if (*from_name != '/') {
 662                         /* this is already a relative link */
 663                         do_symlink(from_name, to_name, target_sb);
 664                         /* XXX: from_name may point outside of destdir. */
 665                         metadata_log(to_name, "link", NULL, from_name, NULL, 0);
 666                         return;
 667                 }
 668
 669                 /* Resolve pathnames. */
 670                 if (realpath(from_name, src) == NULL)
 671                         err(EX_OSERR, "%s: realpath", from_name);
 672
 673                 /*
 674                  * The last component of to_name may be a symlink,
 675                  * so use realpath to resolve only the directory.
 676                  */
 677                 cp = dirname(to_name);
 678                 if (realpath(cp, dst) == NULL)
 679                         err(EX_OSERR, "%s: realpath", cp);
 680                 /* .. and add the last component. */
 681                 if (strcmp(dst, "/") != 0) {
 682                         if (strlcat(dst, "/", sizeof(dst)) > sizeof(dst))
 683                                 errx(1, "resolved pathname too long");
 684                 }
 685                 cp = basename(to_name);
 686                 if (strlcat(dst, cp, sizeof(dst)) > sizeof(dst))
 687                         errx(1, "resolved pathname too long");
 688
 689                 /* Trim common path components. */
 690                 for (s = src, d = dst; *s == *d; s++, d++)
 691                         continue;
 692                 while (*s != '/')
 693                         s--, d--;
 694
 695                 /* Count the number of directories we need to backtrack. */
 696                 for (++d, lnk[0] = '\0'; *d; d++)
 697                         if (*d == '/')
 698                                 (void)strlcat(lnk, "../", sizeof(lnk));
 699
 700                 (void)strlcat(lnk, ++s, sizeof(lnk));
 701
 702                 do_symlink(lnk, to_name, target_sb);
 703                 /* XXX: Link may point outside of destdir. */
 704                 metadata_log(to_name, "link", NULL, lnk, NULL, 0);
 705                 return;
 706         }
 707
 708         /*
 709          * If absolute or relative was not specified, try the names the
 710          * user provided.
 711          */
 712         do_symlink(from_name, to_name, target_sb);
 713         /* XXX: from_name may point outside of destdir. */
 714         metadata_log(to_name, "link", NULL, from_name, NULL, 0);
 715 }
 716
 717 /*
 718  * install --
 719  *      build a path name and install the file
 720  */
 721 static void
 722 install(const char *from_name, const char *to_name, u_long fset, u_int flags)
 723 {
 724         struct stat from_sb, temp_sb, to_sb;
 725         struct timeval tvb[2];
 726         int devnull, files_match, from_fd, serrno, target;
 727         int tempcopy, temp_fd, to_fd;
 728         char backup[MAXPATHLEN], *p, pathbuf[MAXPATHLEN], tempfile[MAXPATHLEN];
 729         char *digestresult;
 730
 731         files_match = 0;
 732         from_fd = -1;
 733         to_fd = -1;
 734
 735         /* If try to install NULL file to a directory, fails. */
 736         if (flags & DIRECTORY || strcmp(from_name, _PATH_DEVNULL)) {
 737                 if (!dolink) {
 738                         if (stat(from_name, &from_sb))
 739                                 err(EX_OSERR, "%s", from_name);
 740                         if (!S_ISREG(from_sb.st_mode)) {
 741                                 errno = EFTYPE;
 742                                 err(EX_OSERR, "%s", from_name);
 743                         }
 744                 }
 745                 /* Build the target path. */
 746                 if (flags & DIRECTORY) {
 747                         (void)snprintf(pathbuf, sizeof(pathbuf), "%s/%s",
 748                             to_name,
 749                             (p = strrchr(from_name, '/')) ? ++p : from_name);
 750                         to_name = pathbuf;
 751                 }
 752                 devnull = 0;
 753         } else {
 754                 devnull = 1;
 755         }
 756
 757         if (!dolink)
 758                 target = (stat(to_name, &to_sb) == 0);
 759         else
 760                 target = (lstat(to_name, &to_sb) == 0);
 761
 762         if (dolink) {
 763                 if (target && !safecopy) {
 764                         if (to_sb.st_mode & S_IFDIR && rmdir(to_name) == -1)
 765                                 err(EX_OSERR, "%s", to_name);
 766                         if (to_sb.st_flags & NOCHANGEBITS)
 767                                 (void)chflags(to_name,
 768                                     to_sb.st_flags & ~NOCHANGEBITS);
 769                         unlink(to_name);
 770                 }
 771                 makelink(from_name, to_name, target ? &to_sb : NULL);
 772                 return;
 773         }
 774
 775         /* Only install to regular files. */
 776         if (target && !S_ISREG(to_sb.st_mode)) {
 777                 errno = EFTYPE;
 778                 warn("%s", to_name);
 779                 return;
 780         }
 781
 782         /* Only copy safe if the target exists. */
 783         tempcopy = safecopy && target;
 784
 785         if (!devnull && (from_fd = open(from_name, O_RDONLY, 0)) < 0)
 786                 err(EX_OSERR, "%s", from_name);
 787
 788         /* If we don't strip, we can compare first. */
 789         if (docompare && !dostrip && target) {
 790                 if ((to_fd = open(to_name, O_RDONLY, 0)) < 0)
 791                         err(EX_OSERR, "%s", to_name);
 792                 if (devnull)
 793                         files_match = to_sb.st_size == 0;
 794                 else
 795                         files_match = !(compare(from_fd, from_name,
 796                             (size_t)from_sb.st_size, to_fd,
 797                             to_name, (size_t)to_sb.st_size, &digestresult));
 798
 799                 /* Close "to" file unless we match. */
 800                 if (!files_match)
 801                         (void)close(to_fd);
 802         }
 803
 804         if (!files_match) {
 805                 if (tempcopy) {
 806                         to_fd = create_tempfile(to_name, tempfile,
 807                             sizeof(tempfile));
 808                         if (to_fd < 0)
 809                                 err(EX_OSERR, "%s", tempfile);
 810                 } else {
 811                         if ((to_fd = create_newfile(to_name, target,
 812                             &to_sb)) < 0)
 813                                 err(EX_OSERR, "%s", to_name);
 814                         if (verbose)
 815                                 (void)printf("install: %s -> %s\n",
 816                                     from_name, to_name);
 817                 }
 818                 if (!devnull)
 819                         digestresult = copy(from_fd, from_name, to_fd,
 820                              tempcopy ? tempfile : to_name, from_sb.st_size);
 821                 else
 822                         digestresult = NULL;
 823         }
 824
 825         if (dostrip) {
 826                 strip(tempcopy ? tempfile : to_name);
 827
 828                 /*
 829                  * Re-open our fd on the target, in case we used a strip
 830                  * that does not work in-place -- like GNU binutils strip.
 831                  */
 832                 close(to_fd);
 833                 to_fd = open(tempcopy ? tempfile : to_name, O_RDONLY, 0);
 834                 if (to_fd < 0)
 835                         err(EX_OSERR, "stripping %s", to_name);
 836         }
 837
 838         /*
 839          * Compare the stripped temp file with the target.
 840          */
 841         if (docompare && dostrip && target) {
 842                 temp_fd = to_fd;
 843
 844                 /* Re-open to_fd using the real target name. */
 845                 if ((to_fd = open(to_name, O_RDONLY, 0)) < 0)
 846                         err(EX_OSERR, "%s", to_name);
 847
 848                 if (fstat(temp_fd, &temp_sb)) {
 849                         serrno = errno;
 850                         (void)unlink(tempfile);
 851                         errno = serrno;
 852                         err(EX_OSERR, "%s", tempfile);
 853                 }
 854
 855                 if (compare(temp_fd, tempfile, (size_t)temp_sb.st_size, to_fd,
 856                             to_name, (size_t)to_sb.st_size, &digestresult)
 857                             == 0) {
 858                         /*
 859                          * If target has more than one link we need to
 860                          * replace it in order to snap the extra links.
 861                          * Need to preserve target file times, though.
 862                          */
 863                         if (to_sb.st_nlink != 1) {
 864                                 tvb[0].tv_sec = to_sb.st_atime;
 865                                 tvb[0].tv_usec = 0;
 866                                 tvb[1].tv_sec = to_sb.st_mtime;
 867                                 tvb[1].tv_usec = 0;
 868                                 (void)utimes(tempfile, tvb);
 869                         } else {
 870                                 files_match = 1;
 871                                 (void)unlink(tempfile);
 872                         }
 873                         (void) close(temp_fd);
 874                 }
 875         }
 876
 877         if (dostrip && (!docompare || !target))
 878                 digestresult = digest_file(tempfile);
 879
 880         /*
 881          * Move the new file into place if doing a safe copy
 882          * and the files are different (or just not compared).
 883          */
 884         if (tempcopy && !files_match) {
 885                 /* Try to turn off the immutable bits. */
 886                 if (to_sb.st_flags & NOCHANGEBITS)
 887                         (void)chflags(to_name, to_sb.st_flags & ~NOCHANGEBITS);
 888                 if (dobackup) {
 889                         if ((size_t)snprintf(backup, MAXPATHLEN, "%s%s", to_name,
 890                             suffix) != strlen(to_name) + strlen(suffix)) {
 891                                 unlink(tempfile);
 892                                 errx(EX_OSERR, "%s: backup filename too long",
 893                                     to_name);
 894                         }
 895                         if (verbose)
 896                                 (void)printf("install: %s -> %s\n", to_name, backup);
 897                         if (rename(to_name, backup) < 0) {
 898                                 serrno = errno;
 899                                 unlink(tempfile);
 900                                 errno = serrno;
 901                                 err(EX_OSERR, "rename: %s to %s", to_name,
 902                                      backup);
 903                         }
 904                 }
 905                 if (verbose)
 906                         (void)printf("install: %s -> %s\n", from_name, to_name);
 907                 if (rename(tempfile, to_name) < 0) {
 908                         serrno = errno;
 909                         unlink(tempfile);
 910                         errno = serrno;
 911                         err(EX_OSERR, "rename: %s to %s",
 912                             tempfile, to_name);
 913                 }
 914
 915                 /* Re-open to_fd so we aren't hosed by the rename(2). */
 916                 (void) close(to_fd);
 917                 if ((to_fd = open(to_name, O_RDONLY, 0)) < 0)
 918                         err(EX_OSERR, "%s", to_name);
 919         }
 920
 921         /*
 922          * Preserve the timestamp of the source file if necessary.
 923          */
 924         if (dopreserve && !files_match && !devnull) {
 925                 tvb[0].tv_sec = from_sb.st_atime;
 926                 tvb[0].tv_usec = 0;
 927                 tvb[1].tv_sec = from_sb.st_mtime;
 928                 tvb[1].tv_usec = 0;
 929                 (void)utimes(to_name, tvb);
 930         }
 931
 932         if (fstat(to_fd, &to_sb) == -1) {
 933                 serrno = errno;
 934                 (void)unlink(to_name);
 935                 errno = serrno;
 936                 err(EX_OSERR, "%s", to_name);
 937         }
 938
 939         /*
 940          * Set owner, group, mode for target; do the chown first,
 941          * chown may lose the setuid bits.
 942          */
 943         if (!dounpriv && ((gid != (gid_t)-1 && gid != to_sb.st_gid) ||
 944             (uid != (uid_t)-1 && uid != to_sb.st_uid) ||
 945             (mode != (to_sb.st_mode & ALLPERMS)))) {
 946                 /* Try to turn off the immutable bits. */
 947                 if (to_sb.st_flags & NOCHANGEBITS)
 948                         (void)fchflags(to_fd, to_sb.st_flags & ~NOCHANGEBITS);
 949         }
 950
 951         if (!dounpriv &
 952             (gid != (gid_t)-1 && gid != to_sb.st_gid) ||
 953             (uid != (uid_t)-1 && uid != to_sb.st_uid))
 954                 if (fchown(to_fd, uid, gid) == -1) {
 955                         serrno = errno;
 956                         (void)unlink(to_name);
 957                         errno = serrno;
 958                         err(EX_OSERR,"%s: chown/chgrp", to_name);
 959                 }
 960
 961         if (mode != (to_sb.st_mode & ALLPERMS)) {
 962                 if (fchmod(to_fd,
 963                      dounpriv ? mode & (S_IRWXU|S_IRWXG|S_IRWXO) : mode)) {
 964                         serrno = errno;
 965                         (void)unlink(to_name);
 966                         errno = serrno;
 967                         err(EX_OSERR, "%s: chmod", to_name);
 968                 }
 969         }
 970
 971         /*
 972          * If provided a set of flags, set them, otherwise, preserve the
 973          * flags, except for the dump flag.
 974          * NFS does not support flags.  Ignore EOPNOTSUPP flags if we're just
 975          * trying to turn off UF_NODUMP.  If we're trying to set real flags,
 976          * then warn if the fs doesn't support it, otherwise fail.
 977          */
 978         if (!dounpriv & !devnull && (flags & SETFLAGS ||
 979             (from_sb.st_flags & ~UF_NODUMP) != to_sb.st_flags) &&
 980             fchflags(to_fd,
 981             flags & SETFLAGS ? fset : from_sb.st_flags & ~UF_NODUMP)) {
 982                 if (flags & SETFLAGS) {
 983                         if (errno == EOPNOTSUPP)
 984                                 warn("%s: chflags", to_name);
 985                         else {
 986                                 serrno = errno;
 987                                 (void)unlink(to_name);
 988                                 errno = serrno;
 989                                 err(EX_OSERR, "%s: chflags", to_name);
 990                         }
 991                 }
 992         }
 993
 994         (void)close(to_fd);
 995         if (!devnull)
 996                 (void)close(from_fd);
 997
 998         metadata_log(to_name, "file", tvb, NULL, digestresult, to_sb.st_size);
 999         free(digestresult);
1000 }
1001
1002 /*
1003  * compare --
1004  *      compare two files; non-zero means files differ
1005  */
1006 static int
1007 compare(int from_fd, const char *from_name __unused, size_t from_len,
1008         int to_fd, const char *to_name __unused, size_t to_len,
1009         char **dresp)
1010 {
1011         char *p, *q;
1012         int rv;
1013         int done_compare;
1014         DIGEST_CTX ctx;
1015
1016         rv = 0;
1017         if (from_len != to_len)
1018                 return 1;
1019
1020         if (from_len <= MAX_CMP_SIZE) {
1021                 if (dresp != NULL)
1022                         digest_init(&ctx);
1023                 done_compare = 0;
1024                 if (trymmap(from_fd) && trymmap(to_fd)) {
1025                         p = mmap(NULL, from_len, PROT_READ, MAP_SHARED,
1026                             from_fd, (off_t)0);
1027                         if (p == MAP_FAILED)
1028                                 goto out;
1029                         q = mmap(NULL, from_len, PROT_READ, MAP_SHARED,
1030                             to_fd, (off_t)0);
1031                         if (q == MAP_FAILED) {
1032                                 munmap(p, from_len);
1033                                 goto out;
1034                         }
1035
1036                         rv = memcmp(p, q, from_len);
1037                         if (dresp != NULL)
1038                                 digest_update(&ctx, p, from_len);
1039                         munmap(p, from_len);
1040                         munmap(q, from_len);
1041                         done_compare = 1;
1042                 }
1043         out:
1044                 if (!done_compare) {
1045                         char buf1[MAXBSIZE];
1046                         char buf2[MAXBSIZE];
1047                         int n1, n2;
1048
1049                         rv = 0;
1050                         lseek(from_fd, 0, SEEK_SET);
1051                         lseek(to_fd, 0, SEEK_SET);
1052                         while (rv == 0) {
1053                                 n1 = read(from_fd, buf1, sizeof(buf1));
1054                                 if (n1 == 0)
1055                                         break;          /* EOF */
1056                                 else if (n1 > 0) {
1057                                         n2 = read(to_fd, buf2, n1);
1058                                         if (n2 == n1)
1059                                                 rv = memcmp(buf1, buf2, n1);
1060                                         else
1061                                                 rv = 1; /* out of sync */
1062                                 } else
1063                                         rv = 1;         /* read failure */
1064                                 digest_update(&ctx, buf1, n1);
1065                         }
1066                         lseek(from_fd, 0, SEEK_SET);
1067                         lseek(to_fd, 0, SEEK_SET);
1068                 }
1069         } else
1070                 rv = 1; /* don't bother in this case */
1071
1072         if (dresp != NULL) {
1073                 if (rv == 0)
1074                         *dresp = digest_end(&ctx, NULL);
1075                 else
1076                         (void)digest_end(&ctx, NULL);
1077         }
1078
1079         return rv;
1080 }
1081
1082 /*
1083  * create_tempfile --
1084  *      create a temporary file based on path and open it
1085  */
1086 static int
1087 create_tempfile(const char *path, char *temp, size_t tsize)
1088 {
1089         char *p;
1090
1091         (void)strncpy(temp, path, tsize);
1092         temp[tsize - 1] = '\0';
1093         if ((p = strrchr(temp, '/')) != NULL)
1094                 p++;
1095         else
1096                 p = temp;
1097         (void)strncpy(p, "INS@XXXX", &temp[tsize - 1] - p);
1098         temp[tsize - 1] = '\0';
1099         return (mkstemp(temp));
1100 }
1101
1102 /*
1103  * create_newfile --
1104  *      create a new file, overwriting an existing one if necessary
1105  */
1106 static int
1107 create_newfile(const char *path, int target, struct stat *sbp)
1108 {
1109         char backup[MAXPATHLEN];
1110         int saved_errno = 0;
1111         int newfd;
1112
1113         if (target) {
1114                 /*
1115                  * Unlink now... avoid ETXTBSY errors later.  Try to turn
1116                  * off the append/immutable bits -- if we fail, go ahead,
1117                  * it might work.
1118                  */
1119                 if (sbp->st_flags & NOCHANGEBITS)
1120                         (void)chflags(path, sbp->st_flags & ~NOCHANGEBITS);
1121
1122                 if (dobackup) {
1123                         if ((size_t)snprintf(backup, MAXPATHLEN, "%s%s",
1124                             path, suffix) != strlen(path) + strlen(suffix))
1125                                 errx(EX_OSERR, "%s: backup filename too long",
1126                                     path);
1127                         (void)snprintf(backup, MAXPATHLEN, "%s%s",
1128                             path, suffix);
1129                         if (verbose)
1130                                 (void)printf("install: %s -> %s\n",
1131                                     path, backup);
1132                         if (rename(path, backup) < 0)
1133                                 err(EX_OSERR, "rename: %s to %s", path, backup);
1134                 } else
1135                         if (unlink(path) < 0)
1136                                 saved_errno = errno;
1137         }
1138
1139         newfd = open(path, O_CREAT | O_RDWR | O_TRUNC, S_IRUSR | S_IWUSR);
1140         if (newfd < 0 && saved_errno != 0)
1141                 errno = saved_errno;
1142         return newfd;
1143 }
1144
1145 /*
1146  * copy --
1147  *      copy from one file to another
1148  */
1149 static char *
1150 copy(int from_fd, const char *from_name, int to_fd, const char *to_name,
1151     off_t size)
1152 {
1153         int nr, nw;
1154         int serrno;
1155         char *p;
1156         char buf[MAXBSIZE];
1157         int done_copy;
1158         DIGEST_CTX ctx;
1159
1160         /* Rewind file descriptors. */
1161         if (lseek(from_fd, (off_t)0, SEEK_SET) == (off_t)-1)
1162                 err(EX_OSERR, "lseek: %s", from_name);
1163         if (lseek(to_fd, (off_t)0, SEEK_SET) == (off_t)-1)
1164                 err(EX_OSERR, "lseek: %s", to_name);
1165
1166         digest_init(&ctx);
1167
1168         /*
1169          * Mmap and write if less than 8M (the limit is so we don't totally
1170          * trash memory on big files.  This is really a minor hack, but it
1171          * wins some CPU back.
1172          */
1173         done_copy = 0;
1174         if (size <= 8 * 1048576 && trymmap(from_fd) &&
1175             (p = mmap(NULL, (size_t)size, PROT_READ, MAP_SHARED,
1176                     from_fd, (off_t)0)) != MAP_FAILED) {
1177                 nw = write(to_fd, p, size);
1178                 if (nw != size) {
1179                         serrno = errno;
1180                         (void)unlink(to_name);
1181                         if (nw >= 0) {
1182                                 errx(EX_OSERR,
1183      "short write to %s: %jd bytes written, %jd bytes asked to write",
1184                                     to_name, (uintmax_t)nw, (uintmax_t)size);
1185                         } else {
1186                                 errno = serrno;
1187                                 err(EX_OSERR, "%s", to_name);
1188                         }
1189                 }
1190                 digest_update(&ctx, p, size);
1191                 (void)munmap(p, size);
1192                 done_copy = 1;
1193         }
1194         if (!done_copy) {
1195                 while ((nr = read(from_fd, buf, sizeof(buf))) > 0) {
1196                         if ((nw = write(to_fd, buf, nr)) != nr) {
1197                                 serrno = errno;
1198                                 (void)unlink(to_name);
1199                                 if (nw >= 0) {
1200                                         errx(EX_OSERR,
1201      "short write to %s: %jd bytes written, %jd bytes asked to write",
1202                                             to_name, (uintmax_t)nw,
1203                                             (uintmax_t)size);
1204                                 } else {
1205                                         errno = serrno;
1206                                         err(EX_OSERR, "%s", to_name);
1207                                 }
1208                         }
1209                         digest_update(&ctx, buf, nr);
1210                 }
1211                 if (nr != 0) {
1212                         serrno = errno;
1213                         (void)unlink(to_name);
1214                         errno = serrno;
1215                         err(EX_OSERR, "%s", from_name);
1216                 }
1217         }
1218         return (digest_end(&ctx, NULL));
1219 }
1220
1221 /*
1222  * strip --
1223  *      use strip(1) to strip the target file
1224  */
1225 static void
1226 strip(const char *to_name)
1227 {
1228         const char *stripbin;
1229         const char *args[3];
1230         pid_t pid;
1231         int error, status;
1232
1233         stripbin = getenv("STRIPBIN");
1234         if (stripbin == NULL)
1235                 stripbin = "strip";
1236         args[0] = stripbin;
1237         args[1] = to_name;
1238         args[2] = NULL;
1239         error = posix_spawnp(&pid, stripbin, NULL, NULL,
1240             __DECONST(char **, args), environ);
1241         if (error != 0) {
1242                 (void)unlink(to_name);
1243                 errc(error == EAGAIN || error == EPROCLIM || error == ENOMEM ?
1244                     EX_TEMPFAIL : EX_OSERR, error, "spawn %s", stripbin);
1245         }
1246         if (waitpid(pid, &status, 0) == -1) {
1247                 error = errno;
1248                 (void)unlink(to_name);
1249                 errc(EX_SOFTWARE, error, "wait");
1250                 /* NOTREACHED */
1251         }
1252         if (status != 0) {
1253                 (void)unlink(to_name);
1254                 errx(EX_SOFTWARE, "strip command %s failed on %s",
1255                     stripbin, to_name);
1256         }
1257 }
1258
1259 /*
1260  * install_dir --
1261  *      build directory hierarchy
1262  */
1263 static void
1264 install_dir(char *path)
1265 {
1266         char *p;
1267         struct stat sb;
1268         int ch;
1269
1270         for (p = path;; ++p)
1271                 if (!*p || (p != path && *p  == '/')) {
1272                         ch = *p;
1273                         *p = '\0';
1274 again:
1275                         if (stat(path, &sb) < 0) {
1276                                 if (errno != ENOENT)
1277                                         err(EX_OSERR, "stat %s", path);
1278                                 if (mkdir(path, 0755) < 0) {
1279                                         if (errno == EEXIST)
1280                                                 goto again;
1281                                         err(EX_OSERR, "mkdir %s", path);
1282                                 }
1283                                 if (verbose)
1284                                         (void)printf("install: mkdir %s\n",
1285                                             path);
1286                         } else if (!S_ISDIR(sb.st_mode))
1287                                 errx(EX_OSERR, "%s exists but is not a directory", path);
1288                         if (!(*p = ch))
1289                                 break;
1290                 }
1291
1292         if (!dounpriv) {
1293                 if ((gid != (gid_t)-1 || uid != (uid_t)-1) &&
1294                     chown(path, uid, gid))
1295                         warn("chown %u:%u %s", uid, gid, path);
1296                 /* XXXBED: should we do the chmod in the dounpriv case? */
1297                 if (chmod(path, mode))
1298                         warn("chmod %o %s", mode, path);
1299         }
1300         metadata_log(path, "dir", NULL, NULL, NULL, 0);
1301 }
1302
1303 /*
1304  * metadata_log --
1305  *      if metafp is not NULL, output mtree(8) full path name and settings to
1306  *      metafp, to allow permissions to be set correctly by other tools,
1307  *      or to allow integrity checks to be performed.
1308  */
1309 static void
1310 metadata_log(const char *path, const char *type, struct timeval *tv,
1311         const char *slink, const char *digestresult, off_t size)
1312 {
1313         static const char extra[] = { ' ', '\t', '\n', '\\', '#', '\0' };
1314         const char *p;
1315         char *buf;
1316         size_t destlen;
1317         struct flock metalog_lock;
1318
1319         if (!metafp)
1320                 return;
1321         /* Buffer for strsvis(3). */
1322         buf = (char *)malloc(4 * strlen(path) + 1);
1323         if (buf == NULL) {
1324                 warnx("%s", strerror(ENOMEM));
1325                 return;
1326         }
1327
1328         /* Lock log file. */
1329         metalog_lock.l_start = 0;
1330         metalog_lock.l_len = 0;
1331         metalog_lock.l_whence = SEEK_SET;
1332         metalog_lock.l_type = F_WRLCK;
1333         if (fcntl(fileno(metafp), F_SETLKW, &metalog_lock) == -1) {
1334                 warn("can't lock %s", metafile);
1335                 free(buf);
1336                 return;
1337         }
1338
1339         /* Remove destdir. */
1340         p = path;
1341         if (destdir) {
1342                 destlen = strlen(destdir);
1343                 if (strncmp(p, destdir, destlen) == 0 &&
1344                     (p[destlen] == '/' || p[destlen] == '\0'))
1345                         p += destlen;
1346         }
1347         while (*p && *p == '/')
1348                 p++;
1349         strsvis(buf, p, VIS_OCTAL, extra);
1350         p = buf;
1351         /* Print details. */
1352         fprintf(metafp, ".%s%s type=%s", *p ? "/" : "", p, type);
1353         if (owner)
1354                 fprintf(metafp, " uname=%s", owner);
1355         if (group)
1356                 fprintf(metafp, " gname=%s", group);
1357         fprintf(metafp, " mode=%#o", mode);
1358         if (slink) {
1359                 strsvis(buf, slink, VIS_CSTYLE, extra); /* encode link */
1360                 fprintf(metafp, " link=%s", buf);
1361         }
1362         if (*type == 'f') /* type=file */
1363                 fprintf(metafp, " size=%lld", (long long)size);
1364         if (tv != NULL && dopreserve)
1365                 fprintf(metafp, " time=%lld.%ld",
1366                         (long long)tv[1].tv_sec, (long)tv[1].tv_usec);
1367         if (digestresult && digest)
1368                 fprintf(metafp, " %s=%s", digest, digestresult);
1369         if (fflags)
1370                 fprintf(metafp, " flags=%s", fflags);
1371         if (tags)
1372                 fprintf(metafp, " tags=%s", tags);
1373         fputc('\n', metafp);
1374         /* Flush line. */
1375         fflush(metafp);
1376
1377         /* Unlock log file. */
1378         metalog_lock.l_type = F_UNLCK;
1379         if (fcntl(fileno(metafp), F_SETLKW, &metalog_lock) == -1)
1380                 warn("can't unlock %s", metafile);
1381         free(buf);
1382 }
1383
1384 /*
1385  * usage --
1386  *      print a usage message and die
1387  */
1388 static void
1389 usage(void)
1390 {
1391         (void)fprintf(stderr,
1392 "usage: install [-bCcpSsUv] [-f flags] [-g group] [-m mode] [-o owner]\n"
1393 "               [-M log] [-D dest] [-h hash] [-T tags]\n"
1394 "               [-B suffix] [-l linkflags] [-N dbdir]\n"
1395 "               file1 file2\n"
1396 "       install [-bCcpSsUv] [-f flags] [-g group] [-m mode] [-o owner]\n"
1397 "               [-M log] [-D dest] [-h hash] [-T tags]\n"
1398 "               [-B suffix] [-l linkflags] [-N dbdir]\n"
1399 "               file1 ... fileN directory\n"
1400 "       install -dU [-vU] [-g group] [-m mode] [-N dbdir] [-o owner]\n"
1401 "               [-M log] [-D dest] [-h hash] [-T tags]\n"
1402 "               directory ...\n");
1403         exit(EX_USAGE);
1404         /* NOTREACHED */
1405 }
1406
1407 /*
1408  * trymmap --
1409  *      return true (1) if mmap should be tried, false (0) if not.
1410  */
1411 static int
1412 trymmap(int fd)
1413 {
1414 /*
1415  * The ifdef is for bootstrapping - f_fstypename doesn't exist in
1416  * pre-Lite2-merge systems.
1417  */
1418 #ifdef MFSNAMELEN
1419         struct statfs stfs;
1420
1421         if (fstatfs(fd, &stfs) != 0)
1422                 return (0);
1423         if (strcmp(stfs.f_fstypename, "ufs") == 0 ||
1424             strcmp(stfs.f_fstypename, "cd9660") == 0)
1425                 return (1);
1426 #endif
1427         return (0);
1428 }