1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2014 Lennart Poettering
8 systemd is free software; you can redistribute it and/or modify it
9 under the terms of the GNU Lesser General Public License as published by
10 the Free Software Foundation; either version 2.1 of the License, or
11 (at your option) any later version.
13 systemd is distributed in the hope that it will be useful, but
14 WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 Lesser General Public License for more details.
18 You should have received a copy of the GNU Lesser General Public License
19 along with systemd; If not, see <http://www.gnu.org/licenses/>.
22 #include <sys/sendfile.h>
27 int copy_bytes(int fdf, int fdt, off_t max_bytes) {
28 bool try_sendfile = true;
37 if (max_bytes != (off_t) -1) {
42 if ((off_t) m > max_bytes)
43 m = (size_t) max_bytes;
46 /* First try sendfile(), unless we already tried */
49 n = sendfile(fdt, fdf, NULL, m);
51 if (errno != EINVAL && errno != ENOSYS)
55 /* use fallback below */
56 } else if (n == 0) /* EOF */
63 /* As a fallback just copy bits by hand */
68 n = read(fdf, buf, m);
74 r = loop_write(fdt, buf, n, false);
81 if (max_bytes != (off_t) -1) {
82 assert(max_bytes >= n);
90 static int fd_copy_symlink(int df, const char *from, const struct stat *st, int dt, const char *to) {
91 _cleanup_free_ char *target = NULL;
98 r = readlinkat_malloc(df, from, &target);
102 if (symlinkat(target, dt, to) < 0)
105 if (fchownat(dt, to, st->st_uid, st->st_gid, AT_SYMLINK_NOFOLLOW) < 0)
111 static int fd_copy_regular(int df, const char *from, const struct stat *st, int dt, const char *to) {
112 _cleanup_close_ int fdf = -1, fdt = -1;
119 fdf = openat(df, from, O_RDONLY|O_CLOEXEC|O_NOCTTY|O_NOFOLLOW);
123 fdt = openat(dt, to, O_WRONLY|O_CREAT|O_EXCL|O_CLOEXEC|O_NOCTTY|O_NOFOLLOW, st->st_mode & 07777);
127 r = copy_bytes(fdf, fdt, (off_t) -1);
133 if (fchown(fdt, st->st_uid, st->st_gid) < 0)
136 if (fchmod(fdt, st->st_mode & 07777) < 0)
150 static int fd_copy_fifo(int df, const char *from, const struct stat *st, int dt, const char *to) {
157 r = mkfifoat(dt, to, st->st_mode & 07777);
161 if (fchownat(dt, to, st->st_uid, st->st_gid, AT_SYMLINK_NOFOLLOW) < 0)
164 if (fchmodat(dt, to, st->st_mode & 07777, 0) < 0)
170 static int fd_copy_node(int df, const char *from, const struct stat *st, int dt, const char *to) {
177 r = mknodat(dt, to, st->st_mode, st->st_rdev);
181 if (fchownat(dt, to, st->st_uid, st->st_gid, AT_SYMLINK_NOFOLLOW) < 0)
184 if (fchmodat(dt, to, st->st_mode & 07777, 0) < 0)
190 static int fd_copy_directory(int df, const char *from, const struct stat *st, int dt, const char *to, dev_t original_device, bool merge) {
191 _cleanup_close_ int fdf = -1, fdt = -1;
192 _cleanup_closedir_ DIR *d = NULL;
201 fdf = openat(df, from, O_RDONLY|O_DIRECTORY|O_CLOEXEC|O_NOCTTY|O_NOFOLLOW);
210 r = mkdirat(dt, to, st->st_mode & 07777);
213 else if (errno == EEXIST && merge)
218 fdt = openat(dt, to, O_RDONLY|O_DIRECTORY|O_CLOEXEC|O_NOCTTY|O_NOFOLLOW);
225 if (fchown(fdt, st->st_uid, st->st_gid) < 0)
228 if (fchmod(fdt, st->st_mode & 07777) < 0)
232 FOREACH_DIRENT(de, d, return -errno) {
236 if (fstatat(dirfd(d), de->d_name, &buf, AT_SYMLINK_NOFOLLOW) < 0) {
241 if (buf.st_dev != original_device)
244 if (S_ISREG(buf.st_mode))
245 q = fd_copy_regular(dirfd(d), de->d_name, &buf, fdt, de->d_name);
246 else if (S_ISDIR(buf.st_mode))
247 q = fd_copy_directory(dirfd(d), de->d_name, &buf, fdt, de->d_name, original_device, merge);
248 else if (S_ISLNK(buf.st_mode))
249 q = fd_copy_symlink(dirfd(d), de->d_name, &buf, fdt, de->d_name);
250 else if (S_ISFIFO(buf.st_mode))
251 q = fd_copy_fifo(dirfd(d), de->d_name, &buf, fdt, de->d_name);
252 else if (S_ISBLK(buf.st_mode) || S_ISCHR(buf.st_mode))
253 q = fd_copy_node(dirfd(d), de->d_name, &buf, fdt, de->d_name);
257 if (q == -EEXIST && merge)
267 int copy_tree(const char *from, const char *to, bool merge) {
273 if (lstat(from, &st) < 0)
276 if (S_ISREG(st.st_mode))
277 return fd_copy_regular(AT_FDCWD, from, &st, AT_FDCWD, to);
278 else if (S_ISDIR(st.st_mode))
279 return fd_copy_directory(AT_FDCWD, from, &st, AT_FDCWD, to, st.st_dev, merge);
280 else if (S_ISLNK(st.st_mode))
281 return fd_copy_symlink(AT_FDCWD, from, &st, AT_FDCWD, to);
282 else if (S_ISFIFO(st.st_mode))
283 return fd_copy_fifo(AT_FDCWD, from, &st, AT_FDCWD, to);
284 else if (S_ISBLK(st.st_mode) || S_ISCHR(st.st_mode))
285 return fd_copy_node(AT_FDCWD, from, &st, AT_FDCWD, to);
290 int copy_file_fd(const char *from, int fdt) {
291 _cleanup_close_ int fdf = -1;
296 fdf = open(from, O_RDONLY|O_CLOEXEC|O_NOCTTY);
300 return copy_bytes(fdf, fdt, (off_t) -1);
303 int copy_file(const char *from, const char *to, int flags, mode_t mode) {
309 fdt = open(to, flags|O_WRONLY|O_CREAT|O_CLOEXEC|O_NOCTTY, mode);
313 r = copy_file_fd(from, fdt);
320 if (close(fdt) < 0) {