From 448ffe2842b2b1e0d21f905403489d3f865672d8 Mon Sep 17 00:00:00 2001
From: Brian Potchik <brian@vector35.com>
Date: Mon, 4 Dec 2017 13:12:06 -0500
Subject: Add MediumLevelIL Function Recognizer.

---
 functionrecognizer.cpp | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'functionrecognizer.cpp')

diff --git a/functionrecognizer.cpp b/functionrecognizer.cpp
index 32c7245c..34f1db80 100644
--- a/functionrecognizer.cpp
+++ b/functionrecognizer.cpp
@@ -38,11 +38,22 @@ bool FunctionRecognizer::RecognizeLowLevelILCallback(void* ctxt, BNBinaryView* d
 }
 
 
+bool FunctionRecognizer::RecognizeMediumLevelILCallback(void* ctxt, BNBinaryView* data, BNFunction* func, BNMediumLevelILFunction* il)
+{
+	FunctionRecognizer* recog = (FunctionRecognizer*)ctxt;
+	Ref<BinaryView> dataObj = new BinaryView(BNNewViewReference(data));
+	Ref<Function> funcObj = new Function(BNNewFunctionReference(func));
+	Ref<MediumLevelILFunction> ilObj = new MediumLevelILFunction(BNNewMediumLevelILFunctionReference(il));
+	return recog->RecognizeMediumLevelIL(dataObj, funcObj, ilObj);
+}
+
+
 void FunctionRecognizer::RegisterGlobalRecognizer(FunctionRecognizer* recog)
 {
 	BNFunctionRecognizer reg;
 	reg.context = recog;
 	reg.recognizeLowLevelIL = RecognizeLowLevelILCallback;
+	reg.recognizeMediumLevelIL = RecognizeMediumLevelILCallback;
 	BNRegisterGlobalFunctionRecognizer(&reg);
 }
 
@@ -52,6 +63,7 @@ void FunctionRecognizer::RegisterArchitectureFunctionRecognizer(Architecture* ar
 	BNFunctionRecognizer reg;
 	reg.context = recog;
 	reg.recognizeLowLevelIL = RecognizeLowLevelILCallback;
+	reg.recognizeMediumLevelIL = RecognizeMediumLevelILCallback;
 	BNRegisterArchitectureFunctionRecognizer(arch->GetObject(), &reg);
 }
 
@@ -60,3 +72,9 @@ bool FunctionRecognizer::RecognizeLowLevelIL(BinaryView*, Function*, LowLevelILF
 {
 	return false;
 }
+
+
+bool FunctionRecognizer::RecognizeMediumLevelIL(BinaryView*, Function*, MediumLevelILFunction*)
+{
+	return false;
+}
-- 
cgit v1.3.1