Skip to main content

vm/
native.rs

1use super::FnVariant;
2use crate::JITRunTime;
3use crate::memory::{alloc_dynamic, alloc_struct_bytes, take_dynamic_return};
4use anyhow::{Result, anyhow};
5use cranelift::prelude::AbiParam;
6use cranelift_module::{Linkage, Module};
7use dynamic::{Dynamic, Type};
8use parser::{BinaryOp, Expr, ExprKind, Span};
9use rand::RngExt;
10use std::borrow::Cow;
11use std::collections::BTreeMap;
12use std::fmt::Write as _;
13use std::sync::{RwLock, Weak};
14
15#[derive(Clone, Debug)]
16pub struct ZustCallback {
17    pub fn_ptr: usize,
18    pub ret_ty: Type,
19    pub explicit_arg_len: usize,
20    pub captures: Vec<Dynamic>,
21}
22
23impl ZustCallback {
24    pub fn new(fn_ptr: usize, ret_ty: Type, captures: Vec<Dynamic>) -> Self {
25        Self { fn_ptr, ret_ty, explicit_arg_len: usize::MAX, captures }
26    }
27
28    pub fn new_with_arg_len(fn_ptr: usize, ret_ty: Type, explicit_arg_len: usize, captures: Vec<Dynamic>) -> Self {
29        Self { fn_ptr, ret_ty, explicit_arg_len, captures }
30    }
31
32    pub fn call0(&self) -> Result<Dynamic> {
33        self.call(Vec::new())
34    }
35
36    pub fn call1(&self, arg: Dynamic) -> Result<Dynamic> {
37        self.call(vec![arg])
38    }
39
40    pub fn call(&self, mut args: Vec<Dynamic>) -> Result<Dynamic> {
41        if self.explicit_arg_len != usize::MAX {
42            args.truncate(self.explicit_arg_len);
43            while args.len() < self.explicit_arg_len {
44                args.push(Dynamic::Null);
45            }
46        }
47        let args: Vec<Box<Dynamic>> = args.into_iter().map(Box::new).collect();
48        let mut ptrs: Vec<*const Dynamic> = args.iter().map(|arg| arg.as_ref() as *const Dynamic).collect();
49        ptrs.extend(self.captures.iter().map(|value| value as *const Dynamic));
50        call_jit_isolated(|| unsafe { call_callback(self.fn_ptr as *const u8, &self.ret_ty, &ptrs) })
51    }
52}
53
54extern "C" fn any_clone(addr: *const Dynamic) -> *const Dynamic {
55    //在堆上分配内存 复制 addr 到内存中
56    unsafe {
57        let cloned_value = (*addr).deep_clone();
58        alloc_dynamic(cloned_value)
59    }
60}
61
62extern "C" fn any_null() -> *const Dynamic {
63    //在堆上分配内存 复制 addr 到内存中
64    alloc_dynamic(Dynamic::Null)
65}
66
67extern "C" fn print(addr: *const Dynamic) {
68    if !addr.is_null() {
69        unsafe {
70            println!("{}", (*addr).to_string());
71        }
72    }
73}
74
75extern "C" fn sqrt(value: f64) -> f64 {
76    value.sqrt()
77}
78
79extern "C" fn log_any(addr: *const Dynamic) {
80    if addr.is_null() {
81        log::debug!("{:?}", Dynamic::Null);
82    } else {
83        log::debug!("{:?}", unsafe { &*addr });
84    }
85}
86
87extern "C" fn any_is_map(addr: *const Dynamic) -> bool {
88    !addr.is_null() && unsafe { (*addr).is_map() }
89}
90
91extern "C" fn any_is_list(addr: *const Dynamic) -> bool {
92    !addr.is_null() && unsafe { (*addr).is_list() }
93}
94
95extern "C" fn any_is_string(addr: *const Dynamic) -> bool {
96    !addr.is_null() && unsafe { (*addr).is_str() }
97}
98
99extern "C" fn any_is_null(addr: *const Dynamic) -> bool {
100    addr.is_null() || unsafe { (*addr).is_null() }
101}
102
103extern "C" fn random(start: *const Dynamic, stop: *const Dynamic) -> *const Dynamic {
104    if !start.is_null() && !stop.is_null() {
105        let mut rng = rand::rng();
106        unsafe {
107            if (&*start).is_int() {
108                let start = (*start).as_int().unwrap_or(0);
109                let stop = (*stop).as_int().unwrap_or(100);
110                return alloc_dynamic(Dynamic::I64(rng.random_range(start..stop)));
111            } else if (&*start).is_f32() || (&*start).is_f64() {
112                let start = (*start).as_float().unwrap_or(0.0);
113                let stop = (*stop).as_float().unwrap_or(1.0);
114                return alloc_dynamic(Dynamic::F64(rng.random_range(start..stop)));
115            }
116        }
117    }
118    alloc_dynamic(Dynamic::Null)
119}
120
121extern "C" fn uuid() -> *const Dynamic {
122    alloc_dynamic(uuid::Uuid::new_v4().to_string().into())
123}
124
125pub(crate) extern "C" fn struct_alloc(size: i64) -> *mut u8 {
126    let size = size.max(0) as usize;
127    let ptr = alloc_struct_bytes(size);
128    unsafe {
129        std::ptr::write_bytes(ptr, 0, size);
130    }
131    ptr
132}
133
134pub(crate) extern "C" fn repeat_fill(dst: *mut u8, pattern: u64, width: i64, len: i64) {
135    if dst.is_null() || width <= 0 || len <= 0 {
136        return;
137    }
138    let width = width as usize;
139    let len = len as usize;
140    let bytes = pattern.to_le_bytes();
141    unsafe {
142        if width == 1 {
143            std::ptr::write_bytes(dst, bytes[0], len);
144            return;
145        }
146        if !matches!(width, 2 | 4 | 8) {
147            return;
148        }
149        for idx in 0..len {
150            std::ptr::copy_nonoverlapping(bytes.as_ptr(), dst.add(idx * width), width);
151        }
152    }
153}
154
155pub(crate) extern "C" fn strcat(left: *const Dynamic, right: *const Dynamic) -> *const Dynamic {
156    let left = if left.is_null() { "" } else { unsafe { (&*left).as_str() } };
157    let right = if right.is_null() { "" } else { unsafe { (&*right).as_str() } };
158    let mut out = String::with_capacity(left.len() + right.len());
159    out.push_str(left);
160    out.push_str(right);
161    alloc_dynamic(Dynamic::StringBuf(out))
162}
163
164pub(crate) extern "C" fn strcat_i64(left: *const Dynamic, right: i64) -> *const Dynamic {
165    let left = if left.is_null() { "" } else { unsafe { (&*left).as_str() } };
166    let mut out = String::with_capacity(left.len() + 20);
167    out.push_str(left);
168    let _ = write!(&mut out, "{right}");
169    alloc_dynamic(Dynamic::StringBuf(out))
170}
171
172pub(crate) extern "C" fn strcat_assign(left: *mut Dynamic, right: *const Dynamic) -> *const Dynamic {
173    if left.is_null() {
174        return strcat(left, right);
175    }
176    let suffix = if right.is_null() {
177        Cow::Borrowed("")
178    } else if std::ptr::eq(left as *const Dynamic, right) {
179        Cow::Owned(unsafe { (&*right).to_string() })
180    } else {
181        let right = unsafe { &*right };
182        if right.is_str() { Cow::Borrowed(right.as_str()) } else { Cow::Owned(right.to_string()) }
183    };
184    unsafe {
185        match &mut *left {
186            Dynamic::StringBuf(text) => text.push_str(suffix.as_ref()),
187            Dynamic::String(text) => {
188                let mut out = String::with_capacity(text.len() + suffix.len());
189                out.push_str(text.as_str());
190                out.push_str(suffix.as_ref());
191                *left = Dynamic::StringBuf(out);
192            }
193            value => {
194                let prefix = value.to_string();
195                let mut out = String::with_capacity(prefix.len() + suffix.len());
196                out.push_str(&prefix);
197                out.push_str(suffix.as_ref());
198                *value = Dynamic::StringBuf(out);
199            }
200        }
201    }
202    left
203}
204
205pub(crate) extern "C" fn struct_from_ptr(addr: i64, ty: i64) -> *const Dynamic {
206    let ty = unsafe { (&*(ty as *const Type)).clone() };
207    alloc_dynamic(Dynamic::Struct { addr: addr as usize, ty })
208}
209
210pub(crate) extern "C" fn array_from_ptr(addr: i64, ty: i64) -> *const Dynamic {
211    if addr == 0 || ty == 0 {
212        return alloc_dynamic(Dynamic::Null);
213    }
214    let ty = unsafe { &*(ty as *const Type) };
215    alloc_dynamic(dynamic_from_ptr(addr as *const u8, ty))
216}
217
218pub(crate) extern "C" fn array_to_ptr(dst: *mut u8, value: *const Dynamic, ty: i64) {
219    if dst.is_null() || value.is_null() || ty == 0 {
220        return;
221    }
222    let ty = unsafe { &*(ty as *const Type) };
223    write_dynamic_to_ptr(dst, unsafe { &*value }, ty);
224}
225
226fn dynamic_from_ptr(addr: *const u8, ty: &Type) -> Dynamic {
227    if addr.is_null() {
228        return Dynamic::Null;
229    }
230    match ty {
231        Type::Bool => Dynamic::Bool(unsafe { std::ptr::read_unaligned(addr) } != 0),
232        Type::I8 => Dynamic::I8(unsafe { std::ptr::read_unaligned(addr as *const i8) }),
233        Type::U8 => Dynamic::U8(unsafe { std::ptr::read_unaligned(addr) }),
234        Type::I16 => Dynamic::I16(unsafe { std::ptr::read_unaligned(addr as *const i16) }),
235        Type::U16 => Dynamic::U16(unsafe { std::ptr::read_unaligned(addr as *const u16) }),
236        Type::I32 => Dynamic::I32(unsafe { std::ptr::read_unaligned(addr as *const i32) }),
237        Type::U32 => Dynamic::U32(unsafe { std::ptr::read_unaligned(addr as *const u32) }),
238        Type::I64 => Dynamic::I64(unsafe { std::ptr::read_unaligned(addr as *const i64) }),
239        Type::U64 => Dynamic::U64(unsafe { std::ptr::read_unaligned(addr as *const u64) }),
240        Type::F32 => Dynamic::F32(unsafe { std::ptr::read_unaligned(addr as *const f32) }),
241        Type::F64 => Dynamic::F64(unsafe { std::ptr::read_unaligned(addr as *const f64) }),
242        Type::Array(elem_ty, len) => {
243            let width = elem_ty.storage_width() as usize;
244            let values = (0..*len as usize).map(|idx| unsafe { dynamic_from_ptr(addr.add(idx * width), elem_ty) }).collect();
245            Dynamic::list(values)
246        }
247        Type::Struct { fields, .. } => {
248            let mut map = BTreeMap::new();
249            let (_, offsets) = Type::struct_layout(fields);
250            for ((name, field_ty), offset) in fields.iter().zip(offsets) {
251                let value = unsafe { dynamic_from_ptr(addr.add(offset as usize), field_ty) };
252                map.insert(name.clone(), value);
253            }
254            Dynamic::map(map)
255        }
256        _ => {
257            let ptr = unsafe { std::ptr::read_unaligned(addr as *const *const Dynamic) };
258            if ptr.is_null() { Dynamic::Null } else { unsafe { (&*ptr).deep_clone() } }
259        }
260    }
261}
262
263fn write_dynamic_to_ptr(dst: *mut u8, value: &Dynamic, ty: &Type) {
264    if dst.is_null() {
265        return;
266    }
267    match ty {
268        Type::Bool => unsafe { std::ptr::write_unaligned(dst, if value.is_true() { 1 } else { 0 }) },
269        Type::I8 => unsafe { std::ptr::write_unaligned(dst as *mut i8, value.clone().try_into().unwrap_or_default()) },
270        Type::U8 => unsafe { std::ptr::write_unaligned(dst, value.clone().try_into().unwrap_or_default()) },
271        Type::I16 => unsafe { std::ptr::write_unaligned(dst as *mut i16, value.clone().try_into().unwrap_or_default()) },
272        Type::U16 => unsafe { std::ptr::write_unaligned(dst as *mut u16, value.clone().try_into().unwrap_or_default()) },
273        Type::I32 => unsafe { std::ptr::write_unaligned(dst as *mut i32, value.clone().try_into().unwrap_or_default()) },
274        Type::U32 => unsafe { std::ptr::write_unaligned(dst as *mut u32, value.clone().try_into().unwrap_or_default()) },
275        Type::I64 => unsafe { std::ptr::write_unaligned(dst as *mut i64, value.clone().try_into().unwrap_or_default()) },
276        Type::U64 => unsafe { std::ptr::write_unaligned(dst as *mut u64, value.clone().try_into().unwrap_or_default()) },
277        Type::F32 => unsafe { std::ptr::write_unaligned(dst as *mut f32, f32::try_from(value.clone()).unwrap_or_default()) },
278        Type::F64 => unsafe { std::ptr::write_unaligned(dst as *mut f64, value.clone().try_into().unwrap_or_default()) },
279        Type::Array(elem_ty, len) => {
280            let width = elem_ty.storage_width() as usize;
281            for idx in 0..*len as usize {
282                let item = value.get_idx(idx).unwrap_or(Dynamic::Null);
283                unsafe { write_dynamic_to_ptr(dst.add(idx * width), &item, elem_ty) };
284            }
285        }
286        Type::Struct { fields, .. } => {
287            let (_, offsets) = Type::struct_layout(fields);
288            for ((name, field_ty), offset) in fields.iter().zip(offsets) {
289                let item = value.get_dynamic(name.as_str()).unwrap_or(Dynamic::Null);
290                unsafe { write_dynamic_to_ptr(dst.add(offset as usize), &item, field_ty) };
291            }
292        }
293        _ => {
294            let ptr = alloc_dynamic(value.deep_clone());
295            unsafe { std::ptr::write_unaligned(dst as *mut usize, ptr as usize) };
296        }
297    }
298}
299
300pub(crate) extern "C" fn import_with_vm(context: *const Weak<RwLock<JITRunTime>>, addr: *const Dynamic, path: *const Dynamic) -> bool {
301    if addr.is_null() || path.is_null() {
302        return false;
303    }
304    super::with_vm_context(context, |jit| {
305        let name = unsafe { &*addr }.as_str();
306        let path = unsafe { &*path }.as_str();
307        jit.import(name, path)
308    })
309    .map_err(|e| log::error!("import failed: {e:#}"))
310    .is_ok()
311}
312
313pub(crate) extern "C" fn spawn_with_vm(context: *const Weak<RwLock<JITRunTime>>, fn_name: *const Dynamic, args: *const Dynamic) -> bool {
314    if context.is_null() || fn_name.is_null() {
315        return false;
316    }
317    let fn_name = unsafe { (&*fn_name).as_str().to_string() };
318    if fn_name.is_empty() {
319        return false;
320    }
321    let args = if args.is_null() { Dynamic::Null } else { unsafe { (&*args).deep_clone() } };
322    let context = unsafe { (&*context).clone() };
323    std::thread::Builder::new()
324        .name(format!("zust:{fn_name}"))
325        .spawn(move || {
326            if let Err(err) = spawn_run(context, fn_name.as_str(), args) {
327                log::error!("spawn {fn_name} failed: {err:?}");
328            }
329        })
330        .is_ok()
331}
332
333fn spawn_args(args: Dynamic) -> Vec<Dynamic> {
334    match args {
335        Dynamic::Null => Vec::new(),
336        Dynamic::List(values) => values.read().unwrap().iter().map(Dynamic::deep_clone).collect(),
337        value => vec![value],
338    }
339}
340
341fn spawn_run(context: Weak<RwLock<JITRunTime>>, fn_name: &str, args: Dynamic) -> Result<()> {
342    let args = spawn_args(args);
343    if args.len() > 16 {
344        anyhow::bail!("spawn supports at most 16 args, got {}", args.len());
345    }
346    let arg_tys = vec![Type::Any; args.len()];
347    let (ptr, ret_ty) = super::with_vm_context(&context as *const Weak<RwLock<JITRunTime>>, |vm| vm.jit.write().unwrap().get_fn_ptr(fn_name, &arg_tys))?;
348    let args: Vec<Box<Dynamic>> = args.into_iter().map(Box::new).collect();
349    let ptrs: Vec<*const Dynamic> = args.iter().map(|arg| arg.as_ref() as *const Dynamic).collect();
350    call_jit_isolated(|| unsafe { call_spawned(ptr, &ret_ty, &ptrs) })
351}
352
353pub(crate) extern "C" fn spawn_ptr(fn_ptr: i64, ret_ty: i64, args: *const Dynamic) -> bool {
354    if fn_ptr == 0 || ret_ty == 0 {
355        return false;
356    }
357    let fn_ptr = fn_ptr as usize;
358    let ret_ty = unsafe { (&*(ret_ty as *const Type)).clone() };
359    let args = if args.is_null() { Dynamic::Null } else { unsafe { (&*args).deep_clone() } };
360    std::thread::Builder::new()
361        .name("zust:closure".to_string())
362        .spawn(move || {
363            if let Err(err) = spawn_run_ptr(fn_ptr, ret_ty, args) {
364                log::error!("spawn closure failed: {err:?}");
365            }
366        })
367        .is_ok()
368}
369
370pub(crate) extern "C" fn callback_new(fn_ptr: i64, ret_ty: i64, explicit_arg_len: i64, captures: *const Dynamic) -> *const Dynamic {
371    if fn_ptr == 0 || ret_ty == 0 {
372        return alloc_dynamic(Dynamic::Null);
373    }
374    let ret_ty = unsafe { (&*(ret_ty as *const Type)).clone() };
375    let explicit_arg_len = usize::try_from(explicit_arg_len).unwrap_or(usize::MAX);
376    let captures = if captures.is_null() {
377        Vec::new()
378    } else {
379        match unsafe { &*captures } {
380            Dynamic::List(values) => values.read().unwrap().iter().map(Dynamic::deep_clone).collect(),
381            value => vec![value.deep_clone()],
382        }
383    };
384    alloc_dynamic(Dynamic::custom(ZustCallback::new_with_arg_len(fn_ptr as usize, ret_ty, explicit_arg_len, captures)))
385}
386
387fn spawn_run_ptr(fn_ptr: usize, ret_ty: Type, args: Dynamic) -> Result<()> {
388    let args = spawn_args(args);
389    if args.len() > 16 {
390        anyhow::bail!("spawn supports at most 16 args, got {}", args.len());
391    }
392    let args: Vec<Box<Dynamic>> = args.into_iter().map(Box::new).collect();
393    let ptrs: Vec<*const Dynamic> = args.iter().map(|arg| arg.as_ref() as *const Dynamic).collect();
394    call_jit_isolated(|| unsafe { call_spawned(fn_ptr as *const u8, &ret_ty, &ptrs) })
395}
396
397/// 脚本执行的隔离边界。
398///
399/// JIT 代码无法返回 `Result`,运行期错误(整数除零等)通过 [`dynamic`] 的线程
400/// 局部 fault 标志上报(由 `__vm_arith_fault` 与 `Dynamic` 的除法守卫设置)。这里
401/// 在调用前清掉陈旧标志,调用后读取它,把"脚本出错"降级为一次失败的 `Result`,
402/// 而不是让进程崩溃。`catch_unwind` 额外兜住宿主侧 Rust 代码(参数编组、返回值
403/// 取出)的 panic。
404///
405/// 注意:若 panic 发生在 JIT 的 `extern "C"` 帧之内再向外穿越,Rust 默认会 abort,
406/// `catch_unwind` 无法拦截——这类路径靠的是各 native 助手不 panic(除零已改为置
407/// fault 标志)。
408pub(crate) fn call_jit_isolated<T>(f: impl FnOnce() -> Result<T>) -> Result<T> {
409    let _ = dynamic::take_fault();
410    let outcome = std::panic::catch_unwind(std::panic::AssertUnwindSafe(f));
411    match outcome {
412        Ok(inner) => match dynamic::take_fault() {
413            Some(reason) => Err(anyhow!("脚本运行期错误: {}", reason)),
414            None => inner,
415        },
416        Err(_) => Err(anyhow!("脚本执行 panic,已隔离")),
417    }
418}
419
420unsafe fn call_callback(ptr: *const u8, ret_ty: &Type, args: &[*const Dynamic]) -> Result<Dynamic> {
421    macro_rules! callback_arg_ty {
422        ($arg:ident) => {
423            *const Dynamic
424        };
425    }
426
427    macro_rules! callback_args {
428        ($body:ident $(, $extra:tt)*) => {
429            match args {
430                [] => $body!($($extra),*;),
431                [a] => $body!($($extra),*; a),
432                [a, b] => $body!($($extra),*; a, b),
433                [a, b, c] => $body!($($extra),*; a, b, c),
434                [a, b, c, d] => $body!($($extra),*; a, b, c, d),
435                [a, b, c, d, e] => $body!($($extra),*; a, b, c, d, e),
436                [a, b, c, d, e, f] => $body!($($extra),*; a, b, c, d, e, f),
437                [a, b, c, d, e, f, g] => $body!($($extra),*; a, b, c, d, e, f, g),
438                [a, b, c, d, e, f, g, h] => $body!($($extra),*; a, b, c, d, e, f, g, h),
439                [a, b, c, d, e, f, g, h, i] => $body!($($extra),*; a, b, c, d, e, f, g, h, i),
440                [a, b, c, d, e, f, g, h, i, j] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j),
441                [a, b, c, d, e, f, g, h, i, j, k] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k),
442                [a, b, c, d, e, f, g, h, i, j, k, l] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k, l),
443                [a, b, c, d, e, f, g, h, i, j, k, l, m] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k, l, m),
444                [a, b, c, d, e, f, g, h, i, j, k, l, m, n] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k, l, m, n),
445                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k, l, m, n, o),
446                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p),
447                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q),
448                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r),
449                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s),
450                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t),
451                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u),
452                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v),
453                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v, w] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v, w),
454                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v, w, x] => $body!($($extra),*; a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v, w, x),
455                _ => anyhow::bail!("callback supports at most 24 args including captures, got {}", args.len()),
456            }
457        };
458    }
459
460    macro_rules! call_void_body {
461        (; $($arg:ident),*) => {{
462            let fn_ptr: extern "C" fn($(callback_arg_ty!($arg)),*) = unsafe { std::mem::transmute(ptr) };
463            fn_ptr($(*$arg),*)
464        }};
465    }
466
467    macro_rules! call_void {
468        () => {
469            callback_args!(call_void_body)
470        };
471    }
472
473    macro_rules! call_ret_body {
474        ($ret:ty, $dynamic:expr; $($arg:ident),*) => {{
475            let fn_ptr: extern "C" fn($(callback_arg_ty!($arg)),*) -> $ret = unsafe { std::mem::transmute(ptr) };
476            $dynamic(fn_ptr($(*$arg),*))
477        }};
478    }
479
480    macro_rules! call_ret {
481        ($ret:ty, $dynamic:expr) => {{ callback_args!(call_ret_body, $ret, $dynamic) }};
482    }
483
484    if ret_ty.is_void() {
485        call_void!();
486        Ok(Dynamic::Null)
487    } else if ret_ty.is_bool() {
488        call_ret!(bool, |value| Ok(Dynamic::Bool(value)))
489    } else if ret_ty.is_float() {
490        if ret_ty.is_f64() { call_ret!(f64, |value| Ok(Dynamic::F64(value))) } else { call_ret!(f32, |value| Ok(Dynamic::F32(value))) }
491    } else if ret_ty.is_int() || ret_ty.is_uint() {
492        match ret_ty {
493            Type::I8 => call_ret!(i8, |value| Ok(Dynamic::I8(value))),
494            Type::U8 => call_ret!(u8, |value| Ok(Dynamic::U8(value))),
495            Type::I16 => call_ret!(i16, |value| Ok(Dynamic::I16(value))),
496            Type::U16 => call_ret!(u16, |value| Ok(Dynamic::U16(value))),
497            Type::I32 => call_ret!(i32, |value| Ok(Dynamic::I32(value))),
498            Type::U32 => call_ret!(u32, |value| Ok(Dynamic::U32(value))),
499            Type::I64 => call_ret!(i64, |value| Ok(Dynamic::I64(value))),
500            Type::U64 => call_ret!(u64, |value| Ok(Dynamic::U64(value))),
501            _ => unreachable!(),
502        }
503    } else if ret_ty.is_struct() || ret_ty.is_array() || ret_ty.is_vec() {
504        log::warn!("callback returns {ret_ty:?} — not supported, discarding");
505        call_ret!(*const u8, |_| Ok(Dynamic::Null))
506    } else {
507        call_ret!(*const Dynamic, |ptr| unsafe { Ok((*take_dynamic_return(ptr)).deep_clone()) })
508    }
509}
510
511unsafe fn call_spawned(ptr: *const u8, ret_ty: &Type, args: &[*const Dynamic]) -> Result<()> {
512    macro_rules! call_void {
513        () => {
514            match args {
515                [] => unsafe { std::mem::transmute::<_, extern "C" fn()>(ptr)() },
516                [a] => unsafe { std::mem::transmute::<_, extern "C" fn(*const Dynamic)>(ptr)(*a) },
517                [a, b] => unsafe { std::mem::transmute::<_, extern "C" fn(*const Dynamic, *const Dynamic)>(ptr)(*a, *b) },
518                [a, b, c] => unsafe { std::mem::transmute::<_, extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic)>(ptr)(*a, *b, *c) },
519                [a, b, c, d] => unsafe { std::mem::transmute::<_, extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic)>(ptr)(*a, *b, *c, *d) },
520                [a, b, c, d, e] => unsafe { std::mem::transmute::<_, extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic)>(ptr)(*a, *b, *c, *d, *e) },
521                [a, b, c, d, e, f] => unsafe { std::mem::transmute::<_, extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic)>(ptr)(*a, *b, *c, *d, *e, *f) },
522                [a, b, c, d, e, f, g] => unsafe {
523                    std::mem::transmute::<_, extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic)>(ptr)(*a, *b, *c, *d, *e, *f, *g)
524                },
525                [a, b, c, d, e, f, g, h] => unsafe {
526                    std::mem::transmute::<_, extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic)>(ptr)(
527                        *a, *b, *c, *d, *e, *f, *g, *h,
528                    )
529                },
530                [a, b, c, d, e, f, g, h, i] => unsafe {
531                    std::mem::transmute::<_, extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic)>(ptr)(
532                        *a, *b, *c, *d, *e, *f, *g, *h, *i,
533                    )
534                },
535                [a, b, c, d, e, f, g, h, i, j] => unsafe {
536                    std::mem::transmute::<_, extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic)>(
537                        ptr,
538                    )(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j)
539                },
540                [a, b, c, d, e, f, g, h, i, j, k] => unsafe {
541                    std::mem::transmute::<
542                        _,
543                        extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic),
544                    >(ptr)(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j, *k)
545                },
546                [a, b, c, d, e, f, g, h, i, j, k, l] => unsafe {
547                    std::mem::transmute::<
548                        _,
549                        extern "C" fn(
550                            *const Dynamic,
551                            *const Dynamic,
552                            *const Dynamic,
553                            *const Dynamic,
554                            *const Dynamic,
555                            *const Dynamic,
556                            *const Dynamic,
557                            *const Dynamic,
558                            *const Dynamic,
559                            *const Dynamic,
560                            *const Dynamic,
561                            *const Dynamic,
562                        ),
563                    >(ptr)(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j, *k, *l)
564                },
565                [a, b, c, d, e, f, g, h, i, j, k, l, m] => unsafe {
566                    std::mem::transmute::<
567                        _,
568                        extern "C" fn(
569                            *const Dynamic,
570                            *const Dynamic,
571                            *const Dynamic,
572                            *const Dynamic,
573                            *const Dynamic,
574                            *const Dynamic,
575                            *const Dynamic,
576                            *const Dynamic,
577                            *const Dynamic,
578                            *const Dynamic,
579                            *const Dynamic,
580                            *const Dynamic,
581                            *const Dynamic,
582                        ),
583                    >(ptr)(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j, *k, *l, *m)
584                },
585                [a, b, c, d, e, f, g, h, i, j, k, l, m, n] => unsafe {
586                    std::mem::transmute::<
587                        _,
588                        extern "C" fn(
589                            *const Dynamic,
590                            *const Dynamic,
591                            *const Dynamic,
592                            *const Dynamic,
593                            *const Dynamic,
594                            *const Dynamic,
595                            *const Dynamic,
596                            *const Dynamic,
597                            *const Dynamic,
598                            *const Dynamic,
599                            *const Dynamic,
600                            *const Dynamic,
601                            *const Dynamic,
602                            *const Dynamic,
603                        ),
604                    >(ptr)(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j, *k, *l, *m, *n)
605                },
606                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o] => unsafe {
607                    std::mem::transmute::<
608                        _,
609                        extern "C" fn(
610                            *const Dynamic,
611                            *const Dynamic,
612                            *const Dynamic,
613                            *const Dynamic,
614                            *const Dynamic,
615                            *const Dynamic,
616                            *const Dynamic,
617                            *const Dynamic,
618                            *const Dynamic,
619                            *const Dynamic,
620                            *const Dynamic,
621                            *const Dynamic,
622                            *const Dynamic,
623                            *const Dynamic,
624                            *const Dynamic,
625                        ),
626                    >(ptr)(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j, *k, *l, *m, *n, *o)
627                },
628                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p] => unsafe {
629                    std::mem::transmute::<
630                        _,
631                        extern "C" fn(
632                            *const Dynamic,
633                            *const Dynamic,
634                            *const Dynamic,
635                            *const Dynamic,
636                            *const Dynamic,
637                            *const Dynamic,
638                            *const Dynamic,
639                            *const Dynamic,
640                            *const Dynamic,
641                            *const Dynamic,
642                            *const Dynamic,
643                            *const Dynamic,
644                            *const Dynamic,
645                            *const Dynamic,
646                            *const Dynamic,
647                            *const Dynamic,
648                        ),
649                    >(ptr)(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j, *k, *l, *m, *n, *o, *p)
650                },
651                _ => unreachable!(),
652            }
653        };
654    }
655
656    if ret_ty.is_void() {
657        call_void!();
658        return Ok(());
659    }
660
661    macro_rules! call_ret {
662        ($ret:ty, $drop_result:expr) => {
663            match args {
664                [] => {
665                    let fn_ptr: extern "C" fn() -> $ret = unsafe { std::mem::transmute(ptr) };
666                    $drop_result(fn_ptr());
667                }
668                [a] => {
669                    let fn_ptr: extern "C" fn(*const Dynamic) -> $ret = unsafe { std::mem::transmute(ptr) };
670                    $drop_result(fn_ptr(*a));
671                }
672                [a, b] => {
673                    let fn_ptr: extern "C" fn(*const Dynamic, *const Dynamic) -> $ret = unsafe { std::mem::transmute(ptr) };
674                    $drop_result(fn_ptr(*a, *b));
675                }
676                [a, b, c] => {
677                    let fn_ptr: extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic) -> $ret = unsafe { std::mem::transmute(ptr) };
678                    $drop_result(fn_ptr(*a, *b, *c));
679                }
680                [a, b, c, d] => {
681                    let fn_ptr: extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic) -> $ret = unsafe { std::mem::transmute(ptr) };
682                    $drop_result(fn_ptr(*a, *b, *c, *d));
683                }
684                [a, b, c, d, e] => {
685                    let fn_ptr: extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic) -> $ret = unsafe { std::mem::transmute(ptr) };
686                    $drop_result(fn_ptr(*a, *b, *c, *d, *e));
687                }
688                [a, b, c, d, e, f] => {
689                    let fn_ptr: extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic) -> $ret = unsafe { std::mem::transmute(ptr) };
690                    $drop_result(fn_ptr(*a, *b, *c, *d, *e, *f));
691                }
692                [a, b, c, d, e, f, g] => {
693                    let fn_ptr: extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic) -> $ret = unsafe { std::mem::transmute(ptr) };
694                    $drop_result(fn_ptr(*a, *b, *c, *d, *e, *f, *g));
695                }
696                [a, b, c, d, e, f, g, h] => {
697                    let fn_ptr: extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic) -> $ret = unsafe { std::mem::transmute(ptr) };
698                    $drop_result(fn_ptr(*a, *b, *c, *d, *e, *f, *g, *h));
699                }
700                [a, b, c, d, e, f, g, h, i] => {
701                    let fn_ptr: extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic) -> $ret =
702                        unsafe { std::mem::transmute(ptr) };
703                    $drop_result(fn_ptr(*a, *b, *c, *d, *e, *f, *g, *h, *i));
704                }
705                [a, b, c, d, e, f, g, h, i, j] => {
706                    let fn_ptr: extern "C" fn(*const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic, *const Dynamic) -> $ret =
707                        unsafe { std::mem::transmute(ptr) };
708                    $drop_result(fn_ptr(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j));
709                }
710                [a, b, c, d, e, f, g, h, i, j, k] => {
711                    let fn_ptr: extern "C" fn(
712                        *const Dynamic,
713                        *const Dynamic,
714                        *const Dynamic,
715                        *const Dynamic,
716                        *const Dynamic,
717                        *const Dynamic,
718                        *const Dynamic,
719                        *const Dynamic,
720                        *const Dynamic,
721                        *const Dynamic,
722                        *const Dynamic,
723                    ) -> $ret = unsafe { std::mem::transmute(ptr) };
724                    $drop_result(fn_ptr(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j, *k));
725                }
726                [a, b, c, d, e, f, g, h, i, j, k, l] => {
727                    let fn_ptr: extern "C" fn(
728                        *const Dynamic,
729                        *const Dynamic,
730                        *const Dynamic,
731                        *const Dynamic,
732                        *const Dynamic,
733                        *const Dynamic,
734                        *const Dynamic,
735                        *const Dynamic,
736                        *const Dynamic,
737                        *const Dynamic,
738                        *const Dynamic,
739                        *const Dynamic,
740                    ) -> $ret = unsafe { std::mem::transmute(ptr) };
741                    $drop_result(fn_ptr(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j, *k, *l));
742                }
743                [a, b, c, d, e, f, g, h, i, j, k, l, m] => {
744                    let fn_ptr: extern "C" fn(
745                        *const Dynamic,
746                        *const Dynamic,
747                        *const Dynamic,
748                        *const Dynamic,
749                        *const Dynamic,
750                        *const Dynamic,
751                        *const Dynamic,
752                        *const Dynamic,
753                        *const Dynamic,
754                        *const Dynamic,
755                        *const Dynamic,
756                        *const Dynamic,
757                        *const Dynamic,
758                    ) -> $ret = unsafe { std::mem::transmute(ptr) };
759                    $drop_result(fn_ptr(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j, *k, *l, *m));
760                }
761                [a, b, c, d, e, f, g, h, i, j, k, l, m, n] => {
762                    let fn_ptr: extern "C" fn(
763                        *const Dynamic,
764                        *const Dynamic,
765                        *const Dynamic,
766                        *const Dynamic,
767                        *const Dynamic,
768                        *const Dynamic,
769                        *const Dynamic,
770                        *const Dynamic,
771                        *const Dynamic,
772                        *const Dynamic,
773                        *const Dynamic,
774                        *const Dynamic,
775                        *const Dynamic,
776                        *const Dynamic,
777                    ) -> $ret = unsafe { std::mem::transmute(ptr) };
778                    $drop_result(fn_ptr(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j, *k, *l, *m, *n));
779                }
780                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o] => {
781                    let fn_ptr: extern "C" fn(
782                        *const Dynamic,
783                        *const Dynamic,
784                        *const Dynamic,
785                        *const Dynamic,
786                        *const Dynamic,
787                        *const Dynamic,
788                        *const Dynamic,
789                        *const Dynamic,
790                        *const Dynamic,
791                        *const Dynamic,
792                        *const Dynamic,
793                        *const Dynamic,
794                        *const Dynamic,
795                        *const Dynamic,
796                        *const Dynamic,
797                    ) -> $ret = unsafe { std::mem::transmute(ptr) };
798                    $drop_result(fn_ptr(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j, *k, *l, *m, *n, *o));
799                }
800                [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p] => {
801                    let fn_ptr: extern "C" fn(
802                        *const Dynamic,
803                        *const Dynamic,
804                        *const Dynamic,
805                        *const Dynamic,
806                        *const Dynamic,
807                        *const Dynamic,
808                        *const Dynamic,
809                        *const Dynamic,
810                        *const Dynamic,
811                        *const Dynamic,
812                        *const Dynamic,
813                        *const Dynamic,
814                        *const Dynamic,
815                        *const Dynamic,
816                        *const Dynamic,
817                        *const Dynamic,
818                    ) -> $ret = unsafe { std::mem::transmute(ptr) };
819                    $drop_result(fn_ptr(*a, *b, *c, *d, *e, *f, *g, *h, *i, *j, *k, *l, *m, *n, *o, *p));
820                }
821                _ => unreachable!(),
822            }
823        };
824    }
825
826    if ret_ty.is_bool() {
827        call_ret!(bool, |_| {});
828    } else if ret_ty.is_float() {
829        if ret_ty.is_f64() {
830            call_ret!(f64, |_| {});
831        } else {
832            call_ret!(f32, |_| {});
833        }
834    } else if ret_ty.is_int() || ret_ty.is_uint() {
835        match ret_ty {
836            Type::I8 => call_ret!(i8, |_| {}),
837            Type::U8 => call_ret!(u8, |_| {}),
838            Type::I16 => call_ret!(i16, |_| {}),
839            Type::U16 => call_ret!(u16, |_| {}),
840            Type::I32 => call_ret!(i32, |_| {}),
841            Type::U32 => call_ret!(u32, |_| {}),
842            Type::I64 => call_ret!(i64, |_| {}),
843            Type::U64 => call_ret!(u64, |_| {}),
844            _ => unreachable!(),
845        }
846    } else if ret_ty.is_struct() || ret_ty.is_array() || ret_ty.is_vec() {
847        log::warn!("spawned fn returns {ret_ty:?} — not supported, discarding");
848        call_ret!(*const u8, |_| {});
849    } else {
850        call_ret!(*const Dynamic, |ptr| drop(unsafe { take_dynamic_return(ptr) }));
851    }
852    Ok(())
853}
854
855extern "C" fn any_len(addr: *const Dynamic) -> i64 {
856    if addr.is_null() { 0 } else { unsafe { (&*addr).len() as i64 } }
857}
858
859extern "C" fn any_keys(addr: *const Dynamic) -> *const Dynamic {
860    if addr.is_null() {
861        return alloc_dynamic(Dynamic::list(Vec::new()));
862    }
863    let keys = match unsafe { &*addr } {
864        Dynamic::Map(map) => map.read().unwrap().keys().map(|key| Dynamic::from(key.as_str())).collect(),
865        _ => Vec::new(),
866    };
867    alloc_dynamic(Dynamic::list(keys))
868}
869
870extern "C" fn any_iter(addr: *const Dynamic) -> *const Dynamic {
871    if addr.is_null() { any_null() } else { alloc_dynamic(unsafe { (*addr).clone().into_iter() }) }
872}
873
874extern "C" fn any_next(addr: *const Dynamic) -> *const Dynamic {
875    if addr.is_null() { any_null() } else { alloc_dynamic(unsafe { (&mut *(addr as *mut Dynamic)).next().unwrap_or(Dynamic::Null) }) }
876}
877
878extern "C" fn any_next_pair(addr: *const Dynamic) -> *const Dynamic {
879    if addr.is_null() { any_null() } else { alloc_dynamic(unsafe { (&mut *(addr as *mut Dynamic)).next_pair().unwrap_or(Dynamic::Null) }) }
880}
881
882extern "C" fn any_push(addr: *mut Dynamic, value: *mut Dynamic) {
883    if !addr.is_null() && !value.is_null() {
884        unsafe {
885            (&mut *addr).push_dynamic((&*value).clone());
886        }
887    }
888}
889
890extern "C" fn any_pop(addr: *mut Dynamic) -> *const Dynamic {
891    if addr.is_null() { any_null() } else { alloc_dynamic(unsafe { (*addr).pop().unwrap_or(Dynamic::Null) }) }
892}
893
894extern "C" fn get_key(addr: *const Dynamic, key: *const Dynamic) -> *const Dynamic {
895    if addr.is_null() || key.is_null() {
896        any_null()
897    } else {
898        let key: &str = unsafe { &*key }.as_str();
899        alloc_dynamic(unsafe { (*addr).get_dynamic(key).unwrap_or(Dynamic::Null) })
900    }
901}
902
903extern "C" fn del_key(addr: *const Dynamic, key: *const Dynamic) -> *const Dynamic {
904    if addr.is_null() || key.is_null() {
905        any_null()
906    } else {
907        let key: &str = unsafe { &*key }.as_str();
908        alloc_dynamic(unsafe { (*addr).remove_dynamic(key).unwrap_or(Dynamic::Null) })
909    }
910}
911
912extern "C" fn contains(addr: *const Dynamic, key: *const Dynamic) -> bool {
913    if addr.is_null() || key.is_null() {
914        false
915    } else {
916        let key: &str = unsafe { &*key }.as_str();
917        unsafe { (*addr).contains(key) }
918    }
919}
920
921extern "C" fn starts_with(addr: *const Dynamic, prefix: *const Dynamic) -> bool {
922    if addr.is_null() || prefix.is_null() {
923        false
924    } else {
925        let prefix: &str = unsafe { &*prefix }.as_str();
926        unsafe { (*addr).starts_with(prefix) }
927    }
928}
929
930extern "C" fn get_idx(addr: *const Dynamic, idx: i64) -> *const Dynamic {
931    if addr.is_null() { any_null() } else { alloc_dynamic(unsafe { (*addr).get_idx(idx as usize).unwrap_or(Dynamic::Null) }) }
932}
933
934macro_rules! myvec_list_native {
935    ($push:ident, $get_idx:ident, $set_idx:ident, $vec:ident, $dynamic:ident, $ty:ty, $fallback:expr) => {
936        extern "C" fn $push(addr: *mut Dynamic, value: $ty) {
937            if addr.is_null() {
938                return;
939            }
940            unsafe {
941                match &mut *addr {
942                    Dynamic::$vec(values) => values.push(value),
943                    list => {
944                        list.push_dynamic(Dynamic::$dynamic(value));
945                    }
946                }
947            }
948        }
949
950        extern "C" fn $get_idx(addr: *const Dynamic, idx: i64) -> $ty {
951            if addr.is_null() {
952                return <$ty>::default();
953            }
954            let Ok(idx) = usize::try_from(idx) else {
955                return <$ty>::default();
956            };
957            unsafe {
958                match &*addr {
959                    Dynamic::$vec(values) => values.get(idx).unwrap_or_default(),
960                    values => values.get_idx(idx).and_then($fallback).unwrap_or_default(),
961                }
962            }
963        }
964
965        extern "C" fn $set_idx(addr: *mut Dynamic, idx: i64, value: $ty) {
966            if addr.is_null() {
967                return;
968            }
969            let Ok(idx) = usize::try_from(idx) else {
970                return;
971            };
972            unsafe {
973                match &mut *addr {
974                    Dynamic::$vec(values) => values.set(idx, value),
975                    list => list.set_idx(idx, Dynamic::$dynamic(value)),
976                }
977            }
978        }
979    };
980}
981
982macro_rules! stdvec_list_native {
983    ($push:ident, $get_idx:ident, $set_idx:ident, $vec:ident, $dynamic:ident, $ty:ty, $fallback:expr) => {
984        extern "C" fn $push(addr: *mut Dynamic, value: $ty) {
985            if addr.is_null() {
986                return;
987            }
988            unsafe {
989                match &mut *addr {
990                    Dynamic::$vec(values) => values.push(value),
991                    list => {
992                        list.push_dynamic(Dynamic::$dynamic(value));
993                    }
994                }
995            }
996        }
997
998        extern "C" fn $get_idx(addr: *const Dynamic, idx: i64) -> $ty {
999            if addr.is_null() {
1000                return <$ty>::default();
1001            }
1002            let Ok(idx) = usize::try_from(idx) else {
1003                return <$ty>::default();
1004            };
1005            unsafe {
1006                match &*addr {
1007                    Dynamic::$vec(values) => values.get(idx).copied().unwrap_or_default(),
1008                    values => values.get_idx(idx).and_then($fallback).unwrap_or_default(),
1009                }
1010            }
1011        }
1012
1013        extern "C" fn $set_idx(addr: *mut Dynamic, idx: i64, value: $ty) {
1014            if addr.is_null() {
1015                return;
1016            }
1017            let Ok(idx) = usize::try_from(idx) else {
1018                return;
1019            };
1020            unsafe {
1021                match &mut *addr {
1022                    Dynamic::$vec(values) => {
1023                        if let Some(slot) = values.get_mut(idx) {
1024                            *slot = value;
1025                        }
1026                    }
1027                    list => list.set_idx(idx, Dynamic::$dynamic(value)),
1028                }
1029            }
1030        }
1031    };
1032}
1033
1034myvec_list_native!(list_i8_push, list_i8_get_idx, list_i8_set_idx, VecI8, I8, i8, |value: Dynamic| value.as_int().map(|value| value as i8));
1035myvec_list_native!(list_u16_push, list_u16_get_idx, list_u16_set_idx, VecU16, U16, u16, |value: Dynamic| value.as_uint().map(|value| value as u16));
1036myvec_list_native!(list_i16_push, list_i16_get_idx, list_i16_set_idx, VecI16, I16, i16, |value: Dynamic| value.as_int().map(|value| value as i16));
1037myvec_list_native!(list_u32_push, list_u32_get_idx, list_u32_set_idx, VecU32, U32, u32, |value: Dynamic| value.as_uint().map(|value| value as u32));
1038myvec_list_native!(list_i32_push, list_i32_get_idx, list_i32_set_idx, VecI32, I32, i32, |value: Dynamic| value.as_int().map(|value| value as i32));
1039myvec_list_native!(list_f32_push, list_f32_get_idx, list_f32_set_idx, VecF32, F32, f32, |value: Dynamic| value.as_float().map(|value| value as f32));
1040stdvec_list_native!(list_u64_push, list_u64_get_idx, list_u64_set_idx, VecU64, U64, u64, |value: Dynamic| value.as_uint());
1041stdvec_list_native!(list_i64_push, list_i64_get_idx, list_i64_set_idx, VecI64, I64, i64, |value: Dynamic| value.as_int());
1042stdvec_list_native!(list_f64_push, list_f64_get_idx, list_f64_set_idx, VecF64, F64, f64, |value: Dynamic| value.as_float());
1043
1044extern "C" fn list_bool_push(addr: *mut Dynamic, value: bool) {
1045    if !addr.is_null() {
1046        unsafe {
1047            (&mut *addr).push_dynamic(Dynamic::Bool(value));
1048        }
1049    }
1050}
1051
1052extern "C" fn list_bool_get_idx(addr: *const Dynamic, idx: i64) -> bool {
1053    if addr.is_null() {
1054        return false;
1055    }
1056    let Ok(idx) = usize::try_from(idx) else {
1057        return false;
1058    };
1059    unsafe { (&*addr).get_idx(idx).and_then(|value| value.as_bool()).unwrap_or(false) }
1060}
1061
1062extern "C" fn list_bool_set_idx(addr: *mut Dynamic, idx: i64, value: bool) {
1063    if addr.is_null() {
1064        return;
1065    }
1066    let Ok(idx) = usize::try_from(idx) else {
1067        return;
1068    };
1069    unsafe {
1070        (&mut *addr).set_idx(idx, Dynamic::Bool(value));
1071    }
1072}
1073
1074extern "C" fn list_u8_push(addr: *mut Dynamic, value: u8) {
1075    if !addr.is_null() {
1076        unsafe {
1077            (&mut *addr).push_dynamic(Dynamic::U8(value));
1078        }
1079    }
1080}
1081
1082extern "C" fn list_u8_get_idx(addr: *const Dynamic, idx: i64) -> u8 {
1083    if addr.is_null() {
1084        return 0;
1085    }
1086    let Ok(idx) = usize::try_from(idx) else {
1087        return 0;
1088    };
1089    unsafe { (&*addr).get_idx(idx).and_then(|value| value.as_uint()).map(|value| value as u8).unwrap_or(0) }
1090}
1091
1092extern "C" fn list_u8_set_idx(addr: *mut Dynamic, idx: i64, value: u8) {
1093    if addr.is_null() {
1094        return;
1095    }
1096    let Ok(idx) = usize::try_from(idx) else {
1097        return;
1098    };
1099    unsafe {
1100        (&mut *addr).set_idx(idx, Dynamic::U8(value));
1101    }
1102}
1103
1104extern "C" fn list_str_push(addr: *mut Dynamic, value: *const Dynamic) {
1105    if addr.is_null() || value.is_null() {
1106        return;
1107    }
1108    unsafe {
1109        (&mut *addr).push_dynamic((&*value).clone());
1110    }
1111}
1112
1113extern "C" fn list_str_get_idx(addr: *const Dynamic, idx: i64) -> *const Dynamic {
1114    if addr.is_null() {
1115        return any_null();
1116    };
1117    let Ok(idx) = usize::try_from(idx) else {
1118        return any_null();
1119    };
1120    if let Some(value) = unsafe { (&*addr).get_idx(idx) } { alloc_dynamic(value) } else { any_null() }
1121}
1122
1123extern "C" fn list_str_set_idx(addr: *mut Dynamic, idx: i64, value: *const Dynamic) {
1124    if addr.is_null() || value.is_null() {
1125        return;
1126    }
1127    let Ok(idx) = usize::try_from(idx) else {
1128        return;
1129    };
1130    unsafe {
1131        (&mut *addr).set_idx(idx, (&*value).clone());
1132    }
1133}
1134
1135extern "C" fn slice(addr: *const Dynamic, start: i64, stop: *const Dynamic, inclusive: bool) -> *const Dynamic {
1136    if addr.is_null() {
1137        return any_null();
1138    }
1139
1140    let value = unsafe { &*addr };
1141    let len = value.len() as i64;
1142    let start = start.clamp(0, len) as usize;
1143    let mut stop = if stop.is_null() {
1144        len
1145    } else {
1146        let raw = unsafe { &*stop };
1147        if raw.is_null() { len } else { raw.as_int().unwrap_or(len) }
1148    };
1149    if inclusive && stop < len {
1150        stop += 1;
1151    }
1152    let stop = stop.clamp(start as i64, len) as usize;
1153
1154    let sliced = if value.is_str() {
1155        Dynamic::from(value.as_str().chars().skip(start).take(stop.saturating_sub(start)).collect::<String>())
1156    } else {
1157        match value {
1158            Dynamic::List(list) => Dynamic::list(list.read().unwrap()[start..stop].to_vec()),
1159            _ => Dynamic::Null,
1160        }
1161    };
1162    alloc_dynamic(sliced)
1163}
1164
1165extern "C" fn set_key(addr: *mut Dynamic, key: *const Dynamic, value: *const Dynamic) {
1166    if addr.is_null() || key.is_null() {
1167        return;
1168    }
1169    let key: &str = unsafe { &*key }.as_str();
1170    unsafe { (&mut *addr).set_dynamic(key.into(), (&*value).clone()) }
1171}
1172
1173extern "C" fn set_idx(addr: *mut Dynamic, idx: i64, value: *const Dynamic) {
1174    if addr.is_null() {
1175        return;
1176    }
1177    unsafe { (&mut *addr).set_idx(idx as usize, (&*value).clone()) }
1178}
1179
1180extern "C" fn any_from_i64(v: i64) -> *const Dynamic {
1181    alloc_dynamic(Dynamic::I64(v))
1182}
1183
1184extern "C" fn any_from_u64(v: u64) -> *const Dynamic {
1185    alloc_dynamic(Dynamic::U64(v))
1186}
1187
1188extern "C" fn any_from_bool(v: bool) -> *const Dynamic {
1189    alloc_dynamic(Dynamic::Bool(v))
1190}
1191
1192extern "C" fn any_to_i64(addr: *const Dynamic) -> i64 {
1193    if addr.is_null() {
1194        return 0;
1195    }
1196    unsafe {
1197        let value = &*addr;
1198        value
1199            .as_int()
1200            .or_else(|| value.as_float().map(|value| value as i64))
1201            .or_else(|| {
1202                let text = value.as_str();
1203                let text = text.trim();
1204                if text.is_empty() { None } else { text.parse::<i64>().ok().or_else(|| text.parse::<f64>().ok().map(|value| value as i64)) }
1205            })
1206            .unwrap_or(0)
1207    }
1208}
1209
1210extern "C" fn any_to_bool(addr: *const Dynamic) -> bool {
1211    if addr.is_null() {
1212        return false;
1213    }
1214    unsafe {
1215        let value = &*addr;
1216        if let Some(v) = value.as_bool() {
1217            v
1218        } else if let Some(v) = value.as_int() {
1219            v != 0
1220        } else if let Some(v) = value.as_float() {
1221            v != 0.0
1222        } else {
1223            !value.is_null()
1224        }
1225    }
1226}
1227
1228extern "C" fn any_from_f64(v: f64) -> *const Dynamic {
1229    alloc_dynamic(Dynamic::F64(v))
1230}
1231
1232extern "C" fn any_split(addr: *mut Dynamic, s: *const Dynamic) -> *const Dynamic {
1233    if addr.is_null() || s.is_null() {
1234        return any_null();
1235    }
1236    let s: &str = unsafe { &*s }.as_str();
1237    alloc_dynamic(unsafe { (&*addr).clone() }.split(s))
1238}
1239
1240extern "C" fn any_to_f64(addr: *const Dynamic) -> f64 {
1241    if addr.is_null() {
1242        return 0.0;
1243    }
1244    unsafe {
1245        let value = &*addr;
1246        value.as_float().or_else(|| value.as_str().trim().parse::<f64>().ok()).unwrap_or(0.0)
1247    }
1248}
1249
1250extern "C" fn any_to_string(addr: *const Dynamic) -> *const Dynamic {
1251    if addr.is_null() {
1252        return alloc_dynamic(Dynamic::from(""));
1253    }
1254    alloc_dynamic(Dynamic::from(unsafe { &*addr }.to_string()))
1255}
1256
1257extern "C" fn any_binary(left: *const Dynamic, op: i32, right: *const Dynamic) -> *const Dynamic {
1258    if left.is_null() {
1259        if right.is_null() {
1260            return any_null();
1261        }
1262        return alloc_dynamic(unsafe { (&*right).clone() });
1263    }
1264    if right.is_null() {
1265        return alloc_dynamic(unsafe { (&*left).clone() });
1266    }
1267    let op = BinaryOp::try_from(op).unwrap_or(BinaryOp::Unknow);
1268    if op == BinaryOp::Add {
1269        let (left_value, right_value) = unsafe { (&*left, &*right) };
1270        if left_value.is_str() || right_value.is_str() {
1271            return alloc_dynamic(left_value.clone() + right_value.clone());
1272        }
1273    }
1274    unsafe {
1275        let expr = Expr::new(
1276            ExprKind::Binary { left: Box::new(Expr::new(ExprKind::Value((&*left).clone()), Span::default())), op, right: Box::new(Expr::new(ExprKind::Value((&*right).clone()), Span::default())) },
1277            Span::default(),
1278        );
1279        alloc_dynamic(expr.compact().unwrap_or(Dynamic::Null))
1280    }
1281}
1282
1283extern "C" fn any_logic(left: *const Dynamic, op: i32, right: *const Dynamic) -> bool {
1284    let op = BinaryOp::try_from(op).unwrap_or(BinaryOp::Unknow);
1285    unsafe {
1286        let expr = Expr::new(
1287            ExprKind::Binary { left: Box::new(Expr::new(ExprKind::Value((&*left).clone()), Span::default())), op, right: Box::new(Expr::new(ExprKind::Value((&*right).clone()), Span::default())) },
1288            Span::default(),
1289        );
1290        expr.compact().and_then(|r| r.as_bool()).unwrap_or(false)
1291    }
1292}
1293
1294pub const STD: [(&str, &[Type], Type, *const u8); 5] = [
1295    ("print", &[Type::Any], Type::Void, print as *const u8),
1296    ("sqrt", &[Type::F64], Type::F64, sqrt as *const u8),
1297    ("log", &[Type::Any], Type::Void, log_any as *const u8),
1298    ("uuid", &[], Type::Any, uuid as *const u8),
1299    ("rand", &[Type::Any, Type::Any], Type::Any, random as *const u8),
1300];
1301
1302pub const ANY: [(&str, &[Type], Type, *const u8); 68] = [
1303    ("Any::null", &[], Type::Any, any_null as *const u8),
1304    ("Any::is_map", &[Type::Any], Type::Bool, any_is_map as *const u8),
1305    ("Any::is_list", &[Type::Any], Type::Bool, any_is_list as *const u8),
1306    ("Any::is_string", &[Type::Any], Type::Bool, any_is_string as *const u8),
1307    ("Any::is_null", &[Type::Any], Type::Bool, any_is_null as *const u8),
1308    ("Any::clone", &[Type::Any], Type::Any, any_clone as *const u8),
1309    ("Any::len", &[Type::Any], Type::I32, any_len as *const u8),
1310    ("Any::keys", &[Type::Any], Type::Any, any_keys as *const u8),
1311    ("Any::split", &[Type::Any, Type::Any], Type::Any, any_split as *const u8),
1312    ("Any::push", &[Type::Any, Type::Any], Type::Void, any_push as *const u8),
1313    ("Any::pop", &[Type::Any], Type::Any, any_pop as *const u8),
1314    ("Any::get_idx", &[Type::Any, Type::I64], Type::Any, get_idx as *const u8),
1315    ("Any::push_bool", &[Type::Any, Type::Bool], Type::Void, list_bool_push as *const u8),
1316    ("Any::get_idx_bool", &[Type::Any, Type::I64], Type::Bool, list_bool_get_idx as *const u8),
1317    ("Any::set_idx_bool", &[Type::Any, Type::I64, Type::Bool], Type::Void, list_bool_set_idx as *const u8),
1318    ("Any::push_u8", &[Type::Any, Type::U8], Type::Void, list_u8_push as *const u8),
1319    ("Any::get_idx_u8", &[Type::Any, Type::I64], Type::U8, list_u8_get_idx as *const u8),
1320    ("Any::set_idx_u8", &[Type::Any, Type::I64, Type::U8], Type::Void, list_u8_set_idx as *const u8),
1321    ("Any::push_i8", &[Type::Any, Type::I8], Type::Void, list_i8_push as *const u8),
1322    ("Any::get_idx_i8", &[Type::Any, Type::I64], Type::I8, list_i8_get_idx as *const u8),
1323    ("Any::set_idx_i8", &[Type::Any, Type::I64, Type::I8], Type::Void, list_i8_set_idx as *const u8),
1324    ("Any::push_u16", &[Type::Any, Type::U16], Type::Void, list_u16_push as *const u8),
1325    ("Any::get_idx_u16", &[Type::Any, Type::I64], Type::U16, list_u16_get_idx as *const u8),
1326    ("Any::set_idx_u16", &[Type::Any, Type::I64, Type::U16], Type::Void, list_u16_set_idx as *const u8),
1327    ("Any::push_i16", &[Type::Any, Type::I16], Type::Void, list_i16_push as *const u8),
1328    ("Any::get_idx_i16", &[Type::Any, Type::I64], Type::I16, list_i16_get_idx as *const u8),
1329    ("Any::set_idx_i16", &[Type::Any, Type::I64, Type::I16], Type::Void, list_i16_set_idx as *const u8),
1330    ("Any::push_u32", &[Type::Any, Type::U32], Type::Void, list_u32_push as *const u8),
1331    ("Any::get_idx_u32", &[Type::Any, Type::I64], Type::U32, list_u32_get_idx as *const u8),
1332    ("Any::set_idx_u32", &[Type::Any, Type::I64, Type::U32], Type::Void, list_u32_set_idx as *const u8),
1333    ("Any::push_i32", &[Type::Any, Type::I32], Type::Void, list_i32_push as *const u8),
1334    ("Any::get_idx_i32", &[Type::Any, Type::I64], Type::I32, list_i32_get_idx as *const u8),
1335    ("Any::set_idx_i32", &[Type::Any, Type::I64, Type::I32], Type::Void, list_i32_set_idx as *const u8),
1336    ("Any::push_f32", &[Type::Any, Type::F32], Type::Void, list_f32_push as *const u8),
1337    ("Any::get_idx_f32", &[Type::Any, Type::I64], Type::F32, list_f32_get_idx as *const u8),
1338    ("Any::set_idx_f32", &[Type::Any, Type::I64, Type::F32], Type::Void, list_f32_set_idx as *const u8),
1339    ("Any::push_u64", &[Type::Any, Type::U64], Type::Void, list_u64_push as *const u8),
1340    ("Any::get_idx_u64", &[Type::Any, Type::I64], Type::U64, list_u64_get_idx as *const u8),
1341    ("Any::set_idx_u64", &[Type::Any, Type::I64, Type::U64], Type::Void, list_u64_set_idx as *const u8),
1342    ("Any::push_i64", &[Type::Any, Type::I64], Type::Void, list_i64_push as *const u8),
1343    ("Any::get_idx_i64", &[Type::Any, Type::I64], Type::I64, list_i64_get_idx as *const u8),
1344    ("Any::set_idx_i64", &[Type::Any, Type::I64, Type::I64], Type::Void, list_i64_set_idx as *const u8),
1345    ("Any::push_f64", &[Type::Any, Type::F64], Type::Void, list_f64_push as *const u8),
1346    ("Any::get_idx_f64", &[Type::Any, Type::I64], Type::F64, list_f64_get_idx as *const u8),
1347    ("Any::set_idx_f64", &[Type::Any, Type::I64, Type::F64], Type::Void, list_f64_set_idx as *const u8),
1348    ("Any::push_str", &[Type::Any, Type::Str], Type::Void, list_str_push as *const u8),
1349    ("Any::get_idx_str", &[Type::Any, Type::I64], Type::Str, list_str_get_idx as *const u8),
1350    ("Any::set_idx_str", &[Type::Any, Type::I64, Type::Str], Type::Void, list_str_set_idx as *const u8),
1351    ("Any::slice", &[Type::Any, Type::I64, Type::Any, Type::Bool], Type::Any, slice as *const u8),
1352    ("Any::contains", &[Type::Any, Type::Any], Type::Bool, contains as *const u8),
1353    ("Any::starts_with", &[Type::Any, Type::Any], Type::Bool, starts_with as *const u8),
1354    ("Any::get_key", &[Type::Any, Type::Any], Type::Any, get_key as *const u8),
1355    ("Any::del_key", &[Type::Any, Type::Any], Type::Any, del_key as *const u8),
1356    ("Any::set_idx", &[Type::Any, Type::I64, Type::Any], Type::Void, set_idx as *const u8),
1357    ("Any::set_key", &[Type::Any, Type::Any, Type::Any], Type::Void, set_key as *const u8),
1358    ("Any::from_i64", &[Type::I64], Type::Any, any_from_i64 as *const u8),
1359    ("Any::from_u64", &[Type::U64], Type::Any, any_from_u64 as *const u8),
1360    ("Any::from_bool", &[Type::Bool], Type::Any, any_from_bool as *const u8),
1361    ("Any::to_i64", &[Type::Any], Type::I64, any_to_i64 as *const u8),
1362    ("Any::to_bool", &[Type::Any], Type::Bool, any_to_bool as *const u8),
1363    ("Any::from_f64", &[Type::F64], Type::Any, any_from_f64 as *const u8),
1364    ("Any::to_f64", &[Type::Any], Type::F64, any_to_f64 as *const u8),
1365    ("Any::to_string", &[Type::Any], Type::Str, any_to_string as *const u8),
1366    ("Any::binary", &[Type::Any, Type::I32, Type::Any], Type::Any, any_binary as *const u8),
1367    ("Any::logic", &[Type::Any, Type::I32, Type::Any], Type::Bool, any_logic as *const u8),
1368    ("Any::iter", &[Type::Any], Type::Any, any_iter as *const u8),
1369    ("Any::next", &[Type::Any], Type::Any, any_next as *const u8),
1370    ("Any::next_pair", &[Type::Any], Type::Any, any_next_pair as *const u8),
1371];
1372
1373use std::rc::Rc;
1374impl JITRunTime {
1375    pub fn add_native_ptr(&mut self, full_name: &str, name: &str, arg_tys: &[Type], ret_ty: Type, fn_ptr: *const u8) -> Result<u32> {
1376        self.native_symbols.write().unwrap().insert(full_name.to_string(), fn_ptr as usize);
1377        self.add_native(full_name, name, arg_tys, ret_ty)
1378    }
1379
1380    pub(crate) fn add_context_native_ptr(&mut self, full_name: &str, name: &str, arg_tys: &[Type], ret_ty: Type, fn_ptr: *const u8) -> Result<u32> {
1381        self.native_symbols.write().unwrap().insert(full_name.to_string(), fn_ptr as usize);
1382        self.add_context_native(full_name, name, arg_tys, ret_ty)
1383    }
1384
1385    pub fn add_native(&mut self, full_name: &str, name: &str, arg_tys: &[Type], ret_ty: Type) -> Result<u32> {
1386        let fn_ty = Type::Fn { tys: arg_tys.to_vec(), ret: Rc::new(ret_ty.clone()) };
1387        let id = self.compiler.add_symbol(name, compiler::Symbol::Native(fn_ty.clone()));
1388        let sig = self.get_sig(arg_tys, ret_ty)?;
1389        let fn_id = self.module.declare_function(full_name, Linkage::Import, &sig)?;
1390        self.fns.insert(id, FnVariant::Native { ty: fn_ty, fn_id, context: None });
1391        Ok(id)
1392    }
1393
1394    pub(crate) fn add_context_native(&mut self, full_name: &str, name: &str, arg_tys: &[Type], ret_ty: Type) -> Result<u32> {
1395        let fn_ty = Type::Fn { tys: arg_tys.to_vec(), ret: Rc::new(ret_ty.clone()) };
1396        let id = self.compiler.add_symbol(name, compiler::Symbol::Native(fn_ty.clone()));
1397        let mut sig = self.module.make_signature();
1398        sig.params.push(AbiParam::new(crate::ptr_type()));
1399        for arg in arg_tys.iter() {
1400            sig.params.push(AbiParam::new(crate::get_type(arg)?));
1401        }
1402        if !ret_ty.is_void() {
1403            sig.returns.push(AbiParam::new(crate::get_type(&ret_ty)?));
1404        }
1405        let fn_id = self.module.declare_function(full_name, Linkage::Import, &sig)?;
1406        self.fns.insert(id, FnVariant::Native { ty: fn_ty, fn_id, context: Some(self.owner_context_ptr()) });
1407        Ok(id)
1408    }
1409}