Skip to main content

luaur_code_gen/methods/
assembly_builder_x_64_place_shift.rs

1use crate::enums::category_x_64::CategoryX64;
2use crate::enums::size_x_64::SizeX64;
3use crate::macros::codegen_assert::CODEGEN_ASSERT;
4use crate::records::assembly_builder_x_64::AssemblyBuilderX64;
5use crate::records::operand_x_64::OperandX64;
6use crate::records::register_x_64::RegisterX64;
7
8impl AssemblyBuilderX64 {
9    pub fn place_shift(
10        &mut self,
11        name: *const core::ffi::c_char,
12        lhs: OperandX64,
13        rhs: OperandX64,
14        opreg: u8,
15    ) {
16        if self.log_text {
17            self.log_c_char_operand_x_64_operand_x_64(name, lhs, rhs);
18        }
19
20        let cl = RegisterX64 {
21            bits: (1u8 << RegisterX64::INDEX_SHIFT) | SizeX64::byte as u8,
22        };
23
24        if !(lhs.cat == CategoryX64::reg || lhs.cat == CategoryX64::mem) {
25            luaur_common::LUAU_DEBUGBREAK!();
26        }
27        if !(rhs.cat == CategoryX64::imm || (rhs.cat == CategoryX64::reg && rhs.base == cl)) {
28            luaur_common::LUAU_DEBUGBREAK!();
29        }
30
31        let size = lhs.base.size();
32
33        self.place_rex_register_x_64(lhs.base);
34
35        if rhs.cat == CategoryX64::imm && rhs.imm == 1 {
36            self.place(if size == SizeX64::byte { 0xd0 } else { 0xd1 });
37            self.place_mod_reg_mem(lhs, opreg, 0);
38        } else if rhs.cat == CategoryX64::imm {
39            if !((rhs.imm as i8) as i32 == rhs.imm) {
40                luaur_common::LUAU_DEBUGBREAK!();
41            }
42
43            self.place(if size == SizeX64::byte { 0xc0 } else { 0xc1 });
44            self.place_mod_reg_mem(lhs, opreg, 1);
45            self.place_imm_8(rhs.imm);
46        } else {
47            self.place(if size == SizeX64::byte { 0xd2 } else { 0xd3 });
48            self.place_mod_reg_mem(lhs, opreg, 0);
49        }
50
51        self.commit();
52    }
53}