luaur-bytecode 0.1.2

Luau bytecode format and builder (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

use crate::records::bc_block::BcBlock;
use crate::records::bc_get_table_ks::BcGetTableKS;
use crate::records::bc_inst_helper::BcInstHelper;
use crate::records::bc_move::BcMove;
use crate::records::bc_op::BcOp;
use crate::records::bc_ref::BcRef;
use crate::records::call_inliner::CallInliner;
use crate::type_aliases::reg::Reg;
use luaur_common::macros::luau_assert::LUAU_ASSERT;

impl<'a> CallInliner<'a> {
    pub fn replace_namecall(
        &mut self,
        namecall: BcOp,
        prev_block: &mut BcRef<'a, BcBlock>,
    ) -> BcOp {
        // move LOP_NAMECALL to call block
        let caller_ptr: *mut crate::records::bc_function::BcFunction = self.caller;
        let inst_ref = unsafe { (&*caller_ptr).inst(namecall) };
        let mut helper = BcInstHelper {
            graph: unsafe { &mut *caller_ptr },
            inst: inst_ref,
        };
        let table = helper.get_bc_op(0);
        let hint_op = helper.get_bc_op(1);
        let hint = unsafe { helper.graph.imm_op(hint_op).value.valueInt as u32 };
        let key = helper.get_bc_op(2).index;
        helper.prepend_to(self.call.base.operator_deref().block);
        prev_block.operator_deref_mut().ops.pop_back();
        LUAU_ASSERT!(self.target_reg == helper.get_out_reg());
        let table_reg: Reg = helper.get_out_reg() + 1;
        drop(helper);

        // and replace it with LOP_MOVE + LOP_GETTABLEKS
        let mut move_helper = BcMove::<crate::records::bc_function::VmConst>::create(self.caller);
        move_helper.set_out_reg(table_reg);
        move_helper.set_src(table);
        move_helper.append_to(prev_block.op);
        let move_op = move_helper.op();
        drop(move_helper);

        let mut get_table_ks_helper =
            BcGetTableKS::<crate::records::bc_function::VmConst>::create(self.caller);
        get_table_ks_helper.set_source(move_op);
        get_table_ks_helper.set_hint(hint);
        get_table_ks_helper.set_key(key);
        get_table_ks_helper.set_out_reg(self.target_reg);
        get_table_ks_helper.append_to(prev_block.op);

        get_table_ks_helper.op()
    }
}