X86FrameLowering.cpp revision e4d18de5d15c6c1d4d279788b35d0cd1ab237e82
13c2f0a11cce5a1e828e20675fa8467b624795e0aNick Lewycky//=======- X86FrameLowering.cpp - X86 Frame Information --------*- C++ -*-====// 233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov// 333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov// The LLVM Compiler Infrastructure 433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov// 533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov// This file is distributed under the University of Illinois Open Source 633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov// License. See LICENSE.TXT for details. 733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov// 833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov//===----------------------------------------------------------------------===// 933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov// 1016c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikov// This file contains the X86 implementation of TargetFrameLowering class. 1133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov// 1233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov//===----------------------------------------------------------------------===// 1333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 1416c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikov#include "X86FrameLowering.h" 1533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov#include "X86InstrBuilder.h" 1633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov#include "X86InstrInfo.h" 1733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov#include "X86MachineFunctionInfo.h" 1876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola#include "X86Subtarget.h" 19d9e3385ced2dc887e2fe8e1c071bd2611e4d3edeAnton Korobeynikov#include "X86TargetMachine.h" 2033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov#include "llvm/Function.h" 2133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov#include "llvm/CodeGen/MachineFrameInfo.h" 2233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov#include "llvm/CodeGen/MachineFunction.h" 2333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov#include "llvm/CodeGen/MachineInstrBuilder.h" 2433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov#include "llvm/CodeGen/MachineModuleInfo.h" 2533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov#include "llvm/CodeGen/MachineRegisterInfo.h" 26f0adba9a7ec8a3031876575a6ffb7db5f1b6f855Rafael Espindola#include "llvm/MC/MCAsmInfo.h" 276a6b8c3e96b9e1ca7092eafd0cfb219cbbfbdfc4Bill Wendling#include "llvm/MC/MCSymbol.h" 2833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov#include "llvm/Target/TargetData.h" 2933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov#include "llvm/Target/TargetOptions.h" 3033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov#include "llvm/Support/CommandLine.h" 317158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng#include "llvm/ADT/SmallSet.h" 3233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 3333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikovusing namespace llvm; 3433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 3533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov// FIXME: completely move here. 3633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikovextern cl::opt<bool> ForceStackAlign; 3733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 3816c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikovbool X86FrameLowering::hasReservedCallFrame(const MachineFunction &MF) const { 39d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov return !MF.getFrameInfo()->hasVarSizedObjects(); 40d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov} 41d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov 42d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov/// hasFP - Return true if the specified function should have a dedicated frame 43d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov/// pointer register. This is true if the function has variable sized allocas 44d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov/// or if frame pointer elimination is disabled. 4516c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikovbool X86FrameLowering::hasFP(const MachineFunction &MF) const { 46d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov const MachineFrameInfo *MFI = MF.getFrameInfo(); 47d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov const MachineModuleInfo &MMI = MF.getMMI(); 48d9e3385ced2dc887e2fe8e1c071bd2611e4d3edeAnton Korobeynikov const TargetRegisterInfo *RI = TM.getRegisterInfo(); 49d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov 508a8d479214745c82ef00f08d4e4f1c173b5f9ce2Nick Lewycky return (MF.getTarget().Options.DisableFramePointerElim(MF) || 51d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov RI->needsStackRealignment(MF) || 52d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov MFI->hasVarSizedObjects() || 53d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov MFI->isFrameAddressTaken() || 54d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov MF.getInfo<X86MachineFunctionInfo>()->getForceFramePointer() || 55d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov MMI.callsUnwindInit()); 56d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov} 57d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov 5833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikovstatic unsigned getSUBriOpcode(unsigned is64Bit, int64_t Imm) { 5933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (is64Bit) { 6033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (isInt<8>(Imm)) 6133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov return X86::SUB64ri8; 6233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov return X86::SUB64ri32; 6333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 6433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (isInt<8>(Imm)) 6533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov return X86::SUB32ri8; 6633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov return X86::SUB32ri; 6733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 6833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov} 6933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 7033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikovstatic unsigned getADDriOpcode(unsigned is64Bit, int64_t Imm) { 7133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (is64Bit) { 7233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (isInt<8>(Imm)) 7333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov return X86::ADD64ri8; 7433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov return X86::ADD64ri32; 7533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 7633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (isInt<8>(Imm)) 7733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov return X86::ADD32ri8; 7833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov return X86::ADD32ri; 7933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 8033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov} 8133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 827158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng/// findDeadCallerSavedReg - Return a caller-saved register that isn't live 837158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng/// when it reaches the "return" instruction. We can then pop a stack object 847158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng/// to this register without worry about clobbering it. 857158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Chengstatic unsigned findDeadCallerSavedReg(MachineBasicBlock &MBB, 867158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng MachineBasicBlock::iterator &MBBI, 877158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng const TargetRegisterInfo &TRI, 887158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng bool Is64Bit) { 897158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng const MachineFunction *MF = MBB.getParent(); 907158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng const Function *F = MF->getFunction(); 917158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng if (!F || MF->getMMI().callsEHReturn()) 927158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng return 0; 937158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng 947158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng static const unsigned CallerSavedRegs32Bit[] = { 9532a183c84ad0cbe492119c37f1a7941ace61dd79Andrew Trick X86::EAX, X86::EDX, X86::ECX, 0 967158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng }; 977158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng 987158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng static const unsigned CallerSavedRegs64Bit[] = { 997158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng X86::RAX, X86::RDX, X86::RCX, X86::RSI, X86::RDI, 10032a183c84ad0cbe492119c37f1a7941ace61dd79Andrew Trick X86::R8, X86::R9, X86::R10, X86::R11, 0 1017158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng }; 1027158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng 1037158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng unsigned Opc = MBBI->getOpcode(); 1047158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng switch (Opc) { 1057158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng default: return 0; 1067158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng case X86::RET: 1077158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng case X86::RETI: 1087158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng case X86::TCRETURNdi: 1097158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng case X86::TCRETURNri: 1107158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng case X86::TCRETURNmi: 1117158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng case X86::TCRETURNdi64: 1127158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng case X86::TCRETURNri64: 1137158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng case X86::TCRETURNmi64: 1147158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng case X86::EH_RETURN: 1157158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng case X86::EH_RETURN64: { 1167158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng SmallSet<unsigned, 8> Uses; 1177158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng for (unsigned i = 0, e = MBBI->getNumOperands(); i != e; ++i) { 1187158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng MachineOperand &MO = MBBI->getOperand(i); 1197158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng if (!MO.isReg() || MO.isDef()) 1207158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng continue; 1217158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng unsigned Reg = MO.getReg(); 1227158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng if (!Reg) 1237158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng continue; 1247158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng for (const unsigned *AsI = TRI.getOverlaps(Reg); *AsI; ++AsI) 1257158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng Uses.insert(*AsI); 1267158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng } 1277158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng 1287158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng const unsigned *CS = Is64Bit ? CallerSavedRegs64Bit : CallerSavedRegs32Bit; 1297158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng for (; *CS; ++CS) 1307158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng if (!Uses.count(*CS)) 1317158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng return *CS; 1327158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng } 1337158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng } 1347158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng 1357158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng return 0; 1367158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng} 1377158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng 1387158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng 13933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// emitSPUpdate - Emit a series of instructions to increment / decrement the 14033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// stack pointer by a constant value. 14133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikovstatic 14233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikovvoid emitSPUpdate(MachineBasicBlock &MBB, MachineBasicBlock::iterator &MBBI, 1437158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng unsigned StackPtr, int64_t NumBytes, 1447158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng bool Is64Bit, const TargetInstrInfo &TII, 1457158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng const TargetRegisterInfo &TRI) { 14633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool isSub = NumBytes < 0; 14733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t Offset = isSub ? -NumBytes : NumBytes; 14833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned Opc = isSub ? 14933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov getSUBriOpcode(Is64Bit, Offset) : 15033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov getADDriOpcode(Is64Bit, Offset); 15133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t Chunk = (1LL << 31) - 1; 15233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov DebugLoc DL = MBB.findDebugLoc(MBBI); 15333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 15433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov while (Offset) { 15533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t ThisVal = (Offset > Chunk) ? Chunk : Offset; 1567158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng if (ThisVal == (Is64Bit ? 8 : 4)) { 1577158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng // Use push / pop instead. 1587158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng unsigned Reg = isSub 1591e08cd1eaef8acbcfaf7db48d859a29583c29897Dale Johannesen ? (unsigned)(Is64Bit ? X86::RAX : X86::EAX) 1607158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng : findDeadCallerSavedReg(MBB, MBBI, TRI, Is64Bit); 1617158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng if (Reg) { 1627158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng Opc = isSub 1637158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng ? (Is64Bit ? X86::PUSH64r : X86::PUSH32r) 1647158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng : (Is64Bit ? X86::POP64r : X86::POP32r); 165aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis MachineInstr *MI = BuildMI(MBB, MBBI, DL, TII.get(Opc)) 1667158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng .addReg(Reg, getDefRegState(!isSub) | getUndefRegState(isSub)); 167aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis if (isSub) 168aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis MI->setFlag(MachineInstr::FrameSetup); 1697158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng Offset -= ThisVal; 1707158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng continue; 1717158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng } 1727158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng } 1737158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng 17433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineInstr *MI = 17533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get(Opc), StackPtr) 1767158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng .addReg(StackPtr) 1777158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng .addImm(ThisVal); 178aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis if (isSub) 179aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis MI->setFlag(MachineInstr::FrameSetup); 18033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MI->getOperand(3).setIsDead(); // The EFLAGS implicit def is dead. 18133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Offset -= ThisVal; 18233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 18333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov} 18433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 18533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// mergeSPUpdatesUp - Merge two stack-manipulating instructions upper iterator. 18633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikovstatic 18733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikovvoid mergeSPUpdatesUp(MachineBasicBlock &MBB, MachineBasicBlock::iterator &MBBI, 18833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned StackPtr, uint64_t *NumBytes = NULL) { 18933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (MBBI == MBB.begin()) return; 19033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 19133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock::iterator PI = prior(MBBI); 19233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned Opc = PI->getOpcode(); 19333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if ((Opc == X86::ADD64ri32 || Opc == X86::ADD64ri8 || 19433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Opc == X86::ADD32ri || Opc == X86::ADD32ri8) && 19533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov PI->getOperand(0).getReg() == StackPtr) { 19633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (NumBytes) 19733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov *NumBytes += PI->getOperand(2).getImm(); 19833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBB.erase(PI); 19933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if ((Opc == X86::SUB64ri32 || Opc == X86::SUB64ri8 || 20033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Opc == X86::SUB32ri || Opc == X86::SUB32ri8) && 20133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov PI->getOperand(0).getReg() == StackPtr) { 20233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (NumBytes) 20333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov *NumBytes -= PI->getOperand(2).getImm(); 20433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBB.erase(PI); 20533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 20633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov} 20733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 20833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// mergeSPUpdatesDown - Merge two stack-manipulating instructions lower iterator. 20933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikovstatic 21033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikovvoid mergeSPUpdatesDown(MachineBasicBlock &MBB, 21133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock::iterator &MBBI, 21233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned StackPtr, uint64_t *NumBytes = NULL) { 213fc9261279aa140542cf9b7c2c384d000ad97aca0Sanjoy Das // FIXME: THIS ISN'T RUN!!! 21433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov return; 21533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 21633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (MBBI == MBB.end()) return; 21733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 21833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock::iterator NI = llvm::next(MBBI); 21933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (NI == MBB.end()) return; 22033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 22133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned Opc = NI->getOpcode(); 22233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if ((Opc == X86::ADD64ri32 || Opc == X86::ADD64ri8 || 22333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Opc == X86::ADD32ri || Opc == X86::ADD32ri8) && 22433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NI->getOperand(0).getReg() == StackPtr) { 22533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (NumBytes) 22633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov *NumBytes -= NI->getOperand(2).getImm(); 22733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBB.erase(NI); 22833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBBI = NI; 22933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if ((Opc == X86::SUB64ri32 || Opc == X86::SUB64ri8 || 23033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Opc == X86::SUB32ri || Opc == X86::SUB32ri8) && 23133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NI->getOperand(0).getReg() == StackPtr) { 23233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (NumBytes) 23333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov *NumBytes += NI->getOperand(2).getImm(); 23433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBB.erase(NI); 23533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBBI = NI; 23633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 23733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov} 23833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 23933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// mergeSPUpdates - Checks the instruction before/after the passed 24033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// instruction. If it is an ADD/SUB instruction it is deleted argument and the 24133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// stack adjustment is returned as a positive value for ADD and a negative for 24233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// SUB. 24333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikovstatic int mergeSPUpdates(MachineBasicBlock &MBB, 24433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock::iterator &MBBI, 24533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned StackPtr, 24633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool doMergeWithPrevious) { 24733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if ((doMergeWithPrevious && MBBI == MBB.begin()) || 24833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov (!doMergeWithPrevious && MBBI == MBB.end())) 24933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov return 0; 25033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 25133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock::iterator PI = doMergeWithPrevious ? prior(MBBI) : MBBI; 25233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock::iterator NI = doMergeWithPrevious ? 0 : llvm::next(MBBI); 25333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned Opc = PI->getOpcode(); 25433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int Offset = 0; 25533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 25633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if ((Opc == X86::ADD64ri32 || Opc == X86::ADD64ri8 || 25733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Opc == X86::ADD32ri || Opc == X86::ADD32ri8) && 25833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov PI->getOperand(0).getReg() == StackPtr){ 25933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Offset += PI->getOperand(2).getImm(); 26033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBB.erase(PI); 26133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (!doMergeWithPrevious) MBBI = NI; 26233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if ((Opc == X86::SUB64ri32 || Opc == X86::SUB64ri8 || 26333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Opc == X86::SUB32ri || Opc == X86::SUB32ri8) && 26433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov PI->getOperand(0).getReg() == StackPtr) { 26533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Offset -= PI->getOperand(2).getImm(); 26633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBB.erase(PI); 26733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (!doMergeWithPrevious) MBBI = NI; 26833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 26933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 27033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov return Offset; 27133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov} 27233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 27333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikovstatic bool isEAXLiveIn(MachineFunction &MF) { 27433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov for (MachineRegisterInfo::livein_iterator II = MF.getRegInfo().livein_begin(), 27533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov EE = MF.getRegInfo().livein_end(); II != EE; ++II) { 27633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned Reg = II->first; 27733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 27833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (Reg == X86::EAX || Reg == X86::AX || 27933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Reg == X86::AH || Reg == X86::AL) 28033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov return true; 28133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 28233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 28333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov return false; 28433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov} 28533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 28616c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikovvoid X86FrameLowering::emitCalleeSavedFrameMoves(MachineFunction &MF, 28709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling MCSymbol *Label, 28809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned FramePtr) const { 28933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineFrameInfo *MFI = MF.getFrameInfo(); 29033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineModuleInfo &MMI = MF.getMMI(); 29133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 29233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Add callee saved registers to move list. 29333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov const std::vector<CalleeSavedInfo> &CSI = MFI->getCalleeSavedInfo(); 29433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (CSI.empty()) return; 29533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 29633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov std::vector<MachineMove> &Moves = MMI.getFrameMoves(); 297d9e3385ced2dc887e2fe8e1c071bd2611e4d3edeAnton Korobeynikov const TargetData *TD = TM.getTargetData(); 298d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov bool HasFP = hasFP(MF); 29933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 30033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Calculate amount of bytes used for return address storing. 301e749911372c75e4e68f83a32b4092f4ffb0d2793Anton Korobeynikov int stackGrowth = -TD->getPointerSize(); 30233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 30333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // FIXME: This is dirty hack. The code itself is pretty mess right now. 30433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // It should be rewritten from scratch and generalized sometimes. 30533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 3067a2bdde0a0eebcd2125055e0eacaca040f0b766cChris Lattner // Determine maximum offset (minimum due to stack growth). 30733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int64_t MaxOffset = 0; 30833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov for (std::vector<CalleeSavedInfo>::const_iterator 30933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov I = CSI.begin(), E = CSI.end(); I != E; ++I) 31033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MaxOffset = std::min(MaxOffset, 31133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MFI->getObjectOffset(I->getFrameIdx())); 31233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 31333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Calculate offsets. 31433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int64_t saveAreaOffset = (HasFP ? 3 : 2) * stackGrowth; 31533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov for (std::vector<CalleeSavedInfo>::const_iterator 31633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov I = CSI.begin(), E = CSI.end(); I != E; ++I) { 31733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int64_t Offset = MFI->getObjectOffset(I->getFrameIdx()); 31833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned Reg = I->getReg(); 31933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Offset = MaxOffset - Offset + saveAreaOffset; 32033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 32133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Don't output a new machine move if we're re-saving the frame 32233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // pointer. This happens when the PrologEpilogInserter has inserted an extra 32333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // "PUSH" of the frame pointer -- the "emitPrologue" method automatically 32433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // generates one when frame pointers are used. If we generate a "machine 32533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // move" for this extra "PUSH", the linker will lose track of the fact that 32633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // the frame pointer should have the value of the first "PUSH" when it's 32733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // trying to unwind. 3282763538609fd455d63c192b320c73fb5d48c3e47NAKAMURA Takumi // 32933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // FIXME: This looks inelegant. It's possibly correct, but it's covering up 33033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // another bug. I.e., one where we generate a prolog like this: 33133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // 33233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // pushl %ebp 33333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // movl %esp, %ebp 33433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // pushl %ebp 33533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // pushl %esi 33633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // ... 33733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // 33833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // The immediate re-push of EBP is unnecessary. At the least, it's an 33933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // optimization bug. EBP can be used as a scratch register in certain 34033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // cases, but probably not when we have a frame pointer. 34133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (HasFP && FramePtr == Reg) 34233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov continue; 34333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 34433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation CSDst(MachineLocation::VirtualFP, Offset); 34533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation CSSrc(Reg); 34633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(Label, CSDst, CSSrc)); 34733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 34833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov} 34933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 35009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling/// getCompactUnwindRegNum - Get the compact unwind number for a given 35109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling/// register. The number corresponds to the enum lists in 35209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling/// compact_unwind_encoding.h. 35309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendlingstatic int getCompactUnwindRegNum(const unsigned *CURegs, unsigned Reg) { 35410e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling for (int Idx = 1; *CURegs; ++CURegs, ++Idx) 35509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (*CURegs == Reg) 35609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return Idx; 35709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 35809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return -1; 35909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling} 36009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 36157a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling// Number of registers that can be saved in a compact unwind encoding. 36257a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling#define CU_NUM_SAVED_REGS 6 36357a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling 36409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling/// encodeCompactUnwindRegistersWithoutFrame - Create the permutation encoding 36509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling/// used with frameless stacks. It is passed the number of registers to be saved 36609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling/// and an array of the registers saved. 36757a3cd2f326269dd1431cf430db50239a93450f2Bill Wendlingstatic uint32_t 36857a3cd2f326269dd1431cf430db50239a93450f2Bill WendlingencodeCompactUnwindRegistersWithoutFrame(unsigned SavedRegs[CU_NUM_SAVED_REGS], 36957a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling unsigned RegCount, bool Is64Bit) { 37009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // The saved registers are numbered from 1 to 6. In order to encode the order 37109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // in which they were saved, we re-number them according to their place in the 37209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // register order. The re-numbering is relative to the last re-numbered 37309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // register. E.g., if we have registers {6, 2, 4, 5} saved in that order: 37409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // 37509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Orig Re-Num 37609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // ---- ------ 37709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // 6 6 37809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // 2 2 37909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // 4 3 38009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // 5 3 38109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // 38209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling static const unsigned CU32BitRegs[] = { 38309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling X86::EBX, X86::ECX, X86::EDX, X86::EDI, X86::ESI, X86::EBP, 0 38409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling }; 38509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling static const unsigned CU64BitRegs[] = { 38609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling X86::RBX, X86::R12, X86::R13, X86::R14, X86::R15, X86::RBP, 0 38709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling }; 38809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling const unsigned *CURegs = (Is64Bit ? CU64BitRegs : CU32BitRegs); 38909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 39010e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling for (unsigned i = 0; i != CU_NUM_SAVED_REGS; ++i) { 39109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling int CUReg = getCompactUnwindRegNum(CURegs, SavedRegs[i]); 39209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (CUReg == -1) return ~0U; 39309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling SavedRegs[i] = CUReg; 39479df986c60094e9ea29f081295aea08b1680a999Bill Wendling } 39509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 39610e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling // Reverse the list. 39710e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling std::swap(SavedRegs[0], SavedRegs[5]); 39810e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling std::swap(SavedRegs[1], SavedRegs[4]); 39910e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling std::swap(SavedRegs[2], SavedRegs[3]); 40010e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling 40157a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling uint32_t RenumRegs[CU_NUM_SAVED_REGS]; 40257a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling for (unsigned i = CU_NUM_SAVED_REGS - RegCount; i < CU_NUM_SAVED_REGS; ++i) { 40309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned Countless = 0; 40457a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling for (unsigned j = CU_NUM_SAVED_REGS - RegCount; j < i; ++j) 40509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (SavedRegs[j] < SavedRegs[i]) 40609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling ++Countless; 40709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 40809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling RenumRegs[i] = SavedRegs[i] - Countless - 1; 40909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } 41009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 41109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Take the renumbered values and encode them into a 10-bit number. 41209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling uint32_t permutationEncoding = 0; 41309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling switch (RegCount) { 41409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling case 6: 41509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling permutationEncoding |= 120 * RenumRegs[0] + 24 * RenumRegs[1] 41609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling + 6 * RenumRegs[2] + 2 * RenumRegs[3] 41709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling + RenumRegs[4]; 41809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling break; 41909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling case 5: 42009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling permutationEncoding |= 120 * RenumRegs[1] + 24 * RenumRegs[2] 42109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling + 6 * RenumRegs[3] + 2 * RenumRegs[4] 42209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling + RenumRegs[5]; 42309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling break; 42409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling case 4: 42509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling permutationEncoding |= 60 * RenumRegs[2] + 12 * RenumRegs[3] 42609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling + 3 * RenumRegs[4] + RenumRegs[5]; 42709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling break; 42809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling case 3: 42909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling permutationEncoding |= 20 * RenumRegs[3] + 4 * RenumRegs[4] 43009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling + RenumRegs[5]; 43109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling break; 43209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling case 2: 43309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling permutationEncoding |= 5 * RenumRegs[4] + RenumRegs[5]; 43409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling break; 43509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling case 1: 43609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling permutationEncoding |= RenumRegs[5]; 43709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling break; 43809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } 43909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 44009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling assert((permutationEncoding & 0x3FF) == permutationEncoding && 44109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling "Invalid compact register encoding!"); 44209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return permutationEncoding; 44309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling} 44409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 44509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling/// encodeCompactUnwindRegistersWithFrame - Return the registers encoded for a 44609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling/// compact encoding with a frame pointer. 44757a3cd2f326269dd1431cf430db50239a93450f2Bill Wendlingstatic uint32_t 44857a3cd2f326269dd1431cf430db50239a93450f2Bill WendlingencodeCompactUnwindRegistersWithFrame(unsigned SavedRegs[CU_NUM_SAVED_REGS], 44957a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling bool Is64Bit) { 45009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling static const unsigned CU32BitRegs[] = { 45109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling X86::EBX, X86::ECX, X86::EDX, X86::EDI, X86::ESI, X86::EBP, 0 45209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling }; 45309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling static const unsigned CU64BitRegs[] = { 45409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling X86::RBX, X86::R12, X86::R13, X86::R14, X86::R15, X86::RBP, 0 45509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling }; 45609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling const unsigned *CURegs = (Is64Bit ? CU64BitRegs : CU32BitRegs); 45709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 458e2eb92578a1c5c6b890d8c6db41bb6b4d2662625Benjamin Kramer // FIXME: The code below is WRONG and breaks tests on i386, see 459e2eb92578a1c5c6b890d8c6db41bb6b4d2662625Benjamin Kramer // SingleSource/Regression/C++/EH/ctor_dtor_count.exec 460e2eb92578a1c5c6b890d8c6db41bb6b4d2662625Benjamin Kramer // SingleSource/Regression/C++/EH/exception_spec_test.exec 461e2eb92578a1c5c6b890d8c6db41bb6b4d2662625Benjamin Kramer // SingleSource/Regression/C++/EH/function_try_block.exec 462e2eb92578a1c5c6b890d8c6db41bb6b4d2662625Benjamin Kramer // SingleSource/Regression/C++/EH/throw_rethrow_test.exec 463e2eb92578a1c5c6b890d8c6db41bb6b4d2662625Benjamin Kramer return ~0U; 464e2eb92578a1c5c6b890d8c6db41bb6b4d2662625Benjamin Kramer 46509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Encode the registers in the order they were saved, 3-bits per register. The 46609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // registers are numbered from 1 to 6. 46709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling uint32_t RegEnc = 0; 468dec1f996152d4292133e81527ad710fbc1280946Jakob Stoklund Olesen for (int I = 0; I != 6; ++I) { 46909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned Reg = SavedRegs[I]; 47009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (Reg == 0) break; 47109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling int CURegNum = getCompactUnwindRegNum(CURegs, Reg); 47209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (CURegNum == -1) 47309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return ~0U; 47480caf9c2737d4f1bf5bae3a283fe9d538f5e2970Bill Wendling 47580caf9c2737d4f1bf5bae3a283fe9d538f5e2970Bill Wendling // Encode the 3-bit register number in order, skipping over 3-bits for each 47680caf9c2737d4f1bf5bae3a283fe9d538f5e2970Bill Wendling // register. 47779df986c60094e9ea29f081295aea08b1680a999Bill Wendling RegEnc |= (CURegNum & 0x7) << ((5 - I) * 3); 47809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } 47909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 480dec1f996152d4292133e81527ad710fbc1280946Jakob Stoklund Olesen assert((RegEnc & 0x3FFFF) == RegEnc && "Invalid compact register encoding!"); 48109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return RegEnc; 48209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling} 48309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 48409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendlinguint32_t X86FrameLowering::getCompactUnwindEncoding(MachineFunction &MF) const { 48509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling const X86RegisterInfo *RegInfo = TM.getRegisterInfo(); 48609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned FramePtr = RegInfo->getFrameRegister(MF); 48709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned StackPtr = RegInfo->getStackRegister(); 48809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 48909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling bool Is64Bit = STI.is64Bit(); 49009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling bool HasFP = hasFP(MF); 49109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 49257a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling unsigned SavedRegs[CU_NUM_SAVED_REGS] = { 0, 0, 0, 0, 0, 0 }; 49310e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling unsigned SavedRegIdx = 0; 49409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 49509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned OffsetSize = (Is64Bit ? 8 : 4); 49609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 49709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned PushInstr = (Is64Bit ? X86::PUSH64r : X86::PUSH32r); 49809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned PushInstrSize = 1; 49909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned MoveInstr = (Is64Bit ? X86::MOV64rr : X86::MOV32rr); 50009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned MoveInstrSize = (Is64Bit ? 3 : 2); 50109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned SubtractInstrIdx = (Is64Bit ? 3 : 2); 50209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 503de77055a68a3fc495e01b682f00059af3e38822eBill Wendling unsigned StackDivide = (Is64Bit ? 8 : 4); 504de77055a68a3fc495e01b682f00059af3e38822eBill Wendling 50509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned InstrOffset = 0; 50609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned StackAdjust = 0; 50757a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling unsigned StackSize = 0; 50809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 50909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling MachineBasicBlock &MBB = MF.front(); // Prologue is in entry BB. 51009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling bool ExpectEnd = false; 51109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling for (MachineBasicBlock::iterator 51209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling MBBI = MBB.begin(), MBBE = MBB.end(); MBBI != MBBE; ++MBBI) { 51309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling MachineInstr &MI = *MBBI; 51409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned Opc = MI.getOpcode(); 51509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (Opc == X86::PROLOG_LABEL) continue; 51609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (!MI.getFlag(MachineInstr::FrameSetup)) break; 51709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 51809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // We don't exect any more prolog instructions. 51909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (ExpectEnd) return 0; 52009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 52109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (Opc == PushInstr) { 52209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // If there are too many saved registers, we cannot use compact encoding. 52310e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling if (SavedRegIdx >= CU_NUM_SAVED_REGS) return 0; 52409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 52510e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling SavedRegs[SavedRegIdx++] = MI.getOperand(0).getReg(); 52657a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling StackAdjust += OffsetSize; 52709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling InstrOffset += PushInstrSize; 52809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } else if (Opc == MoveInstr) { 52909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned SrcReg = MI.getOperand(1).getReg(); 53009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned DstReg = MI.getOperand(0).getReg(); 53109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 53209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (DstReg != FramePtr || SrcReg != StackPtr) 53309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return 0; 53409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 53557a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling StackAdjust = 0; 53609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling memset(SavedRegs, 0, sizeof(SavedRegs)); 53710e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling SavedRegIdx = 0; 53809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling InstrOffset += MoveInstrSize; 53984d518af1991f581b748c4d11dbeb1c54573556bBill Wendling } else if (Opc == X86::SUB64ri32 || Opc == X86::SUB64ri8 || 54084d518af1991f581b748c4d11dbeb1c54573556bBill Wendling Opc == X86::SUB32ri || Opc == X86::SUB32ri8) { 54157a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling if (StackSize) 54257a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling // We already have a stack size. 54309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return 0; 54409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 54509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (!MI.getOperand(0).isReg() || 54609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling MI.getOperand(0).getReg() != MI.getOperand(1).getReg() || 54709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling MI.getOperand(0).getReg() != StackPtr || !MI.getOperand(2).isImm()) 54809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // We need this to be a stack adjustment pointer. Something like: 54909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // 55009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // %RSP<def> = SUB64ri8 %RSP, 48 55109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return 0; 55209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 55357a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling StackSize = MI.getOperand(2).getImm() / StackDivide; 55409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling SubtractInstrIdx += InstrOffset; 55509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling ExpectEnd = true; 55609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } 55709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } 55809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 55909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Encode that we are using EBP/RBP as the frame pointer. 56009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling uint32_t CompactUnwindEncoding = 0; 56157a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling StackAdjust /= StackDivide; 56209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (HasFP) { 56357a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling if ((StackAdjust & 0xFF) != StackAdjust) 56409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Offset was too big for compact encoding. 56509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return 0; 56609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 56709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Get the encoding of the saved registers when we have a frame pointer. 56809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling uint32_t RegEnc = encodeCompactUnwindRegistersWithFrame(SavedRegs, Is64Bit); 5695b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling if (RegEnc == ~0U) return 0; 57009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 57109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling CompactUnwindEncoding |= 0x01000000; 57257a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling CompactUnwindEncoding |= (StackAdjust & 0xFF) << 16; 57309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling CompactUnwindEncoding |= RegEnc & 0x7FFF; 57409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } else { 575b3ec329c14fa3d37b5963f188db54ba03c6b97c0Bill Wendling ++StackAdjust; 576b3ec329c14fa3d37b5963f188db54ba03c6b97c0Bill Wendling uint32_t TotalStackSize = StackAdjust + StackSize; 577581ac2723c5cb5182114092591eb5e21ceea2e77Bill Wendling if ((TotalStackSize & 0xFF) == TotalStackSize) { 5785b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling // Frameless stack with a small stack size. 57909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling CompactUnwindEncoding |= 0x02000000; 5805b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling 5815b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling // Encode the stack size. 582581ac2723c5cb5182114092591eb5e21ceea2e77Bill Wendling CompactUnwindEncoding |= (TotalStackSize & 0xFF) << 16; 58309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } else { 58457a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling if ((StackAdjust & 0x7) != StackAdjust) 58509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // The extra stack adjustments are too big for us to handle. 58609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return 0; 58709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 58809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Frameless stack with an offset too large for us to encode compactly. 58909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling CompactUnwindEncoding |= 0x03000000; 59009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 59109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Encode the offset to the nnnnnn value in the 'subl $nnnnnn, ESP' 59209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // instruction. 59309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling CompactUnwindEncoding |= (SubtractInstrIdx & 0xFF) << 16; 59409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 59557a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling // Encode any extra stack stack adjustments (done via push instructions). 59657a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling CompactUnwindEncoding |= (StackAdjust & 0x7) << 13; 59709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } 59809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 5995b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling // Encode the number of registers saved. 60010e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling CompactUnwindEncoding |= (SavedRegIdx & 0x7) << 10; 60175e14e0ebd29bb2b1893948037c0ae5df4e09a41Bill Wendling 60209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Get the encoding of the saved registers when we don't have a frame 60309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // pointer. 60457a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling uint32_t RegEnc = 60510e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling encodeCompactUnwindRegistersWithoutFrame(SavedRegs, SavedRegIdx, 60657a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling Is64Bit); 60709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (RegEnc == ~0U) return 0; 6085b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling 6095b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling // Encode the register encoding. 61009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling CompactUnwindEncoding |= RegEnc & 0x3FF; 61109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } 61209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 61309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return CompactUnwindEncoding; 61409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling} 61509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 61633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// emitPrologue - Push callee-saved registers onto the stack, which 61733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// automatically adjust the stack pointer. Adjust the stack pointer to allocate 61833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// space for local variables. Also emit labels used by the exception handler to 61933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// generate the exception handling frames. 62016c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikovvoid X86FrameLowering::emitPrologue(MachineFunction &MF) const { 62133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock &MBB = MF.front(); // Prologue goes in entry BB. 62233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock::iterator MBBI = MBB.begin(); 62333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineFrameInfo *MFI = MF.getFrameInfo(); 62433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov const Function *Fn = MF.getFunction(); 625d9e3385ced2dc887e2fe8e1c071bd2611e4d3edeAnton Korobeynikov const X86RegisterInfo *RegInfo = TM.getRegisterInfo(); 626d9e3385ced2dc887e2fe8e1c071bd2611e4d3edeAnton Korobeynikov const X86InstrInfo &TII = *TM.getInstrInfo(); 62733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineModuleInfo &MMI = MF.getMMI(); 62833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); 62933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool needsFrameMoves = MMI.hasDebugInfo() || 630fc2bb8c4448fa884d79e437cc2d2627a7d7740a8Rafael Espindola Fn->needsUnwindTableEntry(); 63133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t MaxAlign = MFI->getMaxAlignment(); // Desired stack alignment. 63233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t StackSize = MFI->getStackSize(); // Number of bytes to allocate. 633d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov bool HasFP = hasFP(MF); 63433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool Is64Bit = STI.is64Bit(); 63533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool IsWin64 = STI.isTargetWin64(); 63633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned StackAlign = getStackAlignment(); 63733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned SlotSize = RegInfo->getSlotSize(); 63833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned FramePtr = RegInfo->getFrameRegister(MF); 63933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned StackPtr = RegInfo->getStackRegister(); 64033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov DebugLoc DL; 64133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 64233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If we're forcing a stack realignment we can't rely on just the frame 64333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // info, we need to know the ABI stack alignment as well in case we 64433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // have a call out. Otherwise just make sure we have some alignment - we'll 64533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // go with the minimum SlotSize. 64633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (ForceStackAlign) { 64733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (MFI->hasCalls()) 64833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MaxAlign = (StackAlign > MaxAlign) ? StackAlign : MaxAlign; 64933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov else if (MaxAlign < SlotSize) 65033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MaxAlign = SlotSize; 65133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 65233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 65333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Add RETADDR move area to callee saved frame size. 65433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int TailCallReturnAddrDelta = X86FI->getTCReturnAddrDelta(); 65533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (TailCallReturnAddrDelta < 0) 65633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov X86FI->setCalleeSavedFrameSize( 65733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov X86FI->getCalleeSavedFrameSize() - TailCallReturnAddrDelta); 65833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 65933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If this is x86-64 and the Red Zone is not disabled, if we are a leaf 66033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // function, and use up to 128 bytes of stack space, don't have a frame 66133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // pointer, calls, or dynamic alloca then we do not need to adjust the 66233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // stack pointer (we fit in the Red Zone). 66333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (Is64Bit && !Fn->hasFnAttr(Attribute::NoRedZone) && 66433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov !RegInfo->needsStackRealignment(MF) && 6658a8d479214745c82ef00f08d4e4f1c173b5f9ce2Nick Lewycky !MFI->hasVarSizedObjects() && // No dynamic alloca. 6668a8d479214745c82ef00f08d4e4f1c173b5f9ce2Nick Lewycky !MFI->adjustsStack() && // No calls. 6678a8d479214745c82ef00f08d4e4f1c173b5f9ce2Nick Lewycky !IsWin64 && // Win64 has no Red Zone 6688a8d479214745c82ef00f08d4e4f1c173b5f9ce2Nick Lewycky !MF.getTarget().Options.EnableSegmentedStacks) { // Regular stack 66933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t MinSize = X86FI->getCalleeSavedFrameSize(); 67033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (HasFP) MinSize += SlotSize; 67133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov StackSize = std::max(MinSize, StackSize > 128 ? StackSize - 128 : 0); 67233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MFI->setStackSize(StackSize); 67333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 67433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 67533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Insert stack pointer adjustment for later moving of return addr. Only 67633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // applies to tail call optimized functions where the callee argument stack 67733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // size is bigger than the callers. 67833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (TailCallReturnAddrDelta < 0) { 67933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineInstr *MI = 68033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 68133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov TII.get(getSUBriOpcode(Is64Bit, -TailCallReturnAddrDelta)), 68233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov StackPtr) 68333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov .addReg(StackPtr) 684aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .addImm(-TailCallReturnAddrDelta) 685aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .setMIFlag(MachineInstr::FrameSetup); 68633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MI->getOperand(3).setIsDead(); // The EFLAGS implicit def is dead. 68733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 68833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 68933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Mapping for machine moves: 69033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // 69133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // DST: VirtualFP AND 69233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // SRC: VirtualFP => DW_CFA_def_cfa_offset 69333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // ELSE => DW_CFA_def_cfa 69433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // 69533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // SRC: VirtualFP AND 69633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // DST: Register => DW_CFA_def_cfa_register 69733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // 69833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // ELSE 69933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // OFFSET < 0 => DW_CFA_offset_extended_sf 70033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // REG < 64 => DW_CFA_offset + Reg 70133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // ELSE => DW_CFA_offset_extended 70233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 70333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov std::vector<MachineMove> &Moves = MMI.getFrameMoves(); 70433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov const TargetData *TD = MF.getTarget().getTargetData(); 70533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t NumBytes = 0; 70633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int stackGrowth = -TD->getPointerSize(); 70733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 70833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (HasFP) { 70933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Calculate required stack adjustment. 71033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t FrameSize = StackSize - SlotSize; 71133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (RegInfo->needsStackRealignment(MF)) 71233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov FrameSize = (FrameSize + MaxAlign - 1) / MaxAlign * MaxAlign; 71333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 71433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NumBytes = FrameSize - X86FI->getCalleeSavedFrameSize(); 71533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 71633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Get the offset of the stack slot for the EBP register, which is 71733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // guaranteed to be the last slot by processFunctionBeforeFrameFinalized. 71833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Update the frame offset adjustment. 71933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MFI->setOffsetAdjustment(-NumBytes); 72033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 72133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Save EBP/RBP into the appropriate stack slot. 72233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get(Is64Bit ? X86::PUSH64r : X86::PUSH32r)) 723aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .addReg(FramePtr, RegState::Kill) 724aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .setMIFlag(MachineInstr::FrameSetup); 72533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 72633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (needsFrameMoves) { 72733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Mark the place where EBP/RBP was saved. 72833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MCSymbol *FrameLabel = MMI.getContext().CreateTempSymbol(); 729fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling BuildMI(MBB, MBBI, DL, TII.get(X86::PROLOG_LABEL)) 730fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addSym(FrameLabel); 73133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 73233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Define the current CFA rule to use the provided offset. 73333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (StackSize) { 73433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPDst(MachineLocation::VirtualFP); 73533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPSrc(MachineLocation::VirtualFP, 2 * stackGrowth); 73633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(FrameLabel, SPDst, SPSrc)); 73733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 73833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPDst(StackPtr); 73933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPSrc(StackPtr, stackGrowth); 74033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(FrameLabel, SPDst, SPSrc)); 74133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 74233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 74333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Change the rule for the FramePtr to be an "offset" rule. 74433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation FPDst(MachineLocation::VirtualFP, 2 * stackGrowth); 74533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation FPSrc(FramePtr); 74633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(FrameLabel, FPDst, FPSrc)); 74733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 74833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 74909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Update EBP with the new base value. 75033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 75133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov TII.get(Is64Bit ? X86::MOV64rr : X86::MOV32rr), FramePtr) 752aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .addReg(StackPtr) 753aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .setMIFlag(MachineInstr::FrameSetup); 75433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 75533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (needsFrameMoves) { 75633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Mark effective beginning of when frame pointer becomes valid. 75733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MCSymbol *FrameLabel = MMI.getContext().CreateTempSymbol(); 758fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling BuildMI(MBB, MBBI, DL, TII.get(X86::PROLOG_LABEL)) 759fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addSym(FrameLabel); 76033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 76133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Define the current CFA to use the EBP/RBP register. 76233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation FPDst(FramePtr); 76333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation FPSrc(MachineLocation::VirtualFP); 76433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(FrameLabel, FPDst, FPSrc)); 76533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 76633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 76733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Mark the FramePtr as live-in in every block except the entry. 76833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov for (MachineFunction::iterator I = llvm::next(MF.begin()), E = MF.end(); 76933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov I != E; ++I) 77033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov I->addLiveIn(FramePtr); 77133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 77233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Realign stack 77333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (RegInfo->needsStackRealignment(MF)) { 77433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineInstr *MI = 77533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 776fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling TII.get(Is64Bit ? X86::AND64ri32 : X86::AND32ri), StackPtr) 777fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addReg(StackPtr) 778fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addImm(-MaxAlign) 779fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .setMIFlag(MachineInstr::FrameSetup); 78033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 78133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // The EFLAGS implicit def is dead. 78233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MI->getOperand(3).setIsDead(); 78333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 78433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 78533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NumBytes = StackSize - X86FI->getCalleeSavedFrameSize(); 78633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 78733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 78833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Skip the callee-saved push instructions. 78933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool PushedRegs = false; 79033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int StackOffset = 2 * stackGrowth; 79133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 79233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov while (MBBI != MBB.end() && 79333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov (MBBI->getOpcode() == X86::PUSH32r || 79433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBBI->getOpcode() == X86::PUSH64r)) { 79533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov PushedRegs = true; 796fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling MBBI->setFlag(MachineInstr::FrameSetup); 79733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov ++MBBI; 79833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 79933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (!HasFP && needsFrameMoves) { 80033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Mark callee-saved push instruction. 80133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MCSymbol *Label = MMI.getContext().CreateTempSymbol(); 80233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get(X86::PROLOG_LABEL)).addSym(Label); 80333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 80433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Define the current CFA rule to use the provided offset. 80509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned Ptr = StackSize ? MachineLocation::VirtualFP : StackPtr; 80633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPDst(Ptr); 80733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPSrc(Ptr, StackOffset); 80833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(Label, SPDst, SPSrc)); 80933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov StackOffset += stackGrowth; 81033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 81133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 81233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 81333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov DL = MBB.findDebugLoc(MBBI); 81433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 81533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If there is an SUB32ri of ESP immediately before this instruction, merge 81633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // the two. This can be the case when tail call elimination is enabled and 81733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // the callee has more arguments then the caller. 81833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NumBytes -= mergeSPUpdates(MBB, MBBI, StackPtr, true); 81933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 82033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If there is an ADD32ri or SUB32ri of ESP immediately after this 82133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // instruction, merge the two instructions. 82233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov mergeSPUpdatesDown(MBB, MBBI, StackPtr, &NumBytes); 82333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 82433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Adjust stack pointer: ESP -= numbytes. 82533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 82633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Windows and cygwin/mingw require a prologue helper routine when allocating 82733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // more than 4K bytes on the stack. Windows uses __chkstk and cygwin/mingw 82833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // uses __alloca. __alloca and the 32-bit version of __chkstk will probe the 82933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // stack and adjust the stack pointer in one go. The 64-bit version of 83033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // __chkstk is only responsible for probing the stack. The 64-bit prologue is 83133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // responsible for adjusting the stack pointer. Touching the stack at 4K 83233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // increments is necessary to ensure that the guard pages used by the OS 83333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // virtual memory manager are allocated in correct sequence. 834a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (NumBytes >= 4096 && STI.isTargetCOFF() && !STI.isTargetEnvMacho()) { 835a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi const char *StackProbeSymbol; 836a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi bool isSPUpdateNeeded = false; 837a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi 838a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (Is64Bit) { 839a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (STI.isTargetCygMing()) 840a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi StackProbeSymbol = "___chkstk"; 841a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi else { 842a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi StackProbeSymbol = "__chkstk"; 843a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi isSPUpdateNeeded = true; 844a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi } 845a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi } else if (STI.isTargetCygMing()) 846a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi StackProbeSymbol = "_alloca"; 847a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi else 848a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi StackProbeSymbol = "_chkstk"; 849a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi 85033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Check whether EAX is livein for this function. 85133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool isEAXAlive = isEAXLiveIn(MF); 85233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 853a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (isEAXAlive) { 854a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // Sanity check that EAX is not livein for this function. 855a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // It should not be, so throw an assert. 856a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi assert(!Is64Bit && "EAX is livein in x64 case!"); 857a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi 85833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Save EAX 85933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get(X86::PUSH32r)) 860fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addReg(X86::EAX, RegState::Kill) 861fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .setMIFlag(MachineInstr::FrameSetup); 862a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi } 86333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 864a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (Is64Bit) { 865a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // Handle the 64-bit Windows ABI case where we need to call __chkstk. 866a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // Function prologue is responsible for adjusting the stack pointer. 867a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi BuildMI(MBB, MBBI, DL, TII.get(X86::MOV64ri), X86::RAX) 868fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addImm(NumBytes) 869fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .setMIFlag(MachineInstr::FrameSetup); 870a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi } else { 871a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // Allocate NumBytes-4 bytes on stack in case of isEAXAlive. 872a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // We'll also use 4 already allocated bytes for EAX. 87333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get(X86::MOV32ri), X86::EAX) 874fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addImm(isEAXAlive ? NumBytes - 4 : NumBytes) 875fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .setMIFlag(MachineInstr::FrameSetup); 876a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi } 877a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi 878a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi BuildMI(MBB, MBBI, DL, 879a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi TII.get(Is64Bit ? X86::W64ALLOCA : X86::CALLpcrel32)) 880a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi .addExternalSymbol(StackProbeSymbol) 881a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi .addReg(StackPtr, RegState::Define | RegState::Implicit) 882fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addReg(X86::EFLAGS, RegState::Define | RegState::Implicit) 883fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .setMIFlag(MachineInstr::FrameSetup); 884a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi 885a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // MSVC x64's __chkstk needs to adjust %rsp. 886a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // FIXME: %rax preserves the offset and should be available. 887a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (isSPUpdateNeeded) 888a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi emitSPUpdate(MBB, MBBI, StackPtr, -(int64_t)NumBytes, Is64Bit, 889a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi TII, *RegInfo); 890a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi 891a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (isEAXAlive) { 892a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // Restore EAX 893a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi MachineInstr *MI = addRegOffset(BuildMI(MF, DL, TII.get(X86::MOV32rm), 894a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi X86::EAX), 895a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi StackPtr, false, NumBytes - 4); 896fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling MI->setFlag(MachineInstr::FrameSetup); 897a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi MBB.insert(MBBI, MI); 89833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 89933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if (NumBytes) 9007158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng emitSPUpdate(MBB, MBBI, StackPtr, -(int64_t)NumBytes, Is64Bit, 9017158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng TII, *RegInfo); 90233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 903f0adba9a7ec8a3031876575a6ffb7db5f1b6f855Rafael Espindola if (( (!HasFP && NumBytes) || PushedRegs) && needsFrameMoves) { 90433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Mark end of stack pointer adjustment. 90533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MCSymbol *Label = MMI.getContext().CreateTempSymbol(); 906fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling BuildMI(MBB, MBBI, DL, TII.get(X86::PROLOG_LABEL)) 907fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addSym(Label); 90833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 90933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (!HasFP && NumBytes) { 91033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Define the current CFA rule to use the provided offset. 91133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (StackSize) { 91233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPDst(MachineLocation::VirtualFP); 91333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPSrc(MachineLocation::VirtualFP, 91433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov -StackSize + stackGrowth); 91533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(Label, SPDst, SPSrc)); 91633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 91733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPDst(StackPtr); 91833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPSrc(StackPtr, stackGrowth); 91933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(Label, SPDst, SPSrc)); 92033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 92133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 92233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 92333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Emit DWARF info specifying the offsets of the callee-saved registers. 92433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (PushedRegs) 92533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov emitCalleeSavedFrameMoves(MF, Label, HasFP ? FramePtr : StackPtr); 92633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 92709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 92809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Darwin 10.7 and greater has support for compact unwind encoding. 929c8725d11f8756c57bdbceccc61062a9d560261c5Bill Wendling if (STI.getTargetTriple().isMacOSX() && 930ac86d43eae8bbfe5284d8e5fa17c141d1a7b1194Eli Friedman !STI.getTargetTriple().isMacOSXVersionLT(10, 7)) 93109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling MMI.setCompactUnwindEncoding(getCompactUnwindEncoding(MF)); 93233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov} 93333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 93416c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikovvoid X86FrameLowering::emitEpilogue(MachineFunction &MF, 9353c2f0a11cce5a1e828e20675fa8467b624795e0aNick Lewycky MachineBasicBlock &MBB) const { 93633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov const MachineFrameInfo *MFI = MF.getFrameInfo(); 93733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); 938d9e3385ced2dc887e2fe8e1c071bd2611e4d3edeAnton Korobeynikov const X86RegisterInfo *RegInfo = TM.getRegisterInfo(); 939d9e3385ced2dc887e2fe8e1c071bd2611e4d3edeAnton Korobeynikov const X86InstrInfo &TII = *TM.getInstrInfo(); 9404f28c1c71450c711e96aa283de53739d8b4504cdJakob Stoklund Olesen MachineBasicBlock::iterator MBBI = MBB.getLastNonDebugInstr(); 9414f28c1c71450c711e96aa283de53739d8b4504cdJakob Stoklund Olesen assert(MBBI != MBB.end() && "Returning block has no instructions"); 94233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned RetOpcode = MBBI->getOpcode(); 94333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov DebugLoc DL = MBBI->getDebugLoc(); 94433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool Is64Bit = STI.is64Bit(); 94533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned StackAlign = getStackAlignment(); 94633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned SlotSize = RegInfo->getSlotSize(); 94733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned FramePtr = RegInfo->getFrameRegister(MF); 94833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned StackPtr = RegInfo->getStackRegister(); 94933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 95033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov switch (RetOpcode) { 95133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov default: 95233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov llvm_unreachable("Can only insert epilog into returning blocks"); 95333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::RET: 95433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::RETI: 95533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::TCRETURNdi: 95633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::TCRETURNri: 95733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::TCRETURNmi: 95833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::TCRETURNdi64: 95933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::TCRETURNri64: 96033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::TCRETURNmi64: 96133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::EH_RETURN: 96233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::EH_RETURN64: 96333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov break; // These are ok 96433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 96533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 96633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Get the number of bytes to allocate from the FrameInfo. 96733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t StackSize = MFI->getStackSize(); 96833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t MaxAlign = MFI->getMaxAlignment(); 96933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned CSSize = X86FI->getCalleeSavedFrameSize(); 97033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t NumBytes = 0; 97133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 97233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If we're forcing a stack realignment we can't rely on just the frame 97333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // info, we need to know the ABI stack alignment as well in case we 97433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // have a call out. Otherwise just make sure we have some alignment - we'll 97533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // go with the minimum. 97633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (ForceStackAlign) { 97733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (MFI->hasCalls()) 97833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MaxAlign = (StackAlign > MaxAlign) ? StackAlign : MaxAlign; 97933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov else 98033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MaxAlign = MaxAlign ? MaxAlign : 4; 98133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 98233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 983d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov if (hasFP(MF)) { 98433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Calculate required stack adjustment. 98533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t FrameSize = StackSize - SlotSize; 98633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (RegInfo->needsStackRealignment(MF)) 98733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov FrameSize = (FrameSize + MaxAlign - 1)/MaxAlign*MaxAlign; 98833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 98933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NumBytes = FrameSize - CSSize; 99033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 99133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Pop EBP. 99233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 99333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov TII.get(Is64Bit ? X86::POP64r : X86::POP32r), FramePtr); 99433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 99533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NumBytes = StackSize - CSSize; 99633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 99733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 99833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Skip the callee-saved pop instructions. 99933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock::iterator LastCSPop = MBBI; 100033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov while (MBBI != MBB.begin()) { 100133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock::iterator PI = prior(MBBI); 100233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned Opc = PI->getOpcode(); 100333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 10044f28c1c71450c711e96aa283de53739d8b4504cdJakob Stoklund Olesen if (Opc != X86::POP32r && Opc != X86::POP64r && Opc != X86::DBG_VALUE && 10055a96b3dad2f634c9081c8b2b6c2575441dc5a2bdEvan Cheng !PI->isTerminator()) 100633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov break; 100733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 100833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov --MBBI; 100933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 101033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 101133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov DL = MBBI->getDebugLoc(); 101233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 101333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If there is an ADD32ri or SUB32ri of ESP immediately before this 101433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // instruction, merge the two instructions. 101533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (NumBytes || MFI->hasVarSizedObjects()) 101633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov mergeSPUpdatesUp(MBB, MBBI, StackPtr, &NumBytes); 101733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 101833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If dynamic alloca is used, then reset esp to point to the last callee-saved 101933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // slot before popping them off! Same applies for the case, when stack was 102033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // realigned. 102133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (RegInfo->needsStackRealignment(MF)) { 102233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // We cannot use LEA here, because stack pointer was realigned. We need to 102333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // deallocate local frame back. 102433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (CSSize) { 10257158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng emitSPUpdate(MBB, MBBI, StackPtr, NumBytes, Is64Bit, TII, *RegInfo); 102633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBBI = prior(LastCSPop); 102733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 102833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 102933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 103033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov TII.get(Is64Bit ? X86::MOV64rr : X86::MOV32rr), 103133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov StackPtr).addReg(FramePtr); 103233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if (MFI->hasVarSizedObjects()) { 103333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (CSSize) { 103433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned Opc = Is64Bit ? X86::LEA64r : X86::LEA32r; 103533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineInstr *MI = 103633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov addRegOffset(BuildMI(MF, DL, TII.get(Opc), StackPtr), 103733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov FramePtr, false, -CSSize); 103833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBB.insert(MBBI, MI); 103933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 104033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 104133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov TII.get(Is64Bit ? X86::MOV64rr : X86::MOV32rr), StackPtr) 104233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov .addReg(FramePtr); 104333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 104433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if (NumBytes) { 104533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Adjust stack pointer back: ESP += numbytes. 10467158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng emitSPUpdate(MBB, MBBI, StackPtr, NumBytes, Is64Bit, TII, *RegInfo); 104733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 104833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 104933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // We're returning from function via eh_return. 105033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (RetOpcode == X86::EH_RETURN || RetOpcode == X86::EH_RETURN64) { 10514f28c1c71450c711e96aa283de53739d8b4504cdJakob Stoklund Olesen MBBI = MBB.getLastNonDebugInstr(); 105233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineOperand &DestAddr = MBBI->getOperand(0); 105333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov assert(DestAddr.isReg() && "Offset should be in register!"); 105433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 105533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov TII.get(Is64Bit ? X86::MOV64rr : X86::MOV32rr), 105633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov StackPtr).addReg(DestAddr.getReg()); 105733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if (RetOpcode == X86::TCRETURNri || RetOpcode == X86::TCRETURNdi || 105833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov RetOpcode == X86::TCRETURNmi || 105933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov RetOpcode == X86::TCRETURNri64 || RetOpcode == X86::TCRETURNdi64 || 106033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov RetOpcode == X86::TCRETURNmi64) { 106133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool isMem = RetOpcode == X86::TCRETURNmi || RetOpcode == X86::TCRETURNmi64; 106233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Tail call return: adjust the stack pointer and jump to callee. 1063f7ca976e74eafeeab0e9097f0fb07d6bb447415bJakob Stoklund Olesen MBBI = MBB.getLastNonDebugInstr(); 106433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineOperand &JumpTarget = MBBI->getOperand(0); 106533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineOperand &StackAdjust = MBBI->getOperand(isMem ? 5 : 1); 106633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov assert(StackAdjust.isImm() && "Expecting immediate value."); 106733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 106833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Adjust stack pointer. 106933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int StackAdj = StackAdjust.getImm(); 107033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int MaxTCDelta = X86FI->getTCReturnAddrDelta(); 107133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int Offset = 0; 107233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov assert(MaxTCDelta <= 0 && "MaxTCDelta should never be positive"); 107333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 107433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Incoporate the retaddr area. 107533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Offset = StackAdj-MaxTCDelta; 107633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov assert(Offset >= 0 && "Offset should never be negative"); 107733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 107833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (Offset) { 10797a2bdde0a0eebcd2125055e0eacaca040f0b766cChris Lattner // Check for possible merge with preceding ADD instruction. 108033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Offset += mergeSPUpdates(MBB, MBBI, StackPtr, true); 10817158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng emitSPUpdate(MBB, MBBI, StackPtr, Offset, Is64Bit, TII, *RegInfo); 108233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 108333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 108433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Jump to label or value in register. 108533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (RetOpcode == X86::TCRETURNdi || RetOpcode == X86::TCRETURNdi64) { 10863d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng MachineInstrBuilder MIB = 10873d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng BuildMI(MBB, MBBI, DL, TII.get((RetOpcode == X86::TCRETURNdi) 10883d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng ? X86::TAILJMPd : X86::TAILJMPd64)); 10893d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng if (JumpTarget.isGlobal()) 10903d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng MIB.addGlobalAddress(JumpTarget.getGlobal(), JumpTarget.getOffset(), 10913d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng JumpTarget.getTargetFlags()); 10923d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng else { 10933d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng assert(JumpTarget.isSymbol()); 10943d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng MIB.addExternalSymbol(JumpTarget.getSymbolName(), 10953d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng JumpTarget.getTargetFlags()); 10963d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng } 109733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if (RetOpcode == X86::TCRETURNmi || RetOpcode == X86::TCRETURNmi64) { 109833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineInstrBuilder MIB = 109933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get((RetOpcode == X86::TCRETURNmi) 110033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov ? X86::TAILJMPm : X86::TAILJMPm64)); 110133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov for (unsigned i = 0; i != 5; ++i) 110233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MIB.addOperand(MBBI->getOperand(i)); 110333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if (RetOpcode == X86::TCRETURNri64) { 110433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get(X86::TAILJMPr64)). 110533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov addReg(JumpTarget.getReg(), RegState::Kill); 110633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 110733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get(X86::TAILJMPr)). 110833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov addReg(JumpTarget.getReg(), RegState::Kill); 110933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 111033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 111133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineInstr *NewMI = prior(MBBI); 111233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov for (unsigned i = 2, e = MBBI->getNumOperands(); i != e; ++i) 111333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NewMI->addOperand(MBBI->getOperand(i)); 111433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 111533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Delete the pseudo instruction TCRETURN. 111633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBB.erase(MBBI); 111733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if ((RetOpcode == X86::RET || RetOpcode == X86::RETI) && 111833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov (X86FI->getTCReturnAddrDelta() < 0)) { 111933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Add the return addr area delta back since we are not tail calling. 112033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int delta = -1*X86FI->getTCReturnAddrDelta(); 11214f28c1c71450c711e96aa283de53739d8b4504cdJakob Stoklund Olesen MBBI = MBB.getLastNonDebugInstr(); 112233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 11237a2bdde0a0eebcd2125055e0eacaca040f0b766cChris Lattner // Check for possible merge with preceding ADD instruction. 112433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov delta += mergeSPUpdates(MBB, MBBI, StackPtr, true); 11257158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng emitSPUpdate(MBB, MBBI, StackPtr, delta, Is64Bit, TII, *RegInfo); 112633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 112733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov} 1128d9e3385ced2dc887e2fe8e1c071bd2611e4d3edeAnton Korobeynikov 112916c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikovint X86FrameLowering::getFrameIndexOffset(const MachineFunction &MF, int FI) const { 113082f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov const X86RegisterInfo *RI = 113182f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov static_cast<const X86RegisterInfo*>(MF.getTarget().getRegisterInfo()); 113282f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov const MachineFrameInfo *MFI = MF.getFrameInfo(); 113382f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov int Offset = MFI->getObjectOffset(FI) - getOffsetOfLocalArea(); 113482f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov uint64_t StackSize = MFI->getStackSize(); 113582f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov 113682f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov if (RI->needsStackRealignment(MF)) { 113782f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov if (FI < 0) { 113882f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov // Skip the saved EBP. 113982f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov Offset += RI->getSlotSize(); 114082f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov } else { 114117001ce25cc205ac1cd2604492c2bce310964220Duncan Sands assert((-(Offset + StackSize)) % MFI->getObjectAlignment(FI) == 0); 114282f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov return Offset + StackSize; 114382f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov } 114482f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov // FIXME: Support tail calls 114582f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov } else { 114682f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov if (!hasFP(MF)) 114782f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov return Offset + StackSize; 114882f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov 114982f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov // Skip the saved EBP. 115082f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov Offset += RI->getSlotSize(); 115182f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov 115282f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov // Skip the RETADDR move area 115382f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov const X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); 115482f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov int TailCallReturnAddrDelta = X86FI->getTCReturnAddrDelta(); 115582f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov if (TailCallReturnAddrDelta < 0) 115682f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov Offset -= TailCallReturnAddrDelta; 115782f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov } 115882f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov 115982f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov return Offset; 116082f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov} 1161cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 116216c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikovbool X86FrameLowering::spillCalleeSavedRegisters(MachineBasicBlock &MBB, 1163cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov MachineBasicBlock::iterator MI, 1164cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov const std::vector<CalleeSavedInfo> &CSI, 1165cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov const TargetRegisterInfo *TRI) const { 1166cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov if (CSI.empty()) 1167cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov return false; 1168cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1169cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov DebugLoc DL = MBB.findDebugLoc(MI); 1170cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1171cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov MachineFunction &MF = *MBB.getParent(); 1172cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1173cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned SlotSize = STI.is64Bit() ? 8 : 4; 1174cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned FPReg = TRI->getFrameRegister(MF); 1175cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned CalleeFrameSize = 0; 1176cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1177cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov const TargetInstrInfo &TII = *MF.getTarget().getInstrInfo(); 1178cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); 1179cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1180419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // Push GPRs. It increases frame size. 1181cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned Opc = STI.is64Bit() ? X86::PUSH64r : X86::PUSH32r; 1182cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov for (unsigned i = CSI.size(); i != 0; --i) { 1183cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned Reg = CSI[i-1].getReg(); 1184419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi if (!X86::GR64RegClass.contains(Reg) && 1185419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi !X86::GR32RegClass.contains(Reg)) 1186419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi continue; 1187cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov // Add the callee-saved register as live-in. It's killed at the spill. 1188cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov MBB.addLiveIn(Reg); 1189cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov if (Reg == FPReg) 1190cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov // X86RegisterInfo::emitPrologue will handle spilling of frame register. 1191cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov continue; 1192419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi CalleeFrameSize += SlotSize; 1193aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis BuildMI(MBB, MI, DL, TII.get(Opc)).addReg(Reg, RegState::Kill) 1194aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .setMIFlag(MachineInstr::FrameSetup); 1195cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov } 1196cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1197cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov X86FI->setCalleeSavedFrameSize(CalleeFrameSize); 1198419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi 1199419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // Make XMM regs spilled. X86 does not have ability of push/pop XMM. 1200419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // It can be done by spilling XMMs to stack frame. 1201419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // Note that only Win64 ABI might spill XMMs. 1202419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi for (unsigned i = CSI.size(); i != 0; --i) { 1203419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi unsigned Reg = CSI[i-1].getReg(); 1204419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi if (X86::GR64RegClass.contains(Reg) || 1205419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi X86::GR32RegClass.contains(Reg)) 1206419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi continue; 1207419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // Add the callee-saved register as live-in. It's killed at the spill. 1208419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi MBB.addLiveIn(Reg); 1209419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi const TargetRegisterClass *RC = TRI->getMinimalPhysRegClass(Reg); 1210419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi TII.storeRegToStackSlot(MBB, MI, Reg, true, CSI[i-1].getFrameIdx(), 1211419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi RC, TRI); 1212419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi } 1213419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi 1214cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov return true; 1215cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov} 1216cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 121716c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikovbool X86FrameLowering::restoreCalleeSavedRegisters(MachineBasicBlock &MBB, 1218cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov MachineBasicBlock::iterator MI, 1219cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov const std::vector<CalleeSavedInfo> &CSI, 1220cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov const TargetRegisterInfo *TRI) const { 1221cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov if (CSI.empty()) 1222cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov return false; 1223cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1224cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov DebugLoc DL = MBB.findDebugLoc(MI); 1225cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1226cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov MachineFunction &MF = *MBB.getParent(); 1227cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov const TargetInstrInfo &TII = *MF.getTarget().getInstrInfo(); 1228419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi 1229419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // Reload XMMs from stack frame. 1230419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi for (unsigned i = 0, e = CSI.size(); i != e; ++i) { 1231419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi unsigned Reg = CSI[i].getReg(); 1232419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi if (X86::GR64RegClass.contains(Reg) || 1233419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi X86::GR32RegClass.contains(Reg)) 1234419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi continue; 1235419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi const TargetRegisterClass *RC = TRI->getMinimalPhysRegClass(Reg); 1236419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi TII.loadRegFromStackSlot(MBB, MI, Reg, CSI[i].getFrameIdx(), 1237419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi RC, TRI); 1238419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi } 1239419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi 1240419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // POP GPRs. 1241cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned FPReg = TRI->getFrameRegister(MF); 1242cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned Opc = STI.is64Bit() ? X86::POP64r : X86::POP32r; 1243cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov for (unsigned i = 0, e = CSI.size(); i != e; ++i) { 1244cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned Reg = CSI[i].getReg(); 1245419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi if (!X86::GR64RegClass.contains(Reg) && 1246419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi !X86::GR32RegClass.contains(Reg)) 1247419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi continue; 1248cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov if (Reg == FPReg) 1249cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov // X86RegisterInfo::emitEpilogue will handle restoring of frame register. 1250cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov continue; 1251419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi BuildMI(MBB, MI, DL, TII.get(Opc), Reg); 1252cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov } 1253cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov return true; 1254cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov} 125594c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov 125694c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikovvoid 125716c29b5f285f375be53dabaa73e3e91107485fe4Anton KorobeynikovX86FrameLowering::processFunctionBeforeCalleeSavedScan(MachineFunction &MF, 125894c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov RegScavenger *RS) const { 125994c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov MachineFrameInfo *MFI = MF.getFrameInfo(); 126094c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov const X86RegisterInfo *RegInfo = TM.getRegisterInfo(); 126194c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov unsigned SlotSize = RegInfo->getSlotSize(); 126294c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov 126394c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); 126494c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov int32_t TailCallReturnAddrDelta = X86FI->getTCReturnAddrDelta(); 126594c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov 126694c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov if (TailCallReturnAddrDelta < 0) { 126794c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // create RETURNADDR area 126894c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // arg 126994c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // arg 127094c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // RETADDR 127194c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // { ... 127294c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // RETADDR area 127394c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // ... 127494c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // } 127594c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // [EBP] 127694c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov MFI->CreateFixedObject(-TailCallReturnAddrDelta, 127794c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov (-1U*SlotSize)+TailCallReturnAddrDelta, true); 127894c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov } 127994c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov 128094c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov if (hasFP(MF)) { 128194c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov assert((TailCallReturnAddrDelta <= 0) && 128294c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov "The Delta should always be zero or negative"); 128316c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikov const TargetFrameLowering &TFI = *MF.getTarget().getFrameLowering(); 128494c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov 128594c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // Create a frame entry for the EBP register that must be saved. 128694c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov int FrameIdx = MFI->CreateFixedObject(SlotSize, 128794c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov -(int)SlotSize + 128894c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov TFI.getOffsetOfLocalArea() + 128994c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov TailCallReturnAddrDelta, 129094c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov true); 129194c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov assert(FrameIdx == MFI->getObjectIndexBegin() && 129294c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov "Slot for EBP register must be last in order to be found!"); 129317001ce25cc205ac1cd2604492c2bce310964220Duncan Sands (void)FrameIdx; 129494c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov } 129594c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov} 129676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 129776927d758657b3a511c73467ec5a7288795c1513Rafael Espindolastatic bool 129876927d758657b3a511c73467ec5a7288795c1513Rafael EspindolaHasNestArgument(const MachineFunction *MF) { 129976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola const Function *F = MF->getFunction(); 130076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola for (Function::const_arg_iterator I = F->arg_begin(), E = F->arg_end(); 130176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola I != E; I++) { 130276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (I->hasNestAttr()) 130376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola return true; 130476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 130576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola return false; 130676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola} 130776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 13082028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola 13092028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola/// GetScratchRegister - Get a register for performing work in the segmented 13102028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola/// stack prologue. Depending on platform and the properties of the function 13112028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola/// either one or two registers will be needed. Set primary to true for 13122028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola/// the first register, false for the second. 131376927d758657b3a511c73467ec5a7288795c1513Rafael Espindolastatic unsigned 13142028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael EspindolaGetScratchRegister(bool Is64Bit, const MachineFunction &MF, bool Primary) { 131576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (Is64Bit) { 13162028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola return Primary ? X86::R11 : X86::R12; 131776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } else { 131876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola CallingConv::ID CallingConvention = MF.getFunction()->getCallingConv(); 131976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola bool IsNested = HasNestArgument(&MF); 132076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 132125cd4ff97e882ad5039dec15b0c9be8fef062b6bRafael Espindola if (CallingConvention == CallingConv::X86_FastCall || 132225cd4ff97e882ad5039dec15b0c9be8fef062b6bRafael Espindola CallingConvention == CallingConv::Fast) { 132376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (IsNested) { 1324e81abfd30b7c81735e9f3d79006c2653690aa592Rafael Espindola report_fatal_error("Segmented stacks does not support fastcall with " 1325e81abfd30b7c81735e9f3d79006c2653690aa592Rafael Espindola "nested function."); 132676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola return -1; 132776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } else { 13282028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola return Primary ? X86::EAX : X86::ECX; 132976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 133076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } else { 133176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (IsNested) 13322028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola return Primary ? X86::EDX : X86::EAX; 133376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola else 13342028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola return Primary ? X86::ECX : X86::EAX; 133576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 133676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 133776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola} 133876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 1339199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das// The stack limit in the TCB is set to this many bytes above the actual stack 1340199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das// limit. 1341199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Dasstatic const uint64_t kSplitStackAvailable = 256; 1342199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das 134376927d758657b3a511c73467ec5a7288795c1513Rafael Espindolavoid 134476927d758657b3a511c73467ec5a7288795c1513Rafael EspindolaX86FrameLowering::adjustForSegmentedStacks(MachineFunction &MF) const { 134576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MachineBasicBlock &prologueMBB = MF.front(); 134676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MachineFrameInfo *MFI = MF.getFrameInfo(); 134776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola const X86InstrInfo &TII = *TM.getInstrInfo(); 134876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola uint64_t StackSize; 134976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola bool Is64Bit = STI.is64Bit(); 135076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola unsigned TlsReg, TlsOffset; 135176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola DebugLoc DL; 135276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola const X86Subtarget *ST = &MF.getTarget().getSubtarget<X86Subtarget>(); 135376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 13542028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola unsigned ScratchReg = GetScratchRegister(Is64Bit, MF, true); 135576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola assert(!MF.getRegInfo().isLiveIn(ScratchReg) && 135676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola "Scratch register is live-in"); 135776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 135876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (MF.getFunction()->isVarArg()) 135976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola report_fatal_error("Segmented stacks do not support vararg functions."); 1360e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola if (!ST->isTargetLinux() && !ST->isTargetDarwin() && !ST->isTargetWin32()) 1361e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola report_fatal_error("Segmented stacks supported only on linux, darwin and win32."); 136276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 136376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MachineBasicBlock *allocMBB = MF.CreateMachineBasicBlock(); 136476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MachineBasicBlock *checkMBB = MF.CreateMachineBasicBlock(); 136576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); 136676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola bool IsNested = false; 136776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 136876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // We need to know if the function has a nest argument only in 64 bit mode. 136976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (Is64Bit) 137076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola IsNested = HasNestArgument(&MF); 137176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 13724e68054b20725f6ec1cac33630258f749fe5debeBill Wendling // The MOV R10, RAX needs to be in a different block, since the RET we emit in 13734e68054b20725f6ec1cac33630258f749fe5debeBill Wendling // allocMBB needs to be last (terminating) instruction. 13744e68054b20725f6ec1cac33630258f749fe5debeBill Wendling 137576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola for (MachineBasicBlock::livein_iterator i = prologueMBB.livein_begin(), 137676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola e = prologueMBB.livein_end(); i != e; i++) { 137776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola allocMBB->addLiveIn(*i); 137876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola checkMBB->addLiveIn(*i); 137976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 138076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 1381e840e88239cf92a065cbf5f5b9c7d18bc139c0e1Rafael Espindola if (IsNested) 138276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola allocMBB->addLiveIn(X86::R10); 138376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 138476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MF.push_front(allocMBB); 138576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MF.push_front(checkMBB); 138676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 138776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // Eventually StackSize will be calculated by a link-time pass; which will 138876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // also decide whether checking code needs to be injected into this particular 138976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // prologue. 139076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola StackSize = MFI->getStackSize(); 139176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 13922028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola // When the frame size is less than 256 we just compare the stack 13932028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola // boundary directly to the value of the stack pointer, per gcc. 13942028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola bool CompareStackPointer = StackSize < kSplitStackAvailable; 13952028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola 139676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // Read the limit off the current stacklet off the stack_guard location. 139776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (Is64Bit) { 13982028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola if (ST->isTargetLinux()) { 13992028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola TlsReg = X86::FS; 14002028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola TlsOffset = 0x70; 14012028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola } else if (ST->isTargetDarwin()) { 14022028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola TlsReg = X86::GS; 14032028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola TlsOffset = 0x60 + 90*8; // See pthread_machdep.h. Steal TLS slot 90. 1404e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola } else { 1405e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola report_fatal_error("Segmented stacks not supported on this platform."); 14062028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola } 140776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 14082028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola if (CompareStackPointer) 1409199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das ScratchReg = X86::RSP; 1410199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das else 1411199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das BuildMI(checkMBB, DL, TII.get(X86::LEA64r), ScratchReg).addReg(X86::RSP) 1412014f7a3b3798580d5aac80b83bcb67e03d302fa4Rafael Espindola .addImm(1).addReg(0).addImm(-StackSize).addReg(0); 1413199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das 141476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(checkMBB, DL, TII.get(X86::CMP64rm)).addReg(ScratchReg) 1415014f7a3b3798580d5aac80b83bcb67e03d302fa4Rafael Espindola .addReg(0).addImm(1).addReg(0).addImm(TlsOffset).addReg(TlsReg); 141676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } else { 1417e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola if (ST->isTargetLinux()) { 1418e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola TlsReg = X86::GS; 1419e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola TlsOffset = 0x30; 1420e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola } else if (ST->isTargetDarwin()) { 1421e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola TlsReg = X86::GS; 1422e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola TlsOffset = 0x48 + 90*4; 1423e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola } else if (ST->isTargetWin32()) { 1424e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola TlsReg = X86::FS; 1425e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola TlsOffset = 0x14; // pvArbitrary, reserved for application use 1426e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola } else { 1427e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola report_fatal_error("Segmented stacks not supported on this platform."); 1428e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola } 142976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 14302028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola if (CompareStackPointer) 1431199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das ScratchReg = X86::ESP; 1432199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das else 1433199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das BuildMI(checkMBB, DL, TII.get(X86::LEA32r), ScratchReg).addReg(X86::ESP) 1434014f7a3b3798580d5aac80b83bcb67e03d302fa4Rafael Espindola .addImm(1).addReg(0).addImm(-StackSize).addReg(0); 1435199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das 1436e4d18de5d15c6c1d4d279788b35d0cd1ab237e82Rafael Espindola if (ST->isTargetLinux() || ST->isTargetWin32()) { 14372028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola BuildMI(checkMBB, DL, TII.get(X86::CMP32rm)).addReg(ScratchReg) 14382028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola .addReg(0).addImm(0).addReg(0).addImm(TlsOffset).addReg(TlsReg); 14392028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola } else if (ST->isTargetDarwin()) { 14402028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola 14412028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola // TlsOffset doesn't fit into a mod r/m byte so we need an extra register 14422028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola unsigned ScratchReg2; 14432028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola bool SaveScratch2; 14442028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola if (CompareStackPointer) { 14452028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola // The primary scratch register is available for holding the TLS offset 14462028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola ScratchReg2 = GetScratchRegister(Is64Bit, MF, true); 14472028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola SaveScratch2 = false; 14482028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola } else { 14492028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola // Need to use a second register to hold the TLS offset 14502028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola ScratchReg2 = GetScratchRegister(Is64Bit, MF, false); 14512028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola 14522028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola // Unfortunately, with fastcc the second scratch register may hold an arg 14532028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola SaveScratch2 = MF.getRegInfo().isLiveIn(ScratchReg2); 14542028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola } 14552028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola 14562028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola // If Scratch2 is live-in then it needs to be saved 14572028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola assert((!MF.getRegInfo().isLiveIn(ScratchReg2) || SaveScratch2) && 14582028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola "Scratch register is live-in and not saved"); 14592028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola 14602028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola if (SaveScratch2) 14612028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola BuildMI(checkMBB, DL, TII.get(X86::PUSH32r)) 14622028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola .addReg(ScratchReg2, RegState::Kill); 14632028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola 14642028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola BuildMI(checkMBB, DL, TII.get(X86::MOV32ri), ScratchReg2) 14652028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola .addImm(TlsOffset); 14662028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola BuildMI(checkMBB, DL, TII.get(X86::CMP32rm)) 14672028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola .addReg(ScratchReg) 14682028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola .addReg(ScratchReg2).addImm(1).addReg(0) 14692028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola .addImm(0) 14702028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola .addReg(TlsReg); 14712028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola 14722028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola if (SaveScratch2) 14732028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola BuildMI(checkMBB, DL, TII.get(X86::POP32r), ScratchReg2); 14742028b793e1fd1a8dd4d99b0b7c9972865d5e806aRafael Espindola } 147576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 147676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 147776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // This jump is taken if SP >= (Stacklet Limit + Stack Space required). 147876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // It jumps to normal execution of the function body. 1479313c7038319422cff0b2ea1015e180575cab4b7aRafael Espindola BuildMI(checkMBB, DL, TII.get(X86::JA_4)).addMBB(&prologueMBB); 148076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 148176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // On 32 bit we first push the arguments size and then the frame size. On 64 148276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // bit, we pass the stack frame size in r10 and the argument size in r11. 148376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (Is64Bit) { 148476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // Functions with nested arguments use R10, so it needs to be saved across 148576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // the call to _morestack 148676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 148776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (IsNested) 148876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::MOV64rr), X86::RAX).addReg(X86::R10); 148976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 149076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::MOV64ri), X86::R10) 149176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addImm(StackSize); 149276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::MOV64ri), X86::R11) 149376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addImm(X86FI->getArgumentStackSize()); 149476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MF.getRegInfo().setPhysRegUsed(X86::R10); 149576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MF.getRegInfo().setPhysRegUsed(X86::R11); 149676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } else { 149776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::PUSHi32)) 149876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addImm(X86FI->getArgumentStackSize()); 149976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::PUSHi32)) 150076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addImm(StackSize); 150176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 150276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 150376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // __morestack is in libgcc 150476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (Is64Bit) 150576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::CALL64pcrel32)) 150676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addExternalSymbol("__morestack"); 150776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola else 150876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::CALLpcrel32)) 150976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addExternalSymbol("__morestack"); 151076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 15114e68054b20725f6ec1cac33630258f749fe5debeBill Wendling if (IsNested) 1512e840e88239cf92a065cbf5f5b9c7d18bc139c0e1Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::MORESTACK_RET_RESTORE_R10)); 1513e840e88239cf92a065cbf5f5b9c7d18bc139c0e1Rafael Espindola else 1514e840e88239cf92a065cbf5f5b9c7d18bc139c0e1Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::MORESTACK_RET)); 151576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 1516e840e88239cf92a065cbf5f5b9c7d18bc139c0e1Rafael Espindola allocMBB->addSuccessor(&prologueMBB); 151782222c20be24adda7c218f3fdaf2e0ae049c955bBill Wendling 151876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola checkMBB->addSuccessor(allocMBB); 151976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola checkMBB->addSuccessor(&prologueMBB); 152076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 152151f0c7641983469cbd29f8862a121645471a885aJakob Stoklund Olesen#ifdef XDEBUG 152276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MF.verify(); 152376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola#endif 152476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola} 1525