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 //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
  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;
 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 }