X-Git-Url: https://git.siccegge.de//index.cgi?p=frida%2Ffrida.git;a=blobdiff_plain;f=src%2Fdisassembler%2FFunction.hxx;h=57833baaf1ccbf210b62867caed574248bfb40e3;hp=ccca54c2198ec22b247ba27e264bac30e526908a;hb=9252262693432b33dbe4ffc60d79bbdc6fbb5f66;hpb=c55f9e2b5f5ff83817ab5187eb0122600a4c5cea

diff --git a/src/disassembler/Function.hxx b/src/disassembler/Function.hxx
index ccca54c..57833ba 100644
--- a/src/disassembler/Function.hxx
+++ b/src/disassembler/Function.hxx
@@ -2,6 +2,7 @@
 #define INCLUDE__Function_hxx
 
 #include "disassembler/BasicBlock.hxx"
+#include <map>
 
 class Function {
 public:
@@ -17,9 +18,18 @@ public:
 	std::string getName() const {
 		return name;
 	}
+
+	void addBasicBlock(BasicBlock* block) {
+		_blocks.insert(std::make_pair(block->getStartAddress(), block));
+	}
+
+	std::map<uint64_t, BasicBlock*>& blocks() {
+		return _blocks;
+	}
 private:
 	std::string name;
 	uint64_t start_address;
+	std::map<uint64_t, BasicBlock*> _blocks;
 };
 
 #endif