39 lines
1.3 KiB
Python
39 lines
1.3 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, paddingLen, outFile, execPath, offset):
|
|
self.__binary = binary
|
|
self.__gadgets = gadgets
|
|
self.paddingLen = paddingLen
|
|
self.outFile = outFile
|
|
self.execPath = execPath
|
|
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.paddingLen, self.outFile, self.execPath, 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.paddingLen, self.__offset)
|
|
|
|
else:
|
|
print("\n[Error] ROPMaker.__handlerArch - Arch not supported yet for the rop chain generation")
|
|
|