2 * Copyright (c) 1987, 1993
3 * The Regents of the University of California. All rights reserved.
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 4. Neither the name of the University nor the names of its contributors
14 * may be used to endorse or promote products derived from this software
15 * without specific prior written permission.
17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 static const char copyright[] =
32 "@(#) Copyright (c) 1987, 1993\n\
33 The Regents of the University of California. All rights reserved.\n";
38 static char sccsid[] = "@(#)xinstall.c 8.1 (Berkeley) 7/21/93";
42 #include <sys/cdefs.h>
43 __FBSDID("$FreeBSD$");
45 #include <sys/param.h>
47 #include <sys/mount.h>
65 /* Bootstrap aid - this doesn't exist in most older releases */
67 #define MAP_FAILED ((void *)-1) /* from <sys/mman.h> */
70 #define MAX_CMP_SIZE (16 * 1024 * 1024)
72 #define DIRECTORY 0x01 /* Tell install it's a directory. */
73 #define SETFLAGS 0x02 /* Tell install to set flags. */
74 #define NOCHANGEBITS (UF_IMMUTABLE | UF_APPEND | SF_IMMUTABLE | SF_APPEND)
75 #define BACKUP_SUFFIX ".old"
77 static struct passwd *pp;
78 static struct group *gp;
81 static int dobackup, docompare, dodir, dopreserve, dostrip, nommap, safecopy,
83 static mode_t mode = S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH;
84 static const char *suffix = BACKUP_SUFFIX;
86 static int compare(int, const char *, size_t, int, const char *, size_t);
87 static void copy(int, const char *, int, const char *, off_t);
88 static int create_newfile(const char *, int, struct stat *);
89 static int create_tempfile(const char *, char *, size_t);
90 static void install(const char *, const char *, u_long, u_int);
91 static void install_dir(char *);
92 static u_long numeric_id(const char *, const char *);
93 static void strip(const char *);
94 static int trymmap(int);
95 static void usage(void);
98 main(int argc, char *argv[])
100 struct stat from_sb, to_sb;
106 const char *group, *owner, *to_name;
109 group = owner = NULL;
110 while ((ch = getopt(argc, argv, "B:bCcdf:g:Mm:o:pSsv")) != -1)
122 /* For backwards compatibility. */
129 if (strtofflags(&flags, &fset, NULL))
130 errx(EX_USAGE, "%s: invalid flag", flags);
140 if (!(set = setmode(optarg)))
141 errx(EX_USAGE, "invalid file mode: %s",
143 mode = getmode(set, 0);
150 docompare = dopreserve = 1;
168 /* some options make no sense when creating directories */
169 if (dostrip && dodir) {
170 warnx("-d and -s may not be specified together");
174 if (getenv("DONTSTRIP") != NULL) {
175 warnx("DONTSTRIP set - will not strip installed binaries");
179 /* must have at least two arguments, except when creating directories */
180 if (argc == 0 || (argc == 1 && !dodir))
183 /* need to make a temp copy so we can compare stripped version */
184 if (docompare && dostrip)
187 /* get group and owner id's */
189 if ((gp = getgrnam(group)) != NULL)
192 gid = (gid_t)numeric_id(group, "group");
197 if ((pp = getpwnam(owner)) != NULL)
200 uid = (uid_t)numeric_id(owner, "user");
205 for (; *argv != NULL; ++argv)
211 no_target = stat(to_name = argv[argc - 1], &to_sb);
212 if (!no_target && S_ISDIR(to_sb.st_mode)) {
213 for (; *argv != to_name; ++argv)
214 install(*argv, to_name, fset, iflags | DIRECTORY);
219 /* can't do file1 file2 directory/file */
222 warnx("target directory `%s' does not exist",
225 warnx("target `%s' is not a directory",
231 if (stat(*argv, &from_sb))
232 err(EX_OSERR, "%s", *argv);
233 if (!S_ISREG(to_sb.st_mode)) {
235 err(EX_OSERR, "%s", to_name);
237 if (to_sb.st_dev == from_sb.st_dev &&
238 to_sb.st_ino == from_sb.st_ino)
240 "%s and %s are the same file", *argv, to_name);
242 install(*argv, to_name, fset, iflags);
248 numeric_id(const char *name, const char *type)
255 * We know that uid_t's and gid_t's are unsigned longs.
258 val = strtoul(name, &ep, 10);
260 err(EX_NOUSER, "%s", name);
262 errx(EX_NOUSER, "unknown %s %s", type, name);
268 * build a path name and install the file
271 install(const char *from_name, const char *to_name, u_long fset, u_int flags)
273 struct stat from_sb, temp_sb, to_sb;
274 struct timeval tvb[2];
275 int devnull, files_match, from_fd, serrno, target;
276 int tempcopy, temp_fd, to_fd;
277 char backup[MAXPATHLEN], *p, pathbuf[MAXPATHLEN], tempfile[MAXPATHLEN];
283 /* If try to install NULL file to a directory, fails. */
284 if (flags & DIRECTORY || strcmp(from_name, _PATH_DEVNULL)) {
285 if (stat(from_name, &from_sb))
286 err(EX_OSERR, "%s", from_name);
287 if (!S_ISREG(from_sb.st_mode)) {
289 err(EX_OSERR, "%s", from_name);
291 /* Build the target path. */
292 if (flags & DIRECTORY) {
293 (void)snprintf(pathbuf, sizeof(pathbuf), "%s/%s",
295 (p = strrchr(from_name, '/')) ? ++p : from_name);
303 target = stat(to_name, &to_sb) == 0;
305 /* Only install to regular files. */
306 if (target && !S_ISREG(to_sb.st_mode)) {
312 /* Only copy safe if the target exists. */
313 tempcopy = safecopy && target;
315 if (!devnull && (from_fd = open(from_name, O_RDONLY, 0)) < 0)
316 err(EX_OSERR, "%s", from_name);
318 /* If we don't strip, we can compare first. */
319 if (docompare && !dostrip && target) {
320 if ((to_fd = open(to_name, O_RDONLY, 0)) < 0)
321 err(EX_OSERR, "%s", to_name);
323 files_match = to_sb.st_size == 0;
325 files_match = !(compare(from_fd, from_name,
326 (size_t)from_sb.st_size, to_fd,
327 to_name, (size_t)to_sb.st_size));
329 /* Close "to" file unless we match. */
336 to_fd = create_tempfile(to_name, tempfile,
339 err(EX_OSERR, "%s", tempfile);
341 if ((to_fd = create_newfile(to_name, target,
343 err(EX_OSERR, "%s", to_name);
345 (void)printf("install: %s -> %s\n",
349 copy(from_fd, from_name, to_fd,
350 tempcopy ? tempfile : to_name, from_sb.st_size);
354 strip(tempcopy ? tempfile : to_name);
357 * Re-open our fd on the target, in case we used a strip
358 * that does not work in-place -- like GNU binutils strip.
361 to_fd = open(tempcopy ? tempfile : to_name, O_RDONLY, 0);
363 err(EX_OSERR, "stripping %s", to_name);
367 * Compare the stripped temp file with the target.
369 if (docompare && dostrip && target) {
372 /* Re-open to_fd using the real target name. */
373 if ((to_fd = open(to_name, O_RDONLY, 0)) < 0)
374 err(EX_OSERR, "%s", to_name);
376 if (fstat(temp_fd, &temp_sb)) {
378 (void)unlink(tempfile);
380 err(EX_OSERR, "%s", tempfile);
383 if (compare(temp_fd, tempfile, (size_t)temp_sb.st_size, to_fd,
384 to_name, (size_t)to_sb.st_size) == 0) {
386 * If target has more than one link we need to
387 * replace it in order to snap the extra links.
388 * Need to preserve target file times, though.
390 if (to_sb.st_nlink != 1) {
391 tvb[0].tv_sec = to_sb.st_atime;
393 tvb[1].tv_sec = to_sb.st_mtime;
395 (void)utimes(tempfile, tvb);
398 (void)unlink(tempfile);
400 (void) close(temp_fd);
405 * Move the new file into place if doing a safe copy
406 * and the files are different (or just not compared).
408 if (tempcopy && !files_match) {
409 /* Try to turn off the immutable bits. */
410 if (to_sb.st_flags & NOCHANGEBITS)
411 (void)chflags(to_name, to_sb.st_flags & ~NOCHANGEBITS);
413 if ((size_t)snprintf(backup, MAXPATHLEN, "%s%s", to_name,
414 suffix) != strlen(to_name) + strlen(suffix)) {
416 errx(EX_OSERR, "%s: backup filename too long",
420 (void)printf("install: %s -> %s\n", to_name, backup);
421 if (rename(to_name, backup) < 0) {
425 err(EX_OSERR, "rename: %s to %s", to_name,
430 (void)printf("install: %s -> %s\n", from_name, to_name);
431 if (rename(tempfile, to_name) < 0) {
435 err(EX_OSERR, "rename: %s to %s",
439 /* Re-open to_fd so we aren't hosed by the rename(2). */
441 if ((to_fd = open(to_name, O_RDONLY, 0)) < 0)
442 err(EX_OSERR, "%s", to_name);
446 * Preserve the timestamp of the source file if necessary.
448 if (dopreserve && !files_match && !devnull) {
449 tvb[0].tv_sec = from_sb.st_atime;
451 tvb[1].tv_sec = from_sb.st_mtime;
453 (void)utimes(to_name, tvb);
456 if (fstat(to_fd, &to_sb) == -1) {
458 (void)unlink(to_name);
460 err(EX_OSERR, "%s", to_name);
464 * Set owner, group, mode for target; do the chown first,
465 * chown may lose the setuid bits.
467 if ((gid != (gid_t)-1 && gid != to_sb.st_gid) ||
468 (uid != (uid_t)-1 && uid != to_sb.st_uid) ||
469 (mode != (to_sb.st_mode & ALLPERMS))) {
470 /* Try to turn off the immutable bits. */
471 if (to_sb.st_flags & NOCHANGEBITS)
472 (void)fchflags(to_fd, to_sb.st_flags & ~NOCHANGEBITS);
475 if ((gid != (gid_t)-1 && gid != to_sb.st_gid) ||
476 (uid != (uid_t)-1 && uid != to_sb.st_uid))
477 if (fchown(to_fd, uid, gid) == -1) {
479 (void)unlink(to_name);
481 err(EX_OSERR,"%s: chown/chgrp", to_name);
484 if (mode != (to_sb.st_mode & ALLPERMS))
485 if (fchmod(to_fd, mode)) {
487 (void)unlink(to_name);
489 err(EX_OSERR, "%s: chmod", to_name);
493 * If provided a set of flags, set them, otherwise, preserve the
494 * flags, except for the dump flag.
495 * NFS does not support flags. Ignore EOPNOTSUPP flags if we're just
496 * trying to turn off UF_NODUMP. If we're trying to set real flags,
497 * then warn if the fs doesn't support it, otherwise fail.
499 if (!devnull && (flags & SETFLAGS ||
500 (from_sb.st_flags & ~UF_NODUMP) != to_sb.st_flags) &&
502 flags & SETFLAGS ? fset : from_sb.st_flags & ~UF_NODUMP)) {
503 if (flags & SETFLAGS) {
504 if (errno == EOPNOTSUPP)
505 warn("%s: chflags", to_name);
508 (void)unlink(to_name);
510 err(EX_OSERR, "%s: chflags", to_name);
517 (void)close(from_fd);
522 * compare two files; non-zero means files differ
525 compare(int from_fd, const char *from_name __unused, size_t from_len,
526 int to_fd, const char *to_name __unused, size_t to_len)
533 if (from_len != to_len)
536 if (from_len <= MAX_CMP_SIZE) {
538 if (trymmap(from_fd) && trymmap(to_fd)) {
539 p = mmap(NULL, from_len, PROT_READ, MAP_SHARED, from_fd, (off_t)0);
540 if (p == (char *)MAP_FAILED)
542 q = mmap(NULL, from_len, PROT_READ, MAP_SHARED, to_fd, (off_t)0);
543 if (q == (char *)MAP_FAILED) {
548 rv = memcmp(p, q, from_len);
560 lseek(from_fd, 0, SEEK_SET);
561 lseek(to_fd, 0, SEEK_SET);
563 n1 = read(from_fd, buf1, sizeof(buf1));
567 n2 = read(to_fd, buf2, n1);
569 rv = memcmp(buf1, buf2, n1);
571 rv = 1; /* out of sync */
573 rv = 1; /* read failure */
575 lseek(from_fd, 0, SEEK_SET);
576 lseek(to_fd, 0, SEEK_SET);
579 rv = 1; /* don't bother in this case */
586 * create a temporary file based on path and open it
589 create_tempfile(const char *path, char *temp, size_t tsize)
593 (void)strncpy(temp, path, tsize);
594 temp[tsize - 1] = '\0';
595 if ((p = strrchr(temp, '/')) != NULL)
599 (void)strncpy(p, "INS@XXXX", &temp[tsize - 1] - p);
600 temp[tsize - 1] = '\0';
601 return (mkstemp(temp));
606 * create a new file, overwriting an existing one if necessary
609 create_newfile(const char *path, int target, struct stat *sbp)
611 char backup[MAXPATHLEN];
617 * Unlink now... avoid ETXTBSY errors later. Try to turn
618 * off the append/immutable bits -- if we fail, go ahead,
621 if (sbp->st_flags & NOCHANGEBITS)
622 (void)chflags(path, sbp->st_flags & ~NOCHANGEBITS);
625 if ((size_t)snprintf(backup, MAXPATHLEN, "%s%s",
626 path, suffix) != strlen(path) + strlen(suffix))
627 errx(EX_OSERR, "%s: backup filename too long",
629 (void)snprintf(backup, MAXPATHLEN, "%s%s",
632 (void)printf("install: %s -> %s\n",
634 if (rename(path, backup) < 0)
635 err(EX_OSERR, "rename: %s to %s", path, backup);
637 if (unlink(path) < 0)
641 newfd = open(path, O_CREAT | O_RDWR | O_TRUNC, S_IRUSR | S_IWUSR);
642 if (newfd < 0 && saved_errno != 0)
649 * copy from one file to another
652 copy(int from_fd, const char *from_name, int to_fd, const char *to_name,
657 char *p, buf[MAXBSIZE];
660 /* Rewind file descriptors. */
661 if (lseek(from_fd, (off_t)0, SEEK_SET) == (off_t)-1)
662 err(EX_OSERR, "lseek: %s", from_name);
663 if (lseek(to_fd, (off_t)0, SEEK_SET) == (off_t)-1)
664 err(EX_OSERR, "lseek: %s", to_name);
667 * Mmap and write if less than 8M (the limit is so we don't totally
668 * trash memory on big files. This is really a minor hack, but it
669 * wins some CPU back.
672 if (size <= 8 * 1048576 && trymmap(from_fd) &&
673 (p = mmap(NULL, (size_t)size, PROT_READ, MAP_SHARED,
674 from_fd, (off_t)0)) != (char *)MAP_FAILED) {
675 nw = write(to_fd, p, size);
678 (void)unlink(to_name);
681 "short write to %s: %jd bytes written, %jd bytes asked to write",
682 to_name, (uintmax_t)nw, (uintmax_t)size);
685 err(EX_OSERR, "%s", to_name);
691 while ((nr = read(from_fd, buf, sizeof(buf))) > 0)
692 if ((nw = write(to_fd, buf, nr)) != nr) {
694 (void)unlink(to_name);
697 "short write to %s: %jd bytes written, %jd bytes asked to write",
698 to_name, (uintmax_t)nw,
702 err(EX_OSERR, "%s", to_name);
707 (void)unlink(to_name);
709 err(EX_OSERR, "%s", from_name);
716 * use strip(1) to strip the target file
719 strip(const char *to_name)
721 const char *stripbin;
727 (void)unlink(to_name);
729 err(EX_TEMPFAIL, "fork");
731 stripbin = getenv("STRIPBIN");
732 if (stripbin == NULL)
734 execlp(stripbin, stripbin, to_name, (char *)NULL);
735 err(EX_OSERR, "exec(%s)", stripbin);
737 if (wait(&status) == -1 || status) {
739 (void)unlink(to_name);
740 errc(EX_SOFTWARE, serrno, "wait");
748 * build directory hierarchy
751 install_dir(char *path)
758 if (!*p || (p != path && *p == '/')) {
761 if (stat(path, &sb)) {
762 if (errno != ENOENT || mkdir(path, 0755) < 0) {
763 err(EX_OSERR, "mkdir %s", path);
766 (void)printf("install: mkdir %s\n",
768 } else if (!S_ISDIR(sb.st_mode))
769 errx(EX_OSERR, "%s exists but is not a directory", path);
774 if ((gid != (gid_t)-1 || uid != (uid_t)-1) && chown(path, uid, gid))
775 warn("chown %u:%u %s", uid, gid, path);
776 if (chmod(path, mode))
777 warn("chmod %o %s", mode, path);
782 * print a usage message and die
787 (void)fprintf(stderr,
788 "usage: install [-bCcMpSsv] [-B suffix] [-f flags] [-g group] [-m mode]\n"
789 " [-o owner] file1 file2\n"
790 " install [-bCcMpSsv] [-B suffix] [-f flags] [-g group] [-m mode]\n"
791 " [-o owner] file1 ... fileN directory\n"
792 " install -d [-v] [-g group] [-m mode] [-o owner] directory ...\n");
799 * return true (1) if mmap should be tried, false (0) if not.
805 * The ifdef is for bootstrapping - f_fstypename doesn't exist in
806 * pre-Lite2-merge systems.
811 if (nommap || fstatfs(fd, &stfs) != 0)
813 if (strcmp(stfs.f_fstypename, "ufs") == 0 ||
814 strcmp(stfs.f_fstypename, "cd9660") == 0)