mv.c revision 50471 
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
38static 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
45static char sccsid[] = "@(#)mv.c	8.2 (Berkeley) 4/2/94";
46#endif
47static const char rcsid[] =
48  "$FreeBSD: head/bin/mv/mv.c 50471 1999-08-27 23:15:48Z peter $";
49#endif /* not lint */
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 <stdio.h>
61#include <stdlib.h>
62#include <string.h>
63#include <unistd.h>
64
65#include "pathnames.h"
66
67int fflg, iflg;
68
69int	copy __P((char *, char *));
70int	do_move __P((char *, char *));
71int	fastcopy __P((char *, char *, struct stat *));
72void	usage __P((void));
73
74int
75main(argc, argv)
76	int argc;
77	char *argv[];
78{
79	register int baselen, len, rval;
80	register char *p, *endp;
81	struct stat sb;
82	int ch;
83	char path[MAXPATHLEN];
84
85	while ((ch = getopt(argc, argv, "fi")) != -1)
86		switch (ch) {
87		case 'i':
88			iflg = 1;
89			fflg = 0;
90			break;
91		case 'f':
92			fflg = 1;
93			iflg = 0;
94			break;
95		default:
96			usage();
97		}
98	argc -= optind;
99	argv += optind;
100
101	if (argc < 2)
102		usage();
103
104	/*
105	 * If the stat on the target fails or the target isn't a directory,
106	 * try the move.  More than 2 arguments is an error in this case.
107	 */
108	if (stat(argv[argc - 1], &sb) || !S_ISDIR(sb.st_mode)) {
109		if (argc > 2)
110			usage();
111		exit(do_move(argv[0], argv[1]));
112	}
113
114	/* It's a directory, move each file into it. */
115	if (strlen(argv[argc - 1]) > sizeof(path) - 1)
116		errx(1, "%s: destination pathname too long", *argv);
117	(void)strcpy(path, argv[argc - 1]);
118	baselen = strlen(path);
119	endp = &path[baselen];
120	if (!baselen || *(endp - 1) != '/') {
121		*endp++ = '/';
122		++baselen;
123	}
124	for (rval = 0; --argc; ++argv) {
125		/*
126		 * Find the last component of the source pathname.  It
127		 * may have trailing slashes.
128		 */
129		p = *argv + strlen(*argv);
130		while (p != *argv && p[-1] == '/')
131			--p;
132		while (p != *argv && p[-1] != '/')
133			--p;
134
135		if ((baselen + (len = strlen(p))) >= MAXPATHLEN) {
136			warnx("%s: destination pathname too long", *argv);
137			rval = 1;
138		} else {
139			memmove(endp, p, len + 1);
140			if (do_move(*argv, path))
141				rval = 1;
142		}
143	}
144	exit(rval);
145}
146
147int
148do_move(from, to)
149	char *from, *to;
150{
151	struct stat sb;
152	int ask, ch, first;
153	char modep[15];
154
155	/*
156	 * Check access.  If interactive and file exists, ask user if it
157	 * should be replaced.  Otherwise if file exists but isn't writable
158	 * make sure the user wants to clobber it.
159	 */
160	if (!fflg && !access(to, F_OK)) {
161
162		/* prompt only if source exist */
163	        if (lstat(from, &sb) == -1) {
164			warn("%s", from);
165			return (1);
166		}
167
168#define YESNO "(y/n [n]) "
169		ask = 0;
170		if (iflg) {
171			(void)fprintf(stderr, "overwrite %s? %s", to, YESNO);
172			ask = 1;
173		} else if (access(to, W_OK) && !stat(to, &sb)) {
174			strmode(sb.st_mode, modep);
175			(void)fprintf(stderr, "override %s%s%s/%s for %s? %s",
176			    modep + 1, modep[9] == ' ' ? "" : " ",
177			    user_from_uid(sb.st_uid, 0),
178			    group_from_gid(sb.st_gid, 0), to, YESNO);
179			ask = 1;
180		}
181		if (ask) {
182			first = ch = getchar();
183			while (ch != '\n' && ch != EOF)
184				ch = getchar();
185			if (first != 'y' && first != 'Y') {
186				(void)fprintf(stderr, "not overwritten\n");
187				return (0);
188			}
189		}
190	}
191	if (!rename(from, to))
192		return (0);
193
194	if (errno == EXDEV) {
195		struct statfs sfs;
196		char path[MAXPATHLEN];
197
198		/* Can't mv(1) a mount point. */
199		if (realpath(from, path) == NULL) {
200			warnx("cannot resolve %s: %s", from, path);
201			return (1);
202		}
203		if (!statfs(path, &sfs) && !strcmp(path, sfs.f_mntonname)) {
204			warnx("cannot rename a mount point");
205			return (1);
206		}
207	} else {
208		warn("rename %s to %s", from, to);
209		return (1);
210	}
211
212	/*
213	 * If rename fails because we're trying to cross devices, and
214	 * it's a regular file, do the copy internally; otherwise, use
215	 * cp and rm.
216	 */
217	if (stat(from, &sb)) {
218		warn("%s", from);
219		return (1);
220	}
221	return (S_ISREG(sb.st_mode) ?
222	    fastcopy(from, to, &sb) : copy(from, to));
223}
224
225int
226fastcopy(from, to, sbp)
227	char *from, *to;
228	struct stat *sbp;
229{
230	struct timeval tval[2];
231	static u_int blen;
232	static char *bp;
233	mode_t oldmode;
234	register int nread, from_fd, to_fd;
235
236	if ((from_fd = open(from, O_RDONLY, 0)) < 0) {
237		warn("%s", from);
238		return (1);
239	}
240	if (blen < sbp->st_blksize) {
241		if (bp != NULL)
242			free(bp);
243		if ((bp = malloc(sbp->st_blksize)) == NULL) {
244			blen = 0;
245			warnx("malloc failed");
246			return (1);
247		}
248		blen = sbp->st_blksize;
249	}
250	while ((to_fd =
251	    open(to, O_CREAT | O_EXCL | O_TRUNC | O_WRONLY, 0)) < 0) {
252		if (errno == EEXIST && unlink(to) == 0)
253			continue;
254		warn("%s", to);
255		(void)close(from_fd);
256		return (1);
257	}
258	while ((nread = read(from_fd, bp, blen)) > 0)
259		if (write(to_fd, bp, nread) != nread) {
260			warn("%s", to);
261			goto err;
262		}
263	if (nread < 0) {
264		warn("%s", from);
265err:		if (unlink(to))
266			warn("%s: remove", to);
267		(void)close(from_fd);
268		(void)close(to_fd);
269		return (1);
270	}
271	(void)close(from_fd);
272
273	oldmode = sbp->st_mode & ALLPERMS;
274	if (fchown(to_fd, sbp->st_uid, sbp->st_gid)) {
275		warn("%s: set owner/group (was: %lu/%lu)", to,
276		    (u_long)sbp->st_uid, (u_long)sbp->st_gid);
277		if (oldmode & (S_ISUID | S_ISGID)) {
278			warnx(
279"%s: owner/group changed; clearing suid/sgid (mode was 0%03o)",
280			    to, oldmode);
281			sbp->st_mode &= ~(S_ISUID | S_ISGID);
282		}
283	}
284	if (fchmod(to_fd, sbp->st_mode))
285		warn("%s: set mode (was: 0%03o)", to, oldmode);
286
287	tval[0].tv_sec = sbp->st_atime;
288	tval[1].tv_sec = sbp->st_mtime;
289	tval[0].tv_usec = tval[1].tv_usec = 0;
290	if (utimes(to, tval))
291		warn("%s: set times", to);
292
293	if (close(to_fd)) {
294		warn("%s", to);
295		return (1);
296	}
297
298	if (unlink(from)) {
299		warn("%s: remove", from);
300		return (1);
301	}
302	return (0);
303}
304
305int
306copy(from, to)
307	char *from, *to;
308{
309	int pid, status;
310
311	if ((pid = fork()) == 0) {
312		execl(_PATH_CP, "mv", "-PRp", from, to, NULL);
313		warn("%s", _PATH_CP);
314		_exit(1);
315	}
316	if (waitpid(pid, &status, 0) == -1) {
317		warn("%s: waitpid", _PATH_CP);
318		return (1);
319	}
320	if (!WIFEXITED(status)) {
321		warn("%s: did not terminate normally", _PATH_CP);
322		return (1);
323	}
324	if (WEXITSTATUS(status)) {
325		warn("%s: terminated with %d (non-zero) status",
326		    _PATH_CP, WEXITSTATUS(status));
327		return (1);
328	}
329	if (!(pid = vfork())) {
330		execl(_PATH_RM, "mv", "-rf", from, NULL);
331		warn("%s", _PATH_RM);
332		_exit(1);
333	}
334	if (waitpid(pid, &status, 0) == -1) {
335		warn("%s: waitpid", _PATH_RM);
336		return (1);
337	}
338	if (!WIFEXITED(status)) {
339		warn("%s: did not terminate normally", _PATH_RM);
340		return (1);
341	}
342	if (WEXITSTATUS(status)) {
343		warn("%s: terminated with %d (non-zero) status",
344		    _PATH_RM, WEXITSTATUS(status));
345		return (1);
346	}
347	return (0);
348}
349
350void
351usage()
352{
353	(void)fprintf(stderr, "%s\n%s\n",
354		      "usage: mv [-f | -i] source target",
355		      "       mv [-f | -i] source ... directory");
356	exit(1);
357}
358