2018-04-06 07:34:45 +00:00
|
|
|
|
//! Manually manage memory through raw pointers.
|
2014-04-07 21:00:19 +00:00
|
|
|
|
//!
|
2016-03-01 12:44:48 +00:00
|
|
|
|
//! *[See also the pointer primitive types](../../std/primitive.pointer.html).*
|
2018-04-06 07:34:45 +00:00
|
|
|
|
//!
|
|
|
|
|
//! # Safety
|
|
|
|
|
//!
|
2018-08-29 12:34:59 +00:00
|
|
|
|
//! Many functions in this module take raw pointers as arguments and read from
|
|
|
|
|
//! or write to them. For this to be safe, these pointers must be *valid*.
|
|
|
|
|
//! Whether a pointer is valid depends on the operation it is used for
|
|
|
|
|
//! (read or write), and the extent of the memory that is accessed (i.e.,
|
|
|
|
|
//! how many bytes are read/written). Most functions use `*mut T` and `*const T`
|
|
|
|
|
//! to access only a single value, in which case the documentation omits the size
|
|
|
|
|
//! and implicitly assumes it to be `size_of::<T>()` bytes.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
//!
|
2019-02-09 21:23:30 +00:00
|
|
|
|
//! The precise rules for validity are not determined yet. The guarantees that are
|
2018-08-29 17:27:20 +00:00
|
|
|
|
//! provided at this point are very minimal:
|
2018-07-04 18:30:23 +00:00
|
|
|
|
//!
|
2018-08-29 12:34:59 +00:00
|
|
|
|
//! * A [null] pointer is *never* valid, not even for accesses of [size zero][zst].
|
2018-07-04 19:02:26 +00:00
|
|
|
|
//! * All pointers (except for the null pointer) are valid for all operations of
|
|
|
|
|
//! [size zero][zst].
|
2019-11-05 20:50:55 +00:00
|
|
|
|
//! * For a pointer to be valid, it is necessary, but not always sufficient, that the pointer
|
2020-03-06 11:13:55 +00:00
|
|
|
|
//! be *dereferenceable*: the memory range of the given size starting at the pointer must all be
|
2019-11-05 08:55:33 +00:00
|
|
|
|
//! within the bounds of a single allocated object. Note that in Rust,
|
|
|
|
|
//! every (stack-allocated) variable is considered a separate allocated object.
|
2018-09-01 16:00:10 +00:00
|
|
|
|
//! * All accesses performed by functions in this module are *non-atomic* in the sense
|
|
|
|
|
//! of [atomic operations] used to synchronize between threads. This means it is
|
|
|
|
|
//! undefined behavior to perform two concurrent accesses to the same location from different
|
|
|
|
|
//! threads unless both accesses only read from memory. Notice that this explicitly
|
|
|
|
|
//! includes [`read_volatile`] and [`write_volatile`]: Volatile accesses cannot
|
|
|
|
|
//! be used for inter-thread synchronization.
|
2018-08-29 12:34:59 +00:00
|
|
|
|
//! * The result of casting a reference to a pointer is valid for as long as the
|
|
|
|
|
//! underlying object is live and no reference (just raw pointers) is used to
|
|
|
|
|
//! access the same memory.
|
2018-07-04 18:30:23 +00:00
|
|
|
|
//!
|
2018-10-22 16:21:55 +00:00
|
|
|
|
//! These axioms, along with careful use of [`offset`] for pointer arithmetic,
|
2018-08-29 17:27:20 +00:00
|
|
|
|
//! are enough to correctly implement many useful things in unsafe code. Stronger guarantees
|
|
|
|
|
//! will be provided eventually, as the [aliasing] rules are being determined. For more
|
2018-07-04 18:30:23 +00:00
|
|
|
|
//! information, see the [book] as well as the section in the reference devoted
|
2018-06-17 09:45:38 +00:00
|
|
|
|
//! to [undefined behavior][ub].
|
2018-06-15 23:00:07 +00:00
|
|
|
|
//!
|
|
|
|
|
//! ## Alignment
|
2018-04-06 07:34:45 +00:00
|
|
|
|
//!
|
2018-09-01 15:50:54 +00:00
|
|
|
|
//! Valid raw pointers as defined above are not necessarily properly aligned (where
|
2018-09-11 05:30:30 +00:00
|
|
|
|
//! "proper" alignment is defined by the pointee type, i.e., `*const T` must be
|
2018-08-31 15:01:53 +00:00
|
|
|
|
//! aligned to `mem::align_of::<T>()`). However, most functions require their
|
|
|
|
|
//! arguments to be properly aligned, and will explicitly state
|
2018-06-17 09:45:38 +00:00
|
|
|
|
//! this requirement in their documentation. Notable exceptions to this are
|
2018-06-05 18:22:40 +00:00
|
|
|
|
//! [`read_unaligned`] and [`write_unaligned`].
|
2018-05-24 03:55:39 +00:00
|
|
|
|
//!
|
2018-08-29 12:34:59 +00:00
|
|
|
|
//! When a function requires proper alignment, it does so even if the access
|
|
|
|
|
//! has size 0, i.e., even if memory is not actually touched. Consider using
|
|
|
|
|
//! [`NonNull::dangling`] in such cases.
|
|
|
|
|
//!
|
2018-06-17 09:45:38 +00:00
|
|
|
|
//! [aliasing]: ../../nomicon/aliasing.html
|
2018-11-21 00:49:47 +00:00
|
|
|
|
//! [book]: ../../book/ch19-01-unsafe-rust.html#dereferencing-a-raw-pointer
|
2018-06-17 09:45:38 +00:00
|
|
|
|
//! [ub]: ../../reference/behavior-considered-undefined.html
|
2018-07-04 18:30:23 +00:00
|
|
|
|
//! [zst]: ../../nomicon/exotic-sizes.html#zero-sized-types-zsts
|
2020-09-08 21:36:36 +00:00
|
|
|
|
//! [atomic operations]: crate::sync::atomic
|
2018-07-04 18:30:23 +00:00
|
|
|
|
//! [`offset`]: ../../std/primitive.pointer.html#method.offset
|
2012-03-10 08:04:09 +00:00
|
|
|
|
|
2015-01-24 05:48:20 +00:00
|
|
|
|
#![stable(feature = "rust1", since = "1.0.0")]
|
2014-12-19 16:57:12 +00:00
|
|
|
|
|
2019-12-22 22:42:04 +00:00
|
|
|
|
use crate::cmp::Ordering;
|
2019-04-15 02:23:21 +00:00
|
|
|
|
use crate::fmt;
|
|
|
|
|
use crate::hash;
|
2020-06-06 09:56:58 +00:00
|
|
|
|
use crate::intrinsics::{self, abort, is_aligned_and_not_null, is_nonoverlapping};
|
2019-04-15 02:23:21 +00:00
|
|
|
|
use crate::mem::{self, MaybeUninit};
|
2013-02-28 16:57:33 +00:00
|
|
|
|
|
2015-02-23 19:39:16 +00:00
|
|
|
|
#[stable(feature = "rust1", since = "1.0.0")]
|
2020-04-23 00:59:30 +00:00
|
|
|
|
#[doc(inline)]
|
2019-04-15 02:23:21 +00:00
|
|
|
|
pub use crate::intrinsics::copy_nonoverlapping;
|
2014-12-09 01:12:35 +00:00
|
|
|
|
|
2015-02-23 19:39:16 +00:00
|
|
|
|
#[stable(feature = "rust1", since = "1.0.0")]
|
2020-04-23 00:59:30 +00:00
|
|
|
|
#[doc(inline)]
|
2019-04-15 02:23:21 +00:00
|
|
|
|
pub use crate::intrinsics::copy;
|
2014-12-09 01:12:35 +00:00
|
|
|
|
|
2015-02-23 19:39:16 +00:00
|
|
|
|
#[stable(feature = "rust1", since = "1.0.0")]
|
2020-04-23 00:59:30 +00:00
|
|
|
|
#[doc(inline)]
|
2019-04-15 02:23:21 +00:00
|
|
|
|
pub use crate::intrinsics::write_bytes;
|
2014-12-03 22:21:51 +00:00
|
|
|
|
|
2019-05-25 07:03:45 +00:00
|
|
|
|
mod non_null;
|
|
|
|
|
#[stable(feature = "nonnull", since = "1.25.0")]
|
|
|
|
|
pub use non_null::NonNull;
|
|
|
|
|
|
|
|
|
|
mod unique;
|
2019-12-21 11:16:18 +00:00
|
|
|
|
#[unstable(feature = "ptr_internals", issue = "none")]
|
2019-05-25 07:03:45 +00:00
|
|
|
|
pub use unique::Unique;
|
|
|
|
|
|
2019-12-08 20:22:18 +00:00
|
|
|
|
mod const_ptr;
|
|
|
|
|
mod mut_ptr;
|
|
|
|
|
|
2017-03-13 23:08:21 +00:00
|
|
|
|
/// Executes the destructor (if any) of the pointed-to value.
|
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// This is semantically equivalent to calling [`ptr::read`] and discarding
|
|
|
|
|
/// the result, but has the following advantages:
|
2017-03-13 23:08:21 +00:00
|
|
|
|
///
|
|
|
|
|
/// * It is *required* to use `drop_in_place` to drop unsized types like
|
|
|
|
|
/// trait objects, because they can't be read out onto the stack and
|
|
|
|
|
/// dropped normally.
|
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// * It is friendlier to the optimizer to do this over [`ptr::read`] when
|
2018-11-27 02:59:49 +00:00
|
|
|
|
/// dropping manually allocated memory (e.g., when writing Box/Rc/Vec),
|
2017-03-13 23:08:21 +00:00
|
|
|
|
/// as the compiler doesn't need to prove that it's sound to elide the
|
|
|
|
|
/// copy.
|
|
|
|
|
///
|
2020-04-27 21:50:55 +00:00
|
|
|
|
/// * It can be used to drop [pinned] data when `T` is not `repr(packed)`
|
|
|
|
|
/// (pinned data must not be moved before it is dropped).
|
|
|
|
|
///
|
2019-07-03 18:13:42 +00:00
|
|
|
|
/// Unaligned values cannot be dropped in place, they must be copied to an aligned
|
2020-04-27 21:50:55 +00:00
|
|
|
|
/// location first using [`ptr::read_unaligned`]. For packed structs, this move is
|
|
|
|
|
/// done automatically by the compiler. This means the fields of packed structs
|
|
|
|
|
/// are not dropped in-place.
|
2019-07-03 18:13:42 +00:00
|
|
|
|
///
|
2020-09-08 21:36:36 +00:00
|
|
|
|
/// [`ptr::read`]: self::read
|
|
|
|
|
/// [`ptr::read_unaligned`]: self::read_unaligned
|
|
|
|
|
/// [pinned]: crate::pin
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2017-08-17 21:45:01 +00:00
|
|
|
|
/// # Safety
|
2017-03-13 23:08:21 +00:00
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// Behavior is undefined if any of the following conditions are violated:
|
|
|
|
|
///
|
2020-02-14 23:34:15 +00:00
|
|
|
|
/// * `to_drop` must be [valid] for both reads and writes.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2019-07-03 18:13:42 +00:00
|
|
|
|
/// * `to_drop` must be properly aligned.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2020-02-14 23:34:15 +00:00
|
|
|
|
/// * The value `to_drop` points to must be valid for dropping, which may mean it must uphold
|
|
|
|
|
/// additional invariants - this is type-dependent.
|
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// Additionally, if `T` is not [`Copy`], using the pointed-to value after
|
|
|
|
|
/// calling `drop_in_place` can cause undefined behavior. Note that `*to_drop =
|
2018-11-10 12:31:49 +00:00
|
|
|
|
/// foo` counts as a use because it will cause the value to be dropped
|
2020-09-09 02:24:57 +00:00
|
|
|
|
/// again. [`write()`] can be used to overwrite data without causing it to be
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// dropped.
|
|
|
|
|
///
|
2018-08-30 14:19:05 +00:00
|
|
|
|
/// Note that even if `T` has size `0`, the pointer must be non-NULL and properly aligned.
|
2018-08-29 12:34:59 +00:00
|
|
|
|
///
|
2020-09-09 20:42:57 +00:00
|
|
|
|
/// [valid]: self#safety
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// Manually remove the last item from a vector:
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
/// use std::rc::Rc;
|
|
|
|
|
///
|
|
|
|
|
/// let last = Rc::new(1);
|
|
|
|
|
/// let weak = Rc::downgrade(&last);
|
|
|
|
|
///
|
|
|
|
|
/// let mut v = vec![Rc::new(0), last];
|
|
|
|
|
///
|
|
|
|
|
/// unsafe {
|
2018-08-31 16:25:42 +00:00
|
|
|
|
/// // Get a raw pointer to the last element in `v`.
|
|
|
|
|
/// let ptr = &mut v[1] as *mut _;
|
2019-02-09 21:23:30 +00:00
|
|
|
|
/// // Shorten `v` to prevent the last item from being dropped. We do that first,
|
2018-08-31 15:01:53 +00:00
|
|
|
|
/// // to prevent issues if the `drop_in_place` below panics.
|
|
|
|
|
/// v.set_len(1);
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// // Without a call `drop_in_place`, the last item would never be dropped,
|
|
|
|
|
/// // and the memory it manages would be leaked.
|
2018-08-31 16:25:42 +00:00
|
|
|
|
/// ptr::drop_in_place(ptr);
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// }
|
|
|
|
|
///
|
|
|
|
|
/// assert_eq!(v, &[0.into()]);
|
|
|
|
|
///
|
|
|
|
|
/// // Ensure that the last item was dropped.
|
|
|
|
|
/// assert!(weak.upgrade().is_none());
|
|
|
|
|
/// ```
|
2018-08-30 15:07:24 +00:00
|
|
|
|
///
|
2018-09-17 13:14:19 +00:00
|
|
|
|
/// Notice that the compiler performs this copy automatically when dropping packed structs,
|
|
|
|
|
/// i.e., you do not usually have to worry about such issues unless you call `drop_in_place`
|
|
|
|
|
/// manually.
|
2017-03-13 23:08:21 +00:00
|
|
|
|
#[stable(feature = "drop_in_place", since = "1.8.0")]
|
2017-09-28 08:30:25 +00:00
|
|
|
|
#[lang = "drop_in_place"]
|
2017-03-13 23:08:21 +00:00
|
|
|
|
#[allow(unconditional_recursion)]
|
2020-01-16 21:18:17 +00:00
|
|
|
|
pub unsafe fn drop_in_place<T: ?Sized>(to_drop: *mut T) {
|
2017-03-13 23:08:21 +00:00
|
|
|
|
// Code here does not matter - this is replaced by the
|
|
|
|
|
// real drop glue by the compiler.
|
2020-06-30 17:10:22 +00:00
|
|
|
|
|
|
|
|
|
// SAFETY: see comment above
|
|
|
|
|
unsafe { drop_in_place(to_drop) }
|
2017-03-13 23:08:21 +00:00
|
|
|
|
}
|
|
|
|
|
|
2014-12-09 01:12:35 +00:00
|
|
|
|
/// Creates a null raw pointer.
|
2014-04-07 21:00:19 +00:00
|
|
|
|
///
|
2014-12-09 01:12:35 +00:00
|
|
|
|
/// # Examples
|
2014-04-07 21:00:19 +00:00
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
2015-02-13 19:30:31 +00:00
|
|
|
|
/// let p: *const i32 = ptr::null();
|
2014-04-07 21:00:19 +00:00
|
|
|
|
/// assert!(p.is_null());
|
|
|
|
|
/// ```
|
2019-10-02 12:22:37 +00:00
|
|
|
|
#[inline(always)]
|
2015-01-24 05:48:20 +00:00
|
|
|
|
#[stable(feature = "rust1", since = "1.0.0")]
|
2018-10-31 18:53:17 +00:00
|
|
|
|
#[rustc_promotable]
|
2019-12-18 17:00:59 +00:00
|
|
|
|
#[rustc_const_stable(feature = "const_ptr_null", since = "1.32.0")]
|
2019-12-07 04:18:12 +00:00
|
|
|
|
pub const fn null<T>() -> *const T {
|
|
|
|
|
0 as *const T
|
|
|
|
|
}
|
2012-04-04 04:56:16 +00:00
|
|
|
|
|
2014-12-09 01:12:35 +00:00
|
|
|
|
/// Creates a null mutable raw pointer.
|
2014-04-07 21:00:19 +00:00
|
|
|
|
///
|
2014-12-09 01:12:35 +00:00
|
|
|
|
/// # Examples
|
2014-04-07 21:00:19 +00:00
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
2015-02-13 19:30:31 +00:00
|
|
|
|
/// let p: *mut i32 = ptr::null_mut();
|
2014-04-07 21:00:19 +00:00
|
|
|
|
/// assert!(p.is_null());
|
|
|
|
|
/// ```
|
2019-10-02 12:22:37 +00:00
|
|
|
|
#[inline(always)]
|
2015-01-24 05:48:20 +00:00
|
|
|
|
#[stable(feature = "rust1", since = "1.0.0")]
|
2018-10-31 18:53:17 +00:00
|
|
|
|
#[rustc_promotable]
|
2019-12-18 17:00:59 +00:00
|
|
|
|
#[rustc_const_stable(feature = "const_ptr_null", since = "1.32.0")]
|
2019-12-07 04:18:12 +00:00
|
|
|
|
pub const fn null_mut<T>() -> *mut T {
|
|
|
|
|
0 as *mut T
|
|
|
|
|
}
|
2012-09-14 23:12:18 +00:00
|
|
|
|
|
2019-06-19 07:21:44 +00:00
|
|
|
|
#[repr(C)]
|
|
|
|
|
pub(crate) union Repr<T> {
|
|
|
|
|
pub(crate) rust: *const [T],
|
|
|
|
|
rust_mut: *mut [T],
|
|
|
|
|
pub(crate) raw: FatPtr<T>,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[repr(C)]
|
|
|
|
|
pub(crate) struct FatPtr<T> {
|
|
|
|
|
data: *const T,
|
|
|
|
|
pub(crate) len: usize,
|
|
|
|
|
}
|
|
|
|
|
|
2019-11-05 08:32:47 +00:00
|
|
|
|
/// Forms a raw slice from a pointer and a length.
|
2019-06-19 07:21:44 +00:00
|
|
|
|
///
|
|
|
|
|
/// The `len` argument is the number of **elements**, not the number of bytes.
|
|
|
|
|
///
|
2019-11-05 08:32:47 +00:00
|
|
|
|
/// This function is safe, but actually using the return value is unsafe.
|
2020-09-08 21:36:36 +00:00
|
|
|
|
/// See the documentation of [`slice::from_raw_parts`] for slice safety requirements.
|
2019-11-05 08:32:47 +00:00
|
|
|
|
///
|
2020-09-08 21:36:36 +00:00
|
|
|
|
/// [`slice::from_raw_parts`]: crate::slice::from_raw_parts
|
2019-11-05 08:32:47 +00:00
|
|
|
|
///
|
2019-06-19 07:21:44 +00:00
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// ```rust
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
|
|
|
|
/// // create a slice pointer when starting out with a pointer to the first element
|
2020-01-14 21:45:11 +00:00
|
|
|
|
/// let x = [5, 6, 7];
|
2020-04-23 20:22:27 +00:00
|
|
|
|
/// let raw_pointer = x.as_ptr();
|
|
|
|
|
/// let slice = ptr::slice_from_raw_parts(raw_pointer, 3);
|
2019-06-19 07:21:44 +00:00
|
|
|
|
/// assert_eq!(unsafe { &*slice }[2], 7);
|
|
|
|
|
/// ```
|
|
|
|
|
#[inline]
|
2020-01-14 21:45:11 +00:00
|
|
|
|
#[stable(feature = "slice_from_raw_parts", since = "1.42.0")]
|
2019-12-20 20:42:03 +00:00
|
|
|
|
#[rustc_const_unstable(feature = "const_slice_from_raw_parts", issue = "67456")]
|
|
|
|
|
pub const fn slice_from_raw_parts<T>(data: *const T, len: usize) -> *const [T] {
|
2020-04-23 20:32:20 +00:00
|
|
|
|
// SAFETY: Accessing the value from the `Repr` union is safe since *const [T]
|
|
|
|
|
// and FatPtr have the same memory layouts. Only std can make this
|
|
|
|
|
// guarantee.
|
2019-06-19 07:21:44 +00:00
|
|
|
|
unsafe { Repr { raw: FatPtr { data, len } }.rust }
|
|
|
|
|
}
|
|
|
|
|
|
2019-11-05 08:32:47 +00:00
|
|
|
|
/// Performs the same functionality as [`slice_from_raw_parts`], except that a
|
2019-11-05 20:50:55 +00:00
|
|
|
|
/// raw mutable slice is returned, as opposed to a raw immutable slice.
|
2019-06-19 07:21:44 +00:00
|
|
|
|
///
|
2019-11-05 08:32:47 +00:00
|
|
|
|
/// See the documentation of [`slice_from_raw_parts`] for more details.
|
2019-06-19 07:21:44 +00:00
|
|
|
|
///
|
2019-11-05 08:32:47 +00:00
|
|
|
|
/// This function is safe, but actually using the return value is unsafe.
|
2020-09-08 21:36:36 +00:00
|
|
|
|
/// See the documentation of [`slice::from_raw_parts_mut`] for slice safety requirements.
|
2019-11-05 08:32:47 +00:00
|
|
|
|
///
|
2020-09-08 21:36:36 +00:00
|
|
|
|
/// [`slice::from_raw_parts_mut`]: crate::slice::from_raw_parts_mut
|
2020-04-23 20:32:20 +00:00
|
|
|
|
///
|
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// ```rust
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
|
|
|
|
/// let x = &mut [5, 6, 7];
|
|
|
|
|
/// let raw_pointer = x.as_mut_ptr();
|
|
|
|
|
/// let slice = ptr::slice_from_raw_parts_mut(raw_pointer, 3);
|
|
|
|
|
///
|
|
|
|
|
/// unsafe {
|
|
|
|
|
/// (*slice)[2] = 99; // assign a value at an index in the slice
|
|
|
|
|
/// };
|
|
|
|
|
///
|
|
|
|
|
/// assert_eq!(unsafe { &*slice }[2], 99);
|
|
|
|
|
/// ```
|
2019-06-19 07:21:44 +00:00
|
|
|
|
#[inline]
|
2020-01-14 21:45:11 +00:00
|
|
|
|
#[stable(feature = "slice_from_raw_parts", since = "1.42.0")]
|
2019-12-20 20:42:03 +00:00
|
|
|
|
#[rustc_const_unstable(feature = "const_slice_from_raw_parts", issue = "67456")]
|
|
|
|
|
pub const fn slice_from_raw_parts_mut<T>(data: *mut T, len: usize) -> *mut [T] {
|
2020-04-23 20:32:20 +00:00
|
|
|
|
// SAFETY: Accessing the value from the `Repr` union is safe since *mut [T]
|
|
|
|
|
// and FatPtr have the same memory layouts
|
2019-06-19 07:21:44 +00:00
|
|
|
|
unsafe { Repr { raw: FatPtr { data, len } }.rust_mut }
|
|
|
|
|
}
|
|
|
|
|
|
2014-12-09 01:12:35 +00:00
|
|
|
|
/// Swaps the values at two mutable locations of the same type, without
|
2017-12-03 23:02:17 +00:00
|
|
|
|
/// deinitializing either.
|
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// But for the following two exceptions, this function is semantically
|
|
|
|
|
/// equivalent to [`mem::swap`]:
|
|
|
|
|
///
|
|
|
|
|
/// * It operates on raw pointers instead of references. When references are
|
|
|
|
|
/// available, [`mem::swap`] should be preferred.
|
|
|
|
|
///
|
|
|
|
|
/// * The two pointed-to values may overlap. If the values do overlap, then the
|
|
|
|
|
/// overlapping region of memory from `x` will be used. This is demonstrated
|
2018-09-18 19:14:31 +00:00
|
|
|
|
/// in the second example below.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2014-12-09 01:12:35 +00:00
|
|
|
|
/// # Safety
|
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// Behavior is undefined if any of the following conditions are violated:
|
|
|
|
|
///
|
2020-02-15 12:58:54 +00:00
|
|
|
|
/// * Both `x` and `y` must be [valid] for both reads and writes.
|
2017-01-07 18:41:16 +00:00
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// * Both `x` and `y` must be properly aligned.
|
2017-01-07 18:41:16 +00:00
|
|
|
|
///
|
2018-08-30 14:19:05 +00:00
|
|
|
|
/// Note that even if `T` has size `0`, the pointers must be non-NULL and properly aligned.
|
2018-08-29 12:34:59 +00:00
|
|
|
|
///
|
2020-09-09 20:42:57 +00:00
|
|
|
|
/// [valid]: self#safety
|
2017-12-03 23:02:17 +00:00
|
|
|
|
///
|
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// Swapping two non-overlapping regions:
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
|
|
|
|
/// let mut array = [0, 1, 2, 3];
|
|
|
|
|
///
|
2018-09-18 19:14:31 +00:00
|
|
|
|
/// let x = array[0..].as_mut_ptr() as *mut [u32; 2]; // this is `array[0..2]`
|
|
|
|
|
/// let y = array[2..].as_mut_ptr() as *mut [u32; 2]; // this is `array[2..4]`
|
2017-12-03 23:02:17 +00:00
|
|
|
|
///
|
|
|
|
|
/// unsafe {
|
|
|
|
|
/// ptr::swap(x, y);
|
|
|
|
|
/// assert_eq!([2, 3, 0, 1], array);
|
|
|
|
|
/// }
|
|
|
|
|
/// ```
|
|
|
|
|
///
|
|
|
|
|
/// Swapping two overlapping regions:
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
|
|
|
|
/// let mut array = [0, 1, 2, 3];
|
|
|
|
|
///
|
2018-09-18 19:14:31 +00:00
|
|
|
|
/// let x = array[0..].as_mut_ptr() as *mut [u32; 3]; // this is `array[0..3]`
|
|
|
|
|
/// let y = array[1..].as_mut_ptr() as *mut [u32; 3]; // this is `array[1..4]`
|
2017-12-03 23:02:17 +00:00
|
|
|
|
///
|
|
|
|
|
/// unsafe {
|
|
|
|
|
/// ptr::swap(x, y);
|
2018-09-18 19:14:31 +00:00
|
|
|
|
/// // The indices `1..3` of the slice overlap between `x` and `y`.
|
|
|
|
|
/// // Reasonable results would be for to them be `[2, 3]`, so that indices `0..3` are
|
|
|
|
|
/// // `[1, 2, 3]` (matching `y` before the `swap`); or for them to be `[0, 1]`
|
|
|
|
|
/// // so that indices `1..4` are `[0, 1, 2]` (matching `x` before the `swap`).
|
|
|
|
|
/// // This implementation is defined to make the latter choice.
|
2017-12-03 23:02:17 +00:00
|
|
|
|
/// assert_eq!([1, 0, 1, 2], array);
|
|
|
|
|
/// }
|
|
|
|
|
/// ```
|
2013-05-31 14:21:29 +00:00
|
|
|
|
#[inline]
|
2015-01-24 05:48:20 +00:00
|
|
|
|
#[stable(feature = "rust1", since = "1.0.0")]
|
2014-02-14 23:42:01 +00:00
|
|
|
|
pub unsafe fn swap<T>(x: *mut T, y: *mut T) {
|
2018-10-12 06:59:38 +00:00
|
|
|
|
// Give ourselves some scratch space to work with.
|
|
|
|
|
// We do not have to worry about drops: `MaybeUninit` does nothing when dropped.
|
2019-03-18 21:45:02 +00:00
|
|
|
|
let mut tmp = MaybeUninit::<T>::uninit();
|
2013-05-31 14:21:29 +00:00
|
|
|
|
|
|
|
|
|
// Perform the swap
|
2020-06-30 17:10:22 +00:00
|
|
|
|
// SAFETY: the caller must guarantee that `x` and `y` are
|
|
|
|
|
// valid for writes and properly aligned. `tmp` cannot be
|
|
|
|
|
// overlapping either `x` or `y` because `tmp` was just allocated
|
|
|
|
|
// on the stack as a separate allocated object.
|
|
|
|
|
unsafe {
|
|
|
|
|
copy_nonoverlapping(x, tmp.as_mut_ptr(), 1);
|
|
|
|
|
copy(y, x, 1); // `x` and `y` may overlap
|
|
|
|
|
copy_nonoverlapping(tmp.as_ptr(), y, 1);
|
|
|
|
|
}
|
2013-05-31 14:21:29 +00:00
|
|
|
|
}
|
|
|
|
|
|
2018-05-29 00:41:19 +00:00
|
|
|
|
/// Swaps `count * size_of::<T>()` bytes between the two regions of memory
|
|
|
|
|
/// beginning at `x` and `y`. The two regions must *not* overlap.
|
2017-06-21 06:48:15 +00:00
|
|
|
|
///
|
|
|
|
|
/// # Safety
|
|
|
|
|
///
|
2018-05-29 00:41:19 +00:00
|
|
|
|
/// Behavior is undefined if any of the following conditions are violated:
|
|
|
|
|
///
|
2020-02-15 12:58:54 +00:00
|
|
|
|
/// * Both `x` and `y` must be [valid] for both reads and writes of `count *
|
2018-06-17 09:45:38 +00:00
|
|
|
|
/// size_of::<T>()` bytes.
|
2018-05-29 00:41:19 +00:00
|
|
|
|
///
|
2018-06-17 09:45:38 +00:00
|
|
|
|
/// * Both `x` and `y` must be properly aligned.
|
2018-05-29 00:41:19 +00:00
|
|
|
|
///
|
2018-06-17 09:45:38 +00:00
|
|
|
|
/// * The region of memory beginning at `x` with a size of `count *
|
|
|
|
|
/// size_of::<T>()` bytes must *not* overlap with the region of memory
|
|
|
|
|
/// beginning at `y` with the same size.
|
2018-05-29 00:41:19 +00:00
|
|
|
|
///
|
2018-08-30 14:26:48 +00:00
|
|
|
|
/// Note that even if the effectively copied size (`count * size_of::<T>()`) is `0`,
|
|
|
|
|
/// the pointers must be non-NULL and properly aligned.
|
2018-08-29 12:34:59 +00:00
|
|
|
|
///
|
2020-09-09 20:42:57 +00:00
|
|
|
|
/// [valid]: self#safety
|
2017-06-21 06:48:15 +00:00
|
|
|
|
///
|
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// Basic usage:
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
|
|
|
|
/// let mut x = [1, 2, 3, 4];
|
|
|
|
|
/// let mut y = [7, 8, 9];
|
|
|
|
|
///
|
|
|
|
|
/// unsafe {
|
|
|
|
|
/// ptr::swap_nonoverlapping(x.as_mut_ptr(), y.as_mut_ptr(), 2);
|
|
|
|
|
/// }
|
|
|
|
|
///
|
|
|
|
|
/// assert_eq!(x, [7, 8, 3, 4]);
|
|
|
|
|
/// assert_eq!(y, [1, 2, 9]);
|
|
|
|
|
/// ```
|
|
|
|
|
#[inline]
|
2018-04-17 04:36:13 +00:00
|
|
|
|
#[stable(feature = "swap_nonoverlapping", since = "1.27.0")]
|
2017-06-21 06:48:15 +00:00
|
|
|
|
pub unsafe fn swap_nonoverlapping<T>(x: *mut T, y: *mut T, count: usize) {
|
2020-06-06 09:56:58 +00:00
|
|
|
|
if cfg!(debug_assertions)
|
|
|
|
|
&& !(is_aligned_and_not_null(x)
|
|
|
|
|
&& is_aligned_and_not_null(y)
|
|
|
|
|
&& is_nonoverlapping(x, y, count))
|
|
|
|
|
{
|
|
|
|
|
// Not panicking to keep codegen impact smaller.
|
|
|
|
|
abort();
|
|
|
|
|
}
|
2020-02-16 13:44:16 +00:00
|
|
|
|
|
2017-06-21 06:48:15 +00:00
|
|
|
|
let x = x as *mut u8;
|
|
|
|
|
let y = y as *mut u8;
|
|
|
|
|
let len = mem::size_of::<T>() * count;
|
2020-06-30 17:10:22 +00:00
|
|
|
|
// SAFETY: the caller must guarantee that `x` and `y` are
|
|
|
|
|
// valid for writes and properly aligned.
|
|
|
|
|
unsafe { swap_nonoverlapping_bytes(x, y, len) }
|
2017-06-21 06:48:15 +00:00
|
|
|
|
}
|
|
|
|
|
|
2018-07-04 09:48:30 +00:00
|
|
|
|
#[inline]
|
|
|
|
|
pub(crate) unsafe fn swap_nonoverlapping_one<T>(x: *mut T, y: *mut T) {
|
|
|
|
|
// For types smaller than the block optimization below,
|
|
|
|
|
// just swap directly to avoid pessimizing codegen.
|
|
|
|
|
if mem::size_of::<T>() < 32 {
|
2020-06-30 17:10:22 +00:00
|
|
|
|
// SAFETY: the caller must guarantee that `x` and `y` are valid
|
|
|
|
|
// for writes, properly aligned, and non-overlapping.
|
|
|
|
|
unsafe {
|
|
|
|
|
let z = read(x);
|
|
|
|
|
copy_nonoverlapping(y, x, 1);
|
|
|
|
|
write(y, z);
|
|
|
|
|
}
|
2018-07-04 09:48:30 +00:00
|
|
|
|
} else {
|
2020-06-30 17:10:22 +00:00
|
|
|
|
// SAFETY: the caller must uphold the safety contract for `swap_nonoverlapping`.
|
|
|
|
|
unsafe { swap_nonoverlapping(x, y, 1) };
|
2018-07-04 09:48:30 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2017-06-21 06:48:15 +00:00
|
|
|
|
#[inline]
|
|
|
|
|
unsafe fn swap_nonoverlapping_bytes(x: *mut u8, y: *mut u8, len: usize) {
|
|
|
|
|
// The approach here is to utilize simd to swap x & y efficiently. Testing reveals
|
2018-11-10 12:31:49 +00:00
|
|
|
|
// that swapping either 32 bytes or 64 bytes at a time is most efficient for Intel
|
2017-06-21 06:48:15 +00:00
|
|
|
|
// Haswell E processors. LLVM is more able to optimize if we give a struct a
|
|
|
|
|
// #[repr(simd)], even if we don't actually use this struct directly.
|
|
|
|
|
//
|
|
|
|
|
// FIXME repr(simd) broken on emscripten and redox
|
2019-05-06 20:02:12 +00:00
|
|
|
|
#[cfg_attr(not(any(target_os = "emscripten", target_os = "redox")), repr(simd))]
|
2017-06-21 06:48:15 +00:00
|
|
|
|
struct Block(u64, u64, u64, u64);
|
|
|
|
|
struct UnalignedBlock(u64, u64, u64, u64);
|
|
|
|
|
|
|
|
|
|
let block_size = mem::size_of::<Block>();
|
|
|
|
|
|
|
|
|
|
// Loop through x & y, copying them `Block` at a time
|
|
|
|
|
// The optimizer should unroll the loop fully for most types
|
|
|
|
|
// N.B. We can't use a for loop as the `range` impl calls `mem::swap` recursively
|
|
|
|
|
let mut i = 0;
|
|
|
|
|
while i + block_size <= len {
|
|
|
|
|
// Create some uninitialized memory as scratch space
|
|
|
|
|
// Declaring `t` here avoids aligning the stack when this loop is unused
|
2019-03-18 21:45:02 +00:00
|
|
|
|
let mut t = mem::MaybeUninit::<Block>::uninit();
|
2018-10-10 10:04:07 +00:00
|
|
|
|
let t = t.as_mut_ptr() as *mut u8;
|
2017-06-21 06:48:15 +00:00
|
|
|
|
|
2020-06-30 17:10:22 +00:00
|
|
|
|
// SAFETY: As `i < len`, and as the caller must guarantee that `x` and `y` are valid
|
|
|
|
|
// for `len` bytes, `x + i` and `y + i` must be valid adresses, which fulfills the
|
|
|
|
|
// safety contract for `add`.
|
|
|
|
|
//
|
|
|
|
|
// Also, the caller must guarantee that `x` and `y` are valid for writes, properly aligned,
|
|
|
|
|
// and non-overlapping, which fulfills the safety contract for `copy_nonoverlapping`.
|
|
|
|
|
unsafe {
|
|
|
|
|
let x = x.add(i);
|
|
|
|
|
let y = y.add(i);
|
|
|
|
|
|
|
|
|
|
// Swap a block of bytes of x & y, using t as a temporary buffer
|
|
|
|
|
// This should be optimized into efficient SIMD operations where available
|
|
|
|
|
copy_nonoverlapping(x, t, block_size);
|
|
|
|
|
copy_nonoverlapping(y, x, block_size);
|
|
|
|
|
copy_nonoverlapping(t, y, block_size);
|
|
|
|
|
}
|
2017-06-21 06:48:15 +00:00
|
|
|
|
i += block_size;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if i < len {
|
|
|
|
|
// Swap any remaining bytes
|
2019-03-18 21:45:02 +00:00
|
|
|
|
let mut t = mem::MaybeUninit::<UnalignedBlock>::uninit();
|
2017-06-21 06:48:15 +00:00
|
|
|
|
let rem = len - i;
|
|
|
|
|
|
2018-10-10 10:04:07 +00:00
|
|
|
|
let t = t.as_mut_ptr() as *mut u8;
|
2017-06-21 06:48:15 +00:00
|
|
|
|
|
2020-06-30 17:10:22 +00:00
|
|
|
|
// SAFETY: see previous safety comment.
|
|
|
|
|
unsafe {
|
|
|
|
|
let x = x.add(i);
|
|
|
|
|
let y = y.add(i);
|
|
|
|
|
|
|
|
|
|
copy_nonoverlapping(x, t, rem);
|
|
|
|
|
copy_nonoverlapping(y, x, rem);
|
|
|
|
|
copy_nonoverlapping(t, y, rem);
|
|
|
|
|
}
|
2017-06-21 06:48:15 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2018-08-30 14:26:48 +00:00
|
|
|
|
/// Moves `src` into the pointed `dst`, returning the previous `dst` value.
|
2018-05-28 12:36:14 +00:00
|
|
|
|
///
|
|
|
|
|
/// Neither value is dropped.
|
2014-12-09 01:12:35 +00:00
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// This function is semantically equivalent to [`mem::replace`] except that it
|
|
|
|
|
/// operates on raw pointers instead of references. When references are
|
|
|
|
|
/// available, [`mem::replace`] should be preferred.
|
|
|
|
|
///
|
2014-12-09 01:12:35 +00:00
|
|
|
|
/// # Safety
|
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// Behavior is undefined if any of the following conditions are violated:
|
|
|
|
|
///
|
2020-01-31 14:18:27 +00:00
|
|
|
|
/// * `dst` must be [valid] for both reads and writes.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2018-08-30 14:26:48 +00:00
|
|
|
|
/// * `dst` must be properly aligned.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2020-01-31 14:22:51 +00:00
|
|
|
|
/// * `dst` must point to a properly initialized value of type `T`.
|
|
|
|
|
///
|
2018-08-30 14:19:05 +00:00
|
|
|
|
/// Note that even if `T` has size `0`, the pointer must be non-NULL and properly aligned.
|
2018-08-29 12:34:59 +00:00
|
|
|
|
///
|
2020-09-09 20:42:57 +00:00
|
|
|
|
/// [valid]: self#safety
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
|
|
|
|
/// let mut rust = vec!['b', 'u', 's', 't'];
|
|
|
|
|
///
|
|
|
|
|
/// // `mem::replace` would have the same effect without requiring the unsafe
|
|
|
|
|
/// // block.
|
|
|
|
|
/// let b = unsafe {
|
|
|
|
|
/// ptr::replace(&mut rust[0], 'r')
|
|
|
|
|
/// };
|
|
|
|
|
///
|
|
|
|
|
/// assert_eq!(b, 'b');
|
|
|
|
|
/// assert_eq!(rust, &['r', 'u', 's', 't']);
|
|
|
|
|
/// ```
|
2013-06-18 21:45:18 +00:00
|
|
|
|
#[inline]
|
2015-01-24 05:48:20 +00:00
|
|
|
|
#[stable(feature = "rust1", since = "1.0.0")]
|
2018-08-30 14:26:48 +00:00
|
|
|
|
pub unsafe fn replace<T>(dst: *mut T, mut src: T) -> T {
|
2020-06-30 17:10:22 +00:00
|
|
|
|
// SAFETY: the caller must guarantee that `dst` is valid to be
|
|
|
|
|
// cast to a mutable reference (valid for writes, aligned, initialized),
|
|
|
|
|
// and cannot overlap `src` since `dst` must point to a distinct
|
|
|
|
|
// allocated object.
|
|
|
|
|
unsafe {
|
|
|
|
|
mem::swap(&mut *dst, &mut src); // cannot overlap
|
|
|
|
|
}
|
2013-05-31 14:21:29 +00:00
|
|
|
|
src
|
|
|
|
|
}
|
|
|
|
|
|
2015-02-06 00:57:28 +00:00
|
|
|
|
/// Reads the value from `src` without moving it. This leaves the
|
2014-12-09 01:12:35 +00:00
|
|
|
|
/// memory in `src` unchanged.
|
|
|
|
|
///
|
|
|
|
|
/// # Safety
|
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// Behavior is undefined if any of the following conditions are violated:
|
|
|
|
|
///
|
2018-06-17 09:45:38 +00:00
|
|
|
|
/// * `src` must be [valid] for reads.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
|
|
|
|
/// * `src` must be properly aligned. Use [`read_unaligned`] if this is not the
|
|
|
|
|
/// case.
|
2016-04-14 16:42:00 +00:00
|
|
|
|
///
|
2020-01-31 14:22:51 +00:00
|
|
|
|
/// * `src` must point to a properly initialized value of type `T`.
|
|
|
|
|
///
|
2018-08-30 14:19:05 +00:00
|
|
|
|
/// Note that even if `T` has size `0`, the pointer must be non-NULL and properly aligned.
|
2016-12-12 02:51:22 +00:00
|
|
|
|
///
|
2016-04-14 16:42:00 +00:00
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// Basic usage:
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// let x = 12;
|
|
|
|
|
/// let y = &x as *const i32;
|
|
|
|
|
///
|
2016-08-21 20:49:09 +00:00
|
|
|
|
/// unsafe {
|
|
|
|
|
/// assert_eq!(std::ptr::read(y), 12);
|
|
|
|
|
/// }
|
2016-04-14 16:42:00 +00:00
|
|
|
|
/// ```
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
|
|
|
|
/// Manually implement [`mem::swap`]:
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
|
|
|
|
/// fn swap<T>(a: &mut T, b: &mut T) {
|
|
|
|
|
/// unsafe {
|
|
|
|
|
/// // Create a bitwise copy of the value at `a` in `tmp`.
|
|
|
|
|
/// let tmp = ptr::read(a);
|
|
|
|
|
///
|
|
|
|
|
/// // Exiting at this point (either by explicitly returning or by
|
|
|
|
|
/// // calling a function which panics) would cause the value in `tmp` to
|
|
|
|
|
/// // be dropped while the same value is still referenced by `a`. This
|
|
|
|
|
/// // could trigger undefined behavior if `T` is not `Copy`.
|
|
|
|
|
///
|
|
|
|
|
/// // Create a bitwise copy of the value at `b` in `a`.
|
|
|
|
|
/// // This is safe because mutable references cannot alias.
|
|
|
|
|
/// ptr::copy_nonoverlapping(b, a, 1);
|
|
|
|
|
///
|
|
|
|
|
/// // As above, exiting here could trigger undefined behavior because
|
|
|
|
|
/// // the same value is referenced by `a` and `b`.
|
|
|
|
|
///
|
|
|
|
|
/// // Move `tmp` into `b`.
|
|
|
|
|
/// ptr::write(b, tmp);
|
2018-09-17 13:14:19 +00:00
|
|
|
|
///
|
|
|
|
|
/// // `tmp` has been moved (`write` takes ownership of its second argument),
|
|
|
|
|
/// // so nothing is dropped implicitly here.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// }
|
|
|
|
|
/// }
|
|
|
|
|
///
|
|
|
|
|
/// let mut foo = "foo".to_owned();
|
|
|
|
|
/// let mut bar = "bar".to_owned();
|
|
|
|
|
///
|
|
|
|
|
/// swap(&mut foo, &mut bar);
|
|
|
|
|
///
|
|
|
|
|
/// assert_eq!(foo, "bar");
|
|
|
|
|
/// assert_eq!(bar, "foo");
|
|
|
|
|
/// ```
|
|
|
|
|
///
|
2018-09-17 13:21:20 +00:00
|
|
|
|
/// ## Ownership of the Returned Value
|
|
|
|
|
///
|
|
|
|
|
/// `read` creates a bitwise copy of `T`, regardless of whether `T` is [`Copy`].
|
|
|
|
|
/// If `T` is not [`Copy`], using both the returned value and the value at
|
2019-02-09 21:23:30 +00:00
|
|
|
|
/// `*src` can violate memory safety. Note that assigning to `*src` counts as a
|
2018-09-17 13:21:20 +00:00
|
|
|
|
/// use because it will attempt to drop the value at `*src`.
|
|
|
|
|
///
|
2020-09-09 02:24:57 +00:00
|
|
|
|
/// [`write()`] can be used to overwrite data without causing it to be dropped.
|
2018-09-17 13:21:20 +00:00
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
|
|
|
|
/// let mut s = String::from("foo");
|
|
|
|
|
/// unsafe {
|
|
|
|
|
/// // `s2` now points to the same underlying memory as `s`.
|
|
|
|
|
/// let mut s2: String = ptr::read(&s);
|
|
|
|
|
///
|
|
|
|
|
/// assert_eq!(s2, "foo");
|
|
|
|
|
///
|
|
|
|
|
/// // Assigning to `s2` causes its original value to be dropped. Beyond
|
|
|
|
|
/// // this point, `s` must no longer be used, as the underlying memory has
|
|
|
|
|
/// // been freed.
|
|
|
|
|
/// s2 = String::default();
|
|
|
|
|
/// assert_eq!(s2, "");
|
|
|
|
|
///
|
|
|
|
|
/// // Assigning to `s` would cause the old value to be dropped again,
|
|
|
|
|
/// // resulting in undefined behavior.
|
|
|
|
|
/// // s = String::from("bar"); // ERROR
|
|
|
|
|
///
|
|
|
|
|
/// // `ptr::write` can be used to overwrite a value without dropping it.
|
|
|
|
|
/// ptr::write(&mut s, String::from("bar"));
|
|
|
|
|
/// }
|
|
|
|
|
///
|
|
|
|
|
/// assert_eq!(s, "bar");
|
|
|
|
|
/// ```
|
|
|
|
|
///
|
2020-09-09 20:42:57 +00:00
|
|
|
|
/// [valid]: self#safety
|
2017-07-20 18:14:13 +00:00
|
|
|
|
#[inline]
|
2015-01-24 05:48:20 +00:00
|
|
|
|
#[stable(feature = "rust1", since = "1.0.0")]
|
2014-06-25 19:47:34 +00:00
|
|
|
|
pub unsafe fn read<T>(src: *const T) -> T {
|
2020-02-16 13:44:16 +00:00
|
|
|
|
// `copy_nonoverlapping` takes care of debug_assert.
|
2019-03-18 21:45:02 +00:00
|
|
|
|
let mut tmp = MaybeUninit::<T>::uninit();
|
2020-06-30 17:10:22 +00:00
|
|
|
|
// SAFETY: the caller must guarantee that `src` is valid for reads.
|
|
|
|
|
// `src` cannot overlap `tmp` because `tmp` was just allocated on
|
|
|
|
|
// the stack as a separate allocated object.
|
|
|
|
|
//
|
|
|
|
|
// Also, since we just wrote a valid value into `tmp`, it is guaranteed
|
|
|
|
|
// to be properly initialized.
|
|
|
|
|
unsafe {
|
|
|
|
|
copy_nonoverlapping(src, tmp.as_mut_ptr(), 1);
|
|
|
|
|
tmp.assume_init()
|
|
|
|
|
}
|
2013-06-20 19:13:22 +00:00
|
|
|
|
}
|
|
|
|
|
|
2016-12-12 02:51:22 +00:00
|
|
|
|
/// Reads the value from `src` without moving it. This leaves the
|
|
|
|
|
/// memory in `src` unchanged.
|
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// Unlike [`read`], `read_unaligned` works with unaligned pointers.
|
2016-12-12 02:51:22 +00:00
|
|
|
|
///
|
|
|
|
|
/// # Safety
|
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// Behavior is undefined if any of the following conditions are violated:
|
|
|
|
|
///
|
2018-06-17 09:45:38 +00:00
|
|
|
|
/// * `src` must be [valid] for reads.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2020-01-31 14:22:51 +00:00
|
|
|
|
/// * `src` must point to a properly initialized value of type `T`.
|
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// Like [`read`], `read_unaligned` creates a bitwise copy of `T`, regardless of
|
2019-02-09 21:23:30 +00:00
|
|
|
|
/// whether `T` is [`Copy`]. If `T` is not [`Copy`], using both the returned
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// value and the value at `*src` can [violate memory safety][read-ownership].
|
|
|
|
|
///
|
2018-12-28 14:00:33 +00:00
|
|
|
|
/// Note that even if `T` has size `0`, the pointer must be non-NULL.
|
2018-08-29 12:34:59 +00:00
|
|
|
|
///
|
2020-09-08 21:36:36 +00:00
|
|
|
|
/// [read-ownership]: read#ownership-of-the-returned-value
|
2020-09-09 20:42:57 +00:00
|
|
|
|
/// [valid]: self#safety
|
2016-12-12 02:51:22 +00:00
|
|
|
|
///
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// ## On `packed` structs
|
2016-12-12 02:51:22 +00:00
|
|
|
|
///
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// It is currently impossible to create raw pointers to unaligned fields
|
|
|
|
|
/// of a packed struct.
|
2016-12-12 02:51:22 +00:00
|
|
|
|
///
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// Attempting to create a raw pointer to an `unaligned` struct field with
|
|
|
|
|
/// an expression such as `&packed.unaligned as *const FieldType` creates an
|
|
|
|
|
/// intermediate unaligned reference before converting that to a raw pointer.
|
|
|
|
|
/// That this reference is temporary and immediately cast is inconsequential
|
|
|
|
|
/// as the compiler always expects references to be properly aligned.
|
|
|
|
|
/// As a result, using `&packed.unaligned as *const FieldType` causes immediate
|
|
|
|
|
/// *undefined behavior* in your program.
|
2016-12-12 02:51:22 +00:00
|
|
|
|
///
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// An example of what not to do and how this relates to `read_unaligned` is:
|
|
|
|
|
///
|
2019-07-04 07:54:37 +00:00
|
|
|
|
/// ```no_run
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// #[repr(packed, C)]
|
|
|
|
|
/// struct Packed {
|
|
|
|
|
/// _padding: u8,
|
|
|
|
|
/// unaligned: u32,
|
2016-12-12 02:51:22 +00:00
|
|
|
|
/// }
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// let packed = Packed {
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// _padding: 0x00,
|
|
|
|
|
/// unaligned: 0x01020304,
|
|
|
|
|
/// };
|
|
|
|
|
///
|
|
|
|
|
/// let v = unsafe {
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// // Here we attempt to take the address of a 32-bit integer which is not aligned.
|
|
|
|
|
/// let unaligned =
|
|
|
|
|
/// // A temporary unaligned reference is created here which results in
|
|
|
|
|
/// // undefined behavior regardless of whether the reference is used or not.
|
|
|
|
|
/// &packed.unaligned
|
|
|
|
|
/// // Casting to a raw pointer doesn't help; the mistake already happened.
|
|
|
|
|
/// as *const u32;
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// let v = std::ptr::read_unaligned(unaligned);
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
|
|
|
|
/// v
|
|
|
|
|
/// };
|
2016-12-12 02:51:22 +00:00
|
|
|
|
/// ```
|
2019-07-03 02:05:05 +00:00
|
|
|
|
///
|
2019-07-04 07:54:37 +00:00
|
|
|
|
/// Accessing unaligned fields directly with e.g. `packed.unaligned` is safe however.
|
2019-07-03 02:05:05 +00:00
|
|
|
|
// FIXME: Update docs based on outcome of RFC #2582 and friends.
|
2019-07-08 14:03:29 +00:00
|
|
|
|
///
|
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// Read an usize value from a byte buffer:
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::mem;
|
|
|
|
|
///
|
|
|
|
|
/// fn read_usize(x: &[u8]) -> usize {
|
|
|
|
|
/// assert!(x.len() >= mem::size_of::<usize>());
|
|
|
|
|
///
|
|
|
|
|
/// let ptr = x.as_ptr() as *const usize;
|
|
|
|
|
///
|
|
|
|
|
/// unsafe { ptr.read_unaligned() }
|
|
|
|
|
/// }
|
|
|
|
|
/// ```
|
2017-07-20 18:14:13 +00:00
|
|
|
|
#[inline]
|
2017-03-15 03:51:29 +00:00
|
|
|
|
#[stable(feature = "ptr_unaligned", since = "1.17.0")]
|
2016-12-12 02:51:22 +00:00
|
|
|
|
pub unsafe fn read_unaligned<T>(src: *const T) -> T {
|
2020-02-16 13:44:16 +00:00
|
|
|
|
// `copy_nonoverlapping` takes care of debug_assert.
|
2019-03-18 21:45:02 +00:00
|
|
|
|
let mut tmp = MaybeUninit::<T>::uninit();
|
2020-06-30 17:10:22 +00:00
|
|
|
|
// SAFETY: the caller must guarantee that `src` is valid for reads.
|
|
|
|
|
// `src` cannot overlap `tmp` because `tmp` was just allocated on
|
|
|
|
|
// the stack as a separate allocated object.
|
|
|
|
|
//
|
|
|
|
|
// Also, since we just wrote a valid value into `tmp`, it is guaranteed
|
|
|
|
|
// to be properly initialized.
|
|
|
|
|
unsafe {
|
|
|
|
|
copy_nonoverlapping(src as *const u8, tmp.as_mut_ptr() as *mut u8, mem::size_of::<T>());
|
|
|
|
|
tmp.assume_init()
|
|
|
|
|
}
|
2016-12-12 02:51:22 +00:00
|
|
|
|
}
|
|
|
|
|
|
2014-12-19 16:57:12 +00:00
|
|
|
|
/// Overwrites a memory location with the given value without reading or
|
|
|
|
|
/// dropping the old value.
|
2014-05-30 00:40:18 +00:00
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// `write` does not drop the contents of `dst`. This is safe, but it could leak
|
2018-08-30 15:07:24 +00:00
|
|
|
|
/// allocations or resources, so care should be taken not to overwrite an object
|
2015-10-11 11:40:47 +00:00
|
|
|
|
/// that should be dropped.
|
2014-12-09 01:12:35 +00:00
|
|
|
|
///
|
2017-03-09 12:45:52 +00:00
|
|
|
|
/// Additionally, it does not drop `src`. Semantically, `src` is moved into the
|
|
|
|
|
/// location pointed to by `dst`.
|
2017-03-07 22:39:49 +00:00
|
|
|
|
///
|
2015-01-07 01:53:18 +00:00
|
|
|
|
/// This is appropriate for initializing uninitialized memory, or overwriting
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// memory that has previously been [`read`] from.
|
2016-04-14 16:42:00 +00:00
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// # Safety
|
|
|
|
|
///
|
|
|
|
|
/// Behavior is undefined if any of the following conditions are violated:
|
2016-04-14 16:42:00 +00:00
|
|
|
|
///
|
2018-06-17 09:45:38 +00:00
|
|
|
|
/// * `dst` must be [valid] for writes.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
|
|
|
|
/// * `dst` must be properly aligned. Use [`write_unaligned`] if this is not the
|
|
|
|
|
/// case.
|
|
|
|
|
///
|
2018-08-30 14:19:05 +00:00
|
|
|
|
/// Note that even if `T` has size `0`, the pointer must be non-NULL and properly aligned.
|
2018-08-29 12:34:59 +00:00
|
|
|
|
///
|
2020-09-09 20:42:57 +00:00
|
|
|
|
/// [valid]: self#safety
|
2016-12-12 02:51:22 +00:00
|
|
|
|
///
|
2016-04-14 16:42:00 +00:00
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// Basic usage:
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// let mut x = 0;
|
|
|
|
|
/// let y = &mut x as *mut i32;
|
|
|
|
|
/// let z = 12;
|
|
|
|
|
///
|
|
|
|
|
/// unsafe {
|
|
|
|
|
/// std::ptr::write(y, z);
|
2016-08-21 20:49:09 +00:00
|
|
|
|
/// assert_eq!(std::ptr::read(y), 12);
|
2016-04-14 16:42:00 +00:00
|
|
|
|
/// }
|
|
|
|
|
/// ```
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
|
|
|
|
/// Manually implement [`mem::swap`]:
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
|
|
|
|
/// fn swap<T>(a: &mut T, b: &mut T) {
|
|
|
|
|
/// unsafe {
|
2018-09-17 13:14:19 +00:00
|
|
|
|
/// // Create a bitwise copy of the value at `a` in `tmp`.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// let tmp = ptr::read(a);
|
2018-09-17 13:14:19 +00:00
|
|
|
|
///
|
|
|
|
|
/// // Exiting at this point (either by explicitly returning or by
|
|
|
|
|
/// // calling a function which panics) would cause the value in `tmp` to
|
|
|
|
|
/// // be dropped while the same value is still referenced by `a`. This
|
|
|
|
|
/// // could trigger undefined behavior if `T` is not `Copy`.
|
|
|
|
|
///
|
|
|
|
|
/// // Create a bitwise copy of the value at `b` in `a`.
|
|
|
|
|
/// // This is safe because mutable references cannot alias.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// ptr::copy_nonoverlapping(b, a, 1);
|
2018-09-17 13:14:19 +00:00
|
|
|
|
///
|
|
|
|
|
/// // As above, exiting here could trigger undefined behavior because
|
|
|
|
|
/// // the same value is referenced by `a` and `b`.
|
|
|
|
|
///
|
|
|
|
|
/// // Move `tmp` into `b`.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// ptr::write(b, tmp);
|
2018-09-17 13:14:19 +00:00
|
|
|
|
///
|
|
|
|
|
/// // `tmp` has been moved (`write` takes ownership of its second argument),
|
|
|
|
|
/// // so nothing is dropped implicitly here.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// }
|
|
|
|
|
/// }
|
|
|
|
|
///
|
|
|
|
|
/// let mut foo = "foo".to_owned();
|
|
|
|
|
/// let mut bar = "bar".to_owned();
|
|
|
|
|
///
|
|
|
|
|
/// swap(&mut foo, &mut bar);
|
|
|
|
|
///
|
|
|
|
|
/// assert_eq!(foo, "bar");
|
|
|
|
|
/// assert_eq!(bar, "foo");
|
|
|
|
|
/// ```
|
2014-05-30 00:40:18 +00:00
|
|
|
|
#[inline]
|
2015-01-24 05:48:20 +00:00
|
|
|
|
#[stable(feature = "rust1", since = "1.0.0")]
|
2014-05-30 00:40:18 +00:00
|
|
|
|
pub unsafe fn write<T>(dst: *mut T, src: T) {
|
2020-06-06 09:56:58 +00:00
|
|
|
|
if cfg!(debug_assertions) && !is_aligned_and_not_null(dst) {
|
|
|
|
|
// Not panicking to keep codegen impact smaller.
|
|
|
|
|
abort();
|
|
|
|
|
}
|
2020-06-30 17:10:22 +00:00
|
|
|
|
// SAFETY: the caller must uphold the safety contract for `move_val_init`.
|
|
|
|
|
unsafe { intrinsics::move_val_init(&mut *dst, src) }
|
2014-05-30 00:40:18 +00:00
|
|
|
|
}
|
|
|
|
|
|
2016-12-12 02:51:22 +00:00
|
|
|
|
/// Overwrites a memory location with the given value without reading or
|
|
|
|
|
/// dropping the old value.
|
|
|
|
|
///
|
2020-09-09 02:24:57 +00:00
|
|
|
|
/// Unlike [`write()`], the pointer may be unaligned.
|
2018-05-17 17:19:41 +00:00
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// `write_unaligned` does not drop the contents of `dst`. This is safe, but it
|
2018-08-30 15:07:24 +00:00
|
|
|
|
/// could leak allocations or resources, so care should be taken not to overwrite
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// an object that should be dropped.
|
2016-12-12 02:51:22 +00:00
|
|
|
|
///
|
2017-03-09 12:45:52 +00:00
|
|
|
|
/// Additionally, it does not drop `src`. Semantically, `src` is moved into the
|
|
|
|
|
/// location pointed to by `dst`.
|
|
|
|
|
///
|
2016-12-12 02:51:22 +00:00
|
|
|
|
/// This is appropriate for initializing uninitialized memory, or overwriting
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// memory that has previously been read with [`read_unaligned`].
|
|
|
|
|
///
|
|
|
|
|
/// # Safety
|
|
|
|
|
///
|
|
|
|
|
/// Behavior is undefined if any of the following conditions are violated:
|
|
|
|
|
///
|
2018-06-17 09:45:38 +00:00
|
|
|
|
/// * `dst` must be [valid] for writes.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2018-12-28 14:00:33 +00:00
|
|
|
|
/// Note that even if `T` has size `0`, the pointer must be non-NULL.
|
2018-08-29 12:34:59 +00:00
|
|
|
|
///
|
2020-09-09 20:42:57 +00:00
|
|
|
|
/// [valid]: self#safety
|
2016-12-12 02:51:22 +00:00
|
|
|
|
///
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// ## On `packed` structs
|
2016-12-12 02:51:22 +00:00
|
|
|
|
///
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// It is currently impossible to create raw pointers to unaligned fields
|
|
|
|
|
/// of a packed struct.
|
2016-12-12 02:51:22 +00:00
|
|
|
|
///
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// Attempting to create a raw pointer to an `unaligned` struct field with
|
|
|
|
|
/// an expression such as `&packed.unaligned as *const FieldType` creates an
|
|
|
|
|
/// intermediate unaligned reference before converting that to a raw pointer.
|
|
|
|
|
/// That this reference is temporary and immediately cast is inconsequential
|
|
|
|
|
/// as the compiler always expects references to be properly aligned.
|
|
|
|
|
/// As a result, using `&packed.unaligned as *const FieldType` causes immediate
|
|
|
|
|
/// *undefined behavior* in your program.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// An example of what not to do and how this relates to `write_unaligned` is:
|
|
|
|
|
///
|
2019-07-04 07:54:37 +00:00
|
|
|
|
/// ```no_run
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// #[repr(packed, C)]
|
|
|
|
|
/// struct Packed {
|
|
|
|
|
/// _padding: u8,
|
|
|
|
|
/// unaligned: u32,
|
|
|
|
|
/// }
|
|
|
|
|
///
|
|
|
|
|
/// let v = 0x01020304;
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// let mut packed: Packed = unsafe { std::mem::zeroed() };
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// let v = unsafe {
|
|
|
|
|
/// // Here we attempt to take the address of a 32-bit integer which is not aligned.
|
|
|
|
|
/// let unaligned =
|
|
|
|
|
/// // A temporary unaligned reference is created here which results in
|
|
|
|
|
/// // undefined behavior regardless of whether the reference is used or not.
|
|
|
|
|
/// &mut packed.unaligned
|
|
|
|
|
/// // Casting to a raw pointer doesn't help; the mistake already happened.
|
|
|
|
|
/// as *mut u32;
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// std::ptr::write_unaligned(unaligned, v);
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
2019-07-03 02:05:05 +00:00
|
|
|
|
/// v
|
|
|
|
|
/// };
|
2018-05-17 17:19:41 +00:00
|
|
|
|
/// ```
|
2019-07-03 02:05:05 +00:00
|
|
|
|
///
|
2019-07-04 07:54:37 +00:00
|
|
|
|
/// Accessing unaligned fields directly with e.g. `packed.unaligned` is safe however.
|
2019-07-03 02:05:05 +00:00
|
|
|
|
// FIXME: Update docs based on outcome of RFC #2582 and friends.
|
2019-07-08 14:03:29 +00:00
|
|
|
|
///
|
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// Write an usize value to a byte buffer:
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::mem;
|
|
|
|
|
///
|
|
|
|
|
/// fn write_usize(x: &mut [u8], val: usize) {
|
|
|
|
|
/// assert!(x.len() >= mem::size_of::<usize>());
|
|
|
|
|
///
|
|
|
|
|
/// let ptr = x.as_mut_ptr() as *mut usize;
|
|
|
|
|
///
|
|
|
|
|
/// unsafe { ptr.write_unaligned(val) }
|
|
|
|
|
/// }
|
|
|
|
|
/// ```
|
2016-12-12 02:51:22 +00:00
|
|
|
|
#[inline]
|
2017-03-15 03:51:29 +00:00
|
|
|
|
#[stable(feature = "ptr_unaligned", since = "1.17.0")]
|
2016-12-12 02:51:22 +00:00
|
|
|
|
pub unsafe fn write_unaligned<T>(dst: *mut T, src: T) {
|
2020-06-30 17:10:22 +00:00
|
|
|
|
// SAFETY: the caller must guarantee that `dst` is valid for writes.
|
|
|
|
|
// `dst` cannot overlap `src` because the caller has mutable access
|
|
|
|
|
// to `dst` while `src` is owned by this function.
|
|
|
|
|
unsafe {
|
|
|
|
|
// `copy_nonoverlapping` takes care of debug_assert.
|
|
|
|
|
copy_nonoverlapping(&src as *const T as *const u8, dst as *mut u8, mem::size_of::<T>());
|
|
|
|
|
}
|
2016-12-12 02:51:22 +00:00
|
|
|
|
mem::forget(src);
|
|
|
|
|
}
|
|
|
|
|
|
2016-02-18 20:01:11 +00:00
|
|
|
|
/// Performs a volatile read of the value from `src` without moving it. This
|
|
|
|
|
/// leaves the memory in `src` unchanged.
|
|
|
|
|
///
|
|
|
|
|
/// Volatile operations are intended to act on I/O memory, and are guaranteed
|
|
|
|
|
/// to not be elided or reordered by the compiler across other volatile
|
2016-04-07 17:42:53 +00:00
|
|
|
|
/// operations.
|
2016-02-18 20:01:11 +00:00
|
|
|
|
///
|
2016-04-07 17:42:53 +00:00
|
|
|
|
/// # Notes
|
|
|
|
|
///
|
|
|
|
|
/// Rust does not currently have a rigorously and formally defined memory model,
|
|
|
|
|
/// so the precise semantics of what "volatile" means here is subject to change
|
|
|
|
|
/// over time. That being said, the semantics will almost always end up pretty
|
|
|
|
|
/// similar to [C11's definition of volatile][c11].
|
|
|
|
|
///
|
2017-08-13 09:28:04 +00:00
|
|
|
|
/// The compiler shouldn't change the relative order or number of volatile
|
|
|
|
|
/// memory operations. However, volatile memory operations on zero-sized types
|
2019-02-09 22:16:58 +00:00
|
|
|
|
/// (e.g., if a zero-sized type is passed to `read_volatile`) are noops
|
2017-08-13 09:28:04 +00:00
|
|
|
|
/// and may be ignored.
|
2017-08-12 08:43:59 +00:00
|
|
|
|
///
|
2016-04-07 17:42:53 +00:00
|
|
|
|
/// [c11]: http://www.open-std.org/jtc1/sc22/wg14/www/docs/n1570.pdf
|
2016-02-18 20:01:11 +00:00
|
|
|
|
///
|
|
|
|
|
/// # Safety
|
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// Behavior is undefined if any of the following conditions are violated:
|
|
|
|
|
///
|
2018-06-17 09:45:38 +00:00
|
|
|
|
/// * `src` must be [valid] for reads.
|
2018-04-06 07:34:45 +00:00
|
|
|
|
///
|
|
|
|
|
/// * `src` must be properly aligned.
|
|
|
|
|
///
|
2020-01-31 14:22:51 +00:00
|
|
|
|
/// * `src` must point to a properly initialized value of type `T`.
|
|
|
|
|
///
|
2019-05-18 16:03:12 +00:00
|
|
|
|
/// Like [`read`], `read_volatile` creates a bitwise copy of `T`, regardless of
|
2019-02-09 21:23:30 +00:00
|
|
|
|
/// whether `T` is [`Copy`]. If `T` is not [`Copy`], using both the returned
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// value and the value at `*src` can [violate memory safety][read-ownership].
|
|
|
|
|
/// However, storing non-[`Copy`] types in volatile memory is almost certainly
|
|
|
|
|
/// incorrect.
|
|
|
|
|
///
|
2018-08-30 14:19:05 +00:00
|
|
|
|
/// Note that even if `T` has size `0`, the pointer must be non-NULL and properly aligned.
|
2018-08-29 12:34:59 +00:00
|
|
|
|
///
|
2020-09-09 20:42:57 +00:00
|
|
|
|
/// [valid]: self#safety
|
2020-09-08 21:36:36 +00:00
|
|
|
|
/// [read-ownership]: read#ownership-of-the-returned-value
|
2016-04-14 16:42:00 +00:00
|
|
|
|
///
|
2018-08-03 10:15:00 +00:00
|
|
|
|
/// Just like in C, whether an operation is volatile has no bearing whatsoever
|
|
|
|
|
/// on questions involving concurrent access from multiple threads. Volatile
|
|
|
|
|
/// accesses behave exactly like non-atomic accesses in that regard. In particular,
|
|
|
|
|
/// a race between a `read_volatile` and any write operation to the same location
|
|
|
|
|
/// is undefined behavior.
|
|
|
|
|
///
|
2016-04-14 16:42:00 +00:00
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// Basic usage:
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// let x = 12;
|
|
|
|
|
/// let y = &x as *const i32;
|
|
|
|
|
///
|
2016-08-21 20:49:09 +00:00
|
|
|
|
/// unsafe {
|
|
|
|
|
/// assert_eq!(std::ptr::read_volatile(y), 12);
|
|
|
|
|
/// }
|
2016-04-14 16:42:00 +00:00
|
|
|
|
/// ```
|
2016-02-18 20:01:11 +00:00
|
|
|
|
#[inline]
|
2016-04-07 17:42:53 +00:00
|
|
|
|
#[stable(feature = "volatile", since = "1.9.0")]
|
2016-02-18 20:01:11 +00:00
|
|
|
|
pub unsafe fn read_volatile<T>(src: *const T) -> T {
|
2020-06-06 09:56:58 +00:00
|
|
|
|
if cfg!(debug_assertions) && !is_aligned_and_not_null(src) {
|
|
|
|
|
// Not panicking to keep codegen impact smaller.
|
|
|
|
|
abort();
|
|
|
|
|
}
|
2020-06-30 17:10:22 +00:00
|
|
|
|
// SAFETY: the caller must uphold the safety contract for `volatile_load`.
|
|
|
|
|
unsafe { intrinsics::volatile_load(src) }
|
2016-02-18 20:01:11 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Performs a volatile write of a memory location with the given value without
|
|
|
|
|
/// reading or dropping the old value.
|
|
|
|
|
///
|
|
|
|
|
/// Volatile operations are intended to act on I/O memory, and are guaranteed
|
|
|
|
|
/// to not be elided or reordered by the compiler across other volatile
|
2016-04-07 17:42:53 +00:00
|
|
|
|
/// operations.
|
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// `write_volatile` does not drop the contents of `dst`. This is safe, but it
|
2018-08-30 15:07:24 +00:00
|
|
|
|
/// could leak allocations or resources, so care should be taken not to overwrite
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// an object that should be dropped.
|
|
|
|
|
///
|
|
|
|
|
/// Additionally, it does not drop `src`. Semantically, `src` is moved into the
|
|
|
|
|
/// location pointed to by `dst`.
|
|
|
|
|
///
|
2016-04-07 17:42:53 +00:00
|
|
|
|
/// # Notes
|
2016-02-18 20:01:11 +00:00
|
|
|
|
///
|
2016-04-07 17:42:53 +00:00
|
|
|
|
/// Rust does not currently have a rigorously and formally defined memory model,
|
|
|
|
|
/// so the precise semantics of what "volatile" means here is subject to change
|
|
|
|
|
/// over time. That being said, the semantics will almost always end up pretty
|
|
|
|
|
/// similar to [C11's definition of volatile][c11].
|
|
|
|
|
///
|
2017-08-13 09:28:04 +00:00
|
|
|
|
/// The compiler shouldn't change the relative order or number of volatile
|
|
|
|
|
/// memory operations. However, volatile memory operations on zero-sized types
|
2019-02-09 22:16:58 +00:00
|
|
|
|
/// (e.g., if a zero-sized type is passed to `write_volatile`) are noops
|
2017-08-13 09:28:04 +00:00
|
|
|
|
/// and may be ignored.
|
2017-08-12 08:43:59 +00:00
|
|
|
|
///
|
2016-04-07 17:42:53 +00:00
|
|
|
|
/// [c11]: http://www.open-std.org/jtc1/sc22/wg14/www/docs/n1570.pdf
|
2016-02-18 20:01:11 +00:00
|
|
|
|
///
|
|
|
|
|
/// # Safety
|
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// Behavior is undefined if any of the following conditions are violated:
|
2016-02-18 20:01:11 +00:00
|
|
|
|
///
|
2018-06-17 09:45:38 +00:00
|
|
|
|
/// * `dst` must be [valid] for writes.
|
2016-02-18 20:01:11 +00:00
|
|
|
|
///
|
2018-04-06 07:34:45 +00:00
|
|
|
|
/// * `dst` must be properly aligned.
|
|
|
|
|
///
|
2018-08-30 14:19:05 +00:00
|
|
|
|
/// Note that even if `T` has size `0`, the pointer must be non-NULL and properly aligned.
|
2018-08-29 12:34:59 +00:00
|
|
|
|
///
|
2020-09-09 20:42:57 +00:00
|
|
|
|
/// [valid]: self#safety
|
2016-04-14 16:42:00 +00:00
|
|
|
|
///
|
2018-08-03 10:15:00 +00:00
|
|
|
|
/// Just like in C, whether an operation is volatile has no bearing whatsoever
|
|
|
|
|
/// on questions involving concurrent access from multiple threads. Volatile
|
|
|
|
|
/// accesses behave exactly like non-atomic accesses in that regard. In particular,
|
2020-09-09 02:24:57 +00:00
|
|
|
|
/// a race between a `write_volatile` and any other operation (reading or writing)
|
2018-08-03 10:15:00 +00:00
|
|
|
|
/// on the same location is undefined behavior.
|
|
|
|
|
///
|
2016-04-14 16:42:00 +00:00
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// Basic usage:
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// let mut x = 0;
|
|
|
|
|
/// let y = &mut x as *mut i32;
|
|
|
|
|
/// let z = 12;
|
|
|
|
|
///
|
|
|
|
|
/// unsafe {
|
|
|
|
|
/// std::ptr::write_volatile(y, z);
|
2016-08-21 20:49:09 +00:00
|
|
|
|
/// assert_eq!(std::ptr::read_volatile(y), 12);
|
2016-04-14 16:42:00 +00:00
|
|
|
|
/// }
|
|
|
|
|
/// ```
|
2016-02-18 20:01:11 +00:00
|
|
|
|
#[inline]
|
2016-04-07 17:42:53 +00:00
|
|
|
|
#[stable(feature = "volatile", since = "1.9.0")]
|
2016-02-18 20:01:11 +00:00
|
|
|
|
pub unsafe fn write_volatile<T>(dst: *mut T, src: T) {
|
2020-06-06 09:56:58 +00:00
|
|
|
|
if cfg!(debug_assertions) && !is_aligned_and_not_null(dst) {
|
|
|
|
|
// Not panicking to keep codegen impact smaller.
|
|
|
|
|
abort();
|
|
|
|
|
}
|
2020-06-30 17:10:22 +00:00
|
|
|
|
// SAFETY: the caller must uphold the safety contract for `volatile_store`.
|
|
|
|
|
unsafe {
|
|
|
|
|
intrinsics::volatile_store(dst, src);
|
|
|
|
|
}
|
2016-02-18 20:01:11 +00:00
|
|
|
|
}
|
|
|
|
|
|
2018-04-18 15:38:12 +00:00
|
|
|
|
/// Align pointer `p`.
|
|
|
|
|
///
|
|
|
|
|
/// Calculate offset (in terms of elements of `stride` stride) that has to be applied
|
|
|
|
|
/// to pointer `p` so that pointer `p` would get aligned to `a`.
|
|
|
|
|
///
|
|
|
|
|
/// Note: This implementation has been carefully tailored to not panic. It is UB for this to panic.
|
|
|
|
|
/// The only real change that can be made here is change of `INV_TABLE_MOD_16` and associated
|
|
|
|
|
/// constants.
|
|
|
|
|
///
|
|
|
|
|
/// If we ever decide to make it possible to call the intrinsic with `a` that is not a
|
|
|
|
|
/// power-of-two, it will probably be more prudent to just change to a naive implementation rather
|
2018-08-19 13:30:23 +00:00
|
|
|
|
/// than trying to adapt this to accommodate that change.
|
2018-04-18 15:38:12 +00:00
|
|
|
|
///
|
|
|
|
|
/// Any questions go to @nagisa.
|
2019-12-07 04:18:12 +00:00
|
|
|
|
#[lang = "align_offset"]
|
2018-05-03 21:49:22 +00:00
|
|
|
|
pub(crate) unsafe fn align_offset<T: Sized>(p: *const T, a: usize) -> usize {
|
2020-08-16 13:59:43 +00:00
|
|
|
|
// FIXME(#75598): Direct use of these intrinsics improves codegen significantly at opt-level <=
|
|
|
|
|
// 1, where the method versions of these operations are not inlined.
|
|
|
|
|
use intrinsics::{unchecked_shl, unchecked_shr, unchecked_sub, wrapping_mul, wrapping_sub};
|
|
|
|
|
|
2018-04-18 15:38:12 +00:00
|
|
|
|
/// Calculate multiplicative modular inverse of `x` modulo `m`.
|
|
|
|
|
///
|
2020-08-16 14:10:54 +00:00
|
|
|
|
/// This implementation is tailored for `align_offset` and has following preconditions:
|
2018-04-18 15:38:12 +00:00
|
|
|
|
///
|
|
|
|
|
/// * `m` is a power-of-two;
|
|
|
|
|
/// * `x < m`; (if `x ≥ m`, pass in `x % m` instead)
|
|
|
|
|
///
|
|
|
|
|
/// Implementation of this function shall not panic. Ever.
|
2018-05-03 21:49:22 +00:00
|
|
|
|
#[inline]
|
2020-08-16 13:59:43 +00:00
|
|
|
|
unsafe fn mod_inv(x: usize, m: usize) -> usize {
|
2018-04-18 15:38:12 +00:00
|
|
|
|
/// Multiplicative modular inverse table modulo 2⁴ = 16.
|
|
|
|
|
///
|
2018-11-27 02:59:49 +00:00
|
|
|
|
/// Note, that this table does not contain values where inverse does not exist (i.e., for
|
2018-04-18 15:38:12 +00:00
|
|
|
|
/// `0⁻¹ mod 16`, `2⁻¹ mod 16`, etc.)
|
2018-09-24 15:01:46 +00:00
|
|
|
|
const INV_TABLE_MOD_16: [u8; 8] = [1, 11, 13, 7, 9, 3, 5, 15];
|
2018-04-18 15:38:12 +00:00
|
|
|
|
/// Modulo for which the `INV_TABLE_MOD_16` is intended.
|
|
|
|
|
const INV_TABLE_MOD: usize = 16;
|
|
|
|
|
/// INV_TABLE_MOD²
|
|
|
|
|
const INV_TABLE_MOD_SQUARED: usize = INV_TABLE_MOD * INV_TABLE_MOD;
|
|
|
|
|
|
2018-09-24 15:01:46 +00:00
|
|
|
|
let table_inverse = INV_TABLE_MOD_16[(x & (INV_TABLE_MOD - 1)) >> 1] as usize;
|
2020-08-16 13:59:43 +00:00
|
|
|
|
// SAFETY: `m` is required to be a power-of-two, hence non-zero.
|
|
|
|
|
let m_minus_one = unsafe { unchecked_sub(m, 1) };
|
2018-04-18 15:38:12 +00:00
|
|
|
|
if m <= INV_TABLE_MOD {
|
2020-08-16 13:59:43 +00:00
|
|
|
|
table_inverse & m_minus_one
|
2018-04-18 15:38:12 +00:00
|
|
|
|
} else {
|
|
|
|
|
// We iterate "up" using the following formula:
|
|
|
|
|
//
|
|
|
|
|
// $$ xy ≡ 1 (mod 2ⁿ) → xy (2 - xy) ≡ 1 (mod 2²ⁿ) $$
|
|
|
|
|
//
|
|
|
|
|
// until 2²ⁿ ≥ m. Then we can reduce to our desired `m` by taking the result `mod m`.
|
|
|
|
|
let mut inverse = table_inverse;
|
|
|
|
|
let mut going_mod = INV_TABLE_MOD_SQUARED;
|
|
|
|
|
loop {
|
|
|
|
|
// y = y * (2 - xy) mod n
|
|
|
|
|
//
|
|
|
|
|
// Note, that we use wrapping operations here intentionally – the original formula
|
2018-11-27 02:59:49 +00:00
|
|
|
|
// uses e.g., subtraction `mod n`. It is entirely fine to do them `mod
|
2020-06-02 07:59:11 +00:00
|
|
|
|
// usize::MAX` instead, because we take the result `mod n` at the end
|
2018-04-18 15:38:12 +00:00
|
|
|
|
// anyway.
|
2020-08-16 13:59:43 +00:00
|
|
|
|
inverse = wrapping_mul(inverse, wrapping_sub(2usize, wrapping_mul(x, inverse)));
|
2020-01-30 19:04:24 +00:00
|
|
|
|
if going_mod >= m {
|
2020-08-16 13:59:43 +00:00
|
|
|
|
return inverse & m_minus_one;
|
2018-04-18 15:38:12 +00:00
|
|
|
|
}
|
2020-08-16 13:59:43 +00:00
|
|
|
|
going_mod = wrapping_mul(going_mod, going_mod);
|
2018-04-18 15:38:12 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2019-04-15 02:23:21 +00:00
|
|
|
|
let stride = mem::size_of::<T>();
|
2020-08-16 14:10:54 +00:00
|
|
|
|
// SAFETY: `a` is a power-of-two, therefore non-zero.
|
2020-08-16 13:59:43 +00:00
|
|
|
|
let a_minus_one = unsafe { unchecked_sub(a, 1) };
|
2020-08-16 14:10:54 +00:00
|
|
|
|
if stride == 1 {
|
|
|
|
|
// `stride == 1` case can be computed more efficiently through `-p (mod a)`.
|
|
|
|
|
return wrapping_sub(0, p as usize) & a_minus_one;
|
|
|
|
|
}
|
2018-04-18 15:38:12 +00:00
|
|
|
|
|
2020-08-16 14:10:54 +00:00
|
|
|
|
let pmoda = p as usize & a_minus_one;
|
2018-04-18 15:38:12 +00:00
|
|
|
|
if pmoda == 0 {
|
|
|
|
|
// Already aligned. Yay!
|
|
|
|
|
return 0;
|
2020-08-16 14:10:54 +00:00
|
|
|
|
} else if stride == 0 {
|
|
|
|
|
// If the pointer is not aligned, and the element is zero-sized, then no amount of
|
|
|
|
|
// elements will ever align the pointer.
|
|
|
|
|
return usize::MAX;
|
2018-05-03 21:49:22 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
let smoda = stride & a_minus_one;
|
2020-08-16 13:59:43 +00:00
|
|
|
|
// SAFETY: a is power-of-two hence non-zero. stride == 0 case is handled above.
|
2020-06-30 17:10:22 +00:00
|
|
|
|
let gcdpow = unsafe { intrinsics::cttz_nonzero(stride).min(intrinsics::cttz_nonzero(a)) };
|
2020-08-16 13:59:43 +00:00
|
|
|
|
// SAFETY: gcdpow has an upper-bound that’s at most the number of bits in an usize.
|
|
|
|
|
let gcd = unsafe { unchecked_shl(1usize, gcdpow) };
|
2018-05-03 21:49:22 +00:00
|
|
|
|
|
2020-08-16 13:59:43 +00:00
|
|
|
|
// SAFETY: gcd is always greater or equal to 1.
|
|
|
|
|
if p as usize & unsafe { unchecked_sub(gcd, 1) } == 0 {
|
2018-09-24 15:01:46 +00:00
|
|
|
|
// This branch solves for the following linear congruence equation:
|
2018-04-18 15:38:12 +00:00
|
|
|
|
//
|
2020-01-28 21:14:04 +00:00
|
|
|
|
// ` p + so = 0 mod a `
|
2018-04-18 15:38:12 +00:00
|
|
|
|
//
|
2020-01-28 21:14:04 +00:00
|
|
|
|
// `p` here is the pointer value, `s` - stride of `T`, `o` offset in `T`s, and `a` - the
|
2018-09-24 15:01:46 +00:00
|
|
|
|
// requested alignment.
|
2018-04-18 15:38:12 +00:00
|
|
|
|
//
|
2020-08-16 13:59:43 +00:00
|
|
|
|
// With `g = gcd(a, s)`, and the above condition asserting that `p` is also divisible by
|
|
|
|
|
// `g`, we can denote `a' = a/g`, `s' = s/g`, `p' = p/g`, then this becomes equivalent to:
|
2018-04-18 15:38:12 +00:00
|
|
|
|
//
|
2020-01-28 21:14:04 +00:00
|
|
|
|
// ` p' + s'o = 0 mod a' `
|
|
|
|
|
// ` o = (a' - (p' mod a')) * (s'^-1 mod a') `
|
2018-04-18 15:38:12 +00:00
|
|
|
|
//
|
2020-01-28 21:14:04 +00:00
|
|
|
|
// The first term is "the relative alignment of `p` to `a`" (divided by the `g`), the second
|
|
|
|
|
// term is "how does incrementing `p` by `s` bytes change the relative alignment of `p`" (again
|
|
|
|
|
// divided by `g`).
|
|
|
|
|
// Division by `g` is necessary to make the inverse well formed if `a` and `s` are not
|
|
|
|
|
// co-prime.
|
|
|
|
|
//
|
|
|
|
|
// Furthermore, the result produced by this solution is not "minimal", so it is necessary
|
|
|
|
|
// to take the result `o mod lcm(s, a)`. We can replace `lcm(s, a)` with just a `a'`.
|
2020-08-16 13:59:43 +00:00
|
|
|
|
|
|
|
|
|
// SAFETY: `gcdpow` has an upper-bound not greater than the number of trailing 0-bits in
|
|
|
|
|
// `a`.
|
|
|
|
|
let a2 = unsafe { unchecked_shr(a, gcdpow) };
|
|
|
|
|
// SAFETY: `a2` is non-zero. Shifting `a` by `gcdpow` cannot shift out any of the set bits
|
|
|
|
|
// in `a` (of which it has exactly one).
|
|
|
|
|
let a2minus1 = unsafe { unchecked_sub(a2, 1) };
|
|
|
|
|
// SAFETY: `gcdpow` has an upper-bound not greater than the number of trailing 0-bits in
|
|
|
|
|
// `a`.
|
|
|
|
|
let s2 = unsafe { unchecked_shr(smoda, gcdpow) };
|
|
|
|
|
// SAFETY: `gcdpow` has an upper-bound not greater than the number of trailing 0-bits in
|
|
|
|
|
// `a`. Furthermore, the subtraction cannot overflow, because `a2 = a >> gcdpow` will
|
|
|
|
|
// always be strictly greater than `(p % a) >> gcdpow`.
|
|
|
|
|
let minusp2 = unsafe { unchecked_sub(a2, unchecked_shr(pmoda, gcdpow)) };
|
|
|
|
|
// SAFETY: `a2` is a power-of-two, as proven above. `s2` is strictly less than `a2`
|
|
|
|
|
// because `(s % a) >> gcdpow` is strictly less than `a >> gcdpow`.
|
|
|
|
|
return wrapping_mul(minusp2, unsafe { mod_inv(s2, a2) }) & a2minus1;
|
2018-04-18 15:38:12 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Cannot be aligned at all.
|
2020-06-02 07:59:11 +00:00
|
|
|
|
usize::MAX
|
2018-04-18 15:38:12 +00:00
|
|
|
|
}
|
|
|
|
|
|
2019-02-09 22:16:58 +00:00
|
|
|
|
/// Compares raw pointers for equality.
|
2016-09-15 09:19:19 +00:00
|
|
|
|
///
|
|
|
|
|
/// This is the same as using the `==` operator, but less generic:
|
|
|
|
|
/// the arguments have to be `*const T` raw pointers,
|
|
|
|
|
/// not anything that implements `PartialEq`.
|
|
|
|
|
///
|
|
|
|
|
/// This can be used to compare `&T` references (which coerce to `*const T` implicitly)
|
|
|
|
|
/// by their address rather than comparing the values they point to
|
|
|
|
|
/// (which is what the `PartialEq for &T` implementation does).
|
|
|
|
|
///
|
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
|
|
|
|
/// let five = 5;
|
|
|
|
|
/// let other_five = 5;
|
|
|
|
|
/// let five_ref = &five;
|
|
|
|
|
/// let same_five_ref = &five;
|
|
|
|
|
/// let other_five_ref = &other_five;
|
|
|
|
|
///
|
|
|
|
|
/// assert!(five_ref == same_five_ref);
|
|
|
|
|
/// assert!(ptr::eq(five_ref, same_five_ref));
|
2019-03-25 20:38:12 +00:00
|
|
|
|
///
|
|
|
|
|
/// assert!(five_ref == other_five_ref);
|
2016-09-15 09:19:19 +00:00
|
|
|
|
/// assert!(!ptr::eq(five_ref, other_five_ref));
|
|
|
|
|
/// ```
|
2019-03-25 21:19:47 +00:00
|
|
|
|
///
|
|
|
|
|
/// Slices are also compared by their length (fat pointers):
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// let a = [1, 2, 3];
|
|
|
|
|
/// assert!(std::ptr::eq(&a[..3], &a[..3]));
|
|
|
|
|
/// assert!(!std::ptr::eq(&a[..2], &a[..3]));
|
|
|
|
|
/// assert!(!std::ptr::eq(&a[0..2], &a[1..3]));
|
|
|
|
|
/// ```
|
|
|
|
|
///
|
|
|
|
|
/// Traits are also compared by their implementation:
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// #[repr(transparent)]
|
|
|
|
|
/// struct Wrapper { member: i32 }
|
|
|
|
|
///
|
|
|
|
|
/// trait Trait {}
|
|
|
|
|
/// impl Trait for Wrapper {}
|
|
|
|
|
/// impl Trait for i32 {}
|
|
|
|
|
///
|
2019-10-01 11:55:46 +00:00
|
|
|
|
/// let wrapper = Wrapper { member: 10 };
|
|
|
|
|
///
|
|
|
|
|
/// // Pointers have equal addresses.
|
|
|
|
|
/// assert!(std::ptr::eq(
|
|
|
|
|
/// &wrapper as *const Wrapper as *const u8,
|
|
|
|
|
/// &wrapper.member as *const i32 as *const u8
|
|
|
|
|
/// ));
|
|
|
|
|
///
|
|
|
|
|
/// // Objects have equal addresses, but `Trait` has different implementations.
|
|
|
|
|
/// assert!(!std::ptr::eq(
|
|
|
|
|
/// &wrapper as &dyn Trait,
|
|
|
|
|
/// &wrapper.member as &dyn Trait,
|
|
|
|
|
/// ));
|
|
|
|
|
/// assert!(!std::ptr::eq(
|
|
|
|
|
/// &wrapper as &dyn Trait as *const dyn Trait,
|
|
|
|
|
/// &wrapper.member as &dyn Trait as *const dyn Trait,
|
|
|
|
|
/// ));
|
|
|
|
|
///
|
|
|
|
|
/// // Converting the reference to a `*const u8` compares by address.
|
|
|
|
|
/// assert!(std::ptr::eq(
|
|
|
|
|
/// &wrapper as &dyn Trait as *const dyn Trait as *const u8,
|
|
|
|
|
/// &wrapper.member as &dyn Trait as *const dyn Trait as *const u8,
|
|
|
|
|
/// ));
|
2019-03-25 21:19:47 +00:00
|
|
|
|
/// ```
|
2017-03-15 14:58:27 +00:00
|
|
|
|
#[stable(feature = "ptr_eq", since = "1.17.0")]
|
2016-09-15 09:19:19 +00:00
|
|
|
|
#[inline]
|
|
|
|
|
pub fn eq<T: ?Sized>(a: *const T, b: *const T) -> bool {
|
|
|
|
|
a == b
|
|
|
|
|
}
|
|
|
|
|
|
2018-12-07 16:33:32 +00:00
|
|
|
|
/// Hash a raw pointer.
|
|
|
|
|
///
|
2018-12-07 16:34:53 +00:00
|
|
|
|
/// This can be used to hash a `&T` reference (which coerces to `*const T` implicitly)
|
2018-12-07 16:33:32 +00:00
|
|
|
|
/// by its address rather than the value it points to
|
|
|
|
|
/// (which is what the `Hash for &T` implementation does).
|
2018-11-26 18:36:03 +00:00
|
|
|
|
///
|
|
|
|
|
/// # Examples
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// use std::collections::hash_map::DefaultHasher;
|
2018-11-27 20:09:18 +00:00
|
|
|
|
/// use std::hash::{Hash, Hasher};
|
2018-11-26 18:36:03 +00:00
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
|
|
|
|
/// let five = 5;
|
|
|
|
|
/// let five_ref = &five;
|
|
|
|
|
///
|
|
|
|
|
/// let mut hasher = DefaultHasher::new();
|
2018-11-26 21:31:12 +00:00
|
|
|
|
/// ptr::hash(five_ref, &mut hasher);
|
2018-11-27 16:46:24 +00:00
|
|
|
|
/// let actual = hasher.finish();
|
|
|
|
|
///
|
|
|
|
|
/// let mut hasher = DefaultHasher::new();
|
2018-11-27 20:09:18 +00:00
|
|
|
|
/// (five_ref as *const i32).hash(&mut hasher);
|
2018-11-27 16:46:24 +00:00
|
|
|
|
/// let expected = hasher.finish();
|
|
|
|
|
///
|
|
|
|
|
/// assert_eq!(actual, expected);
|
2018-11-26 18:36:03 +00:00
|
|
|
|
/// ```
|
2019-04-01 13:36:07 +00:00
|
|
|
|
#[stable(feature = "ptr_hash", since = "1.35.0")]
|
2018-12-12 17:41:03 +00:00
|
|
|
|
pub fn hash<T: ?Sized, S: hash::Hasher>(hashee: *const T, into: &mut S) {
|
2019-04-15 02:23:21 +00:00
|
|
|
|
use crate::hash::Hash;
|
2018-12-04 08:06:26 +00:00
|
|
|
|
hashee.hash(into);
|
2018-11-26 18:36:03 +00:00
|
|
|
|
}
|
|
|
|
|
|
2015-09-13 15:11:10 +00:00
|
|
|
|
// Impls for function pointers
|
|
|
|
|
macro_rules! fnptr_impls_safety_abi {
|
|
|
|
|
($FnTy: ty, $($Arg: ident),*) => {
|
|
|
|
|
#[stable(feature = "fnptr_impls", since = "1.4.0")]
|
|
|
|
|
impl<Ret, $($Arg),*> PartialEq for $FnTy {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn eq(&self, other: &Self) -> bool {
|
|
|
|
|
*self as usize == *other as usize
|
|
|
|
|
}
|
2013-08-21 13:31:02 +00:00
|
|
|
|
}
|
2015-09-13 15:11:10 +00:00
|
|
|
|
|
|
|
|
|
#[stable(feature = "fnptr_impls", since = "1.4.0")]
|
|
|
|
|
impl<Ret, $($Arg),*> Eq for $FnTy {}
|
|
|
|
|
|
|
|
|
|
#[stable(feature = "fnptr_impls", since = "1.4.0")]
|
|
|
|
|
impl<Ret, $($Arg),*> PartialOrd for $FnTy {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
|
|
|
|
|
(*self as usize).partial_cmp(&(*other as usize))
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[stable(feature = "fnptr_impls", since = "1.4.0")]
|
|
|
|
|
impl<Ret, $($Arg),*> Ord for $FnTy {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn cmp(&self, other: &Self) -> Ordering {
|
|
|
|
|
(*self as usize).cmp(&(*other as usize))
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[stable(feature = "fnptr_impls", since = "1.4.0")]
|
|
|
|
|
impl<Ret, $($Arg),*> hash::Hash for $FnTy {
|
|
|
|
|
fn hash<HH: hash::Hasher>(&self, state: &mut HH) {
|
|
|
|
|
state.write_usize(*self as usize)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[stable(feature = "fnptr_impls", since = "1.4.0")]
|
|
|
|
|
impl<Ret, $($Arg),*> fmt::Pointer for $FnTy {
|
2019-04-18 23:37:12 +00:00
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
2019-06-08 13:57:40 +00:00
|
|
|
|
// HACK: The intermediate cast as usize is required for AVR
|
|
|
|
|
// so that the address space of the source function pointer
|
|
|
|
|
// is preserved in the final function pointer.
|
|
|
|
|
//
|
|
|
|
|
// https://github.com/avr-rust/rust/issues/143
|
|
|
|
|
fmt::Pointer::fmt(&(*self as usize as *const ()), f)
|
2015-09-13 15:11:10 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[stable(feature = "fnptr_impls", since = "1.4.0")]
|
|
|
|
|
impl<Ret, $($Arg),*> fmt::Debug for $FnTy {
|
2019-04-18 23:37:12 +00:00
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
2019-06-08 13:57:40 +00:00
|
|
|
|
// HACK: The intermediate cast as usize is required for AVR
|
|
|
|
|
// so that the address space of the source function pointer
|
|
|
|
|
// is preserved in the final function pointer.
|
|
|
|
|
//
|
|
|
|
|
// https://github.com/avr-rust/rust/issues/143
|
|
|
|
|
fmt::Pointer::fmt(&(*self as usize as *const ()), f)
|
2013-08-21 13:31:02 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
2014-11-14 17:18:10 +00:00
|
|
|
|
}
|
2013-08-21 13:31:02 +00:00
|
|
|
|
}
|
|
|
|
|
|
2015-09-13 15:11:10 +00:00
|
|
|
|
macro_rules! fnptr_impls_args {
|
2016-07-16 21:15:15 +00:00
|
|
|
|
($($Arg: ident),+) => {
|
2019-05-29 18:05:43 +00:00
|
|
|
|
fnptr_impls_safety_abi! { extern "Rust" fn($($Arg),+) -> Ret, $($Arg),+ }
|
|
|
|
|
fnptr_impls_safety_abi! { extern "C" fn($($Arg),+) -> Ret, $($Arg),+ }
|
|
|
|
|
fnptr_impls_safety_abi! { extern "C" fn($($Arg),+ , ...) -> Ret, $($Arg),+ }
|
|
|
|
|
fnptr_impls_safety_abi! { unsafe extern "Rust" fn($($Arg),+) -> Ret, $($Arg),+ }
|
|
|
|
|
fnptr_impls_safety_abi! { unsafe extern "C" fn($($Arg),+) -> Ret, $($Arg),+ }
|
|
|
|
|
fnptr_impls_safety_abi! { unsafe extern "C" fn($($Arg),+ , ...) -> Ret, $($Arg),+ }
|
2016-07-16 21:15:15 +00:00
|
|
|
|
};
|
|
|
|
|
() => {
|
|
|
|
|
// No variadic functions with 0 parameters
|
|
|
|
|
fnptr_impls_safety_abi! { extern "Rust" fn() -> Ret, }
|
|
|
|
|
fnptr_impls_safety_abi! { extern "C" fn() -> Ret, }
|
|
|
|
|
fnptr_impls_safety_abi! { unsafe extern "Rust" fn() -> Ret, }
|
|
|
|
|
fnptr_impls_safety_abi! { unsafe extern "C" fn() -> Ret, }
|
|
|
|
|
};
|
2015-09-13 15:11:10 +00:00
|
|
|
|
}
|
|
|
|
|
|
2019-12-07 04:18:12 +00:00
|
|
|
|
fnptr_impls_args! {}
|
2015-09-13 15:11:10 +00:00
|
|
|
|
fnptr_impls_args! { A }
|
|
|
|
|
fnptr_impls_args! { A, B }
|
|
|
|
|
fnptr_impls_args! { A, B, C }
|
|
|
|
|
fnptr_impls_args! { A, B, C, D }
|
|
|
|
|
fnptr_impls_args! { A, B, C, D, E }
|
2015-09-21 08:51:30 +00:00
|
|
|
|
fnptr_impls_args! { A, B, C, D, E, F }
|
|
|
|
|
fnptr_impls_args! { A, B, C, D, E, F, G }
|
|
|
|
|
fnptr_impls_args! { A, B, C, D, E, F, G, H }
|
|
|
|
|
fnptr_impls_args! { A, B, C, D, E, F, G, H, I }
|
|
|
|
|
fnptr_impls_args! { A, B, C, D, E, F, G, H, I, J }
|
|
|
|
|
fnptr_impls_args! { A, B, C, D, E, F, G, H, I, J, K }
|
|
|
|
|
fnptr_impls_args! { A, B, C, D, E, F, G, H, I, J, K, L }
|
2020-06-12 16:17:30 +00:00
|
|
|
|
|
|
|
|
|
/// Create a `const` raw pointer to a place, without creating an intermediate reference.
|
|
|
|
|
///
|
|
|
|
|
/// Creating a reference with `&`/`&mut` is only allowed if the pointer is properly aligned
|
|
|
|
|
/// and points to initialized data. For cases where those requirements do not hold,
|
|
|
|
|
/// raw pointers should be used instead. However, `&expr as *const _` creates a reference
|
|
|
|
|
/// before casting it to a raw pointer, and that reference is subject to the same rules
|
|
|
|
|
/// as all other references. This macro can create a raw pointer *without* creating
|
|
|
|
|
/// a reference first.
|
|
|
|
|
///
|
|
|
|
|
/// # Example
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// #![feature(raw_ref_macros)]
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
|
|
|
|
/// #[repr(packed)]
|
|
|
|
|
/// struct Packed {
|
|
|
|
|
/// f1: u8,
|
|
|
|
|
/// f2: u16,
|
|
|
|
|
/// }
|
|
|
|
|
///
|
|
|
|
|
/// let packed = Packed { f1: 1, f2: 2 };
|
|
|
|
|
/// // `&packed.f2` would create an unaligned reference, and thus be Undefined Behavior!
|
|
|
|
|
/// let raw_f2 = ptr::raw_const!(packed.f2);
|
|
|
|
|
/// assert_eq!(unsafe { raw_f2.read_unaligned() }, 2);
|
|
|
|
|
/// ```
|
2020-06-16 07:25:29 +00:00
|
|
|
|
#[unstable(feature = "raw_ref_macros", issue = "73394")]
|
2020-06-12 16:17:30 +00:00
|
|
|
|
#[rustc_macro_transparency = "semitransparent"]
|
|
|
|
|
#[allow_internal_unstable(raw_ref_op)]
|
|
|
|
|
pub macro raw_const($e:expr) {
|
|
|
|
|
&raw const $e
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Create a `mut` raw pointer to a place, without creating an intermediate reference.
|
|
|
|
|
///
|
|
|
|
|
/// Creating a reference with `&`/`&mut` is only allowed if the pointer is properly aligned
|
|
|
|
|
/// and points to initialized data. For cases where those requirements do not hold,
|
|
|
|
|
/// raw pointers should be used instead. However, `&mut expr as *mut _` creates a reference
|
|
|
|
|
/// before casting it to a raw pointer, and that reference is subject to the same rules
|
|
|
|
|
/// as all other references. This macro can create a raw pointer *without* creating
|
|
|
|
|
/// a reference first.
|
|
|
|
|
///
|
|
|
|
|
/// # Example
|
|
|
|
|
///
|
|
|
|
|
/// ```
|
|
|
|
|
/// #![feature(raw_ref_macros)]
|
|
|
|
|
/// use std::ptr;
|
|
|
|
|
///
|
|
|
|
|
/// #[repr(packed)]
|
|
|
|
|
/// struct Packed {
|
|
|
|
|
/// f1: u8,
|
|
|
|
|
/// f2: u16,
|
|
|
|
|
/// }
|
|
|
|
|
///
|
|
|
|
|
/// let mut packed = Packed { f1: 1, f2: 2 };
|
|
|
|
|
/// // `&mut packed.f2` would create an unaligned reference, and thus be Undefined Behavior!
|
|
|
|
|
/// let raw_f2 = ptr::raw_mut!(packed.f2);
|
|
|
|
|
/// unsafe { raw_f2.write_unaligned(42); }
|
|
|
|
|
/// assert_eq!({packed.f2}, 42); // `{...}` forces copying the field instead of creating a reference.
|
|
|
|
|
/// ```
|
2020-06-16 07:25:29 +00:00
|
|
|
|
#[unstable(feature = "raw_ref_macros", issue = "73394")]
|
2020-06-12 16:17:30 +00:00
|
|
|
|
#[rustc_macro_transparency = "semitransparent"]
|
|
|
|
|
#[allow_internal_unstable(raw_ref_op)]
|
|
|
|
|
pub macro raw_mut($e:expr) {
|
|
|
|
|
&raw mut $e
|
|
|
|
|
}
|