]> git.lizzy.rs Git - rust.git/blob - library/core/src/ptr/mod.rs
Rollup merge of #93057 - frengor:iter_collect_into, r=m-ou-se
[rust.git] / library / core / src / ptr / mod.rs
1 //! Manually manage memory through raw pointers.
2 //!
3 //! *[See also the pointer primitive types](pointer).*
4 //!
5 //! # Safety
6 //!
7 //! Many functions in this module take raw pointers as arguments and read from
8 //! or write to them. For this to be safe, these pointers must be *valid*.
9 //! Whether a pointer is valid depends on the operation it is used for
10 //! (read or write), and the extent of the memory that is accessed (i.e.,
11 //! how many bytes are read/written). Most functions use `*mut T` and `*const T`
12 //! to access only a single value, in which case the documentation omits the size
13 //! and implicitly assumes it to be `size_of::<T>()` bytes.
14 //!
15 //! The precise rules for validity are not determined yet. The guarantees that are
16 //! provided at this point are very minimal:
17 //!
18 //! * A [null] pointer is *never* valid, not even for accesses of [size zero][zst].
19 //! * For a pointer to be valid, it is necessary, but not always sufficient, that the pointer
20 //!   be *dereferenceable*: the memory range of the given size starting at the pointer must all be
21 //!   within the bounds of a single allocated object. Note that in Rust,
22 //!   every (stack-allocated) variable is considered a separate allocated object.
23 //! * Even for operations of [size zero][zst], the pointer must not be pointing to deallocated
24 //!   memory, i.e., deallocation makes pointers invalid even for zero-sized operations. However,
25 //!   casting any non-zero integer *literal* to a pointer is valid for zero-sized accesses, even if
26 //!   some memory happens to exist at that address and gets deallocated. This corresponds to writing
27 //!   your own allocator: allocating zero-sized objects is not very hard. The canonical way to
28 //!   obtain a pointer that is valid for zero-sized accesses is [`NonNull::dangling`].
29 //! * All accesses performed by functions in this module are *non-atomic* in the sense
30 //!   of [atomic operations] used to synchronize between threads. This means it is
31 //!   undefined behavior to perform two concurrent accesses to the same location from different
32 //!   threads unless both accesses only read from memory. Notice that this explicitly
33 //!   includes [`read_volatile`] and [`write_volatile`]: Volatile accesses cannot
34 //!   be used for inter-thread synchronization.
35 //! * The result of casting a reference to a pointer is valid for as long as the
36 //!   underlying object is live and no reference (just raw pointers) is used to
37 //!   access the same memory.
38 //!
39 //! These axioms, along with careful use of [`offset`] for pointer arithmetic,
40 //! are enough to correctly implement many useful things in unsafe code. Stronger guarantees
41 //! will be provided eventually, as the [aliasing] rules are being determined. For more
42 //! information, see the [book] as well as the section in the reference devoted
43 //! to [undefined behavior][ub].
44 //!
45 //! ## Alignment
46 //!
47 //! Valid raw pointers as defined above are not necessarily properly aligned (where
48 //! "proper" alignment is defined by the pointee type, i.e., `*const T` must be
49 //! aligned to `mem::align_of::<T>()`). However, most functions require their
50 //! arguments to be properly aligned, and will explicitly state
51 //! this requirement in their documentation. Notable exceptions to this are
52 //! [`read_unaligned`] and [`write_unaligned`].
53 //!
54 //! When a function requires proper alignment, it does so even if the access
55 //! has size 0, i.e., even if memory is not actually touched. Consider using
56 //! [`NonNull::dangling`] in such cases.
57 //!
58 //! ## Allocated object
59 //!
60 //! For several operations, such as [`offset`] or field projections (`expr.field`), the notion of an
61 //! "allocated object" becomes relevant. An allocated object is a contiguous region of memory.
62 //! Common examples of allocated objects include stack-allocated variables (each variable is a
63 //! separate allocated object), heap allocations (each allocation created by the global allocator is
64 //! a separate allocated object), and `static` variables.
65 //!
66 //! [aliasing]: ../../nomicon/aliasing.html
67 //! [book]: ../../book/ch19-01-unsafe-rust.html#dereferencing-a-raw-pointer
68 //! [ub]: ../../reference/behavior-considered-undefined.html
69 //! [zst]: ../../nomicon/exotic-sizes.html#zero-sized-types-zsts
70 //! [atomic operations]: crate::sync::atomic
71 //! [`offset`]: pointer::offset
72
73 #![stable(feature = "rust1", since = "1.0.0")]
74
75 use crate::cmp::Ordering;
76 use crate::fmt;
77 use crate::hash;
78 use crate::intrinsics::{self, abort, is_aligned_and_not_null};
79 use crate::mem::{self, MaybeUninit};
80
81 #[stable(feature = "rust1", since = "1.0.0")]
82 #[doc(inline)]
83 pub use crate::intrinsics::copy_nonoverlapping;
84
85 #[stable(feature = "rust1", since = "1.0.0")]
86 #[doc(inline)]
87 pub use crate::intrinsics::copy;
88
89 #[stable(feature = "rust1", since = "1.0.0")]
90 #[doc(inline)]
91 pub use crate::intrinsics::write_bytes;
92
93 mod metadata;
94 pub(crate) use metadata::PtrRepr;
95 #[unstable(feature = "ptr_metadata", issue = "81513")]
96 pub use metadata::{from_raw_parts, from_raw_parts_mut, metadata, DynMetadata, Pointee, Thin};
97
98 mod non_null;
99 #[stable(feature = "nonnull", since = "1.25.0")]
100 pub use non_null::NonNull;
101
102 mod unique;
103 #[unstable(feature = "ptr_internals", issue = "none")]
104 pub use unique::Unique;
105
106 mod const_ptr;
107 mod mut_ptr;
108
109 /// Executes the destructor (if any) of the pointed-to value.
110 ///
111 /// This is semantically equivalent to calling [`ptr::read`] and discarding
112 /// the result, but has the following advantages:
113 ///
114 /// * It is *required* to use `drop_in_place` to drop unsized types like
115 ///   trait objects, because they can't be read out onto the stack and
116 ///   dropped normally.
117 ///
118 /// * It is friendlier to the optimizer to do this over [`ptr::read`] when
119 ///   dropping manually allocated memory (e.g., in the implementations of
120 ///   `Box`/`Rc`/`Vec`), as the compiler doesn't need to prove that it's
121 ///   sound to elide the copy.
122 ///
123 /// * It can be used to drop [pinned] data when `T` is not `repr(packed)`
124 ///   (pinned data must not be moved before it is dropped).
125 ///
126 /// Unaligned values cannot be dropped in place, they must be copied to an aligned
127 /// location first using [`ptr::read_unaligned`]. For packed structs, this move is
128 /// done automatically by the compiler. This means the fields of packed structs
129 /// are not dropped in-place.
130 ///
131 /// [`ptr::read`]: self::read
132 /// [`ptr::read_unaligned`]: self::read_unaligned
133 /// [pinned]: crate::pin
134 ///
135 /// # Safety
136 ///
137 /// Behavior is undefined if any of the following conditions are violated:
138 ///
139 /// * `to_drop` must be [valid] for both reads and writes.
140 ///
141 /// * `to_drop` must be properly aligned.
142 ///
143 /// * The value `to_drop` points to must be valid for dropping, which may mean it must uphold
144 ///   additional invariants - this is type-dependent.
145 ///
146 /// Additionally, if `T` is not [`Copy`], using the pointed-to value after
147 /// calling `drop_in_place` can cause undefined behavior. Note that `*to_drop =
148 /// foo` counts as a use because it will cause the value to be dropped
149 /// again. [`write()`] can be used to overwrite data without causing it to be
150 /// dropped.
151 ///
152 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
153 ///
154 /// [valid]: self#safety
155 ///
156 /// # Examples
157 ///
158 /// Manually remove the last item from a vector:
159 ///
160 /// ```
161 /// use std::ptr;
162 /// use std::rc::Rc;
163 ///
164 /// let last = Rc::new(1);
165 /// let weak = Rc::downgrade(&last);
166 ///
167 /// let mut v = vec![Rc::new(0), last];
168 ///
169 /// unsafe {
170 ///     // Get a raw pointer to the last element in `v`.
171 ///     let ptr = &mut v[1] as *mut _;
172 ///     // Shorten `v` to prevent the last item from being dropped. We do that first,
173 ///     // to prevent issues if the `drop_in_place` below panics.
174 ///     v.set_len(1);
175 ///     // Without a call `drop_in_place`, the last item would never be dropped,
176 ///     // and the memory it manages would be leaked.
177 ///     ptr::drop_in_place(ptr);
178 /// }
179 ///
180 /// assert_eq!(v, &[0.into()]);
181 ///
182 /// // Ensure that the last item was dropped.
183 /// assert!(weak.upgrade().is_none());
184 /// ```
185 #[stable(feature = "drop_in_place", since = "1.8.0")]
186 #[lang = "drop_in_place"]
187 #[allow(unconditional_recursion)]
188 pub unsafe fn drop_in_place<T: ?Sized>(to_drop: *mut T) {
189     // Code here does not matter - this is replaced by the
190     // real drop glue by the compiler.
191
192     // SAFETY: see comment above
193     unsafe { drop_in_place(to_drop) }
194 }
195
196 /// Creates a null raw pointer.
197 ///
198 /// # Examples
199 ///
200 /// ```
201 /// use std::ptr;
202 ///
203 /// let p: *const i32 = ptr::null();
204 /// assert!(p.is_null());
205 /// ```
206 #[inline(always)]
207 #[must_use]
208 #[stable(feature = "rust1", since = "1.0.0")]
209 #[rustc_promotable]
210 #[rustc_const_stable(feature = "const_ptr_null", since = "1.24.0")]
211 #[rustc_diagnostic_item = "ptr_null"]
212 pub const fn null<T>() -> *const T {
213     0 as *const T
214 }
215
216 /// Creates a null mutable raw pointer.
217 ///
218 /// # Examples
219 ///
220 /// ```
221 /// use std::ptr;
222 ///
223 /// let p: *mut i32 = ptr::null_mut();
224 /// assert!(p.is_null());
225 /// ```
226 #[inline(always)]
227 #[must_use]
228 #[stable(feature = "rust1", since = "1.0.0")]
229 #[rustc_promotable]
230 #[rustc_const_stable(feature = "const_ptr_null", since = "1.24.0")]
231 #[rustc_diagnostic_item = "ptr_null_mut"]
232 pub const fn null_mut<T>() -> *mut T {
233     0 as *mut T
234 }
235
236 /// Forms a raw slice from a pointer and a length.
237 ///
238 /// The `len` argument is the number of **elements**, not the number of bytes.
239 ///
240 /// This function is safe, but actually using the return value is unsafe.
241 /// See the documentation of [`slice::from_raw_parts`] for slice safety requirements.
242 ///
243 /// [`slice::from_raw_parts`]: crate::slice::from_raw_parts
244 ///
245 /// # Examples
246 ///
247 /// ```rust
248 /// use std::ptr;
249 ///
250 /// // create a slice pointer when starting out with a pointer to the first element
251 /// let x = [5, 6, 7];
252 /// let raw_pointer = x.as_ptr();
253 /// let slice = ptr::slice_from_raw_parts(raw_pointer, 3);
254 /// assert_eq!(unsafe { &*slice }[2], 7);
255 /// ```
256 #[inline]
257 #[stable(feature = "slice_from_raw_parts", since = "1.42.0")]
258 #[rustc_const_unstable(feature = "const_slice_from_raw_parts", issue = "67456")]
259 pub const fn slice_from_raw_parts<T>(data: *const T, len: usize) -> *const [T] {
260     from_raw_parts(data.cast(), len)
261 }
262
263 /// Performs the same functionality as [`slice_from_raw_parts`], except that a
264 /// raw mutable slice is returned, as opposed to a raw immutable slice.
265 ///
266 /// See the documentation of [`slice_from_raw_parts`] for more details.
267 ///
268 /// This function is safe, but actually using the return value is unsafe.
269 /// See the documentation of [`slice::from_raw_parts_mut`] for slice safety requirements.
270 ///
271 /// [`slice::from_raw_parts_mut`]: crate::slice::from_raw_parts_mut
272 ///
273 /// # Examples
274 ///
275 /// ```rust
276 /// use std::ptr;
277 ///
278 /// let x = &mut [5, 6, 7];
279 /// let raw_pointer = x.as_mut_ptr();
280 /// let slice = ptr::slice_from_raw_parts_mut(raw_pointer, 3);
281 ///
282 /// unsafe {
283 ///     (*slice)[2] = 99; // assign a value at an index in the slice
284 /// };
285 ///
286 /// assert_eq!(unsafe { &*slice }[2], 99);
287 /// ```
288 #[inline]
289 #[stable(feature = "slice_from_raw_parts", since = "1.42.0")]
290 #[rustc_const_unstable(feature = "const_slice_from_raw_parts", issue = "67456")]
291 pub const fn slice_from_raw_parts_mut<T>(data: *mut T, len: usize) -> *mut [T] {
292     from_raw_parts_mut(data.cast(), len)
293 }
294
295 /// Swaps the values at two mutable locations of the same type, without
296 /// deinitializing either.
297 ///
298 /// But for the following two exceptions, this function is semantically
299 /// equivalent to [`mem::swap`]:
300 ///
301 /// * It operates on raw pointers instead of references. When references are
302 ///   available, [`mem::swap`] should be preferred.
303 ///
304 /// * The two pointed-to values may overlap. If the values do overlap, then the
305 ///   overlapping region of memory from `x` will be used. This is demonstrated
306 ///   in the second example below.
307 ///
308 /// # Safety
309 ///
310 /// Behavior is undefined if any of the following conditions are violated:
311 ///
312 /// * Both `x` and `y` must be [valid] for both reads and writes.
313 ///
314 /// * Both `x` and `y` must be properly aligned.
315 ///
316 /// Note that even if `T` has size `0`, the pointers must be non-null and properly aligned.
317 ///
318 /// [valid]: self#safety
319 ///
320 /// # Examples
321 ///
322 /// Swapping two non-overlapping regions:
323 ///
324 /// ```
325 /// use std::ptr;
326 ///
327 /// let mut array = [0, 1, 2, 3];
328 ///
329 /// let x = array[0..].as_mut_ptr() as *mut [u32; 2]; // this is `array[0..2]`
330 /// let y = array[2..].as_mut_ptr() as *mut [u32; 2]; // this is `array[2..4]`
331 ///
332 /// unsafe {
333 ///     ptr::swap(x, y);
334 ///     assert_eq!([2, 3, 0, 1], array);
335 /// }
336 /// ```
337 ///
338 /// Swapping two overlapping regions:
339 ///
340 /// ```
341 /// use std::ptr;
342 ///
343 /// let mut array: [i32; 4] = [0, 1, 2, 3];
344 ///
345 /// let array_ptr: *mut i32 = array.as_mut_ptr();
346 ///
347 /// let x = array_ptr as *mut [i32; 3]; // this is `array[0..3]`
348 /// let y = unsafe { array_ptr.add(1) } as *mut [i32; 3]; // this is `array[1..4]`
349 ///
350 /// unsafe {
351 ///     ptr::swap(x, y);
352 ///     // The indices `1..3` of the slice overlap between `x` and `y`.
353 ///     // Reasonable results would be for to them be `[2, 3]`, so that indices `0..3` are
354 ///     // `[1, 2, 3]` (matching `y` before the `swap`); or for them to be `[0, 1]`
355 ///     // so that indices `1..4` are `[0, 1, 2]` (matching `x` before the `swap`).
356 ///     // This implementation is defined to make the latter choice.
357 ///     assert_eq!([1, 0, 1, 2], array);
358 /// }
359 /// ```
360 #[inline]
361 #[stable(feature = "rust1", since = "1.0.0")]
362 #[rustc_const_unstable(feature = "const_swap", issue = "83163")]
363 pub const unsafe fn swap<T>(x: *mut T, y: *mut T) {
364     // Give ourselves some scratch space to work with.
365     // We do not have to worry about drops: `MaybeUninit` does nothing when dropped.
366     let mut tmp = MaybeUninit::<T>::uninit();
367
368     // Perform the swap
369     // SAFETY: the caller must guarantee that `x` and `y` are
370     // valid for writes and properly aligned. `tmp` cannot be
371     // overlapping either `x` or `y` because `tmp` was just allocated
372     // on the stack as a separate allocated object.
373     unsafe {
374         copy_nonoverlapping(x, tmp.as_mut_ptr(), 1);
375         copy(y, x, 1); // `x` and `y` may overlap
376         copy_nonoverlapping(tmp.as_ptr(), y, 1);
377     }
378 }
379
380 /// Swaps `count * size_of::<T>()` bytes between the two regions of memory
381 /// beginning at `x` and `y`. The two regions must *not* overlap.
382 ///
383 /// # Safety
384 ///
385 /// Behavior is undefined if any of the following conditions are violated:
386 ///
387 /// * Both `x` and `y` must be [valid] for both reads and writes of `count *
388 ///   size_of::<T>()` bytes.
389 ///
390 /// * Both `x` and `y` must be properly aligned.
391 ///
392 /// * The region of memory beginning at `x` with a size of `count *
393 ///   size_of::<T>()` bytes must *not* overlap with the region of memory
394 ///   beginning at `y` with the same size.
395 ///
396 /// Note that even if the effectively copied size (`count * size_of::<T>()`) is `0`,
397 /// the pointers must be non-null and properly aligned.
398 ///
399 /// [valid]: self#safety
400 ///
401 /// # Examples
402 ///
403 /// Basic usage:
404 ///
405 /// ```
406 /// use std::ptr;
407 ///
408 /// let mut x = [1, 2, 3, 4];
409 /// let mut y = [7, 8, 9];
410 ///
411 /// unsafe {
412 ///     ptr::swap_nonoverlapping(x.as_mut_ptr(), y.as_mut_ptr(), 2);
413 /// }
414 ///
415 /// assert_eq!(x, [7, 8, 3, 4]);
416 /// assert_eq!(y, [1, 2, 9]);
417 /// ```
418 #[inline]
419 #[stable(feature = "swap_nonoverlapping", since = "1.27.0")]
420 #[rustc_const_unstable(feature = "const_swap", issue = "83163")]
421 pub const unsafe fn swap_nonoverlapping<T>(x: *mut T, y: *mut T, count: usize) {
422     #[allow(unused)]
423     macro_rules! attempt_swap_as_chunks {
424         ($ChunkTy:ty) => {
425             if mem::align_of::<T>() >= mem::align_of::<$ChunkTy>()
426                 && mem::size_of::<T>() % mem::size_of::<$ChunkTy>() == 0
427             {
428                 let x: *mut MaybeUninit<$ChunkTy> = x.cast();
429                 let y: *mut MaybeUninit<$ChunkTy> = y.cast();
430                 let count = count * (mem::size_of::<T>() / mem::size_of::<$ChunkTy>());
431                 // SAFETY: these are the same bytes that the caller promised were
432                 // ok, just typed as `MaybeUninit<ChunkTy>`s instead of as `T`s.
433                 // The `if` condition above ensures that we're not violating
434                 // alignment requirements, and that the division is exact so
435                 // that we don't lose any bytes off the end.
436                 return unsafe { swap_nonoverlapping_simple(x, y, count) };
437             }
438         };
439     }
440
441     // NOTE(scottmcm) MIRI is disabled here as reading in smaller units is a
442     // pessimization for it.  Also, if the type contains any unaligned pointers,
443     // copying those over multiple reads is difficult to support.
444     #[cfg(not(miri))]
445     {
446         // Split up the slice into small power-of-two-sized chunks that LLVM is able
447         // to vectorize (unless it's a special type with more-than-pointer alignment,
448         // because we don't want to pessimize things like slices of SIMD vectors.)
449         if mem::align_of::<T>() <= mem::size_of::<usize>()
450             && (!mem::size_of::<T>().is_power_of_two()
451                 || mem::size_of::<T>() > mem::size_of::<usize>() * 2)
452         {
453             attempt_swap_as_chunks!(usize);
454             attempt_swap_as_chunks!(u8);
455         }
456     }
457
458     // SAFETY: Same preconditions as this function
459     unsafe { swap_nonoverlapping_simple(x, y, count) }
460 }
461
462 /// Same behaviour and safety conditions as [`swap_nonoverlapping`]
463 ///
464 /// LLVM can vectorize this (at least it can for the power-of-two-sized types
465 /// `swap_nonoverlapping` tries to use) so no need to manually SIMD it.
466 #[inline]
467 #[rustc_const_unstable(feature = "const_swap", issue = "83163")]
468 const unsafe fn swap_nonoverlapping_simple<T>(x: *mut T, y: *mut T, count: usize) {
469     let mut i = 0;
470     while i < count {
471         let x: &mut T =
472             // SAFETY: By precondition, `i` is in-bounds because it's below `n`
473             unsafe { &mut *x.add(i) };
474         let y: &mut T =
475             // SAFETY: By precondition, `i` is in-bounds because it's below `n`
476             // and it's distinct from `x` since the ranges are non-overlapping
477             unsafe { &mut *y.add(i) };
478         mem::swap_simple(x, y);
479
480         i += 1;
481     }
482 }
483
484 /// Moves `src` into the pointed `dst`, returning the previous `dst` value.
485 ///
486 /// Neither value is dropped.
487 ///
488 /// This function is semantically equivalent to [`mem::replace`] except that it
489 /// operates on raw pointers instead of references. When references are
490 /// available, [`mem::replace`] should be preferred.
491 ///
492 /// # Safety
493 ///
494 /// Behavior is undefined if any of the following conditions are violated:
495 ///
496 /// * `dst` must be [valid] for both reads and writes.
497 ///
498 /// * `dst` must be properly aligned.
499 ///
500 /// * `dst` must point to a properly initialized value of type `T`.
501 ///
502 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
503 ///
504 /// [valid]: self#safety
505 ///
506 /// # Examples
507 ///
508 /// ```
509 /// use std::ptr;
510 ///
511 /// let mut rust = vec!['b', 'u', 's', 't'];
512 ///
513 /// // `mem::replace` would have the same effect without requiring the unsafe
514 /// // block.
515 /// let b = unsafe {
516 ///     ptr::replace(&mut rust[0], 'r')
517 /// };
518 ///
519 /// assert_eq!(b, 'b');
520 /// assert_eq!(rust, &['r', 'u', 's', 't']);
521 /// ```
522 #[inline]
523 #[stable(feature = "rust1", since = "1.0.0")]
524 #[rustc_const_unstable(feature = "const_replace", issue = "83164")]
525 pub const unsafe fn replace<T>(dst: *mut T, mut src: T) -> T {
526     // SAFETY: the caller must guarantee that `dst` is valid to be
527     // cast to a mutable reference (valid for writes, aligned, initialized),
528     // and cannot overlap `src` since `dst` must point to a distinct
529     // allocated object.
530     unsafe {
531         mem::swap(&mut *dst, &mut src); // cannot overlap
532     }
533     src
534 }
535
536 /// Reads the value from `src` without moving it. This leaves the
537 /// memory in `src` unchanged.
538 ///
539 /// # Safety
540 ///
541 /// Behavior is undefined if any of the following conditions are violated:
542 ///
543 /// * `src` must be [valid] for reads.
544 ///
545 /// * `src` must be properly aligned. Use [`read_unaligned`] if this is not the
546 ///   case.
547 ///
548 /// * `src` must point to a properly initialized value of type `T`.
549 ///
550 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
551 ///
552 /// # Examples
553 ///
554 /// Basic usage:
555 ///
556 /// ```
557 /// let x = 12;
558 /// let y = &x as *const i32;
559 ///
560 /// unsafe {
561 ///     assert_eq!(std::ptr::read(y), 12);
562 /// }
563 /// ```
564 ///
565 /// Manually implement [`mem::swap`]:
566 ///
567 /// ```
568 /// use std::ptr;
569 ///
570 /// fn swap<T>(a: &mut T, b: &mut T) {
571 ///     unsafe {
572 ///         // Create a bitwise copy of the value at `a` in `tmp`.
573 ///         let tmp = ptr::read(a);
574 ///
575 ///         // Exiting at this point (either by explicitly returning or by
576 ///         // calling a function which panics) would cause the value in `tmp` to
577 ///         // be dropped while the same value is still referenced by `a`. This
578 ///         // could trigger undefined behavior if `T` is not `Copy`.
579 ///
580 ///         // Create a bitwise copy of the value at `b` in `a`.
581 ///         // This is safe because mutable references cannot alias.
582 ///         ptr::copy_nonoverlapping(b, a, 1);
583 ///
584 ///         // As above, exiting here could trigger undefined behavior because
585 ///         // the same value is referenced by `a` and `b`.
586 ///
587 ///         // Move `tmp` into `b`.
588 ///         ptr::write(b, tmp);
589 ///
590 ///         // `tmp` has been moved (`write` takes ownership of its second argument),
591 ///         // so nothing is dropped implicitly here.
592 ///     }
593 /// }
594 ///
595 /// let mut foo = "foo".to_owned();
596 /// let mut bar = "bar".to_owned();
597 ///
598 /// swap(&mut foo, &mut bar);
599 ///
600 /// assert_eq!(foo, "bar");
601 /// assert_eq!(bar, "foo");
602 /// ```
603 ///
604 /// ## Ownership of the Returned Value
605 ///
606 /// `read` creates a bitwise copy of `T`, regardless of whether `T` is [`Copy`].
607 /// If `T` is not [`Copy`], using both the returned value and the value at
608 /// `*src` can violate memory safety. Note that assigning to `*src` counts as a
609 /// use because it will attempt to drop the value at `*src`.
610 ///
611 /// [`write()`] can be used to overwrite data without causing it to be dropped.
612 ///
613 /// ```
614 /// use std::ptr;
615 ///
616 /// let mut s = String::from("foo");
617 /// unsafe {
618 ///     // `s2` now points to the same underlying memory as `s`.
619 ///     let mut s2: String = ptr::read(&s);
620 ///
621 ///     assert_eq!(s2, "foo");
622 ///
623 ///     // Assigning to `s2` causes its original value to be dropped. Beyond
624 ///     // this point, `s` must no longer be used, as the underlying memory has
625 ///     // been freed.
626 ///     s2 = String::default();
627 ///     assert_eq!(s2, "");
628 ///
629 ///     // Assigning to `s` would cause the old value to be dropped again,
630 ///     // resulting in undefined behavior.
631 ///     // s = String::from("bar"); // ERROR
632 ///
633 ///     // `ptr::write` can be used to overwrite a value without dropping it.
634 ///     ptr::write(&mut s, String::from("bar"));
635 /// }
636 ///
637 /// assert_eq!(s, "bar");
638 /// ```
639 ///
640 /// [valid]: self#safety
641 #[inline]
642 #[stable(feature = "rust1", since = "1.0.0")]
643 #[rustc_const_unstable(feature = "const_ptr_read", issue = "80377")]
644 pub const unsafe fn read<T>(src: *const T) -> T {
645     // We are calling the intrinsics directly to avoid function calls in the generated code
646     // as `intrinsics::copy_nonoverlapping` is a wrapper function.
647     extern "rust-intrinsic" {
648         #[rustc_const_unstable(feature = "const_intrinsic_copy", issue = "80697")]
649         fn copy_nonoverlapping<T>(src: *const T, dst: *mut T, count: usize);
650     }
651
652     let mut tmp = MaybeUninit::<T>::uninit();
653     // SAFETY: the caller must guarantee that `src` is valid for reads.
654     // `src` cannot overlap `tmp` because `tmp` was just allocated on
655     // the stack as a separate allocated object.
656     //
657     // Also, since we just wrote a valid value into `tmp`, it is guaranteed
658     // to be properly initialized.
659     unsafe {
660         copy_nonoverlapping(src, tmp.as_mut_ptr(), 1);
661         tmp.assume_init()
662     }
663 }
664
665 /// Reads the value from `src` without moving it. This leaves the
666 /// memory in `src` unchanged.
667 ///
668 /// Unlike [`read`], `read_unaligned` works with unaligned pointers.
669 ///
670 /// # Safety
671 ///
672 /// Behavior is undefined if any of the following conditions are violated:
673 ///
674 /// * `src` must be [valid] for reads.
675 ///
676 /// * `src` must point to a properly initialized value of type `T`.
677 ///
678 /// Like [`read`], `read_unaligned` creates a bitwise copy of `T`, regardless of
679 /// whether `T` is [`Copy`]. If `T` is not [`Copy`], using both the returned
680 /// value and the value at `*src` can [violate memory safety][read-ownership].
681 ///
682 /// Note that even if `T` has size `0`, the pointer must be non-null.
683 ///
684 /// [read-ownership]: read#ownership-of-the-returned-value
685 /// [valid]: self#safety
686 ///
687 /// ## On `packed` structs
688 ///
689 /// Attempting to create a raw pointer to an `unaligned` struct field with
690 /// an expression such as `&packed.unaligned as *const FieldType` creates an
691 /// intermediate unaligned reference before converting that to a raw pointer.
692 /// That this reference is temporary and immediately cast is inconsequential
693 /// as the compiler always expects references to be properly aligned.
694 /// As a result, using `&packed.unaligned as *const FieldType` causes immediate
695 /// *undefined behavior* in your program.
696 ///
697 /// Instead you must use the [`ptr::addr_of!`](addr_of) macro to
698 /// create the pointer. You may use that returned pointer together with this
699 /// function.
700 ///
701 /// An example of what not to do and how this relates to `read_unaligned` is:
702 ///
703 /// ```
704 /// #[repr(packed, C)]
705 /// struct Packed {
706 ///     _padding: u8,
707 ///     unaligned: u32,
708 /// }
709 ///
710 /// let packed = Packed {
711 ///     _padding: 0x00,
712 ///     unaligned: 0x01020304,
713 /// };
714 ///
715 /// // Take the address of a 32-bit integer which is not aligned.
716 /// // In contrast to `&packed.unaligned as *const _`, this has no undefined behavior.
717 /// let unaligned = std::ptr::addr_of!(packed.unaligned);
718 ///
719 /// let v = unsafe { std::ptr::read_unaligned(unaligned) };
720 /// assert_eq!(v, 0x01020304);
721 /// ```
722 ///
723 /// Accessing unaligned fields directly with e.g. `packed.unaligned` is safe however.
724 ///
725 /// # Examples
726 ///
727 /// Read a usize value from a byte buffer:
728 ///
729 /// ```
730 /// use std::mem;
731 ///
732 /// fn read_usize(x: &[u8]) -> usize {
733 ///     assert!(x.len() >= mem::size_of::<usize>());
734 ///
735 ///     let ptr = x.as_ptr() as *const usize;
736 ///
737 ///     unsafe { ptr.read_unaligned() }
738 /// }
739 /// ```
740 #[inline]
741 #[stable(feature = "ptr_unaligned", since = "1.17.0")]
742 #[rustc_const_unstable(feature = "const_ptr_read", issue = "80377")]
743 pub const unsafe fn read_unaligned<T>(src: *const T) -> T {
744     let mut tmp = MaybeUninit::<T>::uninit();
745     // SAFETY: the caller must guarantee that `src` is valid for reads.
746     // `src` cannot overlap `tmp` because `tmp` was just allocated on
747     // the stack as a separate allocated object.
748     //
749     // Also, since we just wrote a valid value into `tmp`, it is guaranteed
750     // to be properly initialized.
751     unsafe {
752         copy_nonoverlapping(src as *const u8, tmp.as_mut_ptr() as *mut u8, mem::size_of::<T>());
753         tmp.assume_init()
754     }
755 }
756
757 /// Overwrites a memory location with the given value without reading or
758 /// dropping the old value.
759 ///
760 /// `write` does not drop the contents of `dst`. This is safe, but it could leak
761 /// allocations or resources, so care should be taken not to overwrite an object
762 /// that should be dropped.
763 ///
764 /// Additionally, it does not drop `src`. Semantically, `src` is moved into the
765 /// location pointed to by `dst`.
766 ///
767 /// This is appropriate for initializing uninitialized memory, or overwriting
768 /// memory that has previously been [`read`] from.
769 ///
770 /// # Safety
771 ///
772 /// Behavior is undefined if any of the following conditions are violated:
773 ///
774 /// * `dst` must be [valid] for writes.
775 ///
776 /// * `dst` must be properly aligned. Use [`write_unaligned`] if this is not the
777 ///   case.
778 ///
779 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
780 ///
781 /// [valid]: self#safety
782 ///
783 /// # Examples
784 ///
785 /// Basic usage:
786 ///
787 /// ```
788 /// let mut x = 0;
789 /// let y = &mut x as *mut i32;
790 /// let z = 12;
791 ///
792 /// unsafe {
793 ///     std::ptr::write(y, z);
794 ///     assert_eq!(std::ptr::read(y), 12);
795 /// }
796 /// ```
797 ///
798 /// Manually implement [`mem::swap`]:
799 ///
800 /// ```
801 /// use std::ptr;
802 ///
803 /// fn swap<T>(a: &mut T, b: &mut T) {
804 ///     unsafe {
805 ///         // Create a bitwise copy of the value at `a` in `tmp`.
806 ///         let tmp = ptr::read(a);
807 ///
808 ///         // Exiting at this point (either by explicitly returning or by
809 ///         // calling a function which panics) would cause the value in `tmp` to
810 ///         // be dropped while the same value is still referenced by `a`. This
811 ///         // could trigger undefined behavior if `T` is not `Copy`.
812 ///
813 ///         // Create a bitwise copy of the value at `b` in `a`.
814 ///         // This is safe because mutable references cannot alias.
815 ///         ptr::copy_nonoverlapping(b, a, 1);
816 ///
817 ///         // As above, exiting here could trigger undefined behavior because
818 ///         // the same value is referenced by `a` and `b`.
819 ///
820 ///         // Move `tmp` into `b`.
821 ///         ptr::write(b, tmp);
822 ///
823 ///         // `tmp` has been moved (`write` takes ownership of its second argument),
824 ///         // so nothing is dropped implicitly here.
825 ///     }
826 /// }
827 ///
828 /// let mut foo = "foo".to_owned();
829 /// let mut bar = "bar".to_owned();
830 ///
831 /// swap(&mut foo, &mut bar);
832 ///
833 /// assert_eq!(foo, "bar");
834 /// assert_eq!(bar, "foo");
835 /// ```
836 #[inline]
837 #[stable(feature = "rust1", since = "1.0.0")]
838 #[rustc_const_unstable(feature = "const_ptr_write", issue = "86302")]
839 pub const unsafe fn write<T>(dst: *mut T, src: T) {
840     // We are calling the intrinsics directly to avoid function calls in the generated code
841     // as `intrinsics::copy_nonoverlapping` is a wrapper function.
842     extern "rust-intrinsic" {
843         #[rustc_const_unstable(feature = "const_intrinsic_copy", issue = "80697")]
844         fn copy_nonoverlapping<T>(src: *const T, dst: *mut T, count: usize);
845     }
846
847     // SAFETY: the caller must guarantee that `dst` is valid for writes.
848     // `dst` cannot overlap `src` because the caller has mutable access
849     // to `dst` while `src` is owned by this function.
850     unsafe {
851         copy_nonoverlapping(&src as *const T, dst, 1);
852         intrinsics::forget(src);
853     }
854 }
855
856 /// Overwrites a memory location with the given value without reading or
857 /// dropping the old value.
858 ///
859 /// Unlike [`write()`], the pointer may be unaligned.
860 ///
861 /// `write_unaligned` does not drop the contents of `dst`. This is safe, but it
862 /// could leak allocations or resources, so care should be taken not to overwrite
863 /// an object that should be dropped.
864 ///
865 /// Additionally, it does not drop `src`. Semantically, `src` is moved into the
866 /// location pointed to by `dst`.
867 ///
868 /// This is appropriate for initializing uninitialized memory, or overwriting
869 /// memory that has previously been read with [`read_unaligned`].
870 ///
871 /// # Safety
872 ///
873 /// Behavior is undefined if any of the following conditions are violated:
874 ///
875 /// * `dst` must be [valid] for writes.
876 ///
877 /// Note that even if `T` has size `0`, the pointer must be non-null.
878 ///
879 /// [valid]: self#safety
880 ///
881 /// ## On `packed` structs
882 ///
883 /// Attempting to create a raw pointer to an `unaligned` struct field with
884 /// an expression such as `&packed.unaligned as *const FieldType` creates an
885 /// intermediate unaligned reference before converting that to a raw pointer.
886 /// That this reference is temporary and immediately cast is inconsequential
887 /// as the compiler always expects references to be properly aligned.
888 /// As a result, using `&packed.unaligned as *const FieldType` causes immediate
889 /// *undefined behavior* in your program.
890 ///
891 /// Instead you must use the [`ptr::addr_of_mut!`](addr_of_mut)
892 /// macro to create the pointer. You may use that returned pointer together with
893 /// this function.
894 ///
895 /// An example of how to do it and how this relates to `write_unaligned` is:
896 ///
897 /// ```
898 /// #[repr(packed, C)]
899 /// struct Packed {
900 ///     _padding: u8,
901 ///     unaligned: u32,
902 /// }
903 ///
904 /// let mut packed: Packed = unsafe { std::mem::zeroed() };
905 ///
906 /// // Take the address of a 32-bit integer which is not aligned.
907 /// // In contrast to `&packed.unaligned as *mut _`, this has no undefined behavior.
908 /// let unaligned = std::ptr::addr_of_mut!(packed.unaligned);
909 ///
910 /// unsafe { std::ptr::write_unaligned(unaligned, 42) };
911 ///
912 /// assert_eq!({packed.unaligned}, 42); // `{...}` forces copying the field instead of creating a reference.
913 /// ```
914 ///
915 /// Accessing unaligned fields directly with e.g. `packed.unaligned` is safe however
916 /// (as can be seen in the `assert_eq!` above).
917 ///
918 /// # Examples
919 ///
920 /// Write a usize value to a byte buffer:
921 ///
922 /// ```
923 /// use std::mem;
924 ///
925 /// fn write_usize(x: &mut [u8], val: usize) {
926 ///     assert!(x.len() >= mem::size_of::<usize>());
927 ///
928 ///     let ptr = x.as_mut_ptr() as *mut usize;
929 ///
930 ///     unsafe { ptr.write_unaligned(val) }
931 /// }
932 /// ```
933 #[inline]
934 #[stable(feature = "ptr_unaligned", since = "1.17.0")]
935 #[rustc_const_unstable(feature = "const_ptr_write", issue = "86302")]
936 pub const unsafe fn write_unaligned<T>(dst: *mut T, src: T) {
937     // SAFETY: the caller must guarantee that `dst` is valid for writes.
938     // `dst` cannot overlap `src` because the caller has mutable access
939     // to `dst` while `src` is owned by this function.
940     unsafe {
941         copy_nonoverlapping(&src as *const T as *const u8, dst as *mut u8, mem::size_of::<T>());
942         // We are calling the intrinsic directly to avoid function calls in the generated code.
943         intrinsics::forget(src);
944     }
945 }
946
947 /// Performs a volatile read of the value from `src` without moving it. This
948 /// leaves the memory in `src` unchanged.
949 ///
950 /// Volatile operations are intended to act on I/O memory, and are guaranteed
951 /// to not be elided or reordered by the compiler across other volatile
952 /// operations.
953 ///
954 /// # Notes
955 ///
956 /// Rust does not currently have a rigorously and formally defined memory model,
957 /// so the precise semantics of what "volatile" means here is subject to change
958 /// over time. That being said, the semantics will almost always end up pretty
959 /// similar to [C11's definition of volatile][c11].
960 ///
961 /// The compiler shouldn't change the relative order or number of volatile
962 /// memory operations. However, volatile memory operations on zero-sized types
963 /// (e.g., if a zero-sized type is passed to `read_volatile`) are noops
964 /// and may be ignored.
965 ///
966 /// [c11]: http://www.open-std.org/jtc1/sc22/wg14/www/docs/n1570.pdf
967 ///
968 /// # Safety
969 ///
970 /// Behavior is undefined if any of the following conditions are violated:
971 ///
972 /// * `src` must be [valid] for reads.
973 ///
974 /// * `src` must be properly aligned.
975 ///
976 /// * `src` must point to a properly initialized value of type `T`.
977 ///
978 /// Like [`read`], `read_volatile` creates a bitwise copy of `T`, regardless of
979 /// whether `T` is [`Copy`]. If `T` is not [`Copy`], using both the returned
980 /// value and the value at `*src` can [violate memory safety][read-ownership].
981 /// However, storing non-[`Copy`] types in volatile memory is almost certainly
982 /// incorrect.
983 ///
984 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
985 ///
986 /// [valid]: self#safety
987 /// [read-ownership]: read#ownership-of-the-returned-value
988 ///
989 /// Just like in C, whether an operation is volatile has no bearing whatsoever
990 /// on questions involving concurrent access from multiple threads. Volatile
991 /// accesses behave exactly like non-atomic accesses in that regard. In particular,
992 /// a race between a `read_volatile` and any write operation to the same location
993 /// is undefined behavior.
994 ///
995 /// # Examples
996 ///
997 /// Basic usage:
998 ///
999 /// ```
1000 /// let x = 12;
1001 /// let y = &x as *const i32;
1002 ///
1003 /// unsafe {
1004 ///     assert_eq!(std::ptr::read_volatile(y), 12);
1005 /// }
1006 /// ```
1007 #[inline]
1008 #[stable(feature = "volatile", since = "1.9.0")]
1009 pub unsafe fn read_volatile<T>(src: *const T) -> T {
1010     if cfg!(debug_assertions) && !is_aligned_and_not_null(src) {
1011         // Not panicking to keep codegen impact smaller.
1012         abort();
1013     }
1014     // SAFETY: the caller must uphold the safety contract for `volatile_load`.
1015     unsafe { intrinsics::volatile_load(src) }
1016 }
1017
1018 /// Performs a volatile write of a memory location with the given value without
1019 /// reading or dropping the old value.
1020 ///
1021 /// Volatile operations are intended to act on I/O memory, and are guaranteed
1022 /// to not be elided or reordered by the compiler across other volatile
1023 /// operations.
1024 ///
1025 /// `write_volatile` does not drop the contents of `dst`. This is safe, but it
1026 /// could leak allocations or resources, so care should be taken not to overwrite
1027 /// an object that should be dropped.
1028 ///
1029 /// Additionally, it does not drop `src`. Semantically, `src` is moved into the
1030 /// location pointed to by `dst`.
1031 ///
1032 /// # Notes
1033 ///
1034 /// Rust does not currently have a rigorously and formally defined memory model,
1035 /// so the precise semantics of what "volatile" means here is subject to change
1036 /// over time. That being said, the semantics will almost always end up pretty
1037 /// similar to [C11's definition of volatile][c11].
1038 ///
1039 /// The compiler shouldn't change the relative order or number of volatile
1040 /// memory operations. However, volatile memory operations on zero-sized types
1041 /// (e.g., if a zero-sized type is passed to `write_volatile`) are noops
1042 /// and may be ignored.
1043 ///
1044 /// [c11]: http://www.open-std.org/jtc1/sc22/wg14/www/docs/n1570.pdf
1045 ///
1046 /// # Safety
1047 ///
1048 /// Behavior is undefined if any of the following conditions are violated:
1049 ///
1050 /// * `dst` must be [valid] for writes.
1051 ///
1052 /// * `dst` must be properly aligned.
1053 ///
1054 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
1055 ///
1056 /// [valid]: self#safety
1057 ///
1058 /// Just like in C, whether an operation is volatile has no bearing whatsoever
1059 /// on questions involving concurrent access from multiple threads. Volatile
1060 /// accesses behave exactly like non-atomic accesses in that regard. In particular,
1061 /// a race between a `write_volatile` and any other operation (reading or writing)
1062 /// on the same location is undefined behavior.
1063 ///
1064 /// # Examples
1065 ///
1066 /// Basic usage:
1067 ///
1068 /// ```
1069 /// let mut x = 0;
1070 /// let y = &mut x as *mut i32;
1071 /// let z = 12;
1072 ///
1073 /// unsafe {
1074 ///     std::ptr::write_volatile(y, z);
1075 ///     assert_eq!(std::ptr::read_volatile(y), 12);
1076 /// }
1077 /// ```
1078 #[inline]
1079 #[stable(feature = "volatile", since = "1.9.0")]
1080 pub unsafe fn write_volatile<T>(dst: *mut T, src: T) {
1081     if cfg!(debug_assertions) && !is_aligned_and_not_null(dst) {
1082         // Not panicking to keep codegen impact smaller.
1083         abort();
1084     }
1085     // SAFETY: the caller must uphold the safety contract for `volatile_store`.
1086     unsafe {
1087         intrinsics::volatile_store(dst, src);
1088     }
1089 }
1090
1091 /// Align pointer `p`.
1092 ///
1093 /// Calculate offset (in terms of elements of `stride` stride) that has to be applied
1094 /// to pointer `p` so that pointer `p` would get aligned to `a`.
1095 ///
1096 /// Note: This implementation has been carefully tailored to not panic. It is UB for this to panic.
1097 /// The only real change that can be made here is change of `INV_TABLE_MOD_16` and associated
1098 /// constants.
1099 ///
1100 /// If we ever decide to make it possible to call the intrinsic with `a` that is not a
1101 /// power-of-two, it will probably be more prudent to just change to a naive implementation rather
1102 /// than trying to adapt this to accommodate that change.
1103 ///
1104 /// Any questions go to @nagisa.
1105 #[lang = "align_offset"]
1106 pub(crate) unsafe fn align_offset<T: Sized>(p: *const T, a: usize) -> usize {
1107     // FIXME(#75598): Direct use of these intrinsics improves codegen significantly at opt-level <=
1108     // 1, where the method versions of these operations are not inlined.
1109     use intrinsics::{
1110         unchecked_shl, unchecked_shr, unchecked_sub, wrapping_add, wrapping_mul, wrapping_sub,
1111     };
1112
1113     /// Calculate multiplicative modular inverse of `x` modulo `m`.
1114     ///
1115     /// This implementation is tailored for `align_offset` and has following preconditions:
1116     ///
1117     /// * `m` is a power-of-two;
1118     /// * `x < m`; (if `x ≥ m`, pass in `x % m` instead)
1119     ///
1120     /// Implementation of this function shall not panic. Ever.
1121     #[inline]
1122     unsafe fn mod_inv(x: usize, m: usize) -> usize {
1123         /// Multiplicative modular inverse table modulo 2⁴ = 16.
1124         ///
1125         /// Note, that this table does not contain values where inverse does not exist (i.e., for
1126         /// `0⁻¹ mod 16`, `2⁻¹ mod 16`, etc.)
1127         const INV_TABLE_MOD_16: [u8; 8] = [1, 11, 13, 7, 9, 3, 5, 15];
1128         /// Modulo for which the `INV_TABLE_MOD_16` is intended.
1129         const INV_TABLE_MOD: usize = 16;
1130         /// INV_TABLE_MOD²
1131         const INV_TABLE_MOD_SQUARED: usize = INV_TABLE_MOD * INV_TABLE_MOD;
1132
1133         let table_inverse = INV_TABLE_MOD_16[(x & (INV_TABLE_MOD - 1)) >> 1] as usize;
1134         // SAFETY: `m` is required to be a power-of-two, hence non-zero.
1135         let m_minus_one = unsafe { unchecked_sub(m, 1) };
1136         if m <= INV_TABLE_MOD {
1137             table_inverse & m_minus_one
1138         } else {
1139             // We iterate "up" using the following formula:
1140             //
1141             // $$ xy ≡ 1 (mod 2ⁿ) → xy (2 - xy) ≡ 1 (mod 2²ⁿ) $$
1142             //
1143             // until 2²ⁿ ≥ m. Then we can reduce to our desired `m` by taking the result `mod m`.
1144             let mut inverse = table_inverse;
1145             let mut going_mod = INV_TABLE_MOD_SQUARED;
1146             loop {
1147                 // y = y * (2 - xy) mod n
1148                 //
1149                 // Note, that we use wrapping operations here intentionally – the original formula
1150                 // uses e.g., subtraction `mod n`. It is entirely fine to do them `mod
1151                 // usize::MAX` instead, because we take the result `mod n` at the end
1152                 // anyway.
1153                 inverse = wrapping_mul(inverse, wrapping_sub(2usize, wrapping_mul(x, inverse)));
1154                 if going_mod >= m {
1155                     return inverse & m_minus_one;
1156                 }
1157                 going_mod = wrapping_mul(going_mod, going_mod);
1158             }
1159         }
1160     }
1161
1162     let stride = mem::size_of::<T>();
1163     // SAFETY: `a` is a power-of-two, therefore non-zero.
1164     let a_minus_one = unsafe { unchecked_sub(a, 1) };
1165     if stride == 1 {
1166         // `stride == 1` case can be computed more simply through `-p (mod a)`, but doing so
1167         // inhibits LLVM's ability to select instructions like `lea`. Instead we compute
1168         //
1169         //    round_up_to_next_alignment(p, a) - p
1170         //
1171         // which distributes operations around the load-bearing, but pessimizing `and` sufficiently
1172         // for LLVM to be able to utilize the various optimizations it knows about.
1173         return wrapping_sub(
1174             wrapping_add(p as usize, a_minus_one) & wrapping_sub(0, a),
1175             p as usize,
1176         );
1177     }
1178
1179     let pmoda = p as usize & a_minus_one;
1180     if pmoda == 0 {
1181         // Already aligned. Yay!
1182         return 0;
1183     } else if stride == 0 {
1184         // If the pointer is not aligned, and the element is zero-sized, then no amount of
1185         // elements will ever align the pointer.
1186         return usize::MAX;
1187     }
1188
1189     let smoda = stride & a_minus_one;
1190     // SAFETY: a is power-of-two hence non-zero. stride == 0 case is handled above.
1191     let gcdpow = unsafe { intrinsics::cttz_nonzero(stride).min(intrinsics::cttz_nonzero(a)) };
1192     // SAFETY: gcdpow has an upper-bound that’s at most the number of bits in a usize.
1193     let gcd = unsafe { unchecked_shl(1usize, gcdpow) };
1194
1195     // SAFETY: gcd is always greater or equal to 1.
1196     if p as usize & unsafe { unchecked_sub(gcd, 1) } == 0 {
1197         // This branch solves for the following linear congruence equation:
1198         //
1199         // ` p + so = 0 mod a `
1200         //
1201         // `p` here is the pointer value, `s` - stride of `T`, `o` offset in `T`s, and `a` - the
1202         // requested alignment.
1203         //
1204         // With `g = gcd(a, s)`, and the above condition asserting that `p` is also divisible by
1205         // `g`, we can denote `a' = a/g`, `s' = s/g`, `p' = p/g`, then this becomes equivalent to:
1206         //
1207         // ` p' + s'o = 0 mod a' `
1208         // ` o = (a' - (p' mod a')) * (s'^-1 mod a') `
1209         //
1210         // The first term is "the relative alignment of `p` to `a`" (divided by the `g`), the second
1211         // term is "how does incrementing `p` by `s` bytes change the relative alignment of `p`" (again
1212         // divided by `g`).
1213         // Division by `g` is necessary to make the inverse well formed if `a` and `s` are not
1214         // co-prime.
1215         //
1216         // Furthermore, the result produced by this solution is not "minimal", so it is necessary
1217         // to take the result `o mod lcm(s, a)`. We can replace `lcm(s, a)` with just a `a'`.
1218
1219         // SAFETY: `gcdpow` has an upper-bound not greater than the number of trailing 0-bits in
1220         // `a`.
1221         let a2 = unsafe { unchecked_shr(a, gcdpow) };
1222         // SAFETY: `a2` is non-zero. Shifting `a` by `gcdpow` cannot shift out any of the set bits
1223         // in `a` (of which it has exactly one).
1224         let a2minus1 = unsafe { unchecked_sub(a2, 1) };
1225         // SAFETY: `gcdpow` has an upper-bound not greater than the number of trailing 0-bits in
1226         // `a`.
1227         let s2 = unsafe { unchecked_shr(smoda, gcdpow) };
1228         // SAFETY: `gcdpow` has an upper-bound not greater than the number of trailing 0-bits in
1229         // `a`. Furthermore, the subtraction cannot overflow, because `a2 = a >> gcdpow` will
1230         // always be strictly greater than `(p % a) >> gcdpow`.
1231         let minusp2 = unsafe { unchecked_sub(a2, unchecked_shr(pmoda, gcdpow)) };
1232         // SAFETY: `a2` is a power-of-two, as proven above. `s2` is strictly less than `a2`
1233         // because `(s % a) >> gcdpow` is strictly less than `a >> gcdpow`.
1234         return wrapping_mul(minusp2, unsafe { mod_inv(s2, a2) }) & a2minus1;
1235     }
1236
1237     // Cannot be aligned at all.
1238     usize::MAX
1239 }
1240
1241 /// Compares raw pointers for equality.
1242 ///
1243 /// This is the same as using the `==` operator, but less generic:
1244 /// the arguments have to be `*const T` raw pointers,
1245 /// not anything that implements `PartialEq`.
1246 ///
1247 /// This can be used to compare `&T` references (which coerce to `*const T` implicitly)
1248 /// by their address rather than comparing the values they point to
1249 /// (which is what the `PartialEq for &T` implementation does).
1250 ///
1251 /// # Examples
1252 ///
1253 /// ```
1254 /// use std::ptr;
1255 ///
1256 /// let five = 5;
1257 /// let other_five = 5;
1258 /// let five_ref = &five;
1259 /// let same_five_ref = &five;
1260 /// let other_five_ref = &other_five;
1261 ///
1262 /// assert!(five_ref == same_five_ref);
1263 /// assert!(ptr::eq(five_ref, same_five_ref));
1264 ///
1265 /// assert!(five_ref == other_five_ref);
1266 /// assert!(!ptr::eq(five_ref, other_five_ref));
1267 /// ```
1268 ///
1269 /// Slices are also compared by their length (fat pointers):
1270 ///
1271 /// ```
1272 /// let a = [1, 2, 3];
1273 /// assert!(std::ptr::eq(&a[..3], &a[..3]));
1274 /// assert!(!std::ptr::eq(&a[..2], &a[..3]));
1275 /// assert!(!std::ptr::eq(&a[0..2], &a[1..3]));
1276 /// ```
1277 ///
1278 /// Traits are also compared by their implementation:
1279 ///
1280 /// ```
1281 /// #[repr(transparent)]
1282 /// struct Wrapper { member: i32 }
1283 ///
1284 /// trait Trait {}
1285 /// impl Trait for Wrapper {}
1286 /// impl Trait for i32 {}
1287 ///
1288 /// let wrapper = Wrapper { member: 10 };
1289 ///
1290 /// // Pointers have equal addresses.
1291 /// assert!(std::ptr::eq(
1292 ///     &wrapper as *const Wrapper as *const u8,
1293 ///     &wrapper.member as *const i32 as *const u8
1294 /// ));
1295 ///
1296 /// // Objects have equal addresses, but `Trait` has different implementations.
1297 /// assert!(!std::ptr::eq(
1298 ///     &wrapper as &dyn Trait,
1299 ///     &wrapper.member as &dyn Trait,
1300 /// ));
1301 /// assert!(!std::ptr::eq(
1302 ///     &wrapper as &dyn Trait as *const dyn Trait,
1303 ///     &wrapper.member as &dyn Trait as *const dyn Trait,
1304 /// ));
1305 ///
1306 /// // Converting the reference to a `*const u8` compares by address.
1307 /// assert!(std::ptr::eq(
1308 ///     &wrapper as &dyn Trait as *const dyn Trait as *const u8,
1309 ///     &wrapper.member as &dyn Trait as *const dyn Trait as *const u8,
1310 /// ));
1311 /// ```
1312 #[stable(feature = "ptr_eq", since = "1.17.0")]
1313 #[inline]
1314 pub fn eq<T: ?Sized>(a: *const T, b: *const T) -> bool {
1315     a == b
1316 }
1317
1318 /// Hash a raw pointer.
1319 ///
1320 /// This can be used to hash a `&T` reference (which coerces to `*const T` implicitly)
1321 /// by its address rather than the value it points to
1322 /// (which is what the `Hash for &T` implementation does).
1323 ///
1324 /// # Examples
1325 ///
1326 /// ```
1327 /// use std::collections::hash_map::DefaultHasher;
1328 /// use std::hash::{Hash, Hasher};
1329 /// use std::ptr;
1330 ///
1331 /// let five = 5;
1332 /// let five_ref = &five;
1333 ///
1334 /// let mut hasher = DefaultHasher::new();
1335 /// ptr::hash(five_ref, &mut hasher);
1336 /// let actual = hasher.finish();
1337 ///
1338 /// let mut hasher = DefaultHasher::new();
1339 /// (five_ref as *const i32).hash(&mut hasher);
1340 /// let expected = hasher.finish();
1341 ///
1342 /// assert_eq!(actual, expected);
1343 /// ```
1344 #[stable(feature = "ptr_hash", since = "1.35.0")]
1345 pub fn hash<T: ?Sized, S: hash::Hasher>(hashee: *const T, into: &mut S) {
1346     use crate::hash::Hash;
1347     hashee.hash(into);
1348 }
1349
1350 // Impls for function pointers
1351 macro_rules! fnptr_impls_safety_abi {
1352     ($FnTy: ty, $($Arg: ident),*) => {
1353         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1354         impl<Ret, $($Arg),*> PartialEq for $FnTy {
1355             #[inline]
1356             fn eq(&self, other: &Self) -> bool {
1357                 *self as usize == *other as usize
1358             }
1359         }
1360
1361         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1362         impl<Ret, $($Arg),*> Eq for $FnTy {}
1363
1364         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1365         impl<Ret, $($Arg),*> PartialOrd for $FnTy {
1366             #[inline]
1367             fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
1368                 (*self as usize).partial_cmp(&(*other as usize))
1369             }
1370         }
1371
1372         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1373         impl<Ret, $($Arg),*> Ord for $FnTy {
1374             #[inline]
1375             fn cmp(&self, other: &Self) -> Ordering {
1376                 (*self as usize).cmp(&(*other as usize))
1377             }
1378         }
1379
1380         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1381         impl<Ret, $($Arg),*> hash::Hash for $FnTy {
1382             fn hash<HH: hash::Hasher>(&self, state: &mut HH) {
1383                 state.write_usize(*self as usize)
1384             }
1385         }
1386
1387         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1388         impl<Ret, $($Arg),*> fmt::Pointer for $FnTy {
1389             fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1390                 // HACK: The intermediate cast as usize is required for AVR
1391                 // so that the address space of the source function pointer
1392                 // is preserved in the final function pointer.
1393                 //
1394                 // https://github.com/avr-rust/rust/issues/143
1395                 fmt::Pointer::fmt(&(*self as usize as *const ()), f)
1396             }
1397         }
1398
1399         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1400         impl<Ret, $($Arg),*> fmt::Debug for $FnTy {
1401             fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1402                 // HACK: The intermediate cast as usize is required for AVR
1403                 // so that the address space of the source function pointer
1404                 // is preserved in the final function pointer.
1405                 //
1406                 // https://github.com/avr-rust/rust/issues/143
1407                 fmt::Pointer::fmt(&(*self as usize as *const ()), f)
1408             }
1409         }
1410     }
1411 }
1412
1413 macro_rules! fnptr_impls_args {
1414     ($($Arg: ident),+) => {
1415         fnptr_impls_safety_abi! { extern "Rust" fn($($Arg),+) -> Ret, $($Arg),+ }
1416         fnptr_impls_safety_abi! { extern "C" fn($($Arg),+) -> Ret, $($Arg),+ }
1417         fnptr_impls_safety_abi! { extern "C" fn($($Arg),+ , ...) -> Ret, $($Arg),+ }
1418         fnptr_impls_safety_abi! { unsafe extern "Rust" fn($($Arg),+) -> Ret, $($Arg),+ }
1419         fnptr_impls_safety_abi! { unsafe extern "C" fn($($Arg),+) -> Ret, $($Arg),+ }
1420         fnptr_impls_safety_abi! { unsafe extern "C" fn($($Arg),+ , ...) -> Ret, $($Arg),+ }
1421     };
1422     () => {
1423         // No variadic functions with 0 parameters
1424         fnptr_impls_safety_abi! { extern "Rust" fn() -> Ret, }
1425         fnptr_impls_safety_abi! { extern "C" fn() -> Ret, }
1426         fnptr_impls_safety_abi! { unsafe extern "Rust" fn() -> Ret, }
1427         fnptr_impls_safety_abi! { unsafe extern "C" fn() -> Ret, }
1428     };
1429 }
1430
1431 fnptr_impls_args! {}
1432 fnptr_impls_args! { A }
1433 fnptr_impls_args! { A, B }
1434 fnptr_impls_args! { A, B, C }
1435 fnptr_impls_args! { A, B, C, D }
1436 fnptr_impls_args! { A, B, C, D, E }
1437 fnptr_impls_args! { A, B, C, D, E, F }
1438 fnptr_impls_args! { A, B, C, D, E, F, G }
1439 fnptr_impls_args! { A, B, C, D, E, F, G, H }
1440 fnptr_impls_args! { A, B, C, D, E, F, G, H, I }
1441 fnptr_impls_args! { A, B, C, D, E, F, G, H, I, J }
1442 fnptr_impls_args! { A, B, C, D, E, F, G, H, I, J, K }
1443 fnptr_impls_args! { A, B, C, D, E, F, G, H, I, J, K, L }
1444
1445 /// Create a `const` raw pointer to a place, without creating an intermediate reference.
1446 ///
1447 /// Creating a reference with `&`/`&mut` is only allowed if the pointer is properly aligned
1448 /// and points to initialized data. For cases where those requirements do not hold,
1449 /// raw pointers should be used instead. However, `&expr as *const _` creates a reference
1450 /// before casting it to a raw pointer, and that reference is subject to the same rules
1451 /// as all other references. This macro can create a raw pointer *without* creating
1452 /// a reference first.
1453 ///
1454 /// Note, however, that the `expr` in `addr_of!(expr)` is still subject to all
1455 /// the usual rules. In particular, `addr_of!(*ptr::null())` is Undefined
1456 /// Behavior because it dereferences a null pointer.
1457 ///
1458 /// # Example
1459 ///
1460 /// ```
1461 /// use std::ptr;
1462 ///
1463 /// #[repr(packed)]
1464 /// struct Packed {
1465 ///     f1: u8,
1466 ///     f2: u16,
1467 /// }
1468 ///
1469 /// let packed = Packed { f1: 1, f2: 2 };
1470 /// // `&packed.f2` would create an unaligned reference, and thus be Undefined Behavior!
1471 /// let raw_f2 = ptr::addr_of!(packed.f2);
1472 /// assert_eq!(unsafe { raw_f2.read_unaligned() }, 2);
1473 /// ```
1474 ///
1475 /// See [`addr_of_mut`] for how to create a pointer to unininitialized data.
1476 /// Doing that with `addr_of` would not make much sense since one could only
1477 /// read the data, and that would be Undefined Behavior.
1478 #[stable(feature = "raw_ref_macros", since = "1.51.0")]
1479 #[rustc_macro_transparency = "semitransparent"]
1480 #[allow_internal_unstable(raw_ref_op)]
1481 pub macro addr_of($place:expr) {
1482     &raw const $place
1483 }
1484
1485 /// Create a `mut` raw pointer to a place, without creating an intermediate reference.
1486 ///
1487 /// Creating a reference with `&`/`&mut` is only allowed if the pointer is properly aligned
1488 /// and points to initialized data. For cases where those requirements do not hold,
1489 /// raw pointers should be used instead. However, `&mut expr as *mut _` creates a reference
1490 /// before casting it to a raw pointer, and that reference is subject to the same rules
1491 /// as all other references. This macro can create a raw pointer *without* creating
1492 /// a reference first.
1493 ///
1494 /// Note, however, that the `expr` in `addr_of_mut!(expr)` is still subject to all
1495 /// the usual rules. In particular, `addr_of_mut!(*ptr::null_mut())` is Undefined
1496 /// Behavior because it dereferences a null pointer.
1497 ///
1498 /// # Examples
1499 ///
1500 /// **Creating a pointer to unaligned data:**
1501 ///
1502 /// ```
1503 /// use std::ptr;
1504 ///
1505 /// #[repr(packed)]
1506 /// struct Packed {
1507 ///     f1: u8,
1508 ///     f2: u16,
1509 /// }
1510 ///
1511 /// let mut packed = Packed { f1: 1, f2: 2 };
1512 /// // `&mut packed.f2` would create an unaligned reference, and thus be Undefined Behavior!
1513 /// let raw_f2 = ptr::addr_of_mut!(packed.f2);
1514 /// unsafe { raw_f2.write_unaligned(42); }
1515 /// assert_eq!({packed.f2}, 42); // `{...}` forces copying the field instead of creating a reference.
1516 /// ```
1517 ///
1518 /// **Creating a pointer to uninitialized data:**
1519 ///
1520 /// ```rust
1521 /// use std::{ptr, mem::MaybeUninit};
1522 ///
1523 /// struct Demo {
1524 ///     field: bool,
1525 /// }
1526 ///
1527 /// let mut uninit = MaybeUninit::<Demo>::uninit();
1528 /// // `&uninit.as_mut().field` would create a reference to an uninitialized `bool`,
1529 /// // and thus be Undefined Behavior!
1530 /// let f1_ptr = unsafe { ptr::addr_of_mut!((*uninit.as_mut_ptr()).field) };
1531 /// unsafe { f1_ptr.write(true); }
1532 /// let init = unsafe { uninit.assume_init() };
1533 /// ```
1534 #[stable(feature = "raw_ref_macros", since = "1.51.0")]
1535 #[rustc_macro_transparency = "semitransparent"]
1536 #[allow_internal_unstable(raw_ref_op)]
1537 pub macro addr_of_mut($place:expr) {
1538     &raw mut $place
1539 }