Fix up some silly macros and use xmalloc and friends exclusively.
[oweals/busybox.git] / cp_mv.c
diff --git a/cp_mv.c b/cp_mv.c
index 2ba8662b914fc23afef5522a1cf92fcb7bd51544..fb48d3c5cecc5c4698e97a1102d7aa7de48977cf 100644 (file)
--- a/cp_mv.c
+++ b/cp_mv.c
@@ -1,8 +1,9 @@
+/* vi: set sw=4 ts=4: */
 /*
  * Mini `cp' and `mv' implementation for BusyBox.
  *
  *
- * Copyright (C) 1999 by Lineo, inc.
+ * Copyright (C) 1999,2000,2001 by Lineo, inc.
  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
  *
  * Copyright (C) 2000 by BitterSweet Enterprises, LLC. (GPL)
  *
  */
 
-#include "internal.h"
+#include <stdio.h>
+#include <time.h>
+#include <utime.h>
+#include <dirent.h>
+#include <sys/param.h>
+#include <setjmp.h>
+#include <string.h>
+#include <unistd.h>
+#include <errno.h>
+#include <getopt.h>
+#include <stdlib.h>
+#include "busybox.h"
 #define BB_DECLARE_EXTERN
 #define bb_need_name_too_long
 #define bb_need_omitting_directory
 #define bb_need_not_a_directory
 #include "messages.c"
 
-#include <stdio.h>
-#include <time.h>
-#include <utime.h>
-#include <dirent.h>
-#include <sys/param.h>
 
-#define is_cp 0
-#define is_mv 1
-static const char *dz;         /* dollar zero, .bss */
-static int dz_i;               /* index,       .bss */
-static const char *cp_mv_usage[] =     /* .rodata */
-{
-     "cp [OPTION]... SOURCE DEST\n"
-     "   or: cp [OPTION]... SOURCE... DIRECTORY\n\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 possible\n"
-     "\t-R\tcopy directories recursively\n"
-     ,
-     "mv SOURCE DEST\n"
-     "   or: mv SOURCE... DIRECTORY\n\n"
-     "Rename SOURCE to DEST, or move SOURCE(s) to DIRECTORY.\n"
-     "Warning!!  This is not GNU `mv'.  It does not preserve hard links.\n"
-};
+static const int is_cp = 0;
+static const int is_mv = 1;
+static int         dz_i;               /* index into cp_mv_usage */
 
-extern int cp_mv_main(int argc, char **argv)
-{
-    __label__ name_too_long__exit;
-    __label__ exit_false;
 
-    int         recursiveFlag;
-    int         followLinks;
-    int         preserveFlag;
+static int recursiveFlag;
+static int followLinks;
+static int preserveFlag;
+static int forceFlag;
+
+static const char *baseSrcName;
+static int                srcDirFlag;
+static struct stat srcStatBuf;
+
+static char               *pBaseDestName;
+static size_t     baseDestLen;
+static int                destDirFlag;
+static struct stat destStatBuf;
 
-    const char *baseSrcName;
-    int         srcDirFlag;
+static jmp_buf      catch;
+static volatile int mv_Action_first_time;
 
-    char        baseDestName[PATH_MAX + 1];
-    size_t      baseDestLen;
-    int         destDirFlag;
+static void name_too_long__exit (void) __attribute__((noreturn));
+
+static
+void name_too_long__exit (void)
+{
+       error_msg_and_die(name_too_long);
+}
 
-    void fill_baseDest_buf(char *_buf, size_t *_buflen)
-       {
-           const char *srcBasename;
-           if ((srcBasename = strrchr(baseSrcName, '/')) == NULL)
-           {
+static void
+fill_baseDest_buf(char *_buf, size_t * _buflen) {
+       const char *srcBasename;
+       if ((srcBasename = strrchr(baseSrcName, '/')) == NULL) {
                srcBasename = baseSrcName;
-               if (_buf[*_buflen - 1] != '/')
-               {
-                   if (++(*_buflen) > PATH_MAX)
-                       goto name_too_long__exit;
-                   strcat(_buf, "/");
+               if (_buf[*_buflen - 1] != '/') {
+                       if (++(*_buflen) > BUFSIZ)
+                               name_too_long__exit();
+                       strcat(_buf, "/");
                }
-           }
-           if (*_buflen + strlen(srcBasename) > PATH_MAX)
-               goto name_too_long__exit;
-           strcat(_buf, srcBasename);
-           return;
        }
+       if (*_buflen + strlen(srcBasename) > BUFSIZ)
+               name_too_long__exit();
+       strcat(_buf, srcBasename);
+       return;
+       
+}
+
+static int
+cp_mv_Action(const char *fileName, struct stat *statbuf, void* junk)
+{
+       char            destName[BUFSIZ + 1];
+       size_t          destLen;
+       const char *srcBasename;
+       char       *name;
 
-    int fileAction(const char *fileName, struct stat *statbuf)
-       {
-           __label__   return_false;
-           char        destName[PATH_MAX + 1];
-           size_t      destLen;
-           const char *srcBasename;
-           
-           strcpy(destName, baseDestName);
-           destLen = strlen(destName);
-           
-           if (srcDirFlag == TRUE)
-           {
-               if (recursiveFlag == FALSE)
-               {
-                   fprintf(stderr, omitting_directory, "cp", baseSrcName);
-                   return TRUE;
+       strcpy(destName, pBaseDestName);
+       destLen = strlen(destName);
+
+       if (srcDirFlag == TRUE) {
+               if (recursiveFlag == FALSE) {
+                       error_msg(omitting_directory, baseSrcName);
+                       return TRUE;
                }
                srcBasename = (strstr(fileName, baseSrcName)
-                              + strlen(baseSrcName));
-               if (destLen + strlen(srcBasename) > PATH_MAX)
-               {
-                   fprintf(stderr, name_too_long, "cp");
-                   goto return_false;
+                                          + strlen(baseSrcName));
+
+               if (destLen + strlen(srcBasename) > BUFSIZ) {
+                       error_msg(name_too_long);
+                       return FALSE;
                }
                strcat(destName, srcBasename);
-           }
-           else if (destDirFlag == TRUE)
-           {
+       }
+       else if (destDirFlag == TRUE) {
                fill_baseDest_buf(&destName[0], &destLen);
-           }
-           else
-           {
+       }
+       else {
                srcBasename = baseSrcName;
-           }
-           return copyFile(fileName, destName, preserveFlag, followLinks);
-           
-       return_false:
-           return FALSE;
        }
+       if (mv_Action_first_time && (dz_i == is_mv)) {
+               mv_Action_first_time = errno = 0;
+               if (rename(fileName, destName) < 0 && errno != EXDEV) {
+                       perror_msg("rename(%s, %s)", fileName, destName);
+                       goto do_copyFile;       /* Try anyway... */
+               }
+               else if (errno == EXDEV)
+                       goto do_copyFile;
+               else
+                       longjmp(catch, 1);      /* succeeded with rename() */
+       }
+ do_copyFile:
+       if (preserveFlag == TRUE && statbuf->st_nlink > 1) {
+               if (is_in_ino_dev_hashtable(statbuf, &name)) {
+                       if (link(name, destName) < 0) {
+                               perror_msg("link(%s, %s)", name, destName);
+                               return FALSE;
+                       }
+                       return TRUE;
+               }
+               else {
+                       add_to_ino_dev_hashtable(statbuf, destName);
+               }
+       }
+       return copy_file(fileName, destName, preserveFlag, followLinks, forceFlag);
+}
 
-    int rmfileAction(const char *fileName, struct stat* statbuf)
-        {
-             if (unlink(fileName) < 0 ) {
-                  perror(fileName);
-                  return FALSE;
-             }
-             return TRUE;
-        }
-
-    int rmdirAction(const char *fileName, struct stat* statbuf)
-        {
-             if (rmdir(fileName) < 0 ) {
-                  perror(fileName);
-                  return FALSE;
-             }
-             return TRUE;
-        }
-
-    if ((dz = strrchr(*argv, '/')) == NULL) dz = *argv; else dz++;
-    if (*dz == 'c' && *(dz + 1) == 'p') dz_i = is_cp; else dz_i = is_mv;
-    if (argc < 3) usage(cp_mv_usage[dz_i]);
-    argc--;
-    argv++;
+static int
+rm_Action(const char *fileName, struct stat *statbuf, void* junk)
+{
+       int status = TRUE;
 
-    if (dz_i == is_cp)
-    {
-       recursiveFlag = preserveFlag = FALSE;
-       followLinks = TRUE;
-       while (**argv == '-')
-       {
-           while (*++(*argv))
-           {
-               switch (**argv)
-               {
-               case 'a':
-                   followLinks   = FALSE;
-                   preserveFlag  = TRUE;
-                   recursiveFlag = TRUE;
-                   break;
-               case 'd':
-                   followLinks   = FALSE;
-                   break;
-               case 'p':
-                   preserveFlag  = TRUE;
-                   break;
-               case 'R':
-                   recursiveFlag = TRUE;
-                   break;
-               default:
-                   usage(cp_mv_usage[is_cp]);
+       if (S_ISDIR(statbuf->st_mode)) {
+               if (rmdir(fileName) < 0) {
+                       perror_msg("rmdir(%s)", fileName);
+                       status = FALSE;
                }
-           }
-           argc--;
-           argv++;
+       } else if (unlink(fileName) < 0) {
+               perror_msg("unlink(%s)", fileName);
+               status = FALSE;
        }
-    }
-    else /* (dz_i == is_mv) */
-    {
-       recursiveFlag = preserveFlag = TRUE;
-       followLinks = FALSE;
-    }
+       return status;
+}
 
-    if (strlen(argv[argc - 1]) > PATH_MAX)
-    {
-       fprintf(stderr, name_too_long, "cp");
-       goto exit_false;
-    }
-    strcpy(baseDestName, argv[argc - 1]);
-    baseDestLen = strlen(baseDestName);
-    if (baseDestLen == 0) goto exit_false;
+extern int cp_mv_main(int argc, char **argv)
+{
+       volatile int i;
+       int c;
+       RESERVE_BB_BUFFER(baseDestName,BUFSIZ + 1);
+       pBaseDestName = baseDestName; /* available globally */
 
-    destDirFlag = isDirectory(baseDestName, TRUE);
-    if ((argc > 3) && destDirFlag == FALSE)
-    {
-       fprintf(stderr, not_a_directory, "cp", baseDestName);
-       goto exit_false;
-    }
+       if (*applet_name == 'c' && *(applet_name + 1) == 'p')
+               dz_i = is_cp;
+       else
+               dz_i = is_mv;
+       if (argc < 3)
+               show_usage();
 
-    while (argc-- > 1)
-    {
-       size_t srcLen;
-       int    flags_memo;
+       if (dz_i == is_cp) {
+               recursiveFlag = preserveFlag = forceFlag = FALSE;
+               followLinks = TRUE;
+               while ((c = getopt(argc, argv, "adpRf")) != EOF) {
+                               switch (c) {
+                               case 'a':
+                                       followLinks = FALSE;
+                                       preserveFlag = TRUE;
+                                       recursiveFlag = TRUE;
+                                       break;
+                               case 'd':
+                                       followLinks = FALSE;
+                                       break;
+                               case 'p':
+                                       preserveFlag = TRUE;
+                                       break;
+                               case 'R':
+                                       recursiveFlag = TRUE;
+                                       break;
+                               case 'f':
+                                       forceFlag = TRUE;
+                                       break;
+                               default:
+                                       show_usage();
+                               }
+               }
+               if ((argc - optind) < 2) {
+                       show_usage();
+               }
+       } else {                                        /* (dz_i == is_mv) */
+               /* Initialize optind to 1, since in libc5 optind
+                * is not initialized until getopt() is called
+                * (or until sneaky programmers force it...). */
+               optind = 1;
+               recursiveFlag = preserveFlag = TRUE;
+               followLinks = FALSE;
+       }
+       
 
-       baseSrcName = *(argv++);
+       if (strlen(argv[argc - 1]) > BUFSIZ) {
+               error_msg(name_too_long);
+               goto exit_false;
+       }
+       strcpy(baseDestName, argv[argc - 1]);
+       baseDestLen = strlen(baseDestName);
+       if (baseDestLen == 0)
+               goto exit_false;
 
-       if ((srcLen = strlen(baseSrcName)) > PATH_MAX)
-           goto name_too_long__exit;
+       destDirFlag = is_directory(baseDestName, TRUE, &destStatBuf);
+       if (argc - optind > 2 && destDirFlag == FALSE) {
+               error_msg(not_a_directory, baseDestName);
+               goto exit_false;
+       }
 
-       if (srcLen == 0) continue;
+       for (i = optind; i < (argc-1); i++) {
+               size_t srcLen;
+               volatile int flags_memo;
+               int        status;
 
-       srcDirFlag = isDirectory(baseSrcName, followLinks);
+               baseSrcName=argv[i];
 
-       if ((flags_memo = (recursiveFlag == TRUE &&
-                          srcDirFlag == TRUE && destDirFlag == TRUE)))
-       {
-           fill_baseDest_buf(&baseDestName[0], &baseDestLen);
-       }
-       if (recursiveAction(baseSrcName,
-                           recursiveFlag, followLinks, FALSE,
-                           fileAction, fileAction)
-           == FALSE) goto exit_false;
+               if ((srcLen = strlen(baseSrcName)) > BUFSIZ)
+                       name_too_long__exit();
+
+               if (srcLen == 0) continue; /* "" */
 
-       if (dz_i == is_mv &&
-           recursiveAction(baseSrcName,
-                           recursiveFlag, followLinks, TRUE,
-                           rmfileAction, rmdirAction)
-           == FALSE) goto exit_false;
+               srcDirFlag = is_directory(baseSrcName, followLinks, &srcStatBuf);
 
-       if (flags_memo) *(baseDestName + baseDestLen) = '\0';
-    }
+               if ((flags_memo = (recursiveFlag == TRUE &&
+                                                  srcDirFlag == TRUE && destDirFlag == TRUE))) {
 
-    exit TRUE;
+                       struct stat sb;
+                       int                     state = 0;
+                       char            *pushd, *d, *p;
 
- name_too_long__exit:
-    fprintf(stderr, name_too_long, "cp");
+                       if ((pushd = getcwd(NULL, BUFSIZ + 1)) == NULL) {
+                               perror_msg("getcwd()");
+                               continue;
+                       }
+                       if (chdir(baseDestName) < 0) {
+                               perror_msg("chdir(%s)", baseSrcName);
+                               continue;
+                       }
+                       if ((d = getcwd(NULL, BUFSIZ + 1)) == NULL) {
+                               perror_msg("getcwd()");
+                               continue;
+                       }
+                       while (!state && *d != '\0') {
+                               if (stat(d, &sb) < 0) { /* stat not lstat - always dereference targets */
+                                       perror_msg("stat(%s)", d);
+                                       state = -1;
+                                       continue;
+                               }
+                               if ((sb.st_ino == srcStatBuf.st_ino) &&
+                                       (sb.st_dev == srcStatBuf.st_dev)) {
+                                       error_msg("Cannot %s `%s' into a subdirectory of itself, "
+                                                       "`%s/%s'", applet_name, baseSrcName,
+                                                       baseDestName, baseSrcName);
+                                       state = -1;
+                                       continue;
+                               }
+                               if ((p = strrchr(d, '/')) != NULL) {
+                                       *p = '\0';
+                               }
+                       }
+                       if (chdir(pushd) < 0) {
+                               perror_msg("chdir(%s)", pushd);
+                               free(pushd);
+                               free(d);
+                               continue;
+                       }
+                       free(pushd);
+                       free(d);
+                       if (state < 0)
+                               continue;
+                       else
+                               fill_baseDest_buf(baseDestName, &baseDestLen);
+               }
+               status = setjmp(catch);
+               if (status == 0) {
+                       mv_Action_first_time = 1;
+                       if (recursive_action(baseSrcName,
+                                                               recursiveFlag, followLinks, FALSE,
+                                                               cp_mv_Action, cp_mv_Action, NULL) == FALSE) goto exit_false;
+                       if (dz_i == is_mv &&
+                               recursive_action(baseSrcName,
+                                                               recursiveFlag, followLinks, TRUE,
+                                                               rm_Action, rm_Action, NULL) == FALSE) goto exit_false;
+               }               
+               if (flags_memo)
+                       *(baseDestName + baseDestLen) = '\0';
+       }
+       return EXIT_SUCCESS;
  exit_false:
-    exit FALSE;
+       return EXIT_FAILURE;
 }
 
-// Local Variables:
-// c-file-style: "k&r"
-// c-basic-offset: 4
-// End:
+/*
+Local Variables:
+c-file-style: "linux"
+c-basic-offset: 4
+tab-width: 4
+End:
+*/