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

use libzeropool::{
    fawkes_crypto::ff_uint::PrimeField,
    fawkes_crypto::ff_uint::{Num, NumRepr, Uint},
    native::key::{derive_key_a, derive_key_eta},
    native::params::PoolParams,
};

pub fn reduce_sk<Fs: PrimeField>(seed: &[u8]) -> Num<Fs> {
    Num::<Fs>::from_uint_reduced(NumRepr(Uint::from_little_endian(seed)))
}

#[derive(Clone)]
pub struct Keys<P: PoolParams> {
    pub sk: Num<P::Fs>,
    pub a: Num<P::Fr>,
    pub eta: Num<P::Fr>,
}

impl<P: PoolParams> Keys<P> {
    pub fn derive(sk: Num<P::Fs>, params: &P) -> Self {
        let a = derive_key_a(sk, params).x;
        let eta = derive_key_eta(a, params);

        Keys { sk, a, eta }
    }
}