Skip to main content

zerocopy/pointer/
ptr.rs

1// SPDX-License-Identifier: BSD-2-Clause OR Apache-2.0 OR MIT
2//
3// Copyright 2023 The Fuchsia Authors
4//
5// Licensed under a BSD-style license <LICENSE-BSD>, Apache License, Version 2.0
6// <LICENSE-APACHE or https://www.apache.org/licenses/LICENSE-2.0>, or the MIT
7// license <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your option.
8// This file may not be copied, modified, or distributed except according to
9// those terms.
10
11#![allow(missing_docs)]
12
13use core::{
14    fmt::{Debug, Formatter},
15    marker::PhantomData,
16};
17
18use crate::{
19    pointer::{
20        inner::PtrInner,
21        invariant::*,
22        transmute::{MutationCompatible, SizeEq, TransmuteFromPtr},
23    },
24    AlignmentError, CastError, CastType, KnownLayout, SizeError, TryFromBytes, ValidityError,
25};
26
27/// Module used to gate access to [`Ptr`]'s fields.
28mod def {
29    #[cfg(doc)]
30    use super::super::invariant;
31    use super::*;
32
33    /// A raw pointer with more restrictions.
34    ///
35    /// `Ptr<T>` is similar to [`NonNull<T>`], but it is more restrictive in the
36    /// following ways (note that these requirements only hold of non-zero-sized
37    /// referents):
38    /// - It must derive from a valid allocation.
39    /// - It must reference a byte range which is contained inside the
40    ///   allocation from which it derives.
41    ///   - As a consequence, the byte range it references must have a size
42    ///     which does not overflow `isize`.
43    ///
44    /// Depending on how `Ptr` is parameterized, it may have additional
45    /// invariants:
46    /// - `ptr` conforms to the aliasing invariant of
47    ///   [`I::Aliasing`](invariant::Aliasing).
48    /// - `ptr` conforms to the alignment invariant of
49    ///   [`I::Alignment`](invariant::Alignment).
50    /// - `ptr` conforms to the validity invariant of
51    ///   [`I::Validity`](invariant::Validity).
52    ///
53    /// `Ptr<'a, T>` is [covariant] in `'a` and invariant in `T`.
54    ///
55    /// [`NonNull<T>`]: core::ptr::NonNull
56    /// [covariant]: https://doc.rust-lang.org/reference/subtyping.html
57    pub struct Ptr<'a, T, I>
58    where
59        T: ?Sized,
60        I: Invariants,
61    {
62        /// # Invariants
63        ///
64        /// 0. `ptr` conforms to the aliasing invariant of
65        ///    [`I::Aliasing`](invariant::Aliasing).
66        /// 1. `ptr` conforms to the alignment invariant of
67        ///    [`I::Alignment`](invariant::Alignment).
68        /// 2. `ptr` conforms to the validity invariant of
69        ///    [`I::Validity`](invariant::Validity).
70        // SAFETY: `PtrInner<'a, T>` is covariant in `'a` and invariant in `T`.
71        ptr: PtrInner<'a, T>,
72        _invariants: PhantomData<I>,
73    }
74
75    impl<'a, T, I> Ptr<'a, T, I>
76    where
77        T: 'a + ?Sized,
78        I: Invariants,
79    {
80        /// Constructs a new `Ptr` from a [`PtrInner`].
81        ///
82        /// # Safety
83        ///
84        /// The caller promises that:
85        ///
86        /// 0. `ptr` conforms to the aliasing invariant of
87        ///    [`I::Aliasing`](invariant::Aliasing).
88        /// 1. `ptr` conforms to the alignment invariant of
89        ///    [`I::Alignment`](invariant::Alignment).
90        /// 2. `ptr` conforms to the validity invariant of
91        ///    [`I::Validity`](invariant::Validity).
92        pub(crate) unsafe fn from_inner(ptr: PtrInner<'a, T>) -> Ptr<'a, T, I> {
93            // SAFETY: The caller has promised to satisfy all safety invariants
94            // of `Ptr`.
95            Self { ptr, _invariants: PhantomData }
96        }
97
98        /// Converts this `Ptr<T>` to a [`PtrInner<T>`].
99        ///
100        /// Note that this method does not consume `self`. The caller should
101        /// watch out for `unsafe` code which uses the returned value in a way
102        /// that violates the safety invariants of `self`.
103        #[inline]
104        #[must_use]
105        pub fn as_inner(&self) -> PtrInner<'a, T> {
106            self.ptr
107        }
108    }
109}
110
111#[allow(unreachable_pub)] // This is a false positive on our MSRV toolchain.
112pub use def::Ptr;
113
114/// External trait implementations on [`Ptr`].
115mod _external {
116    use super::*;
117
118    /// SAFETY: Shared pointers are safely `Copy`. `Ptr`'s other invariants
119    /// (besides aliasing) are unaffected by the number of references that exist
120    /// to `Ptr`'s referent. The notable cases are:
121    /// - Alignment is a property of the referent type (`T`) and the address,
122    ///   both of which are unchanged
123    /// - Let `S(T, V)` be the set of bit values permitted to appear in the
124    ///   referent of a `Ptr<T, I: Invariants<Validity = V>>`. Since this copy
125    ///   does not change `I::Validity` or `T`, `S(T, I::Validity)` is also
126    ///   unchanged.
127    ///
128    ///   We are required to guarantee that the referents of the original `Ptr`
129    ///   and of the copy (which, of course, are actually the same since they
130    ///   live in the same byte address range) both remain in the set `S(T,
131    ///   I::Validity)`. Since this invariant holds on the original `Ptr`, it
132    ///   cannot be violated by the original `Ptr`, and thus the original `Ptr`
133    ///   cannot be used to violate this invariant on the copy. The inverse
134    ///   holds as well.
135    impl<'a, T, I> Copy for Ptr<'a, T, I>
136    where
137        T: 'a + ?Sized,
138        I: Invariants<Aliasing = Shared>,
139    {
140    }
141
142    /// SAFETY: See the safety comment on `Copy`.
143    impl<'a, T, I> Clone for Ptr<'a, T, I>
144    where
145        T: 'a + ?Sized,
146        I: Invariants<Aliasing = Shared>,
147    {
148        #[inline]
149        fn clone(&self) -> Self {
150            *self
151        }
152    }
153
154    impl<'a, T, I> Debug for Ptr<'a, T, I>
155    where
156        T: 'a + ?Sized,
157        I: Invariants,
158    {
159        #[inline]
160        fn fmt(&self, f: &mut Formatter<'_>) -> core::fmt::Result {
161            self.as_inner().as_non_null().fmt(f)
162        }
163    }
164}
165
166/// Methods for converting to and from `Ptr` and Rust's safe reference types.
167mod _conversions {
168    use super::*;
169    use crate::pointer::cast::{CastExact, CastSized, IdCast};
170
171    /// `&'a T` → `Ptr<'a, T>`
172    impl<'a, T> Ptr<'a, T, (Shared, Aligned, Safe)>
173    where
174        T: 'a + ?Sized,
175    {
176        /// Constructs a `Ptr` from a shared reference.
177        #[inline(always)]
178        pub fn from_ref(ptr: &'a T) -> Self {
179            let inner = PtrInner::from_ref(ptr);
180            // SAFETY:
181            // 0. `ptr`, by invariant on `&'a T`, conforms to the aliasing
182            //    invariant of `Shared`.
183            // 1. `ptr`, by invariant on `&'a T`, conforms to the alignment
184            //    invariant of `Aligned`.
185            // 2. `ptr`'s referent, by invariant on `&'a T`, is a bit-valid `T`.
186            //    This satisfies the requirement that a `Ptr<T, (_, _, Safe)>`
187            //    point to a bit-valid `T`. Even if `T` permits interior
188            //    mutation, this invariant guarantees that the returned `Ptr` can
189            //    only ever be used to modify the referent to store bit-valid
190            //    `T`s, which ensures that the returned `Ptr` cannot be used to
191            //    violate the soundness of the original `ptr: &'a T` or of any
192            //    other references that may exist to the same referent.
193            unsafe { Self::from_inner(inner) }
194        }
195    }
196
197    /// `&'a mut T` → `Ptr<'a, T>`
198    impl<'a, T> Ptr<'a, T, (Exclusive, Aligned, Safe)>
199    where
200        T: 'a + ?Sized,
201    {
202        /// Constructs a `Ptr` from an exclusive reference.
203        #[inline(always)]
204        pub fn from_mut(ptr: &'a mut T) -> Self {
205            let inner = PtrInner::from_mut(ptr);
206            // SAFETY:
207            // 0. `ptr`, by invariant on `&'a mut T`, conforms to the aliasing
208            //    invariant of `Exclusive`.
209            // 1. `ptr`, by invariant on `&'a mut T`, conforms to the alignment
210            //    invariant of `Aligned`.
211            // 2. `ptr`'s referent, by invariant on `&'a mut T`, is a bit-valid
212            //    `T`. This satisfies the requirement that a `Ptr<T, (_, _,
213            //    Safe)>` point to a bit-valid `T`. This invariant guarantees
214            //    that the returned `Ptr` can only ever be used to modify the
215            //    referent to store bit-valid `T`s, which ensures that the
216            //    returned `Ptr` cannot be used to violate the soundness of the
217            //    original `ptr: &'a mut T`.
218            unsafe { Self::from_inner(inner) }
219        }
220    }
221
222    /// `Ptr<'a, T>` → `&'a T`
223    impl<'a, T, I> Ptr<'a, T, I>
224    where
225        T: 'a + ?Sized,
226        I: Invariants<Alignment = Aligned, Validity = Safe>,
227        I::Aliasing: Reference,
228    {
229        /// Converts `self` to a shared reference.
230        // This consumes `self`, not `&self`, because `self` is, logically, a
231        // pointer. For `I::Aliasing = invariant::Shared`, `Self: Copy`, and so
232        // this doesn't prevent the caller from still using the pointer after
233        // calling `as_ref`.
234        #[allow(clippy::wrong_self_convention)]
235        #[inline]
236        #[must_use]
237        pub fn as_ref(self) -> &'a T {
238            let raw = self.as_inner().as_non_null();
239            // SAFETY: `self` satisfies the `Aligned` invariant, so we know that
240            // `raw` is validly-aligned for `T`.
241            #[cfg(all(miri, __ZEROCOPY_INTERNAL_USE_ONLY_DEV_MODE))]
242            unsafe {
243                crate::util::miri_promise_symbolic_alignment(
244                    raw.as_ptr().cast(),
245                    core::mem::align_of_val_raw(raw.as_ptr()),
246                );
247            }
248            // SAFETY: This invocation of `NonNull::as_ref` satisfies its
249            // documented safety preconditions:
250            //
251            // 1. The pointer is properly aligned. This is ensured by-contract
252            //    on `Ptr`, because the `I::Alignment` is `Aligned`.
253            //
254            // 2. If the pointer's referent is not zero-sized, then the pointer
255            //    must be “dereferenceable” in the sense defined in the module
256            //    documentation; i.e.:
257            //
258            //    > The memory range of the given size starting at the pointer
259            //    > must all be within the bounds of a single allocated object.
260            //    > [2]
261            //
262            //   This is ensured by contract on all `PtrInner`s.
263            //
264            // 3. The pointer must point to a validly-initialized instance of
265            //    `T`. This is ensured by-contract on `Ptr`, because the
266            //    `I::Validity` is `Safe`.
267            //
268            // 4. You must enforce Rust’s aliasing rules. This is ensured by
269            //    contract on `Ptr`, because `I::Aliasing: Reference`. Either it
270            //    is `Shared` or `Exclusive`. If it is `Shared`, other
271            //    references may not mutate the referent outside of
272            //    `UnsafeCell`s.
273            //
274            // [1]: https://doc.rust-lang.org/std/ptr/struct.NonNull.html#method.as_ref
275            // [2]: https://doc.rust-lang.org/std/ptr/index.html#safety
276            unsafe { raw.as_ref() }
277        }
278    }
279
280    impl<'a, T, I> Ptr<'a, T, I>
281    where
282        T: 'a + ?Sized,
283        I: Invariants,
284        I::Aliasing: Reference,
285    {
286        /// Reborrows `self`, producing another `Ptr`.
287        ///
288        /// Since `self` is borrowed mutably, this prevents any methods from
289        /// being called on `self` as long as the returned `Ptr` exists.
290        #[inline]
291        #[must_use]
292        #[allow(clippy::needless_lifetimes)] // Allows us to name the lifetime in the safety comment below.
293        pub fn reborrow<'b>(&'b mut self) -> Ptr<'b, T, I>
294        where
295            'a: 'b,
296        {
297            // SAFETY: The following all hold by invariant on `self`, and thus
298            // hold of `ptr = self.as_inner()`:
299            // 0. SEE BELOW.
300            // 1. `ptr` conforms to the alignment invariant of
301            //    [`I::Alignment`](invariant::Alignment).
302            // 2. `ptr` conforms to the validity invariant of
303            //    [`I::Validity`](invariant::Validity). `self` and the returned
304            //    `Ptr` permit the same bit values in their referents since they
305            //    have the same referent type (`T`) and the same validity
306            //    (`I::Validity`). Thus, regardless of what mutation is
307            //    permitted (`Exclusive` aliasing or `Shared`-aliased interior
308            //    mutation), neither can be used to write a value to the
309            //    referent which violates the other's validity invariant.
310            //
311            // For aliasing (0 above), since `I::Aliasing: Reference`,
312            // there are two cases for `I::Aliasing`:
313            // - For `invariant::Shared`: `'a` outlives `'b`, and so the
314            //   returned `Ptr` does not permit accessing the referent any
315            //   longer than is possible via `self`. For shared aliasing, it is
316            //   sound for multiple `Ptr`s to exist simultaneously which
317            //   reference the same memory, so creating a new one is not
318            //   problematic.
319            // - For `invariant::Exclusive`: Since `self` is `&'b mut` and we
320            //   return a `Ptr` with lifetime `'b`, `self` is inaccessible to
321            //   the caller for the lifetime `'b` - in other words, `self` is
322            //   inaccessible to the caller as long as the returned `Ptr`
323            //   exists. Since `self` is an exclusive `Ptr`, no other live
324            //   references or `Ptr`s may exist which refer to the same memory
325            //   while `self` is live. Thus, as long as the returned `Ptr`
326            //   exists, no other references or `Ptr`s which refer to the same
327            //   memory may be live.
328            unsafe { Ptr::from_inner(self.as_inner()) }
329        }
330
331        /// Reborrows `self` as shared, producing another `Ptr` with `Shared`
332        /// aliasing.
333        ///
334        /// Since `self` is borrowed mutably, this prevents any methods from
335        /// being called on `self` as long as the returned `Ptr` exists.
336        #[inline]
337        #[must_use]
338        #[allow(clippy::needless_lifetimes)] // Allows us to name the lifetime in the safety comment below.
339        pub fn reborrow_shared<'b>(&'b mut self) -> Ptr<'b, T, (Shared, I::Alignment, I::Validity)>
340        where
341            'a: 'b,
342        {
343            // SAFETY: The following all hold by invariant on `self`, and thus
344            // hold of `ptr = self.as_inner()`:
345            // 0. SEE BELOW.
346            // 1. `ptr` conforms to the alignment invariant of
347            //    [`I::Alignment`](invariant::Alignment).
348            // 2. `ptr` conforms to the validity invariant of
349            //    [`I::Validity`](invariant::Validity). `self` and the returned
350            //    `Ptr` permit the same bit values in their referents since they
351            //    have the same referent type (`T`) and the same validity
352            //    (`I::Validity`). Thus, regardless of what mutation is
353            //    permitted (`Exclusive` aliasing or `Shared`-aliased interior
354            //    mutation), neither can be used to write a value to the
355            //    referent which violates the other's validity invariant.
356            //
357            // For aliasing (0 above), since `I::Aliasing: Reference`,
358            // there are two cases for `I::Aliasing`:
359            // - For `invariant::Shared`: `'a` outlives `'b`, and so the
360            //   returned `Ptr` does not permit accessing the referent any
361            //   longer than is possible via `self`. For shared aliasing, it is
362            //   sound for multiple `Ptr`s to exist simultaneously which
363            //   reference the same memory, so creating a new one is not
364            //   problematic.
365            // - For `invariant::Exclusive`: Since `self` is `&'b mut` and we
366            //   return a `Ptr` with lifetime `'b`, `self` is inaccessible to
367            //   the caller for the lifetime `'b` - in other words, `self` is
368            //   inaccessible to the caller as long as the returned `Ptr`
369            //   exists. Since `self` is an exclusive `Ptr`, no other live
370            //   references or `Ptr`s may exist which refer to the same memory
371            //   while `self` is live. Thus, as long as the returned `Ptr`
372            //   exists, no other references or `Ptr`s which refer to the same
373            //   memory may be live.
374            unsafe { Ptr::from_inner(self.as_inner()) }
375        }
376    }
377
378    /// `Ptr<'a, T>` → `&'a mut T`
379    impl<'a, T> Ptr<'a, T, (Exclusive, Aligned, Safe)>
380    where
381        T: 'a + ?Sized,
382    {
383        /// Converts `self` to a mutable reference.
384        #[allow(clippy::wrong_self_convention)]
385        #[inline]
386        #[must_use]
387        pub fn as_mut(self) -> &'a mut T {
388            let mut raw = self.as_inner().as_non_null();
389            // SAFETY: `self` satisfies the `Aligned` invariant, so we know that
390            // `raw` is validly-aligned for `T`.
391            #[cfg(all(miri, __ZEROCOPY_INTERNAL_USE_ONLY_DEV_MODE))]
392            unsafe {
393                crate::util::miri_promise_symbolic_alignment(
394                    raw.as_ptr().cast(),
395                    core::mem::align_of_val_raw(raw.as_ptr()),
396                );
397            }
398            // SAFETY: This invocation of `NonNull::as_mut` satisfies its
399            // documented safety preconditions:
400            //
401            // 1. The pointer is properly aligned. This is ensured by-contract
402            //    on `Ptr`, because the `ALIGNMENT_INVARIANT` is `Aligned`.
403            //
404            // 2. If the pointer's referent is not zero-sized, then the pointer
405            //    must be “dereferenceable” in the sense defined in the module
406            //    documentation; i.e.:
407            //
408            //    > The memory range of the given size starting at the pointer
409            //    > must all be within the bounds of a single allocated object.
410            //    > [2]
411            //
412            //   This is ensured by contract on all `PtrInner`s.
413            //
414            // 3. The pointer must point to a validly-initialized instance of
415            //    `T`. This is ensured by-contract on `Ptr`, because the validity
416            //    invariant is `Safe`.
417            //
418            // 4. You must enforce Rust’s aliasing rules. This is ensured by
419            //    contract on `Ptr`, because the `ALIASING_INVARIANT` is
420            //    `Exclusive`.
421            //
422            // [1]: https://doc.rust-lang.org/std/ptr/struct.NonNull.html#method.as_mut
423            // [2]: https://doc.rust-lang.org/std/ptr/index.html#safety
424            unsafe { raw.as_mut() }
425        }
426    }
427
428    /// `Ptr<'a, T>` → `Ptr<'a, U>`
429    impl<'a, T: ?Sized, I> Ptr<'a, T, I>
430    where
431        I: Invariants,
432    {
433        /// Reinterprets the same byte region as `U` with validity `V`.
434        ///
435        /// This preserves the aliasing invariant, uses [`SizeEq`] to select a
436        /// [`CastExact`] implementation, and conservatively forgets alignment.
437        #[must_use]
438        #[inline(always)]
439        pub fn transmute<U, V, R>(self) -> Ptr<'a, U, (I::Aliasing, Unaligned, V)>
440        where
441            V: Validity,
442            U: TransmuteFromPtr<T, I::Aliasing, I::Validity, V, <U as SizeEq<T>>::CastFrom, R>
443                + SizeEq<T>
444                + ?Sized,
445        {
446            self.transmute_with::<U, V, <U as SizeEq<T>>::CastFrom, R>()
447        }
448
449        /// Reinterprets the same byte region as `U` with validity `V` using
450        /// `C`.
451        ///
452        /// `C: CastExact` preserves the byte region. The aliasing invariant is
453        /// preserved, while alignment is conservatively forgotten because the
454        /// destination type may have a different alignment requirement.
455        #[inline]
456        #[must_use]
457        pub fn transmute_with<U, V, C, R>(self) -> Ptr<'a, U, (I::Aliasing, Unaligned, V)>
458        where
459            V: Validity,
460            U: TransmuteFromPtr<T, I::Aliasing, I::Validity, V, C, R> + ?Sized,
461            C: CastExact<T, U>,
462        {
463            // SAFETY:
464            // - By `C: CastExact`, `C` preserves referent address, and so we
465            //   don't need to consider projections in the following safety
466            //   arguments.
467            // - If aliasing is `Shared`, then by `U: TransmuteFromPtr<T>`, at
468            //   least one of the following holds:
469            //   - `T: Immutable` and `U: Immutable`, in which case it is
470            //     trivially sound for shared code to operate on a `&T` and `&U`
471            //     at the same time, as neither can perform interior mutation
472            //   - It is directly guaranteed that it is sound for shared code to
473            //     operate on these references simultaneously
474            // - By `U: TransmuteFromPtr<T, I::Aliasing, I::Validity, V, C>`, it
475            //   is sound to perform this transmute using `C`.
476            unsafe { self.project_transmute_unchecked::<_, _, C>() }
477        }
478
479        /// Changes only the validity invariant to `V`.
480        ///
481        /// The referent type and byte region are unchanged, so this preserves
482        /// the existing aliasing and alignment invariants.
483        #[inline]
484        #[must_use]
485        pub fn recall_validity<V, R>(self) -> Ptr<'a, T, (I::Aliasing, I::Alignment, V)>
486        where
487            V: Validity,
488            T: TransmuteFromPtr<T, I::Aliasing, I::Validity, V, IdCast, R>,
489        {
490            let ptr = self.transmute_with::<T, V, IdCast, R>();
491            // SAFETY: `self` and `ptr` have the same address and referent type.
492            // Therefore, if `self` satisfies `I::Alignment`, then so does
493            // `ptr`.
494            unsafe { ptr.assume_alignment::<I::Alignment>() }
495        }
496
497        /// Projects and/or transmutes to a different (unsized) referent type
498        /// without checking interior mutability.
499        ///
500        /// Callers should prefer [`cast`] or [`project`] where possible.
501        ///
502        /// [`cast`]: Ptr::cast
503        /// [`project`]: Ptr::project
504        ///
505        /// # Safety
506        ///
507        /// The caller promises that:
508        /// - If `I::Aliasing` is [`Shared`], it must not be possible for safe
509        ///   code, operating on a `&T` and `&U`, with the referents of `self`
510        ///   and `self.project_transmute_unchecked()`, respectively, to cause
511        ///   undefined behavior.
512        /// - It is sound to project and/or transmute a pointer of type `T` with
513        ///   aliasing `I::Aliasing` and validity `I::Validity` to a pointer of
514        ///   type `U` with aliasing `I::Aliasing` and validity `V`. This is a
515        ///   subtle soundness requirement that is a function of `T`, `U`,
516        ///   `I::Aliasing`, `I::Validity`, and `V`, and may depend upon the
517        ///   presence, absence, or specific location of `UnsafeCell`s in `T`
518        ///   and/or `U`, and on whether interior mutation is ever permitted via
519        ///   those `UnsafeCell`s. See [`Validity`] for more details.
520        #[inline]
521        #[must_use]
522        pub unsafe fn project_transmute_unchecked<U: ?Sized, V, P>(
523            self,
524        ) -> Ptr<'a, U, (I::Aliasing, Unaligned, V)>
525        where
526            V: Validity,
527            P: crate::pointer::cast::Project<T, U>,
528        {
529            let ptr = self.as_inner().project::<_, P>();
530
531            // SAFETY:
532            //
533            // The following safety arguments rely on the fact that `P: Project`
534            // guarantees that `P` is a referent-preserving or -shrinking
535            // projection. Thus, `ptr` addresses a subset of the bytes of
536            // `*self`, and so certain properties that hold of `*self` also hold
537            // of `*ptr`.
538            //
539            // 0. `ptr` conforms to the aliasing invariant of `I::Aliasing`:
540            //    - `Exclusive`: `self` is the only `Ptr` or reference which is
541            //      permitted to read or modify the referent for the lifetime
542            //      `'a`. Since we consume `self` by value, the returned pointer
543            //      remains the only `Ptr` or reference which is permitted to
544            //      read or modify the referent for the lifetime `'a`.
545            //    - `Shared`: Since `self` has aliasing `Shared`, we know that
546            //      no other code may mutate the referent during the lifetime
547            //      `'a`, except via `UnsafeCell`s, and except as permitted by
548            //      `T`'s library safety invariants. The caller promises that
549            //      any safe operations which can be permitted on a `&T` and a
550            //      `&U` simultaneously must be sound. Thus, no operations on a
551            //      `&U` could violate `&T`'s library safety invariants, and
552            //      vice-versa. Since any mutation via shared references outside
553            //      of `UnsafeCell`s is unsound, this must be impossible using
554            //      `&T` and `&U`.
555            //    - `Inaccessible`: There are no restrictions we need to uphold.
556            // 1. `ptr` trivially satisfies the alignment invariant `Unaligned`.
557            // 2. The caller promises that the returned pointer satisfies the
558            //    validity invariant `V` with respect to its referent type, `U`.
559            unsafe { Ptr::from_inner(ptr) }
560        }
561    }
562
563    /// `Ptr<'a, T, (_, _, _)>` → `Ptr<'a, Unalign<T>, (_, Aligned, _)>`
564    impl<'a, T, I> Ptr<'a, T, I>
565    where
566        I: Invariants,
567    {
568        /// Converts a `Ptr` an unaligned `T` into a `Ptr` to an aligned
569        /// `Unalign<T>`.
570        #[inline]
571        #[must_use]
572        pub fn into_unalign(
573            self,
574        ) -> Ptr<'a, crate::Unalign<T>, (I::Aliasing, Aligned, I::Validity)> {
575            // FIXME(#1359): This should be a `transmute_with` call.
576            // Unfortunately, to avoid blanket impl conflicts, we only implement
577            // `TransmuteFrom<T>` for `Unalign<T>` (and vice versa) specifically
578            // for `Safe` validity, not for all validity types.
579
580            // SAFETY:
581            // - By `CastSized: Cast`, `CastSized` preserves referent address,
582            //   and so we don't need to consider projections in the following
583            //   safety arguments.
584            // - Since `Unalign<T>` has the same layout as `T`, the returned
585            //   pointer refers to `UnsafeCell`s at the same locations as
586            //   `self`.
587            // - `Unalign<T>` promises to have the same bit validity as `T`. By
588            //   invariant on `Validity`, the set of bit patterns allowed in the
589            //   referent of a `Ptr<X, (_, _, V)>` is only a function of the
590            //   validity of `X` and of `V`. Thus, the set of bit patterns
591            //   allowed in the referent of a `Ptr<T, (_, _, I::Validity)>` is
592            //   the same as the set of bit patterns allowed in the referent of
593            //   a `Ptr<Unalign<T>, (_, _, I::Validity)>`. As a result, `self`
594            //   and the returned `Ptr` permit the same set of bit patterns in
595            //   their referents, and so neither can be used to violate the
596            //   validity of the other.
597            let ptr = unsafe { self.project_transmute_unchecked::<_, _, CastSized>() };
598            ptr.bikeshed_recall_aligned()
599        }
600    }
601
602    impl<'a, T, I> Ptr<'a, T, I>
603    where
604        T: ?Sized,
605        I: Invariants<Validity = Safe>,
606        I::Aliasing: Reference,
607    {
608        /// Reads the referent.
609        #[must_use]
610        #[inline(always)]
611        pub fn read<R>(self) -> T
612        where
613            T: Copy,
614            T: Read<I::Aliasing, R>,
615        {
616            <I::Alignment as Alignment>::read(self)
617        }
618
619        /// Views the value as an aligned reference.
620        ///
621        /// This is only available if `T` is [`Unaligned`].
622        #[must_use]
623        #[inline]
624        pub fn unaligned_as_ref(self) -> &'a T
625        where
626            T: crate::Unaligned,
627        {
628            self.bikeshed_recall_aligned().as_ref()
629        }
630    }
631}
632
633/// State transitions between invariants.
634mod _transitions {
635    use super::*;
636    use crate::{
637        pointer::{cast::IdCast, transmute::TryTransmuteFromPtr},
638        ReadOnly,
639    };
640
641    impl<'a, T, I> Ptr<'a, T, I>
642    where
643        T: 'a + ?Sized,
644        I: Invariants,
645    {
646        /// Assumes that `self` satisfies the invariants `H`.
647        ///
648        /// # Safety
649        ///
650        /// The caller promises that `self` satisfies the invariants `H`.
651        unsafe fn assume_invariants<H: Invariants>(self) -> Ptr<'a, T, H> {
652            // SAFETY: The caller has promised to satisfy all parameterized
653            // invariants of `Ptr`. `Ptr`'s other invariants are satisfied
654            // by-contract by the source `Ptr`.
655            unsafe { Ptr::from_inner(self.as_inner()) }
656        }
657
658        /// Helps the type system unify two distinct invariant types which are
659        /// actually the same.
660        #[inline]
661        #[must_use]
662        pub fn unify_invariants<
663            H: Invariants<Aliasing = I::Aliasing, Alignment = I::Alignment, Validity = I::Validity>,
664        >(
665            self,
666        ) -> Ptr<'a, T, H> {
667            // SAFETY: The associated type bounds on `H` ensure that the
668            // invariants are unchanged.
669            unsafe { self.assume_invariants::<H>() }
670        }
671
672        /// Downgrades `self` to shared aliasing.
673        #[inline]
674        #[must_use]
675        pub(super) fn into_shared(self) -> Ptr<'a, T, (Shared, I::Alignment, I::Validity)>
676        where
677            I::Aliasing: Reference,
678        {
679            // SAFETY: `I::Aliasing: Reference` guarantees that the source
680            // aliasing is either `Shared` or `Exclusive`. `Shared` requires no
681            // transition. If it is `Exclusive`, consuming `self` ends the only
682            // permitted access to the referent, so the returned pointer can
683            // soundly carry `Shared` aliasing. The referent type, alignment,
684            // and validity invariants are unchanged.
685            unsafe { self.assume_invariants() }
686        }
687
688        /// Assumes that `self`'s referent is validly-aligned for `T` if
689        /// required by `A`.
690        ///
691        /// # Safety
692        ///
693        /// The caller promises that `self`'s referent conforms to the alignment
694        /// invariant of `T` if required by `A`.
695        #[inline]
696        pub(crate) unsafe fn assume_alignment<A: Alignment>(
697            self,
698        ) -> Ptr<'a, T, (I::Aliasing, A, I::Validity)> {
699            // SAFETY: The caller promises that `self`'s referent is
700            // well-aligned for `T` if required by `A` .
701            unsafe { self.assume_invariants() }
702        }
703
704        /// Checks the `self`'s alignment at runtime, returning an aligned `Ptr`
705        /// on success.
706        #[inline]
707        pub fn try_into_aligned(
708            self,
709        ) -> Result<Ptr<'a, T, (I::Aliasing, Aligned, I::Validity)>, AlignmentError<Self, T>>
710        where
711            T: Sized,
712        {
713            if let Err(err) =
714                crate::util::validate_aligned_to::<_, T>(self.as_inner().as_non_null())
715            {
716                return Err(err.with_src(self));
717            }
718
719            // SAFETY: We just checked the alignment.
720            Ok(unsafe { self.assume_alignment::<Aligned>() })
721        }
722
723        /// Recalls that `self`'s referent is validly-aligned for `T`.
724        #[inline]
725        // FIXME(#859): Reconsider the name of this method before making it
726        // public.
727        #[must_use]
728        pub fn bikeshed_recall_aligned(self) -> Ptr<'a, T, (I::Aliasing, Aligned, I::Validity)>
729        where
730            T: crate::Unaligned,
731        {
732            // SAFETY: The bound `T: Unaligned` ensures that `T` has no
733            // non-trivial alignment requirement.
734            unsafe { self.assume_alignment::<Aligned>() }
735        }
736
737        /// Assumes that `self`'s referent conforms to the validity requirement
738        /// of `V`.
739        ///
740        /// # Safety
741        ///
742        /// The caller promises that `self`'s referent conforms to the validity
743        /// requirement of `V`.
744        #[must_use]
745        #[inline]
746        pub unsafe fn assume_validity<V: Validity>(
747            self,
748        ) -> Ptr<'a, T, (I::Aliasing, I::Alignment, V)> {
749            // SAFETY: The caller promises that `self`'s referent conforms to
750            // the validity requirement of `V`.
751            unsafe { self.assume_invariants() }
752        }
753
754        /// A shorthand for `self.assume_validity<invariant::Initialized>()`.
755        ///
756        /// # Safety
757        ///
758        /// The caller promises to uphold the safety preconditions of
759        /// `self.assume_validity<invariant::Initialized>()`.
760        #[must_use]
761        #[inline]
762        pub unsafe fn assume_initialized(
763            self,
764        ) -> Ptr<'a, T, (I::Aliasing, I::Alignment, Initialized)> {
765            // SAFETY: The caller has promised to uphold the safety
766            // preconditions.
767            unsafe { self.assume_validity::<Initialized>() }
768        }
769
770        /// A shorthand for `self.assume_validity<Safe>()`.
771        ///
772        /// # Safety
773        ///
774        /// The caller promises to uphold the safety preconditions of
775        /// `self.assume_validity<Safe>()`.
776        #[must_use]
777        #[inline]
778        pub unsafe fn assume_safe(self) -> Ptr<'a, T, (I::Aliasing, I::Alignment, Safe)> {
779            // SAFETY: The caller has promised to uphold the safety
780            // preconditions.
781            unsafe { self.assume_validity::<Safe>() }
782        }
783
784        /// Checks that `self`'s referent is validly initialized for `T`,
785        /// returning a `Ptr` with `Safe` on success.
786        ///
787        /// # Panics
788        ///
789        /// This method will panic if
790        /// [`T::is_safe`][TryFromBytes::is_safe] panics.
791        ///
792        /// # Safety
793        ///
794        /// On error, unsafe code may rely on this method's returned
795        /// `ValidityError` containing `self`.
796        #[inline]
797        pub fn try_into_safe<R, S>(
798            mut self,
799        ) -> Result<Ptr<'a, T, (I::Aliasing, I::Alignment, Safe)>, ValidityError<Self, T>>
800        where
801            T: TryFromBytes
802                + Read<I::Aliasing, R>
803                + TryTransmuteFromPtr<T, I::Aliasing, I::Validity, Safe, IdCast, S>,
804            ReadOnly<T>: Read<I::Aliasing, R>,
805            I::Aliasing: Reference,
806            I: Invariants<Validity = Initialized>,
807        {
808            // This call may panic. If that happens, it doesn't cause any
809            // soundness issues, as we have not generated any invalid state
810            // which we need to fix before returning.
811            if T::is_safe(self.reborrow().transmute::<_, _, _>().reborrow_shared()) {
812                // SAFETY: If `T::is_safe` returns true, code may assume that
813                // `self` contains a valid `T`, so its referent conforms to
814                // `Safe` for `T`. By `T: TryTransmuteFromPtr<T, I::Aliasing,
815                // I::Validity, Safe>`, given that condition, changing `self`'s
816                // validity to `Safe` is sound.
817                Ok(unsafe { self.assume_safe() })
818            } else {
819                Err(ValidityError::new(self))
820            }
821        }
822
823        /// Forgets that `self`'s referent is validly-aligned for `T`.
824        #[inline]
825        #[must_use]
826        pub fn forget_aligned(self) -> Ptr<'a, T, (I::Aliasing, Unaligned, I::Validity)> {
827            // SAFETY: `Unaligned` is less restrictive than `Aligned`.
828            unsafe { self.assume_invariants() }
829        }
830    }
831}
832
833/// Casts of the referent type.
834#[allow(unreachable_pub)] // False positive on MSRV
835pub use _casts::TryWithError;
836mod _casts {
837    use core::cell::UnsafeCell;
838
839    use super::*;
840    use crate::{
841        pointer::cast::{AsBytesCast, Cast},
842        HasTag, ProjectField,
843    };
844
845    impl<'a, T, I> Ptr<'a, T, I>
846    where
847        T: 'a + ?Sized,
848        I: Invariants,
849    {
850        /// Casts to a different referent type without checking interior
851        /// mutability.
852        ///
853        /// Callers should prefer [`cast`][Ptr::cast] where possible.
854        ///
855        /// # Safety
856        ///
857        /// If `I::Aliasing` is [`Shared`], it must not be possible for safe
858        /// code, operating on a `&T` and `&U` with the same referent
859        /// simultaneously, to cause undefined behavior.
860        #[inline]
861        #[must_use]
862        pub unsafe fn cast_unchecked<U, C: Cast<T, U>>(
863            self,
864        ) -> Ptr<'a, U, (I::Aliasing, Unaligned, I::Validity)>
865        where
866            U: 'a + CastableFrom<T, I::Validity, I::Validity> + ?Sized,
867        {
868            // SAFETY:
869            // - By `C: Cast`, `C` preserves the address of the referent.
870            // - If `I::Aliasing` is [`Shared`], the caller promises that it
871            //   is not possible for safe code, operating on a `&T` and `&U`
872            //   with the same referent simultaneously, to cause undefined
873            //   behavior.
874            // - By `U: CastableFrom<T, I::Validity, I::Validity>`,
875            //   `I::Validity` is either `Uninit` or `Initialized`. In both
876            //   cases, the bit validity `I::Validity` has the same semantics
877            //   regardless of referent type. In other words, the set of allowed
878            //   referent values for `Ptr<T, (_, _, I::Validity)>` and `Ptr<U,
879            //   (_, _, I::Validity)>` are identical. As a consequence, neither
880            //   `self` nor the returned `Ptr` can be used to write values which
881            //   are invalid for the other.
882            unsafe { self.project_transmute_unchecked::<_, _, C>() }
883        }
884
885        /// Casts to a different referent type.
886        #[inline]
887        #[must_use]
888        pub fn cast<U, C, R>(self) -> Ptr<'a, U, (I::Aliasing, Unaligned, I::Validity)>
889        where
890            T: MutationCompatible<U, I::Aliasing, I::Validity, I::Validity, R>,
891            U: 'a + ?Sized + CastableFrom<T, I::Validity, I::Validity>,
892            C: Cast<T, U>,
893        {
894            // SAFETY: Because `T: MutationCompatible<U, I::Aliasing, R>`, one
895            // of the following holds:
896            // - `T: Read<I::Aliasing>` and `U: Read<I::Aliasing>`, in which
897            //   case one of the following holds:
898            //   - `I::Aliasing` is `Exclusive`
899            //   - `T` and `U` are both `Immutable`
900            // - It is sound for safe code to operate on `&T` and `&U` with the
901            //   same referent simultaneously.
902            unsafe { self.cast_unchecked::<_, C>() }
903        }
904
905        #[inline(always)]
906        pub fn project<Client, F, const VARIANT_ID: i128, const FIELD_ID: i128>(
907            mut self,
908        ) -> Result<Ptr<'a, T::Type, T::Invariants>, T::Error>
909        where
910            T: ProjectField<Client, F, I, VARIANT_ID, FIELD_ID>,
911            I::Aliasing: Reference,
912        {
913            use crate::pointer::cast::Projection;
914            match <T as ProjectField<Client, F, I, VARIANT_ID, FIELD_ID>>::is_projectable(
915                self.reborrow().project_tag::<Client>(),
916            ) {
917                Ok(()) => {
918                    let inner = self.as_inner();
919                    let projected =
920                        inner.project::<_, Projection<Client, F, VARIANT_ID, FIELD_ID>>();
921                    // SAFETY: By `T: ProjectField<Client, F, I, VARIANT_ID, FIELD_ID>`,
922                    // for `self: Ptr<'_, T, I>` such that `T::is_projectable`
923                    // (which we've verified in this match arm),
924                    // `T::project(self.as_inner())` conforms to
925                    // `T::Invariants`. The `projected` pointer satisfies these
926                    // invariants because it is produced by way of an
927                    // abstraction that is equivalent to
928                    // `T::project(ptr.as_inner())`: by invariant on
929                    // `PtrInner::project`, `projected` is guaranteed to address
930                    // the subset of the bytes of `inner`'s referent addressed
931                    // by `Projection::project(inner)`, and by invariant on
932                    // `Projection`, `Projection::project` is implemented by
933                    // delegating to an implementation of `HasField::project`.
934                    Ok(unsafe { Ptr::from_inner(projected) })
935                }
936                Err(err) => Err(err),
937            }
938        }
939
940        #[must_use]
941        #[inline(always)]
942        pub fn project_tag<Client>(
943            self,
944        ) -> Ptr<'a, <T as HasTag<Client>>::Tag, (Shared, I::Alignment, I::Validity)>
945        where
946            T: HasTag<Client>,
947            I::Aliasing: Reference,
948        {
949            let ptr = self.into_shared();
950            // SAFETY: By invariant on `ProjectToTag`, it is sound to project
951            // `ptr` to a shared tag pointer with the same validity invariant.
952            let ptr = unsafe {
953                ptr.project_transmute_unchecked::<_, _, <T as HasTag<Client>>::ProjectToTag>()
954            };
955            // SAFETY: By invariant on `ProjectToTag`, the projected pointer has
956            // the same alignment as `self`.
957            unsafe { ptr.assume_alignment() }
958        }
959
960        /// Attempts to transform the pointer, restoring the original on
961        /// failure.
962        ///
963        /// # Safety
964        ///
965        /// If `I::Aliasing != Shared`, then if `f` returns `Err(err)`, no copy
966        /// of `f`'s argument must exist outside of `err`.
967        #[inline(always)]
968        pub(crate) unsafe fn try_with_unchecked<U, J, E, F>(
969            self,
970            f: F,
971        ) -> Result<Ptr<'a, U, J>, E::Mapped>
972        where
973            U: 'a + ?Sized,
974            J: Invariants<Aliasing = I::Aliasing>,
975            E: TryWithError<Self>,
976            F: FnOnce(Ptr<'a, T, I>) -> Result<Ptr<'a, U, J>, E>,
977        {
978            let old_inner = self.as_inner();
979            #[rustfmt::skip]
980            let res = f(self).map_err(#[inline(always)] move |err: E| {
981                err.map(#[inline(always)] |src| {
982                    drop(src);
983
984                    // SAFETY:
985                    // 0. Aliasing is either `Shared` or `Exclusive`:
986                    //    - If aliasing is `Shared`, then it cannot violate
987                    //      aliasing make another copy of this pointer (in fact,
988                    //      using `I::Aliasing = Shared`, we could have just
989                    //      cloned `self`).
990                    //    - If aliasing is `Exclusive`, then `f` is not allowed
991                    //      to make another copy of `self`. In `map_err`, we are
992                    //      consuming the only value in the returned `Result`.
993                    //      By invariant on `E: TryWithError<Self>`, that `err:
994                    //      E` only contains a single `Self` and no other
995                    //      non-ZST fields which could be `Ptr`s or references
996                    //      to `self`'s referent. By the same invariant, `map`
997                    //      consumes this single `Self` and passes it to this
998                    //      closure. Since `self` was, by invariant on
999                    //      `Exclusive`, the only `Ptr` or reference live for
1000                    //      `'a` with this referent, and since we `drop(src)`
1001                    //      above, there are no copies left, and so we are
1002                    //      creating the only copy.
1003                    // 1. `self` conforms to `I::Aliasing` by invariant on
1004                    //    `Ptr`, and `old_inner` has the same address, so it
1005                    //    does too.
1006                    // 2. `f` could not have violated `self`'s validity without
1007                    //    itself being unsound. Assuming that `f` is sound, the
1008                    //    referent of `self` is still valid for `T`.
1009                    unsafe { Ptr::from_inner(old_inner) }
1010                })
1011            });
1012            res
1013        }
1014
1015        /// Attempts to transform the pointer, restoring the original on
1016        /// failure.
1017        #[inline(always)]
1018        pub fn try_with<U, J, E, F>(self, f: F) -> Result<Ptr<'a, U, J>, E::Mapped>
1019        where
1020            U: 'a + ?Sized,
1021            J: Invariants<Aliasing = I::Aliasing>,
1022            E: TryWithError<Self>,
1023            F: FnOnce(Ptr<'a, T, I>) -> Result<Ptr<'a, U, J>, E>,
1024            I: Invariants<Aliasing = Shared>,
1025        {
1026            // SAFETY: `I::Aliasing = Shared`, so the safety condition does not
1027            // apply.
1028            unsafe { self.try_with_unchecked(f) }
1029        }
1030    }
1031
1032    /// # Safety
1033    ///
1034    /// `Self` only contains a single `Self::Inner`, and `Self::Mapped` only
1035    /// contains a single `MappedInner`. Other than that, `Self` and
1036    /// `Self::Mapped` contain no non-ZST fields.
1037    ///
1038    /// `map` must pass ownership of `self`'s sole `Self::Inner` to `f`.
1039    pub unsafe trait TryWithError<MappedInner> {
1040        type Inner;
1041        type Mapped;
1042        fn map<F: FnOnce(Self::Inner) -> MappedInner>(self, f: F) -> Self::Mapped;
1043    }
1044
1045    impl<'a, T, I> Ptr<'a, T, I>
1046    where
1047        T: 'a + KnownLayout + ?Sized,
1048        I: Invariants,
1049    {
1050        /// Casts this pointer-to-initialized into a pointer-to-bytes.
1051        #[allow(clippy::wrong_self_convention)]
1052        #[must_use]
1053        #[inline]
1054        pub fn as_bytes<R>(self) -> Ptr<'a, [u8], (I::Aliasing, Aligned, Safe)>
1055        where
1056            [u8]: TransmuteFromPtr<T, I::Aliasing, I::Validity, Safe, AsBytesCast, R>,
1057        {
1058            self.transmute_with::<[u8], Safe, AsBytesCast, _>().bikeshed_recall_aligned()
1059        }
1060    }
1061
1062    impl<'a, T, I, const N: usize> Ptr<'a, [T; N], I>
1063    where
1064        T: 'a,
1065        I: Invariants,
1066    {
1067        /// Casts this pointer-to-array into a slice.
1068        #[allow(clippy::wrong_self_convention)]
1069        #[inline]
1070        #[must_use]
1071        pub fn as_slice(self) -> Ptr<'a, [T], I> {
1072            let slice = self.as_inner().as_slice();
1073            // SAFETY: Note that, by post-condition on `PtrInner::as_slice`,
1074            // `slice` refers to the same byte range as `self.as_inner()`.
1075            //
1076            // 0. Thus, `slice` conforms to the aliasing invariant of
1077            //    `I::Aliasing` because `self` does.
1078            // 1. By the above lemma, `slice` conforms to the alignment
1079            //    invariant of `I::Alignment` because `self` does.
1080            // 2. Since `[T; N]` and `[T]` have the same bit validity [1][2],
1081            //    and since `self` and the returned `Ptr` have the same validity
1082            //    invariant, neither `self` nor the returned `Ptr` can be used
1083            //    to write a value to the referent which violates the other's
1084            //    validity invariant.
1085            //
1086            // [1] Per https://doc.rust-lang.org/1.81.0/reference/type-layout.html#array-layout:
1087            //
1088            //   An array of `[T; N]` has a size of `size_of::<T>() * N` and the
1089            //   same alignment of `T`. Arrays are laid out so that the
1090            //   zero-based `nth` element of the array is offset from the start
1091            //   of the array by `n * size_of::<T>()` bytes.
1092            //
1093            //   ...
1094            //
1095            //   Slices have the same layout as the section of the array they
1096            //   slice.
1097            //
1098            // [2] Per https://doc.rust-lang.org/1.81.0/reference/types/array.html#array-types:
1099            //
1100            //   All elements of arrays are always initialized
1101            unsafe { Ptr::from_inner(slice) }
1102        }
1103    }
1104
1105    /// For caller convenience, these methods are generic over alignment
1106    /// invariant. In practice, the referent is always well-aligned, because the
1107    /// alignment of `[u8]` is 1.
1108    impl<'a, I> Ptr<'a, [u8], I>
1109    where
1110        I: Invariants<Validity = Safe>,
1111    {
1112        /// Attempts to cast `self` to a `U` using the given cast type.
1113        ///
1114        /// If `U` is a slice DST and pointer metadata (`meta`) is provided,
1115        /// then the cast will only succeed if it would produce an object with
1116        /// the given metadata.
1117        ///
1118        /// Returns `None` if the resulting `U` would be invalidly-aligned, if
1119        /// no `U` can fit in `self`, or if the provided pointer metadata
1120        /// describes an invalid instance of `U`. On success, returns a pointer
1121        /// to the largest-possible `U` which fits in `self`.
1122        ///
1123        /// # Safety
1124        ///
1125        /// The caller may assume that this implementation is correct, and may
1126        /// rely on that assumption for the soundness of their code. In
1127        /// particular, the caller may assume that, if `try_cast_into` returns
1128        /// `Some((ptr, remainder))`, then `ptr` and `remainder` refer to
1129        /// non-overlapping byte ranges within `self`, and that `ptr` and
1130        /// `remainder` entirely cover `self`. Finally:
1131        /// - If this is a prefix cast, `ptr` has the same address as `self`.
1132        /// - If this is a suffix cast, `remainder` has the same address as
1133        ///   `self`.
1134        #[inline(always)]
1135        pub fn try_cast_into<U, R>(
1136            self,
1137            cast_type: CastType,
1138            meta: Option<U::PointerMetadata>,
1139        ) -> Result<
1140            (Ptr<'a, U, (I::Aliasing, Aligned, Initialized)>, Ptr<'a, [u8], I>),
1141            CastError<Self, U>,
1142        >
1143        where
1144            I::Aliasing: Reference,
1145            U: 'a + ?Sized + KnownLayout + Read<I::Aliasing, R>,
1146        {
1147            let (inner, remainder) = self.as_inner().try_cast_into(cast_type, meta).map_err(
1148                #[inline(always)]
1149                |err| {
1150                    err.map_src(
1151                        #[inline(always)]
1152                        |inner|
1153                    // SAFETY: `PtrInner::try_cast_into` promises to return its
1154                    // original argument on error, which was originally produced
1155                    // by `self.as_inner()`, which is guaranteed to satisfy
1156                    // `Ptr`'s invariants.
1157                    unsafe { Ptr::from_inner(inner) },
1158                    )
1159                },
1160            )?;
1161
1162            // SAFETY:
1163            // 0. Since `U: Read<I::Aliasing, _>`, either:
1164            //    - `I::Aliasing` is `Exclusive`, in which case both `src` and
1165            //      `ptr` conform to `Exclusive`
1166            //    - `I::Aliasing` is `Shared` and `U` is `Immutable` (we already
1167            //      know that `[u8]: Immutable`). In this case, neither `U` nor
1168            //      `[u8]` permit mutation, and so `Shared` aliasing is
1169            //      satisfied.
1170            // 1. `ptr` conforms to the alignment invariant of `Aligned` because
1171            //    it is derived from `try_cast_into`, which promises that the
1172            //    object described by `target` is validly aligned for `U`.
1173            // 2. By trait bound, `self` - and thus `target` - is a bit-valid
1174            //    `[u8]`. `Ptr<[u8], (_, _, Safe)>` and `Ptr<_, (_, _,
1175            //    Initialized)>` have the same bit validity, and so neither
1176            //    `self` nor `res` can be used to write a value to the referent
1177            //    which violates the other's validity invariant.
1178            let res = unsafe { Ptr::from_inner(inner) };
1179
1180            // SAFETY:
1181            // 0. `self` and `remainder` both have the type `[u8]`. Thus, they
1182            //    have `UnsafeCell`s at the same locations. Type casting does not
1183            //    affect aliasing.
1184            // 1. `[u8]` has no alignment requirement.
1185            // 2. `self` has validity `Safe` and has type `[u8]`. Since
1186            //    `remainder` references a subset of `self`'s referent, it is
1187            //    also a bit-valid `[u8]`. Thus, neither `self` nor `remainder`
1188            //    can be used to write a value to the referent which violates the
1189            //    other's validity invariant.
1190            let remainder = unsafe { Ptr::from_inner(remainder) };
1191
1192            Ok((res, remainder))
1193        }
1194
1195        /// Attempts to cast `self` into a `U`, failing if all of the bytes of
1196        /// `self` cannot be treated as a `U`.
1197        ///
1198        /// In particular, this method fails if `self` is not validly-aligned
1199        /// for `U` or if `self`'s size is not a valid size for `U`.
1200        ///
1201        /// # Safety
1202        ///
1203        /// On success, the caller may assume that the returned pointer
1204        /// references the same byte range as `self`.
1205        #[allow(unused)]
1206        #[inline(always)]
1207        pub fn try_cast_into_no_leftover<U, R>(
1208            self,
1209            meta: Option<U::PointerMetadata>,
1210        ) -> Result<Ptr<'a, U, (I::Aliasing, Aligned, Initialized)>, CastError<Self, U>>
1211        where
1212            I::Aliasing: Reference,
1213            U: 'a + ?Sized + KnownLayout + Read<I::Aliasing, R>,
1214            [u8]: Read<I::Aliasing, R>,
1215        {
1216            // SAFETY: The provided closure returns the only copy of `slf`.
1217            unsafe {
1218                self.try_with_unchecked(
1219                    #[inline(always)]
1220                    |slf| match slf.try_cast_into(CastType::Prefix, meta) {
1221                        Ok((slf, remainder)) => {
1222                            if remainder.is_empty() {
1223                                Ok(slf)
1224                            } else {
1225                                Err(CastError::Size(SizeError::<_, U>::new(())))
1226                            }
1227                        }
1228                        Err(err) => Err(err.map_src(
1229                            #[inline(always)]
1230                            |_slf| (),
1231                        )),
1232                    },
1233                )
1234            }
1235        }
1236    }
1237
1238    impl<'a, T, I> Ptr<'a, UnsafeCell<T>, I>
1239    where
1240        T: 'a + ?Sized,
1241        I: Invariants<Aliasing = Exclusive>,
1242    {
1243        /// Converts this `Ptr` into a pointer to the underlying data.
1244        ///
1245        /// This call borrows the `UnsafeCell` mutably (at compile-time) which
1246        /// guarantees that we possess the only reference.
1247        ///
1248        /// This is like [`UnsafeCell::get_mut`], but for `Ptr`.
1249        ///
1250        /// [`UnsafeCell::get_mut`]: core::cell::UnsafeCell::get_mut
1251        #[must_use]
1252        #[inline(always)]
1253        pub fn get_mut(self) -> Ptr<'a, T, I> {
1254            // SAFETY: As described below, `UnsafeCell<T>` has the same size
1255            // as `T: ?Sized` (same static size or same DST layout). Thus,
1256            // `*const UnsafeCell<T> as *const T` is a size-preserving cast.
1257            define_cast!(unsafe { Cast<T: ?Sized> = UnsafeCell<T> => T });
1258
1259            // SAFETY:
1260            // - Aliasing is `Exclusive`, and so we are not required to promise
1261            //   anything about the locations of `UnsafeCell`s.
1262            // - `UnsafeCell<T>` has the same bit validity as `T` [1].
1263            //   Technically the term "representation" doesn't guarantee this,
1264            //   but the subsequent sentence in the documentation makes it clear
1265            //   that this is the intention.
1266            //
1267            //   By invariant on `Validity`, since `T` and `UnsafeCell<T>` have
1268            //   the same bit validity, then the set of values which may appear
1269            //   in the referent of a `Ptr<T, (_, _, V)>` is the same as the set
1270            //   which may appear in the referent of a `Ptr<UnsafeCell<T>, (_,
1271            //   _, V)>`. Thus, neither `self` nor `ptr` may be used to write a
1272            //   value to the referent which would violate the other's validity
1273            //   invariant.
1274            //
1275            // [1] Per https://doc.rust-lang.org/1.81.0/core/cell/struct.UnsafeCell.html#memory-layout:
1276            //
1277            //   `UnsafeCell<T>` has the same in-memory representation as its
1278            //   inner type `T`. A consequence of this guarantee is that it is
1279            //   possible to convert between `T` and `UnsafeCell<T>`.
1280            let ptr = unsafe { self.project_transmute_unchecked::<_, _, Cast>() };
1281
1282            // SAFETY: `UnsafeCell<T>` has the same alignment as `T` [1],
1283            // and so if `self` is guaranteed to be aligned, then so is the
1284            // returned `Ptr`.
1285            //
1286            // [1] Per https://doc.rust-lang.org/1.81.0/core/cell/struct.UnsafeCell.html#memory-layout:
1287            //
1288            //   `UnsafeCell<T>` has the same in-memory representation as
1289            //   its inner type `T`. A consequence of this guarantee is that
1290            //   it is possible to convert between `T` and `UnsafeCell<T>`.
1291            let ptr = unsafe { ptr.assume_alignment::<I::Alignment>() };
1292            ptr.unify_invariants()
1293        }
1294    }
1295}
1296
1297/// Projections through the referent.
1298mod _project {
1299    use super::*;
1300
1301    impl<'a, T, I> Ptr<'a, [T], I>
1302    where
1303        T: 'a,
1304        I: Invariants,
1305        I::Aliasing: Reference,
1306    {
1307        /// Iteratively projects the elements `Ptr<T>` from `Ptr<[T]>`.
1308        #[inline]
1309        pub fn iter(self) -> impl Iterator<Item = Ptr<'a, T, I>> {
1310            // SAFETY:
1311            // 0. `elem` conforms to the aliasing invariant of `I::Aliasing`:
1312            //    - `Exclusive`: `self` is consumed by value, and therefore
1313            //      cannot be used to access the slice while any yielded
1314            //      element `Ptr` is live. Each non-zero-sized element is a
1315            //      disjoint byte range within the slice, and zero-sized
1316            //      elements address no bytes, so distinct yielded element
1317            //      `Ptr`s do not alias each other.
1318            //    - `Shared`: It is sound for multiple shared `Ptr`s to exist
1319            //      simultaneously which reference the same memory.
1320            // 1. `elem`, conditionally, conforms to the validity invariant of
1321            //    `I::Alignment`. If `elem` is projected from data well-aligned
1322            //    for `[T]`, `elem` will be valid for `T`.
1323            // 2. `elem` conforms to the validity invariant of `I::Validity`.
1324            //    Per https://doc.rust-lang.org/1.81.0/reference/type-layout.html#array-layout:
1325            //
1326            //      Slices have the same layout as the section of the array they
1327            //      slice.
1328            //
1329            //    Arrays are laid out so that the zero-based `nth` element of
1330            //    the array is offset from the start of the array by `n *
1331            //    size_of::<T>()` bytes. Thus, `elem` addresses a valid `T`
1332            //    within the slice. Since `self` satisfies `I::Validity`, `elem`
1333            //    also satisfies `I::Validity`.
1334            self.as_inner().iter().map(
1335                #[inline(always)]
1336                |elem| unsafe { Ptr::from_inner(elem) },
1337            )
1338        }
1339    }
1340
1341    #[allow(clippy::needless_lifetimes)]
1342    impl<'a, T, I> Ptr<'a, T, I>
1343    where
1344        T: 'a + ?Sized + KnownLayout<PointerMetadata = usize>,
1345        I: Invariants,
1346    {
1347        /// The number of slice elements in the object referenced by `self`.
1348        #[inline]
1349        #[must_use]
1350        pub fn len(&self) -> usize {
1351            self.as_inner().meta().get()
1352        }
1353
1354        /// Returns `true` if the slice pointer has a length of 0.
1355        #[inline]
1356        #[must_use]
1357        pub fn is_empty(&self) -> bool {
1358            self.len() == 0
1359        }
1360    }
1361}
1362
1363#[cfg(test)]
1364mod tests {
1365    use core::mem::{self, MaybeUninit};
1366
1367    use super::*;
1368    #[allow(unused)] // Needed on our MSRV, but considered unused on later toolchains.
1369    use crate::util::AsAddress;
1370    use crate::{pointer::BecauseImmutable, util::testutil::AU64, FromBytes, Immutable};
1371
1372    #[test]
1373    fn test_project_tag_downgrades_aliasing() {
1374        #[allow(dead_code)]
1375        #[derive(zerocopy_derive::Project)]
1376        #[repr(u8)]
1377        enum Enum {
1378            Variant(u8),
1379        }
1380
1381        let mut value = Enum::Variant(0);
1382        let _: Ptr<
1383            '_,
1384            <Enum as crate::HasTag<crate::project_clients::ProjectDerive>>::Tag,
1385            (Shared, Aligned, Safe),
1386        > = Ptr::from_mut(&mut value).project_tag::<crate::project_clients::ProjectDerive>();
1387    }
1388
1389    mod test_ptr_try_cast_into_soundness {
1390        use super::*;
1391
1392        // This test is designed so that if `Ptr::try_cast_into_xxx` are
1393        // buggy, it will manifest as unsoundness that Miri can detect.
1394
1395        // - If `size_of::<T>() == 0`, `N == 4`
1396        // - Else, `N == 4 * size_of::<T>()`
1397        //
1398        // Each test will be run for each metadata in `metas`.
1399        fn test<T, I, const N: usize>(metas: I)
1400        where
1401            T: ?Sized + KnownLayout + Immutable + FromBytes,
1402            I: IntoIterator<Item = Option<T::PointerMetadata>> + Clone,
1403        {
1404            let mut bytes = [MaybeUninit::<u8>::uninit(); N];
1405            let initialized = [MaybeUninit::new(0u8); N];
1406            for start in 0..=bytes.len() {
1407                for end in start..=bytes.len() {
1408                    // Set all bytes to uninitialized other than those in
1409                    // the range we're going to pass to `try_cast_from`.
1410                    // This allows Miri to detect out-of-bounds reads
1411                    // because they read uninitialized memory. Without this,
1412                    // some out-of-bounds reads would still be in-bounds of
1413                    // `bytes`, and so might spuriously be accepted.
1414                    bytes = [MaybeUninit::<u8>::uninit(); N];
1415                    let bytes = &mut bytes[start..end];
1416                    // Initialize only the byte range we're going to pass to
1417                    // `try_cast_from`.
1418                    bytes.copy_from_slice(&initialized[start..end]);
1419
1420                    let bytes = {
1421                        let bytes: *const [MaybeUninit<u8>] = bytes;
1422                        #[allow(clippy::as_conversions)]
1423                        let bytes = bytes as *const [u8];
1424                        // SAFETY: We just initialized these bytes to valid
1425                        // `u8`s.
1426                        unsafe { &*bytes }
1427                    };
1428
1429                    // SAFETY: The bytes in `slf` must be initialized.
1430                    unsafe fn validate_and_get_len<
1431                        T: ?Sized + KnownLayout + FromBytes + Immutable,
1432                    >(
1433                        slf: Ptr<'_, T, (Shared, Aligned, Initialized)>,
1434                    ) -> usize {
1435                        let t = slf.recall_validity().as_ref();
1436
1437                        let bytes = {
1438                            let len = mem::size_of_val(t);
1439                            let t: *const T = t;
1440                            // SAFETY:
1441                            // - We know `t`'s bytes are all initialized
1442                            //   because we just read it from `slf`, which
1443                            //   points to an initialized range of bytes. If
1444                            //   there's a bug and this doesn't hold, then
1445                            //   that's exactly what we're hoping Miri will
1446                            //   catch!
1447                            // - Since `T: FromBytes`, `T` doesn't contain
1448                            //   any `UnsafeCell`s, so it's okay for `t: T`
1449                            //   and a `&[u8]` to the same memory to be
1450                            //   alive concurrently.
1451                            unsafe { core::slice::from_raw_parts(t.cast::<u8>(), len) }
1452                        };
1453
1454                        // This assertion ensures that `t`'s bytes are read
1455                        // and compared to another value, which in turn
1456                        // ensures that Miri gets a chance to notice if any
1457                        // of `t`'s bytes are uninitialized, which they
1458                        // shouldn't be (see the comment above).
1459                        assert_eq!(bytes, vec![0u8; bytes.len()]);
1460
1461                        mem::size_of_val(t)
1462                    }
1463
1464                    for meta in metas.clone().into_iter() {
1465                        for cast_type in [CastType::Prefix, CastType::Suffix] {
1466                            if let Ok((slf, remaining)) = Ptr::from_ref(bytes)
1467                                .try_cast_into::<T, BecauseImmutable>(cast_type, meta)
1468                            {
1469                                // SAFETY: All bytes in `bytes` have been
1470                                // initialized.
1471                                let len = unsafe { validate_and_get_len(slf) };
1472                                assert_eq!(remaining.len(), bytes.len() - len);
1473                                #[allow(unstable_name_collisions)]
1474                                let bytes_addr = bytes.as_ptr().addr();
1475                                #[allow(unstable_name_collisions)]
1476                                let remaining_addr = remaining.as_inner().as_ptr().addr();
1477                                match cast_type {
1478                                    CastType::Prefix => {
1479                                        assert_eq!(remaining_addr, bytes_addr + len)
1480                                    }
1481                                    CastType::Suffix => assert_eq!(remaining_addr, bytes_addr),
1482                                }
1483
1484                                if let Some(want) = meta {
1485                                    let got =
1486                                        KnownLayout::pointer_to_metadata(slf.as_inner().as_ptr());
1487                                    assert_eq!(got, want);
1488                                }
1489                            }
1490                        }
1491
1492                        if let Ok(slf) = Ptr::from_ref(bytes)
1493                            .try_cast_into_no_leftover::<T, BecauseImmutable>(meta)
1494                        {
1495                            // SAFETY: All bytes in `bytes` have been
1496                            // initialized.
1497                            let len = unsafe { validate_and_get_len(slf) };
1498                            assert_eq!(len, bytes.len());
1499
1500                            if let Some(want) = meta {
1501                                let got = KnownLayout::pointer_to_metadata(slf.as_inner().as_ptr());
1502                                assert_eq!(got, want);
1503                            }
1504                        }
1505                    }
1506                }
1507            }
1508        }
1509
1510        #[derive(FromBytes, KnownLayout, Immutable)]
1511        #[repr(C)]
1512        struct SliceDst<T> {
1513            a: u8,
1514            trailing: [T],
1515        }
1516
1517        // Each test case becomes its own `#[test]` function. We do this because
1518        // this test in particular takes far, far longer to execute under Miri
1519        // than all of our other tests combined. Previously, we had these
1520        // execute sequentially in a single test function. We run Miri tests in
1521        // parallel in CI, but this test being sequential meant that most of
1522        // that parallelism was wasted, as all other tests would finish in a
1523        // fraction of the total execution time, leaving this test to execute on
1524        // a single thread for the remainder of the test. By putting each test
1525        // case in its own function, we permit better use of available
1526        // parallelism.
1527        macro_rules! test {
1528            ($test_name:ident: $ty:ty) => {
1529                #[test]
1530                #[allow(non_snake_case)]
1531                fn $test_name() {
1532                    const S: usize = core::mem::size_of::<$ty>();
1533                    const N: usize = if S == 0 { 4 } else { S * 4 };
1534                    test::<$ty, _, N>([None]);
1535
1536                    // If `$ty` is a ZST, then we can't pass `None` as the
1537                    // pointer metadata, or else computing the correct trailing
1538                    // slice length will panic.
1539                    if S == 0 {
1540                        test::<[$ty], _, N>([Some(0), Some(1), Some(2), Some(3)]);
1541                        test::<SliceDst<$ty>, _, N>([Some(0), Some(1), Some(2), Some(3)]);
1542                    } else {
1543                        test::<[$ty], _, N>([None, Some(0), Some(1), Some(2), Some(3)]);
1544                        test::<SliceDst<$ty>, _, N>([None, Some(0), Some(1), Some(2), Some(3)]);
1545                    }
1546                }
1547            };
1548            ($ty:ident) => {
1549                test!($ty: $ty);
1550            };
1551            ($($ty:ident),*) => { $(test!($ty);)* }
1552        }
1553
1554        test!(empty_tuple: ());
1555        test!(u8, u16, u32, u64, usize, AU64);
1556        test!(i8, i16, i32, i64, isize);
1557        test!(f32, f64);
1558    }
1559
1560    #[test]
1561    fn test_try_cast_into_explicit_count() {
1562        macro_rules! test {
1563            ($ty:ty, $bytes:expr, $elems:expr, $expect:expr) => {{
1564                let bytes = [0u8; $bytes];
1565                let ptr = Ptr::from_ref(&bytes[..]);
1566                let res =
1567                    ptr.try_cast_into::<$ty, BecauseImmutable>(CastType::Prefix, Some($elems));
1568                if let Some(expect) = $expect {
1569                    let (ptr, _) = res.unwrap();
1570                    assert_eq!(KnownLayout::pointer_to_metadata(ptr.as_inner().as_ptr()), expect);
1571                } else {
1572                    let _ = res.unwrap_err();
1573                }
1574            }};
1575        }
1576
1577        #[derive(KnownLayout, Immutable)]
1578        #[repr(C)]
1579        struct ZstDst {
1580            u: [u8; 8],
1581            slc: [()],
1582        }
1583
1584        test!(ZstDst, 8, 0, Some(0));
1585        test!(ZstDst, 7, 0, None);
1586
1587        test!(ZstDst, 8, usize::MAX, Some(usize::MAX));
1588        test!(ZstDst, 7, usize::MAX, None);
1589
1590        #[derive(KnownLayout, Immutable)]
1591        #[repr(C)]
1592        struct Dst {
1593            u: [u8; 8],
1594            slc: [u8],
1595        }
1596
1597        test!(Dst, 8, 0, Some(0));
1598        test!(Dst, 7, 0, None);
1599
1600        test!(Dst, 9, 1, Some(1));
1601        test!(Dst, 8, 1, None);
1602
1603        // If we didn't properly check for overflow, this would cause the
1604        // metadata to overflow to 0, and thus the cast would spuriously
1605        // succeed.
1606        test!(Dst, 8, usize::MAX - 8 + 1, None);
1607    }
1608
1609    #[test]
1610    fn test_try_cast_into_no_leftover_restores_original_slice() {
1611        let bytes = [0u8; 4];
1612        let ptr = Ptr::from_ref(&bytes[..]);
1613        let res = ptr.try_cast_into_no_leftover::<[u8; 2], BecauseImmutable>(None);
1614        match res {
1615            Ok(_) => panic!("should have failed due to leftover bytes"),
1616            Err(CastError::Size(e)) => {
1617                assert_eq!(e.into_src().len(), 4, "Should return original slice length");
1618            }
1619            Err(e) => panic!("wrong error type: {:?}", e),
1620        }
1621    }
1622
1623    #[test]
1624    fn test_iter_exclusive_yields_disjoint_ptrs() {
1625        let mut arr = [0u8, 1, 2, 3];
1626
1627        {
1628            let mut iter = Ptr::from_mut(&mut arr[..]).iter();
1629            let first = iter.next().unwrap().as_mut();
1630            let second = iter.next().unwrap().as_mut();
1631
1632            *first = 10;
1633            *second = 20;
1634            *first = 30;
1635        }
1636
1637        assert_eq!(arr, [30, 20, 2, 3]);
1638    }
1639}