coreutils/install.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Copyright (C) 2003 by Glenn McGrath <bug1@iinet.net.au>
   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
  19 static const char install_longopts[] =
  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 #if ENABLE_SELINUX
  27         "context\0"             Required_argument "Z"
  28         "preserve_context\0"    No_argument       "\xff"
  29         "preserve-context\0"    No_argument       "\xff"
  30 #endif
  31         "\0";
  32 #endif
  33
  34
  35 #if ENABLE_SELINUX
  36 static bool use_default_selinux_context = 1;
  37
  38 static 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
  69 int install_main(int argc, char **argv);
  70 int install_main(int argc, char **argv)
  71 {
  72         struct stat statbuf;
  73         mode_t mode;
  74         uid_t uid;
  75         gid_t gid;
  76         const char *gid_str;
  77         const char *uid_str;
  78         const char *mode_str;
  79         int copy_flags = FILEUTILS_DEREFERENCE | FILEUTILS_FORCE;
  80         int ret = EXIT_SUCCESS, flags, i, isdir;
  81 #if ENABLE_SELINUX
  82         security_context_t scontext;
  83 #endif
  84         enum {
  85                 OPT_CMD           =  0x1,
  86                 OPT_DIRECTORY     =  0x2,
  87                 OPT_PRESERVE_TIME =  0x4,
  88                 OPT_STRIP         =  0x8,
  89                 OPT_GROUP         = 0x10,
  90                 OPT_MODE          = 0x20,
  91                 OPT_OWNER         = 0x40,
  92 #if ENABLE_SELINUX
  93                 OPT_SET_SECURITY_CONTEXT = 0x80,
  94                 OPT_PRESERVE_SECURITY_CONTEXT = 0x100,
  95 #endif
  96         };
  97
  98 #if ENABLE_FEATURE_INSTALL_LONG_OPTIONS
  99         applet_long_options = install_longopts;
 100 #endif
 101         opt_complementary = "s--d:d--s" USE_SELINUX(":Z--\xff:\xff--Z");
 102         /* -c exists for backwards compatibility, it's needed */
 103
 104         flags = getopt32(argc, argv, "cdpsg:m:o:" USE_SELINUX("Z:"),
 105                         &gid_str, &mode_str, &uid_str USE_SELINUX(, &scontext));
 106
 107 #if ENABLE_SELINUX
 108         if (flags & OPT_PRESERVE_SECURITY_CONTEXT) {
 109                 use_default_selinux_context = 0;
 110                 copy_flags |= FILEUTILS_PRESERVE_SECURITY_CONTEXT;
 111                 selinux_or_die();
 112         }
 113         if (flags & OPT_SET_SECURITY_CONTEXT) {
 114                 selinux_or_die();
 115                 setfscreatecon_or_die(scontext);
 116                 use_default_selinux_context = 0;
 117                 copy_flags |= FILEUTILS_SET_SECURITY_CONTEXT;
 118         }
 119 #endif
 120
 121         /* preserve access and modification time, this is GNU behaviour, BSD only preserves modification time */
 122         if (flags & OPT_PRESERVE_TIME) {
 123                 copy_flags |= FILEUTILS_PRESERVE_STATUS;
 124         }
 125         mode = 0666;
 126         if (flags & OPT_MODE) bb_parse_mode(mode_str, &mode);
 127         uid = (flags & OPT_OWNER) ? get_ug_id(uid_str, xuname2uid) : getuid();
 128         gid = (flags & OPT_GROUP) ? get_ug_id(gid_str, xgroup2gid) : getgid();
 129         if (flags & (OPT_OWNER|OPT_GROUP)) umask(0);
 130
 131         /* Create directories
 132          * don't use bb_make_directory() as it can't change uid or gid
 133          * perhaps bb_make_directory() should be improved.
 134          */
 135         if (flags & OPT_DIRECTORY) {
 136                 for (argv += optind; *argv; argv++) {
 137                         char *old_argv_ptr = *argv + 1;
 138                         char *argv_ptr;
 139                         do {
 140                                 argv_ptr = strchr(old_argv_ptr, '/');
 141                                 old_argv_ptr = argv_ptr;
 142                                 if (argv_ptr) {
 143                                         *argv_ptr = '\0';
 144                                         old_argv_ptr++;
 145                                 }
 146                                 if (mkdir(*argv, mode | 0111) == -1) {
 147                                         if (errno != EEXIST) {
 148                                                 bb_perror_msg("cannot create %s", *argv);
 149                                                 ret = EXIT_FAILURE;
 150                                                 break;
 151                                         }
 152                                 }
 153                                 if ((flags & (OPT_OWNER|OPT_GROUP))
 154                                  && lchown(*argv, uid, gid) == -1
 155                                 ) {
 156                                         bb_perror_msg("cannot change ownership of %s", *argv);
 157                                         ret = EXIT_FAILURE;
 158                                         break;
 159                                 }
 160                                 if (argv_ptr) {
 161                                         *argv_ptr = '/';
 162                                 }
 163                         } while (old_argv_ptr);
 164                 }
 165                 return ret;
 166         }
 167
 168         /* coreutils install resolves link in this case, don't use lstat */
 169         isdir = stat(argv[argc - 1], &statbuf) < 0 ? 0 : S_ISDIR(statbuf.st_mode);
 170
 171         for (i = optind; i < argc - 1; i++) {
 172                 char *dest;
 173
 174                 dest = argv[argc - 1];
 175                 if (isdir)
 176                         dest = concat_path_file(argv[argc - 1], basename(argv[i]));
 177                 ret |= copy_file(argv[i], dest, copy_flags);
 178
 179                 /* Set the file mode */
 180                 if ((flags & OPT_MODE) && chmod(dest, mode) == -1) {
 181                         bb_perror_msg("cannot change permissions of %s", dest);
 182                         ret = EXIT_FAILURE;
 183                 }
 184 #if ENABLE_SELINUX
 185                 if (use_default_selinux_context)
 186                         setdefaultfilecon(dest);
 187 #endif
 188                 /* Set the user and group id */
 189                 if ((flags & (OPT_OWNER|OPT_GROUP))
 190                  && lchown(dest, uid, gid) == -1
 191                 ) {
 192                         bb_perror_msg("cannot change ownership of %s", dest);
 193                         ret = EXIT_FAILURE;
 194                 }
 195                 if (flags & 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                                 bb_perror_msg("strip");
 202                                 ret = EXIT_FAILURE;
 203                         }
 204                 }
 205                 if (ENABLE_FEATURE_CLEAN_UP && isdir) free(dest);
 206         }
 207
 208         return ret;
 209 }