#include "include_llvm.hxx"
#include "disassembler/Disassembler.hxx"
-#include "disassembler/BasicBlock.hxx"
-#include "disassembler/Function.hxx"
-#include "disassembler/llvm/LLVMBasicBlock.hxx"
-#include "disassembler/llvm/LLVMFunction.hxx"
+class Function;
+class BasicBlock;
+Disassembler * createLLVMDisassembler(const std::string& filename, InformationManager* manager);
+
+template <typename ObjectType>
class LLVMDisassembler
: public Disassembler {
public:
- LLVMDisassembler(const std::string& filename, InformationManager* manager);
+ LLVMDisassembler(const std::string& filename, InformationManager* manager,
+ llvm::object::ObjectFile* file = NULL);
virtual ~LLVMDisassembler();
void start();
void getSymbols() {}
- uint64_t entryAddress() {}
-
- void forEachFunction(std::function<void (uint64_t, Function*)> callback);
- void printEachInstruction(uint64_t start, uint64_t end,
- std::function<void (uint8_t*, size_t, const std::string&)> fun);
-
- BasicBlock * getBasicBlock(uint64_t address) {
- return blocks[address];
- }
+ uint64_t entryAddress();
Function * disassembleFunctionAt(uint64_t address, const std::string& name = "");
-
-protected:
- bool isFunctionCall(uint64_t address) {return false;}
- bool isJump(uint64_t address) {return false;}
+ std::vector<Instruction> getInstructions(const BasicBlock* block);
private:
// http://llvm.org/docs/doxygen/html/MCObjectDisassembler_8cpp_source.html +197
- void disassembleFunction(LLVMFunction* function);
- void splitBlocks();
+ void disassembleFunction(Function* function);
+ void splitBlocks(Function* fun);
void disassemble();
+ llvm::object::SectionRef getTextSection();
void readSymbols();
void readSections();
+ void readDynamicSymbols();
log4cxx::LoggerPtr logger;
- std::map<uint64_t, LLVMBasicBlock*> blocks;
- std::map<uint64_t, LLVMFunction*> functions;
llvm::Triple triple;
std::shared_ptr<llvm::object::Binary> binary;
std::unique_ptr<const llvm::MCRegisterInfo> MRI;
std::unique_ptr<const llvm::MCAsmInfo> AsmInfo;
- std::unique_ptr<llvm::MCModule> Mod;
+// std::unique_ptr<llvm::MCModule> Mod;
std::unique_ptr<llvm::MCInstPrinter> IP;
std::unique_ptr<llvm::MCDisassembler> DisAsm;
std::unique_ptr<const llvm::MCObjectFileInfo> MOFI;
std::map<std::string, llvm::object::SectionRef> sections;
std::map<std::string, llvm::object::SymbolRef> symbols;
InformationManager * manager;
+ uint64_t _entryAddress;
};
#endif