X-Git-Url: https://git.siccegge.de//index.cgi?p=frida%2Ffrida.git;a=blobdiff_plain;f=src%2Fdisassembler%2FBasicBlock.hxx;h=6d623d0f84116535962de5cd30c5ff14299fb55b;hp=c994d9ea02c77373f8c5f81a2ade0cf2516b9508;hb=9e283567ac56a6433e832d0fa38cf534a0cb8f9f;hpb=d84d4109b1f5bd5f87732b7e49860639d963a553

diff --git a/src/disassembler/BasicBlock.hxx b/src/disassembler/BasicBlock.hxx
index c994d9e..6d623d0 100644
--- a/src/disassembler/BasicBlock.hxx
+++ b/src/disassembler/BasicBlock.hxx
@@ -1,20 +1,31 @@
 #ifndef INCLUDE__BasicBlock_hxx
 #define INCLUDE__BasicBlock_hxx
 
+#include <cassert>
+
 class BasicBlock {
 public:
-    BasicBlock() {}
+	BasicBlock() {
+		next_blocks[0] = 0;
+		next_blocks[1] = 0;
+	}
+
+	uint64_t getStartAddress() const {
+		return start_address;
+	}
 
-    uint64_t getStartAddress() const {
-        return start_address;
-    }
+	uint64_t getEndAddress() const {
+		return end_address;
+	}
 
-    uint64_t getEndAddress() const {
-        return end_address;
-    }
+	uint64_t getNextBlock(size_t index) const {
+		assert(index < 2);
+		return next_blocks[index];
+	}
 
-	BasicBlock * const * getNextBlocks() const {
-		return next_blocks;
+	void setNextBlock(size_t index, uint64_t address) {
+		assert(index < 2);
+		next_blocks[index] = address;
 	}
 
 	void setStartAddress(uint64_t address) {
@@ -26,10 +37,10 @@ public:
 	}
 
 private:
-    uint64_t start_address;
-    uint64_t end_address;
+	uint64_t start_address;
+	uint64_t end_address;
 
-    BasicBlock * next_blocks[2];
+	uint64_t next_blocks[2];
 };
 
 #endif