forked from Mirrors/wasm3
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
459 lines
14 KiB
C
459 lines
14 KiB
C
//
|
|
// m3_api_uvwasi.c
|
|
//
|
|
// Created by Colin J. Ihrig on 4/20/20.
|
|
// Copyright © 2020 Colin J. Ihrig. All rights reserved.
|
|
|
|
//
|
|
|
|
#define _POSIX_C_SOURCE 200809L
|
|
|
|
#include "m3_api_wasi.h"
|
|
|
|
#include "m3_api_defs.h"
|
|
#include "m3_env.h"
|
|
#include "m3_exception.h"
|
|
|
|
#if defined(d_m3HasUVWASI)
|
|
|
|
typedef uint32_t __wasi_size_t;
|
|
#include "extra/wasi_core.h"
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#include "uvwasi.h"
|
|
|
|
#ifdef __APPLE__
|
|
# include <crt_externs.h>
|
|
# define environ (*_NSGetEnviron())
|
|
#elif !defined(_MSC_VER)
|
|
extern char** environ;
|
|
#endif
|
|
|
|
uvwasi_t uvwasi;
|
|
|
|
typedef struct wasi_iovec_t
|
|
{
|
|
__wasi_size_t buf;
|
|
__wasi_size_t buf_len;
|
|
} wasi_iovec_t;
|
|
|
|
/*
|
|
* WASI API implementation
|
|
*/
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_args_get)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArgMem (u32* , argv)
|
|
m3ApiGetArgMem (char* , argv_buf)
|
|
|
|
if (runtime == NULL) { m3ApiReturn(__WASI_EINVAL); }
|
|
|
|
for (u32 i = 0; i < runtime->argc; ++i)
|
|
{
|
|
argv[i] = m3ApiPtrToOffset (argv_buf);
|
|
|
|
size_t len = strlen (runtime->argv [i]);
|
|
memcpy (argv_buf, runtime->argv [i], len);
|
|
argv_buf += len;
|
|
* argv_buf++ = 0;
|
|
}
|
|
|
|
m3ApiReturn(UVWASI_ESUCCESS);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_args_sizes_get)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArgMem (__wasi_size_t* , argc)
|
|
m3ApiGetArgMem (__wasi_size_t* , argv_buf_size)
|
|
|
|
if (runtime == NULL) { m3ApiReturn(__WASI_EINVAL); }
|
|
|
|
*argc = runtime->argc;
|
|
*argv_buf_size = 0;
|
|
for (u32 i = 0; i < runtime->argc; ++i)
|
|
{
|
|
* argv_buf_size += strlen (runtime->argv [i]) + 1;
|
|
}
|
|
m3ApiReturn(UVWASI_ESUCCESS);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_environ_get)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArgMem (u32* , env)
|
|
m3ApiGetArgMem (char* , env_buf)
|
|
|
|
char **environment;
|
|
uvwasi_errno_t err;
|
|
|
|
environment = calloc(uvwasi.envc, sizeof(char *));
|
|
if (environment == NULL) {
|
|
m3ApiReturn(UVWASI_ENOMEM);
|
|
}
|
|
|
|
err = uvwasi_environ_get(&uvwasi, environment, env_buf);
|
|
if (err != UVWASI_ESUCCESS) {
|
|
free(environment);
|
|
m3ApiReturn(err);
|
|
}
|
|
|
|
uint32_t environ_buf_offset = m3ApiPtrToOffset(env_buf);
|
|
|
|
for (u32 i = 0; i < uvwasi.envc; ++i)
|
|
{
|
|
uint32_t offset = environ_buf_offset +
|
|
(environment[i] - environment[0]);
|
|
env[i] = offset;
|
|
}
|
|
|
|
free(environment);
|
|
m3ApiReturn(UVWASI_ESUCCESS);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_environ_sizes_get)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArgMem (__wasi_size_t* , env_count)
|
|
m3ApiGetArgMem (__wasi_size_t* , env_buf_size)
|
|
|
|
size_t count;
|
|
size_t buf_size;
|
|
uvwasi_errno_t err;
|
|
|
|
err = uvwasi_environ_sizes_get(&uvwasi, &count, &buf_size);
|
|
*env_count = count;
|
|
*env_buf_size = buf_size;
|
|
m3ApiReturn(err);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_fd_prestat_dir_name)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArg (uvwasi_fd_t , fd)
|
|
m3ApiGetArgMem (char* , path)
|
|
m3ApiGetArg (size_t , path_len)
|
|
|
|
m3ApiReturn(uvwasi_fd_prestat_dir_name(&uvwasi, fd, path, path_len));
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_fd_prestat_get)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArg (uvwasi_fd_t , fd)
|
|
m3ApiGetArgMem (uint32_t * , buf)
|
|
|
|
uvwasi_prestat_t prestat;
|
|
uvwasi_errno_t err;
|
|
|
|
err = uvwasi_fd_prestat_get(&uvwasi, fd, &prestat);
|
|
if (err != UVWASI_ESUCCESS) {
|
|
m3ApiReturn(err);
|
|
}
|
|
|
|
// TODO(cjihrig): This memory writing logic is wrong.
|
|
*buf = prestat.pr_type;
|
|
*(buf + 1) = prestat.u.dir.pr_name_len;
|
|
m3ApiReturn(UVWASI_ESUCCESS);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_fd_fdstat_get)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArg (uvwasi_fd_t , fd)
|
|
m3ApiGetArgMem (uvwasi_fdstat_t* , fdstat)
|
|
|
|
m3ApiReturn(uvwasi_fd_fdstat_get(&uvwasi, fd, fdstat));
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_fd_fdstat_set_flags)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArg (uvwasi_fd_t , fd)
|
|
m3ApiGetArg (uvwasi_fdflags_t , flags)
|
|
|
|
m3ApiReturn(uvwasi_fd_fdstat_set_flags(&uvwasi, fd, flags));
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_fd_seek)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArg (uvwasi_fd_t , fd)
|
|
m3ApiGetArg (uvwasi_filedelta_t , offset)
|
|
m3ApiGetArg (uvwasi_whence_t , whence)
|
|
m3ApiGetArgMem (uvwasi_filesize_t* , result)
|
|
|
|
m3ApiReturn(uvwasi_fd_seek(&uvwasi, fd, offset, whence, result));
|
|
}
|
|
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_path_open)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArg (uvwasi_fd_t , dirfd)
|
|
m3ApiGetArg (uvwasi_lookupflags_t , dirflags)
|
|
m3ApiGetArgMem (const char * , path)
|
|
m3ApiGetArg (__wasi_size_t , path_len)
|
|
m3ApiGetArg (uvwasi_oflags_t , oflags)
|
|
m3ApiGetArg (uvwasi_rights_t , fs_rights_base)
|
|
m3ApiGetArg (uvwasi_rights_t , fs_rights_inheriting)
|
|
m3ApiGetArg (uvwasi_fdflags_t , fs_flags)
|
|
m3ApiGetArgMem (uvwasi_fd_t * , fd)
|
|
|
|
m3ApiReturn(uvwasi_path_open(&uvwasi,
|
|
dirfd,
|
|
dirflags,
|
|
path,
|
|
path_len,
|
|
oflags,
|
|
fs_rights_base,
|
|
fs_rights_inheriting,
|
|
fs_flags,
|
|
fd));
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_path_filestat_get)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArg (__wasi_fd_t , fd)
|
|
m3ApiGetArg (__wasi_lookupflags_t , flags)
|
|
m3ApiGetArgMem (const char * , path)
|
|
m3ApiGetArg (uint32_t , path_len)
|
|
m3ApiGetArgMem (__wasi_filestat_t * , buf)
|
|
|
|
uvwasi_errno_t ret = uvwasi_path_filestat_get(&uvwasi, fd, flags, path, path_len, buf);
|
|
|
|
m3ApiReturn(ret);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_fd_read)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArg (uvwasi_fd_t , fd)
|
|
m3ApiGetArgMem (wasi_iovec_t* , wasi_iovs)
|
|
m3ApiGetArg (__wasi_size_t , iovs_len)
|
|
m3ApiGetArgMem (__wasi_size_t* , nread)
|
|
|
|
// TODO: check iovs_len
|
|
#if defined(M3_COMPILER_MSVC)
|
|
uvwasi_ciovec_t iovs[32];
|
|
#else
|
|
uvwasi_ciovec_t iovs[iovs_len];
|
|
#endif
|
|
size_t num_read;
|
|
uvwasi_errno_t ret;
|
|
|
|
if (iovs == NULL) {
|
|
m3ApiReturn(UVWASI_ENOMEM);
|
|
}
|
|
|
|
for (__wasi_size_t i = 0; i < iovs_len; ++i) {
|
|
iovs[i].buf = m3ApiOffsetToPtr(wasi_iovs[i].buf);
|
|
iovs[i].buf_len = wasi_iovs[i].buf_len;
|
|
}
|
|
|
|
ret = uvwasi_fd_read(&uvwasi, fd, iovs, iovs_len, &num_read);
|
|
*nread = num_read;
|
|
m3ApiReturn(ret);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_fd_write)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArg (uvwasi_fd_t , fd)
|
|
m3ApiGetArgMem (wasi_iovec_t* , wasi_iovs)
|
|
m3ApiGetArg (__wasi_size_t , iovs_len)
|
|
m3ApiGetArgMem (__wasi_size_t* , nwritten)
|
|
|
|
// TODO: check iovs_len
|
|
|
|
#if defined(M3_COMPILER_MSVC)
|
|
uvwasi_ciovec_t iovs[32];
|
|
#else
|
|
uvwasi_ciovec_t iovs[iovs_len];
|
|
#endif
|
|
size_t num_written;
|
|
uvwasi_errno_t ret;
|
|
|
|
if (iovs == NULL) {
|
|
m3ApiReturn(UVWASI_ENOMEM);
|
|
}
|
|
|
|
for (__wasi_size_t i = 0; i < iovs_len; ++i) {
|
|
iovs[i].buf = m3ApiOffsetToPtr(wasi_iovs[i].buf);
|
|
iovs[i].buf_len = wasi_iovs[i].buf_len;
|
|
}
|
|
|
|
ret = uvwasi_fd_write(&uvwasi, fd, iovs, iovs_len, &num_written);
|
|
*nwritten = num_written;
|
|
m3ApiReturn(ret);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_fd_close)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArg (uvwasi_fd_t, fd)
|
|
|
|
uvwasi_errno_t ret = uvwasi_fd_close(&uvwasi, fd);
|
|
|
|
m3ApiReturn(ret);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_fd_datasync)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArg (uvwasi_fd_t, fd)
|
|
|
|
uvwasi_errno_t ret = uvwasi_fd_datasync(&uvwasi, fd);
|
|
|
|
m3ApiReturn(ret);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_random_get)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArgMem (uint8_t* , buf)
|
|
m3ApiGetArg (__wasi_size_t , buflen)
|
|
|
|
uvwasi_errno_t ret = uvwasi_random_get(&uvwasi, buf, buflen);
|
|
|
|
m3ApiReturn(ret);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_clock_res_get)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArg (uvwasi_clockid_t , wasi_clk_id)
|
|
m3ApiGetArgMem (uvwasi_timestamp_t* , resolution)
|
|
|
|
uvwasi_errno_t ret = uvwasi_clock_res_get(&uvwasi, wasi_clk_id, resolution);
|
|
|
|
m3ApiReturn(ret);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_clock_time_get)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArg (uvwasi_clockid_t , wasi_clk_id)
|
|
m3ApiGetArg (uvwasi_timestamp_t , precision)
|
|
m3ApiGetArgMem (uvwasi_timestamp_t* , time)
|
|
|
|
uvwasi_errno_t ret = uvwasi_clock_time_get(&uvwasi, wasi_clk_id, precision, time);
|
|
|
|
m3ApiReturn(ret);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_poll_oneoff)
|
|
{
|
|
m3ApiReturnType (uint32_t)
|
|
m3ApiGetArgMem (const __wasi_subscription_t* , in)
|
|
m3ApiGetArgMem (__wasi_event_t* , out)
|
|
m3ApiGetArg (__wasi_size_t , nsubscriptions)
|
|
m3ApiGetArgMem (__wasi_size_t* , nevents)
|
|
|
|
uvwasi_errno_t ret = uvwasi_poll_oneoff(&uvwasi, in, out, nsubscriptions, nevents);
|
|
|
|
m3ApiReturn(ret);
|
|
}
|
|
|
|
m3ApiRawFunction(m3_wasi_unstable_proc_exit)
|
|
{
|
|
m3ApiGetArg (uint32_t, code)
|
|
|
|
runtime->exit_code = code;
|
|
|
|
m3ApiTrap(m3Err_trapExit);
|
|
}
|
|
|
|
|
|
static
|
|
M3Result SuppressLookupFailure(M3Result i_result)
|
|
{
|
|
if (i_result == m3Err_functionLookupFailed)
|
|
return m3Err_none;
|
|
else
|
|
return i_result;
|
|
}
|
|
|
|
|
|
M3Result m3_LinkWASI (IM3Module module)
|
|
{
|
|
M3Result result = m3Err_none;
|
|
|
|
// TODO(cjihrig): uvwasi currently implements 'wasi_snapshot_preview1' but
|
|
// call it 'wasi_unstable' here for compatibility with the other wasm3 WASI
|
|
// implementations.
|
|
const char* wasi = "wasi_unstable";
|
|
|
|
uvwasi_options_t init_options;
|
|
uvwasi_errno_t ret;
|
|
|
|
init_options.in = 0;
|
|
init_options.out = 1;
|
|
init_options.err = 2;
|
|
init_options.fd_table_size = 3;
|
|
// runtime->argc is not initialized at this point. However, it's fine to
|
|
// use the runtime instead of uvwasi to implement the two WASI functions for
|
|
// working with command line arguments.
|
|
init_options.argc = 0;
|
|
init_options.argv = NULL;
|
|
init_options.envp = (char**) environ;
|
|
|
|
// TODO(cjihrig): This requires better support for the --dir command line
|
|
// flag to implement properly. For now, just let WASI applications access
|
|
// the current working directory as the sandboxed root directory.
|
|
|
|
uvwasi_preopen_t preopens[1];
|
|
preopens[0].mapped_path = "/";
|
|
preopens[0].real_path = ".";
|
|
|
|
init_options.preopenc = 1;
|
|
init_options.preopens = &preopens;
|
|
init_options.allocator = NULL;
|
|
|
|
ret = uvwasi_init(&uvwasi, &init_options);
|
|
|
|
// uvwasi_init() returns WASI errors, which don't really map to m3 Errors,
|
|
// so return unknown error for now.
|
|
if (ret != UVWASI_ESUCCESS) {
|
|
result = m3Err_unknownError;
|
|
return result;
|
|
}
|
|
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "args_sizes_get", "i(**)", &m3_wasi_unstable_args_sizes_get)));
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "environ_sizes_get", "i(**)", &m3_wasi_unstable_environ_sizes_get)));
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "args_get", "i(**)", &m3_wasi_unstable_args_get)));
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "environ_get", "i(**)", &m3_wasi_unstable_environ_get)));
|
|
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "fd_prestat_dir_name", "i(i*i)", &m3_wasi_unstable_fd_prestat_dir_name)));
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "fd_prestat_get", "i(i*)", &m3_wasi_unstable_fd_prestat_get)));
|
|
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "path_open", "i(ii*iiIIi*)", &m3_wasi_unstable_path_open)));
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "path_filestat_get", "i(ii*i*)", &m3_wasi_unstable_path_filestat_get)));
|
|
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "fd_fdstat_get", "i(i*)", &m3_wasi_unstable_fd_fdstat_get)));
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "fd_fdstat_set_flags", "i(ii)", &m3_wasi_unstable_fd_fdstat_set_flags)));
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "fd_write", "i(i*i*)", &m3_wasi_unstable_fd_write)));
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "fd_read", "i(i*i*)", &m3_wasi_unstable_fd_read)));
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "fd_seek", "i(iIi*)", &m3_wasi_unstable_fd_seek)));
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "fd_datasync", "i(i)", &m3_wasi_unstable_fd_datasync)));
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "fd_close", "i(i)", &m3_wasi_unstable_fd_close)));
|
|
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "random_get", "i(*i)", &m3_wasi_unstable_random_get)));
|
|
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "clock_res_get", "i(i*)", &m3_wasi_unstable_clock_res_get)));
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "clock_time_get", "i(iI*)", &m3_wasi_unstable_clock_time_get)));
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "poll_oneoff", "i(**i*)", &m3_wasi_unstable_poll_oneoff)));
|
|
|
|
_ (SuppressLookupFailure (m3_LinkRawFunction (module, wasi, "proc_exit", "v(i)", &m3_wasi_unstable_proc_exit)));
|
|
|
|
_catch:
|
|
return result;
|
|
}
|
|
|
|
#endif // d_m3HasUVWASI
|
|
|