luaur-vm 0.1.1

The Luau register virtual machine and standard library (Rust).
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

//! Node: `cxx:Function:Luau.VM:VM/src/lapi.cpp:73:pseudo2addr`
//! Source: `VM/src/lapi.cpp:65-97` (hand-ported; includes the file-static
//! `getcurrenv` helper inlined here since it was never a graph node)

use crate::macros::api_check::api_check;
use crate::macros::curr_func::curr_func;
use crate::macros::lua_environindex::LUA_ENVIRONINDEX;
use crate::macros::lua_globalsindex::LUA_GLOBALSINDEX;
use crate::macros::lua_ispseudo::lua_ispseudo;
use crate::macros::lua_o_nilobject::luaO_nilobject;
use crate::macros::lua_registryindex::LUA_REGISTRYINDEX;
use crate::macros::registry::registry;
use crate::macros::sethvalue::sethvalue;
use crate::records::closure::Closure;
use crate::records::lua_table::LuaTable;
use crate::type_aliases::lua_state::lua_State;
use crate::type_aliases::stk_id::StkId;
use crate::type_aliases::t_value::TValue;

unsafe fn getcurrenv(L: *mut lua_State) -> *mut LuaTable {
    if (*L).ci == (*L).base_ci {
        // no enclosing function? use global table as environment
        (*L).gt
    } else {
        let func = curr_func!(L);
        (*func).env
    }
}

pub unsafe fn pseudo_2_addr(L: *mut lua_State, idx: core::ffi::c_int) -> StkId {
    api_check!(L, lua_ispseudo(idx));
    match idx {
        // pseudo-indices
        LUA_REGISTRYINDEX => registry!(L) as *const TValue as *mut TValue,
        LUA_ENVIRONINDEX => {
            let tmp = &mut (*(*L).global).pseudotemp as *mut TValue;
            sethvalue!(L, tmp, getcurrenv(L));
            tmp
        }
        LUA_GLOBALSINDEX => {
            let tmp = &mut (*(*L).global).pseudotemp as *mut TValue;
            sethvalue!(L, tmp, (*L).gt);
            tmp
        }
        _ => {
            let func = curr_func!(L);
            let idx = LUA_GLOBALSINDEX - idx;
            if idx <= (*func).nupvalues as i32 {
                let c = &mut (*func).inner.c;
                c.upvals.as_mut_ptr().add((idx - 1) as usize)
            } else {
                luaO_nilobject as *mut TValue
            }
        }
    }
}