From ab99d0927bb4b47a5dabc94a3eecaedd3e5d024b Mon Sep 17 00:00:00 2001
From: Avril <flanchan@cumallover.me>
Date: Wed, 24 Mar 2021 18:17:03 +0000
Subject: [PATCH] test pointer is safely writeable

---
 src/lib.rs | 35 ++++++++++++++++++++++++-----------
 1 file changed, 24 insertions(+), 11 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 4519f4a..2649456 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -9,19 +9,32 @@ mod tests {
     {
 	use std::ffi::c_void;
 
-	let mut size: usize = 100;
-	extern "C" fn callback(ptr: *mut c_void, data: *mut c_void)
-	{
-	    let size = unsafe {&mut *(data as *mut usize)};
-	    println!("From callback! Size is {}", *size);
+	let size: usize = 100;
+	let output = {
+	    let mut size: usize = size;
+	    extern "C" fn callback(ptr: *mut c_void, data: *mut c_void)
+	    {
+		let size = unsafe {&mut *(data as *mut usize)};
+		let slice = unsafe {
+		    std::ptr::write_bytes(ptr, 0, *size);
+		    std::slice::from_raw_parts_mut(ptr as *mut u8, *size)
+		};
+		println!("From callback! Size is {}", slice.len());
 
-	    *size = 0;
-	}
+		for (i, x) in (0..).zip(slice.iter_mut())
+		{
+		    *x = i as u8;
+		}
 
-	unsafe {
-	    super::ffi::alloca_trampoline(size, callback, &mut size as *mut usize as *mut _);
-	}
+		*size = slice.iter().map(|&x| x as usize).sum::<usize>();
+	    }
 
-	assert_eq!(size, 0);
+	    unsafe {
+		super::ffi::alloca_trampoline(size, callback, &mut size as *mut usize as *mut _);
+	    }
+	    size
+	};
+
+	assert_eq!(output, (0..size).sum::<usize>());
     }
 }