From b15ea74b0b5e0afab9c935894eb39610408b6678 Mon Sep 17 00:00:00 2001 From: Rusty Wagner Date: Thu, 5 Apr 2018 17:34:55 -0400 Subject: Add SSA variable liveness API for MLIL --- binaryninjaapi.h | 1 + binaryninjacore.h | 2 ++ mediumlevelil.cpp | 6 ++++++ python/mediumlevelil.py | 14 ++++++++++++++ 4 files changed, 23 insertions(+) diff --git a/binaryninjaapi.h b/binaryninjaapi.h index 81b5105d..05c8c4e3 100644 --- a/binaryninjaapi.h +++ b/binaryninjaapi.h @@ -3230,6 +3230,7 @@ namespace BinaryNinja size_t GetSSAMemoryDefinition(size_t version) const; std::set GetSSAVarUses(const SSAVariable& var) const; std::set GetSSAMemoryUses(size_t version) const; + bool IsSSAVarLive(const SSAVariable& var) const; std::set GetVariableDefinitions(const Variable& var) const; std::set GetVariableUses(const Variable& var) const; diff --git a/binaryninjacore.h b/binaryninjacore.h index 69c5e61b..216301c0 100644 --- a/binaryninjacore.h +++ b/binaryninjacore.h @@ -2830,6 +2830,8 @@ extern "C" size_t version, size_t* count); BINARYNINJACOREAPI size_t* BNGetMediumLevelILSSAMemoryUses(BNMediumLevelILFunction* func, size_t version, size_t* count); + BINARYNINJACOREAPI bool BNIsMediumLevelILSSAVarLive(BNMediumLevelILFunction* func, + const BNVariable* var, size_t version); BINARYNINJACOREAPI size_t* BNGetMediumLevelILVariableDefinitions(BNMediumLevelILFunction* func, const BNVariable* var, size_t* count); diff --git a/mediumlevelil.cpp b/mediumlevelil.cpp index 0a28bd49..1795a54d 100644 --- a/mediumlevelil.cpp +++ b/mediumlevelil.cpp @@ -486,6 +486,12 @@ set MediumLevelILFunction::GetSSAMemoryUses(size_t version) const } +bool MediumLevelILFunction::IsSSAVarLive(const SSAVariable& var) const +{ + return BNIsMediumLevelILSSAVarLive(m_object, &var.var, var.version); +} + + set MediumLevelILFunction::GetVariableDefinitions(const Variable& var) const { size_t count; diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py index caf19e8e..d87b5bf7 100644 --- a/python/mediumlevelil.py +++ b/python/mediumlevelil.py @@ -851,6 +851,20 @@ class MediumLevelILFunction(object): core.BNFreeILInstructionList(instrs) return result + def is_ssa_var_live(self, ssa_var): + """ + ``is_ssa_var_live`` determines if ``ssa_var`` is live at any point in the function + + :param SSAVariable ssa_var: the SSA variable to query + :return: whether the variable is live at any point in the function + :rtype: bool + """ + var_data = core.BNVariable() + var_data.type = ssa_var.var.source_type + var_data.index = ssa_var.var.index + var_data.storage = ssa_var.var.storage + return core.BNIsMediumLevelILSSAVarLive(self.handle, var_data, ssa_var.version) + def get_var_definitions(self, var): count = ctypes.c_ulonglong() var_data = core.BNVariable() -- cgit v1.3.1