diff --git a/crates/wasi-common/src/virtfs/pipe.rs b/crates/wasi-common/src/virtfs/pipe.rs index 3067172f12..ef253b72ae 100644 --- a/crates/wasi-common/src/virtfs/pipe.rs +++ b/crates/wasi-common/src/virtfs/pipe.rs @@ -12,9 +12,8 @@ use crate::handle::{Handle, HandleRights}; use crate::wasi::{types, Errno, Result}; use std::any::Any; -use std::cell::{Cell, Ref, RefCell}; use std::io::{self, Read, Write}; -use std::rc::Rc; +use std::sync::{Arc, RwLock}; /// A virtual pipe read end. /// @@ -27,10 +26,19 @@ use std::rc::Rc; /// let stdin = ReadPipe::from("hello from stdin!"); /// ctx.stdin(stdin); /// ``` -#[derive(Clone, Debug)] +#[derive(Debug)] pub struct ReadPipe { - rights: Cell, - reader: Rc>, + rights: RwLock, + reader: Arc>, +} + +impl Clone for ReadPipe { + fn clone(&self) -> Self { + Self { + rights: RwLock::new(*self.rights.read().unwrap()), + reader: self.reader.clone(), + } + } } impl ReadPipe { @@ -38,16 +46,16 @@ impl ReadPipe { /// /// All `Handle` read operations delegate to reading from this underlying reader. pub fn new(r: R) -> Self { - Self::from_shared(Rc::new(RefCell::new(r))) + Self::from_shared(Arc::new(RwLock::new(r))) } /// Create a new pipe from a shareable `Read` type. /// /// All `Handle` read operations delegate to reading from this underlying reader. - pub fn from_shared(reader: Rc>) -> Self { + pub fn from_shared(reader: Arc>) -> Self { use types::Rights; Self { - rights: Cell::new(HandleRights::from_base( + rights: RwLock::new(HandleRights::from_base( Rights::FD_DATASYNC | Rights::FD_FDSTAT_SET_FLAGS | Rights::FD_READ @@ -63,8 +71,8 @@ impl ReadPipe { /// /// This will fail with `Err(self)` if multiple references to the underlying `R` exist. pub fn try_into_inner(mut self) -> std::result::Result { - match Rc::try_unwrap(self.reader) { - Ok(rc) => Ok(RefCell::into_inner(rc)), + match Arc::try_unwrap(self.reader) { + Ok(rc) => Ok(RwLock::into_inner(rc).unwrap()), Err(reader) => { self.reader = reader; Err(self) @@ -103,10 +111,7 @@ impl Handle for ReadPipe { } fn try_clone(&self) -> io::Result> { - Ok(Box::new(Self { - rights: self.rights.clone(), - reader: self.reader.clone(), - })) + Ok(Box::new(self.clone())) } fn get_file_type(&self) -> types::Filetype { @@ -114,11 +119,11 @@ impl Handle for ReadPipe { } fn get_rights(&self) -> HandleRights { - self.rights.get() + *self.rights.read().unwrap() } fn set_rights(&self, rights: HandleRights) { - self.rights.set(rights) + *self.rights.write().unwrap() = rights; } fn advise( @@ -161,7 +166,7 @@ impl Handle for ReadPipe { if offset != 0 { return Err(Errno::Spipe); } - Ok(self.reader.borrow_mut().read_vectored(buf)?) + Ok(self.reader.write().unwrap().read_vectored(buf)?) } fn seek(&self, _offset: io::SeekFrom) -> Result { @@ -169,7 +174,7 @@ impl Handle for ReadPipe { } fn read_vectored(&self, iovs: &mut [io::IoSliceMut]) -> Result { - Ok(self.reader.borrow_mut().read_vectored(iovs)?) + Ok(self.reader.write().unwrap().read_vectored(iovs)?) } fn create_directory(&self, _path: &str) -> Result<()> { @@ -223,10 +228,19 @@ impl Handle for ReadPipe { } /// A virtual pipe write end. -#[derive(Clone, Debug)] +#[derive(Debug)] pub struct WritePipe { - rights: Cell, - writer: Rc>, + rights: RwLock, + writer: Arc>, +} + +impl Clone for WritePipe { + fn clone(&self) -> Self { + Self { + rights: RwLock::new(*self.rights.read().unwrap()), + writer: self.writer.clone(), + } + } } impl WritePipe { @@ -234,16 +248,16 @@ impl WritePipe { /// /// All `Handle` write operations delegate to writing to this underlying writer. pub fn new(w: W) -> Self { - Self::from_shared(Rc::new(RefCell::new(w))) + Self::from_shared(Arc::new(RwLock::new(w))) } /// Create a new pipe from a shareable `Write` type. /// /// All `Handle` write operations delegate to writing to this underlying writer. - pub fn from_shared(writer: Rc>) -> Self { + pub fn from_shared(writer: Arc>) -> Self { use types::Rights; Self { - rights: Cell::new(HandleRights::from_base( + rights: RwLock::new(HandleRights::from_base( Rights::FD_DATASYNC | Rights::FD_FDSTAT_SET_FLAGS | Rights::FD_SYNC @@ -259,8 +273,8 @@ impl WritePipe { /// /// This will fail with `Err(self)` if multiple references to the underlying `W` exist. pub fn try_into_inner(mut self) -> std::result::Result { - match Rc::try_unwrap(self.writer) { - Ok(rc) => Ok(RefCell::into_inner(rc)), + match Arc::try_unwrap(self.writer) { + Ok(rc) => Ok(RwLock::into_inner(rc).unwrap()), Err(writer) => { self.writer = writer; Err(self) @@ -274,11 +288,6 @@ impl WritePipe>> { pub fn new_in_memory() -> Self { Self::new(io::Cursor::new(vec![])) } - - /// Get a reference to the bytes contained in the underlying `Vec` buffer. - pub fn as_slice(&self) -> Ref<[u8]> { - Ref::map(self.writer.borrow(), |c| c.get_ref().as_slice()) - } } impl Handle for WritePipe { @@ -287,10 +296,7 @@ impl Handle for WritePipe { } fn try_clone(&self) -> io::Result> { - Ok(Box::new(Self { - rights: self.rights.clone(), - writer: self.writer.clone(), - })) + Ok(Box::new(self.clone())) } fn get_file_type(&self) -> types::Filetype { @@ -298,11 +304,11 @@ impl Handle for WritePipe { } fn get_rights(&self) -> HandleRights { - self.rights.get() + *self.rights.read().unwrap() } fn set_rights(&self, rights: HandleRights) { - self.rights.set(rights) + *self.rights.write().unwrap() = rights; } fn advise( @@ -345,7 +351,7 @@ impl Handle for WritePipe { if offset != 0 { return Err(Errno::Spipe); } - Ok(self.writer.borrow_mut().write_vectored(buf)?) + Ok(self.writer.write().unwrap().write_vectored(buf)?) } fn seek(&self, _offset: io::SeekFrom) -> Result { @@ -353,7 +359,7 @@ impl Handle for WritePipe { } fn write_vectored(&self, iovs: &[io::IoSlice]) -> Result { - Ok(self.writer.borrow_mut().write_vectored(iovs)?) + Ok(self.writer.write().unwrap().write_vectored(iovs)?) } fn create_directory(&self, _path: &str) -> Result<()> {