ce1d870d54
Mod for TikTok 47.1.3 (arm64) in Material 3 style: bottom bar, profile menu, TikTok settings and comments; downloads without the watermark; switches for ads, live streams and the Shop tab; badges of other mods next to names and the niche mod badge (members service); app icons; mod settings screen. Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
import unittest
|
|
|
|
from lihont import bar_patch
|
|
|
|
ON_RESUME = """\
|
|
.class public Lcom/ss/android/ugc/aweme/main/MainActivity;
|
|
.super LX/0tPX;
|
|
|
|
.method public onResume()V
|
|
.registers 5
|
|
|
|
const v0, 0x117de
|
|
|
|
invoke-static {v0}, Lcom/bytedance/pumbaa/utility/method_id/MethodIDManager;->push(I)Z
|
|
|
|
move-result v3
|
|
|
|
const/4 v0, 0x1
|
|
|
|
const-string v2, "com.ss.android.ugc.aweme.main.MainActivity"
|
|
|
|
const-string v1, "onResume"
|
|
|
|
invoke-static {v2, v1, v0}, Lcom/bytedance/apm/agent/v2/instrumentation/ActivityAgent;->onTrace(Ljava/lang/String;Ljava/lang/String;Z)V
|
|
|
|
invoke-super {p0}, LX/1EKk;->onResume()V
|
|
|
|
const/4 v0, 0x0
|
|
|
|
invoke-static {v2, v1, v0}, Lcom/bytedance/apm/agent/v2/instrumentation/ActivityAgent;->onTrace(Ljava/lang/String;Ljava/lang/String;Z)V
|
|
|
|
return-void
|
|
.end method
|
|
|
|
.method public onStart()V
|
|
.registers 4
|
|
|
|
invoke-super {p0}, LX/1EKk;->onStart()V
|
|
|
|
return-void
|
|
.end method
|
|
"""
|
|
|
|
|
|
class BarPatchTest(unittest.TestCase):
|
|
def test_hook_follows_the_super_call_and_nothing_else_changes(self):
|
|
patched = bar_patch.rewrite_main_activity(ON_RESUME)
|
|
call = " invoke-static {p0}, " + bar_patch.HOOK
|
|
self.assertEqual(1, patched.count(call))
|
|
self.assertLess(patched.index("->onResume()V"), patched.index(call))
|
|
self.assertLess(patched.index(call), patched.index("const/4 v0, 0x0"))
|
|
# onStart's own super call is left alone and the rest of the file is byte for byte the same
|
|
self.assertEqual(ON_RESUME, patched.replace("\n" + call + "\n", "", 1))
|
|
|
|
def test_it_refuses_to_patch_twice(self):
|
|
patched = bar_patch.rewrite_main_activity(ON_RESUME)
|
|
with self.assertRaises(ValueError):
|
|
bar_patch.rewrite_main_activity(patched)
|
|
|
|
def test_it_refuses_a_method_it_does_not_recognise(self):
|
|
with self.assertRaises(ValueError):
|
|
bar_patch.rewrite_main_activity(ON_RESUME.replace("->onResume()V", "->onPause()V"))
|
|
with self.assertRaises(ValueError):
|
|
bar_patch.rewrite_main_activity(".class public Lx;\n.super Ljava/lang/Object;\n")
|
|
|
|
def test_runtime_target_is_the_hook(self):
|
|
self.assertEqual([bar_patch.HOOK], bar_patch.runtime_targets())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|