bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

Move add_to_ancillary_data and AncillaryDataIter to ancillary.rs

Browse source
This commit is contained in:
LinkTed 2020-08-23 14:50:54 +02:00
parent e6984eee6f
commit 8783b06bd2
2 changed files with 86 additions and 87 deletions
Download patch file Download diff file Expand all files Collapse all files

82
library/std/src/sys/unix/net.rs
View file

@ -1,10 +1,8 @@
use crate::cmp;
use crate::ffi::CStr;
use crate::io::{self, IoSlice, IoSliceMut};
use crate::marker::PhantomData;
use crate::mem;
use crate::net::{Shutdown, SocketAddr};
use crate::ptr::null_mut;
use crate::str;
use crate::sys::fd::FileDesc;
use crate::sys_common::net::{getsockopt, setsockopt, sockaddr_to_addr};
@ -20,86 +18,6 @@ pub extern crate libc as netc;
pub type wrlen_t = size_t;
pub struct AncillaryDataIter<'a, T> {
data: &'a [u8],
phantom: crate::marker::PhantomData<T>,
}
impl<'a, T> AncillaryDataIter<'a, T> {
pub fn new(data: &'a [u8]) -> AncillaryDataIter<'a, T> {
AncillaryDataIter { data, phantom: PhantomData }
}
}
impl<'a, T> Iterator for AncillaryDataIter<'a, T> {
type Item = T;
fn next(&mut self) -> Option<T> {
unsafe {
let mut unit = mem::zeroed();
if mem::size_of::<T>() <= self.data.len() {
let unit_ptr: *mut T = &mut unit;
libc::memcpy(unit_ptr.cast(), self.data.as_ptr().cast(), mem::size_of::<T>());
self.data = &self.data[mem::size_of::<T>()..];
Some(unit)
} else {
None
}
}
}
}
pub fn add_to_ancillary_data<T: core::clone::Clone>(
buffer: &mut [u8],
length: &mut usize,
source: &[T],
cmsg_level: libc::c_int,
cmsg_type: libc::c_int,
) -> bool {
let len = (source.len() * mem::size_of::<T>()) as u32;
unsafe {
let additional_space = libc::CMSG_SPACE(len) as usize;
if *length + additional_space > buffer.len() {
return false;
}
libc::memset(buffer[*length..].as_mut_ptr().cast(), 0, additional_space);
*length += additional_space;
let msg = libc::msghdr {
msg_name: null_mut(),
msg_namelen: 0,
msg_iov: null_mut(),
msg_iovlen: 0,
msg_control: buffer.as_mut_ptr().cast(),
msg_controllen: *length,
msg_flags: 0,
};
let mut cmsg = libc::CMSG_FIRSTHDR(&msg);
let mut previous_cmsg = cmsg;
while !cmsg.is_null() {
previous_cmsg = cmsg;
cmsg = libc::CMSG_NXTHDR(&msg, cmsg);
}
if previous_cmsg.is_null() {
return false;
}
(*previous_cmsg).cmsg_level = cmsg_level;
(*previous_cmsg).cmsg_type = cmsg_type;
(*previous_cmsg).cmsg_len = libc::CMSG_LEN(len) as usize;
let data = libc::CMSG_DATA(previous_cmsg).cast();
libc::memcpy(data, source.as_ptr().cast(), len as usize);
}
true
}
pub struct Socket(FileDesc);
pub fn init() {}