Added module to apply patches to IOS

modules/title/iospatcher.py

@@ -0,0 +1,66 @@
+# "modules/title/iospatcher.py" from WiiPy by NinjaCheetah
+# https://github.com/NinjaCheetah/WiiPy
+
+import pathlib
+import libWiiPy
+
+
+def handle_iospatch(args):
+    input_path = pathlib.Path(args.input)
+    if not input_path.exists():
+        raise FileNotFoundError(input_path)
+
+    title = libWiiPy.title.Title()
+    title.load_wad(open(input_path, "rb").read())
+
+    tid = title.tmd.title_id
+    if tid[:8] != "00000001" or tid[8:] == "00000001" or tid[8:] == "00000002":
+        raise ValueError("This WAD does not appear to contain an IOS! Patching cannot continue.")
+
+    patches_applied = False
+
+    if args.version is not None:
+        title.set_title_version(args.version)
+        patches_applied = True
+
+    if args.slot is not None:
+        slot = args.slot
+        if 3 <= slot <= 255:
+            tid = title.tmd.title_id[:-2] + f"{slot:02X}"
+            title.set_title_id(tid)
+        patches_applied = True
+
+    ios_patcher = libWiiPy.title.IOSPatcher()
+    ios_patcher.load(title)
+
+    if args.all is True:
+        ios_patcher.patch_all()
+        patches_applied = True
+    else:
+        if args.fakesigning is True:
+            ios_patcher.patch_fakesigning()
+            patches_applied = True
+        if args.es_identify is True:
+            ios_patcher.patch_es_identify()
+            patches_applied = True
+        if args.nand_access is True:
+            ios_patcher.patch_nand_access()
+            patches_applied = True
+        if args.version_patch is True:
+            ios_patcher.patch_version_patch()
+            patches_applied = True
+
+    if not patches_applied:
+        raise ValueError("No patches were selected! Please select patches to apply.")
+
+    if args.output is not None:
+        output_path = pathlib.Path(args.output)
+        output_file = open(output_path, "wb")
+        output_file.write(ios_patcher.title.dump_wad())
+        output_file.close()
+    else:
+        output_file = open(input_path, "wb")
+        output_file.write(ios_patcher.title.dump_wad())
+        output_file.close()
+
+    print("IOS successfully patched!")