lune-org/lune
1
1
Fork 0
mirror of https://github.com/lune-org/lune.git synced 2025-04-03 01:50:55 +01:00
Code Issues Projects Releases Packages Wiki Activity

Implememt CVoid, StringifyData and CopyData (#243)

Browse source
This commit is contained in:
qwreey 2024-10-18 06:12:53 +00:00
parent 7ee757ac9c
commit 27e250daa5
No known key found for this signature in database
GPG key ID: D28DB79297A214BD
26 changed files with 549 additions and 105 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
crates/lune-std-ffi/README.md
View file

@ -90,3 +90,9 @@ Add `CTypeInfo:add(target, from1, from2, ...)` and `:sub` `:mul` `:div` `:mod` `
Add bor band and such bit-related operation
> Luau only supports 32bit bit operations
wchar and wstring support
string(null ending) / buffer support
void support

18
crates/lune-std-ffi/src/c/arr_info.rs
View file

@ -59,7 +59,7 @@ impl CArrInfo {
self.length
}
pub fn get_type(&self) -> Type {
pub fn get_middle_type(&self) -> Type {
self.struct_type.clone()
}
@ -87,7 +87,6 @@ impl FfiSize for CArrInfo {
}
}
impl FfiConvert for CArrInfo {
// FIXME: FfiBox, FfiRef support required
unsafe fn value_into_data<'lua>(
&self,
lua: &'lua Lua,
@ -132,6 +131,20 @@ impl FfiConvert for CArrInfo {
}
Ok(LuaValue::Table(table))
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
dst.get_pointer()
.byte_offset(dst_offset)
.copy_from(src.get_pointer().byte_offset(src_offset), self.get_size());
Ok(())
}
}
impl LuaUserData for CArrInfo {
@ -157,6 +170,7 @@ impl LuaUserData for CArrInfo {
method_provider::provide_box(methods);
method_provider::provide_read_data(methods);
method_provider::provide_write_data(methods);
method_provider::provide_copy_data(methods);
methods.add_method("offset", |_, this, offset: isize| {
if this.length > (offset as usize) && offset >= 0 {

121
crates/lune-std-ffi/src/c/helper.rs
View file

@ -2,7 +2,7 @@ use libffi::middle::Type;
use lune_utils::fmt::{pretty_format_value, ValueFormatConfig};
use mlua::prelude::*;
use super::{ctype_helper, CArrInfo, CFnInfo, CPtrInfo, CStructInfo};
use super::{ctype_helper, void_info::CVoidInfo, CArrInfo, CFnInfo, CPtrInfo, CStructInfo};
use crate::{
data::{BoxData, GetFfiData},
ffi::{FfiConvert, FfiSize},
@ -84,17 +84,104 @@ pub mod method_provider {
);
}
pub fn provide_copy_data<'lua, Target, M>(methods: &mut M)
where
Target: FfiSize + FfiConvert,
M: LuaUserDataMethods<'lua, Target>,
{
methods.add_method(
"copyData",
|lua,
this,
(dst, src, dst_offset, src_offset): (
LuaAnyUserData,
LuaAnyUserData,
Option<isize>,
Option<isize>,
)| {
let dst_offset = dst_offset.unwrap_or(0);
let src_offset = src_offset.unwrap_or(0);
let dst = &dst.get_ffi_data()?;
// use or functions
if !dst.check_boundary(dst_offset, this.get_size()) {
return Err(LuaError::external("Out of bounds"));
}
if !dst.is_writable() {
return Err(LuaError::external("Unwritable data handle"));
}
let src = &src.get_ffi_data()?;
if !src.check_boundary(dst_offset, this.get_size()) {
return Err(LuaError::external("Out of bounds"));
}
if !src.is_readable() {
return Err(LuaError::external("Unreadable value data handle"));
}
unsafe { this.copy_data(lua, dst_offset, src_offset, dst, src) }
},
);
}
pub fn provide_stringify_data<'lua, Target, M>(methods: &mut M)
where
Target: FfiSize + FfiConvert,
M: LuaUserDataMethods<'lua, Target>,
{
methods.add_method(
"stringifyData",
|lua, this, (target, offset): (LuaAnyUserData, Option<isize>)| unsafe {
this.stringify_data(lua, offset.unwrap_or(0), &target.get_ffi_data()?)
},
);
}
pub fn provide_box<'lua, Target, M>(methods: &mut M)
where
Target: FfiSize + FfiConvert,
M: LuaUserDataMethods<'lua, Target>,
{
methods.add_method("box", |lua, this, table: LuaValue| {
methods.add_method("box", |lua, this, value: LuaValue| {
let result = lua.create_userdata(BoxData::new(this.get_size()))?;
unsafe { this.value_into_data(lua, 0, &result.get_ffi_data()?, table)? };
unsafe { this.value_into_data(lua, 0, &result.get_ffi_data()?, value)? };
Ok(result)
});
}
// FIXME: Buffer support should be part of another PR
// pub fn provide_write_buffer<'lua, Target, M>(methods: &mut M)
// where
// Target: FfiSize + FfiConvert,
// M: LuaUserDataMethods<'lua, Target>,
// {
// methods.add_method(
// "writeBuffer",
// |lua, this, (target, value, offset): (LuaValue, LuaValue, Option<isize>)| {
// if !target.is_buffer() {
// return Err(LuaError::external(format!(
// "Argument target must be a buffer, got {}",
// target.type_name()
// )));
// }
// target.to_pointer()
// target.as_userdata().unwrap().to_pointer()
// let offset = offset.unwrap_or(0);
// let data_handle = &target.get_ffi_data()?;
// // use or functions
// if !data_handle.check_boundary(offset, this.get_size()) {
// return Err(LuaError::external("Out of bounds"));
// }
// if !data_handle.is_writable() {
// return Err(LuaError::external("Unwritable data handle"));
// }
// unsafe { this.value_into_data(lua, offset, data_handle, value) }
// },
// );
// }
}
pub fn get_userdata(value: LuaValue) -> LuaResult<LuaAnyUserData> {
@ -143,28 +230,32 @@ pub unsafe fn get_conv_list(table: &LuaTable) -> LuaResult<Vec<*const dyn FfiCon
create_list(table, |userdata| get_conv(userdata))
}
pub fn get_size(this: &LuaAnyUserData) -> LuaResult<usize> {
if this.is::<CStructInfo>() {
Ok(this.borrow::<CStructInfo>()?.get_size())
} else if this.is::<CArrInfo>() {
Ok(this.borrow::<CArrInfo>()?.get_size())
} else if this.is::<CPtrInfo>() {
Ok(this.borrow::<CPtrInfo>()?.get_size())
pub fn get_size(userdata: &LuaAnyUserData) -> LuaResult<usize> {
if userdata.is::<CStructInfo>() {
Ok(userdata.borrow::<CStructInfo>()?.get_size())
} else if userdata.is::<CArrInfo>() {
Ok(userdata.borrow::<CArrInfo>()?.get_size())
} else if userdata.is::<CPtrInfo>() {
Ok(userdata.borrow::<CPtrInfo>()?.get_size())
} else if userdata.is::<CVoidInfo>() {
Ok(userdata.borrow::<CVoidInfo>()?.get_size())
} else {
ctype_helper::get_size(this)
ctype_helper::get_size(userdata)
}
}
// get libffi_type from any c-type userdata
pub fn get_middle_type(userdata: &LuaAnyUserData) -> LuaResult<Type> {
if userdata.is::<CStructInfo>() {
Ok(userdata.borrow::<CStructInfo>()?.get_type())
Ok(userdata.borrow::<CStructInfo>()?.get_middle_type())
} else if let Some(middle_type) = ctype_helper::get_middle_type(userdata)? {
Ok(middle_type)
} else if userdata.is::<CArrInfo>() {
Ok(userdata.borrow::<CArrInfo>()?.get_type())
Ok(userdata.borrow::<CArrInfo>()?.get_middle_type())
} else if userdata.is::<CPtrInfo>() {
Ok(CPtrInfo::get_type())
Ok(CPtrInfo::get_middle_type())
} else if userdata.is::<CVoidInfo>() {
Ok(CVoidInfo::get_middle_type())
} else {
Err(LuaError::external(format!(
"Unexpected field. CStruct, CType, CString or CArr is required for element but got {}",
@ -210,6 +301,8 @@ pub fn get_tag_name(userdata: &LuaAnyUserData) -> LuaResult<String> {
String::from("CPtr")
} else if userdata.is::<CFnInfo>() {
String::from("CFn")
} else if userdata.is::<CVoidInfo>() {
String::from("CVoid")
} else if ctype_helper::is_ctype(userdata) {
String::from("CType")
} else {

1
crates/lune-std-ffi/src/c/mod.rs
View file

@ -5,6 +5,7 @@ mod ptr_info;
mod struct_info;
mod type_info;
mod types;
mod void_info;
pub use self::{
arr_info::CArrInfo,

30
crates/lune-std-ffi/src/c/ptr_info.rs
View file

@ -45,21 +45,10 @@ impl FfiConvert for CPtrInfo {
let value_userdata = value
.as_userdata()
.ok_or_else(|| LuaError::external("CPtrInfo:writeRef only allows data"))?;
data_handle
.check_boundary(offset, self.get_size())
.then_some(())
.ok_or_else(|| LuaError::external("Out of bounds"))?;
data_handle
.is_writable()
.then_some(())
.ok_or_else(|| LuaError::external("Unwritable data handle"))?;
*data_handle
.get_pointer()
.byte_offset(offset)
.cast::<*mut ()>() = value_userdata.get_ffi_data()?.get_pointer();
Ok(())
}
@ -70,10 +59,6 @@ impl FfiConvert for CPtrInfo {
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<LuaValue<'lua>> {
if !data_handle.check_boundary(offset, SIZE_OF_POINTER) {
return Err(LuaError::external("Out of bounds"));
}
Ok(LuaValue::UserData(lua.create_userdata(RefData::new(
unsafe { data_handle.get_pointer().byte_offset(offset) },
if self.inner_is_cptr {
@ -84,6 +69,19 @@ impl FfiConvert for CPtrInfo {
RefBounds::new(0, self.inner_size),
))?))
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<*mut ()>() =
src.get_pointer().byte_offset(src_offset);
Ok(())
}
}
impl CPtrInfo {
@ -118,7 +116,7 @@ impl CPtrInfo {
}
// Return void*
pub fn get_type() -> Type {
pub fn get_middle_type() -> Type {
Type::pointer()
}
}

102
crates/lune-std-ffi/src/c/struct_info.rs
View file

@ -1,11 +1,11 @@
use std::{cell::Ref, vec::Vec};
use libffi::{low, middle::Type, raw};
use libffi::{low, middle::Type, raw::ffi_get_struct_offsets};
use mlua::prelude::*;
use super::{association_names::CSTRUCT_INNER, helper, method_provider};
use crate::ffi::{
association, libffi_helper::FFI_STATUS_NAMES, FfiConvert, FfiData, FfiSignedness, FfiSize,
association, libffi_helper::ffi_status_assert, FfiConvert, FfiData, FfiSignedness, FfiSize,
};
pub struct CStructInfo {
@ -15,6 +15,21 @@ pub struct CStructInfo {
inner_conv_list: Vec<*const dyn FfiConvert>,
}
fn get_field_table<'lua>(
lua: &'lua Lua,
userdata: &LuaAnyUserData<'lua>,
) -> LuaResult<LuaTable<'lua>> {
let value = association::get(lua, CSTRUCT_INNER, userdata)?
.ok_or_else(|| LuaError::external("Failed to get inner field table. not found"))?;
if let LuaValue::Table(table) = value {
Ok(table)
} else {
Err(LuaError::external(
"Failed to get inner field table. not a table",
))
}
}
impl CStructInfo {
pub fn new(fields: Vec<Type>, inner_conv_list: Vec<*const dyn FfiConvert>) -> LuaResult<Self> {
let len = fields.len();
@ -23,17 +38,11 @@ impl CStructInfo {
// Get field offsets with ffi_get_struct_offsets
unsafe {
let offset_result: raw::ffi_status = raw::ffi_get_struct_offsets(
ffi_status_assert(ffi_get_struct_offsets(
low::ffi_abi_FFI_DEFAULT_ABI,
middle_type.as_raw_ptr(),
inner_offset_list.as_mut_ptr(),
);
if offset_result != raw::ffi_status_FFI_OK {
return Err(LuaError::external(format!(
"ffi_get_struct_offsets failed. expected result {}, got {}",
FFI_STATUS_NAMES[0], FFI_STATUS_NAMES[offset_result as usize]
)));
}
))?;
inner_offset_list.set_len(len);
}
@ -50,7 +59,7 @@ impl CStructInfo {
})
}
// Create new CStruct UserData with LuaTable.
// Create new CStruct UserData from LuaTable.
// Lock and hold table for .inner ref
pub fn from_table<'lua>(
lua: &'lua Lua,
@ -60,7 +69,6 @@ impl CStructInfo {
.create_userdata(Self::new(helper::get_middle_type_list(&table)?, unsafe {
helper::get_conv_list(&table)?
})?)?;
table.set_readonly(true);
association::set(lua, CSTRUCT_INNER, &cstruct, table)?;
Ok(cstruct)
@ -69,24 +77,20 @@ impl CStructInfo {
// Stringify cstruct for pretty printing like:
// <CStruct( u8, i32, size = 8 )>
pub fn stringify(lua: &Lua, userdata: &LuaAnyUserData) -> LuaResult<String> {
if let LuaValue::Table(fields) = association::get(lua, CSTRUCT_INNER, userdata)?
.ok_or_else(|| LuaError::external("Field table not found"))?
{
let mut result = String::from(" ");
for i in 0..fields.raw_len() {
let child: LuaAnyUserData = fields.raw_get(i + 1)?;
let pretty_formatted = helper::pretty_format(lua, &child)?;
result.push_str(format!("{pretty_formatted}, ").as_str());
}
let fields = get_field_table(lua, userdata)?;
let mut stringified = String::from(" ");
// size of
result.push_str(
format!("size = {} ", userdata.borrow::<CStructInfo>()?.get_size()).as_str(),
);
Ok(result)
} else {
Err(LuaError::external("failed to get inner type table."))
// children
for i in 0..fields.raw_len() {
let child: LuaAnyUserData = fields.raw_get(i + 1)?;
let pretty_formatted = helper::pretty_format(lua, &child)?;
stringified.push_str(format!("{pretty_formatted}, ").as_str());
}
// size of
stringified
.push_str(format!("size = {} ", userdata.borrow::<CStructInfo>()?.get_size()).as_str());
Ok(stringified)
}
// Get byte offset of nth field
@ -99,7 +103,7 @@ impl CStructInfo {
Ok(offset)
}
pub fn get_type(&self) -> Type {
pub fn get_middle_type(&self) -> Type {
self.middle_type.clone()
}
}
@ -128,7 +132,6 @@ impl FfiConvert for CStructInfo {
for (index, conv) in self.inner_conv_list.iter().enumerate() {
let field_offset = self.offset(index)? as isize;
let data: LuaValue = table.get(index + 1)?;
conv.as_ref().unwrap().value_into_data(
lua,
field_offset + offset,
@ -157,6 +160,19 @@ impl FfiConvert for CStructInfo {
}
Ok(LuaValue::Table(table))
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
dst.get_pointer()
.byte_offset(dst_offset)
.copy_from(src.get_pointer().byte_offset(src_offset), self.get_size());
Ok(())
}
}
impl LuaUserData for CStructInfo {
@ -175,22 +191,16 @@ impl LuaUserData for CStructInfo {
method_provider::provide_box(methods);
method_provider::provide_read_data(methods);
method_provider::provide_write_data(methods);
method_provider::provide_copy_data(methods);
methods.add_method("offset", |_, this, index: usize| {
let offset = this.offset(index)?;
Ok(offset)
});
// Simply pass type in the locked table used when first creating this object.
// By referencing the table to struct, the types inside do not disappear
methods.add_function("field", |lua, (this, field): (LuaAnyUserData, usize)| {
if let LuaValue::Table(fields) = association::get(lua, CSTRUCT_INNER, this)?
.ok_or_else(|| LuaError::external("Field table not found"))?
{
let value: LuaValue = fields.raw_get(field + 1)?;
Ok(value)
} else {
Err(LuaError::external("Failed to read field table"))
}
});
methods.add_method("offset", |_, this, index: usize| this.offset(index));
// Get nth field type userdata
methods.add_function(
"field",
|lua, (this, field_index): (LuaAnyUserData, usize)| {
let field_table = get_field_table(lua, &this)?;
field_table.raw_get::<_, LuaAnyUserData>(field_index + 1)
},
);
}
}

2
crates/lune-std-ffi/src/c/type_info.rs
View file

@ -108,6 +108,8 @@ where
method_provider::provide_box(methods);
method_provider::provide_read_data(methods);
method_provider::provide_write_data(methods);
method_provider::provide_copy_data(methods);
method_provider::provide_stringify_data(methods);
methods.add_function(
"cast",

21
crates/lune-std-ffi/src/c/types/f32.rs
View file

@ -18,7 +18,6 @@ impl FfiConvert for CTypeInfo<f32> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -54,4 +53,24 @@ impl FfiConvert for CTypeInfo<f32> {
};
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<f32>() =
*src.get_pointer().byte_offset(src_offset).cast::<f32>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle.get_pointer().byte_offset(offset).cast::<f32>()).to_string())
}
}

21
crates/lune-std-ffi/src/c/types/f64.rs
View file

@ -18,7 +18,6 @@ impl FfiConvert for CTypeInfo<f64> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -54,4 +53,24 @@ impl FfiConvert for CTypeInfo<f64> {
};
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<f64>() =
*src.get_pointer().byte_offset(src_offset).cast::<f64>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle.get_pointer().byte_offset(offset).cast::<f64>()).to_string())
}
}

21
crates/lune-std-ffi/src/c/types/i128.rs
View file

@ -18,7 +18,6 @@ impl FfiConvert for CTypeInfo<i128> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -54,4 +53,24 @@ impl FfiConvert for CTypeInfo<i128> {
};
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<i128>() =
*src.get_pointer().byte_offset(src_offset).cast::<i128>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle.get_pointer().byte_offset(offset).cast::<i128>()).to_string())
}
}

21
crates/lune-std-ffi/src/c/types/i16.rs
View file

@ -18,7 +18,6 @@ impl FfiConvert for CTypeInfo<i16> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -54,4 +53,24 @@ impl FfiConvert for CTypeInfo<i16> {
};
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<i16>() =
*src.get_pointer().byte_offset(src_offset).cast::<i16>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle.get_pointer().byte_offset(offset).cast::<i16>()).to_string())
}
}

21
crates/lune-std-ffi/src/c/types/i32.rs
View file

@ -18,7 +18,6 @@ impl FfiConvert for CTypeInfo<i32> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -54,4 +53,24 @@ impl FfiConvert for CTypeInfo<i32> {
};
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<i32>() =
*src.get_pointer().byte_offset(src_offset).cast::<i32>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle.get_pointer().byte_offset(offset).cast::<i32>()).to_string())
}
}

21
crates/lune-std-ffi/src/c/types/i64.rs
View file

@ -18,7 +18,6 @@ impl FfiConvert for CTypeInfo<i64> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -54,4 +53,24 @@ impl FfiConvert for CTypeInfo<i64> {
};
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<i64>() =
*src.get_pointer().byte_offset(src_offset).cast::<i64>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle.get_pointer().byte_offset(offset).cast::<i64>()).to_string())
}
}

21
crates/lune-std-ffi/src/c/types/i8.rs
View file

@ -18,7 +18,6 @@ impl FfiConvert for CTypeInfo<i8> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -49,4 +48,24 @@ impl FfiConvert for CTypeInfo<i8> {
unsafe { (*data_handle.get_pointer().byte_offset(offset).cast::<i8>()).into_lua(lua)? };
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<i8>() =
*src.get_pointer().byte_offset(src_offset).cast::<i8>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle.get_pointer().byte_offset(offset).cast::<i8>()).to_string())
}
}

25
crates/lune-std-ffi/src/c/types/isize.rs
View file

@ -18,7 +18,6 @@ impl FfiConvert for CTypeInfo<isize> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -61,4 +60,28 @@ impl FfiConvert for CTypeInfo<isize> {
};
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<isize>() =
*src.get_pointer().byte_offset(src_offset).cast::<isize>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle
.get_pointer()
.byte_offset(offset)
.cast::<isize>())
.to_string())
}
}

21
crates/lune-std-ffi/src/c/types/u128.rs
View file

@ -18,7 +18,6 @@ impl FfiConvert for CTypeInfo<u128> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -54,4 +53,24 @@ impl FfiConvert for CTypeInfo<u128> {
};
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<u128>() =
*src.get_pointer().byte_offset(src_offset).cast::<u128>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle.get_pointer().byte_offset(offset).cast::<u128>()).to_string())
}
}

21
crates/lune-std-ffi/src/c/types/u16.rs
View file

@ -19,7 +19,6 @@ impl FfiConvert for CTypeInfo<u16> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -55,4 +54,24 @@ impl FfiConvert for CTypeInfo<u16> {
};
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<u16>() =
*src.get_pointer().byte_offset(src_offset).cast::<u16>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle.get_pointer().byte_offset(offset).cast::<u16>()).to_string())
}
}

21
crates/lune-std-ffi/src/c/types/u32.rs
View file

@ -18,7 +18,6 @@ impl FfiConvert for CTypeInfo<u32> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -54,4 +53,24 @@ impl FfiConvert for CTypeInfo<u32> {
};
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<u32>() =
*src.get_pointer().byte_offset(src_offset).cast::<u32>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle.get_pointer().byte_offset(offset).cast::<f32>()).to_string())
}
}

21
crates/lune-std-ffi/src/c/types/u64.rs
View file

@ -18,7 +18,6 @@ impl FfiConvert for CTypeInfo<u64> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -54,4 +53,24 @@ impl FfiConvert for CTypeInfo<u64> {
};
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<u64>() =
*src.get_pointer().byte_offset(src_offset).cast::<u64>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle.get_pointer().byte_offset(offset).cast::<u64>()).to_string())
}
}

21
crates/lune-std-ffi/src/c/types/u8.rs
View file

@ -19,7 +19,6 @@ impl FfiConvert for CTypeInfo<u8> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -52,4 +51,24 @@ impl FfiConvert for CTypeInfo<u8> {
unsafe { (*data_handle.get_pointer().byte_offset(offset).cast::<u8>()).into_lua(lua)? };
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<u8>() =
*src.get_pointer().byte_offset(src_offset).cast::<u8>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle.get_pointer().byte_offset(offset).cast::<u8>()).to_string())
}
}

25
crates/lune-std-ffi/src/c/types/usize.rs
View file

@ -18,7 +18,6 @@ impl FfiConvert for CTypeInfo<usize> {
unsafe fn value_into_data<'lua>(
&self,
_lua: &'lua Lua,
// _type_userdata: &LuaAnyUserData<'lua>,
offset: isize,
data_handle: &Ref<dyn FfiData>,
value: LuaValue<'lua>,
@ -61,4 +60,28 @@ impl FfiConvert for CTypeInfo<usize> {
};
Ok(value)
}
unsafe fn copy_data(
&self,
_lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()> {
*dst.get_pointer().byte_offset(dst_offset).cast::<usize>() =
*src.get_pointer().byte_offset(src_offset).cast::<usize>();
Ok(())
}
unsafe fn stringify_data(
&self,
_lua: &Lua,
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Ok((*data_handle
.get_pointer()
.byte_offset(offset)
.cast::<usize>())
.to_string())
}
}

25
crates/lune-std-ffi/src/c/void_info.rs
View file

@ -1 +1,24 @@
use core::ffi::c_void;
use libffi::middle::Type;
use mlua::prelude::*;
use crate::ffi::{FfiSignedness, FfiSize};
pub struct CVoidInfo();
impl FfiSignedness for CVoidInfo {
fn get_signedness(&self) -> bool {
false
}
}
impl FfiSize for CVoidInfo {
fn get_size(&self) -> usize {
0
}
}
impl CVoidInfo {
pub fn get_middle_type() -> Type {
Type::void()
}
}
impl LuaUserData for CVoidInfo {}

15
crates/lune-std-ffi/src/data/closure_data.rs
View file

@ -9,7 +9,7 @@ use mlua::prelude::*;
use super::ref_data::{RefBounds, RefData, RefFlag};
use crate::ffi::{
libffi_helper::{FFI_STATUS_NAMES, SIZE_OF_POINTER},
libffi_helper::{ffi_status_assert, SIZE_OF_POINTER},
FfiArg, FfiData, FfiResult,
};
@ -97,7 +97,7 @@ impl ClosureData {
func,
};
let prep_result = unsafe {
ffi_status_assert(unsafe {
ffi_prep_closure_loc(
closure,
cif,
@ -105,16 +105,9 @@ impl ClosureData {
ptr::from_ref(&closure_data).cast::<c_void>().cast_mut(),
code.as_mut_ptr(),
)
};
})?;
if prep_result != 0 {
Err(LuaError::external(format!(
"ffi_get_struct_offsets failed. expected result {}, got {}",
FFI_STATUS_NAMES[0], FFI_STATUS_NAMES[prep_result as usize]
)))
} else {
Ok(closure_data)
}
Ok(closure_data)
}
}

11
crates/lune-std-ffi/src/ffi/libffi_helper.rs
View file

@ -35,3 +35,14 @@ pub const FFI_STATUS_NAMES: [&str; 4] = [
"ffi_status_FFI_BAD_ABI",
"ffi_status_FFI_BAD_ARGTYPE",
];
pub fn ffi_status_assert(result: raw::ffi_status) -> LuaResult<()> {
if result == raw::ffi_status_FFI_OK {
Ok(())
} else {
Err(LuaError::external(format!(
"ffi_status assertion failed. expected result {}, got {}",
FFI_STATUS_NAMES[0], FFI_STATUS_NAMES[result as usize]
)))
}
}

18
crates/lune-std-ffi/src/ffi/mod.rs
View file

@ -37,6 +37,24 @@ pub trait FfiConvert {
offset: isize,
data_handle: &Ref<dyn FfiData>,
) -> LuaResult<LuaValue<'lua>>;
unsafe fn copy_data(
&self,
lua: &Lua,
dst_offset: isize,
src_offset: isize,
dst: &Ref<dyn FfiData>,
src: &Ref<dyn FfiData>,
) -> LuaResult<()>;
unsafe fn stringify_data(
&self,
_lua: &Lua,
_offset: isize,
_data_handle: &Ref<dyn FfiData>,
) -> LuaResult<String> {
Err(LuaError::external("stringify not implemented"))
}
}
pub trait FfiData {

3
types/ffi.luau
View file

@ -11,6 +11,7 @@ export type CTypeInfo<T, R> = {
box: (self: CTypeInfo<T, R>, val: R) -> Box,
readData: (self: CTypeInfo<T, R>, target: (Ref|Box), offset: number?) -> R,
writeData: (self: CTypeInfo<T, R>, target: (Ref|Box), value: R, offset: number?) -> (),
stringifyData: (self: CTypeInfo<T, R>, target: (Ref|Box), offset: number?) -> string,
-- FIXME: recursive types; 'intoType' should be CTypes
cast: (self: CTypeInfo<T, R>, intoType: any, fromData: (Ref|Box), intoData: (Ref|Box)) -> (),
@ -27,7 +28,6 @@ export type CPtrInfo<T> = {
readRef: (self: CPtrInfo<T>, target: (Ref|Box), offset: number?) -> Ref,
writeRef: (self: CPtrInfo<T>, target: (Ref|Box), value: (Ref|Box), offset: number?) -> (),
}
export type CArrInfo<T, R> = {
@ -42,6 +42,7 @@ export type CArrInfo<T, R> = {
box: (self: CArrInfo<T, R>, table: { T }) -> Box,
readData: (self: CArrInfo<T, R>, target: (Ref|Box), offset: number?) -> { T },
writeData: (self: CArrInfo<T, R>, target: (Ref|Box), value: { T }, offset: number?) -> (),
copyData: (self: CArrInfo<T, R>, dst: (Ref|Box), src: (Ref|Box), dst_offset: number?, src_offset: number?) -> (),
offset: (self: CArrInfo<T, R>, offset: number) -> number,
}