X-Git-Url: https://git.draconx.ca/gitweb/upkg.git/blobdiff_plain/eedd5d24e385f09ec109fffb558797e3c6e2889d..d42bc5c2d8dcf388b424c5bbc4b6f25416f689f1:/src/upkg.c diff --git a/src/upkg.c b/src/upkg.c index 038eb39..50f5dda 100644 --- a/src/upkg.c +++ b/src/upkg.c @@ -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 @@ -21,6 +21,7 @@ #include #include #include +#include #include #include @@ -35,6 +36,7 @@ enum { MODE_INFO, + MODE_LIST, MODE_EXPORT, MODE_MAX }; @@ -42,9 +44,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' }, @@ -178,25 +181,61 @@ void print_guid(unsigned char guid[static 16]) } } +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) +{ + printf("%s", pkg->name); + export_print_name(export); +} + +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); + } +} + 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); } } } +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(struct upkg *pkg) { printf("Version: %u\n", pkg->version); @@ -212,33 +251,159 @@ int package_info(struct upkg *pkg) printf("Names: %lu\n", pkg->name_count); if (verbose >= 3) { for (unsigned long i = 0; i < pkg->name_count; i++) { - printf("\t%s\n", upkg_get_name(pkg, i)); + printf("%lu - %s\n", i, upkg_get_name(pkg, i)); } } printf("Exports: %lu\n", pkg->export_count); if (verbose >= 1) print_upkg_exports(pkg); printf("Imports: %lu\n", pkg->import_count); + if (verbose >= 1) print_upkg_imports(pkg); + + 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) { + const char *class, *package; + + class = upkg_export_class(upkg, i, &package); + printf(" (%s.%s)", package, class); + } + + putchar('\n'); + } + + return 0; +} + +static int object_dump_properties(GTypeModule *pkg, unsigned long 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 export(struct upkg *pkg, GObject *obj, unsigned idx) +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; + int ret = 0; + + export = upkg_get_export(upkg, idx); + export_print_fullname(pkg, export); + + class = upkg_export_class(upkg, idx, &package); + if (class) + printf(" (%s.%s)\n", package, class); + else + printf(" (failed to get class)\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", upkg_export_name(pkg, idx), name); of = fopen(name, "wb"); if (!of) { perror(name); @@ -259,28 +424,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); } @@ -340,10 +515,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(U_PKG(pkg)->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(); @@ -380,6 +564,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;