aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAshwini Sharma <ak.ashwini1981@gmail.com>2014-05-02 06:17:48 -0500
committerAshwini Sharma <ak.ashwini1981@gmail.com>2014-05-02 06:17:48 -0500
commita547cf11686a878d2fd1a42a05719b78903009ad (patch)
tree036226786f17a32c4446a60c1d84eb2ed8e10fce
parentc54fdc92310566ccff7a96ad415b9959f5087c93 (diff)
downloadtoybox-a547cf11686a878d2fd1a42a05719b78903009ad.tar.gz
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.
-rw-r--r--toys/other/lsattr.c320
1 files changed, 320 insertions, 0 deletions
diff --git a/toys/other/lsattr.c b/toys/other/lsattr.c
new file mode 100644
index 00000000..937ea608
--- /dev/null
+++ b/toys/other/lsattr.c
@@ -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.
+}