| ... | ... | @@ -2,24 +2,198 @@ |
| 2 | 2 | //! This lowers AIR into MIR. |
| 3 | 3 | const std = @import("std"); |
| 4 | 4 | const assert = std.debug.assert; |
| 5 | const mem = std.mem; |
| 6 | const Allocator = mem.Allocator; |
| 5 | 7 | const builtin = @import("builtin"); |
| 6 | 8 | const link = @import("../../link.zig"); |
| 7 | 9 | const Module = @import("../../Module.zig"); |
| 10 | const ErrorMsg = Module.ErrorMsg; |
| 8 | 11 | const Air = @import("../../Air.zig"); |
| 9 | 12 | const Mir = @import("Mir.zig"); |
| 10 | 13 | const Emit = @import("Emit.zig"); |
| 11 | 14 | const Liveness = @import("../../Liveness.zig"); |
| 12 | | const build_options = @import("build_options"); |
| 13 | | |
| 15 | const Type = @import("../../type.zig").Type; |
| 14 | 16 | const GenerateSymbolError = @import("../../codegen.zig").GenerateSymbolError; |
| 15 | 17 | const FnResult = @import("../../codegen.zig").FnResult; |
| 16 | 18 | const DebugInfoOutput = @import("../../codegen.zig").DebugInfoOutput; |
| 17 | 19 | |
| 20 | const build_options = @import("build_options"); |
| 21 | |
| 18 | 22 | const bits = @import("bits.zig"); |
| 19 | 23 | const abi = @import("abi.zig"); |
| 24 | const Register = bits.Register; |
| 20 | 25 | |
| 21 | 26 | const Self = @This(); |
| 22 | 27 | |
| 28 | const InnerError = error{ |
| 29 | OutOfMemory, |
| 30 | CodegenFail, |
| 31 | OutOfRegisters, |
| 32 | }; |
| 33 | |
| 34 | gpa: Allocator, |
| 35 | air: Air, |
| 36 | liveness: Liveness, |
| 37 | bin_file: *link.File, |
| 38 | target: *const std.Target, |
| 39 | mod_fn: *const Module.Fn, |
| 40 | code: *std.ArrayList(u8), |
| 41 | debug_output: DebugInfoOutput, |
| 42 | err_msg: ?*ErrorMsg, |
| 43 | args: []MCValue, |
| 44 | ret_mcv: MCValue, |
| 45 | fn_type: Type, |
| 46 | arg_index: usize, |
| 47 | src_loc: Module.SrcLoc, |
| 48 | stack_align: u32, |
| 49 | |
| 50 | /// MIR Instructions |
| 51 | mir_instructions: std.MultiArrayList(Mir.Inst) = .{}, |
| 52 | /// MIR extra data |
| 53 | mir_extra: std.ArrayListUnmanaged(u32) = .{}, |
| 54 | |
| 55 | /// Byte offset within the source file of the ending curly. |
| 56 | end_di_line: u32, |
| 57 | end_di_column: u32, |
| 58 | |
| 59 | /// The value is an offset into the `Function` `code` from the beginning. |
| 60 | /// To perform the reloc, write 32-bit signed little-endian integer |
| 61 | /// which is a relative jump, based on the address following the reloc. |
| 62 | exitlude_jump_relocs: std.ArrayListUnmanaged(usize) = .{}, |
| 63 | |
| 64 | /// Whenever there is a runtime branch, we push a Branch onto this stack, |
| 65 | /// and pop it off when the runtime branch joins. This provides an "overlay" |
| 66 | /// of the table of mappings from instructions to `MCValue` from within the branch. |
| 67 | /// This way we can modify the `MCValue` for an instruction in different ways |
| 68 | /// within different branches. Special consideration is needed when a branch |
| 69 | /// joins with its parent, to make sure all instructions have the same MCValue |
| 70 | /// across each runtime branch upon joining. |
| 71 | branch_stack: *std.ArrayList(Branch), |
| 72 | |
| 73 | // Key is the block instruction |
| 74 | blocks: std.AutoHashMapUnmanaged(Air.Inst.Index, BlockData) = .{}, |
| 75 | |
| 76 | /// Maps offset to what is stored there. |
| 77 | stack: std.AutoHashMapUnmanaged(u32, StackAllocation) = .{}, |
| 78 | |
| 79 | /// Offset from the stack base, representing the end of the stack frame. |
| 80 | max_end_stack: u32 = 0, |
| 81 | /// Represents the current end stack offset. If there is no existing slot |
| 82 | /// to place a new stack allocation, it goes here, and then bumps `max_end_stack`. |
| 83 | next_stack_offset: u32 = 0, |
| 84 | |
| 85 | /// Debug field, used to find bugs in the compiler. |
| 86 | air_bookkeeping: @TypeOf(air_bookkeeping_init) = air_bookkeeping_init, |
| 87 | |
| 88 | const air_bookkeeping_init = if (std.debug.runtime_safety) @as(usize, 0) else {}; |
| 89 | |
| 90 | const MCValue = union(enum) { |
| 91 | /// No runtime bits. `void` types, empty structs, u0, enums with 1 tag, etc. |
| 92 | /// TODO Look into deleting this tag and using `dead` instead, since every use |
| 93 | /// of MCValue.none should be instead looking at the type and noticing it is 0 bits. |
| 94 | none, |
| 95 | /// Control flow will not allow this value to be observed. |
| 96 | unreach, |
| 97 | /// No more references to this value remain. |
| 98 | dead, |
| 99 | /// The value is undefined. |
| 100 | undef, |
| 101 | /// A pointer-sized integer that fits in a register. |
| 102 | /// If the type is a pointer, this is the pointer address in virtual address space. |
| 103 | immediate: u64, |
| 104 | /// The value is in a target-specific register. |
| 105 | register: Register, |
| 106 | /// The value is in memory at a hard-coded address. |
| 107 | /// If the type is a pointer, it means the pointer address is at this memory location. |
| 108 | memory: u64, |
| 109 | /// The value is one of the stack variables. |
| 110 | /// If the type is a pointer, it means the pointer address is in the stack at this offset. |
| 111 | stack_offset: u32, |
| 112 | /// The value is a pointer to one of the stack variables (payload is stack offset). |
| 113 | ptr_stack_offset: u32, |
| 114 | |
| 115 | fn isMemory(mcv: MCValue) bool { |
| 116 | return switch (mcv) { |
| 117 | .memory, .stack_offset => true, |
| 118 | else => false, |
| 119 | }; |
| 120 | } |
| 121 | |
| 122 | fn isImmediate(mcv: MCValue) bool { |
| 123 | return switch (mcv) { |
| 124 | .immediate => true, |
| 125 | else => false, |
| 126 | }; |
| 127 | } |
| 128 | |
| 129 | fn isMutable(mcv: MCValue) bool { |
| 130 | return switch (mcv) { |
| 131 | .none => unreachable, |
| 132 | .unreach => unreachable, |
| 133 | .dead => unreachable, |
| 134 | |
| 135 | .immediate, |
| 136 | .memory, |
| 137 | .ptr_stack_offset, |
| 138 | .undef, |
| 139 | => false, |
| 140 | |
| 141 | .register, |
| 142 | .stack_offset, |
| 143 | => true, |
| 144 | }; |
| 145 | } |
| 146 | }; |
| 147 | |
| 148 | const Branch = struct { |
| 149 | inst_table: std.AutoArrayHashMapUnmanaged(Air.Inst.Index, MCValue) = .{}, |
| 150 | |
| 151 | fn deinit(self: *Branch, gpa: Allocator) void { |
| 152 | self.inst_table.deinit(gpa); |
| 153 | self.* = undefined; |
| 154 | } |
| 155 | }; |
| 156 | |
| 157 | const StackAllocation = struct { |
| 158 | inst: Air.Inst.Index, |
| 159 | /// TODO do we need size? should be determined by inst.ty.abiSize() |
| 160 | size: u32, |
| 161 | }; |
| 162 | |
| 163 | const BlockData = struct { |
| 164 | relocs: std.ArrayListUnmanaged(Reloc), |
| 165 | /// The first break instruction encounters `null` here and chooses a |
| 166 | /// machine code value for the block result, populating this field. |
| 167 | /// Following break instructions encounter that value and use it for |
| 168 | /// the location to store their block results. |
| 169 | mcv: MCValue, |
| 170 | }; |
| 171 | |
| 172 | const Reloc = union(enum) { |
| 173 | /// The value is an offset into the `Function` `code` from the beginning. |
| 174 | /// To perform the reloc, write 32-bit signed little-endian integer |
| 175 | /// which is a relative jump, based on the address following the reloc. |
| 176 | rel32: usize, |
| 177 | /// A branch in the ARM instruction set |
| 178 | arm_branch: struct { |
| 179 | pos: usize, |
| 180 | cond: @import("../arm/bits.zig").Condition, |
| 181 | }, |
| 182 | }; |
| 183 | |
| 184 | const CallMCValues = struct { |
| 185 | args: []MCValue, |
| 186 | return_value: MCValue, |
| 187 | stack_byte_count: u32, |
| 188 | stack_align: u32, |
| 189 | |
| 190 | fn deinit(self: *CallMCValues, func: *Self) void { |
| 191 | func.gpa.free(self.args); |
| 192 | self.* = undefined; |
| 193 | } |
| 194 | }; |
| 195 | |
| 196 | |
| 23 | 197 | pub fn generate( |
| 24 | 198 | bin_file: *link.File, |
| 25 | 199 | src_loc: Module.SrcLoc, |
| ... | ... | @@ -29,19 +203,110 @@ pub fn generate( |
| 29 | 203 | code: *std.ArrayList(u8), |
| 30 | 204 | debug_output: DebugInfoOutput, |
| 31 | 205 | ) GenerateSymbolError!FnResult { |
| 32 | | _ = bin_file; |
| 33 | | _ = src_loc; |
| 34 | | _ = module_fn; |
| 35 | | _ = air; |
| 36 | | _ = liveness; |
| 37 | | _ = code; |
| 38 | | _ = debug_output; |
| 39 | | |
| 40 | 206 | if (build_options.skip_non_native and builtin.cpu.arch != bin_file.options.target.cpu.arch) { |
| 41 | 207 | @panic("Attempted to compile for architecture that was disabled by build configuration"); |
| 42 | 208 | } |
| 43 | 209 | |
| 44 | 210 | assert(module_fn.owner_decl.has_tv); |
| 211 | const fn_type = module_fn.owner_decl.ty; |
| 212 | |
| 213 | var branch_stack = std.ArrayList(Branch).init(bin_file.allocator); |
| 214 | defer { |
| 215 | assert(branch_stack.items.len == 1); |
| 216 | branch_stack.items[0].deinit(bin_file.allocator); |
| 217 | branch_stack.deinit(); |
| 218 | } |
| 219 | try branch_stack.append(.{}); |
| 220 | |
| 221 | var function = Self{ |
| 222 | .gpa = bin_file.allocator, |
| 223 | .air = air, |
| 224 | .liveness = liveness, |
| 225 | .target = &bin_file.options.target, |
| 226 | .bin_file = bin_file, |
| 227 | .mod_fn = module_fn, |
| 228 | .code = code, |
| 229 | .debug_output = debug_output, |
| 230 | .err_msg = null, |
| 231 | .args = undefined, // populated after `resolveCallingConventionValues` |
| 232 | .ret_mcv = undefined, // populated after `resolveCallingConventionValues` |
| 233 | .fn_type = fn_type, |
| 234 | .arg_index = 0, |
| 235 | .branch_stack = &branch_stack, |
| 236 | .src_loc = src_loc, |
| 237 | .stack_align = undefined, |
| 238 | .end_di_line = module_fn.rbrace_line, |
| 239 | .end_di_column = module_fn.rbrace_column, |
| 240 | }; |
| 241 | defer function.stack.deinit(bin_file.allocator); |
| 242 | defer function.blocks.deinit(bin_file.allocator); |
| 243 | defer function.exitlude_jump_relocs.deinit(bin_file.allocator); |
| 244 | |
| 245 | var call_info = function.resolveCallingConventionValues(fn_type) catch |err| switch (err) { |
| 246 | error.CodegenFail => return FnResult{ .fail = function.err_msg.? }, |
| 247 | error.OutOfRegisters => return FnResult{ |
| 248 | .fail = try ErrorMsg.create(bin_file.allocator, src_loc, "CodeGen ran out of registers. This is a bug in the Zig compiler.", .{}), |
| 249 | }, |
| 250 | else => |e| return e, |
| 251 | }; |
| 252 | defer call_info.deinit(&function); |
| 253 | |
| 254 | function.args = call_info.args; |
| 255 | function.ret_mcv = call_info.return_value; |
| 256 | function.stack_align = call_info.stack_align; |
| 257 | function.max_end_stack = call_info.stack_byte_count; |
| 258 | |
| 259 | function.gen() catch |err| switch (err) { |
| 260 | error.CodegenFail => return FnResult{ .fail = function.err_msg.? }, |
| 261 | error.OutOfRegisters => return FnResult{ |
| 262 | .fail = try ErrorMsg.create(bin_file.allocator, src_loc, "CodeGen ran out of registers. This is a bug in the Zig compiler.", .{}), |
| 263 | }, |
| 264 | else => |e| return e, |
| 265 | }; |
| 266 | |
| 267 | var mir = Mir{ |
| 268 | .instructions = function.mir_instructions.toOwnedSlice(), |
| 269 | .extra = function.mir_extra.toOwnedSlice(bin_file.allocator), |
| 270 | }; |
| 271 | defer mir.deinit(bin_file.allocator); |
| 272 | |
| 273 | var emit = Emit{ |
| 274 | .mir = mir, |
| 275 | .bin_file = bin_file, |
| 276 | .debug_output = debug_output, |
| 277 | .target = &bin_file.options.target, |
| 278 | .src_loc = src_loc, |
| 279 | .code = code, |
| 280 | .prev_di_pc = 0, |
| 281 | .prev_di_line = module_fn.lbrace_line, |
| 282 | .prev_di_column = module_fn.lbrace_column, |
| 283 | }; |
| 284 | defer emit.deinit(); |
| 285 | |
| 286 | emit.emitMir() catch |err| switch (err) { |
| 287 | error.EmitFail => return FnResult{ .fail = emit.err_msg.? }, |
| 288 | else => |e| return e, |
| 289 | }; |
| 290 | |
| 291 | if (function.err_msg) |em| { |
| 292 | return FnResult{ .fail = em }; |
| 293 | } else { |
| 294 | return FnResult{ .appended = {} }; |
| 295 | } |
| 296 | } |
| 297 | |
| 298 | /// Caller must call `CallMCValues.deinit`. |
| 299 | fn resolveCallingConventionValues(self: *Self, fn_ty: Type) !CallMCValues { |
| 300 | _ = self; |
| 301 | _ = fn_ty; |
| 302 | |
| 303 | @panic("TODO implement resolveCallingConventionValues"); |
| 304 | } |
| 305 | |
| 306 | |
| 307 | /// Caller must call `CallMCValues.deinit`. |
| 308 | fn gen(self: *Self) !void { |
| 309 | _ = self; |
| 45 | 310 | |
| 46 | | @panic("TODO implement SPARCv9 codegen"); |
| 311 | @panic("TODO implement gen"); |
| 47 | 312 | } |