From 68a9e96c98dbed4e4154a13ffec7b9c94876661a Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Mon, 27 Apr 2020 23:14:44 -0400
Subject: Add names for goto labels

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

(limited to 'highlevelil.cpp')

diff --git a/highlevelil.cpp b/highlevelil.cpp
index f97b11d3..264af7c3 100644
--- a/highlevelil.cpp
+++ b/highlevelil.cpp
@@ -460,3 +460,23 @@ Ref<FlowGraph> HighLevelILFunction::CreateFunctionGraph(DisassemblySettings* set
 	BNFlowGraph* graph = BNCreateHighLevelILFunctionGraph(m_object, settings ? settings->GetObject() : nullptr);
 	return new CoreFlowGraph(graph);
 }
+
+
+size_t HighLevelILFunction::GetExprIndexForLabel(uint64_t label)
+{
+	return BNGetHighLevelILExprIndexForLabel(m_object, label);
+}
+
+
+set<size_t> HighLevelILFunction::GetUsesForLabel(uint64_t label)
+{
+	size_t count;
+	size_t* uses = BNGetHighLevelILUsesForLabel(m_object, label, &count);
+
+	set<size_t> result;
+	for (size_t i = 0; i < count; i++)
+		result.insert(uses[i]);
+
+	BNFreeILInstructionList(uses);
+	return result;
+}
-- 
cgit v1.3.1