From cecb2af5e8ec590500379e4b103531649de4e5a2 Mon Sep 17 00:00:00 2001
From: Ryan Snyder <ryan@vector35.com>
Date: Mon, 20 Aug 2018 18:58:00 -0400
Subject: api: add methods to get basic block objects from IL instrs

---
 mediumlevelil.cpp | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'mediumlevelil.cpp')

diff --git a/mediumlevelil.cpp b/mediumlevelil.cpp
index 1795a54d..cac45dd6 100644
--- a/mediumlevelil.cpp
+++ b/mediumlevelil.cpp
@@ -404,6 +404,15 @@ vector<Ref<BasicBlock>> MediumLevelILFunction::GetBasicBlocks() const
 }
 
 
+Ref<BasicBlock> MediumLevelILFunction::GetBasicBlockForInstruction(size_t i) const
+{
+	BNBasicBlock* block = BNGetMediumLevelILBasicBlockForInstruction(m_object, i);
+	if (!block)
+		return nullptr;
+	return new BasicBlock(block);
+}
+
+
 Ref<MediumLevelILFunction> MediumLevelILFunction::GetSSAForm() const
 {
 	BNMediumLevelILFunction* func = BNGetMediumLevelILSSAForm(m_object);
-- 
cgit v1.3.1