diff --git a/CHANGELOG.md b/CHANGELOG.md index 2f34e61..ee102fe 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -12,6 +12,8 @@ ### Bug Fixes +- [\#70](https://github.com/arkworks-rs/r1cs-std/pull/70) Fix soundness issues of `mul_by_inverse` for field gadgets. + ## v0.3.0 ### Breaking changes diff --git a/src/fields/mod.rs b/src/fields/mod.rs index d6294df..0d622c7 100644 --- a/src/fields/mod.rs +++ b/src/fields/mod.rs @@ -155,15 +155,10 @@ pub trait FieldVar: /// Computes `result` such that `self * result == Self::one()`. fn inverse(&self) -> Result; - /// Returns `(self / d)`. but requires fewer constraints than `self * d.inverse()`. - /// It is up to the caller to ensure that `d` is non-zero, - /// since in that case the result is unconstrained. + /// Returns `(self / d)`. + /// The constraint system will be unsatisfiable when `d = 0`. fn mul_by_inverse(&self, d: &Self) -> Result { - let d_inv = if self.is_constant() || d.is_constant() { - d.inverse()? - } else { - Self::new_witness(self.cs(), || Ok(d.value()?.inverse().unwrap_or(F::zero())))? - }; + let d_inv = d.inverse()?; Ok(d_inv * self) } diff --git a/src/groups/curves/short_weierstrass/non_zero_affine.rs b/src/groups/curves/short_weierstrass/non_zero_affine.rs index 3881dd5..c4dbea3 100644 --- a/src/groups/curves/short_weierstrass/non_zero_affine.rs +++ b/src/groups/curves/short_weierstrass/non_zero_affine.rs @@ -164,20 +164,20 @@ where } #[cfg(test)] -mod test { +mod test_non_zero_affine { use crate::alloc::AllocVar; use crate::fields::fp::{AllocatedFp, FpVar}; use crate::groups::curves::short_weierstrass::non_zero_affine::NonZeroAffineVar; use crate::groups::curves::short_weierstrass::ProjectiveVar; use crate::groups::CurveVar; use crate::R1CSVar; - use ark_ec::SWModelParameters; + use ark_ec::{ProjectiveCurve, SWModelParameters}; use ark_relations::r1cs::ConstraintSystem; use ark_std::{vec::Vec, One}; use ark_test_curves::bls12_381::{g1::Parameters as G1Parameters, Fq}; #[test] - fn test_non_zero_affine_cost() { + fn correctness_test_1() { let cs = ConstraintSystem::::new_ref(); let x = FpVar::Var( @@ -216,7 +216,7 @@ mod test { sum = sum + elem; } - let sum = sum.value().unwrap(); + let sum = sum.value().unwrap().into_affine(); (sum.x, sum.y) }; @@ -242,4 +242,79 @@ mod test { assert_eq!(sum_a.0, sum_b.0); assert_eq!(sum_a.1, sum_b.1); } + + #[test] + fn correctness_test_2() { + let cs = ConstraintSystem::::new_ref(); + + let x = FpVar::Var( + AllocatedFp::::new_witness(cs.clone(), || { + Ok(G1Parameters::AFFINE_GENERATOR_COEFFS.0) + }) + .unwrap(), + ); + let y = FpVar::Var( + AllocatedFp::::new_witness(cs.clone(), || { + Ok(G1Parameters::AFFINE_GENERATOR_COEFFS.1) + }) + .unwrap(), + ); + + // The following code tests `double_and_add`. + let sum_a = { + let a = ProjectiveVar::>::new( + x.clone(), + y.clone(), + FpVar::Constant(Fq::one()), + ); + + let mut cur = a.clone(); + cur.double_in_place().unwrap(); + for _ in 1..10 { + cur.double_in_place().unwrap(); + cur = cur + &a; + } + + let sum = cur.value().unwrap().into_affine(); + (sum.x, sum.y) + }; + + let sum_b = { + let a = NonZeroAffineVar::>::new(x, y); + + let mut cur = a.double().unwrap(); + for _ in 1..10 { + cur = cur.double_and_add(&a).unwrap(); + } + + (cur.x.value().unwrap(), cur.y.value().unwrap()) + }; + + assert!(cs.is_satisfied().unwrap()); + assert_eq!(sum_a.0, sum_b.0); + assert_eq!(sum_a.1, sum_b.1); + } + + #[test] + fn soundness_test() { + let cs = ConstraintSystem::::new_ref(); + + let x = FpVar::Var( + AllocatedFp::::new_witness(cs.clone(), || { + Ok(G1Parameters::AFFINE_GENERATOR_COEFFS.0) + }) + .unwrap(), + ); + let y = FpVar::Var( + AllocatedFp::::new_witness(cs.clone(), || { + Ok(G1Parameters::AFFINE_GENERATOR_COEFFS.1) + }) + .unwrap(), + ); + + let a = NonZeroAffineVar::>::new(x, y); + + let _ = a.double_and_add(&a).unwrap(); + assert!(!cs.is_satisfied().unwrap()); + } }