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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

macro_rules! impl_state {
    ($state:ident, $rate:ident, $padding:ty, $doc:expr) => {

        #[allow(non_camel_case_types)]
        #[derive(Clone, Default)]
        #[doc=$doc]
        pub struct $state {
            state: Sha3State,
            buffer: BlockBuffer<$rate>,
        }

        impl $state {
            fn absorb(&mut self, input: &[u8]) {
                let s = &mut self.state;
                self.buffer.input_block(input, |b| s.absorb_block(b));
            }

            fn apply_padding(&mut self) {
                let buf = self.buffer.pad_with::<$padding>()
                    .expect("we never use input_lazy");
                self.state.absorb_block(buf);
            }
        }
    }
}

macro_rules! sha3_impl {
    ($state:ident, $output_size:ident, $rate:ident, $padding:ty, $doc:expr) => {
        impl_state!($state, $rate, $padding, $doc);

        impl BlockInput for $state {
            type BlockSize = $rate;
        }

        impl Update for $state {
            fn update(&mut self, input: impl AsRef<[u8]>) {
                self.absorb(input.as_ref())
            }
        }

        impl FixedOutputDirty for $state {
            type OutputSize = $output_size;

            fn finalize_into_dirty(&mut self, out: &mut digest::Output<Self>) {
                self.apply_padding();

                let n = out.len();
                self.state.as_bytes(|state| {
                    out.copy_from_slice(&state[..n]);
                });
            }
        }

        impl Reset for $state {
            fn reset(&mut self) {
                self.state = Default::default();
                self.buffer.reset();
            }
        }

        opaque_debug::implement!($state);
        digest::impl_write!($state);
    };
}

macro_rules! shake_impl {
    ($state:ident, $rate:ident, $padding:ty, $doc:expr) => {
        impl_state!($state, $rate, $padding, $doc);

        impl Update for $state {
            fn update(&mut self, input: impl AsRef<[u8]>) {
                self.absorb(input.as_ref())
            }
        }

        impl ExtendableOutputDirty for $state {
            type Reader = Sha3XofReader;

            fn finalize_xof_dirty(&mut self) -> Sha3XofReader {
                self.apply_padding();
                let r = $rate::to_usize();
                Sha3XofReader::new(self.state.clone(), r)
            }
        }

        impl Reset for $state {
            fn reset(&mut self) {
                self.state = Default::default();
                self.buffer.reset();
            }
        }

        opaque_debug::implement!($state);
        digest::impl_write!($state);
    };
}