Make sure added capacity doesn't overflow usize::MAX

rust-ndarray · May 7, 2023 · 3913586 · 3913586
1 parent d695430
commit 3913586
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/impl_owned_array.rs b/src/impl_owned_array.rs
@@ -731,6 +731,9 @@ impl<A, D> Array<A, D>
         debug_assert!(axis.index() < self.ndim());
         let self_dim = self.raw_dim();
         let remaining_shape = self_dim.remove_axis(axis);
+
+        // Make sure added capacity doesn't overflow
+        debug_assert!(remaining_shape.size().checked_mul(additional).is_some());
         let len_to_append = remaining_shape.size() * additional;
 
         // Make sure new capacity is still in bounds