Coverage Report

Created: 2026-08-05 07:20

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/rust/registry/src/index.crates.io-1949cf8c6b5b557f/smallvec-1.15.2/src/lib.rs
Line
Count
Source
1
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
2
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
3
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
4
// option. This file may not be copied, modified, or distributed
5
// except according to those terms.
6
7
//! Small vectors in various sizes. These store a certain number of elements inline, and fall back
8
//! to the heap for larger allocations.  This can be a useful optimization for improving cache
9
//! locality and reducing allocator traffic for workloads that fit within the inline buffer.
10
//!
11
//! ## `no_std` support
12
//!
13
//! By default, `smallvec` does not depend on `std`.  However, the optional
14
//! `write` feature implements the `std::io::Write` trait for vectors of `u8`.
15
//! When this feature is enabled, `smallvec` depends on `std`.
16
//!
17
//! ## Optional features
18
//!
19
//! ### `serde`
20
//!
21
//! When this optional dependency is enabled, `SmallVec` implements the `serde::Serialize` and
22
//! `serde::Deserialize` traits.
23
//!
24
//! ### `write`
25
//!
26
//! When this feature is enabled, `SmallVec<[u8; _]>` implements the `std::io::Write` trait.
27
//! This feature is not compatible with `#![no_std]` programs.
28
//!
29
//! ### `union`
30
//!
31
//! **This feature requires Rust 1.49.**
32
//!
33
//! When the `union` feature is enabled `smallvec` will track its state (inline or spilled)
34
//! without the use of an enum tag, reducing the size of the `smallvec` by one machine word.
35
//! This means that there is potentially no space overhead compared to `Vec`.
36
//! Note that `smallvec` can still be larger than `Vec` if the inline buffer is larger than two
37
//! machine words.
38
//!
39
//! To use this feature add `features = ["union"]` in the `smallvec` section of Cargo.toml.
40
//! Note that this feature requires Rust 1.49.
41
//!
42
//! Tracking issue: [rust-lang/rust#55149](https://github.com/rust-lang/rust/issues/55149)
43
//!
44
//! ### `const_generics`
45
//!
46
//! **This feature requires Rust 1.51.**
47
//!
48
//! When this feature is enabled, `SmallVec` works with any arrays of any size, not just a fixed
49
//! list of sizes.
50
//!
51
//! ### `const_new`
52
//!
53
//! **This feature requires Rust 1.51.**
54
//!
55
//! This feature exposes the functions [`SmallVec::new_const`], [`SmallVec::from_const`], and [`smallvec_inline`] which enables the `SmallVec` to be initialized from a const context.
56
//! For details, see the
57
//! [Rust Reference](https://doc.rust-lang.org/reference/const_eval.html#const-functions).
58
//!
59
//! ### `drain_filter`
60
//!
61
//! **This feature is unstable.** It may change to match the unstable `drain_filter` method in libstd.
62
//!
63
//! Enables the `drain_filter` method, which produces an iterator that calls a user-provided
64
//! closure to determine which elements of the vector to remove and yield from the iterator.
65
//!
66
//! ### `drain_keep_rest`
67
//!
68
//! **This feature is unstable.** It may change to match the unstable `drain_keep_rest` method in libstd.
69
//!
70
//! Enables the `DrainFilter::keep_rest` method.
71
//!
72
//! ### `specialization`
73
//!
74
//! **This feature is unstable and requires a nightly build of the Rust toolchain.**
75
//!
76
//! When this feature is enabled, `SmallVec::from(slice)` has improved performance for slices
77
//! of `Copy` types.  (Without this feature, you can use `SmallVec::from_slice` to get optimal
78
//! performance for `Copy` types.)
79
//!
80
//! Tracking issue: [rust-lang/rust#31844](https://github.com/rust-lang/rust/issues/31844)
81
//!
82
//! ### `may_dangle`
83
//!
84
//! **This feature is unstable and requires a nightly build of the Rust toolchain.**
85
//!
86
//! This feature makes the Rust compiler less strict about use of vectors that contain borrowed
87
//! references. For details, see the
88
//! [Rustonomicon](https://doc.rust-lang.org/1.42.0/nomicon/dropck.html#an-escape-hatch).
89
//!
90
//! Tracking issue: [rust-lang/rust#34761](https://github.com/rust-lang/rust/issues/34761)
91
92
#![no_std]
93
#![cfg_attr(docsrs, feature(doc_cfg))]
94
#![cfg_attr(feature = "specialization", allow(incomplete_features))]
95
#![cfg_attr(feature = "specialization", feature(specialization))]
96
#![cfg_attr(feature = "may_dangle", feature(dropck_eyepatch))]
97
#![cfg_attr(
98
    feature = "debugger_visualizer",
99
    feature(debugger_visualizer),
100
    debugger_visualizer(natvis_file = "../debug_metadata/smallvec.natvis")
101
)]
102
#![deny(missing_docs)]
103
104
#[doc(hidden)]
105
pub extern crate alloc;
106
107
#[cfg(any(test, feature = "write"))]
108
extern crate std;
109
110
#[cfg(test)]
111
mod tests;
112
113
#[allow(deprecated)]
114
use alloc::alloc::{Layout, LayoutErr};
115
use alloc::boxed::Box;
116
use alloc::{vec, vec::Vec};
117
use core::borrow::{Borrow, BorrowMut};
118
use core::cmp;
119
use core::fmt;
120
use core::hash::{Hash, Hasher};
121
use core::hint::unreachable_unchecked;
122
use core::iter::{repeat, FromIterator, FusedIterator, IntoIterator};
123
use core::mem;
124
use core::mem::MaybeUninit;
125
use core::ops::{self, Range, RangeBounds};
126
use core::ptr::{self, NonNull};
127
use core::slice::{self, SliceIndex};
128
129
#[cfg(feature = "malloc_size_of")]
130
use malloc_size_of::{MallocShallowSizeOf, MallocSizeOf, MallocSizeOfOps};
131
132
#[cfg(feature = "serde")]
133
use serde::{
134
    de::{Deserialize, Deserializer, SeqAccess, Visitor},
135
    ser::{Serialize, SerializeSeq, Serializer},
136
};
137
138
#[cfg(feature = "serde")]
139
use core::marker::PhantomData;
140
141
#[cfg(feature = "write")]
142
use std::io;
143
144
#[cfg(feature = "drain_keep_rest")]
145
use core::mem::ManuallyDrop;
146
147
/// Creates a [`SmallVec`] containing the arguments.
148
///
149
/// `smallvec!` allows `SmallVec`s to be defined with the same syntax as array expressions.
150
/// There are two forms of this macro:
151
///
152
/// - Create a [`SmallVec`] containing a given list of elements:
153
///
154
/// ```
155
/// # use smallvec::{smallvec, SmallVec};
156
/// # fn main() {
157
/// let v: SmallVec<[_; 128]> = smallvec![1, 2, 3];
158
/// assert_eq!(v[0], 1);
159
/// assert_eq!(v[1], 2);
160
/// assert_eq!(v[2], 3);
161
/// # }
162
/// ```
163
///
164
/// - Create a [`SmallVec`] from a given element and size:
165
///
166
/// ```
167
/// # use smallvec::{smallvec, SmallVec};
168
/// # fn main() {
169
/// let v: SmallVec<[_; 10]> = smallvec![1; 3];
170
/// assert_eq!(v, SmallVec::from_buf([1, 1, 1]));
171
/// # }
172
/// ```
173
///
174
/// Note that unlike array expressions this syntax supports all elements
175
/// which implement [`Clone`] and the number of elements doesn't have to be
176
/// a constant.
177
///
178
/// This will use `clone` to duplicate an expression, so one should be careful
179
/// using this with types having a nonstandard `Clone` implementation. For
180
/// example, `smallvec![Rc::new(1); 5]` will create a vector of five references
181
/// to the same boxed integer value, not five references pointing to independently
182
/// boxed integers.
183
#[macro_export]
184
macro_rules! smallvec {
185
    // count helper: transform any expression into 1
186
    (@one $x:expr) => (1usize);
187
    () => (
188
        $crate::SmallVec::new()
189
    );
190
    ($elem:expr; $n:expr) => ({
191
        $crate::SmallVec::from_elem($elem, $n)
192
    });
193
    ($($x:expr),+$(,)?) => ({
194
        let count = 0usize $(+ $crate::smallvec!(@one $x))+;
195
        let mut vec = $crate::SmallVec::new();
196
        if count <= vec.inline_size() {
197
            $(vec.push($x);)*
198
            vec
199
        } else {
200
            $crate::SmallVec::from_vec($crate::alloc::vec![$($x,)+])
201
        }
202
    });
203
}
204
205
/// Creates an inline [`SmallVec`] containing the arguments. This macro is enabled by the feature `const_new`.
206
///
207
/// `smallvec_inline!` allows `SmallVec`s to be defined with the same syntax as array expressions in `const` contexts.
208
/// The inline storage `A` will always be an array of the size specified by the arguments.
209
/// There are two forms of this macro:
210
///
211
/// - Create a [`SmallVec`] containing a given list of elements:
212
///
213
/// ```
214
/// # use smallvec::{smallvec_inline, SmallVec};
215
/// # fn main() {
216
/// const V: SmallVec<[i32; 3]> = smallvec_inline![1, 2, 3];
217
/// assert_eq!(V[0], 1);
218
/// assert_eq!(V[1], 2);
219
/// assert_eq!(V[2], 3);
220
/// # }
221
/// ```
222
///
223
/// - Create a [`SmallVec`] from a given element and size:
224
///
225
/// ```
226
/// # use smallvec::{smallvec_inline, SmallVec};
227
/// # fn main() {
228
/// const V: SmallVec<[i32; 3]> = smallvec_inline![1; 3];
229
/// assert_eq!(V, SmallVec::from_buf([1, 1, 1]));
230
/// # }
231
/// ```
232
///
233
/// Note that the behavior mimics that of array expressions, in contrast to [`smallvec`].
234
#[cfg(feature = "const_new")]
235
#[cfg_attr(docsrs, doc(cfg(feature = "const_new")))]
236
#[macro_export]
237
macro_rules! smallvec_inline {
238
    // count helper: transform any expression into 1
239
    (@one $x:expr) => (1usize);
240
    ($elem:expr; $n:expr) => ({
241
        $crate::SmallVec::<[_; $n]>::from_const([$elem; $n])
242
    });
243
    ($($x:expr),+ $(,)?) => ({
244
        const N: usize = 0usize $(+ $crate::smallvec_inline!(@one $x))*;
245
        $crate::SmallVec::<[_; N]>::from_const([$($x,)*])
246
    });
247
}
248
249
/// `panic!()` in debug builds, optimization hint in release.
250
#[cfg(not(feature = "union"))]
251
macro_rules! debug_unreachable {
252
    () => {
253
        debug_unreachable!("entered unreachable code")
254
    };
255
    ($e:expr) => {
256
        if cfg!(debug_assertions) {
257
            panic!($e);
258
        } else {
259
            unreachable_unchecked();
260
        }
261
    };
262
}
263
264
/// Trait to be implemented by a collection that can be extended from a slice
265
///
266
/// ## Example
267
///
268
/// ```rust
269
/// use smallvec::{ExtendFromSlice, SmallVec};
270
///
271
/// fn initialize<V: ExtendFromSlice<u8>>(v: &mut V) {
272
///     v.extend_from_slice(b"Test!");
273
/// }
274
///
275
/// let mut vec = Vec::new();
276
/// initialize(&mut vec);
277
/// assert_eq!(&vec, b"Test!");
278
///
279
/// let mut small_vec = SmallVec::<[u8; 8]>::new();
280
/// initialize(&mut small_vec);
281
/// assert_eq!(&small_vec as &[_], b"Test!");
282
/// ```
283
#[doc(hidden)]
284
#[deprecated]
285
pub trait ExtendFromSlice<T> {
286
    /// Extends a collection from a slice of its element type
287
    fn extend_from_slice(&mut self, other: &[T]);
288
}
289
290
#[allow(deprecated)]
291
impl<T: Clone> ExtendFromSlice<T> for Vec<T> {
292
0
    fn extend_from_slice(&mut self, other: &[T]) {
293
0
        Vec::extend_from_slice(self, other)
294
0
    }
295
}
296
297
/// Error type for APIs with fallible heap allocation
298
#[derive(Debug)]
299
pub enum CollectionAllocErr {
300
    /// Overflow `usize::MAX` or other error during size computation
301
    CapacityOverflow,
302
    /// The allocator return an error
303
    AllocErr {
304
        /// The layout that was passed to the allocator
305
        layout: Layout,
306
    },
307
}
308
309
impl fmt::Display for CollectionAllocErr {
310
0
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
311
0
        write!(f, "Allocation error: {:?}", self)
312
0
    }
313
}
314
315
#[allow(deprecated)]
316
impl From<LayoutErr> for CollectionAllocErr {
317
0
    fn from(_: LayoutErr) -> Self {
318
0
        CollectionAllocErr::CapacityOverflow
319
0
    }
320
}
321
322
2.94k
fn infallible<T>(result: Result<T, CollectionAllocErr>) -> T {
323
0
    match result {
324
2.94k
        Ok(x) => x,
325
0
        Err(CollectionAllocErr::CapacityOverflow) => panic!("capacity overflow"),
326
0
        Err(CollectionAllocErr::AllocErr { layout }) => alloc::alloc::handle_alloc_error(layout),
327
    }
328
2.94k
}
Unexecuted instantiation: smallvec::infallible::<()>
Unexecuted instantiation: smallvec::infallible::<_>
smallvec::infallible::<()>
Line
Count
Source
322
2.94k
fn infallible<T>(result: Result<T, CollectionAllocErr>) -> T {
323
0
    match result {
324
2.94k
        Ok(x) => x,
325
0
        Err(CollectionAllocErr::CapacityOverflow) => panic!("capacity overflow"),
326
0
        Err(CollectionAllocErr::AllocErr { layout }) => alloc::alloc::handle_alloc_error(layout),
327
    }
328
2.94k
}
329
330
/// FIXME: use `Layout::array` when we require a Rust version where it’s stable
331
/// <https://github.com/rust-lang/rust/issues/55724>
332
0
fn layout_array<T>(n: usize) -> Result<Layout, CollectionAllocErr> {
333
0
    let size = mem::size_of::<T>()
334
0
        .checked_mul(n)
335
0
        .ok_or(CollectionAllocErr::CapacityOverflow)?;
336
0
    let align = mem::align_of::<T>();
337
0
    Layout::from_size_align(size, align).map_err(|_| CollectionAllocErr::CapacityOverflow)
338
0
}
Unexecuted instantiation: smallvec::layout_array::<core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>>
Unexecuted instantiation: smallvec::layout_array::<vm_virtio::checked_descriptor::CheckedDescriptor>
Unexecuted instantiation: smallvec::layout_array::<(vm_memory::guest_memory::GuestAddress, usize)>
Unexecuted instantiation: smallvec::layout_array::<(vm_memory::guest_memory::GuestAddress, u32)>
Unexecuted instantiation: smallvec::layout_array::<_>
Unexecuted instantiation: smallvec::layout_array::<(vm_memory::guest_memory::GuestAddress, usize)>
339
340
0
unsafe fn deallocate<T>(ptr: NonNull<T>, capacity: usize) {
341
    // This unwrap should succeed since the same did when allocating.
342
0
    let layout = layout_array::<T>(capacity).unwrap();
343
0
    alloc::alloc::dealloc(ptr.as_ptr() as *mut u8, layout)
344
0
}
Unexecuted instantiation: smallvec::deallocate::<core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>>
Unexecuted instantiation: smallvec::deallocate::<vm_virtio::checked_descriptor::CheckedDescriptor>
Unexecuted instantiation: smallvec::deallocate::<(vm_memory::guest_memory::GuestAddress, usize)>
Unexecuted instantiation: smallvec::deallocate::<(vm_memory::guest_memory::GuestAddress, u32)>
Unexecuted instantiation: smallvec::deallocate::<_>
Unexecuted instantiation: smallvec::deallocate::<(vm_memory::guest_memory::GuestAddress, usize)>
345
346
/// An iterator that removes the items from a `SmallVec` and yields them by value.
347
///
348
/// Returned from [`SmallVec::drain`][1].
349
///
350
/// [1]: struct.SmallVec.html#method.drain
351
pub struct Drain<'a, T: 'a + Array> {
352
    tail_start: usize,
353
    tail_len: usize,
354
    iter: slice::Iter<'a, T::Item>,
355
    vec: NonNull<SmallVec<T>>,
356
}
357
358
impl<'a, T: 'a + Array> fmt::Debug for Drain<'a, T>
359
where
360
    T::Item: fmt::Debug,
361
{
362
0
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
363
0
        f.debug_tuple("Drain").field(&self.iter.as_slice()).finish()
364
0
    }
365
}
366
367
unsafe impl<'a, T: Sync + Array> Sync for Drain<'a, T> {}
368
unsafe impl<'a, T: Send + Array> Send for Drain<'a, T> {}
369
370
impl<'a, T: 'a + Array> Iterator for Drain<'a, T> {
371
    type Item = T::Item;
372
373
    #[inline]
374
0
    fn next(&mut self) -> Option<T::Item> {
375
0
        self.iter
376
0
            .next()
377
0
            .map(|reference| unsafe { ptr::read(reference) })
378
0
    }
379
380
    #[inline]
381
0
    fn size_hint(&self) -> (usize, Option<usize>) {
382
0
        self.iter.size_hint()
383
0
    }
384
}
385
386
impl<'a, T: 'a + Array> DoubleEndedIterator for Drain<'a, T> {
387
    #[inline]
388
0
    fn next_back(&mut self) -> Option<T::Item> {
389
0
        self.iter
390
0
            .next_back()
391
0
            .map(|reference| unsafe { ptr::read(reference) })
392
0
    }
393
}
394
395
impl<'a, T: Array> ExactSizeIterator for Drain<'a, T> {
396
    #[inline]
397
0
    fn len(&self) -> usize {
398
0
        self.iter.len()
399
0
    }
400
}
401
402
impl<'a, T: Array> FusedIterator for Drain<'a, T> {}
403
404
impl<'a, T: 'a + Array> Drop for Drain<'a, T> {
405
0
    fn drop(&mut self) {
406
0
        self.for_each(drop);
407
408
0
        if self.tail_len > 0 {
409
            unsafe {
410
0
                let source_vec = self.vec.as_mut();
411
412
                // memmove back untouched tail, update to new length
413
0
                let start = source_vec.len();
414
0
                let tail = self.tail_start;
415
0
                if tail != start {
416
0
                    // as_mut_ptr creates a &mut, invalidating other pointers.
417
0
                    // This pattern avoids calling it with a pointer already present.
418
0
                    let ptr = source_vec.as_mut_ptr();
419
0
                    let src = ptr.add(tail);
420
0
                    let dst = ptr.add(start);
421
0
                    ptr::copy(src, dst, self.tail_len);
422
0
                }
423
0
                source_vec.set_len(start + self.tail_len);
424
            }
425
0
        }
426
0
    }
427
}
428
429
#[cfg(feature = "drain_filter")]
430
/// An iterator which uses a closure to determine if an element should be removed.
431
///
432
/// Returned from [`SmallVec::drain_filter`][1].
433
///
434
/// [1]: struct.SmallVec.html#method.drain_filter
435
pub struct DrainFilter<'a, T, F>
436
where
437
    F: FnMut(&mut T::Item) -> bool,
438
    T: Array,
439
{
440
    vec: &'a mut SmallVec<T>,
441
    /// The index of the item that will be inspected by the next call to `next`.
442
    idx: usize,
443
    /// The number of items that have been drained (removed) thus far.
444
    del: usize,
445
    /// The original length of `vec` prior to draining.
446
    old_len: usize,
447
    /// The filter test predicate.
448
    pred: F,
449
    /// A flag that indicates a panic has occurred in the filter test predicate.
450
    /// This is used as a hint in the drop implementation to prevent consumption
451
    /// of the remainder of the `DrainFilter`. Any unprocessed items will be
452
    /// backshifted in the `vec`, but no further items will be dropped or
453
    /// tested by the filter predicate.
454
    panic_flag: bool,
455
}
456
457
#[cfg(feature = "drain_filter")]
458
impl <T, F> fmt::Debug for DrainFilter<'_, T, F>
459
where
460
    F: FnMut(&mut T::Item) -> bool,
461
    T: Array,
462
    T::Item: fmt::Debug,
463
{
464
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
465
        f.debug_tuple("DrainFilter").field(&self.vec.as_slice()).finish()
466
    }
467
}
468
469
#[cfg(feature = "drain_filter")]
470
impl <T, F> Iterator for DrainFilter<'_, T, F>
471
where
472
    F: FnMut(&mut T::Item) -> bool,
473
    T: Array,
474
{
475
    type Item = T::Item;
476
477
    fn next(&mut self) -> Option<T::Item>
478
    {
479
        unsafe {
480
            while self.idx < self.old_len {
481
                let i = self.idx;
482
                let v = slice::from_raw_parts_mut(self.vec.as_mut_ptr(), self.old_len);
483
                self.panic_flag = true;
484
                let drained = (self.pred)(&mut v[i]);
485
                self.panic_flag = false;
486
                // Update the index *after* the predicate is called. If the index
487
                // is updated prior and the predicate panics, the element at this
488
                // index would be leaked.
489
                self.idx += 1;
490
                if drained {
491
                    self.del += 1;
492
                    return Some(ptr::read(&v[i]));
493
                } else if self.del > 0 {
494
                    let del = self.del;
495
                    let src: *const Self::Item = &v[i];
496
                    let dst: *mut Self::Item = &mut v[i - del];
497
                    ptr::copy_nonoverlapping(src, dst, 1);
498
                }
499
            }
500
            None
501
        }
502
    }
503
504
    fn size_hint(&self) -> (usize, Option<usize>) {
505
        (0, Some(self.old_len - self.idx))
506
    }
507
}
508
509
#[cfg(feature = "drain_filter")]
510
impl <T, F> Drop for DrainFilter<'_, T, F>
511
where
512
    F: FnMut(&mut T::Item) -> bool,
513
    T: Array,
514
{
515
    fn drop(&mut self) {
516
        struct BackshiftOnDrop<'a, 'b, T, F>
517
        where
518
            F: FnMut(&mut T::Item) -> bool,
519
            T: Array
520
        {
521
            drain: &'b mut DrainFilter<'a, T, F>,
522
        }
523
524
        impl<'a, 'b, T, F> Drop for BackshiftOnDrop<'a, 'b, T, F>
525
        where
526
            F: FnMut(&mut T::Item) -> bool,
527
            T: Array
528
        {
529
            fn drop(&mut self) {
530
                unsafe {
531
                    if self.drain.idx < self.drain.old_len && self.drain.del > 0 {
532
                        // This is a pretty messed up state, and there isn't really an
533
                        // obviously right thing to do. We don't want to keep trying
534
                        // to execute `pred`, so we just backshift all the unprocessed
535
                        // elements and tell the vec that they still exist. The backshift
536
                        // is required to prevent a double-drop of the last successfully
537
                        // drained item prior to a panic in the predicate.
538
                        let ptr = self.drain.vec.as_mut_ptr();
539
                        let src = ptr.add(self.drain.idx);
540
                        let dst = src.sub(self.drain.del);
541
                        let tail_len = self.drain.old_len - self.drain.idx;
542
                        src.copy_to(dst, tail_len);
543
                    }
544
                    self.drain.vec.set_len(self.drain.old_len - self.drain.del);
545
                }
546
            }
547
        }
548
549
        let backshift = BackshiftOnDrop { drain: self };
550
551
        // Attempt to consume any remaining elements if the filter predicate
552
        // has not yet panicked. We'll backshift any remaining elements
553
        // whether we've already panicked or if the consumption here panics.
554
        if !backshift.drain.panic_flag {
555
            backshift.drain.for_each(drop);
556
        }
557
    }
558
}
559
560
#[cfg(feature = "drain_keep_rest")]
561
impl <T, F> DrainFilter<'_, T, F>
562
where
563
    F: FnMut(&mut T::Item) -> bool,
564
    T: Array
565
{
566
    /// Keep unyielded elements in the source `Vec`.
567
    ///
568
    /// # Examples
569
    ///
570
    /// ```
571
    /// # use smallvec::{smallvec, SmallVec};
572
    ///
573
    /// let mut vec: SmallVec<[char; 2]> = smallvec!['a', 'b', 'c'];
574
    /// let mut drain = vec.drain_filter(|_| true);
575
    ///
576
    /// assert_eq!(drain.next().unwrap(), 'a');
577
    ///
578
    /// // This call keeps 'b' and 'c' in the vec.
579
    /// drain.keep_rest();
580
    ///
581
    /// // If we wouldn't call `keep_rest()`,
582
    /// // `vec` would be empty.
583
    /// assert_eq!(vec, SmallVec::<[char; 2]>::from_slice(&['b', 'c']));
584
    /// ```
585
    pub fn keep_rest(self)
586
    {
587
        // At this moment layout looks like this:
588
        //
589
        //  _____________________/-- old_len
590
        // /                     \
591
        // [kept] [yielded] [tail]
592
        //        \_______/ ^-- idx
593
        //                \-- del
594
        //
595
        // Normally `Drop` impl would drop [tail] (via .for_each(drop), ie still calling `pred`)
596
        //
597
        // 1. Move [tail] after [kept]
598
        // 2. Update length of the original vec to `old_len - del`
599
        //    a. In case of ZST, this is the only thing we want to do
600
        // 3. Do *not* drop self, as everything is put in a consistent state already, there is nothing to do
601
        let mut this = ManuallyDrop::new(self);
602
603
        unsafe {
604
            // ZSTs have no identity, so we don't need to move them around.
605
            let needs_move = mem::size_of::<T::Item>() != 0;
606
607
            if needs_move && this.idx < this.old_len && this.del > 0 {
608
                let ptr = this.vec.as_mut_ptr();
609
                let src = ptr.add(this.idx);
610
                let dst = src.sub(this.del);
611
                let tail_len = this.old_len - this.idx;
612
                src.copy_to(dst, tail_len);
613
            }
614
615
            let new_len = this.old_len - this.del;
616
            this.vec.set_len(new_len);
617
        }
618
    }
619
}
620
621
#[cfg(feature = "union")]
622
union SmallVecData<A: Array> {
623
    inline: core::mem::ManuallyDrop<MaybeUninit<A>>,
624
    heap: (NonNull<A::Item>, usize),
625
}
626
627
#[cfg(all(feature = "union", feature = "const_new"))]
628
impl<T, const N: usize> SmallVecData<[T; N]> {
629
    #[cfg_attr(docsrs, doc(cfg(feature = "const_new")))]
630
    #[inline]
631
    const fn from_const(inline: MaybeUninit<[T; N]>) -> Self {
632
        SmallVecData {
633
            inline: core::mem::ManuallyDrop::new(inline),
634
        }
635
    }
636
}
637
638
#[cfg(feature = "union")]
639
impl<A: Array> SmallVecData<A> {
640
    #[inline]
641
    unsafe fn inline(&self) -> ConstNonNull<A::Item> {
642
        ConstNonNull::new(self.inline.as_ptr() as *const A::Item).unwrap()
643
    }
644
    #[inline]
645
    unsafe fn inline_mut(&mut self) -> NonNull<A::Item> {
646
        NonNull::new(self.inline.as_mut_ptr() as *mut A::Item).unwrap()
647
    }
648
    #[inline]
649
    fn from_inline(inline: MaybeUninit<A>) -> SmallVecData<A> {
650
        SmallVecData {
651
            inline: core::mem::ManuallyDrop::new(inline),
652
        }
653
    }
654
    // Workaround for https://github.com/rust-lang/rust/issues/157743: when from_inline is
655
    // called with MaybeUninit::uninit(), rustc 1.93+ GVN propagates const <uninit> into the
656
    // ManuallyDrop::new() aggregate, causing LLVM to materialize a global constant that
657
    // MemCpyOpt then collapses into a memset over the whole struct. Using assume_init() of a
658
    // doubly-wrapped MaybeUninit produces Immediate::Uninit instead of const <uninit>, which
659
    // codegen handles as undef without emitting any global. This function also avoids
660
    // introducing an intermediate local that would inflate stack frames in debug builds.
661
    #[inline]
662
    fn empty() -> SmallVecData<A> {
663
        // SAFETY: ManuallyDrop<MaybeUninit<A>> is valid for any bit pattern including
664
        // uninitialized bytes, so assume_init() on a MaybeUninit of that type is sound.
665
        SmallVecData { inline: unsafe { MaybeUninit::uninit().assume_init() } }
666
    }
667
    #[inline]
668
    unsafe fn into_inline(self) -> MaybeUninit<A> {
669
        core::mem::ManuallyDrop::into_inner(self.inline)
670
    }
671
    #[inline]
672
    unsafe fn heap(&self) -> (ConstNonNull<A::Item>, usize) {
673
        (ConstNonNull(self.heap.0), self.heap.1)
674
    }
675
    #[inline]
676
    unsafe fn heap_mut(&mut self) -> (NonNull<A::Item>, &mut usize) {
677
        let h = &mut self.heap;
678
        (h.0, &mut h.1)
679
    }
680
    #[inline]
681
    fn from_heap(ptr: NonNull<A::Item>, len: usize) -> SmallVecData<A> {
682
        SmallVecData { heap: (ptr, len) }
683
    }
684
}
685
686
#[cfg(not(feature = "union"))]
687
enum SmallVecData<A: Array> {
688
    Inline(MaybeUninit<A>),
689
    // Using NonNull and NonZero here allows to reduce size of `SmallVec`.
690
    Heap {
691
        // Since we never allocate on heap
692
        // unless our capacity is bigger than inline capacity
693
        // heap capacity cannot be less than 1.
694
        // Therefore, pointer cannot be null too.
695
        ptr: NonNull<A::Item>,
696
        len: usize,
697
    },
698
}
699
700
#[cfg(all(not(feature = "union"), feature = "const_new"))]
701
impl<T, const N: usize> SmallVecData<[T; N]> {
702
    #[cfg_attr(docsrs, doc(cfg(feature = "const_new")))]
703
    #[inline]
704
    const fn from_const(inline: MaybeUninit<[T; N]>) -> Self {
705
        SmallVecData::Inline(inline)
706
    }
707
}
708
709
#[cfg(not(feature = "union"))]
710
impl<A: Array> SmallVecData<A> {
711
    #[inline]
712
8.47k
    unsafe fn inline(&self) -> ConstNonNull<A::Item> {
713
8.47k
        match self {
714
8.47k
            SmallVecData::Inline(a) => ConstNonNull::new(a.as_ptr() as *const A::Item).unwrap(),
715
0
            _ => debug_unreachable!(),
716
        }
717
8.47k
    }
Unexecuted instantiation: <smallvec::SmallVecData<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::inline
Unexecuted instantiation: <smallvec::SmallVecData<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::inline
<smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::inline
Line
Count
Source
712
8.47k
    unsafe fn inline(&self) -> ConstNonNull<A::Item> {
713
8.47k
        match self {
714
8.47k
            SmallVecData::Inline(a) => ConstNonNull::new(a.as_ptr() as *const A::Item).unwrap(),
715
0
            _ => debug_unreachable!(),
716
        }
717
8.47k
    }
Unexecuted instantiation: <smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::inline
Unexecuted instantiation: <smallvec::SmallVecData<_>>::inline
718
    #[inline]
719
8.82k
    unsafe fn inline_mut(&mut self) -> NonNull<A::Item> {
720
8.82k
        match self {
721
8.82k
            SmallVecData::Inline(a) => NonNull::new(a.as_mut_ptr() as *mut A::Item).unwrap(),
722
0
            _ => debug_unreachable!(),
723
        }
724
8.82k
    }
Unexecuted instantiation: <smallvec::SmallVecData<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::inline_mut
Unexecuted instantiation: <smallvec::SmallVecData<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::inline_mut
Unexecuted instantiation: <smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::inline_mut
<smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::inline_mut
Line
Count
Source
719
8.82k
    unsafe fn inline_mut(&mut self) -> NonNull<A::Item> {
720
8.82k
        match self {
721
8.82k
            SmallVecData::Inline(a) => NonNull::new(a.as_mut_ptr() as *mut A::Item).unwrap(),
722
0
            _ => debug_unreachable!(),
723
        }
724
8.82k
    }
Unexecuted instantiation: <smallvec::SmallVecData<_>>::inline_mut
725
    #[inline]
726
0
    fn from_inline(inline: MaybeUninit<A>) -> SmallVecData<A> {
727
0
        SmallVecData::Inline(inline)
728
0
    }
729
    // See the comment on the union variant's empty() for why this exists.
730
    #[inline]
731
2.94k
    fn empty() -> SmallVecData<A> {
732
        // SAFETY: MaybeUninit<A> is valid for any bit pattern including uninitialized bytes,
733
        // so assume_init() on a MaybeUninit of that type is sound.
734
2.94k
        SmallVecData::Inline(unsafe { MaybeUninit::uninit().assume_init() })
735
2.94k
    }
Unexecuted instantiation: <smallvec::SmallVecData<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::empty
Unexecuted instantiation: <smallvec::SmallVecData<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::empty
Unexecuted instantiation: <smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::empty
Unexecuted instantiation: <smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::empty
Unexecuted instantiation: <smallvec::SmallVecData<_>>::empty
<smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::empty
Line
Count
Source
731
2.94k
    fn empty() -> SmallVecData<A> {
732
        // SAFETY: MaybeUninit<A> is valid for any bit pattern including uninitialized bytes,
733
        // so assume_init() on a MaybeUninit of that type is sound.
734
2.94k
        SmallVecData::Inline(unsafe { MaybeUninit::uninit().assume_init() })
735
2.94k
    }
736
    #[inline]
737
0
    unsafe fn into_inline(self) -> MaybeUninit<A> {
738
0
        match self {
739
0
            SmallVecData::Inline(a) => a,
740
0
            _ => debug_unreachable!(),
741
        }
742
0
    }
743
    #[inline]
744
0
    unsafe fn heap(&self) -> (ConstNonNull<A::Item>, usize) {
745
0
        match self {
746
0
            SmallVecData::Heap { ptr, len } => (ConstNonNull(*ptr), *len),
747
0
            _ => debug_unreachable!(),
748
        }
749
0
    }
Unexecuted instantiation: <smallvec::SmallVecData<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::heap
Unexecuted instantiation: <smallvec::SmallVecData<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::heap
Unexecuted instantiation: <smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::heap
Unexecuted instantiation: <smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::heap
Unexecuted instantiation: <smallvec::SmallVecData<_>>::heap
750
    #[inline]
751
0
    unsafe fn heap_mut(&mut self) -> (NonNull<A::Item>, &mut usize) {
752
0
        match self {
753
0
            SmallVecData::Heap { ptr, len } => (*ptr, len),
754
0
            _ => debug_unreachable!(),
755
        }
756
0
    }
Unexecuted instantiation: <smallvec::SmallVecData<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::heap_mut
Unexecuted instantiation: <smallvec::SmallVecData<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::heap_mut
Unexecuted instantiation: <smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::heap_mut
Unexecuted instantiation: <smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::heap_mut
Unexecuted instantiation: <smallvec::SmallVecData<_>>::heap_mut
757
    #[inline]
758
0
    fn from_heap(ptr: NonNull<A::Item>, len: usize) -> SmallVecData<A> {
759
0
        SmallVecData::Heap { ptr, len }
760
0
    }
Unexecuted instantiation: <smallvec::SmallVecData<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::from_heap
Unexecuted instantiation: <smallvec::SmallVecData<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::from_heap
Unexecuted instantiation: <smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::from_heap
Unexecuted instantiation: <smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::from_heap
Unexecuted instantiation: <smallvec::SmallVecData<_>>::from_heap
Unexecuted instantiation: <smallvec::SmallVecData<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::from_heap
761
}
762
763
unsafe impl<A: Array + Send> Send for SmallVecData<A> {}
764
unsafe impl<A: Array + Sync> Sync for SmallVecData<A> {}
765
766
/// A `Vec`-like container that can store a small number of elements inline.
767
///
768
/// `SmallVec` acts like a vector, but can store a limited amount of data inline within the
769
/// `SmallVec` struct rather than in a separate allocation.  If the data exceeds this limit, the
770
/// `SmallVec` will "spill" its data onto the heap, allocating a new buffer to hold it.
771
///
772
/// The amount of data that a `SmallVec` can store inline depends on its backing store. The backing
773
/// store can be any type that implements the `Array` trait; usually it is a small fixed-sized
774
/// array.  For example a `SmallVec<[u64; 8]>` can hold up to eight 64-bit integers inline.
775
///
776
/// ## Example
777
///
778
/// ```rust
779
/// use smallvec::SmallVec;
780
/// let mut v = SmallVec::<[u8; 4]>::new(); // initialize an empty vector
781
///
782
/// // The vector can hold up to 4 items without spilling onto the heap.
783
/// v.extend(0..4);
784
/// assert_eq!(v.len(), 4);
785
/// assert!(!v.spilled());
786
///
787
/// // Pushing another element will force the buffer to spill:
788
/// v.push(4);
789
/// assert_eq!(v.len(), 5);
790
/// assert!(v.spilled());
791
/// ```
792
pub struct SmallVec<A: Array> {
793
    // The capacity field is used to determine which of the storage variants is active:
794
    // If capacity <= Self::inline_capacity() then the inline variant is used and capacity holds the current length of the vector (number of elements actually in use).
795
    // If capacity > Self::inline_capacity() then the heap variant is used and capacity holds the size of the memory allocation.
796
    capacity: usize,
797
    data: SmallVecData<A>,
798
}
799
800
impl<A: Array> SmallVec<A> {
801
    /// Construct an empty vector
802
    #[inline]
803
2.94k
    pub fn new() -> SmallVec<A> {
804
        // Try to detect invalid custom implementations of `Array`. Hopefully,
805
        // this check should be optimized away entirely for valid ones.
806
2.94k
        assert!(
807
2.94k
            mem::size_of::<A>() == A::size() * mem::size_of::<A::Item>()
808
2.94k
                && mem::align_of::<A>() >= mem::align_of::<A::Item>()
809
        );
810
2.94k
        SmallVec {
811
2.94k
            capacity: 0,
812
2.94k
            data: SmallVecData::empty(),
813
2.94k
        }
814
2.94k
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::new
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::new
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::new
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::new
Unexecuted instantiation: <smallvec::SmallVec<_>>::new
<smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::new
Line
Count
Source
803
2.94k
    pub fn new() -> SmallVec<A> {
804
        // Try to detect invalid custom implementations of `Array`. Hopefully,
805
        // this check should be optimized away entirely for valid ones.
806
2.94k
        assert!(
807
2.94k
            mem::size_of::<A>() == A::size() * mem::size_of::<A::Item>()
808
2.94k
                && mem::align_of::<A>() >= mem::align_of::<A::Item>()
809
        );
810
2.94k
        SmallVec {
811
2.94k
            capacity: 0,
812
2.94k
            data: SmallVecData::empty(),
813
2.94k
        }
814
2.94k
    }
815
816
    /// Construct an empty vector with enough capacity pre-allocated to store at least `n`
817
    /// elements.
818
    ///
819
    /// Will create a heap allocation only if `n` is larger than the inline capacity.
820
    ///
821
    /// ```
822
    /// # use smallvec::SmallVec;
823
    ///
824
    /// let v: SmallVec<[u8; 3]> = SmallVec::with_capacity(100);
825
    ///
826
    /// assert!(v.is_empty());
827
    /// assert!(v.capacity() >= 100);
828
    /// ```
829
    #[inline]
830
0
    pub fn with_capacity(n: usize) -> Self {
831
0
        let mut v = SmallVec::new();
832
0
        v.reserve_exact(n);
833
0
        v
834
0
    }
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::with_capacity
Unexecuted instantiation: <smallvec::SmallVec<_>>::with_capacity
835
836
    /// Construct a new `SmallVec` from a `Vec<A::Item>`.
837
    ///
838
    /// Elements will be copied to the inline buffer if `vec.capacity() <= Self::inline_capacity()`.
839
    ///
840
    /// ```rust
841
    /// use smallvec::SmallVec;
842
    ///
843
    /// let vec = vec![1, 2, 3, 4, 5];
844
    /// let small_vec: SmallVec<[_; 3]> = SmallVec::from_vec(vec);
845
    ///
846
    /// assert_eq!(&*small_vec, &[1, 2, 3, 4, 5]);
847
    /// ```
848
    #[inline]
849
0
    pub fn from_vec(mut vec: Vec<A::Item>) -> SmallVec<A> {
850
0
        if vec.capacity() <= Self::inline_capacity() {
851
            // Cannot use Vec with smaller capacity
852
            // because we use value of `Self::capacity` field as indicator.
853
            unsafe {
854
0
                let mut data = SmallVecData::<A>::empty();
855
0
                let len = vec.len();
856
0
                vec.set_len(0);
857
0
                ptr::copy_nonoverlapping(vec.as_ptr(), data.inline_mut().as_ptr(), len);
858
859
0
                SmallVec {
860
0
                    capacity: len,
861
0
                    data,
862
0
                }
863
            }
864
        } else {
865
0
            let (ptr, cap, len) = (vec.as_mut_ptr(), vec.capacity(), vec.len());
866
0
            mem::forget(vec);
867
0
            let ptr = NonNull::new(ptr)
868
                // See docs: https://doc.rust-lang.org/std/vec/struct.Vec.html#method.as_mut_ptr
869
0
                .expect("Cannot be null by `Vec` invariant");
870
871
0
            SmallVec {
872
0
                capacity: cap,
873
0
                data: SmallVecData::from_heap(ptr, len),
874
0
            }
875
        }
876
0
    }
877
878
    /// Constructs a new `SmallVec` on the stack from an `A` without
879
    /// copying elements.
880
    ///
881
    /// ```rust
882
    /// use smallvec::SmallVec;
883
    ///
884
    /// let buf = [1, 2, 3, 4, 5];
885
    /// let small_vec: SmallVec<_> = SmallVec::from_buf(buf);
886
    ///
887
    /// assert_eq!(&*small_vec, &[1, 2, 3, 4, 5]);
888
    /// ```
889
    #[inline]
890
0
    pub fn from_buf(buf: A) -> SmallVec<A> {
891
0
        SmallVec {
892
0
            capacity: A::size(),
893
0
            data: SmallVecData::from_inline(MaybeUninit::new(buf)),
894
0
        }
895
0
    }
896
897
    /// Constructs a new `SmallVec` on the stack from an `A` without
898
    /// copying elements. Also sets the length, which must be less or
899
    /// equal to the size of `buf`.
900
    ///
901
    /// ```rust
902
    /// use smallvec::SmallVec;
903
    ///
904
    /// let buf = [1, 2, 3, 4, 5, 0, 0, 0];
905
    /// let small_vec: SmallVec<_> = SmallVec::from_buf_and_len(buf, 5);
906
    ///
907
    /// assert_eq!(&*small_vec, &[1, 2, 3, 4, 5]);
908
    /// ```
909
    #[inline]
910
0
    pub fn from_buf_and_len(buf: A, len: usize) -> SmallVec<A> {
911
0
        assert!(len <= A::size());
912
0
        unsafe { SmallVec::from_buf_and_len_unchecked(MaybeUninit::new(buf), len) }
913
0
    }
914
915
    /// Constructs a new `SmallVec` on the stack from an `A` without
916
    /// copying elements. Also sets the length. The user is responsible
917
    /// for ensuring that `len <= A::size()`.
918
    ///
919
    /// ```rust
920
    /// use smallvec::SmallVec;
921
    /// use std::mem::MaybeUninit;
922
    ///
923
    /// let buf = [1, 2, 3, 4, 5, 0, 0, 0];
924
    /// let small_vec: SmallVec<_> = unsafe {
925
    ///     SmallVec::from_buf_and_len_unchecked(MaybeUninit::new(buf), 5)
926
    /// };
927
    ///
928
    /// assert_eq!(&*small_vec, &[1, 2, 3, 4, 5]);
929
    /// ```
930
    #[inline]
931
0
    pub unsafe fn from_buf_and_len_unchecked(buf: MaybeUninit<A>, len: usize) -> SmallVec<A> {
932
0
        SmallVec {
933
0
            capacity: len,
934
0
            data: SmallVecData::from_inline(buf),
935
0
        }
936
0
    }
937
938
    /// Sets the length of a vector.
939
    ///
940
    /// This will explicitly set the size of the vector, without actually
941
    /// modifying its buffers, so it is up to the caller to ensure that the
942
    /// vector is actually the specified size.
943
0
    pub unsafe fn set_len(&mut self, new_len: usize) {
944
0
        let (_, len_ptr, _) = self.triple_mut();
945
0
        *len_ptr = new_len;
946
0
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::set_len
Unexecuted instantiation: <smallvec::SmallVec<_>>::set_len
947
948
    /// The maximum number of elements this vector can hold inline
949
    #[inline]
950
37.5k
    fn inline_capacity() -> usize {
951
37.5k
        if mem::size_of::<A::Item>() > 0 {
952
37.5k
            A::size()
953
        } else {
954
            // For zero-size items code like `ptr.add(offset)` always returns the same pointer.
955
            // Therefore all items are at the same address,
956
            // and any array size has capacity for infinitely many items.
957
            // The capacity is limited by the bit width of the length field.
958
            //
959
            // `Vec` also does this:
960
            // https://github.com/rust-lang/rust/blob/1.44.0/src/liballoc/raw_vec.rs#L186
961
            //
962
            // In our case, this also ensures that a smallvec of zero-size items never spills,
963
            // and we never try to allocate zero bytes which `std::alloc::alloc` disallows.
964
0
            core::usize::MAX
965
        }
966
37.5k
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::inline_capacity
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::inline_capacity
<smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::inline_capacity
Line
Count
Source
950
37.5k
    fn inline_capacity() -> usize {
951
37.5k
        if mem::size_of::<A::Item>() > 0 {
952
37.5k
            A::size()
953
        } else {
954
            // For zero-size items code like `ptr.add(offset)` always returns the same pointer.
955
            // Therefore all items are at the same address,
956
            // and any array size has capacity for infinitely many items.
957
            // The capacity is limited by the bit width of the length field.
958
            //
959
            // `Vec` also does this:
960
            // https://github.com/rust-lang/rust/blob/1.44.0/src/liballoc/raw_vec.rs#L186
961
            //
962
            // In our case, this also ensures that a smallvec of zero-size items never spills,
963
            // and we never try to allocate zero bytes which `std::alloc::alloc` disallows.
964
0
            core::usize::MAX
965
        }
966
37.5k
    }
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::inline_capacity
Unexecuted instantiation: <smallvec::SmallVec<_>>::inline_capacity
967
968
    /// The maximum number of elements this vector can hold inline
969
    #[inline]
970
0
    pub fn inline_size(&self) -> usize {
971
0
        Self::inline_capacity()
972
0
    }
973
974
    /// The number of elements stored in the vector
975
    #[inline]
976
0
    pub fn len(&self) -> usize {
977
0
        self.triple().1
978
0
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::len
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::len
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::len
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::len
Unexecuted instantiation: <smallvec::SmallVec<_>>::len
979
980
    /// Returns `true` if the vector is empty
981
    #[inline]
982
0
    pub fn is_empty(&self) -> bool {
983
0
        self.len() == 0
984
0
    }
985
986
    /// The number of items the vector can hold without reallocating
987
    #[inline]
988
0
    pub fn capacity(&self) -> usize {
989
0
        self.triple().2
990
0
    }
991
992
    /// Returns a tuple with (data ptr, len, capacity)
993
    /// Useful to get all `SmallVec` properties with a single check of the current storage variant.
994
    #[inline]
995
8.47k
    fn triple(&self) -> (ConstNonNull<A::Item>, usize, usize) {
996
        unsafe {
997
8.47k
            if self.spilled() {
998
0
                let (ptr, len) = self.data.heap();
999
0
                (ptr, len, self.capacity)
1000
            } else {
1001
8.47k
                (self.data.inline(), self.capacity, Self::inline_capacity())
1002
            }
1003
        }
1004
8.47k
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::triple
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::triple
<smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::triple
Line
Count
Source
995
8.47k
    fn triple(&self) -> (ConstNonNull<A::Item>, usize, usize) {
996
        unsafe {
997
8.47k
            if self.spilled() {
998
0
                let (ptr, len) = self.data.heap();
999
0
                (ptr, len, self.capacity)
1000
            } else {
1001
8.47k
                (self.data.inline(), self.capacity, Self::inline_capacity())
1002
            }
1003
        }
1004
8.47k
    }
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::triple
Unexecuted instantiation: <smallvec::SmallVec<_>>::triple
1005
1006
    /// Returns a tuple with (data ptr, len ptr, capacity)
1007
    #[inline]
1008
8.82k
    fn triple_mut(&mut self) -> (NonNull<A::Item>, &mut usize, usize) {
1009
        unsafe {
1010
8.82k
            if self.spilled() {
1011
0
                let (ptr, len_ptr) = self.data.heap_mut();
1012
0
                (ptr, len_ptr, self.capacity)
1013
            } else {
1014
8.82k
                (
1015
8.82k
                    self.data.inline_mut(),
1016
8.82k
                    &mut self.capacity,
1017
8.82k
                    Self::inline_capacity(),
1018
8.82k
                )
1019
            }
1020
        }
1021
8.82k
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::triple_mut
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::triple_mut
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::triple_mut
<smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::triple_mut
Line
Count
Source
1008
8.82k
    fn triple_mut(&mut self) -> (NonNull<A::Item>, &mut usize, usize) {
1009
        unsafe {
1010
8.82k
            if self.spilled() {
1011
0
                let (ptr, len_ptr) = self.data.heap_mut();
1012
0
                (ptr, len_ptr, self.capacity)
1013
            } else {
1014
8.82k
                (
1015
8.82k
                    self.data.inline_mut(),
1016
8.82k
                    &mut self.capacity,
1017
8.82k
                    Self::inline_capacity(),
1018
8.82k
                )
1019
            }
1020
        }
1021
8.82k
    }
Unexecuted instantiation: <smallvec::SmallVec<_>>::triple_mut
1022
1023
    /// Returns `true` if the data has spilled into a separate heap-allocated buffer.
1024
    #[inline]
1025
20.2k
    pub fn spilled(&self) -> bool {
1026
20.2k
        self.capacity > Self::inline_capacity()
1027
20.2k
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::spilled
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::spilled
<smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::spilled
Line
Count
Source
1025
20.2k
    pub fn spilled(&self) -> bool {
1026
20.2k
        self.capacity > Self::inline_capacity()
1027
20.2k
    }
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::spilled
Unexecuted instantiation: <smallvec::SmallVec<_>>::spilled
1028
1029
    /// Creates a draining iterator that removes the specified range in the vector
1030
    /// and yields the removed items.
1031
    ///
1032
    /// Note 1: The element range is removed even if the iterator is only
1033
    /// partially consumed or not consumed at all.
1034
    ///
1035
    /// Note 2: It is unspecified how many elements are removed from the vector
1036
    /// if the `Drain` value is leaked.
1037
    ///
1038
    /// # Panics
1039
    ///
1040
    /// Panics if the starting point is greater than the end point or if
1041
    /// the end point is greater than the length of the vector.
1042
0
    pub fn drain<R>(&mut self, range: R) -> Drain<'_, A>
1043
0
    where
1044
0
        R: RangeBounds<usize>,
1045
    {
1046
        use core::ops::Bound::*;
1047
1048
0
        let len = self.len();
1049
0
        let start = match range.start_bound() {
1050
0
            Included(&n) => n,
1051
0
            Excluded(&n) => n.checked_add(1).expect("Range start out of bounds"),
1052
0
            Unbounded => 0,
1053
        };
1054
0
        let end = match range.end_bound() {
1055
0
            Included(&n) => n.checked_add(1).expect("Range end out of bounds"),
1056
0
            Excluded(&n) => n,
1057
0
            Unbounded => len,
1058
        };
1059
1060
0
        assert!(start <= end);
1061
0
        assert!(end <= len);
1062
1063
        unsafe {
1064
0
            self.set_len(start);
1065
1066
0
            let range_slice = slice::from_raw_parts(self.as_ptr().add(start), end - start);
1067
1068
0
            Drain {
1069
0
                tail_start: end,
1070
0
                tail_len: len - end,
1071
0
                iter: range_slice.iter(),
1072
0
                // Since self is a &mut, passing it to a function would invalidate the slice iterator.
1073
0
                vec: NonNull::new_unchecked(self as *mut _),
1074
0
            }
1075
        }
1076
0
    }
1077
1078
    #[cfg(feature = "drain_filter")]
1079
    /// Creates an iterator which uses a closure to determine if an element should be removed.
1080
    ///
1081
    /// If the closure returns true, the element is removed and yielded. If the closure returns
1082
    /// false, the element will remain in the vector and will not be yielded by the iterator.
1083
    ///
1084
    /// Using this method is equivalent to the following code:
1085
    /// ```
1086
    /// # use smallvec::SmallVec;
1087
    /// # let some_predicate = |x: &mut i32| { *x == 2 || *x == 3 || *x == 6 };
1088
    /// # let mut vec: SmallVec<[i32; 8]> = SmallVec::from_slice(&[1i32, 2, 3, 4, 5, 6]);
1089
    /// let mut i = 0;
1090
    /// while i < vec.len() {
1091
    ///     if some_predicate(&mut vec[i]) {
1092
    ///         let val = vec.remove(i);
1093
    ///         // your code here
1094
    ///     } else {
1095
    ///         i += 1;
1096
    ///     }
1097
    /// }
1098
    ///
1099
    /// # assert_eq!(vec, SmallVec::<[i32; 8]>::from_slice(&[1i32, 4, 5]));
1100
    /// ```
1101
    /// ///
1102
    /// But `drain_filter` is easier to use. `drain_filter` is also more efficient,
1103
    /// because it can backshift the elements of the array in bulk.
1104
    ///
1105
    /// Note that `drain_filter` also lets you mutate every element in the filter closure,
1106
    /// regardless of whether you choose to keep or remove it.
1107
    ///
1108
    /// # Examples
1109
    ///
1110
    /// Splitting an array into evens and odds, reusing the original allocation:
1111
    ///
1112
    /// ```
1113
    /// # use smallvec::SmallVec;
1114
    /// let mut numbers: SmallVec<[i32; 16]> = SmallVec::from_slice(&[1i32, 2, 3, 4, 5, 6, 8, 9, 11, 13, 14, 15]);
1115
    ///
1116
    /// let evens = numbers.drain_filter(|x| *x % 2 == 0).collect::<SmallVec<[i32; 16]>>();
1117
    /// let odds = numbers;
1118
    ///
1119
    /// assert_eq!(evens, SmallVec::<[i32; 16]>::from_slice(&[2i32, 4, 6, 8, 14]));
1120
    /// assert_eq!(odds, SmallVec::<[i32; 16]>::from_slice(&[1i32, 3, 5, 9, 11, 13, 15]));
1121
    /// ```
1122
    pub fn drain_filter<F>(&mut self, filter: F) -> DrainFilter<'_, A, F,>
1123
    where
1124
        F: FnMut(&mut A::Item) -> bool,
1125
    {
1126
        let old_len = self.len();
1127
1128
        // Guard against us getting leaked (leak amplification)
1129
        unsafe {
1130
            self.set_len(0);
1131
        }
1132
1133
        DrainFilter { vec: self, idx: 0, del: 0, old_len, pred: filter, panic_flag: false }
1134
    }
1135
1136
    /// Append an item to the vector.
1137
    #[inline]
1138
0
    pub fn push(&mut self, value: A::Item) {
1139
        unsafe {
1140
0
            let (mut ptr, mut len, cap) = self.triple_mut();
1141
0
            if *len == cap {
1142
0
                self.reserve_one_unchecked();
1143
0
                let (heap_ptr, heap_len) = self.data.heap_mut();
1144
0
                ptr = heap_ptr;
1145
0
                len = heap_len;
1146
0
            }
1147
0
            ptr::write(ptr.as_ptr().add(*len), value);
1148
0
            *len += 1;
1149
        }
1150
0
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::push
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::push
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::push
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::push
Unexecuted instantiation: <smallvec::SmallVec<_>>::push
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::push
1151
1152
    /// Remove an item from the end of the vector and return it, or None if empty.
1153
    #[inline]
1154
0
    pub fn pop(&mut self) -> Option<A::Item> {
1155
        unsafe {
1156
0
            let (ptr, len_ptr, _) = self.triple_mut();
1157
0
            let ptr: *const _ = ptr.as_ptr();
1158
0
            if *len_ptr == 0 {
1159
0
                return None;
1160
0
            }
1161
0
            let last_index = *len_ptr - 1;
1162
0
            *len_ptr = last_index;
1163
0
            Some(ptr::read(ptr.add(last_index)))
1164
        }
1165
0
    }
1166
1167
    /// Moves all the elements of `other` into `self`, leaving `other` empty.
1168
    ///
1169
    /// # Example
1170
    ///
1171
    /// ```
1172
    /// # use smallvec::{SmallVec, smallvec};
1173
    /// let mut v0: SmallVec<[u8; 16]> = smallvec![1, 2, 3];
1174
    /// let mut v1: SmallVec<[u8; 32]> = smallvec![4, 5, 6];
1175
    /// v0.append(&mut v1);
1176
    /// assert_eq!(*v0, [1, 2, 3, 4, 5, 6]);
1177
    /// assert_eq!(*v1, []);
1178
    /// ```
1179
0
    pub fn append<B>(&mut self, other: &mut SmallVec<B>)
1180
0
    where
1181
0
        B: Array<Item = A::Item>,
1182
    {
1183
0
        self.extend(other.drain(..))
1184
0
    }
1185
1186
    /// Re-allocate to set the capacity to `max(new_cap, inline_size())`.
1187
    ///
1188
    /// Panics if `new_cap` is less than the vector's length
1189
    /// or if the capacity computation overflows `usize`.
1190
0
    pub fn grow(&mut self, new_cap: usize) {
1191
0
        infallible(self.try_grow(new_cap))
1192
0
    }
1193
1194
    /// Re-allocate to set the capacity to `max(new_cap, inline_size())`.
1195
    ///
1196
    /// Panics if `new_cap` is less than the vector's length
1197
0
    pub fn try_grow(&mut self, new_cap: usize) -> Result<(), CollectionAllocErr> {
1198
        unsafe {
1199
0
            let unspilled = !self.spilled();
1200
0
            let (ptr, &mut len, cap) = self.triple_mut();
1201
0
            assert!(new_cap >= len);
1202
0
            if new_cap <= Self::inline_capacity() {
1203
0
                if unspilled {
1204
0
                    return Ok(());
1205
0
                }
1206
0
                self.data = SmallVecData::empty();
1207
0
                ptr::copy_nonoverlapping(ptr.as_ptr(), self.data.inline_mut().as_ptr(), len);
1208
0
                self.capacity = len;
1209
0
                deallocate(ptr, cap);
1210
0
            } else if new_cap != cap {
1211
0
                let layout = layout_array::<A::Item>(new_cap)?;
1212
0
                debug_assert!(layout.size() > 0);
1213
                let new_alloc;
1214
0
                if unspilled {
1215
0
                    new_alloc = NonNull::new(alloc::alloc::alloc(layout))
1216
0
                        .ok_or(CollectionAllocErr::AllocErr { layout })?
1217
0
                        .cast();
1218
0
                    ptr::copy_nonoverlapping(ptr.as_ptr(), new_alloc.as_ptr(), len);
1219
                } else {
1220
                    // This should never fail since the same succeeded
1221
                    // when previously allocating `ptr`.
1222
0
                    let old_layout = layout_array::<A::Item>(cap)?;
1223
1224
0
                    let new_ptr =
1225
0
                        alloc::alloc::realloc(ptr.as_ptr() as *mut u8, old_layout, layout.size());
1226
0
                    new_alloc = NonNull::new(new_ptr)
1227
0
                        .ok_or(CollectionAllocErr::AllocErr { layout })?
1228
0
                        .cast();
1229
                }
1230
0
                self.data = SmallVecData::from_heap(new_alloc, len);
1231
0
                self.capacity = new_cap;
1232
0
            }
1233
0
            Ok(())
1234
        }
1235
0
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::try_grow
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::try_grow
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::try_grow
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::try_grow
Unexecuted instantiation: <smallvec::SmallVec<_>>::try_grow
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::try_grow
1236
1237
    /// Reserve capacity for `additional` more elements to be inserted.
1238
    ///
1239
    /// May reserve more space to avoid frequent reallocations.
1240
    ///
1241
    /// Panics if the capacity computation overflows `usize`.
1242
    #[inline]
1243
2.94k
    pub fn reserve(&mut self, additional: usize) {
1244
2.94k
        infallible(self.try_reserve(additional))
1245
2.94k
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::reserve
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::reserve
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::reserve
Unexecuted instantiation: <smallvec::SmallVec<_>>::reserve
<smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::reserve
Line
Count
Source
1243
2.94k
    pub fn reserve(&mut self, additional: usize) {
1244
2.94k
        infallible(self.try_reserve(additional))
1245
2.94k
    }
1246
1247
    /// Internal method used to grow in push() and insert(), where we know already we have to grow.
1248
    #[cold]
1249
0
    fn reserve_one_unchecked(&mut self) {
1250
0
        debug_assert_eq!(self.len(), self.capacity());
1251
0
        let new_cap = self.len()
1252
0
            .checked_add(1)
1253
0
            .and_then(usize::checked_next_power_of_two)
1254
0
            .expect("capacity overflow");
1255
0
        infallible(self.try_grow(new_cap))
1256
0
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::reserve_one_unchecked
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::reserve_one_unchecked
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::reserve_one_unchecked
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::reserve_one_unchecked
Unexecuted instantiation: <smallvec::SmallVec<_>>::reserve_one_unchecked
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::reserve_one_unchecked
1257
1258
    /// Reserve capacity for `additional` more elements to be inserted.
1259
    ///
1260
    /// May reserve more space to avoid frequent reallocations.
1261
2.94k
    pub fn try_reserve(&mut self, additional: usize) -> Result<(), CollectionAllocErr> {
1262
        // prefer triple_mut() even if triple() would work so that the optimizer removes duplicated
1263
        // calls to it from callers.
1264
2.94k
        let (_, &mut len, cap) = self.triple_mut();
1265
2.94k
        if cap - len >= additional {
1266
2.94k
            return Ok(());
1267
0
        }
1268
0
        let new_cap = len
1269
0
            .checked_add(additional)
1270
0
            .and_then(usize::checked_next_power_of_two)
1271
0
            .ok_or(CollectionAllocErr::CapacityOverflow)?;
1272
0
        self.try_grow(new_cap)
1273
2.94k
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::try_reserve
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]>>::try_reserve
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::try_reserve
Unexecuted instantiation: <smallvec::SmallVec<_>>::try_reserve
<smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]>>::try_reserve
Line
Count
Source
1261
2.94k
    pub fn try_reserve(&mut self, additional: usize) -> Result<(), CollectionAllocErr> {
1262
        // prefer triple_mut() even if triple() would work so that the optimizer removes duplicated
1263
        // calls to it from callers.
1264
2.94k
        let (_, &mut len, cap) = self.triple_mut();
1265
2.94k
        if cap - len >= additional {
1266
2.94k
            return Ok(());
1267
0
        }
1268
0
        let new_cap = len
1269
0
            .checked_add(additional)
1270
0
            .and_then(usize::checked_next_power_of_two)
1271
0
            .ok_or(CollectionAllocErr::CapacityOverflow)?;
1272
0
        self.try_grow(new_cap)
1273
2.94k
    }
1274
1275
    /// Reserve the minimum capacity for `additional` more elements to be inserted.
1276
    ///
1277
    /// Panics if the new capacity overflows `usize`.
1278
0
    pub fn reserve_exact(&mut self, additional: usize) {
1279
0
        infallible(self.try_reserve_exact(additional))
1280
0
    }
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::reserve_exact
Unexecuted instantiation: <smallvec::SmallVec<_>>::reserve_exact
1281
1282
    /// Reserve the minimum capacity for `additional` more elements to be inserted.
1283
0
    pub fn try_reserve_exact(&mut self, additional: usize) -> Result<(), CollectionAllocErr> {
1284
0
        let (_, &mut len, cap) = self.triple_mut();
1285
0
        if cap - len >= additional {
1286
0
            return Ok(());
1287
0
        }
1288
0
        let new_cap = len
1289
0
            .checked_add(additional)
1290
0
            .ok_or(CollectionAllocErr::CapacityOverflow)?;
1291
0
        self.try_grow(new_cap)
1292
0
    }
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]>>::try_reserve_exact
Unexecuted instantiation: <smallvec::SmallVec<_>>::try_reserve_exact
1293
1294
    /// Shrink the capacity of the vector as much as possible.
1295
    ///
1296
    /// When possible, this will move data from an external heap buffer to the vector's inline
1297
    /// storage.
1298
0
    pub fn shrink_to_fit(&mut self) {
1299
0
        if !self.spilled() {
1300
0
            return;
1301
0
        }
1302
0
        let len = self.len();
1303
0
        if self.inline_size() >= len {
1304
0
            unsafe {
1305
0
                let (ptr, len) = self.data.heap();
1306
0
                self.data = SmallVecData::empty();
1307
0
                ptr::copy_nonoverlapping(ptr.as_ptr(), self.data.inline_mut().as_ptr(), len);
1308
0
                deallocate(ptr.0, self.capacity);
1309
0
                self.capacity = len;
1310
0
            }
1311
0
        } else if self.capacity() > len {
1312
0
            self.grow(len);
1313
0
        }
1314
0
    }
1315
1316
    /// Shorten the vector, keeping the first `len` elements and dropping the rest.
1317
    ///
1318
    /// If `len` is greater than or equal to the vector's current length, this has no
1319
    /// effect.
1320
    ///
1321
    /// This does not re-allocate.  If you want the vector's capacity to shrink, call
1322
    /// `shrink_to_fit` after truncating.
1323
0
    pub fn truncate(&mut self, len: usize) {
1324
        unsafe {
1325
0
            let (ptr, len_ptr, _) = self.triple_mut();
1326
0
            let ptr = ptr.as_ptr();
1327
0
            while len < *len_ptr {
1328
0
                let last_index = *len_ptr - 1;
1329
0
                *len_ptr = last_index;
1330
0
                ptr::drop_in_place(ptr.add(last_index));
1331
0
            }
1332
        }
1333
0
    }
1334
1335
    /// Extracts a slice containing the entire vector.
1336
    ///
1337
    /// Equivalent to `&s[..]`.
1338
0
    pub fn as_slice(&self) -> &[A::Item] {
1339
0
        self
1340
0
    }
1341
1342
    /// Extracts a mutable slice of the entire vector.
1343
    ///
1344
    /// Equivalent to `&mut s[..]`.
1345
0
    pub fn as_mut_slice(&mut self) -> &mut [A::Item] {
1346
0
        self
1347
0
    }
1348
1349
    /// Remove the element at position `index`, replacing it with the last element.
1350
    ///
1351
    /// This does not preserve ordering, but is O(1).
1352
    ///
1353
    /// Panics if `index` is out of bounds.
1354
    #[inline]
1355
0
    pub fn swap_remove(&mut self, index: usize) -> A::Item {
1356
0
        let len = self.len();
1357
0
        self.swap(len - 1, index);
1358
0
        self.pop()
1359
0
            .unwrap_or_else(|| unsafe { unreachable_unchecked() })
1360
0
    }
1361
1362
    /// Remove all elements from the vector.
1363
    #[inline]
1364
0
    pub fn clear(&mut self) {
1365
0
        self.truncate(0);
1366
0
    }
1367
1368
    /// Remove and return the element at position `index`, shifting all elements after it to the
1369
    /// left.
1370
    ///
1371
    /// Panics if `index` is out of bounds.
1372
0
    pub fn remove(&mut self, index: usize) -> A::Item {
1373
        unsafe {
1374
0
            let (ptr, len_ptr, _) = self.triple_mut();
1375
0
            let len = *len_ptr;
1376
0
            assert!(index < len);
1377
0
            *len_ptr = len - 1;
1378
0
            let ptr = ptr.as_ptr().add(index);
1379
0
            let item = ptr::read(ptr);
1380
0
            ptr::copy(ptr.add(1), ptr, len - index - 1);
1381
0
            item
1382
        }
1383
0
    }
1384
1385
    /// Insert an element at position `index`, shifting all elements after it to the right.
1386
    ///
1387
    /// Panics if `index > len`.
1388
0
    pub fn insert(&mut self, index: usize, element: A::Item) {
1389
        unsafe {
1390
0
            let (mut ptr, mut len_ptr, cap) = self.triple_mut();
1391
0
            if *len_ptr == cap {
1392
0
                self.reserve_one_unchecked();
1393
0
                let (heap_ptr, heap_len_ptr) = self.data.heap_mut();
1394
0
                ptr = heap_ptr;
1395
0
                len_ptr = heap_len_ptr;
1396
0
            }
1397
0
            let mut ptr = ptr.as_ptr();
1398
0
            let len = *len_ptr;
1399
0
            if index > len {
1400
0
                panic!("index exceeds length");
1401
0
            }
1402
            // SAFETY: add is UB if index > len, but we panicked first
1403
0
            ptr = ptr.add(index);
1404
0
            if index < len {
1405
0
                // Shift element to the right of `index`.
1406
0
                ptr::copy(ptr, ptr.add(1), len - index);
1407
0
            }
1408
0
            *len_ptr = len + 1;
1409
0
            ptr::write(ptr, element);
1410
        }
1411
0
    }
1412
1413
    /// Insert multiple elements at position `index`, shifting all following elements toward the
1414
    /// back.
1415
0
    pub fn insert_many<I: IntoIterator<Item = A::Item>>(&mut self, index: usize, iterable: I) {
1416
0
        let mut iter = iterable.into_iter();
1417
0
        if index == self.len() {
1418
0
            return self.extend(iter);
1419
0
        }
1420
1421
0
        let (lower_size_bound, _) = iter.size_hint();
1422
0
        assert!(lower_size_bound <= core::isize::MAX as usize); // Ensure offset is indexable
1423
0
        assert!(index + lower_size_bound >= index); // Protect against overflow
1424
1425
0
        let mut num_added = 0;
1426
0
        let old_len = self.len();
1427
0
        assert!(index <= old_len);
1428
1429
        unsafe {
1430
            // Reserve space for `lower_size_bound` elements.
1431
0
            self.reserve(lower_size_bound);
1432
0
            let start = self.as_mut_ptr();
1433
0
            let ptr = start.add(index);
1434
1435
            // Move the trailing elements.
1436
0
            ptr::copy(ptr, ptr.add(lower_size_bound), old_len - index);
1437
1438
            // In case the iterator panics, don't double-drop the items we just copied above.
1439
0
            self.set_len(0);
1440
0
            let mut guard = DropOnPanic {
1441
0
                start,
1442
0
                skip: index..(index + lower_size_bound),
1443
0
                len: old_len + lower_size_bound,
1444
0
            };
1445
1446
            // The set_len above invalidates the previous pointers, so we must re-create them.
1447
0
            let start = self.as_mut_ptr();
1448
0
            let ptr = start.add(index);
1449
1450
0
            while num_added < lower_size_bound {
1451
0
                let element = match iter.next() {
1452
0
                    Some(x) => x,
1453
0
                    None => break,
1454
                };
1455
0
                let cur = ptr.add(num_added);
1456
0
                ptr::write(cur, element);
1457
0
                guard.skip.start += 1;
1458
0
                num_added += 1;
1459
            }
1460
1461
0
            if num_added < lower_size_bound {
1462
0
                // Iterator provided fewer elements than the hint. Move the tail backward.
1463
0
                ptr::copy(
1464
0
                    ptr.add(lower_size_bound),
1465
0
                    ptr.add(num_added),
1466
0
                    old_len - index,
1467
0
                );
1468
0
            }
1469
            // There are no more duplicate or uninitialized slots, so the guard is not needed.
1470
0
            self.set_len(old_len + num_added);
1471
0
            mem::forget(guard);
1472
        }
1473
1474
        // Insert any remaining elements one-by-one.
1475
0
        for element in iter {
1476
0
            self.insert(index + num_added, element);
1477
0
            num_added += 1;
1478
0
        }
1479
1480
        struct DropOnPanic<T> {
1481
            start: *mut T,
1482
            skip: Range<usize>, // Space we copied-out-of, but haven't written-to yet.
1483
            len: usize,
1484
        }
1485
1486
        impl<T> Drop for DropOnPanic<T> {
1487
0
            fn drop(&mut self) {
1488
0
                for i in 0..self.len {
1489
0
                    if !self.skip.contains(&i) {
1490
0
                        unsafe {
1491
0
                            ptr::drop_in_place(self.start.add(i));
1492
0
                        }
1493
0
                    }
1494
                }
1495
0
            }
1496
        }
1497
0
    }
1498
1499
    /// Convert a `SmallVec` to a `Vec`, without reallocating if the `SmallVec` has already spilled onto
1500
    /// the heap.
1501
0
    pub fn into_vec(mut self) -> Vec<A::Item> {
1502
0
        if self.spilled() {
1503
            unsafe {
1504
0
                let (ptr, &mut len) = self.data.heap_mut();
1505
0
                let v = Vec::from_raw_parts(ptr.as_ptr(), len, self.capacity);
1506
0
                mem::forget(self);
1507
0
                v
1508
            }
1509
        } else {
1510
0
            self.into_iter().collect()
1511
        }
1512
0
    }
1513
1514
    /// Converts a `SmallVec` into a `Box<[T]>` without reallocating if the `SmallVec` has already spilled
1515
    /// onto the heap.
1516
    ///
1517
    /// Note that this will drop any excess capacity.
1518
0
    pub fn into_boxed_slice(self) -> Box<[A::Item]> {
1519
0
        self.into_vec().into_boxed_slice()
1520
0
    }
1521
1522
    /// Convert the `SmallVec` into an `A` if possible. Otherwise return `Err(Self)`.
1523
    ///
1524
    /// This method returns `Err(Self)` if the `SmallVec` is too short (and the `A` contains uninitialized elements),
1525
    /// or if the `SmallVec` is too long (and all the elements were spilled to the heap).
1526
0
    pub fn into_inner(self) -> Result<A, Self> {
1527
0
        if self.spilled() || self.len() != A::size() {
1528
            // Note: A::size, not Self::inline_capacity
1529
0
            Err(self)
1530
        } else {
1531
            unsafe {
1532
0
                let data = ptr::read(&self.data);
1533
0
                mem::forget(self);
1534
0
                Ok(data.into_inline().assume_init())
1535
            }
1536
        }
1537
0
    }
1538
1539
    /// Retains only the elements specified by the predicate.
1540
    ///
1541
    /// In other words, remove all elements `e` such that `f(&e)` returns `false`.
1542
    /// This method operates in place and preserves the order of the retained
1543
    /// elements.
1544
0
    pub fn retain<F: FnMut(&mut A::Item) -> bool>(&mut self, mut f: F) {
1545
0
        let mut del = 0;
1546
0
        let len = self.len();
1547
0
        for i in 0..len {
1548
0
            if !f(&mut self[i]) {
1549
0
                del += 1;
1550
0
            } else if del > 0 {
1551
0
                self.swap(i - del, i);
1552
0
            }
1553
        }
1554
0
        self.truncate(len - del);
1555
0
    }
1556
1557
    /// Retains only the elements specified by the predicate.
1558
    ///
1559
    /// This method is identical in behaviour to [`retain`]; it is included only
1560
    /// to maintain api-compatibility with `std::Vec`, where the methods are
1561
    /// separate for historical reasons.
1562
0
    pub fn retain_mut<F: FnMut(&mut A::Item) -> bool>(&mut self, f: F) {
1563
0
        self.retain(f)
1564
0
    }
1565
1566
    /// Removes consecutive duplicate elements.
1567
0
    pub fn dedup(&mut self)
1568
0
    where
1569
0
        A::Item: PartialEq<A::Item>,
1570
    {
1571
0
        self.dedup_by(|a, b| a == b);
1572
0
    }
1573
1574
    /// Removes consecutive duplicate elements using the given equality relation.
1575
0
    pub fn dedup_by<F>(&mut self, mut same_bucket: F)
1576
0
    where
1577
0
        F: FnMut(&mut A::Item, &mut A::Item) -> bool,
1578
    {
1579
        // See the implementation of Vec::dedup_by in the
1580
        // standard library for an explanation of this algorithm.
1581
0
        let len = self.len();
1582
0
        if len <= 1 {
1583
0
            return;
1584
0
        }
1585
1586
0
        let ptr = self.as_mut_ptr();
1587
0
        let mut w: usize = 1;
1588
1589
        unsafe {
1590
0
            for r in 1..len {
1591
0
                let p_r = ptr.add(r);
1592
0
                let p_wm1 = ptr.add(w - 1);
1593
0
                if !same_bucket(&mut *p_r, &mut *p_wm1) {
1594
0
                    if r != w {
1595
0
                        let p_w = p_wm1.add(1);
1596
0
                        mem::swap(&mut *p_r, &mut *p_w);
1597
0
                    }
1598
0
                    w += 1;
1599
0
                }
1600
            }
1601
        }
1602
1603
0
        self.truncate(w);
1604
0
    }
1605
1606
    /// Removes consecutive elements that map to the same key.
1607
0
    pub fn dedup_by_key<F, K>(&mut self, mut key: F)
1608
0
    where
1609
0
        F: FnMut(&mut A::Item) -> K,
1610
0
        K: PartialEq<K>,
1611
    {
1612
0
        self.dedup_by(|a, b| key(a) == key(b));
1613
0
    }
1614
1615
    /// Resizes the `SmallVec` in-place so that `len` is equal to `new_len`.
1616
    ///
1617
    /// If `new_len` is greater than `len`, the `SmallVec` is extended by the difference, with each
1618
    /// additional slot filled with the result of calling the closure `f`. The return values from `f`
1619
    /// will end up in the `SmallVec` in the order they have been generated.
1620
    ///
1621
    /// If `new_len` is less than `len`, the `SmallVec` is simply truncated.
1622
    ///
1623
    /// This method uses a closure to create new values on every push. If you'd rather `Clone` a given
1624
    /// value, use `resize`. If you want to use the `Default` trait to generate values, you can pass
1625
    /// `Default::default()` as the second argument.
1626
    ///
1627
    /// Added for `std::vec::Vec` compatibility (added in Rust 1.33.0)
1628
    ///
1629
    /// ```
1630
    /// # use smallvec::{smallvec, SmallVec};
1631
    /// let mut vec : SmallVec<[_; 4]> = smallvec![1, 2, 3];
1632
    /// vec.resize_with(5, Default::default);
1633
    /// assert_eq!(&*vec, &[1, 2, 3, 0, 0]);
1634
    ///
1635
    /// let mut vec : SmallVec<[_; 4]> = smallvec![];
1636
    /// let mut p = 1;
1637
    /// vec.resize_with(4, || { p *= 2; p });
1638
    /// assert_eq!(&*vec, &[2, 4, 8, 16]);
1639
    /// ```
1640
0
    pub fn resize_with<F>(&mut self, new_len: usize, f: F)
1641
0
    where
1642
0
        F: FnMut() -> A::Item,
1643
    {
1644
0
        let old_len = self.len();
1645
0
        if old_len < new_len {
1646
0
            let mut f = f;
1647
0
            let additional = new_len - old_len;
1648
0
            self.reserve(additional);
1649
0
            for _ in 0..additional {
1650
0
                self.push(f());
1651
0
            }
1652
0
        } else if old_len > new_len {
1653
0
            self.truncate(new_len);
1654
0
        }
1655
0
    }
1656
1657
    /// Creates a `SmallVec` directly from the raw components of another
1658
    /// `SmallVec`.
1659
    ///
1660
    /// # Safety
1661
    ///
1662
    /// This is highly unsafe, due to the number of invariants that aren't
1663
    /// checked:
1664
    ///
1665
    /// * `ptr` needs to have been previously allocated via `SmallVec` for its
1666
    ///   spilled storage (at least, it's highly likely to be incorrect if it
1667
    ///   wasn't).
1668
    /// * `ptr`'s `A::Item` type needs to be the same size and alignment that
1669
    ///   it was allocated with
1670
    /// * `length` needs to be less than or equal to `capacity`.
1671
    /// * `capacity` needs to be the capacity that the pointer was allocated
1672
    ///   with.
1673
    ///
1674
    /// Violating these may cause problems like corrupting the allocator's
1675
    /// internal data structures.
1676
    ///
1677
    /// Additionally, `capacity` must be greater than the amount of inline
1678
    /// storage `A` has; that is, the new `SmallVec` must need to spill over
1679
    /// into heap allocated storage. This condition is asserted against.
1680
    ///
1681
    /// The ownership of `ptr` is effectively transferred to the
1682
    /// `SmallVec` which may then deallocate, reallocate or change the
1683
    /// contents of memory pointed to by the pointer at will. Ensure
1684
    /// that nothing else uses the pointer after calling this
1685
    /// function.
1686
    ///
1687
    /// # Examples
1688
    ///
1689
    /// ```
1690
    /// # use smallvec::{smallvec, SmallVec};
1691
    /// use std::mem;
1692
    /// use std::ptr;
1693
    ///
1694
    /// fn main() {
1695
    ///     let mut v: SmallVec<[_; 1]> = smallvec![1, 2, 3];
1696
    ///
1697
    ///     // Pull out the important parts of `v`.
1698
    ///     let p = v.as_mut_ptr();
1699
    ///     let len = v.len();
1700
    ///     let cap = v.capacity();
1701
    ///     let spilled = v.spilled();
1702
    ///
1703
    ///     unsafe {
1704
    ///         // Forget all about `v`. The heap allocation that stored the
1705
    ///         // three values won't be deallocated.
1706
    ///         mem::forget(v);
1707
    ///
1708
    ///         // Overwrite memory with [4, 5, 6].
1709
    ///         //
1710
    ///         // This is only safe if `spilled` is true! Otherwise, we are
1711
    ///         // writing into the old `SmallVec`'s inline storage on the
1712
    ///         // stack.
1713
    ///         assert!(spilled);
1714
    ///         for i in 0..len {
1715
    ///             ptr::write(p.add(i), 4 + i);
1716
    ///         }
1717
    ///
1718
    ///         // Put everything back together into a SmallVec with a different
1719
    ///         // amount of inline storage, but which is still less than `cap`.
1720
    ///         let rebuilt = SmallVec::<[_; 2]>::from_raw_parts(p, len, cap);
1721
    ///         assert_eq!(&*rebuilt, &[4, 5, 6]);
1722
    ///     }
1723
    /// }
1724
    #[inline]
1725
0
    pub unsafe fn from_raw_parts(ptr: *mut A::Item, length: usize, capacity: usize) -> SmallVec<A> {
1726
        // SAFETY: We require caller to provide same ptr as we alloc
1727
        // and we never alloc null pointer.
1728
0
        let ptr = unsafe {
1729
0
            debug_assert!(!ptr.is_null(), "Called `from_raw_parts` with null pointer.");
1730
0
            NonNull::new_unchecked(ptr)
1731
        };
1732
0
        assert!(capacity > Self::inline_capacity());
1733
0
        SmallVec {
1734
0
            capacity,
1735
0
            data: SmallVecData::from_heap(ptr, length),
1736
0
        }
1737
0
    }
1738
1739
    /// Returns a raw pointer to the vector's buffer.
1740
0
    pub fn as_ptr(&self) -> *const A::Item {
1741
        // We shadow the slice method of the same name to avoid going through
1742
        // `deref`, which creates an intermediate reference that may place
1743
        // additional safety constraints on the contents of the slice.
1744
0
        self.triple().0.as_ptr()
1745
0
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]>>::as_ptr
Unexecuted instantiation: <smallvec::SmallVec<_>>::as_ptr
1746
1747
    /// Returns a raw mutable pointer to the vector's buffer.
1748
0
    pub fn as_mut_ptr(&mut self) -> *mut A::Item {
1749
        // We shadow the slice method of the same name to avoid going through
1750
        // `deref_mut`, which creates an intermediate reference that may place
1751
        // additional safety constraints on the contents of the slice.
1752
0
        self.triple_mut().0.as_ptr()
1753
0
    }
1754
}
1755
1756
impl<A: Array> SmallVec<A>
1757
where
1758
    A::Item: Copy,
1759
{
1760
    /// Copy the elements from a slice into a new `SmallVec`.
1761
    ///
1762
    /// For slices of `Copy` types, this is more efficient than `SmallVec::from(slice)`.
1763
0
    pub fn from_slice(slice: &[A::Item]) -> Self {
1764
0
        let len = slice.len();
1765
0
        if len <= Self::inline_capacity() {
1766
0
            SmallVec {
1767
0
                capacity: len,
1768
0
                data: SmallVecData::from_inline(unsafe {
1769
0
                    let mut data: MaybeUninit<A> = MaybeUninit::uninit();
1770
0
                    ptr::copy_nonoverlapping(
1771
0
                        slice.as_ptr(),
1772
0
                        data.as_mut_ptr() as *mut A::Item,
1773
0
                        len,
1774
0
                    );
1775
0
                    data
1776
0
                }),
1777
0
            }
1778
        } else {
1779
0
            let mut b = slice.to_vec();
1780
0
            let cap = b.capacity();
1781
0
            let ptr = NonNull::new(b.as_mut_ptr()).expect("Vec always contain non null pointers.");
1782
0
            mem::forget(b);
1783
0
            SmallVec {
1784
0
                capacity: cap,
1785
0
                data: SmallVecData::from_heap(ptr, len),
1786
0
            }
1787
        }
1788
0
    }
1789
1790
    /// Copy elements from a slice into the vector at position `index`, shifting any following
1791
    /// elements toward the back.
1792
    ///
1793
    /// For slices of `Copy` types, this is more efficient than `insert`.
1794
    #[inline]
1795
0
    pub fn insert_from_slice(&mut self, index: usize, slice: &[A::Item]) {
1796
0
        self.reserve(slice.len());
1797
1798
0
        let len = self.len();
1799
0
        assert!(index <= len);
1800
1801
0
        unsafe {
1802
0
            let slice_ptr = slice.as_ptr();
1803
0
            let ptr = self.as_mut_ptr().add(index);
1804
0
            ptr::copy(ptr, ptr.add(slice.len()), len - index);
1805
0
            ptr::copy_nonoverlapping(slice_ptr, ptr, slice.len());
1806
0
            self.set_len(len + slice.len());
1807
0
        }
1808
0
    }
1809
1810
    /// Copy elements from a slice and append them to the vector.
1811
    ///
1812
    /// For slices of `Copy` types, this is more efficient than `extend`.
1813
    #[inline]
1814
0
    pub fn extend_from_slice(&mut self, slice: &[A::Item]) {
1815
0
        let len = self.len();
1816
0
        self.insert_from_slice(len, slice);
1817
0
    }
1818
}
1819
1820
impl<A: Array> SmallVec<A>
1821
where
1822
    A::Item: Clone,
1823
{
1824
    /// Resizes the vector so that its length is equal to `len`.
1825
    ///
1826
    /// If `len` is less than the current length, the vector simply truncated.
1827
    ///
1828
    /// If `len` is greater than the current length, `value` is appended to the
1829
    /// vector until its length equals `len`.
1830
0
    pub fn resize(&mut self, len: usize, value: A::Item) {
1831
0
        let old_len = self.len();
1832
1833
0
        if len > old_len {
1834
0
            self.extend(repeat(value).take(len - old_len));
1835
0
        } else {
1836
0
            self.truncate(len);
1837
0
        }
1838
0
    }
1839
1840
    /// Creates a `SmallVec` with `n` copies of `elem`.
1841
    /// ```
1842
    /// use smallvec::SmallVec;
1843
    ///
1844
    /// let v = SmallVec::<[char; 128]>::from_elem('d', 2);
1845
    /// assert_eq!(v, SmallVec::from_buf(['d', 'd']));
1846
    /// ```
1847
0
    pub fn from_elem(elem: A::Item, n: usize) -> Self {
1848
0
        if n > Self::inline_capacity() {
1849
0
            vec![elem; n].into()
1850
        } else {
1851
0
            let mut v = SmallVec::<A>::new();
1852
            unsafe {
1853
0
                let (ptr, len_ptr, _) = v.triple_mut();
1854
0
                let ptr = ptr.as_ptr();
1855
0
                let mut local_len = SetLenOnDrop::new(len_ptr);
1856
1857
0
                for i in 0..n {
1858
0
                    ::core::ptr::write(ptr.add(i), elem.clone());
1859
0
                    local_len.increment_len(1);
1860
0
                }
1861
            }
1862
0
            v
1863
        }
1864
0
    }
1865
}
1866
1867
impl<A: Array> ops::Deref for SmallVec<A> {
1868
    type Target = [A::Item];
1869
    #[inline]
1870
8.47k
    fn deref(&self) -> &[A::Item] {
1871
        unsafe {
1872
8.47k
            let (ptr, len, _) = self.triple();
1873
8.47k
            slice::from_raw_parts(ptr.as_ptr(), len)
1874
        }
1875
8.47k
    }
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]> as core::ops::deref::Deref>::deref
<smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]> as core::ops::deref::Deref>::deref
Line
Count
Source
1870
8.47k
    fn deref(&self) -> &[A::Item] {
1871
        unsafe {
1872
8.47k
            let (ptr, len, _) = self.triple();
1873
8.47k
            slice::from_raw_parts(ptr.as_ptr(), len)
1874
        }
1875
8.47k
    }
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]> as core::ops::deref::Deref>::deref
Unexecuted instantiation: <smallvec::SmallVec<_> as core::ops::deref::Deref>::deref
1876
}
1877
1878
impl<A: Array> ops::DerefMut for SmallVec<A> {
1879
    #[inline]
1880
2.94k
    fn deref_mut(&mut self) -> &mut [A::Item] {
1881
2.94k
        unsafe {
1882
2.94k
            let (ptr, &mut len, _) = self.triple_mut();
1883
2.94k
            slice::from_raw_parts_mut(ptr.as_ptr(), len)
1884
2.94k
        }
1885
2.94k
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]> as core::ops::deref::DerefMut>::deref_mut
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]> as core::ops::deref::DerefMut>::deref_mut
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]> as core::ops::deref::DerefMut>::deref_mut
<smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]> as core::ops::deref::DerefMut>::deref_mut
Line
Count
Source
1880
2.94k
    fn deref_mut(&mut self) -> &mut [A::Item] {
1881
2.94k
        unsafe {
1882
2.94k
            let (ptr, &mut len, _) = self.triple_mut();
1883
2.94k
            slice::from_raw_parts_mut(ptr.as_ptr(), len)
1884
2.94k
        }
1885
2.94k
    }
Unexecuted instantiation: <smallvec::SmallVec<_> as core::ops::deref::DerefMut>::deref_mut
1886
}
1887
1888
impl<A: Array> AsRef<[A::Item]> for SmallVec<A> {
1889
    #[inline]
1890
0
    fn as_ref(&self) -> &[A::Item] {
1891
0
        self
1892
0
    }
1893
}
1894
1895
impl<A: Array> AsMut<[A::Item]> for SmallVec<A> {
1896
    #[inline]
1897
0
    fn as_mut(&mut self) -> &mut [A::Item] {
1898
0
        self
1899
0
    }
1900
}
1901
1902
impl<A: Array> Borrow<[A::Item]> for SmallVec<A> {
1903
    #[inline]
1904
0
    fn borrow(&self) -> &[A::Item] {
1905
0
        self
1906
0
    }
1907
}
1908
1909
impl<A: Array> BorrowMut<[A::Item]> for SmallVec<A> {
1910
    #[inline]
1911
0
    fn borrow_mut(&mut self) -> &mut [A::Item] {
1912
0
        self
1913
0
    }
1914
}
1915
1916
#[cfg(feature = "write")]
1917
#[cfg_attr(docsrs, doc(cfg(feature = "write")))]
1918
impl<A: Array<Item = u8>> io::Write for SmallVec<A> {
1919
    #[inline]
1920
    fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
1921
        self.extend_from_slice(buf);
1922
        Ok(buf.len())
1923
    }
1924
1925
    #[inline]
1926
    fn write_all(&mut self, buf: &[u8]) -> io::Result<()> {
1927
        self.extend_from_slice(buf);
1928
        Ok(())
1929
    }
1930
1931
    #[inline]
1932
    fn flush(&mut self) -> io::Result<()> {
1933
        Ok(())
1934
    }
1935
}
1936
1937
#[cfg(feature = "serde")]
1938
#[cfg_attr(docsrs, doc(cfg(feature = "serde")))]
1939
impl<A: Array> Serialize for SmallVec<A>
1940
where
1941
    A::Item: Serialize,
1942
{
1943
    fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
1944
        let mut state = serializer.serialize_seq(Some(self.len()))?;
1945
        for item in self {
1946
            state.serialize_element(&item)?;
1947
        }
1948
        state.end()
1949
    }
1950
}
1951
1952
#[cfg(feature = "serde")]
1953
#[cfg_attr(docsrs, doc(cfg(feature = "serde")))]
1954
impl<'de, A: Array> Deserialize<'de> for SmallVec<A>
1955
where
1956
    A::Item: Deserialize<'de>,
1957
{
1958
    fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
1959
        deserializer.deserialize_seq(SmallVecVisitor {
1960
            phantom: PhantomData,
1961
        })
1962
    }
1963
}
1964
1965
#[cfg(feature = "serde")]
1966
struct SmallVecVisitor<A> {
1967
    phantom: PhantomData<A>,
1968
}
1969
1970
#[cfg(feature = "serde")]
1971
impl<'de, A: Array> Visitor<'de> for SmallVecVisitor<A>
1972
where
1973
    A::Item: Deserialize<'de>,
1974
{
1975
    type Value = SmallVec<A>;
1976
1977
    fn expecting(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
1978
        formatter.write_str("a sequence")
1979
    }
1980
1981
    fn visit_seq<B>(self, mut seq: B) -> Result<Self::Value, B::Error>
1982
    where
1983
        B: SeqAccess<'de>,
1984
    {
1985
        use serde::de::Error;
1986
        let len = seq.size_hint().unwrap_or(0);
1987
        let mut values = SmallVec::new();
1988
        values.try_reserve(len).map_err(B::Error::custom)?;
1989
1990
        while let Some(value) = seq.next_element()? {
1991
            values.push(value);
1992
        }
1993
1994
        Ok(values)
1995
    }
1996
}
1997
1998
#[cfg(feature = "malloc_size_of")]
1999
impl<A: Array> MallocShallowSizeOf for SmallVec<A> {
2000
    fn shallow_size_of(&self, ops: &mut MallocSizeOfOps) -> usize {
2001
        if self.spilled() {
2002
            unsafe { ops.malloc_size_of(self.as_ptr()) }
2003
        } else {
2004
            0
2005
        }
2006
    }
2007
}
2008
2009
#[cfg(feature = "malloc_size_of")]
2010
impl<A> MallocSizeOf for SmallVec<A>
2011
where
2012
    A: Array,
2013
    A::Item: MallocSizeOf,
2014
{
2015
    fn size_of(&self, ops: &mut MallocSizeOfOps) -> usize {
2016
        let mut n = self.shallow_size_of(ops);
2017
        for elem in self.iter() {
2018
            n += elem.size_of(ops);
2019
        }
2020
        n
2021
    }
2022
}
2023
2024
#[cfg(feature = "specialization")]
2025
trait SpecFrom<A: Array, S> {
2026
    fn spec_from(slice: S) -> SmallVec<A>;
2027
}
2028
2029
#[cfg(feature = "specialization")]
2030
mod specialization;
2031
2032
#[cfg(feature = "arbitrary")]
2033
mod arbitrary;
2034
2035
#[cfg(feature = "specialization")]
2036
impl<'a, A: Array> SpecFrom<A, &'a [A::Item]> for SmallVec<A>
2037
where
2038
    A::Item: Copy,
2039
{
2040
    #[inline]
2041
    fn spec_from(slice: &'a [A::Item]) -> SmallVec<A> {
2042
        SmallVec::from_slice(slice)
2043
    }
2044
}
2045
2046
impl<'a, A: Array> From<&'a [A::Item]> for SmallVec<A>
2047
where
2048
    A::Item: Clone,
2049
{
2050
    #[cfg(not(feature = "specialization"))]
2051
    #[inline]
2052
0
    fn from(slice: &'a [A::Item]) -> SmallVec<A> {
2053
0
        slice.iter().cloned().collect()
2054
0
    }
2055
2056
    #[cfg(feature = "specialization")]
2057
    #[inline]
2058
    fn from(slice: &'a [A::Item]) -> SmallVec<A> {
2059
        SmallVec::spec_from(slice)
2060
    }
2061
}
2062
2063
impl<A: Array> From<Vec<A::Item>> for SmallVec<A> {
2064
    #[inline]
2065
0
    fn from(vec: Vec<A::Item>) -> SmallVec<A> {
2066
0
        SmallVec::from_vec(vec)
2067
0
    }
2068
}
2069
2070
impl<A: Array> From<A> for SmallVec<A> {
2071
    #[inline]
2072
0
    fn from(array: A) -> SmallVec<A> {
2073
0
        SmallVec::from_buf(array)
2074
0
    }
2075
}
2076
2077
impl<A: Array, I: SliceIndex<[A::Item]>> ops::Index<I> for SmallVec<A> {
2078
    type Output = I::Output;
2079
2080
0
    fn index(&self, index: I) -> &I::Output {
2081
0
        &(**self)[index]
2082
0
    }
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]> as core::ops::index::Index<usize>>::index
Unexecuted instantiation: <smallvec::SmallVec<_> as core::ops::index::Index<_>>::index
2083
}
2084
2085
impl<A: Array, I: SliceIndex<[A::Item]>> ops::IndexMut<I> for SmallVec<A> {
2086
2.94k
    fn index_mut(&mut self, index: I) -> &mut I::Output {
2087
2.94k
        &mut (&mut **self)[index]
2088
2.94k
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]> as core::ops::index::IndexMut<core::ops::range::RangeFull>>::index_mut
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]> as core::ops::index::IndexMut<core::ops::range::RangeFull>>::index_mut
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]> as core::ops::index::IndexMut<core::ops::range::RangeFull>>::index_mut
<smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]> as core::ops::index::IndexMut<core::ops::range::RangeFull>>::index_mut
Line
Count
Source
2086
2.94k
    fn index_mut(&mut self, index: I) -> &mut I::Output {
2087
2.94k
        &mut (&mut **self)[index]
2088
2.94k
    }
Unexecuted instantiation: <smallvec::SmallVec<_> as core::ops::index::IndexMut<_>>::index_mut
2089
}
2090
2091
#[allow(deprecated)]
2092
impl<A: Array> ExtendFromSlice<A::Item> for SmallVec<A>
2093
where
2094
    A::Item: Copy,
2095
{
2096
0
    fn extend_from_slice(&mut self, other: &[A::Item]) {
2097
0
        SmallVec::extend_from_slice(self, other)
2098
0
    }
2099
}
2100
2101
impl<A: Array> FromIterator<A::Item> for SmallVec<A> {
2102
    #[inline]
2103
2.94k
    fn from_iter<I: IntoIterator<Item = A::Item>>(iterable: I) -> SmallVec<A> {
2104
2.94k
        let mut v = SmallVec::new();
2105
2.94k
        v.extend(iterable);
2106
2.94k
        v
2107
2.94k
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]> as core::iter::traits::collect::FromIterator<core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>>>::from_iter::<vm_virtio::checked_descriptor::CheckedDescriptorIter<vm_memory::atomic::GuestMemoryLoadGuard<vm_memory::region::GuestRegionCollection<vm_memory::mmap::GuestRegionMmap<vm_memory::bitmap::backend::atomic_bitmap::AtomicBitmap>>>>>
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]> as core::iter::traits::collect::FromIterator<vm_virtio::checked_descriptor::CheckedDescriptor>>::from_iter::<core::iter::adapters::GenericShunt<vm_virtio::checked_descriptor::CheckedDescriptorIter<vm_memory::atomic::GuestMemoryLoadGuard<vm_memory::region::GuestRegionCollection<vm_memory::mmap::GuestRegionMmap<vm_memory::bitmap::backend::atomic_bitmap::AtomicBitmap>>>>, core::result::Result<core::convert::Infallible, vm_memory::guest_memory::GuestAddress>>>
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]> as core::iter::traits::collect::FromIterator<(vm_memory::guest_memory::GuestAddress, usize)>>::from_iter::<core::iter::adapters::GenericShunt<core::iter::adapters::map::Map<core::iter::adapters::filter::Filter<core::iter::adapters::copied::Copied<core::slice::iter::Iter<(vm_memory::guest_memory::GuestAddress, u32)>>, <block::io_impl::async_io::guest_memory_target::GuestMemoryTarget>::new<vm_memory::bitmap::backend::atomic_bitmap::AtomicBitmap>::{closure#0}>, <block::io_impl::async_io::guest_memory_target::GuestMemoryTarget>::new<vm_memory::bitmap::backend::atomic_bitmap::AtomicBitmap>::{closure#1}>, core::result::Result<core::convert::Infallible, vm_memory::guest_memory::Error>>>
Unexecuted instantiation: <smallvec::SmallVec<_> as core::iter::traits::collect::FromIterator<<_ as smallvec::Array>::Item>>::from_iter::<_>
<smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]> as core::iter::traits::collect::FromIterator<(vm_memory::guest_memory::GuestAddress, usize)>>::from_iter::<core::iter::adapters::GenericShunt<core::iter::adapters::map::Map<core::iter::adapters::filter::Filter<core::iter::adapters::copied::Copied<core::slice::iter::Iter<(vm_memory::guest_memory::GuestAddress, u32)>>, <block::io_impl::async_io::guest_memory_target::GuestMemoryTarget>::new<()>::{closure#0}>, <block::io_impl::async_io::guest_memory_target::GuestMemoryTarget>::new<()>::{closure#1}>, core::result::Result<core::convert::Infallible, vm_memory::guest_memory::Error>>>
Line
Count
Source
2103
2.94k
    fn from_iter<I: IntoIterator<Item = A::Item>>(iterable: I) -> SmallVec<A> {
2104
2.94k
        let mut v = SmallVec::new();
2105
2.94k
        v.extend(iterable);
2106
2.94k
        v
2107
2.94k
    }
2108
}
2109
2110
impl<A: Array> Extend<A::Item> for SmallVec<A> {
2111
2.94k
    fn extend<I: IntoIterator<Item = A::Item>>(&mut self, iterable: I) {
2112
2.94k
        let mut iter = iterable.into_iter();
2113
2.94k
        let (lower_size_bound, _) = iter.size_hint();
2114
2.94k
        self.reserve(lower_size_bound);
2115
2116
        unsafe {
2117
2.94k
            let (ptr, len_ptr, cap) = self.triple_mut();
2118
2.94k
            let ptr = ptr.as_ptr();
2119
2.94k
            let mut len = SetLenOnDrop::new(len_ptr);
2120
5.88k
            while len.get() < cap {
2121
2.94k
                if let Some(out) = iter.next() {
2122
2.94k
                    ptr::write(ptr.add(len.get()), out);
2123
2.94k
                    len.increment_len(1);
2124
2.94k
                } else {
2125
0
                    return;
2126
                }
2127
            }
2128
        }
2129
2130
2.94k
        for elem in iter {
2131
0
            self.push(elem);
2132
0
        }
2133
2.94k
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]> as core::iter::traits::collect::Extend<core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>>>::extend::<vm_virtio::checked_descriptor::CheckedDescriptorIter<vm_memory::atomic::GuestMemoryLoadGuard<vm_memory::region::GuestRegionCollection<vm_memory::mmap::GuestRegionMmap<vm_memory::bitmap::backend::atomic_bitmap::AtomicBitmap>>>>>
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]> as core::iter::traits::collect::Extend<vm_virtio::checked_descriptor::CheckedDescriptor>>::extend::<core::iter::adapters::GenericShunt<vm_virtio::checked_descriptor::CheckedDescriptorIter<vm_memory::atomic::GuestMemoryLoadGuard<vm_memory::region::GuestRegionCollection<vm_memory::mmap::GuestRegionMmap<vm_memory::bitmap::backend::atomic_bitmap::AtomicBitmap>>>>, core::result::Result<core::convert::Infallible, vm_memory::guest_memory::GuestAddress>>>
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]> as core::iter::traits::collect::Extend<(vm_memory::guest_memory::GuestAddress, usize)>>::extend::<core::iter::adapters::GenericShunt<core::iter::adapters::map::Map<core::iter::adapters::filter::Filter<core::iter::adapters::copied::Copied<core::slice::iter::Iter<(vm_memory::guest_memory::GuestAddress, u32)>>, <block::io_impl::async_io::guest_memory_target::GuestMemoryTarget>::new<vm_memory::bitmap::backend::atomic_bitmap::AtomicBitmap>::{closure#0}>, <block::io_impl::async_io::guest_memory_target::GuestMemoryTarget>::new<vm_memory::bitmap::backend::atomic_bitmap::AtomicBitmap>::{closure#1}>, core::result::Result<core::convert::Infallible, vm_memory::guest_memory::Error>>>
Unexecuted instantiation: <smallvec::SmallVec<_> as core::iter::traits::collect::Extend<<_ as smallvec::Array>::Item>>::extend::<_>
<smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]> as core::iter::traits::collect::Extend<(vm_memory::guest_memory::GuestAddress, usize)>>::extend::<core::iter::adapters::GenericShunt<core::iter::adapters::map::Map<core::iter::adapters::filter::Filter<core::iter::adapters::copied::Copied<core::slice::iter::Iter<(vm_memory::guest_memory::GuestAddress, u32)>>, <block::io_impl::async_io::guest_memory_target::GuestMemoryTarget>::new<()>::{closure#0}>, <block::io_impl::async_io::guest_memory_target::GuestMemoryTarget>::new<()>::{closure#1}>, core::result::Result<core::convert::Infallible, vm_memory::guest_memory::Error>>>
Line
Count
Source
2111
2.94k
    fn extend<I: IntoIterator<Item = A::Item>>(&mut self, iterable: I) {
2112
2.94k
        let mut iter = iterable.into_iter();
2113
2.94k
        let (lower_size_bound, _) = iter.size_hint();
2114
2.94k
        self.reserve(lower_size_bound);
2115
2116
        unsafe {
2117
2.94k
            let (ptr, len_ptr, cap) = self.triple_mut();
2118
2.94k
            let ptr = ptr.as_ptr();
2119
2.94k
            let mut len = SetLenOnDrop::new(len_ptr);
2120
5.88k
            while len.get() < cap {
2121
2.94k
                if let Some(out) = iter.next() {
2122
2.94k
                    ptr::write(ptr.add(len.get()), out);
2123
2.94k
                    len.increment_len(1);
2124
2.94k
                } else {
2125
0
                    return;
2126
                }
2127
            }
2128
        }
2129
2130
2.94k
        for elem in iter {
2131
0
            self.push(elem);
2132
0
        }
2133
2.94k
    }
2134
}
2135
2136
impl<A: Array> fmt::Debug for SmallVec<A>
2137
where
2138
    A::Item: fmt::Debug,
2139
{
2140
0
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2141
0
        f.debug_list().entries(self.iter()).finish()
2142
0
    }
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]> as core::fmt::Debug>::fmt
Unexecuted instantiation: <smallvec::SmallVec<_> as core::fmt::Debug>::fmt
2143
}
2144
2145
impl<A: Array> Default for SmallVec<A> {
2146
    #[inline]
2147
0
    fn default() -> SmallVec<A> {
2148
0
        SmallVec::new()
2149
0
    }
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]> as core::default::Default>::default
Unexecuted instantiation: <smallvec::SmallVec<_> as core::default::Default>::default
2150
}
2151
2152
#[cfg(feature = "may_dangle")]
2153
unsafe impl<#[may_dangle] A: Array> Drop for SmallVec<A> {
2154
    fn drop(&mut self) {
2155
        unsafe {
2156
            if self.spilled() {
2157
                let (ptr, &mut len) = self.data.heap_mut();
2158
                Vec::from_raw_parts(ptr.as_ptr(), len, self.capacity);
2159
            } else {
2160
                ptr::drop_in_place(&mut self[..]);
2161
            }
2162
        }
2163
    }
2164
}
2165
2166
#[cfg(not(feature = "may_dangle"))]
2167
impl<A: Array> Drop for SmallVec<A> {
2168
2.94k
    fn drop(&mut self) {
2169
        unsafe {
2170
2.94k
            if self.spilled() {
2171
0
                let (ptr, &mut len) = self.data.heap_mut();
2172
0
                drop(Vec::from_raw_parts(ptr.as_ptr(), len, self.capacity));
2173
2.94k
            } else {
2174
2.94k
                ptr::drop_in_place(&mut self[..]);
2175
2.94k
            }
2176
        }
2177
2.94k
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]> as core::ops::drop::Drop>::drop
Unexecuted instantiation: <smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]> as core::ops::drop::Drop>::drop
Unexecuted instantiation: <smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]> as core::ops::drop::Drop>::drop
<smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, usize); 1]> as core::ops::drop::Drop>::drop
Line
Count
Source
2168
2.94k
    fn drop(&mut self) {
2169
        unsafe {
2170
2.94k
            if self.spilled() {
2171
0
                let (ptr, &mut len) = self.data.heap_mut();
2172
0
                drop(Vec::from_raw_parts(ptr.as_ptr(), len, self.capacity));
2173
2.94k
            } else {
2174
2.94k
                ptr::drop_in_place(&mut self[..]);
2175
2.94k
            }
2176
        }
2177
2.94k
    }
Unexecuted instantiation: <smallvec::SmallVec<_> as core::ops::drop::Drop>::drop
2178
}
2179
2180
impl<A: Array> Clone for SmallVec<A>
2181
where
2182
    A::Item: Clone,
2183
{
2184
    #[inline]
2185
0
    fn clone(&self) -> SmallVec<A> {
2186
0
        SmallVec::from(self.as_slice())
2187
0
    }
2188
2189
0
    fn clone_from(&mut self, source: &Self) {
2190
        // Inspired from `impl Clone for Vec`.
2191
2192
        // drop anything that will not be overwritten
2193
0
        self.truncate(source.len());
2194
2195
        // self.len <= other.len due to the truncate above, so the
2196
        // slices here are always in-bounds.
2197
0
        let (init, tail) = source.split_at(self.len());
2198
2199
        // reuse the contained values' allocations/resources.
2200
0
        self.clone_from_slice(init);
2201
0
        self.extend(tail.iter().cloned());
2202
0
    }
2203
}
2204
2205
impl<A: Array, B: Array> PartialEq<SmallVec<B>> for SmallVec<A>
2206
where
2207
    A::Item: PartialEq<B::Item>,
2208
{
2209
    #[inline]
2210
0
    fn eq(&self, other: &SmallVec<B>) -> bool {
2211
0
        self[..] == other[..]
2212
0
    }
2213
}
2214
2215
impl<A: Array> Eq for SmallVec<A> where A::Item: Eq {}
2216
2217
impl<A: Array> PartialOrd for SmallVec<A>
2218
where
2219
    A::Item: PartialOrd,
2220
{
2221
    #[inline]
2222
0
    fn partial_cmp(&self, other: &SmallVec<A>) -> Option<cmp::Ordering> {
2223
0
        PartialOrd::partial_cmp(&**self, &**other)
2224
0
    }
2225
}
2226
2227
impl<A: Array> Ord for SmallVec<A>
2228
where
2229
    A::Item: Ord,
2230
{
2231
    #[inline]
2232
0
    fn cmp(&self, other: &SmallVec<A>) -> cmp::Ordering {
2233
0
        Ord::cmp(&**self, &**other)
2234
0
    }
2235
}
2236
2237
impl<A: Array> Hash for SmallVec<A>
2238
where
2239
    A::Item: Hash,
2240
{
2241
0
    fn hash<H: Hasher>(&self, state: &mut H) {
2242
0
        (**self).hash(state)
2243
0
    }
2244
}
2245
2246
unsafe impl<A: Array> Send for SmallVec<A> where A::Item: Send {}
2247
2248
/// An iterator that consumes a `SmallVec` and yields its items by value.
2249
///
2250
/// Returned from [`SmallVec::into_iter`][1].
2251
///
2252
/// [1]: struct.SmallVec.html#method.into_iter
2253
pub struct IntoIter<A: Array> {
2254
    data: SmallVec<A>,
2255
    current: usize,
2256
    end: usize,
2257
}
2258
2259
impl<A: Array> fmt::Debug for IntoIter<A>
2260
where
2261
    A::Item: fmt::Debug,
2262
{
2263
0
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2264
0
        f.debug_tuple("IntoIter").field(&self.as_slice()).finish()
2265
0
    }
2266
}
2267
2268
impl<A: Array + Clone> Clone for IntoIter<A>
2269
where
2270
    A::Item: Clone,
2271
{
2272
0
    fn clone(&self) -> IntoIter<A> {
2273
0
        SmallVec::from(self.as_slice()).into_iter()
2274
0
    }
2275
}
2276
2277
impl<A: Array> Drop for IntoIter<A> {
2278
0
    fn drop(&mut self) {
2279
0
        for _ in self {}
2280
0
    }
Unexecuted instantiation: <smallvec::IntoIter<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]> as core::ops::drop::Drop>::drop
Unexecuted instantiation: <smallvec::IntoIter<_> as core::ops::drop::Drop>::drop
2281
}
2282
2283
impl<A: Array> Iterator for IntoIter<A> {
2284
    type Item = A::Item;
2285
2286
    #[inline]
2287
0
    fn next(&mut self) -> Option<A::Item> {
2288
0
        if self.current == self.end {
2289
0
            None
2290
        } else {
2291
            unsafe {
2292
0
                let current = self.current;
2293
0
                self.current += 1;
2294
0
                Some(ptr::read(self.data.as_ptr().add(current)))
2295
            }
2296
        }
2297
0
    }
Unexecuted instantiation: <smallvec::IntoIter<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]> as core::iter::traits::iterator::Iterator>::next
Unexecuted instantiation: <smallvec::IntoIter<_> as core::iter::traits::iterator::Iterator>::next
2298
2299
    #[inline]
2300
0
    fn size_hint(&self) -> (usize, Option<usize>) {
2301
0
        let size = self.end - self.current;
2302
0
        (size, Some(size))
2303
0
    }
2304
}
2305
2306
impl<A: Array> DoubleEndedIterator for IntoIter<A> {
2307
    #[inline]
2308
0
    fn next_back(&mut self) -> Option<A::Item> {
2309
0
        if self.current == self.end {
2310
0
            None
2311
        } else {
2312
            unsafe {
2313
0
                self.end -= 1;
2314
0
                Some(ptr::read(self.data.as_ptr().add(self.end)))
2315
            }
2316
        }
2317
0
    }
2318
}
2319
2320
impl<A: Array> ExactSizeIterator for IntoIter<A> {}
2321
impl<A: Array> FusedIterator for IntoIter<A> {}
2322
2323
impl<A: Array> IntoIter<A> {
2324
    /// Returns the remaining items of this iterator as a slice.
2325
0
    pub fn as_slice(&self) -> &[A::Item] {
2326
0
        let len = self.end - self.current;
2327
0
        unsafe { core::slice::from_raw_parts(self.data.as_ptr().add(self.current), len) }
2328
0
    }
2329
2330
    /// Returns the remaining items of this iterator as a mutable slice.
2331
0
    pub fn as_mut_slice(&mut self) -> &mut [A::Item] {
2332
0
        let len = self.end - self.current;
2333
0
        unsafe { core::slice::from_raw_parts_mut(self.data.as_mut_ptr().add(self.current), len) }
2334
0
    }
2335
}
2336
2337
impl<A: Array> IntoIterator for SmallVec<A> {
2338
    type IntoIter = IntoIter<A>;
2339
    type Item = A::Item;
2340
0
    fn into_iter(mut self) -> Self::IntoIter {
2341
        unsafe {
2342
            // Set SmallVec len to zero as `IntoIter` drop handles dropping of the elements
2343
0
            let len = self.len();
2344
0
            self.set_len(0);
2345
0
            IntoIter {
2346
0
                data: self,
2347
0
                current: 0,
2348
0
                end: len,
2349
0
            }
2350
        }
2351
0
    }
Unexecuted instantiation: <smallvec::SmallVec<[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4]> as core::iter::traits::collect::IntoIterator>::into_iter
Unexecuted instantiation: <smallvec::SmallVec<_> as core::iter::traits::collect::IntoIterator>::into_iter
2352
}
2353
2354
impl<'a, A: Array> IntoIterator for &'a SmallVec<A> {
2355
    type IntoIter = slice::Iter<'a, A::Item>;
2356
    type Item = &'a A::Item;
2357
0
    fn into_iter(self) -> Self::IntoIter {
2358
0
        self.iter()
2359
0
    }
Unexecuted instantiation: <&smallvec::SmallVec<[vm_virtio::checked_descriptor::CheckedDescriptor; 4]> as core::iter::traits::collect::IntoIterator>::into_iter
Unexecuted instantiation: <&smallvec::SmallVec<[(vm_memory::guest_memory::GuestAddress, u32); 32]> as core::iter::traits::collect::IntoIterator>::into_iter
Unexecuted instantiation: <&smallvec::SmallVec<_> as core::iter::traits::collect::IntoIterator>::into_iter
2360
}
2361
2362
impl<'a, A: Array> IntoIterator for &'a mut SmallVec<A> {
2363
    type IntoIter = slice::IterMut<'a, A::Item>;
2364
    type Item = &'a mut A::Item;
2365
0
    fn into_iter(self) -> Self::IntoIter {
2366
0
        self.iter_mut()
2367
0
    }
2368
}
2369
2370
/// Types that can be used as the backing store for a [`SmallVec`].
2371
pub unsafe trait Array {
2372
    /// The type of the array's elements.
2373
    type Item;
2374
    /// Returns the number of items the array can hold.
2375
    fn size() -> usize;
2376
}
2377
2378
/// Set the length of the vec when the `SetLenOnDrop` value goes out of scope.
2379
///
2380
/// Copied from <https://github.com/rust-lang/rust/pull/36355>
2381
struct SetLenOnDrop<'a> {
2382
    len: &'a mut usize,
2383
    local_len: usize,
2384
}
2385
2386
impl<'a> SetLenOnDrop<'a> {
2387
    #[inline]
2388
2.94k
    fn new(len: &'a mut usize) -> Self {
2389
2.94k
        SetLenOnDrop {
2390
2.94k
            local_len: *len,
2391
2.94k
            len,
2392
2.94k
        }
2393
2.94k
    }
Unexecuted instantiation: <smallvec::SetLenOnDrop>::new
Unexecuted instantiation: <smallvec::SetLenOnDrop>::new
<smallvec::SetLenOnDrop>::new
Line
Count
Source
2388
2.94k
    fn new(len: &'a mut usize) -> Self {
2389
2.94k
        SetLenOnDrop {
2390
2.94k
            local_len: *len,
2391
2.94k
            len,
2392
2.94k
        }
2393
2.94k
    }
2394
2395
    #[inline]
2396
8.82k
    fn get(&self) -> usize {
2397
8.82k
        self.local_len
2398
8.82k
    }
Unexecuted instantiation: <smallvec::SetLenOnDrop>::get
Unexecuted instantiation: <smallvec::SetLenOnDrop>::get
<smallvec::SetLenOnDrop>::get
Line
Count
Source
2396
8.82k
    fn get(&self) -> usize {
2397
8.82k
        self.local_len
2398
8.82k
    }
2399
2400
    #[inline]
2401
2.94k
    fn increment_len(&mut self, increment: usize) {
2402
2.94k
        self.local_len += increment;
2403
2.94k
    }
Unexecuted instantiation: <smallvec::SetLenOnDrop>::increment_len
Unexecuted instantiation: <smallvec::SetLenOnDrop>::increment_len
<smallvec::SetLenOnDrop>::increment_len
Line
Count
Source
2401
2.94k
    fn increment_len(&mut self, increment: usize) {
2402
2.94k
        self.local_len += increment;
2403
2.94k
    }
2404
}
2405
2406
impl<'a> Drop for SetLenOnDrop<'a> {
2407
    #[inline]
2408
2.94k
    fn drop(&mut self) {
2409
2.94k
        *self.len = self.local_len;
2410
2.94k
    }
Unexecuted instantiation: <smallvec::SetLenOnDrop as core::ops::drop::Drop>::drop
Unexecuted instantiation: <smallvec::SetLenOnDrop as core::ops::drop::Drop>::drop
<smallvec::SetLenOnDrop as core::ops::drop::Drop>::drop
Line
Count
Source
2408
2.94k
    fn drop(&mut self) {
2409
2.94k
        *self.len = self.local_len;
2410
2.94k
    }
2411
}
2412
2413
#[cfg(feature = "const_new")]
2414
impl<T, const N: usize> SmallVec<[T; N]> {
2415
    /// Construct an empty vector.
2416
    ///
2417
    /// This is a `const` version of [`SmallVec::new`] that is enabled by the feature `const_new`, with the limitation that it only works for arrays.
2418
    #[cfg_attr(docsrs, doc(cfg(feature = "const_new")))]
2419
    #[inline]
2420
    pub const fn new_const() -> Self {
2421
        SmallVec {
2422
            capacity: 0,
2423
            data: SmallVecData::from_const(MaybeUninit::uninit()),
2424
        }
2425
    }
2426
2427
    /// The array passed as an argument is moved to be an inline version of `SmallVec`.
2428
    ///
2429
    /// This is a `const` version of [`SmallVec::from_buf`] that is enabled by the feature `const_new`, with the limitation that it only works for arrays.
2430
    #[cfg_attr(docsrs, doc(cfg(feature = "const_new")))]
2431
    #[inline]
2432
    pub const fn from_const(items: [T; N]) -> Self {
2433
        SmallVec {
2434
            capacity: N,
2435
            data: SmallVecData::from_const(MaybeUninit::new(items)),
2436
        }
2437
    }
2438
2439
    /// Constructs a new `SmallVec` on the stack from an array without
2440
    /// copying elements. Also sets the length. The user is responsible
2441
    /// for ensuring that `len <= N`.
2442
    /// 
2443
    /// This is a `const` version of [`SmallVec::from_buf_and_len_unchecked`] that is enabled by the feature `const_new`, with the limitation that it only works for arrays.
2444
    #[cfg_attr(docsrs, doc(cfg(feature = "const_new")))]
2445
    #[inline]
2446
    pub const unsafe fn from_const_with_len_unchecked(items: [T; N], len: usize) -> Self {
2447
        SmallVec {
2448
            capacity: len,
2449
            data: SmallVecData::from_const(MaybeUninit::new(items)),
2450
        }
2451
    }
2452
}
2453
2454
#[cfg(feature = "const_generics")]
2455
#[cfg_attr(docsrs, doc(cfg(feature = "const_generics")))]
2456
unsafe impl<T, const N: usize> Array for [T; N] {
2457
    type Item = T;
2458
    #[inline]
2459
    fn size() -> usize {
2460
        N
2461
    }
2462
}
2463
2464
#[cfg(not(feature = "const_generics"))]
2465
macro_rules! impl_array(
2466
    ($($size:expr),+) => {
2467
        $(
2468
            unsafe impl<T> Array for [T; $size] {
2469
                type Item = T;
2470
                #[inline]
2471
40.4k
                fn size() -> usize { $size }
Unexecuted instantiation: <[core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>; 4] as smallvec::Array>::size
Unexecuted instantiation: <[vm_virtio::checked_descriptor::CheckedDescriptor; 4] as smallvec::Array>::size
Unexecuted instantiation: <[(vm_memory::guest_memory::GuestAddress, u32); 32] as smallvec::Array>::size
<[(vm_memory::guest_memory::GuestAddress, usize); 1] as smallvec::Array>::size
Line
Count
Source
2471
40.4k
                fn size() -> usize { $size }
Unexecuted instantiation: <[_; 5] as smallvec::Array>::size
Unexecuted instantiation: <[_; 6] as smallvec::Array>::size
Unexecuted instantiation: <[_; 7] as smallvec::Array>::size
Unexecuted instantiation: <[_; 8] as smallvec::Array>::size
Unexecuted instantiation: <[_; 9] as smallvec::Array>::size
Unexecuted instantiation: <[_; 10] as smallvec::Array>::size
Unexecuted instantiation: <[_; 11] as smallvec::Array>::size
Unexecuted instantiation: <[_; 12] as smallvec::Array>::size
Unexecuted instantiation: <[_; 13] as smallvec::Array>::size
Unexecuted instantiation: <[_; 14] as smallvec::Array>::size
Unexecuted instantiation: <[_; 2048] as smallvec::Array>::size
Unexecuted instantiation: <[_; 4096] as smallvec::Array>::size
Unexecuted instantiation: <[_; 8192] as smallvec::Array>::size
Unexecuted instantiation: <[_; 16384] as smallvec::Array>::size
Unexecuted instantiation: <[_; 24576] as smallvec::Array>::size
Unexecuted instantiation: <[_; 32768] as smallvec::Array>::size
Unexecuted instantiation: <[_; 65536] as smallvec::Array>::size
Unexecuted instantiation: <[_; 131072] as smallvec::Array>::size
Unexecuted instantiation: <[_; 262144] as smallvec::Array>::size
Unexecuted instantiation: <[_; 393216] as smallvec::Array>::size
Unexecuted instantiation: <[_; 524288] as smallvec::Array>::size
Unexecuted instantiation: <[_; 1048576] as smallvec::Array>::size
Unexecuted instantiation: <[_; 15] as smallvec::Array>::size
Unexecuted instantiation: <[_; 16] as smallvec::Array>::size
Unexecuted instantiation: <[_; 17] as smallvec::Array>::size
Unexecuted instantiation: <[_; 18] as smallvec::Array>::size
Unexecuted instantiation: <[_; 19] as smallvec::Array>::size
Unexecuted instantiation: <[_; 20] as smallvec::Array>::size
Unexecuted instantiation: <[_; 21] as smallvec::Array>::size
Unexecuted instantiation: <[_; 22] as smallvec::Array>::size
Unexecuted instantiation: <[_; 23] as smallvec::Array>::size
Unexecuted instantiation: <[_; 24] as smallvec::Array>::size
Unexecuted instantiation: <[_; 25] as smallvec::Array>::size
Unexecuted instantiation: <[_; 26] as smallvec::Array>::size
Unexecuted instantiation: <[_; 27] as smallvec::Array>::size
Unexecuted instantiation: <[_; 28] as smallvec::Array>::size
Unexecuted instantiation: <[_; 29] as smallvec::Array>::size
Unexecuted instantiation: <[_; 30] as smallvec::Array>::size
Unexecuted instantiation: <[_; 31] as smallvec::Array>::size
Unexecuted instantiation: <[_; 32] as smallvec::Array>::size
Unexecuted instantiation: <[_; 36] as smallvec::Array>::size
Unexecuted instantiation: <[_; 64] as smallvec::Array>::size
Unexecuted instantiation: <[_; 96] as smallvec::Array>::size
Unexecuted instantiation: <[_; 128] as smallvec::Array>::size
Unexecuted instantiation: <[_; 256] as smallvec::Array>::size
Unexecuted instantiation: <[_; 512] as smallvec::Array>::size
Unexecuted instantiation: <[_; 1024] as smallvec::Array>::size
Unexecuted instantiation: <[_; 1536] as smallvec::Array>::size
Unexecuted instantiation: <[_; 0] as smallvec::Array>::size
Unexecuted instantiation: <[_; 1] as smallvec::Array>::size
Unexecuted instantiation: <[_; 2] as smallvec::Array>::size
Unexecuted instantiation: <[_; 3] as smallvec::Array>::size
Unexecuted instantiation: <[_; 4] as smallvec::Array>::size
2472
            }
2473
        )+
2474
    }
2475
);
2476
2477
#[cfg(not(feature = "const_generics"))]
2478
impl_array!(
2479
    0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25,
2480
    26, 27, 28, 29, 30, 31, 32, 36, 0x40, 0x60, 0x80, 0x100, 0x200, 0x400, 0x600, 0x800, 0x1000,
2481
    0x2000, 0x4000, 0x6000, 0x8000, 0x10000, 0x20000, 0x40000, 0x60000, 0x80000, 0x10_0000
2482
);
2483
2484
/// Convenience trait for constructing a `SmallVec`
2485
pub trait ToSmallVec<A: Array> {
2486
    /// Construct a new `SmallVec` from a slice.
2487
    fn to_smallvec(&self) -> SmallVec<A>;
2488
}
2489
2490
impl<A: Array> ToSmallVec<A> for [A::Item]
2491
where
2492
    A::Item: Copy,
2493
{
2494
    #[inline]
2495
0
    fn to_smallvec(&self) -> SmallVec<A> {
2496
0
        SmallVec::from_slice(self)
2497
0
    }
2498
}
2499
2500
// Immutable counterpart for `NonNull<T>`.
2501
#[repr(transparent)]
2502
struct ConstNonNull<T>(NonNull<T>);
2503
2504
impl<T> ConstNonNull<T> {
2505
    #[inline]
2506
8.47k
    fn new(ptr: *const T) -> Option<Self> {
2507
8.47k
        NonNull::new(ptr as *mut T).map(Self)
2508
8.47k
    }
Unexecuted instantiation: <smallvec::ConstNonNull<core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>>>::new
Unexecuted instantiation: <smallvec::ConstNonNull<vm_virtio::checked_descriptor::CheckedDescriptor>>::new
<smallvec::ConstNonNull<(vm_memory::guest_memory::GuestAddress, usize)>>::new
Line
Count
Source
2506
8.47k
    fn new(ptr: *const T) -> Option<Self> {
2507
8.47k
        NonNull::new(ptr as *mut T).map(Self)
2508
8.47k
    }
Unexecuted instantiation: <smallvec::ConstNonNull<(vm_memory::guest_memory::GuestAddress, u32)>>::new
Unexecuted instantiation: <smallvec::ConstNonNull<_>>::new
2509
    #[inline]
2510
8.47k
    fn as_ptr(self) -> *const T {
2511
8.47k
        self.0.as_ptr()
2512
8.47k
    }
Unexecuted instantiation: <smallvec::ConstNonNull<core::result::Result<vm_virtio::checked_descriptor::CheckedDescriptor, vm_memory::guest_memory::GuestAddress>>>::as_ptr
Unexecuted instantiation: <smallvec::ConstNonNull<vm_virtio::checked_descriptor::CheckedDescriptor>>::as_ptr
<smallvec::ConstNonNull<(vm_memory::guest_memory::GuestAddress, usize)>>::as_ptr
Line
Count
Source
2510
8.47k
    fn as_ptr(self) -> *const T {
2511
8.47k
        self.0.as_ptr()
2512
8.47k
    }
Unexecuted instantiation: <smallvec::ConstNonNull<(vm_memory::guest_memory::GuestAddress, u32)>>::as_ptr
Unexecuted instantiation: <smallvec::ConstNonNull<_>>::as_ptr
2513
}
2514
2515
impl<T> Clone for ConstNonNull<T> {
2516
    #[inline]
2517
0
    fn clone(&self) -> Self {
2518
0
        *self
2519
0
    }
2520
}
2521
2522
impl<T> Copy for ConstNonNull<T> {}
2523
2524
#[cfg(feature = "impl_bincode")]
2525
use bincode::{
2526
    de::{BorrowDecoder, Decode, Decoder, read::Reader},
2527
    enc::{Encode, Encoder, write::Writer},
2528
    error::{DecodeError, EncodeError},
2529
    BorrowDecode,
2530
};
2531
2532
#[cfg(feature = "impl_bincode")]
2533
impl<A, Context> Decode<Context> for SmallVec<A>
2534
where
2535
    A: Array,
2536
    A::Item: Decode<Context>,
2537
{
2538
    fn decode<D: Decoder<Context = Context>>(decoder: &mut D) -> Result<Self, DecodeError> {
2539
        use core::convert::TryInto;
2540
        let len = u64::decode(decoder)?;
2541
        let len = len.try_into().map_err(|_| DecodeError::OutsideUsizeRange(len))?;
2542
        decoder.claim_container_read::<A::Item>(len)?;
2543
2544
        let mut vec = SmallVec::with_capacity(len);
2545
        if unty::type_equal::<A::Item, u8>() {
2546
            // Initialize the smallvec's buffer.  Note that we need to do this through
2547
            // the raw pointer as we cannot name the type [u8; N] even though A::Item is u8.
2548
            let ptr = vec.as_mut_ptr();
2549
            // SAFETY: A::Item is u8 and the smallvec has been allocated with enough capacity
2550
            unsafe {
2551
                core::ptr::write_bytes(ptr, 0, len);
2552
                vec.set_len(len);
2553
            }
2554
            // Read the data into the smallvec's buffer.
2555
            let slice = vec.as_mut_slice();
2556
            // SAFETY: A::Item is u8
2557
            let slice = unsafe { core::mem::transmute::<&mut [A::Item], &mut [u8]>(slice) };
2558
            decoder.reader().read(slice)?;
2559
        } else {
2560
            for _ in 0..len {
2561
                decoder.unclaim_bytes_read(core::mem::size_of::<A::Item>());
2562
                vec.push(A::Item::decode(decoder)?);
2563
            }
2564
        }
2565
        Ok(vec)
2566
    }
2567
}
2568
2569
#[cfg(feature = "impl_bincode")]
2570
impl<'de, A, Context> BorrowDecode<'de, Context> for SmallVec<A>
2571
where
2572
    A: Array,
2573
    A::Item: BorrowDecode<'de, Context>,
2574
{
2575
    fn borrow_decode<D: BorrowDecoder<'de, Context = Context>>(decoder: &mut D) -> Result<Self, DecodeError> {
2576
        use core::convert::TryInto;
2577
        let len = u64::decode(decoder)?;
2578
        let len = len.try_into().map_err(|_| DecodeError::OutsideUsizeRange(len))?;
2579
        decoder.claim_container_read::<A::Item>(len)?;
2580
2581
        let mut vec = SmallVec::with_capacity(len);
2582
        if unty::type_equal::<A::Item, u8>() {
2583
            // Initialize the smallvec's buffer.  Note that we need to do this through
2584
            // the raw pointer as we cannot name the type [u8; N] even though A::Item is u8.
2585
            let ptr = vec.as_mut_ptr();
2586
            // SAFETY: A::Item is u8 and the smallvec has been allocated with enough capacity
2587
            unsafe {
2588
                core::ptr::write_bytes(ptr, 0, len);
2589
                vec.set_len(len);
2590
            }
2591
            // Read the data into the smallvec's buffer.
2592
            let slice = vec.as_mut_slice();
2593
            // SAFETY: A::Item is u8
2594
            let slice = unsafe { core::mem::transmute::<&mut [A::Item], &mut [u8]>(slice) };
2595
            decoder.reader().read(slice)?;
2596
        } else {
2597
            for _ in 0..len {
2598
                decoder.unclaim_bytes_read(core::mem::size_of::<A::Item>());
2599
                vec.push(A::Item::borrow_decode(decoder)?);
2600
            }
2601
        }
2602
        Ok(vec)
2603
    }
2604
}
2605
2606
#[cfg(feature = "impl_bincode")]
2607
impl<A> Encode for SmallVec<A>
2608
where
2609
    A: Array,
2610
    A::Item: Encode,
2611
{
2612
    fn encode<E: Encoder>(&self, encoder: &mut E) -> Result<(), EncodeError> {
2613
        (self.len() as u64).encode(encoder)?;
2614
        if unty::type_equal::<A::Item, u8>() {
2615
            // Safety: A::Item is u8
2616
            let slice: &[u8] = unsafe { core::mem::transmute(self.as_slice()) };
2617
            encoder.writer().write(slice)?;
2618
        } else {
2619
            for item in self.iter() {
2620
                item.encode(encoder)?;
2621
            }
2622
        }
2623
        Ok(())
2624
    }
2625
}