-
Notifications
You must be signed in to change notification settings - Fork 9
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'refactor/move_circom_witness_ext' of github.com:TaceoLa…
…bs/collaborative-circom into refactor/move_circom_witness_ext
- Loading branch information
Showing
11 changed files
with
336 additions
and
35 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,6 +1,200 @@ | ||
| use super::{CircomGroth16Prover, IoResult}; | ||
| use ark_ec::pairing::Pairing; | ||
| use ark_ff::PrimeField; | ||
| use mpc_core::protocols::shamir::{network::ShamirNetwork, ShamirProtocol}; | ||
| use itertools::izip; | ||
| use mpc_core::protocols::shamir::{ | ||
| arithmetic, network::ShamirNetwork, pointshare, ShamirPointShare, ShamirPrimeFieldShare, | ||
| ShamirProtocol, | ||
| }; | ||
|
|
||
| pub struct ShamirGroth16Driver<F: PrimeField, N: ShamirNetwork> { | ||
| protocol: ShamirProtocol<F, N>, | ||
| } | ||
|
|
||
| impl<F: PrimeField, N: ShamirNetwork> ShamirGroth16Driver<F, N> { | ||
| pub fn new(protocol: ShamirProtocol<F, N>) -> Self { | ||
| Self { protocol } | ||
| } | ||
| } | ||
|
|
||
| impl<P: Pairing, N: ShamirNetwork> CircomGroth16Prover<P> | ||
| for ShamirGroth16Driver<P::ScalarField, N> | ||
| { | ||
| type ArithmeticShare = ShamirPrimeFieldShare<P::ScalarField>; | ||
| type PointShareG1 = ShamirPointShare<P::G1>; | ||
| type PointShareG2 = ShamirPointShare<P::G2>; | ||
|
|
||
| type PartyID = usize; | ||
|
|
||
| async fn rand(&mut self) -> IoResult<Self::ArithmeticShare> { | ||
| self.protocol.rand().await | ||
| } | ||
|
|
||
| fn get_party_id(&self) -> Self::PartyID { | ||
| self.protocol.network.get_id() | ||
| } | ||
|
|
||
| fn fork(&mut self) -> Self { | ||
| todo!() | ||
| } | ||
|
|
||
| fn evaluate_constraint( | ||
| _party_id: Self::PartyID, | ||
| lhs: &[(<P as Pairing>::ScalarField, usize)], | ||
| public_inputs: &[<P as Pairing>::ScalarField], | ||
| private_witness: &[Self::ArithmeticShare], | ||
| ) -> Self::ArithmeticShare { | ||
| let mut acc = Self::ArithmeticShare::default(); | ||
| for (coeff, index) in lhs { | ||
| if index < &public_inputs.len() { | ||
| let val = public_inputs[*index]; | ||
| let mul_result = val * coeff; | ||
| arithmetic::add_assign_public(&mut acc, mul_result); | ||
| } else { | ||
| let current_witness = private_witness[*index - public_inputs.len()]; | ||
| arithmetic::add_assign(&mut acc, arithmetic::mul_public(current_witness, *coeff)); | ||
| } | ||
| } | ||
| acc | ||
| } | ||
|
|
||
| fn promote_to_trivial_shares( | ||
| id: Self::PartyID, | ||
| public_values: &[<P as Pairing>::ScalarField], | ||
| ) -> Vec<Self::ArithmeticShare> { | ||
| todo!() | ||
| } | ||
|
|
||
| fn sub_assign_vec(a: &mut [Self::ArithmeticShare], b: &[Self::ArithmeticShare]) { | ||
| for (a, b) in izip!(a, b) { | ||
| arithmetic::sub_assign(a, *b); | ||
| } | ||
| } | ||
|
|
||
| async fn mul( | ||
| &mut self, | ||
| a: Self::ArithmeticShare, | ||
| b: Self::ArithmeticShare, | ||
| ) -> IoResult<Self::ArithmeticShare> { | ||
| arithmetic::mul(a, b, &mut self.protocol).await | ||
| } | ||
|
|
||
| async fn mul_vec( | ||
| &mut self, | ||
| a: &[Self::ArithmeticShare], | ||
| b: &[Self::ArithmeticShare], | ||
| ) -> IoResult<Vec<Self::ArithmeticShare>> { | ||
| arithmetic::mul_vec(a, b, &mut self.protocol).await | ||
| } | ||
|
|
||
| fn fft_in_place<D: ark_poly::EvaluationDomain<<P as Pairing>::ScalarField>>( | ||
| data: &mut Vec<Self::ArithmeticShare>, | ||
| domain: &D, | ||
| ) { | ||
| domain.fft_in_place(data) | ||
| } | ||
|
|
||
| fn ifft_in_place<D: ark_poly::EvaluationDomain<<P as Pairing>::ScalarField>>( | ||
| data: &mut Vec<Self::ArithmeticShare>, | ||
| domain: &D, | ||
| ) { | ||
| domain.ifft_in_place(data) | ||
| } | ||
|
|
||
| fn ifft<D: ark_poly::EvaluationDomain<<P as Pairing>::ScalarField>>( | ||
| data: &[Self::ArithmeticShare], | ||
| domain: &D, | ||
| ) -> Vec<Self::ArithmeticShare> { | ||
| domain.ifft(&data) | ||
| } | ||
|
|
||
| fn distribute_powers_and_mul_by_const( | ||
| coeffs: &mut [Self::ArithmeticShare], | ||
| g: <P as Pairing>::ScalarField, | ||
| c: <P as Pairing>::ScalarField, | ||
| ) { | ||
| let mut pow = c; | ||
| for share in coeffs.iter_mut() { | ||
| arithmetic::mul_assign_public(share, pow); | ||
| pow *= g; | ||
| } | ||
| } | ||
|
|
||
| fn msm_public_points_g1( | ||
| points: &[<P as Pairing>::G1Affine], | ||
| scalars: &[Self::ArithmeticShare], | ||
| ) -> Self::PointShareG1 { | ||
| pointshare::msm_public_points(points, scalars) | ||
| } | ||
|
|
||
| fn msm_public_points_g2( | ||
| points: &[<P as Pairing>::G2Affine], | ||
| scalars: &[Self::ArithmeticShare], | ||
| ) -> Self::PointShareG2 { | ||
| pointshare::msm_public_points(points, scalars) | ||
| } | ||
|
|
||
| fn scalar_mul_public_point_g1( | ||
| a: &<P as Pairing>::G1, | ||
| b: Self::ArithmeticShare, | ||
| ) -> Self::PointShareG1 { | ||
| pointshare::scalar_mul_public_point(b, a) | ||
| } | ||
|
|
||
| fn add_assign_points_g1(a: &mut Self::PointShareG1, b: &Self::PointShareG1) { | ||
| pointshare::add_assign(a, b) | ||
| } | ||
|
|
||
| fn add_assign_points_public_g1( | ||
| _id: Self::PartyID, | ||
| a: &mut Self::PointShareG1, | ||
| b: &<P as Pairing>::G1, | ||
| ) { | ||
| pointshare::add_assign_public(a, b) | ||
| } | ||
|
|
||
| async fn open_point_g1(&mut self, a: &Self::PointShareG1) -> IoResult<<P as Pairing>::G1> { | ||
| pointshare::open_point(a, &mut self.protocol).await | ||
| } | ||
|
|
||
| async fn scalar_mul_g1( | ||
| &mut self, | ||
| a: &Self::PointShareG1, | ||
| b: Self::ArithmeticShare, | ||
| ) -> IoResult<Self::PointShareG1> { | ||
| pointshare::scalar_mul(a, b, &mut self.protocol).await | ||
| } | ||
|
|
||
| fn sub_assign_points_g1(a: &mut Self::PointShareG1, b: &Self::PointShareG1) { | ||
| pointshare::sub_assign(a, b); | ||
| } | ||
|
|
||
| fn scalar_mul_public_point_g2( | ||
| a: &<P as Pairing>::G2, | ||
| b: Self::ArithmeticShare, | ||
| ) -> Self::PointShareG2 { | ||
| pointshare::scalar_mul_public_point(b, a) | ||
| } | ||
|
|
||
| fn add_assign_points_g2(a: &mut Self::PointShareG2, b: &Self::PointShareG2) { | ||
| pointshare::add_assign(a, b) | ||
| } | ||
|
|
||
| fn add_assign_points_public_g2( | ||
| _id: Self::PartyID, | ||
| a: &mut Self::PointShareG2, | ||
| b: &<P as Pairing>::G2, | ||
| ) { | ||
| pointshare::add_assign_public(a, b) | ||
| } | ||
|
|
||
| async fn open_two_points( | ||
| &mut self, | ||
| a: Self::PointShareG1, | ||
| b: Self::PointShareG2, | ||
| ) -> std::io::Result<(<P as Pairing>::G1, <P as Pairing>::G2)> { | ||
| let a_res = pointshare::open_point(&a, &mut self.protocol).await?; | ||
| let b_res = pointshare::open_point(&b, &mut self.protocol).await?; | ||
| Ok((a_res, b_res)) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.