sonobe/src/folding/nova/decider.rs
src/folding/nova/decider.rsRaw
1/// This file implements the onchain (Ethereum's EVM) decider circuit. For non-ethereum use cases,
2/// other more efficient approaches can be used.
3use ark_crypto_primitives::crh::poseidon::constraints::CRHParametersVar;
4use ark_crypto_primitives::sponge::{poseidon::PoseidonConfig, Absorb};
5use ark_ec::{CurveGroup, Group};
6use ark_ff::PrimeField;
7use ark_r1cs_std::{
8 alloc::{AllocVar, AllocationMode},
9 boolean::Boolean,
10 eq::EqGadget,
11 fields::{fp::FpVar, nonnative::NonNativeFieldVar, FieldVar},
12 groups::GroupOpsBounds,
13 prelude::CurveVar,
14 ToConstraintFieldGadget,
15};
16use ark_relations::r1cs::{ConstraintSynthesizer, ConstraintSystemRef, Namespace, SynthesisError};
17use ark_std::{One, Zero};
18use core::{borrow::Borrow, marker::PhantomData};
19
20use crate::ccs::r1cs::R1CS;
21use crate::commitment::{pedersen::Params as PedersenParams, CommitmentProver};
22use crate::folding::nova::{
23 circuits::{CommittedInstanceVar, CF1, CF2},
24 ivc::IVC,
25 CommittedInstance, Witness,
26};
27use crate::frontend::FCircuit;
28use crate::utils::gadgets::{
29 hadamard, mat_vec_mul_sparse, vec_add, vec_scalar_mul, SparseMatrixVar,
30};
31
32#[derive(Debug, Clone)]
33pub struct RelaxedR1CSGadget<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>> {
34 _f: PhantomData<F>,
35 _cf: PhantomData<CF>,
36 _fv: PhantomData<FV>,
37}
38impl<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>> RelaxedR1CSGadget<F, CF, FV> {
39 /// performs the RelaxedR1CS check (Az∘Bz==uCz+E)
40 pub fn check(
41 r1cs: R1CSVar<F, CF, FV>,
42 E: Vec<FV>,
43 u: FV,
44 z: Vec<FV>,
45 ) -> Result<(), SynthesisError> {
46 let Az = mat_vec_mul_sparse(r1cs.A, z.clone());
47 let Bz = mat_vec_mul_sparse(r1cs.B, z.clone());
48 let Cz = mat_vec_mul_sparse(r1cs.C, z.clone());
49 let uCz = vec_scalar_mul(&Cz, &u);
50 let uCzE = vec_add(&uCz, &E)?;
51 let AzBz = hadamard(&Az, &Bz)?;
52 for i in 0..AzBz.len() {
53 AzBz[i].enforce_equal(&uCzE[i].clone())?;
54 }
55 Ok(())
56 }
57}
58
59#[derive(Debug, Clone)]
60pub struct R1CSVar<F: PrimeField, CF: PrimeField, FV: FieldVar<F, CF>> {
61 _f: PhantomData<F>,
62 _cf: PhantomData<CF>,
63 _fv: PhantomData<FV>,
64 pub A: SparseMatrixVar<F, CF, FV>,
65 pub B: SparseMatrixVar<F, CF, FV>,
66 pub C: SparseMatrixVar<F, CF, FV>,
67}
68
69impl<F, CF, FV> AllocVar<R1CS<F>, CF> for R1CSVar<F, CF, FV>
70where
71 F: PrimeField,
72 CF: PrimeField,
73 FV: FieldVar<F, CF>,
74{
75 fn new_variable<T: Borrow<R1CS<F>>>(
76 cs: impl Into<Namespace<CF>>,
77 f: impl FnOnce() -> Result<T, SynthesisError>,
78 _mode: AllocationMode,
79 ) -> Result<Self, SynthesisError> {
80 f().and_then(|val| {
81 let cs = cs.into();
82
83 let A = SparseMatrixVar::<F, CF, FV>::new_constant(cs.clone(), &val.borrow().A)?;
84 let B = SparseMatrixVar::<F, CF, FV>::new_constant(cs.clone(), &val.borrow().B)?;
85 let C = SparseMatrixVar::<F, CF, FV>::new_constant(cs.clone(), &val.borrow().C)?;
86
87 Ok(Self {
88 _f: PhantomData,
89 _cf: PhantomData,
90 _fv: PhantomData,
91 A,
92 B,
93 C,
94 })
95 })
96 }
97}
98
99/// In-circuit representation of the Witness associated to the CommittedInstance.
100#[derive(Debug, Clone)]
101pub struct WitnessVar<C: CurveGroup> {
102 pub E: Vec<FpVar<C::ScalarField>>,
103 pub rE: FpVar<C::ScalarField>,
104 pub W: Vec<FpVar<C::ScalarField>>,
105 pub rW: FpVar<C::ScalarField>,
106}
107
108impl<C> AllocVar<Witness<C>, CF1<C>> for WitnessVar<C>
109where
110 C: CurveGroup,
111 <C as ark_ec::CurveGroup>::BaseField: PrimeField,
112{
113 fn new_variable<T: Borrow<Witness<C>>>(
114 cs: impl Into<Namespace<CF1<C>>>,
115 f: impl FnOnce() -> Result<T, SynthesisError>,
116 mode: AllocationMode,
117 ) -> Result<Self, SynthesisError> {
118 f().and_then(|val| {
119 let cs = cs.into();
120
121 let E: Vec<FpVar<C::ScalarField>> =
122 Vec::new_variable(cs.clone(), || Ok(val.borrow().E.clone()), mode)?;
123 let rE =
124 FpVar::<C::ScalarField>::new_variable(cs.clone(), || Ok(val.borrow().rE), mode)?;
125
126 let W: Vec<FpVar<C::ScalarField>> =
127 Vec::new_variable(cs.clone(), || Ok(val.borrow().W.clone()), mode)?;
128 let rW =
129 FpVar::<C::ScalarField>::new_variable(cs.clone(), || Ok(val.borrow().rW), mode)?;
130
131 Ok(Self { E, rE, W, rW })
132 })
133 }
134}
135
136/// In-circuit representation of the Witness associated to the CommittedInstance, but with
137/// non-native representation, since it is used to represent the CycleFold witness.
138#[derive(Debug, Clone)]
139pub struct CycleFoldWitnessVar<C: CurveGroup> {
140 pub E: Vec<NonNativeFieldVar<C::ScalarField, CF2<C>>>,
141 pub rE: NonNativeFieldVar<C::ScalarField, CF2<C>>,
142 pub W: Vec<NonNativeFieldVar<C::ScalarField, CF2<C>>>,
143 pub rW: NonNativeFieldVar<C::ScalarField, CF2<C>>,
144}
145
146impl<C> AllocVar<Witness<C>, CF2<C>> for CycleFoldWitnessVar<C>
147where
148 C: CurveGroup,
149 <C as ark_ec::CurveGroup>::BaseField: PrimeField,
150{
151 fn new_variable<T: Borrow<Witness<C>>>(
152 cs: impl Into<Namespace<CF2<C>>>,
153 f: impl FnOnce() -> Result<T, SynthesisError>,
154 mode: AllocationMode,
155 ) -> Result<Self, SynthesisError> {
156 f().and_then(|val| {
157 let cs = cs.into();
158
159 let E: Vec<NonNativeFieldVar<C::ScalarField, CF2<C>>> =
160 Vec::new_variable(cs.clone(), || Ok(val.borrow().E.clone()), mode)?;
161 let rE = NonNativeFieldVar::<C::ScalarField, CF2<C>>::new_variable(
162 cs.clone(),
163 || Ok(val.borrow().rE),
164 mode,
165 )?;
166
167 let W: Vec<NonNativeFieldVar<C::ScalarField, CF2<C>>> =
168 Vec::new_variable(cs.clone(), || Ok(val.borrow().W.clone()), mode)?;
169 let rW = NonNativeFieldVar::<C::ScalarField, CF2<C>>::new_variable(
170 cs.clone(),
171 || Ok(val.borrow().rW),
172 mode,
173 )?;
174
175 Ok(Self { E, rE, W, rW })
176 })
177 }
178}
179
180/// Circuit that implements the in-circuit checks needed for the onchain (Ethereum's EVM)
181/// verification.
182pub struct DeciderCircuit<C1, GC1, C2, GC2, CP1, CP2>
183where
184 C1: CurveGroup,
185 GC1: CurveVar<C1, CF2<C1>>,
186 C2: CurveGroup,
187 GC2: CurveVar<C2, CF2<C2>>,
188 CP1: CommitmentProver<C1>,
189 CP2: CommitmentProver<C2>,
190{
191 _c1: PhantomData<C1>,
192 _gc1: PhantomData<GC1>,
193 _c2: PhantomData<C2>,
194 _gc2: PhantomData<GC2>,
195 _cp1: PhantomData<CP1>,
196 _cp2: PhantomData<CP2>,
197
198 /// E vector's length of the Nova instance witness
199 pub E_len: usize,
200 /// E vector's length of the CycleFold instance witness
201 pub cf_E_len: usize,
202 /// R1CS of the Augmented Function circuit
203 pub r1cs: R1CS<C1::ScalarField>,
204 /// R1CS of the CycleFold circuit
205 pub cf_r1cs: R1CS<C2::ScalarField>,
206 /// CycleFold PedersenParams over C2
207 pub cf_pedersen_params: PedersenParams<C2>,
208 pub poseidon_config: PoseidonConfig<CF1<C1>>,
209 pub i: Option<CF1<C1>>,
210 /// initial state
211 pub z_0: Option<Vec<C1::ScalarField>>,
212 /// current i-th state
213 pub z_i: Option<Vec<C1::ScalarField>>,
214 /// Nova instances
215 pub u_i: Option<CommittedInstance<C1>>,
216 pub w_i: Option<Witness<C1>>,
217 pub U_i: Option<CommittedInstance<C1>>,
218 pub W_i: Option<Witness<C1>>,
219 /// CycleFold running instance
220 pub cf_U_i: Option<CommittedInstance<C2>>,
221 pub cf_W_i: Option<Witness<C2>>,
222}
223impl<C1, GC1, C2, GC2, CP1, CP2> DeciderCircuit<C1, GC1, C2, GC2, CP1, CP2>
224where
225 C1: CurveGroup,
226 C2: CurveGroup,
227 GC1: CurveVar<C1, CF2<C1>>,
228 GC2: CurveVar<C2, CF2<C2>>,
229 CP1: CommitmentProver<C1>,
230 // enforce that the CP2 is Pedersen commitment, since we're at Ethereum's EVM decider
231 CP2: CommitmentProver<C2, Params = PedersenParams<C2>>,
232{
233 pub fn from_ivc<FC: FCircuit<C1::ScalarField>>(
234 ivc: IVC<C1, GC1, C2, GC2, FC, CP1, CP2>,
235 ) -> Self {
236 Self {
237 _c1: PhantomData,
238 _gc1: PhantomData,
239 _c2: PhantomData,
240 _gc2: PhantomData,
241 _cp1: PhantomData,
242 _cp2: PhantomData,
243
244 E_len: ivc.W_i.E.len(),
245 cf_E_len: ivc.cf_W_i.E.len(),
246 r1cs: ivc.r1cs,
247 cf_r1cs: ivc.cf_r1cs,
248 cf_pedersen_params: ivc.cf_cm_params,
249 poseidon_config: ivc.poseidon_config,
250 i: Some(ivc.i),
251 z_0: Some(ivc.z_0),
252 z_i: Some(ivc.z_i),
253 u_i: Some(ivc.u_i),
254 w_i: Some(ivc.w_i),
255 U_i: Some(ivc.U_i),
256 W_i: Some(ivc.W_i),
257 cf_U_i: Some(ivc.cf_U_i),
258 cf_W_i: Some(ivc.cf_W_i),
259 }
260 }
261}
262
263impl<C1, GC1, C2, GC2, CP1, CP2> ConstraintSynthesizer<CF1<C1>>
264 for DeciderCircuit<C1, GC1, C2, GC2, CP1, CP2>
265where
266 C1: CurveGroup,
267 C2: CurveGroup,
268 GC1: CurveVar<C1, CF2<C1>>,
269 GC2: CurveVar<C2, CF2<C2>>,
270 CP1: CommitmentProver<C1>,
271 CP2: CommitmentProver<C2>,
272 <C1 as CurveGroup>::BaseField: PrimeField,
273 <C2 as CurveGroup>::BaseField: PrimeField,
274 <C1 as Group>::ScalarField: Absorb,
275 <C2 as Group>::ScalarField: Absorb,
276 C1: CurveGroup<BaseField = C2::ScalarField, ScalarField = C2::BaseField>,
277 for<'b> &'b GC2: GroupOpsBounds<'b, C2, GC2>,
278{
279 fn generate_constraints(self, cs: ConstraintSystemRef<CF1<C1>>) -> Result<(), SynthesisError> {
280 let r1cs =
281 R1CSVar::<C1::ScalarField, CF1<C1>, FpVar<CF1<C1>>>::new_witness(cs.clone(), || {
282 Ok(self.r1cs.clone())
283 })?;
284
285 let i = FpVar::<CF1<C1>>::new_witness(cs.clone(), || {
286 Ok(self.i.unwrap_or_else(CF1::<C1>::zero))
287 })?;
288 let z_0 = Vec::<FpVar<CF1<C1>>>::new_witness(cs.clone(), || {
289 Ok(self.z_0.unwrap_or(vec![CF1::<C1>::zero()]))
290 })?;
291 let z_i = Vec::<FpVar<CF1<C1>>>::new_witness(cs.clone(), || {
292 Ok(self.z_i.unwrap_or(vec![CF1::<C1>::zero()]))
293 })?;
294
295 let u_dummy_native = CommittedInstance::<C1>::dummy(1);
296 let w_dummy_native = Witness::<C1>::new(
297 vec![C1::ScalarField::zero(); self.r1cs.A.n_cols - 2 /* (2=1+1, since u_i.x.len=1) */],
298 self.E_len,
299 );
300
301 let u_i = CommittedInstanceVar::<C1>::new_witness(cs.clone(), || {
302 Ok(self.u_i.unwrap_or(u_dummy_native.clone()))
303 })?;
304 let w_i = WitnessVar::<C1>::new_witness(cs.clone(), || {
305 Ok(self.w_i.unwrap_or(w_dummy_native.clone()))
306 })?;
307 let U_i = CommittedInstanceVar::<C1>::new_witness(cs.clone(), || {
308 Ok(self.U_i.unwrap_or(u_dummy_native.clone()))
309 })?;
310 let W_i = WitnessVar::<C1>::new_witness(cs.clone(), || {
311 Ok(self.W_i.unwrap_or(w_dummy_native.clone()))
312 })?;
313
314 let crh_params = CRHParametersVar::<C1::ScalarField>::new_constant(
315 cs.clone(),
316 self.poseidon_config.clone(),
317 )?;
318
319 // 1. check RelaxedR1CS of u_i
320 let z_u: Vec<FpVar<CF1<C1>>> = [
321 vec![FpVar::<CF1<C1>>::one()],
322 u_i.x.to_vec(),
323 w_i.W.to_vec(),
324 ]
325 .concat();
326 RelaxedR1CSGadget::<C1::ScalarField, CF1<C1>, FpVar<CF1<C1>>>::check(
327 r1cs.clone(),
328 w_i.E,
329 u_i.u.clone(),
330 z_u,
331 )?;
332
333 // 2. check RelaxedR1CS of U_i
334 let z_U: Vec<FpVar<CF1<C1>>> =
335 [vec![U_i.u.clone()], U_i.x.to_vec(), W_i.W.to_vec()].concat();
336 RelaxedR1CSGadget::<C1::ScalarField, CF1<C1>, FpVar<CF1<C1>>>::check(
337 r1cs,
338 W_i.E,
339 U_i.u.clone(),
340 z_U,
341 )?;
342
343 // 3. u_i.cmE==cm(0), u_i.u==1
344 // Here zero_x & zero_y are the x & y coordinates of the zero point affine representation.
345 let zero_x = NonNativeFieldVar::<C1::BaseField, C1::ScalarField>::new_constant(
346 cs.clone(),
347 C1::BaseField::zero(),
348 )?
349 .to_constraint_field()?;
350 let zero_y = NonNativeFieldVar::<C1::BaseField, C1::ScalarField>::new_constant(
351 cs.clone(),
352 C1::BaseField::one(),
353 )?
354 .to_constraint_field()?;
355 (u_i.cmE.x.is_eq(&zero_x)?).enforce_equal(&Boolean::TRUE)?;
356 (u_i.cmE.y.is_eq(&zero_y)?).enforce_equal(&Boolean::TRUE)?;
357 (u_i.u.is_one()?).enforce_equal(&Boolean::TRUE)?;
358
359 // 4. u_i.x == H(i, z_0, z_i, U_i)
360 let u_i_x = U_i
361 .clone()
362 .hash(&crh_params, i.clone(), z_0.clone(), z_i.clone())?;
363 (u_i.x[0]).enforce_equal(&u_i_x)?;
364
365 // The following two checks (and their respective allocations) are disabled for normal
366 // tests since they take ~24.5M constraints and would take several minutes (and RAM) to run
367 // the test
368 #[cfg(not(test))]
369 {
370 // imports here instead of at the top of the file, so we avoid having multiple
371 // `#[cfg(not(test))]
372 use crate::commitment::pedersen::PedersenGadget;
373 use crate::folding::nova::cyclefold::{CycleFoldCommittedInstanceVar, CF_IO_LEN};
374 use ark_r1cs_std::ToBitsGadget;
375
376 let cf_r1cs = R1CSVar::<
377 C1::BaseField,
378 CF1<C1>,
379 NonNativeFieldVar<C1::BaseField, CF1<C1>>,
380 >::new_witness(cs.clone(), || Ok(self.cf_r1cs.clone()))?;
381
382 let cf_u_dummy_native = CommittedInstance::<C2>::dummy(CF_IO_LEN);
383 let w_dummy_native = Witness::<C2>::new(
384 vec![C2::ScalarField::zero(); self.cf_r1cs.A.n_cols - 1 - self.cf_r1cs.l],
385 self.cf_E_len,
386 );
387 let cf_U_i = CycleFoldCommittedInstanceVar::<C2, GC2>::new_witness(cs.clone(), || {
388 Ok(self.cf_U_i.unwrap_or_else(|| cf_u_dummy_native.clone()))
389 })?;
390 let cf_W_i = CycleFoldWitnessVar::<C2>::new_witness(cs.clone(), || {
391 Ok(self.cf_W_i.unwrap_or(w_dummy_native.clone()))
392 })?;
393
394 // 5. check Pedersen commitments of cf_U_i.{cmE, cmW}
395 let H = GC2::new_constant(cs.clone(), self.cf_pedersen_params.h)?;
396 let G = Vec::<GC2>::new_constant(cs.clone(), self.cf_pedersen_params.generators)?;
397 let cf_W_i_E_bits: Vec<Vec<Boolean<CF1<C1>>>> = cf_W_i
398 .E
399 .iter()
400 .map(|E_i| E_i.to_bits_le().unwrap())
401 .collect();
402 let cf_W_i_W_bits: Vec<Vec<Boolean<CF1<C1>>>> = cf_W_i
403 .W
404 .iter()
405 .map(|W_i| W_i.to_bits_le().unwrap())
406 .collect();
407
408 let computed_cmE = PedersenGadget::<C2, GC2>::commit(
409 H.clone(),
410 G.clone(),
411 cf_W_i_E_bits,
412 cf_W_i.rE.to_bits_le()?,
413 )?;
414 cf_U_i.cmE.enforce_equal(&computed_cmE)?;
415 let computed_cmW =
416 PedersenGadget::<C2, GC2>::commit(H, G, cf_W_i_W_bits, cf_W_i.rW.to_bits_le()?)?;
417 cf_U_i.cmW.enforce_equal(&computed_cmW)?;
418
419 // 6. check RelaxedR1CS of cf_U_i
420 let cf_z_U: Vec<NonNativeFieldVar<C2::ScalarField, CF1<C1>>> =
421 [vec![cf_U_i.u.clone()], cf_U_i.x.to_vec(), cf_W_i.W.to_vec()].concat();
422 RelaxedR1CSGadget::<
423 C2::ScalarField,
424 CF1<C1>,
425 NonNativeFieldVar<C2::ScalarField, CF1<C1>>,
426 >::check(cf_r1cs, cf_W_i.E, cf_U_i.u.clone(), cf_z_U)?;
427 }
428
429 Ok(())
430 }
431}
432
433#[cfg(test)]
434pub mod tests {
435 use super::*;
436 use ark_crypto_primitives::crh::{
437 sha256::{
438 constraints::{Sha256Gadget, UnitVar},
439 Sha256,
440 },
441 CRHScheme, CRHSchemeGadget,
442 };
443 use ark_ff::BigInteger;
444 use ark_pallas::{constraints::GVar, Fq, Fr, Projective};
445 use ark_r1cs_std::{
446 alloc::AllocVar,
447 bits::uint8::UInt8,
448 eq::EqGadget,
449 fields::{fp::FpVar, nonnative::NonNativeFieldVar},
450 };
451 use ark_relations::r1cs::ConstraintSystem;
452 use ark_vesta::{constraints::GVar as GVar2, Projective as Projective2};
453
454 use crate::commitment::pedersen::Pedersen;
455 use crate::folding::nova::ivc::tests::get_pedersen_params_len;
456 use crate::frontend::tests::{CubicFCircuit, CustomFCircuit, WrapperCircuit};
457 use crate::transcript::poseidon::tests::poseidon_test_config;
458
459 use crate::ccs::r1cs::{extract_r1cs, extract_w_x};
460 use crate::ccs::r1cs::{
461 tests::{get_test_r1cs, get_test_z},
462 R1CS,
463 };
464
465 #[test]
466 fn test_relaxed_r1cs_small_gadget_handcrafted() {
467 let r1cs: R1CS<Fr> = get_test_r1cs();
468 let rel_r1cs = r1cs.clone().relax();
469 let z = get_test_z(3);
470
471 let cs = ConstraintSystem::<Fr>::new_ref();
472
473 let zVar = Vec::<FpVar<Fr>>::new_witness(cs.clone(), || Ok(z)).unwrap();
474 let EVar = Vec::<FpVar<Fr>>::new_witness(cs.clone(), || Ok(rel_r1cs.E)).unwrap();
475 let uVar = FpVar::<Fr>::new_witness(cs.clone(), || Ok(rel_r1cs.u)).unwrap();
476 let r1csVar = R1CSVar::<Fr, Fr, FpVar<Fr>>::new_witness(cs.clone(), || Ok(r1cs)).unwrap();
477
478 RelaxedR1CSGadget::<Fr, Fr, FpVar<Fr>>::check(r1csVar, EVar, uVar, zVar).unwrap();
479 assert!(cs.is_satisfied().unwrap());
480 }
481
482 // gets as input a circuit that implements the ConstraintSynthesizer trait, and that has been
483 // initialized.
484 fn test_relaxed_r1cs_gadget<CS: ConstraintSynthesizer<Fr>>(circuit: CS) {
485 let cs = ConstraintSystem::<Fr>::new_ref();
486
487 circuit.generate_constraints(cs.clone()).unwrap();
488 cs.finalize();
489 assert!(cs.is_satisfied().unwrap());
490
491 let cs = cs.into_inner().unwrap();
492
493 let r1cs = extract_r1cs::<Fr>(&cs);
494 let (w, x) = extract_w_x::<Fr>(&cs);
495 let z = [vec![Fr::one()], x, w].concat();
496 r1cs.check_relation(&z).unwrap();
497
498 let relaxed_r1cs = r1cs.clone().relax();
499 relaxed_r1cs.check_relation(&z).unwrap();
500
501 // set new CS for the circuit that checks the RelaxedR1CS of our original circuit
502 let cs = ConstraintSystem::<Fr>::new_ref();
503 // prepare the inputs for our circuit
504 let zVar = Vec::<FpVar<Fr>>::new_witness(cs.clone(), || Ok(z)).unwrap();
505 let EVar = Vec::<FpVar<Fr>>::new_witness(cs.clone(), || Ok(relaxed_r1cs.E)).unwrap();
506 let uVar = FpVar::<Fr>::new_witness(cs.clone(), || Ok(relaxed_r1cs.u)).unwrap();
507 let r1csVar = R1CSVar::<Fr, Fr, FpVar<Fr>>::new_witness(cs.clone(), || Ok(r1cs)).unwrap();
508
509 RelaxedR1CSGadget::<Fr, Fr, FpVar<Fr>>::check(r1csVar, EVar, uVar, zVar).unwrap();
510 assert!(cs.is_satisfied().unwrap());
511 }
512
513 #[test]
514 fn test_relaxed_r1cs_small_gadget_arkworks() {
515 let z_i = vec![Fr::from(3_u32)];
516 let cubic_circuit = CubicFCircuit::<Fr>::new(());
517 let circuit = WrapperCircuit::<Fr, CubicFCircuit<Fr>> {
518 FC: cubic_circuit,
519 z_i: Some(z_i.clone()),
520 z_i1: Some(cubic_circuit.step_native(z_i).unwrap()),
521 };
522
523 test_relaxed_r1cs_gadget(circuit);
524 }
525
526 struct Sha256TestCircuit<F: PrimeField> {
527 _f: PhantomData<F>,
528 pub x: Vec<u8>,
529 pub y: Vec<u8>,
530 }
531 impl<F: PrimeField> ConstraintSynthesizer<F> for Sha256TestCircuit<F> {
532 fn generate_constraints(self, cs: ConstraintSystemRef<F>) -> Result<(), SynthesisError> {
533 let x = Vec::<UInt8<F>>::new_witness(cs.clone(), || Ok(self.x))?;
534 let y = Vec::<UInt8<F>>::new_input(cs.clone(), || Ok(self.y))?;
535
536 let unitVar = UnitVar::default();
537 let comp_y = <Sha256Gadget<F> as CRHSchemeGadget<Sha256, F>>::evaluate(&unitVar, &x)?;
538 comp_y.0.enforce_equal(&y)?;
539 Ok(())
540 }
541 }
542 #[test]
543 fn test_relaxed_r1cs_medium_gadget_arkworks() {
544 let x = Fr::from(5_u32).into_bigint().to_bytes_le();
545 let y = <Sha256 as CRHScheme>::evaluate(&(), x.clone()).unwrap();
546
547 let circuit = Sha256TestCircuit::<Fr> {
548 _f: PhantomData,
549 x,
550 y,
551 };
552 test_relaxed_r1cs_gadget(circuit);
553 }
554
555 #[test]
556 fn test_relaxed_r1cs_custom_circuit() {
557 let n_constraints = 10_000;
558 let custom_circuit = CustomFCircuit::<Fr>::new(n_constraints);
559 let z_i = vec![Fr::from(5_u32)];
560 let circuit = WrapperCircuit::<Fr, CustomFCircuit<Fr>> {
561 FC: custom_circuit,
562 z_i: Some(z_i.clone()),
563 z_i1: Some(custom_circuit.step_native(z_i).unwrap()),
564 };
565 test_relaxed_r1cs_gadget(circuit);
566 }
567
568 #[test]
569 fn test_relaxed_r1cs_nonnative_circuit() {
570 let cs = ConstraintSystem::<Fq>::new_ref();
571 // in practice we would use CycleFoldCircuit, but is a very big circuit (when computed
572 // non-natively inside the RelaxedR1CS circuit), so in order to have a short test we use a
573 // custom circuit.
574 let custom_circuit = CustomFCircuit::<Fq>::new(10);
575 let z_i = vec![Fq::from(5_u32)];
576 let circuit = WrapperCircuit::<Fq, CustomFCircuit<Fq>> {
577 FC: custom_circuit,
578 z_i: Some(z_i.clone()),
579 z_i1: Some(custom_circuit.step_native(z_i).unwrap()),
580 };
581 circuit.generate_constraints(cs.clone()).unwrap();
582 cs.finalize();
583 let cs = cs.into_inner().unwrap();
584 let r1cs = extract_r1cs::<Fq>(&cs);
585 let (w, x) = extract_w_x::<Fq>(&cs);
586 let z = [vec![Fq::one()], x, w].concat();
587
588 let relaxed_r1cs = r1cs.clone().relax();
589
590 // natively
591 let cs = ConstraintSystem::<Fq>::new_ref();
592 let zVar = Vec::<FpVar<Fq>>::new_witness(cs.clone(), || Ok(z.clone())).unwrap();
593 let EVar =
594 Vec::<FpVar<Fq>>::new_witness(cs.clone(), || Ok(relaxed_r1cs.clone().E)).unwrap();
595 let uVar = FpVar::<Fq>::new_witness(cs.clone(), || Ok(relaxed_r1cs.u)).unwrap();
596 let r1csVar =
597 R1CSVar::<Fq, Fq, FpVar<Fq>>::new_witness(cs.clone(), || Ok(r1cs.clone())).unwrap();
598 RelaxedR1CSGadget::<Fq, Fq, FpVar<Fq>>::check(r1csVar, EVar, uVar, zVar).unwrap();
599
600 // non-natively
601 let cs = ConstraintSystem::<Fr>::new_ref();
602 let zVar = Vec::<NonNativeFieldVar<Fq, Fr>>::new_witness(cs.clone(), || Ok(z)).unwrap();
603 let EVar = Vec::<NonNativeFieldVar<Fq, Fr>>::new_witness(cs.clone(), || Ok(relaxed_r1cs.E))
604 .unwrap();
605 let uVar =
606 NonNativeFieldVar::<Fq, Fr>::new_witness(cs.clone(), || Ok(relaxed_r1cs.u)).unwrap();
607 let r1csVar =
608 R1CSVar::<Fq, Fr, NonNativeFieldVar<Fq, Fr>>::new_witness(cs.clone(), || Ok(r1cs))
609 .unwrap();
610 RelaxedR1CSGadget::<Fq, Fr, NonNativeFieldVar<Fq, Fr>>::check(r1csVar, EVar, uVar, zVar)
611 .unwrap();
612 }
613
614 #[test]
615 fn test_decider_circuit() {
616 let mut rng = ark_std::test_rng();
617 let poseidon_config = poseidon_test_config::<Fr>();
618
619 let F_circuit = CubicFCircuit::<Fr>::new(());
620 let z_0 = vec![Fr::from(3_u32)];
621
622 let (pedersen_len, cf_pedersen_len) =
623 get_pedersen_params_len::<CubicFCircuit<Fr>>(&poseidon_config, F_circuit).unwrap();
624 // generate the Pedersen params
625 let pedersen_params = Pedersen::<Projective>::new_params(&mut rng, pedersen_len);
626 let cf_pedersen_params = Pedersen::<Projective2>::new_params(&mut rng, cf_pedersen_len);
627
628 // generate an IVC and do a step of it
629 let mut ivc = IVC::<
630 Projective,
631 GVar,
632 Projective2,
633 GVar2,
634 CubicFCircuit<Fr>,
635 Pedersen<Projective>,
636 Pedersen<Projective2>,
637 >::new(
638 poseidon_config,
639 pedersen_params,
640 cf_pedersen_params,
641 F_circuit,
642 z_0.clone(),
643 )
644 .unwrap();
645 ivc.prove_step().unwrap();
646 ivc.verify(z_0, 1).unwrap();
647
648 // load the DeciderCircuit from the generated IVC
649 let decider_circuit = DeciderCircuit::<
650 Projective,
651 GVar,
652 Projective2,
653 GVar2,
654 Pedersen<Projective>,
655 Pedersen<Projective2>,
656 >::from_ivc(ivc);
657
658 let cs = ConstraintSystem::<Fr>::new_ref();
659
660 // generate the constraints and check that are satisfied by the inputs
661 decider_circuit.generate_constraints(cs.clone()).unwrap();
662 assert!(cs.is_satisfied().unwrap());
663 dbg!(cs.num_constraints());
664 }
665}