Skip to main content

luaur_vm/functions/
lua_objlen.rs

1use core::ffi::c_int;
2
3use crate::enums::lua_type::lua_Type;
4use crate::functions::index_2_addr::index_2_addr;
5use crate::functions::lua_h_getn::lua_h_getn;
6use crate::macros::bufvalue::bufvalue;
7use crate::macros::hvalue::hvalue;
8use crate::macros::tsvalue::tsvalue;
9use crate::macros::ttype::ttype;
10use crate::macros::uvalue::uvalue;
11use crate::type_aliases::lua_state::lua_State;
12use crate::type_aliases::stk_id::StkId;
13
14#[allow(non_snake_case)]
15pub unsafe fn lua_objlen(L: *mut lua_State, idx: c_int) -> c_int {
16    let o: StkId = index_2_addr(L, idx);
17    let tt = ttype!(o);
18
19    if tt == lua_Type::LUA_TSTRING as i32 {
20        (*tsvalue!(o)).len as i32
21    } else if tt == lua_Type::LUA_TUSERDATA as i32 {
22        (*uvalue!(o)).len as i32
23    } else if tt == lua_Type::LUA_TBUFFER as i32 {
24        (*bufvalue!(o)).len as i32
25    } else if tt == lua_Type::LUA_TTABLE as i32 {
26        // The current stub for lua_h_getn takes 0 arguments and returns ().
27        // However, the C++ source calls it with a LuaTable pointer and expects an int.
28        // We must cast the function to the correct signature to match the C++ logic.
29        let luaH_getn_ptr = lua_h_getn as *const ();
30        let luaH_getn_real = core::mem::transmute::<
31            *const (),
32            unsafe fn(*mut crate::records::lua_table::LuaTable) -> c_int,
33        >(luaH_getn_ptr);
34
35        luaH_getn_real(hvalue!(o))
36    } else {
37        0
38    }
39}