busybox/coreutils/cp.c
<<
>>
Prefs
   1/* vi: set sw=4 ts=4: */
   2/*
   3 * Mini cp implementation for busybox
   4 *
   5 * Copyright (C) 2000 by Matt Kraai <kraai@alumni.carnegiemellon.edu>
   6 * SELinux support by Yuichi Nakamura <ynakam@hitachisoft.jp>
   7 *
   8 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
   9 */
  10/* Mar 16, 2003      Manuel Novoa III   (mjn3@codepoet.org)
  11 *
  12 * Size reduction.
  13 */
  14//config:config CP
  15//config:       bool "cp (10 kb)"
  16//config:       default y
  17//config:       help
  18//config:       cp is used to copy files and directories.
  19//config:
  20//config:config FEATURE_CP_LONG_OPTIONS
  21//config:       bool "Enable long options"
  22//config:       default y
  23//config:       depends on CP && LONG_OPTS
  24//config:       help
  25//config:       Enable long options.
  26//config:       Also add support for --parents option.
  27//config:
  28//config:config FEATURE_CP_REFLINK
  29//config:       bool "Enable --reflink[=auto]"
  30//config:       default y
  31//config:       depends on FEATURE_CP_LONG_OPTIONS
  32
  33//applet:IF_CP(APPLET_NOEXEC(cp, cp, BB_DIR_BIN, BB_SUID_DROP, cp))
  34/* NOEXEC despite cases when it can be a "runner" (cp -r LARGE_DIR NEW_DIR) */
  35
  36//kbuild:lib-$(CONFIG_CP) += cp.o
  37
  38/* http://www.opengroup.org/onlinepubs/007904975/utilities/cp.html */
  39
  40// Options of cp from GNU coreutils 6.10:
  41// -a, --archive
  42// -f, --force
  43// -i, --interactive
  44// -l, --link
  45// -L, --dereference
  46// -P, --no-dereference
  47// -R, -r, --recursive
  48// -s, --symbolic-link
  49// -v, --verbose
  50// -H   follow command-line symbolic links in SOURCE
  51// -d   same as --no-dereference --preserve=links
  52// -p   same as --preserve=mode,ownership,timestamps
  53// -c   same as --preserve=context
  54// -u, --update
  55//      copy only when the SOURCE file is newer than the destination
  56//      file or when the destination file is missing
  57// --remove-destination
  58//      remove each existing destination file before attempting to open
  59// --parents
  60//      use full source file name under DIRECTORY
  61// -T, --no-target-directory
  62//      treat DEST as a normal file
  63// NOT SUPPORTED IN BBOX:
  64// --backup[=CONTROL]
  65//      make a backup of each existing destination file
  66// -b   like --backup but does not accept an argument
  67// --copy-contents
  68//      copy contents of special files when recursive
  69// --preserve[=ATTR_LIST]
  70//      preserve attributes (default: mode,ownership,timestamps),
  71//      if possible additional attributes: security context,links,all
  72// --no-preserve=ATTR_LIST
  73// --sparse=WHEN
  74//      control creation of sparse files
  75// --strip-trailing-slashes
  76//      remove any trailing slashes from each SOURCE argument
  77// -S, --suffix=SUFFIX
  78//      override the usual backup suffix
  79// -t, --target-directory=DIRECTORY
  80//      copy all SOURCE arguments into DIRECTORY
  81// -x, --one-file-system
  82//      stay on this file system
  83// -Z, --context=CONTEXT
  84//      (SELinux) set SELinux security context of copy to CONTEXT
  85
  86//usage:#define cp_trivial_usage
  87//usage:       "[-arPLHpfinlsTu] SOURCE DEST\n"
  88//usage:       "or: cp [-arPLHpfinlsu] SOURCE... { -t DIRECTORY | DIRECTORY }"
  89//usage:#define cp_full_usage "\n\n"
  90//usage:       "Copy SOURCEs to DEST\n"
  91//usage:     "\n        -a      Same as -dpR"
  92//usage:        IF_SELINUX(
  93//usage:     "\n        -c      Preserve security context"
  94//usage:        )
  95//usage:     "\n        -R,-r   Recurse"
  96//usage:     "\n        -d,-P   Preserve symlinks (default if -R)"
  97//usage:     "\n        -L      Follow all symlinks"
  98//usage:     "\n        -H      Follow symlinks on command line"
  99//usage:     "\n        -p      Preserve file attributes if possible"
 100//usage:     "\n        -f      Overwrite"
 101//usage:     "\n        -i      Prompt before overwrite"
 102//usage:     "\n        -n      Don't overwrite"
 103//usage:     "\n        -l,-s   Create (sym)links"
 104//usage:     "\n        -T      Refuse to copy if DEST is a directory"
 105//usage:     "\n        -t DIR  Copy all SOURCEs into DIR"
 106//usage:     "\n        -u      Copy only newer files"
 107
 108#include "libbb.h"
 109#include "libcoreutils/coreutils.h"
 110
 111/* This is a NOEXEC applet. Be very careful! */
 112
 113int cp_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 114int cp_main(int argc, char **argv)
 115{
 116        struct stat source_stat;
 117        struct stat dest_stat;
 118        const char *last;
 119        const char *dest;
 120        int s_flags;
 121        int d_flags;
 122        int flags;
 123        int status;
 124#if ENABLE_FEATURE_CP_LONG_OPTIONS
 125        enum {
 126                /*OPT_rmdest  = FILEUTILS_RMDEST = 1 << FILEUTILS_CP_OPTBITS */
 127                OPT_parents = 1 << (FILEUTILS_CP_OPTBITS+1),
 128                OPT_reflink = 1 << (FILEUTILS_CP_OPTBITS+2),
 129        };
 130# if ENABLE_FEATURE_CP_REFLINK
 131        char *reflink = NULL;
 132# endif
 133        flags = getopt32long(argv, "^"
 134                FILEUTILS_CP_OPTSTR
 135                "\0"
 136                // Need at least one argument. (Usually two+, but -t DIR can have only one)
 137                // Soft- and hardlinking doesn't mix
 138                // -P and -d are the same (-P is POSIX, -d is GNU)
 139                // -r and -R are the same
 140                // -R (and therefore -r) turns on -d (coreutils does this)
 141                // -a = -pdR
 142                // -i overrides -n and vice versa (last wins)
 143                "-1:l--s:s--l:Pd:rRd:Rd:apdR:i-n:n-i",
 144                "archive\0"        No_argument "a"
 145                "force\0"          No_argument "f"
 146                "interactive\0"    No_argument "i"
 147                "no-clobber\0"     No_argument "n"
 148                "link\0"           No_argument "l"
 149                "dereference\0"    No_argument "L"
 150                "no-dereference\0" No_argument "P"
 151                "recursive\0"      No_argument "R"
 152                "symbolic-link\0"  No_argument "s"
 153                "no-target-directory\0" No_argument "T"
 154                "target-directory\0" Required_argument "t"
 155                "verbose\0"        No_argument "v"
 156                "update\0"         No_argument "u"
 157                "remove-destination\0" No_argument "\xff"
 158                "parents\0"        No_argument "\xfe"
 159# if ENABLE_FEATURE_CP_REFLINK
 160                "reflink\0"        Optional_argument "\xfd"
 161# endif
 162                , &last
 163# if ENABLE_FEATURE_CP_REFLINK
 164                , &reflink
 165# endif
 166        );
 167# if ENABLE_FEATURE_CP_REFLINK
 168        BUILD_BUG_ON((int)OPT_reflink != (int)FILEUTILS_REFLINK);
 169        if (flags & FILEUTILS_REFLINK) {
 170                if (!reflink)
 171                        flags |= FILEUTILS_REFLINK_ALWAYS;
 172                else if (strcmp(reflink, "always") == 0)
 173                        flags |= FILEUTILS_REFLINK_ALWAYS;
 174                else if (strcmp(reflink, "auto") != 0)
 175                        bb_show_usage();
 176        }
 177# endif
 178#else
 179        flags = getopt32(argv, "^"
 180                FILEUTILS_CP_OPTSTR
 181                "\0"
 182                "-1:l--s:s--l:Pd:rRd:Rd:apdR"
 183                , &last
 184        );
 185#endif
 186        argc -= optind;
 187        argv += optind;
 188        /* Reverse this bit. If there is -d, bit is not set: */
 189        flags ^= FILEUTILS_DEREFERENCE;
 190        /* coreutils 6.9 compat:
 191         * by default, "cp" derefs symlinks (creates regular dest files),
 192         * but "cp -R" does not. We switch off deref if -r or -R (see above).
 193         * However, "cp -RL" must still deref symlinks: */
 194        if (flags & FILEUTILS_DEREF_SOFTLINK) /* -L */
 195                flags |= FILEUTILS_DEREFERENCE;
 196
 197#if ENABLE_SELINUX
 198        if (flags & FILEUTILS_PRESERVE_SECURITY_CONTEXT) {
 199                selinux_or_die();
 200        }
 201#endif
 202
 203        status = EXIT_SUCCESS;
 204        if (!(flags & FILEUTILS_TARGET_DIR)) {
 205                last = argv[argc - 1];
 206                if (argc < 2)
 207                        bb_show_usage();
 208                if (argc != 2) {
 209                        if (flags & FILEUTILS_NO_TARGET_DIR)
 210                                bb_show_usage();
 211                        /* "cp A B C... DIR" - target must be dir */
 212                } else /* argc == 2 */ {
 213                        /* "cp A B" - only case where target can be not a dir */
 214                        s_flags = cp_mv_stat2(*argv, &source_stat,
 215                                        (flags & FILEUTILS_DEREFERENCE) ? stat : lstat);
 216                        if (s_flags < 0) /* error other than ENOENT */
 217                                return EXIT_FAILURE;
 218                        d_flags = cp_mv_stat(last, &dest_stat);
 219                        if (d_flags < 0) /* error other than ENOENT */
 220                                return EXIT_FAILURE;
 221
 222                        if (flags & FILEUTILS_NO_TARGET_DIR) { /* -T */
 223                                if (!(s_flags & 2) && (d_flags & 2))
 224                                        /* cp -T NOTDIR DIR */
 225                                        bb_error_msg_and_die("'%s' is a directory", last);
 226                        }
 227
 228#if ENABLE_FEATURE_CP_LONG_OPTIONS
 229                        //bb_error_msg("flags:%x FILEUTILS_RMDEST:%x OPT_parents:%x",
 230                        //      flags, FILEUTILS_RMDEST, OPT_parents);
 231                        if (flags & OPT_parents) {
 232                                if (!(d_flags & 2)) {
 233                                        bb_simple_error_msg_and_die("with --parents, the destination must be a directory");
 234                                }
 235                        }
 236                        if (flags & FILEUTILS_RMDEST) {
 237                                flags |= FILEUTILS_FORCE;
 238                        }
 239#endif
 240
 241                        /* ...if neither is a directory...  */
 242                        if (!((s_flags | d_flags) & 2)
 243                            /* ...or: recursing, the 1st is a directory, and the 2nd doesn't exist... */
 244                         || ((flags & FILEUTILS_RECUR) && (s_flags & 2) && !d_flags)
 245                         || (flags & FILEUTILS_NO_TARGET_DIR)
 246                        ) {
 247                                /* Do a simple copy */
 248                                dest = last;
 249                                goto DO_COPY; /* NB: argc==2 -> *++argv==last */
 250                        }
 251                }
 252        }
 253        /* else: last is DIR from "-t DIR" */
 254
 255        while (1) {
 256#if ENABLE_FEATURE_CP_LONG_OPTIONS
 257                if (flags & OPT_parents) {
 258                        char *dest_dup;
 259                        char *dest_dir;
 260                        dest = concat_path_file(last, *argv);
 261                        dest_dup = xstrdup(dest);
 262                        dest_dir = dirname(dest_dup);
 263                        if (bb_make_directory(dest_dir, -1, FILEUTILS_RECUR)) {
 264                                return EXIT_FAILURE;
 265                        }
 266                        free(dest_dup);
 267                        goto DO_COPY;
 268                }
 269#endif
 270                dest = concat_path_file(last, bb_get_last_path_component_strip(*argv));
 271 DO_COPY:
 272                if (copy_file(*argv, dest, flags) < 0) {
 273                        status = EXIT_FAILURE;
 274                }
 275                if (!*++argv || *argv == last) {
 276                        /* possibly leaking dest... */
 277                        break;
 278                }
 279                /* don't move up: dest may be == last and not malloced! */
 280                free((void*)dest);
 281        }
 282
 283        /* Exit. We are NOEXEC, not NOFORK. We do exit at the end of main() */
 284        return status;
 285}
 286