From 5841af2db8e8dcf4e0da0c438ac040c5fa90038b Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Tue, 21 Aug 2018 15:47:34 -0400
Subject: Add return hint MLIL instruction (used in intermediate stages, not
 emitted in final forms)

---
 mediumlevelilinstruction.cpp | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'mediumlevelilinstruction.cpp')

diff --git a/mediumlevelilinstruction.cpp b/mediumlevelilinstruction.cpp
index 9cf82b08..93587c34 100644
--- a/mediumlevelilinstruction.cpp
+++ b/mediumlevelilinstruction.cpp
@@ -122,6 +122,7 @@ unordered_map<BNMediumLevelILOperation, vector<MediumLevelILOperandUsage>>
 		{MLIL_ADDRESS_OF_FIELD, {SourceVariableMediumLevelOperandUsage, OffsetMediumLevelOperandUsage}},
 		{MLIL_JUMP, {DestExprMediumLevelOperandUsage}},
 		{MLIL_JUMP_TO, {DestExprMediumLevelOperandUsage, TargetListMediumLevelOperandUsage}},
+		{MLIL_RET_HINT, {DestExprMediumLevelOperandUsage}},
 		{MLIL_CALL, {OutputVariablesMediumLevelOperandUsage, DestExprMediumLevelOperandUsage,
 			ParameterExprsMediumLevelOperandUsage}},
 		{MLIL_CALL_UNTYPED, {OutputVariablesSubExprMediumLevelOperandUsage, DestExprMediumLevelOperandUsage,
@@ -1322,6 +1323,7 @@ void MediumLevelILInstruction::VisitExprs(const std::function<bool(const MediumL
 	case MLIL_BOOL_TO_INT:
 	case MLIL_JUMP:
 	case MLIL_JUMP_TO:
+	case MLIL_RET_HINT:
 	case MLIL_IF:
 	case MLIL_UNIMPL_MEM:
 	case MLIL_LOAD:
@@ -1591,6 +1593,7 @@ ExprId MediumLevelILInstruction::CopyTo(MediumLevelILFunction* dest,
 	case MLIL_LOW_PART:
 	case MLIL_BOOL_TO_INT:
 	case MLIL_JUMP:
+	case MLIL_RET_HINT:
 	case MLIL_UNIMPL_MEM:
 	case MLIL_FSQRT:
 	case MLIL_FNEG:
@@ -2497,6 +2500,12 @@ ExprId MediumLevelILFunction::JumpTo(ExprId dest, const vector<BNMediumLevelILLa
 }
 
 
+ExprId MediumLevelILFunction::ReturnHint(ExprId dest, const ILSourceLocation& loc)
+{
+	return AddExprWithLocation(MLIL_RET_HINT, loc, 0, dest);
+}
+
+
 ExprId MediumLevelILFunction::Call(const vector<Variable>& output, ExprId dest,
 	const vector<ExprId>& params, const ILSourceLocation& loc)
 {
-- 
cgit v1.3.1