summaryrefslogtreecommitdiff
path: root/unpack/rdsquashfs.c
diff options
context:
space:
mode:
Diffstat (limited to 'unpack/rdsquashfs.c')
-rw-r--r--unpack/rdsquashfs.c197
1 files changed, 18 insertions, 179 deletions
diff --git a/unpack/rdsquashfs.c b/unpack/rdsquashfs.c
index a24d6c3..27a0cf5 100644
--- a/unpack/rdsquashfs.c
+++ b/unpack/rdsquashfs.c
@@ -1,72 +1,6 @@
/* SPDX-License-Identifier: GPL-3.0-or-later */
#include "rdsquashfs.h"
-enum {
- OP_NONE = 0,
- OP_LS,
- OP_CAT,
- OP_UNPACK,
- OP_DESCRIBE,
-};
-
-static struct option long_opts[] = {
- { "list", required_argument, NULL, 'l' },
- { "cat", required_argument, NULL, 'c' },
- { "unpack-root", required_argument, NULL, 'p' },
- { "unpack-path", required_argument, NULL, 'u' },
- { "no-dev", no_argument, NULL, 'D' },
- { "no-sock", no_argument, NULL, 'S' },
- { "no-fifo", no_argument, NULL, 'F' },
- { "no-slink", no_argument, NULL, 'L' },
- { "no-empty-dir", no_argument, NULL, 'E' },
- { "describe", no_argument, NULL, 'd' },
- { "chmod", no_argument, NULL, 'C' },
- { "chown", no_argument, NULL, 'O' },
- { "quiet", no_argument, NULL, 'q' },
- { "help", no_argument, NULL, 'h' },
- { "version", no_argument, NULL, 'V' },
-};
-
-static const char *short_opts = "l:c:u:p:DSFLCOEdqhV";
-
-static const char *help_string =
-"Usage: %s [OPTIONS] <squashfs-file>\n"
-"\n"
-"View or extract the contents of a squashfs image.\n"
-"\n"
-"Possible options:\n"
-"\n"
-" --list, -l <path> Produce a directory listing for a given path in\n"
-" the squashfs image.\n"
-" --cat, -c <path> If the specified path is a regular file in the,\n"
-" image, dump its contents to stdout.\n"
-" --unpack-path, -u <path> Unpack this sub directory from the image. To\n"
-" unpack everything, simply specify /.\n"
-" --describe, -d Produce a file listing from the image.\n"
-"\n"
-" --unpack-root, -p <path> If used with --unpack-path, this is where the\n"
-" data unpacked to. If used with --describe, this\n"
-" is used as a prefix for the input path of\n"
-" regular files.\n"
-"\n"
-" --no-dev, -D Do not unpack device special files.\n"
-" --no-sock, -S Do not unpack socket files.\n"
-" --no-fifo, -F Do not unpack named pipes.\n"
-" --no-slink, -L Do not unpack symbolic links.\n"
-" --no-empty-dir, -E Do not unpack directories that would end up\n"
-" empty after applying the above rules.\n"
-" --chmod, -C Change permission flags of unpacked files to\n"
-" those store in the squashfs image.\n"
-" --chown, -O Change ownership of unpacked files to the\n"
-" UID/GID set in the squashfs image.\n"
-" --quiet, -q Do not print out progress while unpacking.\n"
-"\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;
@@ -100,114 +34,22 @@ static tree_node_t *find_node(tree_node_t *n, const char *path)
return n;
}
-static char *get_path(char *old, const char *arg)
-{
- char *path;
-
- free(old);
-
- path = strdup(arg);
- if (path == NULL) {
- perror("processing arguments");
- exit(EXIT_FAILURE);
- }
-
- if (canonicalize_name(path)) {
- fprintf(stderr, "Invalid path: %s\n", arg);
- free(path);
- exit(EXIT_FAILURE);
- }
-
- return path;
-}
-
int main(int argc, char **argv)
{
- int i, status = EXIT_FAILURE, op = OP_NONE;
- const char *unpack_root = NULL;
- int rdtree_flags = 0, flags = 0;
+ int status = EXIT_FAILURE;
data_reader_t *data = NULL;
- char *cmdpath = NULL;
sqfs_super_t super;
compressor_t *cmp;
tree_node_t *n;
+ options_t opt;
fstree_t fs;
int sqfsfd;
- for (;;) {
- i = getopt_long(argc, argv, short_opts, long_opts, NULL);
- if (i == -1)
- break;
-
- switch (i) {
- case 'D':
- rdtree_flags |= RDTREE_NO_DEVICES;
- break;
- case 'S':
- rdtree_flags |= RDTREE_NO_SOCKETS;
- break;
- case 'F':
- rdtree_flags |= RDTREE_NO_FIFO;
- break;
- case 'L':
- rdtree_flags |= RDTREE_NO_SLINKS;
- break;
- case 'C':
- flags |= UNPACK_CHMOD;
- break;
- case 'O':
- flags |= UNPACK_CHOWN;
- break;
- case 'E':
- rdtree_flags |= RDTREE_NO_EMPTY;
- break;
- case 'c':
- op = OP_CAT;
- cmdpath = get_path(cmdpath, optarg);
- break;
- case 'd':
- op = OP_DESCRIBE;
- break;
- case 'l':
- op = OP_LS;
- cmdpath = get_path(cmdpath, optarg);
- break;
- case 'p':
- unpack_root = optarg;
- break;
- case 'u':
- op = OP_UNPACK;
- cmdpath = get_path(cmdpath, optarg);
- break;
- case 'q':
- flags |= UNPACK_QUIET;
- 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;
- }
- }
-
- if (op == OP_NONE) {
- fputs("No opteration specified\n", stderr);
- goto fail_arg;
- }
-
- if (optind >= argc) {
- fputs("Missing image argument\n", stderr);
- goto fail_arg;
- }
+ process_command_line(&opt, argc, argv);
- sqfsfd = open(argv[optind], O_RDONLY);
+ sqfsfd = open(opt.image_name, O_RDONLY);
if (sqfsfd < 0) {
- perror(argv[optind]);
+ perror(opt.image_name);
goto out_cmd;
}
@@ -240,28 +82,29 @@ int main(int argc, char **argv)
goto out_cmp;
}
- if (deserialize_fstree(&fs, &super, cmp, sqfsfd, rdtree_flags))
+ if (deserialize_fstree(&fs, &super, cmp, sqfsfd, opt.rdtree_flags))
goto out_cmp;
- switch (op) {
+ switch (opt.op) {
case OP_LS:
- n = find_node(fs.root, cmdpath);
+ n = find_node(fs.root, opt.cmdpath);
if (n == NULL) {
- perror(cmdpath);
+ perror(opt.cmdpath);
goto out_fs;
}
list_files(n);
break;
case OP_CAT:
- n = find_node(fs.root, cmdpath);
+ n = find_node(fs.root, opt.cmdpath);
if (n == NULL) {
- perror(cmdpath);
+ perror(opt.cmdpath);
goto out_fs;
}
if (!S_ISREG(n->mode)) {
- fprintf(stderr, "/%s: not a regular file\n", cmdpath);
+ fprintf(stderr, "/%s: not a regular file\n",
+ opt.cmdpath);
goto out_fs;
}
@@ -273,9 +116,9 @@ int main(int argc, char **argv)
goto out_fs;
break;
case OP_UNPACK:
- n = find_node(fs.root, cmdpath);
+ n = find_node(fs.root, opt.cmdpath);
if (n == NULL) {
- perror(cmdpath);
+ perror(opt.cmdpath);
goto out_fs;
}
@@ -283,11 +126,11 @@ int main(int argc, char **argv)
if (data == NULL)
goto out_fs;
- if (restore_fstree(unpack_root, n, data, flags))
+ if (restore_fstree(opt.unpack_root, n, data, opt.flags))
goto out_fs;
break;
case OP_DESCRIBE:
- describe_tree(fs.root, unpack_root);
+ describe_tree(fs.root, opt.unpack_root);
break;
}
@@ -301,10 +144,6 @@ out_cmp:
out_fd:
close(sqfsfd);
out_cmd:
- free(cmdpath);
+ free(opt.cmdpath);
return status;
-fail_arg:
- fprintf(stderr, "Try `%s --help' for more information.\n", __progname);
- free(cmdpath);
- return EXIT_FAILURE;
}