triptych 0.1.1

An experimental Rust implementation of the Triptych zero-knowledge proving system
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
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
108
109
110
111
112
113
114
115
116

// Copyright (c) 2024, The Tari Project
// SPDX-License-Identifier: BSD-3-Clause

use alloc::sync::Arc;

use curve25519_dalek::{RistrettoPoint, Scalar};
use rand_core::CryptoRngCore;
use snafu::prelude::*;
use zeroize::{Zeroize, ZeroizeOnDrop, Zeroizing};

use crate::parallel::TriptychParameters;

/// A Triptych proof witness.
///
/// The witness consists of a signing key, an auxiliary key, and an index where the corresponding verification and
/// auxiliary verification keys will appear in  the [`TriptychInputSet`](`crate::statement::TriptychInputSet`) of a
/// [`TriptychStatement`](`crate::statement::TriptychStatement`). It also contains [`TriptychParameters`].
#[derive(Zeroize, ZeroizeOnDrop)]
pub struct TriptychWitness {
    #[zeroize(skip)]
    params: Arc<TriptychParameters>,
    l: u32,
    r: Scalar,
    r1: Scalar,
}

/// Errors that can arise relating to [`TriptychWitness`].
#[derive(Debug, Snafu)]
pub enum WitnessError {
    /// An invalid parameter was provided.
    #[snafu(display("An invalid parameter was provided"))]
    InvalidParameter,
}

impl TriptychWitness {
    /// Generate a new [`TriptychWitness`] from secret data.
    ///
    /// The signing key `r` and auxiliary key `r1` must be nonzero, and the index `l` must be valid for the
    /// [`TriptychParameters`] `params`. If any of these conditions is not met, returns a [`WitnessError`].
    ///
    /// If you'd like a [`TriptychWitness`] generated securely for you, use [`TriptychWitness::random`] instead.
    #[allow(non_snake_case)]
    pub fn new(params: &Arc<TriptychParameters>, l: u32, r: &Scalar, r1: &Scalar) -> Result<Self, WitnessError> {
        if r == &Scalar::ZERO || r1 == &Scalar::ZERO {
            return Err(WitnessError::InvalidParameter);
        }
        if l >= params.get_N() {
            return Err(WitnessError::InvalidParameter);
        }

        Ok(Self {
            params: params.clone(),
            l,
            r: *r,
            r1: *r1,
        })
    }

    /// Generate a new random [`TriptychWitness`].
    ///
    /// You must provide [`TriptychParameters`] `params` and a [`CryptoRngCore`] random number generator `rng`.
    /// This will generate a [`TriptychWitness`] with a cryptographically-secure signing key and random index.
    ///
    /// If you'd rather provide your own secret data, use [`TriptychWitness::new`] instead.
    #[allow(clippy::cast_possible_truncation)]
    pub fn random<R: CryptoRngCore>(params: &Arc<TriptychParameters>, rng: &mut R) -> Self {
        // Generate a random index using wide reduction
        // This can't truncate since `N` is bounded by `u32`
        // It is also defined since `N > 0`
        #[allow(clippy::arithmetic_side_effects)]
        let l = (rng.as_rngcore().next_u64() % u64::from(params.get_N())) as u32;

        Self {
            params: params.clone(),
            l,
            r: Scalar::random(rng),
            r1: Scalar::random(rng),
        }
    }

    /// Get the [`TriptychParameters`] from this [`TriptychWitness`].
    pub fn get_params(&self) -> &Arc<TriptychParameters> {
        &self.params
    }

    /// Get the index from this [`TriptychWitness`].
    pub fn get_l(&self) -> u32 {
        self.l
    }

    /// Get the signing key from this [`TriptychWitness`].
    pub fn get_r(&self) -> &Scalar {
        &self.r
    }

    /// Get the auxiliary key from this [`TriptychWitness`].
    pub fn get_r1(&self) -> &Scalar {
        &self.r1
    }

    /// Compute the linking tag for the [`TriptychWitness`] signing key.
    #[allow(non_snake_case)]
    pub fn compute_linking_tag(&self) -> RistrettoPoint {
        *Zeroizing::new(self.r.invert()) * self.params.get_U()
    }

    /// Compute the verification key for the [`TriptychWitness`] signing key.
    pub fn compute_verification_key(&self) -> RistrettoPoint {
        self.r * self.params.get_G()
    }

    /// Compute the auxiliary verification key for the [`TriptychWitness`] signing key.
    pub fn compute_auxiliary_verification_key(&self) -> RistrettoPoint {
        self.r1 * self.params.get_G1()
    }
}