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