mirror of
https://github.com/dobin/SuperMega
synced 2026-06-02 17:27:10 +00:00
refactor: carrier->injectable so i have sane carrier- & payload locations (+cleaner injector)
This commit is contained in:
@@ -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
@@ -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):
|
||||
|
||||
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user