Add scripts

Author: larsenv

wc24decrypt.py

@@ -0,0 +1,49 @@
+import os
+import sys
+from binascii import unhexlify
+from Crypto.Cipher import AES
+from requests import get
+
+if len(sys.argv) < 4:
+    print("Usage: python wc24decrypt.py <input file / url> <output file> <key / wc24pubk.mod>")
+    sys.exit(1)
+
+if sys.argv[1][:4] == "http":
+    with open("temp", "wb") as f:
+        f.write(get(sys.argv[1]).content)
+        f.close()
+    input = open("temp", "rb")
+else:
+    input = open(sys.argv[1], "rb")
+
+if os.path.exists(sys.argv[3]):
+    with open(sys.argv[3], "rb") as f:
+        if os.path.getsize(sys.argv[3]) == 16:
+            key = f.read()
+        elif os.path.getsize(sys.argv[3]) == 544:
+            f.seek(512)
+            key = f.read(16)
+        else:
+            print("Error: Input file is not a 16-byte key or a wc24pubk.mod")
+            sys.exit(1)
+else:
+    if len(sys.argv[3]) != 32:
+        print("Error: Key is not 16 bytes")
+        sys.exit(1)
+    key = unhexlify(sys.argv[3])
+
+input.seek(48)
+iv = input.read(16)
+input.seek(320)
+data = input.read()
+
+aes = AES.new(key, AES.MODE_OFB, iv=iv)
+
+with open(sys.argv[2], "wb") as f:
+    f.write(aes.decrypt(data))
+    f.close()
+
+if os.path.exists("temp"):
+    os.remove("temp")
+
+print("Completed Successfully")

wc24encrypt.py

@@ -0,0 +1,107 @@
+import argparse
+import os
+import rsa
+import struct
+from binascii import unhexlify
+from Crypto.Cipher import AES
+from nlzss import encode_file
+
+def u8(data):
+    return struct.pack(">B", data)
+
+
+def u16(data):
+    return struct.pack(">H", data)
+
+
+def u32(data):
+    return struct.pack(">I", data)
+
+parser = argparse.ArgumentParser(description="Sign / Encrypt WiiConnect24 files.")
+parser.add_argument("-t", "--type",
+                        type=str, nargs="+",
+                        help="Type of file. Set either enc for encrypted file, or dec for non-encrypted file.")
+parser.add_argument("-in", "--input",
+                        type=str, nargs="+",
+                        help="Input file.")
+parser.add_argument("-out", "--output",
+                        type=str, nargs="+",
+                        help="Output file.")
+parser.add_argument("-c", "--compress",
+                        type=str, nargs="+",
+                        help="If set, this will compress the file before signing with LZ10.")
+parser.add_argument("-key", "--aes-key",
+                        type=str, nargs="+",
+                        help="AES key in hex or a path.")
+parser.add_argument("-iv", "--iv-key",
+                        type=str, nargs="+",
+                        help="AES IV in hex or a path.")
+parser.add_argument("-rsa", "--rsa-key-path",
+                        type=str, nargs="+",
+                        help="RSA private key path. If not specified, it will use the private key in Private.pem if it exists.")
+
+args = parser.parse_args()
+
+if args.compress is not None:
+    encode_file(in_path=args.input[0], out_path="temp")
+    filename = "temp"
+else:
+    filename = args.input[0]
+
+with open(filename, "rb") as f:
+    data = f.read()
+
+if args.rsa_key_path is not None:
+    rsa_key_path = args.rsa_key_path[0]
+else:
+    rsa_key_path = "Private.pem"
+
+with open(rsa_key_path, "rb") as source_file:
+    private_key_data = source_file.read()
+
+private_key = rsa.PrivateKey.load_pkcs1(private_key_data, "PEM")
+
+signature = rsa.sign(data, private_key, "SHA-1")
+
+if args.type[0] == "enc":
+    if args.iv_key is not None:
+        try:
+            iv = unhexlify(args.iv_key[0])
+        except:
+            iv = open(args.iv_key[0], "rb").read()
+    else:
+        iv = os.urandom(16)
+
+    try:
+        key = unhexlify(args.aes_key[0])
+    except:
+        key = open(args.aes_key[0], "rb").read()
+
+    aes = AES.new(key, AES.MODE_OFB, iv=iv)
+    processed = aes.encrypt(data)
+elif args.type[0] == "dec":
+    processed = data
+
+content = {}
+
+content["magic"] = b"WC24" if args.type[0] == "enc" else u32(0)
+content["version"] = u32(1) if args.type[0] == "enc" else u32(0)
+content["filler"] = u32(0)
+content["crypt_type"] = u8(1) if args.type[0] == "enc" else u8(0)
+content["pad"] = u8(0) * 3
+content["reserved"] = u8(0) * 32
+content["iv"] = iv if args.type[0] == "enc" else u8(0) * 16
+content["signature"] = signature
+content["data"] = processed
+
+if os.path.exists(args.output[0]):
+    os.remove(args.output[0])
+
+if args.type[0] == "dec":
+    os.remove("temp")
+
+for values in content.values():
+    with open(args.output[0], "ab+") as f:
+        f.write(values)
+
+print("Completed Successfully")