blob: f1acc1d75fbd5a29b2247becde3da6e019ffd4dd [file] [log] [blame]
Nick Lewycky4288f9e2013-03-09 09:32:16 +00001//===- lib/MC/ELFObjectWriter.cpp - ELF File Writer -----------------------===//
Matt Fleming6c1ad482010-08-16 18:57:57 +00002//
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
Chandler Carruthed0881b2012-12-03 16:50:05 +000014#include "llvm/MC/MCELFObjectWriter.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000015#include "llvm/ADT/STLExtras.h"
Rafael Espindola29abd972011-12-22 03:38:00 +000016#include "llvm/ADT/SmallPtrSet.h"
17#include "llvm/ADT/SmallString.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000018#include "llvm/ADT/StringMap.h"
Evan Cheng5928e692011-07-25 23:24:55 +000019#include "llvm/MC/MCAsmBackend.h"
David Blaikie8019bf82014-04-10 21:53:53 +000020#include "llvm/MC/MCAsmInfo.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000021#include "llvm/MC/MCAsmLayout.h"
Rafael Espindola29abd972011-12-22 03:38:00 +000022#include "llvm/MC/MCAssembler.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000023#include "llvm/MC/MCContext.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000024#include "llvm/MC/MCExpr.h"
Craig Topper6e80c282012-03-26 06:58:25 +000025#include "llvm/MC/MCFixupKindInfo.h"
26#include "llvm/MC/MCObjectWriter.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000027#include "llvm/MC/MCSectionELF.h"
Rafael Espindolaa8695762015-06-02 00:25:12 +000028#include "llvm/MC/MCSymbolELF.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000029#include "llvm/MC/MCValue.h"
Rafael Espindola97de4742014-07-03 02:01:39 +000030#include "llvm/MC/StringTableBuilder.h"
David Blaikie8019bf82014-04-10 21:53:53 +000031#include "llvm/Support/Compression.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000032#include "llvm/Support/Debug.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000033#include "llvm/Support/ELF.h"
Chandler Carruthd9903882015-01-14 11:23:27 +000034#include "llvm/Support/Endian.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000035#include "llvm/Support/ErrorHandling.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000036#include <vector>
37using namespace llvm;
38
Jason W Kimc09e7262011-05-11 22:53:06 +000039#undef DEBUG_TYPE
40#define DEBUG_TYPE "reloc-info"
41
Rafael Espindola29abd972011-12-22 03:38:00 +000042namespace {
Rafael Espindola0ce09712014-03-25 22:43:53 +000043
Rafael Espindola10be0832014-03-25 23:44:25 +000044typedef DenseMap<const MCSectionELF *, uint32_t> SectionIndexMapTy;
45
Rafael Espindola91fd2772015-04-29 21:09:32 +000046class ELFObjectWriter;
47
Rafael Espindola10be0832014-03-25 23:44:25 +000048class SymbolTableWriter {
Rafael Espindola91fd2772015-04-29 21:09:32 +000049 ELFObjectWriter &EWriter;
Rafael Espindola10be0832014-03-25 23:44:25 +000050 bool Is64Bit;
Rafael Espindola10be0832014-03-25 23:44:25 +000051
Rafael Espindola91fd2772015-04-29 21:09:32 +000052 // indexes we are going to write to .symtab_shndx.
53 std::vector<uint32_t> ShndxIndexes;
Rafael Espindola10be0832014-03-25 23:44:25 +000054
55 // The numbel of symbols written so far.
56 unsigned NumWritten;
57
58 void createSymtabShndx();
59
Rafael Espindola91fd2772015-04-29 21:09:32 +000060 template <typename T> void write(T Value);
Rafael Espindola10be0832014-03-25 23:44:25 +000061
62public:
Rafael Espindola91fd2772015-04-29 21:09:32 +000063 SymbolTableWriter(ELFObjectWriter &EWriter, bool Is64Bit);
Rafael Espindola10be0832014-03-25 23:44:25 +000064
65 void writeSymbol(uint32_t name, uint8_t info, uint64_t value, uint64_t size,
66 uint8_t other, uint32_t shndx, bool Reserved);
Rafael Espindola91fd2772015-04-29 21:09:32 +000067
68 ArrayRef<uint32_t> getShndxIndexes() const { return ShndxIndexes; }
Rafael Espindola10be0832014-03-25 23:44:25 +000069};
70
Rafael Espindola29abd972011-12-22 03:38:00 +000071class ELFObjectWriter : public MCObjectWriter {
Rafael Espindola29abd972011-12-22 03:38:00 +000072 static bool isFixupKindPCRel(const MCAssembler &Asm, unsigned Kind);
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +000073 static uint64_t SymbolValue(const MCSymbol &Sym, const MCAsmLayout &Layout);
Rafael Espindola95fb9b92015-06-02 20:38:46 +000074 static bool isInSymtab(const MCAsmLayout &Layout, const MCSymbolELF &Symbol,
Rafael Espindola29abd972011-12-22 03:38:00 +000075 bool Used, bool Renamed);
Rafael Espindola29abd972011-12-22 03:38:00 +000076
Rafael Espindola6cd21802015-04-07 22:35:40 +000077 /// Helper struct for containing some precomputed information on symbols.
Rafael Espindola29abd972011-12-22 03:38:00 +000078 struct ELFSymbolData {
Rafael Espindolaa8695762015-06-02 00:25:12 +000079 const MCSymbolELF *Symbol;
Rafael Espindola29abd972011-12-22 03:38:00 +000080 uint32_t SectionIndex;
Hans Wennborg83e6e1e2014-04-30 16:25:02 +000081 StringRef Name;
Rafael Espindola29abd972011-12-22 03:38:00 +000082
83 // Support lexicographic sorting.
84 bool operator<(const ELFSymbolData &RHS) const {
Rafael Espindola95fb9b92015-06-02 20:38:46 +000085 unsigned LHSType = Symbol->getType();
86 unsigned RHSType = RHS.Symbol->getType();
Rafael Espindolab6613022014-10-17 01:48:58 +000087 if (LHSType == ELF::STT_SECTION && RHSType != ELF::STT_SECTION)
88 return false;
89 if (LHSType != ELF::STT_SECTION && RHSType == ELF::STT_SECTION)
90 return true;
91 if (LHSType == ELF::STT_SECTION && RHSType == ELF::STT_SECTION)
92 return SectionIndex < RHS.SectionIndex;
Hans Wennborg83e6e1e2014-04-30 16:25:02 +000093 return Name < RHS.Name;
Rafael Espindola29abd972011-12-22 03:38:00 +000094 }
95 };
96
Rafael Espindola29abd972011-12-22 03:38:00 +000097 /// The target specific ELF writer instance.
Ahmed Charles56440fd2014-03-06 05:51:42 +000098 std::unique_ptr<MCELFObjectTargetWriter> TargetObjectWriter;
Rafael Espindola29abd972011-12-22 03:38:00 +000099
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000100 DenseMap<const MCSymbolELF *, const MCSymbolELF *> Renames;
Rafael Espindola29abd972011-12-22 03:38:00 +0000101
Rafael Espindola34948e52015-04-30 00:45:46 +0000102 llvm::DenseMap<const MCSectionELF *, std::vector<ELFRelocationEntry>>
103 Relocations;
Rafael Espindola29abd972011-12-22 03:38:00 +0000104
105 /// @}
106 /// @name Symbol Table Data
107 /// @{
108
Hans Wennborg83e6e1e2014-04-30 16:25:02 +0000109 StringTableBuilder StrTabBuilder;
Rafael Espindola29abd972011-12-22 03:38:00 +0000110
111 /// @}
112
Rafael Espindola29abd972011-12-22 03:38:00 +0000113 // This holds the symbol table index of the last local symbol.
114 unsigned LastLocalSymbolIndex;
115 // This holds the .strtab section index.
116 unsigned StringTableIndex;
117 // This holds the .symtab section index.
118 unsigned SymbolTableIndex;
Rafael Espindola29abd972011-12-22 03:38:00 +0000119
Rafael Espindola03d7abb2015-04-30 20:53:27 +0000120 // Sections in the order they are to be output in the section table.
Rafael Espindola08850b32015-05-25 21:56:55 +0000121 std::vector<const MCSectionELF *> SectionTable;
122 unsigned addToSectionTable(const MCSectionELF *Sec);
Rafael Espindola29abd972011-12-22 03:38:00 +0000123
Rafael Espindola29abd972011-12-22 03:38:00 +0000124 // TargetObjectWriter wrappers.
Rafael Espindola29abd972011-12-22 03:38:00 +0000125 bool is64Bit() const { return TargetObjectWriter->is64Bit(); }
126 bool hasRelocationAddend() const {
127 return TargetObjectWriter->hasRelocationAddend();
128 }
Rafael Espindola29abd972011-12-22 03:38:00 +0000129 unsigned GetRelocType(const MCValue &Target, const MCFixup &Fixup,
Rafael Espindolac03f44c2014-03-27 20:49:35 +0000130 bool IsPCRel) const {
131 return TargetObjectWriter->GetRelocType(Target, Fixup, IsPCRel);
Rafael Espindola29abd972011-12-22 03:38:00 +0000132 }
133
Rafael Espindolab20fbb82015-06-05 18:21:00 +0000134 void align(unsigned Alignment);
135
Rafael Espindola29abd972011-12-22 03:38:00 +0000136 public:
Rafael Espindola5560a4c2015-04-14 22:14:34 +0000137 ELFObjectWriter(MCELFObjectTargetWriter *MOTW, raw_pwrite_stream &OS,
Rafael Espindola0ce09712014-03-25 22:43:53 +0000138 bool IsLittleEndian)
Rafael Espindolae2b355d2015-05-28 15:20:00 +0000139 : MCObjectWriter(OS, IsLittleEndian), TargetObjectWriter(MOTW) {}
Rafael Espindola29abd972011-12-22 03:38:00 +0000140
Yaron Keren7773a722015-03-23 18:35:01 +0000141 void reset() override {
Yaron Keren7773a722015-03-23 18:35:01 +0000142 Renames.clear();
143 Relocations.clear();
Yaron Keren7773a722015-03-23 18:35:01 +0000144 StrTabBuilder.clear();
Yaron Kerenf3465e12015-05-13 15:17:19 +0000145 SectionTable.clear();
Yaron Keren7773a722015-03-23 18:35:01 +0000146 MCObjectWriter::reset();
147 }
148
Alexander Kornienkof817c1c2015-04-11 02:11:45 +0000149 ~ELFObjectWriter() override;
Rafael Espindola29abd972011-12-22 03:38:00 +0000150
151 void WriteWord(uint64_t W) {
152 if (is64Bit())
Jim Grosbach36e60e92015-06-04 22:24:41 +0000153 write64(W);
Rafael Espindola29abd972011-12-22 03:38:00 +0000154 else
Jim Grosbach36e60e92015-06-04 22:24:41 +0000155 write32(W);
Rafael Espindola29abd972011-12-22 03:38:00 +0000156 }
157
Rafael Espindola91fd2772015-04-29 21:09:32 +0000158 template <typename T> void write(T Val) {
159 if (IsLittleEndian)
David Majnemerabdb2d2a2015-09-01 16:19:03 +0000160 support::endian::Writer<support::little>(getStream()).write(Val);
Rafael Espindola91fd2772015-04-29 21:09:32 +0000161 else
David Majnemerabdb2d2a2015-09-01 16:19:03 +0000162 support::endian::Writer<support::big>(getStream()).write(Val);
Rafael Espindola91fd2772015-04-29 21:09:32 +0000163 }
164
Rafael Espindola8c7829b2015-04-29 20:39:37 +0000165 void writeHeader(const MCAssembler &Asm);
Rafael Espindola29abd972011-12-22 03:38:00 +0000166
Rafael Espindolae48421f2015-05-28 20:25:29 +0000167 void writeSymbol(SymbolTableWriter &Writer, uint32_t StringIndex,
168 ELFSymbolData &MSD, const MCAsmLayout &Layout);
Rafael Espindola29abd972011-12-22 03:38:00 +0000169
Rafael Espindola91fd2772015-04-29 21:09:32 +0000170 // Start and end offset of each section
Rafael Espindolabda19802015-04-30 14:21:49 +0000171 typedef std::map<const MCSectionELF *, std::pair<uint64_t, uint64_t>>
172 SectionOffsetsTy;
Rafael Espindola91fd2772015-04-29 21:09:32 +0000173
Rafael Espindolab60c8292014-04-29 12:46:50 +0000174 bool shouldRelocateWithSymbol(const MCAssembler &Asm,
175 const MCSymbolRefExpr *RefA,
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000176 const MCSymbol *Sym, uint64_t C,
Rafael Espindola5904e122014-03-29 06:26:49 +0000177 unsigned Type) const;
178
Jim Grosbach36e60e92015-06-04 22:24:41 +0000179 void recordRelocation(MCAssembler &Asm, const MCAsmLayout &Layout,
Craig Topper34a61bc2014-03-08 07:02:02 +0000180 const MCFragment *Fragment, const MCFixup &Fixup,
Rafael Espindola5904e122014-03-29 06:26:49 +0000181 MCValue Target, bool &IsPCRel,
182 uint64_t &FixedValue) override;
Rafael Espindola29abd972011-12-22 03:38:00 +0000183
Rafael Espindola89feff32015-04-28 22:59:58 +0000184 // Map from a signature symbol to the group section index
185 typedef DenseMap<const MCSymbol *, unsigned> RevGroupMapTy;
Rafael Espindola29abd972011-12-22 03:38:00 +0000186
Rafael Espindola022bb762014-03-24 03:43:21 +0000187 /// Compute the symbol table data
Rafael Espindola29abd972011-12-22 03:38:00 +0000188 ///
Rafael Espindola073ee7d2012-08-27 16:04:24 +0000189 /// \param Asm - The assembler.
190 /// \param SectionIndexMap - Maps a section to its index.
191 /// \param RevGroupMap - Maps a signature symbol to the group section.
Rafael Espindola022bb762014-03-24 03:43:21 +0000192 void computeSymbolTable(MCAssembler &Asm, const MCAsmLayout &Layout,
Rafael Espindola29abd972011-12-22 03:38:00 +0000193 const SectionIndexMapTy &SectionIndexMap,
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000194 const RevGroupMapTy &RevGroupMap,
195 SectionOffsetsTy &SectionOffsets);
Rafael Espindola29abd972011-12-22 03:38:00 +0000196
Rafael Espindola0a82ad72015-05-21 20:43:13 +0000197 MCSectionELF *createRelocationSection(MCContext &Ctx,
Rafael Espindola0709a7b2015-05-21 19:20:38 +0000198 const MCSectionELF &Sec);
Rafael Espindola29abd972011-12-22 03:38:00 +0000199
Rafael Espindolab1863912015-04-30 21:10:06 +0000200 const MCSectionELF *createStringTable(MCContext &Ctx);
Rafael Espindola29abd972011-12-22 03:38:00 +0000201
Jim Grosbach56ed0bb2015-06-04 23:25:54 +0000202 void executePostLayoutBinding(MCAssembler &Asm,
Craig Topper34a61bc2014-03-08 07:02:02 +0000203 const MCAsmLayout &Layout) override;
Rafael Espindola29abd972011-12-22 03:38:00 +0000204
Rafael Espindola57c80832015-06-04 20:55:49 +0000205 void writeSectionHeader(const MCAsmLayout &Layout,
Rafael Espindola29abd972011-12-22 03:38:00 +0000206 const SectionIndexMapTy &SectionIndexMap,
Rafael Espindolaa8201692015-04-28 15:26:21 +0000207 const SectionOffsetsTy &SectionOffsets);
Rafael Espindola29abd972011-12-22 03:38:00 +0000208
Rafael Espindolae15b1b72015-05-27 13:30:50 +0000209 void writeSectionData(const MCAssembler &Asm, MCSection &Sec,
Rafael Espindola868b3f42015-04-30 21:51:58 +0000210 const MCAsmLayout &Layout);
211
Rafael Espindola29abd972011-12-22 03:38:00 +0000212 void WriteSecHdrEntry(uint32_t Name, uint32_t Type, uint64_t Flags,
Rafael Espindola868b3f42015-04-30 21:51:58 +0000213 uint64_t Address, uint64_t Offset, uint64_t Size,
214 uint32_t Link, uint32_t Info, uint64_t Alignment,
215 uint64_t EntrySize);
Rafael Espindola29abd972011-12-22 03:38:00 +0000216
Rafael Espindola34948e52015-04-30 00:45:46 +0000217 void writeRelocations(const MCAssembler &Asm, const MCSectionELF &Sec);
Rafael Espindola29abd972011-12-22 03:38:00 +0000218
Jim Grosbach36e60e92015-06-04 22:24:41 +0000219 bool isSymbolRefDifferenceFullyResolvedImpl(const MCAssembler &Asm,
Duncan P. N. Exon Smithd81ba532015-05-16 01:01:55 +0000220 const MCSymbol &SymA,
221 const MCFragment &FB,
222 bool InSet,
223 bool IsPCRel) const override;
Rafael Espindola29abd972011-12-22 03:38:00 +0000224
Duncan P. N. Exon Smith5266ad92015-05-20 15:10:03 +0000225 bool isWeak(const MCSymbol &Sym) const override;
Rafael Espindolaf275ad82015-03-25 13:16:53 +0000226
Jim Grosbach36e60e92015-06-04 22:24:41 +0000227 void writeObject(MCAssembler &Asm, const MCAsmLayout &Layout) override;
Rafael Espindolae92c1bf2015-05-21 19:42:35 +0000228 void writeSection(const SectionIndexMapTy &SectionIndexMap,
Rafael Espindola28687582015-05-21 19:36:43 +0000229 uint32_t GroupSymbolIndex, uint64_t Offset, uint64_t Size,
Rafael Espindola29abd972011-12-22 03:38:00 +0000230 const MCSectionELF &Section);
231 };
Alexander Kornienkof00654e2015-06-23 09:49:53 +0000232}
Rafael Espindola29abd972011-12-22 03:38:00 +0000233
Rafael Espindolab20fbb82015-06-05 18:21:00 +0000234void ELFObjectWriter::align(unsigned Alignment) {
David Majnemerabdb2d2a2015-09-01 16:19:03 +0000235 uint64_t Padding = OffsetToAlignment(getStream().tell(), Alignment);
Rafael Espindolab20fbb82015-06-05 18:21:00 +0000236 WriteZeros(Padding);
237}
238
Rafael Espindola08850b32015-05-25 21:56:55 +0000239unsigned ELFObjectWriter::addToSectionTable(const MCSectionELF *Sec) {
Rafael Espindola03d7abb2015-04-30 20:53:27 +0000240 SectionTable.push_back(Sec);
Rafael Espindola5960cee2015-05-22 23:58:30 +0000241 StrTabBuilder.add(Sec->getSectionName());
Rafael Espindola03d7abb2015-04-30 20:53:27 +0000242 return SectionTable.size();
243}
244
Rafael Espindola10be0832014-03-25 23:44:25 +0000245void SymbolTableWriter::createSymtabShndx() {
Rafael Espindola91fd2772015-04-29 21:09:32 +0000246 if (!ShndxIndexes.empty())
Rafael Espindola10be0832014-03-25 23:44:25 +0000247 return;
248
Rafael Espindola91fd2772015-04-29 21:09:32 +0000249 ShndxIndexes.resize(NumWritten);
Rafael Espindola10be0832014-03-25 23:44:25 +0000250}
251
Rafael Espindola91fd2772015-04-29 21:09:32 +0000252template <typename T> void SymbolTableWriter::write(T Value) {
253 EWriter.write(Value);
Rafael Espindola10be0832014-03-25 23:44:25 +0000254}
255
Rafael Espindola91fd2772015-04-29 21:09:32 +0000256SymbolTableWriter::SymbolTableWriter(ELFObjectWriter &EWriter, bool Is64Bit)
257 : EWriter(EWriter), Is64Bit(Is64Bit), NumWritten(0) {}
Rafael Espindola10be0832014-03-25 23:44:25 +0000258
259void SymbolTableWriter::writeSymbol(uint32_t name, uint8_t info, uint64_t value,
260 uint64_t size, uint8_t other,
261 uint32_t shndx, bool Reserved) {
262 bool LargeIndex = shndx >= ELF::SHN_LORESERVE && !Reserved;
263
264 if (LargeIndex)
265 createSymtabShndx();
266
Rafael Espindola91fd2772015-04-29 21:09:32 +0000267 if (!ShndxIndexes.empty()) {
Rafael Espindola10be0832014-03-25 23:44:25 +0000268 if (LargeIndex)
Rafael Espindola91fd2772015-04-29 21:09:32 +0000269 ShndxIndexes.push_back(shndx);
Rafael Espindola10be0832014-03-25 23:44:25 +0000270 else
Rafael Espindola91fd2772015-04-29 21:09:32 +0000271 ShndxIndexes.push_back(0);
Rafael Espindola10be0832014-03-25 23:44:25 +0000272 }
273
274 uint16_t Index = LargeIndex ? uint16_t(ELF::SHN_XINDEX) : shndx;
275
Rafael Espindola10be0832014-03-25 23:44:25 +0000276 if (Is64Bit) {
Rafael Espindola91fd2772015-04-29 21:09:32 +0000277 write(name); // st_name
278 write(info); // st_info
279 write(other); // st_other
280 write(Index); // st_shndx
281 write(value); // st_value
282 write(size); // st_size
Rafael Espindola10be0832014-03-25 23:44:25 +0000283 } else {
Rafael Espindola91fd2772015-04-29 21:09:32 +0000284 write(name); // st_name
285 write(uint32_t(value)); // st_value
286 write(uint32_t(size)); // st_size
287 write(info); // st_info
288 write(other); // st_other
289 write(Index); // st_shndx
Rafael Espindola10be0832014-03-25 23:44:25 +0000290 }
291
292 ++NumWritten;
293}
294
Jan Sjödin30a52de2011-02-28 21:45:04 +0000295bool ELFObjectWriter::isFixupKindPCRel(const MCAssembler &Asm, unsigned Kind) {
296 const MCFixupKindInfo &FKI =
297 Asm.getBackend().getFixupKindInfo((MCFixupKind) Kind);
298
299 return FKI.Flags & MCFixupKindInfo::FKF_IsPCRel;
300}
301
Jason W Kim96f4c012010-11-15 16:18:39 +0000302ELFObjectWriter::~ELFObjectWriter()
303{}
304
Matt Fleming6c1ad482010-08-16 18:57:57 +0000305// Emit the ELF header.
Rafael Espindola8c7829b2015-04-29 20:39:37 +0000306void ELFObjectWriter::writeHeader(const MCAssembler &Asm) {
Matt Fleming6c1ad482010-08-16 18:57:57 +0000307 // ELF Header
308 // ----------
309 //
310 // Note
311 // ----
312 // emitWord method behaves differently for ELF32 and ELF64, writing
313 // 4 bytes in the former and 8 in the latter.
314
Jim Grosbach36e60e92015-06-04 22:24:41 +0000315 writeBytes(ELF::ElfMagic); // e_ident[EI_MAG0] to e_ident[EI_MAG3]
Matt Fleming6c1ad482010-08-16 18:57:57 +0000316
Jim Grosbach36e60e92015-06-04 22:24:41 +0000317 write8(is64Bit() ? ELF::ELFCLASS64 : ELF::ELFCLASS32); // e_ident[EI_CLASS]
Matt Fleming6c1ad482010-08-16 18:57:57 +0000318
319 // e_ident[EI_DATA]
Jim Grosbach36e60e92015-06-04 22:24:41 +0000320 write8(isLittleEndian() ? ELF::ELFDATA2LSB : ELF::ELFDATA2MSB);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000321
Jim Grosbach36e60e92015-06-04 22:24:41 +0000322 write8(ELF::EV_CURRENT); // e_ident[EI_VERSION]
Roman Divacky3b727f52010-09-09 17:57:50 +0000323 // e_ident[EI_OSABI]
Jim Grosbach36e60e92015-06-04 22:24:41 +0000324 write8(TargetObjectWriter->getOSABI());
325 write8(0); // e_ident[EI_ABIVERSION]
Matt Fleming6c1ad482010-08-16 18:57:57 +0000326
327 WriteZeros(ELF::EI_NIDENT - ELF::EI_PAD);
328
Jim Grosbach36e60e92015-06-04 22:24:41 +0000329 write16(ELF::ET_REL); // e_type
Matt Fleming6c1ad482010-08-16 18:57:57 +0000330
Jim Grosbach36e60e92015-06-04 22:24:41 +0000331 write16(TargetObjectWriter->getEMachine()); // e_machine = target
Matt Fleming6c1ad482010-08-16 18:57:57 +0000332
Jim Grosbach36e60e92015-06-04 22:24:41 +0000333 write32(ELF::EV_CURRENT); // e_version
Matt Fleming6c1ad482010-08-16 18:57:57 +0000334 WriteWord(0); // e_entry, no entry point in .o file
335 WriteWord(0); // e_phoff, no program header for .o
Rafael Espindola642a2212015-04-14 22:54:16 +0000336 WriteWord(0); // e_shoff = sec hdr table off in bytes
Matt Fleming6c1ad482010-08-16 18:57:57 +0000337
Jason W Kim4761fba2011-02-04 21:41:11 +0000338 // e_flags = whatever the target wants
Jim Grosbach36e60e92015-06-04 22:24:41 +0000339 write32(Asm.getELFHeaderEFlags());
Matt Fleming6c1ad482010-08-16 18:57:57 +0000340
341 // e_ehsize = ELF header size
Jim Grosbach36e60e92015-06-04 22:24:41 +0000342 write16(is64Bit() ? sizeof(ELF::Elf64_Ehdr) : sizeof(ELF::Elf32_Ehdr));
Matt Fleming6c1ad482010-08-16 18:57:57 +0000343
Jim Grosbach36e60e92015-06-04 22:24:41 +0000344 write16(0); // e_phentsize = prog header entry size
345 write16(0); // e_phnum = # prog header entries = 0
Matt Fleming6c1ad482010-08-16 18:57:57 +0000346
347 // e_shentsize = Section header entry size
Jim Grosbach36e60e92015-06-04 22:24:41 +0000348 write16(is64Bit() ? sizeof(ELF::Elf64_Shdr) : sizeof(ELF::Elf32_Shdr));
Matt Fleming6c1ad482010-08-16 18:57:57 +0000349
350 // e_shnum = # of section header ents
Jim Grosbach36e60e92015-06-04 22:24:41 +0000351 write16(0);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000352
353 // e_shstrndx = Section # of '.shstrtab'
Rafael Espindola5960cee2015-05-22 23:58:30 +0000354 assert(StringTableIndex < ELF::SHN_LORESERVE);
Jim Grosbach36e60e92015-06-04 22:24:41 +0000355 write16(StringTableIndex);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000356}
357
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000358uint64_t ELFObjectWriter::SymbolValue(const MCSymbol &Sym,
Jan Sjödin30a52de2011-02-28 21:45:04 +0000359 const MCAsmLayout &Layout) {
Rafael Espindola4d37b2a2015-05-29 21:45:01 +0000360 if (Sym.isCommon() && Sym.isExternal())
Rafael Espindola14672502015-05-29 17:48:04 +0000361 return Sym.getCommonAlignment();
Rafael Espindola9735f4f2010-09-27 21:23:02 +0000362
Rafael Espindolafee224f2014-04-30 21:51:13 +0000363 uint64_t Res;
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000364 if (!Layout.getSymbolOffset(Sym, Res))
Rafael Espindola553e5eb2014-04-30 16:59:35 +0000365 return 0;
366
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000367 if (Layout.getAssembler().isThumbFunc(&Sym))
Rafael Espindola66f96fe2014-03-21 22:00:29 +0000368 Res |= 1;
Rafael Espindola9735f4f2010-09-27 21:23:02 +0000369
Rafael Espindola66f96fe2014-03-21 22:00:29 +0000370 return Res;
Rafael Espindola9735f4f2010-09-27 21:23:02 +0000371}
372
Jim Grosbach56ed0bb2015-06-04 23:25:54 +0000373void ELFObjectWriter::executePostLayoutBinding(MCAssembler &Asm,
Rafael Espindola93e3cf02010-12-07 00:27:36 +0000374 const MCAsmLayout &Layout) {
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000375 // The presence of symbol versions causes undefined symbols and
376 // versions declared with @@@ to be renamed.
377
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000378 for (const MCSymbol &A : Asm.symbols()) {
379 const auto &Alias = cast<MCSymbolELF>(A);
Rafael Espindola6cd76e62010-10-28 18:33:03 +0000380 // Not an alias.
Rafael Espindola6efaf112014-04-28 17:05:36 +0000381 if (!Alias.isVariable())
Rafael Espindola6cd76e62010-10-28 18:33:03 +0000382 continue;
Rafael Espindola6efaf112014-04-28 17:05:36 +0000383 auto *Ref = dyn_cast<MCSymbolRefExpr>(Alias.getVariableValue());
384 if (!Ref)
385 continue;
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000386 const auto &Symbol = cast<MCSymbolELF>(Ref->getSymbol());
Rafael Espindola6cd76e62010-10-28 18:33:03 +0000387
Benjamin Kramer14807272010-10-27 19:53:52 +0000388 StringRef AliasName = Alias.getName();
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000389 size_t Pos = AliasName.find('@');
390 if (Pos == StringRef::npos)
391 continue;
392
Rafael Espindola6cd76e62010-10-28 18:33:03 +0000393 // Aliases defined with .symvar copy the binding from the symbol they alias.
394 // This is the first place we are able to copy this information.
Rafael Espindola4d37b2a2015-05-29 21:45:01 +0000395 Alias.setExternal(Symbol.isExternal());
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000396 Alias.setBinding(Symbol.getBinding());
Rafael Espindola6cd76e62010-10-28 18:33:03 +0000397
Benjamin Kramer14807272010-10-27 19:53:52 +0000398 StringRef Rest = AliasName.substr(Pos);
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000399 if (!Symbol.isUndefined() && !Rest.startswith("@@@"))
400 continue;
401
Rafael Espindola26496e62010-10-27 17:56:18 +0000402 // FIXME: produce a better error message.
403 if (Symbol.isUndefined() && Rest.startswith("@@") &&
404 !Rest.startswith("@@@"))
405 report_fatal_error("A @@ version cannot be undefined");
406
Benjamin Kramer14807272010-10-27 19:53:52 +0000407 Renames.insert(std::make_pair(&Symbol, &Alias));
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000408 }
409}
410
Roman Divacky5a1c5492014-01-07 20:17:03 +0000411static uint8_t mergeTypeForSet(uint8_t origType, uint8_t newType) {
412 uint8_t Type = newType;
413
414 // Propagation rules:
415 // IFUNC > FUNC > OBJECT > NOTYPE
416 // TLS_OBJECT > OBJECT > NOTYPE
417 //
418 // dont let the new type degrade the old type
419 switch (origType) {
420 default:
421 break;
422 case ELF::STT_GNU_IFUNC:
423 if (Type == ELF::STT_FUNC || Type == ELF::STT_OBJECT ||
424 Type == ELF::STT_NOTYPE || Type == ELF::STT_TLS)
425 Type = ELF::STT_GNU_IFUNC;
426 break;
427 case ELF::STT_FUNC:
428 if (Type == ELF::STT_OBJECT || Type == ELF::STT_NOTYPE ||
429 Type == ELF::STT_TLS)
430 Type = ELF::STT_FUNC;
431 break;
432 case ELF::STT_OBJECT:
433 if (Type == ELF::STT_NOTYPE)
434 Type = ELF::STT_OBJECT;
435 break;
436 case ELF::STT_TLS:
437 if (Type == ELF::STT_OBJECT || Type == ELF::STT_NOTYPE ||
438 Type == ELF::STT_GNU_IFUNC || Type == ELF::STT_FUNC)
439 Type = ELF::STT_TLS;
440 break;
441 }
442
443 return Type;
444}
445
Rafael Espindolae48421f2015-05-28 20:25:29 +0000446void ELFObjectWriter::writeSymbol(SymbolTableWriter &Writer,
447 uint32_t StringIndex, ELFSymbolData &MSD,
Daniel Dunbarfe0c28f2010-11-13 07:33:40 +0000448 const MCAsmLayout &Layout) {
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000449 const auto &Symbol = cast<MCSymbolELF>(*MSD.Symbol);
Rafael Espindolaa8695762015-06-02 00:25:12 +0000450 const MCSymbolELF *Base =
451 cast_or_null<MCSymbolELF>(Layout.getBaseSymbol(Symbol));
Rafael Espindola85a84912014-03-26 00:16:43 +0000452
453 // This has to be in sync with when computeSymbolTable uses SHN_ABS or
454 // SHN_COMMON.
Rafael Espindola4d37b2a2015-05-29 21:45:01 +0000455 bool IsReserved = !Base || Symbol.isCommon();
Rafael Espindola3fe87a12010-10-31 00:16:26 +0000456
Jack Carter2f8d9d92013-02-19 21:57:35 +0000457 // Binding and Type share the same byte as upper and lower nibbles
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000458 uint8_t Binding = Symbol.getBinding();
459 uint8_t Type = Symbol.getType();
Rafael Espindolaa6e3a592014-03-23 03:33:20 +0000460 if (Base) {
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000461 Type = mergeTypeForSet(Type, Base->getType());
Rafael Espindolaa6e3a592014-03-23 03:33:20 +0000462 }
Rafael Espindolaf8794ff2015-06-04 00:47:43 +0000463 uint8_t Info = (Binding << 4) | Type;
Jack Carter2f8d9d92013-02-19 21:57:35 +0000464
Joerg Sonnenbergerfc184732013-10-29 01:06:17 +0000465 // Other and Visibility share the same byte with Visibility using the lower
Jack Carter2f8d9d92013-02-19 21:57:35 +0000466 // 2 bits
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000467 uint8_t Visibility = Symbol.getVisibility();
Rafael Espindola8c006ee2015-06-04 05:59:23 +0000468 uint8_t Other = Symbol.getOther() | Visibility;
Rafael Espindola5f2d6a52010-10-06 21:02:29 +0000469
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000470 uint64_t Value = SymbolValue(*MSD.Symbol, Layout);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000471 uint64_t Size = 0;
Matt Fleming6c1ad482010-08-16 18:57:57 +0000472
Rafael Espindola7c23cba2015-05-29 17:24:52 +0000473 const MCExpr *ESize = MSD.Symbol->getSize();
Rafael Espindolaa041ef12014-03-27 00:28:24 +0000474 if (!ESize && Base)
Rafael Espindola7c23cba2015-05-29 17:24:52 +0000475 ESize = Base->getSize();
Rafael Espindolaf0591c12010-09-21 00:24:38 +0000476
Rafael Espindola73c0ae72010-12-22 16:03:00 +0000477 if (ESize) {
478 int64_t Res;
Rafael Espindola94a88d72015-04-06 15:27:57 +0000479 if (!ESize->evaluateKnownAbsolute(Res, Layout))
Rafael Espindola73c0ae72010-12-22 16:03:00 +0000480 report_fatal_error("Size expression must be absolute.");
481 Size = Res;
Matt Fleming6c1ad482010-08-16 18:57:57 +0000482 }
483
484 // Write out the symbol table entry
Rafael Espindolae48421f2015-05-28 20:25:29 +0000485 Writer.writeSymbol(StringIndex, Info, Value, Size, Other, MSD.SectionIndex,
486 IsReserved);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000487}
488
Rafael Espindola5904e122014-03-29 06:26:49 +0000489// It is always valid to create a relocation with a symbol. It is preferable
490// to use a relocation with a section if that is possible. Using the section
491// allows us to omit some local symbols from the symbol table.
Rafael Espindolab60c8292014-04-29 12:46:50 +0000492bool ELFObjectWriter::shouldRelocateWithSymbol(const MCAssembler &Asm,
493 const MCSymbolRefExpr *RefA,
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000494 const MCSymbol *S, uint64_t C,
Rafael Espindola5904e122014-03-29 06:26:49 +0000495 unsigned Type) const {
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000496 const auto *Sym = cast_or_null<MCSymbolELF>(S);
Rafael Espindola5904e122014-03-29 06:26:49 +0000497 // A PCRel relocation to an absolute value has no symbol (or section). We
498 // represent that with a relocation to a null section.
499 if (!RefA)
500 return false;
Rafael Espindola240028d2010-11-14 23:53:26 +0000501
Rafael Espindola5904e122014-03-29 06:26:49 +0000502 MCSymbolRefExpr::VariantKind Kind = RefA->getKind();
503 switch (Kind) {
504 default:
505 break;
506 // The .odp creation emits a relocation against the symbol ".TOC." which
507 // create a R_PPC64_TOC relocation. However the relocation symbol name
508 // in final object creation should be NULL, since the symbol does not
509 // really exist, it is just the reference to TOC base for the current
510 // object file. Since the symbol is undefined, returning false results
511 // in a relocation with a null section which is the desired result.
512 case MCSymbolRefExpr::VK_PPC_TOCBASE:
513 return false;
514
515 // These VariantKind cause the relocation to refer to something other than
516 // the symbol itself, like a linker generated table. Since the address of
517 // symbol is not relevant, we cannot replace the symbol with the
518 // section and patch the difference in the addend.
519 case MCSymbolRefExpr::VK_GOT:
520 case MCSymbolRefExpr::VK_PLT:
521 case MCSymbolRefExpr::VK_GOTPCREL:
522 case MCSymbolRefExpr::VK_Mips_GOT:
523 case MCSymbolRefExpr::VK_PPC_GOT_LO:
524 case MCSymbolRefExpr::VK_PPC_GOT_HI:
525 case MCSymbolRefExpr::VK_PPC_GOT_HA:
526 return true;
Rafael Espindola240028d2010-11-14 23:53:26 +0000527 }
Rafael Espindola240028d2010-11-14 23:53:26 +0000528
Rafael Espindola5904e122014-03-29 06:26:49 +0000529 // An undefined symbol is not in any section, so the relocation has to point
530 // to the symbol itself.
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000531 assert(Sym && "Expected a symbol");
532 if (Sym->isUndefined())
Rafael Espindola5904e122014-03-29 06:26:49 +0000533 return true;
534
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000535 unsigned Binding = Sym->getBinding();
Rafael Espindola5904e122014-03-29 06:26:49 +0000536 switch(Binding) {
537 default:
538 llvm_unreachable("Invalid Binding");
539 case ELF::STB_LOCAL:
540 break;
541 case ELF::STB_WEAK:
542 // If the symbol is weak, it might be overridden by a symbol in another
543 // file. The relocation has to point to the symbol so that the linker
544 // can update it.
545 return true;
546 case ELF::STB_GLOBAL:
547 // Global ELF symbols can be preempted by the dynamic linker. The relocation
548 // has to point to the symbol for a reason analogous to the STB_WEAK case.
549 return true;
Rafael Espindola8c3039b2010-11-15 16:33:49 +0000550 }
Rafael Espindola75d65b92010-09-25 05:42:19 +0000551
Rafael Espindola5904e122014-03-29 06:26:49 +0000552 // If a relocation points to a mergeable section, we have to be careful.
553 // If the offset is zero, a relocation with the section will encode the
554 // same information. With a non-zero offset, the situation is different.
555 // For example, a relocation can point 42 bytes past the end of a string.
556 // If we change such a relocation to use the section, the linker would think
557 // that it pointed to another string and subtracting 42 at runtime will
558 // produce the wrong value.
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000559 auto &Sec = cast<MCSectionELF>(Sym->getSection());
Rafael Espindola5904e122014-03-29 06:26:49 +0000560 unsigned Flags = Sec.getFlags();
561 if (Flags & ELF::SHF_MERGE) {
562 if (C != 0)
563 return true;
Rafael Espindolab1b49782014-04-02 12:15:20 +0000564
565 // It looks like gold has a bug (http://sourceware.org/PR16794) and can
566 // only handle section relocations to mergeable sections if using RELA.
567 if (!hasRelocationAddend())
568 return true;
Rafael Espindola9f75d5d2010-11-24 21:57:39 +0000569 }
570
Rafael Espindola5904e122014-03-29 06:26:49 +0000571 // Most TLS relocations use a got, so they need the symbol. Even those that
Rafael Espindola11527a12014-10-06 12:33:27 +0000572 // are just an offset (@tpoff), require a symbol in gold versions before
573 // 5efeedf61e4fe720fd3e9a08e6c91c10abb66d42 (2014-09-26) which fixed
574 // http://sourceware.org/PR16773.
Rafael Espindola5904e122014-03-29 06:26:49 +0000575 if (Flags & ELF::SHF_TLS)
576 return true;
Rafael Espindolad7565c32010-10-05 23:57:26 +0000577
Rafael Espindola9ef84412014-04-11 19:18:01 +0000578 // If the symbol is a thumb function the final relocation must set the lowest
579 // bit. With a symbol that is done by just having the symbol have that bit
580 // set, so we would lose the bit if we relocated with the section.
581 // FIXME: We could use the section but add the bit to the relocation value.
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000582 if (Asm.isThumbFunc(Sym))
Rafael Espindola9ef84412014-04-11 19:18:01 +0000583 return true;
584
Rafael Espindolaece40ca2015-05-29 18:26:09 +0000585 if (TargetObjectWriter->needsRelocateWithSymbol(*Sym, Type))
Rafael Espindola5904e122014-03-29 06:26:49 +0000586 return true;
587 return false;
Rafael Espindola75d65b92010-09-25 05:42:19 +0000588}
589
Rafael Espindoladb8a5862015-04-17 20:05:17 +0000590// True if the assembler knows nothing about the final value of the symbol.
591// This doesn't cover the comdat issues, since in those cases the assembler
592// can at least know that all symbols in the section will move together.
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000593static bool isWeak(const MCSymbolELF &Sym) {
594 if (Sym.getType() == ELF::STT_GNU_IFUNC)
Rafael Espindolaa635d832015-04-17 08:46:11 +0000595 return true;
596
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000597 switch (Sym.getBinding()) {
Rafael Espindolaa635d832015-04-17 08:46:11 +0000598 default:
599 llvm_unreachable("Unknown binding");
600 case ELF::STB_LOCAL:
601 return false;
602 case ELF::STB_GLOBAL:
Rafael Espindoladb8a5862015-04-17 20:05:17 +0000603 return false;
Rafael Espindolaa635d832015-04-17 08:46:11 +0000604 case ELF::STB_WEAK:
605 case ELF::STB_GNU_UNIQUE:
606 return true;
607 }
Rafael Espindolac9e70682015-03-24 23:48:44 +0000608}
609
Jim Grosbach36e60e92015-06-04 22:24:41 +0000610void ELFObjectWriter::recordRelocation(MCAssembler &Asm,
Jason W Kim495c2bb2010-12-06 21:57:34 +0000611 const MCAsmLayout &Layout,
612 const MCFragment *Fragment,
Rafael Espindola26585542015-01-19 21:11:14 +0000613 const MCFixup &Fixup, MCValue Target,
614 bool &IsPCRel, uint64_t &FixedValue) {
Rafael Espindola7549f872015-05-26 00:36:57 +0000615 const MCSectionELF &FixupSection = cast<MCSectionELF>(*Fragment->getParent());
Rafael Espindola5904e122014-03-29 06:26:49 +0000616 uint64_t C = Target.getConstant();
617 uint64_t FixupOffset = Layout.getFragmentOffset(Fragment) + Fixup.getOffset();
Jason W Kim495c2bb2010-12-06 21:57:34 +0000618
Rafael Espindola5904e122014-03-29 06:26:49 +0000619 if (const MCSymbolRefExpr *RefB = Target.getSymB()) {
620 assert(RefB->getKind() == MCSymbolRefExpr::VK_None &&
621 "Should not have constructed this");
Jason W Kim495c2bb2010-12-06 21:57:34 +0000622
Rafael Espindola5904e122014-03-29 06:26:49 +0000623 // Let A, B and C being the components of Target and R be the location of
624 // the fixup. If the fixup is not pcrel, we want to compute (A - B + C).
625 // If it is pcrel, we want to compute (A - B + C - R).
Jason W Kim495c2bb2010-12-06 21:57:34 +0000626
Rafael Espindola5904e122014-03-29 06:26:49 +0000627 // In general, ELF has no relocations for -B. It can only represent (A + C)
628 // or (A + C - R). If B = R + K and the relocation is not pcrel, we can
629 // replace B to implement it: (A - R - K + C)
630 if (IsPCRel)
Jim Grosbach6f482002015-05-18 18:43:14 +0000631 Asm.getContext().reportFatalError(
Rafael Espindola5904e122014-03-29 06:26:49 +0000632 Fixup.getLoc(),
633 "No relocation available to represent this relative expression");
David Majnemera45a1762014-01-06 07:39:46 +0000634
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000635 const auto &SymB = cast<MCSymbolELF>(RefB->getSymbol());
Jason W Kim495c2bb2010-12-06 21:57:34 +0000636
Rafael Espindola5904e122014-03-29 06:26:49 +0000637 if (SymB.isUndefined())
Jim Grosbach6f482002015-05-18 18:43:14 +0000638 Asm.getContext().reportFatalError(
Rafael Espindola5904e122014-03-29 06:26:49 +0000639 Fixup.getLoc(),
640 Twine("symbol '") + SymB.getName() +
641 "' can not be undefined in a subtraction expression");
Jason W Kim495c2bb2010-12-06 21:57:34 +0000642
Rafael Espindola5904e122014-03-29 06:26:49 +0000643 assert(!SymB.isAbsolute() && "Should have been folded");
644 const MCSection &SecB = SymB.getSection();
Rafael Espindola34948e52015-04-30 00:45:46 +0000645 if (&SecB != &FixupSection)
Jim Grosbach6f482002015-05-18 18:43:14 +0000646 Asm.getContext().reportFatalError(
Rafael Espindola5904e122014-03-29 06:26:49 +0000647 Fixup.getLoc(), "Cannot represent a difference across sections");
Jason W Kim495c2bb2010-12-06 21:57:34 +0000648
Rafael Espindolae3b2acf2015-05-29 18:47:23 +0000649 if (::isWeak(SymB))
Jim Grosbach6f482002015-05-18 18:43:14 +0000650 Asm.getContext().reportFatalError(
Rafael Espindolac9e70682015-03-24 23:48:44 +0000651 Fixup.getLoc(), "Cannot represent a subtraction with a weak symbol");
652
Duncan P. N. Exon Smith2a404832015-05-19 23:53:20 +0000653 uint64_t SymBOffset = Layout.getSymbolOffset(SymB);
Rafael Espindola5904e122014-03-29 06:26:49 +0000654 uint64_t K = SymBOffset - FixupOffset;
655 IsPCRel = true;
656 C -= K;
Jason W Kim495c2bb2010-12-06 21:57:34 +0000657 }
658
Rafael Espindola5904e122014-03-29 06:26:49 +0000659 // We either rejected the fixup or folded B into C at this point.
660 const MCSymbolRefExpr *RefA = Target.getSymA();
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000661 const auto *SymA = RefA ? cast<MCSymbolELF>(&RefA->getSymbol()) : nullptr;
Rafael Espindola5904e122014-03-29 06:26:49 +0000662
Rafael Espindola212fdde2015-06-03 21:52:06 +0000663 bool ViaWeakRef = false;
664 if (SymA && SymA->isVariable()) {
665 const MCExpr *Expr = SymA->getVariableValue();
666 if (const auto *Inner = dyn_cast<MCSymbolRefExpr>(Expr)) {
667 if (Inner->getKind() == MCSymbolRefExpr::VK_WEAKREF) {
668 SymA = cast<MCSymbolELF>(&Inner->getSymbol());
669 ViaWeakRef = true;
670 }
671 }
672 }
673
Rafael Espindolac03f44c2014-03-27 20:49:35 +0000674 unsigned Type = GetRelocType(Target, Fixup, IsPCRel);
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000675 bool RelocateWithSymbol = shouldRelocateWithSymbol(Asm, RefA, SymA, C, Type);
Rafael Espindola5904e122014-03-29 06:26:49 +0000676 if (!RelocateWithSymbol && SymA && !SymA->isUndefined())
Duncan P. N. Exon Smith2a404832015-05-19 23:53:20 +0000677 C += Layout.getSymbolOffset(*SymA);
Rafael Espindola5904e122014-03-29 06:26:49 +0000678
679 uint64_t Addend = 0;
680 if (hasRelocationAddend()) {
681 Addend = C;
682 C = 0;
683 }
684
685 FixedValue = C;
686
Rafael Espindola5904e122014-03-29 06:26:49 +0000687 if (!RelocateWithSymbol) {
688 const MCSection *SecA =
689 (SymA && !SymA->isUndefined()) ? &SymA->getSection() : nullptr;
Rafael Espindolab6613022014-10-17 01:48:58 +0000690 auto *ELFSec = cast_or_null<MCSectionELF>(SecA);
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000691 const auto *SectionSymbol =
692 ELFSec ? cast<MCSymbolELF>(ELFSec->getBeginSymbol()) : nullptr;
Rafael Espindolaa401eee2015-06-04 15:33:30 +0000693 if (SectionSymbol)
694 SectionSymbol->setUsedInReloc();
Rafael Espindolab6613022014-10-17 01:48:58 +0000695 ELFRelocationEntry Rec(FixupOffset, SectionSymbol, Type, Addend);
Rafael Espindola34948e52015-04-30 00:45:46 +0000696 Relocations[&FixupSection].push_back(Rec);
Rafael Espindola5904e122014-03-29 06:26:49 +0000697 return;
698 }
Roman Divackydfbecd12011-08-04 19:08:19 +0000699
Rafael Espindola5904e122014-03-29 06:26:49 +0000700 if (SymA) {
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000701 if (const MCSymbolELF *R = Renames.lookup(SymA))
Rafael Espindola5904e122014-03-29 06:26:49 +0000702 SymA = R;
Rafael Espindolad7facaf2011-08-04 13:05:26 +0000703
Rafael Espindola212fdde2015-06-03 21:52:06 +0000704 if (ViaWeakRef)
705 SymA->setIsWeakrefUsedInReloc();
Rafael Espindola5904e122014-03-29 06:26:49 +0000706 else
Rafael Espindolaada43f62015-06-03 21:30:10 +0000707 SymA->setUsedInReloc();
Rafael Espindola5904e122014-03-29 06:26:49 +0000708 }
709 ELFRelocationEntry Rec(FixupOffset, SymA, Type, Addend);
Rafael Espindola34948e52015-04-30 00:45:46 +0000710 Relocations[&FixupSection].push_back(Rec);
Rafael Espindola5904e122014-03-29 06:26:49 +0000711 return;
Jason W Kim495c2bb2010-12-06 21:57:34 +0000712}
713
Rafael Espindola3b5ee552014-04-28 13:39:57 +0000714bool ELFObjectWriter::isInSymtab(const MCAsmLayout &Layout,
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000715 const MCSymbolELF &Symbol, bool Used,
Rafael Espindola3b5ee552014-04-28 13:39:57 +0000716 bool Renamed) {
Rafael Espindola7fadc0e2014-03-20 02:12:01 +0000717 if (Symbol.isVariable()) {
718 const MCExpr *Expr = Symbol.getVariableValue();
719 if (const MCSymbolRefExpr *Ref = dyn_cast<MCSymbolRefExpr>(Expr)) {
720 if (Ref->getKind() == MCSymbolRefExpr::VK_WEAKREF)
721 return false;
722 }
723 }
Rafael Espindola16145972010-11-01 14:28:48 +0000724
Rafael Espindolaee8d1512010-10-19 19:31:37 +0000725 if (Used)
726 return true;
727
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000728 if (Renamed)
729 return false;
730
Rafael Espindolaef1e8632015-06-03 21:23:21 +0000731 if (Symbol.isVariable() && Symbol.isUndefined()) {
732 // FIXME: this is here just to diagnose the case of a var = commmon_sym.
733 Layout.getBaseSymbol(Symbol);
734 return false;
Rafael Espindola3b5ee552014-04-28 13:39:57 +0000735 }
Rafael Espindola9e18e962011-02-23 20:22:07 +0000736
Rafael Espindolaef1e8632015-06-03 21:23:21 +0000737 if (Symbol.isUndefined() && !Symbol.isBindingSet())
Rafael Espindolaa5efd6a2010-10-27 14:44:52 +0000738 return false;
739
Rafael Espindolaee8d1512010-10-19 19:31:37 +0000740 if (Symbol.isTemporary())
Rafael Espindolab1d07892010-10-05 18:01:23 +0000741 return false;
742
Rafael Espindolaa401eee2015-06-04 15:33:30 +0000743 if (Symbol.getType() == ELF::STT_SECTION)
744 return false;
745
Rafael Espindolab1d07892010-10-05 18:01:23 +0000746 return true;
747}
748
Rafael Espindolaea1b3942015-04-07 19:00:17 +0000749void ELFObjectWriter::computeSymbolTable(
750 MCAssembler &Asm, const MCAsmLayout &Layout,
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000751 const SectionIndexMapTy &SectionIndexMap, const RevGroupMapTy &RevGroupMap,
752 SectionOffsetsTy &SectionOffsets) {
Rafael Espindola5960cee2015-05-22 23:58:30 +0000753 MCContext &Ctx = Asm.getContext();
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000754 SymbolTableWriter Writer(*this, is64Bit());
755
Rafael Espindola5960cee2015-05-22 23:58:30 +0000756 // Symbol table
757 unsigned EntrySize = is64Bit() ? ELF::SYMENTRY_SIZE64 : ELF::SYMENTRY_SIZE32;
758 MCSectionELF *SymtabSection =
759 Ctx.getELFSection(".symtab", ELF::SHT_SYMTAB, 0, EntrySize, "");
760 SymtabSection->setAlignment(is64Bit() ? 8 : 4);
761 SymbolTableIndex = addToSectionTable(SymtabSection);
762
Rafael Espindolab20fbb82015-06-05 18:21:00 +0000763 align(SymtabSection->getAlignment());
David Majnemerabdb2d2a2015-09-01 16:19:03 +0000764 uint64_t SecStart = getStream().tell();
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000765
766 // The first entry is the undefined symbol entry.
767 Writer.writeSymbol(0, 0, 0, 0, 0, 0, false);
768
Rafael Espindolacfbd35c2015-05-28 20:11:34 +0000769 std::vector<ELFSymbolData> LocalSymbolData;
770 std::vector<ELFSymbolData> ExternalSymbolData;
Rafael Espindolacfbd35c2015-05-28 20:11:34 +0000771
Rafael Espindolabee6e9f2010-10-14 16:34:44 +0000772 // Add the data for the symbols.
Rafael Espindola5960cee2015-05-22 23:58:30 +0000773 bool HasLargeSectionIndex = false;
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000774 for (const MCSymbol &S : Asm.symbols()) {
775 const auto &Symbol = cast<MCSymbolELF>(S);
Rafael Espindolaada43f62015-06-03 21:30:10 +0000776 bool Used = Symbol.isUsedInReloc();
Rafael Espindola212fdde2015-06-03 21:52:06 +0000777 bool WeakrefUsed = Symbol.isWeakrefUsedInReloc();
Rafael Espindola8c52a9b2015-06-03 21:41:59 +0000778 bool isSignature = Symbol.isSignature();
Rafael Espindola7d0ba342010-11-14 04:17:37 +0000779
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000780 if (!isInSymtab(Layout, Symbol, Used || WeakrefUsed || isSignature,
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000781 Renames.count(&Symbol)))
Matt Fleming6c1ad482010-08-16 18:57:57 +0000782 continue;
783
Rafael Espindola6dff8142015-06-25 20:10:45 +0000784 if (Symbol.isTemporary() && Symbol.isUndefined())
785 Ctx.reportFatalError(SMLoc(), "Undefined temporary");
786
Matt Fleming6c1ad482010-08-16 18:57:57 +0000787 ELFSymbolData MSD;
Rafael Espindolaa8695762015-06-02 00:25:12 +0000788 MSD.Symbol = cast<MCSymbolELF>(&Symbol);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000789
Rafael Espindola8c52a9b2015-06-03 21:41:59 +0000790 bool Local = Symbol.getBinding() == ELF::STB_LOCAL;
Rafael Espindola6dff8142015-06-25 20:10:45 +0000791 assert(Local || !Symbol.isTemporary());
792
Rafael Espindolaf4b44302015-05-29 15:07:27 +0000793 if (Symbol.isAbsolute()) {
Rafael Espindola022bb762014-03-24 03:43:21 +0000794 MSD.SectionIndex = ELF::SHN_ABS;
Rafael Espindola14672502015-05-29 17:48:04 +0000795 } else if (Symbol.isCommon()) {
Rafael Espindolabee6e9f2010-10-14 16:34:44 +0000796 assert(!Local);
Rafael Espindolaf0591c12010-09-21 00:24:38 +0000797 MSD.SectionIndex = ELF::SHN_COMMON;
Rafael Espindolaf4b44302015-05-29 15:07:27 +0000798 } else if (Symbol.isUndefined()) {
Rafael Espindola5960cee2015-05-22 23:58:30 +0000799 if (isSignature && !Used) {
Rafael Espindola89feff32015-04-28 22:59:58 +0000800 MSD.SectionIndex = RevGroupMap.lookup(&Symbol);
Rafael Espindola5960cee2015-05-22 23:58:30 +0000801 if (MSD.SectionIndex >= ELF::SHN_LORESERVE)
802 HasLargeSectionIndex = true;
803 } else {
Rafael Espindola7d0ba342010-11-14 04:17:37 +0000804 MSD.SectionIndex = ELF::SHN_UNDEF;
Rafael Espindola5960cee2015-05-22 23:58:30 +0000805 }
Matt Fleming6c1ad482010-08-16 18:57:57 +0000806 } else {
Rafael Espindolad6340032010-11-10 21:51:05 +0000807 const MCSectionELF &Section =
Rafael Espindolaf4b44302015-05-29 15:07:27 +0000808 static_cast<const MCSectionELF &>(Symbol.getSection());
Rafael Espindolad6340032010-11-10 21:51:05 +0000809 MSD.SectionIndex = SectionIndexMap.lookup(&Section);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000810 assert(MSD.SectionIndex && "Invalid section index!");
Rafael Espindola5960cee2015-05-22 23:58:30 +0000811 if (MSD.SectionIndex >= ELF::SHN_LORESERVE)
812 HasLargeSectionIndex = true;
Rafael Espindolafbcf0db2010-10-15 15:39:06 +0000813 }
814
Rafael Espindola5a67ed12015-01-22 14:20:45 +0000815 // The @@@ in symbol version is replaced with @ in undefined symbols and @@
816 // in defined ones.
817 //
818 // FIXME: All name handling should be done before we get to the writer,
NAKAMURA Takumib01d86b2015-02-25 11:02:00 +0000819 // including dealing with GNU-style version suffixes. Fixing this isn't
Rafael Espindola5a67ed12015-01-22 14:20:45 +0000820 // trivial.
821 //
822 // We thus have to be careful to not perform the symbol version replacement
823 // blindly:
824 //
825 // The ELF format is used on Windows by the MCJIT engine. Thus, on
826 // Windows, the ELFObjectWriter can encounter symbols mangled using the MS
827 // Visual Studio C++ name mangling scheme. Symbols mangled using the MSVC
828 // C++ name mangling can legally have "@@@" as a sub-string. In that case,
829 // the EFLObjectWriter should not interpret the "@@@" sub-string as
830 // specifying GNU-style symbol versioning. The ELFObjectWriter therefore
831 // checks for the MSVC C++ name mangling prefix which is either "?", "@?",
832 // "__imp_?" or "__imp_@?".
833 //
834 // It would have been interesting to perform the MS mangling prefix check
835 // only when the target triple is of the form *-pc-windows-elf. But, it
836 // seems that this information is not easily accessible from the
837 // ELFObjectWriter.
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000838 StringRef Name = Symbol.getName();
Evgeniy Stepanov9e0d41a2015-06-22 23:36:03 +0000839 SmallString<32> Buf;
Rafael Espindola5a67ed12015-01-22 14:20:45 +0000840 if (!Name.startswith("?") && !Name.startswith("@?") &&
841 !Name.startswith("__imp_?") && !Name.startswith("__imp_@?")) {
842 // This symbol isn't following the MSVC C++ name mangling convention. We
843 // can thus safely interpret the @@@ in symbol names as specifying symbol
844 // versioning.
Rafael Espindola5a67ed12015-01-22 14:20:45 +0000845 size_t Pos = Name.find("@@@");
846 if (Pos != StringRef::npos) {
847 Buf += Name.substr(0, Pos);
848 unsigned Skip = MSD.SectionIndex == ELF::SHN_UNDEF ? 2 : 1;
849 Buf += Name.substr(Pos + Skip);
850 Name = Buf;
851 }
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000852 }
Rafael Espindolab6613022014-10-17 01:48:58 +0000853
854 // Sections have their own string table
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000855 if (Symbol.getType() != ELF::STT_SECTION)
Rafael Espindolab6613022014-10-17 01:48:58 +0000856 MSD.Name = StrTabBuilder.add(Name);
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000857
Rafael Espindola10d23872015-05-29 14:20:40 +0000858 if (Local)
Rafael Espindolaa5efd6a2010-10-27 14:44:52 +0000859 LocalSymbolData.push_back(MSD);
860 else
861 ExternalSymbolData.push_back(MSD);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000862 }
863
Rafael Espindola2cd19512015-07-02 16:59:57 +0000864 // This holds the .symtab_shndx section index.
865 unsigned SymtabShndxSectionIndex = 0;
866
Rafael Espindola5960cee2015-05-22 23:58:30 +0000867 if (HasLargeSectionIndex) {
868 MCSectionELF *SymtabShndxSection =
869 Ctx.getELFSection(".symtab_shndxr", ELF::SHT_SYMTAB_SHNDX, 0, 4, "");
870 SymtabShndxSectionIndex = addToSectionTable(SymtabShndxSection);
871 SymtabShndxSection->setAlignment(4);
872 }
873
Rafael Espindola1fd36272015-05-28 19:29:15 +0000874 ArrayRef<std::string> FileNames = Asm.getFileNames();
875 for (const std::string &Name : FileNames)
Rafael Espindola66f3c9c2015-05-28 18:03:20 +0000876 StrTabBuilder.add(Name);
Hans Wennborg83e6e1e2014-04-30 16:25:02 +0000877
Hans Wennborgf26bfc12014-09-29 22:43:20 +0000878 StrTabBuilder.finalize(StringTableBuilder::ELF);
Hans Wennborg83e6e1e2014-04-30 16:25:02 +0000879
Rafael Espindola0cbea292015-05-28 20:00:13 +0000880 for (const std::string &Name : FileNames)
881 Writer.writeSymbol(StrTabBuilder.getOffset(Name),
882 ELF::STT_FILE | ELF::STB_LOCAL, 0, 0, ELF::STV_DEFAULT,
883 ELF::SHN_ABS, true);
884
Matt Fleming6c1ad482010-08-16 18:57:57 +0000885 // Symbols are required to be in lexicographic order.
886 array_pod_sort(LocalSymbolData.begin(), LocalSymbolData.end());
887 array_pod_sort(ExternalSymbolData.begin(), ExternalSymbolData.end());
Matt Fleming6c1ad482010-08-16 18:57:57 +0000888
889 // Set the symbol indices. Local symbols must come before all other
890 // symbols with non-local bindings.
Rafael Espindola1fd36272015-05-28 19:29:15 +0000891 unsigned Index = FileNames.size() + 1;
Rafael Espindola0e3decf2010-11-14 03:12:24 +0000892
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000893 for (ELFSymbolData &MSD : LocalSymbolData) {
Daniel Jasper41de8022015-06-23 11:31:32 +0000894 unsigned StringIndex = MSD.Symbol->getType() == ELF::STT_SECTION
895 ? 0
896 : StrTabBuilder.getOffset(MSD.Name);
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000897 MSD.Symbol->setIndex(Index++);
Rafael Espindolae48421f2015-05-28 20:25:29 +0000898 writeSymbol(Writer, StringIndex, MSD, Layout);
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000899 }
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000900
901 // Write the symbol table entries.
Rafael Espindola0cbea292015-05-28 20:00:13 +0000902 LastLocalSymbolIndex = Index;
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000903
Rafael Espindoladcda9972015-05-28 19:43:20 +0000904 for (ELFSymbolData &MSD : ExternalSymbolData) {
Rafael Espindolae48421f2015-05-28 20:25:29 +0000905 unsigned StringIndex = StrTabBuilder.getOffset(MSD.Name);
Rafael Espindola0cbea292015-05-28 20:00:13 +0000906 MSD.Symbol->setIndex(Index++);
Rafael Espindolae48421f2015-05-28 20:25:29 +0000907 writeSymbol(Writer, StringIndex, MSD, Layout);
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000908 assert(MSD.Symbol->getBinding() != ELF::STB_LOCAL);
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000909 }
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000910
David Majnemerabdb2d2a2015-09-01 16:19:03 +0000911 uint64_t SecEnd = getStream().tell();
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000912 SectionOffsets[SymtabSection] = std::make_pair(SecStart, SecEnd);
913
914 ArrayRef<uint32_t> ShndxIndexes = Writer.getShndxIndexes();
915 if (ShndxIndexes.empty()) {
916 assert(SymtabShndxSectionIndex == 0);
917 return;
918 }
919 assert(SymtabShndxSectionIndex != 0);
920
David Majnemerabdb2d2a2015-09-01 16:19:03 +0000921 SecStart = getStream().tell();
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000922 const MCSectionELF *SymtabShndxSection =
923 SectionTable[SymtabShndxSectionIndex - 1];
924 for (uint32_t Index : ShndxIndexes)
925 write(Index);
David Majnemerabdb2d2a2015-09-01 16:19:03 +0000926 SecEnd = getStream().tell();
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000927 SectionOffsets[SymtabShndxSection] = std::make_pair(SecStart, SecEnd);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000928}
929
Rafael Espindola0709a7b2015-05-21 19:20:38 +0000930MCSectionELF *
Rafael Espindola0a82ad72015-05-21 20:43:13 +0000931ELFObjectWriter::createRelocationSection(MCContext &Ctx,
Rafael Espindolabda19802015-04-30 14:21:49 +0000932 const MCSectionELF &Sec) {
Rafael Espindola34948e52015-04-30 00:45:46 +0000933 if (Relocations[&Sec].empty())
Rafael Espindolabda19802015-04-30 14:21:49 +0000934 return nullptr;
Rafael Espindola1557fd62011-03-20 18:44:20 +0000935
Rafael Espindola34948e52015-04-30 00:45:46 +0000936 const StringRef SectionName = Sec.getSectionName();
Rafael Espindolaead85492015-02-17 20:31:13 +0000937 std::string RelaSectionName = hasRelocationAddend() ? ".rela" : ".rel";
938 RelaSectionName += SectionName;
Benjamin Kramerfd054152010-08-17 17:56:13 +0000939
Rafael Espindolaead85492015-02-17 20:31:13 +0000940 unsigned EntrySize;
941 if (hasRelocationAddend())
942 EntrySize = is64Bit() ? sizeof(ELF::Elf64_Rela) : sizeof(ELF::Elf32_Rela);
943 else
944 EntrySize = is64Bit() ? sizeof(ELF::Elf64_Rel) : sizeof(ELF::Elf32_Rel);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000945
Rafael Espindolaead85492015-02-17 20:31:13 +0000946 unsigned Flags = 0;
Rafael Espindola34948e52015-04-30 00:45:46 +0000947 if (Sec.getFlags() & ELF::SHF_GROUP)
Rafael Espindolaead85492015-02-17 20:31:13 +0000948 Flags = ELF::SHF_GROUP;
Rafael Espindolaead85492015-02-17 20:31:13 +0000949
Rafael Espindola0709a7b2015-05-21 19:20:38 +0000950 MCSectionELF *RelaSection = Ctx.createELFRelSection(
Rafael Espindolaead85492015-02-17 20:31:13 +0000951 RelaSectionName, hasRelocationAddend() ? ELF::SHT_RELA : ELF::SHT_REL,
Rafael Espindola34948e52015-04-30 00:45:46 +0000952 Flags, EntrySize, Sec.getGroup(), &Sec);
Rafael Espindola0a82ad72015-05-21 20:43:13 +0000953 RelaSection->setAlignment(is64Bit() ? 8 : 4);
Rafael Espindolabda19802015-04-30 14:21:49 +0000954 return RelaSection;
Rafael Espindola1557fd62011-03-20 18:44:20 +0000955}
Matt Fleming6c1ad482010-08-16 18:57:57 +0000956
David Blaikie8019bf82014-04-10 21:53:53 +0000957// Include the debug info compression header:
958// "ZLIB" followed by 8 bytes representing the uncompressed size of the section,
959// useful for consumers to preallocate a buffer to decompress into.
David Blaikie76d3a3c2014-04-18 21:52:26 +0000960static bool
David Blaikiee06e8012014-04-11 22:11:50 +0000961prependCompressionHeader(uint64_t Size,
962 SmallVectorImpl<char> &CompressedContents) {
Benjamin Kramer7149aab2015-03-01 18:09:56 +0000963 const StringRef Magic = "ZLIB";
David Blaikie76d3a3c2014-04-18 21:52:26 +0000964 if (Size <= Magic.size() + sizeof(Size) + CompressedContents.size())
965 return false;
David Blaikie8019bf82014-04-10 21:53:53 +0000966 if (sys::IsLittleEndianHost)
Artyom Skrobov9aea8432014-06-14 13:18:07 +0000967 sys::swapByteOrder(Size);
David Blaikie8019bf82014-04-10 21:53:53 +0000968 CompressedContents.insert(CompressedContents.begin(),
969 Magic.size() + sizeof(Size), 0);
970 std::copy(Magic.begin(), Magic.end(), CompressedContents.begin());
971 std::copy(reinterpret_cast<char *>(&Size),
972 reinterpret_cast<char *>(&Size + 1),
973 CompressedContents.begin() + Magic.size());
David Blaikie76d3a3c2014-04-18 21:52:26 +0000974 return true;
David Blaikie8019bf82014-04-10 21:53:53 +0000975}
976
Rafael Espindolae15b1b72015-05-27 13:30:50 +0000977void ELFObjectWriter::writeSectionData(const MCAssembler &Asm, MCSection &Sec,
Rafael Espindola868b3f42015-04-30 21:51:58 +0000978 const MCAsmLayout &Layout) {
Rafael Espindolae15b1b72015-05-27 13:30:50 +0000979 MCSectionELF &Section = static_cast<MCSectionELF &>(Sec);
Rafael Espindola868b3f42015-04-30 21:51:58 +0000980 StringRef SectionName = Section.getSectionName();
David Blaikie8019bf82014-04-10 21:53:53 +0000981
Rafael Espindola868b3f42015-04-30 21:51:58 +0000982 // Compressing debug_frame requires handling alignment fragments which is
983 // more work (possibly generalizing MCAssembler.cpp:writeFragment to allow
984 // for writing to arbitrary buffers) for little benefit.
985 if (!Asm.getContext().getAsmInfo()->compressDebugSections() ||
986 !SectionName.startswith(".debug_") || SectionName == ".debug_frame") {
Rafael Espindola64acc7f2015-05-26 02:17:21 +0000987 Asm.writeSectionData(&Section, Layout);
Rafael Espindola868b3f42015-04-30 21:51:58 +0000988 return;
989 }
990
David Majnemerabdb2d2a2015-09-01 16:19:03 +0000991 SmallVector<char, 128> UncompressedData;
992 raw_svector_ostream VecOS(UncompressedData);
993 raw_pwrite_stream &OldStream = getStream();
994 setStream(VecOS);
995 Asm.writeSectionData(&Section, Layout);
996 setStream(OldStream);
David Blaikie8019bf82014-04-10 21:53:53 +0000997
Rafael Espindola868b3f42015-04-30 21:51:58 +0000998 SmallVector<char, 128> CompressedContents;
David Blaikie8019bf82014-04-10 21:53:53 +0000999 zlib::Status Success = zlib::compress(
1000 StringRef(UncompressedData.data(), UncompressedData.size()),
1001 CompressedContents);
Rafael Espindola868b3f42015-04-30 21:51:58 +00001002 if (Success != zlib::StatusOK) {
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001003 getStream() << UncompressedData;
David Blaikie8019bf82014-04-10 21:53:53 +00001004 return;
Rafael Espindola868b3f42015-04-30 21:51:58 +00001005 }
David Blaikie8019bf82014-04-10 21:53:53 +00001006
Rafael Espindola868b3f42015-04-30 21:51:58 +00001007 if (!prependCompressionHeader(UncompressedData.size(), CompressedContents)) {
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001008 getStream() << UncompressedData;
Rafael Espindola868b3f42015-04-30 21:51:58 +00001009 return;
1010 }
David Blaikie8019bf82014-04-10 21:53:53 +00001011 Asm.getContext().renameELFSection(&Section,
1012 (".z" + SectionName.drop_front(1)).str());
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001013 getStream() << CompressedContents;
David Blaikie8019bf82014-04-10 21:53:53 +00001014}
1015
Daniel Dunbarfe0c28f2010-11-13 07:33:40 +00001016void ELFObjectWriter::WriteSecHdrEntry(uint32_t Name, uint32_t Type,
1017 uint64_t Flags, uint64_t Address,
1018 uint64_t Offset, uint64_t Size,
1019 uint32_t Link, uint32_t Info,
1020 uint64_t Alignment,
1021 uint64_t EntrySize) {
Jim Grosbach36e60e92015-06-04 22:24:41 +00001022 write32(Name); // sh_name: index into string table
1023 write32(Type); // sh_type
Matt Fleming6c1ad482010-08-16 18:57:57 +00001024 WriteWord(Flags); // sh_flags
1025 WriteWord(Address); // sh_addr
1026 WriteWord(Offset); // sh_offset
1027 WriteWord(Size); // sh_size
Jim Grosbach36e60e92015-06-04 22:24:41 +00001028 write32(Link); // sh_link
1029 write32(Info); // sh_info
Matt Fleming6c1ad482010-08-16 18:57:57 +00001030 WriteWord(Alignment); // sh_addralign
1031 WriteWord(EntrySize); // sh_entsize
1032}
1033
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001034void ELFObjectWriter::writeRelocations(const MCAssembler &Asm,
Rafael Espindola34948e52015-04-30 00:45:46 +00001035 const MCSectionELF &Sec) {
1036 std::vector<ELFRelocationEntry> &Relocs = Relocations[&Sec];
Akira Hatanaka64ad2cf2012-03-23 23:06:45 +00001037
Petar Jovanovic0380d0b2015-04-14 13:23:34 +00001038 // Sort the relocation entries. Most targets just sort by Offset, but some
1039 // (e.g., MIPS) have additional constraints.
1040 TargetObjectWriter->sortRelocs(Asm, Relocs);
Matt Fleming6c1ad482010-08-16 18:57:57 +00001041
1042 for (unsigned i = 0, e = Relocs.size(); i != e; ++i) {
Rafael Espindola5904e122014-03-29 06:26:49 +00001043 const ELFRelocationEntry &Entry = Relocs[e - i - 1];
Rafael Espindola5e9ed902015-05-28 20:53:09 +00001044 unsigned Index = Entry.Symbol ? Entry.Symbol->getIndex() : 0;
Rafael Espindola5904e122014-03-29 06:26:49 +00001045
Rafael Espindolafdaae0d2010-12-18 03:27:34 +00001046 if (is64Bit()) {
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001047 write(Entry.Offset);
Jack Carter8ad0c272012-06-27 22:28:30 +00001048 if (TargetObjectWriter->isN64()) {
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001049 write(uint32_t(Index));
Benjamin Kramer6c3c3492010-09-09 18:01:29 +00001050
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001051 write(TargetObjectWriter->getRSsym(Entry.Type));
1052 write(TargetObjectWriter->getRType3(Entry.Type));
1053 write(TargetObjectWriter->getRType2(Entry.Type));
1054 write(TargetObjectWriter->getRType(Entry.Type));
Rafael Espindola5904e122014-03-29 06:26:49 +00001055 } else {
Jack Carter8ad0c272012-06-27 22:28:30 +00001056 struct ELF::Elf64_Rela ERE64;
Rafael Espindola5904e122014-03-29 06:26:49 +00001057 ERE64.setSymbolAndType(Index, Entry.Type);
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001058 write(ERE64.r_info);
Jack Carter8ad0c272012-06-27 22:28:30 +00001059 }
Rafael Espindolafdaae0d2010-12-18 03:27:34 +00001060 if (hasRelocationAddend())
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001061 write(Entry.Addend);
Benjamin Kramer6c3c3492010-09-09 18:01:29 +00001062 } else {
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001063 write(uint32_t(Entry.Offset));
Benjamin Kramer6c3c3492010-09-09 18:01:29 +00001064
Rafael Espindolabce26a12010-10-05 15:11:03 +00001065 struct ELF::Elf32_Rela ERE32;
Rafael Espindola5904e122014-03-29 06:26:49 +00001066 ERE32.setSymbolAndType(Index, Entry.Type);
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001067 write(ERE32.r_info);
Benjamin Kramer6c3c3492010-09-09 18:01:29 +00001068
Rafael Espindolafdaae0d2010-12-18 03:27:34 +00001069 if (hasRelocationAddend())
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001070 write(uint32_t(Entry.Addend));
Benjamin Kramer6c3c3492010-09-09 18:01:29 +00001071 }
Matt Fleming6c1ad482010-08-16 18:57:57 +00001072 }
1073}
1074
Rafael Espindolab1863912015-04-30 21:10:06 +00001075const MCSectionELF *ELFObjectWriter::createStringTable(MCContext &Ctx) {
Rafael Espindola08850b32015-05-25 21:56:55 +00001076 const MCSectionELF *StrtabSection = SectionTable[StringTableIndex - 1];
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001077 getStream() << StrTabBuilder.data();
Rafael Espindolabda19802015-04-30 14:21:49 +00001078 return StrtabSection;
Rafael Espindolaa3e9a222010-11-11 18:13:52 +00001079}
1080
Rafael Espindolae92c1bf2015-05-21 19:42:35 +00001081void ELFObjectWriter::writeSection(const SectionIndexMapTy &SectionIndexMap,
1082 uint32_t GroupSymbolIndex, uint64_t Offset,
1083 uint64_t Size, const MCSectionELF &Section) {
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001084 uint64_t sh_link = 0;
1085 uint64_t sh_info = 0;
1086
1087 switch(Section.getType()) {
Rafael Espindola86fe2fe2015-04-06 03:16:51 +00001088 default:
1089 // Nothing to do.
1090 break;
1091
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001092 case ELF::SHT_DYNAMIC:
Rafael Espindola74ef4802015-04-30 21:20:06 +00001093 llvm_unreachable("SHT_DYNAMIC in a relocatable object");
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001094
1095 case ELF::SHT_REL:
1096 case ELF::SHT_RELA: {
Rafael Espindola3a7c0eb2015-02-17 20:37:50 +00001097 sh_link = SymbolTableIndex;
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001098 assert(sh_link && ".symtab not found");
Rafael Espindolab73b70e2015-04-06 03:09:30 +00001099 const MCSectionELF *InfoSection = Section.getAssociatedSection();
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001100 sh_info = SectionIndexMap.lookup(InfoSection);
1101 break;
1102 }
1103
1104 case ELF::SHT_SYMTAB:
1105 case ELF::SHT_DYNSYM:
1106 sh_link = StringTableIndex;
1107 sh_info = LastLocalSymbolIndex;
1108 break;
1109
1110 case ELF::SHT_SYMTAB_SHNDX:
1111 sh_link = SymbolTableIndex;
1112 break;
1113
Nick Lewyckyc6ac5f72011-10-07 23:28:32 +00001114 case ELF::SHT_GROUP:
Rafael Espindolaa3e9a222010-11-11 18:13:52 +00001115 sh_link = SymbolTableIndex;
1116 sh_info = GroupSymbolIndex;
1117 break;
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001118 }
1119
Logan Chien4b724422013-02-05 14:18:59 +00001120 if (TargetObjectWriter->getEMachine() == ELF::EM_ARM &&
Rafael Espindola61e8ce32015-04-06 04:25:18 +00001121 Section.getType() == ELF::SHT_ARM_EXIDX)
1122 sh_link = SectionIndexMap.lookup(Section.getAssociatedSection());
Logan Chien4b724422013-02-05 14:18:59 +00001123
Rafael Espindola5960cee2015-05-22 23:58:30 +00001124 WriteSecHdrEntry(StrTabBuilder.getOffset(Section.getSectionName()),
Rafael Espindola28687582015-05-21 19:36:43 +00001125 Section.getType(), Section.getFlags(), 0, Offset, Size,
1126 sh_link, sh_info, Section.getAlignment(),
1127 Section.getEntrySize());
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001128}
1129
Rafael Espindola7fe7e052015-02-17 20:40:59 +00001130void ELFObjectWriter::writeSectionHeader(
Rafael Espindola57c80832015-06-04 20:55:49 +00001131 const MCAsmLayout &Layout, const SectionIndexMapTy &SectionIndexMap,
Rafael Espindolaa8201692015-04-28 15:26:21 +00001132 const SectionOffsetsTy &SectionOffsets) {
Rafael Espindolab1863912015-04-30 21:10:06 +00001133 const unsigned NumSections = SectionTable.size();
Rafael Espindolaa3e9a222010-11-11 18:13:52 +00001134
Matt Fleming6c1ad482010-08-16 18:57:57 +00001135 // Null section first.
Rafael Espindola3fe87a12010-10-31 00:16:26 +00001136 uint64_t FirstSectionSize =
Rafael Espindolabf0db6c2015-04-15 13:07:47 +00001137 (NumSections + 1) >= ELF::SHN_LORESERVE ? NumSections + 1 : 0;
Rafael Espindola88d1f632015-04-29 20:25:24 +00001138 WriteSecHdrEntry(0, 0, 0, 0, 0, FirstSectionSize, 0, 0, 0, 0);
Matt Fleming6c1ad482010-08-16 18:57:57 +00001139
Rafael Espindola08850b32015-05-25 21:56:55 +00001140 for (const MCSectionELF *Section : SectionTable) {
Rafael Espindolaa3e9a222010-11-11 18:13:52 +00001141 uint32_t GroupSymbolIndex;
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001142 unsigned Type = Section->getType();
1143 if (Type != ELF::SHT_GROUP)
Rafael Espindolaa3e9a222010-11-11 18:13:52 +00001144 GroupSymbolIndex = 0;
1145 else
Rafael Espindola5e9ed902015-05-28 20:53:09 +00001146 GroupSymbolIndex = Section->getGroup()->getIndex();
Matt Fleming6c1ad482010-08-16 18:57:57 +00001147
Rafael Espindolabda19802015-04-30 14:21:49 +00001148 const std::pair<uint64_t, uint64_t> &Offsets =
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001149 SectionOffsets.find(Section)->second;
Rafael Espindola1aa20fc2015-05-21 19:46:39 +00001150 uint64_t Size;
Rafael Espindola5a1e80b2015-05-26 02:00:36 +00001151 if (Type == ELF::SHT_NOBITS)
1152 Size = Layout.getSectionAddressSize(Section);
1153 else
Rafael Espindola1aa20fc2015-05-21 19:46:39 +00001154 Size = Offsets.second - Offsets.first;
Rafael Espindola60ebca92010-12-02 03:09:06 +00001155
Rafael Espindolae92c1bf2015-05-21 19:42:35 +00001156 writeSection(SectionIndexMap, GroupSymbolIndex, Offsets.first, Size,
Rafael Espindola28687582015-05-21 19:36:43 +00001157 *Section);
Matt Fleming6c1ad482010-08-16 18:57:57 +00001158 }
1159}
1160
Jim Grosbach36e60e92015-06-04 22:24:41 +00001161void ELFObjectWriter::writeObject(MCAssembler &Asm,
Rafael Espindola1557fd62011-03-20 18:44:20 +00001162 const MCAsmLayout &Layout) {
Rafael Espindolabda19802015-04-30 14:21:49 +00001163 MCContext &Ctx = Asm.getContext();
Rafael Espindola5960cee2015-05-22 23:58:30 +00001164 MCSectionELF *StrtabSection =
1165 Ctx.getELFSection(".strtab", ELF::SHT_STRTAB, 0);
1166 StringTableIndex = addToSectionTable(StrtabSection);
Rafael Espindolabda19802015-04-30 14:21:49 +00001167
Rafael Espindola1557fd62011-03-20 18:44:20 +00001168 RevGroupMapTy RevGroupMap;
1169 SectionIndexMapTy SectionIndexMap;
1170
Rafael Espindolabda19802015-04-30 14:21:49 +00001171 std::map<const MCSymbol *, std::vector<const MCSectionELF *>> GroupMembers;
Rafael Espindola1557fd62011-03-20 18:44:20 +00001172
Rafael Espindola1557fd62011-03-20 18:44:20 +00001173 // Write out the ELF header ...
Rafael Espindola8c7829b2015-04-29 20:39:37 +00001174 writeHeader(Asm);
Rafael Espindola1557fd62011-03-20 18:44:20 +00001175
Rafael Espindola7230f802015-04-08 11:41:24 +00001176 // ... then the sections ...
Rafael Espindolabda19802015-04-30 14:21:49 +00001177 SectionOffsetsTy SectionOffsets;
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001178 std::vector<MCSectionELF *> Groups;
1179 std::vector<MCSectionELF *> Relocations;
Rafael Espindolae15b1b72015-05-27 13:30:50 +00001180 for (MCSection &Sec : Asm) {
1181 MCSectionELF &Section = static_cast<MCSectionELF &>(Sec);
Rafael Espindolabda19802015-04-30 14:21:49 +00001182
Rafael Espindolab20fbb82015-06-05 18:21:00 +00001183 align(Section.getAlignment());
Rafael Espindola1557fd62011-03-20 18:44:20 +00001184
Rafael Espindola642a2212015-04-14 22:54:16 +00001185 // Remember the offset into the file for this section.
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001186 uint64_t SecStart = getStream().tell();
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001187
Rafael Espindola0ccf9b72015-06-02 21:30:13 +00001188 const MCSymbolELF *SignatureSymbol = Section.getGroup();
Rafael Espindolae15b1b72015-05-27 13:30:50 +00001189 writeSectionData(Asm, Section, Layout);
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001190
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001191 uint64_t SecEnd = getStream().tell();
Rafael Espindolabda19802015-04-30 14:21:49 +00001192 SectionOffsets[&Section] = std::make_pair(SecStart, SecEnd);
1193
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001194 MCSectionELF *RelSection = createRelocationSection(Ctx, Section);
Rafael Espindolabda19802015-04-30 14:21:49 +00001195
1196 if (SignatureSymbol) {
Rafael Espindolab5d316b2015-05-29 20:21:02 +00001197 Asm.registerSymbol(*SignatureSymbol);
Rafael Espindolabda19802015-04-30 14:21:49 +00001198 unsigned &GroupIdx = RevGroupMap[SignatureSymbol];
1199 if (!GroupIdx) {
Rafael Espindola0709a7b2015-05-21 19:20:38 +00001200 MCSectionELF *Group = Ctx.createELFGroupSection(SignatureSymbol);
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001201 GroupIdx = addToSectionTable(Group);
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001202 Group->setAlignment(4);
1203 Groups.push_back(Group);
Rafael Espindolabda19802015-04-30 14:21:49 +00001204 }
Rafael Espindola7830fc82015-06-05 17:54:25 +00001205 std::vector<const MCSectionELF *> &Members =
1206 GroupMembers[SignatureSymbol];
1207 Members.push_back(&Section);
Rafael Espindolabda19802015-04-30 14:21:49 +00001208 if (RelSection)
Rafael Espindola7830fc82015-06-05 17:54:25 +00001209 Members.push_back(RelSection);
Rafael Espindolabda19802015-04-30 14:21:49 +00001210 }
1211
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001212 SectionIndexMap[&Section] = addToSectionTable(&Section);
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001213 if (RelSection) {
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001214 SectionIndexMap[RelSection] = addToSectionTable(RelSection);
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001215 Relocations.push_back(RelSection);
1216 }
Rafael Espindolabda19802015-04-30 14:21:49 +00001217 }
1218
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001219 for (MCSectionELF *Group : Groups) {
Rafael Espindolab20fbb82015-06-05 18:21:00 +00001220 align(Group->getAlignment());
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001221
1222 // Remember the offset into the file for this section.
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001223 uint64_t SecStart = getStream().tell();
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001224
1225 const MCSymbol *SignatureSymbol = Group->getGroup();
1226 assert(SignatureSymbol);
1227 write(uint32_t(ELF::GRP_COMDAT));
1228 for (const MCSectionELF *Member : GroupMembers[SignatureSymbol]) {
1229 uint32_t SecIndex = SectionIndexMap.lookup(Member);
1230 write(SecIndex);
1231 }
1232
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001233 uint64_t SecEnd = getStream().tell();
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001234 SectionOffsets[Group] = std::make_pair(SecStart, SecEnd);
1235 }
1236
1237 // Compute symbol table information.
Rafael Espindolaaa486e92015-05-28 17:54:01 +00001238 computeSymbolTable(Asm, Layout, SectionIndexMap, RevGroupMap, SectionOffsets);
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001239
1240 for (MCSectionELF *RelSection : Relocations) {
Rafael Espindolab20fbb82015-06-05 18:21:00 +00001241 align(RelSection->getAlignment());
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001242
1243 // Remember the offset into the file for this section.
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001244 uint64_t SecStart = getStream().tell();
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001245
1246 writeRelocations(Asm, *RelSection->getAssociatedSection());
1247
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001248 uint64_t SecEnd = getStream().tell();
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001249 SectionOffsets[RelSection] = std::make_pair(SecStart, SecEnd);
Rafael Espindola642a2212015-04-14 22:54:16 +00001250 }
1251
Rafael Espindola88d1f632015-04-29 20:25:24 +00001252 {
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001253 uint64_t SecStart = getStream().tell();
Rafael Espindolab1863912015-04-30 21:10:06 +00001254 const MCSectionELF *Sec = createStringTable(Ctx);
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001255 uint64_t SecEnd = getStream().tell();
Rafael Espindolabda19802015-04-30 14:21:49 +00001256 SectionOffsets[Sec] = std::make_pair(SecStart, SecEnd);
Rafael Espindola88abc392015-04-29 20:34:31 +00001257 }
1258
Rafael Espindola642a2212015-04-14 22:54:16 +00001259 uint64_t NaturalAlignment = is64Bit() ? 8 : 4;
Rafael Espindolab20fbb82015-06-05 18:21:00 +00001260 align(NaturalAlignment);
Rafael Espindola1557fd62011-03-20 18:44:20 +00001261
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001262 const unsigned SectionHeaderOffset = getStream().tell();
Rafael Espindola642a2212015-04-14 22:54:16 +00001263
Rafael Espindola1557fd62011-03-20 18:44:20 +00001264 // ... then the section header table ...
Rafael Espindola57c80832015-06-04 20:55:49 +00001265 writeSectionHeader(Layout, SectionIndexMap, SectionOffsets);
Rafael Espindola642a2212015-04-14 22:54:16 +00001266
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001267 uint16_t NumSections = (SectionTable.size() + 1 >= ELF::SHN_LORESERVE)
Aaron Ballman9cab7322015-04-30 14:03:12 +00001268 ? (uint16_t)ELF::SHN_UNDEF
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001269 : SectionTable.size() + 1;
Rafael Espindola8c7829b2015-04-29 20:39:37 +00001270 if (sys::IsLittleEndianHost != IsLittleEndian)
1271 sys::swapByteOrder(NumSections);
1272 unsigned NumSectionsOffset;
1273
Rafael Espindola642a2212015-04-14 22:54:16 +00001274 if (is64Bit()) {
1275 uint64_t Val = SectionHeaderOffset;
1276 if (sys::IsLittleEndianHost != IsLittleEndian)
1277 sys::swapByteOrder(Val);
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001278 getStream().pwrite(reinterpret_cast<char *>(&Val), sizeof(Val),
1279 offsetof(ELF::Elf64_Ehdr, e_shoff));
Rafael Espindola8c7829b2015-04-29 20:39:37 +00001280 NumSectionsOffset = offsetof(ELF::Elf64_Ehdr, e_shnum);
Rafael Espindola642a2212015-04-14 22:54:16 +00001281 } else {
1282 uint32_t Val = SectionHeaderOffset;
1283 if (sys::IsLittleEndianHost != IsLittleEndian)
1284 sys::swapByteOrder(Val);
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001285 getStream().pwrite(reinterpret_cast<char *>(&Val), sizeof(Val),
1286 offsetof(ELF::Elf32_Ehdr, e_shoff));
Rafael Espindola8c7829b2015-04-29 20:39:37 +00001287 NumSectionsOffset = offsetof(ELF::Elf32_Ehdr, e_shnum);
Rafael Espindola642a2212015-04-14 22:54:16 +00001288 }
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001289 getStream().pwrite(reinterpret_cast<char *>(&NumSections),
1290 sizeof(NumSections), NumSectionsOffset);
Rafael Espindola1557fd62011-03-20 18:44:20 +00001291}
1292
Jim Grosbach36e60e92015-06-04 22:24:41 +00001293bool ELFObjectWriter::isSymbolRefDifferenceFullyResolvedImpl(
Rafael Espindola95fb9b92015-06-02 20:38:46 +00001294 const MCAssembler &Asm, const MCSymbol &SA, const MCFragment &FB,
Rafael Espindola35d61892015-04-17 21:15:17 +00001295 bool InSet, bool IsPCRel) const {
Rafael Espindola95fb9b92015-06-02 20:38:46 +00001296 const auto &SymA = cast<MCSymbolELF>(SA);
Rafael Espindola35d61892015-04-17 21:15:17 +00001297 if (IsPCRel) {
1298 assert(!InSet);
Rafael Espindolae3b2acf2015-05-29 18:47:23 +00001299 if (::isWeak(SymA))
Rafael Espindola35d61892015-04-17 21:15:17 +00001300 return false;
1301 }
Jim Grosbach36e60e92015-06-04 22:24:41 +00001302 return MCObjectWriter::isSymbolRefDifferenceFullyResolvedImpl(Asm, SymA, FB,
Rafael Espindola35d61892015-04-17 21:15:17 +00001303 InSet, IsPCRel);
Eli Friedmand92d17b2011-03-03 07:24:36 +00001304}
1305
Rafael Espindola95fb9b92015-06-02 20:38:46 +00001306bool ELFObjectWriter::isWeak(const MCSymbol &S) const {
1307 const auto &Sym = cast<MCSymbolELF>(S);
Rafael Espindolae3b2acf2015-05-29 18:47:23 +00001308 if (::isWeak(Sym))
Rafael Espindoladb8a5862015-04-17 20:05:17 +00001309 return true;
1310
Rafael Espindoladb8a5862015-04-17 20:05:17 +00001311 // It is invalid to replace a reference to a global in a comdat
1312 // with a reference to a local since out of comdat references
1313 // to a local are forbidden.
1314 // We could try to return false for more cases, like the reference
1315 // being in the same comdat or Sym being an alias to another global,
1316 // but it is not clear if it is worth the effort.
Rafael Espindola95fb9b92015-06-02 20:38:46 +00001317 if (Sym.getBinding() != ELF::STB_GLOBAL)
Rafael Espindola29c82702015-04-20 12:44:06 +00001318 return false;
Rafael Espindoladb8a5862015-04-17 20:05:17 +00001319
Rafael Espindola29c82702015-04-20 12:44:06 +00001320 if (!Sym.isInSection())
1321 return false;
1322
1323 const auto &Sec = cast<MCSectionELF>(Sym.getSection());
1324 return Sec.getGroup();
Rafael Espindolaf275ad82015-03-25 13:16:53 +00001325}
1326
Rafael Espindola6b5e56c2010-12-17 17:45:22 +00001327MCObjectWriter *llvm::createELFObjectWriter(MCELFObjectTargetWriter *MOTW,
Rafael Espindola5560a4c2015-04-14 22:14:34 +00001328 raw_pwrite_stream &OS,
Rafael Espindolafdaae0d2010-12-18 03:27:34 +00001329 bool IsLittleEndian) {
Rafael Espindola34a68af2011-12-22 03:24:43 +00001330 return new ELFObjectWriter(MOTW, OS, IsLittleEndian);
Rafael Espindolab264d332011-12-21 17:30:17 +00001331}