#include "module.h"
#include "uobject.h"
#include "exportable.h"
+#include "loadable.h"
-static const char *progname = "upkg";
+enum {
+ MODE_INFO,
+ MODE_EXPORT,
+ MODE_MAX
+};
+
+int verbose = 0;
-static const char *sopts = "VH";
+static const char *progname = "upkg";
+static const char *sopts = "ixvVH";
static const struct option lopts[] = {
+ { "info", 0, NULL, 'i' },
+ { "export", 0, NULL, 'x' },
+ { "verbose", 0, NULL, 'v' },
{ "version", 0, NULL, 'V' },
{ "help", 0, NULL, 'H' },
{ 0 }
);
}
-void print_usage(void)
+void print_usage(FILE *f)
{
- printf("Usage: %s [options] package\n", progname);
+ fprintf(f, "Usage: %s [options] package\n", progname);
}
void print_help(void)
{
- print_usage();
+ 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];
}
}
+void print_upkg_exports(struct upkg *pkg)
+{
+ for (unsigned i = 0; i < pkg->export_count; i++) {
+ const char *name, *package, *class;
+
+ name = upkg_export_name(pkg, i);
+ class = upkg_export_class(pkg, i, &package);
+
+ if (!name || !class)
+ continue;
+
+ printf("%u - %s (%s.%s)\n", i+1, name, package, class);
+ }
+}
+
+int package_info(struct upkg *pkg)
+{
+ printf("Version: %u\n", pkg->version);
+ printf("License: %u\n", pkg->license);
+ printf("GUID: ");
+ print_guid(pkg->guid);
+
+ printf("Flags: %lx\n", pkg->flags);
+ if (verbose >= 1) {
+ print_upkg_flags("\t", pkg->flags);
+ }
+
+ if (verbose >= 1) {
+ printf("Names: %lu\n", pkg->name_count);
+ if (verbose >= 2) {
+ for (unsigned long i = 0; i < pkg->name_count; i++) {
+ printf("\t%s\n", 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);
+
+ return EXIT_SUCCESS;
+}
+
+static int export(struct upkg *pkg, GObject *obj, unsigned idx)
+{
+ char name[256];
+ FILE *of;
+
+ if (u_object_deserialize(obj, pkg, idx) != 0) {
+ return -1;
+ }
+
+ if (U_OBJECT_IS_LOADABLE(obj) && u_object_load(obj) != 0) {
+ return -1;
+ }
+
+ u_object_export_name(obj, name, sizeof name);
+
+ printf("exporting %s to %s\n", upkg_export_name(pkg, idx), name);
+ of = fopen(name, "wb");
+ if (!of) {
+ perror(name);
+ return -1;
+ }
+
+ u_object_export(obj, of);
+
+ if (fclose(of) != 0) {
+ perror(name);
+ return -1;
+ }
+
+ if (U_OBJECT_IS_LOADABLE(obj)) {
+ u_object_unload(obj);
+ }
+
+ return 0;
+}
+
+int package_export(struct upkg *pkg)
+{
+ const char *class, *package;
+ GObject *obj;
+ GType type;
+ int rc = EXIT_SUCCESS;
+
+ 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;
+ }
+
+ type = module_get_class(package, class);
+ if (!type) continue;
+
+ obj = g_object_new(type, NULL);
+ if (U_OBJECT_IS_EXPORTABLE(obj)) {
+ if (export(pkg, obj, i) != 0) {
+ rc = EXIT_FAILURE;
+ }
+ }
+ g_object_unref(obj);
+ }
+
+ return rc;
+}
+
int main(int argc, char **argv)
{
struct upkg *pkg;
- int opt;
+ unsigned mode = MODE_INFO;
+ int opt, rc = EXIT_FAILURE;
if (argc > 0) progname = argv[0];
while ((opt = getopt_long(argc, argv, sopts, lopts, NULL)) != -1) {
switch (opt) {
+ case 'i':
+ mode = MODE_INFO;
+ break;
+ case 'x':
+ mode = MODE_EXPORT;
+ break;
+ case 'v':
+ verbose++;
+ break;
case 'V':
print_version();
return EXIT_SUCCESS;
print_help();
return EXIT_SUCCESS;
default:
- print_usage();
+ print_usage(stderr);
return EXIT_FAILURE;
}
}
- if (argc < 2) {
- fprintf(stderr, "usage: upkg file\n");
+ if (argv[optind] == NULL) {
+ print_usage(stderr);
return EXIT_FAILURE;
}
if (module_init() != 0)
return EXIT_FAILURE;
- pkg = upkg_fopen(argv[1]);
+ pkg = upkg_fopen(argv[optind]);
if (!pkg) {
fprintf(stderr, "failed to open package!\n");
return EXIT_FAILURE;
}
- printf("Version: %u\n", pkg->version);
- printf("License: %u\n", pkg->license);
- printf("GUID: ");
- print_guid(pkg->guid);
-
- printf("Flags: %lx\n", pkg->flags);
- print_upkg_flags("\t", pkg->flags);
-
- printf("Names: %lu\n", pkg->name_count);
- for (unsigned long i = 0; i < pkg->name_count; i++) {
- printf("\t%s\n", upkg_get_name(pkg, i));
+ switch (mode) {
+ case MODE_INFO:
+ rc = package_info(pkg);
+ break;
+ case MODE_EXPORT:
+ rc = package_export(pkg);
+ break;
}
- printf("Exports: %lu\n", pkg->export_count);
- printf("Imports: %lu\n", pkg->import_count);
-
- GObject *music = g_object_new(module_get_class("Engine", "Music"), NULL);
- if (!music)
- return EXIT_FAILURE;
- struct upkg_file *f = upkg_export_open(pkg, 0);
- u_object_deserialize(music, f);
-
- char name[256];
- u_object_export_name(music, name, sizeof name);
- FILE *fp = fopen(name, "wb");
- if (!fp) return EXIT_FAILURE;
- u_object_export(music, fp);
- fclose(fp);
- printf("Wrote %s\n", name);
-
upkg_close(pkg);
module_exit();
-
- return 0;
+ return rc;
}