1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

//! Contains support for writing smart contracts.

pub mod account;
pub mod runtime;
pub mod storage;
pub mod system;

use alloc::{
    alloc::{alloc, Layout},
    vec::Vec,
};
use core::{mem, ptr::NonNull};

use casperlabs_types::{bytesrepr::ToBytes, ApiError};

use crate::unwrap_or_revert::UnwrapOrRevert;

/// Calculates size and alignment for an array of T.
const fn size_align_for_array<T>(n: usize) -> (usize, usize) {
    (n * mem::size_of::<T>(), mem::align_of::<T>())
}

/// Allocates bytes
pub fn alloc_bytes(n: usize) -> NonNull<u8> {
    let (size, align) = size_align_for_array::<u8>(n);
    // We treat allocated memory as raw bytes, that will be later passed to deserializer which also
    // operates on raw bytes.
    let layout = Layout::from_size_align(size, align)
        .map_err(|_| ApiError::AllocLayout)
        .unwrap_or_revert();
    let raw_ptr = unsafe { alloc(layout) };
    NonNull::new(raw_ptr)
        .ok_or(ApiError::OutOfMemory)
        .unwrap_or_revert()
}

fn to_ptr<T: ToBytes>(t: T) -> (*const u8, usize, Vec<u8>) {
    let bytes = t.into_bytes().unwrap_or_revert();
    let ptr = bytes.as_ptr();
    let size = bytes.len();
    (ptr, size, bytes)
}