xref: /llvm-project/llvm/lib/CodeGen/TargetLoweringObjectFileImpl.cpp (revision 8f21294897befee48f9f72734ea1b0ad4c920aa0)
1 //===- llvm/CodeGen/TargetLoweringObjectFileImpl.cpp - Object File Info ---===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file implements classes used to handle lowerings specific to common
10 // object file formats.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "llvm/CodeGen/TargetLoweringObjectFileImpl.h"
15 #include "llvm/ADT/SmallString.h"
16 #include "llvm/ADT/SmallVector.h"
17 #include "llvm/ADT/StringExtras.h"
18 #include "llvm/ADT/StringRef.h"
19 #include "llvm/BinaryFormat/COFF.h"
20 #include "llvm/BinaryFormat/Dwarf.h"
21 #include "llvm/BinaryFormat/ELF.h"
22 #include "llvm/BinaryFormat/MachO.h"
23 #include "llvm/BinaryFormat/Wasm.h"
24 #include "llvm/CodeGen/BasicBlockSectionUtils.h"
25 #include "llvm/CodeGen/MachineBasicBlock.h"
26 #include "llvm/CodeGen/MachineFunction.h"
27 #include "llvm/CodeGen/MachineModuleInfo.h"
28 #include "llvm/CodeGen/MachineModuleInfoImpls.h"
29 #include "llvm/IR/Comdat.h"
30 #include "llvm/IR/Constants.h"
31 #include "llvm/IR/DataLayout.h"
32 #include "llvm/IR/DerivedTypes.h"
33 #include "llvm/IR/DiagnosticInfo.h"
34 #include "llvm/IR/DiagnosticPrinter.h"
35 #include "llvm/IR/Function.h"
36 #include "llvm/IR/GlobalAlias.h"
37 #include "llvm/IR/GlobalObject.h"
38 #include "llvm/IR/GlobalValue.h"
39 #include "llvm/IR/GlobalVariable.h"
40 #include "llvm/IR/Mangler.h"
41 #include "llvm/IR/Metadata.h"
42 #include "llvm/IR/Module.h"
43 #include "llvm/IR/PseudoProbe.h"
44 #include "llvm/IR/Type.h"
45 #include "llvm/MC/MCAsmInfo.h"
46 #include "llvm/MC/MCContext.h"
47 #include "llvm/MC/MCExpr.h"
48 #include "llvm/MC/MCSectionCOFF.h"
49 #include "llvm/MC/MCSectionELF.h"
50 #include "llvm/MC/MCSectionGOFF.h"
51 #include "llvm/MC/MCSectionMachO.h"
52 #include "llvm/MC/MCSectionWasm.h"
53 #include "llvm/MC/MCSectionXCOFF.h"
54 #include "llvm/MC/MCStreamer.h"
55 #include "llvm/MC/MCSymbol.h"
56 #include "llvm/MC/MCSymbolELF.h"
57 #include "llvm/MC/MCValue.h"
58 #include "llvm/MC/SectionKind.h"
59 #include "llvm/ProfileData/InstrProf.h"
60 #include "llvm/Support/Base64.h"
61 #include "llvm/Support/Casting.h"
62 #include "llvm/Support/CodeGen.h"
63 #include "llvm/Support/ErrorHandling.h"
64 #include "llvm/Support/Format.h"
65 #include "llvm/Support/raw_ostream.h"
66 #include "llvm/Target/TargetMachine.h"
67 #include "llvm/TargetParser/Triple.h"
68 #include <cassert>
69 #include <string>
70 
71 using namespace llvm;
72 using namespace dwarf;
73 
74 static cl::opt<bool> JumpTableInFunctionSection(
75     "jumptable-in-function-section", cl::Hidden, cl::init(false),
76     cl::desc("Putting Jump Table in function section"));
77 
78 static void GetObjCImageInfo(Module &M, unsigned &Version, unsigned &Flags,
79                              StringRef &Section) {
80   SmallVector<Module::ModuleFlagEntry, 8> ModuleFlags;
81   M.getModuleFlagsMetadata(ModuleFlags);
82 
83   for (const auto &MFE: ModuleFlags) {
84     // Ignore flags with 'Require' behaviour.
85     if (MFE.Behavior == Module::Require)
86       continue;
87 
88     StringRef Key = MFE.Key->getString();
89     if (Key == "Objective-C Image Info Version") {
90       Version = mdconst::extract<ConstantInt>(MFE.Val)->getZExtValue();
91     } else if (Key == "Objective-C Garbage Collection" ||
92                Key == "Objective-C GC Only" ||
93                Key == "Objective-C Is Simulated" ||
94                Key == "Objective-C Class Properties" ||
95                Key == "Objective-C Image Swift Version") {
96       Flags |= mdconst::extract<ConstantInt>(MFE.Val)->getZExtValue();
97     } else if (Key == "Objective-C Image Info Section") {
98       Section = cast<MDString>(MFE.Val)->getString();
99     }
100     // Backend generates L_OBJC_IMAGE_INFO from Swift ABI version + major + minor +
101     // "Objective-C Garbage Collection".
102     else if (Key == "Swift ABI Version") {
103       Flags |= (mdconst::extract<ConstantInt>(MFE.Val)->getZExtValue()) << 8;
104     } else if (Key == "Swift Major Version") {
105       Flags |= (mdconst::extract<ConstantInt>(MFE.Val)->getZExtValue()) << 24;
106     } else if (Key == "Swift Minor Version") {
107       Flags |= (mdconst::extract<ConstantInt>(MFE.Val)->getZExtValue()) << 16;
108     }
109   }
110 }
111 
112 //===----------------------------------------------------------------------===//
113 //                                  ELF
114 //===----------------------------------------------------------------------===//
115 
116 TargetLoweringObjectFileELF::TargetLoweringObjectFileELF() {
117   SupportDSOLocalEquivalentLowering = true;
118 }
119 
120 void TargetLoweringObjectFileELF::Initialize(MCContext &Ctx,
121                                              const TargetMachine &TgtM) {
122   TargetLoweringObjectFile::Initialize(Ctx, TgtM);
123 
124   CodeModel::Model CM = TgtM.getCodeModel();
125   InitializeELF(TgtM.Options.UseInitArray);
126 
127   switch (TgtM.getTargetTriple().getArch()) {
128   case Triple::arm:
129   case Triple::armeb:
130   case Triple::thumb:
131   case Triple::thumbeb:
132     if (Ctx.getAsmInfo()->getExceptionHandlingType() == ExceptionHandling::ARM)
133       break;
134     // Fallthrough if not using EHABI
135     [[fallthrough]];
136   case Triple::ppc:
137   case Triple::ppcle:
138   case Triple::x86:
139     PersonalityEncoding = isPositionIndependent()
140                               ? dwarf::DW_EH_PE_indirect |
141                                     dwarf::DW_EH_PE_pcrel |
142                                     dwarf::DW_EH_PE_sdata4
143                               : dwarf::DW_EH_PE_absptr;
144     LSDAEncoding = isPositionIndependent()
145                        ? dwarf::DW_EH_PE_pcrel | dwarf::DW_EH_PE_sdata4
146                        : dwarf::DW_EH_PE_absptr;
147     TTypeEncoding = isPositionIndependent()
148                         ? dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
149                               dwarf::DW_EH_PE_sdata4
150                         : dwarf::DW_EH_PE_absptr;
151     break;
152   case Triple::x86_64:
153     if (isPositionIndependent()) {
154       PersonalityEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
155         ((CM == CodeModel::Small || CM == CodeModel::Medium)
156          ? dwarf::DW_EH_PE_sdata4 : dwarf::DW_EH_PE_sdata8);
157       LSDAEncoding = dwarf::DW_EH_PE_pcrel |
158         (CM == CodeModel::Small
159          ? dwarf::DW_EH_PE_sdata4 : dwarf::DW_EH_PE_sdata8);
160       TTypeEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
161         ((CM == CodeModel::Small || CM == CodeModel::Medium)
162          ? dwarf::DW_EH_PE_sdata4 : dwarf::DW_EH_PE_sdata8);
163     } else {
164       PersonalityEncoding =
165         (CM == CodeModel::Small || CM == CodeModel::Medium)
166         ? dwarf::DW_EH_PE_udata4 : dwarf::DW_EH_PE_absptr;
167       LSDAEncoding = (CM == CodeModel::Small)
168         ? dwarf::DW_EH_PE_udata4 : dwarf::DW_EH_PE_absptr;
169       TTypeEncoding = (CM == CodeModel::Small)
170         ? dwarf::DW_EH_PE_udata4 : dwarf::DW_EH_PE_absptr;
171     }
172     break;
173   case Triple::hexagon:
174     PersonalityEncoding = dwarf::DW_EH_PE_absptr;
175     LSDAEncoding = dwarf::DW_EH_PE_absptr;
176     TTypeEncoding = dwarf::DW_EH_PE_absptr;
177     if (isPositionIndependent()) {
178       PersonalityEncoding |= dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel;
179       LSDAEncoding |= dwarf::DW_EH_PE_pcrel;
180       TTypeEncoding |= dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel;
181     }
182     break;
183   case Triple::aarch64:
184   case Triple::aarch64_be:
185   case Triple::aarch64_32:
186     // The small model guarantees static code/data size < 4GB, but not where it
187     // will be in memory. Most of these could end up >2GB away so even a signed
188     // pc-relative 32-bit address is insufficient, theoretically.
189     //
190     // Use DW_EH_PE_indirect even for -fno-pic to avoid copy relocations.
191     LSDAEncoding = dwarf::DW_EH_PE_pcrel |
192                    (TgtM.getTargetTriple().getEnvironment() == Triple::GNUILP32
193                         ? dwarf::DW_EH_PE_sdata4
194                         : dwarf::DW_EH_PE_sdata8);
195     PersonalityEncoding = LSDAEncoding | dwarf::DW_EH_PE_indirect;
196     TTypeEncoding = LSDAEncoding | dwarf::DW_EH_PE_indirect;
197     break;
198   case Triple::lanai:
199     LSDAEncoding = dwarf::DW_EH_PE_absptr;
200     PersonalityEncoding = dwarf::DW_EH_PE_absptr;
201     TTypeEncoding = dwarf::DW_EH_PE_absptr;
202     break;
203   case Triple::mips:
204   case Triple::mipsel:
205   case Triple::mips64:
206   case Triple::mips64el:
207     // MIPS uses indirect pointer to refer personality functions and types, so
208     // that the eh_frame section can be read-only. DW.ref.personality will be
209     // generated for relocation.
210     PersonalityEncoding = dwarf::DW_EH_PE_indirect;
211     // FIXME: The N64 ABI probably ought to use DW_EH_PE_sdata8 but we can't
212     //        identify N64 from just a triple.
213     TTypeEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
214                     dwarf::DW_EH_PE_sdata4;
215 
216     // FreeBSD must be explicit about the data size and using pcrel since it's
217     // assembler/linker won't do the automatic conversion that the Linux tools
218     // do.
219     if (isPositionIndependent() || TgtM.getTargetTriple().isOSFreeBSD()) {
220       PersonalityEncoding |= dwarf::DW_EH_PE_pcrel | dwarf::DW_EH_PE_sdata4;
221       LSDAEncoding = dwarf::DW_EH_PE_pcrel | dwarf::DW_EH_PE_sdata4;
222     }
223     break;
224   case Triple::ppc64:
225   case Triple::ppc64le:
226     PersonalityEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
227       dwarf::DW_EH_PE_udata8;
228     LSDAEncoding = dwarf::DW_EH_PE_pcrel | dwarf::DW_EH_PE_udata8;
229     TTypeEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
230       dwarf::DW_EH_PE_udata8;
231     break;
232   case Triple::sparcel:
233   case Triple::sparc:
234     if (isPositionIndependent()) {
235       LSDAEncoding = dwarf::DW_EH_PE_pcrel | dwarf::DW_EH_PE_sdata4;
236       PersonalityEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
237         dwarf::DW_EH_PE_sdata4;
238       TTypeEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
239         dwarf::DW_EH_PE_sdata4;
240     } else {
241       LSDAEncoding = dwarf::DW_EH_PE_absptr;
242       PersonalityEncoding = dwarf::DW_EH_PE_absptr;
243       TTypeEncoding = dwarf::DW_EH_PE_absptr;
244     }
245     CallSiteEncoding = dwarf::DW_EH_PE_udata4;
246     break;
247   case Triple::riscv32:
248   case Triple::riscv64:
249     LSDAEncoding = dwarf::DW_EH_PE_pcrel | dwarf::DW_EH_PE_sdata4;
250     PersonalityEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
251                           dwarf::DW_EH_PE_sdata4;
252     TTypeEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
253                     dwarf::DW_EH_PE_sdata4;
254     CallSiteEncoding = dwarf::DW_EH_PE_udata4;
255     break;
256   case Triple::sparcv9:
257     LSDAEncoding = dwarf::DW_EH_PE_pcrel | dwarf::DW_EH_PE_sdata4;
258     if (isPositionIndependent()) {
259       PersonalityEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
260         dwarf::DW_EH_PE_sdata4;
261       TTypeEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
262         dwarf::DW_EH_PE_sdata4;
263     } else {
264       PersonalityEncoding = dwarf::DW_EH_PE_absptr;
265       TTypeEncoding = dwarf::DW_EH_PE_absptr;
266     }
267     break;
268   case Triple::systemz:
269     // All currently-defined code models guarantee that 4-byte PC-relative
270     // values will be in range.
271     if (isPositionIndependent()) {
272       PersonalityEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
273         dwarf::DW_EH_PE_sdata4;
274       LSDAEncoding = dwarf::DW_EH_PE_pcrel | dwarf::DW_EH_PE_sdata4;
275       TTypeEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
276         dwarf::DW_EH_PE_sdata4;
277     } else {
278       PersonalityEncoding = dwarf::DW_EH_PE_absptr;
279       LSDAEncoding = dwarf::DW_EH_PE_absptr;
280       TTypeEncoding = dwarf::DW_EH_PE_absptr;
281     }
282     break;
283   case Triple::loongarch32:
284   case Triple::loongarch64:
285     LSDAEncoding = dwarf::DW_EH_PE_pcrel | dwarf::DW_EH_PE_sdata4;
286     PersonalityEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
287                           dwarf::DW_EH_PE_sdata4;
288     TTypeEncoding = dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel |
289                     dwarf::DW_EH_PE_sdata4;
290     break;
291   default:
292     break;
293   }
294 }
295 
296 void TargetLoweringObjectFileELF::getModuleMetadata(Module &M) {
297   SmallVector<GlobalValue *, 4> Vec;
298   collectUsedGlobalVariables(M, Vec, false);
299   for (GlobalValue *GV : Vec)
300     if (auto *GO = dyn_cast<GlobalObject>(GV))
301       Used.insert(GO);
302 }
303 
304 void TargetLoweringObjectFileELF::emitModuleMetadata(MCStreamer &Streamer,
305                                                      Module &M) const {
306   auto &C = getContext();
307 
308   if (NamedMDNode *LinkerOptions = M.getNamedMetadata("llvm.linker.options")) {
309     auto *S = C.getELFSection(".linker-options", ELF::SHT_LLVM_LINKER_OPTIONS,
310                               ELF::SHF_EXCLUDE);
311 
312     Streamer.switchSection(S);
313 
314     for (const auto *Operand : LinkerOptions->operands()) {
315       if (cast<MDNode>(Operand)->getNumOperands() != 2)
316         report_fatal_error("invalid llvm.linker.options");
317       for (const auto &Option : cast<MDNode>(Operand)->operands()) {
318         Streamer.emitBytes(cast<MDString>(Option)->getString());
319         Streamer.emitInt8(0);
320       }
321     }
322   }
323 
324   if (NamedMDNode *DependentLibraries = M.getNamedMetadata("llvm.dependent-libraries")) {
325     auto *S = C.getELFSection(".deplibs", ELF::SHT_LLVM_DEPENDENT_LIBRARIES,
326                               ELF::SHF_MERGE | ELF::SHF_STRINGS, 1);
327 
328     Streamer.switchSection(S);
329 
330     for (const auto *Operand : DependentLibraries->operands()) {
331       Streamer.emitBytes(
332           cast<MDString>(cast<MDNode>(Operand)->getOperand(0))->getString());
333       Streamer.emitInt8(0);
334     }
335   }
336 
337   if (NamedMDNode *FuncInfo = M.getNamedMetadata(PseudoProbeDescMetadataName)) {
338     // Emit a descriptor for every function including functions that have an
339     // available external linkage. We may not want this for imported functions
340     // that has code in another thinLTO module but we don't have a good way to
341     // tell them apart from inline functions defined in header files. Therefore
342     // we put each descriptor in a separate comdat section and rely on the
343     // linker to deduplicate.
344     for (const auto *Operand : FuncInfo->operands()) {
345       const auto *MD = cast<MDNode>(Operand);
346       auto *GUID = mdconst::dyn_extract<ConstantInt>(MD->getOperand(0));
347       auto *Hash = mdconst::dyn_extract<ConstantInt>(MD->getOperand(1));
348       auto *Name = cast<MDString>(MD->getOperand(2));
349       auto *S = C.getObjectFileInfo()->getPseudoProbeDescSection(
350           TM->getFunctionSections() ? Name->getString() : StringRef());
351 
352       Streamer.switchSection(S);
353       Streamer.emitInt64(GUID->getZExtValue());
354       Streamer.emitInt64(Hash->getZExtValue());
355       Streamer.emitULEB128IntValue(Name->getString().size());
356       Streamer.emitBytes(Name->getString());
357     }
358   }
359 
360   if (NamedMDNode *LLVMStats = M.getNamedMetadata("llvm.stats")) {
361     // Emit the metadata for llvm statistics into .llvm_stats section, which is
362     // formatted as a list of key/value pair, the value is base64 encoded.
363     auto *S = C.getObjectFileInfo()->getLLVMStatsSection();
364     Streamer.switchSection(S);
365     for (const auto *Operand : LLVMStats->operands()) {
366       const auto *MD = cast<MDNode>(Operand);
367       assert(MD->getNumOperands() % 2 == 0 &&
368              ("Operand num should be even for a list of key/value pair"));
369       for (size_t I = 0; I < MD->getNumOperands(); I += 2) {
370         // Encode the key string size.
371         auto *Key = cast<MDString>(MD->getOperand(I));
372         Streamer.emitULEB128IntValue(Key->getString().size());
373         Streamer.emitBytes(Key->getString());
374         // Encode the value into a Base64 string.
375         std::string Value = encodeBase64(
376             Twine(mdconst::dyn_extract<ConstantInt>(MD->getOperand(I + 1))
377                       ->getZExtValue())
378                 .str());
379         Streamer.emitULEB128IntValue(Value.size());
380         Streamer.emitBytes(Value);
381       }
382     }
383   }
384 
385   unsigned Version = 0;
386   unsigned Flags = 0;
387   StringRef Section;
388 
389   GetObjCImageInfo(M, Version, Flags, Section);
390   if (!Section.empty()) {
391     auto *S = C.getELFSection(Section, ELF::SHT_PROGBITS, ELF::SHF_ALLOC);
392     Streamer.switchSection(S);
393     Streamer.emitLabel(C.getOrCreateSymbol(StringRef("OBJC_IMAGE_INFO")));
394     Streamer.emitInt32(Version);
395     Streamer.emitInt32(Flags);
396     Streamer.addBlankLine();
397   }
398 
399   emitCGProfileMetadata(Streamer, M);
400 }
401 
402 MCSymbol *TargetLoweringObjectFileELF::getCFIPersonalitySymbol(
403     const GlobalValue *GV, const TargetMachine &TM,
404     MachineModuleInfo *MMI) const {
405   unsigned Encoding = getPersonalityEncoding();
406   if ((Encoding & 0x80) == DW_EH_PE_indirect)
407     return getContext().getOrCreateSymbol(StringRef("DW.ref.") +
408                                           TM.getSymbol(GV)->getName());
409   if ((Encoding & 0x70) == DW_EH_PE_absptr)
410     return TM.getSymbol(GV);
411   report_fatal_error("We do not support this DWARF encoding yet!");
412 }
413 
414 void TargetLoweringObjectFileELF::emitPersonalityValue(
415     MCStreamer &Streamer, const DataLayout &DL, const MCSymbol *Sym) const {
416   SmallString<64> NameData("DW.ref.");
417   NameData += Sym->getName();
418   MCSymbolELF *Label =
419       cast<MCSymbolELF>(getContext().getOrCreateSymbol(NameData));
420   Streamer.emitSymbolAttribute(Label, MCSA_Hidden);
421   Streamer.emitSymbolAttribute(Label, MCSA_Weak);
422   unsigned Flags = ELF::SHF_ALLOC | ELF::SHF_WRITE | ELF::SHF_GROUP;
423   MCSection *Sec = getContext().getELFNamedSection(".data", Label->getName(),
424                                                    ELF::SHT_PROGBITS, Flags, 0);
425   unsigned Size = DL.getPointerSize();
426   Streamer.switchSection(Sec);
427   Streamer.emitValueToAlignment(DL.getPointerABIAlignment(0));
428   Streamer.emitSymbolAttribute(Label, MCSA_ELF_TypeObject);
429   const MCExpr *E = MCConstantExpr::create(Size, getContext());
430   Streamer.emitELFSize(Label, E);
431   Streamer.emitLabel(Label);
432 
433   Streamer.emitSymbolValue(Sym, Size);
434 }
435 
436 const MCExpr *TargetLoweringObjectFileELF::getTTypeGlobalReference(
437     const GlobalValue *GV, unsigned Encoding, const TargetMachine &TM,
438     MachineModuleInfo *MMI, MCStreamer &Streamer) const {
439   if (Encoding & DW_EH_PE_indirect) {
440     MachineModuleInfoELF &ELFMMI = MMI->getObjFileInfo<MachineModuleInfoELF>();
441 
442     MCSymbol *SSym = getSymbolWithGlobalValueBase(GV, ".DW.stub", TM);
443 
444     // Add information about the stub reference to ELFMMI so that the stub
445     // gets emitted by the asmprinter.
446     MachineModuleInfoImpl::StubValueTy &StubSym = ELFMMI.getGVStubEntry(SSym);
447     if (!StubSym.getPointer()) {
448       MCSymbol *Sym = TM.getSymbol(GV);
449       StubSym = MachineModuleInfoImpl::StubValueTy(Sym, !GV->hasLocalLinkage());
450     }
451 
452     return TargetLoweringObjectFile::
453       getTTypeReference(MCSymbolRefExpr::create(SSym, getContext()),
454                         Encoding & ~DW_EH_PE_indirect, Streamer);
455   }
456 
457   return TargetLoweringObjectFile::getTTypeGlobalReference(GV, Encoding, TM,
458                                                            MMI, Streamer);
459 }
460 
461 static SectionKind getELFKindForNamedSection(StringRef Name, SectionKind K) {
462   // N.B.: The defaults used in here are not the same ones used in MC.
463   // We follow gcc, MC follows gas. For example, given ".section .eh_frame",
464   // both gas and MC will produce a section with no flags. Given
465   // section(".eh_frame") gcc will produce:
466   //
467   //   .section   .eh_frame,"a",@progbits
468 
469   if (Name == getInstrProfSectionName(IPSK_covmap, Triple::ELF,
470                                       /*AddSegmentInfo=*/false) ||
471       Name == getInstrProfSectionName(IPSK_covfun, Triple::ELF,
472                                       /*AddSegmentInfo=*/false) ||
473       Name == getInstrProfSectionName(IPSK_covdata, Triple::ELF,
474                                       /*AddSegmentInfo=*/false) ||
475       Name == getInstrProfSectionName(IPSK_covname, Triple::ELF,
476                                       /*AddSegmentInfo=*/false) ||
477       Name == ".llvmbc" || Name == ".llvmcmd")
478     return SectionKind::getMetadata();
479 
480   if (!Name.starts_with(".")) return K;
481 
482   // Default implementation based on some magic section names.
483   if (Name == ".bss" || Name.starts_with(".bss.") ||
484       Name.starts_with(".gnu.linkonce.b.") ||
485       Name.starts_with(".llvm.linkonce.b.") || Name == ".sbss" ||
486       Name.starts_with(".sbss.") || Name.starts_with(".gnu.linkonce.sb.") ||
487       Name.starts_with(".llvm.linkonce.sb."))
488     return SectionKind::getBSS();
489 
490   if (Name == ".tdata" || Name.starts_with(".tdata.") ||
491       Name.starts_with(".gnu.linkonce.td.") ||
492       Name.starts_with(".llvm.linkonce.td."))
493     return SectionKind::getThreadData();
494 
495   if (Name == ".tbss" || Name.starts_with(".tbss.") ||
496       Name.starts_with(".gnu.linkonce.tb.") ||
497       Name.starts_with(".llvm.linkonce.tb."))
498     return SectionKind::getThreadBSS();
499 
500   return K;
501 }
502 
503 static bool hasPrefix(StringRef SectionName, StringRef Prefix) {
504   return SectionName.consume_front(Prefix) &&
505          (SectionName.empty() || SectionName[0] == '.');
506 }
507 
508 static unsigned getELFSectionType(StringRef Name, SectionKind K) {
509   // Use SHT_NOTE for section whose name starts with ".note" to allow
510   // emitting ELF notes from C variable declaration.
511   // See https://gcc.gnu.org/bugzilla/show_bug.cgi?id=77609
512   if (Name.starts_with(".note"))
513     return ELF::SHT_NOTE;
514 
515   if (hasPrefix(Name, ".init_array"))
516     return ELF::SHT_INIT_ARRAY;
517 
518   if (hasPrefix(Name, ".fini_array"))
519     return ELF::SHT_FINI_ARRAY;
520 
521   if (hasPrefix(Name, ".preinit_array"))
522     return ELF::SHT_PREINIT_ARRAY;
523 
524   if (hasPrefix(Name, ".llvm.offloading"))
525     return ELF::SHT_LLVM_OFFLOADING;
526 
527   if (K.isBSS() || K.isThreadBSS())
528     return ELF::SHT_NOBITS;
529 
530   return ELF::SHT_PROGBITS;
531 }
532 
533 static unsigned getELFSectionFlags(SectionKind K) {
534   unsigned Flags = 0;
535 
536   if (!K.isMetadata() && !K.isExclude())
537     Flags |= ELF::SHF_ALLOC;
538 
539   if (K.isExclude())
540     Flags |= ELF::SHF_EXCLUDE;
541 
542   if (K.isText())
543     Flags |= ELF::SHF_EXECINSTR;
544 
545   if (K.isExecuteOnly())
546     Flags |= ELF::SHF_ARM_PURECODE;
547 
548   if (K.isWriteable())
549     Flags |= ELF::SHF_WRITE;
550 
551   if (K.isThreadLocal())
552     Flags |= ELF::SHF_TLS;
553 
554   if (K.isMergeableCString() || K.isMergeableConst())
555     Flags |= ELF::SHF_MERGE;
556 
557   if (K.isMergeableCString())
558     Flags |= ELF::SHF_STRINGS;
559 
560   return Flags;
561 }
562 
563 static const Comdat *getELFComdat(const GlobalValue *GV) {
564   const Comdat *C = GV->getComdat();
565   if (!C)
566     return nullptr;
567 
568   if (C->getSelectionKind() != Comdat::Any &&
569       C->getSelectionKind() != Comdat::NoDeduplicate)
570     report_fatal_error("ELF COMDATs only support SelectionKind::Any and "
571                        "SelectionKind::NoDeduplicate, '" +
572                        C->getName() + "' cannot be lowered.");
573 
574   return C;
575 }
576 
577 static const MCSymbolELF *getLinkedToSymbol(const GlobalObject *GO,
578                                             const TargetMachine &TM) {
579   MDNode *MD = GO->getMetadata(LLVMContext::MD_associated);
580   if (!MD)
581     return nullptr;
582 
583   auto *VM = cast<ValueAsMetadata>(MD->getOperand(0).get());
584   auto *OtherGV = dyn_cast<GlobalValue>(VM->getValue());
585   return OtherGV ? dyn_cast<MCSymbolELF>(TM.getSymbol(OtherGV)) : nullptr;
586 }
587 
588 static unsigned getEntrySizeForKind(SectionKind Kind) {
589   if (Kind.isMergeable1ByteCString())
590     return 1;
591   else if (Kind.isMergeable2ByteCString())
592     return 2;
593   else if (Kind.isMergeable4ByteCString())
594     return 4;
595   else if (Kind.isMergeableConst4())
596     return 4;
597   else if (Kind.isMergeableConst8())
598     return 8;
599   else if (Kind.isMergeableConst16())
600     return 16;
601   else if (Kind.isMergeableConst32())
602     return 32;
603   else {
604     // We shouldn't have mergeable C strings or mergeable constants that we
605     // didn't handle above.
606     assert(!Kind.isMergeableCString() && "unknown string width");
607     assert(!Kind.isMergeableConst() && "unknown data width");
608     return 0;
609   }
610 }
611 
612 /// Return the section prefix name used by options FunctionsSections and
613 /// DataSections.
614 static StringRef getSectionPrefixForGlobal(SectionKind Kind, bool IsLarge) {
615   if (Kind.isText())
616     return IsLarge ? ".ltext" : ".text";
617   if (Kind.isReadOnly())
618     return IsLarge ? ".lrodata" : ".rodata";
619   if (Kind.isBSS())
620     return IsLarge ? ".lbss" : ".bss";
621   if (Kind.isThreadData())
622     return ".tdata";
623   if (Kind.isThreadBSS())
624     return ".tbss";
625   if (Kind.isData())
626     return IsLarge ? ".ldata" : ".data";
627   if (Kind.isReadOnlyWithRel())
628     return IsLarge ? ".ldata.rel.ro" : ".data.rel.ro";
629   llvm_unreachable("Unknown section kind");
630 }
631 
632 static SmallString<128>
633 getELFSectionNameForGlobal(const GlobalObject *GO, SectionKind Kind,
634                            Mangler &Mang, const TargetMachine &TM,
635                            unsigned EntrySize, bool UniqueSectionName) {
636   SmallString<128> Name =
637       getSectionPrefixForGlobal(Kind, TM.isLargeGlobalValue(GO));
638   if (Kind.isMergeableCString()) {
639     // We also need alignment here.
640     // FIXME: this is getting the alignment of the character, not the
641     // alignment of the global!
642     Align Alignment = GO->getParent()->getDataLayout().getPreferredAlign(
643         cast<GlobalVariable>(GO));
644 
645     Name += ".str";
646     Name += utostr(EntrySize);
647     Name += ".";
648     Name += utostr(Alignment.value());
649   } else if (Kind.isMergeableConst()) {
650     Name += ".cst";
651     Name += utostr(EntrySize);
652   }
653 
654   bool HasPrefix = false;
655   if (const auto *F = dyn_cast<Function>(GO)) {
656     if (std::optional<StringRef> Prefix = F->getSectionPrefix()) {
657       raw_svector_ostream(Name) << '.' << *Prefix;
658       HasPrefix = true;
659     }
660   }
661 
662   if (UniqueSectionName) {
663     Name.push_back('.');
664     TM.getNameWithPrefix(Name, GO, Mang, /*MayAlwaysUsePrivate*/true);
665   } else if (HasPrefix)
666     // For distinguishing between .text.${text-section-prefix}. (with trailing
667     // dot) and .text.${function-name}
668     Name.push_back('.');
669   return Name;
670 }
671 
672 namespace {
673 class LoweringDiagnosticInfo : public DiagnosticInfo {
674   const Twine &Msg;
675 
676 public:
677   LoweringDiagnosticInfo(const Twine &DiagMsg,
678                          DiagnosticSeverity Severity = DS_Error)
679       : DiagnosticInfo(DK_Lowering, Severity), Msg(DiagMsg) {}
680   void print(DiagnosticPrinter &DP) const override { DP << Msg; }
681 };
682 }
683 
684 /// Calculate an appropriate unique ID for a section, and update Flags,
685 /// EntrySize and NextUniqueID where appropriate.
686 static unsigned
687 calcUniqueIDUpdateFlagsAndSize(const GlobalObject *GO, StringRef SectionName,
688                                SectionKind Kind, const TargetMachine &TM,
689                                MCContext &Ctx, Mangler &Mang, unsigned &Flags,
690                                unsigned &EntrySize, unsigned &NextUniqueID,
691                                const bool Retain, const bool ForceUnique) {
692   // Increment uniqueID if we are forced to emit a unique section.
693   // This works perfectly fine with section attribute or pragma section as the
694   // sections with the same name are grouped together by the assembler.
695   if (ForceUnique)
696     return NextUniqueID++;
697 
698   // A section can have at most one associated section. Put each global with
699   // MD_associated in a unique section.
700   const bool Associated = GO->getMetadata(LLVMContext::MD_associated);
701   if (Associated) {
702     Flags |= ELF::SHF_LINK_ORDER;
703     return NextUniqueID++;
704   }
705 
706   if (Retain) {
707     if (TM.getTargetTriple().isOSSolaris())
708       Flags |= ELF::SHF_SUNW_NODISCARD;
709     else if (Ctx.getAsmInfo()->useIntegratedAssembler() ||
710              Ctx.getAsmInfo()->binutilsIsAtLeast(2, 36))
711       Flags |= ELF::SHF_GNU_RETAIN;
712     return NextUniqueID++;
713   }
714 
715   // If two symbols with differing sizes end up in the same mergeable section
716   // that section can be assigned an incorrect entry size. To avoid this we
717   // usually put symbols of the same size into distinct mergeable sections with
718   // the same name. Doing so relies on the ",unique ," assembly feature. This
719   // feature is not avalible until bintuils version 2.35
720   // (https://sourceware.org/bugzilla/show_bug.cgi?id=25380).
721   const bool SupportsUnique = Ctx.getAsmInfo()->useIntegratedAssembler() ||
722                               Ctx.getAsmInfo()->binutilsIsAtLeast(2, 35);
723   if (!SupportsUnique) {
724     Flags &= ~ELF::SHF_MERGE;
725     EntrySize = 0;
726     return MCContext::GenericSectionID;
727   }
728 
729   const bool SymbolMergeable = Flags & ELF::SHF_MERGE;
730   const bool SeenSectionNameBefore =
731       Ctx.isELFGenericMergeableSection(SectionName);
732   // If this is the first ocurrence of this section name, treat it as the
733   // generic section
734   if (!SymbolMergeable && !SeenSectionNameBefore) {
735     if (TM.getSeparateNamedSections())
736       return NextUniqueID++;
737     else
738       return MCContext::GenericSectionID;
739   }
740 
741   // Symbols must be placed into sections with compatible entry sizes. Generate
742   // unique sections for symbols that have not been assigned to compatible
743   // sections.
744   const auto PreviousID =
745       Ctx.getELFUniqueIDForEntsize(SectionName, Flags, EntrySize);
746   if (PreviousID && (!TM.getSeparateNamedSections() ||
747                      *PreviousID == MCContext::GenericSectionID))
748     return *PreviousID;
749 
750   // If the user has specified the same section name as would be created
751   // implicitly for this symbol e.g. .rodata.str1.1, then we don't need
752   // to unique the section as the entry size for this symbol will be
753   // compatible with implicitly created sections.
754   SmallString<128> ImplicitSectionNameStem =
755       getELFSectionNameForGlobal(GO, Kind, Mang, TM, EntrySize, false);
756   if (SymbolMergeable &&
757       Ctx.isELFImplicitMergeableSectionNamePrefix(SectionName) &&
758       SectionName.starts_with(ImplicitSectionNameStem))
759     return MCContext::GenericSectionID;
760 
761   // We have seen this section name before, but with different flags or entity
762   // size. Create a new unique ID.
763   return NextUniqueID++;
764 }
765 
766 static std::tuple<StringRef, bool, unsigned>
767 getGlobalObjectInfo(const GlobalObject *GO, const TargetMachine &TM) {
768   StringRef Group = "";
769   bool IsComdat = false;
770   unsigned Flags = 0;
771   if (const Comdat *C = getELFComdat(GO)) {
772     Flags |= ELF::SHF_GROUP;
773     Group = C->getName();
774     IsComdat = C->getSelectionKind() == Comdat::Any;
775   }
776   if (TM.isLargeGlobalValue(GO))
777     Flags |= ELF::SHF_X86_64_LARGE;
778   return {Group, IsComdat, Flags};
779 }
780 
781 static MCSection *selectExplicitSectionGlobal(
782     const GlobalObject *GO, SectionKind Kind, const TargetMachine &TM,
783     MCContext &Ctx, Mangler &Mang, unsigned &NextUniqueID,
784     bool Retain, bool ForceUnique) {
785   StringRef SectionName = GO->getSection();
786 
787   // Check if '#pragma clang section' name is applicable.
788   // Note that pragma directive overrides -ffunction-section, -fdata-section
789   // and so section name is exactly as user specified and not uniqued.
790   const GlobalVariable *GV = dyn_cast<GlobalVariable>(GO);
791   if (GV && GV->hasImplicitSection()) {
792     auto Attrs = GV->getAttributes();
793     if (Attrs.hasAttribute("bss-section") && Kind.isBSS()) {
794       SectionName = Attrs.getAttribute("bss-section").getValueAsString();
795     } else if (Attrs.hasAttribute("rodata-section") && Kind.isReadOnly()) {
796       SectionName = Attrs.getAttribute("rodata-section").getValueAsString();
797     } else if (Attrs.hasAttribute("relro-section") && Kind.isReadOnlyWithRel()) {
798       SectionName = Attrs.getAttribute("relro-section").getValueAsString();
799     } else if (Attrs.hasAttribute("data-section") && Kind.isData()) {
800       SectionName = Attrs.getAttribute("data-section").getValueAsString();
801     }
802   }
803 
804   // Infer section flags from the section name if we can.
805   Kind = getELFKindForNamedSection(SectionName, Kind);
806 
807   unsigned Flags = getELFSectionFlags(Kind);
808   auto [Group, IsComdat, ExtraFlags] = getGlobalObjectInfo(GO, TM);
809   Flags |= ExtraFlags;
810 
811   unsigned EntrySize = getEntrySizeForKind(Kind);
812   const unsigned UniqueID = calcUniqueIDUpdateFlagsAndSize(
813       GO, SectionName, Kind, TM, Ctx, Mang, Flags, EntrySize, NextUniqueID,
814       Retain, ForceUnique);
815 
816   const MCSymbolELF *LinkedToSym = getLinkedToSymbol(GO, TM);
817   MCSectionELF *Section = Ctx.getELFSection(
818       SectionName, getELFSectionType(SectionName, Kind), Flags, EntrySize,
819       Group, IsComdat, UniqueID, LinkedToSym);
820   // Make sure that we did not get some other section with incompatible sh_link.
821   // This should not be possible due to UniqueID code above.
822   assert(Section->getLinkedToSymbol() == LinkedToSym &&
823          "Associated symbol mismatch between sections");
824 
825   if (!(Ctx.getAsmInfo()->useIntegratedAssembler() ||
826         Ctx.getAsmInfo()->binutilsIsAtLeast(2, 35))) {
827     // If we are using GNU as before 2.35, then this symbol might have
828     // been placed in an incompatible mergeable section. Emit an error if this
829     // is the case to avoid creating broken output.
830     if ((Section->getFlags() & ELF::SHF_MERGE) &&
831         (Section->getEntrySize() != getEntrySizeForKind(Kind)))
832       GO->getContext().diagnose(LoweringDiagnosticInfo(
833           "Symbol '" + GO->getName() + "' from module '" +
834           (GO->getParent() ? GO->getParent()->getSourceFileName() : "unknown") +
835           "' required a section with entry-size=" +
836           Twine(getEntrySizeForKind(Kind)) + " but was placed in section '" +
837           SectionName + "' with entry-size=" + Twine(Section->getEntrySize()) +
838           ": Explicit assignment by pragma or attribute of an incompatible "
839           "symbol to this section?"));
840   }
841 
842   return Section;
843 }
844 
845 MCSection *TargetLoweringObjectFileELF::getExplicitSectionGlobal(
846     const GlobalObject *GO, SectionKind Kind, const TargetMachine &TM) const {
847   return selectExplicitSectionGlobal(GO, Kind, TM, getContext(), getMangler(),
848                                      NextUniqueID, Used.count(GO),
849                                      /* ForceUnique = */false);
850 }
851 
852 static MCSectionELF *selectELFSectionForGlobal(
853     MCContext &Ctx, const GlobalObject *GO, SectionKind Kind, Mangler &Mang,
854     const TargetMachine &TM, bool EmitUniqueSection, unsigned Flags,
855     unsigned *NextUniqueID, const MCSymbolELF *AssociatedSymbol) {
856 
857   auto [Group, IsComdat, ExtraFlags] = getGlobalObjectInfo(GO, TM);
858   Flags |= ExtraFlags;
859 
860   // Get the section entry size based on the kind.
861   unsigned EntrySize = getEntrySizeForKind(Kind);
862 
863   bool UniqueSectionName = false;
864   unsigned UniqueID = MCContext::GenericSectionID;
865   if (EmitUniqueSection) {
866     if (TM.getUniqueSectionNames()) {
867       UniqueSectionName = true;
868     } else {
869       UniqueID = *NextUniqueID;
870       (*NextUniqueID)++;
871     }
872   }
873   SmallString<128> Name = getELFSectionNameForGlobal(
874       GO, Kind, Mang, TM, EntrySize, UniqueSectionName);
875 
876   // Use 0 as the unique ID for execute-only text.
877   if (Kind.isExecuteOnly())
878     UniqueID = 0;
879   return Ctx.getELFSection(Name, getELFSectionType(Name, Kind), Flags,
880                            EntrySize, Group, IsComdat, UniqueID,
881                            AssociatedSymbol);
882 }
883 
884 static MCSection *selectELFSectionForGlobal(
885     MCContext &Ctx, const GlobalObject *GO, SectionKind Kind, Mangler &Mang,
886     const TargetMachine &TM, bool Retain, bool EmitUniqueSection,
887     unsigned Flags, unsigned *NextUniqueID) {
888   const MCSymbolELF *LinkedToSym = getLinkedToSymbol(GO, TM);
889   if (LinkedToSym) {
890     EmitUniqueSection = true;
891     Flags |= ELF::SHF_LINK_ORDER;
892   }
893   if (Retain) {
894     if (TM.getTargetTriple().isOSSolaris()) {
895       EmitUniqueSection = true;
896       Flags |= ELF::SHF_SUNW_NODISCARD;
897     } else if (Ctx.getAsmInfo()->useIntegratedAssembler() ||
898                Ctx.getAsmInfo()->binutilsIsAtLeast(2, 36)) {
899       EmitUniqueSection = true;
900       Flags |= ELF::SHF_GNU_RETAIN;
901     }
902   }
903 
904   MCSectionELF *Section = selectELFSectionForGlobal(
905       Ctx, GO, Kind, Mang, TM, EmitUniqueSection, Flags,
906       NextUniqueID, LinkedToSym);
907   assert(Section->getLinkedToSymbol() == LinkedToSym);
908   return Section;
909 }
910 
911 MCSection *TargetLoweringObjectFileELF::SelectSectionForGlobal(
912     const GlobalObject *GO, SectionKind Kind, const TargetMachine &TM) const {
913   unsigned Flags = getELFSectionFlags(Kind);
914 
915   // If we have -ffunction-section or -fdata-section then we should emit the
916   // global value to a uniqued section specifically for it.
917   bool EmitUniqueSection = false;
918   if (!(Flags & ELF::SHF_MERGE) && !Kind.isCommon()) {
919     if (Kind.isText())
920       EmitUniqueSection = TM.getFunctionSections();
921     else
922       EmitUniqueSection = TM.getDataSections();
923   }
924   EmitUniqueSection |= GO->hasComdat();
925   return selectELFSectionForGlobal(getContext(), GO, Kind, getMangler(), TM,
926                                    Used.count(GO), EmitUniqueSection, Flags,
927                                    &NextUniqueID);
928 }
929 
930 MCSection *TargetLoweringObjectFileELF::getUniqueSectionForFunction(
931     const Function &F, const TargetMachine &TM) const {
932   SectionKind Kind = SectionKind::getText();
933   unsigned Flags = getELFSectionFlags(Kind);
934   // If the function's section names is pre-determined via pragma or a
935   // section attribute, call selectExplicitSectionGlobal.
936   if (F.hasSection())
937     return selectExplicitSectionGlobal(
938         &F, Kind, TM, getContext(), getMangler(), NextUniqueID,
939         Used.count(&F), /* ForceUnique = */true);
940   else
941     return selectELFSectionForGlobal(
942         getContext(), &F, Kind, getMangler(), TM, Used.count(&F),
943         /*EmitUniqueSection=*/true, Flags, &NextUniqueID);
944 }
945 
946 MCSection *TargetLoweringObjectFileELF::getSectionForJumpTable(
947     const Function &F, const TargetMachine &TM) const {
948   // If the function can be removed, produce a unique section so that
949   // the table doesn't prevent the removal.
950   const Comdat *C = F.getComdat();
951   bool EmitUniqueSection = TM.getFunctionSections() || C;
952   if (!EmitUniqueSection)
953     return ReadOnlySection;
954 
955   return selectELFSectionForGlobal(getContext(), &F, SectionKind::getReadOnly(),
956                                    getMangler(), TM, EmitUniqueSection,
957                                    ELF::SHF_ALLOC, &NextUniqueID,
958                                    /* AssociatedSymbol */ nullptr);
959 }
960 
961 MCSection *TargetLoweringObjectFileELF::getSectionForLSDA(
962     const Function &F, const MCSymbol &FnSym, const TargetMachine &TM) const {
963   // If neither COMDAT nor function sections, use the monolithic LSDA section.
964   // Re-use this path if LSDASection is null as in the Arm EHABI.
965   if (!LSDASection || (!F.hasComdat() && !TM.getFunctionSections()))
966     return LSDASection;
967 
968   const auto *LSDA = cast<MCSectionELF>(LSDASection);
969   unsigned Flags = LSDA->getFlags();
970   const MCSymbolELF *LinkedToSym = nullptr;
971   StringRef Group;
972   bool IsComdat = false;
973   if (const Comdat *C = getELFComdat(&F)) {
974     Flags |= ELF::SHF_GROUP;
975     Group = C->getName();
976     IsComdat = C->getSelectionKind() == Comdat::Any;
977   }
978   // Use SHF_LINK_ORDER to facilitate --gc-sections if we can use GNU ld>=2.36
979   // or LLD, which support mixed SHF_LINK_ORDER & non-SHF_LINK_ORDER.
980   if (TM.getFunctionSections() &&
981       (getContext().getAsmInfo()->useIntegratedAssembler() &&
982        getContext().getAsmInfo()->binutilsIsAtLeast(2, 36))) {
983     Flags |= ELF::SHF_LINK_ORDER;
984     LinkedToSym = cast<MCSymbolELF>(&FnSym);
985   }
986 
987   // Append the function name as the suffix like GCC, assuming
988   // -funique-section-names applies to .gcc_except_table sections.
989   return getContext().getELFSection(
990       (TM.getUniqueSectionNames() ? LSDA->getName() + "." + F.getName()
991                                   : LSDA->getName()),
992       LSDA->getType(), Flags, 0, Group, IsComdat, MCSection::NonUniqueID,
993       LinkedToSym);
994 }
995 
996 bool TargetLoweringObjectFileELF::shouldPutJumpTableInFunctionSection(
997     bool UsesLabelDifference, const Function &F) const {
998   // We can always create relative relocations, so use another section
999   // that can be marked non-executable.
1000   return false;
1001 }
1002 
1003 /// Given a mergeable constant with the specified size and relocation
1004 /// information, return a section that it should be placed in.
1005 MCSection *TargetLoweringObjectFileELF::getSectionForConstant(
1006     const DataLayout &DL, SectionKind Kind, const Constant *C,
1007     Align &Alignment) const {
1008   if (Kind.isMergeableConst4() && MergeableConst4Section)
1009     return MergeableConst4Section;
1010   if (Kind.isMergeableConst8() && MergeableConst8Section)
1011     return MergeableConst8Section;
1012   if (Kind.isMergeableConst16() && MergeableConst16Section)
1013     return MergeableConst16Section;
1014   if (Kind.isMergeableConst32() && MergeableConst32Section)
1015     return MergeableConst32Section;
1016   if (Kind.isReadOnly())
1017     return ReadOnlySection;
1018 
1019   assert(Kind.isReadOnlyWithRel() && "Unknown section kind");
1020   return DataRelROSection;
1021 }
1022 
1023 /// Returns a unique section for the given machine basic block.
1024 MCSection *TargetLoweringObjectFileELF::getSectionForMachineBasicBlock(
1025     const Function &F, const MachineBasicBlock &MBB,
1026     const TargetMachine &TM) const {
1027   assert(MBB.isBeginSection() && "Basic block does not start a section!");
1028   unsigned UniqueID = MCContext::GenericSectionID;
1029 
1030   // For cold sections use the .text.split. prefix along with the parent
1031   // function name. All cold blocks for the same function go to the same
1032   // section. Similarly all exception blocks are grouped by symbol name
1033   // under the .text.eh prefix. For regular sections, we either use a unique
1034   // name, or a unique ID for the section.
1035   SmallString<128> Name;
1036   StringRef FunctionSectionName = MBB.getParent()->getSection()->getName();
1037   if (FunctionSectionName == ".text" ||
1038       FunctionSectionName.starts_with(".text.")) {
1039     // Function is in a regular .text section.
1040     StringRef FunctionName = MBB.getParent()->getName();
1041     if (MBB.getSectionID() == MBBSectionID::ColdSectionID) {
1042       Name += BBSectionsColdTextPrefix;
1043       Name += FunctionName;
1044     } else if (MBB.getSectionID() == MBBSectionID::ExceptionSectionID) {
1045       Name += ".text.eh.";
1046       Name += FunctionName;
1047     } else {
1048       Name += FunctionSectionName;
1049       if (TM.getUniqueBasicBlockSectionNames()) {
1050         if (!Name.ends_with("."))
1051           Name += ".";
1052         Name += MBB.getSymbol()->getName();
1053       } else {
1054         UniqueID = NextUniqueID++;
1055       }
1056     }
1057   } else {
1058     // If the original function has a custom non-dot-text section, then emit
1059     // all basic block sections into that section too, each with a unique id.
1060     Name = FunctionSectionName;
1061     UniqueID = NextUniqueID++;
1062   }
1063 
1064   unsigned Flags = ELF::SHF_ALLOC | ELF::SHF_EXECINSTR;
1065   std::string GroupName;
1066   if (F.hasComdat()) {
1067     Flags |= ELF::SHF_GROUP;
1068     GroupName = F.getComdat()->getName().str();
1069   }
1070   return getContext().getELFSection(Name, ELF::SHT_PROGBITS, Flags,
1071                                     0 /* Entry Size */, GroupName,
1072                                     F.hasComdat(), UniqueID, nullptr);
1073 }
1074 
1075 static MCSectionELF *getStaticStructorSection(MCContext &Ctx, bool UseInitArray,
1076                                               bool IsCtor, unsigned Priority,
1077                                               const MCSymbol *KeySym) {
1078   std::string Name;
1079   unsigned Type;
1080   unsigned Flags = ELF::SHF_ALLOC | ELF::SHF_WRITE;
1081   StringRef Comdat = KeySym ? KeySym->getName() : "";
1082 
1083   if (KeySym)
1084     Flags |= ELF::SHF_GROUP;
1085 
1086   if (UseInitArray) {
1087     if (IsCtor) {
1088       Type = ELF::SHT_INIT_ARRAY;
1089       Name = ".init_array";
1090     } else {
1091       Type = ELF::SHT_FINI_ARRAY;
1092       Name = ".fini_array";
1093     }
1094     if (Priority != 65535) {
1095       Name += '.';
1096       Name += utostr(Priority);
1097     }
1098   } else {
1099     // The default scheme is .ctor / .dtor, so we have to invert the priority
1100     // numbering.
1101     if (IsCtor)
1102       Name = ".ctors";
1103     else
1104       Name = ".dtors";
1105     if (Priority != 65535)
1106       raw_string_ostream(Name) << format(".%05u", 65535 - Priority);
1107     Type = ELF::SHT_PROGBITS;
1108   }
1109 
1110   return Ctx.getELFSection(Name, Type, Flags, 0, Comdat, /*IsComdat=*/true);
1111 }
1112 
1113 MCSection *TargetLoweringObjectFileELF::getStaticCtorSection(
1114     unsigned Priority, const MCSymbol *KeySym) const {
1115   return getStaticStructorSection(getContext(), UseInitArray, true, Priority,
1116                                   KeySym);
1117 }
1118 
1119 MCSection *TargetLoweringObjectFileELF::getStaticDtorSection(
1120     unsigned Priority, const MCSymbol *KeySym) const {
1121   return getStaticStructorSection(getContext(), UseInitArray, false, Priority,
1122                                   KeySym);
1123 }
1124 
1125 const MCExpr *TargetLoweringObjectFileELF::lowerRelativeReference(
1126     const GlobalValue *LHS, const GlobalValue *RHS,
1127     const TargetMachine &TM) const {
1128   // We may only use a PLT-relative relocation to refer to unnamed_addr
1129   // functions.
1130   if (!LHS->hasGlobalUnnamedAddr() || !LHS->getValueType()->isFunctionTy())
1131     return nullptr;
1132 
1133   // Basic correctness checks.
1134   if (LHS->getType()->getPointerAddressSpace() != 0 ||
1135       RHS->getType()->getPointerAddressSpace() != 0 || LHS->isThreadLocal() ||
1136       RHS->isThreadLocal())
1137     return nullptr;
1138 
1139   return MCBinaryExpr::createSub(
1140       MCSymbolRefExpr::create(TM.getSymbol(LHS), PLTRelativeVariantKind,
1141                               getContext()),
1142       MCSymbolRefExpr::create(TM.getSymbol(RHS), getContext()), getContext());
1143 }
1144 
1145 const MCExpr *TargetLoweringObjectFileELF::lowerDSOLocalEquivalent(
1146     const DSOLocalEquivalent *Equiv, const TargetMachine &TM) const {
1147   assert(supportDSOLocalEquivalentLowering());
1148 
1149   const auto *GV = Equiv->getGlobalValue();
1150 
1151   // A PLT entry is not needed for dso_local globals.
1152   if (GV->isDSOLocal() || GV->isImplicitDSOLocal())
1153     return MCSymbolRefExpr::create(TM.getSymbol(GV), getContext());
1154 
1155   return MCSymbolRefExpr::create(TM.getSymbol(GV), PLTRelativeVariantKind,
1156                                  getContext());
1157 }
1158 
1159 MCSection *TargetLoweringObjectFileELF::getSectionForCommandLines() const {
1160   // Use ".GCC.command.line" since this feature is to support clang's
1161   // -frecord-gcc-switches which in turn attempts to mimic GCC's switch of the
1162   // same name.
1163   return getContext().getELFSection(".GCC.command.line", ELF::SHT_PROGBITS,
1164                                     ELF::SHF_MERGE | ELF::SHF_STRINGS, 1);
1165 }
1166 
1167 void
1168 TargetLoweringObjectFileELF::InitializeELF(bool UseInitArray_) {
1169   UseInitArray = UseInitArray_;
1170   MCContext &Ctx = getContext();
1171   if (!UseInitArray) {
1172     StaticCtorSection = Ctx.getELFSection(".ctors", ELF::SHT_PROGBITS,
1173                                           ELF::SHF_ALLOC | ELF::SHF_WRITE);
1174 
1175     StaticDtorSection = Ctx.getELFSection(".dtors", ELF::SHT_PROGBITS,
1176                                           ELF::SHF_ALLOC | ELF::SHF_WRITE);
1177     return;
1178   }
1179 
1180   StaticCtorSection = Ctx.getELFSection(".init_array", ELF::SHT_INIT_ARRAY,
1181                                         ELF::SHF_WRITE | ELF::SHF_ALLOC);
1182   StaticDtorSection = Ctx.getELFSection(".fini_array", ELF::SHT_FINI_ARRAY,
1183                                         ELF::SHF_WRITE | ELF::SHF_ALLOC);
1184 }
1185 
1186 //===----------------------------------------------------------------------===//
1187 //                                 MachO
1188 //===----------------------------------------------------------------------===//
1189 
1190 TargetLoweringObjectFileMachO::TargetLoweringObjectFileMachO() {
1191   SupportIndirectSymViaGOTPCRel = true;
1192 }
1193 
1194 void TargetLoweringObjectFileMachO::Initialize(MCContext &Ctx,
1195                                                const TargetMachine &TM) {
1196   TargetLoweringObjectFile::Initialize(Ctx, TM);
1197   if (TM.getRelocationModel() == Reloc::Static) {
1198     StaticCtorSection = Ctx.getMachOSection("__TEXT", "__constructor", 0,
1199                                             SectionKind::getData());
1200     StaticDtorSection = Ctx.getMachOSection("__TEXT", "__destructor", 0,
1201                                             SectionKind::getData());
1202   } else {
1203     StaticCtorSection = Ctx.getMachOSection("__DATA", "__mod_init_func",
1204                                             MachO::S_MOD_INIT_FUNC_POINTERS,
1205                                             SectionKind::getData());
1206     StaticDtorSection = Ctx.getMachOSection("__DATA", "__mod_term_func",
1207                                             MachO::S_MOD_TERM_FUNC_POINTERS,
1208                                             SectionKind::getData());
1209   }
1210 
1211   PersonalityEncoding =
1212       dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel | dwarf::DW_EH_PE_sdata4;
1213   LSDAEncoding = dwarf::DW_EH_PE_pcrel;
1214   TTypeEncoding =
1215       dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_pcrel | dwarf::DW_EH_PE_sdata4;
1216 }
1217 
1218 MCSection *TargetLoweringObjectFileMachO::getStaticDtorSection(
1219     unsigned Priority, const MCSymbol *KeySym) const {
1220   return StaticDtorSection;
1221   // In userspace, we lower global destructors via atexit(), but kernel/kext
1222   // environments do not provide this function so we still need to support the
1223   // legacy way here.
1224   // See the -disable-atexit-based-global-dtor-lowering CodeGen flag for more
1225   // context.
1226 }
1227 
1228 void TargetLoweringObjectFileMachO::emitModuleMetadata(MCStreamer &Streamer,
1229                                                        Module &M) const {
1230   // Emit the linker options if present.
1231   if (auto *LinkerOptions = M.getNamedMetadata("llvm.linker.options")) {
1232     for (const auto *Option : LinkerOptions->operands()) {
1233       SmallVector<std::string, 4> StrOptions;
1234       for (const auto &Piece : cast<MDNode>(Option)->operands())
1235         StrOptions.push_back(std::string(cast<MDString>(Piece)->getString()));
1236       Streamer.emitLinkerOptions(StrOptions);
1237     }
1238   }
1239 
1240   unsigned VersionVal = 0;
1241   unsigned ImageInfoFlags = 0;
1242   StringRef SectionVal;
1243 
1244   GetObjCImageInfo(M, VersionVal, ImageInfoFlags, SectionVal);
1245   emitCGProfileMetadata(Streamer, M);
1246 
1247   // The section is mandatory. If we don't have it, then we don't have GC info.
1248   if (SectionVal.empty())
1249     return;
1250 
1251   StringRef Segment, Section;
1252   unsigned TAA = 0, StubSize = 0;
1253   bool TAAParsed;
1254   if (Error E = MCSectionMachO::ParseSectionSpecifier(
1255           SectionVal, Segment, Section, TAA, TAAParsed, StubSize)) {
1256     // If invalid, report the error with report_fatal_error.
1257     report_fatal_error("Invalid section specifier '" + Section +
1258                        "': " + toString(std::move(E)) + ".");
1259   }
1260 
1261   // Get the section.
1262   MCSectionMachO *S = getContext().getMachOSection(
1263       Segment, Section, TAA, StubSize, SectionKind::getData());
1264   Streamer.switchSection(S);
1265   Streamer.emitLabel(getContext().
1266                      getOrCreateSymbol(StringRef("L_OBJC_IMAGE_INFO")));
1267   Streamer.emitInt32(VersionVal);
1268   Streamer.emitInt32(ImageInfoFlags);
1269   Streamer.addBlankLine();
1270 }
1271 
1272 static void checkMachOComdat(const GlobalValue *GV) {
1273   const Comdat *C = GV->getComdat();
1274   if (!C)
1275     return;
1276 
1277   report_fatal_error("MachO doesn't support COMDATs, '" + C->getName() +
1278                      "' cannot be lowered.");
1279 }
1280 
1281 MCSection *TargetLoweringObjectFileMachO::getExplicitSectionGlobal(
1282     const GlobalObject *GO, SectionKind Kind, const TargetMachine &TM) const {
1283 
1284   StringRef SectionName = GO->getSection();
1285 
1286   const GlobalVariable *GV = dyn_cast<GlobalVariable>(GO);
1287   if (GV && GV->hasImplicitSection()) {
1288     auto Attrs = GV->getAttributes();
1289     if (Attrs.hasAttribute("bss-section") && Kind.isBSS()) {
1290       SectionName = Attrs.getAttribute("bss-section").getValueAsString();
1291     } else if (Attrs.hasAttribute("rodata-section") && Kind.isReadOnly()) {
1292       SectionName = Attrs.getAttribute("rodata-section").getValueAsString();
1293     } else if (Attrs.hasAttribute("relro-section") && Kind.isReadOnlyWithRel()) {
1294       SectionName = Attrs.getAttribute("relro-section").getValueAsString();
1295     } else if (Attrs.hasAttribute("data-section") && Kind.isData()) {
1296       SectionName = Attrs.getAttribute("data-section").getValueAsString();
1297     }
1298   }
1299 
1300   // Parse the section specifier and create it if valid.
1301   StringRef Segment, Section;
1302   unsigned TAA = 0, StubSize = 0;
1303   bool TAAParsed;
1304 
1305   checkMachOComdat(GO);
1306 
1307   if (Error E = MCSectionMachO::ParseSectionSpecifier(
1308           SectionName, Segment, Section, TAA, TAAParsed, StubSize)) {
1309     // If invalid, report the error with report_fatal_error.
1310     report_fatal_error("Global variable '" + GO->getName() +
1311                        "' has an invalid section specifier '" +
1312                        GO->getSection() + "': " + toString(std::move(E)) + ".");
1313   }
1314 
1315   // Get the section.
1316   MCSectionMachO *S =
1317       getContext().getMachOSection(Segment, Section, TAA, StubSize, Kind);
1318 
1319   // If TAA wasn't set by ParseSectionSpecifier() above,
1320   // use the value returned by getMachOSection() as a default.
1321   if (!TAAParsed)
1322     TAA = S->getTypeAndAttributes();
1323 
1324   // Okay, now that we got the section, verify that the TAA & StubSize agree.
1325   // If the user declared multiple globals with different section flags, we need
1326   // to reject it here.
1327   if (S->getTypeAndAttributes() != TAA || S->getStubSize() != StubSize) {
1328     // If invalid, report the error with report_fatal_error.
1329     report_fatal_error("Global variable '" + GO->getName() +
1330                        "' section type or attributes does not match previous"
1331                        " section specifier");
1332   }
1333 
1334   return S;
1335 }
1336 
1337 MCSection *TargetLoweringObjectFileMachO::SelectSectionForGlobal(
1338     const GlobalObject *GO, SectionKind Kind, const TargetMachine &TM) const {
1339   checkMachOComdat(GO);
1340 
1341   // Handle thread local data.
1342   if (Kind.isThreadBSS()) return TLSBSSSection;
1343   if (Kind.isThreadData()) return TLSDataSection;
1344 
1345   if (Kind.isText())
1346     return GO->isWeakForLinker() ? TextCoalSection : TextSection;
1347 
1348   // If this is weak/linkonce, put this in a coalescable section, either in text
1349   // or data depending on if it is writable.
1350   if (GO->isWeakForLinker()) {
1351     if (Kind.isReadOnly())
1352       return ConstTextCoalSection;
1353     if (Kind.isReadOnlyWithRel())
1354       return ConstDataCoalSection;
1355     return DataCoalSection;
1356   }
1357 
1358   // FIXME: Alignment check should be handled by section classifier.
1359   if (Kind.isMergeable1ByteCString() &&
1360       GO->getParent()->getDataLayout().getPreferredAlign(
1361           cast<GlobalVariable>(GO)) < Align(32))
1362     return CStringSection;
1363 
1364   // Do not put 16-bit arrays in the UString section if they have an
1365   // externally visible label, this runs into issues with certain linker
1366   // versions.
1367   if (Kind.isMergeable2ByteCString() && !GO->hasExternalLinkage() &&
1368       GO->getParent()->getDataLayout().getPreferredAlign(
1369           cast<GlobalVariable>(GO)) < Align(32))
1370     return UStringSection;
1371 
1372   // With MachO only variables whose corresponding symbol starts with 'l' or
1373   // 'L' can be merged, so we only try merging GVs with private linkage.
1374   if (GO->hasPrivateLinkage() && Kind.isMergeableConst()) {
1375     if (Kind.isMergeableConst4())
1376       return FourByteConstantSection;
1377     if (Kind.isMergeableConst8())
1378       return EightByteConstantSection;
1379     if (Kind.isMergeableConst16())
1380       return SixteenByteConstantSection;
1381   }
1382 
1383   // Otherwise, if it is readonly, but not something we can specially optimize,
1384   // just drop it in .const.
1385   if (Kind.isReadOnly())
1386     return ReadOnlySection;
1387 
1388   // If this is marked const, put it into a const section.  But if the dynamic
1389   // linker needs to write to it, put it in the data segment.
1390   if (Kind.isReadOnlyWithRel())
1391     return ConstDataSection;
1392 
1393   // Put zero initialized globals with strong external linkage in the
1394   // DATA, __common section with the .zerofill directive.
1395   if (Kind.isBSSExtern())
1396     return DataCommonSection;
1397 
1398   // Put zero initialized globals with local linkage in __DATA,__bss directive
1399   // with the .zerofill directive (aka .lcomm).
1400   if (Kind.isBSSLocal())
1401     return DataBSSSection;
1402 
1403   // Otherwise, just drop the variable in the normal data section.
1404   return DataSection;
1405 }
1406 
1407 MCSection *TargetLoweringObjectFileMachO::getSectionForConstant(
1408     const DataLayout &DL, SectionKind Kind, const Constant *C,
1409     Align &Alignment) const {
1410   // If this constant requires a relocation, we have to put it in the data
1411   // segment, not in the text segment.
1412   if (Kind.isData() || Kind.isReadOnlyWithRel())
1413     return ConstDataSection;
1414 
1415   if (Kind.isMergeableConst4())
1416     return FourByteConstantSection;
1417   if (Kind.isMergeableConst8())
1418     return EightByteConstantSection;
1419   if (Kind.isMergeableConst16())
1420     return SixteenByteConstantSection;
1421   return ReadOnlySection;  // .const
1422 }
1423 
1424 MCSection *TargetLoweringObjectFileMachO::getSectionForCommandLines() const {
1425   return getContext().getMachOSection("__TEXT", "__command_line", 0,
1426                                       SectionKind::getReadOnly());
1427 }
1428 
1429 const MCExpr *TargetLoweringObjectFileMachO::getTTypeGlobalReference(
1430     const GlobalValue *GV, unsigned Encoding, const TargetMachine &TM,
1431     MachineModuleInfo *MMI, MCStreamer &Streamer) const {
1432   // The mach-o version of this method defaults to returning a stub reference.
1433 
1434   if (Encoding & DW_EH_PE_indirect) {
1435     MachineModuleInfoMachO &MachOMMI =
1436       MMI->getObjFileInfo<MachineModuleInfoMachO>();
1437 
1438     MCSymbol *SSym = getSymbolWithGlobalValueBase(GV, "$non_lazy_ptr", TM);
1439 
1440     // Add information about the stub reference to MachOMMI so that the stub
1441     // gets emitted by the asmprinter.
1442     MachineModuleInfoImpl::StubValueTy &StubSym = MachOMMI.getGVStubEntry(SSym);
1443     if (!StubSym.getPointer()) {
1444       MCSymbol *Sym = TM.getSymbol(GV);
1445       StubSym = MachineModuleInfoImpl::StubValueTy(Sym, !GV->hasLocalLinkage());
1446     }
1447 
1448     return TargetLoweringObjectFile::
1449       getTTypeReference(MCSymbolRefExpr::create(SSym, getContext()),
1450                         Encoding & ~DW_EH_PE_indirect, Streamer);
1451   }
1452 
1453   return TargetLoweringObjectFile::getTTypeGlobalReference(GV, Encoding, TM,
1454                                                            MMI, Streamer);
1455 }
1456 
1457 MCSymbol *TargetLoweringObjectFileMachO::getCFIPersonalitySymbol(
1458     const GlobalValue *GV, const TargetMachine &TM,
1459     MachineModuleInfo *MMI) const {
1460   // The mach-o version of this method defaults to returning a stub reference.
1461   MachineModuleInfoMachO &MachOMMI =
1462     MMI->getObjFileInfo<MachineModuleInfoMachO>();
1463 
1464   MCSymbol *SSym = getSymbolWithGlobalValueBase(GV, "$non_lazy_ptr", TM);
1465 
1466   // Add information about the stub reference to MachOMMI so that the stub
1467   // gets emitted by the asmprinter.
1468   MachineModuleInfoImpl::StubValueTy &StubSym = MachOMMI.getGVStubEntry(SSym);
1469   if (!StubSym.getPointer()) {
1470     MCSymbol *Sym = TM.getSymbol(GV);
1471     StubSym = MachineModuleInfoImpl::StubValueTy(Sym, !GV->hasLocalLinkage());
1472   }
1473 
1474   return SSym;
1475 }
1476 
1477 const MCExpr *TargetLoweringObjectFileMachO::getIndirectSymViaGOTPCRel(
1478     const GlobalValue *GV, const MCSymbol *Sym, const MCValue &MV,
1479     int64_t Offset, MachineModuleInfo *MMI, MCStreamer &Streamer) const {
1480   // Although MachO 32-bit targets do not explicitly have a GOTPCREL relocation
1481   // as 64-bit do, we replace the GOT equivalent by accessing the final symbol
1482   // through a non_lazy_ptr stub instead. One advantage is that it allows the
1483   // computation of deltas to final external symbols. Example:
1484   //
1485   //    _extgotequiv:
1486   //       .long   _extfoo
1487   //
1488   //    _delta:
1489   //       .long   _extgotequiv-_delta
1490   //
1491   // is transformed to:
1492   //
1493   //    _delta:
1494   //       .long   L_extfoo$non_lazy_ptr-(_delta+0)
1495   //
1496   //       .section        __IMPORT,__pointers,non_lazy_symbol_pointers
1497   //    L_extfoo$non_lazy_ptr:
1498   //       .indirect_symbol        _extfoo
1499   //       .long   0
1500   //
1501   // The indirect symbol table (and sections of non_lazy_symbol_pointers type)
1502   // may point to both local (same translation unit) and global (other
1503   // translation units) symbols. Example:
1504   //
1505   // .section __DATA,__pointers,non_lazy_symbol_pointers
1506   // L1:
1507   //    .indirect_symbol _myGlobal
1508   //    .long 0
1509   // L2:
1510   //    .indirect_symbol _myLocal
1511   //    .long _myLocal
1512   //
1513   // If the symbol is local, instead of the symbol's index, the assembler
1514   // places the constant INDIRECT_SYMBOL_LOCAL into the indirect symbol table.
1515   // Then the linker will notice the constant in the table and will look at the
1516   // content of the symbol.
1517   MachineModuleInfoMachO &MachOMMI =
1518     MMI->getObjFileInfo<MachineModuleInfoMachO>();
1519   MCContext &Ctx = getContext();
1520 
1521   // The offset must consider the original displacement from the base symbol
1522   // since 32-bit targets don't have a GOTPCREL to fold the PC displacement.
1523   Offset = -MV.getConstant();
1524   const MCSymbol *BaseSym = &MV.getSymB()->getSymbol();
1525 
1526   // Access the final symbol via sym$non_lazy_ptr and generate the appropriated
1527   // non_lazy_ptr stubs.
1528   SmallString<128> Name;
1529   StringRef Suffix = "$non_lazy_ptr";
1530   Name += MMI->getModule()->getDataLayout().getPrivateGlobalPrefix();
1531   Name += Sym->getName();
1532   Name += Suffix;
1533   MCSymbol *Stub = Ctx.getOrCreateSymbol(Name);
1534 
1535   MachineModuleInfoImpl::StubValueTy &StubSym = MachOMMI.getGVStubEntry(Stub);
1536 
1537   if (!StubSym.getPointer())
1538     StubSym = MachineModuleInfoImpl::StubValueTy(const_cast<MCSymbol *>(Sym),
1539                                                  !GV->hasLocalLinkage());
1540 
1541   const MCExpr *BSymExpr =
1542     MCSymbolRefExpr::create(BaseSym, MCSymbolRefExpr::VK_None, Ctx);
1543   const MCExpr *LHS =
1544     MCSymbolRefExpr::create(Stub, MCSymbolRefExpr::VK_None, Ctx);
1545 
1546   if (!Offset)
1547     return MCBinaryExpr::createSub(LHS, BSymExpr, Ctx);
1548 
1549   const MCExpr *RHS =
1550     MCBinaryExpr::createAdd(BSymExpr, MCConstantExpr::create(Offset, Ctx), Ctx);
1551   return MCBinaryExpr::createSub(LHS, RHS, Ctx);
1552 }
1553 
1554 static bool canUsePrivateLabel(const MCAsmInfo &AsmInfo,
1555                                const MCSection &Section) {
1556   if (!AsmInfo.isSectionAtomizableBySymbols(Section))
1557     return true;
1558 
1559   // FIXME: we should be able to use private labels for sections that can't be
1560   // dead-stripped (there's no issue with blocking atomization there), but `ld
1561   // -r` sometimes drops the no_dead_strip attribute from sections so for safety
1562   // we don't allow it.
1563   return false;
1564 }
1565 
1566 void TargetLoweringObjectFileMachO::getNameWithPrefix(
1567     SmallVectorImpl<char> &OutName, const GlobalValue *GV,
1568     const TargetMachine &TM) const {
1569   bool CannotUsePrivateLabel = true;
1570   if (auto *GO = GV->getAliaseeObject()) {
1571     SectionKind GOKind = TargetLoweringObjectFile::getKindForGlobal(GO, TM);
1572     const MCSection *TheSection = SectionForGlobal(GO, GOKind, TM);
1573     CannotUsePrivateLabel =
1574         !canUsePrivateLabel(*TM.getMCAsmInfo(), *TheSection);
1575   }
1576   getMangler().getNameWithPrefix(OutName, GV, CannotUsePrivateLabel);
1577 }
1578 
1579 //===----------------------------------------------------------------------===//
1580 //                                  COFF
1581 //===----------------------------------------------------------------------===//
1582 
1583 static unsigned
1584 getCOFFSectionFlags(SectionKind K, const TargetMachine &TM) {
1585   unsigned Flags = 0;
1586   bool isThumb = TM.getTargetTriple().getArch() == Triple::thumb;
1587 
1588   if (K.isMetadata())
1589     Flags |=
1590       COFF::IMAGE_SCN_MEM_DISCARDABLE;
1591   else if (K.isExclude())
1592     Flags |=
1593       COFF::IMAGE_SCN_LNK_REMOVE | COFF::IMAGE_SCN_MEM_DISCARDABLE;
1594   else if (K.isText())
1595     Flags |=
1596       COFF::IMAGE_SCN_MEM_EXECUTE |
1597       COFF::IMAGE_SCN_MEM_READ |
1598       COFF::IMAGE_SCN_CNT_CODE |
1599       (isThumb ? COFF::IMAGE_SCN_MEM_16BIT : (COFF::SectionCharacteristics)0);
1600   else if (K.isBSS())
1601     Flags |=
1602       COFF::IMAGE_SCN_CNT_UNINITIALIZED_DATA |
1603       COFF::IMAGE_SCN_MEM_READ |
1604       COFF::IMAGE_SCN_MEM_WRITE;
1605   else if (K.isThreadLocal())
1606     Flags |=
1607       COFF::IMAGE_SCN_CNT_INITIALIZED_DATA |
1608       COFF::IMAGE_SCN_MEM_READ |
1609       COFF::IMAGE_SCN_MEM_WRITE;
1610   else if (K.isReadOnly() || K.isReadOnlyWithRel())
1611     Flags |=
1612       COFF::IMAGE_SCN_CNT_INITIALIZED_DATA |
1613       COFF::IMAGE_SCN_MEM_READ;
1614   else if (K.isWriteable())
1615     Flags |=
1616       COFF::IMAGE_SCN_CNT_INITIALIZED_DATA |
1617       COFF::IMAGE_SCN_MEM_READ |
1618       COFF::IMAGE_SCN_MEM_WRITE;
1619 
1620   return Flags;
1621 }
1622 
1623 static const GlobalValue *getComdatGVForCOFF(const GlobalValue *GV) {
1624   const Comdat *C = GV->getComdat();
1625   assert(C && "expected GV to have a Comdat!");
1626 
1627   StringRef ComdatGVName = C->getName();
1628   const GlobalValue *ComdatGV = GV->getParent()->getNamedValue(ComdatGVName);
1629   if (!ComdatGV)
1630     report_fatal_error("Associative COMDAT symbol '" + ComdatGVName +
1631                        "' does not exist.");
1632 
1633   if (ComdatGV->getComdat() != C)
1634     report_fatal_error("Associative COMDAT symbol '" + ComdatGVName +
1635                        "' is not a key for its COMDAT.");
1636 
1637   return ComdatGV;
1638 }
1639 
1640 static int getSelectionForCOFF(const GlobalValue *GV) {
1641   if (const Comdat *C = GV->getComdat()) {
1642     const GlobalValue *ComdatKey = getComdatGVForCOFF(GV);
1643     if (const auto *GA = dyn_cast<GlobalAlias>(ComdatKey))
1644       ComdatKey = GA->getAliaseeObject();
1645     if (ComdatKey == GV) {
1646       switch (C->getSelectionKind()) {
1647       case Comdat::Any:
1648         return COFF::IMAGE_COMDAT_SELECT_ANY;
1649       case Comdat::ExactMatch:
1650         return COFF::IMAGE_COMDAT_SELECT_EXACT_MATCH;
1651       case Comdat::Largest:
1652         return COFF::IMAGE_COMDAT_SELECT_LARGEST;
1653       case Comdat::NoDeduplicate:
1654         return COFF::IMAGE_COMDAT_SELECT_NODUPLICATES;
1655       case Comdat::SameSize:
1656         return COFF::IMAGE_COMDAT_SELECT_SAME_SIZE;
1657       }
1658     } else {
1659       return COFF::IMAGE_COMDAT_SELECT_ASSOCIATIVE;
1660     }
1661   }
1662   return 0;
1663 }
1664 
1665 MCSection *TargetLoweringObjectFileCOFF::getExplicitSectionGlobal(
1666     const GlobalObject *GO, SectionKind Kind, const TargetMachine &TM) const {
1667   StringRef Name = GO->getSection();
1668   if (Name == getInstrProfSectionName(IPSK_covmap, Triple::COFF,
1669                                       /*AddSegmentInfo=*/false) ||
1670       Name == getInstrProfSectionName(IPSK_covfun, Triple::COFF,
1671                                       /*AddSegmentInfo=*/false) ||
1672       Name == getInstrProfSectionName(IPSK_covdata, Triple::COFF,
1673                                       /*AddSegmentInfo=*/false) ||
1674       Name == getInstrProfSectionName(IPSK_covname, Triple::COFF,
1675                                       /*AddSegmentInfo=*/false))
1676     Kind = SectionKind::getMetadata();
1677   int Selection = 0;
1678   unsigned Characteristics = getCOFFSectionFlags(Kind, TM);
1679   StringRef COMDATSymName = "";
1680   if (GO->hasComdat()) {
1681     Selection = getSelectionForCOFF(GO);
1682     const GlobalValue *ComdatGV;
1683     if (Selection == COFF::IMAGE_COMDAT_SELECT_ASSOCIATIVE)
1684       ComdatGV = getComdatGVForCOFF(GO);
1685     else
1686       ComdatGV = GO;
1687 
1688     if (!ComdatGV->hasPrivateLinkage()) {
1689       MCSymbol *Sym = TM.getSymbol(ComdatGV);
1690       COMDATSymName = Sym->getName();
1691       Characteristics |= COFF::IMAGE_SCN_LNK_COMDAT;
1692     } else {
1693       Selection = 0;
1694     }
1695   }
1696 
1697   return getContext().getCOFFSection(Name, Characteristics, Kind, COMDATSymName,
1698                                      Selection);
1699 }
1700 
1701 static StringRef getCOFFSectionNameForUniqueGlobal(SectionKind Kind) {
1702   if (Kind.isText())
1703     return ".text";
1704   if (Kind.isBSS())
1705     return ".bss";
1706   if (Kind.isThreadLocal())
1707     return ".tls$";
1708   if (Kind.isReadOnly() || Kind.isReadOnlyWithRel())
1709     return ".rdata";
1710   return ".data";
1711 }
1712 
1713 MCSection *TargetLoweringObjectFileCOFF::SelectSectionForGlobal(
1714     const GlobalObject *GO, SectionKind Kind, const TargetMachine &TM) const {
1715   // If we have -ffunction-sections then we should emit the global value to a
1716   // uniqued section specifically for it.
1717   bool EmitUniquedSection;
1718   if (Kind.isText())
1719     EmitUniquedSection = TM.getFunctionSections();
1720   else
1721     EmitUniquedSection = TM.getDataSections();
1722 
1723   if ((EmitUniquedSection && !Kind.isCommon()) || GO->hasComdat()) {
1724     SmallString<256> Name = getCOFFSectionNameForUniqueGlobal(Kind);
1725 
1726     unsigned Characteristics = getCOFFSectionFlags(Kind, TM);
1727 
1728     Characteristics |= COFF::IMAGE_SCN_LNK_COMDAT;
1729     int Selection = getSelectionForCOFF(GO);
1730     if (!Selection)
1731       Selection = COFF::IMAGE_COMDAT_SELECT_NODUPLICATES;
1732     const GlobalValue *ComdatGV;
1733     if (GO->hasComdat())
1734       ComdatGV = getComdatGVForCOFF(GO);
1735     else
1736       ComdatGV = GO;
1737 
1738     unsigned UniqueID = MCContext::GenericSectionID;
1739     if (EmitUniquedSection)
1740       UniqueID = NextUniqueID++;
1741 
1742     if (!ComdatGV->hasPrivateLinkage()) {
1743       MCSymbol *Sym = TM.getSymbol(ComdatGV);
1744       StringRef COMDATSymName = Sym->getName();
1745 
1746       if (const auto *F = dyn_cast<Function>(GO))
1747         if (std::optional<StringRef> Prefix = F->getSectionPrefix())
1748           raw_svector_ostream(Name) << '$' << *Prefix;
1749 
1750       // Append "$symbol" to the section name *before* IR-level mangling is
1751       // applied when targetting mingw. This is what GCC does, and the ld.bfd
1752       // COFF linker will not properly handle comdats otherwise.
1753       if (getContext().getTargetTriple().isWindowsGNUEnvironment())
1754         raw_svector_ostream(Name) << '$' << ComdatGV->getName();
1755 
1756       return getContext().getCOFFSection(Name, Characteristics, Kind,
1757                                          COMDATSymName, Selection, UniqueID);
1758     } else {
1759       SmallString<256> TmpData;
1760       getMangler().getNameWithPrefix(TmpData, GO, /*CannotUsePrivateLabel=*/true);
1761       return getContext().getCOFFSection(Name, Characteristics, Kind, TmpData,
1762                                          Selection, UniqueID);
1763     }
1764   }
1765 
1766   if (Kind.isText())
1767     return TextSection;
1768 
1769   if (Kind.isThreadLocal())
1770     return TLSDataSection;
1771 
1772   if (Kind.isReadOnly() || Kind.isReadOnlyWithRel())
1773     return ReadOnlySection;
1774 
1775   // Note: we claim that common symbols are put in BSSSection, but they are
1776   // really emitted with the magic .comm directive, which creates a symbol table
1777   // entry but not a section.
1778   if (Kind.isBSS() || Kind.isCommon())
1779     return BSSSection;
1780 
1781   return DataSection;
1782 }
1783 
1784 void TargetLoweringObjectFileCOFF::getNameWithPrefix(
1785     SmallVectorImpl<char> &OutName, const GlobalValue *GV,
1786     const TargetMachine &TM) const {
1787   bool CannotUsePrivateLabel = false;
1788   if (GV->hasPrivateLinkage() &&
1789       ((isa<Function>(GV) && TM.getFunctionSections()) ||
1790        (isa<GlobalVariable>(GV) && TM.getDataSections())))
1791     CannotUsePrivateLabel = true;
1792 
1793   getMangler().getNameWithPrefix(OutName, GV, CannotUsePrivateLabel);
1794 }
1795 
1796 MCSection *TargetLoweringObjectFileCOFF::getSectionForJumpTable(
1797     const Function &F, const TargetMachine &TM) const {
1798   // If the function can be removed, produce a unique section so that
1799   // the table doesn't prevent the removal.
1800   const Comdat *C = F.getComdat();
1801   bool EmitUniqueSection = TM.getFunctionSections() || C;
1802   if (!EmitUniqueSection)
1803     return ReadOnlySection;
1804 
1805   // FIXME: we should produce a symbol for F instead.
1806   if (F.hasPrivateLinkage())
1807     return ReadOnlySection;
1808 
1809   MCSymbol *Sym = TM.getSymbol(&F);
1810   StringRef COMDATSymName = Sym->getName();
1811 
1812   SectionKind Kind = SectionKind::getReadOnly();
1813   StringRef SecName = getCOFFSectionNameForUniqueGlobal(Kind);
1814   unsigned Characteristics = getCOFFSectionFlags(Kind, TM);
1815   Characteristics |= COFF::IMAGE_SCN_LNK_COMDAT;
1816   unsigned UniqueID = NextUniqueID++;
1817 
1818   return getContext().getCOFFSection(
1819       SecName, Characteristics, Kind, COMDATSymName,
1820       COFF::IMAGE_COMDAT_SELECT_ASSOCIATIVE, UniqueID);
1821 }
1822 
1823 bool TargetLoweringObjectFileCOFF::shouldPutJumpTableInFunctionSection(
1824     bool UsesLabelDifference, const Function &F) const {
1825   if (TM->getTargetTriple().getArch() == Triple::x86_64) {
1826     if (!JumpTableInFunctionSection) {
1827       // We can always create relative relocations, so use another section
1828       // that can be marked non-executable.
1829       return false;
1830     }
1831   }
1832   return TargetLoweringObjectFile::shouldPutJumpTableInFunctionSection(
1833     UsesLabelDifference, F);
1834 }
1835 
1836 void TargetLoweringObjectFileCOFF::emitModuleMetadata(MCStreamer &Streamer,
1837                                                       Module &M) const {
1838   emitLinkerDirectives(Streamer, M);
1839 
1840   unsigned Version = 0;
1841   unsigned Flags = 0;
1842   StringRef Section;
1843 
1844   GetObjCImageInfo(M, Version, Flags, Section);
1845   if (!Section.empty()) {
1846     auto &C = getContext();
1847     auto *S = C.getCOFFSection(Section,
1848                                COFF::IMAGE_SCN_CNT_INITIALIZED_DATA |
1849                                    COFF::IMAGE_SCN_MEM_READ,
1850                                SectionKind::getReadOnly());
1851     Streamer.switchSection(S);
1852     Streamer.emitLabel(C.getOrCreateSymbol(StringRef("OBJC_IMAGE_INFO")));
1853     Streamer.emitInt32(Version);
1854     Streamer.emitInt32(Flags);
1855     Streamer.addBlankLine();
1856   }
1857 
1858   emitCGProfileMetadata(Streamer, M);
1859 }
1860 
1861 void TargetLoweringObjectFileCOFF::emitLinkerDirectives(
1862     MCStreamer &Streamer, Module &M) const {
1863   if (NamedMDNode *LinkerOptions = M.getNamedMetadata("llvm.linker.options")) {
1864     // Emit the linker options to the linker .drectve section.  According to the
1865     // spec, this section is a space-separated string containing flags for
1866     // linker.
1867     MCSection *Sec = getDrectveSection();
1868     Streamer.switchSection(Sec);
1869     for (const auto *Option : LinkerOptions->operands()) {
1870       for (const auto &Piece : cast<MDNode>(Option)->operands()) {
1871         // Lead with a space for consistency with our dllexport implementation.
1872         std::string Directive(" ");
1873         Directive.append(std::string(cast<MDString>(Piece)->getString()));
1874         Streamer.emitBytes(Directive);
1875       }
1876     }
1877   }
1878 
1879   // Emit /EXPORT: flags for each exported global as necessary.
1880   std::string Flags;
1881   for (const GlobalValue &GV : M.global_values()) {
1882     raw_string_ostream OS(Flags);
1883     emitLinkerFlagsForGlobalCOFF(OS, &GV, getContext().getTargetTriple(),
1884                                  getMangler());
1885     OS.flush();
1886     if (!Flags.empty()) {
1887       Streamer.switchSection(getDrectveSection());
1888       Streamer.emitBytes(Flags);
1889     }
1890     Flags.clear();
1891   }
1892 
1893   // Emit /INCLUDE: flags for each used global as necessary.
1894   if (const auto *LU = M.getNamedGlobal("llvm.used")) {
1895     assert(LU->hasInitializer() && "expected llvm.used to have an initializer");
1896     assert(isa<ArrayType>(LU->getValueType()) &&
1897            "expected llvm.used to be an array type");
1898     if (const auto *A = cast<ConstantArray>(LU->getInitializer())) {
1899       for (const Value *Op : A->operands()) {
1900         const auto *GV = cast<GlobalValue>(Op->stripPointerCasts());
1901         // Global symbols with internal or private linkage are not visible to
1902         // the linker, and thus would cause an error when the linker tried to
1903         // preserve the symbol due to the `/include:` directive.
1904         if (GV->hasLocalLinkage())
1905           continue;
1906 
1907         raw_string_ostream OS(Flags);
1908         emitLinkerFlagsForUsedCOFF(OS, GV, getContext().getTargetTriple(),
1909                                    getMangler());
1910         OS.flush();
1911 
1912         if (!Flags.empty()) {
1913           Streamer.switchSection(getDrectveSection());
1914           Streamer.emitBytes(Flags);
1915         }
1916         Flags.clear();
1917       }
1918     }
1919   }
1920 }
1921 
1922 void TargetLoweringObjectFileCOFF::Initialize(MCContext &Ctx,
1923                                               const TargetMachine &TM) {
1924   TargetLoweringObjectFile::Initialize(Ctx, TM);
1925   this->TM = &TM;
1926   const Triple &T = TM.getTargetTriple();
1927   if (T.isWindowsMSVCEnvironment() || T.isWindowsItaniumEnvironment()) {
1928     StaticCtorSection =
1929         Ctx.getCOFFSection(".CRT$XCU", COFF::IMAGE_SCN_CNT_INITIALIZED_DATA |
1930                                            COFF::IMAGE_SCN_MEM_READ,
1931                            SectionKind::getReadOnly());
1932     StaticDtorSection =
1933         Ctx.getCOFFSection(".CRT$XTX", COFF::IMAGE_SCN_CNT_INITIALIZED_DATA |
1934                                            COFF::IMAGE_SCN_MEM_READ,
1935                            SectionKind::getReadOnly());
1936   } else {
1937     StaticCtorSection = Ctx.getCOFFSection(
1938         ".ctors", COFF::IMAGE_SCN_CNT_INITIALIZED_DATA |
1939                       COFF::IMAGE_SCN_MEM_READ | COFF::IMAGE_SCN_MEM_WRITE,
1940         SectionKind::getData());
1941     StaticDtorSection = Ctx.getCOFFSection(
1942         ".dtors", COFF::IMAGE_SCN_CNT_INITIALIZED_DATA |
1943                       COFF::IMAGE_SCN_MEM_READ | COFF::IMAGE_SCN_MEM_WRITE,
1944         SectionKind::getData());
1945   }
1946 }
1947 
1948 static MCSectionCOFF *getCOFFStaticStructorSection(MCContext &Ctx,
1949                                                    const Triple &T, bool IsCtor,
1950                                                    unsigned Priority,
1951                                                    const MCSymbol *KeySym,
1952                                                    MCSectionCOFF *Default) {
1953   if (T.isWindowsMSVCEnvironment() || T.isWindowsItaniumEnvironment()) {
1954     // If the priority is the default, use .CRT$XCU, possibly associative.
1955     if (Priority == 65535)
1956       return Ctx.getAssociativeCOFFSection(Default, KeySym, 0);
1957 
1958     // Otherwise, we need to compute a new section name. Low priorities should
1959     // run earlier. The linker will sort sections ASCII-betically, and we need a
1960     // string that sorts between .CRT$XCA and .CRT$XCU. In the general case, we
1961     // make a name like ".CRT$XCT12345", since that runs before .CRT$XCU. Really
1962     // low priorities need to sort before 'L', since the CRT uses that
1963     // internally, so we use ".CRT$XCA00001" for them. We have a contract with
1964     // the frontend that "init_seg(compiler)" corresponds to priority 200 and
1965     // "init_seg(lib)" corresponds to priority 400, and those respectively use
1966     // 'C' and 'L' without the priority suffix. Priorities between 200 and 400
1967     // use 'C' with the priority as a suffix.
1968     SmallString<24> Name;
1969     char LastLetter = 'T';
1970     bool AddPrioritySuffix = Priority != 200 && Priority != 400;
1971     if (Priority < 200)
1972       LastLetter = 'A';
1973     else if (Priority < 400)
1974       LastLetter = 'C';
1975     else if (Priority == 400)
1976       LastLetter = 'L';
1977     raw_svector_ostream OS(Name);
1978     OS << ".CRT$X" << (IsCtor ? "C" : "T") << LastLetter;
1979     if (AddPrioritySuffix)
1980       OS << format("%05u", Priority);
1981     MCSectionCOFF *Sec = Ctx.getCOFFSection(
1982         Name, COFF::IMAGE_SCN_CNT_INITIALIZED_DATA | COFF::IMAGE_SCN_MEM_READ,
1983         SectionKind::getReadOnly());
1984     return Ctx.getAssociativeCOFFSection(Sec, KeySym, 0);
1985   }
1986 
1987   std::string Name = IsCtor ? ".ctors" : ".dtors";
1988   if (Priority != 65535)
1989     raw_string_ostream(Name) << format(".%05u", 65535 - Priority);
1990 
1991   return Ctx.getAssociativeCOFFSection(
1992       Ctx.getCOFFSection(Name, COFF::IMAGE_SCN_CNT_INITIALIZED_DATA |
1993                                    COFF::IMAGE_SCN_MEM_READ |
1994                                    COFF::IMAGE_SCN_MEM_WRITE,
1995                          SectionKind::getData()),
1996       KeySym, 0);
1997 }
1998 
1999 MCSection *TargetLoweringObjectFileCOFF::getStaticCtorSection(
2000     unsigned Priority, const MCSymbol *KeySym) const {
2001   return getCOFFStaticStructorSection(
2002       getContext(), getContext().getTargetTriple(), true, Priority, KeySym,
2003       cast<MCSectionCOFF>(StaticCtorSection));
2004 }
2005 
2006 MCSection *TargetLoweringObjectFileCOFF::getStaticDtorSection(
2007     unsigned Priority, const MCSymbol *KeySym) const {
2008   return getCOFFStaticStructorSection(
2009       getContext(), getContext().getTargetTriple(), false, Priority, KeySym,
2010       cast<MCSectionCOFF>(StaticDtorSection));
2011 }
2012 
2013 const MCExpr *TargetLoweringObjectFileCOFF::lowerRelativeReference(
2014     const GlobalValue *LHS, const GlobalValue *RHS,
2015     const TargetMachine &TM) const {
2016   const Triple &T = TM.getTargetTriple();
2017   if (T.isOSCygMing())
2018     return nullptr;
2019 
2020   // Our symbols should exist in address space zero, cowardly no-op if
2021   // otherwise.
2022   if (LHS->getType()->getPointerAddressSpace() != 0 ||
2023       RHS->getType()->getPointerAddressSpace() != 0)
2024     return nullptr;
2025 
2026   // Both ptrtoint instructions must wrap global objects:
2027   // - Only global variables are eligible for image relative relocations.
2028   // - The subtrahend refers to the special symbol __ImageBase, a GlobalVariable.
2029   // We expect __ImageBase to be a global variable without a section, externally
2030   // defined.
2031   //
2032   // It should look something like this: @__ImageBase = external constant i8
2033   if (!isa<GlobalObject>(LHS) || !isa<GlobalVariable>(RHS) ||
2034       LHS->isThreadLocal() || RHS->isThreadLocal() ||
2035       RHS->getName() != "__ImageBase" || !RHS->hasExternalLinkage() ||
2036       cast<GlobalVariable>(RHS)->hasInitializer() || RHS->hasSection())
2037     return nullptr;
2038 
2039   return MCSymbolRefExpr::create(TM.getSymbol(LHS),
2040                                  MCSymbolRefExpr::VK_COFF_IMGREL32,
2041                                  getContext());
2042 }
2043 
2044 static std::string APIntToHexString(const APInt &AI) {
2045   unsigned Width = (AI.getBitWidth() / 8) * 2;
2046   std::string HexString = toString(AI, 16, /*Signed=*/false);
2047   llvm::transform(HexString, HexString.begin(), tolower);
2048   unsigned Size = HexString.size();
2049   assert(Width >= Size && "hex string is too large!");
2050   HexString.insert(HexString.begin(), Width - Size, '0');
2051 
2052   return HexString;
2053 }
2054 
2055 static std::string scalarConstantToHexString(const Constant *C) {
2056   Type *Ty = C->getType();
2057   if (isa<UndefValue>(C)) {
2058     return APIntToHexString(APInt::getZero(Ty->getPrimitiveSizeInBits()));
2059   } else if (const auto *CFP = dyn_cast<ConstantFP>(C)) {
2060     return APIntToHexString(CFP->getValueAPF().bitcastToAPInt());
2061   } else if (const auto *CI = dyn_cast<ConstantInt>(C)) {
2062     return APIntToHexString(CI->getValue());
2063   } else {
2064     unsigned NumElements;
2065     if (auto *VTy = dyn_cast<VectorType>(Ty))
2066       NumElements = cast<FixedVectorType>(VTy)->getNumElements();
2067     else
2068       NumElements = Ty->getArrayNumElements();
2069     std::string HexString;
2070     for (int I = NumElements - 1, E = -1; I != E; --I)
2071       HexString += scalarConstantToHexString(C->getAggregateElement(I));
2072     return HexString;
2073   }
2074 }
2075 
2076 MCSection *TargetLoweringObjectFileCOFF::getSectionForConstant(
2077     const DataLayout &DL, SectionKind Kind, const Constant *C,
2078     Align &Alignment) const {
2079   if (Kind.isMergeableConst() && C &&
2080       getContext().getAsmInfo()->hasCOFFComdatConstants()) {
2081     // This creates comdat sections with the given symbol name, but unless
2082     // AsmPrinter::GetCPISymbol actually makes the symbol global, the symbol
2083     // will be created with a null storage class, which makes GNU binutils
2084     // error out.
2085     const unsigned Characteristics = COFF::IMAGE_SCN_CNT_INITIALIZED_DATA |
2086                                      COFF::IMAGE_SCN_MEM_READ |
2087                                      COFF::IMAGE_SCN_LNK_COMDAT;
2088     std::string COMDATSymName;
2089     if (Kind.isMergeableConst4()) {
2090       if (Alignment <= 4) {
2091         COMDATSymName = "__real@" + scalarConstantToHexString(C);
2092         Alignment = Align(4);
2093       }
2094     } else if (Kind.isMergeableConst8()) {
2095       if (Alignment <= 8) {
2096         COMDATSymName = "__real@" + scalarConstantToHexString(C);
2097         Alignment = Align(8);
2098       }
2099     } else if (Kind.isMergeableConst16()) {
2100       // FIXME: These may not be appropriate for non-x86 architectures.
2101       if (Alignment <= 16) {
2102         COMDATSymName = "__xmm@" + scalarConstantToHexString(C);
2103         Alignment = Align(16);
2104       }
2105     } else if (Kind.isMergeableConst32()) {
2106       if (Alignment <= 32) {
2107         COMDATSymName = "__ymm@" + scalarConstantToHexString(C);
2108         Alignment = Align(32);
2109       }
2110     }
2111 
2112     if (!COMDATSymName.empty())
2113       return getContext().getCOFFSection(".rdata", Characteristics, Kind,
2114                                          COMDATSymName,
2115                                          COFF::IMAGE_COMDAT_SELECT_ANY);
2116   }
2117 
2118   return TargetLoweringObjectFile::getSectionForConstant(DL, Kind, C,
2119                                                          Alignment);
2120 }
2121 
2122 //===----------------------------------------------------------------------===//
2123 //                                  Wasm
2124 //===----------------------------------------------------------------------===//
2125 
2126 static const Comdat *getWasmComdat(const GlobalValue *GV) {
2127   const Comdat *C = GV->getComdat();
2128   if (!C)
2129     return nullptr;
2130 
2131   if (C->getSelectionKind() != Comdat::Any)
2132     report_fatal_error("WebAssembly COMDATs only support "
2133                        "SelectionKind::Any, '" + C->getName() + "' cannot be "
2134                        "lowered.");
2135 
2136   return C;
2137 }
2138 
2139 static unsigned getWasmSectionFlags(SectionKind K, bool Retain) {
2140   unsigned Flags = 0;
2141 
2142   if (K.isThreadLocal())
2143     Flags |= wasm::WASM_SEG_FLAG_TLS;
2144 
2145   if (K.isMergeableCString())
2146     Flags |= wasm::WASM_SEG_FLAG_STRINGS;
2147 
2148   if (Retain)
2149     Flags |= wasm::WASM_SEG_FLAG_RETAIN;
2150 
2151   // TODO(sbc): Add suport for K.isMergeableConst()
2152 
2153   return Flags;
2154 }
2155 
2156 void TargetLoweringObjectFileWasm::getModuleMetadata(Module &M) {
2157   SmallVector<GlobalValue *, 4> Vec;
2158   collectUsedGlobalVariables(M, Vec, false);
2159   for (GlobalValue *GV : Vec)
2160     if (auto *GO = dyn_cast<GlobalObject>(GV))
2161       Used.insert(GO);
2162 }
2163 
2164 MCSection *TargetLoweringObjectFileWasm::getExplicitSectionGlobal(
2165     const GlobalObject *GO, SectionKind Kind, const TargetMachine &TM) const {
2166   // We don't support explict section names for functions in the wasm object
2167   // format.  Each function has to be in its own unique section.
2168   if (isa<Function>(GO)) {
2169     return SelectSectionForGlobal(GO, Kind, TM);
2170   }
2171 
2172   StringRef Name = GO->getSection();
2173 
2174   // Certain data sections we treat as named custom sections rather than
2175   // segments within the data section.
2176   // This could be avoided if all data segements (the wasm sense) were
2177   // represented as their own sections (in the llvm sense).
2178   // TODO(sbc): https://github.com/WebAssembly/tool-conventions/issues/138
2179   if (Name == ".llvmcmd" || Name == ".llvmbc")
2180     Kind = SectionKind::getMetadata();
2181 
2182   StringRef Group = "";
2183   if (const Comdat *C = getWasmComdat(GO)) {
2184     Group = C->getName();
2185   }
2186 
2187   unsigned Flags = getWasmSectionFlags(Kind, Used.count(GO));
2188   MCSectionWasm *Section = getContext().getWasmSection(
2189       Name, Kind, Flags, Group, MCContext::GenericSectionID);
2190 
2191   return Section;
2192 }
2193 
2194 static MCSectionWasm *
2195 selectWasmSectionForGlobal(MCContext &Ctx, const GlobalObject *GO,
2196                            SectionKind Kind, Mangler &Mang,
2197                            const TargetMachine &TM, bool EmitUniqueSection,
2198                            unsigned *NextUniqueID, bool Retain) {
2199   StringRef Group = "";
2200   if (const Comdat *C = getWasmComdat(GO)) {
2201     Group = C->getName();
2202   }
2203 
2204   bool UniqueSectionNames = TM.getUniqueSectionNames();
2205   SmallString<128> Name = getSectionPrefixForGlobal(Kind, /*IsLarge=*/false);
2206 
2207   if (const auto *F = dyn_cast<Function>(GO)) {
2208     const auto &OptionalPrefix = F->getSectionPrefix();
2209     if (OptionalPrefix)
2210       raw_svector_ostream(Name) << '.' << *OptionalPrefix;
2211   }
2212 
2213   if (EmitUniqueSection && UniqueSectionNames) {
2214     Name.push_back('.');
2215     TM.getNameWithPrefix(Name, GO, Mang, true);
2216   }
2217   unsigned UniqueID = MCContext::GenericSectionID;
2218   if (EmitUniqueSection && !UniqueSectionNames) {
2219     UniqueID = *NextUniqueID;
2220     (*NextUniqueID)++;
2221   }
2222 
2223   unsigned Flags = getWasmSectionFlags(Kind, Retain);
2224   return Ctx.getWasmSection(Name, Kind, Flags, Group, UniqueID);
2225 }
2226 
2227 MCSection *TargetLoweringObjectFileWasm::SelectSectionForGlobal(
2228     const GlobalObject *GO, SectionKind Kind, const TargetMachine &TM) const {
2229 
2230   if (Kind.isCommon())
2231     report_fatal_error("mergable sections not supported yet on wasm");
2232 
2233   // If we have -ffunction-section or -fdata-section then we should emit the
2234   // global value to a uniqued section specifically for it.
2235   bool EmitUniqueSection = false;
2236   if (Kind.isText())
2237     EmitUniqueSection = TM.getFunctionSections();
2238   else
2239     EmitUniqueSection = TM.getDataSections();
2240   EmitUniqueSection |= GO->hasComdat();
2241   bool Retain = Used.count(GO);
2242   EmitUniqueSection |= Retain;
2243 
2244   return selectWasmSectionForGlobal(getContext(), GO, Kind, getMangler(), TM,
2245                                     EmitUniqueSection, &NextUniqueID, Retain);
2246 }
2247 
2248 bool TargetLoweringObjectFileWasm::shouldPutJumpTableInFunctionSection(
2249     bool UsesLabelDifference, const Function &F) const {
2250   // We can always create relative relocations, so use another section
2251   // that can be marked non-executable.
2252   return false;
2253 }
2254 
2255 const MCExpr *TargetLoweringObjectFileWasm::lowerRelativeReference(
2256     const GlobalValue *LHS, const GlobalValue *RHS,
2257     const TargetMachine &TM) const {
2258   // We may only use a PLT-relative relocation to refer to unnamed_addr
2259   // functions.
2260   if (!LHS->hasGlobalUnnamedAddr() || !LHS->getValueType()->isFunctionTy())
2261     return nullptr;
2262 
2263   // Basic correctness checks.
2264   if (LHS->getType()->getPointerAddressSpace() != 0 ||
2265       RHS->getType()->getPointerAddressSpace() != 0 || LHS->isThreadLocal() ||
2266       RHS->isThreadLocal())
2267     return nullptr;
2268 
2269   return MCBinaryExpr::createSub(
2270       MCSymbolRefExpr::create(TM.getSymbol(LHS), MCSymbolRefExpr::VK_None,
2271                               getContext()),
2272       MCSymbolRefExpr::create(TM.getSymbol(RHS), getContext()), getContext());
2273 }
2274 
2275 void TargetLoweringObjectFileWasm::InitializeWasm() {
2276   StaticCtorSection =
2277       getContext().getWasmSection(".init_array", SectionKind::getData());
2278 
2279   // We don't use PersonalityEncoding and LSDAEncoding because we don't emit
2280   // .cfi directives. We use TTypeEncoding to encode typeinfo global variables.
2281   TTypeEncoding = dwarf::DW_EH_PE_absptr;
2282 }
2283 
2284 MCSection *TargetLoweringObjectFileWasm::getStaticCtorSection(
2285     unsigned Priority, const MCSymbol *KeySym) const {
2286   return Priority == UINT16_MAX ?
2287          StaticCtorSection :
2288          getContext().getWasmSection(".init_array." + utostr(Priority),
2289                                      SectionKind::getData());
2290 }
2291 
2292 MCSection *TargetLoweringObjectFileWasm::getStaticDtorSection(
2293     unsigned Priority, const MCSymbol *KeySym) const {
2294   report_fatal_error("@llvm.global_dtors should have been lowered already");
2295 }
2296 
2297 //===----------------------------------------------------------------------===//
2298 //                                  XCOFF
2299 //===----------------------------------------------------------------------===//
2300 bool TargetLoweringObjectFileXCOFF::ShouldEmitEHBlock(
2301     const MachineFunction *MF) {
2302   if (!MF->getLandingPads().empty())
2303     return true;
2304 
2305   const Function &F = MF->getFunction();
2306   if (!F.hasPersonalityFn() || !F.needsUnwindTableEntry())
2307     return false;
2308 
2309   const GlobalValue *Per =
2310       dyn_cast<GlobalValue>(F.getPersonalityFn()->stripPointerCasts());
2311   assert(Per && "Personality routine is not a GlobalValue type.");
2312   if (isNoOpWithoutInvoke(classifyEHPersonality(Per)))
2313     return false;
2314 
2315   return true;
2316 }
2317 
2318 bool TargetLoweringObjectFileXCOFF::ShouldSetSSPCanaryBitInTB(
2319     const MachineFunction *MF) {
2320   const Function &F = MF->getFunction();
2321   if (!F.hasStackProtectorFnAttr())
2322     return false;
2323   // FIXME: check presence of canary word
2324   // There are cases that the stack protectors are not really inserted even if
2325   // the attributes are on.
2326   return true;
2327 }
2328 
2329 MCSymbol *
2330 TargetLoweringObjectFileXCOFF::getEHInfoTableSymbol(const MachineFunction *MF) {
2331   MCSymbol *EHInfoSym = MF->getMMI().getContext().getOrCreateSymbol(
2332       "__ehinfo." + Twine(MF->getFunctionNumber()));
2333   cast<MCSymbolXCOFF>(EHInfoSym)->setEHInfo();
2334   return EHInfoSym;
2335 }
2336 
2337 MCSymbol *
2338 TargetLoweringObjectFileXCOFF::getTargetSymbol(const GlobalValue *GV,
2339                                                const TargetMachine &TM) const {
2340   // We always use a qualname symbol for a GV that represents
2341   // a declaration, a function descriptor, or a common symbol.
2342   // If a GV represents a GlobalVariable and -fdata-sections is enabled, we
2343   // also return a qualname so that a label symbol could be avoided.
2344   // It is inherently ambiguous when the GO represents the address of a
2345   // function, as the GO could either represent a function descriptor or a
2346   // function entry point. We choose to always return a function descriptor
2347   // here.
2348   if (const GlobalObject *GO = dyn_cast<GlobalObject>(GV)) {
2349     if (GO->isDeclarationForLinker())
2350       return cast<MCSectionXCOFF>(getSectionForExternalReference(GO, TM))
2351           ->getQualNameSymbol();
2352 
2353     if (const GlobalVariable *GVar = dyn_cast<GlobalVariable>(GV))
2354       if (GVar->hasAttribute("toc-data"))
2355         return cast<MCSectionXCOFF>(
2356                    SectionForGlobal(GVar, SectionKind::getData(), TM))
2357             ->getQualNameSymbol();
2358 
2359     SectionKind GOKind = getKindForGlobal(GO, TM);
2360     if (GOKind.isText())
2361       return cast<MCSectionXCOFF>(
2362                  getSectionForFunctionDescriptor(cast<Function>(GO), TM))
2363           ->getQualNameSymbol();
2364     if ((TM.getDataSections() && !GO->hasSection()) || GO->hasCommonLinkage() ||
2365         GOKind.isBSSLocal() || GOKind.isThreadBSSLocal())
2366       return cast<MCSectionXCOFF>(SectionForGlobal(GO, GOKind, TM))
2367           ->getQualNameSymbol();
2368   }
2369 
2370   // For all other cases, fall back to getSymbol to return the unqualified name.
2371   return nullptr;
2372 }
2373 
2374 MCSection *TargetLoweringObjectFileXCOFF::getExplicitSectionGlobal(
2375     const GlobalObject *GO, SectionKind Kind, const TargetMachine &TM) const {
2376   if (!GO->hasSection())
2377     report_fatal_error("#pragma clang section is not yet supported");
2378 
2379   StringRef SectionName = GO->getSection();
2380 
2381   // Handle the XCOFF::TD case first, then deal with the rest.
2382   if (const GlobalVariable *GVar = dyn_cast<GlobalVariable>(GO))
2383     if (GVar->hasAttribute("toc-data"))
2384       return getContext().getXCOFFSection(
2385           SectionName, Kind,
2386           XCOFF::CsectProperties(/*MappingClass*/ XCOFF::XMC_TD, XCOFF::XTY_SD),
2387           /* MultiSymbolsAllowed*/ true);
2388 
2389   XCOFF::StorageMappingClass MappingClass;
2390   if (Kind.isText())
2391     MappingClass = XCOFF::XMC_PR;
2392   else if (Kind.isData() || Kind.isBSS())
2393     MappingClass = XCOFF::XMC_RW;
2394   else if (Kind.isReadOnlyWithRel())
2395     MappingClass =
2396         TM.Options.XCOFFReadOnlyPointers ? XCOFF::XMC_RO : XCOFF::XMC_RW;
2397   else if (Kind.isReadOnly())
2398     MappingClass = XCOFF::XMC_RO;
2399   else
2400     report_fatal_error("XCOFF other section types not yet implemented.");
2401 
2402   return getContext().getXCOFFSection(
2403       SectionName, Kind, XCOFF::CsectProperties(MappingClass, XCOFF::XTY_SD),
2404       /* MultiSymbolsAllowed*/ true);
2405 }
2406 
2407 MCSection *TargetLoweringObjectFileXCOFF::getSectionForExternalReference(
2408     const GlobalObject *GO, const TargetMachine &TM) const {
2409   assert(GO->isDeclarationForLinker() &&
2410          "Tried to get ER section for a defined global.");
2411 
2412   SmallString<128> Name;
2413   getNameWithPrefix(Name, GO, TM);
2414 
2415   // AIX TLS local-dynamic does not need the external reference for the
2416   // "_$TLSML" symbol.
2417   if (GO->getThreadLocalMode() == GlobalVariable::LocalDynamicTLSModel &&
2418       GO->hasName() && GO->getName() == "_$TLSML") {
2419     return getContext().getXCOFFSection(
2420         Name, SectionKind::getData(),
2421         XCOFF::CsectProperties(XCOFF::XMC_TC, XCOFF::XTY_SD));
2422   }
2423 
2424   XCOFF::StorageMappingClass SMC =
2425       isa<Function>(GO) ? XCOFF::XMC_DS : XCOFF::XMC_UA;
2426   if (GO->isThreadLocal())
2427     SMC = XCOFF::XMC_UL;
2428 
2429   if (const GlobalVariable *GVar = dyn_cast<GlobalVariable>(GO))
2430     if (GVar->hasAttribute("toc-data"))
2431       SMC = XCOFF::XMC_TD;
2432 
2433   // Externals go into a csect of type ER.
2434   return getContext().getXCOFFSection(
2435       Name, SectionKind::getMetadata(),
2436       XCOFF::CsectProperties(SMC, XCOFF::XTY_ER));
2437 }
2438 
2439 MCSection *TargetLoweringObjectFileXCOFF::SelectSectionForGlobal(
2440     const GlobalObject *GO, SectionKind Kind, const TargetMachine &TM) const {
2441   // Handle the XCOFF::TD case first, then deal with the rest.
2442   if (const GlobalVariable *GVar = dyn_cast<GlobalVariable>(GO))
2443     if (GVar->hasAttribute("toc-data")) {
2444       SmallString<128> Name;
2445       getNameWithPrefix(Name, GO, TM);
2446       XCOFF::SymbolType symType =
2447           GO->hasCommonLinkage() ? XCOFF::XTY_CM : XCOFF::XTY_SD;
2448       return getContext().getXCOFFSection(
2449           Name, Kind, XCOFF::CsectProperties(XCOFF::XMC_TD, symType),
2450           /* MultiSymbolsAllowed*/ true);
2451     }
2452 
2453   // Common symbols go into a csect with matching name which will get mapped
2454   // into the .bss section.
2455   // Zero-initialized local TLS symbols go into a csect with matching name which
2456   // will get mapped into the .tbss section.
2457   if (Kind.isBSSLocal() || GO->hasCommonLinkage() || Kind.isThreadBSSLocal()) {
2458     SmallString<128> Name;
2459     getNameWithPrefix(Name, GO, TM);
2460     XCOFF::StorageMappingClass SMC = Kind.isBSSLocal() ? XCOFF::XMC_BS
2461                                      : Kind.isCommon() ? XCOFF::XMC_RW
2462                                                        : XCOFF::XMC_UL;
2463     return getContext().getXCOFFSection(
2464         Name, Kind, XCOFF::CsectProperties(SMC, XCOFF::XTY_CM));
2465   }
2466 
2467   if (Kind.isText()) {
2468     if (TM.getFunctionSections()) {
2469       return cast<MCSymbolXCOFF>(getFunctionEntryPointSymbol(GO, TM))
2470           ->getRepresentedCsect();
2471     }
2472     return TextSection;
2473   }
2474 
2475   if (TM.Options.XCOFFReadOnlyPointers && Kind.isReadOnlyWithRel()) {
2476     if (!TM.getDataSections())
2477       report_fatal_error(
2478           "ReadOnlyPointers is supported only if data sections is turned on");
2479 
2480     SmallString<128> Name;
2481     getNameWithPrefix(Name, GO, TM);
2482     return getContext().getXCOFFSection(
2483         Name, SectionKind::getReadOnly(),
2484         XCOFF::CsectProperties(XCOFF::XMC_RO, XCOFF::XTY_SD));
2485   }
2486 
2487   // For BSS kind, zero initialized data must be emitted to the .data section
2488   // because external linkage control sections that get mapped to the .bss
2489   // section will be linked as tentative defintions, which is only appropriate
2490   // for SectionKind::Common.
2491   if (Kind.isData() || Kind.isReadOnlyWithRel() || Kind.isBSS()) {
2492     if (TM.getDataSections()) {
2493       SmallString<128> Name;
2494       getNameWithPrefix(Name, GO, TM);
2495       return getContext().getXCOFFSection(
2496           Name, SectionKind::getData(),
2497           XCOFF::CsectProperties(XCOFF::XMC_RW, XCOFF::XTY_SD));
2498     }
2499     return DataSection;
2500   }
2501 
2502   if (Kind.isReadOnly()) {
2503     if (TM.getDataSections()) {
2504       SmallString<128> Name;
2505       getNameWithPrefix(Name, GO, TM);
2506       return getContext().getXCOFFSection(
2507           Name, SectionKind::getReadOnly(),
2508           XCOFF::CsectProperties(XCOFF::XMC_RO, XCOFF::XTY_SD));
2509     }
2510     return ReadOnlySection;
2511   }
2512 
2513   // External/weak TLS data and initialized local TLS data are not eligible
2514   // to be put into common csect. If data sections are enabled, thread
2515   // data are emitted into separate sections. Otherwise, thread data
2516   // are emitted into the .tdata section.
2517   if (Kind.isThreadLocal()) {
2518     if (TM.getDataSections()) {
2519       SmallString<128> Name;
2520       getNameWithPrefix(Name, GO, TM);
2521       return getContext().getXCOFFSection(
2522           Name, Kind, XCOFF::CsectProperties(XCOFF::XMC_TL, XCOFF::XTY_SD));
2523     }
2524     return TLSDataSection;
2525   }
2526 
2527   report_fatal_error("XCOFF other section types not yet implemented.");
2528 }
2529 
2530 MCSection *TargetLoweringObjectFileXCOFF::getSectionForJumpTable(
2531     const Function &F, const TargetMachine &TM) const {
2532   assert (!F.getComdat() && "Comdat not supported on XCOFF.");
2533 
2534   if (!TM.getFunctionSections())
2535     return ReadOnlySection;
2536 
2537   // If the function can be removed, produce a unique section so that
2538   // the table doesn't prevent the removal.
2539   SmallString<128> NameStr(".rodata.jmp..");
2540   getNameWithPrefix(NameStr, &F, TM);
2541   return getContext().getXCOFFSection(
2542       NameStr, SectionKind::getReadOnly(),
2543       XCOFF::CsectProperties(XCOFF::XMC_RO, XCOFF::XTY_SD));
2544 }
2545 
2546 bool TargetLoweringObjectFileXCOFF::shouldPutJumpTableInFunctionSection(
2547     bool UsesLabelDifference, const Function &F) const {
2548   return false;
2549 }
2550 
2551 /// Given a mergeable constant with the specified size and relocation
2552 /// information, return a section that it should be placed in.
2553 MCSection *TargetLoweringObjectFileXCOFF::getSectionForConstant(
2554     const DataLayout &DL, SectionKind Kind, const Constant *C,
2555     Align &Alignment) const {
2556   // TODO: Enable emiting constant pool to unique sections when we support it.
2557   if (Alignment > Align(16))
2558     report_fatal_error("Alignments greater than 16 not yet supported.");
2559 
2560   if (Alignment == Align(8)) {
2561     assert(ReadOnly8Section && "Section should always be initialized.");
2562     return ReadOnly8Section;
2563   }
2564 
2565   if (Alignment == Align(16)) {
2566     assert(ReadOnly16Section && "Section should always be initialized.");
2567     return ReadOnly16Section;
2568   }
2569 
2570   return ReadOnlySection;
2571 }
2572 
2573 void TargetLoweringObjectFileXCOFF::Initialize(MCContext &Ctx,
2574                                                const TargetMachine &TgtM) {
2575   TargetLoweringObjectFile::Initialize(Ctx, TgtM);
2576   TTypeEncoding =
2577       dwarf::DW_EH_PE_indirect | dwarf::DW_EH_PE_datarel |
2578       (TgtM.getTargetTriple().isArch32Bit() ? dwarf::DW_EH_PE_sdata4
2579                                             : dwarf::DW_EH_PE_sdata8);
2580   PersonalityEncoding = 0;
2581   LSDAEncoding = 0;
2582   CallSiteEncoding = dwarf::DW_EH_PE_udata4;
2583 
2584   // AIX debug for thread local location is not ready. And for integrated as
2585   // mode, the relocatable address for the thread local variable will cause
2586   // linker error. So disable the location attribute generation for thread local
2587   // variables for now.
2588   // FIXME: when TLS debug on AIX is ready, remove this setting.
2589   SupportDebugThreadLocalLocation = false;
2590 }
2591 
2592 MCSection *TargetLoweringObjectFileXCOFF::getStaticCtorSection(
2593 	unsigned Priority, const MCSymbol *KeySym) const {
2594   report_fatal_error("no static constructor section on AIX");
2595 }
2596 
2597 MCSection *TargetLoweringObjectFileXCOFF::getStaticDtorSection(
2598 	unsigned Priority, const MCSymbol *KeySym) const {
2599   report_fatal_error("no static destructor section on AIX");
2600 }
2601 
2602 const MCExpr *TargetLoweringObjectFileXCOFF::lowerRelativeReference(
2603     const GlobalValue *LHS, const GlobalValue *RHS,
2604     const TargetMachine &TM) const {
2605   /* Not implemented yet, but don't crash, return nullptr. */
2606   return nullptr;
2607 }
2608 
2609 XCOFF::StorageClass
2610 TargetLoweringObjectFileXCOFF::getStorageClassForGlobal(const GlobalValue *GV) {
2611   assert(!isa<GlobalIFunc>(GV) && "GlobalIFunc is not supported on AIX.");
2612 
2613   switch (GV->getLinkage()) {
2614   case GlobalValue::InternalLinkage:
2615   case GlobalValue::PrivateLinkage:
2616     return XCOFF::C_HIDEXT;
2617   case GlobalValue::ExternalLinkage:
2618   case GlobalValue::CommonLinkage:
2619   case GlobalValue::AvailableExternallyLinkage:
2620     return XCOFF::C_EXT;
2621   case GlobalValue::ExternalWeakLinkage:
2622   case GlobalValue::LinkOnceAnyLinkage:
2623   case GlobalValue::LinkOnceODRLinkage:
2624   case GlobalValue::WeakAnyLinkage:
2625   case GlobalValue::WeakODRLinkage:
2626     return XCOFF::C_WEAKEXT;
2627   case GlobalValue::AppendingLinkage:
2628     report_fatal_error(
2629         "There is no mapping that implements AppendingLinkage for XCOFF.");
2630   }
2631   llvm_unreachable("Unknown linkage type!");
2632 }
2633 
2634 MCSymbol *TargetLoweringObjectFileXCOFF::getFunctionEntryPointSymbol(
2635     const GlobalValue *Func, const TargetMachine &TM) const {
2636   assert((isa<Function>(Func) ||
2637           (isa<GlobalAlias>(Func) &&
2638            isa_and_nonnull<Function>(
2639                cast<GlobalAlias>(Func)->getAliaseeObject()))) &&
2640          "Func must be a function or an alias which has a function as base "
2641          "object.");
2642 
2643   SmallString<128> NameStr;
2644   NameStr.push_back('.');
2645   getNameWithPrefix(NameStr, Func, TM);
2646 
2647   // When -function-sections is enabled and explicit section is not specified,
2648   // it's not necessary to emit function entry point label any more. We will use
2649   // function entry point csect instead. And for function delcarations, the
2650   // undefined symbols gets treated as csect with XTY_ER property.
2651   if (((TM.getFunctionSections() && !Func->hasSection()) ||
2652        Func->isDeclarationForLinker()) &&
2653       isa<Function>(Func)) {
2654     return getContext()
2655         .getXCOFFSection(
2656             NameStr, SectionKind::getText(),
2657             XCOFF::CsectProperties(XCOFF::XMC_PR, Func->isDeclarationForLinker()
2658                                                       ? XCOFF::XTY_ER
2659                                                       : XCOFF::XTY_SD))
2660         ->getQualNameSymbol();
2661   }
2662 
2663   return getContext().getOrCreateSymbol(NameStr);
2664 }
2665 
2666 MCSection *TargetLoweringObjectFileXCOFF::getSectionForFunctionDescriptor(
2667     const Function *F, const TargetMachine &TM) const {
2668   SmallString<128> NameStr;
2669   getNameWithPrefix(NameStr, F, TM);
2670   return getContext().getXCOFFSection(
2671       NameStr, SectionKind::getData(),
2672       XCOFF::CsectProperties(XCOFF::XMC_DS, XCOFF::XTY_SD));
2673 }
2674 
2675 MCSection *TargetLoweringObjectFileXCOFF::getSectionForTOCEntry(
2676     const MCSymbol *Sym, const TargetMachine &TM) const {
2677   const XCOFF::StorageMappingClass SMC = [](const MCSymbol *Sym,
2678                                             const TargetMachine &TM) {
2679     const MCSymbolXCOFF *XSym = cast<MCSymbolXCOFF>(Sym);
2680 
2681     // The "_$TLSML" symbol for TLS local-dynamic mode requires XMC_TC,
2682     // otherwise the AIX assembler will complain.
2683     if (XSym->getSymbolTableName() == "_$TLSML")
2684       return XCOFF::XMC_TC;
2685 
2686     // Use large code model toc entries for ehinfo symbols as they are
2687     // never referenced directly. The runtime loads their TOC entry
2688     // addresses from the trace-back table.
2689     if (XSym->isEHInfo())
2690       return XCOFF::XMC_TE;
2691 
2692     // If the symbol does not have a code model specified use the module value.
2693     if (!XSym->hasPerSymbolCodeModel())
2694       return TM.getCodeModel() == CodeModel::Large ? XCOFF::XMC_TE
2695                                                    : XCOFF::XMC_TC;
2696 
2697     return XSym->getPerSymbolCodeModel() == MCSymbolXCOFF::CM_Large
2698                ? XCOFF::XMC_TE
2699                : XCOFF::XMC_TC;
2700   }(Sym, TM);
2701 
2702   return getContext().getXCOFFSection(
2703       cast<MCSymbolXCOFF>(Sym)->getSymbolTableName(), SectionKind::getData(),
2704       XCOFF::CsectProperties(SMC, XCOFF::XTY_SD));
2705 }
2706 
2707 MCSection *TargetLoweringObjectFileXCOFF::getSectionForLSDA(
2708     const Function &F, const MCSymbol &FnSym, const TargetMachine &TM) const {
2709   auto *LSDA = cast<MCSectionXCOFF>(LSDASection);
2710   if (TM.getFunctionSections()) {
2711     // If option -ffunction-sections is on, append the function name to the
2712     // name of the LSDA csect so that each function has its own LSDA csect.
2713     // This helps the linker to garbage-collect EH info of unused functions.
2714     SmallString<128> NameStr = LSDA->getName();
2715     raw_svector_ostream(NameStr) << '.' << F.getName();
2716     LSDA = getContext().getXCOFFSection(NameStr, LSDA->getKind(),
2717                                         LSDA->getCsectProp());
2718   }
2719   return LSDA;
2720 }
2721 //===----------------------------------------------------------------------===//
2722 //                                  GOFF
2723 //===----------------------------------------------------------------------===//
2724 TargetLoweringObjectFileGOFF::TargetLoweringObjectFileGOFF() = default;
2725 
2726 MCSection *TargetLoweringObjectFileGOFF::getExplicitSectionGlobal(
2727     const GlobalObject *GO, SectionKind Kind, const TargetMachine &TM) const {
2728   return SelectSectionForGlobal(GO, Kind, TM);
2729 }
2730 
2731 MCSection *TargetLoweringObjectFileGOFF::getSectionForLSDA(
2732     const Function &F, const MCSymbol &FnSym, const TargetMachine &TM) const {
2733   std::string Name = ".gcc_exception_table." + F.getName().str();
2734   return getContext().getGOFFSection(Name, SectionKind::getData(), nullptr,
2735                                      nullptr);
2736 }
2737 
2738 MCSection *TargetLoweringObjectFileGOFF::SelectSectionForGlobal(
2739     const GlobalObject *GO, SectionKind Kind, const TargetMachine &TM) const {
2740   auto *Symbol = TM.getSymbol(GO);
2741   if (Kind.isBSS())
2742     return getContext().getGOFFSection(Symbol->getName(), SectionKind::getBSS(),
2743                                        nullptr, nullptr);
2744 
2745   return getContext().getObjectFileInfo()->getTextSection();
2746 }
2747