Validate

crates/floretta-cli/src/bin/floretta.rs

@@ -32,7 +32,7 @@ pub fn main() -> anyhow::Result<()> {
             wat::parse_bytes(&stdin)?.into_owned()
         }
     };
-    let after = floretta::autodiff(&before);
+    let after = floretta::autodiff(&before)?;
     if args.wat {
         let string = wasmprinter::print_bytes(after)?;
         match args.output {

crates/floretta/src/lib.rs

@@ -1,3 +1,7 @@
-pub fn autodiff(wasm_module: &[u8]) -> Vec<u8> {
-    wasm_module.to_vec()
+//! Reverse-mode automatic differentiation for WebAssembly.
+
+/// Apply reverse-mode automatic differentiation to a WebAssembly module.
+pub fn autodiff(wasm_module: &[u8]) -> wasmparser::Result<Vec<u8>> {
+    wasmparser::validate(wasm_module)?;
+    Ok(wasm_module.to_vec())
 }