xref: /freebsd/usr.bin/xinstall/xinstall.c (revision d3d381b2b194b4d24853e92eecef55f262688d1a)
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
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 *
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
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
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 *
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
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 *
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
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
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
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, *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 		for (s = src, d = dst; *s == *d; s++, d++)
713 			continue;
714 		while (*s != '/')
715 			s--, d--;
716 
717 		/* Count the number of directories we need to backtrack. */
718 		for (++d, lnk[0] = '\0'; *d; d++)
719 			if (*d == '/')
720 				(void)strlcat(lnk, "../", sizeof(lnk));
721 
722 		(void)strlcat(lnk, ++s, sizeof(lnk));
723 
724 		do_symlink(lnk, to_name, target_sb);
725 		/* XXX: Link may point outside of destdir. */
726 		metadata_log(to_name, "link", NULL, lnk, NULL, 0);
727 		return;
728 	}
729 
730 	/*
731 	 * If absolute or relative was not specified, try the names the
732 	 * user provided.
733 	 */
734 	do_symlink(from_name, to_name, target_sb);
735 	/* XXX: from_name may point outside of destdir. */
736 	metadata_log(to_name, "link", NULL, from_name, NULL, 0);
737 }
738 
739 /*
740  * install --
741  *	build a path name and install the file
742  */
743 static void
744 install(const char *from_name, const char *to_name, u_long fset, u_int flags)
745 {
746 	struct stat from_sb, temp_sb, to_sb;
747 	struct timespec tsb[2];
748 	int devnull, files_match, from_fd, serrno, target;
749 	int tempcopy, temp_fd, to_fd;
750 	char backup[MAXPATHLEN], *p, pathbuf[MAXPATHLEN], tempfile[MAXPATHLEN];
751 	char *digestresult;
752 
753 	files_match = 0;
754 	from_fd = -1;
755 	to_fd = -1;
756 
757 	/* If try to install NULL file to a directory, fails. */
758 	if (flags & DIRECTORY || strcmp(from_name, _PATH_DEVNULL)) {
759 		if (!dolink) {
760 			if (stat(from_name, &from_sb))
761 				err(EX_OSERR, "%s", from_name);
762 			if (!S_ISREG(from_sb.st_mode)) {
763 				errno = EFTYPE;
764 				err(EX_OSERR, "%s", from_name);
765 			}
766 		}
767 		/* Build the target path. */
768 		if (flags & DIRECTORY) {
769 			(void)snprintf(pathbuf, sizeof(pathbuf), "%s%s%s",
770 			    to_name,
771 			    to_name[strlen(to_name) - 1] == '/' ? "" : "/",
772 			    (p = strrchr(from_name, '/')) ? ++p : from_name);
773 			to_name = pathbuf;
774 		}
775 		devnull = 0;
776 	} else {
777 		devnull = 1;
778 	}
779 
780 	target = (lstat(to_name, &to_sb) == 0);
781 
782 	if (dolink) {
783 		if (target && !safecopy) {
784 			if (to_sb.st_mode & S_IFDIR && rmdir(to_name) == -1)
785 				err(EX_OSERR, "%s", to_name);
786 #if HAVE_STRUCT_STAT_ST_FLAGS
787 			if (to_sb.st_flags & NOCHANGEBITS)
788 				(void)chflags(to_name,
789 				    to_sb.st_flags & ~NOCHANGEBITS);
790 #endif
791 			unlink(to_name);
792 		}
793 		makelink(from_name, to_name, target ? &to_sb : NULL);
794 		return;
795 	}
796 
797 	if (target && !S_ISREG(to_sb.st_mode) && !S_ISLNK(to_sb.st_mode)) {
798 		errno = EFTYPE;
799 		warn("%s", to_name);
800 		return;
801 	}
802 
803 	/* Only copy safe if the target exists. */
804 	tempcopy = safecopy && target;
805 
806 	if (!devnull && (from_fd = open(from_name, O_RDONLY, 0)) < 0)
807 		err(EX_OSERR, "%s", from_name);
808 
809 	/* If we don't strip, we can compare first. */
810 	if (docompare && !dostrip && target && S_ISREG(to_sb.st_mode)) {
811 		if ((to_fd = open(to_name, O_RDONLY, 0)) < 0)
812 			err(EX_OSERR, "%s", to_name);
813 		if (devnull)
814 			files_match = to_sb.st_size == 0;
815 		else
816 			files_match = !(compare(from_fd, from_name,
817 			    (size_t)from_sb.st_size, to_fd,
818 			    to_name, (size_t)to_sb.st_size, &digestresult));
819 
820 		/* Close "to" file unless we match. */
821 		if (!files_match)
822 			(void)close(to_fd);
823 	}
824 
825 	if (!files_match) {
826 		if (tempcopy) {
827 			to_fd = create_tempfile(to_name, tempfile,
828 			    sizeof(tempfile));
829 			if (to_fd < 0)
830 				err(EX_OSERR, "%s", tempfile);
831 		} else {
832 			if ((to_fd = create_newfile(to_name, target,
833 			    &to_sb)) < 0)
834 				err(EX_OSERR, "%s", to_name);
835 			if (verbose)
836 				(void)printf("install: %s -> %s\n",
837 				    from_name, to_name);
838 		}
839 		if (!devnull)
840 			digestresult = copy(from_fd, from_name, to_fd,
841 			     tempcopy ? tempfile : to_name, from_sb.st_size);
842 		else
843 			digestresult = NULL;
844 	}
845 
846 	if (dostrip) {
847 		strip(tempcopy ? tempfile : to_name);
848 
849 		/*
850 		 * Re-open our fd on the target, in case we used a strip
851 		 * that does not work in-place -- like GNU binutils strip.
852 		 */
853 		close(to_fd);
854 		to_fd = open(tempcopy ? tempfile : to_name, O_RDONLY, 0);
855 		if (to_fd < 0)
856 			err(EX_OSERR, "stripping %s", to_name);
857 	}
858 
859 	/*
860 	 * Compare the stripped temp file with the target.
861 	 */
862 	if (docompare && dostrip && target && S_ISREG(to_sb.st_mode)) {
863 		temp_fd = to_fd;
864 
865 		/* Re-open to_fd using the real target name. */
866 		if ((to_fd = open(to_name, O_RDONLY, 0)) < 0)
867 			err(EX_OSERR, "%s", to_name);
868 
869 		if (fstat(temp_fd, &temp_sb)) {
870 			serrno = errno;
871 			(void)unlink(tempfile);
872 			errno = serrno;
873 			err(EX_OSERR, "%s", tempfile);
874 		}
875 
876 		if (compare(temp_fd, tempfile, (size_t)temp_sb.st_size, to_fd,
877 			    to_name, (size_t)to_sb.st_size, &digestresult)
878 			    == 0) {
879 			/*
880 			 * If target has more than one link we need to
881 			 * replace it in order to snap the extra links.
882 			 * Need to preserve target file times, though.
883 			 */
884 			if (to_sb.st_nlink != 1) {
885 				tsb[0] = to_sb.st_atim;
886 				tsb[1] = to_sb.st_mtim;
887 				(void)utimensat(AT_FDCWD, tempfile, tsb, 0);
888 			} else {
889 				files_match = 1;
890 				(void)unlink(tempfile);
891 			}
892 			(void) close(temp_fd);
893 		}
894 	} else if (dostrip)
895 		digestresult = digest_file(tempfile);
896 
897 	/*
898 	 * Move the new file into place if doing a safe copy
899 	 * and the files are different (or just not compared).
900 	 */
901 	if (tempcopy && !files_match) {
902 #if HAVE_STRUCT_STAT_ST_FLAGS
903 		/* Try to turn off the immutable bits. */
904 		if (to_sb.st_flags & NOCHANGEBITS)
905 			(void)chflags(to_name, to_sb.st_flags & ~NOCHANGEBITS);
906 #endif
907 		if (dobackup) {
908 			if ((size_t)snprintf(backup, MAXPATHLEN, "%s%s", to_name,
909 			    suffix) != strlen(to_name) + strlen(suffix)) {
910 				unlink(tempfile);
911 				errx(EX_OSERR, "%s: backup filename too long",
912 				    to_name);
913 			}
914 			if (verbose)
915 				(void)printf("install: %s -> %s\n", to_name, backup);
916 			if (unlink(backup) < 0 && errno != ENOENT) {
917 				serrno = errno;
918 #if HAVE_STRUCT_STAT_ST_FLAGS
919 				if (to_sb.st_flags & NOCHANGEBITS)
920 					(void)chflags(to_name, to_sb.st_flags);
921 #endif
922 				unlink(tempfile);
923 				errno = serrno;
924 				err(EX_OSERR, "unlink: %s", backup);
925 			}
926 			if (link(to_name, backup) < 0) {
927 				serrno = errno;
928 				unlink(tempfile);
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 				errno = serrno;
934 				err(EX_OSERR, "link: %s to %s", to_name,
935 				     backup);
936 			}
937 		}
938 		if (verbose)
939 			(void)printf("install: %s -> %s\n", from_name, to_name);
940 		if (rename(tempfile, to_name) < 0) {
941 			serrno = errno;
942 			unlink(tempfile);
943 			errno = serrno;
944 			err(EX_OSERR, "rename: %s to %s",
945 			    tempfile, to_name);
946 		}
947 
948 		/* Re-open to_fd so we aren't hosed by the rename(2). */
949 		(void) close(to_fd);
950 		if ((to_fd = open(to_name, O_RDONLY, 0)) < 0)
951 			err(EX_OSERR, "%s", to_name);
952 	}
953 
954 	/*
955 	 * Preserve the timestamp of the source file if necessary.
956 	 */
957 	if (dopreserve && !files_match && !devnull) {
958 		tsb[0] = from_sb.st_atim;
959 		tsb[1] = from_sb.st_mtim;
960 		(void)utimensat(AT_FDCWD, to_name, tsb, 0);
961 	}
962 
963 	if (fstat(to_fd, &to_sb) == -1) {
964 		serrno = errno;
965 		(void)unlink(to_name);
966 		errno = serrno;
967 		err(EX_OSERR, "%s", to_name);
968 	}
969 
970 	/*
971 	 * Set owner, group, mode for target; do the chown first,
972 	 * chown may lose the setuid bits.
973 	 */
974 	if (!dounpriv && ((gid != (gid_t)-1 && gid != to_sb.st_gid) ||
975 	    (uid != (uid_t)-1 && uid != to_sb.st_uid) ||
976 	    (mode != (to_sb.st_mode & ALLPERMS)))) {
977 #if HAVE_STRUCT_STAT_ST_FLAGS
978 		/* Try to turn off the immutable bits. */
979 		if (to_sb.st_flags & NOCHANGEBITS)
980 			(void)fchflags(to_fd, to_sb.st_flags & ~NOCHANGEBITS);
981 #endif
982 	}
983 
984 	if (!dounpriv &
985 	    (gid != (gid_t)-1 && gid != to_sb.st_gid) ||
986 	    (uid != (uid_t)-1 && uid != to_sb.st_uid))
987 		if (fchown(to_fd, uid, gid) == -1) {
988 			serrno = errno;
989 			(void)unlink(to_name);
990 			errno = serrno;
991 			err(EX_OSERR,"%s: chown/chgrp", to_name);
992 		}
993 
994 	if (mode != (to_sb.st_mode & ALLPERMS)) {
995 		if (fchmod(to_fd,
996 		     dounpriv ? mode & (S_IRWXU|S_IRWXG|S_IRWXO) : mode)) {
997 			serrno = errno;
998 			(void)unlink(to_name);
999 			errno = serrno;
1000 			err(EX_OSERR, "%s: chmod", to_name);
1001 		}
1002 	}
1003 #if HAVE_STRUCT_STAT_ST_FLAGS
1004 	/*
1005 	 * If provided a set of flags, set them, otherwise, preserve the
1006 	 * flags, except for the dump flag.
1007 	 * NFS does not support flags.  Ignore EOPNOTSUPP flags if we're just
1008 	 * trying to turn off UF_NODUMP.  If we're trying to set real flags,
1009 	 * then warn if the fs doesn't support it, otherwise fail.
1010 	 */
1011 	if (!dounpriv & !devnull && (flags & SETFLAGS ||
1012 	    (from_sb.st_flags & ~UF_NODUMP) != to_sb.st_flags) &&
1013 	    fchflags(to_fd,
1014 	    flags & SETFLAGS ? fset : from_sb.st_flags & ~UF_NODUMP)) {
1015 		if (flags & SETFLAGS) {
1016 			if (errno == EOPNOTSUPP)
1017 				warn("%s: chflags", to_name);
1018 			else {
1019 				serrno = errno;
1020 				(void)unlink(to_name);
1021 				errno = serrno;
1022 				err(EX_OSERR, "%s: chflags", to_name);
1023 			}
1024 		}
1025 	}
1026 #endif
1027 
1028 	(void)close(to_fd);
1029 	if (!devnull)
1030 		(void)close(from_fd);
1031 
1032 	metadata_log(to_name, "file", tsb, NULL, digestresult, to_sb.st_size);
1033 	free(digestresult);
1034 }
1035 
1036 /*
1037  * compare --
1038  *	compare two files; non-zero means files differ
1039  */
1040 static int
1041 compare(int from_fd, const char *from_name __unused, size_t from_len,
1042 	int to_fd, const char *to_name __unused, size_t to_len,
1043 	char **dresp)
1044 {
1045 	char *p, *q;
1046 	int rv;
1047 	int done_compare;
1048 	DIGEST_CTX ctx;
1049 
1050 	rv = 0;
1051 	if (from_len != to_len)
1052 		return 1;
1053 
1054 	if (from_len <= MAX_CMP_SIZE) {
1055 		if (dresp != NULL)
1056 			digest_init(&ctx);
1057 		done_compare = 0;
1058 		if (trymmap(from_fd) && trymmap(to_fd)) {
1059 			p = mmap(NULL, from_len, PROT_READ, MAP_SHARED,
1060 			    from_fd, (off_t)0);
1061 			if (p == MAP_FAILED)
1062 				goto out;
1063 			q = mmap(NULL, from_len, PROT_READ, MAP_SHARED,
1064 			    to_fd, (off_t)0);
1065 			if (q == MAP_FAILED) {
1066 				munmap(p, from_len);
1067 				goto out;
1068 			}
1069 
1070 			rv = memcmp(p, q, from_len);
1071 			if (dresp != NULL)
1072 				digest_update(&ctx, p, from_len);
1073 			munmap(p, from_len);
1074 			munmap(q, from_len);
1075 			done_compare = 1;
1076 		}
1077 	out:
1078 		if (!done_compare) {
1079 			char buf1[MAXBSIZE];
1080 			char buf2[MAXBSIZE];
1081 			int n1, n2;
1082 
1083 			rv = 0;
1084 			lseek(from_fd, 0, SEEK_SET);
1085 			lseek(to_fd, 0, SEEK_SET);
1086 			while (rv == 0) {
1087 				n1 = read(from_fd, buf1, sizeof(buf1));
1088 				if (n1 == 0)
1089 					break;		/* EOF */
1090 				else if (n1 > 0) {
1091 					n2 = read(to_fd, buf2, n1);
1092 					if (n2 == n1)
1093 						rv = memcmp(buf1, buf2, n1);
1094 					else
1095 						rv = 1;	/* out of sync */
1096 				} else
1097 					rv = 1;		/* read failure */
1098 				digest_update(&ctx, buf1, n1);
1099 			}
1100 			lseek(from_fd, 0, SEEK_SET);
1101 			lseek(to_fd, 0, SEEK_SET);
1102 		}
1103 	} else
1104 		rv = 1;	/* don't bother in this case */
1105 
1106 	if (dresp != NULL) {
1107 		if (rv == 0)
1108 			*dresp = digest_end(&ctx, NULL);
1109 		else
1110 			(void)digest_end(&ctx, NULL);
1111 	}
1112 
1113 	return rv;
1114 }
1115 
1116 /*
1117  * create_tempfile --
1118  *	create a temporary file based on path and open it
1119  */
1120 static int
1121 create_tempfile(const char *path, char *temp, size_t tsize)
1122 {
1123 	char *p;
1124 
1125 	(void)strncpy(temp, path, tsize);
1126 	temp[tsize - 1] = '\0';
1127 	if ((p = strrchr(temp, '/')) != NULL)
1128 		p++;
1129 	else
1130 		p = temp;
1131 	(void)strncpy(p, "INS@XXXX", &temp[tsize - 1] - p);
1132 	temp[tsize - 1] = '\0';
1133 	return (mkstemp(temp));
1134 }
1135 
1136 /*
1137  * create_newfile --
1138  *	create a new file, overwriting an existing one if necessary
1139  */
1140 static int
1141 create_newfile(const char *path, int target, struct stat *sbp)
1142 {
1143 	char backup[MAXPATHLEN];
1144 	int saved_errno = 0;
1145 	int newfd;
1146 
1147 	if (target) {
1148 		/*
1149 		 * Unlink now... avoid ETXTBSY errors later.  Try to turn
1150 		 * off the append/immutable bits -- if we fail, go ahead,
1151 		 * it might work.
1152 		 */
1153 #if HAVE_STRUCT_STAT_ST_FLAGS
1154 		if (sbp->st_flags & NOCHANGEBITS)
1155 			(void)chflags(path, sbp->st_flags & ~NOCHANGEBITS);
1156 #endif
1157 
1158 		if (dobackup) {
1159 			if ((size_t)snprintf(backup, MAXPATHLEN, "%s%s",
1160 			    path, suffix) != strlen(path) + strlen(suffix)) {
1161 				saved_errno = errno;
1162 #if HAVE_STRUCT_STAT_ST_FLAGS
1163 				if (sbp->st_flags & NOCHANGEBITS)
1164 					(void)chflags(path, sbp->st_flags);
1165 #endif
1166 				errno = saved_errno;
1167 				errx(EX_OSERR, "%s: backup filename too long",
1168 				    path);
1169 			}
1170 			(void)snprintf(backup, MAXPATHLEN, "%s%s",
1171 			    path, suffix);
1172 			if (verbose)
1173 				(void)printf("install: %s -> %s\n",
1174 				    path, backup);
1175 			if (rename(path, backup) < 0) {
1176 				saved_errno = errno;
1177 #if HAVE_STRUCT_STAT_ST_FLAGS
1178 				if (sbp->st_flags & NOCHANGEBITS)
1179 					(void)chflags(path, sbp->st_flags);
1180 #endif
1181 				errno = saved_errno;
1182 				err(EX_OSERR, "rename: %s to %s", path, backup);
1183 			}
1184 		} else
1185 			if (unlink(path) < 0)
1186 				saved_errno = errno;
1187 	}
1188 
1189 	newfd = open(path, O_CREAT | O_RDWR | O_TRUNC, S_IRUSR | S_IWUSR);
1190 	if (newfd < 0 && saved_errno != 0)
1191 		errno = saved_errno;
1192 	return newfd;
1193 }
1194 
1195 /*
1196  * copy --
1197  *	copy from one file to another
1198  */
1199 static char *
1200 copy(int from_fd, const char *from_name, int to_fd, const char *to_name,
1201     off_t size)
1202 {
1203 	int nr, nw;
1204 	int serrno;
1205 	char *p;
1206 	char buf[MAXBSIZE];
1207 	int done_copy;
1208 	DIGEST_CTX ctx;
1209 
1210 	/* Rewind file descriptors. */
1211 	if (lseek(from_fd, (off_t)0, SEEK_SET) == (off_t)-1)
1212 		err(EX_OSERR, "lseek: %s", from_name);
1213 	if (lseek(to_fd, (off_t)0, SEEK_SET) == (off_t)-1)
1214 		err(EX_OSERR, "lseek: %s", to_name);
1215 
1216 	digest_init(&ctx);
1217 
1218 	/*
1219 	 * Mmap and write if less than 8M (the limit is so we don't totally
1220 	 * trash memory on big files.  This is really a minor hack, but it
1221 	 * wins some CPU back.
1222 	 */
1223 	done_copy = 0;
1224 	if (size <= 8 * 1048576 && trymmap(from_fd) &&
1225 	    (p = mmap(NULL, (size_t)size, PROT_READ, MAP_SHARED,
1226 		    from_fd, (off_t)0)) != MAP_FAILED) {
1227 		nw = write(to_fd, p, size);
1228 		if (nw != size) {
1229 			serrno = errno;
1230 			(void)unlink(to_name);
1231 			if (nw >= 0) {
1232 				errx(EX_OSERR,
1233      "short write to %s: %jd bytes written, %jd bytes asked to write",
1234 				    to_name, (uintmax_t)nw, (uintmax_t)size);
1235 			} else {
1236 				errno = serrno;
1237 				err(EX_OSERR, "%s", to_name);
1238 			}
1239 		}
1240 		digest_update(&ctx, p, size);
1241 		(void)munmap(p, size);
1242 		done_copy = 1;
1243 	}
1244 	if (!done_copy) {
1245 		while ((nr = read(from_fd, buf, sizeof(buf))) > 0) {
1246 			if ((nw = write(to_fd, buf, nr)) != nr) {
1247 				serrno = errno;
1248 				(void)unlink(to_name);
1249 				if (nw >= 0) {
1250 					errx(EX_OSERR,
1251      "short write to %s: %jd bytes written, %jd bytes asked to write",
1252 					    to_name, (uintmax_t)nw,
1253 					    (uintmax_t)size);
1254 				} else {
1255 					errno = serrno;
1256 					err(EX_OSERR, "%s", to_name);
1257 				}
1258 			}
1259 			digest_update(&ctx, buf, nr);
1260 		}
1261 		if (nr != 0) {
1262 			serrno = errno;
1263 			(void)unlink(to_name);
1264 			errno = serrno;
1265 			err(EX_OSERR, "%s", from_name);
1266 		}
1267 	}
1268 	if (safecopy && fsync(to_fd) == -1) {
1269 		serrno = errno;
1270 		(void)unlink(to_name);
1271 		errno = serrno;
1272 		err(EX_OSERR, "fsync failed for %s", to_name);
1273 	}
1274 	return (digest_end(&ctx, NULL));
1275 }
1276 
1277 /*
1278  * strip --
1279  *	use strip(1) to strip the target file
1280  */
1281 static void
1282 strip(const char *to_name)
1283 {
1284 	const char *stripbin;
1285 	const char *args[3];
1286 	pid_t pid;
1287 	int error, status;
1288 
1289 	stripbin = getenv("STRIPBIN");
1290 	if (stripbin == NULL)
1291 		stripbin = "strip";
1292 	args[0] = stripbin;
1293 	args[1] = to_name;
1294 	args[2] = NULL;
1295 	error = posix_spawnp(&pid, stripbin, NULL, NULL,
1296 	    __DECONST(char **, args), environ);
1297 	if (error != 0) {
1298 		(void)unlink(to_name);
1299 		errc(error == EAGAIN || error == EPROCLIM || error == ENOMEM ?
1300 		    EX_TEMPFAIL : EX_OSERR, error, "spawn %s", stripbin);
1301 	}
1302 	if (waitpid(pid, &status, 0) == -1) {
1303 		error = errno;
1304 		(void)unlink(to_name);
1305 		errc(EX_SOFTWARE, error, "wait");
1306 		/* NOTREACHED */
1307 	}
1308 	if (status != 0) {
1309 		(void)unlink(to_name);
1310 		errx(EX_SOFTWARE, "strip command %s failed on %s",
1311 		    stripbin, to_name);
1312 	}
1313 }
1314 
1315 /*
1316  * install_dir --
1317  *	build directory hierarchy
1318  */
1319 static void
1320 install_dir(char *path)
1321 {
1322 	char *p;
1323 	struct stat sb;
1324 	int ch, tried_mkdir;
1325 
1326 	for (p = path;; ++p)
1327 		if (!*p || (p != path && *p  == '/')) {
1328 			tried_mkdir = 0;
1329 			ch = *p;
1330 			*p = '\0';
1331 again:
1332 			if (stat(path, &sb) < 0) {
1333 				if (errno != ENOENT || tried_mkdir)
1334 					err(EX_OSERR, "stat %s", path);
1335 				if (mkdir(path, 0755) < 0) {
1336 					tried_mkdir = 1;
1337 					if (errno == EEXIST)
1338 						goto again;
1339 					err(EX_OSERR, "mkdir %s", path);
1340 				}
1341 				if (verbose)
1342 					(void)printf("install: mkdir %s\n",
1343 					    path);
1344 			} else if (!S_ISDIR(sb.st_mode))
1345 				errx(EX_OSERR, "%s exists but is not a directory", path);
1346 			if (!(*p = ch))
1347 				break;
1348  		}
1349 
1350 	if (!dounpriv) {
1351 		if ((gid != (gid_t)-1 || uid != (uid_t)-1) &&
1352 		    chown(path, uid, gid))
1353 			warn("chown %u:%u %s", uid, gid, path);
1354 		/* XXXBED: should we do the chmod in the dounpriv case? */
1355 		if (chmod(path, mode))
1356 			warn("chmod %o %s", mode, path);
1357 	}
1358 	metadata_log(path, "dir", NULL, NULL, NULL, 0);
1359 }
1360 
1361 /*
1362  * metadata_log --
1363  *	if metafp is not NULL, output mtree(8) full path name and settings to
1364  *	metafp, to allow permissions to be set correctly by other tools,
1365  *	or to allow integrity checks to be performed.
1366  */
1367 static void
1368 metadata_log(const char *path, const char *type, struct timespec *ts,
1369 	const char *slink, const char *digestresult, off_t size)
1370 {
1371 	static const char extra[] = { ' ', '\t', '\n', '\\', '#', '\0' };
1372 	const char *p;
1373 	char *buf;
1374 	size_t destlen;
1375 	struct flock metalog_lock;
1376 
1377 	if (!metafp)
1378 		return;
1379 	/* Buffer for strsvis(3). */
1380 	buf = (char *)malloc(4 * strlen(path) + 1);
1381 	if (buf == NULL) {
1382 		warnx("%s", strerror(ENOMEM));
1383 		return;
1384 	}
1385 
1386 	/* Lock log file. */
1387 	metalog_lock.l_start = 0;
1388 	metalog_lock.l_len = 0;
1389 	metalog_lock.l_whence = SEEK_SET;
1390 	metalog_lock.l_type = F_WRLCK;
1391 	if (fcntl(fileno(metafp), F_SETLKW, &metalog_lock) == -1) {
1392 		warn("can't lock %s", metafile);
1393 		free(buf);
1394 		return;
1395 	}
1396 
1397 	/* Remove destdir. */
1398 	p = path;
1399 	if (destdir) {
1400 		destlen = strlen(destdir);
1401 		if (strncmp(p, destdir, destlen) == 0 &&
1402 		    (p[destlen] == '/' || p[destlen] == '\0'))
1403 			p += destlen;
1404 	}
1405 	while (*p && *p == '/')
1406 		p++;
1407 	strsvis(buf, p, VIS_OCTAL, extra);
1408 	p = buf;
1409 	/* Print details. */
1410 	fprintf(metafp, ".%s%s type=%s", *p ? "/" : "", p, type);
1411 	if (owner)
1412 		fprintf(metafp, " uname=%s", owner);
1413 	if (group)
1414 		fprintf(metafp, " gname=%s", group);
1415 	fprintf(metafp, " mode=%#o", mode);
1416 	if (slink) {
1417 		strsvis(buf, slink, VIS_CSTYLE, extra);	/* encode link */
1418 		fprintf(metafp, " link=%s", buf);
1419 	}
1420 	if (*type == 'f') /* type=file */
1421 		fprintf(metafp, " size=%lld", (long long)size);
1422 	if (ts != NULL && dopreserve)
1423 		fprintf(metafp, " time=%lld.%09ld",
1424 			(long long)ts[1].tv_sec, ts[1].tv_nsec);
1425 	if (digestresult && digest)
1426 		fprintf(metafp, " %s=%s", digest, digestresult);
1427 	if (fflags)
1428 		fprintf(metafp, " flags=%s", fflags);
1429 	if (tags)
1430 		fprintf(metafp, " tags=%s", tags);
1431 	fputc('\n', metafp);
1432 	/* Flush line. */
1433 	fflush(metafp);
1434 
1435 	/* Unlock log file. */
1436 	metalog_lock.l_type = F_UNLCK;
1437 	if (fcntl(fileno(metafp), F_SETLKW, &metalog_lock) == -1)
1438 		warn("can't unlock %s", metafile);
1439 	free(buf);
1440 }
1441 
1442 /*
1443  * usage --
1444  *	print a usage message and die
1445  */
1446 static void
1447 usage(void)
1448 {
1449 	(void)fprintf(stderr,
1450 "usage: install [-bCcpSsUv] [-f flags] [-g group] [-m mode] [-o owner]\n"
1451 "               [-M log] [-D dest] [-h hash] [-T tags]\n"
1452 "               [-B suffix] [-l linkflags] [-N dbdir]\n"
1453 "               file1 file2\n"
1454 "       install [-bCcpSsUv] [-f flags] [-g group] [-m mode] [-o owner]\n"
1455 "               [-M log] [-D dest] [-h hash] [-T tags]\n"
1456 "               [-B suffix] [-l linkflags] [-N dbdir]\n"
1457 "               file1 ... fileN directory\n"
1458 "       install -dU [-vU] [-g group] [-m mode] [-N dbdir] [-o owner]\n"
1459 "               [-M log] [-D dest] [-h hash] [-T tags]\n"
1460 "               directory ...\n");
1461 	exit(EX_USAGE);
1462 	/* NOTREACHED */
1463 }
1464 
1465 /*
1466  * trymmap --
1467  *	return true (1) if mmap should be tried, false (0) if not.
1468  */
1469 static int
1470 trymmap(int fd)
1471 {
1472 /*
1473  * The ifdef is for bootstrapping - f_fstypename doesn't exist in
1474  * pre-Lite2-merge systems.
1475  */
1476 #ifdef MFSNAMELEN
1477 	struct statfs stfs;
1478 
1479 	if (fstatfs(fd, &stfs) != 0)
1480 		return (0);
1481 	if (strcmp(stfs.f_fstypename, "ufs") == 0 ||
1482 	    strcmp(stfs.f_fstypename, "cd9660") == 0)
1483 		return (1);
1484 #endif
1485 	return (0);
1486 }
1487