luaur-code-gen 0.1.0

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
54
55
56
57

use crate::enums::ir_cmd::IrCmd;
use crate::records::ir_builder::IrBuilder;
use crate::type_aliases::instruction_ir_builder::Instruction;
use luaur_common::macros::luau_insn_a::LUAU_INSN_A;
use luaur_common::macros::luau_insn_d::LUAU_INSN_D;
use luaur_vm::enums::lua_type::lua_Type;
use luaur_vm::macros::lu_tag_iterator::LU_TAG_ITERATOR;

#[allow(non_snake_case)]
pub fn translate_inst_for_g_prep_next(build: &mut IrBuilder, pc: *const Instruction, pcpos: i32) {
    let ra = LUAU_INSN_A(unsafe { *pc }) as u8;

    let target = build.block_at_inst((pcpos + 1 + LUAU_INSN_D(unsafe { *pc })) as u32);
    let fallback = build.fallback_block(pcpos as u32);

    build.check_safe_env(pcpos);

    let vm_reg_ra_plus_1 = build.vm_reg(ra + 1);
    let tag_b = build.inst_ir_cmd_ir_op(IrCmd::LOAD_TAG, vm_reg_ra_plus_1);
    let const_tag_table = build.const_tag(lua_Type::LUA_TTABLE as u8);
    build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::CHECK_TAG, tag_b, const_tag_table, fallback);

    let vm_reg_ra_plus_2 = build.vm_reg(ra + 2);
    let tag_c = build.inst_ir_cmd_ir_op(IrCmd::LOAD_TAG, vm_reg_ra_plus_2);
    let const_tag_nil = build.const_tag(0);
    build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::CHECK_TAG, tag_c, const_tag_nil, fallback);

    let vm_reg_ra = build.vm_reg(ra);
    build.inst_ir_cmd_ir_op_ir_op(IrCmd::STORE_TAG, vm_reg_ra, const_tag_nil);

    let vm_reg_ra_plus_2 = build.vm_reg(ra + 2);
    let const_int_zero = build.const_int(0);
    build.inst_ir_cmd_ir_op_ir_op(IrCmd::STORE_POINTER, vm_reg_ra_plus_2, const_int_zero);

    let const_int_lu_tag_iterator = build.const_int(LU_TAG_ITERATOR);
    build.inst_ir_cmd_ir_op_ir_op(
        IrCmd::STORE_EXTRA,
        vm_reg_ra_plus_2,
        const_int_lu_tag_iterator,
    );

    let vm_reg_ra_plus_2 = build.vm_reg(ra + 2);
    let const_tag_lightuserdata = build.const_tag(6);
    build.inst_ir_cmd_ir_op_ir_op(IrCmd::STORE_TAG, vm_reg_ra_plus_2, const_tag_lightuserdata);

    build.inst_ir_cmd_ir_op(IrCmd::JUMP, target);

    build.begin_block(fallback);
    let const_uint_pcpos = build.const_uint(pcpos as u32);
    let vm_reg_ra = build.vm_reg(ra);
    build.inst_ir_cmd_ir_op_ir_op_ir_op(
        IrCmd::FORGPREP_XNEXT_FALLBACK,
        const_uint_pcpos,
        vm_reg_ra,
        target,
    );
}