zerocopy/layout.rs
1// SPDX-License-Identifier: (BSD-2-Clause OR Apache-2.0) OR MIT
2//
3// Copyright 2024 The Fuchsia Authors
4//
5// Licensed under the 2-Clause BSD License <LICENSE-BSD or
6// https://opensource.org/license/bsd-2-clause>, Apache License, Version 2.0
7// <LICENSE-APACHE or https://www.apache.org/licenses/LICENSE-2.0>, or the MIT
8// license <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your option.
9// This file may not be copied, modified, or distributed except according to
10// those terms.
11
12use core::{mem, num::NonZeroUsize};
13
14use crate::util;
15
16/// The target pointer width, counted in bits.
17const POINTER_WIDTH_BITS: usize = mem::size_of::<usize>() * 8;
18
19/// The layout of a type which might be dynamically-sized.
20///
21/// `DstLayout` describes the layout of sized types, slice types, and "slice
22/// DSTs" - ie, those that are known by the type system to have a trailing slice
23/// (as distinguished from `dyn Trait` types - such types *might* have a
24/// trailing slice type, but the type system isn't aware of it).
25///
26/// Note that `DstLayout` does not have any internal invariants, so no guarantee
27/// is made that a `DstLayout` conforms to any of Rust's requirements regarding
28/// the layout of real Rust types or instances of types.
29#[doc(hidden)]
30#[allow(missing_debug_implementations, missing_copy_implementations)]
31#[cfg_attr(any(kani, test), derive(Debug, PartialEq, Eq))]
32#[derive(Copy, Clone)]
33pub struct DstLayout {
34 pub(crate) align: NonZeroUsize,
35 pub(crate) size_info: SizeInfo,
36 // Is it guaranteed statically (without knowing a value's runtime metadata)
37 // that the top-level type contains no padding? This does *not* apply
38 // recursively - for example, `[(u8, u16)]` has `statically_shallow_unpadded
39 // = true` even though this type likely has padding inside each `(u8, u16)`.
40 pub(crate) statically_shallow_unpadded: bool,
41}
42
43#[cfg_attr(any(kani, test), derive(Debug, PartialEq, Eq))]
44#[derive(Copy, Clone)]
45pub(crate) enum SizeInfo<E = usize> {
46 Sized { size: usize },
47 SliceDst(TrailingSliceLayout<E>),
48}
49
50#[cfg_attr(any(kani, test), derive(Debug, PartialEq, Eq))]
51#[derive(Copy, Clone)]
52pub(crate) struct TrailingSliceLayout<E = usize> {
53 // The offset of the first byte of the trailing slice field. Note that this
54 // is NOT the same as the minimum size of the type. For example, consider
55 // the following type:
56 //
57 // struct Foo {
58 // a: u16,
59 // b: u8,
60 // c: [u8],
61 // }
62 //
63 // In `Foo`, `c` is at byte offset 3. When `c.len() == 0`, `c` is followed
64 // by a padding byte.
65 pub(crate) offset: usize,
66 // The size of the element type of the trailing slice field.
67 pub(crate) elem_size: E,
68}
69
70impl SizeInfo {
71 /// Attempts to create a `SizeInfo` from `Self` in which `elem_size` is a
72 /// `NonZeroUsize`. If `elem_size` is 0, returns `None`.
73 #[allow(unused)]
74 #[cfg_attr(not(zerocopy_inline_always), inline)]
75 #[cfg_attr(zerocopy_inline_always, inline(always))]
76 const fn try_to_nonzero_elem_size(&self) -> Option<SizeInfo<NonZeroUsize>> {
77 Some(match *self {
78 SizeInfo::Sized { size } => SizeInfo::Sized { size },
79 SizeInfo::SliceDst(TrailingSliceLayout { offset, elem_size }) => {
80 if let Some(elem_size) = NonZeroUsize::new(elem_size) {
81 SizeInfo::SliceDst(TrailingSliceLayout { offset, elem_size })
82 } else {
83 return None;
84 }
85 }
86 })
87 }
88}
89
90#[doc(hidden)]
91#[derive(Copy, Clone)]
92#[cfg_attr(test, derive(Debug))]
93#[allow(missing_debug_implementations)]
94pub enum CastType {
95 Prefix,
96 Suffix,
97}
98
99#[cfg_attr(test, derive(Debug))]
100pub(crate) enum MetadataCastError {
101 Alignment,
102 Size,
103}
104
105impl DstLayout {
106 /// The minimum possible alignment of a type.
107 const MIN_ALIGN: NonZeroUsize = match NonZeroUsize::new(1) {
108 Some(min_align) => min_align,
109 None => const_unreachable!(),
110 };
111
112 /// The maximum theoretic possible alignment of a type.
113 ///
114 /// For compatibility with future Rust versions, this is defined as the
115 /// maximum power-of-two that fits into a `usize`. See also
116 /// [`DstLayout::CURRENT_MAX_ALIGN`].
117 pub(crate) const THEORETICAL_MAX_ALIGN: NonZeroUsize =
118 match NonZeroUsize::new(1 << (POINTER_WIDTH_BITS - 1)) {
119 Some(max_align) => max_align,
120 None => const_unreachable!(),
121 };
122
123 /// The current, documented max alignment of a type \[1\].
124 ///
125 /// \[1\] Per <https://doc.rust-lang.org/reference/type-layout.html#the-alignment-modifiers>:
126 ///
127 /// The alignment value must be a power of two from 1 up to
128 /// 2<sup>29</sup>.
129 #[cfg(not(kani))]
130 #[cfg(not(target_pointer_width = "16"))]
131 pub(crate) const CURRENT_MAX_ALIGN: NonZeroUsize = match NonZeroUsize::new(1 << 28) {
132 Some(max_align) => max_align,
133 None => const_unreachable!(),
134 };
135
136 #[cfg(not(kani))]
137 #[cfg(target_pointer_width = "16")]
138 pub(crate) const CURRENT_MAX_ALIGN: NonZeroUsize = match NonZeroUsize::new(1 << 15) {
139 Some(max_align) => max_align,
140 None => const_unreachable!(),
141 };
142
143 /// The maximum size of an allocation \[1\].
144 ///
145 /// \[1\] Per <https://doc.rust-lang.org/1.91.1/std/ptr/index.html#allocation>:
146 ///
147 /// For any allocation with base `address`, `size`, and a set of `addresses`,
148 /// the following are guaranteed: [..]
149 ///
150 /// - `size <= isize::MAX`
151 ///
152 #[allow(clippy::as_conversions)]
153 pub(crate) const MAX_SIZE: usize = isize::MAX as usize;
154
155 /// Assumes that this layout lacks static shallow padding.
156 ///
157 /// # Panics
158 ///
159 /// This method does not panic.
160 ///
161 /// # Safety
162 ///
163 /// If `self` describes the size and alignment of type that lacks static
164 /// shallow padding, unsafe code may assume that the result of this method
165 /// accurately reflects the size, alignment, and lack of static shallow
166 /// padding of that type.
167 const fn assume_shallow_unpadded(self) -> Self {
168 Self { statically_shallow_unpadded: true, ..self }
169 }
170
171 /// Constructs a `DstLayout` for a zero-sized type with `repr_align`
172 /// alignment (or 1). If `repr_align` is provided, then it must be a power
173 /// of two.
174 ///
175 /// # Panics
176 ///
177 /// This function panics if the supplied `repr_align` is not a power of two.
178 ///
179 /// # Safety
180 ///
181 /// Unsafe code may assume that the contract of this function is satisfied.
182 #[doc(hidden)]
183 #[must_use]
184 #[inline]
185 pub const fn new_zst(repr_align: Option<NonZeroUsize>) -> DstLayout {
186 let align = match repr_align {
187 Some(align) => align,
188 None => Self::MIN_ALIGN,
189 };
190
191 const_assert!(align.get().is_power_of_two());
192
193 DstLayout {
194 align,
195 size_info: SizeInfo::Sized { size: 0 },
196 statically_shallow_unpadded: true,
197 }
198 }
199
200 /// Constructs a `DstLayout` which describes `T` and assumes `T` may contain
201 /// padding.
202 ///
203 /// # Safety
204 ///
205 /// Unsafe code may assume that `DstLayout` is the correct layout for `T`.
206 #[doc(hidden)]
207 #[must_use]
208 #[inline]
209 pub const fn for_type<T>() -> DstLayout {
210 // SAFETY: `align` is correct by construction. `T: Sized`, and so it is
211 // sound to initialize `size_info` to `SizeInfo::Sized { size }`; the
212 // `size` field is also correct by construction. `unpadded` can safely
213 // default to `false`.
214 DstLayout {
215 align: match NonZeroUsize::new(mem::align_of::<T>()) {
216 Some(align) => align,
217 None => const_unreachable!(),
218 },
219 size_info: SizeInfo::Sized { size: mem::size_of::<T>() },
220 statically_shallow_unpadded: false,
221 }
222 }
223
224 /// Constructs a `DstLayout` which describes a `T` that does not contain
225 /// padding.
226 ///
227 /// # Safety
228 ///
229 /// Unsafe code may assume that `DstLayout` is the correct layout for `T`.
230 #[doc(hidden)]
231 #[must_use]
232 #[inline]
233 pub const fn for_unpadded_type<T>() -> DstLayout {
234 Self::for_type::<T>().assume_shallow_unpadded()
235 }
236
237 /// Constructs a `DstLayout` which describes `[T]`.
238 ///
239 /// # Safety
240 ///
241 /// Unsafe code may assume that `DstLayout` is the correct layout for `[T]`.
242 pub(crate) const fn for_slice<T>() -> DstLayout {
243 // SAFETY: The alignment of a slice is equal to the alignment of its
244 // element type, and so `align` is initialized correctly.
245 //
246 // Since this is just a slice type, there is no offset between the
247 // beginning of the type and the beginning of the slice, so it is
248 // correct to set `offset: 0`. The `elem_size` is correct by
249 // construction. Since `[T]` is a (degenerate case of a) slice DST, it
250 // is correct to initialize `size_info` to `SizeInfo::SliceDst`.
251 DstLayout {
252 align: match NonZeroUsize::new(mem::align_of::<T>()) {
253 Some(align) => align,
254 None => const_unreachable!(),
255 },
256 size_info: SizeInfo::SliceDst(TrailingSliceLayout {
257 offset: 0,
258 elem_size: mem::size_of::<T>(),
259 }),
260 statically_shallow_unpadded: true,
261 }
262 }
263
264 /// Constructs a complete `DstLayout` reflecting a `repr(C)` struct with the
265 /// given alignment modifiers and fields.
266 ///
267 /// This method cannot be used to match the layout of a record with the
268 /// default representation, as that representation is mostly unspecified.
269 ///
270 /// # Safety
271 ///
272 /// For any definition of a `repr(C)` struct, if this method is invoked with
273 /// alignment modifiers and fields corresponding to that definition, the
274 /// resulting `DstLayout` will correctly encode the layout of that struct.
275 ///
276 /// We make no guarantees to the behavior of this method when it is invoked
277 /// with arguments that cannot correspond to a valid `repr(C)` struct.
278 #[must_use]
279 #[inline]
280 pub const fn for_repr_c_struct(
281 repr_align: Option<NonZeroUsize>,
282 repr_packed: Option<NonZeroUsize>,
283 fields: &[DstLayout],
284 ) -> DstLayout {
285 let mut layout = DstLayout::new_zst(repr_align);
286
287 let mut i = 0;
288 #[allow(clippy::arithmetic_side_effects)]
289 while i < fields.len() {
290 #[allow(clippy::indexing_slicing)]
291 let field = fields[i];
292 layout = layout.extend(field, repr_packed);
293 i += 1;
294 }
295
296 layout = layout.pad_to_align();
297
298 // SAFETY: `layout` accurately describes the layout of a `repr(C)`
299 // struct with `repr_align` or `repr_packed` alignment modifications and
300 // the given `fields`. The `layout` is constructed using a sequence of
301 // invocations of `DstLayout::{new_zst,extend,pad_to_align}`. The
302 // documentation of these items vows that invocations in this manner
303 // will accurately describe a type, so long as:
304 //
305 // - that type is `repr(C)`,
306 // - its fields are enumerated in the order they appear,
307 // - the presence of `repr_align` and `repr_packed` are correctly accounted for.
308 //
309 // We respect all three of these preconditions above.
310 layout
311 }
312
313 /// Like `Layout::extend`, this creates a layout that describes a record
314 /// whose layout consists of `self` followed by `next` that includes the
315 /// necessary inter-field padding, but not any trailing padding.
316 ///
317 /// In order to match the layout of a `#[repr(C)]` struct, this method
318 /// should be invoked for each field in declaration order. To add trailing
319 /// padding, call `DstLayout::pad_to_align` after extending the layout for
320 /// all fields. If `self` corresponds to a type marked with
321 /// `repr(packed(N))`, then `repr_packed` should be set to `Some(N)`,
322 /// otherwise `None`.
323 ///
324 /// This method cannot be used to match the layout of a record with the
325 /// default representation, as that representation is mostly unspecified.
326 ///
327 /// # Safety
328 ///
329 /// If a (potentially hypothetical) valid `repr(C)` Rust type begins with
330 /// fields whose layout are `self`, and those fields are immediately
331 /// followed by a field whose layout is `field`, then unsafe code may rely
332 /// on `self.extend(field, repr_packed)` producing a layout that correctly
333 /// encompasses those two components.
334 ///
335 /// We make no guarantees to the behavior of this method if these fragments
336 /// cannot appear in a valid Rust type (e.g., the concatenation of the
337 /// layouts would lead to a size larger than `isize::MAX`).
338 #[doc(hidden)]
339 #[must_use]
340 #[inline]
341 pub const fn extend(self, field: DstLayout, repr_packed: Option<NonZeroUsize>) -> Self {
342 use util::{max, min, padding_needed_for};
343
344 // If `repr_packed` is `None`, there are no alignment constraints, and
345 // the value can be defaulted to `THEORETICAL_MAX_ALIGN`.
346 let max_align = match repr_packed {
347 Some(max_align) => max_align,
348 None => Self::THEORETICAL_MAX_ALIGN,
349 };
350
351 const_assert!(max_align.get().is_power_of_two());
352
353 // We use Kani to prove that this method is robust to future increases
354 // in Rust's maximum allowed alignment. However, if such a change ever
355 // actually occurs, we'd like to be notified via assertion failures.
356 #[cfg(not(kani))]
357 {
358 const_debug_assert!(self.align.get() <= DstLayout::CURRENT_MAX_ALIGN.get());
359 const_debug_assert!(field.align.get() <= DstLayout::CURRENT_MAX_ALIGN.get());
360 if let Some(repr_packed) = repr_packed {
361 const_debug_assert!(repr_packed.get() <= DstLayout::CURRENT_MAX_ALIGN.get());
362 }
363 }
364
365 // The field's alignment is clamped by `repr_packed` (i.e., the
366 // `repr(packed(N))` attribute, if any) [1].
367 //
368 // [1] Per https://doc.rust-lang.org/reference/type-layout.html#the-alignment-modifiers:
369 //
370 // The alignments of each field, for the purpose of positioning
371 // fields, is the smaller of the specified alignment and the alignment
372 // of the field's type.
373 let field_align = min(field.align, max_align);
374
375 // The struct's alignment is the maximum of its previous alignment and
376 // `field_align`.
377 let align = max(self.align, field_align);
378
379 let (interfield_padding, size_info) = match self.size_info {
380 // If the layout is already a DST, we panic; DSTs cannot be extended
381 // with additional fields.
382 SizeInfo::SliceDst(..) => const_panic!("Cannot extend a DST with additional fields."),
383
384 SizeInfo::Sized { size: preceding_size } => {
385 // Compute the minimum amount of inter-field padding needed to
386 // satisfy the field's alignment, and offset of the trailing
387 // field. [1]
388 //
389 // [1] Per https://doc.rust-lang.org/reference/type-layout.html#the-alignment-modifiers:
390 //
391 // Inter-field padding is guaranteed to be the minimum
392 // required in order to satisfy each field's (possibly
393 // altered) alignment.
394 let padding = padding_needed_for(preceding_size, field_align);
395
396 // This will not panic (and is proven to not panic, with Kani)
397 // if the layout components can correspond to a leading layout
398 // fragment of a valid Rust type, but may panic otherwise (e.g.,
399 // combining or aligning the components would create a size
400 // exceeding `isize::MAX`).
401 let offset = match preceding_size.checked_add(padding) {
402 Some(offset) => offset,
403 None => const_panic!("Adding padding to `self`'s size overflows `usize`."),
404 };
405
406 (
407 padding,
408 match field.size_info {
409 SizeInfo::Sized { size: field_size } => {
410 // If the trailing field is sized, the resulting layout
411 // will be sized. Its size will be the sum of the
412 // preceding layout, the size of the new field, and the
413 // size of inter-field padding between the two.
414 //
415 // This will not panic (and is proven with Kani to not
416 // panic) if the layout components can correspond to a
417 // leading layout fragment of a valid Rust type, but may
418 // panic otherwise (e.g., combining or aligning the
419 // components would create a size exceeding
420 // `usize::MAX`).
421 let size = match offset.checked_add(field_size) {
422 Some(size) => size,
423 None => const_panic!("`field` cannot be appended without the total size overflowing `usize`"),
424 };
425 SizeInfo::Sized { size }
426 }
427 SizeInfo::SliceDst(TrailingSliceLayout {
428 offset: trailing_offset,
429 elem_size,
430 }) => {
431 // If the trailing field is dynamically sized, so too
432 // will the resulting layout. The offset of the trailing
433 // slice component is the sum of the offset of the
434 // trailing field and the trailing slice offset within
435 // that field.
436 //
437 // This will not panic (and is proven with Kani to not
438 // panic) if the layout components can correspond to a
439 // leading layout fragment of a valid Rust type, but may
440 // panic otherwise (e.g., combining or aligning the
441 // components would create a size exceeding
442 // `usize::MAX`).
443 let offset = match offset.checked_add(trailing_offset) {
444 Some(offset) => offset,
445 None => const_panic!("`field` cannot be appended without the total size overflowing `usize`"),
446 };
447 SizeInfo::SliceDst(TrailingSliceLayout { offset, elem_size })
448 }
449 },
450 )
451 }
452 };
453
454 let statically_shallow_unpadded = self.statically_shallow_unpadded
455 && field.statically_shallow_unpadded
456 && interfield_padding == 0;
457
458 DstLayout { align, size_info, statically_shallow_unpadded }
459 }
460
461 /// Like `Layout::pad_to_align`, this routine rounds the size of this layout
462 /// up to the nearest multiple of this type's alignment or `repr_packed`
463 /// (whichever is less). This method leaves DST layouts unchanged, since the
464 /// trailing padding of DSTs is computed at runtime.
465 ///
466 /// The accompanying boolean is `true` if the resulting composition of
467 /// fields necessitated static (as opposed to dynamic) padding; otherwise
468 /// `false`.
469 ///
470 /// In order to match the layout of a `#[repr(C)]` struct, this method
471 /// should be invoked after the invocations of [`DstLayout::extend`]. If
472 /// `self` corresponds to a type marked with `repr(packed(N))`, then
473 /// `repr_packed` should be set to `Some(N)`, otherwise `None`.
474 ///
475 /// This method cannot be used to match the layout of a record with the
476 /// default representation, as that representation is mostly unspecified.
477 ///
478 /// # Safety
479 ///
480 /// If a (potentially hypothetical) valid `repr(C)` type begins with fields
481 /// whose layout are `self` followed only by zero or more bytes of trailing
482 /// padding (not included in `self`), then unsafe code may rely on
483 /// `self.pad_to_align(repr_packed)` producing a layout that correctly
484 /// encapsulates the layout of that type.
485 ///
486 /// We make no guarantees to the behavior of this method if `self` cannot
487 /// appear in a valid Rust type (e.g., because the addition of trailing
488 /// padding would lead to a size larger than `isize::MAX`).
489 #[doc(hidden)]
490 #[must_use]
491 #[inline]
492 pub const fn pad_to_align(self) -> Self {
493 use util::padding_needed_for;
494
495 let (static_padding, size_info) = match self.size_info {
496 // For sized layouts, we add the minimum amount of trailing padding
497 // needed to satisfy alignment.
498 SizeInfo::Sized { size: unpadded_size } => {
499 let padding = padding_needed_for(unpadded_size, self.align);
500 let size = match unpadded_size.checked_add(padding) {
501 Some(size) => size,
502 None => const_panic!("Adding padding caused size to overflow `usize`."),
503 };
504 (padding, SizeInfo::Sized { size })
505 }
506 // For DST layouts, trailing padding depends on the length of the
507 // trailing DST and is computed at runtime. This does not alter the
508 // offset or element size of the layout, so we leave `size_info`
509 // unchanged.
510 size_info @ SizeInfo::SliceDst(_) => (0, size_info),
511 };
512
513 let statically_shallow_unpadded = self.statically_shallow_unpadded && static_padding == 0;
514
515 DstLayout { align: self.align, size_info, statically_shallow_unpadded }
516 }
517
518 /// Produces `true` if `self` requires static padding; otherwise `false`.
519 #[must_use]
520 #[inline(always)]
521 pub const fn requires_static_padding(self) -> bool {
522 !self.statically_shallow_unpadded
523 }
524
525 /// Produces `true` if there exists any metadata for which a type of layout
526 /// `self` would require dynamic trailing padding; otherwise `false`.
527 #[must_use]
528 #[inline(always)]
529 pub const fn requires_dynamic_padding(self) -> bool {
530 // A `% self.align.get()` cannot panic, since `align` is non-zero.
531 #[allow(clippy::arithmetic_side_effects)]
532 match self.size_info {
533 SizeInfo::Sized { .. } => false,
534 SizeInfo::SliceDst(trailing_slice_layout) => {
535 // SAFETY: This predicate is formally proved sound by
536 // `proofs::prove_requires_dynamic_padding`.
537 trailing_slice_layout.offset % self.align.get() != 0
538 || trailing_slice_layout.elem_size % self.align.get() != 0
539 }
540 }
541 }
542
543 /// Validates that a cast is sound from a layout perspective.
544 ///
545 /// Validates that the size and alignment requirements of a type with the
546 /// layout described in `self` would not be violated by performing a
547 /// `cast_type` cast from a pointer with address `addr` which refers to a
548 /// memory region of size `bytes_len`.
549 ///
550 /// If the cast is valid, `validate_cast_and_convert_metadata` returns
551 /// `(elems, split_at)`. If `self` describes a dynamically-sized type, then
552 /// `elems` is the maximum number of trailing slice elements for which a
553 /// cast would be valid (for sized types, `elem` is meaningless and should
554 /// be ignored). `split_at` is the index at which to split the memory region
555 /// in order for the prefix (suffix) to contain the result of the cast, and
556 /// in order for the remaining suffix (prefix) to contain the leftover
557 /// bytes.
558 ///
559 /// There are three conditions under which a cast can fail:
560 /// - The smallest possible value for the type is larger than the provided
561 /// memory region
562 /// - A prefix cast is requested, and `addr` does not satisfy `self`'s
563 /// alignment requirement
564 /// - A suffix cast is requested, and `addr + bytes_len` does not satisfy
565 /// `self`'s alignment requirement (as a consequence, since all instances
566 /// of the type are a multiple of its alignment, no size for the type will
567 /// result in a starting address which is properly aligned)
568 ///
569 /// # Safety
570 ///
571 /// The caller may assume that this implementation is correct, and may rely
572 /// on that assumption for the soundness of their code. In particular, the
573 /// caller may assume that, if `validate_cast_and_convert_metadata` returns
574 /// `Some((elems, split_at))`, then:
575 /// - A pointer to the type (for dynamically sized types, this includes
576 /// `elems` as its pointer metadata) describes an object of size `size <=
577 /// bytes_len`
578 /// - If this is a prefix cast:
579 /// - `addr` satisfies `self`'s alignment
580 /// - `size == split_at`
581 /// - If this is a suffix cast:
582 /// - `split_at == bytes_len - size`
583 /// - `addr + split_at` satisfies `self`'s alignment
584 ///
585 /// Note that this method does *not* ensure that a pointer constructed from
586 /// its return values will be a valid pointer. In particular, this method
587 /// does not reason about `isize` overflow, which is a requirement of many
588 /// Rust pointer APIs, and may at some point be determined to be a validity
589 /// invariant of pointer types themselves. This should never be a problem so
590 /// long as the arguments to this method are derived from a known-valid
591 /// pointer (e.g., one derived from a safe Rust reference), but it is
592 /// nonetheless the caller's responsibility to justify that pointer
593 /// arithmetic will not overflow based on a safety argument *other than* the
594 /// mere fact that this method returned successfully.
595 ///
596 /// # Panics
597 ///
598 /// `validate_cast_and_convert_metadata` will panic if `self` describes a
599 /// DST whose trailing slice element is zero-sized.
600 ///
601 /// If `addr + bytes_len` overflows `usize`,
602 /// `validate_cast_and_convert_metadata` may panic, or it may return
603 /// incorrect results. No guarantees are made about when
604 /// `validate_cast_and_convert_metadata` will panic. The caller should not
605 /// rely on `validate_cast_and_convert_metadata` panicking in any particular
606 /// condition, even if `debug_assertions` are enabled.
607 #[allow(unused)]
608 #[inline(always)]
609 pub(crate) const fn validate_cast_and_convert_metadata(
610 &self,
611 addr: usize,
612 bytes_len: usize,
613 cast_type: CastType,
614 ) -> Result<(usize, usize), MetadataCastError> {
615 // `debug_assert!`, but with `#[allow(clippy::arithmetic_side_effects)]`.
616 macro_rules! __const_debug_assert {
617 ($e:expr $(, $msg:expr)?) => {
618 const_debug_assert!({
619 #[allow(clippy::arithmetic_side_effects)]
620 let e = $e;
621 e
622 } $(, $msg)?);
623 };
624 }
625
626 // Note that, in practice, `self` is always a compile-time constant. We
627 // do this check earlier than needed to ensure that we always panic as a
628 // result of bugs in the program (such as calling this function on an
629 // invalid type) instead of allowing this panic to be hidden if the cast
630 // would have failed anyway for runtime reasons (such as a too-small
631 // memory region).
632 //
633 // FIXME(#67): Once our MSRV is 1.65, use let-else:
634 // https://blog.rust-lang.org/2022/11/03/Rust-1.65.0.html#let-else-statements
635 let size_info = match self.size_info.try_to_nonzero_elem_size() {
636 Some(size_info) => size_info,
637 None => const_panic!("attempted to cast to slice type with zero-sized element"),
638 };
639
640 // Precondition
641 __const_debug_assert!(
642 addr.checked_add(bytes_len).is_some(),
643 "`addr` + `bytes_len` > usize::MAX"
644 );
645
646 // Alignment checks go in their own block to avoid introducing variables
647 // into the top-level scope.
648 {
649 // We check alignment for `addr` (for prefix casts) or `addr +
650 // bytes_len` (for suffix casts). For a prefix cast, the correctness
651 // of this check is trivial - `addr` is the address the object will
652 // live at.
653 //
654 // For a suffix cast, we know that all valid sizes for the type are
655 // a multiple of the alignment (and by safety precondition, we know
656 // `DstLayout` may only describe valid Rust types). Thus, a
657 // validly-sized instance which lives at a validly-aligned address
658 // must also end at a validly-aligned address. Thus, if the end
659 // address for a suffix cast (`addr + bytes_len`) is not aligned,
660 // then no valid start address will be aligned either.
661 let offset = match cast_type {
662 CastType::Prefix => 0,
663 CastType::Suffix => bytes_len,
664 };
665
666 // Addition is guaranteed not to overflow because `offset <=
667 // bytes_len`, and `addr + bytes_len <= usize::MAX` is a
668 // precondition of this method. Modulus is guaranteed not to divide
669 // by 0 because `align` is non-zero.
670 #[allow(clippy::arithmetic_side_effects)]
671 if (addr + offset) % self.align.get() != 0 {
672 return Err(MetadataCastError::Alignment);
673 }
674 }
675
676 let (elems, self_bytes) = match size_info {
677 SizeInfo::Sized { size } => {
678 if size > bytes_len {
679 return Err(MetadataCastError::Size);
680 }
681 (0, size)
682 }
683 SizeInfo::SliceDst(TrailingSliceLayout { offset, elem_size }) => {
684 // Calculate the maximum number of bytes that could be consumed
685 // - any number of bytes larger than this will either not be a
686 // multiple of the alignment, or will be larger than
687 // `bytes_len`.
688 let max_total_bytes =
689 util::round_down_to_next_multiple_of_alignment(bytes_len, self.align);
690 // Calculate the maximum number of bytes that could be consumed
691 // by the trailing slice.
692 //
693 // FIXME(#67): Once our MSRV is 1.65, use let-else:
694 // https://blog.rust-lang.org/2022/11/03/Rust-1.65.0.html#let-else-statements
695 let max_slice_and_padding_bytes = match max_total_bytes.checked_sub(offset) {
696 Some(max) => max,
697 // `bytes_len` too small even for 0 trailing slice elements.
698 None => return Err(MetadataCastError::Size),
699 };
700
701 // Calculate the number of elements that fit in
702 // `max_slice_and_padding_bytes`; any remaining bytes will be
703 // considered padding.
704 //
705 // Guaranteed not to divide by zero: `elem_size` is non-zero.
706 #[allow(clippy::arithmetic_side_effects)]
707 let elems = max_slice_and_padding_bytes / elem_size.get();
708 // Guaranteed not to overflow on multiplication: `usize::MAX >=
709 // max_slice_and_padding_bytes >= (max_slice_and_padding_bytes /
710 // elem_size) * elem_size`.
711 //
712 // Guaranteed not to overflow on addition:
713 // - max_slice_and_padding_bytes == max_total_bytes - offset
714 // - elems * elem_size <= max_slice_and_padding_bytes == max_total_bytes - offset
715 // - elems * elem_size + offset <= max_total_bytes <= usize::MAX
716 #[allow(clippy::arithmetic_side_effects)]
717 let without_padding = offset + elems * elem_size.get();
718 // `self_bytes` is equal to the offset bytes plus the bytes
719 // consumed by the trailing slice plus any padding bytes
720 // required to satisfy the alignment. Note that we have computed
721 // the maximum number of trailing slice elements that could fit
722 // in `self_bytes`, so any padding is guaranteed to be less than
723 // the size of an extra element.
724 //
725 // Guaranteed not to overflow:
726 // - By previous comment: without_padding == elems * elem_size +
727 // offset <= max_total_bytes
728 // - By construction, `max_total_bytes` is a multiple of
729 // `self.align`.
730 // - At most, adding padding needed to round `without_padding`
731 // up to the next multiple of the alignment will bring
732 // `self_bytes` up to `max_total_bytes`.
733 #[allow(clippy::arithmetic_side_effects)]
734 let self_bytes =
735 without_padding + util::padding_needed_for(without_padding, self.align);
736 (elems, self_bytes)
737 }
738 };
739
740 __const_debug_assert!(self_bytes <= bytes_len);
741
742 let split_at = match cast_type {
743 CastType::Prefix => self_bytes,
744 // Guaranteed not to underflow:
745 // - In the `Sized` branch, only returns `size` if `size <=
746 // bytes_len`.
747 // - In the `SliceDst` branch, calculates `self_bytes <=
748 // max_toatl_bytes`, which is upper-bounded by `bytes_len`.
749 #[allow(clippy::arithmetic_side_effects)]
750 CastType::Suffix => bytes_len - self_bytes,
751 };
752
753 Ok((elems, split_at))
754 }
755}
756
757pub(crate) use cast_from::CastFrom;
758mod cast_from {
759 use crate::*;
760
761 pub(crate) struct CastFrom<Dst: ?Sized> {
762 _never: core::convert::Infallible,
763 _marker: PhantomData<Dst>,
764 }
765
766 // SAFETY: The implementation of `Project::project` preserves the address
767 // of the referent – it only modifies pointer metadata.
768 unsafe impl<Src, Dst> crate::pointer::cast::Cast<Src, Dst> for CastFrom<Dst>
769 where
770 Src: KnownLayout + ?Sized,
771 Dst: KnownLayout + ?Sized,
772 {
773 }
774
775 // SAFETY: The implementation of `Project::project` preserves the size of
776 // the referent (see inline comments for a more detailed proof of this).
777 unsafe impl<Src, Dst> crate::pointer::cast::CastExact<Src, Dst> for CastFrom<Dst>
778 where
779 Src: KnownLayout + ?Sized,
780 Dst: KnownLayout + ?Sized,
781 {
782 }
783
784 // SAFETY: `project` produces a pointer which refers to the same referent
785 // bytes as its input, or to a subset of them (see inline comments for a
786 // more detailed proof of this). It does this using provenance-preserving
787 // operations.
788 unsafe impl<Src, Dst> crate::pointer::cast::Project<Src, Dst> for CastFrom<Dst>
789 where
790 Src: KnownLayout + ?Sized,
791 Dst: KnownLayout + ?Sized,
792 {
793 /// # PME
794 ///
795 /// Generates a post-monomorphization error if it is not possible to
796 /// implement soundly.
797 //
798 // FIXME(#1817): Support Sized->Unsized and Unsized->Sized casts
799 fn project(src: PtrInner<'_, Src>) -> *mut Dst {
800 /// The parameters required in order to perform a pointer cast from
801 /// `Src` to `Dst`.
802 ///
803 /// These are a compile-time function of the layouts of `Src`
804 /// and `Dst`.
805 ///
806 /// # Safety
807 ///
808 /// `Src`'s alignment must not be smaller than `Dst`'s alignment.
809 struct CastParams<Src: ?Sized, Dst: ?Sized> {
810 inner: CastParamsInner,
811 _src: PhantomData<Src>,
812 _dst: PhantomData<Dst>,
813 }
814
815 #[derive(Copy, Clone)]
816 enum CastParamsInner {
817 // At compile time (specifically, post-monomorphization time),
818 // we need to compute two things:
819 // - Whether, given *any* `*Src`, it is possible to construct a
820 // `*Dst` which addresses the same number of bytes (ie,
821 // whether, for any `Src` pointer metadata, there exists `Dst`
822 // pointer metadata that addresses the same number of bytes)
823 // - If this is possible, any information necessary to perform
824 // the `Src`->`Dst` metadata conversion at runtime.
825 //
826 // Assume that `Src` and `Dst` are slice DSTs, and define:
827 // - `S_OFF = Src::LAYOUT.size_info.offset`
828 // - `S_ELEM = Src::LAYOUT.size_info.elem_size`
829 // - `D_OFF = Dst::LAYOUT.size_info.offset`
830 // - `D_ELEM = Dst::LAYOUT.size_info.elem_size`
831 //
832 // We are trying to solve the following equation:
833 //
834 // D_OFF + d_meta * D_ELEM = S_OFF + s_meta * S_ELEM
835 //
836 // At runtime, we will be attempting to compute `d_meta`, given
837 // `s_meta` (a runtime value) and all other parameters (which
838 // are compile-time values). We can solve like so:
839 //
840 // D_OFF + d_meta * D_ELEM = S_OFF + s_meta * S_ELEM
841 //
842 // d_meta * D_ELEM = S_OFF - D_OFF + s_meta * S_ELEM
843 //
844 // d_meta = (S_OFF - D_OFF + s_meta * S_ELEM)/D_ELEM
845 //
846 // Since `d_meta` will be a `usize`, we need the right-hand side
847 // to be an integer, and this needs to hold for *any* value of
848 // `s_meta` (in order for our conversion to be infallible - ie,
849 // to not have to reject certain values of `s_meta` at runtime).
850 // This means that:
851 //
852 // - `s_meta * S_ELEM` must be a multiple of `D_ELEM`
853 // - Since this must hold for any value of `s_meta`, `S_ELEM`
854 // must be a multiple of `D_ELEM`
855 // - `S_OFF - D_OFF` must be a multiple of `D_ELEM`
856 //
857 // Thus, let `OFFSET_DELTA_ELEMS = (S_OFF - D_OFF)/D_ELEM` and
858 // `ELEM_MULTIPLE = S_ELEM/D_ELEM`. We can rewrite the above
859 // expression as:
860 //
861 // d_meta = (S_OFF - D_OFF + s_meta * S_ELEM)/D_ELEM
862 //
863 // d_meta = OFFSET_DELTA_ELEMS + s_meta * ELEM_MULTIPLE
864 //
865 // Thus, we just need to compute the following and confirm that
866 // they have integer solutions in order to both a) determine
867 // whether infallible `Src` -> `Dst` casts are possible and, b)
868 // pre-compute the parameters necessary to perform those casts
869 // at runtime. These parameters are encapsulated in
870 // `CastParams`, which acts as a witness that such infallible
871 // casts are possible.
872 /// The parameters required in order to perform an
873 /// unsized-to-unsized pointer cast from `Src` to `Dst` as
874 /// described above.
875 ///
876 /// # Safety
877 ///
878 /// `Src` and `Dst` must both be slice DSTs.
879 ///
880 /// `offset_delta_elems` and `elem_multiple` must be valid as
881 /// described above.
882 UnsizedToUnsized { offset_delta_elems: usize, elem_multiple: usize },
883
884 /// The metadata of a `Dst` which has the same size as `Src:
885 /// Sized`.
886 ///
887 /// # Safety
888 ///
889 /// `Src: Sized` and `Dst` must be a slice DST.
890 ///
891 /// A raw `Dst` pointer with metadata `dst_meta` must address
892 /// `size_of::<Src>()` bytes.
893 SizedToUnsized { dst_meta: usize },
894
895 /// The metadata of a `Dst` which has the same size as `Src:
896 /// Sized`.
897 ///
898 /// # Safety
899 ///
900 /// `Src` and `Dst` must both be `Sized` and `size_of::<Src>()
901 /// == size_of::<Dst>()`.
902 SizedToSized,
903 }
904
905 impl<Src: ?Sized, Dst: ?Sized> Copy for CastParams<Src, Dst> {}
906 impl<Src: ?Sized, Dst: ?Sized> Clone for CastParams<Src, Dst> {
907 fn clone(&self) -> Self {
908 *self
909 }
910 }
911
912 impl<Src: ?Sized, Dst: ?Sized> CastParams<Src, Dst> {
913 const fn try_compute(
914 src: &DstLayout,
915 dst: &DstLayout,
916 ) -> Option<CastParams<Src, Dst>> {
917 if src.align.get() < dst.align.get() {
918 return None;
919 }
920
921 let inner = match (src.size_info, dst.size_info) {
922 (
923 SizeInfo::Sized { size: src_size },
924 SizeInfo::Sized { size: dst_size },
925 ) => {
926 if src_size != dst_size {
927 return None;
928 }
929
930 // SAFETY: We checked above that `src_size ==
931 // dst_size`.
932 CastParamsInner::SizedToSized
933 }
934 (SizeInfo::Sized { size: src_size }, SizeInfo::SliceDst(dst)) => {
935 let offset_delta = if let Some(od) = src_size.checked_sub(dst.offset) {
936 od
937 } else {
938 return None;
939 };
940
941 let dst_elem_size = if let Some(e) = NonZeroUsize::new(dst.elem_size) {
942 e
943 } else {
944 return None;
945 };
946
947 // PANICS: `dst_elem_size: NonZeroUsize`, so this won't
948 // divide by zero.
949 #[allow(clippy::arithmetic_side_effects)]
950 let delta_mod_other_elem = offset_delta % dst_elem_size.get();
951
952 if delta_mod_other_elem != 0 {
953 return None;
954 }
955
956 // PANICS: `dst_elem_size: NonZeroUsize`, so this won't
957 // divide by zero.
958 #[allow(clippy::arithmetic_side_effects)]
959 let dst_meta = offset_delta / dst_elem_size.get();
960
961 // SAFETY: The preceding math ensures that a `Dst`
962 // with `dst_meta` addresses `src_size` bytes.
963 CastParamsInner::SizedToUnsized { dst_meta }
964 }
965 (SizeInfo::SliceDst(src), SizeInfo::SliceDst(dst)) => {
966 let offset_delta = if let Some(od) = src.offset.checked_sub(dst.offset)
967 {
968 od
969 } else {
970 return None;
971 };
972
973 let dst_elem_size = if let Some(e) = NonZeroUsize::new(dst.elem_size) {
974 e
975 } else {
976 return None;
977 };
978
979 // PANICS: `dst_elem_size: NonZeroUsize`, so this won't
980 // divide by zero.
981 #[allow(clippy::arithmetic_side_effects)]
982 let delta_mod_other_elem = offset_delta % dst_elem_size.get();
983
984 // PANICS: `dst_elem_size: NonZeroUsize`, so this won't
985 // divide by zero.
986 #[allow(clippy::arithmetic_side_effects)]
987 let elem_remainder = src.elem_size % dst_elem_size.get();
988
989 if delta_mod_other_elem != 0
990 || src.elem_size < dst.elem_size
991 || elem_remainder != 0
992 {
993 return None;
994 }
995
996 // PANICS: `dst_elem_size: NonZeroUsize`, so this won't
997 // divide by zero.
998 #[allow(clippy::arithmetic_side_effects)]
999 let offset_delta_elems = offset_delta / dst_elem_size.get();
1000
1001 // PANICS: `dst_elem_size: NonZeroUsize`, so this won't
1002 // divide by zero.
1003 #[allow(clippy::arithmetic_side_effects)]
1004 let elem_multiple = src.elem_size / dst_elem_size.get();
1005
1006 CastParamsInner::UnsizedToUnsized {
1007 // SAFETY: We checked above that this is an exact ratio.
1008 offset_delta_elems,
1009 // SAFETY: We checked above that this is an exact ratio.
1010 elem_multiple,
1011 }
1012 }
1013 _ => return None,
1014 };
1015
1016 // SAFETY: We checked above that `src.align >= dst.align`.
1017 Some(CastParams { inner, _src: PhantomData, _dst: PhantomData })
1018 }
1019 }
1020
1021 impl<Src: KnownLayout + ?Sized, Dst: KnownLayout + ?Sized> CastParams<Src, Dst> {
1022 /// # Safety
1023 ///
1024 /// `src_meta` describes a `Src` whose size is no larger than
1025 /// `isize::MAX`.
1026 ///
1027 /// The returned metadata describes a `Dst` of the same size as
1028 /// the original `Src`.
1029 #[inline(always)]
1030 unsafe fn cast_metadata(
1031 self,
1032 src_meta: Src::PointerMetadata,
1033 ) -> Dst::PointerMetadata {
1034 #[allow(unused)]
1035 use crate::util::polyfills::*;
1036
1037 let dst_meta = match self.inner {
1038 CastParamsInner::UnsizedToUnsized { offset_delta_elems, elem_multiple } => {
1039 let src_meta = src_meta.to_elem_count();
1040 #[allow(
1041 unstable_name_collisions,
1042 clippy::multiple_unsafe_ops_per_block
1043 )]
1044 // SAFETY: `self` is a witness that the following
1045 // equation holds:
1046 //
1047 // D_OFF + d_meta * D_ELEM = S_OFF + s_meta * S_ELEM
1048 //
1049 // Since the caller promises that `src_meta` is
1050 // valid `Src` metadata, this math will not
1051 // overflow, and the returned value will describe a
1052 // `Dst` of the same size.
1053 unsafe {
1054 offset_delta_elems
1055 .unchecked_add(src_meta.unchecked_mul(elem_multiple))
1056 }
1057 }
1058 CastParamsInner::SizedToUnsized { dst_meta } => dst_meta,
1059 CastParamsInner::SizedToSized => 0,
1060 };
1061 Dst::PointerMetadata::from_elem_count(dst_meta)
1062 }
1063 }
1064
1065 trait Params<Src: ?Sized> {
1066 const CAST_PARAMS: CastParams<Src, Self>;
1067 }
1068
1069 impl<Src, Dst> Params<Src> for Dst
1070 where
1071 Src: KnownLayout + ?Sized,
1072 Dst: KnownLayout + ?Sized,
1073 {
1074 const CAST_PARAMS: CastParams<Src, Dst> =
1075 match CastParams::try_compute(&Src::LAYOUT, &Dst::LAYOUT) {
1076 Some(params) => params,
1077 None => const_panic!(
1078 "cannot `transmute_ref!` or `transmute_mut!` between incompatible types"
1079 ),
1080 };
1081 }
1082
1083 let src_meta = <Src as KnownLayout>::pointer_to_metadata(src.as_ptr());
1084 let params = <Dst as Params<Src>>::CAST_PARAMS;
1085
1086 // SAFETY: `src: PtrInner` guarantees that `src`'s referent is zero
1087 // bytes or lives in a single allocation, which means that it is no
1088 // larger than `isize::MAX` bytes [1].
1089 //
1090 // [1] https://doc.rust-lang.org/1.92.0/std/ptr/index.html#allocation
1091 let dst_meta = unsafe { params.cast_metadata(src_meta) };
1092
1093 <Dst as KnownLayout>::raw_from_ptr_len(src.as_non_null().cast(), dst_meta).as_ptr()
1094 }
1095 }
1096}
1097
1098// FIXME(#67): For some reason, on our MSRV toolchain, this `allow` isn't
1099// enforced despite having `#![allow(unknown_lints)]` at the crate root, but
1100// putting it here works. Once our MSRV is high enough that this bug has been
1101// fixed, remove this `allow`.
1102#[allow(unknown_lints)]
1103#[cfg(test)]
1104mod tests {
1105 use super::*;
1106
1107 #[test]
1108 fn test_dst_layout_for_slice() {
1109 let layout = DstLayout::for_slice::<u32>();
1110 match layout.size_info {
1111 SizeInfo::SliceDst(TrailingSliceLayout { offset, elem_size }) => {
1112 assert_eq!(offset, 0);
1113 assert_eq!(elem_size, 4);
1114 }
1115 _ => panic!("Expected SliceDst"),
1116 }
1117 assert_eq!(layout.align.get(), 4);
1118 }
1119
1120 /// Tests of when a sized `DstLayout` is extended with a sized field.
1121 #[allow(clippy::decimal_literal_representation)]
1122 #[test]
1123 fn test_dst_layout_extend_sized_with_sized() {
1124 // This macro constructs a layout corresponding to a `u8` and extends it
1125 // with a zero-sized trailing field of given alignment `n`. The macro
1126 // tests that the resulting layout has both size and alignment `min(n,
1127 // P)` for all valid values of `repr(packed(P))`.
1128 macro_rules! test_align_is_size {
1129 ($n:expr) => {
1130 let base = DstLayout::for_type::<u8>();
1131 let trailing_field = DstLayout::for_type::<elain::Align<$n>>();
1132
1133 let packs =
1134 core::iter::once(None).chain((0..29).map(|p| NonZeroUsize::new(2usize.pow(p))));
1135
1136 for pack in packs {
1137 let composite = base.extend(trailing_field, pack);
1138 let max_align = pack.unwrap_or(DstLayout::CURRENT_MAX_ALIGN);
1139 let align = $n.min(max_align.get());
1140 assert_eq!(
1141 composite,
1142 DstLayout {
1143 align: NonZeroUsize::new(align).unwrap(),
1144 size_info: SizeInfo::Sized { size: align },
1145 statically_shallow_unpadded: false,
1146 }
1147 )
1148 }
1149 };
1150 }
1151
1152 test_align_is_size!(1);
1153 test_align_is_size!(2);
1154 test_align_is_size!(4);
1155 test_align_is_size!(8);
1156 test_align_is_size!(16);
1157 test_align_is_size!(32);
1158 test_align_is_size!(64);
1159 test_align_is_size!(128);
1160 test_align_is_size!(256);
1161 test_align_is_size!(512);
1162 test_align_is_size!(1024);
1163 test_align_is_size!(2048);
1164 test_align_is_size!(4096);
1165 test_align_is_size!(8192);
1166 test_align_is_size!(16384);
1167 test_align_is_size!(32768);
1168 test_align_is_size!(65536);
1169 test_align_is_size!(131072);
1170 test_align_is_size!(262144);
1171 test_align_is_size!(524288);
1172 test_align_is_size!(1048576);
1173 test_align_is_size!(2097152);
1174 test_align_is_size!(4194304);
1175 test_align_is_size!(8388608);
1176 test_align_is_size!(16777216);
1177 test_align_is_size!(33554432);
1178 test_align_is_size!(67108864);
1179 test_align_is_size!(33554432);
1180 test_align_is_size!(134217728);
1181 test_align_is_size!(268435456);
1182 }
1183
1184 /// Tests of when a sized `DstLayout` is extended with a DST field.
1185 #[test]
1186 fn test_dst_layout_extend_sized_with_dst() {
1187 // Test that for all combinations of real-world alignments and
1188 // `repr_packed` values, that the extension of a sized `DstLayout`` with
1189 // a DST field correctly computes the trailing offset in the composite
1190 // layout.
1191
1192 let aligns = (0..29).map(|p| NonZeroUsize::new(2usize.pow(p)).unwrap());
1193 let packs = core::iter::once(None).chain(aligns.clone().map(Some));
1194
1195 for align in aligns {
1196 for pack in packs.clone() {
1197 let base = DstLayout::for_type::<u8>();
1198 let elem_size = 42;
1199 let trailing_field_offset = 11;
1200
1201 let trailing_field = DstLayout {
1202 align,
1203 size_info: SizeInfo::SliceDst(TrailingSliceLayout { elem_size, offset: 11 }),
1204 statically_shallow_unpadded: false,
1205 };
1206
1207 let composite = base.extend(trailing_field, pack);
1208
1209 let max_align = pack.unwrap_or(DstLayout::CURRENT_MAX_ALIGN).get();
1210
1211 let align = align.get().min(max_align);
1212
1213 assert_eq!(
1214 composite,
1215 DstLayout {
1216 align: NonZeroUsize::new(align).unwrap(),
1217 size_info: SizeInfo::SliceDst(TrailingSliceLayout {
1218 elem_size,
1219 offset: align + trailing_field_offset,
1220 }),
1221 statically_shallow_unpadded: false,
1222 }
1223 )
1224 }
1225 }
1226 }
1227
1228 /// Tests that calling `pad_to_align` on a sized `DstLayout` adds the
1229 /// expected amount of trailing padding.
1230 #[test]
1231 fn test_dst_layout_pad_to_align_with_sized() {
1232 // For all valid alignments `align`, construct a one-byte layout aligned
1233 // to `align`, call `pad_to_align`, and assert that the size of the
1234 // resulting layout is equal to `align`.
1235 for align in (0..29).map(|p| NonZeroUsize::new(2usize.pow(p)).unwrap()) {
1236 let layout = DstLayout {
1237 align,
1238 size_info: SizeInfo::Sized { size: 1 },
1239 statically_shallow_unpadded: true,
1240 };
1241
1242 assert_eq!(
1243 layout.pad_to_align(),
1244 DstLayout {
1245 align,
1246 size_info: SizeInfo::Sized { size: align.get() },
1247 statically_shallow_unpadded: align.get() == 1
1248 }
1249 );
1250 }
1251
1252 // Test explicitly-provided combinations of unpadded and padded
1253 // counterparts.
1254
1255 macro_rules! test {
1256 (unpadded { size: $unpadded_size:expr, align: $unpadded_align:expr }
1257 => padded { size: $padded_size:expr, align: $padded_align:expr }) => {
1258 let unpadded = DstLayout {
1259 align: NonZeroUsize::new($unpadded_align).unwrap(),
1260 size_info: SizeInfo::Sized { size: $unpadded_size },
1261 statically_shallow_unpadded: false,
1262 };
1263 let padded = unpadded.pad_to_align();
1264
1265 assert_eq!(
1266 padded,
1267 DstLayout {
1268 align: NonZeroUsize::new($padded_align).unwrap(),
1269 size_info: SizeInfo::Sized { size: $padded_size },
1270 statically_shallow_unpadded: false,
1271 }
1272 );
1273 };
1274 }
1275
1276 test!(unpadded { size: 0, align: 4 } => padded { size: 0, align: 4 });
1277 test!(unpadded { size: 1, align: 4 } => padded { size: 4, align: 4 });
1278 test!(unpadded { size: 2, align: 4 } => padded { size: 4, align: 4 });
1279 test!(unpadded { size: 3, align: 4 } => padded { size: 4, align: 4 });
1280 test!(unpadded { size: 4, align: 4 } => padded { size: 4, align: 4 });
1281 test!(unpadded { size: 5, align: 4 } => padded { size: 8, align: 4 });
1282 test!(unpadded { size: 6, align: 4 } => padded { size: 8, align: 4 });
1283 test!(unpadded { size: 7, align: 4 } => padded { size: 8, align: 4 });
1284 test!(unpadded { size: 8, align: 4 } => padded { size: 8, align: 4 });
1285
1286 let current_max_align = DstLayout::CURRENT_MAX_ALIGN.get();
1287
1288 test!(unpadded { size: 1, align: current_max_align }
1289 => padded { size: current_max_align, align: current_max_align });
1290
1291 test!(unpadded { size: current_max_align + 1, align: current_max_align }
1292 => padded { size: current_max_align * 2, align: current_max_align });
1293 }
1294
1295 /// Tests that calling `pad_to_align` on a DST `DstLayout` is a no-op.
1296 #[test]
1297 fn test_dst_layout_pad_to_align_with_dst() {
1298 for align in (0..29).map(|p| NonZeroUsize::new(2usize.pow(p)).unwrap()) {
1299 for offset in 0..10 {
1300 for elem_size in 0..10 {
1301 let layout = DstLayout {
1302 align,
1303 size_info: SizeInfo::SliceDst(TrailingSliceLayout { offset, elem_size }),
1304 statically_shallow_unpadded: false,
1305 };
1306 assert_eq!(layout.pad_to_align(), layout);
1307 }
1308 }
1309 }
1310 }
1311
1312 // This test takes a long time when running under Miri, so we skip it in
1313 // that case. This is acceptable because this is a logic test that doesn't
1314 // attempt to expose UB.
1315 #[test]
1316 #[cfg_attr(miri, ignore)]
1317 fn test_validate_cast_and_convert_metadata() {
1318 #[allow(non_local_definitions)]
1319 impl From<usize> for SizeInfo {
1320 fn from(size: usize) -> SizeInfo {
1321 SizeInfo::Sized { size }
1322 }
1323 }
1324
1325 #[allow(non_local_definitions)]
1326 impl From<(usize, usize)> for SizeInfo {
1327 fn from((offset, elem_size): (usize, usize)) -> SizeInfo {
1328 SizeInfo::SliceDst(TrailingSliceLayout { offset, elem_size })
1329 }
1330 }
1331
1332 fn layout<S: Into<SizeInfo>>(s: S, align: usize) -> DstLayout {
1333 DstLayout {
1334 size_info: s.into(),
1335 align: NonZeroUsize::new(align).unwrap(),
1336 statically_shallow_unpadded: false,
1337 }
1338 }
1339
1340 /// This macro accepts arguments in the form of:
1341 ///
1342 /// layout(_, _).validate(_, _, _), Ok(Some((_, _)))
1343 /// | | | | | | |
1344 /// size ---------+ | | | | | |
1345 /// align -----------+ | | | | |
1346 /// addr ------------------------+ | | | |
1347 /// bytes_len ----------------------+ | | |
1348 /// cast_type -------------------------+ | |
1349 /// elems ------------------------------------------+ |
1350 /// split_at ------------------------------------------+
1351 ///
1352 /// `.validate` is shorthand for `.validate_cast_and_convert_metadata`
1353 /// for brevity.
1354 ///
1355 /// Each argument can either be an iterator or a wildcard. Each
1356 /// wildcarded variable is implicitly replaced by an iterator over a
1357 /// representative sample of values for that variable. Each `test!`
1358 /// invocation iterates over every combination of values provided by
1359 /// each variable's iterator (ie, the cartesian product) and validates
1360 /// that the results are expected.
1361 ///
1362 /// The final argument uses the same syntax, but it has a different
1363 /// meaning:
1364 /// - If it is `Ok(pat)`, then the pattern `pat` is supplied to
1365 /// a matching assert to validate the computed result for each
1366 /// combination of input values.
1367 /// - If it is `Err(Some(msg) | None)`, then `test!` validates that the
1368 /// call to `validate_cast_and_convert_metadata` panics with the given
1369 /// panic message or, if the current Rust toolchain version is too
1370 /// early to support panicking in `const fn`s, panics with *some*
1371 /// message. In the latter case, the `const_panic!` macro is used,
1372 /// which emits code which causes a non-panicking error at const eval
1373 /// time, but which does panic when invoked at runtime. Thus, it is
1374 /// merely difficult to predict the *value* of this panic. We deem
1375 /// that testing against the real panic strings on stable and nightly
1376 /// toolchains is enough to ensure correctness.
1377 ///
1378 /// Note that the meta-variables that match these variables have the
1379 /// `tt` type, and some valid expressions are not valid `tt`s (such as
1380 /// `a..b`). In this case, wrap the expression in parentheses, and it
1381 /// will become valid `tt`.
1382 macro_rules! test {
1383 (
1384 layout($size:tt, $align:tt)
1385 .validate($addr:tt, $bytes_len:tt, $cast_type:tt), $expect:pat $(,)?
1386 ) => {
1387 itertools::iproduct!(
1388 test!(@generate_size $size),
1389 test!(@generate_align $align),
1390 test!(@generate_usize $addr),
1391 test!(@generate_usize $bytes_len),
1392 test!(@generate_cast_type $cast_type)
1393 ).for_each(|(size_info, align, addr, bytes_len, cast_type)| {
1394 // Temporarily disable the panic hook installed by the test
1395 // harness. If we don't do this, all panic messages will be
1396 // kept in an internal log. On its own, this isn't a
1397 // problem, but if a non-caught panic ever happens (ie, in
1398 // code later in this test not in this macro), all of the
1399 // previously-buffered messages will be dumped, hiding the
1400 // real culprit.
1401 let previous_hook = std::panic::take_hook();
1402 // I don't understand why, but this seems to be required in
1403 // addition to the previous line.
1404 std::panic::set_hook(Box::new(|_| {}));
1405 let actual = std::panic::catch_unwind(|| {
1406 layout(size_info, align).validate_cast_and_convert_metadata(addr, bytes_len, cast_type)
1407 }).map_err(|d| {
1408 let msg = d.downcast::<&'static str>().ok().map(|s| *s.as_ref());
1409 assert!(msg.is_some() || cfg!(no_zerocopy_panic_in_const_and_vec_try_reserve_1_57_0), "non-string panic messages are not permitted when usage of panic in const fn is enabled");
1410 msg
1411 });
1412 std::panic::set_hook(previous_hook);
1413
1414 assert!(
1415 matches!(actual, $expect),
1416 "layout({:?}, {}).validate_cast_and_convert_metadata({}, {}, {:?})" ,size_info, align, addr, bytes_len, cast_type
1417 );
1418 });
1419 };
1420 (@generate_usize _) => { 0..8 };
1421 // Generate sizes for both Sized and !Sized types.
1422 (@generate_size _) => {
1423 test!(@generate_size (_)).chain(test!(@generate_size (_, _)))
1424 };
1425 // Generate sizes for both Sized and !Sized types by chaining
1426 // specified iterators for each.
1427 (@generate_size ($sized_sizes:tt | $unsized_sizes:tt)) => {
1428 test!(@generate_size ($sized_sizes)).chain(test!(@generate_size $unsized_sizes))
1429 };
1430 // Generate sizes for Sized types.
1431 (@generate_size (_)) => { test!(@generate_size (0..8)) };
1432 (@generate_size ($sizes:expr)) => { $sizes.into_iter().map(Into::<SizeInfo>::into) };
1433 // Generate sizes for !Sized types.
1434 (@generate_size ($min_sizes:tt, $elem_sizes:tt)) => {
1435 itertools::iproduct!(
1436 test!(@generate_min_size $min_sizes),
1437 test!(@generate_elem_size $elem_sizes)
1438 ).map(Into::<SizeInfo>::into)
1439 };
1440 (@generate_fixed_size _) => { (0..8).into_iter().map(Into::<SizeInfo>::into) };
1441 (@generate_min_size _) => { 0..8 };
1442 (@generate_elem_size _) => { 1..8 };
1443 (@generate_align _) => { [1, 2, 4, 8, 16] };
1444 (@generate_opt_usize _) => { [None].into_iter().chain((0..8).map(Some).into_iter()) };
1445 (@generate_cast_type _) => { [CastType::Prefix, CastType::Suffix] };
1446 (@generate_cast_type $variant:ident) => { [CastType::$variant] };
1447 // Some expressions need to be wrapped in parentheses in order to be
1448 // valid `tt`s (required by the top match pattern). See the comment
1449 // below for more details. This arm removes these parentheses to
1450 // avoid generating an `unused_parens` warning.
1451 (@$_:ident ($vals:expr)) => { $vals };
1452 (@$_:ident $vals:expr) => { $vals };
1453 }
1454
1455 const EVENS: [usize; 8] = [0, 2, 4, 6, 8, 10, 12, 14];
1456 const ODDS: [usize; 8] = [1, 3, 5, 7, 9, 11, 13, 15];
1457
1458 // base_size is too big for the memory region.
1459 test!(
1460 layout(((1..8) | ((1..8), (1..8))), _).validate([0], [0], _),
1461 Ok(Err(MetadataCastError::Size))
1462 );
1463 test!(
1464 layout(((2..8) | ((2..8), (2..8))), _).validate([0], [1], Prefix),
1465 Ok(Err(MetadataCastError::Size))
1466 );
1467 test!(
1468 layout(((2..8) | ((2..8), (2..8))), _).validate([0x1000_0000 - 1], [1], Suffix),
1469 Ok(Err(MetadataCastError::Size))
1470 );
1471
1472 // addr is unaligned for prefix cast
1473 test!(layout(_, [2]).validate(ODDS, _, Prefix), Ok(Err(MetadataCastError::Alignment)));
1474 test!(layout(_, [2]).validate(ODDS, _, Prefix), Ok(Err(MetadataCastError::Alignment)));
1475
1476 // addr is aligned, but end of buffer is unaligned for suffix cast
1477 test!(layout(_, [2]).validate(EVENS, ODDS, Suffix), Ok(Err(MetadataCastError::Alignment)));
1478 test!(layout(_, [2]).validate(EVENS, ODDS, Suffix), Ok(Err(MetadataCastError::Alignment)));
1479
1480 // Unfortunately, these constants cannot easily be used in the
1481 // implementation of `validate_cast_and_convert_metadata`, since
1482 // `panic!` consumes a string literal, not an expression.
1483 //
1484 // It's important that these messages be in a separate module. If they
1485 // were at the function's top level, we'd pass them to `test!` as, e.g.,
1486 // `Err(TRAILING)`, which would run into a subtle Rust footgun - the
1487 // `TRAILING` identifier would be treated as a pattern to match rather
1488 // than a value to check for equality.
1489 mod msgs {
1490 pub(super) const TRAILING: &str =
1491 "attempted to cast to slice type with zero-sized element";
1492 pub(super) const OVERFLOW: &str = "`addr` + `bytes_len` > usize::MAX";
1493 }
1494
1495 // casts with ZST trailing element types are unsupported
1496 test!(layout((_, [0]), _).validate(_, _, _), Err(Some(msgs::TRAILING) | None),);
1497
1498 // addr + bytes_len must not overflow usize
1499 test!(layout(_, _).validate([usize::MAX], (1..100), _), Err(Some(msgs::OVERFLOW) | None));
1500 test!(layout(_, _).validate((1..100), [usize::MAX], _), Err(Some(msgs::OVERFLOW) | None));
1501 test!(
1502 layout(_, _).validate(
1503 [usize::MAX / 2 + 1, usize::MAX],
1504 [usize::MAX / 2 + 1, usize::MAX],
1505 _
1506 ),
1507 Err(Some(msgs::OVERFLOW) | None)
1508 );
1509
1510 // Validates that `validate_cast_and_convert_metadata` satisfies its own
1511 // documented safety postconditions, and also a few other properties
1512 // that aren't documented but we want to guarantee anyway.
1513 fn validate_behavior(
1514 (layout, addr, bytes_len, cast_type): (DstLayout, usize, usize, CastType),
1515 ) {
1516 if let Ok((elems, split_at)) =
1517 layout.validate_cast_and_convert_metadata(addr, bytes_len, cast_type)
1518 {
1519 let (size_info, align) = (layout.size_info, layout.align);
1520 let debug_str = format!(
1521 "layout({:?}, {}).validate_cast_and_convert_metadata({}, {}, {:?}) => ({}, {})",
1522 size_info, align, addr, bytes_len, cast_type, elems, split_at
1523 );
1524
1525 // If this is a sized type (no trailing slice), then `elems` is
1526 // meaningless, but in practice we set it to 0. Callers are not
1527 // allowed to rely on this, but a lot of math is nicer if
1528 // they're able to, and some callers might accidentally do that.
1529 let sized = matches!(layout.size_info, SizeInfo::Sized { .. });
1530 assert!(!(sized && elems != 0), "{}", debug_str);
1531
1532 let resulting_size = match layout.size_info {
1533 SizeInfo::Sized { size } => size,
1534 SizeInfo::SliceDst(TrailingSliceLayout { offset, elem_size }) => {
1535 let padded_size = |elems| {
1536 let without_padding = offset + elems * elem_size;
1537 without_padding + util::padding_needed_for(without_padding, align)
1538 };
1539
1540 let resulting_size = padded_size(elems);
1541 // Test that `validate_cast_and_convert_metadata`
1542 // computed the largest possible value that fits in the
1543 // given range.
1544 assert!(padded_size(elems + 1) > bytes_len, "{}", debug_str);
1545 resulting_size
1546 }
1547 };
1548
1549 // Test safety postconditions guaranteed by
1550 // `validate_cast_and_convert_metadata`.
1551 assert!(resulting_size <= bytes_len, "{}", debug_str);
1552 match cast_type {
1553 CastType::Prefix => {
1554 assert_eq!(addr % align, 0, "{}", debug_str);
1555 assert_eq!(resulting_size, split_at, "{}", debug_str);
1556 }
1557 CastType::Suffix => {
1558 assert_eq!(split_at, bytes_len - resulting_size, "{}", debug_str);
1559 assert_eq!((addr + split_at) % align, 0, "{}", debug_str);
1560 }
1561 }
1562 } else {
1563 let min_size = match layout.size_info {
1564 SizeInfo::Sized { size } => size,
1565 SizeInfo::SliceDst(TrailingSliceLayout { offset, .. }) => {
1566 offset + util::padding_needed_for(offset, layout.align)
1567 }
1568 };
1569
1570 // If a cast is invalid, it is either because...
1571 // 1. there are insufficient bytes at the given region for type:
1572 let insufficient_bytes = bytes_len < min_size;
1573 // 2. performing the cast would misalign type:
1574 let base = match cast_type {
1575 CastType::Prefix => 0,
1576 CastType::Suffix => bytes_len,
1577 };
1578 let misaligned = (base + addr) % layout.align != 0;
1579
1580 assert!(insufficient_bytes || misaligned);
1581 }
1582 }
1583
1584 let sizes = 0..8;
1585 let elem_sizes = 1..8;
1586 let size_infos = sizes
1587 .clone()
1588 .map(Into::<SizeInfo>::into)
1589 .chain(itertools::iproduct!(sizes, elem_sizes).map(Into::<SizeInfo>::into));
1590 let layouts = itertools::iproduct!(size_infos, [1, 2, 4, 8, 16, 32])
1591 .filter(|(size_info, align)| !matches!(size_info, SizeInfo::Sized { size } if size % align != 0))
1592 .map(|(size_info, align)| layout(size_info, align));
1593 itertools::iproduct!(layouts, 0..8, 0..8, [CastType::Prefix, CastType::Suffix])
1594 .for_each(validate_behavior);
1595 }
1596
1597 #[test]
1598 #[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS)]
1599 fn test_validate_rust_layout() {
1600 use core::{
1601 convert::TryInto as _,
1602 ptr::{self, NonNull},
1603 };
1604
1605 use crate::util::testutil::*;
1606
1607 // This test synthesizes pointers with various metadata and uses Rust's
1608 // built-in APIs to confirm that Rust makes decisions about type layout
1609 // which are consistent with what we believe is guaranteed by the
1610 // language. If this test fails, it doesn't just mean our code is wrong
1611 // - it means we're misunderstanding the language's guarantees.
1612
1613 #[derive(Debug)]
1614 struct MacroArgs {
1615 offset: usize,
1616 align: NonZeroUsize,
1617 elem_size: Option<usize>,
1618 }
1619
1620 /// # Safety
1621 ///
1622 /// `test` promises to only call `addr_of_slice_field` on a `NonNull<T>`
1623 /// which points to a valid `T`.
1624 ///
1625 /// `with_elems` must produce a pointer which points to a valid `T`.
1626 fn test<T: ?Sized, W: Fn(usize) -> NonNull<T>>(
1627 args: MacroArgs,
1628 with_elems: W,
1629 addr_of_slice_field: Option<fn(NonNull<T>) -> NonNull<u8>>,
1630 ) {
1631 let dst = args.elem_size.is_some();
1632 let layout = {
1633 let size_info = match args.elem_size {
1634 Some(elem_size) => {
1635 SizeInfo::SliceDst(TrailingSliceLayout { offset: args.offset, elem_size })
1636 }
1637 None => SizeInfo::Sized {
1638 // Rust only supports types whose sizes are a multiple
1639 // of their alignment. If the macro created a type like
1640 // this:
1641 //
1642 // #[repr(C, align(2))]
1643 // struct Foo([u8; 1]);
1644 //
1645 // ...then Rust will automatically round the type's size
1646 // up to 2.
1647 size: args.offset + util::padding_needed_for(args.offset, args.align),
1648 },
1649 };
1650 DstLayout { size_info, align: args.align, statically_shallow_unpadded: false }
1651 };
1652
1653 for elems in 0..128 {
1654 let ptr = with_elems(elems);
1655
1656 if let Some(addr_of_slice_field) = addr_of_slice_field {
1657 let slc_field_ptr = addr_of_slice_field(ptr).as_ptr();
1658 // SAFETY: Both `slc_field_ptr` and `ptr` are pointers to
1659 // the same valid Rust object.
1660 // Work around https://github.com/rust-lang/rust-clippy/issues/12280
1661 let offset: usize =
1662 unsafe { slc_field_ptr.byte_offset_from(ptr.as_ptr()).try_into().unwrap() };
1663 assert_eq!(offset, args.offset);
1664 }
1665
1666 // SAFETY: `ptr` points to a valid `T`.
1667 #[allow(clippy::multiple_unsafe_ops_per_block)]
1668 let (size, align) = unsafe {
1669 (mem::size_of_val_raw(ptr.as_ptr()), mem::align_of_val_raw(ptr.as_ptr()))
1670 };
1671
1672 // Avoid expensive allocation when running under Miri.
1673 let assert_msg = if !cfg!(miri) {
1674 format!("\n{:?}\nsize:{}, align:{}", args, size, align)
1675 } else {
1676 String::new()
1677 };
1678
1679 let without_padding =
1680 args.offset + args.elem_size.map(|elem_size| elems * elem_size).unwrap_or(0);
1681 assert!(size >= without_padding, "{}", assert_msg);
1682 assert_eq!(align, args.align.get(), "{}", assert_msg);
1683
1684 // This encodes the most important part of the test: our
1685 // understanding of how Rust determines the layout of repr(C)
1686 // types. Sized repr(C) types are trivial, but DST types have
1687 // some subtlety. Note that:
1688 // - For sized types, `without_padding` is just the size of the
1689 // type that we constructed for `Foo`. Since we may have
1690 // requested a larger alignment, `Foo` may actually be larger
1691 // than this, hence `padding_needed_for`.
1692 // - For unsized types, `without_padding` is dynamically
1693 // computed from the offset, the element size, and element
1694 // count. We expect that the size of the object should be
1695 // `offset + elem_size * elems` rounded up to the next
1696 // alignment.
1697 let expected_size =
1698 without_padding + util::padding_needed_for(without_padding, args.align);
1699 assert_eq!(expected_size, size, "{}", assert_msg);
1700
1701 // For zero-sized element types,
1702 // `validate_cast_and_convert_metadata` just panics, so we skip
1703 // testing those types.
1704 if args.elem_size.map(|elem_size| elem_size > 0).unwrap_or(true) {
1705 let addr = ptr.addr().get();
1706 let (got_elems, got_split_at) = layout
1707 .validate_cast_and_convert_metadata(addr, size, CastType::Prefix)
1708 .unwrap();
1709 // Avoid expensive allocation when running under Miri.
1710 let assert_msg = if !cfg!(miri) {
1711 format!(
1712 "{}\nvalidate_cast_and_convert_metadata({}, {})",
1713 assert_msg, addr, size,
1714 )
1715 } else {
1716 String::new()
1717 };
1718 assert_eq!(got_split_at, size, "{}", assert_msg);
1719 if dst {
1720 assert!(got_elems >= elems, "{}", assert_msg);
1721 if got_elems != elems {
1722 // If `validate_cast_and_convert_metadata`
1723 // returned more elements than `elems`, that
1724 // means that `elems` is not the maximum number
1725 // of elements that can fit in `size` - in other
1726 // words, there is enough padding at the end of
1727 // the value to fit at least one more element.
1728 // If we use this metadata to synthesize a
1729 // pointer, despite having a different element
1730 // count, we still expect it to have the same
1731 // size.
1732 let got_ptr = with_elems(got_elems);
1733 // SAFETY: `got_ptr` is a pointer to a valid `T`.
1734 let size_of_got_ptr = unsafe { mem::size_of_val_raw(got_ptr.as_ptr()) };
1735 assert_eq!(size_of_got_ptr, size, "{}", assert_msg);
1736 }
1737 } else {
1738 // For sized casts, the returned element value is
1739 // technically meaningless, and we don't guarantee any
1740 // particular value. In practice, it's always zero.
1741 assert_eq!(got_elems, 0, "{}", assert_msg)
1742 }
1743 }
1744 }
1745 }
1746
1747 macro_rules! validate_against_rust {
1748 ($offset:literal, $align:literal $(, $elem_size:literal)?) => {{
1749 #[repr(C, align($align))]
1750 struct Foo([u8; $offset]$(, [[u8; $elem_size]])?);
1751
1752 let args = MacroArgs {
1753 offset: $offset,
1754 align: $align.try_into().unwrap(),
1755 elem_size: {
1756 #[allow(unused)]
1757 let ret = None::<usize>;
1758 $(let ret = Some($elem_size);)?
1759 ret
1760 }
1761 };
1762
1763 #[repr(C, align($align))]
1764 struct FooAlign;
1765 // Create an aligned buffer to use in order to synthesize
1766 // pointers to `Foo`. We don't ever load values from these
1767 // pointers - we just do arithmetic on them - so having a "real"
1768 // block of memory as opposed to a validly-aligned-but-dangling
1769 // pointer is only necessary to make Miri happy since we run it
1770 // with "strict provenance" checking enabled.
1771 let aligned_buf = Align::<_, FooAlign>::new([0u8; 1024]);
1772 let with_elems = |elems| {
1773 let slc = NonNull::slice_from_raw_parts(NonNull::from(&aligned_buf.t), elems);
1774 #[allow(clippy::as_conversions)]
1775 NonNull::new(slc.as_ptr() as *mut Foo).unwrap()
1776 };
1777 let addr_of_slice_field = {
1778 #[allow(unused)]
1779 let f = None::<fn(NonNull<Foo>) -> NonNull<u8>>;
1780 $(
1781 // SAFETY: `test` promises to only call `f` with a `ptr`
1782 // to a valid `Foo`.
1783 let f: Option<fn(NonNull<Foo>) -> NonNull<u8>> = Some(|ptr: NonNull<Foo>| unsafe {
1784 NonNull::new(ptr::addr_of_mut!((*ptr.as_ptr()).1)).unwrap().cast::<u8>()
1785 });
1786 let _ = $elem_size;
1787 )?
1788 f
1789 };
1790
1791 test::<Foo, _>(args, with_elems, addr_of_slice_field);
1792 }};
1793 }
1794
1795 // Every permutation of:
1796 // - offset in [0, 4]
1797 // - align in [1, 16]
1798 // - elem_size in [0, 4] (plus no elem_size)
1799 validate_against_rust!(0, 1);
1800 validate_against_rust!(0, 1, 0);
1801 validate_against_rust!(0, 1, 1);
1802 validate_against_rust!(0, 1, 2);
1803 validate_against_rust!(0, 1, 3);
1804 validate_against_rust!(0, 1, 4);
1805 validate_against_rust!(0, 2);
1806 validate_against_rust!(0, 2, 0);
1807 validate_against_rust!(0, 2, 1);
1808 validate_against_rust!(0, 2, 2);
1809 validate_against_rust!(0, 2, 3);
1810 validate_against_rust!(0, 2, 4);
1811 validate_against_rust!(0, 4);
1812 validate_against_rust!(0, 4, 0);
1813 validate_against_rust!(0, 4, 1);
1814 validate_against_rust!(0, 4, 2);
1815 validate_against_rust!(0, 4, 3);
1816 validate_against_rust!(0, 4, 4);
1817 validate_against_rust!(0, 8);
1818 validate_against_rust!(0, 8, 0);
1819 validate_against_rust!(0, 8, 1);
1820 validate_against_rust!(0, 8, 2);
1821 validate_against_rust!(0, 8, 3);
1822 validate_against_rust!(0, 8, 4);
1823 validate_against_rust!(0, 16);
1824 validate_against_rust!(0, 16, 0);
1825 validate_against_rust!(0, 16, 1);
1826 validate_against_rust!(0, 16, 2);
1827 validate_against_rust!(0, 16, 3);
1828 validate_against_rust!(0, 16, 4);
1829 validate_against_rust!(1, 1);
1830 validate_against_rust!(1, 1, 0);
1831 validate_against_rust!(1, 1, 1);
1832 validate_against_rust!(1, 1, 2);
1833 validate_against_rust!(1, 1, 3);
1834 validate_against_rust!(1, 1, 4);
1835 validate_against_rust!(1, 2);
1836 validate_against_rust!(1, 2, 0);
1837 validate_against_rust!(1, 2, 1);
1838 validate_against_rust!(1, 2, 2);
1839 validate_against_rust!(1, 2, 3);
1840 validate_against_rust!(1, 2, 4);
1841 validate_against_rust!(1, 4);
1842 validate_against_rust!(1, 4, 0);
1843 validate_against_rust!(1, 4, 1);
1844 validate_against_rust!(1, 4, 2);
1845 validate_against_rust!(1, 4, 3);
1846 validate_against_rust!(1, 4, 4);
1847 validate_against_rust!(1, 8);
1848 validate_against_rust!(1, 8, 0);
1849 validate_against_rust!(1, 8, 1);
1850 validate_against_rust!(1, 8, 2);
1851 validate_against_rust!(1, 8, 3);
1852 validate_against_rust!(1, 8, 4);
1853 validate_against_rust!(1, 16);
1854 validate_against_rust!(1, 16, 0);
1855 validate_against_rust!(1, 16, 1);
1856 validate_against_rust!(1, 16, 2);
1857 validate_against_rust!(1, 16, 3);
1858 validate_against_rust!(1, 16, 4);
1859 validate_against_rust!(2, 1);
1860 validate_against_rust!(2, 1, 0);
1861 validate_against_rust!(2, 1, 1);
1862 validate_against_rust!(2, 1, 2);
1863 validate_against_rust!(2, 1, 3);
1864 validate_against_rust!(2, 1, 4);
1865 validate_against_rust!(2, 2);
1866 validate_against_rust!(2, 2, 0);
1867 validate_against_rust!(2, 2, 1);
1868 validate_against_rust!(2, 2, 2);
1869 validate_against_rust!(2, 2, 3);
1870 validate_against_rust!(2, 2, 4);
1871 validate_against_rust!(2, 4);
1872 validate_against_rust!(2, 4, 0);
1873 validate_against_rust!(2, 4, 1);
1874 validate_against_rust!(2, 4, 2);
1875 validate_against_rust!(2, 4, 3);
1876 validate_against_rust!(2, 4, 4);
1877 validate_against_rust!(2, 8);
1878 validate_against_rust!(2, 8, 0);
1879 validate_against_rust!(2, 8, 1);
1880 validate_against_rust!(2, 8, 2);
1881 validate_against_rust!(2, 8, 3);
1882 validate_against_rust!(2, 8, 4);
1883 validate_against_rust!(2, 16);
1884 validate_against_rust!(2, 16, 0);
1885 validate_against_rust!(2, 16, 1);
1886 validate_against_rust!(2, 16, 2);
1887 validate_against_rust!(2, 16, 3);
1888 validate_against_rust!(2, 16, 4);
1889 validate_against_rust!(3, 1);
1890 validate_against_rust!(3, 1, 0);
1891 validate_against_rust!(3, 1, 1);
1892 validate_against_rust!(3, 1, 2);
1893 validate_against_rust!(3, 1, 3);
1894 validate_against_rust!(3, 1, 4);
1895 validate_against_rust!(3, 2);
1896 validate_against_rust!(3, 2, 0);
1897 validate_against_rust!(3, 2, 1);
1898 validate_against_rust!(3, 2, 2);
1899 validate_against_rust!(3, 2, 3);
1900 validate_against_rust!(3, 2, 4);
1901 validate_against_rust!(3, 4);
1902 validate_against_rust!(3, 4, 0);
1903 validate_against_rust!(3, 4, 1);
1904 validate_against_rust!(3, 4, 2);
1905 validate_against_rust!(3, 4, 3);
1906 validate_against_rust!(3, 4, 4);
1907 validate_against_rust!(3, 8);
1908 validate_against_rust!(3, 8, 0);
1909 validate_against_rust!(3, 8, 1);
1910 validate_against_rust!(3, 8, 2);
1911 validate_against_rust!(3, 8, 3);
1912 validate_against_rust!(3, 8, 4);
1913 validate_against_rust!(3, 16);
1914 validate_against_rust!(3, 16, 0);
1915 validate_against_rust!(3, 16, 1);
1916 validate_against_rust!(3, 16, 2);
1917 validate_against_rust!(3, 16, 3);
1918 validate_against_rust!(3, 16, 4);
1919 validate_against_rust!(4, 1);
1920 validate_against_rust!(4, 1, 0);
1921 validate_against_rust!(4, 1, 1);
1922 validate_against_rust!(4, 1, 2);
1923 validate_against_rust!(4, 1, 3);
1924 validate_against_rust!(4, 1, 4);
1925 validate_against_rust!(4, 2);
1926 validate_against_rust!(4, 2, 0);
1927 validate_against_rust!(4, 2, 1);
1928 validate_against_rust!(4, 2, 2);
1929 validate_against_rust!(4, 2, 3);
1930 validate_against_rust!(4, 2, 4);
1931 validate_against_rust!(4, 4);
1932 validate_against_rust!(4, 4, 0);
1933 validate_against_rust!(4, 4, 1);
1934 validate_against_rust!(4, 4, 2);
1935 validate_against_rust!(4, 4, 3);
1936 validate_against_rust!(4, 4, 4);
1937 validate_against_rust!(4, 8);
1938 validate_against_rust!(4, 8, 0);
1939 validate_against_rust!(4, 8, 1);
1940 validate_against_rust!(4, 8, 2);
1941 validate_against_rust!(4, 8, 3);
1942 validate_against_rust!(4, 8, 4);
1943 validate_against_rust!(4, 16);
1944 validate_against_rust!(4, 16, 0);
1945 validate_against_rust!(4, 16, 1);
1946 validate_against_rust!(4, 16, 2);
1947 validate_against_rust!(4, 16, 3);
1948 validate_against_rust!(4, 16, 4);
1949 }
1950}
1951
1952#[cfg(kani)]
1953mod proofs {
1954 use core::alloc::Layout;
1955
1956 use super::*;
1957
1958 impl kani::Arbitrary for DstLayout {
1959 fn any() -> Self {
1960 let align: NonZeroUsize = kani::any();
1961 let size_info: SizeInfo = kani::any();
1962
1963 kani::assume(align.is_power_of_two());
1964 kani::assume(align < DstLayout::THEORETICAL_MAX_ALIGN);
1965
1966 // For testing purposes, we most care about instantiations of
1967 // `DstLayout` that can correspond to actual Rust types. We use
1968 // `Layout` to verify that our `DstLayout` satisfies the validity
1969 // conditions of Rust layouts.
1970 kani::assume(
1971 match size_info {
1972 SizeInfo::Sized { size } => Layout::from_size_align(size, align.get()),
1973 SizeInfo::SliceDst(TrailingSliceLayout { offset, elem_size: _ }) => {
1974 // `SliceDst` cannot encode an exact size, but we know
1975 // it is at least `offset` bytes.
1976 Layout::from_size_align(offset, align.get())
1977 }
1978 }
1979 .is_ok(),
1980 );
1981
1982 Self { align: align, size_info: size_info, statically_shallow_unpadded: kani::any() }
1983 }
1984 }
1985
1986 impl kani::Arbitrary for SizeInfo {
1987 fn any() -> Self {
1988 let is_sized: bool = kani::any();
1989
1990 match is_sized {
1991 true => {
1992 let size: usize = kani::any();
1993
1994 kani::assume(size <= DstLayout::MAX_SIZE);
1995
1996 SizeInfo::Sized { size }
1997 }
1998 false => SizeInfo::SliceDst(kani::any()),
1999 }
2000 }
2001 }
2002
2003 impl kani::Arbitrary for TrailingSliceLayout {
2004 fn any() -> Self {
2005 let elem_size: usize = kani::any();
2006 let offset: usize = kani::any();
2007
2008 kani::assume(elem_size < DstLayout::MAX_SIZE);
2009 kani::assume(offset < DstLayout::MAX_SIZE);
2010
2011 TrailingSliceLayout { elem_size, offset }
2012 }
2013 }
2014
2015 #[kani::proof]
2016 fn prove_requires_dynamic_padding() {
2017 let layout: DstLayout = kani::any();
2018
2019 let SizeInfo::SliceDst(size_info) = layout.size_info else {
2020 kani::assume(false);
2021 loop {}
2022 };
2023
2024 let meta: usize = kani::any();
2025
2026 let Some(trailing_slice_size) = size_info.elem_size.checked_mul(meta) else {
2027 // The `trailing_slice_size` exceeds `usize::MAX`; `meta` is invalid.
2028 kani::assume(false);
2029 loop {}
2030 };
2031
2032 let Some(unpadded_size) = size_info.offset.checked_add(trailing_slice_size) else {
2033 // The `unpadded_size` exceeds `usize::MAX`; `meta`` is invalid.
2034 kani::assume(false);
2035 loop {}
2036 };
2037
2038 if unpadded_size >= DstLayout::MAX_SIZE {
2039 // The `unpadded_size` exceeds `isize::MAX`; `meta` is invalid.
2040 kani::assume(false);
2041 loop {}
2042 }
2043
2044 let trailing_padding = util::padding_needed_for(unpadded_size, layout.align);
2045
2046 if !layout.requires_dynamic_padding() {
2047 assert!(trailing_padding == 0);
2048 }
2049 }
2050
2051 #[kani::proof]
2052 fn prove_dst_layout_extend() {
2053 use crate::util::{max, min, padding_needed_for};
2054
2055 let base: DstLayout = kani::any();
2056 let field: DstLayout = kani::any();
2057 let packed: Option<NonZeroUsize> = kani::any();
2058
2059 if let Some(max_align) = packed {
2060 kani::assume(max_align.is_power_of_two());
2061 kani::assume(base.align <= max_align);
2062 }
2063
2064 // The base can only be extended if it's sized.
2065 kani::assume(matches!(base.size_info, SizeInfo::Sized { .. }));
2066 let base_size = if let SizeInfo::Sized { size } = base.size_info {
2067 size
2068 } else {
2069 unreachable!();
2070 };
2071
2072 // Under the above conditions, `DstLayout::extend` will not panic.
2073 let composite = base.extend(field, packed);
2074
2075 // The field's alignment is clamped by `max_align` (i.e., the
2076 // `packed` attribute, if any) [1].
2077 //
2078 // [1] Per https://doc.rust-lang.org/reference/type-layout.html#the-alignment-modifiers:
2079 //
2080 // The alignments of each field, for the purpose of positioning
2081 // fields, is the smaller of the specified alignment and the
2082 // alignment of the field's type.
2083 let field_align = min(field.align, packed.unwrap_or(DstLayout::THEORETICAL_MAX_ALIGN));
2084
2085 // The struct's alignment is the maximum of its previous alignment and
2086 // `field_align`.
2087 assert_eq!(composite.align, max(base.align, field_align));
2088
2089 // Compute the minimum amount of inter-field padding needed to
2090 // satisfy the field's alignment, and offset of the trailing field.
2091 // [1]
2092 //
2093 // [1] Per https://doc.rust-lang.org/reference/type-layout.html#the-alignment-modifiers:
2094 //
2095 // Inter-field padding is guaranteed to be the minimum required in
2096 // order to satisfy each field's (possibly altered) alignment.
2097 let padding = padding_needed_for(base_size, field_align);
2098 let offset = base_size + padding;
2099
2100 // For testing purposes, we'll also construct `alloc::Layout`
2101 // stand-ins for `DstLayout`, and show that `extend` behaves
2102 // comparably on both types.
2103 let base_analog = Layout::from_size_align(base_size, base.align.get()).unwrap();
2104
2105 match field.size_info {
2106 SizeInfo::Sized { size: field_size } => {
2107 if let SizeInfo::Sized { size: composite_size } = composite.size_info {
2108 // If the trailing field is sized, the resulting layout will
2109 // be sized. Its size will be the sum of the preceding
2110 // layout, the size of the new field, and the size of
2111 // inter-field padding between the two.
2112 assert_eq!(composite_size, offset + field_size);
2113
2114 let field_analog =
2115 Layout::from_size_align(field_size, field_align.get()).unwrap();
2116
2117 if let Ok((actual_composite, actual_offset)) = base_analog.extend(field_analog)
2118 {
2119 assert_eq!(actual_offset, offset);
2120 assert_eq!(actual_composite.size(), composite_size);
2121 assert_eq!(actual_composite.align(), composite.align.get());
2122 } else {
2123 // An error here reflects that composite of `base`
2124 // and `field` cannot correspond to a real Rust type
2125 // fragment, because such a fragment would violate
2126 // the basic invariants of a valid Rust layout. At
2127 // the time of writing, `DstLayout` is a little more
2128 // permissive than `Layout`, so we don't assert
2129 // anything in this branch (e.g., unreachability).
2130 }
2131 } else {
2132 panic!("The composite of two sized layouts must be sized.")
2133 }
2134 }
2135 SizeInfo::SliceDst(TrailingSliceLayout {
2136 offset: field_offset,
2137 elem_size: field_elem_size,
2138 }) => {
2139 if let SizeInfo::SliceDst(TrailingSliceLayout {
2140 offset: composite_offset,
2141 elem_size: composite_elem_size,
2142 }) = composite.size_info
2143 {
2144 // The offset of the trailing slice component is the sum
2145 // of the offset of the trailing field and the trailing
2146 // slice offset within that field.
2147 assert_eq!(composite_offset, offset + field_offset);
2148 // The elem size is unchanged.
2149 assert_eq!(composite_elem_size, field_elem_size);
2150
2151 let field_analog =
2152 Layout::from_size_align(field_offset, field_align.get()).unwrap();
2153
2154 if let Ok((actual_composite, actual_offset)) = base_analog.extend(field_analog)
2155 {
2156 assert_eq!(actual_offset, offset);
2157 assert_eq!(actual_composite.size(), composite_offset);
2158 assert_eq!(actual_composite.align(), composite.align.get());
2159 } else {
2160 // An error here reflects that composite of `base`
2161 // and `field` cannot correspond to a real Rust type
2162 // fragment, because such a fragment would violate
2163 // the basic invariants of a valid Rust layout. At
2164 // the time of writing, `DstLayout` is a little more
2165 // permissive than `Layout`, so we don't assert
2166 // anything in this branch (e.g., unreachability).
2167 }
2168 } else {
2169 panic!("The extension of a layout with a DST must result in a DST.")
2170 }
2171 }
2172 }
2173 }
2174
2175 #[kani::proof]
2176 #[kani::should_panic]
2177 fn prove_dst_layout_extend_dst_panics() {
2178 let base: DstLayout = kani::any();
2179 let field: DstLayout = kani::any();
2180 let packed: Option<NonZeroUsize> = kani::any();
2181
2182 if let Some(max_align) = packed {
2183 kani::assume(max_align.is_power_of_two());
2184 kani::assume(base.align <= max_align);
2185 }
2186
2187 kani::assume(matches!(base.size_info, SizeInfo::SliceDst(..)));
2188
2189 let _ = base.extend(field, packed);
2190 }
2191
2192 #[kani::proof]
2193 fn prove_dst_layout_pad_to_align() {
2194 use crate::util::padding_needed_for;
2195
2196 let layout: DstLayout = kani::any();
2197
2198 let padded = layout.pad_to_align();
2199
2200 // Calling `pad_to_align` does not alter the `DstLayout`'s alignment.
2201 assert_eq!(padded.align, layout.align);
2202
2203 if let SizeInfo::Sized { size: unpadded_size } = layout.size_info {
2204 if let SizeInfo::Sized { size: padded_size } = padded.size_info {
2205 // If the layout is sized, it will remain sized after padding is
2206 // added. Its sum will be its unpadded size and the size of the
2207 // trailing padding needed to satisfy its alignment
2208 // requirements.
2209 let padding = padding_needed_for(unpadded_size, layout.align);
2210 assert_eq!(padded_size, unpadded_size + padding);
2211
2212 // Prove that calling `DstLayout::pad_to_align` behaves
2213 // identically to `Layout::pad_to_align`.
2214 let layout_analog =
2215 Layout::from_size_align(unpadded_size, layout.align.get()).unwrap();
2216 let padded_analog = layout_analog.pad_to_align();
2217 assert_eq!(padded_analog.align(), layout.align.get());
2218 assert_eq!(padded_analog.size(), padded_size);
2219 } else {
2220 panic!("The padding of a sized layout must result in a sized layout.")
2221 }
2222 } else {
2223 // If the layout is a DST, padding cannot be statically added.
2224 assert_eq!(padded.size_info, layout.size_info);
2225 }
2226 }
2227}