atsamd-hal 0.23.3

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)
    }
}

impl CryptoRng for Trng {}

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