From 1bd786c9d75b1b5ae6727378593ad02b9b1f27d1 Mon Sep 17 00:00:00 2001
From: Jean-Marie 'Histausse' Mineau <histausse@protonmail.com>
Date: Sun, 28 Jan 2024 23:37:15 +0100
Subject: [PATCH] update test script

---
 test.py | 110 ++++++++++++++++++++++++++++++++------------------------
 1 file changed, 64 insertions(+), 46 deletions(-)

diff --git a/test.py b/test.py
index 47384f7..f4214c6 100644
--- a/test.py
+++ b/test.py
@@ -4,6 +4,7 @@ FORMAT = "[%(levelname)s] %(name)s %(filename)s:%(lineno)d: %(message)s"
 logging.basicConfig(format=FORMAT)
 logging.getLogger().setLevel(logging.DEBUG)
 
+import json
 import androscalpel as asc
 import zipfile as z
 from androscalpel import *
@@ -27,49 +28,66 @@ clazz = apk.classes[clazz_id]
 method = clazz.virtual_methods[method_id]
 code = method.code
 
-# logging.getLogger().setLevel(logging.ERROR)
-#
-# print(f"Code of {method_id}")
-# for i in code.insns:
-#    print(i)
-#
-# new_insns = []
-# for i in code.insns:
-#    if isinstance(i, asc.ins.ConstString):
-#        if i.lit == "Hello":
-#            i = asc.ins.ConstString(i.reg, DexString("Degemer Mat"))
-#        elif i.lit == "Bye":
-#            i = asc.ins.ConstString(i.reg, DexString("Kenavo"))
-#    new_insns.append(i)
-#
-## This need improving!
-# code = asc.Code(code.registers_size, code.ins_size, code.outs_size, new_insns)
-# apk.set_method_code(method_id, code)
-## apk.set_method_code(method.descriptor, code)
-#
-# clazz = apk.classes[clazz_id]
-# method = clazz.virtual_methods[method_id]
-# code = method.code
-#
-# print(f"Code of {method_id}")
-# for i in code.insns:
-#    print(i)
-#
-# dex_raw = apk.gen_raw_dex()
-# assert len(dex_raw) == 1
-# with open(DEX_NAME, "wb") as file:
-#    file.write(dex_raw[0])
-#
-#
-# with open(DEX_NAME, "rb") as file:
-#    dex = file.read()
-# new_apk = asc.Apk()
-# new_apk.add_dex_file(dex)
-#
-# clazz = new_apk.classes[clazz_id]
-# method = clazz.virtual_methods[method_id]
-# code = method.code
-#
-# print(f"Code of {method_id} in new apk")
-# for i in code.insns:
-#    print(i)
+logging.getLogger().setLevel(logging.ERROR)
+
+print(f"[+] Code of {method_id} ")
+for i in code.insns:
+    print(f"    {i}")
+print("[+] Modify code")
+new_insns = []
+for i in code.insns:
+    if isinstance(i, asc.ins.ConstString):
+        if i.lit == "Hello":
+            i = asc.ins.ConstString(i.reg, DexString("Degemer Mat"))
+        elif i.lit == "Bye":
+            i = asc.ins.ConstString(i.reg, DexString("Kenavo"))
+    new_insns.append(i)
+
+# This need improving!
+code = asc.Code(code.registers_size, code.ins_size, code.outs_size, new_insns)
+apk.set_method_code(method_id, code)
+# apk.set_method_code(method.descriptor, code)
+
+
+clazz = apk.classes[clazz_id]
+method = clazz.virtual_methods[method_id]
+code = method.code
+print(f"[+] New code of {method_id} ")
+for i in code.insns:
+    print(f"    {i}")
+
+# Strip class for debugging
+classes = list(
+    filter(
+        lambda x: x
+        not in [
+            IdType("Lcom/example/testapplication/ui/home/HomeViewModel;"),
+            IdType("Landroidx/navigation/NavDeepLink$Builder;"),
+            IdType("Landroidx/constraintlayout/core/widgets/ConstraintWidget$1;"),
+        ],
+        apk.classes.keys(),
+    )
+)
+for cls in classes:
+    apk.remove_class(cls)
+
+print("[+] Recompile")
+
+dex_raw = apk.gen_raw_dex()
+assert len(dex_raw) == 1
+with open(DEX_NAME, "wb") as file:
+    file.write(dex_raw[0])
+
+print("[+] Load new dex")
+with open(DEX_NAME, "rb") as file:
+    dex = file.read()
+new_apk = asc.Apk()
+new_apk.add_dex_file(dex)
+
+clazz = new_apk.classes[clazz_id]
+method = clazz.virtual_methods[method_id]
+code = method.code
+
+print(f"[+] Code of {method_id} in new apk")
+for i in code.insns:
+    print(f"    {i}")