lib-q-prf 0.0.5

Legendre and Gold (power-residue) PRFs over large prime fields (Fiat-Shamir transcript building blocks)
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
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

//! SHAKE256 helpers for domain-separated key derivation.

use crypto_bigint::{
    CtEq,
    NonZero,
    U256,
    U512,
};
use lib_q_sha3::{
    ExtendableOutput,
    Update,
    XofReader,
};

use crate::error::PrfError;

/// Expand XOF output until we obtain a non-zero field element `< p` (rejection sampling).
pub fn shake256_to_field_u256(seed: &[u8], label: &[u8], p: &U256) -> Result<U256, PrfError> {
    let nz_p = NonZero::new(*p)
        .into_option()
        .ok_or(PrfError::InvalidParam)?;
    let mut ctr: u32 = 0;
    loop {
        let mut h = lib_q_sha3::Shake256::default();
        h.update(label);
        h.update(&(ctr as u64).to_le_bytes());
        h.update(seed);
        let mut wide = [0u8; 64];
        let mut r = h.finalize_xof();
        XofReader::read(&mut r, &mut wide);
        let candidate = U256::from_le_slice(&wide[..32]);
        let reduced = candidate.rem_vartime(&nz_p);
        if !bool::from(reduced.ct_eq(&U256::ZERO)) {
            return Ok(reduced);
        }
        ctr = ctr.wrapping_add(1);
        if ctr > 1024 {
            return Err(PrfError::InvalidParam);
        }
    }
}

/// Same for [`U512`] / 64-byte absorb.
pub fn shake256_to_field_u512(seed: &[u8], label: &[u8], p: &U512) -> Result<U512, PrfError> {
    let nz_p = NonZero::new(*p)
        .into_option()
        .ok_or(PrfError::InvalidParam)?;
    let mut ctr: u32 = 0;
    loop {
        let mut h = lib_q_sha3::Shake256::default();
        h.update(label);
        h.update(&(ctr as u64).to_le_bytes());
        h.update(seed);
        let mut wide = [0u8; 128];
        let mut r = h.finalize_xof();
        XofReader::read(&mut r, &mut wide);
        let candidate = U512::from_le_slice(&wide[..64]);
        let reduced = candidate.rem_vartime(&nz_p);
        if !bool::from(reduced.ct_eq(&U512::ZERO)) {
            return Ok(reduced);
        }
        ctr = ctr.wrapping_add(1);
        if ctr > 1024 {
            return Err(PrfError::InvalidParam);
        }
    }
}