feat(proofs): Wrap ledger proofs in Arc and share them

Author: tizoc

ledger/src/proofs/block.rs

@@ -1855,7 +1855,7 @@ pub(super) fn generate_block_proof(
     } = params;
 
     let (txn_snark_statement, txn_snark_proof) =
-        ledger_proof_opt(ledger_proof.as_ref(), next_state)?;
+        ledger_proof_opt(ledger_proof.as_deref(), next_state)?;
     let prev_state_proof = &chain.proof;
 
     let (new_state_hash, previous_proof_statements) = block_main(

mina-p2p-messages/src/v2/generated.rs

@@ -271,7 +271,7 @@ pub struct ProverExtendBlockchainInputStableV2 {
     pub chain: BlockchainSnarkBlockchainStableV2,
     pub next_state: MinaStateProtocolStateValueStableV2,
     pub block: MinaStateSnarkTransitionValueStableV2,
-    pub ledger_proof: Option<LedgerProofProdStableV2>,
+    pub ledger_proof: Option<Arc<LedgerProofProdStableV2>>,
     pub prover_state: ConsensusStakeProofStableV2,
     pub pending_coinbase: MinaBasePendingCoinbaseWitnessStableV2,
 }

node/src/block_producer/block_producer_state.rs

@@ -81,7 +81,7 @@ pub enum BlockProducerCurrentState {
         /// `protocol_state.blockchain_state.body_reference`
         diff_hash: v2::ConsensusBodyReferenceStableV1,
         staged_ledger_hash: v2::MinaBaseStagedLedgerHashStableV1,
-        emitted_ledger_proof: Option<Box<v2::LedgerProofProdStableV2>>,
+        emitted_ledger_proof: Option<Arc<v2::LedgerProofProdStableV2>>,
         pending_coinbase_update: v2::MinaBasePendingCoinbaseUpdateStableV1,
         pending_coinbase_witness: v2::MinaBasePendingCoinbaseWitnessStableV2,
         stake_proof_sparse_ledger: v2::MinaBaseSparseLedgerBaseStableV2,
@@ -91,7 +91,7 @@ pub enum BlockProducerCurrentState {
         won_slot: BlockProducerWonSlot,
         /// Chain that we are extending.
         chain: Vec<AppliedBlock>,
-        emitted_ledger_proof: Option<Box<v2::LedgerProofProdStableV2>>,
+        emitted_ledger_proof: Option<Arc<v2::LedgerProofProdStableV2>>,
         pending_coinbase_update: v2::MinaBasePendingCoinbaseUpdateStableV1,
         pending_coinbase_witness: v2::MinaBasePendingCoinbaseWitnessStableV2,
         stake_proof_sparse_ledger: v2::MinaBaseSparseLedgerBaseStableV2,
@@ -103,7 +103,7 @@ pub enum BlockProducerCurrentState {
         won_slot: BlockProducerWonSlot,
         /// Chain that we are extending.
         chain: Vec<AppliedBlock>,
-        emitted_ledger_proof: Option<Box<v2::LedgerProofProdStableV2>>,
+        emitted_ledger_proof: Option<Arc<v2::LedgerProofProdStableV2>>,
         pending_coinbase_update: v2::MinaBasePendingCoinbaseUpdateStableV1,
         pending_coinbase_witness: v2::MinaBasePendingCoinbaseWitnessStableV2,
         stake_proof_sparse_ledger: v2::MinaBaseSparseLedgerBaseStableV2,

node/src/block_producer_effectful/block_producer_effectful_effects.rs

@@ -164,7 +164,7 @@ pub fn block_producer_effects<S: crate::Service>(
                             .clone(),
                         pending_coinbase_update: pending_coinbase_update.clone(),
                     },
-                    ledger_proof: emitted_ledger_proof.as_ref().map(|proof| (**proof).clone()),
+                    ledger_proof: emitted_ledger_proof.clone(),
                     prover_state: ConsensusStakeProofStableV2 {
                         delegator: won_slot.delegator.1.into(),
                         delegator_pk: won_slot.delegator.0.clone(),

node/src/block_producer_effectful/block_producer_effectful_service.rs

@@ -1,3 +1,5 @@
+use std::sync::Arc;
+
 use ledger::proofs::provers::BlockProver;
 use mina_p2p_messages::v2::{
     ConsensusBodyReferenceStableV1, LedgerProofProdStableV2, MinaBasePendingCoinbaseUpdateStableV1,
@@ -13,7 +15,7 @@ pub struct StagedLedgerDiffCreateOutput {
     /// `protocol_state.blockchain_state.body_reference`
     pub diff_hash: ConsensusBodyReferenceStableV1,
     pub staged_ledger_hash: MinaBaseStagedLedgerHashStableV1,
-    pub emitted_ledger_proof: Option<Box<LedgerProofProdStableV2>>,
+    pub emitted_ledger_proof: Option<Arc<LedgerProofProdStableV2>>,
     pub pending_coinbase_update: MinaBasePendingCoinbaseUpdateStableV1,
     pub pending_coinbase_witness: MinaBasePendingCoinbaseWitnessStableV2,
     pub stake_proof_sparse_ledger: MinaBaseSparseLedgerBaseStableV2,

node/src/ledger/ledger_service.rs

@@ -1049,7 +1049,7 @@ impl LedgerCtx {
             emitted_ledger_proof: res
                 .ledger_proof
                 .map(|(proof, ..)| (&proof).into())
-                .map(Box::new),
+                .map(Arc::new),
             pending_coinbase_update: (&res.pending_coinbase_update.1).into(),
             pending_coinbase_witness: MinaBasePendingCoinbaseWitnessStableV2 {
                 pending_coinbases: pending_coinbase_witness,