lib.rs

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