1 files changed, 56 insertions, 0 deletions

diff --git a/utils/nwztools/upgtools/fwp.c b/utils/nwztools/upgtools/fwp.c
new file mode 100644
index 0000000000..e739b8caef
--- /dev/null
+++ b/utils/nwztools/upgtools/fwp.c
@@ -0,0 +1,56 @@
+/***************************************************************************
+ *             __________               __   ___.
+ *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
+ *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
+ *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
+ *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
+ *                     \/            \/     \/    \/            \/
+ * $Id$
+ *
+ * Copyright (C) 2012 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 <stdio.h>
+#include "fwp.h"
+#include "misc.h"
+#include "mg.h"
+#include <string.h>
+
+int fwp_read(void *in, int size, void *out, uint8_t *key)
+{
+    return mg_decrypt_fw(in, size, out, key);
+}
+
+static uint8_t g_key[NWZ_KEY_SIZE];
+
+void fwp_setkey(char key[NWZ_KEY_SIZE])
+{
+    memcpy(g_key, key, NWZ_KEY_SIZE);
+}
+
+int fwp_crypt(void *buf, int size, int mode)
+{
+    while(size >= NWZ_KEY_SIZE)
+    {
+        if(mode)
+            mg_decrypt_pass(buf, NWZ_KEY_SIZE, buf, g_key);
+        else
+            mg_encrypt_pass(buf, NWZ_KEY_SIZE, buf, g_key);
+        buf += NWZ_KEY_SIZE;
+        size -= NWZ_KEY_SIZE;
+    }
+    if(size != 0)
+    {
+        cprintf(GREY, "Cannot fwp_crypt non-multiple of 8!\n");
+        return -1;
+    }
+    return 0;
+}
\ No newline at end of file