marine/examples/ipfs_node/wasm/ipfs_rpc/src/mem.rs

/*
 * Copyright 2020 Fluence Labs Limited
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

use crate::log_utf8_string;
use std::alloc::{alloc as global_alloc, dealloc as global_dealloc, Layout};
use std::ptr::NonNull;

/// Allocates memory area of specified size and returns its address.
#[no_mangle]
pub unsafe fn allocate(size: usize) -> NonNull<u8> {
    let layout: Layout = Layout::from_size_align(size, std::mem::align_of::<u8>()).unwrap();

    let msg = format!("wasm_rpc: calling allocate with {}\n", size);
    log_utf8_string(msg.as_ptr() as _, msg.len() as _);

    NonNull::new_unchecked(global_alloc(layout))
}

/// Deallocates memory area for provided memory pointer and size.
#[no_mangle]
pub unsafe fn deallocate(ptr: NonNull<u8>, size: usize) {
    let layout = Layout::from_size_align(size, std::mem::align_of::<u8>()).unwrap();

    let msg = format!("wasm_rpc: calling deallocate with {:?} {}\n", ptr, size);
    log_utf8_string(msg.as_ptr() as _, msg.len() as _);

    global_dealloc(ptr.as_ptr(), layout);
}