about summary refs log tree commit diff
path: root/tvix/eval/src/disassembler.rs
diff options
context:
space:
mode:
Diffstat (limited to 'tvix/eval/src/disassembler.rs')
-rw-r--r--tvix/eval/src/disassembler.rs37
1 files changed, 37 insertions, 0 deletions
diff --git a/tvix/eval/src/disassembler.rs b/tvix/eval/src/disassembler.rs
new file mode 100644
index 000000000000..98a6dac9afc4
--- /dev/null
+++ b/tvix/eval/src/disassembler.rs
@@ -0,0 +1,37 @@
+//! Implements methods for disassembling and printing a representation
+//! of compiled code, as well as tracing the runtime stack during
+//! execution.
+use std::io::{Stderr, Write};
+use tabwriter::TabWriter;
+
+use crate::opcode::OpCode;
+use crate::value::Value;
+
+/// Helper struct to trace runtime values and automatically flush the
+/// output after the value is dropped (i.e. in both success and
+/// failure exits from the VM).
+pub struct Tracer(TabWriter<Stderr>);
+
+impl Tracer {
+    pub fn new() -> Self {
+        let mut tw = TabWriter::new(std::io::stderr());
+        write!(&mut tw, "=== runtime trace ===\n").ok();
+        Tracer(tw)
+    }
+
+    pub fn trace(&mut self, op: &OpCode, ip: usize, stack: &[Value]) {
+        write!(&mut self.0, "{:04} {:?}\t[ ", ip, op).ok();
+
+        for val in stack {
+            write!(&mut self.0, "{} ", val).ok();
+        }
+
+        write!(&mut self.0, "]\n").ok();
+    }
+}
+
+impl Drop for Tracer {
+    fn drop(&mut self) {
+        self.0.flush().ok();
+    }
+}