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