|
| 1 | +//@compile-flags: -Zmiri-permissive-provenance |
| 2 | +#![deny(unsafe_op_in_unsafe_fn)] |
| 3 | +//! This does some tricky ptr-int-casting. |
| 4 | +
|
| 5 | +use core::alloc::{GlobalAlloc, Layout}; |
| 6 | +use std::alloc::System; |
| 7 | + |
| 8 | +/// # Safety |
| 9 | +/// `ptr` must be valid for writes of `len` bytes |
| 10 | +unsafe fn volatile_write_zeroize_mem(ptr: *mut u8, len: usize) { |
| 11 | + for i in 0..len { |
| 12 | + // ptr as usize + i can't overlow because `ptr` is valid for writes of `len` |
| 13 | + let ptr_new: *mut u8 = ((ptr as usize) + i) as *mut u8; |
| 14 | + // SAFETY: `ptr` is valid for writes of `len` bytes, so `ptr_new` is valid for a |
| 15 | + // byte write |
| 16 | + unsafe { |
| 17 | + core::ptr::write_volatile(ptr_new, 0u8); |
| 18 | + } |
| 19 | + } |
| 20 | +} |
| 21 | + |
| 22 | +pub struct ZeroizeAlloc; |
| 23 | + |
| 24 | +unsafe impl GlobalAlloc for ZeroizeAlloc { |
| 25 | + unsafe fn alloc(&self, layout: Layout) -> *mut u8 { |
| 26 | + // SAFETY: uphold by caller |
| 27 | + unsafe { System.alloc(layout) } |
| 28 | + } |
| 29 | + |
| 30 | + unsafe fn dealloc(&self, ptr: *mut u8, layout: Layout) { |
| 31 | + // securely wipe the deallocated memory |
| 32 | + // SAFETY: `ptr` is valid for writes of `layout.size()` bytes since it was |
| 33 | + // previously successfully allocated (by the safety assumption on this function) |
| 34 | + // and not yet deallocated |
| 35 | + unsafe { |
| 36 | + volatile_write_zeroize_mem(ptr, layout.size()); |
| 37 | + } |
| 38 | + // SAFETY: uphold by caller |
| 39 | + unsafe { System.dealloc(ptr, layout) } |
| 40 | + } |
| 41 | + |
| 42 | + unsafe fn alloc_zeroed(&self, layout: Layout) -> *mut u8 { |
| 43 | + // SAFETY: uphold by caller |
| 44 | + unsafe { System.alloc_zeroed(layout) } |
| 45 | + } |
| 46 | +} |
| 47 | + |
| 48 | +#[global_allocator] |
| 49 | +static GLOBAL: ZeroizeAlloc = ZeroizeAlloc; |
| 50 | + |
| 51 | +fn main() { |
| 52 | + let layout = Layout::new::<[u8; 16]>(); |
| 53 | + let ptr = unsafe { std::alloc::alloc_zeroed(layout) }; |
| 54 | + unsafe { |
| 55 | + std::alloc::dealloc(ptr, layout); |
| 56 | + } |
| 57 | +} |
0 commit comments