cleanup code w/ pointers in std a little

This commit is contained in:
Maybe Waffle 2022-08-01 21:50:21 +04:00
parent 1f5d8d49eb
commit 127b6c4c18
8 changed files with 19 additions and 15 deletions

View File

@ -74,7 +74,7 @@ use crate::ptr;
/// { /// {
/// return null_mut(); /// return null_mut();
/// }; /// };
/// (self.arena.get() as *mut u8).add(allocated) /// self.arena.get().cast::<u8>().add(allocated)
/// } /// }
/// unsafe fn dealloc(&self, _ptr: *mut u8, _layout: Layout) {} /// unsafe fn dealloc(&self, _ptr: *mut u8, _layout: Layout) {}
/// } /// }

View File

@ -1270,8 +1270,8 @@ impl<T: ?Sized> *const T {
/// # fn foo(n: usize) { /// # fn foo(n: usize) {
/// # use std::mem::align_of; /// # use std::mem::align_of;
/// # unsafe { /// # unsafe {
/// let x = [5u8, 6u8, 7u8, 8u8, 9u8]; /// let x = [5u8, 6, 7, 8, 9];
/// let ptr = x.as_ptr().add(n) as *const u8; /// let ptr = x.as_ptr().add(n);
/// let offset = ptr.align_offset(align_of::<u16>()); /// let offset = ptr.align_offset(align_of::<u16>());
/// if offset < x.len() - n - 1 { /// if offset < x.len() - n - 1 {
/// let u16_ptr = ptr.add(offset) as *const u16; /// let u16_ptr = ptr.add(offset) as *const u16;

View File

@ -1548,12 +1548,13 @@ impl<T: ?Sized> *mut T {
/// # fn foo(n: usize) { /// # fn foo(n: usize) {
/// # use std::mem::align_of; /// # use std::mem::align_of;
/// # unsafe { /// # unsafe {
/// let x = [5u8, 6u8, 7u8, 8u8, 9u8]; /// let mut x = [5u8, 6, 7, 8, 9];
/// let ptr = x.as_ptr().add(n) as *const u8; /// let ptr = x.as_mut_ptr().add(n);
/// let offset = ptr.align_offset(align_of::<u16>()); /// let offset = ptr.align_offset(align_of::<u16>());
/// if offset < x.len() - n - 1 { /// if offset < x.len() - n - 1 {
/// let u16_ptr = ptr.add(offset) as *const u16; /// let u16_ptr = ptr.add(offset) as *mut u16;
/// assert_ne!(*u16_ptr, 500); /// assert_ne!(*u16_ptr, 500);
/// *u16_ptr = 0;
/// } else { /// } else {
/// // while the pointer can be aligned via `offset`, it would point /// // while the pointer can be aligned via `offset`, it would point
/// // outside the allocation /// // outside the allocation

View File

@ -92,7 +92,7 @@ impl<'a, T> Iter<'a, T> {
assume(!ptr.is_null()); assume(!ptr.is_null());
let end = if mem::size_of::<T>() == 0 { let end = if mem::size_of::<T>() == 0 {
(ptr as *const u8).wrapping_add(slice.len()) as *const T ptr.wrapping_byte_add(slice.len())
} else { } else {
ptr.add(slice.len()) ptr.add(slice.len())
}; };
@ -228,7 +228,7 @@ impl<'a, T> IterMut<'a, T> {
assume(!ptr.is_null()); assume(!ptr.is_null());
let end = if mem::size_of::<T>() == 0 { let end = if mem::size_of::<T>() == 0 {
(ptr as *mut u8).wrapping_add(slice.len()) as *mut T ptr.wrapping_byte_add(slice.len())
} else { } else {
ptr.add(slice.len()) ptr.add(slice.len())
}; };

View File

@ -3,7 +3,7 @@ const DATA: [u16; 2] = [u16::from_ne_bytes([0x01, 0x23]), u16::from_ne_bytes([0x
const fn unaligned_ptr() -> *const u16 { const fn unaligned_ptr() -> *const u16 {
// Since DATA.as_ptr() is aligned to two bytes, adding 1 byte to that produces an unaligned *const u16 // Since DATA.as_ptr() is aligned to two bytes, adding 1 byte to that produces an unaligned *const u16
unsafe { (DATA.as_ptr() as *const u8).add(1) as *const u16 } unsafe { DATA.as_ptr().byte_add(1) }
} }
#[test] #[test]
@ -67,7 +67,7 @@ fn write() {
const fn write_unaligned() -> [u16; 2] { const fn write_unaligned() -> [u16; 2] {
let mut two_aligned = [0u16; 2]; let mut two_aligned = [0u16; 2];
unsafe { unsafe {
let unaligned_ptr = (two_aligned.as_mut_ptr() as *mut u8).add(1) as *mut u16; let unaligned_ptr = two_aligned.as_mut_ptr().byte_add(1);
ptr::write_unaligned(unaligned_ptr, u16::from_ne_bytes([0x23, 0x45])); ptr::write_unaligned(unaligned_ptr, u16::from_ne_bytes([0x23, 0x45]));
} }
two_aligned two_aligned
@ -91,7 +91,7 @@ fn mut_ptr_write() {
const fn write_unaligned() -> [u16; 2] { const fn write_unaligned() -> [u16; 2] {
let mut two_aligned = [0u16; 2]; let mut two_aligned = [0u16; 2];
unsafe { unsafe {
let unaligned_ptr = (two_aligned.as_mut_ptr() as *mut u8).add(1) as *mut u16; let unaligned_ptr = two_aligned.as_mut_ptr().byte_add(1);
unaligned_ptr.write_unaligned(u16::from_ne_bytes([0x23, 0x45])); unaligned_ptr.write_unaligned(u16::from_ne_bytes([0x23, 0x45]));
} }
two_aligned two_aligned

View File

@ -14,6 +14,7 @@
#![feature(const_maybe_uninit_assume_init_read)] #![feature(const_maybe_uninit_assume_init_read)]
#![feature(const_nonnull_new)] #![feature(const_nonnull_new)]
#![feature(const_num_from_num)] #![feature(const_num_from_num)]
#![feature(const_pointer_byte_offsets)]
#![feature(const_ptr_as_ref)] #![feature(const_ptr_as_ref)]
#![feature(const_ptr_read)] #![feature(const_ptr_read)]
#![feature(const_ptr_write)] #![feature(const_ptr_write)]
@ -74,6 +75,7 @@
#![feature(never_type)] #![feature(never_type)]
#![feature(unwrap_infallible)] #![feature(unwrap_infallible)]
#![feature(result_into_ok_or_err)] #![feature(result_into_ok_or_err)]
#![feature(pointer_byte_offsets)]
#![feature(portable_simd)] #![feature(portable_simd)]
#![feature(ptr_metadata)] #![feature(ptr_metadata)]
#![feature(once_cell)] #![feature(once_cell)]

View File

@ -115,7 +115,7 @@ pub unsafe trait UserSafe {
/// * the pointer is null. /// * the pointer is null.
/// * the pointed-to range is not in user memory. /// * the pointed-to range is not in user memory.
unsafe fn check_ptr(ptr: *const Self) { unsafe fn check_ptr(ptr: *const Self) {
let is_aligned = |p| -> bool { 0 == (p as usize) & (Self::align_of() - 1) }; let is_aligned = |p: *const u8| -> bool { 0 == p.addr() & (Self::align_of() - 1) };
assert!(is_aligned(ptr as *const u8)); assert!(is_aligned(ptr as *const u8));
assert!(is_user_range(ptr as _, mem::size_of_val(unsafe { &*ptr }))); assert!(is_user_range(ptr as _, mem::size_of_val(unsafe { &*ptr })));

View File

@ -1,15 +1,16 @@
use crate::alloc::{GlobalAlloc, Layout, System}; use crate::alloc::{GlobalAlloc, Layout, System};
use crate::ptr::null_mut;
#[stable(feature = "alloc_system_type", since = "1.28.0")] #[stable(feature = "alloc_system_type", since = "1.28.0")]
unsafe impl GlobalAlloc for System { unsafe impl GlobalAlloc for System {
#[inline] #[inline]
unsafe fn alloc(&self, _layout: Layout) -> *mut u8 { unsafe fn alloc(&self, _layout: Layout) -> *mut u8 {
0 as *mut u8 null_mut()
} }
#[inline] #[inline]
unsafe fn alloc_zeroed(&self, _layout: Layout) -> *mut u8 { unsafe fn alloc_zeroed(&self, _layout: Layout) -> *mut u8 {
0 as *mut u8 null_mut()
} }
#[inline] #[inline]
@ -17,6 +18,6 @@ unsafe impl GlobalAlloc for System {
#[inline] #[inline]
unsafe fn realloc(&self, _ptr: *mut u8, _layout: Layout, _new_size: usize) -> *mut u8 { unsafe fn realloc(&self, _ptr: *mut u8, _layout: Layout, _new_size: usize) -> *mut u8 {
0 as *mut u8 null_mut()
} }
} }