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

use crate::functions::get_native_proto_exec_data_header_native_proto_exec_data_alt_b::get_native_proto_exec_data_header;
use crate::macros::codegen_assert::CODEGEN_ASSERT;
use crate::type_aliases::native_proto_exec_data_ptr::NativeProtoExecDataPtr;
use alloc::vec::Vec;
use luaur_common::enums::luau_opcode::LuauOpcode;
use luaur_vm::records::proto::Proto;
use luaur_vm::type_aliases::instruction::Instruction;

static K_CODE_ENTRY_INSN: Instruction = LuauOpcode::LOP_NATIVECALL as u32;

pub fn bind_native_protos(
    module_protos: &Vec<*mut Proto>,
    native_protos: &mut Vec<NativeProtoExecDataPtr>,
    _release: bool,
) -> u32 {
    let mut protos_bound = 0u32;
    let mut proto_it = 0usize;

    for native_proto in native_protos.iter_mut() {
        let header = unsafe { &*get_native_proto_exec_data_header(native_proto.as_ptr()) };

        while proto_it != module_protos.len()
            && unsafe { (*module_protos[proto_it]).bytecodeid as u32 } != header.bytecode_id
        {
            proto_it += 1;
        }

        CODEGEN_ASSERT!(proto_it != module_protos.len());

        let proto = module_protos[proto_it];

        unsafe {
            (*proto).execdata = native_proto.as_ptr().cast();
            (*proto).exectarget = header.entry_offset_or_address as usize;
            (*proto).codeentry = &K_CODE_ENTRY_INSN;
        }

        protos_bound += 1;
    }

    protos_bound
}