From 061b83f19b348bd5b6b017020845b34cc246a877 Mon Sep 17 00:00:00 2001 From: Dobin Date: Sat, 27 Apr 2024 14:29:29 +0100 Subject: [PATCH] tests: fix forgotten --- tests/test_derbackdoorer.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/tests/test_derbackdoorer.py b/tests/test_derbackdoorer.py index e6a7d56..fa22bce 100644 --- a/tests/test_derbackdoorer.py +++ b/tests/test_derbackdoorer.py @@ -22,7 +22,7 @@ class DerBackdoorerTest(unittest.TestCase): addr = function_backdoorer.find_suitable_instruction_addr(superpe.get_entrypoint()) 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[2], 0x44) self.assertEqual(trampoline_compiled[3], 0x33) @@ -39,13 +39,11 @@ class DerBackdoorerTest(unittest.TestCase): addr = function_backdoorer.find_suitable_instruction_addr(superpe.get_entrypoint()) 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[2], 0x44) self.assertEqual(trampoline_compiled[3], 0x33) - self.assertEqual(trampoline_compiled[4], 0x22) - self.assertEqual(trampoline_compiled[5], 0x51) - self.assertEqual(trampoline_compiled[6], 0x01) + self.assertEqual(trampoline_compiled[6], 0x02) self.assertEqual(trampoline_compiled[10], 0xff) self.assertEqual(trampoline_reloc_offset, 2) \ No newline at end of file