X-Git-Url: http://git.draconx.ca/gitweb/liblbx.git/blobdiff_plain/b144af9a1a50583d9177ea4d2b0f94eee4178f84..e8921298baed414f18f74eaed34917758231c1a0:/src/lbxtool.c

diff --git a/src/lbxtool.c b/src/lbxtool.c
index 5435d5f..1d80eeb 100644
--- a/src/lbxtool.c
+++ b/src/lbxtool.c
@@ -1,11 +1,42 @@
+/* 2ooM: The Master of Orion II Reverse Engineering Project
+ * Simple command-line tool to extract LBX archive files.
+ * Copyright (C) 2006-2008 Nick Bowler
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
 #define _GNU_SOURCE
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <getopt.h>
+#include <fnmatch.h>
 
+#include "tools.h"
 #include "lbx.h"
 
+static void printusage(void)
+{
+	puts("usage: lbxtool [-l|-x] [-v] [-f path] [file ...]");
+}
+
+static void printhelp(void)
+{
+	printusage();
+	puts("For now, see the man page for detailed help.");
+}
+
 static const char *progname;
 #define errmsg(fmt, ...) (\
 	fprintf(stderr, "%s: " fmt, progname, __VA_ARGS__)\
@@ -17,7 +48,26 @@ enum {
 	MODE_EXTRACT,
 };
 
-int list(FILE *f, const char *name, int verbose) {
+int filematch(char **argv, const char *name)
+{
+	int rc, i;
+
+	for (i = 0; argv[i]; i++) {
+		switch(fnmatch(argv[i], name, 0)) {
+		case 0:
+			return 0;
+		case FNM_NOMATCH:
+			break;
+		default:
+			errmsg("error matching glob: %s.\n", argv[i]);
+			return 1;
+		}
+	}
+
+	return i ? -1: 0;
+}
+
+int list(FILE *f, const char *name, int verbose, char **argv) {
 	LBX *lbx;
 	size_t nfiles;
 	unsigned int i;
@@ -37,6 +87,13 @@ int list(FILE *f, const char *name, int verbose) {
 		struct lbx_statbuf stat;
 
 		lbx_stat(lbx, i, &stat);
+
+		switch (filematch(argv, stat.name)) {
+		case -1: continue;
+		case  0: break;
+		default: goto err;
+		}
+
 		printf("%s", stat.name);
 		if (verbose) {
 			printf(" size=%zu bytes", stat.size);
@@ -47,9 +104,12 @@ int list(FILE *f, const char *name, int verbose) {
 
 	lbx_close(lbx);
 	return EXIT_SUCCESS;
+err:
+	lbx_close(lbx);
+	return EXIT_FAILURE;
 }
 
-int extract(FILE *f, const char *name, int verbose) {
+int extract(FILE *f, const char *name, int verbose, char **argv) {
 	LBX *lbx;
 	size_t nfiles;
 	unsigned int i;
@@ -69,13 +129,21 @@ int extract(FILE *f, const char *name, int verbose) {
 		struct lbx_statbuf stat;
 		size_t rc;
 		FILE *of;
+		int j;
 
 		lbx_stat(lbx, i, &stat);
+
+		switch (filematch(argv, stat.name)) {
+		case -1: continue;
+		case  0: break;
+		default: goto err;
+		}
+
 		of = fopen(stat.name, "wbx");
 		if (!of) {
 			errmsg("failed to create output file %s: %m.\n",
 			                                     stat.name);
-			break;
+			goto err;
 		}
 
 		if (verbose) printf("extracting %s...\n", stat.name);
@@ -84,13 +152,16 @@ int extract(FILE *f, const char *name, int verbose) {
 			if (rc == 0) remove(stat.name);
 			errmsg("error extracting %s: %s.\n",
 			       stat.name, lbx_strerror());
-			break;
+			goto err;
 		}
 		if (verbose) printf("wrote %zu bytes.\n", rc);
 	}
 
 	lbx_close(lbx);
 	return EXIT_SUCCESS;
+err:
+	lbx_close(lbx);
+	return EXIT_FAILURE;
 }
 
 int main(int argc, char **argv)
@@ -100,7 +171,7 @@ int main(int argc, char **argv)
 	FILE *f  = stdin;
 	int opt;
 
-	static const char         *sopts   = "lxf:i:v";
+	static const char         *sopts   = "lxf:i:vV";
 	static const struct option lopts[] = {
 		{ "list",    0, NULL, 'l' },
 		{ "extract", 0, NULL, 'x' },
@@ -110,6 +181,10 @@ int main(int argc, char **argv)
 
 		{ "verbose", 0, NULL, 'v' },
 
+		{ "version", 0, NULL, 'V' },
+		{ "usage",   0, NULL, 'U' },
+		{ "help",    0, NULL, 'H' },
+
 		{ 0 }
 	};
 
@@ -141,6 +216,15 @@ int main(int argc, char **argv)
 		case 'v':
 			verbose = 1;
 			break;
+		case 'V':
+			puts(VERSION_BOILERPLATE("lbxtool"));
+			return EXIT_SUCCESS;
+		case 'U':
+			printusage();
+			return EXIT_SUCCESS;
+		case 'H':
+			printhelp();
+			return EXIT_SUCCESS;
 		default:
 			return EXIT_FAILURE;
 		}
@@ -148,9 +232,9 @@ int main(int argc, char **argv)
 
 	switch (mode) {
 	case MODE_LIST:
-		return list(f, name, verbose);
+		return list(f, name, verbose, &argv[optind]);
 	case MODE_EXTRACT:
-		return extract(f, name, verbose);
+		return extract(f, name, verbose, &argv[optind]);
 	}
 
 	fprintf(stderr, "%s: you must specify a mode.\n", progname);