From 032080750a88a1a437784b8c95cb1747aa907d0c Mon Sep 17 00:00:00 2001
From: Mason Reed <mason@vector35.com>
Date: Fri, 30 May 2025 16:42:48 -0400
Subject: [Rust] Finish out the `PossibleValueSet` implementation to include
 range and lists of values

This requires us to allocate on the rust side of the API, this is quite easy to mess up but the usage of it is limited.
---
 rust/tests/medium_level_il.rs | 72 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 72 insertions(+)

(limited to 'rust/tests')

diff --git a/rust/tests/medium_level_il.rs b/rust/tests/medium_level_il.rs
index 7d4d8af1..de79a47d 100644
--- a/rust/tests/medium_level_il.rs
+++ b/rust/tests/medium_level_il.rs
@@ -4,6 +4,7 @@ use binaryninja::medium_level_il::{
     MediumLevelExpressionIndex, MediumLevelILInstructionKind, MediumLevelILLiftedInstructionKind,
     MediumLevelInstructionIndex,
 };
+use binaryninja::variable::{PossibleValueSet, ValueRange};
 use std::path::PathBuf;
 
 #[test]
@@ -117,3 +118,74 @@ fn test_mlil_basic_blocks() {
         }
     }
 }
+
+#[test]
+fn test_mlil_possible_values() {
+    let _session = Session::new().expect("Failed to initialize session");
+    let out_dir = env!("OUT_DIR").parse::<PathBuf>().unwrap();
+    let view = binaryninja::load(out_dir.join("atox.obj")).expect("Failed to create view");
+    let image_base = view.original_image_base();
+
+    let functions = view.functions_containing(image_base + 0x0002af9a);
+    assert_eq!(functions.len(), 1);
+    let function = functions.get(0);
+    let mlil_function = function.medium_level_il().unwrap();
+
+    // 0 @ 0002af40  (MLIL_SET_VAR.d edi_1 = (MLIL_VAR.d edi))
+    let instr_0 = mlil_function
+        .instruction_from_index(MediumLevelInstructionIndex(0))
+        .expect("Failed to get instruction");
+    let lifted_instr_0 = instr_0.lift();
+    match lifted_instr_0.kind {
+        MediumLevelILLiftedInstructionKind::SetVar(op) => match op.src.kind {
+            MediumLevelILLiftedInstructionKind::Var(var) => {
+                let var = var.src;
+                let pvs_value = PossibleValueSet::SignedRangeValue {
+                    value: 5,
+                    ranges: vec![ValueRange {
+                        start: -5,
+                        end: 0,
+                        step: 1,
+                    }],
+                };
+                mlil_function
+                    .set_user_var_value(&var, op.src.address, pvs_value.clone(), false)
+                    .expect("Failed to set user var value");
+                let var_values = mlil_function.user_var_values();
+                assert_eq!(var_values.len(), 1);
+                let var_value = var_values.get(0);
+                assert_eq!(var_value.value, pvs_value);
+                assert_eq!(var_value.def_site.addr, op.src.address);
+                assert_eq!(var_value.variable, var);
+                assert_eq!(var_value.after, false);
+            }
+            _ => panic!("Expected Var"),
+        },
+        _ => panic!("Expected SetVar"),
+    }
+
+    // 21 @ 0002af9a  (MLIL_RET return (MLIL_VAR.d eax_2))
+    let instr_21 = mlil_function
+        .instruction_from_index(MediumLevelInstructionIndex(21))
+        .expect("Failed to get instruction");
+    let lifted_instr_21 = instr_21.lift();
+    match lifted_instr_21.kind {
+        MediumLevelILLiftedInstructionKind::Ret(ret) => {
+            // This should be the eax variable, with the signed range.
+            let eax_var_lifted = ret.src.get(0).unwrap();
+            // TODO: I really dislike that we have lifted and unlifted versions of the same thing.
+            let eax_var = mlil_function
+                .instruction_from_expr_index(eax_var_lifted.expr_index)
+                .unwrap();
+            let eax_var_pvs = eax_var.possible_values();
+            match eax_var_pvs {
+                PossibleValueSet::SignedRangeValue { value, ranges } => {
+                    assert_eq!(value, -48);
+                    assert_eq!(ranges.len(), 2);
+                }
+                _ => panic!("Expected SignedRangeValue"),
+            }
+        }
+        _ => panic!("Expected Ret"),
+    }
+}
-- 
cgit v1.3.1