add Function::{set_user_parameter_variables, set_auto_parameter_varia…

…bles} methods
Vector35 · May 14, 2024 · 479b585 · 479b585
1 parent 352e24b
commit 479b585
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/rust/src/function.rs b/rust/src/function.rs
@@ -764,6 +764,40 @@ impl Function {
         }
     }
 
+    pub fn set_user_parameter_variables<I>(&self, values: I, confidence: u8)
+    where
+        I: IntoIterator<Item = Variable>,
+    {
+        let mut vars: Box<[BNVariable]> = values.into_iter().map(|var| var.raw()).collect();
+        unsafe {
+            BNSetUserFunctionParameterVariables(
+                self.handle,
+                &mut BNParameterVariablesWithConfidence {
+                    vars: vars.as_mut_ptr(),
+                    count: vars.len(),
+                    confidence,
+                },
+            )
+        }
+    }
+
+    pub fn set_auto_parameter_variables<I>(&self, values: I, confidence: u8)
+    where
+        I: IntoIterator<Item = Variable>,
+    {
+        let mut vars: Box<[BNVariable]> = values.into_iter().map(|var| var.raw()).collect();
+        unsafe {
+            BNSetAutoFunctionParameterVariables(
+                self.handle,
+                &mut BNParameterVariablesWithConfidence {
+                    vars: vars.as_mut_ptr(),
+                    count: vars.len(),
+                    confidence,
+                },
+            )
+        }
+    }
+
     pub fn parameter_at(
         &self,
         addr: u64,