mirror of
https://github.com/dobin/SuperMega
synced 2026-06-03 01:27:11 +00:00
Dobin
59 lines
2.2 KiB
Python
59 lines
2.2 KiB
Python
import shutil
|
|
from typing import List
|
|
import unittest
|
|
import logging
|
|
|
|
from model.exehost import ExeHost
|
|
from model.defs import *
|
|
from pe.pehelper import extract_code_from_exe_file
|
|
from utils import hexdump
|
|
from observer import observer
|
|
from model.defs import *
|
|
from pe.derbackdoorer import FunctionBackdoorer
|
|
from pe.superpe import SuperPe
|
|
|
|
|
|
class DerBackdoorerTest(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
observer.active = False
|
|
|
|
|
|
def test_function_backdoorer_exe(self):
|
|
shellcode = b"\x90" * 200
|
|
superpe = SuperPe(PATH_EXES + "iattest-full.exe")
|
|
function_backdoorer = FunctionBackdoorer(superpe, shellcode)
|
|
|
|
instr = function_backdoorer.find_suitable_instruction_addr(superpe.get_entrypoint(), 128, 5)
|
|
self.assertIsNotNone(instr)
|
|
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
|
|
superpe = SuperPe(PATH_EXES + "libbz2-1.dll")
|
|
function_backdoorer = FunctionBackdoorer(superpe, shellcode)
|
|
|
|
instr = function_backdoorer.find_suitable_instruction_addr(superpe.get_entrypoint(), 128, 5)
|
|
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) |