diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 14:29:10 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 14:29:10 +0000 |
commit | 2aa4a82499d4becd2284cdb482213d541b8804dd (patch) | |
tree | b80bf8bf13c3766139fbacc530efd0dd9d54394c /third_party/rust/parity-wasm/src/io.rs | |
parent | Initial commit. (diff) | |
download | firefox-2aa4a82499d4becd2284cdb482213d541b8804dd.tar.xz firefox-2aa4a82499d4becd2284cdb482213d541b8804dd.zip |
Adding upstream version 86.0.1.upstream/86.0.1upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/rust/parity-wasm/src/io.rs')
-rw-r--r-- | third_party/rust/parity-wasm/src/io.rs | 122 |
1 files changed, 122 insertions, 0 deletions
diff --git a/third_party/rust/parity-wasm/src/io.rs b/third_party/rust/parity-wasm/src/io.rs new file mode 100644 index 0000000000..c027b7102a --- /dev/null +++ b/third_party/rust/parity-wasm/src/io.rs @@ -0,0 +1,122 @@ +//! Simple abstractions for the IO operations. +//! +//! Basically it just a replacement for the std::io that is usable from +//! the `no_std` environment. + +#[cfg(feature="std")] +use std::io; + +/// IO specific error. +#[derive(Debug)] +pub enum Error { + /// Some unexpected data left in the buffer after reading all data. + TrailingData, + + /// Unexpected End-Of-File + UnexpectedEof, + + /// Invalid data is encountered. + InvalidData, + + #[cfg(feature = "std")] + IoError(std::io::Error), +} + +/// IO specific Result. +pub type Result<T> = core::result::Result<T, Error>; + +pub trait Write { + /// Write a buffer of data into this write. + /// + /// All data is written at once. + fn write(&mut self, buf: &[u8]) -> Result<()>; +} + +pub trait Read { + /// Read a data from this read to a buffer. + /// + /// If there is not enough data in this read then `UnexpectedEof` will be returned. + fn read(&mut self, buf: &mut [u8]) -> Result<()>; +} + +/// Reader that saves the last position. +pub struct Cursor<T> { + inner: T, + pos: usize, +} + +impl<T> Cursor<T> { + pub fn new(inner: T) -> Cursor<T> { + Cursor { + inner, + pos: 0, + } + } + + pub fn position(&self) -> usize { + self.pos + } +} + +impl<T: AsRef<[u8]>> Read for Cursor<T> { + fn read(&mut self, buf: &mut [u8]) -> Result<()> { + let slice = self.inner.as_ref(); + let remainder = slice.len() - self.pos; + let requested = buf.len(); + if requested > remainder { + return Err(Error::UnexpectedEof); + } + buf.copy_from_slice(&slice[self.pos..(self.pos + requested)]); + self.pos += requested; + Ok(()) + } +} + +#[cfg(not(feature = "std"))] +impl Write for alloc::vec::Vec<u8> { + fn write(&mut self, buf: &[u8]) -> Result<()> { + self.extend(buf); + Ok(()) + } +} + +#[cfg(feature = "std")] +impl<T: io::Read> Read for T { + fn read(&mut self, buf: &mut [u8]) -> Result<()> { + self.read_exact(buf) + .map_err(Error::IoError) + } +} + +#[cfg(feature = "std")] +impl<T: io::Write> Write for T { + fn write(&mut self, buf: &[u8]) -> Result<()> { + self.write_all(buf).map_err(Error::IoError) + } +} + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn cursor() { + let mut cursor = Cursor::new(vec![0xFFu8, 0x7Fu8]); + assert_eq!(cursor.position(), 0); + + let mut buf = [0u8]; + assert!(cursor.read(&mut buf[..]).is_ok()); + assert_eq!(cursor.position(), 1); + assert_eq!(buf[0], 0xFFu8); + assert!(cursor.read(&mut buf[..]).is_ok()); + assert_eq!(buf[0], 0x7Fu8); + assert_eq!(cursor.position(), 2); + } + + #[test] + fn overflow_in_cursor() { + let mut cursor = Cursor::new(vec![0u8]); + let mut buf = [0, 1, 2]; + assert!(cursor.read(&mut buf[..]).is_err()); + } +} |