use blstrs::{G1Projective, Scalar};
use digest::Digest;
use group::{ff::Field, Group};
use merlin::Transcript;
use rand::SeedableRng;
use rand_chacha::ChaCha20Rng;
use sha3::Sha3_256;
use crate::errors::ProofError;
pub trait TranscriptProtocol {
fn rangeproof_domain_sep(&mut self, n: u64, m: u64);
fn innerproduct_domain_sep(&mut self, n: u64);
fn r1cs_domain_sep(&mut self);
fn r1cs_1phase_domain_sep(&mut self);
fn r1cs_2phase_domain_sep(&mut self);
fn append_scalar(&mut self, label: &'static [u8], scalar: &Scalar);
fn append_point(&mut self, label: &'static [u8], point: &G1Projective);
fn validate_and_append_point(
&mut self,
label: &'static [u8],
point: &G1Projective,
) -> Result<(), ProofError>;
fn challenge_scalar(&mut self, label: &'static [u8]) -> Scalar;
}
impl TranscriptProtocol for Transcript {
fn rangeproof_domain_sep(&mut self, n: u64, m: u64) {
self.append_message(b"dom-sep", b"rangeproof v1");
self.append_u64(b"n", n);
self.append_u64(b"m", m);
}
fn innerproduct_domain_sep(&mut self, n: u64) {
self.append_message(b"dom-sep", b"ipp v1");
self.append_u64(b"n", n);
}
fn r1cs_domain_sep(&mut self) {
self.append_message(b"dom-sep", b"r1cs v1");
}
fn r1cs_1phase_domain_sep(&mut self) {
self.append_message(b"dom-sep", b"r1cs-1phase");
}
fn r1cs_2phase_domain_sep(&mut self) {
self.append_message(b"dom-sep", b"r1cs-2phase");
}
fn append_scalar(&mut self, label: &'static [u8], scalar: &Scalar) {
self.append_message(label, &scalar.to_bytes_le());
}
fn append_point(&mut self, label: &'static [u8], point: &G1Projective) {
self.append_message(label, &point.to_compressed());
}
fn validate_and_append_point(
&mut self,
label: &'static [u8],
point: &G1Projective,
) -> Result<(), ProofError> {
if bool::from(point.is_identity()) {
Err(ProofError::VerificationError)
} else {
Ok(self.append_message(label, &point.to_compressed()))
}
}
fn challenge_scalar(&mut self, label: &'static [u8]) -> Scalar {
let mut buf = [0u8; 64];
self.challenge_bytes(label, &mut buf);
let mut sha3 = Sha3_256::new();
sha3.update(b"TranscriptChallenge");
sha3.update(buf);
let mut rng = ChaCha20Rng::from_seed(sha3.finalize().into());
Scalar::random(&mut rng)
}
}