cleanup: rdsquashfs: move command line option processing to options.c

Signed-off-by: David Oberhollenzer <david.oberhollenzer@sigma-star.at>

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;
 }