From 42cb99079d4ff0a5aa49e7730bb73500d13f35dd Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Mon, 13 Mar 2017 23:03:41 -0400
Subject: Branch dependence APIs for path sensitive analysis

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

(limited to 'mediumlevelil.cpp')

diff --git a/mediumlevelil.cpp b/mediumlevelil.cpp
index 04b1cffd..dcd1fe8e 100644
--- a/mediumlevelil.cpp
+++ b/mediumlevelil.cpp
@@ -456,6 +456,26 @@ size_t MediumLevelILFunction::GetSSAMemoryIndexAtInstruction(size_t instr) const
 }
 
 
+BNILBranchDependence MediumLevelILFunction::GetBranchDependenceAtInstruction(size_t curInstr, size_t branchInstr) const
+{
+	return BNGetMediumLevelILBranchDependence(m_object, curInstr, branchInstr);
+}
+
+
+map<size_t, BNILBranchDependence> MediumLevelILFunction::GetAllBranchDependenceAtInstruction(size_t instr) const
+{
+	size_t count;
+	BNILBranchInstructionAndDependence* deps = BNGetAllMediumLevelILBranchDependence(m_object, instr, &count);
+
+	map<size_t, BNILBranchDependence> result;
+	for (size_t i = 0; i < count; i++)
+		result[deps[i].branch] = deps[i].dependence;
+
+	BNFreeILBranchDependenceList(deps);
+	return result;
+}
+
+
 Ref<LowLevelILFunction> MediumLevelILFunction::GetLowLevelIL() const
 {
 	BNLowLevelILFunction* func = BNGetLowLevelILForMediumLevelIL(m_object);
-- 
cgit v1.3.1