• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Copyright 2022 The Fuchsia Authors
2 //
3 // Licensed under a BSD-style license <LICENSE-BSD>, Apache License, Version 2.0
4 // <LICENSE-APACHE or https://www.apache.org/licenses/LICENSE-2.0>, or the MIT
5 // license <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your option.
6 // This file may not be copied, modified, or distributed except according to
7 // those terms.
8 
9 //! Utilities used by macros and by `zerocopy-derive`.
10 //!
11 //! These are defined here `zerocopy` rather than in code generated by macros or
12 //! by `zerocopy-derive` so that they can be compiled once rather than
13 //! recompiled for every invocation (e.g., if they were defined in generated
14 //! code, then deriving `IntoBytes` and `FromBytes` on three different types
15 //! would result in the code in question being emitted and compiled six
16 //! different times).
17 
18 #![allow(missing_debug_implementations)]
19 
20 use core::mem::{self, ManuallyDrop};
21 
22 // TODO(#29), TODO(https://github.com/rust-lang/rust/issues/69835): Remove this
23 // `cfg` when `size_of_val_raw` is stabilized.
24 #[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
25 use core::ptr::{self, NonNull};
26 
27 use crate::{
28     pointer::invariant::{self, BecauseExclusive, BecauseImmutable, Invariants},
29     FromBytes, Immutable, IntoBytes, Ptr, TryFromBytes, ValidityError,
30 };
31 
32 /// Projects the type of the field at `Index` in `Self`.
33 ///
34 /// The `Index` parameter is any sort of handle that identifies the field; its
35 /// definition is the obligation of the implementer.
36 ///
37 /// # Safety
38 ///
39 /// Unsafe code may assume that this accurately reflects the definition of
40 /// `Self`.
41 pub unsafe trait Field<Index> {
42     /// The type of the field at `Index`.
43     type Type: ?Sized;
44 }
45 
46 #[cfg_attr(
47     zerocopy_diagnostic_on_unimplemented_1_78_0,
48     diagnostic::on_unimplemented(
49         message = "`{T}` has inter-field padding",
50         label = "types with padding cannot implement `IntoBytes`",
51         note = "consider using `zerocopy::Unalign` to lower the alignment of individual fields",
52         note = "consider adding explicit fields where padding would be",
53         note = "consider using `#[repr(packed)]` to remove inter-field padding"
54     )
55 )]
56 pub trait PaddingFree<T: ?Sized, const HAS_PADDING: bool> {}
57 impl<T: ?Sized> PaddingFree<T, false> for () {}
58 
59 /// A type whose size is equal to `align_of::<T>()`.
60 #[repr(C)]
61 pub struct AlignOf<T> {
62     // This field ensures that:
63     // - The size is always at least 1 (the minimum possible alignment).
64     // - If the alignment is greater than 1, Rust has to round up to the next
65     //   multiple of it in order to make sure that `Align`'s size is a multiple
66     //   of that alignment. Without this field, its size could be 0, which is a
67     //   valid multiple of any alignment.
68     _u: u8,
69     _a: [T; 0],
70 }
71 
72 impl<T> AlignOf<T> {
73     #[inline(never)] // Make `missing_inline_in_public_items` happy.
74     #[cfg_attr(
75         all(coverage_nightly, __ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS),
76         coverage(off)
77     )]
into_t(self) -> T78     pub fn into_t(self) -> T {
79         unreachable!()
80     }
81 }
82 
83 /// A type whose size is equal to `max(align_of::<T>(), align_of::<U>())`.
84 #[repr(C)]
85 pub union MaxAlignsOf<T, U> {
86     _t: ManuallyDrop<AlignOf<T>>,
87     _u: ManuallyDrop<AlignOf<U>>,
88 }
89 
90 impl<T, U> MaxAlignsOf<T, U> {
91     #[inline(never)] // Make `missing_inline_in_public_items` happy.
92     #[cfg_attr(
93         all(coverage_nightly, __ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS),
94         coverage(off)
95     )]
new(_t: T, _u: U) -> MaxAlignsOf<T, U>96     pub fn new(_t: T, _u: U) -> MaxAlignsOf<T, U> {
97         unreachable!()
98     }
99 }
100 
101 #[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
102 const _64K: usize = 1 << 16;
103 
104 // TODO(#29), TODO(https://github.com/rust-lang/rust/issues/69835): Remove this
105 // `cfg` when `size_of_val_raw` is stabilized.
106 #[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
107 #[repr(C, align(65536))]
108 struct Aligned64kAllocation([u8; _64K]);
109 
110 /// A pointer to an aligned allocation of size 2^16.
111 ///
112 /// # Safety
113 ///
114 /// `ALIGNED_64K_ALLOCATION` is guaranteed to point to the entirety of an
115 /// allocation with size and alignment 2^16, and to have valid provenance.
116 // TODO(#29), TODO(https://github.com/rust-lang/rust/issues/69835): Remove this
117 // `cfg` when `size_of_val_raw` is stabilized.
118 #[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
119 pub const ALIGNED_64K_ALLOCATION: NonNull<[u8]> = {
120     const REF: &Aligned64kAllocation = &Aligned64kAllocation([0; _64K]);
121     let ptr: *const Aligned64kAllocation = REF;
122     let ptr: *const [u8] = ptr::slice_from_raw_parts(ptr.cast(), _64K);
123     // SAFETY:
124     // - `ptr` is derived from a Rust reference, which is guaranteed to be
125     //   non-null.
126     // - `ptr` is derived from an `&Aligned64kAllocation`, which has size and
127     //   alignment `_64K` as promised. Its length is initialized to `_64K`,
128     //   which means that it refers to the entire allocation.
129     // - `ptr` is derived from a Rust reference, which is guaranteed to have
130     //   valid provenance.
131     //
132     // TODO(#429): Once `NonNull::new_unchecked` docs document that it preserves
133     // provenance, cite those docs.
134     // TODO: Replace this `as` with `ptr.cast_mut()` once our MSRV >= 1.65
135     #[allow(clippy::as_conversions)]
136     unsafe {
137         NonNull::new_unchecked(ptr as *mut _)
138     }
139 };
140 
141 /// Computes the offset of the base of the field `$trailing_field_name` within
142 /// the type `$ty`.
143 ///
144 /// `trailing_field_offset!` produces code which is valid in a `const` context.
145 // TODO(#29), TODO(https://github.com/rust-lang/rust/issues/69835): Remove this
146 // `cfg` when `size_of_val_raw` is stabilized.
147 #[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
148 #[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
149 #[macro_export]
150 macro_rules! trailing_field_offset {
151     ($ty:ty, $trailing_field_name:tt) => {{
152         let min_size = {
153             let zero_elems: *const [()] =
154                 $crate::util::macro_util::core_reexport::ptr::slice_from_raw_parts(
155                     // Work around https://github.com/rust-lang/rust-clippy/issues/12280
156                     #[allow(clippy::incompatible_msrv)]
157                     $crate::util::macro_util::core_reexport::ptr::NonNull::<()>::dangling()
158                         .as_ptr()
159                         .cast_const(),
160                     0,
161                 );
162             // SAFETY:
163             // - If `$ty` is `Sized`, `size_of_val_raw` is always safe to call.
164             // - Otherwise:
165             //   - If `$ty` is not a slice DST, this pointer conversion will
166             //     fail due to "mismatched vtable kinds", and compilation will
167             //     fail.
168             //   - If `$ty` is a slice DST, we have constructed `zero_elems` to
169             //     have zero trailing slice elements. Per the `size_of_val_raw`
170             //     docs, "For the special case where the dynamic tail length is
171             //     0, this function is safe to call." [1]
172             //
173             // [1] https://doc.rust-lang.org/nightly/std/mem/fn.size_of_val_raw.html
174             unsafe {
175                 #[allow(clippy::as_conversions)]
176                 $crate::util::macro_util::core_reexport::mem::size_of_val_raw(
177                     zero_elems as *const $ty,
178                 )
179             }
180         };
181 
182         assert!(min_size <= _64K);
183 
184         #[allow(clippy::as_conversions)]
185         let ptr = ALIGNED_64K_ALLOCATION.as_ptr() as *const $ty;
186 
187         // SAFETY:
188         // - Thanks to the preceding `assert!`, we know that the value with zero
189         //   elements fits in `_64K` bytes, and thus in the allocation addressed
190         //   by `ALIGNED_64K_ALLOCATION`. The offset of the trailing field is
191         //   guaranteed to be no larger than this size, so this field projection
192         //   is guaranteed to remain in-bounds of its allocation.
193         // - Because the minimum size is no larger than `_64K` bytes, and
194         //   because an object's size must always be a multiple of its alignment
195         //   [1], we know that `$ty`'s alignment is no larger than `_64K`. The
196         //   allocation addressed by `ALIGNED_64K_ALLOCATION` is guaranteed to
197         //   be aligned to `_64K`, so `ptr` is guaranteed to satisfy `$ty`'s
198         //   alignment.
199         // - As required by `addr_of!`, we do not write through `field`.
200         //
201         //   Note that, as of [2], this requirement is technically unnecessary
202         //   for Rust versions >= 1.75.0, but no harm in guaranteeing it anyway
203         //   until we bump our MSRV.
204         //
205         // [1] Per https://doc.rust-lang.org/reference/type-layout.html:
206         //
207         //   The size of a value is always a multiple of its alignment.
208         //
209         // [2] https://github.com/rust-lang/reference/pull/1387
210         let field = unsafe {
211             $crate::util::macro_util::core_reexport::ptr::addr_of!((*ptr).$trailing_field_name)
212         };
213         // SAFETY:
214         // - Both `ptr` and `field` are derived from the same allocated object.
215         // - By the preceding safety comment, `field` is in bounds of that
216         //   allocated object.
217         // - The distance, in bytes, between `ptr` and `field` is required to be
218         //   a multiple of the size of `u8`, which is trivially true because
219         //   `u8`'s size is 1.
220         // - The distance, in bytes, cannot overflow `isize`. This is guaranteed
221         //   because no allocated object can have a size larger than can fit in
222         //   `isize`. [1]
223         // - The distance being in-bounds cannot rely on wrapping around the
224         //   address space. This is guaranteed because the same is guaranteed of
225         //   allocated objects. [1]
226         //
227         // [1] TODO(#429), TODO(https://github.com/rust-lang/rust/pull/116675):
228         //     Once these are guaranteed in the Reference, cite it.
229         let offset = unsafe { field.cast::<u8>().offset_from(ptr.cast::<u8>()) };
230         // Guaranteed not to be lossy: `field` comes after `ptr`, so the offset
231         // from `ptr` to `field` is guaranteed to be positive.
232         assert!(offset >= 0);
233         Some(
234             #[allow(clippy::as_conversions)]
235             {
236                 offset as usize
237             },
238         )
239     }};
240 }
241 
242 /// Computes alignment of `$ty: ?Sized`.
243 ///
244 /// `align_of!` produces code which is valid in a `const` context.
245 // TODO(#29), TODO(https://github.com/rust-lang/rust/issues/69835): Remove this
246 // `cfg` when `size_of_val_raw` is stabilized.
247 #[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
248 #[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
249 #[macro_export]
250 macro_rules! align_of {
251     ($ty:ty) => {{
252         // SAFETY: `OffsetOfTrailingIsAlignment` is `repr(C)`, and its layout is
253         // guaranteed [1] to begin with the single-byte layout for `_byte`,
254         // followed by the padding needed to align `_trailing`, then the layout
255         // for `_trailing`, and finally any trailing padding bytes needed to
256         // correctly-align the entire struct.
257         //
258         // This macro computes the alignment of `$ty` by counting the number of
259         // bytes preceeding `_trailing`. For instance, if the alignment of `$ty`
260         // is `1`, then no padding is required align `_trailing` and it will be
261         // located immediately after `_byte` at offset 1. If the alignment of
262         // `$ty` is 2, then a single padding byte is required before
263         // `_trailing`, and `_trailing` will be located at offset 2.
264 
265         // This correspondence between offset and alignment holds for all valid
266         // Rust alignments, and we confirm this exhaustively (or, at least up to
267         // the maximum alignment supported by `trailing_field_offset!`) in
268         // `test_align_of_dst`.
269         //
270         // [1]: https://doc.rust-lang.org/nomicon/other-reprs.html#reprc
271 
272         #[repr(C)]
273         struct OffsetOfTrailingIsAlignment {
274             _byte: u8,
275             _trailing: $ty,
276         }
277 
278         trailing_field_offset!(OffsetOfTrailingIsAlignment, _trailing)
279     }};
280 }
281 
282 mod size_to_tag {
283     pub trait SizeToTag<const SIZE: usize> {
284         type Tag;
285     }
286 
287     impl SizeToTag<1> for () {
288         type Tag = u8;
289     }
290     impl SizeToTag<2> for () {
291         type Tag = u16;
292     }
293     impl SizeToTag<4> for () {
294         type Tag = u32;
295     }
296     impl SizeToTag<8> for () {
297         type Tag = u64;
298     }
299     impl SizeToTag<16> for () {
300         type Tag = u128;
301     }
302 }
303 
304 /// An alias for the unsigned integer of the given size in bytes.
305 #[doc(hidden)]
306 pub type SizeToTag<const SIZE: usize> = <() as size_to_tag::SizeToTag<SIZE>>::Tag;
307 
308 // We put `Sized` in its own module so it can have the same name as the standard
309 // library `Sized` without shadowing it in the parent module.
310 #[cfg(zerocopy_diagnostic_on_unimplemented_1_78_0)]
311 mod __size_of {
312     #[diagnostic::on_unimplemented(
313         message = "`{Self}` is unsized",
314         label = "`IntoBytes` needs all field types to be `Sized` in order to determine whether there is inter-field padding",
315         note = "consider using `#[repr(packed)]` to remove inter-field padding",
316         note = "`IntoBytes` does not require the fields of `#[repr(packed)]` types to be `Sized`"
317     )]
318     pub trait Sized: core::marker::Sized {}
319     impl<T: core::marker::Sized> Sized for T {}
320 
321     #[inline(always)]
322     #[must_use]
323     #[allow(clippy::needless_maybe_sized)]
size_of<T: Sized + ?core::marker::Sized>() -> usize324     pub const fn size_of<T: Sized + ?core::marker::Sized>() -> usize {
325         core::mem::size_of::<T>()
326     }
327 }
328 
329 #[cfg(zerocopy_diagnostic_on_unimplemented_1_78_0)]
330 pub use __size_of::size_of;
331 #[cfg(not(zerocopy_diagnostic_on_unimplemented_1_78_0))]
332 pub use core::mem::size_of;
333 
334 /// Does the struct type `$t` have padding?
335 ///
336 /// `$ts` is the list of the type of every field in `$t`. `$t` must be a
337 /// struct type, or else `struct_has_padding!`'s result may be meaningless.
338 ///
339 /// Note that `struct_has_padding!`'s results are independent of `repcr` since
340 /// they only consider the size of the type and the sizes of the fields.
341 /// Whatever the repr, the size of the type already takes into account any
342 /// padding that the compiler has decided to add. Structs with well-defined
343 /// representations (such as `repr(C)`) can use this macro to check for padding.
344 /// Note that while this may yield some consistent value for some `repr(Rust)`
345 /// structs, it is not guaranteed across platforms or compilations.
346 #[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
347 #[macro_export]
348 macro_rules! struct_has_padding {
349     ($t:ty, [$($ts:ty),*]) => {
350         ::zerocopy::util::macro_util::size_of::<$t>() > 0 $(+ ::zerocopy::util::macro_util::size_of::<$ts>())*
351     };
352 }
353 
354 /// Does the union type `$t` have padding?
355 ///
356 /// `$ts` is the list of the type of every field in `$t`. `$t` must be a
357 /// union type, or else `union_has_padding!`'s result may be meaningless.
358 ///
359 /// Note that `union_has_padding!`'s results are independent of `repr` since
360 /// they only consider the size of the type and the sizes of the fields.
361 /// Whatever the repr, the size of the type already takes into account any
362 /// padding that the compiler has decided to add. Unions with well-defined
363 /// representations (such as `repr(C)`) can use this macro to check for padding.
364 /// Note that while this may yield some consistent value for some `repr(Rust)`
365 /// unions, it is not guaranteed across platforms or compilations.
366 #[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
367 #[macro_export]
368 macro_rules! union_has_padding {
369     ($t:ty, [$($ts:ty),*]) => {
370         false $(|| ::zerocopy::util::macro_util::size_of::<$t>() != ::zerocopy::util::macro_util::size_of::<$ts>())*
371     };
372 }
373 
374 /// Does the enum type `$t` have padding?
375 ///
376 /// `$disc` is the type of the enum tag, and `$ts` is a list of fields in each
377 /// square-bracket-delimited variant. `$t` must be an enum, or else
378 /// `enum_has_padding!`'s result may be meaningless. An enum has padding if any
379 /// of its variant structs [1][2] contain padding, and so all of the variants of
380 /// an enum must be "full" in order for the enum to not have padding.
381 ///
382 /// The results of `enum_has_padding!` require that the enum is not
383 /// `repr(Rust)`, as `repr(Rust)` enums may niche the enum's tag and reduce the
384 /// total number of bytes required to represent the enum as a result. As long as
385 /// the enum is `repr(C)`, `repr(int)`, or `repr(C, int)`, this will
386 /// consistently return whether the enum contains any padding bytes.
387 ///
388 /// [1]: https://doc.rust-lang.org/1.81.0/reference/type-layout.html#reprc-enums-with-fields
389 /// [2]: https://doc.rust-lang.org/1.81.0/reference/type-layout.html#primitive-representation-of-enums-with-fields
390 #[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
391 #[macro_export]
392 macro_rules! enum_has_padding {
393     ($t:ty, $disc:ty, $([$($ts:ty),*]),*) => {
394         false $(
395             || ::zerocopy::util::macro_util::size_of::<$t>()
396                 != (
397                     ::zerocopy::util::macro_util::size_of::<$disc>()
398                     $(+ ::zerocopy::util::macro_util::size_of::<$ts>())*
399                 )
400         )*
401     }
402 }
403 
404 /// Does `t` have alignment greater than or equal to `u`?  If not, this macro
405 /// produces a compile error. It must be invoked in a dead codepath. This is
406 /// used in `transmute_ref!` and `transmute_mut!`.
407 #[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
408 #[macro_export]
409 macro_rules! assert_align_gt_eq {
410     ($t:ident, $u: ident) => {{
411         // The comments here should be read in the context of this macro's
412         // invocations in `transmute_ref!` and `transmute_mut!`.
413         if false {
414             // The type wildcard in this bound is inferred to be `T` because
415             // `align_of.into_t()` is assigned to `t` (which has type `T`).
416             let align_of: $crate::util::macro_util::AlignOf<_> = unreachable!();
417             $t = align_of.into_t();
418             // `max_aligns` is inferred to have type `MaxAlignsOf<T, U>` because
419             // of the inferred types of `t` and `u`.
420             let mut max_aligns = $crate::util::macro_util::MaxAlignsOf::new($t, $u);
421 
422             // This transmute will only compile successfully if
423             // `align_of::<T>() == max(align_of::<T>(), align_of::<U>())` - in
424             // other words, if `align_of::<T>() >= align_of::<U>()`.
425             //
426             // SAFETY: This code is never run.
427             max_aligns = unsafe {
428                 // Clippy: We can't annotate the types; this macro is designed
429                 // to infer the types from the calling context.
430                 #[allow(clippy::missing_transmute_annotations)]
431                 $crate::util::macro_util::core_reexport::mem::transmute(align_of)
432             };
433         } else {
434             loop {}
435         }
436     }};
437 }
438 
439 /// Do `t` and `u` have the same size?  If not, this macro produces a compile
440 /// error. It must be invoked in a dead codepath. This is used in
441 /// `transmute_ref!` and `transmute_mut!`.
442 #[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
443 #[macro_export]
444 macro_rules! assert_size_eq {
445     ($t:ident, $u: ident) => {{
446         // The comments here should be read in the context of this macro's
447         // invocations in `transmute_ref!` and `transmute_mut!`.
448         if false {
449             // SAFETY: This code is never run.
450             $u = unsafe {
451                 // Clippy:
452                 // - It's okay to transmute a type to itself.
453                 // - We can't annotate the types; this macro is designed to
454                 //   infer the types from the calling context.
455                 #[allow(clippy::useless_transmute, clippy::missing_transmute_annotations)]
456                 $crate::util::macro_util::core_reexport::mem::transmute($t)
457             };
458         } else {
459             loop {}
460         }
461     }};
462 }
463 
464 /// Transmutes a reference of one type to a reference of another type.
465 ///
466 /// # Safety
467 ///
468 /// The caller must guarantee that:
469 /// - `Src: IntoBytes + Immutable`
470 /// - `Dst: FromBytes + Immutable`
471 /// - `size_of::<Src>() == size_of::<Dst>()`
472 /// - `align_of::<Src>() >= align_of::<Dst>()`
473 #[inline(always)]
transmute_ref<'dst, 'src: 'dst, Src: 'src, Dst: 'dst>( src: &'src Src, ) -> &'dst Dst474 pub const unsafe fn transmute_ref<'dst, 'src: 'dst, Src: 'src, Dst: 'dst>(
475     src: &'src Src,
476 ) -> &'dst Dst {
477     let src: *const Src = src;
478     let dst = src.cast::<Dst>();
479     // SAFETY:
480     // - We know that it is sound to view the target type of the input reference
481     //   (`Src`) as the target type of the output reference (`Dst`) because the
482     //   caller has guaranteed that `Src: IntoBytes`, `Dst: FromBytes`, and
483     //   `size_of::<Src>() == size_of::<Dst>()`.
484     // - We know that there are no `UnsafeCell`s, and thus we don't have to
485     //   worry about `UnsafeCell` overlap, because `Src: Immutable` and `Dst:
486     //   Immutable`.
487     // - The caller has guaranteed that alignment is not increased.
488     // - We know that the returned lifetime will not outlive the input lifetime
489     //   thanks to the lifetime bounds on this function.
490     //
491     // TODO(#67): Once our MSRV is 1.58, replace this `transmute` with `&*dst`.
492     #[allow(clippy::transmute_ptr_to_ref)]
493     unsafe {
494         mem::transmute(dst)
495     }
496 }
497 
498 /// Transmutes a mutable reference of one type to a mutable reference of another
499 /// type.
500 ///
501 /// # Safety
502 ///
503 /// The caller must guarantee that:
504 /// - `Src: FromBytes + IntoBytes`
505 /// - `Dst: FromBytes + IntoBytes`
506 /// - `size_of::<Src>() == size_of::<Dst>()`
507 /// - `align_of::<Src>() >= align_of::<Dst>()`
508 // TODO(#686): Consider removing the `Immutable` requirement.
509 #[inline(always)]
transmute_mut<'dst, 'src: 'dst, Src: 'src, Dst: 'dst>( src: &'src mut Src, ) -> &'dst mut Dst510 pub unsafe fn transmute_mut<'dst, 'src: 'dst, Src: 'src, Dst: 'dst>(
511     src: &'src mut Src,
512 ) -> &'dst mut Dst {
513     let src: *mut Src = src;
514     let dst = src.cast::<Dst>();
515     // SAFETY:
516     // - We know that it is sound to view the target type of the input reference
517     //   (`Src`) as the target type of the output reference (`Dst`) and
518     //   vice-versa because the caller has guaranteed that `Src: FromBytes +
519     //   IntoBytes`, `Dst: FromBytes + IntoBytes`, and `size_of::<Src>() ==
520     //   size_of::<Dst>()`.
521     // - The caller has guaranteed that alignment is not increased.
522     // - We know that the returned lifetime will not outlive the input lifetime
523     //   thanks to the lifetime bounds on this function.
524     unsafe { &mut *dst }
525 }
526 
527 /// Is a given source a valid instance of `Dst`?
528 ///
529 /// If so, returns `src` casted to a `Ptr<Dst, _>`. Otherwise returns `None`.
530 ///
531 /// # Safety
532 ///
533 /// Unsafe code may assume that, if `try_cast_or_pme(src)` returns `Ok`,
534 /// `*src` is a bit-valid instance of `Dst`, and that the size of `Src` is
535 /// greater than or equal to the size of `Dst`.
536 ///
537 /// Unsafe code may assume that, if `try_cast_or_pme(src)` returns `Err`, the
538 /// encapsulated `Ptr` value is the original `src`. `try_cast_or_pme` cannot
539 /// guarantee that the referent has not been modified, as it calls user-defined
540 /// code (`TryFromBytes::is_bit_valid`).
541 ///
542 /// # Panics
543 ///
544 /// `try_cast_or_pme` may either produce a post-monomorphization error or a
545 /// panic if `Dst` not the same size as `Src`. Otherwise, `try_cast_or_pme`
546 /// panics under the same circumstances as [`is_bit_valid`].
547 ///
548 /// [`is_bit_valid`]: TryFromBytes::is_bit_valid
549 #[doc(hidden)]
550 #[inline]
try_cast_or_pme<Src, Dst, I, R>( src: Ptr<'_, Src, I>, ) -> Result< Ptr<'_, Dst, (I::Aliasing, invariant::Unaligned, invariant::Valid)>, ValidityError<Ptr<'_, Src, I>, Dst>, > where Src: invariant::Read<I::Aliasing, R>, Dst: TryFromBytes + invariant::Read<I::Aliasing, R>, I: Invariants<Validity = invariant::Initialized>, I::Aliasing: invariant::Reference,551 fn try_cast_or_pme<Src, Dst, I, R>(
552     src: Ptr<'_, Src, I>,
553 ) -> Result<
554     Ptr<'_, Dst, (I::Aliasing, invariant::Unaligned, invariant::Valid)>,
555     ValidityError<Ptr<'_, Src, I>, Dst>,
556 >
557 where
558     // TODO(#2226): There should be a `Src: FromBytes` bound here, but doing so
559     // requires deeper surgery.
560     Src: invariant::Read<I::Aliasing, R>,
561     Dst: TryFromBytes + invariant::Read<I::Aliasing, R>,
562     I: Invariants<Validity = invariant::Initialized>,
563     I::Aliasing: invariant::Reference,
564 {
565     static_assert!(Src, Dst => mem::size_of::<Dst>() == mem::size_of::<Src>());
566 
567     // SAFETY: This is a pointer cast, satisfying the following properties:
568     // - `p as *mut Dst` addresses a subset of the `bytes` addressed by `src`,
569     //   because we assert above that the size of `Dst` equal to the size of
570     //   `Src`.
571     // - `p as *mut Dst` is a provenance-preserving cast
572     #[allow(clippy::as_conversions)]
573     let c_ptr = unsafe { src.cast_unsized(|p| p as *mut Dst) };
574 
575     match c_ptr.try_into_valid() {
576         Ok(ptr) => Ok(ptr),
577         Err(err) => {
578             // Re-cast `Ptr<Dst>` to `Ptr<Src>`.
579             let ptr = err.into_src();
580             // SAFETY: This is a pointer cast, satisfying the following
581             // properties:
582             // - `p as *mut Src` addresses a subset of the `bytes` addressed by
583             //   `ptr`, because we assert above that the size of `Dst` is equal
584             //   to the size of `Src`.
585             // - `p as *mut Src` is a provenance-preserving cast
586             #[allow(clippy::as_conversions)]
587             let ptr = unsafe { ptr.cast_unsized(|p| p as *mut Src) };
588             // SAFETY: `ptr` is `src`, and has the same alignment invariant.
589             let ptr = unsafe { ptr.assume_alignment::<I::Alignment>() };
590             // SAFETY: `ptr` is `src` and has the same validity invariant.
591             let ptr = unsafe { ptr.assume_validity::<I::Validity>() };
592             Err(ValidityError::new(ptr.unify_invariants()))
593         }
594     }
595 }
596 
597 /// Attempts to transmute `Src` into `Dst`.
598 ///
599 /// A helper for `try_transmute!`.
600 ///
601 /// # Panics
602 ///
603 /// `try_transmute` may either produce a post-monomorphization error or a panic
604 /// if `Dst` is bigger than `Src`. Otherwise, `try_transmute` panics under the
605 /// same circumstances as [`is_bit_valid`].
606 ///
607 /// [`is_bit_valid`]: TryFromBytes::is_bit_valid
608 #[inline(always)]
try_transmute<Src, Dst>(src: Src) -> Result<Dst, ValidityError<Src, Dst>> where Src: IntoBytes, Dst: TryFromBytes,609 pub fn try_transmute<Src, Dst>(src: Src) -> Result<Dst, ValidityError<Src, Dst>>
610 where
611     Src: IntoBytes,
612     Dst: TryFromBytes,
613 {
614     static_assert!(Src, Dst => mem::size_of::<Dst>() == mem::size_of::<Src>());
615 
616     let mu_src = mem::MaybeUninit::new(src);
617     // SAFETY: By invariant on `&`, the following are satisfied:
618     // - `&mu_src` is valid for reads
619     // - `&mu_src` is properly aligned
620     // - `&mu_src`'s referent is bit-valid
621     let mu_src_copy = unsafe { core::ptr::read(&mu_src) };
622     // SAFETY: `MaybeUninit` has no validity constraints.
623     let mut mu_dst: mem::MaybeUninit<Dst> =
624         unsafe { crate::util::transmute_unchecked(mu_src_copy) };
625 
626     let ptr = Ptr::from_mut(&mut mu_dst);
627 
628     // SAFETY: Since `Src: IntoBytes`, and since `size_of::<Src>() ==
629     // size_of::<Dst>()` by the preceding assertion, all of `mu_dst`'s bytes are
630     // initialized.
631     let ptr = unsafe { ptr.assume_validity::<invariant::Initialized>() };
632 
633     // SAFETY: `MaybeUninit<T>` and `T` have the same size [1], so this cast
634     // preserves the referent's size. This cast preserves provenance.
635     //
636     // [1] Per https://doc.rust-lang.org/1.81.0/std/mem/union.MaybeUninit.html#layout-1:
637     //
638     //   `MaybeUninit<T>` is guaranteed to have the same size, alignment, and
639     //   ABI as `T`
640     let ptr: Ptr<'_, Dst, _> =
641         unsafe { ptr.cast_unsized(|mu: *mut mem::MaybeUninit<Dst>| mu.cast()) };
642 
643     if Dst::is_bit_valid(ptr.forget_aligned()) {
644         // SAFETY: Since `Dst::is_bit_valid`, we know that `ptr`'s referent is
645         // bit-valid for `Dst`. `ptr` points to `mu_dst`, and no intervening
646         // operations have mutated it, so it is a bit-valid `Dst`.
647         Ok(unsafe { mu_dst.assume_init() })
648     } else {
649         // SAFETY: `mu_src` was constructed from `src` and never modified, so it
650         // is still bit-valid.
651         Err(ValidityError::new(unsafe { mu_src.assume_init() }))
652     }
653 }
654 
655 /// Attempts to transmute `&Src` into `&Dst`.
656 ///
657 /// A helper for `try_transmute_ref!`.
658 ///
659 /// # Panics
660 ///
661 /// `try_transmute_ref` may either produce a post-monomorphization error or a
662 /// panic if `Dst` is bigger or has a stricter alignment requirement than `Src`.
663 /// Otherwise, `try_transmute_ref` panics under the same circumstances as
664 /// [`is_bit_valid`].
665 ///
666 /// [`is_bit_valid`]: TryFromBytes::is_bit_valid
667 #[inline(always)]
try_transmute_ref<Src, Dst>(src: &Src) -> Result<&Dst, ValidityError<&Src, Dst>> where Src: IntoBytes + Immutable, Dst: TryFromBytes + Immutable,668 pub fn try_transmute_ref<Src, Dst>(src: &Src) -> Result<&Dst, ValidityError<&Src, Dst>>
669 where
670     Src: IntoBytes + Immutable,
671     Dst: TryFromBytes + Immutable,
672 {
673     let ptr = Ptr::from_ref(src);
674     let ptr = ptr.bikeshed_recall_initialized_immutable();
675     match try_cast_or_pme::<Src, Dst, _, BecauseImmutable>(ptr) {
676         Ok(ptr) => {
677             static_assert!(Src, Dst => mem::align_of::<Dst>() <= mem::align_of::<Src>());
678             // SAFETY: We have checked that `Dst` does not have a stricter
679             // alignment requirement than `Src`.
680             let ptr = unsafe { ptr.assume_alignment::<invariant::Aligned>() };
681             Ok(ptr.as_ref())
682         }
683         Err(err) => Err(err.map_src(|ptr| {
684             // SAFETY: Because `Src: Immutable` and we create a `Ptr` via
685             // `Ptr::from_ref`, the resulting `Ptr` is a shared-and-`Immutable`
686             // `Ptr`, which does not permit mutation of its referent. Therefore,
687             // no mutation could have happened during the call to
688             // `try_cast_or_pme` (any such mutation would be unsound).
689             //
690             // `try_cast_or_pme` promises to return its original argument, and
691             // so we know that we are getting back the same `ptr` that we
692             // originally passed, and that `ptr` was a bit-valid `Src`.
693             let ptr = unsafe { ptr.assume_valid() };
694             ptr.as_ref()
695         })),
696     }
697 }
698 
699 /// Attempts to transmute `&mut Src` into `&mut Dst`.
700 ///
701 /// A helper for `try_transmute_mut!`.
702 ///
703 /// # Panics
704 ///
705 /// `try_transmute_mut` may either produce a post-monomorphization error or a
706 /// panic if `Dst` is bigger or has a stricter alignment requirement than `Src`.
707 /// Otherwise, `try_transmute_mut` panics under the same circumstances as
708 /// [`is_bit_valid`].
709 ///
710 /// [`is_bit_valid`]: TryFromBytes::is_bit_valid
711 #[inline(always)]
try_transmute_mut<Src, Dst>(src: &mut Src) -> Result<&mut Dst, ValidityError<&mut Src, Dst>> where Src: FromBytes + IntoBytes, Dst: TryFromBytes + IntoBytes,712 pub fn try_transmute_mut<Src, Dst>(src: &mut Src) -> Result<&mut Dst, ValidityError<&mut Src, Dst>>
713 where
714     Src: FromBytes + IntoBytes,
715     Dst: TryFromBytes + IntoBytes,
716 {
717     let ptr = Ptr::from_mut(src);
718     let ptr = ptr.bikeshed_recall_initialized_from_bytes();
719     match try_cast_or_pme::<Src, Dst, _, BecauseExclusive>(ptr) {
720         Ok(ptr) => {
721             static_assert!(Src, Dst => mem::align_of::<Dst>() <= mem::align_of::<Src>());
722             // SAFETY: We have checked that `Dst` does not have a stricter
723             // alignment requirement than `Src`.
724             let ptr = unsafe { ptr.assume_alignment::<invariant::Aligned>() };
725             Ok(ptr.as_mut())
726         }
727         Err(err) => Err(err.map_src(|ptr| ptr.bikeshed_recall_valid().as_mut())),
728     }
729 }
730 
731 /// A function which emits a warning if its return value is not used.
732 #[must_use]
733 #[inline(always)]
must_use<T>(t: T) -> T734 pub const fn must_use<T>(t: T) -> T {
735     t
736 }
737 
738 // NOTE: We can't change this to a `pub use core as core_reexport` until [1] is
739 // fixed or we update to a semver-breaking version (as of this writing, 0.8.0)
740 // on the `main` branch.
741 //
742 // [1] https://github.com/obi1kenobi/cargo-semver-checks/issues/573
743 pub mod core_reexport {
744     pub use core::*;
745 
746     pub mod mem {
747         pub use core::mem::*;
748     }
749 }
750 
751 #[cfg(test)]
752 mod tests {
753     use super::*;
754     use crate::util::testutil::*;
755 
756     #[test]
test_align_of()757     fn test_align_of() {
758         macro_rules! test {
759             ($ty:ty) => {
760                 assert_eq!(mem::size_of::<AlignOf<$ty>>(), mem::align_of::<$ty>());
761             };
762         }
763 
764         test!(());
765         test!(u8);
766         test!(AU64);
767         test!([AU64; 2]);
768     }
769 
770     #[test]
test_max_aligns_of()771     fn test_max_aligns_of() {
772         macro_rules! test {
773             ($t:ty, $u:ty) => {
774                 assert_eq!(
775                     mem::size_of::<MaxAlignsOf<$t, $u>>(),
776                     core::cmp::max(mem::align_of::<$t>(), mem::align_of::<$u>())
777                 );
778             };
779         }
780 
781         test!(u8, u8);
782         test!(u8, AU64);
783         test!(AU64, u8);
784     }
785 
786     #[test]
test_typed_align_check()787     fn test_typed_align_check() {
788         // Test that the type-based alignment check used in
789         // `assert_align_gt_eq!` behaves as expected.
790 
791         macro_rules! assert_t_align_gteq_u_align {
792             ($t:ty, $u:ty, $gteq:expr) => {
793                 assert_eq!(
794                     mem::size_of::<MaxAlignsOf<$t, $u>>() == mem::size_of::<AlignOf<$t>>(),
795                     $gteq
796                 );
797             };
798         }
799 
800         assert_t_align_gteq_u_align!(u8, u8, true);
801         assert_t_align_gteq_u_align!(AU64, AU64, true);
802         assert_t_align_gteq_u_align!(AU64, u8, true);
803         assert_t_align_gteq_u_align!(u8, AU64, false);
804     }
805 
806     // TODO(#29), TODO(https://github.com/rust-lang/rust/issues/69835): Remove
807     // this `cfg` when `size_of_val_raw` is stabilized.
808     #[allow(clippy::decimal_literal_representation)]
809     #[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
810     #[test]
test_trailing_field_offset()811     fn test_trailing_field_offset() {
812         assert_eq!(mem::align_of::<Aligned64kAllocation>(), _64K);
813 
814         macro_rules! test {
815             (#[$cfg:meta] ($($ts:ty),* ; $trailing_field_ty:ty) => $expect:expr) => {{
816                 #[$cfg]
817                 struct Test($(#[allow(dead_code)] $ts,)* #[allow(dead_code)] $trailing_field_ty);
818                 assert_eq!(test!(@offset $($ts),* ; $trailing_field_ty), $expect);
819             }};
820             (#[$cfg:meta] $(#[$cfgs:meta])* ($($ts:ty),* ; $trailing_field_ty:ty) => $expect:expr) => {
821                 test!(#[$cfg] ($($ts),* ; $trailing_field_ty) => $expect);
822                 test!($(#[$cfgs])* ($($ts),* ; $trailing_field_ty) => $expect);
823             };
824             (@offset ; $_trailing:ty) => { trailing_field_offset!(Test, 0) };
825             (@offset $_t:ty ; $_trailing:ty) => { trailing_field_offset!(Test, 1) };
826         }
827 
828         test!(#[repr(C)] #[repr(transparent)] #[repr(packed)](; u8) => Some(0));
829         test!(#[repr(C)] #[repr(transparent)] #[repr(packed)](; [u8]) => Some(0));
830         test!(#[repr(C)] #[repr(C, packed)] (u8; u8) => Some(1));
831         test!(#[repr(C)] (; AU64) => Some(0));
832         test!(#[repr(C)] (; [AU64]) => Some(0));
833         test!(#[repr(C)] (u8; AU64) => Some(8));
834         test!(#[repr(C)] (u8; [AU64]) => Some(8));
835         test!(#[repr(C)] (; Nested<u8, AU64>) => Some(0));
836         test!(#[repr(C)] (; Nested<u8, [AU64]>) => Some(0));
837         test!(#[repr(C)] (u8; Nested<u8, AU64>) => Some(8));
838         test!(#[repr(C)] (u8; Nested<u8, [AU64]>) => Some(8));
839 
840         // Test that `packed(N)` limits the offset of the trailing field.
841         test!(#[repr(C, packed(        1))] (u8; elain::Align<        2>) => Some(        1));
842         test!(#[repr(C, packed(        2))] (u8; elain::Align<        4>) => Some(        2));
843         test!(#[repr(C, packed(        4))] (u8; elain::Align<        8>) => Some(        4));
844         test!(#[repr(C, packed(        8))] (u8; elain::Align<       16>) => Some(        8));
845         test!(#[repr(C, packed(       16))] (u8; elain::Align<       32>) => Some(       16));
846         test!(#[repr(C, packed(       32))] (u8; elain::Align<       64>) => Some(       32));
847         test!(#[repr(C, packed(       64))] (u8; elain::Align<      128>) => Some(       64));
848         test!(#[repr(C, packed(      128))] (u8; elain::Align<      256>) => Some(      128));
849         test!(#[repr(C, packed(      256))] (u8; elain::Align<      512>) => Some(      256));
850         test!(#[repr(C, packed(      512))] (u8; elain::Align<     1024>) => Some(      512));
851         test!(#[repr(C, packed(     1024))] (u8; elain::Align<     2048>) => Some(     1024));
852         test!(#[repr(C, packed(     2048))] (u8; elain::Align<     4096>) => Some(     2048));
853         test!(#[repr(C, packed(     4096))] (u8; elain::Align<     8192>) => Some(     4096));
854         test!(#[repr(C, packed(     8192))] (u8; elain::Align<    16384>) => Some(     8192));
855         test!(#[repr(C, packed(    16384))] (u8; elain::Align<    32768>) => Some(    16384));
856         test!(#[repr(C, packed(    32768))] (u8; elain::Align<    65536>) => Some(    32768));
857         test!(#[repr(C, packed(    65536))] (u8; elain::Align<   131072>) => Some(    65536));
858         /* Alignments above 65536 are not yet supported.
859         test!(#[repr(C, packed(   131072))] (u8; elain::Align<   262144>) => Some(   131072));
860         test!(#[repr(C, packed(   262144))] (u8; elain::Align<   524288>) => Some(   262144));
861         test!(#[repr(C, packed(   524288))] (u8; elain::Align<  1048576>) => Some(   524288));
862         test!(#[repr(C, packed(  1048576))] (u8; elain::Align<  2097152>) => Some(  1048576));
863         test!(#[repr(C, packed(  2097152))] (u8; elain::Align<  4194304>) => Some(  2097152));
864         test!(#[repr(C, packed(  4194304))] (u8; elain::Align<  8388608>) => Some(  4194304));
865         test!(#[repr(C, packed(  8388608))] (u8; elain::Align< 16777216>) => Some(  8388608));
866         test!(#[repr(C, packed( 16777216))] (u8; elain::Align< 33554432>) => Some( 16777216));
867         test!(#[repr(C, packed( 33554432))] (u8; elain::Align< 67108864>) => Some( 33554432));
868         test!(#[repr(C, packed( 67108864))] (u8; elain::Align< 33554432>) => Some( 67108864));
869         test!(#[repr(C, packed( 33554432))] (u8; elain::Align<134217728>) => Some( 33554432));
870         test!(#[repr(C, packed(134217728))] (u8; elain::Align<268435456>) => Some(134217728));
871         test!(#[repr(C, packed(268435456))] (u8; elain::Align<268435456>) => Some(268435456));
872         */
873 
874         // Test that `align(N)` does not limit the offset of the trailing field.
875         test!(#[repr(C, align(        1))] (u8; elain::Align<        2>) => Some(        2));
876         test!(#[repr(C, align(        2))] (u8; elain::Align<        4>) => Some(        4));
877         test!(#[repr(C, align(        4))] (u8; elain::Align<        8>) => Some(        8));
878         test!(#[repr(C, align(        8))] (u8; elain::Align<       16>) => Some(       16));
879         test!(#[repr(C, align(       16))] (u8; elain::Align<       32>) => Some(       32));
880         test!(#[repr(C, align(       32))] (u8; elain::Align<       64>) => Some(       64));
881         test!(#[repr(C, align(       64))] (u8; elain::Align<      128>) => Some(      128));
882         test!(#[repr(C, align(      128))] (u8; elain::Align<      256>) => Some(      256));
883         test!(#[repr(C, align(      256))] (u8; elain::Align<      512>) => Some(      512));
884         test!(#[repr(C, align(      512))] (u8; elain::Align<     1024>) => Some(     1024));
885         test!(#[repr(C, align(     1024))] (u8; elain::Align<     2048>) => Some(     2048));
886         test!(#[repr(C, align(     2048))] (u8; elain::Align<     4096>) => Some(     4096));
887         test!(#[repr(C, align(     4096))] (u8; elain::Align<     8192>) => Some(     8192));
888         test!(#[repr(C, align(     8192))] (u8; elain::Align<    16384>) => Some(    16384));
889         test!(#[repr(C, align(    16384))] (u8; elain::Align<    32768>) => Some(    32768));
890         test!(#[repr(C, align(    32768))] (u8; elain::Align<    65536>) => Some(    65536));
891         /* Alignments above 65536 are not yet supported.
892         test!(#[repr(C, align(    65536))] (u8; elain::Align<   131072>) => Some(   131072));
893         test!(#[repr(C, align(   131072))] (u8; elain::Align<   262144>) => Some(   262144));
894         test!(#[repr(C, align(   262144))] (u8; elain::Align<   524288>) => Some(   524288));
895         test!(#[repr(C, align(   524288))] (u8; elain::Align<  1048576>) => Some(  1048576));
896         test!(#[repr(C, align(  1048576))] (u8; elain::Align<  2097152>) => Some(  2097152));
897         test!(#[repr(C, align(  2097152))] (u8; elain::Align<  4194304>) => Some(  4194304));
898         test!(#[repr(C, align(  4194304))] (u8; elain::Align<  8388608>) => Some(  8388608));
899         test!(#[repr(C, align(  8388608))] (u8; elain::Align< 16777216>) => Some( 16777216));
900         test!(#[repr(C, align( 16777216))] (u8; elain::Align< 33554432>) => Some( 33554432));
901         test!(#[repr(C, align( 33554432))] (u8; elain::Align< 67108864>) => Some( 67108864));
902         test!(#[repr(C, align( 67108864))] (u8; elain::Align< 33554432>) => Some( 33554432));
903         test!(#[repr(C, align( 33554432))] (u8; elain::Align<134217728>) => Some(134217728));
904         test!(#[repr(C, align(134217728))] (u8; elain::Align<268435456>) => Some(268435456));
905         */
906     }
907 
908     // TODO(#29), TODO(https://github.com/rust-lang/rust/issues/69835): Remove
909     // this `cfg` when `size_of_val_raw` is stabilized.
910     #[allow(clippy::decimal_literal_representation)]
911     #[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
912     #[test]
test_align_of_dst()913     fn test_align_of_dst() {
914         // Test that `align_of!` correctly computes the alignment of DSTs.
915         assert_eq!(align_of!([elain::Align<1>]), Some(1));
916         assert_eq!(align_of!([elain::Align<2>]), Some(2));
917         assert_eq!(align_of!([elain::Align<4>]), Some(4));
918         assert_eq!(align_of!([elain::Align<8>]), Some(8));
919         assert_eq!(align_of!([elain::Align<16>]), Some(16));
920         assert_eq!(align_of!([elain::Align<32>]), Some(32));
921         assert_eq!(align_of!([elain::Align<64>]), Some(64));
922         assert_eq!(align_of!([elain::Align<128>]), Some(128));
923         assert_eq!(align_of!([elain::Align<256>]), Some(256));
924         assert_eq!(align_of!([elain::Align<512>]), Some(512));
925         assert_eq!(align_of!([elain::Align<1024>]), Some(1024));
926         assert_eq!(align_of!([elain::Align<2048>]), Some(2048));
927         assert_eq!(align_of!([elain::Align<4096>]), Some(4096));
928         assert_eq!(align_of!([elain::Align<8192>]), Some(8192));
929         assert_eq!(align_of!([elain::Align<16384>]), Some(16384));
930         assert_eq!(align_of!([elain::Align<32768>]), Some(32768));
931         assert_eq!(align_of!([elain::Align<65536>]), Some(65536));
932         /* Alignments above 65536 are not yet supported.
933         assert_eq!(align_of!([elain::Align<131072>]), Some(131072));
934         assert_eq!(align_of!([elain::Align<262144>]), Some(262144));
935         assert_eq!(align_of!([elain::Align<524288>]), Some(524288));
936         assert_eq!(align_of!([elain::Align<1048576>]), Some(1048576));
937         assert_eq!(align_of!([elain::Align<2097152>]), Some(2097152));
938         assert_eq!(align_of!([elain::Align<4194304>]), Some(4194304));
939         assert_eq!(align_of!([elain::Align<8388608>]), Some(8388608));
940         assert_eq!(align_of!([elain::Align<16777216>]), Some(16777216));
941         assert_eq!(align_of!([elain::Align<33554432>]), Some(33554432));
942         assert_eq!(align_of!([elain::Align<67108864>]), Some(67108864));
943         assert_eq!(align_of!([elain::Align<33554432>]), Some(33554432));
944         assert_eq!(align_of!([elain::Align<134217728>]), Some(134217728));
945         assert_eq!(align_of!([elain::Align<268435456>]), Some(268435456));
946         */
947     }
948 
949     #[test]
test_enum_casts()950     fn test_enum_casts() {
951         // Test that casting the variants of enums with signed integer reprs to
952         // unsigned integers obeys expected signed -> unsigned casting rules.
953 
954         #[repr(i8)]
955         enum ReprI8 {
956             MinusOne = -1,
957             Zero = 0,
958             Min = i8::MIN,
959             Max = i8::MAX,
960         }
961 
962         #[allow(clippy::as_conversions)]
963         let x = ReprI8::MinusOne as u8;
964         assert_eq!(x, u8::MAX);
965 
966         #[allow(clippy::as_conversions)]
967         let x = ReprI8::Zero as u8;
968         assert_eq!(x, 0);
969 
970         #[allow(clippy::as_conversions)]
971         let x = ReprI8::Min as u8;
972         assert_eq!(x, 128);
973 
974         #[allow(clippy::as_conversions)]
975         let x = ReprI8::Max as u8;
976         assert_eq!(x, 127);
977     }
978 
979     #[test]
test_struct_has_padding()980     fn test_struct_has_padding() {
981         // Test that, for each provided repr, `struct_has_padding!` reports the
982         // expected value.
983         macro_rules! test {
984             (#[$cfg:meta] ($($ts:ty),*) => $expect:expr) => {{
985                 #[$cfg]
986                 struct Test($(#[allow(dead_code)] $ts),*);
987                 assert_eq!(struct_has_padding!(Test, [$($ts),*]), $expect);
988             }};
989             (#[$cfg:meta] $(#[$cfgs:meta])* ($($ts:ty),*) => $expect:expr) => {
990                 test!(#[$cfg] ($($ts),*) => $expect);
991                 test!($(#[$cfgs])* ($($ts),*) => $expect);
992             };
993         }
994 
995         test!(#[repr(C)] #[repr(transparent)] #[repr(packed)] () => false);
996         test!(#[repr(C)] #[repr(transparent)] #[repr(packed)] (u8) => false);
997         test!(#[repr(C)] #[repr(transparent)] #[repr(packed)] (u8, ()) => false);
998         test!(#[repr(C)] #[repr(packed)] (u8, u8) => false);
999 
1000         test!(#[repr(C)] (u8, AU64) => true);
1001         // Rust won't let you put `#[repr(packed)]` on a type which contains a
1002         // `#[repr(align(n > 1))]` type (`AU64`), so we have to use `u64` here.
1003         // It's not ideal, but it definitely has align > 1 on /some/ of our CI
1004         // targets, and this isn't a particularly complex macro we're testing
1005         // anyway.
1006         test!(#[repr(packed)] (u8, u64) => false);
1007     }
1008 
1009     #[test]
test_union_has_padding()1010     fn test_union_has_padding() {
1011         // Test that, for each provided repr, `union_has_padding!` reports the
1012         // expected value.
1013         macro_rules! test {
1014             (#[$cfg:meta] {$($fs:ident: $ts:ty),*} => $expect:expr) => {{
1015                 #[$cfg]
1016                 #[allow(unused)] // fields are never read
1017                 union Test{ $($fs: $ts),* }
1018                 assert_eq!(union_has_padding!(Test, [$($ts),*]), $expect);
1019             }};
1020             (#[$cfg:meta] $(#[$cfgs:meta])* {$($fs:ident: $ts:ty),*} => $expect:expr) => {
1021                 test!(#[$cfg] {$($fs: $ts),*} => $expect);
1022                 test!($(#[$cfgs])* {$($fs: $ts),*} => $expect);
1023             };
1024         }
1025 
1026         test!(#[repr(C)] #[repr(packed)] {a: u8} => false);
1027         test!(#[repr(C)] #[repr(packed)] {a: u8, b: u8} => false);
1028 
1029         // Rust won't let you put `#[repr(packed)]` on a type which contains a
1030         // `#[repr(align(n > 1))]` type (`AU64`), so we have to use `u64` here.
1031         // It's not ideal, but it definitely has align > 1 on /some/ of our CI
1032         // targets, and this isn't a particularly complex macro we're testing
1033         // anyway.
1034         test!(#[repr(C)] #[repr(packed)] {a: u8, b: u64} => true);
1035     }
1036 
1037     #[test]
test_enum_has_padding()1038     fn test_enum_has_padding() {
1039         // Test that, for each provided repr, `enum_has_padding!` reports the
1040         // expected value.
1041         macro_rules! test {
1042             (#[repr($disc:ident $(, $c:ident)?)] { $($vs:ident ($($ts:ty),*),)* } => $expect:expr) => {
1043                 test!(@case #[repr($disc $(, $c)?)] { $($vs ($($ts),*),)* } => $expect);
1044             };
1045             (#[repr($disc:ident $(, $c:ident)?)] #[$cfg:meta] $(#[$cfgs:meta])* { $($vs:ident ($($ts:ty),*),)* } => $expect:expr) => {
1046                 test!(@case #[repr($disc $(, $c)?)] #[$cfg] { $($vs ($($ts),*),)* } => $expect);
1047                 test!(#[repr($disc $(, $c)?)] $(#[$cfgs])* { $($vs ($($ts),*),)* } => $expect);
1048             };
1049             (@case #[repr($disc:ident $(, $c:ident)?)] $(#[$cfg:meta])? { $($vs:ident ($($ts:ty),*),)* } => $expect:expr) => {{
1050                 #[repr($disc $(, $c)?)]
1051                 $(#[$cfg])?
1052                 #[allow(unused)] // variants and fields are never used
1053                 enum Test {
1054                     $($vs ($($ts),*),)*
1055                 }
1056                 assert_eq!(
1057                     enum_has_padding!(Test, $disc, $([$($ts),*]),*),
1058                     $expect
1059                 );
1060             }};
1061         }
1062 
1063         #[allow(unused)]
1064         #[repr(align(2))]
1065         struct U16(u16);
1066 
1067         #[allow(unused)]
1068         #[repr(align(4))]
1069         struct U32(u32);
1070 
1071         test!(#[repr(u8)] #[repr(C)] {
1072             A(u8),
1073         } => false);
1074         test!(#[repr(u16)] #[repr(C)] {
1075             A(u8, u8),
1076             B(U16),
1077         } => false);
1078         test!(#[repr(u32)] #[repr(C)] {
1079             A(u8, u8, u8, u8),
1080             B(U16, u8, u8),
1081             C(u8, u8, U16),
1082             D(U16, U16),
1083             E(U32),
1084         } => false);
1085 
1086         // `repr(int)` can pack the discriminant more efficiently
1087         test!(#[repr(u8)] {
1088             A(u8, U16),
1089         } => false);
1090         test!(#[repr(u8)] {
1091             A(u8, U16, U32),
1092         } => false);
1093 
1094         // `repr(C)` cannot
1095         test!(#[repr(u8, C)] {
1096             A(u8, U16),
1097         } => true);
1098         test!(#[repr(u8, C)] {
1099             A(u8, u8, u8, U32),
1100         } => true);
1101 
1102         // And field ordering can always cause problems
1103         test!(#[repr(u8)] #[repr(C)] {
1104             A(U16, u8),
1105         } => true);
1106         test!(#[repr(u8)] #[repr(C)] {
1107             A(U32, u8, u8, u8),
1108         } => true);
1109     }
1110 }
1111