* 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 "busybox.h"
#define BB_DECLARE_EXTERN
#define bb_need_name_too_long
#define bb_need_omitting_directory
#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>
-#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)
+
+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;
+
+static jmp_buf catch;
+static volatile int mv_Action_first_time;
+
+static void name_too_long__exit (void) __attribute__((noreturn));
+
+static
+void name_too_long__exit (void)
{
- __label__ name_too_long__exit;
- __label__ exit_false;
-
- int recursiveFlag;
- int followLinks;
- int preserveFlag;
-
- const char *baseSrcName;
- int srcDirFlag;
- struct stat srcStatBuf;
-
- char baseDestName[PATH_MAX + 1];
- size_t baseDestLen;
- int destDirFlag;
- struct stat destStatBuf;
-
- 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, "/");
- }
+ error_msg_and_die(name_too_long);
+}
+
+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) > 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;
+
+}
- int fileAction(const char *fileName, struct stat *statbuf) {
- char destName[PATH_MAX + 1];
- size_t destLen;
- const char *srcBasename;
+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;
- strcpy(destName, baseDestName);
- destLen = strlen(destName);
+ strcpy(destName, pBaseDestName);
+ destLen = strlen(destName);
- if (srcDirFlag == TRUE) {
- if (recursiveFlag == FALSE) {
- fprintf(stderr, omitting_directory, "cp", baseSrcName);
- return TRUE;
- }
- srcBasename = (strstr(fileName, baseSrcName)
- + strlen(baseSrcName));
+ 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");
+ if (destLen + strlen(srcBasename) > BUFSIZ) {
+ error_msg(name_too_long);
+ return FALSE;
+ }
+ strcat(destName, srcBasename);
+ }
+ else if (destDirFlag == TRUE) {
+ fill_baseDest_buf(&destName[0], &destLen);
+ }
+ else {
+ srcBasename = baseSrcName;
+ }
+ 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;
}
- strcat(destName, srcBasename);
- } else if (destDirFlag == TRUE) {
- fill_baseDest_buf(&destName[0], &destLen);
- } else {
- srcBasename = baseSrcName;
+ return TRUE;
}
- return copyFile(fileName, destName, preserveFlag, followLinks);
- }
-
- int rmfileAction(const char *fileName, struct stat *statbuf) {
- if (unlink(fileName) < 0) {
- perror(fileName);
- return FALSE;
+ else {
+ add_to_ino_dev_hashtable(statbuf, destName);
}
- return TRUE;
}
+ return copy_file(fileName, destName, preserveFlag, followLinks, forceFlag);
+}
- int rmdirAction(const char *fileName, struct stat *statbuf) {
+static int
+rm_Action(const char *fileName, struct stat *statbuf, void* junk)
+{
+ int status = TRUE;
+
+ if (S_ISDIR(statbuf->st_mode)) {
if (rmdir(fileName) < 0) {
- perror(fileName);
- return FALSE;
+ perror_msg("rmdir(%s)", fileName);
+ status = FALSE;
}
- return TRUE;
+ } else if (unlink(fileName) < 0) {
+ perror_msg("unlink(%s)", fileName);
+ status = FALSE;
}
+ return status;
+}
- if ((dz = strrchr(*argv, '/')) == NULL)
- dz = *argv;
- else
- dz++;
- if (*dz == 'c' && *(dz + 1) == 'p')
+extern int cp_mv_main(int argc, char **argv)
+{
+ volatile int i;
+ int c;
+ RESERVE_BB_BUFFER(baseDestName,BUFSIZ + 1);
+ pBaseDestName = baseDestName; /* available globally */
+
+ if (*applet_name == 'c' && *(applet_name + 1) == 'p')
dz_i = is_cp;
else
dz_i = is_mv;
if (argc < 3)
- usage(cp_mv_usage[dz_i]);
- argc--;
- argv++;
+ show_usage();
if (dz_i == is_cp) {
- recursiveFlag = preserveFlag = FALSE;
+ recursiveFlag = preserveFlag = forceFlag = FALSE;
followLinks = TRUE;
- while (**argv == '-') {
- while (*++(*argv)) {
- switch (**argv) {
+ while ((c = getopt(argc, argv, "adpRf")) != EOF) {
+ switch (c) {
case 'a':
followLinks = FALSE;
preserveFlag = TRUE;
case 'R':
recursiveFlag = TRUE;
break;
+ case 'f':
+ forceFlag = TRUE;
+ break;
default:
- usage(cp_mv_usage[is_cp]);
+ show_usage();
}
- }
- argc--;
- argv++;
+ }
+ 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;
}
+
- if (strlen(argv[argc - 1]) > PATH_MAX) {
- fprintf(stderr, name_too_long, "cp");
+ if (strlen(argv[argc - 1]) > BUFSIZ) {
+ error_msg(name_too_long);
goto exit_false;
}
strcpy(baseDestName, argv[argc - 1]);
if (baseDestLen == 0)
goto exit_false;
- destDirFlag = isDirectory(baseDestName, TRUE, &destStatBuf);
- if ((argc > 3) && destDirFlag == FALSE) {
- fprintf(stderr, not_a_directory, "cp", baseDestName);
+ destDirFlag = is_directory(baseDestName, TRUE, &destStatBuf);
+ if (argc - optind > 2 && destDirFlag == FALSE) {
+ error_msg(not_a_directory, baseDestName);
goto exit_false;
}
- while (argc-- > 1) {
+ for (i = optind; i < (argc-1); i++) {
size_t srcLen;
- int flags_memo;
+ volatile int flags_memo;
+ int status;
- baseSrcName = *(argv++);
+ baseSrcName=argv[i];
- if ((srcLen = strlen(baseSrcName)) > PATH_MAX)
- goto name_too_long__exit;
+ if ((srcLen = strlen(baseSrcName)) > BUFSIZ)
+ name_too_long__exit();
- if (srcLen == 0)
- continue;
+ if (srcLen == 0) continue; /* "" */
- srcDirFlag = isDirectory(baseSrcName, followLinks, &srcStatBuf);
+ srcDirFlag = is_directory(baseSrcName, followLinks, &srcStatBuf);
if ((flags_memo = (recursiveFlag == TRUE &&
srcDirFlag == TRUE && destDirFlag == TRUE))) {
- if ((destStatBuf.st_ino == srcStatBuf.st_ino) &&
- (destStatBuf.st_rdev == srcStatBuf.st_rdev)) {
- fprintf(stderr,
- "%s: Cannot %s `%s' into a subdirectory of itself, `%s/%s'\n",
- dz, dz, baseSrcName, baseDestName, baseSrcName);
- continue;
+
+ struct stat sb;
+ int state = 0;
+ char *pushd, *d, *p;
+
+ 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);
}
- if (recursiveAction(baseSrcName,
- recursiveFlag, followLinks, FALSE,
- fileAction, fileAction) == FALSE)
- goto exit_false;
-
- if (dz_i == is_mv &&
- recursiveAction(baseSrcName,
- recursiveFlag, followLinks, TRUE,
- rmfileAction, rmdirAction) == FALSE)
- goto exit_false;
-
+ 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';
}
-
- exit TRUE;
-
- name_too_long__exit:
- fprintf(stderr, name_too_long, "cp");
- exit_false:
- exit FALSE;
+ return EXIT_SUCCESS;
+ exit_false:
+ return EXIT_FAILURE;
}
-// Local Variables:
-// c-file-style: "linux"
-// tab-width: 4
-// End:
+/*
+Local Variables:
+c-file-style: "linux"
+c-basic-offset: 4
+tab-width: 4
+End:
+*/