[821] | 1 | /* vi: set sw=4 ts=4: */
|
---|
| 2 | /*
|
---|
| 3 | * Mini copy_file implementation for busybox
|
---|
| 4 | *
|
---|
| 5 | * Copyright (C) 2001 by Matt Kraai <kraai@alumni.carnegiemellon.edu>
|
---|
| 6 | *
|
---|
| 7 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
|
---|
| 8 | *
|
---|
| 9 | */
|
---|
| 10 |
|
---|
| 11 | #include "libbb.h"
|
---|
| 12 | #include <utime.h>
|
---|
| 13 | #include <errno.h>
|
---|
| 14 |
|
---|
| 15 | int copy_file(const char *source, const char *dest, int flags)
|
---|
| 16 | {
|
---|
| 17 | struct stat source_stat;
|
---|
| 18 | struct stat dest_stat;
|
---|
| 19 | int dest_exists = 0;
|
---|
| 20 | int status = 0;
|
---|
| 21 |
|
---|
| 22 | if ((!(flags & FILEUTILS_DEREFERENCE) &&
|
---|
| 23 | lstat(source, &source_stat) < 0) ||
|
---|
| 24 | ((flags & FILEUTILS_DEREFERENCE) &&
|
---|
| 25 | stat(source, &source_stat) < 0)) {
|
---|
| 26 | bb_perror_msg("%s", source);
|
---|
| 27 | return -1;
|
---|
| 28 | }
|
---|
| 29 |
|
---|
| 30 | if (lstat(dest, &dest_stat) < 0) {
|
---|
| 31 | if (errno != ENOENT) {
|
---|
| 32 | bb_perror_msg("unable to stat `%s'", dest);
|
---|
| 33 | return -1;
|
---|
| 34 | }
|
---|
| 35 | } else {
|
---|
| 36 | if (source_stat.st_dev == dest_stat.st_dev &&
|
---|
| 37 | source_stat.st_ino == dest_stat.st_ino)
|
---|
| 38 | {
|
---|
| 39 | bb_error_msg("`%s' and `%s' are the same file", source, dest);
|
---|
| 40 | return -1;
|
---|
| 41 | }
|
---|
| 42 | dest_exists = 1;
|
---|
| 43 | }
|
---|
| 44 |
|
---|
| 45 | if (S_ISDIR(source_stat.st_mode)) {
|
---|
| 46 | DIR *dp;
|
---|
| 47 | struct dirent *d;
|
---|
| 48 | mode_t saved_umask = 0;
|
---|
| 49 |
|
---|
| 50 | if (!(flags & FILEUTILS_RECUR)) {
|
---|
| 51 | bb_error_msg("%s: omitting directory", source);
|
---|
| 52 | return -1;
|
---|
| 53 | }
|
---|
| 54 |
|
---|
| 55 | /* Create DEST. */
|
---|
| 56 | if (dest_exists) {
|
---|
| 57 | if (!S_ISDIR(dest_stat.st_mode)) {
|
---|
| 58 | bb_error_msg("`%s' is not a directory", dest);
|
---|
| 59 | return -1;
|
---|
| 60 | }
|
---|
| 61 | } else {
|
---|
| 62 | mode_t mode;
|
---|
| 63 | saved_umask = umask(0);
|
---|
| 64 |
|
---|
| 65 | mode = source_stat.st_mode;
|
---|
| 66 | if (!(flags & FILEUTILS_PRESERVE_STATUS))
|
---|
| 67 | mode = source_stat.st_mode & ~saved_umask;
|
---|
| 68 | mode |= S_IRWXU;
|
---|
| 69 |
|
---|
| 70 | if (mkdir(dest, mode) < 0) {
|
---|
| 71 | umask(saved_umask);
|
---|
| 72 | bb_perror_msg("cannot create directory `%s'", dest);
|
---|
| 73 | return -1;
|
---|
| 74 | }
|
---|
| 75 |
|
---|
| 76 | umask(saved_umask);
|
---|
| 77 | }
|
---|
| 78 |
|
---|
| 79 | /* Recursively copy files in SOURCE. */
|
---|
| 80 | if ((dp = bb_opendir(source)) == NULL) {
|
---|
| 81 | status = -1;
|
---|
| 82 | goto preserve_status;
|
---|
| 83 | }
|
---|
| 84 |
|
---|
| 85 | while ((d = readdir(dp)) != NULL) {
|
---|
| 86 | char *new_source, *new_dest;
|
---|
| 87 |
|
---|
| 88 | new_source = concat_subpath_file(source, d->d_name);
|
---|
| 89 | if(new_source == NULL)
|
---|
| 90 | continue;
|
---|
| 91 | new_dest = concat_path_file(dest, d->d_name);
|
---|
| 92 | if (copy_file(new_source, new_dest, flags) < 0)
|
---|
| 93 | status = -1;
|
---|
| 94 | free(new_source);
|
---|
| 95 | free(new_dest);
|
---|
| 96 | }
|
---|
| 97 | /* closedir have only EBADF error, but "dp" not changes */
|
---|
| 98 | closedir(dp);
|
---|
| 99 |
|
---|
| 100 | if (!dest_exists &&
|
---|
| 101 | chmod(dest, source_stat.st_mode & ~saved_umask) < 0) {
|
---|
| 102 | bb_perror_msg("unable to change permissions of `%s'", dest);
|
---|
| 103 | status = -1;
|
---|
| 104 | }
|
---|
| 105 | } else if (S_ISREG(source_stat.st_mode) ||
|
---|
| 106 | (S_ISLNK(source_stat.st_mode) && (flags & FILEUTILS_DEREFERENCE)))
|
---|
| 107 | {
|
---|
| 108 | int src_fd;
|
---|
| 109 | int dst_fd;
|
---|
| 110 | if (ENABLE_FEATURE_PRESERVE_HARDLINKS) {
|
---|
| 111 | char *link_name;
|
---|
| 112 |
|
---|
| 113 | if (!(flags & FILEUTILS_DEREFERENCE) &&
|
---|
| 114 | is_in_ino_dev_hashtable(&source_stat, &link_name)) {
|
---|
| 115 | if (link(link_name, dest) < 0) {
|
---|
| 116 | bb_perror_msg("unable to link `%s'", dest);
|
---|
| 117 | return -1;
|
---|
| 118 | }
|
---|
| 119 |
|
---|
| 120 | return 0;
|
---|
| 121 | }
|
---|
| 122 | add_to_ino_dev_hashtable(&source_stat, dest);
|
---|
| 123 | }
|
---|
| 124 | src_fd = open(source, O_RDONLY);
|
---|
| 125 | if (src_fd == -1) {
|
---|
| 126 | bb_perror_msg("unable to open `%s'", source);
|
---|
| 127 | return(-1);
|
---|
| 128 | }
|
---|
| 129 |
|
---|
| 130 | if (dest_exists) {
|
---|
| 131 | if (flags & FILEUTILS_INTERACTIVE) {
|
---|
| 132 | fprintf(stderr, "%s: overwrite `%s'? ", bb_applet_name, dest);
|
---|
| 133 | if (!bb_ask_confirmation()) {
|
---|
| 134 | close (src_fd);
|
---|
| 135 | return 0;
|
---|
| 136 | }
|
---|
| 137 | }
|
---|
| 138 |
|
---|
| 139 | dst_fd = open(dest, O_WRONLY|O_TRUNC);
|
---|
| 140 | if (dst_fd == -1) {
|
---|
| 141 | if (!(flags & FILEUTILS_FORCE)) {
|
---|
| 142 | bb_perror_msg("unable to open `%s'", dest);
|
---|
| 143 | close(src_fd);
|
---|
| 144 | return -1;
|
---|
| 145 | }
|
---|
| 146 |
|
---|
| 147 | if (unlink(dest) < 0) {
|
---|
| 148 | bb_perror_msg("unable to remove `%s'", dest);
|
---|
| 149 | close(src_fd);
|
---|
| 150 | return -1;
|
---|
| 151 | }
|
---|
| 152 |
|
---|
| 153 | goto dest_removed;
|
---|
| 154 | }
|
---|
| 155 | } else {
|
---|
| 156 | dest_removed:
|
---|
| 157 | dst_fd = open(dest, O_WRONLY|O_CREAT, source_stat.st_mode);
|
---|
| 158 | if (dst_fd == -1) {
|
---|
| 159 | bb_perror_msg("unable to open `%s'", dest);
|
---|
| 160 | close(src_fd);
|
---|
| 161 | return(-1);
|
---|
| 162 | }
|
---|
| 163 | }
|
---|
| 164 |
|
---|
| 165 | if (bb_copyfd_eof(src_fd, dst_fd) == -1)
|
---|
| 166 | status = -1;
|
---|
| 167 |
|
---|
| 168 | if (close(dst_fd) < 0) {
|
---|
| 169 | bb_perror_msg("unable to close `%s'", dest);
|
---|
| 170 | status = -1;
|
---|
| 171 | }
|
---|
| 172 |
|
---|
| 173 | if (close(src_fd) < 0) {
|
---|
| 174 | bb_perror_msg("unable to close `%s'", source);
|
---|
| 175 | status = -1;
|
---|
| 176 | }
|
---|
| 177 | } else if (S_ISBLK(source_stat.st_mode) || S_ISCHR(source_stat.st_mode) ||
|
---|
| 178 | S_ISSOCK(source_stat.st_mode) || S_ISFIFO(source_stat.st_mode) ||
|
---|
| 179 | S_ISLNK(source_stat.st_mode)) {
|
---|
| 180 |
|
---|
| 181 | if (dest_exists) {
|
---|
| 182 | if((flags & FILEUTILS_FORCE) == 0) {
|
---|
| 183 | fprintf(stderr, "`%s' exists\n", dest);
|
---|
| 184 | return -1;
|
---|
| 185 | }
|
---|
| 186 | if(unlink(dest) < 0) {
|
---|
| 187 | bb_perror_msg("unable to remove `%s'", dest);
|
---|
| 188 | return -1;
|
---|
| 189 | }
|
---|
| 190 | }
|
---|
| 191 | if (S_ISFIFO(source_stat.st_mode)) {
|
---|
| 192 | if (mkfifo(dest, source_stat.st_mode) < 0) {
|
---|
| 193 | bb_perror_msg("cannot create fifo `%s'", dest);
|
---|
| 194 | return -1;
|
---|
| 195 | }
|
---|
| 196 | } else if (S_ISLNK(source_stat.st_mode)) {
|
---|
| 197 | char *lpath;
|
---|
| 198 |
|
---|
| 199 | lpath = xreadlink(source);
|
---|
| 200 | if (symlink(lpath, dest) < 0) {
|
---|
| 201 | bb_perror_msg("cannot create symlink `%s'", dest);
|
---|
| 202 | return -1;
|
---|
| 203 | }
|
---|
| 204 | free(lpath);
|
---|
| 205 |
|
---|
| 206 | if (flags & FILEUTILS_PRESERVE_STATUS)
|
---|
| 207 | if (lchown(dest, source_stat.st_uid, source_stat.st_gid) < 0)
|
---|
| 208 | bb_perror_msg("unable to preserve ownership of `%s'", dest);
|
---|
| 209 |
|
---|
| 210 | return 0;
|
---|
| 211 |
|
---|
| 212 | } else {
|
---|
| 213 | if (mknod(dest, source_stat.st_mode, source_stat.st_rdev) < 0) {
|
---|
| 214 | bb_perror_msg("unable to create `%s'", dest);
|
---|
| 215 | return -1;
|
---|
| 216 | }
|
---|
| 217 | }
|
---|
| 218 | } else {
|
---|
| 219 | bb_error_msg("internal error: unrecognized file type");
|
---|
| 220 | return -1;
|
---|
| 221 | }
|
---|
| 222 |
|
---|
| 223 | preserve_status:
|
---|
| 224 |
|
---|
| 225 | if (flags & FILEUTILS_PRESERVE_STATUS) {
|
---|
| 226 | struct utimbuf times;
|
---|
| 227 | char *msg="unable to preserve %s of `%s'";
|
---|
| 228 |
|
---|
| 229 | times.actime = source_stat.st_atime;
|
---|
| 230 | times.modtime = source_stat.st_mtime;
|
---|
| 231 | if (utime(dest, ×) < 0)
|
---|
| 232 | bb_perror_msg(msg, "times", dest);
|
---|
| 233 | if (chown(dest, source_stat.st_uid, source_stat.st_gid) < 0) {
|
---|
| 234 | source_stat.st_mode &= ~(S_ISUID | S_ISGID);
|
---|
| 235 | bb_perror_msg(msg, "ownership", dest);
|
---|
| 236 | }
|
---|
| 237 | if (chmod(dest, source_stat.st_mode) < 0)
|
---|
| 238 | bb_perror_msg(msg, "permissions", dest);
|
---|
| 239 | }
|
---|
| 240 |
|
---|
| 241 | return status;
|
---|
| 242 | }
|
---|