lib/Target/X86/X86AsmPrinter.cpp

   1 //===-- X86AsmPrinter.cpp - Convert X86 LLVM code to AT&T assembly --------===//
   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 a printer that converts from our internal representation
  11 // of machine-dependent LLVM code to X86 machine code.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "X86AsmPrinter.h"
  16 #include "InstPrinter/X86ATTInstPrinter.h"
  17 #include "MCTargetDesc/X86BaseInfo.h"
  18 #include "X86COFFMachineModuleInfo.h"
  19 #include "X86InstrInfo.h"
  20 #include "X86MachineFunctionInfo.h"
  21 #include "llvm/ADT/SmallString.h"
  22 #include "llvm/CodeGen/MachineModuleInfoImpls.h"
  23 #include "llvm/CodeGen/MachineValueType.h"
  24 #include "llvm/CodeGen/TargetLoweringObjectFileImpl.h"
  25 #include "llvm/IR/DebugInfo.h"
  26 #include "llvm/IR/DerivedTypes.h"
  27 #include "llvm/IR/Mangler.h"
  28 #include "llvm/IR/Module.h"
  29 #include "llvm/IR/Type.h"
  30 #include "llvm/MC/MCAsmInfo.h"
  31 #include "llvm/MC/MCContext.h"
  32 #include "llvm/MC/MCExpr.h"
  33 #include "llvm/MC/MCSectionMachO.h"
  34 #include "llvm/MC/MCStreamer.h"
  35 #include "llvm/MC/MCSymbol.h"
  36 #include "llvm/Support/COFF.h"
  37 #include "llvm/Support/Debug.h"
  38 #include "llvm/Support/ErrorHandling.h"
  39 #include "llvm/Support/TargetRegistry.h"
  40 using namespace llvm;
  41
  42 //===----------------------------------------------------------------------===//
  43 // Primitive Helper Functions.
  44 //===----------------------------------------------------------------------===//
  45
  46 /// runOnMachineFunction - Emit the function body.
  47 ///
  48 bool X86AsmPrinter::runOnMachineFunction(MachineFunction &MF) {
  49   SetupMachineFunction(MF);
  50
  51   if (Subtarget->isTargetCOFF()) {
  52     bool Intrn = MF.getFunction()->hasInternalLinkage();
  53     OutStreamer.BeginCOFFSymbolDef(CurrentFnSym);
  54     OutStreamer.EmitCOFFSymbolStorageClass(Intrn ? COFF::IMAGE_SYM_CLASS_STATIC
  55                                               : COFF::IMAGE_SYM_CLASS_EXTERNAL);
  56     OutStreamer.EmitCOFFSymbolType(COFF::IMAGE_SYM_DTYPE_FUNCTION
  57                                                << COFF::SCT_COMPLEX_TYPE_SHIFT);
  58     OutStreamer.EndCOFFSymbolDef();
  59   }
  60
  61   // Have common code print out the function header with linkage info etc.
  62   EmitFunctionHeader();
  63
  64   // Emit the rest of the function body.
  65   EmitFunctionBody();
  66
  67   // We didn't modify anything.
  68   return false;
  69 }
  70
  71 /// printSymbolOperand - Print a raw symbol reference operand.  This handles
  72 /// jump tables, constant pools, global address and external symbols, all of
  73 /// which print to a label with various suffixes for relocation types etc.
  74 static void printSymbolOperand(X86AsmPrinter &P, const MachineOperand &MO,
  75                                raw_ostream &O) {
  76   switch (MO.getType()) {
  77   default: llvm_unreachable("unknown symbol type!");
  78   case MachineOperand::MO_ConstantPoolIndex:
  79     O << *P.GetCPISymbol(MO.getIndex());
  80     P.printOffset(MO.getOffset(), O);
  81     break;
  82   case MachineOperand::MO_GlobalAddress: {
  83     const GlobalValue *GV = MO.getGlobal();
  84
  85     MCSymbol *GVSym;
  86     if (MO.getTargetFlags() == X86II::MO_DARWIN_STUB)
  87       GVSym = P.getSymbolWithGlobalValueBase(GV, "$stub");
  88     else if (MO.getTargetFlags() == X86II::MO_DARWIN_NONLAZY ||
  89              MO.getTargetFlags() == X86II::MO_DARWIN_NONLAZY_PIC_BASE ||
  90              MO.getTargetFlags() == X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE)
  91       GVSym = P.getSymbolWithGlobalValueBase(GV, "$non_lazy_ptr");
  92     else
  93       GVSym = P.getSymbol(GV);
  94
  95     // Handle dllimport linkage.
  96     if (MO.getTargetFlags() == X86II::MO_DLLIMPORT)
  97       GVSym =
  98           P.OutContext.GetOrCreateSymbol(Twine("__imp_") + GVSym->getName());
  99
 100     if (MO.getTargetFlags() == X86II::MO_DARWIN_NONLAZY ||
 101         MO.getTargetFlags() == X86II::MO_DARWIN_NONLAZY_PIC_BASE) {
 102       MCSymbol *Sym = P.getSymbolWithGlobalValueBase(GV, "$non_lazy_ptr");
 103       MachineModuleInfoImpl::StubValueTy &StubSym =
 104           P.MMI->getObjFileInfo<MachineModuleInfoMachO>().getGVStubEntry(Sym);
 105       if (!StubSym.getPointer())
 106         StubSym = MachineModuleInfoImpl::
 107           StubValueTy(P.getSymbol(GV), !GV->hasInternalLinkage());
 108     } else if (MO.getTargetFlags() == X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE){
 109       MCSymbol *Sym = P.getSymbolWithGlobalValueBase(GV, "$non_lazy_ptr");
 110       MachineModuleInfoImpl::StubValueTy &StubSym =
 111           P.MMI->getObjFileInfo<MachineModuleInfoMachO>().getHiddenGVStubEntry(
 112               Sym);
 113       if (!StubSym.getPointer())
 114         StubSym = MachineModuleInfoImpl::
 115           StubValueTy(P.getSymbol(GV), !GV->hasInternalLinkage());
 116     } else if (MO.getTargetFlags() == X86II::MO_DARWIN_STUB) {
 117       MCSymbol *Sym = P.getSymbolWithGlobalValueBase(GV, "$stub");
 118       MachineModuleInfoImpl::StubValueTy &StubSym =
 119           P.MMI->getObjFileInfo<MachineModuleInfoMachO>().getFnStubEntry(Sym);
 120       if (!StubSym.getPointer())
 121         StubSym = MachineModuleInfoImpl::
 122           StubValueTy(P.getSymbol(GV), !GV->hasInternalLinkage());
 123     }
 124
 125     // If the name begins with a dollar-sign, enclose it in parens.  We do this
 126     // to avoid having it look like an integer immediate to the assembler.
 127     if (GVSym->getName()[0] != '$')
 128       O << *GVSym;
 129     else
 130       O << '(' << *GVSym << ')';
 131     P.printOffset(MO.getOffset(), O);
 132     break;
 133   }
 134   }
 135
 136   switch (MO.getTargetFlags()) {
 137   default:
 138     llvm_unreachable("Unknown target flag on GV operand");
 139   case X86II::MO_NO_FLAG:    // No flag.
 140     break;
 141   case X86II::MO_DARWIN_NONLAZY:
 142   case X86II::MO_DLLIMPORT:
 143   case X86II::MO_DARWIN_STUB:
 144     // These affect the name of the symbol, not any suffix.
 145     break;
 146   case X86II::MO_GOT_ABSOLUTE_ADDRESS:
 147     O << " + [.-" << *P.MF->getPICBaseSymbol() << ']';
 148     break;
 149   case X86II::MO_PIC_BASE_OFFSET:
 150   case X86II::MO_DARWIN_NONLAZY_PIC_BASE:
 151   case X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE:
 152     O << '-' << *P.MF->getPICBaseSymbol();
 153     break;
 154   case X86II::MO_TLSGD:     O << "@TLSGD";     break;
 155   case X86II::MO_TLSLD:     O << "@TLSLD";     break;
 156   case X86II::MO_TLSLDM:    O << "@TLSLDM";    break;
 157   case X86II::MO_GOTTPOFF:  O << "@GOTTPOFF";  break;
 158   case X86II::MO_INDNTPOFF: O << "@INDNTPOFF"; break;
 159   case X86II::MO_TPOFF:     O << "@TPOFF";     break;
 160   case X86II::MO_DTPOFF:    O << "@DTPOFF";    break;
 161   case X86II::MO_NTPOFF:    O << "@NTPOFF";    break;
 162   case X86II::MO_GOTNTPOFF: O << "@GOTNTPOFF"; break;
 163   case X86II::MO_GOTPCREL:  O << "@GOTPCREL";  break;
 164   case X86II::MO_GOT:       O << "@GOT";       break;
 165   case X86II::MO_GOTOFF:    O << "@GOTOFF";    break;
 166   case X86II::MO_PLT:       O << "@PLT";       break;
 167   case X86II::MO_TLVP:      O << "@TLVP";      break;
 168   case X86II::MO_TLVP_PIC_BASE:
 169     O << "@TLVP" << '-' << *P.MF->getPICBaseSymbol();
 170     break;
 171   case X86II::MO_SECREL:    O << "@SECREL32";  break;
 172   }
 173 }
 174
 175 static void printOperand(X86AsmPrinter &P, const MachineInstr *MI,
 176                          unsigned OpNo, raw_ostream &O,
 177                          const char *Modifier = nullptr, unsigned AsmVariant = 0);
 178
 179 /// printPCRelImm - This is used to print an immediate value that ends up
 180 /// being encoded as a pc-relative value.  These print slightly differently, for
 181 /// example, a $ is not emitted.
 182 static void printPCRelImm(X86AsmPrinter &P, const MachineInstr *MI,
 183                           unsigned OpNo, raw_ostream &O) {
 184   const MachineOperand &MO = MI->getOperand(OpNo);
 185   switch (MO.getType()) {
 186   default: llvm_unreachable("Unknown pcrel immediate operand");
 187   case MachineOperand::MO_Register:
 188     // pc-relativeness was handled when computing the value in the reg.
 189     printOperand(P, MI, OpNo, O);
 190     return;
 191   case MachineOperand::MO_Immediate:
 192     O << MO.getImm();
 193     return;
 194   case MachineOperand::MO_GlobalAddress:
 195     printSymbolOperand(P, MO, O);
 196     return;
 197   }
 198 }
 199
 200 static void printOperand(X86AsmPrinter &P, const MachineInstr *MI,
 201                          unsigned OpNo, raw_ostream &O, const char *Modifier,
 202                          unsigned AsmVariant) {
 203   const MachineOperand &MO = MI->getOperand(OpNo);
 204   switch (MO.getType()) {
 205   default: llvm_unreachable("unknown operand type!");
 206   case MachineOperand::MO_Register: {
 207     // FIXME: Enumerating AsmVariant, so we can remove magic number.
 208     if (AsmVariant == 0) O << '%';
 209     unsigned Reg = MO.getReg();
 210     if (Modifier && strncmp(Modifier, "subreg", strlen("subreg")) == 0) {
 211       MVT::SimpleValueType VT = (strcmp(Modifier+6,"64") == 0) ?
 212         MVT::i64 : ((strcmp(Modifier+6, "32") == 0) ? MVT::i32 :
 213                     ((strcmp(Modifier+6,"16") == 0) ? MVT::i16 : MVT::i8));
 214       Reg = getX86SubSuperRegister(Reg, VT);
 215     }
 216     O << X86ATTInstPrinter::getRegisterName(Reg);
 217     return;
 218   }
 219
 220   case MachineOperand::MO_Immediate:
 221     if (AsmVariant == 0) O << '$';
 222     O << MO.getImm();
 223     return;
 224
 225   case MachineOperand::MO_GlobalAddress: {
 226     if (AsmVariant == 0) O << '$';
 227     printSymbolOperand(P, MO, O);
 228     break;
 229   }
 230   }
 231 }
 232
 233 static void printLeaMemReference(X86AsmPrinter &P, const MachineInstr *MI,
 234                                  unsigned Op, raw_ostream &O,
 235                                  const char *Modifier = nullptr) {
 236   const MachineOperand &BaseReg  = MI->getOperand(Op+X86::AddrBaseReg);
 237   const MachineOperand &IndexReg = MI->getOperand(Op+X86::AddrIndexReg);
 238   const MachineOperand &DispSpec = MI->getOperand(Op+X86::AddrDisp);
 239
 240   // If we really don't want to print out (rip), don't.
 241   bool HasBaseReg = BaseReg.getReg() != 0;
 242   if (HasBaseReg && Modifier && !strcmp(Modifier, "no-rip") &&
 243       BaseReg.getReg() == X86::RIP)
 244     HasBaseReg = false;
 245
 246   // HasParenPart - True if we will print out the () part of the mem ref.
 247   bool HasParenPart = IndexReg.getReg() || HasBaseReg;
 248
 249   switch (DispSpec.getType()) {
 250   default:
 251     llvm_unreachable("unknown operand type!");
 252   case MachineOperand::MO_Immediate: {
 253     int DispVal = DispSpec.getImm();
 254     if (DispVal || !HasParenPart)
 255       O << DispVal;
 256     break;
 257   }
 258   case MachineOperand::MO_GlobalAddress:
 259   case MachineOperand::MO_ConstantPoolIndex:
 260     printSymbolOperand(P, DispSpec, O);
 261   }
 262
 263   if (Modifier && strcmp(Modifier, "H") == 0)
 264     O << "+8";
 265
 266   if (HasParenPart) {
 267     assert(IndexReg.getReg() != X86::ESP &&
 268            "X86 doesn't allow scaling by ESP");
 269
 270     O << '(';
 271     if (HasBaseReg)
 272       printOperand(P, MI, Op+X86::AddrBaseReg, O, Modifier);
 273
 274     if (IndexReg.getReg()) {
 275       O << ',';
 276       printOperand(P, MI, Op+X86::AddrIndexReg, O, Modifier);
 277       unsigned ScaleVal = MI->getOperand(Op+X86::AddrScaleAmt).getImm();
 278       if (ScaleVal != 1)
 279         O << ',' << ScaleVal;
 280     }
 281     O << ')';
 282   }
 283 }
 284
 285 static void printMemReference(X86AsmPrinter &P, const MachineInstr *MI,
 286                               unsigned Op, raw_ostream &O,
 287                               const char *Modifier = nullptr) {
 288   assert(isMem(MI, Op) && "Invalid memory reference!");
 289   const MachineOperand &Segment = MI->getOperand(Op+X86::AddrSegmentReg);
 290   if (Segment.getReg()) {
 291     printOperand(P, MI, Op+X86::AddrSegmentReg, O, Modifier);
 292     O << ':';
 293   }
 294   printLeaMemReference(P, MI, Op, O, Modifier);
 295 }
 296
 297 static void printIntelMemReference(X86AsmPrinter &P, const MachineInstr *MI,
 298                                    unsigned Op, raw_ostream &O,
 299                                    const char *Modifier = nullptr,
 300                                    unsigned AsmVariant = 1) {
 301   const MachineOperand &BaseReg  = MI->getOperand(Op+X86::AddrBaseReg);
 302   unsigned ScaleVal = MI->getOperand(Op+X86::AddrScaleAmt).getImm();
 303   const MachineOperand &IndexReg = MI->getOperand(Op+X86::AddrIndexReg);
 304   const MachineOperand &DispSpec = MI->getOperand(Op+X86::AddrDisp);
 305   const MachineOperand &SegReg   = MI->getOperand(Op+X86::AddrSegmentReg);
 306
 307   // If this has a segment register, print it.
 308   if (SegReg.getReg()) {
 309     printOperand(P, MI, Op+X86::AddrSegmentReg, O, Modifier, AsmVariant);
 310     O << ':';
 311   }
 312
 313   O << '[';
 314
 315   bool NeedPlus = false;
 316   if (BaseReg.getReg()) {
 317     printOperand(P, MI, Op+X86::AddrBaseReg, O, Modifier, AsmVariant);
 318     NeedPlus = true;
 319   }
 320
 321   if (IndexReg.getReg()) {
 322     if (NeedPlus) O << " + ";
 323     if (ScaleVal != 1)
 324       O << ScaleVal << '*';
 325     printOperand(P, MI, Op+X86::AddrIndexReg, O, Modifier, AsmVariant);
 326     NeedPlus = true;
 327   }
 328
 329   if (!DispSpec.isImm()) {
 330     if (NeedPlus) O << " + ";
 331     printOperand(P, MI, Op+X86::AddrDisp, O, Modifier, AsmVariant);
 332   } else {
 333     int64_t DispVal = DispSpec.getImm();
 334     if (DispVal || (!IndexReg.getReg() && !BaseReg.getReg())) {
 335       if (NeedPlus) {
 336         if (DispVal > 0)
 337           O << " + ";
 338         else {
 339           O << " - ";
 340           DispVal = -DispVal;
 341         }
 342       }
 343       O << DispVal;
 344     }
 345   }
 346   O << ']';
 347 }
 348
 349 static bool printAsmMRegister(X86AsmPrinter &P, const MachineOperand &MO,
 350                               char Mode, raw_ostream &O) {
 351   unsigned Reg = MO.getReg();
 352   switch (Mode) {
 353   default: return true;  // Unknown mode.
 354   case 'b': // Print QImode register
 355     Reg = getX86SubSuperRegister(Reg, MVT::i8);
 356     break;
 357   case 'h': // Print QImode high register
 358     Reg = getX86SubSuperRegister(Reg, MVT::i8, true);
 359     break;
 360   case 'w': // Print HImode register
 361     Reg = getX86SubSuperRegister(Reg, MVT::i16);
 362     break;
 363   case 'k': // Print SImode register
 364     Reg = getX86SubSuperRegister(Reg, MVT::i32);
 365     break;
 366   case 'q':
 367     // Print 64-bit register names if 64-bit integer registers are available.
 368     // Otherwise, print 32-bit register names.
 369     MVT::SimpleValueType Ty = P.getSubtarget().is64Bit() ? MVT::i64 : MVT::i32;
 370     Reg = getX86SubSuperRegister(Reg, Ty);
 371     break;
 372   }
 373
 374   O << '%' << X86ATTInstPrinter::getRegisterName(Reg);
 375   return false;
 376 }
 377
 378 /// PrintAsmOperand - Print out an operand for an inline asm expression.
 379 ///
 380 bool X86AsmPrinter::PrintAsmOperand(const MachineInstr *MI, unsigned OpNo,
 381                                     unsigned AsmVariant,
 382                                     const char *ExtraCode, raw_ostream &O) {
 383   // Does this asm operand have a single letter operand modifier?
 384   if (ExtraCode && ExtraCode[0]) {
 385     if (ExtraCode[1] != 0) return true; // Unknown modifier.
 386
 387     const MachineOperand &MO = MI->getOperand(OpNo);
 388
 389     switch (ExtraCode[0]) {
 390     default:
 391       // See if this is a generic print operand
 392       return AsmPrinter::PrintAsmOperand(MI, OpNo, AsmVariant, ExtraCode, O);
 393     case 'a': // This is an address.  Currently only 'i' and 'r' are expected.
 394       switch (MO.getType()) {
 395       default:
 396         return true;
 397       case MachineOperand::MO_Immediate:
 398         O << MO.getImm();
 399         return false;
 400       case MachineOperand::MO_ConstantPoolIndex:
 401       case MachineOperand::MO_JumpTableIndex:
 402       case MachineOperand::MO_ExternalSymbol:
 403         llvm_unreachable("unexpected operand type!");
 404       case MachineOperand::MO_GlobalAddress:
 405         printSymbolOperand(*this, MO, O);
 406         if (Subtarget->isPICStyleRIPRel())
 407           O << "(%rip)";
 408         return false;
 409       case MachineOperand::MO_Register:
 410         O << '(';
 411         printOperand(*this, MI, OpNo, O);
 412         O << ')';
 413         return false;
 414       }
 415
 416     case 'c': // Don't print "$" before a global var name or constant.
 417       switch (MO.getType()) {
 418       default:
 419         printOperand(*this, MI, OpNo, O);
 420         break;
 421       case MachineOperand::MO_Immediate:
 422         O << MO.getImm();
 423         break;
 424       case MachineOperand::MO_ConstantPoolIndex:
 425       case MachineOperand::MO_JumpTableIndex:
 426       case MachineOperand::MO_ExternalSymbol:
 427         llvm_unreachable("unexpected operand type!");
 428       case MachineOperand::MO_GlobalAddress:
 429         printSymbolOperand(*this, MO, O);
 430         break;
 431       }
 432       return false;
 433
 434     case 'A': // Print '*' before a register (it must be a register)
 435       if (MO.isReg()) {
 436         O << '*';
 437         printOperand(*this, MI, OpNo, O);
 438         return false;
 439       }
 440       return true;
 441
 442     case 'b': // Print QImode register
 443     case 'h': // Print QImode high register
 444     case 'w': // Print HImode register
 445     case 'k': // Print SImode register
 446     case 'q': // Print DImode register
 447       if (MO.isReg())
 448         return printAsmMRegister(*this, MO, ExtraCode[0], O);
 449       printOperand(*this, MI, OpNo, O);
 450       return false;
 451
 452     case 'P': // This is the operand of a call, treat specially.
 453       printPCRelImm(*this, MI, OpNo, O);
 454       return false;
 455
 456     case 'n':  // Negate the immediate or print a '-' before the operand.
 457       // Note: this is a temporary solution. It should be handled target
 458       // independently as part of the 'MC' work.
 459       if (MO.isImm()) {
 460         O << -MO.getImm();
 461         return false;
 462       }
 463       O << '-';
 464     }
 465   }
 466
 467   printOperand(*this, MI, OpNo, O, /*Modifier*/ nullptr, AsmVariant);
 468   return false;
 469 }
 470
 471 bool X86AsmPrinter::PrintAsmMemoryOperand(const MachineInstr *MI,
 472                                           unsigned OpNo, unsigned AsmVariant,
 473                                           const char *ExtraCode,
 474                                           raw_ostream &O) {
 475   if (AsmVariant) {
 476     printIntelMemReference(*this, MI, OpNo, O);
 477     return false;
 478   }
 479
 480   if (ExtraCode && ExtraCode[0]) {
 481     if (ExtraCode[1] != 0) return true; // Unknown modifier.
 482
 483     switch (ExtraCode[0]) {
 484     default: return true;  // Unknown modifier.
 485     case 'b': // Print QImode register
 486     case 'h': // Print QImode high register
 487     case 'w': // Print HImode register
 488     case 'k': // Print SImode register
 489     case 'q': // Print SImode register
 490       // These only apply to registers, ignore on mem.
 491       break;
 492     case 'H':
 493       printMemReference(*this, MI, OpNo, O, "H");
 494       return false;
 495     case 'P': // Don't print @PLT, but do print as memory.
 496       printMemReference(*this, MI, OpNo, O, "no-rip");
 497       return false;
 498     }
 499   }
 500   printMemReference(*this, MI, OpNo, O);
 501   return false;
 502 }
 503
 504 void X86AsmPrinter::EmitStartOfAsmFile(Module &M) {
 505   if (Subtarget->isTargetMacho())
 506     OutStreamer.SwitchSection(getObjFileLowering().getTextSection());
 507
 508   if (Subtarget->isTargetCOFF()) {
 509     // Emit an absolute @feat.00 symbol.  This appears to be some kind of
 510     // compiler features bitfield read by link.exe.
 511     if (!Subtarget->is64Bit()) {
 512       MCSymbol *S = MMI->getContext().GetOrCreateSymbol(StringRef("@feat.00"));
 513       OutStreamer.BeginCOFFSymbolDef(S);
 514       OutStreamer.EmitCOFFSymbolStorageClass(COFF::IMAGE_SYM_CLASS_STATIC);
 515       OutStreamer.EmitCOFFSymbolType(COFF::IMAGE_SYM_DTYPE_NULL);
 516       OutStreamer.EndCOFFSymbolDef();
 517       // According to the PE-COFF spec, the LSB of this value marks the object
 518       // for "registered SEH".  This means that all SEH handler entry points
 519       // must be registered in .sxdata.  Use of any unregistered handlers will
 520       // cause the process to terminate immediately.  LLVM does not know how to
 521       // register any SEH handlers, so its object files should be safe.
 522       S->setAbsolute();
 523       OutStreamer.EmitSymbolAttribute(S, MCSA_Global);
 524       OutStreamer.EmitAssignment(
 525           S, MCConstantExpr::Create(int64_t(1), MMI->getContext()));
 526     }
 527   }
 528 }
 529
 530
 531 void X86AsmPrinter::EmitEndOfAsmFile(Module &M) {
 532   if (Subtarget->isTargetMacho()) {
 533     // All darwin targets use mach-o.
 534     MachineModuleInfoMachO &MMIMacho =
 535       MMI->getObjFileInfo<MachineModuleInfoMachO>();
 536
 537     // Output stubs for dynamically-linked functions.
 538     MachineModuleInfoMachO::SymbolListTy Stubs;
 539
 540     Stubs = MMIMacho.GetFnStubList();
 541     if (!Stubs.empty()) {
 542       const MCSection *TheSection =
 543         OutContext.getMachOSection("__IMPORT", "__jump_table",
 544                                    MachO::S_SYMBOL_STUBS |
 545                                    MachO::S_ATTR_SELF_MODIFYING_CODE |
 546                                    MachO::S_ATTR_PURE_INSTRUCTIONS,
 547                                    5, SectionKind::getMetadata());
 548       OutStreamer.SwitchSection(TheSection);
 549
 550       for (unsigned i = 0, e = Stubs.size(); i != e; ++i) {
 551         // L_foo$stub:
 552         OutStreamer.EmitLabel(Stubs[i].first);
 553         //   .indirect_symbol _foo
 554         OutStreamer.EmitSymbolAttribute(Stubs[i].second.getPointer(),
 555                                         MCSA_IndirectSymbol);
 556         // hlt; hlt; hlt; hlt; hlt     hlt = 0xf4.
 557         const char HltInsts[] = "\xf4\xf4\xf4\xf4\xf4";
 558         OutStreamer.EmitBytes(StringRef(HltInsts, 5));
 559       }
 560
 561       Stubs.clear();
 562       OutStreamer.AddBlankLine();
 563     }
 564
 565     // Output stubs for external and common global variables.
 566     Stubs = MMIMacho.GetGVStubList();
 567     if (!Stubs.empty()) {
 568       const MCSection *TheSection =
 569         OutContext.getMachOSection("__IMPORT", "__pointers",
 570                                    MachO::S_NON_LAZY_SYMBOL_POINTERS,
 571                                    SectionKind::getMetadata());
 572       OutStreamer.SwitchSection(TheSection);
 573
 574       for (unsigned i = 0, e = Stubs.size(); i != e; ++i) {
 575         // L_foo$non_lazy_ptr:
 576         OutStreamer.EmitLabel(Stubs[i].first);
 577         // .indirect_symbol _foo
 578         MachineModuleInfoImpl::StubValueTy &MCSym = Stubs[i].second;
 579         OutStreamer.EmitSymbolAttribute(MCSym.getPointer(),
 580                                         MCSA_IndirectSymbol);
 581         // .long 0
 582         if (MCSym.getInt())
 583           // External to current translation unit.
 584           OutStreamer.EmitIntValue(0, 4/*size*/);
 585         else
 586           // Internal to current translation unit.
 587           //
 588           // When we place the LSDA into the TEXT section, the type info
 589           // pointers need to be indirect and pc-rel. We accomplish this by
 590           // using NLPs.  However, sometimes the types are local to the file. So
 591           // we need to fill in the value for the NLP in those cases.
 592           OutStreamer.EmitValue(MCSymbolRefExpr::Create(MCSym.getPointer(),
 593                                                         OutContext), 4/*size*/);
 594       }
 595       Stubs.clear();
 596       OutStreamer.AddBlankLine();
 597     }
 598
 599     Stubs = MMIMacho.GetHiddenGVStubList();
 600     if (!Stubs.empty()) {
 601       OutStreamer.SwitchSection(getObjFileLowering().getDataSection());
 602       EmitAlignment(2);
 603
 604       for (unsigned i = 0, e = Stubs.size(); i != e; ++i) {
 605         // L_foo$non_lazy_ptr:
 606         OutStreamer.EmitLabel(Stubs[i].first);
 607         // .long _foo
 608         OutStreamer.EmitValue(MCSymbolRefExpr::
 609                               Create(Stubs[i].second.getPointer(),
 610                                      OutContext), 4/*size*/);
 611       }
 612       Stubs.clear();
 613       OutStreamer.AddBlankLine();
 614     }
 615
 616     SM.serializeToStackMapSection();
 617
 618     // Funny Darwin hack: This flag tells the linker that no global symbols
 619     // contain code that falls through to other global symbols (e.g. the obvious
 620     // implementation of multiple entry points).  If this doesn't occur, the
 621     // linker can safely perform dead code stripping.  Since LLVM never
 622     // generates code that does this, it is always safe to set.
 623     OutStreamer.EmitAssemblerFlag(MCAF_SubsectionsViaSymbols);
 624   }
 625
 626   if (Subtarget->isTargetKnownWindowsMSVC() && MMI->usesVAFloatArgument()) {
 627     StringRef SymbolName = Subtarget->is64Bit() ? "_fltused" : "__fltused";
 628     MCSymbol *S = MMI->getContext().GetOrCreateSymbol(SymbolName);
 629     OutStreamer.EmitSymbolAttribute(S, MCSA_Global);
 630   }
 631
 632   if (Subtarget->isTargetCOFF()) {
 633     X86COFFMachineModuleInfo &COFFMMI =
 634       MMI->getObjFileInfo<X86COFFMachineModuleInfo>();
 635
 636     // Emit type information for external functions
 637     typedef X86COFFMachineModuleInfo::externals_iterator externals_iterator;
 638     for (externals_iterator I = COFFMMI.externals_begin(),
 639                             E = COFFMMI.externals_end();
 640                             I != E; ++I) {
 641       OutStreamer.BeginCOFFSymbolDef(CurrentFnSym);
 642       OutStreamer.EmitCOFFSymbolStorageClass(COFF::IMAGE_SYM_CLASS_EXTERNAL);
 643       OutStreamer.EmitCOFFSymbolType(COFF::IMAGE_SYM_DTYPE_FUNCTION
 644                                                << COFF::SCT_COMPLEX_TYPE_SHIFT);
 645       OutStreamer.EndCOFFSymbolDef();
 646     }
 647
 648     // Necessary for dllexport support
 649     std::vector<const MCSymbol*> DLLExportedFns, DLLExportedGlobals;
 650
 651     for (Module::const_iterator I = M.begin(), E = M.end(); I != E; ++I)
 652       if (I->hasDLLExportStorageClass())
 653         DLLExportedFns.push_back(getSymbol(I));
 654
 655     for (Module::const_global_iterator I = M.global_begin(),
 656            E = M.global_end(); I != E; ++I)
 657       if (I->hasDLLExportStorageClass())
 658         DLLExportedGlobals.push_back(getSymbol(I));
 659
 660     for (Module::const_alias_iterator I = M.alias_begin(), E = M.alias_end();
 661                                       I != E; ++I) {
 662       const GlobalValue *GV = I;
 663       if (!GV->hasDLLExportStorageClass())
 664         continue;
 665
 666       while (const GlobalAlias *A = dyn_cast<GlobalAlias>(GV))
 667         GV = A->getAliasedGlobal();
 668
 669       if (isa<Function>(GV))
 670         DLLExportedFns.push_back(getSymbol(I));
 671       else if (isa<GlobalVariable>(GV))
 672         DLLExportedGlobals.push_back(getSymbol(I));
 673     }
 674
 675     // Output linker support code for dllexported globals on windows.
 676     if (!DLLExportedGlobals.empty() || !DLLExportedFns.empty()) {
 677       const TargetLoweringObjectFileCOFF &TLOFCOFF =
 678         static_cast<const TargetLoweringObjectFileCOFF&>(getObjFileLowering());
 679
 680       OutStreamer.SwitchSection(TLOFCOFF.getDrectveSection());
 681       SmallString<128> name;
 682       for (unsigned i = 0, e = DLLExportedGlobals.size(); i != e; ++i) {
 683         if (Subtarget->isTargetKnownWindowsMSVC())
 684           name = " /EXPORT:";
 685         else
 686           name = " -export:";
 687         name += DLLExportedGlobals[i]->getName();
 688         if (Subtarget->isTargetKnownWindowsMSVC())
 689           name += ",DATA";
 690         else
 691         name += ",data";
 692         OutStreamer.EmitBytes(name);
 693       }
 694
 695       for (unsigned i = 0, e = DLLExportedFns.size(); i != e; ++i) {
 696         if (Subtarget->isTargetKnownWindowsMSVC())
 697           name = " /EXPORT:";
 698         else
 699           name = " -export:";
 700         name += DLLExportedFns[i]->getName();
 701         OutStreamer.EmitBytes(name);
 702       }
 703     }
 704   }
 705
 706   if (Subtarget->isTargetELF()) {
 707     const TargetLoweringObjectFileELF &TLOFELF =
 708       static_cast<const TargetLoweringObjectFileELF &>(getObjFileLowering());
 709
 710     MachineModuleInfoELF &MMIELF = MMI->getObjFileInfo<MachineModuleInfoELF>();
 711
 712     // Output stubs for external and common global variables.
 713     MachineModuleInfoELF::SymbolListTy Stubs = MMIELF.GetGVStubList();
 714     if (!Stubs.empty()) {
 715       OutStreamer.SwitchSection(TLOFELF.getDataRelSection());
 716       const DataLayout *TD = TM.getDataLayout();
 717
 718       for (unsigned i = 0, e = Stubs.size(); i != e; ++i) {
 719         OutStreamer.EmitLabel(Stubs[i].first);
 720         OutStreamer.EmitSymbolValue(Stubs[i].second.getPointer(),
 721                                     TD->getPointerSize());
 722       }
 723       Stubs.clear();
 724     }
 725   }
 726 }
 727
 728 //===----------------------------------------------------------------------===//
 729 // Target Registry Stuff
 730 //===----------------------------------------------------------------------===//
 731
 732 // Force static initialization.
 733 extern "C" void LLVMInitializeX86AsmPrinter() {
 734   RegisterAsmPrinter<X86AsmPrinter> X(TheX86_32Target);
 735   RegisterAsmPrinter<X86AsmPrinter> Y(TheX86_64Target);
 736 }