Skip to main content

zerocopy/
wrappers.rs

1// SPDX-License-Identifier: BSD-2-Clause OR Apache-2.0 OR MIT
2//
3// Copyright 2023 The Fuchsia Authors
4//
5// Licensed under a BSD-style license <LICENSE-BSD>, Apache License, Version 2.0
6// <LICENSE-APACHE or https://www.apache.org/licenses/LICENSE-2.0>, or the MIT
7// license <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your option.
8// This file may not be copied, modified, or distributed except according to
9// those terms.
10
11use core::{borrow::Borrow, fmt, hash::Hash};
12
13use super::*;
14use crate::pointer::{invariant::Safe, SizeEq, TransmuteFrom};
15
16/// A type with no alignment requirement.
17///
18/// An `Unalign` wraps a `T`, removing any alignment requirement. `Unalign<T>`
19/// has the same size and bit validity as `T`, but not necessarily the same
20/// alignment [or ABI]. This is useful if a type with an alignment requirement
21/// needs to be read from a chunk of memory which provides no alignment
22/// guarantees.
23///
24/// Since `Unalign` has no alignment requirement, the inner `T` may not be
25/// properly aligned in memory. There are five ways to access the inner `T`:
26/// - by value, using [`get`] or [`into_inner`]
27/// - by reference inside of a callback, using [`update`]
28/// - fallibly by reference, using [`try_deref`] or [`try_deref_mut`]; these can
29///   fail if the `Unalign` does not satisfy `T`'s alignment requirement at
30///   runtime
31/// - unsafely by reference, using [`deref_unchecked`] or
32///   [`deref_mut_unchecked`]; it is the caller's responsibility to ensure that
33///   the `Unalign` satisfies `T`'s alignment requirement
34/// - (where `T: Unaligned`) infallibly by reference, using [`Deref::deref`] or
35///   [`DerefMut::deref_mut`]
36///
37/// [or ABI]: https://github.com/google/zerocopy/issues/164
38/// [`get`]: Unalign::get
39/// [`into_inner`]: Unalign::into_inner
40/// [`update`]: Unalign::update
41/// [`try_deref`]: Unalign::try_deref
42/// [`try_deref_mut`]: Unalign::try_deref_mut
43/// [`deref_unchecked`]: Unalign::deref_unchecked
44/// [`deref_mut_unchecked`]: Unalign::deref_mut_unchecked
45///
46/// # Example
47///
48/// In this example, we need `EthernetFrame` to have no alignment requirement -
49/// and thus implement [`Unaligned`]. `EtherType` is `#[repr(u16)]` and so
50/// cannot implement `Unaligned`. We use `Unalign` to relax `EtherType`'s
51/// alignment requirement so that `EthernetFrame` has no alignment requirement
52/// and can implement `Unaligned`.
53///
54/// ```rust
55/// use zerocopy::*;
56/// # use zerocopy_derive::*;
57/// # #[derive(FromBytes, KnownLayout, Immutable, Unaligned)] #[repr(C)] struct Mac([u8; 6]);
58///
59/// # #[derive(PartialEq, Copy, Clone, Debug)]
60/// #[derive(TryFromBytes, KnownLayout, Immutable)]
61/// #[repr(u16)]
62/// enum EtherType {
63///     Ipv4 = 0x0800u16.to_be(),
64///     Arp = 0x0806u16.to_be(),
65///     Ipv6 = 0x86DDu16.to_be(),
66///     # /*
67///     ...
68///     # */
69/// }
70///
71/// #[derive(TryFromBytes, KnownLayout, Immutable, Unaligned)]
72/// #[repr(C)]
73/// struct EthernetFrame {
74///     src: Mac,
75///     dst: Mac,
76///     ethertype: Unalign<EtherType>,
77///     payload: [u8],
78/// }
79///
80/// let bytes = &[
81///     # 0, 1, 2, 3, 4, 5,
82///     # 6, 7, 8, 9, 10, 11,
83///     # /*
84///     ...
85///     # */
86///     0x86, 0xDD,            // EtherType
87///     0xDE, 0xAD, 0xBE, 0xEF // Payload
88/// ][..];
89///
90/// // PANICS: Guaranteed not to panic because `bytes` is of the right
91/// // length, has the right contents, and `EthernetFrame` has no
92/// // alignment requirement.
93/// let packet = EthernetFrame::try_ref_from_bytes(&bytes).unwrap();
94///
95/// assert_eq!(packet.ethertype.get(), EtherType::Ipv6);
96/// assert_eq!(packet.payload, [0xDE, 0xAD, 0xBE, 0xEF]);
97/// ```
98///
99/// # Safety
100///
101/// `Unalign<T>` is guaranteed to have the same size and bit validity as `T`,
102/// and to have [`UnsafeCell`]s covering the same byte ranges as `T`.
103/// `Unalign<T>` is guaranteed to have alignment 1.
104// NOTE: This type is sound to use with types that need to be dropped. The
105// reason is that the compiler-generated drop code automatically moves all
106// values to aligned memory slots before dropping them in-place. This is not
107// well-documented, but it's hinted at in places like [1] and [2]. However, this
108// also means that `T` must be `Sized`; unless something changes, we can never
109// support unsized `T`. [3]
110//
111// [1] https://github.com/rust-lang/rust/issues/54148#issuecomment-420529646
112// [2] https://github.com/google/zerocopy/pull/126#discussion_r1018512323
113// [3] https://github.com/google/zerocopy/issues/209
114#[allow(missing_debug_implementations)]
115#[derive(Default, Copy)]
116#[cfg_attr(any(feature = "derive", test), derive(Immutable, FromBytes, IntoBytes, Unaligned))]
117#[repr(C, packed)]
118pub struct Unalign<T>(T);
119
120// We do not use `derive(KnownLayout)` on `Unalign`, because the derive is not
121// smart enough to realize that `Unalign<T>` is always sized and thus emits a
122// `KnownLayout` impl bounded on `T: KnownLayout.` This is overly restrictive.
123impl_known_layout!(T => Unalign<T>);
124
125// FIXME(https://github.com/rust-lang/rust-clippy/issues/16087): Move these
126// attributes below the comment once this Clippy bug is fixed.
127#[cfg_attr(
128    all(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS, any(feature = "derive", test)),
129    expect(unused_unsafe)
130)]
131#[cfg_attr(
132    all(
133        not(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS),
134        any(feature = "derive", test)
135    ),
136    allow(unused_unsafe)
137)]
138// SAFETY:
139// - `Unalign<T>` promises to have alignment 1, and so we don't require that `T:
140//   Unaligned`.
141// - `Unalign<T>` has the same bit validity as `T`, and so it is `FromZeros`,
142//   `FromBytes`, or `IntoBytes` exactly when `T` is as well.
143// - `Immutable`: `Unalign<T>` has the same fields as `T`, so it permits interior
144//   mutation exactly when `T` does.
145// - `TryFromBytes`: `Unalign<T>` has the same the same bit validity as `T`, so
146//   `T::is_safe` is a sound implementation of `is_safe`.
147//
148#[allow(clippy::multiple_unsafe_ops_per_block)]
149const _: () = unsafe {
150    impl_or_verify!(T => Unaligned for Unalign<T>);
151    impl_or_verify!(T: Immutable => Immutable for Unalign<T>);
152    impl_or_verify!(
153        T: TryFromBytes => TryFromBytes for Unalign<T>;
154        |c| T::is_safe(c.transmute::<_, _, BecauseImmutable>())
155    );
156    impl_or_verify!(T: FromZeros => FromZeros for Unalign<T>);
157    impl_or_verify!(T: FromBytes => FromBytes for Unalign<T>);
158    impl_or_verify!(T: IntoBytes => IntoBytes for Unalign<T>);
159};
160
161// Note that `Unalign: Clone` only if `T: Copy`. Since the inner `T` may not be
162// aligned, there's no way to safely call `T::clone`, and so a `T: Clone` bound
163// is not sufficient to implement `Clone` for `Unalign`.
164impl<T: Copy> Clone for Unalign<T> {
165    #[inline(always)]
166    fn clone(&self) -> Unalign<T> {
167        *self
168    }
169}
170
171impl<T> Unalign<T> {
172    /// Constructs a new `Unalign`.
173    #[inline(always)]
174    pub const fn new(val: T) -> Unalign<T> {
175        Unalign(val)
176    }
177
178    /// Consumes `self`, returning the inner `T`.
179    #[inline(always)]
180    pub const fn into_inner(self) -> T {
181        // SAFETY: Since `Unalign` is `#[repr(C, packed)]`, it has the same size
182        // and bit validity as `T`.
183        //
184        // We do this instead of just destructuring in order to prevent
185        // `Unalign`'s `Drop::drop` from being run, since dropping is not
186        // supported in `const fn`s.
187        //
188        // FIXME(https://github.com/rust-lang/rust/issues/73255): Destructure
189        // instead of using unsafe.
190        unsafe { crate::util::transmute_unchecked(self) }
191    }
192
193    /// Attempts to return a reference to the wrapped `T`, failing if `self` is
194    /// not properly aligned.
195    ///
196    /// If `self` does not satisfy `align_of::<T>()`, then `try_deref` returns
197    /// `Err`.
198    ///
199    /// If `T: Unaligned`, then `Unalign<T>` implements [`Deref`], and callers
200    /// may prefer [`Deref::deref`], which is infallible.
201    #[inline(always)]
202    pub fn try_deref(&self) -> Result<&T, AlignmentError<&Self, T>> {
203        let inner = Ptr::from_ref(self).transmute();
204        match inner.try_into_aligned() {
205            Ok(aligned) => Ok(aligned.as_ref()),
206            Err(err) => Err(err.map_src(
207                #[inline(always)]
208                |src| src.into_unalign().as_ref(),
209            )),
210        }
211    }
212
213    /// Attempts to return a mutable reference to the wrapped `T`, failing if
214    /// `self` is not properly aligned.
215    ///
216    /// If `self` does not satisfy `align_of::<T>()`, then `try_deref` returns
217    /// `Err`.
218    ///
219    /// If `T: Unaligned`, then `Unalign<T>` implements [`DerefMut`], and
220    /// callers may prefer [`DerefMut::deref_mut`], which is infallible.
221    #[inline(always)]
222    pub fn try_deref_mut(&mut self) -> Result<&mut T, AlignmentError<&mut Self, T>> {
223        let inner = Ptr::from_mut(self).transmute::<_, _, (_, (_, _))>();
224        match inner.try_into_aligned() {
225            Ok(aligned) => Ok(aligned.as_mut()),
226            Err(err) => Err(err.map_src(|src| src.into_unalign().as_mut())),
227        }
228    }
229
230    /// Returns a reference to the wrapped `T` without checking alignment.
231    ///
232    /// If `T: Unaligned`, then `Unalign<T>` implements[ `Deref`], and callers
233    /// may prefer [`Deref::deref`], which is safe.
234    ///
235    /// # Safety
236    ///
237    /// The caller must guarantee that `self` satisfies `align_of::<T>()`.
238    #[inline(always)]
239    pub const unsafe fn deref_unchecked(&self) -> &T {
240        // SAFETY: `Unalign<T>` is `repr(transparent)`, so there is a valid `T`
241        // at the same memory location as `self`. It has no alignment guarantee,
242        // but the caller has promised that `self` is properly aligned, so we
243        // know that it is sound to create a reference to `T` at this memory
244        // location.
245        //
246        // We use `mem::transmute` instead of `&*self.get_ptr()` because
247        // dereferencing pointers is not stable in `const` on our current MSRV
248        // (1.56 as of this writing).
249        unsafe { mem::transmute(self) }
250    }
251
252    /// Returns a mutable reference to the wrapped `T` without checking
253    /// alignment.
254    ///
255    /// If `T: Unaligned`, then `Unalign<T>` implements[ `DerefMut`], and
256    /// callers may prefer [`DerefMut::deref_mut`], which is safe.
257    ///
258    /// # Safety
259    ///
260    /// The caller must guarantee that `self` satisfies `align_of::<T>()`.
261    #[inline(always)]
262    pub unsafe fn deref_mut_unchecked(&mut self) -> &mut T {
263        // SAFETY: `self.get_mut_ptr()` returns a raw pointer to a valid `T` at
264        // the same memory location as `self`. It has no alignment guarantee,
265        // but the caller has promised that `self` is properly aligned, so we
266        // know that the pointer itself is aligned, and thus that it is sound to
267        // create a reference to a `T` at this memory location.
268        unsafe { &mut *self.get_mut_ptr() }
269    }
270
271    /// Gets an unaligned raw pointer to the inner `T`.
272    ///
273    /// # Safety
274    ///
275    /// The returned raw pointer is not necessarily aligned to
276    /// `align_of::<T>()`. Most functions which operate on raw pointers require
277    /// those pointers to be aligned, so calling those functions with the result
278    /// of `get_ptr` will result in undefined behavior if alignment is not
279    /// guaranteed using some out-of-band mechanism. In general, the only
280    /// functions which are safe to call with this pointer are those which are
281    /// explicitly documented as being sound to use with an unaligned pointer,
282    /// such as [`read_unaligned`].
283    ///
284    /// Even if the caller is permitted to mutate `self` (e.g. they have
285    /// ownership or a mutable borrow), it is not guaranteed to be sound to
286    /// write through the returned pointer. If writing is required, prefer
287    /// [`get_mut_ptr`] instead.
288    ///
289    /// [`read_unaligned`]: core::ptr::read_unaligned
290    /// [`get_mut_ptr`]: Unalign::get_mut_ptr
291    #[inline(always)]
292    pub const fn get_ptr(&self) -> *const T {
293        ptr::addr_of!(self.0)
294    }
295
296    /// Gets an unaligned mutable raw pointer to the inner `T`.
297    ///
298    /// # Safety
299    ///
300    /// The returned raw pointer is not necessarily aligned to
301    /// `align_of::<T>()`. Most functions which operate on raw pointers require
302    /// those pointers to be aligned, so calling those functions with the result
303    /// of `get_ptr` will result in undefined behavior if alignment is not
304    /// guaranteed using some out-of-band mechanism. In general, the only
305    /// functions which are safe to call with this pointer are those which are
306    /// explicitly documented as being sound to use with an unaligned pointer,
307    /// such as [`read_unaligned`].
308    ///
309    /// [`read_unaligned`]: core::ptr::read_unaligned
310    // FIXME(https://github.com/rust-lang/rust/issues/57349): Make this `const`.
311    #[inline(always)]
312    pub fn get_mut_ptr(&mut self) -> *mut T {
313        ptr::addr_of_mut!(self.0)
314    }
315
316    /// Sets the inner `T`, dropping the previous value.
317    // FIXME(https://github.com/rust-lang/rust/issues/57349): Make this `const`.
318    #[inline(always)]
319    pub fn set(&mut self, t: T) {
320        *self = Unalign::new(t);
321    }
322
323    /// Updates the inner `T` by calling a function on it.
324    ///
325    /// If [`T: Unaligned`], then `Unalign<T>` implements [`DerefMut`], and that
326    /// impl should be preferred over this method when performing updates, as it
327    /// will usually be faster and more ergonomic.
328    ///
329    /// For large types, this method may be expensive, as it requires copying
330    /// `2 * size_of::<T>()` bytes. \[1\]
331    ///
332    /// \[1\] Since the inner `T` may not be aligned, it would not be sound to
333    /// invoke `f` on it directly. Instead, `update` moves it into a
334    /// properly-aligned location in the local stack frame, calls `f` on it, and
335    /// then moves it back to its original location in `self`.
336    ///
337    /// [`T: Unaligned`]: Unaligned
338    #[inline]
339    pub fn update<O, F: FnOnce(&mut T) -> O>(&mut self, f: F) -> O {
340        if mem::align_of::<T>() == 1 {
341            // While we advise callers to use `DerefMut` when `T: Unaligned`,
342            // not all callers will be able to guarantee `T: Unaligned` in all
343            // cases. In particular, callers who are themselves providing an API
344            // which is generic over `T` may sometimes be called by *their*
345            // callers with `T` such that `align_of::<T>() == 1`, but cannot
346            // guarantee this in the general case. Thus, this optimization may
347            // sometimes be helpful.
348
349            // SAFETY: Since `T`'s alignment is 1, `self` satisfies its
350            // alignment by definition.
351            let t = unsafe { self.deref_mut_unchecked() };
352            return f(t);
353        }
354
355        // On drop, this moves `copy` out of itself and uses `ptr::write` to
356        // overwrite `slf`.
357        struct WriteBackOnDrop<T> {
358            copy: ManuallyDrop<T>,
359            slf: *mut Unalign<T>,
360        }
361
362        impl<T> Drop for WriteBackOnDrop<T> {
363            fn drop(&mut self) {
364                // SAFETY: We never use `copy` again as required by
365                // `ManuallyDrop::take`.
366                let copy = unsafe { ManuallyDrop::take(&mut self.copy) };
367                // SAFETY: `slf` is the raw pointer value of `self`. We know it
368                // is valid for writes and properly aligned because `self` is a
369                // mutable reference, which guarantees both of these properties.
370                unsafe { ptr::write(self.slf, Unalign::new(copy)) };
371            }
372        }
373
374        // SAFETY: We know that `self` is valid for reads, properly aligned, and
375        // points to an initialized `Unalign<T>` because it is a mutable
376        // reference, which guarantees all of these properties.
377        //
378        // Since `T: !Copy`, it would be unsound in the general case to allow
379        // both the original `Unalign<T>` and the copy to be used by safe code.
380        // We guarantee that the copy is used to overwrite the original in the
381        // `Drop::drop` impl of `WriteBackOnDrop`. So long as this `drop` is
382        // called before any other safe code executes, soundness is upheld.
383        // While this method can terminate in two ways (by returning normally or
384        // by unwinding due to a panic in `f`), in both cases, `write_back` is
385        // dropped - and its `drop` called - before any other safe code can
386        // execute.
387        let copy = unsafe { ptr::read(self) }.into_inner();
388        let mut write_back = WriteBackOnDrop { copy: ManuallyDrop::new(copy), slf: self };
389
390        let ret = f(&mut write_back.copy);
391
392        drop(write_back);
393        ret
394    }
395}
396
397impl<T: Copy> Unalign<T> {
398    /// Gets a copy of the inner `T`.
399    // FIXME(https://github.com/rust-lang/rust/issues/57349): Make this `const`.
400    #[inline(always)]
401    pub fn get(&self) -> T {
402        let Unalign(val) = *self;
403        val
404    }
405}
406
407impl<T: Unaligned> Deref for Unalign<T> {
408    type Target = T;
409
410    #[inline(always)]
411    fn deref(&self) -> &T {
412        Ptr::from_ref(self).transmute().bikeshed_recall_aligned().as_ref()
413    }
414}
415
416impl<T: Unaligned> DerefMut for Unalign<T> {
417    #[inline(always)]
418    fn deref_mut(&mut self) -> &mut T {
419        Ptr::from_mut(self).transmute::<_, _, (_, (_, _))>().bikeshed_recall_aligned().as_mut()
420    }
421}
422
423impl<T: Unaligned + PartialOrd> PartialOrd<Unalign<T>> for Unalign<T> {
424    #[inline(always)]
425    fn partial_cmp(&self, other: &Unalign<T>) -> Option<Ordering> {
426        PartialOrd::partial_cmp(self.deref(), other.deref())
427    }
428}
429
430impl<T: Unaligned + Ord> Ord for Unalign<T> {
431    #[inline(always)]
432    fn cmp(&self, other: &Unalign<T>) -> Ordering {
433        Ord::cmp(self.deref(), other.deref())
434    }
435}
436
437impl<T: Unaligned + PartialEq> PartialEq<Unalign<T>> for Unalign<T> {
438    #[inline(always)]
439    fn eq(&self, other: &Unalign<T>) -> bool {
440        PartialEq::eq(self.deref(), other.deref())
441    }
442}
443
444impl<T: Unaligned + Eq> Eq for Unalign<T> {}
445
446impl<T: Unaligned + Hash> Hash for Unalign<T> {
447    #[inline(always)]
448    fn hash<H>(&self, state: &mut H)
449    where
450        H: Hasher,
451    {
452        self.deref().hash(state);
453    }
454}
455
456impl<T: Unaligned + Debug> Debug for Unalign<T> {
457    #[inline(always)]
458    fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
459        Debug::fmt(self.deref(), f)
460    }
461}
462
463impl<T: Unaligned + Display> Display for Unalign<T> {
464    #[inline(always)]
465    fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
466        Display::fmt(self.deref(), f)
467    }
468}
469
470/// A wrapper type to construct uninitialized instances of `T`.
471///
472/// `MaybeUninit` is identical to the [standard library
473/// `MaybeUninit`][core-maybe-uninit] type except that it supports unsized
474/// types.
475///
476/// # Layout
477///
478/// The same layout guarantees and caveats apply to `MaybeUninit<T>` as apply to
479/// the [standard library `MaybeUninit`][core-maybe-uninit] with one exception:
480/// for `T: !Sized`, there is no single value for `T`'s size. Instead, for such
481/// types, the following are guaranteed:
482/// - Every [valid size][valid-size] for `T` is a valid size for
483///   `MaybeUninit<T>` and vice versa
484/// - Given `t: *const T` and `m: *const MaybeUninit<T>` with identical fat
485///   pointer metadata, `t` and `m` address the same number of bytes (and
486///   likewise for `*mut`)
487///
488/// [core-maybe-uninit]: core::mem::MaybeUninit
489/// [valid-size]: crate::KnownLayout#what-is-a-valid-size
490#[repr(transparent)]
491#[doc(hidden)]
492pub struct MaybeUninit<T: ?Sized + KnownLayout>(
493    // SAFETY: `MaybeUninit<T>` has the same size as `T`, because (by invariant
494    // on `T::MaybeUninit`) `T::MaybeUninit` has `T::LAYOUT` identical to `T`,
495    // and because (invariant on `T::LAYOUT`) we can trust that `LAYOUT`
496    // accurately reflects the layout of `T`. By invariant on `T::MaybeUninit`,
497    // it admits uninitialized bytes in all positions. Because `MaybeUninit` is
498    // marked `repr(transparent)`, these properties additionally hold true for
499    // `Self`.
500    T::MaybeUninit,
501);
502
503#[doc(hidden)]
504impl<T: ?Sized + KnownLayout> MaybeUninit<T> {
505    /// Constructs a `MaybeUninit<T>` initialized with the given value.
506    #[inline(always)]
507    pub fn new(val: T) -> Self
508    where
509        T: Sized,
510        Self: Sized,
511    {
512        // SAFETY: It is valid to transmute `val` to `MaybeUninit<T>` because it
513        // is both valid to transmute `val` to `T::MaybeUninit`, and it is valid
514        // to transmute from `T::MaybeUninit` to `MaybeUninit<T>`.
515        //
516        // First, it is valid to transmute `val` to `T::MaybeUninit` because, by
517        // invariant on `T::MaybeUninit`:
518        // - For `T: Sized`, `T` and `T::MaybeUninit` have the same size.
519        // - All byte sequences of the correct size are valid values of
520        //   `T::MaybeUninit`.
521        //
522        // Second, it is additionally valid to transmute from `T::MaybeUninit`
523        // to `MaybeUninit<T>`, because `MaybeUninit<T>` is a
524        // `repr(transparent)` wrapper around `T::MaybeUninit`.
525        //
526        // These two transmutes are collapsed into one so we don't need to add a
527        // `T::MaybeUninit: Sized` bound to this function's `where` clause.
528        unsafe { crate::util::transmute_unchecked(val) }
529    }
530
531    /// Constructs an uninitialized `MaybeUninit<T>`.
532    #[must_use]
533    #[inline(always)]
534    pub fn uninit() -> Self
535    where
536        T: Sized,
537        Self: Sized,
538    {
539        let uninit = CoreMaybeUninit::<T>::uninit();
540        // SAFETY: It is valid to transmute from `CoreMaybeUninit<T>` to
541        // `MaybeUninit<T>` since they both admit uninitialized bytes in all
542        // positions, and they have the same size (i.e., that of `T`).
543        //
544        // `MaybeUninit<T>` has the same size as `T`, because (by invariant on
545        // `T::MaybeUninit`) `T::MaybeUninit` has `T::LAYOUT` identical to `T`,
546        // and because (invariant on `T::LAYOUT`) we can trust that `LAYOUT`
547        // accurately reflects the layout of `T`.
548        //
549        // `CoreMaybeUninit<T>` has the same size as `T` [1] and admits
550        // uninitialized bytes in all positions.
551        //
552        // [1] Per https://doc.rust-lang.org/1.81.0/std/mem/union.MaybeUninit.html#layout-1:
553        //
554        //   `MaybeUninit<T>` is guaranteed to have the same size, alignment,
555        //   and ABI as `T`
556        unsafe { crate::util::transmute_unchecked(uninit) }
557    }
558
559    /// Creates a `Box<MaybeUninit<T>>`.
560    ///
561    /// This function is useful for allocating large, uninit values on the heap
562    /// without ever creating a temporary instance of `Self` on the stack.
563    ///
564    /// # Errors
565    ///
566    /// Returns an error on allocation failure. Allocation failure is guaranteed
567    /// never to cause a panic or an abort.
568    #[cfg(feature = "alloc")]
569    #[inline]
570    pub fn new_boxed_uninit(meta: T::PointerMetadata) -> Result<Box<Self>, AllocError> {
571        // SAFETY: `alloc::alloc::alloc_zeroed` is a valid argument of
572        // `new_box`. The referent of the pointer returned by `alloc` (and,
573        // consequently, the `Box` derived from it) is a valid instance of
574        // `Self`, because `Self` is `MaybeUninit` and thus admits arbitrary
575        // (un)initialized bytes.
576        unsafe { crate::util::new_box(meta, alloc::alloc::alloc) }
577    }
578
579    /// Extracts the value from the `MaybeUninit<T>` container.
580    ///
581    /// # Safety
582    ///
583    /// The caller must ensure that `self` is in an bit-valid state. Depending
584    /// on subsequent use, it may also need to be in a library-valid state.
585    #[inline(always)]
586    pub unsafe fn assume_init(self) -> T
587    where
588        T: Sized,
589        Self: Sized,
590    {
591        // SAFETY: The caller guarantees that `self` is in an bit-valid state.
592        unsafe { crate::util::transmute_unchecked(self) }
593    }
594}
595
596impl<T: ?Sized + KnownLayout> fmt::Debug for MaybeUninit<T> {
597    #[inline]
598    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
599        f.pad(core::any::type_name::<Self>())
600    }
601}
602
603#[allow(unreachable_pub)] // False positive on MSRV
604#[doc(hidden)]
605pub use read_only_def::*;
606mod read_only_def {
607    /// A read-only wrapper.
608    ///
609    /// A `ReadOnly<T>` disables any interior mutability in `T`, ensuring that
610    /// a `&ReadOnly<T>` is genuinely read-only. Thus, `ReadOnly<T>` is
611    /// [`Immutable`] regardless of whether `T` is.
612    ///
613    /// Note that `&mut ReadOnly<T>` still permits mutation – the read-only
614    /// property only applies to shared references.
615    ///
616    /// `ReadOnly<T>` implements [`Copy`] and [`Clone`] when `T: Copy`. Cloning
617    /// copies the value without calling `T::clone`. Trait implementations that
618    /// expose or operate on a shared reference to `T` require `T: Immutable`.
619    /// Constructing a wrapper or accessing its contents through [`AsMut`]
620    /// does not require `T: Immutable`.
621    ///
622    /// [`Immutable`]: crate::Immutable
623    #[derive(Copy)]
624    #[repr(transparent)]
625    pub struct ReadOnly<T: ?Sized> {
626        // INVARIANT: `inner` is never mutated through a `&ReadOnly<T>`
627        // reference.
628        inner: T,
629    }
630
631    impl<T> ReadOnly<T> {
632        /// Creates a new `ReadOnly`.
633        #[must_use]
634        #[inline(always)]
635        pub const fn new(t: T) -> ReadOnly<T> {
636            ReadOnly { inner: t }
637        }
638
639        /// Returns the inner value.
640        #[must_use]
641        #[inline(always)]
642        pub fn into_inner(r: ReadOnly<T>) -> T {
643            r.inner
644        }
645    }
646
647    impl<T: ?Sized> ReadOnly<T> {
648        #[inline(always)]
649        pub(crate) fn as_mut(r: &mut ReadOnly<T>) -> &mut T {
650            // SAFETY: `r: &mut ReadOnly`, so this doesn't violate the invariant
651            // that `inner` is never mutated through a `&ReadOnly<T>` reference.
652            &mut r.inner
653        }
654
655        /// # Safety
656        ///
657        /// The caller promises not to mutate the referent (i.e., via interior
658        /// mutation).
659        pub(crate) const unsafe fn as_ref_unchecked(r: &ReadOnly<T>) -> &T {
660            // SAFETY: The caller promises not to mutate the referent.
661            &r.inner
662        }
663    }
664}
665
666// SAFETY: `ReadOnly<T>` is a `#[repr(transparent)` wrapper around `T`.
667const _: () = unsafe {
668    unsafe_impl_known_layout!(T: ?Sized + KnownLayout => #[repr(T)] ReadOnly<T>);
669};
670
671#[allow(clippy::multiple_unsafe_ops_per_block)]
672// SAFETY:
673// - `ReadOnly<T>` has the same alignment as `T`, and so it is `Unaligned`
674//   exactly when `T` is as well.
675// - `ReadOnly<T>` has the same bit validity as `T`, and so this `is_safe`
676//   implementation is correct, and thus the `TryFromBytes` impl is sound.
677// - `ReadOnly<T>` has the same bit validity as `T`, and so it is `FromZeros`,
678//   `FromBytes`, and `IntoBytes` exactly when `T` is as well.
679const _: () = unsafe {
680    unsafe_impl!(T: ?Sized + Unaligned => Unaligned for ReadOnly<T>);
681    unsafe_impl!(
682        T: ?Sized + TryFromBytes => TryFromBytes for ReadOnly<T>;
683        |c| T::is_safe(c.cast::<_, <ReadOnly<T> as SizeEq<ReadOnly<ReadOnly<T>>>>::CastFrom, _>())
684    );
685    unsafe_impl!(T: ?Sized + FromZeros => FromZeros for ReadOnly<T>);
686    unsafe_impl!(T: ?Sized + FromBytes => FromBytes for ReadOnly<T>);
687    unsafe_impl!(T: ?Sized + IntoBytes => IntoBytes for ReadOnly<T>);
688};
689
690// SAFETY: By invariant, `inner` is never mutated through a `&ReadOnly<T>`
691// reference.
692const _: () = unsafe {
693    unsafe_impl!(T: ?Sized => Immutable for ReadOnly<T>);
694};
695
696const _: () = {
697    use crate::pointer::cast::CastExact;
698
699    // SAFETY: `ReadOnly<T>` has the same layout as `T`.
700    define_cast!(unsafe { pub CastFromReadOnly<T: ?Sized> = ReadOnly<T> => T});
701    // SAFETY: `ReadOnly<T>` has the same layout as `T`.
702    unsafe impl<T: ?Sized> CastExact<ReadOnly<T>, T> for CastFromReadOnly {}
703    // SAFETY: `ReadOnly<T>` has the same layout as `T`.
704    define_cast!(unsafe { pub CastToReadOnly<T: ?Sized> = T => ReadOnly<T>});
705    // SAFETY: `ReadOnly<T>` has the same layout as `T`.
706    unsafe impl<T: ?Sized> CastExact<T, ReadOnly<T>> for CastToReadOnly {}
707
708    impl<T: ?Sized> SizeEq<ReadOnly<T>> for T {
709        type CastFrom = CastFromReadOnly;
710    }
711
712    impl<T: ?Sized> SizeEq<T> for ReadOnly<T> {
713        type CastFrom = CastToReadOnly;
714    }
715};
716
717// SAFETY: `ReadOnly<T>` is a `#[repr(transparent)]` wrapper around `T`, and so
718// it has the same bit validity as `T`.
719unsafe impl<T: ?Sized> TransmuteFrom<T, Safe, Safe> for ReadOnly<T> {}
720
721// SAFETY: `ReadOnly<T>` is a `#[repr(transparent)]` wrapper around `T`, and so
722// it has the same bit validity as `T`.
723unsafe impl<T: ?Sized> TransmuteFrom<ReadOnly<T>, Safe, Safe> for T {}
724
725impl<T: Default> Default for ReadOnly<T> {
726    #[inline(always)]
727    fn default() -> Self {
728        Self::new(Default::default())
729    }
730}
731
732// Copying avoids calling `T::clone`, which could mutate through a shared
733// reference to `T` and violate `ReadOnly`'s invariant.
734impl<T: Copy> Clone for ReadOnly<T> {
735    #[inline(always)]
736    fn clone(&self) -> Self {
737        *self
738    }
739}
740
741impl<T> From<T> for ReadOnly<T> {
742    #[inline(always)]
743    fn from(t: T) -> Self {
744        Self::new(t)
745    }
746}
747
748impl<'a, T: ?Sized + Immutable> From<&'a T> for &'a ReadOnly<T> {
749    #[inline(always)]
750    fn from(t: &'a T) -> &'a ReadOnly<T> {
751        let ro = Ptr::from_ref(t).transmute::<_, _, (_, _)>();
752        // SAFETY: `ReadOnly<T>` has the same alignment as `T`, and
753        // `Ptr::from_ref` produces an aligned `Ptr`.
754        let ro = unsafe { ro.assume_alignment() };
755        ro.as_ref()
756    }
757}
758
759impl<T: ?Sized + Immutable> Deref for ReadOnly<T> {
760    type Target = T;
761
762    #[inline(always)]
763    fn deref(&self) -> &Self::Target {
764        // SAFETY: By `T: Immutable`, `&T` doesn't permit interior mutation.
765        unsafe { ReadOnly::as_ref_unchecked(self) }
766    }
767}
768
769impl<T: ?Sized + Immutable> DerefMut for ReadOnly<T> {
770    #[inline(always)]
771    fn deref_mut(&mut self) -> &mut Self::Target {
772        ReadOnly::as_mut(self)
773    }
774}
775
776impl<T: ?Sized + Immutable> AsRef<T> for ReadOnly<T> {
777    #[inline(always)]
778    fn as_ref(&self) -> &T {
779        self.deref()
780    }
781}
782
783impl<T: ?Sized + Immutable> Borrow<T> for ReadOnly<T> {
784    #[inline(always)]
785    fn borrow(&self) -> &T {
786        self.deref()
787    }
788}
789
790impl<T: ?Sized> AsMut<T> for ReadOnly<T> {
791    #[inline(always)]
792    fn as_mut(&mut self) -> &mut T {
793        ReadOnly::as_mut(self)
794    }
795}
796
797// Delegate shared access through `Deref`, which requires `T: Immutable` and
798// thus prevents interior mutation of the wrapped value by these trait methods.
799impl<T: ?Sized + Immutable + PartialEq> PartialEq for ReadOnly<T> {
800    #[inline(always)]
801    fn eq(&self, other: &Self) -> bool {
802        self.deref().eq(other.deref())
803    }
804}
805
806impl<T: ?Sized + Immutable + Eq> Eq for ReadOnly<T> {}
807
808impl<T: ?Sized + Immutable + PartialOrd> PartialOrd for ReadOnly<T> {
809    #[inline(always)]
810    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
811        self.deref().partial_cmp(other.deref())
812    }
813}
814
815impl<T: ?Sized + Immutable + Ord> Ord for ReadOnly<T> {
816    #[inline(always)]
817    fn cmp(&self, other: &Self) -> Ordering {
818        self.deref().cmp(other.deref())
819    }
820}
821
822impl<T: ?Sized + Immutable + Hash> Hash for ReadOnly<T> {
823    #[inline(always)]
824    fn hash<H: Hasher>(&self, state: &mut H) {
825        self.deref().hash(state);
826    }
827}
828
829macro_rules! impl_read_only_fmt {
830    ($($trait:ident),* $(,)?) => {$(
831        impl<T: ?Sized + Immutable + fmt::$trait> fmt::$trait for ReadOnly<T> {
832            #[inline(always)]
833            fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
834                fmt::$trait::fmt(self.deref(), f)
835            }
836        }
837    )*};
838}
839
840impl_read_only_fmt!(Debug, Display, Binary, Octal, LowerHex, UpperHex, LowerExp, UpperExp);
841
842// SAFETY: See safety comment on `ProjectToTag`.
843unsafe impl<T: HasTag<Client> + ?Sized, Client> HasTag<Client> for ReadOnly<T> {
844    #[allow(clippy::missing_inline_in_public_items)]
845    fn only_derive_is_allowed_to_implement_this_trait()
846    where
847        Self: Sized,
848    {
849    }
850
851    type Tag = <T as HasTag<Client>>::Tag;
852
853    // SAFETY: `<T as SizeEq<ReadOnly<T>>>::CastFrom` is a no-op projection that
854    // produces a pointer with the same referent. By invariant, for any
855    // `Ptr<'_, T, I>` it is sound to use
856    // `<T as HasTag<Client>>::ProjectToTag` to project to a
857    // shared `Ptr<'_, <T as HasTag<Client>>::Tag,
858    // (_, I::Alignment, I::Validity)>`. Since `ReadOnly<T>` has the same
859    // layout and validity as `T`, the same is true of projecting from a
860    // `Ptr<'_, ReadOnly<T>, I>`.
861    type ProjectToTag = crate::pointer::cast::TransitiveProject<
862        T,
863        <T as SizeEq<ReadOnly<T>>>::CastFrom,
864        <T as HasTag<Client>>::ProjectToTag,
865    >;
866}
867
868// SAFETY: `ReadOnly<T>` is a `#[repr(transparent)]` wrapper around `T`, and so
869// has the same fields at the same offsets. Thus, it satisfies the safety
870// invariants of `HasField<Client, Field, VARIANT_ID, FIELD_ID>` for field `f`
871// exactly when `T` does, as guaranteed by the `T: HasField` bound:
872// - If `VARIANT_ID` is `STRUCT_VARIANT_ID`, `UNION_VARIANT_ID`, or
873//   `REPR_C_UNION_VARIANT_ID`, then `T` has the layout of a struct, non-C union,
874//   or C union type respectively. Since `ReadOnly<T>` is a transparent wrapper
875//   around `T`, it does too. Otherwise, `VARIANT_ID` is the identifier ID of an
876//   enum variant; `T` has the layout of an enum type, and `ReadOnly<T>` does too.
877// - By `T: HasField<_, _, _, FIELD_ID>`:
878//   - `T` has a field `f` with name `n` such that
879//     `FIELD_ID = zerocopy::ident_id!(n)` or at index `i` such that
880//     `FIELD_ID = zerocopy::ident_id!(i)`.
881//   - `Field` has the same visibility as `f`.
882//   - `T::Type` has the same type as `f`. Thus, `ReadOnly<T::Type>` has the
883//     same type as `f`, wrapped in `ReadOnly`.
884//
885// `project` satisfies its post-condition – namely, that the returned pointer
886// refers to a non-strict subset of the bytes of `slf`'s referent, and has the
887// same provenance as `slf` – because all intermediate operations satisfy those
888// same conditions.
889unsafe impl<T, Client, Field, const VARIANT_ID: i128, const FIELD_ID: i128>
890    HasField<Client, Field, VARIANT_ID, FIELD_ID> for ReadOnly<T>
891where
892    T: HasField<Client, Field, VARIANT_ID, FIELD_ID> + ?Sized,
893{
894    #[allow(clippy::missing_inline_in_public_items)]
895    fn only_derive_is_allowed_to_implement_this_trait()
896    where
897        Self: Sized,
898    {
899    }
900
901    type Type = ReadOnly<T::Type>;
902
903    #[inline(always)]
904    fn project(slf: PtrInner<'_, Self>) -> *mut ReadOnly<T::Type> {
905        slf.project::<_, <T as SizeEq<ReadOnly<T>>>::CastFrom>()
906            .project::<_, crate::pointer::cast::Projection<Client, Field, VARIANT_ID, FIELD_ID>>()
907            .project::<_, <ReadOnly<T::Type> as SizeEq<T::Type>>::CastFrom>()
908            .as_non_null()
909            .as_ptr()
910    }
911}
912
913// SAFETY: `ReadOnly<T>` is a `#[repr(transparent)]` wrapper around `T`, and so
914// has the same fields at the same offsets. `is_projectable` simply delegates to
915// `T::is_projectable`, which is sound because a `Ptr<'_, ReadOnly<T>, I>` will
916// be projectable exactly when a `Ptr<'_, T, I>` referent is.
917unsafe impl<T, Client, Field, I, const VARIANT_ID: i128, const FIELD_ID: i128>
918    ProjectField<Client, Field, I, VARIANT_ID, FIELD_ID> for ReadOnly<T>
919where
920    T: ProjectField<Client, Field, I, VARIANT_ID, FIELD_ID> + ?Sized,
921    I: invariant::Invariants,
922{
923    #[allow(clippy::missing_inline_in_public_items)]
924    fn only_derive_is_allowed_to_implement_this_trait()
925    where
926        Self: Sized,
927    {
928    }
929
930    type Invariants = <T as ProjectField<Client, Field, I, VARIANT_ID, FIELD_ID>>::Invariants;
931
932    type Error = <T as ProjectField<Client, Field, I, VARIANT_ID, FIELD_ID>>::Error;
933
934    #[inline(always)]
935    fn is_projectable<'a>(
936        ptr: Ptr<'a, <Self as HasTag<Client>>::Tag, (invariant::Shared, I::Alignment, I::Validity)>,
937    ) -> Result<(), Self::Error> {
938        <T as ProjectField<Client, Field, I, VARIANT_ID, FIELD_ID>>::is_projectable(ptr)
939    }
940}
941
942#[cfg(test)]
943mod tests {
944    use core::panic::AssertUnwindSafe;
945
946    use super::*;
947    use crate::util::testutil::*;
948
949    #[test]
950    #[allow(clippy::clone_on_copy, clippy::non_canonical_clone_impl)]
951    fn test_read_only_copy_clone() {
952        // This type deliberately does not implement `Immutable`. Copying the
953        // wrapper must not require it or invoke the inner `Clone` impl.
954        #[derive(Copy, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
955        struct PanickingClone(u16);
956
957        impl Clone for PanickingClone {
958            fn clone(&self) -> Self {
959                panic!("ReadOnly must not call the inner Clone implementation")
960            }
961        }
962
963        static_assertions::assert_not_impl_any!(
964            ReadOnly<PanickingClone>: Debug, PartialEq, Eq, PartialOrd, Ord,
965            Hash, AsRef<PanickingClone>, Borrow<PanickingClone>
966        );
967        static_assertions::assert_not_impl_any!(
968            ReadOnly<Cell<u8>>: Copy, Clone, Deref, AsRef<Cell<u8>>, Borrow<Cell<u8>>
969        );
970
971        let value = ReadOnly::new(PanickingClone(42));
972        let copy = value;
973        let clone = value.clone();
974        assert_eq!(ReadOnly::into_inner(copy).0, 42);
975        assert_eq!(ReadOnly::into_inner(clone).0, 42);
976    }
977
978    #[test]
979    fn test_read_only_mutable_access() {
980        // Construction and exclusive access also support interior mutability.
981        let mut value = ReadOnly::<Cell<u8>>::default();
982        let inner: &mut Cell<u8> = value.as_mut();
983        inner.set(42);
984        assert_eq!(ReadOnly::into_inner(value).get(), 42);
985
986        let value = ReadOnly::from(Cell::new(7));
987        assert_eq!(ReadOnly::into_inner(value).get(), 7);
988    }
989
990    #[test]
991    fn test_read_only_borrowed_lookup() {
992        // Borrowed lookups require equality, hashing, and ordering to agree
993        // between the wrapper and the inner value.
994        let mut hash_map = std::collections::HashMap::new();
995        hash_map.insert(ReadOnly::new(42u16), true);
996        assert_eq!(hash_map.get(&42u16), Some(&true));
997        assert_eq!(hash_map.get(&43u16), None);
998
999        let mut tree = std::collections::BTreeMap::new();
1000        tree.insert(ReadOnly::new(42u16), true);
1001        assert_eq!(tree.get(&42u16), Some(&true));
1002        assert_eq!(tree.get(&43u16), None);
1003
1004        // Preserve partial orders, including incomparable values.
1005        let nan = ReadOnly::new(f32::NAN);
1006        assert_eq!(nan.partial_cmp(&nan), None);
1007        assert_ne!(nan, nan);
1008    }
1009
1010    #[test]
1011    fn test_read_only_unsized() {
1012        let bytes = [1u8, 2];
1013        let value: &ReadOnly<[u8]> = (&bytes[..]).into();
1014        let as_ref: &[u8] = value.as_ref();
1015        let borrowed: &[u8] = value.borrow();
1016        assert_eq!(as_ref, bytes);
1017        assert_eq!(borrowed, bytes);
1018        assert_eq!(value, value);
1019        assert_eq!(value.cmp(value), Ordering::Equal);
1020
1021        let text: &ReadOnly<str> = "hello".into();
1022        assert_eq!(format!("{:.3}", text), "hel");
1023    }
1024
1025    #[test]
1026    fn test_read_only_formatting() {
1027        let value = ReadOnly::new(42u16);
1028        // Check that formatting flags are forwarded as well as the value.
1029        assert_eq!(format!("{:04?}", value), "0042");
1030        assert_eq!(format!("{:04}", value), "0042");
1031        assert_eq!(format!("{:#010b}", value), "0b00101010");
1032        assert_eq!(format!("{:#06o}", value), "0o0052");
1033        assert_eq!(format!("{:#06x}", value), "0x002a");
1034        assert_eq!(format!("{:#06X}", value), "0x002A");
1035        assert_eq!(format!("{:.2e}", value), "4.20e1");
1036        assert_eq!(format!("{:.2E}", value), "4.20E1");
1037    }
1038
1039    #[test]
1040    fn test_unalign() {
1041        // Test methods that don't depend on alignment.
1042        let mut u = Unalign::new(AU64(123));
1043        assert_eq!(u.get(), AU64(123));
1044        assert_eq!(u.into_inner(), AU64(123));
1045        assert_eq!(u.get_ptr(), <*const _>::cast::<AU64>(&u));
1046        assert_eq!(u.get_mut_ptr(), <*mut _>::cast::<AU64>(&mut u));
1047        u.set(AU64(321));
1048        assert_eq!(u.get(), AU64(321));
1049
1050        // Test methods that depend on alignment (when alignment is satisfied).
1051        let mut u: Align<_, AU64> = Align::new(Unalign::new(AU64(123)));
1052        assert_eq!(u.t.try_deref().unwrap(), &AU64(123));
1053        assert_eq!(u.t.try_deref_mut().unwrap(), &mut AU64(123));
1054        // SAFETY: The `Align<_, AU64>` guarantees proper alignment.
1055        assert_eq!(unsafe { u.t.deref_unchecked() }, &AU64(123));
1056        // SAFETY: The `Align<_, AU64>` guarantees proper alignment.
1057        assert_eq!(unsafe { u.t.deref_mut_unchecked() }, &mut AU64(123));
1058        *u.t.try_deref_mut().unwrap() = AU64(321);
1059        assert_eq!(u.t.get(), AU64(321));
1060
1061        // Test methods that depend on alignment (when alignment is not
1062        // satisfied).
1063        let mut u: ForceUnalign<_, AU64> = ForceUnalign::new(Unalign::new(AU64(123)));
1064        assert!(matches!(u.t.try_deref(), Err(AlignmentError { .. })));
1065        assert!(matches!(u.t.try_deref_mut(), Err(AlignmentError { .. })));
1066
1067        // Test methods that depend on `T: Unaligned`.
1068        let mut u = Unalign::new(123u8);
1069        assert_eq!(u.try_deref(), Ok(&123));
1070        assert_eq!(u.try_deref_mut(), Ok(&mut 123));
1071        assert_eq!(u.deref(), &123);
1072        assert_eq!(u.deref_mut(), &mut 123);
1073        *u = 21;
1074        assert_eq!(u.get(), 21);
1075
1076        // Test that some `Unalign` functions and methods are `const`.
1077        const _UNALIGN: Unalign<u64> = Unalign::new(0);
1078        const _UNALIGN_PTR: *const u64 = _UNALIGN.get_ptr();
1079        const _U64: u64 = _UNALIGN.into_inner();
1080        // Make sure all code is considered "used".
1081        //
1082        // FIXME(https://github.com/rust-lang/rust/issues/104084): Remove this
1083        // attribute.
1084        #[allow(dead_code)]
1085        const _: () = {
1086            let x: Align<_, AU64> = Align::new(Unalign::new(AU64(123)));
1087            // Make sure that `deref_unchecked` is `const`.
1088            //
1089            // SAFETY: The `Align<_, AU64>` guarantees proper alignment.
1090            let au64 = unsafe { x.t.deref_unchecked() };
1091            match au64 {
1092                AU64(123) => {}
1093                _ => const_unreachable!(),
1094            }
1095        };
1096    }
1097
1098    #[test]
1099    fn test_unalign_update() {
1100        let mut u = Unalign::new(AU64(123));
1101        u.update(|a| a.0 += 1);
1102        assert_eq!(u.get(), AU64(124));
1103
1104        // Test that, even if the callback panics, the original is still
1105        // correctly overwritten. Use a `Box` so that Miri is more likely to
1106        // catch any unsoundness (which would likely result in two `Box`es for
1107        // the same heap object, which is the sort of thing that Miri would
1108        // probably catch).
1109        let mut u = Unalign::new(Box::new(AU64(123)));
1110        let res = std::panic::catch_unwind(AssertUnwindSafe(|| {
1111            u.update(|a| {
1112                a.0 += 1;
1113                panic!();
1114            })
1115        }));
1116        assert!(res.is_err());
1117        assert_eq!(u.into_inner(), Box::new(AU64(124)));
1118
1119        // Test the align_of::<T>() == 1 optimization.
1120        let mut u = Unalign::new([0u8, 1]);
1121        u.update(|a| a[0] += 1);
1122        assert_eq!(u.get(), [1u8, 1]);
1123    }
1124
1125    #[test]
1126    fn test_unalign_copy_clone() {
1127        // Test that `Copy` and `Clone` do not cause soundness issues. This test
1128        // is mainly meant to exercise UB that would be caught by Miri.
1129
1130        // `u.t` is definitely not validly-aligned for `AU64`'s alignment of 8.
1131        let u = ForceUnalign::<_, AU64>::new(Unalign::new(AU64(123)));
1132        #[allow(clippy::clone_on_copy)]
1133        let v = u.t.clone();
1134        let w = u.t;
1135        assert_eq!(u.t.get(), v.get());
1136        assert_eq!(u.t.get(), w.get());
1137        assert_eq!(v.get(), w.get());
1138    }
1139
1140    #[test]
1141    fn test_unalign_trait_impls() {
1142        let zero = Unalign::new(0u8);
1143        let one = Unalign::new(1u8);
1144
1145        assert!(zero < one);
1146        assert_eq!(PartialOrd::partial_cmp(&zero, &one), Some(Ordering::Less));
1147        assert_eq!(Ord::cmp(&zero, &one), Ordering::Less);
1148
1149        assert_ne!(zero, one);
1150        assert_eq!(zero, zero);
1151        assert!(!PartialEq::eq(&zero, &one));
1152        assert!(PartialEq::eq(&zero, &zero));
1153
1154        fn hash<T: Hash>(t: &T) -> u64 {
1155            let mut h = std::collections::hash_map::DefaultHasher::new();
1156            t.hash(&mut h);
1157            h.finish()
1158        }
1159
1160        assert_eq!(hash(&zero), hash(&0u8));
1161        assert_eq!(hash(&one), hash(&1u8));
1162
1163        assert_eq!(format!("{:?}", zero), format!("{:?}", 0u8));
1164        assert_eq!(format!("{:?}", one), format!("{:?}", 1u8));
1165        assert_eq!(format!("{}", zero), format!("{}", 0u8));
1166        assert_eq!(format!("{}", one), format!("{}", 1u8));
1167    }
1168
1169    #[test]
1170    #[allow(clippy::as_conversions)]
1171    fn test_maybe_uninit() {
1172        // int
1173        {
1174            let input = 42;
1175            let uninit = MaybeUninit::new(input);
1176            // SAFETY: `uninit` is in an initialized state
1177            let output = unsafe { uninit.assume_init() };
1178            assert_eq!(input, output);
1179        }
1180
1181        // thin ref
1182        {
1183            let input = 42;
1184            let uninit = MaybeUninit::new(&input);
1185            // SAFETY: `uninit` is in an initialized state
1186            let output = unsafe { uninit.assume_init() };
1187            assert_eq!(&input as *const _, output as *const _);
1188            assert_eq!(input, *output);
1189        }
1190
1191        // wide ref
1192        {
1193            let input = [1, 2, 3, 4];
1194            let uninit = MaybeUninit::new(&input[..]);
1195            // SAFETY: `uninit` is in an initialized state
1196            let output = unsafe { uninit.assume_init() };
1197            assert_eq!(&input[..] as *const _, output as *const _);
1198            assert_eq!(input, *output);
1199        }
1200    }
1201    #[test]
1202    fn test_maybe_uninit_uninit() {
1203        let _uninit = MaybeUninit::<u8>::uninit();
1204        // Cannot check value, but can check it compiles and runs
1205    }
1206
1207    #[test]
1208    #[cfg(feature = "alloc")]
1209    fn test_maybe_uninit_new_boxed_uninit() {
1210        let _boxed = MaybeUninit::<u8>::new_boxed_uninit(()).unwrap();
1211    }
1212
1213    #[test]
1214    fn test_maybe_uninit_debug() {
1215        let uninit = MaybeUninit::<u8>::uninit();
1216        assert!(format!("{:?}", uninit).contains("MaybeUninit"));
1217    }
1218}