1 /* vi: set sw=4 ts=4: */
3 * Mini `cp' and `mv' implementation for BusyBox.
6 * Copyright (C) 1999 by Lineo, inc.
7 * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
9 * Copyright (C) 2000 by BitterSweet Enterprises, LLC. (GPL)
10 * Extensively modified and rewritten by Karl M. Hegbloom <karlheg@debian.org>
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.
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.
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
29 #define BB_DECLARE_EXTERN
30 #define bb_need_name_too_long
31 #define bb_need_omitting_directory
32 #define bb_need_not_a_directory
39 #include <sys/param.h>
47 static int dz_i; /* index into cp_mv_usage */
48 static const char *dz; /* dollar zero, .bss */
49 static const char *cp_mv_usage[] = /* .rodata */
51 "cp [OPTION]... SOURCE DEST\n"
52 " or: cp [OPTION]... SOURCE... DIRECTORY\n"
53 #ifndef BB_FEATURE_TRIVIAL_HELP
54 "\nCopies SOURCE to DEST, or multiple SOURCE(s) to DIRECTORY.\n"
56 "\t-a\tSame as -dpR\n"
57 "\t-d\tPreserves links\n"
58 "\t-p\tPreserves file attributes if possible\n"
59 "\t-f\tforce (implied; ignored) - always set\n"
60 "\t-R\tCopies directories recursively\n"
64 " or: mv SOURCE... DIRECTORY\n"
65 #ifndef BB_FEATURE_TRIVIAL_HELP
66 "\nRename SOURCE to DEST, or move SOURCE(s) to DIRECTORY.\n"
70 static int recursiveFlag;
71 static int followLinks;
72 static int preserveFlag;
75 static const char *baseSrcName;
76 static int srcDirFlag;
77 static struct stat srcStatBuf;
79 static char baseDestName[BUFSIZ + 1];
80 static size_t baseDestLen;
81 static int destDirFlag;
82 static struct stat destStatBuf;
85 static volatile int mv_Action_first_time;
87 static void name_too_long__exit (void) __attribute__((noreturn));
90 void name_too_long__exit (void)
92 fprintf(stderr, name_too_long, dz);
97 fill_baseDest_buf(char *_buf, size_t * _buflen) {
98 const char *srcBasename;
99 if ((srcBasename = strrchr(baseSrcName, '/')) == NULL) {
100 srcBasename = baseSrcName;
101 if (_buf[*_buflen - 1] != '/') {
102 if (++(*_buflen) > BUFSIZ)
103 name_too_long__exit();
107 if (*_buflen + strlen(srcBasename) > BUFSIZ)
108 name_too_long__exit();
109 strcat(_buf, srcBasename);
115 cp_mv_Action(const char *fileName, struct stat *statbuf, void* junk)
117 char destName[BUFSIZ + 1];
119 const char *srcBasename;
122 strcpy(destName, baseDestName);
123 destLen = strlen(destName);
125 if (srcDirFlag == TRUE) {
126 if (recursiveFlag == FALSE) {
127 fprintf(stderr, omitting_directory, dz, baseSrcName);
130 srcBasename = (strstr(fileName, baseSrcName)
131 + strlen(baseSrcName));
133 if (destLen + strlen(srcBasename) > BUFSIZ) {
134 fprintf(stderr, name_too_long, dz);
137 strcat(destName, srcBasename);
139 else if (destDirFlag == TRUE) {
140 fill_baseDest_buf(&destName[0], &destLen);
143 srcBasename = baseSrcName;
145 if (mv_Action_first_time && (dz_i == is_mv)) {
146 mv_Action_first_time = errno = 0;
147 if (rename(fileName, destName) < 0 && errno != EXDEV) {
148 fprintf(stderr, "%s: rename(%s, %s): %s\n",
149 dz, fileName, destName, strerror(errno));
150 goto do_copyFile; /* Try anyway... */
152 else if (errno == EXDEV)
155 longjmp(catch, 1); /* succeeded with rename() */
158 if (preserveFlag == TRUE && statbuf->st_nlink > 1) {
159 if (is_in_ino_dev_hashtable(statbuf, &name)) {
160 if (link(name, destName) < 0) {
161 fprintf(stderr, "%s: link(%s, %s): %s\n",
162 dz, name, destName, strerror(errno));
168 add_to_ino_dev_hashtable(statbuf, destName);
171 return copyFile(fileName, destName, preserveFlag, followLinks, forceFlag);
175 rm_Action(const char *fileName, struct stat *statbuf, void* junk)
179 if (S_ISDIR(statbuf->st_mode)) {
180 if (rmdir(fileName) < 0) {
181 fprintf(stderr, "%s: rmdir(%s): %s\n", dz, fileName, strerror(errno));
184 } else if (unlink(fileName) < 0) {
185 fprintf(stderr, "%s: unlink(%s): %s\n", dz, fileName, strerror(errno));
191 extern int cp_mv_main(int argc, char **argv)
193 dz = *argv; /* already basename'd by busybox.c:main */
194 if (*dz == 'c' && *(dz + 1) == 'p')
199 usage(cp_mv_usage[dz_i]);
204 recursiveFlag = preserveFlag = forceFlag = FALSE;
206 while (--argc >= 0 && *argv && **argv) {
207 while (**argv == '-') {
213 recursiveFlag = TRUE;
222 recursiveFlag = TRUE;
228 usage(cp_mv_usage[is_cp]);
235 fatalError("cp: missing file argument\n");
237 } else { /* (dz_i == is_mv) */
238 recursiveFlag = preserveFlag = TRUE;
243 if (strlen(argv[argc - 1]) > BUFSIZ) {
244 fprintf(stderr, name_too_long, "cp");
247 strcpy(baseDestName, argv[argc - 1]);
248 baseDestLen = strlen(baseDestName);
249 if (baseDestLen == 0)
252 destDirFlag = isDirectory(baseDestName, TRUE, &destStatBuf);
253 if ((argc > 3) && destDirFlag == FALSE) {
254 fprintf(stderr, not_a_directory, "cp", baseDestName);
260 volatile int flags_memo;
263 baseSrcName = *(argv++);
265 if ((srcLen = strlen(baseSrcName)) > BUFSIZ)
266 name_too_long__exit();
268 if (srcLen == 0) continue; /* "" */
270 srcDirFlag = isDirectory(baseSrcName, followLinks, &srcStatBuf);
272 if ((flags_memo = (recursiveFlag == TRUE &&
273 srcDirFlag == TRUE && destDirFlag == TRUE))) {
279 if ((pushd = getcwd(NULL, BUFSIZ + 1)) == NULL) {
280 fprintf(stderr, "%s: getcwd(): %s\n", dz, strerror(errno));
283 if (chdir(baseDestName) < 0) {
284 fprintf(stderr, "%s: chdir(%s): %s\n", dz, baseSrcName, strerror(errno));
287 if ((d = getcwd(NULL, BUFSIZ + 1)) == NULL) {
288 fprintf(stderr, "%s: getcwd(): %s\n", dz, strerror(errno));
291 while (!state && *d != '\0') {
292 if (stat(d, &sb) < 0) { /* stat not lstat - always dereference targets */
293 fprintf(stderr, "%s: stat(%s) :%s\n", dz, d, strerror(errno));
297 if ((sb.st_ino == srcStatBuf.st_ino) &&
298 (sb.st_dev == srcStatBuf.st_dev)) {
300 "%s: Cannot %s `%s' "
301 "into a subdirectory of itself, `%s/%s'\n",
302 dz, dz, baseSrcName, baseDestName, baseSrcName);
306 if ((p = strrchr(d, '/')) != NULL) {
310 if (chdir(pushd) < 0) {
311 fprintf(stderr, "%s: chdir(%s): %s\n", dz, pushd, strerror(errno));
321 fill_baseDest_buf(baseDestName, &baseDestLen);
323 status = setjmp(catch);
325 mv_Action_first_time = 1;
326 if (recursiveAction(baseSrcName,
327 recursiveFlag, followLinks, FALSE,
328 cp_mv_Action, cp_mv_Action, NULL) == FALSE) goto exit_false;
330 recursiveAction(baseSrcName,
331 recursiveFlag, followLinks, TRUE,
332 rm_Action, rm_Action, NULL) == FALSE) goto exit_false;
335 *(baseDestName + baseDestLen) = '\0';
345 c-file-style: "linux"