]> git.lizzy.rs Git - rust.git/blob - library/core/src/mem/maybe_uninit.rs
Auto merge of #75137 - Aaron1011:fix/hygiene-skip-expndata, r=petrochenkov
[rust.git] / library / core / src / mem / maybe_uninit.rs
1 use crate::any::type_name;
2 use crate::fmt;
3 use crate::intrinsics;
4 use crate::mem::ManuallyDrop;
5
6 // ignore-tidy-undocumented-unsafe
7
8 /// A wrapper type to construct uninitialized instances of `T`.
9 ///
10 /// # Initialization invariant
11 ///
12 /// The compiler, in general, assumes that a variable is properly initialized
13 /// according to the requirements of the variable's type. For example, a variable of
14 /// reference type must be aligned and non-NULL. This is an invariant that must
15 /// *always* be upheld, even in unsafe code. As a consequence, zero-initializing a
16 /// variable of reference type causes instantaneous [undefined behavior][ub],
17 /// no matter whether that reference ever gets used to access memory:
18 ///
19 /// ```rust,no_run
20 /// # #![allow(invalid_value)]
21 /// use std::mem::{self, MaybeUninit};
22 ///
23 /// let x: &i32 = unsafe { mem::zeroed() }; // undefined behavior! ⚠️
24 /// // The equivalent code with `MaybeUninit<&i32>`:
25 /// let x: &i32 = unsafe { MaybeUninit::zeroed().assume_init() }; // undefined behavior! ⚠️
26 /// ```
27 ///
28 /// This is exploited by the compiler for various optimizations, such as eliding
29 /// run-time checks and optimizing `enum` layout.
30 ///
31 /// Similarly, entirely uninitialized memory may have any content, while a `bool` must
32 /// always be `true` or `false`. Hence, creating an uninitialized `bool` is undefined behavior:
33 ///
34 /// ```rust,no_run
35 /// # #![allow(invalid_value)]
36 /// use std::mem::{self, MaybeUninit};
37 ///
38 /// let b: bool = unsafe { mem::uninitialized() }; // undefined behavior! ⚠️
39 /// // The equivalent code with `MaybeUninit<bool>`:
40 /// let b: bool = unsafe { MaybeUninit::uninit().assume_init() }; // undefined behavior! ⚠️
41 /// ```
42 ///
43 /// Moreover, uninitialized memory is special in that the compiler knows that
44 /// it does not have a fixed value. This makes it undefined behavior to have
45 /// uninitialized data in a variable even if that variable has an integer type,
46 /// which otherwise can hold any *fixed* bit pattern:
47 ///
48 /// ```rust,no_run
49 /// # #![allow(invalid_value)]
50 /// use std::mem::{self, MaybeUninit};
51 ///
52 /// let x: i32 = unsafe { mem::uninitialized() }; // undefined behavior! ⚠️
53 /// // The equivalent code with `MaybeUninit<i32>`:
54 /// let x: i32 = unsafe { MaybeUninit::uninit().assume_init() }; // undefined behavior! ⚠️
55 /// ```
56 /// (Notice that the rules around uninitialized integers are not finalized yet, but
57 /// until they are, it is advisable to avoid them.)
58 ///
59 /// On top of that, remember that most types have additional invariants beyond merely
60 /// being considered initialized at the type level. For example, a `1`-initialized [`Vec<T>`]
61 /// is considered initialized (under the current implementation; this does not constitute
62 /// a stable guarantee) because the only requirement the compiler knows about it
63 /// is that the data pointer must be non-null. Creating such a `Vec<T>` does not cause
64 /// *immediate* undefined behavior, but will cause undefined behavior with most
65 /// safe operations (including dropping it).
66 ///
67 /// [`Vec<T>`]: ../../std/vec/struct.Vec.html
68 ///
69 /// # Examples
70 ///
71 /// `MaybeUninit<T>` serves to enable unsafe code to deal with uninitialized data.
72 /// It is a signal to the compiler indicating that the data here might *not*
73 /// be initialized:
74 ///
75 /// ```rust
76 /// use std::mem::MaybeUninit;
77 ///
78 /// // Create an explicitly uninitialized reference. The compiler knows that data inside
79 /// // a `MaybeUninit<T>` may be invalid, and hence this is not UB:
80 /// let mut x = MaybeUninit::<&i32>::uninit();
81 /// // Set it to a valid value.
82 /// unsafe { x.as_mut_ptr().write(&0); }
83 /// // Extract the initialized data -- this is only allowed *after* properly
84 /// // initializing `x`!
85 /// let x = unsafe { x.assume_init() };
86 /// ```
87 ///
88 /// The compiler then knows to not make any incorrect assumptions or optimizations on this code.
89 ///
90 /// You can think of `MaybeUninit<T>` as being a bit like `Option<T>` but without
91 /// any of the run-time tracking and without any of the safety checks.
92 ///
93 /// ## out-pointers
94 ///
95 /// You can use `MaybeUninit<T>` to implement "out-pointers": instead of returning data
96 /// from a function, pass it a pointer to some (uninitialized) memory to put the
97 /// result into. This can be useful when it is important for the caller to control
98 /// how the memory the result is stored in gets allocated, and you want to avoid
99 /// unnecessary moves.
100 ///
101 /// ```
102 /// use std::mem::MaybeUninit;
103 ///
104 /// unsafe fn make_vec(out: *mut Vec<i32>) {
105 ///     // `write` does not drop the old contents, which is important.
106 ///     out.write(vec![1, 2, 3]);
107 /// }
108 ///
109 /// let mut v = MaybeUninit::uninit();
110 /// unsafe { make_vec(v.as_mut_ptr()); }
111 /// // Now we know `v` is initialized! This also makes sure the vector gets
112 /// // properly dropped.
113 /// let v = unsafe { v.assume_init() };
114 /// assert_eq!(&v, &[1, 2, 3]);
115 /// ```
116 ///
117 /// ## Initializing an array element-by-element
118 ///
119 /// `MaybeUninit<T>` can be used to initialize a large array element-by-element:
120 ///
121 /// ```
122 /// use std::mem::{self, MaybeUninit};
123 ///
124 /// let data = {
125 ///     // Create an uninitialized array of `MaybeUninit`. The `assume_init` is
126 ///     // safe because the type we are claiming to have initialized here is a
127 ///     // bunch of `MaybeUninit`s, which do not require initialization.
128 ///     let mut data: [MaybeUninit<Vec<u32>>; 1000] = unsafe {
129 ///         MaybeUninit::uninit().assume_init()
130 ///     };
131 ///
132 ///     // Dropping a `MaybeUninit` does nothing. Thus using raw pointer
133 ///     // assignment instead of `ptr::write` does not cause the old
134 ///     // uninitialized value to be dropped. Also if there is a panic during
135 ///     // this loop, we have a memory leak, but there is no memory safety
136 ///     // issue.
137 ///     for elem in &mut data[..] {
138 ///         *elem = MaybeUninit::new(vec![42]);
139 ///     }
140 ///
141 ///     // Everything is initialized. Transmute the array to the
142 ///     // initialized type.
143 ///     unsafe { mem::transmute::<_, [Vec<u32>; 1000]>(data) }
144 /// };
145 ///
146 /// assert_eq!(&data[0], &[42]);
147 /// ```
148 ///
149 /// You can also work with partially initialized arrays, which could
150 /// be found in low-level datastructures.
151 ///
152 /// ```
153 /// use std::mem::MaybeUninit;
154 /// use std::ptr;
155 ///
156 /// // Create an uninitialized array of `MaybeUninit`. The `assume_init` is
157 /// // safe because the type we are claiming to have initialized here is a
158 /// // bunch of `MaybeUninit`s, which do not require initialization.
159 /// let mut data: [MaybeUninit<String>; 1000] = unsafe { MaybeUninit::uninit().assume_init() };
160 /// // Count the number of elements we have assigned.
161 /// let mut data_len: usize = 0;
162 ///
163 /// for elem in &mut data[0..500] {
164 ///     *elem = MaybeUninit::new(String::from("hello"));
165 ///     data_len += 1;
166 /// }
167 ///
168 /// // For each item in the array, drop if we allocated it.
169 /// for elem in &mut data[0..data_len] {
170 ///     unsafe { ptr::drop_in_place(elem.as_mut_ptr()); }
171 /// }
172 /// ```
173 ///
174 /// ## Initializing a struct field-by-field
175 ///
176 /// There is currently no supported way to create a raw pointer or reference
177 /// to a field of a struct inside `MaybeUninit<Struct>`. That means it is not possible
178 /// to create a struct by calling `MaybeUninit::uninit::<Struct>()` and then writing
179 /// to its fields.
180 ///
181 /// [ub]: ../../reference/behavior-considered-undefined.html
182 ///
183 /// # Layout
184 ///
185 /// `MaybeUninit<T>` is guaranteed to have the same size, alignment, and ABI as `T`:
186 ///
187 /// ```rust
188 /// use std::mem::{MaybeUninit, size_of, align_of};
189 /// assert_eq!(size_of::<MaybeUninit<u64>>(), size_of::<u64>());
190 /// assert_eq!(align_of::<MaybeUninit<u64>>(), align_of::<u64>());
191 /// ```
192 ///
193 /// However remember that a type *containing* a `MaybeUninit<T>` is not necessarily the same
194 /// layout; Rust does not in general guarantee that the fields of a `Foo<T>` have the same order as
195 /// a `Foo<U>` even if `T` and `U` have the same size and alignment. Furthermore because any bit
196 /// value is valid for a `MaybeUninit<T>` the compiler can't apply non-zero/niche-filling
197 /// optimizations, potentially resulting in a larger size:
198 ///
199 /// ```rust
200 /// # use std::mem::{MaybeUninit, size_of};
201 /// assert_eq!(size_of::<Option<bool>>(), 1);
202 /// assert_eq!(size_of::<Option<MaybeUninit<bool>>>(), 2);
203 /// ```
204 ///
205 /// If `T` is FFI-safe, then so is `MaybeUninit<T>`.
206 ///
207 /// While `MaybeUninit` is `#[repr(transparent)]` (indicating it guarantees the same size,
208 /// alignment, and ABI as `T`), this does *not* change any of the previous caveats. `Option<T>` and
209 /// `Option<MaybeUninit<T>>` may still have different sizes, and types containing a field of type
210 /// `T` may be laid out (and sized) differently than if that field were `MaybeUninit<T>`.
211 /// `MaybeUninit` is a union type, and `#[repr(transparent)]` on unions is unstable (see [the
212 /// tracking issue](https://github.com/rust-lang/rust/issues/60405)). Over time, the exact
213 /// guarantees of `#[repr(transparent)]` on unions may evolve, and `MaybeUninit` may or may not
214 /// remain `#[repr(transparent)]`. That said, `MaybeUninit<T>` will *always* guarantee that it has
215 /// the same size, alignment, and ABI as `T`; it's just that the way `MaybeUninit` implements that
216 /// guarantee may evolve.
217 #[stable(feature = "maybe_uninit", since = "1.36.0")]
218 // Lang item so we can wrap other types in it. This is useful for generators.
219 #[lang = "maybe_uninit"]
220 #[derive(Copy)]
221 #[repr(transparent)]
222 pub union MaybeUninit<T> {
223     uninit: (),
224     value: ManuallyDrop<T>,
225 }
226
227 #[stable(feature = "maybe_uninit", since = "1.36.0")]
228 impl<T: Copy> Clone for MaybeUninit<T> {
229     #[inline(always)]
230     fn clone(&self) -> Self {
231         // Not calling `T::clone()`, we cannot know if we are initialized enough for that.
232         *self
233     }
234 }
235
236 #[stable(feature = "maybe_uninit_debug", since = "1.41.0")]
237 impl<T> fmt::Debug for MaybeUninit<T> {
238     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
239         f.pad(type_name::<Self>())
240     }
241 }
242
243 impl<T> MaybeUninit<T> {
244     /// Creates a new `MaybeUninit<T>` initialized with the given value.
245     /// It is safe to call [`assume_init`] on the return value of this function.
246     ///
247     /// Note that dropping a `MaybeUninit<T>` will never call `T`'s drop code.
248     /// It is your responsibility to make sure `T` gets dropped if it got initialized.
249     ///
250     /// [`assume_init`]: #method.assume_init
251     #[stable(feature = "maybe_uninit", since = "1.36.0")]
252     #[rustc_const_stable(feature = "const_maybe_uninit", since = "1.36.0")]
253     #[inline(always)]
254     pub const fn new(val: T) -> MaybeUninit<T> {
255         MaybeUninit { value: ManuallyDrop::new(val) }
256     }
257
258     /// Creates a new `MaybeUninit<T>` in an uninitialized state.
259     ///
260     /// Note that dropping a `MaybeUninit<T>` will never call `T`'s drop code.
261     /// It is your responsibility to make sure `T` gets dropped if it got initialized.
262     ///
263     /// See the [type-level documentation][type] for some examples.
264     ///
265     /// [type]: union.MaybeUninit.html
266     #[stable(feature = "maybe_uninit", since = "1.36.0")]
267     #[rustc_const_stable(feature = "const_maybe_uninit", since = "1.36.0")]
268     #[inline(always)]
269     #[rustc_diagnostic_item = "maybe_uninit_uninit"]
270     pub const fn uninit() -> MaybeUninit<T> {
271         MaybeUninit { uninit: () }
272     }
273
274     /// Create a new array of `MaybeUninit<T>` items, in an uninitialized state.
275     ///
276     /// Note: in a future Rust version this method may become unnecessary
277     /// when array literal syntax allows
278     /// [repeating const expressions](https://github.com/rust-lang/rust/issues/49147).
279     /// The example below could then use `let mut buf = [MaybeUninit::<u8>::uninit(); 32];`.
280     ///
281     /// # Examples
282     ///
283     /// ```no_run
284     /// #![feature(maybe_uninit_uninit_array, maybe_uninit_extra, maybe_uninit_slice_assume_init)]
285     ///
286     /// use std::mem::MaybeUninit;
287     ///
288     /// extern "C" {
289     ///     fn read_into_buffer(ptr: *mut u8, max_len: usize) -> usize;
290     /// }
291     ///
292     /// /// Returns a (possibly smaller) slice of data that was actually read
293     /// fn read(buf: &mut [MaybeUninit<u8>]) -> &[u8] {
294     ///     unsafe {
295     ///         let len = read_into_buffer(buf.as_mut_ptr() as *mut u8, buf.len());
296     ///         MaybeUninit::slice_get_ref(&buf[..len])
297     ///     }
298     /// }
299     ///
300     /// let mut buf: [MaybeUninit<u8>; 32] = MaybeUninit::uninit_array();
301     /// let data = read(&mut buf);
302     /// ```
303     #[unstable(feature = "maybe_uninit_uninit_array", issue = "none")]
304     #[inline(always)]
305     pub fn uninit_array<const LEN: usize>() -> [Self; LEN] {
306         unsafe { MaybeUninit::<[MaybeUninit<T>; LEN]>::uninit().assume_init() }
307     }
308
309     /// A promotable constant, equivalent to `uninit()`.
310     #[unstable(
311         feature = "internal_uninit_const",
312         issue = "none",
313         reason = "hack to work around promotability"
314     )]
315     pub const UNINIT: Self = Self::uninit();
316
317     /// Creates a new `MaybeUninit<T>` in an uninitialized state, with the memory being
318     /// filled with `0` bytes. It depends on `T` whether that already makes for
319     /// proper initialization. For example, `MaybeUninit<usize>::zeroed()` is initialized,
320     /// but `MaybeUninit<&'static i32>::zeroed()` is not because references must not
321     /// be null.
322     ///
323     /// Note that dropping a `MaybeUninit<T>` will never call `T`'s drop code.
324     /// It is your responsibility to make sure `T` gets dropped if it got initialized.
325     ///
326     /// # Example
327     ///
328     /// Correct usage of this function: initializing a struct with zero, where all
329     /// fields of the struct can hold the bit-pattern 0 as a valid value.
330     ///
331     /// ```rust
332     /// use std::mem::MaybeUninit;
333     ///
334     /// let x = MaybeUninit::<(u8, bool)>::zeroed();
335     /// let x = unsafe { x.assume_init() };
336     /// assert_eq!(x, (0, false));
337     /// ```
338     ///
339     /// *Incorrect* usage of this function: calling `x.zeroed().assume_init()`
340     /// when `0` is not a valid bit-pattern for the type:
341     ///
342     /// ```rust,no_run
343     /// use std::mem::MaybeUninit;
344     ///
345     /// enum NotZero { One = 1, Two = 2 };
346     ///
347     /// let x = MaybeUninit::<(u8, NotZero)>::zeroed();
348     /// let x = unsafe { x.assume_init() };
349     /// // Inside a pair, we create a `NotZero` that does not have a valid discriminant.
350     /// // This is undefined behavior. ⚠️
351     /// ```
352     #[stable(feature = "maybe_uninit", since = "1.36.0")]
353     #[inline]
354     #[rustc_diagnostic_item = "maybe_uninit_zeroed"]
355     pub fn zeroed() -> MaybeUninit<T> {
356         let mut u = MaybeUninit::<T>::uninit();
357         unsafe {
358             u.as_mut_ptr().write_bytes(0u8, 1);
359         }
360         u
361     }
362
363     /// Sets the value of the `MaybeUninit<T>`. This overwrites any previous value
364     /// without dropping it, so be careful not to use this twice unless you want to
365     /// skip running the destructor. For your convenience, this also returns a mutable
366     /// reference to the (now safely initialized) contents of `self`.
367     #[unstable(feature = "maybe_uninit_extra", issue = "63567")]
368     #[inline(always)]
369     pub fn write(&mut self, val: T) -> &mut T {
370         unsafe {
371             self.value = ManuallyDrop::new(val);
372             self.get_mut()
373         }
374     }
375
376     /// Gets a pointer to the contained value. Reading from this pointer or turning it
377     /// into a reference is undefined behavior unless the `MaybeUninit<T>` is initialized.
378     /// Writing to memory that this pointer (non-transitively) points to is undefined behavior
379     /// (except inside an `UnsafeCell<T>`).
380     ///
381     /// # Examples
382     ///
383     /// Correct usage of this method:
384     ///
385     /// ```rust
386     /// use std::mem::MaybeUninit;
387     ///
388     /// let mut x = MaybeUninit::<Vec<u32>>::uninit();
389     /// unsafe { x.as_mut_ptr().write(vec![0,1,2]); }
390     /// // Create a reference into the `MaybeUninit<T>`. This is okay because we initialized it.
391     /// let x_vec = unsafe { &*x.as_ptr() };
392     /// assert_eq!(x_vec.len(), 3);
393     /// ```
394     ///
395     /// *Incorrect* usage of this method:
396     ///
397     /// ```rust,no_run
398     /// use std::mem::MaybeUninit;
399     ///
400     /// let x = MaybeUninit::<Vec<u32>>::uninit();
401     /// let x_vec = unsafe { &*x.as_ptr() };
402     /// // We have created a reference to an uninitialized vector! This is undefined behavior. ⚠️
403     /// ```
404     ///
405     /// (Notice that the rules around references to uninitialized data are not finalized yet, but
406     /// until they are, it is advisable to avoid them.)
407     #[stable(feature = "maybe_uninit", since = "1.36.0")]
408     #[rustc_const_unstable(feature = "const_maybe_uninit_as_ptr", issue = "75251")]
409     #[inline(always)]
410     pub const fn as_ptr(&self) -> *const T {
411         // `MaybeUninit` and `ManuallyDrop` are both `repr(transparent)` so we can cast the pointer.
412         self as *const _ as *const T
413     }
414
415     /// Gets a mutable pointer to the contained value. Reading from this pointer or turning it
416     /// into a reference is undefined behavior unless the `MaybeUninit<T>` is initialized.
417     ///
418     /// # Examples
419     ///
420     /// Correct usage of this method:
421     ///
422     /// ```rust
423     /// use std::mem::MaybeUninit;
424     ///
425     /// let mut x = MaybeUninit::<Vec<u32>>::uninit();
426     /// unsafe { x.as_mut_ptr().write(vec![0,1,2]); }
427     /// // Create a reference into the `MaybeUninit<Vec<u32>>`.
428     /// // This is okay because we initialized it.
429     /// let x_vec = unsafe { &mut *x.as_mut_ptr() };
430     /// x_vec.push(3);
431     /// assert_eq!(x_vec.len(), 4);
432     /// ```
433     ///
434     /// *Incorrect* usage of this method:
435     ///
436     /// ```rust,no_run
437     /// use std::mem::MaybeUninit;
438     ///
439     /// let mut x = MaybeUninit::<Vec<u32>>::uninit();
440     /// let x_vec = unsafe { &mut *x.as_mut_ptr() };
441     /// // We have created a reference to an uninitialized vector! This is undefined behavior. ⚠️
442     /// ```
443     ///
444     /// (Notice that the rules around references to uninitialized data are not finalized yet, but
445     /// until they are, it is advisable to avoid them.)
446     #[stable(feature = "maybe_uninit", since = "1.36.0")]
447     #[rustc_const_unstable(feature = "const_maybe_uninit_as_ptr", issue = "75251")]
448     #[inline(always)]
449     pub const fn as_mut_ptr(&mut self) -> *mut T {
450         // `MaybeUninit` and `ManuallyDrop` are both `repr(transparent)` so we can cast the pointer.
451         self as *mut _ as *mut T
452     }
453
454     /// Extracts the value from the `MaybeUninit<T>` container. This is a great way
455     /// to ensure that the data will get dropped, because the resulting `T` is
456     /// subject to the usual drop handling.
457     ///
458     /// # Safety
459     ///
460     /// It is up to the caller to guarantee that the `MaybeUninit<T>` really is in an initialized
461     /// state. Calling this when the content is not yet fully initialized causes immediate undefined
462     /// behavior. The [type-level documentation][inv] contains more information about
463     /// this initialization invariant.
464     ///
465     /// [inv]: #initialization-invariant
466     ///
467     /// On top of that, remember that most types have additional invariants beyond merely
468     /// being considered initialized at the type level. For example, a `1`-initialized [`Vec<T>`]
469     /// is considered initialized (under the current implementation; this does not constitute
470     /// a stable guarantee) because the only requirement the compiler knows about it
471     /// is that the data pointer must be non-null. Creating such a `Vec<T>` does not cause
472     /// *immediate* undefined behavior, but will cause undefined behavior with most
473     /// safe operations (including dropping it).
474     ///
475     /// # Examples
476     ///
477     /// Correct usage of this method:
478     ///
479     /// ```rust
480     /// use std::mem::MaybeUninit;
481     ///
482     /// let mut x = MaybeUninit::<bool>::uninit();
483     /// unsafe { x.as_mut_ptr().write(true); }
484     /// let x_init = unsafe { x.assume_init() };
485     /// assert_eq!(x_init, true);
486     /// ```
487     ///
488     /// *Incorrect* usage of this method:
489     ///
490     /// ```rust,no_run
491     /// use std::mem::MaybeUninit;
492     ///
493     /// let x = MaybeUninit::<Vec<u32>>::uninit();
494     /// let x_init = unsafe { x.assume_init() };
495     /// // `x` had not been initialized yet, so this last line caused undefined behavior. ⚠️
496     /// ```
497     #[stable(feature = "maybe_uninit", since = "1.36.0")]
498     #[inline(always)]
499     #[rustc_diagnostic_item = "assume_init"]
500     pub unsafe fn assume_init(self) -> T {
501         // SAFETY: the caller must guarantee that `self` is initialized.
502         // This also means that `self` must be a `value` variant.
503         unsafe {
504             intrinsics::assert_inhabited::<T>();
505             ManuallyDrop::into_inner(self.value)
506         }
507     }
508
509     /// Reads the value from the `MaybeUninit<T>` container. The resulting `T` is subject
510     /// to the usual drop handling.
511     ///
512     /// Whenever possible, it is preferable to use [`assume_init`] instead, which
513     /// prevents duplicating the content of the `MaybeUninit<T>`.
514     ///
515     /// # Safety
516     ///
517     /// It is up to the caller to guarantee that the `MaybeUninit<T>` really is in an initialized
518     /// state. Calling this when the content is not yet fully initialized causes undefined
519     /// behavior. The [type-level documentation][inv] contains more information about
520     /// this initialization invariant.
521     ///
522     /// Moreover, this leaves a copy of the same data behind in the `MaybeUninit<T>`. When using
523     /// multiple copies of the data (by calling `read` multiple times, or first
524     /// calling `read` and then [`assume_init`]), it is your responsibility
525     /// to ensure that that data may indeed be duplicated.
526     ///
527     /// [inv]: #initialization-invariant
528     /// [`assume_init`]: #method.assume_init
529     ///
530     /// # Examples
531     ///
532     /// Correct usage of this method:
533     ///
534     /// ```rust
535     /// #![feature(maybe_uninit_extra)]
536     /// use std::mem::MaybeUninit;
537     ///
538     /// let mut x = MaybeUninit::<u32>::uninit();
539     /// x.write(13);
540     /// let x1 = unsafe { x.read() };
541     /// // `u32` is `Copy`, so we may read multiple times.
542     /// let x2 = unsafe { x.read() };
543     /// assert_eq!(x1, x2);
544     ///
545     /// let mut x = MaybeUninit::<Option<Vec<u32>>>::uninit();
546     /// x.write(None);
547     /// let x1 = unsafe { x.read() };
548     /// // Duplicating a `None` value is okay, so we may read multiple times.
549     /// let x2 = unsafe { x.read() };
550     /// assert_eq!(x1, x2);
551     /// ```
552     ///
553     /// *Incorrect* usage of this method:
554     ///
555     /// ```rust,no_run
556     /// #![feature(maybe_uninit_extra)]
557     /// use std::mem::MaybeUninit;
558     ///
559     /// let mut x = MaybeUninit::<Option<Vec<u32>>>::uninit();
560     /// x.write(Some(vec![0,1,2]));
561     /// let x1 = unsafe { x.read() };
562     /// let x2 = unsafe { x.read() };
563     /// // We now created two copies of the same vector, leading to a double-free ⚠️ when
564     /// // they both get dropped!
565     /// ```
566     #[unstable(feature = "maybe_uninit_extra", issue = "63567")]
567     #[inline(always)]
568     pub unsafe fn read(&self) -> T {
569         // SAFETY: the caller must guarantee that `self` is initialized.
570         // Reading from `self.as_ptr()` is safe since `self` should be initialized.
571         unsafe {
572             intrinsics::assert_inhabited::<T>();
573             self.as_ptr().read()
574         }
575     }
576
577     /// Gets a shared reference to the contained value.
578     ///
579     /// This can be useful when we want to access a `MaybeUninit` that has been
580     /// initialized but don't have ownership of the `MaybeUninit` (preventing the use
581     /// of `.assume_init()`).
582     ///
583     /// # Safety
584     ///
585     /// Calling this when the content is not yet fully initialized causes undefined
586     /// behavior: it is up to the caller to guarantee that the `MaybeUninit<T>` really
587     /// is in an initialized state.
588     ///
589     /// # Examples
590     ///
591     /// ### Correct usage of this method:
592     ///
593     /// ```rust
594     /// #![feature(maybe_uninit_ref)]
595     /// use std::mem::MaybeUninit;
596     ///
597     /// let mut x = MaybeUninit::<Vec<u32>>::uninit();
598     /// // Initialize `x`:
599     /// unsafe { x.as_mut_ptr().write(vec![1, 2, 3]); }
600     /// // Now that our `MaybeUninit<_>` is known to be initialized, it is okay to
601     /// // create a shared reference to it:
602     /// let x: &Vec<u32> = unsafe {
603     ///     // Safety: `x` has been initialized.
604     ///     x.get_ref()
605     /// };
606     /// assert_eq!(x, &vec![1, 2, 3]);
607     /// ```
608     ///
609     /// ### *Incorrect* usages of this method:
610     ///
611     /// ```rust,no_run
612     /// #![feature(maybe_uninit_ref)]
613     /// use std::mem::MaybeUninit;
614     ///
615     /// let x = MaybeUninit::<Vec<u32>>::uninit();
616     /// let x_vec: &Vec<u32> = unsafe { x.get_ref() };
617     /// // We have created a reference to an uninitialized vector! This is undefined behavior. ⚠️
618     /// ```
619     ///
620     /// ```rust,no_run
621     /// #![feature(maybe_uninit_ref)]
622     /// use std::{cell::Cell, mem::MaybeUninit};
623     ///
624     /// let b = MaybeUninit::<Cell<bool>>::uninit();
625     /// // Initialize the `MaybeUninit` using `Cell::set`:
626     /// unsafe {
627     ///     b.get_ref().set(true);
628     ///  // ^^^^^^^^^^^
629     ///  // Reference to an uninitialized `Cell<bool>`: UB!
630     /// }
631     /// ```
632     #[unstable(feature = "maybe_uninit_ref", issue = "63568")]
633     #[inline(always)]
634     pub unsafe fn get_ref(&self) -> &T {
635         // SAFETY: the caller must guarantee that `self` is initialized.
636         // This also means that `self` must be a `value` variant.
637         unsafe {
638             intrinsics::assert_inhabited::<T>();
639             &*self.value
640         }
641     }
642
643     /// Gets a mutable (unique) reference to the contained value.
644     ///
645     /// This can be useful when we want to access a `MaybeUninit` that has been
646     /// initialized but don't have ownership of the `MaybeUninit` (preventing the use
647     /// of `.assume_init()`).
648     ///
649     /// # Safety
650     ///
651     /// Calling this when the content is not yet fully initialized causes undefined
652     /// behavior: it is up to the caller to guarantee that the `MaybeUninit<T>` really
653     /// is in an initialized state. For instance, `.get_mut()` cannot be used to
654     /// initialize a `MaybeUninit`.
655     ///
656     /// # Examples
657     ///
658     /// ### Correct usage of this method:
659     ///
660     /// ```rust
661     /// #![feature(maybe_uninit_ref)]
662     /// use std::mem::MaybeUninit;
663     ///
664     /// # unsafe extern "C" fn initialize_buffer(buf: *mut [u8; 2048]) { *buf = [0; 2048] }
665     /// # #[cfg(FALSE)]
666     /// extern "C" {
667     ///     /// Initializes *all* the bytes of the input buffer.
668     ///     fn initialize_buffer(buf: *mut [u8; 2048]);
669     /// }
670     ///
671     /// let mut buf = MaybeUninit::<[u8; 2048]>::uninit();
672     ///
673     /// // Initialize `buf`:
674     /// unsafe { initialize_buffer(buf.as_mut_ptr()); }
675     /// // Now we know that `buf` has been initialized, so we could `.assume_init()` it.
676     /// // However, using `.assume_init()` may trigger a `memcpy` of the 2048 bytes.
677     /// // To assert our buffer has been initialized without copying it, we upgrade
678     /// // the `&mut MaybeUninit<[u8; 2048]>` to a `&mut [u8; 2048]`:
679     /// let buf: &mut [u8; 2048] = unsafe {
680     ///     // Safety: `buf` has been initialized.
681     ///     buf.get_mut()
682     /// };
683     ///
684     /// // Now we can use `buf` as a normal slice:
685     /// buf.sort_unstable();
686     /// assert!(
687     ///     buf.windows(2).all(|pair| pair[0] <= pair[1]),
688     ///     "buffer is sorted",
689     /// );
690     /// ```
691     ///
692     /// ### *Incorrect* usages of this method:
693     ///
694     /// You cannot use `.get_mut()` to initialize a value:
695     ///
696     /// ```rust,no_run
697     /// #![feature(maybe_uninit_ref)]
698     /// use std::mem::MaybeUninit;
699     ///
700     /// let mut b = MaybeUninit::<bool>::uninit();
701     /// unsafe {
702     ///     *b.get_mut() = true;
703     ///     // We have created a (mutable) reference to an uninitialized `bool`!
704     ///     // This is undefined behavior. ⚠️
705     /// }
706     /// ```
707     ///
708     /// For instance, you cannot [`Read`] into an uninitialized buffer:
709     ///
710     /// [`Read`]: https://doc.rust-lang.org/std/io/trait.Read.html
711     ///
712     /// ```rust,no_run
713     /// #![feature(maybe_uninit_ref)]
714     /// use std::{io, mem::MaybeUninit};
715     ///
716     /// fn read_chunk (reader: &'_ mut dyn io::Read) -> io::Result<[u8; 64]>
717     /// {
718     ///     let mut buffer = MaybeUninit::<[u8; 64]>::uninit();
719     ///     reader.read_exact(unsafe { buffer.get_mut() })?;
720     ///                             // ^^^^^^^^^^^^^^^^
721     ///                             // (mutable) reference to uninitialized memory!
722     ///                             // This is undefined behavior.
723     ///     Ok(unsafe { buffer.assume_init() })
724     /// }
725     /// ```
726     ///
727     /// Nor can you use direct field access to do field-by-field gradual initialization:
728     ///
729     /// ```rust,no_run
730     /// #![feature(maybe_uninit_ref)]
731     /// use std::{mem::MaybeUninit, ptr};
732     ///
733     /// struct Foo {
734     ///     a: u32,
735     ///     b: u8,
736     /// }
737     ///
738     /// let foo: Foo = unsafe {
739     ///     let mut foo = MaybeUninit::<Foo>::uninit();
740     ///     ptr::write(&mut foo.get_mut().a as *mut u32, 1337);
741     ///                  // ^^^^^^^^^^^^^
742     ///                  // (mutable) reference to uninitialized memory!
743     ///                  // This is undefined behavior.
744     ///     ptr::write(&mut foo.get_mut().b as *mut u8, 42);
745     ///                  // ^^^^^^^^^^^^^
746     ///                  // (mutable) reference to uninitialized memory!
747     ///                  // This is undefined behavior.
748     ///     foo.assume_init()
749     /// };
750     /// ```
751     // FIXME(#53491): We currently rely on the above being incorrect, i.e., we have references
752     // to uninitialized data (e.g., in `libcore/fmt/float.rs`).  We should make
753     // a final decision about the rules before stabilization.
754     #[unstable(feature = "maybe_uninit_ref", issue = "63568")]
755     #[inline(always)]
756     pub unsafe fn get_mut(&mut self) -> &mut T {
757         // SAFETY: the caller must guarantee that `self` is initialized.
758         // This also means that `self` must be a `value` variant.
759         unsafe {
760             intrinsics::assert_inhabited::<T>();
761             &mut *self.value
762         }
763     }
764
765     /// Assuming all the elements are initialized, get a slice to them.
766     ///
767     /// # Safety
768     ///
769     /// It is up to the caller to guarantee that the `MaybeUninit<T>` elements
770     /// really are in an initialized state.
771     /// Calling this when the content is not yet fully initialized causes undefined behavior.
772     #[unstable(feature = "maybe_uninit_slice_assume_init", issue = "none")]
773     #[inline(always)]
774     pub unsafe fn slice_get_ref(slice: &[Self]) -> &[T] {
775         // SAFETY: casting slice to a `*const [T]` is safe since the caller guarantees that
776         // `slice` is initialized, and`MaybeUninit` is guaranteed to have the same layout as `T`.
777         // The pointer obtained is valid since it refers to memory owned by `slice` which is a
778         // reference and thus guaranteed to be valid for reads.
779         unsafe { &*(slice as *const [Self] as *const [T]) }
780     }
781
782     /// Assuming all the elements are initialized, get a mutable slice to them.
783     ///
784     /// # Safety
785     ///
786     /// It is up to the caller to guarantee that the `MaybeUninit<T>` elements
787     /// really are in an initialized state.
788     /// Calling this when the content is not yet fully initialized causes undefined behavior.
789     #[unstable(feature = "maybe_uninit_slice_assume_init", issue = "none")]
790     #[inline(always)]
791     pub unsafe fn slice_get_mut(slice: &mut [Self]) -> &mut [T] {
792         // SAFETY: similar to safety notes for `slice_get_ref`, but we have a
793         // mutable reference which is also guaranteed to be valid for writes.
794         unsafe { &mut *(slice as *mut [Self] as *mut [T]) }
795     }
796
797     /// Gets a pointer to the first element of the array.
798     #[unstable(feature = "maybe_uninit_slice", issue = "63569")]
799     #[inline(always)]
800     pub fn first_ptr(this: &[MaybeUninit<T>]) -> *const T {
801         this as *const [MaybeUninit<T>] as *const T
802     }
803
804     /// Gets a mutable pointer to the first element of the array.
805     #[unstable(feature = "maybe_uninit_slice", issue = "63569")]
806     #[inline(always)]
807     pub fn first_ptr_mut(this: &mut [MaybeUninit<T>]) -> *mut T {
808         this as *mut [MaybeUninit<T>] as *mut T
809     }
810 }