refactor: make address type more clear

This commit is contained in:
Dobin
2024-02-25 20:33:19 +00:00
parent 5301ef7b2e
commit 16438e799b
3 changed files with 5 additions and 5 deletions
+3 -3
View File
@@ -124,15 +124,15 @@ class ExeHost():
self.iat[dll_name].append({
"dll_name": dll_name,
"func_name": imp_name,
"func_addr": imp_addr
"iat_vaddr": imp_addr
})
def get_addr_of_iat_function(self, func_name: str) -> int:
def get_vaddr_of_iatentry(self, func_name: str) -> int:
for dll_name in self.iat:
for entry in self.iat[dll_name]:
if entry["func_name"] == func_name:
return entry["func_addr"]
return entry["iat_vaddr"]
return None
## Other