]> git.draconx.ca Git - upkg.git/blobdiff - src/upkg.c
Stop using gnulib's flexmember module.
[upkg.git] / src / upkg.c
index 989bf798f322e10d5d546d8f1e99d63ee3385868..db9a747d39eb1ccedaaa6ed49d61f45b5a09c418 100644 (file)
@@ -1,19 +1,19 @@
 /*
- *  upkg: tool for manipulating Unreal Tournament packages.
- *  Copyright (C) 2009 Nick Bowler
+ * upkg: tool for manipulating Unreal Tournament packages.
+ * Copyright © 2009-2012, 2022 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 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.
+ * 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 <http://www.gnu.org/licenses/>.
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
  */
 
 #include <config.h>
@@ -21,6 +21,7 @@
 #include <stdlib.h>
 #include <string.h>
 #include <errno.h>
+#include <assert.h>
 #include <getopt.h>
 
 #include <glib-object.h>
 #include <uobject/package.h>
 #include <uobject/vfs.h>
 
+#include "help.h"
+
 enum {
        MODE_INFO,
+       MODE_LIST,
        MODE_EXPORT,
        MODE_MAX
 };
@@ -42,49 +46,49 @@ enum {
 int verbose = 0;
 
 static const char *progname = "upkg";
-static const char *sopts = "ixvf:VH";
-static const struct option lopts[] = {
-       { "info",     0, NULL, 'i' },
-       { "export",   0, NULL, 'x' },
-       { "file",     1, NULL, 'f' },
-       { "verbose",  0, NULL, 'v' },
-       { "version",  0, NULL, 'V' },
-       { "help",     0, NULL, 'H' },
-       { 0 }
-};
 
-void print_version(void)
+#include "upkgopts.h"
+static const char sopts[] = SOPT_STRING;
+static const struct option lopts[] = { LOPTS_INITIALIZER, {0} };
+
+static void print_version(void)
 {
        printf("%s\n", PACKAGE_STRING);
-       puts("\
-Copyright (C) 2009 Nick Bowler.\n\
-License GPLv3+: GNU GPL version 3 or later <http://gnu.org/licenses/gpl.html>.\n\
-This is free software: you are free to change and redistribute it.\n\
-There is NO WARRANTY, to the extent permitted by law."
-       );
+       puts("Copyright (C) 2022 Nick Bowler.");
+       puts("License GPLv3+: GNU GPL version 3 or any later version.");
+       puts("This is free software: you are free to change and redistribute it.");
+       puts("There is NO WARRANTY, to the extent permitted by law.");
 }
 
-void print_usage(FILE *f)
+static void print_usage(FILE *f)
 {
        fprintf(f, "Usage: %s [options] [object ...]\n", progname);
+       if (f != stdout)
+               fprintf(f, "Try %s --help for more information.\n", progname);
 }
 
-void print_help(void)
+static void print_help(void)
 {
+       const struct option *opt;
+
        print_usage(stdout);
-       puts("Detailed help coming soon.  Until then, I'll just list my options.");
-       for (unsigned i = 0; lopts[i].name; i++) {
-               const struct option *o = &lopts[i];
-               printf("\t--%s", o->name);
-
-               if (o->has_arg == 1) {
-                       printf("=val");
-               } else if (o->has_arg == 2) {
-                       printf("[=val]");
-               }
 
-               printf(" (-%c)\n", o->val);
+       putchar('\n');
+       puts("Options:");
+       for (opt = lopts; opt->name; opt++) {
+               struct lopt_help help;
+
+               if (!lopt_get_help(opt, &help))
+                       continue;
+
+               help_print_option(opt, help.arg, help.desc, 20);
        }
+       putchar('\n');
+
+       puts("For more information, see the upkg(1) man page.");
+       putchar('\n');
+
+       printf("Report bugs to <%s>.\n", PACKAGE_BUGREPORT);
 }
 
 static void print_upkg_flags(const char *prefix, unsigned long flags)
@@ -167,7 +171,7 @@ static void print_upkg_object_flags(const char *prefix, unsigned long flags)
                printf("%sDebugDestroy\n", prefix);
 }
 
-void print_guid(unsigned char guid[static 16])
+static void print_guid(unsigned char *guid)
 {
        for (unsigned i = 0; i < 16; i++) {
                printf("%02hhx", guid[i]);
@@ -178,18 +182,64 @@ void print_guid(unsigned char guid[static 16])
        }
 }
 
-void print_upkg_exports(struct upkg *pkg)
+static void export_print_name(const struct upkg_export *e)
+{
+       if (e) {
+               export_print_name(e->parent);
+               printf(".%s", e->name);
+       }
+}
+
+static void
+export_print_fullname(GTypeModule *pkg, const struct upkg_export *export)
 {
-       for (unsigned i = 0; i < pkg->export_count; i++) {
-               const struct upkg_export *export = upkg_get_export(pkg, i);
-               const char *package, *class;
+       printf("%s", pkg->name);
+       export_print_name(export);
+}
 
-               class = upkg_export_class(pkg, i, &package);
+static void import_print_fullname(const struct upkg_import *import)
+{
+       if (import) {
+               import_print_fullname(import->parent);
+               if (import->parent)
+                       putchar('.');
+               printf("%s", import->name);
+       }
+}
 
-               if (!class)
-                       continue;
+static void
+export_print_class(GTypeModule *pkg, const struct upkg_export *export)
+{
+       struct upkg *upkg = U_PKG(pkg)->pkg;
+
+       if (export->class < 0) {
+               const struct upkg_import *class;
+
+               class = upkg_get_import(upkg, -(export->class+1));
+               import_print_fullname(class);
+       } else if (export->class > 0) {
+               const struct upkg_export *class;
+
+               class = upkg_get_export(upkg, export->class-1);
+               export_print_fullname(pkg, class);
+       } else {
+               printf("Core.Class");
+       }
+}
+
+void print_upkg_exports(GTypeModule *pkg)
+{
+       struct upkg *upkg = U_PKG(pkg)->pkg;
+
+       for (unsigned i = 0; i < upkg->export_count; i++) {
+               const struct upkg_export *export = upkg_get_export(upkg, i);
+
+               printf("%u - ", i+1);
+               export_print_fullname(pkg, export);
+               printf(" (");
+               export_print_class(pkg, export);
+               printf(")\n");
 
-               printf("%u - %s (%s.%s)\n", i+1, export->name, package, class);
                printf("  Flags: %lx\n", export->flags);
                if (verbose >= 2) {
                        print_upkg_object_flags("    ", export->flags);
@@ -197,49 +247,183 @@ void print_upkg_exports(struct upkg *pkg)
        }
 }
 
-int package_info(struct upkg *pkg)
+void print_upkg_imports(struct upkg *pkg)
+{
+       for (unsigned i = 0; i < pkg->import_count; i++) {
+               const struct upkg_import *import = upkg_get_import(pkg, i);
+
+               printf("%u - ", i);
+               import_print_fullname(import);
+               printf(" (%s.%s)\n", import->class_package, import->class_name);
+       }
+}
+
+int package_info(GTypeModule *pkg)
 {
-       printf("Version: %u\n",  pkg->version);
-       printf("License: %u\n",  pkg->license);
+       struct upkg *upkg = U_PKG(pkg)->pkg;
+
+       printf("Version: %u\n", upkg->version);
+       printf("License: %u\n", upkg->license);
        printf("GUID: ");
-       print_guid(pkg->guid);
+       print_guid(upkg->guid);
 
-       printf("Flags:   %lx\n", pkg->flags);
+       printf("Flags:   %lx\n", upkg->flags);
        if (verbose >= 2) {
-               print_upkg_flags("        ", pkg->flags);
+               print_upkg_flags("        ", upkg->flags);
        }
 
-       printf("Names:   %lu\n", pkg->name_count);
+       printf("Names:   %lu\n", upkg->name_count);
        if (verbose >= 3) {
-               for (unsigned long i = 0; i < pkg->name_count; i++) {
-                       printf("\t%s\n", upkg_get_name(pkg, i));
+               for (unsigned long i = 0; i < upkg->name_count; i++) {
+                       printf("%lu - %s\n", i, upkg_get_name(upkg, i));
                }
        }
 
-       printf("Exports: %lu\n", pkg->export_count);
+       printf("Exports: %lu\n", upkg->export_count);
        if (verbose >= 1) print_upkg_exports(pkg);
-       printf("Imports: %lu\n", pkg->import_count);
+       printf("Imports: %lu\n", upkg->import_count);
+       if (verbose >= 1) print_upkg_imports(upkg);
+
+       return 0;
+}
+
+static int package_list(GTypeModule *pkg, long current)
+{
+       struct upkg *upkg = U_PKG(pkg)->pkg;
+       const struct upkg_export *parent = NULL;
+
+       if (current >= 0) {
+               parent = upkg_get_export(upkg, current);
+               assert(parent != NULL);
+       }
+
+       for (unsigned i = 0; i < upkg->export_count; i++) {
+               const struct upkg_export *export = upkg_get_export(upkg, i);
+
+               if (export->parent != parent)
+                       continue;
+
+               export_print_fullname(pkg, export);
+
+               if (verbose >= 1) {
+                       printf(" (");
+                       export_print_class(pkg, export);
+                       printf(")");
+               }
+
+               putchar('\n');
+       }
 
        return 0;
 }
 
-static int export(struct upkg *pkg, GObject *obj, unsigned idx)
+static int object_dump_properties(GTypeModule *pkg, unsigned long idx)
 {
-       const struct upkg_export *export = upkg_get_export(pkg, idx);
+       GParamSpec **props;
+       GObject *obj;
+       unsigned n;
+
+       obj = u_object_new_from_package(pkg, idx);
+       if (!obj) {
+               fprintf(stderr, "%s: failed to load object.\n", progname);
+               return -1;
+       }
+
+       props = g_object_class_list_properties(G_OBJECT_GET_CLASS(obj), &n);
+       for (unsigned i = 0; i < n; i++) {
+               GValue val = {0};
+               char *valstr;
+
+               printf("  property %s:", props[i]->name);
+               fflush(stdout);
+
+               g_value_init(&val, props[i]->value_type);
+               g_object_get_property(obj, props[i]->name, &val);
+
+               if (G_VALUE_HOLDS(&val, U_TYPE_OBJECT)) {
+                       UObject *obj = g_value_get_object(&val);
+
+                       if (obj && obj->pkg_name) {
+                               printf(" %s", obj->pkg_name);
+                       }
+               }
+
+               valstr = g_strdup_value_contents(&val);
+               printf(" %s\n", valstr);
+               g_free(valstr);
+       }
+
+       g_free(props);
+       g_object_unref(obj);
+
+       return 0;
+}
+
+static int object_dump_offsets(GTypeModule *pkg, unsigned long idx)
+{
+       struct upkg_file *f;
+
+       f = upkg_export_open(U_PKG(pkg)->pkg, idx);
+       if (!f) {
+               fprintf(stderr, "%s: failed to open export.\n", progname);
+               return -1;
+       }
+
+       printf("  file size: %lu\n", f->len);
+       printf("  file start: %#lx\n", f->base);
+       printf("  file end: %#lx\n", f->base + f->len);
+
+       upkg_export_close(f);
+
+       return 0;
+}
+
+static int object_info(GTypeModule *pkg, unsigned long idx)
+{
+       struct upkg *upkg = U_PKG(pkg)->pkg;
+       const struct upkg_export *export;
+       int ret = 0;
+
+       export = upkg_get_export(upkg, idx);
+       export_print_fullname(pkg, export);
+
+       printf(" (");
+       export_print_class(pkg, export);
+       printf(")\n");
+
+       if (verbose >= 1 && object_dump_properties(pkg, idx) != 0)
+               ret = -1;
+
+       if (verbose >= 3 && object_dump_offsets(pkg, idx))
+               ret = -1;
+
+       return ret;
+}
+
+static int export(GTypeModule *pkg, GObject *obj, unsigned idx)
+{
+       struct upkg *upkg = U_PKG(pkg)->pkg;
        char name[256];
        FILE *of;
+       int rc;
 
-       if (u_object_deserialize(obj, pkg, idx) != 0) {
+       if (U_OBJECT_IS_LOADABLE(obj) && u_object_load(obj) != 0) {
+               fprintf(stderr, "%s: failed to load export data.\n", progname);
                return -1;
        }
 
-       if (U_OBJECT_IS_LOADABLE(obj) && u_object_load(obj) != 0) {
+       rc = u_object_export_name(obj, name, sizeof name);
+       if (rc <= 0) {
+               /* XXX: We could use a default name here. */
+               fprintf(stderr, "%s: failed to determine export filename.\n",
+                               progname);
                return -1;
        }
 
-       u_object_export_name(obj, name, sizeof name);
+       printf("exporting ");
+       export_print_fullname(pkg, upkg_get_export(upkg, idx));
+       printf(" to %s\n", name);
 
-       printf("exporting %s to %s\n", export->name, name);
        of = fopen(name, "wb");
        if (!of) {
                perror(name);
@@ -260,28 +444,38 @@ static int export(struct upkg *pkg, GObject *obj, unsigned idx)
        return 0;
 }
 
-int package_export(struct upkg *pkg)
+static int object_export(GTypeModule *pkg, unsigned long idx)
 {
-       const char *class, *package;
        GObject *obj;
-       GType type;
-       int ret = 0;
+       int ret = -1;
 
-       for (unsigned i = 0; i < pkg->export_count; i++) {
-               class = upkg_export_class(pkg, i, &package);
-               if (!class) {
-                       fprintf(stderr, "error getting class information.\n");
-                       return -1;
-               }
+       obj = u_object_new_from_package(pkg, idx);
+       if (!obj) {
+               fprintf(stderr, "%s: failed to load object.\n", progname);
+               return -1;
+       }
 
-               type = u_object_module_get_class(package, class);
-               if (!type) continue;
+       if (!U_OBJECT_IS_EXPORTABLE(obj)) {
+               fprintf(stderr, "%s: object is not exportable.\n", progname);
+               goto out;
+       }
 
-               obj = g_object_new(type, NULL);
-               if (U_OBJECT_IS_EXPORTABLE(obj)) {
-                       if (export(pkg, obj, i) != 0) {
-                               ret = -1;
-                       }
+       ret = export(pkg, obj, idx);
+out:
+       g_object_unref(obj);
+       return ret;
+}
+
+int package_export(GTypeModule *pkg)
+{
+       struct upkg *upkg = U_PKG(pkg)->pkg;
+       GObject *obj;
+       int ret = 0;
+
+       for (unsigned i = 0; i < upkg->export_count; i++) {
+               obj = u_object_new_from_package(pkg, i);
+               if (U_OBJECT_IS_EXPORTABLE(obj) && export(pkg, obj, i) != 0) {
+                       ret = -1;
                }
                g_object_unref(obj);
        }
@@ -341,10 +535,19 @@ static int process_object(int mode, const char *objname)
 
        switch (mode) {
        case MODE_INFO:
-               ret = package_info(U_PKG(pkg)->pkg);
+               if (current < 0)
+                       ret = package_info(pkg);
+               else
+                       ret = object_info(pkg, current);
+               break;
+       case MODE_LIST:
+               ret = package_list(pkg, current);
                break;
        case MODE_EXPORT:
-               ret = package_export(U_PKG(pkg)->pkg);
+               if (current < 0)
+                       ret = package_export(pkg);
+               else
+                       ret = object_export(pkg, current);
                break;
        default:
                abort();
@@ -381,6 +584,9 @@ int main(int argc, char **argv)
                case 'i':
                        mode = MODE_INFO;
                        break;
+               case 'l':
+                       mode = MODE_LIST;
+                       break;
                case 'x':
                        mode = MODE_EXPORT;
                        break;