62 lines
1.8 KiB
Rust
62 lines
1.8 KiB
Rust
pub mod clocks;
|
|
pub mod dir;
|
|
pub mod file;
|
|
pub mod sched;
|
|
pub mod stdio;
|
|
|
|
use cap_rand::RngCore;
|
|
use std::cell::RefCell;
|
|
use std::path::Path;
|
|
use std::rc::Rc;
|
|
use wasi_c2::{table::Table, Error, WasiCtx, WasiFile};
|
|
|
|
pub struct WasiCtxBuilder(wasi_c2::WasiCtxBuilder);
|
|
|
|
impl WasiCtxBuilder {
|
|
pub fn new() -> Self {
|
|
WasiCtxBuilder(WasiCtx::builder(
|
|
random(),
|
|
clocks::clocks(),
|
|
Box::new(sched::SyncSched::new()),
|
|
Rc::new(RefCell::new(Table::new())),
|
|
))
|
|
}
|
|
pub fn env(self, var: &str, value: &str) -> Result<Self, wasi_c2::StringArrayError> {
|
|
let s = self.0.env(var, value)?;
|
|
Ok(WasiCtxBuilder(s))
|
|
}
|
|
pub fn arg(self, arg: &str) -> Result<Self, wasi_c2::StringArrayError> {
|
|
let s = self.0.arg(arg)?;
|
|
Ok(WasiCtxBuilder(s))
|
|
}
|
|
pub fn stdin(self, f: Box<dyn WasiFile>) -> Self {
|
|
WasiCtxBuilder(self.0.stdin(f))
|
|
}
|
|
pub fn stdout(self, f: Box<dyn WasiFile>) -> Self {
|
|
WasiCtxBuilder(self.0.stdout(f))
|
|
}
|
|
pub fn stderr(self, f: Box<dyn WasiFile>) -> Self {
|
|
WasiCtxBuilder(self.0.stderr(f))
|
|
}
|
|
pub fn inherit_stdio(self) -> Self {
|
|
self.stdin(Box::new(crate::stdio::stdin()))
|
|
.stdout(Box::new(crate::stdio::stdout()))
|
|
.stderr(Box::new(crate::stdio::stderr()))
|
|
}
|
|
pub fn preopened_dir(
|
|
self,
|
|
dir: cap_std::fs::Dir,
|
|
path: impl AsRef<Path>,
|
|
) -> Result<Self, Error> {
|
|
let dir = Box::new(crate::dir::Dir::from_cap_std(dir));
|
|
Ok(WasiCtxBuilder(self.0.preopened_dir(dir, path)?))
|
|
}
|
|
pub fn build(self) -> Result<WasiCtx, Error> {
|
|
self.0.build()
|
|
}
|
|
}
|
|
|
|
pub fn random() -> RefCell<Box<dyn RngCore>> {
|
|
RefCell::new(Box::new(unsafe { cap_rand::rngs::OsRng::default() }))
|
|
}
|