Home | History | Annotate | Download | only in Mips
      1 //=======- MipsFrameLowering.cpp - Mips Frame Information ------*- C++ -*-====//
      2 //
      3 //                     The LLVM Compiler Infrastructure
      4 //
      5 // This file is distributed under the University of Illinois Open Source
      6 // License. See LICENSE.TXT for details.
      7 //
      8 //===----------------------------------------------------------------------===//
      9 //
     10 // This file contains the Mips implementation of TargetFrameLowering class.
     11 //
     12 //===----------------------------------------------------------------------===//
     13 
     14 #include "MipsFrameLowering.h"
     15 #include "MipsInstrInfo.h"
     16 #include "MipsMachineFunction.h"
     17 #include "llvm/Function.h"
     18 #include "llvm/CodeGen/MachineFrameInfo.h"
     19 #include "llvm/CodeGen/MachineFunction.h"
     20 #include "llvm/CodeGen/MachineInstrBuilder.h"
     21 #include "llvm/CodeGen/MachineModuleInfo.h"
     22 #include "llvm/CodeGen/MachineRegisterInfo.h"
     23 #include "llvm/Target/TargetData.h"
     24 #include "llvm/Target/TargetOptions.h"
     25 #include "llvm/Support/CommandLine.h"
     26 
     27 using namespace llvm;
     28 
     29 
     30 //===----------------------------------------------------------------------===//
     31 //
     32 // Stack Frame Processing methods
     33 // +----------------------------+
     34 //
     35 // The stack is allocated decrementing the stack pointer on
     36 // the first instruction of a function prologue. Once decremented,
     37 // all stack references are done thought a positive offset
     38 // from the stack/frame pointer, so the stack is considering
     39 // to grow up! Otherwise terrible hacks would have to be made
     40 // to get this stack ABI compliant :)
     41 //
     42 //  The stack frame required by the ABI (after call):
     43 //  Offset
     44 //
     45 //  0                 ----------
     46 //  4                 Args to pass
     47 //  .                 saved $GP  (used in PIC)
     48 //  .                 Alloca allocations
     49 //  .                 Local Area
     50 //  .                 CPU "Callee Saved" Registers
     51 //  .                 saved FP
     52 //  .                 saved RA
     53 //  .                 FPU "Callee Saved" Registers
     54 //  StackSize         -----------
     55 //
     56 // Offset - offset from sp after stack allocation on function prologue
     57 //
     58 // The sp is the stack pointer subtracted/added from the stack size
     59 // at the Prologue/Epilogue
     60 //
     61 // References to the previous stack (to obtain arguments) are done
     62 // with offsets that exceeds the stack size: (stacksize+(4*(num_arg-1))
     63 //
     64 // Examples:
     65 // - reference to the actual stack frame
     66 //   for any local area var there is smt like : FI >= 0, StackOffset: 4
     67 //     sw REGX, 4(SP)
     68 //
     69 // - reference to previous stack frame
     70 //   suppose there's a load to the 5th arguments : FI < 0, StackOffset: 16.
     71 //   The emitted instruction will be something like:
     72 //     lw REGX, 16+StackSize(SP)
     73 //
     74 // Since the total stack size is unknown on LowerFormalArguments, all
     75 // stack references (ObjectOffset) created to reference the function
     76 // arguments, are negative numbers. This way, on eliminateFrameIndex it's
     77 // possible to detect those references and the offsets are adjusted to
     78 // their real location.
     79 //
     80 //===----------------------------------------------------------------------===//
     81 
     82 // hasFP - Return true if the specified function should have a dedicated frame
     83 // pointer register.  This is true if the function has variable sized allocas or
     84 // if frame pointer elimination is disabled.
     85 bool MipsFrameLowering::hasFP(const MachineFunction &MF) const {
     86   const MachineFrameInfo *MFI = MF.getFrameInfo();
     87   return DisableFramePointerElim(MF) || MFI->hasVarSizedObjects()
     88       || MFI->isFrameAddressTaken();
     89 }
     90 
     91 bool MipsFrameLowering::targetHandlesStackFrameRounding() const {
     92   return true;
     93 }
     94 
     95 static unsigned AlignOffset(unsigned Offset, unsigned Align) {
     96   return (Offset + Align - 1) / Align * Align;
     97 }
     98 
     99 // expand pair of register and immediate if the immediate doesn't fit in the
    100 // 16-bit offset field.
    101 // e.g.
    102 //  if OrigImm = 0x10000, OrigReg = $sp:
    103 //    generate the following sequence of instrs:
    104 //      lui  $at, hi(0x10000)
    105 //      addu $at, $sp, $at
    106 //
    107 //    (NewReg, NewImm) = ($at, lo(Ox10000))
    108 //    return true
    109 static bool expandRegLargeImmPair(unsigned OrigReg, int OrigImm,
    110                                   unsigned& NewReg, int& NewImm,
    111                                   MachineBasicBlock& MBB,
    112                                   MachineBasicBlock::iterator I) {
    113   // OrigImm fits in the 16-bit field
    114   if (OrigImm < 0x8000 && OrigImm >= -0x8000) {
    115     NewReg = OrigReg;
    116     NewImm = OrigImm;
    117     return false;
    118   }
    119 
    120   MachineFunction* MF = MBB.getParent();
    121   const TargetInstrInfo *TII = MF->getTarget().getInstrInfo();
    122   DebugLoc DL = I->getDebugLoc();
    123   int ImmLo = (short)(OrigImm & 0xffff);
    124   int ImmHi = (((unsigned)OrigImm & 0xffff0000) >> 16) +
    125               ((OrigImm & 0x8000) != 0);
    126 
    127   // FIXME: change this when mips goes MC".
    128   BuildMI(MBB, I, DL, TII->get(Mips::NOAT));
    129   BuildMI(MBB, I, DL, TII->get(Mips::LUi), Mips::AT).addImm(ImmHi);
    130   BuildMI(MBB, I, DL, TII->get(Mips::ADDu), Mips::AT).addReg(OrigReg)
    131                                                      .addReg(Mips::AT);
    132   NewReg = Mips::AT;
    133   NewImm = ImmLo;
    134 
    135   return true;
    136 }
    137 
    138 void MipsFrameLowering::emitPrologue(MachineFunction &MF) const {
    139   MachineBasicBlock &MBB   = MF.front();
    140   MachineFrameInfo *MFI    = MF.getFrameInfo();
    141   MipsFunctionInfo *MipsFI = MF.getInfo<MipsFunctionInfo>();
    142   const MipsRegisterInfo *RegInfo =
    143     static_cast<const MipsRegisterInfo*>(MF.getTarget().getRegisterInfo());
    144   const MipsInstrInfo &TII =
    145     *static_cast<const MipsInstrInfo*>(MF.getTarget().getInstrInfo());
    146   MachineBasicBlock::iterator MBBI = MBB.begin();
    147   DebugLoc dl = MBBI != MBB.end() ? MBBI->getDebugLoc() : DebugLoc();
    148   bool isPIC = (MF.getTarget().getRelocationModel() == Reloc::PIC_);
    149   unsigned NewReg = 0;
    150   int NewImm = 0;
    151   bool ATUsed;
    152 
    153   // First, compute final stack size.
    154   unsigned RegSize = STI.isGP32bit() ? 4 : 8;
    155   unsigned StackAlign = getStackAlignment();
    156   unsigned LocalVarAreaOffset = MipsFI->needGPSaveRestore() ?
    157     (MFI->getObjectOffset(MipsFI->getGPFI()) + RegSize) :
    158     MipsFI->getMaxCallFrameSize();
    159   unsigned StackSize = AlignOffset(LocalVarAreaOffset, StackAlign) +
    160     AlignOffset(MFI->getStackSize(), StackAlign);
    161 
    162    // Update stack size
    163   MFI->setStackSize(StackSize);
    164 
    165   BuildMI(MBB, MBBI, dl, TII.get(Mips::NOREORDER));
    166 
    167   // TODO: check need from GP here.
    168   if (isPIC && STI.isABI_O32())
    169     BuildMI(MBB, MBBI, dl, TII.get(Mips::CPLOAD))
    170       .addReg(RegInfo->getPICCallReg());
    171   BuildMI(MBB, MBBI, dl, TII.get(Mips::NOMACRO));
    172 
    173   // No need to allocate space on the stack.
    174   if (StackSize == 0 && !MFI->adjustsStack()) return;
    175 
    176   MachineModuleInfo &MMI = MF.getMMI();
    177   std::vector<MachineMove> &Moves = MMI.getFrameMoves();
    178   MachineLocation DstML, SrcML;
    179 
    180   // Adjust stack : addi sp, sp, (-imm)
    181   ATUsed = expandRegLargeImmPair(Mips::SP, -StackSize, NewReg, NewImm, MBB,
    182                                  MBBI);
    183   BuildMI(MBB, MBBI, dl, TII.get(Mips::ADDiu), Mips::SP)
    184     .addReg(NewReg).addImm(NewImm);
    185 
    186   // FIXME: change this when mips goes MC".
    187   if (ATUsed)
    188     BuildMI(MBB, MBBI, dl, TII.get(Mips::ATMACRO));
    189 
    190   // emit ".cfi_def_cfa_offset StackSize"
    191   MCSymbol *AdjustSPLabel = MMI.getContext().CreateTempSymbol();
    192   BuildMI(MBB, MBBI, dl,
    193           TII.get(TargetOpcode::PROLOG_LABEL)).addSym(AdjustSPLabel);
    194   DstML = MachineLocation(MachineLocation::VirtualFP);
    195   SrcML = MachineLocation(MachineLocation::VirtualFP, -StackSize);
    196   Moves.push_back(MachineMove(AdjustSPLabel, DstML, SrcML));
    197 
    198   const std::vector<CalleeSavedInfo> &CSI = MFI->getCalleeSavedInfo();
    199 
    200   if (CSI.size()) {
    201     // Find the instruction past the last instruction that saves a callee-saved
    202     // register to the stack.
    203     for (unsigned i = 0; i < CSI.size(); ++i)
    204       ++MBBI;
    205 
    206     // Iterate over list of callee-saved registers and emit .cfi_offset
    207     // directives.
    208     MCSymbol *CSLabel = MMI.getContext().CreateTempSymbol();
    209     BuildMI(MBB, MBBI, dl,
    210             TII.get(TargetOpcode::PROLOG_LABEL)).addSym(CSLabel);
    211 
    212     for (std::vector<CalleeSavedInfo>::const_iterator I = CSI.begin(),
    213            E = CSI.end(); I != E; ++I) {
    214       int64_t Offset = MFI->getObjectOffset(I->getFrameIdx());
    215       unsigned Reg = I->getReg();
    216 
    217       // If Reg is a double precision register, emit two cfa_offsets,
    218       // one for each of the paired single precision registers.
    219       if (Mips::AFGR64RegisterClass->contains(Reg)) {
    220         const unsigned *SubRegs = RegInfo->getSubRegisters(Reg);
    221         MachineLocation DstML0(MachineLocation::VirtualFP, Offset);
    222         MachineLocation DstML1(MachineLocation::VirtualFP, Offset + 4);
    223         MachineLocation SrcML0(*SubRegs);
    224         MachineLocation SrcML1(*(SubRegs + 1));
    225 
    226         if (!STI.isLittle())
    227           std::swap(SrcML0, SrcML1);
    228 
    229         Moves.push_back(MachineMove(CSLabel, DstML0, SrcML0));
    230         Moves.push_back(MachineMove(CSLabel, DstML1, SrcML1));
    231       }
    232       else {
    233         // Reg is either in CPURegs or FGR32.
    234         DstML = MachineLocation(MachineLocation::VirtualFP, Offset);
    235         SrcML = MachineLocation(Reg);
    236         Moves.push_back(MachineMove(CSLabel, DstML, SrcML));
    237       }
    238     }
    239   }
    240 
    241   // if framepointer enabled, set it to point to the stack pointer.
    242   if (hasFP(MF)) {
    243     // Insert instruction "move $fp, $sp" at this location.
    244     BuildMI(MBB, MBBI, dl, TII.get(Mips::ADDu), Mips::FP)
    245       .addReg(Mips::SP).addReg(Mips::ZERO);
    246 
    247     // emit ".cfi_def_cfa_register $fp"
    248     MCSymbol *SetFPLabel = MMI.getContext().CreateTempSymbol();
    249     BuildMI(MBB, MBBI, dl,
    250             TII.get(TargetOpcode::PROLOG_LABEL)).addSym(SetFPLabel);
    251     DstML = MachineLocation(Mips::FP);
    252     SrcML = MachineLocation(MachineLocation::VirtualFP);
    253     Moves.push_back(MachineMove(SetFPLabel, DstML, SrcML));
    254   }
    255 
    256   // Restore GP from the saved stack location
    257   if (MipsFI->needGPSaveRestore()) {
    258     unsigned Offset = MFI->getObjectOffset(MipsFI->getGPFI());
    259     BuildMI(MBB, MBBI, dl, TII.get(Mips::CPRESTORE)).addImm(Offset);
    260 
    261     if (Offset >= 0x8000) {
    262       BuildMI(MBB, llvm::prior(MBBI), dl, TII.get(Mips::MACRO));
    263       BuildMI(MBB, MBBI, dl, TII.get(Mips::NOMACRO));
    264     }
    265   }
    266 }
    267 
    268 void MipsFrameLowering::emitEpilogue(MachineFunction &MF,
    269                                  MachineBasicBlock &MBB) const {
    270   MachineBasicBlock::iterator MBBI = MBB.getLastNonDebugInstr();
    271   MachineFrameInfo *MFI            = MF.getFrameInfo();
    272   const MipsInstrInfo &TII =
    273     *static_cast<const MipsInstrInfo*>(MF.getTarget().getInstrInfo());
    274   DebugLoc dl = MBBI->getDebugLoc();
    275 
    276   // Get the number of bytes from FrameInfo
    277   unsigned StackSize = MFI->getStackSize();
    278 
    279   unsigned NewReg = 0;
    280   int NewImm = 0;
    281   bool ATUsed = false;
    282 
    283   // if framepointer enabled, restore the stack pointer.
    284   if (hasFP(MF)) {
    285     // Find the first instruction that restores a callee-saved register.
    286     MachineBasicBlock::iterator I = MBBI;
    287 
    288     for (unsigned i = 0; i < MFI->getCalleeSavedInfo().size(); ++i)
    289       --I;
    290 
    291     // Insert instruction "move $sp, $fp" at this location.
    292     BuildMI(MBB, I, dl, TII.get(Mips::ADDu), Mips::SP)
    293       .addReg(Mips::FP).addReg(Mips::ZERO);
    294   }
    295 
    296   // adjust stack  : insert addi sp, sp, (imm)
    297   if (StackSize) {
    298     ATUsed = expandRegLargeImmPair(Mips::SP, StackSize, NewReg, NewImm, MBB,
    299                                    MBBI);
    300     BuildMI(MBB, MBBI, dl, TII.get(Mips::ADDiu), Mips::SP)
    301       .addReg(NewReg).addImm(NewImm);
    302 
    303     // FIXME: change this when mips goes MC".
    304     if (ATUsed)
    305       BuildMI(MBB, MBBI, dl, TII.get(Mips::ATMACRO));
    306   }
    307 }
    308 
    309 void MipsFrameLowering::
    310 processFunctionBeforeCalleeSavedScan(MachineFunction &MF,
    311                                      RegScavenger *RS) const {
    312   MachineRegisterInfo& MRI = MF.getRegInfo();
    313 
    314   // FIXME: remove this code if register allocator can correctly mark
    315   //        $fp and $ra used or unused.
    316 
    317   // Mark $fp and $ra as used or unused.
    318   if (hasFP(MF))
    319     MRI.setPhysRegUsed(Mips::FP);
    320 
    321   // The register allocator might determine $ra is used after seeing
    322   // instruction "jr $ra", but we do not want PrologEpilogInserter to insert
    323   // instructions to save/restore $ra unless there is a function call.
    324   // To correct this, $ra is explicitly marked unused if there is no
    325   // function call.
    326   if (MF.getFrameInfo()->hasCalls())
    327     MRI.setPhysRegUsed(Mips::RA);
    328   else
    329     MRI.setPhysRegUnused(Mips::RA);
    330 }
    331