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

// helper functions for tests
use crate::clsag::Clsag;
use crate::keys::PrivateSet;
use crate::member::Member;
use curve25519_dalek::ristretto::{CompressedRistretto, RistrettoPoint};
use curve25519_dalek::scalar::Scalar;

// There is an exact copy of this function in member.rs
// which is being used to generate nonces. The reason is because the code in this
// file should only be used for test
pub fn generate_rand_scalars(num: usize) -> Vec<Scalar> {
    let mut rng = rand::thread_rng();
    let mut scalars = Vec::<Scalar>::with_capacity(num);

    for _ in 0..num {
        scalars.push(Scalar::random(&mut rng));
    }

    scalars
}

pub fn generate_private_set(num: usize) -> PrivateSet {
    let scalars = generate_rand_scalars(num);
    PrivateSet(scalars)
}

pub fn generate_rand_points(num: usize) -> Vec<RistrettoPoint> {
    let mut rng = rand::thread_rng();
    let mut points = Vec::<RistrettoPoint>::with_capacity(num);

    for _ in 0..num {
        points.push(RistrettoPoint::random(&mut rng));
    }

    points
}
pub fn generate_rand_compressed_points(num: usize) -> Vec<CompressedRistretto> {
    let mut rng = rand::thread_rng();
    let mut points = Vec::<CompressedRistretto>::with_capacity(num);

    for _ in 0..num {
        points.push(RistrettoPoint::random(&mut rng).compress());
    }

    points
}

pub fn generate_decoy(num_keys: usize) -> Member {
    let points = generate_rand_points(num_keys);
    Member::new_decoy(points)
}

pub fn generate_decoys(num_decoys: usize, num_keys: usize) -> Vec<Member> {
    let mut decoys: Vec<Member> = Vec::with_capacity(num_decoys);
    for _ in 0..num_decoys {
        decoys.push(generate_decoy(num_keys));
    }
    decoys
}

pub fn generate_signer(num_keys: usize) -> Member {
    let scalars = generate_rand_scalars(num_keys);
    Member::new_signer(scalars)
}

pub fn generate_clsag_with(num_decoys: usize, num_keys: usize) -> Clsag {
    let mut clsag = Clsag::new();

    for _ in 0..num_decoys {
        clsag.add_member(generate_decoy(num_keys));
    }

    clsag
}