64 lines
1.4 KiB
Rust
64 lines
1.4 KiB
Rust
|
use marker::PhantomData;
|
||
|
|
use slice;
|
||
|
|
use sys::c;
|
||
|
|
|
||
|
|
#[repr(transparent)]
|
||
|
|
pub struct IoVec<'a> {
|
||
|
|
vec: c::WSABUF,
|
||
|
|
_p: PhantomData<&'a [u8]>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<'a> IoVec<'a> {
|
||
|
|
#[inline]
|
||
|
|
pub fn new(buf: &'a [u8]) -> IoVec<'a> {
|
||
|
|
assert!(buf.len() <= c::ULONG::max_value() as usize);
|
||
|
|
IoVec {
|
||
|
|
vec: c::WSABUF {
|
||
|
|
len: buf.len() as c::ULONG,
|
||
|
|
buf: buf.as_ptr() as *mut u8 as *mut c::CHAR,
|
||
|
|
},
|
||
|
|
_p: PhantomData,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[inline]
|
||
|
|
pub fn as_slice(&self) -> &'a [u8] {
|
||
|
|
unsafe {
|
||
|
|
slice::from_raw_parts(self.vec.buf as *mut u8, self.vec.len as usize)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub struct IoVecMut<'a> {
|
||
|
|
vec: c::WSABUF,
|
||
|
|
_p: PhantomData<&'a mut [u8]>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<'a> IoVecMut<'a> {
|
||
|
|
#[inline]
|
||
|
|
pub fn new(buf: &'a mut [u8]) -> IoVecMut<'a> {
|
||
|
|
assert!(buf.len() <= c::ULONG::max_value() as usize);
|
||
|
|
IoVecMut {
|
||
|
|
vec: c::WSABUF {
|
||
|
|
len: buf.len() as c::ULONG,
|
||
|
|
buf: buf.as_mut_ptr() as *mut c::CHAR,
|
||
|
|
},
|
||
|
|
_p: PhantomData,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[inline]
|
||
|
|
pub fn as_slice(&self) -> &'a [u8] {
|
||
|
|
unsafe {
|
||
|
|
slice::from_raw_parts(self.vec.buf as *mut u8, self.vec.len as usize)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[inline]
|
||
|
|
pub fn as_mut_slice(&mut self) -> &'a mut [u8] {
|
||
|
|
unsafe {
|
||
|
|
slice::from_raw_parts_mut(self.vec.buf as *mut u8, self.vec.len as usize)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|