use gnulib's progname module
[platform/upstream/coreutils.git] / src / chcon.c
1 /* chcon -- change security context of files
2    Copyright (C) 2005-2008 Free Software Foundation, Inc.
3
4    This program is free software: you can redistribute it and/or modify
5    it under the terms of the GNU General Public License as published by
6    the Free Software Foundation, either version 3 of the License, or
7    (at your option) any later version.
8
9    This program is distributed in the hope that it will be useful,
10    but WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12    GNU General Public License for more details.
13
14    You should have received a copy of the GNU General Public License
15    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
16
17 #include <config.h>
18 #include <stdio.h>
19 #include <sys/types.h>
20 #include <getopt.h>
21
22 #include "system.h"
23 #include "dev-ino.h"
24 #include "error.h"
25 #include "quote.h"
26 #include "quotearg.h"
27 #include "root-dev-ino.h"
28 #include "selinux-at.h"
29 #include "xfts.h"
30
31 /* The official name of this program (e.g., no `g' prefix).  */
32 #define PROGRAM_NAME "chcon"
33
34 #define AUTHORS \
35   proper_name ("Russell Coker"), \
36   proper_name ("Jim Meyering")
37
38 enum Change_status
39 {
40   CH_NOT_APPLIED,
41   CH_SUCCEEDED,
42   CH_FAILED,
43   CH_NO_CHANGE_REQUESTED
44 };
45
46 enum Verbosity
47 {
48   /* Print a message for each file that is processed.  */
49   V_high,
50
51   /* Print a message for each file whose attributes we change.  */
52   V_changes_only,
53
54   /* Do not be verbose.  This is the default. */
55   V_off
56 };
57
58 /* If nonzero, and the systems has support for it, change the context
59    of symbolic links rather than any files they point to.  */
60 static bool affect_symlink_referent;
61
62 /* If true, change the modes of directories recursively. */
63 static bool recurse;
64
65 /* Level of verbosity. */
66 static bool verbose;
67
68 /* Pointer to the device and inode numbers of `/', when --recursive.
69    Otherwise NULL.  */
70 static struct dev_ino *root_dev_ino;
71
72 /* The name of the context file is being given. */
73 static char const *specified_context;
74
75 /* Specific components of the context */
76 static char const *specified_user;
77 static char const *specified_role;
78 static char const *specified_range;
79 static char const *specified_type;
80
81 /* For long options that have no equivalent short option, use a
82    non-character as a pseudo short option, starting with CHAR_MAX + 1.  */
83 enum
84 {
85   DEREFERENCE_OPTION = CHAR_MAX + 1,
86   NO_PRESERVE_ROOT,
87   PRESERVE_ROOT,
88   REFERENCE_FILE_OPTION
89 };
90
91 static struct option const long_options[] =
92 {
93   {"recursive", no_argument, NULL, 'R'},
94   {"dereference", no_argument, NULL, DEREFERENCE_OPTION},
95   {"no-dereference", no_argument, NULL, 'h'},
96   {"no-preserve-root", no_argument, NULL, NO_PRESERVE_ROOT},
97   {"preserve-root", no_argument, NULL, PRESERVE_ROOT},
98   {"reference", required_argument, NULL, REFERENCE_FILE_OPTION},
99   {"user", required_argument, NULL, 'u'},
100   {"role", required_argument, NULL, 'r'},
101   {"type", required_argument, NULL, 't'},
102   {"range", required_argument, NULL, 'l'},
103   {"verbose", no_argument, NULL, 'v'},
104   {GETOPT_HELP_OPTION_DECL},
105   {GETOPT_VERSION_OPTION_DECL},
106   {NULL, 0, NULL, 0}
107 };
108
109 /* Given a security context, CONTEXT, derive a context_t (*RET),
110    setting any portions selected via the global variables, specified_user,
111    specified_role, etc.  */
112 static int
113 compute_context_from_mask (security_context_t context, context_t *ret)
114 {
115   bool ok = true;
116   context_t new_context = context_new (context);
117   if (!new_context)
118     {
119       error (0, errno, _("failed to create security context: %s"),
120              quotearg_colon (context));
121       return 1;
122     }
123
124 #define SET_COMPONENT(C, comp)                                          \
125    do                                                                   \
126      {                                                                  \
127        if (specified_ ## comp                                           \
128            && context_ ## comp ## _set ((C), specified_ ## comp))       \
129          {                                                              \
130             error (0, errno,                                            \
131                    _("failed to set %s security context component to %s"), \
132                    #comp, quote (specified_ ## comp));                  \
133            ok = false;                                                  \
134          }                                                              \
135      }                                                                  \
136    while (0)
137
138   SET_COMPONENT (new_context, user);
139   SET_COMPONENT (new_context, range);
140   SET_COMPONENT (new_context, role);
141   SET_COMPONENT (new_context, type);
142
143   if (!ok)
144     {
145       int saved_errno = errno;
146       context_free (new_context);
147       errno = saved_errno;
148       return 1;
149     }
150
151   *ret = new_context;
152   return 0;
153 }
154
155 /* Change the context of FILE, using specified components.
156    If it is a directory and -R is given, recurse.
157    Return 0 if successful, 1 if errors occurred. */
158
159 static int
160 change_file_context (int fd, char const *file)
161 {
162   security_context_t file_context = NULL;
163   context_t context;
164   security_context_t context_string;
165   int errors = 0;
166
167   if (specified_context == NULL)
168     {
169       int status = (affect_symlink_referent
170                     ? getfileconat (fd, file, &file_context)
171                     : lgetfileconat (fd, file, &file_context));
172
173       if (status < 0 && errno != ENODATA)
174         {
175           error (0, errno, _("failed to get security context of %s"),
176                  quote (file));
177           return 1;
178         }
179
180       /* If the file doesn't have a context, and we're not setting all of
181          the context components, there isn't really an obvious default.
182          Thus, we just give up. */
183       if (file_context == NULL)
184         {
185           error (0, 0, _("can't apply partial context to unlabeled file %s"),
186                  quote (file));
187           return 1;
188         }
189
190       if (compute_context_from_mask (file_context, &context))
191         return 1;
192     }
193   else
194     {
195       /* FIXME: this should be done exactly once, in main.  */
196       context = context_new (specified_context);
197       if (!context)
198         abort ();
199     }
200
201   context_string = context_str (context);
202
203   if (file_context == NULL || ! STREQ (context_string, file_context))
204     {
205       int fail = (affect_symlink_referent
206                   ?  setfileconat (fd, file, context_string)
207                   : lsetfileconat (fd, file, context_string));
208
209       if (fail)
210         {
211           errors = 1;
212           error (0, errno, _("failed to change context of %s to %s"),
213                  quote_n (0, file), quote_n (1, context_string));
214         }
215     }
216
217   context_free (context);
218   freecon (file_context);
219
220   return errors;
221 }
222
223 /* Change the context of FILE.
224    Return true if successful.  This function is called
225    once for every file system object that fts encounters.  */
226
227 static bool
228 process_file (FTS *fts, FTSENT *ent)
229 {
230   char const *file_full_name = ent->fts_path;
231   char const *file = ent->fts_accpath;
232   const struct stat *file_stats = ent->fts_statp;
233   bool ok = true;
234
235   switch (ent->fts_info)
236     {
237     case FTS_D:
238       if (recurse)
239         {
240           if (ROOT_DEV_INO_CHECK (root_dev_ino, ent->fts_statp))
241             {
242               /* This happens e.g., with "chcon -R --preserve-root ... /"
243                  and with "chcon -RH --preserve-root ... symlink-to-root".  */
244               ROOT_DEV_INO_WARN (file_full_name);
245               /* Tell fts not to traverse into this hierarchy.  */
246               fts_set (fts, ent, FTS_SKIP);
247               /* Ensure that we do not process "/" on the second visit.  */
248               ent = fts_read (fts);
249               return false;
250             }
251           return true;
252         }
253       break;
254
255     case FTS_DP:
256       if (! recurse)
257         return true;
258       break;
259
260     case FTS_NS:
261       /* For a top-level file or directory, this FTS_NS (stat failed)
262          indicator is determined at the time of the initial fts_open call.
263          With programs like chmod, chown, and chgrp, that modify
264          permissions, it is possible that the file in question is
265          accessible when control reaches this point.  So, if this is
266          the first time we've seen the FTS_NS for this file, tell
267          fts_read to stat it "again".  */
268       if (ent->fts_level == 0 && ent->fts_number == 0)
269         {
270           ent->fts_number = 1;
271           fts_set (fts, ent, FTS_AGAIN);
272           return true;
273         }
274       error (0, ent->fts_errno, _("cannot access %s"), quote (file_full_name));
275       ok = false;
276       break;
277
278     case FTS_ERR:
279       error (0, ent->fts_errno, _("%s"), quote (file_full_name));
280       ok = false;
281       break;
282
283     case FTS_DNR:
284       error (0, ent->fts_errno, _("cannot read directory %s"),
285              quote (file_full_name));
286       ok = false;
287       break;
288
289     default:
290       break;
291     }
292
293   if (ent->fts_info == FTS_DP
294       && ok && ROOT_DEV_INO_CHECK (root_dev_ino, file_stats))
295     {
296       ROOT_DEV_INO_WARN (file_full_name);
297       ok = false;
298     }
299
300   if (ok)
301     {
302       if (verbose)
303         printf (_("changing security context of %s"),
304                 quote (file_full_name));
305
306       if (change_file_context (fts->fts_cwd_fd, file) != 0)
307         ok = false;
308     }
309
310   if ( ! recurse)
311     fts_set (fts, ent, FTS_SKIP);
312
313   return ok;
314 }
315
316 /* Recursively operate on the specified FILES (the last entry
317    of which is NULL).  BIT_FLAGS controls how fts works.
318    Return true if successful.  */
319
320 static bool
321 process_files (char **files, int bit_flags)
322 {
323   bool ok = true;
324
325   FTS *fts = xfts_open (files, bit_flags, NULL);
326
327   while (1)
328     {
329       FTSENT *ent;
330
331       ent = fts_read (fts);
332       if (ent == NULL)
333         {
334           if (errno != 0)
335             {
336               /* FIXME: try to give a better message  */
337               error (0, errno, _("fts_read failed"));
338               ok = false;
339             }
340           break;
341         }
342
343       ok &= process_file (fts, ent);
344     }
345
346   /* Ignore failure, since the only way it can do so is in failing to
347      return to the original directory, and since we're about to exit,
348      that doesn't matter.  */
349   fts_close (fts);
350
351   return ok;
352 }
353
354 void
355 usage (int status)
356 {
357   if (status != EXIT_SUCCESS)
358     fprintf (stderr, _("Try `%s --help' for more information.\n"),
359              program_name);
360   else
361     {
362       printf (_("\
363 Usage: %s [OPTION]... CONTEXT FILE...\n\
364   or:  %s [OPTION]... [-u USER] [-r ROLE] [-l RANGE] [-t TYPE] FILE...\n\
365   or:  %s [OPTION]... --reference=RFILE FILE...\n\
366 "),
367         program_name, program_name, program_name);
368       fputs (_("\
369 Change the security context of each FILE to CONTEXT.\n\
370 With --reference, change the security context of each FILE to that of RFILE.\n\
371 \n\
372   -c, --changes          like verbose but report only when a change is made\n\
373   -h, --no-dereference   affect symbolic links instead of any referenced file\n\
374 "), stdout);
375       fputs (_("\
376       --reference=RFILE  use RFILE's security context rather than specifying\n\
377                          a CONTEXT value\n\
378   -R, --recursive        operate on files and directories recursively\n\
379   -v, --verbose          output a diagnostic for every file processed\n\
380 "), stdout);
381       fputs (_("\
382   -u, --user=USER        set user USER in the target security context\n\
383   -r, --role=ROLE        set role ROLE in the target security context\n\
384   -t, --type=TYPE        set type TYPE in the target security context\n\
385   -l, --range=RANGE      set range RANGE in the target security context\n\
386 \n\
387 "), stdout);
388       fputs (_("\
389 The following options modify how a hierarchy is traversed when the -R\n\
390 option is also specified.  If more than one is specified, only the final\n\
391 one takes effect.\n\
392 \n\
393   -H                     if a command line argument is a symbolic link\n\
394                          to a directory, traverse it\n\
395   -L                     traverse every symbolic link to a directory\n\
396                          encountered\n\
397   -P                     do not traverse any symbolic links (default)\n\
398 \n\
399 "), stdout);
400       fputs (HELP_OPTION_DESCRIPTION, stdout);
401       fputs (VERSION_OPTION_DESCRIPTION, stdout);
402       emit_bug_reporting_address ();
403     }
404   exit (status);
405 }
406
407 int
408 main (int argc, char **argv)
409 {
410   security_context_t ref_context = NULL;
411
412   /* Bit flags that control how fts works.  */
413   int bit_flags = FTS_PHYSICAL;
414
415   /* 1 if --dereference, 0 if --no-dereference, -1 if neither has been
416      specified.  */
417   int dereference = -1;
418
419   bool ok;
420   bool preserve_root = false;
421   bool component_specified = false;
422   char *reference_file = NULL;
423   int optc;
424
425   initialize_main (&argc, &argv);
426   set_program_name (argv[0]);
427   setlocale (LC_ALL, "");
428   bindtextdomain (PACKAGE, LOCALEDIR);
429   textdomain (PACKAGE);
430
431   atexit (close_stdout);
432
433   while ((optc = getopt_long (argc, argv, "HLPRchvu:r:t:l:", long_options, NULL))
434          != -1)
435     {
436       switch (optc)
437         {
438         case 'H': /* Traverse command-line symlinks-to-directories.  */
439           bit_flags = FTS_COMFOLLOW | FTS_PHYSICAL;
440           break;
441
442         case 'L': /* Traverse all symlinks-to-directories.  */
443           bit_flags = FTS_LOGICAL;
444           break;
445
446         case 'P': /* Traverse no symlinks-to-directories.  */
447           bit_flags = FTS_PHYSICAL;
448           break;
449
450         case 'h': /* --no-dereference: affect symlinks */
451           dereference = 0;
452           break;
453
454         case DEREFERENCE_OPTION: /* --dereference: affect the referent
455                                     of each symlink */
456           dereference = 1;
457           break;
458
459         case NO_PRESERVE_ROOT:
460           preserve_root = false;
461           break;
462
463         case PRESERVE_ROOT:
464           preserve_root = true;
465           break;
466
467         case REFERENCE_FILE_OPTION:
468           reference_file = optarg;
469           break;
470
471         case 'R':
472           recurse = true;
473           break;
474
475         case 'f':
476           /* ignore */
477           break;
478
479         case 'v':
480           verbose = true;
481           break;
482
483         case 'u':
484           specified_user = optarg;
485           component_specified = true;
486           break;
487
488         case 'r':
489           specified_role = optarg;
490           component_specified = true;
491           break;
492
493         case 't':
494           specified_type = optarg;
495           component_specified = true;
496           break;
497
498         case 'l':
499           specified_range = optarg;
500           component_specified = true;
501           break;
502
503         case_GETOPT_HELP_CHAR;
504         case_GETOPT_VERSION_CHAR (PROGRAM_NAME, AUTHORS);
505         default:
506           usage (EXIT_FAILURE);
507         }
508     }
509
510   if (recurse)
511     {
512       if (bit_flags == FTS_PHYSICAL)
513         {
514           if (dereference == 1)
515             error (EXIT_FAILURE, 0,
516                    _("-R --dereference requires either -H or -L"));
517           affect_symlink_referent = false;
518         }
519       else
520         {
521           if (dereference == 0)
522             error (EXIT_FAILURE, 0, _("-R -h requires -P"));
523           affect_symlink_referent = true;
524         }
525     }
526   else
527     {
528       bit_flags = FTS_PHYSICAL;
529       affect_symlink_referent = (dereference != 0);
530     }
531
532   if (argc - optind < (reference_file || component_specified ? 1 : 2))
533     {
534       if (argc <= optind)
535         error (0, 0, _("missing operand"));
536       else
537         error (0, 0, _("missing operand after %s"), quote (argv[argc - 1]));
538       usage (EXIT_FAILURE);
539     }
540
541   if (reference_file)
542     {
543       if (getfilecon (reference_file, &ref_context) < 0)
544         error (EXIT_FAILURE, errno, _("failed to get security context of %s"),
545                quote (reference_file));
546
547       specified_context = ref_context;
548     }
549   else if (component_specified)
550     {
551       /* FIXME: it's already null, so this is a no-op. */
552       specified_context = NULL;
553     }
554   else
555     {
556       context_t context;
557       specified_context = argv[optind++];
558       context = context_new (specified_context);
559       if (!context)
560         error (EXIT_FAILURE, 0, _("invalid context: %s"),
561                quotearg_colon (specified_context));
562       context_free (context);
563     }
564
565   if (reference_file && component_specified)
566     {
567       error (0, 0, _("conflicting security context specifiers given"));
568       usage (1);
569     }
570
571   if (recurse & preserve_root)
572     {
573       static struct dev_ino dev_ino_buf;
574       root_dev_ino = get_root_dev_ino (&dev_ino_buf);
575       if (root_dev_ino == NULL)
576         error (EXIT_FAILURE, errno, _("failed to get attributes of %s"),
577                quote ("/"));
578     }
579   else
580     {
581       root_dev_ino = NULL;
582     }
583
584   ok = process_files (argv + optind, bit_flags | FTS_NOSTAT);
585
586   exit (ok ? EXIT_SUCCESS : EXIT_FAILURE);
587 }