mirror of
https://github.com/dobin/ShellcodeObfuscationLab
synced 2026-06-08 13:53:31 +00:00
Dobin Rutishauser
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
from typing import Iterator
|
|
import random
|
|
import string
|
|
|
|
|
|
# Based on snovvcrash RC4 encryption script: https://gist.github.com/snovvcrash/3533d950be2d96cf52131e8393794d99
|
|
# Stolen from: https://gist.github.com/hsauers5/491f9dde975f1eaa97103427eda50071
|
|
def key_scheduling(key):
|
|
key = [ord(char) for char in key]
|
|
sched = [i for i in range(0, 256)]
|
|
|
|
i = 0
|
|
for j in range(0, 256):
|
|
i = (i + sched[j] + key[j % len(key)]) % 256
|
|
tmp = sched[j]
|
|
sched[j] = sched[i]
|
|
sched[i] = tmp
|
|
|
|
return sched
|
|
|
|
|
|
def stream_generation(sched: list[int]) -> Iterator[bytes]:
|
|
i, j = 0, 0
|
|
while True:
|
|
i = (1 + i) % 256
|
|
j = (sched[i] + j) % 256
|
|
tmp = sched[j]
|
|
sched[j] = sched[i]
|
|
sched[i] = tmp
|
|
yield sched[(sched[i] + sched[j]) % 256]
|
|
|
|
|
|
def encrypt(plaintext: bytes, key: bytes) -> bytes:
|
|
sched = key_scheduling(key)
|
|
key_stream = stream_generation(sched)
|
|
|
|
ciphertext = b''
|
|
for char in plaintext:
|
|
enc = char ^ next(key_stream)
|
|
ciphertext += bytes([enc])
|
|
|
|
return ciphertext
|
|
|
|
|
|
def rc4api(input_file: str) -> bytes:
|
|
# https://stackoverflow.com/a/2257449
|
|
key = ''.join(random.choices(string.ascii_uppercase + string.digits, k=16))
|
|
|
|
with open(input_file, 'rb') as f:
|
|
result = encrypt(plaintext=f.read(), key=key)
|
|
|
|
ret = ""
|
|
ret += 'char _key[] = "{}";'.format(key)
|
|
ret += 'char shellcode[] = {{{}}};'.format(', '.join(hex(x) for x in result))
|
|
return ret
|