To protect your data, the CISO officer has suggested users to enable GitLab 2FA as soon as possible.

mod.rs 4.97 KB
Newer Older
1
pub mod inst_sel;
2
pub mod reg_alloc;
3
pub mod peephole_opt;
4
pub mod code_emission;
5

qinsoon's avatar
qinsoon committed
6
use utils::ByteSize;
qinsoon's avatar
qinsoon committed
7

8
9
10
pub type Word = usize;
pub const WORD_SIZE : ByteSize = 8;

11
12
13
pub const AOT_EMIT_DIR : &'static str = "emit";
pub const AOT_EMIT_CONTEXT_FILE : &'static str = "context.s";

qinsoon's avatar
qinsoon committed
14
// X86_64
qinsoon's avatar
qinsoon committed
15

16
#[cfg(target_arch = "x86_64")]
17
#[path = "arch/x86_64/mod.rs"]
18
pub mod x86_64;
19

qinsoon's avatar
qinsoon committed
20
#[cfg(target_arch = "x86_64")]
qinsoon's avatar
qinsoon committed
21
pub use compiler::backend::x86_64::init_machine_regs_for_func;
qinsoon's avatar
qinsoon committed
22
23

#[cfg(target_arch = "x86_64")]
qinsoon's avatar
qinsoon committed
24
pub use compiler::backend::x86_64::number_of_regs_in_group;
qinsoon's avatar
qinsoon committed
25
#[cfg(target_arch = "x86_64")]
qinsoon's avatar
qinsoon committed
26
27
28
pub use compiler::backend::x86_64::number_of_all_regs;
#[cfg(target_arch = "x86_64")]
pub use compiler::backend::x86_64::all_regs;
29
#[cfg(target_arch = "x86_64")]
30
31
pub use compiler::backend::x86_64::all_usable_regs;
#[cfg(target_arch = "x86_64")]
32
pub use compiler::backend::x86_64::pick_group_for_reg;
33
34
#[cfg(target_arch = "x86_64")]
pub use compiler::backend::x86_64::is_callee_saved;
35
36
37
38
#[cfg(target_arch = "x86_64")]
pub use compiler::backend::x86_64::emit_code;
#[cfg(target_arch = "x86_64")]
pub use compiler::backend::x86_64::emit_context;
qinsoon's avatar
qinsoon committed
39

qinsoon's avatar
qinsoon committed
40
41
// ARM

qinsoon's avatar
qinsoon committed
42
43
44
#[cfg(target_arch = "arm")]
#[path = "arch/arm/mod.rs"]
mod arm;
qinsoon's avatar
qinsoon committed
45
46
47

// common data structure with target specific info

qinsoon's avatar
qinsoon committed
48
use vm::VM;
qinsoon's avatar
qinsoon committed
49
50
use ast::types::*;
use ast::ptr::*;
51
use ast::ir::*;
qinsoon's avatar
qinsoon committed
52
pub fn resolve_backend_type_info (ty: &MuType, vm: &VM) -> BackendTypeInfo {
qinsoon's avatar
qinsoon committed
53
    match ty.v {
qinsoon's avatar
qinsoon committed
54
        // integral
qinsoon's avatar
qinsoon committed
55
        MuType_::Int(size_in_bit) => {
qinsoon's avatar
qinsoon committed
56
57
58
59
60
61
62
63
64
            match size_in_bit {
                8  => BackendTypeInfo{size: 1, alignment: 1, struct_layout: None},
                16 => BackendTypeInfo{size: 2, alignment: 2, struct_layout: None},
                32 => BackendTypeInfo{size: 4, alignment: 4, struct_layout: None},
                64 => BackendTypeInfo{size: 8, alignment: 8, struct_layout: None},
                _ => unimplemented!()
            }
        },
        // pointer of any type
qinsoon's avatar
qinsoon committed
65
66
67
68
69
70
71
72
73
        MuType_::Ref(_)
        | MuType_::IRef(_)
        | MuType_::WeakRef(_)
        | MuType_::UPtr(_)
        | MuType_::FuncRef(_)
        | MuType_::UFuncPtr(_)
        | MuType_::Tagref64
        | MuType_::ThreadRef
        | MuType_::StackRef => BackendTypeInfo{size: 8, alignment: 8, struct_layout: None},
qinsoon's avatar
qinsoon committed
74
        // floating point
qinsoon's avatar
qinsoon committed
75
76
        MuType_::Float => BackendTypeInfo{size: 4, alignment: 4, struct_layout: None},
        MuType_::Double => BackendTypeInfo{size: 8, alignment: 8, struct_layout: None},
qinsoon's avatar
qinsoon committed
77
        // array
qinsoon's avatar
qinsoon committed
78
        MuType_::Array(ref ty, len) => {
79
            let ele_ty = vm.get_backend_type_info(ty.id());
qinsoon's avatar
qinsoon committed
80
81
82
83
            
            BackendTypeInfo{size: ele_ty.size * len, alignment: ele_ty.alignment, struct_layout: None}
        }
        // struct
84
        MuType_::Struct(ref name) => {
qinsoon's avatar
qinsoon committed
85
86
87
88
89
90
91
92
93
94
95
            let read_lock = STRUCT_TAG_MAP.read().unwrap();
            let struc = read_lock.get(name).unwrap();
            let tys = struc.get_tys();            
            
            trace!("layout struct: {}", struc);
            layout_struct(tys, vm)
        }
        // hybrid 
        // - align is the most strict aligned element (from all fix tys and var ty)
        // - size is fixed tys size
        // - layout is fixed tys layout
qinsoon's avatar
qinsoon committed
96
        MuType_::Hybrid(ref fix_tys, ref var_ty) => {
qinsoon's avatar
qinsoon committed
97
98
99
100
            // treat fix_tys as struct
            let mut ret = layout_struct(fix_tys, vm);
            
            // treat var_ty as array (getting its alignment)
101
            let var_align = vm.get_backend_type_info(var_ty.id()).alignment;
qinsoon's avatar
qinsoon committed
102
103
104
105
106
107
108
109
            
            if ret.alignment < var_align {
                ret.alignment = var_align;
            }
            
            ret
        }
        // void
qinsoon's avatar
qinsoon committed
110
        MuType_::Void => BackendTypeInfo{size: 0, alignment: 8, struct_layout: None},
qinsoon's avatar
qinsoon committed
111
        // vector
qinsoon's avatar
qinsoon committed
112
        MuType_::Vector(_, _) => unimplemented!()
qinsoon's avatar
qinsoon committed
113
114
115
    }
}

qinsoon's avatar
qinsoon committed
116
fn layout_struct(tys: &Vec<P<MuType>>, vm: &VM) -> BackendTypeInfo {
qinsoon's avatar
qinsoon committed
117
118
119
120
121
    let mut offsets : Vec<ByteSize> = vec![];
    let mut cur : ByteSize = 0;
    let mut struct_align : ByteSize = 0;
    
    for ty in tys.iter() {
122
        let ty_info = vm.get_backend_type_info(ty.id());
qinsoon's avatar
qinsoon committed
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
        trace!("examining field: {}, {:?}", ty, ty_info);
        
        let align = ty_info.alignment;
        if struct_align < align {
            struct_align = align;
        }
        
        if cur % align != 0 {
            // move cursor to next aligned offset
            cur = (cur / align + 1) * align;
        }
        
        offsets.push(cur);
        trace!("aligned to {}", cur);
        
        cur += ty_info.size;
    }
    
    // if we need padding at the end
    if cur % struct_align != 0 {
        cur = (cur / struct_align + 1) * struct_align;
    }
    
    BackendTypeInfo {
        size: cur,
        alignment: struct_align,
        struct_layout: Some(offsets)
    }
}

qinsoon's avatar
qinsoon committed
153
#[derive(Clone, Debug, RustcEncodable, RustcDecodable)]
qinsoon's avatar
qinsoon committed
154
155
156
157
158
159
pub struct BackendTypeInfo {
    size: ByteSize,
    alignment: ByteSize,
    struct_layout: Option<Vec<ByteSize>>
}

qinsoon's avatar
qinsoon committed
160
#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, RustcEncodable, RustcDecodable)]
qinsoon's avatar
qinsoon committed
161
pub enum RegGroup {GPR, FPR}