Add a basic regression test tool.

[liblbx.git] / src / lbxtool.c

diff --git a/src/lbxtool.c b/src/lbxtool.c
index 96c8ddd06e8ef87c7a016f2923e4cfac9967fa98..687e419bdf297e4f51b333aa98ac5bc194a0bb0f 100644 (file)
--- 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);