277716c89b46e3e23c914a87331303e0b3497b90
[oota-llvm.git] / lib / MC / ELFObjectWriter.cpp
1 //===- lib/MC/ELFObjectWriter.cpp - ELF File Writer -----------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file implements ELF object file writer information.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "llvm/MC/MCELFObjectWriter.h"
15 #include "llvm/ADT/STLExtras.h"
16 #include "llvm/ADT/SmallPtrSet.h"
17 #include "llvm/ADT/SmallString.h"
18 #include "llvm/ADT/StringMap.h"
19 #include "llvm/MC/MCAsmBackend.h"
20 #include "llvm/MC/MCAsmLayout.h"
21 #include "llvm/MC/MCAssembler.h"
22 #include "llvm/MC/MCContext.h"
23 #include "llvm/MC/MCELF.h"
24 #include "llvm/MC/MCELFSymbolFlags.h"
25 #include "llvm/MC/MCExpr.h"
26 #include "llvm/MC/MCFixupKindInfo.h"
27 #include "llvm/MC/MCObjectWriter.h"
28 #include "llvm/MC/MCSectionELF.h"
29 #include "llvm/MC/MCValue.h"
30 #include "llvm/Support/Debug.h"
31 #include "llvm/Support/Endian.h"
32 #include "llvm/Support/ELF.h"
33 #include "llvm/Support/ErrorHandling.h"
34 #include <vector>
35 using namespace llvm;
36
37 #undef  DEBUG_TYPE
38 #define DEBUG_TYPE "reloc-info"
39
40 namespace {
41 class FragmentWriter {
42   bool IsLittleEndian;
43
44 public:
45   FragmentWriter(bool IsLittleEndian);
46   template <typename T> void write(MCDataFragment &F, T Val);
47 };
48
49 typedef DenseMap<const MCSectionELF *, uint32_t> SectionIndexMapTy;
50
51 class SymbolTableWriter {
52   MCAssembler &Asm;
53   FragmentWriter &FWriter;
54   bool Is64Bit;
55   SectionIndexMapTy &SectionIndexMap;
56
57   // The symbol .symtab fragment we are writting to.
58   MCDataFragment *SymtabF;
59
60   // .symtab_shndx fragment we are writting to.
61   MCDataFragment *ShndxF;
62
63   // The numbel of symbols written so far.
64   unsigned NumWritten;
65
66   void createSymtabShndx();
67
68   template <typename T> void write(MCDataFragment &F, T Value);
69
70 public:
71   SymbolTableWriter(MCAssembler &Asm, FragmentWriter &FWriter, bool Is64Bit,
72                     SectionIndexMapTy &SectionIndexMap,
73                     MCDataFragment *SymtabF);
74
75   void writeSymbol(uint32_t name, uint8_t info, uint64_t value, uint64_t size,
76                    uint8_t other, uint32_t shndx, bool Reserved);
77 };
78
79 class ELFObjectWriter : public MCObjectWriter {
80   FragmentWriter FWriter;
81
82   protected:
83
84     static bool isFixupKindPCRel(const MCAssembler &Asm, unsigned Kind);
85     static bool RelocNeedsGOT(MCSymbolRefExpr::VariantKind Variant);
86     static uint64_t SymbolValue(MCSymbolData &Data, const MCAsmLayout &Layout);
87     static bool isInSymtab(const MCAssembler &Asm, const MCSymbolData &Data,
88                            bool Used, bool Renamed);
89     static bool isLocal(const MCSymbolData &Data, bool isSignature,
90                         bool isUsedInReloc);
91     static bool IsELFMetaDataSection(const MCSectionData &SD);
92     static uint64_t DataSectionSize(const MCSectionData &SD);
93     static uint64_t GetSectionFileSize(const MCAsmLayout &Layout,
94                                        const MCSectionData &SD);
95     static uint64_t GetSectionAddressSize(const MCAsmLayout &Layout,
96                                           const MCSectionData &SD);
97
98     void WriteDataSectionData(MCAssembler &Asm,
99                               const MCAsmLayout &Layout,
100                               const MCSectionELF &Section);
101
102     /*static bool isFixupKindX86RIPRel(unsigned Kind) {
103       return Kind == X86::reloc_riprel_4byte ||
104         Kind == X86::reloc_riprel_4byte_movq_load;
105     }*/
106
107     /// ELFSymbolData - Helper struct for containing some precomputed
108     /// information on symbols.
109     struct ELFSymbolData {
110       MCSymbolData *SymbolData;
111       uint64_t StringIndex;
112       uint32_t SectionIndex;
113
114       // Support lexicographic sorting.
115       bool operator<(const ELFSymbolData &RHS) const {
116         return SymbolData->getSymbol().getName() <
117                RHS.SymbolData->getSymbol().getName();
118       }
119     };
120
121     /// The target specific ELF writer instance.
122     std::unique_ptr<MCELFObjectTargetWriter> TargetObjectWriter;
123
124     SmallPtrSet<const MCSymbol *, 16> UsedInReloc;
125     SmallPtrSet<const MCSymbol *, 16> WeakrefUsedInReloc;
126     DenseMap<const MCSymbol *, const MCSymbol *> Renames;
127
128     llvm::DenseMap<const MCSectionData*,
129                    std::vector<ELFRelocationEntry> > Relocations;
130     DenseMap<const MCSection*, uint64_t> SectionStringTableIndex;
131
132     /// @}
133     /// @name Symbol Table Data
134     /// @{
135
136     SmallString<256> StringTable;
137     std::vector<uint64_t> FileSymbolData;
138     std::vector<ELFSymbolData> LocalSymbolData;
139     std::vector<ELFSymbolData> ExternalSymbolData;
140     std::vector<ELFSymbolData> UndefinedSymbolData;
141
142     /// @}
143
144     bool NeedsGOT;
145
146     // This holds the symbol table index of the last local symbol.
147     unsigned LastLocalSymbolIndex;
148     // This holds the .strtab section index.
149     unsigned StringTableIndex;
150     // This holds the .symtab section index.
151     unsigned SymbolTableIndex;
152
153     unsigned ShstrtabIndex;
154
155
156     const MCSymbol *SymbolToReloc(const MCAssembler &Asm,
157                                   const MCValue &Target,
158                                   const MCFragment &F,
159                                   const MCFixup &Fixup,
160                                   bool IsPCRel) const;
161
162     // TargetObjectWriter wrappers.
163     const MCSymbol *ExplicitRelSym(const MCAssembler &Asm,
164                                    const MCValue &Target,
165                                    const MCFragment &F,
166                                    const MCFixup &Fixup,
167                                    bool IsPCRel) const {
168       return TargetObjectWriter->ExplicitRelSym(Asm, Target, F, Fixup, IsPCRel);
169     }
170     const MCSymbol *undefinedExplicitRelSym(const MCValue &Target,
171                                             const MCFixup &Fixup,
172                                             bool IsPCRel) const {
173       return TargetObjectWriter->undefinedExplicitRelSym(Target, Fixup,
174                                                          IsPCRel);
175     }
176
177     bool is64Bit() const { return TargetObjectWriter->is64Bit(); }
178     bool hasRelocationAddend() const {
179       return TargetObjectWriter->hasRelocationAddend();
180     }
181     unsigned GetRelocType(const MCValue &Target, const MCFixup &Fixup,
182                           bool IsPCRel, bool IsRelocWithSymbol,
183                           int64_t Addend) const {
184       return TargetObjectWriter->GetRelocType(Target, Fixup, IsPCRel,
185                                               IsRelocWithSymbol, Addend);
186     }
187
188   public:
189     ELFObjectWriter(MCELFObjectTargetWriter *MOTW, raw_ostream &_OS,
190                     bool IsLittleEndian)
191         : MCObjectWriter(_OS, IsLittleEndian), FWriter(IsLittleEndian),
192           TargetObjectWriter(MOTW), NeedsGOT(false) {}
193
194     virtual ~ELFObjectWriter();
195
196     void WriteWord(uint64_t W) {
197       if (is64Bit())
198         Write64(W);
199       else
200         Write32(W);
201     }
202
203     template <typename T> void write(MCDataFragment &F, T Value) {
204       FWriter.write(F, Value);
205     }
206
207     void WriteHeader(const MCAssembler &Asm,
208                      uint64_t SectionDataSize,
209                      unsigned NumberOfSections);
210
211     void WriteSymbol(SymbolTableWriter &Writer, ELFSymbolData &MSD,
212                      const MCAsmLayout &Layout);
213
214     void WriteSymbolTable(MCDataFragment *SymtabF, MCAssembler &Asm,
215                           const MCAsmLayout &Layout,
216                           SectionIndexMapTy &SectionIndexMap);
217
218     void RecordRelocation(const MCAssembler &Asm, const MCAsmLayout &Layout,
219                           const MCFragment *Fragment, const MCFixup &Fixup,
220                           MCValue Target, uint64_t &FixedValue) override;
221
222     uint64_t getSymbolIndexInSymbolTable(const MCAssembler &Asm,
223                                          const MCSymbol *S);
224
225     // Map from a group section to the signature symbol
226     typedef DenseMap<const MCSectionELF*, const MCSymbol*> GroupMapTy;
227     // Map from a signature symbol to the group section
228     typedef DenseMap<const MCSymbol*, const MCSectionELF*> RevGroupMapTy;
229     // Map from a section to the section with the relocations
230     typedef DenseMap<const MCSectionELF*, const MCSectionELF*> RelMapTy;
231     // Map from a section to its offset
232     typedef DenseMap<const MCSectionELF*, uint64_t> SectionOffsetMapTy;
233
234     /// Compute the symbol table data
235     ///
236     /// \param Asm - The assembler.
237     /// \param SectionIndexMap - Maps a section to its index.
238     /// \param RevGroupMap - Maps a signature symbol to the group section.
239     /// \param NumRegularSections - Number of non-relocation sections.
240     void computeSymbolTable(MCAssembler &Asm, const MCAsmLayout &Layout,
241                             const SectionIndexMapTy &SectionIndexMap,
242                             RevGroupMapTy RevGroupMap,
243                             unsigned NumRegularSections);
244
245     void ComputeIndexMap(MCAssembler &Asm,
246                          SectionIndexMapTy &SectionIndexMap,
247                          const RelMapTy &RelMap);
248
249     void CreateRelocationSections(MCAssembler &Asm, MCAsmLayout &Layout,
250                                   RelMapTy &RelMap);
251
252     void WriteRelocations(MCAssembler &Asm, MCAsmLayout &Layout,
253                           const RelMapTy &RelMap);
254
255     void CreateMetadataSections(MCAssembler &Asm, MCAsmLayout &Layout,
256                                 SectionIndexMapTy &SectionIndexMap,
257                                 const RelMapTy &RelMap);
258
259     // Create the sections that show up in the symbol table. Currently
260     // those are the .note.GNU-stack section and the group sections.
261     void CreateIndexedSections(MCAssembler &Asm, MCAsmLayout &Layout,
262                                GroupMapTy &GroupMap,
263                                RevGroupMapTy &RevGroupMap,
264                                SectionIndexMapTy &SectionIndexMap,
265                                const RelMapTy &RelMap);
266
267     void ExecutePostLayoutBinding(MCAssembler &Asm,
268                                   const MCAsmLayout &Layout) override;
269
270     void WriteSectionHeader(MCAssembler &Asm, const GroupMapTy &GroupMap,
271                             const MCAsmLayout &Layout,
272                             const SectionIndexMapTy &SectionIndexMap,
273                             const SectionOffsetMapTy &SectionOffsetMap);
274
275     void ComputeSectionOrder(MCAssembler &Asm,
276                              std::vector<const MCSectionELF*> &Sections);
277
278     void WriteSecHdrEntry(uint32_t Name, uint32_t Type, uint64_t Flags,
279                           uint64_t Address, uint64_t Offset,
280                           uint64_t Size, uint32_t Link, uint32_t Info,
281                           uint64_t Alignment, uint64_t EntrySize);
282
283     void WriteRelocationsFragment(const MCAssembler &Asm,
284                                   MCDataFragment *F,
285                                   const MCSectionData *SD);
286
287     bool
288     IsSymbolRefDifferenceFullyResolvedImpl(const MCAssembler &Asm,
289                                            const MCSymbolData &DataA,
290                                            const MCFragment &FB,
291                                            bool InSet,
292                                            bool IsPCRel) const override;
293
294     void WriteObject(MCAssembler &Asm, const MCAsmLayout &Layout) override;
295     void WriteSection(MCAssembler &Asm,
296                       const SectionIndexMapTy &SectionIndexMap,
297                       uint32_t GroupSymbolIndex,
298                       uint64_t Offset, uint64_t Size, uint64_t Alignment,
299                       const MCSectionELF &Section);
300   };
301 }
302
303 FragmentWriter::FragmentWriter(bool IsLittleEndian)
304     : IsLittleEndian(IsLittleEndian) {}
305
306 template <typename T> void FragmentWriter::write(MCDataFragment &F, T Val) {
307   if (IsLittleEndian)
308     Val = support::endian::byte_swap<T, support::little>(Val);
309   else
310     Val = support::endian::byte_swap<T, support::big>(Val);
311   const char *Start = (const char *)&Val;
312   F.getContents().append(Start, Start + sizeof(T));
313 }
314
315 void SymbolTableWriter::createSymtabShndx() {
316   if (ShndxF)
317     return;
318
319   MCContext &Ctx = Asm.getContext();
320   const MCSectionELF *SymtabShndxSection =
321       Ctx.getELFSection(".symtab_shndxr", ELF::SHT_SYMTAB_SHNDX, 0,
322                         SectionKind::getReadOnly(), 4, "");
323   MCSectionData *SymtabShndxSD =
324       &Asm.getOrCreateSectionData(*SymtabShndxSection);
325   SymtabShndxSD->setAlignment(4);
326   ShndxF = new MCDataFragment(SymtabShndxSD);
327   unsigned Index = SectionIndexMap.size() + 1;
328   SectionIndexMap[SymtabShndxSection] = Index;
329
330   for (unsigned I = 0; I < NumWritten; ++I)
331     write(*ShndxF, uint32_t(0));
332 }
333
334 template <typename T>
335 void SymbolTableWriter::write(MCDataFragment &F, T Value) {
336   FWriter.write(F, Value);
337 }
338
339 SymbolTableWriter::SymbolTableWriter(MCAssembler &Asm, FragmentWriter &FWriter,
340                                      bool Is64Bit,
341                                      SectionIndexMapTy &SectionIndexMap,
342                                      MCDataFragment *SymtabF)
343     : Asm(Asm), FWriter(FWriter), Is64Bit(Is64Bit),
344       SectionIndexMap(SectionIndexMap), SymtabF(SymtabF), ShndxF(nullptr),
345       NumWritten(0) {}
346
347 void SymbolTableWriter::writeSymbol(uint32_t name, uint8_t info, uint64_t value,
348                                     uint64_t size, uint8_t other,
349                                     uint32_t shndx, bool Reserved) {
350   bool LargeIndex = shndx >= ELF::SHN_LORESERVE && !Reserved;
351
352   if (LargeIndex)
353     createSymtabShndx();
354
355   if (ShndxF) {
356     if (LargeIndex)
357       write(*ShndxF, shndx);
358     else
359       write(*ShndxF, uint32_t(0));
360   }
361
362   uint16_t Index = LargeIndex ? uint16_t(ELF::SHN_XINDEX) : shndx;
363
364   raw_svector_ostream OS(SymtabF->getContents());
365
366   if (Is64Bit) {
367     write(*SymtabF, name);  // st_name
368     write(*SymtabF, info);  // st_info
369     write(*SymtabF, other); // st_other
370     write(*SymtabF, Index); // st_shndx
371     write(*SymtabF, value); // st_value
372     write(*SymtabF, size);  // st_size
373   } else {
374     write(*SymtabF, name);            // st_name
375     write(*SymtabF, uint32_t(value)); // st_value
376     write(*SymtabF, uint32_t(size));  // st_size
377     write(*SymtabF, info);            // st_info
378     write(*SymtabF, other);           // st_other
379     write(*SymtabF, Index);           // st_shndx
380   }
381
382   ++NumWritten;
383 }
384
385 bool ELFObjectWriter::isFixupKindPCRel(const MCAssembler &Asm, unsigned Kind) {
386   const MCFixupKindInfo &FKI =
387     Asm.getBackend().getFixupKindInfo((MCFixupKind) Kind);
388
389   return FKI.Flags & MCFixupKindInfo::FKF_IsPCRel;
390 }
391
392 bool ELFObjectWriter::RelocNeedsGOT(MCSymbolRefExpr::VariantKind Variant) {
393   switch (Variant) {
394   default:
395     return false;
396   case MCSymbolRefExpr::VK_GOT:
397   case MCSymbolRefExpr::VK_PLT:
398   case MCSymbolRefExpr::VK_GOTPCREL:
399   case MCSymbolRefExpr::VK_GOTOFF:
400   case MCSymbolRefExpr::VK_TPOFF:
401   case MCSymbolRefExpr::VK_TLSGD:
402   case MCSymbolRefExpr::VK_GOTTPOFF:
403   case MCSymbolRefExpr::VK_INDNTPOFF:
404   case MCSymbolRefExpr::VK_NTPOFF:
405   case MCSymbolRefExpr::VK_GOTNTPOFF:
406   case MCSymbolRefExpr::VK_TLSLDM:
407   case MCSymbolRefExpr::VK_DTPOFF:
408   case MCSymbolRefExpr::VK_TLSLD:
409     return true;
410   }
411 }
412
413 ELFObjectWriter::~ELFObjectWriter()
414 {}
415
416 // Emit the ELF header.
417 void ELFObjectWriter::WriteHeader(const MCAssembler &Asm,
418                                   uint64_t SectionDataSize,
419                                   unsigned NumberOfSections) {
420   // ELF Header
421   // ----------
422   //
423   // Note
424   // ----
425   // emitWord method behaves differently for ELF32 and ELF64, writing
426   // 4 bytes in the former and 8 in the latter.
427
428   Write8(0x7f); // e_ident[EI_MAG0]
429   Write8('E');  // e_ident[EI_MAG1]
430   Write8('L');  // e_ident[EI_MAG2]
431   Write8('F');  // e_ident[EI_MAG3]
432
433   Write8(is64Bit() ? ELF::ELFCLASS64 : ELF::ELFCLASS32); // e_ident[EI_CLASS]
434
435   // e_ident[EI_DATA]
436   Write8(isLittleEndian() ? ELF::ELFDATA2LSB : ELF::ELFDATA2MSB);
437
438   Write8(ELF::EV_CURRENT);        // e_ident[EI_VERSION]
439   // e_ident[EI_OSABI]
440   Write8(TargetObjectWriter->getOSABI());
441   Write8(0);                  // e_ident[EI_ABIVERSION]
442
443   WriteZeros(ELF::EI_NIDENT - ELF::EI_PAD);
444
445   Write16(ELF::ET_REL);             // e_type
446
447   Write16(TargetObjectWriter->getEMachine()); // e_machine = target
448
449   Write32(ELF::EV_CURRENT);         // e_version
450   WriteWord(0);                    // e_entry, no entry point in .o file
451   WriteWord(0);                    // e_phoff, no program header for .o
452   WriteWord(SectionDataSize + (is64Bit() ? sizeof(ELF::Elf64_Ehdr) :
453             sizeof(ELF::Elf32_Ehdr)));  // e_shoff = sec hdr table off in bytes
454
455   // e_flags = whatever the target wants
456   Write32(Asm.getELFHeaderEFlags());
457
458   // e_ehsize = ELF header size
459   Write16(is64Bit() ? sizeof(ELF::Elf64_Ehdr) : sizeof(ELF::Elf32_Ehdr));
460
461   Write16(0);                  // e_phentsize = prog header entry size
462   Write16(0);                  // e_phnum = # prog header entries = 0
463
464   // e_shentsize = Section header entry size
465   Write16(is64Bit() ? sizeof(ELF::Elf64_Shdr) : sizeof(ELF::Elf32_Shdr));
466
467   // e_shnum     = # of section header ents
468   if (NumberOfSections >= ELF::SHN_LORESERVE)
469     Write16(ELF::SHN_UNDEF);
470   else
471     Write16(NumberOfSections);
472
473   // e_shstrndx  = Section # of '.shstrtab'
474   if (ShstrtabIndex >= ELF::SHN_LORESERVE)
475     Write16(ELF::SHN_XINDEX);
476   else
477     Write16(ShstrtabIndex);
478 }
479
480 uint64_t ELFObjectWriter::SymbolValue(MCSymbolData &OrigData,
481                                       const MCAsmLayout &Layout) {
482   MCSymbolData *Data = &OrigData;
483   if (Data->isCommon() && Data->isExternal())
484     return Data->getCommonAlignment();
485
486   const MCSymbol *Symbol = &Data->getSymbol();
487   bool IsThumbFunc = OrigData.getFlags() & ELF_Other_ThumbFunc;
488
489   uint64_t Res = 0;
490   if (Symbol->isVariable()) {
491     const MCExpr *Expr = Symbol->getVariableValue();
492     MCValue Value;
493     if (!Expr->EvaluateAsRelocatable(Value, &Layout))
494       llvm_unreachable("Invalid expression");
495
496     assert(!Value.getSymB());
497
498     Res = Value.getConstant();
499
500     if (const MCSymbolRefExpr *A = Value.getSymA()) {
501       Symbol = &A->getSymbol();
502       Data = &Layout.getAssembler().getSymbolData(*Symbol);
503     } else {
504       Symbol = 0;
505       Data = 0;
506     }
507   }
508
509   if (IsThumbFunc)
510     Res |= 1;
511
512   if (!Symbol || !Symbol->isInSection())
513     return Res;
514
515   Res += Layout.getSymbolOffset(Data);
516
517   return Res;
518 }
519
520 void ELFObjectWriter::ExecutePostLayoutBinding(MCAssembler &Asm,
521                                                const MCAsmLayout &Layout) {
522   // The presence of symbol versions causes undefined symbols and
523   // versions declared with @@@ to be renamed.
524
525   for (MCAssembler::symbol_iterator it = Asm.symbol_begin(),
526          ie = Asm.symbol_end(); it != ie; ++it) {
527     const MCSymbol &Alias = it->getSymbol();
528     const MCSymbol &Symbol = Alias.AliasedSymbol();
529     MCSymbolData &SD = Asm.getSymbolData(Symbol);
530
531     // Not an alias.
532     if (&Symbol == &Alias)
533       continue;
534
535     StringRef AliasName = Alias.getName();
536     size_t Pos = AliasName.find('@');
537     if (Pos == StringRef::npos)
538       continue;
539
540     // Aliases defined with .symvar copy the binding from the symbol they alias.
541     // This is the first place we are able to copy this information.
542     it->setExternal(SD.isExternal());
543     MCELF::SetBinding(*it, MCELF::GetBinding(SD));
544
545     StringRef Rest = AliasName.substr(Pos);
546     if (!Symbol.isUndefined() && !Rest.startswith("@@@"))
547       continue;
548
549     // FIXME: produce a better error message.
550     if (Symbol.isUndefined() && Rest.startswith("@@") &&
551         !Rest.startswith("@@@"))
552       report_fatal_error("A @@ version cannot be undefined");
553
554     Renames.insert(std::make_pair(&Symbol, &Alias));
555   }
556 }
557
558 static uint8_t mergeTypeForSet(uint8_t origType, uint8_t newType) {
559   uint8_t Type = newType;
560
561   // Propagation rules:
562   // IFUNC > FUNC > OBJECT > NOTYPE
563   // TLS_OBJECT > OBJECT > NOTYPE
564   //
565   // dont let the new type degrade the old type
566   switch (origType) {
567   default:
568     break;
569   case ELF::STT_GNU_IFUNC:
570     if (Type == ELF::STT_FUNC || Type == ELF::STT_OBJECT ||
571         Type == ELF::STT_NOTYPE || Type == ELF::STT_TLS)
572       Type = ELF::STT_GNU_IFUNC;
573     break;
574   case ELF::STT_FUNC:
575     if (Type == ELF::STT_OBJECT || Type == ELF::STT_NOTYPE ||
576         Type == ELF::STT_TLS)
577       Type = ELF::STT_FUNC;
578     break;
579   case ELF::STT_OBJECT:
580     if (Type == ELF::STT_NOTYPE)
581       Type = ELF::STT_OBJECT;
582     break;
583   case ELF::STT_TLS:
584     if (Type == ELF::STT_OBJECT || Type == ELF::STT_NOTYPE ||
585         Type == ELF::STT_GNU_IFUNC || Type == ELF::STT_FUNC)
586       Type = ELF::STT_TLS;
587     break;
588   }
589
590   return Type;
591 }
592
593 static const MCSymbol *getBaseSymbol(const MCAsmLayout &Layout,
594                                      const MCSymbol &Symbol) {
595   if (!Symbol.isVariable())
596     return &Symbol;
597
598   const MCExpr *Expr = Symbol.getVariableValue();
599   MCValue Value;
600   if (!Expr->EvaluateAsRelocatable(Value, &Layout))
601     llvm_unreachable("Invalid Expression");
602   assert(!Value.getSymB());
603   const MCSymbolRefExpr *A = Value.getSymA();
604   if (!A)
605     return nullptr;
606   return getBaseSymbol(Layout, A->getSymbol());
607 }
608
609 void ELFObjectWriter::WriteSymbol(SymbolTableWriter &Writer, ELFSymbolData &MSD,
610                                   const MCAsmLayout &Layout) {
611   MCSymbolData &OrigData = *MSD.SymbolData;
612   const MCSymbol *Base = getBaseSymbol(Layout, OrigData.getSymbol());
613
614   // This has to be in sync with when computeSymbolTable uses SHN_ABS or
615   // SHN_COMMON.
616   bool IsReserved = !Base || OrigData.isCommon();
617
618   // Binding and Type share the same byte as upper and lower nibbles
619   uint8_t Binding = MCELF::GetBinding(OrigData);
620   uint8_t Type = MCELF::GetType(OrigData);
621   MCSymbolData *BaseSD = nullptr;
622   if (Base) {
623     BaseSD = &Layout.getAssembler().getSymbolData(*Base);
624     Type = mergeTypeForSet(Type, MCELF::GetType(*BaseSD));
625   }
626   if (OrigData.getFlags() & ELF_Other_ThumbFunc)
627     Type = ELF::STT_FUNC;
628   uint8_t Info = (Binding << ELF_STB_Shift) | (Type << ELF_STT_Shift);
629
630   // Other and Visibility share the same byte with Visibility using the lower
631   // 2 bits
632   uint8_t Visibility = MCELF::GetVisibility(OrigData);
633   uint8_t Other = MCELF::getOther(OrigData) << (ELF_STO_Shift - ELF_STV_Shift);
634   Other |= Visibility;
635
636   uint64_t Value = SymbolValue(OrigData, Layout);
637   if (OrigData.getFlags() & ELF_Other_ThumbFunc)
638     Value |= 1;
639   uint64_t Size = 0;
640
641   const MCExpr *ESize = OrigData.getSize();
642   if (!ESize && Base)
643     ESize = BaseSD->getSize();
644
645   if (ESize) {
646     int64_t Res;
647     if (!ESize->EvaluateAsAbsolute(Res, Layout))
648       report_fatal_error("Size expression must be absolute.");
649     Size = Res;
650   }
651
652   // Write out the symbol table entry
653   Writer.writeSymbol(MSD.StringIndex, Info, Value, Size, Other,
654                      MSD.SectionIndex, IsReserved);
655 }
656
657 void ELFObjectWriter::WriteSymbolTable(MCDataFragment *SymtabF,
658                                        MCAssembler &Asm,
659                                        const MCAsmLayout &Layout,
660                                        SectionIndexMapTy &SectionIndexMap) {
661   // The string table must be emitted first because we need the index
662   // into the string table for all the symbol names.
663   assert(StringTable.size() && "Missing string table");
664
665   // FIXME: Make sure the start of the symbol table is aligned.
666
667   SymbolTableWriter Writer(Asm, FWriter, is64Bit(), SectionIndexMap, SymtabF);
668
669   // The first entry is the undefined symbol entry.
670   Writer.writeSymbol(0, 0, 0, 0, 0, 0, false);
671
672   for (unsigned i = 0, e = FileSymbolData.size(); i != e; ++i) {
673     Writer.writeSymbol(FileSymbolData[i], ELF::STT_FILE | ELF::STB_LOCAL, 0, 0,
674                        ELF::STV_DEFAULT, ELF::SHN_ABS, true);
675   }
676
677   // Write the symbol table entries.
678   LastLocalSymbolIndex = FileSymbolData.size() + LocalSymbolData.size() + 1;
679
680   for (unsigned i = 0, e = LocalSymbolData.size(); i != e; ++i) {
681     ELFSymbolData &MSD = LocalSymbolData[i];
682     WriteSymbol(Writer, MSD, Layout);
683   }
684
685   // Write out a symbol table entry for each regular section.
686   for (MCAssembler::const_iterator i = Asm.begin(), e = Asm.end(); i != e;
687        ++i) {
688     const MCSectionELF &Section =
689       static_cast<const MCSectionELF&>(i->getSection());
690     if (Section.getType() == ELF::SHT_RELA ||
691         Section.getType() == ELF::SHT_REL ||
692         Section.getType() == ELF::SHT_STRTAB ||
693         Section.getType() == ELF::SHT_SYMTAB ||
694         Section.getType() == ELF::SHT_SYMTAB_SHNDX)
695       continue;
696     Writer.writeSymbol(0, ELF::STT_SECTION, 0, 0, ELF::STV_DEFAULT,
697                        SectionIndexMap.lookup(&Section), false);
698     LastLocalSymbolIndex++;
699   }
700
701   for (unsigned i = 0, e = ExternalSymbolData.size(); i != e; ++i) {
702     ELFSymbolData &MSD = ExternalSymbolData[i];
703     MCSymbolData &Data = *MSD.SymbolData;
704     assert(((Data.getFlags() & ELF_STB_Global) ||
705             (Data.getFlags() & ELF_STB_Weak)) &&
706            "External symbol requires STB_GLOBAL or STB_WEAK flag");
707     WriteSymbol(Writer, MSD, Layout);
708     if (MCELF::GetBinding(Data) == ELF::STB_LOCAL)
709       LastLocalSymbolIndex++;
710   }
711
712   for (unsigned i = 0, e = UndefinedSymbolData.size(); i != e; ++i) {
713     ELFSymbolData &MSD = UndefinedSymbolData[i];
714     MCSymbolData &Data = *MSD.SymbolData;
715     WriteSymbol(Writer, MSD, Layout);
716     if (MCELF::GetBinding(Data) == ELF::STB_LOCAL)
717       LastLocalSymbolIndex++;
718   }
719 }
720
721 const MCSymbol *ELFObjectWriter::SymbolToReloc(const MCAssembler &Asm,
722                                                const MCValue &Target,
723                                                const MCFragment &F,
724                                                const MCFixup &Fixup,
725                                                bool IsPCRel) const {
726   const MCSymbol &Symbol = Target.getSymA()->getSymbol();
727   const MCSymbol &ASymbol = Symbol.AliasedSymbol();
728   const MCSymbol *Renamed = Renames.lookup(&Symbol);
729   const MCSymbolData &SD = Asm.getSymbolData(Symbol);
730
731   if (ASymbol.isUndefined()) {
732     if (Renamed)
733       return Renamed;
734     return undefinedExplicitRelSym(Target, Fixup, IsPCRel);
735   }
736
737   if (SD.isExternal()) {
738     if (Renamed)
739       return Renamed;
740     return &Symbol;
741   }
742
743   const MCSectionELF &Section =
744     static_cast<const MCSectionELF&>(ASymbol.getSection());
745   const SectionKind secKind = Section.getKind();
746
747   if (secKind.isBSS())
748     return ExplicitRelSym(Asm, Target, F, Fixup, IsPCRel);
749
750   if (secKind.isThreadLocal()) {
751     if (Renamed)
752       return Renamed;
753     return &Symbol;
754   }
755
756   MCSymbolRefExpr::VariantKind Kind = Target.getSymA()->getKind();
757   const MCSectionELF &Sec2 =
758     static_cast<const MCSectionELF&>(F.getParent()->getSection());
759
760   if (&Sec2 != &Section &&
761       (Kind == MCSymbolRefExpr::VK_PLT ||
762        Kind == MCSymbolRefExpr::VK_GOTPCREL ||
763        Kind == MCSymbolRefExpr::VK_GOTOFF)) {
764     if (Renamed)
765       return Renamed;
766     return &Symbol;
767   }
768
769   if (Section.getFlags() & ELF::SHF_MERGE) {
770     if (Target.getConstant() == 0)
771       return ExplicitRelSym(Asm, Target, F, Fixup, IsPCRel);
772     if (Renamed)
773       return Renamed;
774     return &Symbol;
775   }
776
777   return ExplicitRelSym(Asm, Target, F, Fixup, IsPCRel);
778
779 }
780
781
782 void ELFObjectWriter::RecordRelocation(const MCAssembler &Asm,
783                                        const MCAsmLayout &Layout,
784                                        const MCFragment *Fragment,
785                                        const MCFixup &Fixup,
786                                        MCValue Target,
787                                        uint64_t &FixedValue) {
788   int64_t Addend = 0;
789   int Index = 0;
790   int64_t Value = Target.getConstant();
791   const MCSymbol *RelocSymbol = NULL;
792
793   bool IsPCRel = isFixupKindPCRel(Asm, Fixup.getKind());
794   if (!Target.isAbsolute()) {
795     const MCSymbol &Symbol = Target.getSymA()->getSymbol();
796     const MCSymbol &ASymbol = Symbol.AliasedSymbol();
797     RelocSymbol = SymbolToReloc(Asm, Target, *Fragment, Fixup, IsPCRel);
798
799     if (const MCSymbolRefExpr *RefB = Target.getSymB()) {
800       const MCSymbol &SymbolB = RefB->getSymbol();
801       MCSymbolData &SDB = Asm.getSymbolData(SymbolB);
802       IsPCRel = true;
803
804       if (!SDB.getFragment())
805         Asm.getContext().FatalError(
806             Fixup.getLoc(),
807             Twine("symbol '") + SymbolB.getName() +
808                 "' can not be undefined in a subtraction expression");
809
810       // Offset of the symbol in the section
811       int64_t a = Layout.getSymbolOffset(&SDB);
812
813       // Offset of the relocation in the section
814       int64_t b = Layout.getFragmentOffset(Fragment) + Fixup.getOffset();
815       Value += b - a;
816     }
817
818     if (!RelocSymbol) {
819       MCSymbolData &SD = Asm.getSymbolData(ASymbol);
820       MCFragment *F = SD.getFragment();
821
822       if (F) {
823         Index = F->getParent()->getOrdinal() + 1;
824         // Offset of the symbol in the section
825         Value += Layout.getSymbolOffset(&SD);
826       } else {
827         Index = 0;
828       }
829     } else {
830       if (Target.getSymA()->getKind() == MCSymbolRefExpr::VK_WEAKREF)
831         WeakrefUsedInReloc.insert(RelocSymbol);
832       else
833         UsedInReloc.insert(RelocSymbol);
834       Index = -1;
835     }
836     Addend = Value;
837     if (hasRelocationAddend())
838       Value = 0;
839   }
840
841   FixedValue = Value;
842   unsigned Type = GetRelocType(Target, Fixup, IsPCRel,
843                                (RelocSymbol != 0), Addend);
844   MCSymbolRefExpr::VariantKind Modifier = Target.isAbsolute() ?
845     MCSymbolRefExpr::VK_None : Target.getSymA()->getKind();
846   if (RelocNeedsGOT(Modifier))
847     NeedsGOT = true;
848
849   uint64_t RelocOffset = Layout.getFragmentOffset(Fragment) +
850     Fixup.getOffset();
851
852   if (!hasRelocationAddend())
853     Addend = 0;
854
855   if (is64Bit())
856     assert(isInt<64>(Addend));
857   else
858     assert(isInt<32>(Addend));
859
860   ELFRelocationEntry ERE(RelocOffset, Index, Type, RelocSymbol, Addend, Fixup);
861   Relocations[Fragment->getParent()].push_back(ERE);
862 }
863
864
865 uint64_t
866 ELFObjectWriter::getSymbolIndexInSymbolTable(const MCAssembler &Asm,
867                                              const MCSymbol *S) {
868   MCSymbolData &SD = Asm.getSymbolData(*S);
869   return SD.getIndex();
870 }
871
872 bool ELFObjectWriter::isInSymtab(const MCAssembler &Asm,
873                                  const MCSymbolData &Data,
874                                  bool Used, bool Renamed) {
875   const MCSymbol &Symbol = Data.getSymbol();
876   if (Symbol.isVariable()) {
877     const MCExpr *Expr = Symbol.getVariableValue();
878     if (const MCSymbolRefExpr *Ref = dyn_cast<MCSymbolRefExpr>(Expr)) {
879       if (Ref->getKind() == MCSymbolRefExpr::VK_WEAKREF)
880         return false;
881     }
882   }
883
884   if (Used)
885     return true;
886
887   if (Renamed)
888     return false;
889
890   if (Symbol.getName() == "_GLOBAL_OFFSET_TABLE_")
891     return true;
892
893   const MCSymbol &A = Symbol.AliasedSymbol();
894   if (Symbol.isVariable() && !A.isVariable() && A.isUndefined())
895     return false;
896
897   bool IsGlobal = MCELF::GetBinding(Data) == ELF::STB_GLOBAL;
898   if (!Symbol.isVariable() && Symbol.isUndefined() && !IsGlobal)
899     return false;
900
901   if (Symbol.isTemporary())
902     return false;
903
904   return true;
905 }
906
907 bool ELFObjectWriter::isLocal(const MCSymbolData &Data, bool isSignature,
908                               bool isUsedInReloc) {
909   if (Data.isExternal())
910     return false;
911
912   const MCSymbol &Symbol = Data.getSymbol();
913   const MCSymbol &RefSymbol = Symbol.AliasedSymbol();
914
915   if (RefSymbol.isUndefined() && !RefSymbol.isVariable()) {
916     if (isSignature && !isUsedInReloc)
917       return true;
918
919     return false;
920   }
921
922   return true;
923 }
924
925 void ELFObjectWriter::ComputeIndexMap(MCAssembler &Asm,
926                                       SectionIndexMapTy &SectionIndexMap,
927                                       const RelMapTy &RelMap) {
928   unsigned Index = 1;
929   for (MCAssembler::iterator it = Asm.begin(),
930          ie = Asm.end(); it != ie; ++it) {
931     const MCSectionELF &Section =
932       static_cast<const MCSectionELF &>(it->getSection());
933     if (Section.getType() != ELF::SHT_GROUP)
934       continue;
935     SectionIndexMap[&Section] = Index++;
936   }
937
938   for (MCAssembler::iterator it = Asm.begin(),
939          ie = Asm.end(); it != ie; ++it) {
940     const MCSectionELF &Section =
941       static_cast<const MCSectionELF &>(it->getSection());
942     if (Section.getType() == ELF::SHT_GROUP ||
943         Section.getType() == ELF::SHT_REL ||
944         Section.getType() == ELF::SHT_RELA)
945       continue;
946     SectionIndexMap[&Section] = Index++;
947     const MCSectionELF *RelSection = RelMap.lookup(&Section);
948     if (RelSection)
949       SectionIndexMap[RelSection] = Index++;
950   }
951 }
952
953 void
954 ELFObjectWriter::computeSymbolTable(MCAssembler &Asm, const MCAsmLayout &Layout,
955                                     const SectionIndexMapTy &SectionIndexMap,
956                                     RevGroupMapTy RevGroupMap,
957                                     unsigned NumRegularSections) {
958   // FIXME: Is this the correct place to do this?
959   // FIXME: Why is an undefined reference to _GLOBAL_OFFSET_TABLE_ needed?
960   if (NeedsGOT) {
961     StringRef Name = "_GLOBAL_OFFSET_TABLE_";
962     MCSymbol *Sym = Asm.getContext().GetOrCreateSymbol(Name);
963     MCSymbolData &Data = Asm.getOrCreateSymbolData(*Sym);
964     Data.setExternal(true);
965     MCELF::SetBinding(Data, ELF::STB_GLOBAL);
966   }
967
968   // Index 0 is always the empty string.
969   StringMap<uint64_t> StringIndexMap;
970   StringTable += '\x00';
971
972   // FIXME: We could optimize suffixes in strtab in the same way we
973   // optimize them in shstrtab.
974
975   for (MCAssembler::const_file_name_iterator it = Asm.file_names_begin(),
976                                             ie = Asm.file_names_end();
977                                             it != ie;
978                                             ++it) {
979     StringRef Name = *it;
980     uint64_t &Entry = StringIndexMap[Name];
981     if (!Entry) {
982       Entry = StringTable.size();
983       StringTable += Name;
984       StringTable += '\x00';
985     }
986     FileSymbolData.push_back(Entry);
987   }
988
989   // Add the data for the symbols.
990   for (MCAssembler::symbol_iterator it = Asm.symbol_begin(),
991          ie = Asm.symbol_end(); it != ie; ++it) {
992     const MCSymbol &Symbol = it->getSymbol();
993
994     bool Used = UsedInReloc.count(&Symbol);
995     bool WeakrefUsed = WeakrefUsedInReloc.count(&Symbol);
996     bool isSignature = RevGroupMap.count(&Symbol);
997
998     if (!isInSymtab(Asm, *it,
999                     Used || WeakrefUsed || isSignature,
1000                     Renames.count(&Symbol)))
1001       continue;
1002
1003     ELFSymbolData MSD;
1004     MSD.SymbolData = it;
1005     const MCSymbol *BaseSymbol = getBaseSymbol(Layout, Symbol);
1006
1007     // Undefined symbols are global, but this is the first place we
1008     // are able to set it.
1009     bool Local = isLocal(*it, isSignature, Used);
1010     if (!Local && MCELF::GetBinding(*it) == ELF::STB_LOCAL) {
1011       assert(BaseSymbol);
1012       MCSymbolData &SD = Asm.getSymbolData(*BaseSymbol);
1013       MCELF::SetBinding(*it, ELF::STB_GLOBAL);
1014       MCELF::SetBinding(SD, ELF::STB_GLOBAL);
1015     }
1016
1017     if (!BaseSymbol) {
1018       MSD.SectionIndex = ELF::SHN_ABS;
1019     } else if (it->isCommon()) {
1020       assert(!Local);
1021       MSD.SectionIndex = ELF::SHN_COMMON;
1022     } else if (BaseSymbol->isUndefined()) {
1023       if (isSignature && !Used)
1024         MSD.SectionIndex = SectionIndexMap.lookup(RevGroupMap[&Symbol]);
1025       else
1026         MSD.SectionIndex = ELF::SHN_UNDEF;
1027       if (!Used && WeakrefUsed)
1028         MCELF::SetBinding(*it, ELF::STB_WEAK);
1029     } else {
1030       const MCSectionELF &Section =
1031         static_cast<const MCSectionELF&>(BaseSymbol->getSection());
1032       MSD.SectionIndex = SectionIndexMap.lookup(&Section);
1033       assert(MSD.SectionIndex && "Invalid section index!");
1034     }
1035
1036     // The @@@ in symbol version is replaced with @ in undefined symbols and
1037     // @@ in defined ones.
1038     StringRef Name = Symbol.getName();
1039     SmallString<32> Buf;
1040
1041     size_t Pos = Name.find("@@@");
1042     if (Pos != StringRef::npos) {
1043       Buf += Name.substr(0, Pos);
1044       unsigned Skip = MSD.SectionIndex == ELF::SHN_UNDEF ? 2 : 1;
1045       Buf += Name.substr(Pos + Skip);
1046       Name = Buf;
1047     }
1048
1049     uint64_t &Entry = StringIndexMap[Name];
1050     if (!Entry) {
1051       Entry = StringTable.size();
1052       StringTable += Name;
1053       StringTable += '\x00';
1054     }
1055     MSD.StringIndex = Entry;
1056     if (MSD.SectionIndex == ELF::SHN_UNDEF)
1057       UndefinedSymbolData.push_back(MSD);
1058     else if (Local)
1059       LocalSymbolData.push_back(MSD);
1060     else
1061       ExternalSymbolData.push_back(MSD);
1062   }
1063
1064   // Symbols are required to be in lexicographic order.
1065   array_pod_sort(LocalSymbolData.begin(), LocalSymbolData.end());
1066   array_pod_sort(ExternalSymbolData.begin(), ExternalSymbolData.end());
1067   array_pod_sort(UndefinedSymbolData.begin(), UndefinedSymbolData.end());
1068
1069   // Set the symbol indices. Local symbols must come before all other
1070   // symbols with non-local bindings.
1071   unsigned Index = FileSymbolData.size() + 1;
1072   for (unsigned i = 0, e = LocalSymbolData.size(); i != e; ++i)
1073     LocalSymbolData[i].SymbolData->setIndex(Index++);
1074
1075   Index += NumRegularSections;
1076
1077   for (unsigned i = 0, e = ExternalSymbolData.size(); i != e; ++i)
1078     ExternalSymbolData[i].SymbolData->setIndex(Index++);
1079   for (unsigned i = 0, e = UndefinedSymbolData.size(); i != e; ++i)
1080     UndefinedSymbolData[i].SymbolData->setIndex(Index++);
1081 }
1082
1083 void ELFObjectWriter::CreateRelocationSections(MCAssembler &Asm,
1084                                                MCAsmLayout &Layout,
1085                                                RelMapTy &RelMap) {
1086   for (MCAssembler::const_iterator it = Asm.begin(),
1087          ie = Asm.end(); it != ie; ++it) {
1088     const MCSectionData &SD = *it;
1089     if (Relocations[&SD].empty())
1090       continue;
1091
1092     MCContext &Ctx = Asm.getContext();
1093     const MCSectionELF &Section =
1094       static_cast<const MCSectionELF&>(SD.getSection());
1095
1096     const StringRef SectionName = Section.getSectionName();
1097     std::string RelaSectionName = hasRelocationAddend() ? ".rela" : ".rel";
1098     RelaSectionName += SectionName;
1099
1100     unsigned EntrySize;
1101     if (hasRelocationAddend())
1102       EntrySize = is64Bit() ? sizeof(ELF::Elf64_Rela) : sizeof(ELF::Elf32_Rela);
1103     else
1104       EntrySize = is64Bit() ? sizeof(ELF::Elf64_Rel) : sizeof(ELF::Elf32_Rel);
1105
1106     unsigned Flags = 0;
1107     StringRef Group = "";
1108     if (Section.getFlags() & ELF::SHF_GROUP) {
1109       Flags = ELF::SHF_GROUP;
1110       Group = Section.getGroup()->getName();
1111     }
1112
1113     const MCSectionELF *RelaSection =
1114       Ctx.getELFSection(RelaSectionName, hasRelocationAddend() ?
1115                         ELF::SHT_RELA : ELF::SHT_REL, Flags,
1116                         SectionKind::getReadOnly(),
1117                         EntrySize, Group);
1118     RelMap[&Section] = RelaSection;
1119     Asm.getOrCreateSectionData(*RelaSection);
1120   }
1121 }
1122
1123 void ELFObjectWriter::WriteRelocations(MCAssembler &Asm, MCAsmLayout &Layout,
1124                                        const RelMapTy &RelMap) {
1125   for (MCAssembler::const_iterator it = Asm.begin(),
1126          ie = Asm.end(); it != ie; ++it) {
1127     const MCSectionData &SD = *it;
1128     const MCSectionELF &Section =
1129       static_cast<const MCSectionELF&>(SD.getSection());
1130
1131     const MCSectionELF *RelaSection = RelMap.lookup(&Section);
1132     if (!RelaSection)
1133       continue;
1134     MCSectionData &RelaSD = Asm.getOrCreateSectionData(*RelaSection);
1135     RelaSD.setAlignment(is64Bit() ? 8 : 4);
1136
1137     MCDataFragment *F = new MCDataFragment(&RelaSD);
1138     WriteRelocationsFragment(Asm, F, &*it);
1139   }
1140 }
1141
1142 void ELFObjectWriter::WriteSecHdrEntry(uint32_t Name, uint32_t Type,
1143                                        uint64_t Flags, uint64_t Address,
1144                                        uint64_t Offset, uint64_t Size,
1145                                        uint32_t Link, uint32_t Info,
1146                                        uint64_t Alignment,
1147                                        uint64_t EntrySize) {
1148   Write32(Name);        // sh_name: index into string table
1149   Write32(Type);        // sh_type
1150   WriteWord(Flags);     // sh_flags
1151   WriteWord(Address);   // sh_addr
1152   WriteWord(Offset);    // sh_offset
1153   WriteWord(Size);      // sh_size
1154   Write32(Link);        // sh_link
1155   Write32(Info);        // sh_info
1156   WriteWord(Alignment); // sh_addralign
1157   WriteWord(EntrySize); // sh_entsize
1158 }
1159
1160 void ELFObjectWriter::WriteRelocationsFragment(const MCAssembler &Asm,
1161                                                MCDataFragment *F,
1162                                                const MCSectionData *SD) {
1163   std::vector<ELFRelocationEntry> &Relocs = Relocations[SD];
1164
1165   // Sort the relocation entries. Most targets just sort by r_offset, but some
1166   // (e.g., MIPS) have additional constraints.
1167   TargetObjectWriter->sortRelocs(Asm, Relocs);
1168
1169   for (unsigned i = 0, e = Relocs.size(); i != e; ++i) {
1170     ELFRelocationEntry entry = Relocs[e - i - 1];
1171
1172     if (!entry.Index)
1173       ;
1174     // FIXME: this is most likely a bug if index overflows.
1175     else if (entry.Index < 0)
1176       entry.Index = getSymbolIndexInSymbolTable(Asm, entry.Symbol);
1177     else
1178       entry.Index += FileSymbolData.size() + LocalSymbolData.size();
1179     if (is64Bit()) {
1180       write(*F, entry.r_offset);
1181       if (TargetObjectWriter->isN64()) {
1182         write(*F, uint32_t(entry.Index));
1183
1184         write(*F, TargetObjectWriter->getRSsym(entry.Type));
1185         write(*F, TargetObjectWriter->getRType3(entry.Type));
1186         write(*F, TargetObjectWriter->getRType2(entry.Type));
1187         write(*F, TargetObjectWriter->getRType(entry.Type));
1188       }
1189       else {
1190         struct ELF::Elf64_Rela ERE64;
1191         ERE64.setSymbolAndType(entry.Index, entry.Type);
1192         write(*F, ERE64.r_info);
1193       }
1194       if (hasRelocationAddend())
1195         write(*F, entry.r_addend);
1196     } else {
1197       write(*F, uint32_t(entry.r_offset));
1198
1199       struct ELF::Elf32_Rela ERE32;
1200       ERE32.setSymbolAndType(entry.Index, entry.Type);
1201       write(*F, ERE32.r_info);
1202
1203       if (hasRelocationAddend())
1204         write(*F, uint32_t(entry.r_addend));
1205     }
1206   }
1207 }
1208
1209 static int compareBySuffix(const MCSectionELF *const *a,
1210                            const MCSectionELF *const *b) {
1211   const StringRef &NameA = (*a)->getSectionName();
1212   const StringRef &NameB = (*b)->getSectionName();
1213   const unsigned sizeA = NameA.size();
1214   const unsigned sizeB = NameB.size();
1215   const unsigned len = std::min(sizeA, sizeB);
1216   for (unsigned int i = 0; i < len; ++i) {
1217     char ca = NameA[sizeA - i - 1];
1218     char cb = NameB[sizeB - i - 1];
1219     if (ca != cb)
1220       return cb - ca;
1221   }
1222
1223   return sizeB - sizeA;
1224 }
1225
1226 void ELFObjectWriter::CreateMetadataSections(MCAssembler &Asm,
1227                                              MCAsmLayout &Layout,
1228                                              SectionIndexMapTy &SectionIndexMap,
1229                                              const RelMapTy &RelMap) {
1230   MCContext &Ctx = Asm.getContext();
1231   MCDataFragment *F;
1232
1233   unsigned EntrySize = is64Bit() ? ELF::SYMENTRY_SIZE64 : ELF::SYMENTRY_SIZE32;
1234
1235   // We construct .shstrtab, .symtab and .strtab in this order to match gnu as.
1236   const MCSectionELF *ShstrtabSection =
1237     Ctx.getELFSection(".shstrtab", ELF::SHT_STRTAB, 0,
1238                       SectionKind::getReadOnly());
1239   MCSectionData &ShstrtabSD = Asm.getOrCreateSectionData(*ShstrtabSection);
1240   ShstrtabSD.setAlignment(1);
1241
1242   const MCSectionELF *SymtabSection =
1243     Ctx.getELFSection(".symtab", ELF::SHT_SYMTAB, 0,
1244                       SectionKind::getReadOnly(),
1245                       EntrySize, "");
1246   MCSectionData &SymtabSD = Asm.getOrCreateSectionData(*SymtabSection);
1247   SymtabSD.setAlignment(is64Bit() ? 8 : 4);
1248
1249   const MCSectionELF *StrtabSection;
1250   StrtabSection = Ctx.getELFSection(".strtab", ELF::SHT_STRTAB, 0,
1251                                     SectionKind::getReadOnly());
1252   MCSectionData &StrtabSD = Asm.getOrCreateSectionData(*StrtabSection);
1253   StrtabSD.setAlignment(1);
1254
1255   ComputeIndexMap(Asm, SectionIndexMap, RelMap);
1256
1257   ShstrtabIndex = SectionIndexMap.lookup(ShstrtabSection);
1258   SymbolTableIndex = SectionIndexMap.lookup(SymtabSection);
1259   StringTableIndex = SectionIndexMap.lookup(StrtabSection);
1260
1261   // Symbol table
1262   F = new MCDataFragment(&SymtabSD);
1263   WriteSymbolTable(F, Asm, Layout, SectionIndexMap);
1264
1265   F = new MCDataFragment(&StrtabSD);
1266   F->getContents().append(StringTable.begin(), StringTable.end());
1267
1268   F = new MCDataFragment(&ShstrtabSD);
1269
1270   std::vector<const MCSectionELF*> Sections;
1271   for (MCAssembler::const_iterator it = Asm.begin(),
1272          ie = Asm.end(); it != ie; ++it) {
1273     const MCSectionELF &Section =
1274       static_cast<const MCSectionELF&>(it->getSection());
1275     Sections.push_back(&Section);
1276   }
1277   array_pod_sort(Sections.begin(), Sections.end(), compareBySuffix);
1278
1279   // Section header string table.
1280   //
1281   // The first entry of a string table holds a null character so skip
1282   // section 0.
1283   uint64_t Index = 1;
1284   F->getContents().push_back('\x00');
1285
1286   for (unsigned int I = 0, E = Sections.size(); I != E; ++I) {
1287     const MCSectionELF &Section = *Sections[I];
1288
1289     StringRef Name = Section.getSectionName();
1290     if (I != 0) {
1291       StringRef PreviousName = Sections[I - 1]->getSectionName();
1292       if (PreviousName.endswith(Name)) {
1293         SectionStringTableIndex[&Section] = Index - Name.size() - 1;
1294         continue;
1295       }
1296     }
1297     // Remember the index into the string table so we can write it
1298     // into the sh_name field of the section header table.
1299     SectionStringTableIndex[&Section] = Index;
1300
1301     Index += Name.size() + 1;
1302     F->getContents().append(Name.begin(), Name.end());
1303     F->getContents().push_back('\x00');
1304   }
1305 }
1306
1307 void ELFObjectWriter::CreateIndexedSections(MCAssembler &Asm,
1308                                             MCAsmLayout &Layout,
1309                                             GroupMapTy &GroupMap,
1310                                             RevGroupMapTy &RevGroupMap,
1311                                             SectionIndexMapTy &SectionIndexMap,
1312                                             const RelMapTy &RelMap) {
1313   // Create the .note.GNU-stack section if needed.
1314   MCContext &Ctx = Asm.getContext();
1315   if (Asm.getNoExecStack()) {
1316     const MCSectionELF *GnuStackSection =
1317       Ctx.getELFSection(".note.GNU-stack", ELF::SHT_PROGBITS, 0,
1318                         SectionKind::getReadOnly());
1319     Asm.getOrCreateSectionData(*GnuStackSection);
1320   }
1321
1322   // Build the groups
1323   for (MCAssembler::const_iterator it = Asm.begin(), ie = Asm.end();
1324        it != ie; ++it) {
1325     const MCSectionELF &Section =
1326       static_cast<const MCSectionELF&>(it->getSection());
1327     if (!(Section.getFlags() & ELF::SHF_GROUP))
1328       continue;
1329
1330     const MCSymbol *SignatureSymbol = Section.getGroup();
1331     Asm.getOrCreateSymbolData(*SignatureSymbol);
1332     const MCSectionELF *&Group = RevGroupMap[SignatureSymbol];
1333     if (!Group) {
1334       Group = Ctx.CreateELFGroupSection();
1335       MCSectionData &Data = Asm.getOrCreateSectionData(*Group);
1336       Data.setAlignment(4);
1337       MCDataFragment *F = new MCDataFragment(&Data);
1338       write(*F, uint32_t(ELF::GRP_COMDAT));
1339     }
1340     GroupMap[Group] = SignatureSymbol;
1341   }
1342
1343   ComputeIndexMap(Asm, SectionIndexMap, RelMap);
1344
1345   // Add sections to the groups
1346   for (MCAssembler::const_iterator it = Asm.begin(), ie = Asm.end();
1347        it != ie; ++it) {
1348     const MCSectionELF &Section =
1349       static_cast<const MCSectionELF&>(it->getSection());
1350     if (!(Section.getFlags() & ELF::SHF_GROUP))
1351       continue;
1352     const MCSectionELF *Group = RevGroupMap[Section.getGroup()];
1353     MCSectionData &Data = Asm.getOrCreateSectionData(*Group);
1354     // FIXME: we could use the previous fragment
1355     MCDataFragment *F = new MCDataFragment(&Data);
1356     uint32_t Index = SectionIndexMap.lookup(&Section);
1357     write(*F, Index);
1358   }
1359 }
1360
1361 void ELFObjectWriter::WriteSection(MCAssembler &Asm,
1362                                    const SectionIndexMapTy &SectionIndexMap,
1363                                    uint32_t GroupSymbolIndex,
1364                                    uint64_t Offset, uint64_t Size,
1365                                    uint64_t Alignment,
1366                                    const MCSectionELF &Section) {
1367   uint64_t sh_link = 0;
1368   uint64_t sh_info = 0;
1369
1370   switch(Section.getType()) {
1371   case ELF::SHT_DYNAMIC:
1372     sh_link = SectionStringTableIndex[&Section];
1373     sh_info = 0;
1374     break;
1375
1376   case ELF::SHT_REL:
1377   case ELF::SHT_RELA: {
1378     const MCSectionELF *SymtabSection;
1379     const MCSectionELF *InfoSection;
1380     SymtabSection = Asm.getContext().getELFSection(".symtab", ELF::SHT_SYMTAB,
1381                                                    0,
1382                                                    SectionKind::getReadOnly());
1383     sh_link = SectionIndexMap.lookup(SymtabSection);
1384     assert(sh_link && ".symtab not found");
1385
1386     // Remove ".rel" and ".rela" prefixes.
1387     unsigned SecNameLen = (Section.getType() == ELF::SHT_REL) ? 4 : 5;
1388     StringRef SectionName = Section.getSectionName().substr(SecNameLen);
1389     StringRef GroupName =
1390         Section.getGroup() ? Section.getGroup()->getName() : "";
1391
1392     InfoSection = Asm.getContext().getELFSection(SectionName, ELF::SHT_PROGBITS,
1393                                                  0, SectionKind::getReadOnly(),
1394                                                  0, GroupName);
1395     sh_info = SectionIndexMap.lookup(InfoSection);
1396     break;
1397   }
1398
1399   case ELF::SHT_SYMTAB:
1400   case ELF::SHT_DYNSYM:
1401     sh_link = StringTableIndex;
1402     sh_info = LastLocalSymbolIndex;
1403     break;
1404
1405   case ELF::SHT_SYMTAB_SHNDX:
1406     sh_link = SymbolTableIndex;
1407     break;
1408
1409   case ELF::SHT_PROGBITS:
1410   case ELF::SHT_STRTAB:
1411   case ELF::SHT_NOBITS:
1412   case ELF::SHT_NOTE:
1413   case ELF::SHT_NULL:
1414   case ELF::SHT_ARM_ATTRIBUTES:
1415   case ELF::SHT_INIT_ARRAY:
1416   case ELF::SHT_FINI_ARRAY:
1417   case ELF::SHT_PREINIT_ARRAY:
1418   case ELF::SHT_X86_64_UNWIND:
1419   case ELF::SHT_MIPS_REGINFO:
1420   case ELF::SHT_MIPS_OPTIONS:
1421     // Nothing to do.
1422     break;
1423
1424   case ELF::SHT_GROUP:
1425     sh_link = SymbolTableIndex;
1426     sh_info = GroupSymbolIndex;
1427     break;
1428
1429   default:
1430     assert(0 && "FIXME: sh_type value not supported!");
1431     break;
1432   }
1433
1434   if (TargetObjectWriter->getEMachine() == ELF::EM_ARM &&
1435       Section.getType() == ELF::SHT_ARM_EXIDX) {
1436     StringRef SecName(Section.getSectionName());
1437     if (SecName == ".ARM.exidx") {
1438       sh_link = SectionIndexMap.lookup(
1439         Asm.getContext().getELFSection(".text",
1440                                        ELF::SHT_PROGBITS,
1441                                        ELF::SHF_EXECINSTR | ELF::SHF_ALLOC,
1442                                        SectionKind::getText()));
1443     } else if (SecName.startswith(".ARM.exidx")) {
1444       StringRef GroupName =
1445           Section.getGroup() ? Section.getGroup()->getName() : "";
1446       sh_link = SectionIndexMap.lookup(Asm.getContext().getELFSection(
1447           SecName.substr(sizeof(".ARM.exidx") - 1), ELF::SHT_PROGBITS,
1448           ELF::SHF_EXECINSTR | ELF::SHF_ALLOC, SectionKind::getText(), 0,
1449           GroupName));
1450     }
1451   }
1452
1453   WriteSecHdrEntry(SectionStringTableIndex[&Section], Section.getType(),
1454                    Section.getFlags(), 0, Offset, Size, sh_link, sh_info,
1455                    Alignment, Section.getEntrySize());
1456 }
1457
1458 bool ELFObjectWriter::IsELFMetaDataSection(const MCSectionData &SD) {
1459   return SD.getOrdinal() == ~UINT32_C(0) &&
1460     !SD.getSection().isVirtualSection();
1461 }
1462
1463 uint64_t ELFObjectWriter::DataSectionSize(const MCSectionData &SD) {
1464   uint64_t Ret = 0;
1465   for (MCSectionData::const_iterator i = SD.begin(), e = SD.end(); i != e;
1466        ++i) {
1467     const MCFragment &F = *i;
1468     assert(F.getKind() == MCFragment::FT_Data);
1469     Ret += cast<MCDataFragment>(F).getContents().size();
1470   }
1471   return Ret;
1472 }
1473
1474 uint64_t ELFObjectWriter::GetSectionFileSize(const MCAsmLayout &Layout,
1475                                              const MCSectionData &SD) {
1476   if (IsELFMetaDataSection(SD))
1477     return DataSectionSize(SD);
1478   return Layout.getSectionFileSize(&SD);
1479 }
1480
1481 uint64_t ELFObjectWriter::GetSectionAddressSize(const MCAsmLayout &Layout,
1482                                                 const MCSectionData &SD) {
1483   if (IsELFMetaDataSection(SD))
1484     return DataSectionSize(SD);
1485   return Layout.getSectionAddressSize(&SD);
1486 }
1487
1488 void ELFObjectWriter::WriteDataSectionData(MCAssembler &Asm,
1489                                            const MCAsmLayout &Layout,
1490                                            const MCSectionELF &Section) {
1491   const MCSectionData &SD = Asm.getOrCreateSectionData(Section);
1492
1493   uint64_t Padding = OffsetToAlignment(OS.tell(), SD.getAlignment());
1494   WriteZeros(Padding);
1495
1496   if (IsELFMetaDataSection(SD)) {
1497     for (MCSectionData::const_iterator i = SD.begin(), e = SD.end(); i != e;
1498          ++i) {
1499       const MCFragment &F = *i;
1500       assert(F.getKind() == MCFragment::FT_Data);
1501       WriteBytes(cast<MCDataFragment>(F).getContents());
1502     }
1503   } else {
1504     Asm.writeSectionData(&SD, Layout);
1505   }
1506 }
1507
1508 void ELFObjectWriter::WriteSectionHeader(MCAssembler &Asm,
1509                                          const GroupMapTy &GroupMap,
1510                                          const MCAsmLayout &Layout,
1511                                       const SectionIndexMapTy &SectionIndexMap,
1512                                    const SectionOffsetMapTy &SectionOffsetMap) {
1513   const unsigned NumSections = Asm.size() + 1;
1514
1515   std::vector<const MCSectionELF*> Sections;
1516   Sections.resize(NumSections - 1);
1517
1518   for (SectionIndexMapTy::const_iterator i=
1519          SectionIndexMap.begin(), e = SectionIndexMap.end(); i != e; ++i) {
1520     const std::pair<const MCSectionELF*, uint32_t> &p = *i;
1521     Sections[p.second - 1] = p.first;
1522   }
1523
1524   // Null section first.
1525   uint64_t FirstSectionSize =
1526     NumSections >= ELF::SHN_LORESERVE ? NumSections : 0;
1527   uint32_t FirstSectionLink =
1528     ShstrtabIndex >= ELF::SHN_LORESERVE ? ShstrtabIndex : 0;
1529   WriteSecHdrEntry(0, 0, 0, 0, 0, FirstSectionSize, FirstSectionLink, 0, 0, 0);
1530
1531   for (unsigned i = 0; i < NumSections - 1; ++i) {
1532     const MCSectionELF &Section = *Sections[i];
1533     const MCSectionData &SD = Asm.getOrCreateSectionData(Section);
1534     uint32_t GroupSymbolIndex;
1535     if (Section.getType() != ELF::SHT_GROUP)
1536       GroupSymbolIndex = 0;
1537     else
1538       GroupSymbolIndex = getSymbolIndexInSymbolTable(Asm,
1539                                                      GroupMap.lookup(&Section));
1540
1541     uint64_t Size = GetSectionAddressSize(Layout, SD);
1542
1543     WriteSection(Asm, SectionIndexMap, GroupSymbolIndex,
1544                  SectionOffsetMap.lookup(&Section), Size,
1545                  SD.getAlignment(), Section);
1546   }
1547 }
1548
1549 void ELFObjectWriter::ComputeSectionOrder(MCAssembler &Asm,
1550                                   std::vector<const MCSectionELF*> &Sections) {
1551   for (MCAssembler::iterator it = Asm.begin(),
1552          ie = Asm.end(); it != ie; ++it) {
1553     const MCSectionELF &Section =
1554       static_cast<const MCSectionELF &>(it->getSection());
1555     if (Section.getType() == ELF::SHT_GROUP)
1556       Sections.push_back(&Section);
1557   }
1558
1559   for (MCAssembler::iterator it = Asm.begin(),
1560          ie = Asm.end(); it != ie; ++it) {
1561     const MCSectionELF &Section =
1562       static_cast<const MCSectionELF &>(it->getSection());
1563     if (Section.getType() != ELF::SHT_GROUP &&
1564         Section.getType() != ELF::SHT_REL &&
1565         Section.getType() != ELF::SHT_RELA)
1566       Sections.push_back(&Section);
1567   }
1568
1569   for (MCAssembler::iterator it = Asm.begin(),
1570          ie = Asm.end(); it != ie; ++it) {
1571     const MCSectionELF &Section =
1572       static_cast<const MCSectionELF &>(it->getSection());
1573     if (Section.getType() == ELF::SHT_REL ||
1574         Section.getType() == ELF::SHT_RELA)
1575       Sections.push_back(&Section);
1576   }
1577 }
1578
1579 void ELFObjectWriter::WriteObject(MCAssembler &Asm,
1580                                   const MCAsmLayout &Layout) {
1581   GroupMapTy GroupMap;
1582   RevGroupMapTy RevGroupMap;
1583   SectionIndexMapTy SectionIndexMap;
1584
1585   unsigned NumUserSections = Asm.size();
1586
1587   DenseMap<const MCSectionELF*, const MCSectionELF*> RelMap;
1588   CreateRelocationSections(Asm, const_cast<MCAsmLayout&>(Layout), RelMap);
1589
1590   const unsigned NumUserAndRelocSections = Asm.size();
1591   CreateIndexedSections(Asm, const_cast<MCAsmLayout&>(Layout), GroupMap,
1592                         RevGroupMap, SectionIndexMap, RelMap);
1593   const unsigned AllSections = Asm.size();
1594   const unsigned NumIndexedSections = AllSections - NumUserAndRelocSections;
1595
1596   unsigned NumRegularSections = NumUserSections + NumIndexedSections;
1597
1598   // Compute symbol table information.
1599   computeSymbolTable(Asm, Layout, SectionIndexMap, RevGroupMap,
1600                      NumRegularSections);
1601
1602   WriteRelocations(Asm, const_cast<MCAsmLayout&>(Layout), RelMap);
1603
1604   CreateMetadataSections(const_cast<MCAssembler&>(Asm),
1605                          const_cast<MCAsmLayout&>(Layout),
1606                          SectionIndexMap,
1607                          RelMap);
1608
1609   uint64_t NaturalAlignment = is64Bit() ? 8 : 4;
1610   uint64_t HeaderSize = is64Bit() ? sizeof(ELF::Elf64_Ehdr) :
1611                                     sizeof(ELF::Elf32_Ehdr);
1612   uint64_t FileOff = HeaderSize;
1613
1614   std::vector<const MCSectionELF*> Sections;
1615   ComputeSectionOrder(Asm, Sections);
1616   unsigned NumSections = Sections.size();
1617   SectionOffsetMapTy SectionOffsetMap;
1618   for (unsigned i = 0; i < NumRegularSections + 1; ++i) {
1619     const MCSectionELF &Section = *Sections[i];
1620     const MCSectionData &SD = Asm.getOrCreateSectionData(Section);
1621
1622     FileOff = RoundUpToAlignment(FileOff, SD.getAlignment());
1623
1624     // Remember the offset into the file for this section.
1625     SectionOffsetMap[&Section] = FileOff;
1626
1627     // Get the size of the section in the output file (including padding).
1628     FileOff += GetSectionFileSize(Layout, SD);
1629   }
1630
1631   FileOff = RoundUpToAlignment(FileOff, NaturalAlignment);
1632
1633   const unsigned SectionHeaderOffset = FileOff - HeaderSize;
1634
1635   uint64_t SectionHeaderEntrySize = is64Bit() ?
1636     sizeof(ELF::Elf64_Shdr) : sizeof(ELF::Elf32_Shdr);
1637   FileOff += (NumSections + 1) * SectionHeaderEntrySize;
1638
1639   for (unsigned i = NumRegularSections + 1; i < NumSections; ++i) {
1640     const MCSectionELF &Section = *Sections[i];
1641     const MCSectionData &SD = Asm.getOrCreateSectionData(Section);
1642
1643     FileOff = RoundUpToAlignment(FileOff, SD.getAlignment());
1644
1645     // Remember the offset into the file for this section.
1646     SectionOffsetMap[&Section] = FileOff;
1647
1648     // Get the size of the section in the output file (including padding).
1649     FileOff += GetSectionFileSize(Layout, SD);
1650   }
1651
1652   // Write out the ELF header ...
1653   WriteHeader(Asm, SectionHeaderOffset, NumSections + 1);
1654
1655   // ... then the regular sections ...
1656   // + because of .shstrtab
1657   for (unsigned i = 0; i < NumRegularSections + 1; ++i)
1658     WriteDataSectionData(Asm, Layout, *Sections[i]);
1659
1660   uint64_t Padding = OffsetToAlignment(OS.tell(), NaturalAlignment);
1661   WriteZeros(Padding);
1662
1663   // ... then the section header table ...
1664   WriteSectionHeader(Asm, GroupMap, Layout, SectionIndexMap,
1665                      SectionOffsetMap);
1666
1667   // ... and then the remaining sections ...
1668   for (unsigned i = NumRegularSections + 1; i < NumSections; ++i)
1669     WriteDataSectionData(Asm, Layout, *Sections[i]);
1670 }
1671
1672 bool
1673 ELFObjectWriter::IsSymbolRefDifferenceFullyResolvedImpl(const MCAssembler &Asm,
1674                                                       const MCSymbolData &DataA,
1675                                                       const MCFragment &FB,
1676                                                       bool InSet,
1677                                                       bool IsPCRel) const {
1678   if (DataA.getFlags() & ELF_STB_Weak || MCELF::GetType(DataA) == ELF::STT_GNU_IFUNC)
1679     return false;
1680   return MCObjectWriter::IsSymbolRefDifferenceFullyResolvedImpl(
1681                                                  Asm, DataA, FB,InSet, IsPCRel);
1682 }
1683
1684 MCObjectWriter *llvm::createELFObjectWriter(MCELFObjectTargetWriter *MOTW,
1685                                             raw_ostream &OS,
1686                                             bool IsLittleEndian) {
1687   return new ELFObjectWriter(MOTW, OS, IsLittleEndian);
1688 }