luaur-code-gen 0.1.2

Native (A64/X64) code generation for Luau (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

use crate::macros::vm_reg::VM_REG;
use luaur_vm::functions::lua_d_performcally::lua_d_performcally;
use luaur_vm::macros::setobj_2_s::setobj_2_s as setobj2s;
use luaur_vm::macros::ttisnil::ttisnil;

use crate::type_aliases::lua_state::LuaState;
use luaur_vm::type_aliases::t_value::TValue;

pub unsafe fn forg_loop_non_table_fallback(
    L: *mut luaur_vm::records::lua_state::lua_State,
    insn_a: i32,
    aux: i32,
) -> i32 {
    let l_ptr = L;
    let base: *mut TValue = (*l_ptr).base;
    let mut ra: *mut TValue = VM_REG!(insn_a, l_ptr, base);

    // note: it's safe to push arguments past top for complicated reasons (see lvmexecute.cpp)
    setobj2s!(l_ptr, ra.add(3 + 2), ra.add(2));
    setobj2s!(l_ptr, ra.add(3 + 1), ra.add(1));
    setobj2s!(l_ptr, ra.add(3), ra);

    (*l_ptr).top = ra.add(3 + 3); // func + 2 args (state and index)
    luaur_common::LUAU_ASSERT!((*l_ptr).top <= (*l_ptr).stack_last);

    if lua_d_performcally(l_ptr, ra.add(3), aux as u8 as i32) {
        return -1; // yield/break, caller must exit native execution
    }

    (*l_ptr).top = (*(*l_ptr).ci).top;

    // recompute ra since stack might have been reallocated
    let base = (*l_ptr).base;
    ra = VM_REG!(insn_a, l_ptr, base);

    // copy first variable back into the iteration index
    setobj2s!(l_ptr, ra.add(2), ra.add(3));

    if ttisnil!(ra.add(3)) {
        0
    } else {
        1
    }
}

#[no_mangle]
pub unsafe extern "C" fn forgLoopNonTableFallback(
    L: *mut luaur_vm::records::lua_state::LuaState,
    insn_a: i32,
    aux: i32,
) -> i32 {
    forg_loop_non_table_fallback(L, insn_a, aux)
}