1 /*
2 * Copyright 2011 Tresys Technology, LLC. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are met:
6 *
7 * 1. Redistributions of source code must retain the above copyright notice,
8 * this list of conditions and the following disclaimer.
9 *
10 * 2. Redistributions in binary form must reproduce the above copyright notice,
11 * this list of conditions and the following disclaimer in the documentation
12 * and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY TRESYS TECHNOLOGY, LLC ``AS IS'' AND ANY EXPRESS
15 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
16 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
17 * EVENT SHALL TRESYS TECHNOLOGY, LLC OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
18 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
19 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
20 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
21 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
22 * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
23 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 *
25 * The views and conclusions contained in the software and documentation are those
26 * of the authors and should not be interpreted as representing official policies,
27 * either expressed or implied, of Tresys Technology, LLC.
28 */
29
30 #include <stdlib.h>
31 #include <stdio.h>
32 #include <stdint.h>
33 #include <string.h>
34 #include <getopt.h>
35 #include <sys/stat.h>
36
37 #ifdef ANDROID
38 #include <cil/cil.h>
39 #else
40 #include <sepol/cil/cil.h>
41 #endif
42 #include <sepol/policydb.h>
43
usage(const char * prog)44 static __attribute__((__noreturn__)) void usage(const char *prog)
45 {
46 printf("Usage: %s [OPTION]... FILE...\n", prog);
47 printf("\n");
48 printf("Options:\n");
49 printf(" -o, --output=<file> write binary policy to <file>\n");
50 printf(" (default: policy.<version>)\n");
51 printf(" -f, --filecontext=<file> write file contexts to <file>\n");
52 printf(" (default: file_contexts)\n");
53 printf(" -t, --target=<type> specify target architecture. may be selinux or\n");
54 printf(" xen. (default: selinux)\n");
55 printf(" -M, --mls true|false build an mls policy. Must be true or false.\n");
56 printf(" This will override the (mls boolean) statement\n");
57 printf(" if present in the policy\n");
58 printf(" -c, --policyvers=<version> build a binary policy with a given <version>\n");
59 printf(" (default: %i)\n", POLICYDB_VERSION_MAX);
60 printf(" -U, --handle-unknown=<action> how to handle unknown classes or permissions.\n");
61 printf(" may be deny, allow, or reject. (default: deny)\n");
62 printf(" This will override the (handleunknown action)\n");
63 printf(" statement if present in the policy\n");
64 printf(" -D, --disable-dontaudit do not add dontaudit rules to the binary policy\n");
65 printf(" -P, --preserve-tunables treat tunables as booleans\n");
66 printf(" -m, --multiple-decls allow some statements to be re-declared\n");
67 printf(" -N, --disable-neverallow do not check neverallow rules\n");
68 printf(" -G, --expand-generated Expand and remove auto-generated attributes\n");
69 printf(" -X, --expand-size <SIZE> Expand type attributes with fewer than <SIZE>\n");
70 printf(" members.\n");
71 printf(" -O, --optimize optimize final policy\n");
72 printf(" -v, --verbose increment verbosity level\n");
73 printf(" -h, --help display usage information\n");
74 exit(1);
75 }
76
main(int argc,char * argv[])77 int main(int argc, char *argv[])
78 {
79 int rc = SEPOL_ERR;
80 sepol_policydb_t *pdb = NULL;
81 struct sepol_policy_file *pf = NULL;
82 FILE *binary = NULL;
83 FILE *file_contexts;
84 FILE *file = NULL;
85 char *buffer = NULL;
86 struct stat filedata;
87 uint32_t file_size;
88 char *output = NULL;
89 char *filecontexts = NULL;
90 struct cil_db *db = NULL;
91 int target = SEPOL_TARGET_SELINUX;
92 int mls = -1;
93 int disable_dontaudit = 0;
94 int multiple_decls = 0;
95 int disable_neverallow = 0;
96 int preserve_tunables = 0;
97 int handle_unknown = -1;
98 int policyvers = POLICYDB_VERSION_MAX;
99 int attrs_expand_generated = 0;
100 int attrs_expand_size = -1;
101 int optimize = 0;
102 int opt_char;
103 int opt_index = 0;
104 char *fc_buf = NULL;
105 size_t fc_size;
106 enum cil_log_level log_level = CIL_ERR;
107 static struct option long_opts[] = {
108 {"help", no_argument, 0, 'h'},
109 {"verbose", no_argument, 0, 'v'},
110 {"target", required_argument, 0, 't'},
111 {"mls", required_argument, 0, 'M'},
112 {"policyversion", required_argument, 0, 'c'},
113 {"handle-unknown", required_argument, 0, 'U'},
114 {"disable-dontaudit", no_argument, 0, 'D'},
115 {"multiple-decls", no_argument, 0, 'm'},
116 {"disable-neverallow", no_argument, 0, 'N'},
117 {"preserve-tunables", no_argument, 0, 'P'},
118 {"output", required_argument, 0, 'o'},
119 {"filecontexts", required_argument, 0, 'f'},
120 {"expand-generated", no_argument, 0, 'G'},
121 {"expand-size", required_argument, 0, 'X'},
122 {"optimize", no_argument, 0, 'O'},
123 {0, 0, 0, 0}
124 };
125 int i;
126
127 while (1) {
128 opt_char = getopt_long(argc, argv, "o:f:U:hvt:M:PDmNOc:GX:n", long_opts, &opt_index);
129 if (opt_char == -1) {
130 break;
131 }
132 switch (opt_char) {
133 case 'v':
134 log_level++;
135 break;
136 case 't':
137 if (!strcmp(optarg, "selinux")) {
138 target = SEPOL_TARGET_SELINUX;
139 } else if (!strcmp(optarg, "xen")) {
140 target = SEPOL_TARGET_XEN;
141 } else {
142 fprintf(stderr, "Unknown target: %s\n", optarg);
143 usage(argv[0]);
144 }
145 break;
146 case 'M':
147 if (!strcasecmp(optarg, "true") || !strcasecmp(optarg, "1")) {
148 mls = 1;
149 } else if (!strcasecmp(optarg, "false") || !strcasecmp(optarg, "0")) {
150 mls = 0;
151 } else {
152 usage(argv[0]);
153 }
154 break;
155 case 'c': {
156 char *endptr = NULL;
157 errno = 0;
158 policyvers = strtol(optarg, &endptr, 10);
159 if (errno != 0 || endptr == optarg || *endptr != '\0') {
160 fprintf(stderr, "Bad policy version: %s\n", optarg);
161 usage(argv[0]);
162 }
163 if (policyvers > POLICYDB_VERSION_MAX || policyvers < POLICYDB_VERSION_MIN) {
164 fprintf(stderr, "Policy version must be between %d and %d\n",
165 POLICYDB_VERSION_MIN, POLICYDB_VERSION_MAX);
166 usage(argv[0]);
167 }
168 break;
169 }
170 case 'U':
171 if (!strcasecmp(optarg, "deny")) {
172 handle_unknown = SEPOL_DENY_UNKNOWN;
173 } else if (!strcasecmp(optarg, "allow")) {
174 handle_unknown = SEPOL_ALLOW_UNKNOWN;
175 } else if (!strcasecmp(optarg, "reject")) {
176 handle_unknown = SEPOL_REJECT_UNKNOWN;
177 } else {
178 usage(argv[0]);
179 }
180 break;
181 case 'D':
182 disable_dontaudit = 1;
183 break;
184 case 'm':
185 multiple_decls = 1;
186 break;
187 case 'N':
188 disable_neverallow = 1;
189 break;
190 case 'P':
191 preserve_tunables = 1;
192 break;
193 case 'o':
194 output = strdup(optarg);
195 break;
196 case 'f':
197 filecontexts = strdup(optarg);
198 break;
199 case 'G':
200 attrs_expand_generated = 1;
201 break;
202 case 'X': {
203 char *endptr = NULL;
204 errno = 0;
205 attrs_expand_size = strtol(optarg, &endptr, 10);
206 if (errno != 0 || endptr == optarg || *endptr != '\0') {
207 fprintf(stderr, "Bad attribute expand size: %s\n", optarg);
208 usage(argv[0]);
209 }
210
211 if (attrs_expand_size < 0) {
212 fprintf(stderr, "Attribute expand size must be > 0\n");
213 usage(argv[0]);
214 }
215 break;
216 }
217 case 'O':
218 optimize = 1;
219 break;
220 case 'h':
221 usage(argv[0]);
222 case '?':
223 break;
224 default:
225 fprintf(stderr, "Unsupported option: %s\n", optarg);
226 usage(argv[0]);
227 }
228 }
229 if (optind >= argc) {
230 fprintf(stderr, "No cil files specified\n");
231 usage(argv[0]);
232 }
233
234 cil_set_log_level(log_level);
235
236 cil_db_init(&db);
237 cil_set_disable_dontaudit(db, disable_dontaudit);
238 cil_set_multiple_decls(db, multiple_decls);
239 cil_set_disable_neverallow(db, disable_neverallow);
240 cil_set_preserve_tunables(db, preserve_tunables);
241 if (handle_unknown != -1) {
242 rc = cil_set_handle_unknown(db, handle_unknown);
243 if (rc != SEPOL_OK) {
244 goto exit;
245 }
246 }
247
248 cil_set_mls(db, mls);
249 cil_set_target_platform(db, target);
250 cil_set_policy_version(db, policyvers);
251 cil_set_attrs_expand_generated(db, attrs_expand_generated);
252 if (attrs_expand_size >= 0) {
253 cil_set_attrs_expand_size(db, (unsigned)attrs_expand_size);
254 }
255
256 for (i = optind; i < argc; i++) {
257 file = fopen(argv[i], "r");
258 if (!file) {
259 fprintf(stderr, "Could not open file: %s\n", argv[i]);
260 rc = SEPOL_ERR;
261 goto exit;
262 }
263 rc = stat(argv[i], &filedata);
264 if (rc == -1) {
265 fprintf(stderr, "Could not stat file: %s\n", argv[i]);
266 rc = SEPOL_ERR;
267 goto exit;
268 }
269 file_size = filedata.st_size;
270
271 if (!file_size) {
272 fclose(file);
273 file = NULL;
274 continue;
275 }
276
277 buffer = malloc(file_size);
278 rc = fread(buffer, file_size, 1, file);
279 if (rc != 1) {
280 fprintf(stderr, "Failure reading file: %s\n", argv[i]);
281 rc = SEPOL_ERR;
282 goto exit;
283 }
284 fclose(file);
285 file = NULL;
286
287 rc = cil_add_file(db, argv[i], buffer, file_size);
288 if (rc != SEPOL_OK) {
289 fprintf(stderr, "Failure adding %s\n", argv[i]);
290 goto exit;
291 }
292
293 free(buffer);
294 buffer = NULL;
295 }
296
297 rc = cil_compile(db);
298 if (rc != SEPOL_OK) {
299 fprintf(stderr, "Failed to compile cildb: %d\n", rc);
300 goto exit;
301 }
302
303 rc = cil_build_policydb(db, &pdb);
304 if (rc != SEPOL_OK) {
305 fprintf(stderr, "Failed to build policydb\n");
306 goto exit;
307 }
308
309 if (optimize) {
310 rc = sepol_policydb_optimize(pdb);
311 if (rc != SEPOL_OK) {
312 fprintf(stderr, "Failed to optimize policydb\n");
313 goto exit;
314 }
315 }
316
317 if (output == NULL) {
318 int size = snprintf(NULL, 0, "policy.%d", policyvers);
319 output = malloc((size + 1) * sizeof(char));
320 if (output == NULL) {
321 fprintf(stderr, "Failed to create output filename\n");
322 rc = SEPOL_ERR;
323 goto exit;
324 }
325 if (snprintf(output, size + 1, "policy.%d", policyvers) != size) {
326 fprintf(stderr, "Failed to create output filename\n");
327 rc = SEPOL_ERR;
328 goto exit;
329 }
330 }
331
332 binary = fopen(output, "w");
333 if (binary == NULL) {
334 fprintf(stderr, "Failure opening binary file for writing\n");
335 rc = SEPOL_ERR;
336 goto exit;
337 }
338
339 rc = sepol_policy_file_create(&pf);
340 if (rc != 0) {
341 fprintf(stderr, "Failed to create policy file: %d\n", rc);
342 goto exit;
343 }
344
345 sepol_policy_file_set_fp(pf, binary);
346
347 rc = sepol_policydb_write(pdb, pf);
348 if (rc != 0) {
349 fprintf(stderr, "Failed to write binary policy: %d\n", rc);
350 goto exit;
351 }
352
353 fclose(binary);
354 binary = NULL;
355
356 rc = cil_filecons_to_string(db, &fc_buf, &fc_size);
357 if (rc != SEPOL_OK) {
358 fprintf(stderr, "Failed to get file context data\n");
359 goto exit;
360 }
361
362 if (filecontexts == NULL) {
363 file_contexts = fopen("file_contexts", "w+");
364 } else {
365 file_contexts = fopen(filecontexts, "w+");
366 }
367
368 if (file_contexts == NULL) {
369 fprintf(stderr, "Failed to open file_contexts file\n");
370 rc = SEPOL_ERR;
371 goto exit;
372 }
373
374 if (fwrite(fc_buf, sizeof(char), fc_size, file_contexts) != fc_size) {
375 fprintf(stderr, "Failed to write file_contexts file\n");
376 rc = SEPOL_ERR;
377 goto exit;
378 }
379
380 fclose(file_contexts);
381 file_contexts = NULL;
382
383 rc = SEPOL_OK;
384
385 exit:
386 if (binary != NULL) {
387 fclose(binary);
388 }
389 if (file != NULL) {
390 fclose(file);
391 }
392 free(buffer);
393 free(output);
394 free(filecontexts);
395 cil_db_destroy(&db);
396 sepol_policydb_free(pdb);
397 sepol_policy_file_free(pf);
398 free(fc_buf);
399 return rc;
400 }
401