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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

use crate::enums::arch::Arch;
use crate::functions::build_entry_function_code_gen_x_64::build_entry_function;
use crate::macros::codegen_assert::CODEGEN_ASSERT;
use crate::records::assembly_builder_x_64::AssemblyBuilderX64;
use crate::records::base_code_gen_context::BaseCodeGenContext;
use crate::records::unwind_builder::UnwindBuilder;

pub fn init_header_functions(code_gen_context: &mut BaseCodeGenContext) -> bool {
    let mut build = AssemblyBuilderX64::assembly_builder_x_64_bool_i32(false, 0);

    unsafe {
        start_info(&mut *code_gen_context.unwind_builder, Arch::X64);
    }

    let entry_locations =
        unsafe { build_entry_function(&mut build, &mut *code_gen_context.unwind_builder) };

    build.finalize();

    unsafe {
        finish_info(&mut *code_gen_context.unwind_builder);
    }

    CODEGEN_ASSERT!(build.data.is_empty());

    let mut code_start: *mut u8 = core::ptr::null_mut();

    if luaur_common::FFlag::LuauCodegenFreeBlocks.get() {
        code_gen_context.gate_allocation_data = code_gen_context.code_allocator.allocate(
            build.data.as_ptr(),
            build.data.len(),
            build.code.as_ptr(),
            build.code.len(),
        );

        if code_gen_context.gate_allocation_data.start.is_null() {
            return false;
        }

        code_start = code_gen_context.gate_allocation_data.code_start;
    } else if !code_gen_context.code_allocator.allocate_deprecated(
        build.data.as_ptr(),
        build.data.len(),
        build.code.as_ptr(),
        build.code.len(),
        &mut code_gen_context.gate_data_deprecated,
        &mut code_gen_context.gate_data_size_deprecated,
        &mut code_start,
    ) {
        return false;
    }

    unsafe {
        set_begin_offset(
            &mut *code_gen_context.unwind_builder,
            build.get_label_offset(&entry_locations.prologueEnd) as usize,
        );

        code_gen_context.context.gateEntry =
            code_start.add(build.get_label_offset(&entry_locations.start) as usize);
        code_gen_context.context.gateExit =
            code_start.add(build.get_label_offset(&entry_locations.epilogueStart) as usize);
    }

    true
}

#[cfg(target_os = "windows")]
unsafe fn start_info(unwind: &mut UnwindBuilder, arch: Arch) {
    (&mut *(unwind as *mut UnwindBuilder)
        .cast::<crate::records::unwind_builder_win::UnwindBuilderWin>())
        .start_info(arch);
}

#[cfg(not(target_os = "windows"))]
unsafe fn start_info(unwind: &mut UnwindBuilder, arch: Arch) {
    (&mut *(unwind as *mut UnwindBuilder)
        .cast::<crate::records::unwind_builder_dwarf_2::UnwindBuilderDwarf2>())
        .start_info(arch);
}

#[cfg(target_os = "windows")]
unsafe fn finish_info(unwind: &mut UnwindBuilder) {
    (&mut *(unwind as *mut UnwindBuilder)
        .cast::<crate::records::unwind_builder_win::UnwindBuilderWin>())
        .finish_info();
}

#[cfg(not(target_os = "windows"))]
unsafe fn finish_info(unwind: &mut UnwindBuilder) {
    (&mut *(unwind as *mut UnwindBuilder)
        .cast::<crate::records::unwind_builder_dwarf_2::UnwindBuilderDwarf2>())
        .finish_info();
}

#[cfg(target_os = "windows")]
unsafe fn set_begin_offset(unwind: &mut UnwindBuilder, begin_offset: usize) {
    (&mut *(unwind as *mut UnwindBuilder)
        .cast::<crate::records::unwind_builder_win::UnwindBuilderWin>())
        .set_begin_offset(begin_offset);
}

#[cfg(not(target_os = "windows"))]
unsafe fn set_begin_offset(unwind: &mut UnwindBuilder, begin_offset: usize) {
    (&mut *(unwind as *mut UnwindBuilder)
        .cast::<crate::records::unwind_builder_dwarf_2::UnwindBuilderDwarf2>())
        .set_begin_offset(begin_offset);
}