]> git.lizzy.rs Git - rust.git/blob - library/core/src/ptr/mod.rs
Auto merge of #2739 - RalfJung:misc, r=RalfJung
[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 //FIXME: mention `ptr::invalid` above, once it is stable.
30 //! * All accesses performed by functions in this module are *non-atomic* in the sense
31 //!   of [atomic operations] used to synchronize between threads. This means it is
32 //!   undefined behavior to perform two concurrent accesses to the same location from different
33 //!   threads unless both accesses only read from memory. Notice that this explicitly
34 //!   includes [`read_volatile`] and [`write_volatile`]: Volatile accesses cannot
35 //!   be used for inter-thread synchronization.
36 //! * The result of casting a reference to a pointer is valid for as long as the
37 //!   underlying object is live and no reference (just raw pointers) is used to
38 //!   access the same memory. That is, reference and pointer accesses cannot be
39 //!   interleaved.
40 //!
41 //! These axioms, along with careful use of [`offset`] for pointer arithmetic,
42 //! are enough to correctly implement many useful things in unsafe code. Stronger guarantees
43 //! will be provided eventually, as the [aliasing] rules are being determined. For more
44 //! information, see the [book] as well as the section in the reference devoted
45 //! to [undefined behavior][ub].
46 //!
47 //! ## Alignment
48 //!
49 //! Valid raw pointers as defined above are not necessarily properly aligned (where
50 //! "proper" alignment is defined by the pointee type, i.e., `*const T` must be
51 //! aligned to `mem::align_of::<T>()`). However, most functions require their
52 //! arguments to be properly aligned, and will explicitly state
53 //! this requirement in their documentation. Notable exceptions to this are
54 //! [`read_unaligned`] and [`write_unaligned`].
55 //!
56 //! When a function requires proper alignment, it does so even if the access
57 //! has size 0, i.e., even if memory is not actually touched. Consider using
58 //! [`NonNull::dangling`] in such cases.
59 //!
60 //! ## Allocated object
61 //!
62 //! For several operations, such as [`offset`] or field projections (`expr.field`), the notion of an
63 //! "allocated object" becomes relevant. An allocated object is a contiguous region of memory.
64 //! Common examples of allocated objects include stack-allocated variables (each variable is a
65 //! separate allocated object), heap allocations (each allocation created by the global allocator is
66 //! a separate allocated object), and `static` variables.
67 //!
68 //! # Strict Provenance
69 //!
70 //! **The following text is non-normative, insufficiently formal, and is an extremely strict
71 //! interpretation of provenance. It's ok if your code doesn't strictly conform to it.**
72 //!
73 //! [Strict Provenance][] is an experimental set of APIs that help tools that try
74 //! to validate the memory-safety of your program's execution. Notably this includes [Miri][]
75 //! and [CHERI][], which can detect when you access out of bounds memory or otherwise violate
76 //! Rust's memory model.
77 //!
78 //! Provenance must exist in some form for any programming
79 //! language compiled for modern computer architectures, but specifying a model for provenance
80 //! in a way that is useful to both compilers and programmers is an ongoing challenge.
81 //! The [Strict Provenance][] experiment seeks to explore the question: *what if we just said you
82 //! couldn't do all the nasty operations that make provenance so messy?*
83 //!
84 //! What APIs would have to be removed? What APIs would have to be added? How much would code
85 //! have to change, and is it worse or better now? Would any patterns become truly inexpressible?
86 //! Could we carve out special exceptions for those patterns? Should we?
87 //!
88 //! A secondary goal of this project is to see if we can disambiguate the many functions of
89 //! pointer<->integer casts enough for the definition of `usize` to be loosened so that it
90 //! isn't *pointer*-sized but address-space/offset/allocation-sized (we'll probably continue
91 //! to conflate these notions). This would potentially make it possible to more efficiently
92 //! target platforms where pointers are larger than offsets, such as CHERI and maybe some
93 //! segmented architectures.
94 //!
95 //! ## Provenance
96 //!
97 //! **This section is *non-normative* and is part of the [Strict Provenance][] experiment.**
98 //!
99 //! Pointers are not *simply* an "integer" or "address". For instance, it's uncontroversial
100 //! to say that a Use After Free is clearly Undefined Behaviour, even if you "get lucky"
101 //! and the freed memory gets reallocated before your read/write (in fact this is the
102 //! worst-case scenario, UAFs would be much less concerning if this didn't happen!).
103 //! To rationalize this claim, pointers need to somehow be *more* than just their addresses:
104 //! they must have provenance.
105 //!
106 //! When an allocation is created, that allocation has a unique Original Pointer. For alloc
107 //! APIs this is literally the pointer the call returns, and for local variables and statics,
108 //! this is the name of the variable/static. This is mildly overloading the term "pointer"
109 //! for the sake of brevity/exposition.
110 //!
111 //! The Original Pointer for an allocation is guaranteed to have unique access to the entire
112 //! allocation and *only* that allocation. In this sense, an allocation can be thought of
113 //! as a "sandbox" that cannot be broken into or out of. *Provenance* is the permission
114 //! to access an allocation's sandbox and has both a *spatial* and *temporal* component:
115 //!
116 //! * Spatial: A range of bytes that the pointer is allowed to access.
117 //! * Temporal: The lifetime (of the allocation) that access to these bytes is tied to.
118 //!
119 //! Spatial provenance makes sure you don't go beyond your sandbox, while temporal provenance
120 //! makes sure that you can't "get lucky" after your permission to access some memory
121 //! has been revoked (either through deallocations or borrows expiring).
122 //!
123 //! Provenance is implicitly shared with all pointers transitively derived from
124 //! The Original Pointer through operations like [`offset`], borrowing, and pointer casts.
125 //! Some operations may *shrink* the derived provenance, limiting how much memory it can
126 //! access or how long it's valid for (i.e. borrowing a subfield and subslicing).
127 //!
128 //! Shrinking provenance cannot be undone: even if you "know" there is a larger allocation, you
129 //! can't derive a pointer with a larger provenance. Similarly, you cannot "recombine"
130 //! two contiguous provenances back into one (i.e. with a `fn merge(&[T], &[T]) -> &[T]`).
131 //!
132 //! A reference to a value always has provenance over exactly the memory that field occupies.
133 //! A reference to a slice always has provenance over exactly the range that slice describes.
134 //!
135 //! If an allocation is deallocated, all pointers with provenance to that allocation become
136 //! invalidated, and effectively lose their provenance.
137 //!
138 //! The strict provenance experiment is mostly only interested in exploring stricter *spatial*
139 //! provenance. In this sense it can be thought of as a subset of the more ambitious and
140 //! formal [Stacked Borrows][] research project, which is what tools like [Miri][] are based on.
141 //! In particular, Stacked Borrows is necessary to properly describe what borrows are allowed
142 //! to do and when they become invalidated. This necessarily involves much more complex
143 //! *temporal* reasoning than simply identifying allocations. Adjusting APIs and code
144 //! for the strict provenance experiment will also greatly help Stacked Borrows.
145 //!
146 //!
147 //! ## Pointer Vs Addresses
148 //!
149 //! **This section is *non-normative* and is part of the [Strict Provenance][] experiment.**
150 //!
151 //! One of the largest historical issues with trying to define provenance is that programmers
152 //! freely convert between pointers and integers. Once you allow for this, it generally becomes
153 //! impossible to accurately track and preserve provenance information, and you need to appeal
154 //! to very complex and unreliable heuristics. But of course, converting between pointers and
155 //! integers is very useful, so what can we do?
156 //!
157 //! Also did you know WASM is actually a "Harvard Architecture"? As in function pointers are
158 //! handled completely differently from data pointers? And we kind of just shipped Rust on WASM
159 //! without really addressing the fact that we let you freely convert between function pointers
160 //! and data pointers, because it mostly Just Works? Let's just put that on the "pointer casts
161 //! are dubious" pile.
162 //!
163 //! Strict Provenance attempts to square these circles by decoupling Rust's traditional conflation
164 //! of pointers and `usize` (and `isize`), and defining a pointer to semantically contain the
165 //! following information:
166 //!
167 //! * The **address-space** it is part of (e.g. "data" vs "code" in WASM).
168 //! * The **address** it points to, which can be represented by a `usize`.
169 //! * The **provenance** it has, defining the memory it has permission to access.
170 //!
171 //! Under Strict Provenance, a usize *cannot* accurately represent a pointer, and converting from
172 //! a pointer to a usize is generally an operation which *only* extracts the address. It is
173 //! therefore *impossible* to construct a valid pointer from a usize because there is no way
174 //! to restore the address-space and provenance. In other words, pointer-integer-pointer
175 //! roundtrips are not possible (in the sense that the resulting pointer is not dereferenceable).
176 //!
177 //! The key insight to making this model *at all* viable is the [`with_addr`][] method:
178 //!
179 //! ```text
180 //!     /// Creates a new pointer with the given address.
181 //!     ///
182 //!     /// This performs the same operation as an `addr as ptr` cast, but copies
183 //!     /// the *address-space* and *provenance* of `self` to the new pointer.
184 //!     /// This allows us to dynamically preserve and propagate this important
185 //!     /// information in a way that is otherwise impossible with a unary cast.
186 //!     ///
187 //!     /// This is equivalent to using `wrapping_offset` to offset `self` to the
188 //!     /// given address, and therefore has all the same capabilities and restrictions.
189 //!     pub fn with_addr(self, addr: usize) -> Self;
190 //! ```
191 //!
192 //! So you're still able to drop down to the address representation and do whatever
193 //! clever bit tricks you want *as long as* you're able to keep around a pointer
194 //! into the allocation you care about that can "reconstitute" the other parts of the pointer.
195 //! Usually this is very easy, because you only are taking a pointer, messing with the address,
196 //! and then immediately converting back to a pointer. To make this use case more ergonomic,
197 //! we provide the [`map_addr`][] method.
198 //!
199 //! To help make it clear that code is "following" Strict Provenance semantics, we also provide an
200 //! [`addr`][] method which promises that the returned address is not part of a
201 //! pointer-usize-pointer roundtrip. In the future we may provide a lint for pointer<->integer
202 //! casts to help you audit if your code conforms to strict provenance.
203 //!
204 //!
205 //! ## Using Strict Provenance
206 //!
207 //! Most code needs no changes to conform to strict provenance, as the only really concerning
208 //! operation that *wasn't* obviously already Undefined Behaviour is casts from usize to a
209 //! pointer. For code which *does* cast a usize to a pointer, the scope of the change depends
210 //! on exactly what you're doing.
211 //!
212 //! In general you just need to make sure that if you want to convert a usize address to a
213 //! pointer and then use that pointer to read/write memory, you need to keep around a pointer
214 //! that has sufficient provenance to perform that read/write itself. In this way all of your
215 //! casts from an address to a pointer are essentially just applying offsets/indexing.
216 //!
217 //! This is generally trivial to do for simple cases like tagged pointers *as long as you
218 //! represent the tagged pointer as an actual pointer and not a usize*. For instance:
219 //!
220 //! ```
221 //! #![feature(strict_provenance)]
222 //!
223 //! unsafe {
224 //!     // A flag we want to pack into our pointer
225 //!     static HAS_DATA: usize = 0x1;
226 //!     static FLAG_MASK: usize = !HAS_DATA;
227 //!
228 //!     // Our value, which must have enough alignment to have spare least-significant-bits.
229 //!     let my_precious_data: u32 = 17;
230 //!     assert!(core::mem::align_of::<u32>() > 1);
231 //!
232 //!     // Create a tagged pointer
233 //!     let ptr = &my_precious_data as *const u32;
234 //!     let tagged = ptr.map_addr(|addr| addr | HAS_DATA);
235 //!
236 //!     // Check the flag:
237 //!     if tagged.addr() & HAS_DATA != 0 {
238 //!         // Untag and read the pointer
239 //!         let data = *tagged.map_addr(|addr| addr & FLAG_MASK);
240 //!         assert_eq!(data, 17);
241 //!     } else {
242 //!         unreachable!()
243 //!     }
244 //! }
245 //! ```
246 //!
247 //! (Yes, if you've been using AtomicUsize for pointers in concurrent datastructures, you should
248 //! be using AtomicPtr instead. If that messes up the way you atomically manipulate pointers,
249 //! we would like to know why, and what needs to be done to fix it.)
250 //!
251 //! Something more complicated and just generally *evil* like an XOR-List requires more significant
252 //! changes like allocating all nodes in a pre-allocated Vec or Arena and using a pointer
253 //! to the whole allocation to reconstitute the XORed addresses.
254 //!
255 //! Situations where a valid pointer *must* be created from just an address, such as baremetal code
256 //! accessing a memory-mapped interface at a fixed address, are an open question on how to support.
257 //! These situations *will* still be allowed, but we might require some kind of "I know what I'm
258 //! doing" annotation to explain the situation to the compiler. It's also possible they need no
259 //! special attention at all, because they're generally accessing memory outside the scope of
260 //! "the abstract machine", or already using "I know what I'm doing" annotations like "volatile".
261 //!
262 //! Under [Strict Provenance] it is Undefined Behaviour to:
263 //!
264 //! * Access memory through a pointer that does not have provenance over that memory.
265 //!
266 //! * [`offset`] a pointer to or from an address it doesn't have provenance over.
267 //!   This means it's always UB to offset a pointer derived from something deallocated,
268 //!   even if the offset is 0. Note that a pointer "one past the end" of its provenance
269 //!   is not actually outside its provenance, it just has 0 bytes it can load/store.
270 //!
271 //! But it *is* still sound to:
272 //!
273 //! * Create an invalid pointer from just an address (see [`ptr::invalid`][]). This can
274 //!   be used for sentinel values like `null` *or* to represent a tagged pointer that will
275 //!   never be dereferenceable. In general, it is always sound for an integer to pretend
276 //!   to be a pointer "for fun" as long as you don't use operations on it which require
277 //!   it to be valid (offset, read, write, etc).
278 //!
279 //! * Forge an allocation of size zero at any sufficiently aligned non-null address.
280 //!   i.e. the usual "ZSTs are fake, do what you want" rules apply *but* this only applies
281 //!   for actual forgery (integers cast to pointers). If you borrow some struct's field
282 //!   that *happens* to be zero-sized, the resulting pointer will have provenance tied to
283 //!   that allocation and it will still get invalidated if the allocation gets deallocated.
284 //!   In the future we may introduce an API to make such a forged allocation explicit.
285 //!
286 //! * [`wrapping_offset`][] a pointer outside its provenance. This includes invalid pointers
287 //!   which have "no" provenance. Unfortunately there may be practical limits on this for a
288 //!   particular platform, and it's an open question as to how to specify this (if at all).
289 //!   Notably, [CHERI][] relies on a compression scheme that can't handle a
290 //!   pointer getting offset "too far" out of bounds. If this happens, the address
291 //!   returned by `addr` will be the value you expect, but the provenance will get invalidated
292 //!   and using it to read/write will fault. The details of this are architecture-specific
293 //!   and based on alignment, but the buffer on either side of the pointer's range is pretty
294 //!   generous (think kilobytes, not bytes).
295 //!
296 //! * Compare arbitrary pointers by address. Addresses *are* just integers and so there is
297 //!   always a coherent answer, even if the pointers are invalid or from different
298 //!   address-spaces/provenances. Of course, comparing addresses from different address-spaces
299 //!   is generally going to be *meaningless*, but so is comparing Kilograms to Meters, and Rust
300 //!   doesn't prevent that either. Similarly, if you get "lucky" and notice that a pointer
301 //!   one-past-the-end is the "same" address as the start of an unrelated allocation, anything
302 //!   you do with that fact is *probably* going to be gibberish. The scope of that gibberish
303 //!   is kept under control by the fact that the two pointers *still* aren't allowed to access
304 //!   the other's allocation (bytes), because they still have different provenance.
305 //!
306 //! * Perform pointer tagging tricks. This falls out of [`wrapping_offset`] but is worth
307 //!   mentioning in more detail because of the limitations of [CHERI][]. Low-bit tagging
308 //!   is very robust, and often doesn't even go out of bounds because types ensure
309 //!   size >= align (and over-aligning actually gives CHERI more flexibility). Anything
310 //!   more complex than this rapidly enters "extremely platform-specific" territory as
311 //!   certain things may or may not be allowed based on specific supported operations.
312 //!   For instance, ARM explicitly supports high-bit tagging, and so CHERI on ARM inherits
313 //!   that and should support it.
314 //!
315 //! ## Pointer-usize-pointer roundtrips and 'exposed' provenance
316 //!
317 //! **This section is *non-normative* and is part of the [Strict Provenance] experiment.**
318 //!
319 //! As discussed above, pointer-usize-pointer roundtrips are not possible under [Strict Provenance].
320 //! However, there exists legacy Rust code that is full of such roundtrips, and legacy platform APIs
321 //! regularly assume that `usize` can capture all the information that makes up a pointer. There
322 //! also might be code that cannot be ported to Strict Provenance (which is something we would [like
323 //! to hear about][Strict Provenance]).
324 //!
325 //! For situations like this, there is a fallback plan, a way to 'opt out' of Strict Provenance.
326 //! However, note that this makes your code a lot harder to specify, and the code will not work
327 //! (well) with tools like [Miri] and [CHERI].
328 //!
329 //! This fallback plan is provided by the [`expose_addr`] and [`from_exposed_addr`] methods (which
330 //! are equivalent to `as` casts between pointers and integers). [`expose_addr`] is a lot like
331 //! [`addr`], but additionally adds the provenance of the pointer to a global list of 'exposed'
332 //! provenances. (This list is purely conceptual, it exists for the purpose of specifying Rust but
333 //! is not materialized in actual executions, except in tools like [Miri].) [`from_exposed_addr`]
334 //! can be used to construct a pointer with one of these previously 'exposed' provenances.
335 //! [`from_exposed_addr`] takes only `addr: usize` as arguments, so unlike in [`with_addr`] there is
336 //! no indication of what the correct provenance for the returned pointer is -- and that is exactly
337 //! what makes pointer-usize-pointer roundtrips so tricky to rigorously specify! There is no
338 //! algorithm that decides which provenance will be used. You can think of this as "guessing" the
339 //! right provenance, and the guess will be "maximally in your favor", in the sense that if there is
340 //! any way to avoid undefined behavior, then that is the guess that will be taken. However, if
341 //! there is *no* previously 'exposed' provenance that justifies the way the returned pointer will
342 //! be used, the program has undefined behavior.
343 //!
344 //! Using [`expose_addr`] or [`from_exposed_addr`] (or the equivalent `as` casts) means that code is
345 //! *not* following Strict Provenance rules. The goal of the Strict Provenance experiment is to
346 //! determine whether it is possible to use Rust without [`expose_addr`] and [`from_exposed_addr`].
347 //! If this is successful, it would be a major win for avoiding specification complexity and to
348 //! facilitate adoption of tools like [CHERI] and [Miri] that can be a big help in increasing the
349 //! confidence in (unsafe) Rust code.
350 //!
351 //! [aliasing]: ../../nomicon/aliasing.html
352 //! [book]: ../../book/ch19-01-unsafe-rust.html#dereferencing-a-raw-pointer
353 //! [ub]: ../../reference/behavior-considered-undefined.html
354 //! [zst]: ../../nomicon/exotic-sizes.html#zero-sized-types-zsts
355 //! [atomic operations]: crate::sync::atomic
356 //! [`offset`]: pointer::offset
357 //! [`wrapping_offset`]: pointer::wrapping_offset
358 //! [`with_addr`]: pointer::with_addr
359 //! [`map_addr`]: pointer::map_addr
360 //! [`addr`]: pointer::addr
361 //! [`ptr::invalid`]: core::ptr::invalid
362 //! [`expose_addr`]: pointer::expose_addr
363 //! [`from_exposed_addr`]: from_exposed_addr
364 //! [Miri]: https://github.com/rust-lang/miri
365 //! [CHERI]: https://www.cl.cam.ac.uk/research/security/ctsrd/cheri/
366 //! [Strict Provenance]: https://github.com/rust-lang/rust/issues/95228
367 //! [Stacked Borrows]: https://plv.mpi-sws.org/rustbelt/stacked-borrows/
368
369 #![stable(feature = "rust1", since = "1.0.0")]
370
371 use crate::cmp::Ordering;
372 use crate::fmt;
373 use crate::hash;
374 use crate::intrinsics::{
375     self, assert_unsafe_precondition, is_aligned_and_not_null, is_nonoverlapping,
376 };
377
378 use crate::mem::{self, MaybeUninit};
379
380 mod alignment;
381 #[unstable(feature = "ptr_alignment_type", issue = "102070")]
382 pub use alignment::Alignment;
383
384 #[stable(feature = "rust1", since = "1.0.0")]
385 #[doc(inline)]
386 pub use crate::intrinsics::copy_nonoverlapping;
387
388 #[stable(feature = "rust1", since = "1.0.0")]
389 #[doc(inline)]
390 pub use crate::intrinsics::copy;
391
392 #[stable(feature = "rust1", since = "1.0.0")]
393 #[doc(inline)]
394 pub use crate::intrinsics::write_bytes;
395
396 mod metadata;
397 #[unstable(feature = "ptr_metadata", issue = "81513")]
398 pub use metadata::{from_raw_parts, from_raw_parts_mut, metadata, DynMetadata, Pointee, Thin};
399
400 mod non_null;
401 #[stable(feature = "nonnull", since = "1.25.0")]
402 pub use non_null::NonNull;
403
404 mod unique;
405 #[unstable(feature = "ptr_internals", issue = "none")]
406 pub use unique::Unique;
407
408 mod const_ptr;
409 mod mut_ptr;
410
411 /// Executes the destructor (if any) of the pointed-to value.
412 ///
413 /// This is semantically equivalent to calling [`ptr::read`] and discarding
414 /// the result, but has the following advantages:
415 ///
416 /// * It is *required* to use `drop_in_place` to drop unsized types like
417 ///   trait objects, because they can't be read out onto the stack and
418 ///   dropped normally.
419 ///
420 /// * It is friendlier to the optimizer to do this over [`ptr::read`] when
421 ///   dropping manually allocated memory (e.g., in the implementations of
422 ///   `Box`/`Rc`/`Vec`), as the compiler doesn't need to prove that it's
423 ///   sound to elide the copy.
424 ///
425 /// * It can be used to drop [pinned] data when `T` is not `repr(packed)`
426 ///   (pinned data must not be moved before it is dropped).
427 ///
428 /// Unaligned values cannot be dropped in place, they must be copied to an aligned
429 /// location first using [`ptr::read_unaligned`]. For packed structs, this move is
430 /// done automatically by the compiler. This means the fields of packed structs
431 /// are not dropped in-place.
432 ///
433 /// [`ptr::read`]: self::read
434 /// [`ptr::read_unaligned`]: self::read_unaligned
435 /// [pinned]: crate::pin
436 ///
437 /// # Safety
438 ///
439 /// Behavior is undefined if any of the following conditions are violated:
440 ///
441 /// * `to_drop` must be [valid] for both reads and writes.
442 ///
443 /// * `to_drop` must be properly aligned.
444 ///
445 /// * The value `to_drop` points to must be valid for dropping, which may mean it must uphold
446 ///   additional invariants - this is type-dependent.
447 ///
448 /// Additionally, if `T` is not [`Copy`], using the pointed-to value after
449 /// calling `drop_in_place` can cause undefined behavior. Note that `*to_drop =
450 /// foo` counts as a use because it will cause the value to be dropped
451 /// again. [`write()`] can be used to overwrite data without causing it to be
452 /// dropped.
453 ///
454 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
455 ///
456 /// [valid]: self#safety
457 ///
458 /// # Examples
459 ///
460 /// Manually remove the last item from a vector:
461 ///
462 /// ```
463 /// use std::ptr;
464 /// use std::rc::Rc;
465 ///
466 /// let last = Rc::new(1);
467 /// let weak = Rc::downgrade(&last);
468 ///
469 /// let mut v = vec![Rc::new(0), last];
470 ///
471 /// unsafe {
472 ///     // Get a raw pointer to the last element in `v`.
473 ///     let ptr = &mut v[1] as *mut _;
474 ///     // Shorten `v` to prevent the last item from being dropped. We do that first,
475 ///     // to prevent issues if the `drop_in_place` below panics.
476 ///     v.set_len(1);
477 ///     // Without a call `drop_in_place`, the last item would never be dropped,
478 ///     // and the memory it manages would be leaked.
479 ///     ptr::drop_in_place(ptr);
480 /// }
481 ///
482 /// assert_eq!(v, &[0.into()]);
483 ///
484 /// // Ensure that the last item was dropped.
485 /// assert!(weak.upgrade().is_none());
486 /// ```
487 #[stable(feature = "drop_in_place", since = "1.8.0")]
488 #[lang = "drop_in_place"]
489 #[allow(unconditional_recursion)]
490 pub unsafe fn drop_in_place<T: ?Sized>(to_drop: *mut T) {
491     // Code here does not matter - this is replaced by the
492     // real drop glue by the compiler.
493
494     // SAFETY: see comment above
495     unsafe { drop_in_place(to_drop) }
496 }
497
498 /// Creates a null raw pointer.
499 ///
500 /// # Examples
501 ///
502 /// ```
503 /// use std::ptr;
504 ///
505 /// let p: *const i32 = ptr::null();
506 /// assert!(p.is_null());
507 /// ```
508 #[inline(always)]
509 #[must_use]
510 #[stable(feature = "rust1", since = "1.0.0")]
511 #[rustc_promotable]
512 #[rustc_const_stable(feature = "const_ptr_null", since = "1.24.0")]
513 #[rustc_allow_const_fn_unstable(ptr_metadata)]
514 #[rustc_diagnostic_item = "ptr_null"]
515 pub const fn null<T: ?Sized + Thin>() -> *const T {
516     from_raw_parts(invalid(0), ())
517 }
518
519 /// Creates an invalid pointer with the given address.
520 ///
521 /// This is different from `addr as *const T`, which creates a pointer that picks up a previously
522 /// exposed provenance. See [`from_exposed_addr`] for more details on that operation.
523 ///
524 /// The module's top-level documentation discusses the precise meaning of an "invalid"
525 /// pointer but essentially this expresses that the pointer is not associated
526 /// with any actual allocation and is little more than a usize address in disguise.
527 ///
528 /// This pointer will have no provenance associated with it and is therefore
529 /// UB to read/write/offset. This mostly exists to facilitate things
530 /// like `ptr::null` and `NonNull::dangling` which make invalid pointers.
531 ///
532 /// (Standard "Zero-Sized-Types get to cheat and lie" caveats apply, although it
533 /// may be desirable to give them their own API just to make that 100% clear.)
534 ///
535 /// This API and its claimed semantics are part of the Strict Provenance experiment,
536 /// see the [module documentation][crate::ptr] for details.
537 #[inline(always)]
538 #[must_use]
539 #[rustc_const_stable(feature = "stable_things_using_strict_provenance", since = "1.61.0")]
540 #[unstable(feature = "strict_provenance", issue = "95228")]
541 pub const fn invalid<T>(addr: usize) -> *const T {
542     // FIXME(strict_provenance_magic): I am magic and should be a compiler intrinsic.
543     // We use transmute rather than a cast so tools like Miri can tell that this
544     // is *not* the same as from_exposed_addr.
545     // SAFETY: every valid integer is also a valid pointer (as long as you don't dereference that
546     // pointer).
547     unsafe { mem::transmute(addr) }
548 }
549
550 /// Creates an invalid mutable pointer with the given address.
551 ///
552 /// This is different from `addr as *mut T`, which creates a pointer that picks up a previously
553 /// exposed provenance. See [`from_exposed_addr_mut`] for more details on that operation.
554 ///
555 /// The module's top-level documentation discusses the precise meaning of an "invalid"
556 /// pointer but essentially this expresses that the pointer is not associated
557 /// with any actual allocation and is little more than a usize address in disguise.
558 ///
559 /// This pointer will have no provenance associated with it and is therefore
560 /// UB to read/write/offset. This mostly exists to facilitate things
561 /// like `ptr::null` and `NonNull::dangling` which make invalid pointers.
562 ///
563 /// (Standard "Zero-Sized-Types get to cheat and lie" caveats apply, although it
564 /// may be desirable to give them their own API just to make that 100% clear.)
565 ///
566 /// This API and its claimed semantics are part of the Strict Provenance experiment,
567 /// see the [module documentation][crate::ptr] for details.
568 #[inline(always)]
569 #[must_use]
570 #[rustc_const_stable(feature = "stable_things_using_strict_provenance", since = "1.61.0")]
571 #[unstable(feature = "strict_provenance", issue = "95228")]
572 pub const fn invalid_mut<T>(addr: usize) -> *mut T {
573     // FIXME(strict_provenance_magic): I am magic and should be a compiler intrinsic.
574     // We use transmute rather than a cast so tools like Miri can tell that this
575     // is *not* the same as from_exposed_addr.
576     // SAFETY: every valid integer is also a valid pointer (as long as you don't dereference that
577     // pointer).
578     unsafe { mem::transmute(addr) }
579 }
580
581 /// Convert an address back to a pointer, picking up a previously 'exposed' provenance.
582 ///
583 /// This is equivalent to `addr as *const T`. The provenance of the returned pointer is that of *any*
584 /// pointer that was previously exposed by passing it to [`expose_addr`][pointer::expose_addr],
585 /// or a `ptr as usize` cast. In addition, memory which is outside the control of the Rust abstract
586 /// machine (MMIO registers, for example) is always considered to be exposed, so long as this memory
587 /// is disjoint from memory that will be used by the abstract machine such as the stack, heap,
588 /// and statics.
589 ///
590 /// If there is no 'exposed' provenance that justifies the way this pointer will be used,
591 /// the program has undefined behavior. In particular, the aliasing rules still apply: pointers
592 /// and references that have been invalidated due to aliasing accesses cannot be used any more,
593 /// even if they have been exposed!
594 ///
595 /// Note that there is no algorithm that decides which provenance will be used. You can think of this
596 /// as "guessing" the right provenance, and the guess will be "maximally in your favor", in the sense
597 /// that if there is any way to avoid undefined behavior (while upholding all aliasing requirements),
598 /// then that is the guess that will be taken.
599 ///
600 /// On platforms with multiple address spaces, it is your responsibility to ensure that the
601 /// address makes sense in the address space that this pointer will be used with.
602 ///
603 /// Using this method means that code is *not* following strict provenance rules. "Guessing" a
604 /// suitable provenance complicates specification and reasoning and may not be supported by
605 /// tools that help you to stay conformant with the Rust memory model, so it is recommended to
606 /// use [`with_addr`][pointer::with_addr] wherever possible.
607 ///
608 /// On most platforms this will produce a value with the same bytes as the address. Platforms
609 /// which need to store additional information in a pointer may not support this operation,
610 /// since it is generally not possible to actually *compute* which provenance the returned
611 /// pointer has to pick up.
612 ///
613 /// This API and its claimed semantics are part of the Strict Provenance experiment, see the
614 /// [module documentation][crate::ptr] for details.
615 #[must_use]
616 #[inline(always)]
617 #[unstable(feature = "strict_provenance", issue = "95228")]
618 #[cfg_attr(miri, track_caller)] // even without panics, this helps for Miri backtraces
619 #[allow(fuzzy_provenance_casts)] // this *is* the strict provenance API one should use instead
620 pub fn from_exposed_addr<T>(addr: usize) -> *const T
621 where
622     T: Sized,
623 {
624     // FIXME(strict_provenance_magic): I am magic and should be a compiler intrinsic.
625     addr as *const T
626 }
627
628 /// Convert an address back to a mutable pointer, picking up a previously 'exposed' provenance.
629 ///
630 /// This is equivalent to `addr as *mut T`. The provenance of the returned pointer is that of *any*
631 /// pointer that was previously passed to [`expose_addr`][pointer::expose_addr] or a `ptr as usize`
632 /// cast. If there is no previously 'exposed' provenance that justifies the way this pointer will be
633 /// used, the program has undefined behavior. Note that there is no algorithm that decides which
634 /// provenance will be used. You can think of this as "guessing" the right provenance, and the guess
635 /// will be "maximally in your favor", in the sense that if there is any way to avoid undefined
636 /// behavior, then that is the guess that will be taken.
637 ///
638 /// On platforms with multiple address spaces, it is your responsibility to ensure that the
639 /// address makes sense in the address space that this pointer will be used with.
640 ///
641 /// Using this method means that code is *not* following strict provenance rules. "Guessing" a
642 /// suitable provenance complicates specification and reasoning and may not be supported by
643 /// tools that help you to stay conformant with the Rust memory model, so it is recommended to
644 /// use [`with_addr`][pointer::with_addr] wherever possible.
645 ///
646 /// On most platforms this will produce a value with the same bytes as the address. Platforms
647 /// which need to store additional information in a pointer may not support this operation,
648 /// since it is generally not possible to actually *compute* which provenance the returned
649 /// pointer has to pick up.
650 ///
651 /// This API and its claimed semantics are part of the Strict Provenance experiment, see the
652 /// [module documentation][crate::ptr] for details.
653 #[must_use]
654 #[inline(always)]
655 #[unstable(feature = "strict_provenance", issue = "95228")]
656 #[cfg_attr(miri, track_caller)] // even without panics, this helps for Miri backtraces
657 #[allow(fuzzy_provenance_casts)] // this *is* the strict provenance API one should use instead
658 pub fn from_exposed_addr_mut<T>(addr: usize) -> *mut T
659 where
660     T: Sized,
661 {
662     // FIXME(strict_provenance_magic): I am magic and should be a compiler intrinsic.
663     addr as *mut T
664 }
665
666 /// Creates a null mutable raw pointer.
667 ///
668 /// # Examples
669 ///
670 /// ```
671 /// use std::ptr;
672 ///
673 /// let p: *mut i32 = ptr::null_mut();
674 /// assert!(p.is_null());
675 /// ```
676 #[inline(always)]
677 #[must_use]
678 #[stable(feature = "rust1", since = "1.0.0")]
679 #[rustc_promotable]
680 #[rustc_const_stable(feature = "const_ptr_null", since = "1.24.0")]
681 #[rustc_allow_const_fn_unstable(ptr_metadata)]
682 #[rustc_diagnostic_item = "ptr_null_mut"]
683 pub const fn null_mut<T: ?Sized + Thin>() -> *mut T {
684     from_raw_parts_mut(invalid_mut(0), ())
685 }
686
687 /// Forms a raw slice from a pointer and a length.
688 ///
689 /// The `len` argument is the number of **elements**, not the number of bytes.
690 ///
691 /// This function is safe, but actually using the return value is unsafe.
692 /// See the documentation of [`slice::from_raw_parts`] for slice safety requirements.
693 ///
694 /// [`slice::from_raw_parts`]: crate::slice::from_raw_parts
695 ///
696 /// # Examples
697 ///
698 /// ```rust
699 /// use std::ptr;
700 ///
701 /// // create a slice pointer when starting out with a pointer to the first element
702 /// let x = [5, 6, 7];
703 /// let raw_pointer = x.as_ptr();
704 /// let slice = ptr::slice_from_raw_parts(raw_pointer, 3);
705 /// assert_eq!(unsafe { &*slice }[2], 7);
706 /// ```
707 #[inline]
708 #[stable(feature = "slice_from_raw_parts", since = "1.42.0")]
709 #[rustc_const_stable(feature = "const_slice_from_raw_parts", since = "1.64.0")]
710 #[rustc_allow_const_fn_unstable(ptr_metadata)]
711 pub const fn slice_from_raw_parts<T>(data: *const T, len: usize) -> *const [T] {
712     from_raw_parts(data.cast(), len)
713 }
714
715 /// Performs the same functionality as [`slice_from_raw_parts`], except that a
716 /// raw mutable slice is returned, as opposed to a raw immutable slice.
717 ///
718 /// See the documentation of [`slice_from_raw_parts`] for more details.
719 ///
720 /// This function is safe, but actually using the return value is unsafe.
721 /// See the documentation of [`slice::from_raw_parts_mut`] for slice safety requirements.
722 ///
723 /// [`slice::from_raw_parts_mut`]: crate::slice::from_raw_parts_mut
724 ///
725 /// # Examples
726 ///
727 /// ```rust
728 /// use std::ptr;
729 ///
730 /// let x = &mut [5, 6, 7];
731 /// let raw_pointer = x.as_mut_ptr();
732 /// let slice = ptr::slice_from_raw_parts_mut(raw_pointer, 3);
733 ///
734 /// unsafe {
735 ///     (*slice)[2] = 99; // assign a value at an index in the slice
736 /// };
737 ///
738 /// assert_eq!(unsafe { &*slice }[2], 99);
739 /// ```
740 #[inline]
741 #[stable(feature = "slice_from_raw_parts", since = "1.42.0")]
742 #[rustc_const_unstable(feature = "const_slice_from_raw_parts_mut", issue = "67456")]
743 pub const fn slice_from_raw_parts_mut<T>(data: *mut T, len: usize) -> *mut [T] {
744     from_raw_parts_mut(data.cast(), len)
745 }
746
747 /// Swaps the values at two mutable locations of the same type, without
748 /// deinitializing either.
749 ///
750 /// But for the following exceptions, this function is semantically
751 /// equivalent to [`mem::swap`]:
752 ///
753 /// * It operates on raw pointers instead of references. When references are
754 ///   available, [`mem::swap`] should be preferred.
755 ///
756 /// * The two pointed-to values may overlap. If the values do overlap, then the
757 ///   overlapping region of memory from `x` will be used. This is demonstrated
758 ///   in the second example below.
759 ///
760 /// * The operation is "untyped" in the sense that data may be uninitialized or otherwise violate
761 ///   the requirements of `T`. The initialization state is preserved exactly.
762 ///
763 /// # Safety
764 ///
765 /// Behavior is undefined if any of the following conditions are violated:
766 ///
767 /// * Both `x` and `y` must be [valid] for both reads and writes.
768 ///
769 /// * Both `x` and `y` must be properly aligned.
770 ///
771 /// Note that even if `T` has size `0`, the pointers must be non-null and properly aligned.
772 ///
773 /// [valid]: self#safety
774 ///
775 /// # Examples
776 ///
777 /// Swapping two non-overlapping regions:
778 ///
779 /// ```
780 /// use std::ptr;
781 ///
782 /// let mut array = [0, 1, 2, 3];
783 ///
784 /// let (x, y) = array.split_at_mut(2);
785 /// let x = x.as_mut_ptr().cast::<[u32; 2]>(); // this is `array[0..2]`
786 /// let y = y.as_mut_ptr().cast::<[u32; 2]>(); // this is `array[2..4]`
787 ///
788 /// unsafe {
789 ///     ptr::swap(x, y);
790 ///     assert_eq!([2, 3, 0, 1], array);
791 /// }
792 /// ```
793 ///
794 /// Swapping two overlapping regions:
795 ///
796 /// ```
797 /// use std::ptr;
798 ///
799 /// let mut array: [i32; 4] = [0, 1, 2, 3];
800 ///
801 /// let array_ptr: *mut i32 = array.as_mut_ptr();
802 ///
803 /// let x = array_ptr as *mut [i32; 3]; // this is `array[0..3]`
804 /// let y = unsafe { array_ptr.add(1) } as *mut [i32; 3]; // this is `array[1..4]`
805 ///
806 /// unsafe {
807 ///     ptr::swap(x, y);
808 ///     // The indices `1..3` of the slice overlap between `x` and `y`.
809 ///     // Reasonable results would be for to them be `[2, 3]`, so that indices `0..3` are
810 ///     // `[1, 2, 3]` (matching `y` before the `swap`); or for them to be `[0, 1]`
811 ///     // so that indices `1..4` are `[0, 1, 2]` (matching `x` before the `swap`).
812 ///     // This implementation is defined to make the latter choice.
813 ///     assert_eq!([1, 0, 1, 2], array);
814 /// }
815 /// ```
816 #[inline]
817 #[stable(feature = "rust1", since = "1.0.0")]
818 #[rustc_const_unstable(feature = "const_swap", issue = "83163")]
819 pub const unsafe fn swap<T>(x: *mut T, y: *mut T) {
820     // Give ourselves some scratch space to work with.
821     // We do not have to worry about drops: `MaybeUninit` does nothing when dropped.
822     let mut tmp = MaybeUninit::<T>::uninit();
823
824     // Perform the swap
825     // SAFETY: the caller must guarantee that `x` and `y` are
826     // valid for writes and properly aligned. `tmp` cannot be
827     // overlapping either `x` or `y` because `tmp` was just allocated
828     // on the stack as a separate allocated object.
829     unsafe {
830         copy_nonoverlapping(x, tmp.as_mut_ptr(), 1);
831         copy(y, x, 1); // `x` and `y` may overlap
832         copy_nonoverlapping(tmp.as_ptr(), y, 1);
833     }
834 }
835
836 /// Swaps `count * size_of::<T>()` bytes between the two regions of memory
837 /// beginning at `x` and `y`. The two regions must *not* overlap.
838 ///
839 /// The operation is "untyped" in the sense that data may be uninitialized or otherwise violate the
840 /// requirements of `T`. The initialization state is preserved exactly.
841 ///
842 /// # Safety
843 ///
844 /// Behavior is undefined if any of the following conditions are violated:
845 ///
846 /// * Both `x` and `y` must be [valid] for both reads and writes of `count *
847 ///   size_of::<T>()` bytes.
848 ///
849 /// * Both `x` and `y` must be properly aligned.
850 ///
851 /// * The region of memory beginning at `x` with a size of `count *
852 ///   size_of::<T>()` bytes must *not* overlap with the region of memory
853 ///   beginning at `y` with the same size.
854 ///
855 /// Note that even if the effectively copied size (`count * size_of::<T>()`) is `0`,
856 /// the pointers must be non-null and properly aligned.
857 ///
858 /// [valid]: self#safety
859 ///
860 /// # Examples
861 ///
862 /// Basic usage:
863 ///
864 /// ```
865 /// use std::ptr;
866 ///
867 /// let mut x = [1, 2, 3, 4];
868 /// let mut y = [7, 8, 9];
869 ///
870 /// unsafe {
871 ///     ptr::swap_nonoverlapping(x.as_mut_ptr(), y.as_mut_ptr(), 2);
872 /// }
873 ///
874 /// assert_eq!(x, [7, 8, 3, 4]);
875 /// assert_eq!(y, [1, 2, 9]);
876 /// ```
877 #[inline]
878 #[stable(feature = "swap_nonoverlapping", since = "1.27.0")]
879 #[rustc_const_unstable(feature = "const_swap", issue = "83163")]
880 pub const unsafe fn swap_nonoverlapping<T>(x: *mut T, y: *mut T, count: usize) {
881     #[allow(unused)]
882     macro_rules! attempt_swap_as_chunks {
883         ($ChunkTy:ty) => {
884             if mem::align_of::<T>() >= mem::align_of::<$ChunkTy>()
885                 && mem::size_of::<T>() % mem::size_of::<$ChunkTy>() == 0
886             {
887                 let x: *mut $ChunkTy = x.cast();
888                 let y: *mut $ChunkTy = y.cast();
889                 let count = count * (mem::size_of::<T>() / mem::size_of::<$ChunkTy>());
890                 // SAFETY: these are the same bytes that the caller promised were
891                 // ok, just typed as `MaybeUninit<ChunkTy>`s instead of as `T`s.
892                 // The `if` condition above ensures that we're not violating
893                 // alignment requirements, and that the division is exact so
894                 // that we don't lose any bytes off the end.
895                 return unsafe { swap_nonoverlapping_simple_untyped(x, y, count) };
896             }
897         };
898     }
899
900     // SAFETY: the caller must guarantee that `x` and `y` are
901     // valid for writes and properly aligned.
902     unsafe {
903         assert_unsafe_precondition!(
904             "ptr::swap_nonoverlapping requires that both pointer arguments are aligned and non-null \
905             and the specified memory ranges do not overlap",
906             [T](x: *mut T, y: *mut T, count: usize) =>
907             is_aligned_and_not_null(x)
908                 && is_aligned_and_not_null(y)
909                 && is_nonoverlapping(x, y, count)
910         );
911     }
912
913     // Split up the slice into small power-of-two-sized chunks that LLVM is able
914     // to vectorize (unless it's a special type with more-than-pointer alignment,
915     // because we don't want to pessimize things like slices of SIMD vectors.)
916     if mem::align_of::<T>() <= mem::size_of::<usize>()
917         && (!mem::size_of::<T>().is_power_of_two()
918             || mem::size_of::<T>() > mem::size_of::<usize>() * 2)
919     {
920         attempt_swap_as_chunks!(usize);
921         attempt_swap_as_chunks!(u8);
922     }
923
924     // SAFETY: Same preconditions as this function
925     unsafe { swap_nonoverlapping_simple_untyped(x, y, count) }
926 }
927
928 /// Same behaviour and safety conditions as [`swap_nonoverlapping`]
929 ///
930 /// LLVM can vectorize this (at least it can for the power-of-two-sized types
931 /// `swap_nonoverlapping` tries to use) so no need to manually SIMD it.
932 #[inline]
933 #[rustc_const_unstable(feature = "const_swap", issue = "83163")]
934 const unsafe fn swap_nonoverlapping_simple_untyped<T>(x: *mut T, y: *mut T, count: usize) {
935     let x = x.cast::<MaybeUninit<T>>();
936     let y = y.cast::<MaybeUninit<T>>();
937     let mut i = 0;
938     while i < count {
939         // SAFETY: By precondition, `i` is in-bounds because it's below `n`
940         let x = unsafe { &mut *x.add(i) };
941         // SAFETY: By precondition, `i` is in-bounds because it's below `n`
942         // and it's distinct from `x` since the ranges are non-overlapping
943         let y = unsafe { &mut *y.add(i) };
944         mem::swap_simple::<MaybeUninit<T>>(x, y);
945
946         i += 1;
947     }
948 }
949
950 /// Moves `src` into the pointed `dst`, returning the previous `dst` value.
951 ///
952 /// Neither value is dropped.
953 ///
954 /// This function is semantically equivalent to [`mem::replace`] except that it
955 /// operates on raw pointers instead of references. When references are
956 /// available, [`mem::replace`] should be preferred.
957 ///
958 /// # Safety
959 ///
960 /// Behavior is undefined if any of the following conditions are violated:
961 ///
962 /// * `dst` must be [valid] for both reads and writes.
963 ///
964 /// * `dst` must be properly aligned.
965 ///
966 /// * `dst` must point to a properly initialized value of type `T`.
967 ///
968 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
969 ///
970 /// [valid]: self#safety
971 ///
972 /// # Examples
973 ///
974 /// ```
975 /// use std::ptr;
976 ///
977 /// let mut rust = vec!['b', 'u', 's', 't'];
978 ///
979 /// // `mem::replace` would have the same effect without requiring the unsafe
980 /// // block.
981 /// let b = unsafe {
982 ///     ptr::replace(&mut rust[0], 'r')
983 /// };
984 ///
985 /// assert_eq!(b, 'b');
986 /// assert_eq!(rust, &['r', 'u', 's', 't']);
987 /// ```
988 #[inline]
989 #[stable(feature = "rust1", since = "1.0.0")]
990 #[rustc_const_unstable(feature = "const_replace", issue = "83164")]
991 pub const unsafe fn replace<T>(dst: *mut T, mut src: T) -> T {
992     // SAFETY: the caller must guarantee that `dst` is valid to be
993     // cast to a mutable reference (valid for writes, aligned, initialized),
994     // and cannot overlap `src` since `dst` must point to a distinct
995     // allocated object.
996     unsafe {
997         assert_unsafe_precondition!(
998             "ptr::replace requires that the pointer argument is aligned and non-null",
999             [T](dst: *mut T) => is_aligned_and_not_null(dst)
1000         );
1001         mem::swap(&mut *dst, &mut src); // cannot overlap
1002     }
1003     src
1004 }
1005
1006 /// Reads the value from `src` without moving it. This leaves the
1007 /// memory in `src` unchanged.
1008 ///
1009 /// # Safety
1010 ///
1011 /// Behavior is undefined if any of the following conditions are violated:
1012 ///
1013 /// * `src` must be [valid] for reads.
1014 ///
1015 /// * `src` must be properly aligned. Use [`read_unaligned`] if this is not the
1016 ///   case.
1017 ///
1018 /// * `src` must point to a properly initialized value of type `T`.
1019 ///
1020 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
1021 ///
1022 /// # Examples
1023 ///
1024 /// Basic usage:
1025 ///
1026 /// ```
1027 /// let x = 12;
1028 /// let y = &x as *const i32;
1029 ///
1030 /// unsafe {
1031 ///     assert_eq!(std::ptr::read(y), 12);
1032 /// }
1033 /// ```
1034 ///
1035 /// Manually implement [`mem::swap`]:
1036 ///
1037 /// ```
1038 /// use std::ptr;
1039 ///
1040 /// fn swap<T>(a: &mut T, b: &mut T) {
1041 ///     unsafe {
1042 ///         // Create a bitwise copy of the value at `a` in `tmp`.
1043 ///         let tmp = ptr::read(a);
1044 ///
1045 ///         // Exiting at this point (either by explicitly returning or by
1046 ///         // calling a function which panics) would cause the value in `tmp` to
1047 ///         // be dropped while the same value is still referenced by `a`. This
1048 ///         // could trigger undefined behavior if `T` is not `Copy`.
1049 ///
1050 ///         // Create a bitwise copy of the value at `b` in `a`.
1051 ///         // This is safe because mutable references cannot alias.
1052 ///         ptr::copy_nonoverlapping(b, a, 1);
1053 ///
1054 ///         // As above, exiting here could trigger undefined behavior because
1055 ///         // the same value is referenced by `a` and `b`.
1056 ///
1057 ///         // Move `tmp` into `b`.
1058 ///         ptr::write(b, tmp);
1059 ///
1060 ///         // `tmp` has been moved (`write` takes ownership of its second argument),
1061 ///         // so nothing is dropped implicitly here.
1062 ///     }
1063 /// }
1064 ///
1065 /// let mut foo = "foo".to_owned();
1066 /// let mut bar = "bar".to_owned();
1067 ///
1068 /// swap(&mut foo, &mut bar);
1069 ///
1070 /// assert_eq!(foo, "bar");
1071 /// assert_eq!(bar, "foo");
1072 /// ```
1073 ///
1074 /// ## Ownership of the Returned Value
1075 ///
1076 /// `read` creates a bitwise copy of `T`, regardless of whether `T` is [`Copy`].
1077 /// If `T` is not [`Copy`], using both the returned value and the value at
1078 /// `*src` can violate memory safety. Note that assigning to `*src` counts as a
1079 /// use because it will attempt to drop the value at `*src`.
1080 ///
1081 /// [`write()`] can be used to overwrite data without causing it to be dropped.
1082 ///
1083 /// ```
1084 /// use std::ptr;
1085 ///
1086 /// let mut s = String::from("foo");
1087 /// unsafe {
1088 ///     // `s2` now points to the same underlying memory as `s`.
1089 ///     let mut s2: String = ptr::read(&s);
1090 ///
1091 ///     assert_eq!(s2, "foo");
1092 ///
1093 ///     // Assigning to `s2` causes its original value to be dropped. Beyond
1094 ///     // this point, `s` must no longer be used, as the underlying memory has
1095 ///     // been freed.
1096 ///     s2 = String::default();
1097 ///     assert_eq!(s2, "");
1098 ///
1099 ///     // Assigning to `s` would cause the old value to be dropped again,
1100 ///     // resulting in undefined behavior.
1101 ///     // s = String::from("bar"); // ERROR
1102 ///
1103 ///     // `ptr::write` can be used to overwrite a value without dropping it.
1104 ///     ptr::write(&mut s, String::from("bar"));
1105 /// }
1106 ///
1107 /// assert_eq!(s, "bar");
1108 /// ```
1109 ///
1110 /// [valid]: self#safety
1111 #[inline]
1112 #[stable(feature = "rust1", since = "1.0.0")]
1113 #[rustc_const_unstable(feature = "const_ptr_read", issue = "80377")]
1114 #[cfg_attr(miri, track_caller)] // even without panics, this helps for Miri backtraces
1115 pub const unsafe fn read<T>(src: *const T) -> T {
1116     // We are calling the intrinsics directly to avoid function calls in the generated code
1117     // as `intrinsics::copy_nonoverlapping` is a wrapper function.
1118     extern "rust-intrinsic" {
1119         #[rustc_const_stable(feature = "const_intrinsic_copy", since = "1.63.0")]
1120         fn copy_nonoverlapping<T>(src: *const T, dst: *mut T, count: usize);
1121     }
1122
1123     let mut tmp = MaybeUninit::<T>::uninit();
1124     // SAFETY: the caller must guarantee that `src` is valid for reads.
1125     // `src` cannot overlap `tmp` because `tmp` was just allocated on
1126     // the stack as a separate allocated object.
1127     //
1128     // Also, since we just wrote a valid value into `tmp`, it is guaranteed
1129     // to be properly initialized.
1130     unsafe {
1131         assert_unsafe_precondition!(
1132             "ptr::read requires that the pointer argument is aligned and non-null",
1133             [T](src: *const T) => is_aligned_and_not_null(src)
1134         );
1135         copy_nonoverlapping(src, tmp.as_mut_ptr(), 1);
1136         tmp.assume_init()
1137     }
1138 }
1139
1140 /// Reads the value from `src` without moving it. This leaves the
1141 /// memory in `src` unchanged.
1142 ///
1143 /// Unlike [`read`], `read_unaligned` works with unaligned pointers.
1144 ///
1145 /// # Safety
1146 ///
1147 /// Behavior is undefined if any of the following conditions are violated:
1148 ///
1149 /// * `src` must be [valid] for reads.
1150 ///
1151 /// * `src` must point to a properly initialized value of type `T`.
1152 ///
1153 /// Like [`read`], `read_unaligned` creates a bitwise copy of `T`, regardless of
1154 /// whether `T` is [`Copy`]. If `T` is not [`Copy`], using both the returned
1155 /// value and the value at `*src` can [violate memory safety][read-ownership].
1156 ///
1157 /// Note that even if `T` has size `0`, the pointer must be non-null.
1158 ///
1159 /// [read-ownership]: read#ownership-of-the-returned-value
1160 /// [valid]: self#safety
1161 ///
1162 /// ## On `packed` structs
1163 ///
1164 /// Attempting to create a raw pointer to an `unaligned` struct field with
1165 /// an expression such as `&packed.unaligned as *const FieldType` creates an
1166 /// intermediate unaligned reference before converting that to a raw pointer.
1167 /// That this reference is temporary and immediately cast is inconsequential
1168 /// as the compiler always expects references to be properly aligned.
1169 /// As a result, using `&packed.unaligned as *const FieldType` causes immediate
1170 /// *undefined behavior* in your program.
1171 ///
1172 /// Instead you must use the [`ptr::addr_of!`](addr_of) macro to
1173 /// create the pointer. You may use that returned pointer together with this
1174 /// function.
1175 ///
1176 /// An example of what not to do and how this relates to `read_unaligned` is:
1177 ///
1178 /// ```
1179 /// #[repr(packed, C)]
1180 /// struct Packed {
1181 ///     _padding: u8,
1182 ///     unaligned: u32,
1183 /// }
1184 ///
1185 /// let packed = Packed {
1186 ///     _padding: 0x00,
1187 ///     unaligned: 0x01020304,
1188 /// };
1189 ///
1190 /// // Take the address of a 32-bit integer which is not aligned.
1191 /// // In contrast to `&packed.unaligned as *const _`, this has no undefined behavior.
1192 /// let unaligned = std::ptr::addr_of!(packed.unaligned);
1193 ///
1194 /// let v = unsafe { std::ptr::read_unaligned(unaligned) };
1195 /// assert_eq!(v, 0x01020304);
1196 /// ```
1197 ///
1198 /// Accessing unaligned fields directly with e.g. `packed.unaligned` is safe however.
1199 ///
1200 /// # Examples
1201 ///
1202 /// Read a usize value from a byte buffer:
1203 ///
1204 /// ```
1205 /// use std::mem;
1206 ///
1207 /// fn read_usize(x: &[u8]) -> usize {
1208 ///     assert!(x.len() >= mem::size_of::<usize>());
1209 ///
1210 ///     let ptr = x.as_ptr() as *const usize;
1211 ///
1212 ///     unsafe { ptr.read_unaligned() }
1213 /// }
1214 /// ```
1215 #[inline]
1216 #[stable(feature = "ptr_unaligned", since = "1.17.0")]
1217 #[rustc_const_unstable(feature = "const_ptr_read", issue = "80377")]
1218 #[cfg_attr(miri, track_caller)] // even without panics, this helps for Miri backtraces
1219 pub const unsafe fn read_unaligned<T>(src: *const T) -> T {
1220     let mut tmp = MaybeUninit::<T>::uninit();
1221     // SAFETY: the caller must guarantee that `src` is valid for reads.
1222     // `src` cannot overlap `tmp` because `tmp` was just allocated on
1223     // the stack as a separate allocated object.
1224     //
1225     // Also, since we just wrote a valid value into `tmp`, it is guaranteed
1226     // to be properly initialized.
1227     unsafe {
1228         copy_nonoverlapping(src as *const u8, tmp.as_mut_ptr() as *mut u8, mem::size_of::<T>());
1229         tmp.assume_init()
1230     }
1231 }
1232
1233 /// Overwrites a memory location with the given value without reading or
1234 /// dropping the old value.
1235 ///
1236 /// `write` does not drop the contents of `dst`. This is safe, but it could leak
1237 /// allocations or resources, so care should be taken not to overwrite an object
1238 /// that should be dropped.
1239 ///
1240 /// Additionally, it does not drop `src`. Semantically, `src` is moved into the
1241 /// location pointed to by `dst`.
1242 ///
1243 /// This is appropriate for initializing uninitialized memory, or overwriting
1244 /// memory that has previously been [`read`] from.
1245 ///
1246 /// # Safety
1247 ///
1248 /// Behavior is undefined if any of the following conditions are violated:
1249 ///
1250 /// * `dst` must be [valid] for writes.
1251 ///
1252 /// * `dst` must be properly aligned. Use [`write_unaligned`] if this is not the
1253 ///   case.
1254 ///
1255 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
1256 ///
1257 /// [valid]: self#safety
1258 ///
1259 /// # Examples
1260 ///
1261 /// Basic usage:
1262 ///
1263 /// ```
1264 /// let mut x = 0;
1265 /// let y = &mut x as *mut i32;
1266 /// let z = 12;
1267 ///
1268 /// unsafe {
1269 ///     std::ptr::write(y, z);
1270 ///     assert_eq!(std::ptr::read(y), 12);
1271 /// }
1272 /// ```
1273 ///
1274 /// Manually implement [`mem::swap`]:
1275 ///
1276 /// ```
1277 /// use std::ptr;
1278 ///
1279 /// fn swap<T>(a: &mut T, b: &mut T) {
1280 ///     unsafe {
1281 ///         // Create a bitwise copy of the value at `a` in `tmp`.
1282 ///         let tmp = ptr::read(a);
1283 ///
1284 ///         // Exiting at this point (either by explicitly returning or by
1285 ///         // calling a function which panics) would cause the value in `tmp` to
1286 ///         // be dropped while the same value is still referenced by `a`. This
1287 ///         // could trigger undefined behavior if `T` is not `Copy`.
1288 ///
1289 ///         // Create a bitwise copy of the value at `b` in `a`.
1290 ///         // This is safe because mutable references cannot alias.
1291 ///         ptr::copy_nonoverlapping(b, a, 1);
1292 ///
1293 ///         // As above, exiting here could trigger undefined behavior because
1294 ///         // the same value is referenced by `a` and `b`.
1295 ///
1296 ///         // Move `tmp` into `b`.
1297 ///         ptr::write(b, tmp);
1298 ///
1299 ///         // `tmp` has been moved (`write` takes ownership of its second argument),
1300 ///         // so nothing is dropped implicitly here.
1301 ///     }
1302 /// }
1303 ///
1304 /// let mut foo = "foo".to_owned();
1305 /// let mut bar = "bar".to_owned();
1306 ///
1307 /// swap(&mut foo, &mut bar);
1308 ///
1309 /// assert_eq!(foo, "bar");
1310 /// assert_eq!(bar, "foo");
1311 /// ```
1312 #[inline]
1313 #[stable(feature = "rust1", since = "1.0.0")]
1314 #[rustc_const_unstable(feature = "const_ptr_write", issue = "86302")]
1315 #[cfg_attr(miri, track_caller)] // even without panics, this helps for Miri backtraces
1316 pub const unsafe fn write<T>(dst: *mut T, src: T) {
1317     // We are calling the intrinsics directly to avoid function calls in the generated code
1318     // as `intrinsics::copy_nonoverlapping` is a wrapper function.
1319     extern "rust-intrinsic" {
1320         #[rustc_const_stable(feature = "const_intrinsic_copy", since = "1.63.0")]
1321         fn copy_nonoverlapping<T>(src: *const T, dst: *mut T, count: usize);
1322     }
1323
1324     // SAFETY: the caller must guarantee that `dst` is valid for writes.
1325     // `dst` cannot overlap `src` because the caller has mutable access
1326     // to `dst` while `src` is owned by this function.
1327     unsafe {
1328         assert_unsafe_precondition!(
1329             "ptr::write requires that the pointer argument is aligned and non-null",
1330             [T](dst: *mut T) => is_aligned_and_not_null(dst)
1331         );
1332         copy_nonoverlapping(&src as *const T, dst, 1);
1333         intrinsics::forget(src);
1334     }
1335 }
1336
1337 /// Overwrites a memory location with the given value without reading or
1338 /// dropping the old value.
1339 ///
1340 /// Unlike [`write()`], the pointer may be unaligned.
1341 ///
1342 /// `write_unaligned` does not drop the contents of `dst`. This is safe, but it
1343 /// could leak allocations or resources, so care should be taken not to overwrite
1344 /// an object that should be dropped.
1345 ///
1346 /// Additionally, it does not drop `src`. Semantically, `src` is moved into the
1347 /// location pointed to by `dst`.
1348 ///
1349 /// This is appropriate for initializing uninitialized memory, or overwriting
1350 /// memory that has previously been read with [`read_unaligned`].
1351 ///
1352 /// # Safety
1353 ///
1354 /// Behavior is undefined if any of the following conditions are violated:
1355 ///
1356 /// * `dst` must be [valid] for writes.
1357 ///
1358 /// Note that even if `T` has size `0`, the pointer must be non-null.
1359 ///
1360 /// [valid]: self#safety
1361 ///
1362 /// ## On `packed` structs
1363 ///
1364 /// Attempting to create a raw pointer to an `unaligned` struct field with
1365 /// an expression such as `&packed.unaligned as *const FieldType` creates an
1366 /// intermediate unaligned reference before converting that to a raw pointer.
1367 /// That this reference is temporary and immediately cast is inconsequential
1368 /// as the compiler always expects references to be properly aligned.
1369 /// As a result, using `&packed.unaligned as *const FieldType` causes immediate
1370 /// *undefined behavior* in your program.
1371 ///
1372 /// Instead you must use the [`ptr::addr_of_mut!`](addr_of_mut)
1373 /// macro to create the pointer. You may use that returned pointer together with
1374 /// this function.
1375 ///
1376 /// An example of how to do it and how this relates to `write_unaligned` is:
1377 ///
1378 /// ```
1379 /// #[repr(packed, C)]
1380 /// struct Packed {
1381 ///     _padding: u8,
1382 ///     unaligned: u32,
1383 /// }
1384 ///
1385 /// let mut packed: Packed = unsafe { std::mem::zeroed() };
1386 ///
1387 /// // Take the address of a 32-bit integer which is not aligned.
1388 /// // In contrast to `&packed.unaligned as *mut _`, this has no undefined behavior.
1389 /// let unaligned = std::ptr::addr_of_mut!(packed.unaligned);
1390 ///
1391 /// unsafe { std::ptr::write_unaligned(unaligned, 42) };
1392 ///
1393 /// assert_eq!({packed.unaligned}, 42); // `{...}` forces copying the field instead of creating a reference.
1394 /// ```
1395 ///
1396 /// Accessing unaligned fields directly with e.g. `packed.unaligned` is safe however
1397 /// (as can be seen in the `assert_eq!` above).
1398 ///
1399 /// # Examples
1400 ///
1401 /// Write a usize value to a byte buffer:
1402 ///
1403 /// ```
1404 /// use std::mem;
1405 ///
1406 /// fn write_usize(x: &mut [u8], val: usize) {
1407 ///     assert!(x.len() >= mem::size_of::<usize>());
1408 ///
1409 ///     let ptr = x.as_mut_ptr() as *mut usize;
1410 ///
1411 ///     unsafe { ptr.write_unaligned(val) }
1412 /// }
1413 /// ```
1414 #[inline]
1415 #[stable(feature = "ptr_unaligned", since = "1.17.0")]
1416 #[rustc_const_unstable(feature = "const_ptr_write", issue = "86302")]
1417 #[cfg_attr(miri, track_caller)] // even without panics, this helps for Miri backtraces
1418 pub const unsafe fn write_unaligned<T>(dst: *mut T, src: T) {
1419     // SAFETY: the caller must guarantee that `dst` is valid for writes.
1420     // `dst` cannot overlap `src` because the caller has mutable access
1421     // to `dst` while `src` is owned by this function.
1422     unsafe {
1423         copy_nonoverlapping(&src as *const T as *const u8, dst as *mut u8, mem::size_of::<T>());
1424         // We are calling the intrinsic directly to avoid function calls in the generated code.
1425         intrinsics::forget(src);
1426     }
1427 }
1428
1429 /// Performs a volatile read of the value from `src` without moving it. This
1430 /// leaves the memory in `src` unchanged.
1431 ///
1432 /// Volatile operations are intended to act on I/O memory, and are guaranteed
1433 /// to not be elided or reordered by the compiler across other volatile
1434 /// operations.
1435 ///
1436 /// # Notes
1437 ///
1438 /// Rust does not currently have a rigorously and formally defined memory model,
1439 /// so the precise semantics of what "volatile" means here is subject to change
1440 /// over time. That being said, the semantics will almost always end up pretty
1441 /// similar to [C11's definition of volatile][c11].
1442 ///
1443 /// The compiler shouldn't change the relative order or number of volatile
1444 /// memory operations. However, volatile memory operations on zero-sized types
1445 /// (e.g., if a zero-sized type is passed to `read_volatile`) are noops
1446 /// and may be ignored.
1447 ///
1448 /// [c11]: http://www.open-std.org/jtc1/sc22/wg14/www/docs/n1570.pdf
1449 ///
1450 /// # Safety
1451 ///
1452 /// Behavior is undefined if any of the following conditions are violated:
1453 ///
1454 /// * `src` must be [valid] for reads.
1455 ///
1456 /// * `src` must be properly aligned.
1457 ///
1458 /// * `src` must point to a properly initialized value of type `T`.
1459 ///
1460 /// Like [`read`], `read_volatile` creates a bitwise copy of `T`, regardless of
1461 /// whether `T` is [`Copy`]. If `T` is not [`Copy`], using both the returned
1462 /// value and the value at `*src` can [violate memory safety][read-ownership].
1463 /// However, storing non-[`Copy`] types in volatile memory is almost certainly
1464 /// incorrect.
1465 ///
1466 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
1467 ///
1468 /// [valid]: self#safety
1469 /// [read-ownership]: read#ownership-of-the-returned-value
1470 ///
1471 /// Just like in C, whether an operation is volatile has no bearing whatsoever
1472 /// on questions involving concurrent access from multiple threads. Volatile
1473 /// accesses behave exactly like non-atomic accesses in that regard. In particular,
1474 /// a race between a `read_volatile` and any write operation to the same location
1475 /// is undefined behavior.
1476 ///
1477 /// # Examples
1478 ///
1479 /// Basic usage:
1480 ///
1481 /// ```
1482 /// let x = 12;
1483 /// let y = &x as *const i32;
1484 ///
1485 /// unsafe {
1486 ///     assert_eq!(std::ptr::read_volatile(y), 12);
1487 /// }
1488 /// ```
1489 #[inline]
1490 #[stable(feature = "volatile", since = "1.9.0")]
1491 #[cfg_attr(miri, track_caller)] // even without panics, this helps for Miri backtraces
1492 pub unsafe fn read_volatile<T>(src: *const T) -> T {
1493     // SAFETY: the caller must uphold the safety contract for `volatile_load`.
1494     unsafe {
1495         assert_unsafe_precondition!(
1496             "ptr::read_volatile requires that the pointer argument is aligned and non-null",
1497             [T](src: *const T) => is_aligned_and_not_null(src)
1498         );
1499         intrinsics::volatile_load(src)
1500     }
1501 }
1502
1503 /// Performs a volatile write of a memory location with the given value without
1504 /// reading or dropping the old value.
1505 ///
1506 /// Volatile operations are intended to act on I/O memory, and are guaranteed
1507 /// to not be elided or reordered by the compiler across other volatile
1508 /// operations.
1509 ///
1510 /// `write_volatile` does not drop the contents of `dst`. This is safe, but it
1511 /// could leak allocations or resources, so care should be taken not to overwrite
1512 /// an object that should be dropped.
1513 ///
1514 /// Additionally, it does not drop `src`. Semantically, `src` is moved into the
1515 /// location pointed to by `dst`.
1516 ///
1517 /// # Notes
1518 ///
1519 /// Rust does not currently have a rigorously and formally defined memory model,
1520 /// so the precise semantics of what "volatile" means here is subject to change
1521 /// over time. That being said, the semantics will almost always end up pretty
1522 /// similar to [C11's definition of volatile][c11].
1523 ///
1524 /// The compiler shouldn't change the relative order or number of volatile
1525 /// memory operations. However, volatile memory operations on zero-sized types
1526 /// (e.g., if a zero-sized type is passed to `write_volatile`) are noops
1527 /// and may be ignored.
1528 ///
1529 /// [c11]: http://www.open-std.org/jtc1/sc22/wg14/www/docs/n1570.pdf
1530 ///
1531 /// # Safety
1532 ///
1533 /// Behavior is undefined if any of the following conditions are violated:
1534 ///
1535 /// * `dst` must be [valid] for writes.
1536 ///
1537 /// * `dst` must be properly aligned.
1538 ///
1539 /// Note that even if `T` has size `0`, the pointer must be non-null and properly aligned.
1540 ///
1541 /// [valid]: self#safety
1542 ///
1543 /// Just like in C, whether an operation is volatile has no bearing whatsoever
1544 /// on questions involving concurrent access from multiple threads. Volatile
1545 /// accesses behave exactly like non-atomic accesses in that regard. In particular,
1546 /// a race between a `write_volatile` and any other operation (reading or writing)
1547 /// on the same location is undefined behavior.
1548 ///
1549 /// # Examples
1550 ///
1551 /// Basic usage:
1552 ///
1553 /// ```
1554 /// let mut x = 0;
1555 /// let y = &mut x as *mut i32;
1556 /// let z = 12;
1557 ///
1558 /// unsafe {
1559 ///     std::ptr::write_volatile(y, z);
1560 ///     assert_eq!(std::ptr::read_volatile(y), 12);
1561 /// }
1562 /// ```
1563 #[inline]
1564 #[stable(feature = "volatile", since = "1.9.0")]
1565 #[cfg_attr(miri, track_caller)] // even without panics, this helps for Miri backtraces
1566 pub unsafe fn write_volatile<T>(dst: *mut T, src: T) {
1567     // SAFETY: the caller must uphold the safety contract for `volatile_store`.
1568     unsafe {
1569         assert_unsafe_precondition!(
1570             "ptr::write_volatile requires that the pointer argument is aligned and non-null",
1571             [T](dst: *mut T) => is_aligned_and_not_null(dst)
1572         );
1573         intrinsics::volatile_store(dst, src);
1574     }
1575 }
1576
1577 /// Align pointer `p`.
1578 ///
1579 /// Calculate offset (in terms of elements of `size_of::<T>()` stride) that has to be applied
1580 /// to pointer `p` so that pointer `p` would get aligned to `a`.
1581 ///
1582 /// # Safety
1583 /// `a` must be a power of two.
1584 ///
1585 /// # Notes
1586 /// This implementation has been carefully tailored to not panic. It is UB for this to panic.
1587 /// The only real change that can be made here is change of `INV_TABLE_MOD_16` and associated
1588 /// constants.
1589 ///
1590 /// If we ever decide to make it possible to call the intrinsic with `a` that is not a
1591 /// power-of-two, it will probably be more prudent to just change to a naive implementation rather
1592 /// than trying to adapt this to accommodate that change.
1593 ///
1594 /// Any questions go to @nagisa.
1595 #[lang = "align_offset"]
1596 pub(crate) const unsafe fn align_offset<T: Sized>(p: *const T, a: usize) -> usize {
1597     // FIXME(#75598): Direct use of these intrinsics improves codegen significantly at opt-level <=
1598     // 1, where the method versions of these operations are not inlined.
1599     use intrinsics::{
1600         cttz_nonzero, exact_div, mul_with_overflow, unchecked_rem, unchecked_shl, unchecked_shr,
1601         unchecked_sub, wrapping_add, wrapping_mul, wrapping_sub,
1602     };
1603
1604     /// Calculate multiplicative modular inverse of `x` modulo `m`.
1605     ///
1606     /// This implementation is tailored for `align_offset` and has following preconditions:
1607     ///
1608     /// * `m` is a power-of-two;
1609     /// * `x < m`; (if `x ≥ m`, pass in `x % m` instead)
1610     ///
1611     /// Implementation of this function shall not panic. Ever.
1612     #[inline]
1613     const unsafe fn mod_inv(x: usize, m: usize) -> usize {
1614         /// Multiplicative modular inverse table modulo 2⁴ = 16.
1615         ///
1616         /// Note, that this table does not contain values where inverse does not exist (i.e., for
1617         /// `0⁻¹ mod 16`, `2⁻¹ mod 16`, etc.)
1618         const INV_TABLE_MOD_16: [u8; 8] = [1, 11, 13, 7, 9, 3, 5, 15];
1619         /// Modulo for which the `INV_TABLE_MOD_16` is intended.
1620         const INV_TABLE_MOD: usize = 16;
1621
1622         // SAFETY: `m` is required to be a power-of-two, hence non-zero.
1623         let m_minus_one = unsafe { unchecked_sub(m, 1) };
1624         let mut inverse = INV_TABLE_MOD_16[(x & (INV_TABLE_MOD - 1)) >> 1] as usize;
1625         let mut mod_gate = INV_TABLE_MOD;
1626         // We iterate "up" using the following formula:
1627         //
1628         // $$ xy ≡ 1 (mod 2ⁿ) → xy (2 - xy) ≡ 1 (mod 2²ⁿ) $$
1629         //
1630         // This application needs to be applied at least until `2²ⁿ ≥ m`, at which point we can
1631         // finally reduce the computation to our desired `m` by taking `inverse mod m`.
1632         //
1633         // This computation is `O(log log m)`, which is to say, that on 64-bit machines this loop
1634         // will always finish in at most 4 iterations.
1635         loop {
1636             // y = y * (2 - xy) mod n
1637             //
1638             // Note, that we use wrapping operations here intentionally – the original formula
1639             // uses e.g., subtraction `mod n`. It is entirely fine to do them `mod
1640             // usize::MAX` instead, because we take the result `mod n` at the end
1641             // anyway.
1642             if mod_gate >= m {
1643                 break;
1644             }
1645             inverse = wrapping_mul(inverse, wrapping_sub(2usize, wrapping_mul(x, inverse)));
1646             let (new_gate, overflow) = mul_with_overflow(mod_gate, mod_gate);
1647             if overflow {
1648                 break;
1649             }
1650             mod_gate = new_gate;
1651         }
1652         inverse & m_minus_one
1653     }
1654
1655     let stride = mem::size_of::<T>();
1656
1657     // SAFETY: This is just an inlined `p.addr()` (which is not
1658     // a `const fn` so we cannot call it).
1659     // During const eval, we hook this function to ensure that the pointer never
1660     // has provenance, making this sound.
1661     let addr: usize = unsafe { mem::transmute(p) };
1662
1663     // SAFETY: `a` is a power-of-two, therefore non-zero.
1664     let a_minus_one = unsafe { unchecked_sub(a, 1) };
1665
1666     if stride == 0 {
1667         // SPECIAL_CASE: handle 0-sized types. No matter how many times we step, the address will
1668         // stay the same, so no offset will be able to align the pointer unless it is already
1669         // aligned. This branch _will_ be optimized out as `stride` is known at compile-time.
1670         let p_mod_a = addr & a_minus_one;
1671         return if p_mod_a == 0 { 0 } else { usize::MAX };
1672     }
1673
1674     // SAFETY: `stride == 0` case has been handled by the special case above.
1675     let a_mod_stride = unsafe { unchecked_rem(a, stride) };
1676     if a_mod_stride == 0 {
1677         // SPECIAL_CASE: In cases where the `a` is divisible by `stride`, byte offset to align a
1678         // pointer can be computed more simply through `-p (mod a)`. In the off-chance the byte
1679         // offset is not a multiple of `stride`, the input pointer was misaligned and no pointer
1680         // offset will be able to produce a `p` aligned to the specified `a`.
1681         //
1682         // The naive `-p (mod a)` equation  inhibits LLVM's ability to select instructions
1683         // like `lea`. We compute `(round_up_to_next_alignment(p, a) - p)` instead. This
1684         // redistributes operations around the load-bearing, but pessimizing `and` instruction
1685         // sufficiently for LLVM to be able to utilize the various optimizations it knows about.
1686         //
1687         // LLVM handles the branch here particularly nicely. If this branch needs to be evaluated
1688         // at runtime, it will produce a mask `if addr_mod_stride == 0 { 0 } else { usize::MAX }`
1689         // in a branch-free way and then bitwise-OR it with whatever result the `-p mod a`
1690         // computation produces.
1691
1692         // SAFETY: `stride == 0` case has been handled by the special case above.
1693         let addr_mod_stride = unsafe { unchecked_rem(addr, stride) };
1694
1695         return if addr_mod_stride == 0 {
1696             let aligned_address = wrapping_add(addr, a_minus_one) & wrapping_sub(0, a);
1697             let byte_offset = wrapping_sub(aligned_address, addr);
1698             // SAFETY: `stride` is non-zero. This is guaranteed to divide exactly as well, because
1699             // addr has been verified to be aligned to the original type’s alignment requirements.
1700             unsafe { exact_div(byte_offset, stride) }
1701         } else {
1702             usize::MAX
1703         };
1704     }
1705
1706     // GENERAL_CASE: From here on we’re handling the very general case where `addr` may be
1707     // misaligned, there isn’t an obvious relationship between `stride` and `a` that we can take an
1708     // advantage of, etc. This case produces machine code that isn’t particularly high quality,
1709     // compared to the special cases above. The code produced here is still within the realm of
1710     // miracles, given the situations this case has to deal with.
1711
1712     // SAFETY: a is power-of-two hence non-zero. stride == 0 case is handled above.
1713     let gcdpow = unsafe { cttz_nonzero(stride).min(cttz_nonzero(a)) };
1714     // SAFETY: gcdpow has an upper-bound that’s at most the number of bits in a usize.
1715     let gcd = unsafe { unchecked_shl(1usize, gcdpow) };
1716     // SAFETY: gcd is always greater or equal to 1.
1717     if addr & unsafe { unchecked_sub(gcd, 1) } == 0 {
1718         // This branch solves for the following linear congruence equation:
1719         //
1720         // ` p + so = 0 mod a `
1721         //
1722         // `p` here is the pointer value, `s` - stride of `T`, `o` offset in `T`s, and `a` - the
1723         // requested alignment.
1724         //
1725         // With `g = gcd(a, s)`, and the above condition asserting that `p` is also divisible by
1726         // `g`, we can denote `a' = a/g`, `s' = s/g`, `p' = p/g`, then this becomes equivalent to:
1727         //
1728         // ` p' + s'o = 0 mod a' `
1729         // ` o = (a' - (p' mod a')) * (s'^-1 mod a') `
1730         //
1731         // The first term is "the relative alignment of `p` to `a`" (divided by the `g`), the
1732         // second term is "how does incrementing `p` by `s` bytes change the relative alignment of
1733         // `p`" (again divided by `g`). Division by `g` is necessary to make the inverse well
1734         // formed if `a` and `s` are not co-prime.
1735         //
1736         // Furthermore, the result produced by this solution is not "minimal", so it is necessary
1737         // to take the result `o mod lcm(s, a)`. This `lcm(s, a)` is the same as `a'`.
1738
1739         // SAFETY: `gcdpow` has an upper-bound not greater than the number of trailing 0-bits in
1740         // `a`.
1741         let a2 = unsafe { unchecked_shr(a, gcdpow) };
1742         // SAFETY: `a2` is non-zero. Shifting `a` by `gcdpow` cannot shift out any of the set bits
1743         // in `a` (of which it has exactly one).
1744         let a2minus1 = unsafe { unchecked_sub(a2, 1) };
1745         // SAFETY: `gcdpow` has an upper-bound not greater than the number of trailing 0-bits in
1746         // `a`.
1747         let s2 = unsafe { unchecked_shr(stride & a_minus_one, gcdpow) };
1748         // SAFETY: `gcdpow` has an upper-bound not greater than the number of trailing 0-bits in
1749         // `a`. Furthermore, the subtraction cannot overflow, because `a2 = a >> gcdpow` will
1750         // always be strictly greater than `(p % a) >> gcdpow`.
1751         let minusp2 = unsafe { unchecked_sub(a2, unchecked_shr(addr & a_minus_one, gcdpow)) };
1752         // SAFETY: `a2` is a power-of-two, as proven above. `s2` is strictly less than `a2`
1753         // because `(s % a) >> gcdpow` is strictly less than `a >> gcdpow`.
1754         return wrapping_mul(minusp2, unsafe { mod_inv(s2, a2) }) & a2minus1;
1755     }
1756
1757     // Cannot be aligned at all.
1758     usize::MAX
1759 }
1760
1761 /// Compares raw pointers for equality.
1762 ///
1763 /// This is the same as using the `==` operator, but less generic:
1764 /// the arguments have to be `*const T` raw pointers,
1765 /// not anything that implements `PartialEq`.
1766 ///
1767 /// This can be used to compare `&T` references (which coerce to `*const T` implicitly)
1768 /// by their address rather than comparing the values they point to
1769 /// (which is what the `PartialEq for &T` implementation does).
1770 ///
1771 /// When comparing wide pointers, both the address and the metadata are tested for equality.
1772 /// However, note that comparing trait object pointers (`*const dyn Trait`) is unreliable: pointers
1773 /// to values of the same underlying type can compare inequal (because vtables are duplicated in
1774 /// multiple codegen units), and pointers to values of *different* underlying type can compare equal
1775 /// (since identical vtables can be deduplicated within a codegen unit).
1776 ///
1777 /// # Examples
1778 ///
1779 /// ```
1780 /// use std::ptr;
1781 ///
1782 /// let five = 5;
1783 /// let other_five = 5;
1784 /// let five_ref = &five;
1785 /// let same_five_ref = &five;
1786 /// let other_five_ref = &other_five;
1787 ///
1788 /// assert!(five_ref == same_five_ref);
1789 /// assert!(ptr::eq(five_ref, same_five_ref));
1790 ///
1791 /// assert!(five_ref == other_five_ref);
1792 /// assert!(!ptr::eq(five_ref, other_five_ref));
1793 /// ```
1794 ///
1795 /// Slices are also compared by their length (fat pointers):
1796 ///
1797 /// ```
1798 /// let a = [1, 2, 3];
1799 /// assert!(std::ptr::eq(&a[..3], &a[..3]));
1800 /// assert!(!std::ptr::eq(&a[..2], &a[..3]));
1801 /// assert!(!std::ptr::eq(&a[0..2], &a[1..3]));
1802 /// ```
1803 #[stable(feature = "ptr_eq", since = "1.17.0")]
1804 #[inline(always)]
1805 pub fn eq<T: ?Sized>(a: *const T, b: *const T) -> bool {
1806     a == b
1807 }
1808
1809 /// Hash a raw pointer.
1810 ///
1811 /// This can be used to hash a `&T` reference (which coerces to `*const T` implicitly)
1812 /// by its address rather than the value it points to
1813 /// (which is what the `Hash for &T` implementation does).
1814 ///
1815 /// # Examples
1816 ///
1817 /// ```
1818 /// use std::collections::hash_map::DefaultHasher;
1819 /// use std::hash::{Hash, Hasher};
1820 /// use std::ptr;
1821 ///
1822 /// let five = 5;
1823 /// let five_ref = &five;
1824 ///
1825 /// let mut hasher = DefaultHasher::new();
1826 /// ptr::hash(five_ref, &mut hasher);
1827 /// let actual = hasher.finish();
1828 ///
1829 /// let mut hasher = DefaultHasher::new();
1830 /// (five_ref as *const i32).hash(&mut hasher);
1831 /// let expected = hasher.finish();
1832 ///
1833 /// assert_eq!(actual, expected);
1834 /// ```
1835 #[stable(feature = "ptr_hash", since = "1.35.0")]
1836 pub fn hash<T: ?Sized, S: hash::Hasher>(hashee: *const T, into: &mut S) {
1837     use crate::hash::Hash;
1838     hashee.hash(into);
1839 }
1840
1841 // If this is a unary fn pointer, it adds a doc comment.
1842 // Otherwise, it hides the docs entirely.
1843 macro_rules! maybe_fnptr_doc {
1844     (@ #[$meta:meta] $item:item) => {
1845         #[doc(hidden)]
1846         #[$meta]
1847         $item
1848     };
1849     ($a:ident @ #[$meta:meta] $item:item) => {
1850         #[doc(fake_variadic)]
1851         #[doc = "This trait is implemented for function pointers with up to twelve arguments."]
1852         #[$meta]
1853         $item
1854     };
1855     ($a:ident $($rest_a:ident)+ @ #[$meta:meta] $item:item) => {
1856         #[doc(hidden)]
1857         #[$meta]
1858         $item
1859     };
1860 }
1861
1862 // FIXME(strict_provenance_magic): function pointers have buggy codegen that
1863 // necessitates casting to a usize to get the backend to do the right thing.
1864 // for now I will break AVR to silence *a billion* lints. We should probably
1865 // have a proper "opaque function pointer type" to handle this kind of thing.
1866
1867 // Impls for function pointers
1868 macro_rules! fnptr_impls_safety_abi {
1869     ($FnTy: ty, $($Arg: ident),*) => {
1870         fnptr_impls_safety_abi! { #[stable(feature = "fnptr_impls", since = "1.4.0")] $FnTy, $($Arg),* }
1871     };
1872     (@c_unwind $FnTy: ty, $($Arg: ident),*) => {
1873         fnptr_impls_safety_abi! { #[unstable(feature = "c_unwind", issue = "74990")] $FnTy, $($Arg),* }
1874     };
1875     (#[$meta:meta] $FnTy: ty, $($Arg: ident),*) => {
1876         maybe_fnptr_doc! {
1877             $($Arg)* @
1878             #[$meta]
1879             impl<Ret, $($Arg),*> PartialEq for $FnTy {
1880                 #[inline]
1881                 fn eq(&self, other: &Self) -> bool {
1882                     *self as usize == *other as usize
1883                 }
1884             }
1885         }
1886
1887         maybe_fnptr_doc! {
1888             $($Arg)* @
1889             #[$meta]
1890             impl<Ret, $($Arg),*> Eq for $FnTy {}
1891         }
1892
1893         maybe_fnptr_doc! {
1894             $($Arg)* @
1895             #[$meta]
1896             impl<Ret, $($Arg),*> PartialOrd for $FnTy {
1897                 #[inline]
1898                 fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
1899                     (*self as usize).partial_cmp(&(*other as usize))
1900                 }
1901             }
1902         }
1903
1904         maybe_fnptr_doc! {
1905             $($Arg)* @
1906             #[$meta]
1907             impl<Ret, $($Arg),*> Ord for $FnTy {
1908                 #[inline]
1909                 fn cmp(&self, other: &Self) -> Ordering {
1910                     (*self as usize).cmp(&(*other as usize))
1911                 }
1912             }
1913         }
1914
1915         maybe_fnptr_doc! {
1916             $($Arg)* @
1917             #[$meta]
1918             impl<Ret, $($Arg),*> hash::Hash for $FnTy {
1919                 fn hash<HH: hash::Hasher>(&self, state: &mut HH) {
1920                     state.write_usize(*self as usize)
1921                 }
1922             }
1923         }
1924
1925         maybe_fnptr_doc! {
1926             $($Arg)* @
1927             #[$meta]
1928             impl<Ret, $($Arg),*> fmt::Pointer for $FnTy {
1929                 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1930                     fmt::pointer_fmt_inner(*self as usize, f)
1931                 }
1932             }
1933         }
1934
1935         maybe_fnptr_doc! {
1936             $($Arg)* @
1937             #[$meta]
1938             impl<Ret, $($Arg),*> fmt::Debug for $FnTy {
1939                 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1940                     fmt::pointer_fmt_inner(*self as usize, f)
1941                 }
1942             }
1943         }
1944     }
1945 }
1946
1947 macro_rules! fnptr_impls_args {
1948     ($($Arg: ident),+) => {
1949         fnptr_impls_safety_abi! { extern "Rust" fn($($Arg),+) -> Ret, $($Arg),+ }
1950         fnptr_impls_safety_abi! { extern "C" fn($($Arg),+) -> Ret, $($Arg),+ }
1951         fnptr_impls_safety_abi! { extern "C" fn($($Arg),+ , ...) -> Ret, $($Arg),+ }
1952         fnptr_impls_safety_abi! { @c_unwind extern "C-unwind" fn($($Arg),+) -> Ret, $($Arg),+ }
1953         fnptr_impls_safety_abi! { @c_unwind extern "C-unwind" fn($($Arg),+ , ...) -> Ret, $($Arg),+ }
1954         fnptr_impls_safety_abi! { unsafe extern "Rust" fn($($Arg),+) -> Ret, $($Arg),+ }
1955         fnptr_impls_safety_abi! { unsafe extern "C" fn($($Arg),+) -> Ret, $($Arg),+ }
1956         fnptr_impls_safety_abi! { unsafe extern "C" fn($($Arg),+ , ...) -> Ret, $($Arg),+ }
1957         fnptr_impls_safety_abi! { @c_unwind unsafe extern "C-unwind" fn($($Arg),+) -> Ret, $($Arg),+ }
1958         fnptr_impls_safety_abi! { @c_unwind unsafe extern "C-unwind" fn($($Arg),+ , ...) -> Ret, $($Arg),+ }
1959     };
1960     () => {
1961         // No variadic functions with 0 parameters
1962         fnptr_impls_safety_abi! { extern "Rust" fn() -> Ret, }
1963         fnptr_impls_safety_abi! { extern "C" fn() -> Ret, }
1964         fnptr_impls_safety_abi! { @c_unwind extern "C-unwind" fn() -> Ret, }
1965         fnptr_impls_safety_abi! { unsafe extern "Rust" fn() -> Ret, }
1966         fnptr_impls_safety_abi! { unsafe extern "C" fn() -> Ret, }
1967         fnptr_impls_safety_abi! { @c_unwind unsafe extern "C-unwind" fn() -> Ret, }
1968     };
1969 }
1970
1971 fnptr_impls_args! {}
1972 fnptr_impls_args! { T }
1973 fnptr_impls_args! { A, B }
1974 fnptr_impls_args! { A, B, C }
1975 fnptr_impls_args! { A, B, C, D }
1976 fnptr_impls_args! { A, B, C, D, E }
1977 fnptr_impls_args! { A, B, C, D, E, F }
1978 fnptr_impls_args! { A, B, C, D, E, F, G }
1979 fnptr_impls_args! { A, B, C, D, E, F, G, H }
1980 fnptr_impls_args! { A, B, C, D, E, F, G, H, I }
1981 fnptr_impls_args! { A, B, C, D, E, F, G, H, I, J }
1982 fnptr_impls_args! { A, B, C, D, E, F, G, H, I, J, K }
1983 fnptr_impls_args! { A, B, C, D, E, F, G, H, I, J, K, L }
1984
1985 /// Create a `const` raw pointer to a place, without creating an intermediate reference.
1986 ///
1987 /// Creating a reference with `&`/`&mut` is only allowed if the pointer is properly aligned
1988 /// and points to initialized data. For cases where those requirements do not hold,
1989 /// raw pointers should be used instead. However, `&expr as *const _` creates a reference
1990 /// before casting it to a raw pointer, and that reference is subject to the same rules
1991 /// as all other references. This macro can create a raw pointer *without* creating
1992 /// a reference first.
1993 ///
1994 /// Note, however, that the `expr` in `addr_of!(expr)` is still subject to all
1995 /// the usual rules. In particular, `addr_of!(*ptr::null())` is Undefined
1996 /// Behavior because it dereferences a null pointer.
1997 ///
1998 /// # Example
1999 ///
2000 /// ```
2001 /// use std::ptr;
2002 ///
2003 /// #[repr(packed)]
2004 /// struct Packed {
2005 ///     f1: u8,
2006 ///     f2: u16,
2007 /// }
2008 ///
2009 /// let packed = Packed { f1: 1, f2: 2 };
2010 /// // `&packed.f2` would create an unaligned reference, and thus be Undefined Behavior!
2011 /// let raw_f2 = ptr::addr_of!(packed.f2);
2012 /// assert_eq!(unsafe { raw_f2.read_unaligned() }, 2);
2013 /// ```
2014 ///
2015 /// See [`addr_of_mut`] for how to create a pointer to unininitialized data.
2016 /// Doing that with `addr_of` would not make much sense since one could only
2017 /// read the data, and that would be Undefined Behavior.
2018 #[stable(feature = "raw_ref_macros", since = "1.51.0")]
2019 #[rustc_macro_transparency = "semitransparent"]
2020 #[allow_internal_unstable(raw_ref_op)]
2021 pub macro addr_of($place:expr) {
2022     &raw const $place
2023 }
2024
2025 /// Create a `mut` raw pointer to a place, without creating an intermediate reference.
2026 ///
2027 /// Creating a reference with `&`/`&mut` is only allowed if the pointer is properly aligned
2028 /// and points to initialized data. For cases where those requirements do not hold,
2029 /// raw pointers should be used instead. However, `&mut expr as *mut _` creates a reference
2030 /// before casting it to a raw pointer, and that reference is subject to the same rules
2031 /// as all other references. This macro can create a raw pointer *without* creating
2032 /// a reference first.
2033 ///
2034 /// Note, however, that the `expr` in `addr_of_mut!(expr)` is still subject to all
2035 /// the usual rules. In particular, `addr_of_mut!(*ptr::null_mut())` is Undefined
2036 /// Behavior because it dereferences a null pointer.
2037 ///
2038 /// # Examples
2039 ///
2040 /// **Creating a pointer to unaligned data:**
2041 ///
2042 /// ```
2043 /// use std::ptr;
2044 ///
2045 /// #[repr(packed)]
2046 /// struct Packed {
2047 ///     f1: u8,
2048 ///     f2: u16,
2049 /// }
2050 ///
2051 /// let mut packed = Packed { f1: 1, f2: 2 };
2052 /// // `&mut packed.f2` would create an unaligned reference, and thus be Undefined Behavior!
2053 /// let raw_f2 = ptr::addr_of_mut!(packed.f2);
2054 /// unsafe { raw_f2.write_unaligned(42); }
2055 /// assert_eq!({packed.f2}, 42); // `{...}` forces copying the field instead of creating a reference.
2056 /// ```
2057 ///
2058 /// **Creating a pointer to uninitialized data:**
2059 ///
2060 /// ```rust
2061 /// use std::{ptr, mem::MaybeUninit};
2062 ///
2063 /// struct Demo {
2064 ///     field: bool,
2065 /// }
2066 ///
2067 /// let mut uninit = MaybeUninit::<Demo>::uninit();
2068 /// // `&uninit.as_mut().field` would create a reference to an uninitialized `bool`,
2069 /// // and thus be Undefined Behavior!
2070 /// let f1_ptr = unsafe { ptr::addr_of_mut!((*uninit.as_mut_ptr()).field) };
2071 /// unsafe { f1_ptr.write(true); }
2072 /// let init = unsafe { uninit.assume_init() };
2073 /// ```
2074 #[stable(feature = "raw_ref_macros", since = "1.51.0")]
2075 #[rustc_macro_transparency = "semitransparent"]
2076 #[allow_internal_unstable(raw_ref_op)]
2077 pub macro addr_of_mut($place:expr) {
2078     &raw mut $place
2079 }