Check for null in the alloc_zeroed example

We should demonstrate good behavior, just like #99198 did for `alloc`.
This commit is contained in:
Josh Stone 2024-11-11 16:30:37 -08:00
parent 81eef2d362
commit 2ddb91acd1

View File

@ -155,11 +155,14 @@ pub unsafe fn realloc(ptr: *mut u8, layout: Layout, new_size: usize) -> *mut u8
/// # Examples /// # Examples
/// ///
/// ``` /// ```
/// use std::alloc::{alloc_zeroed, dealloc, Layout}; /// use std::alloc::{alloc_zeroed, dealloc, handle_alloc_error, Layout};
/// ///
/// unsafe { /// unsafe {
/// let layout = Layout::new::<u16>(); /// let layout = Layout::new::<u16>();
/// let ptr = alloc_zeroed(layout); /// let ptr = alloc_zeroed(layout);
/// if ptr.is_null() {
/// handle_alloc_error(layout);
/// }
/// ///
/// assert_eq!(*(ptr as *mut u16), 0); /// assert_eq!(*(ptr as *mut u16), 0);
/// ///