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
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

pub mod wait_poll;
use crate::memory::WasmAccess;
use crate::{__private::wasip1::*, memory::WasmAccessName};
pub use wait_poll::{DefaultWaitPoll, WaitPoll};

/// Trait for handling the `poll_oneoff` WASI function.
pub trait PollOneoff {
    /// Concurrently polls for events.
    fn poll_oneoff<Wasm: WasmAccess + WasmAccessName + 'static>(
        subscriptions: *const Subscription,
        events: *mut Event,
        nsubscriptions: Size,
        nevents: *mut Size,
    ) -> Errno;
}

/// Default implementation of `PollOneoff`.
pub struct DefaultPoll;

impl PollOneoff for DefaultPoll {
    fn poll_oneoff<Wasm: WasmAccess + WasmAccessName + 'static>(
        subscriptions: *const Subscription,
        events: *mut Event,
        nsubscriptions: Size,
        nevents: *mut Size,
    ) -> Errno {
        let subscriptions = subscriptions as i32;
        let events = events as i32;
        let nsubscriptions = nsubscriptions as i32;
        let nevents = nevents as i32;

        let ret = crate::non_recursive_wasi_snapshot_preview1!(
            poll_oneoff(
                subscriptions: i32,
                events: i32,
                nsubscriptions: i32,
                nevents: i32
            ) -> i32
        );

        unsafe { core::mem::transmute::<u16, Errno>(ret as u16) }
    }
}

/// Plugs the poll ecosystem by defining necessary handlers.
///
/// ```rust,no_run
/// use wasi_virt_layer::prelude::*;
///
/// import_wasm!(test_wasm);
///
/// plug_poll!(DefaultPoll, test_wasm);
/// ```
#[macro_export]
macro_rules! plug_poll {
    ($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 _poll_oneoff>](
                    subscriptions: *const $crate::__private::wasip1::Subscription,
                    events: *mut $crate::__private::wasip1::Event,
                    nsubscriptions: $crate::__private::wasip1::Size,
                    nevents: *mut $crate::__private::wasip1::Size,
                ) -> $crate::__private::wasip1::Errno {
                    $crate::__as_t!(@as_t, $wasm);
                    <$ty as $crate::poll::PollOneoff>::poll_oneoff::<T>(subscriptions, events, nsubscriptions, nevents)
                }
            )*
        }
    };
    ($($wasm:ident),*) => {
        $crate::__as_t!(@through, $($wasm),* => $crate::plug_poll, @inner);
    };
    (@inner, $($wasm:ident),*) => {
        $crate::plug_poll!($crate::poll::DefaultPoll, $($wasm),*);
    };
}