avoid zero sized allocation in false

This commit is contained in:
Brendan Hansknecht 2023-03-13 14:08:11 -07:00
parent b94a046b0d
commit d88894c365
No known key found for this signature in database
GPG key ID: 0EA784685083E75B

View file

@ -134,18 +134,26 @@ pub extern "C" fn rust_main() -> i32 {
unsafe fn call_the_closure(closure_data_ptr: *const u8) -> i64 {
let size = size_Fx_result() as usize;
let layout = Layout::array::<u8>(size).unwrap();
let buffer = std::alloc::alloc(layout) as *mut u8;
if size == 0 {
call_Fx(
// This flags pointer will never get dereferenced
MaybeUninit::uninit().as_ptr(),
closure_data_ptr as *const u8,
std::ptr::null_mut(),
);
} else {
let layout = Layout::array::<u8>(size).unwrap();
let buffer = std::alloc::alloc(layout) as *mut u8;
call_Fx(
// This flags pointer will never get dereferenced
MaybeUninit::uninit().as_ptr(),
closure_data_ptr as *const u8,
buffer as *mut u8,
);
std::alloc::dealloc(buffer, layout);
call_Fx(
// This flags pointer will never get dereferenced
MaybeUninit::uninit().as_ptr(),
closure_data_ptr as *const u8,
buffer as *mut u8,
);
std::alloc::dealloc(buffer, layout);
}
0
}