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 only safe to call if the following conditions hold:
427///
428/// - If `T` is `Sized`, this function is always safe to call.
429/// - If the unsized tail of `T` is:
430/// - a [slice], then the length of the slice tail must be an initialized
431/// integer, and the size of the *entire value*
432/// (dynamic tail length + statically sized prefix) must fit in `isize`.
433/// For the special case where the dynamic tail length is 0, this function
434/// is safe to call.
435// NOTE: the reason this is safe is that if an overflow were to occur already with size 0,
436// then we would stop compilation as even the "statically known" part of the type would
437// already be too big (or the call may be in dead code and optimized away, but then it
438// doesn't matter).
439/// - a [trait object], then the vtable part of the pointer must point
440/// to a valid vtable acquired by an unsizing coercion, and the size
441/// of the *entire value* (dynamic tail length + statically sized prefix)
442/// must fit in `isize`.
443/// - an (unstable) [extern type], then this function is always safe to
444/// call, but may panic or otherwise return the wrong value, as the
445/// extern type's layout is not known. This is the same behavior as
446/// [`size_of_val`] on a reference to a type with an extern type tail.
447/// - otherwise, it is conservatively not allowed to call this function.
448///
449/// [`size_of::<T>()`]: size_of
450/// [trait object]: ../../book/ch17-02-trait-objects.html
451/// [extern type]: ../../unstable-book/language-features/extern-types.html
452///
453/// # Examples
454///
455/// ```
456/// #![feature(layout_for_ptr)]
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#[unstable(feature = "layout_for_ptr", issue = "69835")]
468#[ferrocene::prevalidated]
469pub const unsafe fn size_of_val_raw<T: ?Sized>(val: *const T) -> usize {
470 // SAFETY: the caller must provide a valid raw pointer
471 unsafe { intrinsics::size_of_val(val) }
472}
473
474/// Returns the [ABI]-required minimum alignment of a type in bytes.
475///
476/// Every reference to a value of the type `T` must be a multiple of this number.
477///
478/// This is the alignment used for struct fields. It may be smaller than the preferred alignment.
479///
480/// [ABI]: https://en.wikipedia.org/wiki/Application_binary_interface
481///
482/// # Examples
483///
484/// ```
485/// # #![allow(deprecated)]
486/// use std::mem;
487///
488/// assert_eq!(4, mem::min_align_of::<i32>());
489/// ```
490#[inline]
491#[must_use]
492#[stable(feature = "rust1", since = "1.0.0")]
493#[deprecated(note = "use `align_of` instead", since = "1.2.0", suggestion = "align_of")]
494pub fn min_align_of<T>() -> usize {
495 <T as SizedTypeProperties>::ALIGN
496}
497
498/// Returns the [ABI]-required minimum alignment of the type of the value that `val` points to in
499/// bytes.
500///
501/// Every reference to a value of the type `T` must be a multiple of this number.
502///
503/// [ABI]: https://en.wikipedia.org/wiki/Application_binary_interface
504///
505/// # Examples
506///
507/// ```
508/// # #![allow(deprecated)]
509/// use std::mem;
510///
511/// assert_eq!(4, mem::min_align_of_val(&5i32));
512/// ```
513#[inline]
514#[must_use]
515#[stable(feature = "rust1", since = "1.0.0")]
516#[deprecated(note = "use `align_of_val` instead", since = "1.2.0", suggestion = "align_of_val")]
517pub fn min_align_of_val<T: ?Sized>(val: &T) -> usize {
518 // SAFETY: val is a reference, so it's a valid raw pointer
519 unsafe { intrinsics::align_of_val(val) }
520}
521
522/// Returns the [ABI]-required minimum alignment of a type, in bytes.
523///
524/// Every reference to a value of the type `T` must be a multiple of this number.
525///
526/// This is the alignment used for struct fields. It may be smaller than the preferred alignment.
527///
528/// [ABI]: https://en.wikipedia.org/wiki/Application_binary_interface
529///
530/// # Examples
531///
532/// ```
533/// assert_eq!(4, align_of::<i32>());
534/// ```
535///
536/// (Caution: [it is not guaranteed][type-layout] that the alignment of `i32` is `4`;
537/// that is, the above assertion does not pass on all platforms.)
538///
539/// [type-layout]: ../../reference/type-layout.html#r-layout.primitive
540#[inline(always)]
541#[must_use]
542#[stable(feature = "rust1", since = "1.0.0")]
543#[rustc_promotable]
544#[rustc_const_stable(feature = "const_align_of", since = "1.24.0")]
545#[rustc_diagnostic_item = "mem_align_of"]
546#[ferrocene::prevalidated]
547pub const fn align_of<T>() -> usize {
548 <T as SizedTypeProperties>::ALIGN
549}
550
551/// Returns the [ABI]-required minimum alignment of the type of the value that `val` points to, in
552/// bytes.
553///
554/// This function is identical to [`align_of::<T>()`][align_of] whenever <code>T: [Sized]</code>,
555/// but also supports determining the alignment required by a `dyn Trait` value, which is the
556/// alignment of the underlying concrete type.
557///
558/// [ABI]: https://en.wikipedia.org/wiki/Application_binary_interface
559///
560/// # Examples
561///
562/// ```
563/// assert_eq!(4, align_of_val(&5i32));
564/// ```
565///
566/// (Caution: [it is not guaranteed][type-layout] that the alignment of `i32` is `4`;
567/// that is, this example assertion does not pass on all platforms.)
568///
569/// `dyn` types may have different alignments for different values;
570/// `align_of_val` can be used to learn those alignments:
571///
572/// ```
573/// let a: &dyn ToString = &1234u16;
574/// let b: &dyn ToString = &String::from("abcd");
575///
576/// assert_eq!(align_of_val(a), align_of::<u16>());
577/// assert_eq!(align_of_val(b), align_of::<String>());
578/// ```
579///
580/// [type-layout]: ../../reference/type-layout.html#r-layout.primitive
581#[inline]
582#[must_use]
583#[stable(feature = "rust1", since = "1.0.0")]
584#[rustc_const_stable(feature = "const_align_of_val", since = "1.85.0")]
585#[ferrocene::prevalidated]
586pub const fn align_of_val<T: ?Sized>(val: &T) -> usize {
587 // SAFETY: val is a reference, so it's a valid raw pointer
588 unsafe { intrinsics::align_of_val(val) }
589}
590
591/// Returns the [ABI]-required minimum alignment of the type of the value that `val` points to, in
592/// bytes.
593///
594/// This function is identical to [`align_of_val()`], except that it can be used with raw pointers
595/// in situations where it would be unsound or undesirable to convert them to
596/// [`&` references][primitive@reference] and impose the aliasing rules that come with that.
597///
598/// [ABI]: https://en.wikipedia.org/wiki/Application_binary_interface
599///
600/// # Safety
601///
602/// This function is only safe to call if the following conditions hold:
603///
604/// - If `T` is `Sized`, this function is always safe to call.
605/// - If the unsized tail of `T` is:
606/// - a [slice], then the length of the slice tail must be an initialized
607/// integer, and the size of the *entire value*
608/// (dynamic tail length + statically sized prefix) must fit in `isize`.
609/// For the special case where the dynamic tail length is 0, this function
610/// is safe to call.
611/// - a [trait object], then the vtable part of the pointer must point
612/// to a valid vtable acquired by an unsizing coercion, and the size
613/// of the *entire value* (dynamic tail length + statically sized prefix)
614/// must fit in `isize`.
615/// - an (unstable) [extern type], then this function is always safe to
616/// call, but may panic or otherwise return the wrong value, as the
617/// extern type's layout is not known. This is the same behavior as
618/// [`align_of_val`] on a reference to a type with an extern type tail.
619/// - otherwise, it is conservatively not allowed to call this function.
620///
621/// [trait object]: ../../book/ch17-02-trait-objects.html
622/// [extern type]: ../../unstable-book/language-features/extern-types.html
623///
624/// # Examples
625///
626/// ```
627/// #![feature(layout_for_ptr)]
628/// use std::mem;
629///
630/// assert_eq!(4, unsafe { mem::align_of_val_raw(&5i32) });
631/// ```
632///
633/// (Caution: [it is not guaranteed][type-layout] that the alignment of `i32` is `4`;
634/// that is, the above assertion does not pass on all platforms.)
635///
636/// [type-layout]: ../../reference/type-layout.html#r-layout.primitive
637#[inline]
638#[must_use]
639#[unstable(feature = "layout_for_ptr", issue = "69835")]
640pub const unsafe fn align_of_val_raw<T: ?Sized>(val: *const T) -> usize {
641 // SAFETY: the caller must provide a valid raw pointer
642 unsafe { intrinsics::align_of_val(val) }
643}
644
645/// Returns `true` if dropping values of type `T` matters.
646///
647/// This is purely an optimization hint, and may be implemented conservatively:
648/// it may return `true` for types that don't actually need to be dropped.
649/// As such always returning `true` would be a valid implementation of
650/// this function. However if this function actually returns `false`, then you
651/// can be certain dropping `T` has no side effect.
652///
653/// Low level implementations of things like collections, which need to manually
654/// drop their data, should use this function to avoid unnecessarily
655/// trying to drop all their contents when they are destroyed. This might not
656/// make a difference in release builds (where a loop that has no side-effects
657/// is easily detected and eliminated), but is often a big win for debug builds.
658///
659/// Note that [`drop_in_place`] already performs this check, so if your workload
660/// can be reduced to some small number of [`drop_in_place`] calls, using this is
661/// unnecessary. In particular note that you can [`drop_in_place`] a slice, and that
662/// will do a single needs_drop check for all the values.
663///
664/// Types like Vec therefore just `drop_in_place(&mut self[..])` without using
665/// `needs_drop` explicitly. Types like [`HashMap`], on the other hand, have to drop
666/// values one at a time and should use this API.
667///
668/// [`drop_in_place`]: crate::ptr::drop_in_place
669/// [`HashMap`]: ../../std/collections/struct.HashMap.html
670///
671/// # Examples
672///
673/// Here's an example of how a collection might make use of `needs_drop`:
674///
675/// ```
676/// use std::{mem, ptr};
677///
678/// pub struct MyCollection<T> {
679/// # data: [T; 1],
680/// /* ... */
681/// }
682/// # impl<T> MyCollection<T> {
683/// # fn iter_mut(&mut self) -> &mut [T] { &mut self.data }
684/// # fn free_buffer(&mut self) {}
685/// # }
686///
687/// impl<T> Drop for MyCollection<T> {
688/// fn drop(&mut self) {
689/// unsafe {
690/// // drop the data
691/// if mem::needs_drop::<T>() {
692/// for x in self.iter_mut() {
693/// ptr::drop_in_place(x);
694/// }
695/// }
696/// self.free_buffer();
697/// }
698/// }
699/// }
700/// ```
701#[inline]
702#[must_use]
703#[stable(feature = "needs_drop", since = "1.21.0")]
704#[rustc_const_stable(feature = "const_mem_needs_drop", since = "1.36.0")]
705#[rustc_diagnostic_item = "needs_drop"]
706#[ferrocene::prevalidated]
707pub const fn needs_drop<T: ?Sized>() -> bool {
708 const { intrinsics::needs_drop::<T>() }
709}
710
711/// Returns the value of type `T` represented by the all-zero byte-pattern.
712///
713/// This means that, for example, the padding byte in `(u8, u16)` is not
714/// necessarily zeroed.
715///
716/// There is no guarantee that an all-zero byte-pattern represents a valid value
717/// of some type `T`. For example, the all-zero byte-pattern is not a valid value
718/// for reference types (`&T`, `&mut T`) and function pointers. Using `zeroed`
719/// on such types causes immediate [undefined behavior][ub] because [the Rust
720/// compiler assumes][inv] that there always is a valid value in a variable it
721/// considers initialized.
722///
723/// This has the same effect as [`MaybeUninit::zeroed().assume_init()`][zeroed].
724/// It is useful for FFI sometimes, but should generally be avoided.
725///
726/// [zeroed]: MaybeUninit::zeroed
727/// [ub]: ../../reference/behavior-considered-undefined.html
728/// [inv]: MaybeUninit#initialization-invariant
729///
730/// # Examples
731///
732/// Correct usage of this function: initializing an integer with zero.
733///
734/// ```
735/// use std::mem;
736///
737/// let x: i32 = unsafe { mem::zeroed() };
738/// assert_eq!(0, x);
739/// ```
740///
741/// *Incorrect* usage of this function: initializing a reference with zero.
742///
743/// ```rust,no_run
744/// # #![allow(invalid_value)]
745/// use std::mem;
746///
747/// let _x: &i32 = unsafe { mem::zeroed() }; // Undefined behavior!
748/// let _y: fn() = unsafe { mem::zeroed() }; // And again!
749/// ```
750#[inline(always)]
751#[must_use]
752#[stable(feature = "rust1", since = "1.0.0")]
753#[rustc_diagnostic_item = "mem_zeroed"]
754#[track_caller]
755#[rustc_const_stable(feature = "const_mem_zeroed", since = "1.75.0")]
756#[ferrocene::prevalidated]
757pub const unsafe fn zeroed<T>() -> T {
758 // SAFETY: the caller must guarantee that an all-zero value is valid for `T`.
759 unsafe {
760 intrinsics::assert_zero_valid::<T>();
761 MaybeUninit::zeroed().assume_init()
762 }
763}
764
765/// Bypasses Rust's normal memory-initialization checks by pretending to
766/// produce a value of type `T`, while doing nothing at all.
767///
768/// **This function is deprecated.** Use [`MaybeUninit<T>`] instead.
769/// It also might be slower than using `MaybeUninit<T>` due to mitigations that were put in place to
770/// limit the potential harm caused by incorrect use of this function in legacy code.
771///
772/// The reason for deprecation is that the function basically cannot be used
773/// correctly: it has the same effect as [`MaybeUninit::uninit().assume_init()`][uninit].
774/// As the [`assume_init` documentation][assume_init] explains,
775/// [the Rust compiler assumes][inv] that values are properly initialized.
776///
777/// Truly uninitialized memory like what gets returned here
778/// is special in that the compiler knows that it does not have a fixed value.
779/// This makes it undefined behavior to have uninitialized data in a variable even
780/// if that variable has an integer type.
781///
782/// Therefore, it is immediate undefined behavior to call this function on nearly all types,
783/// including integer types and arrays of integer types, and even if the result is unused.
784///
785/// [uninit]: MaybeUninit::uninit
786/// [assume_init]: MaybeUninit::assume_init
787/// [inv]: MaybeUninit#initialization-invariant
788#[inline(always)]
789#[must_use]
790#[deprecated(since = "1.39.0", note = "use `mem::MaybeUninit` instead")]
791#[stable(feature = "rust1", since = "1.0.0")]
792#[rustc_diagnostic_item = "mem_uninitialized"]
793#[track_caller]
794pub unsafe fn uninitialized<T>() -> T {
795 // SAFETY: the caller must guarantee that an uninitialized value is valid for `T`.
796 unsafe {
797 intrinsics::assert_mem_uninitialized_valid::<T>();
798 let mut val = MaybeUninit::<T>::uninit();
799
800 // Fill memory with 0x01, as an imperfect mitigation for old code that uses this function on
801 // bool, nonnull, and noundef types. But don't do this if we actively want to detect UB.
802 if !cfg!(any(miri, sanitize = "memory")) {
803 val.as_mut_ptr().write_bytes(0x01, 1);
804 }
805
806 val.assume_init()
807 }
808}
809
810/// Swaps the values at two mutable locations, without deinitializing either one.
811///
812/// * If you want to swap with a default or dummy value, see [`take`].
813/// * If you want to swap with a passed value, returning the old value, see [`replace`].
814///
815/// # Examples
816///
817/// ```
818/// use std::mem;
819///
820/// let mut x = 5;
821/// let mut y = 42;
822///
823/// mem::swap(&mut x, &mut y);
824///
825/// assert_eq!(42, x);
826/// assert_eq!(5, y);
827/// ```
828#[inline]
829#[stable(feature = "rust1", since = "1.0.0")]
830#[rustc_const_stable(feature = "const_swap", since = "1.85.0")]
831#[rustc_diagnostic_item = "mem_swap"]
832#[ferrocene::prevalidated]
833pub const fn swap<T>(x: &mut T, y: &mut T) {
834 // SAFETY: `&mut` guarantees these are typed readable and writable
835 // as well as non-overlapping.
836 unsafe { intrinsics::typed_swap_nonoverlapping(x, y) }
837}
838
839/// Replaces `dest` with the default value of `T`, returning the previous `dest` value.
840///
841/// * If you want to replace the values of two variables, see [`swap`].
842/// * If you want to replace with a passed value instead of the default value, see [`replace`].
843///
844/// # Examples
845///
846/// A simple example:
847///
848/// ```
849/// use std::mem;
850///
851/// let mut v: Vec<i32> = vec![1, 2];
852///
853/// let old_v = mem::take(&mut v);
854/// assert_eq!(vec![1, 2], old_v);
855/// assert!(v.is_empty());
856/// ```
857///
858/// `take` allows taking ownership of a struct field by replacing it with an "empty" value.
859/// Without `take` you can run into issues like these:
860///
861/// ```compile_fail,E0507
862/// struct Buffer<T> { buf: Vec<T> }
863///
864/// impl<T> Buffer<T> {
865/// fn get_and_reset(&mut self) -> Vec<T> {
866/// // error: cannot move out of dereference of `&mut`-pointer
867/// let buf = self.buf;
868/// self.buf = Vec::new();
869/// buf
870/// }
871/// }
872/// ```
873///
874/// Note that `T` does not necessarily implement [`Clone`], so it can't even clone and reset
875/// `self.buf`. But `take` can be used to disassociate the original value of `self.buf` from
876/// `self`, allowing it to be returned:
877///
878/// ```
879/// use std::mem;
880///
881/// # struct Buffer<T> { buf: Vec<T> }
882/// impl<T> Buffer<T> {
883/// fn get_and_reset(&mut self) -> Vec<T> {
884/// mem::take(&mut self.buf)
885/// }
886/// }
887///
888/// let mut buffer = Buffer { buf: vec![0, 1] };
889/// assert_eq!(buffer.buf.len(), 2);
890///
891/// assert_eq!(buffer.get_and_reset(), vec![0, 1]);
892/// assert_eq!(buffer.buf.len(), 0);
893/// ```
894#[inline]
895#[stable(feature = "mem_take", since = "1.40.0")]
896#[rustc_const_unstable(feature = "const_default", issue = "143894")]
897#[ferrocene::prevalidated]
898pub const fn take<T: [const] Default>(dest: &mut T) -> T {
899 replace(dest, T::default())
900}
901
902/// Moves `src` into the referenced `dest`, returning the previous `dest` value.
903///
904/// Neither value is dropped.
905///
906/// * If you want to replace the values of two variables, see [`swap`].
907/// * If you want to replace with a default value, see [`take`].
908///
909/// # Examples
910///
911/// A simple example:
912///
913/// ```
914/// use std::mem;
915///
916/// let mut v: Vec<i32> = vec![1, 2];
917///
918/// let old_v = mem::replace(&mut v, vec![3, 4, 5]);
919/// assert_eq!(vec![1, 2], old_v);
920/// assert_eq!(vec![3, 4, 5], v);
921/// ```
922///
923/// `replace` allows consumption of a struct field by replacing it with another value.
924/// Without `replace` you can run into issues like these:
925///
926/// ```compile_fail,E0507
927/// struct Buffer<T> { buf: Vec<T> }
928///
929/// impl<T> Buffer<T> {
930/// fn replace_index(&mut self, i: usize, v: T) -> T {
931/// // error: cannot move out of dereference of `&mut`-pointer
932/// let t = self.buf[i];
933/// self.buf[i] = v;
934/// t
935/// }
936/// }
937/// ```
938///
939/// Note that `T` does not necessarily implement [`Clone`], so we can't even clone `self.buf[i]` to
940/// avoid the move. But `replace` can be used to disassociate the original value at that index from
941/// `self`, allowing it to be returned:
942///
943/// ```
944/// # #![allow(dead_code)]
945/// use std::mem;
946///
947/// # struct Buffer<T> { buf: Vec<T> }
948/// impl<T> Buffer<T> {
949/// fn replace_index(&mut self, i: usize, v: T) -> T {
950/// mem::replace(&mut self.buf[i], v)
951/// }
952/// }
953///
954/// let mut buffer = Buffer { buf: vec![0, 1] };
955/// assert_eq!(buffer.buf[0], 0);
956///
957/// assert_eq!(buffer.replace_index(0, 2), 0);
958/// assert_eq!(buffer.buf[0], 2);
959/// ```
960#[inline]
961#[stable(feature = "rust1", since = "1.0.0")]
962#[must_use = "if you don't need the old value, you can just assign the new value directly"]
963#[rustc_const_stable(feature = "const_replace", since = "1.83.0")]
964#[rustc_diagnostic_item = "mem_replace"]
965#[ferrocene::prevalidated]
966pub const fn replace<T>(dest: &mut T, src: T) -> T {
967 // It may be tempting to use `swap` to avoid `unsafe` here. Don't!
968 // The compiler optimizes the implementation below to two `memcpy`s
969 // while `swap` would require at least three. See PR#83022 for details.
970
971 // SAFETY: We read from `dest` but directly write `src` into it afterwards,
972 // such that the old value is not duplicated. Nothing is dropped and
973 // nothing here can panic.
974 unsafe {
975 // Ideally we wouldn't use the intrinsics here, but going through the
976 // `ptr` methods introduces two unnecessary UbChecks, so until we can
977 // remove those for pointers that come from references, this uses the
978 // intrinsics instead so this stays very cheap in MIR (and debug).
979
980 let result = crate::intrinsics::read_via_copy(dest);
981 crate::intrinsics::write_via_move(dest, src);
982 result
983 }
984}
985
986/// Disposes of a value.
987///
988/// This effectively does nothing for types which implement `Copy`, e.g.
989/// integers. Such values are copied and _then_ moved into the function, so the
990/// value persists after this function call.
991///
992/// This function is not magic; it is literally defined as
993///
994/// ```
995/// pub fn drop<T>(_x: T) {}
996/// ```
997///
998/// Because `_x` is moved into the function, it is automatically [dropped][drop] before
999/// the function returns.
1000///
1001/// [drop]: Drop
1002///
1003/// # Examples
1004///
1005/// Basic usage:
1006///
1007/// ```
1008/// let v = vec![1, 2, 3];
1009///
1010/// drop(v); // explicitly drop the vector
1011/// ```
1012///
1013/// Since [`RefCell`] enforces the borrow rules at runtime, `drop` can
1014/// release a [`RefCell`] borrow:
1015///
1016/// ```
1017/// use std::cell::RefCell;
1018///
1019/// let x = RefCell::new(1);
1020///
1021/// let mut mutable_borrow = x.borrow_mut();
1022/// *mutable_borrow = 1;
1023///
1024/// drop(mutable_borrow); // relinquish the mutable borrow on this slot
1025///
1026/// let borrow = x.borrow();
1027/// println!("{}", *borrow);
1028/// ```
1029///
1030/// Integers and other types implementing [`Copy`] are unaffected by `drop`.
1031///
1032/// ```
1033/// # #![allow(dropping_copy_types)]
1034/// #[derive(Copy, Clone)]
1035/// struct Foo(u8);
1036///
1037/// let x = 1;
1038/// let y = Foo(2);
1039/// drop(x); // a copy of `x` is moved and dropped
1040/// drop(y); // a copy of `y` is moved and dropped
1041///
1042/// println!("x: {}, y: {}", x, y.0); // still available
1043/// ```
1044///
1045/// [`RefCell`]: crate::cell::RefCell
1046#[inline]
1047#[stable(feature = "rust1", since = "1.0.0")]
1048#[rustc_const_unstable(feature = "const_destruct", issue = "133214")]
1049#[rustc_diagnostic_item = "mem_drop"]
1050#[ferrocene::prevalidated]
1051pub const fn drop<T>(_x: T)
1052where
1053 T: [const] Destruct,
1054{
1055}
1056
1057/// Bitwise-copies a value.
1058///
1059/// This function is not magic; it is literally defined as
1060/// ```
1061/// pub const fn copy<T: Copy>(x: &T) -> T { *x }
1062/// ```
1063///
1064/// It is useful when you want to pass a function pointer to a combinator, rather than defining a new closure.
1065///
1066/// Example:
1067/// ```
1068/// #![feature(mem_copy_fn)]
1069/// use core::mem::copy;
1070/// let result_from_ffi_function: Result<(), &i32> = Err(&1);
1071/// let result_copied: Result<(), i32> = result_from_ffi_function.map_err(copy);
1072/// ```
1073#[inline]
1074#[unstable(feature = "mem_copy_fn", issue = "98262")]
1075pub const fn copy<T: Copy>(x: &T) -> T {
1076 *x
1077}
1078
1079/// Interprets `src` as having type `&Dst`, and then reads `src` without moving
1080/// the contained value.
1081///
1082/// This function will unsafely assume the pointer `src` is valid for [`size_of::<Dst>`][size_of]
1083/// bytes by transmuting `&Src` to `&Dst` and then reading the `&Dst` (except that this is done
1084/// in a way that is correct even when `&Dst` has stricter alignment requirements than `&Src`).
1085/// It will also unsafely create a copy of the contained value instead of moving out of `src`.
1086///
1087/// It is not a compile-time error if `Src` and `Dst` have different sizes, but it
1088/// is highly encouraged to only invoke this function where `Src` and `Dst` have the
1089/// same size. This function triggers [undefined behavior][ub] if `Dst` is larger than
1090/// `Src`.
1091///
1092/// [ub]: ../../reference/behavior-considered-undefined.html
1093///
1094/// If you have a raw pointer instead of a reference, you might be looking for
1095/// `src.cast::<Dst>().`[`read_unaligned()`](pointer#method.read_unaligned) instead.
1096///
1097/// # Safety
1098///
1099/// - Requires `size_of_val::<Src>(src) >= size_of::<Dst>()`
1100/// - The first `size_of::<Dst>()` bytes behind `src` must be *readable*
1101/// - The first `size_of::<Dst>()` bytes behind `src` must be *[valid]*
1102/// when interpreted as a `Dst`.
1103///
1104/// On top of that, remember that most types have additional invariants beyond merely
1105/// being considered initialized at the type level. For example, a `1`-initialized [`Vec<T>`]
1106/// is considered initialized (under the current implementation; this does not constitute
1107/// a stable guarantee) because the only requirement the compiler knows about it
1108/// is that the data pointer must be non-null. Creating such a `Vec<T>` does not cause
1109/// *immediate* undefined behavior, but will cause undefined behavior with most
1110/// safe operations (including dropping it).
1111///
1112/// [valid]: ../../reference/behavior-considered-undefined.html#r-undefined.validity
1113/// [`Vec<T>`]: ../../std/vec/struct.Vec.html
1114///
1115/// # Examples
1116///
1117/// ```
1118/// use std::mem;
1119///
1120/// #[repr(packed)]
1121/// struct Foo {
1122/// bar: u8,
1123/// }
1124///
1125/// let foo_array = [10u8];
1126///
1127/// unsafe {
1128/// // Copy the data from 'foo_array' and treat it as a 'Foo'
1129/// let mut foo_struct: Foo = mem::transmute_copy(&foo_array);
1130/// assert_eq!(foo_struct.bar, 10);
1131///
1132/// // Modify the copied data
1133/// foo_struct.bar = 20;
1134/// assert_eq!(foo_struct.bar, 20);
1135/// }
1136///
1137/// // The contents of 'foo_array' should not have changed
1138/// assert_eq!(foo_array, [10]);
1139///
1140/// let bytes: &[u8] = &[1, 2, 3, 4, 5, 6, 7];
1141/// assert_eq!(
1142/// unsafe { mem::transmute_copy::<[u8], u32>(bytes) },
1143/// u32::from_ne_bytes(*bytes.first_chunk().unwrap()),
1144/// );
1145/// ```
1146#[ferrocene::prevalidated]
1147#[inline]
1148#[must_use]
1149#[track_caller]
1150#[stable(feature = "rust1", since = "1.0.0")]
1151#[rustc_const_stable(feature = "const_transmute_copy", since = "1.74.0")]
1152pub const unsafe fn transmute_copy<Src: ?Sized, Dst>(src: &Src) -> Dst {
1153 // library UB because it's possible for the `Src` to be only a subset of the allocation
1154 // and thus for a failure to not be immediate language UB
1155 assert_unsafe_precondition!(
1156 check_library_ub,
1157 "cannot transmute_copy if Dst is larger than Src",
1158 (
1159 src_size: usize = size_of_val::<Src>(src),
1160 dst_size: usize = Dst::SIZE,
1161 ) => src_size >= dst_size
1162 );
1163
1164 // If Dst has a higher alignment requirement, src might not be suitably aligned.
1165 if align_of::<Dst>() > align_of_val::<Src>(src) {
1166 // SAFETY: `src` is a reference which is guaranteed to be valid for reads.
1167 // The caller must guarantee that the actual transmutation is safe.
1168 unsafe { ptr::read_unaligned(src as *const Src as *const Dst) }
1169 } else {
1170 // SAFETY: `src` is a reference which is guaranteed to be valid for reads.
1171 // We just checked that `src as *const Dst` was properly aligned.
1172 // The caller must guarantee that the actual transmutation is safe.
1173 unsafe { ptr::read(src as *const Src as *const Dst) }
1174 }
1175}
1176
1177/// Like [`transmute`], but only initializes the "common prefix" of the first
1178/// `min(size_of::<Src>(), size_of::<Dst>())` bytes of the destination from the
1179/// corresponding bytes of the source.
1180///
1181/// This is equivalent to a "union cast" through a `union` with `#[repr(C)]`.
1182///
1183/// That means some size mismatches are not UB, like `[T; 2]` to `[T; 1]`.
1184/// Increasing size is usually UB from being insufficiently initialized -- like
1185/// `u8` to `u32` -- but isn't always. For example, going from `u8` to
1186/// `#[repr(C, align(4))] AlignedU8(u8);` is sound.
1187///
1188/// Prefer normal `transmute` where possible, for the extra checking, since
1189/// both do exactly the same thing at runtime, if they both compile.
1190///
1191/// # Safety
1192///
1193/// If `size_of::<Src>() >= size_of::<Dst>()`, the first `size_of::<Dst>()` bytes
1194/// of `src` must be be *valid* when interpreted as a `Dst`. (In this case, the
1195/// preconditions are the same as for `transmute_copy(&ManuallyDrop::new(src))`.)
1196///
1197/// If `size_of::<Src>() <= size_of::<Dst>()`, the bytes of `src` padded with
1198/// uninitialized bytes afterwards up to a total size of `size_of::<Dst>()`
1199/// must be *valid* when interpreted as a `Dst`.
1200///
1201/// In both cases, any safety preconditions of the `Dst` type must also be upheld.
1202///
1203/// # Examples
1204///
1205/// ```
1206/// #![feature(transmute_prefix)]
1207/// use std::mem::transmute_prefix;
1208///
1209/// assert_eq!(unsafe { transmute_prefix::<[i32; 4], [i32; 2]>([1, 2, 3, 4]) }, [1, 2]);
1210///
1211/// let expected = if cfg!(target_endian = "little") { 0x34 } else { 0x12 };
1212/// assert_eq!(unsafe { transmute_prefix::<u16, u8>(0x1234) }, expected);
1213///
1214/// // Would be UB because the destination is incompletely initialized.
1215/// // transmute_prefix::<u8, u16>(123)
1216///
1217/// // OK because the destination is allowed to be partially initialized.
1218/// let _: std::mem::MaybeUninit<u16> = unsafe { transmute_prefix(123_u8) };
1219/// ```
1220#[unstable(feature = "transmute_prefix", issue = "155079")]
1221#[rustc_no_writable]
1222pub const unsafe fn transmute_prefix<Src, Dst>(src: Src) -> Dst {
1223 #[repr(C)]
1224 union Transmute<A, B> {
1225 a: ManuallyDrop<A>,
1226 b: ManuallyDrop<B>,
1227 }
1228
1229 match const { Ord::cmp(&Src::SIZE, &Dst::SIZE) } {
1230 // SAFETY: When Dst is bigger, the union is the size of Dst
1231 Ordering::Less => unsafe {
1232 let a = transmute_neo(src);
1233 intrinsics::transmute_unchecked(Transmute::<Src, Dst> { a })
1234 },
1235 // SAFETY: When they're the same size, we can use the MIR primitive
1236 Ordering::Equal => unsafe { intrinsics::transmute_unchecked::<Src, Dst>(src) },
1237 // SAFETY: When Src is bigger, the union is the size of Src
1238 Ordering::Greater => unsafe {
1239 let u: Transmute<Src, Dst> = intrinsics::transmute_unchecked(src);
1240 transmute_neo(u.b)
1241 },
1242 }
1243}
1244
1245/// New version of `transmute`, exposed under this name so it can be iterated upon
1246/// without risking breakage to uses of "real" transmute.
1247///
1248/// Uses a `const`-`assert` to check the sizes instead of typeck hacks,
1249/// but is semantially identical to `transmute` otherwise.
1250///
1251/// It will not be stabilized under this name.
1252///
1253/// # Examples
1254///
1255/// ```
1256/// #![feature(transmute_neo)]
1257/// use std::mem::transmute_neo;
1258///
1259/// assert_eq!(unsafe { transmute_neo::<f32, u32>(0.0) }, 0);
1260/// ```
1261///
1262/// ```compile_fail,E0080
1263/// #![feature(transmute_neo)]
1264/// use std::mem::transmute_neo;
1265///
1266/// unsafe { transmute_neo::<u32, u16>(123) };
1267/// ```
1268#[ferrocene::prevalidated]
1269#[unstable(feature = "transmute_neo", issue = "155079")]
1270#[cfg_attr(miri, track_caller)] // even without panics, this helps for Miri backtraces
1271#[inline]
1272#[rustc_no_writable]
1273pub const unsafe fn transmute_neo<Src, Dst>(src: Src) -> Dst {
1274 const { assert!(Src::SIZE == Dst::SIZE) };
1275
1276 // SAFETY: the const-assert just checked that they're the same size,
1277 // and any other safety invariants need to be upheld by the caller.
1278 unsafe { intrinsics::transmute_unchecked(src) }
1279}
1280
1281/// Opaque type representing the discriminant of an enum.
1282///
1283/// See the [`discriminant`] function in this module for more information.
1284#[stable(feature = "discriminant_value", since = "1.21.0")]
1285#[ferrocene::prevalidated]
1286pub struct Discriminant<T>(<T as DiscriminantKind>::Discriminant);
1287
1288// N.B. These trait implementations cannot be derived because we don't want any bounds on T.
1289
1290#[stable(feature = "discriminant_value", since = "1.21.0")]
1291impl<T> Copy for Discriminant<T> {}
1292
1293#[stable(feature = "discriminant_value", since = "1.21.0")]
1294impl<T> clone::Clone for Discriminant<T> {
1295 fn clone(&self) -> Self {
1296 *self
1297 }
1298}
1299
1300#[doc(hidden)]
1301#[unstable(feature = "trivial_clone", issue = "none")]
1302unsafe impl<T> TrivialClone for Discriminant<T> {}
1303
1304#[stable(feature = "discriminant_value", since = "1.21.0")]
1305impl<T> cmp::PartialEq for Discriminant<T> {
1306 #[ferrocene::prevalidated]
1307 fn eq(&self, rhs: &Self) -> bool {
1308 self.0 == rhs.0
1309 }
1310}
1311
1312#[stable(feature = "discriminant_value", since = "1.21.0")]
1313impl<T> cmp::Eq for Discriminant<T> {}
1314
1315#[stable(feature = "discriminant_value", since = "1.21.0")]
1316impl<T> hash::Hash for Discriminant<T> {
1317 fn hash<H: hash::Hasher>(&self, state: &mut H) {
1318 self.0.hash(state);
1319 }
1320}
1321
1322#[stable(feature = "discriminant_value", since = "1.21.0")]
1323impl<T> fmt::Debug for Discriminant<T> {
1324 #[ferrocene::prevalidated]
1325 fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
1326 fmt.debug_tuple("Discriminant").field(&self.0).finish()
1327 }
1328}
1329
1330/// Returns a value uniquely identifying the enum variant in `v`.
1331///
1332/// If `T` is not an enum, calling this function will not result in undefined behavior, but the
1333/// return value is unspecified.
1334///
1335/// # Stability
1336///
1337/// The discriminant of an enum variant may change if the enum definition changes. A discriminant
1338/// of some variant will not change between compilations with the same compiler. See the [Reference]
1339/// for more information.
1340///
1341/// [Reference]: ../../reference/items/enumerations.html#custom-discriminant-values-for-fieldless-enumerations
1342///
1343/// The value of a [`Discriminant<T>`] is independent of any *free lifetimes* in `T`. As such,
1344/// reading or writing a `Discriminant<Foo<'a>>` as a `Discriminant<Foo<'b>>` (whether via
1345/// [`transmute`] or otherwise) is always sound. Note that this is **not** true for other kinds
1346/// of generic parameters and for higher-ranked lifetimes; `Discriminant<Foo<A>>` and
1347/// `Discriminant<Foo<B>>` as well as `Discriminant<Bar<dyn for<'a> Trait<'a>>>` and
1348/// `Discriminant<Bar<dyn Trait<'static>>>` may be incompatible.
1349///
1350/// # Examples
1351///
1352/// This can be used to compare enums that carry data, while disregarding
1353/// the actual data:
1354///
1355/// ```
1356/// use std::mem;
1357///
1358/// enum Foo { A(&'static str), B(i32), C(i32) }
1359///
1360/// assert_eq!(mem::discriminant(&Foo::A("bar")), mem::discriminant(&Foo::A("baz")));
1361/// assert_eq!(mem::discriminant(&Foo::B(1)), mem::discriminant(&Foo::B(2)));
1362/// assert_ne!(mem::discriminant(&Foo::B(3)), mem::discriminant(&Foo::C(3)));
1363/// ```
1364///
1365/// ## Accessing the numeric value of the discriminant
1366///
1367/// Note that it is *undefined behavior* to [`transmute`] from [`Discriminant`] to a primitive!
1368///
1369/// If an enum has only unit variants, then the numeric value of the discriminant can be accessed
1370/// with an [`as`] cast:
1371///
1372/// ```
1373/// enum Enum {
1374/// Foo,
1375/// Bar,
1376/// Baz,
1377/// }
1378///
1379/// assert_eq!(0, Enum::Foo as isize);
1380/// assert_eq!(1, Enum::Bar as isize);
1381/// assert_eq!(2, Enum::Baz as isize);
1382/// ```
1383///
1384/// If an enum has opted-in to having a [primitive representation] for its discriminant,
1385/// then it's possible to use pointers to read the memory location storing the discriminant.
1386/// That **cannot** be done for enums using the [default representation], however, as it's
1387/// undefined what layout the discriminant has and where it's stored — it might not even be
1388/// stored at all!
1389///
1390/// [`as`]: ../../std/keyword.as.html
1391/// [primitive representation]: ../../reference/type-layout.html#primitive-representations
1392/// [default representation]: ../../reference/type-layout.html#the-default-representation
1393/// ```
1394/// #[repr(u8)]
1395/// enum Enum {
1396/// Unit,
1397/// Tuple(bool),
1398/// Struct { a: bool },
1399/// }
1400///
1401/// impl Enum {
1402/// fn discriminant(&self) -> u8 {
1403/// // SAFETY: Because `Self` is marked `repr(u8)`, its layout is a `repr(C)` `union`
1404/// // between `repr(C)` structs, each of which has the `u8` discriminant as its first
1405/// // field, so we can read the discriminant without offsetting the pointer.
1406/// unsafe { *<*const _>::from(self).cast::<u8>() }
1407/// }
1408/// }
1409///
1410/// let unit_like = Enum::Unit;
1411/// let tuple_like = Enum::Tuple(true);
1412/// let struct_like = Enum::Struct { a: false };
1413/// assert_eq!(0, unit_like.discriminant());
1414/// assert_eq!(1, tuple_like.discriminant());
1415/// assert_eq!(2, struct_like.discriminant());
1416///
1417/// // ⚠️ This is undefined behavior. Don't do this. ⚠️
1418/// // assert_eq!(0, unsafe { std::mem::transmute::<_, u8>(std::mem::discriminant(&unit_like)) });
1419/// ```
1420#[stable(feature = "discriminant_value", since = "1.21.0")]
1421#[rustc_const_stable(feature = "const_discriminant", since = "1.75.0")]
1422#[rustc_diagnostic_item = "mem_discriminant"]
1423#[cfg_attr(miri, track_caller)] // even without panics, this helps for Miri backtraces
1424#[ferrocene::prevalidated]
1425pub const fn discriminant<T>(v: &T) -> Discriminant<T> {
1426 Discriminant(intrinsics::discriminant_value(v))
1427}
1428
1429/// Returns the number of variants in the enum type `T`.
1430///
1431/// If `T` is not an enum, calling this function will not result in undefined behavior, but the
1432/// return value is unspecified. Equally, if `T` is an enum with more variants than `usize::MAX`
1433/// the return value is unspecified. Uninhabited variants will be counted.
1434///
1435/// Note that an enum may be expanded with additional variants in the future
1436/// as a non-breaking change, for example if it is marked `#[non_exhaustive]`,
1437/// which will change the result of this function.
1438///
1439/// # Examples
1440///
1441/// ```
1442/// # #![feature(never_type)]
1443/// # #![feature(variant_count)]
1444///
1445/// use std::mem;
1446///
1447/// enum Void {}
1448/// enum Foo { A(&'static str), B(i32), C(i32) }
1449///
1450/// assert_eq!(mem::variant_count::<Void>(), 0);
1451/// assert_eq!(mem::variant_count::<Foo>(), 3);
1452///
1453/// assert_eq!(mem::variant_count::<Option<!>>(), 2);
1454/// assert_eq!(mem::variant_count::<Result<!, !>>(), 2);
1455/// ```
1456#[inline(always)]
1457#[must_use]
1458#[unstable(feature = "variant_count", issue = "73662")]
1459#[rustc_const_unstable(feature = "variant_count", issue = "73662")]
1460#[rustc_diagnostic_item = "mem_variant_count"]
1461pub const fn variant_count<T>() -> usize {
1462 const { intrinsics::variant_count::<T>() }
1463}
1464
1465/// Provides associated constants for various useful properties of types,
1466/// to give them a canonical form in our code and make them easier to read.
1467///
1468/// This is here only to simplify all the ZST checks we need in the library.
1469/// It's not on a stabilization track right now.
1470#[doc(hidden)]
1471#[unstable(feature = "sized_type_properties", issue = "none")]
1472pub trait SizedTypeProperties: Sized {
1473 #[doc(hidden)]
1474 #[unstable(feature = "sized_type_properties", issue = "none")]
1475 #[lang = "mem_size_const"]
1476 const SIZE: usize = intrinsics::size_of::<Self>();
1477
1478 #[doc(hidden)]
1479 #[unstable(feature = "sized_type_properties", issue = "none")]
1480 #[lang = "mem_align_const"]
1481 const ALIGN: usize = intrinsics::align_of::<Self>();
1482
1483 #[doc(hidden)]
1484 #[unstable(feature = "ptr_alignment_type", issue = "102070")]
1485 const ALIGNMENT: Alignment = {
1486 // This can't panic since type alignment is always a power of two.
1487 Alignment::new(Self::ALIGN).unwrap()
1488 };
1489
1490 /// `true` if this type requires no storage.
1491 /// `false` if its [size](size_of) is greater than zero.
1492 ///
1493 /// # Examples
1494 ///
1495 /// ```
1496 /// #![feature(sized_type_properties)]
1497 /// use core::mem::SizedTypeProperties;
1498 ///
1499 /// fn do_something_with<T>() {
1500 /// if T::IS_ZST {
1501 /// // ... special approach ...
1502 /// } else {
1503 /// // ... the normal thing ...
1504 /// }
1505 /// }
1506 ///
1507 /// struct MyUnit;
1508 /// assert!(MyUnit::IS_ZST);
1509 ///
1510 /// // For negative checks, consider using UFCS to emphasize the negation
1511 /// assert!(!<i32>::IS_ZST);
1512 /// // As it can sometimes hide in the type otherwise
1513 /// assert!(!String::IS_ZST);
1514 /// ```
1515 #[doc(hidden)]
1516 #[unstable(feature = "sized_type_properties", issue = "none")]
1517 const IS_ZST: bool = Self::SIZE == 0;
1518
1519 #[doc(hidden)]
1520 #[unstable(feature = "sized_type_properties", issue = "none")]
1521 const LAYOUT: Layout = {
1522 // SAFETY: if the type is instantiated, rustc already ensures that its
1523 // layout is valid. Use the unchecked constructor to avoid inserting a
1524 // panicking codepath that needs to be optimized out.
1525 unsafe { Layout::from_size_align_unchecked(Self::SIZE, Self::ALIGN) }
1526 };
1527
1528 /// The largest safe length for a `[Self]`.
1529 ///
1530 /// Anything larger than this would make `size_of_val` overflow `isize::MAX`,
1531 /// which is never allowed for a single object.
1532 #[doc(hidden)]
1533 #[unstable(feature = "sized_type_properties", issue = "none")]
1534 const MAX_SLICE_LEN: usize = match Self::SIZE {
1535 0 => usize::MAX,
1536 n => (isize::MAX as usize) / n,
1537 };
1538}
1539#[doc(hidden)]
1540#[unstable(feature = "sized_type_properties", issue = "none")]
1541impl<T> SizedTypeProperties for T {}
1542
1543/// Expands to the offset in bytes of a field from the beginning of the given type.
1544///
1545/// The type may be a `struct`, `enum`, `union`, or tuple.
1546///
1547/// The field may be a nested field (`field1.field2`), but not an array index.
1548/// The field must be visible to the call site.
1549///
1550/// The offset is returned as a [`usize`].
1551///
1552/// # Offsets of, and in, dynamically sized types
1553///
1554/// The field’s type must be [`Sized`], but it may be located in a [dynamically sized] container.
1555/// If the field type is dynamically sized, then you cannot use `offset_of!` (since the field's
1556/// alignment, and therefore its offset, may also be dynamic) and must take the offset from an
1557/// actual pointer to the container instead.
1558///
1559/// ```
1560/// # use core::mem;
1561/// # use core::fmt::Debug;
1562/// #[repr(C)]
1563/// pub struct Struct<T: ?Sized> {
1564/// a: u8,
1565/// b: T,
1566/// }
1567///
1568/// #[derive(Debug)]
1569/// #[repr(C, align(4))]
1570/// struct Align4(u32);
1571///
1572/// assert_eq!(mem::offset_of!(Struct<dyn Debug>, a), 0); // OK — Sized field
1573/// assert_eq!(mem::offset_of!(Struct<Align4>, b), 4); // OK — not DST
1574///
1575/// // assert_eq!(mem::offset_of!(Struct<dyn Debug>, b), 1);
1576/// // ^^^ error[E0277]: ... cannot be known at compilation time
1577///
1578/// // To obtain the offset of a !Sized field, examine a concrete value
1579/// // instead of using offset_of!.
1580/// let value: Struct<Align4> = Struct { a: 1, b: Align4(2) };
1581/// let ref_unsized: &Struct<dyn Debug> = &value;
1582/// let offset_of_b = unsafe {
1583/// (&raw const ref_unsized.b).byte_offset_from_unsigned(ref_unsized)
1584/// };
1585/// assert_eq!(offset_of_b, 4);
1586/// ```
1587///
1588/// If you need to obtain the offset of a field of a `!Sized` type, then, since the offset may
1589/// depend on the particular value being stored (in particular, `dyn Trait` values have a
1590/// dynamically-determined alignment), you must retrieve the offset from a specific reference
1591/// or pointer, and so you cannot use `offset_of!` to work without one.
1592///
1593/// # Layout is subject to change
1594///
1595/// Note that type layout is, in general, [subject to change and
1596/// platform-specific](https://doc.rust-lang.org/reference/type-layout.html). If
1597/// layout stability is required, consider using an [explicit `repr` attribute].
1598///
1599/// Rust guarantees that the offset of a given field within a given type will not
1600/// change over the lifetime of the program. However, two different compilations of
1601/// the same program may result in different layouts. Also, even within a single
1602/// program execution, no guarantees are made about types which are *similar* but
1603/// not *identical*, e.g.:
1604///
1605/// ```
1606/// struct Wrapper<T, U>(T, U);
1607///
1608/// type A = Wrapper<u8, u8>;
1609/// type B = Wrapper<u8, i8>;
1610///
1611/// // Not necessarily identical even though `u8` and `i8` have the same layout!
1612/// // assert_eq!(mem::offset_of!(A, 1), mem::offset_of!(B, 1));
1613///
1614/// #[repr(transparent)]
1615/// struct U8(u8);
1616///
1617/// type C = Wrapper<u8, U8>;
1618///
1619/// // Not necessarily identical even though `u8` and `U8` have the same layout!
1620/// // assert_eq!(mem::offset_of!(A, 1), mem::offset_of!(C, 1));
1621///
1622/// struct Empty<T>(core::marker::PhantomData<T>);
1623///
1624/// // Not necessarily identical even though `PhantomData` always has the same layout!
1625/// // assert_eq!(mem::offset_of!(Empty<u8>, 0), mem::offset_of!(Empty<i8>, 0));
1626/// ```
1627///
1628/// [explicit `repr` attribute]: https://doc.rust-lang.org/reference/type-layout.html#representations
1629///
1630/// # Unstable features
1631///
1632/// The following unstable features expand the functionality of `offset_of!`:
1633///
1634/// * [`offset_of_enum`] — allows `enum` variants to be traversed as if they were fields.
1635/// * [`offset_of_slice`] — allows getting the offset of a field of type `[T]`.
1636///
1637/// # Examples
1638///
1639/// ```
1640/// use std::mem;
1641/// #[repr(C)]
1642/// struct FieldStruct {
1643/// first: u8,
1644/// second: u16,
1645/// third: u8
1646/// }
1647///
1648/// assert_eq!(mem::offset_of!(FieldStruct, first), 0);
1649/// assert_eq!(mem::offset_of!(FieldStruct, second), 2);
1650/// assert_eq!(mem::offset_of!(FieldStruct, third), 4);
1651///
1652/// #[repr(C)]
1653/// struct NestedA {
1654/// b: NestedB
1655/// }
1656///
1657/// #[repr(C)]
1658/// struct NestedB(u8);
1659///
1660/// assert_eq!(mem::offset_of!(NestedA, b.0), 0);
1661/// ```
1662///
1663/// [dynamically sized]: https://doc.rust-lang.org/reference/dynamically-sized-types.html
1664/// [`offset_of_enum`]: https://doc.rust-lang.org/nightly/unstable-book/language-features/offset-of-enum.html
1665/// [`offset_of_slice`]: https://doc.rust-lang.org/nightly/unstable-book/language-features/offset-of-slice.html
1666#[stable(feature = "offset_of", since = "1.77.0")]
1667#[diagnostic::on_unmatched_args(
1668 note = "this macro expects a container type and a (nested) field path, like `offset_of!(Type, field)`"
1669)]
1670#[doc(alias = "memoffset")]
1671#[allow_internal_unstable(builtin_syntax, core_intrinsics)]
1672#[diagnostic::opaque]
1673pub macro offset_of($Container:ty, $($fields:expr)+ $(,)?) {
1674 const { builtin # offset_of($Container, $($fields)+) }
1675}
1676
1677/// Create a fresh instance of the inhabited ZST type `T`.
1678///
1679/// Prefer this to [`zeroed`] or [`uninitialized`] or [`transmute_copy`]
1680/// in places where you know that `T` is zero-sized, but don't have a bound
1681/// (such as [`Default`]) that would allow you to instantiate it using safe code.
1682///
1683/// If you're not sure whether `T` is an inhabited ZST, then you should be
1684/// using [`MaybeUninit`], not this function.
1685///
1686/// # Panics
1687///
1688/// If `size_of::<T>() != 0`.
1689///
1690/// # Safety
1691///
1692/// - `T` must be *[inhabited]*, i.e. possible to construct. This means that types
1693/// like zero-variant enums and [`!`] are unsound to conjure.
1694/// - You must use the value only in ways which do not violate any *safety*
1695/// invariants of the type.
1696///
1697/// While it's easy to create a *valid* instance of an inhabited ZST, since having
1698/// no bits in its representation means there's only one possible value, that
1699/// doesn't mean that it's always *sound* to do so.
1700///
1701/// For example, a library could design zero-sized tokens that are `!Default + !Clone`, limiting
1702/// their creation to functions that initialize some state or establish a scope. Conjuring such a
1703/// token could break invariants and lead to unsoundness.
1704///
1705/// # Examples
1706///
1707/// ```
1708/// #![feature(mem_conjure_zst)]
1709/// use std::mem::conjure_zst;
1710///
1711/// assert_eq!(unsafe { conjure_zst::<()>() }, ());
1712/// assert_eq!(unsafe { conjure_zst::<[i32; 0]>() }, []);
1713/// ```
1714///
1715/// [inhabited]: https://doc.rust-lang.org/reference/glossary.html#inhabited
1716#[unstable(feature = "mem_conjure_zst", issue = "95383")]
1717#[rustc_const_unstable(feature = "mem_conjure_zst", issue = "95383")]
1718#[ferrocene::prevalidated]
1719pub const unsafe fn conjure_zst<T>() -> T {
1720 #[ferrocene::annotation(
1721 "This assertion only runs in compilation, meaning that it cannot be covered in runtime"
1722 )]
1723 // Ferrocene addition: add curly braces in order to apply annotation to whole const_assert.
1724 {
1725 const_assert!(
1726 T::IS_ZST,
1727 "mem::conjure_zst invoked on a non-zero-sized type",
1728 "mem::conjure_zst invoked on type {name}, which is not zero-sized",
1729 name: &str = crate::any::type_name::<T>()
1730 );
1731 }
1732
1733 // SAFETY: because the caller must guarantee that it's inhabited and zero-sized,
1734 // there's nothing in the representation that needs to be set.
1735 // `assume_init` calls `assert_inhabited`, so we don't need to here.
1736 unsafe {
1737 #[allow(clippy::uninit_assumed_init)]
1738 MaybeUninit::uninit().assume_init()
1739 }
1740}