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 6cad001dcf74c68ea2f0113c078487c767af7e68..57833baaf1ccbf210b62867caed574248bfb40e3 100644 (file)
--- a/src/disassembler/Function.hxx
+++ b/src/disassembler/Function.hxx
@@ -2,22 +2,34 @@
 #define INCLUDE__Function_hxx
 
 #include "disassembler/BasicBlock.hxx"
+#include <map>
 
 class Function {
 public:
-       Function(const std::string& name) {
-               this->name = name;
+       Function(const std::string& name, uint64_t start_address)
+               : name(name)
+               , start_address(start_address) {
+       }
+
+       uint64_t getStartAddress() const {
+               return start_address;
        }
 
        std::string getName() const {
                return name;
        }
 
-       BasicBlock * getEntry();
+       void addBasicBlock(BasicBlock* block) {
+               _blocks.insert(std::make_pair(block->getStartAddress(), block));
+       }
 
+       std::map<uint64_t, BasicBlock*>& blocks() {
+               return _blocks;
+       }
 private:
        std::string name;
-       BasicBlock * start;
+       uint64_t start_address;
+       std::map<uint64_t, BasicBlock*> _blocks;
 };
 
 #endif