X86FrameLowering.cpp revision dec1f996152d4292133e81527ad710fbc1280946
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 45809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Encode the registers in the order they were saved, 3-bits per register. The 45909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // registers are numbered from 1 to 6. 46009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling uint32_t RegEnc = 0; 461dec1f996152d4292133e81527ad710fbc1280946Jakob Stoklund Olesen for (int I = 0; I != 6; ++I) { 46209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned Reg = SavedRegs[I]; 46309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (Reg == 0) break; 46409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling int CURegNum = getCompactUnwindRegNum(CURegs, Reg); 46509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (CURegNum == -1) 46609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return ~0U; 46780caf9c2737d4f1bf5bae3a283fe9d538f5e2970Bill Wendling 46880caf9c2737d4f1bf5bae3a283fe9d538f5e2970Bill Wendling // Encode the 3-bit register number in order, skipping over 3-bits for each 46980caf9c2737d4f1bf5bae3a283fe9d538f5e2970Bill Wendling // register. 47079df986c60094e9ea29f081295aea08b1680a999Bill Wendling RegEnc |= (CURegNum & 0x7) << ((5 - I) * 3); 47109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } 47209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 473dec1f996152d4292133e81527ad710fbc1280946Jakob Stoklund Olesen assert((RegEnc & 0x3FFFF) == RegEnc && "Invalid compact register encoding!"); 47409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return RegEnc; 47509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling} 47609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 47709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendlinguint32_t X86FrameLowering::getCompactUnwindEncoding(MachineFunction &MF) const { 47809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling const X86RegisterInfo *RegInfo = TM.getRegisterInfo(); 47909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned FramePtr = RegInfo->getFrameRegister(MF); 48009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned StackPtr = RegInfo->getStackRegister(); 48109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 48209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling bool Is64Bit = STI.is64Bit(); 48309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling bool HasFP = hasFP(MF); 48409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 48557a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling unsigned SavedRegs[CU_NUM_SAVED_REGS] = { 0, 0, 0, 0, 0, 0 }; 48610e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling unsigned SavedRegIdx = 0; 48709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 48809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned OffsetSize = (Is64Bit ? 8 : 4); 48909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 49009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned PushInstr = (Is64Bit ? X86::PUSH64r : X86::PUSH32r); 49109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned PushInstrSize = 1; 49209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned MoveInstr = (Is64Bit ? X86::MOV64rr : X86::MOV32rr); 49309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned MoveInstrSize = (Is64Bit ? 3 : 2); 49409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned SubtractInstrIdx = (Is64Bit ? 3 : 2); 49509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 496de77055a68a3fc495e01b682f00059af3e38822eBill Wendling unsigned StackDivide = (Is64Bit ? 8 : 4); 497de77055a68a3fc495e01b682f00059af3e38822eBill Wendling 49809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned InstrOffset = 0; 49909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned StackAdjust = 0; 50057a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling unsigned StackSize = 0; 50109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 50209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling MachineBasicBlock &MBB = MF.front(); // Prologue is in entry BB. 50309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling bool ExpectEnd = false; 50409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling for (MachineBasicBlock::iterator 50509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling MBBI = MBB.begin(), MBBE = MBB.end(); MBBI != MBBE; ++MBBI) { 50609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling MachineInstr &MI = *MBBI; 50709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned Opc = MI.getOpcode(); 50809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (Opc == X86::PROLOG_LABEL) continue; 50909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (!MI.getFlag(MachineInstr::FrameSetup)) break; 51009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 51109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // We don't exect any more prolog instructions. 51209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (ExpectEnd) return 0; 51309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 51409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (Opc == PushInstr) { 51509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // If there are too many saved registers, we cannot use compact encoding. 51610e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling if (SavedRegIdx >= CU_NUM_SAVED_REGS) return 0; 51709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 51810e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling SavedRegs[SavedRegIdx++] = MI.getOperand(0).getReg(); 51957a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling StackAdjust += OffsetSize; 52009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling InstrOffset += PushInstrSize; 52109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } else if (Opc == MoveInstr) { 52209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned SrcReg = MI.getOperand(1).getReg(); 52309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned DstReg = MI.getOperand(0).getReg(); 52409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 52509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (DstReg != FramePtr || SrcReg != StackPtr) 52609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return 0; 52709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 52857a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling StackAdjust = 0; 52909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling memset(SavedRegs, 0, sizeof(SavedRegs)); 53010e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling SavedRegIdx = 0; 53109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling InstrOffset += MoveInstrSize; 53284d518af1991f581b748c4d11dbeb1c54573556bBill Wendling } else if (Opc == X86::SUB64ri32 || Opc == X86::SUB64ri8 || 53384d518af1991f581b748c4d11dbeb1c54573556bBill Wendling Opc == X86::SUB32ri || Opc == X86::SUB32ri8) { 53457a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling if (StackSize) 53557a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling // We already have a stack size. 53609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return 0; 53709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 53809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (!MI.getOperand(0).isReg() || 53909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling MI.getOperand(0).getReg() != MI.getOperand(1).getReg() || 54009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling MI.getOperand(0).getReg() != StackPtr || !MI.getOperand(2).isImm()) 54109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // We need this to be a stack adjustment pointer. Something like: 54209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // 54309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // %RSP<def> = SUB64ri8 %RSP, 48 54409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return 0; 54509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 54657a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling StackSize = MI.getOperand(2).getImm() / StackDivide; 54709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling SubtractInstrIdx += InstrOffset; 54809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling ExpectEnd = true; 54909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } 55009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } 55109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 55209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Encode that we are using EBP/RBP as the frame pointer. 55309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling uint32_t CompactUnwindEncoding = 0; 55457a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling StackAdjust /= StackDivide; 55509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (HasFP) { 55657a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling if ((StackAdjust & 0xFF) != StackAdjust) 55709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Offset was too big for compact encoding. 55809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return 0; 55909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 56009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Get the encoding of the saved registers when we have a frame pointer. 56109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling uint32_t RegEnc = encodeCompactUnwindRegistersWithFrame(SavedRegs, Is64Bit); 5625b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling if (RegEnc == ~0U) return 0; 56309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 56409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling CompactUnwindEncoding |= 0x01000000; 56557a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling CompactUnwindEncoding |= (StackAdjust & 0xFF) << 16; 56609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling CompactUnwindEncoding |= RegEnc & 0x7FFF; 56709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } else { 568b3ec329c14fa3d37b5963f188db54ba03c6b97c0Bill Wendling ++StackAdjust; 569b3ec329c14fa3d37b5963f188db54ba03c6b97c0Bill Wendling uint32_t TotalStackSize = StackAdjust + StackSize; 570581ac2723c5cb5182114092591eb5e21ceea2e77Bill Wendling if ((TotalStackSize & 0xFF) == TotalStackSize) { 5715b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling // Frameless stack with a small stack size. 57209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling CompactUnwindEncoding |= 0x02000000; 5735b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling 5745b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling // Encode the stack size. 575581ac2723c5cb5182114092591eb5e21ceea2e77Bill Wendling CompactUnwindEncoding |= (TotalStackSize & 0xFF) << 16; 57609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } else { 57757a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling if ((StackAdjust & 0x7) != StackAdjust) 57809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // The extra stack adjustments are too big for us to handle. 57909b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return 0; 58009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 58109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Frameless stack with an offset too large for us to encode compactly. 58209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling CompactUnwindEncoding |= 0x03000000; 58309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 58409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Encode the offset to the nnnnnn value in the 'subl $nnnnnn, ESP' 58509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // instruction. 58609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling CompactUnwindEncoding |= (SubtractInstrIdx & 0xFF) << 16; 58709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 58857a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling // Encode any extra stack stack adjustments (done via push instructions). 58957a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling CompactUnwindEncoding |= (StackAdjust & 0x7) << 13; 59009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } 59109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 5925b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling // Encode the number of registers saved. 59310e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling CompactUnwindEncoding |= (SavedRegIdx & 0x7) << 10; 59475e14e0ebd29bb2b1893948037c0ae5df4e09a41Bill Wendling 59509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Get the encoding of the saved registers when we don't have a frame 59609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // pointer. 59757a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling uint32_t RegEnc = 59810e412ec6b761952b95710203e6d2d89f4fee53aBill Wendling encodeCompactUnwindRegistersWithoutFrame(SavedRegs, SavedRegIdx, 59957a3cd2f326269dd1431cf430db50239a93450f2Bill Wendling Is64Bit); 60009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling if (RegEnc == ~0U) return 0; 6015b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling 6025b2c4978ce56455689c515c9d74cc1d92871f3bbBill Wendling // Encode the register encoding. 60309b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling CompactUnwindEncoding |= RegEnc & 0x3FF; 60409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling } 60509b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 60609b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling return CompactUnwindEncoding; 60709b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling} 60809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 60933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// emitPrologue - Push callee-saved registers onto the stack, which 61033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// automatically adjust the stack pointer. Adjust the stack pointer to allocate 61133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// space for local variables. Also emit labels used by the exception handler to 61233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov/// generate the exception handling frames. 61316c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikovvoid X86FrameLowering::emitPrologue(MachineFunction &MF) const { 61433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock &MBB = MF.front(); // Prologue goes in entry BB. 61533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock::iterator MBBI = MBB.begin(); 61633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineFrameInfo *MFI = MF.getFrameInfo(); 61733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov const Function *Fn = MF.getFunction(); 618d9e3385ced2dc887e2fe8e1c071bd2611e4d3edeAnton Korobeynikov const X86RegisterInfo *RegInfo = TM.getRegisterInfo(); 619d9e3385ced2dc887e2fe8e1c071bd2611e4d3edeAnton Korobeynikov const X86InstrInfo &TII = *TM.getInstrInfo(); 62033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineModuleInfo &MMI = MF.getMMI(); 62133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); 62233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool needsFrameMoves = MMI.hasDebugInfo() || 623fc2bb8c4448fa884d79e437cc2d2627a7d7740a8Rafael Espindola Fn->needsUnwindTableEntry(); 62433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t MaxAlign = MFI->getMaxAlignment(); // Desired stack alignment. 62533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t StackSize = MFI->getStackSize(); // Number of bytes to allocate. 626d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov bool HasFP = hasFP(MF); 62733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool Is64Bit = STI.is64Bit(); 62833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool IsWin64 = STI.isTargetWin64(); 62933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned StackAlign = getStackAlignment(); 63033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned SlotSize = RegInfo->getSlotSize(); 63133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned FramePtr = RegInfo->getFrameRegister(MF); 63233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned StackPtr = RegInfo->getStackRegister(); 63333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov DebugLoc DL; 63433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 63533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If we're forcing a stack realignment we can't rely on just the frame 63633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // info, we need to know the ABI stack alignment as well in case we 63733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // have a call out. Otherwise just make sure we have some alignment - we'll 63833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // go with the minimum SlotSize. 63933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (ForceStackAlign) { 64033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (MFI->hasCalls()) 64133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MaxAlign = (StackAlign > MaxAlign) ? StackAlign : MaxAlign; 64233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov else if (MaxAlign < SlotSize) 64333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MaxAlign = SlotSize; 64433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 64533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 64633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Add RETADDR move area to callee saved frame size. 64733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int TailCallReturnAddrDelta = X86FI->getTCReturnAddrDelta(); 64833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (TailCallReturnAddrDelta < 0) 64933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov X86FI->setCalleeSavedFrameSize( 65033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov X86FI->getCalleeSavedFrameSize() - TailCallReturnAddrDelta); 65133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 65233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If this is x86-64 and the Red Zone is not disabled, if we are a leaf 65333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // function, and use up to 128 bytes of stack space, don't have a frame 65433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // pointer, calls, or dynamic alloca then we do not need to adjust the 65533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // stack pointer (we fit in the Red Zone). 65633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (Is64Bit && !Fn->hasFnAttr(Attribute::NoRedZone) && 65733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov !RegInfo->needsStackRealignment(MF) && 6588a8d479214745c82ef00f08d4e4f1c173b5f9ce2Nick Lewycky !MFI->hasVarSizedObjects() && // No dynamic alloca. 6598a8d479214745c82ef00f08d4e4f1c173b5f9ce2Nick Lewycky !MFI->adjustsStack() && // No calls. 6608a8d479214745c82ef00f08d4e4f1c173b5f9ce2Nick Lewycky !IsWin64 && // Win64 has no Red Zone 6618a8d479214745c82ef00f08d4e4f1c173b5f9ce2Nick Lewycky !MF.getTarget().Options.EnableSegmentedStacks) { // Regular stack 66233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t MinSize = X86FI->getCalleeSavedFrameSize(); 66333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (HasFP) MinSize += SlotSize; 66433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov StackSize = std::max(MinSize, StackSize > 128 ? StackSize - 128 : 0); 66533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MFI->setStackSize(StackSize); 66633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 66733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 66833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Insert stack pointer adjustment for later moving of return addr. Only 66933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // applies to tail call optimized functions where the callee argument stack 67033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // size is bigger than the callers. 67133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (TailCallReturnAddrDelta < 0) { 67233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineInstr *MI = 67333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 67433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov TII.get(getSUBriOpcode(Is64Bit, -TailCallReturnAddrDelta)), 67533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov StackPtr) 67633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov .addReg(StackPtr) 677aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .addImm(-TailCallReturnAddrDelta) 678aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .setMIFlag(MachineInstr::FrameSetup); 67933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MI->getOperand(3).setIsDead(); // The EFLAGS implicit def is dead. 68033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 68133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 68233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Mapping for machine moves: 68333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // 68433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // DST: VirtualFP AND 68533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // SRC: VirtualFP => DW_CFA_def_cfa_offset 68633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // ELSE => DW_CFA_def_cfa 68733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // 68833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // SRC: VirtualFP AND 68933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // DST: Register => DW_CFA_def_cfa_register 69033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // 69133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // ELSE 69233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // OFFSET < 0 => DW_CFA_offset_extended_sf 69333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // REG < 64 => DW_CFA_offset + Reg 69433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // ELSE => DW_CFA_offset_extended 69533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 69633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov std::vector<MachineMove> &Moves = MMI.getFrameMoves(); 69733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov const TargetData *TD = MF.getTarget().getTargetData(); 69833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t NumBytes = 0; 69933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int stackGrowth = -TD->getPointerSize(); 70033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 70133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (HasFP) { 70233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Calculate required stack adjustment. 70333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t FrameSize = StackSize - SlotSize; 70433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (RegInfo->needsStackRealignment(MF)) 70533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov FrameSize = (FrameSize + MaxAlign - 1) / MaxAlign * MaxAlign; 70633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 70733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NumBytes = FrameSize - X86FI->getCalleeSavedFrameSize(); 70833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 70933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Get the offset of the stack slot for the EBP register, which is 71033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // guaranteed to be the last slot by processFunctionBeforeFrameFinalized. 71133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Update the frame offset adjustment. 71233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MFI->setOffsetAdjustment(-NumBytes); 71333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 71433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Save EBP/RBP into the appropriate stack slot. 71533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get(Is64Bit ? X86::PUSH64r : X86::PUSH32r)) 716aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .addReg(FramePtr, RegState::Kill) 717aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .setMIFlag(MachineInstr::FrameSetup); 71833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 71933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (needsFrameMoves) { 72033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Mark the place where EBP/RBP was saved. 72133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MCSymbol *FrameLabel = MMI.getContext().CreateTempSymbol(); 722fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling BuildMI(MBB, MBBI, DL, TII.get(X86::PROLOG_LABEL)) 723fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addSym(FrameLabel); 72433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 72533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Define the current CFA rule to use the provided offset. 72633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (StackSize) { 72733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPDst(MachineLocation::VirtualFP); 72833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPSrc(MachineLocation::VirtualFP, 2 * stackGrowth); 72933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(FrameLabel, SPDst, SPSrc)); 73033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 73133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPDst(StackPtr); 73233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPSrc(StackPtr, stackGrowth); 73333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(FrameLabel, SPDst, SPSrc)); 73433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 73533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 73633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Change the rule for the FramePtr to be an "offset" rule. 73733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation FPDst(MachineLocation::VirtualFP, 2 * stackGrowth); 73833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation FPSrc(FramePtr); 73933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(FrameLabel, FPDst, FPSrc)); 74033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 74133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 74209b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Update EBP with the new base value. 74333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 74433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov TII.get(Is64Bit ? X86::MOV64rr : X86::MOV32rr), FramePtr) 745aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .addReg(StackPtr) 746aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .setMIFlag(MachineInstr::FrameSetup); 74733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 74833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (needsFrameMoves) { 74933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Mark effective beginning of when frame pointer becomes valid. 75033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MCSymbol *FrameLabel = MMI.getContext().CreateTempSymbol(); 751fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling BuildMI(MBB, MBBI, DL, TII.get(X86::PROLOG_LABEL)) 752fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addSym(FrameLabel); 75333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 75433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Define the current CFA to use the EBP/RBP register. 75533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation FPDst(FramePtr); 75633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation FPSrc(MachineLocation::VirtualFP); 75733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(FrameLabel, FPDst, FPSrc)); 75833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 75933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 76033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Mark the FramePtr as live-in in every block except the entry. 76133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov for (MachineFunction::iterator I = llvm::next(MF.begin()), E = MF.end(); 76233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov I != E; ++I) 76333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov I->addLiveIn(FramePtr); 76433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 76533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Realign stack 76633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (RegInfo->needsStackRealignment(MF)) { 76733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineInstr *MI = 76833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 769fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling TII.get(Is64Bit ? X86::AND64ri32 : X86::AND32ri), StackPtr) 770fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addReg(StackPtr) 771fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addImm(-MaxAlign) 772fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .setMIFlag(MachineInstr::FrameSetup); 77333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 77433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // The EFLAGS implicit def is dead. 77533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MI->getOperand(3).setIsDead(); 77633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 77733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 77833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NumBytes = StackSize - X86FI->getCalleeSavedFrameSize(); 77933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 78033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 78133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Skip the callee-saved push instructions. 78233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool PushedRegs = false; 78333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int StackOffset = 2 * stackGrowth; 78433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 78533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov while (MBBI != MBB.end() && 78633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov (MBBI->getOpcode() == X86::PUSH32r || 78733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBBI->getOpcode() == X86::PUSH64r)) { 78833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov PushedRegs = true; 789fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling MBBI->setFlag(MachineInstr::FrameSetup); 79033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov ++MBBI; 79133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 79233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (!HasFP && needsFrameMoves) { 79333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Mark callee-saved push instruction. 79433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MCSymbol *Label = MMI.getContext().CreateTempSymbol(); 79533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get(X86::PROLOG_LABEL)).addSym(Label); 79633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 79733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Define the current CFA rule to use the provided offset. 79809b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling unsigned Ptr = StackSize ? MachineLocation::VirtualFP : StackPtr; 79933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPDst(Ptr); 80033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPSrc(Ptr, StackOffset); 80133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(Label, SPDst, SPSrc)); 80233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov StackOffset += stackGrowth; 80333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 80433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 80533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 80633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov DL = MBB.findDebugLoc(MBBI); 80733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 80833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If there is an SUB32ri of ESP immediately before this instruction, merge 80933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // the two. This can be the case when tail call elimination is enabled and 81033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // the callee has more arguments then the caller. 81133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NumBytes -= mergeSPUpdates(MBB, MBBI, StackPtr, true); 81233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 81333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If there is an ADD32ri or SUB32ri of ESP immediately after this 81433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // instruction, merge the two instructions. 81533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov mergeSPUpdatesDown(MBB, MBBI, StackPtr, &NumBytes); 81633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 81733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Adjust stack pointer: ESP -= numbytes. 81833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 81933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Windows and cygwin/mingw require a prologue helper routine when allocating 82033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // more than 4K bytes on the stack. Windows uses __chkstk and cygwin/mingw 82133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // uses __alloca. __alloca and the 32-bit version of __chkstk will probe the 82233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // stack and adjust the stack pointer in one go. The 64-bit version of 82333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // __chkstk is only responsible for probing the stack. The 64-bit prologue is 82433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // responsible for adjusting the stack pointer. Touching the stack at 4K 82533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // increments is necessary to ensure that the guard pages used by the OS 82633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // virtual memory manager are allocated in correct sequence. 827a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (NumBytes >= 4096 && STI.isTargetCOFF() && !STI.isTargetEnvMacho()) { 828a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi const char *StackProbeSymbol; 829a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi bool isSPUpdateNeeded = false; 830a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi 831a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (Is64Bit) { 832a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (STI.isTargetCygMing()) 833a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi StackProbeSymbol = "___chkstk"; 834a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi else { 835a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi StackProbeSymbol = "__chkstk"; 836a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi isSPUpdateNeeded = true; 837a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi } 838a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi } else if (STI.isTargetCygMing()) 839a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi StackProbeSymbol = "_alloca"; 840a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi else 841a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi StackProbeSymbol = "_chkstk"; 842a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi 84333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Check whether EAX is livein for this function. 84433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool isEAXAlive = isEAXLiveIn(MF); 84533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 846a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (isEAXAlive) { 847a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // Sanity check that EAX is not livein for this function. 848a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // It should not be, so throw an assert. 849a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi assert(!Is64Bit && "EAX is livein in x64 case!"); 850a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi 85133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Save EAX 85233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get(X86::PUSH32r)) 853fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addReg(X86::EAX, RegState::Kill) 854fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .setMIFlag(MachineInstr::FrameSetup); 855a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi } 85633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 857a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (Is64Bit) { 858a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // Handle the 64-bit Windows ABI case where we need to call __chkstk. 859a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // Function prologue is responsible for adjusting the stack pointer. 860a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi BuildMI(MBB, MBBI, DL, TII.get(X86::MOV64ri), X86::RAX) 861fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addImm(NumBytes) 862fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .setMIFlag(MachineInstr::FrameSetup); 863a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi } else { 864a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // Allocate NumBytes-4 bytes on stack in case of isEAXAlive. 865a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // We'll also use 4 already allocated bytes for EAX. 86633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get(X86::MOV32ri), X86::EAX) 867fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addImm(isEAXAlive ? NumBytes - 4 : NumBytes) 868fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .setMIFlag(MachineInstr::FrameSetup); 869a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi } 870a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi 871a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi BuildMI(MBB, MBBI, DL, 872a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi TII.get(Is64Bit ? X86::W64ALLOCA : X86::CALLpcrel32)) 873a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi .addExternalSymbol(StackProbeSymbol) 874a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi .addReg(StackPtr, RegState::Define | RegState::Implicit) 875fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addReg(X86::EFLAGS, RegState::Define | RegState::Implicit) 876fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .setMIFlag(MachineInstr::FrameSetup); 877a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi 878a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // MSVC x64's __chkstk needs to adjust %rsp. 879a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // FIXME: %rax preserves the offset and should be available. 880a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (isSPUpdateNeeded) 881a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi emitSPUpdate(MBB, MBBI, StackPtr, -(int64_t)NumBytes, Is64Bit, 882a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi TII, *RegInfo); 883a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi 884a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi if (isEAXAlive) { 885a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi // Restore EAX 886a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi MachineInstr *MI = addRegOffset(BuildMI(MF, DL, TII.get(X86::MOV32rm), 887a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi X86::EAX), 888a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi StackPtr, false, NumBytes - 4); 889fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling MI->setFlag(MachineInstr::FrameSetup); 890a2e0762fae050464494a50c8b5d53ac2f4ba738cNAKAMURA Takumi MBB.insert(MBBI, MI); 89133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 89233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if (NumBytes) 8937158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng emitSPUpdate(MBB, MBBI, StackPtr, -(int64_t)NumBytes, Is64Bit, 8947158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng TII, *RegInfo); 89533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 896f0adba9a7ec8a3031876575a6ffb7db5f1b6f855Rafael Espindola if (( (!HasFP && NumBytes) || PushedRegs) && needsFrameMoves) { 89733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Mark end of stack pointer adjustment. 89833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MCSymbol *Label = MMI.getContext().CreateTempSymbol(); 899fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling BuildMI(MBB, MBBI, DL, TII.get(X86::PROLOG_LABEL)) 900fb4eb165d66b8ca5ae09c37d98d3ed937454b6afBill Wendling .addSym(Label); 90133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 90233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (!HasFP && NumBytes) { 90333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Define the current CFA rule to use the provided offset. 90433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (StackSize) { 90533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPDst(MachineLocation::VirtualFP); 90633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPSrc(MachineLocation::VirtualFP, 90733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov -StackSize + stackGrowth); 90833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(Label, SPDst, SPSrc)); 90933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 91033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPDst(StackPtr); 91133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineLocation SPSrc(StackPtr, stackGrowth); 91233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Moves.push_back(MachineMove(Label, SPDst, SPSrc)); 91333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 91433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 91533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 91633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Emit DWARF info specifying the offsets of the callee-saved registers. 91733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (PushedRegs) 91833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov emitCalleeSavedFrameMoves(MF, Label, HasFP ? FramePtr : StackPtr); 91933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 92009b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling 92109b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling // Darwin 10.7 and greater has support for compact unwind encoding. 922c8725d11f8756c57bdbceccc61062a9d560261c5Bill Wendling if (STI.getTargetTriple().isMacOSX() && 923ac86d43eae8bbfe5284d8e5fa17c141d1a7b1194Eli Friedman !STI.getTargetTriple().isMacOSXVersionLT(10, 7)) 92409b02c8dddf822e84c30cc08e7759d6ceebb2334Bill Wendling MMI.setCompactUnwindEncoding(getCompactUnwindEncoding(MF)); 92533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov} 92633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 92716c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikovvoid X86FrameLowering::emitEpilogue(MachineFunction &MF, 9283c2f0a11cce5a1e828e20675fa8467b624795e0aNick Lewycky MachineBasicBlock &MBB) const { 92933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov const MachineFrameInfo *MFI = MF.getFrameInfo(); 93033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); 931d9e3385ced2dc887e2fe8e1c071bd2611e4d3edeAnton Korobeynikov const X86RegisterInfo *RegInfo = TM.getRegisterInfo(); 932d9e3385ced2dc887e2fe8e1c071bd2611e4d3edeAnton Korobeynikov const X86InstrInfo &TII = *TM.getInstrInfo(); 9334f28c1c71450c711e96aa283de53739d8b4504cdJakob Stoklund Olesen MachineBasicBlock::iterator MBBI = MBB.getLastNonDebugInstr(); 9344f28c1c71450c711e96aa283de53739d8b4504cdJakob Stoklund Olesen assert(MBBI != MBB.end() && "Returning block has no instructions"); 93533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned RetOpcode = MBBI->getOpcode(); 93633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov DebugLoc DL = MBBI->getDebugLoc(); 93733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool Is64Bit = STI.is64Bit(); 93833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned StackAlign = getStackAlignment(); 93933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned SlotSize = RegInfo->getSlotSize(); 94033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned FramePtr = RegInfo->getFrameRegister(MF); 94133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned StackPtr = RegInfo->getStackRegister(); 94233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 94333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov switch (RetOpcode) { 94433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov default: 94533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov llvm_unreachable("Can only insert epilog into returning blocks"); 94633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::RET: 94733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::RETI: 94833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::TCRETURNdi: 94933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::TCRETURNri: 95033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::TCRETURNmi: 95133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::TCRETURNdi64: 95233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::TCRETURNri64: 95333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::TCRETURNmi64: 95433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::EH_RETURN: 95533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov case X86::EH_RETURN64: 95633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov break; // These are ok 95733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 95833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 95933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Get the number of bytes to allocate from the FrameInfo. 96033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t StackSize = MFI->getStackSize(); 96133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t MaxAlign = MFI->getMaxAlignment(); 96233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned CSSize = X86FI->getCalleeSavedFrameSize(); 96333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t NumBytes = 0; 96433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 96533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If we're forcing a stack realignment we can't rely on just the frame 96633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // info, we need to know the ABI stack alignment as well in case we 96733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // have a call out. Otherwise just make sure we have some alignment - we'll 96833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // go with the minimum. 96933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (ForceStackAlign) { 97033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (MFI->hasCalls()) 97133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MaxAlign = (StackAlign > MaxAlign) ? StackAlign : MaxAlign; 97233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov else 97333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MaxAlign = MaxAlign ? MaxAlign : 4; 97433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 97533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 976d0c38176690e9602a93a20a43f1bd084564a8116Anton Korobeynikov if (hasFP(MF)) { 97733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Calculate required stack adjustment. 97833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov uint64_t FrameSize = StackSize - SlotSize; 97933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (RegInfo->needsStackRealignment(MF)) 98033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov FrameSize = (FrameSize + MaxAlign - 1)/MaxAlign*MaxAlign; 98133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 98233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NumBytes = FrameSize - CSSize; 98333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 98433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Pop EBP. 98533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 98633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov TII.get(Is64Bit ? X86::POP64r : X86::POP32r), FramePtr); 98733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 98833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NumBytes = StackSize - CSSize; 98933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 99033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 99133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Skip the callee-saved pop instructions. 99233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock::iterator LastCSPop = MBBI; 99333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov while (MBBI != MBB.begin()) { 99433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineBasicBlock::iterator PI = prior(MBBI); 99533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned Opc = PI->getOpcode(); 99633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 9974f28c1c71450c711e96aa283de53739d8b4504cdJakob Stoklund Olesen if (Opc != X86::POP32r && Opc != X86::POP64r && Opc != X86::DBG_VALUE && 9985a96b3dad2f634c9081c8b2b6c2575441dc5a2bdEvan Cheng !PI->isTerminator()) 99933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov break; 100033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 100133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov --MBBI; 100233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 100333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 100433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov DL = MBBI->getDebugLoc(); 100533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 100633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If there is an ADD32ri or SUB32ri of ESP immediately before this 100733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // instruction, merge the two instructions. 100833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (NumBytes || MFI->hasVarSizedObjects()) 100933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov mergeSPUpdatesUp(MBB, MBBI, StackPtr, &NumBytes); 101033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 101133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // If dynamic alloca is used, then reset esp to point to the last callee-saved 101233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // slot before popping them off! Same applies for the case, when stack was 101333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // realigned. 101433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (RegInfo->needsStackRealignment(MF)) { 101533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // We cannot use LEA here, because stack pointer was realigned. We need to 101633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // deallocate local frame back. 101733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (CSSize) { 10187158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng emitSPUpdate(MBB, MBBI, StackPtr, NumBytes, Is64Bit, TII, *RegInfo); 101933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBBI = prior(LastCSPop); 102033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 102133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 102233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 102333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov TII.get(Is64Bit ? X86::MOV64rr : X86::MOV32rr), 102433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov StackPtr).addReg(FramePtr); 102533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if (MFI->hasVarSizedObjects()) { 102633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (CSSize) { 102733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov unsigned Opc = Is64Bit ? X86::LEA64r : X86::LEA32r; 102833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineInstr *MI = 102933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov addRegOffset(BuildMI(MF, DL, TII.get(Opc), StackPtr), 103033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov FramePtr, false, -CSSize); 103133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBB.insert(MBBI, MI); 103233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 103333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 103433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov TII.get(Is64Bit ? X86::MOV64rr : X86::MOV32rr), StackPtr) 103533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov .addReg(FramePtr); 103633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 103733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if (NumBytes) { 103833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Adjust stack pointer back: ESP += numbytes. 10397158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng emitSPUpdate(MBB, MBBI, StackPtr, NumBytes, Is64Bit, TII, *RegInfo); 104033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 104133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 104233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // We're returning from function via eh_return. 104333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (RetOpcode == X86::EH_RETURN || RetOpcode == X86::EH_RETURN64) { 10444f28c1c71450c711e96aa283de53739d8b4504cdJakob Stoklund Olesen MBBI = MBB.getLastNonDebugInstr(); 104533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineOperand &DestAddr = MBBI->getOperand(0); 104633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov assert(DestAddr.isReg() && "Offset should be in register!"); 104733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, 104833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov TII.get(Is64Bit ? X86::MOV64rr : X86::MOV32rr), 104933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov StackPtr).addReg(DestAddr.getReg()); 105033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if (RetOpcode == X86::TCRETURNri || RetOpcode == X86::TCRETURNdi || 105133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov RetOpcode == X86::TCRETURNmi || 105233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov RetOpcode == X86::TCRETURNri64 || RetOpcode == X86::TCRETURNdi64 || 105333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov RetOpcode == X86::TCRETURNmi64) { 105433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov bool isMem = RetOpcode == X86::TCRETURNmi || RetOpcode == X86::TCRETURNmi64; 105533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Tail call return: adjust the stack pointer and jump to callee. 1056f7ca976e74eafeeab0e9097f0fb07d6bb447415bJakob Stoklund Olesen MBBI = MBB.getLastNonDebugInstr(); 105733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineOperand &JumpTarget = MBBI->getOperand(0); 105833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineOperand &StackAdjust = MBBI->getOperand(isMem ? 5 : 1); 105933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov assert(StackAdjust.isImm() && "Expecting immediate value."); 106033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 106133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Adjust stack pointer. 106233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int StackAdj = StackAdjust.getImm(); 106333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int MaxTCDelta = X86FI->getTCReturnAddrDelta(); 106433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int Offset = 0; 106533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov assert(MaxTCDelta <= 0 && "MaxTCDelta should never be positive"); 106633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 106733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Incoporate the retaddr area. 106833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Offset = StackAdj-MaxTCDelta; 106933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov assert(Offset >= 0 && "Offset should never be negative"); 107033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 107133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (Offset) { 10727a2bdde0a0eebcd2125055e0eacaca040f0b766cChris Lattner // Check for possible merge with preceding ADD instruction. 107333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov Offset += mergeSPUpdates(MBB, MBBI, StackPtr, true); 10747158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng emitSPUpdate(MBB, MBBI, StackPtr, Offset, Is64Bit, TII, *RegInfo); 107533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 107633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 107733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Jump to label or value in register. 107833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov if (RetOpcode == X86::TCRETURNdi || RetOpcode == X86::TCRETURNdi64) { 10793d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng MachineInstrBuilder MIB = 10803d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng BuildMI(MBB, MBBI, DL, TII.get((RetOpcode == X86::TCRETURNdi) 10813d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng ? X86::TAILJMPd : X86::TAILJMPd64)); 10823d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng if (JumpTarget.isGlobal()) 10833d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng MIB.addGlobalAddress(JumpTarget.getGlobal(), JumpTarget.getOffset(), 10843d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng JumpTarget.getTargetFlags()); 10853d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng else { 10863d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng assert(JumpTarget.isSymbol()); 10873d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng MIB.addExternalSymbol(JumpTarget.getSymbolName(), 10883d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng JumpTarget.getTargetFlags()); 10893d2125c9dbac695c93f42c0f59fd040e413fd711Evan Cheng } 109033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if (RetOpcode == X86::TCRETURNmi || RetOpcode == X86::TCRETURNmi64) { 109133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineInstrBuilder MIB = 109233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get((RetOpcode == X86::TCRETURNmi) 109333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov ? X86::TAILJMPm : X86::TAILJMPm64)); 109433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov for (unsigned i = 0; i != 5; ++i) 109533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MIB.addOperand(MBBI->getOperand(i)); 109633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if (RetOpcode == X86::TCRETURNri64) { 109733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get(X86::TAILJMPr64)). 109833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov addReg(JumpTarget.getReg(), RegState::Kill); 109933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else { 110033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov BuildMI(MBB, MBBI, DL, TII.get(X86::TAILJMPr)). 110133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov addReg(JumpTarget.getReg(), RegState::Kill); 110233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 110333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 110433464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MachineInstr *NewMI = prior(MBBI); 110533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov for (unsigned i = 2, e = MBBI->getNumOperands(); i != e; ++i) 110633464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov NewMI->addOperand(MBBI->getOperand(i)); 110733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 110833464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Delete the pseudo instruction TCRETURN. 110933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov MBB.erase(MBBI); 111033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } else if ((RetOpcode == X86::RET || RetOpcode == X86::RETI) && 111133464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov (X86FI->getTCReturnAddrDelta() < 0)) { 111233464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov // Add the return addr area delta back since we are not tail calling. 111333464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov int delta = -1*X86FI->getTCReturnAddrDelta(); 11144f28c1c71450c711e96aa283de53739d8b4504cdJakob Stoklund Olesen MBBI = MBB.getLastNonDebugInstr(); 111533464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov 11167a2bdde0a0eebcd2125055e0eacaca040f0b766cChris Lattner // Check for possible merge with preceding ADD instruction. 111733464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov delta += mergeSPUpdates(MBB, MBBI, StackPtr, true); 11187158e08b8e619f4dcac9834c57f5f8afd6eea2ebEvan Cheng emitSPUpdate(MBB, MBBI, StackPtr, delta, Is64Bit, TII, *RegInfo); 111933464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov } 112033464912237efaa0ed7060829e66b59055bdd48bAnton Korobeynikov} 1121d9e3385ced2dc887e2fe8e1c071bd2611e4d3edeAnton Korobeynikov 112216c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikovint X86FrameLowering::getFrameIndexOffset(const MachineFunction &MF, int FI) const { 112382f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov const X86RegisterInfo *RI = 112482f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov static_cast<const X86RegisterInfo*>(MF.getTarget().getRegisterInfo()); 112582f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov const MachineFrameInfo *MFI = MF.getFrameInfo(); 112682f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov int Offset = MFI->getObjectOffset(FI) - getOffsetOfLocalArea(); 112782f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov uint64_t StackSize = MFI->getStackSize(); 112882f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov 112982f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov if (RI->needsStackRealignment(MF)) { 113082f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov if (FI < 0) { 113182f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov // Skip the saved EBP. 113282f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov Offset += RI->getSlotSize(); 113382f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov } else { 113417001ce25cc205ac1cd2604492c2bce310964220Duncan Sands assert((-(Offset + StackSize)) % MFI->getObjectAlignment(FI) == 0); 113582f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov return Offset + StackSize; 113682f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov } 113782f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov // FIXME: Support tail calls 113882f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov } else { 113982f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov if (!hasFP(MF)) 114082f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov return Offset + StackSize; 114182f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov 114282f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov // Skip the saved EBP. 114382f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov Offset += RI->getSlotSize(); 114482f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov 114582f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov // Skip the RETADDR move area 114682f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov const X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); 114782f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov int TailCallReturnAddrDelta = X86FI->getTCReturnAddrDelta(); 114882f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov if (TailCallReturnAddrDelta < 0) 114982f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov Offset -= TailCallReturnAddrDelta; 115082f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov } 115182f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov 115282f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov return Offset; 115382f58740c76b42af8370247b23677a0318f6dde8Anton Korobeynikov} 1154cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 115516c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikovbool X86FrameLowering::spillCalleeSavedRegisters(MachineBasicBlock &MBB, 1156cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov MachineBasicBlock::iterator MI, 1157cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov const std::vector<CalleeSavedInfo> &CSI, 1158cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov const TargetRegisterInfo *TRI) const { 1159cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov if (CSI.empty()) 1160cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov return false; 1161cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1162cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov DebugLoc DL = MBB.findDebugLoc(MI); 1163cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1164cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov MachineFunction &MF = *MBB.getParent(); 1165cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1166cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned SlotSize = STI.is64Bit() ? 8 : 4; 1167cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned FPReg = TRI->getFrameRegister(MF); 1168cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned CalleeFrameSize = 0; 1169cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1170cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov const TargetInstrInfo &TII = *MF.getTarget().getInstrInfo(); 1171cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); 1172cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1173419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // Push GPRs. It increases frame size. 1174cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned Opc = STI.is64Bit() ? X86::PUSH64r : X86::PUSH32r; 1175cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov for (unsigned i = CSI.size(); i != 0; --i) { 1176cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned Reg = CSI[i-1].getReg(); 1177419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi if (!X86::GR64RegClass.contains(Reg) && 1178419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi !X86::GR32RegClass.contains(Reg)) 1179419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi continue; 1180cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov // Add the callee-saved register as live-in. It's killed at the spill. 1181cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov MBB.addLiveIn(Reg); 1182cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov if (Reg == FPReg) 1183cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov // X86RegisterInfo::emitPrologue will handle spilling of frame register. 1184cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov continue; 1185419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi CalleeFrameSize += SlotSize; 1186aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis BuildMI(MBB, MI, DL, TII.get(Opc)).addReg(Reg, RegState::Kill) 1187aff232a5941c9ffb7ad52e08f81ad53794fed56bCharles Davis .setMIFlag(MachineInstr::FrameSetup); 1188cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov } 1189cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1190cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov X86FI->setCalleeSavedFrameSize(CalleeFrameSize); 1191419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi 1192419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // Make XMM regs spilled. X86 does not have ability of push/pop XMM. 1193419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // It can be done by spilling XMMs to stack frame. 1194419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // Note that only Win64 ABI might spill XMMs. 1195419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi for (unsigned i = CSI.size(); i != 0; --i) { 1196419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi unsigned Reg = CSI[i-1].getReg(); 1197419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi if (X86::GR64RegClass.contains(Reg) || 1198419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi X86::GR32RegClass.contains(Reg)) 1199419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi continue; 1200419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // Add the callee-saved register as live-in. It's killed at the spill. 1201419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi MBB.addLiveIn(Reg); 1202419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi const TargetRegisterClass *RC = TRI->getMinimalPhysRegClass(Reg); 1203419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi TII.storeRegToStackSlot(MBB, MI, Reg, true, CSI[i-1].getFrameIdx(), 1204419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi RC, TRI); 1205419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi } 1206419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi 1207cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov return true; 1208cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov} 1209cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 121016c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikovbool X86FrameLowering::restoreCalleeSavedRegisters(MachineBasicBlock &MBB, 1211cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov MachineBasicBlock::iterator MI, 1212cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov const std::vector<CalleeSavedInfo> &CSI, 1213cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov const TargetRegisterInfo *TRI) const { 1214cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov if (CSI.empty()) 1215cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov return false; 1216cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1217cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov DebugLoc DL = MBB.findDebugLoc(MI); 1218cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov 1219cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov MachineFunction &MF = *MBB.getParent(); 1220cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov const TargetInstrInfo &TII = *MF.getTarget().getInstrInfo(); 1221419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi 1222419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // Reload XMMs from stack frame. 1223419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi for (unsigned i = 0, e = CSI.size(); i != e; ++i) { 1224419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi unsigned Reg = CSI[i].getReg(); 1225419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi if (X86::GR64RegClass.contains(Reg) || 1226419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi X86::GR32RegClass.contains(Reg)) 1227419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi continue; 1228419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi const TargetRegisterClass *RC = TRI->getMinimalPhysRegClass(Reg); 1229419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi TII.loadRegFromStackSlot(MBB, MI, Reg, CSI[i].getFrameIdx(), 1230419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi RC, TRI); 1231419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi } 1232419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi 1233419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi // POP GPRs. 1234cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned FPReg = TRI->getFrameRegister(MF); 1235cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned Opc = STI.is64Bit() ? X86::POP64r : X86::POP32r; 1236cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov for (unsigned i = 0, e = CSI.size(); i != e; ++i) { 1237cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov unsigned Reg = CSI[i].getReg(); 1238419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi if (!X86::GR64RegClass.contains(Reg) && 1239419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi !X86::GR32RegClass.contains(Reg)) 1240419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi continue; 1241cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov if (Reg == FPReg) 1242cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov // X86RegisterInfo::emitEpilogue will handle restoring of frame register. 1243cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov continue; 1244419f23278388a8999b5c0bbe73ce823f63747f28NAKAMURA Takumi BuildMI(MBB, MI, DL, TII.get(Opc), Reg); 1245cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov } 1246cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov return true; 1247cd775ceff0b25a0b026f643a7990c2924bd310a3Anton Korobeynikov} 124894c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov 124994c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikovvoid 125016c29b5f285f375be53dabaa73e3e91107485fe4Anton KorobeynikovX86FrameLowering::processFunctionBeforeCalleeSavedScan(MachineFunction &MF, 125194c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov RegScavenger *RS) const { 125294c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov MachineFrameInfo *MFI = MF.getFrameInfo(); 125394c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov const X86RegisterInfo *RegInfo = TM.getRegisterInfo(); 125494c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov unsigned SlotSize = RegInfo->getSlotSize(); 125594c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov 125694c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); 125794c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov int32_t TailCallReturnAddrDelta = X86FI->getTCReturnAddrDelta(); 125894c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov 125994c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov if (TailCallReturnAddrDelta < 0) { 126094c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // create RETURNADDR area 126194c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // arg 126294c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // arg 126394c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // RETADDR 126494c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // { ... 126594c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // RETADDR area 126694c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // ... 126794c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // } 126894c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // [EBP] 126994c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov MFI->CreateFixedObject(-TailCallReturnAddrDelta, 127094c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov (-1U*SlotSize)+TailCallReturnAddrDelta, true); 127194c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov } 127294c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov 127394c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov if (hasFP(MF)) { 127494c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov assert((TailCallReturnAddrDelta <= 0) && 127594c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov "The Delta should always be zero or negative"); 127616c29b5f285f375be53dabaa73e3e91107485fe4Anton Korobeynikov const TargetFrameLowering &TFI = *MF.getTarget().getFrameLowering(); 127794c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov 127894c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov // Create a frame entry for the EBP register that must be saved. 127994c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov int FrameIdx = MFI->CreateFixedObject(SlotSize, 128094c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov -(int)SlotSize + 128194c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov TFI.getOffsetOfLocalArea() + 128294c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov TailCallReturnAddrDelta, 128394c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov true); 128494c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov assert(FrameIdx == MFI->getObjectIndexBegin() && 128594c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov "Slot for EBP register must be last in order to be found!"); 128617001ce25cc205ac1cd2604492c2bce310964220Duncan Sands (void)FrameIdx; 128794c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov } 128894c5ae08750f314bc3cf1bf882b686244a3927d9Anton Korobeynikov} 128976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 129076927d758657b3a511c73467ec5a7288795c1513Rafael Espindolastatic bool 129176927d758657b3a511c73467ec5a7288795c1513Rafael EspindolaHasNestArgument(const MachineFunction *MF) { 129276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola const Function *F = MF->getFunction(); 129376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola for (Function::const_arg_iterator I = F->arg_begin(), E = F->arg_end(); 129476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola I != E; I++) { 129576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (I->hasNestAttr()) 129676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola return true; 129776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 129876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola return false; 129976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola} 130076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 130176927d758657b3a511c73467ec5a7288795c1513Rafael Espindolastatic unsigned 130276927d758657b3a511c73467ec5a7288795c1513Rafael EspindolaGetScratchRegister(bool Is64Bit, const MachineFunction &MF) { 130376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (Is64Bit) { 130476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola return X86::R11; 130576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } else { 130676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola CallingConv::ID CallingConvention = MF.getFunction()->getCallingConv(); 130776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola bool IsNested = HasNestArgument(&MF); 130876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 130976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (CallingConvention == CallingConv::X86_FastCall) { 131076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (IsNested) { 1311e81abfd30b7c81735e9f3d79006c2653690aa592Rafael Espindola report_fatal_error("Segmented stacks does not support fastcall with " 1312e81abfd30b7c81735e9f3d79006c2653690aa592Rafael Espindola "nested function."); 131376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola return -1; 131476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } else { 131576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola return X86::EAX; 131676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 131776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } else { 131876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (IsNested) 131976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola return X86::EDX; 132076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola else 132176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola return X86::ECX; 132276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 132376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 132476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola} 132576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 1326199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das// The stack limit in the TCB is set to this many bytes above the actual stack 1327199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das// limit. 1328199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Dasstatic const uint64_t kSplitStackAvailable = 256; 1329199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das 133076927d758657b3a511c73467ec5a7288795c1513Rafael Espindolavoid 133176927d758657b3a511c73467ec5a7288795c1513Rafael EspindolaX86FrameLowering::adjustForSegmentedStacks(MachineFunction &MF) const { 133276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MachineBasicBlock &prologueMBB = MF.front(); 133376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MachineFrameInfo *MFI = MF.getFrameInfo(); 133476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola const X86InstrInfo &TII = *TM.getInstrInfo(); 133576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola uint64_t StackSize; 133676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola bool Is64Bit = STI.is64Bit(); 133776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola unsigned TlsReg, TlsOffset; 133876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola DebugLoc DL; 133976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola const X86Subtarget *ST = &MF.getTarget().getSubtarget<X86Subtarget>(); 134076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 134176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola unsigned ScratchReg = GetScratchRegister(Is64Bit, MF); 134276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola assert(!MF.getRegInfo().isLiveIn(ScratchReg) && 134376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola "Scratch register is live-in"); 134476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 134576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (MF.getFunction()->isVarArg()) 134676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola report_fatal_error("Segmented stacks do not support vararg functions."); 134776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (!ST->isTargetLinux()) 134876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola report_fatal_error("Segmented stacks supported only on linux."); 134976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 135076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MachineBasicBlock *allocMBB = MF.CreateMachineBasicBlock(); 135176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MachineBasicBlock *checkMBB = MF.CreateMachineBasicBlock(); 135276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola X86MachineFunctionInfo *X86FI = MF.getInfo<X86MachineFunctionInfo>(); 135376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola bool IsNested = false; 135476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 135576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // We need to know if the function has a nest argument only in 64 bit mode. 135676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (Is64Bit) 135776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola IsNested = HasNestArgument(&MF); 135876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 13594e68054b20725f6ec1cac33630258f749fe5debeBill Wendling // The MOV R10, RAX needs to be in a different block, since the RET we emit in 13604e68054b20725f6ec1cac33630258f749fe5debeBill Wendling // allocMBB needs to be last (terminating) instruction. 13614e68054b20725f6ec1cac33630258f749fe5debeBill Wendling 136276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola for (MachineBasicBlock::livein_iterator i = prologueMBB.livein_begin(), 136376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola e = prologueMBB.livein_end(); i != e; i++) { 136476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola allocMBB->addLiveIn(*i); 136576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola checkMBB->addLiveIn(*i); 136676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 136776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 1368e840e88239cf92a065cbf5f5b9c7d18bc139c0e1Rafael Espindola if (IsNested) 136976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola allocMBB->addLiveIn(X86::R10); 137076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 137176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MF.push_front(allocMBB); 137276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MF.push_front(checkMBB); 137376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 137476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // Eventually StackSize will be calculated by a link-time pass; which will 137576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // also decide whether checking code needs to be injected into this particular 137676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // prologue. 137776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola StackSize = MFI->getStackSize(); 137876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 137976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // Read the limit off the current stacklet off the stack_guard location. 138076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (Is64Bit) { 138176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola TlsReg = X86::FS; 138276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola TlsOffset = 0x70; 138376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 1384199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das if (StackSize < kSplitStackAvailable) 1385199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das ScratchReg = X86::RSP; 1386199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das else 1387199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das BuildMI(checkMBB, DL, TII.get(X86::LEA64r), ScratchReg).addReg(X86::RSP) 1388199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das .addImm(0).addReg(0).addImm(-StackSize).addReg(0); 1389199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das 139076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(checkMBB, DL, TII.get(X86::CMP64rm)).addReg(ScratchReg) 139176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addReg(0).addImm(0).addReg(0).addImm(TlsOffset).addReg(TlsReg); 139276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } else { 139376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola TlsReg = X86::GS; 139476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola TlsOffset = 0x30; 139576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 1396199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das if (StackSize < kSplitStackAvailable) 1397199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das ScratchReg = X86::ESP; 1398199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das else 1399199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das BuildMI(checkMBB, DL, TII.get(X86::LEA32r), ScratchReg).addReg(X86::ESP) 1400199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das .addImm(0).addReg(0).addImm(-StackSize).addReg(0); 1401199ce33b3bfa0b6293946c1b835da2a1fbc8cab4Sanjoy Das 140276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(checkMBB, DL, TII.get(X86::CMP32rm)).addReg(ScratchReg) 140376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addReg(0).addImm(0).addReg(0).addImm(TlsOffset).addReg(TlsReg); 140476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 140576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 140676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // This jump is taken if SP >= (Stacklet Limit + Stack Space required). 140776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // It jumps to normal execution of the function body. 140876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(checkMBB, DL, TII.get(X86::JG_4)).addMBB(&prologueMBB); 140976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 141076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // On 32 bit we first push the arguments size and then the frame size. On 64 141176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // bit, we pass the stack frame size in r10 and the argument size in r11. 141276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (Is64Bit) { 141376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // Functions with nested arguments use R10, so it needs to be saved across 141476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // the call to _morestack 141576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 141676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (IsNested) 141776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::MOV64rr), X86::RAX).addReg(X86::R10); 141876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 141976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::MOV64ri), X86::R10) 142076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addImm(StackSize); 142176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::MOV64ri), X86::R11) 142276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addImm(X86FI->getArgumentStackSize()); 142376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MF.getRegInfo().setPhysRegUsed(X86::R10); 142476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MF.getRegInfo().setPhysRegUsed(X86::R11); 142576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } else { 142676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::PUSHi32)) 142776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addImm(X86FI->getArgumentStackSize()); 142876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::PUSHi32)) 142976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addImm(StackSize); 143076927d758657b3a511c73467ec5a7288795c1513Rafael Espindola } 143176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 143276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola // __morestack is in libgcc 143376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola if (Is64Bit) 143476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::CALL64pcrel32)) 143576927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addExternalSymbol("__morestack"); 143676927d758657b3a511c73467ec5a7288795c1513Rafael Espindola else 143776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::CALLpcrel32)) 143876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola .addExternalSymbol("__morestack"); 143976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 14404e68054b20725f6ec1cac33630258f749fe5debeBill Wendling if (IsNested) 1441e840e88239cf92a065cbf5f5b9c7d18bc139c0e1Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::MORESTACK_RET_RESTORE_R10)); 1442e840e88239cf92a065cbf5f5b9c7d18bc139c0e1Rafael Espindola else 1443e840e88239cf92a065cbf5f5b9c7d18bc139c0e1Rafael Espindola BuildMI(allocMBB, DL, TII.get(X86::MORESTACK_RET)); 144476927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 1445e840e88239cf92a065cbf5f5b9c7d18bc139c0e1Rafael Espindola allocMBB->addSuccessor(&prologueMBB); 144682222c20be24adda7c218f3fdaf2e0ae049c955bBill Wendling 144776927d758657b3a511c73467ec5a7288795c1513Rafael Espindola checkMBB->addSuccessor(allocMBB); 144876927d758657b3a511c73467ec5a7288795c1513Rafael Espindola checkMBB->addSuccessor(&prologueMBB); 144976927d758657b3a511c73467ec5a7288795c1513Rafael Espindola 145051f0c7641983469cbd29f8862a121645471a885aJakob Stoklund Olesen#ifdef XDEBUG 145176927d758657b3a511c73467ec5a7288795c1513Rafael Espindola MF.verify(); 145276927d758657b3a511c73467ec5a7288795c1513Rafael Espindola#endif 145376927d758657b3a511c73467ec5a7288795c1513Rafael Espindola} 1454