1 | use super::{AlgorithmName, XofReaderCore}; |
2 | use crate::XofReader; |
3 | use block_buffer::EagerBuffer; |
4 | use core::fmt; |
5 | use crypto_common::typenum::{IsLess, Le, NonZero, U256}; |
6 | |
7 | /// Wrapper around [`XofReaderCore`] implementations. |
8 | /// |
9 | /// It handles data buffering and implements the mid-level traits. |
10 | #[derive (Clone, Default)] |
11 | pub struct XofReaderCoreWrapper<T> |
12 | where |
13 | T: XofReaderCore, |
14 | T::BlockSize: IsLess<U256>, |
15 | Le<T::BlockSize, U256>: NonZero, |
16 | { |
17 | pub(super) core: T, |
18 | pub(super) buffer: EagerBuffer<T::BlockSize>, |
19 | } |
20 | |
21 | impl<T> fmt::Debug for XofReaderCoreWrapper<T> |
22 | where |
23 | T: XofReaderCore + AlgorithmName, |
24 | T::BlockSize: IsLess<U256>, |
25 | Le<T::BlockSize, U256>: NonZero, |
26 | { |
27 | fn fmt(&self, f: &mut fmt::Formatter<'_>) -> Result<(), fmt::Error> { |
28 | T::write_alg_name(f)?; |
29 | f.write_str(data:" { .. }" ) |
30 | } |
31 | } |
32 | |
33 | impl<T> XofReader for XofReaderCoreWrapper<T> |
34 | where |
35 | T: XofReaderCore, |
36 | T::BlockSize: IsLess<U256>, |
37 | Le<T::BlockSize, U256>: NonZero, |
38 | { |
39 | #[inline ] |
40 | fn read(&mut self, buffer: &mut [u8]) { |
41 | let Self { core: &mut T, buffer: buf: &mut BlockBuffer<::BlockSize, …> } = self; |
42 | buf.set_data(data:buffer, |blocks: &mut [GenericArray::BlockSize>]| { |
43 | for block: &mut GenericArray::BlockSize> in blocks { |
44 | *block = core.read_block(); |
45 | } |
46 | }); |
47 | } |
48 | } |
49 | |
50 | #[cfg (feature = "std" )] |
51 | #[cfg_attr (docsrs, doc(cfg(feature = "std" )))] |
52 | impl<T> std::io::Read for XofReaderCoreWrapper<T> |
53 | where |
54 | T: XofReaderCore, |
55 | T::BlockSize: IsLess<U256>, |
56 | Le<T::BlockSize, U256>: NonZero, |
57 | { |
58 | #[inline ] |
59 | fn read(&mut self, buf: &mut [u8]) -> std::io::Result<usize> { |
60 | XofReader::read(self, buffer:buf); |
61 | Ok(buf.len()) |
62 | } |
63 | } |
64 | |