feat: add proof compression (Lightprotocol#435)

* feat(prover.js): add proof compression

* feat: proof compression

Author: ananas-block

Cargo.lock

@@ -23,7 +23,8 @@ light-merkle-tree-program = { version = "0.3.1", path = "../merkle-tree", featur
 solana-security-txt = "1.1.0"
 
 # Light Deps
-groth16-solana = "0.0.2"
+groth16-solana = { git= "https://github.com/Lightprotocol/groth16-solana", branch="master"}
+
 light-macros = { version = "0.3.1", path = "../../macros/light" }
 light-verifier-sdk = { version = "0.3.1", path = "../../verifier-sdk" }
 

pnpm-lock.yaml

@@ -33,7 +33,7 @@ impl Config for TransactionConfig {
 pub mod light_psp10in2out {
     use std::marker::PhantomData;
 
-    use light_verifier_sdk::light_transaction::{Amounts, Proof};
+    use light_verifier_sdk::light_transaction::{Amounts, ProofCompressed};
 
     use super::*;
 
@@ -89,7 +89,7 @@ pub mod light_psp10in2out {
             InstructionDataCompressedTransferSecond::try_deserialize_unchecked(
                 &mut [vec![0u8; 8], inputs].concat().as_slice(),
             )?;
-        let proof = Proof {
+        let proof = ProofCompressed {
             a: inputs.proof_a,
             b: inputs.proof_b,
             c: inputs.proof_c,
@@ -218,9 +218,9 @@ pub struct LightInstructionSecond<
 #[derive(Debug)]
 #[account]
 pub struct InstructionDataCompressedTransferSecond {
-    proof_a: [u8; 64],
-    proof_b: [u8; 128],
-    proof_c: [u8; 64],
+    proof_a: [u8; 32],
+    proof_b: [u8; 64],
+    proof_c: [u8; 32],
 }
 
 #[derive(Accounts)]

programs/psp10in2out/Cargo.toml

@@ -22,7 +22,8 @@ anchor-spl = "0.28.0"
 
 # Light deps
 aligned-sized = { version = "0.1.0", path = "../../macros/aligned-sized" }
-groth16-solana = "0.0.2"
+groth16-solana = { git= "https://github.com/Lightprotocol/groth16-solana", branch="master"}
+
 light-macros = { version = "0.3.1", path = "../../macros/light" }
 light-merkle-tree-program = { version = "0.3.1", path = "../merkle-tree", features = ["cpi"] }
 light-verifier-sdk = { version = "0.3.1", path = "../../verifier-sdk" }

programs/psp10in2out/src/lib.rs

@@ -2,7 +2,7 @@ use aligned_sized::aligned_sized;
 use anchor_lang::prelude::*;
 use light_macros::light_verifier_accounts;
 use light_verifier_sdk::light_transaction::{
-    Amounts, Message, Proof, Transaction, TransactionInput, VERIFIER_STATE_SEED,
+    Amounts, Message, Transaction, TransactionInput, VERIFIER_STATE_SEED,
 };
 
 pub mod verifying_key;
@@ -36,6 +36,8 @@ pub enum VerifierError {
 
 #[program]
 pub mod light_psp2in2out_storage {
+    use light_verifier_sdk::light_transaction::ProofCompressed;
+
     use super::*;
 
     /// Saves the provided message in a temporary PDA.
@@ -87,7 +89,7 @@ pub mod light_psp2in2out_storage {
                 &mut [vec![0u8; 8], inputs, vec![0u8; 16]].concat().as_slice(),
             )?;
         let message = Message::new(&ctx.accounts.verifier_state.msg);
-        let proof = Proof {
+        let proof = ProofCompressed {
             a: inputs.proof_a,
             b: inputs.proof_b,
             c: inputs.proof_c,
@@ -168,9 +170,9 @@ pub struct LightInstructionSecond<'info> {
 #[derive(Debug)]
 #[account]
 pub struct InstructionDataCompressedTransferSecond {
-    proof_a: [u8; 64],
-    proof_b: [u8; 128],
-    proof_c: [u8; 64],
+    proof_a: [u8; 32],
+    proof_b: [u8; 64],
+    proof_c: [u8; 32],
     input_nullifier: [[u8; 32]; 2],
     output_commitment: [[u8; 32]; 2],
     public_amount_sol: [u8; 32],

programs/psp2in2out-storage/Cargo.toml

@@ -23,7 +23,7 @@ light-merkle-tree-program = { version = "0.3.1", path = "../merkle-tree", featur
 solana-security-txt = "1.1.0"
 
 # Light Deps
-groth16-solana = "0.0.2"
+groth16-solana = { git= "https://github.com/Lightprotocol/groth16-solana", branch="master"}
 light-macros = { version = "0.3.1", path = "../../macros/light" }
 light-verifier-sdk = { version = "0.3.1", path = "../../verifier-sdk" }
 

programs/psp2in2out-storage/src/lib.rs

@@ -1,6 +1,8 @@
 use anchor_lang::prelude::*;
 use light_macros::light_verifier_accounts;
-use light_verifier_sdk::light_transaction::{Amounts, Proof, Transaction, TransactionInput};
+use light_verifier_sdk::light_transaction::{
+    Amounts, ProofCompressed, Transaction, TransactionInput,
+};
 
 pub mod verifying_key;
 use verifying_key::VERIFYINGKEY_TRANSACTION_MASP2_MAIN;
@@ -21,6 +23,7 @@ pub const PROGRAM_ID: &str = "J1RRetZ4ujphU75LP8RadjXMf3sA12yC2R44CF7PmU7i";
 
 #[program]
 pub mod light_psp2in2out {
+
     use super::*;
 
     /// This instruction is the first step of a compressed transaction.
@@ -39,7 +42,7 @@ pub mod light_psp2in2out {
         let len_missing_bytes = 256 - inputs.encrypted_utxos.len();
         let mut enc_utxos = inputs.encrypted_utxos;
         enc_utxos.append(&mut vec![0u8; len_missing_bytes]);
-        let proof = Proof {
+        let proof = ProofCompressed {
             a: inputs.proof_a,
             b: inputs.proof_b,
             c: inputs.proof_c,
@@ -76,9 +79,9 @@ pub struct LightInstruction<'info> {}
 #[derive(Debug)]
 #[account]
 pub struct InstructionDataCompressedTransferFirst {
-    proof_a: [u8; 64],
-    proof_b: [u8; 128],
-    proof_c: [u8; 64],
+    proof_a: [u8; 32],
+    proof_b: [u8; 64],
+    proof_c: [u8; 32],
     public_amount_spl: [u8; 32],
     input_nullifier: [[u8; 32]; 2],
     output_commitment: [[u8; 32]; 2],

programs/psp2in2out/Cargo.toml

@@ -23,7 +23,8 @@ light-merkle-tree-program = { version = "0.3.1", path = "../merkle-tree", featur
 solana-security-txt = "1.1.0"
 
 # Light Deps
-groth16-solana = "0.0.2"
+groth16-solana = { git= "https://github.com/Lightprotocol/groth16-solana", branch="master"}
+
 light-macros = { version = "0.3.1", path = "../../macros/light" }
 light-verifier-sdk = { version = "0.3.1", path = "../../verifier-sdk" }
 bytemuck = "1.14.0"

programs/psp2in2out/src/lib.rs

@@ -2,7 +2,7 @@ use anchor_lang::{prelude::*, solana_program::hash::hash};
 use bytemuck::{Pod, Zeroable};
 use light_macros::{light_verifier_accounts, pubkey};
 use light_verifier_sdk::light_transaction::{
-    Amounts, Config, Proof, Transaction, TransactionInput,
+    Amounts, Config, ProofCompressed, Transaction, TransactionInput,
 };
 
 pub mod verifying_key;
@@ -42,23 +42,23 @@ pub mod light_psp4in4out_app_storage {
     /// This instruction is used to invoke this system verifier and can only be invoked via cpi.
     pub fn compressed_transfer_inputs<'info>(
         ctx: Context<'_, '_, '_, 'info, LightInstruction<'info>>,
-        proof_a: [u8; 64],
-        proof_b: [u8; 128],
-        proof_c: [u8; 64],
+        proof_a: [u8; 32],
+        proof_b: [u8; 64],
+        proof_c: [u8; 32],
         connecting_hash: [u8; 32],
         start_offset: usize,
     ) -> Result<()> {
-        let proof = Proof {
+        let proof = ProofCompressed {
             a: proof_a,
             b: proof_b,
             c: proof_c,
         };
         // + 8 to account for the discriminator
         let end_offset =
             start_offset + 8 + std::mem::size_of::<Psp4In4OutAppStorageVerifierState>();
-
         let verifier_state = Psp4In4OutAppStorageVerifierState::try_deserialize_unchecked(
-            &mut &ctx.accounts.verifier_state.to_account_info().data.borrow()[32..end_offset],
+            &mut &ctx.accounts.verifier_state.to_account_info().data.borrow()
+                [start_offset..end_offset],
         )?;
 
         let public_amount = Amounts {