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
97
98
99
100
101
102
103
104
105
106
107
use crate::{
crh::PedersenCRHParameters,
errors::CRHError,
traits::{CRHParameters, CRH},
};
use bitvec::{order::Lsb0, view::BitView};
use snarkvm_curves::Group;
use snarkvm_fields::{ConstraintFieldError, Field, ToConstraintField};
use rand::Rng;
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
pub struct PedersenCRH<G: Group, const NUM_WINDOWS: usize, const WINDOW_SIZE: usize> {
pub parameters: PedersenCRHParameters<G, NUM_WINDOWS, WINDOW_SIZE>,
}
impl<G: Group, const NUM_WINDOWS: usize, const WINDOW_SIZE: usize> CRH for PedersenCRH<G, NUM_WINDOWS, WINDOW_SIZE> {
type Output = G;
type Parameters = PedersenCRHParameters<G, NUM_WINDOWS, WINDOW_SIZE>;
const INPUT_SIZE_BITS: usize = WINDOW_SIZE * NUM_WINDOWS;
fn setup<R: Rng>(rng: &mut R) -> Self {
Self {
parameters: PedersenCRHParameters::setup(rng),
}
}
fn hash(&self, input: &[u8]) -> Result<Self::Output, CRHError> {
if (input.len() * 8) > WINDOW_SIZE * NUM_WINDOWS {
return Err(CRHError::IncorrectInputLength(input.len(), WINDOW_SIZE, NUM_WINDOWS));
}
let mut input = input;
let mut padded_input = vec![];
if (input.len() * 8) < WINDOW_SIZE * NUM_WINDOWS {
padded_input.extend_from_slice(input);
padded_input.resize((WINDOW_SIZE * NUM_WINDOWS) / 8, 0u8);
input = padded_input.as_slice();
}
if self.parameters.bases.len() != NUM_WINDOWS {
return Err(CRHError::IncorrectParameterSize(
self.parameters.bases[0].len(),
self.parameters.bases.len(),
WINDOW_SIZE,
NUM_WINDOWS,
));
}
let bits = input.view_bits::<Lsb0>();
let result = bits
.chunks(WINDOW_SIZE)
.zip(&self.parameters.bases)
.map(|(bits, powers)| {
let mut encoded = G::zero();
for (bit, base) in bits.iter().zip(powers.iter()) {
if *bit {
encoded += base;
}
}
encoded
})
.fold(G::zero(), |a, b| a + b);
Ok(result)
}
fn parameters(&self) -> &Self::Parameters {
&self.parameters
}
}
impl<G: Group, const NUM_WINDOWS: usize, const WINDOW_SIZE: usize>
From<PedersenCRHParameters<G, NUM_WINDOWS, WINDOW_SIZE>> for PedersenCRH<G, NUM_WINDOWS, WINDOW_SIZE>
{
fn from(parameters: PedersenCRHParameters<G, NUM_WINDOWS, WINDOW_SIZE>) -> Self {
Self { parameters }
}
}
impl<F: Field, G: Group + ToConstraintField<F>, const NUM_WINDOWS: usize, const WINDOW_SIZE: usize> ToConstraintField<F>
for PedersenCRH<G, NUM_WINDOWS, WINDOW_SIZE>
{
#[inline]
fn to_field_elements(&self) -> Result<Vec<F>, ConstraintFieldError> {
self.parameters.to_field_elements()
}
}