Add improved instruction mapping APIs.

1 files changed, 14 insertions, 0 deletions

diff --git a/lowlevelil.cpp b/lowlevelil.cpp
index 13b465f9..ed9c7884 100644
--- a/lowlevelil.cpp
+++ b/lowlevelil.cpp
@@ -798,6 +798,20 @@ size_t LowLevelILFunction::GetMediumLevelILExprIndex(size_t expr) const
 }
 
 
+set<size_t> LowLevelILFunction::GetMediumLevelILExprIndexes(size_t expr) const
+{
+	size_t count;
+	size_t* exprs = BNGetMediumLevelILExprIndexes(m_object, expr, &count);
+
+	set<size_t> result;
+	for (size_t i = 0; i < count; i++)
+		result.insert(exprs[i]);
+
+	BNFreeILInstructionList(exprs);
+	return result;
+}
+
+
 size_t LowLevelILFunction::GetMappedMediumLevelILInstructionIndex(size_t instr) const
 {
 	return BNGetMappedMediumLevelILInstructionIndex(m_object, instr);