runtime(plugin): unify to Nyash ABI v2 dispatch via shim; loader drops legacy nyash_plugin_invoke and records TypeBox invoke_id; all PluginBoxV2 invoke paths use v2 shim

This commit is contained in:
Selfhosting Dev
2025-09-17 20:42:01 +09:00
parent 4ea3ca2685
commit 913ca8b731
4 changed files with 93 additions and 41 deletions

View File

@ -1,9 +1,27 @@
// Host bridge helpers for TypeBox invoke (minimal, v2) // Host bridge helpers for TypeBox invoke (v2)
pub type InvokeFn = // Library-level shim signature used across the runtime (compat convenience)
unsafe extern "C" fn(u32, u32, u32, *const u8, usize, *mut u8, *mut usize) -> i32; pub type InvokeFn = unsafe extern "C" fn(
u32, /* type_id (for dispatch) */
u32, /* method_id */
u32, /* instance_id */
*const u8,
usize,
*mut u8,
*mut usize,
) -> i32;
// Call invoke_id with a temporary output buffer; returns (code, bytes_written, buffer) // Native v2 per-Box signature
pub type BoxInvokeFn = extern "C" fn(
u32, /* instance_id */
u32, /* method_id */
*const u8,
usize,
*mut u8,
*mut usize,
) -> i32;
// Call library-level shim with a temporary output buffer
pub fn invoke_alloc( pub fn invoke_alloc(
invoke: InvokeFn, invoke: InvokeFn,
type_id: u32, type_id: u32,

View File

@ -1,3 +1,4 @@
use super::host_bridge::BoxInvokeFn;
use super::types::{LoadedPluginV2, NyashTypeBoxFfi, PluginBoxMetadata, PluginBoxV2, PluginHandleInner}; use super::types::{LoadedPluginV2, NyashTypeBoxFfi, PluginBoxMetadata, PluginBoxV2, PluginHandleInner};
use crate::bid::{BidError, BidResult}; use crate::bid::{BidError, BidResult};
use crate::box_trait::NyashBox; use crate::box_trait::NyashBox;
@ -12,7 +13,7 @@ fn dbg_on() -> bool {
std::env::var("NYASH_DEBUG_PLUGIN").unwrap_or_default() == "1" std::env::var("NYASH_DEBUG_PLUGIN").unwrap_or_default() == "1"
} }
type BoxInvokeFn = extern "C" fn(u32, u32, *const u8, usize, *mut u8, *mut usize) -> i32; // (alias imported from host_bridge)
#[derive(Debug, Clone, Default)] #[derive(Debug, Clone, Default)]
struct LoadedBoxSpec { struct LoadedBoxSpec {
@ -128,33 +129,24 @@ impl PluginLoaderV2 {
let lib = unsafe { Library::new(&lib_path) }.map_err(|_| BidError::PluginError)?; let lib = unsafe { Library::new(&lib_path) }.map_err(|_| BidError::PluginError)?;
let lib_arc = Arc::new(lib); let lib_arc = Arc::new(lib);
// Resolve required invoke symbol (legacy library-level): nyash_plugin_invoke
unsafe {
let invoke_sym: Symbol<
unsafe extern "C" fn(u32, u32, u32, *const u8, usize, *mut u8, *mut usize) -> i32,
> = lib_arc
.get(b"nyash_plugin_invoke\0")
.map_err(|_| BidError::PluginError)?;
// Optional init (best-effort) // Optional init (best-effort)
unsafe {
if let Ok(init_sym) = if let Ok(init_sym) =
lib_arc.get::<Symbol<unsafe extern "C" fn() -> i32>>(b"nyash_plugin_init\0") lib_arc.get::<Symbol<unsafe extern "C" fn() -> i32>>(b"nyash_plugin_init\0")
{ {
let _ = init_sym(); let _ = init_sym();
} }
}
let loaded = LoadedPluginV2 { let loaded = LoadedPluginV2 {
_lib: lib_arc.clone(), _lib: lib_arc.clone(),
box_types: lib_def.boxes.clone(), box_types: lib_def.boxes.clone(),
typeboxes: HashMap::new(), typeboxes: HashMap::new(),
init_fn: None, init_fn: None,
invoke_fn: *invoke_sym,
}; };
self.plugins self.plugins
.write() .write()
.map_err(|_| BidError::PluginError)? .map_err(|_| BidError::PluginError)?
.insert(lib_name.to_string(), Arc::new(loaded)); .insert(lib_name.to_string(), Arc::new(loaded));
}
// Try to resolve Nyash ABI v2 per-box TypeBox symbols and record invoke_id // Try to resolve Nyash ABI v2 per-box TypeBox symbols and record invoke_id
// Symbol pattern: nyash_typebox_<BoxType> // Symbol pattern: nyash_typebox_<BoxType>
@ -227,13 +219,26 @@ impl PluginLoaderV2 {
None None
} }
/// Lookup per-Box invoke function pointer for given type_id via loaded TypeBox specs
pub fn box_invoke_fn_for_type_id(&self, type_id: u32) -> Option<BoxInvokeFn> {
let config = self.config.as_ref()?;
let cfg_path = self.config_path.as_ref()?;
let toml_str = std::fs::read_to_string(cfg_path).ok()?;
let toml_value: toml::Value = toml::from_str(&toml_str).ok()?;
let (lib_name, box_type) = self.find_box_by_type_id(config, &toml_value, type_id)?;
let key = (lib_name.to_string(), box_type.to_string());
let map = self.box_specs.read().ok()?;
let spec = map.get(&key)?;
spec.invoke_id
}
pub fn metadata_for_type_id(&self, type_id: u32) -> Option<PluginBoxMetadata> { pub fn metadata_for_type_id(&self, type_id: u32) -> Option<PluginBoxMetadata> {
let config = self.config.as_ref()?; let config = self.config.as_ref()?;
let cfg_path = self.config_path.as_ref()?; let cfg_path = self.config_path.as_ref()?;
let toml_str = std::fs::read_to_string(cfg_path).ok()?; let toml_str = std::fs::read_to_string(cfg_path).ok()?;
let toml_value: toml::Value = toml::from_str(&toml_str).ok()?; let toml_value: toml::Value = toml::from_str(&toml_str).ok()?;
let (lib_name, box_type) = self.find_box_by_type_id(config, &toml_value, type_id)?; let (lib_name, box_type) = self.find_box_by_type_id(config, &toml_value, type_id)?;
let plugin = { let _plugin = {
let plugins = self.plugins.read().ok()?; let plugins = self.plugins.read().ok()?;
plugins.get(lib_name)?.clone() plugins.get(lib_name)?.clone()
}; };
@ -262,7 +267,7 @@ impl PluginLoaderV2 {
lib_name: lib_name.to_string(), lib_name: lib_name.to_string(),
box_type: box_type.to_string(), box_type: box_type.to_string(),
type_id: resolved_type, type_id: resolved_type,
invoke_fn: plugin.invoke_fn, invoke_fn: super::super::nyash_plugin_invoke_v2_shim,
fini_method_id: fini_method, fini_method_id: fini_method,
}) })
} }
@ -278,7 +283,7 @@ impl PluginLoaderV2 {
let toml_value: toml::Value = toml::from_str(&toml_str).ok()?; let toml_value: toml::Value = toml::from_str(&toml_str).ok()?;
let (lib_name, box_type) = self.find_box_by_type_id(config, &toml_value, type_id)?; let (lib_name, box_type) = self.find_box_by_type_id(config, &toml_value, type_id)?;
let plugins = self.plugins.read().ok()?; let plugins = self.plugins.read().ok()?;
let plugin = plugins.get(lib_name)?.clone(); let _plugin = plugins.get(lib_name)?.clone();
let fini_method_id = if let Some(spec) = self let fini_method_id = if let Some(spec) = self
.box_specs .box_specs
.read() .read()
@ -293,7 +298,7 @@ impl PluginLoaderV2 {
let bx = super::types::construct_plugin_box( let bx = super::types::construct_plugin_box(
box_type.to_string(), box_type.to_string(),
type_id, type_id,
plugin.invoke_fn, super::super::nyash_plugin_invoke_v2_shim,
instance_id, instance_id,
fini_method_id, fini_method_id,
); );
@ -329,7 +334,7 @@ impl PluginLoaderV2 {
.map_err(|_| BidError::PluginError)?; .map_err(|_| BidError::PluginError)?;
let config = self.config.as_ref().ok_or(BidError::PluginError)?; let config = self.config.as_ref().ok_or(BidError::PluginError)?;
let plugins = self.plugins.read().unwrap(); let plugins = self.plugins.read().unwrap();
let plugin = plugins.get(lib_name).ok_or(BidError::PluginError)?; let _plugin = plugins.get(lib_name).ok_or(BidError::PluginError)?;
let type_id = if let Some(spec) = self let type_id = if let Some(spec) = self
.box_specs .box_specs
.read() .read()
@ -348,7 +353,7 @@ impl PluginLoaderV2 {
let out_len = out.len(); let out_len = out.len();
let tlv_args = crate::runtime::plugin_ffi_common::encode_empty_args(); let tlv_args = crate::runtime::plugin_ffi_common::encode_empty_args();
let (birth_result, _len, out_vec) = let (birth_result, _len, out_vec) =
super::host_bridge::invoke_alloc(plugin.invoke_fn, type_id, 0, 0, &tlv_args); super::host_bridge::invoke_alloc(super::super::nyash_plugin_invoke_v2_shim, type_id, 0, 0, &tlv_args);
let out = out_vec; let out = out_vec;
if birth_result != 0 || out_len < 4 { if birth_result != 0 || out_len < 4 {
return Err(BidError::PluginError); return Err(BidError::PluginError);
@ -369,7 +374,7 @@ impl PluginLoaderV2 {
}; };
let handle = Arc::new(PluginHandleInner { let handle = Arc::new(PluginHandleInner {
type_id, type_id,
invoke_fn: plugin.invoke_fn, invoke_fn: super::super::nyash_plugin_invoke_v2_shim,
instance_id, instance_id,
fini_method_id: fini_id, fini_method_id: fini_id,
finalized: std::sync::atomic::AtomicBool::new(false), finalized: std::sync::atomic::AtomicBool::new(false),
@ -583,11 +588,11 @@ impl PluginLoaderV2 {
.ok_or(BidError::InvalidMethod)?; .ok_or(BidError::InvalidMethod)?;
// Get plugin handle // Get plugin handle
let plugins = self.plugins.read().map_err(|_| BidError::PluginError)?; let plugins = self.plugins.read().map_err(|_| BidError::PluginError)?;
let plugin = plugins.get(lib_name).ok_or(BidError::PluginError)?; let _plugin = plugins.get(lib_name).ok_or(BidError::PluginError)?;
// Encode TLV args via shared helper (numeric→string→toString) // Encode TLV args via shared helper (numeric→string→toString)
let tlv = crate::runtime::plugin_ffi_common::encode_args(args); let tlv = crate::runtime::plugin_ffi_common::encode_args(args);
let (_code, out_len, out) = super::host_bridge::invoke_alloc( let (_code, out_len, out) = super::host_bridge::invoke_alloc(
plugin.invoke_fn, super::super::nyash_plugin_invoke_v2_shim,
type_id, type_id,
method.method_id, method.method_id,
instance_id, instance_id,
@ -629,7 +634,7 @@ impl PluginLoaderV2 {
{ {
let handle = Arc::new(super::types::PluginHandleInner { let handle = Arc::new(super::types::PluginHandleInner {
type_id: ret_type, type_id: ret_type,
invoke_fn: plugin.invoke_fn, invoke_fn: super::super::nyash_plugin_invoke_v2_shim,
instance_id: inst, instance_id: inst,
fini_method_id: None, fini_method_id: None,
finalized: std::sync::atomic::AtomicBool::new(false), finalized: std::sync::atomic::AtomicBool::new(false),
@ -686,8 +691,7 @@ impl PluginLoaderV2 {
let fini_id = box_conf.methods.get("fini").map(|m| m.method_id); let fini_id = box_conf.methods.get("fini").map(|m| m.method_id);
// Get loaded plugin invoke // Get loaded plugin invoke
let plugins = self.plugins.read().map_err(|_| BidError::PluginError)?; let _plugins = self.plugins.read().map_err(|_| BidError::PluginError)?;
let plugin = plugins.get(lib_name).ok_or(BidError::PluginError)?;
// Call birth (no args TLV) and read returned instance id (little-endian u32 in bytes 0..4) // Call birth (no args TLV) and read returned instance id (little-endian u32 in bytes 0..4)
if dbg_on() { if dbg_on() {
@ -697,8 +701,13 @@ impl PluginLoaderV2 {
); );
} }
let tlv = crate::runtime::plugin_ffi_common::encode_empty_args(); let tlv = crate::runtime::plugin_ffi_common::encode_empty_args();
let (code, out_len, out_buf) = let (code, out_len, out_buf) = super::host_bridge::invoke_alloc(
super::host_bridge::invoke_alloc(plugin.invoke_fn, type_id, birth_id, 0, &tlv); super::super::nyash_plugin_invoke_v2_shim,
type_id,
birth_id,
0,
&tlv,
);
if dbg_on() { if dbg_on() {
eprintln!("[PluginLoaderV2] create_box: box_type={} type_id={} birth_id={} code={} out_len={}", box_type, type_id, birth_id, code, out_len); eprintln!("[PluginLoaderV2] create_box: box_type={} type_id={} birth_id={} code={} out_len={}", box_type, type_id, birth_id, code, out_len);
if out_len > 0 { if out_len > 0 {
@ -717,7 +726,7 @@ impl PluginLoaderV2 {
box_type: box_type.to_string(), box_type: box_type.to_string(),
inner: Arc::new(PluginHandleInner { inner: Arc::new(PluginHandleInner {
type_id, type_id,
invoke_fn: plugin.invoke_fn, invoke_fn: super::super::nyash_plugin_invoke_v2_shim,
instance_id, instance_id,
fini_method_id: fini_id, fini_method_id: fini_id,
finalized: std::sync::atomic::AtomicBool::new(false), finalized: std::sync::atomic::AtomicBool::new(false),

View File

@ -17,6 +17,32 @@ pub fn metadata_for_type_id(type_id: u32) -> Option<PluginBoxMetadata> {
guard.metadata_for_type_id(type_id) guard.metadata_for_type_id(type_id)
} }
/// Resolve per-Box invoke function for a type_id (v2 only)
pub fn box_invoke_for_type_id(type_id: u32) -> Option<super::enabled::host_bridge::BoxInvokeFn> {
let loader = get_global_loader_v2();
let guard = loader.read().ok()?;
guard.box_invoke_fn_for_type_id(type_id)
}
/// Library-level shim to dispatch a v2 per-Box invoke function using type_id
#[no_mangle]
pub extern "C" fn nyash_plugin_invoke_v2_shim(
type_id: u32,
method_id: u32,
instance_id: u32,
args: *const u8,
args_len: usize,
result: *mut u8,
result_len: *mut usize,
) -> i32 {
if let Some(f) = box_invoke_for_type_id(type_id) {
// Safety: Plugin-provided function pointer; adhere to C ABI
return unsafe { f(instance_id, method_id, args, args_len, result, result_len) };
}
// E_PLUGIN (-5) when not found
-5
}
pub fn backend_kind() -> &'static str { pub fn backend_kind() -> &'static str {
"enabled" "enabled"
} }

View File

@ -13,7 +13,6 @@ pub struct LoadedPluginV2 {
pub(super) box_types: Vec<String>, pub(super) box_types: Vec<String>,
pub(super) typeboxes: std::collections::HashMap<String, usize>, pub(super) typeboxes: std::collections::HashMap<String, usize>,
pub(super) init_fn: Option<unsafe extern "C" fn() -> i32>, pub(super) init_fn: Option<unsafe extern "C" fn() -> i32>,
pub(super) invoke_fn: InvokeFn,
} }
#[derive(Clone)] #[derive(Clone)]