X-Git-Url: https://git.siccegge.de//index.cgi?p=frida%2Ffrida.git;a=blobdiff_plain;f=src%2Fdisassembler%2Fllvm%2FLLVMDisassembler.hxx;h=42024ee7c4998839f796c7d55ea43ea45ccd8a2b;hp=4f58a0eb8450078758d5dd6486138cc8d500afda;hb=f5b948bf38d8757f4e9a0d7af4edaa4a6fffec27;hpb=d84d4109b1f5bd5f87732b7e49860639d963a553

diff --git a/src/disassembler/llvm/LLVMDisassembler.hxx b/src/disassembler/llvm/LLVMDisassembler.hxx
index 4f58a0e..42024ee 100644
--- a/src/disassembler/llvm/LLVMDisassembler.hxx
+++ b/src/disassembler/llvm/LLVMDisassembler.hxx
@@ -8,21 +8,22 @@
 #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 LLVMDisassembler
     : public Disassembler {
 public:
     LLVMDisassembler(const std::string& filename);
-    virtual ~LLVMDisassembler() {};
+    virtual ~LLVMDisassembler();
 
 	void getSymbols();
 	uint64_t entryAddress();
 
-    void forEachInstruction(const std::string& name,
-							std::function<void (long, std::string, std::string)> callback)
-		{}
+	void forEachFunction(std::function<void (uint64_t, Function*)> callback);
 
 	BasicBlock * generateControlFlowGraph(const std::string& name);
 	BasicBlock * generateControlFlowGraph(uint64_t address);
@@ -38,7 +39,8 @@ private:
 	void readSections();
 
     log4cxx::LoggerPtr logger;
-	std::map<uint8_t, LLVMBasicBlock*> blocks;
+	std::map<uint64_t, LLVMBasicBlock*> blocks;
+	std::map<uint64_t, LLVMFunction*> functions;
 
     llvm::Triple triple;
     std::shared_ptr<llvm::object::Binary> binary;