]> git.draconx.ca Git - liblbx.git/blobdiff - src/lbxtool.c
Add the tools.h that should have been included a while ago.
[liblbx.git] / src / lbxtool.c
index f9a17d10e9371e7e7aadbf6f011187d6731102de..1d80eeb0bcc5ab2c075b2c20ed5c0e33ce49745e 100644 (file)
@@ -1,13 +1,45 @@
+/* 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 ": %m.\n", progname, __VA_ARGS__)\
+       fprintf(stderr, "%s: " fmt, progname, __VA_ARGS__)\
 )
 
 enum {
@@ -16,23 +48,148 @@ enum {
        MODE_EXTRACT,
 };
 
+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: %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)
 {
-       int mode = MODE_NONE;
+       int mode = MODE_NONE, verbose = 0;
+       const char *name = "stdin";
        FILE *f  = stdin;
        int opt;
 
-       struct option longopts[] = {
+       static const char         *sopts   = "lxf:i:vV";
+       static const struct option lopts[] = {
                { "list",    0, NULL, 'l' },
                { "extract", 0, NULL, 'x' },
 
                { "file",    1, NULL, 'f' },
+               { "index",   1, NULL, 'i' },
+
+               { "verbose", 0, NULL, 'v' },
+
+               { "version", 0, NULL, 'V' },
+               { "usage",   0, NULL, 'U' },
+               { "help",    0, NULL, 'H' },
 
                { 0 }
        };
 
-       progname = argv[0];
-       while ((opt = getopt_long(argc, argv, "lxf:", longopts, NULL)) != -1) {
+       progname = "lbxtool"; /* argv[0]; */
+       while ((opt = getopt_long(argc, argv, sopts, lopts, NULL)) != -1) {
                switch(opt) {
                case 'l':
                        mode = MODE_LIST;
@@ -41,19 +198,45 @@ int main(int argc, char **argv)
                        mode = MODE_EXTRACT;
                        break;
                case 'f':
+                       if (strcmp(optarg, "-") == 0)
+                               break;
+
+                       name = strrchr(optarg, '/');
+                       name = name ? name+1 : optarg;
+
                        f = fopen(optarg, "rb");
                        if (!f) {
-                               errmsg("failed to open file", 0, 4);
+                               errmsg("failed to open file %s: %m\n", optarg);
                                return EXIT_FAILURE;
                        }
                        break;
+               case 'i':
+                       /* FIXME: Add index file support. */
+                       break;
+               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;
                }
        }
 
-       if (mode == MODE_NONE) {
-               fprintf(stderr, "%s: you must specify a mode.\n", progname);
-               return EXIT_FAILURE;
+       switch (mode) {
+       case MODE_LIST:
+               return list(f, name, verbose, &argv[optind]);
+       case MODE_EXTRACT:
+               return extract(f, name, verbose, &argv[optind]);
        }
+
+       fprintf(stderr, "%s: you must specify a mode.\n", progname);
+       return EXIT_FAILURE;
 }