src/libstd/sys/unix/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 #[repr(transparent)]
  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 }