aboutsummaryrefslogtreecommitdiff
path: root/rand/rand_os/src/netbsd.rs
blob: cf4b6c74f5db34bfa57b5a6058f8b1670f3b6ee7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
// 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" }
}