provekit-whir 0.1.1

An implementation of the WHIR polynomial commitment scheme
Documentation 
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

#![cfg(test)]

use super::DuplexSpongeInterface;

#[derive(Clone, Debug)]
pub struct MockSponge<'a> {
    pub absorb: Option<&'a [u8]>,
    pub squeeze: &'a [u8],
}

impl DuplexSpongeInterface for MockSponge<'_> {
    type U = u8;

    fn absorb(&mut self, input: &[Self::U]) -> &mut Self {
        if let Some(absorb) = self.absorb.as_mut() {
            assert!(&absorb[..input.len()] == input);
            *absorb = &absorb[input.len()..];
        }
        self
    }

    fn squeeze(&mut self, output: &mut [Self::U]) -> &mut Self {
        output.copy_from_slice(&self.squeeze[..output.len()]);
        self.squeeze = &self.squeeze[output.len()..];
        self
    }

    fn ratchet(&mut self) -> &mut Self {
        if let Some(absorb) = self.absorb.as_mut() {
            assert!(&absorb[..7] == b"RATCHET");
            *absorb = &absorb[7..];
        }
        assert!(&self.squeeze[..7] == b"RATCHET");
        self.squeeze = &self.squeeze[7..];
        self
    }
}