61 lines · 1.6 KB · Rust
RawDownload
pub use bincode;pub use serde; /// This is the buffer that is used Wasm side./// Note that it mirrors the functionality of/// the `WasiBuffer` found in `plugin_runtime/src/plugin.rs`,/// But has a few different methods.pub struct __Buffer {    pub ptr: u32, // *const u8,    pub len: u32, // usize,} impl __Buffer {    pub fn into_u64(self) -> u64 {        ((self.ptr as u64) << 32) | (self.len as u64)    }     pub fn from_u64(packed: u64) -> Self {        __Buffer {            ptr: (packed >> 32) as u32,            len: packed as u32,        }    }} /// Allocates a buffer with an exact size./// We don't return the size because it has to be passed in anyway.#[no_mangle]pub extern "C" fn __alloc_buffer(len: u32) -> u32 {    let vec = vec![0; len as usize];    let buffer = unsafe { __Buffer::from_vec(vec) };    buffer.ptr} /// Frees a given buffer, requires the size.#[no_mangle]pub extern "C" fn __free_buffer(buffer: u64) {    let vec = unsafe { __Buffer::from_u64(buffer).to_vec() };    std::mem::drop(vec);} impl __Buffer {    #[inline(always)]    pub unsafe fn to_vec(&self) -> Vec<u8> {        core::slice::from_raw_parts(self.ptr as *const u8, self.len as usize).to_vec()    }     #[inline(always)]    pub unsafe fn from_vec(mut vec: Vec<u8>) -> __Buffer {        vec.shrink_to(0);        let ptr = vec.as_ptr() as u32;        let len = vec.len() as u32;        std::mem::forget(vec);        __Buffer { ptr, len }    }} pub mod prelude {    pub use super::{__Buffer, __alloc_buffer};    pub use plugin_macros::{export, import};}