C++程序  |  252行  |  6.85 KB

/*
 * Copyright 2005 The Android Open Source Project
 *
 * Android "cp" replacement.
 *
 * The GNU/Linux "cp" uses O_LARGEFILE in its open() calls, utimes() instead
 * of utime(), and getxattr()/setxattr() instead of chmod().  These are
 * probably "better", but are non-portable, and not necessary for our
 * purposes.
 */
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <getopt.h>
#include <dirent.h>
#include <fcntl.h>
#include <utime.h>
#include <limits.h>
#include <errno.h>
#include <assert.h>
#include <host/CopyFile.h>

/*#define DEBUG_MSGS*/
#ifdef DEBUG_MSGS
# define DBUG(x) printf x
#else
# define DBUG(x) ((void)0)
#endif

#define FSSEP '/'       /* filename separator char */


/*
 * Process the command-line file arguments.
 *
 * Returns 0 on success.
 */
int process(int argc, char* const argv[], unsigned int options)
{
    int retVal = 0;
    int i;
    char* stripDest = NULL;
    int stripDestLen;
    bool destMustBeDir = false;
    struct stat sb;

    assert(argc >= 2);

    /*
     * Check for and trim a trailing slash on the last arg.
     *
     * It's useful to be able to say "cp foo bar/" when you want to copy
     * a single file into a directory.  If you say "cp foo bar", and "bar"
     * does not exist, it will create "bar", when what you really wanted
     * was for the cp command to fail with "directory does not exist".
     */
    stripDestLen = strlen(argv[argc-1]);
    stripDest = malloc(stripDestLen+1);
    memcpy(stripDest, argv[argc-1], stripDestLen+1);
    if (stripDest[stripDestLen-1] == FSSEP) {
        stripDest[--stripDestLen] = '\0';
        destMustBeDir = true;
    }

    if (argc > 2)
        destMustBeDir = true;

    /*
     * Start with a quick check to ensure that, if we're expecting to copy
     * to a directory, the target already exists and is actually a directory.
     * It's okay if it's a symlink to a directory.
     *
     * If it turns out to be a directory, go ahead and raise the
     * destMustBeDir flag so we do some path concatenation below.
     */
    if (stat(stripDest, &sb) < 0) {
        if (destMustBeDir) {
            if (errno == ENOENT)
                fprintf(stderr,
                    "acp: destination directory '%s' does not exist\n",
                    stripDest);
            else
                fprintf(stderr, "acp: unable to stat dest dir\n");
            retVal = 1;
            goto bail;
        }
    } else {
        if (S_ISDIR(sb.st_mode)) {
            DBUG(("--- dest exists and is a dir, setting flag\n"));
            destMustBeDir = true;
        } else if (destMustBeDir) {
            fprintf(stderr,
                "acp: destination '%s' is not a directory\n",
                stripDest);
            retVal = 1;
            goto bail;
        }
    }

    /*
     * Copying files.
     *
     * Strip trailing slashes off.  They shouldn't be there, but
     * sometimes file completion will put them in for directories.
     *
     * The observed behavior of GNU and BSD cp is that they print warnings
     * if something fails, but continue on.  If any part fails, the command
     * exits with an error status.
     */
    for (i = 0; i < argc-1; i++) {
        const char* srcName;
        char* src;
        char* dst;
        int copyResult;
        int srcLen;

        /* make a copy of the source name, and strip trailing '/' */
        srcLen = strlen(argv[i]);
        src = malloc(srcLen+1);
        memcpy(src, argv[i], srcLen+1);

        if (src[srcLen-1] == FSSEP)
            src[--srcLen] = '\0';

        /* find just the name part */
        srcName = strrchr(src, FSSEP);
        if (srcName == NULL) {
            srcName = src;
        } else {
            srcName++;
            assert(*srcName != '\0');
        }
        
        if (destMustBeDir) {
            /* concatenate dest dir and src name */
            int srcNameLen = strlen(srcName);

            dst = malloc(stripDestLen +1 + srcNameLen +1);
            memcpy(dst, stripDest, stripDestLen);
            dst[stripDestLen] = FSSEP;
            memcpy(dst + stripDestLen+1, srcName, srcNameLen+1);
        } else {
            /* simple */
            dst = stripDest;
        }

        /*
         * Copy the source to the destination.
         */
        copyResult = copyFile(src, dst, options);

        if (copyResult != 0)
            retVal = 1;

        free(src);
        if (dst != stripDest)
            free(dst);
    }

bail:
    free(stripDest);
    return retVal;
}

/*
 * Set up the options.
 */
int main(int argc, char* const argv[])
{
    bool wantUsage;
    int ic, retVal;
    int verboseLevel;
    unsigned int options;

    verboseLevel = 0;
    options = 0;
    wantUsage = false;

    while (1) {
        ic = getopt(argc, argv, "defprtuv");
        if (ic < 0)
            break;

        switch (ic) {
            case 'd':
                options |= COPY_NO_DEREFERENCE;
                break;
            case 'e':
                options |= COPY_TRY_EXE;
                break;
            case 'f':
                options |= COPY_FORCE;
                break;
            case 'p':
                options |= COPY_PERMISSIONS;
                break;
            case 't':
                options |= COPY_TIMESTAMPS;
                break;
            case 'r':
                options |= COPY_RECURSIVE;
                break;
            case 'u':
                options |= COPY_UPDATE_ONLY;
                break;
            case 'v':
                verboseLevel++;
                break;
            default:
                fprintf(stderr, "Unexpected arg -%c\n", ic);
                wantUsage = true;
                break;
        }

        if (wantUsage)
            break;
    }

    options |= verboseLevel & COPY_VERBOSE_MASK;

    if (optind == argc-1) {
        fprintf(stderr, "acp: missing destination file\n");
        return 2;
    } else if (optind+2 > argc)
        wantUsage = true;

    if (wantUsage) {
        fprintf(stderr, "Usage: acp [OPTION]... SOURCE DEST\n");
        fprintf(stderr, "  or:  acp [OPTION]... SOURCE... DIRECTORY\n");
        fprintf(stderr, "\nOptions:\n");
        fprintf(stderr, "  -d  never follow (dereference) symbolic links\n");
        fprintf(stderr, "  -e  if source file doesn't exist, try adding "
                        "'.exe' [Win32 only]\n");
        fprintf(stderr, "  -f  use force, removing existing file if it's "
                        "not writeable\n");
        fprintf(stderr, "  -p  preserve mode, ownership\n");
        fprintf(stderr, "  -r  recursive copy\n");
        fprintf(stderr, "  -t  preserve timestamps\n");
        fprintf(stderr, "  -u  update only: don't copy if dest is newer\n");
        fprintf(stderr, "  -v  verbose output (-vv is more verbose)\n");
        return 2;
    }

    retVal = process(argc-optind, argv+optind, options);
    DBUG(("EXIT: %d\n", retVal));
    return retVal;
}