Jean Wolter writes:

[oweals/busybox.git] / coreutils / cp.c

diff --git a/coreutils/cp.c b/coreutils/cp.c
index 94b4ab024010ff324e6fa7a8ec6f94cfaa586f89..6a82f6bffcc61a4257eb2600ec351d44f469cb89 100644 (file)
--- a/coreutils/cp.c
+++ b/coreutils/cp.c
@@ -1,7 +1,8 @@
+/* vi: set sw=4 ts=4: */
 /*
  * Mini cp implementation for busybox
  *
- * Copyright (C) 1998 by Erik Andersen <andersee@debian.org>
+ * Copyright (C) 2000 by Matt Kraai <kraai@alumni.carnegiemellon.edu>
  *
  * 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
@@ -19,89 +20,100 @@
  *
  */
 
-#include "internal.h"
-#include <stdio.h>
-#include <time.h>
-#include <utime.h>
-#include <dirent.h>
-
-const char cp_usage[] = "cp [OPTION]... SOURCE DEST\n"
-    "   or: cp [OPTION]... SOURCE... DIRECTORY\n"
-    "Copy SOURCE to DEST, or multiple SOURCE(s) to DIRECTORY.\n"
-    "\n"
-    "\t-a\tsame as -dpR\n"
-    "\t-d\tpreserve links\n"
-    "\t-p\tpreserve file attributes if possable\n"
-    "\t-R\tcopy directories recursively\n";
+/* BB_AUDIT SUSv3 defects - unsupported options -H, -L, and -P. */
+/* BB_AUDIT GNU defects - only extension options supported are -a and -d.  */
+/* http://www.opengroup.org/onlinepubs/007904975/utilities/cp.html */
 
+/* Mar 16, 2003      Manuel Novoa III   (mjn3@codepoet.org)
+ *
+ * Size reduction.
+ */
 
-static int recursiveFlag = FALSE;
-static int followLinks = FALSE;
-static int preserveFlag = FALSE;
-static const char *srcName;
-static const char *destName;
-
+#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 <assert.h>
+#include "busybox.h"
+#include "libcoreutils/coreutils.h"
 
-static int fileAction(const char *fileName)
-{
-    char newdestName[NAME_MAX];
-    strcpy(newdestName, destName);
-    strcat(newdestName, fileName+(strlen(srcName)));
-    return (copyFile(fileName, newdestName, preserveFlag, followLinks));
-}
+/* WARNING!! ORDER IS IMPORTANT!! */
+static const char cp_opts[] = "pdRfiar";
 
 extern 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 = 0;
+
+       /* Since these are enums, #if tests will not work.  So use assert()s. */
+       assert(FILEUTILS_PRESERVE_STATUS == 1);
+       assert(FILEUTILS_DEREFERENCE == 2);
+       assert(FILEUTILS_RECUR == 4);
+       assert(FILEUTILS_FORCE == 8);
+       assert(FILEUTILS_INTERACTIVE == 16);
 
-    int dirFlag;
+       flags = bb_getopt_ulflags(argc, argv, cp_opts);
 
-    if (argc < 3) {
-       fprintf(stderr, "Usage: %s", cp_usage);
-       exit (FALSE);
-    }
-    argc--;
-    argv++;
+       if (flags & 32) {
+               flags |= (FILEUTILS_PRESERVE_STATUS | FILEUTILS_RECUR | FILEUTILS_DEREFERENCE);
+       }
+       if (flags & 64) {
+               /* Make -r a synonym for -R,
+                * -r was marked as obsolete in SUSv3, but is included for compatability
+                */
+               flags |= FILEUTILS_RECUR;
+       }
 
-    /* Parse any options */
-    while (**argv == '-') {
-       while (*++(*argv))
-           switch (**argv) {
-           case 'a':
-               followLinks = TRUE;
-               preserveFlag = TRUE;
-               recursiveFlag = TRUE;
-               break;
-           case 'd':
-               followLinks = TRUE;
-               break;
-           case 'p':
-               preserveFlag = TRUE;
-               break;
-           case 'R':
-               recursiveFlag = TRUE;
-               break;
-           default:
-               fprintf(stderr, "Usage: %s\n", cp_usage);
-               exit(FALSE);
-           }
-       argc--;
-       argv++;
-    }
+       flags ^= FILEUTILS_DEREFERENCE;         /* The sense of this flag was reversed. */
 
+       if (optind + 2 > argc) {
+               bb_show_usage();
+       }
 
-    destName = argv[argc - 1];
+       last = argv[argc - 1];
+       argv += optind;
 
-    dirFlag = isDirectory(destName);
+       /* If there are only two arguments and...  */
+       if (optind + 2 == argc) {
+               s_flags = cp_mv_stat2(*argv, &source_stat,
+                                                                (flags & FILEUTILS_DEREFERENCE) ? stat : lstat);
+               if ((s_flags < 0) || ((d_flags = cp_mv_stat(last, &dest_stat)) < 0)) {
+                       exit(EXIT_FAILURE);
+               }
+               /* ...if neither is a directory or...  */
+               if ( !((s_flags | d_flags) & 2) ||
+                       /* ...recursing, the 1st is a directory, and the 2nd doesn't exist... */
+                       /* ((flags & FILEUTILS_RECUR) && (s_flags & 2) && !d_flags) */
+                       /* Simplify the above since FILEUTILS_RECUR >> 1 == 2. */
+                       ((((flags & FILEUTILS_RECUR) >> 1) & s_flags) && !d_flags)
+               ) {
+                       /* ...do a simple copy.  */
+                               dest = last;
+                               goto DO_COPY; /* Note: optind+2==argc implies argv[1]==last below. */
+               }
+       }
 
-    if ((argc > 3) && !dirFlag) {
-       fprintf(stderr, "%s: not a directory\n", destName);
-       exit (FALSE);
-    }
+       do {
+               dest = concat_path_file(last, bb_get_last_path_component(*argv));
+       DO_COPY:
+               if (copy_file(*argv, dest, flags) < 0) {
+                       status = 1;
+               }
+               if (*++argv == last) {
+                       break;
+               }
+               free((void *) dest);
+       } while (1);
 
-    while (argc-- >= 2) {
-       srcName = *(argv++);
-       exit( recursiveAction(srcName, recursiveFlag, followLinks,
-                              fileAction, fileAction));
-    }
-    exit( TRUE);
+       exit(status);
 }