]> git.draconx.ca Git - upkg.git/blobdiff - src/upkg.c
uobject: Add a helper function to load objects from packages.
[upkg.git] / src / upkg.c
index 19f1449a3e1aa53ab550438c43886d88909ad433..c13a867f7b1dfd8cbdbf21977c537ccc89d8da19 100644 (file)
@@ -1,6 +1,6 @@
 /*
  *  upkg: tool for manipulating Unreal Tournament packages.
- *  Copyright (C) 2009 Nick Bowler
+ *  Copyright © 2009-2011 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
@@ -19,6 +19,8 @@
 #include <config.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <string.h>
+#include <errno.h>
 #include <getopt.h>
 
 #include <glib-object.h>
@@ -33,6 +35,7 @@
 
 enum {
        MODE_INFO,
+       MODE_LIST,
        MODE_EXPORT,
        MODE_MAX
 };
@@ -40,9 +43,10 @@ enum {
 int verbose = 0;
 
 static const char *progname = "upkg";
-static const char *sopts = "ixvf:VH";
+static const char *sopts = "ilxvf:VH";
 static const struct option lopts[] = {
        { "info",     0, NULL, 'i' },
+       { "list",     0, NULL, 'l' },
        { "export",   0, NULL, 'x' },
        { "file",     1, NULL, 'f' },
        { "verbose",  0, NULL, 'v' },
@@ -64,7 +68,7 @@ There is NO WARRANTY, to the extent permitted by law."
 
 void print_usage(FILE *f)
 {
-       fprintf(f, "Usage: %s [options]\n", progname);
+       fprintf(f, "Usage: %s [options] [object ...]\n", progname);
 }
 
 void print_help(void)
@@ -179,18 +183,18 @@ void print_guid(unsigned char guid[static 16])
 void print_upkg_exports(struct upkg *pkg)
 {
        for (unsigned i = 0; i < pkg->export_count; i++) {
-               const char *name, *package, *class;
+               const struct upkg_export *export = upkg_get_export(pkg, i);
+               const char *package, *class;
 
-               name  = upkg_export_name(pkg, i);
                class = upkg_export_class(pkg, i, &package);
 
-               if (!name || !class)
+               if (!class)
                        continue;
 
-               printf("%u - %s (%s.%s)\n", i+1, name, package, class);
-               printf("  Flags: %lx\n", upkg_export_flags(pkg, i));
+               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("    ", upkg_export_flags(pkg, i));
+                       print_upkg_object_flags("    ", export->flags);
                }
        }
 }
@@ -218,11 +222,96 @@ int package_info(struct upkg *pkg)
        if (verbose >= 1) print_upkg_exports(pkg);
        printf("Imports: %lu\n", pkg->import_count);
 
-       return EXIT_SUCCESS;
+       return 0;
+}
+
+static void export_print_name(struct upkg *upkg, const struct upkg_export *e)
+{
+       if (e) {
+               export_print_name(upkg, upkg_get_export(upkg, e->package-1));
+               printf(".%s", e->name);
+       }
+}
+
+static void
+export_print_fullname(GTypeModule *pkg, const struct upkg_export *export)
+{
+       printf("%s", pkg->name);
+       export_print_name(U_PKG(pkg)->pkg, export);
+}
+
+static int package_list(GTypeModule *pkg, long parent)
+{
+       struct upkg *upkg = U_PKG(pkg)->pkg;
+       const struct upkg_export *export;
+
+       for (unsigned i = 0; i < upkg->export_count; i++) {
+               export = upkg_get_export(upkg, i);
+
+               if (export->package != parent)
+                       continue;
+
+               export_print_fullname(pkg, export);
+
+               if (verbose >= 1) {
+                       const char *class, *package;
+
+                       class = upkg_export_class(upkg, i, &package);
+                       printf(" (%s.%s)", package, class);
+               }
+
+               putchar('\n');
+       }
+}
+
+static int object_info(GTypeModule *pkg, unsigned long idx)
+{
+       struct upkg *upkg = U_PKG(pkg)->pkg;
+       const struct upkg_export *export;
+       const char *class, *package;
+
+       export = upkg_get_export(upkg, idx);
+       export_print_fullname(pkg, export);
+
+       class = upkg_export_class(upkg, idx, &package);
+       printf(" (%s.%s)\n", package, class);
+
+       if (verbose >= 1) {
+               GParamSpec **props;
+               GObject *obj;
+               unsigned n;
+
+               obj = u_object_new_from_package(upkg, 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("  %s: ", props[i]->name);
+                       fflush(stdout);
+
+                       g_value_init(&val, props[i]->value_type);
+                       g_object_get_property(obj, props[i]->name, &val);
+                       valstr = g_strdup_value_contents(&val);
+                       printf("%s\n", valstr);
+               }
+               free(props);
+
+               g_object_unref(obj);
+       }
+
+       return 0;
 }
 
 static int export(struct upkg *pkg, GObject *obj, unsigned idx)
 {
+       const struct upkg_export *export = upkg_get_export(pkg, idx);
        char name[256];
        FILE *of;
 
@@ -236,7 +325,7 @@ static int export(struct upkg *pkg, GObject *obj, unsigned idx)
 
        u_object_export_name(obj, name, sizeof name);
 
-       printf("exporting %s to %s\n", upkg_export_name(pkg, idx), name);
+       printf("exporting %s to %s\n", export->name, name);
        of = fopen(name, "wb");
        if (!of) {
                perror(name);
@@ -262,13 +351,13 @@ int package_export(struct upkg *pkg)
        const char *class, *package;
        GObject *obj;
        GType type;
-       int rc = EXIT_SUCCESS;
+       int ret = 0;
 
        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 EXIT_FAILURE;
+                       return -1;
                }
 
                type = u_object_module_get_class(package, class);
@@ -277,21 +366,93 @@ int package_export(struct upkg *pkg)
                obj = g_object_new(type, NULL);
                if (U_OBJECT_IS_EXPORTABLE(obj)) {
                        if (export(pkg, obj, i) != 0) {
-                               rc = EXIT_FAILURE;
+                               ret = -1;
                        }
                }
                g_object_unref(obj);
        }
 
-       return rc;
+       return ret;
 }
 
-int main(int argc, char **argv)
+static int process_object(int mode, const char *objname)
 {
+       char work[strlen(objname)+1], *p = work, *c;
        GTypeModule *pkg;
+       long current = -1;
+       int ret = 0;
+
+       strcpy(work, objname);
+
+       /* First, find and open the actual package. */
+       c = strchr(p, '.');
+       if (c) *c = 0;
+
+       pkg = u_pkg_open(p);
+       if (!pkg || !g_type_module_use(pkg)) {
+               fprintf(stderr, "%s: %s: failed to open package.\n",
+                               progname, work);
+               goto out;
+       }
+
+       if (!U_PKG(pkg)->pkg) {
+               if (u_pkg_is_native(pkg)) {
+                       fprintf(stderr, "%s: %s: not a UObject package.\n",
+                                       progname, work);
+               } else {
+                       fprintf(stderr, "%s: %s: package not found.\n",
+                                       progname, work);
+               }
+
+               goto out_unuse;
+       }
+
+       /* Resolve the hierarchical reference. */
+       while (c) {
+               p = c+1;
+               c = strchr(p, '.');
+               if (c) *c = 0;
+
+               current = upkg_export_find(U_PKG(pkg)->pkg, current, p);
+               if (current == -1) {
+                       /* We want to print the full name up to this point. */
+                       strcpy(work, objname);
+                       if (c) *c = 0;
+
+                       fprintf(stderr, "%s: %s: object not found.\n",
+                                       progname, work);
+                       goto out_unuse;
+               }
+       }
+
+       switch (mode) {
+       case MODE_INFO:
+               if (current < 0)
+                       ret = package_info(U_PKG(pkg)->pkg);
+               else
+                       ret = object_info(pkg, current);
+               break;
+       case MODE_LIST:
+               ret = package_list(pkg, current+1);
+               break;
+       case MODE_EXPORT:
+               ret = package_export(U_PKG(pkg)->pkg);
+               break;
+       default:
+               abort();
+       }
+
+out_unuse:
+       g_type_module_unuse(pkg);
+out:
+       return ret;
+}
+
+int main(int argc, char **argv)
+{
        const char *pkgname = NULL;
        unsigned mode = MODE_INFO;
-       int opt, rc = EXIT_FAILURE;
+       int opt, rc;
 
        if (argc > 0) progname = argv[0];
 
@@ -312,6 +473,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;
@@ -330,40 +494,22 @@ int main(int argc, char **argv)
                }
        }
 
-       if (!pkgname) {
+       if (!pkgname && !argv[optind]) {
                print_usage(stderr);
                return EXIT_FAILURE;
        }
 
-       pkg = u_pkg_open(pkgname);
-       if (!pkg || !g_type_module_use(pkg)) {
-               fprintf(stderr, "%s: %s: failed to open package.\n",
-                               progname, pkgname);
-               return EXIT_FAILURE;
-       }
-
-       if (!U_PKG(pkg)->pkg) {
-               if (u_pkg_is_native(pkg)) {
-                       fprintf(stderr, "%s: %s: not a UObject package.\n",
-                                       progname, pkgname);
-               } else {
-                       fprintf(stderr, "%s: %s: package not found.\n",
-                                       progname, pkgname);
+       rc = EXIT_SUCCESS;
+       if (argv[optind]) {
+               for (int i = optind; i < argc; i++) {
+                       if (process_object(mode, argv[i]) != 0)
+                               rc = EXIT_FAILURE;
                }
-
-               return EXIT_FAILURE;
-       }
-
-       switch (mode) {
-       case MODE_INFO:
-               rc = package_info(U_PKG(pkg)->pkg);
-               break;
-       case MODE_EXPORT:
-               rc = package_export(U_PKG(pkg)->pkg);
-               break;
+       } else {
+               if (process_object(mode, pkgname) != 0)
+                       rc = EXIT_FAILURE;
        }
 
-       g_type_module_unuse(pkg);
        u_object_module_exit();
        u_pkg_vfs_exit();
        return rc;