lib/Target/X86/AsmPrinter/X86MCInstLower.cpp

   1 //===-- X86MCInstLower.cpp - Convert X86 MachineInstr to an MCInst --------===//
   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 code to lower X86 MachineInstrs to their corresponding
  11 // MCInst records.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15
  16 #include "X86ATTAsmPrinter.h"
  17 #include "X86MCAsmInfo.h"
  18 #include "llvm/MC/MCContext.h"
  19 #include "llvm/MC/MCExpr.h"
  20 #include "llvm/MC/MCInst.h"
  21 #include "llvm/MC/MCStreamer.h"
  22 #include "llvm/Support/FormattedStream.h"
  23 #include "llvm/Support/Mangler.h"
  24 #include "llvm/ADT/SmallString.h"
  25 using namespace llvm;
  26
  27 MCSymbol *X86ATTAsmPrinter::GetPICBaseSymbol() {
  28   // FIXME: the actual label generated doesn't matter here!  Just mangle in
  29   // something unique (the function number) with Private prefix.
  30   SmallString<60> Name;
  31
  32   if (Subtarget->isTargetDarwin()) {
  33     raw_svector_ostream(Name) << 'L' << getFunctionNumber() << "$pb";
  34   } else {
  35     assert(Subtarget->isTargetELF() && "Don't know how to print PIC label!");
  36     raw_svector_ostream(Name) << ".Lllvm$" << getFunctionNumber()<<".$piclabel";
  37   }
  38   return OutContext.GetOrCreateSymbol(Name.str());
  39 }
  40
  41
  42
  43 static void lower_lea64_32mem(MCInst *MI, unsigned OpNo) {
  44   // Convert registers in the addr mode according to subreg64.
  45   for (unsigned i = 0; i != 4; ++i) {
  46     if (!MI->getOperand(i).isReg()) continue;
  47
  48     unsigned Reg = MI->getOperand(i).getReg();
  49     if (Reg == 0) continue;
  50
  51     MI->getOperand(i).setReg(getX86SubSuperRegister(Reg, MVT::i64));
  52   }
  53 }
  54
  55 /// LowerGlobalAddressOperand - Lower an MO_GlobalAddress operand to an
  56 /// MCOperand.
  57 MCSymbol *X86ATTAsmPrinter::GetGlobalAddressSymbol(const MachineOperand &MO) {
  58   const GlobalValue *GV = MO.getGlobal();
  59
  60   const char *Suffix = "";
  61   if (MO.getTargetFlags() == X86II::MO_DARWIN_STUB)
  62     Suffix = "$stub";
  63   else if (MO.getTargetFlags() == X86II::MO_DARWIN_NONLAZY ||
  64            MO.getTargetFlags() == X86II::MO_DARWIN_NONLAZY_PIC_BASE ||
  65            MO.getTargetFlags() == X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE)
  66     Suffix = "$non_lazy_ptr";
  67
  68   std::string Name = Mang->getMangledName(GV, Suffix, Suffix[0] != '\0');
  69   if (Subtarget->isTargetCygMing())
  70     DecorateCygMingName(Name, GV);
  71
  72   switch (MO.getTargetFlags()) {
  73   default: llvm_unreachable("Unknown target flag on GV operand");
  74   case X86II::MO_NO_FLAG:                // No flag.
  75   case X86II::MO_GOT_ABSOLUTE_ADDRESS:   // Doesn't modify symbol name.
  76   case X86II::MO_PIC_BASE_OFFSET:        // Doesn't modify symbol name.
  77     break;
  78   case X86II::MO_DLLIMPORT:
  79     // Handle dllimport linkage.
  80     Name = "__imp_" + Name;
  81     break;
  82   case X86II::MO_DARWIN_NONLAZY:
  83   case X86II::MO_DARWIN_NONLAZY_PIC_BASE:
  84     GVStubs[Name] = Mang->getMangledName(GV);
  85     break;
  86   case X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE:
  87     HiddenGVStubs[Name] = Mang->getMangledName(GV);
  88     break;
  89   case X86II::MO_DARWIN_STUB:
  90     FnStubs[Name] = Mang->getMangledName(GV);
  91     break;
  92   // FIXME: These probably should be a modifier on the symbol or something??
  93   case X86II::MO_TLSGD:     Name += "@TLSGD";     break;
  94   case X86II::MO_GOTTPOFF:  Name += "@GOTTPOFF";  break;
  95   case X86II::MO_INDNTPOFF: Name += "@INDNTPOFF"; break;
  96   case X86II::MO_TPOFF:     Name += "@TPOFF";     break;
  97   case X86II::MO_NTPOFF:    Name += "@NTPOFF";    break;
  98   case X86II::MO_GOTPCREL:  Name += "@GOTPCREL";  break;
  99   case X86II::MO_GOT:       Name += "@GOT";       break;
 100   case X86II::MO_GOTOFF:    Name += "@GOTOFF";    break;
 101   case X86II::MO_PLT:       Name += "@PLT";       break;
 102   }
 103
 104   return OutContext.GetOrCreateSymbol(Name);
 105 }
 106
 107 MCSymbol *X86ATTAsmPrinter::GetExternalSymbolSymbol(const MachineOperand &MO) {
 108   std::string Name = Mang->makeNameProper(MO.getSymbolName());
 109   if (MO.getTargetFlags() == X86II::MO_DARWIN_STUB) {
 110     FnStubs[Name+"$stub"] = Name;
 111     Name += "$stub";
 112   }
 113
 114   return OutContext.GetOrCreateSymbol(Name);
 115 }
 116
 117 MCSymbol *X86ATTAsmPrinter::GetJumpTableSymbol(const MachineOperand &MO) {
 118   SmallString<256> Name;
 119   raw_svector_ostream(Name) << MAI->getPrivateGlobalPrefix() << "JTI"
 120     << getFunctionNumber() << '_' << MO.getIndex();
 121
 122   MCSymbol *NegatedSymbol = 0;
 123   switch (MO.getTargetFlags()) {
 124   default:
 125     llvm_unreachable("Unknown target flag on GV operand");
 126   case X86II::MO_NO_FLAG:    // No flag.
 127   case X86II::MO_PIC_BASE_OFFSET:
 128   case X86II::MO_DARWIN_NONLAZY_PIC_BASE:
 129   case X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE:
 130     break;
 131     // Subtract the pic base.
 132     NegatedSymbol = GetPICBaseSymbol();
 133     break;
 134   }
 135
 136   // Create a symbol for the name.
 137   return OutContext.GetOrCreateSymbol(Name.str());
 138 }
 139
 140
 141 MCSymbol *X86ATTAsmPrinter::
 142 GetConstantPoolIndexSymbol(const MachineOperand &MO) {
 143   SmallString<256> Name;
 144   raw_svector_ostream(Name) << MAI->getPrivateGlobalPrefix() << "CPI"
 145   << getFunctionNumber() << '_' << MO.getIndex();
 146
 147   switch (MO.getTargetFlags()) {
 148   default:
 149     llvm_unreachable("Unknown target flag on GV operand");
 150   case X86II::MO_NO_FLAG:    // No flag.
 151   case X86II::MO_PIC_BASE_OFFSET:
 152   case X86II::MO_DARWIN_NONLAZY_PIC_BASE:
 153   case X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE:
 154     break;
 155   }
 156
 157   // Create a symbol for the name.
 158   return OutContext.GetOrCreateSymbol(Name.str());
 159 }
 160
 161 MCOperand X86ATTAsmPrinter::LowerSymbolOperand(const MachineOperand &MO,
 162                                                MCSymbol *Sym) {
 163   // FIXME: We would like an efficient form for this, so we don't have to do a
 164   // lot of extra uniquing.
 165   const MCExpr *Expr = MCSymbolRefExpr::Create(Sym, OutContext);
 166
 167   switch (MO.getTargetFlags()) {
 168   default: llvm_unreachable("Unknown target flag on GV operand");
 169   case X86II::MO_NO_FLAG:    // No flag.
 170
 171   // These affect the name of the symbol, not any suffix.
 172   case X86II::MO_DARWIN_NONLAZY:
 173   case X86II::MO_DLLIMPORT:
 174   case X86II::MO_DARWIN_STUB:
 175   case X86II::MO_TLSGD:
 176   case X86II::MO_GOTTPOFF:
 177   case X86II::MO_INDNTPOFF:
 178   case X86II::MO_TPOFF:
 179   case X86II::MO_NTPOFF:
 180   case X86II::MO_GOTPCREL:
 181   case X86II::MO_GOT:
 182   case X86II::MO_GOTOFF:
 183   case X86II::MO_PLT:
 184     break;
 185   case X86II::MO_PIC_BASE_OFFSET:
 186   case X86II::MO_DARWIN_NONLAZY_PIC_BASE:
 187   case X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE:
 188     // Subtract the pic base.
 189     Expr = MCBinaryExpr::CreateSub(Expr,
 190                                    MCSymbolRefExpr::Create(GetPICBaseSymbol(),
 191                                                            OutContext),
 192                                    OutContext);
 193     break;
 194   case X86II::MO_GOT_ABSOLUTE_ADDRESS: {
 195     // For this, we want to print something like:
 196     //   MYSYMBOL + (. - PICBASE)
 197     // However, we can't generate a ".", so just emit a new label here and refer
 198     // to it.  We know that this operand flag occurs at most once per function.
 199     SmallString<64> Name;
 200     raw_svector_ostream(Name) << MAI->getPrivateGlobalPrefix() << "picbaseref"
 201       << getFunctionNumber();
 202     MCSymbol *DotSym = OutContext.GetOrCreateSymbol(Name.str());
 203     OutStreamer.EmitLabel(DotSym);
 204
 205     const MCExpr *DotExpr = MCSymbolRefExpr::Create(DotSym, OutContext);
 206     const MCExpr *PICBase = MCSymbolRefExpr::Create(GetPICBaseSymbol(),
 207                                                     OutContext);
 208     DotExpr = MCBinaryExpr::CreateSub(DotExpr, PICBase, OutContext);
 209     Expr = MCBinaryExpr::CreateAdd(Expr, DotExpr, OutContext);
 210     break;
 211   }
 212   }
 213
 214   if (!MO.isJTI() && MO.getOffset())
 215     Expr = MCBinaryExpr::CreateAdd(Expr, MCConstantExpr::Create(MO.getOffset(),
 216                                                                 OutContext),
 217                                    OutContext);
 218   return MCOperand::CreateExpr(Expr);
 219 }
 220
 221 void X86ATTAsmPrinter::
 222 printInstructionThroughMCStreamer(const MachineInstr *MI) {
 223
 224   MCInst TmpInst;
 225
 226   switch (MI->getOpcode()) {
 227   case TargetInstrInfo::DBG_LABEL:
 228   case TargetInstrInfo::EH_LABEL:
 229   case TargetInstrInfo::GC_LABEL:
 230     printLabel(MI);
 231     return;
 232   case TargetInstrInfo::INLINEASM:
 233     O << '\t';
 234     printInlineAsm(MI);
 235     return;
 236   case TargetInstrInfo::IMPLICIT_DEF:
 237     printImplicitDef(MI);
 238     return;
 239   case X86::MOVPC32r: {
 240     // This is a pseudo op for a two instruction sequence with a label, which
 241     // looks like:
 242     //     call "L1$pb"
 243     // "L1$pb":
 244     //     popl %esi
 245
 246     // Emit the call.
 247     MCSymbol *PICBase = GetPICBaseSymbol();
 248     TmpInst.setOpcode(X86::CALLpcrel32);
 249     // FIXME: We would like an efficient form for this, so we don't have to do a
 250     // lot of extra uniquing.
 251     TmpInst.addOperand(MCOperand::CreateExpr(MCSymbolRefExpr::Create(PICBase,
 252                                                                  OutContext)));
 253     printInstruction(&TmpInst);
 254
 255     // Emit the label.
 256     OutStreamer.EmitLabel(PICBase);
 257
 258     // popl $reg
 259     TmpInst.setOpcode(X86::POP32r);
 260     TmpInst.getOperand(0) = MCOperand::CreateReg(MI->getOperand(0).getReg());
 261     printInstruction(&TmpInst);
 262     return;
 263     }
 264   }
 265
 266   TmpInst.setOpcode(MI->getOpcode());
 267
 268   for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
 269     const MachineOperand &MO = MI->getOperand(i);
 270
 271     MCOperand MCOp;
 272     switch (MO.getType()) {
 273     default:
 274       O.flush();
 275       errs() << "Cannot lower operand #" << i << " of :" << *MI;
 276       llvm_unreachable("Unimp");
 277     case MachineOperand::MO_Register:
 278       MCOp = MCOperand::CreateReg(MO.getReg());
 279       break;
 280     case MachineOperand::MO_Immediate:
 281       MCOp = MCOperand::CreateImm(MO.getImm());
 282       break;
 283     case MachineOperand::MO_MachineBasicBlock:
 284       MCOp = MCOperand::CreateMBBLabel(getFunctionNumber(),
 285                                        MO.getMBB()->getNumber());
 286       break;
 287     case MachineOperand::MO_GlobalAddress:
 288       MCOp = LowerSymbolOperand(MO, GetGlobalAddressSymbol(MO));
 289       break;
 290     case MachineOperand::MO_ExternalSymbol:
 291       MCOp = LowerSymbolOperand(MO, GetExternalSymbolSymbol(MO));
 292       break;
 293     case MachineOperand::MO_JumpTableIndex:
 294       MCOp = LowerSymbolOperand(MO, GetJumpTableSymbol(MO));
 295       break;
 296     case MachineOperand::MO_ConstantPoolIndex:
 297       MCOp = LowerSymbolOperand(MO, GetConstantPoolIndexSymbol(MO));
 298       break;
 299     }
 300
 301     TmpInst.addOperand(MCOp);
 302   }
 303
 304   switch (TmpInst.getOpcode()) {
 305   case X86::LEA64_32r:
 306     // Handle the 'subreg rewriting' for the lea64_32mem operand.
 307     lower_lea64_32mem(&TmpInst, 1);
 308     break;
 309   }
 310
 311   printInstruction(&TmpInst);
 312 }