Recursive disassembler

[frida/frida.git] / src / disassembler / Disassembler.hxx

diff --git a/src/disassembler/Disassembler.hxx b/src/disassembler/Disassembler.hxx
index 13c36b6d30a481a7d2892a81a644f1f377ad9464..14083262f5027ac7b22057c009d4dfb1feeb8ff0 100644 (file)
--- a/src/disassembler/Disassembler.hxx
+++ b/src/disassembler/Disassembler.hxx
@@ -1,15 +1,29 @@
 #ifndef INCLUDE__Disassembler_hxx
 #define INCLUDE__Disassembler_hxx
 
-#include "Target.hxx"
-#include "include.hxx"
+#include <string>
+#include <functional>
+
+#include "disassembler/BasicBlock.hxx"
+
 
 class Disassembler {
 public:
-    Disassembler(const std::string& filename);
-private:
-    llvm::object::Binary* _binary;
-    Target _target;
+       Disassembler(const std::string& filename) {}
+    virtual ~Disassembler() {}
+
+       void getSymbols();
+       uint64_t entryAddress();
+
+    virtual void forEachInstruction(const std::string& name, std::function<void (long, std::string, std::string)> callback) = 0;
+
+    // http://llvm.org/docs/doxygen/html/MCObjectDisassembler_8cpp_source.html +197
+       BasicBlock * generateControlFlowGraph(const std::string& name);
+       BasicBlock * generateControlFlowGraph(uint64_t address);
+
+protected:
+    virtual bool isFunctionCall(uint64_t address) = 0;
+    virtual bool isJump(uint64_t address) = 0;
 };
 
 #endif