1 // Copyright 2018 Developers of the Rand project.
2 //
3 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
4 // https://www.apache.org/licenses/LICENSE-2.0> or the MIT license
5 // <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your
6 // option. This file may not be copied, modified, or distributed
7 // except according to those terms.
8 
9 //! Implementation for FreeBSD and NetBSD
10 use crate::{util_libc::sys_fill_exact, Error};
11 use core::ptr;
12 
kern_arnd(buf: &mut [u8]) -> libc::ssize_t13 fn kern_arnd(buf: &mut [u8]) -> libc::ssize_t {
14     static MIB: [libc::c_int; 2] = [libc::CTL_KERN, libc::KERN_ARND];
15     let mut len = buf.len();
16     let ret = unsafe {
17         libc::sysctl(
18             MIB.as_ptr(),
19             MIB.len() as libc::c_uint,
20             buf.as_mut_ptr() as *mut _,
21             &mut len,
22             ptr::null(),
23             0,
24         )
25     };
26     if ret == -1 {
27         -1
28     } else {
29         len as libc::ssize_t
30     }
31 }
32 
getrandom_inner(dest: &mut [u8]) -> Result<(), Error>33 pub fn getrandom_inner(dest: &mut [u8]) -> Result<(), Error> {
34     #[cfg(target_os = "freebsd")]
35     {
36         use crate::util_libc::Weak;
37         static GETRANDOM: Weak = unsafe { Weak::new("getrandom\0") };
38         type GetRandomFn =
39             unsafe extern "C" fn(*mut u8, libc::size_t, libc::c_uint) -> libc::ssize_t;
40 
41         if let Some(fptr) = GETRANDOM.ptr() {
42             let func: GetRandomFn = unsafe { core::mem::transmute(fptr) };
43             return sys_fill_exact(dest, |buf| unsafe { func(buf.as_mut_ptr(), buf.len(), 0) });
44         }
45     }
46     // Both FreeBSD and NetBSD will only return up to 256 bytes at a time, and
47     // older NetBSD kernels will fail on longer buffers.
48     for chunk in dest.chunks_mut(256) {
49         sys_fill_exact(chunk, kern_arnd)?
50     }
51     Ok(())
52 }
53