Add SetExprType API to MLIL/HLIL

Add get/set_expr_type to MLIL/HLIL Python APi

1 files changed, 16 insertions, 0 deletions

diff --git a/highlevelil.cpp b/highlevelil.cpp
index d0b3b4e2..7d3789d9 100644
--- a/highlevelil.cpp
+++ b/highlevelil.cpp
@@ -501,6 +501,22 @@ Confidence<Ref<Type>> HighLevelILFunction::GetExprType(const HighLevelILInstruct
 }
 
 
+void HighLevelILFunction::SetExprType(size_t expr, const Confidence<Ref<Type>>& type)
+{
+	BNTypeWithConfidence tc;
+	tc.type = type->GetObject();
+	tc.confidence = type.GetConfidence();
+	BNSetHighLevelILExprType(m_object, expr, &tc);
+}
+
+
+void HighLevelILFunction::SetExprType(const BinaryNinja::HighLevelILInstruction& expr,
+										const Confidence<Ref<BinaryNinja::Type>>& type)
+{
+	SetExprType(expr.exprIndex, type);
+}
+
+
 void HighLevelILFunction::VisitAllExprs(const function<bool(const HighLevelILInstruction& expr)>& func)
 {
 	GetRootExpr().VisitExprs([&](const HighLevelILInstruction& expr) { return func(expr); });