Check size on insertion

solana-labs · Apr 23, 2021 · b685910 · b685910
1 parent 7becc7d
commit b685910
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ledger/src/blockstore.rs b/ledger/src/blockstore.rs
@@ -1381,6 +1381,7 @@ impl Blockstore {
         shred: &Shred,
         write_batch: &mut WriteBatch,
     ) -> Result<Vec<(u32, u32)>> {
+        use crate::shred::SHRED_PAYLOAD_SIZE;
         let slot = shred.slot();
         let index = u64::from(shred.index());
 
@@ -1418,7 +1419,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(SHRED_PAYLOAD_SIZE, shred.data_header.size.into())],
         )?;
         data_index.set_present(index, true);
         let newly_completed_data_sets = update_slot_meta(