X-Git-Url: https://git.draconx.ca/gitweb/liblbx.git/blobdiff_plain/c907294abea66bfa9739fcd6a08415f4ea1142f8..8325f5adeef29ffc7032c832d525a0822e19d74c:/src/lbxtool.c

diff --git a/src/lbxtool.c b/src/lbxtool.c
index 96c8ddd..687e419 100644
--- a/src/lbxtool.c
+++ b/src/lbxtool.c
@@ -3,6 +3,7 @@
 #include <stdlib.h>
 #include <string.h>
 #include <getopt.h>
+#include <fnmatch.h>
 
 #include "lbx.h"
 
@@ -17,18 +18,120 @@ enum {
 	MODE_EXTRACT,
 };
 
-int list(FILE *f, const char *name, int verbose) {
-	LBX *lbx = lbx_fopen(f, name);
+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;
+
+	lbx = lbx_fopen(f, name);
 	if (!lbx) {
 		errmsg("failed to open archive: %s.\n", lbx_strerror());
 		return EXIT_FAILURE;
 	}
 
+	nfiles = lbx_numfiles(lbx);
 	if (verbose) {
-		printf("Files in archive: %zd\n", lbx_numfiles(lbx));
+		printf("Files in archive: %zu\n", nfiles);
+	}
+
+	for (i = 0; i < nfiles; i++) {
+		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);
+		}
+
+		putchar('\n');
 	}
 
 	lbx_close(lbx);
+	return EXIT_SUCCESS;
+err:
+	lbx_close(lbx);
+	return EXIT_FAILURE;
+}
+
+int extract(FILE *f, const char *name, int verbose, char **argv) {
+	LBX *lbx;
+	size_t nfiles;
+	unsigned int i;
+
+	lbx = lbx_fopen(f, name);
+	if (!lbx) {
+		errmsg("failed to open archive: %s.\n", lbx_strerror());
+		return EXIT_FAILURE;
+	}
+
+	nfiles = lbx_numfiles(lbx);
+	if (verbose) {
+		printf("Files in archive: %zu\n", nfiles);
+	}
+
+	for (i = 0; i < nfiles; i++) {
+		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);
+			goto err;
+		}
+
+		if (verbose) printf("extracting %s...\n", stat.name);
+		rc = lbx_extract(lbx, i, of);
+		if (rc < stat.size) {
+			if (rc == 0) remove(stat.name);
+			errmsg("error extracting %s: %s.\n",
+			       stat.name, lbx_strerror());
+			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)
@@ -86,9 +189,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 EXIT_SUCCESS;
+		return extract(f, name, verbose, &argv[optind]);
 	}
 
 	fprintf(stderr, "%s: you must specify a mode.\n", progname);