lib.rs

 1pub use bincode;
 2pub use serde;
 3
 4// TODO: move the implementation to one place?
 5pub struct __Buffer {
 6    pub ptr: u32, // *const u8,
 7    pub len: u32, // usize,
 8}
 9
10impl __Buffer {
11    pub fn into_u64(self) -> u64 {
12        ((self.ptr as u64) << 32) | (self.len as u64)
13    }
14
15    pub fn from_u64(packed: u64) -> Self {
16        __Buffer {
17            ptr: (packed >> 32) as u32,
18            len: packed as u32,
19        }
20    }
21}
22
23/// Allocates a buffer with an exact size.
24/// We don't return the size because it has to be passed in anyway.
25#[no_mangle]
26pub extern "C" fn __alloc_buffer(len: u32) -> u32 {
27    let vec = vec![0; len as usize];
28    let buffer = unsafe { __Buffer::from_vec(vec) };
29    return buffer.ptr;
30}
31
32/// Frees a given buffer, requires the size.
33#[no_mangle]
34pub extern "C" fn __free_buffer(buffer: u64) {
35    let vec = unsafe { __Buffer::from_u64(buffer).to_vec() };
36    std::mem::drop(vec);
37}
38
39impl __Buffer {
40    #[inline(always)]
41    pub unsafe fn to_vec(&self) -> Vec<u8> {
42        core::slice::from_raw_parts(self.ptr as *const u8, self.len as usize).to_vec()
43    }
44
45    #[inline(always)]
46    pub unsafe fn from_vec(mut vec: Vec<u8>) -> __Buffer {
47        vec.shrink_to(0);
48        let ptr = vec.as_ptr() as u32;
49        let len = vec.len() as u32;
50        std::mem::forget(vec);
51        __Buffer { ptr, len }
52    }
53}
54
55pub mod prelude {
56    pub use super::{__Buffer, __alloc_buffer};
57    pub use plugin_macros::{export, import};
58}