rand_os/src/netbsd.rs

// Copyright 2018 Developers of the Rand project.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// https://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

//! Implementation for NetBSD

use rand_core::Error;
use super::random_device;
use super::OsRngImpl;

use std::fs::File;
use std::io::Read;
use std::sync::atomic::{AtomicBool, ATOMIC_BOOL_INIT, Ordering};

#[derive(Clone, Debug)]
pub struct OsRng { initialized: bool }

impl OsRngImpl for OsRng {
    fn new() -> Result<OsRng, Error> {
        random_device::open("/dev/urandom", &|p| File::open(p))?;
        Ok(OsRng { initialized: false })
    }

    fn fill_chunk(&mut self, dest: &mut [u8]) -> Result<(), Error> {
        random_device::read(dest)
    }

    // Read a single byte from `/dev/random` to determine if the OS RNG is
    // already seeded. NetBSD always blocks if not yet ready.
    fn test_initialized(&mut self, dest: &mut [u8], _blocking: bool)
        -> Result<usize, Error>
    {
        static OS_RNG_INITIALIZED: AtomicBool = ATOMIC_BOOL_INIT;
        if !self.initialized {
            self.initialized = OS_RNG_INITIALIZED.load(Ordering::Relaxed);
        }
        if self.initialized { return Ok(0); }

        info!("OsRng: testing random device /dev/random");
        let mut file =
            File::open("/dev/random").map_err(random_device::map_err)?;
        file.read(&mut dest[..1]).map_err(random_device::map_err)?;

        OS_RNG_INITIALIZED.store(true, Ordering::Relaxed);
        self.initialized = true;
        Ok(1)
    }

    fn method_str(&self) -> &'static str { "/dev/urandom" }
}