blob: fb48d3c5cecc5c4698e97a1102d7aa7de48977cf [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 *
Eric Andersen8ec10a92001-01-27 09:33:39 +00006 * Copyright (C) 1999,2000,2001 by Lineo, inc.
Erik Andersenfac10d72000-02-07 05:29:42 +00007 * 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
Erik Andersenfac10d72000-02-07 05:29:42 +000028#include <stdio.h>
29#include <time.h>
30#include <utime.h>
31#include <dirent.h>
32#include <sys/param.h>
Erik Andersen61677fe2000-04-13 01:18:56 +000033#include <setjmp.h>
Erik Andersen029011b2000-03-04 21:19:32 +000034#include <string.h>
35#include <unistd.h>
36#include <errno.h>
Eric Andersenf4c02262000-07-13 18:42:58 +000037#include <getopt.h>
Eric Andersened3ef502001-01-27 08:24:39 +000038#include <stdlib.h>
Eric Andersencbe31da2001-02-20 06:14:08 +000039#include "busybox.h"
40#define BB_DECLARE_EXTERN
41#define bb_need_name_too_long
42#define bb_need_omitting_directory
43#define bb_need_not_a_directory
44#include "messages.c"
45
Erik Andersenfac10d72000-02-07 05:29:42 +000046
Mark Whitley59ab0252001-01-23 22:30:04 +000047static const int is_cp = 0;
48static const int is_mv = 1;
Erik Andersen029011b2000-03-04 21:19:32 +000049static int dz_i; /* index into cp_mv_usage */
Matt Kraai3bd8bd82000-07-14 23:28:47 +000050
Erik Andersenfac10d72000-02-07 05:29:42 +000051
Erik Andersen029011b2000-03-04 21:19:32 +000052static int recursiveFlag;
53static int followLinks;
54static int preserveFlag;
Erik Andersen59b9e872000-05-10 05:05:45 +000055static int forceFlag;
Erik Andersen029011b2000-03-04 21:19:32 +000056
57static const char *baseSrcName;
58static int srcDirFlag;
59static struct stat srcStatBuf;
60
Mark Whitley59ab0252001-01-23 22:30:04 +000061static char *pBaseDestName;
Erik Andersen029011b2000-03-04 21:19:32 +000062static size_t baseDestLen;
63static int destDirFlag;
64static struct stat destStatBuf;
65
66static jmp_buf catch;
67static volatile int mv_Action_first_time;
68
69static void name_too_long__exit (void) __attribute__((noreturn));
70
71static
72void name_too_long__exit (void)
73{
Mark Whitleyf57c9442000-12-07 19:56:48 +000074 error_msg_and_die(name_too_long);
Erik Andersen029011b2000-03-04 21:19:32 +000075}
76
77static void
78fill_baseDest_buf(char *_buf, size_t * _buflen) {
79 const char *srcBasename;
80 if ((srcBasename = strrchr(baseSrcName, '/')) == NULL) {
81 srcBasename = baseSrcName;
82 if (_buf[*_buflen - 1] != '/') {
Erik Andersen4f3f7572000-04-28 00:18:56 +000083 if (++(*_buflen) > BUFSIZ)
Erik Andersen029011b2000-03-04 21:19:32 +000084 name_too_long__exit();
85 strcat(_buf, "/");
86 }
87 }
Erik Andersen4f3f7572000-04-28 00:18:56 +000088 if (*_buflen + strlen(srcBasename) > BUFSIZ)
Erik Andersen029011b2000-03-04 21:19:32 +000089 name_too_long__exit();
90 strcat(_buf, srcBasename);
91 return;
92
93}
94
95static int
Erik Andersen3364d782000-03-28 00:58:14 +000096cp_mv_Action(const char *fileName, struct stat *statbuf, void* junk)
Erik Andersen029011b2000-03-04 21:19:32 +000097{
Erik Andersen4f3f7572000-04-28 00:18:56 +000098 char destName[BUFSIZ + 1];
Erik Andersen029011b2000-03-04 21:19:32 +000099 size_t destLen;
100 const char *srcBasename;
101 char *name;
102
Mark Whitley59ab0252001-01-23 22:30:04 +0000103 strcpy(destName, pBaseDestName);
Erik Andersen029011b2000-03-04 21:19:32 +0000104 destLen = strlen(destName);
105
106 if (srcDirFlag == TRUE) {
107 if (recursiveFlag == FALSE) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000108 error_msg(omitting_directory, baseSrcName);
Erik Andersen029011b2000-03-04 21:19:32 +0000109 return TRUE;
110 }
111 srcBasename = (strstr(fileName, baseSrcName)
112 + strlen(baseSrcName));
113
Erik Andersen4f3f7572000-04-28 00:18:56 +0000114 if (destLen + strlen(srcBasename) > BUFSIZ) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000115 error_msg(name_too_long);
Erik Andersen029011b2000-03-04 21:19:32 +0000116 return FALSE;
117 }
118 strcat(destName, srcBasename);
119 }
120 else if (destDirFlag == TRUE) {
121 fill_baseDest_buf(&destName[0], &destLen);
122 }
123 else {
124 srcBasename = baseSrcName;
125 }
126 if (mv_Action_first_time && (dz_i == is_mv)) {
127 mv_Action_first_time = errno = 0;
128 if (rename(fileName, destName) < 0 && errno != EXDEV) {
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000129 perror_msg("rename(%s, %s)", fileName, destName);
Erik Andersen029011b2000-03-04 21:19:32 +0000130 goto do_copyFile; /* Try anyway... */
131 }
132 else if (errno == EXDEV)
133 goto do_copyFile;
134 else
135 longjmp(catch, 1); /* succeeded with rename() */
136 }
137 do_copyFile:
138 if (preserveFlag == TRUE && statbuf->st_nlink > 1) {
139 if (is_in_ino_dev_hashtable(statbuf, &name)) {
140 if (link(name, destName) < 0) {
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000141 perror_msg("link(%s, %s)", name, destName);
Erik Andersen029011b2000-03-04 21:19:32 +0000142 return FALSE;
143 }
144 return TRUE;
145 }
146 else {
147 add_to_ino_dev_hashtable(statbuf, destName);
148 }
149 }
Mark Whitleyf57c9442000-12-07 19:56:48 +0000150 return copy_file(fileName, destName, preserveFlag, followLinks, forceFlag);
Erik Andersen029011b2000-03-04 21:19:32 +0000151}
152
153static int
Erik Andersen3364d782000-03-28 00:58:14 +0000154rm_Action(const char *fileName, struct stat *statbuf, void* junk)
Erik Andersen029011b2000-03-04 21:19:32 +0000155{
156 int status = TRUE;
157
158 if (S_ISDIR(statbuf->st_mode)) {
159 if (rmdir(fileName) < 0) {
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000160 perror_msg("rmdir(%s)", fileName);
Erik Andersen029011b2000-03-04 21:19:32 +0000161 status = FALSE;
162 }
163 } else if (unlink(fileName) < 0) {
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000164 perror_msg("unlink(%s)", fileName);
Erik Andersen029011b2000-03-04 21:19:32 +0000165 status = FALSE;
166 }
167 return status;
168}
169
Erik Andersenfac10d72000-02-07 05:29:42 +0000170extern int cp_mv_main(int argc, char **argv)
171{
Eric Andersene7413a92000-07-14 06:19:41 +0000172 volatile int i;
Pavel Roskin93c0d9f2000-07-14 15:18:57 +0000173 int c;
Eric Andersend35c2152001-01-25 23:49:09 +0000174 RESERVE_BB_BUFFER(baseDestName,BUFSIZ + 1);
175 pBaseDestName = baseDestName; /* available globally */
Eric Andersenf4c02262000-07-13 18:42:58 +0000176
Matt Kraaie714bce2000-07-12 00:53:06 +0000177 if (*applet_name == 'c' && *(applet_name + 1) == 'p')
Erik Andersene49d5ec2000-02-08 19:58:47 +0000178 dz_i = is_cp;
179 else
180 dz_i = is_mv;
181 if (argc < 3)
Eric Andersen67991cf2001-02-14 21:23:06 +0000182 show_usage();
Erik Andersene49d5ec2000-02-08 19:58:47 +0000183
184 if (dz_i == is_cp) {
Erik Andersen59b9e872000-05-10 05:05:45 +0000185 recursiveFlag = preserveFlag = forceFlag = FALSE;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000186 followLinks = TRUE;
Eric Andersenf4c02262000-07-13 18:42:58 +0000187 while ((c = getopt(argc, argv, "adpRf")) != EOF) {
188 switch (c) {
Pavel Roskinb3d235f2000-06-08 18:06:37 +0000189 case 'a':
190 followLinks = FALSE;
191 preserveFlag = TRUE;
192 recursiveFlag = TRUE;
193 break;
194 case 'd':
195 followLinks = FALSE;
196 break;
197 case 'p':
198 preserveFlag = TRUE;
199 break;
200 case 'R':
201 recursiveFlag = TRUE;
202 break;
203 case 'f':
204 forceFlag = TRUE;
205 break;
206 default:
Eric Andersen67991cf2001-02-14 21:23:06 +0000207 show_usage();
Erik Andersene49d5ec2000-02-08 19:58:47 +0000208 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000209 }
Eric Andersenf4c02262000-07-13 18:42:58 +0000210 if ((argc - optind) < 2) {
Eric Andersen67991cf2001-02-14 21:23:06 +0000211 show_usage();
Eric Andersen815e9042000-06-06 16:15:23 +0000212 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000213 } else { /* (dz_i == is_mv) */
Eric Andersen2da90d32000-09-23 20:55:59 +0000214 /* Initialize optind to 1, since in libc5 optind
215 * is not initialized until getopt() is called
216 * (or until sneaky programmers force it...). */
217 optind = 1;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000218 recursiveFlag = preserveFlag = TRUE;
219 followLinks = FALSE;
220 }
Eric Andersen815e9042000-06-06 16:15:23 +0000221
Erik Andersene49d5ec2000-02-08 19:58:47 +0000222
Erik Andersen4f3f7572000-04-28 00:18:56 +0000223 if (strlen(argv[argc - 1]) > BUFSIZ) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000224 error_msg(name_too_long);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000225 goto exit_false;
226 }
227 strcpy(baseDestName, argv[argc - 1]);
228 baseDestLen = strlen(baseDestName);
229 if (baseDestLen == 0)
230 goto exit_false;
231
Mark Whitleyf57c9442000-12-07 19:56:48 +0000232 destDirFlag = is_directory(baseDestName, TRUE, &destStatBuf);
Matt Kraaiadbf0182000-08-06 15:36:50 +0000233 if (argc - optind > 2 && destDirFlag == FALSE) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000234 error_msg(not_a_directory, baseDestName);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000235 goto exit_false;
236 }
237
Eric Andersenf4c02262000-07-13 18:42:58 +0000238 for (i = optind; i < (argc-1); i++) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000239 size_t srcLen;
Erik Andersen029011b2000-03-04 21:19:32 +0000240 volatile int flags_memo;
241 int status;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000242
Eric Andersenf4c02262000-07-13 18:42:58 +0000243 baseSrcName=argv[i];
Erik Andersene49d5ec2000-02-08 19:58:47 +0000244
Erik Andersen4f3f7572000-04-28 00:18:56 +0000245 if ((srcLen = strlen(baseSrcName)) > BUFSIZ)
Erik Andersen029011b2000-03-04 21:19:32 +0000246 name_too_long__exit();
Erik Andersene49d5ec2000-02-08 19:58:47 +0000247
Erik Andersen029011b2000-03-04 21:19:32 +0000248 if (srcLen == 0) continue; /* "" */
Erik Andersene49d5ec2000-02-08 19:58:47 +0000249
Mark Whitleyf57c9442000-12-07 19:56:48 +0000250 srcDirFlag = is_directory(baseSrcName, followLinks, &srcStatBuf);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000251
252 if ((flags_memo = (recursiveFlag == TRUE &&
253 srcDirFlag == TRUE && destDirFlag == TRUE))) {
Erik Andersen029011b2000-03-04 21:19:32 +0000254
255 struct stat sb;
256 int state = 0;
257 char *pushd, *d, *p;
258
Erik Andersen4f3f7572000-04-28 00:18:56 +0000259 if ((pushd = getcwd(NULL, BUFSIZ + 1)) == NULL) {
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000260 perror_msg("getcwd()");
Erik Andersen029011b2000-03-04 21:19:32 +0000261 continue;
262 }
263 if (chdir(baseDestName) < 0) {
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000264 perror_msg("chdir(%s)", baseSrcName);
Erik Andersen029011b2000-03-04 21:19:32 +0000265 continue;
266 }
Erik Andersen4f3f7572000-04-28 00:18:56 +0000267 if ((d = getcwd(NULL, BUFSIZ + 1)) == NULL) {
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000268 perror_msg("getcwd()");
Erik Andersen029011b2000-03-04 21:19:32 +0000269 continue;
270 }
271 while (!state && *d != '\0') {
272 if (stat(d, &sb) < 0) { /* stat not lstat - always dereference targets */
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000273 perror_msg("stat(%s)", d);
Erik Andersen029011b2000-03-04 21:19:32 +0000274 state = -1;
275 continue;
Erik Andersen9ffdaa62000-02-11 21:55:04 +0000276 }
Erik Andersen029011b2000-03-04 21:19:32 +0000277 if ((sb.st_ino == srcStatBuf.st_ino) &&
278 (sb.st_dev == srcStatBuf.st_dev)) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000279 error_msg("Cannot %s `%s' into a subdirectory of itself, "
Matt Kraaidd19c692001-01-31 19:00:21 +0000280 "`%s/%s'", applet_name, baseSrcName,
Matt Kraaid537a952000-07-14 01:51:25 +0000281 baseDestName, baseSrcName);
Erik Andersen029011b2000-03-04 21:19:32 +0000282 state = -1;
283 continue;
284 }
285 if ((p = strrchr(d, '/')) != NULL) {
286 *p = '\0';
287 }
288 }
289 if (chdir(pushd) < 0) {
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000290 perror_msg("chdir(%s)", pushd);
Erik Andersen029011b2000-03-04 21:19:32 +0000291 free(pushd);
292 free(d);
293 continue;
294 }
295 free(pushd);
296 free(d);
297 if (state < 0)
298 continue;
299 else
Erik Andersen9ffdaa62000-02-11 21:55:04 +0000300 fill_baseDest_buf(baseDestName, &baseDestLen);
Erik Andersenfac10d72000-02-07 05:29:42 +0000301 }
Erik Andersen029011b2000-03-04 21:19:32 +0000302 status = setjmp(catch);
303 if (status == 0) {
304 mv_Action_first_time = 1;
Mark Whitleyf57c9442000-12-07 19:56:48 +0000305 if (recursive_action(baseSrcName,
Erik Andersen029011b2000-03-04 21:19:32 +0000306 recursiveFlag, followLinks, FALSE,
Erik Andersen3364d782000-03-28 00:58:14 +0000307 cp_mv_Action, cp_mv_Action, NULL) == FALSE) goto exit_false;
Erik Andersen029011b2000-03-04 21:19:32 +0000308 if (dz_i == is_mv &&
Mark Whitleyf57c9442000-12-07 19:56:48 +0000309 recursive_action(baseSrcName,
Erik Andersen029011b2000-03-04 21:19:32 +0000310 recursiveFlag, followLinks, TRUE,
Erik Andersen3364d782000-03-28 00:58:14 +0000311 rm_Action, rm_Action, NULL) == FALSE) goto exit_false;
Erik Andersen029011b2000-03-04 21:19:32 +0000312 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000313 if (flags_memo)
314 *(baseDestName + baseDestLen) = '\0';
Erik Andersenfac10d72000-02-07 05:29:42 +0000315 }
Matt Kraai3e856ce2000-12-01 02:55:13 +0000316 return EXIT_SUCCESS;
Erik Andersen029011b2000-03-04 21:19:32 +0000317 exit_false:
Matt Kraai3e856ce2000-12-01 02:55:13 +0000318 return EXIT_FAILURE;
Erik Andersenfac10d72000-02-07 05:29:42 +0000319}
320
Erik Andersen029011b2000-03-04 21:19:32 +0000321/*
322Local Variables:
323c-file-style: "linux"
324c-basic-offset: 4
325tab-width: 4
326End:
327*/