include ROPGadget

ROPgadget/ropgadget/ropchain/ropmaker.py

@@ -0,0 +1,35 @@
+## -*- 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")
+