feat: add TranscriptVar trait (privacy-scaling-explorations#42)

* feat: add TranscriptVar trait * Update src/transcript/poseidon.rs Co-authored-by: arnaucube <[email protected]> --------- Co-authored-by: arnaucube <[email protected]>
amit0365 · Dec 5, 2023 · 14a0b46 · 14a0b46
1 parent 905ba44
commit 14a0b46
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 7 deletions.
diff --git a/src/transcript/mod.rs b/src/transcript/mod.rs
@@ -1,5 +1,8 @@
 use crate::Error;
 use ark_ec::CurveGroup;
+use ark_ff::PrimeField;
+use ark_r1cs_std::{boolean::Boolean, fields::fp::FpVar};
+use ark_relations::r1cs::{ConstraintSystemRef, SynthesisError};
 use ark_std::fmt::Debug;
 
 pub mod poseidon;
@@ -16,3 +19,16 @@ pub trait Transcript<C: CurveGroup> {
     fn get_challenge_nbits(&mut self, nbits: usize) -> Vec<bool>;
     fn get_challenges(&mut self, n: usize) -> Vec<C::ScalarField>;
 }
+
+pub trait TranscriptVar<F: PrimeField> {
+    type TranscriptVarConfig: Debug;
+
+    fn new(cs: ConstraintSystemRef<F>, poseidon_config: &Self::TranscriptVarConfig) -> Self;
+    fn absorb(&mut self, v: FpVar<F>) -> Result<(), SynthesisError>;
+    fn absorb_vec(&mut self, v: &[FpVar<F>]) -> Result<(), SynthesisError>;
+    fn get_challenge(&mut self) -> Result<FpVar<F>, SynthesisError>;
+    /// returns the bit representation of the challenge, we use its output in-circuit for the
+    /// `GC.scalar_mul_le` method.
+    fn get_challenge_nbits(&mut self, nbits: usize) -> Result<Vec<Boolean<F>>, SynthesisError>;
+    fn get_challenges(&mut self, n: usize) -> Result<Vec<FpVar<F>>, SynthesisError>;
+}
diff --git a/src/transcript/poseidon.rs b/src/transcript/poseidon.rs
@@ -12,6 +12,8 @@ use ark_std::{One, Zero};
 use crate::transcript::Transcript;
 use crate::Error;
 
+use super::TranscriptVar;
+
 /// PoseidonTranscript implements the Transcript trait using the Poseidon hash
 pub struct PoseidonTranscript<C: CurveGroup>
 where
@@ -84,29 +86,31 @@ fn prepare_point<C: CurveGroup>(p: &C) -> Result<Vec<C::ScalarField>, Error> {
 pub struct PoseidonTranscriptVar<F: PrimeField> {
     sponge: PoseidonSpongeVar<F>,
 }
-impl<F: PrimeField> PoseidonTranscriptVar<F> {
-    pub fn new(cs: ConstraintSystemRef<F>, poseidon_config: &PoseidonConfig<F>) -> Self {
+impl<F: PrimeField> TranscriptVar<F> for PoseidonTranscriptVar<F> {
+    type TranscriptVarConfig = PoseidonConfig<F>;
+
+    fn new(cs: ConstraintSystemRef<F>, poseidon_config: &Self::TranscriptVarConfig) -> Self {
         let sponge = PoseidonSpongeVar::<F>::new(cs, poseidon_config);
         Self { sponge }
     }
-    pub fn absorb(&mut self, v: FpVar<F>) -> Result<(), SynthesisError> {
+    fn absorb(&mut self, v: FpVar<F>) -> Result<(), SynthesisError> {
         self.sponge.absorb(&v)
     }
-    pub fn absorb_vec(&mut self, v: &[FpVar<F>]) -> Result<(), SynthesisError> {
+    fn absorb_vec(&mut self, v: &[FpVar<F>]) -> Result<(), SynthesisError> {
         self.sponge.absorb(&v)
     }
-    pub fn get_challenge(&mut self) -> Result<FpVar<F>, SynthesisError> {
+    fn get_challenge(&mut self) -> Result<FpVar<F>, SynthesisError> {
         let c = self.sponge.squeeze_field_elements(1)?;
         self.sponge.absorb(&c[0])?;
         Ok(c[0].clone())
     }
 
     /// returns the bit representation of the challenge, we use its output in-circuit for the
     /// `GC.scalar_mul_le` method.
-    pub fn get_challenge_nbits(&mut self, nbits: usize) -> Result<Vec<Boolean<F>>, SynthesisError> {
+    fn get_challenge_nbits(&mut self, nbits: usize) -> Result<Vec<Boolean<F>>, SynthesisError> {
         self.sponge.squeeze_bits(nbits)
     }
-    pub fn get_challenges(&mut self, n: usize) -> Result<Vec<FpVar<F>>, SynthesisError> {
+    fn get_challenges(&mut self, n: usize) -> Result<Vec<FpVar<F>>, SynthesisError> {
         let c = self.sponge.squeeze_field_elements(n)?;
         self.sponge.absorb(&c)?;
         Ok(c)