Check size on insertion

solana-labs · Apr 22, 2021 · 66c9c61 · 66c9c61
1 parent f7e0988
commit 66c9c61
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ledger/src/blockstore.rs b/ledger/src/blockstore.rs
@@ -1385,6 +1385,7 @@ impl Blockstore {
         shred: &Shred,
         write_batch: &mut WriteBatch,
     ) -> Result<Vec<(u32, u32)>> {
+        use crate::shred::SIZE_OF_DATA_SHRED_PAYLOAD;
         let slot = shred.slot();
         let index = u64::from(shred.index());
 
@@ -1422,7 +1423,7 @@ impl Blockstore {
             (slot, index),
             // Payload will be padded out to SHRED_PAYLOAD_SIZE
             // But only need to store the bytes within data_header.size
-            &shred.payload[..shred.data_header.size as usize],
+            &shred.payload[..cmp::min(SIZE_OF_DATA_SHRED_PAYLOAD, shred.data_header.size.into())],
         )?;
         data_index.set_present(index, true);
         let newly_completed_data_sets = update_slot_meta(