Update plugin dependencies

This commit is contained in:
Isaac Clayton 2022-06-03 11:18:54 +02:00
parent 0cf64d6fba
commit 35b2eff29c
4 changed files with 30 additions and 33 deletions

24
Cargo.lock generated
View file

@ -3614,10 +3614,21 @@ name = "plugin"
version = "0.1.0" version = "0.1.0"
dependencies = [ dependencies = [
"bincode", "bincode",
"rust_plugin_macros", "plugin_macros",
"serde", "serde",
] ]
[[package]]
name = "plugin_macros"
version = "0.1.0"
dependencies = [
"bincode",
"proc-macro2",
"quote",
"serde",
"syn",
]
[[package]] [[package]]
name = "png" name = "png"
version = "0.16.8" version = "0.16.8"
@ -4320,17 +4331,6 @@ dependencies = [
"walkdir", "walkdir",
] ]
[[package]]
name = "rust_plugin_macros"
version = "0.1.0"
dependencies = [
"bincode",
"proc-macro2",
"quote",
"serde",
"syn",
]
[[package]] [[package]]
name = "rustc-demangle" name = "rustc-demangle"
version = "0.1.21" version = "0.1.21"

View file

@ -6,4 +6,4 @@ edition = "2021"
[dependencies] [dependencies]
serde = "1.0" serde = "1.0"
bincode = "1.3" bincode = "1.3"
rust_plugin_macros = { path = "../plugin_macros" } plugin_macros = { path = "../plugin_macros" }

View file

@ -1,7 +1,7 @@
#[repr(C)] #[repr(C)]
pub struct Buffer { pub struct __Buffer {
ptr: *const u8, pub ptr: *const u8,
len: usize, pub len: usize,
} }
/// Allocates a buffer with an exact size. /// Allocates a buffer with an exact size.
@ -9,7 +9,7 @@ pub struct Buffer {
#[no_mangle] #[no_mangle]
pub extern "C" fn __alloc_buffer(len: usize) -> *const u8 { pub extern "C" fn __alloc_buffer(len: usize) -> *const u8 {
let vec = vec![0; len]; let vec = vec![0; len];
let buffer = unsafe { Buffer::from_vec(vec) }; let buffer = unsafe { __Buffer::from_vec(vec) };
return buffer.ptr; return buffer.ptr;
} }
@ -21,31 +21,30 @@ pub extern "C" fn __alloc_buffer(len: usize) -> *const u8 {
// std::mem::drop(vec); // std::mem::drop(vec);
// } // }
impl Buffer { impl __Buffer {
#[inline(always)] #[inline(always)]
pub unsafe fn to_vec(&self) -> Vec<u8> { pub unsafe fn to_vec(&self) -> Vec<u8> {
slice::from_raw_parts(self.ptr, self.len).to_vec() core::slice::from_raw_parts(self.ptr, self.len).to_vec()
} }
#[inline(always)] #[inline(always)]
pub unsafe fn from_vec(mut vec: Vec<u8>) -> Buffer { pub unsafe fn from_vec(mut vec: Vec<u8>) -> __Buffer {
vec.shrink_to(0); vec.shrink_to(0);
let ptr = vec.as_ptr(); let ptr = vec.as_ptr();
let len = vec.len(); let len = vec.len();
std::mem::forget(vec); std::mem::forget(vec);
Buffer { ptr, len } __Buffer { ptr, len }
} }
#[inline(always)] #[inline(always)]
pub fn leak_to_heap(self) -> *const Buffer { pub fn leak_to_heap(self) -> *const __Buffer {
let boxed = Box::new(self); let boxed = Box::new(self);
let ptr = Box::<Buffer>::into_raw(boxed) as *const Buffer; let ptr = Box::<__Buffer>::into_raw(boxed) as *const __Buffer;
return ptr; return ptr;
} }
} }
pub mod prelude { pub mod prelude {
pub use super::{Buffer, __alloc_buffer}; pub use super::{__Buffer, __alloc_buffer};
#[macro_use]
pub use plugin_macros::bind; pub use plugin_macros::bind;
} }

View file

@ -2,7 +2,7 @@ use core::panic;
use proc_macro::TokenStream; use proc_macro::TokenStream;
use quote::{format_ident, quote}; use quote::{format_ident, quote};
use syn::{parse_macro_input, ItemFn, VisPublic, Visibility}; use syn::{parse_macro_input, ItemFn, Visibility};
#[proc_macro_attribute] #[proc_macro_attribute]
pub fn bind(args: TokenStream, function: TokenStream) -> TokenStream { pub fn bind(args: TokenStream, function: TokenStream) -> TokenStream {
@ -20,25 +20,23 @@ pub fn bind(args: TokenStream, function: TokenStream) -> TokenStream {
let outer_fn_name = format_ident!("__{}", inner_fn_name); let outer_fn_name = format_ident!("__{}", inner_fn_name);
TokenStream::from(quote! { TokenStream::from(quote! {
use serde;
#[no_mangle] #[no_mangle]
#inner_fn #inner_fn
#[no_mangle] #[no_mangle]
pub extern "C" fn #outer_fn_name(ptr: *const u8, len: usize) -> *const Buffer { pub extern "C" fn #outer_fn_name(ptr: *const u8, len: usize) -> *const ::plugin::__Buffer {
// setup // setup
let buffer = Buffer { ptr, len }; let buffer = ::plugin::__Buffer { ptr, len };
let data = unsafe { buffer.to_vec() }; let data = unsafe { buffer.to_vec() };
// operation // operation
let argument = bincode::deserialize(&data).unwrap(); let argument = ::bincode::deserialize(&data).unwrap();
let result = #inner_fn_name(argument); let result = #inner_fn_name(argument);
let new_data: Result<Vec<u8>, _> = bincode::serialize(&result); let new_data: Result<Vec<u8>, _> = ::bincode::serialize(&result);
let new_data = new_data.unwrap(); let new_data = new_data.unwrap();
// teardown // teardown
let new_buffer = unsafe { Buffer::from_vec(new_data) }; let new_buffer = unsafe { ::plugin::__Buffer::from_vec(new_data) };
return new_buffer.leak_to_heap(); return new_buffer.leak_to_heap();
} }
}) })