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