8b48ce7fb7
1. This makes it easier for implementors to deal with internal APIs. 2. This matches the signatures of the WASI Snapshot traits. Although it is likely true that these methods would have to become immutable in order to implement threading efficiently, threading will impact a large number of existing traits. So this change is practical for now with an already-unavoidable change required for threading. Signed-off-by: Nathaniel McCallum <nathaniel@profian.com>
117 lines
3.2 KiB
Rust
117 lines
3.2 KiB
Rust
use crate::clocks::WasiClocks;
|
|
use crate::dir::{DirCaps, DirEntry, WasiDir};
|
|
use crate::file::{FileCaps, FileEntry, WasiFile};
|
|
use crate::sched::WasiSched;
|
|
use crate::string_array::{StringArray, StringArrayError};
|
|
use crate::table::Table;
|
|
use crate::Error;
|
|
use cap_rand::RngCore;
|
|
use std::path::{Path, PathBuf};
|
|
|
|
pub struct WasiCtx {
|
|
pub args: StringArray,
|
|
pub env: StringArray,
|
|
pub random: Box<dyn RngCore + Send + Sync>,
|
|
pub clocks: WasiClocks,
|
|
pub sched: Box<dyn WasiSched>,
|
|
pub table: Table,
|
|
}
|
|
|
|
impl WasiCtx {
|
|
pub fn new(
|
|
random: Box<dyn RngCore + Send + Sync>,
|
|
clocks: WasiClocks,
|
|
sched: Box<dyn WasiSched>,
|
|
table: Table,
|
|
) -> Self {
|
|
let mut s = WasiCtx {
|
|
args: StringArray::new(),
|
|
env: StringArray::new(),
|
|
random,
|
|
clocks,
|
|
sched,
|
|
table,
|
|
};
|
|
s.set_stdin(Box::new(crate::pipe::ReadPipe::new(std::io::empty())));
|
|
s.set_stdout(Box::new(crate::pipe::WritePipe::new(std::io::sink())));
|
|
s.set_stderr(Box::new(crate::pipe::WritePipe::new(std::io::sink())));
|
|
s
|
|
}
|
|
|
|
pub fn insert_file(&mut self, fd: u32, file: Box<dyn WasiFile>, caps: FileCaps) {
|
|
self.table()
|
|
.insert_at(fd, Box::new(FileEntry::new(caps, file)));
|
|
}
|
|
|
|
pub fn insert_dir(
|
|
&mut self,
|
|
fd: u32,
|
|
dir: Box<dyn WasiDir>,
|
|
caps: DirCaps,
|
|
file_caps: FileCaps,
|
|
path: PathBuf,
|
|
) {
|
|
self.table().insert_at(
|
|
fd,
|
|
Box::new(DirEntry::new(caps, file_caps, Some(path), dir)),
|
|
);
|
|
}
|
|
|
|
pub fn table(&mut self) -> &mut Table {
|
|
&mut self.table
|
|
}
|
|
|
|
pub fn push_arg(&mut self, arg: &str) -> Result<(), StringArrayError> {
|
|
self.args.push(arg.to_owned())
|
|
}
|
|
|
|
pub fn push_env(&mut self, var: &str, value: &str) -> Result<(), StringArrayError> {
|
|
self.env.push(format!("{}={}", var, value))?;
|
|
Ok(())
|
|
}
|
|
|
|
pub fn set_stdin(&mut self, mut f: Box<dyn WasiFile>) {
|
|
let rights = Self::stdio_rights(&mut *f);
|
|
self.insert_file(0, f, rights);
|
|
}
|
|
|
|
pub fn set_stdout(&mut self, mut f: Box<dyn WasiFile>) {
|
|
let rights = Self::stdio_rights(&mut *f);
|
|
self.insert_file(1, f, rights);
|
|
}
|
|
|
|
pub fn set_stderr(&mut self, mut f: Box<dyn WasiFile>) {
|
|
let rights = Self::stdio_rights(&mut *f);
|
|
self.insert_file(2, f, rights);
|
|
}
|
|
|
|
fn stdio_rights(f: &mut dyn WasiFile) -> FileCaps {
|
|
let mut rights = FileCaps::all();
|
|
|
|
// If `f` is a tty, restrict the `tell` and `seek` capabilities, so
|
|
// that wasi-libc's `isatty` correctly detects the file descriptor
|
|
// as a tty.
|
|
if f.isatty() {
|
|
rights &= !(FileCaps::TELL | FileCaps::SEEK);
|
|
}
|
|
|
|
rights
|
|
}
|
|
|
|
pub fn push_preopened_dir(
|
|
&mut self,
|
|
dir: Box<dyn WasiDir>,
|
|
path: impl AsRef<Path>,
|
|
) -> Result<(), Error> {
|
|
let caps = DirCaps::all();
|
|
let file_caps = FileCaps::all();
|
|
self.table().push(Box::new(DirEntry::new(
|
|
caps,
|
|
file_caps,
|
|
Some(path.as_ref().to_owned()),
|
|
dir,
|
|
)))?;
|
|
Ok(())
|
|
}
|
|
}
|