coreutils/install.c

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