pulley: Add a simple disassembler example

pulley/Cargo.toml

@@ -20,6 +20,8 @@ sptr = { workspace = true }
 
 [dev-dependencies]
 env_logger = { workspace = true }
+object = { workspace = true, features = ['std'] }
+anyhow = { workspace = true, features = ['std'] }
 
 [features]
 std = []
@@ -31,3 +33,7 @@ interp = ["decode", "encode"]
 
 [package.metadata.docs.rs]
 all-features = true
+
+[[example]]
+name = "objdump"
+required-features = ["disas"]

pulley/examples/objdump.rs

@@ -0,0 +1,48 @@
+//! Small helper utility to disassemble `*.cwasm` files produced by Wasmtime.
+//!
+//! Run with:
+//!
+//!     cargo run --example objdump -F disas -p pulley-interpreter foo.cwasm
+
+use anyhow::{bail, Result};
+use object::{File, Object as _, ObjectSection, ObjectSymbol, SectionKind, SymbolKind};
+use pulley_interpreter::decode::Decoder;
+use pulley_interpreter::disas::Disassembler;
+
+fn main() -> Result<()> {
+    let cwasm = std::fs::read(std::env::args().nth(1).unwrap())?;
+
+    let image = File::parse(&cwasm[..])?;
+
+    let text = match image.sections().find(|s| s.kind() == SectionKind::Text) {
+        Some(section) => section.data()?,
+        None => bail!("no text section"),
+    };
+
+    for sym in image.symbols() {
+        if !sym.is_definition() {
+            continue;
+        }
+        if sym.kind() != SymbolKind::Text {
+            continue;
+        }
+        let address = sym.address();
+        let size = sym.size();
+        if size == 0 {
+            continue;
+        }
+
+        let name = sym.name()?;
+        let code = &text[address as usize..][..size as usize];
+
+        println!("{address:#08x}: <{name}>:");
+        let mut disas = Disassembler::new(code);
+        disas.start_offset(address as usize);
+        let result = Decoder::decode_all(&mut disas);
+        println!("{}", disas.disas());
+        if let Err(e) = result {
+            println!("        : error disassembling: {e:?}");
+        }
+    }
+    Ok(())
+}

pulley/src/disas.rs

@@ -17,6 +17,7 @@ pub struct Disassembler<'a> {
     raw_bytecode: &'a [u8],
     bytecode: SafeBytecodeStream<'a>,
     disas: String,
+    start_offset: usize,
     start: usize,
     temp: String,
     offsets: bool,
@@ -39,6 +40,7 @@ impl<'a> Disassembler<'a> {
             bytecode: SafeBytecodeStream::new(bytecode),
             disas: String::new(),
             start: 0,
+            start_offset: 0,
             temp: String::new(),
             offsets: true,
             hexdump: true,
@@ -61,6 +63,16 @@ impl<'a> Disassembler<'a> {
         self
     }
 
+    /// Configures the offset that this function starts from, if it doesn't
+    /// start from 0.
+    ///
+    /// This can possibly be useful when a single function at a time is being
+    /// disassembled.
+    pub fn start_offset(&mut self, offset: usize) -> &mut Self {
+        self.start_offset = offset;
+        self
+    }
+
     /// Get the disassembly thus far.
     pub fn disas(&self) -> &str {
         &self.disas
@@ -73,7 +85,7 @@ impl<'a> Disassembler<'a> {
                 write!(&mut self.temp, ",").unwrap();
             }
             write!(&mut self.temp, " ").unwrap();
-            val.disas(self.start, &mut self.temp);
+            val.disas(self.start + self.start_offset, &mut self.temp);
         }
     }
 }
@@ -216,7 +228,7 @@ impl<'a> OpVisitor for Disassembler<'a> {
 
     fn after_visit(&mut self) {
         if self.offsets {
-            write!(&mut self.disas, "{:8x}: ", self.start).unwrap();
+            write!(&mut self.disas, "{:8x}: ", self.start + self.start_offset).unwrap();
         }
         if self.hexdump {
             let size = self.bytecode.position() - self.start;