forked from AbleOS/holey-bytes
86 lines
2.6 KiB
Rust
86 lines
2.6 KiB
Rust
|
use rhai::{CustomType, Engine, ImmutableString};
|
||
|
|
|
||
|
|
use {
|
||
|
|
crate::{object::SymbolRef, SharedObject},
|
||
|
|
rhai::Module,
|
||
|
|
};
|
||
|
|
|
||
|
|
macro_rules! gen_data_insertions {
|
||
|
|
($module:expr, $obj:expr, [$($ty:ident),* $(,)?] $(,)?) => {{
|
||
|
|
let (module, obj) = ($module, $obj);
|
||
|
|
$({
|
||
|
|
let obj = ::std::rc::Rc::clone(obj);
|
||
|
|
let hash = module.set_native_fn(stringify!($ty), move |arr: ::rhai::Array| {
|
||
|
|
let obj = &mut *obj.borrow_mut();
|
||
|
|
let symbol = obj.symbol($crate::object::Section::Data);
|
||
|
|
|
||
|
|
obj.sections
|
||
|
|
.data
|
||
|
|
.reserve(arr.len() * ::std::mem::size_of::<$ty>());
|
||
|
|
|
||
|
|
for item in arr {
|
||
|
|
obj.sections.data.extend(
|
||
|
|
match item.as_int() {
|
||
|
|
Ok(num) => $ty::try_from(num).map_err(|_| "i64".to_owned()),
|
||
|
|
Err(ty) => Err(ty.to_owned()),
|
||
|
|
}
|
||
|
|
.map_err(|err| {
|
||
|
|
::rhai::EvalAltResult::ErrorMismatchDataType(
|
||
|
|
stringify!($ty).to_owned(),
|
||
|
|
err,
|
||
|
|
::rhai::Position::NONE,
|
||
|
|
)
|
||
|
|
})?
|
||
|
|
.to_le_bytes(),
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok(DataRef {
|
||
|
|
symbol,
|
||
|
|
len: obj.sections.data.len() - symbol.0,
|
||
|
|
})
|
||
|
|
});
|
||
|
|
|
||
|
|
module.update_fn_namespace(hash, ::rhai::FnNamespace::Global);
|
||
|
|
})*
|
||
|
|
}};
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Clone, Copy, Debug)]
|
||
|
|
pub struct DataRef {
|
||
|
|
pub symbol: SymbolRef,
|
||
|
|
pub len: usize,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl CustomType for DataRef {
|
||
|
|
fn build(mut builder: rhai::TypeBuilder<Self>) {
|
||
|
|
builder
|
||
|
|
.with_name("DataRef")
|
||
|
|
.with_get("symbol", |this: &mut Self| this.symbol)
|
||
|
|
.with_get("len", |this: &mut Self| this.len as u64 as i64);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn module(engine: &mut Engine, obj: SharedObject) -> Module {
|
||
|
|
let mut module = Module::new();
|
||
|
|
gen_data_insertions!(&mut module, &obj, [i8, i16, i32, i64]);
|
||
|
|
|
||
|
|
{
|
||
|
|
let hash = module.set_native_fn("str", move |s: ImmutableString| {
|
||
|
|
let obj = &mut *obj.borrow_mut();
|
||
|
|
let symbol = obj.symbol(crate::object::Section::Data);
|
||
|
|
|
||
|
|
obj.sections.data.extend(s.as_bytes());
|
||
|
|
Ok(DataRef {
|
||
|
|
symbol,
|
||
|
|
len: s.len(),
|
||
|
|
})
|
||
|
|
});
|
||
|
|
|
||
|
|
module.update_fn_namespace(hash, rhai::FnNamespace::Global);
|
||
|
|
}
|
||
|
|
|
||
|
|
engine.build_type::<DataRef>();
|
||
|
|
module
|
||
|
|
}
|