capstone isn't disassembling the whole block, dumb heuristics for growth

master
JoYo 2019-02-24 06:34:26 +00:00
parent 0b709669b0
commit ad6277c0ee
1 changed files with 9 additions and 4 deletions

View File

@ -55,13 +55,18 @@ def generation(queue: Queue, shellcode: bytes):
def growth(*, shellcode: bytes, length: int) -> bytes:
for mnemonic, op_str in disasm(shellcode):
if mnemonic == 'nop':
return bytes(shellcode)
if length <= len(shellcode):
return bytes(shellcode)
opcodes = disasm(shellcode)
if len(shellcode) > len(opcodes) * 8:
return bytes(shellcode)
for mnemonic, op_str in opcodes:
if mnemonic == 'nop':
return bytes(shellcode)
shellcode = bytearray(shellcode)
shellcode += b'\x90'