take a snapshot of all bundle accounts before sim (jito-foundation#13) (

jito-foundation#615)

bundle/src/bundle_execution.rs

@@ -17,8 +17,9 @@ use {
         transaction::{SanitizedTransaction, TransactionError, VersionedTransaction},
     },
     solana_svm::{
-        account_loader::TransactionLoadResult, account_overrides::AccountOverrides,
-        transaction_processor::ExecutionRecordingConfig,
+        account_loader::TransactionLoadResult,
+        account_overrides::AccountOverrides,
+        transaction_processor::{ExecutionRecordingConfig, TransactionProcessingCallback},
         transaction_results::TransactionExecutionResult,
     },
     solana_transaction_status::{token_balances::TransactionTokenBalances, PreBalanceInfo},
@@ -264,6 +265,18 @@ pub fn load_and_execute_bundle<'a>(
                 account_overrides.upsert_account_overrides(
                     bank.get_account_overrides_for_simulation(&account_keys),
                 );
+
+                // An unfrozen bank's state is always changing.
+                // By taking a snapshot of the accounts we're mocking out grabbing their locks.
+                // **Note** this does not prevent race conditions, just mocks preventing them.
+                if !bank.is_frozen() {
+                    for pk in account_keys.iter() {
+                        // Save on a disk read.
+                        if account_overrides.get(pk).is_none() {
+                            account_overrides.set_account(pk, bank.get_account_shared_data(pk));
+                        }
+                    }
+                }
             });
     }