e2fsprogs/chattr.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * chattr.c             - Change file attributes on an ext2 file system
   4  *
   5  * Copyright (C) 1993, 1994  Remy Card <card@masi.ibp.fr>
   6  *                           Laboratoire MASI, Institut Blaise Pascal
   7  *                           Universite Pierre et Marie Curie (Paris VI)
   8  *
   9  * This file can be redistributed under the terms of the GNU General
  10  * Public License
  11  */
  12
  13 /*
  14  * History:
  15  * 93/10/30     - Creation
  16  * 93/11/13     - Replace stat() calls by lstat() to avoid loops
  17  * 94/02/27     - Integrated in Ted's distribution
  18  * 98/12/29     - Ignore symlinks when working recursively (G M Sipe)
  19  * 98/12/29     - Display version info only when -V specified (G M Sipe)
  20  */
  21
  22 //usage:#define chattr_trivial_usage
  23 //usage:       "[-R] [-+=AacDdijsStTu] [-v VERSION] [FILE]..."
  24 //usage:#define chattr_full_usage "\n\n"
  25 //usage:       "Change file attributes on an ext2 fs\n"
  26 //usage:     "\nModifiers:"
  27 //usage:     "\n        -       Remove attributes"
  28 //usage:     "\n        +       Add attributes"
  29 //usage:     "\n        =       Set attributes"
  30 //usage:     "\nAttributes:"
  31 //usage:     "\n        A       Don't track atime"
  32 //usage:     "\n        a       Append mode only"
  33 //usage:     "\n        c       Enable compress"
  34 //usage:     "\n        D       Write dir contents synchronously"
  35 //usage:     "\n        d       Don't backup with dump"
  36 //usage:     "\n        i       Cannot be modified (immutable)"
  37 //usage:     "\n        j       Write all data to journal first"
  38 //usage:     "\n        s       Zero disk storage when deleted"
  39 //usage:     "\n        S       Write file contents synchronously"
  40 //usage:     "\n        t       Disable tail-merging of partial blocks with other files"
  41 //usage:     "\n        u       Allow file to be undeleted"
  42 //usage:     "\nOptions:"
  43 //usage:     "\n        -R      Recurse"
  44 //usage:     "\n        -v      Set the file's version/generation number"
  45
  46 #include "libbb.h"
  47 #include "e2fs_lib.h"
  48
  49 #define OPT_ADD 1
  50 #define OPT_REM 2
  51 #define OPT_SET 4
  52 #define OPT_SET_VER 8
  53
  54 struct globals {
  55         unsigned long version;
  56         unsigned long af;
  57         unsigned long rf;
  58         smallint flags;
  59         smallint recursive;
  60 };
  61
  62 static unsigned long get_flag(char c)
  63 {
  64         const char *fp = strchr(e2attr_flags_sname_chattr, c);
  65         if (fp)
  66                 return e2attr_flags_value_chattr[fp - e2attr_flags_sname_chattr];
  67         bb_show_usage();
  68 }
  69
  70 static int decode_arg(const char *arg, struct globals *gp)
  71 {
  72         unsigned long *fl;
  73         char opt = *arg++;
  74
  75         fl = &gp->af;
  76         if (opt == '-') {
  77                 gp->flags |= OPT_REM;
  78                 fl = &gp->rf;
  79         } else if (opt == '+') {
  80                 gp->flags |= OPT_ADD;
  81         } else if (opt == '=') {
  82                 gp->flags |= OPT_SET;
  83         } else
  84                 return 0;
  85
  86         while (*arg)
  87                 *fl |= get_flag(*arg++);
  88
  89         return 1;
  90 }
  91
  92 static void change_attributes(const char *name, struct globals *gp);
  93
  94 static int FAST_FUNC chattr_dir_proc(const char *dir_name, struct dirent *de, void *gp)
  95 {
  96         char *path = concat_subpath_file(dir_name, de->d_name);
  97         /* path is NULL if de->d_name is "." or "..", else... */
  98         if (path) {
  99                 change_attributes(path, gp);
 100                 free(path);
 101         }
 102         return 0;
 103 }
 104
 105 static void change_attributes(const char *name, struct globals *gp)
 106 {
 107         unsigned long fsflags;
 108         struct stat st;
 109
 110         if (lstat(name, &st) != 0) {
 111                 bb_perror_msg("stat %s", name);
 112                 return;
 113         }
 114         if (S_ISLNK(st.st_mode) && gp->recursive)
 115                 return;
 116
 117         /* Don't try to open device files, fifos etc.  We probably
 118          * ought to display an error if the file was explicitly given
 119          * on the command line (whether or not recursive was
 120          * requested).  */
 121         if (!S_ISREG(st.st_mode) && !S_ISLNK(st.st_mode) && !S_ISDIR(st.st_mode))
 122                 return;
 123
 124         if (gp->flags & OPT_SET_VER)
 125                 if (fsetversion(name, gp->version) != 0)
 126                         bb_perror_msg("setting version on %s", name);
 127
 128         if (gp->flags & OPT_SET) {
 129                 fsflags = gp->af;
 130         } else {
 131                 if (fgetflags(name, &fsflags) != 0) {
 132                         bb_perror_msg("reading flags on %s", name);
 133                         goto skip_setflags;
 134                 }
 135                 /*if (gp->flags & OPT_REM) - not needed, rf is zero otherwise */
 136                         fsflags &= ~gp->rf;
 137                 /*if (gp->flags & OPT_ADD) - not needed, af is zero otherwise */
 138                         fsflags |= gp->af;
 139                 /* What is this? And why it's not done for SET case? */
 140                 if (!S_ISDIR(st.st_mode))
 141                         fsflags &= ~EXT2_DIRSYNC_FL;
 142         }
 143         if (fsetflags(name, fsflags) != 0)
 144                 bb_perror_msg("setting flags on %s", name);
 145
 146  skip_setflags:
 147         if (gp->recursive && S_ISDIR(st.st_mode))
 148                 iterate_on_dir(name, chattr_dir_proc, gp);
 149 }
 150
 151 int chattr_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 152 int chattr_main(int argc UNUSED_PARAM, char **argv)
 153 {
 154         struct globals g;
 155         char *arg;
 156
 157         memset(&g, 0, sizeof(g));
 158
 159         /* parse the args */
 160         while ((arg = *++argv)) {
 161                 /* take care of -R and -v <version> */
 162                 if (arg[0] == '-'
 163                  && (arg[1] == 'R' || arg[1] == 'v')
 164                  && !arg[2]
 165                 ) {
 166                         if (arg[1] == 'R') {
 167                                 g.recursive = 1;
 168                                 continue;
 169                         }
 170                         /* arg[1] == 'v' */
 171                         if (!*++argv)
 172                                 bb_show_usage();
 173                         g.version = xatoul(*argv);
 174                         g.flags |= OPT_SET_VER;
 175                         continue;
 176                 }
 177
 178                 if (!decode_arg(arg, &g))
 179                         break;
 180         }
 181
 182         /* run sanity checks on all the arguments given us */
 183         if (!*argv)
 184                 bb_show_usage();
 185         if ((g.flags & OPT_SET) && (g.flags & (OPT_ADD|OPT_REM)))
 186                 bb_error_msg_and_die("= is incompatible with - and +");
 187         if (g.rf & g.af)
 188                 bb_error_msg_and_die("can't set and unset a flag");
 189         if (!g.flags)
 190                 bb_error_msg_and_die("must use '-v', =, - or +");
 191
 192         /* now run chattr on all the files passed to us */
 193         do change_attributes(*argv, &g); while (*++argv);
 194
 195         return EXIT_SUCCESS;
 196 }