claenups for previous commit
[oweals/busybox.git] / coreutils / mv.c
index ae0ee92e49d345773f40300fb44372bb30f03cae..6e11197a1d35a926ce7e8a747220606aadfca8ef 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.
  */
+//config:config MV
+//config:      bool "mv (9.8 kb)"
+//config:      default y
+//config:      help
+//config:      mv is used to move or rename files or directories.
 
-#include <sys/types.h>
-#include <sys/stat.h>
-#include <unistd.h>
-#include <dirent.h>
-#include <errno.h>
-#include <stdlib.h>
-#include "busybox.h"
-#include "libcoreutils/coreutils.h"
+//applet:IF_MV(APPLET_NOEXEC(mv, mv, BB_DIR_BIN, BB_SUID_DROP, mv))
+/* NOEXEC despite cases when it can be a "runner" (mv LARGE_DIR OTHER_FS) */
+
+//kbuild:lib-$(CONFIG_MV) += mv.o
+
+//usage:#define mv_trivial_usage
+//usage:       "[-fin] SOURCE DEST\n"
+//usage:       "or: mv [-fin] SOURCE... DIRECTORY"
+//usage:#define mv_full_usage "\n\n"
+//usage:       "Rename SOURCE to DEST, or move SOURCE(s) to DIRECTORY\n"
+//usage:     "\n       -f      Don't prompt before overwriting"
+//usage:     "\n       -i      Interactive, prompt before overwrite"
+//usage:     "\n       -n      Don't overwrite an existing file"
+//usage:
+//usage:#define mv_example_usage
+//usage:       "$ mv /tmp/foo /bin/bar\n"
 
-static const char *fmt = "cannot overwrite %sdirectory with %sdirectory";
+#include "libbb.h"
+#include "libcoreutils/coreutils.h"
 
-extern int mv_main(int argc, char **argv)
+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;
        const char *last;
        const char *dest;
+       unsigned flags;
        int dest_exists;
-       int source_exists;
-       int opt;
-       int flags = 0;
        int status = 0;
+       int copy_flag = 0;
 
-       while ((opt = getopt(argc, argv, "fi")) > 0) {
-               flags &= ~(FILEUTILS_INTERACTIVE | FILEUTILS_FORCE);
-               if (opt == 'i') {
-                       flags |= FILEUTILS_INTERACTIVE;
-               } else if (opt == 'f') {
-                       flags |= FILEUTILS_FORCE;
-               } else {
-                       bb_show_usage();
-               }
-       }
-
-       if (optind + 2 > argc)
-               bb_show_usage();
-
-       last = argv[argc - 1];
+#define OPT_FORCE       (1 << 0)
+#define OPT_INTERACTIVE (1 << 1)
+#define OPT_NOCLOBBER   (1 << 2)
+#define OPT_VERBOSE     ((1 << 3) * ENABLE_FEATURE_VERBOSE)
+       /* Need at least two arguments.
+        * If more than one of -f, -i, -n is specified , only the final one
+        * takes effect (it unsets previous options).
+        */
+       flags = getopt32long(argv, "^"
+                       "finv"
+                       "\0"
+                       "-2:f-in:i-fn:n-fi",
+                       "interactive\0" No_argument "i"
+                       "force\0"       No_argument "f"
+                       "no-clobber\0"  No_argument "n"
+                       IF_FEATURE_VERBOSE(
+                       "verbose\0"     No_argument "v"
+                       )
+       );
+       argc -= optind;
        argv += optind;
+       last = argv[argc - 1];
 
-       if (optind + 2 == argc) {
-               if ((dest_exists = cp_mv_stat(last, &dest_stat)) < 0) {
-                       return 1;
+       if (argc == 2) {
+               dest_exists = cp_mv_stat(last, &dest_stat);
+               if (dest_exists < 0) {
+                       return EXIT_FAILURE;
                }
 
-               if (!(dest_exists & 2)) {
+               if (!(dest_exists & 2)) { /* last is not a directory */
                        dest = last;
                        goto DO_MOVE;
                }
        }
-       
-       do {
-               dest = concat_path_file(last,
-                                                               bb_get_last_path_component(*argv));
 
-               if ((dest_exists = cp_mv_stat(dest, &dest_stat)) < 0) {
+       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;
                }
 
-       DO_MOVE:
-               
-               if (dest_exists && !(flags & FILEUTILS_FORCE) &&
-                       ((access(dest, W_OK) < 0 && isatty(0)) ||
-                        (flags & FILEUTILS_INTERACTIVE))) {
-                                if (fprintf(stderr, "mv: overwrite `%s'? ", dest) < 0) {
-                                        goto RET_1;    /* Ouch! fprintf failed! */
-                                }
-                                if (!bb_ask_confirmation())
-                                        goto RET_0;
-                }
-               
+ DO_MOVE:
+               if (dest_exists) {
+                       if (flags & OPT_NOCLOBBER)
+                               goto RET_0;
+                       if (!(flags & OPT_FORCE)
+                        && ((access(dest, W_OK) < 0 && isatty(0))
+                           || (flags & OPT_INTERACTIVE))
+                       ) {
+                               if (fprintf(stderr, "mv: overwrite '%s'? ", dest) < 0) {
+                                       goto RET_1;  /* Ouch! fprintf failed! */
+                               }
+                               if (!bb_ask_y_confirmation()) {
+                                       goto RET_0;
+                               }
+                       }
+               }
+
                if (rename(*argv, dest) < 0) {
-                       if (errno != EXDEV) {
-                               bb_perror_msg("unable to rename `%s'", *argv);
-                       } else if ((source_exists = cp_mv_stat(*argv, &source_stat)) >= 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 & 2) {
-                                               if (!(source_exists & 2)) {
+                                       if (dest_exists == 3) {
+                                               if (source_exists != 3) {
                                                        bb_error_msg(fmt, "", "non-");
                                                        goto RET_1;
                                                }
                                        } else {
-                                               if (source_exists & 2) {
+                                               if (source_exists == 3) {
                                                        bb_error_msg(fmt, "non-", "");
                                                        goto RET_1;
                                                }
                                        }
                                        if (unlink(dest) < 0) {
-                                               bb_perror_msg("cannot remove `%s'", dest);
+                                               bb_perror_msg("can't remove '%s'", dest);
                                                goto RET_1;
                                        }
                                }
-                               
-                               if ((copy_file(*argv, dest,
-                                                          FILEUTILS_RECUR | FILEUTILS_PRESERVE_STATUS) >= 0)
-                                       && (remove_file(*argv, FILEUTILS_RECUR | FILEUTILS_FORCE) >= 0)
-                                       ) {
+                               /* 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:
+ RET_1:
                        status = 1;
                }
-
-       RET_0:
+ RET_0:
+               if (flags & OPT_VERBOSE) {
+                       printf("'%s' -> '%s'\n", *argv, dest);
+               }
                if (dest != last) {
                        free((void *) dest);
                }
-               
        } while (*++argv != last);
-       
-       exit(status);
+
+       return status;
 }