authorgravatar for kubkon@jakubkonka.comJakub Konka <kubkon@jakubkonka.com> 2022-02-07 16:02:43+01:00
committergravatar for kubkon@jakubkonka.comJakub Konka <kubkon@jakubkonka.com> 2022-02-07 20:13:29+01:00
logbbd7969c87e03c9312eeafd9940ef8287a3ad689
tree293c89a8c76f0df820dc86a79102bc8ebbb58d7a
parent0a7801236cf3601ff20b3c1c16cd27c1d089157e

stage2,x64: implement genSetStack for memory operand

1 files changed, 38 insertions(+), 5 deletions(-)

src/arch/x86_64/CodeGen.zig+38-5
......@@ -3487,6 +3487,7 @@ fn genSetStackArg(self: *Self, ty: Type, stack_offset: i32, mcv: MCValue) InnerE
34873487}
34883488
34893489fn genSetStack(self: *Self, ty: Type, stack_offset: i32, mcv: MCValue) InnerError!void {
3490 const abi_size = ty.abiSize(self.target.*);
34903491 switch (mcv) {
34913492 .dead => unreachable,
34923493 .ptr_embedded_in_code => unreachable,
......@@ -3510,7 +3511,6 @@ fn genSetStack(self: *Self, ty: Type, stack_offset: i32, mcv: MCValue) InnerErro
35103511 return self.genSetStack(ty, stack_offset, .{ .register = reg });
35113512 },
35123513 .immediate => |x_big| {
3513 const abi_size = ty.abiSize(self.target.*);
35143514 const adj_off = stack_offset + @intCast(i32, abi_size);
35153515 if (adj_off > 128) {
35163516 return self.fail("TODO implement set stack variable with large stack offset", .{});
......@@ -3583,7 +3583,6 @@ fn genSetStack(self: *Self, ty: Type, stack_offset: i32, mcv: MCValue) InnerErro
35833583 if (stack_offset > math.maxInt(i32)) {
35843584 return self.fail("stack offset too large", .{});
35853585 }
3586 const abi_size = ty.abiSize(self.target.*);
35873586 const adj_off = stack_offset + @intCast(i32, abi_size);
35883587 _ = try self.addInst(.{
35893588 .tag = .mov,
......@@ -3600,11 +3599,46 @@ fn genSetStack(self: *Self, ty: Type, stack_offset: i32, mcv: MCValue) InnerErro
36003599 .got_load,
36013600 .direct_load,
36023601 => {
3603 if (ty.abiSize(self.target.*) <= 8) {
3602 if (abi_size <= 8) {
36043603 const reg = try self.copyToTmpRegister(ty, mcv);
36053604 return self.genSetStack(ty, stack_offset, MCValue{ .register = reg });
36063605 }
3607 return self.fail("TODO implement memcpy for setting stack from {}", .{mcv});
3606
3607 try self.register_manager.getReg(.rax, null);
3608 try self.register_manager.getReg(.rcx, null);
3609
3610 self.register_manager.freezeRegs(&.{ .rax, .rcx, .rbp });
3611 defer self.register_manager.unfreezeRegs(&.{ .rax, .rcx, .rbp });
3612
3613 const addr_reg: Register = blk: {
3614 switch (mcv) {
3615 .memory => |addr| {
3616 const reg = try self.copyToTmpRegister(Type.usize, .{ .immediate = addr });
3617 break :blk reg;
3618 },
3619 else => {
3620 return self.fail("TODO implement memcpy for setting stack from {}", .{mcv});
3621 },
3622 }
3623 };
3624
3625 self.register_manager.freezeRegs(&.{addr_reg});
3626 defer self.register_manager.unfreezeRegs(&.{addr_reg});
3627
3628 const regs = try self.register_manager.allocRegs(2, .{ null, null });
3629 const count_reg = regs[0];
3630 const tmp_reg = regs[1];
3631
3632 // TODO allow for abi_size to be u64
3633 try self.genSetReg(Type.u32, count_reg, .{ .immediate = @intCast(u32, abi_size) });
3634
3635 return self.genInlineMemcpy(
3636 -(stack_offset + @intCast(i32, abi_size)),
3637 .rbp,
3638 addr_reg.to64(),
3639 count_reg.to64(),
3640 tmp_reg.to8(),
3641 );
36083642 },
36093643 .ptr_stack_offset => {
36103644 const reg = try self.copyToTmpRegister(ty, mcv);
......@@ -3616,7 +3650,6 @@ fn genSetStack(self: *Self, ty: Type, stack_offset: i32, mcv: MCValue) InnerErro
36163650 return;
36173651 }
36183652
3619 const abi_size = ty.abiSize(self.target.*);
36203653 if (abi_size <= 8) {
36213654 const reg = try self.copyToTmpRegister(ty, mcv);
36223655 return self.genSetStack(ty, stack_offset, MCValue{ .register = reg });