busybox/coreutils/install.c

/* vi: set sw=4 ts=4: */
/*
 * Copyright (C) 2003 by Glenn McGrath
 * SELinux support: by Yuichi Nakamura <ynakam@hitachisoft.jp>
 *
 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
 */

#include "libbb.h"
#include "libcoreutils/coreutils.h"

#if ENABLE_FEATURE_INSTALL_LONG_OPTIONS
static const char install_longopts[] ALIGN1 =
	"directory\0"           No_argument       "d"
	"preserve-timestamps\0" No_argument       "p"
	"strip\0"               No_argument       "s"
	"group\0"               Required_argument "g"
	"mode\0"                Required_argument "m"
	"owner\0"               Required_argument "o"
/* autofs build insists of using -b --suffix=.orig */
/* TODO? (short option for --suffix is -S) */
#if ENABLE_SELINUX
	"context\0"             Required_argument "Z"
	"preserve_context\0"    No_argument       "\xff"
	"preserve-context\0"    No_argument       "\xff"
#endif
	;
#endif


#if ENABLE_SELINUX
static void setdefaultfilecon(const char *path)
{
	struct stat s;
	security_context_t scontext = NULL;

	if (!is_selinux_enabled()) {
		return;
	}
	if (lstat(path, &s) != 0) {
		return;
	}

	if (matchpathcon(path, s.st_mode, &scontext) < 0) {
		goto out;
	}
	if (strcmp(scontext, "<<none>>") == 0) {
		goto out;
	}

	if (lsetfilecon(path, scontext) < 0) {
		if (errno != ENOTSUP) {
			bb_perror_msg("warning: failed to change context"
					" of %s to %s", path, scontext);
		}
	}

 out:
	freecon(scontext);
}

#endif

int install_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int install_main(int argc, char **argv)
{
	struct stat statbuf;
	mode_t mode;
	uid_t uid;
	gid_t gid;
	char *arg, *last;
	const char *gid_str;
	const char *uid_str;
	const char *mode_str;
	int copy_flags = FILEUTILS_DEREFERENCE | FILEUTILS_FORCE;
	int opts;
	int min_args = 1;
	int ret = EXIT_SUCCESS;
	int isdir = 0;
#if ENABLE_SELINUX
	security_context_t scontext;
	bool use_default_selinux_context = 1;
#endif
	enum {
		OPT_c             = 1 << 0,
		OPT_v             = 1 << 1,
		OPT_b             = 1 << 2,
		OPT_MKDIR_LEADING = 1 << 3,
		OPT_DIRECTORY     = 1 << 4,
		OPT_PRESERVE_TIME = 1 << 5,
		OPT_STRIP         = 1 << 6,
		OPT_GROUP         = 1 << 7,
		OPT_MODE          = 1 << 8,
		OPT_OWNER         = 1 << 9,
#if ENABLE_SELINUX
		OPT_SET_SECURITY_CONTEXT = 1 << 10,
		OPT_PRESERVE_SECURITY_CONTEXT = 1 << 11,
#endif
	};

#if ENABLE_FEATURE_INSTALL_LONG_OPTIONS
	applet_long_options = install_longopts;
#endif
	opt_complementary = "s--d:d--s" IF_FEATURE_INSTALL_LONG_OPTIONS(IF_SELINUX(":Z--\xff:\xff--Z"));
	/* -c exists for backwards compatibility, it's needed */
	/* -v is ignored ("print name of each created directory") */
	/* -b is ignored ("make a backup of each existing destination file") */
	opts = getopt32(argv, "cvb" "Ddpsg:m:o:" IF_SELINUX("Z:"),
			&gid_str, &mode_str, &uid_str IF_SELINUX(, &scontext));
	argc -= optind;
	argv += optind;

#if ENABLE_SELINUX
	if (opts & (OPT_PRESERVE_SECURITY_CONTEXT|OPT_SET_SECURITY_CONTEXT)) {
		selinux_or_die();
		use_default_selinux_context = 0;
		if (opts & OPT_PRESERVE_SECURITY_CONTEXT) {
			copy_flags |= FILEUTILS_PRESERVE_SECURITY_CONTEXT;
		}
		if (opts & OPT_SET_SECURITY_CONTEXT) {
			setfscreatecon_or_die(scontext);
			copy_flags |= FILEUTILS_SET_SECURITY_CONTEXT;
		}
	}
#endif

	/* preserve access and modification time, this is GNU behaviour,
	 * BSD only preserves modification time */
	if (opts & OPT_PRESERVE_TIME) {
		copy_flags |= FILEUTILS_PRESERVE_STATUS;
	}
	mode = 0755; /* GNU coreutils 6.10 compat */
	if (opts & OPT_MODE)
		bb_parse_mode(mode_str, &mode);
	uid = (opts & OPT_OWNER) ? get_ug_id(uid_str, xuname2uid) : getuid();
	gid = (opts & OPT_GROUP) ? get_ug_id(gid_str, xgroup2gid) : getgid();

	last = argv[argc - 1];
	if (!(opts & OPT_DIRECTORY)) {
		argv[argc - 1] = NULL;
		min_args++;

		/* coreutils install resolves link in this case, don't use lstat */
		isdir = stat(last, &statbuf) < 0 ? 0 : S_ISDIR(statbuf.st_mode);
	}

	if (argc < min_args)
		bb_show_usage();

	while ((arg = *argv++) != NULL) {
		char *dest = last;
		if (opts & OPT_DIRECTORY) {
			dest = arg;
			/* GNU coreutils 6.9 does not set uid:gid
			 * on intermediate created directories
			 * (only on last one) */
			if (bb_make_directory(dest, 0755, FILEUTILS_RECUR)) {
				ret = EXIT_FAILURE;
				goto next;
			}
		} else {
			if (opts & OPT_MKDIR_LEADING) {
				char *ddir = xstrdup(dest);
				bb_make_directory(dirname(ddir), 0755, FILEUTILS_RECUR);
				/* errors are not checked. copy_file
				 * will fail if dir is not created. */
				free(ddir);
			}
			if (isdir)
				dest = concat_path_file(last, basename(arg));
			if (copy_file(arg, dest, copy_flags)) {
				/* copy is not made */
				ret = EXIT_FAILURE;
				goto next;
			}
		}

		/* Set the file mode (always, not only with -m).
		 * GNU coreutils 6.10 is not affected by umask. */
		if (chmod(dest, mode) == -1) {
			bb_perror_msg("can't change %s of %s", "permissions", dest);
			ret = EXIT_FAILURE;
		}
#if ENABLE_SELINUX
		if (use_default_selinux_context)
			setdefaultfilecon(dest);
#endif
		/* Set the user and group id */
		if ((opts & (OPT_OWNER|OPT_GROUP))
		 && lchown(dest, uid, gid) == -1
		) {
			bb_perror_msg("can't change %s of %s", "ownership", dest);
			ret = EXIT_FAILURE;
		}
		if (opts & OPT_STRIP) {
			char *args[3];
			args[0] = (char*)"strip";
			args[1] = dest;
			args[2] = NULL;
			if (spawn_and_wait(args)) {
				bb_perror_msg("strip");
				ret = EXIT_FAILURE;
			}
		}
 next:
		if (ENABLE_FEATURE_CLEAN_UP && isdir)
			free(dest);
	}

	return ret;
}
1	niro	532	/* vi: set sw=4 ts=4: */
2			/*
3	niro	816	* Copyright (C) 2003 by Glenn McGrath
4			* SELinux support: by Yuichi Nakamura <ynakam@hitachisoft.jp>
5	niro	532	*
6			* Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
7			*/
8
9	niro	816	#include "libbb.h"
10	niro	532	#include "libcoreutils/coreutils.h"
11
12			#if ENABLE_FEATURE_INSTALL_LONG_OPTIONS
13	niro	816	static const char install_longopts[] ALIGN1 =
14			"directory\0" No_argument "d"
15			"preserve-timestamps\0" No_argument "p"
16			"strip\0" No_argument "s"
17			"group\0" Required_argument "g"
18			"mode\0" Required_argument "m"
19			"owner\0" Required_argument "o"
20			/* autofs build insists of using -b --suffix=.orig */
21			/* TODO? (short option for --suffix is -S) */
22			#if ENABLE_SELINUX
23			"context\0" Required_argument "Z"
24			"preserve_context\0" No_argument "\xff"
25			"preserve-context\0" No_argument "\xff"
26	niro	532	#endif
27	niro	816	;
28			#endif
29	niro	532
30	niro	816
31			#if ENABLE_SELINUX
32			static void setdefaultfilecon(const char *path)
33			{
34			struct stat s;
35			security_context_t scontext = NULL;
36
37			if (!is_selinux_enabled()) {
38			return;
39			}
40			if (lstat(path, &s) != 0) {
41			return;
42			}
43
44			if (matchpathcon(path, s.st_mode, &scontext) < 0) {
45			goto out;
46			}
47			if (strcmp(scontext, "<<none>>") == 0) {
48			goto out;
49			}
50
51			if (lsetfilecon(path, scontext) < 0) {
52			if (errno != ENOTSUP) {
53			bb_perror_msg("warning: failed to change context"
54			" of %s to %s", path, scontext);
55			}
56			}
57
58			out:
59			freecon(scontext);
60			}
61
62			#endif
63
64			int install_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
65	niro	532	int install_main(int argc, char **argv)
66			{
67			struct stat statbuf;
68			mode_t mode;
69			uid_t uid;
70			gid_t gid;
71	niro	816	char arg, last;
72	niro	532	const char *gid_str;
73			const char *uid_str;
74			const char *mode_str;
75			int copy_flags = FILEUTILS_DEREFERENCE \| FILEUTILS_FORCE;
76	niro	816	int opts;
77			int min_args = 1;
78			int ret = EXIT_SUCCESS;
79			int isdir = 0;
80			#if ENABLE_SELINUX
81			security_context_t scontext;
82			bool use_default_selinux_context = 1;
83			#endif
84	niro	532	enum {
85	niro	816	OPT_c = 1 << 0,
86			OPT_v = 1 << 1,
87			OPT_b = 1 << 2,
88			OPT_MKDIR_LEADING = 1 << 3,
89			OPT_DIRECTORY = 1 << 4,
90			OPT_PRESERVE_TIME = 1 << 5,
91			OPT_STRIP = 1 << 6,
92			OPT_GROUP = 1 << 7,
93			OPT_MODE = 1 << 8,
94			OPT_OWNER = 1 << 9,
95			#if ENABLE_SELINUX
96			OPT_SET_SECURITY_CONTEXT = 1 << 10,
97			OPT_PRESERVE_SECURITY_CONTEXT = 1 << 11,
98			#endif
99	niro	532	};
100
101			#if ENABLE_FEATURE_INSTALL_LONG_OPTIONS
102	niro	816	applet_long_options = install_longopts;
103	niro	532	#endif
104	niro	984	opt_complementary = "s--d:d--s" IF_FEATURE_INSTALL_LONG_OPTIONS(IF_SELINUX(":Z--\xff:\xff--Z"));
105	niro	816	/* -c exists for backwards compatibility, it's needed */
106			/* -v is ignored ("print name of each created directory") */
107			/* -b is ignored ("make a backup of each existing destination file") */
108	niro	984	opts = getopt32(argv, "cvb" "Ddpsg:m:o:" IF_SELINUX("Z:"),
109			&gid_str, &mode_str, &uid_str IF_SELINUX(, &scontext));
110	niro	816	argc -= optind;
111			argv += optind;
112	niro	532
113	niro	816	#if ENABLE_SELINUX
114			if (opts & (OPT_PRESERVE_SECURITY_CONTEXT\|OPT_SET_SECURITY_CONTEXT)) {
115			selinux_or_die();
116			use_default_selinux_context = 0;
117			if (opts & OPT_PRESERVE_SECURITY_CONTEXT) {
118			copy_flags \|= FILEUTILS_PRESERVE_SECURITY_CONTEXT;
119			}
120			if (opts & OPT_SET_SECURITY_CONTEXT) {
121			setfscreatecon_or_die(scontext);
122			copy_flags \|= FILEUTILS_SET_SECURITY_CONTEXT;
123			}
124			}
125			#endif
126
127			/* preserve access and modification time, this is GNU behaviour,
128			* BSD only preserves modification time */
129			if (opts & OPT_PRESERVE_TIME) {
130	niro	532	copy_flags \|= FILEUTILS_PRESERVE_STATUS;
131			}
132	niro	984	mode = 0755; /* GNU coreutils 6.10 compat */
133	niro	816	if (opts & OPT_MODE)
134			bb_parse_mode(mode_str, &mode);
135			uid = (opts & OPT_OWNER) ? get_ug_id(uid_str, xuname2uid) : getuid();
136			gid = (opts & OPT_GROUP) ? get_ug_id(gid_str, xgroup2gid) : getgid();
137	niro	532
138	niro	816	last = argv[argc - 1];
139			if (!(opts & OPT_DIRECTORY)) {
140			argv[argc - 1] = NULL;
141			min_args++;
142
143			/* coreutils install resolves link in this case, don't use lstat */
144			isdir = stat(last, &statbuf) < 0 ? 0 : S_ISDIR(statbuf.st_mode);
145	niro	532	}
146
147	niro	816	if (argc < min_args)
148			bb_show_usage();
149	niro	532
150	niro	816	while ((arg = *argv++) != NULL) {
151			char *dest = last;
152			if (opts & OPT_DIRECTORY) {
153			dest = arg;
154			/* GNU coreutils 6.9 does not set uid:gid
155			* on intermediate created directories
156			* (only on last one) */
157			if (bb_make_directory(dest, 0755, FILEUTILS_RECUR)) {
158			ret = EXIT_FAILURE;
159			goto next;
160			}
161			} else {
162			if (opts & OPT_MKDIR_LEADING) {
163			char *ddir = xstrdup(dest);
164			bb_make_directory(dirname(ddir), 0755, FILEUTILS_RECUR);
165			/* errors are not checked. copy_file
166			* will fail if dir is not created. */
167			free(ddir);
168			}
169			if (isdir)
170			dest = concat_path_file(last, basename(arg));
171			if (copy_file(arg, dest, copy_flags)) {
172			/* copy is not made */
173			ret = EXIT_FAILURE;
174			goto next;
175			}
176			}
177	niro	532
178	niro	984	/* Set the file mode (always, not only with -m).
179			* GNU coreutils 6.10 is not affected by umask. */
180			if (chmod(dest, mode) == -1) {
181	niro	816	bb_perror_msg("can't change %s of %s", "permissions", dest);
182	niro	532	ret = EXIT_FAILURE;
183			}
184	niro	816	#if ENABLE_SELINUX
185			if (use_default_selinux_context)
186			setdefaultfilecon(dest);
187			#endif
188	niro	532	/* Set the user and group id */
189	niro	816	if ((opts & (OPT_OWNER\|OPT_GROUP))
190	niro	532	&& lchown(dest, uid, gid) == -1
191			) {
192	niro	816	bb_perror_msg("can't change %s of %s", "ownership", dest);
193	niro	532	ret = EXIT_FAILURE;
194			}
195	niro	816	if (opts & OPT_STRIP) {
196			char *args[3];
197			args[0] = (char*)"strip";
198			args[1] = dest;
199			args[2] = NULL;
200			if (spawn_and_wait(args)) {
201	niro	532	bb_perror_msg("strip");
202			ret = EXIT_FAILURE;
203			}
204			}
205	niro	816	next:
206			if (ENABLE_FEATURE_CLEAN_UP && isdir)
207			free(dest);
208	niro	532	}
209
210			return ret;
211			}