Skip to main content

core/mem/
mod.rs

1//! Basic functions for dealing with memory, values, and types.
2//!
3//! The contents of this module can be seen as belonging to a few families:
4//!
5//! * [`drop`], [`replace`], [`swap`], and [`take`]
6//!   are safe functions for moving values in particular ways.
7//!   They are useful in everyday Rust code.
8//!
9//! * [`size_of`], [`size_of_val`], [`align_of`], [`align_of_val`], and [`offset_of`]
10//!   give information about the representation of values in memory.
11//!
12//! * [`discriminant`]
13//!   allows comparing the variants of [`enum`] values while ignoring their fields.
14//!
15//! * [`forget`] and [`ManuallyDrop`]
16//!   prevent destructors from running, which is used in certain kinds of ownership transfer.
17//!   [`needs_drop`]
18//!   tells you whether a type’s destructor even does anything.
19//!
20//! * [`transmute`], [`transmute_copy`], and [`MaybeUninit`]
21//!   convert and construct values in [`unsafe`] ways.
22//!
23//! See also the [`alloc`] and [`ptr`] modules for more primitive operations on memory.
24//!
25// core::alloc exists but doesn’t contain all the items we want to discuss
26//! [`alloc`]: ../../std/alloc/index.html
27//! [`enum`]: ../../std/keyword.enum.html
28//! [`ptr`]: crate::ptr
29//! [`unsafe`]: ../../std/keyword.unsafe.html
30
31#![stable(feature = "rust1", since = "1.0.0")]
32
33use crate::alloc::Layout;
34use crate::clone::TrivialClone;
35use crate::cmp::Ordering;
36use crate::marker::{Destruct, DiscriminantKind};
37use crate::panic::const_assert;
38use crate::ub_checks::assert_unsafe_precondition;
39use crate::{clone, cmp, fmt, hash, intrinsics, ptr};
40
41mod alignment;
42#[unstable(feature = "ptr_alignment_type", issue = "102070")]
43pub use alignment::Alignment;
44
45mod manually_drop;
46#[stable(feature = "manually_drop", since = "1.20.0")]
47pub use manually_drop::ManuallyDrop;
48
49mod maybe_uninit;
50#[stable(feature = "maybe_uninit", since = "1.36.0")]
51pub use maybe_uninit::MaybeUninit;
52
53mod maybe_dangling;
54#[unstable(feature = "maybe_dangling", issue = "118166")]
55pub use maybe_dangling::MaybeDangling;
56
57mod transmutability;
58#[unstable(feature = "transmutability", issue = "99571")]
59pub use transmutability::{Assume, TransmuteFrom};
60
61mod drop_guard;
62#[unstable(feature = "drop_guard", issue = "144426")]
63pub use drop_guard::DropGuard;
64
65// This one has to be a re-export (rather than wrapping the underlying intrinsic) so that we can do
66// the special magic "types have equal size" check at the call site.
67#[stable(feature = "rust1", since = "1.0.0")]
68#[doc(inline)]
69pub use crate::intrinsics::transmute;
70
71#[unstable(feature = "type_info", issue = "146922")]
72pub mod type_info;
73
74/// Takes ownership and "forgets" about the value **without running its destructor**.
75///
76/// Any resources the value manages, such as heap memory or a file handle, will linger
77/// forever in an unreachable state. However, it does not guarantee that pointers
78/// to this memory will remain valid.
79///
80/// * If you want to leak memory, see [`Box::leak`].
81/// * If you want to obtain a raw pointer to the memory, see [`Box::into_raw`].
82/// * If you want to dispose of a value properly, running its destructor, see
83///   [`mem::drop`].
84///
85/// # Safety
86///
87/// `forget` is not marked as `unsafe`, because Rust's safety guarantees
88/// do not include a guarantee that destructors will always run. For example,
89/// a program can create a reference cycle using [`Rc`][rc], or call
90/// [`process::exit`][exit] to exit without running destructors. Thus, allowing
91/// `mem::forget` from safe code does not fundamentally change Rust's safety
92/// guarantees.
93///
94/// That said, leaking resources such as memory or I/O objects is usually undesirable.
95/// The need comes up in some specialized use cases for FFI or unsafe code, but even
96/// then, [`ManuallyDrop`] is typically preferred.
97///
98/// Because forgetting a value is allowed, any `unsafe` code you write must
99/// allow for this possibility. You cannot return a value and expect that the
100/// caller will necessarily run the value's destructor.
101///
102/// [rc]: ../../std/rc/struct.Rc.html
103/// [exit]: ../../std/process/fn.exit.html
104///
105/// # Examples
106///
107/// The canonical safe use of `mem::forget` is to circumvent a value's destructor
108/// implemented by the `Drop` trait. For example, this will leak a `File`, i.e. reclaim
109/// the space taken by the variable but never close the underlying system resource:
110///
111/// ```no_run
112/// use std::mem;
113/// use std::fs::File;
114///
115/// let file = File::open("foo.txt").unwrap();
116/// mem::forget(file);
117/// ```
118///
119/// This is useful when the ownership of the underlying resource was previously
120/// transferred to code outside of Rust, for example by transmitting the raw
121/// file descriptor to C code.
122///
123/// # Relationship with `ManuallyDrop`
124///
125/// While `mem::forget` can also be used to transfer *memory* ownership, doing so is error-prone.
126/// [`ManuallyDrop`] should be used instead. Consider, for example, this code:
127///
128/// ```
129/// use std::mem;
130///
131/// let mut v = vec![65, 122];
132/// // Build a `String` using the contents of `v`
133/// let s = unsafe { String::from_raw_parts(v.as_mut_ptr(), v.len(), v.capacity()) };
134/// // leak `v` because its memory is now managed by `s`
135/// mem::forget(v);  // ERROR - v is invalid and must not be passed to a function
136/// assert_eq!(s, "Az");
137/// // `s` is implicitly dropped and its memory deallocated.
138/// ```
139///
140/// There are two issues with the above example:
141///
142/// * If more code were added between the construction of `String` and the invocation of
143///   `mem::forget()`, a panic within it would cause a double free because the same memory
144///   is handled by both `v` and `s`.
145/// * After calling `v.as_mut_ptr()` and transmitting the ownership of the data to `s`,
146///   the `v` value is invalid. Even when a value is just moved to `mem::forget` (which won't
147///   inspect it), some types have strict requirements on their values that
148///   make them invalid when dangling or no longer owned. Using invalid values in any
149///   way, including passing them to or returning them from functions, constitutes
150///   undefined behavior and may break the assumptions made by the compiler.
151///
152/// Switching to `ManuallyDrop` avoids both issues:
153///
154/// ```
155/// use std::mem::ManuallyDrop;
156///
157/// let v = vec![65, 122];
158/// // Before we disassemble `v` into its raw parts, make sure it
159/// // does not get dropped!
160/// let mut v = ManuallyDrop::new(v);
161/// // Now disassemble `v`. These operations cannot panic, so there cannot be a leak.
162/// let (ptr, len, cap) = (v.as_mut_ptr(), v.len(), v.capacity());
163/// // Finally, build a `String`.
164/// let s = unsafe { String::from_raw_parts(ptr, len, cap) };
165/// assert_eq!(s, "Az");
166/// // `s` is implicitly dropped and its memory deallocated.
167/// ```
168///
169/// `ManuallyDrop` robustly prevents double-free because we disable `v`'s destructor
170/// before doing anything else. `mem::forget()` doesn't allow this because it consumes its
171/// argument, forcing us to call it only after extracting anything we need from `v`. Even
172/// if a panic were introduced between construction of `ManuallyDrop` and building the
173/// string (which cannot happen in the code as shown), it would result in a leak and not a
174/// double free. In other words, `ManuallyDrop` errs on the side of leaking instead of
175/// erring on the side of (double-)dropping.
176///
177/// Also, `ManuallyDrop` prevents us from having to "touch" `v` after transferring the
178/// ownership to `s` — the final step of interacting with `v` to dispose of it without
179/// running its destructor is entirely avoided.
180///
181/// [`Box`]: ../../std/boxed/struct.Box.html
182/// [`Box::leak`]: ../../std/boxed/struct.Box.html#method.leak
183/// [`Box::into_raw`]: ../../std/boxed/struct.Box.html#method.into_raw
184/// [`mem::drop`]: drop
185/// [ub]: ../../reference/behavior-considered-undefined.html
186#[inline]
187#[rustc_const_stable(feature = "const_forget", since = "1.46.0")]
188#[stable(feature = "rust1", since = "1.0.0")]
189#[rustc_diagnostic_item = "mem_forget"]
190#[rustc_no_writable]
191#[ferrocene::prevalidated]
192pub const fn forget<T>(t: T) {
193    let _ = ManuallyDrop::new(t);
194}
195
196/// Like [`forget`], but also accepts unsized values.
197///
198/// While Rust does not permit unsized locals since its removal in [#111942] it is
199/// still possible to call functions with unsized values from a function argument
200/// or place expression.
201///
202/// ```rust
203/// #![feature(unsized_fn_params, forget_unsized)]
204/// #![allow(internal_features)]
205///
206/// use std::mem::forget_unsized;
207///
208/// pub fn in_place() {
209///     forget_unsized(*Box::<str>::from("str"));
210/// }
211///
212/// pub fn param(x: str) {
213///     forget_unsized(x);
214/// }
215/// ```
216///
217/// This works because the compiler will alter these functions to pass the parameter
218/// by reference instead. This trick is necessary to support `Box<dyn FnOnce()>: FnOnce()`.
219/// See [#68304] and [#71170] for more information.
220///
221/// [#111942]: https://github.com/rust-lang/rust/issues/111942
222/// [#68304]: https://github.com/rust-lang/rust/issues/68304
223/// [#71170]: https://github.com/rust-lang/rust/pull/71170
224#[inline]
225#[unstable(feature = "forget_unsized", issue = "none")]
226pub fn forget_unsized<T: ?Sized>(t: T) {
227    intrinsics::forget(t)
228}
229
230/// Returns the size of a type in bytes.
231///
232/// More specifically, this is the offset in bytes between successive elements
233/// in an array with that item type including alignment padding. Thus, for any
234/// type `T` and length `n`, `[T; n]` has a size of `n * size_of::<T>()`.
235///
236/// In general, the size of a type is not stable across compilations, but
237/// specific types such as primitives are.
238///
239/// The following table gives the size for primitives.
240///
241/// Type | `size_of::<Type>()`
242/// ---- | ---------------
243/// () | 0
244/// bool | 1
245/// u8 | 1
246/// u16 | 2
247/// u32 | 4
248/// u64 | 8
249/// u128 | 16
250/// i8 | 1
251/// i16 | 2
252/// i32 | 4
253/// i64 | 8
254/// i128 | 16
255/// f32 | 4
256/// f64 | 8
257/// char | 4
258///
259/// Furthermore, `usize` and `isize` have the same size.
260///
261/// The types [`*const T`], `&T`, [`Box<T>`], [`Option<&T>`], and `Option<Box<T>>` all have
262/// the same size. If `T` is `Sized`, all of those types have the same size as `usize`.
263///
264/// The mutability of a pointer does not change its size. As such, `&T` and `&mut T`
265/// have the same size. Likewise for `*const T` and `*mut T`.
266///
267/// # Size of `#[repr(C)]` items
268///
269/// The `C` representation for items has a defined layout. With this layout,
270/// the size of items is also stable as long as all fields have a stable size.
271///
272/// ## Size of Structs
273///
274/// For `struct`s, the size is determined by the following algorithm.
275///
276/// For each field in the struct ordered by declaration order:
277///
278/// 1. Add the size of the field.
279/// 2. Round up the current size to the nearest multiple of the next field's [alignment].
280///
281/// Finally, round the size of the struct to the nearest multiple of its [alignment].
282/// The alignment of the struct is usually the largest alignment of all its
283/// fields; this can be changed with the use of `repr(align(N))`.
284///
285/// Unlike `C`, zero sized structs are not rounded up to one byte in size.
286///
287/// ## Size of Enums
288///
289/// Enums that carry no data other than the discriminant have the same size as C enums
290/// on the platform they are compiled for.
291///
292/// ## Size of Unions
293///
294/// The size of a union is the size of its largest field.
295///
296/// Unlike `C`, zero sized unions are not rounded up to one byte in size.
297///
298/// # Examples
299///
300/// ```
301/// // Some primitives
302/// assert_eq!(4, size_of::<i32>());
303/// assert_eq!(8, size_of::<f64>());
304/// assert_eq!(0, size_of::<()>());
305///
306/// // Some arrays
307/// assert_eq!(8, size_of::<[i32; 2]>());
308/// assert_eq!(12, size_of::<[i32; 3]>());
309/// assert_eq!(0, size_of::<[i32; 0]>());
310///
311///
312/// // Pointer size equality
313/// assert_eq!(size_of::<&i32>(), size_of::<*const i32>());
314/// assert_eq!(size_of::<&i32>(), size_of::<Box<i32>>());
315/// assert_eq!(size_of::<&i32>(), size_of::<Option<&i32>>());
316/// assert_eq!(size_of::<Box<i32>>(), size_of::<Option<Box<i32>>>());
317/// ```
318///
319/// Using `#[repr(C)]`.
320///
321/// ```
322/// #[repr(C)]
323/// struct FieldStruct {
324///     first: u8,
325///     second: u16,
326///     third: u8
327/// }
328///
329/// // The size of the first field is 1, so add 1 to the size. Size is 1.
330/// // The alignment of the second field is 2, so add 1 to the size for padding. Size is 2.
331/// // The size of the second field is 2, so add 2 to the size. Size is 4.
332/// // The alignment of the third field is 1, so add 0 to the size for padding. Size is 4.
333/// // The size of the third field is 1, so add 1 to the size. Size is 5.
334/// // Finally, the alignment of the struct is 2 (because the largest alignment amongst its
335/// // fields is 2), so add 1 to the size for padding. Size is 6.
336/// assert_eq!(6, size_of::<FieldStruct>());
337///
338/// #[repr(C)]
339/// struct TupleStruct(u8, u16, u8);
340///
341/// // Tuple structs follow the same rules.
342/// assert_eq!(6, size_of::<TupleStruct>());
343///
344/// // Note that reordering the fields can lower the size. We can remove both padding bytes
345/// // by putting `third` before `second`.
346/// #[repr(C)]
347/// struct FieldStructOptimized {
348///     first: u8,
349///     third: u8,
350///     second: u16
351/// }
352///
353/// assert_eq!(4, size_of::<FieldStructOptimized>());
354///
355/// // Union size is the size of the largest field.
356/// #[repr(C)]
357/// union ExampleUnion {
358///     smaller: u8,
359///     larger: u16
360/// }
361///
362/// assert_eq!(2, size_of::<ExampleUnion>());
363/// ```
364///
365/// [alignment]: align_of
366/// [`*const T`]: primitive@pointer
367/// [`Box<T>`]: ../../std/boxed/struct.Box.html
368/// [`Option<&T>`]: crate::option::Option
369///
370#[inline(always)]
371#[must_use]
372#[stable(feature = "rust1", since = "1.0.0")]
373#[rustc_promotable]
374#[rustc_const_stable(feature = "const_mem_size_of", since = "1.24.0")]
375#[rustc_diagnostic_item = "mem_size_of"]
376#[ferrocene::prevalidated]
377pub const fn size_of<T>() -> usize {
378    // By making this a constant, we also guarantee that the constant can be successfully evaluated
379    // in any program execution that actually executes `size_of`. Which is relevant because the
380    // constant can fail to evaluate if the type is too big. Someone might do something cursed where
381    // soundness relies on a certain type not being too big, and they check that by just invoking
382    // size_of on the type to ensure it exists, so if we fully DCE'd size_of calls that would be
383    // considered unsound... but by making this a constant, it participates in the usual "required
384    // consts" system, and we are safe.
385    <T as SizedTypeProperties>::SIZE
386}
387
388/// Returns the size of the pointed-to value in bytes.
389///
390/// This is usually the same as [`size_of::<T>()`]. However, when `T` *has* no
391/// statically-known size, e.g., a slice [`[T]`][slice] or a [trait object],
392/// then `size_of_val` can be used to get the dynamically-known size.
393///
394/// [trait object]: ../../book/ch17-02-trait-objects.html
395///
396/// # Examples
397///
398/// ```
399/// assert_eq!(4, size_of_val(&5i32));
400///
401/// let x: [u8; 13] = [0; 13];
402/// let y: &[u8] = &x;
403/// assert_eq!(13, size_of_val(y));
404/// ```
405///
406/// [`size_of::<T>()`]: size_of
407#[inline]
408#[must_use]
409#[stable(feature = "rust1", since = "1.0.0")]
410#[rustc_const_stable(feature = "const_size_of_val", since = "1.85.0")]
411#[rustc_diagnostic_item = "mem_size_of_val"]
412#[ferrocene::prevalidated]
413pub const fn size_of_val<T: ?Sized>(val: &T) -> usize {
414    // SAFETY: `val` is a reference, so it's a valid raw pointer
415    unsafe { intrinsics::size_of_val(val) }
416}
417
418/// Returns the size of the pointed-to value in bytes.
419///
420/// This is usually the same as [`size_of::<T>()`]. However, when `T` *has* no
421/// statically-known size, e.g., a slice [`[T]`][slice] or a [trait object],
422/// then `size_of_val_raw` can be used to get the dynamically-known size.
423///
424/// # Safety
425///
426/// This function is safe to call if the pointer is safe to reborrow as `&T`
427/// (in which case you could also call [`size_of_val`]).
428/// Otherwise, the following conditions must hold:
429///
430/// - If `T` is `Sized`, this function is always safe to call.
431/// - If the *unsized tail* of `T` is:
432///     - a [slice] `[U]`, `str`, or a [trait object] `dyn Trait`, then the size of the *entire value*
433///       (dynamic tail length + statically sized prefix) must fit in `isize`.
434///       For the special case where the dynamic tail length is 0, this function
435///       is safe to call.
436//        NOTE: the reason this is safe is that if an overflow were to occur already with size 0,
437//        then we would stop compilation as even the "statically known" part of the type would
438//        already be too big (or the call may be in dead code and optimized away, but then it
439//        doesn't matter).
440///     - No other kind of unsized tail currently exists that satisfies the trait bounds for this
441///       function. If more kinds of unsized tails get introduced in the future, the documentation
442///       of this function will have to be extended before it can be used for such types.
443///
444/// Here, *unsized tail* refers to the type obtained by recursively descending through the last
445/// field of a tuple or struct until we arrived at a built-in unsized type.
446///
447/// As a consequence of these rules, it is the case that whenever it is allowed to convert `val`
448/// into a shared reference, then it is also allowed to invoke this function.
449///
450/// [`size_of::<T>()`]: size_of
451/// [trait object]: ../../book/ch17-02-trait-objects.html
452/// [extern type]: ../../unstable-book/language-features/extern-types.html
453///
454/// # Examples
455///
456/// ```
457/// use std::mem;
458///
459/// assert_eq!(4, size_of_val(&5i32));
460///
461/// let x: [u8; 13] = [0; 13];
462/// let y: &[u8] = &x;
463/// assert_eq!(13, unsafe { mem::size_of_val_raw(y) });
464/// ```
465#[inline]
466#[must_use]
467#[stable(feature = "layout_for_ptr", since = "CURRENT_RUSTC_VERSION")]
468#[rustc_const_stable(feature = "layout_for_ptr", since = "CURRENT_RUSTC_VERSION")]
469#[ferrocene::prevalidated]
470pub const unsafe fn size_of_val_raw<T: ?Sized>(val: *const T) -> usize {
471    // SAFETY: the caller must provide a valid raw pointer
472    unsafe { intrinsics::size_of_val(val) }
473}
474
475/// Returns the [ABI]-required minimum alignment of a type in bytes.
476///
477/// Every reference to a value of the type `T` must be a multiple of this number.
478///
479/// This is the alignment used for struct fields. It may be smaller than the preferred alignment.
480///
481/// [ABI]: https://en.wikipedia.org/wiki/Application_binary_interface
482///
483/// # Examples
484///
485/// ```
486/// # #![allow(deprecated)]
487/// use std::mem;
488///
489/// assert_eq!(4, mem::min_align_of::<i32>());
490/// ```
491#[inline]
492#[must_use]
493#[stable(feature = "rust1", since = "1.0.0")]
494#[deprecated(note = "use `align_of` instead", since = "1.2.0", suggestion = "align_of")]
495pub fn min_align_of<T>() -> usize {
496    <T as SizedTypeProperties>::ALIGN
497}
498
499/// Returns the [ABI]-required minimum alignment of the type of the value that `val` points to in
500/// bytes.
501///
502/// Every reference to a value of the type `T` must be a multiple of this number.
503///
504/// [ABI]: https://en.wikipedia.org/wiki/Application_binary_interface
505///
506/// # Examples
507///
508/// ```
509/// # #![allow(deprecated)]
510/// use std::mem;
511///
512/// assert_eq!(4, mem::min_align_of_val(&5i32));
513/// ```
514#[inline]
515#[must_use]
516#[stable(feature = "rust1", since = "1.0.0")]
517#[deprecated(note = "use `align_of_val` instead", since = "1.2.0", suggestion = "align_of_val")]
518pub fn min_align_of_val<T: ?Sized>(val: &T) -> usize {
519    // SAFETY: val is a reference, so it's a valid raw pointer
520    unsafe { intrinsics::align_of_val(val) }
521}
522
523/// Returns the [ABI]-required minimum alignment of a type, in bytes.
524///
525/// Every reference to a value of the type `T` must be a multiple of this number.
526///
527/// This is the alignment used for struct fields. It may be smaller than the preferred alignment.
528///
529/// [ABI]: https://en.wikipedia.org/wiki/Application_binary_interface
530///
531/// # Examples
532///
533/// ```
534/// assert_eq!(4, align_of::<i32>());
535/// ```
536///
537/// (Caution: [it is not guaranteed][type-layout] that the alignment of `i32` is `4`;
538/// that is, the above assertion does not pass on all platforms.)
539///
540/// [type-layout]: ../../reference/type-layout.html#r-layout.primitive
541#[inline(always)]
542#[must_use]
543#[stable(feature = "rust1", since = "1.0.0")]
544#[rustc_promotable]
545#[rustc_const_stable(feature = "const_align_of", since = "1.24.0")]
546#[rustc_diagnostic_item = "mem_align_of"]
547#[ferrocene::prevalidated]
548pub const fn align_of<T>() -> usize {
549    <T as SizedTypeProperties>::ALIGN
550}
551
552/// Returns the [ABI]-required minimum alignment of the type of the value that `val` points to, in
553/// bytes.
554///
555/// This function is identical to [`align_of::<T>()`][align_of] whenever <code>T: [Sized]</code>,
556/// but also supports determining the alignment required by a `dyn Trait` value, which is the
557/// alignment of the underlying concrete type.
558///
559/// [ABI]: https://en.wikipedia.org/wiki/Application_binary_interface
560///
561/// # Examples
562///
563/// ```
564/// assert_eq!(4, align_of_val(&5i32));
565/// ```
566///
567/// (Caution: [it is not guaranteed][type-layout] that the alignment of `i32` is `4`;
568/// that is, this example assertion does not pass on all platforms.)
569///
570/// `dyn` types may have different alignments for different values;
571/// `align_of_val` can be used to learn those alignments:
572///
573/// ```
574/// let a: &dyn ToString = &1234u16;
575/// let b: &dyn ToString = &String::from("abcd");
576///
577/// assert_eq!(align_of_val(a), align_of::<u16>());
578/// assert_eq!(align_of_val(b), align_of::<String>());
579/// ```
580///
581/// [type-layout]: ../../reference/type-layout.html#r-layout.primitive
582#[inline]
583#[must_use]
584#[stable(feature = "rust1", since = "1.0.0")]
585#[rustc_const_stable(feature = "const_align_of_val", since = "1.85.0")]
586#[ferrocene::prevalidated]
587pub const fn align_of_val<T: ?Sized>(val: &T) -> usize {
588    // SAFETY: val is a reference, so it's a valid raw pointer
589    unsafe { intrinsics::align_of_val(val) }
590}
591
592/// Returns the [ABI]-required minimum alignment of the type of the value that `val` points to, in
593/// bytes.
594///
595/// This function is identical to [`align_of_val()`], except that it can be used with raw pointers
596/// in situations where it would be unsound or undesirable to convert them to
597/// [`&` references][primitive@reference] and impose the aliasing rules that come with that.
598///
599/// [ABI]: https://en.wikipedia.org/wiki/Application_binary_interface
600///
601/// # Safety
602///
603/// This function is safe to call if the pointer is safe to reborrow as `&T`
604/// (in which case you could also call [`align_of_val`]).
605/// Otherwise, the following conditions must hold:
606///
607/// - If `T` is `Sized`, this function is always safe to call.
608/// - If the unsized tail of `T` is:
609///     - a [slice] `[U]`, `str`, or a [trait object] `dyn Trait`, then the size of the *entire value*
610///       (dynamic tail length + statically sized prefix) must fit in `isize`.
611///       For the special case where the dynamic tail length is 0, this function
612///       is safe to call.
613//        NOTE: the reason this is safe is that if an overflow were to occur already with size 0,
614//        then we would stop compilation as even the "statically known" part of the type would
615//        already be too big (or the call may be in dead code and optimized away, but then it
616//        doesn't matter).
617///     - No other kind of unsized tail currently exists that satisfies the trait bounds for this
618///       function. If more kinds of unsized tails get introduced in the future, the documentation
619///       of this function will have to be extended before it can be used for such types.
620///
621/// Here, *unsized tail* refers to the type obtained by recursively descending through the last
622/// field of a tuple or struct until we arrived at a built-in unsized type.
623///
624/// As a consequence of these rules, it is the case that whenever it is allowed to convert `val`
625/// into a shared reference, then it is also allowed to invoke this function.
626///
627/// [trait object]: ../../book/ch17-02-trait-objects.html
628/// [extern type]: ../../unstable-book/language-features/extern-types.html
629///
630/// # Examples
631///
632/// ```
633/// use std::mem;
634///
635/// assert_eq!(4, unsafe { mem::align_of_val_raw(&5i32) });
636/// ```
637///
638/// (Caution: [it is not guaranteed][type-layout] that the alignment of `i32` is `4`;
639/// that is, the above assertion does not pass on all platforms.)
640///
641/// [type-layout]: ../../reference/type-layout.html#r-layout.primitive
642#[inline]
643#[must_use]
644#[stable(feature = "layout_for_ptr", since = "CURRENT_RUSTC_VERSION")]
645#[rustc_const_stable(feature = "layout_for_ptr", since = "CURRENT_RUSTC_VERSION")]
646pub const unsafe fn align_of_val_raw<T: ?Sized>(val: *const T) -> usize {
647    // SAFETY: the caller must provide a valid raw pointer
648    unsafe { intrinsics::align_of_val(val) }
649}
650
651/// Returns `true` if dropping values of type `T` matters.
652///
653/// This is purely an optimization hint, and may be implemented conservatively:
654/// it may return `true` for types that don't actually need to be dropped.
655/// As such always returning `true` would be a valid implementation of
656/// this function. However if this function actually returns `false`, then you
657/// can be certain dropping `T` has no side effect.
658///
659/// Low level implementations of things like collections, which need to manually
660/// drop their data, should use this function to avoid unnecessarily
661/// trying to drop all their contents when they are destroyed. This might not
662/// make a difference in release builds (where a loop that has no side-effects
663/// is easily detected and eliminated), but is often a big win for debug builds.
664///
665/// Note that [`drop_in_place`] already performs this check, so if your workload
666/// can be reduced to some small number of [`drop_in_place`] calls, using this is
667/// unnecessary. In particular note that you can [`drop_in_place`] a slice, and that
668/// will do a single needs_drop check for all the values.
669///
670/// Types like Vec therefore just `drop_in_place(&mut self[..])` without using
671/// `needs_drop` explicitly. Types like [`HashMap`], on the other hand, have to drop
672/// values one at a time and should use this API.
673///
674/// [`drop_in_place`]: crate::ptr::drop_in_place
675/// [`HashMap`]: ../../std/collections/struct.HashMap.html
676///
677/// # Examples
678///
679/// Here's an example of how a collection might make use of `needs_drop`:
680///
681/// ```
682/// use std::{mem, ptr};
683///
684/// pub struct MyCollection<T> {
685/// #   data: [T; 1],
686///     /* ... */
687/// }
688/// # impl<T> MyCollection<T> {
689/// #   fn iter_mut(&mut self) -> &mut [T] { &mut self.data }
690/// #   fn free_buffer(&mut self) {}
691/// # }
692///
693/// impl<T> Drop for MyCollection<T> {
694///     fn drop(&mut self) {
695///         unsafe {
696///             // drop the data
697///             if mem::needs_drop::<T>() {
698///                 for x in self.iter_mut() {
699///                     ptr::drop_in_place(x);
700///                 }
701///             }
702///             self.free_buffer();
703///         }
704///     }
705/// }
706/// ```
707#[inline]
708#[must_use]
709#[stable(feature = "needs_drop", since = "1.21.0")]
710#[rustc_const_stable(feature = "const_mem_needs_drop", since = "1.36.0")]
711#[rustc_diagnostic_item = "needs_drop"]
712#[ferrocene::prevalidated]
713pub const fn needs_drop<T: ?Sized>() -> bool {
714    const { intrinsics::needs_drop::<T>() }
715}
716
717/// Returns the value of type `T` represented by the all-zero byte-pattern.
718///
719/// This means that, for example, the padding byte in `(u8, u16)` is not
720/// necessarily zeroed.
721///
722/// There is no guarantee that an all-zero byte-pattern represents a valid value
723/// of some type `T`. For example, the all-zero byte-pattern is not a valid value
724/// for reference types (`&T`, `&mut T`) and function pointers. Using `zeroed`
725/// on such types causes immediate [undefined behavior][ub] because [the Rust
726/// compiler assumes][inv] that there always is a valid value in a variable it
727/// considers initialized.
728///
729/// This has the same effect as [`MaybeUninit::zeroed().assume_init()`][zeroed].
730/// It is useful for FFI sometimes, but should generally be avoided.
731///
732/// [zeroed]: MaybeUninit::zeroed
733/// [ub]: ../../reference/behavior-considered-undefined.html
734/// [inv]: MaybeUninit#initialization-invariant
735///
736/// # Examples
737///
738/// Correct usage of this function: initializing an integer with zero.
739///
740/// ```
741/// use std::mem;
742///
743/// let x: i32 = unsafe { mem::zeroed() };
744/// assert_eq!(0, x);
745/// ```
746///
747/// *Incorrect* usage of this function: initializing a reference with zero.
748///
749/// ```rust,no_run
750/// # #![allow(invalid_value)]
751/// use std::mem;
752///
753/// let _x: &i32 = unsafe { mem::zeroed() }; // Undefined behavior!
754/// let _y: fn() = unsafe { mem::zeroed() }; // And again!
755/// ```
756#[inline(always)]
757#[must_use]
758#[stable(feature = "rust1", since = "1.0.0")]
759#[rustc_diagnostic_item = "mem_zeroed"]
760#[track_caller]
761#[rustc_const_stable(feature = "const_mem_zeroed", since = "1.75.0")]
762#[ferrocene::prevalidated]
763pub const unsafe fn zeroed<T>() -> T {
764    // SAFETY: the caller must guarantee that an all-zero value is valid for `T`.
765    unsafe {
766        intrinsics::assert_zero_valid::<T>();
767        MaybeUninit::zeroed().assume_init()
768    }
769}
770
771/// Bypasses Rust's normal memory-initialization checks by pretending to
772/// produce a value of type `T`, while doing nothing at all.
773///
774/// **This function is deprecated.** Use [`MaybeUninit<T>`] instead.
775/// It also might be slower than using `MaybeUninit<T>` due to mitigations that were put in place to
776/// limit the potential harm caused by incorrect use of this function in legacy code.
777///
778/// The reason for deprecation is that the function basically cannot be used
779/// correctly: it has the same effect as [`MaybeUninit::uninit().assume_init()`][uninit].
780/// As the [`assume_init` documentation][assume_init] explains,
781/// [the Rust compiler assumes][inv] that values are properly initialized.
782///
783/// Truly uninitialized memory like what gets returned here
784/// is special in that the compiler knows that it does not have a fixed value.
785/// This makes it undefined behavior to have uninitialized data in a variable even
786/// if that variable has an integer type.
787///
788/// Therefore, it is immediate undefined behavior to call this function on nearly all types,
789/// including integer types and arrays of integer types, and even if the result is unused.
790///
791/// [uninit]: MaybeUninit::uninit
792/// [assume_init]: MaybeUninit::assume_init
793/// [inv]: MaybeUninit#initialization-invariant
794#[inline(always)]
795#[must_use]
796#[deprecated(since = "1.39.0", note = "use `mem::MaybeUninit` instead")]
797#[stable(feature = "rust1", since = "1.0.0")]
798#[rustc_diagnostic_item = "mem_uninitialized"]
799#[track_caller]
800pub unsafe fn uninitialized<T>() -> T {
801    // SAFETY: the caller must guarantee that an uninitialized value is valid for `T`.
802    unsafe {
803        intrinsics::assert_mem_uninitialized_valid::<T>();
804        let mut val = MaybeUninit::<T>::uninit();
805
806        // Fill memory with 0x01, as an imperfect mitigation for old code that uses this function on
807        // bool, nonnull, and noundef types. But don't do this if we actively want to detect UB.
808        if !cfg!(any(miri, sanitize = "memory")) {
809            val.as_mut_ptr().write_bytes(0x01, 1);
810        }
811
812        val.assume_init()
813    }
814}
815
816/// Swaps the values at two mutable locations, without deinitializing either one.
817///
818/// * If you want to swap with a default or dummy value, see [`take`].
819/// * If you want to swap with a passed value, returning the old value, see [`replace`].
820///
821/// # Examples
822///
823/// ```
824/// use std::mem;
825///
826/// let mut x = 5;
827/// let mut y = 42;
828///
829/// mem::swap(&mut x, &mut y);
830///
831/// assert_eq!(42, x);
832/// assert_eq!(5, y);
833/// ```
834#[inline]
835#[stable(feature = "rust1", since = "1.0.0")]
836#[rustc_const_stable(feature = "const_swap", since = "1.85.0")]
837#[rustc_diagnostic_item = "mem_swap"]
838#[ferrocene::prevalidated]
839pub const fn swap<T>(x: &mut T, y: &mut T) {
840    // SAFETY: `&mut` guarantees these are typed readable and writable
841    // as well as non-overlapping.
842    unsafe { intrinsics::typed_swap_nonoverlapping(x, y) }
843}
844
845/// Replaces `dest` with the default value of `T`, returning the previous `dest` value.
846///
847/// * If you want to replace the values of two variables, see [`swap`].
848/// * If you want to replace with a passed value instead of the default value, see [`replace`].
849///
850/// # Examples
851///
852/// A simple example:
853///
854/// ```
855/// use std::mem;
856///
857/// let mut v: Vec<i32> = vec![1, 2];
858///
859/// let old_v = mem::take(&mut v);
860/// assert_eq!(vec![1, 2], old_v);
861/// assert!(v.is_empty());
862/// ```
863///
864/// `take` allows taking ownership of a struct field by replacing it with an "empty" value.
865/// Without `take` you can run into issues like these:
866///
867/// ```compile_fail,E0507
868/// struct Buffer<T> { buf: Vec<T> }
869///
870/// impl<T> Buffer<T> {
871///     fn get_and_reset(&mut self) -> Vec<T> {
872///         // error: cannot move out of dereference of `&mut`-pointer
873///         let buf = self.buf;
874///         self.buf = Vec::new();
875///         buf
876///     }
877/// }
878/// ```
879///
880/// Note that `T` does not necessarily implement [`Clone`], so it can't even clone and reset
881/// `self.buf`. But `take` can be used to disassociate the original value of `self.buf` from
882/// `self`, allowing it to be returned:
883///
884/// ```
885/// use std::mem;
886///
887/// # struct Buffer<T> { buf: Vec<T> }
888/// impl<T> Buffer<T> {
889///     fn get_and_reset(&mut self) -> Vec<T> {
890///         mem::take(&mut self.buf)
891///     }
892/// }
893///
894/// let mut buffer = Buffer { buf: vec![0, 1] };
895/// assert_eq!(buffer.buf.len(), 2);
896///
897/// assert_eq!(buffer.get_and_reset(), vec![0, 1]);
898/// assert_eq!(buffer.buf.len(), 0);
899/// ```
900#[inline]
901#[stable(feature = "mem_take", since = "1.40.0")]
902#[rustc_const_unstable(feature = "const_default", issue = "143894")]
903#[ferrocene::prevalidated]
904pub const fn take<T: [const] Default>(dest: &mut T) -> T {
905    replace(dest, T::default())
906}
907
908/// Moves `src` into the referenced `dest`, returning the previous `dest` value.
909///
910/// Neither value is dropped.
911///
912/// * If you want to replace the values of two variables, see [`swap`].
913/// * If you want to replace with a default value, see [`take`].
914///
915/// # Examples
916///
917/// A simple example:
918///
919/// ```
920/// use std::mem;
921///
922/// let mut v: Vec<i32> = vec![1, 2];
923///
924/// let old_v = mem::replace(&mut v, vec![3, 4, 5]);
925/// assert_eq!(vec![1, 2], old_v);
926/// assert_eq!(vec![3, 4, 5], v);
927/// ```
928///
929/// `replace` allows consumption of a struct field by replacing it with another value.
930/// Without `replace` you can run into issues like these:
931///
932/// ```compile_fail,E0507
933/// struct Buffer<T> { buf: Vec<T> }
934///
935/// impl<T> Buffer<T> {
936///     fn replace_index(&mut self, i: usize, v: T) -> T {
937///         // error: cannot move out of dereference of `&mut`-pointer
938///         let t = self.buf[i];
939///         self.buf[i] = v;
940///         t
941///     }
942/// }
943/// ```
944///
945/// Note that `T` does not necessarily implement [`Clone`], so we can't even clone `self.buf[i]` to
946/// avoid the move. But `replace` can be used to disassociate the original value at that index from
947/// `self`, allowing it to be returned:
948///
949/// ```
950/// # #![allow(dead_code)]
951/// use std::mem;
952///
953/// # struct Buffer<T> { buf: Vec<T> }
954/// impl<T> Buffer<T> {
955///     fn replace_index(&mut self, i: usize, v: T) -> T {
956///         mem::replace(&mut self.buf[i], v)
957///     }
958/// }
959///
960/// let mut buffer = Buffer { buf: vec![0, 1] };
961/// assert_eq!(buffer.buf[0], 0);
962///
963/// assert_eq!(buffer.replace_index(0, 2), 0);
964/// assert_eq!(buffer.buf[0], 2);
965/// ```
966#[inline]
967#[stable(feature = "rust1", since = "1.0.0")]
968#[must_use = "if you don't need the old value, you can just assign the new value directly"]
969#[rustc_const_stable(feature = "const_replace", since = "1.83.0")]
970#[rustc_diagnostic_item = "mem_replace"]
971#[ferrocene::prevalidated]
972pub const fn replace<T>(dest: &mut T, src: T) -> T {
973    // It may be tempting to use `swap` to avoid `unsafe` here. Don't!
974    // The compiler optimizes the implementation below to two `memcpy`s
975    // while `swap` would require at least three. See PR#83022 for details.
976
977    // SAFETY: We read from `dest` but directly write `src` into it afterwards,
978    // such that the old value is not duplicated. Nothing is dropped and
979    // nothing here can panic.
980    unsafe {
981        // Ideally we wouldn't use the intrinsics here, but going through the
982        // `ptr` methods introduces two unnecessary UbChecks, so until we can
983        // remove those for pointers that come from references, this uses the
984        // intrinsics instead so this stays very cheap in MIR (and debug).
985
986        let result = crate::intrinsics::read_via_copy(dest);
987        crate::intrinsics::write_via_move(dest, src);
988        result
989    }
990}
991
992/// Disposes of a value.
993///
994/// This effectively does nothing for types which implement `Copy`, e.g.
995/// integers. Such values are copied and _then_ moved into the function, so the
996/// value persists after this function call.
997///
998/// This function is not magic; it is literally defined as
999///
1000/// ```
1001/// pub fn drop<T>(_x: T) {}
1002/// ```
1003///
1004/// Because `_x` is moved into the function, it is automatically [dropped][drop] before
1005/// the function returns.
1006///
1007/// [drop]: Drop
1008///
1009/// # Examples
1010///
1011/// Basic usage:
1012///
1013/// ```
1014/// let v = vec![1, 2, 3];
1015///
1016/// drop(v); // explicitly drop the vector
1017/// ```
1018///
1019/// Since [`RefCell`] enforces the borrow rules at runtime, `drop` can
1020/// release a [`RefCell`] borrow:
1021///
1022/// ```
1023/// use std::cell::RefCell;
1024///
1025/// let x = RefCell::new(1);
1026///
1027/// let mut mutable_borrow = x.borrow_mut();
1028/// *mutable_borrow = 1;
1029///
1030/// drop(mutable_borrow); // relinquish the mutable borrow on this slot
1031///
1032/// let borrow = x.borrow();
1033/// println!("{}", *borrow);
1034/// ```
1035///
1036/// Integers and other types implementing [`Copy`] are unaffected by `drop`.
1037///
1038/// ```
1039/// # #![allow(dropping_copy_types)]
1040/// #[derive(Copy, Clone)]
1041/// struct Foo(u8);
1042///
1043/// let x = 1;
1044/// let y = Foo(2);
1045/// drop(x); // a copy of `x` is moved and dropped
1046/// drop(y); // a copy of `y` is moved and dropped
1047///
1048/// println!("x: {}, y: {}", x, y.0); // still available
1049/// ```
1050///
1051/// [`RefCell`]: crate::cell::RefCell
1052#[inline]
1053#[stable(feature = "rust1", since = "1.0.0")]
1054#[rustc_const_unstable(feature = "const_destruct", issue = "133214")]
1055#[rustc_diagnostic_item = "mem_drop"]
1056#[ferrocene::prevalidated]
1057pub const fn drop<T>(_x: T)
1058where
1059    T: [const] Destruct,
1060{
1061}
1062
1063/// Bitwise-copies a value.
1064///
1065/// This function is not magic; it is literally defined as
1066/// ```
1067/// pub const fn copy<T: Copy>(x: &T) -> T { *x }
1068/// ```
1069///
1070/// It is useful when you want to pass a function pointer to a combinator, rather than defining a new closure.
1071///
1072/// Example:
1073/// ```
1074/// #![feature(mem_copy_fn)]
1075/// use core::mem::copy;
1076/// let result_from_ffi_function: Result<(), &i32> = Err(&1);
1077/// let result_copied: Result<(), i32> = result_from_ffi_function.map_err(copy);
1078/// ```
1079#[inline]
1080#[unstable(feature = "mem_copy_fn", issue = "98262")]
1081pub const fn copy<T: Copy>(x: &T) -> T {
1082    *x
1083}
1084
1085/// Interprets `src` as having type `&Dst`, and then reads `src` without moving
1086/// the contained value.
1087///
1088/// This function will unsafely assume the pointer `src` is valid for [`size_of::<Dst>`][size_of]
1089/// bytes by transmuting `&Src` to `&Dst` and then reading the `&Dst` (except that this is done
1090/// in a way that is correct even when `&Dst` has stricter alignment requirements than `&Src`).
1091/// It will also unsafely create a copy of the contained value instead of moving out of `src`.
1092///
1093/// It is not a compile-time error if `Src` and `Dst` have different sizes, but it
1094/// is highly encouraged to only invoke this function where `Src` and `Dst` have the
1095/// same size. This function triggers [undefined behavior][ub] if `Dst` is larger than
1096/// `Src`.
1097///
1098/// [ub]: ../../reference/behavior-considered-undefined.html
1099///
1100/// If you have a raw pointer instead of a reference, you might be looking for
1101/// `src.cast::<Dst>().`[`read_unaligned()`](pointer#method.read_unaligned) instead.
1102///
1103/// # Safety
1104///
1105/// - Requires `size_of_val::<Src>(src) >= size_of::<Dst>()`
1106/// - The first `size_of::<Dst>()` bytes behind `src` must be *readable*
1107/// - The first `size_of::<Dst>()` bytes behind `src` must be *[valid]*
1108///   when interpreted as a `Dst`.
1109///
1110/// On top of that, remember that most types have additional invariants beyond merely
1111/// being considered initialized at the type level. For example, a `1`-initialized [`Vec<T>`]
1112/// is considered initialized (under the current implementation; this does not constitute
1113/// a stable guarantee) because the only requirement the compiler knows about it
1114/// is that the data pointer must be non-null. Creating such a `Vec<T>` does not cause
1115/// *immediate* undefined behavior, but will cause undefined behavior with most
1116/// safe operations (including dropping it).
1117///
1118/// [valid]: ../../reference/behavior-considered-undefined.html#r-undefined.validity
1119/// [`Vec<T>`]: ../../std/vec/struct.Vec.html
1120///
1121/// # Examples
1122///
1123/// ```
1124/// use std::mem;
1125///
1126/// #[repr(packed)]
1127/// struct Foo {
1128///     bar: u8,
1129/// }
1130///
1131/// let foo_array = [10u8];
1132///
1133/// unsafe {
1134///     // Copy the data from 'foo_array' and treat it as a 'Foo'
1135///     let mut foo_struct: Foo = mem::transmute_copy(&foo_array);
1136///     assert_eq!(foo_struct.bar, 10);
1137///
1138///     // Modify the copied data
1139///     foo_struct.bar = 20;
1140///     assert_eq!(foo_struct.bar, 20);
1141/// }
1142///
1143/// // The contents of 'foo_array' should not have changed
1144/// assert_eq!(foo_array, [10]);
1145///
1146/// let bytes: &[u8] = &[1, 2, 3, 4, 5, 6, 7];
1147/// assert_eq!(
1148///     unsafe { mem::transmute_copy::<[u8], u32>(bytes) },
1149///     u32::from_ne_bytes(*bytes.first_chunk().unwrap()),
1150/// );
1151/// ```
1152#[ferrocene::prevalidated]
1153#[inline]
1154#[must_use]
1155#[track_caller]
1156#[stable(feature = "rust1", since = "1.0.0")]
1157#[rustc_const_stable(feature = "const_transmute_copy", since = "1.74.0")]
1158pub const unsafe fn transmute_copy<Src: ?Sized, Dst>(src: &Src) -> Dst {
1159    // library UB because it's possible for the `Src` to be only a subset of the allocation
1160    // and thus for a failure to not be immediate language UB
1161    assert_unsafe_precondition!(
1162        check_library_ub,
1163        "cannot transmute_copy if Dst is larger than Src",
1164        (
1165            src_size: usize = size_of_val::<Src>(src),
1166            dst_size: usize = Dst::SIZE,
1167        ) => src_size >= dst_size
1168    );
1169
1170    // If Dst has a higher alignment requirement, src might not be suitably aligned.
1171    if align_of::<Dst>() > align_of_val::<Src>(src) {
1172        // SAFETY: `src` is a reference which is guaranteed to be valid for reads.
1173        // The caller must guarantee that the actual transmutation is safe.
1174        unsafe { ptr::read_unaligned(src as *const Src as *const Dst) }
1175    } else {
1176        // SAFETY: `src` is a reference which is guaranteed to be valid for reads.
1177        // We just checked that `src as *const Dst` was properly aligned.
1178        // The caller must guarantee that the actual transmutation is safe.
1179        unsafe { ptr::read(src as *const Src as *const Dst) }
1180    }
1181}
1182
1183/// Like [`transmute`], but only initializes the "common prefix" of the first
1184/// `min(size_of::<Src>(), size_of::<Dst>())` bytes of the destination from the
1185/// corresponding bytes of the source.
1186///
1187/// This is equivalent to a "union cast" through a `union` with `#[repr(C)]`.
1188///
1189/// That means some size mismatches are not UB, like `[T; 2]` to `[T; 1]`.
1190/// Increasing size is usually UB from being insufficiently initialized -- like
1191/// `u8` to `u32` -- but isn't always.  For example, going from `u8` to
1192/// `#[repr(C, align(4))] AlignedU8(u8);` is sound.
1193///
1194/// Prefer normal `transmute` where possible, for the extra checking, since
1195/// both do exactly the same thing at runtime, if they both compile.
1196///
1197/// # Safety
1198///
1199/// If `size_of::<Src>() >= size_of::<Dst>()`, the first `size_of::<Dst>()` bytes
1200/// of `src` must be be *valid* when interpreted as a `Dst`.  (In this case, the
1201/// preconditions are the same as for `transmute_copy(&ManuallyDrop::new(src))`.)
1202///
1203/// If `size_of::<Src>() <= size_of::<Dst>()`, the bytes of `src` padded with
1204/// uninitialized bytes afterwards up to a total size of `size_of::<Dst>()`
1205/// must be *valid* when interpreted as a `Dst`.
1206///
1207/// In both cases, any safety preconditions of the `Dst` type must also be upheld.
1208///
1209/// # Examples
1210///
1211/// ```
1212/// #![feature(transmute_prefix)]
1213/// use std::mem::transmute_prefix;
1214///
1215/// assert_eq!(unsafe { transmute_prefix::<[i32; 4], [i32; 2]>([1, 2, 3, 4]) }, [1, 2]);
1216///
1217/// let expected = if cfg!(target_endian = "little") { 0x34 } else { 0x12 };
1218/// assert_eq!(unsafe { transmute_prefix::<u16, u8>(0x1234) }, expected);
1219///
1220/// // Would be UB because the destination is incompletely initialized.
1221/// // transmute_prefix::<u8, u16>(123)
1222///
1223/// // OK because the destination is allowed to be partially initialized.
1224/// let _: std::mem::MaybeUninit<u16> = unsafe { transmute_prefix(123_u8) };
1225/// ```
1226#[unstable(feature = "transmute_prefix", issue = "155079")]
1227#[rustc_no_writable]
1228pub const unsafe fn transmute_prefix<Src, Dst>(src: Src) -> Dst {
1229    #[repr(C)]
1230    union Transmute<A, B> {
1231        a: ManuallyDrop<A>,
1232        b: ManuallyDrop<B>,
1233    }
1234
1235    match const { Ord::cmp(&Src::SIZE, &Dst::SIZE) } {
1236        // SAFETY: When Dst is bigger, the union is the size of Dst
1237        Ordering::Less => unsafe {
1238            let a = transmute_neo(src);
1239            intrinsics::transmute_unchecked(Transmute::<Src, Dst> { a })
1240        },
1241        // SAFETY: When they're the same size, we can use the MIR primitive
1242        Ordering::Equal => unsafe { intrinsics::transmute_unchecked::<Src, Dst>(src) },
1243        // SAFETY: When Src is bigger, the union is the size of Src
1244        Ordering::Greater => unsafe {
1245            let u: Transmute<Src, Dst> = intrinsics::transmute_unchecked(src);
1246            transmute_neo(u.b)
1247        },
1248    }
1249}
1250
1251/// New version of `transmute`, exposed under this name so it can be iterated upon
1252/// without risking breakage to uses of "real" transmute.
1253///
1254/// Uses a `const`-`assert` to check the sizes instead of typeck hacks,
1255/// but is semantially identical to `transmute` otherwise.
1256///
1257/// It will not be stabilized under this name.
1258///
1259/// # Examples
1260///
1261/// ```
1262/// #![feature(transmute_neo)]
1263/// use std::mem::transmute_neo;
1264///
1265/// assert_eq!(unsafe { transmute_neo::<f32, u32>(0.0) }, 0);
1266/// ```
1267///
1268/// ```compile_fail,E0080
1269/// #![feature(transmute_neo)]
1270/// use std::mem::transmute_neo;
1271///
1272/// unsafe { transmute_neo::<u32, u16>(123) };
1273/// ```
1274#[ferrocene::prevalidated]
1275#[unstable(feature = "transmute_neo", issue = "155079")]
1276#[cfg_attr(miri, track_caller)] // even without panics, this helps for Miri backtraces
1277#[inline]
1278#[rustc_no_writable]
1279pub const unsafe fn transmute_neo<Src, Dst>(src: Src) -> Dst {
1280    const { assert!(Src::SIZE == Dst::SIZE) };
1281
1282    // SAFETY: the const-assert just checked that they're the same size,
1283    // and any other safety invariants need to be upheld by the caller.
1284    unsafe { intrinsics::transmute_unchecked(src) }
1285}
1286
1287/// Opaque type representing the discriminant of an enum.
1288///
1289/// See the [`discriminant`] function in this module for more information.
1290#[stable(feature = "discriminant_value", since = "1.21.0")]
1291#[ferrocene::prevalidated]
1292pub struct Discriminant<T>(<T as DiscriminantKind>::Discriminant);
1293
1294// N.B. These trait implementations cannot be derived because we don't want any bounds on T.
1295
1296#[stable(feature = "discriminant_value", since = "1.21.0")]
1297impl<T> Copy for Discriminant<T> {}
1298
1299#[stable(feature = "discriminant_value", since = "1.21.0")]
1300impl<T> clone::Clone for Discriminant<T> {
1301    fn clone(&self) -> Self {
1302        *self
1303    }
1304}
1305
1306#[doc(hidden)]
1307#[unstable(feature = "trivial_clone", issue = "none")]
1308unsafe impl<T> TrivialClone for Discriminant<T> {}
1309
1310#[stable(feature = "discriminant_value", since = "1.21.0")]
1311impl<T> cmp::PartialEq for Discriminant<T> {
1312    #[ferrocene::prevalidated]
1313    fn eq(&self, rhs: &Self) -> bool {
1314        self.0 == rhs.0
1315    }
1316}
1317
1318#[stable(feature = "discriminant_value", since = "1.21.0")]
1319impl<T> cmp::Eq for Discriminant<T> {}
1320
1321#[stable(feature = "discriminant_value", since = "1.21.0")]
1322impl<T> hash::Hash for Discriminant<T> {
1323    fn hash<H: hash::Hasher>(&self, state: &mut H) {
1324        self.0.hash(state);
1325    }
1326}
1327
1328#[stable(feature = "discriminant_value", since = "1.21.0")]
1329impl<T> fmt::Debug for Discriminant<T> {
1330    #[ferrocene::prevalidated]
1331    fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
1332        fmt.debug_tuple("Discriminant").field(&self.0).finish()
1333    }
1334}
1335
1336/// Returns a value uniquely identifying the enum variant in `v`.
1337///
1338/// If `T` is not an enum, calling this function will not result in undefined behavior, but the
1339/// return value is unspecified.
1340///
1341/// # Stability
1342///
1343/// The discriminant of an enum variant may change if the enum definition changes. A discriminant
1344/// of some variant will not change between compilations with the same compiler. See the [Reference]
1345/// for more information.
1346///
1347/// [Reference]: ../../reference/items/enumerations.html#custom-discriminant-values-for-fieldless-enumerations
1348///
1349/// The value of a [`Discriminant<T>`] is independent of any *free lifetimes* in `T`. As such,
1350/// reading or writing a `Discriminant<Foo<'a>>` as a `Discriminant<Foo<'b>>` (whether via
1351/// [`transmute`] or otherwise) is always sound. Note that this is **not** true for other kinds
1352/// of generic parameters and for higher-ranked lifetimes; `Discriminant<Foo<A>>` and
1353/// `Discriminant<Foo<B>>` as well as `Discriminant<Bar<dyn for<'a> Trait<'a>>>` and
1354/// `Discriminant<Bar<dyn Trait<'static>>>` may be incompatible.
1355///
1356/// # Examples
1357///
1358/// This can be used to compare enums that carry data, while disregarding
1359/// the actual data:
1360///
1361/// ```
1362/// use std::mem;
1363///
1364/// enum Foo { A(&'static str), B(i32), C(i32) }
1365///
1366/// assert_eq!(mem::discriminant(&Foo::A("bar")), mem::discriminant(&Foo::A("baz")));
1367/// assert_eq!(mem::discriminant(&Foo::B(1)), mem::discriminant(&Foo::B(2)));
1368/// assert_ne!(mem::discriminant(&Foo::B(3)), mem::discriminant(&Foo::C(3)));
1369/// ```
1370///
1371/// ## Accessing the numeric value of the discriminant
1372///
1373/// Note that it is *undefined behavior* to [`transmute`] from [`Discriminant`] to a primitive!
1374///
1375/// If an enum has only unit variants, then the numeric value of the discriminant can be accessed
1376/// with an [`as`] cast:
1377///
1378/// ```
1379/// enum Enum {
1380///     Foo,
1381///     Bar,
1382///     Baz,
1383/// }
1384///
1385/// assert_eq!(0, Enum::Foo as isize);
1386/// assert_eq!(1, Enum::Bar as isize);
1387/// assert_eq!(2, Enum::Baz as isize);
1388/// ```
1389///
1390/// If an enum has opted-in to having a [primitive representation] for its discriminant,
1391/// then it's possible to use pointers to read the memory location storing the discriminant.
1392/// That **cannot** be done for enums using the [default representation], however, as it's
1393/// undefined what layout the discriminant has and where it's stored — it might not even be
1394/// stored at all!
1395///
1396/// [`as`]: ../../std/keyword.as.html
1397/// [primitive representation]: ../../reference/type-layout.html#primitive-representations
1398/// [default representation]: ../../reference/type-layout.html#the-default-representation
1399/// ```
1400/// #[repr(u8)]
1401/// enum Enum {
1402///     Unit,
1403///     Tuple(bool),
1404///     Struct { a: bool },
1405/// }
1406///
1407/// impl Enum {
1408///     fn discriminant(&self) -> u8 {
1409///         // SAFETY: Because `Self` is marked `repr(u8)`, its layout is a `repr(C)` `union`
1410///         // between `repr(C)` structs, each of which has the `u8` discriminant as its first
1411///         // field, so we can read the discriminant without offsetting the pointer.
1412///         unsafe { *<*const _>::from(self).cast::<u8>() }
1413///     }
1414/// }
1415///
1416/// let unit_like = Enum::Unit;
1417/// let tuple_like = Enum::Tuple(true);
1418/// let struct_like = Enum::Struct { a: false };
1419/// assert_eq!(0, unit_like.discriminant());
1420/// assert_eq!(1, tuple_like.discriminant());
1421/// assert_eq!(2, struct_like.discriminant());
1422///
1423/// // ⚠️ This is undefined behavior. Don't do this. ⚠️
1424/// // assert_eq!(0, unsafe { std::mem::transmute::<_, u8>(std::mem::discriminant(&unit_like)) });
1425/// ```
1426#[stable(feature = "discriminant_value", since = "1.21.0")]
1427#[rustc_const_stable(feature = "const_discriminant", since = "1.75.0")]
1428#[rustc_diagnostic_item = "mem_discriminant"]
1429#[cfg_attr(miri, track_caller)] // even without panics, this helps for Miri backtraces
1430#[ferrocene::prevalidated]
1431pub const fn discriminant<T>(v: &T) -> Discriminant<T> {
1432    Discriminant(intrinsics::discriminant_value(v))
1433}
1434
1435/// Returns the number of variants in the enum type `T`.
1436///
1437/// If `T` is not an enum, calling this function will not result in undefined behavior, but the
1438/// return value is unspecified. Equally, if `T` is an enum with more variants than `usize::MAX`
1439/// the return value is unspecified. Uninhabited variants will be counted.
1440///
1441/// Note that an enum may be expanded with additional variants in the future
1442/// as a non-breaking change, for example if it is marked `#[non_exhaustive]`,
1443/// which will change the result of this function.
1444///
1445/// # Examples
1446///
1447/// ```
1448/// # #![feature(never_type)]
1449/// # #![feature(variant_count)]
1450///
1451/// use std::mem;
1452///
1453/// enum Void {}
1454/// enum Foo { A(&'static str), B(i32), C(i32) }
1455///
1456/// assert_eq!(mem::variant_count::<Void>(), 0);
1457/// assert_eq!(mem::variant_count::<Foo>(), 3);
1458///
1459/// assert_eq!(mem::variant_count::<Option<!>>(), 2);
1460/// assert_eq!(mem::variant_count::<Result<!, !>>(), 2);
1461/// ```
1462#[inline(always)]
1463#[must_use]
1464#[unstable(feature = "variant_count", issue = "73662")]
1465#[rustc_const_unstable(feature = "variant_count", issue = "73662")]
1466#[rustc_diagnostic_item = "mem_variant_count"]
1467pub const fn variant_count<T>() -> usize {
1468    const { intrinsics::variant_count::<T>() }
1469}
1470
1471/// Provides associated constants for various useful properties of types,
1472/// to give them a canonical form in our code and make them easier to read.
1473///
1474/// This is here only to simplify all the ZST checks we need in the library.
1475/// It's not on a stabilization track right now.
1476#[doc(hidden)]
1477#[unstable(feature = "sized_type_properties", issue = "none")]
1478pub trait SizedTypeProperties: Sized {
1479    #[doc(hidden)]
1480    #[unstable(feature = "sized_type_properties", issue = "none")]
1481    #[lang = "mem_size_const"]
1482    const SIZE: usize = intrinsics::size_of::<Self>();
1483
1484    #[doc(hidden)]
1485    #[unstable(feature = "sized_type_properties", issue = "none")]
1486    #[lang = "mem_align_const"]
1487    const ALIGN: usize = intrinsics::align_of::<Self>();
1488
1489    #[doc(hidden)]
1490    #[unstable(feature = "ptr_alignment_type", issue = "102070")]
1491    const ALIGNMENT: Alignment = {
1492        // This can't panic since type alignment is always a power of two.
1493        Alignment::new(Self::ALIGN).unwrap()
1494    };
1495
1496    /// `true` if this type requires no storage.
1497    /// `false` if its [size](size_of) is greater than zero.
1498    ///
1499    /// # Examples
1500    ///
1501    /// ```
1502    /// #![feature(sized_type_properties)]
1503    /// use core::mem::SizedTypeProperties;
1504    ///
1505    /// fn do_something_with<T>() {
1506    ///     if T::IS_ZST {
1507    ///         // ... special approach ...
1508    ///     } else {
1509    ///         // ... the normal thing ...
1510    ///     }
1511    /// }
1512    ///
1513    /// struct MyUnit;
1514    /// assert!(MyUnit::IS_ZST);
1515    ///
1516    /// // For negative checks, consider using UFCS to emphasize the negation
1517    /// assert!(!<i32>::IS_ZST);
1518    /// // As it can sometimes hide in the type otherwise
1519    /// assert!(!String::IS_ZST);
1520    /// ```
1521    #[doc(hidden)]
1522    #[unstable(feature = "sized_type_properties", issue = "none")]
1523    const IS_ZST: bool = Self::SIZE == 0;
1524
1525    #[doc(hidden)]
1526    #[unstable(feature = "sized_type_properties", issue = "none")]
1527    const LAYOUT: Layout = {
1528        // SAFETY: if the type is instantiated, rustc already ensures that its
1529        // layout is valid. Use the unchecked constructor to avoid inserting a
1530        // panicking codepath that needs to be optimized out.
1531        unsafe { Layout::from_size_align_unchecked(Self::SIZE, Self::ALIGN) }
1532    };
1533
1534    /// The largest safe length for a `[Self]`.
1535    ///
1536    /// Anything larger than this would make `size_of_val` overflow `isize::MAX`,
1537    /// which is never allowed for a single object.
1538    #[doc(hidden)]
1539    #[unstable(feature = "sized_type_properties", issue = "none")]
1540    const MAX_SLICE_LEN: usize = match Self::SIZE {
1541        0 => usize::MAX,
1542        n => (isize::MAX as usize) / n,
1543    };
1544}
1545#[doc(hidden)]
1546#[unstable(feature = "sized_type_properties", issue = "none")]
1547impl<T> SizedTypeProperties for T {}
1548
1549/// Expands to the offset in bytes of a field from the beginning of the given type.
1550///
1551/// The type may be a `struct`, `enum`, `union`, or tuple.
1552///
1553/// The field may be a nested field (`field1.field2`), but not an array index.
1554/// The field must be visible to the call site.
1555///
1556/// The offset is returned as a [`usize`].
1557///
1558/// # Offsets of, and in, dynamically sized types
1559///
1560/// The field’s type must be [`Sized`], but it may be located in a [dynamically sized] container.
1561/// If the field type is dynamically sized, then you cannot use `offset_of!` (since the field's
1562/// alignment, and therefore its offset, may also be dynamic) and must take the offset from an
1563/// actual pointer to the container instead.
1564///
1565/// ```
1566/// # use core::mem;
1567/// # use core::fmt::Debug;
1568/// #[repr(C)]
1569/// pub struct Struct<T: ?Sized> {
1570///     a: u8,
1571///     b: T,
1572/// }
1573///
1574/// #[derive(Debug)]
1575/// #[repr(C, align(4))]
1576/// struct Align4(u32);
1577///
1578/// assert_eq!(mem::offset_of!(Struct<dyn Debug>, a), 0); // OK — Sized field
1579/// assert_eq!(mem::offset_of!(Struct<Align4>, b), 4); // OK — not DST
1580///
1581/// // assert_eq!(mem::offset_of!(Struct<dyn Debug>, b), 1);
1582/// // ^^^ error[E0277]: ... cannot be known at compilation time
1583///
1584/// // To obtain the offset of a !Sized field, examine a concrete value
1585/// // instead of using offset_of!.
1586/// let value: Struct<Align4> = Struct { a: 1, b: Align4(2) };
1587/// let ref_unsized: &Struct<dyn Debug> = &value;
1588/// let offset_of_b = unsafe {
1589///     (&raw const ref_unsized.b).byte_offset_from_unsigned(ref_unsized)
1590/// };
1591/// assert_eq!(offset_of_b, 4);
1592/// ```
1593///
1594/// If you need to obtain the offset of a field of a `!Sized` type, then, since the offset may
1595/// depend on the particular value being stored (in particular, `dyn Trait` values have a
1596/// dynamically-determined alignment), you must retrieve the offset from a specific reference
1597/// or pointer, and so you cannot use `offset_of!` to work without one.
1598///
1599/// # Layout is subject to change
1600///
1601/// Note that type layout is, in general, [subject to change and
1602/// platform-specific](https://doc.rust-lang.org/reference/type-layout.html). If
1603/// layout stability is required, consider using an [explicit `repr` attribute].
1604///
1605/// Rust guarantees that the offset of a given field within a given type will not
1606/// change over the lifetime of the program. However, two different compilations of
1607/// the same program may result in different layouts. Also, even within a single
1608/// program execution, no guarantees are made about types which are *similar* but
1609/// not *identical*, e.g.:
1610///
1611/// ```
1612/// struct Wrapper<T, U>(T, U);
1613///
1614/// type A = Wrapper<u8, u8>;
1615/// type B = Wrapper<u8, i8>;
1616///
1617/// // Not necessarily identical even though `u8` and `i8` have the same layout!
1618/// // assert_eq!(mem::offset_of!(A, 1), mem::offset_of!(B, 1));
1619///
1620/// #[repr(transparent)]
1621/// struct U8(u8);
1622///
1623/// type C = Wrapper<u8, U8>;
1624///
1625/// // Not necessarily identical even though `u8` and `U8` have the same layout!
1626/// // assert_eq!(mem::offset_of!(A, 1), mem::offset_of!(C, 1));
1627///
1628/// struct Empty<T>(core::marker::PhantomData<T>);
1629///
1630/// // Not necessarily identical even though `PhantomData` always has the same layout!
1631/// // assert_eq!(mem::offset_of!(Empty<u8>, 0), mem::offset_of!(Empty<i8>, 0));
1632/// ```
1633///
1634/// [explicit `repr` attribute]: https://doc.rust-lang.org/reference/type-layout.html#representations
1635///
1636/// # Unstable features
1637///
1638/// The following unstable features expand the functionality of `offset_of!`:
1639///
1640/// * [`offset_of_enum`] — allows `enum` variants to be traversed as if they were fields.
1641/// * [`offset_of_slice`] — allows getting the offset of a field of type `[T]`.
1642///
1643/// # Examples
1644///
1645/// ```
1646/// use std::mem;
1647/// #[repr(C)]
1648/// struct FieldStruct {
1649///     first: u8,
1650///     second: u16,
1651///     third: u8
1652/// }
1653///
1654/// assert_eq!(mem::offset_of!(FieldStruct, first), 0);
1655/// assert_eq!(mem::offset_of!(FieldStruct, second), 2);
1656/// assert_eq!(mem::offset_of!(FieldStruct, third), 4);
1657///
1658/// #[repr(C)]
1659/// struct NestedA {
1660///     b: NestedB
1661/// }
1662///
1663/// #[repr(C)]
1664/// struct NestedB(u8);
1665///
1666/// assert_eq!(mem::offset_of!(NestedA, b.0), 0);
1667/// ```
1668///
1669/// [dynamically sized]: https://doc.rust-lang.org/reference/dynamically-sized-types.html
1670/// [`offset_of_enum`]: https://doc.rust-lang.org/nightly/unstable-book/language-features/offset-of-enum.html
1671/// [`offset_of_slice`]: https://doc.rust-lang.org/nightly/unstable-book/language-features/offset-of-slice.html
1672#[stable(feature = "offset_of", since = "1.77.0")]
1673#[diagnostic::on_unmatched_args(
1674    note = "this macro expects a container type and a (nested) field path, like `offset_of!(Type, field)`"
1675)]
1676#[doc(alias = "memoffset")]
1677#[allow_internal_unstable(builtin_syntax, core_intrinsics)]
1678#[diagnostic::opaque]
1679pub macro offset_of($Container:ty, $($fields:expr)+ $(,)?) {
1680    const { builtin # offset_of($Container, $($fields)+) }
1681}
1682
1683/// Create a fresh instance of the inhabited ZST type `T`.
1684///
1685/// Prefer this to [`zeroed`] or [`uninitialized`] or [`transmute_copy`]
1686/// in places where you know that `T` is zero-sized, but don't have a bound
1687/// (such as [`Default`]) that would allow you to instantiate it using safe code.
1688///
1689/// If you're not sure whether `T` is an inhabited ZST, then you should be
1690/// using [`MaybeUninit`], not this function.
1691///
1692/// # Panics
1693///
1694/// If `size_of::<T>() != 0`.
1695///
1696/// # Safety
1697///
1698/// - `T` must be *[inhabited]*, i.e. possible to construct. This means that types
1699///   like zero-variant enums and [`!`] are unsound to conjure.
1700/// - You must use the value only in ways which do not violate any *safety*
1701///   invariants of the type.
1702///
1703/// While it's easy to create a *valid* instance of an inhabited ZST, since having
1704/// no bits in its representation means there's only one possible value, that
1705/// doesn't mean that it's always *sound* to do so.
1706///
1707/// For example, a library could design zero-sized tokens that are `!Default + !Clone`, limiting
1708/// their creation to functions that initialize some state or establish a scope. Conjuring such a
1709/// token could break invariants and lead to unsoundness.
1710///
1711/// # Examples
1712///
1713/// ```
1714/// #![feature(mem_conjure_zst)]
1715/// use std::mem::conjure_zst;
1716///
1717/// assert_eq!(unsafe { conjure_zst::<()>() }, ());
1718/// assert_eq!(unsafe { conjure_zst::<[i32; 0]>() }, []);
1719/// ```
1720///
1721/// [inhabited]: https://doc.rust-lang.org/reference/glossary.html#inhabited
1722#[unstable(feature = "mem_conjure_zst", issue = "95383")]
1723#[rustc_const_unstable(feature = "mem_conjure_zst", issue = "95383")]
1724#[ferrocene::prevalidated]
1725pub const unsafe fn conjure_zst<T>() -> T {
1726    #[ferrocene::annotation(
1727        "This assertion only runs in compilation, meaning that it cannot be covered in runtime"
1728    )]
1729    // Ferrocene addition: add curly braces in order to apply annotation to whole const_assert.
1730    {
1731        const_assert!(
1732            T::IS_ZST,
1733            "mem::conjure_zst invoked on a non-zero-sized type",
1734            "mem::conjure_zst invoked on type {name}, which is not zero-sized",
1735            name: &str = crate::any::type_name::<T>()
1736        );
1737    }
1738
1739    // SAFETY: because the caller must guarantee that it's inhabited and zero-sized,
1740    // there's nothing in the representation that needs to be set.
1741    // `assume_init` calls `assert_inhabited`, so we don't need to here.
1742    unsafe {
1743        #[allow(clippy::uninit_assumed_init)]
1744        MaybeUninit::uninit().assume_init()
1745    }
1746}