luaur_code_gen/functions/
translate_inst_set_table.rs

1use crate::enums::ir_cmd::IrCmd;
2use crate::functions::is_userdata_bytecode_type::is_userdata_bytecode_type;
3use crate::records::fallback_stream_scope::FallbackStreamScope;
4use crate::records::ir_builder::IrBuilder;
5use crate::type_aliases::instruction_ir_translation::Instruction;
6use luaur_common::enums::luau_bytecode_type::LuauBytecodeType;
7use luaur_common::macros::luau_insn_a::LUAU_INSN_A;
8use luaur_common::macros::luau_insn_b::LUAU_INSN_B;
9use luaur_common::macros::luau_insn_c::LUAU_INSN_C;
10use luaur_vm::enums::lua_type::lua_Type;
11
12pub fn translate_inst_set_table(build: &mut IrBuilder, pc: *const Instruction, pcpos: i32) {
13    let insn = unsafe { *pc };
14    let ra = LUAU_INSN_A(insn) as u8;
15    let rb = LUAU_INSN_B(insn) as u8;
16    let rc = LUAU_INSN_C(insn) as u8;
17
18    let bc_types = build.function.get_bytecode_types_at(pcpos);
19
20    if is_userdata_bytecode_type(bc_types.a)
21        || bc_types.b == LuauBytecodeType::LBC_TYPE_STRING.0 as u8
22    {
23        let savedpc_arg = build.const_uint((pcpos + 1) as u32);
24        build.inst_ir_cmd_ir_op(IrCmd::SET_SAVEDPC, savedpc_arg);
25        let reg_ra = build.vm_reg(ra);
26        let reg_rb = build.vm_reg(rb);
27        let reg_rc = build.vm_reg(rc);
28        build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::SET_TABLE, reg_ra, reg_rb, reg_rc);
29        return;
30    }
31
32    let fallback = build.fallback_block(pcpos as u32);
33
34    let reg_rb = build.vm_reg(rb);
35    let tb = build.inst_ir_cmd_ir_op(IrCmd::LOAD_TAG, reg_rb);
36    let const_tag_table = build.const_tag(lua_Type::LUA_TTABLE as u8);
37    let exit_or_fallback = if bc_types.a == LuauBytecodeType::LBC_TYPE_TABLE.0 as u8 {
38        build.vm_exit(pcpos as u32)
39    } else {
40        fallback
41    };
42    build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::CHECK_TAG, tb, const_tag_table, exit_or_fallback);
43
44    let reg_rc = build.vm_reg(rc);
45    let tc = build.inst_ir_cmd_ir_op(IrCmd::LOAD_TAG, reg_rc);
46    let const_tag_number = build.const_tag(lua_Type::LUA_TNUMBER as u8);
47    let exit_or_fallback_c = if bc_types.b == LuauBytecodeType::LBC_TYPE_NUMBER.0 as u8 {
48        build.vm_exit(pcpos as u32)
49    } else {
50        fallback
51    };
52    build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::CHECK_TAG, tc, const_tag_number, exit_or_fallback_c);
53
54    let vb = build.inst_ir_cmd_ir_op(IrCmd::LOAD_POINTER, reg_rb);
55    let vc = build.inst_ir_cmd_ir_op(IrCmd::LOAD_DOUBLE, reg_rc);
56
57    let index = build.inst_ir_cmd_ir_op_ir_op(IrCmd::TRY_NUM_TO_INDEX, vc, fallback);
58
59    let const_int_one = build.const_int(1);
60    let index = build.inst_ir_cmd_ir_op_ir_op(IrCmd::SUB_INT, index, const_int_one);
61
62    build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::CHECK_ARRAY_SIZE, vb, index, fallback);
63    build.inst_ir_cmd_ir_op_ir_op(IrCmd::CHECK_NO_METATABLE, vb, fallback);
64    build.inst_ir_cmd_ir_op_ir_op(IrCmd::CHECK_READONLY, vb, fallback);
65
66    let arr_el = build.inst_ir_cmd_ir_op_ir_op(IrCmd::GET_ARR_ADDR, vb, index);
67
68    let reg_ra = build.vm_reg(ra);
69    let tva = build.inst_ir_cmd_ir_op(IrCmd::LOAD_TVALUE, reg_ra);
70    build.inst_ir_cmd_ir_op_ir_op(IrCmd::STORE_TVALUE, arr_el, tva);
71
72    let undef = build.undef();
73    build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::BARRIER_TABLE_FORWARD, vb, reg_ra, undef);
74
75    let next = build.block_at_inst((pcpos + 1) as u32);
76    let mut scope = FallbackStreamScope::new(build, fallback, next);
77    let build = &mut *scope.build;
78
79    let savedpc_arg = build.const_uint((pcpos + 1) as u32);
80    build.inst_ir_cmd_ir_op(IrCmd::SET_SAVEDPC, savedpc_arg);
81    let reg_ra = build.vm_reg(ra);
82    let reg_rb = build.vm_reg(rb);
83    let reg_rc = build.vm_reg(rc);
84    build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::SET_TABLE, reg_ra, reg_rb, reg_rc);
85    build.inst_ir_cmd_ir_op(IrCmd::JUMP, next);
86}
luaur_code_gen/functions/translate_inst_set_table.rs

luaur_code_gen/functions/
translate_inst_set_table.rs
