Skip to main content

zerocopy/util/
macro_util.rs

1// SPDX-License-Identifier: (BSD-2-Clause OR Apache-2.0) OR MIT
2//
3// Copyright 2022 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
11//! Utilities used by macros and by `zerocopy-derive`.
12//!
13//! These are defined here `zerocopy` rather than in code generated by macros or
14//! by `zerocopy-derive` so that they can be compiled once rather than
15//! recompiled for every invocation (e.g., if they were defined in generated
16//! code, then deriving `IntoBytes` and `FromBytes` on three different types
17//! would result in the code in question being emitted and compiled six
18//! different times).
19
20#![allow(missing_debug_implementations)]
21
22// FIXME(#29), FIXME(https://github.com/rust-lang/rust/issues/69835): Remove
23// this `cfg` when `size_of_val_raw` is stabilized.
24#[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
25#[cfg(not(target_pointer_width = "16"))]
26use core::ptr::{self, NonNull};
27use core::{marker::PhantomData, mem, num::Wrapping};
28
29use crate::{
30    pointer::{
31        cast::CastSizedExact,
32        invariant::{Aligned, Initialized, Safe},
33        BecauseImmutable,
34    },
35    FromBytes, Immutable, IntoBytes, KnownLayout, Ptr, ReadOnly, TryFromBytes, ValidityError,
36};
37
38/// Projects the type of the field at `Index` in `Self` without regard for field
39/// privacy.
40///
41/// The `Index` parameter is any sort of handle that identifies the field; its
42/// definition is the obligation of the implementer.
43///
44/// # Safety
45///
46/// Unsafe code may assume that this accurately reflects the definition of
47/// `Self`.
48pub unsafe trait Field<Index> {
49    /// The type of the field at `Index`.
50    type Type: ?Sized;
51}
52
53#[cfg_attr(
54    not(no_zerocopy_diagnostic_on_unimplemented_1_78_0),
55    diagnostic::on_unimplemented(
56        message = "`{T}` has {PADDING_BYTES} total byte(s) of padding",
57        label = "types with padding cannot implement `IntoBytes`",
58        note = "consider using `zerocopy::Unalign` to lower the alignment of individual fields",
59        note = "consider adding explicit fields where padding would be",
60        note = "consider using `#[repr(packed)]` to remove padding"
61    )
62)]
63pub trait PaddingFree<T: ?Sized, const PADDING_BYTES: usize> {}
64impl<T: ?Sized> PaddingFree<T, 0> for () {}
65
66// FIXME(#1112): In the slice DST case, we should delegate to *both*
67// `PaddingFree` *and* `DynamicPaddingFree` (and probably rename `PaddingFree`
68// to `StaticPaddingFree` or something - or introduce a third trait with that
69// name) so that we can have more clear error messages.
70
71#[cfg_attr(
72    not(no_zerocopy_diagnostic_on_unimplemented_1_78_0),
73    diagnostic::on_unimplemented(
74        message = "`{T}` has one or more padding bytes",
75        label = "types with padding cannot implement `IntoBytes`",
76        note = "consider using `zerocopy::Unalign` to lower the alignment of individual fields",
77        note = "consider adding explicit fields where padding would be",
78        note = "consider using `#[repr(packed)]` to remove padding"
79    )
80)]
81pub trait DynamicPaddingFree<T: ?Sized, const HAS_PADDING: bool> {}
82impl<T: ?Sized> DynamicPaddingFree<T, false> for () {}
83
84#[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
85#[cfg(not(target_pointer_width = "16"))]
86const _64K: usize = 1 << 16;
87
88// FIXME(#29), FIXME(https://github.com/rust-lang/rust/issues/69835): Remove
89// this `cfg` when `size_of_val_raw` is stabilized.
90#[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
91#[cfg(not(target_pointer_width = "16"))]
92#[repr(C, align(65536))]
93struct Aligned64kAllocation([u8; _64K]);
94
95/// A pointer to an aligned allocation of size 2^16.
96///
97/// # Safety
98///
99/// `ALIGNED_64K_ALLOCATION` is guaranteed to point to the entirety of an
100/// allocation with size and alignment 2^16, and to have valid provenance.
101// FIXME(#29), FIXME(https://github.com/rust-lang/rust/issues/69835): Remove
102// this `cfg` when `size_of_val_raw` is stabilized.
103#[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
104#[cfg(not(target_pointer_width = "16"))]
105pub const ALIGNED_64K_ALLOCATION: NonNull<[u8]> = {
106    const REF: &Aligned64kAllocation = &Aligned64kAllocation([0; _64K]);
107    let ptr: *const Aligned64kAllocation = REF;
108    let ptr: *const [u8] = ptr::slice_from_raw_parts(ptr.cast(), _64K);
109    // SAFETY:
110    // - `ptr` is derived from a Rust reference, which is guaranteed to be
111    //   non-null.
112    // - `ptr` is derived from an `&Aligned64kAllocation`, which has size and
113    //   alignment `_64K` as promised. Its length is initialized to `_64K`,
114    //   which means that it refers to the entire allocation.
115    // - `ptr` is derived from a Rust reference, which is guaranteed to have
116    //   valid provenance.
117    //
118    // FIXME(#429): Once `NonNull::new_unchecked` docs document that it
119    // preserves provenance, cite those docs.
120    // FIXME: Replace this `as` with `ptr.cast_mut()` once our MSRV >= 1.65
121    #[allow(clippy::as_conversions)]
122    unsafe {
123        NonNull::new_unchecked(ptr as *mut _)
124    }
125};
126
127/// Computes the offset of the base of the field `$trailing_field_name` within
128/// the type `$ty`.
129///
130/// `trailing_field_offset!` produces code which is valid in a `const` context.
131// FIXME(#29), FIXME(https://github.com/rust-lang/rust/issues/69835): Remove
132// this `cfg` when `size_of_val_raw` is stabilized.
133#[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
134#[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
135#[macro_export]
136macro_rules! trailing_field_offset {
137    ($ty:ty, $trailing_field_name:tt) => {{
138        let min_size = {
139            let zero_elems: *const [()] =
140                $crate::util::macro_util::core_reexport::ptr::slice_from_raw_parts(
141                    $crate::util::macro_util::core_reexport::ptr::NonNull::<()>::dangling()
142                        .as_ptr()
143                        .cast_const(),
144                    0,
145                );
146            // SAFETY:
147            // - If `$ty` is `Sized`, `size_of_val_raw` is always safe to call.
148            // - Otherwise:
149            //   - If `$ty` is not a slice DST, this pointer conversion will
150            //     fail due to "mismatched vtable kinds", and compilation will
151            //     fail.
152            //   - If `$ty` is a slice DST, we have constructed `zero_elems` to
153            //     have zero trailing slice elements. Per the `size_of_val_raw`
154            //     docs, "For the special case where the dynamic tail length is
155            //     0, this function is safe to call." [1]
156            //
157            // [1] https://doc.rust-lang.org/nightly/std/mem/fn.size_of_val_raw.html
158            unsafe {
159                #[allow(clippy::as_conversions)]
160                $crate::util::macro_util::core_reexport::mem::size_of_val_raw(
161                    zero_elems as *const $ty,
162                )
163            }
164        };
165
166        assert!(min_size <= _64K);
167
168        #[allow(clippy::as_conversions)]
169        let ptr = ALIGNED_64K_ALLOCATION.as_ptr() as *const $ty;
170
171        // SAFETY:
172        // - Thanks to the preceding `assert!`, we know that the value with zero
173        //   elements fits in `_64K` bytes, and thus in the allocation addressed
174        //   by `ALIGNED_64K_ALLOCATION`. The offset of the trailing field is
175        //   guaranteed to be no larger than this size, so this field projection
176        //   is guaranteed to remain in-bounds of its allocation.
177        // - Because the minimum size is no larger than `_64K` bytes, and
178        //   because an object's size must always be a multiple of its alignment
179        //   [1], we know that `$ty`'s alignment is no larger than `_64K`. The
180        //   allocation addressed by `ALIGNED_64K_ALLOCATION` is guaranteed to
181        //   be aligned to `_64K`, so `ptr` is guaranteed to satisfy `$ty`'s
182        //   alignment.
183        // - As required by `addr_of!`, we do not write through `field`.
184        //
185        //   Note that, as of [2], this requirement is technically unnecessary
186        //   for Rust versions >= 1.75.0, but no harm in guaranteeing it anyway
187        //   until we bump our MSRV.
188        //
189        // [1] Per https://doc.rust-lang.org/reference/type-layout.html:
190        //
191        //   The size of a value is always a multiple of its alignment.
192        //
193        // [2] https://github.com/rust-lang/reference/pull/1387
194        let field = unsafe {
195            $crate::util::macro_util::core_reexport::ptr::addr_of!((*ptr).$trailing_field_name)
196        };
197        // SAFETY:
198        // - Both `ptr` and `field` are derived from the same allocated object.
199        // - By the preceding safety comment, `field` is in bounds of that
200        //   allocated object.
201        // - The distance, in bytes, between `ptr` and `field` is required to be
202        //   a multiple of the size of `u8`, which is trivially true because
203        //   `u8`'s size is 1.
204        // - The distance, in bytes, cannot overflow `isize`. This is guaranteed
205        //   because no allocated object can have a size larger than can fit in
206        //   `isize`. [1]
207        // - The distance being in-bounds cannot rely on wrapping around the
208        //   address space. This is guaranteed because the same is guaranteed of
209        //   allocated objects. [1]
210        //
211        // [1] FIXME(#429), FIXME(https://github.com/rust-lang/rust/pull/116675):
212        //     Once these are guaranteed in the Reference, cite it.
213        let offset = unsafe { field.cast::<u8>().offset_from(ptr.cast::<u8>()) };
214        // Guaranteed not to be lossy: `field` comes after `ptr`, so the offset
215        // from `ptr` to `field` is guaranteed to be positive.
216        assert!(offset >= 0);
217        Some(
218            #[allow(clippy::as_conversions)]
219            {
220                offset as usize
221            },
222        )
223    }};
224}
225
226/// Computes alignment of `$ty: ?Sized`.
227///
228/// `align_of!` produces code which is valid in a `const` context.
229// FIXME(#29), FIXME(https://github.com/rust-lang/rust/issues/69835): Remove
230// this `cfg` when `size_of_val_raw` is stabilized.
231#[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
232#[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
233#[macro_export]
234macro_rules! align_of {
235    ($ty:ty) => {{
236        // SAFETY: `OffsetOfTrailingIsAlignment` is `repr(C)`, and its layout is
237        // guaranteed [1] to begin with the single-byte layout for `_byte`,
238        // followed by the padding needed to align `_trailing`, then the layout
239        // for `_trailing`, and finally any trailing padding bytes needed to
240        // correctly-align the entire struct.
241        //
242        // This macro computes the alignment of `$ty` by counting the number of
243        // bytes preceding `_trailing`. For instance, if the alignment of `$ty`
244        // is `1`, then no padding is required align `_trailing` and it will be
245        // located immediately after `_byte` at offset 1. If the alignment of
246        // `$ty` is 2, then a single padding byte is required before
247        // `_trailing`, and `_trailing` will be located at offset 2.
248
249        // This correspondence between offset and alignment holds for all valid
250        // Rust alignments, and we confirm this exhaustively (or, at least up to
251        // the maximum alignment supported by `trailing_field_offset!`) in
252        // `test_align_of_dst`.
253        //
254        // [1]: https://doc.rust-lang.org/nomicon/other-reprs.html#reprc
255
256        #[repr(C)]
257        struct OffsetOfTrailingIsAlignment {
258            _byte: u8,
259            _trailing: $ty,
260        }
261
262        trailing_field_offset!(OffsetOfTrailingIsAlignment, _trailing)
263    }};
264}
265
266mod size_to_tag {
267    pub trait SizeToTag<const SIZE: usize> {
268        type Tag;
269    }
270
271    impl SizeToTag<1> for () {
272        type Tag = u8;
273    }
274    impl SizeToTag<2> for () {
275        type Tag = u16;
276    }
277    impl SizeToTag<4> for () {
278        type Tag = u32;
279    }
280    impl SizeToTag<8> for () {
281        type Tag = u64;
282    }
283    impl SizeToTag<16> for () {
284        type Tag = u128;
285    }
286}
287
288/// An alias for the unsigned integer of the given size in bytes.
289#[doc(hidden)]
290pub type SizeToTag<const SIZE: usize> = <() as size_to_tag::SizeToTag<SIZE>>::Tag;
291
292// We put `Sized` in its own module so it can have the same name as the standard
293// library `Sized` without shadowing it in the parent module.
294#[cfg(not(no_zerocopy_diagnostic_on_unimplemented_1_78_0))]
295mod __size_of {
296    #[diagnostic::on_unimplemented(
297        message = "`{Self}` is unsized",
298        label = "`IntoBytes` needs all field types to be `Sized` in order to determine whether there is padding",
299        note = "consider using `#[repr(packed)]` to remove padding",
300        note = "`IntoBytes` does not require the fields of `#[repr(packed)]` types to be `Sized`"
301    )]
302    pub trait Sized: core::marker::Sized {}
303    impl<T: core::marker::Sized> Sized for T {}
304
305    #[inline(always)]
306    #[must_use]
307    #[allow(clippy::needless_maybe_sized)]
308    pub const fn size_of<T: Sized + ?core::marker::Sized>() -> usize {
309        core::mem::size_of::<T>()
310    }
311}
312
313#[cfg(no_zerocopy_diagnostic_on_unimplemented_1_78_0)]
314pub use core::mem::size_of;
315
316#[cfg(not(no_zerocopy_diagnostic_on_unimplemented_1_78_0))]
317pub use __size_of::size_of;
318
319/// How many padding bytes does the struct type `$t` have?
320///
321/// `$ts` is the list of the type of every field in `$t`. `$t` must be a struct
322/// type, or else `struct_padding!`'s result may be meaningless.
323///
324/// Note that `struct_padding!`'s results are independent of `repcr` since they
325/// only consider the size of the type and the sizes of the fields. Whatever the
326/// repr, the size of the type already takes into account any padding that the
327/// compiler has decided to add. Structs with well-defined representations (such
328/// as `repr(C)`) can use this macro to check for padding. Note that while this
329/// may yield some consistent value for some `repr(Rust)` structs, it is not
330/// guaranteed across platforms or compilations.
331#[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
332#[macro_export]
333macro_rules! struct_padding {
334    ($t:ty, $_align:expr, $_packed:expr, [$($ts:ty),*]) => {{
335        // The `align` and `packed` directives can be ignored here. Regardless
336        // of if and how they are set, comparing the size of `$t` to the sum of
337        // its field sizes is a reliable indicator of the presence of padding.
338        $crate::util::macro_util::size_of::<$t>() - (0 $(+ $crate::util::macro_util::size_of::<$ts>())*)
339    }};
340}
341
342/// Does the `repr(C)` struct type `$t` have padding?
343///
344/// `$ts` is the list of the type of every field in `$t`. `$t` must be a
345/// `repr(C)` struct type, or else `struct_has_padding!`'s result may be
346/// meaningless.
347#[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
348#[macro_export]
349macro_rules! repr_c_struct_has_padding {
350    ($t:ty, $align:expr, $packed:expr, [$($ts:tt),*]) => {{
351        let layout = $crate::DstLayout::for_repr_c_struct(
352            $align,
353            $packed,
354            &[$($crate::repr_c_struct_has_padding!(@field $ts),)*]
355        );
356        layout.requires_static_padding() || layout.requires_dynamic_padding()
357    }};
358    (@field ([$t:ty])) => {
359        <[$t] as $crate::KnownLayout>::LAYOUT
360    };
361    (@field ($t:ty)) => {
362        $crate::DstLayout::for_unpadded_type::<$t>()
363    };
364    (@field [$t:ty]) => {
365        <[$t] as $crate::KnownLayout>::LAYOUT
366    };
367    (@field $t:ty) => {
368        $crate::DstLayout::for_unpadded_type::<$t>()
369    };
370}
371
372/// Does the union type `$t` have padding?
373///
374/// `$ts` is the list of the type of every field in `$t`. `$t` must be a union
375/// type, or else `union_padding!`'s result may be meaningless.
376///
377/// Note that `union_padding!`'s results are independent of `repr` since they
378/// only consider the size of the type and the sizes of the fields. Whatever the
379/// repr, the size of the type already takes into account any padding that the
380/// compiler has decided to add. Unions with well-defined representations (such
381/// as `repr(C)`) can use this macro to check for padding. Note that while this
382/// may yield some consistent value for some `repr(Rust)` unions, it is not
383/// guaranteed across platforms or compilations.
384#[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
385#[macro_export]
386macro_rules! union_padding {
387    ($t:ty, $_align:expr, $_packed:expr, [$($ts:ty),*]) => {{
388        // The `align` and `packed` directives can be ignored here. Regardless
389        // of if and how they are set, comparing the size of `$t` to each of its
390        // field sizes is a reliable indicator of the presence of padding.
391        let mut max = 0;
392        $({
393            let padding = $crate::util::macro_util::size_of::<$t>() - $crate::util::macro_util::size_of::<$ts>();
394            if padding > max {
395                max = padding;
396            }
397        })*
398        max
399    }};
400}
401
402/// How many padding bytes does the enum type `$t` have?
403///
404/// `$disc` is the type of the enum tag, and `$ts` is a list of fields in each
405/// square-bracket-delimited variant. `$t` must be an enum, or else
406/// `enum_padding!`'s result may be meaningless. An enum has padding if any of
407/// its variant structs [1][2] contain padding, and so all of the variants of an
408/// enum must be "full" in order for the enum to not have padding.
409///
410/// The results of `enum_padding!` require that the enum is not `repr(Rust)`, as
411/// `repr(Rust)` enums may niche the enum's tag and reduce the total number of
412/// bytes required to represent the enum as a result. As long as the enum is
413/// `repr(C)`, `repr(int)`, or `repr(C, int)`, this will consistently return
414/// whether the enum contains any padding bytes.
415///
416/// [1]: https://doc.rust-lang.org/1.81.0/reference/type-layout.html#reprc-enums-with-fields
417/// [2]: https://doc.rust-lang.org/1.81.0/reference/type-layout.html#primitive-representation-of-enums-with-fields
418#[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
419#[macro_export]
420macro_rules! enum_padding {
421    ($t:ty, $_align:expr, $packed:expr, $disc:ty, $([$($ts:ty),*]),*) => {{
422        // The `align` and `packed` directives are irrelevant. `$align` can be
423        // ignored because regardless of if and how it is set, comparing the
424        // size of `$t` to each of its field sizes is a reliable indicator of
425        // the presence of padding. `$packed` is irrelevant because it is
426        // forbidden on enums.
427        #[allow(clippy::as_conversions)]
428        const _: [(); 1] = [(); $packed.is_none() as usize];
429        let mut max = 0;
430        $({
431            let padding = $crate::util::macro_util::size_of::<$t>()
432                - (
433                    $crate::util::macro_util::size_of::<$disc>()
434                    $(+ $crate::util::macro_util::size_of::<$ts>())*
435                );
436            if padding > max {
437                max = padding;
438            }
439        })*
440        max
441    }};
442}
443
444/// Unwraps an infallible `Result`.
445#[doc(hidden)]
446#[macro_export]
447macro_rules! into_inner {
448    ($e:expr) => {
449        match $e {
450            $crate::util::macro_util::core_reexport::result::Result::Ok(e) => e,
451            $crate::util::macro_util::core_reexport::result::Result::Err(i) => match i {},
452        }
453    };
454}
455
456/// Translates an identifier or tuple index into a numeric identifier.
457#[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
458#[macro_export]
459macro_rules! ident_id {
460    ($field:ident) => {
461        $crate::util::macro_util::hash_name(stringify!($field))
462    };
463    ($field:literal) => {
464        $field
465    };
466}
467
468/// Computes the hash of a string.
469///
470/// NOTE(#2749) on hash collisions: This function's output only needs to be
471/// deterministic within a particular compilation. Thus, if a user ever reports
472/// a hash collision (very unlikely given the <= 16-byte special case), we can
473/// strengthen the hash function at that point and publish a new version. Since
474/// this is computed at compile time on small strings, we can easily use more
475/// expensive and higher-quality hash functions if need be.
476#[inline(always)]
477#[must_use]
478#[allow(clippy::as_conversions, clippy::indexing_slicing, clippy::arithmetic_side_effects)]
479pub const fn hash_name(name: &str) -> i128 {
480    let name = name.as_bytes();
481
482    // We guarantee freedom from hash collisions between any two strings of
483    // length 16 or less by having the hashes of such strings be equal to
484    // their value. There is still a possibility that such strings will have
485    // the same value as the hash of a string of length > 16.
486    if name.len() <= size_of::<u128>() {
487        let mut bytes = [0u8; 16];
488
489        let mut i = 0;
490        while i < name.len() {
491            bytes[i] = name[i];
492            i += 1;
493        }
494
495        return i128::from_ne_bytes(bytes);
496    };
497
498    // An implementation of FxHasher, although returning a u128. Probably
499    // not as strong as it could be, but probably more collision resistant
500    // than normal 64-bit FxHasher.
501    let mut hash = 0u128;
502    let mut i = 0;
503    while i < name.len() {
504        // This is just FxHasher's `0x517cc1b727220a95` constant
505        // concatenated back-to-back.
506        const K: u128 = 0x517cc1b727220a95517cc1b727220a95;
507        hash = (hash.rotate_left(5) ^ (name[i] as u128)).wrapping_mul(K);
508        i += 1;
509    }
510    i128::from_ne_bytes(hash.to_ne_bytes())
511}
512
513/// Attempts to transmute `Src` into `Dst`.
514///
515/// A helper for `try_transmute!`.
516///
517/// # Panics
518///
519/// `try_transmute` may either produce a post-monomorphization error or a panic
520/// if `Dst` is bigger than `Src`. Otherwise, `try_transmute` panics under the
521/// same circumstances as [`is_safe`].
522///
523/// [`is_safe`]: TryFromBytes::is_safe
524#[inline(always)]
525pub fn try_transmute<Src, Dst>(src: Src) -> Result<Dst, ValidityError<Src, Dst>>
526where
527    Src: IntoBytes,
528    Dst: TryFromBytes,
529{
530    static_assert!(Src, Dst => mem::size_of::<Dst>() == mem::size_of::<Src>());
531
532    let src = mem::ManuallyDrop::new(ReadOnly::new(src));
533    let mut ptr =
534        Ptr::from_ref(&*src).transmute_with::<ReadOnly<Dst>, Initialized, CastSizedExact, _>();
535    if Dst::is_safe(ptr.reborrow_shared()) {
536        // SAFETY: `read_unaligned` requires that `ptr` be valid for reads and
537        // point to a properly initialized `ReadOnly<Dst>` [1]. `ptr` was
538        // derived from a live reference to `src`; `CastSizedExact` preserves
539        // that reference's address, provenance, and exact byte extent. Thus,
540        // the entire range remains live and valid for reads. The reference is
541        // non-null even when `ReadOnly<Dst>` is zero-sized, and the cast
542        // preserves that address, satisfying [1]'s explicit ZST requirement.
543        //
544        // `Src: IntoBytes` and the `ReadOnly<Src>` bridge let `transmute_with`
545        // establish that every byte in that exact range is initialized.
546        // `Dst::is_safe` then established that the same bytes are bit-valid for
547        // `Dst`, and thus for `ReadOnly<Dst>`. The candidate was a shared
548        // `ReadOnly` pointer, so it could not mutate the bytes; no operation
549        // occurs between validation and this read which could do so. Alignment
550        // is not an additional precondition because [1] expressly permits
551        // unaligned pointers.
552        //
553        // `read_unaligned` leaves the source memory unchanged and makes a
554        // bitwise copy [1]. The source is wrapped in `ManuallyDrop`, which
555        // inhibits its destructor [2]. After this read, neither `src` nor
556        // `ptr` is used to access the inner value, and the wrapper's eventual
557        // drop does not drop it. Thus, only the returned `Dst` is subsequently
558        // used or dropped; there is no duplicate ownership use.
559        //
560        // [1] Per https://doc.rust-lang.org/1.56.0/std/ptr/fn.read_unaligned.html:
561        //
562        //     Reads the value from `src` without moving it. This leaves the
563        //     memory in `src` unchanged.
564        //
565        //     Unlike `read`, `read_unaligned` works with unaligned pointers.
566        //
567        //     Behavior is undefined if any of the following conditions are
568        //     violated:
569        //
570        //     * `src` must be valid for reads.
571        //
572        //     * `src` must point to a properly initialized value of type `T`.
573        //
574        //     Like `read`, `read_unaligned` creates a bitwise copy of `T`,
575        //     regardless of whether `T` is `Copy`. If `T` is not `Copy`, using
576        //     both the returned value and the value at `*src` can violate
577        //     memory safety.
578        //
579        //     Note that even if `T` has size `0`, the pointer must be non-null.
580        //
581        // [2] Per https://doc.rust-lang.org/1.56.0/std/mem/struct.ManuallyDrop.html:
582        //
583        //     A wrapper to inhibit compiler from automatically calling `T`'s
584        //     destructor.
585        let dst = unsafe { core::ptr::read_unaligned(ptr.as_inner().as_ptr()) };
586        Ok(ReadOnly::into_inner(dst))
587    } else {
588        let src = ReadOnly::into_inner(mem::ManuallyDrop::into_inner(src));
589        Err(ValidityError::new(src))
590    }
591}
592
593/// See `try_transmute_ref!` documentation.
594pub trait TryTransmuteRefDst<'a> {
595    type Dst: ?Sized;
596
597    /// See `try_transmute_ref!` documentation.
598    fn try_transmute_ref(self) -> Result<&'a Self::Dst, ValidityError<&'a Self::Src, Self::Dst>>
599    where
600        Self: TryTransmuteRefSrc<'a>,
601        Self::Src: IntoBytes + Immutable + KnownLayout,
602        Self::Dst: TryFromBytes + Immutable + KnownLayout;
603}
604
605pub trait TryTransmuteRefSrc<'a> {
606    type Src: ?Sized;
607}
608
609impl<'a, Src, Dst> TryTransmuteRefSrc<'a> for Wrap<&'a Src, &'a Dst>
610where
611    Src: ?Sized,
612    Dst: ?Sized,
613{
614    type Src = Src;
615}
616
617impl<'a, Src, Dst> TryTransmuteRefDst<'a> for Wrap<&'a Src, &'a Dst>
618where
619    Src: IntoBytes + Immutable + KnownLayout + ?Sized,
620    Dst: TryFromBytes + Immutable + KnownLayout + ?Sized,
621{
622    type Dst = Dst;
623
624    #[inline(always)]
625    fn try_transmute_ref(
626        self,
627    ) -> Result<
628        &'a Dst,
629        ValidityError<&'a <Wrap<&'a Src, &'a Dst> as TryTransmuteRefSrc<'a>>::Src, Dst>,
630    > {
631        let ptr = Ptr::from_ref(self.0);
632        #[rustfmt::skip]
633        let res = ptr.try_with(#[inline(always)] |ptr| {
634            let ptr = ptr.recall_validity::<Initialized, _>();
635            let ptr = ptr.cast::<_, crate::layout::CastFrom<Dst>, _>();
636            ptr.try_into_safe()
637        });
638        match res {
639            Ok(ptr) => {
640                static_assert!(Src: ?Sized + KnownLayout, Dst: ?Sized + KnownLayout => {
641                    Src::LAYOUT.align.get() >= Dst::LAYOUT.align.get()
642                }, "cannot transmute reference when destination type has higher alignment than source type");
643                // SAFETY: We have checked that `Dst` does not have a stricter
644                // alignment requirement than `Src`.
645                let ptr = unsafe { ptr.assume_alignment::<Aligned>() };
646                Ok(ptr.as_ref())
647            }
648            Err(err) => Err(err.map_src(Ptr::as_ref)),
649        }
650    }
651}
652
653pub trait TryTransmuteMutDst<'a> {
654    type Dst: ?Sized;
655
656    /// See `try_transmute_mut!` documentation.
657    fn try_transmute_mut(
658        self,
659    ) -> Result<&'a mut Self::Dst, ValidityError<&'a mut Self::Src, Self::Dst>>
660    where
661        Self: TryTransmuteMutSrc<'a>,
662        Self::Src: IntoBytes,
663        Self::Dst: TryFromBytes;
664}
665
666pub trait TryTransmuteMutSrc<'a> {
667    type Src: ?Sized;
668}
669
670impl<'a, Src, Dst> TryTransmuteMutSrc<'a> for Wrap<&'a mut Src, &'a mut Dst>
671where
672    Src: ?Sized,
673    Dst: ?Sized,
674{
675    type Src = Src;
676}
677
678impl<'a, Src, Dst> TryTransmuteMutDst<'a> for Wrap<&'a mut Src, &'a mut Dst>
679where
680    Src: FromBytes + IntoBytes + KnownLayout + ?Sized,
681    Dst: TryFromBytes + IntoBytes + KnownLayout + ?Sized,
682{
683    type Dst = Dst;
684
685    #[inline(always)]
686    fn try_transmute_mut(
687        self,
688    ) -> Result<
689        &'a mut Dst,
690        ValidityError<&'a mut <Wrap<&'a mut Src, &'a mut Dst> as TryTransmuteMutSrc<'a>>::Src, Dst>,
691    > {
692        let ptr = Ptr::from_mut(self.0);
693        // SAFETY: The provided closure returns the only copy of `ptr`.
694        #[rustfmt::skip]
695        let res = unsafe {
696            ptr.try_with_unchecked(#[inline(always)] |ptr| {
697                let ptr = ptr.recall_validity::<Initialized, (_, (_, _))>();
698                let ptr = ptr.cast::<_, crate::layout::CastFrom<Dst>, _>();
699                ptr.try_into_safe()
700            })
701        };
702        match res {
703            Ok(ptr) => {
704                static_assert!(Src: ?Sized + KnownLayout, Dst: ?Sized + KnownLayout => {
705                    Src::LAYOUT.align.get() >= Dst::LAYOUT.align.get()
706                }, "cannot transmute reference when destination type has higher alignment than source type");
707                // SAFETY: We have checked that `Dst` does not have a stricter
708                // alignment requirement than `Src`.
709                let ptr = unsafe { ptr.assume_alignment::<Aligned>() };
710                Ok(ptr.as_mut())
711            }
712            Err(err) => Err(err.map_src(Ptr::as_mut)),
713        }
714    }
715}
716
717// Used in `transmute_ref!` and friends.
718//
719// This permits us to use the autoref specialization trick to dispatch to
720// associated functions for `transmute_ref` and `transmute_mut` when both `Src`
721// and `Dst` are `Sized`, and to trait methods otherwise. The associated
722// functions, unlike the trait methods, do not require a `KnownLayout` bound.
723// This permits us to add support for transmuting references to unsized types
724// without breaking backwards-compatibility (on v0.8.x) with the old
725// implementation, which did not require a `KnownLayout` bound to transmute
726// sized types.
727#[derive(Copy, Clone)]
728pub struct Wrap<Src, Dst>(pub Src, pub PhantomData<Dst>);
729
730impl<Src, Dst> Wrap<Src, Dst> {
731    #[inline(always)]
732    pub const fn new(src: Src) -> Self {
733        Wrap(src, PhantomData)
734    }
735}
736
737impl<'a, Src, Dst> Wrap<&'a Src, &'a Dst>
738where
739    Src: ?Sized,
740    Dst: ?Sized,
741{
742    #[allow(clippy::must_use_candidate, clippy::missing_inline_in_public_items, clippy::empty_loop)]
743    pub const fn transmute_ref_inference_helper(self) -> &'a Dst {
744        loop {}
745    }
746}
747
748impl<'a, Src, Dst> Wrap<&'a Src, &'a Dst> {
749    /// # Safety
750    /// The caller must guarantee that:
751    /// - `Src: IntoBytes + Immutable`
752    /// - `Dst: FromBytes + Immutable`
753    ///
754    /// # PME
755    ///
756    /// Instantiating this method PMEs unless both:
757    /// - `mem::size_of::<Dst>() == mem::size_of::<Src>()`
758    /// - `mem::align_of::<Dst>() <= mem::align_of::<Src>()`
759    #[inline(always)]
760    #[must_use]
761    pub const unsafe fn transmute_ref(self) -> &'a Dst {
762        static_assert!(Src, Dst => mem::size_of::<Dst>() == mem::size_of::<Src>());
763        static_assert!(Src, Dst => mem::align_of::<Dst>() <= mem::align_of::<Src>());
764
765        let src: *const Src = self.0;
766        let dst = src.cast::<Dst>();
767        // SAFETY:
768        // - We know that it is sound to view the target type of the input
769        //   reference (`Src`) as the target type of the output reference
770        //   (`Dst`) because the caller has guaranteed that `Src: IntoBytes`,
771        //   `Dst: FromBytes`, and `size_of::<Src>() == size_of::<Dst>()`.
772        // - We know that there are no `UnsafeCell`s, and thus we don't have to
773        //   worry about `UnsafeCell` overlap, because `Src: Immutable` and
774        //   `Dst: Immutable`.
775        // - The caller has guaranteed that alignment is not increased.
776        // - We know that the returned lifetime will not outlive the input
777        //   lifetime thanks to the lifetime bounds on this function.
778        //
779        // FIXME(#67): Once our MSRV is 1.58, replace this `transmute` with
780        // `&*dst`.
781        #[allow(clippy::transmute_ptr_to_ref)]
782        unsafe {
783            mem::transmute(dst)
784        }
785    }
786
787    #[inline(always)]
788    pub fn try_transmute_ref(self) -> Result<&'a Dst, ValidityError<&'a Src, Dst>>
789    where
790        Src: IntoBytes + Immutable,
791        Dst: TryFromBytes + Immutable,
792    {
793        static_assert!(Src => mem::align_of::<Src>() == mem::align_of::<Wrapping<Src>>());
794        static_assert!(Dst => mem::align_of::<Dst>() == mem::align_of::<Wrapping<Dst>>());
795
796        // SAFETY: By the preceding assert, `Src` and `Wrapping<Src>` have the
797        // same alignment.
798        let src: &Wrapping<Src> =
799            unsafe { crate::util::transmute_ref::<_, _, BecauseImmutable>(self.0) };
800        let src = Wrap::new(src);
801        <Wrap<&'a Wrapping<Src>, &'a Wrapping<Dst>> as TryTransmuteRefDst<'a>>::try_transmute_ref(
802            src,
803        )
804        .map(
805            // SAFETY: By the preceding assert, `Dst` and `Wrapping<Dst>` have
806            // the same alignment.
807            #[inline(always)]
808            |dst| unsafe { crate::util::transmute_ref::<_, _, BecauseImmutable>(dst) },
809        )
810        .map_err(
811            #[inline(always)]
812            |err| {
813                // SAFETY: By the preceding assert, `Src` and `Wrapping<Src>` have the
814                // same alignment.
815                ValidityError::new(unsafe {
816                    crate::util::transmute_ref::<_, _, BecauseImmutable>(err.into_src())
817                })
818            },
819        )
820    }
821}
822
823impl<'a, Src, Dst> Wrap<&'a mut Src, &'a mut Dst>
824where
825    Src: ?Sized,
826    Dst: ?Sized,
827{
828    #[allow(clippy::must_use_candidate, clippy::missing_inline_in_public_items, clippy::empty_loop)]
829    pub fn transmute_mut_inference_helper(self) -> &'a mut Dst {
830        loop {}
831    }
832}
833
834impl<'a, Src, Dst> Wrap<&'a mut Src, &'a mut Dst> {
835    /// Transmutes a mutable reference of one type to a mutable reference of
836    /// another type.
837    ///
838    /// # PME
839    ///
840    /// Instantiating this method PMEs unless both:
841    /// - `mem::size_of::<Dst>() == mem::size_of::<Src>()`
842    /// - `mem::align_of::<Dst>() <= mem::align_of::<Src>()`
843    #[inline(always)]
844    #[must_use]
845    pub fn transmute_mut(self) -> &'a mut Dst
846    where
847        Src: FromBytes + IntoBytes,
848        Dst: FromBytes + IntoBytes,
849    {
850        static_assert!(Src, Dst => mem::size_of::<Dst>() == mem::size_of::<Src>());
851        static_assert!(Src, Dst => mem::align_of::<Dst>() <= mem::align_of::<Src>());
852
853        let src: *mut Src = self.0;
854        let dst = src.cast::<Dst>();
855        // SAFETY:
856        // - We know that it is sound to view the target type of the input
857        //   reference (`Src`) as the target type of the output reference
858        //   (`Dst`) and vice-versa because `Src: FromBytes + IntoBytes`, `Dst:
859        //   FromBytes + IntoBytes`, and (as asserted above) `size_of::<Src>()
860        //   == size_of::<Dst>()`.
861        // - We asserted above that alignment will not increase.
862        // - We know that the returned lifetime will not outlive the input
863        //   lifetime thanks to the lifetime bounds on this function.
864        unsafe { &mut *dst }
865    }
866
867    #[inline(always)]
868    pub fn try_transmute_mut(self) -> Result<&'a mut Dst, ValidityError<&'a mut Src, Dst>>
869    where
870        Src: FromBytes + IntoBytes,
871        Dst: TryFromBytes + IntoBytes,
872    {
873        static_assert!(Src => mem::align_of::<Src>() == mem::align_of::<Wrapping<Src>>());
874        static_assert!(Dst => mem::align_of::<Dst>() == mem::align_of::<Wrapping<Dst>>());
875
876        // SAFETY: By the preceding assert, `Src` and `Wrapping<Src>` have the
877        // same alignment.
878        let src: &mut Wrapping<Src> =
879            unsafe { crate::util::transmute_mut::<_, _, (_, (_, _))>(self.0) };
880        let src = Wrap::new(src);
881        <Wrap<&'a mut Wrapping<Src>, &'a mut Wrapping<Dst>> as TryTransmuteMutDst<'a>>
882            ::try_transmute_mut(src)
883            // SAFETY: By the preceding assert, `Dst` and `Wrapping<Dst>` have the
884            // same alignment.
885            .map(|dst| unsafe { crate::util::transmute_mut::<_, _, (_, (_, _))>(dst) })
886            .map_err(|err| {
887                // SAFETY: By the preceding assert, `Src` and `Wrapping<Src>` have the
888                // same alignment.
889                ValidityError::new(unsafe {
890                    crate::util::transmute_mut::<_, _, (_, (_, _))>(err.into_src())
891                })
892            })
893    }
894}
895
896pub trait TransmuteRefDst<'a> {
897    type Dst: ?Sized;
898
899    #[must_use]
900    fn transmute_ref(self) -> &'a Self::Dst;
901}
902
903impl<'a, Src: ?Sized, Dst: ?Sized> TransmuteRefDst<'a> for Wrap<&'a Src, &'a Dst>
904where
905    Src: KnownLayout + IntoBytes + Immutable,
906    Dst: KnownLayout<PointerMetadata = usize> + FromBytes + Immutable,
907{
908    type Dst = Dst;
909
910    #[inline(always)]
911    fn transmute_ref(self) -> &'a Dst {
912        let ptr = Ptr::from_ref(self.0)
913            .recall_validity::<Initialized, _>()
914            .transmute_with::<Dst, Initialized, crate::layout::CastFrom<Dst>, (crate::pointer::BecauseMutationCompatible, _)>()
915            .recall_validity::<Safe, _>();
916
917        static_assert!(Src: ?Sized + KnownLayout, Dst: ?Sized + KnownLayout => {
918            Src::LAYOUT.align.get() >= Dst::LAYOUT.align.get()
919        }, "cannot transmute reference when destination type has higher alignment than source type");
920
921        // SAFETY: The preceding `static_assert!` ensures that
922        // `Src::LAYOUT.align >= Dst::LAYOUT.align`. Since `self` is
923        // validly-aligned for `Src`, it is also validly-aligned for `Dst`.
924        let ptr = unsafe { ptr.assume_alignment() };
925
926        ptr.as_ref()
927    }
928}
929
930pub trait TransmuteMutDst<'a> {
931    type Dst: ?Sized;
932    #[must_use]
933    fn transmute_mut(self) -> &'a mut Self::Dst;
934}
935
936impl<'a, Src: ?Sized, Dst: ?Sized> TransmuteMutDst<'a> for Wrap<&'a mut Src, &'a mut Dst>
937where
938    Src: KnownLayout + FromBytes + IntoBytes,
939    Dst: KnownLayout<PointerMetadata = usize> + FromBytes + IntoBytes,
940{
941    type Dst = Dst;
942
943    #[inline(always)]
944    fn transmute_mut(self) -> &'a mut Dst {
945        let ptr = Ptr::from_mut(self.0)
946            .recall_validity::<Initialized, (_, (_, _))>()
947            .transmute_with::<Dst, Initialized, crate::layout::CastFrom<Dst>, _>()
948            .recall_validity::<Safe, (_, (_, _))>();
949
950        static_assert!(Src: ?Sized + KnownLayout, Dst: ?Sized + KnownLayout => {
951            Src::LAYOUT.align.get() >= Dst::LAYOUT.align.get()
952        }, "cannot transmute reference when destination type has higher alignment than source type");
953
954        // SAFETY: The preceding `static_assert!` ensures that
955        // `Src::LAYOUT.align >= Dst::LAYOUT.align`. Since `self` is
956        // validly-aligned for `Src`, it is also validly-aligned for `Dst`.
957        let ptr = unsafe { ptr.assume_alignment() };
958
959        ptr.as_mut()
960    }
961}
962
963/// A function which emits a warning if its return value is not used.
964#[must_use]
965#[inline(always)]
966pub const fn must_use<T>(t: T) -> T {
967    t
968}
969
970// NOTE: We can't change this to a `pub use core as core_reexport` until [1] is
971// fixed or we update to a semver-breaking version (as of this writing, 0.8.0)
972// on the `main` branch.
973//
974// [1] https://github.com/obi1kenobi/cargo-semver-checks/issues/573
975pub mod core_reexport {
976    pub use core::*;
977
978    pub mod mem {
979        pub use core::mem::*;
980    }
981}
982
983/// Projects `Self` to itself.
984///
985/// This is used by macros to have the compiler verify that two syntactically
986/// similar types actually resolve to the same type. The blanket implementation
987/// ensures that no other implementation can project a type to anything other
988/// than itself.
989#[doc(hidden)]
990pub trait Identity {
991    /// `Self`.
992    type Type: ?Sized;
993}
994
995impl<T: ?Sized> Identity for T {
996    type Type = T;
997}
998
999#[cfg(test)]
1000mod tests {
1001    use core::num::NonZeroUsize;
1002
1003    use crate::util::testutil::*;
1004
1005    #[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
1006    mod nightly {
1007        use super::super::*;
1008        use crate::util::testutil::*;
1009
1010        // FIXME(#29), FIXME(https://github.com/rust-lang/rust/issues/69835):
1011        // Remove this `cfg` when `size_of_val_raw` is stabilized.
1012        #[allow(clippy::decimal_literal_representation)]
1013        #[test]
1014        fn test_trailing_field_offset() {
1015            assert_eq!(mem::align_of::<Aligned64kAllocation>(), _64K);
1016
1017            macro_rules! test {
1018                (#[$cfg:meta] ($($ts:ty),* ; $trailing_field_ty:ty) => $expect:expr) => {{
1019                    #[$cfg]
1020                    struct Test($(#[allow(dead_code)] $ts,)* #[allow(dead_code)] $trailing_field_ty);
1021                    assert_eq!(test!(@offset $($ts),* ; $trailing_field_ty), $expect);
1022                }};
1023                (#[$cfg:meta] $(#[$cfgs:meta])* ($($ts:ty),* ; $trailing_field_ty:ty) => $expect:expr) => {
1024                    test!(#[$cfg] ($($ts),* ; $trailing_field_ty) => $expect);
1025                    test!($(#[$cfgs])* ($($ts),* ; $trailing_field_ty) => $expect);
1026                };
1027                (@offset ; $_trailing:ty) => { trailing_field_offset!(Test, 0) };
1028                (@offset $_t:ty ; $_trailing:ty) => { trailing_field_offset!(Test, 1) };
1029            }
1030
1031            test!(#[repr(C)] #[repr(transparent)] #[repr(packed)](; u8) => Some(0));
1032            test!(#[repr(C)] #[repr(transparent)] #[repr(packed)](; [u8]) => Some(0));
1033            test!(#[repr(C)] #[repr(C, packed)] (u8; u8) => Some(1));
1034            test!(#[repr(C)] (; AU64) => Some(0));
1035            test!(#[repr(C)] (; [AU64]) => Some(0));
1036            test!(#[repr(C)] (u8; AU64) => Some(8));
1037            test!(#[repr(C)] (u8; [AU64]) => Some(8));
1038
1039            #[derive(
1040                Immutable, FromBytes, Eq, PartialEq, Ord, PartialOrd, Default, Debug, Copy, Clone,
1041            )]
1042            #[repr(C)]
1043            pub(crate) struct Nested<T, U: ?Sized> {
1044                _t: T,
1045                _u: U,
1046            }
1047
1048            test!(#[repr(C)] (; Nested<u8, AU64>) => Some(0));
1049            test!(#[repr(C)] (; Nested<u8, [AU64]>) => Some(0));
1050            test!(#[repr(C)] (u8; Nested<u8, AU64>) => Some(8));
1051            test!(#[repr(C)] (u8; Nested<u8, [AU64]>) => Some(8));
1052
1053            // Test that `packed(N)` limits the offset of the trailing field.
1054            test!(#[repr(C, packed(        1))] (u8; elain::Align<        2>) => Some(        1));
1055            test!(#[repr(C, packed(        2))] (u8; elain::Align<        4>) => Some(        2));
1056            test!(#[repr(C, packed(        4))] (u8; elain::Align<        8>) => Some(        4));
1057            test!(#[repr(C, packed(        8))] (u8; elain::Align<       16>) => Some(        8));
1058            test!(#[repr(C, packed(       16))] (u8; elain::Align<       32>) => Some(       16));
1059            test!(#[repr(C, packed(       32))] (u8; elain::Align<       64>) => Some(       32));
1060            test!(#[repr(C, packed(       64))] (u8; elain::Align<      128>) => Some(       64));
1061            test!(#[repr(C, packed(      128))] (u8; elain::Align<      256>) => Some(      128));
1062            test!(#[repr(C, packed(      256))] (u8; elain::Align<      512>) => Some(      256));
1063            test!(#[repr(C, packed(      512))] (u8; elain::Align<     1024>) => Some(      512));
1064            test!(#[repr(C, packed(     1024))] (u8; elain::Align<     2048>) => Some(     1024));
1065            test!(#[repr(C, packed(     2048))] (u8; elain::Align<     4096>) => Some(     2048));
1066            test!(#[repr(C, packed(     4096))] (u8; elain::Align<     8192>) => Some(     4096));
1067            test!(#[repr(C, packed(     8192))] (u8; elain::Align<    16384>) => Some(     8192));
1068            test!(#[repr(C, packed(    16384))] (u8; elain::Align<    32768>) => Some(    16384));
1069            test!(#[repr(C, packed(    32768))] (u8; elain::Align<    65536>) => Some(    32768));
1070            test!(#[repr(C, packed(    65536))] (u8; elain::Align<   131072>) => Some(    65536));
1071            /* Alignments above 65536 are not yet supported.
1072            test!(#[repr(C, packed(   131072))] (u8; elain::Align<   262144>) => Some(   131072));
1073            test!(#[repr(C, packed(   262144))] (u8; elain::Align<   524288>) => Some(   262144));
1074            test!(#[repr(C, packed(   524288))] (u8; elain::Align<  1048576>) => Some(   524288));
1075            test!(#[repr(C, packed(  1048576))] (u8; elain::Align<  2097152>) => Some(  1048576));
1076            test!(#[repr(C, packed(  2097152))] (u8; elain::Align<  4194304>) => Some(  2097152));
1077            test!(#[repr(C, packed(  4194304))] (u8; elain::Align<  8388608>) => Some(  4194304));
1078            test!(#[repr(C, packed(  8388608))] (u8; elain::Align< 16777216>) => Some(  8388608));
1079            test!(#[repr(C, packed( 16777216))] (u8; elain::Align< 33554432>) => Some( 16777216));
1080            test!(#[repr(C, packed( 33554432))] (u8; elain::Align< 67108864>) => Some( 33554432));
1081            test!(#[repr(C, packed( 67108864))] (u8; elain::Align< 33554432>) => Some( 67108864));
1082            test!(#[repr(C, packed( 33554432))] (u8; elain::Align<134217728>) => Some( 33554432));
1083            test!(#[repr(C, packed(134217728))] (u8; elain::Align<268435456>) => Some(134217728));
1084            test!(#[repr(C, packed(268435456))] (u8; elain::Align<268435456>) => Some(268435456));
1085            */
1086
1087            // Test that `align(N)` does not limit the offset of the trailing field.
1088            test!(#[repr(C, align(        1))] (u8; elain::Align<        2>) => Some(        2));
1089            test!(#[repr(C, align(        2))] (u8; elain::Align<        4>) => Some(        4));
1090            test!(#[repr(C, align(        4))] (u8; elain::Align<        8>) => Some(        8));
1091            test!(#[repr(C, align(        8))] (u8; elain::Align<       16>) => Some(       16));
1092            test!(#[repr(C, align(       16))] (u8; elain::Align<       32>) => Some(       32));
1093            test!(#[repr(C, align(       32))] (u8; elain::Align<       64>) => Some(       64));
1094            test!(#[repr(C, align(       64))] (u8; elain::Align<      128>) => Some(      128));
1095            test!(#[repr(C, align(      128))] (u8; elain::Align<      256>) => Some(      256));
1096            test!(#[repr(C, align(      256))] (u8; elain::Align<      512>) => Some(      512));
1097            test!(#[repr(C, align(      512))] (u8; elain::Align<     1024>) => Some(     1024));
1098            test!(#[repr(C, align(     1024))] (u8; elain::Align<     2048>) => Some(     2048));
1099            test!(#[repr(C, align(     2048))] (u8; elain::Align<     4096>) => Some(     4096));
1100            test!(#[repr(C, align(     4096))] (u8; elain::Align<     8192>) => Some(     8192));
1101            test!(#[repr(C, align(     8192))] (u8; elain::Align<    16384>) => Some(    16384));
1102            test!(#[repr(C, align(    16384))] (u8; elain::Align<    32768>) => Some(    32768));
1103            test!(#[repr(C, align(    32768))] (u8; elain::Align<    65536>) => Some(    65536));
1104            /* Alignments above 65536 are not yet supported.
1105            test!(#[repr(C, align(    65536))] (u8; elain::Align<   131072>) => Some(   131072));
1106            test!(#[repr(C, align(   131072))] (u8; elain::Align<   262144>) => Some(   262144));
1107            test!(#[repr(C, align(   262144))] (u8; elain::Align<   524288>) => Some(   524288));
1108            test!(#[repr(C, align(   524288))] (u8; elain::Align<  1048576>) => Some(  1048576));
1109            test!(#[repr(C, align(  1048576))] (u8; elain::Align<  2097152>) => Some(  2097152));
1110            test!(#[repr(C, align(  2097152))] (u8; elain::Align<  4194304>) => Some(  4194304));
1111            test!(#[repr(C, align(  4194304))] (u8; elain::Align<  8388608>) => Some(  8388608));
1112            test!(#[repr(C, align(  8388608))] (u8; elain::Align< 16777216>) => Some( 16777216));
1113            test!(#[repr(C, align( 16777216))] (u8; elain::Align< 33554432>) => Some( 33554432));
1114            test!(#[repr(C, align( 33554432))] (u8; elain::Align< 67108864>) => Some( 67108864));
1115            test!(#[repr(C, align( 67108864))] (u8; elain::Align< 33554432>) => Some( 33554432));
1116            test!(#[repr(C, align( 33554432))] (u8; elain::Align<134217728>) => Some(134217728));
1117            test!(#[repr(C, align(134217728))] (u8; elain::Align<268435456>) => Some(268435456));
1118            */
1119        }
1120
1121        // FIXME(#29), FIXME(https://github.com/rust-lang/rust/issues/69835):
1122        // Remove this `cfg` when `size_of_val_raw` is stabilized.
1123        #[allow(clippy::decimal_literal_representation)]
1124        #[test]
1125        fn test_align_of_dst() {
1126            // Test that `align_of!` correctly computes the alignment of DSTs.
1127            assert_eq!(align_of!([elain::Align<1>]), Some(1));
1128            assert_eq!(align_of!([elain::Align<2>]), Some(2));
1129            assert_eq!(align_of!([elain::Align<4>]), Some(4));
1130            assert_eq!(align_of!([elain::Align<8>]), Some(8));
1131            assert_eq!(align_of!([elain::Align<16>]), Some(16));
1132            assert_eq!(align_of!([elain::Align<32>]), Some(32));
1133            assert_eq!(align_of!([elain::Align<64>]), Some(64));
1134            assert_eq!(align_of!([elain::Align<128>]), Some(128));
1135            assert_eq!(align_of!([elain::Align<256>]), Some(256));
1136            assert_eq!(align_of!([elain::Align<512>]), Some(512));
1137            assert_eq!(align_of!([elain::Align<1024>]), Some(1024));
1138            assert_eq!(align_of!([elain::Align<2048>]), Some(2048));
1139            assert_eq!(align_of!([elain::Align<4096>]), Some(4096));
1140            assert_eq!(align_of!([elain::Align<8192>]), Some(8192));
1141            assert_eq!(align_of!([elain::Align<16384>]), Some(16384));
1142            assert_eq!(align_of!([elain::Align<32768>]), Some(32768));
1143            assert_eq!(align_of!([elain::Align<65536>]), Some(65536));
1144            /* Alignments above 65536 are not yet supported.
1145            assert_eq!(align_of!([elain::Align<131072>]), Some(131072));
1146            assert_eq!(align_of!([elain::Align<262144>]), Some(262144));
1147            assert_eq!(align_of!([elain::Align<524288>]), Some(524288));
1148            assert_eq!(align_of!([elain::Align<1048576>]), Some(1048576));
1149            assert_eq!(align_of!([elain::Align<2097152>]), Some(2097152));
1150            assert_eq!(align_of!([elain::Align<4194304>]), Some(4194304));
1151            assert_eq!(align_of!([elain::Align<8388608>]), Some(8388608));
1152            assert_eq!(align_of!([elain::Align<16777216>]), Some(16777216));
1153            assert_eq!(align_of!([elain::Align<33554432>]), Some(33554432));
1154            assert_eq!(align_of!([elain::Align<67108864>]), Some(67108864));
1155            assert_eq!(align_of!([elain::Align<33554432>]), Some(33554432));
1156            assert_eq!(align_of!([elain::Align<134217728>]), Some(134217728));
1157            assert_eq!(align_of!([elain::Align<268435456>]), Some(268435456));
1158            */
1159        }
1160    }
1161
1162    #[test]
1163    fn test_enum_casts() {
1164        // Test that casting the variants of enums with signed integer reprs to
1165        // unsigned integers obeys expected signed -> unsigned casting rules.
1166
1167        #[repr(i8)]
1168        enum ReprI8 {
1169            MinusOne = -1,
1170            Zero = 0,
1171            Min = i8::MIN,
1172            Max = i8::MAX,
1173        }
1174
1175        #[allow(clippy::as_conversions)]
1176        let x = ReprI8::MinusOne as u8;
1177        assert_eq!(x, u8::MAX);
1178
1179        #[allow(clippy::as_conversions)]
1180        let x = ReprI8::Zero as u8;
1181        assert_eq!(x, 0);
1182
1183        #[allow(clippy::as_conversions)]
1184        let x = ReprI8::Min as u8;
1185        assert_eq!(x, 128);
1186
1187        #[allow(clippy::as_conversions)]
1188        let x = ReprI8::Max as u8;
1189        assert_eq!(x, 127);
1190    }
1191
1192    #[test]
1193    fn test_struct_padding() {
1194        // Test that, for each provided repr, `struct_padding!` reports the
1195        // expected value.
1196        macro_rules! test {
1197            (#[$cfg:meta] ($($ts:ty),*) => $expect:expr) => {{
1198                #[$cfg]
1199                #[allow(dead_code)]
1200                struct Test($($ts),*);
1201                assert_eq!(struct_padding!(Test, None::<NonZeroUsize>, None::<NonZeroUsize>, [$($ts),*]), $expect);
1202            }};
1203            (#[$cfg:meta] $(#[$cfgs:meta])* ($($ts:ty),*) => $expect:expr) => {
1204                test!(#[$cfg] ($($ts),*) => $expect);
1205                test!($(#[$cfgs])* ($($ts),*) => $expect);
1206            };
1207        }
1208
1209        test!(#[repr(C)] #[repr(transparent)] #[repr(packed)] () => 0);
1210        test!(#[repr(C)] #[repr(transparent)] #[repr(packed)] (u8) => 0);
1211        test!(#[repr(C)] #[repr(transparent)] #[repr(packed)] (u8, ()) => 0);
1212        test!(#[repr(C)] #[repr(packed)] (u8, u8) => 0);
1213
1214        test!(#[repr(C)] (u8, AU64) => 7);
1215        // Rust won't let you put `#[repr(packed)]` on a type which contains a
1216        // `#[repr(align(n > 1))]` type (`AU64`), so we have to use `u64` here.
1217        // It's not ideal, but it definitely has align > 1 on /some/ of our CI
1218        // targets, and this isn't a particularly complex macro we're testing
1219        // anyway.
1220        test!(#[repr(packed)] (u8, u64) => 0);
1221    }
1222
1223    #[test]
1224    fn test_repr_c_struct_padding() {
1225        // Test that, for each provided repr, `repr_c_struct_padding!` reports
1226        // the expected value.
1227        macro_rules! test {
1228            (($($ts:tt),*) => $expect:expr) => {{
1229                #[repr(C)]
1230                #[allow(dead_code)]
1231                struct Test($($ts),*);
1232                assert_eq!(repr_c_struct_has_padding!(Test, None::<NonZeroUsize>, None::<NonZeroUsize>, [$($ts),*]), $expect);
1233            }};
1234        }
1235
1236        // Test static padding
1237        test!(() => false);
1238        test!(([u8]) => false);
1239        test!((u8) => false);
1240        test!((u8, [u8]) => false);
1241        test!((u8, ()) => false);
1242        test!((u8, (), [u8]) => false);
1243        test!((u8, u8) => false);
1244        test!((u8, u8, [u8]) => false);
1245
1246        test!((u8, AU64) => true);
1247        test!((u8, AU64, [u8]) => true);
1248
1249        // Test dynamic padding
1250        test!((AU64, [AU64]) => false);
1251        test!((u8, [AU64]) => true);
1252
1253        #[repr(align(4))]
1254        struct AU32(#[allow(unused)] u32);
1255        test!((AU64, [AU64]) => false);
1256        test!((AU64, [AU32]) => true);
1257    }
1258
1259    #[test]
1260    fn test_union_padding() {
1261        // Test that, for each provided repr, `union_padding!` reports the
1262        // expected value.
1263        macro_rules! test {
1264            (#[$cfg:meta] {$($fs:ident: $ts:ty),*} => $expect:expr) => {{
1265                #[$cfg]
1266                #[allow(unused)] // fields are never read
1267                union Test{ $($fs: $ts),* }
1268                assert_eq!(union_padding!(Test, None::<NonZeroUsize>, None::<usize>, [$($ts),*]), $expect);
1269            }};
1270            (#[$cfg:meta] $(#[$cfgs:meta])* {$($fs:ident: $ts:ty),*} => $expect:expr) => {
1271                test!(#[$cfg] {$($fs: $ts),*} => $expect);
1272                test!($(#[$cfgs])* {$($fs: $ts),*} => $expect);
1273            };
1274        }
1275
1276        test!(#[repr(C)] #[repr(packed)] {a: u8} => 0);
1277        test!(#[repr(C)] #[repr(packed)] {a: u8, b: u8} => 0);
1278
1279        // Rust won't let you put `#[repr(packed)]` on a type which contains a
1280        // `#[repr(align(n > 1))]` type (`AU64`), so we have to use `u64` here.
1281        // It's not ideal, but it definitely has align > 1 on /some/ of our CI
1282        // targets, and this isn't a particularly complex macro we're testing
1283        // anyway.
1284        test!(#[repr(C)] #[repr(packed)] {a: u8, b: u64} => 7);
1285    }
1286
1287    #[test]
1288    fn test_enum_padding() {
1289        // Test that, for each provided repr, `enum_has_padding!` reports the
1290        // expected value.
1291        macro_rules! test {
1292            (#[repr($disc:ident $(, $c:ident)?)] { $($vs:ident ($($ts:ty),*),)* } => $expect:expr) => {
1293                test!(@case #[repr($disc $(, $c)?)] { $($vs ($($ts),*),)* } => $expect);
1294            };
1295            (#[repr($disc:ident $(, $c:ident)?)] #[$cfg:meta] $(#[$cfgs:meta])* { $($vs:ident ($($ts:ty),*),)* } => $expect:expr) => {
1296                test!(@case #[repr($disc $(, $c)?)] #[$cfg] { $($vs ($($ts),*),)* } => $expect);
1297                test!(#[repr($disc $(, $c)?)] $(#[$cfgs])* { $($vs ($($ts),*),)* } => $expect);
1298            };
1299            (@case #[repr($disc:ident $(, $c:ident)?)] $(#[$cfg:meta])? { $($vs:ident ($($ts:ty),*),)* } => $expect:expr) => {{
1300                #[repr($disc $(, $c)?)]
1301                $(#[$cfg])?
1302                #[allow(unused)] // variants and fields are never used
1303                enum Test {
1304                    $($vs ($($ts),*),)*
1305                }
1306                assert_eq!(
1307                    enum_padding!(Test, None::<NonZeroUsize>, None::<NonZeroUsize>, $disc, $([$($ts),*]),*),
1308                    $expect
1309                );
1310            }};
1311        }
1312
1313        #[allow(unused)]
1314        #[repr(align(2))]
1315        struct U16(u16);
1316
1317        #[allow(unused)]
1318        #[repr(align(4))]
1319        struct U32(u32);
1320
1321        test!(#[repr(u8)] #[repr(C)] {
1322            A(u8),
1323        } => 0);
1324        test!(#[repr(u16)] #[repr(C)] {
1325            A(u8, u8),
1326            B(U16),
1327        } => 0);
1328        test!(#[repr(u32)] #[repr(C)] {
1329            A(u8, u8, u8, u8),
1330            B(U16, u8, u8),
1331            C(u8, u8, U16),
1332            D(U16, U16),
1333            E(U32),
1334        } => 0);
1335
1336        // `repr(int)` can pack the discriminant more efficiently
1337        test!(#[repr(u8)] {
1338            A(u8, U16),
1339        } => 0);
1340        test!(#[repr(u8)] {
1341            A(u8, U16, U32),
1342        } => 0);
1343
1344        // `repr(C)` cannot
1345        test!(#[repr(u8, C)] {
1346            A(u8, U16),
1347        } => 2);
1348        test!(#[repr(u8, C)] {
1349            A(u8, u8, u8, U32),
1350        } => 4);
1351
1352        // And field ordering can always cause problems
1353        test!(#[repr(u8)] #[repr(C)] {
1354            A(U16, u8),
1355        } => 2);
1356        test!(#[repr(u8)] #[repr(C)] {
1357            A(U32, u8, u8, u8),
1358        } => 4);
1359    }
1360}