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
61 #include <sys/mount.h>
62 #include <get_compat.h>
64 #include <sys/clonefile.h>
66 #define COMPAT_MODE(a,b) (1)
67 #endif /* __APPLE__ */
70 #define cp_pct(x,y) (int)(100.0 * (double)(x) / (double)(y))
72 /* Memory strategy threshold, in pages: if physmem is larger then this, use a
74 #define PHYSPAGES_THRESHOLD (32*1024)
76 /* Maximum buffer size in bytes - do not allow it to grow larger than this */
77 #define BUFSIZE_MAX (2*1024*1024)
79 /* Small (default) buffer size in bytes. It's inefficient for this to be
80 * smaller than MAXPHYS */
81 #define BUFSIZE_SMALL (MAXPHYS)
84 copy_file(const FTSENT
*entp
, int dne
)
86 static char *buf
= NULL
;
87 static size_t bufsize
;
89 int ch
, checkch
, from_fd
, rval
, to_fd
;
95 char resp
[] = {'\0', '\0'};
96 #ifdef VM_AND_BUFFER_CACHE_SYNCHRONIZED
102 if ((from_fd
= open(entp
->fts_path
, O_RDONLY
, 0)) == -1) {
103 warn("%s", entp
->fts_path
);
107 fs
= entp
->fts_statp
;
110 * If the file exists and we're interactive, verify with the user.
111 * If the file DNE, set the mode to be the from file, minus setuid
112 * bits, modified by the umask; arguably wrong, but it makes copying
113 * executables work right and it's been that way forever. (The
114 * other choice is 666 or'ed with the execute bits on the from file
115 * modified by the umask.)
118 #define YESNO "(y/n [n]) "
121 printf("%s not overwritten\n", to
.p_path
);
122 (void)close(from_fd
);
125 (void)fprintf(stderr
, "overwrite %s? %s",
128 /* Load user specified locale */
129 setlocale(LC_MESSAGES
, "");
131 checkch
= ch
= getchar();
132 while (ch
!= '\n' && ch
!= EOF
)
135 /* only care about the first character */
138 if (rpmatch(resp
) != 1) {
139 (void)close(from_fd
);
140 (void)fprintf(stderr
, "not overwritten\n");
146 (void)unlink(to
.p_path
);
147 int error
= clonefile(entp
->fts_path
, to
.p_path
, 0);
149 warn("%s: clonefile failed", to
.p_path
);
150 (void)close(from_fd
);
151 return error
== 0 ?
0 : 1;
154 if (COMPAT_MODE("bin/cp", "unix2003")) {
155 /* first try to overwrite existing destination file name */
156 to_fd
= open(to
.p_path
, O_WRONLY
| O_TRUNC
, 0);
159 /* Only if it fails remove file and create a new one */
160 (void)unlink(to
.p_path
);
161 to_fd
= open(to
.p_path
, O_WRONLY
| O_TRUNC
| O_CREAT
,
162 fs
->st_mode
& ~(S_ISUID
| S_ISGID
));
167 /* remove existing destination file name,
168 * create a new file */
169 (void)unlink(to
.p_path
);
170 to_fd
= open(to
.p_path
, O_WRONLY
| O_TRUNC
| O_CREAT
,
171 fs
->st_mode
& ~(S_ISUID
| S_ISGID
));
173 /* overwrite existing destination file name */
174 to_fd
= open(to
.p_path
, O_WRONLY
| O_TRUNC
, 0);
179 int error
= clonefile(entp
->fts_path
, to
.p_path
, 0);
181 warn("%s: clonefile failed", to
.p_path
);
182 (void)close(from_fd
);
183 return error
== 0 ?
0 : 1;
186 to_fd
= open(to
.p_path
, O_WRONLY
| O_TRUNC
| O_CREAT
,
187 fs
->st_mode
& ~(S_ISUID
| S_ISGID
));
191 warn("%s", to
.p_path
);
192 (void)close(from_fd
);
199 if (S_ISREG(fs
->st_mode
)) {
203 * Pre-allocate blocks for the destination file if it
206 if (fstatfs(to_fd
, &sfs
) == 0 &&
207 strcmp(sfs
.f_fstypename
, "acfs") == 0) {
211 fst
.fst_posmode
= F_PEOFPOSMODE
;
213 fst
.fst_length
= fs
->st_size
;
215 (void) fcntl(to_fd
, F_PREALLOCATE
, &fst
);
218 #endif /* __APPLE__ */
220 if (fstat(to_fd
, &to_stat
) != -1) {
221 mode
= to_stat
.st_mode
;
222 if ((mode
& (S_IRWXG
|S_IRWXO
))
223 && fchmod(to_fd
, mode
& ~(S_IRWXG
|S_IRWXO
))) {
224 if (errno
!= EPERM
) /* we have write access but do not own the file */
225 warn("%s: fchmod failed", to
.p_path
);
229 warn("%s", to
.p_path
);
232 * Mmap and write if less than 8M (the limit is so we don't totally
233 * trash memory on big files. This is really a minor hack, but it
234 * wins some CPU back.
236 #ifdef VM_AND_BUFFER_CACHE_SYNCHRONIZED
237 if (S_ISREG(fs
->st_mode
) && fs
->st_size
> 0 &&
238 fs
->st_size
<= 8 * 1048576) {
239 if ((p
= mmap(NULL
, (size_t)fs
->st_size
, PROT_READ
,
240 MAP_SHARED
, from_fd
, (off_t
)0)) == MAP_FAILED
) {
241 warn("%s", entp
->fts_path
);
245 for (bufp
= p
, wresid
= fs
->st_size
; ;
246 bufp
+= wcount
, wresid
-= (size_t)wcount
) {
247 wcount
= write(to_fd
, bufp
, wresid
);
251 (void)fprintf(stderr
,
253 entp
->fts_path
, to
.p_path
,
254 cp_pct(wtotal
, fs
->st_size
));
257 if (wcount
>= (ssize_t
)wresid
|| wcount
<= 0)
260 if (wcount
!= (ssize_t
)wresid
) {
261 warn("%s", to
.p_path
);
264 /* Some systems don't unmap on close(2). */
265 if (munmap(p
, fs
->st_size
) < 0) {
266 warn("%s", entp
->fts_path
);
275 * Note that buf and bufsize are static. If
276 * malloc() fails, it will fail at the start
277 * and not copy only some files.
279 if (sysconf(_SC_PHYS_PAGES
) >
281 bufsize
= MIN(BUFSIZE_MAX
, MAXPHYS
* 8);
283 bufsize
= BUFSIZE_SMALL
;
284 buf
= malloc(bufsize
);
286 err(1, "Not enough memory");
289 while ((rcount
= read(from_fd
, buf
, bufsize
)) > 0) {
290 for (bufp
= buf
, wresid
= rcount
; ;
291 bufp
+= wcount
, wresid
-= wcount
) {
292 wcount
= write(to_fd
, bufp
, wresid
);
296 (void)fprintf(stderr
,
298 entp
->fts_path
, to
.p_path
,
299 cp_pct(wtotal
, fs
->st_size
));
302 if (wcount
>= (ssize_t
)wresid
|| wcount
<= 0)
305 if (wcount
!= (ssize_t
)wresid
) {
306 warn("%s", to
.p_path
);
312 warn("%s", entp
->fts_path
);
318 * Don't remove the target even after an error. The target might
319 * not be a regular file, or its attributes might be important,
320 * or its contents might be irreplaceable. It would only be safe
321 * to remove it if we created it and its length is 0.
324 if (fchmod(to_fd
, mode
))
325 warn("%s: fchmod failed", to
.p_path
);
327 /* do these before setfile in case copyfile changes mtime */
328 if (!Xflag
&& S_ISREG(fs
->st_mode
)) { /* skip devices, etc */
329 if (fcopyfile(from_fd
, to_fd
, NULL
, COPYFILE_XATTR
) < 0)
330 warn("%s: could not copy extended attributes to %s", entp
->fts_path
, to
.p_path
);
332 if (pflag
&& setfile(fs
, to_fd
))
335 /* If this ACL denies writeattr then setfile will fail... */
336 if (fcopyfile(from_fd
, to_fd
, NULL
, COPYFILE_ACL
) < 0)
337 warn("%s: could not copy ACL to %s", entp
->fts_path
, to
.p_path
);
339 #else /* !__APPLE__ */
340 if (pflag
&& setfile(fs
, to_fd
))
342 if (pflag
&& preserve_fd_acls(from_fd
, to_fd
) != 0)
344 #endif /* __APPLE__ */
345 (void)close(from_fd
);
347 warn("%s", to
.p_path
);
354 copy_link(const FTSENT
*p
, int exists
)
357 char llink
[PATH_MAX
];
359 if ((len
= readlink(p
->fts_path
, llink
, sizeof(llink
) - 1)) == -1) {
360 warn("readlink: %s", p
->fts_path
);
364 if (exists
&& unlink(to
.p_path
)) {
365 warn("unlink: %s", to
.p_path
);
368 if (symlink(llink
, to
.p_path
)) {
369 warn("symlink: %s", llink
);
374 if (copyfile(p
->fts_path
, to
.p_path
, NULL
, COPYFILE_XATTR
| COPYFILE_NOFOLLOW_SRC
) <0)
375 warn("%s: could not copy extended attributes to %s",
376 p
->fts_path
, to
.p_path
);
378 return (pflag ?
setfile(p
->fts_statp
, -1) : 0);
382 copy_fifo(struct stat
*from_stat
, int exists
)
384 if (exists
&& unlink(to
.p_path
)) {
385 warn("unlink: %s", to
.p_path
);
388 if (mkfifo(to
.p_path
, from_stat
->st_mode
)) {
389 warn("mkfifo: %s", to
.p_path
);
392 return (pflag ?
setfile(from_stat
, -1) : 0);
396 copy_special(struct stat
*from_stat
, int exists
)
398 if (exists
&& unlink(to
.p_path
)) {
399 warn("unlink: %s", to
.p_path
);
402 if (mknod(to
.p_path
, from_stat
->st_mode
, from_stat
->st_rdev
)) {
403 warn("mknod: %s", to
.p_path
);
406 return (pflag ?
setfile(from_stat
, -1) : 0);
410 setfile(struct stat
*fs
, int fd
)
412 struct attrlist ts_req
= {};
414 int rval
, gotstat
, islink
, fdval
;
416 struct timespec mtime
;
417 struct timespec atime
;
422 islink
= !fdval
&& S_ISLNK(fs
->st_mode
);
423 fs
->st_mode
&= S_ISUID
| S_ISGID
| S_ISVTX
| S_IRWXU
| S_IRWXG
| S_IRWXO
;
424 unsigned int options
= islink ? FSOPT_NOFOLLOW
: 0;
426 ts_req
.bitmapcount
= ATTR_BIT_MAP_COUNT
;
427 ts_req
.commonattr
= ATTR_CMN_MODTIME
| ATTR_CMN_ACCTIME
;
428 set_ts
.mtime
= fs
->st_mtimespec
;
429 set_ts
.atime
= fs
->st_atimespec
;
431 if (fdval ?
fsetattrlist(fd
, &ts_req
, &set_ts
, sizeof(set_ts
), options
) :
432 setattrlist(to
.p_path
, &ts_req
, &set_ts
, sizeof(set_ts
), options
)) {
433 warn("%ssetattrlist: %s", fdval ?
"f" : "", to
.p_path
);
436 if (fdval ?
fstat(fd
, &ts
) : (islink ?
lstat(to
.p_path
, &ts
) :
437 stat(to
.p_path
, &ts
))) {
441 ts
.st_mode
&= S_ISUID
| S_ISGID
| S_ISVTX
| S_IRWXU
| S_IRWXG
| S_IRWXO
;
444 * Changing the ownership probably won't succeed, unless we're root
445 * or POSIX_CHOWN_RESTRICTED is not set. Set uid/gid before setting
446 * the mode; current BSD behavior is to remove all setuid bits on
447 * chown. If chown fails, lose setuid/setgid bits.
449 if (!gotstat
|| fs
->st_uid
!= ts
.st_uid
|| fs
->st_gid
!= ts
.st_gid
) {
450 if (fdval ?
fchown(fd
, fs
->st_uid
, fs
->st_gid
) : (islink ?
451 lchown(to
.p_path
, fs
->st_uid
, fs
->st_gid
) :
452 chown(to
.p_path
, fs
->st_uid
, fs
->st_gid
))) {
453 if (errno
!= EPERM
) {
454 warn("%schown: %s", fdval ?
"f" : (islink ?
"l" : ""), to
.p_path
);
457 fs
->st_mode
&= ~(S_ISUID
| S_ISGID
);
461 if (!gotstat
|| fs
->st_mode
!= ts
.st_mode
) {
462 if (fdval ?
fchmod(fd
, fs
->st_mode
) : (islink ?
463 lchmod(to
.p_path
, fs
->st_mode
) :
464 chmod(to
.p_path
, fs
->st_mode
))) {
465 warn("%schmod: %s", fdval ?
"f" : (islink ?
"l" : ""), to
.p_path
);
470 if (!gotstat
|| fs
->st_flags
!= ts
.st_flags
) {
471 if (fdval ?
fchflags(fd
, fs
->st_flags
) : (islink ?
472 lchflags(to
.p_path
, fs
->st_flags
) :
473 chflags(to
.p_path
, fs
->st_flags
))) {
474 if (errno
!= EPERM
) {
475 warn("%schflags: %s", fdval ?
"f" : (islink ?
"l" : ""), to
.p_path
);
485 preserve_fd_acls(int source_fd
, int dest_fd
)
490 if (fpathconf(source_fd
, _PC_ACL_EXTENDED
) != 1 ||
491 fpathconf(dest_fd
, _PC_ACL_EXTENDED
) != 1)
493 acl
= acl_get_fd(source_fd
);
495 warn("failed to get acl entries while setting %s", to
.p_path
);
498 aclp
= &acl
->ats_acl
;
499 if (aclp
->acl_cnt
== 3)
501 if (acl_set_fd(dest_fd
, acl
) < 0) {
502 warn("failed to set acl entries for %s", to
.p_path
);
509 preserve_dir_acls(struct stat
*fs
, char *source_dir
, char *dest_dir
)
511 acl_t (*aclgetf
)(const char *, acl_type_t
);
512 int (*aclsetf
)(const char *, acl_type_t
, acl_t
);
516 if (pathconf(source_dir
, _PC_ACL_EXTENDED
) != 1 ||
517 pathconf(dest_dir
, _PC_ACL_EXTENDED
) != 1)
520 * If the file is a link we will not follow it
522 if (S_ISLNK(fs
->st_mode
)) {
523 aclgetf
= acl_get_link_np
;
524 aclsetf
= acl_set_link_np
;
526 aclgetf
= acl_get_file
;
527 aclsetf
= acl_set_file
;
530 * Even if there is no ACL_TYPE_DEFAULT entry here, a zero
531 * size ACL will be returned. So it is not safe to simply
532 * check the pointer to see if the default ACL is present.
534 acl
= aclgetf(source_dir
, ACL_TYPE_DEFAULT
);
536 warn("failed to get default acl entries on %s",
540 aclp
= &acl
->ats_acl
;
541 if (aclp
->acl_cnt
!= 0 && aclsetf(dest_dir
,
542 ACL_TYPE_DEFAULT
, acl
) < 0) {
543 warn("failed to set default acl entries on %s",
547 acl
= aclgetf(source_dir
, ACL_TYPE_ACCESS
);
549 warn("failed to get acl entries on %s", source_dir
);
552 aclp
= &acl
->ats_acl
;
553 if (aclsetf(dest_dir
, ACL_TYPE_ACCESS
, acl
) < 0) {
554 warn("failed to set acl entries on %s", dest_dir
);
559 #endif /* !__APPLE__ */
565 if (COMPAT_MODE("bin/cp", "unix2003")) {
566 (void)fprintf(stderr
, "%s\n%s\n",
567 "usage: cp [-R [-H | -L | -P]] [-fi | -n] [-apvXc] source_file target_file",
568 " cp [-R [-H | -L | -P]] [-fi | -n] [-apvXc] source_file ... "
571 (void)fprintf(stderr
, "%s\n%s\n",
572 "usage: cp [-R [-H | -L | -P]] [-f | -i | -n] [-apvXc] source_file target_file",
573 " cp [-R [-H | -L | -P]] [-f | -i | -n] [-apvXc] source_file ... "