Backports sim_bundle improvements (#407)

Cargo.toml

@@ -147,7 +147,7 @@ backoff = "0.4.0"
 base64 = "0.21.2"
 bincode = "1.3.3"
 bitflags = "1.3.1"
-blake3 = "=1.3.1" # TODO (LB): pinned here
+blake3 = "1.3.3"
 block-buffer = "0.10.4"
 borsh = "0.10.3"
 bs58 = "0.4.0"

bundle/Cargo.toml

@@ -11,11 +11,14 @@ license = { workspace = true }
 edition = { workspace = true }
 
 [dependencies]
+anchor-lang = { workspace = true }
 itertools = { workspace = true }
 log = { workspace = true }
+serde = { workspace = true }
 solana-ledger = { workspace = true }
 solana-logger = { workspace = true }
 solana-measure = { workspace = true }
+solana-poh = { workspace = true }
 solana-program-runtime = { workspace = true }
 solana-runtime = { workspace = true }
 solana-sdk = { workspace = true }

bundle/src/bundle_execution.rs

@@ -1,6 +1,6 @@
 use {
     itertools::izip,
-    log::{debug, trace},
+    log::*,
     solana_ledger::token_balances::collect_token_balances,
     solana_measure::{measure::Measure, measure_us},
     solana_program_runtime::timings::ExecuteTimings,
@@ -77,16 +77,24 @@ impl<'a> LoadAndExecuteBundleOutput<'a> {
 
 #[derive(Clone, Debug, Error)]
 pub enum LoadAndExecuteBundleError {
-    #[error("The bundle processing time was exceeded")]
+    #[error("Bundle execution timed out")]
     ProcessingTimeExceeded(Duration),
 
-    #[error("A transaction in the bundle encountered a lock error")]
+    #[error(
+        "A transaction in the bundle encountered a lock error: [signature={:?}, transaction_error={:?}]",
+        signature,
+        transaction_error
+    )]
     LockError {
         signature: Signature,
         transaction_error: TransactionError,
     },
 
-    #[error("A transaction in the bundle failed to execute")]
+    #[error(
+        "A transaction in the bundle failed to execute: [signature={:?}, execution_result={:?}",
+        signature,
+        execution_result
+    )]
     TransactionError {
         signature: Signature,
         // Box reduces the size between variants in the Error
@@ -318,7 +326,7 @@ pub fn load_and_execute_bundle<'a>(
         });
         saturating_add_assign!(metrics.collect_balances_us, collect_balances_us);
 
-        let end = max(
+        let end = min(
             chunk_start.saturating_add(batch.sanitized_transactions().len()),
             pre_execution_accounts.len(),
         );
@@ -495,7 +503,7 @@ mod tests {
         solana_ledger::genesis_utils::create_genesis_config,
         solana_runtime::{bank::Bank, genesis_utils::GenesisConfigInfo},
         solana_sdk::{
-            bundle::{derive_bundle_id_from_sanizited_transactions, SanitizedBundle},
+            bundle::{derive_bundle_id_from_sanitized_transactions, SanitizedBundle},
             clock::MAX_PROCESSING_AGE,
             pubkey::Pubkey,
             signature::{Keypair, Signer},
@@ -525,7 +533,7 @@ mod tests {
             .map(|tx| SanitizedTransaction::try_from_legacy_transaction(tx.clone()).unwrap())
             .collect();
 
-        let bundle_id = derive_bundle_id_from_sanizited_transactions(&transactions);
+        let bundle_id = derive_bundle_id_from_sanitized_transactions(&transactions);
 
         SanitizedBundle {
             transactions,

bundle/src/lib.rs

@@ -1 +1,60 @@
+use {
+    crate::bundle_execution::LoadAndExecuteBundleError,
+    anchor_lang::error::Error,
+    serde::{Deserialize, Serialize},
+    solana_poh::poh_recorder::PohRecorderError,
+    solana_sdk::pubkey::Pubkey,
+    thiserror::Error,
+};
+
 pub mod bundle_execution;
+
+#[derive(Error, Debug, Clone, Serialize, Deserialize, PartialEq)]
+pub enum TipError {
+    #[error("account is missing from bank: {0}")]
+    AccountMissing(Pubkey),
+
+    #[error("Anchor error: {0}")]
+    AnchorError(String),
+
+    #[error("Lock error")]
+    LockError,
+
+    #[error("Error executing initialize programs")]
+    InitializeProgramsError,
+
+    #[error("Error cranking tip programs")]
+    CrankTipError,
+}
+
+impl From<anchor_lang::error::Error> for TipError {
+    fn from(anchor_err: Error) -> Self {
+        match anchor_err {
+            Error::AnchorError(e) => Self::AnchorError(e.error_msg),
+            Error::ProgramError(e) => Self::AnchorError(e.to_string()),
+        }
+    }
+}
+
+pub type BundleExecutionResult<T> = Result<T, BundleExecutionError>;
+
+#[derive(Error, Debug, Clone)]
+pub enum BundleExecutionError {
+    #[error("The bank has hit the max allotted time for processing transactions")]
+    BankProcessingTimeLimitReached,
+
+    #[error("The bundle exceeds the cost model")]
+    ExceedsCostModel,
+
+    #[error("Runtime error while executing the bundle: {0}")]
+    TransactionFailure(#[from] LoadAndExecuteBundleError),
+
+    #[error("Error locking bundle because a transaction is malformed")]
+    LockError,
+
+    #[error("PoH record error: {0}")]
+    PohRecordError(#[from] PohRecorderError),
+
+    #[error("Tip payment error {0}")]
+    TipError(#[from] TipError),
+}

ci/check-crates.sh

@@ -31,6 +31,9 @@ printf "%s\n" "${files[@]}"
 error_count=0
 for file in "${files[@]}"; do
   read -r crate_name package_publish workspace < <(toml get "$file" . | jq -r '(.package.name | tostring)+" "+(.package.publish | tostring)+" "+(.workspace | tostring)')
+  if [ "$crate_name" == "solana-bundle" ]; then
+    continue
+  fi
   echo "=== $crate_name ($file) ==="
 
   if [[ $package_publish = 'false' ]]; then

core/Cargo.toml

@@ -83,6 +83,7 @@ matches = { workspace = true }
 raptorq = { workspace = true }
 serde_json = { workspace = true }
 serial_test = { workspace = true }
+solana-bundle = { workspace = true }
 solana-logger = { workspace = true }
 solana-program-runtime = { workspace = true }
 solana-program-test = { workspace = true }

core/src/bundle_stage.rs

@@ -43,7 +43,6 @@ mod bundle_packet_receiver;
 mod bundle_reserved_space_manager;
 pub(crate) mod bundle_stage_leader_metrics;
 mod committer;
-pub mod result;
 
 const MAX_BUNDLE_RETRY_DURATION: Duration = Duration::from_millis(10);
 const SLOT_BOUNDARY_CHECK_PERIOD: Duration = Duration::from_millis(10);

core/src/bundle_stage/bundle_consumer.rs

@@ -6,18 +6,20 @@ use {
             bundle_reserved_space_manager::BundleReservedSpaceManager,
             bundle_stage_leader_metrics::BundleStageLeaderMetrics,
             committer::Committer,
-            result::{BundleExecutionError, BundleExecutionResult},
         },
         consensus_cache_updater::ConsensusCacheUpdater,
         immutable_deserialized_bundle::ImmutableDeserializedBundle,
         leader_slot_banking_stage_metrics::ProcessTransactionsSummary,
         leader_slot_banking_stage_timing_metrics::LeaderExecuteAndCommitTimings,
         proxy::block_engine_stage::BlockBuilderFeeInfo,
         qos_service::QosService,
-        tip_manager::{TipManager, TipPaymentError},
+        tip_manager::TipManager,
         unprocessed_transaction_storage::UnprocessedTransactionStorage,
     },
-    solana_bundle::bundle_execution::{load_and_execute_bundle, BundleExecutionMetrics},
+    solana_bundle::{
+        bundle_execution::{load_and_execute_bundle, BundleExecutionMetrics},
+        BundleExecutionError, BundleExecutionResult, TipError,
+    },
     solana_gossip::cluster_info::ClusterInfo,
     solana_measure::{measure, measure_us},
     solana_poh::poh_recorder::{BankStart, RecordTransactionsSummary, TransactionRecorder},
@@ -246,7 +248,9 @@ impl BundleConsumer {
                         .increment_process_packets_transactions_us(measure);
                     r
                 }
-                Err(e) => Err(e.into()),
+                Err(_) => {
+                    Err(BundleExecutionError::LockError)
+                }
             })
             .collect::<Vec<_>>());
 
@@ -283,7 +287,7 @@ impl BundleConsumer {
             &bank_start.bank_creation_time,
             bank_start.working_bank.ns_per_slot,
         ) {
-            return Err(BundleExecutionError::BankProcessingDone);
+            return Err(BundleExecutionError::BankProcessingTimeLimitReached);
         }
 
         if Self::bundle_touches_tip_pdas(
@@ -364,7 +368,7 @@ impl BundleConsumer {
 
             let locked_init_tip_programs_bundle = bundle_account_locker
                 .prepare_locked_bundle(&bundle, &bank_start.working_bank)
-                .map_err(|_| BundleExecutionError::TipError(TipPaymentError::LockError))?;
+                .map_err(|_| BundleExecutionError::TipError(TipError::LockError))?;
 
             Self::update_qos_and_execute_record_commit_bundle(
                 committer,
@@ -386,7 +390,7 @@ impl BundleConsumer {
                     locked_init_tip_programs_bundle.sanitized_bundle().bundle_id,
                     e
                 );
-                BundleExecutionError::TipError(TipPaymentError::InitializeProgramsError)
+                BundleExecutionError::TipError(TipError::InitializeProgramsError)
             })?;
 
             bundle_stage_leader_metrics
@@ -417,7 +421,7 @@ impl BundleConsumer {
 
             let locked_tip_crank_bundle = bundle_account_locker
                 .prepare_locked_bundle(&bundle, &bank_start.working_bank)
-                .map_err(|_| BundleExecutionError::TipError(TipPaymentError::LockError))?;
+                .map_err(|_| BundleExecutionError::TipError(TipError::LockError))?;
 
             Self::update_qos_and_execute_record_commit_bundle(
                 committer,
@@ -439,7 +443,7 @@ impl BundleConsumer {
                     locked_tip_crank_bundle.sanitized_bundle().bundle_id,
                     e
                 );
-                BundleExecutionError::TipError(TipPaymentError::CrankTipError)
+                BundleExecutionError::TipError(TipError::CrankTipError)
             })?;
 
             bundle_stage_leader_metrics
@@ -555,7 +559,7 @@ impl BundleConsumer {
             .accumulate_process_transactions_summary(&ProcessTransactionsSummary {
                 reached_max_poh_height: matches!(
                     result.result,
-                    Err(BundleExecutionError::BankProcessingDone)
+                    Err(BundleExecutionError::BankProcessingTimeLimitReached)
                         | Err(BundleExecutionError::PohRecordError(_))
                 ),
                 transactions_attempted_execution_count: sanitized_bundle.transactions.len(),

core/src/bundle_stage/bundle_packet_receiver.rs

@@ -165,15 +165,11 @@ impl BundleReceiver {
 mod tests {
     use {
         super::*,
-        crate::{
-            bundle_stage::{
-                bundle_account_locker::BundleAccountLockerError, result::BundleExecutionError,
-            },
-            tip_manager::TipPaymentError,
-        },
         crossbeam_channel::unbounded,
         rand::{thread_rng, RngCore},
-        solana_bundle::bundle_execution::LoadAndExecuteBundleError,
+        solana_bundle::{
+            bundle_execution::LoadAndExecuteBundleError, BundleExecutionError, TipError,
+        },
         solana_ledger::genesis_utils::create_genesis_config,
         solana_perf::packet::PacketBatch,
         solana_poh::poh_recorder::PohRecorderError,
@@ -474,7 +470,7 @@ mod tests {
                 let mut results = vec![Ok(()); bundles_to_process.len()];
 
                 (bank_processing_done_index..bundles_to_process.len()).for_each(|index| {
-                    results[index] = Err(BundleExecutionError::BankProcessingDone);
+                    results[index] = Err(BundleExecutionError::BankProcessingTimeLimitReached);
                 });
                 results
             }
@@ -538,7 +534,7 @@ mod tests {
             |bundles_to_process, _stats| {
                 assert_bundles_same(&bundles, bundles_to_process);
                 vec![
-                    Err(BundleExecutionError::ExecutionError(
+                    Err(BundleExecutionError::TransactionFailure(
                         LoadAndExecuteBundleError::ProcessingTimeExceeded(Duration::from_secs(1)),
                     ));
                     bundles_to_process.len()
@@ -591,7 +587,7 @@ mod tests {
             |bundles_to_process, _stats| {
                 assert_bundles_same(&bundles, bundles_to_process);
                 vec![
-                    Err(BundleExecutionError::TipError(TipPaymentError::LockError));
+                    Err(BundleExecutionError::TipError(TipError::LockError));
                     bundles_to_process.len()
                 ]
             }
@@ -640,12 +636,7 @@ mod tests {
             &mut bundle_stage_leader_metrics,
             &HashSet::default(),
             |bundles_to_process, _stats| {
-                vec![
-                    Err(BundleExecutionError::LockError(
-                        BundleAccountLockerError::LockingError
-                    ));
-                    bundles_to_process.len()
-                ]
+                vec![Err(BundleExecutionError::LockError); bundles_to_process.len()]
             }
         ));
         assert_eq!(bundle_storage.unprocessed_bundles_len(), 0);