Add prover check for imput's first element

lambdaclass · Jan 13, 2025 · f16f6e2 · f16f6e2
1 parent 9c717ef
commit f16f6e2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/examples/baby-snark/src/prover.rs b/examples/baby-snark/src/prover.rs
@@ -10,6 +10,7 @@ pub struct Proof {
 #[derive(Debug)]
 pub enum Error {
     WrongWitness,
+    FirstInputElementIsNotOne,
 }
 
 pub struct Prover;
@@ -19,6 +20,9 @@ impl Prover {
         ssp: &SquareSpanProgram,
         pk: &ProvingKey,
     ) -> Result<Proof, Error> {
+        if inputs[0].ne(&FrElement::one()) {
+            return Err(Error::FirstInputElementIsNotOne);
+        }
         if !ssp.check_valid(inputs) {
             return Err(Error::WrongWitness);
         }