1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
import sys
from argparse import ArgumentParser
from miasm.loader import pe_init
parser = ArgumentParser(description="Create a PE from a shellcode")
parser.add_argument("filename",
help="x86 shellcode filename")
parser.add_argument("-p", "--pename",
help="new PE filename (default is 'sc_pe.exe')",
default="sc_pe.exe")
parser.add_argument("-w", "--word-size",
help="word size (default is 32 bits)",
choices=[32, 64],
type=int,
default=32)
args = parser.parse_args()
data = open(args.filename, 'rb').read()
pe = pe_init.PE(wsize=args.word_size)
s_text = pe.SHList.add_section(name="text", addr=0x1000, data=data)
pe.Opthdr.AddressOfEntryPoint = s_text.addr
open(args.pename, 'wb').write(bytes(pe))
|