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

tests: improve functionbackdoorer unittest

Browse Source
This commit is contained in:
Dobin
2024-04-20 19:08:51 +01:00
parent bc6cc4df2d
commit 1f768f0f15
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_derbackdoorer.py
+18
View File
@@ -29,6 +29,15 @@ class DerBackdoorerTest(unittest.TestCase):
self.assertEqual(instr.mnemonic, "jne")
self.assertEqual(instr.address, 0x1701)
trampoline_compiled, trampoline_reloc_offset = function_backdoorer.get_trampoline(instr)
print(hexdump(trampoline_compiled))
self.assertEqual(trampoline_compiled[0], 0x48)
self.assertEqual(trampoline_compiled[2], 0x00)
self.assertEqual(trampoline_compiled[5], 0x40)
self.assertEqual(trampoline_compiled[6], 0x01)
self.assertEqual(trampoline_compiled[10], 0xff)
self.assertEqual(trampoline_reloc_offset, 2)
def test_function_backdoorer_dll(self):
shellcode = b"\x90" * 200
@@ -39,3 +48,12 @@ class DerBackdoorerTest(unittest.TestCase):
self.assertIsNotNone(instr)
self.assertEqual(instr.mnemonic, "jne")
self.assertEqual(instr.address, 0x1220)
trampoline_compiled, trampoline_reloc_offset = function_backdoorer.get_trampoline(instr)
print(hexdump(trampoline_compiled))
self.assertEqual(trampoline_compiled[0], 0x48)
self.assertEqual(trampoline_compiled[2], 0x00)
self.assertEqual(trampoline_compiled[5], 0xf1)
self.assertEqual(trampoline_compiled[6], 0x01)
self.assertEqual(trampoline_compiled[10], 0xff)
self.assertEqual(trampoline_reloc_offset, 2)