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 tarball for details.
   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#if ENABLE_FEATURE_MV_LONG_OPTIONS
  20static const char mv_longopts[] ALIGN1 =
  21        "interactive\0" No_argument "i"
  22        "force\0"       No_argument "f"
  23        ;
  24#endif
  25
  26#define OPT_FILEUTILS_FORCE       1
  27#define OPT_FILEUTILS_INTERACTIVE 2
  28
  29int mv_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  30int mv_main(int argc, char **argv)
  31{
  32        struct stat dest_stat;
  33        const char *last;
  34        const char *dest;
  35        unsigned flags;
  36        int dest_exists;
  37        int status = 0;
  38        int copy_flag = 0;
  39
  40#if ENABLE_FEATURE_MV_LONG_OPTIONS
  41        applet_long_options = mv_longopts;
  42#endif
  43        // Need at least two arguments
  44        // -f unsets -i, -i unsets -f
  45        opt_complementary = "-2:f-i:i-f";
  46        flags = getopt32(argv, "fi");
  47        argc -= optind;
  48        argv += optind;
  49        last = argv[argc - 1];
  50
  51        if (argc == 2) {
  52                dest_exists = cp_mv_stat(last, &dest_stat);
  53                if (dest_exists < 0) {
  54                        return EXIT_FAILURE;
  55                }
  56
  57                if (!(dest_exists & 2)) { /* last is not a directory */
  58                        dest = last;
  59                        goto DO_MOVE;
  60                }
  61        }
  62
  63        do {
  64                dest = concat_path_file(last, bb_get_last_path_component_strip(*argv));
  65                dest_exists = cp_mv_stat(dest, &dest_stat);
  66                if (dest_exists < 0) {
  67                        goto RET_1;
  68                }
  69
  70 DO_MOVE:
  71                if (dest_exists
  72                 && !(flags & OPT_FILEUTILS_FORCE)
  73                 && ((access(dest, W_OK) < 0 && isatty(0))
  74                    || (flags & OPT_FILEUTILS_INTERACTIVE))
  75                ) {
  76                        if (fprintf(stderr, "mv: overwrite '%s'? ", dest) < 0) {
  77                                goto RET_1;     /* Ouch! fprintf failed! */
  78                        }
  79                        if (!bb_ask_confirmation()) {
  80                                goto RET_0;
  81                        }
  82                }
  83                if (rename(*argv, dest) < 0) {
  84                        struct stat source_stat;
  85                        int source_exists;
  86
  87                        if (errno != EXDEV
  88                         || (source_exists = cp_mv_stat2(*argv, &source_stat, lstat)) < 1
  89                        ) {
  90                                bb_perror_msg("cannot rename '%s'", *argv);
  91                        } else {
  92                                static const char fmt[] ALIGN1 =
  93                                        "cannot overwrite %sdirectory with %sdirectory";
  94
  95                                if (dest_exists) {
  96                                        if (dest_exists == 3) {
  97                                                if (source_exists != 3) {
  98                                                        bb_error_msg(fmt, "", "non-");
  99                                                        goto RET_1;
 100                                                }
 101                                        } else {
 102                                                if (source_exists == 3) {
 103                                                        bb_error_msg(fmt, "non-", "");
 104                                                        goto RET_1;
 105                                                }
 106                                        }
 107                                        if (unlink(dest) < 0) {
 108                                                bb_perror_msg("cannot remove '%s'", dest);
 109                                                goto RET_1;
 110                                        }
 111                                }
 112                                /* FILEUTILS_RECUR also prevents nasties like
 113                                 * "read from device and write contents to dst"
 114                                 * instead of "create same device node" */
 115                                copy_flag = FILEUTILS_RECUR | FILEUTILS_PRESERVE_STATUS;
 116#if ENABLE_SELINUX
 117                                copy_flag |= FILEUTILS_PRESERVE_SECURITY_CONTEXT;
 118#endif
 119                                if ((copy_file(*argv, dest, copy_flag) >= 0)
 120                                 && (remove_file(*argv, FILEUTILS_RECUR | FILEUTILS_FORCE) >= 0)
 121                                ) {
 122                                        goto RET_0;
 123                                }
 124                        }
 125 RET_1:
 126                        status = 1;
 127                }
 128 RET_0:
 129                if (dest != last) {
 130                        free((void *) dest);
 131                }
 132        } while (*++argv != last);
 133
 134        return status;
 135}
 136