src/disassembler/Disassembler.hxx

   1 #ifndef INCLUDE__Disassembler_hxx
   2 #define INCLUDE__Disassembler_hxx
   3
   4 #include <string>
   5 #include <functional>
   6
   7 #include "disassembler/BasicBlock.hxx"
   8 #include "disassembler/Function.hxx"
   9
  10
  11 class Disassembler {
  12 public:
  13         Disassembler(const std::string& filename) {}
  14     virtual ~Disassembler() {}
  15
  16         void getSymbols();
  17         uint64_t entryAddress();
  18
  19         virtual void forEachFunction(std::function<void (uint64_t, Function*)> callback) = 0;
  20     // virtual void forEachInstruction(const std::string& name, std::function<void (long, std::string, std::string)> callback) = 0;
  21
  22     // http://llvm.org/docs/doxygen/html/MCObjectDisassembler_8cpp_source.html +197
  23         BasicBlock * generateControlFlowGraph(const std::string& name);
  24         BasicBlock * generateControlFlowGraph(uint64_t address);
  25
  26 protected:
  27     virtual bool isFunctionCall(uint64_t address) = 0;
  28     virtual bool isJump(uint64_t address) = 0;
  29 };
  30
  31 #endif