source: MondoRescue/branches/3.2/mindi-busybox/coreutils/cp.c@ 3232

/* vi: set sw=4 ts=4: */
/*
 * Mini cp implementation for busybox
 *
 * Copyright (C) 2000 by Matt Kraai <kraai@alumni.carnegiemellon.edu>
 * SELinux support by Yuichi Nakamura <ynakam@hitachisoft.jp>
 *
 * 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.
 */

//usage:#define cp_trivial_usage
//usage:       "[OPTIONS] SOURCE... DEST"
//usage:#define cp_full_usage "\n\n"
//usage:       "Copy SOURCE(s) to DEST\n"
//usage:     "\n    -a  Same as -dpR"
//usage:    IF_SELINUX(
//usage:     "\n    -c  Preserve security context"
//usage:    )
//usage:     "\n    -R,-r   Recurse"
//usage:     "\n    -d,-P   Preserve symlinks (default if -R)"
//usage:     "\n    -L  Follow all symlinks"
//usage:     "\n    -H  Follow symlinks on command line"
//usage:     "\n    -p  Preserve file attributes if possible"
//usage:     "\n    -f  Overwrite"
//usage:     "\n    -i  Prompt before overwrite"
//usage:     "\n    -l,-s   Create (sym)links"

#include "libbb.h"
#include "libcoreutils/coreutils.h"

/* This is a NOEXEC applet. Be very careful! */

int cp_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int cp_main(int argc, char **argv)
{
    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
    };

    // 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 (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 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 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 */
        }
    }

    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;
}