mirror of
https://github.com/dobin/SuperMega
synced 2026-06-03 01:27:11 +00:00
Dobin
31 lines
701 B
Python
31 lines
701 B
Python
from typing import Dict, List
|
|
import logging
|
|
|
|
from model.exehost import ExeHost
|
|
|
|
logger = logging.getLogger("ExeHost")
|
|
|
|
|
|
class IatEntry():
|
|
def __init__(self, name: str, placeholder: bytes):
|
|
self.name: str = name # Function Name, like "VirtualAlloc"
|
|
self.placeholder: bytes = placeholder # Random bytes as placeholder
|
|
|
|
|
|
|
|
class Carrier():
|
|
def __init__(self):
|
|
self.iat_requests: List[IatEntry] = []
|
|
|
|
|
|
def init(self):
|
|
pass
|
|
|
|
|
|
def add_iat_request(self, func_name: str, placeholder: bytes):
|
|
self.iat_requests.append(IatEntry(func_name, placeholder))
|
|
|
|
|
|
def get_all_iat_requests(self) -> List[IatEntry]:
|
|
return self.iat_requests
|