refactor: carrier->injectable so i have sane carrier- & payload locations (+cleaner injector)

This commit is contained in:
Dobin Rutishauser
2024-06-23 14:11:30 +02:00
parent 29db10cbe6
commit e10f9c7fda
8 changed files with 335 additions and 281 deletions
+2 -2
View File
@@ -6,7 +6,7 @@ from model.defs import *
from pe.superpe import SuperPe, PeSection
logger = logging.getLogger("Carrier")
logger = logging.getLogger("Injectable")
class IatRequest():
@@ -39,7 +39,7 @@ class DataReuseEntry():
self.references.append(DataReuseReference(placeholder, register))
class Carrier():
class Injectable():
def __init__(self, exe_file: str):
self.iat_requests: List[IatRequest] = []
self.reusedata_fixups: List[DataReuseEntry] = []
+3 -3
View File
@@ -4,7 +4,7 @@ import shutil
from model.defs import *
from model.payload import Payload
from model.settings import Settings
from model.carrier import Carrier
from model.injectable import Injectable
logger = logging.getLogger("Project")
@@ -22,7 +22,7 @@ class Project():
self.comment: str = ""
self.settings: Settings = settings
self.payload: Payload = Payload(self.settings.payload_path)
self.carrier: Carrier = Carrier(self.settings.inject_exe_in)
self.injectable: Injectable = Injectable(self.settings.inject_exe_in)
self.project_dir: str = ""
self.project_exe: str = ""
@@ -30,7 +30,7 @@ class Project():
def init(self):
self.payload.init()
self.carrier.init()
self.injectable.init()
def prepare_project(project_name, settings):
+4 -2
View File
@@ -11,15 +11,17 @@ class RangeManager:
self.min = min
self.max = max
self.intervals.add(Interval(0, min))
def merge_overlaps(self):
self.intervals.merge_overlaps(strict=False)
def print_all(self):
logger.info("Min: {} Max: {}".format(self.min, self.max))
print("Min: {} Max: {}".format(self.min, self.max))
for i in self.intervals:
logger.info("Interval: {}-{}".format(i.begin, i.end))
print("Interval: {}-{}".format(i.begin, i.end))
def add_range(self, start, end):