ls: add support for -H
[oweals/busybox.git] / coreutils / cp.c
index 254445f029703ed9db189906a5328ebcb0626798..ab17b39a69f2fcc272601b748747a97072f73ed5 100644 (file)
 /*
  * Mini cp implementation for busybox
  *
- *
  * Copyright (C) 2000 by Matt Kraai <kraai@alumni.carnegiemellon.edu>
+ * SELinux support by Yuichi Nakamura <ynakam@hitachisoft.jp>
  *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
+ */
+
+/* http://www.opengroup.org/onlinepubs/007904975/utilities/cp.html */
+
+/* Mar 16, 2003      Manuel Novoa III   (mjn3@codepoet.org)
  *
+ * Size reduction.
  */
 
-#include <sys/types.h>
-#include <sys/stat.h>
-#include <unistd.h>
-#include <fcntl.h>
-#include <utime.h>
-#include <errno.h>
-#include <dirent.h>
-#include <stdlib.h>
+#include "libbb.h"
+#include "libcoreutils/coreutils.h"
 
-#include "busybox.h"
+/* This is a NOEXEC applet. Be very careful! */
 
-extern int cp_main(int argc, char **argv)
+int cp_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int cp_main(int argc, char **argv)
 {
-       int status = 0;
-       int opt;
-       int flags = 0;
-       int i;
+       struct stat source_stat;
+       struct stat dest_stat;
+       const char *last;
+       const char *dest;
+       int s_flags;
+       int d_flags;
+       int flags;
+       int status;
+       enum {
+               OPT_a = 1 << (sizeof(FILEUTILS_CP_OPTSTR)-1),
+               OPT_r = 1 << (sizeof(FILEUTILS_CP_OPTSTR)),
+               OPT_P = 1 << (sizeof(FILEUTILS_CP_OPTSTR)+1),
+               OPT_v = 1 << (sizeof(FILEUTILS_CP_OPTSTR)+2),
+#if ENABLE_FEATURE_CP_LONG_OPTIONS
+               OPT_parents = 1 << (sizeof(FILEUTILS_CP_OPTSTR)+3),
+#endif
+       };
 
-       while ((opt = getopt(argc, argv, "adfipR")) != -1)
-               switch (opt) {
-               case 'a':
-                       flags |= CP_PRESERVE_STATUS | CP_RECUR;
-                       /* fallthrough */
-               case 'd':
-                       flags |= CP_PRESERVE_SYMLINKS;
-                       break;
-               case 'f':
-                       flags |= CP_FORCE;
-                       break;
-               case 'i':
-                       flags |= CP_INTERACTIVE;
-                       break;
-               case 'p':
-                       flags |= CP_PRESERVE_STATUS;
-                       break;
-               case 'R':
-                       flags |= CP_RECUR;
-                       break;
-               default:
-                       show_usage();
-               }
-       
-       if (optind + 2 > argc)
-               show_usage();
+       // Need at least two arguments
+       // Soft- and hardlinking doesn't mix
+       // -P and -d are the same (-P is POSIX, -d is GNU)
+       // -r and -R are the same
+       // -R (and therefore -r) turns on -d (coreutils does this)
+       // -a = -pdR
+       opt_complementary = "-2:l--s:s--l:Pd:rRd:Rd:apdR";
+#if ENABLE_FEATURE_CP_LONG_OPTIONS
+       applet_long_options =
+               "archive\0"        No_argument "a"
+               "force\0"          No_argument "f"
+               "interactive\0"    No_argument "i"
+               "link\0"           No_argument "l"
+               "dereference\0"    No_argument "L"
+               "no-dereference\0" No_argument "P"
+               "recursive\0"      No_argument "R"
+               "symbolic-link\0"  No_argument "s"
+               "verbose\0"        No_argument "v"
+               "parents\0"        No_argument "\xff"
+               ;
+#endif
+       // -v (--verbose) is ignored
+       flags = getopt32(argv, FILEUTILS_CP_OPTSTR "arPv");
+       /* Options of cp from GNU coreutils 6.10:
+        * -a, --archive
+        * -f, --force
+        * -i, --interactive
+        * -l, --link
+        * -L, --dereference
+        * -P, --no-dereference
+        * -R, -r, --recursive
+        * -s, --symbolic-link
+        * -v, --verbose
+        * -H   follow command-line symbolic links in SOURCE
+        * -d   same as --no-dereference --preserve=links
+        * -p   same as --preserve=mode,ownership,timestamps
+        * -c   same as --preserve=context
+        * --parents
+        *      use full source file name under DIRECTORY
+        * NOT SUPPORTED IN BBOX:
+        * --backup[=CONTROL]
+        *      make a backup of each existing destination file
+        * -b   like --backup but does not accept an argument
+        * --copy-contents
+        *      copy contents of special files when recursive
+        * --preserve[=ATTR_LIST]
+        *      preserve attributes (default: mode,ownership,timestamps),
+        *      if possible additional attributes: security context,links,all
+        * --no-preserve=ATTR_LIST
+        * --remove-destination
+        *      remove  each existing destination file before attempting to open
+        * --sparse=WHEN
+        *      control creation of sparse files
+        * --strip-trailing-slashes
+        *      remove any trailing slashes from each SOURCE argument
+        * -S, --suffix=SUFFIX
+        *      override the usual backup suffix
+        * -t, --target-directory=DIRECTORY
+        *      copy all SOURCE arguments into DIRECTORY
+        * -T, --no-target-directory
+        *      treat DEST as a normal file
+        * -u, --update
+        *      copy only when the SOURCE file is newer than the destination
+        *      file or when the destination file is missing
+        * -x, --one-file-system
+        *      stay on this file system
+        * -Z, --context=CONTEXT
+        *      (SELinux) set SELinux security context of copy to CONTEXT
+        */
+       argc -= optind;
+       argv += optind;
+       /* Reverse this bit. If there is -d, bit is not set: */
+       flags ^= FILEUTILS_DEREFERENCE;
+       /* coreutils 6.9 compat:
+        * by default, "cp" derefs symlinks (creates regular dest files),
+        * but "cp -R" does not. We switch off deref if -r or -R (see above).
+        * However, "cp -RL" must still deref symlinks: */
+       if (flags & FILEUTILS_DEREF_SOFTLINK) /* -L */
+               flags |= FILEUTILS_DEREFERENCE;
 
+#if ENABLE_SELINUX
+       if (flags & FILEUTILS_PRESERVE_SECURITY_CONTEXT) {
+               selinux_or_die();
+       }
+#endif
+
+       status = EXIT_SUCCESS;
+       last = argv[argc - 1];
        /* If there are only two arguments and...  */
-       if (optind + 2 == argc) {
-               struct stat source_stat;
-               struct stat dest_stat;
-               int source_exists = 1;
-               int dest_exists = 1;
+       if (argc == 2) {
+               s_flags = cp_mv_stat2(*argv, &source_stat,
+                               (flags & FILEUTILS_DEREFERENCE) ? stat : lstat);
+               if (s_flags < 0)
+                       return EXIT_FAILURE;
+               d_flags = cp_mv_stat(last, &dest_stat);
+               if (d_flags < 0)
+                       return EXIT_FAILURE;
 
-               if (((flags & CP_PRESERVE_SYMLINKS) &&
-                               lstat(argv[optind], &source_stat) < 0) ||
-                               (!(flags & CP_PRESERVE_SYMLINKS) &&
-                                stat(argv[optind], &source_stat))) {
-                       if (errno != ENOENT)
-                               perror_msg_and_die("unable to stat `%s'", argv[optind]);
-                       source_exists = 0;
+#if ENABLE_FEATURE_CP_LONG_OPTIONS
+               if (flags & OPT_parents) {
+                       if (!(d_flags & 2)) {
+                               bb_error_msg_and_die("with --parents, the destination must be a directory");
+                       }
                }
+#endif
 
-               if (stat(argv[optind + 1], &dest_stat) < 0) {
-                       if (errno != ENOENT)
-                               perror_msg_and_die("unable to stat `%s'", argv[optind + 1]);
-                       dest_exists = 0;
-               }
-               
-               /* ...if neither is a directory or...  */
-               if (((!source_exists || !S_ISDIR(source_stat.st_mode)) &&
-                               (!dest_exists || !S_ISDIR(dest_stat.st_mode))) ||
-                               /* ...recursing, the first is a directory, and the
-                                * second doesn't exist, then... */
-                               ((flags & CP_RECUR) && S_ISDIR(source_stat.st_mode) &&
-                                !dest_exists)) {
-                       /* ...do a simple copy.  */
-                       if (copy_file(argv[optind], argv[optind + 1], flags) < 0)
-                               status = 1;
-                       return status;
+               /* ...if neither is a directory...  */
+               if (!((s_flags | d_flags) & 2)
+                   /* ...or: recursing, the 1st is a directory, and the 2nd doesn't exist... */
+                || ((flags & FILEUTILS_RECUR) && (s_flags & 2) && !d_flags)
+               ) {
+                       /* Do a simple copy */
+                       dest = last;
+                       goto DO_COPY; /* NB: argc==2 -> *++argv==last */
                }
        }
 
-       for (i = optind; i < argc - 1; i++) {
-               char *dest = concat_path_file(argv[argc - 1],
-                               get_last_path_component(argv[i]));
-               if (copy_file(argv[i], dest, flags) < 0)
-                       status = 1;
-               free(dest);
+       while (1) {
+#if ENABLE_FEATURE_CP_LONG_OPTIONS
+               if (flags & OPT_parents) {
+                       char *dest_dup;
+                       char *dest_dir;
+                       dest = concat_path_file(last, *argv);
+                       dest_dup = xstrdup(dest);
+                       dest_dir = dirname(dest_dup);
+                       if (bb_make_directory(dest_dir, -1, FILEUTILS_RECUR)) {
+                               return EXIT_FAILURE;
+                       }
+                       free(dest_dup);
+                       goto DO_COPY;
+               }
+#endif
+               dest = concat_path_file(last, bb_get_last_path_component_strip(*argv));
+ DO_COPY:
+               if (copy_file(*argv, dest, flags) < 0) {
+                       status = EXIT_FAILURE;
+               }
+               if (*++argv == last) {
+                       /* possibly leaking dest... */
+                       break;
+               }
+               /* don't move up: dest may be == last and not malloced! */
+               free((void*)dest);
        }
 
+       /* Exit. We are NOEXEC, not NOFORK. We do exit at the end of main() */
        return status;
 }