refactor: some cleanup

This commit is contained in:
Dobin Rutishauser
2024-06-23 17:17:24 +02:00
parent 720d85c994
commit 240b5e6494
2 changed files with 10 additions and 6 deletions
+6 -3
View File
@@ -23,19 +23,22 @@ class Injector():
def __init__(
self,
carrier_shc: bytes,
settings: Settings,
payload: Payload,
injectable: Injectable,
payload: Payload):
settings: Settings):
self.carrier_shc = carrier_shc
self.settings = settings
self.injectable = injectable
self.payload = payload
# superpe is a representation of the exe file. We gonna modify it, and save it at the end.
self.superpe = SuperPe(settings.inject_exe_in)
# reuse from injectable
#self.superpe = SuperPe(settings.inject_exe_in)
self.superpe = injectable.superpe
self.function_backdoorer = FunctionBackdoorer(self.superpe)
# to find space for carrier and payload
# for some combination of settings HACK
self.payload_rva = None
self.carrier_rva = None
self.init_addresses()
+4 -3
View File
@@ -199,9 +199,10 @@ def start_real(settings: Settings):
# INJECT loader into an exe and do IAT & data references. Big task.
injector = phases.injector.Injector(
carrier_shellcode,
settings,
project.injectable,
project.payload)
project.payload,
project.injectable,
settings)
injector.inject_exe()
#observer.add_code_file("exe_final", extract_code_from_exe_file_ep(settings.inject_exe_out, 300))