dep-protobuf/rust/cpp_kernel/rust_alloc_for_cpp_api.rs
Protobuf Team Bot c07de7c9df Change to proto2_rust C prefix and proto2::rust C++ namespace
PiperOrigin-RevId: 648791688
2024-07-02 11:44:17 -07:00

20 lines
733 B
Rust

// Protocol Buffers - Google's data interchange format
// Copyright 2023 Google LLC. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd
use std::alloc::{alloc, Layout};
#[no_mangle]
extern "C" fn proto2_rust_alloc(size: usize, align: usize) -> *mut u8 {
if size == 0 {
// A 0-sized layout is legal but the global allocator isn't required to support
// it so return a dangling pointer instead.
std::ptr::null_mut::<u8>().wrapping_add(align)
} else {
let layout = Layout::from_size_align(size, align).unwrap();
unsafe { alloc(layout) }
}
}