1 /* vi: set sw=4 ts=4: */
2 /*
3 * Copyright (C) 2003 by Glenn McGrath
4 * SELinux support: by Yuichi Nakamura <ynakam@hitachisoft.jp>
5 *
6 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
7 */
8 //config:config INSTALL
9 //config: bool "install (12 kb)"
10 //config: default y
11 //config: help
12 //config: Copy files and set attributes.
13 //config:
14 //config:config FEATURE_INSTALL_LONG_OPTIONS
15 //config: bool "Enable long options"
16 //config: default y
17 //config: depends on INSTALL && LONG_OPTS
18
19 //applet:IF_INSTALL(APPLET(install, BB_DIR_USR_BIN, BB_SUID_DROP))
20
21 //kbuild:lib-$(CONFIG_INSTALL) += install.o
22
23 /* -v, -b, -c are ignored */
24 //usage:#define install_trivial_usage
25 //usage: "[-cdDsp] [-o USER] [-g GRP] [-m MODE] [-t DIR] [SOURCE]... DEST"
26 //usage:#define install_full_usage "\n\n"
27 //usage: "Copy files and set attributes\n"
28 //usage: "\n -c Just copy (default)"
29 //usage: "\n -d Create directories"
30 //usage: "\n -D Create leading target directories"
31 //usage: "\n -s Strip symbol table"
32 //usage: "\n -p Preserve date"
33 //usage: "\n -o USER Set ownership"
34 //usage: "\n -g GRP Set group ownership"
35 //usage: "\n -m MODE Set permissions"
36 //usage: "\n -t DIR Install to DIR"
37 //usage: IF_SELINUX(
38 //usage: "\n -Z Set security context"
39 //usage: )
40
41 #include "libbb.h"
42 #include "libcoreutils/coreutils.h"
43
44 #if ENABLE_FEATURE_INSTALL_LONG_OPTIONS
45 static const char install_longopts[] ALIGN1 =
46 IF_FEATURE_VERBOSE(
47 "verbose\0" No_argument "v"
48 )
49 "directory\0" No_argument "d"
50 "preserve-timestamps\0" No_argument "p"
51 "strip\0" No_argument "s"
52 "group\0" Required_argument "g"
53 "mode\0" Required_argument "m"
54 "owner\0" Required_argument "o"
55 "target-directory\0" Required_argument "t"
56 /* autofs build insists of using -b --suffix=.orig */
57 /* TODO? (short option for --suffix is -S) */
58 # if ENABLE_SELINUX
59 "context\0" Required_argument "Z"
60 "preserve_context\0" No_argument "\xff"
61 "preserve-context\0" No_argument "\xff"
62 # endif
63 ;
64 # define GETOPT32 getopt32long
65 # define LONGOPTS install_longopts,
66 #else
67 # define GETOPT32 getopt32
68 # define LONGOPTS
69 #endif
70
71
72 #if ENABLE_SELINUX
setdefaultfilecon(const char * path)73 static void setdefaultfilecon(const char *path)
74 {
75 struct stat s;
76 security_context_t scontext = NULL;
77
78 if (!is_selinux_enabled()) {
79 return;
80 }
81 if (lstat(path, &s) != 0) {
82 return;
83 }
84
85 if (matchpathcon(path, s.st_mode, &scontext) < 0) {
86 goto out;
87 }
88 if (strcmp(scontext, "<<none>>") == 0) {
89 goto out;
90 }
91
92 if (lsetfilecon(path, scontext) < 0) {
93 if (errno != ENOTSUP) {
94 bb_perror_msg("warning: can't change context"
95 " of %s to %s", path, scontext);
96 }
97 }
98
99 out:
100 freecon(scontext);
101 }
102
103 #endif
104
105 int install_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
install_main(int argc,char ** argv)106 int install_main(int argc, char **argv)
107 {
108 struct stat statbuf;
109 mode_t mode;
110 uid_t uid;
111 gid_t gid;
112 char *arg, *last;
113 const char *gid_str;
114 const char *uid_str;
115 const char *mode_str;
116 int mkdir_flags = FILEUTILS_RECUR;
117 int copy_flags = FILEUTILS_DEREFERENCE | FILEUTILS_FORCE;
118 int opts;
119 int ret = EXIT_SUCCESS;
120 int isdir;
121 #if ENABLE_SELINUX
122 security_context_t scontext;
123 bool use_default_selinux_context = 1;
124 #endif
125 enum {
126 OPT_c = 1 << 0,
127 OPT_v = 1 << 1,
128 OPT_b = 1 << 2,
129 OPT_MKDIR_LEADING = 1 << 3,
130 OPT_DIRECTORY = 1 << 4,
131 OPT_PRESERVE_TIME = 1 << 5,
132 OPT_STRIP = 1 << 6,
133 OPT_GROUP = 1 << 7,
134 OPT_MODE = 1 << 8,
135 OPT_OWNER = 1 << 9,
136 OPT_TARGET = 1 << 10,
137 #if ENABLE_SELINUX
138 OPT_SET_SECURITY_CONTEXT = 1 << 11,
139 OPT_PRESERVE_SECURITY_CONTEXT = 1 << 12,
140 #endif
141 };
142
143 /* -c exists for backwards compatibility, it's needed */
144 /* -b is ignored ("make a backup of each existing destination file") */
145 opts = GETOPT32(argv, "^"
146 "cvb" "Ddpsg:m:o:t:" IF_SELINUX("Z:")
147 "\0"
148 "t--d:d--t:s--d:d--s"
149 IF_FEATURE_INSTALL_LONG_OPTIONS(IF_SELINUX(":Z--\xff:\xff--Z")),
150 LONGOPTS
151 &gid_str, &mode_str, &uid_str, &last
152 IF_SELINUX(, &scontext)
153 );
154 argc -= optind;
155 argv += optind;
156
157 #if ENABLE_SELINUX
158 if (opts & (OPT_PRESERVE_SECURITY_CONTEXT|OPT_SET_SECURITY_CONTEXT)) {
159 selinux_or_die();
160 use_default_selinux_context = 0;
161 if (opts & OPT_PRESERVE_SECURITY_CONTEXT) {
162 copy_flags |= FILEUTILS_PRESERVE_SECURITY_CONTEXT;
163 }
164 if (opts & OPT_SET_SECURITY_CONTEXT) {
165 setfscreatecon_or_die(scontext);
166 copy_flags |= FILEUTILS_SET_SECURITY_CONTEXT;
167 }
168 }
169 #endif
170
171 if ((opts & OPT_v) && FILEUTILS_VERBOSE) {
172 mkdir_flags |= FILEUTILS_VERBOSE;
173 copy_flags |= FILEUTILS_VERBOSE;
174 }
175
176 /* preserve access and modification time, this is GNU behaviour,
177 * BSD only preserves modification time */
178 if (opts & OPT_PRESERVE_TIME) {
179 copy_flags |= FILEUTILS_PRESERVE_STATUS;
180 }
181 mode = 0755; /* GNU coreutils 6.10 compat */
182 if (opts & OPT_MODE)
183 mode = bb_parse_mode(mode_str, mode);
184 uid = (opts & OPT_OWNER) ? get_ug_id(uid_str, xuname2uid) : getuid();
185 gid = (opts & OPT_GROUP) ? get_ug_id(gid_str, xgroup2gid) : getgid();
186
187 /* If -t DIR is in use, then isdir=true, last="DIR" */
188 isdir = (opts & OPT_TARGET);
189 if (!(opts & (OPT_TARGET|OPT_DIRECTORY))) {
190 /* Neither -t DIR nor -d is in use */
191 argc--;
192 last = argv[argc];
193 argv[argc] = NULL;
194 /* coreutils install resolves link in this case, don't use lstat */
195 isdir = stat(last, &statbuf) < 0 ? 0 : S_ISDIR(statbuf.st_mode);
196 }
197
198 if (argc < 1)
199 bb_show_usage();
200
201 while ((arg = *argv++) != NULL) {
202 char *dest;
203
204 if (opts & OPT_DIRECTORY) {
205 dest = arg;
206 /* GNU coreutils 6.9 does not set uid:gid
207 * on intermediate created directories
208 * (only on last one) */
209 if (bb_make_directory(dest, 0755, mkdir_flags)) {
210 ret = EXIT_FAILURE;
211 goto next;
212 }
213 } else {
214 dest = last;
215 if (opts & OPT_MKDIR_LEADING) {
216 char *ddir = xstrdup(dest);
217 /*
218 * -D -t DIR1/DIR2/F3 FILE: create DIR1/DIR2/F3, copy FILE there
219 * -D FILE DIR1/DIR2/F3: create DIR1/DIR2, copy FILE there as F3
220 */
221 bb_make_directory((opts & OPT_TARGET) ? ddir : dirname(ddir), 0755, mkdir_flags);
222 /* errors are not checked. copy_file
223 * will fail if dir is not created.
224 */
225 free(ddir);
226 }
227 if (isdir)
228 dest = concat_path_file(last, bb_basename(arg));
229 if (copy_file(arg, dest, copy_flags) != 0) {
230 /* copy is not made */
231 ret = EXIT_FAILURE;
232 goto next;
233 }
234 if (opts & OPT_STRIP) {
235 char *args[4];
236 args[0] = (char*)"strip";
237 args[1] = (char*)"-p"; /* -p --preserve-dates */
238 args[2] = dest;
239 args[3] = NULL;
240 if (spawn_and_wait(args)) {
241 bb_simple_perror_msg("strip");
242 ret = EXIT_FAILURE;
243 }
244 }
245 }
246
247 /* Set the file mode (always, not only with -m).
248 * GNU coreutils 6.10 is not affected by umask. */
249 if (chmod(dest, mode) == -1) {
250 bb_perror_msg("can't change %s of %s", "permissions", dest);
251 ret = EXIT_FAILURE;
252 }
253 #if ENABLE_SELINUX
254 if (use_default_selinux_context)
255 setdefaultfilecon(dest);
256 #endif
257 /* Set the user and group id */
258 if ((opts & (OPT_OWNER|OPT_GROUP))
259 && lchown(dest, uid, gid) == -1
260 ) {
261 bb_perror_msg("can't change %s of %s", "ownership", dest);
262 ret = EXIT_FAILURE;
263 }
264 next:
265 if (ENABLE_FEATURE_CLEAN_UP && isdir)
266 free(dest);
267 }
268
269 return ret;
270 }
271