T0b1/wasmtime
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Modify fd_prestat to hold C-str together with its length

Browse Source
This commit is contained in:
Jakub Konka
2019-05-01 17:05:59 +02:00
committed by Dan Gohman
parent f7245f40a1
commit d6b2faeeb4
3 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
wasmtime-wasi/sandboxed-system-primitives/src/posix.c
View File

@@ -274,8 +274,10 @@ static bool fd_prestats_grow(
return false;
// Mark all new file descriptors as unused.
for (size_t i = pt->size; i < size; ++i)
prestats[i].dir = NULL;
for (size_t i = pt->size; i < size; ++i) {
prestats[i].dir_name = NULL;
prestats[i].dir_name_len = 0;
}
pt->prestats = prestats;
pt->size = size;
}
@@ -295,7 +297,8 @@ bool fd_prestats_insert(
return false;
}
pt->prestats[fd].dir = strdup(dir);
pt->prestats[fd].dir_name = strdup(dir);
pt->prestats[fd].dir_name_len = strlen(dir);
rwlock_unlock(&pt->lock);
return true;
}
@@ -310,7 +313,7 @@ static __wasi_errno_t fd_prestats_get_entry(
if (fd >= pt->size)
return __WASI_EBADF;
struct fd_prestat *prestat = &pt->prestats[fd];
if (prestat->dir == NULL)
if (prestat->dir_name == NULL)
return __WASI_EBADF;
*ret = prestat;
@@ -664,12 +667,12 @@ __wasi_errno_t wasmtime_ssp_fd_prestat_dir_name(
rwlock_unlock(&prestats->lock);
return error;
}
if (path_len != strlen(prestat->dir)) {
if (path_len != prestat->dir_name_len) {
rwlock_unlock(&prestats->lock);
return EINVAL;
}
memcpy(path, prestat->dir, path_len);
memcpy(path, prestat->dir_name, path_len);
rwlock_unlock(&prestats->lock);

3
wasmtime-wasi/sandboxed-system-primitives/src/posix.h
View File

@@ -21,7 +21,8 @@ struct fd_entry;
struct syscalls;
struct fd_prestat {
const char *dir;
const char *dir_name;
size_t dir_name_len;
};
struct fd_table {

7
wasmtime-wasi/src/host_impls.rs
View File

@@ -98,7 +98,7 @@ fn fd_prestats_get_entry(
}
let prestat = unsafe { &*pt.prestats.add(fd as usize) };
if prestat.dir == ::std::ptr::null() {
if prestat.dir_name == ::std::ptr::null() {
return None;
}
@@ -133,10 +133,9 @@ pub fn wasmtime_ssp_fd_prestat_get(
rwlock_rdlock!(prestats);
let ret_code = if let Some(prestat) = fd_prestats_get_entry(prestats, fd) {
(*buf).pr_type = host::__WASI_PREOPENTYPE_DIR;
buf.pr_type = host::__WASI_PREOPENTYPE_DIR;
unsafe {
let dir_name = ::std::ffi::CStr::from_ptr((*prestat).dir).to_str().unwrap();
(*buf).u.dir.pr_name_len = dir_name.len();
buf.u.dir.pr_name_len = prestat.dir_name_len;
}
host::__WASI_ESUCCESS
} else {