8644502b588115fe9c230698e0b67e5dfa644f6f
[oweals/busybox.git] / 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  * Licensed under GPLv2, see file LICENSE in this source tree.
9  */
10
11 //usage:#define chcon_trivial_usage
12 //usage:       "[OPTIONS] CONTEXT FILE..."
13 //usage:       "\n      chcon [OPTIONS] [-u USER] [-r ROLE] [-l RANGE] [-t TYPE] FILE..."
14 //usage:        IF_FEATURE_CHCON_LONG_OPTIONS(
15 //usage:       "\n      chcon [OPTIONS] --reference=RFILE FILE..."
16 //usage:        )
17 //usage:#define chcon_full_usage "\n\n"
18 //usage:       "Change the security context of each FILE to CONTEXT\n"
19 //usage:        IF_FEATURE_CHCON_LONG_OPTIONS(
20 //usage:     "\n        -v,--verbose            Verbose"
21 //usage:     "\n        -c,--changes            Report changes made"
22 //usage:     "\n        -h,--no-dereference     Affect symlinks instead of their targets"
23 //usage:     "\n        -f,--silent,--quiet     Suppress most error messages"
24 //usage:     "\n        --reference=RFILE       Use RFILE's group instead of using a CONTEXT value"
25 //usage:     "\n        -u,--user=USER          Set user/role/type/range in the target"
26 //usage:     "\n        -r,--role=ROLE          security context"
27 //usage:     "\n        -t,--type=TYPE"
28 //usage:     "\n        -l,--range=RANGE"
29 //usage:     "\n        -R,--recursive          Recurse"
30 //usage:        )
31 //usage:        IF_NOT_FEATURE_CHCON_LONG_OPTIONS(
32 //usage:     "\n        -v      Verbose"
33 //usage:     "\n        -c      Report changes made"
34 //usage:     "\n        -h      Affect symlinks instead of their targets"
35 //usage:     "\n        -f      Suppress most error messages"
36 //usage:     "\n        -u USER Set user/role/type/range in the target security context"
37 //usage:     "\n        -r ROLE"
38 //usage:     "\n        -t TYPE"
39 //usage:     "\n        -l RNG"
40 //usage:     "\n        -R      Recurse"
41 //usage:        )
42
43 #include <getopt.h>
44 #include <selinux/context.h>
45
46 #include "libbb.h"
47
48 #define OPT_RECURSIVE           (1<<0)  /* 'R' */
49 #define OPT_CHANHES             (1<<1)  /* 'c' */
50 #define OPT_NODEREFERENCE       (1<<2)  /* 'h' */
51 #define OPT_QUIET               (1<<3)  /* 'f' */
52 #define OPT_USER                (1<<4)  /* 'u' */
53 #define OPT_ROLE                (1<<5)  /* 'r' */
54 #define OPT_TYPE                (1<<6)  /* 't' */
55 #define OPT_RANGE               (1<<7)  /* 'l' */
56 #define OPT_VERBOSE             (1<<8)  /* 'v' */
57 #define OPT_REFERENCE           ((1<<9) * ENABLE_FEATURE_CHCON_LONG_OPTIONS)
58 #define OPT_COMPONENT_SPECIFIED (OPT_USER | OPT_ROLE | OPT_TYPE | OPT_RANGE)
59
60 static char *user = NULL;
61 static char *role = NULL;
62 static char *type = NULL;
63 static char *range = NULL;
64 static char *specified_context = NULL;
65
66 static int FAST_FUNC change_filedir_context(
67                 const char *fname,
68                 struct stat *stbuf UNUSED_PARAM,
69                 void *userData UNUSED_PARAM,
70                 int depth UNUSED_PARAM)
71 {
72         context_t context = NULL;
73         security_context_t file_context = NULL;
74         security_context_t context_string;
75         int rc = FALSE;
76         int status = 0;
77
78         if (option_mask32 & OPT_NODEREFERENCE) {
79                 status = lgetfilecon(fname, &file_context);
80         } else {
81                 status = getfilecon(fname, &file_context);
82         }
83         if (status < 0 && errno != ENODATA) {
84                 if ((option_mask32 & OPT_QUIET) == 0)
85                         bb_error_msg("can't obtain security context: %s", fname);
86                 goto skip;
87         }
88
89         if (file_context == NULL && specified_context == NULL) {
90                 bb_error_msg("can't apply partial context to unlabeled file %s", fname);
91                 goto skip;
92         }
93
94         if (specified_context == NULL) {
95                 context = set_security_context_component(file_context,
96                                                          user, role, type, range);
97                 if (!context) {
98                         bb_error_msg("can't compute security context from %s", file_context);
99                         goto skip;
100                 }
101         } else {
102                 context = context_new(specified_context);
103                 if (!context) {
104                         bb_error_msg("invalid context: %s", specified_context);
105                         goto skip;
106                 }
107         }
108
109         context_string = context_str(context);
110         if (!context_string) {
111                 bb_error_msg("can't obtain security context in text expression");
112                 goto skip;
113         }
114
115         if (file_context == NULL || strcmp(context_string, file_context) != 0) {
116                 int fail;
117
118                 if (option_mask32 & OPT_NODEREFERENCE) {
119                         fail = lsetfilecon(fname, context_string);
120                 } else {
121                         fail = setfilecon(fname, context_string);
122                 }
123                 if ((option_mask32 & OPT_VERBOSE) || ((option_mask32 & OPT_CHANHES) && !fail)) {
124                         printf(!fail
125                                ? "context of %s changed to %s\n"
126                                : "can't change context of %s to %s\n",
127                                fname, context_string);
128                 }
129                 if (!fail) {
130                         rc = TRUE;
131                 } else if ((option_mask32 & OPT_QUIET) == 0) {
132                         bb_error_msg("can't change context of %s to %s",
133                                      fname, context_string);
134                 }
135         } else if (option_mask32 & OPT_VERBOSE) {
136                 printf("context of %s retained as %s\n", fname, context_string);
137                 rc = TRUE;
138         }
139 skip:
140         context_free(context);
141         freecon(file_context);
142
143         return rc;
144 }
145
146 #if ENABLE_FEATURE_CHCON_LONG_OPTIONS
147 static const char chcon_longopts[] ALIGN1 =
148         "recursive\0"      No_argument       "R"
149         "changes\0"        No_argument       "c"
150         "no-dereference\0" No_argument       "h"
151         "silent\0"         No_argument       "f"
152         "quiet\0"          No_argument       "f"
153         "user\0"           Required_argument "u"
154         "role\0"           Required_argument "r"
155         "type\0"           Required_argument "t"
156         "range\0"          Required_argument "l"
157         "verbose\0"        No_argument       "v"
158         "reference\0"      Required_argument "\xff" /* no short option */
159         ;
160 #endif
161
162 int chcon_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
163 int chcon_main(int argc UNUSED_PARAM, char **argv)
164 {
165         char *reference_file;
166         char *fname;
167         int i, errors = 0;
168
169 #if ENABLE_FEATURE_CHCON_LONG_OPTIONS
170         applet_long_options = chcon_longopts;
171 #endif
172         opt_complementary = "-1"  /* at least 1 param */
173                 ":?"  /* error if exclusivity constraints are violated */
174 #if ENABLE_FEATURE_CHCON_LONG_OPTIONS
175                 ":\xff--urtl:u--\xff:r--\xff:t--\xff:l--\xff"
176 #endif
177                 ":f--v:v--f";  /* 'verbose' and 'quiet' are exclusive */
178         getopt32(argv, "Rchfu:r:t:l:v",
179                 &user, &role, &type, &range, &reference_file);
180         argv += optind;
181
182 #if ENABLE_FEATURE_CHCON_LONG_OPTIONS
183         if (option_mask32 & OPT_REFERENCE) {
184                 /* FIXME: lgetfilecon() should be used when '-h' is specified.
185                    But current implementation follows the original one. */
186                 if (getfilecon(reference_file, &specified_context) < 0)
187                         bb_perror_msg_and_die("getfilecon('%s') failed", reference_file);
188         } else
189 #endif
190         if ((option_mask32 & OPT_COMPONENT_SPECIFIED) == 0) {
191                 specified_context = *argv++;
192                 /* specified_context is never NULL -
193                  * "-1" in opt_complementary prevents this. */
194                 if (!argv[0])
195                         bb_error_msg_and_die("too few arguments");
196         }
197
198         for (i = 0; (fname = argv[i]) != NULL; i++) {
199                 int fname_len = strlen(fname);
200                 while (fname_len > 1 && fname[fname_len - 1] == '/')
201                         fname_len--;
202                 fname[fname_len] = '\0';
203
204                 if (recursive_action(fname,
205                                      1<<option_mask32 & OPT_RECURSIVE,
206                                      change_filedir_context,
207                                      change_filedir_context,
208                                      NULL, 0) != TRUE)
209                         errors = 1;
210         }
211         return errors;
212 }