mirror of
https://github.com/dobin/SuperMega
synced 2026-06-02 17:27:10 +00:00
Dobin
51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
import logging
|
|
|
|
from model import *
|
|
from config import config
|
|
from observer import observer
|
|
from pe.pehelper import *
|
|
from helper import *
|
|
|
|
logger = logging.getLogger("Assembler")
|
|
|
|
|
|
def asm_to_shellcode(asm_in: FilePath, build_exe: FilePath) -> bytes:
|
|
"""Takes ASM source file asm_in, compiles it into build_exe, extracts its code section and write into shellcode_out"""
|
|
logger.info("--[ Assemble to exe: {} -> {}".format(asm_in, build_exe))
|
|
run_process_checkret([
|
|
config.get("path_ml64"),
|
|
asm_in,
|
|
"/link",
|
|
"/OUT:{}".format(build_exe),
|
|
"/entry:AlignRSP" # "/entry:main",
|
|
])
|
|
if not os.path.isfile(build_exe):
|
|
raise Exception("Compiling failed")
|
|
code = extract_code_from_exe_file(build_exe)
|
|
return code
|
|
|
|
|
|
def merge_loader_payload(
|
|
shellcode_in: bytes,
|
|
payload_data: bytes,
|
|
decoder_style: DecoderStyle
|
|
) -> bytes:
|
|
payload_data = encode_payload(payload_data, decoder_style)
|
|
|
|
logger.info("---[ Size: Carrier: {} and Payload: {} Sum: {} ".format(
|
|
len(shellcode_in), len(payload_data), len(shellcode_in)+len(payload_data)))
|
|
|
|
return shellcode_in + payload_data
|
|
|
|
|
|
def encode_payload(payload: bytes, decoder_style: DecoderStyle) -> bytes:
|
|
if decoder_style == DecoderStyle.PLAIN_1:
|
|
return payload
|
|
elif decoder_style == DecoderStyle.XOR_1:
|
|
xor_key = config.xor_key
|
|
logger.info("---[ XOR payload with key 0x{:X}".format(xor_key))
|
|
xored = bytes([byte ^ xor_key for byte in payload])
|
|
return xored
|
|
else:
|
|
raise Exception("Unknown decoder style")
|