diff --git a/tests/test_datareuse.py b/tests/test_datareuse.py index 3ead267..5645a90 100644 --- a/tests/test_datareuse.py +++ b/tests/test_datareuse.py @@ -3,13 +3,13 @@ from typing import List import unittest import logging import os - +from model.defs import * from model.exehost import ExeHost from phases.datareuse import ReusedataAsmFileParser class DataReuseTest(unittest.TestCase): def test_relocation_list(self): - exe_host = ExeHost("data/exes/7z.exe") + exe_host = ExeHost(PATH_EXES + "7z.exe") exe_host.init() relocs = exe_host.get_relocations_for_section(".rdata") @@ -22,7 +22,7 @@ class DataReuseTest(unittest.TestCase): def test_largestgap(self): - exe_host = ExeHost("data/exes/7z.exe") + exe_host = ExeHost(PATH_EXES + "7z.exe") exe_host.init() rm = exe_host.get_rdata_relocmanager() start, stop = rm.find_hole(100) diff --git a/tests/test_derbackdoorer.py b/tests/test_derbackdoorer.py index b097d5c..c424a3d 100644 --- a/tests/test_derbackdoorer.py +++ b/tests/test_derbackdoorer.py @@ -6,9 +6,9 @@ import logging from model.exehost import ExeHost from model.defs import * from pe.pehelper import extract_code_from_exe_file -from helper import hexdump +from utils import hexdump from observer import observer - +from model.defs import * from pe.derbackdoorer import PeBackdoor @@ -28,13 +28,13 @@ class DerBackdoorerTest(unittest.TestCase): def test_backdoor_ep(self): # Write example shellcode - shellcode_path = "data/exes/shellcode.test" + shellcode_path = PATH_EXES + "shellcode.test" shellcode = b"\x90" * 200 with open(shellcode_path, "wb") as f: f.write(shellcode) - exe_path = "data/exes/iattest-full.exe" - exe_out_path = "data/exes/iattest-full-test.exe" + exe_path = PATH_EXES + "iattest-full.exe" + exe_out_path = PATH_EXES + "iattest-full.test.exe" shutil.copyfile(exe_path, exe_out_path) @@ -59,12 +59,12 @@ class DerBackdoorerTest(unittest.TestCase): def test_backdoor_hijack(self): # Write example shellcode shellcode = b"\x90" * 200 - with open("data/exes/shellcode.test", "wb") as f: + with open(PATH_EXES + "shellcode.test", "wb") as f: f.write(shellcode) - shellcode_path = "data/exes/shellcode.test" - exe_path = "data/exes/7z.exe" - exe_out_path = "data/exes/7z-test.exe" + shellcode_path = PATH_EXES + "shellcode.test" + exe_path = PATH_EXES + "7z.exe" + exe_out_path = PATH_EXES + "7z.test.exe" shutil.copyfile(exe_path, exe_out_path) diff --git a/tests/test_rangemanager.py b/tests/test_rangemanager.py index 689d15a..7365462 100644 --- a/tests/test_rangemanager.py +++ b/tests/test_rangemanager.py @@ -2,6 +2,7 @@ from typing import List import unittest import logging +from model.defs import * from pe.superpe import SuperPe from model.exehost import ExeHost from model.rangemanager import RangeManager @@ -22,7 +23,7 @@ class RangeManagerTest(unittest.TestCase): def test_relocmanager(self): - exehost = ExeHost("data/exes/procexp64.exe") + exehost = ExeHost(PATH_EXES + "procexp64.exe") exehost.init() section = exehost.superpe.get_section_by_name(".rdata") rm = exehost.get_rdata_relocmanager()