bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

Avoid overflow in VecDeque::with_capacity_in().

Browse source
This commit is contained in:
Hans Kratz 2021-10-18 13:15:45 +02:00
parent 5e02151318
commit 4a37b9cbff
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
library/alloc/src/collections/vec_deque/mod.rs
View file

@ -543,9 +543,9 @@ impl<T, A: Allocator> VecDeque<T, A> {
/// ``` /// ```
#[unstable(feature = "allocator_api", issue = "32838")] #[unstable(feature = "allocator_api", issue = "32838")]
pub fn with_capacity_in(capacity: usize, alloc: A) -> VecDeque<T, A> { pub fn with_capacity_in(capacity: usize, alloc: A) -> VecDeque<T, A> {
assert!(capacity < 1_usize << usize::BITS - 1, "capacity overflow");
// +1 since the ringbuffer always leaves one space empty // +1 since the ringbuffer always leaves one space empty
let cap = cmp::max(capacity + 1, MINIMUM_CAPACITY + 1).next_power_of_two(); let cap = cmp::max(capacity + 1, MINIMUM_CAPACITY + 1).next_power_of_two();
assert!(cap > capacity, "capacity overflow");
VecDeque { tail: 0, head: 0, buf: RawVec::with_capacity_in(cap, alloc) } VecDeque { tail: 0, head: 0, buf: RawVec::with_capacity_in(cap, alloc) }
} }