diff options
Diffstat (limited to 'toys')
-rw-r--r-- | toys/other/lsattr.c | 320 |
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. +} |