mirror of
https://github.com/roc-lang/roc.git
synced 2025-09-29 23:04:49 +00:00
wasm_interp: support WASI argc/argv, and writing to stdout/stderr
This commit is contained in:
Brian Carroll
parent
736630b53f
commit
284dc6fa51
4 changed files with 386 additions and 69 deletions
|
|
@ -1,54 +1,345 @@
|
|||
use roc_wasm_module::Value;
|
||||
use std::io::{self, Write};
|
||||
|
||||
pub const MODULE_NAME: &str = "wasi_snapshot_preview1";
|
||||
|
||||
pub fn dispatch(function_name: &str, arguments: &[Value], _memory: &mut [u8]) -> Option<Value> {
|
||||
match function_name {
|
||||
"args_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"args_sizes_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"environ_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"environ_sizes_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"clock_res_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"clock_time_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_advise" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_allocate" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_close" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_datasync" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_fdstat_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_fdstat_set_flags" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_fdstat_set_rights" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_filestat_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_filestat_set_size" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_filestat_set_times" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_pread" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_prestat_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_prestat_dir_name" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_pwrite" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_read" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_readdir" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_renumber" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_seek" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_sync" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_tell" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_write" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_create_directory" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_filestat_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_filestat_set_times" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_link" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_open" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_readlink" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_remove_directory" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_rename" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_symlink" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_unlink_file" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"poll_oneoff" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"proc_exit" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"proc_raise" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"sched_yield" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"random_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"sock_recv" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"sock_send" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"sock_shutdown" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
_ => panic!("Unknown WASI function {}({:?})", function_name, arguments),
|
||||
pub struct WasiDispatcher<'a> {
|
||||
pub args: &'a [&'a String],
|
||||
}
|
||||
|
||||
/// Implementation of WASI syscalls
|
||||
/// References for other engines:
|
||||
/// https://github.com/wasmerio/wasmer/blob/ef8d2f651ed29b4b06fdc2070eb8189922c54d82/lib/wasi/src/syscalls/mod.rs
|
||||
/// https://github.com/wasm3/wasm3/blob/045040a97345e636b8be4f3086e6db59cdcc785f/source/extra/wasi_core.h
|
||||
impl<'a> WasiDispatcher<'a> {
|
||||
pub fn new(args: &'a [&'a String]) -> Self {
|
||||
WasiDispatcher { args }
|
||||
}
|
||||
|
||||
pub fn dispatch(
|
||||
&mut self,
|
||||
function_name: &str,
|
||||
arguments: &[Value],
|
||||
memory: &mut [u8],
|
||||
) -> Option<Value> {
|
||||
let success_code = Some(Value::I32(Errno::Success as i32));
|
||||
|
||||
match function_name {
|
||||
"args_get" => {
|
||||
// uint8_t ** argv,
|
||||
let mut ptr_ptr_argv = arguments[0].unwrap_i32() as usize;
|
||||
// uint8_t * argv_buf
|
||||
let mut ptr_argv_buf = arguments[1].unwrap_i32() as usize;
|
||||
|
||||
for arg in self.args {
|
||||
write_u32(memory, ptr_ptr_argv, ptr_argv_buf as u32);
|
||||
let bytes_target = &mut memory[ptr_argv_buf..][..arg.len()];
|
||||
bytes_target.copy_from_slice(arg.as_bytes());
|
||||
memory[ptr_argv_buf + arg.len()] = 0; // C string zero termination
|
||||
ptr_argv_buf += arg.len() + 1;
|
||||
ptr_ptr_argv += 4;
|
||||
}
|
||||
|
||||
success_code
|
||||
}
|
||||
"args_sizes_get" => {
|
||||
// (i32, i32) -> i32
|
||||
|
||||
// number of string arguments
|
||||
let ptr_argc = arguments[0].unwrap_i32() as usize;
|
||||
// size of string arguments buffer
|
||||
let ptr_argv_buf_size = arguments[1].unwrap_i32() as usize;
|
||||
|
||||
let argc = self.args.len() as u32;
|
||||
write_u32(memory, ptr_argc, argc);
|
||||
|
||||
let argv_buf_size: u32 = self.args.iter().map(|a| 1 + a.len() as u32).sum();
|
||||
write_u32(memory, ptr_argv_buf_size, argv_buf_size);
|
||||
|
||||
success_code
|
||||
}
|
||||
"environ_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"environ_sizes_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"clock_res_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"clock_time_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_advise" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_allocate" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_close" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_datasync" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_fdstat_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_fdstat_set_flags" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_fdstat_set_rights" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_filestat_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_filestat_set_size" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_filestat_set_times" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_pread" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_prestat_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_prestat_dir_name" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_pwrite" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_read" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_readdir" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_renumber" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_seek" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_sync" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_tell" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"fd_write" => {
|
||||
// file descriptor
|
||||
let fd = arguments[0].unwrap_i32();
|
||||
// Array of IO vectors
|
||||
let ptr_iovs = arguments[1].unwrap_i32() as usize;
|
||||
// Length of array
|
||||
let iovs_len = arguments[2].unwrap_i32();
|
||||
// Out param: number of bytes written
|
||||
let ptr_nwritten = arguments[3].unwrap_i32() as usize;
|
||||
|
||||
let mut write_lock = match fd {
|
||||
1 => Ok(io::stdout().lock()),
|
||||
2 => Err(io::stderr().lock()),
|
||||
_ => return Some(Value::I32(Errno::Inval as i32)),
|
||||
};
|
||||
|
||||
let mut n_written: i32 = 0;
|
||||
let mut negative_length_count = 0;
|
||||
for _ in 0..iovs_len {
|
||||
// https://man7.org/linux/man-pages/man2/readv.2.html
|
||||
// struct iovec {
|
||||
// void *iov_base; /* Starting address */
|
||||
// size_t iov_len; /* Number of bytes to transfer */
|
||||
// };
|
||||
let iov_base = read_u32(memory, ptr_iovs) as usize;
|
||||
let iov_len = read_i32(memory, ptr_iovs + 4);
|
||||
if iov_len < 0 {
|
||||
// I found negative-length iov's when I implemented this in JS for the web REPL (see wasi.js)
|
||||
// I'm not sure why, but this solution worked, and it's the same WASI libc - there's only one.
|
||||
n_written += iov_len;
|
||||
negative_length_count += 1;
|
||||
continue;
|
||||
}
|
||||
let bytes = &memory[iov_base..][..iov_len as usize];
|
||||
|
||||
match &mut write_lock {
|
||||
Ok(stdout) => {
|
||||
n_written += stdout.write(bytes).unwrap() as i32;
|
||||
}
|
||||
Err(stderr) => {
|
||||
n_written += stderr.write(bytes).unwrap() as i32;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
write_i32(memory, ptr_nwritten, n_written);
|
||||
if negative_length_count > 0 {
|
||||
// Let's see if we ever get this message. If not, we can remove this negative-length stuff.
|
||||
eprintln!(
|
||||
"WASI DEV INFO: found {} negative-length iovecs.",
|
||||
negative_length_count
|
||||
);
|
||||
}
|
||||
|
||||
success_code
|
||||
}
|
||||
"path_create_directory" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_filestat_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_filestat_set_times" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_link" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_open" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_readlink" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_remove_directory" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_rename" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_symlink" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"path_unlink_file" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"poll_oneoff" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"proc_exit" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"proc_raise" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"sched_yield" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"random_get" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"sock_recv" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"sock_send" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
"sock_shutdown" => todo!("WASI {}({:?})", function_name, arguments),
|
||||
_ => panic!("Unknown WASI function {}({:?})", function_name, arguments),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn read_u32(memory: &mut [u8], addr: usize) -> u32 {
|
||||
let mut bytes = [0; 4];
|
||||
bytes.copy_from_slice(&memory[addr..][..4]);
|
||||
u32::from_le_bytes(bytes)
|
||||
}
|
||||
|
||||
fn read_i32(memory: &mut [u8], addr: usize) -> i32 {
|
||||
let mut bytes = [0; 4];
|
||||
bytes.copy_from_slice(&memory[addr..][..4]);
|
||||
i32::from_le_bytes(bytes)
|
||||
}
|
||||
|
||||
fn write_u32(memory: &mut [u8], addr: usize, value: u32) {
|
||||
memory[addr..][..4].copy_from_slice(&value.to_le_bytes());
|
||||
}
|
||||
|
||||
fn write_i32(memory: &mut [u8], addr: usize, value: i32) {
|
||||
memory[addr..][..4].copy_from_slice(&value.to_le_bytes());
|
||||
}
|
||||
|
||||
/// Error codes returned by functions.
|
||||
/// Not all of these error codes are returned by the functions provided by this
|
||||
/// API; some are used in higher-level library layers, and others are provided
|
||||
/// merely for alignment with POSIX.
|
||||
#[repr(u8)]
|
||||
#[derive(Clone, Copy, PartialEq, Eq)]
|
||||
pub enum Errno {
|
||||
/// No error occurred. System call completed successfully.
|
||||
Success,
|
||||
/// Argument list too long.
|
||||
Toobig,
|
||||
/// Permission denied.
|
||||
Access,
|
||||
/// Address in use.
|
||||
Addrinuse,
|
||||
/// Address not available.
|
||||
Addrnotavail,
|
||||
/// Address family not supported.
|
||||
Afnosupport,
|
||||
/// Resource unavailable, or operation would block.
|
||||
Again,
|
||||
/// Connection already in progress.
|
||||
Already,
|
||||
/// Bad file descriptor.
|
||||
Badf,
|
||||
/// Bad message.
|
||||
Badmsg,
|
||||
/// Device or resource busy.
|
||||
Busy,
|
||||
/// Operation canceled.
|
||||
Canceled,
|
||||
/// No child processes.
|
||||
Child,
|
||||
/// Connection aborted.
|
||||
Connaborted,
|
||||
/// Connection refused.
|
||||
Connrefused,
|
||||
/// Connection reset.
|
||||
Connreset,
|
||||
/// Resource deadlock would occur.
|
||||
Deadlk,
|
||||
/// Destination address required.
|
||||
Destaddrreq,
|
||||
/// Mathematics argument out of domain of function.
|
||||
Dom,
|
||||
/// Reserved.
|
||||
Dquot,
|
||||
/// File exists.
|
||||
Exist,
|
||||
/// Bad address.
|
||||
Fault,
|
||||
/// File too large.
|
||||
Fbig,
|
||||
/// Host is unreachable.
|
||||
Hostunreach,
|
||||
/// Identifier removed.
|
||||
Idrm,
|
||||
/// Illegal byte sequence.
|
||||
Ilseq,
|
||||
/// Operation in progress.
|
||||
Inprogress,
|
||||
/// Interrupted function.
|
||||
Intr,
|
||||
/// Invalid argument.
|
||||
Inval,
|
||||
/// I/O error.
|
||||
Io,
|
||||
/// Socket is connected.
|
||||
Isconn,
|
||||
/// Is a directory.
|
||||
Isdir,
|
||||
/// Too many levels of symbolic links.
|
||||
Loop,
|
||||
/// File descriptor value too large.
|
||||
Mfile,
|
||||
/// Too many links.
|
||||
Mlink,
|
||||
/// Message too large.
|
||||
Msgsize,
|
||||
/// Reserved.
|
||||
Multihop,
|
||||
/// Filename too long.
|
||||
Nametoolong,
|
||||
/// Network is down.
|
||||
Netdown,
|
||||
/// Connection aborted by network.
|
||||
Netreset,
|
||||
/// Network unreachable.
|
||||
Netunreach,
|
||||
/// Too many files open in system.
|
||||
Nfile,
|
||||
/// No buffer space available.
|
||||
Nobufs,
|
||||
/// No such device.
|
||||
Nodev,
|
||||
/// No such file or directory.
|
||||
Noent,
|
||||
/// Executable file format error.
|
||||
Noexec,
|
||||
/// No locks available.
|
||||
Nolck,
|
||||
/// Reserved.
|
||||
Nolink,
|
||||
/// Not enough space.
|
||||
Nomem,
|
||||
/// No message of the desired type.
|
||||
Nomsg,
|
||||
/// Protocol not available.
|
||||
Noprotoopt,
|
||||
/// No space left on device.
|
||||
Nospc,
|
||||
/// Function not supported.
|
||||
Nosys,
|
||||
/// The socket is not connected.
|
||||
Notconn,
|
||||
/// Not a directory or a symbolic link to a directory.
|
||||
Notdir,
|
||||
/// Directory not empty.
|
||||
Notempty,
|
||||
/// State not recoverable.
|
||||
Notrecoverable,
|
||||
/// Not a socket.
|
||||
Notsock,
|
||||
/// Not supported, or operation not supported on socket.
|
||||
Notsup,
|
||||
/// Inappropriate I/O control operation.
|
||||
Notty,
|
||||
/// No such device or address.
|
||||
Nxio,
|
||||
/// Value too large to be stored in data type.
|
||||
Overflow,
|
||||
/// Previous owner died.
|
||||
Ownerdead,
|
||||
/// Operation not permitted.
|
||||
Perm,
|
||||
/// Broken pipe.
|
||||
Pipe,
|
||||
/// Protocol error.
|
||||
Proto,
|
||||
/// Protocol not supported.
|
||||
Protonosupport,
|
||||
/// Protocol wrong type for socket.
|
||||
Prototype,
|
||||
/// Result too large.
|
||||
Range,
|
||||
/// Read-only file system.
|
||||
Rofs,
|
||||
/// Invalid seek.
|
||||
Spipe,
|
||||
/// No such process.
|
||||
Srch,
|
||||
/// Reserved.
|
||||
Stale,
|
||||
/// Connection timed out.
|
||||
Timedout,
|
||||
/// Text file busy.
|
||||
Txtbsy,
|
||||
/// Cross-device link.
|
||||
Xdev,
|
||||
/// Extension: Capabilities insufficient.
|
||||
Notcapable,
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue