xref: /wasmtime-44.0.1/crates/fuzzing/tests/oom/instance_pre.rs (revision c7d25dfd)

#![cfg(arc_try_new)]

use wasmtime::{Config, Engine, Linker, Module, Result, Store};
use wasmtime_fuzzing::oom::OomTest;

#[test]
fn instance_pre_instantiate() -> Result<()> {
    let module_bytes = {
        let mut config = Config::new();
        config.concurrency_support(false);
        let engine = Engine::new(&config)?;
        let module = Module::new(
            &engine,
            r#"
                (module
                    (import "module" "func" (func (param i32) (result i32)))

                    (memory (export "memory") 1)
                    (data (i32.const 0) "a")

                    (table (export "table") 1 funcref)
                    (elem (i32.const 0) func 1)

                    (func (export "func") (param i32) (result i32)
                        (call 0 (local.get 0))
                    )
                )
            "#,
        )?;
        module.serialize()?
    };

    let mut config = Config::new();
    config.enable_compiler(false);
    config.concurrency_support(false);

    let engine = Engine::new(&config)?;

    let mut linker = Linker::<()>::new(&engine);
    linker.func_wrap("module", "func", |x: i32| x * 2)?;

    let module = unsafe { Module::deserialize(&engine, &module_bytes)? };
    let instance_pre = linker.instantiate_pre(&module)?;

    OomTest::new().test(|| {
        let mut store = Store::try_new(&engine, ())?;
        let _ = instance_pre.instantiate(&mut store)?;
        Ok(())
    })
}