refactor: IAT stuff to ExeHost

phases/assembler.py

@@ -3,7 +3,7 @@ import logging
 from model import *
 from config import config
 from observer import observer
-from pehelper import *
+from peparser.pehelper import *
 from helper import *
 
 logger = logging.getLogger("Assembler")

phases/compiler.py

@@ -136,7 +136,7 @@ def fixup_asm_file(filename: FilePath, payload_len: int, short_call_patching: bo
     return True
 
 
-def get_function_stubs(asm_in: FilePath):
+def get_function_stubs(asm_in: FilePath) -> List[str]:
     functions = []
 
     with open(asm_in, 'r', encoding='utf-8') as asmfile:

phases/injector.py

@@ -6,7 +6,7 @@ import time
 import tempfile
 import logging
 
-from pehelper import *
+from peparser.pehelper import *
 from model.exehost import *
 from observer import observer
 from helper import rbrunmode_str