2 * Copyright (c) 1991, 1993, 1994
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
32 static char sccsid
[] = "@(#)utils.c 8.3 (Berkeley) 4/1/94";
35 #include <sys/cdefs.h>
36 __FBSDID("$FreeBSD: src/bin/cp/utils.c,v 1.46 2005/09/05 04:36:08 csjp Exp $");
38 #include <sys/types.h>
40 #include <sys/param.h>
42 #ifdef VM_AND_BUFFER_CACHE_SYNCHRONIZED
60 #include <sys/mount.h>
61 #include <get_compat.h>
63 #define COMPAT_MODE(a,b) (1)
64 #endif /* __APPLE__ */
67 #define cp_pct(x,y) (int)(100.0 * (double)(x) / (double)(y))
70 copy_file(const FTSENT
*entp
, int dne
)
72 static char buf
[MAXBSIZE
];
74 int ch
, checkch
, from_fd
, rcount
, rval
, to_fd
;
79 #ifdef VM_AND_BUFFER_CACHE_SYNCHRONIZED
83 if ((from_fd
= open(entp
->fts_path
, O_RDONLY
, 0)) == -1) {
84 warn("%s", entp
->fts_path
);
91 * If the file exists and we're interactive, verify with the user.
92 * If the file DNE, set the mode to be the from file, minus setuid
93 * bits, modified by the umask; arguably wrong, but it makes copying
94 * executables work right and it's been that way forever. (The
95 * other choice is 666 or'ed with the execute bits on the from file
96 * modified by the umask.)
99 #define YESNO "(y/n [n]) "
102 printf("%s not overwritten\n", to
.p_path
);
103 (void)close(from_fd
);
106 (void)fprintf(stderr
, "overwrite %s? %s",
108 checkch
= ch
= getchar();
109 while (ch
!= '\n' && ch
!= EOF
)
111 if (checkch
!= 'y' && checkch
!= 'Y') {
112 (void)close(from_fd
);
113 (void)fprintf(stderr
, "not overwritten\n");
118 if (COMPAT_MODE("bin/cp", "unix2003")) {
119 /* first try to overwrite existing destination file name */
120 to_fd
= open(to
.p_path
, O_WRONLY
| O_TRUNC
, 0);
123 /* Only if it fails remove file and create a new one */
124 (void)unlink(to
.p_path
);
125 to_fd
= open(to
.p_path
, O_WRONLY
| O_TRUNC
| O_CREAT
,
126 fs
->st_mode
& ~(S_ISUID
| S_ISGID
));
131 /* remove existing destination file name,
132 * create a new file */
133 (void)unlink(to
.p_path
);
134 to_fd
= open(to
.p_path
, O_WRONLY
| O_TRUNC
| O_CREAT
,
135 fs
->st_mode
& ~(S_ISUID
| S_ISGID
));
137 /* overwrite existing destination file name */
138 to_fd
= open(to
.p_path
, O_WRONLY
| O_TRUNC
, 0);
141 to_fd
= open(to
.p_path
, O_WRONLY
| O_TRUNC
| O_CREAT
,
142 fs
->st_mode
& ~(S_ISUID
| S_ISGID
));
145 warn("%s", to
.p_path
);
146 (void)close(from_fd
);
153 if (S_ISREG(fs
->st_mode
)) {
157 * Pre-allocate blocks for the destination file if it
160 if (fstatfs(to_fd
, &sfs
) == 0 &&
161 strcmp(sfs
.f_fstypename
, "acfs") == 0) {
165 fst
.fst_posmode
= F_PEOFPOSMODE
;
167 fst
.fst_length
= fs
->st_size
;
169 (void) fcntl(to_fd
, F_PREALLOCATE
, &fst
);
172 #endif /* __APPLE__ */
175 * Mmap and write if less than 8M (the limit is so we don't totally
176 * trash memory on big files. This is really a minor hack, but it
177 * wins some CPU back.
179 #ifdef VM_AND_BUFFER_CACHE_SYNCHRONIZED
180 if (S_ISREG(fs
->st_mode
) && fs
->st_size
> 0 &&
181 fs
->st_size
<= 8 * 1048576) {
182 if ((p
= mmap(NULL
, (size_t)fs
->st_size
, PROT_READ
,
183 MAP_SHARED
, from_fd
, (off_t
)0)) == MAP_FAILED
) {
184 warn("%s", entp
->fts_path
);
188 for (bufp
= p
, wresid
= fs
->st_size
; ;
189 bufp
+= wcount
, wresid
-= (size_t)wcount
) {
190 wcount
= write(to_fd
, bufp
, wresid
);
194 (void)fprintf(stderr
,
196 entp
->fts_path
, to
.p_path
,
197 cp_pct(wtotal
, fs
->st_size
));
200 if (wcount
>= (ssize_t
)wresid
|| wcount
<= 0)
203 if (wcount
!= (ssize_t
)wresid
) {
204 warn("%s", to
.p_path
);
207 /* Some systems don't unmap on close(2). */
208 if (munmap(p
, fs
->st_size
) < 0) {
209 warn("%s", entp
->fts_path
);
217 while ((rcount
= read(from_fd
, buf
, MAXBSIZE
)) > 0) {
218 for (bufp
= buf
, wresid
= rcount
; ;
219 bufp
+= wcount
, wresid
-= wcount
) {
220 wcount
= write(to_fd
, bufp
, wresid
);
224 (void)fprintf(stderr
,
226 entp
->fts_path
, to
.p_path
,
227 cp_pct(wtotal
, fs
->st_size
));
230 if (wcount
>= (ssize_t
)wresid
|| wcount
<= 0)
233 if (wcount
!= (ssize_t
)wresid
) {
234 warn("%s", to
.p_path
);
240 warn("%s", entp
->fts_path
);
246 * Don't remove the target even after an error. The target might
247 * not be a regular file, or its attributes might be important,
248 * or its contents might be irreplaceable. It would only be safe
249 * to remove it if we created it and its length is 0.
253 /* do these before setfile in case copyfile changes mtime */
254 if (!Xflag
&& S_ISREG(fs
->st_mode
)) { /* skip devices, etc */
255 if (fcopyfile(from_fd
, to_fd
, NULL
, COPYFILE_XATTR
) < 0)
256 warn("%s: could not copy extended attributes to %s", entp
->fts_path
, to
.p_path
);
258 if (pflag
&& setfile(fs
, to_fd
))
261 /* If this ACL denies writeattr then setfile will fail... */
262 if (fcopyfile(from_fd
, to_fd
, NULL
, COPYFILE_ACL
) < 0)
263 warn("%s: could not copy ACL to %s", entp
->fts_path
, to
.p_path
);
265 #else /* !__APPLE__ */
266 if (pflag
&& setfile(fs
, to_fd
))
268 if (pflag
&& preserve_fd_acls(from_fd
, to_fd
) != 0)
270 #endif /* __APPLE__ */
271 (void)close(from_fd
);
273 warn("%s", to
.p_path
);
280 copy_link(const FTSENT
*p
, int exists
)
283 char llink
[PATH_MAX
];
285 if ((len
= readlink(p
->fts_path
, llink
, sizeof(llink
) - 1)) == -1) {
286 warn("readlink: %s", p
->fts_path
);
290 if (exists
&& unlink(to
.p_path
)) {
291 warn("unlink: %s", to
.p_path
);
294 if (symlink(llink
, to
.p_path
)) {
295 warn("symlink: %s", llink
);
300 if (copyfile(p
->fts_path
, to
.p_path
, NULL
, COPYFILE_XATTR
| COPYFILE_NOFOLLOW_SRC
) <0)
301 warn("%s: could not copy extended attributes to %s",
302 p
->fts_path
, to
.p_path
);
304 return (pflag ?
setfile(p
->fts_statp
, -1) : 0);
308 copy_fifo(struct stat
*from_stat
, int exists
)
310 if (exists
&& unlink(to
.p_path
)) {
311 warn("unlink: %s", to
.p_path
);
314 if (mkfifo(to
.p_path
, from_stat
->st_mode
)) {
315 warn("mkfifo: %s", to
.p_path
);
318 return (pflag ?
setfile(from_stat
, -1) : 0);
322 copy_special(struct stat
*from_stat
, int exists
)
324 if (exists
&& unlink(to
.p_path
)) {
325 warn("unlink: %s", to
.p_path
);
328 if (mknod(to
.p_path
, from_stat
->st_mode
, from_stat
->st_rdev
)) {
329 warn("mknod: %s", to
.p_path
);
332 return (pflag ?
setfile(from_stat
, -1) : 0);
336 setfile(struct stat
*fs
, int fd
)
338 static struct timeval tv
[2];
340 int rval
, gotstat
, islink
, fdval
;
344 islink
= !fdval
&& S_ISLNK(fs
->st_mode
);
345 fs
->st_mode
&= S_ISUID
| S_ISGID
| S_ISVTX
|
346 S_IRWXU
| S_IRWXG
| S_IRWXO
;
348 TIMESPEC_TO_TIMEVAL(&tv
[0], &fs
->st_atimespec
);
349 TIMESPEC_TO_TIMEVAL(&tv
[1], &fs
->st_mtimespec
);
351 if (islink ?
0 : utimes(to
.p_path
, tv
)) {
353 if (islink ?
lutimes(to
.p_path
, tv
) : utimes(to
.p_path
, tv
)) {
354 #endif /* __APPLE__ */
355 warn("%sutimes: %s", islink ?
"l" : "", to
.p_path
);
358 if (fdval ?
fstat(fd
, &ts
) :
359 (islink ?
lstat(to
.p_path
, &ts
) : stat(to
.p_path
, &ts
)))
363 ts
.st_mode
&= S_ISUID
| S_ISGID
| S_ISVTX
|
364 S_IRWXU
| S_IRWXG
| S_IRWXO
;
367 * Changing the ownership probably won't succeed, unless we're root
368 * or POSIX_CHOWN_RESTRICTED is not set. Set uid/gid before setting
369 * the mode; current BSD behavior is to remove all setuid bits on
370 * chown. If chown fails, lose setuid/setgid bits.
372 if (!gotstat
|| fs
->st_uid
!= ts
.st_uid
|| fs
->st_gid
!= ts
.st_gid
)
373 if (fdval ?
fchown(fd
, fs
->st_uid
, fs
->st_gid
) :
374 (islink ?
lchown(to
.p_path
, fs
->st_uid
, fs
->st_gid
) :
375 chown(to
.p_path
, fs
->st_uid
, fs
->st_gid
))) {
376 if (errno
!= EPERM
) {
377 warn("%schown: %s", islink ?
"l" : "", to
.p_path
);
380 fs
->st_mode
&= ~(S_ISUID
| S_ISGID
);
383 if (!gotstat
|| fs
->st_mode
!= ts
.st_mode
)
384 if (fdval ?
fchmod(fd
, fs
->st_mode
) :
385 (islink ?
lchmod(to
.p_path
, fs
->st_mode
) :
386 chmod(to
.p_path
, fs
->st_mode
))) {
387 warn("%schmod: %s", islink ?
"l" : "", to
.p_path
);
391 if (!gotstat
|| fs
->st_flags
!= ts
.st_flags
)
393 fchflags(fd
, fs
->st_flags
) :
394 (islink ?
lchflags(to
.p_path
, fs
->st_flags
) :
395 chflags(to
.p_path
, fs
->st_flags
))) {
396 warn("%schflags: %s", islink ?
"l" : "", to
.p_path
);
405 preserve_fd_acls(int source_fd
, int dest_fd
)
410 if (fpathconf(source_fd
, _PC_ACL_EXTENDED
) != 1 ||
411 fpathconf(dest_fd
, _PC_ACL_EXTENDED
) != 1)
413 acl
= acl_get_fd(source_fd
);
415 warn("failed to get acl entries while setting %s", to
.p_path
);
418 aclp
= &acl
->ats_acl
;
419 if (aclp
->acl_cnt
== 3)
421 if (acl_set_fd(dest_fd
, acl
) < 0) {
422 warn("failed to set acl entries for %s", to
.p_path
);
429 preserve_dir_acls(struct stat
*fs
, char *source_dir
, char *dest_dir
)
431 acl_t (*aclgetf
)(const char *, acl_type_t
);
432 int (*aclsetf
)(const char *, acl_type_t
, acl_t
);
436 if (pathconf(source_dir
, _PC_ACL_EXTENDED
) != 1 ||
437 pathconf(dest_dir
, _PC_ACL_EXTENDED
) != 1)
440 * If the file is a link we will not follow it
442 if (S_ISLNK(fs
->st_mode
)) {
443 aclgetf
= acl_get_link_np
;
444 aclsetf
= acl_set_link_np
;
446 aclgetf
= acl_get_file
;
447 aclsetf
= acl_set_file
;
450 * Even if there is no ACL_TYPE_DEFAULT entry here, a zero
451 * size ACL will be returned. So it is not safe to simply
452 * check the pointer to see if the default ACL is present.
454 acl
= aclgetf(source_dir
, ACL_TYPE_DEFAULT
);
456 warn("failed to get default acl entries on %s",
460 aclp
= &acl
->ats_acl
;
461 if (aclp
->acl_cnt
!= 0 && aclsetf(dest_dir
,
462 ACL_TYPE_DEFAULT
, acl
) < 0) {
463 warn("failed to set default acl entries on %s",
467 acl
= aclgetf(source_dir
, ACL_TYPE_ACCESS
);
469 warn("failed to get acl entries on %s", source_dir
);
472 aclp
= &acl
->ats_acl
;
473 if (aclsetf(dest_dir
, ACL_TYPE_ACCESS
, acl
) < 0) {
474 warn("failed to set acl entries on %s", dest_dir
);
479 #endif /* !__APPLE__ */
485 if (COMPAT_MODE("bin/cp", "unix2003")) {
486 (void)fprintf(stderr
, "%s\n%s\n",
487 "usage: cp [-R [-H | -L | -P]] [-fi | -n] [-pvX] source_file target_file",
488 " cp [-R [-H | -L | -P]] [-fi | -n] [-pvX] source_file ... "
491 (void)fprintf(stderr
, "%s\n%s\n",
492 "usage: cp [-R [-H | -L | -P]] [-f | -i | -n] [-pvX] source_file target_file",
493 " cp [-R [-H | -L | -P]] [-f | -i | -n] [-pvX] source_file ... "