git.maemo.org Git - busybox4maemo/blob - selinux/chcon.c

   1 /*
   2  * chcon -- change security context, based on coreutils-5.97-13
   3  *
   4  * Port to busybox: KaiGai Kohei <kaigai@kaigai.gr.jp>
   5  *
   6  * Copyright (C) 2006 - 2007 KaiGai Kohei <kaigai@kaigai.gr.jp>
   7  */
   8 #include <getopt.h>
   9 #include <selinux/context.h>
  10
  11 #include "libbb.h"
  12
  13 #define OPT_RECURSIVE           (1<<0)  /* 'R' */
  14 #define OPT_CHANHES             (1<<1)  /* 'c' */
  15 #define OPT_NODEREFERENCE       (1<<2)  /* 'h' */
  16 #define OPT_QUIET               (1<<3)  /* 'f' */
  17 #define OPT_USER                (1<<4)  /* 'u' */
  18 #define OPT_ROLE                (1<<5)  /* 'r' */
  19 #define OPT_TYPE                (1<<6)  /* 't' */
  20 #define OPT_RANGE               (1<<7)  /* 'l' */
  21 #define OPT_VERBOSE             (1<<8)  /* 'v' */
  22 #define OPT_REFERENCE           ((1<<9) * ENABLE_FEATURE_CHCON_LONG_OPTIONS)
  23 #define OPT_COMPONENT_SPECIFIED (OPT_USER | OPT_ROLE | OPT_TYPE | OPT_RANGE)
  24
  25 static char *user = NULL;
  26 static char *role = NULL;
  27 static char *type = NULL;
  28 static char *range = NULL;
  29 static char *specified_context = NULL;
  30
  31 static int change_filedir_context(
  32                 const char *fname,
  33                 struct stat *stbuf ATTRIBUTE_UNUSED,
  34                 void *userData ATTRIBUTE_UNUSED,
  35                 int depth ATTRIBUTE_UNUSED)
  36 {
  37         context_t context = NULL;
  38         security_context_t file_context = NULL;
  39         security_context_t context_string;
  40         int rc = FALSE;
  41         int status = 0;
  42
  43         if (option_mask32 & OPT_NODEREFERENCE) {
  44                 status = lgetfilecon(fname, &file_context);
  45         } else {
  46                 status = getfilecon(fname, &file_context);
  47         }
  48         if (status < 0 && errno != ENODATA) {
  49                 if ((option_mask32 & OPT_QUIET) == 0)
  50                         bb_error_msg("cannot obtain security context: %s", fname);
  51                 goto skip;
  52         }
  53
  54         if (file_context == NULL && specified_context == NULL) {
  55                 bb_error_msg("cannot apply partial context to unlabeled file %s", fname);
  56                 goto skip;
  57         }
  58
  59         if (specified_context == NULL) {
  60                 context = set_security_context_component(file_context,
  61                                                          user, role, type, range);
  62                 if (!context) {
  63                         bb_error_msg("cannot compute security context from %s", file_context);
  64                         goto skip;
  65                 }
  66         } else {
  67                 context = context_new(specified_context);
  68                 if (!context) {
  69                         bb_error_msg("invalid context: %s", specified_context);
  70                         goto skip;
  71                 }
  72         }
  73
  74         context_string = context_str(context);
  75         if (!context_string) {
  76                 bb_error_msg("cannot obtain security context in text expression");
  77                 goto skip;
  78         }
  79
  80         if (file_context == NULL || strcmp(context_string, file_context) != 0) {
  81                 int fail;
  82
  83                 if (option_mask32 & OPT_NODEREFERENCE) {
  84                         fail = lsetfilecon(fname, context_string);
  85                 } else {
  86                         fail = setfilecon(fname, context_string);
  87                 }
  88                 if ((option_mask32 & OPT_VERBOSE) || ((option_mask32 & OPT_CHANHES) && !fail)) {
  89                         printf(!fail
  90                                ? "context of %s changed to %s\n"
  91                                : "failed to change context of %s to %s\n",
  92                                fname, context_string);
  93                 }
  94                 if (!fail) {
  95                         rc = TRUE;
  96                 } else if ((option_mask32 & OPT_QUIET) == 0) {
  97                         bb_error_msg("failed to change context of %s to %s",
  98                                      fname, context_string);
  99                 }
 100         } else if (option_mask32 & OPT_VERBOSE) {
 101                 printf("context of %s retained as %s\n", fname, context_string);
 102                 rc = TRUE;
 103         }
 104 skip:
 105         context_free(context);
 106         freecon(file_context);
 107
 108         return rc;
 109 }
 110
 111 #if ENABLE_FEATURE_CHCON_LONG_OPTIONS
 112 static const char chcon_longopts[] ALIGN1 =
 113         "recursive\0"      No_argument       "R"
 114         "changes\0"        No_argument       "c"
 115         "no-dereference\0" No_argument       "h"
 116         "silent\0"         No_argument       "f"
 117         "quiet\0"          No_argument       "f"
 118         "user\0"           Required_argument "u"
 119         "role\0"           Required_argument "r"
 120         "type\0"           Required_argument "t"
 121         "range\0"          Required_argument "l"
 122         "verbose\0"        No_argument       "v"
 123         "reference\0"      Required_argument "\xff" /* no short option */
 124         ;
 125 #endif
 126
 127 int chcon_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 128 int chcon_main(int argc ATTRIBUTE_UNUSED, char **argv)
 129 {
 130         char *reference_file;
 131         char *fname;
 132         int i, errors = 0;
 133
 134 #if ENABLE_FEATURE_CHCON_LONG_OPTIONS
 135         applet_long_options = chcon_longopts;
 136 #endif
 137         opt_complementary = "-1"  /* at least 1 param */
 138                 ":?"  /* error if exclusivity constraints are violated */
 139 #if ENABLE_FEATURE_CHCON_LONG_OPTIONS
 140                 ":\xff--urtl:u--\xff:r--\xff:t--\xff:l--\xff"
 141 #endif
 142                 ":f--v:v--f";  /* 'verbose' and 'quiet' are exclusive */
 143         getopt32(argv, "Rchfu:r:t:l:v",
 144                 &user, &role, &type, &range, &reference_file);
 145         argv += optind;
 146
 147 #if ENABLE_FEATURE_CHCON_LONG_OPTIONS
 148         if (option_mask32 & OPT_REFERENCE) {
 149                 /* FIXME: lgetfilecon() should be used when '-h' is specified.
 150                    But current implementation follows the original one. */
 151                 if (getfilecon(reference_file, &specified_context) < 0)
 152                         bb_perror_msg_and_die("getfilecon('%s') failed", reference_file);
 153         } else
 154 #endif
 155         if ((option_mask32 & OPT_COMPONENT_SPECIFIED) == 0) {
 156                 specified_context = *argv++;
 157                 /* specified_context is never NULL -
 158                  * "-1" in opt_complementary prevents this. */
 159                 if (!argv[0])
 160                         bb_error_msg_and_die("too few arguments");
 161         }
 162
 163         for (i = 0; (fname = argv[i]) != NULL; i++) {
 164                 int fname_len = strlen(fname);
 165                 while (fname_len > 1 && fname[fname_len - 1] == '/')
 166                         fname_len--;
 167                 fname[fname_len] = '\0';
 168
 169                 if (recursive_action(fname,
 170                                      1<<option_mask32 & OPT_RECURSIVE,
 171                                      change_filedir_context,
 172                                      change_filedir_context,
 173                                      NULL, 0) != TRUE)
 174                         errors = 1;
 175         }
 176         return errors;
 177 }