stake-pool: Add checked math to big vec calc (solana-labs#2524)

Nikolay-everstake · Oct 18, 2021 · 7f0278c · 7f0278c
1 parent f97c61a
commit 7f0278c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/stake-pool/program/src/big_vec.rs b/stake-pool/program/src/big_vec.rs
@@ -90,7 +90,10 @@ impl<'data> BigVec<'data> {
         len: usize,
     ) -> Result<Vec<&'data mut T>, ProgramError> {
         let vec_len = self.len();
-        if skip + len > vec_len as usize {
+        let last_item_index = skip
+            .checked_add(len)
+            .ok_or(ProgramError::AccountDataTooSmall)?;
+        if last_item_index > vec_len as usize {
             return Err(ProgramError::AccountDataTooSmall);
         }