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