busybox/coreutils/mv.c
<<
>>
Prefs
   1/* vi: set sw=4 ts=4: */
   2/*
   3 * Mini mv implementation for busybox
   4 *
   5 * Copyright (C) 2000 by Matt Kraai <kraai@alumni.carnegiemellon.edu>
   6 * SELinux support by Yuichi Nakamura <ynakam@hitachisoft.jp>
   7 *
   8 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
   9 */
  10
  11/* Mar 16, 2003      Manuel Novoa III   (mjn3@codepoet.org)
  12 *
  13 * Size reduction and improved error checking.
  14 */
  15
  16#include "libbb.h"
  17#include "libcoreutils/coreutils.h"
  18
  19//usage:#define mv_trivial_usage
  20//usage:       "[-fin] SOURCE DEST\n"
  21//usage:       "or: mv [-fin] SOURCE... DIRECTORY"
  22//usage:#define mv_full_usage "\n\n"
  23//usage:       "Rename SOURCE to DEST, or move SOURCE(s) to DIRECTORY\n"
  24//usage:     "\n        -f      Don't prompt before overwriting"
  25//usage:     "\n        -i      Interactive, prompt before overwrite"
  26//usage:     "\n        -n      Don't overwrite an existing file"
  27//usage:
  28//usage:#define mv_example_usage
  29//usage:       "$ mv /tmp/foo /bin/bar\n"
  30
  31#if ENABLE_FEATURE_MV_LONG_OPTIONS
  32static const char mv_longopts[] ALIGN1 =
  33        "interactive\0" No_argument "i"
  34        "force\0"       No_argument "f"
  35        "no-clobber\0"  No_argument "n"
  36        ;
  37#endif
  38
  39#define OPT_FILEUTILS_FORCE       1
  40#define OPT_FILEUTILS_INTERACTIVE 2
  41#define OPT_FILEUTILS_NOCLOBBER   4
  42
  43int mv_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  44int mv_main(int argc, char **argv)
  45{
  46        struct stat dest_stat;
  47        const char *last;
  48        const char *dest;
  49        unsigned flags;
  50        int dest_exists;
  51        int status = 0;
  52        int copy_flag = 0;
  53
  54#if ENABLE_FEATURE_MV_LONG_OPTIONS
  55        applet_long_options = mv_longopts;
  56#endif
  57        /* Need at least two arguments.
  58         * If more than one of -f, -i, -n is specified , only the final one
  59         * takes effect (it unsets previous options). */
  60        opt_complementary = "-2:f-in:i-fn:n-fi";
  61        flags = getopt32(argv, "fin");
  62        argc -= optind;
  63        argv += optind;
  64        last = argv[argc - 1];
  65
  66        if (argc == 2) {
  67                dest_exists = cp_mv_stat(last, &dest_stat);
  68                if (dest_exists < 0) {
  69                        return EXIT_FAILURE;
  70                }
  71
  72                if (!(dest_exists & 2)) { /* last is not a directory */
  73                        dest = last;
  74                        goto DO_MOVE;
  75                }
  76        }
  77
  78        do {
  79                dest = concat_path_file(last, bb_get_last_path_component_strip(*argv));
  80                dest_exists = cp_mv_stat(dest, &dest_stat);
  81                if (dest_exists < 0) {
  82                        goto RET_1;
  83                }
  84
  85 DO_MOVE:
  86                if (dest_exists) {
  87                        if (flags & OPT_FILEUTILS_NOCLOBBER)
  88                                goto RET_0;
  89                        if (!(flags & OPT_FILEUTILS_FORCE)
  90                         && ((access(dest, W_OK) < 0 && isatty(0))
  91                            || (flags & OPT_FILEUTILS_INTERACTIVE))
  92                        ) {
  93                                if (fprintf(stderr, "mv: overwrite '%s'? ", dest) < 0) {
  94                                        goto RET_1;  /* Ouch! fprintf failed! */
  95                                }
  96                                if (!bb_ask_confirmation()) {
  97                                        goto RET_0;
  98                                }
  99                        }
 100                }
 101
 102                if (rename(*argv, dest) < 0) {
 103                        struct stat source_stat;
 104                        int source_exists;
 105
 106                        if (errno != EXDEV
 107                         || (source_exists = cp_mv_stat2(*argv, &source_stat, lstat)) < 1
 108                        ) {
 109                                bb_perror_msg("can't rename '%s'", *argv);
 110                        } else {
 111                                static const char fmt[] ALIGN1 =
 112                                        "can't overwrite %sdirectory with %sdirectory";
 113
 114                                if (dest_exists) {
 115                                        if (dest_exists == 3) {
 116                                                if (source_exists != 3) {
 117                                                        bb_error_msg(fmt, "", "non-");
 118                                                        goto RET_1;
 119                                                }
 120                                        } else {
 121                                                if (source_exists == 3) {
 122                                                        bb_error_msg(fmt, "non-", "");
 123                                                        goto RET_1;
 124                                                }
 125                                        }
 126                                        if (unlink(dest) < 0) {
 127                                                bb_perror_msg("can't remove '%s'", dest);
 128                                                goto RET_1;
 129                                        }
 130                                }
 131                                /* FILEUTILS_RECUR also prevents nasties like
 132                                 * "read from device and write contents to dst"
 133                                 * instead of "create same device node" */
 134                                copy_flag = FILEUTILS_RECUR | FILEUTILS_PRESERVE_STATUS;
 135#if ENABLE_SELINUX
 136                                copy_flag |= FILEUTILS_PRESERVE_SECURITY_CONTEXT;
 137#endif
 138                                if ((copy_file(*argv, dest, copy_flag) >= 0)
 139                                 && (remove_file(*argv, FILEUTILS_RECUR | FILEUTILS_FORCE) >= 0)
 140                                ) {
 141                                        goto RET_0;
 142                                }
 143                        }
 144 RET_1:
 145                        status = 1;
 146                }
 147 RET_0:
 148                if (dest != last) {
 149                        free((void *) dest);
 150                }
 151        } while (*++argv != last);
 152
 153        return status;
 154}
 155