src/libstd/sys/wasi/io.rs

   1 use crate::marker::PhantomData;
   2 use crate::slice;
   3
   4 #[derive(Copy, Clone)]
   5 #[repr(transparent)]
   6 pub struct IoSlice<'a> {
   7     vec: wasi::Ciovec,
   8     _p: PhantomData<&'a [u8]>,
   9 }
  10
  11 impl<'a> IoSlice<'a> {
  12     #[inline]
  13     pub fn new(buf: &'a [u8]) -> IoSlice<'a> {
  14         IoSlice { vec: wasi::Ciovec { buf: buf.as_ptr(), buf_len: buf.len() }, _p: PhantomData }
  15     }
  16
  17     #[inline]
  18     pub fn advance(&mut self, n: usize) {
  19         if self.vec.buf_len < n {
  20             panic!("advancing IoSlice beyond its length");
  21         }
  22
  23         unsafe {
  24             self.vec.buf_len -= n;
  25             self.vec.buf = self.vec.buf.add(n);
  26         }
  27     }
  28
  29     #[inline]
  30     pub fn as_slice(&self) -> &[u8] {
  31         unsafe { slice::from_raw_parts(self.vec.buf as *const u8, self.vec.buf_len) }
  32     }
  33 }
  34
  35 #[repr(transparent)]
  36 pub struct IoSliceMut<'a> {
  37     vec: wasi::Iovec,
  38     _p: PhantomData<&'a mut [u8]>,
  39 }
  40
  41 impl<'a> IoSliceMut<'a> {
  42     #[inline]
  43     pub fn new(buf: &'a mut [u8]) -> IoSliceMut<'a> {
  44         IoSliceMut {
  45             vec: wasi::Iovec { buf: buf.as_mut_ptr(), buf_len: buf.len() },
  46             _p: PhantomData,
  47         }
  48     }
  49
  50     #[inline]
  51     pub fn advance(&mut self, n: usize) {
  52         if self.vec.buf_len < n {
  53             panic!("advancing IoSlice beyond its length");
  54         }
  55
  56         unsafe {
  57             self.vec.buf_len -= n;
  58             self.vec.buf = self.vec.buf.add(n);
  59         }
  60     }
  61
  62     #[inline]
  63     pub fn as_slice(&self) -> &[u8] {
  64         unsafe { slice::from_raw_parts(self.vec.buf as *const u8, self.vec.buf_len) }
  65     }
  66
  67     #[inline]
  68     pub fn as_mut_slice(&mut self) -> &mut [u8] {
  69         unsafe { slice::from_raw_parts_mut(self.vec.buf as *mut u8, self.vec.buf_len) }
  70     }
  71 }