blob: b334206fb3ac8b0c84e5e0cf765758c55f0c102a [file] [log] [blame]
Erik Andersene49d5ec2000-02-08 19:58:47 +00001/* vi: set sw=4 ts=4: */
Erik Andersenfac10d72000-02-07 05:29:42 +00002/*
3 * Mini `cp' and `mv' implementation for BusyBox.
4 *
5 *
6 * Copyright (C) 1999 by Lineo, inc.
7 * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
8 *
9 * Copyright (C) 2000 by BitterSweet Enterprises, LLC. (GPL)
10 * Extensively modified and rewritten by Karl M. Hegbloom <karlheg@debian.org>
11 *
12 * This program is free software; you can redistribute it and/or modify
13 * it under the terms of the GNU General Public License as published by
14 * the Free Software Foundation; either version 2 of the License, or
15 * (at your option) any later version.
16 *
17 * This program is distributed in the hope that it will be useful,
18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
20 * General Public License for more details.
21 *
22 * You should have received a copy of the GNU General Public License
23 * along with this program; if not, write to the Free Software
24 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
25 *
26 */
27
Eric Andersen3570a342000-09-25 21:45:58 +000028#include "busybox.h"
Erik Andersenfac10d72000-02-07 05:29:42 +000029#define BB_DECLARE_EXTERN
30#define bb_need_name_too_long
31#define bb_need_omitting_directory
32#define bb_need_not_a_directory
33#include "messages.c"
34
35#include <stdio.h>
36#include <time.h>
37#include <utime.h>
38#include <dirent.h>
39#include <sys/param.h>
Erik Andersen61677fe2000-04-13 01:18:56 +000040#include <setjmp.h>
Erik Andersen029011b2000-03-04 21:19:32 +000041#include <string.h>
42#include <unistd.h>
43#include <errno.h>
Eric Andersenf4c02262000-07-13 18:42:58 +000044#include <getopt.h>
Erik Andersenfac10d72000-02-07 05:29:42 +000045
46#define is_cp 0
47#define is_mv 1
Erik Andersen029011b2000-03-04 21:19:32 +000048static int dz_i; /* index into cp_mv_usage */
Matt Kraai3bd8bd82000-07-14 23:28:47 +000049
Erik Andersene49d5ec2000-02-08 19:58:47 +000050static const char *cp_mv_usage[] = /* .rodata */
Erik Andersenfac10d72000-02-07 05:29:42 +000051{
Matt Kraai3bd8bd82000-07-14 23:28:47 +000052 cp_usage,
53 mv_usage
Erik Andersenfac10d72000-02-07 05:29:42 +000054};
55
Erik Andersen029011b2000-03-04 21:19:32 +000056static int recursiveFlag;
57static int followLinks;
58static int preserveFlag;
Erik Andersen59b9e872000-05-10 05:05:45 +000059static int forceFlag;
Erik Andersen029011b2000-03-04 21:19:32 +000060
61static const char *baseSrcName;
62static int srcDirFlag;
63static struct stat srcStatBuf;
64
Erik Andersen4f3f7572000-04-28 00:18:56 +000065static char baseDestName[BUFSIZ + 1];
Erik Andersen029011b2000-03-04 21:19:32 +000066static size_t baseDestLen;
67static int destDirFlag;
68static struct stat destStatBuf;
69
70static jmp_buf catch;
71static volatile int mv_Action_first_time;
72
73static void name_too_long__exit (void) __attribute__((noreturn));
74
75static
76void name_too_long__exit (void)
77{
Mark Whitleyf57c9442000-12-07 19:56:48 +000078 error_msg_and_die(name_too_long);
Erik Andersen029011b2000-03-04 21:19:32 +000079}
80
81static void
82fill_baseDest_buf(char *_buf, size_t * _buflen) {
83 const char *srcBasename;
84 if ((srcBasename = strrchr(baseSrcName, '/')) == NULL) {
85 srcBasename = baseSrcName;
86 if (_buf[*_buflen - 1] != '/') {
Erik Andersen4f3f7572000-04-28 00:18:56 +000087 if (++(*_buflen) > BUFSIZ)
Erik Andersen029011b2000-03-04 21:19:32 +000088 name_too_long__exit();
89 strcat(_buf, "/");
90 }
91 }
Erik Andersen4f3f7572000-04-28 00:18:56 +000092 if (*_buflen + strlen(srcBasename) > BUFSIZ)
Erik Andersen029011b2000-03-04 21:19:32 +000093 name_too_long__exit();
94 strcat(_buf, srcBasename);
95 return;
96
97}
98
99static int
Erik Andersen3364d782000-03-28 00:58:14 +0000100cp_mv_Action(const char *fileName, struct stat *statbuf, void* junk)
Erik Andersen029011b2000-03-04 21:19:32 +0000101{
Erik Andersen4f3f7572000-04-28 00:18:56 +0000102 char destName[BUFSIZ + 1];
Erik Andersen029011b2000-03-04 21:19:32 +0000103 size_t destLen;
104 const char *srcBasename;
105 char *name;
106
107 strcpy(destName, baseDestName);
108 destLen = strlen(destName);
109
110 if (srcDirFlag == TRUE) {
111 if (recursiveFlag == FALSE) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000112 error_msg(omitting_directory, baseSrcName);
Erik Andersen029011b2000-03-04 21:19:32 +0000113 return TRUE;
114 }
115 srcBasename = (strstr(fileName, baseSrcName)
116 + strlen(baseSrcName));
117
Erik Andersen4f3f7572000-04-28 00:18:56 +0000118 if (destLen + strlen(srcBasename) > BUFSIZ) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000119 error_msg(name_too_long);
Erik Andersen029011b2000-03-04 21:19:32 +0000120 return FALSE;
121 }
122 strcat(destName, srcBasename);
123 }
124 else if (destDirFlag == TRUE) {
125 fill_baseDest_buf(&destName[0], &destLen);
126 }
127 else {
128 srcBasename = baseSrcName;
129 }
130 if (mv_Action_first_time && (dz_i == is_mv)) {
131 mv_Action_first_time = errno = 0;
132 if (rename(fileName, destName) < 0 && errno != EXDEV) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000133 error_msg("rename(%s, %s): %s\n", fileName, destName,
Matt Kraaid537a952000-07-14 01:51:25 +0000134 strerror(errno));
Erik Andersen029011b2000-03-04 21:19:32 +0000135 goto do_copyFile; /* Try anyway... */
136 }
137 else if (errno == EXDEV)
138 goto do_copyFile;
139 else
140 longjmp(catch, 1); /* succeeded with rename() */
141 }
142 do_copyFile:
143 if (preserveFlag == TRUE && statbuf->st_nlink > 1) {
144 if (is_in_ino_dev_hashtable(statbuf, &name)) {
145 if (link(name, destName) < 0) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000146 error_msg("link(%s, %s): %s\n", name, destName, strerror(errno));
Erik Andersen029011b2000-03-04 21:19:32 +0000147 return FALSE;
148 }
149 return TRUE;
150 }
151 else {
152 add_to_ino_dev_hashtable(statbuf, destName);
153 }
154 }
Mark Whitleyf57c9442000-12-07 19:56:48 +0000155 return copy_file(fileName, destName, preserveFlag, followLinks, forceFlag);
Erik Andersen029011b2000-03-04 21:19:32 +0000156}
157
158static int
Erik Andersen3364d782000-03-28 00:58:14 +0000159rm_Action(const char *fileName, struct stat *statbuf, void* junk)
Erik Andersen029011b2000-03-04 21:19:32 +0000160{
161 int status = TRUE;
162
163 if (S_ISDIR(statbuf->st_mode)) {
164 if (rmdir(fileName) < 0) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000165 error_msg("rmdir(%s): %s\n", fileName, strerror(errno));
Erik Andersen029011b2000-03-04 21:19:32 +0000166 status = FALSE;
167 }
168 } else if (unlink(fileName) < 0) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000169 error_msg("unlink(%s): %s\n", fileName, strerror(errno));
Erik Andersen029011b2000-03-04 21:19:32 +0000170 status = FALSE;
171 }
172 return status;
173}
174
Erik Andersenfac10d72000-02-07 05:29:42 +0000175extern int cp_mv_main(int argc, char **argv)
176{
Eric Andersene7413a92000-07-14 06:19:41 +0000177 volatile int i;
Pavel Roskin93c0d9f2000-07-14 15:18:57 +0000178 int c;
Eric Andersenf4c02262000-07-13 18:42:58 +0000179
Matt Kraaie714bce2000-07-12 00:53:06 +0000180 if (*applet_name == 'c' && *(applet_name + 1) == 'p')
Erik Andersene49d5ec2000-02-08 19:58:47 +0000181 dz_i = is_cp;
182 else
183 dz_i = is_mv;
184 if (argc < 3)
185 usage(cp_mv_usage[dz_i]);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000186
187 if (dz_i == is_cp) {
Erik Andersen59b9e872000-05-10 05:05:45 +0000188 recursiveFlag = preserveFlag = forceFlag = FALSE;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000189 followLinks = TRUE;
Eric Andersenf4c02262000-07-13 18:42:58 +0000190 while ((c = getopt(argc, argv, "adpRf")) != EOF) {
191 switch (c) {
Pavel Roskinb3d235f2000-06-08 18:06:37 +0000192 case 'a':
193 followLinks = FALSE;
194 preserveFlag = TRUE;
195 recursiveFlag = TRUE;
196 break;
197 case 'd':
198 followLinks = FALSE;
199 break;
200 case 'p':
201 preserveFlag = TRUE;
202 break;
203 case 'R':
204 recursiveFlag = TRUE;
205 break;
206 case 'f':
207 forceFlag = TRUE;
208 break;
209 default:
210 usage(cp_mv_usage[is_cp]);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000211 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000212 }
Eric Andersenf4c02262000-07-13 18:42:58 +0000213 if ((argc - optind) < 2) {
Pavel Roskinb3d235f2000-06-08 18:06:37 +0000214 usage(cp_mv_usage[dz_i]);
Eric Andersen815e9042000-06-06 16:15:23 +0000215 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000216 } else { /* (dz_i == is_mv) */
Eric Andersen2da90d32000-09-23 20:55:59 +0000217 /* Initialize optind to 1, since in libc5 optind
218 * is not initialized until getopt() is called
219 * (or until sneaky programmers force it...). */
220 optind = 1;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000221 recursiveFlag = preserveFlag = TRUE;
222 followLinks = FALSE;
223 }
Eric Andersen815e9042000-06-06 16:15:23 +0000224
Erik Andersene49d5ec2000-02-08 19:58:47 +0000225
Erik Andersen4f3f7572000-04-28 00:18:56 +0000226 if (strlen(argv[argc - 1]) > BUFSIZ) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000227 error_msg(name_too_long);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000228 goto exit_false;
229 }
230 strcpy(baseDestName, argv[argc - 1]);
231 baseDestLen = strlen(baseDestName);
232 if (baseDestLen == 0)
233 goto exit_false;
234
Mark Whitleyf57c9442000-12-07 19:56:48 +0000235 destDirFlag = is_directory(baseDestName, TRUE, &destStatBuf);
Matt Kraaiadbf0182000-08-06 15:36:50 +0000236 if (argc - optind > 2 && destDirFlag == FALSE) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000237 error_msg(not_a_directory, baseDestName);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000238 goto exit_false;
239 }
240
Eric Andersenf4c02262000-07-13 18:42:58 +0000241 for (i = optind; i < (argc-1); i++) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000242 size_t srcLen;
Erik Andersen029011b2000-03-04 21:19:32 +0000243 volatile int flags_memo;
244 int status;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000245
Eric Andersenf4c02262000-07-13 18:42:58 +0000246 baseSrcName=argv[i];
Erik Andersene49d5ec2000-02-08 19:58:47 +0000247
Erik Andersen4f3f7572000-04-28 00:18:56 +0000248 if ((srcLen = strlen(baseSrcName)) > BUFSIZ)
Erik Andersen029011b2000-03-04 21:19:32 +0000249 name_too_long__exit();
Erik Andersene49d5ec2000-02-08 19:58:47 +0000250
Erik Andersen029011b2000-03-04 21:19:32 +0000251 if (srcLen == 0) continue; /* "" */
Erik Andersene49d5ec2000-02-08 19:58:47 +0000252
Mark Whitleyf57c9442000-12-07 19:56:48 +0000253 srcDirFlag = is_directory(baseSrcName, followLinks, &srcStatBuf);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000254
255 if ((flags_memo = (recursiveFlag == TRUE &&
256 srcDirFlag == TRUE && destDirFlag == TRUE))) {
Erik Andersen029011b2000-03-04 21:19:32 +0000257
258 struct stat sb;
259 int state = 0;
260 char *pushd, *d, *p;
261
Erik Andersen4f3f7572000-04-28 00:18:56 +0000262 if ((pushd = getcwd(NULL, BUFSIZ + 1)) == NULL) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000263 error_msg("getcwd(): %s\n", strerror(errno));
Erik Andersen029011b2000-03-04 21:19:32 +0000264 continue;
265 }
266 if (chdir(baseDestName) < 0) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000267 error_msg("chdir(%s): %s\n", baseSrcName, strerror(errno));
Erik Andersen029011b2000-03-04 21:19:32 +0000268 continue;
269 }
Erik Andersen4f3f7572000-04-28 00:18:56 +0000270 if ((d = getcwd(NULL, BUFSIZ + 1)) == NULL) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000271 error_msg("getcwd(): %s\n", strerror(errno));
Erik Andersen029011b2000-03-04 21:19:32 +0000272 continue;
273 }
274 while (!state && *d != '\0') {
275 if (stat(d, &sb) < 0) { /* stat not lstat - always dereference targets */
Mark Whitleyf57c9442000-12-07 19:56:48 +0000276 error_msg("stat(%s): %s\n", d, strerror(errno));
Erik Andersen029011b2000-03-04 21:19:32 +0000277 state = -1;
278 continue;
Erik Andersen9ffdaa62000-02-11 21:55:04 +0000279 }
Erik Andersen029011b2000-03-04 21:19:32 +0000280 if ((sb.st_ino == srcStatBuf.st_ino) &&
281 (sb.st_dev == srcStatBuf.st_dev)) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000282 error_msg("Cannot %s `%s' into a subdirectory of itself, "
Matt Kraaid537a952000-07-14 01:51:25 +0000283 "`%s/%s'\n", applet_name, baseSrcName,
284 baseDestName, baseSrcName);
Erik Andersen029011b2000-03-04 21:19:32 +0000285 state = -1;
286 continue;
287 }
288 if ((p = strrchr(d, '/')) != NULL) {
289 *p = '\0';
290 }
291 }
292 if (chdir(pushd) < 0) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000293 error_msg("chdir(%s): %s\n", pushd, strerror(errno));
Erik Andersen029011b2000-03-04 21:19:32 +0000294 free(pushd);
295 free(d);
296 continue;
297 }
298 free(pushd);
299 free(d);
300 if (state < 0)
301 continue;
302 else
Erik Andersen9ffdaa62000-02-11 21:55:04 +0000303 fill_baseDest_buf(baseDestName, &baseDestLen);
Erik Andersenfac10d72000-02-07 05:29:42 +0000304 }
Erik Andersen029011b2000-03-04 21:19:32 +0000305 status = setjmp(catch);
306 if (status == 0) {
307 mv_Action_first_time = 1;
Mark Whitleyf57c9442000-12-07 19:56:48 +0000308 if (recursive_action(baseSrcName,
Erik Andersen029011b2000-03-04 21:19:32 +0000309 recursiveFlag, followLinks, FALSE,
Erik Andersen3364d782000-03-28 00:58:14 +0000310 cp_mv_Action, cp_mv_Action, NULL) == FALSE) goto exit_false;
Erik Andersen029011b2000-03-04 21:19:32 +0000311 if (dz_i == is_mv &&
Mark Whitleyf57c9442000-12-07 19:56:48 +0000312 recursive_action(baseSrcName,
Erik Andersen029011b2000-03-04 21:19:32 +0000313 recursiveFlag, followLinks, TRUE,
Erik Andersen3364d782000-03-28 00:58:14 +0000314 rm_Action, rm_Action, NULL) == FALSE) goto exit_false;
Erik Andersen029011b2000-03-04 21:19:32 +0000315 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000316 if (flags_memo)
317 *(baseDestName + baseDestLen) = '\0';
Erik Andersenfac10d72000-02-07 05:29:42 +0000318 }
Matt Kraai3e856ce2000-12-01 02:55:13 +0000319 return EXIT_SUCCESS;
Erik Andersen029011b2000-03-04 21:19:32 +0000320 exit_false:
Matt Kraai3e856ce2000-12-01 02:55:13 +0000321 return EXIT_FAILURE;
Erik Andersenfac10d72000-02-07 05:29:42 +0000322}
323
Erik Andersen029011b2000-03-04 21:19:32 +0000324/*
325Local Variables:
326c-file-style: "linux"
327c-basic-offset: 4
328tab-width: 4
329End:
330*/