coreutils/cp.c

   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
  11 /* http://www.opengroup.org/onlinepubs/007904975/utilities/cp.html */
  12
  13 /* Mar 16, 2003      Manuel Novoa III   (mjn3@codepoet.org)
  14  *
  15  * Size reduction.
  16  */
  17
  18 //usage:#define cp_trivial_usage
  19 //usage:       "[OPTIONS] SOURCE DEST"
  20 //usage:#define cp_full_usage "\n\n"
  21 //usage:       "Copy SOURCE to DEST, or multiple SOURCE(s) to DIRECTORY\n"
  22 //usage:     "\nOptions:"
  23 //usage:     "\n        -a      Same as -dpR"
  24 //usage:        IF_SELINUX(
  25 //usage:     "\n        -c      Preserve security context"
  26 //usage:        )
  27 //usage:     "\n        -R,-r   Recurse"
  28 //usage:     "\n        -d,-P   Preserve symlinks (default if -R)"
  29 //usage:     "\n        -L      Follow all symlinks"
  30 //usage:     "\n        -H      Follow symlinks on command line"
  31 //usage:     "\n        -p      Preserve file attributes if possible"
  32 //usage:     "\n        -f      Overwrite"
  33 //usage:     "\n        -i      Prompt before overwrite"
  34 //usage:     "\n        -l,-s   Create (sym)links"
  35
  36 #include "libbb.h"
  37 #include "libcoreutils/coreutils.h"
  38
  39 /* This is a NOEXEC applet. Be very careful! */
  40
  41 int cp_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  42 int cp_main(int argc, char **argv)
  43 {
  44         struct stat source_stat;
  45         struct stat dest_stat;
  46         const char *last;
  47         const char *dest;
  48         int s_flags;
  49         int d_flags;
  50         int flags;
  51         int status;
  52         enum {
  53                 OPT_a = 1 << (sizeof(FILEUTILS_CP_OPTSTR)-1),
  54                 OPT_r = 1 << (sizeof(FILEUTILS_CP_OPTSTR)),
  55                 OPT_P = 1 << (sizeof(FILEUTILS_CP_OPTSTR)+1),
  56                 OPT_v = 1 << (sizeof(FILEUTILS_CP_OPTSTR)+2),
  57 #if ENABLE_FEATURE_CP_LONG_OPTIONS
  58                 OPT_parents = 1 << (sizeof(FILEUTILS_CP_OPTSTR)+3),
  59 #endif
  60         };
  61
  62         // Need at least two arguments
  63         // Soft- and hardlinking doesn't mix
  64         // -P and -d are the same (-P is POSIX, -d is GNU)
  65         // -r and -R are the same
  66         // -R (and therefore -r) turns on -d (coreutils does this)
  67         // -a = -pdR
  68         opt_complementary = "-2:l--s:s--l:Pd:rRd:Rd:apdR";
  69 #if ENABLE_FEATURE_CP_LONG_OPTIONS
  70         applet_long_options =
  71                 "archive\0"        No_argument "a"
  72                 "force\0"          No_argument "f"
  73                 "interactive\0"    No_argument "i"
  74                 "link\0"           No_argument "l"
  75                 "dereference\0"    No_argument "L"
  76                 "no-dereference\0" No_argument "P"
  77                 "recursive\0"      No_argument "R"
  78                 "symbolic-link\0"  No_argument "s"
  79                 "verbose\0"        No_argument "v"
  80                 "parents\0"        No_argument "\xff"
  81                 ;
  82 #endif
  83         // -v (--verbose) is ignored
  84         flags = getopt32(argv, FILEUTILS_CP_OPTSTR "arPv");
  85         /* Options of cp from GNU coreutils 6.10:
  86          * -a, --archive
  87          * -f, --force
  88          * -i, --interactive
  89          * -l, --link
  90          * -L, --dereference
  91          * -P, --no-dereference
  92          * -R, -r, --recursive
  93          * -s, --symbolic-link
  94          * -v, --verbose
  95          * -H   follow command-line symbolic links in SOURCE
  96          * -d   same as --no-dereference --preserve=links
  97          * -p   same as --preserve=mode,ownership,timestamps
  98          * -c   same as --preserve=context
  99          * --parents
 100          *      use full source file name under DIRECTORY
 101          * NOT SUPPORTED IN BBOX:
 102          * --backup[=CONTROL]
 103          *      make a backup of each existing destination file
 104          * -b   like --backup but does not accept an argument
 105          * --copy-contents
 106          *      copy contents of special files when recursive
 107          * --preserve[=ATTR_LIST]
 108          *      preserve attributes (default: mode,ownership,timestamps),
 109          *      if possible additional attributes: security context,links,all
 110          * --no-preserve=ATTR_LIST
 111          * --remove-destination
 112          *      remove  each existing destination file before attempting to open
 113          * --sparse=WHEN
 114          *      control creation of sparse files
 115          * --strip-trailing-slashes
 116          *      remove any trailing slashes from each SOURCE argument
 117          * -S, --suffix=SUFFIX
 118          *      override the usual backup suffix
 119          * -t, --target-directory=DIRECTORY
 120          *      copy all SOURCE arguments into DIRECTORY
 121          * -T, --no-target-directory
 122          *      treat DEST as a normal file
 123          * -u, --update
 124          *      copy only when the SOURCE file is newer than the destination
 125          *      file or when the destination file is missing
 126          * -x, --one-file-system
 127          *      stay on this file system
 128          * -Z, --context=CONTEXT
 129          *      (SELinux) set SELinux security context of copy to CONTEXT
 130          */
 131         argc -= optind;
 132         argv += optind;
 133         /* Reverse this bit. If there is -d, bit is not set: */
 134         flags ^= FILEUTILS_DEREFERENCE;
 135         /* coreutils 6.9 compat:
 136          * by default, "cp" derefs symlinks (creates regular dest files),
 137          * but "cp -R" does not. We switch off deref if -r or -R (see above).
 138          * However, "cp -RL" must still deref symlinks: */
 139         if (flags & FILEUTILS_DEREF_SOFTLINK) /* -L */
 140                 flags |= FILEUTILS_DEREFERENCE;
 141
 142 #if ENABLE_SELINUX
 143         if (flags & FILEUTILS_PRESERVE_SECURITY_CONTEXT) {
 144                 selinux_or_die();
 145         }
 146 #endif
 147
 148         status = EXIT_SUCCESS;
 149         last = argv[argc - 1];
 150         /* If there are only two arguments and...  */
 151         if (argc == 2) {
 152                 s_flags = cp_mv_stat2(*argv, &source_stat,
 153                                 (flags & FILEUTILS_DEREFERENCE) ? stat : lstat);
 154                 if (s_flags < 0)
 155                         return EXIT_FAILURE;
 156                 d_flags = cp_mv_stat(last, &dest_stat);
 157                 if (d_flags < 0)
 158                         return EXIT_FAILURE;
 159
 160 #if ENABLE_FEATURE_CP_LONG_OPTIONS
 161                 if (flags & OPT_parents) {
 162                         if (!(d_flags & 2)) {
 163                                 bb_error_msg_and_die("with --parents, the destination must be a directory");
 164                         }
 165                 }
 166 #endif
 167
 168                 /* ...if neither is a directory...  */
 169                 if (!((s_flags | d_flags) & 2)
 170                     /* ...or: recursing, the 1st is a directory, and the 2nd doesn't exist... */
 171                  || ((flags & FILEUTILS_RECUR) && (s_flags & 2) && !d_flags)
 172                 ) {
 173                         /* Do a simple copy */
 174                         dest = last;
 175                         goto DO_COPY; /* NB: argc==2 -> *++argv==last */
 176                 }
 177         }
 178
 179         while (1) {
 180 #if ENABLE_FEATURE_CP_LONG_OPTIONS
 181                 if (flags & OPT_parents) {
 182                         char *dest_dup;
 183                         char *dest_dir;
 184                         dest = concat_path_file(last, *argv);
 185                         dest_dup = xstrdup(dest);
 186                         dest_dir = dirname(dest_dup);
 187                         if (bb_make_directory(dest_dir, -1, FILEUTILS_RECUR)) {
 188                                 return EXIT_FAILURE;
 189                         }
 190                         free(dest_dup);
 191                         goto DO_COPY;
 192                 }
 193 #endif
 194                 dest = concat_path_file(last, bb_get_last_path_component_strip(*argv));
 195  DO_COPY:
 196                 if (copy_file(*argv, dest, flags) < 0) {
 197                         status = EXIT_FAILURE;
 198                 }
 199                 if (*++argv == last) {
 200                         /* possibly leaking dest... */
 201                         break;
 202                 }
 203                 /* don't move up: dest may be == last and not malloced! */
 204                 free((void*)dest);
 205         }
 206
 207         /* Exit. We are NOEXEC, not NOFORK. We do exit at the end of main() */
 208         return status;
 209 }