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