wasmtime/lib/wast/src/wast.rs

use cranelift_codegen::isa;
use std::collections::HashMap;
use std::fs;
use std::io;
use std::io::Read;
use std::path::Path;
use std::str;
use wabt::script::{self, Action, Command, CommandKind, ModuleBinary, ScriptParser};
use wasmtime_execute::{Code, InstanceWorld, InvokeOutcome, Value};

struct Instances {
    current: Option<InstanceWorld>,
    namespace: HashMap<String, InstanceWorld>,
    code: Code,
}

impl Instances {
    pub fn new() -> Self {
        Self {
            current: None,
            namespace: HashMap::new(),
            code: Code::new(),
        }
    }

    fn instantiate(&mut self, isa: &isa::TargetIsa, module: ModuleBinary) -> InstanceWorld {
        InstanceWorld::new(&mut self.code, isa, &module.into_vec()).unwrap()
    }

    pub fn define_unnamed_module(&mut self, isa: &isa::TargetIsa, module: ModuleBinary) {
        self.current = Some(self.instantiate(isa, module));
    }

    pub fn define_named_module(
        &mut self,
        isa: &isa::TargetIsa,
        name: String,
        module: ModuleBinary,
    ) {
        let world = self.instantiate(isa, module);
        self.namespace.insert(name, world);
    }

    pub fn perform_action(&mut self, isa: &isa::TargetIsa, action: Action) -> InvokeOutcome {
        match action {
            Action::Invoke {
                module,
                field,
                args,
            } => {
                let mut value_args = Vec::with_capacity(args.len());
                for a in args {
                    value_args.push(match a {
                        script::Value::I32(i) => Value::I32(i),
                        script::Value::I64(i) => Value::I64(i),
                        script::Value::F32(i) => Value::F32(i.to_bits()),
                        script::Value::F64(i) => Value::F64(i.to_bits()),
                    });
                }
                match module {
                    None => match self.current {
                        None => panic!("invoke performed with no module present"),
                        Some(ref mut instance_world) => instance_world
                            .invoke(&mut self.code, isa, &field, &value_args)
                            .expect(&format!("error invoking {} in current module", field)),
                    },
                    Some(name) => self
                        .namespace
                        .get_mut(&name)
                        .expect(&format!("module {} not declared", name))
                        .invoke(&mut self.code, isa, &field, &value_args)
                        .expect(&format!("error invoking {} in module {}", field, name)),
                }
            }
            _ => panic!("unsupported action {:?}", action),
        }
    }
}

/// Run a wast script from a byte buffer.
pub fn wast_buffer(name: &str, isa: &isa::TargetIsa, wast: &[u8]) {
    let mut parser = ScriptParser::from_str(str::from_utf8(wast).unwrap()).unwrap();
    let mut instances = Instances::new();

    while let Some(Command { kind, line }) = parser.next().unwrap() {
        match kind {
            CommandKind::Module { module, name } => {
                if let Some(name) = name {
                    instances.define_named_module(&*isa, name, module.clone());
                }

                instances.define_unnamed_module(&*isa, module)
            }
            CommandKind::PerformAction(action) => match instances.perform_action(&*isa, action) {
                InvokeOutcome::Returned { .. } => {}
                InvokeOutcome::Trapped { message } => {
                    panic!("{}:{}: a trap occurred: {}", name, line, message);
                }
            },
            CommandKind::AssertReturn { action, expected } => {
                match instances.perform_action(&*isa, action) {
                    InvokeOutcome::Returned { values } => {
                        for (v, e) in values.iter().zip(expected.iter()) {
                            match *e {
                                script::Value::I32(x) => {
                                    assert_eq!(x, v.unwrap_i32(), "at {}:{}", name, line)
                                }
                                script::Value::I64(x) => {
                                    assert_eq!(x, v.unwrap_i64(), "at {}:{}", name, line)
                                }
                                script::Value::F32(x) => {
                                    assert_eq!(x.to_bits(), v.unwrap_f32(), "at {}:{}", name, line)
                                }
                                script::Value::F64(x) => {
                                    assert_eq!(x.to_bits(), v.unwrap_f64(), "at {}:{}", name, line)
                                }
                            };
                        }
                    }
                    InvokeOutcome::Trapped { message } => {
                        panic!(
                            "{}:{}: expected normal return, but a trap occurred: {}",
                            name, line, message
                        );
                    }
                }
            }
            CommandKind::AssertTrap { action, message } => {
                match instances.perform_action(&*isa, action) {
                    InvokeOutcome::Returned { values } => panic!(
                        "{}:{}: expected trap, but invoke returned with {:?}",
                        name, line, values
                    ),
                    InvokeOutcome::Trapped {
                        message: trap_message,
                    } => {
                        println!(
                            "{}:{}: TODO: Check the trap message: expected {}, got {}",
                            name, line, message, trap_message
                        );
                    }
                }
            }
            CommandKind::AssertExhaustion { action } => {
                match instances.perform_action(&*isa, action) {
                    InvokeOutcome::Returned { values } => panic!(
                        "{}:{}: expected exhaustion, but invoke returned with {:?}",
                        name, line, values
                    ),
                    InvokeOutcome::Trapped { message } => {
                        println!(
                            "{}:{}: TODO: Check the exhaustion message: {}",
                            name, line, message
                        );
                    }
                }
            }
            command => {
                println!("{}:{}: TODO: implement {:?}", name, line, command);
            }
        }
    }
}

/// Run a wast script from a file.
pub fn wast_file(path: &Path, isa: &isa::TargetIsa) -> Result<(), String> {
    let wast = read_to_end(path).map_err(|e| e.to_string())?;
    wast_buffer(&path.display().to_string(), isa, &wast);
    Ok(())
}

fn read_to_end(path: &Path) -> Result<Vec<u8>, io::Error> {
    let mut buf: Vec<u8> = Vec::new();
    let mut file = fs::File::open(path)?;
    file.read_to_end(&mut buf)?;
    Ok(buf)
}