lib.rs

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