atsamd_hal/peripherals/
trng.rs

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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
use crate::pac::{self, Mclk};

use rand_core::{CryptoRng, RngCore};

use crate::ehal_02::blocking::rng::Read;

pub struct Trng(pac::Trng);

impl Trng {
    pub fn new(mclk: &mut Mclk, trng: pac::Trng) -> Trng {
        mclk.apbcmask().modify(|_, w| w.trng_().set_bit());
        trng.ctrla().modify(|_, w| w.enable().set_bit());
        Self(trng)
    }

    pub fn random(&self, buf: &mut [u8]) {
        for chunk in buf.chunks_mut(4) {
            chunk.copy_from_slice(&self.random_u32().to_le_bytes()[..chunk.len()]);
        }
    }

    pub fn random_u8(&self) -> u8 {
        self.random_u32() as u8
    }

    pub fn random_u16(&self) -> u16 {
        self.random_u32() as u16
    }

    pub fn random_u32(&self) -> u32 {
        while self.0.intflag().read().datardy().bit_is_clear() {}
        self.0.data().read().bits()
    }

    pub fn random_u64(&self) -> u64 {
        while self.0.intflag().read().datardy().bit_is_clear() {}
        let lower_half = self.0.data().read().bits() as u64;
        while self.0.intflag().read().datardy().bit_is_clear() {}
        let upper_half = self.0.data().read().bits() as u64;
        (upper_half << 32) | lower_half
    }
}

impl RngCore for Trng {
    fn next_u32(&mut self) -> u32 {
        self.random_u32()
    }

    fn next_u64(&mut self) -> u64 {
        self.random_u64()
    }

    fn fill_bytes(&mut self, dest: &mut [u8]) {
        self.random(dest)
    }

    fn try_fill_bytes(&mut self, dest: &mut [u8]) -> Result<(), rand_core::Error> {
        self.fill_bytes(dest);
        Ok(())
    }
}

impl CryptoRng for Trng {}

impl Read for Trng {
    type Error = ();
    fn read(&mut self, buffer: &mut [u8]) -> Result<(), Self::Error> {
        self.random(buffer);
        Ok(())
    }
}