add MediumLevelILFunction::ssa_variable_definition

rust/src/mlil/function.rs

@@ -7,7 +7,7 @@ use crate::basicblock::BasicBlock;
 use crate::function::{Function, Location};
 use crate::rc::{Array, CoreArrayProvider, CoreArrayProviderInner, Ref, RefCountable};
 use crate::string::BnStrCompatible;
-use crate::types::{Conf, PossibleValueSet, Type, UserVariableValues, Variable};
+use crate::types::{Conf, PossibleValueSet, SSAVariable, Type, UserVariableValues, Variable};
 
 use super::{MediumLevelILBlock, MediumLevelILInstruction, MediumLevelILLiftedInstruction};
 
@@ -449,6 +449,18 @@ impl MediumLevelILFunction {
             )
         }
     }
+
+    /// Gets the instruction that contains the given SSA variable's definition.
+    ///
+    /// Since SSA variables can only be defined once, this will return the single instruction where that occurs.
+    /// For SSA variable version 0s, which don't have definitions, this will return None instead.
+    pub fn ssa_variable_definition(&self, var: SSAVariable) -> Option<MediumLevelILInstruction> {
+        let result = unsafe {
+            BNGetMediumLevelILSSAVarDefinition(self.handle, &var.variable.raw(), var.version)
+        };
+        (result < self.instruction_count())
+            .then(|| MediumLevelILInstruction::new(self.to_owned(), result))
+    }
 }
 
 impl ToOwned for MediumLevelILFunction {