Jia Liu | b22310f | 2012-02-18 12:03:15 +0000 | [diff] [blame] | 1 | //===-- X86FrameLowering.cpp - X86 Frame Information ----------------------===// |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
Anton Korobeynikov | 2f93128 | 2011-01-10 12:39:04 +0000 | [diff] [blame] | 10 | // This file contains the X86 implementation of TargetFrameLowering class. |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
Anton Korobeynikov | 2f93128 | 2011-01-10 12:39:04 +0000 | [diff] [blame] | 14 | #include "X86FrameLowering.h" |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 15 | #include "X86InstrBuilder.h" |
| 16 | #include "X86InstrInfo.h" |
| 17 | #include "X86MachineFunctionInfo.h" |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 18 | #include "X86Subtarget.h" |
Anton Korobeynikov | 14ee344 | 2010-11-18 23:25:52 +0000 | [diff] [blame] | 19 | #include "X86TargetMachine.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 20 | #include "llvm/ADT/SmallSet.h" |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 21 | #include "llvm/CodeGen/MachineFrameInfo.h" |
| 22 | #include "llvm/CodeGen/MachineFunction.h" |
| 23 | #include "llvm/CodeGen/MachineInstrBuilder.h" |
| 24 | #include "llvm/CodeGen/MachineModuleInfo.h" |
| 25 | #include "llvm/CodeGen/MachineRegisterInfo.h" |
Chandler Carruth | 9fb823b | 2013-01-02 11:36:10 +0000 | [diff] [blame] | 26 | #include "llvm/IR/DataLayout.h" |
| 27 | #include "llvm/IR/Function.h" |
Rafael Espindola | a01cdb0 | 2011-04-15 15:11:06 +0000 | [diff] [blame] | 28 | #include "llvm/MC/MCAsmInfo.h" |
Bill Wendling | b6adf46 | 2011-07-07 00:54:13 +0000 | [diff] [blame] | 29 | #include "llvm/MC/MCSymbol.h" |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 30 | #include "llvm/Support/CommandLine.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 31 | #include "llvm/Target/TargetOptions.h" |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 32 | #include "llvm/Support/Debug.h" |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 33 | |
| 34 | using namespace llvm; |
| 35 | |
| 36 | // FIXME: completely move here. |
| 37 | extern cl::opt<bool> ForceStackAlign; |
| 38 | |
Anton Korobeynikov | 2f93128 | 2011-01-10 12:39:04 +0000 | [diff] [blame] | 39 | bool X86FrameLowering::hasReservedCallFrame(const MachineFunction &MF) const { |
Anton Korobeynikov | 0eecf5d | 2010-11-18 21:19:35 +0000 | [diff] [blame] | 40 | return !MF.getFrameInfo()->hasVarSizedObjects(); |
| 41 | } |
| 42 | |
| 43 | /// hasFP - Return true if the specified function should have a dedicated frame |
| 44 | /// pointer register. This is true if the function has variable sized allocas |
| 45 | /// or if frame pointer elimination is disabled. |
Anton Korobeynikov | 2f93128 | 2011-01-10 12:39:04 +0000 | [diff] [blame] | 46 | bool X86FrameLowering::hasFP(const MachineFunction &MF) const { |
Anton Korobeynikov | 0eecf5d | 2010-11-18 21:19:35 +0000 | [diff] [blame] | 47 | const MachineFrameInfo *MFI = MF.getFrameInfo(); |
| 48 | const MachineModuleInfo &MMI = MF.getMMI(); |
Eric Christopher | 11b05cc | 2014-06-05 00:09:05 +0000 | [diff] [blame] | 49 | const TargetRegisterInfo *RegInfo = MF.getTarget().getRegisterInfo(); |
Anton Korobeynikov | 0eecf5d | 2010-11-18 21:19:35 +0000 | [diff] [blame] | 50 | |
Nick Lewycky | 50f02cb | 2011-12-02 22:16:29 +0000 | [diff] [blame] | 51 | return (MF.getTarget().Options.DisableFramePointerElim(MF) || |
Chad Rosier | 20b79dc | 2012-05-23 23:45:10 +0000 | [diff] [blame] | 52 | RegInfo->needsStackRealignment(MF) || |
Anton Korobeynikov | 0eecf5d | 2010-11-18 21:19:35 +0000 | [diff] [blame] | 53 | MFI->hasVarSizedObjects() || |
Reid Kleckner | ee08897 | 2013-12-10 18:27:32 +0000 | [diff] [blame] | 54 | MFI->isFrameAddressTaken() || MFI->hasInlineAsmWithSPAdjust() || |
Anton Korobeynikov | 0eecf5d | 2010-11-18 21:19:35 +0000 | [diff] [blame] | 55 | MF.getInfo<X86MachineFunctionInfo>()->getForceFramePointer() || |
Jakob Stoklund Olesen | 321d41a | 2012-06-22 03:04:27 +0000 | [diff] [blame] | 56 | MMI.callsUnwindInit() || MMI.callsEHReturn()); |
Anton Korobeynikov | 0eecf5d | 2010-11-18 21:19:35 +0000 | [diff] [blame] | 57 | } |
| 58 | |
Eli Bendersky | 8da8716 | 2013-02-21 20:05:00 +0000 | [diff] [blame] | 59 | static unsigned getSUBriOpcode(unsigned IsLP64, int64_t Imm) { |
| 60 | if (IsLP64) { |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 61 | if (isInt<8>(Imm)) |
| 62 | return X86::SUB64ri8; |
| 63 | return X86::SUB64ri32; |
| 64 | } else { |
| 65 | if (isInt<8>(Imm)) |
| 66 | return X86::SUB32ri8; |
| 67 | return X86::SUB32ri; |
| 68 | } |
| 69 | } |
| 70 | |
Eli Bendersky | ef4558a | 2013-02-06 20:43:57 +0000 | [diff] [blame] | 71 | static unsigned getADDriOpcode(unsigned IsLP64, int64_t Imm) { |
| 72 | if (IsLP64) { |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 73 | if (isInt<8>(Imm)) |
| 74 | return X86::ADD64ri8; |
| 75 | return X86::ADD64ri32; |
| 76 | } else { |
| 77 | if (isInt<8>(Imm)) |
| 78 | return X86::ADD32ri8; |
| 79 | return X86::ADD32ri; |
| 80 | } |
| 81 | } |
| 82 | |
Eli Bendersky | ef4558a | 2013-02-06 20:43:57 +0000 | [diff] [blame] | 83 | static unsigned getLEArOpcode(unsigned IsLP64) { |
| 84 | return IsLP64 ? X86::LEA64r : X86::LEA32r; |
Evan Cheng | 1b81fdd | 2012-02-07 22:50:41 +0000 | [diff] [blame] | 85 | } |
| 86 | |
Evan Cheng | 65089fc | 2011-01-03 22:53:22 +0000 | [diff] [blame] | 87 | /// findDeadCallerSavedReg - Return a caller-saved register that isn't live |
| 88 | /// when it reaches the "return" instruction. We can then pop a stack object |
| 89 | /// to this register without worry about clobbering it. |
| 90 | static unsigned findDeadCallerSavedReg(MachineBasicBlock &MBB, |
| 91 | MachineBasicBlock::iterator &MBBI, |
| 92 | const TargetRegisterInfo &TRI, |
| 93 | bool Is64Bit) { |
| 94 | const MachineFunction *MF = MBB.getParent(); |
| 95 | const Function *F = MF->getFunction(); |
| 96 | if (!F || MF->getMMI().callsEHReturn()) |
| 97 | return 0; |
| 98 | |
Craig Topper | 1d32658 | 2012-03-04 10:43:23 +0000 | [diff] [blame] | 99 | static const uint16_t CallerSavedRegs32Bit[] = { |
Andrew Trick | 210bf83 | 2011-08-12 00:49:19 +0000 | [diff] [blame] | 100 | X86::EAX, X86::EDX, X86::ECX, 0 |
Evan Cheng | 65089fc | 2011-01-03 22:53:22 +0000 | [diff] [blame] | 101 | }; |
| 102 | |
Craig Topper | 1d32658 | 2012-03-04 10:43:23 +0000 | [diff] [blame] | 103 | static const uint16_t CallerSavedRegs64Bit[] = { |
Evan Cheng | 65089fc | 2011-01-03 22:53:22 +0000 | [diff] [blame] | 104 | X86::RAX, X86::RDX, X86::RCX, X86::RSI, X86::RDI, |
Andrew Trick | 210bf83 | 2011-08-12 00:49:19 +0000 | [diff] [blame] | 105 | X86::R8, X86::R9, X86::R10, X86::R11, 0 |
Evan Cheng | 65089fc | 2011-01-03 22:53:22 +0000 | [diff] [blame] | 106 | }; |
| 107 | |
| 108 | unsigned Opc = MBBI->getOpcode(); |
| 109 | switch (Opc) { |
| 110 | default: return 0; |
David Woodhouse | 79dd505 | 2014-01-08 12:58:07 +0000 | [diff] [blame] | 111 | case X86::RETL: |
| 112 | case X86::RETQ: |
David Woodhouse | 4e033b0 | 2014-01-13 14:05:59 +0000 | [diff] [blame] | 113 | case X86::RETIL: |
| 114 | case X86::RETIQ: |
Evan Cheng | 65089fc | 2011-01-03 22:53:22 +0000 | [diff] [blame] | 115 | case X86::TCRETURNdi: |
| 116 | case X86::TCRETURNri: |
| 117 | case X86::TCRETURNmi: |
| 118 | case X86::TCRETURNdi64: |
| 119 | case X86::TCRETURNri64: |
| 120 | case X86::TCRETURNmi64: |
| 121 | case X86::EH_RETURN: |
| 122 | case X86::EH_RETURN64: { |
Craig Topper | 1d32658 | 2012-03-04 10:43:23 +0000 | [diff] [blame] | 123 | SmallSet<uint16_t, 8> Uses; |
Evan Cheng | 65089fc | 2011-01-03 22:53:22 +0000 | [diff] [blame] | 124 | for (unsigned i = 0, e = MBBI->getNumOperands(); i != e; ++i) { |
| 125 | MachineOperand &MO = MBBI->getOperand(i); |
| 126 | if (!MO.isReg() || MO.isDef()) |
| 127 | continue; |
| 128 | unsigned Reg = MO.getReg(); |
| 129 | if (!Reg) |
| 130 | continue; |
Jakob Stoklund Olesen | 54038d7 | 2012-06-01 23:28:30 +0000 | [diff] [blame] | 131 | for (MCRegAliasIterator AI(Reg, &TRI, true); AI.isValid(); ++AI) |
| 132 | Uses.insert(*AI); |
Evan Cheng | 65089fc | 2011-01-03 22:53:22 +0000 | [diff] [blame] | 133 | } |
| 134 | |
Craig Topper | 1d32658 | 2012-03-04 10:43:23 +0000 | [diff] [blame] | 135 | const uint16_t *CS = Is64Bit ? CallerSavedRegs64Bit : CallerSavedRegs32Bit; |
Evan Cheng | 65089fc | 2011-01-03 22:53:22 +0000 | [diff] [blame] | 136 | for (; *CS; ++CS) |
| 137 | if (!Uses.count(*CS)) |
| 138 | return *CS; |
| 139 | } |
| 140 | } |
| 141 | |
| 142 | return 0; |
| 143 | } |
| 144 | |
| 145 | |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 146 | /// emitSPUpdate - Emit a series of instructions to increment / decrement the |
| 147 | /// stack pointer by a constant value. |
| 148 | static |
| 149 | void emitSPUpdate(MachineBasicBlock &MBB, MachineBasicBlock::iterator &MBBI, |
Evan Cheng | 65089fc | 2011-01-03 22:53:22 +0000 | [diff] [blame] | 150 | unsigned StackPtr, int64_t NumBytes, |
Eli Bendersky | 44a40ca | 2013-02-05 21:53:29 +0000 | [diff] [blame] | 151 | bool Is64Bit, bool IsLP64, bool UseLEA, |
Eric Christopher | f4fba5c | 2012-10-03 08:10:01 +0000 | [diff] [blame] | 152 | const TargetInstrInfo &TII, const TargetRegisterInfo &TRI) { |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 153 | bool isSub = NumBytes < 0; |
| 154 | uint64_t Offset = isSub ? -NumBytes : NumBytes; |
Evan Cheng | 1b81fdd | 2012-02-07 22:50:41 +0000 | [diff] [blame] | 155 | unsigned Opc; |
| 156 | if (UseLEA) |
Eli Bendersky | ef4558a | 2013-02-06 20:43:57 +0000 | [diff] [blame] | 157 | Opc = getLEArOpcode(IsLP64); |
Evan Cheng | 1b81fdd | 2012-02-07 22:50:41 +0000 | [diff] [blame] | 158 | else |
| 159 | Opc = isSub |
Eli Bendersky | 44a40ca | 2013-02-05 21:53:29 +0000 | [diff] [blame] | 160 | ? getSUBriOpcode(IsLP64, Offset) |
| 161 | : getADDriOpcode(IsLP64, Offset); |
Evan Cheng | 1b81fdd | 2012-02-07 22:50:41 +0000 | [diff] [blame] | 162 | |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 163 | uint64_t Chunk = (1LL << 31) - 1; |
Eric Christopher | f4fba5c | 2012-10-03 08:10:01 +0000 | [diff] [blame] | 164 | DebugLoc DL = MBB.findDebugLoc(MBBI); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 165 | |
| 166 | while (Offset) { |
| 167 | uint64_t ThisVal = (Offset > Chunk) ? Chunk : Offset; |
Evan Cheng | 65089fc | 2011-01-03 22:53:22 +0000 | [diff] [blame] | 168 | if (ThisVal == (Is64Bit ? 8 : 4)) { |
| 169 | // Use push / pop instead. |
| 170 | unsigned Reg = isSub |
Dale Johannesen | e45a238 | 2011-01-04 19:31:24 +0000 | [diff] [blame] | 171 | ? (unsigned)(Is64Bit ? X86::RAX : X86::EAX) |
Evan Cheng | 65089fc | 2011-01-03 22:53:22 +0000 | [diff] [blame] | 172 | : findDeadCallerSavedReg(MBB, MBBI, TRI, Is64Bit); |
| 173 | if (Reg) { |
| 174 | Opc = isSub |
| 175 | ? (Is64Bit ? X86::PUSH64r : X86::PUSH32r) |
| 176 | : (Is64Bit ? X86::POP64r : X86::POP32r); |
Charles Davis | 7ed40cb | 2011-06-12 01:45:54 +0000 | [diff] [blame] | 177 | MachineInstr *MI = BuildMI(MBB, MBBI, DL, TII.get(Opc)) |
Evan Cheng | 65089fc | 2011-01-03 22:53:22 +0000 | [diff] [blame] | 178 | .addReg(Reg, getDefRegState(!isSub) | getUndefRegState(isSub)); |
Charles Davis | 7ed40cb | 2011-06-12 01:45:54 +0000 | [diff] [blame] | 179 | if (isSub) |
| 180 | MI->setFlag(MachineInstr::FrameSetup); |
Evan Cheng | 65089fc | 2011-01-03 22:53:22 +0000 | [diff] [blame] | 181 | Offset -= ThisVal; |
| 182 | continue; |
| 183 | } |
| 184 | } |
| 185 | |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 186 | MachineInstr *MI = nullptr; |
Evan Cheng | 1b81fdd | 2012-02-07 22:50:41 +0000 | [diff] [blame] | 187 | |
| 188 | if (UseLEA) { |
| 189 | MI = addRegOffset(BuildMI(MBB, MBBI, DL, TII.get(Opc), StackPtr), |
| 190 | StackPtr, false, isSub ? -ThisVal : ThisVal); |
| 191 | } else { |
| 192 | MI = BuildMI(MBB, MBBI, DL, TII.get(Opc), StackPtr) |
| 193 | .addReg(StackPtr) |
| 194 | .addImm(ThisVal); |
| 195 | MI->getOperand(3).setIsDead(); // The EFLAGS implicit def is dead. |
| 196 | } |
| 197 | |
Charles Davis | 7ed40cb | 2011-06-12 01:45:54 +0000 | [diff] [blame] | 198 | if (isSub) |
| 199 | MI->setFlag(MachineInstr::FrameSetup); |
Evan Cheng | 1b81fdd | 2012-02-07 22:50:41 +0000 | [diff] [blame] | 200 | |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 201 | Offset -= ThisVal; |
| 202 | } |
| 203 | } |
| 204 | |
| 205 | /// mergeSPUpdatesUp - Merge two stack-manipulating instructions upper iterator. |
| 206 | static |
| 207 | void mergeSPUpdatesUp(MachineBasicBlock &MBB, MachineBasicBlock::iterator &MBBI, |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 208 | unsigned StackPtr, uint64_t *NumBytes = nullptr) { |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 209 | if (MBBI == MBB.begin()) return; |
| 210 | |
Benjamin Kramer | b6d0bd4 | 2014-03-02 12:27:27 +0000 | [diff] [blame] | 211 | MachineBasicBlock::iterator PI = std::prev(MBBI); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 212 | unsigned Opc = PI->getOpcode(); |
| 213 | if ((Opc == X86::ADD64ri32 || Opc == X86::ADD64ri8 || |
Evan Cheng | 1b81fdd | 2012-02-07 22:50:41 +0000 | [diff] [blame] | 214 | Opc == X86::ADD32ri || Opc == X86::ADD32ri8 || |
| 215 | Opc == X86::LEA32r || Opc == X86::LEA64_32r) && |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 216 | PI->getOperand(0).getReg() == StackPtr) { |
| 217 | if (NumBytes) |
| 218 | *NumBytes += PI->getOperand(2).getImm(); |
| 219 | MBB.erase(PI); |
| 220 | } else if ((Opc == X86::SUB64ri32 || Opc == X86::SUB64ri8 || |
| 221 | Opc == X86::SUB32ri || Opc == X86::SUB32ri8) && |
| 222 | PI->getOperand(0).getReg() == StackPtr) { |
| 223 | if (NumBytes) |
| 224 | *NumBytes -= PI->getOperand(2).getImm(); |
| 225 | MBB.erase(PI); |
| 226 | } |
| 227 | } |
| 228 | |
Eric Christopher | 4237bf1 | 2014-04-29 00:16:33 +0000 | [diff] [blame] | 229 | /// mergeSPUpdatesDown - Merge two stack-manipulating instructions lower |
| 230 | /// iterator. |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 231 | static |
| 232 | void mergeSPUpdatesDown(MachineBasicBlock &MBB, |
| 233 | MachineBasicBlock::iterator &MBBI, |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 234 | unsigned StackPtr, uint64_t *NumBytes = nullptr) { |
Sanjoy Das | f60485c | 2011-12-01 19:15:08 +0000 | [diff] [blame] | 235 | // FIXME: THIS ISN'T RUN!!! |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 236 | return; |
| 237 | |
| 238 | if (MBBI == MBB.end()) return; |
| 239 | |
Benjamin Kramer | b6d0bd4 | 2014-03-02 12:27:27 +0000 | [diff] [blame] | 240 | MachineBasicBlock::iterator NI = std::next(MBBI); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 241 | if (NI == MBB.end()) return; |
| 242 | |
| 243 | unsigned Opc = NI->getOpcode(); |
| 244 | if ((Opc == X86::ADD64ri32 || Opc == X86::ADD64ri8 || |
| 245 | Opc == X86::ADD32ri || Opc == X86::ADD32ri8) && |
| 246 | NI->getOperand(0).getReg() == StackPtr) { |
| 247 | if (NumBytes) |
| 248 | *NumBytes -= NI->getOperand(2).getImm(); |
| 249 | MBB.erase(NI); |
| 250 | MBBI = NI; |
| 251 | } else if ((Opc == X86::SUB64ri32 || Opc == X86::SUB64ri8 || |
| 252 | Opc == X86::SUB32ri || Opc == X86::SUB32ri8) && |
| 253 | NI->getOperand(0).getReg() == StackPtr) { |
| 254 | if (NumBytes) |
| 255 | *NumBytes += NI->getOperand(2).getImm(); |
| 256 | MBB.erase(NI); |
| 257 | MBBI = NI; |
| 258 | } |
| 259 | } |
| 260 | |
| 261 | /// mergeSPUpdates - Checks the instruction before/after the passed |
Eric Christopher | 4237bf1 | 2014-04-29 00:16:33 +0000 | [diff] [blame] | 262 | /// instruction. If it is an ADD/SUB/LEA instruction it is deleted argument and |
| 263 | /// the stack adjustment is returned as a positive value for ADD/LEA and a |
| 264 | /// negative for SUB. |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 265 | static int mergeSPUpdates(MachineBasicBlock &MBB, |
Eric Christopher | 4237bf1 | 2014-04-29 00:16:33 +0000 | [diff] [blame] | 266 | MachineBasicBlock::iterator &MBBI, unsigned StackPtr, |
| 267 | bool doMergeWithPrevious) { |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 268 | if ((doMergeWithPrevious && MBBI == MBB.begin()) || |
| 269 | (!doMergeWithPrevious && MBBI == MBB.end())) |
| 270 | return 0; |
| 271 | |
Benjamin Kramer | b6d0bd4 | 2014-03-02 12:27:27 +0000 | [diff] [blame] | 272 | MachineBasicBlock::iterator PI = doMergeWithPrevious ? std::prev(MBBI) : MBBI; |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 273 | MachineBasicBlock::iterator NI = doMergeWithPrevious ? nullptr |
| 274 | : std::next(MBBI); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 275 | unsigned Opc = PI->getOpcode(); |
| 276 | int Offset = 0; |
| 277 | |
| 278 | if ((Opc == X86::ADD64ri32 || Opc == X86::ADD64ri8 || |
Evan Cheng | 1b81fdd | 2012-02-07 22:50:41 +0000 | [diff] [blame] | 279 | Opc == X86::ADD32ri || Opc == X86::ADD32ri8 || |
| 280 | Opc == X86::LEA32r || Opc == X86::LEA64_32r) && |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 281 | PI->getOperand(0).getReg() == StackPtr){ |
| 282 | Offset += PI->getOperand(2).getImm(); |
| 283 | MBB.erase(PI); |
| 284 | if (!doMergeWithPrevious) MBBI = NI; |
| 285 | } else if ((Opc == X86::SUB64ri32 || Opc == X86::SUB64ri8 || |
| 286 | Opc == X86::SUB32ri || Opc == X86::SUB32ri8) && |
| 287 | PI->getOperand(0).getReg() == StackPtr) { |
| 288 | Offset -= PI->getOperand(2).getImm(); |
| 289 | MBB.erase(PI); |
| 290 | if (!doMergeWithPrevious) MBBI = NI; |
| 291 | } |
| 292 | |
| 293 | return Offset; |
| 294 | } |
| 295 | |
| 296 | static bool isEAXLiveIn(MachineFunction &MF) { |
| 297 | for (MachineRegisterInfo::livein_iterator II = MF.getRegInfo().livein_begin(), |
| 298 | EE = MF.getRegInfo().livein_end(); II != EE; ++II) { |
| 299 | unsigned Reg = II->first; |
| 300 | |
| 301 | if (Reg == X86::EAX || Reg == X86::AX || |
| 302 | Reg == X86::AH || Reg == X86::AL) |
| 303 | return true; |
| 304 | } |
| 305 | |
| 306 | return false; |
| 307 | } |
| 308 | |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 309 | void |
| 310 | X86FrameLowering::emitCalleeSavedFrameMoves(MachineBasicBlock &MBB, |
| 311 | MachineBasicBlock::iterator MBBI, |
| 312 | DebugLoc DL) const { |
Rafael Espindola | b1f25f1 | 2014-03-07 06:08:31 +0000 | [diff] [blame] | 313 | MachineFunction &MF = *MBB.getParent(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 314 | MachineFrameInfo *MFI = MF.getFrameInfo(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 315 | MachineModuleInfo &MMI = MF.getMMI(); |
Bill Wendling | bc07a89 | 2013-06-18 07:20:20 +0000 | [diff] [blame] | 316 | const MCRegisterInfo *MRI = MMI.getContext().getRegisterInfo(); |
Eric Christopher | 11b05cc | 2014-06-05 00:09:05 +0000 | [diff] [blame] | 317 | const TargetInstrInfo &TII = *MF.getTarget().getInstrInfo(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 318 | |
| 319 | // Add callee saved registers to move list. |
| 320 | const std::vector<CalleeSavedInfo> &CSI = MFI->getCalleeSavedInfo(); |
| 321 | if (CSI.empty()) return; |
| 322 | |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 323 | // Calculate offsets. |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 324 | for (std::vector<CalleeSavedInfo>::const_iterator |
| 325 | I = CSI.begin(), E = CSI.end(); I != E; ++I) { |
| 326 | int64_t Offset = MFI->getObjectOffset(I->getFrameIdx()); |
| 327 | unsigned Reg = I->getReg(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 328 | |
Bill Wendling | bc07a89 | 2013-06-18 07:20:20 +0000 | [diff] [blame] | 329 | unsigned DwarfReg = MRI->getDwarfRegNum(Reg, true); |
Rafael Espindola | b1f25f1 | 2014-03-07 06:08:31 +0000 | [diff] [blame] | 330 | unsigned CFIIndex = |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 331 | MMI.addFrameInst(MCCFIInstruction::createOffset(nullptr, DwarfReg, |
| 332 | Offset)); |
Eric Christopher | 612bb69 | 2014-04-29 00:16:46 +0000 | [diff] [blame] | 333 | BuildMI(MBB, MBBI, DL, TII.get(TargetOpcode::CFI_INSTRUCTION)) |
| 334 | .addCFIIndex(CFIIndex); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 335 | } |
| 336 | } |
| 337 | |
Nadav Rotem | 1bef5a0 | 2012-12-23 07:30:09 +0000 | [diff] [blame] | 338 | /// usesTheStack - This function checks if any of the users of EFLAGS |
Nadav Rotem | d5aae98 | 2012-12-21 23:48:49 +0000 | [diff] [blame] | 339 | /// copies the EFLAGS. We know that the code that lowers COPY of EFLAGS has |
| 340 | /// to use the stack, and if we don't adjust the stack we clobber the first |
| 341 | /// frame index. |
Nadav Rotem | 1bef5a0 | 2012-12-23 07:30:09 +0000 | [diff] [blame] | 342 | /// See X86InstrInfo::copyPhysReg. |
Bill Wendling | 2851907 | 2013-08-15 18:46:14 +0000 | [diff] [blame] | 343 | static bool usesTheStack(const MachineFunction &MF) { |
| 344 | const MachineRegisterInfo &MRI = MF.getRegInfo(); |
Nadav Rotem | d5aae98 | 2012-12-21 23:48:49 +0000 | [diff] [blame] | 345 | |
Owen Anderson | 16c6bf4 | 2014-03-13 23:12:04 +0000 | [diff] [blame] | 346 | for (MachineRegisterInfo::reg_instr_iterator |
| 347 | ri = MRI.reg_instr_begin(X86::EFLAGS), re = MRI.reg_instr_end(); |
| 348 | ri != re; ++ri) |
Nadav Rotem | d5aae98 | 2012-12-21 23:48:49 +0000 | [diff] [blame] | 349 | if (ri->isCopy()) |
| 350 | return true; |
| 351 | |
| 352 | return false; |
| 353 | } |
| 354 | |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 355 | /// emitPrologue - Push callee-saved registers onto the stack, which |
| 356 | /// automatically adjust the stack pointer. Adjust the stack pointer to allocate |
| 357 | /// space for local variables. Also emit labels used by the exception handler to |
| 358 | /// generate the exception handling frames. |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 359 | |
| 360 | /* |
| 361 | Here's a gist of what gets emitted: |
| 362 | |
| 363 | ; Establish frame pointer, if needed |
| 364 | [if needs FP] |
| 365 | push %rbp |
| 366 | .cfi_def_cfa_offset 16 |
| 367 | .cfi_offset %rbp, -16 |
| 368 | .seh_pushreg %rpb |
| 369 | mov %rsp, %rbp |
| 370 | .cfi_def_cfa_register %rbp |
| 371 | |
| 372 | ; Spill general-purpose registers |
| 373 | [for all callee-saved GPRs] |
| 374 | pushq %<reg> |
| 375 | [if not needs FP] |
| 376 | .cfi_def_cfa_offset (offset from RETADDR) |
| 377 | .seh_pushreg %<reg> |
| 378 | |
| 379 | ; If the required stack alignment > default stack alignment |
| 380 | ; rsp needs to be re-aligned. This creates a "re-alignment gap" |
| 381 | ; of unknown size in the stack frame. |
| 382 | [if stack needs re-alignment] |
| 383 | and $MASK, %rsp |
| 384 | |
| 385 | ; Allocate space for locals |
| 386 | [if target is Windows and allocated space > 4096 bytes] |
| 387 | ; Windows needs special care for allocations larger |
| 388 | ; than one page. |
| 389 | mov $NNN, %rax |
| 390 | call ___chkstk_ms/___chkstk |
| 391 | sub %rax, %rsp |
| 392 | [else] |
| 393 | sub $NNN, %rsp |
| 394 | |
| 395 | [if needs FP] |
| 396 | .seh_stackalloc (size of XMM spill slots) |
| 397 | .seh_setframe %rbp, SEHFrameOffset ; = size of all spill slots |
| 398 | [else] |
| 399 | .seh_stackalloc NNN |
| 400 | |
| 401 | ; Spill XMMs |
| 402 | ; Note, that while only Windows 64 ABI specifies XMMs as callee-preserved, |
| 403 | ; they may get spilled on any platform, if the current function |
| 404 | ; calls @llvm.eh.unwind.init |
| 405 | [if needs FP] |
| 406 | [for all callee-saved XMM registers] |
| 407 | movaps %<xmm reg>, -MMM(%rbp) |
| 408 | [for all callee-saved XMM registers] |
| 409 | .seh_savexmm %<xmm reg>, (-MMM + SEHFrameOffset) |
| 410 | ; i.e. the offset relative to (%rbp - SEHFrameOffset) |
| 411 | [else] |
| 412 | [for all callee-saved XMM registers] |
| 413 | movaps %<xmm reg>, KKK(%rsp) |
| 414 | [for all callee-saved XMM registers] |
| 415 | .seh_savexmm %<xmm reg>, KKK |
| 416 | |
| 417 | .seh_endprologue |
| 418 | |
| 419 | [if needs base pointer] |
| 420 | mov %rsp, %rbx |
| 421 | |
| 422 | ; Emit CFI info |
| 423 | [if needs FP] |
| 424 | [for all callee-saved registers] |
| 425 | .cfi_offset %<reg>, (offset from %rbp) |
| 426 | [else] |
| 427 | .cfi_def_cfa_offset (offset from RETADDR) |
| 428 | [for all callee-saved registers] |
| 429 | .cfi_offset %<reg>, (offset from %rsp) |
| 430 | |
| 431 | Notes: |
| 432 | - .seh directives are emitted only for Windows 64 ABI |
| 433 | - .cfi directives are emitted for all other ABIs |
| 434 | - for 32-bit code, substitute %e?? registers for %r?? |
| 435 | */ |
| 436 | |
Anton Korobeynikov | 2f93128 | 2011-01-10 12:39:04 +0000 | [diff] [blame] | 437 | void X86FrameLowering::emitPrologue(MachineFunction &MF) const { |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 438 | MachineBasicBlock &MBB = MF.front(); // Prologue goes in entry BB. |
| 439 | MachineBasicBlock::iterator MBBI = MBB.begin(); |
| 440 | MachineFrameInfo *MFI = MF.getFrameInfo(); |
| 441 | const Function *Fn = MF.getFunction(); |
Eric Christopher | 11b05cc | 2014-06-05 00:09:05 +0000 | [diff] [blame] | 442 | const X86RegisterInfo *RegInfo = |
| 443 | static_cast<const X86RegisterInfo *>(MF.getTarget().getRegisterInfo()); |
| 444 | const TargetInstrInfo &TII = *MF.getTarget().getInstrInfo(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 445 | MachineModuleInfo &MMI = MF.getMMI(); |
| 446 | X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 447 | uint64_t MaxAlign = MFI->getMaxAlignment(); // Desired stack alignment. |
| 448 | uint64_t StackSize = MFI->getStackSize(); // Number of bytes to allocate. |
Anton Korobeynikov | 0eecf5d | 2010-11-18 21:19:35 +0000 | [diff] [blame] | 449 | bool HasFP = hasFP(MF); |
Eric Christopher | f438164 | 2014-06-05 22:00:31 +0000 | [diff] [blame] | 450 | const X86Subtarget &STI = MF.getTarget().getSubtarget<X86Subtarget>(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 451 | bool Is64Bit = STI.is64Bit(); |
Eli Bendersky | 44a40ca | 2013-02-05 21:53:29 +0000 | [diff] [blame] | 452 | bool IsLP64 = STI.isTarget64BitLP64(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 453 | bool IsWin64 = STI.isTargetWin64(); |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 454 | bool IsSEH = |
| 455 | MF.getTarget().getMCAsmInfo()->getExceptionHandlingType() == |
| 456 | ExceptionHandling::Win64; // Not necessarily synonymous with IsWin64. |
| 457 | bool NeedsWin64SEH = IsSEH && Fn->needsUnwindTableEntry(); |
| 458 | bool NeedsDwarfCFI = |
| 459 | !IsSEH && (MMI.hasDebugInfo() || Fn->needsUnwindTableEntry()); |
Evan Cheng | 1b81fdd | 2012-02-07 22:50:41 +0000 | [diff] [blame] | 460 | bool UseLEA = STI.useLeaForSP(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 461 | unsigned StackAlign = getStackAlignment(); |
| 462 | unsigned SlotSize = RegInfo->getSlotSize(); |
| 463 | unsigned FramePtr = RegInfo->getFrameRegister(MF); |
| 464 | unsigned StackPtr = RegInfo->getStackRegister(); |
Chad Rosier | bdb08ac | 2012-07-10 17:45:53 +0000 | [diff] [blame] | 465 | unsigned BasePtr = RegInfo->getBaseRegister(); |
Bill Wendling | f27e331 | 2013-09-10 00:20:27 +0000 | [diff] [blame] | 466 | DebugLoc DL; |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 467 | |
| 468 | // If we're forcing a stack realignment we can't rely on just the frame |
| 469 | // info, we need to know the ABI stack alignment as well in case we |
| 470 | // have a call out. Otherwise just make sure we have some alignment - we'll |
| 471 | // go with the minimum SlotSize. |
| 472 | if (ForceStackAlign) { |
| 473 | if (MFI->hasCalls()) |
| 474 | MaxAlign = (StackAlign > MaxAlign) ? StackAlign : MaxAlign; |
| 475 | else if (MaxAlign < SlotSize) |
| 476 | MaxAlign = SlotSize; |
| 477 | } |
| 478 | |
| 479 | // Add RETADDR move area to callee saved frame size. |
| 480 | int TailCallReturnAddrDelta = X86FI->getTCReturnAddrDelta(); |
| 481 | if (TailCallReturnAddrDelta < 0) |
| 482 | X86FI->setCalleeSavedFrameSize( |
| 483 | X86FI->getCalleeSavedFrameSize() - TailCallReturnAddrDelta); |
| 484 | |
| 485 | // If this is x86-64 and the Red Zone is not disabled, if we are a leaf |
| 486 | // function, and use up to 128 bytes of stack space, don't have a frame |
| 487 | // pointer, calls, or dynamic alloca then we do not need to adjust the |
Nadav Rotem | d5aae98 | 2012-12-21 23:48:49 +0000 | [diff] [blame] | 488 | // stack pointer (we fit in the Red Zone). We also check that we don't |
| 489 | // push and pop from the stack. |
Bill Wendling | 698e84f | 2012-12-30 10:32:01 +0000 | [diff] [blame] | 490 | if (Is64Bit && !Fn->getAttributes().hasAttribute(AttributeSet::FunctionIndex, |
| 491 | Attribute::NoRedZone) && |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 492 | !RegInfo->needsStackRealignment(MF) && |
Nick Lewycky | 50f02cb | 2011-12-02 22:16:29 +0000 | [diff] [blame] | 493 | !MFI->hasVarSizedObjects() && // No dynamic alloca. |
| 494 | !MFI->adjustsStack() && // No calls. |
| 495 | !IsWin64 && // Win64 has no Red Zone |
Nadav Rotem | 1bef5a0 | 2012-12-23 07:30:09 +0000 | [diff] [blame] | 496 | !usesTheStack(MF) && // Don't push and pop. |
Reid Kleckner | 9c65821 | 2014-04-10 22:58:43 +0000 | [diff] [blame] | 497 | !MF.shouldSplitStack()) { // Regular stack |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 498 | uint64_t MinSize = X86FI->getCalleeSavedFrameSize(); |
| 499 | if (HasFP) MinSize += SlotSize; |
| 500 | StackSize = std::max(MinSize, StackSize > 128 ? StackSize - 128 : 0); |
| 501 | MFI->setStackSize(StackSize); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 502 | } |
| 503 | |
| 504 | // Insert stack pointer adjustment for later moving of return addr. Only |
| 505 | // applies to tail call optimized functions where the callee argument stack |
| 506 | // size is bigger than the callers. |
| 507 | if (TailCallReturnAddrDelta < 0) { |
| 508 | MachineInstr *MI = |
| 509 | BuildMI(MBB, MBBI, DL, |
Eli Bendersky | 44a40ca | 2013-02-05 21:53:29 +0000 | [diff] [blame] | 510 | TII.get(getSUBriOpcode(IsLP64, -TailCallReturnAddrDelta)), |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 511 | StackPtr) |
| 512 | .addReg(StackPtr) |
Charles Davis | 7ed40cb | 2011-06-12 01:45:54 +0000 | [diff] [blame] | 513 | .addImm(-TailCallReturnAddrDelta) |
| 514 | .setMIFlag(MachineInstr::FrameSetup); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 515 | MI->getOperand(3).setIsDead(); // The EFLAGS implicit def is dead. |
| 516 | } |
| 517 | |
| 518 | // Mapping for machine moves: |
| 519 | // |
| 520 | // DST: VirtualFP AND |
| 521 | // SRC: VirtualFP => DW_CFA_def_cfa_offset |
| 522 | // ELSE => DW_CFA_def_cfa |
| 523 | // |
| 524 | // SRC: VirtualFP AND |
| 525 | // DST: Register => DW_CFA_def_cfa_register |
| 526 | // |
| 527 | // ELSE |
| 528 | // OFFSET < 0 => DW_CFA_offset_extended_sf |
| 529 | // REG < 64 => DW_CFA_offset + Reg |
| 530 | // ELSE => DW_CFA_offset_extended |
| 531 | |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 532 | uint64_t NumBytes = 0; |
Michael Liao | 6d810bd | 2012-10-25 06:29:14 +0000 | [diff] [blame] | 533 | int stackGrowth = -SlotSize; |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 534 | |
| 535 | if (HasFP) { |
| 536 | // Calculate required stack adjustment. |
| 537 | uint64_t FrameSize = StackSize - SlotSize; |
Alexey Samsonov | dcc1291 | 2012-07-16 06:54:09 +0000 | [diff] [blame] | 538 | if (RegInfo->needsStackRealignment(MF)) { |
| 539 | // Callee-saved registers are pushed on stack before the stack |
| 540 | // is realigned. |
| 541 | FrameSize -= X86FI->getCalleeSavedFrameSize(); |
| 542 | NumBytes = (FrameSize + MaxAlign - 1) / MaxAlign * MaxAlign; |
| 543 | } else { |
| 544 | NumBytes = FrameSize - X86FI->getCalleeSavedFrameSize(); |
| 545 | } |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 546 | |
| 547 | // Get the offset of the stack slot for the EBP register, which is |
| 548 | // guaranteed to be the last slot by processFunctionBeforeFrameFinalized. |
| 549 | // Update the frame offset adjustment. |
| 550 | MFI->setOffsetAdjustment(-NumBytes); |
| 551 | |
| 552 | // Save EBP/RBP into the appropriate stack slot. |
| 553 | BuildMI(MBB, MBBI, DL, TII.get(Is64Bit ? X86::PUSH64r : X86::PUSH32r)) |
Charles Davis | 7ed40cb | 2011-06-12 01:45:54 +0000 | [diff] [blame] | 554 | .addReg(FramePtr, RegState::Kill) |
| 555 | .setMIFlag(MachineInstr::FrameSetup); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 556 | |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 557 | if (NeedsDwarfCFI) { |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 558 | // Mark the place where EBP/RBP was saved. |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 559 | // Define the current CFA rule to use the provided offset. |
Rafael Espindola | 84ee6c4 | 2013-05-15 22:27:35 +0000 | [diff] [blame] | 560 | assert(StackSize); |
Rafael Espindola | b1f25f1 | 2014-03-07 06:08:31 +0000 | [diff] [blame] | 561 | unsigned CFIIndex = MMI.addFrameInst( |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 562 | MCCFIInstruction::createDefCfaOffset(nullptr, 2 * stackGrowth)); |
Eric Christopher | 612bb69 | 2014-04-29 00:16:46 +0000 | [diff] [blame] | 563 | BuildMI(MBB, MBBI, DL, TII.get(TargetOpcode::CFI_INSTRUCTION)) |
Rafael Espindola | b1f25f1 | 2014-03-07 06:08:31 +0000 | [diff] [blame] | 564 | .addCFIIndex(CFIIndex); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 565 | |
| 566 | // Change the rule for the FramePtr to be an "offset" rule. |
Rafael Espindola | b08d2c2 | 2013-05-16 21:02:15 +0000 | [diff] [blame] | 567 | unsigned DwarfFramePtr = RegInfo->getDwarfRegNum(FramePtr, true); |
Rafael Espindola | b1f25f1 | 2014-03-07 06:08:31 +0000 | [diff] [blame] | 568 | CFIIndex = MMI.addFrameInst( |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 569 | MCCFIInstruction::createOffset(nullptr, |
| 570 | DwarfFramePtr, 2 * stackGrowth)); |
Eric Christopher | 612bb69 | 2014-04-29 00:16:46 +0000 | [diff] [blame] | 571 | BuildMI(MBB, MBBI, DL, TII.get(TargetOpcode::CFI_INSTRUCTION)) |
Rafael Espindola | b1f25f1 | 2014-03-07 06:08:31 +0000 | [diff] [blame] | 572 | .addCFIIndex(CFIIndex); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 573 | } |
| 574 | |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 575 | if (NeedsWin64SEH) { |
| 576 | BuildMI(MBB, MBBI, DL, TII.get(X86::SEH_PushReg)) |
| 577 | .addImm(FramePtr) |
| 578 | .setMIFlag(MachineInstr::FrameSetup); |
| 579 | } |
| 580 | |
Bill Wendling | b97270d | 2011-07-25 18:00:28 +0000 | [diff] [blame] | 581 | // Update EBP with the new base value. |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 582 | BuildMI(MBB, MBBI, DL, |
| 583 | TII.get(Is64Bit ? X86::MOV64rr : X86::MOV32rr), FramePtr) |
Charles Davis | 7ed40cb | 2011-06-12 01:45:54 +0000 | [diff] [blame] | 584 | .addReg(StackPtr) |
| 585 | .setMIFlag(MachineInstr::FrameSetup); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 586 | |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 587 | if (NeedsDwarfCFI) { |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 588 | // Mark effective beginning of when frame pointer becomes valid. |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 589 | // Define the current CFA to use the EBP/RBP register. |
Rafael Espindola | b08d2c2 | 2013-05-16 21:02:15 +0000 | [diff] [blame] | 590 | unsigned DwarfFramePtr = RegInfo->getDwarfRegNum(FramePtr, true); |
Rafael Espindola | b1f25f1 | 2014-03-07 06:08:31 +0000 | [diff] [blame] | 591 | unsigned CFIIndex = MMI.addFrameInst( |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 592 | MCCFIInstruction::createDefCfaRegister(nullptr, DwarfFramePtr)); |
Eric Christopher | 612bb69 | 2014-04-29 00:16:46 +0000 | [diff] [blame] | 593 | BuildMI(MBB, MBBI, DL, TII.get(TargetOpcode::CFI_INSTRUCTION)) |
Rafael Espindola | b1f25f1 | 2014-03-07 06:08:31 +0000 | [diff] [blame] | 594 | .addCFIIndex(CFIIndex); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 595 | } |
| 596 | |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 597 | // Mark the FramePtr as live-in in every block. |
| 598 | for (MachineFunction::iterator I = MF.begin(), E = MF.end(); I != E; ++I) |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 599 | I->addLiveIn(FramePtr); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 600 | } else { |
| 601 | NumBytes = StackSize - X86FI->getCalleeSavedFrameSize(); |
| 602 | } |
| 603 | |
| 604 | // Skip the callee-saved push instructions. |
| 605 | bool PushedRegs = false; |
| 606 | int StackOffset = 2 * stackGrowth; |
| 607 | |
| 608 | while (MBBI != MBB.end() && |
| 609 | (MBBI->getOpcode() == X86::PUSH32r || |
| 610 | MBBI->getOpcode() == X86::PUSH64r)) { |
| 611 | PushedRegs = true; |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 612 | unsigned Reg = MBBI->getOperand(0).getReg(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 613 | ++MBBI; |
| 614 | |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 615 | if (!HasFP && NeedsDwarfCFI) { |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 616 | // Mark callee-saved push instruction. |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 617 | // Define the current CFA rule to use the provided offset. |
Rafael Espindola | 7242186 | 2013-05-16 04:59:17 +0000 | [diff] [blame] | 618 | assert(StackSize); |
Rafael Espindola | b1f25f1 | 2014-03-07 06:08:31 +0000 | [diff] [blame] | 619 | unsigned CFIIndex = MMI.addFrameInst( |
| 620 | MCCFIInstruction::createDefCfaOffset(nullptr, StackOffset)); |
Eric Christopher | 612bb69 | 2014-04-29 00:16:46 +0000 | [diff] [blame] | 621 | BuildMI(MBB, MBBI, DL, TII.get(TargetOpcode::CFI_INSTRUCTION)) |
Rafael Espindola | b1f25f1 | 2014-03-07 06:08:31 +0000 | [diff] [blame] | 622 | .addCFIIndex(CFIIndex); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 623 | StackOffset += stackGrowth; |
| 624 | } |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 625 | |
| 626 | if (NeedsWin64SEH) { |
| 627 | BuildMI(MBB, MBBI, DL, TII.get(X86::SEH_PushReg)).addImm(Reg).setMIFlag( |
| 628 | MachineInstr::FrameSetup); |
| 629 | } |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 630 | } |
| 631 | |
Alexey Samsonov | dcc1291 | 2012-07-16 06:54:09 +0000 | [diff] [blame] | 632 | // Realign stack after we pushed callee-saved registers (so that we'll be |
| 633 | // able to calculate their offsets from the frame pointer). |
Alexey Samsonov | dcc1291 | 2012-07-16 06:54:09 +0000 | [diff] [blame] | 634 | if (RegInfo->needsStackRealignment(MF)) { |
| 635 | assert(HasFP && "There should be a frame pointer if stack is realigned."); |
| 636 | MachineInstr *MI = |
| 637 | BuildMI(MBB, MBBI, DL, |
| 638 | TII.get(Is64Bit ? X86::AND64ri32 : X86::AND32ri), StackPtr) |
| 639 | .addReg(StackPtr) |
| 640 | .addImm(-MaxAlign) |
| 641 | .setMIFlag(MachineInstr::FrameSetup); |
| 642 | |
| 643 | // The EFLAGS implicit def is dead. |
| 644 | MI->getOperand(3).setIsDead(); |
| 645 | } |
| 646 | |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 647 | // If there is an SUB32ri of ESP immediately before this instruction, merge |
| 648 | // the two. This can be the case when tail call elimination is enabled and |
| 649 | // the callee has more arguments then the caller. |
| 650 | NumBytes -= mergeSPUpdates(MBB, MBBI, StackPtr, true); |
| 651 | |
| 652 | // If there is an ADD32ri or SUB32ri of ESP immediately after this |
| 653 | // instruction, merge the two instructions. |
| 654 | mergeSPUpdatesDown(MBB, MBBI, StackPtr, &NumBytes); |
| 655 | |
| 656 | // Adjust stack pointer: ESP -= numbytes. |
| 657 | |
| 658 | // Windows and cygwin/mingw require a prologue helper routine when allocating |
| 659 | // more than 4K bytes on the stack. Windows uses __chkstk and cygwin/mingw |
| 660 | // uses __alloca. __alloca and the 32-bit version of __chkstk will probe the |
| 661 | // stack and adjust the stack pointer in one go. The 64-bit version of |
| 662 | // __chkstk is only responsible for probing the stack. The 64-bit prologue is |
| 663 | // responsible for adjusting the stack pointer. Touching the stack at 4K |
| 664 | // increments is necessary to ensure that the guard pages used by the OS |
| 665 | // virtual memory manager are allocated in correct sequence. |
Tim Northover | 9653eb5 | 2013-12-10 16:57:43 +0000 | [diff] [blame] | 666 | if (NumBytes >= 4096 && STI.isOSWindows() && !STI.isTargetMacho()) { |
NAKAMURA Takumi | 521eb7c | 2011-03-24 07:07:00 +0000 | [diff] [blame] | 667 | const char *StackProbeSymbol; |
NAKAMURA Takumi | 521eb7c | 2011-03-24 07:07:00 +0000 | [diff] [blame] | 668 | |
| 669 | if (Is64Bit) { |
Kai Nacke | 87b23ae | 2013-12-13 05:37:05 +0000 | [diff] [blame] | 670 | if (STI.isTargetCygMing()) { |
| 671 | StackProbeSymbol = "___chkstk_ms"; |
| 672 | } else { |
NAKAMURA Takumi | 521eb7c | 2011-03-24 07:07:00 +0000 | [diff] [blame] | 673 | StackProbeSymbol = "__chkstk"; |
NAKAMURA Takumi | 521eb7c | 2011-03-24 07:07:00 +0000 | [diff] [blame] | 674 | } |
| 675 | } else if (STI.isTargetCygMing()) |
| 676 | StackProbeSymbol = "_alloca"; |
| 677 | else |
| 678 | StackProbeSymbol = "_chkstk"; |
| 679 | |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 680 | // Check whether EAX is livein for this function. |
| 681 | bool isEAXAlive = isEAXLiveIn(MF); |
| 682 | |
NAKAMURA Takumi | 521eb7c | 2011-03-24 07:07:00 +0000 | [diff] [blame] | 683 | if (isEAXAlive) { |
| 684 | // Sanity check that EAX is not livein for this function. |
| 685 | // It should not be, so throw an assert. |
| 686 | assert(!Is64Bit && "EAX is livein in x64 case!"); |
| 687 | |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 688 | // Save EAX |
| 689 | BuildMI(MBB, MBBI, DL, TII.get(X86::PUSH32r)) |
Bill Wendling | 28b6e12 | 2011-07-21 00:44:56 +0000 | [diff] [blame] | 690 | .addReg(X86::EAX, RegState::Kill) |
| 691 | .setMIFlag(MachineInstr::FrameSetup); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 692 | } |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 693 | |
NAKAMURA Takumi | 521eb7c | 2011-03-24 07:07:00 +0000 | [diff] [blame] | 694 | if (Is64Bit) { |
| 695 | // Handle the 64-bit Windows ABI case where we need to call __chkstk. |
| 696 | // Function prologue is responsible for adjusting the stack pointer. |
| 697 | BuildMI(MBB, MBBI, DL, TII.get(X86::MOV64ri), X86::RAX) |
Bill Wendling | 28b6e12 | 2011-07-21 00:44:56 +0000 | [diff] [blame] | 698 | .addImm(NumBytes) |
| 699 | .setMIFlag(MachineInstr::FrameSetup); |
NAKAMURA Takumi | 521eb7c | 2011-03-24 07:07:00 +0000 | [diff] [blame] | 700 | } else { |
| 701 | // Allocate NumBytes-4 bytes on stack in case of isEAXAlive. |
| 702 | // We'll also use 4 already allocated bytes for EAX. |
| 703 | BuildMI(MBB, MBBI, DL, TII.get(X86::MOV32ri), X86::EAX) |
Bill Wendling | 28b6e12 | 2011-07-21 00:44:56 +0000 | [diff] [blame] | 704 | .addImm(isEAXAlive ? NumBytes - 4 : NumBytes) |
| 705 | .setMIFlag(MachineInstr::FrameSetup); |
NAKAMURA Takumi | 521eb7c | 2011-03-24 07:07:00 +0000 | [diff] [blame] | 706 | } |
| 707 | |
| 708 | BuildMI(MBB, MBBI, DL, |
| 709 | TII.get(Is64Bit ? X86::W64ALLOCA : X86::CALLpcrel32)) |
| 710 | .addExternalSymbol(StackProbeSymbol) |
| 711 | .addReg(StackPtr, RegState::Define | RegState::Implicit) |
Bill Wendling | 28b6e12 | 2011-07-21 00:44:56 +0000 | [diff] [blame] | 712 | .addReg(X86::EFLAGS, RegState::Define | RegState::Implicit) |
| 713 | .setMIFlag(MachineInstr::FrameSetup); |
NAKAMURA Takumi | 521eb7c | 2011-03-24 07:07:00 +0000 | [diff] [blame] | 714 | |
Kai Nacke | 87b23ae | 2013-12-13 05:37:05 +0000 | [diff] [blame] | 715 | if (Is64Bit) { |
| 716 | // MSVC x64's __chkstk and cygwin/mingw's ___chkstk_ms do not adjust %rsp |
| 717 | // themself. It also does not clobber %rax so we can reuse it when |
| 718 | // adjusting %rsp. |
Nico Rieck | 51969be | 2013-07-08 11:20:11 +0000 | [diff] [blame] | 719 | BuildMI(MBB, MBBI, DL, TII.get(X86::SUB64rr), StackPtr) |
| 720 | .addReg(StackPtr) |
| 721 | .addReg(X86::RAX) |
| 722 | .setMIFlag(MachineInstr::FrameSetup); |
| 723 | } |
NAKAMURA Takumi | 521eb7c | 2011-03-24 07:07:00 +0000 | [diff] [blame] | 724 | if (isEAXAlive) { |
| 725 | // Restore EAX |
| 726 | MachineInstr *MI = addRegOffset(BuildMI(MF, DL, TII.get(X86::MOV32rm), |
| 727 | X86::EAX), |
| 728 | StackPtr, false, NumBytes - 4); |
Bill Wendling | 28b6e12 | 2011-07-21 00:44:56 +0000 | [diff] [blame] | 729 | MI->setFlag(MachineInstr::FrameSetup); |
NAKAMURA Takumi | 521eb7c | 2011-03-24 07:07:00 +0000 | [diff] [blame] | 730 | MBB.insert(MBBI, MI); |
| 731 | } |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 732 | } else if (NumBytes) { |
Eli Bendersky | 44a40ca | 2013-02-05 21:53:29 +0000 | [diff] [blame] | 733 | emitSPUpdate(MBB, MBBI, StackPtr, -(int64_t)NumBytes, Is64Bit, IsLP64, |
Eric Christopher | f4fba5c | 2012-10-03 08:10:01 +0000 | [diff] [blame] | 734 | UseLEA, TII, *RegInfo); |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 735 | } |
| 736 | |
| 737 | int SEHFrameOffset = 0; |
| 738 | if (NeedsWin64SEH) { |
| 739 | if (HasFP) { |
| 740 | // We need to set frame base offset low enough such that all saved |
| 741 | // register offsets would be positive relative to it, but we can't |
| 742 | // just use NumBytes, because .seh_setframe offset must be <=240. |
| 743 | // So we pretend to have only allocated enough space to spill the |
| 744 | // non-volatile registers. |
| 745 | // We don't care about the rest of stack allocation, because unwinder |
| 746 | // will restore SP to (BP - SEHFrameOffset) |
| 747 | for (const CalleeSavedInfo &Info : MFI->getCalleeSavedInfo()) { |
| 748 | int offset = MFI->getObjectOffset(Info.getFrameIdx()); |
| 749 | SEHFrameOffset = std::max(SEHFrameOffset, abs(offset)); |
| 750 | } |
| 751 | SEHFrameOffset += SEHFrameOffset % 16; // ensure alignmant |
| 752 | |
| 753 | // This only needs to account for XMM spill slots, GPR slots |
| 754 | // are covered by .seh_pushreg's emitted above. |
| 755 | BuildMI(MBB, MBBI, DL, TII.get(X86::SEH_StackAlloc)) |
| 756 | .addImm(SEHFrameOffset - X86FI->getCalleeSavedFrameSize()) |
| 757 | .setMIFlag(MachineInstr::FrameSetup); |
| 758 | |
| 759 | BuildMI(MBB, MBBI, DL, TII.get(X86::SEH_SetFrame)) |
| 760 | .addImm(FramePtr) |
| 761 | .addImm(SEHFrameOffset) |
| 762 | .setMIFlag(MachineInstr::FrameSetup); |
| 763 | } else { |
| 764 | // SP will be the base register for restoring XMMs |
| 765 | if (NumBytes) { |
| 766 | BuildMI(MBB, MBBI, DL, TII.get(X86::SEH_StackAlloc)) |
| 767 | .addImm(NumBytes) |
| 768 | .setMIFlag(MachineInstr::FrameSetup); |
| 769 | } |
| 770 | } |
| 771 | } |
| 772 | |
| 773 | // Skip the rest of register spilling code |
| 774 | while (MBBI != MBB.end() && MBBI->getFlag(MachineInstr::FrameSetup)) |
| 775 | ++MBBI; |
| 776 | |
| 777 | // Emit SEH info for non-GPRs |
| 778 | if (NeedsWin64SEH) { |
| 779 | for (const CalleeSavedInfo &Info : MFI->getCalleeSavedInfo()) { |
| 780 | unsigned Reg = Info.getReg(); |
| 781 | if (X86::GR64RegClass.contains(Reg) || X86::GR32RegClass.contains(Reg)) |
| 782 | continue; |
| 783 | assert(X86::FR64RegClass.contains(Reg) && "Unexpected register class"); |
| 784 | |
| 785 | int Offset = getFrameIndexOffset(MF, Info.getFrameIdx()); |
| 786 | Offset += SEHFrameOffset; |
| 787 | |
| 788 | BuildMI(MBB, MBBI, DL, TII.get(X86::SEH_SaveXMM)) |
| 789 | .addImm(Reg) |
| 790 | .addImm(Offset) |
| 791 | .setMIFlag(MachineInstr::FrameSetup); |
| 792 | } |
| 793 | |
| 794 | BuildMI(MBB, MBBI, DL, TII.get(X86::SEH_EndPrologue)) |
| 795 | .setMIFlag(MachineInstr::FrameSetup); |
| 796 | } |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 797 | |
Chad Rosier | bdb08ac | 2012-07-10 17:45:53 +0000 | [diff] [blame] | 798 | // If we need a base pointer, set it up here. It's whatever the value |
| 799 | // of the stack pointer is at this point. Any variable size objects |
| 800 | // will be allocated after this, so we can still use the base pointer |
| 801 | // to reference locals. |
| 802 | if (RegInfo->hasBasePointer(MF)) { |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 803 | // Update the base pointer with the current stack pointer. |
Chad Rosier | bdb08ac | 2012-07-10 17:45:53 +0000 | [diff] [blame] | 804 | unsigned Opc = Is64Bit ? X86::MOV64rr : X86::MOV32rr; |
| 805 | BuildMI(MBB, MBBI, DL, TII.get(Opc), BasePtr) |
| 806 | .addReg(StackPtr) |
| 807 | .setMIFlag(MachineInstr::FrameSetup); |
Chad Rosier | bdb08ac | 2012-07-10 17:45:53 +0000 | [diff] [blame] | 808 | } |
| 809 | |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 810 | if (((!HasFP && NumBytes) || PushedRegs) && NeedsDwarfCFI) { |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 811 | // Mark end of stack pointer adjustment. |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 812 | if (!HasFP && NumBytes) { |
| 813 | // Define the current CFA rule to use the provided offset. |
Rafael Espindola | 84ee6c4 | 2013-05-15 22:27:35 +0000 | [diff] [blame] | 814 | assert(StackSize); |
Rafael Espindola | b1f25f1 | 2014-03-07 06:08:31 +0000 | [diff] [blame] | 815 | unsigned CFIIndex = MMI.addFrameInst( |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 816 | MCCFIInstruction::createDefCfaOffset(nullptr, |
| 817 | -StackSize + stackGrowth)); |
Rafael Espindola | b1f25f1 | 2014-03-07 06:08:31 +0000 | [diff] [blame] | 818 | |
Eric Christopher | 612bb69 | 2014-04-29 00:16:46 +0000 | [diff] [blame] | 819 | BuildMI(MBB, MBBI, DL, TII.get(TargetOpcode::CFI_INSTRUCTION)) |
Rafael Espindola | b1f25f1 | 2014-03-07 06:08:31 +0000 | [diff] [blame] | 820 | .addCFIIndex(CFIIndex); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 821 | } |
| 822 | |
| 823 | // Emit DWARF info specifying the offsets of the callee-saved registers. |
| 824 | if (PushedRegs) |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 825 | emitCalleeSavedFrameMoves(MBB, MBBI, DL); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 826 | } |
| 827 | } |
| 828 | |
Anton Korobeynikov | 2f93128 | 2011-01-10 12:39:04 +0000 | [diff] [blame] | 829 | void X86FrameLowering::emitEpilogue(MachineFunction &MF, |
Nick Lewycky | 34a425b | 2011-06-14 03:23:52 +0000 | [diff] [blame] | 830 | MachineBasicBlock &MBB) const { |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 831 | const MachineFrameInfo *MFI = MF.getFrameInfo(); |
| 832 | X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); |
Eric Christopher | 11b05cc | 2014-06-05 00:09:05 +0000 | [diff] [blame] | 833 | const X86RegisterInfo *RegInfo = |
| 834 | static_cast<const X86RegisterInfo *>(MF.getTarget().getRegisterInfo()); |
| 835 | const TargetInstrInfo &TII = *MF.getTarget().getInstrInfo(); |
Jakob Stoklund Olesen | 4bc5e38 | 2011-01-13 21:28:52 +0000 | [diff] [blame] | 836 | MachineBasicBlock::iterator MBBI = MBB.getLastNonDebugInstr(); |
| 837 | assert(MBBI != MBB.end() && "Returning block has no instructions"); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 838 | unsigned RetOpcode = MBBI->getOpcode(); |
| 839 | DebugLoc DL = MBBI->getDebugLoc(); |
Eric Christopher | f438164 | 2014-06-05 22:00:31 +0000 | [diff] [blame] | 840 | const X86Subtarget &STI = MF.getTarget().getSubtarget<X86Subtarget>(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 841 | bool Is64Bit = STI.is64Bit(); |
Eli Bendersky | 44a40ca | 2013-02-05 21:53:29 +0000 | [diff] [blame] | 842 | bool IsLP64 = STI.isTarget64BitLP64(); |
Evan Cheng | 1b81fdd | 2012-02-07 22:50:41 +0000 | [diff] [blame] | 843 | bool UseLEA = STI.useLeaForSP(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 844 | unsigned StackAlign = getStackAlignment(); |
| 845 | unsigned SlotSize = RegInfo->getSlotSize(); |
| 846 | unsigned FramePtr = RegInfo->getFrameRegister(MF); |
| 847 | unsigned StackPtr = RegInfo->getStackRegister(); |
| 848 | |
| 849 | switch (RetOpcode) { |
| 850 | default: |
| 851 | llvm_unreachable("Can only insert epilog into returning blocks"); |
David Woodhouse | 79dd505 | 2014-01-08 12:58:07 +0000 | [diff] [blame] | 852 | case X86::RETQ: |
| 853 | case X86::RETL: |
David Woodhouse | 4e033b0 | 2014-01-13 14:05:59 +0000 | [diff] [blame] | 854 | case X86::RETIL: |
| 855 | case X86::RETIQ: |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 856 | case X86::TCRETURNdi: |
| 857 | case X86::TCRETURNri: |
| 858 | case X86::TCRETURNmi: |
| 859 | case X86::TCRETURNdi64: |
| 860 | case X86::TCRETURNri64: |
| 861 | case X86::TCRETURNmi64: |
| 862 | case X86::EH_RETURN: |
| 863 | case X86::EH_RETURN64: |
| 864 | break; // These are ok |
| 865 | } |
| 866 | |
| 867 | // Get the number of bytes to allocate from the FrameInfo. |
| 868 | uint64_t StackSize = MFI->getStackSize(); |
| 869 | uint64_t MaxAlign = MFI->getMaxAlignment(); |
| 870 | unsigned CSSize = X86FI->getCalleeSavedFrameSize(); |
| 871 | uint64_t NumBytes = 0; |
| 872 | |
| 873 | // If we're forcing a stack realignment we can't rely on just the frame |
| 874 | // info, we need to know the ABI stack alignment as well in case we |
| 875 | // have a call out. Otherwise just make sure we have some alignment - we'll |
| 876 | // go with the minimum. |
| 877 | if (ForceStackAlign) { |
| 878 | if (MFI->hasCalls()) |
| 879 | MaxAlign = (StackAlign > MaxAlign) ? StackAlign : MaxAlign; |
| 880 | else |
| 881 | MaxAlign = MaxAlign ? MaxAlign : 4; |
| 882 | } |
| 883 | |
Anton Korobeynikov | 0eecf5d | 2010-11-18 21:19:35 +0000 | [diff] [blame] | 884 | if (hasFP(MF)) { |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 885 | // Calculate required stack adjustment. |
| 886 | uint64_t FrameSize = StackSize - SlotSize; |
Alexey Samsonov | dcc1291 | 2012-07-16 06:54:09 +0000 | [diff] [blame] | 887 | if (RegInfo->needsStackRealignment(MF)) { |
| 888 | // Callee-saved registers were pushed on stack before the stack |
| 889 | // was realigned. |
| 890 | FrameSize -= CSSize; |
| 891 | NumBytes = (FrameSize + MaxAlign - 1) / MaxAlign * MaxAlign; |
| 892 | } else { |
| 893 | NumBytes = FrameSize - CSSize; |
| 894 | } |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 895 | |
| 896 | // Pop EBP. |
| 897 | BuildMI(MBB, MBBI, DL, |
| 898 | TII.get(Is64Bit ? X86::POP64r : X86::POP32r), FramePtr); |
| 899 | } else { |
| 900 | NumBytes = StackSize - CSSize; |
| 901 | } |
| 902 | |
| 903 | // Skip the callee-saved pop instructions. |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 904 | while (MBBI != MBB.begin()) { |
Benjamin Kramer | b6d0bd4 | 2014-03-02 12:27:27 +0000 | [diff] [blame] | 905 | MachineBasicBlock::iterator PI = std::prev(MBBI); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 906 | unsigned Opc = PI->getOpcode(); |
| 907 | |
Jakob Stoklund Olesen | 4bc5e38 | 2011-01-13 21:28:52 +0000 | [diff] [blame] | 908 | if (Opc != X86::POP32r && Opc != X86::POP64r && Opc != X86::DBG_VALUE && |
Evan Cheng | 7f8e563 | 2011-12-07 07:15:52 +0000 | [diff] [blame] | 909 | !PI->isTerminator()) |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 910 | break; |
| 911 | |
| 912 | --MBBI; |
| 913 | } |
Alexey Samsonov | dcc1291 | 2012-07-16 06:54:09 +0000 | [diff] [blame] | 914 | MachineBasicBlock::iterator FirstCSPop = MBBI; |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 915 | |
| 916 | DL = MBBI->getDebugLoc(); |
| 917 | |
| 918 | // If there is an ADD32ri or SUB32ri of ESP immediately before this |
| 919 | // instruction, merge the two instructions. |
| 920 | if (NumBytes || MFI->hasVarSizedObjects()) |
| 921 | mergeSPUpdatesUp(MBB, MBBI, StackPtr, &NumBytes); |
| 922 | |
| 923 | // If dynamic alloca is used, then reset esp to point to the last callee-saved |
| 924 | // slot before popping them off! Same applies for the case, when stack was |
| 925 | // realigned. |
Alexey Samsonov | dcc1291 | 2012-07-16 06:54:09 +0000 | [diff] [blame] | 926 | if (RegInfo->needsStackRealignment(MF) || MFI->hasVarSizedObjects()) { |
| 927 | if (RegInfo->needsStackRealignment(MF)) |
| 928 | MBBI = FirstCSPop; |
| 929 | if (CSSize != 0) { |
Eli Bendersky | ef4558a | 2013-02-06 20:43:57 +0000 | [diff] [blame] | 930 | unsigned Opc = getLEArOpcode(IsLP64); |
Alexey Samsonov | dcc1291 | 2012-07-16 06:54:09 +0000 | [diff] [blame] | 931 | addRegOffset(BuildMI(MBB, MBBI, DL, TII.get(Opc), StackPtr), |
| 932 | FramePtr, false, -CSSize); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 933 | } else { |
Alexey Samsonov | dcc1291 | 2012-07-16 06:54:09 +0000 | [diff] [blame] | 934 | unsigned Opc = (Is64Bit ? X86::MOV64rr : X86::MOV32rr); |
| 935 | BuildMI(MBB, MBBI, DL, TII.get(Opc), StackPtr) |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 936 | .addReg(FramePtr); |
| 937 | } |
| 938 | } else if (NumBytes) { |
| 939 | // Adjust stack pointer back: ESP += numbytes. |
Eli Bendersky | 44a40ca | 2013-02-05 21:53:29 +0000 | [diff] [blame] | 940 | emitSPUpdate(MBB, MBBI, StackPtr, NumBytes, Is64Bit, IsLP64, UseLEA, |
| 941 | TII, *RegInfo); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 942 | } |
| 943 | |
| 944 | // We're returning from function via eh_return. |
| 945 | if (RetOpcode == X86::EH_RETURN || RetOpcode == X86::EH_RETURN64) { |
Jakob Stoklund Olesen | 4bc5e38 | 2011-01-13 21:28:52 +0000 | [diff] [blame] | 946 | MBBI = MBB.getLastNonDebugInstr(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 947 | MachineOperand &DestAddr = MBBI->getOperand(0); |
| 948 | assert(DestAddr.isReg() && "Offset should be in register!"); |
| 949 | BuildMI(MBB, MBBI, DL, |
| 950 | TII.get(Is64Bit ? X86::MOV64rr : X86::MOV32rr), |
| 951 | StackPtr).addReg(DestAddr.getReg()); |
| 952 | } else if (RetOpcode == X86::TCRETURNri || RetOpcode == X86::TCRETURNdi || |
| 953 | RetOpcode == X86::TCRETURNmi || |
| 954 | RetOpcode == X86::TCRETURNri64 || RetOpcode == X86::TCRETURNdi64 || |
| 955 | RetOpcode == X86::TCRETURNmi64) { |
| 956 | bool isMem = RetOpcode == X86::TCRETURNmi || RetOpcode == X86::TCRETURNmi64; |
| 957 | // Tail call return: adjust the stack pointer and jump to callee. |
Jakob Stoklund Olesen | bbb1a54 | 2011-01-13 22:47:43 +0000 | [diff] [blame] | 958 | MBBI = MBB.getLastNonDebugInstr(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 959 | MachineOperand &JumpTarget = MBBI->getOperand(0); |
| 960 | MachineOperand &StackAdjust = MBBI->getOperand(isMem ? 5 : 1); |
| 961 | assert(StackAdjust.isImm() && "Expecting immediate value."); |
| 962 | |
| 963 | // Adjust stack pointer. |
| 964 | int StackAdj = StackAdjust.getImm(); |
| 965 | int MaxTCDelta = X86FI->getTCReturnAddrDelta(); |
| 966 | int Offset = 0; |
| 967 | assert(MaxTCDelta <= 0 && "MaxTCDelta should never be positive"); |
| 968 | |
| 969 | // Incoporate the retaddr area. |
| 970 | Offset = StackAdj-MaxTCDelta; |
| 971 | assert(Offset >= 0 && "Offset should never be negative"); |
| 972 | |
| 973 | if (Offset) { |
Chris Lattner | 0ab5e2c | 2011-04-15 05:18:47 +0000 | [diff] [blame] | 974 | // Check for possible merge with preceding ADD instruction. |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 975 | Offset += mergeSPUpdates(MBB, MBBI, StackPtr, true); |
Eli Bendersky | 44a40ca | 2013-02-05 21:53:29 +0000 | [diff] [blame] | 976 | emitSPUpdate(MBB, MBBI, StackPtr, Offset, Is64Bit, IsLP64, |
| 977 | UseLEA, TII, *RegInfo); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 978 | } |
| 979 | |
| 980 | // Jump to label or value in register. |
| 981 | if (RetOpcode == X86::TCRETURNdi || RetOpcode == X86::TCRETURNdi64) { |
Evan Cheng | d4b0873 | 2010-11-30 23:55:39 +0000 | [diff] [blame] | 982 | MachineInstrBuilder MIB = |
| 983 | BuildMI(MBB, MBBI, DL, TII.get((RetOpcode == X86::TCRETURNdi) |
| 984 | ? X86::TAILJMPd : X86::TAILJMPd64)); |
| 985 | if (JumpTarget.isGlobal()) |
| 986 | MIB.addGlobalAddress(JumpTarget.getGlobal(), JumpTarget.getOffset(), |
| 987 | JumpTarget.getTargetFlags()); |
| 988 | else { |
| 989 | assert(JumpTarget.isSymbol()); |
| 990 | MIB.addExternalSymbol(JumpTarget.getSymbolName(), |
| 991 | JumpTarget.getTargetFlags()); |
| 992 | } |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 993 | } else if (RetOpcode == X86::TCRETURNmi || RetOpcode == X86::TCRETURNmi64) { |
| 994 | MachineInstrBuilder MIB = |
| 995 | BuildMI(MBB, MBBI, DL, TII.get((RetOpcode == X86::TCRETURNmi) |
| 996 | ? X86::TAILJMPm : X86::TAILJMPm64)); |
| 997 | for (unsigned i = 0; i != 5; ++i) |
| 998 | MIB.addOperand(MBBI->getOperand(i)); |
| 999 | } else if (RetOpcode == X86::TCRETURNri64) { |
| 1000 | BuildMI(MBB, MBBI, DL, TII.get(X86::TAILJMPr64)). |
| 1001 | addReg(JumpTarget.getReg(), RegState::Kill); |
| 1002 | } else { |
| 1003 | BuildMI(MBB, MBBI, DL, TII.get(X86::TAILJMPr)). |
| 1004 | addReg(JumpTarget.getReg(), RegState::Kill); |
| 1005 | } |
| 1006 | |
Benjamin Kramer | b6d0bd4 | 2014-03-02 12:27:27 +0000 | [diff] [blame] | 1007 | MachineInstr *NewMI = std::prev(MBBI); |
Jakob Stoklund Olesen | 33f5d14 | 2012-12-20 22:54:02 +0000 | [diff] [blame] | 1008 | NewMI->copyImplicitOps(MF, MBBI); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 1009 | |
| 1010 | // Delete the pseudo instruction TCRETURN. |
| 1011 | MBB.erase(MBBI); |
David Woodhouse | 4e033b0 | 2014-01-13 14:05:59 +0000 | [diff] [blame] | 1012 | } else if ((RetOpcode == X86::RETQ || RetOpcode == X86::RETL || |
| 1013 | RetOpcode == X86::RETIQ || RetOpcode == X86::RETIL) && |
| 1014 | (X86FI->getTCReturnAddrDelta() < 0)) { |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 1015 | // Add the return addr area delta back since we are not tail calling. |
| 1016 | int delta = -1*X86FI->getTCReturnAddrDelta(); |
Jakob Stoklund Olesen | 4bc5e38 | 2011-01-13 21:28:52 +0000 | [diff] [blame] | 1017 | MBBI = MBB.getLastNonDebugInstr(); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 1018 | |
Chris Lattner | 0ab5e2c | 2011-04-15 05:18:47 +0000 | [diff] [blame] | 1019 | // Check for possible merge with preceding ADD instruction. |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 1020 | delta += mergeSPUpdates(MBB, MBBI, StackPtr, true); |
Eli Bendersky | 44a40ca | 2013-02-05 21:53:29 +0000 | [diff] [blame] | 1021 | emitSPUpdate(MBB, MBBI, StackPtr, delta, Is64Bit, IsLP64, UseLEA, TII, |
| 1022 | *RegInfo); |
Anton Korobeynikov | f7183ed | 2010-11-15 00:06:54 +0000 | [diff] [blame] | 1023 | } |
| 1024 | } |
Anton Korobeynikov | 14ee344 | 2010-11-18 23:25:52 +0000 | [diff] [blame] | 1025 | |
Eric Christopher | 4237bf1 | 2014-04-29 00:16:33 +0000 | [diff] [blame] | 1026 | int X86FrameLowering::getFrameIndexOffset(const MachineFunction &MF, |
| 1027 | int FI) const { |
Chad Rosier | 20b79dc | 2012-05-23 23:45:10 +0000 | [diff] [blame] | 1028 | const X86RegisterInfo *RegInfo = |
Anton Korobeynikov | 4687778 | 2010-11-20 15:59:32 +0000 | [diff] [blame] | 1029 | static_cast<const X86RegisterInfo*>(MF.getTarget().getRegisterInfo()); |
| 1030 | const MachineFrameInfo *MFI = MF.getFrameInfo(); |
| 1031 | int Offset = MFI->getObjectOffset(FI) - getOffsetOfLocalArea(); |
| 1032 | uint64_t StackSize = MFI->getStackSize(); |
| 1033 | |
Chad Rosier | bdb08ac | 2012-07-10 17:45:53 +0000 | [diff] [blame] | 1034 | if (RegInfo->hasBasePointer(MF)) { |
| 1035 | assert (hasFP(MF) && "VLAs and dynamic stack realign, but no FP?!"); |
| 1036 | if (FI < 0) { |
| 1037 | // Skip the saved EBP. |
| 1038 | return Offset + RegInfo->getSlotSize(); |
| 1039 | } else { |
| 1040 | assert((-(Offset + StackSize)) % MFI->getObjectAlignment(FI) == 0); |
| 1041 | return Offset + StackSize; |
| 1042 | } |
| 1043 | } else if (RegInfo->needsStackRealignment(MF)) { |
Anton Korobeynikov | 4687778 | 2010-11-20 15:59:32 +0000 | [diff] [blame] | 1044 | if (FI < 0) { |
| 1045 | // Skip the saved EBP. |
Chad Rosier | 20b79dc | 2012-05-23 23:45:10 +0000 | [diff] [blame] | 1046 | return Offset + RegInfo->getSlotSize(); |
Anton Korobeynikov | 4687778 | 2010-11-20 15:59:32 +0000 | [diff] [blame] | 1047 | } else { |
Duncan Sands | d278d35 | 2011-10-18 12:44:00 +0000 | [diff] [blame] | 1048 | assert((-(Offset + StackSize)) % MFI->getObjectAlignment(FI) == 0); |
Anton Korobeynikov | 4687778 | 2010-11-20 15:59:32 +0000 | [diff] [blame] | 1049 | return Offset + StackSize; |
| 1050 | } |
| 1051 | // FIXME: Support tail calls |
| 1052 | } else { |
| 1053 | if (!hasFP(MF)) |
| 1054 | return Offset + StackSize; |
| 1055 | |
| 1056 | // Skip the saved EBP. |
Chad Rosier | 20b79dc | 2012-05-23 23:45:10 +0000 | [diff] [blame] | 1057 | Offset += RegInfo->getSlotSize(); |
Anton Korobeynikov | 4687778 | 2010-11-20 15:59:32 +0000 | [diff] [blame] | 1058 | |
| 1059 | // Skip the RETADDR move area |
| 1060 | const X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); |
| 1061 | int TailCallReturnAddrDelta = X86FI->getTCReturnAddrDelta(); |
| 1062 | if (TailCallReturnAddrDelta < 0) |
| 1063 | Offset -= TailCallReturnAddrDelta; |
| 1064 | } |
| 1065 | |
| 1066 | return Offset; |
| 1067 | } |
Anton Korobeynikov | d08fbd1 | 2010-11-27 23:05:03 +0000 | [diff] [blame] | 1068 | |
Alexey Samsonov | c4b3ad8 | 2012-05-01 15:16:06 +0000 | [diff] [blame] | 1069 | int X86FrameLowering::getFrameIndexReference(const MachineFunction &MF, int FI, |
| 1070 | unsigned &FrameReg) const { |
Chad Rosier | 20b79dc | 2012-05-23 23:45:10 +0000 | [diff] [blame] | 1071 | const X86RegisterInfo *RegInfo = |
Alexey Samsonov | c4b3ad8 | 2012-05-01 15:16:06 +0000 | [diff] [blame] | 1072 | static_cast<const X86RegisterInfo*>(MF.getTarget().getRegisterInfo()); |
| 1073 | // We can't calculate offset from frame pointer if the stack is realigned, |
Chad Rosier | bdb08ac | 2012-07-10 17:45:53 +0000 | [diff] [blame] | 1074 | // so enforce usage of stack/base pointer. The base pointer is used when we |
| 1075 | // have dynamic allocas in addition to dynamic realignment. |
| 1076 | if (RegInfo->hasBasePointer(MF)) |
| 1077 | FrameReg = RegInfo->getBaseRegister(); |
| 1078 | else if (RegInfo->needsStackRealignment(MF)) |
| 1079 | FrameReg = RegInfo->getStackRegister(); |
| 1080 | else |
| 1081 | FrameReg = RegInfo->getFrameRegister(MF); |
Alexey Samsonov | c4b3ad8 | 2012-05-01 15:16:06 +0000 | [diff] [blame] | 1082 | return getFrameIndexOffset(MF, FI); |
| 1083 | } |
| 1084 | |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 1085 | bool X86FrameLowering::assignCalleeSavedSpillSlots( |
| 1086 | MachineFunction &MF, const TargetRegisterInfo *TRI, |
| 1087 | std::vector<CalleeSavedInfo> &CSI) const { |
| 1088 | MachineFrameInfo *MFI = MF.getFrameInfo(); |
| 1089 | const X86RegisterInfo *RegInfo = |
| 1090 | static_cast<const X86RegisterInfo *>(MF.getTarget().getRegisterInfo()); |
| 1091 | unsigned SlotSize = RegInfo->getSlotSize(); |
| 1092 | X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); |
Anton Korobeynikov | d08fbd1 | 2010-11-27 23:05:03 +0000 | [diff] [blame] | 1093 | |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 1094 | unsigned CalleeSavedFrameSize = 0; |
| 1095 | int SpillSlotOffset = getOffsetOfLocalArea() + X86FI->getTCReturnAddrDelta(); |
| 1096 | |
| 1097 | if (hasFP(MF)) { |
| 1098 | // emitPrologue always spills frame register the first thing. |
| 1099 | SpillSlotOffset -= SlotSize; |
| 1100 | MFI->CreateFixedSpillStackObject(SlotSize, SpillSlotOffset); |
| 1101 | |
| 1102 | // Since emitPrologue and emitEpilogue will handle spilling and restoring of |
| 1103 | // the frame register, we can delete it from CSI list and not have to worry |
| 1104 | // about avoiding it later. |
| 1105 | unsigned FPReg = RegInfo->getFrameRegister(MF); |
| 1106 | for (unsigned i = 0; i < CSI.size(); ++i) { |
| 1107 | if (CSI[i].getReg() == FPReg) { |
| 1108 | CSI.erase(CSI.begin() + i); |
| 1109 | break; |
| 1110 | } |
| 1111 | } |
| 1112 | } |
| 1113 | |
| 1114 | // Assign slots for GPRs. It increases frame size. |
| 1115 | for (unsigned i = CSI.size(); i != 0; --i) { |
| 1116 | unsigned Reg = CSI[i - 1].getReg(); |
| 1117 | |
| 1118 | if (!X86::GR64RegClass.contains(Reg) && !X86::GR32RegClass.contains(Reg)) |
| 1119 | continue; |
| 1120 | |
| 1121 | SpillSlotOffset -= SlotSize; |
| 1122 | CalleeSavedFrameSize += SlotSize; |
| 1123 | |
| 1124 | int SlotIndex = MFI->CreateFixedSpillStackObject(SlotSize, SpillSlotOffset); |
| 1125 | CSI[i - 1].setFrameIdx(SlotIndex); |
| 1126 | } |
| 1127 | |
| 1128 | X86FI->setCalleeSavedFrameSize(CalleeSavedFrameSize); |
| 1129 | |
| 1130 | // Assign slots for XMMs. |
| 1131 | for (unsigned i = CSI.size(); i != 0; --i) { |
| 1132 | unsigned Reg = CSI[i - 1].getReg(); |
| 1133 | if (X86::GR64RegClass.contains(Reg) || X86::GR32RegClass.contains(Reg)) |
| 1134 | continue; |
| 1135 | |
| 1136 | const TargetRegisterClass *RC = RegInfo->getMinimalPhysRegClass(Reg); |
| 1137 | // ensure alignment |
| 1138 | SpillSlotOffset -= abs(SpillSlotOffset) % RC->getAlignment(); |
| 1139 | // spill into slot |
| 1140 | SpillSlotOffset -= RC->getSize(); |
| 1141 | int SlotIndex = |
| 1142 | MFI->CreateFixedSpillStackObject(RC->getSize(), SpillSlotOffset); |
| 1143 | CSI[i - 1].setFrameIdx(SlotIndex); |
| 1144 | MFI->ensureMaxAlignment(RC->getAlignment()); |
| 1145 | } |
| 1146 | |
| 1147 | return true; |
| 1148 | } |
| 1149 | |
| 1150 | bool X86FrameLowering::spillCalleeSavedRegisters( |
| 1151 | MachineBasicBlock &MBB, MachineBasicBlock::iterator MI, |
| 1152 | const std::vector<CalleeSavedInfo> &CSI, |
| 1153 | const TargetRegisterInfo *TRI) const { |
Anton Korobeynikov | d08fbd1 | 2010-11-27 23:05:03 +0000 | [diff] [blame] | 1154 | DebugLoc DL = MBB.findDebugLoc(MI); |
| 1155 | |
| 1156 | MachineFunction &MF = *MBB.getParent(); |
Anton Korobeynikov | d08fbd1 | 2010-11-27 23:05:03 +0000 | [diff] [blame] | 1157 | const TargetInstrInfo &TII = *MF.getTarget().getInstrInfo(); |
Eric Christopher | f438164 | 2014-06-05 22:00:31 +0000 | [diff] [blame] | 1158 | const X86Subtarget &STI = MF.getTarget().getSubtarget<X86Subtarget>(); |
Anton Korobeynikov | d08fbd1 | 2010-11-27 23:05:03 +0000 | [diff] [blame] | 1159 | |
NAKAMURA Takumi | d4e5003 | 2011-02-27 08:47:19 +0000 | [diff] [blame] | 1160 | // Push GPRs. It increases frame size. |
Anton Korobeynikov | d08fbd1 | 2010-11-27 23:05:03 +0000 | [diff] [blame] | 1161 | unsigned Opc = STI.is64Bit() ? X86::PUSH64r : X86::PUSH32r; |
| 1162 | for (unsigned i = CSI.size(); i != 0; --i) { |
NAKAMURA Takumi | c403be1 | 2014-06-25 12:40:56 +0000 | [diff] [blame] | 1163 | unsigned Reg = CSI[i - 1].getReg(); |
| 1164 | |
| 1165 | if (!X86::GR64RegClass.contains(Reg) && !X86::GR32RegClass.contains(Reg)) |
NAKAMURA Takumi | d4e5003 | 2011-02-27 08:47:19 +0000 | [diff] [blame] | 1166 | continue; |
Anton Korobeynikov | d08fbd1 | 2010-11-27 23:05:03 +0000 | [diff] [blame] | 1167 | // Add the callee-saved register as live-in. It's killed at the spill. |
| 1168 | MBB.addLiveIn(Reg); |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 1169 | |
Charles Davis | 7ed40cb | 2011-06-12 01:45:54 +0000 | [diff] [blame] | 1170 | BuildMI(MBB, MI, DL, TII.get(Opc)).addReg(Reg, RegState::Kill) |
| 1171 | .setMIFlag(MachineInstr::FrameSetup); |
Anton Korobeynikov | d08fbd1 | 2010-11-27 23:05:03 +0000 | [diff] [blame] | 1172 | } |
| 1173 | |
NAKAMURA Takumi | d4e5003 | 2011-02-27 08:47:19 +0000 | [diff] [blame] | 1174 | // Make XMM regs spilled. X86 does not have ability of push/pop XMM. |
| 1175 | // It can be done by spilling XMMs to stack frame. |
NAKAMURA Takumi | d4e5003 | 2011-02-27 08:47:19 +0000 | [diff] [blame] | 1176 | for (unsigned i = CSI.size(); i != 0; --i) { |
| 1177 | unsigned Reg = CSI[i-1].getReg(); |
| 1178 | if (X86::GR64RegClass.contains(Reg) || |
| 1179 | X86::GR32RegClass.contains(Reg)) |
| 1180 | continue; |
| 1181 | // Add the callee-saved register as live-in. It's killed at the spill. |
| 1182 | MBB.addLiveIn(Reg); |
| 1183 | const TargetRegisterClass *RC = TRI->getMinimalPhysRegClass(Reg); |
NAKAMURA Takumi | c403be1 | 2014-06-25 12:40:56 +0000 | [diff] [blame] | 1184 | |
| 1185 | TII.storeRegToStackSlot(MBB, MI, Reg, true, CSI[i - 1].getFrameIdx(), RC, |
| 1186 | TRI); |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 1187 | --MI; |
| 1188 | MI->setFlag(MachineInstr::FrameSetup); |
| 1189 | ++MI; |
NAKAMURA Takumi | d4e5003 | 2011-02-27 08:47:19 +0000 | [diff] [blame] | 1190 | } |
| 1191 | |
Anton Korobeynikov | d08fbd1 | 2010-11-27 23:05:03 +0000 | [diff] [blame] | 1192 | return true; |
| 1193 | } |
| 1194 | |
Anton Korobeynikov | 2f93128 | 2011-01-10 12:39:04 +0000 | [diff] [blame] | 1195 | bool X86FrameLowering::restoreCalleeSavedRegisters(MachineBasicBlock &MBB, |
Anton Korobeynikov | d08fbd1 | 2010-11-27 23:05:03 +0000 | [diff] [blame] | 1196 | MachineBasicBlock::iterator MI, |
| 1197 | const std::vector<CalleeSavedInfo> &CSI, |
| 1198 | const TargetRegisterInfo *TRI) const { |
| 1199 | if (CSI.empty()) |
| 1200 | return false; |
| 1201 | |
| 1202 | DebugLoc DL = MBB.findDebugLoc(MI); |
| 1203 | |
| 1204 | MachineFunction &MF = *MBB.getParent(); |
| 1205 | const TargetInstrInfo &TII = *MF.getTarget().getInstrInfo(); |
Eric Christopher | f438164 | 2014-06-05 22:00:31 +0000 | [diff] [blame] | 1206 | const X86Subtarget &STI = MF.getTarget().getSubtarget<X86Subtarget>(); |
NAKAMURA Takumi | d4e5003 | 2011-02-27 08:47:19 +0000 | [diff] [blame] | 1207 | |
| 1208 | // Reload XMMs from stack frame. |
| 1209 | for (unsigned i = 0, e = CSI.size(); i != e; ++i) { |
| 1210 | unsigned Reg = CSI[i].getReg(); |
| 1211 | if (X86::GR64RegClass.contains(Reg) || |
| 1212 | X86::GR32RegClass.contains(Reg)) |
| 1213 | continue; |
NAKAMURA Takumi | c403be1 | 2014-06-25 12:40:56 +0000 | [diff] [blame] | 1214 | |
NAKAMURA Takumi | d4e5003 | 2011-02-27 08:47:19 +0000 | [diff] [blame] | 1215 | const TargetRegisterClass *RC = TRI->getMinimalPhysRegClass(Reg); |
NAKAMURA Takumi | c403be1 | 2014-06-25 12:40:56 +0000 | [diff] [blame] | 1216 | TII.loadRegFromStackSlot(MBB, MI, Reg, CSI[i].getFrameIdx(), RC, TRI); |
NAKAMURA Takumi | d4e5003 | 2011-02-27 08:47:19 +0000 | [diff] [blame] | 1217 | } |
| 1218 | |
| 1219 | // POP GPRs. |
Anton Korobeynikov | d08fbd1 | 2010-11-27 23:05:03 +0000 | [diff] [blame] | 1220 | unsigned Opc = STI.is64Bit() ? X86::POP64r : X86::POP32r; |
| 1221 | for (unsigned i = 0, e = CSI.size(); i != e; ++i) { |
| 1222 | unsigned Reg = CSI[i].getReg(); |
NAKAMURA Takumi | d4e5003 | 2011-02-27 08:47:19 +0000 | [diff] [blame] | 1223 | if (!X86::GR64RegClass.contains(Reg) && |
| 1224 | !X86::GR32RegClass.contains(Reg)) |
| 1225 | continue; |
NAKAMURA Takumi | 1db5995 | 2014-06-25 12:41:52 +0000 | [diff] [blame] | 1226 | |
NAKAMURA Takumi | d4e5003 | 2011-02-27 08:47:19 +0000 | [diff] [blame] | 1227 | BuildMI(MBB, MI, DL, TII.get(Opc), Reg); |
Anton Korobeynikov | d08fbd1 | 2010-11-27 23:05:03 +0000 | [diff] [blame] | 1228 | } |
| 1229 | return true; |
| 1230 | } |
Anton Korobeynikov | 7283b8d | 2010-11-27 23:05:25 +0000 | [diff] [blame] | 1231 | |
| 1232 | void |
Anton Korobeynikov | 2f93128 | 2011-01-10 12:39:04 +0000 | [diff] [blame] | 1233 | X86FrameLowering::processFunctionBeforeCalleeSavedScan(MachineFunction &MF, |
Eric Christopher | 11b05cc | 2014-06-05 00:09:05 +0000 | [diff] [blame] | 1234 | RegScavenger *RS) const { |
Anton Korobeynikov | 7283b8d | 2010-11-27 23:05:25 +0000 | [diff] [blame] | 1235 | MachineFrameInfo *MFI = MF.getFrameInfo(); |
Eric Christopher | 11b05cc | 2014-06-05 00:09:05 +0000 | [diff] [blame] | 1236 | const X86RegisterInfo *RegInfo = |
| 1237 | static_cast<const X86RegisterInfo *>(MF.getTarget().getRegisterInfo()); |
Anton Korobeynikov | 7283b8d | 2010-11-27 23:05:25 +0000 | [diff] [blame] | 1238 | unsigned SlotSize = RegInfo->getSlotSize(); |
| 1239 | |
| 1240 | X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); |
Tim Northover | ecc018c | 2013-08-04 09:35:57 +0000 | [diff] [blame] | 1241 | int64_t TailCallReturnAddrDelta = X86FI->getTCReturnAddrDelta(); |
Anton Korobeynikov | 7283b8d | 2010-11-27 23:05:25 +0000 | [diff] [blame] | 1242 | |
| 1243 | if (TailCallReturnAddrDelta < 0) { |
| 1244 | // create RETURNADDR area |
| 1245 | // arg |
| 1246 | // arg |
| 1247 | // RETADDR |
| 1248 | // { ... |
| 1249 | // RETADDR area |
| 1250 | // ... |
| 1251 | // } |
| 1252 | // [EBP] |
| 1253 | MFI->CreateFixedObject(-TailCallReturnAddrDelta, |
Tim Northover | ecc018c | 2013-08-04 09:35:57 +0000 | [diff] [blame] | 1254 | TailCallReturnAddrDelta - SlotSize, true); |
Anton Korobeynikov | 7283b8d | 2010-11-27 23:05:25 +0000 | [diff] [blame] | 1255 | } |
| 1256 | |
Chad Rosier | bdb08ac | 2012-07-10 17:45:53 +0000 | [diff] [blame] | 1257 | // Spill the BasePtr if it's used. |
| 1258 | if (RegInfo->hasBasePointer(MF)) |
| 1259 | MF.getRegInfo().setPhysRegUsed(RegInfo->getBaseRegister()); |
Anton Korobeynikov | 7283b8d | 2010-11-27 23:05:25 +0000 | [diff] [blame] | 1260 | } |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1261 | |
| 1262 | static bool |
| 1263 | HasNestArgument(const MachineFunction *MF) { |
| 1264 | const Function *F = MF->getFunction(); |
| 1265 | for (Function::const_arg_iterator I = F->arg_begin(), E = F->arg_end(); |
| 1266 | I != E; I++) { |
| 1267 | if (I->hasNestAttr()) |
| 1268 | return true; |
| 1269 | } |
| 1270 | return false; |
| 1271 | } |
| 1272 | |
Benjamin Kramer | 53bc37c | 2013-02-18 20:55:12 +0000 | [diff] [blame] | 1273 | /// GetScratchRegister - Get a temp register for performing work in the |
| 1274 | /// segmented stack and the Erlang/HiPE stack prologue. Depending on platform |
| 1275 | /// and the properties of the function either one or two registers will be |
| 1276 | /// needed. Set primary to true for the first register, false for the second. |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1277 | static unsigned |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1278 | GetScratchRegister(bool Is64Bit, const MachineFunction &MF, bool Primary) { |
Benjamin Kramer | 53bc37c | 2013-02-18 20:55:12 +0000 | [diff] [blame] | 1279 | CallingConv::ID CallingConvention = MF.getFunction()->getCallingConv(); |
| 1280 | |
| 1281 | // Erlang stuff. |
| 1282 | if (CallingConvention == CallingConv::HiPE) { |
| 1283 | if (Is64Bit) |
| 1284 | return Primary ? X86::R14 : X86::R13; |
| 1285 | else |
| 1286 | return Primary ? X86::EBX : X86::EDI; |
| 1287 | } |
| 1288 | |
David Blaikie | 46a9f01 | 2012-01-20 21:51:11 +0000 | [diff] [blame] | 1289 | if (Is64Bit) |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1290 | return Primary ? X86::R11 : X86::R12; |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1291 | |
David Blaikie | 46a9f01 | 2012-01-20 21:51:11 +0000 | [diff] [blame] | 1292 | bool IsNested = HasNestArgument(&MF); |
| 1293 | |
| 1294 | if (CallingConvention == CallingConv::X86_FastCall || |
| 1295 | CallingConvention == CallingConv::Fast) { |
| 1296 | if (IsNested) |
| 1297 | report_fatal_error("Segmented stacks does not support fastcall with " |
| 1298 | "nested function."); |
| 1299 | return Primary ? X86::EAX : X86::ECX; |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1300 | } |
David Blaikie | 46a9f01 | 2012-01-20 21:51:11 +0000 | [diff] [blame] | 1301 | if (IsNested) |
| 1302 | return Primary ? X86::EDX : X86::EAX; |
| 1303 | return Primary ? X86::ECX : X86::EAX; |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1304 | } |
| 1305 | |
Sanjoy Das | 006e43b | 2011-12-03 09:32:07 +0000 | [diff] [blame] | 1306 | // The stack limit in the TCB is set to this many bytes above the actual stack |
| 1307 | // limit. |
| 1308 | static const uint64_t kSplitStackAvailable = 256; |
| 1309 | |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1310 | void |
| 1311 | X86FrameLowering::adjustForSegmentedStacks(MachineFunction &MF) const { |
| 1312 | MachineBasicBlock &prologueMBB = MF.front(); |
| 1313 | MachineFrameInfo *MFI = MF.getFrameInfo(); |
Eric Christopher | 11b05cc | 2014-06-05 00:09:05 +0000 | [diff] [blame] | 1314 | const TargetInstrInfo &TII = *MF.getTarget().getInstrInfo(); |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1315 | uint64_t StackSize; |
Eric Christopher | f438164 | 2014-06-05 22:00:31 +0000 | [diff] [blame] | 1316 | const X86Subtarget &STI = MF.getTarget().getSubtarget<X86Subtarget>(); |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1317 | bool Is64Bit = STI.is64Bit(); |
| 1318 | unsigned TlsReg, TlsOffset; |
| 1319 | DebugLoc DL; |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1320 | |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1321 | unsigned ScratchReg = GetScratchRegister(Is64Bit, MF, true); |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1322 | assert(!MF.getRegInfo().isLiveIn(ScratchReg) && |
| 1323 | "Scratch register is live-in"); |
| 1324 | |
| 1325 | if (MF.getFunction()->isVarArg()) |
| 1326 | report_fatal_error("Segmented stacks do not support vararg functions."); |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1327 | if (!STI.isTargetLinux() && !STI.isTargetDarwin() && |
Reid Kleckner | 1011027 | 2014-04-01 18:34:21 +0000 | [diff] [blame] | 1328 | !STI.isTargetWin32() && !STI.isTargetWin64() && !STI.isTargetFreeBSD()) |
Rafael Espindola | 00e861e | 2012-01-12 20:24:30 +0000 | [diff] [blame] | 1329 | report_fatal_error("Segmented stacks not supported on this platform."); |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1330 | |
Tim Northover | f9e798b | 2014-05-22 13:03:43 +0000 | [diff] [blame] | 1331 | // Eventually StackSize will be calculated by a link-time pass; which will |
| 1332 | // also decide whether checking code needs to be injected into this particular |
| 1333 | // prologue. |
| 1334 | StackSize = MFI->getStackSize(); |
| 1335 | |
| 1336 | // Do not generate a prologue for functions with a stack of size zero |
| 1337 | if (StackSize == 0) |
| 1338 | return; |
| 1339 | |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1340 | MachineBasicBlock *allocMBB = MF.CreateMachineBasicBlock(); |
| 1341 | MachineBasicBlock *checkMBB = MF.CreateMachineBasicBlock(); |
| 1342 | X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); |
| 1343 | bool IsNested = false; |
| 1344 | |
| 1345 | // We need to know if the function has a nest argument only in 64 bit mode. |
| 1346 | if (Is64Bit) |
| 1347 | IsNested = HasNestArgument(&MF); |
| 1348 | |
Bill Wendling | 25f6d3e | 2011-10-13 08:24:19 +0000 | [diff] [blame] | 1349 | // The MOV R10, RAX needs to be in a different block, since the RET we emit in |
| 1350 | // allocMBB needs to be last (terminating) instruction. |
Bill Wendling | 25f6d3e | 2011-10-13 08:24:19 +0000 | [diff] [blame] | 1351 | |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1352 | for (MachineBasicBlock::livein_iterator i = prologueMBB.livein_begin(), |
| 1353 | e = prologueMBB.livein_end(); i != e; i++) { |
| 1354 | allocMBB->addLiveIn(*i); |
| 1355 | checkMBB->addLiveIn(*i); |
| 1356 | } |
| 1357 | |
| 1358 | if (IsNested) |
Rafael Espindola | 66393c1 | 2011-10-26 21:12:27 +0000 | [diff] [blame] | 1359 | allocMBB->addLiveIn(X86::R10); |
| 1360 | |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1361 | MF.push_front(allocMBB); |
| 1362 | MF.push_front(checkMBB); |
| 1363 | |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1364 | // When the frame size is less than 256 we just compare the stack |
| 1365 | // boundary directly to the value of the stack pointer, per gcc. |
| 1366 | bool CompareStackPointer = StackSize < kSplitStackAvailable; |
| 1367 | |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1368 | // Read the limit off the current stacklet off the stack_guard location. |
| 1369 | if (Is64Bit) { |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1370 | if (STI.isTargetLinux()) { |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1371 | TlsReg = X86::FS; |
| 1372 | TlsOffset = 0x70; |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1373 | } else if (STI.isTargetDarwin()) { |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1374 | TlsReg = X86::GS; |
| 1375 | TlsOffset = 0x60 + 90*8; // See pthread_machdep.h. Steal TLS slot 90. |
Reid Kleckner | 1011027 | 2014-04-01 18:34:21 +0000 | [diff] [blame] | 1376 | } else if (STI.isTargetWin64()) { |
| 1377 | TlsReg = X86::GS; |
| 1378 | TlsOffset = 0x28; // pvArbitrary, reserved for application use |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1379 | } else if (STI.isTargetFreeBSD()) { |
Rafael Espindola | 00e861e | 2012-01-12 20:24:30 +0000 | [diff] [blame] | 1380 | TlsReg = X86::FS; |
| 1381 | TlsOffset = 0x18; |
Rafael Espindola | 10745d3 | 2012-01-12 20:22:08 +0000 | [diff] [blame] | 1382 | } else { |
| 1383 | report_fatal_error("Segmented stacks not supported on this platform."); |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1384 | } |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1385 | |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1386 | if (CompareStackPointer) |
Sanjoy Das | 006e43b | 2011-12-03 09:32:07 +0000 | [diff] [blame] | 1387 | ScratchReg = X86::RSP; |
| 1388 | else |
| 1389 | BuildMI(checkMBB, DL, TII.get(X86::LEA64r), ScratchReg).addReg(X86::RSP) |
Rafael Espindola | 6635ae1 | 2012-01-11 18:14:03 +0000 | [diff] [blame] | 1390 | .addImm(1).addReg(0).addImm(-StackSize).addReg(0); |
Sanjoy Das | 006e43b | 2011-12-03 09:32:07 +0000 | [diff] [blame] | 1391 | |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1392 | BuildMI(checkMBB, DL, TII.get(X86::CMP64rm)).addReg(ScratchReg) |
Rafael Espindola | 6635ae1 | 2012-01-11 18:14:03 +0000 | [diff] [blame] | 1393 | .addReg(0).addImm(1).addReg(0).addImm(TlsOffset).addReg(TlsReg); |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1394 | } else { |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1395 | if (STI.isTargetLinux()) { |
Rafael Espindola | 10745d3 | 2012-01-12 20:22:08 +0000 | [diff] [blame] | 1396 | TlsReg = X86::GS; |
| 1397 | TlsOffset = 0x30; |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1398 | } else if (STI.isTargetDarwin()) { |
Rafael Espindola | 10745d3 | 2012-01-12 20:22:08 +0000 | [diff] [blame] | 1399 | TlsReg = X86::GS; |
| 1400 | TlsOffset = 0x48 + 90*4; |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1401 | } else if (STI.isTargetWin32()) { |
Rafael Espindola | 10745d3 | 2012-01-12 20:22:08 +0000 | [diff] [blame] | 1402 | TlsReg = X86::FS; |
| 1403 | TlsOffset = 0x14; // pvArbitrary, reserved for application use |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1404 | } else if (STI.isTargetFreeBSD()) { |
Rafael Espindola | 00e861e | 2012-01-12 20:24:30 +0000 | [diff] [blame] | 1405 | report_fatal_error("Segmented stacks not supported on FreeBSD i386."); |
Rafael Espindola | 10745d3 | 2012-01-12 20:22:08 +0000 | [diff] [blame] | 1406 | } else { |
| 1407 | report_fatal_error("Segmented stacks not supported on this platform."); |
| 1408 | } |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1409 | |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1410 | if (CompareStackPointer) |
Sanjoy Das | 006e43b | 2011-12-03 09:32:07 +0000 | [diff] [blame] | 1411 | ScratchReg = X86::ESP; |
| 1412 | else |
| 1413 | BuildMI(checkMBB, DL, TII.get(X86::LEA32r), ScratchReg).addReg(X86::ESP) |
Rafael Espindola | 6635ae1 | 2012-01-11 18:14:03 +0000 | [diff] [blame] | 1414 | .addImm(1).addReg(0).addImm(-StackSize).addReg(0); |
Sanjoy Das | 006e43b | 2011-12-03 09:32:07 +0000 | [diff] [blame] | 1415 | |
Reid Kleckner | 1011027 | 2014-04-01 18:34:21 +0000 | [diff] [blame] | 1416 | if (STI.isTargetLinux() || STI.isTargetWin32() || STI.isTargetWin64()) { |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1417 | BuildMI(checkMBB, DL, TII.get(X86::CMP32rm)).addReg(ScratchReg) |
| 1418 | .addReg(0).addImm(0).addReg(0).addImm(TlsOffset).addReg(TlsReg); |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1419 | } else if (STI.isTargetDarwin()) { |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1420 | |
Eric Christopher | 4237bf1 | 2014-04-29 00:16:33 +0000 | [diff] [blame] | 1421 | // TlsOffset doesn't fit into a mod r/m byte so we need an extra register. |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1422 | unsigned ScratchReg2; |
| 1423 | bool SaveScratch2; |
| 1424 | if (CompareStackPointer) { |
Eric Christopher | 4237bf1 | 2014-04-29 00:16:33 +0000 | [diff] [blame] | 1425 | // The primary scratch register is available for holding the TLS offset. |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1426 | ScratchReg2 = GetScratchRegister(Is64Bit, MF, true); |
| 1427 | SaveScratch2 = false; |
| 1428 | } else { |
| 1429 | // Need to use a second register to hold the TLS offset |
| 1430 | ScratchReg2 = GetScratchRegister(Is64Bit, MF, false); |
| 1431 | |
Eric Christopher | 4237bf1 | 2014-04-29 00:16:33 +0000 | [diff] [blame] | 1432 | // Unfortunately, with fastcc the second scratch register may hold an |
| 1433 | // argument. |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1434 | SaveScratch2 = MF.getRegInfo().isLiveIn(ScratchReg2); |
| 1435 | } |
| 1436 | |
Eric Christopher | 4237bf1 | 2014-04-29 00:16:33 +0000 | [diff] [blame] | 1437 | // If Scratch2 is live-in then it needs to be saved. |
Rafael Espindola | d90466b | 2012-01-11 19:00:37 +0000 | [diff] [blame] | 1438 | assert((!MF.getRegInfo().isLiveIn(ScratchReg2) || SaveScratch2) && |
| 1439 | "Scratch register is live-in and not saved"); |
| 1440 | |
| 1441 | if (SaveScratch2) |
| 1442 | BuildMI(checkMBB, DL, TII.get(X86::PUSH32r)) |
| 1443 | .addReg(ScratchReg2, RegState::Kill); |
| 1444 | |
| 1445 | BuildMI(checkMBB, DL, TII.get(X86::MOV32ri), ScratchReg2) |
| 1446 | .addImm(TlsOffset); |
| 1447 | BuildMI(checkMBB, DL, TII.get(X86::CMP32rm)) |
| 1448 | .addReg(ScratchReg) |
| 1449 | .addReg(ScratchReg2).addImm(1).addReg(0) |
| 1450 | .addImm(0) |
| 1451 | .addReg(TlsReg); |
| 1452 | |
| 1453 | if (SaveScratch2) |
| 1454 | BuildMI(checkMBB, DL, TII.get(X86::POP32r), ScratchReg2); |
| 1455 | } |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1456 | } |
| 1457 | |
| 1458 | // This jump is taken if SP >= (Stacklet Limit + Stack Space required). |
| 1459 | // It jumps to normal execution of the function body. |
Rafael Espindola | 2b89448 | 2012-01-11 18:23:35 +0000 | [diff] [blame] | 1460 | BuildMI(checkMBB, DL, TII.get(X86::JA_4)).addMBB(&prologueMBB); |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1461 | |
| 1462 | // On 32 bit we first push the arguments size and then the frame size. On 64 |
| 1463 | // bit, we pass the stack frame size in r10 and the argument size in r11. |
| 1464 | if (Is64Bit) { |
| 1465 | // Functions with nested arguments use R10, so it needs to be saved across |
| 1466 | // the call to _morestack |
| 1467 | |
| 1468 | if (IsNested) |
| 1469 | BuildMI(allocMBB, DL, TII.get(X86::MOV64rr), X86::RAX).addReg(X86::R10); |
| 1470 | |
| 1471 | BuildMI(allocMBB, DL, TII.get(X86::MOV64ri), X86::R10) |
| 1472 | .addImm(StackSize); |
| 1473 | BuildMI(allocMBB, DL, TII.get(X86::MOV64ri), X86::R11) |
| 1474 | .addImm(X86FI->getArgumentStackSize()); |
| 1475 | MF.getRegInfo().setPhysRegUsed(X86::R10); |
| 1476 | MF.getRegInfo().setPhysRegUsed(X86::R11); |
| 1477 | } else { |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1478 | BuildMI(allocMBB, DL, TII.get(X86::PUSHi32)) |
| 1479 | .addImm(X86FI->getArgumentStackSize()); |
| 1480 | BuildMI(allocMBB, DL, TII.get(X86::PUSHi32)) |
| 1481 | .addImm(StackSize); |
| 1482 | } |
| 1483 | |
| 1484 | // __morestack is in libgcc |
| 1485 | if (Is64Bit) |
| 1486 | BuildMI(allocMBB, DL, TII.get(X86::CALL64pcrel32)) |
| 1487 | .addExternalSymbol("__morestack"); |
| 1488 | else |
| 1489 | BuildMI(allocMBB, DL, TII.get(X86::CALLpcrel32)) |
| 1490 | .addExternalSymbol("__morestack"); |
| 1491 | |
Bill Wendling | 25f6d3e | 2011-10-13 08:24:19 +0000 | [diff] [blame] | 1492 | if (IsNested) |
Rafael Espindola | 66393c1 | 2011-10-26 21:12:27 +0000 | [diff] [blame] | 1493 | BuildMI(allocMBB, DL, TII.get(X86::MORESTACK_RET_RESTORE_R10)); |
| 1494 | else |
| 1495 | BuildMI(allocMBB, DL, TII.get(X86::MORESTACK_RET)); |
Bill Wendling | 25f6d3e | 2011-10-13 08:24:19 +0000 | [diff] [blame] | 1496 | |
Rafael Espindola | 66393c1 | 2011-10-26 21:12:27 +0000 | [diff] [blame] | 1497 | allocMBB->addSuccessor(&prologueMBB); |
Bill Wendling | 25f6d3e | 2011-10-13 08:24:19 +0000 | [diff] [blame] | 1498 | |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1499 | checkMBB->addSuccessor(allocMBB); |
| 1500 | checkMBB->addSuccessor(&prologueMBB); |
| 1501 | |
Jakob Stoklund Olesen | 55cf2ed | 2011-09-24 01:11:19 +0000 | [diff] [blame] | 1502 | #ifdef XDEBUG |
Rafael Espindola | c217421 | 2011-08-30 19:39:58 +0000 | [diff] [blame] | 1503 | MF.verify(); |
| 1504 | #endif |
| 1505 | } |
Benjamin Kramer | 53bc37c | 2013-02-18 20:55:12 +0000 | [diff] [blame] | 1506 | |
Yiannis Tsiouris | d4842e5 | 2013-02-28 16:59:10 +0000 | [diff] [blame] | 1507 | /// Erlang programs may need a special prologue to handle the stack size they |
| 1508 | /// might need at runtime. That is because Erlang/OTP does not implement a C |
| 1509 | /// stack but uses a custom implementation of hybrid stack/heap architecture. |
| 1510 | /// (for more information see Eric Stenman's Ph.D. thesis: |
| 1511 | /// http://publications.uu.se/uu/fulltext/nbn_se_uu_diva-2688.pdf) |
| 1512 | /// |
| 1513 | /// CheckStack: |
Eric Christopher | 4237bf1 | 2014-04-29 00:16:33 +0000 | [diff] [blame] | 1514 | /// temp0 = sp - MaxStack |
| 1515 | /// if( temp0 < SP_LIMIT(P) ) goto IncStack else goto OldStart |
Yiannis Tsiouris | d4842e5 | 2013-02-28 16:59:10 +0000 | [diff] [blame] | 1516 | /// OldStart: |
Eric Christopher | 4237bf1 | 2014-04-29 00:16:33 +0000 | [diff] [blame] | 1517 | /// ... |
Yiannis Tsiouris | d4842e5 | 2013-02-28 16:59:10 +0000 | [diff] [blame] | 1518 | /// IncStack: |
Eric Christopher | 4237bf1 | 2014-04-29 00:16:33 +0000 | [diff] [blame] | 1519 | /// call inc_stack # doubles the stack space |
| 1520 | /// temp0 = sp - MaxStack |
| 1521 | /// if( temp0 < SP_LIMIT(P) ) goto IncStack else goto OldStart |
Benjamin Kramer | 53bc37c | 2013-02-18 20:55:12 +0000 | [diff] [blame] | 1522 | void X86FrameLowering::adjustForHiPEPrologue(MachineFunction &MF) const { |
Eric Christopher | 11b05cc | 2014-06-05 00:09:05 +0000 | [diff] [blame] | 1523 | const TargetInstrInfo &TII = *MF.getTarget().getInstrInfo(); |
Benjamin Kramer | 53bc37c | 2013-02-18 20:55:12 +0000 | [diff] [blame] | 1524 | MachineFrameInfo *MFI = MF.getFrameInfo(); |
Eric Christopher | 11b05cc | 2014-06-05 00:09:05 +0000 | [diff] [blame] | 1525 | const unsigned SlotSize = |
| 1526 | static_cast<const X86RegisterInfo *>(MF.getTarget().getRegisterInfo()) |
| 1527 | ->getSlotSize(); |
Eric Christopher | f438164 | 2014-06-05 22:00:31 +0000 | [diff] [blame] | 1528 | const X86Subtarget &STI = MF.getTarget().getSubtarget<X86Subtarget>(); |
Benjamin Kramer | 53bc37c | 2013-02-18 20:55:12 +0000 | [diff] [blame] | 1529 | const bool Is64Bit = STI.is64Bit(); |
| 1530 | DebugLoc DL; |
| 1531 | // HiPE-specific values |
| 1532 | const unsigned HipeLeafWords = 24; |
| 1533 | const unsigned CCRegisteredArgs = Is64Bit ? 6 : 5; |
| 1534 | const unsigned Guaranteed = HipeLeafWords * SlotSize; |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1535 | unsigned CallerStkArity = MF.getFunction()->arg_size() > CCRegisteredArgs ? |
| 1536 | MF.getFunction()->arg_size() - CCRegisteredArgs : 0; |
| 1537 | unsigned MaxStack = MFI->getStackSize() + CallerStkArity*SlotSize + SlotSize; |
Benjamin Kramer | 53bc37c | 2013-02-18 20:55:12 +0000 | [diff] [blame] | 1538 | |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1539 | assert(STI.isTargetLinux() && |
Benjamin Kramer | 53bc37c | 2013-02-18 20:55:12 +0000 | [diff] [blame] | 1540 | "HiPE prologue is only supported on Linux operating systems."); |
| 1541 | |
| 1542 | // Compute the largest caller's frame that is needed to fit the callees' |
| 1543 | // frames. This 'MaxStack' is computed from: |
| 1544 | // |
| 1545 | // a) the fixed frame size, which is the space needed for all spilled temps, |
| 1546 | // b) outgoing on-stack parameter areas, and |
| 1547 | // c) the minimum stack space this function needs to make available for the |
| 1548 | // functions it calls (a tunable ABI property). |
| 1549 | if (MFI->hasCalls()) { |
| 1550 | unsigned MoreStackForCalls = 0; |
| 1551 | |
| 1552 | for (MachineFunction::iterator MBBI = MF.begin(), MBBE = MF.end(); |
| 1553 | MBBI != MBBE; ++MBBI) |
| 1554 | for (MachineBasicBlock::iterator MI = MBBI->begin(), ME = MBBI->end(); |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1555 | MI != ME; ++MI) { |
| 1556 | if (!MI->isCall()) |
| 1557 | continue; |
Benjamin Kramer | 53bc37c | 2013-02-18 20:55:12 +0000 | [diff] [blame] | 1558 | |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1559 | // Get callee operand. |
| 1560 | const MachineOperand &MO = MI->getOperand(0); |
Benjamin Kramer | 53bc37c | 2013-02-18 20:55:12 +0000 | [diff] [blame] | 1561 | |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1562 | // Only take account of global function calls (no closures etc.). |
| 1563 | if (!MO.isGlobal()) |
| 1564 | continue; |
Benjamin Kramer | 53bc37c | 2013-02-18 20:55:12 +0000 | [diff] [blame] | 1565 | |
Benjamin Kramer | 1cb826b | 2013-02-19 17:32:57 +0000 | [diff] [blame] | 1566 | const Function *F = dyn_cast<Function>(MO.getGlobal()); |
| 1567 | if (!F) |
| 1568 | continue; |
| 1569 | |
| 1570 | // Do not update 'MaxStack' for primitive and built-in functions |
| 1571 | // (encoded with names either starting with "erlang."/"bif_" or not |
| 1572 | // having a ".", such as a simple <Module>.<Function>.<Arity>, or an |
| 1573 | // "_", such as the BIF "suspend_0") as they are executed on another |
| 1574 | // stack. |
| 1575 | if (F->getName().find("erlang.") != StringRef::npos || |
| 1576 | F->getName().find("bif_") != StringRef::npos || |
| 1577 | F->getName().find_first_of("._") == StringRef::npos) |
| 1578 | continue; |
| 1579 | |
| 1580 | unsigned CalleeStkArity = |
| 1581 | F->arg_size() > CCRegisteredArgs ? F->arg_size()-CCRegisteredArgs : 0; |
| 1582 | if (HipeLeafWords - 1 > CalleeStkArity) |
| 1583 | MoreStackForCalls = std::max(MoreStackForCalls, |
| 1584 | (HipeLeafWords - 1 - CalleeStkArity) * SlotSize); |
| 1585 | } |
Benjamin Kramer | 53bc37c | 2013-02-18 20:55:12 +0000 | [diff] [blame] | 1586 | MaxStack += MoreStackForCalls; |
| 1587 | } |
| 1588 | |
| 1589 | // If the stack frame needed is larger than the guaranteed then runtime checks |
| 1590 | // and calls to "inc_stack_0" BIF should be inserted in the assembly prologue. |
| 1591 | if (MaxStack > Guaranteed) { |
| 1592 | MachineBasicBlock &prologueMBB = MF.front(); |
| 1593 | MachineBasicBlock *stackCheckMBB = MF.CreateMachineBasicBlock(); |
| 1594 | MachineBasicBlock *incStackMBB = MF.CreateMachineBasicBlock(); |
| 1595 | |
| 1596 | for (MachineBasicBlock::livein_iterator I = prologueMBB.livein_begin(), |
| 1597 | E = prologueMBB.livein_end(); I != E; I++) { |
| 1598 | stackCheckMBB->addLiveIn(*I); |
| 1599 | incStackMBB->addLiveIn(*I); |
| 1600 | } |
| 1601 | |
| 1602 | MF.push_front(incStackMBB); |
| 1603 | MF.push_front(stackCheckMBB); |
| 1604 | |
| 1605 | unsigned ScratchReg, SPReg, PReg, SPLimitOffset; |
| 1606 | unsigned LEAop, CMPop, CALLop; |
| 1607 | if (Is64Bit) { |
| 1608 | SPReg = X86::RSP; |
| 1609 | PReg = X86::RBP; |
| 1610 | LEAop = X86::LEA64r; |
| 1611 | CMPop = X86::CMP64rm; |
| 1612 | CALLop = X86::CALL64pcrel32; |
| 1613 | SPLimitOffset = 0x90; |
| 1614 | } else { |
| 1615 | SPReg = X86::ESP; |
| 1616 | PReg = X86::EBP; |
| 1617 | LEAop = X86::LEA32r; |
| 1618 | CMPop = X86::CMP32rm; |
| 1619 | CALLop = X86::CALLpcrel32; |
| 1620 | SPLimitOffset = 0x4c; |
| 1621 | } |
| 1622 | |
| 1623 | ScratchReg = GetScratchRegister(Is64Bit, MF, true); |
| 1624 | assert(!MF.getRegInfo().isLiveIn(ScratchReg) && |
| 1625 | "HiPE prologue scratch register is live-in"); |
| 1626 | |
| 1627 | // Create new MBB for StackCheck: |
| 1628 | addRegOffset(BuildMI(stackCheckMBB, DL, TII.get(LEAop), ScratchReg), |
| 1629 | SPReg, false, -MaxStack); |
| 1630 | // SPLimitOffset is in a fixed heap location (pointed by BP). |
| 1631 | addRegOffset(BuildMI(stackCheckMBB, DL, TII.get(CMPop)) |
| 1632 | .addReg(ScratchReg), PReg, false, SPLimitOffset); |
| 1633 | BuildMI(stackCheckMBB, DL, TII.get(X86::JAE_4)).addMBB(&prologueMBB); |
| 1634 | |
| 1635 | // Create new MBB for IncStack: |
| 1636 | BuildMI(incStackMBB, DL, TII.get(CALLop)). |
| 1637 | addExternalSymbol("inc_stack_0"); |
| 1638 | addRegOffset(BuildMI(incStackMBB, DL, TII.get(LEAop), ScratchReg), |
| 1639 | SPReg, false, -MaxStack); |
| 1640 | addRegOffset(BuildMI(incStackMBB, DL, TII.get(CMPop)) |
| 1641 | .addReg(ScratchReg), PReg, false, SPLimitOffset); |
| 1642 | BuildMI(incStackMBB, DL, TII.get(X86::JLE_4)).addMBB(incStackMBB); |
| 1643 | |
| 1644 | stackCheckMBB->addSuccessor(&prologueMBB, 99); |
| 1645 | stackCheckMBB->addSuccessor(incStackMBB, 1); |
| 1646 | incStackMBB->addSuccessor(&prologueMBB, 99); |
| 1647 | incStackMBB->addSuccessor(incStackMBB, 1); |
| 1648 | } |
| 1649 | #ifdef XDEBUG |
| 1650 | MF.verify(); |
| 1651 | #endif |
| 1652 | } |
Eli Bendersky | 8da8716 | 2013-02-21 20:05:00 +0000 | [diff] [blame] | 1653 | |
| 1654 | void X86FrameLowering:: |
| 1655 | eliminateCallFramePseudoInstr(MachineFunction &MF, MachineBasicBlock &MBB, |
| 1656 | MachineBasicBlock::iterator I) const { |
Eric Christopher | 11b05cc | 2014-06-05 00:09:05 +0000 | [diff] [blame] | 1657 | const TargetInstrInfo &TII = *MF.getTarget().getInstrInfo(); |
| 1658 | const X86RegisterInfo &RegInfo = |
| 1659 | *static_cast<const X86RegisterInfo *>(MF.getTarget().getRegisterInfo()); |
Eli Bendersky | 8da8716 | 2013-02-21 20:05:00 +0000 | [diff] [blame] | 1660 | unsigned StackPtr = RegInfo.getStackRegister(); |
| 1661 | bool reseveCallFrame = hasReservedCallFrame(MF); |
| 1662 | int Opcode = I->getOpcode(); |
| 1663 | bool isDestroy = Opcode == TII.getCallFrameDestroyOpcode(); |
Eric Christopher | f438164 | 2014-06-05 22:00:31 +0000 | [diff] [blame] | 1664 | const X86Subtarget &STI = MF.getTarget().getSubtarget<X86Subtarget>(); |
Eli Bendersky | 8da8716 | 2013-02-21 20:05:00 +0000 | [diff] [blame] | 1665 | bool IsLP64 = STI.isTarget64BitLP64(); |
| 1666 | DebugLoc DL = I->getDebugLoc(); |
| 1667 | uint64_t Amount = !reseveCallFrame ? I->getOperand(0).getImm() : 0; |
| 1668 | uint64_t CalleeAmt = isDestroy ? I->getOperand(1).getImm() : 0; |
| 1669 | I = MBB.erase(I); |
| 1670 | |
| 1671 | if (!reseveCallFrame) { |
| 1672 | // If the stack pointer can be changed after prologue, turn the |
| 1673 | // adjcallstackup instruction into a 'sub ESP, <amt>' and the |
| 1674 | // adjcallstackdown instruction into 'add ESP, <amt>' |
| 1675 | // TODO: consider using push / pop instead of sub + store / add |
| 1676 | if (Amount == 0) |
| 1677 | return; |
| 1678 | |
| 1679 | // We need to keep the stack aligned properly. To do this, we round the |
| 1680 | // amount of space needed for the outgoing arguments up to the next |
| 1681 | // alignment boundary. |
Eric Christopher | 52fa659 | 2014-06-05 00:09:08 +0000 | [diff] [blame] | 1682 | unsigned StackAlign = |
| 1683 | MF.getTarget().getFrameLowering()->getStackAlignment(); |
Eli Bendersky | 8da8716 | 2013-02-21 20:05:00 +0000 | [diff] [blame] | 1684 | Amount = (Amount + StackAlign - 1) / StackAlign * StackAlign; |
| 1685 | |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 1686 | MachineInstr *New = nullptr; |
Eli Bendersky | 8da8716 | 2013-02-21 20:05:00 +0000 | [diff] [blame] | 1687 | if (Opcode == TII.getCallFrameSetupOpcode()) { |
| 1688 | New = BuildMI(MF, DL, TII.get(getSUBriOpcode(IsLP64, Amount)), |
| 1689 | StackPtr) |
| 1690 | .addReg(StackPtr) |
| 1691 | .addImm(Amount); |
| 1692 | } else { |
| 1693 | assert(Opcode == TII.getCallFrameDestroyOpcode()); |
| 1694 | |
| 1695 | // Factor out the amount the callee already popped. |
| 1696 | Amount -= CalleeAmt; |
| 1697 | |
| 1698 | if (Amount) { |
| 1699 | unsigned Opc = getADDriOpcode(IsLP64, Amount); |
| 1700 | New = BuildMI(MF, DL, TII.get(Opc), StackPtr) |
| 1701 | .addReg(StackPtr).addImm(Amount); |
| 1702 | } |
| 1703 | } |
| 1704 | |
| 1705 | if (New) { |
| 1706 | // The EFLAGS implicit def is dead. |
| 1707 | New->getOperand(3).setIsDead(); |
| 1708 | |
| 1709 | // Replace the pseudo instruction with a new instruction. |
| 1710 | MBB.insert(I, New); |
| 1711 | } |
| 1712 | |
| 1713 | return; |
| 1714 | } |
| 1715 | |
| 1716 | if (Opcode == TII.getCallFrameDestroyOpcode() && CalleeAmt) { |
| 1717 | // If we are performing frame pointer elimination and if the callee pops |
| 1718 | // something off the stack pointer, add it back. We do this until we have |
| 1719 | // more advanced stack pointer tracking ability. |
| 1720 | unsigned Opc = getSUBriOpcode(IsLP64, CalleeAmt); |
| 1721 | MachineInstr *New = BuildMI(MF, DL, TII.get(Opc), StackPtr) |
| 1722 | .addReg(StackPtr).addImm(CalleeAmt); |
| 1723 | |
| 1724 | // The EFLAGS implicit def is dead. |
| 1725 | New->getOperand(3).setIsDead(); |
| 1726 | |
| 1727 | // We are not tracking the stack pointer adjustment by the callee, so make |
| 1728 | // sure we restore the stack pointer immediately after the call, there may |
| 1729 | // be spill code inserted between the CALL and ADJCALLSTACKUP instructions. |
| 1730 | MachineBasicBlock::iterator B = MBB.begin(); |
Benjamin Kramer | b6d0bd4 | 2014-03-02 12:27:27 +0000 | [diff] [blame] | 1731 | while (I != B && !std::prev(I)->isCall()) |
Eli Bendersky | 8da8716 | 2013-02-21 20:05:00 +0000 | [diff] [blame] | 1732 | --I; |
| 1733 | MBB.insert(I, New); |
| 1734 | } |
| 1735 | } |
| 1736 | |