wasi_virt_layer 0.4.0

A virtual layer for WASI modules
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

use crate::__private::wasip1::*;
use crate::memory::{WasmAccess, WasmAccessName};

/// Trait for handling the `sched_yield` WASI function.
pub trait SchedYield {
    /// Yields the execution of the current thread.
    fn sched_yield<Wasm: WasmAccess + WasmAccessName + 'static>() -> Errno;
}

/// Default implementation of `SchedYield`.
pub struct DefaultSched;

impl SchedYield for DefaultSched {
    fn sched_yield<Wasm: WasmAccess + WasmAccessName + 'static>() -> Errno {
        unsafe {
            crate::transporter::non_recursive_sched_yield();
        }
        ERRNO_SUCCESS
    }
}

/// Plugs the sched ecosystem by defining necessary handlers.
///
/// ```rust,no_run
/// use wasi_virt_layer::prelude::*;
///
/// import_wasm!(test_wasm);
///
/// plug_sched!(DefaultSched, test_wasm);
/// ```
#[macro_export]
macro_rules! plug_sched {
    ($ty:ty, $($wasm:ident),+) => {
        const _: () = {
            type __TYPE = $ty;
        };

        $crate::__private::paste::paste! {
            $(
                #[unsafe(no_mangle)]
                #[cfg(target_os = "wasi")]
                pub unsafe extern "C" fn [<__wasip1_vfs_ $wasm _sched_yield>](
                ) -> $crate::__private::wasip1::Errno {
                    $crate::__as_t!(@as_t, $wasm);
                    <$ty as $crate::sched::SchedYield>::sched_yield::<T>()
                }
            )*
        }
    };
    ($($wasm:ident),*) => {
        $crate::__as_t!(@through, $($wasm),* => $crate::plug_sched, @inner);
    };
    (@inner, $($wasm:ident),*) => {
        $crate::plug_sched!($crate::sched::DefaultSched, $($wasm),*);
    };
}