lineedit: do not hang on error, but return error indicator.
[oweals/busybox.git] / coreutils / mv.c
index b890abf6efcc89c402b3ea4df40469742e8437ec..7f49d5babd5f043c8d2ce140d797a09bed2d5f6e 100644 (file)
 /*
  * Mini mv implementation for busybox
  *
- *
  * Copyright (C) 2000 by Matt Kraai <kraai@alumni.carnegiemellon.edu>
+ * SELinux support by Yuichi Nakamura <ynakam@hitachisoft.jp>
  *
- * 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
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
+ */
+
+/* Mar 16, 2003      Manuel Novoa III   (mjn3@codepoet.org)
  *
+ * Size reduction and improved error checking.
  */
 
-#include <sys/types.h>
-#include <sys/stat.h>
-#include <unistd.h>
-#include <dirent.h>
-#include <errno.h>
-#include <stdlib.h>
+#include "libbb.h"
+#include "libcoreutils/coreutils.h"
 
-#include "busybox.h"
+#if ENABLE_FEATURE_MV_LONG_OPTIONS
+static const char mv_longopts[] ALIGN1 =
+       "interactive\0" No_argument "i"
+       "force\0"       No_argument "f"
+       ;
+#endif
 
-static int flags;
+#define OPT_FILEUTILS_FORCE       1
+#define OPT_FILEUTILS_INTERACTIVE 2
 
-static int manual_rename(const char *source, const char *dest)
+int mv_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int mv_main(int argc, char **argv)
 {
-       struct stat source_stat;
        struct stat dest_stat;
-       int source_exists = 1;
-       int dest_exists = 1;
-
-       if (stat(source, &source_stat) < 0) {
-               if (errno != ENOENT) {
-                       perror_msg("unable to stat `%s'", source);
-                       return -1;
+       const char *last;
+       const char *dest;
+       unsigned flags;
+       int dest_exists;
+       int status = 0;
+       int copy_flag = 0;
+
+#if ENABLE_FEATURE_MV_LONG_OPTIONS
+       applet_long_options = mv_longopts;
+#endif
+       // Need at least two arguments
+       // -f unsets -i, -i unsets -f
+       opt_complementary = "-2:f-i:i-f";
+       flags = getopt32(argv, "fi");
+       argc -= optind;
+       argv += optind;
+       last = argv[argc - 1];
+
+       if (argc == 2) {
+               dest_exists = cp_mv_stat(last, &dest_stat);
+               if (dest_exists < 0) {
+                       return EXIT_FAILURE;
                }
-               source_exists = 0;
-       }
 
-       if (stat(dest, &dest_stat) < 0) {
-               if (errno != ENOENT) {
-                       perror_msg("unable to stat `%s'", dest);
-                       return -1;
+               if (!(dest_exists & 2)) { /* last is not a directory */
+                       dest = last;
+                       goto DO_MOVE;
                }
-               dest_exists = 0;
        }
 
-       if (dest_exists) {
-               if (S_ISDIR(dest_stat.st_mode) &&
-                         (!source_exists || !S_ISDIR(source_stat.st_mode))) {
-                       error_msg("cannot overwrite directory with non-directory");
-                       return -1;
+       do {
+               dest = concat_path_file(last, bb_get_last_path_component_strip(*argv));
+               dest_exists = cp_mv_stat(dest, &dest_stat);
+               if (dest_exists < 0) {
+                       goto RET_1;
                }
 
-               if (!S_ISDIR(dest_stat.st_mode) && source_exists &&
-                               S_ISDIR(source_stat.st_mode)) {
-                       error_msg("cannot overwrite non-directory with directory");
-                       return -1;
+ DO_MOVE:
+               if (dest_exists
+                && !(flags & OPT_FILEUTILS_FORCE)
+                && ((access(dest, W_OK) < 0 && isatty(0))
+                   || (flags & OPT_FILEUTILS_INTERACTIVE))
+               ) {
+                       if (fprintf(stderr, "mv: overwrite '%s'? ", dest) < 0) {
+                               goto RET_1;     /* Ouch! fprintf failed! */
+                       }
+                       if (!bb_ask_confirmation()) {
+                               goto RET_0;
+                       }
                }
-
-               if (unlink(dest) < 0) {
-                       perror_msg("cannot remove `%s'", dest);
-                       return -1;
-               }
-       }
-
-       if (copy_file(source, dest, FILEUTILS_RECUR | FILEUTILS_PRESERVE_STATUS |
-                       FILEUTILS_PRESERVE_SYMLINKS) < 0)
-               return -1;
-
-       if (remove_file(source, FILEUTILS_RECUR | FILEUTILS_FORCE) < 0)
-               return -1;
-
-       return 0;
-}
-
-static int move_file(const char *source, const char *dest)
-{
-       struct stat dest_stat;
-       int dest_exists = 1;
-
-       if (stat(dest, &dest_stat) < 0) {
-               if (errno != ENOENT) {
-                       perror_msg("unable to stat `%s'", dest);
-                       return -1;
-               }
-               dest_exists = 0;
-       }
-
-       if (dest_exists && !(flags & FILEUTILS_FORCE) &&
-                       ((access(dest, W_OK) < 0 && isatty(0)) ||
-                        (flags & FILEUTILS_INTERACTIVE))) {
-               fprintf(stderr, "mv: overwrite `%s'? ", dest);
-               if (!ask_confirmation())
-                       return 0;
-       }
-
-       if (rename(source, dest) < 0) {
-               if (errno == EXDEV)
-                       return manual_rename(source, dest);
-
-               perror_msg("unable to rename `%s'", source);
-               return -1;
-       }
-       
-       return 0;
-}
-
-extern int mv_main(int argc, char **argv)
-{
-       int status = 0;
-       int opt;
-       int i;
-
-       while ((opt = getopt(argc, argv, "fi")) != -1)
-               switch (opt) {
-               case 'f':
-                       flags &= ~FILEUTILS_INTERACTIVE;
-                       flags |= FILEUTILS_FORCE;
-                       break;
-               case 'i':
-                       flags &= ~FILEUTILS_FORCE;
-                       flags |= FILEUTILS_INTERACTIVE;
-                       break;
-               default:
-                       show_usage();
-               }
-
-       if (optind + 2 > argc)
-               show_usage();
-
-       if (optind + 2 == argc) {
-               struct stat dest_stat;
-               int dest_exists = 1;
-
-               if (stat(argv[optind + 1], &dest_stat) < 0) {
-                       if (errno != ENOENT)
-                               perror_msg_and_die("unable to stat `%s'", argv[optind + 1]);
-                       dest_exists = 0;
+               if (rename(*argv, dest) < 0) {
+                       struct stat source_stat;
+                       int source_exists;
+
+                       if (errno != EXDEV
+                        || (source_exists = cp_mv_stat2(*argv, &source_stat, lstat)) < 1
+                       ) {
+                               bb_perror_msg("can't rename '%s'", *argv);
+                       } else {
+                               static const char fmt[] ALIGN1 =
+                                       "can't overwrite %sdirectory with %sdirectory";
+
+                               if (dest_exists) {
+                                       if (dest_exists == 3) {
+                                               if (source_exists != 3) {
+                                                       bb_error_msg(fmt, "", "non-");
+                                                       goto RET_1;
+                                               }
+                                       } else {
+                                               if (source_exists == 3) {
+                                                       bb_error_msg(fmt, "non-", "");
+                                                       goto RET_1;
+                                               }
+                                       }
+                                       if (unlink(dest) < 0) {
+                                               bb_perror_msg("can't remove '%s'", dest);
+                                               goto RET_1;
+                                       }
+                               }
+                               /* FILEUTILS_RECUR also prevents nasties like
+                                * "read from device and write contents to dst"
+                                * instead of "create same device node" */
+                               copy_flag = FILEUTILS_RECUR | FILEUTILS_PRESERVE_STATUS;
+#if ENABLE_SELINUX
+                               copy_flag |= FILEUTILS_PRESERVE_SECURITY_CONTEXT;
+#endif
+                               if ((copy_file(*argv, dest, copy_flag) >= 0)
+                                && (remove_file(*argv, FILEUTILS_RECUR | FILEUTILS_FORCE) >= 0)
+                               ) {
+                                       goto RET_0;
+                               }
+                       }
+ RET_1:
+                       status = 1;
                }
-
-               if (!dest_exists || !S_ISDIR(dest_stat.st_mode)) {
-                       if (move_file(argv[optind], argv[optind + 1]) < 0)
-                               status = 1;
-                       return status;
+ RET_0:
+               if (dest != last) {
+                       free((void *) dest);
                }
-       }
-
-       for (i = optind; i < argc - 1; i++) {
-               char *dest = concat_path_file(argv[argc - 1],
-                               get_last_path_component(argv[i]));
-               if (move_file(argv[i], dest) < 0)
-                       status = 1;
-               free(dest);
-       }
+       } while (*++argv != last);
 
        return status;
 }