zeroize: add proxy_alloc_test

Author: newpavlov

zeroize/tests/alloc.rs

@@ -0,0 +1,48 @@
+use std::alloc::{GlobalAlloc, Layout, System};
+
+use zeroize::Zeroize;
+
+// Allocator that ensures that deallocated data is zeroized.
+struct ProxyAllocator;
+
+unsafe impl GlobalAlloc for ProxyAllocator {
+    unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
+        unsafe { System.alloc(layout) }
+    }
+
+    unsafe fn dealloc(&self, ptr: *mut u8, layout: Layout) {
+        if layout.size() == 160 {
+            for i in 0..layout.size() {
+                let b = unsafe { core::ptr::read(ptr.add(i)) };
+                if b != 0 {
+                    panic!()
+                }
+            }
+        }
+
+        unsafe { System.dealloc(ptr, layout) }
+    }
+}
+
+#[global_allocator]
+static PROXY_ALLOCATOR: ProxyAllocator = ProxyAllocator;
+
+struct SecretBox<S: Zeroize>(Box<S>);
+
+impl<S: Zeroize> SecretBox<S> {
+    fn new(val: S) -> Self {
+        Self(Box::new(val))
+    }
+}
+
+impl<S: Zeroize> Drop for SecretBox<S> {
+    fn drop(&mut self) {
+        self.0.as_mut().zeroize()
+    }
+}
+
+#[test]
+fn proxy_alloc_test() {
+    let _b1 = SecretBox::new([u128::MAX; 10]);
+    let _b2 = SecretBox::new([u8::MAX; 160]);
+}