admin/SuperMega
1
0
Fork 0
mirror of https://github.com/dobin/SuperMega synced 2026-06-02 17:27:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests: fix and reorganize

Browse Source
This commit is contained in:
Dobin
2024-05-08 11:44:37 +01:00
parent b8c834ac56
commit 61f7cf0a5b
9 changed files with 592 additions and 191 deletions
Download Patch File Download Diff File Expand all files Collapse all files
phases/injector.py
+1 -7
View File
@@ -47,15 +47,9 @@ def inject_exe(
# Patch IAT if necessary
if source_style == FunctionInvokeStyle.iat_reuse:
for iatRequest in project.carrier.get_all_iat_requests():
iat_name = superpe.get_iat_name_for("KERNEL32.dll", iatRequest.name)
iat_name = superpe.get_replacement_iat_for("KERNEL32.dll", iatRequest.name)
superpe.patch_iat_entry("KERNEL32.dll", iat_name, iatRequest.name)
#iat_name_a = superpe.get_iat_name_for("KERNEL32.dll", "GetEnvironmentVariableW")
#iat_name_b = superpe.get_iat_name_for("KERNEL32.dll", "VirtualProtect")
#logger.info("Using: {} and {}".format(iat_name_a, iat_name_b))
#superpe.patch_iat_entry("KERNEL32.dll", iat_name_a, "GetEnvironmentVariableW")
#superpe.patch_iat_entry("KERNEL32.dll", iat_name_b, "VirtualProtect")
superpe.pe.parse_data_directories()
shellcode_offset: int = 0 # file offset