diff options
Diffstat (limited to 'tools/acp/acp.c')
-rw-r--r-- | tools/acp/acp.c | 252 |
1 files changed, 252 insertions, 0 deletions
diff --git a/tools/acp/acp.c b/tools/acp/acp.c new file mode 100644 index 0000000..eb1de1f --- /dev/null +++ b/tools/acp/acp.c @@ -0,0 +1,252 @@ +/* + * 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, cc; + char* stripDest = NULL; + int stripDestLen; + struct stat destStat; + 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; +} + |