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
use crate::digest::Digest;
use crate::hashing::keccak;
use crate::hashing::sha3;
macro_rules! digest {
($name: ident, $m:ident, $ctx: ident) => {
#[derive(Clone)]
pub struct $name {
ctx: $m::$ctx,
computed: bool,
}
impl $name {
pub const fn new() -> Self {
Self {
ctx: $m::$ctx::new(),
computed: false,
}
}
}
impl Digest for $name {
fn reset(&mut self) {
self.ctx.reset();
self.computed = false;
}
fn input(&mut self, msg: &[u8]) {
assert!(!self.computed, "context is already finalized, needs reset");
self.ctx.update_mut(msg);
}
fn result(&mut self, slice: &mut [u8]) {
assert!(!self.computed, "context is already finalized, needs reset");
self.computed = true;
slice.copy_from_slice(&self.ctx.finalize_reset());
}
fn output_bits(&self) -> usize {
$m::$name::OUTPUT_BITS
}
fn block_size(&self) -> usize {
$m::$name::BLOCK_BYTES
}
}
};
}
digest!(Sha3_512, sha3, Context512);
digest!(Sha3_384, sha3, Context384);
digest!(Sha3_256, sha3, Context256);
digest!(Sha3_224, sha3, Context224);
digest!(Keccak512, keccak, Context512);
digest!(Keccak384, keccak, Context384);
digest!(Keccak256, keccak, Context256);
digest!(Keccak224, keccak, Context224);