From e558ad4c717e3daaac4bc2414ee8c91c94fe10d2 Mon Sep 17 00:00:00 2001
From: Trent Nelson <trent@solana.com>
Date: Wed, 24 Nov 2021 14:19:22 -0700
Subject: [PATCH] bank: don't panic when getting logs from malformed
 TransactionLogCollector

---
 runtime/src/bank.rs | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/runtime/src/bank.rs b/runtime/src/bank.rs
index 270fc59aa0..9fd74f36cf 100644
--- a/runtime/src/bank.rs
+++ b/runtime/src/bank.rs
@@ -582,10 +582,12 @@ impl TransactionLogCollector {
     ) -> Option<Vec<TransactionLogInfo>> {
         match address {
             None => Some(self.logs.clone()),
-            Some(address) => self
-                .mentioned_address_map
-                .get(address)
-                .map(|log_indices| log_indices.iter().map(|i| self.logs[*i].clone()).collect()),
+            Some(address) => self.mentioned_address_map.get(address).map(|log_indices| {
+                log_indices
+                    .iter()
+                    .filter_map(|i| self.logs.get(*i).cloned())
+                    .collect()
+            }),
         }
     }
 }
@@ -15397,7 +15399,6 @@ pub(crate) mod tests {
     }
 
     #[test]
-    #[should_panic]
     fn test_transaction_log_collector_get_logs_for_address() {
         let address = Pubkey::new_unique();
         let mut mentioned_address_map = HashMap::new();