src/disassembler/llvm/LLVMDisassembler.hxx

   1 #ifndef INCLUDE__LLVMDisassembler_hxx
   2 #define INCLUDE__LLVMDisassembler_hxx
   3
   4 #include <memory>
   5 #include <map>
   6 #include <log4cxx/logger.h>
   7
   8 #include "include_llvm.hxx"
   9
  10 #include "disassembler/Disassembler.hxx"
  11 #include "disassembler/BasicBlock.hxx"
  12 #include "disassembler/Function.hxx"
  13 #include "disassembler/llvm/LLVMBasicBlock.hxx"
  14 #include "disassembler/llvm/LLVMFunction.hxx"
  15
  16
  17 class LLVMDisassembler
  18     : public Disassembler {
  19 public:
  20     LLVMDisassembler(const std::string& filename);
  21     virtual ~LLVMDisassembler();
  22
  23         void getSymbols();
  24         uint64_t entryAddress();
  25
  26         void forEachFunction(std::function<void (uint64_t, Function*)> callback);
  27         void printEachInstruction(uint64_t start, uint64_t end,
  28                                                           std::function<void (uint8_t*, size_t, const std::string&)> fun);
  29
  30         BasicBlock * getBasicBlock(uint64_t address) {
  31                 return blocks[address];
  32         }
  33
  34     Function * disassembleFunctionAt(uint64_t address, const std::string& name = "");
  35
  36 protected:
  37     bool isFunctionCall(uint64_t address) {return false;}
  38     bool isJump(uint64_t address) {return false;}
  39
  40 private:
  41     // http://llvm.org/docs/doxygen/html/MCObjectDisassembler_8cpp_source.html +197
  42     void disassembleFunction(LLVMFunction* function);
  43     void splitBlocks();
  44     void disassemble();
  45
  46         void readSymbols();
  47         void readSections();
  48
  49     log4cxx::LoggerPtr logger;
  50         std::map<uint64_t, LLVMBasicBlock*> blocks;
  51         std::map<uint64_t, LLVMFunction*> functions;
  52
  53     llvm::Triple triple;
  54     std::shared_ptr<llvm::object::Binary> binary;
  55
  56
  57     const llvm::Target * target;
  58     llvm::object::ObjectFile * o;
  59
  60     std::unique_ptr<const llvm::MCRegisterInfo> MRI;
  61     std::unique_ptr<const llvm::MCAsmInfo> AsmInfo;
  62     std::unique_ptr<llvm::MCModule> Mod;
  63     std::unique_ptr<llvm::MCInstPrinter> IP;
  64     std::unique_ptr<llvm::MCDisassembler> DisAsm;
  65     std::unique_ptr<const llvm::MCObjectFileInfo> MOFI;
  66     std::unique_ptr<llvm::MCContext> Ctx;
  67     std::unique_ptr<const llvm::MCInstrAnalysis> MIA;
  68     std::unique_ptr<const llvm::MCSubtargetInfo> STI;
  69     std::unique_ptr<const llvm::MCInstrInfo> MII;
  70     std::unique_ptr<llvm::MCRelocationInfo> RelInfo;
  71     std::unique_ptr<llvm::MCSymbolizer> Symzer;
  72
  73     std::map<std::string, llvm::object::SectionRef> sections;
  74     std::map<std::string, llvm::object::SymbolRef> symbols;
  75 };
  76
  77 #endif