add Function::{reg_value_at, reg_value_after, reg_value_at_exit} methods

rust/src/function.rs

@@ -848,6 +848,60 @@ impl Function {
         let display = self.int_display_type(instr_addr, value, operand, Some(arch));
         (display, name)
     }
+
+    /// Get the value the provided string register address corresponding to the given virtual address
+    ///
+    /// * `addr` - virtual address of the instruction to query
+    /// * `reg` - string value of native register to query
+    /// * `arch` - (optional) Architecture for the given function
+    ///
+    /// # Example
+    /// ```no_run
+    /// # use binaryninja::architecture::{ArchitectureExt, Register};
+    /// # let fun: binaryninja::function::Function = todo!();
+    /// let reg = fun.arch().register_by_name("rdi").unwrap();
+    /// let value = fun.reg_value_at(0x400dbe, reg.id(), None);
+    /// ```
+    pub fn reg_value_at(
+        &self,
+        addr: u64,
+        reg: u32,
+        arch: Option<CoreArchitecture>,
+    ) -> RegisterValue {
+        let arch = arch.unwrap_or_else(|| self.arch());
+        let register = unsafe { BNGetRegisterValueAtInstruction(self.handle, arch.0, addr, reg) };
+        register.into()
+    }
+
+    /// Gets the value instruction address corresponding to the given virtual address
+    ///
+    /// * `addr` - virtual address of the instruction to query
+    /// * `reg` - string value of native register to query
+    /// * `arch` - (optional) Architecture for the given function
+    ///
+    /// # Example
+    /// ```no_run
+    /// # use binaryninja::architecture::{ArchitectureExt, Register};
+    /// # let fun: binaryninja::function::Function = todo!();
+    /// let reg = fun.arch().register_by_name("rdi").unwrap();
+    /// let value = fun.reg_value_after(0x400dbe, reg.id(), None);
+    /// ```
+    pub fn reg_value_after(
+        &self,
+        addr: u64,
+        reg: u32,
+        arch: Option<CoreArchitecture>,
+    ) -> RegisterValue {
+        let arch = arch.unwrap_or_else(|| self.arch());
+        let register =
+            unsafe { BNGetRegisterValueAfterInstruction(self.handle, arch.0, addr, reg) };
+        register.into()
+    }
+
+    pub fn reg_value_at_exit(&self, reg: u32) -> Conf<RegisterValue> {
+        let register = unsafe { BNGetFunctionRegisterValueAtExit(self.handle, reg) };
+        Conf::new(register.value.into(), register.confidence)
+    }
 }
 
 impl fmt::Debug for Function {