123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272 |
- #include "libbb.h"
- #include "libcoreutils/coreutils.h"
- #if ENABLE_FEATURE_INSTALL_LONG_OPTIONS
- static const char install_longopts[] ALIGN1 =
- IF_FEATURE_VERBOSE(
- "verbose\0" No_argument "v"
- )
- "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"
- "target-directory\0" Required_argument "t"
- # if ENABLE_SELINUX
- "context\0" Required_argument "Z"
- "preserve_context\0" No_argument "\xff"
- "preserve-context\0" No_argument "\xff"
- # endif
- ;
- # define GETOPT32 getopt32long
- # define LONGOPTS install_longopts,
- #else
- # define GETOPT32 getopt32
- # define LONGOPTS
- #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: can't 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 mkdir_flags = FILEUTILS_RECUR;
- int copy_flags = FILEUTILS_DEREFERENCE | FILEUTILS_FORCE;
- int opts;
- int ret = EXIT_SUCCESS;
- int isdir;
- #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,
- OPT_TARGET = 1 << 10,
- #if ENABLE_SELINUX
- OPT_SET_SECURITY_CONTEXT = 1 << 11,
- OPT_PRESERVE_SECURITY_CONTEXT = 1 << 12,
- #endif
- };
-
-
- opts = GETOPT32(argv, "^"
- "cvb" "Ddpsg:m:o:t:" IF_SELINUX("Z:")
- "\0"
- "t--d:d--t:s--d:d--s"
- IF_FEATURE_INSTALL_LONG_OPTIONS(IF_SELINUX(":Z--\xff:\xff--Z")),
- LONGOPTS
- &gid_str, &mode_str, &uid_str, &last
- 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
- if ((opts & OPT_v) && FILEUTILS_VERBOSE) {
- mkdir_flags |= FILEUTILS_VERBOSE;
- copy_flags |= FILEUTILS_VERBOSE;
- }
-
- if (opts & OPT_PRESERVE_TIME) {
- copy_flags |= FILEUTILS_PRESERVE_STATUS;
- }
- mode = 0755;
- if (opts & OPT_MODE)
- 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();
-
- isdir = (opts & OPT_TARGET);
- if (!(opts & (OPT_TARGET|OPT_DIRECTORY))) {
-
- argc--;
- last = argv[argc];
- argv[argc] = NULL;
-
- isdir = stat(last, &statbuf) < 0 ? 0 : S_ISDIR(statbuf.st_mode);
- }
- if (argc < 1)
- bb_show_usage();
- while ((arg = *argv++) != NULL) {
- char *dest;
- if (opts & OPT_DIRECTORY) {
- dest = arg;
-
- if (bb_make_directory(dest, 0755, mkdir_flags)) {
- ret = EXIT_FAILURE;
- goto next;
- }
- } else {
- dest = last;
- if (opts & OPT_MKDIR_LEADING) {
- char *ddir = xstrdup(dest);
-
- bb_make_directory((opts & OPT_TARGET) ? ddir : dirname(ddir), 0755, mkdir_flags);
-
- free(ddir);
- }
- if (isdir)
- dest = concat_path_file(last, bb_basename(arg));
- if (copy_file(arg, dest, copy_flags) != 0) {
-
- ret = EXIT_FAILURE;
- goto next;
- }
- if (opts & OPT_STRIP) {
- char *args[4];
- args[0] = (char*)"strip";
- args[1] = (char*)"-p";
- args[2] = dest;
- args[3] = NULL;
- if (spawn_and_wait(args)) {
- bb_simple_perror_msg("strip");
- ret = EXIT_FAILURE;
- }
- }
- }
-
-
- 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 (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
- next:
- if (ENABLE_FEATURE_CLEAN_UP && isdir)
- free(dest);
- }
- return ret;
- }
|