1 files changed, 126 insertions, 0 deletions

diff --git a/utils/mknwzboot/main.c b/utils/mknwzboot/main.c
new file mode 100644
index 0000000000..491a18043b
--- /dev/null
+++ b/utils/mknwzboot/main.c
@@ -0,0 +1,126 @@
+/***************************************************************************
+ *             __________               __   ___.
+ *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
+ *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
+ *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
+ *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
+ *                     \/            \/     \/    \/            \/
+ * $Id$
+ *
+ * Copyright (C) 2011 by Amaury Pouly
+ *
+ * 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 2
+ * of the License, or (at your option) any later version.
+ *
+ * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
+ * KIND, either express or implied.
+ *
+ ****************************************************************************/
+
+#include <getopt.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include "mknwzboot.h"
+
+static void usage(void)
+{
+    printf("Usage: mknwzboot [options | file]...\n");
+    printf("Options:\n");
+    printf("  -h/--help   Display this message\n");
+    printf("  -o <file>   Set output file\n");
+    printf("  -b <file>   Set boot file\n");
+    printf("  -d/--debug  Enable debug output\n");
+    printf("  -x          Dump device informations\n");
+    printf("  -u          Create uninstall update\n");
+    printf("  -m <model>  Specify model\n");
+    exit(1);
+}
+
+int main(int argc, char *argv[])
+{
+    char *outfile = NULL;
+    char *bootfile = NULL;
+    bool debug = false;
+    bool install = true;
+    const char *model = NULL;
+
+    if(argc == 1)
+        usage();
+
+    while(1)
+    {
+        static struct option long_options[] =
+        {
+            {"help", no_argument, 0, 'h'},
+            {"out-file", required_argument, 0, 'o'},
+            {"boot-file", required_argument, 0, 'b'},
+            {"debug", no_argument, 0, 'd'},
+            {"dev-info", no_argument, 0, 'x'},
+            {"uninstall", no_argument, 0, 'u'},
+            {"model", required_argument, 0, 'm'},
+            {0, 0, 0, 0}
+        };
+
+        int c = getopt_long(argc, argv, "ho:b:dxum:", long_options, NULL);
+        if(c == -1)
+            break;
+        switch(c)
+        {
+            case 'd':
+                debug = true;
+                break;
+            case 'h':
+                usage();
+                break;
+            case 'o':
+                outfile = optarg;
+                break;
+            case 'b':
+                bootfile = optarg;
+                break;
+            case 'x':
+                dump_nwz_dev_info("");
+                return 0;
+            case 'u':
+                install = false;
+                break;
+            case 'm':
+                model = optarg;
+                break;
+            default:
+                abort();
+        }
+    }
+
+    if(!outfile)
+    {
+        printf("You must specify an output file\n");
+        return 1;
+    }
+    if(install && !bootfile)
+    {
+        printf("You must specify a boot file for installation\n");
+        return 1;
+    }
+    if(!install && !model)
+    {
+        printf("You must provide a model for uninstallation\n");
+        return 1;
+    }
+    if(optind != argc)
+    {
+        printf("Extra arguments on command line\n");
+        return 1;
+    }
+
+    int err;
+    if(install)
+        err = mknwzboot(bootfile, outfile, debug);
+    else
+        err = mknwzboot_uninst(model, outfile, debug);
+    printf("Result: %d\n", err);
+    return err;
+}