Groth16 encrypted input mode introduced

everx-labs · Jul 7, 2022 · cb903e5 · cb903e5
1 parent 8cae7f9
commit cb903e5
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/src/executor/crypto.rs b/src/executor/crypto.rs
@@ -33,9 +33,9 @@ use ed25519::signature::{Signature, Verifier};
 use std::sync::Arc;
 use ton_types::{BuilderData, Cell, error, GasConsumer, ExceptionCode, UInt256};
 
-use crusty3_zk::{groth16::{verify_proof, prepare_verifying_key, Parameters, verify_groth16_proof_from_byteblob},
-                 bls::{Bls12, Fr}
-                };
+use crusty3_zk::{groth16::{verify_proof, prepare_verifying_key, Parameters, verify_groth16_proof_from_byteblob, verify_encrypted_input_groth16_proof_from_byteblob},
+                 bls::{Bls12, Fr},
+};
 
 const PUBLIC_KEY_BITS:  usize = PUBLIC_KEY_BYTES * 8;
 const SIGNATURE_BITS:   usize = SIGNATURE_BYTES * 8;
@@ -109,11 +109,16 @@ pub(super) fn execute_vergrth16(engine: &mut Engine) -> Failure {
 
             let cell_proof = ctx.engine.finalize_cell(builder)?;
 
-            let mut cell_proof_data = obtain_cells_data(cell_proof).unwrap();
-
-            if cell_proof_data_length % 8 == 0 {
-
-                let result = verify_groth16_proof_from_byteblob::<Bls12>(&cell_proof_data[..]).unwrap();
+            let mut cell_proof_data = obtain_cells_data(cell_proof).unwrap();if cell_proof_data_length % 8 == 0 {
+        let mut result = false;
+        if cell_proof_data[0] == 0 {
+            result = verify_groth16_proof_from_byteblob::<Bls12>(&cell_proof_data[1..]).unwrap();
+        } else if cell_proof_data[0] == 1 {
+            result = verify_encrypted_input_groth16_proof_from_byteblob::<Bls12>(&cell_proof_data[1..]).unwrap();
+        }
+        else {
+            return err!(ExceptionCode::InvalidOpcode);
+        }
 
                 ctx.engine.cc.stack.push(boolean!(result));
                 Ok(ctx)