Files
hakorune/tools/plugin-tester/src/main.rs

678 lines
27 KiB
Rust
Raw Normal View History

//! Nyash Plugin Tester
//!
//! プラグイン開発者向けの診断ツール
//! Box名を決め打ちせず、プラグインから取得する
use clap::Parser;
use colored::*;
use libloading::{Library, Symbol};
use std::ffi::{CStr, CString};
use std::os::raw::{c_char, c_void};
use std::path::PathBuf;
use std::io::Write;
// ============ FFI Types (プラグインと同じ定義) ============
#[repr(C)]
pub struct NyashHostVtable {
pub alloc: unsafe extern "C" fn(size: usize) -> *mut u8,
pub free: unsafe extern "C" fn(ptr: *mut u8),
pub wake: unsafe extern "C" fn(handle: u64),
pub log: unsafe extern "C" fn(level: i32, msg: *const c_char),
}
#[repr(C)]
pub struct NyashMethodInfo {
pub method_id: u32,
pub name: *const c_char,
pub signature: u32,
}
#[repr(C)]
pub struct NyashPluginInfo {
pub type_id: u32,
pub type_name: *const c_char,
pub method_count: usize,
pub methods: *const NyashMethodInfo,
}
// ============ CLI Arguments ============
#[derive(Parser, Debug)]
#[command(name = "plugin-tester")]
#[command(about = "Nyash plugin diagnostic tool", long_about = None)]
struct Args {
/// Action to perform
#[command(subcommand)]
command: Commands,
}
#[derive(clap::Subcommand, Debug)]
enum Commands {
/// Check plugin and display information
Check {
/// Path to plugin .so file
plugin: PathBuf,
},
/// Test plugin lifecycle (birth/fini)
Lifecycle {
/// Path to plugin .so file
plugin: PathBuf,
},
/// File I/O end-to-end test (open/write/read/close)
Io {
/// Path to plugin .so file
plugin: PathBuf,
},
fix(plugins): Fix FileBox plugin I/O operations and add TLV debugging ## FileBox Plugin Fixes ### Issue Resolution - Fixed file write operations not persisting to disk - Fixed TLV tag mismatch preventing read operations - Added comprehensive TLV debugging capability ### Changes Made #### plugins/nyash-filebox-plugin/src/lib.rs - Added file.flush() after write operations to ensure data persistence - Modified tlv_parse_bytes to accept both String(6) and Bytes(7) tags - Resolves compatibility issue with plugin-tester String encoding #### tools/plugin-tester/src/main.rs - Added `tlv-debug` subcommand with detailed TLV analysis - Provides hex dumps, encoding/decoding verification - Full plugin round-trip testing with file I/O validation - Detailed error analysis for debugging TLV protocol issues ## Test Results ### Before Fix ``` INFO: WRITE 8 bytes INFO: READ 0 bytes ← Problem: no data read ✗: Plugin round-trip failed\! ``` ### After Fix ``` INFO: WRITE 8 bytes INFO: READ 8 bytes ← Fixed: data successfully read ✓: Plugin round-trip successful\! ``` ## Technical Details - Root cause: Missing file.flush() + TLV tag type mismatch - Plugin-tester sends String(6), plugin expected Bytes(7) only - File buffer not flushed to disk before close/reopen - Solution: Added flush() + dual tag support for compatibility This completes the core BID-FFI plugin I/O functionality validation. Nyash integration still needs method dispatch optimization. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
2025-08-18 11:21:04 +09:00
/// Debug TLV encoding/decoding with detailed output
TlvDebug {
/// Path to plugin .so file
plugin: PathBuf,
/// Test message to encode/decode
#[arg(short, long, default_value = "Hello TLV Debug!")]
message: String,
},
/// Validate plugin type information against nyash.toml
Typecheck {
/// Path to plugin .so file
plugin: PathBuf,
/// Path to nyash.toml configuration file
#[arg(short, long, default_value = "../../nyash.toml")]
config: PathBuf,
},
}
// ============ Host Functions (テスト用実装) ============
unsafe extern "C" fn test_alloc(size: usize) -> *mut u8 {
let layout = std::alloc::Layout::from_size_align(size, 8).unwrap();
std::alloc::alloc(layout)
}
unsafe extern "C" fn test_free(ptr: *mut u8) {
if !ptr.is_null() {
// サイズ情報が必要だが、簡易実装のため省略
}
}
unsafe extern "C" fn test_wake(_handle: u64) {
// テスト用なので何もしない
}
unsafe extern "C" fn test_log(level: i32, msg: *const c_char) {
if !msg.is_null() {
let c_str = CStr::from_ptr(msg);
let message = c_str.to_string_lossy();
match level {
0 => println!("{}: {}", "DEBUG".blue(), message),
1 => println!("{}: {}", "INFO".green(), message),
2 => println!("{}: {}", "WARN".yellow(), message),
3 => println!("{}: {}", "ERROR".red(), message),
_ => println!("{}: {}", "UNKNOWN".white(), message),
}
}
}
static HOST_VTABLE: NyashHostVtable = NyashHostVtable {
alloc: test_alloc,
free: test_free,
wake: test_wake,
log: test_log,
};
// ============ Main Functions ============
fn main() {
let args = Args::parse();
match args.command {
Commands::Check { plugin } => check_plugin(&plugin),
Commands::Lifecycle { plugin } => test_lifecycle(&plugin),
Commands::Io { plugin } => test_file_io(&plugin),
fix(plugins): Fix FileBox plugin I/O operations and add TLV debugging ## FileBox Plugin Fixes ### Issue Resolution - Fixed file write operations not persisting to disk - Fixed TLV tag mismatch preventing read operations - Added comprehensive TLV debugging capability ### Changes Made #### plugins/nyash-filebox-plugin/src/lib.rs - Added file.flush() after write operations to ensure data persistence - Modified tlv_parse_bytes to accept both String(6) and Bytes(7) tags - Resolves compatibility issue with plugin-tester String encoding #### tools/plugin-tester/src/main.rs - Added `tlv-debug` subcommand with detailed TLV analysis - Provides hex dumps, encoding/decoding verification - Full plugin round-trip testing with file I/O validation - Detailed error analysis for debugging TLV protocol issues ## Test Results ### Before Fix ``` INFO: WRITE 8 bytes INFO: READ 0 bytes ← Problem: no data read ✗: Plugin round-trip failed\! ``` ### After Fix ``` INFO: WRITE 8 bytes INFO: READ 8 bytes ← Fixed: data successfully read ✓: Plugin round-trip successful\! ``` ## Technical Details - Root cause: Missing file.flush() + TLV tag type mismatch - Plugin-tester sends String(6), plugin expected Bytes(7) only - File buffer not flushed to disk before close/reopen - Solution: Added flush() + dual tag support for compatibility This completes the core BID-FFI plugin I/O functionality validation. Nyash integration still needs method dispatch optimization. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
2025-08-18 11:21:04 +09:00
Commands::TlvDebug { plugin, message } => test_tlv_debug(&plugin, &message),
}
}
// ============ Minimal BID-1 TLV Helpers ============
#[repr(C)]
#[derive(Clone, Copy)]
struct TlvHeader { version: u16, argc: u16 }
const TLV_VERSION: u16 = 1;
#[repr(u8)]
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
enum Tag { Bool=1, I32=2, I64=3, F32=4, F64=5, String=6, Bytes=7, Handle=8, Void=9 }
fn tlv_encode_string(s: &str, buf: &mut Vec<u8>) {
let header_pos = buf.len();
buf.extend_from_slice(&[0,0,0,0]);
let mut argc: u16 = 0;
// entry
let bytes = s.as_bytes();
buf.push(Tag::String as u8);
buf.push(0);
buf.extend_from_slice(&(bytes.len() as u16).to_le_bytes());
buf.extend_from_slice(bytes);
argc += 1;
// write header
buf[header_pos..header_pos+2].copy_from_slice(&TLV_VERSION.to_le_bytes());
buf[header_pos+2..header_pos+4].copy_from_slice(&argc.to_le_bytes());
}
fn tlv_encode_two_strings(a: &str, b: &str, buf: &mut Vec<u8>) {
let header_pos = buf.len();
buf.extend_from_slice(&[0,0,0,0]);
let mut argc: u16 = 0;
for s in [a,b] {
let bytes = s.as_bytes();
buf.push(Tag::String as u8);
buf.push(0);
buf.extend_from_slice(&(bytes.len() as u16).to_le_bytes());
buf.extend_from_slice(bytes);
argc += 1;
}
buf[header_pos..header_pos+2].copy_from_slice(&TLV_VERSION.to_le_bytes());
buf[header_pos+2..header_pos+4].copy_from_slice(&argc.to_le_bytes());
}
fn tlv_encode_i32(v: i32, buf: &mut Vec<u8>) {
let header_pos = buf.len();
buf.extend_from_slice(&[0,0,0,0]);
buf.push(Tag::I32 as u8);
buf.push(0);
buf.extend_from_slice(&4u16.to_le_bytes());
buf.extend_from_slice(&v.to_le_bytes());
buf[header_pos..header_pos+2].copy_from_slice(&TLV_VERSION.to_le_bytes());
buf[header_pos+2..header_pos+4].copy_from_slice(&1u16.to_le_bytes());
}
fn tlv_encode_bytes(data: &[u8], buf: &mut Vec<u8>) {
let header_pos = buf.len();
buf.extend_from_slice(&[0,0,0,0]);
buf.push(Tag::Bytes as u8);
buf.push(0);
buf.extend_from_slice(&(data.len() as u16).to_le_bytes());
buf.extend_from_slice(data);
buf[header_pos..header_pos+2].copy_from_slice(&TLV_VERSION.to_le_bytes());
buf[header_pos+2..header_pos+4].copy_from_slice(&1u16.to_le_bytes());
}
fn tlv_decode_first(bytes: &[u8]) -> Option<(u8, &[u8])> {
if bytes.len() < 4 { return None; }
let argc = u16::from_le_bytes([bytes[2], bytes[3]]);
if argc == 0 { return None; }
if bytes.len() < 8 { return None; }
let tag = bytes[4];
let size = u16::from_le_bytes([bytes[6], bytes[7]]) as usize;
if bytes.len() < 8+size { return None; }
Some((tag, &bytes[8..8+size]))
}
fn check_plugin(path: &PathBuf) {
println!("{}", "=== Nyash Plugin Checker ===".bold());
println!("Plugin: {}", path.display());
// プラグインをロード
let library = match unsafe { Library::new(path) } {
Ok(lib) => lib,
Err(e) => {
eprintln!("{}: Failed to load plugin: {}", "ERROR".red(), e);
return;
}
};
println!("{}: Plugin loaded successfully", "".green());
// ABI version確認
unsafe {
let abi_fn: Symbol<unsafe extern "C" fn() -> u32> = match library.get(b"nyash_plugin_abi") {
Ok(f) => f,
Err(e) => {
eprintln!("{}: nyash_plugin_abi not found: {}", "ERROR".red(), e);
return;
}
};
let abi_version = abi_fn();
println!("{}: ABI version: {}", "".green(), abi_version);
if abi_version != 1 {
eprintln!("{}: Unsupported ABI version (expected 1)", "WARNING".yellow());
}
}
// Plugin初期化とBox名取得
unsafe {
let init_fn: Symbol<unsafe extern "C" fn(*const NyashHostVtable, *mut NyashPluginInfo) -> i32> =
match library.get(b"nyash_plugin_init") {
Ok(f) => f,
Err(e) => {
eprintln!("{}: nyash_plugin_init not found: {}", "ERROR".red(), e);
return;
}
};
let mut plugin_info = std::mem::zeroed::<NyashPluginInfo>();
let result = init_fn(&HOST_VTABLE, &mut plugin_info);
if result != 0 {
eprintln!("{}: nyash_plugin_init failed with code {}", "ERROR".red(), result);
return;
}
println!("{}: Plugin initialized", "".green());
// 重要Box名をプラグインから取得決め打ちしない
let box_name = if plugin_info.type_name.is_null() {
"<unknown>".to_string()
} else {
CStr::from_ptr(plugin_info.type_name).to_string_lossy().to_string()
};
println!("\n{}", "Plugin Information:".bold());
println!(" Box Type: {} (ID: {})", box_name.cyan(), plugin_info.type_id);
println!(" Methods: {}", plugin_info.method_count);
// メソッド一覧表示
if plugin_info.method_count > 0 && !plugin_info.methods.is_null() {
println!("\n{}", "Methods:".bold());
let methods = std::slice::from_raw_parts(plugin_info.methods, plugin_info.method_count);
for method in methods {
let method_name = if method.name.is_null() {
"<unnamed>".to_string()
} else {
CStr::from_ptr(method.name).to_string_lossy().to_string()
};
let method_type = match method.method_id {
0 => " (constructor)".yellow(),
id if id == u32::MAX => " (destructor)".yellow(),
_ => "".normal(),
};
println!(" - {} [ID: {}, Sig: 0x{:08X}]{}",
method_name,
method.method_id,
method.signature,
method_type
);
}
}
}
// シャットダウン
unsafe {
if let Ok(shutdown_fn) = library.get::<Symbol<unsafe extern "C" fn()>>(b"nyash_plugin_shutdown") {
shutdown_fn();
println!("\n{}: Plugin shutdown completed", "".green());
}
}
println!("\n{}", "Check completed!".green().bold());
}
fn test_lifecycle(path: &PathBuf) {
println!("{}", "=== Lifecycle Test ===".bold());
println!("Testing birth/fini for: {}", path.display());
// プラグインをロード
let library = match unsafe { Library::new(path) } {
Ok(lib) => lib,
Err(e) => {
eprintln!("{}: Failed to load plugin: {}", "ERROR".red(), e);
return;
}
};
unsafe {
// ABI version
let abi_fn: Symbol<unsafe extern "C" fn() -> u32> = match library.get(b"nyash_plugin_abi") {
Ok(f) => f,
Err(e) => {
eprintln!("{}: nyash_plugin_abi not found: {}", "ERROR".red(), e);
return;
}
};
let abi_version = abi_fn();
println!("{}: ABI version: {}", "".green(), abi_version);
if abi_version != 1 {
eprintln!("{}: Unsupported ABI version (expected 1)", "WARNING".yellow());
}
// init
let init_fn: Symbol<unsafe extern "C" fn(*const NyashHostVtable, *mut NyashPluginInfo) -> i32> =
match library.get(b"nyash_plugin_init") {
Ok(f) => f,
Err(e) => {
eprintln!("{}: nyash_plugin_init not found: {}", "ERROR".red(), e);
return;
}
};
let mut plugin_info = std::mem::zeroed::<NyashPluginInfo>();
let result = init_fn(&HOST_VTABLE, &mut plugin_info);
if result != 0 {
eprintln!("{}: nyash_plugin_init failed with code {}", "ERROR".red(), result);
return;
}
println!("{}: Plugin initialized", "".green());
// invoke
let invoke_fn: Symbol<unsafe extern "C" fn(u32, u32, u32, *const u8, usize, *mut u8, *mut usize) -> i32> =
match library.get(b"nyash_plugin_invoke") {
Ok(f) => f,
Err(e) => {
eprintln!("{}: nyash_plugin_invoke not found: {}", "ERROR".red(), e);
return;
}
};
let type_id = plugin_info.type_id;
println!("{}: BoxType ID = {}", "i".blue(), type_id);
// birth
let mut out = [0u8; 8];
let mut out_len: usize = out.len();
let rc = invoke_fn(type_id, 0, 0, std::ptr::null(), 0, out.as_mut_ptr(), &mut out_len as *mut usize);
if rc != 0 {
eprintln!("{}: birth invoke failed with code {}", "ERROR".red(), rc);
return;
}
if out_len < 4 {
eprintln!("{}: birth returned too small result ({} bytes)", "ERROR".red(), out_len);
return;
}
let instance_id = u32::from_le_bytes(out[0..4].try_into().unwrap());
println!("{}: birth → instance_id={}", "".green(), instance_id);
// fini
let rc = invoke_fn(type_id, u32::MAX, instance_id, std::ptr::null(), 0, std::ptr::null_mut(), std::ptr::null_mut());
if rc != 0 {
eprintln!("{}: fini invoke failed with code {}", "ERROR".red(), rc);
return;
}
println!("{}: fini → instance {} cleaned", "".green(), instance_id);
// shutdown
if let Ok(shutdown_fn) = library.get::<Symbol<unsafe extern "C" fn()>>(b"nyash_plugin_shutdown") {
shutdown_fn();
println!("{}: Plugin shutdown completed", "".green());
}
}
println!("\n{}", "Lifecycle test completed!".green().bold());
}
fn test_file_io(path: &PathBuf) {
println!("{}", "=== File I/O Test ===".bold());
println!("Testing open/write/read/close: {}", path.display());
// Load
let library = match unsafe { Library::new(path) } {
Ok(lib) => lib,
Err(e) => {
eprintln!("{}: Failed to load plugin: {}", "ERROR".red(), e);
return;
}
};
unsafe {
let abi: Symbol<unsafe extern "C" fn() -> u32> = library.get(b"nyash_plugin_abi").unwrap();
println!("{}: ABI version: {}", "".green(), abi());
let init: Symbol<unsafe extern "C" fn(*const NyashHostVtable, *mut NyashPluginInfo) -> i32> = library.get(b"nyash_plugin_init").unwrap();
let mut info = std::mem::zeroed::<NyashPluginInfo>();
assert_eq!(0, init(&HOST_VTABLE, &mut info));
let invoke: Symbol<unsafe extern "C" fn(u32,u32,u32,*const u8,usize,*mut u8,*mut usize)->i32> = library.get(b"nyash_plugin_invoke").unwrap();
let shutdown: Symbol<unsafe extern "C" fn()> = library.get(b"nyash_plugin_shutdown").unwrap();
// birth
let mut buf_len: usize = 0;
let rc = invoke(info.type_id, 0, 0, std::ptr::null(), 0, std::ptr::null_mut(), &mut buf_len as *mut usize);
assert!(rc == -1 && buf_len >= 4, "unexpected birth preflight");
let mut out = vec![0u8; buf_len];
let mut out_len = buf_len;
assert_eq!(0, invoke(info.type_id, 0, 0, std::ptr::null(), 0, out.as_mut_ptr(), &mut out_len as *mut usize));
let instance_id = u32::from_le_bytes(out[0..4].try_into().unwrap());
println!("{}: birth → instance_id={}", "".green(), instance_id);
// open: write mode
let mut args = Vec::new();
let test_path = "plugins/nyash-filebox-plugin/target/test_io.txt";
tlv_encode_two_strings(test_path, "w", &mut args);
let mut res_len: usize = 0;
let rc = invoke(info.type_id, 1, instance_id, args.as_ptr(), args.len(), std::ptr::null_mut(), &mut res_len as *mut usize);
assert!(rc == -1 || rc == 0);
let mut res = vec![0u8; res_len.max(4)];
let mut rl = res_len;
let _ = invoke(info.type_id, 1, instance_id, args.as_ptr(), args.len(), res.as_mut_ptr(), &mut rl as *mut usize);
println!("{}: open(w)", "".green());
// write
let content = b"Hello from plugin-tester!";
let mut wargs = Vec::new();
tlv_encode_bytes(content, &mut wargs);
let mut rlen: usize = 0;
let rc = invoke(info.type_id, 3, instance_id, wargs.as_ptr(), wargs.len(), std::ptr::null_mut(), &mut rlen as *mut usize);
assert!(rc == -1 || rc == 0);
let mut wb = vec![0u8; rlen.max(8)];
let mut rl2 = rlen;
let _ = invoke(info.type_id, 3, instance_id, wargs.as_ptr(), wargs.len(), wb.as_mut_ptr(), &mut rl2 as *mut usize);
if let Some((tag, payload)) = tlv_decode_first(&wb[..rl2]) {
assert_eq!(tag, Tag::I32 as u8);
let mut n = [0u8;4]; n.copy_from_slice(payload);
let written = i32::from_le_bytes(n);
println!("{}: write {} bytes", "".green(), written);
}
// close
let mut clen: usize = 0;
let _ = invoke(info.type_id, 4, instance_id, std::ptr::null(), 0, std::ptr::null_mut(), &mut clen as *mut usize);
let mut cb = vec![0u8; clen.max(4)]; let mut cbl = clen; let _ = invoke(info.type_id, 4, instance_id, std::ptr::null(), 0, cb.as_mut_ptr(), &mut cbl as *mut usize);
println!("{}: close", "".green());
// reopen read
let mut args2 = Vec::new(); tlv_encode_two_strings(test_path, "r", &mut args2);
let mut r0: usize = 0; let _ = invoke(info.type_id, 1, instance_id, args2.as_ptr(), args2.len(), std::ptr::null_mut(), &mut r0 as *mut usize);
let mut ob = vec![0u8; r0.max(4)]; let mut obl=r0; let _=invoke(info.type_id,1,instance_id,args2.as_ptr(),args2.len(),ob.as_mut_ptr(),&mut obl as *mut usize);
println!("{}: open(r)", "".green());
// read 1024
let mut rargs = Vec::new(); tlv_encode_i32(1024, &mut rargs);
let mut rneed: usize = 0; let rc = invoke(info.type_id, 2, instance_id, rargs.as_ptr(), rargs.len(), std::ptr::null_mut(), &mut rneed as *mut usize);
assert!(rc == -1 || rc == 0);
let mut rb = vec![0u8; rneed.max(16)]; let mut rbl=rneed; let rc2=invoke(info.type_id,2,instance_id,rargs.as_ptr(),rargs.len(),rb.as_mut_ptr(),&mut rbl as *mut usize);
if rc2 != 0 { println!("{}: read rc={} (expected 0)", "WARN".yellow(), rc2); }
if let Some((tag, payload)) = tlv_decode_first(&rb[..rbl]) {
assert_eq!(tag, Tag::Bytes as u8);
let s = String::from_utf8_lossy(payload).to_string();
println!("{}: read {} bytes → '{}'", "".green(), payload.len(), s);
} else {
println!("{}: read decode failed (len={})", "WARN".yellow(), rbl);
}
// close & shutdown
let mut clen2: usize = 0; let _=invoke(info.type_id,4,instance_id,std::ptr::null(),0,std::ptr::null_mut(),&mut clen2 as *mut usize);
shutdown();
println!("\n{}", "File I/O test completed!".green().bold());
}
}
fix(plugins): Fix FileBox plugin I/O operations and add TLV debugging ## FileBox Plugin Fixes ### Issue Resolution - Fixed file write operations not persisting to disk - Fixed TLV tag mismatch preventing read operations - Added comprehensive TLV debugging capability ### Changes Made #### plugins/nyash-filebox-plugin/src/lib.rs - Added file.flush() after write operations to ensure data persistence - Modified tlv_parse_bytes to accept both String(6) and Bytes(7) tags - Resolves compatibility issue with plugin-tester String encoding #### tools/plugin-tester/src/main.rs - Added `tlv-debug` subcommand with detailed TLV analysis - Provides hex dumps, encoding/decoding verification - Full plugin round-trip testing with file I/O validation - Detailed error analysis for debugging TLV protocol issues ## Test Results ### Before Fix ``` INFO: WRITE 8 bytes INFO: READ 0 bytes ← Problem: no data read ✗: Plugin round-trip failed\! ``` ### After Fix ``` INFO: WRITE 8 bytes INFO: READ 8 bytes ← Fixed: data successfully read ✓: Plugin round-trip successful\! ``` ## Technical Details - Root cause: Missing file.flush() + TLV tag type mismatch - Plugin-tester sends String(6), plugin expected Bytes(7) only - File buffer not flushed to disk before close/reopen - Solution: Added flush() + dual tag support for compatibility This completes the core BID-FFI plugin I/O functionality validation. Nyash integration still needs method dispatch optimization. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
2025-08-18 11:21:04 +09:00
fn test_tlv_debug(path: &PathBuf, message: &str) {
println!("{}", "=== TLV Debug Test ===".bold());
println!("Testing TLV encoding/decoding with: '{}'", message);
// Load plugin
let library = match unsafe { Library::new(path) } {
Ok(lib) => lib,
Err(e) => {
eprintln!("{}: Failed to load plugin: {}", "ERROR".red(), e);
return;
}
};
unsafe {
let abi: Symbol<unsafe extern "C" fn() -> u32> = library.get(b"nyash_plugin_abi").unwrap();
println!("{}: ABI version: {}", "".green(), abi());
let init: Symbol<unsafe extern "C" fn(*const NyashHostVtable, *mut NyashPluginInfo) -> i32> = library.get(b"nyash_plugin_init").unwrap();
let mut info = std::mem::zeroed::<NyashPluginInfo>();
assert_eq!(0, init(&HOST_VTABLE, &mut info));
let invoke: Symbol<unsafe extern "C" fn(u32,u32,u32,*const u8,usize,*mut u8,*mut usize)->i32> = library.get(b"nyash_plugin_invoke").unwrap();
let shutdown: Symbol<unsafe extern "C" fn()> = library.get(b"nyash_plugin_shutdown").unwrap();
// Test TLV encoding
println!("\n{}", "--- Encoding Test ---".cyan());
let mut encoded = Vec::new();
tlv_encode_string(message, &mut encoded);
println!("Original message: '{}'", message);
println!("Encoded TLV ({} bytes): {:02x?}", encoded.len(), encoded);
// Hex dump for readability
print!("Hex dump: ");
for (i, byte) in encoded.iter().enumerate() {
if i % 16 == 0 && i > 0 { print!("\n "); }
print!("{:02x} ", byte);
}
println!();
// Test TLV decoding
println!("\n{}", "--- Decoding Test ---".cyan());
if let Some((tag, payload)) = tlv_decode_first(&encoded) {
println!("Decoded tag: {} ({})", tag,
match tag {
6 => "String",
7 => "Bytes",
_ => "Unknown"
});
println!("Decoded payload ({} bytes): {:02x?}", payload.len(), payload);
if tag == Tag::String as u8 || tag == Tag::Bytes as u8 {
let decoded_str = String::from_utf8_lossy(payload);
println!("Decoded string: '{}'", decoded_str);
if decoded_str == message {
println!("{}: TLV round-trip successful!", "".green());
} else {
println!("{}: TLV round-trip failed! Expected: '{}', Got: '{}'",
"".red(), message, decoded_str);
}
}
} else {
println!("{}: Failed to decode TLV!", "".red());
}
// Test with plugin write/read
println!("\n{}", "--- Plugin Round-trip Test ---".cyan());
// birth
let mut buf_len: usize = 0;
let rc = invoke(info.type_id, 0, 0, std::ptr::null(), 0, std::ptr::null_mut(), &mut buf_len);
assert!(rc == -1 && buf_len >= 4);
let mut out = vec![0u8; buf_len];
let mut out_len = buf_len;
assert_eq!(0, invoke(info.type_id, 0, 0, std::ptr::null(), 0, out.as_mut_ptr(), &mut out_len));
let instance_id = u32::from_le_bytes(out[0..4].try_into().unwrap());
println!("{}: birth → instance_id={}", "".green(), instance_id);
// Test file write
let test_path = "plugins/nyash-filebox-plugin/target/tlv_debug_test.txt";
let mut args = Vec::new();
tlv_encode_two_strings(test_path, "w", &mut args);
println!("Write args TLV ({} bytes): {:02x?}", args.len(), &args[..args.len().min(32)]);
let mut need: usize = 0;
let _ = invoke(info.type_id, 1, instance_id, args.as_ptr(), args.len(), std::ptr::null_mut(), &mut need);
let mut obuf = vec![0u8; need.max(4)];
let mut olen = need;
let _ = invoke(info.type_id, 1, instance_id, args.as_ptr(), args.len(), obuf.as_mut_ptr(), &mut olen);
println!("{}: open(w) successful", "".green());
// Write test message
let mut write_args = Vec::new();
tlv_encode_string(message, &mut write_args);
println!("Write message TLV ({} bytes): {:02x?}", write_args.len(), &write_args[..write_args.len().min(32)]);
let mut wneed: usize = 0;
let _ = invoke(info.type_id, 3, instance_id, write_args.as_ptr(), write_args.len(), std::ptr::null_mut(), &mut wneed);
let mut wbuf = vec![0u8; wneed.max(4)];
let mut wlen = wneed;
let _ = invoke(info.type_id, 3, instance_id, write_args.as_ptr(), write_args.len(), wbuf.as_mut_ptr(), &mut wlen);
println!("{}: write successful", "".green());
// Close
let mut clen: usize = 0;
let _ = invoke(info.type_id, 4, instance_id, std::ptr::null(), 0, std::ptr::null_mut(), &mut clen);
let mut cb = vec![0u8; clen.max(4)];
let mut cbl = clen;
let _ = invoke(info.type_id, 4, instance_id, std::ptr::null(), 0, cb.as_mut_ptr(), &mut cbl);
println!("{}: close successful", "".green());
// Reopen for read
let mut read_args = Vec::new();
tlv_encode_two_strings(test_path, "r", &mut read_args);
let mut rneed: usize = 0;
let _ = invoke(info.type_id, 1, instance_id, read_args.as_ptr(), read_args.len(), std::ptr::null_mut(), &mut rneed);
let mut robuf = vec![0u8; rneed.max(4)];
let mut rolen = rneed;
let _ = invoke(info.type_id, 1, instance_id, read_args.as_ptr(), read_args.len(), robuf.as_mut_ptr(), &mut rolen);
println!("{}: open(r) successful", "".green());
// Read back
let mut size_args = Vec::new();
tlv_encode_i32(1024, &mut size_args);
let mut read_need: usize = 0;
let rc = invoke(info.type_id, 2, instance_id, size_args.as_ptr(), size_args.len(), std::ptr::null_mut(), &mut read_need);
println!("Read preflight: rc={}, need={} bytes", rc, read_need);
let mut read_buf = vec![0u8; read_need.max(16)];
let mut read_len = read_need;
let rc2 = invoke(info.type_id, 2, instance_id, size_args.as_ptr(), size_args.len(), read_buf.as_mut_ptr(), &mut read_len);
println!("Read actual: rc={}, got={} bytes", rc2, read_len);
if read_len > 0 {
println!("Read result TLV ({} bytes): {:02x?}", read_len, &read_buf[..read_len.min(32)]);
// Try to decode
if let Some((tag, payload)) = tlv_decode_first(&read_buf[..read_len]) {
println!("Read decoded tag: {} ({})", tag,
match tag {
6 => "String",
7 => "Bytes",
_ => "Unknown"
});
let read_message = String::from_utf8_lossy(payload);
println!("Read decoded message: '{}'", read_message);
if read_message == message {
println!("{}: Plugin round-trip successful!", "".green());
} else {
println!("{}: Plugin round-trip failed! Expected: '{}', Got: '{}'",
"".red(), message, read_message);
}
} else {
println!("{}: Failed to decode read result!", "".red());
// Show detailed hex analysis
if read_len >= 4 {
let version = u16::from_le_bytes([read_buf[0], read_buf[1]]);
let argc = u16::from_le_bytes([read_buf[2], read_buf[3]]);
println!("TLV Header analysis: version={}, argc={}", version, argc);
if read_len >= 8 {
let entry_tag = read_buf[4];
let entry_reserved = read_buf[5];
let entry_len = u16::from_le_bytes([read_buf[6], read_buf[7]]);
println!("First entry: tag={}, reserved={}, len={}", entry_tag, entry_reserved, entry_len);
}
}
}
}
shutdown();
println!("\n{}", "TLV Debug test completed!".green().bold());
}
}