zerocopy/lib.rs
1// Copyright 2018 The Fuchsia Authors
2//
3// Licensed under the 2-Clause BSD License <LICENSE-BSD or
4// https://opensource.org/license/bsd-2-clause>, Apache License, Version 2.0
5// <LICENSE-APACHE or https://www.apache.org/licenses/LICENSE-2.0>, or the MIT
6// license <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your option.
7// This file may not be copied, modified, or distributed except according to
8// those terms.
9
10// After updating the following doc comment, make sure to run the following
11// command to update `README.md` based on its contents:
12//
13// cargo -q run --manifest-path tools/Cargo.toml -p generate-readme > README.md
14
15//! ***<span style="font-size: 140%">Fast, safe, <span
16//! style="color:red;">compile error</span>. Pick two.</span>***
17//!
18//! Zerocopy makes zero-cost memory manipulation effortless. We write `unsafe`
19//! so you don't have to.
20//!
21//! *For an overview of what's changed from zerocopy 0.7, check out our [release
22//! notes][release-notes], which include a step-by-step upgrading guide.*
23//!
24//! *Have questions? Need more out of zerocopy? Submit a [customer request
25//! issue][customer-request-issue] or ask the maintainers on
26//! [GitHub][github-q-a] or [Discord][discord]!*
27//!
28//! [customer-request-issue]: https://github.com/google/zerocopy/issues/new/choose
29//! [release-notes]: https://github.com/google/zerocopy/discussions/1680
30//! [github-q-a]: https://github.com/google/zerocopy/discussions/categories/q-a
31//! [discord]: https://discord.gg/MAvWH2R6zk
32//!
33//! # Overview
34//!
35//! ##### Conversion Traits
36//!
37//! Zerocopy provides four derivable traits for zero-cost conversions:
38//! - [`TryFromBytes`] indicates that a type may safely be converted from
39//! certain byte sequences (conditional on runtime checks)
40//! - [`FromZeros`] indicates that a sequence of zero bytes represents a valid
41//! instance of a type
42//! - [`FromBytes`] indicates that a type may safely be converted from an
43//! arbitrary byte sequence
44//! - [`IntoBytes`] indicates that a type may safely be converted *to* a byte
45//! sequence
46//!
47//! These traits support sized types, slices, and [slice DSTs][slice-dsts].
48//!
49//! [slice-dsts]: KnownLayout#dynamically-sized-types
50//!
51//! ##### Marker Traits
52//!
53//! Zerocopy provides three derivable marker traits that do not provide any
54//! functionality themselves, but are required to call certain methods provided
55//! by the conversion traits:
56//! - [`KnownLayout`] indicates that zerocopy can reason about certain layout
57//! qualities of a type
58//! - [`Immutable`] indicates that a type is free from interior mutability,
59//! except by ownership or an exclusive (`&mut`) borrow
60//! - [`Unaligned`] indicates that a type's alignment requirement is 1
61//!
62//! You should generally derive these marker traits whenever possible.
63//!
64//! ##### Conversion Macros
65//!
66//! Zerocopy provides six macros for safe casting between types:
67//!
68//! - ([`try_`][try_transmute])[`transmute`] (conditionally) converts a value of
69//! one type to a value of another type of the same size
70//! - ([`try_`][try_transmute_mut])[`transmute_mut`] (conditionally) converts a
71//! mutable reference of one type to a mutable reference of another type of
72//! the same size
73//! - ([`try_`][try_transmute_ref])[`transmute_ref`] (conditionally) converts a
74//! mutable or immutable reference of one type to an immutable reference of
75//! another type of the same size
76//!
77//! These macros perform *compile-time* size and alignment checks, meaning that
78//! unconditional casts have zero cost at runtime. Conditional casts do not need
79//! to validate size or alignment runtime, but do need to validate contents.
80//!
81//! These macros cannot be used in generic contexts. For generic conversions,
82//! use the methods defined by the [conversion traits](#conversion-traits).
83//!
84//! ##### Byteorder-Aware Numerics
85//!
86//! Zerocopy provides byte-order aware integer types that support these
87//! conversions; see the [`byteorder`] module. These types are especially useful
88//! for network parsing.
89//!
90//! # Cargo Features
91//!
92//! - **`alloc`**
93//! By default, `zerocopy` is `no_std`. When the `alloc` feature is enabled,
94//! the `alloc` crate is added as a dependency, and some allocation-related
95//! functionality is added.
96//!
97//! - **`std`**
98//! By default, `zerocopy` is `no_std`. When the `std` feature is enabled, the
99//! `std` crate is added as a dependency (ie, `no_std` is disabled), and
100//! support for some `std` types is added. `std` implies `alloc`.
101//!
102//! - **`derive`**
103//! Provides derives for the core marker traits via the `zerocopy-derive`
104//! crate. These derives are re-exported from `zerocopy`, so it is not
105//! necessary to depend on `zerocopy-derive` directly.
106//!
107//! However, you may experience better compile times if you instead directly
108//! depend on both `zerocopy` and `zerocopy-derive` in your `Cargo.toml`,
109//! since doing so will allow Rust to compile these crates in parallel. To do
110//! so, do *not* enable the `derive` feature, and list both dependencies in
111//! your `Cargo.toml` with the same leading non-zero version number; e.g:
112//!
113//! ```toml
114//! [dependencies]
115//! zerocopy = "0.X"
116//! zerocopy-derive = "0.X"
117//! ```
118//!
119//! To avoid the risk of [duplicate import errors][duplicate-import-errors] if
120//! one of your dependencies enables zerocopy's `derive` feature, import
121//! derives as `use zerocopy_derive::*` rather than by name (e.g., `use
122//! zerocopy_derive::FromBytes`).
123//!
124//! - **`simd`**
125//! When the `simd` feature is enabled, `FromZeros`, `FromBytes`, and
126//! `IntoBytes` impls are emitted for all stable SIMD types which exist on the
127//! target platform. Note that the layout of SIMD types is not yet stabilized,
128//! so these impls may be removed in the future if layout changes make them
129//! invalid. For more information, see the Unsafe Code Guidelines Reference
130//! page on the [layout of packed SIMD vectors][simd-layout].
131//!
132//! - **`simd-nightly`**
133//! Enables the `simd` feature and adds support for SIMD types which are only
134//! available on nightly. Since these types are unstable, support for any type
135//! may be removed at any point in the future.
136//!
137//! - **`float-nightly`**
138//! Adds support for the unstable `f16` and `f128` types. These types are
139//! not yet fully implemented and may not be supported on all platforms.
140//!
141//! [duplicate-import-errors]: https://github.com/google/zerocopy/issues/1587
142//! [simd-layout]: https://rust-lang.github.io/unsafe-code-guidelines/layout/packed-simd-vectors.html
143//!
144//! # Security Ethos
145//!
146//! Zerocopy is expressly designed for use in security-critical contexts. We
147//! strive to ensure that that zerocopy code is sound under Rust's current
148//! memory model, and *any future memory model*. We ensure this by:
149//! - **...not 'guessing' about Rust's semantics.**
150//! We annotate `unsafe` code with a precise rationale for its soundness that
151//! cites a relevant section of Rust's official documentation. When Rust's
152//! documented semantics are unclear, we work with the Rust Operational
153//! Semantics Team to clarify Rust's documentation.
154//! - **...rigorously testing our implementation.**
155//! We run tests using [Miri], ensuring that zerocopy is sound across a wide
156//! array of supported target platforms of varying endianness and pointer
157//! width, and across both current and experimental memory models of Rust.
158//! - **...formally proving the correctness of our implementation.**
159//! We apply formal verification tools like [Kani][kani] to prove zerocopy's
160//! correctness.
161//!
162//! For more information, see our full [soundness policy].
163//!
164//! [Miri]: https://github.com/rust-lang/miri
165//! [Kani]: https://github.com/model-checking/kani
166//! [soundness policy]: https://github.com/google/zerocopy/blob/main/POLICIES.md#soundness
167//!
168//! # Relationship to Project Safe Transmute
169//!
170//! [Project Safe Transmute] is an official initiative of the Rust Project to
171//! develop language-level support for safer transmutation. The Project consults
172//! with crates like zerocopy to identify aspects of safer transmutation that
173//! would benefit from compiler support, and has developed an [experimental,
174//! compiler-supported analysis][mcp-transmutability] which determines whether,
175//! for a given type, any value of that type may be soundly transmuted into
176//! another type. Once this functionality is sufficiently mature, zerocopy
177//! intends to replace its internal transmutability analysis (implemented by our
178//! custom derives) with the compiler-supported one. This change will likely be
179//! an implementation detail that is invisible to zerocopy's users.
180//!
181//! Project Safe Transmute will not replace the need for most of zerocopy's
182//! higher-level abstractions. The experimental compiler analysis is a tool for
183//! checking the soundness of `unsafe` code, not a tool to avoid writing
184//! `unsafe` code altogether. For the foreseeable future, crates like zerocopy
185//! will still be required in order to provide higher-level abstractions on top
186//! of the building block provided by Project Safe Transmute.
187//!
188//! [Project Safe Transmute]: https://rust-lang.github.io/rfcs/2835-project-safe-transmute.html
189//! [mcp-transmutability]: https://github.com/rust-lang/compiler-team/issues/411
190//!
191//! # MSRV
192//!
193//! See our [MSRV policy].
194//!
195//! [MSRV policy]: https://github.com/google/zerocopy/blob/main/POLICIES.md#msrv
196//!
197//! # Changelog
198//!
199//! Zerocopy uses [GitHub Releases].
200//!
201//! [GitHub Releases]: https://github.com/google/zerocopy/releases
202//!
203//! # Thanks
204//!
205//! Zerocopy is maintained by engineers at Google with help from [many wonderful
206//! contributors][contributors]. Thank you to everyone who has lent a hand in
207//! making Rust a little more secure!
208//!
209//! [contributors]: https://github.com/google/zerocopy/graphs/contributors
210
211// Sometimes we want to use lints which were added after our MSRV.
212// `unknown_lints` is `warn` by default and we deny warnings in CI, so without
213// this attribute, any unknown lint would cause a CI failure when testing with
214// our MSRV.
215#![allow(unknown_lints, non_local_definitions, unreachable_patterns)]
216#![deny(renamed_and_removed_lints)]
217#![deny(
218 anonymous_parameters,
219 deprecated_in_future,
220 late_bound_lifetime_arguments,
221 missing_copy_implementations,
222 missing_debug_implementations,
223 missing_docs,
224 path_statements,
225 patterns_in_fns_without_body,
226 rust_2018_idioms,
227 trivial_numeric_casts,
228 unreachable_pub,
229 unsafe_op_in_unsafe_fn,
230 unused_extern_crates,
231 // We intentionally choose not to deny `unused_qualifications`. When items
232 // are added to the prelude (e.g., `core::mem::size_of`), this has the
233 // consequence of making some uses trigger this lint on the latest toolchain
234 // (e.g., `mem::size_of`), but fixing it (e.g. by replacing with `size_of`)
235 // does not work on older toolchains.
236 //
237 // We tested a more complicated fix in #1413, but ultimately decided that,
238 // since this lint is just a minor style lint, the complexity isn't worth it
239 // - it's fine to occasionally have unused qualifications slip through,
240 // especially since these do not affect our user-facing API in any way.
241 variant_size_differences
242)]
243#![cfg_attr(
244 __ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS,
245 deny(fuzzy_provenance_casts, lossy_provenance_casts)
246)]
247#![deny(
248 clippy::all,
249 clippy::alloc_instead_of_core,
250 clippy::arithmetic_side_effects,
251 clippy::as_underscore,
252 clippy::assertions_on_result_states,
253 clippy::as_conversions,
254 clippy::correctness,
255 clippy::dbg_macro,
256 clippy::decimal_literal_representation,
257 clippy::double_must_use,
258 clippy::get_unwrap,
259 clippy::indexing_slicing,
260 clippy::missing_inline_in_public_items,
261 clippy::missing_safety_doc,
262 clippy::multiple_unsafe_ops_per_block,
263 clippy::must_use_candidate,
264 clippy::must_use_unit,
265 clippy::obfuscated_if_else,
266 clippy::perf,
267 clippy::print_stdout,
268 clippy::return_self_not_must_use,
269 clippy::std_instead_of_core,
270 clippy::style,
271 clippy::suspicious,
272 clippy::todo,
273 clippy::undocumented_unsafe_blocks,
274 clippy::unimplemented,
275 clippy::unnested_or_patterns,
276 clippy::unwrap_used,
277 clippy::use_debug
278)]
279// `clippy::incompatible_msrv` (implied by `clippy::suspicious`): This sometimes
280// has false positives, and we test on our MSRV in CI, so it doesn't help us
281// anyway.
282#![allow(clippy::needless_lifetimes, clippy::type_complexity, clippy::incompatible_msrv)]
283#![deny(
284 rustdoc::bare_urls,
285 rustdoc::broken_intra_doc_links,
286 rustdoc::invalid_codeblock_attributes,
287 rustdoc::invalid_html_tags,
288 rustdoc::invalid_rust_codeblocks,
289 rustdoc::missing_crate_level_docs,
290 rustdoc::private_intra_doc_links
291)]
292// In test code, it makes sense to weight more heavily towards concise, readable
293// code over correct or debuggable code.
294#![cfg_attr(any(test, kani), allow(
295 // In tests, you get line numbers and have access to source code, so panic
296 // messages are less important. You also often unwrap a lot, which would
297 // make expect'ing instead very verbose.
298 clippy::unwrap_used,
299 // In tests, there's no harm to "panic risks" - the worst that can happen is
300 // that your test will fail, and you'll fix it. By contrast, panic risks in
301 // production code introduce the possibly of code panicking unexpectedly "in
302 // the field".
303 clippy::arithmetic_side_effects,
304 clippy::indexing_slicing,
305))]
306#![cfg_attr(not(any(test, kani, feature = "std")), no_std)]
307#![cfg_attr(
308 all(feature = "simd-nightly", target_arch = "arm"),
309 feature(stdarch_arm_neon_intrinsics)
310)]
311#![cfg_attr(
312 all(feature = "simd-nightly", any(target_arch = "powerpc", target_arch = "powerpc64")),
313 feature(stdarch_powerpc)
314)]
315#![cfg_attr(feature = "float-nightly", feature(f16, f128))]
316#![cfg_attr(doc_cfg, feature(doc_cfg))]
317#![cfg_attr(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS, feature(coverage_attribute))]
318#![cfg_attr(
319 any(__ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS, miri),
320 feature(layout_for_ptr)
321)]
322#![cfg_attr(all(test, __ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS), feature(test))]
323
324// This is a hack to allow zerocopy-derive derives to work in this crate. They
325// assume that zerocopy is linked as an extern crate, so they access items from
326// it as `zerocopy::Xxx`. This makes that still work.
327#[cfg(any(feature = "derive", test))]
328extern crate self as zerocopy;
329
330#[cfg(all(test, __ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS))]
331extern crate test;
332
333#[doc(hidden)]
334#[macro_use]
335pub mod util;
336
337pub mod byte_slice;
338pub mod byteorder;
339mod deprecated;
340
341#[cfg(__ZEROCOPY_INTERNAL_USE_ONLY_DEV_MODE)]
342pub mod doctests;
343
344// This module is `pub` so that zerocopy's error types and error handling
345// documentation is grouped together in a cohesive module. In practice, we
346// expect most users to use the re-export of `error`'s items to avoid identifier
347// stuttering.
348pub mod error;
349mod impls;
350#[doc(hidden)]
351pub mod layout;
352mod macros;
353#[doc(hidden)]
354pub mod pointer;
355mod r#ref;
356mod split_at;
357// FIXME(#252): If we make this pub, come up with a better name.
358mod wrappers;
359
360use core::{
361 cell::{Cell, UnsafeCell},
362 cmp::Ordering,
363 fmt::{self, Debug, Display, Formatter},
364 hash::Hasher,
365 marker::PhantomData,
366 mem::{self, ManuallyDrop, MaybeUninit as CoreMaybeUninit},
367 num::{
368 NonZeroI128, NonZeroI16, NonZeroI32, NonZeroI64, NonZeroI8, NonZeroIsize, NonZeroU128,
369 NonZeroU16, NonZeroU32, NonZeroU64, NonZeroU8, NonZeroUsize, Wrapping,
370 },
371 ops::{Deref, DerefMut},
372 ptr::{self, NonNull},
373 slice,
374};
375#[cfg(feature = "std")]
376use std::io;
377
378#[doc(hidden)]
379pub use crate::pointer::invariant::{self, BecauseExclusive};
380#[doc(hidden)]
381pub use crate::pointer::PtrInner;
382pub use crate::{
383 byte_slice::*,
384 byteorder::*,
385 error::*,
386 r#ref::*,
387 split_at::{Split, SplitAt},
388 wrappers::*,
389};
390
391#[cfg(any(feature = "alloc", test, kani))]
392extern crate alloc;
393#[cfg(any(feature = "alloc", test))]
394use alloc::{boxed::Box, vec::Vec};
395#[cfg(any(feature = "alloc", test))]
396use core::alloc::Layout;
397
398use util::MetadataOf;
399
400// Used by `KnownLayout`.
401#[doc(hidden)]
402pub use crate::layout::*;
403// Used by `TryFromBytes::is_bit_valid`.
404#[doc(hidden)]
405pub use crate::pointer::{invariant::BecauseImmutable, Maybe, Ptr};
406// For each trait polyfill, as soon as the corresponding feature is stable, the
407// polyfill import will be unused because method/function resolution will prefer
408// the inherent method/function over a trait method/function. Thus, we suppress
409// the `unused_imports` warning.
410//
411// See the documentation on `util::polyfills` for more information.
412#[allow(unused_imports)]
413use crate::util::polyfills::{self, NonNullExt as _, NumExt as _};
414
415#[cfg(all(test, not(__ZEROCOPY_INTERNAL_USE_ONLY_DEV_MODE)))]
416const _: () = {
417 #[deprecated = "Development of zerocopy using cargo is not supported. Please use `cargo.sh` or `win-cargo.bat` instead."]
418 #[allow(unused)]
419 const WARNING: () = ();
420 #[warn(deprecated)]
421 WARNING
422};
423
424/// Implements [`KnownLayout`].
425///
426/// This derive analyzes various aspects of a type's layout that are needed for
427/// some of zerocopy's APIs. It can be applied to structs, enums, and unions;
428/// e.g.:
429///
430/// ```
431/// # use zerocopy_derive::KnownLayout;
432/// #[derive(KnownLayout)]
433/// struct MyStruct {
434/// # /*
435/// ...
436/// # */
437/// }
438///
439/// #[derive(KnownLayout)]
440/// enum MyEnum {
441/// # V00,
442/// # /*
443/// ...
444/// # */
445/// }
446///
447/// #[derive(KnownLayout)]
448/// union MyUnion {
449/// # variant: u8,
450/// # /*
451/// ...
452/// # */
453/// }
454/// ```
455///
456/// # Limitations
457///
458/// This derive cannot currently be applied to unsized structs without an
459/// explicit `repr` attribute.
460///
461/// Some invocations of this derive run afoul of a [known bug] in Rust's type
462/// privacy checker. For example, this code:
463///
464/// ```compile_fail,E0446
465/// use zerocopy::*;
466/// # use zerocopy_derive::*;
467///
468/// #[derive(KnownLayout)]
469/// #[repr(C)]
470/// pub struct PublicType {
471/// leading: Foo,
472/// trailing: Bar,
473/// }
474///
475/// #[derive(KnownLayout)]
476/// struct Foo;
477///
478/// #[derive(KnownLayout)]
479/// struct Bar;
480/// ```
481///
482/// ...results in a compilation error:
483///
484/// ```text
485/// error[E0446]: private type `Bar` in public interface
486/// --> examples/bug.rs:3:10
487/// |
488/// 3 | #[derive(KnownLayout)]
489/// | ^^^^^^^^^^^ can't leak private type
490/// ...
491/// 14 | struct Bar;
492/// | ---------- `Bar` declared as private
493/// |
494/// = note: this error originates in the derive macro `KnownLayout` (in Nightly builds, run with -Z macro-backtrace for more info)
495/// ```
496///
497/// This issue arises when `#[derive(KnownLayout)]` is applied to `repr(C)`
498/// structs whose trailing field type is less public than the enclosing struct.
499///
500/// To work around this, mark the trailing field type `pub` and annotate it with
501/// `#[doc(hidden)]`; e.g.:
502///
503/// ```no_run
504/// use zerocopy::*;
505/// # use zerocopy_derive::*;
506///
507/// #[derive(KnownLayout)]
508/// #[repr(C)]
509/// pub struct PublicType {
510/// leading: Foo,
511/// trailing: Bar,
512/// }
513///
514/// #[derive(KnownLayout)]
515/// struct Foo;
516///
517/// #[doc(hidden)]
518/// #[derive(KnownLayout)]
519/// pub struct Bar; // <- `Bar` is now also `pub`
520/// ```
521///
522/// [known bug]: https://github.com/rust-lang/rust/issues/45713
523#[cfg(any(feature = "derive", test))]
524#[cfg_attr(doc_cfg, doc(cfg(feature = "derive")))]
525pub use zerocopy_derive::KnownLayout;
526// These exist so that code which was written against the old names will get
527// less confusing error messages when they upgrade to a more recent version of
528// zerocopy. On our MSRV toolchain, the error messages read, for example:
529//
530// error[E0603]: trait `FromZeroes` is private
531// --> examples/deprecated.rs:1:15
532// |
533// 1 | use zerocopy::FromZeroes;
534// | ^^^^^^^^^^ private trait
535// |
536// note: the trait `FromZeroes` is defined here
537// --> /Users/josh/workspace/zerocopy/src/lib.rs:1845:5
538// |
539// 1845 | use FromZeros as FromZeroes;
540// | ^^^^^^^^^^^^^^^^^^^^^^^
541//
542// The "note" provides enough context to make it easy to figure out how to fix
543// the error.
544#[allow(unused)]
545use {FromZeros as FromZeroes, IntoBytes as AsBytes, Ref as LayoutVerified};
546
547/// Indicates that zerocopy can reason about certain aspects of a type's layout.
548///
549/// This trait is required by many of zerocopy's APIs. It supports sized types,
550/// slices, and [slice DSTs](#dynamically-sized-types).
551///
552/// # Implementation
553///
554/// **Do not implement this trait yourself!** Instead, use
555/// [`#[derive(KnownLayout)]`][derive]; e.g.:
556///
557/// ```
558/// # use zerocopy_derive::KnownLayout;
559/// #[derive(KnownLayout)]
560/// struct MyStruct {
561/// # /*
562/// ...
563/// # */
564/// }
565///
566/// #[derive(KnownLayout)]
567/// enum MyEnum {
568/// # /*
569/// ...
570/// # */
571/// }
572///
573/// #[derive(KnownLayout)]
574/// union MyUnion {
575/// # variant: u8,
576/// # /*
577/// ...
578/// # */
579/// }
580/// ```
581///
582/// This derive performs a sophisticated analysis to deduce the layout
583/// characteristics of types. You **must** implement this trait via the derive.
584///
585/// # Dynamically-sized types
586///
587/// `KnownLayout` supports slice-based dynamically sized types ("slice DSTs").
588///
589/// A slice DST is a type whose trailing field is either a slice or another
590/// slice DST, rather than a type with fixed size. For example:
591///
592/// ```
593/// #[repr(C)]
594/// struct PacketHeader {
595/// # /*
596/// ...
597/// # */
598/// }
599///
600/// #[repr(C)]
601/// struct Packet {
602/// header: PacketHeader,
603/// body: [u8],
604/// }
605/// ```
606///
607/// It can be useful to think of slice DSTs as a generalization of slices - in
608/// other words, a normal slice is just the special case of a slice DST with
609/// zero leading fields. In particular:
610/// - Like slices, slice DSTs can have different lengths at runtime
611/// - Like slices, slice DSTs cannot be passed by-value, but only by reference
612/// or via other indirection such as `Box`
613/// - Like slices, a reference (or `Box`, or other pointer type) to a slice DST
614/// encodes the number of elements in the trailing slice field
615///
616/// ## Slice DST layout
617///
618/// Just like other composite Rust types, the layout of a slice DST is not
619/// well-defined unless it is specified using an explicit `#[repr(...)]`
620/// attribute such as `#[repr(C)]`. [Other representations are
621/// supported][reprs], but in this section, we'll use `#[repr(C)]` as our
622/// example.
623///
624/// A `#[repr(C)]` slice DST is laid out [just like sized `#[repr(C)]`
625/// types][repr-c-structs], but the presence of a variable-length field
626/// introduces the possibility of *dynamic padding*. In particular, it may be
627/// necessary to add trailing padding *after* the trailing slice field in order
628/// to satisfy the outer type's alignment, and the amount of padding required
629/// may be a function of the length of the trailing slice field. This is just a
630/// natural consequence of the normal `#[repr(C)]` rules applied to slice DSTs,
631/// but it can result in surprising behavior. For example, consider the
632/// following type:
633///
634/// ```
635/// #[repr(C)]
636/// struct Foo {
637/// a: u32,
638/// b: u8,
639/// z: [u16],
640/// }
641/// ```
642///
643/// Assuming that `u32` has alignment 4 (this is not true on all platforms),
644/// then `Foo` has alignment 4 as well. Here is the smallest possible value for
645/// `Foo`:
646///
647/// ```text
648/// byte offset | 01234567
649/// field | aaaab---
650/// ><
651/// ```
652///
653/// In this value, `z` has length 0. Abiding by `#[repr(C)]`, the lowest offset
654/// that we can place `z` at is 5, but since `z` has alignment 2, we need to
655/// round up to offset 6. This means that there is one byte of padding between
656/// `b` and `z`, then 0 bytes of `z` itself (denoted `><` in this diagram), and
657/// then two bytes of padding after `z` in order to satisfy the overall
658/// alignment of `Foo`. The size of this instance is 8 bytes.
659///
660/// What about if `z` has length 1?
661///
662/// ```text
663/// byte offset | 01234567
664/// field | aaaab-zz
665/// ```
666///
667/// In this instance, `z` has length 1, and thus takes up 2 bytes. That means
668/// that we no longer need padding after `z` in order to satisfy `Foo`'s
669/// alignment. We've now seen two different values of `Foo` with two different
670/// lengths of `z`, but they both have the same size - 8 bytes.
671///
672/// What about if `z` has length 2?
673///
674/// ```text
675/// byte offset | 012345678901
676/// field | aaaab-zzzz--
677/// ```
678///
679/// Now `z` has length 2, and thus takes up 4 bytes. This brings our un-padded
680/// size to 10, and so we now need another 2 bytes of padding after `z` to
681/// satisfy `Foo`'s alignment.
682///
683/// Again, all of this is just a logical consequence of the `#[repr(C)]` rules
684/// applied to slice DSTs, but it can be surprising that the amount of trailing
685/// padding becomes a function of the trailing slice field's length, and thus
686/// can only be computed at runtime.
687///
688/// [reprs]: https://doc.rust-lang.org/reference/type-layout.html#representations
689/// [repr-c-structs]: https://doc.rust-lang.org/reference/type-layout.html#reprc-structs
690///
691/// ## What is a valid size?
692///
693/// There are two places in zerocopy's API that we refer to "a valid size" of a
694/// type. In normal casts or conversions, where the source is a byte slice, we
695/// need to know whether the source byte slice is a valid size of the
696/// destination type. In prefix or suffix casts, we need to know whether *there
697/// exists* a valid size of the destination type which fits in the source byte
698/// slice and, if so, what the largest such size is.
699///
700/// As outlined above, a slice DST's size is defined by the number of elements
701/// in its trailing slice field. However, there is not necessarily a 1-to-1
702/// mapping between trailing slice field length and overall size. As we saw in
703/// the previous section with the type `Foo`, instances with both 0 and 1
704/// elements in the trailing `z` field result in a `Foo` whose size is 8 bytes.
705///
706/// When we say "x is a valid size of `T`", we mean one of two things:
707/// - If `T: Sized`, then we mean that `x == size_of::<T>()`
708/// - If `T` is a slice DST, then we mean that there exists a `len` such that the instance of
709/// `T` with `len` trailing slice elements has size `x`
710///
711/// When we say "largest possible size of `T` that fits in a byte slice", we
712/// mean one of two things:
713/// - If `T: Sized`, then we mean `size_of::<T>()` if the byte slice is at least
714/// `size_of::<T>()` bytes long
715/// - If `T` is a slice DST, then we mean to consider all values, `len`, such
716/// that the instance of `T` with `len` trailing slice elements fits in the
717/// byte slice, and to choose the largest such `len`, if any
718///
719///
720/// # Safety
721///
722/// This trait does not convey any safety guarantees to code outside this crate.
723///
724/// You must not rely on the `#[doc(hidden)]` internals of `KnownLayout`. Future
725/// releases of zerocopy may make backwards-breaking changes to these items,
726/// including changes that only affect soundness, which may cause code which
727/// uses those items to silently become unsound.
728///
729#[cfg_attr(feature = "derive", doc = "[derive]: zerocopy_derive::KnownLayout")]
730#[cfg_attr(
731 not(feature = "derive"),
732 doc = concat!("[derive]: https://docs.rs/zerocopy/", env!("CARGO_PKG_VERSION"), "/zerocopy/derive.KnownLayout.html"),
733)]
734#[cfg_attr(
735 not(no_zerocopy_diagnostic_on_unimplemented_1_78_0),
736 diagnostic::on_unimplemented(note = "Consider adding `#[derive(KnownLayout)]` to `{Self}`")
737)]
738pub unsafe trait KnownLayout {
739 // The `Self: Sized` bound makes it so that `KnownLayout` can still be
740 // object safe. It's not currently object safe thanks to `const LAYOUT`, and
741 // it likely won't be in the future, but there's no reason not to be
742 // forwards-compatible with object safety.
743 #[doc(hidden)]
744 fn only_derive_is_allowed_to_implement_this_trait()
745 where
746 Self: Sized;
747
748 /// The type of metadata stored in a pointer to `Self`.
749 ///
750 /// This is `()` for sized types and [`usize`] for slice DSTs.
751 type PointerMetadata: PointerMetadata;
752
753 /// A maybe-uninitialized analog of `Self`
754 ///
755 /// # Safety
756 ///
757 /// `Self::LAYOUT` and `Self::MaybeUninit::LAYOUT` are identical.
758 /// `Self::MaybeUninit` admits uninitialized bytes in all positions.
759 #[doc(hidden)]
760 type MaybeUninit: ?Sized + KnownLayout<PointerMetadata = Self::PointerMetadata>;
761
762 /// The layout of `Self`.
763 ///
764 /// # Safety
765 ///
766 /// Callers may assume that `LAYOUT` accurately reflects the layout of
767 /// `Self`. In particular:
768 /// - `LAYOUT.align` is equal to `Self`'s alignment
769 /// - If `Self: Sized`, then `LAYOUT.size_info == SizeInfo::Sized { size }`
770 /// where `size == size_of::<Self>()`
771 /// - If `Self` is a slice DST, then `LAYOUT.size_info ==
772 /// SizeInfo::SliceDst(slice_layout)` where:
773 /// - The size, `size`, of an instance of `Self` with `elems` trailing
774 /// slice elements is equal to `slice_layout.offset +
775 /// slice_layout.elem_size * elems` rounded up to the nearest multiple
776 /// of `LAYOUT.align`
777 /// - For such an instance, any bytes in the range `[slice_layout.offset +
778 /// slice_layout.elem_size * elems, size)` are padding and must not be
779 /// assumed to be initialized
780 #[doc(hidden)]
781 const LAYOUT: DstLayout;
782
783 /// SAFETY: The returned pointer has the same address and provenance as
784 /// `bytes`. If `Self` is a DST, the returned pointer's referent has `elems`
785 /// elements in its trailing slice.
786 #[doc(hidden)]
787 fn raw_from_ptr_len(bytes: NonNull<u8>, meta: Self::PointerMetadata) -> NonNull<Self>;
788
789 /// Extracts the metadata from a pointer to `Self`.
790 ///
791 /// # Safety
792 ///
793 /// `pointer_to_metadata` always returns the correct metadata stored in
794 /// `ptr`.
795 #[doc(hidden)]
796 fn pointer_to_metadata(ptr: *mut Self) -> Self::PointerMetadata;
797
798 /// Computes the length of the byte range addressed by `ptr`.
799 ///
800 /// Returns `None` if the resulting length would not fit in an `usize`.
801 ///
802 /// # Safety
803 ///
804 /// Callers may assume that `size_of_val_raw` always returns the correct
805 /// size.
806 ///
807 /// Callers may assume that, if `ptr` addresses a byte range whose length
808 /// fits in an `usize`, this will return `Some`.
809 #[doc(hidden)]
810 #[must_use]
811 #[inline(always)]
812 fn size_of_val_raw(ptr: NonNull<Self>) -> Option<usize> {
813 let meta = Self::pointer_to_metadata(ptr.as_ptr());
814 // SAFETY: `size_for_metadata` promises to only return `None` if the
815 // resulting size would not fit in a `usize`.
816 Self::size_for_metadata(meta)
817 }
818
819 #[doc(hidden)]
820 #[must_use]
821 #[inline(always)]
822 fn raw_dangling() -> NonNull<Self> {
823 let meta = Self::PointerMetadata::from_elem_count(0);
824 Self::raw_from_ptr_len(NonNull::dangling(), meta)
825 }
826
827 /// Computes the size of an object of type `Self` with the given pointer
828 /// metadata.
829 ///
830 /// # Safety
831 ///
832 /// `size_for_metadata` promises to return `None` if and only if the
833 /// resulting size would not fit in a [`usize`]. Note that the returned size
834 /// could exceed the actual maximum valid size of an allocated object,
835 /// [`isize::MAX`].
836 ///
837 /// # Examples
838 ///
839 /// ```
840 /// use zerocopy::KnownLayout;
841 ///
842 /// assert_eq!(u8::size_for_metadata(()), Some(1));
843 /// assert_eq!(u16::size_for_metadata(()), Some(2));
844 /// assert_eq!(<[u8]>::size_for_metadata(42), Some(42));
845 /// assert_eq!(<[u16]>::size_for_metadata(42), Some(84));
846 ///
847 /// // This size exceeds the maximum valid object size (`isize::MAX`):
848 /// assert_eq!(<[u8]>::size_for_metadata(usize::MAX), Some(usize::MAX));
849 ///
850 /// // This size, if computed, would exceed `usize::MAX`:
851 /// assert_eq!(<[u16]>::size_for_metadata(usize::MAX), None);
852 /// ```
853 #[inline(always)]
854 fn size_for_metadata(meta: Self::PointerMetadata) -> Option<usize> {
855 meta.size_for_metadata(Self::LAYOUT)
856 }
857}
858
859/// Efficiently produces the [`TrailingSliceLayout`] of `T`.
860#[inline(always)]
861pub(crate) fn trailing_slice_layout<T>() -> TrailingSliceLayout
862where
863 T: ?Sized + KnownLayout<PointerMetadata = usize>,
864{
865 trait LayoutFacts {
866 const SIZE_INFO: TrailingSliceLayout;
867 }
868
869 impl<T: ?Sized> LayoutFacts for T
870 where
871 T: KnownLayout<PointerMetadata = usize>,
872 {
873 const SIZE_INFO: TrailingSliceLayout = match T::LAYOUT.size_info {
874 crate::SizeInfo::Sized { .. } => const_panic!("unreachable"),
875 crate::SizeInfo::SliceDst(info) => info,
876 };
877 }
878
879 T::SIZE_INFO
880}
881
882/// The metadata associated with a [`KnownLayout`] type.
883#[doc(hidden)]
884pub trait PointerMetadata: Copy + Eq + Debug {
885 /// Constructs a `Self` from an element count.
886 ///
887 /// If `Self = ()`, this returns `()`. If `Self = usize`, this returns
888 /// `elems`. No other types are currently supported.
889 fn from_elem_count(elems: usize) -> Self;
890
891 /// Converts `self` to an element count.
892 ///
893 /// If `Self = ()`, this returns `0`. If `Self = usize`, this returns
894 /// `self`. No other types are currently supported.
895 fn to_elem_count(self) -> usize;
896
897 /// Computes the size of the object with the given layout and pointer
898 /// metadata.
899 ///
900 /// # Panics
901 ///
902 /// If `Self = ()`, `layout` must describe a sized type. If `Self = usize`,
903 /// `layout` must describe a slice DST. Otherwise, `size_for_metadata` may
904 /// panic.
905 ///
906 /// # Safety
907 ///
908 /// `size_for_metadata` promises to only return `None` if the resulting size
909 /// would not fit in a `usize`.
910 fn size_for_metadata(self, layout: DstLayout) -> Option<usize>;
911}
912
913impl PointerMetadata for () {
914 #[inline]
915 #[allow(clippy::unused_unit)]
916 fn from_elem_count(_elems: usize) -> () {}
917
918 #[inline]
919 fn to_elem_count(self) -> usize {
920 0
921 }
922
923 #[inline]
924 fn size_for_metadata(self, layout: DstLayout) -> Option<usize> {
925 match layout.size_info {
926 SizeInfo::Sized { size } => Some(size),
927 // NOTE: This branch is unreachable, but we return `None` rather
928 // than `unreachable!()` to avoid generating panic paths.
929 SizeInfo::SliceDst(_) => None,
930 }
931 }
932}
933
934impl PointerMetadata for usize {
935 #[inline]
936 fn from_elem_count(elems: usize) -> usize {
937 elems
938 }
939
940 #[inline]
941 fn to_elem_count(self) -> usize {
942 self
943 }
944
945 #[inline]
946 fn size_for_metadata(self, layout: DstLayout) -> Option<usize> {
947 match layout.size_info {
948 SizeInfo::SliceDst(TrailingSliceLayout { offset, elem_size }) => {
949 let slice_len = elem_size.checked_mul(self)?;
950 let without_padding = offset.checked_add(slice_len)?;
951 without_padding.checked_add(util::padding_needed_for(without_padding, layout.align))
952 }
953 // NOTE: This branch is unreachable, but we return `None` rather
954 // than `unreachable!()` to avoid generating panic paths.
955 SizeInfo::Sized { .. } => None,
956 }
957 }
958}
959
960// SAFETY: Delegates safety to `DstLayout::for_slice`.
961unsafe impl<T> KnownLayout for [T] {
962 #[allow(clippy::missing_inline_in_public_items, dead_code)]
963 #[cfg_attr(
964 all(coverage_nightly, __ZEROCOPY_INTERNAL_USE_ONLY_NIGHTLY_FEATURES_IN_TESTS),
965 coverage(off)
966 )]
967 fn only_derive_is_allowed_to_implement_this_trait()
968 where
969 Self: Sized,
970 {
971 }
972
973 type PointerMetadata = usize;
974
975 // SAFETY: `CoreMaybeUninit<T>::LAYOUT` and `T::LAYOUT` are identical
976 // because `CoreMaybeUninit<T>` has the same size and alignment as `T` [1].
977 // Consequently, `[CoreMaybeUninit<T>]::LAYOUT` and `[T]::LAYOUT` are
978 // identical, because they both lack a fixed-sized prefix and because they
979 // inherit the alignments of their inner element type (which are identical)
980 // [2][3].
981 //
982 // `[CoreMaybeUninit<T>]` admits uninitialized bytes at all positions
983 // because `CoreMaybeUninit<T>` admits uninitialized bytes at all positions
984 // and because the inner elements of `[CoreMaybeUninit<T>]` are laid out
985 // back-to-back [2][3].
986 //
987 // [1] Per https://doc.rust-lang.org/1.81.0/std/mem/union.MaybeUninit.html#layout-1:
988 //
989 // `MaybeUninit<T>` is guaranteed to have the same size, alignment, and ABI as
990 // `T`
991 //
992 // [2] Per https://doc.rust-lang.org/1.82.0/reference/type-layout.html#slice-layout:
993 //
994 // Slices have the same layout as the section of the array they slice.
995 //
996 // [3] Per https://doc.rust-lang.org/1.82.0/reference/type-layout.html#array-layout:
997 //
998 // An array of `[T; N]` has a size of `size_of::<T>() * N` and the same
999 // alignment of `T`. Arrays are laid out so that the zero-based `nth`
1000 // element of the array is offset from the start of the array by `n *
1001 // size_of::<T>()` bytes.
1002 type MaybeUninit = [CoreMaybeUninit<T>];
1003
1004 const LAYOUT: DstLayout = DstLayout::for_slice::<T>();
1005
1006 // SAFETY: `.cast` preserves address and provenance. The returned pointer
1007 // refers to an object with `elems` elements by construction.
1008 #[inline(always)]
1009 fn raw_from_ptr_len(data: NonNull<u8>, elems: usize) -> NonNull<Self> {
1010 // FIXME(#67): Remove this allow. See NonNullExt for more details.
1011 #[allow(unstable_name_collisions)]
1012 NonNull::slice_from_raw_parts(data.cast::<T>(), elems)
1013 }
1014
1015 #[inline(always)]
1016 fn pointer_to_metadata(ptr: *mut [T]) -> usize {
1017 #[allow(clippy::as_conversions)]
1018 let slc = ptr as *const [()];
1019
1020 // SAFETY:
1021 // - `()` has alignment 1, so `slc` is trivially aligned.
1022 // - `slc` was derived from a non-null pointer.
1023 // - The size is 0 regardless of the length, so it is sound to
1024 // materialize a reference regardless of location.
1025 // - By invariant, `self.ptr` has valid provenance.
1026 let slc = unsafe { &*slc };
1027
1028 // This is correct because the preceding `as` cast preserves the number
1029 // of slice elements. [1]
1030 //
1031 // [1] Per https://doc.rust-lang.org/reference/expressions/operator-expr.html#pointer-to-pointer-cast:
1032 //
1033 // For slice types like `[T]` and `[U]`, the raw pointer types `*const
1034 // [T]`, `*mut [T]`, `*const [U]`, and `*mut [U]` encode the number of
1035 // elements in this slice. Casts between these raw pointer types
1036 // preserve the number of elements. ... The same holds for `str` and
1037 // any compound type whose unsized tail is a slice type, such as
1038 // struct `Foo(i32, [u8])` or `(u64, Foo)`.
1039 slc.len()
1040 }
1041}
1042
1043#[rustfmt::skip]
1044impl_known_layout!(
1045 (),
1046 u8, i8, u16, i16, u32, i32, u64, i64, u128, i128, usize, isize, f32, f64,
1047 bool, char,
1048 NonZeroU8, NonZeroI8, NonZeroU16, NonZeroI16, NonZeroU32, NonZeroI32,
1049 NonZeroU64, NonZeroI64, NonZeroU128, NonZeroI128, NonZeroUsize, NonZeroIsize
1050);
1051#[rustfmt::skip]
1052#[cfg(feature = "float-nightly")]
1053impl_known_layout!(
1054 #[cfg_attr(doc_cfg, doc(cfg(feature = "float-nightly")))]
1055 f16,
1056 #[cfg_attr(doc_cfg, doc(cfg(feature = "float-nightly")))]
1057 f128
1058);
1059#[rustfmt::skip]
1060impl_known_layout!(
1061 T => Option<T>,
1062 T: ?Sized => PhantomData<T>,
1063 T => Wrapping<T>,
1064 T => CoreMaybeUninit<T>,
1065 T: ?Sized => *const T,
1066 T: ?Sized => *mut T,
1067 T: ?Sized => &'_ T,
1068 T: ?Sized => &'_ mut T,
1069);
1070impl_known_layout!(const N: usize, T => [T; N]);
1071
1072// SAFETY: `str` has the same representation as `[u8]`. `ManuallyDrop<T>` [1],
1073// `UnsafeCell<T>` [2], and `Cell<T>` [3] have the same representation as `T`.
1074//
1075// [1] Per https://doc.rust-lang.org/1.85.0/std/mem/struct.ManuallyDrop.html:
1076//
1077// `ManuallyDrop<T>` is guaranteed to have the same layout and bit validity as
1078// `T`
1079//
1080// [2] Per https://doc.rust-lang.org/1.85.0/core/cell/struct.UnsafeCell.html#memory-layout:
1081//
1082// `UnsafeCell<T>` has the same in-memory representation as its inner type
1083// `T`.
1084//
1085// [3] Per https://doc.rust-lang.org/1.85.0/core/cell/struct.Cell.html#memory-layout:
1086//
1087// `Cell<T>` has the same in-memory representation as `T`.
1088#[allow(clippy::multiple_unsafe_ops_per_block)]
1089const _: () = unsafe {
1090 unsafe_impl_known_layout!(
1091 #[repr([u8])]
1092 str
1093 );
1094 unsafe_impl_known_layout!(T: ?Sized + KnownLayout => #[repr(T)] ManuallyDrop<T>);
1095 unsafe_impl_known_layout!(T: ?Sized + KnownLayout => #[repr(T)] UnsafeCell<T>);
1096 unsafe_impl_known_layout!(T: ?Sized + KnownLayout => #[repr(T)] Cell<T>);
1097};
1098
1099// SAFETY:
1100// - By consequence of the invariant on `T::MaybeUninit` that `T::LAYOUT` and
1101// `T::MaybeUninit::LAYOUT` are equal, `T` and `T::MaybeUninit` have the same:
1102// - Fixed prefix size
1103// - Alignment
1104// - (For DSTs) trailing slice element size
1105// - By consequence of the above, referents `T::MaybeUninit` and `T` have the
1106// require the same kind of pointer metadata, and thus it is valid to perform
1107// an `as` cast from `*mut T` and `*mut T::MaybeUninit`, and this operation
1108// preserves referent size (ie, `size_of_val_raw`).
1109const _: () = unsafe {
1110 unsafe_impl_known_layout!(T: ?Sized + KnownLayout => #[repr(T::MaybeUninit)] MaybeUninit<T>)
1111};
1112
1113// FIXME(#196, #2856): Eventually, we'll want to support enums variants and
1114// union fields being treated uniformly since they behave similarly to each
1115// other in terms of projecting validity – specifically, for a type `T` with
1116// validity `V`, if `T` is a struct type, then its fields straightforwardly also
1117// have validity `V`. By contrast, if `T` is an enum or union type, then
1118// validity is not straightforwardly recursive in this way.
1119#[doc(hidden)]
1120pub const STRUCT_VARIANT_ID: i128 = -1;
1121#[doc(hidden)]
1122pub const UNION_VARIANT_ID: i128 = -2;
1123#[doc(hidden)]
1124pub const REPR_C_UNION_VARIANT_ID: i128 = -3;
1125
1126/// # Safety
1127///
1128/// `Self::ProjectToTag` must satisfy its safety invariant.
1129#[doc(hidden)]
1130pub unsafe trait HasTag {
1131 fn only_derive_is_allowed_to_implement_this_trait()
1132 where
1133 Self: Sized;
1134
1135 /// The type's enum tag, or `()` for non-enum types.
1136 type Tag: Immutable;
1137
1138 /// A pointer projection from `Self` to its tag.
1139 ///
1140 /// # Safety
1141 ///
1142 /// It must be the case that, for all `slf: Ptr<'_, Self, I>`, it is sound
1143 /// to project from `slf` to `Ptr<'_, Self::Tag, I>` using this projection.
1144 type ProjectToTag: pointer::cast::Project<Self, Self::Tag>;
1145}
1146
1147/// Projects a given field from `Self`.
1148///
1149/// All implementations of `HasField` for a particular field `f` in `Self`
1150/// should use the same `Field` type; this ensures that `Field` is inferable
1151/// given an explicit `VARIANT_ID` and `FIELD_ID`.
1152///
1153/// # Safety
1154///
1155/// A field `f` is `HasField` for `Self` if and only if:
1156///
1157/// - If `Self` has the layout of a struct or union type, then `VARIANT_ID` is
1158/// `STRUCT_VARIANT_ID` or `UNION_VARIANT_ID` respectively; otherwise, if
1159/// `Self` has the layout of an enum type, `VARIANT_ID` is the numerical index
1160/// of the enum variant in which `f` appears. Note that `Self` does not need
1161/// to actually *be* such a type – it just needs to have the same layout as
1162/// such a type. For example, a `#[repr(transparent)]` wrapper around an enum
1163/// has the same layout as that enum.
1164/// - If `f` has name `n`, `FIELD_ID` is `zerocopy::ident_id!(n)`; otherwise,
1165/// if `f` is at index `i`, `FIELD_ID` is `zerocopy::ident_id!(i)`.
1166/// - `Field` is a type with the same visibility as `f`.
1167/// - `Type` has the same type as `f`.
1168///
1169/// The caller must **not** assume that a pointer's referent being aligned
1170/// implies that calling `project` on that pointer will result in a pointer to
1171/// an aligned referent. For example, `HasField` may be implemented for
1172/// `#[repr(packed)]` structs.
1173///
1174/// The implementation of `project` must satisfy its safety post-condition.
1175#[doc(hidden)]
1176pub unsafe trait HasField<Field, const VARIANT_ID: i128, const FIELD_ID: i128>:
1177 HasTag
1178{
1179 fn only_derive_is_allowed_to_implement_this_trait()
1180 where
1181 Self: Sized;
1182
1183 /// The type of the field.
1184 type Type: ?Sized;
1185
1186 /// Projects from `slf` to the field.
1187 ///
1188 /// Users should generally not call `project` directly, and instead should
1189 /// use high-level APIs like [`PtrInner::project`] or [`Ptr::project`].
1190 ///
1191 /// # Safety
1192 ///
1193 /// The returned pointer refers to a non-strict subset of the bytes of
1194 /// `slf`'s referent, and has the same provenance as `slf`.
1195 #[must_use]
1196 fn project(slf: PtrInner<'_, Self>) -> *mut Self::Type;
1197}
1198
1199/// Projects a given field from `Self`.
1200///
1201/// Implementations of this trait encode the conditions under which a field can
1202/// be projected from a `Ptr<'_, Self, I>`, and how the invariants of that
1203/// [`Ptr`] (`I`) determine the invariants of pointers projected from it. In
1204/// other words, it is a type-level function over invariants; `I` goes in,
1205/// `Self::Invariants` comes out.
1206///
1207/// # Safety
1208///
1209/// `T: ProjectField<Field, I, VARIANT_ID, FIELD_ID>` if, for a
1210/// `ptr: Ptr<'_, T, I>` such that `T::is_projectable(ptr).is_ok()`,
1211/// `<T as HasField<Field, VARIANT_ID, FIELD_ID>>::project(ptr.as_inner())`
1212/// conforms to `T::Invariants`.
1213#[doc(hidden)]
1214pub unsafe trait ProjectField<Field, I, const VARIANT_ID: i128, const FIELD_ID: i128>:
1215 HasField<Field, VARIANT_ID, FIELD_ID>
1216where
1217 I: invariant::Invariants,
1218{
1219 fn only_derive_is_allowed_to_implement_this_trait()
1220 where
1221 Self: Sized;
1222
1223 /// The invariants of the projected field pointer, with respect to the
1224 /// invariants, `I`, of the containing pointer. The aliasing dimension of
1225 /// the invariants is guaranteed to remain unchanged.
1226 type Invariants: invariant::Invariants<Aliasing = I::Aliasing>;
1227
1228 /// The failure mode of projection. `()` if the projection is fallible,
1229 /// otherwise [`core::convert::Infallible`].
1230 type Error;
1231
1232 /// Is the given field projectable from `ptr`?
1233 ///
1234 /// If a field with [`Self::Invariants`] is projectable from the referent,
1235 /// this function produces an `Ok(ptr)` from which the projection can be
1236 /// made; otherwise `Err`.
1237 ///
1238 /// This method must be overriden if the field's projectability depends on
1239 /// the value of the bytes in `ptr`.
1240 #[inline(always)]
1241 fn is_projectable<'a>(_ptr: Ptr<'a, Self::Tag, I>) -> Result<(), Self::Error> {
1242 trait IsInfallible {
1243 const IS_INFALLIBLE: bool;
1244 }
1245
1246 struct Projection<T, Field, I, const VARIANT_ID: i128, const FIELD_ID: i128>(
1247 PhantomData<(Field, I, T)>,
1248 )
1249 where
1250 T: ?Sized + HasField<Field, VARIANT_ID, FIELD_ID>,
1251 I: invariant::Invariants;
1252
1253 impl<T, Field, I, const VARIANT_ID: i128, const FIELD_ID: i128> IsInfallible
1254 for Projection<T, Field, I, VARIANT_ID, FIELD_ID>
1255 where
1256 T: ?Sized + HasField<Field, VARIANT_ID, FIELD_ID>,
1257 I: invariant::Invariants,
1258 {
1259 const IS_INFALLIBLE: bool = {
1260 let is_infallible = match VARIANT_ID {
1261 // For nondestructive projections of struct and union
1262 // fields, the projected field's satisfaction of
1263 // `Invariants` does not depend on the value of the
1264 // referent. This default implementation of `is_projectable`
1265 // is non-destructive, as it does not overwrite any part of
1266 // the referent.
1267 crate::STRUCT_VARIANT_ID | crate::UNION_VARIANT_ID => true,
1268 _enum_variant => {
1269 use crate::invariant::{Validity, ValidityKind};
1270 match I::Validity::KIND {
1271 // The `Uninit` and `Initialized` validity
1272 // invariants do not depend on the enum's tag. In
1273 // particular, we don't actually care about what
1274 // variant is present – we can treat *any* range of
1275 // uninitialized or initialized memory as containing
1276 // an uninitialized or initialized instance of *any*
1277 // type – the type itself is irrelevant.
1278 ValidityKind::Uninit | ValidityKind::Initialized => true,
1279 // The projectability of an enum field from an
1280 // `AsInitialized` or `Valid` state is a dynamic
1281 // property of its tag.
1282 ValidityKind::AsInitialized | ValidityKind::Valid => false,
1283 }
1284 }
1285 };
1286 const_assert!(is_infallible);
1287 is_infallible
1288 };
1289 }
1290
1291 const_assert!(
1292 <Projection<Self, Field, I, VARIANT_ID, FIELD_ID> as IsInfallible>::IS_INFALLIBLE
1293 );
1294
1295 Ok(())
1296 }
1297}
1298
1299/// Analyzes whether a type is [`FromZeros`].
1300///
1301/// This derive analyzes, at compile time, whether the annotated type satisfies
1302/// the [safety conditions] of `FromZeros` and implements `FromZeros` and its
1303/// supertraits if it is sound to do so. This derive can be applied to structs,
1304/// enums, and unions; e.g.:
1305///
1306/// ```
1307/// # use zerocopy_derive::{FromZeros, Immutable};
1308/// #[derive(FromZeros)]
1309/// struct MyStruct {
1310/// # /*
1311/// ...
1312/// # */
1313/// }
1314///
1315/// #[derive(FromZeros)]
1316/// #[repr(u8)]
1317/// enum MyEnum {
1318/// # Variant0,
1319/// # /*
1320/// ...
1321/// # */
1322/// }
1323///
1324/// #[derive(FromZeros, Immutable)]
1325/// union MyUnion {
1326/// # variant: u8,
1327/// # /*
1328/// ...
1329/// # */
1330/// }
1331/// ```
1332///
1333/// [safety conditions]: trait@FromZeros#safety
1334///
1335/// # Analysis
1336///
1337/// *This section describes, roughly, the analysis performed by this derive to
1338/// determine whether it is sound to implement `FromZeros` for a given type.
1339/// Unless you are modifying the implementation of this derive, or attempting to
1340/// manually implement `FromZeros` for a type yourself, you don't need to read
1341/// this section.*
1342///
1343/// If a type has the following properties, then this derive can implement
1344/// `FromZeros` for that type:
1345///
1346/// - If the type is a struct, all of its fields must be `FromZeros`.
1347/// - If the type is an enum:
1348/// - It must have a defined representation (`repr`s `C`, `u8`, `u16`, `u32`,
1349/// `u64`, `usize`, `i8`, `i16`, `i32`, `i64`, or `isize`).
1350/// - It must have a variant with a discriminant/tag of `0`, and its fields
1351/// must be `FromZeros`. See [the reference] for a description of
1352/// discriminant values are specified.
1353/// - The fields of that variant must be `FromZeros`.
1354///
1355/// This analysis is subject to change. Unsafe code may *only* rely on the
1356/// documented [safety conditions] of `FromZeros`, and must *not* rely on the
1357/// implementation details of this derive.
1358///
1359/// [the reference]: https://doc.rust-lang.org/reference/items/enumerations.html#custom-discriminant-values-for-fieldless-enumerations
1360///
1361/// ## Why isn't an explicit representation required for structs?
1362///
1363/// Neither this derive, nor the [safety conditions] of `FromZeros`, requires
1364/// that structs are marked with `#[repr(C)]`.
1365///
1366/// Per the [Rust reference](reference),
1367///
1368/// > The representation of a type can change the padding between fields, but
1369/// > does not change the layout of the fields themselves.
1370///
1371/// [reference]: https://doc.rust-lang.org/reference/type-layout.html#representations
1372///
1373/// Since the layout of structs only consists of padding bytes and field bytes,
1374/// a struct is soundly `FromZeros` if:
1375/// 1. its padding is soundly `FromZeros`, and
1376/// 2. its fields are soundly `FromZeros`.
1377///
1378/// The answer to the first question is always yes: padding bytes do not have
1379/// any validity constraints. A [discussion] of this question in the Unsafe Code
1380/// Guidelines Working Group concluded that it would be virtually unimaginable
1381/// for future versions of rustc to add validity constraints to padding bytes.
1382///
1383/// [discussion]: https://github.com/rust-lang/unsafe-code-guidelines/issues/174
1384///
1385/// Whether a struct is soundly `FromZeros` therefore solely depends on whether
1386/// its fields are `FromZeros`.
1387// FIXME(#146): Document why we don't require an enum to have an explicit `repr`
1388// attribute.
1389#[cfg(any(feature = "derive", test))]
1390#[cfg_attr(doc_cfg, doc(cfg(feature = "derive")))]
1391pub use zerocopy_derive::FromZeros;
1392/// Analyzes whether a type is [`Immutable`].
1393///
1394/// This derive analyzes, at compile time, whether the annotated type satisfies
1395/// the [safety conditions] of `Immutable` and implements `Immutable` if it is
1396/// sound to do so. This derive can be applied to structs, enums, and unions;
1397/// e.g.:
1398///
1399/// ```
1400/// # use zerocopy_derive::Immutable;
1401/// #[derive(Immutable)]
1402/// struct MyStruct {
1403/// # /*
1404/// ...
1405/// # */
1406/// }
1407///
1408/// #[derive(Immutable)]
1409/// enum MyEnum {
1410/// # Variant0,
1411/// # /*
1412/// ...
1413/// # */
1414/// }
1415///
1416/// #[derive(Immutable)]
1417/// union MyUnion {
1418/// # variant: u8,
1419/// # /*
1420/// ...
1421/// # */
1422/// }
1423/// ```
1424///
1425/// # Analysis
1426///
1427/// *This section describes, roughly, the analysis performed by this derive to
1428/// determine whether it is sound to implement `Immutable` for a given type.
1429/// Unless you are modifying the implementation of this derive, you don't need
1430/// to read this section.*
1431///
1432/// If a type has the following properties, then this derive can implement
1433/// `Immutable` for that type:
1434///
1435/// - All fields must be `Immutable`.
1436///
1437/// This analysis is subject to change. Unsafe code may *only* rely on the
1438/// documented [safety conditions] of `Immutable`, and must *not* rely on the
1439/// implementation details of this derive.
1440///
1441/// [safety conditions]: trait@Immutable#safety
1442#[cfg(any(feature = "derive", test))]
1443#[cfg_attr(doc_cfg, doc(cfg(feature = "derive")))]
1444pub use zerocopy_derive::Immutable;
1445
1446/// Types which are free from interior mutability.
1447///
1448/// `T: Immutable` indicates that `T` does not permit interior mutation, except
1449/// by ownership or an exclusive (`&mut`) borrow.
1450///
1451/// # Implementation
1452///
1453/// **Do not implement this trait yourself!** Instead, use
1454/// [`#[derive(Immutable)]`][derive] (requires the `derive` Cargo feature);
1455/// e.g.:
1456///
1457/// ```
1458/// # use zerocopy_derive::Immutable;
1459/// #[derive(Immutable)]
1460/// struct MyStruct {
1461/// # /*
1462/// ...
1463/// # */
1464/// }
1465///
1466/// #[derive(Immutable)]
1467/// enum MyEnum {
1468/// # /*
1469/// ...
1470/// # */
1471/// }
1472///
1473/// #[derive(Immutable)]
1474/// union MyUnion {
1475/// # variant: u8,
1476/// # /*
1477/// ...
1478/// # */
1479/// }
1480/// ```
1481///
1482/// This derive performs a sophisticated, compile-time safety analysis to
1483/// determine whether a type is `Immutable`.
1484///
1485/// # Safety
1486///
1487/// Unsafe code outside of this crate must not make any assumptions about `T`
1488/// based on `T: Immutable`. We reserve the right to relax the requirements for
1489/// `Immutable` in the future, and if unsafe code outside of this crate makes
1490/// assumptions based on `T: Immutable`, future relaxations may cause that code
1491/// to become unsound.
1492///
1493// # Safety (Internal)
1494//
1495// If `T: Immutable`, unsafe code *inside of this crate* may assume that, given
1496// `t: &T`, `t` does not permit interior mutation of its referent. Because
1497// [`UnsafeCell`] is the only type which permits interior mutation, it is
1498// sufficient (though not necessary) to guarantee that `T` contains no
1499// `UnsafeCell`s.
1500//
1501// [`UnsafeCell`]: core::cell::UnsafeCell
1502#[cfg_attr(
1503 feature = "derive",
1504 doc = "[derive]: zerocopy_derive::Immutable",
1505 doc = "[derive-analysis]: zerocopy_derive::Immutable#analysis"
1506)]
1507#[cfg_attr(
1508 not(feature = "derive"),
1509 doc = concat!("[derive]: https://docs.rs/zerocopy/", env!("CARGO_PKG_VERSION"), "/zerocopy/derive.Immutable.html"),
1510 doc = concat!("[derive-analysis]: https://docs.rs/zerocopy/", env!("CARGO_PKG_VERSION"), "/zerocopy/derive.Immutable.html#analysis"),
1511)]
1512#[cfg_attr(
1513 not(no_zerocopy_diagnostic_on_unimplemented_1_78_0),
1514 diagnostic::on_unimplemented(note = "Consider adding `#[derive(Immutable)]` to `{Self}`")
1515)]
1516pub unsafe trait Immutable {
1517 // The `Self: Sized` bound makes it so that `Immutable` is still object
1518 // safe.
1519 #[doc(hidden)]
1520 fn only_derive_is_allowed_to_implement_this_trait()
1521 where
1522 Self: Sized;
1523}
1524
1525/// Implements [`TryFromBytes`].
1526///
1527/// This derive synthesizes the runtime checks required to check whether a
1528/// sequence of initialized bytes corresponds to a valid instance of a type.
1529/// This derive can be applied to structs, enums, and unions; e.g.:
1530///
1531/// ```
1532/// # use zerocopy_derive::{TryFromBytes, Immutable};
1533/// #[derive(TryFromBytes)]
1534/// struct MyStruct {
1535/// # /*
1536/// ...
1537/// # */
1538/// }
1539///
1540/// #[derive(TryFromBytes)]
1541/// #[repr(u8)]
1542/// enum MyEnum {
1543/// # V00,
1544/// # /*
1545/// ...
1546/// # */
1547/// }
1548///
1549/// #[derive(TryFromBytes, Immutable)]
1550/// union MyUnion {
1551/// # variant: u8,
1552/// # /*
1553/// ...
1554/// # */
1555/// }
1556/// ```
1557///
1558/// # Portability
1559///
1560/// To ensure consistent endianness for enums with multi-byte representations,
1561/// explicitly specify and convert each discriminant using `.to_le()` or
1562/// `.to_be()`; e.g.:
1563///
1564/// ```
1565/// # use zerocopy_derive::TryFromBytes;
1566/// // `DataStoreVersion` is encoded in little-endian.
1567/// #[derive(TryFromBytes)]
1568/// #[repr(u32)]
1569/// pub enum DataStoreVersion {
1570/// /// Version 1 of the data store.
1571/// V1 = 9u32.to_le(),
1572///
1573/// /// Version 2 of the data store.
1574/// V2 = 10u32.to_le(),
1575/// }
1576/// ```
1577///
1578/// [safety conditions]: trait@TryFromBytes#safety
1579#[cfg(any(feature = "derive", test))]
1580#[cfg_attr(doc_cfg, doc(cfg(feature = "derive")))]
1581pub use zerocopy_derive::TryFromBytes;
1582
1583/// Types for which some bit patterns are valid.
1584///
1585/// A memory region of the appropriate length which contains initialized bytes
1586/// can be viewed as a `TryFromBytes` type so long as the runtime value of those
1587/// bytes corresponds to a [*valid instance*] of that type. For example,
1588/// [`bool`] is `TryFromBytes`, so zerocopy can transmute a [`u8`] into a
1589/// [`bool`] so long as it first checks that the value of the [`u8`] is `0` or
1590/// `1`.
1591///
1592/// # Implementation
1593///
1594/// **Do not implement this trait yourself!** Instead, use
1595/// [`#[derive(TryFromBytes)]`][derive]; e.g.:
1596///
1597/// ```
1598/// # use zerocopy_derive::{TryFromBytes, Immutable};
1599/// #[derive(TryFromBytes)]
1600/// struct MyStruct {
1601/// # /*
1602/// ...
1603/// # */
1604/// }
1605///
1606/// #[derive(TryFromBytes)]
1607/// #[repr(u8)]
1608/// enum MyEnum {
1609/// # V00,
1610/// # /*
1611/// ...
1612/// # */
1613/// }
1614///
1615/// #[derive(TryFromBytes, Immutable)]
1616/// union MyUnion {
1617/// # variant: u8,
1618/// # /*
1619/// ...
1620/// # */
1621/// }
1622/// ```
1623///
1624/// This derive ensures that the runtime check of whether bytes correspond to a
1625/// valid instance is sound. You **must** implement this trait via the derive.
1626///
1627/// # What is a "valid instance"?
1628///
1629/// In Rust, each type has *bit validity*, which refers to the set of bit
1630/// patterns which may appear in an instance of that type. It is impossible for
1631/// safe Rust code to produce values which violate bit validity (ie, values
1632/// outside of the "valid" set of bit patterns). If `unsafe` code produces an
1633/// invalid value, this is considered [undefined behavior].
1634///
1635/// Rust's bit validity rules are currently being decided, which means that some
1636/// types have three classes of bit patterns: those which are definitely valid,
1637/// and whose validity is documented in the language; those which may or may not
1638/// be considered valid at some point in the future; and those which are
1639/// definitely invalid.
1640///
1641/// Zerocopy takes a conservative approach, and only considers a bit pattern to
1642/// be valid if its validity is a documented guarantee provided by the
1643/// language.
1644///
1645/// For most use cases, Rust's current guarantees align with programmers'
1646/// intuitions about what ought to be valid. As a result, zerocopy's
1647/// conservatism should not affect most users.
1648///
1649/// If you are negatively affected by lack of support for a particular type,
1650/// we encourage you to let us know by [filing an issue][github-repo].
1651///
1652/// # `TryFromBytes` is not symmetrical with [`IntoBytes`]
1653///
1654/// There are some types which implement both `TryFromBytes` and [`IntoBytes`],
1655/// but for which `TryFromBytes` is not guaranteed to accept all byte sequences
1656/// produced by `IntoBytes`. In other words, for some `T: TryFromBytes +
1657/// IntoBytes`, there exist values of `t: T` such that
1658/// `TryFromBytes::try_ref_from_bytes(t.as_bytes()) == None`. Code should not
1659/// generally assume that values produced by `IntoBytes` will necessarily be
1660/// accepted as valid by `TryFromBytes`.
1661///
1662/// # Safety
1663///
1664/// On its own, `T: TryFromBytes` does not make any guarantees about the layout
1665/// or representation of `T`. It merely provides the ability to perform a
1666/// validity check at runtime via methods like [`try_ref_from_bytes`].
1667///
1668/// You must not rely on the `#[doc(hidden)]` internals of `TryFromBytes`.
1669/// Future releases of zerocopy may make backwards-breaking changes to these
1670/// items, including changes that only affect soundness, which may cause code
1671/// which uses those items to silently become unsound.
1672///
1673/// [undefined behavior]: https://raphlinus.github.io/programming/rust/2018/08/17/undefined-behavior.html
1674/// [github-repo]: https://github.com/google/zerocopy
1675/// [`try_ref_from_bytes`]: TryFromBytes::try_ref_from_bytes
1676/// [*valid instance*]: #what-is-a-valid-instance
1677#[cfg_attr(feature = "derive", doc = "[derive]: zerocopy_derive::TryFromBytes")]
1678#[cfg_attr(
1679 not(feature = "derive"),
1680 doc = concat!("[derive]: https://docs.rs/zerocopy/", env!("CARGO_PKG_VERSION"), "/zerocopy/derive.TryFromBytes.html"),
1681)]
1682#[cfg_attr(
1683 not(no_zerocopy_diagnostic_on_unimplemented_1_78_0),
1684 diagnostic::on_unimplemented(note = "Consider adding `#[derive(TryFromBytes)]` to `{Self}`")
1685)]
1686pub unsafe trait TryFromBytes {
1687 // The `Self: Sized` bound makes it so that `TryFromBytes` is still object
1688 // safe.
1689 #[doc(hidden)]
1690 fn only_derive_is_allowed_to_implement_this_trait()
1691 where
1692 Self: Sized;
1693
1694 /// Does a given memory range contain a valid instance of `Self`?
1695 ///
1696 /// # Safety
1697 ///
1698 /// Unsafe code may assume that, if `is_bit_valid(candidate)` returns true,
1699 /// `*candidate` contains a valid `Self`.
1700 ///
1701 /// # Panics
1702 ///
1703 /// `is_bit_valid` may panic. Callers are responsible for ensuring that any
1704 /// `unsafe` code remains sound even in the face of `is_bit_valid`
1705 /// panicking. (We support user-defined validation routines; so long as
1706 /// these routines are not required to be `unsafe`, there is no way to
1707 /// ensure that these do not generate panics.)
1708 ///
1709 /// Besides user-defined validation routines panicking, `is_bit_valid` will
1710 /// either panic or fail to compile if called on a pointer with [`Shared`]
1711 /// aliasing when `Self: !Immutable`.
1712 ///
1713 /// [`UnsafeCell`]: core::cell::UnsafeCell
1714 /// [`Shared`]: invariant::Shared
1715 #[doc(hidden)]
1716 fn is_bit_valid<A>(candidate: Maybe<'_, Self, A>) -> bool
1717 where
1718 A: invariant::Alignment;
1719
1720 /// Attempts to interpret the given `source` as a `&Self`.
1721 ///
1722 /// If the bytes of `source` are a valid instance of `Self`, this method
1723 /// returns a reference to those bytes interpreted as a `Self`. If the
1724 /// length of `source` is not a [valid size of `Self`][valid-size], or if
1725 /// `source` is not appropriately aligned, or if `source` is not a valid
1726 /// instance of `Self`, this returns `Err`. If [`Self:
1727 /// Unaligned`][self-unaligned], you can [infallibly discard the alignment
1728 /// error][ConvertError::from].
1729 ///
1730 /// `Self` may be a sized type, a slice, or a [slice DST][slice-dst].
1731 ///
1732 /// [valid-size]: crate::KnownLayout#what-is-a-valid-size
1733 /// [self-unaligned]: Unaligned
1734 /// [slice-dst]: KnownLayout#dynamically-sized-types
1735 ///
1736 /// # Compile-Time Assertions
1737 ///
1738 /// This method cannot yet be used on unsized types whose dynamically-sized
1739 /// component is zero-sized. Attempting to use this method on such types
1740 /// results in a compile-time assertion error; e.g.:
1741 ///
1742 /// ```compile_fail,E0080
1743 /// use zerocopy::*;
1744 /// # use zerocopy_derive::*;
1745 ///
1746 /// #[derive(TryFromBytes, Immutable, KnownLayout)]
1747 /// #[repr(C)]
1748 /// struct ZSTy {
1749 /// leading_sized: u16,
1750 /// trailing_dst: [()],
1751 /// }
1752 ///
1753 /// let _ = ZSTy::try_ref_from_bytes(0u16.as_bytes()); // âš Compile Error!
1754 /// ```
1755 ///
1756 /// # Examples
1757 ///
1758 /// ```
1759 /// use zerocopy::TryFromBytes;
1760 /// # use zerocopy_derive::*;
1761 ///
1762 /// // The only valid value of this type is the byte `0xC0`
1763 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
1764 /// #[repr(u8)]
1765 /// enum C0 { xC0 = 0xC0 }
1766 ///
1767 /// // The only valid value of this type is the byte sequence `0xC0C0`.
1768 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
1769 /// #[repr(C)]
1770 /// struct C0C0(C0, C0);
1771 ///
1772 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
1773 /// #[repr(C)]
1774 /// struct Packet {
1775 /// magic_number: C0C0,
1776 /// mug_size: u8,
1777 /// temperature: u8,
1778 /// marshmallows: [[u8; 2]],
1779 /// }
1780 ///
1781 /// let bytes = &[0xC0, 0xC0, 240, 77, 0, 1, 2, 3, 4, 5][..];
1782 ///
1783 /// let packet = Packet::try_ref_from_bytes(bytes).unwrap();
1784 ///
1785 /// assert_eq!(packet.mug_size, 240);
1786 /// assert_eq!(packet.temperature, 77);
1787 /// assert_eq!(packet.marshmallows, [[0, 1], [2, 3], [4, 5]]);
1788 ///
1789 /// // These bytes are not valid instance of `Packet`.
1790 /// let bytes = &[0x10, 0xC0, 240, 77, 0, 1, 2, 3, 4, 5][..];
1791 /// assert!(Packet::try_ref_from_bytes(bytes).is_err());
1792 /// ```
1793 ///
1794 #[doc = codegen_section!(
1795 header = "h5",
1796 bench = "try_ref_from_bytes",
1797 format = "coco",
1798 arity = 3,
1799 [
1800 open
1801 @index 1
1802 @title "Sized"
1803 @variant "static_size"
1804 ],
1805 [
1806 @index 2
1807 @title "Unsized"
1808 @variant "dynamic_size"
1809 ],
1810 [
1811 @index 3
1812 @title "Dynamically Padded"
1813 @variant "dynamic_padding"
1814 ]
1815 )]
1816 #[must_use = "has no side effects"]
1817 #[inline]
1818 fn try_ref_from_bytes(source: &[u8]) -> Result<&Self, TryCastError<&[u8], Self>>
1819 where
1820 Self: KnownLayout + Immutable,
1821 {
1822 static_assert_dst_is_not_zst!(Self);
1823 match Ptr::from_ref(source).try_cast_into_no_leftover::<Self, BecauseImmutable>(None) {
1824 Ok(source) => {
1825 // This call may panic. If that happens, it doesn't cause any soundness
1826 // issues, as we have not generated any invalid state which we need to
1827 // fix before returning.
1828 match source.try_into_valid() {
1829 Ok(valid) => Ok(valid.as_ref()),
1830 Err(e) => {
1831 Err(e.map_src(|src| src.as_bytes::<BecauseImmutable>().as_ref()).into())
1832 }
1833 }
1834 }
1835 Err(e) => Err(e.map_src(Ptr::as_ref).into()),
1836 }
1837 }
1838
1839 /// Attempts to interpret the prefix of the given `source` as a `&Self`.
1840 ///
1841 /// This method computes the [largest possible size of `Self`][valid-size]
1842 /// that can fit in the leading bytes of `source`. If that prefix is a valid
1843 /// instance of `Self`, this method returns a reference to those bytes
1844 /// interpreted as `Self`, and a reference to the remaining bytes. If there
1845 /// are insufficient bytes, or if `source` is not appropriately aligned, or
1846 /// if those bytes are not a valid instance of `Self`, this returns `Err`.
1847 /// If [`Self: Unaligned`][self-unaligned], you can [infallibly discard the
1848 /// alignment error][ConvertError::from].
1849 ///
1850 /// `Self` may be a sized type, a slice, or a [slice DST][slice-dst].
1851 ///
1852 /// [valid-size]: crate::KnownLayout#what-is-a-valid-size
1853 /// [self-unaligned]: Unaligned
1854 /// [slice-dst]: KnownLayout#dynamically-sized-types
1855 ///
1856 /// # Compile-Time Assertions
1857 ///
1858 /// This method cannot yet be used on unsized types whose dynamically-sized
1859 /// component is zero-sized. Attempting to use this method on such types
1860 /// results in a compile-time assertion error; e.g.:
1861 ///
1862 /// ```compile_fail,E0080
1863 /// use zerocopy::*;
1864 /// # use zerocopy_derive::*;
1865 ///
1866 /// #[derive(TryFromBytes, Immutable, KnownLayout)]
1867 /// #[repr(C)]
1868 /// struct ZSTy {
1869 /// leading_sized: u16,
1870 /// trailing_dst: [()],
1871 /// }
1872 ///
1873 /// let _ = ZSTy::try_ref_from_prefix(0u16.as_bytes()); // âš Compile Error!
1874 /// ```
1875 ///
1876 /// # Examples
1877 ///
1878 /// ```
1879 /// use zerocopy::TryFromBytes;
1880 /// # use zerocopy_derive::*;
1881 ///
1882 /// // The only valid value of this type is the byte `0xC0`
1883 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
1884 /// #[repr(u8)]
1885 /// enum C0 { xC0 = 0xC0 }
1886 ///
1887 /// // The only valid value of this type is the bytes `0xC0C0`.
1888 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
1889 /// #[repr(C)]
1890 /// struct C0C0(C0, C0);
1891 ///
1892 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
1893 /// #[repr(C)]
1894 /// struct Packet {
1895 /// magic_number: C0C0,
1896 /// mug_size: u8,
1897 /// temperature: u8,
1898 /// marshmallows: [[u8; 2]],
1899 /// }
1900 ///
1901 /// // These are more bytes than are needed to encode a `Packet`.
1902 /// let bytes = &[0xC0, 0xC0, 240, 77, 0, 1, 2, 3, 4, 5, 6][..];
1903 ///
1904 /// let (packet, suffix) = Packet::try_ref_from_prefix(bytes).unwrap();
1905 ///
1906 /// assert_eq!(packet.mug_size, 240);
1907 /// assert_eq!(packet.temperature, 77);
1908 /// assert_eq!(packet.marshmallows, [[0, 1], [2, 3], [4, 5]]);
1909 /// assert_eq!(suffix, &[6u8][..]);
1910 ///
1911 /// // These bytes are not valid instance of `Packet`.
1912 /// let bytes = &[0x10, 0xC0, 240, 77, 0, 1, 2, 3, 4, 5, 6][..];
1913 /// assert!(Packet::try_ref_from_prefix(bytes).is_err());
1914 /// ```
1915 ///
1916 #[doc = codegen_section!(
1917 header = "h5",
1918 bench = "try_ref_from_prefix",
1919 format = "coco",
1920 arity = 3,
1921 [
1922 open
1923 @index 1
1924 @title "Sized"
1925 @variant "static_size"
1926 ],
1927 [
1928 @index 2
1929 @title "Unsized"
1930 @variant "dynamic_size"
1931 ],
1932 [
1933 @index 3
1934 @title "Dynamically Padded"
1935 @variant "dynamic_padding"
1936 ]
1937 )]
1938 #[must_use = "has no side effects"]
1939 #[inline]
1940 fn try_ref_from_prefix(source: &[u8]) -> Result<(&Self, &[u8]), TryCastError<&[u8], Self>>
1941 where
1942 Self: KnownLayout + Immutable,
1943 {
1944 static_assert_dst_is_not_zst!(Self);
1945 try_ref_from_prefix_suffix(source, CastType::Prefix, None)
1946 }
1947
1948 /// Attempts to interpret the suffix of the given `source` as a `&Self`.
1949 ///
1950 /// This method computes the [largest possible size of `Self`][valid-size]
1951 /// that can fit in the trailing bytes of `source`. If that suffix is a
1952 /// valid instance of `Self`, this method returns a reference to those bytes
1953 /// interpreted as `Self`, and a reference to the preceding bytes. If there
1954 /// are insufficient bytes, or if the suffix of `source` would not be
1955 /// appropriately aligned, or if the suffix is not a valid instance of
1956 /// `Self`, this returns `Err`. If [`Self: Unaligned`][self-unaligned], you
1957 /// can [infallibly discard the alignment error][ConvertError::from].
1958 ///
1959 /// `Self` may be a sized type, a slice, or a [slice DST][slice-dst].
1960 ///
1961 /// [valid-size]: crate::KnownLayout#what-is-a-valid-size
1962 /// [self-unaligned]: Unaligned
1963 /// [slice-dst]: KnownLayout#dynamically-sized-types
1964 ///
1965 /// # Compile-Time Assertions
1966 ///
1967 /// This method cannot yet be used on unsized types whose dynamically-sized
1968 /// component is zero-sized. Attempting to use this method on such types
1969 /// results in a compile-time assertion error; e.g.:
1970 ///
1971 /// ```compile_fail,E0080
1972 /// use zerocopy::*;
1973 /// # use zerocopy_derive::*;
1974 ///
1975 /// #[derive(TryFromBytes, Immutable, KnownLayout)]
1976 /// #[repr(C)]
1977 /// struct ZSTy {
1978 /// leading_sized: u16,
1979 /// trailing_dst: [()],
1980 /// }
1981 ///
1982 /// let _ = ZSTy::try_ref_from_suffix(0u16.as_bytes()); // âš Compile Error!
1983 /// ```
1984 ///
1985 /// # Examples
1986 ///
1987 /// ```
1988 /// use zerocopy::TryFromBytes;
1989 /// # use zerocopy_derive::*;
1990 ///
1991 /// // The only valid value of this type is the byte `0xC0`
1992 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
1993 /// #[repr(u8)]
1994 /// enum C0 { xC0 = 0xC0 }
1995 ///
1996 /// // The only valid value of this type is the bytes `0xC0C0`.
1997 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
1998 /// #[repr(C)]
1999 /// struct C0C0(C0, C0);
2000 ///
2001 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
2002 /// #[repr(C)]
2003 /// struct Packet {
2004 /// magic_number: C0C0,
2005 /// mug_size: u8,
2006 /// temperature: u8,
2007 /// marshmallows: [[u8; 2]],
2008 /// }
2009 ///
2010 /// // These are more bytes than are needed to encode a `Packet`.
2011 /// let bytes = &[0, 0xC0, 0xC0, 240, 77, 2, 3, 4, 5, 6, 7][..];
2012 ///
2013 /// let (prefix, packet) = Packet::try_ref_from_suffix(bytes).unwrap();
2014 ///
2015 /// assert_eq!(packet.mug_size, 240);
2016 /// assert_eq!(packet.temperature, 77);
2017 /// assert_eq!(packet.marshmallows, [[2, 3], [4, 5], [6, 7]]);
2018 /// assert_eq!(prefix, &[0u8][..]);
2019 ///
2020 /// // These bytes are not valid instance of `Packet`.
2021 /// let bytes = &[0, 1, 2, 3, 4, 5, 6, 77, 240, 0xC0, 0x10][..];
2022 /// assert!(Packet::try_ref_from_suffix(bytes).is_err());
2023 /// ```
2024 ///
2025 #[doc = codegen_section!(
2026 header = "h5",
2027 bench = "try_ref_from_suffix",
2028 format = "coco",
2029 arity = 3,
2030 [
2031 open
2032 @index 1
2033 @title "Sized"
2034 @variant "static_size"
2035 ],
2036 [
2037 @index 2
2038 @title "Unsized"
2039 @variant "dynamic_size"
2040 ],
2041 [
2042 @index 3
2043 @title "Dynamically Padded"
2044 @variant "dynamic_padding"
2045 ]
2046 )]
2047 #[must_use = "has no side effects"]
2048 #[inline]
2049 fn try_ref_from_suffix(source: &[u8]) -> Result<(&[u8], &Self), TryCastError<&[u8], Self>>
2050 where
2051 Self: KnownLayout + Immutable,
2052 {
2053 static_assert_dst_is_not_zst!(Self);
2054 try_ref_from_prefix_suffix(source, CastType::Suffix, None).map(swap)
2055 }
2056
2057 /// Attempts to interpret the given `source` as a `&mut Self` without
2058 /// copying.
2059 ///
2060 /// If the bytes of `source` are a valid instance of `Self`, this method
2061 /// returns a reference to those bytes interpreted as a `Self`. If the
2062 /// length of `source` is not a [valid size of `Self`][valid-size], or if
2063 /// `source` is not appropriately aligned, or if `source` is not a valid
2064 /// instance of `Self`, this returns `Err`. If [`Self:
2065 /// Unaligned`][self-unaligned], you can [infallibly discard the alignment
2066 /// error][ConvertError::from].
2067 ///
2068 /// `Self` may be a sized type, a slice, or a [slice DST][slice-dst].
2069 ///
2070 /// [valid-size]: crate::KnownLayout#what-is-a-valid-size
2071 /// [self-unaligned]: Unaligned
2072 /// [slice-dst]: KnownLayout#dynamically-sized-types
2073 ///
2074 /// # Compile-Time Assertions
2075 ///
2076 /// This method cannot yet be used on unsized types whose dynamically-sized
2077 /// component is zero-sized. Attempting to use this method on such types
2078 /// results in a compile-time assertion error; e.g.:
2079 ///
2080 /// ```compile_fail,E0080
2081 /// use zerocopy::*;
2082 /// # use zerocopy_derive::*;
2083 ///
2084 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2085 /// #[repr(C, packed)]
2086 /// struct ZSTy {
2087 /// leading_sized: [u8; 2],
2088 /// trailing_dst: [()],
2089 /// }
2090 ///
2091 /// let mut source = [85, 85];
2092 /// let _ = ZSTy::try_mut_from_bytes(&mut source[..]); // âš Compile Error!
2093 /// ```
2094 ///
2095 /// # Examples
2096 ///
2097 /// ```
2098 /// use zerocopy::TryFromBytes;
2099 /// # use zerocopy_derive::*;
2100 ///
2101 /// // The only valid value of this type is the byte `0xC0`
2102 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2103 /// #[repr(u8)]
2104 /// enum C0 { xC0 = 0xC0 }
2105 ///
2106 /// // The only valid value of this type is the bytes `0xC0C0`.
2107 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2108 /// #[repr(C)]
2109 /// struct C0C0(C0, C0);
2110 ///
2111 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2112 /// #[repr(C, packed)]
2113 /// struct Packet {
2114 /// magic_number: C0C0,
2115 /// mug_size: u8,
2116 /// temperature: u8,
2117 /// marshmallows: [[u8; 2]],
2118 /// }
2119 ///
2120 /// let bytes = &mut [0xC0, 0xC0, 240, 77, 0, 1, 2, 3, 4, 5][..];
2121 ///
2122 /// let packet = Packet::try_mut_from_bytes(bytes).unwrap();
2123 ///
2124 /// assert_eq!(packet.mug_size, 240);
2125 /// assert_eq!(packet.temperature, 77);
2126 /// assert_eq!(packet.marshmallows, [[0, 1], [2, 3], [4, 5]]);
2127 ///
2128 /// packet.temperature = 111;
2129 ///
2130 /// assert_eq!(bytes, [0xC0, 0xC0, 240, 111, 0, 1, 2, 3, 4, 5]);
2131 ///
2132 /// // These bytes are not valid instance of `Packet`.
2133 /// let bytes = &mut [0x10, 0xC0, 240, 77, 0, 1, 2, 3, 4, 5, 6][..];
2134 /// assert!(Packet::try_mut_from_bytes(bytes).is_err());
2135 /// ```
2136 ///
2137 #[doc = codegen_header!("h5", "try_mut_from_bytes")]
2138 ///
2139 /// See [`TryFromBytes::try_ref_from_bytes`](#method.try_ref_from_bytes.codegen).
2140 #[must_use = "has no side effects"]
2141 #[inline]
2142 fn try_mut_from_bytes(bytes: &mut [u8]) -> Result<&mut Self, TryCastError<&mut [u8], Self>>
2143 where
2144 Self: KnownLayout + IntoBytes,
2145 {
2146 static_assert_dst_is_not_zst!(Self);
2147 match Ptr::from_mut(bytes).try_cast_into_no_leftover::<Self, BecauseExclusive>(None) {
2148 Ok(source) => {
2149 // This call may panic. If that happens, it doesn't cause any soundness
2150 // issues, as we have not generated any invalid state which we need to
2151 // fix before returning.
2152 match source.try_into_valid() {
2153 Ok(source) => Ok(source.as_mut()),
2154 Err(e) => Err(e.map_src(|src| src.as_bytes().as_mut()).into()),
2155 }
2156 }
2157 Err(e) => Err(e.map_src(Ptr::as_mut).into()),
2158 }
2159 }
2160
2161 /// Attempts to interpret the prefix of the given `source` as a `&mut
2162 /// Self`.
2163 ///
2164 /// This method computes the [largest possible size of `Self`][valid-size]
2165 /// that can fit in the leading bytes of `source`. If that prefix is a valid
2166 /// instance of `Self`, this method returns a reference to those bytes
2167 /// interpreted as `Self`, and a reference to the remaining bytes. If there
2168 /// are insufficient bytes, or if `source` is not appropriately aligned, or
2169 /// if the bytes are not a valid instance of `Self`, this returns `Err`. If
2170 /// [`Self: Unaligned`][self-unaligned], you can [infallibly discard the
2171 /// alignment error][ConvertError::from].
2172 ///
2173 /// `Self` may be a sized type, a slice, or a [slice DST][slice-dst].
2174 ///
2175 /// [valid-size]: crate::KnownLayout#what-is-a-valid-size
2176 /// [self-unaligned]: Unaligned
2177 /// [slice-dst]: KnownLayout#dynamically-sized-types
2178 ///
2179 /// # Compile-Time Assertions
2180 ///
2181 /// This method cannot yet be used on unsized types whose dynamically-sized
2182 /// component is zero-sized. Attempting to use this method on such types
2183 /// results in a compile-time assertion error; e.g.:
2184 ///
2185 /// ```compile_fail,E0080
2186 /// use zerocopy::*;
2187 /// # use zerocopy_derive::*;
2188 ///
2189 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2190 /// #[repr(C, packed)]
2191 /// struct ZSTy {
2192 /// leading_sized: [u8; 2],
2193 /// trailing_dst: [()],
2194 /// }
2195 ///
2196 /// let mut source = [85, 85];
2197 /// let _ = ZSTy::try_mut_from_prefix(&mut source[..]); // âš Compile Error!
2198 /// ```
2199 ///
2200 /// # Examples
2201 ///
2202 /// ```
2203 /// use zerocopy::TryFromBytes;
2204 /// # use zerocopy_derive::*;
2205 ///
2206 /// // The only valid value of this type is the byte `0xC0`
2207 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2208 /// #[repr(u8)]
2209 /// enum C0 { xC0 = 0xC0 }
2210 ///
2211 /// // The only valid value of this type is the bytes `0xC0C0`.
2212 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2213 /// #[repr(C)]
2214 /// struct C0C0(C0, C0);
2215 ///
2216 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2217 /// #[repr(C, packed)]
2218 /// struct Packet {
2219 /// magic_number: C0C0,
2220 /// mug_size: u8,
2221 /// temperature: u8,
2222 /// marshmallows: [[u8; 2]],
2223 /// }
2224 ///
2225 /// // These are more bytes than are needed to encode a `Packet`.
2226 /// let bytes = &mut [0xC0, 0xC0, 240, 77, 0, 1, 2, 3, 4, 5, 6][..];
2227 ///
2228 /// let (packet, suffix) = Packet::try_mut_from_prefix(bytes).unwrap();
2229 ///
2230 /// assert_eq!(packet.mug_size, 240);
2231 /// assert_eq!(packet.temperature, 77);
2232 /// assert_eq!(packet.marshmallows, [[0, 1], [2, 3], [4, 5]]);
2233 /// assert_eq!(suffix, &[6u8][..]);
2234 ///
2235 /// packet.temperature = 111;
2236 /// suffix[0] = 222;
2237 ///
2238 /// assert_eq!(bytes, [0xC0, 0xC0, 240, 111, 0, 1, 2, 3, 4, 5, 222]);
2239 ///
2240 /// // These bytes are not valid instance of `Packet`.
2241 /// let bytes = &mut [0x10, 0xC0, 240, 77, 0, 1, 2, 3, 4, 5, 6][..];
2242 /// assert!(Packet::try_mut_from_prefix(bytes).is_err());
2243 /// ```
2244 ///
2245 #[doc = codegen_header!("h5", "try_mut_from_prefix")]
2246 ///
2247 /// See [`TryFromBytes::try_ref_from_prefix`](#method.try_ref_from_prefix.codegen).
2248 #[must_use = "has no side effects"]
2249 #[inline]
2250 fn try_mut_from_prefix(
2251 source: &mut [u8],
2252 ) -> Result<(&mut Self, &mut [u8]), TryCastError<&mut [u8], Self>>
2253 where
2254 Self: KnownLayout + IntoBytes,
2255 {
2256 static_assert_dst_is_not_zst!(Self);
2257 try_mut_from_prefix_suffix(source, CastType::Prefix, None)
2258 }
2259
2260 /// Attempts to interpret the suffix of the given `source` as a `&mut
2261 /// Self`.
2262 ///
2263 /// This method computes the [largest possible size of `Self`][valid-size]
2264 /// that can fit in the trailing bytes of `source`. If that suffix is a
2265 /// valid instance of `Self`, this method returns a reference to those bytes
2266 /// interpreted as `Self`, and a reference to the preceding bytes. If there
2267 /// are insufficient bytes, or if the suffix of `source` would not be
2268 /// appropriately aligned, or if the suffix is not a valid instance of
2269 /// `Self`, this returns `Err`. If [`Self: Unaligned`][self-unaligned], you
2270 /// can [infallibly discard the alignment error][ConvertError::from].
2271 ///
2272 /// `Self` may be a sized type, a slice, or a [slice DST][slice-dst].
2273 ///
2274 /// [valid-size]: crate::KnownLayout#what-is-a-valid-size
2275 /// [self-unaligned]: Unaligned
2276 /// [slice-dst]: KnownLayout#dynamically-sized-types
2277 ///
2278 /// # Compile-Time Assertions
2279 ///
2280 /// This method cannot yet be used on unsized types whose dynamically-sized
2281 /// component is zero-sized. Attempting to use this method on such types
2282 /// results in a compile-time assertion error; e.g.:
2283 ///
2284 /// ```compile_fail,E0080
2285 /// use zerocopy::*;
2286 /// # use zerocopy_derive::*;
2287 ///
2288 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2289 /// #[repr(C, packed)]
2290 /// struct ZSTy {
2291 /// leading_sized: u16,
2292 /// trailing_dst: [()],
2293 /// }
2294 ///
2295 /// let mut source = [85, 85];
2296 /// let _ = ZSTy::try_mut_from_suffix(&mut source[..]); // âš Compile Error!
2297 /// ```
2298 ///
2299 /// # Examples
2300 ///
2301 /// ```
2302 /// use zerocopy::TryFromBytes;
2303 /// # use zerocopy_derive::*;
2304 ///
2305 /// // The only valid value of this type is the byte `0xC0`
2306 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2307 /// #[repr(u8)]
2308 /// enum C0 { xC0 = 0xC0 }
2309 ///
2310 /// // The only valid value of this type is the bytes `0xC0C0`.
2311 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2312 /// #[repr(C)]
2313 /// struct C0C0(C0, C0);
2314 ///
2315 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2316 /// #[repr(C, packed)]
2317 /// struct Packet {
2318 /// magic_number: C0C0,
2319 /// mug_size: u8,
2320 /// temperature: u8,
2321 /// marshmallows: [[u8; 2]],
2322 /// }
2323 ///
2324 /// // These are more bytes than are needed to encode a `Packet`.
2325 /// let bytes = &mut [0, 0xC0, 0xC0, 240, 77, 2, 3, 4, 5, 6, 7][..];
2326 ///
2327 /// let (prefix, packet) = Packet::try_mut_from_suffix(bytes).unwrap();
2328 ///
2329 /// assert_eq!(packet.mug_size, 240);
2330 /// assert_eq!(packet.temperature, 77);
2331 /// assert_eq!(packet.marshmallows, [[2, 3], [4, 5], [6, 7]]);
2332 /// assert_eq!(prefix, &[0u8][..]);
2333 ///
2334 /// prefix[0] = 111;
2335 /// packet.temperature = 222;
2336 ///
2337 /// assert_eq!(bytes, [111, 0xC0, 0xC0, 240, 222, 2, 3, 4, 5, 6, 7]);
2338 ///
2339 /// // These bytes are not valid instance of `Packet`.
2340 /// let bytes = &mut [0, 1, 2, 3, 4, 5, 6, 77, 240, 0xC0, 0x10][..];
2341 /// assert!(Packet::try_mut_from_suffix(bytes).is_err());
2342 /// ```
2343 ///
2344 #[doc = codegen_header!("h5", "try_mut_from_suffix")]
2345 ///
2346 /// See [`TryFromBytes::try_ref_from_suffix`](#method.try_ref_from_suffix.codegen).
2347 #[must_use = "has no side effects"]
2348 #[inline]
2349 fn try_mut_from_suffix(
2350 source: &mut [u8],
2351 ) -> Result<(&mut [u8], &mut Self), TryCastError<&mut [u8], Self>>
2352 where
2353 Self: KnownLayout + IntoBytes,
2354 {
2355 static_assert_dst_is_not_zst!(Self);
2356 try_mut_from_prefix_suffix(source, CastType::Suffix, None).map(swap)
2357 }
2358
2359 /// Attempts to interpret the given `source` as a `&Self` with a DST length
2360 /// equal to `count`.
2361 ///
2362 /// This method attempts to return a reference to `source` interpreted as a
2363 /// `Self` with `count` trailing elements. If the length of `source` is not
2364 /// equal to the size of `Self` with `count` elements, if `source` is not
2365 /// appropriately aligned, or if `source` does not contain a valid instance
2366 /// of `Self`, this returns `Err`. If [`Self: Unaligned`][self-unaligned],
2367 /// you can [infallibly discard the alignment error][ConvertError::from].
2368 ///
2369 /// [self-unaligned]: Unaligned
2370 /// [slice-dst]: KnownLayout#dynamically-sized-types
2371 ///
2372 /// # Examples
2373 ///
2374 /// ```
2375 /// # #![allow(non_camel_case_types)] // For C0::xC0
2376 /// use zerocopy::TryFromBytes;
2377 /// # use zerocopy_derive::*;
2378 ///
2379 /// // The only valid value of this type is the byte `0xC0`
2380 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
2381 /// #[repr(u8)]
2382 /// enum C0 { xC0 = 0xC0 }
2383 ///
2384 /// // The only valid value of this type is the bytes `0xC0C0`.
2385 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
2386 /// #[repr(C)]
2387 /// struct C0C0(C0, C0);
2388 ///
2389 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
2390 /// #[repr(C)]
2391 /// struct Packet {
2392 /// magic_number: C0C0,
2393 /// mug_size: u8,
2394 /// temperature: u8,
2395 /// marshmallows: [[u8; 2]],
2396 /// }
2397 ///
2398 /// let bytes = &[0xC0, 0xC0, 240, 77, 2, 3, 4, 5, 6, 7][..];
2399 ///
2400 /// let packet = Packet::try_ref_from_bytes_with_elems(bytes, 3).unwrap();
2401 ///
2402 /// assert_eq!(packet.mug_size, 240);
2403 /// assert_eq!(packet.temperature, 77);
2404 /// assert_eq!(packet.marshmallows, [[2, 3], [4, 5], [6, 7]]);
2405 ///
2406 /// // These bytes are not valid instance of `Packet`.
2407 /// let bytes = &[0, 1, 2, 3, 4, 5, 6, 77, 240, 0xC0, 0xC0][..];
2408 /// assert!(Packet::try_ref_from_bytes_with_elems(bytes, 3).is_err());
2409 /// ```
2410 ///
2411 /// Since an explicit `count` is provided, this method supports types with
2412 /// zero-sized trailing slice elements. Methods such as [`try_ref_from_bytes`]
2413 /// which do not take an explicit count do not support such types.
2414 ///
2415 /// ```
2416 /// use core::num::NonZeroU16;
2417 /// use zerocopy::*;
2418 /// # use zerocopy_derive::*;
2419 ///
2420 /// #[derive(TryFromBytes, Immutable, KnownLayout)]
2421 /// #[repr(C)]
2422 /// struct ZSTy {
2423 /// leading_sized: NonZeroU16,
2424 /// trailing_dst: [()],
2425 /// }
2426 ///
2427 /// let src = 0xCAFEu16.as_bytes();
2428 /// let zsty = ZSTy::try_ref_from_bytes_with_elems(src, 42).unwrap();
2429 /// assert_eq!(zsty.trailing_dst.len(), 42);
2430 /// ```
2431 ///
2432 /// [`try_ref_from_bytes`]: TryFromBytes::try_ref_from_bytes
2433 ///
2434 #[doc = codegen_section!(
2435 header = "h5",
2436 bench = "try_ref_from_bytes_with_elems",
2437 format = "coco",
2438 arity = 2,
2439 [
2440 open
2441 @index 1
2442 @title "Unsized"
2443 @variant "dynamic_size"
2444 ],
2445 [
2446 @index 2
2447 @title "Dynamically Padded"
2448 @variant "dynamic_padding"
2449 ]
2450 )]
2451 #[must_use = "has no side effects"]
2452 #[inline]
2453 fn try_ref_from_bytes_with_elems(
2454 source: &[u8],
2455 count: usize,
2456 ) -> Result<&Self, TryCastError<&[u8], Self>>
2457 where
2458 Self: KnownLayout<PointerMetadata = usize> + Immutable,
2459 {
2460 match Ptr::from_ref(source).try_cast_into_no_leftover::<Self, BecauseImmutable>(Some(count))
2461 {
2462 Ok(source) => {
2463 // This call may panic. If that happens, it doesn't cause any soundness
2464 // issues, as we have not generated any invalid state which we need to
2465 // fix before returning.
2466 match source.try_into_valid() {
2467 Ok(source) => Ok(source.as_ref()),
2468 Err(e) => {
2469 Err(e.map_src(|src| src.as_bytes::<BecauseImmutable>().as_ref()).into())
2470 }
2471 }
2472 }
2473 Err(e) => Err(e.map_src(Ptr::as_ref).into()),
2474 }
2475 }
2476
2477 /// Attempts to interpret the prefix of the given `source` as a `&Self` with
2478 /// a DST length equal to `count`.
2479 ///
2480 /// This method attempts to return a reference to the prefix of `source`
2481 /// interpreted as a `Self` with `count` trailing elements, and a reference
2482 /// to the remaining bytes. If the length of `source` is less than the size
2483 /// of `Self` with `count` elements, if `source` is not appropriately
2484 /// aligned, or if the prefix of `source` does not contain a valid instance
2485 /// of `Self`, this returns `Err`. If [`Self: Unaligned`][self-unaligned],
2486 /// you can [infallibly discard the alignment error][ConvertError::from].
2487 ///
2488 /// [self-unaligned]: Unaligned
2489 /// [slice-dst]: KnownLayout#dynamically-sized-types
2490 ///
2491 /// # Examples
2492 ///
2493 /// ```
2494 /// # #![allow(non_camel_case_types)] // For C0::xC0
2495 /// use zerocopy::TryFromBytes;
2496 /// # use zerocopy_derive::*;
2497 ///
2498 /// // The only valid value of this type is the byte `0xC0`
2499 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
2500 /// #[repr(u8)]
2501 /// enum C0 { xC0 = 0xC0 }
2502 ///
2503 /// // The only valid value of this type is the bytes `0xC0C0`.
2504 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
2505 /// #[repr(C)]
2506 /// struct C0C0(C0, C0);
2507 ///
2508 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
2509 /// #[repr(C)]
2510 /// struct Packet {
2511 /// magic_number: C0C0,
2512 /// mug_size: u8,
2513 /// temperature: u8,
2514 /// marshmallows: [[u8; 2]],
2515 /// }
2516 ///
2517 /// let bytes = &[0xC0, 0xC0, 240, 77, 2, 3, 4, 5, 6, 7, 8][..];
2518 ///
2519 /// let (packet, suffix) = Packet::try_ref_from_prefix_with_elems(bytes, 3).unwrap();
2520 ///
2521 /// assert_eq!(packet.mug_size, 240);
2522 /// assert_eq!(packet.temperature, 77);
2523 /// assert_eq!(packet.marshmallows, [[2, 3], [4, 5], [6, 7]]);
2524 /// assert_eq!(suffix, &[8u8][..]);
2525 ///
2526 /// // These bytes are not valid instance of `Packet`.
2527 /// let bytes = &mut [0, 1, 2, 3, 4, 5, 6, 7, 8, 77, 240, 0xC0, 0xC0][..];
2528 /// assert!(Packet::try_ref_from_prefix_with_elems(bytes, 3).is_err());
2529 /// ```
2530 ///
2531 /// Since an explicit `count` is provided, this method supports types with
2532 /// zero-sized trailing slice elements. Methods such as [`try_ref_from_prefix`]
2533 /// which do not take an explicit count do not support such types.
2534 ///
2535 /// ```
2536 /// use core::num::NonZeroU16;
2537 /// use zerocopy::*;
2538 /// # use zerocopy_derive::*;
2539 ///
2540 /// #[derive(TryFromBytes, Immutable, KnownLayout)]
2541 /// #[repr(C)]
2542 /// struct ZSTy {
2543 /// leading_sized: NonZeroU16,
2544 /// trailing_dst: [()],
2545 /// }
2546 ///
2547 /// let src = 0xCAFEu16.as_bytes();
2548 /// let (zsty, _) = ZSTy::try_ref_from_prefix_with_elems(src, 42).unwrap();
2549 /// assert_eq!(zsty.trailing_dst.len(), 42);
2550 /// ```
2551 ///
2552 /// [`try_ref_from_prefix`]: TryFromBytes::try_ref_from_prefix
2553 ///
2554 #[doc = codegen_section!(
2555 header = "h5",
2556 bench = "try_ref_from_prefix_with_elems",
2557 format = "coco",
2558 arity = 2,
2559 [
2560 open
2561 @index 1
2562 @title "Unsized"
2563 @variant "dynamic_size"
2564 ],
2565 [
2566 @index 2
2567 @title "Dynamically Padded"
2568 @variant "dynamic_padding"
2569 ]
2570 )]
2571 #[must_use = "has no side effects"]
2572 #[inline]
2573 fn try_ref_from_prefix_with_elems(
2574 source: &[u8],
2575 count: usize,
2576 ) -> Result<(&Self, &[u8]), TryCastError<&[u8], Self>>
2577 where
2578 Self: KnownLayout<PointerMetadata = usize> + Immutable,
2579 {
2580 try_ref_from_prefix_suffix(source, CastType::Prefix, Some(count))
2581 }
2582
2583 /// Attempts to interpret the suffix of the given `source` as a `&Self` with
2584 /// a DST length equal to `count`.
2585 ///
2586 /// This method attempts to return a reference to the suffix of `source`
2587 /// interpreted as a `Self` with `count` trailing elements, and a reference
2588 /// to the preceding bytes. If the length of `source` is less than the size
2589 /// of `Self` with `count` elements, if the suffix of `source` is not
2590 /// appropriately aligned, or if the suffix of `source` does not contain a
2591 /// valid instance of `Self`, this returns `Err`. If [`Self:
2592 /// Unaligned`][self-unaligned], you can [infallibly discard the alignment
2593 /// error][ConvertError::from].
2594 ///
2595 /// [self-unaligned]: Unaligned
2596 /// [slice-dst]: KnownLayout#dynamically-sized-types
2597 ///
2598 /// # Examples
2599 ///
2600 /// ```
2601 /// # #![allow(non_camel_case_types)] // For C0::xC0
2602 /// use zerocopy::TryFromBytes;
2603 /// # use zerocopy_derive::*;
2604 ///
2605 /// // The only valid value of this type is the byte `0xC0`
2606 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
2607 /// #[repr(u8)]
2608 /// enum C0 { xC0 = 0xC0 }
2609 ///
2610 /// // The only valid value of this type is the bytes `0xC0C0`.
2611 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
2612 /// #[repr(C)]
2613 /// struct C0C0(C0, C0);
2614 ///
2615 /// #[derive(TryFromBytes, KnownLayout, Immutable)]
2616 /// #[repr(C)]
2617 /// struct Packet {
2618 /// magic_number: C0C0,
2619 /// mug_size: u8,
2620 /// temperature: u8,
2621 /// marshmallows: [[u8; 2]],
2622 /// }
2623 ///
2624 /// let bytes = &[123, 0xC0, 0xC0, 240, 77, 2, 3, 4, 5, 6, 7][..];
2625 ///
2626 /// let (prefix, packet) = Packet::try_ref_from_suffix_with_elems(bytes, 3).unwrap();
2627 ///
2628 /// assert_eq!(packet.mug_size, 240);
2629 /// assert_eq!(packet.temperature, 77);
2630 /// assert_eq!(packet.marshmallows, [[2, 3], [4, 5], [6, 7]]);
2631 /// assert_eq!(prefix, &[123u8][..]);
2632 ///
2633 /// // These bytes are not valid instance of `Packet`.
2634 /// let bytes = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 77, 240, 0xC0, 0xC0][..];
2635 /// assert!(Packet::try_ref_from_suffix_with_elems(bytes, 3).is_err());
2636 /// ```
2637 ///
2638 /// Since an explicit `count` is provided, this method supports types with
2639 /// zero-sized trailing slice elements. Methods such as [`try_ref_from_prefix`]
2640 /// which do not take an explicit count do not support such types.
2641 ///
2642 /// ```
2643 /// use core::num::NonZeroU16;
2644 /// use zerocopy::*;
2645 /// # use zerocopy_derive::*;
2646 ///
2647 /// #[derive(TryFromBytes, Immutable, KnownLayout)]
2648 /// #[repr(C)]
2649 /// struct ZSTy {
2650 /// leading_sized: NonZeroU16,
2651 /// trailing_dst: [()],
2652 /// }
2653 ///
2654 /// let src = 0xCAFEu16.as_bytes();
2655 /// let (_, zsty) = ZSTy::try_ref_from_suffix_with_elems(src, 42).unwrap();
2656 /// assert_eq!(zsty.trailing_dst.len(), 42);
2657 /// ```
2658 ///
2659 /// [`try_ref_from_prefix`]: TryFromBytes::try_ref_from_prefix
2660 ///
2661 #[doc = codegen_section!(
2662 header = "h5",
2663 bench = "try_ref_from_suffix_with_elems",
2664 format = "coco",
2665 arity = 2,
2666 [
2667 open
2668 @index 1
2669 @title "Unsized"
2670 @variant "dynamic_size"
2671 ],
2672 [
2673 @index 2
2674 @title "Dynamically Padded"
2675 @variant "dynamic_padding"
2676 ]
2677 )]
2678 #[must_use = "has no side effects"]
2679 #[inline]
2680 fn try_ref_from_suffix_with_elems(
2681 source: &[u8],
2682 count: usize,
2683 ) -> Result<(&[u8], &Self), TryCastError<&[u8], Self>>
2684 where
2685 Self: KnownLayout<PointerMetadata = usize> + Immutable,
2686 {
2687 try_ref_from_prefix_suffix(source, CastType::Suffix, Some(count)).map(swap)
2688 }
2689
2690 /// Attempts to interpret the given `source` as a `&mut Self` with a DST
2691 /// length equal to `count`.
2692 ///
2693 /// This method attempts to return a reference to `source` interpreted as a
2694 /// `Self` with `count` trailing elements. If the length of `source` is not
2695 /// equal to the size of `Self` with `count` elements, if `source` is not
2696 /// appropriately aligned, or if `source` does not contain a valid instance
2697 /// of `Self`, this returns `Err`. If [`Self: Unaligned`][self-unaligned],
2698 /// you can [infallibly discard the alignment error][ConvertError::from].
2699 ///
2700 /// [self-unaligned]: Unaligned
2701 /// [slice-dst]: KnownLayout#dynamically-sized-types
2702 ///
2703 /// # Examples
2704 ///
2705 /// ```
2706 /// # #![allow(non_camel_case_types)] // For C0::xC0
2707 /// use zerocopy::TryFromBytes;
2708 /// # use zerocopy_derive::*;
2709 ///
2710 /// // The only valid value of this type is the byte `0xC0`
2711 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2712 /// #[repr(u8)]
2713 /// enum C0 { xC0 = 0xC0 }
2714 ///
2715 /// // The only valid value of this type is the bytes `0xC0C0`.
2716 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2717 /// #[repr(C)]
2718 /// struct C0C0(C0, C0);
2719 ///
2720 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2721 /// #[repr(C, packed)]
2722 /// struct Packet {
2723 /// magic_number: C0C0,
2724 /// mug_size: u8,
2725 /// temperature: u8,
2726 /// marshmallows: [[u8; 2]],
2727 /// }
2728 ///
2729 /// let bytes = &mut [0xC0, 0xC0, 240, 77, 2, 3, 4, 5, 6, 7][..];
2730 ///
2731 /// let packet = Packet::try_mut_from_bytes_with_elems(bytes, 3).unwrap();
2732 ///
2733 /// assert_eq!(packet.mug_size, 240);
2734 /// assert_eq!(packet.temperature, 77);
2735 /// assert_eq!(packet.marshmallows, [[2, 3], [4, 5], [6, 7]]);
2736 ///
2737 /// packet.temperature = 111;
2738 ///
2739 /// assert_eq!(bytes, [0xC0, 0xC0, 240, 111, 2, 3, 4, 5, 6, 7]);
2740 ///
2741 /// // These bytes are not valid instance of `Packet`.
2742 /// let bytes = &mut [0, 1, 2, 3, 4, 5, 6, 77, 240, 0xC0, 0xC0][..];
2743 /// assert!(Packet::try_mut_from_bytes_with_elems(bytes, 3).is_err());
2744 /// ```
2745 ///
2746 /// Since an explicit `count` is provided, this method supports types with
2747 /// zero-sized trailing slice elements. Methods such as [`try_mut_from_bytes`]
2748 /// which do not take an explicit count do not support such types.
2749 ///
2750 /// ```
2751 /// use core::num::NonZeroU16;
2752 /// use zerocopy::*;
2753 /// # use zerocopy_derive::*;
2754 ///
2755 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2756 /// #[repr(C, packed)]
2757 /// struct ZSTy {
2758 /// leading_sized: NonZeroU16,
2759 /// trailing_dst: [()],
2760 /// }
2761 ///
2762 /// let mut src = 0xCAFEu16;
2763 /// let src = src.as_mut_bytes();
2764 /// let zsty = ZSTy::try_mut_from_bytes_with_elems(src, 42).unwrap();
2765 /// assert_eq!(zsty.trailing_dst.len(), 42);
2766 /// ```
2767 ///
2768 /// [`try_mut_from_bytes`]: TryFromBytes::try_mut_from_bytes
2769 ///
2770 #[doc = codegen_header!("h5", "try_mut_from_bytes_with_elems")]
2771 ///
2772 /// See [`TryFromBytes::try_ref_from_bytes_with_elems`](#method.try_ref_from_bytes_with_elems.codegen).
2773 #[must_use = "has no side effects"]
2774 #[inline]
2775 fn try_mut_from_bytes_with_elems(
2776 source: &mut [u8],
2777 count: usize,
2778 ) -> Result<&mut Self, TryCastError<&mut [u8], Self>>
2779 where
2780 Self: KnownLayout<PointerMetadata = usize> + IntoBytes,
2781 {
2782 match Ptr::from_mut(source).try_cast_into_no_leftover::<Self, BecauseExclusive>(Some(count))
2783 {
2784 Ok(source) => {
2785 // This call may panic. If that happens, it doesn't cause any soundness
2786 // issues, as we have not generated any invalid state which we need to
2787 // fix before returning.
2788 match source.try_into_valid() {
2789 Ok(source) => Ok(source.as_mut()),
2790 Err(e) => Err(e.map_src(|src| src.as_bytes().as_mut()).into()),
2791 }
2792 }
2793 Err(e) => Err(e.map_src(Ptr::as_mut).into()),
2794 }
2795 }
2796
2797 /// Attempts to interpret the prefix of the given `source` as a `&mut Self`
2798 /// with a DST length equal to `count`.
2799 ///
2800 /// This method attempts to return a reference to the prefix of `source`
2801 /// interpreted as a `Self` with `count` trailing elements, and a reference
2802 /// to the remaining bytes. If the length of `source` is less than the size
2803 /// of `Self` with `count` elements, if `source` is not appropriately
2804 /// aligned, or if the prefix of `source` does not contain a valid instance
2805 /// of `Self`, this returns `Err`. If [`Self: Unaligned`][self-unaligned],
2806 /// you can [infallibly discard the alignment error][ConvertError::from].
2807 ///
2808 /// [self-unaligned]: Unaligned
2809 /// [slice-dst]: KnownLayout#dynamically-sized-types
2810 ///
2811 /// # Examples
2812 ///
2813 /// ```
2814 /// # #![allow(non_camel_case_types)] // For C0::xC0
2815 /// use zerocopy::TryFromBytes;
2816 /// # use zerocopy_derive::*;
2817 ///
2818 /// // The only valid value of this type is the byte `0xC0`
2819 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2820 /// #[repr(u8)]
2821 /// enum C0 { xC0 = 0xC0 }
2822 ///
2823 /// // The only valid value of this type is the bytes `0xC0C0`.
2824 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2825 /// #[repr(C)]
2826 /// struct C0C0(C0, C0);
2827 ///
2828 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2829 /// #[repr(C, packed)]
2830 /// struct Packet {
2831 /// magic_number: C0C0,
2832 /// mug_size: u8,
2833 /// temperature: u8,
2834 /// marshmallows: [[u8; 2]],
2835 /// }
2836 ///
2837 /// let bytes = &mut [0xC0, 0xC0, 240, 77, 2, 3, 4, 5, 6, 7, 8][..];
2838 ///
2839 /// let (packet, suffix) = Packet::try_mut_from_prefix_with_elems(bytes, 3).unwrap();
2840 ///
2841 /// assert_eq!(packet.mug_size, 240);
2842 /// assert_eq!(packet.temperature, 77);
2843 /// assert_eq!(packet.marshmallows, [[2, 3], [4, 5], [6, 7]]);
2844 /// assert_eq!(suffix, &[8u8][..]);
2845 ///
2846 /// packet.temperature = 111;
2847 /// suffix[0] = 222;
2848 ///
2849 /// assert_eq!(bytes, [0xC0, 0xC0, 240, 111, 2, 3, 4, 5, 6, 7, 222]);
2850 ///
2851 /// // These bytes are not valid instance of `Packet`.
2852 /// let bytes = &mut [0, 1, 2, 3, 4, 5, 6, 7, 8, 77, 240, 0xC0, 0xC0][..];
2853 /// assert!(Packet::try_mut_from_prefix_with_elems(bytes, 3).is_err());
2854 /// ```
2855 ///
2856 /// Since an explicit `count` is provided, this method supports types with
2857 /// zero-sized trailing slice elements. Methods such as [`try_mut_from_prefix`]
2858 /// which do not take an explicit count do not support such types.
2859 ///
2860 /// ```
2861 /// use core::num::NonZeroU16;
2862 /// use zerocopy::*;
2863 /// # use zerocopy_derive::*;
2864 ///
2865 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2866 /// #[repr(C, packed)]
2867 /// struct ZSTy {
2868 /// leading_sized: NonZeroU16,
2869 /// trailing_dst: [()],
2870 /// }
2871 ///
2872 /// let mut src = 0xCAFEu16;
2873 /// let src = src.as_mut_bytes();
2874 /// let (zsty, _) = ZSTy::try_mut_from_prefix_with_elems(src, 42).unwrap();
2875 /// assert_eq!(zsty.trailing_dst.len(), 42);
2876 /// ```
2877 ///
2878 /// [`try_mut_from_prefix`]: TryFromBytes::try_mut_from_prefix
2879 ///
2880 #[doc = codegen_header!("h5", "try_mut_from_prefix_with_elems")]
2881 ///
2882 /// See [`TryFromBytes::try_ref_from_prefix_with_elems`](#method.try_ref_from_prefix_with_elems.codegen).
2883 #[must_use = "has no side effects"]
2884 #[inline]
2885 fn try_mut_from_prefix_with_elems(
2886 source: &mut [u8],
2887 count: usize,
2888 ) -> Result<(&mut Self, &mut [u8]), TryCastError<&mut [u8], Self>>
2889 where
2890 Self: KnownLayout<PointerMetadata = usize> + IntoBytes,
2891 {
2892 try_mut_from_prefix_suffix(source, CastType::Prefix, Some(count))
2893 }
2894
2895 /// Attempts to interpret the suffix of the given `source` as a `&mut Self`
2896 /// with a DST length equal to `count`.
2897 ///
2898 /// This method attempts to return a reference to the suffix of `source`
2899 /// interpreted as a `Self` with `count` trailing elements, and a reference
2900 /// to the preceding bytes. If the length of `source` is less than the size
2901 /// of `Self` with `count` elements, if the suffix of `source` is not
2902 /// appropriately aligned, or if the suffix of `source` does not contain a
2903 /// valid instance of `Self`, this returns `Err`. If [`Self:
2904 /// Unaligned`][self-unaligned], you can [infallibly discard the alignment
2905 /// error][ConvertError::from].
2906 ///
2907 /// [self-unaligned]: Unaligned
2908 /// [slice-dst]: KnownLayout#dynamically-sized-types
2909 ///
2910 /// # Examples
2911 ///
2912 /// ```
2913 /// # #![allow(non_camel_case_types)] // For C0::xC0
2914 /// use zerocopy::TryFromBytes;
2915 /// # use zerocopy_derive::*;
2916 ///
2917 /// // The only valid value of this type is the byte `0xC0`
2918 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2919 /// #[repr(u8)]
2920 /// enum C0 { xC0 = 0xC0 }
2921 ///
2922 /// // The only valid value of this type is the bytes `0xC0C0`.
2923 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2924 /// #[repr(C)]
2925 /// struct C0C0(C0, C0);
2926 ///
2927 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2928 /// #[repr(C, packed)]
2929 /// struct Packet {
2930 /// magic_number: C0C0,
2931 /// mug_size: u8,
2932 /// temperature: u8,
2933 /// marshmallows: [[u8; 2]],
2934 /// }
2935 ///
2936 /// let bytes = &mut [123, 0xC0, 0xC0, 240, 77, 2, 3, 4, 5, 6, 7][..];
2937 ///
2938 /// let (prefix, packet) = Packet::try_mut_from_suffix_with_elems(bytes, 3).unwrap();
2939 ///
2940 /// assert_eq!(packet.mug_size, 240);
2941 /// assert_eq!(packet.temperature, 77);
2942 /// assert_eq!(packet.marshmallows, [[2, 3], [4, 5], [6, 7]]);
2943 /// assert_eq!(prefix, &[123u8][..]);
2944 ///
2945 /// prefix[0] = 111;
2946 /// packet.temperature = 222;
2947 ///
2948 /// assert_eq!(bytes, [111, 0xC0, 0xC0, 240, 222, 2, 3, 4, 5, 6, 7]);
2949 ///
2950 /// // These bytes are not valid instance of `Packet`.
2951 /// let bytes = &mut [0, 1, 2, 3, 4, 5, 6, 7, 8, 77, 240, 0xC0, 0xC0][..];
2952 /// assert!(Packet::try_mut_from_suffix_with_elems(bytes, 3).is_err());
2953 /// ```
2954 ///
2955 /// Since an explicit `count` is provided, this method supports types with
2956 /// zero-sized trailing slice elements. Methods such as [`try_mut_from_prefix`]
2957 /// which do not take an explicit count do not support such types.
2958 ///
2959 /// ```
2960 /// use core::num::NonZeroU16;
2961 /// use zerocopy::*;
2962 /// # use zerocopy_derive::*;
2963 ///
2964 /// #[derive(TryFromBytes, IntoBytes, KnownLayout)]
2965 /// #[repr(C, packed)]
2966 /// struct ZSTy {
2967 /// leading_sized: NonZeroU16,
2968 /// trailing_dst: [()],
2969 /// }
2970 ///
2971 /// let mut src = 0xCAFEu16;
2972 /// let src = src.as_mut_bytes();
2973 /// let (_, zsty) = ZSTy::try_mut_from_suffix_with_elems(src, 42).unwrap();
2974 /// assert_eq!(zsty.trailing_dst.len(), 42);
2975 /// ```
2976 ///
2977 /// [`try_mut_from_prefix`]: TryFromBytes::try_mut_from_prefix
2978 ///
2979 #[doc = codegen_header!("h5", "try_mut_from_suffix_with_elems")]
2980 ///
2981 /// See [`TryFromBytes::try_ref_from_suffix_with_elems`](#method.try_ref_from_suffix_with_elems.codegen).
2982 #[must_use = "has no side effects"]
2983 #[inline]
2984 fn try_mut_from_suffix_with_elems(
2985 source: &mut [u8],
2986 count: usize,
2987 ) -> Result<(&mut [u8], &mut Self), TryCastError<&mut [u8], Self>>
2988 where
2989 Self: KnownLayout<PointerMetadata = usize> + IntoBytes,
2990 {
2991 try_mut_from_prefix_suffix(source, CastType::Suffix, Some(count)).map(swap)
2992 }
2993
2994 /// Attempts to read the given `source` as a `Self`.
2995 ///
2996 /// If `source.len() != size_of::<Self>()` or the bytes are not a valid
2997 /// instance of `Self`, this returns `Err`.
2998 ///
2999 /// # Examples
3000 ///
3001 /// ```
3002 /// use zerocopy::TryFromBytes;
3003 /// # use zerocopy_derive::*;
3004 ///
3005 /// // The only valid value of this type is the byte `0xC0`
3006 /// #[derive(TryFromBytes)]
3007 /// #[repr(u8)]
3008 /// enum C0 { xC0 = 0xC0 }
3009 ///
3010 /// // The only valid value of this type is the bytes `0xC0C0`.
3011 /// #[derive(TryFromBytes)]
3012 /// #[repr(C)]
3013 /// struct C0C0(C0, C0);
3014 ///
3015 /// #[derive(TryFromBytes)]
3016 /// #[repr(C)]
3017 /// struct Packet {
3018 /// magic_number: C0C0,
3019 /// mug_size: u8,
3020 /// temperature: u8,
3021 /// }
3022 ///
3023 /// let bytes = &[0xC0, 0xC0, 240, 77][..];
3024 ///
3025 /// let packet = Packet::try_read_from_bytes(bytes).unwrap();
3026 ///
3027 /// assert_eq!(packet.mug_size, 240);
3028 /// assert_eq!(packet.temperature, 77);
3029 ///
3030 /// // These bytes are not valid instance of `Packet`.
3031 /// let bytes = &mut [0x10, 0xC0, 240, 77][..];
3032 /// assert!(Packet::try_read_from_bytes(bytes).is_err());
3033 /// ```
3034 ///
3035 /// # Performance Considerations
3036 ///
3037 /// In this version of zerocopy, this method reads the `source` into a
3038 /// well-aligned stack allocation and *then* validates that the allocation
3039 /// is a valid `Self`. This ensures that validation can be performed using
3040 /// aligned reads (which carry a performance advantage over unaligned reads
3041 /// on many platforms) at the cost of an unconditional copy.
3042 ///
3043 #[doc = codegen_section!(
3044 header = "h5",
3045 bench = "try_read_from_bytes",
3046 format = "coco_static_size",
3047 )]
3048 #[must_use = "has no side effects"]
3049 #[inline]
3050 fn try_read_from_bytes(source: &[u8]) -> Result<Self, TryReadError<&[u8], Self>>
3051 where
3052 Self: Sized,
3053 {
3054 // FIXME(#2981): If `align_of::<Self>() == 1`, validate `source` in-place.
3055
3056 let candidate = match CoreMaybeUninit::<Self>::read_from_bytes(source) {
3057 Ok(candidate) => candidate,
3058 Err(e) => {
3059 return Err(TryReadError::Size(e.with_dst()));
3060 }
3061 };
3062 // SAFETY: `candidate` was copied from from `source: &[u8]`, so all of
3063 // its bytes are initialized.
3064 unsafe { try_read_from(source, candidate) }
3065 }
3066
3067 /// Attempts to read a `Self` from the prefix of the given `source`.
3068 ///
3069 /// This attempts to read a `Self` from the first `size_of::<Self>()` bytes
3070 /// of `source`, returning that `Self` and any remaining bytes. If
3071 /// `source.len() < size_of::<Self>()` or the bytes are not a valid instance
3072 /// of `Self`, it returns `Err`.
3073 ///
3074 /// # Examples
3075 ///
3076 /// ```
3077 /// use zerocopy::TryFromBytes;
3078 /// # use zerocopy_derive::*;
3079 ///
3080 /// // The only valid value of this type is the byte `0xC0`
3081 /// #[derive(TryFromBytes)]
3082 /// #[repr(u8)]
3083 /// enum C0 { xC0 = 0xC0 }
3084 ///
3085 /// // The only valid value of this type is the bytes `0xC0C0`.
3086 /// #[derive(TryFromBytes)]
3087 /// #[repr(C)]
3088 /// struct C0C0(C0, C0);
3089 ///
3090 /// #[derive(TryFromBytes)]
3091 /// #[repr(C)]
3092 /// struct Packet {
3093 /// magic_number: C0C0,
3094 /// mug_size: u8,
3095 /// temperature: u8,
3096 /// }
3097 ///
3098 /// // These are more bytes than are needed to encode a `Packet`.
3099 /// let bytes = &[0xC0, 0xC0, 240, 77, 0, 1, 2, 3, 4, 5, 6][..];
3100 ///
3101 /// let (packet, suffix) = Packet::try_read_from_prefix(bytes).unwrap();
3102 ///
3103 /// assert_eq!(packet.mug_size, 240);
3104 /// assert_eq!(packet.temperature, 77);
3105 /// assert_eq!(suffix, &[0u8, 1, 2, 3, 4, 5, 6][..]);
3106 ///
3107 /// // These bytes are not valid instance of `Packet`.
3108 /// let bytes = &[0x10, 0xC0, 240, 77, 0, 1, 2, 3, 4, 5, 6][..];
3109 /// assert!(Packet::try_read_from_prefix(bytes).is_err());
3110 /// ```
3111 ///
3112 /// # Performance Considerations
3113 ///
3114 /// In this version of zerocopy, this method reads the `source` into a
3115 /// well-aligned stack allocation and *then* validates that the allocation
3116 /// is a valid `Self`. This ensures that validation can be performed using
3117 /// aligned reads (which carry a performance advantage over unaligned reads
3118 /// on many platforms) at the cost of an unconditional copy.
3119 ///
3120 #[doc = codegen_section!(
3121 header = "h5",
3122 bench = "try_read_from_prefix",
3123 format = "coco_static_size",
3124 )]
3125 #[must_use = "has no side effects"]
3126 #[inline]
3127 fn try_read_from_prefix(source: &[u8]) -> Result<(Self, &[u8]), TryReadError<&[u8], Self>>
3128 where
3129 Self: Sized,
3130 {
3131 // FIXME(#2981): If `align_of::<Self>() == 1`, validate `source` in-place.
3132
3133 let (candidate, suffix) = match CoreMaybeUninit::<Self>::read_from_prefix(source) {
3134 Ok(candidate) => candidate,
3135 Err(e) => {
3136 return Err(TryReadError::Size(e.with_dst()));
3137 }
3138 };
3139 // SAFETY: `candidate` was copied from from `source: &[u8]`, so all of
3140 // its bytes are initialized.
3141 unsafe { try_read_from(source, candidate).map(|slf| (slf, suffix)) }
3142 }
3143
3144 /// Attempts to read a `Self` from the suffix of the given `source`.
3145 ///
3146 /// This attempts to read a `Self` from the last `size_of::<Self>()` bytes
3147 /// of `source`, returning that `Self` and any preceding bytes. If
3148 /// `source.len() < size_of::<Self>()` or the bytes are not a valid instance
3149 /// of `Self`, it returns `Err`.
3150 ///
3151 /// # Examples
3152 ///
3153 /// ```
3154 /// # #![allow(non_camel_case_types)] // For C0::xC0
3155 /// use zerocopy::TryFromBytes;
3156 /// # use zerocopy_derive::*;
3157 ///
3158 /// // The only valid value of this type is the byte `0xC0`
3159 /// #[derive(TryFromBytes)]
3160 /// #[repr(u8)]
3161 /// enum C0 { xC0 = 0xC0 }
3162 ///
3163 /// // The only valid value of this type is the bytes `0xC0C0`.
3164 /// #[derive(TryFromBytes)]
3165 /// #[repr(C)]
3166 /// struct C0C0(C0, C0);
3167 ///
3168 /// #[derive(TryFromBytes)]
3169 /// #[repr(C)]
3170 /// struct Packet {
3171 /// magic_number: C0C0,
3172 /// mug_size: u8,
3173 /// temperature: u8,
3174 /// }
3175 ///
3176 /// // These are more bytes than are needed to encode a `Packet`.
3177 /// let bytes = &[0, 1, 2, 3, 4, 5, 0xC0, 0xC0, 240, 77][..];
3178 ///
3179 /// let (prefix, packet) = Packet::try_read_from_suffix(bytes).unwrap();
3180 ///
3181 /// assert_eq!(packet.mug_size, 240);
3182 /// assert_eq!(packet.temperature, 77);
3183 /// assert_eq!(prefix, &[0u8, 1, 2, 3, 4, 5][..]);
3184 ///
3185 /// // These bytes are not valid instance of `Packet`.
3186 /// let bytes = &[0, 1, 2, 3, 4, 5, 0x10, 0xC0, 240, 77][..];
3187 /// assert!(Packet::try_read_from_suffix(bytes).is_err());
3188 /// ```
3189 ///
3190 /// # Performance Considerations
3191 ///
3192 /// In this version of zerocopy, this method reads the `source` into a
3193 /// well-aligned stack allocation and *then* validates that the allocation
3194 /// is a valid `Self`. This ensures that validation can be performed using
3195 /// aligned reads (which carry a performance advantage over unaligned reads
3196 /// on many platforms) at the cost of an unconditional copy.
3197 ///
3198 #[doc = codegen_section!(
3199 header = "h5",
3200 bench = "try_read_from_suffix",
3201 format = "coco_static_size",
3202 )]
3203 #[must_use = "has no side effects"]
3204 #[inline]
3205 fn try_read_from_suffix(source: &[u8]) -> Result<(&[u8], Self), TryReadError<&[u8], Self>>
3206 where
3207 Self: Sized,
3208 {
3209 // FIXME(#2981): If `align_of::<Self>() == 1`, validate `source` in-place.
3210
3211 let (prefix, candidate) = match CoreMaybeUninit::<Self>::read_from_suffix(source) {
3212 Ok(candidate) => candidate,
3213 Err(e) => {
3214 return Err(TryReadError::Size(e.with_dst()));
3215 }
3216 };
3217 // SAFETY: `candidate` was copied from from `source: &[u8]`, so all of
3218 // its bytes are initialized.
3219 unsafe { try_read_from(source, candidate).map(|slf| (prefix, slf)) }
3220 }
3221}
3222
3223#[inline(always)]
3224fn try_ref_from_prefix_suffix<T: TryFromBytes + KnownLayout + Immutable + ?Sized>(
3225 source: &[u8],
3226 cast_type: CastType,
3227 meta: Option<T::PointerMetadata>,
3228) -> Result<(&T, &[u8]), TryCastError<&[u8], T>> {
3229 match Ptr::from_ref(source).try_cast_into::<T, BecauseImmutable>(cast_type, meta) {
3230 Ok((source, prefix_suffix)) => {
3231 // This call may panic. If that happens, it doesn't cause any soundness
3232 // issues, as we have not generated any invalid state which we need to
3233 // fix before returning.
3234 match source.try_into_valid() {
3235 Ok(valid) => Ok((valid.as_ref(), prefix_suffix.as_ref())),
3236 Err(e) => Err(e.map_src(|src| src.as_bytes::<BecauseImmutable>().as_ref()).into()),
3237 }
3238 }
3239 Err(e) => Err(e.map_src(Ptr::as_ref).into()),
3240 }
3241}
3242
3243#[inline(always)]
3244fn try_mut_from_prefix_suffix<T: IntoBytes + TryFromBytes + KnownLayout + ?Sized>(
3245 candidate: &mut [u8],
3246 cast_type: CastType,
3247 meta: Option<T::PointerMetadata>,
3248) -> Result<(&mut T, &mut [u8]), TryCastError<&mut [u8], T>> {
3249 match Ptr::from_mut(candidate).try_cast_into::<T, BecauseExclusive>(cast_type, meta) {
3250 Ok((candidate, prefix_suffix)) => {
3251 // This call may panic. If that happens, it doesn't cause any soundness
3252 // issues, as we have not generated any invalid state which we need to
3253 // fix before returning.
3254 match candidate.try_into_valid() {
3255 Ok(valid) => Ok((valid.as_mut(), prefix_suffix.as_mut())),
3256 Err(e) => Err(e.map_src(|src| src.as_bytes().as_mut()).into()),
3257 }
3258 }
3259 Err(e) => Err(e.map_src(Ptr::as_mut).into()),
3260 }
3261}
3262
3263#[inline(always)]
3264fn swap<T, U>((t, u): (T, U)) -> (U, T) {
3265 (u, t)
3266}
3267
3268/// # Safety
3269///
3270/// All bytes of `candidate` must be initialized.
3271#[inline(always)]
3272unsafe fn try_read_from<S, T: TryFromBytes>(
3273 source: S,
3274 mut candidate: CoreMaybeUninit<T>,
3275) -> Result<T, TryReadError<S, T>> {
3276 // We use `from_mut` despite not mutating via `c_ptr` so that we don't need
3277 // to add a `T: Immutable` bound.
3278 let c_ptr = Ptr::from_mut(&mut candidate);
3279 // SAFETY: `c_ptr` has no uninitialized sub-ranges because it derived from
3280 // `candidate`, which the caller promises is entirely initialized. Since
3281 // `candidate` is a `MaybeUninit`, it has no validity requirements, and so
3282 // no values written to an `Initialized` `c_ptr` can violate its validity.
3283 // Since `c_ptr` has `Exclusive` aliasing, no mutations may happen except
3284 // via `c_ptr` so long as it is live, so we don't need to worry about the
3285 // fact that `c_ptr` may have more restricted validity than `candidate`.
3286 let c_ptr = unsafe { c_ptr.assume_validity::<invariant::Initialized>() };
3287 let mut c_ptr = c_ptr.cast::<_, crate::pointer::cast::CastSized, _>();
3288
3289 // Since we don't have `T: KnownLayout`, we hack around that by using
3290 // `Wrapping<T>`, which implements `KnownLayout` even if `T` doesn't.
3291 //
3292 // This call may panic. If that happens, it doesn't cause any soundness
3293 // issues, as we have not generated any invalid state which we need to fix
3294 // before returning.
3295 if !Wrapping::<T>::is_bit_valid(c_ptr.reborrow_shared().forget_aligned()) {
3296 return Err(ValidityError::new(source).into());
3297 }
3298
3299 fn _assert_same_size_and_validity<T>()
3300 where
3301 Wrapping<T>: pointer::TransmuteFrom<T, invariant::Valid, invariant::Valid>,
3302 T: pointer::TransmuteFrom<Wrapping<T>, invariant::Valid, invariant::Valid>,
3303 {
3304 }
3305
3306 _assert_same_size_and_validity::<T>();
3307
3308 // SAFETY: We just validated that `candidate` contains a valid
3309 // `Wrapping<T>`, which has the same size and bit validity as `T`, as
3310 // guaranteed by the preceding type assertion.
3311 Ok(unsafe { candidate.assume_init() })
3312}
3313
3314/// Types for which a sequence of `0` bytes is a valid instance.
3315///
3316/// Any memory region of the appropriate length which is guaranteed to contain
3317/// only zero bytes can be viewed as any `FromZeros` type with no runtime
3318/// overhead. This is useful whenever memory is known to be in a zeroed state,
3319/// such memory returned from some allocation routines.
3320///
3321/// # Warning: Padding bytes
3322///
3323/// Note that, when a value is moved or copied, only the non-padding bytes of
3324/// that value are guaranteed to be preserved. It is unsound to assume that
3325/// values written to padding bytes are preserved after a move or copy. For more
3326/// details, see the [`FromBytes` docs][frombytes-warning-padding-bytes].
3327///
3328/// [frombytes-warning-padding-bytes]: FromBytes#warning-padding-bytes
3329///
3330/// # Implementation
3331///
3332/// **Do not implement this trait yourself!** Instead, use
3333/// [`#[derive(FromZeros)]`][derive]; e.g.:
3334///
3335/// ```
3336/// # use zerocopy_derive::{FromZeros, Immutable};
3337/// #[derive(FromZeros)]
3338/// struct MyStruct {
3339/// # /*
3340/// ...
3341/// # */
3342/// }
3343///
3344/// #[derive(FromZeros)]
3345/// #[repr(u8)]
3346/// enum MyEnum {
3347/// # Variant0,
3348/// # /*
3349/// ...
3350/// # */
3351/// }
3352///
3353/// #[derive(FromZeros, Immutable)]
3354/// union MyUnion {
3355/// # variant: u8,
3356/// # /*
3357/// ...
3358/// # */
3359/// }
3360/// ```
3361///
3362/// This derive performs a sophisticated, compile-time safety analysis to
3363/// determine whether a type is `FromZeros`.
3364///
3365/// # Safety
3366///
3367/// *This section describes what is required in order for `T: FromZeros`, and
3368/// what unsafe code may assume of such types. If you don't plan on implementing
3369/// `FromZeros` manually, and you don't plan on writing unsafe code that
3370/// operates on `FromZeros` types, then you don't need to read this section.*
3371///
3372/// If `T: FromZeros`, then unsafe code may assume that it is sound to produce a
3373/// `T` whose bytes are all initialized to zero. If a type is marked as
3374/// `FromZeros` which violates this contract, it may cause undefined behavior.
3375///
3376/// `#[derive(FromZeros)]` only permits [types which satisfy these
3377/// requirements][derive-analysis].
3378///
3379#[cfg_attr(
3380 feature = "derive",
3381 doc = "[derive]: zerocopy_derive::FromZeros",
3382 doc = "[derive-analysis]: zerocopy_derive::FromZeros#analysis"
3383)]
3384#[cfg_attr(
3385 not(feature = "derive"),
3386 doc = concat!("[derive]: https://docs.rs/zerocopy/", env!("CARGO_PKG_VERSION"), "/zerocopy/derive.FromZeros.html"),
3387 doc = concat!("[derive-analysis]: https://docs.rs/zerocopy/", env!("CARGO_PKG_VERSION"), "/zerocopy/derive.FromZeros.html#analysis"),
3388)]
3389#[cfg_attr(
3390 not(no_zerocopy_diagnostic_on_unimplemented_1_78_0),
3391 diagnostic::on_unimplemented(note = "Consider adding `#[derive(FromZeros)]` to `{Self}`")
3392)]
3393pub unsafe trait FromZeros: TryFromBytes {
3394 // The `Self: Sized` bound makes it so that `FromZeros` is still object
3395 // safe.
3396 #[doc(hidden)]
3397 fn only_derive_is_allowed_to_implement_this_trait()
3398 where
3399 Self: Sized;
3400
3401 /// Overwrites `self` with zeros.
3402 ///
3403 /// Sets every byte in `self` to 0. While this is similar to doing `*self =
3404 /// Self::new_zeroed()`, it differs in that `zero` does not semantically
3405 /// drop the current value and replace it with a new one — it simply
3406 /// modifies the bytes of the existing value.
3407 ///
3408 /// # Examples
3409 ///
3410 /// ```
3411 /// # use zerocopy::FromZeros;
3412 /// # use zerocopy_derive::*;
3413 /// #
3414 /// #[derive(FromZeros)]
3415 /// #[repr(C)]
3416 /// struct PacketHeader {
3417 /// src_port: [u8; 2],
3418 /// dst_port: [u8; 2],
3419 /// length: [u8; 2],
3420 /// checksum: [u8; 2],
3421 /// }
3422 ///
3423 /// let mut header = PacketHeader {
3424 /// src_port: 100u16.to_be_bytes(),
3425 /// dst_port: 200u16.to_be_bytes(),
3426 /// length: 300u16.to_be_bytes(),
3427 /// checksum: 400u16.to_be_bytes(),
3428 /// };
3429 ///
3430 /// header.zero();
3431 ///
3432 /// assert_eq!(header.src_port, [0, 0]);
3433 /// assert_eq!(header.dst_port, [0, 0]);
3434 /// assert_eq!(header.length, [0, 0]);
3435 /// assert_eq!(header.checksum, [0, 0]);
3436 /// ```
3437 #[inline(always)]
3438 fn zero(&mut self) {
3439 let slf: *mut Self = self;
3440 let len = mem::size_of_val(self);
3441 // SAFETY:
3442 // - `self` is guaranteed by the type system to be valid for writes of
3443 // size `size_of_val(self)`.
3444 // - `u8`'s alignment is 1, and thus `self` is guaranteed to be aligned
3445 // as required by `u8`.
3446 // - Since `Self: FromZeros`, the all-zeros instance is a valid instance
3447 // of `Self.`
3448 //
3449 // FIXME(#429): Add references to docs and quotes.
3450 unsafe { ptr::write_bytes(slf.cast::<u8>(), 0, len) };
3451 }
3452
3453 /// Creates an instance of `Self` from zeroed bytes.
3454 ///
3455 /// # Examples
3456 ///
3457 /// ```
3458 /// # use zerocopy::FromZeros;
3459 /// # use zerocopy_derive::*;
3460 /// #
3461 /// #[derive(FromZeros)]
3462 /// #[repr(C)]
3463 /// struct PacketHeader {
3464 /// src_port: [u8; 2],
3465 /// dst_port: [u8; 2],
3466 /// length: [u8; 2],
3467 /// checksum: [u8; 2],
3468 /// }
3469 ///
3470 /// let header: PacketHeader = FromZeros::new_zeroed();
3471 ///
3472 /// assert_eq!(header.src_port, [0, 0]);
3473 /// assert_eq!(header.dst_port, [0, 0]);
3474 /// assert_eq!(header.length, [0, 0]);
3475 /// assert_eq!(header.checksum, [0, 0]);
3476 /// ```
3477 #[must_use = "has no side effects"]
3478 #[inline(always)]
3479 fn new_zeroed() -> Self
3480 where
3481 Self: Sized,
3482 {
3483 // SAFETY: `FromZeros` says that the all-zeros bit pattern is legal.
3484 unsafe { mem::zeroed() }
3485 }
3486
3487 /// Creates a `Box<Self>` from zeroed bytes.
3488 ///
3489 /// This function is useful for allocating large values on the heap and
3490 /// zero-initializing them, without ever creating a temporary instance of
3491 /// `Self` on the stack. For example, `<[u8; 1048576]>::new_box_zeroed()`
3492 /// will allocate `[u8; 1048576]` directly on the heap; it does not require
3493 /// storing `[u8; 1048576]` in a temporary variable on the stack.
3494 ///
3495 /// On systems that use a heap implementation that supports allocating from
3496 /// pre-zeroed memory, using `new_box_zeroed` (or related functions) may
3497 /// have performance benefits.
3498 ///
3499 /// # Errors
3500 ///
3501 /// Returns an error on allocation failure. Allocation failure is guaranteed
3502 /// never to cause a panic or an abort.
3503 #[must_use = "has no side effects (other than allocation)"]
3504 #[cfg(any(feature = "alloc", test))]
3505 #[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
3506 #[inline]
3507 fn new_box_zeroed() -> Result<Box<Self>, AllocError>
3508 where
3509 Self: Sized,
3510 {
3511 // If `T` is a ZST, then return a proper boxed instance of it. There is
3512 // no allocation, but `Box` does require a correct dangling pointer.
3513 let layout = Layout::new::<Self>();
3514 if layout.size() == 0 {
3515 // Construct the `Box` from a dangling pointer to avoid calling
3516 // `Self::new_zeroed`. This ensures that stack space is never
3517 // allocated for `Self` even on lower opt-levels where this branch
3518 // might not get optimized out.
3519
3520 // SAFETY: Per [1], when `T` is a ZST, `Box<T>`'s only validity
3521 // requirements are that the pointer is non-null and sufficiently
3522 // aligned. Per [2], `NonNull::dangling` produces a pointer which
3523 // is sufficiently aligned. Since the produced pointer is a
3524 // `NonNull`, it is non-null.
3525 //
3526 // [1] Per https://doc.rust-lang.org/1.81.0/std/boxed/index.html#memory-layout:
3527 //
3528 // For zero-sized values, the `Box` pointer has to be non-null and sufficiently aligned.
3529 //
3530 // [2] Per https://doc.rust-lang.org/std/ptr/struct.NonNull.html#method.dangling:
3531 //
3532 // Creates a new `NonNull` that is dangling, but well-aligned.
3533 return Ok(unsafe { Box::from_raw(NonNull::dangling().as_ptr()) });
3534 }
3535
3536 // FIXME(#429): Add a "SAFETY" comment and remove this `allow`.
3537 #[allow(clippy::undocumented_unsafe_blocks)]
3538 let ptr = unsafe { alloc::alloc::alloc_zeroed(layout).cast::<Self>() };
3539 if ptr.is_null() {
3540 return Err(AllocError);
3541 }
3542 // FIXME(#429): Add a "SAFETY" comment and remove this `allow`.
3543 #[allow(clippy::undocumented_unsafe_blocks)]
3544 Ok(unsafe { Box::from_raw(ptr) })
3545 }
3546
3547 /// Creates a `Box<[Self]>` (a boxed slice) from zeroed bytes.
3548 ///
3549 /// This function is useful for allocating large values of `[Self]` on the
3550 /// heap and zero-initializing them, without ever creating a temporary
3551 /// instance of `[Self; _]` on the stack. For example,
3552 /// `u8::new_box_slice_zeroed(1048576)` will allocate the slice directly on
3553 /// the heap; it does not require storing the slice on the stack.
3554 ///
3555 /// On systems that use a heap implementation that supports allocating from
3556 /// pre-zeroed memory, using `new_box_slice_zeroed` may have performance
3557 /// benefits.
3558 ///
3559 /// If `Self` is a zero-sized type, then this function will return a
3560 /// `Box<[Self]>` that has the correct `len`. Such a box cannot contain any
3561 /// actual information, but its `len()` property will report the correct
3562 /// value.
3563 ///
3564 /// # Errors
3565 ///
3566 /// Returns an error on allocation failure. Allocation failure is
3567 /// guaranteed never to cause a panic or an abort.
3568 #[must_use = "has no side effects (other than allocation)"]
3569 #[cfg(feature = "alloc")]
3570 #[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
3571 #[inline]
3572 fn new_box_zeroed_with_elems(count: usize) -> Result<Box<Self>, AllocError>
3573 where
3574 Self: KnownLayout<PointerMetadata = usize>,
3575 {
3576 // SAFETY: `alloc::alloc::alloc_zeroed` is a valid argument of
3577 // `new_box`. The referent of the pointer returned by `alloc_zeroed`
3578 // (and, consequently, the `Box` derived from it) is a valid instance of
3579 // `Self`, because `Self` is `FromZeros`.
3580 unsafe { crate::util::new_box(count, alloc::alloc::alloc_zeroed) }
3581 }
3582
3583 #[deprecated(since = "0.8.0", note = "renamed to `FromZeros::new_box_zeroed_with_elems`")]
3584 #[doc(hidden)]
3585 #[cfg(feature = "alloc")]
3586 #[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
3587 #[must_use = "has no side effects (other than allocation)"]
3588 #[inline(always)]
3589 fn new_box_slice_zeroed(len: usize) -> Result<Box<[Self]>, AllocError>
3590 where
3591 Self: Sized,
3592 {
3593 <[Self]>::new_box_zeroed_with_elems(len)
3594 }
3595
3596 /// Creates a `Vec<Self>` from zeroed bytes.
3597 ///
3598 /// This function is useful for allocating large values of `Vec`s and
3599 /// zero-initializing them, without ever creating a temporary instance of
3600 /// `[Self; _]` (or many temporary instances of `Self`) on the stack. For
3601 /// example, `u8::new_vec_zeroed(1048576)` will allocate directly on the
3602 /// heap; it does not require storing intermediate values on the stack.
3603 ///
3604 /// On systems that use a heap implementation that supports allocating from
3605 /// pre-zeroed memory, using `new_vec_zeroed` may have performance benefits.
3606 ///
3607 /// If `Self` is a zero-sized type, then this function will return a
3608 /// `Vec<Self>` that has the correct `len`. Such a `Vec` cannot contain any
3609 /// actual information, but its `len()` property will report the correct
3610 /// value.
3611 ///
3612 /// # Errors
3613 ///
3614 /// Returns an error on allocation failure. Allocation failure is
3615 /// guaranteed never to cause a panic or an abort.
3616 #[must_use = "has no side effects (other than allocation)"]
3617 #[cfg(feature = "alloc")]
3618 #[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
3619 #[inline(always)]
3620 fn new_vec_zeroed(len: usize) -> Result<Vec<Self>, AllocError>
3621 where
3622 Self: Sized,
3623 {
3624 <[Self]>::new_box_zeroed_with_elems(len).map(Into::into)
3625 }
3626
3627 /// Extends a `Vec<Self>` by pushing `additional` new items onto the end of
3628 /// the vector. The new items are initialized with zeros.
3629 #[cfg(not(no_zerocopy_panic_in_const_and_vec_try_reserve_1_57_0))]
3630 #[cfg(feature = "alloc")]
3631 #[cfg_attr(doc_cfg, doc(cfg(all(rust = "1.57.0", feature = "alloc"))))]
3632 #[inline(always)]
3633 fn extend_vec_zeroed(v: &mut Vec<Self>, additional: usize) -> Result<(), AllocError>
3634 where
3635 Self: Sized,
3636 {
3637 // PANICS: We pass `v.len()` for `position`, so the `position > v.len()`
3638 // panic condition is not satisfied.
3639 <Self as FromZeros>::insert_vec_zeroed(v, v.len(), additional)
3640 }
3641
3642 /// Inserts `additional` new items into `Vec<Self>` at `position`. The new
3643 /// items are initialized with zeros.
3644 ///
3645 /// # Panics
3646 ///
3647 /// Panics if `position > v.len()`.
3648 #[cfg(not(no_zerocopy_panic_in_const_and_vec_try_reserve_1_57_0))]
3649 #[cfg(feature = "alloc")]
3650 #[cfg_attr(doc_cfg, doc(cfg(all(rust = "1.57.0", feature = "alloc"))))]
3651 #[inline]
3652 fn insert_vec_zeroed(
3653 v: &mut Vec<Self>,
3654 position: usize,
3655 additional: usize,
3656 ) -> Result<(), AllocError>
3657 where
3658 Self: Sized,
3659 {
3660 assert!(position <= v.len());
3661 // We only conditionally compile on versions on which `try_reserve` is
3662 // stable; the Clippy lint is a false positive.
3663 v.try_reserve(additional).map_err(|_| AllocError)?;
3664 // SAFETY: The `try_reserve` call guarantees that these cannot overflow:
3665 // * `ptr.add(position)`
3666 // * `position + additional`
3667 // * `v.len() + additional`
3668 //
3669 // `v.len() - position` cannot overflow because we asserted that
3670 // `position <= v.len()`.
3671 #[allow(clippy::multiple_unsafe_ops_per_block)]
3672 unsafe {
3673 // This is a potentially overlapping copy.
3674 let ptr = v.as_mut_ptr();
3675 #[allow(clippy::arithmetic_side_effects)]
3676 ptr.add(position).copy_to(ptr.add(position + additional), v.len() - position);
3677 ptr.add(position).write_bytes(0, additional);
3678 #[allow(clippy::arithmetic_side_effects)]
3679 v.set_len(v.len() + additional);
3680 }
3681
3682 Ok(())
3683 }
3684}
3685
3686/// Analyzes whether a type is [`FromBytes`].
3687///
3688/// This derive analyzes, at compile time, whether the annotated type satisfies
3689/// the [safety conditions] of `FromBytes` and implements `FromBytes` and its
3690/// supertraits if it is sound to do so. This derive can be applied to structs,
3691/// enums, and unions;
3692/// e.g.:
3693///
3694/// ```
3695/// # use zerocopy_derive::{FromBytes, FromZeros, Immutable};
3696/// #[derive(FromBytes)]
3697/// struct MyStruct {
3698/// # /*
3699/// ...
3700/// # */
3701/// }
3702///
3703/// #[derive(FromBytes)]
3704/// #[repr(u8)]
3705/// enum MyEnum {
3706/// # V00, V01, V02, V03, V04, V05, V06, V07, V08, V09, V0A, V0B, V0C, V0D, V0E,
3707/// # V0F, V10, V11, V12, V13, V14, V15, V16, V17, V18, V19, V1A, V1B, V1C, V1D,
3708/// # V1E, V1F, V20, V21, V22, V23, V24, V25, V26, V27, V28, V29, V2A, V2B, V2C,
3709/// # V2D, V2E, V2F, V30, V31, V32, V33, V34, V35, V36, V37, V38, V39, V3A, V3B,
3710/// # V3C, V3D, V3E, V3F, V40, V41, V42, V43, V44, V45, V46, V47, V48, V49, V4A,
3711/// # V4B, V4C, V4D, V4E, V4F, V50, V51, V52, V53, V54, V55, V56, V57, V58, V59,
3712/// # V5A, V5B, V5C, V5D, V5E, V5F, V60, V61, V62, V63, V64, V65, V66, V67, V68,
3713/// # V69, V6A, V6B, V6C, V6D, V6E, V6F, V70, V71, V72, V73, V74, V75, V76, V77,
3714/// # V78, V79, V7A, V7B, V7C, V7D, V7E, V7F, V80, V81, V82, V83, V84, V85, V86,
3715/// # V87, V88, V89, V8A, V8B, V8C, V8D, V8E, V8F, V90, V91, V92, V93, V94, V95,
3716/// # V96, V97, V98, V99, V9A, V9B, V9C, V9D, V9E, V9F, VA0, VA1, VA2, VA3, VA4,
3717/// # VA5, VA6, VA7, VA8, VA9, VAA, VAB, VAC, VAD, VAE, VAF, VB0, VB1, VB2, VB3,
3718/// # VB4, VB5, VB6, VB7, VB8, VB9, VBA, VBB, VBC, VBD, VBE, VBF, VC0, VC1, VC2,
3719/// # VC3, VC4, VC5, VC6, VC7, VC8, VC9, VCA, VCB, VCC, VCD, VCE, VCF, VD0, VD1,
3720/// # VD2, VD3, VD4, VD5, VD6, VD7, VD8, VD9, VDA, VDB, VDC, VDD, VDE, VDF, VE0,
3721/// # VE1, VE2, VE3, VE4, VE5, VE6, VE7, VE8, VE9, VEA, VEB, VEC, VED, VEE, VEF,
3722/// # VF0, VF1, VF2, VF3, VF4, VF5, VF6, VF7, VF8, VF9, VFA, VFB, VFC, VFD, VFE,
3723/// # VFF,
3724/// # /*
3725/// ...
3726/// # */
3727/// }
3728///
3729/// #[derive(FromBytes, Immutable)]
3730/// union MyUnion {
3731/// # variant: u8,
3732/// # /*
3733/// ...
3734/// # */
3735/// }
3736/// ```
3737///
3738/// [safety conditions]: trait@FromBytes#safety
3739///
3740/// # Analysis
3741///
3742/// *This section describes, roughly, the analysis performed by this derive to
3743/// determine whether it is sound to implement `FromBytes` for a given type.
3744/// Unless you are modifying the implementation of this derive, or attempting to
3745/// manually implement `FromBytes` for a type yourself, you don't need to read
3746/// this section.*
3747///
3748/// If a type has the following properties, then this derive can implement
3749/// `FromBytes` for that type:
3750///
3751/// - If the type is a struct, all of its fields must be `FromBytes`.
3752/// - If the type is an enum:
3753/// - It must have a defined representation which is one of `u8`, `u16`, `i8`,
3754/// or `i16`.
3755/// - The maximum number of discriminants must be used (so that every possible
3756/// bit pattern is a valid one).
3757/// - Its fields must be `FromBytes`.
3758///
3759/// This analysis is subject to change. Unsafe code may *only* rely on the
3760/// documented [safety conditions] of `FromBytes`, and must *not* rely on the
3761/// implementation details of this derive.
3762///
3763/// ## Why isn't an explicit representation required for structs?
3764///
3765/// Neither this derive, nor the [safety conditions] of `FromBytes`, requires
3766/// that structs are marked with `#[repr(C)]`.
3767///
3768/// Per the [Rust reference](reference),
3769///
3770/// > The representation of a type can change the padding between fields, but
3771/// > does not change the layout of the fields themselves.
3772///
3773/// [reference]: https://doc.rust-lang.org/reference/type-layout.html#representations
3774///
3775/// Since the layout of structs only consists of padding bytes and field bytes,
3776/// a struct is soundly `FromBytes` if:
3777/// 1. its padding is soundly `FromBytes`, and
3778/// 2. its fields are soundly `FromBytes`.
3779///
3780/// The answer to the first question is always yes: padding bytes do not have
3781/// any validity constraints. A [discussion] of this question in the Unsafe Code
3782/// Guidelines Working Group concluded that it would be virtually unimaginable
3783/// for future versions of rustc to add validity constraints to padding bytes.
3784///
3785/// [discussion]: https://github.com/rust-lang/unsafe-code-guidelines/issues/174
3786///
3787/// Whether a struct is soundly `FromBytes` therefore solely depends on whether
3788/// its fields are `FromBytes`.
3789#[cfg(any(feature = "derive", test))]
3790#[cfg_attr(doc_cfg, doc(cfg(feature = "derive")))]
3791pub use zerocopy_derive::FromBytes;
3792
3793/// Types for which any bit pattern is valid.
3794///
3795/// Any memory region of the appropriate length which contains initialized bytes
3796/// can be viewed as any `FromBytes` type with no runtime overhead. This is
3797/// useful for efficiently parsing bytes as structured data.
3798///
3799/// # Warning: Padding bytes
3800///
3801/// Note that, when a value is moved or copied, only the non-padding bytes of
3802/// that value are guaranteed to be preserved. It is unsound to assume that
3803/// values written to padding bytes are preserved after a move or copy. For
3804/// example, the following is unsound:
3805///
3806/// ```rust,no_run
3807/// use core::mem::{size_of, transmute};
3808/// use zerocopy::FromZeros;
3809/// # use zerocopy_derive::*;
3810///
3811/// // Assume `Foo` is a type with padding bytes.
3812/// #[derive(FromZeros, Default)]
3813/// struct Foo {
3814/// # /*
3815/// ...
3816/// # */
3817/// }
3818///
3819/// let mut foo: Foo = Foo::default();
3820/// FromZeros::zero(&mut foo);
3821/// // UNSOUND: Although `FromZeros::zero` writes zeros to all bytes of `foo`,
3822/// // those writes are not guaranteed to be preserved in padding bytes when
3823/// // `foo` is moved, so this may expose padding bytes as `u8`s.
3824/// let foo_bytes: [u8; size_of::<Foo>()] = unsafe { transmute(foo) };
3825/// ```
3826///
3827/// # Implementation
3828///
3829/// **Do not implement this trait yourself!** Instead, use
3830/// [`#[derive(FromBytes)]`][derive]; e.g.:
3831///
3832/// ```
3833/// # use zerocopy_derive::{FromBytes, Immutable};
3834/// #[derive(FromBytes)]
3835/// struct MyStruct {
3836/// # /*
3837/// ...
3838/// # */
3839/// }
3840///
3841/// #[derive(FromBytes)]
3842/// #[repr(u8)]
3843/// enum MyEnum {
3844/// # V00, V01, V02, V03, V04, V05, V06, V07, V08, V09, V0A, V0B, V0C, V0D, V0E,
3845/// # V0F, V10, V11, V12, V13, V14, V15, V16, V17, V18, V19, V1A, V1B, V1C, V1D,
3846/// # V1E, V1F, V20, V21, V22, V23, V24, V25, V26, V27, V28, V29, V2A, V2B, V2C,
3847/// # V2D, V2E, V2F, V30, V31, V32, V33, V34, V35, V36, V37, V38, V39, V3A, V3B,
3848/// # V3C, V3D, V3E, V3F, V40, V41, V42, V43, V44, V45, V46, V47, V48, V49, V4A,
3849/// # V4B, V4C, V4D, V4E, V4F, V50, V51, V52, V53, V54, V55, V56, V57, V58, V59,
3850/// # V5A, V5B, V5C, V5D, V5E, V5F, V60, V61, V62, V63, V64, V65, V66, V67, V68,
3851/// # V69, V6A, V6B, V6C, V6D, V6E, V6F, V70, V71, V72, V73, V74, V75, V76, V77,
3852/// # V78, V79, V7A, V7B, V7C, V7D, V7E, V7F, V80, V81, V82, V83, V84, V85, V86,
3853/// # V87, V88, V89, V8A, V8B, V8C, V8D, V8E, V8F, V90, V91, V92, V93, V94, V95,
3854/// # V96, V97, V98, V99, V9A, V9B, V9C, V9D, V9E, V9F, VA0, VA1, VA2, VA3, VA4,
3855/// # VA5, VA6, VA7, VA8, VA9, VAA, VAB, VAC, VAD, VAE, VAF, VB0, VB1, VB2, VB3,
3856/// # VB4, VB5, VB6, VB7, VB8, VB9, VBA, VBB, VBC, VBD, VBE, VBF, VC0, VC1, VC2,
3857/// # VC3, VC4, VC5, VC6, VC7, VC8, VC9, VCA, VCB, VCC, VCD, VCE, VCF, VD0, VD1,
3858/// # VD2, VD3, VD4, VD5, VD6, VD7, VD8, VD9, VDA, VDB, VDC, VDD, VDE, VDF, VE0,
3859/// # VE1, VE2, VE3, VE4, VE5, VE6, VE7, VE8, VE9, VEA, VEB, VEC, VED, VEE, VEF,
3860/// # VF0, VF1, VF2, VF3, VF4, VF5, VF6, VF7, VF8, VF9, VFA, VFB, VFC, VFD, VFE,
3861/// # VFF,
3862/// # /*
3863/// ...
3864/// # */
3865/// }
3866///
3867/// #[derive(FromBytes, Immutable)]
3868/// union MyUnion {
3869/// # variant: u8,
3870/// # /*
3871/// ...
3872/// # */
3873/// }
3874/// ```
3875///
3876/// This derive performs a sophisticated, compile-time safety analysis to
3877/// determine whether a type is `FromBytes`.
3878///
3879/// # Safety
3880///
3881/// *This section describes what is required in order for `T: FromBytes`, and
3882/// what unsafe code may assume of such types. If you don't plan on implementing
3883/// `FromBytes` manually, and you don't plan on writing unsafe code that
3884/// operates on `FromBytes` types, then you don't need to read this section.*
3885///
3886/// If `T: FromBytes`, then unsafe code may assume that it is sound to produce a
3887/// `T` whose bytes are initialized to any sequence of valid `u8`s (in other
3888/// words, any byte value which is not uninitialized). If a type is marked as
3889/// `FromBytes` which violates this contract, it may cause undefined behavior.
3890///
3891/// `#[derive(FromBytes)]` only permits [types which satisfy these
3892/// requirements][derive-analysis].
3893///
3894#[cfg_attr(
3895 feature = "derive",
3896 doc = "[derive]: zerocopy_derive::FromBytes",
3897 doc = "[derive-analysis]: zerocopy_derive::FromBytes#analysis"
3898)]
3899#[cfg_attr(
3900 not(feature = "derive"),
3901 doc = concat!("[derive]: https://docs.rs/zerocopy/", env!("CARGO_PKG_VERSION"), "/zerocopy/derive.FromBytes.html"),
3902 doc = concat!("[derive-analysis]: https://docs.rs/zerocopy/", env!("CARGO_PKG_VERSION"), "/zerocopy/derive.FromBytes.html#analysis"),
3903)]
3904#[cfg_attr(
3905 not(no_zerocopy_diagnostic_on_unimplemented_1_78_0),
3906 diagnostic::on_unimplemented(note = "Consider adding `#[derive(FromBytes)]` to `{Self}`")
3907)]
3908pub unsafe trait FromBytes: FromZeros {
3909 // The `Self: Sized` bound makes it so that `FromBytes` is still object
3910 // safe.
3911 #[doc(hidden)]
3912 fn only_derive_is_allowed_to_implement_this_trait()
3913 where
3914 Self: Sized;
3915
3916 /// Interprets the given `source` as a `&Self`.
3917 ///
3918 /// This method attempts to return a reference to `source` interpreted as a
3919 /// `Self`. If the length of `source` is not a [valid size of
3920 /// `Self`][valid-size], or if `source` is not appropriately aligned, this
3921 /// returns `Err`. If [`Self: Unaligned`][self-unaligned], you can
3922 /// [infallibly discard the alignment error][size-error-from].
3923 ///
3924 /// `Self` may be a sized type, a slice, or a [slice DST][slice-dst].
3925 ///
3926 /// [valid-size]: crate::KnownLayout#what-is-a-valid-size
3927 /// [self-unaligned]: Unaligned
3928 /// [size-error-from]: error/struct.SizeError.html#method.from-1
3929 /// [slice-dst]: KnownLayout#dynamically-sized-types
3930 ///
3931 /// # Compile-Time Assertions
3932 ///
3933 /// This method cannot yet be used on unsized types whose dynamically-sized
3934 /// component is zero-sized. Attempting to use this method on such types
3935 /// results in a compile-time assertion error; e.g.:
3936 ///
3937 /// ```compile_fail,E0080
3938 /// use zerocopy::*;
3939 /// # use zerocopy_derive::*;
3940 ///
3941 /// #[derive(FromBytes, Immutable, KnownLayout)]
3942 /// #[repr(C)]
3943 /// struct ZSTy {
3944 /// leading_sized: u16,
3945 /// trailing_dst: [()],
3946 /// }
3947 ///
3948 /// let _ = ZSTy::ref_from_bytes(0u16.as_bytes()); // âš Compile Error!
3949 /// ```
3950 ///
3951 /// # Examples
3952 ///
3953 /// ```
3954 /// use zerocopy::FromBytes;
3955 /// # use zerocopy_derive::*;
3956 ///
3957 /// #[derive(FromBytes, KnownLayout, Immutable)]
3958 /// #[repr(C)]
3959 /// struct PacketHeader {
3960 /// src_port: [u8; 2],
3961 /// dst_port: [u8; 2],
3962 /// length: [u8; 2],
3963 /// checksum: [u8; 2],
3964 /// }
3965 ///
3966 /// #[derive(FromBytes, KnownLayout, Immutable)]
3967 /// #[repr(C)]
3968 /// struct Packet {
3969 /// header: PacketHeader,
3970 /// body: [u8],
3971 /// }
3972 ///
3973 /// // These bytes encode a `Packet`.
3974 /// let bytes = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11][..];
3975 ///
3976 /// let packet = Packet::ref_from_bytes(bytes).unwrap();
3977 ///
3978 /// assert_eq!(packet.header.src_port, [0, 1]);
3979 /// assert_eq!(packet.header.dst_port, [2, 3]);
3980 /// assert_eq!(packet.header.length, [4, 5]);
3981 /// assert_eq!(packet.header.checksum, [6, 7]);
3982 /// assert_eq!(packet.body, [8, 9, 10, 11]);
3983 /// ```
3984 ///
3985 #[doc = codegen_section!(
3986 header = "h5",
3987 bench = "ref_from_bytes",
3988 format = "coco",
3989 arity = 3,
3990 [
3991 open
3992 @index 1
3993 @title "Sized"
3994 @variant "static_size"
3995 ],
3996 [
3997 @index 2
3998 @title "Unsized"
3999 @variant "dynamic_size"
4000 ],
4001 [
4002 @index 3
4003 @title "Dynamically Padded"
4004 @variant "dynamic_padding"
4005 ]
4006 )]
4007 #[must_use = "has no side effects"]
4008 #[inline]
4009 fn ref_from_bytes(source: &[u8]) -> Result<&Self, CastError<&[u8], Self>>
4010 where
4011 Self: KnownLayout + Immutable,
4012 {
4013 static_assert_dst_is_not_zst!(Self);
4014 match Ptr::from_ref(source).try_cast_into_no_leftover::<_, BecauseImmutable>(None) {
4015 Ok(ptr) => Ok(ptr.recall_validity().as_ref()),
4016 Err(err) => Err(err.map_src(|src| src.as_ref())),
4017 }
4018 }
4019
4020 /// Interprets the prefix of the given `source` as a `&Self` without
4021 /// copying.
4022 ///
4023 /// This method computes the [largest possible size of `Self`][valid-size]
4024 /// that can fit in the leading bytes of `source`, then attempts to return
4025 /// both a reference to those bytes interpreted as a `Self`, and a reference
4026 /// to the remaining bytes. If there are insufficient bytes, or if `source`
4027 /// is not appropriately aligned, this returns `Err`. If [`Self:
4028 /// Unaligned`][self-unaligned], you can [infallibly discard the alignment
4029 /// error][size-error-from].
4030 ///
4031 /// `Self` may be a sized type, a slice, or a [slice DST][slice-dst].
4032 ///
4033 /// [valid-size]: crate::KnownLayout#what-is-a-valid-size
4034 /// [self-unaligned]: Unaligned
4035 /// [size-error-from]: error/struct.SizeError.html#method.from-1
4036 /// [slice-dst]: KnownLayout#dynamically-sized-types
4037 ///
4038 /// # Compile-Time Assertions
4039 ///
4040 /// This method cannot yet be used on unsized types whose dynamically-sized
4041 /// component is zero-sized. See [`ref_from_prefix_with_elems`], which does
4042 /// support such types. Attempting to use this method on such types results
4043 /// in a compile-time assertion error; e.g.:
4044 ///
4045 /// ```compile_fail,E0080
4046 /// use zerocopy::*;
4047 /// # use zerocopy_derive::*;
4048 ///
4049 /// #[derive(FromBytes, Immutable, KnownLayout)]
4050 /// #[repr(C)]
4051 /// struct ZSTy {
4052 /// leading_sized: u16,
4053 /// trailing_dst: [()],
4054 /// }
4055 ///
4056 /// let _ = ZSTy::ref_from_prefix(0u16.as_bytes()); // âš Compile Error!
4057 /// ```
4058 ///
4059 /// [`ref_from_prefix_with_elems`]: FromBytes::ref_from_prefix_with_elems
4060 ///
4061 /// # Examples
4062 ///
4063 /// ```
4064 /// use zerocopy::FromBytes;
4065 /// # use zerocopy_derive::*;
4066 ///
4067 /// #[derive(FromBytes, KnownLayout, Immutable)]
4068 /// #[repr(C)]
4069 /// struct PacketHeader {
4070 /// src_port: [u8; 2],
4071 /// dst_port: [u8; 2],
4072 /// length: [u8; 2],
4073 /// checksum: [u8; 2],
4074 /// }
4075 ///
4076 /// #[derive(FromBytes, KnownLayout, Immutable)]
4077 /// #[repr(C)]
4078 /// struct Packet {
4079 /// header: PacketHeader,
4080 /// body: [[u8; 2]],
4081 /// }
4082 ///
4083 /// // These are more bytes than are needed to encode a `Packet`.
4084 /// let bytes = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14][..];
4085 ///
4086 /// let (packet, suffix) = Packet::ref_from_prefix(bytes).unwrap();
4087 ///
4088 /// assert_eq!(packet.header.src_port, [0, 1]);
4089 /// assert_eq!(packet.header.dst_port, [2, 3]);
4090 /// assert_eq!(packet.header.length, [4, 5]);
4091 /// assert_eq!(packet.header.checksum, [6, 7]);
4092 /// assert_eq!(packet.body, [[8, 9], [10, 11], [12, 13]]);
4093 /// assert_eq!(suffix, &[14u8][..]);
4094 /// ```
4095 ///
4096 #[doc = codegen_section!(
4097 header = "h5",
4098 bench = "ref_from_prefix",
4099 format = "coco",
4100 arity = 3,
4101 [
4102 open
4103 @index 1
4104 @title "Sized"
4105 @variant "static_size"
4106 ],
4107 [
4108 @index 2
4109 @title "Unsized"
4110 @variant "dynamic_size"
4111 ],
4112 [
4113 @index 3
4114 @title "Dynamically Padded"
4115 @variant "dynamic_padding"
4116 ]
4117 )]
4118 #[must_use = "has no side effects"]
4119 #[inline]
4120 fn ref_from_prefix(source: &[u8]) -> Result<(&Self, &[u8]), CastError<&[u8], Self>>
4121 where
4122 Self: KnownLayout + Immutable,
4123 {
4124 static_assert_dst_is_not_zst!(Self);
4125 ref_from_prefix_suffix(source, None, CastType::Prefix)
4126 }
4127
4128 /// Interprets the suffix of the given bytes as a `&Self`.
4129 ///
4130 /// This method computes the [largest possible size of `Self`][valid-size]
4131 /// that can fit in the trailing bytes of `source`, then attempts to return
4132 /// both a reference to those bytes interpreted as a `Self`, and a reference
4133 /// to the preceding bytes. If there are insufficient bytes, or if that
4134 /// suffix of `source` is not appropriately aligned, this returns `Err`. If
4135 /// [`Self: Unaligned`][self-unaligned], you can [infallibly discard the
4136 /// alignment error][size-error-from].
4137 ///
4138 /// `Self` may be a sized type, a slice, or a [slice DST][slice-dst].
4139 ///
4140 /// [valid-size]: crate::KnownLayout#what-is-a-valid-size
4141 /// [self-unaligned]: Unaligned
4142 /// [size-error-from]: error/struct.SizeError.html#method.from-1
4143 /// [slice-dst]: KnownLayout#dynamically-sized-types
4144 ///
4145 /// # Compile-Time Assertions
4146 ///
4147 /// This method cannot yet be used on unsized types whose dynamically-sized
4148 /// component is zero-sized. See [`ref_from_suffix_with_elems`], which does
4149 /// support such types. Attempting to use this method on such types results
4150 /// in a compile-time assertion error; e.g.:
4151 ///
4152 /// ```compile_fail,E0080
4153 /// use zerocopy::*;
4154 /// # use zerocopy_derive::*;
4155 ///
4156 /// #[derive(FromBytes, Immutable, KnownLayout)]
4157 /// #[repr(C)]
4158 /// struct ZSTy {
4159 /// leading_sized: u16,
4160 /// trailing_dst: [()],
4161 /// }
4162 ///
4163 /// let _ = ZSTy::ref_from_suffix(0u16.as_bytes()); // âš Compile Error!
4164 /// ```
4165 ///
4166 /// [`ref_from_suffix_with_elems`]: FromBytes::ref_from_suffix_with_elems
4167 ///
4168 /// # Examples
4169 ///
4170 /// ```
4171 /// use zerocopy::FromBytes;
4172 /// # use zerocopy_derive::*;
4173 ///
4174 /// #[derive(FromBytes, Immutable, KnownLayout)]
4175 /// #[repr(C)]
4176 /// struct PacketTrailer {
4177 /// frame_check_sequence: [u8; 4],
4178 /// }
4179 ///
4180 /// // These are more bytes than are needed to encode a `PacketTrailer`.
4181 /// let bytes = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9][..];
4182 ///
4183 /// let (prefix, trailer) = PacketTrailer::ref_from_suffix(bytes).unwrap();
4184 ///
4185 /// assert_eq!(prefix, &[0, 1, 2, 3, 4, 5][..]);
4186 /// assert_eq!(trailer.frame_check_sequence, [6, 7, 8, 9]);
4187 /// ```
4188 ///
4189 #[doc = codegen_section!(
4190 header = "h5",
4191 bench = "ref_from_suffix",
4192 format = "coco",
4193 arity = 3,
4194 [
4195 open
4196 @index 1
4197 @title "Sized"
4198 @variant "static_size"
4199 ],
4200 [
4201 @index 2
4202 @title "Unsized"
4203 @variant "dynamic_size"
4204 ],
4205 [
4206 @index 3
4207 @title "Dynamically Padded"
4208 @variant "dynamic_padding"
4209 ]
4210 )]
4211 #[must_use = "has no side effects"]
4212 #[inline]
4213 fn ref_from_suffix(source: &[u8]) -> Result<(&[u8], &Self), CastError<&[u8], Self>>
4214 where
4215 Self: Immutable + KnownLayout,
4216 {
4217 static_assert_dst_is_not_zst!(Self);
4218 ref_from_prefix_suffix(source, None, CastType::Suffix).map(swap)
4219 }
4220
4221 /// Interprets the given `source` as a `&mut Self`.
4222 ///
4223 /// This method attempts to return a reference to `source` interpreted as a
4224 /// `Self`. If the length of `source` is not a [valid size of
4225 /// `Self`][valid-size], or if `source` is not appropriately aligned, this
4226 /// returns `Err`. If [`Self: Unaligned`][self-unaligned], you can
4227 /// [infallibly discard the alignment error][size-error-from].
4228 ///
4229 /// `Self` may be a sized type, a slice, or a [slice DST][slice-dst].
4230 ///
4231 /// [valid-size]: crate::KnownLayout#what-is-a-valid-size
4232 /// [self-unaligned]: Unaligned
4233 /// [size-error-from]: error/struct.SizeError.html#method.from-1
4234 /// [slice-dst]: KnownLayout#dynamically-sized-types
4235 ///
4236 /// # Compile-Time Assertions
4237 ///
4238 /// This method cannot yet be used on unsized types whose dynamically-sized
4239 /// component is zero-sized. See [`mut_from_prefix_with_elems`], which does
4240 /// support such types. Attempting to use this method on such types results
4241 /// in a compile-time assertion error; e.g.:
4242 ///
4243 /// ```compile_fail,E0080
4244 /// use zerocopy::*;
4245 /// # use zerocopy_derive::*;
4246 ///
4247 /// #[derive(FromBytes, Immutable, IntoBytes, KnownLayout)]
4248 /// #[repr(C, packed)]
4249 /// struct ZSTy {
4250 /// leading_sized: [u8; 2],
4251 /// trailing_dst: [()],
4252 /// }
4253 ///
4254 /// let mut source = [85, 85];
4255 /// let _ = ZSTy::mut_from_bytes(&mut source[..]); // âš Compile Error!
4256 /// ```
4257 ///
4258 /// [`mut_from_prefix_with_elems`]: FromBytes::mut_from_prefix_with_elems
4259 ///
4260 /// # Examples
4261 ///
4262 /// ```
4263 /// use zerocopy::FromBytes;
4264 /// # use zerocopy_derive::*;
4265 ///
4266 /// #[derive(FromBytes, IntoBytes, KnownLayout, Immutable)]
4267 /// #[repr(C)]
4268 /// struct PacketHeader {
4269 /// src_port: [u8; 2],
4270 /// dst_port: [u8; 2],
4271 /// length: [u8; 2],
4272 /// checksum: [u8; 2],
4273 /// }
4274 ///
4275 /// // These bytes encode a `PacketHeader`.
4276 /// let bytes = &mut [0, 1, 2, 3, 4, 5, 6, 7][..];
4277 ///
4278 /// let header = PacketHeader::mut_from_bytes(bytes).unwrap();
4279 ///
4280 /// assert_eq!(header.src_port, [0, 1]);
4281 /// assert_eq!(header.dst_port, [2, 3]);
4282 /// assert_eq!(header.length, [4, 5]);
4283 /// assert_eq!(header.checksum, [6, 7]);
4284 ///
4285 /// header.checksum = [0, 0];
4286 ///
4287 /// assert_eq!(bytes, [0, 1, 2, 3, 4, 5, 0, 0]);
4288 ///
4289 /// ```
4290 ///
4291 #[doc = codegen_header!("h5", "mut_from_bytes")]
4292 ///
4293 /// See [`FromBytes::ref_from_bytes`](#method.ref_from_bytes.codegen).
4294 #[must_use = "has no side effects"]
4295 #[inline]
4296 fn mut_from_bytes(source: &mut [u8]) -> Result<&mut Self, CastError<&mut [u8], Self>>
4297 where
4298 Self: IntoBytes + KnownLayout,
4299 {
4300 static_assert_dst_is_not_zst!(Self);
4301 match Ptr::from_mut(source).try_cast_into_no_leftover::<_, BecauseExclusive>(None) {
4302 Ok(ptr) => Ok(ptr.recall_validity::<_, (_, (_, _))>().as_mut()),
4303 Err(err) => Err(err.map_src(|src| src.as_mut())),
4304 }
4305 }
4306
4307 /// Interprets the prefix of the given `source` as a `&mut Self` without
4308 /// copying.
4309 ///
4310 /// This method computes the [largest possible size of `Self`][valid-size]
4311 /// that can fit in the leading bytes of `source`, then attempts to return
4312 /// both a reference to those bytes interpreted as a `Self`, and a reference
4313 /// to the remaining bytes. If there are insufficient bytes, or if `source`
4314 /// is not appropriately aligned, this returns `Err`. If [`Self:
4315 /// Unaligned`][self-unaligned], you can [infallibly discard the alignment
4316 /// error][size-error-from].
4317 ///
4318 /// `Self` may be a sized type, a slice, or a [slice DST][slice-dst].
4319 ///
4320 /// [valid-size]: crate::KnownLayout#what-is-a-valid-size
4321 /// [self-unaligned]: Unaligned
4322 /// [size-error-from]: error/struct.SizeError.html#method.from-1
4323 /// [slice-dst]: KnownLayout#dynamically-sized-types
4324 ///
4325 /// # Compile-Time Assertions
4326 ///
4327 /// This method cannot yet be used on unsized types whose dynamically-sized
4328 /// component is zero-sized. See [`mut_from_suffix_with_elems`], which does
4329 /// support such types. Attempting to use this method on such types results
4330 /// in a compile-time assertion error; e.g.:
4331 ///
4332 /// ```compile_fail,E0080
4333 /// use zerocopy::*;
4334 /// # use zerocopy_derive::*;
4335 ///
4336 /// #[derive(FromBytes, Immutable, IntoBytes, KnownLayout)]
4337 /// #[repr(C, packed)]
4338 /// struct ZSTy {
4339 /// leading_sized: [u8; 2],
4340 /// trailing_dst: [()],
4341 /// }
4342 ///
4343 /// let mut source = [85, 85];
4344 /// let _ = ZSTy::mut_from_prefix(&mut source[..]); // âš Compile Error!
4345 /// ```
4346 ///
4347 /// [`mut_from_suffix_with_elems`]: FromBytes::mut_from_suffix_with_elems
4348 ///
4349 /// # Examples
4350 ///
4351 /// ```
4352 /// use zerocopy::FromBytes;
4353 /// # use zerocopy_derive::*;
4354 ///
4355 /// #[derive(FromBytes, IntoBytes, KnownLayout, Immutable)]
4356 /// #[repr(C)]
4357 /// struct PacketHeader {
4358 /// src_port: [u8; 2],
4359 /// dst_port: [u8; 2],
4360 /// length: [u8; 2],
4361 /// checksum: [u8; 2],
4362 /// }
4363 ///
4364 /// // These are more bytes than are needed to encode a `PacketHeader`.
4365 /// let bytes = &mut [0, 1, 2, 3, 4, 5, 6, 7, 8, 9][..];
4366 ///
4367 /// let (header, body) = PacketHeader::mut_from_prefix(bytes).unwrap();
4368 ///
4369 /// assert_eq!(header.src_port, [0, 1]);
4370 /// assert_eq!(header.dst_port, [2, 3]);
4371 /// assert_eq!(header.length, [4, 5]);
4372 /// assert_eq!(header.checksum, [6, 7]);
4373 /// assert_eq!(body, &[8, 9][..]);
4374 ///
4375 /// header.checksum = [0, 0];
4376 /// body.fill(1);
4377 ///
4378 /// assert_eq!(bytes, [0, 1, 2, 3, 4, 5, 0, 0, 1, 1]);
4379 /// ```
4380 ///
4381 #[doc = codegen_header!("h5", "mut_from_prefix")]
4382 ///
4383 /// See [`FromBytes::ref_from_prefix`](#method.ref_from_prefix.codegen).
4384 #[must_use = "has no side effects"]
4385 #[inline]
4386 fn mut_from_prefix(
4387 source: &mut [u8],
4388 ) -> Result<(&mut Self, &mut [u8]), CastError<&mut [u8], Self>>
4389 where
4390 Self: IntoBytes + KnownLayout,
4391 {
4392 static_assert_dst_is_not_zst!(Self);
4393 mut_from_prefix_suffix(source, None, CastType::Prefix)
4394 }
4395
4396 /// Interprets the suffix of the given `source` as a `&mut Self` without
4397 /// copying.
4398 ///
4399 /// This method computes the [largest possible size of `Self`][valid-size]
4400 /// that can fit in the trailing bytes of `source`, then attempts to return
4401 /// both a reference to those bytes interpreted as a `Self`, and a reference
4402 /// to the preceding bytes. If there are insufficient bytes, or if that
4403 /// suffix of `source` is not appropriately aligned, this returns `Err`. If
4404 /// [`Self: Unaligned`][self-unaligned], you can [infallibly discard the
4405 /// alignment error][size-error-from].
4406 ///
4407 /// `Self` may be a sized type, a slice, or a [slice DST][slice-dst].
4408 ///
4409 /// [valid-size]: crate::KnownLayout#what-is-a-valid-size
4410 /// [self-unaligned]: Unaligned
4411 /// [size-error-from]: error/struct.SizeError.html#method.from-1
4412 /// [slice-dst]: KnownLayout#dynamically-sized-types
4413 ///
4414 /// # Compile-Time Assertions
4415 ///
4416 /// This method cannot yet be used on unsized types whose dynamically-sized
4417 /// component is zero-sized. Attempting to use this method on such types
4418 /// results in a compile-time assertion error; e.g.:
4419 ///
4420 /// ```compile_fail,E0080
4421 /// use zerocopy::*;
4422 /// # use zerocopy_derive::*;
4423 ///
4424 /// #[derive(FromBytes, Immutable, IntoBytes, KnownLayout)]
4425 /// #[repr(C, packed)]
4426 /// struct ZSTy {
4427 /// leading_sized: [u8; 2],
4428 /// trailing_dst: [()],
4429 /// }
4430 ///
4431 /// let mut source = [85, 85];
4432 /// let _ = ZSTy::mut_from_suffix(&mut source[..]); // âš Compile Error!
4433 /// ```
4434 ///
4435 /// # Examples
4436 ///
4437 /// ```
4438 /// use zerocopy::FromBytes;
4439 /// # use zerocopy_derive::*;
4440 ///
4441 /// #[derive(FromBytes, IntoBytes, KnownLayout, Immutable)]
4442 /// #[repr(C)]
4443 /// struct PacketTrailer {
4444 /// frame_check_sequence: [u8; 4],
4445 /// }
4446 ///
4447 /// // These are more bytes than are needed to encode a `PacketTrailer`.
4448 /// let bytes = &mut [0, 1, 2, 3, 4, 5, 6, 7, 8, 9][..];
4449 ///
4450 /// let (prefix, trailer) = PacketTrailer::mut_from_suffix(bytes).unwrap();
4451 ///
4452 /// assert_eq!(prefix, &[0u8, 1, 2, 3, 4, 5][..]);
4453 /// assert_eq!(trailer.frame_check_sequence, [6, 7, 8, 9]);
4454 ///
4455 /// prefix.fill(0);
4456 /// trailer.frame_check_sequence.fill(1);
4457 ///
4458 /// assert_eq!(bytes, [0, 0, 0, 0, 0, 0, 1, 1, 1, 1]);
4459 /// ```
4460 ///
4461 #[doc = codegen_header!("h5", "mut_from_suffix")]
4462 ///
4463 /// See [`FromBytes::ref_from_suffix`](#method.ref_from_suffix.codegen).
4464 #[must_use = "has no side effects"]
4465 #[inline]
4466 fn mut_from_suffix(
4467 source: &mut [u8],
4468 ) -> Result<(&mut [u8], &mut Self), CastError<&mut [u8], Self>>
4469 where
4470 Self: IntoBytes + KnownLayout,
4471 {
4472 static_assert_dst_is_not_zst!(Self);
4473 mut_from_prefix_suffix(source, None, CastType::Suffix).map(swap)
4474 }
4475
4476 /// Interprets the given `source` as a `&Self` with a DST length equal to
4477 /// `count`.
4478 ///
4479 /// This method attempts to return a reference to `source` interpreted as a
4480 /// `Self` with `count` trailing elements. If the length of `source` is not
4481 /// equal to the size of `Self` with `count` elements, or if `source` is not
4482 /// appropriately aligned, this returns `Err`. If [`Self:
4483 /// Unaligned`][self-unaligned], you can [infallibly discard the alignment
4484 /// error][size-error-from].
4485 ///
4486 /// [self-unaligned]: Unaligned
4487 /// [size-error-from]: error/struct.SizeError.html#method.from-1
4488 ///
4489 /// # Examples
4490 ///
4491 /// ```
4492 /// use zerocopy::FromBytes;
4493 /// # use zerocopy_derive::*;
4494 ///
4495 /// # #[derive(Debug, PartialEq, Eq)]
4496 /// #[derive(FromBytes, Immutable)]
4497 /// #[repr(C)]
4498 /// struct Pixel {
4499 /// r: u8,
4500 /// g: u8,
4501 /// b: u8,
4502 /// a: u8,
4503 /// }
4504 ///
4505 /// let bytes = &[0, 1, 2, 3, 4, 5, 6, 7][..];
4506 ///
4507 /// let pixels = <[Pixel]>::ref_from_bytes_with_elems(bytes, 2).unwrap();
4508 ///
4509 /// assert_eq!(pixels, &[
4510 /// Pixel { r: 0, g: 1, b: 2, a: 3 },
4511 /// Pixel { r: 4, g: 5, b: 6, a: 7 },
4512 /// ]);
4513 ///
4514 /// ```
4515 ///
4516 /// Since an explicit `count` is provided, this method supports types with
4517 /// zero-sized trailing slice elements. Methods such as [`ref_from_bytes`]
4518 /// which do not take an explicit count do not support such types.
4519 ///
4520 /// ```
4521 /// use zerocopy::*;
4522 /// # use zerocopy_derive::*;
4523 ///
4524 /// #[derive(FromBytes, Immutable, KnownLayout)]
4525 /// #[repr(C)]
4526 /// struct ZSTy {
4527 /// leading_sized: [u8; 2],
4528 /// trailing_dst: [()],
4529 /// }
4530 ///
4531 /// let src = &[85, 85][..];
4532 /// let zsty = ZSTy::ref_from_bytes_with_elems(src, 42).unwrap();
4533 /// assert_eq!(zsty.trailing_dst.len(), 42);
4534 /// ```
4535 ///
4536 /// [`ref_from_bytes`]: FromBytes::ref_from_bytes
4537 ///
4538 #[doc = codegen_section!(
4539 header = "h5",
4540 bench = "ref_from_bytes_with_elems",
4541 format = "coco",
4542 arity = 2,
4543 [
4544 open
4545 @index 1
4546 @title "Unsized"
4547 @variant "dynamic_size"
4548 ],
4549 [
4550 @index 2
4551 @title "Dynamically Padded"
4552 @variant "dynamic_padding"
4553 ]
4554 )]
4555 #[must_use = "has no side effects"]
4556 #[inline]
4557 fn ref_from_bytes_with_elems(
4558 source: &[u8],
4559 count: usize,
4560 ) -> Result<&Self, CastError<&[u8], Self>>
4561 where
4562 Self: KnownLayout<PointerMetadata = usize> + Immutable,
4563 {
4564 let source = Ptr::from_ref(source);
4565 let maybe_slf = source.try_cast_into_no_leftover::<_, BecauseImmutable>(Some(count));
4566 match maybe_slf {
4567 Ok(slf) => Ok(slf.recall_validity().as_ref()),
4568 Err(err) => Err(err.map_src(|s| s.as_ref())),
4569 }
4570 }
4571
4572 /// Interprets the prefix of the given `source` as a DST `&Self` with length
4573 /// equal to `count`.
4574 ///
4575 /// This method attempts to return a reference to the prefix of `source`
4576 /// interpreted as a `Self` with `count` trailing elements, and a reference
4577 /// to the remaining bytes. If there are insufficient bytes, or if `source`
4578 /// is not appropriately aligned, this returns `Err`. If [`Self:
4579 /// Unaligned`][self-unaligned], you can [infallibly discard the alignment
4580 /// error][size-error-from].
4581 ///
4582 /// [self-unaligned]: Unaligned
4583 /// [size-error-from]: error/struct.SizeError.html#method.from-1
4584 ///
4585 /// # Examples
4586 ///
4587 /// ```
4588 /// use zerocopy::FromBytes;
4589 /// # use zerocopy_derive::*;
4590 ///
4591 /// # #[derive(Debug, PartialEq, Eq)]
4592 /// #[derive(FromBytes, Immutable)]
4593 /// #[repr(C)]
4594 /// struct Pixel {
4595 /// r: u8,
4596 /// g: u8,
4597 /// b: u8,
4598 /// a: u8,
4599 /// }
4600 ///
4601 /// // These are more bytes than are needed to encode two `Pixel`s.
4602 /// let bytes = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9][..];
4603 ///
4604 /// let (pixels, suffix) = <[Pixel]>::ref_from_prefix_with_elems(bytes, 2).unwrap();
4605 ///
4606 /// assert_eq!(pixels, &[
4607 /// Pixel { r: 0, g: 1, b: 2, a: 3 },
4608 /// Pixel { r: 4, g: 5, b: 6, a: 7 },
4609 /// ]);
4610 ///
4611 /// assert_eq!(suffix, &[8, 9]);
4612 /// ```
4613 ///
4614 /// Since an explicit `count` is provided, this method supports types with
4615 /// zero-sized trailing slice elements. Methods such as [`ref_from_prefix`]
4616 /// which do not take an explicit count do not support such types.
4617 ///
4618 /// ```
4619 /// use zerocopy::*;
4620 /// # use zerocopy_derive::*;
4621 ///
4622 /// #[derive(FromBytes, Immutable, KnownLayout)]
4623 /// #[repr(C)]
4624 /// struct ZSTy {
4625 /// leading_sized: [u8; 2],
4626 /// trailing_dst: [()],
4627 /// }
4628 ///
4629 /// let src = &[85, 85][..];
4630 /// let (zsty, _) = ZSTy::ref_from_prefix_with_elems(src, 42).unwrap();
4631 /// assert_eq!(zsty.trailing_dst.len(), 42);
4632 /// ```
4633 ///
4634 /// [`ref_from_prefix`]: FromBytes::ref_from_prefix
4635 ///
4636 #[doc = codegen_section!(
4637 header = "h5",
4638 bench = "ref_from_prefix_with_elems",
4639 format = "coco",
4640 arity = 2,
4641 [
4642 open
4643 @index 1
4644 @title "Unsized"
4645 @variant "dynamic_size"
4646 ],
4647 [
4648 @index 2
4649 @title "Dynamically Padded"
4650 @variant "dynamic_padding"
4651 ]
4652 )]
4653 #[must_use = "has no side effects"]
4654 #[inline]
4655 fn ref_from_prefix_with_elems(
4656 source: &[u8],
4657 count: usize,
4658 ) -> Result<(&Self, &[u8]), CastError<&[u8], Self>>
4659 where
4660 Self: KnownLayout<PointerMetadata = usize> + Immutable,
4661 {
4662 ref_from_prefix_suffix(source, Some(count), CastType::Prefix)
4663 }
4664
4665 /// Interprets the suffix of the given `source` as a DST `&Self` with length
4666 /// equal to `count`.
4667 ///
4668 /// This method attempts to return a reference to the suffix of `source`
4669 /// interpreted as a `Self` with `count` trailing elements, and a reference
4670 /// to the preceding bytes. If there are insufficient bytes, or if that
4671 /// suffix of `source` is not appropriately aligned, this returns `Err`. If
4672 /// [`Self: Unaligned`][self-unaligned], you can [infallibly discard the
4673 /// alignment error][size-error-from].
4674 ///
4675 /// [self-unaligned]: Unaligned
4676 /// [size-error-from]: error/struct.SizeError.html#method.from-1
4677 ///
4678 /// # Examples
4679 ///
4680 /// ```
4681 /// use zerocopy::FromBytes;
4682 /// # use zerocopy_derive::*;
4683 ///
4684 /// # #[derive(Debug, PartialEq, Eq)]
4685 /// #[derive(FromBytes, Immutable)]
4686 /// #[repr(C)]
4687 /// struct Pixel {
4688 /// r: u8,
4689 /// g: u8,
4690 /// b: u8,
4691 /// a: u8,
4692 /// }
4693 ///
4694 /// // These are more bytes than are needed to encode two `Pixel`s.
4695 /// let bytes = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9][..];
4696 ///
4697 /// let (prefix, pixels) = <[Pixel]>::ref_from_suffix_with_elems(bytes, 2).unwrap();
4698 ///
4699 /// assert_eq!(prefix, &[0, 1]);
4700 ///
4701 /// assert_eq!(pixels, &[
4702 /// Pixel { r: 2, g: 3, b: 4, a: 5 },
4703 /// Pixel { r: 6, g: 7, b: 8, a: 9 },
4704 /// ]);
4705 /// ```
4706 ///
4707 /// Since an explicit `count` is provided, this method supports types with
4708 /// zero-sized trailing slice elements. Methods such as [`ref_from_suffix`]
4709 /// which do not take an explicit count do not support such types.
4710 ///
4711 /// ```
4712 /// use zerocopy::*;
4713 /// # use zerocopy_derive::*;
4714 ///
4715 /// #[derive(FromBytes, Immutable, KnownLayout)]
4716 /// #[repr(C)]
4717 /// struct ZSTy {
4718 /// leading_sized: [u8; 2],
4719 /// trailing_dst: [()],
4720 /// }
4721 ///
4722 /// let src = &[85, 85][..];
4723 /// let (_, zsty) = ZSTy::ref_from_suffix_with_elems(src, 42).unwrap();
4724 /// assert_eq!(zsty.trailing_dst.len(), 42);
4725 /// ```
4726 ///
4727 /// [`ref_from_suffix`]: FromBytes::ref_from_suffix
4728 ///
4729 #[doc = codegen_section!(
4730 header = "h5",
4731 bench = "ref_from_suffix_with_elems",
4732 format = "coco",
4733 arity = 2,
4734 [
4735 open
4736 @index 1
4737 @title "Unsized"
4738 @variant "dynamic_size"
4739 ],
4740 [
4741 @index 2
4742 @title "Dynamically Padded"
4743 @variant "dynamic_padding"
4744 ]
4745 )]
4746 #[must_use = "has no side effects"]
4747 #[inline]
4748 fn ref_from_suffix_with_elems(
4749 source: &[u8],
4750 count: usize,
4751 ) -> Result<(&[u8], &Self), CastError<&[u8], Self>>
4752 where
4753 Self: KnownLayout<PointerMetadata = usize> + Immutable,
4754 {
4755 ref_from_prefix_suffix(source, Some(count), CastType::Suffix).map(swap)
4756 }
4757
4758 /// Interprets the given `source` as a `&mut Self` with a DST length equal
4759 /// to `count`.
4760 ///
4761 /// This method attempts to return a reference to `source` interpreted as a
4762 /// `Self` with `count` trailing elements. If the length of `source` is not
4763 /// equal to the size of `Self` with `count` elements, or if `source` is not
4764 /// appropriately aligned, this returns `Err`. If [`Self:
4765 /// Unaligned`][self-unaligned], you can [infallibly discard the alignment
4766 /// error][size-error-from].
4767 ///
4768 /// [self-unaligned]: Unaligned
4769 /// [size-error-from]: error/struct.SizeError.html#method.from-1
4770 ///
4771 /// # Examples
4772 ///
4773 /// ```
4774 /// use zerocopy::FromBytes;
4775 /// # use zerocopy_derive::*;
4776 ///
4777 /// # #[derive(Debug, PartialEq, Eq)]
4778 /// #[derive(KnownLayout, FromBytes, IntoBytes, Immutable)]
4779 /// #[repr(C)]
4780 /// struct Pixel {
4781 /// r: u8,
4782 /// g: u8,
4783 /// b: u8,
4784 /// a: u8,
4785 /// }
4786 ///
4787 /// let bytes = &mut [0, 1, 2, 3, 4, 5, 6, 7][..];
4788 ///
4789 /// let pixels = <[Pixel]>::mut_from_bytes_with_elems(bytes, 2).unwrap();
4790 ///
4791 /// assert_eq!(pixels, &[
4792 /// Pixel { r: 0, g: 1, b: 2, a: 3 },
4793 /// Pixel { r: 4, g: 5, b: 6, a: 7 },
4794 /// ]);
4795 ///
4796 /// pixels[1] = Pixel { r: 0, g: 0, b: 0, a: 0 };
4797 ///
4798 /// assert_eq!(bytes, [0, 1, 2, 3, 0, 0, 0, 0]);
4799 /// ```
4800 ///
4801 /// Since an explicit `count` is provided, this method supports types with
4802 /// zero-sized trailing slice elements. Methods such as [`mut_from_bytes`]
4803 /// which do not take an explicit count do not support such types.
4804 ///
4805 /// ```
4806 /// use zerocopy::*;
4807 /// # use zerocopy_derive::*;
4808 ///
4809 /// #[derive(FromBytes, IntoBytes, Immutable, KnownLayout)]
4810 /// #[repr(C, packed)]
4811 /// struct ZSTy {
4812 /// leading_sized: [u8; 2],
4813 /// trailing_dst: [()],
4814 /// }
4815 ///
4816 /// let src = &mut [85, 85][..];
4817 /// let zsty = ZSTy::mut_from_bytes_with_elems(src, 42).unwrap();
4818 /// assert_eq!(zsty.trailing_dst.len(), 42);
4819 /// ```
4820 ///
4821 /// [`mut_from_bytes`]: FromBytes::mut_from_bytes
4822 ///
4823 #[doc = codegen_header!("h5", "mut_from_bytes_with_elems")]
4824 ///
4825 /// See [`TryFromBytes::ref_from_bytes_with_elems`](#method.ref_from_bytes_with_elems.codegen).
4826 #[must_use = "has no side effects"]
4827 #[inline]
4828 fn mut_from_bytes_with_elems(
4829 source: &mut [u8],
4830 count: usize,
4831 ) -> Result<&mut Self, CastError<&mut [u8], Self>>
4832 where
4833 Self: IntoBytes + KnownLayout<PointerMetadata = usize> + Immutable,
4834 {
4835 let source = Ptr::from_mut(source);
4836 let maybe_slf = source.try_cast_into_no_leftover::<_, BecauseImmutable>(Some(count));
4837 match maybe_slf {
4838 Ok(slf) => Ok(slf.recall_validity::<_, (_, (_, BecauseExclusive))>().as_mut()),
4839 Err(err) => Err(err.map_src(|s| s.as_mut())),
4840 }
4841 }
4842
4843 /// Interprets the prefix of the given `source` as a `&mut Self` with DST
4844 /// length equal to `count`.
4845 ///
4846 /// This method attempts to return a reference to the prefix of `source`
4847 /// interpreted as a `Self` with `count` trailing elements, and a reference
4848 /// to the preceding bytes. If there are insufficient bytes, or if `source`
4849 /// is not appropriately aligned, this returns `Err`. If [`Self:
4850 /// Unaligned`][self-unaligned], you can [infallibly discard the alignment
4851 /// error][size-error-from].
4852 ///
4853 /// [self-unaligned]: Unaligned
4854 /// [size-error-from]: error/struct.SizeError.html#method.from-1
4855 ///
4856 /// # Examples
4857 ///
4858 /// ```
4859 /// use zerocopy::FromBytes;
4860 /// # use zerocopy_derive::*;
4861 ///
4862 /// # #[derive(Debug, PartialEq, Eq)]
4863 /// #[derive(KnownLayout, FromBytes, IntoBytes, Immutable)]
4864 /// #[repr(C)]
4865 /// struct Pixel {
4866 /// r: u8,
4867 /// g: u8,
4868 /// b: u8,
4869 /// a: u8,
4870 /// }
4871 ///
4872 /// // These are more bytes than are needed to encode two `Pixel`s.
4873 /// let bytes = &mut [0, 1, 2, 3, 4, 5, 6, 7, 8, 9][..];
4874 ///
4875 /// let (pixels, suffix) = <[Pixel]>::mut_from_prefix_with_elems(bytes, 2).unwrap();
4876 ///
4877 /// assert_eq!(pixels, &[
4878 /// Pixel { r: 0, g: 1, b: 2, a: 3 },
4879 /// Pixel { r: 4, g: 5, b: 6, a: 7 },
4880 /// ]);
4881 ///
4882 /// assert_eq!(suffix, &[8, 9]);
4883 ///
4884 /// pixels[1] = Pixel { r: 0, g: 0, b: 0, a: 0 };
4885 /// suffix.fill(1);
4886 ///
4887 /// assert_eq!(bytes, [0, 1, 2, 3, 0, 0, 0, 0, 1, 1]);
4888 /// ```
4889 ///
4890 /// Since an explicit `count` is provided, this method supports types with
4891 /// zero-sized trailing slice elements. Methods such as [`mut_from_prefix`]
4892 /// which do not take an explicit count do not support such types.
4893 ///
4894 /// ```
4895 /// use zerocopy::*;
4896 /// # use zerocopy_derive::*;
4897 ///
4898 /// #[derive(FromBytes, IntoBytes, Immutable, KnownLayout)]
4899 /// #[repr(C, packed)]
4900 /// struct ZSTy {
4901 /// leading_sized: [u8; 2],
4902 /// trailing_dst: [()],
4903 /// }
4904 ///
4905 /// let src = &mut [85, 85][..];
4906 /// let (zsty, _) = ZSTy::mut_from_prefix_with_elems(src, 42).unwrap();
4907 /// assert_eq!(zsty.trailing_dst.len(), 42);
4908 /// ```
4909 ///
4910 /// [`mut_from_prefix`]: FromBytes::mut_from_prefix
4911 ///
4912 #[doc = codegen_header!("h5", "mut_from_prefix_with_elems")]
4913 ///
4914 /// See [`TryFromBytes::ref_from_prefix_with_elems`](#method.ref_from_prefix_with_elems.codegen).
4915 #[must_use = "has no side effects"]
4916 #[inline]
4917 fn mut_from_prefix_with_elems(
4918 source: &mut [u8],
4919 count: usize,
4920 ) -> Result<(&mut Self, &mut [u8]), CastError<&mut [u8], Self>>
4921 where
4922 Self: IntoBytes + KnownLayout<PointerMetadata = usize>,
4923 {
4924 mut_from_prefix_suffix(source, Some(count), CastType::Prefix)
4925 }
4926
4927 /// Interprets the suffix of the given `source` as a `&mut Self` with DST
4928 /// length equal to `count`.
4929 ///
4930 /// This method attempts to return a reference to the suffix of `source`
4931 /// interpreted as a `Self` with `count` trailing elements, and a reference
4932 /// to the remaining bytes. If there are insufficient bytes, or if that
4933 /// suffix of `source` is not appropriately aligned, this returns `Err`. If
4934 /// [`Self: Unaligned`][self-unaligned], you can [infallibly discard the
4935 /// alignment error][size-error-from].
4936 ///
4937 /// [self-unaligned]: Unaligned
4938 /// [size-error-from]: error/struct.SizeError.html#method.from-1
4939 ///
4940 /// # Examples
4941 ///
4942 /// ```
4943 /// use zerocopy::FromBytes;
4944 /// # use zerocopy_derive::*;
4945 ///
4946 /// # #[derive(Debug, PartialEq, Eq)]
4947 /// #[derive(FromBytes, IntoBytes, Immutable)]
4948 /// #[repr(C)]
4949 /// struct Pixel {
4950 /// r: u8,
4951 /// g: u8,
4952 /// b: u8,
4953 /// a: u8,
4954 /// }
4955 ///
4956 /// // These are more bytes than are needed to encode two `Pixel`s.
4957 /// let bytes = &mut [0, 1, 2, 3, 4, 5, 6, 7, 8, 9][..];
4958 ///
4959 /// let (prefix, pixels) = <[Pixel]>::mut_from_suffix_with_elems(bytes, 2).unwrap();
4960 ///
4961 /// assert_eq!(prefix, &[0, 1]);
4962 ///
4963 /// assert_eq!(pixels, &[
4964 /// Pixel { r: 2, g: 3, b: 4, a: 5 },
4965 /// Pixel { r: 6, g: 7, b: 8, a: 9 },
4966 /// ]);
4967 ///
4968 /// prefix.fill(9);
4969 /// pixels[1] = Pixel { r: 0, g: 0, b: 0, a: 0 };
4970 ///
4971 /// assert_eq!(bytes, [9, 9, 2, 3, 4, 5, 0, 0, 0, 0]);
4972 /// ```
4973 ///
4974 /// Since an explicit `count` is provided, this method supports types with
4975 /// zero-sized trailing slice elements. Methods such as [`mut_from_suffix`]
4976 /// which do not take an explicit count do not support such types.
4977 ///
4978 /// ```
4979 /// use zerocopy::*;
4980 /// # use zerocopy_derive::*;
4981 ///
4982 /// #[derive(FromBytes, IntoBytes, Immutable, KnownLayout)]
4983 /// #[repr(C, packed)]
4984 /// struct ZSTy {
4985 /// leading_sized: [u8; 2],
4986 /// trailing_dst: [()],
4987 /// }
4988 ///
4989 /// let src = &mut [85, 85][..];
4990 /// let (_, zsty) = ZSTy::mut_from_suffix_with_elems(src, 42).unwrap();
4991 /// assert_eq!(zsty.trailing_dst.len(), 42);
4992 /// ```
4993 ///
4994 /// [`mut_from_suffix`]: FromBytes::mut_from_suffix
4995 ///
4996 #[doc = codegen_header!("h5", "mut_from_suffix_with_elems")]
4997 ///
4998 /// See [`TryFromBytes::ref_from_suffix_with_elems`](#method.ref_from_suffix_with_elems.codegen).
4999 #[must_use = "has no side effects"]
5000 #[inline]
5001 fn mut_from_suffix_with_elems(
5002 source: &mut [u8],
5003 count: usize,
5004 ) -> Result<(&mut [u8], &mut Self), CastError<&mut [u8], Self>>
5005 where
5006 Self: IntoBytes + KnownLayout<PointerMetadata = usize>,
5007 {
5008 mut_from_prefix_suffix(source, Some(count), CastType::Suffix).map(swap)
5009 }
5010
5011 /// Reads a copy of `Self` from the given `source`.
5012 ///
5013 /// If `source.len() != size_of::<Self>()`, `read_from_bytes` returns `Err`.
5014 ///
5015 /// # Examples
5016 ///
5017 /// ```
5018 /// use zerocopy::FromBytes;
5019 /// # use zerocopy_derive::*;
5020 ///
5021 /// #[derive(FromBytes)]
5022 /// #[repr(C)]
5023 /// struct PacketHeader {
5024 /// src_port: [u8; 2],
5025 /// dst_port: [u8; 2],
5026 /// length: [u8; 2],
5027 /// checksum: [u8; 2],
5028 /// }
5029 ///
5030 /// // These bytes encode a `PacketHeader`.
5031 /// let bytes = &[0, 1, 2, 3, 4, 5, 6, 7][..];
5032 ///
5033 /// let header = PacketHeader::read_from_bytes(bytes).unwrap();
5034 ///
5035 /// assert_eq!(header.src_port, [0, 1]);
5036 /// assert_eq!(header.dst_port, [2, 3]);
5037 /// assert_eq!(header.length, [4, 5]);
5038 /// assert_eq!(header.checksum, [6, 7]);
5039 /// ```
5040 ///
5041 #[doc = codegen_section!(
5042 header = "h5",
5043 bench = "read_from_bytes",
5044 format = "coco_static_size",
5045 )]
5046 #[must_use = "has no side effects"]
5047 #[inline]
5048 fn read_from_bytes(source: &[u8]) -> Result<Self, SizeError<&[u8], Self>>
5049 where
5050 Self: Sized,
5051 {
5052 match Ref::<_, Unalign<Self>>::sized_from(source) {
5053 Ok(r) => Ok(Ref::read(&r).into_inner()),
5054 Err(CastError::Size(e)) => Err(e.with_dst()),
5055 Err(CastError::Alignment(_)) => {
5056 // SAFETY: `Unalign<Self>` is trivially aligned, so
5057 // `Ref::sized_from` cannot fail due to unmet alignment
5058 // requirements.
5059 unsafe { core::hint::unreachable_unchecked() }
5060 }
5061 Err(CastError::Validity(i)) => match i {},
5062 }
5063 }
5064
5065 /// Reads a copy of `Self` from the prefix of the given `source`.
5066 ///
5067 /// This attempts to read a `Self` from the first `size_of::<Self>()` bytes
5068 /// of `source`, returning that `Self` and any remaining bytes. If
5069 /// `source.len() < size_of::<Self>()`, it returns `Err`.
5070 ///
5071 /// # Examples
5072 ///
5073 /// ```
5074 /// use zerocopy::FromBytes;
5075 /// # use zerocopy_derive::*;
5076 ///
5077 /// #[derive(FromBytes)]
5078 /// #[repr(C)]
5079 /// struct PacketHeader {
5080 /// src_port: [u8; 2],
5081 /// dst_port: [u8; 2],
5082 /// length: [u8; 2],
5083 /// checksum: [u8; 2],
5084 /// }
5085 ///
5086 /// // These are more bytes than are needed to encode a `PacketHeader`.
5087 /// let bytes = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9][..];
5088 ///
5089 /// let (header, body) = PacketHeader::read_from_prefix(bytes).unwrap();
5090 ///
5091 /// assert_eq!(header.src_port, [0, 1]);
5092 /// assert_eq!(header.dst_port, [2, 3]);
5093 /// assert_eq!(header.length, [4, 5]);
5094 /// assert_eq!(header.checksum, [6, 7]);
5095 /// assert_eq!(body, [8, 9]);
5096 /// ```
5097 ///
5098 #[doc = codegen_section!(
5099 header = "h5",
5100 bench = "read_from_prefix",
5101 format = "coco_static_size",
5102 )]
5103 #[must_use = "has no side effects"]
5104 #[inline]
5105 fn read_from_prefix(source: &[u8]) -> Result<(Self, &[u8]), SizeError<&[u8], Self>>
5106 where
5107 Self: Sized,
5108 {
5109 match Ref::<_, Unalign<Self>>::sized_from_prefix(source) {
5110 Ok((r, suffix)) => Ok((Ref::read(&r).into_inner(), suffix)),
5111 Err(CastError::Size(e)) => Err(e.with_dst()),
5112 Err(CastError::Alignment(_)) => {
5113 // SAFETY: `Unalign<Self>` is trivially aligned, so
5114 // `Ref::sized_from_prefix` cannot fail due to unmet alignment
5115 // requirements.
5116 unsafe { core::hint::unreachable_unchecked() }
5117 }
5118 Err(CastError::Validity(i)) => match i {},
5119 }
5120 }
5121
5122 /// Reads a copy of `Self` from the suffix of the given `source`.
5123 ///
5124 /// This attempts to read a `Self` from the last `size_of::<Self>()` bytes
5125 /// of `source`, returning that `Self` and any preceding bytes. If
5126 /// `source.len() < size_of::<Self>()`, it returns `Err`.
5127 ///
5128 /// # Examples
5129 ///
5130 /// ```
5131 /// use zerocopy::FromBytes;
5132 /// # use zerocopy_derive::*;
5133 ///
5134 /// #[derive(FromBytes)]
5135 /// #[repr(C)]
5136 /// struct PacketTrailer {
5137 /// frame_check_sequence: [u8; 4],
5138 /// }
5139 ///
5140 /// // These are more bytes than are needed to encode a `PacketTrailer`.
5141 /// let bytes = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9][..];
5142 ///
5143 /// let (prefix, trailer) = PacketTrailer::read_from_suffix(bytes).unwrap();
5144 ///
5145 /// assert_eq!(prefix, [0, 1, 2, 3, 4, 5]);
5146 /// assert_eq!(trailer.frame_check_sequence, [6, 7, 8, 9]);
5147 /// ```
5148 ///
5149 #[doc = codegen_section!(
5150 header = "h5",
5151 bench = "read_from_suffix",
5152 format = "coco_static_size",
5153 )]
5154 #[must_use = "has no side effects"]
5155 #[inline]
5156 fn read_from_suffix(source: &[u8]) -> Result<(&[u8], Self), SizeError<&[u8], Self>>
5157 where
5158 Self: Sized,
5159 {
5160 match Ref::<_, Unalign<Self>>::sized_from_suffix(source) {
5161 Ok((prefix, r)) => Ok((prefix, Ref::read(&r).into_inner())),
5162 Err(CastError::Size(e)) => Err(e.with_dst()),
5163 Err(CastError::Alignment(_)) => {
5164 // SAFETY: `Unalign<Self>` is trivially aligned, so
5165 // `Ref::sized_from_suffix` cannot fail due to unmet alignment
5166 // requirements.
5167 unsafe { core::hint::unreachable_unchecked() }
5168 }
5169 Err(CastError::Validity(i)) => match i {},
5170 }
5171 }
5172
5173 /// Reads a copy of `self` from an `io::Read`.
5174 ///
5175 /// This is useful for interfacing with operating system byte sinks (files,
5176 /// sockets, etc.).
5177 ///
5178 /// # Examples
5179 ///
5180 /// ```no_run
5181 /// use zerocopy::{byteorder::big_endian::*, FromBytes};
5182 /// use std::fs::File;
5183 /// # use zerocopy_derive::*;
5184 ///
5185 /// #[derive(FromBytes)]
5186 /// #[repr(C)]
5187 /// struct BitmapFileHeader {
5188 /// signature: [u8; 2],
5189 /// size: U32,
5190 /// reserved: U64,
5191 /// offset: U64,
5192 /// }
5193 ///
5194 /// let mut file = File::open("image.bin").unwrap();
5195 /// let header = BitmapFileHeader::read_from_io(&mut file).unwrap();
5196 /// ```
5197 #[cfg(feature = "std")]
5198 #[cfg_attr(doc_cfg, doc(cfg(feature = "std")))]
5199 #[inline(always)]
5200 fn read_from_io<R>(mut src: R) -> io::Result<Self>
5201 where
5202 Self: Sized,
5203 R: io::Read,
5204 {
5205 // NOTE(#2319, #2320): We do `buf.zero()` separately rather than
5206 // constructing `let buf = CoreMaybeUninit::zeroed()` because, if `Self`
5207 // contains padding bytes, then a typed copy of `CoreMaybeUninit<Self>`
5208 // will not necessarily preserve zeros written to those padding byte
5209 // locations, and so `buf` could contain uninitialized bytes.
5210 let mut buf = CoreMaybeUninit::<Self>::uninit();
5211 buf.zero();
5212
5213 let ptr = Ptr::from_mut(&mut buf);
5214 // SAFETY: After `buf.zero()`, `buf` consists entirely of initialized,
5215 // zeroed bytes. Since `MaybeUninit` has no validity requirements, `ptr`
5216 // cannot be used to write values which will violate `buf`'s bit
5217 // validity. Since `ptr` has `Exclusive` aliasing, nothing other than
5218 // `ptr` may be used to mutate `ptr`'s referent, and so its bit validity
5219 // cannot be violated even though `buf` may have more permissive bit
5220 // validity than `ptr`.
5221 let ptr = unsafe { ptr.assume_validity::<invariant::Initialized>() };
5222 let ptr = ptr.as_bytes();
5223 src.read_exact(ptr.as_mut())?;
5224 // SAFETY: `buf` entirely consists of initialized bytes, and `Self` is
5225 // `FromBytes`.
5226 Ok(unsafe { buf.assume_init() })
5227 }
5228
5229 #[deprecated(since = "0.8.0", note = "renamed to `FromBytes::ref_from_bytes`")]
5230 #[doc(hidden)]
5231 #[must_use = "has no side effects"]
5232 #[inline(always)]
5233 fn ref_from(source: &[u8]) -> Option<&Self>
5234 where
5235 Self: KnownLayout + Immutable,
5236 {
5237 Self::ref_from_bytes(source).ok()
5238 }
5239
5240 #[deprecated(since = "0.8.0", note = "renamed to `FromBytes::mut_from_bytes`")]
5241 #[doc(hidden)]
5242 #[must_use = "has no side effects"]
5243 #[inline(always)]
5244 fn mut_from(source: &mut [u8]) -> Option<&mut Self>
5245 where
5246 Self: KnownLayout + IntoBytes,
5247 {
5248 Self::mut_from_bytes(source).ok()
5249 }
5250
5251 #[deprecated(since = "0.8.0", note = "renamed to `FromBytes::ref_from_prefix_with_elems`")]
5252 #[doc(hidden)]
5253 #[must_use = "has no side effects"]
5254 #[inline(always)]
5255 fn slice_from_prefix(source: &[u8], count: usize) -> Option<(&[Self], &[u8])>
5256 where
5257 Self: Sized + Immutable,
5258 {
5259 <[Self]>::ref_from_prefix_with_elems(source, count).ok()
5260 }
5261
5262 #[deprecated(since = "0.8.0", note = "renamed to `FromBytes::ref_from_suffix_with_elems`")]
5263 #[doc(hidden)]
5264 #[must_use = "has no side effects"]
5265 #[inline(always)]
5266 fn slice_from_suffix(source: &[u8], count: usize) -> Option<(&[u8], &[Self])>
5267 where
5268 Self: Sized + Immutable,
5269 {
5270 <[Self]>::ref_from_suffix_with_elems(source, count).ok()
5271 }
5272
5273 #[deprecated(since = "0.8.0", note = "renamed to `FromBytes::mut_from_prefix_with_elems`")]
5274 #[doc(hidden)]
5275 #[must_use = "has no side effects"]
5276 #[inline(always)]
5277 fn mut_slice_from_prefix(source: &mut [u8], count: usize) -> Option<(&mut [Self], &mut [u8])>
5278 where
5279 Self: Sized + IntoBytes,
5280 {
5281 <[Self]>::mut_from_prefix_with_elems(source, count).ok()
5282 }
5283
5284 #[deprecated(since = "0.8.0", note = "renamed to `FromBytes::mut_from_suffix_with_elems`")]
5285 #[doc(hidden)]
5286 #[must_use = "has no side effects"]
5287 #[inline(always)]
5288 fn mut_slice_from_suffix(source: &mut [u8], count: usize) -> Option<(&mut [u8], &mut [Self])>
5289 where
5290 Self: Sized + IntoBytes,
5291 {
5292 <[Self]>::mut_from_suffix_with_elems(source, count).ok()
5293 }
5294
5295 #[deprecated(since = "0.8.0", note = "renamed to `FromBytes::read_from_bytes`")]
5296 #[doc(hidden)]
5297 #[must_use = "has no side effects"]
5298 #[inline(always)]
5299 fn read_from(source: &[u8]) -> Option<Self>
5300 where
5301 Self: Sized,
5302 {
5303 Self::read_from_bytes(source).ok()
5304 }
5305}
5306
5307/// Interprets the given affix of the given bytes as a `&Self`.
5308///
5309/// This method computes the largest possible size of `Self` that can fit in the
5310/// prefix or suffix bytes of `source`, then attempts to return both a reference
5311/// to those bytes interpreted as a `Self`, and a reference to the excess bytes.
5312/// If there are insufficient bytes, or if that affix of `source` is not
5313/// appropriately aligned, this returns `Err`.
5314#[inline(always)]
5315fn ref_from_prefix_suffix<T: FromBytes + KnownLayout + Immutable + ?Sized>(
5316 source: &[u8],
5317 meta: Option<T::PointerMetadata>,
5318 cast_type: CastType,
5319) -> Result<(&T, &[u8]), CastError<&[u8], T>> {
5320 let (slf, prefix_suffix) = Ptr::from_ref(source)
5321 .try_cast_into::<_, BecauseImmutable>(cast_type, meta)
5322 .map_err(|err| err.map_src(|s| s.as_ref()))?;
5323 Ok((slf.recall_validity().as_ref(), prefix_suffix.as_ref()))
5324}
5325
5326/// Interprets the given affix of the given bytes as a `&mut Self` without
5327/// copying.
5328///
5329/// This method computes the largest possible size of `Self` that can fit in the
5330/// prefix or suffix bytes of `source`, then attempts to return both a reference
5331/// to those bytes interpreted as a `Self`, and a reference to the excess bytes.
5332/// If there are insufficient bytes, or if that affix of `source` is not
5333/// appropriately aligned, this returns `Err`.
5334#[inline(always)]
5335fn mut_from_prefix_suffix<T: FromBytes + IntoBytes + KnownLayout + ?Sized>(
5336 source: &mut [u8],
5337 meta: Option<T::PointerMetadata>,
5338 cast_type: CastType,
5339) -> Result<(&mut T, &mut [u8]), CastError<&mut [u8], T>> {
5340 let (slf, prefix_suffix) = Ptr::from_mut(source)
5341 .try_cast_into::<_, BecauseExclusive>(cast_type, meta)
5342 .map_err(|err| err.map_src(|s| s.as_mut()))?;
5343 Ok((slf.recall_validity::<_, (_, (_, _))>().as_mut(), prefix_suffix.as_mut()))
5344}
5345
5346/// Analyzes whether a type is [`IntoBytes`].
5347///
5348/// This derive analyzes, at compile time, whether the annotated type satisfies
5349/// the [safety conditions] of `IntoBytes` and implements `IntoBytes` if it is
5350/// sound to do so. This derive can be applied to structs and enums (see below
5351/// for union support); e.g.:
5352///
5353/// ```
5354/// # use zerocopy_derive::{IntoBytes};
5355/// #[derive(IntoBytes)]
5356/// #[repr(C)]
5357/// struct MyStruct {
5358/// # /*
5359/// ...
5360/// # */
5361/// }
5362///
5363/// #[derive(IntoBytes)]
5364/// #[repr(u8)]
5365/// enum MyEnum {
5366/// # Variant,
5367/// # /*
5368/// ...
5369/// # */
5370/// }
5371/// ```
5372///
5373/// [safety conditions]: trait@IntoBytes#safety
5374///
5375/// # Error Messages
5376///
5377/// On Rust toolchains prior to 1.78.0, due to the way that the custom derive
5378/// for `IntoBytes` is implemented, you may get an error like this:
5379///
5380/// ```text
5381/// error[E0277]: the trait bound `(): PaddingFree<Foo, true>` is not satisfied
5382/// --> lib.rs:23:10
5383/// |
5384/// 1 | #[derive(IntoBytes)]
5385/// | ^^^^^^^^^ the trait `PaddingFree<Foo, true>` is not implemented for `()`
5386/// |
5387/// = help: the following implementations were found:
5388/// <() as PaddingFree<T, false>>
5389/// ```
5390///
5391/// This error indicates that the type being annotated has padding bytes, which
5392/// is illegal for `IntoBytes` types. Consider reducing the alignment of some
5393/// fields by using types in the [`byteorder`] module, wrapping field types in
5394/// [`Unalign`], adding explicit struct fields where those padding bytes would
5395/// be, or using `#[repr(packed)]`. See the Rust Reference's page on [type
5396/// layout] for more information about type layout and padding.
5397///
5398/// [type layout]: https://doc.rust-lang.org/reference/type-layout.html
5399///
5400/// # Unions
5401///
5402/// Currently, union bit validity is [up in the air][union-validity], and so
5403/// zerocopy does not support `#[derive(IntoBytes)]` on unions by default.
5404/// However, implementing `IntoBytes` on a union type is likely sound on all
5405/// existing Rust toolchains - it's just that it may become unsound in the
5406/// future. You can opt-in to `#[derive(IntoBytes)]` support on unions by
5407/// passing the unstable `zerocopy_derive_union_into_bytes` cfg:
5408///
5409/// ```shell
5410/// $ RUSTFLAGS='--cfg zerocopy_derive_union_into_bytes' cargo build
5411/// ```
5412///
5413/// However, it is your responsibility to ensure that this derive is sound on
5414/// the specific versions of the Rust toolchain you are using! We make no
5415/// stability or soundness guarantees regarding this cfg, and may remove it at
5416/// any point.
5417///
5418/// We are actively working with Rust to stabilize the necessary language
5419/// guarantees to support this in a forwards-compatible way, which will enable
5420/// us to remove the cfg gate. As part of this effort, we need to know how much
5421/// demand there is for this feature. If you would like to use `IntoBytes` on
5422/// unions, [please let us know][discussion].
5423///
5424/// [union-validity]: https://github.com/rust-lang/unsafe-code-guidelines/issues/438
5425/// [discussion]: https://github.com/google/zerocopy/discussions/1802
5426///
5427/// # Analysis
5428///
5429/// *This section describes, roughly, the analysis performed by this derive to
5430/// determine whether it is sound to implement `IntoBytes` for a given type.
5431/// Unless you are modifying the implementation of this derive, or attempting to
5432/// manually implement `IntoBytes` for a type yourself, you don't need to read
5433/// this section.*
5434///
5435/// If a type has the following properties, then this derive can implement
5436/// `IntoBytes` for that type:
5437///
5438/// - If the type is a struct, its fields must be [`IntoBytes`]. Additionally:
5439/// - if the type is `repr(transparent)` or `repr(packed)`, it is
5440/// [`IntoBytes`] if its fields are [`IntoBytes`]; else,
5441/// - if the type is `repr(C)` with at most one field, it is [`IntoBytes`]
5442/// if its field is [`IntoBytes`]; else,
5443/// - if the type has no generic parameters, it is [`IntoBytes`] if the type
5444/// is sized and has no padding bytes; else,
5445/// - if the type is `repr(C)`, its fields must be [`Unaligned`].
5446/// - If the type is an enum:
5447/// - It must have a defined representation (`repr`s `C`, `u8`, `u16`, `u32`,
5448/// `u64`, `usize`, `i8`, `i16`, `i32`, `i64`, or `isize`).
5449/// - It must have no padding bytes.
5450/// - Its fields must be [`IntoBytes`].
5451///
5452/// This analysis is subject to change. Unsafe code may *only* rely on the
5453/// documented [safety conditions] of `FromBytes`, and must *not* rely on the
5454/// implementation details of this derive.
5455///
5456/// [Rust Reference]: https://doc.rust-lang.org/reference/type-layout.html
5457#[cfg(any(feature = "derive", test))]
5458#[cfg_attr(doc_cfg, doc(cfg(feature = "derive")))]
5459pub use zerocopy_derive::IntoBytes;
5460
5461/// Types that can be converted to an immutable slice of initialized bytes.
5462///
5463/// Any `IntoBytes` type can be converted to a slice of initialized bytes of the
5464/// same size. This is useful for efficiently serializing structured data as raw
5465/// bytes.
5466///
5467/// # Implementation
5468///
5469/// **Do not implement this trait yourself!** Instead, use
5470/// [`#[derive(IntoBytes)]`][derive]; e.g.:
5471///
5472/// ```
5473/// # use zerocopy_derive::IntoBytes;
5474/// #[derive(IntoBytes)]
5475/// #[repr(C)]
5476/// struct MyStruct {
5477/// # /*
5478/// ...
5479/// # */
5480/// }
5481///
5482/// #[derive(IntoBytes)]
5483/// #[repr(u8)]
5484/// enum MyEnum {
5485/// # Variant0,
5486/// # /*
5487/// ...
5488/// # */
5489/// }
5490/// ```
5491///
5492/// This derive performs a sophisticated, compile-time safety analysis to
5493/// determine whether a type is `IntoBytes`. See the [derive
5494/// documentation][derive] for guidance on how to interpret error messages
5495/// produced by the derive's analysis.
5496///
5497/// # Safety
5498///
5499/// *This section describes what is required in order for `T: IntoBytes`, and
5500/// what unsafe code may assume of such types. If you don't plan on implementing
5501/// `IntoBytes` manually, and you don't plan on writing unsafe code that
5502/// operates on `IntoBytes` types, then you don't need to read this section.*
5503///
5504/// If `T: IntoBytes`, then unsafe code may assume that it is sound to treat any
5505/// `t: T` as an immutable `[u8]` of length `size_of_val(t)`. If a type is
5506/// marked as `IntoBytes` which violates this contract, it may cause undefined
5507/// behavior.
5508///
5509/// `#[derive(IntoBytes)]` only permits [types which satisfy these
5510/// requirements][derive-analysis].
5511///
5512#[cfg_attr(
5513 feature = "derive",
5514 doc = "[derive]: zerocopy_derive::IntoBytes",
5515 doc = "[derive-analysis]: zerocopy_derive::IntoBytes#analysis"
5516)]
5517#[cfg_attr(
5518 not(feature = "derive"),
5519 doc = concat!("[derive]: https://docs.rs/zerocopy/", env!("CARGO_PKG_VERSION"), "/zerocopy/derive.IntoBytes.html"),
5520 doc = concat!("[derive-analysis]: https://docs.rs/zerocopy/", env!("CARGO_PKG_VERSION"), "/zerocopy/derive.IntoBytes.html#analysis"),
5521)]
5522#[cfg_attr(
5523 not(no_zerocopy_diagnostic_on_unimplemented_1_78_0),
5524 diagnostic::on_unimplemented(note = "Consider adding `#[derive(IntoBytes)]` to `{Self}`")
5525)]
5526pub unsafe trait IntoBytes {
5527 // The `Self: Sized` bound makes it so that this function doesn't prevent
5528 // `IntoBytes` from being object safe. Note that other `IntoBytes` methods
5529 // prevent object safety, but those provide a benefit in exchange for object
5530 // safety. If at some point we remove those methods, change their type
5531 // signatures, or move them out of this trait so that `IntoBytes` is object
5532 // safe again, it's important that this function not prevent object safety.
5533 #[doc(hidden)]
5534 fn only_derive_is_allowed_to_implement_this_trait()
5535 where
5536 Self: Sized;
5537
5538 /// Gets the bytes of this value.
5539 ///
5540 /// # Examples
5541 ///
5542 /// ```
5543 /// use zerocopy::IntoBytes;
5544 /// # use zerocopy_derive::*;
5545 ///
5546 /// #[derive(IntoBytes, Immutable)]
5547 /// #[repr(C)]
5548 /// struct PacketHeader {
5549 /// src_port: [u8; 2],
5550 /// dst_port: [u8; 2],
5551 /// length: [u8; 2],
5552 /// checksum: [u8; 2],
5553 /// }
5554 ///
5555 /// let header = PacketHeader {
5556 /// src_port: [0, 1],
5557 /// dst_port: [2, 3],
5558 /// length: [4, 5],
5559 /// checksum: [6, 7],
5560 /// };
5561 ///
5562 /// let bytes = header.as_bytes();
5563 ///
5564 /// assert_eq!(bytes, [0, 1, 2, 3, 4, 5, 6, 7]);
5565 /// ```
5566 #[must_use = "has no side effects"]
5567 #[inline(always)]
5568 fn as_bytes(&self) -> &[u8]
5569 where
5570 Self: Immutable,
5571 {
5572 // Note that this method does not have a `Self: Sized` bound;
5573 // `size_of_val` works for unsized values too.
5574 let len = mem::size_of_val(self);
5575 let slf: *const Self = self;
5576
5577 // SAFETY:
5578 // - `slf.cast::<u8>()` is valid for reads for `len * size_of::<u8>()`
5579 // many bytes because...
5580 // - `slf` is the same pointer as `self`, and `self` is a reference
5581 // which points to an object whose size is `len`. Thus...
5582 // - The entire region of `len` bytes starting at `slf` is contained
5583 // within a single allocation.
5584 // - `slf` is non-null.
5585 // - `slf` is trivially aligned to `align_of::<u8>() == 1`.
5586 // - `Self: IntoBytes` ensures that all of the bytes of `slf` are
5587 // initialized.
5588 // - Since `slf` is derived from `self`, and `self` is an immutable
5589 // reference, the only other references to this memory region that
5590 // could exist are other immutable references, which by `Self:
5591 // Immutable` don't permit mutation.
5592 // - The total size of the resulting slice is no larger than
5593 // `isize::MAX` because no allocation produced by safe code can be
5594 // larger than `isize::MAX`.
5595 //
5596 // FIXME(#429): Add references to docs and quotes.
5597 unsafe { slice::from_raw_parts(slf.cast::<u8>(), len) }
5598 }
5599
5600 /// Gets the bytes of this value mutably.
5601 ///
5602 /// # Examples
5603 ///
5604 /// ```
5605 /// use zerocopy::IntoBytes;
5606 /// # use zerocopy_derive::*;
5607 ///
5608 /// # #[derive(Eq, PartialEq, Debug)]
5609 /// #[derive(FromBytes, IntoBytes, Immutable)]
5610 /// #[repr(C)]
5611 /// struct PacketHeader {
5612 /// src_port: [u8; 2],
5613 /// dst_port: [u8; 2],
5614 /// length: [u8; 2],
5615 /// checksum: [u8; 2],
5616 /// }
5617 ///
5618 /// let mut header = PacketHeader {
5619 /// src_port: [0, 1],
5620 /// dst_port: [2, 3],
5621 /// length: [4, 5],
5622 /// checksum: [6, 7],
5623 /// };
5624 ///
5625 /// let bytes = header.as_mut_bytes();
5626 ///
5627 /// assert_eq!(bytes, [0, 1, 2, 3, 4, 5, 6, 7]);
5628 ///
5629 /// bytes.reverse();
5630 ///
5631 /// assert_eq!(header, PacketHeader {
5632 /// src_port: [7, 6],
5633 /// dst_port: [5, 4],
5634 /// length: [3, 2],
5635 /// checksum: [1, 0],
5636 /// });
5637 /// ```
5638 #[must_use = "has no side effects"]
5639 #[inline(always)]
5640 fn as_mut_bytes(&mut self) -> &mut [u8]
5641 where
5642 Self: FromBytes,
5643 {
5644 // Note that this method does not have a `Self: Sized` bound;
5645 // `size_of_val` works for unsized values too.
5646 let len = mem::size_of_val(self);
5647 let slf: *mut Self = self;
5648
5649 // SAFETY:
5650 // - `slf.cast::<u8>()` is valid for reads and writes for `len *
5651 // size_of::<u8>()` many bytes because...
5652 // - `slf` is the same pointer as `self`, and `self` is a reference
5653 // which points to an object whose size is `len`. Thus...
5654 // - The entire region of `len` bytes starting at `slf` is contained
5655 // within a single allocation.
5656 // - `slf` is non-null.
5657 // - `slf` is trivially aligned to `align_of::<u8>() == 1`.
5658 // - `Self: IntoBytes` ensures that all of the bytes of `slf` are
5659 // initialized.
5660 // - `Self: FromBytes` ensures that no write to this memory region
5661 // could result in it containing an invalid `Self`.
5662 // - Since `slf` is derived from `self`, and `self` is a mutable
5663 // reference, no other references to this memory region can exist.
5664 // - The total size of the resulting slice is no larger than
5665 // `isize::MAX` because no allocation produced by safe code can be
5666 // larger than `isize::MAX`.
5667 //
5668 // FIXME(#429): Add references to docs and quotes.
5669 unsafe { slice::from_raw_parts_mut(slf.cast::<u8>(), len) }
5670 }
5671
5672 /// Writes a copy of `self` to `dst`.
5673 ///
5674 /// If `dst.len() != size_of_val(self)`, `write_to` returns `Err`.
5675 ///
5676 /// # Examples
5677 ///
5678 /// ```
5679 /// use zerocopy::IntoBytes;
5680 /// # use zerocopy_derive::*;
5681 ///
5682 /// #[derive(IntoBytes, Immutable)]
5683 /// #[repr(C)]
5684 /// struct PacketHeader {
5685 /// src_port: [u8; 2],
5686 /// dst_port: [u8; 2],
5687 /// length: [u8; 2],
5688 /// checksum: [u8; 2],
5689 /// }
5690 ///
5691 /// let header = PacketHeader {
5692 /// src_port: [0, 1],
5693 /// dst_port: [2, 3],
5694 /// length: [4, 5],
5695 /// checksum: [6, 7],
5696 /// };
5697 ///
5698 /// let mut bytes = [0, 0, 0, 0, 0, 0, 0, 0];
5699 ///
5700 /// header.write_to(&mut bytes[..]);
5701 ///
5702 /// assert_eq!(bytes, [0, 1, 2, 3, 4, 5, 6, 7]);
5703 /// ```
5704 ///
5705 /// If too many or too few target bytes are provided, `write_to` returns
5706 /// `Err` and leaves the target bytes unmodified:
5707 ///
5708 /// ```
5709 /// # use zerocopy::IntoBytes;
5710 /// # let header = u128::MAX;
5711 /// let mut excessive_bytes = &mut [0u8; 128][..];
5712 ///
5713 /// let write_result = header.write_to(excessive_bytes);
5714 ///
5715 /// assert!(write_result.is_err());
5716 /// assert_eq!(excessive_bytes, [0u8; 128]);
5717 /// ```
5718 #[must_use = "callers should check the return value to see if the operation succeeded"]
5719 #[inline]
5720 #[allow(clippy::mut_from_ref)] // False positive: `&self -> &mut [u8]`
5721 fn write_to(&self, dst: &mut [u8]) -> Result<(), SizeError<&Self, &mut [u8]>>
5722 where
5723 Self: Immutable,
5724 {
5725 let src = self.as_bytes();
5726 if dst.len() == src.len() {
5727 // SAFETY: Within this branch of the conditional, we have ensured
5728 // that `dst.len()` is equal to `src.len()`. Neither the size of the
5729 // source nor the size of the destination change between the above
5730 // size check and the invocation of `copy_unchecked`.
5731 unsafe { util::copy_unchecked(src, dst) }
5732 Ok(())
5733 } else {
5734 Err(SizeError::new(self))
5735 }
5736 }
5737
5738 /// Writes a copy of `self` to the prefix of `dst`.
5739 ///
5740 /// `write_to_prefix` writes `self` to the first `size_of_val(self)` bytes
5741 /// of `dst`. If `dst.len() < size_of_val(self)`, it returns `Err`.
5742 ///
5743 /// # Examples
5744 ///
5745 /// ```
5746 /// use zerocopy::IntoBytes;
5747 /// # use zerocopy_derive::*;
5748 ///
5749 /// #[derive(IntoBytes, Immutable)]
5750 /// #[repr(C)]
5751 /// struct PacketHeader {
5752 /// src_port: [u8; 2],
5753 /// dst_port: [u8; 2],
5754 /// length: [u8; 2],
5755 /// checksum: [u8; 2],
5756 /// }
5757 ///
5758 /// let header = PacketHeader {
5759 /// src_port: [0, 1],
5760 /// dst_port: [2, 3],
5761 /// length: [4, 5],
5762 /// checksum: [6, 7],
5763 /// };
5764 ///
5765 /// let mut bytes = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0];
5766 ///
5767 /// header.write_to_prefix(&mut bytes[..]);
5768 ///
5769 /// assert_eq!(bytes, [0, 1, 2, 3, 4, 5, 6, 7, 0, 0]);
5770 /// ```
5771 ///
5772 /// If insufficient target bytes are provided, `write_to_prefix` returns
5773 /// `Err` and leaves the target bytes unmodified:
5774 ///
5775 /// ```
5776 /// # use zerocopy::IntoBytes;
5777 /// # let header = u128::MAX;
5778 /// let mut insufficient_bytes = &mut [0, 0][..];
5779 ///
5780 /// let write_result = header.write_to_suffix(insufficient_bytes);
5781 ///
5782 /// assert!(write_result.is_err());
5783 /// assert_eq!(insufficient_bytes, [0, 0]);
5784 /// ```
5785 #[must_use = "callers should check the return value to see if the operation succeeded"]
5786 #[inline]
5787 #[allow(clippy::mut_from_ref)] // False positive: `&self -> &mut [u8]`
5788 fn write_to_prefix(&self, dst: &mut [u8]) -> Result<(), SizeError<&Self, &mut [u8]>>
5789 where
5790 Self: Immutable,
5791 {
5792 let src = self.as_bytes();
5793 match dst.get_mut(..src.len()) {
5794 Some(dst) => {
5795 // SAFETY: Within this branch of the `match`, we have ensured
5796 // through fallible subslicing that `dst.len()` is equal to
5797 // `src.len()`. Neither the size of the source nor the size of
5798 // the destination change between the above subslicing operation
5799 // and the invocation of `copy_unchecked`.
5800 unsafe { util::copy_unchecked(src, dst) }
5801 Ok(())
5802 }
5803 None => Err(SizeError::new(self)),
5804 }
5805 }
5806
5807 /// Writes a copy of `self` to the suffix of `dst`.
5808 ///
5809 /// `write_to_suffix` writes `self` to the last `size_of_val(self)` bytes of
5810 /// `dst`. If `dst.len() < size_of_val(self)`, it returns `Err`.
5811 ///
5812 /// # Examples
5813 ///
5814 /// ```
5815 /// use zerocopy::IntoBytes;
5816 /// # use zerocopy_derive::*;
5817 ///
5818 /// #[derive(IntoBytes, Immutable)]
5819 /// #[repr(C)]
5820 /// struct PacketHeader {
5821 /// src_port: [u8; 2],
5822 /// dst_port: [u8; 2],
5823 /// length: [u8; 2],
5824 /// checksum: [u8; 2],
5825 /// }
5826 ///
5827 /// let header = PacketHeader {
5828 /// src_port: [0, 1],
5829 /// dst_port: [2, 3],
5830 /// length: [4, 5],
5831 /// checksum: [6, 7],
5832 /// };
5833 ///
5834 /// let mut bytes = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0];
5835 ///
5836 /// header.write_to_suffix(&mut bytes[..]);
5837 ///
5838 /// assert_eq!(bytes, [0, 0, 0, 1, 2, 3, 4, 5, 6, 7]);
5839 ///
5840 /// let mut insufficient_bytes = &mut [0, 0][..];
5841 ///
5842 /// let write_result = header.write_to_suffix(insufficient_bytes);
5843 ///
5844 /// assert!(write_result.is_err());
5845 /// assert_eq!(insufficient_bytes, [0, 0]);
5846 /// ```
5847 ///
5848 /// If insufficient target bytes are provided, `write_to_suffix` returns
5849 /// `Err` and leaves the target bytes unmodified:
5850 ///
5851 /// ```
5852 /// # use zerocopy::IntoBytes;
5853 /// # let header = u128::MAX;
5854 /// let mut insufficient_bytes = &mut [0, 0][..];
5855 ///
5856 /// let write_result = header.write_to_suffix(insufficient_bytes);
5857 ///
5858 /// assert!(write_result.is_err());
5859 /// assert_eq!(insufficient_bytes, [0, 0]);
5860 /// ```
5861 #[must_use = "callers should check the return value to see if the operation succeeded"]
5862 #[inline]
5863 #[allow(clippy::mut_from_ref)] // False positive: `&self -> &mut [u8]`
5864 fn write_to_suffix(&self, dst: &mut [u8]) -> Result<(), SizeError<&Self, &mut [u8]>>
5865 where
5866 Self: Immutable,
5867 {
5868 let src = self.as_bytes();
5869 let start = if let Some(start) = dst.len().checked_sub(src.len()) {
5870 start
5871 } else {
5872 return Err(SizeError::new(self));
5873 };
5874 let dst = if let Some(dst) = dst.get_mut(start..) {
5875 dst
5876 } else {
5877 // get_mut() should never return None here. We return a `SizeError`
5878 // rather than .unwrap() because in the event the branch is not
5879 // optimized away, returning a value is generally lighter-weight
5880 // than panicking.
5881 return Err(SizeError::new(self));
5882 };
5883 // SAFETY: Through fallible subslicing of `dst`, we have ensured that
5884 // `dst.len()` is equal to `src.len()`. Neither the size of the source
5885 // nor the size of the destination change between the above subslicing
5886 // operation and the invocation of `copy_unchecked`.
5887 unsafe {
5888 util::copy_unchecked(src, dst);
5889 }
5890 Ok(())
5891 }
5892
5893 /// Writes a copy of `self` to an `io::Write`.
5894 ///
5895 /// This is a shorthand for `dst.write_all(self.as_bytes())`, and is useful
5896 /// for interfacing with operating system byte sinks (files, sockets, etc.).
5897 ///
5898 /// # Examples
5899 ///
5900 /// ```no_run
5901 /// use zerocopy::{byteorder::big_endian::U16, FromBytes, IntoBytes};
5902 /// use std::fs::File;
5903 /// # use zerocopy_derive::*;
5904 ///
5905 /// #[derive(FromBytes, IntoBytes, Immutable, KnownLayout)]
5906 /// #[repr(C, packed)]
5907 /// struct GrayscaleImage {
5908 /// height: U16,
5909 /// width: U16,
5910 /// pixels: [U16],
5911 /// }
5912 ///
5913 /// let image = GrayscaleImage::ref_from_bytes(&[0, 0, 0, 0][..]).unwrap();
5914 /// let mut file = File::create("image.bin").unwrap();
5915 /// image.write_to_io(&mut file).unwrap();
5916 /// ```
5917 ///
5918 /// If the write fails, `write_to_io` returns `Err` and a partial write may
5919 /// have occurred; e.g.:
5920 ///
5921 /// ```
5922 /// # use zerocopy::IntoBytes;
5923 ///
5924 /// let src = u128::MAX;
5925 /// let mut dst = [0u8; 2];
5926 ///
5927 /// let write_result = src.write_to_io(&mut dst[..]);
5928 ///
5929 /// assert!(write_result.is_err());
5930 /// assert_eq!(dst, [255, 255]);
5931 /// ```
5932 #[cfg(feature = "std")]
5933 #[cfg_attr(doc_cfg, doc(cfg(feature = "std")))]
5934 #[inline(always)]
5935 fn write_to_io<W>(&self, mut dst: W) -> io::Result<()>
5936 where
5937 Self: Immutable,
5938 W: io::Write,
5939 {
5940 dst.write_all(self.as_bytes())
5941 }
5942
5943 #[deprecated(since = "0.8.0", note = "`IntoBytes::as_bytes_mut` was renamed to `as_mut_bytes`")]
5944 #[doc(hidden)]
5945 #[inline]
5946 fn as_bytes_mut(&mut self) -> &mut [u8]
5947 where
5948 Self: FromBytes,
5949 {
5950 self.as_mut_bytes()
5951 }
5952}
5953
5954/// Analyzes whether a type is [`Unaligned`].
5955///
5956/// This derive analyzes, at compile time, whether the annotated type satisfies
5957/// the [safety conditions] of `Unaligned` and implements `Unaligned` if it is
5958/// sound to do so. This derive can be applied to structs, enums, and unions;
5959/// e.g.:
5960///
5961/// ```
5962/// # use zerocopy_derive::Unaligned;
5963/// #[derive(Unaligned)]
5964/// #[repr(C)]
5965/// struct MyStruct {
5966/// # /*
5967/// ...
5968/// # */
5969/// }
5970///
5971/// #[derive(Unaligned)]
5972/// #[repr(u8)]
5973/// enum MyEnum {
5974/// # Variant0,
5975/// # /*
5976/// ...
5977/// # */
5978/// }
5979///
5980/// #[derive(Unaligned)]
5981/// #[repr(packed)]
5982/// union MyUnion {
5983/// # variant: u8,
5984/// # /*
5985/// ...
5986/// # */
5987/// }
5988/// ```
5989///
5990/// # Analysis
5991///
5992/// *This section describes, roughly, the analysis performed by this derive to
5993/// determine whether it is sound to implement `Unaligned` for a given type.
5994/// Unless you are modifying the implementation of this derive, or attempting to
5995/// manually implement `Unaligned` for a type yourself, you don't need to read
5996/// this section.*
5997///
5998/// If a type has the following properties, then this derive can implement
5999/// `Unaligned` for that type:
6000///
6001/// - If the type is a struct or union:
6002/// - If `repr(align(N))` is provided, `N` must equal 1.
6003/// - If the type is `repr(C)` or `repr(transparent)`, all fields must be
6004/// [`Unaligned`].
6005/// - If the type is not `repr(C)` or `repr(transparent)`, it must be
6006/// `repr(packed)` or `repr(packed(1))`.
6007/// - If the type is an enum:
6008/// - If `repr(align(N))` is provided, `N` must equal 1.
6009/// - It must be a field-less enum (meaning that all variants have no fields).
6010/// - It must be `repr(i8)` or `repr(u8)`.
6011///
6012/// [safety conditions]: trait@Unaligned#safety
6013#[cfg(any(feature = "derive", test))]
6014#[cfg_attr(doc_cfg, doc(cfg(feature = "derive")))]
6015pub use zerocopy_derive::Unaligned;
6016
6017/// Types with no alignment requirement.
6018///
6019/// If `T: Unaligned`, then `align_of::<T>() == 1`.
6020///
6021/// # Implementation
6022///
6023/// **Do not implement this trait yourself!** Instead, use
6024/// [`#[derive(Unaligned)]`][derive]; e.g.:
6025///
6026/// ```
6027/// # use zerocopy_derive::Unaligned;
6028/// #[derive(Unaligned)]
6029/// #[repr(C)]
6030/// struct MyStruct {
6031/// # /*
6032/// ...
6033/// # */
6034/// }
6035///
6036/// #[derive(Unaligned)]
6037/// #[repr(u8)]
6038/// enum MyEnum {
6039/// # Variant0,
6040/// # /*
6041/// ...
6042/// # */
6043/// }
6044///
6045/// #[derive(Unaligned)]
6046/// #[repr(packed)]
6047/// union MyUnion {
6048/// # variant: u8,
6049/// # /*
6050/// ...
6051/// # */
6052/// }
6053/// ```
6054///
6055/// This derive performs a sophisticated, compile-time safety analysis to
6056/// determine whether a type is `Unaligned`.
6057///
6058/// # Safety
6059///
6060/// *This section describes what is required in order for `T: Unaligned`, and
6061/// what unsafe code may assume of such types. If you don't plan on implementing
6062/// `Unaligned` manually, and you don't plan on writing unsafe code that
6063/// operates on `Unaligned` types, then you don't need to read this section.*
6064///
6065/// If `T: Unaligned`, then unsafe code may assume that it is sound to produce a
6066/// reference to `T` at any memory location regardless of alignment. If a type
6067/// is marked as `Unaligned` which violates this contract, it may cause
6068/// undefined behavior.
6069///
6070/// `#[derive(Unaligned)]` only permits [types which satisfy these
6071/// requirements][derive-analysis].
6072///
6073#[cfg_attr(
6074 feature = "derive",
6075 doc = "[derive]: zerocopy_derive::Unaligned",
6076 doc = "[derive-analysis]: zerocopy_derive::Unaligned#analysis"
6077)]
6078#[cfg_attr(
6079 not(feature = "derive"),
6080 doc = concat!("[derive]: https://docs.rs/zerocopy/", env!("CARGO_PKG_VERSION"), "/zerocopy/derive.Unaligned.html"),
6081 doc = concat!("[derive-analysis]: https://docs.rs/zerocopy/", env!("CARGO_PKG_VERSION"), "/zerocopy/derive.Unaligned.html#analysis"),
6082)]
6083#[cfg_attr(
6084 not(no_zerocopy_diagnostic_on_unimplemented_1_78_0),
6085 diagnostic::on_unimplemented(note = "Consider adding `#[derive(Unaligned)]` to `{Self}`")
6086)]
6087pub unsafe trait Unaligned {
6088 // The `Self: Sized` bound makes it so that `Unaligned` is still object
6089 // safe.
6090 #[doc(hidden)]
6091 fn only_derive_is_allowed_to_implement_this_trait()
6092 where
6093 Self: Sized;
6094}
6095
6096/// Derives optimized [`PartialEq`] and [`Eq`] implementations.
6097///
6098/// This derive can be applied to structs and enums implementing both
6099/// [`Immutable`] and [`IntoBytes`]; e.g.:
6100///
6101/// ```
6102/// # use zerocopy_derive::{ByteEq, Immutable, IntoBytes};
6103/// #[derive(ByteEq, Immutable, IntoBytes)]
6104/// #[repr(C)]
6105/// struct MyStruct {
6106/// # /*
6107/// ...
6108/// # */
6109/// }
6110///
6111/// #[derive(ByteEq, Immutable, IntoBytes)]
6112/// #[repr(u8)]
6113/// enum MyEnum {
6114/// # Variant,
6115/// # /*
6116/// ...
6117/// # */
6118/// }
6119/// ```
6120///
6121/// The standard library's [`derive(Eq, PartialEq)`][derive@PartialEq] computes
6122/// equality by individually comparing each field. Instead, the implementation
6123/// of [`PartialEq::eq`] emitted by `derive(ByteHash)` converts the entirety of
6124/// `self` and `other` to byte slices and compares those slices for equality.
6125/// This may have performance advantages.
6126#[cfg(any(feature = "derive", test))]
6127#[cfg_attr(doc_cfg, doc(cfg(feature = "derive")))]
6128pub use zerocopy_derive::ByteEq;
6129/// Derives an optimized [`Hash`] implementation.
6130///
6131/// This derive can be applied to structs and enums implementing both
6132/// [`Immutable`] and [`IntoBytes`]; e.g.:
6133///
6134/// ```
6135/// # use zerocopy_derive::{ByteHash, Immutable, IntoBytes};
6136/// #[derive(ByteHash, Immutable, IntoBytes)]
6137/// #[repr(C)]
6138/// struct MyStruct {
6139/// # /*
6140/// ...
6141/// # */
6142/// }
6143///
6144/// #[derive(ByteHash, Immutable, IntoBytes)]
6145/// #[repr(u8)]
6146/// enum MyEnum {
6147/// # Variant,
6148/// # /*
6149/// ...
6150/// # */
6151/// }
6152/// ```
6153///
6154/// The standard library's [`derive(Hash)`][derive@Hash] produces hashes by
6155/// individually hashing each field and combining the results. Instead, the
6156/// implementations of [`Hash::hash()`] and [`Hash::hash_slice()`] generated by
6157/// `derive(ByteHash)` convert the entirety of `self` to a byte slice and hashes
6158/// it in a single call to [`Hasher::write()`]. This may have performance
6159/// advantages.
6160///
6161/// [`Hash`]: core::hash::Hash
6162/// [`Hash::hash()`]: core::hash::Hash::hash()
6163/// [`Hash::hash_slice()`]: core::hash::Hash::hash_slice()
6164#[cfg(any(feature = "derive", test))]
6165#[cfg_attr(doc_cfg, doc(cfg(feature = "derive")))]
6166pub use zerocopy_derive::ByteHash;
6167/// Implements [`SplitAt`].
6168///
6169/// This derive can be applied to structs; e.g.:
6170///
6171/// ```
6172/// # use zerocopy_derive::{ByteEq, Immutable, IntoBytes};
6173/// #[derive(ByteEq, Immutable, IntoBytes)]
6174/// #[repr(C)]
6175/// struct MyStruct {
6176/// # /*
6177/// ...
6178/// # */
6179/// }
6180/// ```
6181#[cfg(any(feature = "derive", test))]
6182#[cfg_attr(doc_cfg, doc(cfg(feature = "derive")))]
6183pub use zerocopy_derive::SplitAt;
6184
6185#[cfg(feature = "alloc")]
6186#[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
6187#[cfg(not(no_zerocopy_panic_in_const_and_vec_try_reserve_1_57_0))]
6188mod alloc_support {
6189 use super::*;
6190
6191 /// Extends a `Vec<T>` by pushing `additional` new items onto the end of the
6192 /// vector. The new items are initialized with zeros.
6193 #[cfg(not(no_zerocopy_panic_in_const_and_vec_try_reserve_1_57_0))]
6194 #[doc(hidden)]
6195 #[deprecated(since = "0.8.0", note = "moved to `FromZeros`")]
6196 #[inline(always)]
6197 pub fn extend_vec_zeroed<T: FromZeros>(
6198 v: &mut Vec<T>,
6199 additional: usize,
6200 ) -> Result<(), AllocError> {
6201 <T as FromZeros>::extend_vec_zeroed(v, additional)
6202 }
6203
6204 /// Inserts `additional` new items into `Vec<T>` at `position`. The new
6205 /// items are initialized with zeros.
6206 ///
6207 /// # Panics
6208 ///
6209 /// Panics if `position > v.len()`.
6210 #[cfg(not(no_zerocopy_panic_in_const_and_vec_try_reserve_1_57_0))]
6211 #[doc(hidden)]
6212 #[deprecated(since = "0.8.0", note = "moved to `FromZeros`")]
6213 #[inline(always)]
6214 pub fn insert_vec_zeroed<T: FromZeros>(
6215 v: &mut Vec<T>,
6216 position: usize,
6217 additional: usize,
6218 ) -> Result<(), AllocError> {
6219 <T as FromZeros>::insert_vec_zeroed(v, position, additional)
6220 }
6221}
6222
6223#[cfg(feature = "alloc")]
6224#[cfg(not(no_zerocopy_panic_in_const_and_vec_try_reserve_1_57_0))]
6225#[doc(hidden)]
6226pub use alloc_support::*;
6227
6228#[cfg(test)]
6229#[allow(clippy::assertions_on_result_states, clippy::unreadable_literal)]
6230mod tests {
6231 use static_assertions::assert_impl_all;
6232
6233 use super::*;
6234 use crate::util::testutil::*;
6235
6236 // An unsized type.
6237 //
6238 // This is used to test the custom derives of our traits. The `[u8]` type
6239 // gets a hand-rolled impl, so it doesn't exercise our custom derives.
6240 #[derive(Debug, Eq, PartialEq, FromBytes, IntoBytes, Unaligned, Immutable)]
6241 #[repr(transparent)]
6242 struct Unsized([u8]);
6243
6244 impl Unsized {
6245 fn from_mut_slice(slc: &mut [u8]) -> &mut Unsized {
6246 // SAFETY: This *probably* sound - since the layouts of `[u8]` and
6247 // `Unsized` are the same, so are the layouts of `&mut [u8]` and
6248 // `&mut Unsized`. [1] Even if it turns out that this isn't actually
6249 // guaranteed by the language spec, we can just change this since
6250 // it's in test code.
6251 //
6252 // [1] https://github.com/rust-lang/unsafe-code-guidelines/issues/375
6253 unsafe { mem::transmute(slc) }
6254 }
6255 }
6256
6257 #[test]
6258 fn test_known_layout() {
6259 // Test that `$ty` and `ManuallyDrop<$ty>` have the expected layout.
6260 // Test that `PhantomData<$ty>` has the same layout as `()` regardless
6261 // of `$ty`.
6262 macro_rules! test {
6263 ($ty:ty, $expect:expr) => {
6264 let expect = $expect;
6265 assert_eq!(<$ty as KnownLayout>::LAYOUT, expect);
6266 assert_eq!(<ManuallyDrop<$ty> as KnownLayout>::LAYOUT, expect);
6267 assert_eq!(<PhantomData<$ty> as KnownLayout>::LAYOUT, <() as KnownLayout>::LAYOUT);
6268 };
6269 }
6270
6271 let layout =
6272 |offset, align, trailing_slice_elem_size, statically_shallow_unpadded| DstLayout {
6273 align: NonZeroUsize::new(align).unwrap(),
6274 size_info: match trailing_slice_elem_size {
6275 None => SizeInfo::Sized { size: offset },
6276 Some(elem_size) => {
6277 SizeInfo::SliceDst(TrailingSliceLayout { offset, elem_size })
6278 }
6279 },
6280 statically_shallow_unpadded,
6281 };
6282
6283 test!((), layout(0, 1, None, false));
6284 test!(u8, layout(1, 1, None, false));
6285 // Use `align_of` because `u64` alignment may be smaller than 8 on some
6286 // platforms.
6287 test!(u64, layout(8, mem::align_of::<u64>(), None, false));
6288 test!(AU64, layout(8, 8, None, false));
6289
6290 test!(Option<&'static ()>, usize::LAYOUT);
6291
6292 test!([()], layout(0, 1, Some(0), true));
6293 test!([u8], layout(0, 1, Some(1), true));
6294 test!(str, layout(0, 1, Some(1), true));
6295 }
6296
6297 #[cfg(feature = "derive")]
6298 #[test]
6299 fn test_known_layout_derive() {
6300 // In this and other files (`late_compile_pass.rs`,
6301 // `mid_compile_pass.rs`, and `struct.rs`), we test success and failure
6302 // modes of `derive(KnownLayout)` for the following combination of
6303 // properties:
6304 //
6305 // +------------+--------------------------------------+-----------+
6306 // | | trailing field properties | |
6307 // | `repr(C)`? | generic? | `KnownLayout`? | `Sized`? | Type Name |
6308 // |------------+----------+----------------+----------+-----------|
6309 // | N | N | N | N | KL00 |
6310 // | N | N | N | Y | KL01 |
6311 // | N | N | Y | N | KL02 |
6312 // | N | N | Y | Y | KL03 |
6313 // | N | Y | N | N | KL04 |
6314 // | N | Y | N | Y | KL05 |
6315 // | N | Y | Y | N | KL06 |
6316 // | N | Y | Y | Y | KL07 |
6317 // | Y | N | N | N | KL08 |
6318 // | Y | N | N | Y | KL09 |
6319 // | Y | N | Y | N | KL10 |
6320 // | Y | N | Y | Y | KL11 |
6321 // | Y | Y | N | N | KL12 |
6322 // | Y | Y | N | Y | KL13 |
6323 // | Y | Y | Y | N | KL14 |
6324 // | Y | Y | Y | Y | KL15 |
6325 // +------------+----------+----------------+----------+-----------+
6326
6327 struct NotKnownLayout<T = ()> {
6328 _t: T,
6329 }
6330
6331 #[derive(KnownLayout)]
6332 #[repr(C)]
6333 struct AlignSize<const ALIGN: usize, const SIZE: usize>
6334 where
6335 elain::Align<ALIGN>: elain::Alignment,
6336 {
6337 _align: elain::Align<ALIGN>,
6338 size: [u8; SIZE],
6339 }
6340
6341 type AU16 = AlignSize<2, 2>;
6342 type AU32 = AlignSize<4, 4>;
6343
6344 fn _assert_kl<T: ?Sized + KnownLayout>(_: &T) {}
6345
6346 let sized_layout = |align, size| DstLayout {
6347 align: NonZeroUsize::new(align).unwrap(),
6348 size_info: SizeInfo::Sized { size },
6349 statically_shallow_unpadded: false,
6350 };
6351
6352 let unsized_layout = |align, elem_size, offset, statically_shallow_unpadded| DstLayout {
6353 align: NonZeroUsize::new(align).unwrap(),
6354 size_info: SizeInfo::SliceDst(TrailingSliceLayout { offset, elem_size }),
6355 statically_shallow_unpadded,
6356 };
6357
6358 // | `repr(C)`? | generic? | `KnownLayout`? | `Sized`? | Type Name |
6359 // | N | N | N | Y | KL01 |
6360 #[allow(dead_code)]
6361 #[derive(KnownLayout)]
6362 struct KL01(NotKnownLayout<AU32>, NotKnownLayout<AU16>);
6363
6364 let expected = DstLayout::for_type::<KL01>();
6365
6366 assert_eq!(<KL01 as KnownLayout>::LAYOUT, expected);
6367 assert_eq!(<KL01 as KnownLayout>::LAYOUT, sized_layout(4, 8));
6368
6369 // ...with `align(N)`:
6370 #[allow(dead_code)]
6371 #[derive(KnownLayout)]
6372 #[repr(align(64))]
6373 struct KL01Align(NotKnownLayout<AU32>, NotKnownLayout<AU16>);
6374
6375 let expected = DstLayout::for_type::<KL01Align>();
6376
6377 assert_eq!(<KL01Align as KnownLayout>::LAYOUT, expected);
6378 assert_eq!(<KL01Align as KnownLayout>::LAYOUT, sized_layout(64, 64));
6379
6380 // ...with `packed`:
6381 #[allow(dead_code)]
6382 #[derive(KnownLayout)]
6383 #[repr(packed)]
6384 struct KL01Packed(NotKnownLayout<AU32>, NotKnownLayout<AU16>);
6385
6386 let expected = DstLayout::for_type::<KL01Packed>();
6387
6388 assert_eq!(<KL01Packed as KnownLayout>::LAYOUT, expected);
6389 assert_eq!(<KL01Packed as KnownLayout>::LAYOUT, sized_layout(1, 6));
6390
6391 // ...with `packed(N)`:
6392 #[allow(dead_code)]
6393 #[derive(KnownLayout)]
6394 #[repr(packed(2))]
6395 struct KL01PackedN(NotKnownLayout<AU32>, NotKnownLayout<AU16>);
6396
6397 assert_impl_all!(KL01PackedN: KnownLayout);
6398
6399 let expected = DstLayout::for_type::<KL01PackedN>();
6400
6401 assert_eq!(<KL01PackedN as KnownLayout>::LAYOUT, expected);
6402 assert_eq!(<KL01PackedN as KnownLayout>::LAYOUT, sized_layout(2, 6));
6403
6404 // | `repr(C)`? | generic? | `KnownLayout`? | `Sized`? | Type Name |
6405 // | N | N | Y | Y | KL03 |
6406 #[allow(dead_code)]
6407 #[derive(KnownLayout)]
6408 struct KL03(NotKnownLayout, u8);
6409
6410 let expected = DstLayout::for_type::<KL03>();
6411
6412 assert_eq!(<KL03 as KnownLayout>::LAYOUT, expected);
6413 assert_eq!(<KL03 as KnownLayout>::LAYOUT, sized_layout(1, 1));
6414
6415 // ... with `align(N)`
6416 #[allow(dead_code)]
6417 #[derive(KnownLayout)]
6418 #[repr(align(64))]
6419 struct KL03Align(NotKnownLayout<AU32>, u8);
6420
6421 let expected = DstLayout::for_type::<KL03Align>();
6422
6423 assert_eq!(<KL03Align as KnownLayout>::LAYOUT, expected);
6424 assert_eq!(<KL03Align as KnownLayout>::LAYOUT, sized_layout(64, 64));
6425
6426 // ... with `packed`:
6427 #[allow(dead_code)]
6428 #[derive(KnownLayout)]
6429 #[repr(packed)]
6430 struct KL03Packed(NotKnownLayout<AU32>, u8);
6431
6432 let expected = DstLayout::for_type::<KL03Packed>();
6433
6434 assert_eq!(<KL03Packed as KnownLayout>::LAYOUT, expected);
6435 assert_eq!(<KL03Packed as KnownLayout>::LAYOUT, sized_layout(1, 5));
6436
6437 // ... with `packed(N)`
6438 #[allow(dead_code)]
6439 #[derive(KnownLayout)]
6440 #[repr(packed(2))]
6441 struct KL03PackedN(NotKnownLayout<AU32>, u8);
6442
6443 assert_impl_all!(KL03PackedN: KnownLayout);
6444
6445 let expected = DstLayout::for_type::<KL03PackedN>();
6446
6447 assert_eq!(<KL03PackedN as KnownLayout>::LAYOUT, expected);
6448 assert_eq!(<KL03PackedN as KnownLayout>::LAYOUT, sized_layout(2, 6));
6449
6450 // | `repr(C)`? | generic? | `KnownLayout`? | `Sized`? | Type Name |
6451 // | N | Y | N | Y | KL05 |
6452 #[allow(dead_code)]
6453 #[derive(KnownLayout)]
6454 struct KL05<T>(u8, T);
6455
6456 fn _test_kl05<T>(t: T) -> impl KnownLayout {
6457 KL05(0u8, t)
6458 }
6459
6460 // | `repr(C)`? | generic? | `KnownLayout`? | `Sized`? | Type Name |
6461 // | N | Y | Y | Y | KL07 |
6462 #[allow(dead_code)]
6463 #[derive(KnownLayout)]
6464 struct KL07<T: KnownLayout>(u8, T);
6465
6466 fn _test_kl07<T: KnownLayout>(t: T) -> impl KnownLayout {
6467 let _ = KL07(0u8, t);
6468 }
6469
6470 // | `repr(C)`? | generic? | `KnownLayout`? | `Sized`? | Type Name |
6471 // | Y | N | Y | N | KL10 |
6472 #[allow(dead_code)]
6473 #[derive(KnownLayout)]
6474 #[repr(C)]
6475 struct KL10(NotKnownLayout<AU32>, [u8]);
6476
6477 let expected = DstLayout::new_zst(None)
6478 .extend(DstLayout::for_type::<NotKnownLayout<AU32>>(), None)
6479 .extend(<[u8] as KnownLayout>::LAYOUT, None)
6480 .pad_to_align();
6481
6482 assert_eq!(<KL10 as KnownLayout>::LAYOUT, expected);
6483 assert_eq!(<KL10 as KnownLayout>::LAYOUT, unsized_layout(4, 1, 4, false));
6484
6485 // ...with `align(N)`:
6486 #[allow(dead_code)]
6487 #[derive(KnownLayout)]
6488 #[repr(C, align(64))]
6489 struct KL10Align(NotKnownLayout<AU32>, [u8]);
6490
6491 let repr_align = NonZeroUsize::new(64);
6492
6493 let expected = DstLayout::new_zst(repr_align)
6494 .extend(DstLayout::for_type::<NotKnownLayout<AU32>>(), None)
6495 .extend(<[u8] as KnownLayout>::LAYOUT, None)
6496 .pad_to_align();
6497
6498 assert_eq!(<KL10Align as KnownLayout>::LAYOUT, expected);
6499 assert_eq!(<KL10Align as KnownLayout>::LAYOUT, unsized_layout(64, 1, 4, false));
6500
6501 // ...with `packed`:
6502 #[allow(dead_code)]
6503 #[derive(KnownLayout)]
6504 #[repr(C, packed)]
6505 struct KL10Packed(NotKnownLayout<AU32>, [u8]);
6506
6507 let repr_packed = NonZeroUsize::new(1);
6508
6509 let expected = DstLayout::new_zst(None)
6510 .extend(DstLayout::for_type::<NotKnownLayout<AU32>>(), repr_packed)
6511 .extend(<[u8] as KnownLayout>::LAYOUT, repr_packed)
6512 .pad_to_align();
6513
6514 assert_eq!(<KL10Packed as KnownLayout>::LAYOUT, expected);
6515 assert_eq!(<KL10Packed as KnownLayout>::LAYOUT, unsized_layout(1, 1, 4, false));
6516
6517 // ...with `packed(N)`:
6518 #[allow(dead_code)]
6519 #[derive(KnownLayout)]
6520 #[repr(C, packed(2))]
6521 struct KL10PackedN(NotKnownLayout<AU32>, [u8]);
6522
6523 let repr_packed = NonZeroUsize::new(2);
6524
6525 let expected = DstLayout::new_zst(None)
6526 .extend(DstLayout::for_type::<NotKnownLayout<AU32>>(), repr_packed)
6527 .extend(<[u8] as KnownLayout>::LAYOUT, repr_packed)
6528 .pad_to_align();
6529
6530 assert_eq!(<KL10PackedN as KnownLayout>::LAYOUT, expected);
6531 assert_eq!(<KL10PackedN as KnownLayout>::LAYOUT, unsized_layout(2, 1, 4, false));
6532
6533 // | `repr(C)`? | generic? | `KnownLayout`? | `Sized`? | Type Name |
6534 // | Y | N | Y | Y | KL11 |
6535 #[allow(dead_code)]
6536 #[derive(KnownLayout)]
6537 #[repr(C)]
6538 struct KL11(NotKnownLayout<AU64>, u8);
6539
6540 let expected = DstLayout::new_zst(None)
6541 .extend(DstLayout::for_type::<NotKnownLayout<AU64>>(), None)
6542 .extend(<u8 as KnownLayout>::LAYOUT, None)
6543 .pad_to_align();
6544
6545 assert_eq!(<KL11 as KnownLayout>::LAYOUT, expected);
6546 assert_eq!(<KL11 as KnownLayout>::LAYOUT, sized_layout(8, 16));
6547
6548 // ...with `align(N)`:
6549 #[allow(dead_code)]
6550 #[derive(KnownLayout)]
6551 #[repr(C, align(64))]
6552 struct KL11Align(NotKnownLayout<AU64>, u8);
6553
6554 let repr_align = NonZeroUsize::new(64);
6555
6556 let expected = DstLayout::new_zst(repr_align)
6557 .extend(DstLayout::for_type::<NotKnownLayout<AU64>>(), None)
6558 .extend(<u8 as KnownLayout>::LAYOUT, None)
6559 .pad_to_align();
6560
6561 assert_eq!(<KL11Align as KnownLayout>::LAYOUT, expected);
6562 assert_eq!(<KL11Align as KnownLayout>::LAYOUT, sized_layout(64, 64));
6563
6564 // ...with `packed`:
6565 #[allow(dead_code)]
6566 #[derive(KnownLayout)]
6567 #[repr(C, packed)]
6568 struct KL11Packed(NotKnownLayout<AU64>, u8);
6569
6570 let repr_packed = NonZeroUsize::new(1);
6571
6572 let expected = DstLayout::new_zst(None)
6573 .extend(DstLayout::for_type::<NotKnownLayout<AU64>>(), repr_packed)
6574 .extend(<u8 as KnownLayout>::LAYOUT, repr_packed)
6575 .pad_to_align();
6576
6577 assert_eq!(<KL11Packed as KnownLayout>::LAYOUT, expected);
6578 assert_eq!(<KL11Packed as KnownLayout>::LAYOUT, sized_layout(1, 9));
6579
6580 // ...with `packed(N)`:
6581 #[allow(dead_code)]
6582 #[derive(KnownLayout)]
6583 #[repr(C, packed(2))]
6584 struct KL11PackedN(NotKnownLayout<AU64>, u8);
6585
6586 let repr_packed = NonZeroUsize::new(2);
6587
6588 let expected = DstLayout::new_zst(None)
6589 .extend(DstLayout::for_type::<NotKnownLayout<AU64>>(), repr_packed)
6590 .extend(<u8 as KnownLayout>::LAYOUT, repr_packed)
6591 .pad_to_align();
6592
6593 assert_eq!(<KL11PackedN as KnownLayout>::LAYOUT, expected);
6594 assert_eq!(<KL11PackedN as KnownLayout>::LAYOUT, sized_layout(2, 10));
6595
6596 // | `repr(C)`? | generic? | `KnownLayout`? | `Sized`? | Type Name |
6597 // | Y | Y | Y | N | KL14 |
6598 #[allow(dead_code)]
6599 #[derive(KnownLayout)]
6600 #[repr(C)]
6601 struct KL14<T: ?Sized + KnownLayout>(u8, T);
6602
6603 fn _test_kl14<T: ?Sized + KnownLayout>(kl: &KL14<T>) {
6604 _assert_kl(kl)
6605 }
6606
6607 // | `repr(C)`? | generic? | `KnownLayout`? | `Sized`? | Type Name |
6608 // | Y | Y | Y | Y | KL15 |
6609 #[allow(dead_code)]
6610 #[derive(KnownLayout)]
6611 #[repr(C)]
6612 struct KL15<T: KnownLayout>(u8, T);
6613
6614 fn _test_kl15<T: KnownLayout>(t: T) -> impl KnownLayout {
6615 let _ = KL15(0u8, t);
6616 }
6617
6618 // Test a variety of combinations of field types:
6619 // - ()
6620 // - u8
6621 // - AU16
6622 // - [()]
6623 // - [u8]
6624 // - [AU16]
6625
6626 #[allow(clippy::upper_case_acronyms, dead_code)]
6627 #[derive(KnownLayout)]
6628 #[repr(C)]
6629 struct KLTU<T, U: ?Sized>(T, U);
6630
6631 assert_eq!(<KLTU<(), ()> as KnownLayout>::LAYOUT, sized_layout(1, 0));
6632
6633 assert_eq!(<KLTU<(), u8> as KnownLayout>::LAYOUT, sized_layout(1, 1));
6634
6635 assert_eq!(<KLTU<(), AU16> as KnownLayout>::LAYOUT, sized_layout(2, 2));
6636
6637 assert_eq!(<KLTU<(), [()]> as KnownLayout>::LAYOUT, unsized_layout(1, 0, 0, false));
6638
6639 assert_eq!(<KLTU<(), [u8]> as KnownLayout>::LAYOUT, unsized_layout(1, 1, 0, false));
6640
6641 assert_eq!(<KLTU<(), [AU16]> as KnownLayout>::LAYOUT, unsized_layout(2, 2, 0, false));
6642
6643 assert_eq!(<KLTU<u8, ()> as KnownLayout>::LAYOUT, sized_layout(1, 1));
6644
6645 assert_eq!(<KLTU<u8, u8> as KnownLayout>::LAYOUT, sized_layout(1, 2));
6646
6647 assert_eq!(<KLTU<u8, AU16> as KnownLayout>::LAYOUT, sized_layout(2, 4));
6648
6649 assert_eq!(<KLTU<u8, [()]> as KnownLayout>::LAYOUT, unsized_layout(1, 0, 1, false));
6650
6651 assert_eq!(<KLTU<u8, [u8]> as KnownLayout>::LAYOUT, unsized_layout(1, 1, 1, false));
6652
6653 assert_eq!(<KLTU<u8, [AU16]> as KnownLayout>::LAYOUT, unsized_layout(2, 2, 2, false));
6654
6655 assert_eq!(<KLTU<AU16, ()> as KnownLayout>::LAYOUT, sized_layout(2, 2));
6656
6657 assert_eq!(<KLTU<AU16, u8> as KnownLayout>::LAYOUT, sized_layout(2, 4));
6658
6659 assert_eq!(<KLTU<AU16, AU16> as KnownLayout>::LAYOUT, sized_layout(2, 4));
6660
6661 assert_eq!(<KLTU<AU16, [()]> as KnownLayout>::LAYOUT, unsized_layout(2, 0, 2, false));
6662
6663 assert_eq!(<KLTU<AU16, [u8]> as KnownLayout>::LAYOUT, unsized_layout(2, 1, 2, false));
6664
6665 assert_eq!(<KLTU<AU16, [AU16]> as KnownLayout>::LAYOUT, unsized_layout(2, 2, 2, false));
6666
6667 // Test a variety of field counts.
6668
6669 #[derive(KnownLayout)]
6670 #[repr(C)]
6671 struct KLF0;
6672
6673 assert_eq!(<KLF0 as KnownLayout>::LAYOUT, sized_layout(1, 0));
6674
6675 #[derive(KnownLayout)]
6676 #[repr(C)]
6677 struct KLF1([u8]);
6678
6679 assert_eq!(<KLF1 as KnownLayout>::LAYOUT, unsized_layout(1, 1, 0, true));
6680
6681 #[derive(KnownLayout)]
6682 #[repr(C)]
6683 struct KLF2(NotKnownLayout<u8>, [u8]);
6684
6685 assert_eq!(<KLF2 as KnownLayout>::LAYOUT, unsized_layout(1, 1, 1, false));
6686
6687 #[derive(KnownLayout)]
6688 #[repr(C)]
6689 struct KLF3(NotKnownLayout<u8>, NotKnownLayout<AU16>, [u8]);
6690
6691 assert_eq!(<KLF3 as KnownLayout>::LAYOUT, unsized_layout(2, 1, 4, false));
6692
6693 #[derive(KnownLayout)]
6694 #[repr(C)]
6695 struct KLF4(NotKnownLayout<u8>, NotKnownLayout<AU16>, NotKnownLayout<AU32>, [u8]);
6696
6697 assert_eq!(<KLF4 as KnownLayout>::LAYOUT, unsized_layout(4, 1, 8, false));
6698 }
6699
6700 #[test]
6701 fn test_object_safety() {
6702 fn _takes_immutable(_: &dyn Immutable) {}
6703 fn _takes_unaligned(_: &dyn Unaligned) {}
6704 }
6705
6706 #[test]
6707 fn test_from_zeros_only() {
6708 // Test types that implement `FromZeros` but not `FromBytes`.
6709
6710 assert!(!bool::new_zeroed());
6711 assert_eq!(char::new_zeroed(), '\0');
6712
6713 #[cfg(feature = "alloc")]
6714 {
6715 assert_eq!(bool::new_box_zeroed(), Ok(Box::new(false)));
6716 assert_eq!(char::new_box_zeroed(), Ok(Box::new('\0')));
6717
6718 assert_eq!(
6719 <[bool]>::new_box_zeroed_with_elems(3).unwrap().as_ref(),
6720 [false, false, false]
6721 );
6722 assert_eq!(
6723 <[char]>::new_box_zeroed_with_elems(3).unwrap().as_ref(),
6724 ['\0', '\0', '\0']
6725 );
6726
6727 assert_eq!(bool::new_vec_zeroed(3).unwrap().as_ref(), [false, false, false]);
6728 assert_eq!(char::new_vec_zeroed(3).unwrap().as_ref(), ['\0', '\0', '\0']);
6729 }
6730
6731 let mut string = "hello".to_string();
6732 let s: &mut str = string.as_mut();
6733 assert_eq!(s, "hello");
6734 s.zero();
6735 assert_eq!(s, "\0\0\0\0\0");
6736 }
6737
6738 #[test]
6739 fn test_zst_count_preserved() {
6740 // Test that, when an explicit count is provided to for a type with a
6741 // ZST trailing slice element, that count is preserved. This is
6742 // important since, for such types, all element counts result in objects
6743 // of the same size, and so the correct behavior is ambiguous. However,
6744 // preserving the count as requested by the user is the behavior that we
6745 // document publicly.
6746
6747 // FromZeros methods
6748 #[cfg(feature = "alloc")]
6749 assert_eq!(<[()]>::new_box_zeroed_with_elems(3).unwrap().len(), 3);
6750 #[cfg(feature = "alloc")]
6751 assert_eq!(<()>::new_vec_zeroed(3).unwrap().len(), 3);
6752
6753 // FromBytes methods
6754 assert_eq!(<[()]>::ref_from_bytes_with_elems(&[][..], 3).unwrap().len(), 3);
6755 assert_eq!(<[()]>::ref_from_prefix_with_elems(&[][..], 3).unwrap().0.len(), 3);
6756 assert_eq!(<[()]>::ref_from_suffix_with_elems(&[][..], 3).unwrap().1.len(), 3);
6757 assert_eq!(<[()]>::mut_from_bytes_with_elems(&mut [][..], 3).unwrap().len(), 3);
6758 assert_eq!(<[()]>::mut_from_prefix_with_elems(&mut [][..], 3).unwrap().0.len(), 3);
6759 assert_eq!(<[()]>::mut_from_suffix_with_elems(&mut [][..], 3).unwrap().1.len(), 3);
6760 }
6761
6762 #[test]
6763 fn test_read_write() {
6764 const VAL: u64 = 0x12345678;
6765 #[cfg(target_endian = "big")]
6766 const VAL_BYTES: [u8; 8] = VAL.to_be_bytes();
6767 #[cfg(target_endian = "little")]
6768 const VAL_BYTES: [u8; 8] = VAL.to_le_bytes();
6769 const ZEROS: [u8; 8] = [0u8; 8];
6770
6771 // Test `FromBytes::{read_from, read_from_prefix, read_from_suffix}`.
6772
6773 assert_eq!(u64::read_from_bytes(&VAL_BYTES[..]), Ok(VAL));
6774 // The first 8 bytes are from `VAL_BYTES` and the second 8 bytes are all
6775 // zeros.
6776 let bytes_with_prefix: [u8; 16] = transmute!([VAL_BYTES, [0; 8]]);
6777 assert_eq!(u64::read_from_prefix(&bytes_with_prefix[..]), Ok((VAL, &ZEROS[..])));
6778 assert_eq!(u64::read_from_suffix(&bytes_with_prefix[..]), Ok((&VAL_BYTES[..], 0)));
6779 // The first 8 bytes are all zeros and the second 8 bytes are from
6780 // `VAL_BYTES`
6781 let bytes_with_suffix: [u8; 16] = transmute!([[0; 8], VAL_BYTES]);
6782 assert_eq!(u64::read_from_prefix(&bytes_with_suffix[..]), Ok((0, &VAL_BYTES[..])));
6783 assert_eq!(u64::read_from_suffix(&bytes_with_suffix[..]), Ok((&ZEROS[..], VAL)));
6784
6785 // Test `IntoBytes::{write_to, write_to_prefix, write_to_suffix}`.
6786
6787 let mut bytes = [0u8; 8];
6788 assert_eq!(VAL.write_to(&mut bytes[..]), Ok(()));
6789 assert_eq!(bytes, VAL_BYTES);
6790 let mut bytes = [0u8; 16];
6791 assert_eq!(VAL.write_to_prefix(&mut bytes[..]), Ok(()));
6792 let want: [u8; 16] = transmute!([VAL_BYTES, [0; 8]]);
6793 assert_eq!(bytes, want);
6794 let mut bytes = [0u8; 16];
6795 assert_eq!(VAL.write_to_suffix(&mut bytes[..]), Ok(()));
6796 let want: [u8; 16] = transmute!([[0; 8], VAL_BYTES]);
6797 assert_eq!(bytes, want);
6798 }
6799
6800 #[test]
6801 #[cfg(feature = "std")]
6802 fn test_read_io_with_padding_soundness() {
6803 // This test is designed to exhibit potential UB in
6804 // `FromBytes::read_from_io`. (see #2319, #2320).
6805
6806 // On most platforms (where `align_of::<u16>() == 2`), `WithPadding`
6807 // will have inter-field padding between `x` and `y`.
6808 #[derive(FromBytes)]
6809 #[repr(C)]
6810 struct WithPadding {
6811 x: u8,
6812 y: u16,
6813 }
6814 struct ReadsInRead;
6815 impl std::io::Read for ReadsInRead {
6816 fn read(&mut self, buf: &mut [u8]) -> std::io::Result<usize> {
6817 // This body branches on every byte of `buf`, ensuring that it
6818 // exhibits UB if any byte of `buf` is uninitialized.
6819 if buf.iter().all(|&x| x == 0) {
6820 Ok(buf.len())
6821 } else {
6822 buf.iter_mut().for_each(|x| *x = 0);
6823 Ok(buf.len())
6824 }
6825 }
6826 }
6827 assert!(matches!(WithPadding::read_from_io(ReadsInRead), Ok(WithPadding { x: 0, y: 0 })));
6828 }
6829
6830 #[test]
6831 #[cfg(feature = "std")]
6832 fn test_read_write_io() {
6833 let mut long_buffer = [0, 0, 0, 0];
6834 assert!(matches!(u16::MAX.write_to_io(&mut long_buffer[..]), Ok(())));
6835 assert_eq!(long_buffer, [255, 255, 0, 0]);
6836 assert!(matches!(u16::read_from_io(&long_buffer[..]), Ok(u16::MAX)));
6837
6838 let mut short_buffer = [0, 0];
6839 assert!(u32::MAX.write_to_io(&mut short_buffer[..]).is_err());
6840 assert_eq!(short_buffer, [255, 255]);
6841 assert!(u32::read_from_io(&short_buffer[..]).is_err());
6842 }
6843
6844 #[test]
6845 fn test_try_from_bytes_try_read_from() {
6846 assert_eq!(<bool as TryFromBytes>::try_read_from_bytes(&[0]), Ok(false));
6847 assert_eq!(<bool as TryFromBytes>::try_read_from_bytes(&[1]), Ok(true));
6848
6849 assert_eq!(<bool as TryFromBytes>::try_read_from_prefix(&[0, 2]), Ok((false, &[2][..])));
6850 assert_eq!(<bool as TryFromBytes>::try_read_from_prefix(&[1, 2]), Ok((true, &[2][..])));
6851
6852 assert_eq!(<bool as TryFromBytes>::try_read_from_suffix(&[2, 0]), Ok((&[2][..], false)));
6853 assert_eq!(<bool as TryFromBytes>::try_read_from_suffix(&[2, 1]), Ok((&[2][..], true)));
6854
6855 // If we don't pass enough bytes, it fails.
6856 assert!(matches!(
6857 <u8 as TryFromBytes>::try_read_from_bytes(&[]),
6858 Err(TryReadError::Size(_))
6859 ));
6860 assert!(matches!(
6861 <u8 as TryFromBytes>::try_read_from_prefix(&[]),
6862 Err(TryReadError::Size(_))
6863 ));
6864 assert!(matches!(
6865 <u8 as TryFromBytes>::try_read_from_suffix(&[]),
6866 Err(TryReadError::Size(_))
6867 ));
6868
6869 // If we pass too many bytes, it fails.
6870 assert!(matches!(
6871 <u8 as TryFromBytes>::try_read_from_bytes(&[0, 0]),
6872 Err(TryReadError::Size(_))
6873 ));
6874
6875 // If we pass an invalid value, it fails.
6876 assert!(matches!(
6877 <bool as TryFromBytes>::try_read_from_bytes(&[2]),
6878 Err(TryReadError::Validity(_))
6879 ));
6880 assert!(matches!(
6881 <bool as TryFromBytes>::try_read_from_prefix(&[2, 0]),
6882 Err(TryReadError::Validity(_))
6883 ));
6884 assert!(matches!(
6885 <bool as TryFromBytes>::try_read_from_suffix(&[0, 2]),
6886 Err(TryReadError::Validity(_))
6887 ));
6888
6889 // Reading from a misaligned buffer should still succeed. Since `AU64`'s
6890 // alignment is 8, and since we read from two adjacent addresses one
6891 // byte apart, it is guaranteed that at least one of them (though
6892 // possibly both) will be misaligned.
6893 let bytes: [u8; 9] = [0, 0, 0, 0, 0, 0, 0, 0, 0];
6894 assert_eq!(<AU64 as TryFromBytes>::try_read_from_bytes(&bytes[..8]), Ok(AU64(0)));
6895 assert_eq!(<AU64 as TryFromBytes>::try_read_from_bytes(&bytes[1..9]), Ok(AU64(0)));
6896
6897 assert_eq!(
6898 <AU64 as TryFromBytes>::try_read_from_prefix(&bytes[..8]),
6899 Ok((AU64(0), &[][..]))
6900 );
6901 assert_eq!(
6902 <AU64 as TryFromBytes>::try_read_from_prefix(&bytes[1..9]),
6903 Ok((AU64(0), &[][..]))
6904 );
6905
6906 assert_eq!(
6907 <AU64 as TryFromBytes>::try_read_from_suffix(&bytes[..8]),
6908 Ok((&[][..], AU64(0)))
6909 );
6910 assert_eq!(
6911 <AU64 as TryFromBytes>::try_read_from_suffix(&bytes[1..9]),
6912 Ok((&[][..], AU64(0)))
6913 );
6914 }
6915
6916 #[test]
6917 fn test_ref_from_mut_from_bytes() {
6918 // Test `FromBytes::{ref_from_bytes, mut_from_bytes}{,_prefix,Suffix}`
6919 // success cases. Exhaustive coverage for these methods is covered by
6920 // the `Ref` tests above, which these helper methods defer to.
6921
6922 let mut buf =
6923 Align::<[u8; 16], AU64>::new([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]);
6924
6925 assert_eq!(
6926 AU64::ref_from_bytes(&buf.t[8..]).unwrap().0.to_ne_bytes(),
6927 [8, 9, 10, 11, 12, 13, 14, 15]
6928 );
6929 let suffix = AU64::mut_from_bytes(&mut buf.t[8..]).unwrap();
6930 suffix.0 = 0x0101010101010101;
6931 // The `[u8:9]` is a non-half size of the full buffer, which would catch
6932 // `from_prefix` having the same implementation as `from_suffix` (issues #506, #511).
6933 assert_eq!(
6934 <[u8; 9]>::ref_from_suffix(&buf.t[..]).unwrap(),
6935 (&[0, 1, 2, 3, 4, 5, 6][..], &[7u8, 1, 1, 1, 1, 1, 1, 1, 1])
6936 );
6937 let (prefix, suffix) = AU64::mut_from_suffix(&mut buf.t[1..]).unwrap();
6938 assert_eq!(prefix, &mut [1u8, 2, 3, 4, 5, 6, 7][..]);
6939 suffix.0 = 0x0202020202020202;
6940 let (prefix, suffix) = <[u8; 10]>::mut_from_suffix(&mut buf.t[..]).unwrap();
6941 assert_eq!(prefix, &mut [0u8, 1, 2, 3, 4, 5][..]);
6942 suffix[0] = 42;
6943 assert_eq!(
6944 <[u8; 9]>::ref_from_prefix(&buf.t[..]).unwrap(),
6945 (&[0u8, 1, 2, 3, 4, 5, 42, 7, 2], &[2u8, 2, 2, 2, 2, 2, 2][..])
6946 );
6947 <[u8; 2]>::mut_from_prefix(&mut buf.t[..]).unwrap().0[1] = 30;
6948 assert_eq!(buf.t, [0, 30, 2, 3, 4, 5, 42, 7, 2, 2, 2, 2, 2, 2, 2, 2]);
6949 }
6950
6951 #[test]
6952 fn test_ref_from_mut_from_bytes_error() {
6953 // Test `FromBytes::{ref_from_bytes, mut_from_bytes}{,_prefix,Suffix}`
6954 // error cases.
6955
6956 // Fail because the buffer is too large.
6957 let mut buf = Align::<[u8; 16], AU64>::default();
6958 // `buf.t` should be aligned to 8, so only the length check should fail.
6959 assert!(AU64::ref_from_bytes(&buf.t[..]).is_err());
6960 assert!(AU64::mut_from_bytes(&mut buf.t[..]).is_err());
6961 assert!(<[u8; 8]>::ref_from_bytes(&buf.t[..]).is_err());
6962 assert!(<[u8; 8]>::mut_from_bytes(&mut buf.t[..]).is_err());
6963
6964 // Fail because the buffer is too small.
6965 let mut buf = Align::<[u8; 4], AU64>::default();
6966 assert!(AU64::ref_from_bytes(&buf.t[..]).is_err());
6967 assert!(AU64::mut_from_bytes(&mut buf.t[..]).is_err());
6968 assert!(<[u8; 8]>::ref_from_bytes(&buf.t[..]).is_err());
6969 assert!(<[u8; 8]>::mut_from_bytes(&mut buf.t[..]).is_err());
6970 assert!(AU64::ref_from_prefix(&buf.t[..]).is_err());
6971 assert!(AU64::mut_from_prefix(&mut buf.t[..]).is_err());
6972 assert!(AU64::ref_from_suffix(&buf.t[..]).is_err());
6973 assert!(AU64::mut_from_suffix(&mut buf.t[..]).is_err());
6974 assert!(<[u8; 8]>::ref_from_prefix(&buf.t[..]).is_err());
6975 assert!(<[u8; 8]>::mut_from_prefix(&mut buf.t[..]).is_err());
6976 assert!(<[u8; 8]>::ref_from_suffix(&buf.t[..]).is_err());
6977 assert!(<[u8; 8]>::mut_from_suffix(&mut buf.t[..]).is_err());
6978
6979 // Fail because the alignment is insufficient.
6980 let mut buf = Align::<[u8; 13], AU64>::default();
6981 assert!(AU64::ref_from_bytes(&buf.t[1..]).is_err());
6982 assert!(AU64::mut_from_bytes(&mut buf.t[1..]).is_err());
6983 assert!(AU64::ref_from_bytes(&buf.t[1..]).is_err());
6984 assert!(AU64::mut_from_bytes(&mut buf.t[1..]).is_err());
6985 assert!(AU64::ref_from_prefix(&buf.t[1..]).is_err());
6986 assert!(AU64::mut_from_prefix(&mut buf.t[1..]).is_err());
6987 assert!(AU64::ref_from_suffix(&buf.t[..]).is_err());
6988 assert!(AU64::mut_from_suffix(&mut buf.t[..]).is_err());
6989 }
6990
6991 #[test]
6992 fn test_to_methods() {
6993 /// Run a series of tests by calling `IntoBytes` methods on `t`.
6994 ///
6995 /// `bytes` is the expected byte sequence returned from `t.as_bytes()`
6996 /// before `t` has been modified. `post_mutation` is the expected
6997 /// sequence returned from `t.as_bytes()` after `t.as_mut_bytes()[0]`
6998 /// has had its bits flipped (by applying `^= 0xFF`).
6999 ///
7000 /// `N` is the size of `t` in bytes.
7001 fn test<T: FromBytes + IntoBytes + Immutable + Debug + Eq + ?Sized, const N: usize>(
7002 t: &mut T,
7003 bytes: &[u8],
7004 post_mutation: &T,
7005 ) {
7006 // Test that we can access the underlying bytes, and that we get the
7007 // right bytes and the right number of bytes.
7008 assert_eq!(t.as_bytes(), bytes);
7009
7010 // Test that changes to the underlying byte slices are reflected in
7011 // the original object.
7012 t.as_mut_bytes()[0] ^= 0xFF;
7013 assert_eq!(t, post_mutation);
7014 t.as_mut_bytes()[0] ^= 0xFF;
7015
7016 // `write_to` rejects slices that are too small or too large.
7017 assert!(t.write_to(&mut vec![0; N - 1][..]).is_err());
7018 assert!(t.write_to(&mut vec![0; N + 1][..]).is_err());
7019
7020 // `write_to` works as expected.
7021 let mut bytes = [0; N];
7022 assert_eq!(t.write_to(&mut bytes[..]), Ok(()));
7023 assert_eq!(bytes, t.as_bytes());
7024
7025 // `write_to_prefix` rejects slices that are too small.
7026 assert!(t.write_to_prefix(&mut vec![0; N - 1][..]).is_err());
7027
7028 // `write_to_prefix` works with exact-sized slices.
7029 let mut bytes = [0; N];
7030 assert_eq!(t.write_to_prefix(&mut bytes[..]), Ok(()));
7031 assert_eq!(bytes, t.as_bytes());
7032
7033 // `write_to_prefix` works with too-large slices, and any bytes past
7034 // the prefix aren't modified.
7035 let mut too_many_bytes = vec![0; N + 1];
7036 too_many_bytes[N] = 123;
7037 assert_eq!(t.write_to_prefix(&mut too_many_bytes[..]), Ok(()));
7038 assert_eq!(&too_many_bytes[..N], t.as_bytes());
7039 assert_eq!(too_many_bytes[N], 123);
7040
7041 // `write_to_suffix` rejects slices that are too small.
7042 assert!(t.write_to_suffix(&mut vec![0; N - 1][..]).is_err());
7043
7044 // `write_to_suffix` works with exact-sized slices.
7045 let mut bytes = [0; N];
7046 assert_eq!(t.write_to_suffix(&mut bytes[..]), Ok(()));
7047 assert_eq!(bytes, t.as_bytes());
7048
7049 // `write_to_suffix` works with too-large slices, and any bytes
7050 // before the suffix aren't modified.
7051 let mut too_many_bytes = vec![0; N + 1];
7052 too_many_bytes[0] = 123;
7053 assert_eq!(t.write_to_suffix(&mut too_many_bytes[..]), Ok(()));
7054 assert_eq!(&too_many_bytes[1..], t.as_bytes());
7055 assert_eq!(too_many_bytes[0], 123);
7056 }
7057
7058 #[derive(Debug, Eq, PartialEq, FromBytes, IntoBytes, Immutable)]
7059 #[repr(C)]
7060 struct Foo {
7061 a: u32,
7062 b: Wrapping<u32>,
7063 c: Option<NonZeroU32>,
7064 }
7065
7066 let expected_bytes: Vec<u8> = if cfg!(target_endian = "little") {
7067 vec![1, 0, 0, 0, 2, 0, 0, 0, 0, 0, 0, 0]
7068 } else {
7069 vec![0, 0, 0, 1, 0, 0, 0, 2, 0, 0, 0, 0]
7070 };
7071 let post_mutation_expected_a =
7072 if cfg!(target_endian = "little") { 0x00_00_00_FE } else { 0xFF_00_00_01 };
7073 test::<_, 12>(
7074 &mut Foo { a: 1, b: Wrapping(2), c: None },
7075 expected_bytes.as_bytes(),
7076 &Foo { a: post_mutation_expected_a, b: Wrapping(2), c: None },
7077 );
7078 test::<_, 3>(
7079 Unsized::from_mut_slice(&mut [1, 2, 3]),
7080 &[1, 2, 3],
7081 Unsized::from_mut_slice(&mut [0xFE, 2, 3]),
7082 );
7083 }
7084
7085 #[test]
7086 fn test_array() {
7087 #[derive(FromBytes, IntoBytes, Immutable)]
7088 #[repr(C)]
7089 struct Foo {
7090 a: [u16; 33],
7091 }
7092
7093 let foo = Foo { a: [0xFFFF; 33] };
7094 let expected = [0xFFu8; 66];
7095 assert_eq!(foo.as_bytes(), &expected[..]);
7096 }
7097
7098 #[test]
7099 fn test_new_zeroed() {
7100 assert!(!bool::new_zeroed());
7101 assert_eq!(u64::new_zeroed(), 0);
7102 // This test exists in order to exercise unsafe code, especially when
7103 // running under Miri.
7104 #[allow(clippy::unit_cmp)]
7105 {
7106 assert_eq!(<()>::new_zeroed(), ());
7107 }
7108 }
7109
7110 #[test]
7111 fn test_transparent_packed_generic_struct() {
7112 #[derive(IntoBytes, FromBytes, Unaligned)]
7113 #[repr(transparent)]
7114 #[allow(dead_code)] // We never construct this type
7115 struct Foo<T> {
7116 _t: T,
7117 _phantom: PhantomData<()>,
7118 }
7119
7120 assert_impl_all!(Foo<u32>: FromZeros, FromBytes, IntoBytes);
7121 assert_impl_all!(Foo<u8>: Unaligned);
7122
7123 #[derive(IntoBytes, FromBytes, Unaligned)]
7124 #[repr(C, packed)]
7125 #[allow(dead_code)] // We never construct this type
7126 struct Bar<T, U> {
7127 _t: T,
7128 _u: U,
7129 }
7130
7131 assert_impl_all!(Bar<u8, AU64>: FromZeros, FromBytes, IntoBytes, Unaligned);
7132 }
7133
7134 #[cfg(feature = "alloc")]
7135 mod alloc {
7136 use super::*;
7137
7138 #[cfg(not(no_zerocopy_panic_in_const_and_vec_try_reserve_1_57_0))]
7139 #[test]
7140 fn test_extend_vec_zeroed() {
7141 // Test extending when there is an existing allocation.
7142 let mut v = vec![100u16, 200, 300];
7143 FromZeros::extend_vec_zeroed(&mut v, 3).unwrap();
7144 assert_eq!(v.len(), 6);
7145 assert_eq!(&*v, &[100, 200, 300, 0, 0, 0]);
7146 drop(v);
7147
7148 // Test extending when there is no existing allocation.
7149 let mut v: Vec<u64> = Vec::new();
7150 FromZeros::extend_vec_zeroed(&mut v, 3).unwrap();
7151 assert_eq!(v.len(), 3);
7152 assert_eq!(&*v, &[0, 0, 0]);
7153 drop(v);
7154 }
7155
7156 #[cfg(not(no_zerocopy_panic_in_const_and_vec_try_reserve_1_57_0))]
7157 #[test]
7158 fn test_extend_vec_zeroed_zst() {
7159 // Test extending when there is an existing (fake) allocation.
7160 let mut v = vec![(), (), ()];
7161 <()>::extend_vec_zeroed(&mut v, 3).unwrap();
7162 assert_eq!(v.len(), 6);
7163 assert_eq!(&*v, &[(), (), (), (), (), ()]);
7164 drop(v);
7165
7166 // Test extending when there is no existing (fake) allocation.
7167 let mut v: Vec<()> = Vec::new();
7168 <()>::extend_vec_zeroed(&mut v, 3).unwrap();
7169 assert_eq!(&*v, &[(), (), ()]);
7170 drop(v);
7171 }
7172
7173 #[cfg(not(no_zerocopy_panic_in_const_and_vec_try_reserve_1_57_0))]
7174 #[test]
7175 fn test_insert_vec_zeroed() {
7176 // Insert at start (no existing allocation).
7177 let mut v: Vec<u64> = Vec::new();
7178 u64::insert_vec_zeroed(&mut v, 0, 2).unwrap();
7179 assert_eq!(v.len(), 2);
7180 assert_eq!(&*v, &[0, 0]);
7181 drop(v);
7182
7183 // Insert at start.
7184 let mut v = vec![100u64, 200, 300];
7185 u64::insert_vec_zeroed(&mut v, 0, 2).unwrap();
7186 assert_eq!(v.len(), 5);
7187 assert_eq!(&*v, &[0, 0, 100, 200, 300]);
7188 drop(v);
7189
7190 // Insert at middle.
7191 let mut v = vec![100u64, 200, 300];
7192 u64::insert_vec_zeroed(&mut v, 1, 1).unwrap();
7193 assert_eq!(v.len(), 4);
7194 assert_eq!(&*v, &[100, 0, 200, 300]);
7195 drop(v);
7196
7197 // Insert at end.
7198 let mut v = vec![100u64, 200, 300];
7199 u64::insert_vec_zeroed(&mut v, 3, 1).unwrap();
7200 assert_eq!(v.len(), 4);
7201 assert_eq!(&*v, &[100, 200, 300, 0]);
7202 drop(v);
7203 }
7204
7205 #[cfg(not(no_zerocopy_panic_in_const_and_vec_try_reserve_1_57_0))]
7206 #[test]
7207 fn test_insert_vec_zeroed_zst() {
7208 // Insert at start (no existing fake allocation).
7209 let mut v: Vec<()> = Vec::new();
7210 <()>::insert_vec_zeroed(&mut v, 0, 2).unwrap();
7211 assert_eq!(v.len(), 2);
7212 assert_eq!(&*v, &[(), ()]);
7213 drop(v);
7214
7215 // Insert at start.
7216 let mut v = vec![(), (), ()];
7217 <()>::insert_vec_zeroed(&mut v, 0, 2).unwrap();
7218 assert_eq!(v.len(), 5);
7219 assert_eq!(&*v, &[(), (), (), (), ()]);
7220 drop(v);
7221
7222 // Insert at middle.
7223 let mut v = vec![(), (), ()];
7224 <()>::insert_vec_zeroed(&mut v, 1, 1).unwrap();
7225 assert_eq!(v.len(), 4);
7226 assert_eq!(&*v, &[(), (), (), ()]);
7227 drop(v);
7228
7229 // Insert at end.
7230 let mut v = vec![(), (), ()];
7231 <()>::insert_vec_zeroed(&mut v, 3, 1).unwrap();
7232 assert_eq!(v.len(), 4);
7233 assert_eq!(&*v, &[(), (), (), ()]);
7234 drop(v);
7235 }
7236
7237 #[test]
7238 fn test_new_box_zeroed() {
7239 assert_eq!(u64::new_box_zeroed(), Ok(Box::new(0)));
7240 }
7241
7242 #[test]
7243 fn test_new_box_zeroed_array() {
7244 drop(<[u32; 0x1000]>::new_box_zeroed());
7245 }
7246
7247 #[test]
7248 fn test_new_box_zeroed_zst() {
7249 // This test exists in order to exercise unsafe code, especially
7250 // when running under Miri.
7251 #[allow(clippy::unit_cmp)]
7252 {
7253 assert_eq!(<()>::new_box_zeroed(), Ok(Box::new(())));
7254 }
7255 }
7256
7257 #[test]
7258 fn test_new_box_zeroed_with_elems() {
7259 let mut s: Box<[u64]> = <[u64]>::new_box_zeroed_with_elems(3).unwrap();
7260 assert_eq!(s.len(), 3);
7261 assert_eq!(&*s, &[0, 0, 0]);
7262 s[1] = 3;
7263 assert_eq!(&*s, &[0, 3, 0]);
7264 }
7265
7266 #[test]
7267 fn test_new_box_zeroed_with_elems_empty() {
7268 let s: Box<[u64]> = <[u64]>::new_box_zeroed_with_elems(0).unwrap();
7269 assert_eq!(s.len(), 0);
7270 }
7271
7272 #[test]
7273 fn test_new_box_zeroed_with_elems_zst() {
7274 let mut s: Box<[()]> = <[()]>::new_box_zeroed_with_elems(3).unwrap();
7275 assert_eq!(s.len(), 3);
7276 assert!(s.get(10).is_none());
7277 // This test exists in order to exercise unsafe code, especially
7278 // when running under Miri.
7279 #[allow(clippy::unit_cmp)]
7280 {
7281 assert_eq!(s[1], ());
7282 }
7283 s[2] = ();
7284 }
7285
7286 #[test]
7287 fn test_new_box_zeroed_with_elems_zst_empty() {
7288 let s: Box<[()]> = <[()]>::new_box_zeroed_with_elems(0).unwrap();
7289 assert_eq!(s.len(), 0);
7290 }
7291
7292 #[test]
7293 fn new_box_zeroed_with_elems_errors() {
7294 assert_eq!(<[u16]>::new_box_zeroed_with_elems(usize::MAX), Err(AllocError));
7295
7296 let max = <usize as core::convert::TryFrom<_>>::try_from(isize::MAX).unwrap();
7297 assert_eq!(
7298 <[u16]>::new_box_zeroed_with_elems((max / mem::size_of::<u16>()) + 1),
7299 Err(AllocError)
7300 );
7301 }
7302 }
7303
7304 #[test]
7305 #[allow(deprecated)]
7306 fn test_deprecated_from_bytes() {
7307 let val = 0u32;
7308 let bytes = val.as_bytes();
7309
7310 assert!(u32::ref_from(bytes).is_some());
7311 // mut_from needs mut bytes
7312 let mut val = 0u32;
7313 let mut_bytes = val.as_mut_bytes();
7314 assert!(u32::mut_from(mut_bytes).is_some());
7315
7316 assert!(u32::read_from(bytes).is_some());
7317
7318 let (slc, rest) = <u32>::slice_from_prefix(bytes, 0).unwrap();
7319 assert!(slc.is_empty());
7320 assert_eq!(rest.len(), 4);
7321
7322 let (rest, slc) = <u32>::slice_from_suffix(bytes, 0).unwrap();
7323 assert!(slc.is_empty());
7324 assert_eq!(rest.len(), 4);
7325
7326 let (slc, rest) = <u32>::mut_slice_from_prefix(mut_bytes, 0).unwrap();
7327 assert!(slc.is_empty());
7328 assert_eq!(rest.len(), 4);
7329
7330 let (rest, slc) = <u32>::mut_slice_from_suffix(mut_bytes, 0).unwrap();
7331 assert!(slc.is_empty());
7332 assert_eq!(rest.len(), 4);
7333 }
7334
7335 #[test]
7336 fn test_try_ref_from_prefix_suffix() {
7337 use crate::util::testutil::Align;
7338 let bytes = &Align::<[u8; 4], u32>::new([0u8; 4]).t[..];
7339 let (r, rest): (&u32, &[u8]) = u32::try_ref_from_prefix(bytes).unwrap();
7340 assert_eq!(*r, 0);
7341 assert_eq!(rest.len(), 0);
7342
7343 let (rest, r): (&[u8], &u32) = u32::try_ref_from_suffix(bytes).unwrap();
7344 assert_eq!(*r, 0);
7345 assert_eq!(rest.len(), 0);
7346 }
7347
7348 #[test]
7349 fn test_raw_dangling() {
7350 use crate::util::AsAddress;
7351 let ptr: NonNull<u32> = u32::raw_dangling();
7352 assert_eq!(AsAddress::addr(ptr), 1);
7353
7354 let ptr: NonNull<[u32]> = <[u32]>::raw_dangling();
7355 assert_eq!(AsAddress::addr(ptr), 1);
7356 }
7357
7358 #[test]
7359 fn test_try_ref_from_prefix_with_elems() {
7360 use crate::util::testutil::Align;
7361 let bytes = &Align::<[u8; 8], u32>::new([0u8; 8]).t[..];
7362 let (r, rest): (&[u32], &[u8]) = <[u32]>::try_ref_from_prefix_with_elems(bytes, 2).unwrap();
7363 assert_eq!(r.len(), 2);
7364 assert_eq!(rest.len(), 0);
7365 }
7366
7367 #[test]
7368 fn test_try_ref_from_suffix_with_elems() {
7369 use crate::util::testutil::Align;
7370 let bytes = &Align::<[u8; 8], u32>::new([0u8; 8]).t[..];
7371 let (rest, r): (&[u8], &[u32]) = <[u32]>::try_ref_from_suffix_with_elems(bytes, 2).unwrap();
7372 assert_eq!(r.len(), 2);
7373 assert_eq!(rest.len(), 0);
7374 }
7375}