X-Git-Url: https://git.draconx.ca/gitweb/liblbx.git/blobdiff_plain/b144af9a1a50583d9177ea4d2b0f94eee4178f84..91743040c1a98baaffc82ffff8306c2c9fd07285:/src/lbxtool.c diff --git a/src/lbxtool.c b/src/lbxtool.c index 5435d5f..13a9ba6 100644 --- a/src/lbxtool.c +++ b/src/lbxtool.c @@ -1,11 +1,43 @@ -#define _GNU_SOURCE +/* + * 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 3 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, see . + */ +#include #include #include #include #include +#include +#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,17 +49,29 @@ enum { MODE_EXTRACT, }; -int list(FILE *f, const char *name, int verbose) { - LBX *lbx; - size_t nfiles; - unsigned int i; +int filematch(char **argv, const char *name) +{ + int i; - lbx = lbx_fopen(f, name); - if (!lbx) { - errmsg("failed to open archive: %s.\n", lbx_strerror()); - return EXIT_FAILURE; + 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(LBX *lbx, const char *name, int verbose, char **argv) { + size_t nfiles; + unsigned int i; + nfiles = lbx_numfiles(lbx); if (verbose) { printf("Files in archive: %zu\n", nfiles); @@ -37,6 +81,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: return EXIT_FAILURE; + } + printf("%s", stat.name); if (verbose) { printf(" size=%zu bytes", stat.size); @@ -45,21 +96,13 @@ int list(FILE *f, const char *name, int verbose) { putchar('\n'); } - lbx_close(lbx); return EXIT_SUCCESS; } -int extract(FILE *f, const char *name, int verbose) { - LBX *lbx; +int extract(LBX *lbx, const char *name, int verbose, char **argv) { 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); @@ -69,13 +112,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: return EXIT_FAILURE; + } + of = fopen(stat.name, "wbx"); if (!of) { errmsg("failed to create output file %s: %m.\n", stat.name); - break; + return EXIT_FAILURE; } if (verbose) printf("extracting %s...\n", stat.name); @@ -84,23 +135,22 @@ 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; + return EXIT_FAILURE; } if (verbose) printf("wrote %zu bytes.\n", rc); } - lbx_close(lbx); return EXIT_SUCCESS; } int main(int argc, char **argv) { - int mode = MODE_NONE, verbose = 0; + int mode = MODE_NONE, verbose = 0, opt, rc = EXIT_FAILURE; + struct lbx_pipe_state state = { .f = stdin }; const char *name = "stdin"; - FILE *f = stdin; - int opt; + LBX *lbx; - 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 +160,10 @@ int main(int argc, char **argv) { "verbose", 0, NULL, 'v' }, + { "version", 0, NULL, 'V' }, + { "usage", 0, NULL, 'U' }, + { "help", 0, NULL, 'H' }, + { 0 } }; @@ -129,8 +183,7 @@ int main(int argc, char **argv) name = strrchr(optarg, '/'); name = name ? name+1 : optarg; - f = fopen(optarg, "rb"); - if (!f) { + if (!freopen(optarg, "rb", state.f)) { errmsg("failed to open file %s: %m\n", optarg); return EXIT_FAILURE; } @@ -141,18 +194,41 @@ 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; } } + if (fseek(state.f, 0, SEEK_CUR) == 0) + lbx = lbx_open(state.f, &lbx_default_fops, NULL, name); + else + lbx = lbx_open(&state, &lbx_pipe_fops, NULL, name); + + if (!lbx) { + errmsg("failed to open archive: %s.\n", lbx_strerror()); + return EXIT_FAILURE; + } + switch (mode) { case MODE_LIST: - return list(f, name, verbose); + rc = list(lbx, name, verbose, &argv[optind]); + break; case MODE_EXTRACT: - return extract(f, name, verbose); + rc = extract(lbx, name, verbose, &argv[optind]); + break; + default: + fprintf(stderr, "%s: you must specify a mode.\n", progname); } - fprintf(stderr, "%s: you must specify a mode.\n", progname); - return EXIT_FAILURE; + lbx_close(lbx); + return rc; }