From 0e8987bb46b8aa33c5898d5effed25f1124e03dd Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Fri, 3 Jan 2020 21:51:50 -0500
Subject: Add types to HLIL expressions

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

(limited to 'highlevelil.cpp')

diff --git a/highlevelil.cpp b/highlevelil.cpp
index 679344ce..a3c9e5b2 100644
--- a/highlevelil.cpp
+++ b/highlevelil.cpp
@@ -295,6 +295,21 @@ vector<DisassemblyTextLine> HighLevelILFunction::GetExprText(const HighLevelILIn
 }
 
 
+Confidence<Ref<Type>> HighLevelILFunction::GetExprType(size_t expr)
+{
+	BNTypeWithConfidence result = BNGetHighLevelILExprType(m_object, expr);
+	if (!result.type)
+		return nullptr;
+	return Confidence<Ref<Type>>(new Type(result.type), result.confidence);
+}
+
+
+Confidence<Ref<Type>> HighLevelILFunction::GetExprType(const HighLevelILInstruction& expr)
+{
+	return GetExprType(expr.exprIndex);
+}
+
+
 void HighLevelILFunction::VisitAllExprs(
 	const function<bool(const HighLevelILInstruction& expr)>& func)
 {
-- 
cgit v1.3.1