Add types to HLIL expressions

1 files changed, 15 insertions, 0 deletions

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)
 {