admin/SuperMega
1
0
Fork 0
mirror of https://github.com/dobin/SuperMega synced 2026-06-03 01:27:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests: fix forgotten

Browse Source
This commit is contained in:
Dobin
2024-04-27 14:29:29 +01:00
parent aca1ed46bc
commit 061b83f19b
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_derbackdoorer.py
+4 -6
View File
@@ -22,7 +22,7 @@ class DerBackdoorerTest(unittest.TestCase):
addr = function_backdoorer.find_suitable_instruction_addr(superpe.get_entrypoint()) addr = function_backdoorer.find_suitable_instruction_addr(superpe.get_entrypoint())
self.assertEqual(addr, 0x1304) self.assertEqual(addr, 0x1304)
trampoline_compiled, trampline_text, trampoline_reloc_offset = function_backdoorer.get_trampoline(addr, 0x11223344) trampoline_compiled, _, trampoline_reloc_offset = function_backdoorer.get_trampoline(addr, 0x11223344)
self.assertEqual(trampoline_compiled[0], 0x48) self.assertEqual(trampoline_compiled[0], 0x48)
self.assertEqual(trampoline_compiled[2], 0x44) self.assertEqual(trampoline_compiled[2], 0x44)
self.assertEqual(trampoline_compiled[3], 0x33) self.assertEqual(trampoline_compiled[3], 0x33)
@@ -39,13 +39,11 @@ class DerBackdoorerTest(unittest.TestCase):
addr = function_backdoorer.find_suitable_instruction_addr(superpe.get_entrypoint()) addr = function_backdoorer.find_suitable_instruction_addr(superpe.get_entrypoint())
self.assertEqual(addr, 0x135D) self.assertEqual(addr, 0x135D)
trampoline_compiled, trampoline_reloc_offset = function_backdoorer.get_trampoline(addr, 0x11223344) trampoline_compiled, _, trampoline_reloc_offset = function_backdoorer.get_trampoline(addr, 0x11223344)
self.assertEqual(trampoline_compiled[0], 0x48) self.assertEqual(trampoline_compiled[0], 0x48)
self.assertEqual(trampoline_compiled[2], 0x44) self.assertEqual(trampoline_compiled[2], 0x44)
self.assertEqual(trampoline_compiled[3], 0x33) self.assertEqual(trampoline_compiled[3], 0x33)
self.assertEqual(trampoline_compiled[4], 0x22) self.assertEqual(trampoline_compiled[6], 0x02)
self.assertEqual(trampoline_compiled[5], 0x51)
self.assertEqual(trampoline_compiled[6], 0x01)
self.assertEqual(trampoline_compiled[10], 0xff) self.assertEqual(trampoline_compiled[10], 0xff)
self.assertEqual(trampoline_reloc_offset, 2) self.assertEqual(trampoline_reloc_offset, 2)