src/libstd/sys/vxworks/io.rs

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