fix: 🩹 Use checked_add instead of saturating_add

Moonsong-Labs · Dec 26, 2024 · 7364a27 · 7364a27
1 parent aa89328
commit 7364a27
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pallets/file-system/src/utils.rs b/pallets/file-system/src/utils.rs
@@ -1274,10 +1274,11 @@ where
             match <T::PaymentStreams as PaymentStreamsInterface>::get_dynamic_rate_payment_stream_amount_provided(&bsp_id, &storage_request_metadata.owner) {
 				Some(previous_amount_provided) => {
 					// Update the payment stream.
+                    let new_amount_provided = &previous_amount_provided.checked_add(storage_request_metadata.size).ok_or(ArithmeticError::Overflow)?;
 					<T::PaymentStreams as PaymentStreamsInterface>::update_dynamic_rate_payment_stream(
 						&bsp_id,
 						&storage_request_metadata.owner,
-						&(previous_amount_provided.saturating_add(storage_request_metadata.size)),
+						new_amount_provided,
 					)?;
 				},
 				None => {