From 754fb4d12a85cdf8c791725002c9a1e78894d8c0 Mon Sep 17 00:00:00 2001
From: Brian Potchik <brian@vector35.com>
Date: Wed, 4 Aug 2021 10:48:57 -0400
Subject: Expose some BasicBlock APIs.

---
 basicblock.cpp | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'basicblock.cpp')

diff --git a/basicblock.cpp b/basicblock.cpp
index 3e3fccef..1675afdf 100644
--- a/basicblock.cpp
+++ b/basicblock.cpp
@@ -200,6 +200,12 @@ bool BasicBlock::CanExit() const
 }
 
 
+void BasicBlock::SetCanExit(bool value)
+{
+	BNBasicBlockSetCanExit(m_object, value);
+}
+
+
 set<Ref<BasicBlock>> BasicBlock::GetDominators(bool post) const
 {
 	size_t count;
@@ -481,3 +487,12 @@ bool BasicBlock::GetInstructionContainingAddress(uint64_t addr, uint64_t* start)
 {
 	return BNGetBasicBlockInstructionContainingAddress(m_object, addr, start);
 }
+
+
+Ref<BasicBlock> BasicBlock::GetSourceBlock() const
+{
+	BNBasicBlock* block = BNGetBasicBlockSourceBlock(m_object);
+	if (!block)
+		return nullptr;
+	return new BasicBlock(block);
+}
-- 
cgit v1.3.1