sha3/
state.rs

1use core::convert::TryInto;
2
3const PLEN: usize = 25;
4const DEFAULT_ROUND_COUNT: usize = 24;
5
6#[derive(Clone)]
7pub(crate) struct Sha3State {
8    pub state: [u64; PLEN],
9    round_count: usize,
10}
11
12impl Default for Sha3State {
13    fn default() -> Self {
14        Self {
15            state: [0u64; PLEN],
16            round_count: DEFAULT_ROUND_COUNT,
17        }
18    }
19}
20
21impl Sha3State {
22    pub(crate) fn new(round_count: usize) -> Self {
23        Self {
24            state: [0u64; PLEN],
25            round_count,
26        }
27    }
28
29    #[inline(always)]
30    pub(crate) fn absorb_block(&mut self, block: &[u8]) {
31        debug_assert_eq!(block.len() % 8, 0);
32
33        for (b, s) in block.chunks_exact(8).zip(self.state.iter_mut()) {
34            *s ^= u64::from_le_bytes(b.try_into().unwrap());
35        }
36
37        keccak::p1600(&mut self.state, self.round_count);
38    }
39
40    #[inline(always)]
41    pub(crate) fn as_bytes(&self, out: &mut [u8]) {
42        for (o, s) in out.chunks_mut(8).zip(self.state.iter()) {
43            o.copy_from_slice(&s.to_le_bytes()[..o.len()]);
44        }
45    }
46
47    #[inline(always)]
48    pub(crate) fn permute(&mut self) {
49        keccak::p1600(&mut self.state, self.round_count);
50    }
51}
sha3/state.rs

sha3/
state.rs
