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