]> git.lizzy.rs Git - rust.git/blob - library/core/src/ptr/mod.rs
Rollup merge of #92714 - yanganto:ignore-message, r=Mark-Simulacrum
[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     macro_rules! attempt_swap_as_chunks {
423         ($ChunkTy:ty) => {
424             if mem::align_of::<T>() >= mem::align_of::<$ChunkTy>()
425                 && mem::size_of::<T>() % mem::size_of::<$ChunkTy>() == 0
426             {
427                 let x: *mut MaybeUninit<$ChunkTy> = x.cast();
428                 let y: *mut MaybeUninit<$ChunkTy> = y.cast();
429                 let count = count * (mem::size_of::<T>() / mem::size_of::<$ChunkTy>());
430                 // SAFETY: these are the same bytes that the caller promised were
431                 // ok, just typed as `MaybeUninit<ChunkTy>`s instead of as `T`s.
432                 // The `if` condition above ensures that we're not violating
433                 // alignment requirements, and that the division is exact so
434                 // that we don't lose any bytes off the end.
435                 return unsafe { swap_nonoverlapping_simple(x, y, count) };
436             }
437         };
438     }
439
440     // Split up the slice into small power-of-two-sized chunks that LLVM is able
441     // to vectorize (unless it's a special type with more-than-pointer alignment,
442     // because we don't want to pessimize things like slices of SIMD vectors.)
443     if mem::align_of::<T>() <= mem::size_of::<usize>()
444         && (!mem::size_of::<T>().is_power_of_two()
445             || mem::size_of::<T>() > mem::size_of::<usize>() * 2)
446     {
447         attempt_swap_as_chunks!(usize);
448         attempt_swap_as_chunks!(u8);
449     }
450
451     // SAFETY: Same preconditions as this function
452     unsafe { swap_nonoverlapping_simple(x, y, count) }
453 }
454
455 /// Same behaviour and safety conditions as [`swap_nonoverlapping`]
456 ///
457 /// LLVM can vectorize this (at least it can for the power-of-two-sized types
458 /// `swap_nonoverlapping` tries to use) so no need to manually SIMD it.
459 #[inline]
460 #[rustc_const_unstable(feature = "const_swap", issue = "83163")]
461 const unsafe fn swap_nonoverlapping_simple<T>(x: *mut T, y: *mut T, count: usize) {
462     let mut i = 0;
463     while i < count {
464         let x: &mut T =
465             // SAFETY: By precondition, `i` is in-bounds because it's below `n`
466             unsafe { &mut *x.add(i) };
467         let y: &mut T =
468             // SAFETY: By precondition, `i` is in-bounds because it's below `n`
469             // and it's distinct from `x` since the ranges are non-overlapping
470             unsafe { &mut *y.add(i) };
471         mem::swap_simple(x, y);
472
473         i += 1;
474     }
475 }
476
477 /// Moves `src` into the pointed `dst`, returning the previous `dst` value.
478 ///
479 /// Neither value is dropped.
480 ///
481 /// This function is semantically equivalent to [`mem::replace`] except that it
482 /// operates on raw pointers instead of references. When references are
483 /// available, [`mem::replace`] should be preferred.
484 ///
485 /// # Safety
486 ///
487 /// Behavior is undefined if any of the following conditions are violated:
488 ///
489 /// * `dst` must be [valid] for both reads and writes.
490 ///
491 /// * `dst` must be properly aligned.
492 ///
493 /// * `dst` must point to a properly initialized value of type `T`.
494 ///
495 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
496 ///
497 /// [valid]: self#safety
498 ///
499 /// # Examples
500 ///
501 /// ```
502 /// use std::ptr;
503 ///
504 /// let mut rust = vec!['b', 'u', 's', 't'];
505 ///
506 /// // `mem::replace` would have the same effect without requiring the unsafe
507 /// // block.
508 /// let b = unsafe {
509 ///     ptr::replace(&mut rust[0], 'r')
510 /// };
511 ///
512 /// assert_eq!(b, 'b');
513 /// assert_eq!(rust, &['r', 'u', 's', 't']);
514 /// ```
515 #[inline]
516 #[stable(feature = "rust1", since = "1.0.0")]
517 #[rustc_const_unstable(feature = "const_replace", issue = "83164")]
518 pub const unsafe fn replace<T>(dst: *mut T, mut src: T) -> T {
519     // SAFETY: the caller must guarantee that `dst` is valid to be
520     // cast to a mutable reference (valid for writes, aligned, initialized),
521     // and cannot overlap `src` since `dst` must point to a distinct
522     // allocated object.
523     unsafe {
524         mem::swap(&mut *dst, &mut src); // cannot overlap
525     }
526     src
527 }
528
529 /// Reads the value from `src` without moving it. This leaves the
530 /// memory in `src` unchanged.
531 ///
532 /// # Safety
533 ///
534 /// Behavior is undefined if any of the following conditions are violated:
535 ///
536 /// * `src` must be [valid] for reads.
537 ///
538 /// * `src` must be properly aligned. Use [`read_unaligned`] if this is not the
539 ///   case.
540 ///
541 /// * `src` must point to a properly initialized value of type `T`.
542 ///
543 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
544 ///
545 /// # Examples
546 ///
547 /// Basic usage:
548 ///
549 /// ```
550 /// let x = 12;
551 /// let y = &x as *const i32;
552 ///
553 /// unsafe {
554 ///     assert_eq!(std::ptr::read(y), 12);
555 /// }
556 /// ```
557 ///
558 /// Manually implement [`mem::swap`]:
559 ///
560 /// ```
561 /// use std::ptr;
562 ///
563 /// fn swap<T>(a: &mut T, b: &mut T) {
564 ///     unsafe {
565 ///         // Create a bitwise copy of the value at `a` in `tmp`.
566 ///         let tmp = ptr::read(a);
567 ///
568 ///         // Exiting at this point (either by explicitly returning or by
569 ///         // calling a function which panics) would cause the value in `tmp` to
570 ///         // be dropped while the same value is still referenced by `a`. This
571 ///         // could trigger undefined behavior if `T` is not `Copy`.
572 ///
573 ///         // Create a bitwise copy of the value at `b` in `a`.
574 ///         // This is safe because mutable references cannot alias.
575 ///         ptr::copy_nonoverlapping(b, a, 1);
576 ///
577 ///         // As above, exiting here could trigger undefined behavior because
578 ///         // the same value is referenced by `a` and `b`.
579 ///
580 ///         // Move `tmp` into `b`.
581 ///         ptr::write(b, tmp);
582 ///
583 ///         // `tmp` has been moved (`write` takes ownership of its second argument),
584 ///         // so nothing is dropped implicitly here.
585 ///     }
586 /// }
587 ///
588 /// let mut foo = "foo".to_owned();
589 /// let mut bar = "bar".to_owned();
590 ///
591 /// swap(&mut foo, &mut bar);
592 ///
593 /// assert_eq!(foo, "bar");
594 /// assert_eq!(bar, "foo");
595 /// ```
596 ///
597 /// ## Ownership of the Returned Value
598 ///
599 /// `read` creates a bitwise copy of `T`, regardless of whether `T` is [`Copy`].
600 /// If `T` is not [`Copy`], using both the returned value and the value at
601 /// `*src` can violate memory safety. Note that assigning to `*src` counts as a
602 /// use because it will attempt to drop the value at `*src`.
603 ///
604 /// [`write()`] can be used to overwrite data without causing it to be dropped.
605 ///
606 /// ```
607 /// use std::ptr;
608 ///
609 /// let mut s = String::from("foo");
610 /// unsafe {
611 ///     // `s2` now points to the same underlying memory as `s`.
612 ///     let mut s2: String = ptr::read(&s);
613 ///
614 ///     assert_eq!(s2, "foo");
615 ///
616 ///     // Assigning to `s2` causes its original value to be dropped. Beyond
617 ///     // this point, `s` must no longer be used, as the underlying memory has
618 ///     // been freed.
619 ///     s2 = String::default();
620 ///     assert_eq!(s2, "");
621 ///
622 ///     // Assigning to `s` would cause the old value to be dropped again,
623 ///     // resulting in undefined behavior.
624 ///     // s = String::from("bar"); // ERROR
625 ///
626 ///     // `ptr::write` can be used to overwrite a value without dropping it.
627 ///     ptr::write(&mut s, String::from("bar"));
628 /// }
629 ///
630 /// assert_eq!(s, "bar");
631 /// ```
632 ///
633 /// [valid]: self#safety
634 #[inline]
635 #[stable(feature = "rust1", since = "1.0.0")]
636 #[rustc_const_unstable(feature = "const_ptr_read", issue = "80377")]
637 pub const unsafe fn read<T>(src: *const T) -> T {
638     // We are calling the intrinsics directly to avoid function calls in the generated code
639     // as `intrinsics::copy_nonoverlapping` is a wrapper function.
640     extern "rust-intrinsic" {
641         #[rustc_const_unstable(feature = "const_intrinsic_copy", issue = "80697")]
642         fn copy_nonoverlapping<T>(src: *const T, dst: *mut T, count: usize);
643     }
644
645     let mut tmp = MaybeUninit::<T>::uninit();
646     // SAFETY: the caller must guarantee that `src` is valid for reads.
647     // `src` cannot overlap `tmp` because `tmp` was just allocated on
648     // the stack as a separate allocated object.
649     //
650     // Also, since we just wrote a valid value into `tmp`, it is guaranteed
651     // to be properly initialized.
652     unsafe {
653         copy_nonoverlapping(src, tmp.as_mut_ptr(), 1);
654         tmp.assume_init()
655     }
656 }
657
658 /// Reads the value from `src` without moving it. This leaves the
659 /// memory in `src` unchanged.
660 ///
661 /// Unlike [`read`], `read_unaligned` works with unaligned pointers.
662 ///
663 /// # Safety
664 ///
665 /// Behavior is undefined if any of the following conditions are violated:
666 ///
667 /// * `src` must be [valid] for reads.
668 ///
669 /// * `src` must point to a properly initialized value of type `T`.
670 ///
671 /// Like [`read`], `read_unaligned` creates a bitwise copy of `T`, regardless of
672 /// whether `T` is [`Copy`]. If `T` is not [`Copy`], using both the returned
673 /// value and the value at `*src` can [violate memory safety][read-ownership].
674 ///
675 /// Note that even if `T` has size `0`, the pointer must be non-null.
676 ///
677 /// [read-ownership]: read#ownership-of-the-returned-value
678 /// [valid]: self#safety
679 ///
680 /// ## On `packed` structs
681 ///
682 /// Attempting to create a raw pointer to an `unaligned` struct field with
683 /// an expression such as `&packed.unaligned as *const FieldType` creates an
684 /// intermediate unaligned reference before converting that to a raw pointer.
685 /// That this reference is temporary and immediately cast is inconsequential
686 /// as the compiler always expects references to be properly aligned.
687 /// As a result, using `&packed.unaligned as *const FieldType` causes immediate
688 /// *undefined behavior* in your program.
689 ///
690 /// Instead you must use the [`ptr::addr_of!`](addr_of) macro to
691 /// create the pointer. You may use that returned pointer together with this
692 /// function.
693 ///
694 /// An example of what not to do and how this relates to `read_unaligned` is:
695 ///
696 /// ```
697 /// #[repr(packed, C)]
698 /// struct Packed {
699 ///     _padding: u8,
700 ///     unaligned: u32,
701 /// }
702 ///
703 /// let packed = Packed {
704 ///     _padding: 0x00,
705 ///     unaligned: 0x01020304,
706 /// };
707 ///
708 /// // Take the address of a 32-bit integer which is not aligned.
709 /// // In contrast to `&packed.unaligned as *const _`, this has no undefined behavior.
710 /// let unaligned = std::ptr::addr_of!(packed.unaligned);
711 ///
712 /// let v = unsafe { std::ptr::read_unaligned(unaligned) };
713 /// assert_eq!(v, 0x01020304);
714 /// ```
715 ///
716 /// Accessing unaligned fields directly with e.g. `packed.unaligned` is safe however.
717 ///
718 /// # Examples
719 ///
720 /// Read a usize value from a byte buffer:
721 ///
722 /// ```
723 /// use std::mem;
724 ///
725 /// fn read_usize(x: &[u8]) -> usize {
726 ///     assert!(x.len() >= mem::size_of::<usize>());
727 ///
728 ///     let ptr = x.as_ptr() as *const usize;
729 ///
730 ///     unsafe { ptr.read_unaligned() }
731 /// }
732 /// ```
733 #[inline]
734 #[stable(feature = "ptr_unaligned", since = "1.17.0")]
735 #[rustc_const_unstable(feature = "const_ptr_read", issue = "80377")]
736 pub const unsafe fn read_unaligned<T>(src: *const T) -> T {
737     let mut tmp = MaybeUninit::<T>::uninit();
738     // SAFETY: the caller must guarantee that `src` is valid for reads.
739     // `src` cannot overlap `tmp` because `tmp` was just allocated on
740     // the stack as a separate allocated object.
741     //
742     // Also, since we just wrote a valid value into `tmp`, it is guaranteed
743     // to be properly initialized.
744     unsafe {
745         copy_nonoverlapping(src as *const u8, tmp.as_mut_ptr() as *mut u8, mem::size_of::<T>());
746         tmp.assume_init()
747     }
748 }
749
750 /// Overwrites a memory location with the given value without reading or
751 /// dropping the old value.
752 ///
753 /// `write` does not drop the contents of `dst`. This is safe, but it could leak
754 /// allocations or resources, so care should be taken not to overwrite an object
755 /// that should be dropped.
756 ///
757 /// Additionally, it does not drop `src`. Semantically, `src` is moved into the
758 /// location pointed to by `dst`.
759 ///
760 /// This is appropriate for initializing uninitialized memory, or overwriting
761 /// memory that has previously been [`read`] from.
762 ///
763 /// # Safety
764 ///
765 /// Behavior is undefined if any of the following conditions are violated:
766 ///
767 /// * `dst` must be [valid] for writes.
768 ///
769 /// * `dst` must be properly aligned. Use [`write_unaligned`] if this is not the
770 ///   case.
771 ///
772 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
773 ///
774 /// [valid]: self#safety
775 ///
776 /// # Examples
777 ///
778 /// Basic usage:
779 ///
780 /// ```
781 /// let mut x = 0;
782 /// let y = &mut x as *mut i32;
783 /// let z = 12;
784 ///
785 /// unsafe {
786 ///     std::ptr::write(y, z);
787 ///     assert_eq!(std::ptr::read(y), 12);
788 /// }
789 /// ```
790 ///
791 /// Manually implement [`mem::swap`]:
792 ///
793 /// ```
794 /// use std::ptr;
795 ///
796 /// fn swap<T>(a: &mut T, b: &mut T) {
797 ///     unsafe {
798 ///         // Create a bitwise copy of the value at `a` in `tmp`.
799 ///         let tmp = ptr::read(a);
800 ///
801 ///         // Exiting at this point (either by explicitly returning or by
802 ///         // calling a function which panics) would cause the value in `tmp` to
803 ///         // be dropped while the same value is still referenced by `a`. This
804 ///         // could trigger undefined behavior if `T` is not `Copy`.
805 ///
806 ///         // Create a bitwise copy of the value at `b` in `a`.
807 ///         // This is safe because mutable references cannot alias.
808 ///         ptr::copy_nonoverlapping(b, a, 1);
809 ///
810 ///         // As above, exiting here could trigger undefined behavior because
811 ///         // the same value is referenced by `a` and `b`.
812 ///
813 ///         // Move `tmp` into `b`.
814 ///         ptr::write(b, tmp);
815 ///
816 ///         // `tmp` has been moved (`write` takes ownership of its second argument),
817 ///         // so nothing is dropped implicitly here.
818 ///     }
819 /// }
820 ///
821 /// let mut foo = "foo".to_owned();
822 /// let mut bar = "bar".to_owned();
823 ///
824 /// swap(&mut foo, &mut bar);
825 ///
826 /// assert_eq!(foo, "bar");
827 /// assert_eq!(bar, "foo");
828 /// ```
829 #[inline]
830 #[stable(feature = "rust1", since = "1.0.0")]
831 #[rustc_const_unstable(feature = "const_ptr_write", issue = "86302")]
832 pub const unsafe fn write<T>(dst: *mut T, src: T) {
833     // We are calling the intrinsics directly to avoid function calls in the generated code
834     // as `intrinsics::copy_nonoverlapping` is a wrapper function.
835     extern "rust-intrinsic" {
836         #[rustc_const_unstable(feature = "const_intrinsic_copy", issue = "80697")]
837         fn copy_nonoverlapping<T>(src: *const T, dst: *mut T, count: usize);
838     }
839
840     // SAFETY: the caller must guarantee that `dst` is valid for writes.
841     // `dst` cannot overlap `src` because the caller has mutable access
842     // to `dst` while `src` is owned by this function.
843     unsafe {
844         copy_nonoverlapping(&src as *const T, dst, 1);
845         intrinsics::forget(src);
846     }
847 }
848
849 /// Overwrites a memory location with the given value without reading or
850 /// dropping the old value.
851 ///
852 /// Unlike [`write()`], the pointer may be unaligned.
853 ///
854 /// `write_unaligned` does not drop the contents of `dst`. This is safe, but it
855 /// could leak allocations or resources, so care should be taken not to overwrite
856 /// an object that should be dropped.
857 ///
858 /// Additionally, it does not drop `src`. Semantically, `src` is moved into the
859 /// location pointed to by `dst`.
860 ///
861 /// This is appropriate for initializing uninitialized memory, or overwriting
862 /// memory that has previously been read with [`read_unaligned`].
863 ///
864 /// # Safety
865 ///
866 /// Behavior is undefined if any of the following conditions are violated:
867 ///
868 /// * `dst` must be [valid] for writes.
869 ///
870 /// Note that even if `T` has size `0`, the pointer must be non-null.
871 ///
872 /// [valid]: self#safety
873 ///
874 /// ## On `packed` structs
875 ///
876 /// Attempting to create a raw pointer to an `unaligned` struct field with
877 /// an expression such as `&packed.unaligned as *const FieldType` creates an
878 /// intermediate unaligned reference before converting that to a raw pointer.
879 /// That this reference is temporary and immediately cast is inconsequential
880 /// as the compiler always expects references to be properly aligned.
881 /// As a result, using `&packed.unaligned as *const FieldType` causes immediate
882 /// *undefined behavior* in your program.
883 ///
884 /// Instead you must use the [`ptr::addr_of_mut!`](addr_of_mut)
885 /// macro to create the pointer. You may use that returned pointer together with
886 /// this function.
887 ///
888 /// An example of how to do it and how this relates to `write_unaligned` is:
889 ///
890 /// ```
891 /// #[repr(packed, C)]
892 /// struct Packed {
893 ///     _padding: u8,
894 ///     unaligned: u32,
895 /// }
896 ///
897 /// let mut packed: Packed = unsafe { std::mem::zeroed() };
898 ///
899 /// // Take the address of a 32-bit integer which is not aligned.
900 /// // In contrast to `&packed.unaligned as *mut _`, this has no undefined behavior.
901 /// let unaligned = std::ptr::addr_of_mut!(packed.unaligned);
902 ///
903 /// unsafe { std::ptr::write_unaligned(unaligned, 42) };
904 ///
905 /// assert_eq!({packed.unaligned}, 42); // `{...}` forces copying the field instead of creating a reference.
906 /// ```
907 ///
908 /// Accessing unaligned fields directly with e.g. `packed.unaligned` is safe however
909 /// (as can be seen in the `assert_eq!` above).
910 ///
911 /// # Examples
912 ///
913 /// Write a usize value to a byte buffer:
914 ///
915 /// ```
916 /// use std::mem;
917 ///
918 /// fn write_usize(x: &mut [u8], val: usize) {
919 ///     assert!(x.len() >= mem::size_of::<usize>());
920 ///
921 ///     let ptr = x.as_mut_ptr() as *mut usize;
922 ///
923 ///     unsafe { ptr.write_unaligned(val) }
924 /// }
925 /// ```
926 #[inline]
927 #[stable(feature = "ptr_unaligned", since = "1.17.0")]
928 #[rustc_const_unstable(feature = "const_ptr_write", issue = "86302")]
929 pub const unsafe fn write_unaligned<T>(dst: *mut T, src: T) {
930     // SAFETY: the caller must guarantee that `dst` is valid for writes.
931     // `dst` cannot overlap `src` because the caller has mutable access
932     // to `dst` while `src` is owned by this function.
933     unsafe {
934         copy_nonoverlapping(&src as *const T as *const u8, dst as *mut u8, mem::size_of::<T>());
935         // We are calling the intrinsic directly to avoid function calls in the generated code.
936         intrinsics::forget(src);
937     }
938 }
939
940 /// Performs a volatile read of the value from `src` without moving it. This
941 /// leaves the memory in `src` unchanged.
942 ///
943 /// Volatile operations are intended to act on I/O memory, and are guaranteed
944 /// to not be elided or reordered by the compiler across other volatile
945 /// operations.
946 ///
947 /// # Notes
948 ///
949 /// Rust does not currently have a rigorously and formally defined memory model,
950 /// so the precise semantics of what "volatile" means here is subject to change
951 /// over time. That being said, the semantics will almost always end up pretty
952 /// similar to [C11's definition of volatile][c11].
953 ///
954 /// The compiler shouldn't change the relative order or number of volatile
955 /// memory operations. However, volatile memory operations on zero-sized types
956 /// (e.g., if a zero-sized type is passed to `read_volatile`) are noops
957 /// and may be ignored.
958 ///
959 /// [c11]: http://www.open-std.org/jtc1/sc22/wg14/www/docs/n1570.pdf
960 ///
961 /// # Safety
962 ///
963 /// Behavior is undefined if any of the following conditions are violated:
964 ///
965 /// * `src` must be [valid] for reads.
966 ///
967 /// * `src` must be properly aligned.
968 ///
969 /// * `src` must point to a properly initialized value of type `T`.
970 ///
971 /// Like [`read`], `read_volatile` creates a bitwise copy of `T`, regardless of
972 /// whether `T` is [`Copy`]. If `T` is not [`Copy`], using both the returned
973 /// value and the value at `*src` can [violate memory safety][read-ownership].
974 /// However, storing non-[`Copy`] types in volatile memory is almost certainly
975 /// incorrect.
976 ///
977 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
978 ///
979 /// [valid]: self#safety
980 /// [read-ownership]: read#ownership-of-the-returned-value
981 ///
982 /// Just like in C, whether an operation is volatile has no bearing whatsoever
983 /// on questions involving concurrent access from multiple threads. Volatile
984 /// accesses behave exactly like non-atomic accesses in that regard. In particular,
985 /// a race between a `read_volatile` and any write operation to the same location
986 /// is undefined behavior.
987 ///
988 /// # Examples
989 ///
990 /// Basic usage:
991 ///
992 /// ```
993 /// let x = 12;
994 /// let y = &x as *const i32;
995 ///
996 /// unsafe {
997 ///     assert_eq!(std::ptr::read_volatile(y), 12);
998 /// }
999 /// ```
1000 #[inline]
1001 #[stable(feature = "volatile", since = "1.9.0")]
1002 pub unsafe fn read_volatile<T>(src: *const T) -> T {
1003     if cfg!(debug_assertions) && !is_aligned_and_not_null(src) {
1004         // Not panicking to keep codegen impact smaller.
1005         abort();
1006     }
1007     // SAFETY: the caller must uphold the safety contract for `volatile_load`.
1008     unsafe { intrinsics::volatile_load(src) }
1009 }
1010
1011 /// Performs a volatile write of a memory location with the given value without
1012 /// reading or dropping the old value.
1013 ///
1014 /// Volatile operations are intended to act on I/O memory, and are guaranteed
1015 /// to not be elided or reordered by the compiler across other volatile
1016 /// operations.
1017 ///
1018 /// `write_volatile` does not drop the contents of `dst`. This is safe, but it
1019 /// could leak allocations or resources, so care should be taken not to overwrite
1020 /// an object that should be dropped.
1021 ///
1022 /// Additionally, it does not drop `src`. Semantically, `src` is moved into the
1023 /// location pointed to by `dst`.
1024 ///
1025 /// # Notes
1026 ///
1027 /// Rust does not currently have a rigorously and formally defined memory model,
1028 /// so the precise semantics of what "volatile" means here is subject to change
1029 /// over time. That being said, the semantics will almost always end up pretty
1030 /// similar to [C11's definition of volatile][c11].
1031 ///
1032 /// The compiler shouldn't change the relative order or number of volatile
1033 /// memory operations. However, volatile memory operations on zero-sized types
1034 /// (e.g., if a zero-sized type is passed to `write_volatile`) are noops
1035 /// and may be ignored.
1036 ///
1037 /// [c11]: http://www.open-std.org/jtc1/sc22/wg14/www/docs/n1570.pdf
1038 ///
1039 /// # Safety
1040 ///
1041 /// Behavior is undefined if any of the following conditions are violated:
1042 ///
1043 /// * `dst` must be [valid] for writes.
1044 ///
1045 /// * `dst` must be properly aligned.
1046 ///
1047 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
1048 ///
1049 /// [valid]: self#safety
1050 ///
1051 /// Just like in C, whether an operation is volatile has no bearing whatsoever
1052 /// on questions involving concurrent access from multiple threads. Volatile
1053 /// accesses behave exactly like non-atomic accesses in that regard. In particular,
1054 /// a race between a `write_volatile` and any other operation (reading or writing)
1055 /// on the same location is undefined behavior.
1056 ///
1057 /// # Examples
1058 ///
1059 /// Basic usage:
1060 ///
1061 /// ```
1062 /// let mut x = 0;
1063 /// let y = &mut x as *mut i32;
1064 /// let z = 12;
1065 ///
1066 /// unsafe {
1067 ///     std::ptr::write_volatile(y, z);
1068 ///     assert_eq!(std::ptr::read_volatile(y), 12);
1069 /// }
1070 /// ```
1071 #[inline]
1072 #[stable(feature = "volatile", since = "1.9.0")]
1073 pub unsafe fn write_volatile<T>(dst: *mut T, src: T) {
1074     if cfg!(debug_assertions) && !is_aligned_and_not_null(dst) {
1075         // Not panicking to keep codegen impact smaller.
1076         abort();
1077     }
1078     // SAFETY: the caller must uphold the safety contract for `volatile_store`.
1079     unsafe {
1080         intrinsics::volatile_store(dst, src);
1081     }
1082 }
1083
1084 /// Align pointer `p`.
1085 ///
1086 /// Calculate offset (in terms of elements of `stride` stride) that has to be applied
1087 /// to pointer `p` so that pointer `p` would get aligned to `a`.
1088 ///
1089 /// Note: This implementation has been carefully tailored to not panic. It is UB for this to panic.
1090 /// The only real change that can be made here is change of `INV_TABLE_MOD_16` and associated
1091 /// constants.
1092 ///
1093 /// If we ever decide to make it possible to call the intrinsic with `a` that is not a
1094 /// power-of-two, it will probably be more prudent to just change to a naive implementation rather
1095 /// than trying to adapt this to accommodate that change.
1096 ///
1097 /// Any questions go to @nagisa.
1098 #[lang = "align_offset"]
1099 pub(crate) unsafe fn align_offset<T: Sized>(p: *const T, a: usize) -> usize {
1100     // FIXME(#75598): Direct use of these intrinsics improves codegen significantly at opt-level <=
1101     // 1, where the method versions of these operations are not inlined.
1102     use intrinsics::{
1103         unchecked_shl, unchecked_shr, unchecked_sub, wrapping_add, wrapping_mul, wrapping_sub,
1104     };
1105
1106     /// Calculate multiplicative modular inverse of `x` modulo `m`.
1107     ///
1108     /// This implementation is tailored for `align_offset` and has following preconditions:
1109     ///
1110     /// * `m` is a power-of-two;
1111     /// * `x < m`; (if `x ≥ m`, pass in `x % m` instead)
1112     ///
1113     /// Implementation of this function shall not panic. Ever.
1114     #[inline]
1115     unsafe fn mod_inv(x: usize, m: usize) -> usize {
1116         /// Multiplicative modular inverse table modulo 2⁴ = 16.
1117         ///
1118         /// Note, that this table does not contain values where inverse does not exist (i.e., for
1119         /// `0⁻¹ mod 16`, `2⁻¹ mod 16`, etc.)
1120         const INV_TABLE_MOD_16: [u8; 8] = [1, 11, 13, 7, 9, 3, 5, 15];
1121         /// Modulo for which the `INV_TABLE_MOD_16` is intended.
1122         const INV_TABLE_MOD: usize = 16;
1123         /// INV_TABLE_MOD²
1124         const INV_TABLE_MOD_SQUARED: usize = INV_TABLE_MOD * INV_TABLE_MOD;
1125
1126         let table_inverse = INV_TABLE_MOD_16[(x & (INV_TABLE_MOD - 1)) >> 1] as usize;
1127         // SAFETY: `m` is required to be a power-of-two, hence non-zero.
1128         let m_minus_one = unsafe { unchecked_sub(m, 1) };
1129         if m <= INV_TABLE_MOD {
1130             table_inverse & m_minus_one
1131         } else {
1132             // We iterate "up" using the following formula:
1133             //
1134             // $$ xy ≡ 1 (mod 2ⁿ) → xy (2 - xy) ≡ 1 (mod 2²ⁿ) $$
1135             //
1136             // until 2²ⁿ ≥ m. Then we can reduce to our desired `m` by taking the result `mod m`.
1137             let mut inverse = table_inverse;
1138             let mut going_mod = INV_TABLE_MOD_SQUARED;
1139             loop {
1140                 // y = y * (2 - xy) mod n
1141                 //
1142                 // Note, that we use wrapping operations here intentionally – the original formula
1143                 // uses e.g., subtraction `mod n`. It is entirely fine to do them `mod
1144                 // usize::MAX` instead, because we take the result `mod n` at the end
1145                 // anyway.
1146                 inverse = wrapping_mul(inverse, wrapping_sub(2usize, wrapping_mul(x, inverse)));
1147                 if going_mod >= m {
1148                     return inverse & m_minus_one;
1149                 }
1150                 going_mod = wrapping_mul(going_mod, going_mod);
1151             }
1152         }
1153     }
1154
1155     let stride = mem::size_of::<T>();
1156     // SAFETY: `a` is a power-of-two, therefore non-zero.
1157     let a_minus_one = unsafe { unchecked_sub(a, 1) };
1158     if stride == 1 {
1159         // `stride == 1` case can be computed more simply through `-p (mod a)`, but doing so
1160         // inhibits LLVM's ability to select instructions like `lea`. Instead we compute
1161         //
1162         //    round_up_to_next_alignment(p, a) - p
1163         //
1164         // which distributes operations around the load-bearing, but pessimizing `and` sufficiently
1165         // for LLVM to be able to utilize the various optimizations it knows about.
1166         return wrapping_sub(
1167             wrapping_add(p as usize, a_minus_one) & wrapping_sub(0, a),
1168             p as usize,
1169         );
1170     }
1171
1172     let pmoda = p as usize & a_minus_one;
1173     if pmoda == 0 {
1174         // Already aligned. Yay!
1175         return 0;
1176     } else if stride == 0 {
1177         // If the pointer is not aligned, and the element is zero-sized, then no amount of
1178         // elements will ever align the pointer.
1179         return usize::MAX;
1180     }
1181
1182     let smoda = stride & a_minus_one;
1183     // SAFETY: a is power-of-two hence non-zero. stride == 0 case is handled above.
1184     let gcdpow = unsafe { intrinsics::cttz_nonzero(stride).min(intrinsics::cttz_nonzero(a)) };
1185     // SAFETY: gcdpow has an upper-bound that’s at most the number of bits in a usize.
1186     let gcd = unsafe { unchecked_shl(1usize, gcdpow) };
1187
1188     // SAFETY: gcd is always greater or equal to 1.
1189     if p as usize & unsafe { unchecked_sub(gcd, 1) } == 0 {
1190         // This branch solves for the following linear congruence equation:
1191         //
1192         // ` p + so = 0 mod a `
1193         //
1194         // `p` here is the pointer value, `s` - stride of `T`, `o` offset in `T`s, and `a` - the
1195         // requested alignment.
1196         //
1197         // With `g = gcd(a, s)`, and the above condition asserting that `p` is also divisible by
1198         // `g`, we can denote `a' = a/g`, `s' = s/g`, `p' = p/g`, then this becomes equivalent to:
1199         //
1200         // ` p' + s'o = 0 mod a' `
1201         // ` o = (a' - (p' mod a')) * (s'^-1 mod a') `
1202         //
1203         // The first term is "the relative alignment of `p` to `a`" (divided by the `g`), the second
1204         // term is "how does incrementing `p` by `s` bytes change the relative alignment of `p`" (again
1205         // divided by `g`).
1206         // Division by `g` is necessary to make the inverse well formed if `a` and `s` are not
1207         // co-prime.
1208         //
1209         // Furthermore, the result produced by this solution is not "minimal", so it is necessary
1210         // to take the result `o mod lcm(s, a)`. We can replace `lcm(s, a)` with just a `a'`.
1211
1212         // SAFETY: `gcdpow` has an upper-bound not greater than the number of trailing 0-bits in
1213         // `a`.
1214         let a2 = unsafe { unchecked_shr(a, gcdpow) };
1215         // SAFETY: `a2` is non-zero. Shifting `a` by `gcdpow` cannot shift out any of the set bits
1216         // in `a` (of which it has exactly one).
1217         let a2minus1 = unsafe { unchecked_sub(a2, 1) };
1218         // SAFETY: `gcdpow` has an upper-bound not greater than the number of trailing 0-bits in
1219         // `a`.
1220         let s2 = unsafe { unchecked_shr(smoda, gcdpow) };
1221         // SAFETY: `gcdpow` has an upper-bound not greater than the number of trailing 0-bits in
1222         // `a`. Furthermore, the subtraction cannot overflow, because `a2 = a >> gcdpow` will
1223         // always be strictly greater than `(p % a) >> gcdpow`.
1224         let minusp2 = unsafe { unchecked_sub(a2, unchecked_shr(pmoda, gcdpow)) };
1225         // SAFETY: `a2` is a power-of-two, as proven above. `s2` is strictly less than `a2`
1226         // because `(s % a) >> gcdpow` is strictly less than `a >> gcdpow`.
1227         return wrapping_mul(minusp2, unsafe { mod_inv(s2, a2) }) & a2minus1;
1228     }
1229
1230     // Cannot be aligned at all.
1231     usize::MAX
1232 }
1233
1234 /// Compares raw pointers for equality.
1235 ///
1236 /// This is the same as using the `==` operator, but less generic:
1237 /// the arguments have to be `*const T` raw pointers,
1238 /// not anything that implements `PartialEq`.
1239 ///
1240 /// This can be used to compare `&T` references (which coerce to `*const T` implicitly)
1241 /// by their address rather than comparing the values they point to
1242 /// (which is what the `PartialEq for &T` implementation does).
1243 ///
1244 /// # Examples
1245 ///
1246 /// ```
1247 /// use std::ptr;
1248 ///
1249 /// let five = 5;
1250 /// let other_five = 5;
1251 /// let five_ref = &five;
1252 /// let same_five_ref = &five;
1253 /// let other_five_ref = &other_five;
1254 ///
1255 /// assert!(five_ref == same_five_ref);
1256 /// assert!(ptr::eq(five_ref, same_five_ref));
1257 ///
1258 /// assert!(five_ref == other_five_ref);
1259 /// assert!(!ptr::eq(five_ref, other_five_ref));
1260 /// ```
1261 ///
1262 /// Slices are also compared by their length (fat pointers):
1263 ///
1264 /// ```
1265 /// let a = [1, 2, 3];
1266 /// assert!(std::ptr::eq(&a[..3], &a[..3]));
1267 /// assert!(!std::ptr::eq(&a[..2], &a[..3]));
1268 /// assert!(!std::ptr::eq(&a[0..2], &a[1..3]));
1269 /// ```
1270 ///
1271 /// Traits are also compared by their implementation:
1272 ///
1273 /// ```
1274 /// #[repr(transparent)]
1275 /// struct Wrapper { member: i32 }
1276 ///
1277 /// trait Trait {}
1278 /// impl Trait for Wrapper {}
1279 /// impl Trait for i32 {}
1280 ///
1281 /// let wrapper = Wrapper { member: 10 };
1282 ///
1283 /// // Pointers have equal addresses.
1284 /// assert!(std::ptr::eq(
1285 ///     &wrapper as *const Wrapper as *const u8,
1286 ///     &wrapper.member as *const i32 as *const u8
1287 /// ));
1288 ///
1289 /// // Objects have equal addresses, but `Trait` has different implementations.
1290 /// assert!(!std::ptr::eq(
1291 ///     &wrapper as &dyn Trait,
1292 ///     &wrapper.member as &dyn Trait,
1293 /// ));
1294 /// assert!(!std::ptr::eq(
1295 ///     &wrapper as &dyn Trait as *const dyn Trait,
1296 ///     &wrapper.member as &dyn Trait as *const dyn Trait,
1297 /// ));
1298 ///
1299 /// // Converting the reference to a `*const u8` compares by address.
1300 /// assert!(std::ptr::eq(
1301 ///     &wrapper as &dyn Trait as *const dyn Trait as *const u8,
1302 ///     &wrapper.member as &dyn Trait as *const dyn Trait as *const u8,
1303 /// ));
1304 /// ```
1305 #[stable(feature = "ptr_eq", since = "1.17.0")]
1306 #[inline]
1307 pub fn eq<T: ?Sized>(a: *const T, b: *const T) -> bool {
1308     a == b
1309 }
1310
1311 /// Hash a raw pointer.
1312 ///
1313 /// This can be used to hash a `&T` reference (which coerces to `*const T` implicitly)
1314 /// by its address rather than the value it points to
1315 /// (which is what the `Hash for &T` implementation does).
1316 ///
1317 /// # Examples
1318 ///
1319 /// ```
1320 /// use std::collections::hash_map::DefaultHasher;
1321 /// use std::hash::{Hash, Hasher};
1322 /// use std::ptr;
1323 ///
1324 /// let five = 5;
1325 /// let five_ref = &five;
1326 ///
1327 /// let mut hasher = DefaultHasher::new();
1328 /// ptr::hash(five_ref, &mut hasher);
1329 /// let actual = hasher.finish();
1330 ///
1331 /// let mut hasher = DefaultHasher::new();
1332 /// (five_ref as *const i32).hash(&mut hasher);
1333 /// let expected = hasher.finish();
1334 ///
1335 /// assert_eq!(actual, expected);
1336 /// ```
1337 #[stable(feature = "ptr_hash", since = "1.35.0")]
1338 pub fn hash<T: ?Sized, S: hash::Hasher>(hashee: *const T, into: &mut S) {
1339     use crate::hash::Hash;
1340     hashee.hash(into);
1341 }
1342
1343 // Impls for function pointers
1344 macro_rules! fnptr_impls_safety_abi {
1345     ($FnTy: ty, $($Arg: ident),*) => {
1346         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1347         impl<Ret, $($Arg),*> PartialEq for $FnTy {
1348             #[inline]
1349             fn eq(&self, other: &Self) -> bool {
1350                 *self as usize == *other as usize
1351             }
1352         }
1353
1354         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1355         impl<Ret, $($Arg),*> Eq for $FnTy {}
1356
1357         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1358         impl<Ret, $($Arg),*> PartialOrd for $FnTy {
1359             #[inline]
1360             fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
1361                 (*self as usize).partial_cmp(&(*other as usize))
1362             }
1363         }
1364
1365         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1366         impl<Ret, $($Arg),*> Ord for $FnTy {
1367             #[inline]
1368             fn cmp(&self, other: &Self) -> Ordering {
1369                 (*self as usize).cmp(&(*other as usize))
1370             }
1371         }
1372
1373         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1374         impl<Ret, $($Arg),*> hash::Hash for $FnTy {
1375             fn hash<HH: hash::Hasher>(&self, state: &mut HH) {
1376                 state.write_usize(*self as usize)
1377             }
1378         }
1379
1380         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1381         impl<Ret, $($Arg),*> fmt::Pointer for $FnTy {
1382             fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1383                 // HACK: The intermediate cast as usize is required for AVR
1384                 // so that the address space of the source function pointer
1385                 // is preserved in the final function pointer.
1386                 //
1387                 // https://github.com/avr-rust/rust/issues/143
1388                 fmt::Pointer::fmt(&(*self as usize as *const ()), f)
1389             }
1390         }
1391
1392         #[stable(feature = "fnptr_impls", since = "1.4.0")]
1393         impl<Ret, $($Arg),*> fmt::Debug for $FnTy {
1394             fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1395                 // HACK: The intermediate cast as usize is required for AVR
1396                 // so that the address space of the source function pointer
1397                 // is preserved in the final function pointer.
1398                 //
1399                 // https://github.com/avr-rust/rust/issues/143
1400                 fmt::Pointer::fmt(&(*self as usize as *const ()), f)
1401             }
1402         }
1403     }
1404 }
1405
1406 macro_rules! fnptr_impls_args {
1407     ($($Arg: ident),+) => {
1408         fnptr_impls_safety_abi! { extern "Rust" fn($($Arg),+) -> Ret, $($Arg),+ }
1409         fnptr_impls_safety_abi! { extern "C" fn($($Arg),+) -> Ret, $($Arg),+ }
1410         fnptr_impls_safety_abi! { extern "C" fn($($Arg),+ , ...) -> Ret, $($Arg),+ }
1411         fnptr_impls_safety_abi! { unsafe extern "Rust" fn($($Arg),+) -> Ret, $($Arg),+ }
1412         fnptr_impls_safety_abi! { unsafe extern "C" fn($($Arg),+) -> Ret, $($Arg),+ }
1413         fnptr_impls_safety_abi! { unsafe extern "C" fn($($Arg),+ , ...) -> Ret, $($Arg),+ }
1414     };
1415     () => {
1416         // No variadic functions with 0 parameters
1417         fnptr_impls_safety_abi! { extern "Rust" fn() -> Ret, }
1418         fnptr_impls_safety_abi! { extern "C" fn() -> Ret, }
1419         fnptr_impls_safety_abi! { unsafe extern "Rust" fn() -> Ret, }
1420         fnptr_impls_safety_abi! { unsafe extern "C" fn() -> Ret, }
1421     };
1422 }
1423
1424 fnptr_impls_args! {}
1425 fnptr_impls_args! { A }
1426 fnptr_impls_args! { A, B }
1427 fnptr_impls_args! { A, B, C }
1428 fnptr_impls_args! { A, B, C, D }
1429 fnptr_impls_args! { A, B, C, D, E }
1430 fnptr_impls_args! { A, B, C, D, E, F }
1431 fnptr_impls_args! { A, B, C, D, E, F, G }
1432 fnptr_impls_args! { A, B, C, D, E, F, G, H }
1433 fnptr_impls_args! { A, B, C, D, E, F, G, H, I }
1434 fnptr_impls_args! { A, B, C, D, E, F, G, H, I, J }
1435 fnptr_impls_args! { A, B, C, D, E, F, G, H, I, J, K }
1436 fnptr_impls_args! { A, B, C, D, E, F, G, H, I, J, K, L }
1437
1438 /// Create a `const` raw pointer to a place, without creating an intermediate reference.
1439 ///
1440 /// Creating a reference with `&`/`&mut` is only allowed if the pointer is properly aligned
1441 /// and points to initialized data. For cases where those requirements do not hold,
1442 /// raw pointers should be used instead. However, `&expr as *const _` creates a reference
1443 /// before casting it to a raw pointer, and that reference is subject to the same rules
1444 /// as all other references. This macro can create a raw pointer *without* creating
1445 /// a reference first.
1446 ///
1447 /// Note, however, that the `expr` in `addr_of!(expr)` is still subject to all
1448 /// the usual rules. In particular, `addr_of!(*ptr::null())` is Undefined
1449 /// Behavior because it dereferences a null pointer.
1450 ///
1451 /// # Example
1452 ///
1453 /// ```
1454 /// use std::ptr;
1455 ///
1456 /// #[repr(packed)]
1457 /// struct Packed {
1458 ///     f1: u8,
1459 ///     f2: u16,
1460 /// }
1461 ///
1462 /// let packed = Packed { f1: 1, f2: 2 };
1463 /// // `&packed.f2` would create an unaligned reference, and thus be Undefined Behavior!
1464 /// let raw_f2 = ptr::addr_of!(packed.f2);
1465 /// assert_eq!(unsafe { raw_f2.read_unaligned() }, 2);
1466 /// ```
1467 ///
1468 /// See [`addr_of_mut`] for how to create a pointer to unininitialized data.
1469 /// Doing that with `addr_of` would not make much sense since one could only
1470 /// read the data, and that would be Undefined Behavior.
1471 #[stable(feature = "raw_ref_macros", since = "1.51.0")]
1472 #[rustc_macro_transparency = "semitransparent"]
1473 #[allow_internal_unstable(raw_ref_op)]
1474 pub macro addr_of($place:expr) {
1475     &raw const $place
1476 }
1477
1478 /// Create a `mut` raw pointer to a place, without creating an intermediate reference.
1479 ///
1480 /// Creating a reference with `&`/`&mut` is only allowed if the pointer is properly aligned
1481 /// and points to initialized data. For cases where those requirements do not hold,
1482 /// raw pointers should be used instead. However, `&mut expr as *mut _` creates a reference
1483 /// before casting it to a raw pointer, and that reference is subject to the same rules
1484 /// as all other references. This macro can create a raw pointer *without* creating
1485 /// a reference first.
1486 ///
1487 /// Note, however, that the `expr` in `addr_of_mut!(expr)` is still subject to all
1488 /// the usual rules. In particular, `addr_of_mut!(*ptr::null_mut())` is Undefined
1489 /// Behavior because it dereferences a null pointer.
1490 ///
1491 /// # Examples
1492 ///
1493 /// **Creating a pointer to unaligned data:**
1494 ///
1495 /// ```
1496 /// use std::ptr;
1497 ///
1498 /// #[repr(packed)]
1499 /// struct Packed {
1500 ///     f1: u8,
1501 ///     f2: u16,
1502 /// }
1503 ///
1504 /// let mut packed = Packed { f1: 1, f2: 2 };
1505 /// // `&mut packed.f2` would create an unaligned reference, and thus be Undefined Behavior!
1506 /// let raw_f2 = ptr::addr_of_mut!(packed.f2);
1507 /// unsafe { raw_f2.write_unaligned(42); }
1508 /// assert_eq!({packed.f2}, 42); // `{...}` forces copying the field instead of creating a reference.
1509 /// ```
1510 ///
1511 /// **Creating a pointer to uninitialized data:**
1512 ///
1513 /// ```rust
1514 /// use std::{ptr, mem::MaybeUninit};
1515 ///
1516 /// struct Demo {
1517 ///     field: bool,
1518 /// }
1519 ///
1520 /// let mut uninit = MaybeUninit::<Demo>::uninit();
1521 /// // `&uninit.as_mut().field` would create a reference to an uninitialized `bool`,
1522 /// // and thus be Undefined Behavior!
1523 /// let f1_ptr = unsafe { ptr::addr_of_mut!((*uninit.as_mut_ptr()).field) };
1524 /// unsafe { f1_ptr.write(true); }
1525 /// let init = unsafe { uninit.assume_init() };
1526 /// ```
1527 #[stable(feature = "raw_ref_macros", since = "1.51.0")]
1528 #[rustc_macro_transparency = "semitransparent"]
1529 #[allow_internal_unstable(raw_ref_op)]
1530 pub macro addr_of_mut($place:expr) {
1531     &raw mut $place
1532 }