5c51940100
* Refactor use of Refs and RefMuts in wasi-common This commit refactors the use of `Ref`s and `RefMut`s in `wasi-common`. Now, `Entry` is stored behind an `Rc` inside the `EntryTable`. The `Entry` itself on the other hand now stores rights behind a `RefCell` and the descriptor as `Rc<RefCell<..>>` combo to enable easy reference tracking and interior mutability which is required down the line in a couple of syscalls. In essence, this implies that we no longer have need for mutable accessor to `Entry` from `WasiCtx`, and so all related methods go away (`get_entry_mut`, etc.). While here, I've also simplified handling and aggregating of rights on the `Entry` object. Instead of storing base and inheriting rights as separate fields, they are now aggregated into one struct `EntryRights` which features convenient constructors for each possible combination; i.e., when only base rights are set, or both base and inheriting are set, or both are left as empty. Since we do need to be able to mutate those rights down the line in `fdstat_set_rights` syscall, this object is kept behind a `RefCell` (note no `Rc` since we don't need to pass it around anywhere). The descriptor field in `Entry` is now kept behind `Rc<RefCell<..>>` combo since we not only need to mutate it down the line, but we also need to be able to pass it around (as part of the machinery making `poll_oneoff` work). I've also removed `as_file` and `try_clone` methods on `Descriptor` struct since they were adding more noise than necessary, and making them work with `Rc` was unnecessarily complicated. Finally, I've converted the `get_dir_from_os_handle` function into a method attached to the `OsHandle` itself, called `dir_stream`. IMHO, it makes more sense to have it there directly as a method than as a separate function. * Use Cell for types that are Copy
47 lines
1.4 KiB
Rust
47 lines
1.4 KiB
Rust
use crate::wasi::Result;
|
|
use std::fs;
|
|
use std::ops::Deref;
|
|
use std::os::unix::prelude::{AsRawFd, RawFd};
|
|
use yanix::dir::Dir;
|
|
|
|
#[derive(Debug)]
|
|
pub(crate) struct OsHandle(fs::File);
|
|
|
|
impl OsHandle {
|
|
pub(crate) fn dir_stream(&self) -> Result<Box<Dir>> {
|
|
// We need to duplicate the fd, because `opendir(3)`:
|
|
// After a successful call to fdopendir(), fd is used internally by the implementation,
|
|
// and should not otherwise be used by the application.
|
|
// `opendir(3p)` also says that it's undefined behavior to
|
|
// modify the state of the fd in a different way than by accessing DIR*.
|
|
//
|
|
// Still, rewinddir will be needed because the two file descriptors
|
|
// share progress. But we can safely execute closedir now.
|
|
let fd = self.0.try_clone()?;
|
|
// TODO This doesn't look very clean. Can we do something about it?
|
|
// Boxing is needed here in order to satisfy `yanix`'s trait requirement for the `DirIter`
|
|
// where `T: Deref<Target = Dir>`.
|
|
Ok(Box::new(Dir::from(fd)?))
|
|
}
|
|
}
|
|
|
|
impl From<fs::File> for OsHandle {
|
|
fn from(file: fs::File) -> Self {
|
|
Self(file)
|
|
}
|
|
}
|
|
|
|
impl AsRawFd for OsHandle {
|
|
fn as_raw_fd(&self) -> RawFd {
|
|
self.0.as_raw_fd()
|
|
}
|
|
}
|
|
|
|
impl Deref for OsHandle {
|
|
type Target = fs::File;
|
|
|
|
fn deref(&self) -> &Self::Target {
|
|
&self.0
|
|
}
|
|
}
|