From 8ba8388e12ab32ad937c4514f7a907d2530f1ef8 Mon Sep 17 00:00:00 2001
From: Mason Reed <mason@vector35.com>
Date: Sat, 31 May 2025 10:45:26 -0400
Subject: [Rust] Retrieve architecture if missing from location passed to
 `MediumLevelILFunction::instruction_from_index`

Fixes https://github.com/Vector35/binaryninja-api/issues/6876
---
 rust/tests/medium_level_il.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'rust/tests')

diff --git a/rust/tests/medium_level_il.rs b/rust/tests/medium_level_il.rs
index de79a47d..a921965f 100644
--- a/rust/tests/medium_level_il.rs
+++ b/rust/tests/medium_level_il.rs
@@ -164,9 +164,14 @@ fn test_mlil_possible_values() {
         _ => panic!("Expected SetVar"),
     }
 
+    // As an aside, we also test to make sure `instruction_index_at` works.
+    let instr_21_idx = mlil_function
+        .instruction_index_at(image_base + 0x0002af9a)
+        .expect("Failed to get instruction index");
+
     // 21 @ 0002af9a  (MLIL_RET return (MLIL_VAR.d eax_2))
     let instr_21 = mlil_function
-        .instruction_from_index(MediumLevelInstructionIndex(21))
+        .instruction_from_index(instr_21_idx)
         .expect("Failed to get instruction");
     let lifted_instr_21 = instr_21.lift();
     match lifted_instr_21.kind {
-- 
cgit v1.3.1