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
  12 class Function;
  13 class BasicBlock;
  14
  15 Disassembler * createLLVMDisassembler(const std::string& filename, InformationManager* manager);
  16
  17 template <typename ObjectType>
  18 class LLVMDisassembler
  19         : public Disassembler {
  20 public:
  21         LLVMDisassembler(const std::string& filename, InformationManager* manager,
  22                          llvm::object::ObjectFile* file = NULL);
  23         virtual ~LLVMDisassembler();
  24
  25         void start();
  26         void getSymbols() {}
  27         uint64_t entryAddress();
  28
  29         void printEachInstruction(uint64_t start, uint64_t end,
  30                                   std::function<void (uint8_t*, size_t, const std::string&,
  31                                                       const std::string&)> fun);
  32
  33         Function * disassembleFunctionAt(uint64_t address, const std::string& name = "");
  34
  35 private:
  36         // http://llvm.org/docs/doxygen/html/MCObjectDisassembler_8cpp_source.html +197
  37         void disassembleFunction(Function* function);
  38         void splitBlocks(Function* fun);
  39         void disassemble();
  40         llvm::object::SectionRef getTextSection();
  41
  42         void readSymbols();
  43         void readSections();
  44         void readDynamicSymbols();
  45
  46         log4cxx::LoggerPtr logger;
  47
  48         llvm::Triple triple;
  49         std::shared_ptr<llvm::object::Binary> binary;
  50
  51         const llvm::Target * target;
  52         llvm::object::ObjectFile * o;
  53
  54         std::unique_ptr<const llvm::MCRegisterInfo> MRI;
  55         std::unique_ptr<const llvm::MCAsmInfo> AsmInfo;
  56         std::unique_ptr<llvm::MCModule> Mod;
  57         std::unique_ptr<llvm::MCInstPrinter> IP;
  58         std::unique_ptr<llvm::MCDisassembler> DisAsm;
  59         std::unique_ptr<const llvm::MCObjectFileInfo> MOFI;
  60         std::unique_ptr<llvm::MCContext> Ctx;
  61         std::unique_ptr<const llvm::MCInstrAnalysis> MIA;
  62         std::unique_ptr<const llvm::MCSubtargetInfo> STI;
  63         std::unique_ptr<const llvm::MCInstrInfo> MII;
  64         std::unique_ptr<llvm::MCRelocationInfo> RelInfo;
  65         std::unique_ptr<llvm::MCSymbolizer> Symzer;
  66
  67         std::map<std::string, llvm::object::SectionRef> sections;
  68         std::map<std::string, llvm::object::SymbolRef> symbols;
  69         InformationManager * manager;
  70         uint64_t _entryAddress;
  71 };
  72
  73 #endif