You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.1 KiB
41 lines
1.1 KiB
## -*- coding: utf-8 -*-
|
|
##
|
|
## Jonathan Salwan - 2014-05-13
|
|
##
|
|
## http://shell-storm.org
|
|
## http://twitter.com/JonathanSalwan
|
|
##
|
|
|
|
from capstone import *
|
|
|
|
from ropgadget.ropchain.arch.ropmakerx64 import *
|
|
from ropgadget.ropchain.arch.ropmakerx86 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")
|