Coverage Report

Created: 2025-08-28 06:06

/rust/registry/src/index.crates.io-6f17d22bba15001f/zerocopy-0.7.34/src/macro_util.rs
Line
Count
Source (jump to first uncovered line)
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 `AsBytes` and `FromBytes` on three different types would
15
//! result in the code in question being emitted and compiled six different
16
//! times).
17
18
#![allow(missing_debug_implementations)]
19
20
use core::{marker::PhantomData, mem::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(__INTERNAL_USE_ONLY_NIGHLTY_FEATURES_IN_TESTS)]
25
use core::ptr::{self, NonNull};
26
27
/// A compile-time check that should be one particular value.
28
pub trait ShouldBe<const VALUE: bool> {}
29
30
/// A struct for checking whether `T` contains padding.
31
pub struct HasPadding<T: ?Sized, const VALUE: bool>(PhantomData<T>);
32
33
impl<T: ?Sized, const VALUE: bool> ShouldBe<VALUE> for HasPadding<T, VALUE> {}
34
35
/// A type whose size is equal to `align_of::<T>()`.
36
#[repr(C)]
37
pub struct AlignOf<T> {
38
    // This field ensures that:
39
    // - The size is always at least 1 (the minimum possible alignment).
40
    // - If the alignment is greater than 1, Rust has to round up to the next
41
    //   multiple of it in order to make sure that `Align`'s size is a multiple
42
    //   of that alignment. Without this field, its size could be 0, which is a
43
    //   valid multiple of any alignment.
44
    _u: u8,
45
    _a: [T; 0],
46
}
47
48
impl<T> AlignOf<T> {
49
    #[inline(never)] // Make `missing_inline_in_public_items` happy.
50
0
    pub fn into_t(self) -> T {
51
0
        unreachable!()
52
    }
53
}
54
55
/// A type whose size is equal to `max(align_of::<T>(), align_of::<U>())`.
56
#[repr(C)]
57
pub union MaxAlignsOf<T, U> {
58
    _t: ManuallyDrop<AlignOf<T>>,
59
    _u: ManuallyDrop<AlignOf<U>>,
60
}
61
62
impl<T, U> MaxAlignsOf<T, U> {
63
    #[inline(never)] // Make `missing_inline_in_public_items` happy.
64
0
    pub fn new(_t: T, _u: U) -> MaxAlignsOf<T, U> {
65
0
        unreachable!()
66
    }
67
}
68
69
const _64K: usize = 1 << 16;
70
71
// TODO(#29), TODO(https://github.com/rust-lang/rust/issues/69835): Remove this
72
// `cfg` when `size_of_val_raw` is stabilized.
73
#[cfg(__INTERNAL_USE_ONLY_NIGHLTY_FEATURES_IN_TESTS)]
74
#[repr(C, align(65536))]
75
struct Aligned64kAllocation([u8; _64K]);
76
77
/// A pointer to an aligned allocation of size 2^16.
78
///
79
/// # Safety
80
///
81
/// `ALIGNED_64K_ALLOCATION` is guaranteed to point to the entirety of an
82
/// allocation with size and alignment 2^16, and to have valid provenance.
83
// TODO(#29), TODO(https://github.com/rust-lang/rust/issues/69835): Remove this
84
// `cfg` when `size_of_val_raw` is stabilized.
85
#[cfg(__INTERNAL_USE_ONLY_NIGHLTY_FEATURES_IN_TESTS)]
86
pub const ALIGNED_64K_ALLOCATION: NonNull<[u8]> = {
87
    const REF: &Aligned64kAllocation = &Aligned64kAllocation([0; _64K]);
88
    let ptr: *const Aligned64kAllocation = REF;
89
    let ptr: *const [u8] = ptr::slice_from_raw_parts(ptr.cast(), _64K);
90
    // SAFETY:
91
    // - `ptr` is derived from a Rust reference, which is guaranteed to be
92
    //   non-null.
93
    // - `ptr` is derived from an `&Aligned64kAllocation`, which has size and
94
    //   alignment `_64K` as promised. Its length is initialized to `_64K`,
95
    //   which means that it refers to the entire allocation.
96
    // - `ptr` is derived from a Rust reference, which is guaranteed to have
97
    //   valid provenance.
98
    //
99
    // TODO(#429): Once `NonNull::new_unchecked` docs document that it preserves
100
    // provenance, cite those docs.
101
    // TODO: Replace this `as` with `ptr.cast_mut()` once our MSRV >= 1.65
102
    #[allow(clippy::as_conversions)]
103
    unsafe {
104
        NonNull::new_unchecked(ptr as *mut _)
105
    }
106
};
107
108
/// Computes the offset of the base of the field `$trailing_field_name` within
109
/// the type `$ty`.
110
///
111
/// `trailing_field_offset!` produces code which is valid in a `const` context.
112
// TODO(#29), TODO(https://github.com/rust-lang/rust/issues/69835): Remove this
113
// `cfg` when `size_of_val_raw` is stabilized.
114
#[cfg(__INTERNAL_USE_ONLY_NIGHLTY_FEATURES_IN_TESTS)]
115
#[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
116
#[macro_export]
117
macro_rules! trailing_field_offset {
118
    ($ty:ty, $trailing_field_name:tt) => {{
119
        let min_size = {
120
            let zero_elems: *const [()] =
121
                $crate::macro_util::core_reexport::ptr::slice_from_raw_parts(
122
                    $crate::macro_util::core_reexport::ptr::NonNull::<()>::dangling()
123
                        .as_ptr()
124
                        .cast_const(),
125
                    0,
126
                );
127
            // SAFETY:
128
            // - If `$ty` is `Sized`, `size_of_val_raw` is always safe to call.
129
            // - Otherwise:
130
            //   - If `$ty` is not a slice DST, this pointer conversion will
131
            //     fail due to "mismatched vtable kinds", and compilation will
132
            //     fail.
133
            //   - If `$ty` is a slice DST, the safety requirement is that "the
134
            //     length of the slice tail must be an initialized integer, and
135
            //     the size of the entire value (dynamic tail length +
136
            //     statically sized prefix) must fit in isize." The length is
137
            //     initialized to 0 above, and Rust guarantees that no type's
138
            //     minimum size may overflow `isize`. [1]
139
            //
140
            // [1] TODO(#429),
141
            // TODO(https://github.com/rust-lang/unsafe-code-guidelines/issues/465#issuecomment-1782206516):
142
            // Citation for this?
143
            unsafe {
144
                #[allow(clippy::as_conversions)]
145
                $crate::macro_util::core_reexport::mem::size_of_val_raw(zero_elems as *const $ty)
146
            }
147
        };
148
149
        assert!(min_size <= _64K);
150
151
        #[allow(clippy::as_conversions)]
152
        let ptr = ALIGNED_64K_ALLOCATION.as_ptr() as *const $ty;
153
154
        // SAFETY:
155
        // - Thanks to the preceding `assert!`, we know that the value with zero
156
        //   elements fits in `_64K` bytes, and thus in the allocation addressed
157
        //   by `ALIGNED_64K_ALLOCATION`. The offset of the trailing field is
158
        //   guaranteed to be no larger than this size, so this field projection
159
        //   is guaranteed to remain in-bounds of its allocation.
160
        // - Because the minimum size is no larger than `_64K` bytes, and
161
        //   because an object's size must always be a multiple of its alignment
162
        //   [1], we know that `$ty`'s alignment is no larger than `_64K`. The
163
        //   allocation addressed by `ALIGNED_64K_ALLOCATION` is guaranteed to
164
        //   be aligned to `_64K`, so `ptr` is guaranteed to satisfy `$ty`'s
165
        //   alignment.
166
        //
167
        //   Note that, as of [2], this requirement is technically unnecessary
168
        //   for Rust versions >= 1.75.0, but no harm in guaranteeing it anyway
169
        //   until we bump our MSRV.
170
        //
171
        // [1] Per https://doc.rust-lang.org/reference/type-layout.html:
172
        //
173
        //   The size of a value is always a multiple of its alignment.
174
        //
175
        // [2] https://github.com/rust-lang/reference/pull/1387
176
        let field = unsafe {
177
            $crate::macro_util::core_reexport::ptr::addr_of!((*ptr).$trailing_field_name)
178
        };
179
        // SAFETY:
180
        // - Both `ptr` and `field` are derived from the same allocated object.
181
        // - By the preceding safety comment, `field` is in bounds of that
182
        //   allocated object.
183
        // - The distance, in bytes, between `ptr` and `field` is required to be
184
        //   a multiple of the size of `u8`, which is trivially true because
185
        //   `u8`'s size is 1.
186
        // - The distance, in bytes, cannot overflow `isize`. This is guaranteed
187
        //   because no allocated object can have a size larger than can fit in
188
        //   `isize`. [1]
189
        // - The distance being in-bounds cannot rely on wrapping around the
190
        //   address space. This is guaranteed because the same is guaranteed of
191
        //   allocated objects. [1]
192
        //
193
        // [1] TODO(#429), TODO(https://github.com/rust-lang/rust/pull/116675):
194
        //     Once these are guaranteed in the Reference, cite it.
195
        let offset = unsafe { field.cast::<u8>().offset_from(ptr.cast::<u8>()) };
196
        // Guaranteed not to be lossy: `field` comes after `ptr`, so the offset
197
        // from `ptr` to `field` is guaranteed to be positive.
198
        assert!(offset >= 0);
199
        Some(
200
            #[allow(clippy::as_conversions)]
201
            {
202
                offset as usize
203
            },
204
        )
205
    }};
206
}
207
208
/// Computes alignment of `$ty: ?Sized`.
209
///
210
/// `align_of!` produces code which is valid in a `const` context.
211
// TODO(#29), TODO(https://github.com/rust-lang/rust/issues/69835): Remove this
212
// `cfg` when `size_of_val_raw` is stabilized.
213
#[cfg(__INTERNAL_USE_ONLY_NIGHLTY_FEATURES_IN_TESTS)]
214
#[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
215
#[macro_export]
216
macro_rules! align_of {
217
    ($ty:ty) => {{
218
        // SAFETY: `OffsetOfTrailingIsAlignment` is `repr(C)`, and its layout is
219
        // guaranteed [1] to begin with the single-byte layout for `_byte`,
220
        // followed by the padding needed to align `_trailing`, then the layout
221
        // for `_trailing`, and finally any trailing padding bytes needed to
222
        // correctly-align the entire struct.
223
        //
224
        // This macro computes the alignment of `$ty` by counting the number of
225
        // bytes preceeding `_trailing`. For instance, if the alignment of `$ty`
226
        // is `1`, then no padding is required align `_trailing` and it will be
227
        // located immediately after `_byte` at offset 1. If the alignment of
228
        // `$ty` is 2, then a single padding byte is required before
229
        // `_trailing`, and `_trailing` will be located at offset 2.
230
231
        // This correspondence between offset and alignment holds for all valid
232
        // Rust alignments, and we confirm this exhaustively (or, at least up to
233
        // the maximum alignment supported by `trailing_field_offset!`) in
234
        // `test_align_of_dst`.
235
        //
236
        // [1]: https://doc.rust-lang.org/nomicon/other-reprs.html#reprc
237
238
        #[repr(C)]
239
        struct OffsetOfTrailingIsAlignment {
240
            _byte: u8,
241
            _trailing: $ty,
242
        }
243
244
        trailing_field_offset!(OffsetOfTrailingIsAlignment, _trailing)
245
    }};
246
}
247
248
/// Does the struct type `$t` have padding?
249
///
250
/// `$ts` is the list of the type of every field in `$t`. `$t` must be a
251
/// struct type, or else `struct_has_padding!`'s result may be meaningless.
252
///
253
/// Note that `struct_has_padding!`'s results are independent of `repr` since
254
/// they only consider the size of the type and the sizes of the fields.
255
/// Whatever the repr, the size of the type already takes into account any
256
/// padding that the compiler has decided to add. Structs with well-defined
257
/// representations (such as `repr(C)`) can use this macro to check for padding.
258
/// Note that while this may yield some consistent value for some `repr(Rust)`
259
/// structs, it is not guaranteed across platforms or compilations.
260
#[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
261
#[macro_export]
262
macro_rules! struct_has_padding {
263
    ($t:ty, $($ts:ty),*) => {
264
        core::mem::size_of::<$t>() > 0 $(+ core::mem::size_of::<$ts>())*
265
    };
266
}
267
268
/// Does the union type `$t` have padding?
269
///
270
/// `$ts` is the list of the type of every field in `$t`. `$t` must be a
271
/// union type, or else `union_has_padding!`'s result may be meaningless.
272
///
273
/// Note that `union_has_padding!`'s results are independent of `repr` since
274
/// they only consider the size of the type and the sizes of the fields.
275
/// Whatever the repr, the size of the type already takes into account any
276
/// padding that the compiler has decided to add. Unions with well-defined
277
/// representations (such as `repr(C)`) can use this macro to check for padding.
278
/// Note that while this may yield some consistent value for some `repr(Rust)`
279
/// unions, it is not guaranteed across platforms or compilations.
280
#[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
281
#[macro_export]
282
macro_rules! union_has_padding {
283
    ($t:ty, $($ts:ty),*) => {
284
        false $(|| core::mem::size_of::<$t>() != core::mem::size_of::<$ts>())*
285
    };
286
}
287
288
/// Does `t` have alignment greater than or equal to `u`?  If not, this macro
289
/// produces a compile error. It must be invoked in a dead codepath. This is
290
/// used in `transmute_ref!` and `transmute_mut!`.
291
#[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
292
#[macro_export]
293
macro_rules! assert_align_gt_eq {
294
    ($t:ident, $u: ident) => {{
295
        // The comments here should be read in the context of this macro's
296
        // invocations in `transmute_ref!` and `transmute_mut!`.
297
        #[allow(clippy::missing_transmute_annotations)]
298
        if false {
299
            // The type wildcard in this bound is inferred to be `T` because
300
            // `align_of.into_t()` is assigned to `t` (which has type `T`).
301
            let align_of: $crate::macro_util::AlignOf<_> = unreachable!();
302
            $t = align_of.into_t();
303
            // `max_aligns` is inferred to have type `MaxAlignsOf<T, U>` because
304
            // of the inferred types of `t` and `u`.
305
            let mut max_aligns = $crate::macro_util::MaxAlignsOf::new($t, $u);
306
307
            // This transmute will only compile successfully if
308
            // `align_of::<T>() == max(align_of::<T>(), align_of::<U>())` - in
309
            // other words, if `align_of::<T>() >= align_of::<U>()`.
310
            //
311
            // SAFETY: This code is never run.
312
            max_aligns = unsafe { $crate::macro_util::core_reexport::mem::transmute(align_of) };
313
        } else {
314
            loop {}
315
        }
316
    }};
317
}
318
319
/// Do `t` and `u` have the same size?  If not, this macro produces a compile
320
/// error. It must be invoked in a dead codepath. This is used in
321
/// `transmute_ref!` and `transmute_mut!`.
322
#[doc(hidden)] // `#[macro_export]` bypasses this module's `#[doc(hidden)]`.
323
#[macro_export]
324
macro_rules! assert_size_eq {
325
    ($t:ident, $u: ident) => {{
326
        // The comments here should be read in the context of this macro's
327
        // invocations in `transmute_ref!` and `transmute_mut!`.
328
        if false {
329
            // SAFETY: This code is never run.
330
            $u = unsafe {
331
                // Clippy: It's okay to transmute a type to itself.
332
                #[allow(clippy::useless_transmute, clippy::missing_transmute_annotations)]
333
                $crate::macro_util::core_reexport::mem::transmute($t)
334
            };
335
        } else {
336
            loop {}
337
        }
338
    }};
339
}
340
341
/// Transmutes a reference of one type to a reference of another type.
342
///
343
/// # Safety
344
///
345
/// The caller must guarantee that:
346
/// - `Src: AsBytes`
347
/// - `Dst: FromBytes`
348
/// - `size_of::<Src>() == size_of::<Dst>()`
349
/// - `align_of::<Src>() >= align_of::<Dst>()`
350
#[inline(always)]
351
0
pub const unsafe fn transmute_ref<'dst, 'src: 'dst, Src: 'src, Dst: 'dst>(
352
0
    src: &'src Src,
353
0
) -> &'dst Dst {
354
0
    let src: *const Src = src;
355
0
    let dst = src.cast::<Dst>();
356
0
    // SAFETY:
357
0
    // - We know that it is sound to view the target type of the input reference
358
0
    //   (`Src`) as the target type of the output reference (`Dst`) because the
359
0
    //   caller has guaranteed that `Src: AsBytes`, `Dst: FromBytes`, and
360
0
    //   `size_of::<Src>() == size_of::<Dst>()`.
361
0
    // - We know that there are no `UnsafeCell`s, and thus we don't have to
362
0
    //   worry about `UnsafeCell` overlap, because `Src: AsBytes` and `Dst:
363
0
    //   FromBytes` both forbid `UnsafeCell`s.
364
0
    // - The caller has guaranteed that alignment is not increased.
365
0
    // - We know that the returned lifetime will not outlive the input lifetime
366
0
    //   thanks to the lifetime bounds on this function.
367
0
    unsafe { &*dst }
368
0
}
369
370
/// Transmutes a mutable reference of one type to a mutable reference of another
371
/// type.
372
///
373
/// # Safety
374
///
375
/// The caller must guarantee that:
376
/// - `Src: FromBytes + AsBytes`
377
/// - `Dst: FromBytes + AsBytes`
378
/// - `size_of::<Src>() == size_of::<Dst>()`
379
/// - `align_of::<Src>() >= align_of::<Dst>()`
380
#[inline(always)]
381
0
pub unsafe fn transmute_mut<'dst, 'src: 'dst, Src: 'src, Dst: 'dst>(
382
0
    src: &'src mut Src,
383
0
) -> &'dst mut Dst {
384
0
    let src: *mut Src = src;
385
0
    let dst = src.cast::<Dst>();
386
0
    // SAFETY:
387
0
    // - We know that it is sound to view the target type of the input reference
388
0
    //   (`Src`) as the target type of the output reference (`Dst`) and
389
0
    //   vice-versa because the caller has guaranteed that `Src: FromBytes +
390
0
    //   AsBytes`, `Dst: FromBytes + AsBytes`, and `size_of::<Src>() ==
391
0
    //   size_of::<Dst>()`.
392
0
    // - We know that there are no `UnsafeCell`s, and thus we don't have to
393
0
    //   worry about `UnsafeCell` overlap, because `Src: FromBytes + AsBytes`
394
0
    //   and `Dst: FromBytes + AsBytes` forbid `UnsafeCell`s.
395
0
    // - The caller has guaranteed that alignment is not increased.
396
0
    // - We know that the returned lifetime will not outlive the input lifetime
397
0
    //   thanks to the lifetime bounds on this function.
398
0
    unsafe { &mut *dst }
399
0
}
400
401
// NOTE: We can't change this to a `pub use core as core_reexport` until [1] is
402
// fixed or we update to a semver-breaking version (as of this writing, 0.8.0)
403
// on the `main` branch.
404
//
405
// [1] https://github.com/obi1kenobi/cargo-semver-checks/issues/573
406
pub mod core_reexport {
407
    pub use core::*;
408
409
    pub mod mem {
410
        pub use core::mem::*;
411
    }
412
}
413
414
#[cfg(test)]
415
mod tests {
416
    use core::mem;
417
418
    use super::*;
419
    use crate::util::testutil::*;
420
421
    #[test]
422
    fn test_align_of() {
423
        macro_rules! test {
424
            ($ty:ty) => {
425
                assert_eq!(mem::size_of::<AlignOf<$ty>>(), mem::align_of::<$ty>());
426
            };
427
        }
428
429
        test!(());
430
        test!(u8);
431
        test!(AU64);
432
        test!([AU64; 2]);
433
    }
434
435
    #[test]
436
    fn test_max_aligns_of() {
437
        macro_rules! test {
438
            ($t:ty, $u:ty) => {
439
                assert_eq!(
440
                    mem::size_of::<MaxAlignsOf<$t, $u>>(),
441
                    core::cmp::max(mem::align_of::<$t>(), mem::align_of::<$u>())
442
                );
443
            };
444
        }
445
446
        test!(u8, u8);
447
        test!(u8, AU64);
448
        test!(AU64, u8);
449
    }
450
451
    #[test]
452
    fn test_typed_align_check() {
453
        // Test that the type-based alignment check used in
454
        // `assert_align_gt_eq!` behaves as expected.
455
456
        macro_rules! assert_t_align_gteq_u_align {
457
            ($t:ty, $u:ty, $gteq:expr) => {
458
                assert_eq!(
459
                    mem::size_of::<MaxAlignsOf<$t, $u>>() == mem::size_of::<AlignOf<$t>>(),
460
                    $gteq
461
                );
462
            };
463
        }
464
465
        assert_t_align_gteq_u_align!(u8, u8, true);
466
        assert_t_align_gteq_u_align!(AU64, AU64, true);
467
        assert_t_align_gteq_u_align!(AU64, u8, true);
468
        assert_t_align_gteq_u_align!(u8, AU64, false);
469
    }
470
471
    // TODO(#29), TODO(https://github.com/rust-lang/rust/issues/69835): Remove
472
    // this `cfg` when `size_of_val_raw` is stabilized.
473
    #[allow(clippy::decimal_literal_representation)]
474
    #[cfg(__INTERNAL_USE_ONLY_NIGHLTY_FEATURES_IN_TESTS)]
475
    #[test]
476
    fn test_trailing_field_offset() {
477
        assert_eq!(mem::align_of::<Aligned64kAllocation>(), _64K);
478
479
        macro_rules! test {
480
            (#[$cfg:meta] ($($ts:ty),* ; $trailing_field_ty:ty) => $expect:expr) => {{
481
                #[$cfg]
482
                #[allow(dead_code)] // fields are never read
483
                struct Test($($ts,)* $trailing_field_ty);
484
                assert_eq!(test!(@offset $($ts),* ; $trailing_field_ty), $expect);
485
            }};
486
            (#[$cfg:meta] $(#[$cfgs:meta])* ($($ts:ty),* ; $trailing_field_ty:ty) => $expect:expr) => {
487
                test!(#[$cfg] ($($ts),* ; $trailing_field_ty) => $expect);
488
                test!($(#[$cfgs])* ($($ts),* ; $trailing_field_ty) => $expect);
489
            };
490
            (@offset ; $_trailing:ty) => { trailing_field_offset!(Test, 0) };
491
            (@offset $_t:ty ; $_trailing:ty) => { trailing_field_offset!(Test, 1) };
492
        }
493
494
        test!(#[repr(C)] #[repr(transparent)] #[repr(packed)](; u8) => Some(0));
495
        test!(#[repr(C)] #[repr(transparent)] #[repr(packed)](; [u8]) => Some(0));
496
        test!(#[repr(C)] #[repr(packed)] (u8; u8) => Some(1));
497
        test!(#[repr(C)] (; AU64) => Some(0));
498
        test!(#[repr(C)] (; [AU64]) => Some(0));
499
        test!(#[repr(C)] (u8; AU64) => Some(8));
500
        test!(#[repr(C)] (u8; [AU64]) => Some(8));
501
        test!(#[repr(C)] (; Nested<u8, AU64>) => Some(0));
502
        test!(#[repr(C)] (; Nested<u8, [AU64]>) => Some(0));
503
        test!(#[repr(C)] (u8; Nested<u8, AU64>) => Some(8));
504
        test!(#[repr(C)] (u8; Nested<u8, [AU64]>) => Some(8));
505
506
        // Test that `packed(N)` limits the offset of the trailing field.
507
        test!(#[repr(C, packed(        1))] (u8; elain::Align<        2>) => Some(        1));
508
        test!(#[repr(C, packed(        2))] (u8; elain::Align<        4>) => Some(        2));
509
        test!(#[repr(C, packed(        4))] (u8; elain::Align<        8>) => Some(        4));
510
        test!(#[repr(C, packed(        8))] (u8; elain::Align<       16>) => Some(        8));
511
        test!(#[repr(C, packed(       16))] (u8; elain::Align<       32>) => Some(       16));
512
        test!(#[repr(C, packed(       32))] (u8; elain::Align<       64>) => Some(       32));
513
        test!(#[repr(C, packed(       64))] (u8; elain::Align<      128>) => Some(       64));
514
        test!(#[repr(C, packed(      128))] (u8; elain::Align<      256>) => Some(      128));
515
        test!(#[repr(C, packed(      256))] (u8; elain::Align<      512>) => Some(      256));
516
        test!(#[repr(C, packed(      512))] (u8; elain::Align<     1024>) => Some(      512));
517
        test!(#[repr(C, packed(     1024))] (u8; elain::Align<     2048>) => Some(     1024));
518
        test!(#[repr(C, packed(     2048))] (u8; elain::Align<     4096>) => Some(     2048));
519
        test!(#[repr(C, packed(     4096))] (u8; elain::Align<     8192>) => Some(     4096));
520
        test!(#[repr(C, packed(     8192))] (u8; elain::Align<    16384>) => Some(     8192));
521
        test!(#[repr(C, packed(    16384))] (u8; elain::Align<    32768>) => Some(    16384));
522
        test!(#[repr(C, packed(    32768))] (u8; elain::Align<    65536>) => Some(    32768));
523
        test!(#[repr(C, packed(    65536))] (u8; elain::Align<   131072>) => Some(    65536));
524
        /* Alignments above 65536 are not yet supported.
525
        test!(#[repr(C, packed(   131072))] (u8; elain::Align<   262144>) => Some(   131072));
526
        test!(#[repr(C, packed(   262144))] (u8; elain::Align<   524288>) => Some(   262144));
527
        test!(#[repr(C, packed(   524288))] (u8; elain::Align<  1048576>) => Some(   524288));
528
        test!(#[repr(C, packed(  1048576))] (u8; elain::Align<  2097152>) => Some(  1048576));
529
        test!(#[repr(C, packed(  2097152))] (u8; elain::Align<  4194304>) => Some(  2097152));
530
        test!(#[repr(C, packed(  4194304))] (u8; elain::Align<  8388608>) => Some(  4194304));
531
        test!(#[repr(C, packed(  8388608))] (u8; elain::Align< 16777216>) => Some(  8388608));
532
        test!(#[repr(C, packed( 16777216))] (u8; elain::Align< 33554432>) => Some( 16777216));
533
        test!(#[repr(C, packed( 33554432))] (u8; elain::Align< 67108864>) => Some( 33554432));
534
        test!(#[repr(C, packed( 67108864))] (u8; elain::Align< 33554432>) => Some( 67108864));
535
        test!(#[repr(C, packed( 33554432))] (u8; elain::Align<134217728>) => Some( 33554432));
536
        test!(#[repr(C, packed(134217728))] (u8; elain::Align<268435456>) => Some(134217728));
537
        test!(#[repr(C, packed(268435456))] (u8; elain::Align<268435456>) => Some(268435456));
538
        */
539
540
        // Test that `align(N)` does not limit the offset of the trailing field.
541
        test!(#[repr(C, align(        1))] (u8; elain::Align<        2>) => Some(        2));
542
        test!(#[repr(C, align(        2))] (u8; elain::Align<        4>) => Some(        4));
543
        test!(#[repr(C, align(        4))] (u8; elain::Align<        8>) => Some(        8));
544
        test!(#[repr(C, align(        8))] (u8; elain::Align<       16>) => Some(       16));
545
        test!(#[repr(C, align(       16))] (u8; elain::Align<       32>) => Some(       32));
546
        test!(#[repr(C, align(       32))] (u8; elain::Align<       64>) => Some(       64));
547
        test!(#[repr(C, align(       64))] (u8; elain::Align<      128>) => Some(      128));
548
        test!(#[repr(C, align(      128))] (u8; elain::Align<      256>) => Some(      256));
549
        test!(#[repr(C, align(      256))] (u8; elain::Align<      512>) => Some(      512));
550
        test!(#[repr(C, align(      512))] (u8; elain::Align<     1024>) => Some(     1024));
551
        test!(#[repr(C, align(     1024))] (u8; elain::Align<     2048>) => Some(     2048));
552
        test!(#[repr(C, align(     2048))] (u8; elain::Align<     4096>) => Some(     4096));
553
        test!(#[repr(C, align(     4096))] (u8; elain::Align<     8192>) => Some(     8192));
554
        test!(#[repr(C, align(     8192))] (u8; elain::Align<    16384>) => Some(    16384));
555
        test!(#[repr(C, align(    16384))] (u8; elain::Align<    32768>) => Some(    32768));
556
        test!(#[repr(C, align(    32768))] (u8; elain::Align<    65536>) => Some(    65536));
557
        /* Alignments above 65536 are not yet supported.
558
        test!(#[repr(C, align(    65536))] (u8; elain::Align<   131072>) => Some(   131072));
559
        test!(#[repr(C, align(   131072))] (u8; elain::Align<   262144>) => Some(   262144));
560
        test!(#[repr(C, align(   262144))] (u8; elain::Align<   524288>) => Some(   524288));
561
        test!(#[repr(C, align(   524288))] (u8; elain::Align<  1048576>) => Some(  1048576));
562
        test!(#[repr(C, align(  1048576))] (u8; elain::Align<  2097152>) => Some(  2097152));
563
        test!(#[repr(C, align(  2097152))] (u8; elain::Align<  4194304>) => Some(  4194304));
564
        test!(#[repr(C, align(  4194304))] (u8; elain::Align<  8388608>) => Some(  8388608));
565
        test!(#[repr(C, align(  8388608))] (u8; elain::Align< 16777216>) => Some( 16777216));
566
        test!(#[repr(C, align( 16777216))] (u8; elain::Align< 33554432>) => Some( 33554432));
567
        test!(#[repr(C, align( 33554432))] (u8; elain::Align< 67108864>) => Some( 67108864));
568
        test!(#[repr(C, align( 67108864))] (u8; elain::Align< 33554432>) => Some( 33554432));
569
        test!(#[repr(C, align( 33554432))] (u8; elain::Align<134217728>) => Some(134217728));
570
        test!(#[repr(C, align(134217728))] (u8; elain::Align<268435456>) => Some(268435456));
571
        */
572
    }
573
574
    // TODO(#29), TODO(https://github.com/rust-lang/rust/issues/69835): Remove
575
    // this `cfg` when `size_of_val_raw` is stabilized.
576
    #[allow(clippy::decimal_literal_representation)]
577
    #[cfg(__INTERNAL_USE_ONLY_NIGHLTY_FEATURES_IN_TESTS)]
578
    #[test]
579
    fn test_align_of_dst() {
580
        // Test that `align_of!` correctly computes the alignment of DSTs.
581
        assert_eq!(align_of!([elain::Align<1>]), Some(1));
582
        assert_eq!(align_of!([elain::Align<2>]), Some(2));
583
        assert_eq!(align_of!([elain::Align<4>]), Some(4));
584
        assert_eq!(align_of!([elain::Align<8>]), Some(8));
585
        assert_eq!(align_of!([elain::Align<16>]), Some(16));
586
        assert_eq!(align_of!([elain::Align<32>]), Some(32));
587
        assert_eq!(align_of!([elain::Align<64>]), Some(64));
588
        assert_eq!(align_of!([elain::Align<128>]), Some(128));
589
        assert_eq!(align_of!([elain::Align<256>]), Some(256));
590
        assert_eq!(align_of!([elain::Align<512>]), Some(512));
591
        assert_eq!(align_of!([elain::Align<1024>]), Some(1024));
592
        assert_eq!(align_of!([elain::Align<2048>]), Some(2048));
593
        assert_eq!(align_of!([elain::Align<4096>]), Some(4096));
594
        assert_eq!(align_of!([elain::Align<8192>]), Some(8192));
595
        assert_eq!(align_of!([elain::Align<16384>]), Some(16384));
596
        assert_eq!(align_of!([elain::Align<32768>]), Some(32768));
597
        assert_eq!(align_of!([elain::Align<65536>]), Some(65536));
598
        /* Alignments above 65536 are not yet supported.
599
        assert_eq!(align_of!([elain::Align<131072>]), Some(131072));
600
        assert_eq!(align_of!([elain::Align<262144>]), Some(262144));
601
        assert_eq!(align_of!([elain::Align<524288>]), Some(524288));
602
        assert_eq!(align_of!([elain::Align<1048576>]), Some(1048576));
603
        assert_eq!(align_of!([elain::Align<2097152>]), Some(2097152));
604
        assert_eq!(align_of!([elain::Align<4194304>]), Some(4194304));
605
        assert_eq!(align_of!([elain::Align<8388608>]), Some(8388608));
606
        assert_eq!(align_of!([elain::Align<16777216>]), Some(16777216));
607
        assert_eq!(align_of!([elain::Align<33554432>]), Some(33554432));
608
        assert_eq!(align_of!([elain::Align<67108864>]), Some(67108864));
609
        assert_eq!(align_of!([elain::Align<33554432>]), Some(33554432));
610
        assert_eq!(align_of!([elain::Align<134217728>]), Some(134217728));
611
        assert_eq!(align_of!([elain::Align<268435456>]), Some(268435456));
612
        */
613
    }
614
615
    #[test]
616
    fn test_struct_has_padding() {
617
        // Test that, for each provided repr, `struct_has_padding!` reports the
618
        // expected value.
619
        macro_rules! test {
620
            (#[$cfg:meta] ($($ts:ty),*) => $expect:expr) => {{
621
                #[$cfg]
622
                #[allow(dead_code)] // fields are never read
623
                struct Test($($ts),*);
624
                assert_eq!(struct_has_padding!(Test, $($ts),*), $expect);
625
            }};
626
            (#[$cfg:meta] $(#[$cfgs:meta])* ($($ts:ty),*) => $expect:expr) => {
627
                test!(#[$cfg] ($($ts),*) => $expect);
628
                test!($(#[$cfgs])* ($($ts),*) => $expect);
629
            };
630
        }
631
632
        test!(#[repr(C)] #[repr(transparent)] #[repr(packed)] () => false);
633
        test!(#[repr(C)] #[repr(transparent)] #[repr(packed)] (u8) => false);
634
        test!(#[repr(C)] #[repr(transparent)] #[repr(packed)] (u8, ()) => false);
635
        test!(#[repr(C)] #[repr(packed)] (u8, u8) => false);
636
637
        test!(#[repr(C)] (u8, AU64) => true);
638
        // Rust won't let you put `#[repr(packed)]` on a type which contains a
639
        // `#[repr(align(n > 1))]` type (`AU64`), so we have to use `u64` here.
640
        // It's not ideal, but it definitely has align > 1 on /some/ of our CI
641
        // targets, and this isn't a particularly complex macro we're testing
642
        // anyway.
643
        test!(#[repr(packed)] (u8, u64) => false);
644
    }
645
646
    #[test]
647
    fn test_union_has_padding() {
648
        // Test that, for each provided repr, `union_has_padding!` reports the
649
        // expected value.
650
        macro_rules! test {
651
            (#[$cfg:meta] {$($fs:ident: $ts:ty),*} => $expect:expr) => {{
652
                #[$cfg]
653
                #[allow(unused)] // fields are never read
654
                union Test{ $($fs: $ts),* }
655
                assert_eq!(union_has_padding!(Test, $($ts),*), $expect);
656
            }};
657
            (#[$cfg:meta] $(#[$cfgs:meta])* {$($fs:ident: $ts:ty),*} => $expect:expr) => {
658
                test!(#[$cfg] {$($fs: $ts),*} => $expect);
659
                test!($(#[$cfgs])* {$($fs: $ts),*} => $expect);
660
            };
661
        }
662
663
        test!(#[repr(C)] #[repr(packed)] {a: u8} => false);
664
        test!(#[repr(C)] #[repr(packed)] {a: u8, b: u8} => false);
665
666
        // Rust won't let you put `#[repr(packed)]` on a type which contains a
667
        // `#[repr(align(n > 1))]` type (`AU64`), so we have to use `u64` here.
668
        // It's not ideal, but it definitely has align > 1 on /some/ of our CI
669
        // targets, and this isn't a particularly complex macro we're testing
670
        // anyway.
671
        test!(#[repr(C)] #[repr(packed)] {a: u8, b: u64} => true);
672
    }
673
}