changeset 1272:17935382d2c1 draft

Attached is the patch for lsattr and chattr implementation. This is the modified and better version from the last one I sent. It is having the both the mains (lsattr_main() and chattr_main() ) in single file. Also removed the dependency of additional file in lib, as common code is in the same .c file.
author Ashwini Sharma <ak.ashwini1981@gmail.com>
date Fri, 02 May 2014 06:17:48 -0500
parents c214de62b18b
children 79e847fec774
files toys/other/lsattr.c
diffstat 1 files changed, 320 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/toys/other/lsattr.c	Fri May 02 06:17:48 2014 -0500
@@ -0,0 +1,320 @@
+/* lsattr.c - List file attributes on a Linux second extended file system.
+ *
+ * Copyright 2013 Ranjan Kumar <ranjankumar.bth@gmail.com>
+ * Copyright 2013 Kyungwan Han <asura321@gmail.com>
+ *
+ * No Standard.
+
+USE_LSATTR(NEWTOY(lsattr, "vldaR", TOYFLAG_BIN))
+USE_CHATTR(NEWTOY(chattr, NULL, TOYFLAG_BIN))
+
+config LSATTR
+  bool "lsattr"
+  default y
+  help
+    usage: lsattr [-Radlv] [Files...]
+
+    List file attributes on a Linux second extended file system.
+
+    -R Recursively list attributes of directories and their contents.
+    -a List all files in directories, including files that start with '.'.
+    -d List directories like other files, rather than listing their contents.
+    -l List long flag names.
+    -v List the file's version/generation number.
+
+config CHATTR
+  bool "chattr"
+  default y
+  help
+    usage: chattr [-R] [-+=AacDdijsStTu] [-v version] [File...]
+
+    Change file attributes on a Linux second extended file system.
+
+    Operators:
+      '-' Remove attributes.
+      '+' Add attributes.
+      '=' Set attributes.
+
+    Attributes:
+      A  Don't track atime.
+      a  Append mode only.
+      c  Enable compress.
+      D  Write dir contents synchronously.
+      d  Don't backup with dump.
+      i  Cannot be modified (immutable).
+      j  Write all data to journal first.
+      s  Zero disk storage when deleted.
+      S  Write file contents synchronously.
+      t  Disable tail-merging of partial blocks with other files.
+      u  Allow file to be undeleted.
+      -R Recurse.
+      -v Set the file's version/generation number.
+
+*/
+#define FOR_lsattr
+#include "toys.h"
+#include <linux/fs.h>
+
+static struct ext2_attr {
+  char *name;
+  unsigned long flag;
+  char opt;
+} e2attrs[] = {
+  {"Secure_Deletion",               FS_SECRM_FL,        's'}, // Secure deletion
+  {"Undelete",                      FS_UNRM_FL,         'u'}, // Undelete
+  {"Compression_Requested",         FS_COMPR_FL,        'c'}, // Compress file
+  {"Synchronous_Updates",           FS_SYNC_FL,         'S'}, // Synchronous updates
+  {"Immutable",                     FS_IMMUTABLE_FL,    'i'}, // Immutable file
+  {"Append_Only",                   FS_APPEND_FL,       'a'}, // writes to file may only append
+  {"No_Dump",                       FS_NODUMP_FL,       'd'}, // do not dump file
+  {"No_Atime",                      FS_NOATIME_FL,      'A'}, // do not update atime
+  {"Indexed_directory",             FS_INDEX_FL,        'I'}, // hash-indexed directory
+  {"Journaled_Data",                FS_JOURNAL_DATA_FL, 'j'}, // file data should be journaled
+  {"No_Tailmerging",                FS_NOTAIL_FL,       't'}, // file tail should not be merged
+  {"Synchronous_Directory_Updates", FS_DIRSYNC_FL,      'D'}, // dirsync behaviour (directories only)
+  {"Top_of_Directory_Hierarchies",  FS_TOPDIR_FL,       'T'}, // Top of directory hierarchies
+  {NULL,                            -1,                   0},
+};
+
+// Get file flags on a Linux second extended file system.
+static int ext2_getflag(int fd, struct stat *sb, unsigned long *flag)
+{
+  if(!S_ISREG(sb->st_mode) && !S_ISDIR(sb->st_mode)) {
+    errno = EOPNOTSUPP;
+    return -1;
+  }
+  return (ioctl(fd, FS_IOC_GETFLAGS, (void*)flag));
+}
+
+static void print_file_attr(char *path)
+{
+  unsigned long flag = 0, version = 0;
+  int fd;
+  struct stat sb;
+
+  if (!stat(path, &sb) && !S_ISREG(sb.st_mode) && !S_ISDIR(sb.st_mode)) {
+    errno = EOPNOTSUPP;
+    goto LABEL1;
+  }
+  if (-1 == (fd=open(path, O_RDONLY | O_NONBLOCK))) goto LABEL1;
+
+  if (toys.optflags & FLAG_v) { 
+    if (ioctl(fd, FS_IOC_GETVERSION, (void*)&version) < 0) goto LABEL2;
+    xprintf("%5lu ", version);
+  }
+
+  if (ext2_getflag(fd, &sb, &flag) < 0) perror_msg("reading flags '%s'", path);
+  else {
+    struct ext2_attr *ptr = e2attrs;
+
+    if (toys.optflags & FLAG_l) {
+      int name_found = 0;
+
+      xprintf("%-50s ", path);
+      for (; ptr->name; ptr++) {
+        if (flag & ptr->flag) {
+          if (name_found) xprintf(", "); //for formatting.
+          xprintf("%s", ptr->name);
+          name_found = 1;
+        }
+      }
+      if (!name_found) xprintf("---");
+      xputc('\n');
+    } else {
+      int index = 0;
+
+      for (; ptr->name; ptr++)
+        toybuf[index++] = (flag & ptr->flag) ? ptr->opt : '-';
+      toybuf[index] = '\0';
+      xprintf("%s %s\n", toybuf, path);
+    }
+  }
+  xclose(fd);
+  return;
+LABEL2: xclose(fd);
+LABEL1: perror_msg("reading '%s'", path);
+}
+
+// Get directory information.
+static int retell_dir(struct dirtree *root)
+{
+  char *fpath = NULL;
+  
+  if (root->data == -1) {
+    xputc('\n');
+    return 0;
+  }
+  if (S_ISDIR(root->st.st_mode) && !root->parent) 
+    return (DIRTREE_RECURSE | DIRTREE_COMEAGAIN);
+
+  fpath = dirtree_path(root, NULL);
+  //Special case: with '-a' option and '.'/'..' also included in printing list.
+  if ((root->name[0] != '.') || (toys.optflags & FLAG_a)) {
+    print_file_attr(fpath);
+    if (S_ISDIR(root->st.st_mode) && (toys.optflags & FLAG_R)
+        && dirtree_notdotdot(root)) {
+      xprintf("\n%s:\n", fpath);
+      free(fpath);
+      return (DIRTREE_RECURSE | DIRTREE_COMEAGAIN);
+    }
+  }
+  free(fpath);
+  return 0;
+}
+
+void lsattr_main(void)
+{
+  if (!*toys.optargs) dirtree_read(".", retell_dir);
+  else
+    for (; *toys.optargs;  toys.optargs++) {
+      struct stat sb;
+
+      if (lstat(*toys.optargs, &sb)) perror_msg("stat '%s'", *toys.optargs);
+      else if (S_ISDIR(sb.st_mode) && !(toys.optflags & FLAG_d))
+        dirtree_read(*toys.optargs, retell_dir);
+      else print_file_attr(*toys.optargs);// to handle "./Filename" or "./Dir"
+    }
+}
+
+// Switch gears from lsattr to chattr.
+#define CLEANUP_lsattr
+#define FOR_chattr
+#include "generated/flags.h"
+
+static struct _chattr {
+  unsigned long add, rm, set, version;
+  unsigned char vflag, recursive;
+} chattr;
+
+static inline void chattr_help(void)
+{
+  toys.exithelp++;
+  error_exit("Invalid Argument");
+}
+
+// Set file flags on a Linux second extended file system.
+static inline int ext2_setflag(int fd, struct stat *sb, unsigned long flag)
+{
+  if (!S_ISREG(sb->st_mode) && !S_ISDIR(sb->st_mode)) {
+    errno = EOPNOTSUPP;
+    return -1;
+  }
+  return (ioctl(fd, FS_IOC_SETFLAGS, (void*)&flag));
+}
+
+static unsigned long get_flag_val(char ch)
+{
+  struct ext2_attr *ptr = e2attrs;
+
+  for (; ptr->name; ptr++)
+    if (ptr->opt == ch) return ptr->flag;
+  chattr_help(); // if no match found then Show help
+  return 0; // silent warning.
+}
+
+// Parse command line argument and fill the chattr structure.
+static void parse_cmdline_arg(char ***argv)
+{
+  char *arg = **argv, *ptr = NULL;
+
+  while (arg) {
+    switch (arg[0]) {
+      case '-':
+        for (ptr = ++arg; *ptr; ptr++) {
+          if (*ptr == 'R') {
+            chattr.recursive = 1;
+            continue;
+          } else if (*ptr == 'v') {// get version from next argv.
+            char *endptr;
+
+            errno = 0;
+            arg = *(*argv += 1);
+            if (!arg) chattr_help();
+            if (*arg == '-') perror_exit("Invalid Number '%s'", arg);
+            chattr.version = strtoul(arg, &endptr, 0);
+            if (errno || *endptr) perror_exit("bad version '%s'", arg);
+            chattr.vflag = 1;
+            continue;
+          } else chattr.rm |= get_flag_val(*ptr);
+        }
+        break;
+      case '+':
+        for (ptr = ++arg; *ptr; ptr++)
+          chattr.add |= get_flag_val(*ptr);
+        break;
+      case '=':
+        for (ptr = ++arg; *ptr; ptr++)
+          chattr.set |= get_flag_val(*ptr);
+        break;
+      default: return;
+    }
+    arg = *(*argv += 1);
+  }
+}
+
+// Update attribute of given file.
+static int update_attr(struct dirtree *root)
+{
+  unsigned long fval = 0;
+  char *fpath = NULL;
+  int fd;
+
+  if (!dirtree_notdotdot(root)) return 0;
+
+  /*
+   * if file is a link and recursive is set or file is not regular+link+dir
+   * (like fifo or dev file) then escape the file.
+   */
+  if ((S_ISLNK(root->st.st_mode) && chattr.recursive)
+    || (!S_ISREG(root->st.st_mode) && !S_ISLNK(root->st.st_mode)
+      && !S_ISDIR(root->st.st_mode)))
+    return 0;
+
+  fpath = dirtree_path(root, NULL);
+  if (-1 == (fd=open(fpath, O_RDONLY | O_NONBLOCK))) {
+    free(fpath);
+    return DIRTREE_ABORT;
+  }
+  // Get current attr of file.
+  if (ext2_getflag(fd, &(root->st), &fval) < 0) {
+    perror_msg("read flags of '%s'", fpath);
+    free(fpath);
+    xclose(fd);
+    return DIRTREE_ABORT;
+  }
+  if (chattr.set) { // for '=' operator.
+    if (ext2_setflag(fd, &(root->st), chattr.set) < 0)
+      perror_msg("setting flags '%s'", fpath);
+  } else { // for '-' / '+' operator.
+    fval &= ~(chattr.rm);
+    fval |= chattr.add;
+    if (!S_ISDIR(root->st.st_mode)) fval &= ~FS_DIRSYNC_FL;
+    if (ext2_setflag(fd, &(root->st), fval) < 0)
+      perror_msg("setting flags '%s'", fpath);
+  }
+  if (chattr.vflag) { // set file version
+    if (ioctl(fd, FS_IOC_SETVERSION, (void*)&chattr.version) < 0)
+      perror_msg("while setting version on '%s'", fpath);
+  }
+  free(fpath);
+  xclose(fd);
+
+  if (S_ISDIR(root->st.st_mode) && chattr.recursive) return DIRTREE_RECURSE;
+  return 0;
+}
+
+void chattr_main(void)
+{
+  char **argv = toys.optargs;
+
+  memset(&chattr, 0, sizeof(struct _chattr));
+  parse_cmdline_arg(&argv);
+  if (!*argv) chattr_help();
+  if (chattr.set && (chattr.add || chattr.rm))
+    error_exit("'=' is incompatible with '-' and '+'");
+  if (chattr.rm & chattr.add) error_exit("Can't set and unset same flag.");
+  if (!(chattr.add || chattr.rm || chattr.set || chattr.vflag))
+    error_exit(("Must use '-v', '=', '-' or '+'"));
+  for (; *argv; argv++) dirtree_read(*argv, update_attr);
+  toys.exitval = 0; //always set success at this point.
+}