src/libstd/sys/vxworks/io.rs

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