X-Git-Url: https://git.siccegge.de//index.cgi?p=frida%2Ffrida.git;a=blobdiff_plain;f=src%2Fdisassembler%2Fllvm%2FLLVMDisassembler.hxx;h=2713ac6f32f561df00e252a53b0678c4fe6cc4d9;hp=d26bb126f8dee24fc592afc058a5cfd6410c7c62;hb=bc36ee5d21eac1d9bd1ab1c2e7003101dbbec604;hpb=231dae075375e7d57982f5107b86294fbe726b33

diff --git a/src/disassembler/llvm/LLVMDisassembler.hxx b/src/disassembler/llvm/LLVMDisassembler.hxx
index d26bb12..2713ac6 100644
--- a/src/disassembler/llvm/LLVMDisassembler.hxx
+++ b/src/disassembler/llvm/LLVMDisassembler.hxx
@@ -8,25 +8,29 @@
 #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;
+class LLVMBasicBlock;
 
+Disassembler * createLLVMDisassembler(const std::string& filename, InformationManager* manager);
+
+template <typename ELFT>
 class LLVMDisassembler
 	: public Disassembler {
 public:
-	LLVMDisassembler(const std::string& filename, InformationManager* manager);
+	LLVMDisassembler(const std::string& filename, InformationManager* manager,
+	                 llvm::object::ELFObjectFile<ELFT>* file = NULL);
 	virtual ~LLVMDisassembler();
 
 	void start();
 	void getSymbols() {}
-	uint64_t entryAddress() {}
+	uint64_t entryAddress() {return _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);
+	                          std::function<void (uint8_t*, size_t, const std::string&,
+	                                              const std::string&)> fun);
 
 	BasicBlock * getBasicBlock(uint64_t address) {
 		return blocks[address];
@@ -39,23 +43,26 @@ protected:
 	bool isJump(uint64_t address) {return false;}
 
 private:
+	typedef llvm::object::ELFFile<ELFT> ELFO;
+
 	// 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();
 
 	void readSymbols();
 	void readSections();
+	void readDynamicSymbols();
 
 	log4cxx::LoggerPtr logger;
 	std::map<uint64_t, LLVMBasicBlock*> blocks;
-	std::map<uint64_t, LLVMFunction*> functions;
+	std::map<uint64_t, Function*> functions;
 
 	llvm::Triple triple;
 	std::shared_ptr<llvm::object::Binary> binary;
 
 	const llvm::Target * target;
-	llvm::object::ObjectFile * o;
+	llvm::object::ELFObjectFile<ELFT> * o;
 
 	std::unique_ptr<const llvm::MCRegisterInfo> MRI;
 	std::unique_ptr<const llvm::MCAsmInfo> AsmInfo;
@@ -73,6 +80,7 @@ private:
 	std::map<std::string, llvm::object::SectionRef> sections;
 	std::map<std::string, llvm::object::SymbolRef> symbols;
 	InformationManager * manager;
+	uint64_t _entryAddress;
 };
 
 #endif