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

refactor: project -> settings and model/

Browse Source
This commit is contained in:
Dobin
2024-02-25 08:47:25 +00:00
parent b1dd3481ed
commit d61f358ae4
18 changed files with 185 additions and 157 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_asm.py
+9 -9
View File
@@ -4,8 +4,8 @@ import unittest
import logging
from phases.compiler import fixup_asm_file, fixup_iat_reuse
from model import ExeInfo
from defs import *
from model.exehost import ExeHost
from model.defs import *
from observer import observer
@@ -44,15 +44,15 @@ class AsmTest(unittest.TestCase):
path_working: FilePath = "tests/data/iat_reuse_pre_fixup.asm.test"
shutil.copy(path_in, path_working)
exe_info = ExeInfo()
exe_host = ExeHost()
fixup_iat_reuse(path_working, exe_info)
self.assertTrue(len(exe_info.iat_resolves), 2)
fixup_iat_reuse(path_working, exe_host)
self.assertTrue(len(exe_host.iat_resolves), 2)
self.assertTrue("GetEnvironmentVariableW" in exe_info.iat_resolves)
self.assertEqual(exe_info.iat_resolves["GetEnvironmentVariableW"].name, "GetEnvironmentVariableW")
self.assertEqual(exe_info.iat_resolves["GetEnvironmentVariableW"].addr, 0)
self.assertTrue(len(exe_info.iat_resolves["GetEnvironmentVariableW"].id), 6) # 6 random bytes
self.assertTrue("GetEnvironmentVariableW" in exe_host.iat_resolves)
self.assertEqual(exe_host.iat_resolves["GetEnvironmentVariableW"].name, "GetEnvironmentVariableW")
self.assertEqual(exe_host.iat_resolves["GetEnvironmentVariableW"].addr, 0)
self.assertTrue(len(exe_host.iat_resolves["GetEnvironmentVariableW"].id), 6) # 6 random bytes
with open(path_working, "r") as f:
lines = f.readlines()