library/std/src/sys/windows/io.rs

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