Home | History | Annotate | Download | only in CodeGen
      1 //===-- llvm/CodeGen/TargetLoweringObjectFileImpl.cpp - Object File Info --===//
      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 implements classes used to handle lowerings specific to common
     11 // object file formats.
     12 //
     13 //===----------------------------------------------------------------------===//
     14 
     15 #include "llvm/CodeGen/TargetLoweringObjectFileImpl.h"
     16 #include "llvm/ADT/SmallString.h"
     17 #include "llvm/ADT/StringExtras.h"
     18 #include "llvm/ADT/Triple.h"
     19 #include "llvm/CodeGen/MachineModuleInfoImpls.h"
     20 #include "llvm/IR/Constants.h"
     21 #include "llvm/IR/DataLayout.h"
     22 #include "llvm/IR/DerivedTypes.h"
     23 #include "llvm/IR/Function.h"
     24 #include "llvm/IR/GlobalVariable.h"
     25 #include "llvm/IR/Mangler.h"
     26 #include "llvm/IR/Module.h"
     27 #include "llvm/MC/MCContext.h"
     28 #include "llvm/MC/MCExpr.h"
     29 #include "llvm/MC/MCSectionCOFF.h"
     30 #include "llvm/MC/MCSectionELF.h"
     31 #include "llvm/MC/MCSectionMachO.h"
     32 #include "llvm/MC/MCStreamer.h"
     33 #include "llvm/MC/MCSymbol.h"
     34 #include "llvm/Support/Dwarf.h"
     35 #include "llvm/Support/ELF.h"
     36 #include "llvm/Support/ErrorHandling.h"
     37 #include "llvm/Support/raw_ostream.h"
     38 #include "llvm/Target/TargetLowering.h"
     39 #include "llvm/Target/TargetMachine.h"
     40 using namespace llvm;
     41 using namespace dwarf;
     42 
     43 //===----------------------------------------------------------------------===//
     44 //                                  ELF
     45 //===----------------------------------------------------------------------===//
     46 
     47 MCSymbol *TargetLoweringObjectFileELF::getCFIPersonalitySymbol(
     48     const GlobalValue *GV, Mangler &Mang, const TargetMachine &TM,
     49     MachineModuleInfo *MMI) const {
     50   unsigned Encoding = getPersonalityEncoding();
     51   if ((Encoding & 0x80) == dwarf::DW_EH_PE_indirect)
     52     return getContext().GetOrCreateSymbol(StringRef("DW.ref.") +
     53                                           TM.getSymbol(GV, Mang)->getName());
     54   if ((Encoding & 0x70) == dwarf::DW_EH_PE_absptr)
     55     return TM.getSymbol(GV, Mang);
     56   report_fatal_error("We do not support this DWARF encoding yet!");
     57 }
     58 
     59 void TargetLoweringObjectFileELF::emitPersonalityValue(MCStreamer &Streamer,
     60                                                        const TargetMachine &TM,
     61                                                        const MCSymbol *Sym) const {
     62   SmallString<64> NameData("DW.ref.");
     63   NameData += Sym->getName();
     64   MCSymbol *Label = getContext().GetOrCreateSymbol(NameData);
     65   Streamer.EmitSymbolAttribute(Label, MCSA_Hidden);
     66   Streamer.EmitSymbolAttribute(Label, MCSA_Weak);
     67   StringRef Prefix = ".data.";
     68   NameData.insert(NameData.begin(), Prefix.begin(), Prefix.end());
     69   unsigned Flags = ELF::SHF_ALLOC | ELF::SHF_WRITE | ELF::SHF_GROUP;
     70   const MCSection *Sec = getContext().getELFSection(NameData,
     71                                                     ELF::SHT_PROGBITS,
     72                                                     Flags,
     73                                                     SectionKind::getDataRel(),
     74                                                     0, Label->getName());
     75   unsigned Size = TM.getDataLayout()->getPointerSize();
     76   Streamer.SwitchSection(Sec);
     77   Streamer.EmitValueToAlignment(TM.getDataLayout()->getPointerABIAlignment());
     78   Streamer.EmitSymbolAttribute(Label, MCSA_ELF_TypeObject);
     79   const MCExpr *E = MCConstantExpr::Create(Size, getContext());
     80   Streamer.EmitELFSize(Label, E);
     81   Streamer.EmitLabel(Label);
     82 
     83   Streamer.EmitSymbolValue(Sym, Size);
     84 }
     85 
     86 const MCExpr *TargetLoweringObjectFileELF::getTTypeGlobalReference(
     87     const GlobalValue *GV, unsigned Encoding, Mangler &Mang,
     88     const TargetMachine &TM, MachineModuleInfo *MMI,
     89     MCStreamer &Streamer) const {
     90 
     91   if (Encoding & dwarf::DW_EH_PE_indirect) {
     92     MachineModuleInfoELF &ELFMMI = MMI->getObjFileInfo<MachineModuleInfoELF>();
     93 
     94     MCSymbol *SSym = getSymbolWithGlobalValueBase(GV, ".DW.stub", Mang, TM);
     95 
     96     // Add information about the stub reference to ELFMMI so that the stub
     97     // gets emitted by the asmprinter.
     98     MachineModuleInfoImpl::StubValueTy &StubSym = ELFMMI.getGVStubEntry(SSym);
     99     if (!StubSym.getPointer()) {
    100       MCSymbol *Sym = TM.getSymbol(GV, Mang);
    101       StubSym = MachineModuleInfoImpl::StubValueTy(Sym, !GV->hasLocalLinkage());
    102     }
    103 
    104     return TargetLoweringObjectFile::
    105       getTTypeReference(MCSymbolRefExpr::Create(SSym, getContext()),
    106                         Encoding & ~dwarf::DW_EH_PE_indirect, Streamer);
    107   }
    108 
    109   return TargetLoweringObjectFile::
    110     getTTypeGlobalReference(GV, Encoding, Mang, TM, MMI, Streamer);
    111 }
    112 
    113 static SectionKind
    114 getELFKindForNamedSection(StringRef Name, SectionKind K) {
    115   // N.B.: The defaults used in here are no the same ones used in MC.
    116   // We follow gcc, MC follows gas. For example, given ".section .eh_frame",
    117   // both gas and MC will produce a section with no flags. Given
    118   // section(".eh_frame") gcc will produce:
    119   //
    120   //   .section   .eh_frame,"a",@progbits
    121   if (Name.empty() || Name[0] != '.') return K;
    122 
    123   // Some lame default implementation based on some magic section names.
    124   if (Name == ".bss" ||
    125       Name.startswith(".bss.") ||
    126       Name.startswith(".gnu.linkonce.b.") ||
    127       Name.startswith(".llvm.linkonce.b.") ||
    128       Name == ".sbss" ||
    129       Name.startswith(".sbss.") ||
    130       Name.startswith(".gnu.linkonce.sb.") ||
    131       Name.startswith(".llvm.linkonce.sb."))
    132     return SectionKind::getBSS();
    133 
    134   if (Name == ".tdata" ||
    135       Name.startswith(".tdata.") ||
    136       Name.startswith(".gnu.linkonce.td.") ||
    137       Name.startswith(".llvm.linkonce.td."))
    138     return SectionKind::getThreadData();
    139 
    140   if (Name == ".tbss" ||
    141       Name.startswith(".tbss.") ||
    142       Name.startswith(".gnu.linkonce.tb.") ||
    143       Name.startswith(".llvm.linkonce.tb."))
    144     return SectionKind::getThreadBSS();
    145 
    146   return K;
    147 }
    148 
    149 
    150 static unsigned getELFSectionType(StringRef Name, SectionKind K) {
    151 
    152   if (Name == ".init_array")
    153     return ELF::SHT_INIT_ARRAY;
    154 
    155   if (Name == ".fini_array")
    156     return ELF::SHT_FINI_ARRAY;
    157 
    158   if (Name == ".preinit_array")
    159     return ELF::SHT_PREINIT_ARRAY;
    160 
    161   if (K.isBSS() || K.isThreadBSS())
    162     return ELF::SHT_NOBITS;
    163 
    164   return ELF::SHT_PROGBITS;
    165 }
    166 
    167 
    168 static unsigned
    169 getELFSectionFlags(SectionKind K) {
    170   unsigned Flags = 0;
    171 
    172   if (!K.isMetadata())
    173     Flags |= ELF::SHF_ALLOC;
    174 
    175   if (K.isText())
    176     Flags |= ELF::SHF_EXECINSTR;
    177 
    178   if (K.isWriteable())
    179     Flags |= ELF::SHF_WRITE;
    180 
    181   if (K.isThreadLocal())
    182     Flags |= ELF::SHF_TLS;
    183 
    184   // K.isMergeableConst() is left out to honour PR4650
    185   if (K.isMergeableCString() || K.isMergeableConst4() ||
    186       K.isMergeableConst8() || K.isMergeableConst16())
    187     Flags |= ELF::SHF_MERGE;
    188 
    189   if (K.isMergeableCString())
    190     Flags |= ELF::SHF_STRINGS;
    191 
    192   return Flags;
    193 }
    194 
    195 static const Comdat *getELFComdat(const GlobalValue *GV) {
    196   const Comdat *C = GV->getComdat();
    197   if (!C)
    198     return nullptr;
    199 
    200   if (C->getSelectionKind() != Comdat::Any)
    201     report_fatal_error("ELF COMDATs only support SelectionKind::Any, '" +
    202                        C->getName() + "' cannot be lowered.");
    203 
    204   return C;
    205 }
    206 
    207 const MCSection *TargetLoweringObjectFileELF::getExplicitSectionGlobal(
    208     const GlobalValue *GV, SectionKind Kind, Mangler &Mang,
    209     const TargetMachine &TM) const {
    210   StringRef SectionName = GV->getSection();
    211 
    212   // Infer section flags from the section name if we can.
    213   Kind = getELFKindForNamedSection(SectionName, Kind);
    214 
    215   StringRef Group = "";
    216   unsigned Flags = getELFSectionFlags(Kind);
    217   if (const Comdat *C = getELFComdat(GV)) {
    218     Group = C->getName();
    219     Flags |= ELF::SHF_GROUP;
    220   }
    221   return getContext().getELFSection(SectionName,
    222                                     getELFSectionType(SectionName, Kind), Flags,
    223                                     Kind, /*EntrySize=*/0, Group);
    224 }
    225 
    226 /// getSectionPrefixForGlobal - Return the section prefix name used by options
    227 /// FunctionsSections and DataSections.
    228 static StringRef getSectionPrefixForGlobal(SectionKind Kind) {
    229   if (Kind.isText())                 return ".text.";
    230   if (Kind.isReadOnly())             return ".rodata.";
    231   if (Kind.isBSS())                  return ".bss.";
    232 
    233   if (Kind.isThreadData())           return ".tdata.";
    234   if (Kind.isThreadBSS())            return ".tbss.";
    235 
    236   if (Kind.isDataNoRel())            return ".data.";
    237   if (Kind.isDataRelLocal())         return ".data.rel.local.";
    238   if (Kind.isDataRel())              return ".data.rel.";
    239   if (Kind.isReadOnlyWithRelLocal()) return ".data.rel.ro.local.";
    240 
    241   assert(Kind.isReadOnlyWithRel() && "Unknown section kind");
    242   return ".data.rel.ro.";
    243 }
    244 
    245 const MCSection *TargetLoweringObjectFileELF::
    246 SelectSectionForGlobal(const GlobalValue *GV, SectionKind Kind,
    247                        Mangler &Mang, const TargetMachine &TM) const {
    248   // If we have -ffunction-section or -fdata-section then we should emit the
    249   // global value to a uniqued section specifically for it.
    250   bool EmitUniquedSection;
    251   if (Kind.isText())
    252     EmitUniquedSection = TM.getFunctionSections();
    253   else
    254     EmitUniquedSection = TM.getDataSections();
    255 
    256   // If this global is linkonce/weak and the target handles this by emitting it
    257   // into a 'uniqued' section name, create and return the section now.
    258   if ((GV->isWeakForLinker() || EmitUniquedSection || GV->hasComdat()) &&
    259       !Kind.isCommon()) {
    260     StringRef Prefix = getSectionPrefixForGlobal(Kind);
    261 
    262     SmallString<128> Name(Prefix);
    263     TM.getNameWithPrefix(Name, GV, Mang, true);
    264 
    265     StringRef Group = "";
    266     unsigned Flags = getELFSectionFlags(Kind);
    267     if (GV->isWeakForLinker() || GV->hasComdat()) {
    268       if (const Comdat *C = getELFComdat(GV))
    269         Group = C->getName();
    270       else
    271         Group = Name.substr(Prefix.size());
    272       Flags |= ELF::SHF_GROUP;
    273     }
    274 
    275     return getContext().getELFSection(Name.str(),
    276                                       getELFSectionType(Name.str(), Kind),
    277                                       Flags, Kind, 0, Group);
    278   }
    279 
    280   if (Kind.isText()) return TextSection;
    281 
    282   if (Kind.isMergeable1ByteCString() ||
    283       Kind.isMergeable2ByteCString() ||
    284       Kind.isMergeable4ByteCString()) {
    285 
    286     // We also need alignment here.
    287     // FIXME: this is getting the alignment of the character, not the
    288     // alignment of the global!
    289     unsigned Align =
    290       TM.getDataLayout()->getPreferredAlignment(cast<GlobalVariable>(GV));
    291 
    292     const char *SizeSpec = ".rodata.str1.";
    293     if (Kind.isMergeable2ByteCString())
    294       SizeSpec = ".rodata.str2.";
    295     else if (Kind.isMergeable4ByteCString())
    296       SizeSpec = ".rodata.str4.";
    297     else
    298       assert(Kind.isMergeable1ByteCString() && "unknown string width");
    299 
    300 
    301     std::string Name = SizeSpec + utostr(Align);
    302     return getContext().getELFSection(Name, ELF::SHT_PROGBITS,
    303                                       ELF::SHF_ALLOC |
    304                                       ELF::SHF_MERGE |
    305                                       ELF::SHF_STRINGS,
    306                                       Kind);
    307   }
    308 
    309   if (Kind.isMergeableConst()) {
    310     if (Kind.isMergeableConst4() && MergeableConst4Section)
    311       return MergeableConst4Section;
    312     if (Kind.isMergeableConst8() && MergeableConst8Section)
    313       return MergeableConst8Section;
    314     if (Kind.isMergeableConst16() && MergeableConst16Section)
    315       return MergeableConst16Section;
    316     return ReadOnlySection;  // .const
    317   }
    318 
    319   if (Kind.isReadOnly())             return ReadOnlySection;
    320 
    321   if (Kind.isThreadData())           return TLSDataSection;
    322   if (Kind.isThreadBSS())            return TLSBSSSection;
    323 
    324   // Note: we claim that common symbols are put in BSSSection, but they are
    325   // really emitted with the magic .comm directive, which creates a symbol table
    326   // entry but not a section.
    327   if (Kind.isBSS() || Kind.isCommon()) return BSSSection;
    328 
    329   if (Kind.isDataNoRel())            return DataSection;
    330   if (Kind.isDataRelLocal())         return DataRelLocalSection;
    331   if (Kind.isDataRel())              return DataRelSection;
    332   if (Kind.isReadOnlyWithRelLocal()) return DataRelROLocalSection;
    333 
    334   assert(Kind.isReadOnlyWithRel() && "Unknown section kind");
    335   return DataRelROSection;
    336 }
    337 
    338 /// getSectionForConstant - Given a mergeable constant with the
    339 /// specified size and relocation information, return a section that it
    340 /// should be placed in.
    341 const MCSection *TargetLoweringObjectFileELF::
    342 getSectionForConstant(SectionKind Kind) const {
    343   if (Kind.isMergeableConst4() && MergeableConst4Section)
    344     return MergeableConst4Section;
    345   if (Kind.isMergeableConst8() && MergeableConst8Section)
    346     return MergeableConst8Section;
    347   if (Kind.isMergeableConst16() && MergeableConst16Section)
    348     return MergeableConst16Section;
    349   if (Kind.isReadOnly())
    350     return ReadOnlySection;
    351 
    352   if (Kind.isReadOnlyWithRelLocal()) return DataRelROLocalSection;
    353   assert(Kind.isReadOnlyWithRel() && "Unknown section kind");
    354   return DataRelROSection;
    355 }
    356 
    357 const MCSection *TargetLoweringObjectFileELF::getStaticCtorSection(
    358     unsigned Priority, const MCSymbol *KeySym) const {
    359   // The default scheme is .ctor / .dtor, so we have to invert the priority
    360   // numbering.
    361   if (Priority == 65535)
    362     return StaticCtorSection;
    363 
    364   if (UseInitArray) {
    365     std::string Name = std::string(".init_array.") + utostr(Priority);
    366     return getContext().getELFSection(Name, ELF::SHT_INIT_ARRAY,
    367                                       ELF::SHF_ALLOC | ELF::SHF_WRITE,
    368                                       SectionKind::getDataRel());
    369   } else {
    370     std::string Name = std::string(".ctors.") + utostr(65535 - Priority);
    371     return getContext().getELFSection(Name, ELF::SHT_PROGBITS,
    372                                       ELF::SHF_ALLOC |ELF::SHF_WRITE,
    373                                       SectionKind::getDataRel());
    374   }
    375 }
    376 
    377 const MCSection *TargetLoweringObjectFileELF::getStaticDtorSection(
    378     unsigned Priority, const MCSymbol *KeySym) const {
    379   // The default scheme is .ctor / .dtor, so we have to invert the priority
    380   // numbering.
    381   if (Priority == 65535)
    382     return StaticDtorSection;
    383 
    384   if (UseInitArray) {
    385     std::string Name = std::string(".fini_array.") + utostr(Priority);
    386     return getContext().getELFSection(Name, ELF::SHT_FINI_ARRAY,
    387                                       ELF::SHF_ALLOC | ELF::SHF_WRITE,
    388                                       SectionKind::getDataRel());
    389   } else {
    390     std::string Name = std::string(".dtors.") + utostr(65535 - Priority);
    391     return getContext().getELFSection(Name, ELF::SHT_PROGBITS,
    392                                       ELF::SHF_ALLOC |ELF::SHF_WRITE,
    393                                       SectionKind::getDataRel());
    394   }
    395 }
    396 
    397 void
    398 TargetLoweringObjectFileELF::InitializeELF(bool UseInitArray_) {
    399   UseInitArray = UseInitArray_;
    400   if (!UseInitArray)
    401     return;
    402 
    403   StaticCtorSection =
    404     getContext().getELFSection(".init_array", ELF::SHT_INIT_ARRAY,
    405                                ELF::SHF_WRITE |
    406                                ELF::SHF_ALLOC,
    407                                SectionKind::getDataRel());
    408   StaticDtorSection =
    409     getContext().getELFSection(".fini_array", ELF::SHT_FINI_ARRAY,
    410                                ELF::SHF_WRITE |
    411                                ELF::SHF_ALLOC,
    412                                SectionKind::getDataRel());
    413 }
    414 
    415 //===----------------------------------------------------------------------===//
    416 //                                 MachO
    417 //===----------------------------------------------------------------------===//
    418 
    419 /// getDepLibFromLinkerOpt - Extract the dependent library name from a linker
    420 /// option string. Returns StringRef() if the option does not specify a library.
    421 StringRef TargetLoweringObjectFileMachO::
    422 getDepLibFromLinkerOpt(StringRef LinkerOption) const {
    423   const char *LibCmd = "-l";
    424   if (LinkerOption.startswith(LibCmd))
    425     return LinkerOption.substr(strlen(LibCmd));
    426   return StringRef();
    427 }
    428 
    429 /// emitModuleFlags - Perform code emission for module flags.
    430 void TargetLoweringObjectFileMachO::
    431 emitModuleFlags(MCStreamer &Streamer,
    432                 ArrayRef<Module::ModuleFlagEntry> ModuleFlags,
    433                 Mangler &Mang, const TargetMachine &TM) const {
    434   unsigned VersionVal = 0;
    435   unsigned ImageInfoFlags = 0;
    436   MDNode *LinkerOptions = nullptr;
    437   StringRef SectionVal;
    438 
    439   for (ArrayRef<Module::ModuleFlagEntry>::iterator
    440          i = ModuleFlags.begin(), e = ModuleFlags.end(); i != e; ++i) {
    441     const Module::ModuleFlagEntry &MFE = *i;
    442 
    443     // Ignore flags with 'Require' behavior.
    444     if (MFE.Behavior == Module::Require)
    445       continue;
    446 
    447     StringRef Key = MFE.Key->getString();
    448     Value *Val = MFE.Val;
    449 
    450     if (Key == "Objective-C Image Info Version") {
    451       VersionVal = cast<ConstantInt>(Val)->getZExtValue();
    452     } else if (Key == "Objective-C Garbage Collection" ||
    453                Key == "Objective-C GC Only" ||
    454                Key == "Objective-C Is Simulated") {
    455       ImageInfoFlags |= cast<ConstantInt>(Val)->getZExtValue();
    456     } else if (Key == "Objective-C Image Info Section") {
    457       SectionVal = cast<MDString>(Val)->getString();
    458     } else if (Key == "Linker Options") {
    459       LinkerOptions = cast<MDNode>(Val);
    460     }
    461   }
    462 
    463   // Emit the linker options if present.
    464   if (LinkerOptions) {
    465     for (unsigned i = 0, e = LinkerOptions->getNumOperands(); i != e; ++i) {
    466       MDNode *MDOptions = cast<MDNode>(LinkerOptions->getOperand(i));
    467       SmallVector<std::string, 4> StrOptions;
    468 
    469       // Convert to strings.
    470       for (unsigned ii = 0, ie = MDOptions->getNumOperands(); ii != ie; ++ii) {
    471         MDString *MDOption = cast<MDString>(MDOptions->getOperand(ii));
    472         StrOptions.push_back(MDOption->getString());
    473       }
    474 
    475       Streamer.EmitLinkerOptions(StrOptions);
    476     }
    477   }
    478 
    479   // The section is mandatory. If we don't have it, then we don't have GC info.
    480   if (SectionVal.empty()) return;
    481 
    482   StringRef Segment, Section;
    483   unsigned TAA = 0, StubSize = 0;
    484   bool TAAParsed;
    485   std::string ErrorCode =
    486     MCSectionMachO::ParseSectionSpecifier(SectionVal, Segment, Section,
    487                                           TAA, TAAParsed, StubSize);
    488   if (!ErrorCode.empty())
    489     // If invalid, report the error with report_fatal_error.
    490     report_fatal_error("Invalid section specifier '" + Section + "': " +
    491                        ErrorCode + ".");
    492 
    493   // Get the section.
    494   const MCSectionMachO *S =
    495     getContext().getMachOSection(Segment, Section, TAA, StubSize,
    496                                  SectionKind::getDataNoRel());
    497   Streamer.SwitchSection(S);
    498   Streamer.EmitLabel(getContext().
    499                      GetOrCreateSymbol(StringRef("L_OBJC_IMAGE_INFO")));
    500   Streamer.EmitIntValue(VersionVal, 4);
    501   Streamer.EmitIntValue(ImageInfoFlags, 4);
    502   Streamer.AddBlankLine();
    503 }
    504 
    505 static void checkMachOComdat(const GlobalValue *GV) {
    506   const Comdat *C = GV->getComdat();
    507   if (!C)
    508     return;
    509 
    510   report_fatal_error("MachO doesn't support COMDATs, '" + C->getName() +
    511                      "' cannot be lowered.");
    512 }
    513 
    514 const MCSection *TargetLoweringObjectFileMachO::getExplicitSectionGlobal(
    515     const GlobalValue *GV, SectionKind Kind, Mangler &Mang,
    516     const TargetMachine &TM) const {
    517   // Parse the section specifier and create it if valid.
    518   StringRef Segment, Section;
    519   unsigned TAA = 0, StubSize = 0;
    520   bool TAAParsed;
    521 
    522   checkMachOComdat(GV);
    523 
    524   std::string ErrorCode =
    525     MCSectionMachO::ParseSectionSpecifier(GV->getSection(), Segment, Section,
    526                                           TAA, TAAParsed, StubSize);
    527   if (!ErrorCode.empty()) {
    528     // If invalid, report the error with report_fatal_error.
    529     report_fatal_error("Global variable '" + GV->getName() +
    530                        "' has an invalid section specifier '" +
    531                        GV->getSection() + "': " + ErrorCode + ".");
    532   }
    533 
    534   // Get the section.
    535   const MCSectionMachO *S =
    536     getContext().getMachOSection(Segment, Section, TAA, StubSize, Kind);
    537 
    538   // If TAA wasn't set by ParseSectionSpecifier() above,
    539   // use the value returned by getMachOSection() as a default.
    540   if (!TAAParsed)
    541     TAA = S->getTypeAndAttributes();
    542 
    543   // Okay, now that we got the section, verify that the TAA & StubSize agree.
    544   // If the user declared multiple globals with different section flags, we need
    545   // to reject it here.
    546   if (S->getTypeAndAttributes() != TAA || S->getStubSize() != StubSize) {
    547     // If invalid, report the error with report_fatal_error.
    548     report_fatal_error("Global variable '" + GV->getName() +
    549                        "' section type or attributes does not match previous"
    550                        " section specifier");
    551   }
    552 
    553   return S;
    554 }
    555 
    556 bool TargetLoweringObjectFileMachO::isSectionAtomizableBySymbols(
    557     const MCSection &Section) const {
    558     const MCSectionMachO &SMO = static_cast<const MCSectionMachO&>(Section);
    559 
    560     // Sections holding 1 byte strings are atomized based on the data
    561     // they contain.
    562     // Sections holding 2 byte strings require symbols in order to be
    563     // atomized.
    564     // There is no dedicated section for 4 byte strings.
    565     if (SMO.getKind().isMergeable1ByteCString())
    566       return false;
    567 
    568     if (SMO.getSegmentName() == "__DATA" &&
    569         SMO.getSectionName() == "__cfstring")
    570       return false;
    571 
    572     switch (SMO.getType()) {
    573     default:
    574       return true;
    575 
    576       // These sections are atomized at the element boundaries without using
    577       // symbols.
    578     case MachO::S_4BYTE_LITERALS:
    579     case MachO::S_8BYTE_LITERALS:
    580     case MachO::S_16BYTE_LITERALS:
    581     case MachO::S_LITERAL_POINTERS:
    582     case MachO::S_NON_LAZY_SYMBOL_POINTERS:
    583     case MachO::S_LAZY_SYMBOL_POINTERS:
    584     case MachO::S_MOD_INIT_FUNC_POINTERS:
    585     case MachO::S_MOD_TERM_FUNC_POINTERS:
    586     case MachO::S_INTERPOSING:
    587       return false;
    588     }
    589 }
    590 
    591 const MCSection *TargetLoweringObjectFileMachO::
    592 SelectSectionForGlobal(const GlobalValue *GV, SectionKind Kind,
    593                        Mangler &Mang, const TargetMachine &TM) const {
    594   checkMachOComdat(GV);
    595 
    596   // Handle thread local data.
    597   if (Kind.isThreadBSS()) return TLSBSSSection;
    598   if (Kind.isThreadData()) return TLSDataSection;
    599 
    600   if (Kind.isText())
    601     return GV->isWeakForLinker() ? TextCoalSection : TextSection;
    602 
    603   // If this is weak/linkonce, put this in a coalescable section, either in text
    604   // or data depending on if it is writable.
    605   if (GV->isWeakForLinker()) {
    606     if (Kind.isReadOnly())
    607       return ConstTextCoalSection;
    608     return DataCoalSection;
    609   }
    610 
    611   // FIXME: Alignment check should be handled by section classifier.
    612   if (Kind.isMergeable1ByteCString() &&
    613       TM.getDataLayout()->getPreferredAlignment(cast<GlobalVariable>(GV)) < 32)
    614     return CStringSection;
    615 
    616   // Do not put 16-bit arrays in the UString section if they have an
    617   // externally visible label, this runs into issues with certain linker
    618   // versions.
    619   if (Kind.isMergeable2ByteCString() && !GV->hasExternalLinkage() &&
    620       TM.getDataLayout()->getPreferredAlignment(cast<GlobalVariable>(GV)) < 32)
    621     return UStringSection;
    622 
    623   if (Kind.isMergeableConst()) {
    624     if (Kind.isMergeableConst4())
    625       return FourByteConstantSection;
    626     if (Kind.isMergeableConst8())
    627       return EightByteConstantSection;
    628     if (Kind.isMergeableConst16())
    629       return SixteenByteConstantSection;
    630   }
    631 
    632   // Otherwise, if it is readonly, but not something we can specially optimize,
    633   // just drop it in .const.
    634   if (Kind.isReadOnly())
    635     return ReadOnlySection;
    636 
    637   // If this is marked const, put it into a const section.  But if the dynamic
    638   // linker needs to write to it, put it in the data segment.
    639   if (Kind.isReadOnlyWithRel())
    640     return ConstDataSection;
    641 
    642   // Put zero initialized globals with strong external linkage in the
    643   // DATA, __common section with the .zerofill directive.
    644   if (Kind.isBSSExtern())
    645     return DataCommonSection;
    646 
    647   // Put zero initialized globals with local linkage in __DATA,__bss directive
    648   // with the .zerofill directive (aka .lcomm).
    649   if (Kind.isBSSLocal())
    650     return DataBSSSection;
    651 
    652   // Otherwise, just drop the variable in the normal data section.
    653   return DataSection;
    654 }
    655 
    656 const MCSection *
    657 TargetLoweringObjectFileMachO::getSectionForConstant(SectionKind Kind) const {
    658   // If this constant requires a relocation, we have to put it in the data
    659   // segment, not in the text segment.
    660   if (Kind.isDataRel() || Kind.isReadOnlyWithRel())
    661     return ConstDataSection;
    662 
    663   if (Kind.isMergeableConst4())
    664     return FourByteConstantSection;
    665   if (Kind.isMergeableConst8())
    666     return EightByteConstantSection;
    667   if (Kind.isMergeableConst16())
    668     return SixteenByteConstantSection;
    669   return ReadOnlySection;  // .const
    670 }
    671 
    672 const MCExpr *TargetLoweringObjectFileMachO::getTTypeGlobalReference(
    673     const GlobalValue *GV, unsigned Encoding, Mangler &Mang,
    674     const TargetMachine &TM, MachineModuleInfo *MMI,
    675     MCStreamer &Streamer) const {
    676   // The mach-o version of this method defaults to returning a stub reference.
    677 
    678   if (Encoding & DW_EH_PE_indirect) {
    679     MachineModuleInfoMachO &MachOMMI =
    680       MMI->getObjFileInfo<MachineModuleInfoMachO>();
    681 
    682     MCSymbol *SSym =
    683         getSymbolWithGlobalValueBase(GV, "$non_lazy_ptr", Mang, TM);
    684 
    685     // Add information about the stub reference to MachOMMI so that the stub
    686     // gets emitted by the asmprinter.
    687     MachineModuleInfoImpl::StubValueTy &StubSym =
    688       GV->hasHiddenVisibility() ? MachOMMI.getHiddenGVStubEntry(SSym) :
    689                                   MachOMMI.getGVStubEntry(SSym);
    690     if (!StubSym.getPointer()) {
    691       MCSymbol *Sym = TM.getSymbol(GV, Mang);
    692       StubSym = MachineModuleInfoImpl::StubValueTy(Sym, !GV->hasLocalLinkage());
    693     }
    694 
    695     return TargetLoweringObjectFile::
    696       getTTypeReference(MCSymbolRefExpr::Create(SSym, getContext()),
    697                         Encoding & ~dwarf::DW_EH_PE_indirect, Streamer);
    698   }
    699 
    700   return TargetLoweringObjectFile::getTTypeGlobalReference(GV, Encoding, Mang,
    701                                                            TM, MMI, Streamer);
    702 }
    703 
    704 MCSymbol *TargetLoweringObjectFileMachO::getCFIPersonalitySymbol(
    705     const GlobalValue *GV, Mangler &Mang, const TargetMachine &TM,
    706     MachineModuleInfo *MMI) const {
    707   // The mach-o version of this method defaults to returning a stub reference.
    708   MachineModuleInfoMachO &MachOMMI =
    709     MMI->getObjFileInfo<MachineModuleInfoMachO>();
    710 
    711   MCSymbol *SSym = getSymbolWithGlobalValueBase(GV, "$non_lazy_ptr", Mang, TM);
    712 
    713   // Add information about the stub reference to MachOMMI so that the stub
    714   // gets emitted by the asmprinter.
    715   MachineModuleInfoImpl::StubValueTy &StubSym = MachOMMI.getGVStubEntry(SSym);
    716   if (!StubSym.getPointer()) {
    717     MCSymbol *Sym = TM.getSymbol(GV, Mang);
    718     StubSym = MachineModuleInfoImpl::StubValueTy(Sym, !GV->hasLocalLinkage());
    719   }
    720 
    721   return SSym;
    722 }
    723 
    724 //===----------------------------------------------------------------------===//
    725 //                                  COFF
    726 //===----------------------------------------------------------------------===//
    727 
    728 static unsigned
    729 getCOFFSectionFlags(SectionKind K) {
    730   unsigned Flags = 0;
    731 
    732   if (K.isMetadata())
    733     Flags |=
    734       COFF::IMAGE_SCN_MEM_DISCARDABLE;
    735   else if (K.isText())
    736     Flags |=
    737       COFF::IMAGE_SCN_MEM_EXECUTE |
    738       COFF::IMAGE_SCN_MEM_READ |
    739       COFF::IMAGE_SCN_CNT_CODE;
    740   else if (K.isBSS ())
    741     Flags |=
    742       COFF::IMAGE_SCN_CNT_UNINITIALIZED_DATA |
    743       COFF::IMAGE_SCN_MEM_READ |
    744       COFF::IMAGE_SCN_MEM_WRITE;
    745   else if (K.isThreadLocal())
    746     Flags |=
    747       COFF::IMAGE_SCN_CNT_INITIALIZED_DATA |
    748       COFF::IMAGE_SCN_MEM_READ |
    749       COFF::IMAGE_SCN_MEM_WRITE;
    750   else if (K.isReadOnly())
    751     Flags |=
    752       COFF::IMAGE_SCN_CNT_INITIALIZED_DATA |
    753       COFF::IMAGE_SCN_MEM_READ;
    754   else if (K.isWriteable())
    755     Flags |=
    756       COFF::IMAGE_SCN_CNT_INITIALIZED_DATA |
    757       COFF::IMAGE_SCN_MEM_READ |
    758       COFF::IMAGE_SCN_MEM_WRITE;
    759 
    760   return Flags;
    761 }
    762 
    763 static const GlobalValue *getComdatGVForCOFF(const GlobalValue *GV) {
    764   const Comdat *C = GV->getComdat();
    765   assert(C && "expected GV to have a Comdat!");
    766 
    767   StringRef ComdatGVName = C->getName();
    768   const GlobalValue *ComdatGV = GV->getParent()->getNamedValue(ComdatGVName);
    769   if (!ComdatGV)
    770     report_fatal_error("Associative COMDAT symbol '" + ComdatGVName +
    771                        "' does not exist.");
    772 
    773   if (ComdatGV->getComdat() != C)
    774     report_fatal_error("Associative COMDAT symbol '" + ComdatGVName +
    775                        "' is not a key for it's COMDAT.");
    776 
    777   return ComdatGV;
    778 }
    779 
    780 static int getSelectionForCOFF(const GlobalValue *GV) {
    781   if (const Comdat *C = GV->getComdat()) {
    782     const GlobalValue *ComdatKey = getComdatGVForCOFF(GV);
    783     if (const auto *GA = dyn_cast<GlobalAlias>(ComdatKey))
    784       ComdatKey = GA->getBaseObject();
    785     if (ComdatKey == GV) {
    786       switch (C->getSelectionKind()) {
    787       case Comdat::Any:
    788         return COFF::IMAGE_COMDAT_SELECT_ANY;
    789       case Comdat::ExactMatch:
    790         return COFF::IMAGE_COMDAT_SELECT_EXACT_MATCH;
    791       case Comdat::Largest:
    792         return COFF::IMAGE_COMDAT_SELECT_LARGEST;
    793       case Comdat::NoDuplicates:
    794         return COFF::IMAGE_COMDAT_SELECT_NODUPLICATES;
    795       case Comdat::SameSize:
    796         return COFF::IMAGE_COMDAT_SELECT_SAME_SIZE;
    797       }
    798     } else {
    799       return COFF::IMAGE_COMDAT_SELECT_ASSOCIATIVE;
    800     }
    801   } else if (GV->isWeakForLinker()) {
    802     return COFF::IMAGE_COMDAT_SELECT_ANY;
    803   }
    804   return 0;
    805 }
    806 
    807 const MCSection *TargetLoweringObjectFileCOFF::getExplicitSectionGlobal(
    808     const GlobalValue *GV, SectionKind Kind, Mangler &Mang,
    809     const TargetMachine &TM) const {
    810   int Selection = 0;
    811   unsigned Characteristics = getCOFFSectionFlags(Kind);
    812   StringRef Name = GV->getSection();
    813   StringRef COMDATSymName = "";
    814   if ((GV->isWeakForLinker() || GV->hasComdat()) && !Kind.isCommon()) {
    815     Selection = getSelectionForCOFF(GV);
    816     const GlobalValue *ComdatGV;
    817     if (Selection == COFF::IMAGE_COMDAT_SELECT_ASSOCIATIVE)
    818       ComdatGV = getComdatGVForCOFF(GV);
    819     else
    820       ComdatGV = GV;
    821 
    822     if (!ComdatGV->hasPrivateLinkage()) {
    823       MCSymbol *Sym = TM.getSymbol(ComdatGV, Mang);
    824       COMDATSymName = Sym->getName();
    825       Characteristics |= COFF::IMAGE_SCN_LNK_COMDAT;
    826     } else {
    827       Selection = 0;
    828     }
    829   }
    830   return getContext().getCOFFSection(Name,
    831                                      Characteristics,
    832                                      Kind,
    833                                      COMDATSymName,
    834                                      Selection);
    835 }
    836 
    837 static const char *getCOFFSectionNameForUniqueGlobal(SectionKind Kind) {
    838   if (Kind.isText())
    839     return ".text";
    840   if (Kind.isBSS())
    841     return ".bss";
    842   if (Kind.isThreadLocal())
    843     return ".tls$";
    844   if (Kind.isWriteable())
    845     return ".data";
    846   return ".rdata";
    847 }
    848 
    849 
    850 const MCSection *TargetLoweringObjectFileCOFF::
    851 SelectSectionForGlobal(const GlobalValue *GV, SectionKind Kind,
    852                        Mangler &Mang, const TargetMachine &TM) const {
    853   // If we have -ffunction-sections then we should emit the global value to a
    854   // uniqued section specifically for it.
    855   bool EmitUniquedSection;
    856   if (Kind.isText())
    857     EmitUniquedSection = TM.getFunctionSections();
    858   else
    859     EmitUniquedSection = TM.getDataSections();
    860 
    861   // If this global is linkonce/weak and the target handles this by emitting it
    862   // into a 'uniqued' section name, create and return the section now.
    863   // Section names depend on the name of the symbol which is not feasible if the
    864   // symbol has private linkage.
    865   if ((GV->isWeakForLinker() || EmitUniquedSection || GV->hasComdat()) &&
    866       !Kind.isCommon()) {
    867     const char *Name = getCOFFSectionNameForUniqueGlobal(Kind);
    868     unsigned Characteristics = getCOFFSectionFlags(Kind);
    869 
    870     Characteristics |= COFF::IMAGE_SCN_LNK_COMDAT;
    871     int Selection = getSelectionForCOFF(GV);
    872     if (!Selection)
    873       Selection = COFF::IMAGE_COMDAT_SELECT_NODUPLICATES;
    874     const GlobalValue *ComdatGV;
    875     if (GV->hasComdat())
    876       ComdatGV = getComdatGVForCOFF(GV);
    877     else
    878       ComdatGV = GV;
    879 
    880     if (!ComdatGV->hasPrivateLinkage()) {
    881       MCSymbol *Sym = TM.getSymbol(ComdatGV, Mang);
    882       StringRef COMDATSymName = Sym->getName();
    883       return getContext().getCOFFSection(Name, Characteristics, Kind,
    884                                          COMDATSymName, Selection);
    885     }
    886   }
    887 
    888   if (Kind.isText())
    889     return TextSection;
    890 
    891   if (Kind.isThreadLocal())
    892     return TLSDataSection;
    893 
    894   if (Kind.isReadOnly())
    895     return ReadOnlySection;
    896 
    897   // Note: we claim that common symbols are put in BSSSection, but they are
    898   // really emitted with the magic .comm directive, which creates a symbol table
    899   // entry but not a section.
    900   if (Kind.isBSS() || Kind.isCommon())
    901     return BSSSection;
    902 
    903   return DataSection;
    904 }
    905 
    906 StringRef TargetLoweringObjectFileCOFF::
    907 getDepLibFromLinkerOpt(StringRef LinkerOption) const {
    908   const char *LibCmd = "/DEFAULTLIB:";
    909   if (LinkerOption.startswith(LibCmd))
    910     return LinkerOption.substr(strlen(LibCmd));
    911   return StringRef();
    912 }
    913 
    914 void TargetLoweringObjectFileCOFF::
    915 emitModuleFlags(MCStreamer &Streamer,
    916                 ArrayRef<Module::ModuleFlagEntry> ModuleFlags,
    917                 Mangler &Mang, const TargetMachine &TM) const {
    918   MDNode *LinkerOptions = nullptr;
    919 
    920   // Look for the "Linker Options" flag, since it's the only one we support.
    921   for (ArrayRef<Module::ModuleFlagEntry>::iterator
    922        i = ModuleFlags.begin(), e = ModuleFlags.end(); i != e; ++i) {
    923     const Module::ModuleFlagEntry &MFE = *i;
    924     StringRef Key = MFE.Key->getString();
    925     Value *Val = MFE.Val;
    926     if (Key == "Linker Options") {
    927       LinkerOptions = cast<MDNode>(Val);
    928       break;
    929     }
    930   }
    931   if (!LinkerOptions)
    932     return;
    933 
    934   // Emit the linker options to the linker .drectve section.  According to the
    935   // spec, this section is a space-separated string containing flags for linker.
    936   const MCSection *Sec = getDrectveSection();
    937   Streamer.SwitchSection(Sec);
    938   for (unsigned i = 0, e = LinkerOptions->getNumOperands(); i != e; ++i) {
    939     MDNode *MDOptions = cast<MDNode>(LinkerOptions->getOperand(i));
    940     for (unsigned ii = 0, ie = MDOptions->getNumOperands(); ii != ie; ++ii) {
    941       MDString *MDOption = cast<MDString>(MDOptions->getOperand(ii));
    942       StringRef Op = MDOption->getString();
    943       // Lead with a space for consistency with our dllexport implementation.
    944       std::string Escaped(" ");
    945       if (Op.find(" ") != StringRef::npos) {
    946         // The PE-COFF spec says args with spaces must be quoted.  It doesn't say
    947         // how to escape quotes, but it probably uses this algorithm:
    948         // http://msdn.microsoft.com/en-us/library/17w5ykft(v=vs.85).aspx
    949         // FIXME: Reuse escaping code from Support/Windows/Program.inc
    950         Escaped.push_back('\"');
    951         Escaped.append(Op);
    952         Escaped.push_back('\"');
    953       } else {
    954         Escaped.append(Op);
    955       }
    956       Streamer.EmitBytes(Escaped);
    957     }
    958   }
    959 }
    960 
    961 static const MCSection *getAssociativeCOFFSection(MCContext &Ctx,
    962                                                   const MCSection *Sec,
    963                                                   const MCSymbol *KeySym) {
    964   // Return the normal section if we don't have to be associative.
    965   if (!KeySym)
    966     return Sec;
    967 
    968   // Make an associative section with the same name and kind as the normal
    969   // section.
    970   const MCSectionCOFF *SecCOFF = cast<MCSectionCOFF>(Sec);
    971   unsigned Characteristics =
    972       SecCOFF->getCharacteristics() | COFF::IMAGE_SCN_LNK_COMDAT;
    973   return Ctx.getCOFFSection(SecCOFF->getSectionName(), Characteristics,
    974                             SecCOFF->getKind(), KeySym->getName(),
    975                             COFF::IMAGE_COMDAT_SELECT_ASSOCIATIVE);
    976 }
    977 
    978 const MCSection *TargetLoweringObjectFileCOFF::getStaticCtorSection(
    979     unsigned Priority, const MCSymbol *KeySym) const {
    980   return getAssociativeCOFFSection(getContext(), StaticCtorSection, KeySym);
    981 }
    982 
    983 const MCSection *TargetLoweringObjectFileCOFF::getStaticDtorSection(
    984     unsigned Priority, const MCSymbol *KeySym) const {
    985   return getAssociativeCOFFSection(getContext(), StaticDtorSection, KeySym);
    986 }
    987