Annotation of src/usr.bin/xinstall/xinstall.c, Revision 1.53
1.53 ! miod 1: /* $OpenBSD: xinstall.c,v 1.52 2012/09/14 00:00:29 millert Exp $ */
1.2 deraadt 2: /* $NetBSD: xinstall.c,v 1.9 1995/12/20 10:25:17 jonathan Exp $ */
1.1 deraadt 3:
4: /*
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.
1.34 millert 16: * 3. Neither the name of the University nor the names of its contributors
1.1 deraadt 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: #include <sys/param.h>
34: #include <sys/wait.h>
35: #include <sys/mman.h>
36: #include <sys/stat.h>
37:
38: #include <ctype.h>
1.4 millert 39: #include <err.h>
1.1 deraadt 40: #include <errno.h>
41: #include <fcntl.h>
42: #include <grp.h>
43: #include <paths.h>
44: #include <pwd.h>
45: #include <stdio.h>
46: #include <stdlib.h>
47: #include <string.h>
48: #include <unistd.h>
1.4 millert 49: #include <sysexits.h>
50: #include <utime.h>
1.1 deraadt 51:
52: #include "pathnames.h"
53:
1.19 millert 54: #define DIRECTORY 0x01 /* Tell install it's a directory. */
55: #define SETFLAGS 0x02 /* Tell install to set flags. */
56: #define NOCHANGEBITS (UF_IMMUTABLE | UF_APPEND | SF_IMMUTABLE | SF_APPEND)
57: #define BACKUP_SUFFIX ".old"
58:
1.1 deraadt 59: struct passwd *pp;
60: struct group *gp;
1.19 millert 61: int dobackup, docompare, dodir, dopreserve, dostrip, safecopy;
1.1 deraadt 62: int mode = S_IRWXU|S_IRGRP|S_IXGRP|S_IROTH|S_IXOTH;
1.4 millert 63: char pathbuf[MAXPATHLEN], tempfile[MAXPATHLEN];
1.19 millert 64: char *suffix = BACKUP_SUFFIX;
1.1 deraadt 65: uid_t uid;
66: gid_t gid;
67:
1.31 millert 68: void copy(int, char *, int, char *, off_t, int);
69: int compare(int, const char *, size_t, int, const char *, size_t);
70: void install(char *, char *, u_long, u_int);
71: void install_dir(char *);
72: void strip(char *);
73: void usage(void);
74: int create_newfile(char *, struct stat *);
75: int create_tempfile(char *, char *, size_t);
76: int file_write(int, char *, size_t, int *, int *, int);
1.33 tedu 77: void file_flush(int, int);
1.1 deraadt 78:
79: int
1.35 deraadt 80: main(int argc, char *argv[])
1.1 deraadt 81: {
82: struct stat from_sb, to_sb;
1.41 otto 83: void *set;
1.23 mickey 84: u_int32_t fset;
1.1 deraadt 85: u_int iflags;
86: int ch, no_target;
87: char *flags, *to_name, *group = NULL, *owner = NULL;
88:
89: iflags = 0;
1.47 jsg 90: while ((ch = getopt(argc, argv, "B:bCcdf:g:m:o:pSs")) != -1)
1.1 deraadt 91: switch((char)ch) {
1.4 millert 92: case 'C':
93: docompare = 1;
94: break;
1.19 millert 95: case 'B':
96: suffix = optarg;
97: /* fall through; -B implies -b */
98: case 'b':
99: dobackup = 1;
100: break;
1.1 deraadt 101: case 'c':
1.4 millert 102: /* For backwards compatibility. */
1.1 deraadt 103: break;
104: case 'f':
105: flags = optarg;
1.23 mickey 106: if (strtofflags(&flags, &fset, NULL))
1.4 millert 107: errx(EX_USAGE, "%s: invalid flag", flags);
1.1 deraadt 108: iflags |= SETFLAGS;
109: break;
110: case 'g':
111: group = optarg;
112: break;
113: case 'm':
114: if (!(set = setmode(optarg)))
1.4 millert 115: errx(EX_USAGE, "%s: invalid file mode", optarg);
1.1 deraadt 116: mode = getmode(set, 0);
1.16 deraadt 117: free(set);
1.1 deraadt 118: break;
119: case 'o':
120: owner = optarg;
121: break;
1.4 millert 122: case 'p':
123: docompare = dopreserve = 1;
124: break;
125: case 'S':
126: safecopy = 1;
127: break;
1.1 deraadt 128: case 's':
129: dostrip = 1;
130: break;
131: case 'd':
132: dodir = 1;
133: break;
134: case '?':
135: default:
136: usage();
137: }
138: argc -= optind;
139: argv += optind;
140:
1.4 millert 141: /* some options make no sense when creating directories */
142: if ((safecopy || docompare || dostrip) && dodir)
1.1 deraadt 143: usage();
144:
145: /* must have at least two arguments, except when creating directories */
146: if (argc < 2 && !dodir)
147: usage();
148:
1.4 millert 149: /* need to make a temp copy so we can compare stripped version */
150: if (docompare && dostrip)
151: safecopy = 1;
152:
1.1 deraadt 153: /* get group and owner id's */
154: if (group && !(gp = getgrnam(group)) && !isdigit(*group))
1.4 millert 155: errx(EX_NOUSER, "unknown group %s", group);
156: gid = (group) ? ((gp) ? gp->gr_gid : (gid_t)strtoul(group, NULL, 10)) : (gid_t)-1;
1.1 deraadt 157: if (owner && !(pp = getpwnam(owner)) && !isdigit(*owner))
1.4 millert 158: errx(EX_NOUSER, "unknown user %s", owner);
159: uid = (owner) ? ((pp) ? pp->pw_uid : (uid_t)strtoul(owner, NULL, 10)) : (uid_t)-1;
1.1 deraadt 160:
161: if (dodir) {
162: for (; *argv != NULL; ++argv)
163: install_dir(*argv);
1.4 millert 164: exit(EX_OK);
1.1 deraadt 165: /* NOTREACHED */
166: }
167:
168: no_target = stat(to_name = argv[argc - 1], &to_sb);
169: if (!no_target && S_ISDIR(to_sb.st_mode)) {
170: for (; *argv != to_name; ++argv)
171: install(*argv, to_name, fset, iflags | DIRECTORY);
1.4 millert 172: exit(EX_OK);
173: /* NOTREACHED */
1.1 deraadt 174: }
175:
176: /* can't do file1 file2 directory/file */
177: if (argc != 2)
1.32 millert 178: errx(EX_OSERR, "Target: %s", argv[argc-1]);
1.1 deraadt 179:
180: if (!no_target) {
181: if (stat(*argv, &from_sb))
1.4 millert 182: err(EX_OSERR, "%s", *argv);
1.1 deraadt 183: if (!S_ISREG(to_sb.st_mode))
1.4 millert 184: errx(EX_OSERR, "%s: %s", to_name, strerror(EFTYPE));
1.1 deraadt 185: if (to_sb.st_dev == from_sb.st_dev &&
186: to_sb.st_ino == from_sb.st_ino)
1.4 millert 187: errx(EX_USAGE, "%s and %s are the same file", *argv, to_name);
1.1 deraadt 188: }
189: install(*argv, to_name, fset, iflags);
1.4 millert 190: exit(EX_OK);
191: /* NOTREACHED */
1.1 deraadt 192: }
193:
194: /*
195: * install --
196: * build a path name and install the file
197: */
198: void
1.35 deraadt 199: install(char *from_name, char *to_name, u_long fset, u_int flags)
1.1 deraadt 200: {
201: struct stat from_sb, to_sb;
1.4 millert 202: struct utimbuf utb;
203: int devnull, from_fd, to_fd, serrno, files_match = 0;
1.1 deraadt 204: char *p;
205:
1.4 millert 206: (void)memset((void *)&from_sb, 0, sizeof(from_sb));
207: (void)memset((void *)&to_sb, 0, sizeof(to_sb));
208:
1.1 deraadt 209: /* If try to install NULL file to a directory, fails. */
210: if (flags & DIRECTORY || strcmp(from_name, _PATH_DEVNULL)) {
211: if (stat(from_name, &from_sb))
1.4 millert 212: err(EX_OSERR, "%s", from_name);
1.1 deraadt 213: if (!S_ISREG(from_sb.st_mode))
1.4 millert 214: errx(EX_OSERR, "%s: %s", from_name, strerror(EFTYPE));
1.1 deraadt 215: /* Build the target path. */
216: if (flags & DIRECTORY) {
217: (void)snprintf(pathbuf, sizeof(pathbuf), "%s/%s",
218: to_name,
1.11 millert 219: (p = strrchr(from_name, '/')) ? ++p : from_name);
1.1 deraadt 220: to_name = pathbuf;
221: }
222: devnull = 0;
223: } else {
224: devnull = 1;
225: }
226:
1.4 millert 227: if (stat(to_name, &to_sb) == 0) {
228: /* Only compare against regular files. */
229: if (docompare && !S_ISREG(to_sb.st_mode)) {
230: docompare = 0;
231: warnx("%s: %s", to_name, strerror(EFTYPE));
232: }
233: } else if (docompare) {
234: /* File does not exist so silently ignore compare flag. */
235: docompare = 0;
236: }
237:
1.52 millert 238: if (!devnull) {
239: if ((from_fd = open(from_name, O_RDONLY, 0)) < 0)
240: err(EX_OSERR, "%s", from_name);
241: }
242:
1.4 millert 243: if (safecopy) {
244: to_fd = create_tempfile(to_name, tempfile, sizeof(tempfile));
245: if (to_fd < 0)
246: err(EX_OSERR, "%s", tempfile);
247: } else if (docompare && !dostrip) {
248: if ((to_fd = open(to_name, O_RDONLY, 0)) < 0)
249: err(EX_OSERR, "%s", to_name);
250: } else {
251: if ((to_fd = create_newfile(to_name, &to_sb)) < 0)
252: err(EX_OSERR, "%s", to_name);
253: }
254:
1.1 deraadt 255: if (!devnull) {
1.4 millert 256: if (docompare && !safecopy) {
257: files_match = !(compare(from_fd, from_name,
258: (size_t)from_sb.st_size, to_fd,
259: to_name, (size_t)to_sb.st_size));
260:
261: /* Truncate "to" file for copy unless we match */
262: if (!files_match) {
263: (void)close(to_fd);
264: if ((to_fd = create_newfile(to_name, &to_sb)) < 0)
265: err(EX_OSERR, "%s", to_name);
266: }
1.1 deraadt 267: }
1.4 millert 268: if (!files_match)
269: copy(from_fd, from_name, to_fd,
1.17 millert 270: safecopy ? tempfile : to_name, from_sb.st_size,
271: ((off_t)from_sb.st_blocks * S_BLKSIZE < from_sb.st_size));
1.1 deraadt 272: }
1.2 deraadt 273:
274: if (dostrip) {
1.4 millert 275: strip(safecopy ? tempfile : to_name);
1.2 deraadt 276:
277: /*
278: * Re-open our fd on the target, in case we used a strip
279: * that does not work in-place -- like gnu binutils strip.
280: */
281: close(to_fd);
1.4 millert 282: if ((to_fd = open(safecopy ? tempfile : to_name, O_RDONLY,
283: 0)) < 0)
284: err(EX_OSERR, "stripping %s", to_name);
285: }
286:
287: /*
288: * Compare the (possibly stripped) temp file to the target.
289: */
290: if (safecopy && docompare) {
291: int temp_fd = to_fd;
292: struct stat temp_sb;
293:
294: /* Re-open to_fd using the real target name. */
295: if ((to_fd = open(to_name, O_RDONLY, 0)) < 0)
296: err(EX_OSERR, "%s", to_name);
297:
298: if (fstat(temp_fd, &temp_sb)) {
299: serrno = errno;
300: (void)unlink(tempfile);
301: errx(EX_OSERR, "%s: %s", tempfile, strerror(serrno));
302: }
303:
304: if (compare(temp_fd, tempfile, (size_t)temp_sb.st_size, to_fd,
305: to_name, (size_t)to_sb.st_size) == 0) {
306: /*
307: * If target has more than one link we need to
308: * replace it in order to snap the extra links.
309: * Need to preserve target file times, though.
310: */
311: if (to_sb.st_nlink != 1) {
312: utb.actime = to_sb.st_atime;
313: utb.modtime = to_sb.st_mtime;
314: (void)utime(tempfile, &utb);
315: } else {
316: files_match = 1;
317: (void)unlink(tempfile);
318: }
319: }
1.48 phessler 320: (void)close(to_fd);
321: to_fd = temp_fd;
1.4 millert 322: }
323:
324: /*
1.30 millert 325: * Preserve the timestamp of the source file if necessary.
1.4 millert 326: */
327: if (dopreserve && !files_match) {
328: utb.actime = from_sb.st_atime;
329: utb.modtime = from_sb.st_mtime;
1.48 phessler 330: (void)utime(safecopy ? tempfile : to_name, &utb);
1.2 deraadt 331: }
332:
1.1 deraadt 333: /*
334: * Set owner, group, mode for target; do the chown first,
335: * chown may lose the setuid bits.
336: */
1.48 phessler 337: if ((gid != (gid_t)-1 || uid != (uid_t)-1) &&
338: fchown(to_fd, uid, gid)) {
1.1 deraadt 339: serrno = errno;
1.48 phessler 340: (void)unlink(safecopy ? tempfile : to_name);
341: errx(EX_OSERR, "%s: chown/chgrp: %s",
342: safecopy ? tempfile : to_name, strerror(serrno));
1.1 deraadt 343: }
344: if (fchmod(to_fd, mode)) {
345: serrno = errno;
1.48 phessler 346: (void)unlink(safecopy ? tempfile : to_name);
347: errx(EX_OSERR, "%s: chmod: %s", safecopy ? tempfile : to_name,
348: strerror(serrno));
1.1 deraadt 349: }
350:
351: /*
352: * If provided a set of flags, set them, otherwise, preserve the
353: * flags, except for the dump flag.
354: */
355: if (fchflags(to_fd,
356: flags & SETFLAGS ? fset : from_sb.st_flags & ~UF_NODUMP)) {
1.12 millert 357: if (errno != EOPNOTSUPP || (from_sb.st_flags & ~UF_NODUMP) != 0)
1.48 phessler 358: warnx("%s: chflags: %s",
359: safecopy ? tempfile :to_name, strerror(errno));
1.1 deraadt 360: }
361:
362: (void)close(to_fd);
1.18 millert 363: if (!devnull)
364: (void)close(from_fd);
1.48 phessler 365:
366: /*
367: * Move the new file into place if doing a safe copy
368: * and the files are different (or just not compared).
369: */
370: if (safecopy && !files_match) {
371: /* Try to turn off the immutable bits. */
372: if (to_sb.st_flags & (NOCHANGEBITS))
373: (void)chflags(to_name, to_sb.st_flags & ~(NOCHANGEBITS));
374: if (dobackup) {
375: char backup[MAXPATHLEN];
376: (void)snprintf(backup, MAXPATHLEN, "%s%s", to_name,
377: suffix);
378: /* It is ok for the target file not to exist. */
379: if (rename(to_name, backup) < 0 && errno != ENOENT) {
380: serrno = errno;
381: unlink(tempfile);
382: errx(EX_OSERR, "rename: %s to %s: %s", to_name,
383: backup, strerror(serrno));
384: }
385: }
386: if (rename(tempfile, to_name) < 0 ) {
387: serrno = errno;
388: unlink(tempfile);
389: errx(EX_OSERR, "rename: %s to %s: %s", tempfile,
390: to_name, strerror(serrno));
391: }
392: }
1.1 deraadt 393: }
394:
395: /*
396: * copy --
397: * copy from one file to another
398: */
399: void
1.35 deraadt 400: copy(int from_fd, char *from_name, int to_fd, char *to_name, off_t size,
401: int sparse)
1.1 deraadt 402: {
1.29 mpech 403: ssize_t nr, nw;
1.1 deraadt 404: int serrno;
405: char *p, buf[MAXBSIZE];
406:
1.51 millert 407: if (size == 0)
408: return;
409:
1.7 millert 410: /* Rewind file descriptors. */
411: if (lseek(from_fd, (off_t)0, SEEK_SET) == (off_t)-1)
412: err(EX_OSERR, "lseek: %s", from_name);
413: if (lseek(to_fd, (off_t)0, SEEK_SET) == (off_t)-1)
414: err(EX_OSERR, "lseek: %s", to_name);
415:
1.1 deraadt 416: /*
417: * Mmap and write if less than 8M (the limit is so we don't totally
418: * trash memory on big files. This is really a minor hack, but it
1.17 millert 419: * wins some CPU back. Sparse files need special treatment.
1.1 deraadt 420: */
1.51 millert 421: if (!sparse && size <= 8 * 1048576) {
1.50 espie 422: size_t siz;
1.17 millert 423:
1.19 millert 424: if ((p = mmap(NULL, (size_t)size, PROT_READ, MAP_PRIVATE,
1.38 grange 425: from_fd, (off_t)0)) == MAP_FAILED) {
1.6 millert 426: serrno = errno;
427: (void)unlink(to_name);
428: errx(EX_OSERR, "%s: %s", from_name, strerror(serrno));
429: }
1.50 espie 430: madvise(p, size, MADV_SEQUENTIAL);
1.3 deraadt 431: siz = (size_t)size;
1.6 millert 432: if ((nw = write(to_fd, p, siz)) != siz) {
433: serrno = errno;
434: (void)unlink(to_name);
435: errx(EX_OSERR, "%s: %s",
436: to_name, strerror(nw > 0 ? EIO : serrno));
437: }
1.4 millert 438: (void) munmap(p, (size_t)size);
1.1 deraadt 439: } else {
1.17 millert 440: int sz, rem, isem = 1;
441: struct stat sb;
442:
443: /*
444: * Pass the blocksize of the file being written to the write
445: * routine. if the size is zero, use the default S_BLKSIZE.
446: */
447: if (fstat(to_fd, &sb) != 0 || sb.st_blksize == 0)
448: sz = S_BLKSIZE;
449: else
450: sz = sb.st_blksize;
1.20 millert 451: rem = sz;
1.17 millert 452:
453: while ((nr = read(from_fd, buf, sizeof(buf))) > 0) {
454: if (sparse)
455: nw = file_write(to_fd, buf, nr, &rem, &isem, sz);
456: else
457: nw = write(to_fd, buf, nr);
458: if (nw != nr) {
1.1 deraadt 459: serrno = errno;
460: (void)unlink(to_name);
1.4 millert 461: errx(EX_OSERR, "%s: %s",
1.1 deraadt 462: to_name, strerror(nw > 0 ? EIO : serrno));
463: }
1.17 millert 464: }
1.33 tedu 465: if (sparse)
466: file_flush(to_fd, isem);
1.1 deraadt 467: if (nr != 0) {
468: serrno = errno;
469: (void)unlink(to_name);
1.4 millert 470: errx(EX_OSERR, "%s: %s", from_name, strerror(serrno));
1.1 deraadt 471: }
472: }
473: }
474:
475: /*
1.4 millert 476: * compare --
477: * compare two files; non-zero means files differ
478: */
479: int
1.35 deraadt 480: compare(int from_fd, const char *from_name, size_t from_len, int to_fd,
481: const char *to_name, size_t to_len)
1.4 millert 482: {
483: caddr_t p1, p2;
1.21 millert 484: size_t length, remainder;
485: off_t from_off, to_off;
1.4 millert 486: int dfound;
487:
1.50 espie 488: if (from_len == 0 && from_len == to_len)
489: return (0);
490:
1.4 millert 491: if (from_len != to_len)
1.50 espie 492: return (1);
1.4 millert 493:
494: /*
495: * Compare the two files being careful not to mmap
496: * more than 8M at a time.
497: */
1.21 millert 498: from_off = to_off = (off_t)0;
1.4 millert 499: remainder = from_len;
500: do {
501: length = MIN(remainder, 8 * 1048576);
502: remainder -= length;
503:
1.22 mickey 504: if ((p1 = mmap(NULL, length, PROT_READ, MAP_PRIVATE,
1.38 grange 505: from_fd, from_off)) == MAP_FAILED)
1.4 millert 506: err(EX_OSERR, "%s", from_name);
1.22 mickey 507: if ((p2 = mmap(NULL, length, PROT_READ, MAP_PRIVATE,
1.38 grange 508: to_fd, to_off)) == MAP_FAILED)
1.4 millert 509: err(EX_OSERR, "%s", to_name);
1.39 mickey 510: if (length) {
511: madvise(p1, length, MADV_SEQUENTIAL);
512: madvise(p2, length, MADV_SEQUENTIAL);
513: }
1.4 millert 514:
515: dfound = memcmp(p1, p2, length);
516:
517: (void) munmap(p1, length);
518: (void) munmap(p2, length);
1.21 millert 519:
520: from_off += length;
521: to_off += length;
1.4 millert 522:
1.7 millert 523: } while (!dfound && remainder > 0);
1.4 millert 524:
525: return(dfound);
526: }
527:
528: /*
1.1 deraadt 529: * strip --
530: * use strip(1) to strip the target file
531: */
532: void
1.35 deraadt 533: strip(char *to_name)
1.1 deraadt 534: {
535: int serrno, status;
1.26 millert 536: char * volatile path_strip;
1.18 millert 537:
1.13 millert 538: if (issetugid() || (path_strip = getenv("STRIP")) == NULL)
539: path_strip = _PATH_STRIP;
1.1 deraadt 540:
541: switch (vfork()) {
542: case -1:
543: serrno = errno;
544: (void)unlink(to_name);
1.4 millert 545: errx(EX_TEMPFAIL, "forks: %s", strerror(serrno));
1.1 deraadt 546: case 0:
1.44 moritz 547: execl(path_strip, "strip", "--", to_name, (char *)NULL);
1.13 millert 548: warn("%s", path_strip);
1.9 deraadt 549: _exit(EX_OSERR);
1.1 deraadt 550: default:
1.14 millert 551: if (wait(&status) == -1 || !WIFEXITED(status))
1.1 deraadt 552: (void)unlink(to_name);
553: }
554: }
555:
556: /*
557: * install_dir --
1.42 jsg 558: * build directory hierarchy
1.1 deraadt 559: */
560: void
1.35 deraadt 561: install_dir(char *path)
1.1 deraadt 562: {
1.29 mpech 563: char *p;
1.8 imp 564: struct stat sb;
565: int ch;
566:
567: for (p = path;; ++p)
568: if (!*p || (p != path && *p == '/')) {
569: ch = *p;
570: *p = '\0';
571: if (stat(path, &sb)) {
572: if (errno != ENOENT || mkdir(path, 0777) < 0) {
1.4 millert 573: err(EX_OSERR, "%s", path);
1.1 deraadt 574: /* NOTREACHED */
1.8 imp 575: }
576: }
577: if (!(*p = ch))
1.1 deraadt 578: break;
1.8 imp 579: }
1.1 deraadt 580:
1.4 millert 581: if (((gid != (gid_t)-1 || uid != (uid_t)-1) && chown(path, uid, gid)) ||
1.8 imp 582: chmod(path, mode)) {
583: warn("%s", path);
584: }
1.1 deraadt 585: }
586:
587: /*
588: * usage --
589: * print a usage message and die
590: */
591: void
1.35 deraadt 592: usage(void)
1.1 deraadt 593: {
594: (void)fprintf(stderr, "\
1.46 sobrado 595: usage: install [-bCcdpSs] [-B suffix] [-f flags] [-g group] [-m mode] [-o owner]\n source ... target ...\n");
1.4 millert 596: exit(EX_USAGE);
597: /* NOTREACHED */
598: }
599:
600: /*
601: * create_tempfile --
602: * create a temporary file based on path and open it
603: */
604: int
1.35 deraadt 605: create_tempfile(char *path, char *temp, size_t tsize)
1.4 millert 606: {
607: char *p;
608:
1.43 millert 609: strlcpy(temp, path, tsize);
610: if ((p = strrchr(temp, '/')) != NULL)
1.4 millert 611: p++;
612: else
613: p = temp;
1.43 millert 614: *p = '\0';
615: strlcat(p, "INS@XXXXXXXXXX", tsize);
1.4 millert 616:
617: return(mkstemp(temp));
618: }
619:
620: /*
621: * create_newfile --
1.30 millert 622: * create a new file, overwriting an existing one if necessary
1.4 millert 623: */
624: int
1.35 deraadt 625: create_newfile(char *path, struct stat *sbp)
1.4 millert 626: {
1.19 millert 627: char backup[MAXPATHLEN];
628:
1.4 millert 629: /*
630: * Unlink now... avoid ETXTBSY errors later. Try and turn
631: * off the append/immutable bits -- if we fail, go ahead,
632: * it might work.
633: */
634: if (sbp->st_flags & (NOCHANGEBITS))
635: (void)chflags(path, sbp->st_flags & ~(NOCHANGEBITS));
636:
1.19 millert 637: if (dobackup) {
638: (void)snprintf(backup, MAXPATHLEN, "%s%s", path, suffix);
1.28 heko 639: /* It is ok for the target file not to exist. */
640: if (rename(path, backup) < 0 && errno != ENOENT)
641: err(EX_OSERR, "rename: %s to %s (errno %d)", path, backup, errno);
1.53 ! miod 642: } else {
! 643: if (unlink(path) < 0 && errno != ENOENT)
! 644: err(EX_OSERR, "%s", path);
! 645: }
1.4 millert 646:
1.52 millert 647: return(open(path, O_CREAT | O_RDWR | O_EXCL, S_IRUSR | S_IWUSR));
1.17 millert 648: }
649:
650: /*
651: * file_write()
652: * Write/copy a file (during copy or archive extract). This routine knows
653: * how to copy files with lseek holes in it. (Which are read as file
654: * blocks containing all 0's but do not have any file blocks associated
655: * with the data). Typical examples of these are files created by dbm
656: * variants (.pag files). While the file size of these files are huge, the
657: * actual storage is quite small (the files are sparse). The problem is
658: * the holes read as all zeros so are probably stored on the archive that
659: * way (there is no way to determine if the file block is really a hole,
660: * we only know that a file block of all zero's can be a hole).
661: * At this writing, no major archive format knows how to archive files
662: * with holes. However, on extraction (or during copy, -rw) we have to
663: * deal with these files. Without detecting the holes, the files can
664: * consume a lot of file space if just written to disk. This replacement
665: * for write when passed the basic allocation size of a file system block,
666: * uses lseek whenever it detects the input data is all 0 within that
667: * file block. In more detail, the strategy is as follows:
668: * While the input is all zero keep doing an lseek. Keep track of when we
1.45 krw 669: * pass over file block boundaries. Only write when we hit a non zero
1.17 millert 670: * input. once we have written a file block, we continue to write it to
671: * the end (we stop looking at the input). When we reach the start of the
672: * next file block, start checking for zero blocks again. Working on file
1.45 krw 673: * block boundaries significantly reduces the overhead when copying files
1.17 millert 674: * that are NOT very sparse. This overhead (when compared to a write) is
675: * almost below the measurement resolution on many systems. Without it,
676: * files with holes cannot be safely copied. It does has a side effect as
677: * it can put holes into files that did not have them before, but that is
678: * not a problem since the file contents are unchanged (in fact it saves
679: * file space). (Except on paging files for diskless clients. But since we
680: * cannot determine one of those file from here, we ignore them). If this
681: * ever ends up on a system where CTG files are supported and the holes
682: * are not desired, just do a conditional test in those routines that
683: * call file_write() and have it call write() instead. BEFORE CLOSING THE
684: * FILE, make sure to call file_flush() when the last write finishes with
685: * an empty block. A lot of file systems will not create an lseek hole at
686: * the end. In this case we drop a single 0 at the end to force the
687: * trailing 0's in the file.
688: * ---Parameters---
689: * rem: how many bytes left in this file system block
690: * isempt: have we written to the file block yet (is it empty)
691: * sz: basic file block allocation size
692: * cnt: number of bytes on this write
693: * str: buffer to write
694: * Return:
695: * number of bytes written, -1 on write (or lseek) error.
696: */
697:
698: int
1.35 deraadt 699: file_write(int fd, char *str, size_t cnt, int *rem, int *isempt, int sz)
1.17 millert 700: {
1.29 mpech 701: char *pt;
702: char *end;
703: size_t wcnt;
704: char *st = str;
1.17 millert 705:
706: /*
707: * while we have data to process
708: */
709: while (cnt) {
710: if (!*rem) {
711: /*
712: * We are now at the start of file system block again
713: * (or what we think one is...). start looking for
714: * empty blocks again
715: */
716: *isempt = 1;
717: *rem = sz;
718: }
719:
720: /*
721: * only examine up to the end of the current file block or
722: * remaining characters to write, whatever is smaller
723: */
724: wcnt = MIN(cnt, *rem);
725: cnt -= wcnt;
726: *rem -= wcnt;
727: if (*isempt) {
728: /*
729: * have not written to this block yet, so we keep
730: * looking for zero's
731: */
732: pt = st;
733: end = st + wcnt;
734:
735: /*
736: * look for a zero filled buffer
737: */
738: while ((pt < end) && (*pt == '\0'))
739: ++pt;
740:
741: if (pt == end) {
742: /*
743: * skip, buf is empty so far
744: */
745: if (lseek(fd, (off_t)wcnt, SEEK_CUR) < 0) {
746: warn("lseek");
747: return(-1);
748: }
749: st = pt;
750: continue;
751: }
752: /*
753: * drat, the buf is not zero filled
754: */
755: *isempt = 0;
756: }
757:
758: /*
759: * have non-zero data in this file system block, have to write
760: */
761: if (write(fd, st, wcnt) != wcnt) {
762: warn("write");
763: return(-1);
764: }
765: st += wcnt;
766: }
767: return(st - str);
1.33 tedu 768: }
769:
770: /*
771: * file_flush()
772: * when the last file block in a file is zero, many file systems will not
773: * let us create a hole at the end. To get the last block with zeros, we
774: * write the last BYTE with a zero (back up one byte and write a zero).
775: */
776: void
777: file_flush(int fd, int isempt)
778: {
779: static char blnk[] = "\0";
780:
781: /*
782: * silly test, but make sure we are only called when the last block is
783: * filled with all zeros.
784: */
785: if (!isempt)
786: return;
787:
788: /*
789: * move back one byte and write a zero
790: */
791: if (lseek(fd, (off_t)-1, SEEK_CUR) < 0) {
792: warn("Failed seek on file");
793: return;
794: }
795:
796: if (write(fd, blnk, 1) < 0)
797: warn("Failed write to file");
798: return;
1.1 deraadt 799: }