wasmer 7.2.0-alpha.2

High-performance WebAssembly runtime
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

#![cfg(any(feature = "cranelift", feature = "llvm", feature = "singlepass"))]

use std::sync::{Arc, Mutex};

use wasmer::sys::NativeEngineExt as _;
use wasmer::{CompileError, Engine};
use wasmer_types::{CompilationProgress, UserAbort};

#[cfg(feature = "singlepass")]
#[test]
fn test_module_compilation_progress_singlepass() {
    let compiler = wasmer::sys::Singlepass::default();
    let engine: wasmer::Engine = wasmer::sys::EngineBuilder::new(compiler).engine().into();
    test_module_compilation_progress(engine);
}

#[cfg(feature = "singlepass")]
#[test]
fn test_module_compilation_abort_singlepass() {
    let compiler = wasmer::sys::Cranelift::default();
    let engine: wasmer::Engine = wasmer::sys::EngineBuilder::new(compiler).engine().into();
    test_module_compilation_abort(engine);
}

#[cfg(feature = "cranelift")]
#[test]
fn test_module_compilation_progress_cranelift() {
    let compiler = wasmer::sys::Cranelift::default();
    let engine: wasmer::Engine = wasmer::sys::EngineBuilder::new(compiler).engine().into();
    test_module_compilation_progress(engine);
}

#[cfg(feature = "cranelift")]
#[test]
fn test_module_compilation_abort_cranelift() {
    let compiler = wasmer::sys::Cranelift::default();
    let engine: wasmer::Engine = wasmer::sys::EngineBuilder::new(compiler).engine().into();
    test_module_compilation_abort(engine);
}

#[cfg(feature = "llvm")]
#[test]
fn test_module_compilation_progress_llvm() {
    let compiler = wasmer::sys::LLVM::default();
    let engine: wasmer::Engine = wasmer::sys::EngineBuilder::new(compiler).engine().into();
    test_module_compilation_progress(engine);
}

#[cfg(feature = "llvm")]
#[test]
fn test_module_compilation_abort_llvm() {
    let compiler = wasmer::sys::LLVM::default();
    let engine: wasmer::Engine = wasmer::sys::EngineBuilder::new(compiler).engine().into();
    test_module_compilation_abort(engine);
}

const SIMPLE_WAT: &str = r#"(module
  (import "env" "div" (func $div (param i32 i32) (result i32)))
  (func (export "add") (param i32 i32) (result i32)
    local.get 0
    local.get 1
    i32.add)
  (func (export "sub") (param i32 i32) (result i32)
    local.get 0
    local.get 1
    i32.sub)
) "#;

fn test_module_compilation_progress(engine: Engine) {
    let items = Arc::new(Mutex::new(Vec::<CompilationProgress>::new()));

    let cb = wasmer_types::CompilationProgressCallback::new({
        let items = items.clone();
        move |p| {
            items.lock().unwrap().push(p);
            Ok(())
        }
    });
    let _module = engine
        .new_module_with_progress(SIMPLE_WAT.as_bytes(), cb)
        .unwrap();

    let last = items
        .lock()
        .unwrap()
        .last()
        .expect("expected at least one progress item")
        .clone();

    // LLVM/Cranelift compiler uses bitcode size for the total.
    const TOTAL_STEPS: u64 = 2014;
    assert_eq!(last.phase_step_count(), Some(TOTAL_STEPS));
    assert_eq!(last.phase_step(), Some(TOTAL_STEPS));
}

fn test_module_compilation_abort(engine: Engine) {
    let reason = "my reason";
    let cb = wasmer_types::CompilationProgressCallback::new(move |_p| Err(UserAbort::new(reason)));
    let err = engine
        .new_module_with_progress(SIMPLE_WAT.as_bytes(), cb)
        .expect_err("should fail");

    match err {
        CompileError::Aborted(e) => {
            assert_eq!(e.reason(), reason)
        }
        other => {
            panic!("expected CompileError::Aborted, got {:?}", other);
        }
    }
}