* Add Pedersen commitments gadget * Add Nova+CycleFold Decider circuit (for onchain approach) "onchain"==Ethereum's EVM * merge src/decider into src/folding/nova/decider * PR review updatesmain
@ -1,391 +0,0 @@ |
|||||
use ark_ec::CurveGroup;
|
|
||||
use ark_ff::{Field, PrimeField};
|
|
||||
use ark_r1cs_std::{
|
|
||||
alloc::{AllocVar, AllocationMode},
|
|
||||
fields::FieldVar,
|
|
||||
};
|
|
||||
use ark_relations::r1cs::{Namespace, SynthesisError};
|
|
||||
use core::{borrow::Borrow, marker::PhantomData};
|
|
||||
|
|
||||
use crate::ccs::r1cs::RelaxedR1CS;
|
|
||||
use crate::utils::vec::SparseMatrix;
|
|
||||
use crate::Error;
|
|
||||
|
|
||||
pub type ConstraintF<C> = <<C as CurveGroup>::BaseField as Field>::BasePrimeField;
|
|
||||
|
|
||||
#[derive(Debug, Clone)]
|
|
||||
pub struct RelaxedR1CSGadget<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>> {
|
|
||||
_f: PhantomData<F>,
|
|
||||
_cf: PhantomData<CF>,
|
|
||||
_fv: PhantomData<FV>,
|
|
||||
}
|
|
||||
impl<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>> RelaxedR1CSGadget<F, CF, FV> {
|
|
||||
/// performs the RelaxedR1CS check (Az∘Bz==uCz+E)
|
|
||||
pub fn check(rel_r1cs: RelaxedR1CSVar<F, CF, FV>, z: Vec<FV>) -> Result<(), Error> {
|
|
||||
let Az = mat_vec_mul_sparse(rel_r1cs.A, z.clone());
|
|
||||
let Bz = mat_vec_mul_sparse(rel_r1cs.B, z.clone());
|
|
||||
let Cz = mat_vec_mul_sparse(rel_r1cs.C, z.clone());
|
|
||||
let uCz = vec_scalar_mul(&Cz, &rel_r1cs.u);
|
|
||||
let uCzE = vec_add(&uCz, &rel_r1cs.E)?;
|
|
||||
let AzBz = hadamard(&Az, &Bz)?;
|
|
||||
for i in 0..AzBz.len() {
|
|
||||
AzBz[i].enforce_equal(&uCzE[i].clone())?;
|
|
||||
}
|
|
||||
Ok(())
|
|
||||
}
|
|
||||
}
|
|
||||
|
|
||||
fn mat_vec_mul_sparse<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>>(
|
|
||||
m: SparseMatrixVar<F, CF, FV>,
|
|
||||
v: Vec<FV>,
|
|
||||
) -> Vec<FV> {
|
|
||||
let mut res = vec![FV::zero(); m.n_rows];
|
|
||||
for (row_i, row) in m.coeffs.iter().enumerate() {
|
|
||||
for (value, col_i) in row.iter() {
|
|
||||
res[row_i] += value.clone().mul(&v[*col_i].clone());
|
|
||||
}
|
|
||||
}
|
|
||||
res
|
|
||||
}
|
|
||||
pub fn vec_add<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>>(
|
|
||||
a: &Vec<FV>,
|
|
||||
b: &Vec<FV>,
|
|
||||
) -> Result<Vec<FV>, Error> {
|
|
||||
if a.len() != b.len() {
|
|
||||
return Err(Error::NotSameLength(
|
|
||||
"a.len()".to_string(),
|
|
||||
a.len(),
|
|
||||
"b.len()".to_string(),
|
|
||||
b.len(),
|
|
||||
));
|
|
||||
}
|
|
||||
let mut r: Vec<FV> = vec![FV::zero(); a.len()];
|
|
||||
for i in 0..a.len() {
|
|
||||
r[i] = a[i].clone() + b[i].clone();
|
|
||||
}
|
|
||||
Ok(r)
|
|
||||
}
|
|
||||
pub fn vec_scalar_mul<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>>(
|
|
||||
vec: &Vec<FV>,
|
|
||||
c: &FV,
|
|
||||
) -> Vec<FV> {
|
|
||||
let mut result = vec![FV::zero(); vec.len()];
|
|
||||
for (i, a) in vec.iter().enumerate() {
|
|
||||
result[i] = a.clone() * c;
|
|
||||
}
|
|
||||
result
|
|
||||
}
|
|
||||
pub fn hadamard<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>>(
|
|
||||
a: &Vec<FV>,
|
|
||||
b: &Vec<FV>,
|
|
||||
) -> Result<Vec<FV>, Error> {
|
|
||||
if a.len() != b.len() {
|
|
||||
return Err(Error::NotSameLength(
|
|
||||
"a.len()".to_string(),
|
|
||||
a.len(),
|
|
||||
"b.len()".to_string(),
|
|
||||
b.len(),
|
|
||||
));
|
|
||||
}
|
|
||||
let mut r: Vec<FV> = vec![FV::zero(); a.len()];
|
|
||||
for i in 0..a.len() {
|
|
||||
r[i] = a[i].clone() * b[i].clone();
|
|
||||
}
|
|
||||
Ok(r)
|
|
||||
}
|
|
||||
|
|
||||
#[derive(Debug, Clone)]
|
|
||||
pub struct SparseMatrixVar<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>> {
|
|
||||
_f: PhantomData<F>,
|
|
||||
_cf: PhantomData<CF>,
|
|
||||
_fv: PhantomData<FV>,
|
|
||||
pub n_rows: usize,
|
|
||||
pub n_cols: usize,
|
|
||||
// same format as the native SparseMatrix (which follows ark_relations::r1cs::Matrix format
|
|
||||
pub coeffs: Vec<Vec<(FV, usize)>>,
|
|
||||
}
|
|
||||
|
|
||||
impl<F, CF, FV> AllocVar<SparseMatrix<F>, CF> for SparseMatrixVar<F, CF, FV>
|
|
||||
where
|
|
||||
F: PrimeField,
|
|
||||
CF: PrimeField,
|
|
||||
FV: FieldVar<F, CF>,
|
|
||||
{
|
|
||||
fn new_variable<T: Borrow<SparseMatrix<F>>>(
|
|
||||
cs: impl Into<Namespace<CF>>,
|
|
||||
f: impl FnOnce() -> Result<T, SynthesisError>,
|
|
||||
mode: AllocationMode,
|
|
||||
) -> Result<Self, SynthesisError> {
|
|
||||
f().and_then(|val| {
|
|
||||
let cs = cs.into();
|
|
||||
|
|
||||
let mut coeffs: Vec<Vec<(FV, usize)>> = Vec::new();
|
|
||||
for row in val.borrow().coeffs.iter() {
|
|
||||
let mut rowVar: Vec<(FV, usize)> = Vec::new();
|
|
||||
for &(value, col_i) in row.iter() {
|
|
||||
let coeffVar = FV::new_variable(cs.clone(), || Ok(value), mode)?;
|
|
||||
rowVar.push((coeffVar, col_i));
|
|
||||
}
|
|
||||
coeffs.push(rowVar);
|
|
||||
}
|
|
||||
|
|
||||
Ok(Self {
|
|
||||
_f: PhantomData,
|
|
||||
_cf: PhantomData,
|
|
||||
_fv: PhantomData,
|
|
||||
n_rows: val.borrow().n_rows,
|
|
||||
n_cols: val.borrow().n_cols,
|
|
||||
coeffs,
|
|
||||
})
|
|
||||
})
|
|
||||
}
|
|
||||
}
|
|
||||
|
|
||||
#[derive(Debug, Clone)]
|
|
||||
pub struct RelaxedR1CSVar<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>> {
|
|
||||
_f: PhantomData<F>,
|
|
||||
_cf: PhantomData<CF>,
|
|
||||
_fv: PhantomData<FV>,
|
|
||||
pub A: SparseMatrixVar<F, CF, FV>,
|
|
||||
pub B: SparseMatrixVar<F, CF, FV>,
|
|
||||
pub C: SparseMatrixVar<F, CF, FV>,
|
|
||||
pub u: FV,
|
|
||||
pub E: Vec<FV>,
|
|
||||
}
|
|
||||
|
|
||||
impl<F, CF, FV> AllocVar<RelaxedR1CS<F>, CF> for RelaxedR1CSVar<F, CF, FV>
|
|
||||
where
|
|
||||
F: PrimeField,
|
|
||||
CF: PrimeField,
|
|
||||
FV: FieldVar<F, CF>,
|
|
||||
{
|
|
||||
fn new_variable<T: Borrow<RelaxedR1CS<F>>>(
|
|
||||
cs: impl Into<Namespace<CF>>,
|
|
||||
f: impl FnOnce() -> Result<T, SynthesisError>,
|
|
||||
mode: AllocationMode,
|
|
||||
) -> Result<Self, SynthesisError> {
|
|
||||
f().and_then(|val| {
|
|
||||
let cs = cs.into();
|
|
||||
|
|
||||
let A = SparseMatrixVar::<F, CF, FV>::new_constant(cs.clone(), &val.borrow().A)?;
|
|
||||
let B = SparseMatrixVar::<F, CF, FV>::new_constant(cs.clone(), &val.borrow().B)?;
|
|
||||
let C = SparseMatrixVar::<F, CF, FV>::new_constant(cs.clone(), &val.borrow().C)?;
|
|
||||
let E = Vec::<FV>::new_variable(cs.clone(), || Ok(val.borrow().E.clone()), mode)?;
|
|
||||
let u = FV::new_variable(cs.clone(), || Ok(val.borrow().u), mode)?;
|
|
||||
|
|
||||
Ok(Self {
|
|
||||
_f: PhantomData,
|
|
||||
_cf: PhantomData,
|
|
||||
_fv: PhantomData,
|
|
||||
A,
|
|
||||
B,
|
|
||||
C,
|
|
||||
E,
|
|
||||
u,
|
|
||||
})
|
|
||||
})
|
|
||||
}
|
|
||||
}
|
|
||||
|
|
||||
#[cfg(test)]
|
|
||||
mod tests {
|
|
||||
use super::*;
|
|
||||
use ark_crypto_primitives::crh::{
|
|
||||
sha256::{
|
|
||||
constraints::{Sha256Gadget, UnitVar},
|
|
||||
Sha256,
|
|
||||
},
|
|
||||
CRHScheme, CRHSchemeGadget,
|
|
||||
};
|
|
||||
use ark_ff::BigInteger;
|
|
||||
use ark_pallas::{Fq, Fr};
|
|
||||
use ark_r1cs_std::{
|
|
||||
alloc::AllocVar,
|
|
||||
bits::uint8::UInt8,
|
|
||||
eq::EqGadget,
|
|
||||
fields::{fp::FpVar, nonnative::NonNativeFieldVar},
|
|
||||
};
|
|
||||
use ark_relations::r1cs::{
|
|
||||
ConstraintSynthesizer, ConstraintSystem, ConstraintSystemRef, SynthesisError,
|
|
||||
};
|
|
||||
use ark_std::One;
|
|
||||
|
|
||||
use crate::ccs::r1cs::{
|
|
||||
tests::{get_test_r1cs, get_test_z},
|
|
||||
R1CS,
|
|
||||
};
|
|
||||
use crate::frontend::arkworks::{extract_r1cs_and_z, tests::TestCircuit};
|
|
||||
|
|
||||
#[test]
|
|
||||
fn test_relaxed_r1cs_small_gadget_handcrafted() {
|
|
||||
let r1cs: R1CS<Fr> = get_test_r1cs();
|
|
||||
let rel_r1cs = r1cs.relax();
|
|
||||
let z = get_test_z(3);
|
|
||||
|
|
||||
let cs = ConstraintSystem::<Fr>::new_ref();
|
|
||||
|
|
||||
let zVar = Vec::<FpVar<Fr>>::new_witness(cs.clone(), || Ok(z)).unwrap();
|
|
||||
let rel_r1csVar =
|
|
||||
RelaxedR1CSVar::<Fr, Fr, FpVar<Fr>>::new_witness(cs.clone(), || Ok(rel_r1cs)).unwrap();
|
|
||||
|
|
||||
RelaxedR1CSGadget::<Fr, Fr, FpVar<Fr>>::check(rel_r1csVar, zVar).unwrap();
|
|
||||
assert!(cs.is_satisfied().unwrap());
|
|
||||
}
|
|
||||
|
|
||||
// gets as input a circuit that implements the ConstraintSynthesizer trait, and that has been
|
|
||||
// initialized.
|
|
||||
fn test_relaxed_r1cs_gadget<CS: ConstraintSynthesizer<Fr>>(circuit: CS) {
|
|
||||
let cs = ConstraintSystem::<Fr>::new_ref();
|
|
||||
|
|
||||
circuit.generate_constraints(cs.clone()).unwrap();
|
|
||||
cs.finalize();
|
|
||||
assert!(cs.is_satisfied().unwrap());
|
|
||||
|
|
||||
let cs = cs.into_inner().unwrap();
|
|
||||
|
|
||||
let (r1cs, z) = extract_r1cs_and_z::<Fr>(&cs);
|
|
||||
r1cs.check_relation(&z).unwrap();
|
|
||||
|
|
||||
let relaxed_r1cs = r1cs.relax();
|
|
||||
relaxed_r1cs.check_relation(&z).unwrap();
|
|
||||
|
|
||||
// set new CS for the circuit that checks the RelaxedR1CS of our original circuit
|
|
||||
let cs = ConstraintSystem::<Fr>::new_ref();
|
|
||||
// prepare the inputs for our circuit
|
|
||||
let zVar = Vec::<FpVar<Fr>>::new_witness(cs.clone(), || Ok(z)).unwrap();
|
|
||||
let rel_r1csVar =
|
|
||||
RelaxedR1CSVar::<Fr, Fr, FpVar<Fr>>::new_witness(cs.clone(), || Ok(relaxed_r1cs))
|
|
||||
.unwrap();
|
|
||||
|
|
||||
RelaxedR1CSGadget::<Fr, Fr, FpVar<Fr>>::check(rel_r1csVar, zVar).unwrap();
|
|
||||
assert!(cs.is_satisfied().unwrap());
|
|
||||
}
|
|
||||
|
|
||||
#[test]
|
|
||||
fn test_relaxed_r1cs_small_gadget_arkworks() {
|
|
||||
let x = Fr::from(5_u32);
|
|
||||
let y = x * x * x + x + Fr::from(5_u32);
|
|
||||
let circuit = TestCircuit::<Fr> { x, y };
|
|
||||
test_relaxed_r1cs_gadget(circuit);
|
|
||||
}
|
|
||||
|
|
||||
struct Sha256TestCircuit<F: PrimeField> {
|
|
||||
_f: PhantomData<F>,
|
|
||||
pub x: Vec<u8>,
|
|
||||
pub y: Vec<u8>,
|
|
||||
}
|
|
||||
impl<F: PrimeField> ConstraintSynthesizer<F> for Sha256TestCircuit<F> {
|
|
||||
fn generate_constraints(self, cs: ConstraintSystemRef<F>) -> Result<(), SynthesisError> {
|
|
||||
let x = Vec::<UInt8<F>>::new_witness(cs.clone(), || Ok(self.x))?;
|
|
||||
let y = Vec::<UInt8<F>>::new_input(cs.clone(), || Ok(self.y))?;
|
|
||||
|
|
||||
let unitVar = UnitVar::default();
|
|
||||
let comp_y = <Sha256Gadget<F> as CRHSchemeGadget<Sha256, F>>::evaluate(&unitVar, &x)?;
|
|
||||
comp_y.0.enforce_equal(&y)?;
|
|
||||
Ok(())
|
|
||||
}
|
|
||||
}
|
|
||||
#[test]
|
|
||||
fn test_relaxed_r1cs_medium_gadget_arkworks() {
|
|
||||
let x = Fr::from(5_u32).into_bigint().to_bytes_le();
|
|
||||
let y = <Sha256 as CRHScheme>::evaluate(&(), x.clone()).unwrap();
|
|
||||
|
|
||||
let circuit = Sha256TestCircuit::<Fr> {
|
|
||||
_f: PhantomData,
|
|
||||
x,
|
|
||||
y,
|
|
||||
};
|
|
||||
test_relaxed_r1cs_gadget(circuit);
|
|
||||
}
|
|
||||
|
|
||||
// circuit that has the number of constraints specified in the `n_constraints` parameter. Note
|
|
||||
// that the generated circuit will have very sparse matrices, so the resulting constraints
|
|
||||
// number of the RelaxedR1CS gadget must take that into account.
|
|
||||
struct CustomTestCircuit<F: PrimeField> {
|
|
||||
_f: PhantomData<F>,
|
|
||||
pub n_constraints: usize,
|
|
||||
pub x: F,
|
|
||||
pub y: F,
|
|
||||
}
|
|
||||
impl<F: PrimeField> CustomTestCircuit<F> {
|
|
||||
fn new(n_constraints: usize) -> Self {
|
|
||||
let x = F::from(5_u32);
|
|
||||
let mut y = F::one();
|
|
||||
for _ in 0..n_constraints - 1 {
|
|
||||
y *= x;
|
|
||||
}
|
|
||||
Self {
|
|
||||
_f: PhantomData,
|
|
||||
n_constraints,
|
|
||||
x,
|
|
||||
y,
|
|
||||
}
|
|
||||
}
|
|
||||
}
|
|
||||
impl<F: PrimeField> ConstraintSynthesizer<F> for CustomTestCircuit<F> {
|
|
||||
fn generate_constraints(self, cs: ConstraintSystemRef<F>) -> Result<(), SynthesisError> {
|
|
||||
let x = FpVar::<F>::new_witness(cs.clone(), || Ok(self.x))?;
|
|
||||
let y = FpVar::<F>::new_input(cs.clone(), || Ok(self.y))?;
|
|
||||
|
|
||||
let mut comp_y = FpVar::<F>::new_witness(cs.clone(), || Ok(F::one()))?;
|
|
||||
for _ in 0..self.n_constraints - 1 {
|
|
||||
comp_y *= x.clone();
|
|
||||
}
|
|
||||
|
|
||||
comp_y.enforce_equal(&y)?;
|
|
||||
Ok(())
|
|
||||
}
|
|
||||
}
|
|
||||
|
|
||||
#[test]
|
|
||||
fn test_relaxed_r1cs_custom_circuit() {
|
|
||||
let n_constraints = 10_000;
|
|
||||
let x = Fr::from(5_u32);
|
|
||||
let mut y = Fr::one();
|
|
||||
for _ in 0..n_constraints - 1 {
|
|
||||
y *= x;
|
|
||||
}
|
|
||||
|
|
||||
let circuit = CustomTestCircuit::<Fr> {
|
|
||||
_f: PhantomData,
|
|
||||
n_constraints,
|
|
||||
x,
|
|
||||
y,
|
|
||||
};
|
|
||||
test_relaxed_r1cs_gadget(circuit);
|
|
||||
}
|
|
||||
|
|
||||
#[test]
|
|
||||
fn test_relaxed_r1cs_nonnative_circuit() {
|
|
||||
let cs = ConstraintSystem::<Fq>::new_ref();
|
|
||||
// in practice we would use CycleFoldCircuit, but is a very big circuit (when computed
|
|
||||
// non-natively inside the RelaxedR1CS circuit), so in order to have a short test we use a
|
|
||||
// custom circuit.
|
|
||||
let circuit = CustomTestCircuit::<Fq>::new(10);
|
|
||||
circuit.generate_constraints(cs.clone()).unwrap();
|
|
||||
cs.finalize();
|
|
||||
let cs = cs.into_inner().unwrap();
|
|
||||
let (r1cs, z) = extract_r1cs_and_z::<Fq>(&cs);
|
|
||||
|
|
||||
let relaxed_r1cs = r1cs.relax();
|
|
||||
|
|
||||
// natively
|
|
||||
let cs = ConstraintSystem::<Fq>::new_ref();
|
|
||||
let zVar = Vec::<FpVar<Fq>>::new_witness(cs.clone(), || Ok(z.clone())).unwrap();
|
|
||||
let rel_r1csVar = RelaxedR1CSVar::<Fq, Fq, FpVar<Fq>>::new_witness(cs.clone(), || {
|
|
||||
Ok(relaxed_r1cs.clone())
|
|
||||
})
|
|
||||
.unwrap();
|
|
||||
RelaxedR1CSGadget::<Fq, Fq, FpVar<Fq>>::check(rel_r1csVar, zVar).unwrap();
|
|
||||
|
|
||||
// non-natively
|
|
||||
let cs = ConstraintSystem::<Fr>::new_ref();
|
|
||||
let zVar = Vec::<NonNativeFieldVar<Fq, Fr>>::new_witness(cs.clone(), || Ok(z)).unwrap();
|
|
||||
let rel_r1csVar =
|
|
||||
RelaxedR1CSVar::<Fq, Fr, NonNativeFieldVar<Fq, Fr>>::new_witness(cs.clone(), || {
|
|
||||
Ok(relaxed_r1cs)
|
|
||||
})
|
|
||||
.unwrap();
|
|
||||
RelaxedR1CSGadget::<Fq, Fr, NonNativeFieldVar<Fq, Fr>>::check(rel_r1csVar, zVar).unwrap();
|
|
||||
}
|
|
||||
}
|
|
@ -1 +0,0 @@ |
|||||
pub mod circuit;
|
|
@ -0,0 +1,656 @@ |
|||||
|
/// This file implements the onchain (Ethereum's EVM) decider circuit. For non-ethereum use cases,
|
||||
|
/// other more efficient approaches can be used.
|
||||
|
use ark_crypto_primitives::crh::poseidon::constraints::CRHParametersVar;
|
||||
|
use ark_crypto_primitives::sponge::{poseidon::PoseidonConfig, Absorb};
|
||||
|
use ark_ec::{CurveGroup, Group};
|
||||
|
use ark_ff::PrimeField;
|
||||
|
use ark_r1cs_std::{
|
||||
|
alloc::{AllocVar, AllocationMode},
|
||||
|
boolean::Boolean,
|
||||
|
eq::EqGadget,
|
||||
|
fields::{fp::FpVar, nonnative::NonNativeFieldVar, FieldVar},
|
||||
|
groups::GroupOpsBounds,
|
||||
|
prelude::CurveVar,
|
||||
|
ToConstraintFieldGadget,
|
||||
|
};
|
||||
|
use ark_relations::r1cs::{ConstraintSynthesizer, ConstraintSystemRef, Namespace, SynthesisError};
|
||||
|
use ark_std::{One, Zero};
|
||||
|
use core::{borrow::Borrow, marker::PhantomData};
|
||||
|
|
||||
|
use crate::ccs::r1cs::R1CS;
|
||||
|
use crate::folding::nova::{
|
||||
|
circuits::{CommittedInstanceVar, FCircuit, CF1, CF2},
|
||||
|
ivc::IVC,
|
||||
|
CommittedInstance, Witness,
|
||||
|
};
|
||||
|
use crate::pedersen::Params as PedersenParams;
|
||||
|
use crate::utils::gadgets::{
|
||||
|
hadamard, mat_vec_mul_sparse, vec_add, vec_scalar_mul, SparseMatrixVar,
|
||||
|
};
|
||||
|
|
||||
|
#[derive(Debug, Clone)]
|
||||
|
pub struct RelaxedR1CSGadget<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>> {
|
||||
|
_f: PhantomData<F>,
|
||||
|
_cf: PhantomData<CF>,
|
||||
|
_fv: PhantomData<FV>,
|
||||
|
}
|
||||
|
impl<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>> RelaxedR1CSGadget<F, CF, FV> {
|
||||
|
/// performs the RelaxedR1CS check (Az∘Bz==uCz+E)
|
||||
|
pub fn check(
|
||||
|
r1cs: R1CSVar<F, CF, FV>,
|
||||
|
E: Vec<FV>,
|
||||
|
u: FV,
|
||||
|
z: Vec<FV>,
|
||||
|
) -> Result<(), SynthesisError> {
|
||||
|
let Az = mat_vec_mul_sparse(r1cs.A, z.clone());
|
||||
|
let Bz = mat_vec_mul_sparse(r1cs.B, z.clone());
|
||||
|
let Cz = mat_vec_mul_sparse(r1cs.C, z.clone());
|
||||
|
let uCz = vec_scalar_mul(&Cz, &u);
|
||||
|
let uCzE = vec_add(&uCz, &E)?;
|
||||
|
let AzBz = hadamard(&Az, &Bz)?;
|
||||
|
for i in 0..AzBz.len() {
|
||||
|
AzBz[i].enforce_equal(&uCzE[i].clone())?;
|
||||
|
}
|
||||
|
Ok(())
|
||||
|
}
|
||||
|
}
|
||||
|
|
||||
|
#[derive(Debug, Clone)]
|
||||
|
pub struct R1CSVar<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>> {
|
||||
|
_f: PhantomData<F>,
|
||||
|
_cf: PhantomData<CF>,
|
||||
|
_fv: PhantomData<FV>,
|
||||
|
pub A: SparseMatrixVar<F, CF, FV>,
|
||||
|
pub B: SparseMatrixVar<F, CF, FV>,
|
||||
|
pub C: SparseMatrixVar<F, CF, FV>,
|
||||
|
}
|
||||
|
|
||||
|
impl<F, CF, FV> AllocVar<R1CS<F>, CF> for R1CSVar<F, CF, FV>
|
||||
|
where
|
||||
|
F: PrimeField,
|
||||
|
CF: PrimeField,
|
||||
|
FV: FieldVar<F, CF>,
|
||||
|
{
|
||||
|
fn new_variable<T: Borrow<R1CS<F>>>(
|
||||
|
cs: impl Into<Namespace<CF>>,
|
||||
|
f: impl FnOnce() -> Result<T, SynthesisError>,
|
||||
|
_mode: AllocationMode,
|
||||
|
) -> Result<Self, SynthesisError> {
|
||||
|
f().and_then(|val| {
|
||||
|
let cs = cs.into();
|
||||
|
|
||||
|
let A = SparseMatrixVar::<F, CF, FV>::new_constant(cs.clone(), &val.borrow().A)?;
|
||||
|
let B = SparseMatrixVar::<F, CF, FV>::new_constant(cs.clone(), &val.borrow().B)?;
|
||||
|
let C = SparseMatrixVar::<F, CF, FV>::new_constant(cs.clone(), &val.borrow().C)?;
|
||||
|
|
||||
|
Ok(Self {
|
||||
|
_f: PhantomData,
|
||||
|
_cf: PhantomData,
|
||||
|
_fv: PhantomData,
|
||||
|
A,
|
||||
|
B,
|
||||
|
C,
|
||||
|
})
|
||||
|
})
|
||||
|
}
|
||||
|
}
|
||||
|
|
||||
|
/// In-circuit representation of the Witness associated to the CommittedInstance.
|
||||
|
#[derive(Debug, Clone)]
|
||||
|
pub struct WitnessVar<C: CurveGroup> {
|
||||
|
pub E: Vec<FpVar<C::ScalarField>>,
|
||||
|
pub rE: FpVar<C::ScalarField>,
|
||||
|
pub W: Vec<FpVar<C::ScalarField>>,
|
||||
|
pub rW: FpVar<C::ScalarField>,
|
||||
|
}
|
||||
|
|
||||
|
impl<C> AllocVar<Witness<C>, CF1<C>> for WitnessVar<C>
|
||||
|
where
|
||||
|
C: CurveGroup,
|
||||
|
<C as ark_ec::CurveGroup>::BaseField: PrimeField,
|
||||
|
{
|
||||
|
fn new_variable<T: Borrow<Witness<C>>>(
|
||||
|
cs: impl Into<Namespace<CF1<C>>>,
|
||||
|
f: impl FnOnce() -> Result<T, SynthesisError>,
|
||||
|
mode: AllocationMode,
|
||||
|
) -> Result<Self, SynthesisError> {
|
||||
|
f().and_then(|val| {
|
||||
|
let cs = cs.into();
|
||||
|
|
||||
|
let E: Vec<FpVar<C::ScalarField>> =
|
||||
|
Vec::new_variable(cs.clone(), || Ok(val.borrow().E.clone()), mode)?;
|
||||
|
let rE =
|
||||
|
FpVar::<C::ScalarField>::new_variable(cs.clone(), || Ok(val.borrow().rE), mode)?;
|
||||
|
|
||||
|
let W: Vec<FpVar<C::ScalarField>> =
|
||||
|
Vec::new_variable(cs.clone(), || Ok(val.borrow().W.clone()), mode)?;
|
||||
|
let rW =
|
||||
|
FpVar::<C::ScalarField>::new_variable(cs.clone(), || Ok(val.borrow().rW), mode)?;
|
||||
|
|
||||
|
Ok(Self { E, rE, W, rW })
|
||||
|
})
|
||||
|
}
|
||||
|
}
|
||||
|
|
||||
|
/// In-circuit representation of the Witness associated to the CommittedInstance, but with
|
||||
|
/// non-native representation, since it is used to represent the CycleFold witness.
|
||||
|
#[derive(Debug, Clone)]
|
||||
|
pub struct CycleFoldWitnessVar<C: CurveGroup> {
|
||||
|
pub E: Vec<NonNativeFieldVar<C::ScalarField, CF2<C>>>,
|
||||
|
pub rE: NonNativeFieldVar<C::ScalarField, CF2<C>>,
|
||||
|
pub W: Vec<NonNativeFieldVar<C::ScalarField, CF2<C>>>,
|
||||
|
pub rW: NonNativeFieldVar<C::ScalarField, CF2<C>>,
|
||||
|
}
|
||||
|
|
||||
|
impl<C> AllocVar<Witness<C>, CF2<C>> for CycleFoldWitnessVar<C>
|
||||
|
where
|
||||
|
C: CurveGroup,
|
||||
|
<C as ark_ec::CurveGroup>::BaseField: PrimeField,
|
||||
|
{
|
||||
|
fn new_variable<T: Borrow<Witness<C>>>(
|
||||
|
cs: impl Into<Namespace<CF2<C>>>,
|
||||
|
f: impl FnOnce() -> Result<T, SynthesisError>,
|
||||
|
mode: AllocationMode,
|
||||
|
) -> Result<Self, SynthesisError> {
|
||||
|
f().and_then(|val| {
|
||||
|
let cs = cs.into();
|
||||
|
|
||||
|
let E: Vec<NonNativeFieldVar<C::ScalarField, CF2<C>>> =
|
||||
|
Vec::new_variable(cs.clone(), || Ok(val.borrow().E.clone()), mode)?;
|
||||
|
let rE = NonNativeFieldVar::<C::ScalarField, CF2<C>>::new_variable(
|
||||
|
cs.clone(),
|
||||
|
|| Ok(val.borrow().rE),
|
||||
|
mode,
|
||||
|
)?;
|
||||
|
|
||||
|
let W: Vec<NonNativeFieldVar<C::ScalarField, CF2<C>>> =
|
||||
|
Vec::new_variable(cs.clone(), || Ok(val.borrow().W.clone()), mode)?;
|
||||
|
let rW = NonNativeFieldVar::<C::ScalarField, CF2<C>>::new_variable(
|
||||
|
cs.clone(),
|
||||
|
|| Ok(val.borrow().rW),
|
||||
|
mode,
|
||||
|
)?;
|
||||
|
|
||||
|
Ok(Self { E, rE, W, rW })
|
||||
|
})
|
||||
|
}
|
||||
|
}
|
||||
|
|
||||
|
/// Circuit that implements the in-circuit checks needed for the onchain (Ethereum's EVM)
|
||||
|
/// verification.
|
||||
|
pub struct DeciderCircuit<C1, GC1, C2, GC2>
|
||||
|
where
|
||||
|
C1: CurveGroup,
|
||||
|
GC1: CurveVar<C1, CF2<C1>>,
|
||||
|
C2: CurveGroup,
|
||||
|
GC2: CurveVar<C2, CF2<C2>>,
|
||||
|
{
|
||||
|
_c1: PhantomData<C1>,
|
||||
|
_gc1: PhantomData<GC1>,
|
||||
|
_c2: PhantomData<C2>,
|
||||
|
_gc2: PhantomData<GC2>,
|
||||
|
|
||||
|
/// E vector's length of the Nova instance witness
|
||||
|
pub E_len: usize,
|
||||
|
/// E vector's length of the CycleFold instance witness
|
||||
|
pub cf_E_len: usize,
|
||||
|
/// R1CS of the Augmented Function circuit
|
||||
|
pub r1cs: R1CS<C1::ScalarField>,
|
||||
|
/// R1CS of the CycleFold circuit
|
||||
|
pub cf_r1cs: R1CS<C2::ScalarField>,
|
||||
|
/// CycleFold PedersenParams, over C2
|
||||
|
pub cf_pedersen_params: PedersenParams<C2>,
|
||||
|
pub poseidon_config: PoseidonConfig<CF1<C1>>,
|
||||
|
pub i: Option<CF1<C1>>,
|
||||
|
/// initial state
|
||||
|
pub z_0: Option<Vec<C1::ScalarField>>,
|
||||
|
/// current i-th state
|
||||
|
pub z_i: Option<Vec<C1::ScalarField>>,
|
||||
|
/// Nova instances
|
||||
|
pub u_i: Option<CommittedInstance<C1>>,
|
||||
|
pub w_i: Option<Witness<C1>>,
|
||||
|
pub U_i: Option<CommittedInstance<C1>>,
|
||||
|
pub W_i: Option<Witness<C1>>,
|
||||
|
/// CycleFold running instance
|
||||
|
pub cf_U_i: Option<CommittedInstance<C2>>,
|
||||
|
pub cf_W_i: Option<Witness<C2>>,
|
||||
|
}
|
||||
|
impl<C1, GC1, C2, GC2> DeciderCircuit<C1, GC1, C2, GC2>
|
||||
|
where
|
||||
|
C1: CurveGroup,
|
||||
|
C2: CurveGroup,
|
||||
|
GC1: CurveVar<C1, CF2<C1>>,
|
||||
|
GC2: CurveVar<C2, CF2<C2>>,
|
||||
|
{
|
||||
|
pub fn from_ivc<FC: FCircuit<C1::ScalarField>>(ivc: IVC<C1, GC1, C2, GC2, FC>) -> Self {
|
||||
|
Self {
|
||||
|
_c1: PhantomData,
|
||||
|
_gc1: PhantomData,
|
||||
|
_c2: PhantomData,
|
||||
|
_gc2: PhantomData,
|
||||
|
|
||||
|
E_len: ivc.W_i.E.len(),
|
||||
|
cf_E_len: ivc.cf_W_i.E.len(),
|
||||
|
r1cs: ivc.r1cs,
|
||||
|
cf_r1cs: ivc.cf_r1cs,
|
||||
|
cf_pedersen_params: ivc.cf_pedersen_params,
|
||||
|
poseidon_config: ivc.poseidon_config,
|
||||
|
i: Some(ivc.i),
|
||||
|
z_0: Some(ivc.z_0),
|
||||
|
z_i: Some(ivc.z_i),
|
||||
|
u_i: Some(ivc.u_i),
|
||||
|
w_i: Some(ivc.w_i),
|
||||
|
U_i: Some(ivc.U_i),
|
||||
|
W_i: Some(ivc.W_i),
|
||||
|
cf_U_i: Some(ivc.cf_U_i),
|
||||
|
cf_W_i: Some(ivc.cf_W_i),
|
||||
|
}
|
||||
|
}
|
||||
|
}
|
||||
|
|
||||
|
impl<C1, GC1, C2, GC2> ConstraintSynthesizer<CF1<C1>> for DeciderCircuit<C1, GC1, C2, GC2>
|
||||
|
where
|
||||
|
C1: CurveGroup,
|
||||
|
C2: CurveGroup,
|
||||
|
GC1: CurveVar<C1, CF2<C1>>,
|
||||
|
GC2: CurveVar<C2, CF2<C2>>,
|
||||
|
<C1 as CurveGroup>::BaseField: PrimeField,
|
||||
|
<C2 as CurveGroup>::BaseField: PrimeField,
|
||||
|
<C1 as Group>::ScalarField: Absorb,
|
||||
|
<C2 as Group>::ScalarField: Absorb,
|
||||
|
C1: CurveGroup<BaseField = C2::ScalarField, ScalarField = C2::BaseField>,
|
||||
|
for<'a> &'a GC2: GroupOpsBounds<'a, C2, GC2>,
|
||||
|
{
|
||||
|
fn generate_constraints(self, cs: ConstraintSystemRef<CF1<C1>>) -> Result<(), SynthesisError> {
|
||||
|
let r1cs =
|
||||
|
R1CSVar::<C1::ScalarField, CF1<C1>, FpVar<CF1<C1>>>::new_witness(cs.clone(), || {
|
||||
|
Ok(self.r1cs.clone())
|
||||
|
})?;
|
||||
|
|
||||
|
let i = FpVar::<CF1<C1>>::new_witness(cs.clone(), || {
|
||||
|
Ok(self.i.unwrap_or_else(CF1::<C1>::zero))
|
||||
|
})?;
|
||||
|
let z_0 = Vec::<FpVar<CF1<C1>>>::new_witness(cs.clone(), || {
|
||||
|
Ok(self.z_0.unwrap_or(vec![CF1::<C1>::zero()]))
|
||||
|
})?;
|
||||
|
let z_i = Vec::<FpVar<CF1<C1>>>::new_witness(cs.clone(), || {
|
||||
|
Ok(self.z_i.unwrap_or(vec![CF1::<C1>::zero()]))
|
||||
|
})?;
|
||||
|
|
||||
|
let u_dummy_native = CommittedInstance::<C1>::dummy(1);
|
||||
|
let w_dummy_native = Witness::<C1>::new(
|
||||
|
vec![C1::ScalarField::zero(); self.r1cs.A.n_cols - 2 /* (2=1+1, since u_i.x.len=1) */],
|
||||
|
self.E_len,
|
||||
|
);
|
||||
|
|
||||
|
let u_i = CommittedInstanceVar::<C1>::new_witness(cs.clone(), || {
|
||||
|
Ok(self.u_i.unwrap_or(u_dummy_native.clone()))
|
||||
|
})?;
|
||||
|
let w_i = WitnessVar::<C1>::new_witness(cs.clone(), || {
|
||||
|
Ok(self.w_i.unwrap_or(w_dummy_native.clone()))
|
||||
|
})?;
|
||||
|
let U_i = CommittedInstanceVar::<C1>::new_witness(cs.clone(), || {
|
||||
|
Ok(self.U_i.unwrap_or(u_dummy_native.clone()))
|
||||
|
})?;
|
||||
|
let W_i = WitnessVar::<C1>::new_witness(cs.clone(), || {
|
||||
|
Ok(self.W_i.unwrap_or(w_dummy_native.clone()))
|
||||
|
})?;
|
||||
|
|
||||
|
let crh_params = CRHParametersVar::<C1::ScalarField>::new_constant(
|
||||
|
cs.clone(),
|
||||
|
self.poseidon_config.clone(),
|
||||
|
)?;
|
||||
|
|
||||
|
// 1. check RelaxedR1CS of u_i
|
||||
|
let z_u: Vec<FpVar<CF1<C1>>> = [
|
||||
|
vec![FpVar::<CF1<C1>>::one()],
|
||||
|
u_i.x.to_vec(),
|
||||
|
w_i.W.to_vec(),
|
||||
|
]
|
||||
|
.concat();
|
||||
|
RelaxedR1CSGadget::<C1::ScalarField, CF1<C1>, FpVar<CF1<C1>>>::check(
|
||||
|
r1cs.clone(),
|
||||
|
w_i.E,
|
||||
|
u_i.u.clone(),
|
||||
|
z_u,
|
||||
|
)?;
|
||||
|
|
||||
|
// 2. check RelaxedR1CS of U_i
|
||||
|
let z_U: Vec<FpVar<CF1<C1>>> =
|
||||
|
[vec![U_i.u.clone()], U_i.x.to_vec(), W_i.W.to_vec()].concat();
|
||||
|
RelaxedR1CSGadget::<C1::ScalarField, CF1<C1>, FpVar<CF1<C1>>>::check(
|
||||
|
r1cs,
|
||||
|
W_i.E,
|
||||
|
U_i.u.clone(),
|
||||
|
z_U,
|
||||
|
)?;
|
||||
|
|
||||
|
// 3. u_i.cmE==cm(0), u_i.u==1
|
||||
|
// Here zero_x & zero_y are the x & y coordinates of the zero point affine representation.
|
||||
|
let zero_x = NonNativeFieldVar::<C1::BaseField, C1::ScalarField>::new_constant(
|
||||
|
cs.clone(),
|
||||
|
C1::BaseField::zero(),
|
||||
|
)?
|
||||
|
.to_constraint_field()?;
|
||||
|
let zero_y = NonNativeFieldVar::<C1::BaseField, C1::ScalarField>::new_constant(
|
||||
|
cs.clone(),
|
||||
|
C1::BaseField::one(),
|
||||
|
)?
|
||||
|
.to_constraint_field()?;
|
||||
|
(u_i.cmE.x.is_eq(&zero_x)?).enforce_equal(&Boolean::TRUE)?;
|
||||
|
(u_i.cmE.y.is_eq(&zero_y)?).enforce_equal(&Boolean::TRUE)?;
|
||||
|
(u_i.u.is_one()?).enforce_equal(&Boolean::TRUE)?;
|
||||
|
|
||||
|
// 4. u_i.x == H(i, z_0, z_i, U_i)
|
||||
|
let u_i_x = U_i
|
||||
|
.clone()
|
||||
|
.hash(&crh_params, i.clone(), z_0.clone(), z_i.clone())?;
|
||||
|
(u_i.x[0]).enforce_equal(&u_i_x)?;
|
||||
|
|
||||
|
// The following two checks (and their respective allocations) are disabled for normal
|
||||
|
// tests since they take ~24.5M constraints and would take several minutes (and RAM) to run
|
||||
|
// the test
|
||||
|
#[cfg(not(test))]
|
||||
|
{
|
||||
|
// imports here instead of at the top of the file, so we avoid having multiple
|
||||
|
// `#[cfg(not(test))]
|
||||
|
use crate::folding::nova::cyclefold::{CycleFoldCommittedInstanceVar, CF_IO_LEN};
|
||||
|
use crate::pedersen::PedersenGadget;
|
||||
|
use ark_r1cs_std::ToBitsGadget;
|
||||
|
|
||||
|
let cf_r1cs = R1CSVar::<
|
||||
|
C1::BaseField,
|
||||
|
CF1<C1>,
|
||||
|
NonNativeFieldVar<C1::BaseField, CF1<C1>>,
|
||||
|
>::new_witness(cs.clone(), || Ok(self.cf_r1cs.clone()))?;
|
||||
|
|
||||
|
let cf_u_dummy_native = CommittedInstance::<C2>::dummy(CF_IO_LEN);
|
||||
|
let w_dummy_native = Witness::<C2>::new(
|
||||
|
vec![C2::ScalarField::zero(); self.cf_r1cs.A.n_cols - 1 - self.cf_r1cs.l],
|
||||
|
self.cf_E_len,
|
||||
|
);
|
||||
|
let cf_U_i = CycleFoldCommittedInstanceVar::<C2, GC2>::new_witness(cs.clone(), || {
|
||||
|
Ok(self.cf_U_i.unwrap_or_else(|| cf_u_dummy_native.clone()))
|
||||
|
})?;
|
||||
|
let cf_W_i = CycleFoldWitnessVar::<C2>::new_witness(cs.clone(), || {
|
||||
|
Ok(self.cf_W_i.unwrap_or(w_dummy_native.clone()))
|
||||
|
})?;
|
||||
|
|
||||
|
// 5. check Pedersen commitments of cf_U_i.{cmE, cmW}
|
||||
|
let H = GC2::new_constant(cs.clone(), self.cf_pedersen_params.h)?;
|
||||
|
let G = Vec::<GC2>::new_constant(cs.clone(), self.cf_pedersen_params.generators)?;
|
||||
|
let cf_W_i_E_bits: Vec<Vec<Boolean<CF1<C1>>>> = cf_W_i
|
||||
|
.E
|
||||
|
.iter()
|
||||
|
.map(|E_i| E_i.to_bits_le().unwrap())
|
||||
|
.collect();
|
||||
|
let cf_W_i_W_bits: Vec<Vec<Boolean<CF1<C1>>>> = cf_W_i
|
||||
|
.W
|
||||
|
.iter()
|
||||
|
.map(|W_i| W_i.to_bits_le().unwrap())
|
||||
|
.collect();
|
||||
|
|
||||
|
let computed_cmE = PedersenGadget::<C2, GC2>::commit(
|
||||
|
H.clone(),
|
||||
|
G.clone(),
|
||||
|
cf_W_i_E_bits,
|
||||
|
cf_W_i.rE.to_bits_le()?,
|
||||
|
)?;
|
||||
|
cf_U_i.cmE.enforce_equal(&computed_cmE)?;
|
||||
|
let computed_cmW =
|
||||
|
PedersenGadget::<C2, GC2>::commit(H, G, cf_W_i_W_bits, cf_W_i.rW.to_bits_le()?)?;
|
||||
|
cf_U_i.cmW.enforce_equal(&computed_cmW)?;
|
||||
|
|
||||
|
// 6. check RelaxedR1CS of cf_U_i
|
||||
|
let cf_z_U: Vec<NonNativeFieldVar<C2::ScalarField, CF1<C1>>> =
|
||||
|
[vec![cf_U_i.u.clone()], cf_U_i.x.to_vec(), cf_W_i.W.to_vec()].concat();
|
||||
|
RelaxedR1CSGadget::<
|
||||
|
C2::ScalarField,
|
||||
|
CF1<C1>,
|
||||
|
NonNativeFieldVar<C2::ScalarField, CF1<C1>>,
|
||||
|
>::check(cf_r1cs, cf_W_i.E, cf_U_i.u.clone(), cf_z_U)?;
|
||||
|
}
|
||||
|
|
||||
|
Ok(())
|
||||
|
}
|
||||
|
}
|
||||
|
|
||||
|
#[cfg(test)]
|
||||
|
pub mod tests {
|
||||
|
use super::*;
|
||||
|
use ark_crypto_primitives::crh::{
|
||||
|
sha256::{
|
||||
|
constraints::{Sha256Gadget, UnitVar},
|
||||
|
Sha256,
|
||||
|
},
|
||||
|
CRHScheme, CRHSchemeGadget,
|
||||
|
};
|
||||
|
use ark_ff::BigInteger;
|
||||
|
use ark_pallas::{constraints::GVar, Fq, Fr, Projective};
|
||||
|
use ark_r1cs_std::{
|
||||
|
alloc::AllocVar,
|
||||
|
bits::uint8::UInt8,
|
||||
|
eq::EqGadget,
|
||||
|
fields::{fp::FpVar, nonnative::NonNativeFieldVar},
|
||||
|
};
|
||||
|
use ark_relations::r1cs::ConstraintSystem;
|
||||
|
use ark_vesta::{constraints::GVar as GVar2, Projective as Projective2};
|
||||
|
|
||||
|
use crate::folding::nova::circuits::{tests::TestFCircuit, FCircuit};
|
||||
|
use crate::folding::nova::ivc::IVC;
|
||||
|
use crate::transcript::poseidon::tests::poseidon_test_config;
|
||||
|
|
||||
|
use crate::ccs::r1cs::{
|
||||
|
tests::{get_test_r1cs, get_test_z},
|
||||
|
R1CS,
|
||||
|
};
|
||||
|
use crate::frontend::arkworks::{extract_r1cs_and_z, tests::TestCircuit};
|
||||
|
|
||||
|
#[test]
|
||||
|
fn test_relaxed_r1cs_small_gadget_handcrafted() {
|
||||
|
let r1cs: R1CS<Fr> = get_test_r1cs();
|
||||
|
let rel_r1cs = r1cs.clone().relax();
|
||||
|
let z = get_test_z(3);
|
||||
|
|
||||
|
let cs = ConstraintSystem::<Fr>::new_ref();
|
||||
|
|
||||
|
let zVar = Vec::<FpVar<Fr>>::new_witness(cs.clone(), || Ok(z)).unwrap();
|
||||
|
let EVar = Vec::<FpVar<Fr>>::new_witness(cs.clone(), || Ok(rel_r1cs.E)).unwrap();
|
||||
|
let uVar = FpVar::<Fr>::new_witness(cs.clone(), || Ok(rel_r1cs.u)).unwrap();
|
||||
|
let r1csVar = R1CSVar::<Fr, Fr, FpVar<Fr>>::new_witness(cs.clone(), || Ok(r1cs)).unwrap();
|
||||
|
|
||||
|
RelaxedR1CSGadget::<Fr, Fr, FpVar<Fr>>::check(r1csVar, EVar, uVar, zVar).unwrap();
|
||||
|
assert!(cs.is_satisfied().unwrap());
|
||||
|
}
|
||||
|
|
||||
|
// gets as input a circuit that implements the ConstraintSynthesizer trait, and that has been
|
||||
|
// initialized.
|
||||
|
fn test_relaxed_r1cs_gadget<CS: ConstraintSynthesizer<Fr>>(circuit: CS) {
|
||||
|
let cs = ConstraintSystem::<Fr>::new_ref();
|
||||
|
|
||||
|
circuit.generate_constraints(cs.clone()).unwrap();
|
||||
|
cs.finalize();
|
||||
|
assert!(cs.is_satisfied().unwrap());
|
||||
|
|
||||
|
let cs = cs.into_inner().unwrap();
|
||||
|
|
||||
|
let (r1cs, z) = extract_r1cs_and_z::<Fr>(&cs);
|
||||
|
r1cs.check_relation(&z).unwrap();
|
||||
|
|
||||
|
let relaxed_r1cs = r1cs.clone().relax();
|
||||
|
relaxed_r1cs.check_relation(&z).unwrap();
|
||||
|
|
||||
|
// set new CS for the circuit that checks the RelaxedR1CS of our original circuit
|
||||
|
let cs = ConstraintSystem::<Fr>::new_ref();
|
||||
|
// prepare the inputs for our circuit
|
||||
|
let zVar = Vec::<FpVar<Fr>>::new_witness(cs.clone(), || Ok(z)).unwrap();
|
||||
|
let EVar = Vec::<FpVar<Fr>>::new_witness(cs.clone(), || Ok(relaxed_r1cs.E)).unwrap();
|
||||
|
let uVar = FpVar::<Fr>::new_witness(cs.clone(), || Ok(relaxed_r1cs.u)).unwrap();
|
||||
|
let r1csVar = R1CSVar::<Fr, Fr, FpVar<Fr>>::new_witness(cs.clone(), || Ok(r1cs)).unwrap();
|
||||
|
|
||||
|
RelaxedR1CSGadget::<Fr, Fr, FpVar<Fr>>::check(r1csVar, EVar, uVar, zVar).unwrap();
|
||||
|
assert!(cs.is_satisfied().unwrap());
|
||||
|
}
|
||||
|
|
||||
|
#[test]
|
||||
|
fn test_relaxed_r1cs_small_gadget_arkworks() {
|
||||
|
let x = Fr::from(5_u32);
|
||||
|
let y = x * x * x + x + Fr::from(5_u32);
|
||||
|
let circuit = TestCircuit::<Fr> { x, y };
|
||||
|
test_relaxed_r1cs_gadget(circuit);
|
||||
|
}
|
||||
|
|
||||
|
struct Sha256TestCircuit<F: PrimeField> {
|
||||
|
_f: PhantomData<F>,
|
||||
|
pub x: Vec<u8>,
|
||||
|
pub y: Vec<u8>,
|
||||
|
}
|
||||
|
impl<F: PrimeField> ConstraintSynthesizer<F> for Sha256TestCircuit<F> {
|
||||
|
fn generate_constraints(self, cs: ConstraintSystemRef<F>) -> Result<(), SynthesisError> {
|
||||
|
let x = Vec::<UInt8<F>>::new_witness(cs.clone(), || Ok(self.x))?;
|
||||
|
let y = Vec::<UInt8<F>>::new_input(cs.clone(), || Ok(self.y))?;
|
||||
|
|
||||
|
let unitVar = UnitVar::default();
|
||||
|
let comp_y = <Sha256Gadget<F> as CRHSchemeGadget<Sha256, F>>::evaluate(&unitVar, &x)?;
|
||||
|
comp_y.0.enforce_equal(&y)?;
|
||||
|
Ok(())
|
||||
|
}
|
||||
|
}
|
||||
|
#[test]
|
||||
|
fn test_relaxed_r1cs_medium_gadget_arkworks() {
|
||||
|
let x = Fr::from(5_u32).into_bigint().to_bytes_le();
|
||||
|
let y = <Sha256 as CRHScheme>::evaluate(&(), x.clone()).unwrap();
|
||||
|
|
||||
|
let circuit = Sha256TestCircuit::<Fr> {
|
||||
|
_f: PhantomData,
|
||||
|
x,
|
||||
|
y,
|
||||
|
};
|
||||
|
test_relaxed_r1cs_gadget(circuit);
|
||||
|
}
|
||||
|
|
||||
|
// circuit that has the number of constraints specified in the `n_constraints` parameter. Note
|
||||
|
// that the generated circuit will have very sparse matrices, so the resulting constraints
|
||||
|
// number of the RelaxedR1CS gadget must take that into account.
|
||||
|
struct CustomTestCircuit<F: PrimeField> {
|
||||
|
_f: PhantomData<F>,
|
||||
|
pub n_constraints: usize,
|
||||
|
pub x: F,
|
||||
|
pub y: F,
|
||||
|
}
|
||||
|
impl<F: PrimeField> CustomTestCircuit<F> {
|
||||
|
fn new(n_constraints: usize) -> Self {
|
||||
|
let x = F::from(5_u32);
|
||||
|
let mut y = F::one();
|
||||
|
for _ in 0..n_constraints - 1 {
|
||||
|
y *= x;
|
||||
|
}
|
||||
|
Self {
|
||||
|
_f: PhantomData,
|
||||
|
n_constraints,
|
||||
|
x,
|
||||
|
y,
|
||||
|
}
|
||||
|
}
|
||||
|
}
|
||||
|
impl<F: PrimeField> ConstraintSynthesizer<F> for CustomTestCircuit<F> {
|
||||
|
fn generate_constraints(self, cs: ConstraintSystemRef<F>) -> Result<(), SynthesisError> {
|
||||
|
let x = FpVar::<F>::new_witness(cs.clone(), || Ok(self.x))?;
|
||||
|
let y = FpVar::<F>::new_input(cs.clone(), || Ok(self.y))?;
|
||||
|
|
||||
|
let mut comp_y = FpVar::<F>::new_witness(cs.clone(), || Ok(F::one()))?;
|
||||
|
for _ in 0..self.n_constraints - 1 {
|
||||
|
comp_y *= x.clone();
|
||||
|
}
|
||||
|
|
||||
|
comp_y.enforce_equal(&y)?;
|
||||
|
Ok(())
|
||||
|
}
|
||||
|
}
|
||||
|
|
||||
|
#[test]
|
||||
|
fn test_relaxed_r1cs_custom_circuit() {
|
||||
|
let n_constraints = 10_000;
|
||||
|
let x = Fr::from(5_u32);
|
||||
|
let mut y = Fr::one();
|
||||
|
for _ in 0..n_constraints - 1 {
|
||||
|
y *= x;
|
||||
|
}
|
||||
|
|
||||
|
let circuit = CustomTestCircuit::<Fr> {
|
||||
|
_f: PhantomData,
|
||||
|
n_constraints,
|
||||
|
x,
|
||||
|
y,
|
||||
|
};
|
||||
|
test_relaxed_r1cs_gadget(circuit);
|
||||
|
}
|
||||
|
|
||||
|
#[test]
|
||||
|
fn test_relaxed_r1cs_nonnative_circuit() {
|
||||
|
let cs = ConstraintSystem::<Fq>::new_ref();
|
||||
|
// in practice we would use CycleFoldCircuit, but is a very big circuit (when computed
|
||||
|
// non-natively inside the RelaxedR1CS circuit), so in order to have a short test we use a
|
||||
|
// custom circuit.
|
||||
|
let circuit = CustomTestCircuit::<Fq>::new(10);
|
||||
|
circuit.generate_constraints(cs.clone()).unwrap();
|
||||
|
cs.finalize();
|
||||
|
let cs = cs.into_inner().unwrap();
|
||||
|
let (r1cs, z) = extract_r1cs_and_z::<Fq>(&cs);
|
||||
|
|
||||
|
let relaxed_r1cs = r1cs.clone().relax();
|
||||
|
|
||||
|
// natively
|
||||
|
let cs = ConstraintSystem::<Fq>::new_ref();
|
||||
|
let zVar = Vec::<FpVar<Fq>>::new_witness(cs.clone(), || Ok(z.clone())).unwrap();
|
||||
|
let EVar =
|
||||
|
Vec::<FpVar<Fq>>::new_witness(cs.clone(), || Ok(relaxed_r1cs.clone().E)).unwrap();
|
||||
|
let uVar = FpVar::<Fq>::new_witness(cs.clone(), || Ok(relaxed_r1cs.u)).unwrap();
|
||||
|
let r1csVar =
|
||||
|
R1CSVar::<Fq, Fq, FpVar<Fq>>::new_witness(cs.clone(), || Ok(r1cs.clone())).unwrap();
|
||||
|
RelaxedR1CSGadget::<Fq, Fq, FpVar<Fq>>::check(r1csVar, EVar, uVar, zVar).unwrap();
|
||||
|
|
||||
|
// non-natively
|
||||
|
let cs = ConstraintSystem::<Fr>::new_ref();
|
||||
|
let zVar = Vec::<NonNativeFieldVar<Fq, Fr>>::new_witness(cs.clone(), || Ok(z)).unwrap();
|
||||
|
let EVar = Vec::<NonNativeFieldVar<Fq, Fr>>::new_witness(cs.clone(), || Ok(relaxed_r1cs.E))
|
||||
|
.unwrap();
|
||||
|
let uVar =
|
||||
|
NonNativeFieldVar::<Fq, Fr>::new_witness(cs.clone(), || Ok(relaxed_r1cs.u)).unwrap();
|
||||
|
let r1csVar =
|
||||
|
R1CSVar::<Fq, Fr, NonNativeFieldVar<Fq, Fr>>::new_witness(cs.clone(), || Ok(r1cs))
|
||||
|
.unwrap();
|
||||
|
RelaxedR1CSGadget::<Fq, Fr, NonNativeFieldVar<Fq, Fr>>::check(r1csVar, EVar, uVar, zVar)
|
||||
|
.unwrap();
|
||||
|
}
|
||||
|
|
||||
|
#[test]
|
||||
|
fn test_decider_circuit() {
|
||||
|
let mut rng = ark_std::test_rng();
|
||||
|
let poseidon_config = poseidon_test_config::<Fr>();
|
||||
|
|
||||
|
let F_circuit = TestFCircuit::<Fr>::new();
|
||||
|
let z_0 = vec![Fr::from(3_u32)];
|
||||
|
|
||||
|
// generate an IVC and do a step of it
|
||||
|
let mut ivc = IVC::<Projective, GVar, Projective2, GVar2, TestFCircuit<Fr>>::new(
|
||||
|
&mut rng,
|
||||
|
poseidon_config,
|
||||
|
F_circuit,
|
||||
|
z_0.clone(),
|
||||
|
)
|
||||
|
.unwrap();
|
||||
|
ivc.prove_step().unwrap();
|
||||
|
ivc.verify(z_0, 1).unwrap();
|
||||
|
|
||||
|
// load the DeciderCircuit from the generated IVC
|
||||
|
let decider_circuit = DeciderCircuit::<Projective, GVar, Projective2, GVar2>::from_ivc(ivc);
|
||||
|
|
||||
|
let cs = ConstraintSystem::<Fr>::new_ref();
|
||||
|
|
||||
|
// generate the constraints and check that are satisfied by the inputs
|
||||
|
decider_circuit.generate_constraints(cs.clone()).unwrap();
|
||||
|
assert!(cs.is_satisfied().unwrap());
|
||||
|
dbg!(cs.num_constraints());
|
||||
|
}
|
||||
|
}
|
@ -0,0 +1,105 @@ |
|||||
|
use ark_ff::PrimeField;
|
||||
|
use ark_r1cs_std::{
|
||||
|
alloc::{AllocVar, AllocationMode},
|
||||
|
fields::FieldVar,
|
||||
|
};
|
||||
|
use ark_relations::r1cs::{Namespace, SynthesisError};
|
||||
|
use core::{borrow::Borrow, marker::PhantomData};
|
||||
|
|
||||
|
use crate::utils::vec::SparseMatrix;
|
||||
|
|
||||
|
pub fn mat_vec_mul_sparse<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>>(
|
||||
|
m: SparseMatrixVar<F, CF, FV>,
|
||||
|
v: Vec<FV>,
|
||||
|
) -> Vec<FV> {
|
||||
|
let mut res = vec![FV::zero(); m.n_rows];
|
||||
|
for (row_i, row) in m.coeffs.iter().enumerate() {
|
||||
|
for (value, col_i) in row.iter() {
|
||||
|
res[row_i] += value.clone().mul(&v[*col_i].clone());
|
||||
|
}
|
||||
|
}
|
||||
|
res
|
||||
|
}
|
||||
|
pub fn vec_add<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>>(
|
||||
|
a: &Vec<FV>,
|
||||
|
b: &Vec<FV>,
|
||||
|
) -> Result<Vec<FV>, SynthesisError> {
|
||||
|
if a.len() != b.len() {
|
||||
|
return Err(SynthesisError::Unsatisfiable);
|
||||
|
}
|
||||
|
let mut r: Vec<FV> = vec![FV::zero(); a.len()];
|
||||
|
for i in 0..a.len() {
|
||||
|
r[i] = a[i].clone() + b[i].clone();
|
||||
|
}
|
||||
|
Ok(r)
|
||||
|
}
|
||||
|
pub fn vec_scalar_mul<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>>(
|
||||
|
vec: &Vec<FV>,
|
||||
|
c: &FV,
|
||||
|
) -> Vec<FV> {
|
||||
|
let mut result = vec![FV::zero(); vec.len()];
|
||||
|
for (i, a) in vec.iter().enumerate() {
|
||||
|
result[i] = a.clone() * c;
|
||||
|
}
|
||||
|
result
|
||||
|
}
|
||||
|
pub fn hadamard<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>>(
|
||||
|
a: &Vec<FV>,
|
||||
|
b: &Vec<FV>,
|
||||
|
) -> Result<Vec<FV>, SynthesisError> {
|
||||
|
if a.len() != b.len() {
|
||||
|
return Err(SynthesisError::Unsatisfiable);
|
||||
|
}
|
||||
|
let mut r: Vec<FV> = vec![FV::zero(); a.len()];
|
||||
|
for i in 0..a.len() {
|
||||
|
r[i] = a[i].clone() * b[i].clone();
|
||||
|
}
|
||||
|
Ok(r)
|
||||
|
}
|
||||
|
|
||||
|
#[derive(Debug, Clone)]
|
||||
|
pub struct SparseMatrixVar<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>> {
|
||||
|
_f: PhantomData<F>,
|
||||
|
_cf: PhantomData<CF>,
|
||||
|
_fv: PhantomData<FV>,
|
||||
|
pub n_rows: usize,
|
||||
|
pub n_cols: usize,
|
||||
|
// same format as the native SparseMatrix (which follows ark_relations::r1cs::Matrix format
|
||||
|
pub coeffs: Vec<Vec<(FV, usize)>>,
|
||||
|
}
|
||||
|
|
||||
|
impl<F, CF, FV> AllocVar<SparseMatrix<F>, CF> for SparseMatrixVar<F, CF, FV>
|
||||
|
where
|
||||
|
F: PrimeField,
|
||||
|
CF: PrimeField,
|
||||
|
FV: FieldVar<F, CF>,
|
||||
|
{
|
||||
|
fn new_variable<T: Borrow<SparseMatrix<F>>>(
|
||||
|
cs: impl Into<Namespace<CF>>,
|
||||
|
f: impl FnOnce() -> Result<T, SynthesisError>,
|
||||
|
mode: AllocationMode,
|
||||
|
) -> Result<Self, SynthesisError> {
|
||||
|
f().and_then(|val| {
|
||||
|
let cs = cs.into();
|
||||
|
|
||||
|
let mut coeffs: Vec<Vec<(FV, usize)>> = Vec::new();
|
||||
|
for row in val.borrow().coeffs.iter() {
|
||||
|
let mut rowVar: Vec<(FV, usize)> = Vec::new();
|
||||
|
for &(value, col_i) in row.iter() {
|
||||
|
let coeffVar = FV::new_variable(cs.clone(), || Ok(value), mode)?;
|
||||
|
rowVar.push((coeffVar, col_i));
|
||||
|
}
|
||||
|
coeffs.push(rowVar);
|
||||
|
}
|
||||
|
|
||||
|
Ok(Self {
|
||||
|
_f: PhantomData,
|
||||
|
_cf: PhantomData,
|
||||
|
_fv: PhantomData,
|
||||
|
n_rows: val.borrow().n_rows,
|
||||
|
n_cols: val.borrow().n_cols,
|
||||
|
coeffs,
|
||||
|
})
|
||||
|
})
|
||||
|
}
|
||||
|
}
|