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