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:     "\n        -R      Recurse"
  43 //usage:     "\n        -v      Set the file's version/generation number"
  44
  45 #include "libbb.h"
  46 #include "e2fs_lib.h"
  47
  48 #define OPT_ADD 1
  49 #define OPT_REM 2
  50 #define OPT_SET 4
  51 #define OPT_SET_VER 8
  52
  53 struct globals {
  54         unsigned long version;
  55         unsigned long af;
  56         unsigned long rf;
  57         smallint flags;
  58         smallint recursive;
  59 };
  60
  61 static unsigned long get_flag(char c)
  62 {
  63         const char *fp = strchr(e2attr_flags_sname_chattr, c);
  64         if (fp)
  65                 return e2attr_flags_value_chattr[fp - e2attr_flags_sname_chattr];
  66         bb_show_usage();
  67 }
  68
  69 static int decode_arg(const char *arg, struct globals *gp)
  70 {
  71         unsigned long *fl;
  72         char opt = *arg++;
  73
  74         fl = &gp->af;
  75         if (opt == '-') {
  76                 gp->flags |= OPT_REM;
  77                 fl = &gp->rf;
  78         } else if (opt == '+') {
  79                 gp->flags |= OPT_ADD;
  80         } else if (opt == '=') {
  81                 gp->flags |= OPT_SET;
  82         } else
  83                 return 0;
  84
  85         while (*arg)
  86                 *fl |= get_flag(*arg++);
  87
  88         return 1;
  89 }
  90
  91 static void change_attributes(const char *name, struct globals *gp);
  92
  93 static int FAST_FUNC chattr_dir_proc(const char *dir_name, struct dirent *de, void *gp)
  94 {
  95         char *path = concat_subpath_file(dir_name, de->d_name);
  96         /* path is NULL if de->d_name is "." or "..", else... */
  97         if (path) {
  98                 change_attributes(path, gp);
  99                 free(path);
 100         }
 101         return 0;
 102 }
 103
 104 static void change_attributes(const char *name, struct globals *gp)
 105 {
 106         unsigned long fsflags;
 107         struct stat st;
 108
 109         if (lstat(name, &st) != 0) {
 110                 bb_perror_msg("stat %s", name);
 111                 return;
 112         }
 113         if (S_ISLNK(st.st_mode) && gp->recursive)
 114                 return;
 115
 116         /* Don't try to open device files, fifos etc.  We probably
 117          * ought to display an error if the file was explicitly given
 118          * on the command line (whether or not recursive was
 119          * requested).  */
 120         if (!S_ISREG(st.st_mode) && !S_ISLNK(st.st_mode) && !S_ISDIR(st.st_mode))
 121                 return;
 122
 123         if (gp->flags & OPT_SET_VER)
 124                 if (fsetversion(name, gp->version) != 0)
 125                         bb_perror_msg("setting version on %s", name);
 126
 127         if (gp->flags & OPT_SET) {
 128                 fsflags = gp->af;
 129         } else {
 130                 if (fgetflags(name, &fsflags) != 0) {
 131                         bb_perror_msg("reading flags on %s", name);
 132                         goto skip_setflags;
 133                 }
 134                 /*if (gp->flags & OPT_REM) - not needed, rf is zero otherwise */
 135                         fsflags &= ~gp->rf;
 136                 /*if (gp->flags & OPT_ADD) - not needed, af is zero otherwise */
 137                         fsflags |= gp->af;
 138                 /* What is this? And why it's not done for SET case? */
 139                 if (!S_ISDIR(st.st_mode))
 140                         fsflags &= ~EXT2_DIRSYNC_FL;
 141         }
 142         if (fsetflags(name, fsflags) != 0)
 143                 bb_perror_msg("setting flags on %s", name);
 144
 145  skip_setflags:
 146         if (gp->recursive && S_ISDIR(st.st_mode))
 147                 iterate_on_dir(name, chattr_dir_proc, gp);
 148 }
 149
 150 int chattr_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 151 int chattr_main(int argc UNUSED_PARAM, char **argv)
 152 {
 153         struct globals g;
 154         char *arg;
 155
 156         memset(&g, 0, sizeof(g));
 157
 158         /* parse the args */
 159         while ((arg = *++argv)) {
 160                 /* take care of -R and -v <version> */
 161                 if (arg[0] == '-'
 162                  && (arg[1] == 'R' || arg[1] == 'v')
 163                  && !arg[2]
 164                 ) {
 165                         if (arg[1] == 'R') {
 166                                 g.recursive = 1;
 167                                 continue;
 168                         }
 169                         /* arg[1] == 'v' */
 170                         if (!*++argv)
 171                                 bb_show_usage();
 172                         g.version = xatoul(*argv);
 173                         g.flags |= OPT_SET_VER;
 174                         continue;
 175                 }
 176
 177                 if (!decode_arg(arg, &g))
 178                         break;
 179         }
 180
 181         /* run sanity checks on all the arguments given us */
 182         if (!*argv)
 183                 bb_show_usage();
 184         if ((g.flags & OPT_SET) && (g.flags & (OPT_ADD|OPT_REM)))
 185                 bb_error_msg_and_die("= is incompatible with - and +");
 186         if (g.rf & g.af)
 187                 bb_error_msg_and_die("can't set and unset a flag");
 188         if (!g.flags)
 189                 bb_error_msg_and_die("must use '-v', =, - or +");
 190
 191         /* now run chattr on all the files passed to us */
 192         do change_attributes(*argv, &g); while (*++argv);
 193
 194         return EXIT_SUCCESS;
 195 }