Mappings between low level IL and medium level IL

2 files changed, 96 insertions, 0 deletions

diff --git a/python/lowlevelil.py b/python/lowlevelil.py
index 40cb964c..462d4ade 100644
--- a/python/lowlevelil.py
+++ b/python/lowlevelil.py
@@ -25,6 +25,7 @@ import _binaryninjacore as core
 from .enums import LowLevelILOperation, LowLevelILFlagCondition, InstructionTextTokenType
 import function
 import basicblock
+import mediumlevelil
 
 
 class LowLevelILLabel(object):
@@ -250,6 +251,14 @@ class LowLevelILInstruction(object):
 			core.BNGetLowLevelILNonSSAExprIndex(self.function.handle, self.expr_index))
 
 	@property
+	def mapped_medium_level_il(self):
+		"""Gets the medium level IL expression corresponding to this expression"""
+		expr = self.function.get_mapped_medium_level_il_expr_index(self.expr_index)
+		if expr is None:
+			return None
+		return mediumlevelil.MediumLevelILInstruction(self.function.mapped_medium_level_il, expr)
+
+	@property
 	def value(self):
 		"""Value of expression using static data flow analysis (read-only)"""
 		value = core.BNGetLowLevelILExprValue(self.function.handle, self.expr_index)
@@ -381,6 +390,24 @@ class LowLevelILFunction(object):
 			return None
 		return LowLevelILFunction(self.arch, result, self.source_function)
 
+	@property
+	def medium_level_il(self):
+		"""Medium level IL for this low level IL."""
+		result = core.BNGetMediumLevelILForLowLevelIL(self.handle)
+		if not result:
+			return None
+		return mediumlevelil.MediumLevelILFunction(self.arch, result, self.source_function)
+
+	@property
+	def mapped_medium_level_il(self):
+		"""Medium level IL with mappings between low level IL and medium level IL. Unused stores are not removed.
+		Typically, this should only be used to answer queries on assembly or low level IL where the query is
+		easier to perform on medium level IL."""
+		result = core.BNGetMappedMediumLevelIL(self.handle)
+		if not result:
+			return None
+		return mediumlevelil.MediumLevelILFunction(self.arch, result, self.source_function)
+
 	def __setattr__(self, name, value):
 		try:
 			object.__setattr__(self, name, value)
@@ -1422,6 +1449,24 @@ class LowLevelILFunction(object):
 		core.BNFreeRegisterValue(value)
 		return result
 
+	def get_mapped_medium_level_il_instruction_index(self, instr):
+		med_il = self.mapped_medium_level_il
+		if med_il is None:
+			return None
+		result = core.BNGetMappedMediumLevelILInstructionIndex(self.handle, instr)
+		if result >= core.BNGetMediumLevelILInstructionCount(med_il.handle):
+			return None
+		return result
+
+	def get_mapped_medium_level_il_expr_index(self, expr):
+		med_il = self.mapped_medium_level_il
+		if med_il is None:
+			return None
+		result = core.BNGetMappedMediumLevelILExprIndex(self.handle, expr)
+		if result >= core.BNGetMediumLevelILExprCount(med_il.handle):
+			return None
+		return result
+
 
 class LowLevelILBasicBlock(basicblock.BasicBlock):
 	def __init__(self, view, handle, owner):
diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py
index 9f25ee97..00d7215f 100644
--- a/python/mediumlevelil.py
+++ b/python/mediumlevelil.py
@@ -25,6 +25,7 @@ import _binaryninjacore as core
 from .enums import MediumLevelILOperation, InstructionTextTokenType, ILVariableSourceType
 import function
 import basicblock
+import lowlevelil
 
 
 class MediumLevelILLabel(object):
@@ -258,6 +259,14 @@ class MediumLevelILInstruction(object):
 		core.BNFreeRegisterValue(value)
 		return result
 
+	@property
+	def low_level_il(self):
+		"""Low level IL form of this expression"""
+		expr = self.function.get_low_level_il_expr_index(self.expr_index)
+		if expr is None:
+			return None
+		return lowlevelil.LowLevelILInstruction(self.function.low_level_il.ssa_form, expr)
+
 	def __setattr__(self, name, value):
 		try:
 			object.__setattr__(self, name, value)
@@ -350,6 +359,14 @@ class MediumLevelILFunction(object):
 			return None
 		return MediumLevelILFunction(self.arch, result, self.source_function)
 
+	@property
+	def low_level_il(self):
+		"""Low level IL for this function"""
+		result = core.BNGetLowLevelILForMediumLevelIL(self.handle)
+		if not result:
+			return None
+		return lowlevelil.LowLevelILFunction(self.arch, result, self.source_function)
+
 	def __setattr__(self, name, value):
 		try:
 			object.__setattr__(self, name, value)
@@ -540,6 +557,40 @@ class MediumLevelILFunction(object):
 		core.BNFreeRegisterValue(value)
 		return result
 
+	def get_ssa_var_index_at_instruction(self, var, instr):
+		var_data = core.BNILVariable()
+		var_data.type = var.type
+		var_data.index = var.index
+		var_data.identifier = var.identifier
+		return core.BNGetMediumLevelILSSAVarIndexAtILInstruction(self.handle, var_data, instr)
+
+	def get_ssa_memory_index_at_instruction(self, instr):
+		return core.BNGetMediumLevelILSSAMemoryIndexAtILInstruction(self.handle, instr)
+
+	def get_low_level_il_instruction_index(self, instr):
+		low_il = self.low_level_il
+		if low_il is None:
+			return None
+		low_il = low_il.ssa_form
+		if low_il is None:
+			return None
+		result = core.BNGetLowLevelILInstructionIndex(self.handle, instr)
+		if result >= core.BNGetLowLevelILInstructionCount(low_il.handle):
+			return None
+		return result
+
+	def get_low_level_il_expr_index(self, expr):
+		low_il = self.low_level_il
+		if low_il is None:
+			return None
+		low_il = low_il.ssa_form
+		if low_il is None:
+			return None
+		result = core.BNGetLowLevelILExprIndex(self.handle, expr)
+		if result >= core.BNGetLowLevelILExprCount(low_il.handle):
+			return None
+		return result
+
 
 class MediumLevelILBasicBlock(basicblock.BasicBlock):
 	def __init__(self, view, handle, owner):