luaur-code-gen 0.1.3

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

use crate::records::native_module::NativeModule;
use crate::records::native_module_ref::NativeModuleRef;
use crate::records::shared_code_allocator::SharedCodeAllocator;
use crate::type_aliases::native_proto_exec_data_ptr::NativeProtoExecDataPtr;
use alloc::boxed::Box;
use alloc::vec::Vec;
use core::sync::atomic::Ordering;
use luaur_common::FFlag;

impl SharedCodeAllocator {
    pub fn shared_code_allocator_insert_anonymous_native_module(
        &mut self,
        native_protos: Vec<NativeProtoExecDataPtr>,
        data: *const u8,
        data_size: usize,
        code: *const u8,
        code_size: usize,
    ) -> NativeModuleRef {
        unsafe {
            let native_module = if FFlag::LuauCodegenFreeBlocks.get() {
                let result = (*self.code_allocator).allocate(data, data_size, code, code_size);

                if result.start.is_null() {
                    return NativeModuleRef::default();
                }

                Box::new(
                    NativeModule::native_module_shared_code_allocator_optional_module_id_code_allocation_data_vector_native_proto_exec_data_ptr(
                        self as *mut SharedCodeAllocator,
                        &None,
                        result,
                        native_protos,
                    ),
                )
            } else {
                let mut native_data = core::ptr::null_mut();
                let mut native_data_size = 0;
                let mut code_start = core::ptr::null_mut();

                if !(*self.code_allocator).allocate_deprecated(
                    data,
                    data_size,
                    code,
                    code_size,
                    &mut native_data,
                    &mut native_data_size,
                    &mut code_start,
                ) {
                    return NativeModuleRef::default();
                }

                Box::new(
                    NativeModule::native_module_shared_code_allocator_optional_module_id_u8_vector_native_proto_exec_data_ptr(
                        self as *mut SharedCodeAllocator,
                        &None,
                        code_start,
                        native_protos,
                    ),
                )
            };

            let native_module_ptr = Box::into_raw(native_module);
            self.anonymous_module_count.fetch_add(1, Ordering::Relaxed);

            NativeModuleRef::native_module_ref_native_module(native_module_ptr)
        }
    }
}