1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini cp implementation for busybox
4  *
5  * Copyright (C) 2000 by Matt Kraai <kraai@alumni.carnegiemellon.edu>
6  * SELinux support by Yuichi Nakamura <ynakam@hitachisoft.jp>
7  *
8  * Licensed under GPLv2 or later, see file LICENSE in this source tree.
9  */
10 /* Mar 16, 2003      Manuel Novoa III   (mjn3@codepoet.org)
11  *
12  * Size reduction.
13  */
14 //config:config CP
15 //config:	bool "cp (10 kb)"
16 //config:	default y
17 //config:	help
18 //config:	cp is used to copy files and directories.
19 //config:
20 //config:config FEATURE_CP_LONG_OPTIONS
21 //config:	bool "Enable long options"
22 //config:	default y
23 //config:	depends on CP && LONG_OPTS
24 //config:	help
25 //config:	Enable long options.
26 //config:	Also add support for --parents option.
27 //config:
28 //config:config FEATURE_CP_REFLINK
29 //config:	bool "Enable --reflink[=auto]"
30 //config:	default y
31 //config:	depends on FEATURE_CP_LONG_OPTIONS
32 
33 //applet:IF_CP(APPLET_NOEXEC(cp, cp, BB_DIR_BIN, BB_SUID_DROP, cp))
34 /* NOEXEC despite cases when it can be a "runner" (cp -r LARGE_DIR NEW_DIR) */
35 
36 //kbuild:lib-$(CONFIG_CP) += cp.o
37 
38 /* http://www.opengroup.org/onlinepubs/007904975/utilities/cp.html */
39 
40 // Options of cp from GNU coreutils 6.10:
41 // -a, --archive
42 // -f, --force
43 // -i, --interactive
44 // -l, --link
45 // -L, --dereference
46 // -P, --no-dereference
47 // -R, -r, --recursive
48 // -s, --symbolic-link
49 // -v, --verbose
50 // -H	follow command-line symbolic links in SOURCE
51 // -d	same as --no-dereference --preserve=links
52 // -p	same as --preserve=mode,ownership,timestamps
53 // -c	same as --preserve=context
54 // -u, --update
55 //	copy only when the SOURCE file is newer than the destination
56 //	file or when the destination file is missing
57 // --remove-destination
58 //	remove each existing destination file before attempting to open
59 // --parents
60 //	use full source file name under DIRECTORY
61 // -T, --no-target-directory
62 //	treat DEST as a normal file
63 // NOT SUPPORTED IN BBOX:
64 // --backup[=CONTROL]
65 //	make a backup of each existing destination file
66 // -b	like --backup but does not accept an argument
67 // --copy-contents
68 //	copy contents of special files when recursive
69 // --preserve[=ATTR_LIST]
70 //	preserve attributes (default: mode,ownership,timestamps),
71 //	if possible additional attributes: security context,links,all
72 // --no-preserve=ATTR_LIST
73 // --sparse=WHEN
74 //	control creation of sparse files
75 // --strip-trailing-slashes
76 //	remove any trailing slashes from each SOURCE argument
77 // -S, --suffix=SUFFIX
78 //	override the usual backup suffix
79 // -t, --target-directory=DIRECTORY
80 //	copy all SOURCE arguments into DIRECTORY
81 // -x, --one-file-system
82 //	stay on this file system
83 // -Z, --context=CONTEXT
84 //	(SELinux) set SELinux security context of copy to CONTEXT
85 
86 //usage:#define cp_trivial_usage
87 //usage:       "[-arPLHpfinlsTu] SOURCE DEST\n"
88 //usage:       "or: cp [-arPLHpfinlsu] SOURCE... { -t DIRECTORY | DIRECTORY }"
89 //usage:#define cp_full_usage "\n\n"
90 //usage:       "Copy SOURCEs to DEST\n"
91 //usage:     "\n	-a	Same as -dpR"
92 //usage:	IF_SELINUX(
93 //usage:     "\n	-c	Preserve security context"
94 //usage:	)
95 //usage:     "\n	-R,-r	Recurse"
96 //usage:     "\n	-d,-P	Preserve symlinks (default if -R)"
97 //usage:     "\n	-L	Follow all symlinks"
98 //usage:     "\n	-H	Follow symlinks on command line"
99 //usage:     "\n	-p	Preserve file attributes if possible"
100 //usage:     "\n	-f	Overwrite"
101 //usage:     "\n	-i	Prompt before overwrite"
102 //usage:     "\n	-n	Don't overwrite"
103 //usage:     "\n	-l,-s	Create (sym)links"
104 //usage:     "\n	-T	Refuse to copy if DEST is a directory"
105 //usage:     "\n	-t DIR	Copy all SOURCEs into DIR"
106 //usage:     "\n	-u	Copy only newer files"
107 
108 #include "libbb.h"
109 #include "libcoreutils/coreutils.h"
110 
111 /* This is a NOEXEC applet. Be very careful! */
112 
113 int cp_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
cp_main(int argc,char ** argv)114 int cp_main(int argc, char **argv)
115 {
116 	struct stat source_stat;
117 	struct stat dest_stat;
118 	const char *last;
119 	const char *dest;
120 	int s_flags;
121 	int d_flags;
122 	int flags;
123 	int status;
124 #if ENABLE_FEATURE_CP_LONG_OPTIONS
125 	enum {
126 		/*OPT_rmdest  = FILEUTILS_RMDEST = 1 << FILEUTILS_CP_OPTBITS */
127 		OPT_parents = 1 << (FILEUTILS_CP_OPTBITS+1),
128 		OPT_reflink = 1 << (FILEUTILS_CP_OPTBITS+2),
129 	};
130 # if ENABLE_FEATURE_CP_REFLINK
131 	char *reflink = NULL;
132 # endif
133 	flags = getopt32long(argv, "^"
134 		FILEUTILS_CP_OPTSTR
135 		"\0"
136 		// Need at least one argument. (Usually two+, but -t DIR can have only one)
137 		// Soft- and hardlinking doesn't mix
138 		// -P and -d are the same (-P is POSIX, -d is GNU)
139 		// -r and -R are the same
140 		// -R (and therefore -r) turns on -d (coreutils does this)
141 		// -a = -pdR
142 		// -i overrides -n and vice versa (last wins)
143 		"-1:l--s:s--l:Pd:rRd:Rd:apdR:i-n:n-i",
144 		"archive\0"        No_argument "a"
145 		"force\0"          No_argument "f"
146 		"interactive\0"    No_argument "i"
147 		"no-clobber\0"     No_argument "n"
148 		"link\0"           No_argument "l"
149 		"dereference\0"    No_argument "L"
150 		"no-dereference\0" No_argument "P"
151 		"recursive\0"      No_argument "R"
152 		"symbolic-link\0"  No_argument "s"
153 		"no-target-directory\0" No_argument "T"
154 		"target-directory\0" Required_argument "t"
155 		"verbose\0"        No_argument "v"
156 		"update\0"         No_argument "u"
157 		"remove-destination\0" No_argument "\xff"
158 		"parents\0"        No_argument "\xfe"
159 # if ENABLE_FEATURE_CP_REFLINK
160 		"reflink\0"        Optional_argument "\xfd"
161 # endif
162 		, &last
163 # if ENABLE_FEATURE_CP_REFLINK
164 		, &reflink
165 # endif
166 	);
167 # if ENABLE_FEATURE_CP_REFLINK
168 	BUILD_BUG_ON((int)OPT_reflink != (int)FILEUTILS_REFLINK);
169 	if (flags & FILEUTILS_REFLINK) {
170 		if (!reflink)
171 			flags |= FILEUTILS_REFLINK_ALWAYS;
172 		else if (strcmp(reflink, "always") == 0)
173 			flags |= FILEUTILS_REFLINK_ALWAYS;
174 		else if (strcmp(reflink, "auto") != 0)
175 			bb_show_usage();
176 	}
177 # endif
178 #else
179 	flags = getopt32(argv, "^"
180 		FILEUTILS_CP_OPTSTR
181 		"\0"
182 		"-1:l--s:s--l:Pd:rRd:Rd:apdR"
183 		, &last
184 	);
185 #endif
186 	argc -= optind;
187 	argv += optind;
188 	/* Reverse this bit. If there is -d, bit is not set: */
189 	flags ^= FILEUTILS_DEREFERENCE;
190 	/* coreutils 6.9 compat:
191 	 * by default, "cp" derefs symlinks (creates regular dest files),
192 	 * but "cp -R" does not. We switch off deref if -r or -R (see above).
193 	 * However, "cp -RL" must still deref symlinks: */
194 	if (flags & FILEUTILS_DEREF_SOFTLINK) /* -L */
195 		flags |= FILEUTILS_DEREFERENCE;
196 
197 #if ENABLE_SELINUX
198 	if (flags & FILEUTILS_PRESERVE_SECURITY_CONTEXT) {
199 		selinux_or_die();
200 	}
201 #endif
202 
203 	status = EXIT_SUCCESS;
204 	if (!(flags & FILEUTILS_TARGET_DIR)) {
205 		last = argv[argc - 1];
206 		if (argc < 2)
207 			bb_show_usage();
208 		if (argc != 2) {
209 			if (flags & FILEUTILS_NO_TARGET_DIR)
210 				bb_show_usage();
211 			/* "cp A B C... DIR" - target must be dir */
212 		} else /* argc == 2 */ {
213 			/* "cp A B" - only case where target can be not a dir */
214 			s_flags = cp_mv_stat2(*argv, &source_stat,
215 					(flags & FILEUTILS_DEREFERENCE) ? stat : lstat);
216 			if (s_flags < 0) /* error other than ENOENT */
217 				return EXIT_FAILURE;
218 			d_flags = cp_mv_stat(last, &dest_stat);
219 			if (d_flags < 0) /* error other than ENOENT */
220 				return EXIT_FAILURE;
221 
222 			if (flags & FILEUTILS_NO_TARGET_DIR) { /* -T */
223 				if (!(s_flags & 2) && (d_flags & 2))
224 					/* cp -T NOTDIR DIR */
225 					bb_error_msg_and_die("'%s' is a directory", last);
226 			}
227 
228 #if ENABLE_FEATURE_CP_LONG_OPTIONS
229 			//bb_error_msg("flags:%x FILEUTILS_RMDEST:%x OPT_parents:%x",
230 			//	flags, FILEUTILS_RMDEST, OPT_parents);
231 			if (flags & OPT_parents) {
232 				if (!(d_flags & 2)) {
233 					bb_simple_error_msg_and_die("with --parents, the destination must be a directory");
234 				}
235 			}
236 			if (flags & FILEUTILS_RMDEST) {
237 				flags |= FILEUTILS_FORCE;
238 			}
239 #endif
240 
241 			/* ...if neither is a directory...  */
242 			if (!((s_flags | d_flags) & 2)
243 			    /* ...or: recursing, the 1st is a directory, and the 2nd doesn't exist... */
244 			 || ((flags & FILEUTILS_RECUR) && (s_flags & 2) && !d_flags)
245 			 || (flags & FILEUTILS_NO_TARGET_DIR)
246 			) {
247 				/* Do a simple copy */
248 				dest = last;
249 				goto DO_COPY; /* NB: argc==2 -> *++argv==last */
250 			}
251 		}
252 	}
253 	/* else: last is DIR from "-t DIR" */
254 
255 	while (1) {
256 #if ENABLE_FEATURE_CP_LONG_OPTIONS
257 		if (flags & OPT_parents) {
258 			char *dest_dup;
259 			char *dest_dir;
260 			dest = concat_path_file(last, *argv);
261 			dest_dup = xstrdup(dest);
262 			dest_dir = dirname(dest_dup);
263 			if (bb_make_directory(dest_dir, -1, FILEUTILS_RECUR)) {
264 				return EXIT_FAILURE;
265 			}
266 			free(dest_dup);
267 			goto DO_COPY;
268 		}
269 #endif
270 		dest = concat_path_file(last, bb_get_last_path_component_strip(*argv));
271  DO_COPY:
272 		if (copy_file(*argv, dest, flags) < 0) {
273 			status = EXIT_FAILURE;
274 		}
275 		if (!*++argv || *argv == last) {
276 			/* possibly leaking dest... */
277 			break;
278 		}
279 		/* don't move up: dest may be == last and not malloced! */
280 		free((void*)dest);
281 	}
282 
283 	/* Exit. We are NOEXEC, not NOFORK. We do exit at the end of main() */
284 	return status;
285 }
286