Implement Rust DisassemblyTextRenderer

1 files changed, 46 insertions, 0 deletions

diff --git a/rust/examples/disassemble.rs b/rust/examples/disassemble.rs
new file mode 100644
index 00000000..3f422181
--- /dev/null
+++ b/rust/examples/disassemble.rs
@@ -0,0 +1,46 @@
+use binaryninja::binary_view::{BinaryViewBase, BinaryViewExt};
+use binaryninja::disassembly::{DisassemblyOption, DisassemblySettings, DisassemblyTextRenderer};
+use binaryninja::function::Function;
+
+fn disassemble(func: &Function) {
+    let settings = DisassemblySettings::new();
+    settings.set_option(DisassemblyOption::ShowAddress, false);
+    settings.set_option(DisassemblyOption::WaitForIL, true);
+    settings.set_option(DisassemblyOption::IndentHLILBody, false);
+    settings.set_option(DisassemblyOption::ShowCollapseIndicators, false);
+    settings.set_option(DisassemblyOption::ShowFunctionHeader, false);
+
+    let text_renderer = DisassemblyTextRenderer::from_function(func, Some(&settings));
+    for basic_block in &func.basic_blocks() {
+        for instr_addr in basic_block.iter() {
+            // NOTE: If you want the annotations as well you can call text_renderer.disassembly_text
+            if let Some((text, len)) = text_renderer.instruction_text(instr_addr) {
+                // TODO: This only ever appears to return a single string?
+                let text_string: Vec<_> = text.iter().map(|t| t.to_string()).collect();
+                println!("{}", text_string.join(""));
+            }
+        }
+    }
+}
+
+pub fn main() {
+    let filename = std::env::args().nth(1).expect("No filename provided");
+
+    println!("Starting session...");
+    // This loads all the core architecture, platform, etc plugins
+    let headless_session =
+        binaryninja::headless::Session::new().expect("Failed to initialize session");
+
+    println!("Loading binary...");
+    let bv = headless_session
+        .load(&filename)
+        .expect("Couldn't open file!");
+
+    println!("Filename:  `{}`", bv.file().filename());
+    println!("File size: `{:#x}`", bv.len());
+    println!("Function count: {}", bv.functions().len());
+
+    for func in &bv.functions() {
+        disassemble(func.as_ref());
+    }
+}