1 files changed, 41 insertions, 0 deletions
diff --git a/rust/tests/high_level_il.rs b/rust/tests/high_level_il.rs
new file mode 100644
index 00000000..3a2b7209
--- /dev/null
+++ b/rust/tests/high_level_il.rs
@@ -0,0 +1,41 @@
+use binaryninja::binary_view::BinaryViewExt;
+use binaryninja::headless::Session;
+use binaryninja::high_level_il::{HighLevelILInstructionKind, HighLevelInstructionIndex};
+use rstest::*;
+use std::path::PathBuf;
+
+#[fixture]
+#[once]
+fn session() -> Session {
+    Session::new().expect("Failed to initialize session")
+}
+
+#[rstest]
+fn test_hlil_info(_session: &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 entry_function = view.entry_point_function().unwrap();
+    let hlil_function = entry_function.high_level_il(false).unwrap();
+    let hlil_basic_blocks = hlil_function.basic_blocks();
+    let mut hlil_basic_block_iter = hlil_basic_blocks.iter();
+    let first_basic_block = hlil_basic_block_iter.next().unwrap();
+    let mut hlil_instr_iter = first_basic_block.iter();
+
+    // NOTE: I guess this is right?
+    // 00025f10        (HLIL_BLOCK
+    // 00025f22            (HLIL_RET return (
+    // 00025f22                HLIL_CALL (HLIL_CONST_PTR.d __crt_interlocked_read_32)((HLIL_VAR.d arg1))))
+    // 00025f10        )
+    let instr_0 = hlil_instr_iter.next().unwrap();
+    assert_eq!(instr_0.expr_index, HighLevelInstructionIndex(5));
+    assert_eq!(instr_0.address, 0x00025f22);
+    println!("{:?}", instr_0.kind);
+    match instr_0.kind {
+        HighLevelILInstructionKind::Ret(op) => {
+            assert_eq!(op.first_src, 4);
+            assert_eq!(op.num_srcs, 1);
+        }
+        _ => panic!("Expected Ret"),
+    }
+}