Home | History | Annotate | Line # | Download | only in cp
utils.c revision 1.34
      1  1.34      hira /* $NetBSD: utils.c,v 1.34 2007/10/26 16:21:25 hira Exp $ */
      2   1.3       cgd 
      3   1.1   mycroft /*-
      4   1.1   mycroft  * Copyright (c) 1991, 1993, 1994
      5   1.1   mycroft  *	The Regents of the University of California.  All rights reserved.
      6   1.1   mycroft  *
      7   1.1   mycroft  * Redistribution and use in source and binary forms, with or without
      8   1.1   mycroft  * modification, are permitted provided that the following conditions
      9   1.1   mycroft  * are met:
     10   1.1   mycroft  * 1. Redistributions of source code must retain the above copyright
     11   1.1   mycroft  *    notice, this list of conditions and the following disclaimer.
     12   1.1   mycroft  * 2. Redistributions in binary form must reproduce the above copyright
     13   1.1   mycroft  *    notice, this list of conditions and the following disclaimer in the
     14   1.1   mycroft  *    documentation and/or other materials provided with the distribution.
     15  1.25       agc  * 3. Neither the name of the University nor the names of its contributors
     16   1.1   mycroft  *    may be used to endorse or promote products derived from this software
     17   1.1   mycroft  *    without specific prior written permission.
     18   1.1   mycroft  *
     19   1.1   mycroft  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     20   1.1   mycroft  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     21   1.1   mycroft  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     22   1.1   mycroft  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     23   1.1   mycroft  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     24   1.1   mycroft  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     25   1.1   mycroft  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     26   1.1   mycroft  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     27   1.1   mycroft  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     28   1.1   mycroft  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     29   1.1   mycroft  * SUCH DAMAGE.
     30   1.1   mycroft  */
     31   1.1   mycroft 
     32   1.8   thorpej #include <sys/cdefs.h>
     33   1.1   mycroft #ifndef lint
     34   1.3       cgd #if 0
     35   1.3       cgd static char sccsid[] = "@(#)utils.c	8.3 (Berkeley) 4/1/94";
     36   1.3       cgd #else
     37  1.34      hira __RCSID("$NetBSD: utils.c,v 1.34 2007/10/26 16:21:25 hira Exp $");
     38   1.3       cgd #endif
     39   1.1   mycroft #endif /* not lint */
     40   1.1   mycroft 
     41  1.20       wiz #include <sys/mman.h>
     42   1.1   mycroft #include <sys/param.h>
     43   1.1   mycroft #include <sys/stat.h>
     44   1.1   mycroft #include <sys/time.h>
     45   1.1   mycroft 
     46   1.1   mycroft #include <err.h>
     47   1.1   mycroft #include <errno.h>
     48   1.1   mycroft #include <fcntl.h>
     49   1.1   mycroft #include <fts.h>
     50   1.1   mycroft #include <stdio.h>
     51   1.1   mycroft #include <stdlib.h>
     52   1.1   mycroft #include <string.h>
     53   1.1   mycroft #include <unistd.h>
     54   1.1   mycroft 
     55   1.1   mycroft #include "extern.h"
     56   1.1   mycroft 
     57   1.1   mycroft int
     58  1.20       wiz set_utimes(const char *file, struct stat *fs)
     59  1.16  wsanchez {
     60  1.16  wsanchez     static struct timeval tv[2];
     61  1.16  wsanchez 
     62  1.16  wsanchez     TIMESPEC_TO_TIMEVAL(&tv[0], &fs->st_atimespec);
     63  1.16  wsanchez     TIMESPEC_TO_TIMEVAL(&tv[1], &fs->st_mtimespec);
     64  1.16  wsanchez 
     65  1.17       mjl     if (lutimes(file, tv)) {
     66  1.17       mjl 	warn("lutimes: %s", file);
     67  1.16  wsanchez 	return (1);
     68  1.16  wsanchez     }
     69  1.16  wsanchez     return (0);
     70  1.16  wsanchez }
     71  1.16  wsanchez 
     72  1.16  wsanchez int
     73  1.20       wiz copy_file(FTSENT *entp, int dne)
     74   1.1   mycroft {
     75   1.1   mycroft 	static char buf[MAXBSIZE];
     76   1.1   mycroft 	struct stat to_stat, *fs;
     77  1.32  jschauma 	int ch, checkch, from_fd, rcount, rval, to_fd, tolnk, wcount;
     78   1.1   mycroft 	char *p;
     79   1.1   mycroft 
     80   1.1   mycroft 	if ((from_fd = open(entp->fts_path, O_RDONLY, 0)) == -1) {
     81   1.1   mycroft 		warn("%s", entp->fts_path);
     82   1.1   mycroft 		return (1);
     83   1.1   mycroft 	}
     84   1.1   mycroft 
     85  1.32  jschauma 	to_fd = -1;
     86   1.1   mycroft 	fs = entp->fts_statp;
     87  1.32  jschauma 	tolnk = ((Rflag && !(Lflag || Hflag)) || Pflag);
     88   1.1   mycroft 
     89   1.1   mycroft 	/*
     90   1.1   mycroft 	 * If the file exists and we're interactive, verify with the user.
     91   1.1   mycroft 	 * If the file DNE, set the mode to be the from file, minus setuid
     92   1.1   mycroft 	 * bits, modified by the umask; arguably wrong, but it makes copying
     93   1.1   mycroft 	 * executables work right and it's been that way forever.  (The
     94   1.1   mycroft 	 * other choice is 666 or'ed with the execute bits on the from file
     95   1.1   mycroft 	 * modified by the umask.)
     96   1.1   mycroft 	 */
     97   1.1   mycroft 	if (!dne) {
     98  1.32  jschauma 		struct stat sb;
     99  1.32  jschauma 		int sval;
    100  1.32  jschauma 
    101   1.1   mycroft 		if (iflag) {
    102   1.1   mycroft 			(void)fprintf(stderr, "overwrite %s? ", to.p_path);
    103   1.1   mycroft 			checkch = ch = getchar();
    104   1.1   mycroft 			while (ch != '\n' && ch != EOF)
    105   1.1   mycroft 				ch = getchar();
    106   1.2   mycroft 			if (checkch != 'y' && checkch != 'Y') {
    107   1.1   mycroft 				(void)close(from_fd);
    108   1.1   mycroft 				return (0);
    109   1.1   mycroft 			}
    110   1.1   mycroft 		}
    111  1.32  jschauma 
    112  1.32  jschauma 		sval = tolnk ?
    113  1.32  jschauma 			lstat(to.p_path, &sb) : stat(to.p_path, &sb);
    114  1.32  jschauma 		if (sval == -1) {
    115  1.32  jschauma 			warn("stat: %s", to.p_path);
    116  1.32  jschauma 			return (1);
    117  1.32  jschauma 		}
    118  1.32  jschauma 
    119  1.32  jschauma 		if (!(tolnk && S_ISLNK(sb.st_mode)))
    120  1.32  jschauma 			to_fd = open(to.p_path, O_WRONLY | O_TRUNC, 0);
    121   1.1   mycroft 	} else
    122   1.1   mycroft 		to_fd = open(to.p_path, O_WRONLY | O_TRUNC | O_CREAT,
    123   1.1   mycroft 		    fs->st_mode & ~(S_ISUID | S_ISGID));
    124   1.1   mycroft 
    125  1.32  jschauma 	if (to_fd == -1 && (fflag || tolnk)) {
    126  1.16  wsanchez 		/*
    127  1.16  wsanchez 		 * attempt to remove existing destination file name and
    128  1.16  wsanchez 		 * create a new file
    129  1.16  wsanchez 		 */
    130  1.16  wsanchez 		(void)unlink(to.p_path);
    131  1.16  wsanchez 		to_fd = open(to.p_path, O_WRONLY | O_TRUNC | O_CREAT,
    132  1.16  wsanchez 			     fs->st_mode & ~(S_ISUID | S_ISGID));
    133  1.16  wsanchez 	}
    134  1.16  wsanchez 
    135   1.1   mycroft 	if (to_fd == -1) {
    136   1.1   mycroft 		warn("%s", to.p_path);
    137   1.1   mycroft 		(void)close(from_fd);
    138  1.23    simonb 		return (1);
    139   1.1   mycroft 	}
    140   1.1   mycroft 
    141   1.1   mycroft 	rval = 0;
    142   1.1   mycroft 
    143   1.1   mycroft 	/*
    144  1.16  wsanchez 	 * There's no reason to do anything other than close the file
    145  1.16  wsanchez 	 * now if it's empty, so let's not bother.
    146   1.1   mycroft 	 */
    147  1.19       chs 
    148  1.16  wsanchez 	if (fs->st_size > 0) {
    149  1.19       chs 
    150  1.16  wsanchez 		/*
    151  1.19       chs 		 * Mmap and write if less than 8M (the limit is so
    152  1.19       chs 		 * we don't totally trash memory on big files).
    153  1.19       chs 		 * This is really a minor hack, but it wins some CPU back.
    154  1.16  wsanchez 		 */
    155  1.19       chs 
    156  1.16  wsanchez 		if (fs->st_size <= 8 * 1048576) {
    157  1.29  christos 			size_t fsize = (size_t)fs->st_size;
    158  1.29  christos 			p = mmap(NULL, fsize, PROT_READ, MAP_FILE|MAP_SHARED,
    159  1.29  christos 			    from_fd, (off_t)0);
    160  1.19       chs 			if (p == MAP_FAILED) {
    161  1.19       chs 				goto mmap_failed;
    162  1.16  wsanchez 			} else {
    163  1.19       chs 				(void) madvise(p, (size_t)fs->st_size,
    164  1.19       chs 				     MADV_SEQUENTIAL);
    165  1.29  christos 				if (write(to_fd, p, fsize) !=
    166  1.19       chs 				    fs->st_size) {
    167  1.16  wsanchez 					warn("%s", to.p_path);
    168  1.16  wsanchez 					rval = 1;
    169  1.16  wsanchez 				}
    170  1.29  christos 				if (munmap(p, fsize) < 0) {
    171  1.16  wsanchez 					warn("%s", entp->fts_path);
    172  1.16  wsanchez 					rval = 1;
    173  1.16  wsanchez 				}
    174   1.1   mycroft 			}
    175  1.19       chs 		} else {
    176  1.19       chs mmap_failed:
    177  1.16  wsanchez 			while ((rcount = read(from_fd, buf, MAXBSIZE)) > 0) {
    178  1.29  christos 				wcount = write(to_fd, buf, (size_t)rcount);
    179  1.16  wsanchez 				if (rcount != wcount || wcount == -1) {
    180  1.16  wsanchez 					warn("%s", to.p_path);
    181  1.16  wsanchez 					rval = 1;
    182  1.16  wsanchez 					break;
    183  1.16  wsanchez 				}
    184  1.16  wsanchez 			}
    185  1.16  wsanchez 			if (rcount < 0) {
    186  1.16  wsanchez 				warn("%s", entp->fts_path);
    187   1.1   mycroft 				rval = 1;
    188   1.1   mycroft 			}
    189   1.1   mycroft 		}
    190   1.1   mycroft 	}
    191   1.1   mycroft 
    192   1.1   mycroft 	if (rval == 1) {
    193   1.1   mycroft 		(void)close(from_fd);
    194   1.1   mycroft 		(void)close(to_fd);
    195   1.1   mycroft 		return (1);
    196   1.1   mycroft 	}
    197   1.1   mycroft 
    198   1.1   mycroft 	if (pflag && setfile(fs, to_fd))
    199   1.1   mycroft 		rval = 1;
    200   1.1   mycroft 	/*
    201   1.1   mycroft 	 * If the source was setuid or setgid, lose the bits unless the
    202   1.1   mycroft 	 * copy is owned by the same user and group.
    203   1.1   mycroft 	 */
    204   1.1   mycroft #define	RETAINBITS \
    205   1.1   mycroft 	(S_ISUID | S_ISGID | S_ISVTX | S_IRWXU | S_IRWXG | S_IRWXO)
    206  1.33       jld 	if (!pflag && dne
    207  1.33       jld 	    && fs->st_mode & (S_ISUID | S_ISGID) && fs->st_uid == myuid) {
    208   1.1   mycroft 		if (fstat(to_fd, &to_stat)) {
    209   1.1   mycroft 			warn("%s", to.p_path);
    210   1.1   mycroft 			rval = 1;
    211   1.1   mycroft 		} else if (fs->st_gid == to_stat.st_gid &&
    212   1.1   mycroft 		    fchmod(to_fd, fs->st_mode & RETAINBITS & ~myumask)) {
    213   1.1   mycroft 			warn("%s", to.p_path);
    214   1.1   mycroft 			rval = 1;
    215   1.1   mycroft 		}
    216  1.15   thorpej 	}
    217   1.1   mycroft 	(void)close(from_fd);
    218   1.1   mycroft 	if (close(to_fd)) {
    219   1.1   mycroft 		warn("%s", to.p_path);
    220   1.1   mycroft 		rval = 1;
    221   1.1   mycroft 	}
    222  1.16  wsanchez 	/* set the mod/access times now after close of the fd */
    223  1.16  wsanchez 	if (pflag && set_utimes(to.p_path, fs)) {
    224  1.16  wsanchez 	    rval = 1;
    225  1.16  wsanchez 	}
    226   1.1   mycroft 	return (rval);
    227   1.1   mycroft }
    228   1.1   mycroft 
    229   1.1   mycroft int
    230  1.20       wiz copy_link(FTSENT *p, int exists)
    231   1.1   mycroft {
    232   1.1   mycroft 	int len;
    233  1.13   mycroft 	char target[MAXPATHLEN];
    234   1.1   mycroft 
    235  1.21    provos 	if ((len = readlink(p->fts_path, target, sizeof(target)-1)) == -1) {
    236   1.1   mycroft 		warn("readlink: %s", p->fts_path);
    237   1.1   mycroft 		return (1);
    238   1.1   mycroft 	}
    239  1.13   mycroft 	target[len] = '\0';
    240   1.1   mycroft 	if (exists && unlink(to.p_path)) {
    241   1.1   mycroft 		warn("unlink: %s", to.p_path);
    242   1.1   mycroft 		return (1);
    243   1.1   mycroft 	}
    244  1.13   mycroft 	if (symlink(target, to.p_path)) {
    245  1.13   mycroft 		warn("symlink: %s", target);
    246   1.1   mycroft 		return (1);
    247   1.1   mycroft 	}
    248   1.9     enami 	return (pflag ? setfile(p->fts_statp, 0) : 0);
    249   1.1   mycroft }
    250   1.1   mycroft 
    251   1.1   mycroft int
    252  1.20       wiz copy_fifo(struct stat *from_stat, int exists)
    253   1.1   mycroft {
    254   1.1   mycroft 	if (exists && unlink(to.p_path)) {
    255   1.1   mycroft 		warn("unlink: %s", to.p_path);
    256   1.1   mycroft 		return (1);
    257   1.1   mycroft 	}
    258   1.1   mycroft 	if (mkfifo(to.p_path, from_stat->st_mode)) {
    259   1.1   mycroft 		warn("mkfifo: %s", to.p_path);
    260   1.1   mycroft 		return (1);
    261   1.1   mycroft 	}
    262   1.1   mycroft 	return (pflag ? setfile(from_stat, 0) : 0);
    263   1.1   mycroft }
    264   1.1   mycroft 
    265   1.1   mycroft int
    266  1.20       wiz copy_special(struct stat *from_stat, int exists)
    267   1.1   mycroft {
    268   1.1   mycroft 	if (exists && unlink(to.p_path)) {
    269   1.1   mycroft 		warn("unlink: %s", to.p_path);
    270   1.1   mycroft 		return (1);
    271   1.1   mycroft 	}
    272   1.1   mycroft 	if (mknod(to.p_path, from_stat->st_mode, from_stat->st_rdev)) {
    273   1.1   mycroft 		warn("mknod: %s", to.p_path);
    274   1.1   mycroft 		return (1);
    275   1.1   mycroft 	}
    276   1.1   mycroft 	return (pflag ? setfile(from_stat, 0) : 0);
    277   1.1   mycroft }
    278   1.1   mycroft 
    279   1.1   mycroft 
    280  1.16  wsanchez /*
    281  1.16  wsanchez  * Function: setfile
    282  1.16  wsanchez  *
    283  1.16  wsanchez  * Purpose:
    284  1.16  wsanchez  *   Set the owner/group/permissions for the "to" file to the information
    285  1.16  wsanchez  *   in the stat structure.  If fd is zero, also call set_utimes() to set
    286  1.16  wsanchez  *   the mod/access times.  If fd is non-zero, the caller must do a utimes
    287  1.16  wsanchez  *   itself after close(fd).
    288  1.16  wsanchez  */
    289   1.1   mycroft int
    290  1.20       wiz setfile(struct stat *fs, int fd)
    291   1.1   mycroft {
    292   1.9     enami 	int rval, islink;
    293   1.1   mycroft 
    294   1.1   mycroft 	rval = 0;
    295   1.9     enami 	islink = S_ISLNK(fs->st_mode);
    296   1.1   mycroft 	fs->st_mode &= S_ISUID | S_ISGID | S_IRWXU | S_IRWXG | S_IRWXO;
    297   1.1   mycroft 
    298   1.1   mycroft 	/*
    299   1.1   mycroft 	 * Changing the ownership probably won't succeed, unless we're root
    300   1.1   mycroft 	 * or POSIX_CHOWN_RESTRICTED is not set.  Set uid/gid before setting
    301   1.1   mycroft 	 * the mode; current BSD behavior is to remove all setuid bits on
    302   1.1   mycroft 	 * chown.  If chown fails, lose setuid/setgid bits.
    303   1.1   mycroft 	 */
    304   1.1   mycroft 	if (fd ? fchown(fd, fs->st_uid, fs->st_gid) :
    305   1.9     enami 	    lchown(to.p_path, fs->st_uid, fs->st_gid)) {
    306   1.1   mycroft 		if (errno != EPERM) {
    307   1.1   mycroft 			warn("chown: %s", to.p_path);
    308   1.1   mycroft 			rval = 1;
    309   1.1   mycroft 		}
    310   1.1   mycroft 		fs->st_mode &= ~(S_ISUID | S_ISGID);
    311   1.1   mycroft 	}
    312   1.9     enami 	if (fd ? fchmod(fd, fs->st_mode) : lchmod(to.p_path, fs->st_mode)) {
    313  1.10   mycroft 		warn("chmod: %s", to.p_path);
    314   1.1   mycroft 		rval = 1;
    315   1.1   mycroft 	}
    316   1.1   mycroft 
    317  1.28      elad 	if (!islink && !Nflag) {
    318  1.29  christos 		unsigned long fflags = fs->st_flags;
    319   1.9     enami 		/*
    320   1.9     enami 		 * XXX
    321   1.9     enami 		 * NFS doesn't support chflags; ignore errors unless
    322   1.9     enami 		 * there's reason to believe we're losing bits.
    323   1.9     enami 		 * (Note, this still won't be right if the server
    324   1.9     enami 		 * supports flags and we were trying to *remove* flags
    325   1.9     enami 		 * on a file that we copied, i.e., that we didn't create.)
    326   1.9     enami 		 */
    327   1.9     enami 		errno = 0;
    328  1.29  christos 		if ((fd ? fchflags(fd, fflags) :
    329  1.29  christos 		    chflags(to.p_path, fflags)) == -1)
    330   1.9     enami 			if (errno != EOPNOTSUPP || fs->st_flags != 0) {
    331   1.9     enami 				warn("chflags: %s", to.p_path);
    332   1.9     enami 				rval = 1;
    333   1.9     enami 			}
    334   1.9     enami 	}
    335  1.16  wsanchez 	/* if fd is non-zero, caller must call set_utimes() after close() */
    336  1.16  wsanchez 	if (fd == 0 && set_utimes(to.p_path, fs))
    337  1.16  wsanchez 	    rval = 1;
    338   1.1   mycroft 	return (rval);
    339   1.1   mycroft }
    340   1.1   mycroft 
    341   1.1   mycroft void
    342  1.20       wiz usage(void)
    343   1.1   mycroft {
    344  1.20       wiz 	(void)fprintf(stderr,
    345  1.34      hira 	    "usage: %s [-R [-H | -L | -P]] [-f | -i] [-Npv] src target\n"
    346  1.34      hira 	    "       %s [-R [-H | -L | -P]] [-f | -i] [-Npv] src1 ... srcN directory\n",
    347  1.20       wiz 	    getprogname(), getprogname());
    348   1.1   mycroft 	exit(1);
    349  1.14   mycroft 	/* NOTREACHED */
    350   1.1   mycroft }
    351