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

//! This file contains small helper functions relating to WASM memory management
//! and serialization used throughout the HDK.

use crate::api::G_MEM_STACK;
use holochain_core_types::error::RibosomeEncodingBits;
use holochain_json_api::json::JsonString;
pub use holochain_wasm_utils::api_serialization::validation::*;
use holochain_wasm_utils::memory::{
    allocation::{AllocationError, AllocationResult, WasmAllocation},
    stack::WasmStack,
};
use std::convert::{TryFrom, TryInto};

/// Init global memory stack
pub fn init_global_memory(initial_allocation: WasmAllocation) -> AllocationResult {
    unsafe {
        G_MEM_STACK = Some(WasmStack::try_from(initial_allocation)?);
    }
    Ok(initial_allocation)
}

/// sugar
pub fn init_global_memory_from_ribosome_encoding(
    encoded_value: RibosomeEncodingBits,
) -> AllocationResult {
    init_global_memory(WasmAllocation::try_from_ribosome_encoding(encoded_value)?)
}

/// Serialize output as json in WASM memory
pub fn write_json<J: TryInto<JsonString>>(jsonable: J) -> AllocationResult {
    let mut mem_stack = unsafe {
        match G_MEM_STACK {
            Some(mem_stack) => mem_stack,
            None => return Err(AllocationError::BadStackAlignment),
        }
    };
    mem_stack.write_json(jsonable)
}