feat: .cyan files

cyan/__main__.py

@@ -21,6 +21,10 @@ def main() -> None:
     "-o", "--output", metavar="output",
     help="if unspecified, overwrites input"
   )
+  parser.add_argument(
+    "-z", "--cyan", metavar="cyan",
+    help="the .cyan file to use"
+  )
 
   parser.add_argument(
     "-f", metavar="file", nargs="+",

cyan/logic.py

@@ -38,6 +38,10 @@ def main(parser: ArgumentParser) -> None:
       else:
         sys.exit("[!] main binary is encrypted; exiting")
 
+    if args.cyan is not None:
+      changing = vars(args)
+      tbhutils.parse_cyan(changing, tmpdir)
+
     # this goes before injection,
     # since user might inject their own extensions
     if args.remove_extensions:

cyan/tbhutils.py

@@ -1,13 +1,14 @@
 import os
 import sys
+import json
 import shutil
 import zipfile
 import platform
 import subprocess
 from uuid import uuid4
-from typing import Optional
 from glob import glob, iglob
 from argparse import Namespace
+from typing import Optional, Any
 
 
 def validate_inputs(args: Namespace) -> Optional[str]:
@@ -57,6 +58,9 @@ def validate_inputs(args: Namespace) -> Optional[str]:
   if args.k is not None and not os.path.isfile(args.k):
     sys.exit(f"[!] {args.k} does not exist")
 
+  if args.cyan is not None and not os.path.isfile(args.cyan):
+    sys.exit(f"[!] {args.cyan} does not exist")
+
 
 def get_app(path: str, tmpdir: str, is_ipa: bool) -> str:
   payload = f"{tmpdir}/Payload"
@@ -179,3 +183,29 @@ def make_ipa(tmpdir: str, output: str, level: int) -> None:
   if weird != 0:
     print(f"[?] was unable to zip {weird} file(s) due to timestamps")
 
+
+def parse_cyan(args: dict[str, Any], tmpdir: str) -> None:
+  print("[*] parsing .cyan file..")
+  with zipfile.ZipFile(args["cyan"]) as zf:
+    DOT_PATH = f"{tmpdir}/cyan"
+    os.mkdir(DOT_PATH)
+
+    with zf.open("config.json") as f:
+      config = json.load(f)
+
+    if "f" in config:
+      NAMES = [n for n in zf.namelist() if n.startswith("inject/")]
+      zf.extractall(DOT_PATH, NAMES)
+
+      # ensure not None
+      args["f"] = args["f"] if args["f"] is not None else {}
+      for e in os.scandir(f"{DOT_PATH}/inject"):
+        args["f"][e.name] = e.path
+      del config["f"]
+    if "k" in config:
+      args["k"] = zf.extract("icon.idk", DOT_PATH)
+      del config["k"]
+
+    for k, v in config.items():
+      args[k] = v
+