diff --git a/model.py b/model.py index a08dfaa..167632c 100644 --- a/model.py +++ b/model.py @@ -37,7 +37,7 @@ class ExeInfo(): self.rwx_section = None - def add_capability(self, func_name, placeholder): + def add_iat_resolve(self, func_name, placeholder): self.iat_resolves[func_name] = IatResolve( func_name, placeholder, pehelper.get_addr_for(self.iat, func_name)) diff --git a/phases/compiler.py b/phases/compiler.py index a78f3bd..58388d1 100644 --- a/phases/compiler.py +++ b/phases/compiler.py @@ -149,7 +149,7 @@ def fixup_iat_reuse(filename: FilePath, exe_info): randbytes: bytes = os.urandom(6) lines[idx] = bytes_to_asm_db(randbytes) + "\r\n" - exe_info.add_capability(func_name, randbytes) + exe_info.add_iat_resolve(func_name, randbytes) logger.info(" > Replace func name: {} with {}".format( func_name, randbytes))