mv/mv.c

   1 /*
   2  * Copyright (c) 1989, 1993, 1994
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * This code is derived from software contributed to Berkeley by
   6  * Ken Smith of The State University of New York at Buffalo.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  * 3. All advertising materials mentioning features or use of this software
  17  *    must display the following acknowledgement:
  18  *      This product includes software developed by the University of
  19  *      California, Berkeley and its contributors.
  20  * 4. Neither the name of the University nor the names of its contributors
  21  *    may be used to endorse or promote products derived from this software
  22  *    without specific prior written permission.
  23  *
  24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  34  * SUCH DAMAGE.
  35  */
  36
  37 #ifndef lint
  38 static char const copyright[] =
  39 "@(#) Copyright (c) 1989, 1993, 1994\n\
  40         The Regents of the University of California.  All rights reserved.\n";
  41 #endif /* not lint */
  42
  43 #ifndef lint
  44 #if 0
  45 static char sccsid[] = "@(#)mv.c        8.2 (Berkeley) 4/2/94";
  46 #endif
  47 #endif /* not lint */
  48 #include <sys/cdefs.h>
  49 __RCSID("$FreeBSD: src/bin/mv/mv.c,v 1.39 2002/07/09 17:45:13 johan Exp $");
  50
  51 #include <sys/param.h>
  52 #include <sys/time.h>
  53 #include <sys/wait.h>
  54 #include <sys/stat.h>
  55 #include <sys/mount.h>
  56
  57 #include <err.h>
  58 #include <errno.h>
  59 #include <fcntl.h>
  60 #include <grp.h>
  61 #include <limits.h>
  62 #include <paths.h>
  63 #include <pwd.h>
  64 #include <stdio.h>
  65 #include <stdlib.h>
  66 #include <string.h>
  67 #include <sysexits.h>
  68 #include <unistd.h>
  69
  70 #ifdef __APPLE__
  71 #include <copyfile.h>
  72 #include <sys/mount.h>
  73 #endif
  74
  75 #include "pathnames.h"
  76
  77 int fflg, iflg, nflg, vflg;
  78
  79 int     copy(char *, char *);
  80 int     do_move(char *, char *);
  81 int     fastcopy(char *, char *, struct stat *);
  82 void    usage(void);
  83
  84 int
  85 main(int argc, char *argv[])
  86 {
  87         size_t baselen, len;
  88         int rval;
  89         char *p, *endp;
  90         struct stat sb;
  91 #ifdef __APPLE__
  92         struct stat fsb, tsb;
  93 #endif /* __APPLE__ */
  94         int ch;
  95         char path[PATH_MAX];
  96
  97         while ((ch = getopt(argc, argv, "finv")) != -1)
  98                 switch (ch) {
  99                 case 'i':
 100                         iflg = 1;
 101                         fflg = nflg = 0;
 102                         break;
 103                 case 'f':
 104                         fflg = 1;
 105                         iflg = nflg = 0;
 106                         break;
 107                 case 'n':
 108                         nflg = 1;
 109                         fflg = iflg = 0;
 110                         break;
 111                 case 'v':
 112                         vflg = 1;
 113                         break;
 114                 default:
 115                         usage();
 116                 }
 117         argc -= optind;
 118         argv += optind;
 119
 120         if (argc < 2)
 121                 usage();
 122
 123         /*
 124          * If the stat on the target fails or the target isn't a directory,
 125          * try the move.  More than 2 arguments is an error in this case.
 126          */
 127         if (stat(argv[argc - 1], &sb) || !S_ISDIR(sb.st_mode)) {
 128                 if (argc > 2)
 129                         usage();
 130                 exit(do_move(argv[0], argv[1]));
 131         }
 132
 133 #ifdef __APPLE__
 134         if (argc == 2 && !lstat(argv[0], &fsb) && !lstat(argv[1], &tsb) &&
 135                 fsb.st_ino == tsb.st_ino && fsb.st_dev == tsb.st_dev &&
 136                 fsb.st_gen == tsb.st_gen) {
 137                 /*
 138                  * We appear to be trying to move a directory into itself,
 139                  * but it may be that the filesystem is case insensitive and
 140                  * we are trying to rename the directory to a case-variant.
 141                  * Ignoring trailing slashes, we look for any difference in
 142                  * the directory names.  If there is a difference we do
 143                  * the rename, otherwise we fall-thru to the traditional
 144                  * error.  Note the lstat calls above (rather than stat)
 145                  * permit the renaming of symlinks to case-variants.
 146                  */
 147                 char *q;
 148
 149                 for (p = argv[0] + strlen(argv[0]); p != argv[0]; ) {
 150                         p--;
 151                         if (*p != '/')
 152                                 break;
 153                 }
 154                 for (q = argv[1] + strlen(argv[1]); q != argv[1]; ) {
 155                         q--;
 156                         if (*q != '/')
 157                                 break;
 158                 }
 159                 for ( ; ; p--, q--) {
 160                         if (*p != *q)
 161                                 exit(do_move(argv[0], argv[1]));
 162                         if (*p == '/')
 163                                 break;
 164                         if (p == argv[0]) {
 165                                 if (q == argv[1] || *(q-1) == '/')
 166                                         break;
 167                                 exit(do_move(argv[0], argv[1]));
 168                         }
 169                         if (q == argv[1]) {
 170                                 if (p == argv[0] || *(p-1) == '/')
 171                                         break;
 172                                 exit(do_move(argv[0], argv[1]));
 173                         }
 174                 }
 175         }
 176 #endif /* __APPLE__ */
 177
 178         /* It's a directory, move each file into it. */
 179         if (strlen(argv[argc - 1]) > sizeof(path) - 1)
 180                 errx(1, "%s: destination pathname too long", *argv);
 181         (void)strcpy(path, argv[argc - 1]);
 182         baselen = strlen(path);
 183         endp = &path[baselen];
 184         if (!baselen || *(endp - 1) != '/') {
 185                 *endp++ = '/';
 186                 ++baselen;
 187         }
 188         for (rval = 0; --argc; ++argv) {
 189                 /*
 190                  * Find the last component of the source pathname.  It
 191                  * may have trailing slashes.
 192                  */
 193                 p = *argv + strlen(*argv);
 194                 while (p != *argv && p[-1] == '/')
 195                         --p;
 196                 while (p != *argv && p[-1] != '/')
 197                         --p;
 198
 199                 if ((baselen + (len = strlen(p))) >= PATH_MAX) {
 200                         warnx("%s: destination pathname too long", *argv);
 201                         rval = 1;
 202                 } else {
 203                         memmove(endp, p, (size_t)len + 1);
 204                         if (do_move(*argv, path))
 205                                 rval = 1;
 206                 }
 207         }
 208         exit(rval);
 209 }
 210
 211 int
 212 do_move(char *from, char *to)
 213 {
 214         struct stat sb;
 215         int ask, ch, first;
 216         char modep[15];
 217
 218         /*
 219          * Check access.  If interactive and file exists, ask user if it
 220          * should be replaced.  Otherwise if file exists but isn't writable
 221          * make sure the user wants to clobber it.
 222          */
 223         if (!fflg && !access(to, F_OK)) {
 224
 225                 /* prompt only if source exist */
 226                 if (lstat(from, &sb) == -1) {
 227                         warn("%s", from);
 228                         return (1);
 229                 }
 230
 231 #define YESNO "(y/n [n]) "
 232                 ask = 0;
 233                 if (nflg) {
 234                         if (vflg)
 235                                 printf("%s not overwritten\n", to);
 236                         return (0);
 237                 } else if (iflg) {
 238                         (void)fprintf(stderr, "overwrite %s? %s", to, YESNO);
 239                         ask = 1;
 240                 } else if (access(to, W_OK) && !stat(to, &sb)) {
 241                         strmode(sb.st_mode, modep);
 242                         (void)fprintf(stderr, "override %s%s%s/%s for %s? %s",
 243                             modep + 1, modep[9] == ' ' ? "" : " ",
 244                             user_from_uid((unsigned long)sb.st_uid, 0),
 245                             group_from_gid((unsigned long)sb.st_gid, 0), to, YESNO);
 246                         ask = 1;
 247                 }
 248                 if (ask) {
 249                         first = ch = getchar();
 250                         while (ch != '\n' && ch != EOF)
 251                                 ch = getchar();
 252                         if (first != 'y' && first != 'Y') {
 253                                 (void)fprintf(stderr, "not overwritten\n");
 254                                 return (0);
 255                         }
 256                 }
 257         }
 258         if (!rename(from, to)) {
 259                 if (vflg)
 260                         printf("%s -> %s\n", from, to);
 261                 return (0);
 262         }
 263
 264         if (errno == EXDEV) {
 265                 struct statfs sfs;
 266                 char path[PATH_MAX];
 267
 268                 /* Can't mv(1) a mount point. */
 269                 if (realpath(from, path) == NULL) {
 270                         warnx("cannot resolve %s: %s", from, path);
 271                         return (1);
 272                 }
 273                 if (!statfs(path, &sfs) && !strcmp(path, sfs.f_mntonname)) {
 274                         warnx("cannot rename a mount point");
 275                         return (1);
 276                 }
 277         } else {
 278                 warn("rename %s to %s", from, to);
 279                 return (1);
 280         }
 281
 282         /*
 283          * If rename fails because we're trying to cross devices, and
 284          * it's a regular file, do the copy internally; otherwise, use
 285          * cp and rm.
 286          */
 287         if (lstat(from, &sb)) {
 288                 warn("%s", from);
 289                 return (1);
 290         }
 291         return (S_ISREG(sb.st_mode) ?
 292             fastcopy(from, to, &sb) : copy(from, to));
 293 }
 294
 295 int
 296 fastcopy(char *from, char *to, struct stat *sbp)
 297 {
 298         struct timeval tval[2];
 299         static u_int blen;
 300         static char *bp;
 301         mode_t oldmode;
 302         int nread, from_fd, to_fd;
 303
 304         if ((from_fd = open(from, O_RDONLY, 0)) < 0) {
 305                 warn("%s", from);
 306                 return (1);
 307         }
 308         if (blen < sbp->st_blksize) {
 309                 if (bp != NULL)
 310                         free(bp);
 311                 if ((bp = malloc((size_t)sbp->st_blksize)) == NULL) {
 312                         blen = 0;
 313                         warnx("malloc failed");
 314                         return (1);
 315                 }
 316                 blen = sbp->st_blksize;
 317         }
 318         while ((to_fd =
 319             open(to, O_CREAT | O_EXCL | O_TRUNC | O_WRONLY, 0)) < 0) {
 320                 if (errno == EEXIST && unlink(to) == 0)
 321                         continue;
 322                 warn("%s", to);
 323                 (void)close(from_fd);
 324                 return (1);
 325         }
 326 #ifdef __APPLE__
 327         {
 328                 struct statfs sfs;
 329
 330                 /*
 331                  * Pre-allocate blocks for the destination file if it
 332                  * resides on Xsan.
 333                  */
 334                 if (fstatfs(to_fd, &sfs) == 0 &&
 335                     strcmp(sfs.f_fstypename, "acfs") == 0) {
 336                         fstore_t fst;
 337
 338                         fst.fst_flags = 0;
 339                         fst.fst_posmode = F_PEOFPOSMODE;
 340                         fst.fst_offset = 0;
 341                         fst.fst_length = sbp->st_size;
 342
 343                         (void) fcntl(to_fd, F_PREALLOCATE, &fst);
 344                 }
 345         }
 346 #endif /* __APPLE__ */
 347         while ((nread = read(from_fd, bp, (size_t)blen)) > 0)
 348                 if (write(to_fd, bp, (size_t)nread) != nread) {
 349                         warn("%s", to);
 350                         goto err;
 351                 }
 352         if (nread < 0) {
 353                 warn("%s", from);
 354 err:            if (unlink(to))
 355                         warn("%s: remove", to);
 356                 (void)close(from_fd);
 357                 (void)close(to_fd);
 358                 return (1);
 359         }
 360 #ifdef __APPLE__
 361         copyfile(from, to, 0, COPYFILE_ACL | COPYFILE_XATTR);
 362 #endif
 363         (void)close(from_fd);
 364
 365         oldmode = sbp->st_mode & ALLPERMS;
 366         if (fchown(to_fd, sbp->st_uid, sbp->st_gid)) {
 367                 warn("%s: set owner/group (was: %lu/%lu)", to,
 368                     (u_long)sbp->st_uid, (u_long)sbp->st_gid);
 369                 if (oldmode & (S_ISUID | S_ISGID)) {
 370                         warnx(
 371 "%s: owner/group changed; clearing suid/sgid (mode was 0%03o)",
 372                             to, oldmode);
 373                         sbp->st_mode &= ~(S_ISUID | S_ISGID);
 374                 }
 375         }
 376         if (fchmod(to_fd, sbp->st_mode))
 377                 warn("%s: set mode (was: 0%03o)", to, oldmode);
 378         /*
 379          * XXX
 380          * NFS doesn't support chflags; ignore errors unless there's reason
 381          * to believe we're losing bits.  (Note, this still won't be right
 382          * if the server supports flags and we were trying to *remove* flags
 383          * on a file that we copied, i.e., that we didn't create.)
 384          */
 385         errno = 0;
 386         if (fchflags(to_fd, (u_long)sbp->st_flags))
 387                 if (errno != EOPNOTSUPP || sbp->st_flags != 0)
 388                         warn("%s: set flags (was: 0%07o)", to, sbp->st_flags);
 389
 390         tval[0].tv_sec = sbp->st_atime;
 391         tval[1].tv_sec = sbp->st_mtime;
 392         tval[0].tv_usec = tval[1].tv_usec = 0;
 393         if (utimes(to, tval))
 394                 warn("%s: set times", to);
 395
 396         if (close(to_fd)) {
 397                 warn("%s", to);
 398                 return (1);
 399         }
 400
 401         if (unlink(from)) {
 402                 warn("%s: remove", from);
 403                 return (1);
 404         }
 405         if (vflg)
 406                 printf("%s -> %s\n", from, to);
 407         return (0);
 408 }
 409
 410 int
 411 copy(char *from, char *to)
 412 {
 413         int pid, status;
 414
 415         if ((pid = fork()) == 0) {
 416                 execl(_PATH_CP, "mv", vflg ? "-PRpv" : "-PRp", "--", from, to,
 417                     (char *)NULL);
 418                 warn("%s", _PATH_CP);
 419                 _exit(1);
 420         }
 421         if (waitpid(pid, &status, 0) == -1) {
 422                 warn("%s: waitpid", _PATH_CP);
 423                 return (1);
 424         }
 425         if (!WIFEXITED(status)) {
 426                 warn("%s: did not terminate normally", _PATH_CP);
 427                 return (1);
 428         }
 429         if (WEXITSTATUS(status)) {
 430                 warn("%s: terminated with %d (non-zero) status",
 431                     _PATH_CP, WEXITSTATUS(status));
 432                 return (1);
 433         }
 434         if (!(pid = vfork())) {
 435                 execl(_PATH_RM, "mv", "-rf", "--", from, (char *)NULL);
 436                 warn("%s", _PATH_RM);
 437                 _exit(1);
 438         }
 439         if (waitpid(pid, &status, 0) == -1) {
 440                 warn("%s: waitpid", _PATH_RM);
 441                 return (1);
 442         }
 443         if (!WIFEXITED(status)) {
 444                 warn("%s: did not terminate normally", _PATH_RM);
 445                 return (1);
 446         }
 447         if (WEXITSTATUS(status)) {
 448                 warn("%s: terminated with %d (non-zero) status",
 449                     _PATH_RM, WEXITSTATUS(status));
 450                 return (1);
 451         }
 452         return (0);
 453 }
 454
 455 void
 456 usage(void)
 457 {
 458
 459         (void)fprintf(stderr, "%s\n%s\n",
 460                       "usage: mv [-f | -i | -n] [-v] source target",
 461                       "       mv [-f | -i | -n] [-v] source ... directory");
 462         exit(EX_USAGE);
 463 }