36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
## -*- coding: utf-8 -*-
|
|
##
|
|
## Jonathan Salwan - 2014-05-13
|
|
##
|
|
## http://shell-storm.org
|
|
## http://twitter.com/JonathanSalwan
|
|
##
|
|
|
|
from capstone import *
|
|
from ropgadget.ropchain.arch.ropmakerx86 import *
|
|
from ropgadget.ropchain.arch.ropmakerx64 import *
|
|
|
|
class ROPMaker(object):
|
|
def __init__(self, binary, gadgets, offset):
|
|
self.__binary = binary
|
|
self.__gadgets = gadgets
|
|
self.__offset = offset
|
|
|
|
self.__handlerArch()
|
|
|
|
def __handlerArch(self):
|
|
|
|
if self.__binary.getArch() == CS_ARCH_X86 \
|
|
and self.__binary.getArchMode() == CS_MODE_32 \
|
|
and self.__binary.getFormat() == "ELF":
|
|
ROPMakerX86(self.__binary, self.__gadgets, self.__offset)
|
|
|
|
elif self.__binary.getArch() == CS_ARCH_X86 \
|
|
and self.__binary.getArchMode() == CS_MODE_64 \
|
|
and self.__binary.getFormat() == "ELF":
|
|
ROPMakerX64(self.__binary, self.__gadgets, self.__offset)
|
|
|
|
else:
|
|
print("\n[Error] ROPMaker.__handlerArch - Arch not supported yet for the rop chain generation")
|
|
|