From 2aa2f13131663afa0d13259365f824b05e9a2f02 Mon Sep 17 00:00:00 2001 From: David Oberhollenzer Date: Thu, 2 May 2019 16:30:33 +0200 Subject: unsquashfs: add argument processing, file listing parameter Signed-off-by: David Oberhollenzer --- unpack/Makemodule.am | 1 + unpack/list_files.c | 90 ++++++++++++++++++++++++++++++++++ unpack/unsquashfs.c | 135 ++++++++++++++++++++++++++++++++++++++++++++++----- unpack/unsquashfs.h | 6 +++ 4 files changed, 219 insertions(+), 13 deletions(-) create mode 100644 unpack/list_files.c (limited to 'unpack') diff --git a/unpack/Makemodule.am b/unpack/Makemodule.am index 2a08295..a199375 100644 --- a/unpack/Makemodule.am +++ b/unpack/Makemodule.am @@ -1,5 +1,6 @@ unsquashfs_SOURCES = unpack/unsquashfs.c unpack/tree_node_from_inode.c unsquashfs_SOURCES += unpack/unsquashfs.h unpack/read_fstree.c +unsquashfs_SOURCES += unpack/list_files.c unsquashfs_LDADD = libsquashfs.a libfstree.a libcompress.a libutil.a if WITH_LZMA diff --git a/unpack/list_files.c b/unpack/list_files.c new file mode 100644 index 0000000..aaf0652 --- /dev/null +++ b/unpack/list_files.c @@ -0,0 +1,90 @@ +/* SPDX-License-Identifier: GPL-3.0-or-later */ +#include "unsquashfs.h" + +static void mode_to_str(uint16_t mode, char *p) +{ + switch (mode & S_IFMT) { + case S_IFDIR: *(p++) = 'd'; break; + case S_IFCHR: *(p++) = 'c'; break; + case S_IFBLK: *(p++) = 'b'; break; + case S_IFREG: *(p++) = '-'; break; + case S_IFLNK: *(p++) = 'l'; break; + case S_IFSOCK: *(p++) = 's'; break; + case S_IFIFO: *(p++) = 'p'; break; + default: *(p++) = '?'; break; + } + + *(p++) = (mode & S_IRUSR) ? 'r' : '-'; + *(p++) = (mode & S_IWUSR) ? 'w' : '-'; + + switch (mode & (S_IXUSR | S_ISUID)) { + case S_IXUSR | S_ISUID: *(p++) = 's'; break; + case S_IXUSR: *(p++) = 'x'; break; + case S_ISUID: *(p++) = 'S'; break; + default: *(p++) = '-'; break; + } + + *(p++) = (mode & S_IRGRP) ? 'r' : '-'; + *(p++) = (mode & S_IWGRP) ? 'w' : '-'; + + switch (mode & (S_IXGRP | S_ISGID)) { + case S_IXGRP | S_ISGID: *(p++) = 's'; break; + case S_IXGRP: *(p++) = 'x'; break; + case S_ISGID: *(p++) = 'S'; break; + case 0: *(p++) = '-'; break; + } + + *(p++) = (mode & S_IROTH) ? 'r' : '-'; + *(p++) = (mode & S_IWOTH) ? 'w' : '-'; + + switch (mode & (S_IXOTH | S_ISVTX)) { + case S_IXOTH | S_ISVTX: *(p++) = 't'; break; + case S_IXOTH: *(p++) = 'x'; break; + case S_ISVTX: *(p++) = 'T'; break; + case 0: *(p++) = '-'; break; + } + + *p = '\0'; +} + +static int count_int_chars(unsigned int i) +{ + int count = 1; + + while (i > 10) { + ++count; + i /= 10; + } + + return count; +} + +void list_files(tree_node_t *node) +{ + int i, max_uid_chars = 0, max_gid_chars = 0; + char modestr[12]; + tree_node_t *n; + + if (S_ISDIR(node->mode)) { + for (n = node->data.dir->children; n != NULL; n = n->next) { + i = count_int_chars(n->uid); + max_uid_chars = i > max_uid_chars ? i : max_uid_chars; + + i = count_int_chars(n->gid); + max_gid_chars = i > max_gid_chars ? i : max_gid_chars; + } + + for (n = node->data.dir->children; n != NULL; n = n->next) { + mode_to_str(n->mode, modestr); + + printf("%s %*u/%-*u %s\n", modestr, + max_uid_chars, n->uid, + max_gid_chars, n->gid, n->name); + } + } else { + mode_to_str(node->mode, modestr); + + printf("%s %u/%u %s\n", modestr, + node->uid, node->gid, node->name); + } +} diff --git a/unpack/unsquashfs.c b/unpack/unsquashfs.c index d332dee..b253bae 100644 --- a/unpack/unsquashfs.c +++ b/unpack/unsquashfs.c @@ -1,28 +1,119 @@ /* SPDX-License-Identifier: GPL-3.0-or-later */ #include "unsquashfs.h" +static struct option long_opts[] = { + { "list", required_argument, NULL, 'l' }, + { "help", no_argument, NULL, 'h' }, + { "version", no_argument, NULL, 'V' }, +}; + +static const char *short_opts = "l:hV"; + +static const char *help_string = +"Usage: %s [OPTIONS] \n" +"\n" +"View or extract the contents of a squashfs image.\n" +"\n" +"Possible options:\n" +" --list, -l Produce a directory listing for a given path in the\n" +" squashfs image.\n" +" --help, -h Print help text and exit.\n" +" --version, -V Print version information and exit.\n" +"\n"; + extern const char *__progname; +static tree_node_t *find_node(tree_node_t *n, const char *path) +{ + const char *end; + size_t len; + + while (path != NULL && *path != '\0') { + if (!S_ISDIR(n->mode)) { + errno = ENOTDIR; + return NULL; + } + + end = strchrnul(path, '/'); + len = end - path; + + for (n = n->data.dir->children; n != NULL; n = n->next) { + if (strncmp(path, n->name, len) != 0) + continue; + if (n->name[len] != '\0') + continue; + break; + } + + if (n == NULL) { + errno = ENOENT; + return NULL; + } + + path = *end ? (end + 1) : end; + } + + return n; +} + int main(int argc, char **argv) { - int fd, status = EXIT_FAILURE; + int i, fd, status = EXIT_FAILURE; + char *lspath = NULL; sqfs_super_t super; compressor_t *cmp; fstree_t fs; - if (argc != 2) { - fprintf(stderr, "Usage: %s \n", __progname); - return EXIT_FAILURE; + for (;;) { + i = getopt_long(argc, argv, short_opts, long_opts, NULL); + if (i == -1) + break; + + switch (i) { + case 'l': + if (optarg != NULL) { + free(lspath); + + lspath = strdup(optarg ? optarg : ""); + if (lspath == NULL) { + perror("processing arguments"); + return EXIT_FAILURE; + } + + if (canonicalize_name(lspath)) { + fprintf(stderr, "Invalid pth: %s\n", + optarg); + goto out_cmd; + } + } + break; + case 'h': + printf(help_string, __progname); + status = EXIT_SUCCESS; + goto out_cmd; + case 'V': + print_version(); + status = EXIT_SUCCESS; + goto out_cmd; + default: + goto fail_arg; + } } - fd = open(argv[1], O_RDONLY); + if (optind >= argc) { + fprintf(stderr, "Usage: %s [OPTIONS] \n", + __progname); + goto out_cmd; + } + + fd = open(argv[optind], O_RDONLY); if (fd < 0) { - perror(argv[1]); - return EXIT_FAILURE; + perror(argv[optind]); + goto out_cmd; } if (sqfs_super_read(&super, fd)) - goto out; + goto out_fd; if ((super.version_major != SQFS_VERSION_MAJOR) || (super.version_minor != SQFS_VERSION_MINOR)) { @@ -31,34 +122,52 @@ int main(int argc, char **argv) "We currently only supports version %d.%d (sorry).\n", super.version_major, super.version_minor, SQFS_VERSION_MAJOR, SQFS_VERSION_MINOR); - goto out; + goto out_fd; } if (super.flags & SQFS_FLAG_COMPRESSOR_OPTIONS) { fputs("Image has been built with compressor options.\n" "This is not yet supported.\n", stderr); - goto out; + goto out_fd; } if (!compressor_exists(super.compression_id)) { fputs("Image uses a compressor that has not been built in\n", stderr); - goto out; + goto out_fd; } cmp = compressor_create(super.compression_id, false, super.block_size); if (cmp == NULL) - goto out; + goto out_fd; if (read_fstree(&fs, fd, &super, cmp)) goto out_cmp; + if (lspath != NULL) { + tree_node_t *n = find_node(fs.root, lspath); + + if (n == NULL) { + perror(lspath); + goto out_fs; + } + + list_files(n); + } + status = EXIT_SUCCESS; +out_fs: fstree_cleanup(&fs); out_cmp: cmp->destroy(cmp); -out: +out_fd: close(fd); +out_cmd: + free(lspath); return status; +fail_arg: + fprintf(stderr, "Try `%s --help' for more information.\n", __progname); + free(lspath); + return EXIT_FAILURE; } diff --git a/unpack/unsquashfs.h b/unpack/unsquashfs.h index 5553012..52b3889 100644 --- a/unpack/unsquashfs.h +++ b/unpack/unsquashfs.h @@ -7,10 +7,14 @@ #include "compress.h" #include "id_table.h" #include "fstree.h" +#include "config.h" +#include "util.h" #include #include #include +#include +#include #include #include @@ -21,4 +25,6 @@ tree_node_t *tree_node_from_inode(sqfs_inode_generic_t *inode, int read_fstree(fstree_t *out, int fd, sqfs_super_t *super, compressor_t *cmp); +void list_files(tree_node_t *node); + #endif /* UNSQUASHFS_H */ -- cgit v1.2.3