From 1feeb66f76440cdd52187d21430328780f579e26 Mon Sep 17 00:00:00 2001 From: Dobin Date: Fri, 16 Feb 2024 19:34:16 +0000 Subject: [PATCH] refactor: rename function --- model.py | 2 +- phases/compiler.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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))