e2fsprogs/e2fs_lib.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * See README for additional information
   4  *
   5  * This file can be redistributed under the terms of the GNU Library General
   6  * Public License
   7  */
   8
   9 #include "libbb.h"
  10 #include "e2fs_lib.h"
  11
  12 #define HAVE_EXT2_IOCTLS 1
  13
  14 #if INT_MAX == LONG_MAX
  15 #define IF_LONG_IS_SAME(...) __VA_ARGS__
  16 #define IF_LONG_IS_WIDER(...)
  17 #else
  18 #define IF_LONG_IS_SAME(...)
  19 #define IF_LONG_IS_WIDER(...) __VA_ARGS__
  20 #endif
  21
  22 static void close_silently(int fd)
  23 {
  24         int e = errno;
  25         close(fd);
  26         errno = e;
  27 }
  28
  29
  30 /* Iterate a function on each entry of a directory */
  31 int iterate_on_dir(const char *dir_name,
  32                 int (*func)(const char *, struct dirent *, void *),
  33                 void * private)
  34 {
  35         DIR *dir;
  36         struct dirent *de, *dep;
  37         int max_len, len;
  38
  39         max_len = PATH_MAX + sizeof(struct dirent);
  40         de = xmalloc(max_len+1);
  41         memset(de, 0, max_len+1);
  42
  43         dir = opendir(dir_name);
  44         if (dir == NULL) {
  45                 free(de);
  46                 return -1;
  47         }
  48         while ((dep = readdir(dir))) {
  49                 len = sizeof(struct dirent);
  50                 if (len < dep->d_reclen)
  51                         len = dep->d_reclen;
  52                 if (len > max_len)
  53                         len = max_len;
  54                 memcpy(de, dep, len);
  55                 func(dir_name, de, private);
  56         }
  57         closedir(dir);
  58         free(de);
  59         return 0;
  60 }
  61
  62
  63 /* Get/set a file version on an ext2 file system */
  64 int fgetsetversion(const char *name, unsigned long *get_version, unsigned long set_version)
  65 {
  66 #if HAVE_EXT2_IOCTLS
  67         int fd, r;
  68         IF_LONG_IS_WIDER(int ver;)
  69
  70         fd = open(name, O_NONBLOCK);
  71         if (fd == -1)
  72                 return -1;
  73         if (!get_version) {
  74                 IF_LONG_IS_WIDER(
  75                         ver = (int) set_version;
  76                         r = ioctl(fd, EXT2_IOC_SETVERSION, &ver);
  77                 )
  78                 IF_LONG_IS_SAME(
  79                         r = ioctl(fd, EXT2_IOC_SETVERSION, (void*)&set_version);
  80                 )
  81         } else {
  82                 IF_LONG_IS_WIDER(
  83                         r = ioctl(fd, EXT2_IOC_GETVERSION, &ver);
  84                         *get_version = ver;
  85                 )
  86                 IF_LONG_IS_SAME(
  87                         r = ioctl(fd, EXT2_IOC_GETVERSION, (void*)get_version);
  88                 )
  89         }
  90         close_silently(fd);
  91         return r;
  92 #else /* ! HAVE_EXT2_IOCTLS */
  93         errno = EOPNOTSUPP;
  94         return -1;
  95 #endif /* ! HAVE_EXT2_IOCTLS */
  96 }
  97
  98
  99 /* Get/set a file flags on an ext2 file system */
 100 int fgetsetflags(const char *name, unsigned long *get_flags, unsigned long set_flags)
 101 {
 102 #if HAVE_EXT2_IOCTLS
 103         struct stat buf;
 104         int fd, r;
 105         IF_LONG_IS_WIDER(int f;)
 106
 107         if (stat(name, &buf) == 0 /* stat is ok */
 108          && !S_ISREG(buf.st_mode) && !S_ISDIR(buf.st_mode)
 109         ) {
 110                 goto notsupp;
 111         }
 112         fd = open(name, O_NONBLOCK); /* neither read nor write asked for */
 113         if (fd == -1)
 114                 return -1;
 115
 116         if (!get_flags) {
 117                 IF_LONG_IS_WIDER(
 118                         f = (int) set_flags;
 119                         r = ioctl(fd, EXT2_IOC_SETFLAGS, &f);
 120                 )
 121                 IF_LONG_IS_SAME(
 122                         r = ioctl(fd, EXT2_IOC_SETFLAGS, (void*)&set_flags);
 123                 )
 124         } else {
 125                 IF_LONG_IS_WIDER(
 126                         r = ioctl(fd, EXT2_IOC_GETFLAGS, &f);
 127                         *get_flags = f;
 128                 )
 129                 IF_LONG_IS_SAME(
 130                         r = ioctl(fd, EXT2_IOC_GETFLAGS, (void*)get_flags);
 131                 )
 132         }
 133
 134         close_silently(fd);
 135         return r;
 136  notsupp:
 137 #endif /* HAVE_EXT2_IOCTLS */
 138         errno = EOPNOTSUPP;
 139         return -1;
 140 }
 141
 142
 143 /* Print file attributes on an ext2 file system */
 144 struct flags_name {
 145         unsigned long   flag;
 146         char            short_name;
 147         const char      *long_name;
 148 };
 149
 150 /* TODO: apart from I and (disabled) COMPRESSION flags, this
 151  * is a duplicate of a table from chattr. Merge? */
 152 static const struct flags_name flags_array[] = {
 153         { EXT2_SECRM_FL, 's', "Secure_Deletion" },
 154         { EXT2_UNRM_FL, 'u' , "Undelete" },
 155         { EXT2_SYNC_FL, 'S', "Synchronous_Updates" },
 156         { EXT2_DIRSYNC_FL, 'D', "Synchronous_Directory_Updates" },
 157         { EXT2_IMMUTABLE_FL, 'i', "Immutable" },
 158         { EXT2_APPEND_FL, 'a', "Append_Only" },
 159         { EXT2_NODUMP_FL, 'd', "No_Dump" },
 160         { EXT2_NOATIME_FL, 'A', "No_Atime" },
 161         { EXT2_COMPR_FL, 'c', "Compression_Requested" },
 162 #ifdef ENABLE_COMPRESSION
 163         { EXT2_COMPRBLK_FL, 'B', "Compressed_File" },
 164         { EXT2_DIRTY_FL, 'Z', "Compressed_Dirty_File" },
 165         { EXT2_NOCOMPR_FL, 'X', "Compression_Raw_Access" },
 166         { EXT2_ECOMPR_FL, 'E', "Compression_Error" },
 167 #endif
 168         { EXT3_JOURNAL_DATA_FL, 'j', "Journaled_Data" },
 169         { EXT2_INDEX_FL, 'I', "Indexed_directory" },
 170         { EXT2_NOTAIL_FL, 't', "No_Tailmerging" },
 171         { EXT2_TOPDIR_FL, 'T', "Top_of_Directory_Hierarchies" },
 172         { 0, '\0', NULL }
 173 };
 174
 175 void print_flags(FILE *f, unsigned long flags, unsigned options)
 176 {
 177         const struct flags_name *fp;
 178
 179         if (options & PFOPT_LONG) {
 180                 int first = 1;
 181                 for (fp = flags_array; fp->short_name; fp++) {
 182                         if (flags & fp->flag) {
 183                                 if (!first)
 184                                         fputs(", ", f);
 185                                 fputs(fp->long_name, f);
 186                                 first = 0;
 187                         }
 188                 }
 189                 if (first)
 190                         fputs("---", f);
 191         } else {
 192                 for (fp = flags_array; fp->short_name; fp++) {
 193                         char c = '-';
 194                         if (flags & fp->flag)
 195                                 c = fp->short_name;
 196                         fputc(c, f);
 197                 }
 198         }
 199 }