[Rust] Add LowLevelILSSARegisterKind::id helper

The partial register id is the "main" id so we return that, whereas the full register id is not.

Author: emesare

rust/src/low_level_il.rs

@@ -189,6 +189,13 @@ impl<R: ArchReg> LowLevelILSSARegisterKind<R> {
             | LowLevelILSSARegisterKind::Partial { version, .. } => version,
         }
     }
+
+    pub fn id(&self) -> RegisterId {
+        match *self {
+            LowLevelILSSARegisterKind::Full { kind, .. } => kind.id(),
+            LowLevelILSSARegisterKind::Partial { partial_reg, .. } => partial_reg.id(),
+        }
+    }
 }
 
 #[derive(Copy, Clone, Debug)]

rust/src/low_level_il/function.rs

@@ -267,15 +267,11 @@ impl<M: FunctionMutability> LowLevelILFunction<M, SSA> {
         reg: LowLevelILSSARegisterKind<R>,
     ) -> Vec<LowLevelILInstruction<'_, M, SSA>> {
         use binaryninjacore_sys::BNGetLowLevelILSSARegisterUses;
-        let register_id = match reg {
-            LowLevelILSSARegisterKind::Full { kind, .. } => kind.id(),
-            LowLevelILSSARegisterKind::Partial { partial_reg, .. } => partial_reg.id(),
-        };
         let mut count = 0;
         let instrs = unsafe {
             BNGetLowLevelILSSARegisterUses(
                 self.handle,
-                register_id.into(),
+                reg.id().into(),
                 reg.version() as usize,
                 &mut count,
             )
@@ -295,14 +291,10 @@ impl<M: FunctionMutability> LowLevelILFunction<M, SSA> {
         reg: &LowLevelILSSARegisterKind<R>,
     ) -> Option<LowLevelILInstruction<'_, M, SSA>> {
         use binaryninjacore_sys::BNGetLowLevelILSSARegisterDefinition;
-        let register_id = match reg {
-            LowLevelILSSARegisterKind::Full { kind, .. } => kind.id(),
-            LowLevelILSSARegisterKind::Partial { partial_reg, .. } => partial_reg.id(),
-        };
         let instr_idx = unsafe {
             BNGetLowLevelILSSARegisterDefinition(
                 self.handle,
-                register_id.into(),
+                reg.id().into(),
                 reg.version() as usize,
             )
         };
@@ -315,12 +307,8 @@ impl<M: FunctionMutability> LowLevelILFunction<M, SSA> {
         &self,
         reg: &LowLevelILSSARegisterKind<R>,
     ) -> Option<RegisterValue> {
-        let register_id = match reg {
-            LowLevelILSSARegisterKind::Full { kind, .. } => kind.id(),
-            LowLevelILSSARegisterKind::Partial { partial_reg, .. } => partial_reg.id(),
-        };
         let value = unsafe {
-            BNGetLowLevelILSSARegisterValue(self.handle, register_id.into(), reg.version() as usize)
+            BNGetLowLevelILSSARegisterValue(self.handle, reg.id().into(), reg.version() as usize)
         };
         if value.state == BNRegisterValueType::UndeterminedValue {
             return None;