Split blocks before finishing function

[frida/frida.git] / src / disassembler / Function.hxx

diff --git a/src/disassembler/Function.hxx b/src/disassembler/Function.hxx
index ccca54c2198ec22b247ba27e264bac30e526908a..57833baaf1ccbf210b62867caed574248bfb40e3 100644 (file)
--- 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