yggdrasil/getrandom
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

freebsd: Increase consistancy with libc implementation (#36)

Browse Source
This commit is contained in:
Joseph Richey 2019-06-19 02:54:51 -07:00 committed by Artyom Pavlov
parent 0b87a0974b
commit a91b60b475
1 changed files with 23 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/freebsd.rs
View File

@ -14,21 +14,30 @@ use std::io;
use core::ptr;
use core::num::NonZeroU32;
fn kern_arnd(buf: &mut [u8]) -> Result<usize, Error> {
static MIB: [libc::c_int; 2] = [libc::CTL_KERN, libc::KERN_ARND];
let mut len = buf.len();
let ret = unsafe {
libc::sysctl(
MIB.as_ptr(),
MIB.len() as libc::c_uint,
buf.as_mut_ptr() as *mut _,
&mut len,
ptr::null(),
0,
)
};
if ret == -1 {
error!("freebsd: kern.arandom syscall failed");
return Err(io::Error::last_os_error().into());
}
Ok(len)
}
pub fn getrandom_inner(dest: &mut [u8]) -> Result<(), Error> {
let mib = [libc::CTL_KERN, libc::KERN_ARND];
// kern.arandom permits a maximum buffer size of 256 bytes
for chunk in dest.chunks_mut(256) {
let mut len = chunk.len();
let ret = unsafe {
libc::sysctl(
mib.as_ptr(), mib.len() as libc::c_uint,
chunk.as_mut_ptr() as *mut _, &mut len, ptr::null(), 0,
)
};
if ret == -1 || len != chunk.len() {
error!("freebsd: kern.arandom syscall failed");
return Err(io::Error::last_os_error().into());
}
let mut start = 0;
while start < dest.len() {
start += kern_arnd(&mut dest[start..])?;
}
Ok(())
}