wasmtime_wasi/p3/random/
host.rs

1use crate::p3::bindings::random::{insecure, insecure_seed, random};
2use crate::random::WasiRandomCtx;
3use cap_rand::Rng;
4use cap_rand::distributions::Standard;
5use wasmtime::bail;
6
7impl random::Host for WasiRandomCtx {
8    fn get_random_bytes(&mut self, len: u64) -> wasmtime::Result<Vec<u8>> {
9        if len > self.max_size {
10            bail!("requested len {len:?} exceeds limit {}", self.max_size);
11        }
12        Ok((&mut self.random)
13            .sample_iter(Standard)
14            .take(len as usize)
15            .collect())
16    }
17
18    fn get_random_u64(&mut self) -> wasmtime::Result<u64> {
19        Ok(self.random.sample(Standard))
20    }
21}
22
23impl insecure::Host for WasiRandomCtx {
24    fn get_insecure_random_bytes(&mut self, len: u64) -> wasmtime::Result<Vec<u8>> {
25        if len > self.max_size {
26            bail!("requested len {len:?} exceeds limit {}", self.max_size);
27        }
28        Ok((&mut self.insecure_random)
29            .sample_iter(Standard)
30            .take(len as usize)
31            .collect())
32    }
33
34    fn get_insecure_random_u64(&mut self) -> wasmtime::Result<u64> {
35        Ok(self.insecure_random.sample(Standard))
36    }
37}
38
39impl insecure_seed::Host for WasiRandomCtx {
40    fn get_insecure_seed(&mut self) -> wasmtime::Result<(u64, u64)> {
41        let seed: u128 = self.insecure_random_seed;
42        Ok((seed as u64, (seed >> 64) as u64))
43    }
44}
wasmtime_wasi/p3/random/host.rs

wasmtime_wasi/p3/random/
host.rs
