[Rust] Add `Variable::to_register` helper function

Makes it simpler to get the register from a variable (if it is a register variable)

1 files changed, 11 insertions, 0 deletions

diff --git a/rust/src/variable.rs b/rust/src/variable.rs
index 1921f950..dd7f4cf5 100644
--- a/rust/src/variable.rs
+++ b/rust/src/variable.rs
@@ -1,5 +1,6 @@
 #![allow(unused)]
 
+use crate::architecture::{Architecture, CoreArchitecture, CoreRegister, RegisterId};
 use crate::confidence::Conf;
 use crate::function::{Function, Location};
 use crate::rc::{CoreArrayProvider, CoreArrayProviderInner, Ref};
@@ -387,6 +388,16 @@ impl Variable {
         let raw = BNVariable::from(*self);
         unsafe { BNToVariableIdentifier(&raw) }
     }
+
+    pub fn to_register(&self, arch: CoreArchitecture) -> Option<CoreRegister> {
+        match self.ty {
+            VariableSourceType::RegisterVariableSourceType => {
+                arch.register_from_id(RegisterId(self.storage as u32))
+            }
+            VariableSourceType::StackVariableSourceType => None,
+            VariableSourceType::FlagVariableSourceType => None,
+        }
+    }
 }
 
 impl From<BNVariable> for Variable {