blob: 85bbcab8264705f739ab64ae2d975dd95dd20410 [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
Eugene Zelenkod3a6c892017-02-11 00:27:28 +000014#include "llvm/ADT/ArrayRef.h"
15#include "llvm/ADT/DenseMap.h"
Chandler Carruth6bda14b2017-06-06 11:49:48 +000016#include "llvm/ADT/STLExtras.h"
Rafael Espindola29abd972011-12-22 03:38:00 +000017#include "llvm/ADT/SmallString.h"
Eugene Zelenkod3a6c892017-02-11 00:27:28 +000018#include "llvm/ADT/SmallVector.h"
Eugene Zelenkod3a6c892017-02-11 00:27:28 +000019#include "llvm/ADT/StringRef.h"
20#include "llvm/ADT/Twine.h"
Zachary Turner264b5d92017-06-07 03:48:56 +000021#include "llvm/BinaryFormat/ELF.h"
Rafael Espindolaceecfe5b2017-07-11 23:56:10 +000022#include "llvm/MC/MCAsmBackend.h"
David Blaikie8019bf82014-04-10 21:53:53 +000023#include "llvm/MC/MCAsmInfo.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000024#include "llvm/MC/MCAsmLayout.h"
Rafael Espindola29abd972011-12-22 03:38:00 +000025#include "llvm/MC/MCAssembler.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000026#include "llvm/MC/MCContext.h"
Eugene Zelenkod3a6c892017-02-11 00:27:28 +000027#include "llvm/MC/MCELFObjectWriter.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000028#include "llvm/MC/MCExpr.h"
Eugene Zelenkod3a6c892017-02-11 00:27:28 +000029#include "llvm/MC/MCFixup.h"
Rafael Espindolaceecfe5b2017-07-11 23:56:10 +000030#include "llvm/MC/MCFixupKindInfo.h"
Eugene Zelenkod3a6c892017-02-11 00:27:28 +000031#include "llvm/MC/MCFragment.h"
Craig Topper6e80c282012-03-26 06:58:25 +000032#include "llvm/MC/MCObjectWriter.h"
Eugene Zelenkod3a6c892017-02-11 00:27:28 +000033#include "llvm/MC/MCSection.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000034#include "llvm/MC/MCSectionELF.h"
Eugene Zelenkod3a6c892017-02-11 00:27:28 +000035#include "llvm/MC/MCSymbol.h"
Rafael Espindolaa8695762015-06-02 00:25:12 +000036#include "llvm/MC/MCSymbolELF.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000037#include "llvm/MC/MCValue.h"
Rafael Espindola97de4742014-07-03 02:01:39 +000038#include "llvm/MC/StringTableBuilder.h"
Eugene Zelenkod3a6c892017-02-11 00:27:28 +000039#include "llvm/Support/Allocator.h"
40#include "llvm/Support/Casting.h"
David Blaikie8019bf82014-04-10 21:53:53 +000041#include "llvm/Support/Compression.h"
Chandler Carruthd9903882015-01-14 11:23:27 +000042#include "llvm/Support/Endian.h"
George Rimar167ca4a2017-01-17 15:45:07 +000043#include "llvm/Support/Error.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000044#include "llvm/Support/ErrorHandling.h"
Eugene Zelenkod3a6c892017-02-11 00:27:28 +000045#include "llvm/Support/Host.h"
46#include "llvm/Support/MathExtras.h"
47#include "llvm/Support/SMLoc.h"
Rafael Espindolafc063e82015-10-22 18:32:06 +000048#include "llvm/Support/StringSaver.h"
Eugene Zelenkod3a6c892017-02-11 00:27:28 +000049#include "llvm/Support/SwapByteOrder.h"
50#include "llvm/Support/raw_ostream.h"
51#include <algorithm>
52#include <cassert>
53#include <cstddef>
54#include <cstdint>
55#include <map>
56#include <memory>
57#include <string>
58#include <utility>
Matt Fleming6c1ad482010-08-16 18:57:57 +000059#include <vector>
Eugene Zelenkoecefe5a2016-02-02 18:20:45 +000060
Matt Fleming6c1ad482010-08-16 18:57:57 +000061using namespace llvm;
62
Jason W Kimc09e7262011-05-11 22:53:06 +000063#undef DEBUG_TYPE
64#define DEBUG_TYPE "reloc-info"
65
Rafael Espindola29abd972011-12-22 03:38:00 +000066namespace {
Eugene Zelenkod3a6c892017-02-11 00:27:28 +000067
Eugene Zelenko7975b992017-04-26 22:31:39 +000068using SectionIndexMapTy = DenseMap<const MCSectionELF *, uint32_t>;
Rafael Espindola10be0832014-03-25 23:44:25 +000069
Rafael Espindola91fd2772015-04-29 21:09:32 +000070class ELFObjectWriter;
71
Rafael Espindola10be0832014-03-25 23:44:25 +000072class SymbolTableWriter {
Rafael Espindola91fd2772015-04-29 21:09:32 +000073 ELFObjectWriter &EWriter;
Rafael Espindola10be0832014-03-25 23:44:25 +000074 bool Is64Bit;
Rafael Espindola10be0832014-03-25 23:44:25 +000075
Rafael Espindola91fd2772015-04-29 21:09:32 +000076 // indexes we are going to write to .symtab_shndx.
77 std::vector<uint32_t> ShndxIndexes;
Rafael Espindola10be0832014-03-25 23:44:25 +000078
79 // The numbel of symbols written so far.
80 unsigned NumWritten;
81
82 void createSymtabShndx();
83
Rafael Espindola91fd2772015-04-29 21:09:32 +000084 template <typename T> void write(T Value);
Rafael Espindola10be0832014-03-25 23:44:25 +000085
86public:
Rafael Espindola91fd2772015-04-29 21:09:32 +000087 SymbolTableWriter(ELFObjectWriter &EWriter, bool Is64Bit);
Rafael Espindola10be0832014-03-25 23:44:25 +000088
89 void writeSymbol(uint32_t name, uint8_t info, uint64_t value, uint64_t size,
90 uint8_t other, uint32_t shndx, bool Reserved);
Rafael Espindola91fd2772015-04-29 21:09:32 +000091
92 ArrayRef<uint32_t> getShndxIndexes() const { return ShndxIndexes; }
Rafael Espindola10be0832014-03-25 23:44:25 +000093};
94
Rafael Espindola29abd972011-12-22 03:38:00 +000095class ELFObjectWriter : public MCObjectWriter {
David Blaikiea0fa2622016-04-12 21:45:53 +000096 static uint64_t SymbolValue(const MCSymbol &Sym, const MCAsmLayout &Layout);
97 static bool isInSymtab(const MCAsmLayout &Layout, const MCSymbolELF &Symbol,
98 bool Used, bool Renamed);
Rafael Espindola29abd972011-12-22 03:38:00 +000099
David Blaikiea0fa2622016-04-12 21:45:53 +0000100 /// Helper struct for containing some precomputed information on symbols.
101 struct ELFSymbolData {
102 const MCSymbolELF *Symbol;
103 uint32_t SectionIndex;
104 StringRef Name;
Rafael Espindola29abd972011-12-22 03:38:00 +0000105
David Blaikiea0fa2622016-04-12 21:45:53 +0000106 // Support lexicographic sorting.
107 bool operator<(const ELFSymbolData &RHS) const {
108 unsigned LHSType = Symbol->getType();
109 unsigned RHSType = RHS.Symbol->getType();
110 if (LHSType == ELF::STT_SECTION && RHSType != ELF::STT_SECTION)
111 return false;
112 if (LHSType != ELF::STT_SECTION && RHSType == ELF::STT_SECTION)
113 return true;
114 if (LHSType == ELF::STT_SECTION && RHSType == ELF::STT_SECTION)
115 return SectionIndex < RHS.SectionIndex;
116 return Name < RHS.Name;
Rafael Espindola29abd972011-12-22 03:38:00 +0000117 }
Rafael Espindola29abd972011-12-22 03:38:00 +0000118 };
David Blaikiea0fa2622016-04-12 21:45:53 +0000119
120 /// The target specific ELF writer instance.
121 std::unique_ptr<MCELFObjectTargetWriter> TargetObjectWriter;
122
123 DenseMap<const MCSymbolELF *, const MCSymbolELF *> Renames;
124
Eugene Zelenkod3a6c892017-02-11 00:27:28 +0000125 DenseMap<const MCSectionELF *, std::vector<ELFRelocationEntry>> Relocations;
David Blaikiea0fa2622016-04-12 21:45:53 +0000126
127 /// @}
128 /// @name Symbol Table Data
129 /// @{
130
David Blaikiea0fa2622016-04-12 21:45:53 +0000131 StringTableBuilder StrTabBuilder{StringTableBuilder::ELF};
132
133 /// @}
134
135 // This holds the symbol table index of the last local symbol.
136 unsigned LastLocalSymbolIndex;
137 // This holds the .strtab section index.
138 unsigned StringTableIndex;
139 // This holds the .symtab section index.
140 unsigned SymbolTableIndex;
141
142 // Sections in the order they are to be output in the section table.
143 std::vector<const MCSectionELF *> SectionTable;
144 unsigned addToSectionTable(const MCSectionELF *Sec);
145
146 // TargetObjectWriter wrappers.
147 bool is64Bit() const { return TargetObjectWriter->is64Bit(); }
148 bool hasRelocationAddend() const {
149 return TargetObjectWriter->hasRelocationAddend();
150 }
151 unsigned getRelocType(MCContext &Ctx, const MCValue &Target,
152 const MCFixup &Fixup, bool IsPCRel) const {
153 return TargetObjectWriter->getRelocType(Ctx, Target, Fixup, IsPCRel);
154 }
155
156 void align(unsigned Alignment);
157
George Rimarc91e38c2016-05-27 12:27:32 +0000158 bool maybeWriteCompression(uint64_t Size,
159 SmallVectorImpl<char> &CompressedContents,
160 bool ZLibStyle, unsigned Alignment);
161
David Blaikiea0fa2622016-04-12 21:45:53 +0000162public:
Lang Hamesdcb312b2017-10-09 23:53:15 +0000163 ELFObjectWriter(std::unique_ptr<MCELFObjectTargetWriter> MOTW,
164 raw_pwrite_stream &OS, bool IsLittleEndian)
165 : MCObjectWriter(OS, IsLittleEndian),
166 TargetObjectWriter(std::move(MOTW)) {}
David Blaikiea0fa2622016-04-12 21:45:53 +0000167
Eugene Zelenkod3a6c892017-02-11 00:27:28 +0000168 ~ELFObjectWriter() override = default;
169
David Blaikiea0fa2622016-04-12 21:45:53 +0000170 void reset() override {
171 Renames.clear();
172 Relocations.clear();
173 StrTabBuilder.clear();
174 SectionTable.clear();
175 MCObjectWriter::reset();
176 }
177
David Blaikiea0fa2622016-04-12 21:45:53 +0000178 void WriteWord(uint64_t W) {
179 if (is64Bit())
180 write64(W);
181 else
182 write32(W);
183 }
184
185 template <typename T> void write(T Val) {
186 if (IsLittleEndian)
187 support::endian::Writer<support::little>(getStream()).write(Val);
188 else
189 support::endian::Writer<support::big>(getStream()).write(Val);
190 }
191
192 void writeHeader(const MCAssembler &Asm);
193
194 void writeSymbol(SymbolTableWriter &Writer, uint32_t StringIndex,
195 ELFSymbolData &MSD, const MCAsmLayout &Layout);
196
197 // Start and end offset of each section
Eugene Zelenko7975b992017-04-26 22:31:39 +0000198 using SectionOffsetsTy =
199 std::map<const MCSectionELF *, std::pair<uint64_t, uint64_t>>;
David Blaikiea0fa2622016-04-12 21:45:53 +0000200
201 bool shouldRelocateWithSymbol(const MCAssembler &Asm,
202 const MCSymbolRefExpr *RefA,
203 const MCSymbol *Sym, uint64_t C,
204 unsigned Type) const;
205
206 void recordRelocation(MCAssembler &Asm, const MCAsmLayout &Layout,
207 const MCFragment *Fragment, const MCFixup &Fixup,
Rafael Espindolaceecfe5b2017-07-11 23:56:10 +0000208 MCValue Target, uint64_t &FixedValue) override;
David Blaikiea0fa2622016-04-12 21:45:53 +0000209
210 // Map from a signature symbol to the group section index
Eugene Zelenko7975b992017-04-26 22:31:39 +0000211 using RevGroupMapTy = DenseMap<const MCSymbol *, unsigned>;
David Blaikiea0fa2622016-04-12 21:45:53 +0000212
213 /// Compute the symbol table data
214 ///
215 /// \param Asm - The assembler.
216 /// \param SectionIndexMap - Maps a section to its index.
217 /// \param RevGroupMap - Maps a signature symbol to the group section.
218 void computeSymbolTable(MCAssembler &Asm, const MCAsmLayout &Layout,
219 const SectionIndexMapTy &SectionIndexMap,
220 const RevGroupMapTy &RevGroupMap,
221 SectionOffsetsTy &SectionOffsets);
222
223 MCSectionELF *createRelocationSection(MCContext &Ctx,
224 const MCSectionELF &Sec);
225
226 const MCSectionELF *createStringTable(MCContext &Ctx);
227
228 void executePostLayoutBinding(MCAssembler &Asm,
229 const MCAsmLayout &Layout) override;
230
231 void writeSectionHeader(const MCAsmLayout &Layout,
232 const SectionIndexMapTy &SectionIndexMap,
233 const SectionOffsetsTy &SectionOffsets);
234
235 void writeSectionData(const MCAssembler &Asm, MCSection &Sec,
236 const MCAsmLayout &Layout);
237
238 void WriteSecHdrEntry(uint32_t Name, uint32_t Type, uint64_t Flags,
239 uint64_t Address, uint64_t Offset, uint64_t Size,
240 uint32_t Link, uint32_t Info, uint64_t Alignment,
241 uint64_t EntrySize);
242
243 void writeRelocations(const MCAssembler &Asm, const MCSectionELF &Sec);
244
Vassil Vassilev89e36cc2017-03-06 15:50:59 +0000245 using MCObjectWriter::isSymbolRefDifferenceFullyResolvedImpl;
David Blaikiea0fa2622016-04-12 21:45:53 +0000246 bool isSymbolRefDifferenceFullyResolvedImpl(const MCAssembler &Asm,
247 const MCSymbol &SymA,
248 const MCFragment &FB, bool InSet,
249 bool IsPCRel) const override;
250
David Blaikiea0fa2622016-04-12 21:45:53 +0000251 void writeObject(MCAssembler &Asm, const MCAsmLayout &Layout) override;
252 void writeSection(const SectionIndexMapTy &SectionIndexMap,
253 uint32_t GroupSymbolIndex, uint64_t Offset, uint64_t Size,
254 const MCSectionELF &Section);
255};
Eugene Zelenkod3a6c892017-02-11 00:27:28 +0000256
Eugene Zelenkoecefe5a2016-02-02 18:20:45 +0000257} // end anonymous namespace
Rafael Espindola29abd972011-12-22 03:38:00 +0000258
Rafael Espindolab20fbb82015-06-05 18:21:00 +0000259void ELFObjectWriter::align(unsigned Alignment) {
David Majnemerabdb2d2a2015-09-01 16:19:03 +0000260 uint64_t Padding = OffsetToAlignment(getStream().tell(), Alignment);
Rafael Espindolab20fbb82015-06-05 18:21:00 +0000261 WriteZeros(Padding);
262}
263
Rafael Espindola08850b32015-05-25 21:56:55 +0000264unsigned ELFObjectWriter::addToSectionTable(const MCSectionELF *Sec) {
Rafael Espindola03d7abb2015-04-30 20:53:27 +0000265 SectionTable.push_back(Sec);
Rafael Espindola5960cee2015-05-22 23:58:30 +0000266 StrTabBuilder.add(Sec->getSectionName());
Rafael Espindola03d7abb2015-04-30 20:53:27 +0000267 return SectionTable.size();
268}
269
Rafael Espindola10be0832014-03-25 23:44:25 +0000270void SymbolTableWriter::createSymtabShndx() {
Rafael Espindola91fd2772015-04-29 21:09:32 +0000271 if (!ShndxIndexes.empty())
Rafael Espindola10be0832014-03-25 23:44:25 +0000272 return;
273
Rafael Espindola91fd2772015-04-29 21:09:32 +0000274 ShndxIndexes.resize(NumWritten);
Rafael Espindola10be0832014-03-25 23:44:25 +0000275}
276
Rafael Espindola91fd2772015-04-29 21:09:32 +0000277template <typename T> void SymbolTableWriter::write(T Value) {
278 EWriter.write(Value);
Rafael Espindola10be0832014-03-25 23:44:25 +0000279}
280
Rafael Espindola91fd2772015-04-29 21:09:32 +0000281SymbolTableWriter::SymbolTableWriter(ELFObjectWriter &EWriter, bool Is64Bit)
282 : EWriter(EWriter), Is64Bit(Is64Bit), NumWritten(0) {}
Rafael Espindola10be0832014-03-25 23:44:25 +0000283
284void SymbolTableWriter::writeSymbol(uint32_t name, uint8_t info, uint64_t value,
285 uint64_t size, uint8_t other,
286 uint32_t shndx, bool Reserved) {
287 bool LargeIndex = shndx >= ELF::SHN_LORESERVE && !Reserved;
288
289 if (LargeIndex)
290 createSymtabShndx();
291
Rafael Espindola91fd2772015-04-29 21:09:32 +0000292 if (!ShndxIndexes.empty()) {
Rafael Espindola10be0832014-03-25 23:44:25 +0000293 if (LargeIndex)
Rafael Espindola91fd2772015-04-29 21:09:32 +0000294 ShndxIndexes.push_back(shndx);
Rafael Espindola10be0832014-03-25 23:44:25 +0000295 else
Rafael Espindola91fd2772015-04-29 21:09:32 +0000296 ShndxIndexes.push_back(0);
Rafael Espindola10be0832014-03-25 23:44:25 +0000297 }
298
299 uint16_t Index = LargeIndex ? uint16_t(ELF::SHN_XINDEX) : shndx;
300
Rafael Espindola10be0832014-03-25 23:44:25 +0000301 if (Is64Bit) {
Rafael Espindola91fd2772015-04-29 21:09:32 +0000302 write(name); // st_name
303 write(info); // st_info
304 write(other); // st_other
305 write(Index); // st_shndx
306 write(value); // st_value
307 write(size); // st_size
Rafael Espindola10be0832014-03-25 23:44:25 +0000308 } else {
Rafael Espindola91fd2772015-04-29 21:09:32 +0000309 write(name); // st_name
310 write(uint32_t(value)); // st_value
311 write(uint32_t(size)); // st_size
312 write(info); // st_info
313 write(other); // st_other
314 write(Index); // st_shndx
Rafael Espindola10be0832014-03-25 23:44:25 +0000315 }
316
317 ++NumWritten;
318}
319
Matt Fleming6c1ad482010-08-16 18:57:57 +0000320// Emit the ELF header.
Rafael Espindola8c7829b2015-04-29 20:39:37 +0000321void ELFObjectWriter::writeHeader(const MCAssembler &Asm) {
Matt Fleming6c1ad482010-08-16 18:57:57 +0000322 // ELF Header
323 // ----------
324 //
325 // Note
326 // ----
327 // emitWord method behaves differently for ELF32 and ELF64, writing
328 // 4 bytes in the former and 8 in the latter.
329
Jim Grosbach36e60e92015-06-04 22:24:41 +0000330 writeBytes(ELF::ElfMagic); // e_ident[EI_MAG0] to e_ident[EI_MAG3]
Matt Fleming6c1ad482010-08-16 18:57:57 +0000331
Jim Grosbach36e60e92015-06-04 22:24:41 +0000332 write8(is64Bit() ? ELF::ELFCLASS64 : ELF::ELFCLASS32); // e_ident[EI_CLASS]
Matt Fleming6c1ad482010-08-16 18:57:57 +0000333
334 // e_ident[EI_DATA]
Jim Grosbach36e60e92015-06-04 22:24:41 +0000335 write8(isLittleEndian() ? ELF::ELFDATA2LSB : ELF::ELFDATA2MSB);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000336
Jim Grosbach36e60e92015-06-04 22:24:41 +0000337 write8(ELF::EV_CURRENT); // e_ident[EI_VERSION]
Roman Divacky3b727f52010-09-09 17:57:50 +0000338 // e_ident[EI_OSABI]
Jim Grosbach36e60e92015-06-04 22:24:41 +0000339 write8(TargetObjectWriter->getOSABI());
340 write8(0); // e_ident[EI_ABIVERSION]
Matt Fleming6c1ad482010-08-16 18:57:57 +0000341
342 WriteZeros(ELF::EI_NIDENT - ELF::EI_PAD);
343
Jim Grosbach36e60e92015-06-04 22:24:41 +0000344 write16(ELF::ET_REL); // e_type
Matt Fleming6c1ad482010-08-16 18:57:57 +0000345
Jim Grosbach36e60e92015-06-04 22:24:41 +0000346 write16(TargetObjectWriter->getEMachine()); // e_machine = target
Matt Fleming6c1ad482010-08-16 18:57:57 +0000347
Jim Grosbach36e60e92015-06-04 22:24:41 +0000348 write32(ELF::EV_CURRENT); // e_version
Matt Fleming6c1ad482010-08-16 18:57:57 +0000349 WriteWord(0); // e_entry, no entry point in .o file
350 WriteWord(0); // e_phoff, no program header for .o
Rafael Espindola642a2212015-04-14 22:54:16 +0000351 WriteWord(0); // e_shoff = sec hdr table off in bytes
Matt Fleming6c1ad482010-08-16 18:57:57 +0000352
Jason W Kim4761fba2011-02-04 21:41:11 +0000353 // e_flags = whatever the target wants
Jim Grosbach36e60e92015-06-04 22:24:41 +0000354 write32(Asm.getELFHeaderEFlags());
Matt Fleming6c1ad482010-08-16 18:57:57 +0000355
356 // e_ehsize = ELF header size
Jim Grosbach36e60e92015-06-04 22:24:41 +0000357 write16(is64Bit() ? sizeof(ELF::Elf64_Ehdr) : sizeof(ELF::Elf32_Ehdr));
Matt Fleming6c1ad482010-08-16 18:57:57 +0000358
Jim Grosbach36e60e92015-06-04 22:24:41 +0000359 write16(0); // e_phentsize = prog header entry size
360 write16(0); // e_phnum = # prog header entries = 0
Matt Fleming6c1ad482010-08-16 18:57:57 +0000361
362 // e_shentsize = Section header entry size
Jim Grosbach36e60e92015-06-04 22:24:41 +0000363 write16(is64Bit() ? sizeof(ELF::Elf64_Shdr) : sizeof(ELF::Elf32_Shdr));
Matt Fleming6c1ad482010-08-16 18:57:57 +0000364
365 // e_shnum = # of section header ents
Jim Grosbach36e60e92015-06-04 22:24:41 +0000366 write16(0);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000367
368 // e_shstrndx = Section # of '.shstrtab'
Rafael Espindola5960cee2015-05-22 23:58:30 +0000369 assert(StringTableIndex < ELF::SHN_LORESERVE);
Jim Grosbach36e60e92015-06-04 22:24:41 +0000370 write16(StringTableIndex);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000371}
372
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000373uint64_t ELFObjectWriter::SymbolValue(const MCSymbol &Sym,
Jan Sjödin30a52de2011-02-28 21:45:04 +0000374 const MCAsmLayout &Layout) {
Rafael Espindola4d37b2a2015-05-29 21:45:01 +0000375 if (Sym.isCommon() && Sym.isExternal())
Rafael Espindola14672502015-05-29 17:48:04 +0000376 return Sym.getCommonAlignment();
Rafael Espindola9735f4f2010-09-27 21:23:02 +0000377
Rafael Espindolafee224f2014-04-30 21:51:13 +0000378 uint64_t Res;
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000379 if (!Layout.getSymbolOffset(Sym, Res))
Rafael Espindola553e5eb2014-04-30 16:59:35 +0000380 return 0;
381
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000382 if (Layout.getAssembler().isThumbFunc(&Sym))
Rafael Espindola66f96fe2014-03-21 22:00:29 +0000383 Res |= 1;
Rafael Espindola9735f4f2010-09-27 21:23:02 +0000384
Rafael Espindola66f96fe2014-03-21 22:00:29 +0000385 return Res;
Rafael Espindola9735f4f2010-09-27 21:23:02 +0000386}
387
Jim Grosbach56ed0bb2015-06-04 23:25:54 +0000388void ELFObjectWriter::executePostLayoutBinding(MCAssembler &Asm,
Rafael Espindola93e3cf02010-12-07 00:27:36 +0000389 const MCAsmLayout &Layout) {
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000390 // The presence of symbol versions causes undefined symbols and
391 // versions declared with @@@ to be renamed.
Rafael Espindola47b4d6b2018-03-09 18:42:25 +0000392 for (const std::pair<StringRef, const MCSymbol *> &P : Asm.Symvers) {
393 StringRef AliasName = P.first;
394 const auto &Symbol = cast<MCSymbolELF>(*P.second);
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000395 size_t Pos = AliasName.find('@');
Rafael Espindola47b4d6b2018-03-09 18:42:25 +0000396 assert(Pos != StringRef::npos);
397
398 StringRef Prefix = AliasName.substr(0, Pos);
399 StringRef Rest = AliasName.substr(Pos);
400 StringRef Tail = Rest;
401 if (Rest.startswith("@@@"))
402 Tail = Rest.substr(Symbol.isUndefined() ? 2 : 1);
403
404 auto *Alias =
405 cast<MCSymbolELF>(Asm.getContext().getOrCreateSymbol(Prefix + Tail));
406 Asm.registerSymbol(*Alias);
407 const MCExpr *Value = MCSymbolRefExpr::create(&Symbol, Asm.getContext());
408 Alias->setVariableValue(Value);
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000409
Rafael Espindola6cd76e62010-10-28 18:33:03 +0000410 // Aliases defined with .symvar copy the binding from the symbol they alias.
411 // This is the first place we are able to copy this information.
Rafael Espindola47b4d6b2018-03-09 18:42:25 +0000412 Alias->setExternal(Symbol.isExternal());
413 Alias->setBinding(Symbol.getBinding());
Rafael Espindola6cd76e62010-10-28 18:33:03 +0000414
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000415 if (!Symbol.isUndefined() && !Rest.startswith("@@@"))
416 continue;
417
Rafael Espindola26496e62010-10-27 17:56:18 +0000418 // FIXME: produce a better error message.
419 if (Symbol.isUndefined() && Rest.startswith("@@") &&
420 !Rest.startswith("@@@"))
421 report_fatal_error("A @@ version cannot be undefined");
422
Vlad Tsyrklevich201a1082018-04-27 20:32:34 +0000423 if (Renames.count(&Symbol) && Renames[&Symbol] != Alias)
424 report_fatal_error(llvm::Twine("Multiple symbol versions defined for ") +
425 Symbol.getName());
426
Rafael Espindola47b4d6b2018-03-09 18:42:25 +0000427 Renames.insert(std::make_pair(&Symbol, Alias));
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000428 }
429}
430
Roman Divacky5a1c5492014-01-07 20:17:03 +0000431static uint8_t mergeTypeForSet(uint8_t origType, uint8_t newType) {
432 uint8_t Type = newType;
433
434 // Propagation rules:
435 // IFUNC > FUNC > OBJECT > NOTYPE
436 // TLS_OBJECT > OBJECT > NOTYPE
437 //
438 // dont let the new type degrade the old type
439 switch (origType) {
440 default:
441 break;
442 case ELF::STT_GNU_IFUNC:
443 if (Type == ELF::STT_FUNC || Type == ELF::STT_OBJECT ||
444 Type == ELF::STT_NOTYPE || Type == ELF::STT_TLS)
445 Type = ELF::STT_GNU_IFUNC;
446 break;
447 case ELF::STT_FUNC:
448 if (Type == ELF::STT_OBJECT || Type == ELF::STT_NOTYPE ||
449 Type == ELF::STT_TLS)
450 Type = ELF::STT_FUNC;
451 break;
452 case ELF::STT_OBJECT:
453 if (Type == ELF::STT_NOTYPE)
454 Type = ELF::STT_OBJECT;
455 break;
456 case ELF::STT_TLS:
457 if (Type == ELF::STT_OBJECT || Type == ELF::STT_NOTYPE ||
458 Type == ELF::STT_GNU_IFUNC || Type == ELF::STT_FUNC)
459 Type = ELF::STT_TLS;
460 break;
461 }
462
463 return Type;
464}
465
Rafael Espindolae48421f2015-05-28 20:25:29 +0000466void ELFObjectWriter::writeSymbol(SymbolTableWriter &Writer,
467 uint32_t StringIndex, ELFSymbolData &MSD,
Daniel Dunbarfe0c28f2010-11-13 07:33:40 +0000468 const MCAsmLayout &Layout) {
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000469 const auto &Symbol = cast<MCSymbolELF>(*MSD.Symbol);
Rafael Espindolaa8695762015-06-02 00:25:12 +0000470 const MCSymbolELF *Base =
471 cast_or_null<MCSymbolELF>(Layout.getBaseSymbol(Symbol));
Rafael Espindola85a84912014-03-26 00:16:43 +0000472
473 // This has to be in sync with when computeSymbolTable uses SHN_ABS or
474 // SHN_COMMON.
Rafael Espindola4d37b2a2015-05-29 21:45:01 +0000475 bool IsReserved = !Base || Symbol.isCommon();
Rafael Espindola3fe87a12010-10-31 00:16:26 +0000476
Jack Carter2f8d9d92013-02-19 21:57:35 +0000477 // Binding and Type share the same byte as upper and lower nibbles
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000478 uint8_t Binding = Symbol.getBinding();
479 uint8_t Type = Symbol.getType();
Rafael Espindolaa6e3a592014-03-23 03:33:20 +0000480 if (Base) {
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000481 Type = mergeTypeForSet(Type, Base->getType());
Rafael Espindolaa6e3a592014-03-23 03:33:20 +0000482 }
Rafael Espindolaf8794ff2015-06-04 00:47:43 +0000483 uint8_t Info = (Binding << 4) | Type;
Jack Carter2f8d9d92013-02-19 21:57:35 +0000484
Joerg Sonnenbergerfc184732013-10-29 01:06:17 +0000485 // Other and Visibility share the same byte with Visibility using the lower
Jack Carter2f8d9d92013-02-19 21:57:35 +0000486 // 2 bits
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000487 uint8_t Visibility = Symbol.getVisibility();
Rafael Espindola8c006ee2015-06-04 05:59:23 +0000488 uint8_t Other = Symbol.getOther() | Visibility;
Rafael Espindola5f2d6a52010-10-06 21:02:29 +0000489
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000490 uint64_t Value = SymbolValue(*MSD.Symbol, Layout);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000491 uint64_t Size = 0;
Matt Fleming6c1ad482010-08-16 18:57:57 +0000492
Rafael Espindola7c23cba2015-05-29 17:24:52 +0000493 const MCExpr *ESize = MSD.Symbol->getSize();
Rafael Espindolaa041ef12014-03-27 00:28:24 +0000494 if (!ESize && Base)
Rafael Espindola7c23cba2015-05-29 17:24:52 +0000495 ESize = Base->getSize();
Rafael Espindolaf0591c12010-09-21 00:24:38 +0000496
Rafael Espindola73c0ae72010-12-22 16:03:00 +0000497 if (ESize) {
498 int64_t Res;
Rafael Espindola94a88d72015-04-06 15:27:57 +0000499 if (!ESize->evaluateKnownAbsolute(Res, Layout))
Rafael Espindola73c0ae72010-12-22 16:03:00 +0000500 report_fatal_error("Size expression must be absolute.");
501 Size = Res;
Matt Fleming6c1ad482010-08-16 18:57:57 +0000502 }
503
504 // Write out the symbol table entry
Rafael Espindolae48421f2015-05-28 20:25:29 +0000505 Writer.writeSymbol(StringIndex, Info, Value, Size, Other, MSD.SectionIndex,
506 IsReserved);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000507}
508
Rafael Espindola5904e122014-03-29 06:26:49 +0000509// It is always valid to create a relocation with a symbol. It is preferable
510// to use a relocation with a section if that is possible. Using the section
511// allows us to omit some local symbols from the symbol table.
Rafael Espindolab60c8292014-04-29 12:46:50 +0000512bool ELFObjectWriter::shouldRelocateWithSymbol(const MCAssembler &Asm,
513 const MCSymbolRefExpr *RefA,
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000514 const MCSymbol *S, uint64_t C,
Rafael Espindola5904e122014-03-29 06:26:49 +0000515 unsigned Type) const {
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000516 const auto *Sym = cast_or_null<MCSymbolELF>(S);
Rafael Espindola5904e122014-03-29 06:26:49 +0000517 // A PCRel relocation to an absolute value has no symbol (or section). We
518 // represent that with a relocation to a null section.
519 if (!RefA)
520 return false;
Rafael Espindola240028d2010-11-14 23:53:26 +0000521
Rafael Espindola5904e122014-03-29 06:26:49 +0000522 MCSymbolRefExpr::VariantKind Kind = RefA->getKind();
523 switch (Kind) {
524 default:
525 break;
526 // The .odp creation emits a relocation against the symbol ".TOC." which
527 // create a R_PPC64_TOC relocation. However the relocation symbol name
528 // in final object creation should be NULL, since the symbol does not
529 // really exist, it is just the reference to TOC base for the current
530 // object file. Since the symbol is undefined, returning false results
531 // in a relocation with a null section which is the desired result.
532 case MCSymbolRefExpr::VK_PPC_TOCBASE:
533 return false;
534
535 // These VariantKind cause the relocation to refer to something other than
536 // the symbol itself, like a linker generated table. Since the address of
537 // symbol is not relevant, we cannot replace the symbol with the
538 // section and patch the difference in the addend.
539 case MCSymbolRefExpr::VK_GOT:
540 case MCSymbolRefExpr::VK_PLT:
541 case MCSymbolRefExpr::VK_GOTPCREL:
Rafael Espindola5904e122014-03-29 06:26:49 +0000542 case MCSymbolRefExpr::VK_PPC_GOT_LO:
543 case MCSymbolRefExpr::VK_PPC_GOT_HI:
544 case MCSymbolRefExpr::VK_PPC_GOT_HA:
545 return true;
Rafael Espindola240028d2010-11-14 23:53:26 +0000546 }
Rafael Espindola240028d2010-11-14 23:53:26 +0000547
Rafael Espindola5904e122014-03-29 06:26:49 +0000548 // An undefined symbol is not in any section, so the relocation has to point
549 // to the symbol itself.
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000550 assert(Sym && "Expected a symbol");
551 if (Sym->isUndefined())
Rafael Espindola5904e122014-03-29 06:26:49 +0000552 return true;
553
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000554 unsigned Binding = Sym->getBinding();
Rafael Espindola5904e122014-03-29 06:26:49 +0000555 switch(Binding) {
556 default:
557 llvm_unreachable("Invalid Binding");
558 case ELF::STB_LOCAL:
559 break;
560 case ELF::STB_WEAK:
561 // If the symbol is weak, it might be overridden by a symbol in another
562 // file. The relocation has to point to the symbol so that the linker
563 // can update it.
564 return true;
565 case ELF::STB_GLOBAL:
566 // Global ELF symbols can be preempted by the dynamic linker. The relocation
567 // has to point to the symbol for a reason analogous to the STB_WEAK case.
568 return true;
Rafael Espindola8c3039b2010-11-15 16:33:49 +0000569 }
Rafael Espindola75d65b92010-09-25 05:42:19 +0000570
Rafael Espindola5904e122014-03-29 06:26:49 +0000571 // If a relocation points to a mergeable section, we have to be careful.
572 // If the offset is zero, a relocation with the section will encode the
573 // same information. With a non-zero offset, the situation is different.
574 // For example, a relocation can point 42 bytes past the end of a string.
575 // If we change such a relocation to use the section, the linker would think
576 // that it pointed to another string and subtracting 42 at runtime will
577 // produce the wrong value.
Saleem Abdulrasool9b106ea2016-11-22 04:32:54 +0000578 if (Sym->isInSection()) {
579 auto &Sec = cast<MCSectionELF>(Sym->getSection());
580 unsigned Flags = Sec.getFlags();
581 if (Flags & ELF::SHF_MERGE) {
582 if (C != 0)
583 return true;
Rafael Espindolab1b49782014-04-02 12:15:20 +0000584
Saleem Abdulrasool9b106ea2016-11-22 04:32:54 +0000585 // It looks like gold has a bug (http://sourceware.org/PR16794) and can
586 // only handle section relocations to mergeable sections if using RELA.
587 if (!hasRelocationAddend())
588 return true;
589 }
590
591 // Most TLS relocations use a got, so they need the symbol. Even those that
592 // are just an offset (@tpoff), require a symbol in gold versions before
593 // 5efeedf61e4fe720fd3e9a08e6c91c10abb66d42 (2014-09-26) which fixed
594 // http://sourceware.org/PR16773.
595 if (Flags & ELF::SHF_TLS)
Rafael Espindolab1b49782014-04-02 12:15:20 +0000596 return true;
Rafael Espindola9f75d5d2010-11-24 21:57:39 +0000597 }
598
Rafael Espindola9ef84412014-04-11 19:18:01 +0000599 // If the symbol is a thumb function the final relocation must set the lowest
600 // bit. With a symbol that is done by just having the symbol have that bit
601 // set, so we would lose the bit if we relocated with the section.
602 // FIXME: We could use the section but add the bit to the relocation value.
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000603 if (Asm.isThumbFunc(Sym))
Rafael Espindola9ef84412014-04-11 19:18:01 +0000604 return true;
605
Rafael Espindolaece40ca2015-05-29 18:26:09 +0000606 if (TargetObjectWriter->needsRelocateWithSymbol(*Sym, Type))
Rafael Espindola5904e122014-03-29 06:26:49 +0000607 return true;
608 return false;
Rafael Espindola75d65b92010-09-25 05:42:19 +0000609}
610
Rafael Espindoladb8a5862015-04-17 20:05:17 +0000611// True if the assembler knows nothing about the final value of the symbol.
612// This doesn't cover the comdat issues, since in those cases the assembler
613// can at least know that all symbols in the section will move together.
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000614static bool isWeak(const MCSymbolELF &Sym) {
615 if (Sym.getType() == ELF::STT_GNU_IFUNC)
Rafael Espindolaa635d832015-04-17 08:46:11 +0000616 return true;
617
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000618 switch (Sym.getBinding()) {
Rafael Espindolaa635d832015-04-17 08:46:11 +0000619 default:
620 llvm_unreachable("Unknown binding");
621 case ELF::STB_LOCAL:
622 return false;
623 case ELF::STB_GLOBAL:
Rafael Espindoladb8a5862015-04-17 20:05:17 +0000624 return false;
Rafael Espindolaa635d832015-04-17 08:46:11 +0000625 case ELF::STB_WEAK:
626 case ELF::STB_GNU_UNIQUE:
627 return true;
628 }
Rafael Espindolac9e70682015-03-24 23:48:44 +0000629}
630
Jim Grosbach36e60e92015-06-04 22:24:41 +0000631void ELFObjectWriter::recordRelocation(MCAssembler &Asm,
Jason W Kim495c2bb2010-12-06 21:57:34 +0000632 const MCAsmLayout &Layout,
633 const MCFragment *Fragment,
Rafael Espindola26585542015-01-19 21:11:14 +0000634 const MCFixup &Fixup, MCValue Target,
Rafael Espindolaceecfe5b2017-07-11 23:56:10 +0000635 uint64_t &FixedValue) {
636 MCAsmBackend &Backend = Asm.getBackend();
637 bool IsPCRel = Backend.getFixupKindInfo(Fixup.getKind()).Flags &
638 MCFixupKindInfo::FKF_IsPCRel;
Rafael Espindola7549f872015-05-26 00:36:57 +0000639 const MCSectionELF &FixupSection = cast<MCSectionELF>(*Fragment->getParent());
Rafael Espindola5904e122014-03-29 06:26:49 +0000640 uint64_t C = Target.getConstant();
641 uint64_t FixupOffset = Layout.getFragmentOffset(Fragment) + Fixup.getOffset();
Rafael Espindola00ebfd42016-01-13 22:23:36 +0000642 MCContext &Ctx = Asm.getContext();
Jason W Kim495c2bb2010-12-06 21:57:34 +0000643
Rafael Espindola5904e122014-03-29 06:26:49 +0000644 if (const MCSymbolRefExpr *RefB = Target.getSymB()) {
Rafael Espindola5904e122014-03-29 06:26:49 +0000645 // Let A, B and C being the components of Target and R be the location of
646 // the fixup. If the fixup is not pcrel, we want to compute (A - B + C).
647 // If it is pcrel, we want to compute (A - B + C - R).
Jason W Kim495c2bb2010-12-06 21:57:34 +0000648
Rafael Espindola5904e122014-03-29 06:26:49 +0000649 // In general, ELF has no relocations for -B. It can only represent (A + C)
650 // or (A + C - R). If B = R + K and the relocation is not pcrel, we can
651 // replace B to implement it: (A - R - K + C)
Oliver Stannard9be59af2015-11-17 10:00:43 +0000652 if (IsPCRel) {
Rafael Espindola00ebfd42016-01-13 22:23:36 +0000653 Ctx.reportError(
Rafael Espindola5904e122014-03-29 06:26:49 +0000654 Fixup.getLoc(),
655 "No relocation available to represent this relative expression");
Oliver Stannard9be59af2015-11-17 10:00:43 +0000656 return;
657 }
David Majnemera45a1762014-01-06 07:39:46 +0000658
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000659 const auto &SymB = cast<MCSymbolELF>(RefB->getSymbol());
Jason W Kim495c2bb2010-12-06 21:57:34 +0000660
Oliver Stannard9be59af2015-11-17 10:00:43 +0000661 if (SymB.isUndefined()) {
Rafael Espindola00ebfd42016-01-13 22:23:36 +0000662 Ctx.reportError(Fixup.getLoc(),
663 Twine("symbol '") + SymB.getName() +
664 "' can not be undefined in a subtraction expression");
Oliver Stannard9be59af2015-11-17 10:00:43 +0000665 return;
666 }
Jason W Kim495c2bb2010-12-06 21:57:34 +0000667
Rafael Espindola5904e122014-03-29 06:26:49 +0000668 assert(!SymB.isAbsolute() && "Should have been folded");
669 const MCSection &SecB = SymB.getSection();
Oliver Stannard9be59af2015-11-17 10:00:43 +0000670 if (&SecB != &FixupSection) {
Rafael Espindola00ebfd42016-01-13 22:23:36 +0000671 Ctx.reportError(Fixup.getLoc(),
672 "Cannot represent a difference across sections");
Oliver Stannard9be59af2015-11-17 10:00:43 +0000673 return;
674 }
Jason W Kim495c2bb2010-12-06 21:57:34 +0000675
Duncan P. N. Exon Smith2a404832015-05-19 23:53:20 +0000676 uint64_t SymBOffset = Layout.getSymbolOffset(SymB);
Rafael Espindola5904e122014-03-29 06:26:49 +0000677 uint64_t K = SymBOffset - FixupOffset;
678 IsPCRel = true;
679 C -= K;
Jason W Kim495c2bb2010-12-06 21:57:34 +0000680 }
681
Rafael Espindola5904e122014-03-29 06:26:49 +0000682 // We either rejected the fixup or folded B into C at this point.
683 const MCSymbolRefExpr *RefA = Target.getSymA();
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000684 const auto *SymA = RefA ? cast<MCSymbolELF>(&RefA->getSymbol()) : nullptr;
Rafael Espindola5904e122014-03-29 06:26:49 +0000685
Rafael Espindola212fdde2015-06-03 21:52:06 +0000686 bool ViaWeakRef = false;
687 if (SymA && SymA->isVariable()) {
688 const MCExpr *Expr = SymA->getVariableValue();
689 if (const auto *Inner = dyn_cast<MCSymbolRefExpr>(Expr)) {
690 if (Inner->getKind() == MCSymbolRefExpr::VK_WEAKREF) {
691 SymA = cast<MCSymbolELF>(&Inner->getSymbol());
692 ViaWeakRef = true;
693 }
694 }
695 }
696
Rafael Espindola8340f942016-01-13 22:56:57 +0000697 unsigned Type = getRelocType(Ctx, Target, Fixup, IsPCRel);
Daniel Sandersa463d312016-05-06 13:49:25 +0000698 uint64_t OriginalC = C;
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000699 bool RelocateWithSymbol = shouldRelocateWithSymbol(Asm, RefA, SymA, C, Type);
Rafael Espindola5904e122014-03-29 06:26:49 +0000700 if (!RelocateWithSymbol && SymA && !SymA->isUndefined())
Duncan P. N. Exon Smith2a404832015-05-19 23:53:20 +0000701 C += Layout.getSymbolOffset(*SymA);
Rafael Espindola5904e122014-03-29 06:26:49 +0000702
703 uint64_t Addend = 0;
704 if (hasRelocationAddend()) {
705 Addend = C;
706 C = 0;
707 }
708
709 FixedValue = C;
710
Rafael Espindola5904e122014-03-29 06:26:49 +0000711 if (!RelocateWithSymbol) {
712 const MCSection *SecA =
713 (SymA && !SymA->isUndefined()) ? &SymA->getSection() : nullptr;
Rafael Espindolab6613022014-10-17 01:48:58 +0000714 auto *ELFSec = cast_or_null<MCSectionELF>(SecA);
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000715 const auto *SectionSymbol =
716 ELFSec ? cast<MCSymbolELF>(ELFSec->getBeginSymbol()) : nullptr;
Rafael Espindolaa401eee2015-06-04 15:33:30 +0000717 if (SectionSymbol)
718 SectionSymbol->setUsedInReloc();
Daniel Sandersa463d312016-05-06 13:49:25 +0000719 ELFRelocationEntry Rec(FixupOffset, SectionSymbol, Type, Addend, SymA,
720 OriginalC);
Rafael Espindola34948e52015-04-30 00:45:46 +0000721 Relocations[&FixupSection].push_back(Rec);
Rafael Espindola5904e122014-03-29 06:26:49 +0000722 return;
723 }
Roman Divackydfbecd12011-08-04 19:08:19 +0000724
Daniel Sandersa463d312016-05-06 13:49:25 +0000725 const auto *RenamedSymA = SymA;
Rafael Espindola5904e122014-03-29 06:26:49 +0000726 if (SymA) {
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000727 if (const MCSymbolELF *R = Renames.lookup(SymA))
Daniel Sandersa463d312016-05-06 13:49:25 +0000728 RenamedSymA = R;
Rafael Espindolad7facaf2011-08-04 13:05:26 +0000729
Rafael Espindola212fdde2015-06-03 21:52:06 +0000730 if (ViaWeakRef)
Daniel Sandersa463d312016-05-06 13:49:25 +0000731 RenamedSymA->setIsWeakrefUsedInReloc();
Rafael Espindola5904e122014-03-29 06:26:49 +0000732 else
Daniel Sandersa463d312016-05-06 13:49:25 +0000733 RenamedSymA->setUsedInReloc();
Rafael Espindola5904e122014-03-29 06:26:49 +0000734 }
Daniel Sandersa463d312016-05-06 13:49:25 +0000735 ELFRelocationEntry Rec(FixupOffset, RenamedSymA, Type, Addend, SymA,
736 OriginalC);
Rafael Espindola34948e52015-04-30 00:45:46 +0000737 Relocations[&FixupSection].push_back(Rec);
Jason W Kim495c2bb2010-12-06 21:57:34 +0000738}
739
Rafael Espindola3b5ee552014-04-28 13:39:57 +0000740bool ELFObjectWriter::isInSymtab(const MCAsmLayout &Layout,
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000741 const MCSymbolELF &Symbol, bool Used,
Rafael Espindola3b5ee552014-04-28 13:39:57 +0000742 bool Renamed) {
Rafael Espindola7fadc0e2014-03-20 02:12:01 +0000743 if (Symbol.isVariable()) {
744 const MCExpr *Expr = Symbol.getVariableValue();
745 if (const MCSymbolRefExpr *Ref = dyn_cast<MCSymbolRefExpr>(Expr)) {
746 if (Ref->getKind() == MCSymbolRefExpr::VK_WEAKREF)
747 return false;
748 }
749 }
Rafael Espindola16145972010-11-01 14:28:48 +0000750
Rafael Espindolaee8d1512010-10-19 19:31:37 +0000751 if (Used)
752 return true;
753
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000754 if (Renamed)
755 return false;
756
Rafael Espindolaef1e8632015-06-03 21:23:21 +0000757 if (Symbol.isVariable() && Symbol.isUndefined()) {
758 // FIXME: this is here just to diagnose the case of a var = commmon_sym.
759 Layout.getBaseSymbol(Symbol);
760 return false;
Rafael Espindola3b5ee552014-04-28 13:39:57 +0000761 }
Rafael Espindola9e18e962011-02-23 20:22:07 +0000762
Rafael Espindolaef1e8632015-06-03 21:23:21 +0000763 if (Symbol.isUndefined() && !Symbol.isBindingSet())
Rafael Espindolaa5efd6a2010-10-27 14:44:52 +0000764 return false;
765
Rafael Espindolaee8d1512010-10-19 19:31:37 +0000766 if (Symbol.isTemporary())
Rafael Espindolab1d07892010-10-05 18:01:23 +0000767 return false;
768
Rafael Espindolaa401eee2015-06-04 15:33:30 +0000769 if (Symbol.getType() == ELF::STT_SECTION)
770 return false;
771
Rafael Espindolab1d07892010-10-05 18:01:23 +0000772 return true;
773}
774
Rafael Espindolaea1b3942015-04-07 19:00:17 +0000775void ELFObjectWriter::computeSymbolTable(
776 MCAssembler &Asm, const MCAsmLayout &Layout,
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000777 const SectionIndexMapTy &SectionIndexMap, const RevGroupMapTy &RevGroupMap,
778 SectionOffsetsTy &SectionOffsets) {
Rafael Espindola5960cee2015-05-22 23:58:30 +0000779 MCContext &Ctx = Asm.getContext();
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000780 SymbolTableWriter Writer(*this, is64Bit());
781
Rafael Espindola5960cee2015-05-22 23:58:30 +0000782 // Symbol table
783 unsigned EntrySize = is64Bit() ? ELF::SYMENTRY_SIZE64 : ELF::SYMENTRY_SIZE32;
784 MCSectionELF *SymtabSection =
785 Ctx.getELFSection(".symtab", ELF::SHT_SYMTAB, 0, EntrySize, "");
786 SymtabSection->setAlignment(is64Bit() ? 8 : 4);
787 SymbolTableIndex = addToSectionTable(SymtabSection);
788
Rafael Espindolab20fbb82015-06-05 18:21:00 +0000789 align(SymtabSection->getAlignment());
David Majnemerabdb2d2a2015-09-01 16:19:03 +0000790 uint64_t SecStart = getStream().tell();
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000791
792 // The first entry is the undefined symbol entry.
793 Writer.writeSymbol(0, 0, 0, 0, 0, 0, false);
794
Rafael Espindolacfbd35c2015-05-28 20:11:34 +0000795 std::vector<ELFSymbolData> LocalSymbolData;
796 std::vector<ELFSymbolData> ExternalSymbolData;
Rafael Espindolacfbd35c2015-05-28 20:11:34 +0000797
Rafael Espindolabee6e9f2010-10-14 16:34:44 +0000798 // Add the data for the symbols.
Rafael Espindola5960cee2015-05-22 23:58:30 +0000799 bool HasLargeSectionIndex = false;
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000800 for (const MCSymbol &S : Asm.symbols()) {
801 const auto &Symbol = cast<MCSymbolELF>(S);
Rafael Espindolaada43f62015-06-03 21:30:10 +0000802 bool Used = Symbol.isUsedInReloc();
Rafael Espindola212fdde2015-06-03 21:52:06 +0000803 bool WeakrefUsed = Symbol.isWeakrefUsedInReloc();
Rafael Espindola8c52a9b2015-06-03 21:41:59 +0000804 bool isSignature = Symbol.isSignature();
Rafael Espindola7d0ba342010-11-14 04:17:37 +0000805
Duncan P. N. Exon Smith469f9db2015-05-20 04:39:01 +0000806 if (!isInSymtab(Layout, Symbol, Used || WeakrefUsed || isSignature,
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000807 Renames.count(&Symbol)))
Matt Fleming6c1ad482010-08-16 18:57:57 +0000808 continue;
809
Oliver Stannard9be59af2015-11-17 10:00:43 +0000810 if (Symbol.isTemporary() && Symbol.isUndefined()) {
811 Ctx.reportError(SMLoc(), "Undefined temporary symbol");
812 continue;
813 }
Rafael Espindola6dff8142015-06-25 20:10:45 +0000814
Matt Fleming6c1ad482010-08-16 18:57:57 +0000815 ELFSymbolData MSD;
Rafael Espindolaa8695762015-06-02 00:25:12 +0000816 MSD.Symbol = cast<MCSymbolELF>(&Symbol);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000817
Rafael Espindola8c52a9b2015-06-03 21:41:59 +0000818 bool Local = Symbol.getBinding() == ELF::STB_LOCAL;
Rafael Espindola6dff8142015-06-25 20:10:45 +0000819 assert(Local || !Symbol.isTemporary());
820
Rafael Espindolaf4b44302015-05-29 15:07:27 +0000821 if (Symbol.isAbsolute()) {
Rafael Espindola022bb762014-03-24 03:43:21 +0000822 MSD.SectionIndex = ELF::SHN_ABS;
Rafael Espindola14672502015-05-29 17:48:04 +0000823 } else if (Symbol.isCommon()) {
Rafael Espindolabee6e9f2010-10-14 16:34:44 +0000824 assert(!Local);
Rafael Espindolaf0591c12010-09-21 00:24:38 +0000825 MSD.SectionIndex = ELF::SHN_COMMON;
Rafael Espindolaf4b44302015-05-29 15:07:27 +0000826 } else if (Symbol.isUndefined()) {
Rafael Espindola5960cee2015-05-22 23:58:30 +0000827 if (isSignature && !Used) {
Rafael Espindola89feff32015-04-28 22:59:58 +0000828 MSD.SectionIndex = RevGroupMap.lookup(&Symbol);
Rafael Espindola5960cee2015-05-22 23:58:30 +0000829 if (MSD.SectionIndex >= ELF::SHN_LORESERVE)
830 HasLargeSectionIndex = true;
831 } else {
Rafael Espindola7d0ba342010-11-14 04:17:37 +0000832 MSD.SectionIndex = ELF::SHN_UNDEF;
Rafael Espindola5960cee2015-05-22 23:58:30 +0000833 }
Matt Fleming6c1ad482010-08-16 18:57:57 +0000834 } else {
Rafael Espindolad6340032010-11-10 21:51:05 +0000835 const MCSectionELF &Section =
Rafael Espindolaf4b44302015-05-29 15:07:27 +0000836 static_cast<const MCSectionELF &>(Symbol.getSection());
Rafael Espindolad6340032010-11-10 21:51:05 +0000837 MSD.SectionIndex = SectionIndexMap.lookup(&Section);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000838 assert(MSD.SectionIndex && "Invalid section index!");
Rafael Espindola5960cee2015-05-22 23:58:30 +0000839 if (MSD.SectionIndex >= ELF::SHN_LORESERVE)
840 HasLargeSectionIndex = true;
Rafael Espindolafbcf0db2010-10-15 15:39:06 +0000841 }
842
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000843 StringRef Name = Symbol.getName();
Rafael Espindolab6613022014-10-17 01:48:58 +0000844
845 // Sections have their own string table
Rafael Espindolafc063e82015-10-22 18:32:06 +0000846 if (Symbol.getType() != ELF::STT_SECTION) {
847 MSD.Name = Name;
848 StrTabBuilder.add(Name);
849 }
Rafael Espindolaeb0c2c12010-10-27 15:18:17 +0000850
Rafael Espindola10d23872015-05-29 14:20:40 +0000851 if (Local)
Rafael Espindolaa5efd6a2010-10-27 14:44:52 +0000852 LocalSymbolData.push_back(MSD);
853 else
854 ExternalSymbolData.push_back(MSD);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000855 }
856
Rafael Espindola2cd19512015-07-02 16:59:57 +0000857 // This holds the .symtab_shndx section index.
858 unsigned SymtabShndxSectionIndex = 0;
859
Rafael Espindola5960cee2015-05-22 23:58:30 +0000860 if (HasLargeSectionIndex) {
861 MCSectionELF *SymtabShndxSection =
862 Ctx.getELFSection(".symtab_shndxr", ELF::SHT_SYMTAB_SHNDX, 0, 4, "");
863 SymtabShndxSectionIndex = addToSectionTable(SymtabShndxSection);
864 SymtabShndxSection->setAlignment(4);
865 }
866
Rafael Espindola1fd36272015-05-28 19:29:15 +0000867 ArrayRef<std::string> FileNames = Asm.getFileNames();
868 for (const std::string &Name : FileNames)
Rafael Espindola66f3c9c2015-05-28 18:03:20 +0000869 StrTabBuilder.add(Name);
Hans Wennborg83e6e1e2014-04-30 16:25:02 +0000870
Rafael Espindola21956e42015-10-23 21:48:05 +0000871 StrTabBuilder.finalize();
Hans Wennborg83e6e1e2014-04-30 16:25:02 +0000872
Peter Collingbourne5b75fd92017-03-03 21:22:06 +0000873 // File symbols are emitted first and handled separately from normal symbols,
874 // i.e. a non-STT_FILE symbol with the same name may appear.
Rafael Espindola0cbea292015-05-28 20:00:13 +0000875 for (const std::string &Name : FileNames)
876 Writer.writeSymbol(StrTabBuilder.getOffset(Name),
877 ELF::STT_FILE | ELF::STB_LOCAL, 0, 0, ELF::STV_DEFAULT,
878 ELF::SHN_ABS, true);
879
Matt Fleming6c1ad482010-08-16 18:57:57 +0000880 // Symbols are required to be in lexicographic order.
881 array_pod_sort(LocalSymbolData.begin(), LocalSymbolData.end());
882 array_pod_sort(ExternalSymbolData.begin(), ExternalSymbolData.end());
Matt Fleming6c1ad482010-08-16 18:57:57 +0000883
884 // Set the symbol indices. Local symbols must come before all other
885 // symbols with non-local bindings.
Rafael Espindola1fd36272015-05-28 19:29:15 +0000886 unsigned Index = FileNames.size() + 1;
Rafael Espindola0e3decf2010-11-14 03:12:24 +0000887
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000888 for (ELFSymbolData &MSD : LocalSymbolData) {
Daniel Jasper41de8022015-06-23 11:31:32 +0000889 unsigned StringIndex = MSD.Symbol->getType() == ELF::STT_SECTION
890 ? 0
891 : StrTabBuilder.getOffset(MSD.Name);
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000892 MSD.Symbol->setIndex(Index++);
Rafael Espindolae48421f2015-05-28 20:25:29 +0000893 writeSymbol(Writer, StringIndex, MSD, Layout);
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000894 }
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000895
896 // Write the symbol table entries.
Rafael Espindola0cbea292015-05-28 20:00:13 +0000897 LastLocalSymbolIndex = Index;
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000898
Rafael Espindoladcda9972015-05-28 19:43:20 +0000899 for (ELFSymbolData &MSD : ExternalSymbolData) {
Rafael Espindolae48421f2015-05-28 20:25:29 +0000900 unsigned StringIndex = StrTabBuilder.getOffset(MSD.Name);
Rafael Espindola0cbea292015-05-28 20:00:13 +0000901 MSD.Symbol->setIndex(Index++);
Rafael Espindolae48421f2015-05-28 20:25:29 +0000902 writeSymbol(Writer, StringIndex, MSD, Layout);
Rafael Espindola95fb9b92015-06-02 20:38:46 +0000903 assert(MSD.Symbol->getBinding() != ELF::STB_LOCAL);
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000904 }
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000905
David Majnemerabdb2d2a2015-09-01 16:19:03 +0000906 uint64_t SecEnd = getStream().tell();
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000907 SectionOffsets[SymtabSection] = std::make_pair(SecStart, SecEnd);
908
909 ArrayRef<uint32_t> ShndxIndexes = Writer.getShndxIndexes();
910 if (ShndxIndexes.empty()) {
911 assert(SymtabShndxSectionIndex == 0);
912 return;
913 }
914 assert(SymtabShndxSectionIndex != 0);
915
David Majnemerabdb2d2a2015-09-01 16:19:03 +0000916 SecStart = getStream().tell();
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000917 const MCSectionELF *SymtabShndxSection =
918 SectionTable[SymtabShndxSectionIndex - 1];
919 for (uint32_t Index : ShndxIndexes)
920 write(Index);
David Majnemerabdb2d2a2015-09-01 16:19:03 +0000921 SecEnd = getStream().tell();
Rafael Espindolaaa486e92015-05-28 17:54:01 +0000922 SectionOffsets[SymtabShndxSection] = std::make_pair(SecStart, SecEnd);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000923}
924
Rafael Espindola0709a7b2015-05-21 19:20:38 +0000925MCSectionELF *
Rafael Espindola0a82ad72015-05-21 20:43:13 +0000926ELFObjectWriter::createRelocationSection(MCContext &Ctx,
Rafael Espindolabda19802015-04-30 14:21:49 +0000927 const MCSectionELF &Sec) {
Rafael Espindola34948e52015-04-30 00:45:46 +0000928 if (Relocations[&Sec].empty())
Rafael Espindolabda19802015-04-30 14:21:49 +0000929 return nullptr;
Rafael Espindola1557fd62011-03-20 18:44:20 +0000930
Rafael Espindola34948e52015-04-30 00:45:46 +0000931 const StringRef SectionName = Sec.getSectionName();
Rafael Espindolaead85492015-02-17 20:31:13 +0000932 std::string RelaSectionName = hasRelocationAddend() ? ".rela" : ".rel";
933 RelaSectionName += SectionName;
Benjamin Kramerfd054152010-08-17 17:56:13 +0000934
Rafael Espindolaead85492015-02-17 20:31:13 +0000935 unsigned EntrySize;
936 if (hasRelocationAddend())
937 EntrySize = is64Bit() ? sizeof(ELF::Elf64_Rela) : sizeof(ELF::Elf32_Rela);
938 else
939 EntrySize = is64Bit() ? sizeof(ELF::Elf64_Rel) : sizeof(ELF::Elf32_Rel);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000940
Rafael Espindolaead85492015-02-17 20:31:13 +0000941 unsigned Flags = 0;
Rafael Espindola34948e52015-04-30 00:45:46 +0000942 if (Sec.getFlags() & ELF::SHF_GROUP)
Rafael Espindolaead85492015-02-17 20:31:13 +0000943 Flags = ELF::SHF_GROUP;
Rafael Espindolaead85492015-02-17 20:31:13 +0000944
Rafael Espindola0709a7b2015-05-21 19:20:38 +0000945 MCSectionELF *RelaSection = Ctx.createELFRelSection(
Rafael Espindolaead85492015-02-17 20:31:13 +0000946 RelaSectionName, hasRelocationAddend() ? ELF::SHT_RELA : ELF::SHT_REL,
Rafael Espindola34948e52015-04-30 00:45:46 +0000947 Flags, EntrySize, Sec.getGroup(), &Sec);
Rafael Espindola0a82ad72015-05-21 20:43:13 +0000948 RelaSection->setAlignment(is64Bit() ? 8 : 4);
Rafael Espindolabda19802015-04-30 14:21:49 +0000949 return RelaSection;
Rafael Espindola1557fd62011-03-20 18:44:20 +0000950}
Matt Fleming6c1ad482010-08-16 18:57:57 +0000951
George Rimarc91e38c2016-05-27 12:27:32 +0000952// Include the debug info compression header.
953bool ELFObjectWriter::maybeWriteCompression(
954 uint64_t Size, SmallVectorImpl<char> &CompressedContents, bool ZLibStyle,
955 unsigned Alignment) {
956 if (ZLibStyle) {
957 uint64_t HdrSize =
958 is64Bit() ? sizeof(ELF::Elf32_Chdr) : sizeof(ELF::Elf64_Chdr);
959 if (Size <= HdrSize + CompressedContents.size())
960 return false;
961 // Platform specific header is followed by compressed data.
962 if (is64Bit()) {
963 // Write Elf64_Chdr header.
964 write(static_cast<ELF::Elf64_Word>(ELF::ELFCOMPRESS_ZLIB));
965 write(static_cast<ELF::Elf64_Word>(0)); // ch_reserved field.
966 write(static_cast<ELF::Elf64_Xword>(Size));
967 write(static_cast<ELF::Elf64_Xword>(Alignment));
968 } else {
969 // Write Elf32_Chdr header otherwise.
970 write(static_cast<ELF::Elf32_Word>(ELF::ELFCOMPRESS_ZLIB));
971 write(static_cast<ELF::Elf32_Word>(Size));
972 write(static_cast<ELF::Elf32_Word>(Alignment));
973 }
974 return true;
975 }
976
977 // "ZLIB" followed by 8 bytes representing the uncompressed size of the section,
978 // useful for consumers to preallocate a buffer to decompress into.
Richard Smithb910e562016-05-25 00:14:12 +0000979 const StringRef Magic = "ZLIB";
980 if (Size <= Magic.size() + sizeof(Size) + CompressedContents.size())
981 return false;
George Rimarc91e38c2016-05-27 12:27:32 +0000982 write(ArrayRef<char>(Magic.begin(), Magic.size()));
983 writeBE64(Size);
Richard Smithb910e562016-05-25 00:14:12 +0000984 return true;
985}
986
Rafael Espindolae15b1b72015-05-27 13:30:50 +0000987void ELFObjectWriter::writeSectionData(const MCAssembler &Asm, MCSection &Sec,
Rafael Espindola868b3f42015-04-30 21:51:58 +0000988 const MCAsmLayout &Layout) {
Rafael Espindolae15b1b72015-05-27 13:30:50 +0000989 MCSectionELF &Section = static_cast<MCSectionELF &>(Sec);
Rafael Espindola868b3f42015-04-30 21:51:58 +0000990 StringRef SectionName = Section.getSectionName();
David Blaikie8019bf82014-04-10 21:53:53 +0000991
Saleem Abdulrasool1f62f572017-06-09 00:40:19 +0000992 auto &MC = Asm.getContext();
993 const auto &MAI = MC.getAsmInfo();
994
Rafael Espindola868b3f42015-04-30 21:51:58 +0000995 // Compressing debug_frame requires handling alignment fragments which is
996 // more work (possibly generalizing MCAssembler.cpp:writeFragment to allow
997 // for writing to arbitrary buffers) for little benefit.
George Rimarc91e38c2016-05-27 12:27:32 +0000998 bool CompressionEnabled =
Saleem Abdulrasool1f62f572017-06-09 00:40:19 +0000999 MAI->compressDebugSections() != DebugCompressionType::None;
George Rimarc91e38c2016-05-27 12:27:32 +00001000 if (!CompressionEnabled || !SectionName.startswith(".debug_") ||
1001 SectionName == ".debug_frame") {
Rafael Espindola64acc7f2015-05-26 02:17:21 +00001002 Asm.writeSectionData(&Section, Layout);
Rafael Espindola868b3f42015-04-30 21:51:58 +00001003 return;
1004 }
1005
Saleem Abdulrasool1f62f572017-06-09 00:40:19 +00001006 assert((MAI->compressDebugSections() == DebugCompressionType::Z ||
1007 MAI->compressDebugSections() == DebugCompressionType::GNU) &&
1008 "expected zlib or zlib-gnu style compression");
1009
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001010 SmallVector<char, 128> UncompressedData;
1011 raw_svector_ostream VecOS(UncompressedData);
1012 raw_pwrite_stream &OldStream = getStream();
1013 setStream(VecOS);
1014 Asm.writeSectionData(&Section, Layout);
1015 setStream(OldStream);
David Blaikie8019bf82014-04-10 21:53:53 +00001016
Rafael Espindola868b3f42015-04-30 21:51:58 +00001017 SmallVector<char, 128> CompressedContents;
George Rimar167ca4a2017-01-17 15:45:07 +00001018 if (Error E = zlib::compress(
1019 StringRef(UncompressedData.data(), UncompressedData.size()),
1020 CompressedContents)) {
1021 consumeError(std::move(E));
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001022 getStream() << UncompressedData;
David Blaikie8019bf82014-04-10 21:53:53 +00001023 return;
Rafael Espindola868b3f42015-04-30 21:51:58 +00001024 }
David Blaikie8019bf82014-04-10 21:53:53 +00001025
Saleem Abdulrasool1f62f572017-06-09 00:40:19 +00001026 bool ZlibStyle = MAI->compressDebugSections() == DebugCompressionType::Z;
George Rimarc91e38c2016-05-27 12:27:32 +00001027 if (!maybeWriteCompression(UncompressedData.size(), CompressedContents,
1028 ZlibStyle, Sec.getAlignment())) {
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001029 getStream() << UncompressedData;
Rafael Espindola868b3f42015-04-30 21:51:58 +00001030 return;
1031 }
George Rimarc91e38c2016-05-27 12:27:32 +00001032
1033 if (ZlibStyle)
1034 // Set the compressed flag. That is zlib style.
1035 Section.setFlags(Section.getFlags() | ELF::SHF_COMPRESSED);
1036 else
1037 // Add "z" prefix to section name. This is zlib-gnu style.
Saleem Abdulrasool1f62f572017-06-09 00:40:19 +00001038 MC.renameELFSection(&Section, (".z" + SectionName.drop_front(1)).str());
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001039 getStream() << CompressedContents;
David Blaikie8019bf82014-04-10 21:53:53 +00001040}
1041
Daniel Dunbarfe0c28f2010-11-13 07:33:40 +00001042void ELFObjectWriter::WriteSecHdrEntry(uint32_t Name, uint32_t Type,
1043 uint64_t Flags, uint64_t Address,
1044 uint64_t Offset, uint64_t Size,
1045 uint32_t Link, uint32_t Info,
1046 uint64_t Alignment,
1047 uint64_t EntrySize) {
Jim Grosbach36e60e92015-06-04 22:24:41 +00001048 write32(Name); // sh_name: index into string table
1049 write32(Type); // sh_type
Matt Fleming6c1ad482010-08-16 18:57:57 +00001050 WriteWord(Flags); // sh_flags
1051 WriteWord(Address); // sh_addr
1052 WriteWord(Offset); // sh_offset
1053 WriteWord(Size); // sh_size
Jim Grosbach36e60e92015-06-04 22:24:41 +00001054 write32(Link); // sh_link
1055 write32(Info); // sh_info
Matt Fleming6c1ad482010-08-16 18:57:57 +00001056 WriteWord(Alignment); // sh_addralign
1057 WriteWord(EntrySize); // sh_entsize
1058}
1059
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001060void ELFObjectWriter::writeRelocations(const MCAssembler &Asm,
Rafael Espindola34948e52015-04-30 00:45:46 +00001061 const MCSectionELF &Sec) {
1062 std::vector<ELFRelocationEntry> &Relocs = Relocations[&Sec];
Akira Hatanaka64ad2cf2012-03-23 23:06:45 +00001063
Rafael Espindolaf44db242015-12-17 16:22:06 +00001064 // We record relocations by pushing to the end of a vector. Reverse the vector
1065 // to get the relocations in the order they were created.
1066 // In most cases that is not important, but it can be for special sections
1067 // (.eh_frame) or specific relocations (TLS optimizations on SystemZ).
1068 std::reverse(Relocs.begin(), Relocs.end());
Rafael Espindolad0e16522015-12-17 15:08:24 +00001069
Rafael Espindolaf44db242015-12-17 16:22:06 +00001070 // Sort the relocation entries. MIPS needs this.
Petar Jovanovic0380d0b2015-04-14 13:23:34 +00001071 TargetObjectWriter->sortRelocs(Asm, Relocs);
Matt Fleming6c1ad482010-08-16 18:57:57 +00001072
1073 for (unsigned i = 0, e = Relocs.size(); i != e; ++i) {
Rafael Espindola5904e122014-03-29 06:26:49 +00001074 const ELFRelocationEntry &Entry = Relocs[e - i - 1];
Rafael Espindola5e9ed902015-05-28 20:53:09 +00001075 unsigned Index = Entry.Symbol ? Entry.Symbol->getIndex() : 0;
Rafael Espindola5904e122014-03-29 06:26:49 +00001076
Rafael Espindolafdaae0d2010-12-18 03:27:34 +00001077 if (is64Bit()) {
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001078 write(Entry.Offset);
Simon Atanasyan9f676a72017-09-21 14:04:47 +00001079 if (TargetObjectWriter->getEMachine() == ELF::EM_MIPS) {
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001080 write(uint32_t(Index));
Benjamin Kramer6c3c3492010-09-09 18:01:29 +00001081
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001082 write(TargetObjectWriter->getRSsym(Entry.Type));
1083 write(TargetObjectWriter->getRType3(Entry.Type));
1084 write(TargetObjectWriter->getRType2(Entry.Type));
1085 write(TargetObjectWriter->getRType(Entry.Type));
Rafael Espindola5904e122014-03-29 06:26:49 +00001086 } else {
Jack Carter8ad0c272012-06-27 22:28:30 +00001087 struct ELF::Elf64_Rela ERE64;
Rafael Espindola5904e122014-03-29 06:26:49 +00001088 ERE64.setSymbolAndType(Index, Entry.Type);
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001089 write(ERE64.r_info);
Jack Carter8ad0c272012-06-27 22:28:30 +00001090 }
Rafael Espindolafdaae0d2010-12-18 03:27:34 +00001091 if (hasRelocationAddend())
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001092 write(Entry.Addend);
Benjamin Kramer6c3c3492010-09-09 18:01:29 +00001093 } else {
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001094 write(uint32_t(Entry.Offset));
Benjamin Kramer6c3c3492010-09-09 18:01:29 +00001095
Rafael Espindolabce26a12010-10-05 15:11:03 +00001096 struct ELF::Elf32_Rela ERE32;
Rafael Espindola5904e122014-03-29 06:26:49 +00001097 ERE32.setSymbolAndType(Index, Entry.Type);
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001098 write(ERE32.r_info);
Benjamin Kramer6c3c3492010-09-09 18:01:29 +00001099
Rafael Espindolafdaae0d2010-12-18 03:27:34 +00001100 if (hasRelocationAddend())
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001101 write(uint32_t(Entry.Addend));
Simon Atanasyanede43b72017-09-21 14:04:53 +00001102
1103 if (TargetObjectWriter->getEMachine() == ELF::EM_MIPS) {
1104 if (uint32_t RType = TargetObjectWriter->getRType2(Entry.Type)) {
1105 write(uint32_t(Entry.Offset));
1106
1107 ERE32.setSymbolAndType(0, RType);
1108 write(ERE32.r_info);
1109 write(uint32_t(0));
1110 }
1111 if (uint32_t RType = TargetObjectWriter->getRType3(Entry.Type)) {
1112 write(uint32_t(Entry.Offset));
1113
1114 ERE32.setSymbolAndType(0, RType);
1115 write(ERE32.r_info);
1116 write(uint32_t(0));
1117 }
1118 }
Benjamin Kramer6c3c3492010-09-09 18:01:29 +00001119 }
Matt Fleming6c1ad482010-08-16 18:57:57 +00001120 }
1121}
1122
Rafael Espindolab1863912015-04-30 21:10:06 +00001123const MCSectionELF *ELFObjectWriter::createStringTable(MCContext &Ctx) {
Rafael Espindola08850b32015-05-25 21:56:55 +00001124 const MCSectionELF *StrtabSection = SectionTable[StringTableIndex - 1];
Rafael Espindola39751af2016-10-04 22:43:25 +00001125 StrTabBuilder.write(getStream());
Rafael Espindolabda19802015-04-30 14:21:49 +00001126 return StrtabSection;
Rafael Espindolaa3e9a222010-11-11 18:13:52 +00001127}
1128
Rafael Espindolae92c1bf2015-05-21 19:42:35 +00001129void ELFObjectWriter::writeSection(const SectionIndexMapTy &SectionIndexMap,
1130 uint32_t GroupSymbolIndex, uint64_t Offset,
1131 uint64_t Size, const MCSectionELF &Section) {
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001132 uint64_t sh_link = 0;
1133 uint64_t sh_info = 0;
1134
1135 switch(Section.getType()) {
Rafael Espindola86fe2fe2015-04-06 03:16:51 +00001136 default:
1137 // Nothing to do.
1138 break;
1139
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001140 case ELF::SHT_DYNAMIC:
Rafael Espindola74ef4802015-04-30 21:20:06 +00001141 llvm_unreachable("SHT_DYNAMIC in a relocatable object");
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001142
1143 case ELF::SHT_REL:
1144 case ELF::SHT_RELA: {
Rafael Espindola3a7c0eb2015-02-17 20:37:50 +00001145 sh_link = SymbolTableIndex;
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001146 assert(sh_link && ".symtab not found");
Evgeniy Stepanov43dcf4d2017-03-14 19:28:51 +00001147 const MCSection *InfoSection = Section.getAssociatedSection();
1148 sh_info = SectionIndexMap.lookup(cast<MCSectionELF>(InfoSection));
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001149 break;
1150 }
1151
1152 case ELF::SHT_SYMTAB:
1153 case ELF::SHT_DYNSYM:
1154 sh_link = StringTableIndex;
1155 sh_info = LastLocalSymbolIndex;
1156 break;
1157
1158 case ELF::SHT_SYMTAB_SHNDX:
1159 sh_link = SymbolTableIndex;
1160 break;
1161
Nick Lewyckyc6ac5f72011-10-07 23:28:32 +00001162 case ELF::SHT_GROUP:
Rafael Espindolaa3e9a222010-11-11 18:13:52 +00001163 sh_link = SymbolTableIndex;
1164 sh_info = GroupSymbolIndex;
1165 break;
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001166 }
1167
Evgeniy Stepanov43dcf4d2017-03-14 19:28:51 +00001168 if (Section.getFlags() & ELF::SHF_LINK_ORDER) {
1169 const MCSymbol *Sym = Section.getAssociatedSymbol();
1170 const MCSectionELF *Sec = cast<MCSectionELF>(&Sym->getSection());
1171 sh_link = SectionIndexMap.lookup(Sec);
1172 }
Logan Chien4b724422013-02-05 14:18:59 +00001173
Rafael Espindola5960cee2015-05-22 23:58:30 +00001174 WriteSecHdrEntry(StrTabBuilder.getOffset(Section.getSectionName()),
Rafael Espindola28687582015-05-21 19:36:43 +00001175 Section.getType(), Section.getFlags(), 0, Offset, Size,
1176 sh_link, sh_info, Section.getAlignment(),
1177 Section.getEntrySize());
Rafael Espindola5a8d7812010-11-10 23:36:59 +00001178}
1179
Rafael Espindola7fe7e052015-02-17 20:40:59 +00001180void ELFObjectWriter::writeSectionHeader(
Rafael Espindola57c80832015-06-04 20:55:49 +00001181 const MCAsmLayout &Layout, const SectionIndexMapTy &SectionIndexMap,
Rafael Espindolaa8201692015-04-28 15:26:21 +00001182 const SectionOffsetsTy &SectionOffsets) {
Rafael Espindolab1863912015-04-30 21:10:06 +00001183 const unsigned NumSections = SectionTable.size();
Rafael Espindolaa3e9a222010-11-11 18:13:52 +00001184
Matt Fleming6c1ad482010-08-16 18:57:57 +00001185 // Null section first.
Rafael Espindola3fe87a12010-10-31 00:16:26 +00001186 uint64_t FirstSectionSize =
Rafael Espindolabf0db6c2015-04-15 13:07:47 +00001187 (NumSections + 1) >= ELF::SHN_LORESERVE ? NumSections + 1 : 0;
Rafael Espindola88d1f632015-04-29 20:25:24 +00001188 WriteSecHdrEntry(0, 0, 0, 0, 0, FirstSectionSize, 0, 0, 0, 0);
Matt Fleming6c1ad482010-08-16 18:57:57 +00001189
Rafael Espindola08850b32015-05-25 21:56:55 +00001190 for (const MCSectionELF *Section : SectionTable) {
Rafael Espindolaa3e9a222010-11-11 18:13:52 +00001191 uint32_t GroupSymbolIndex;
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001192 unsigned Type = Section->getType();
1193 if (Type != ELF::SHT_GROUP)
Rafael Espindolaa3e9a222010-11-11 18:13:52 +00001194 GroupSymbolIndex = 0;
1195 else
Rafael Espindola5e9ed902015-05-28 20:53:09 +00001196 GroupSymbolIndex = Section->getGroup()->getIndex();
Matt Fleming6c1ad482010-08-16 18:57:57 +00001197
Rafael Espindolabda19802015-04-30 14:21:49 +00001198 const std::pair<uint64_t, uint64_t> &Offsets =
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001199 SectionOffsets.find(Section)->second;
Rafael Espindola1aa20fc2015-05-21 19:46:39 +00001200 uint64_t Size;
Rafael Espindola5a1e80b2015-05-26 02:00:36 +00001201 if (Type == ELF::SHT_NOBITS)
1202 Size = Layout.getSectionAddressSize(Section);
1203 else
Rafael Espindola1aa20fc2015-05-21 19:46:39 +00001204 Size = Offsets.second - Offsets.first;
Rafael Espindola60ebca92010-12-02 03:09:06 +00001205
Rafael Espindolae92c1bf2015-05-21 19:42:35 +00001206 writeSection(SectionIndexMap, GroupSymbolIndex, Offsets.first, Size,
Rafael Espindola28687582015-05-21 19:36:43 +00001207 *Section);
Matt Fleming6c1ad482010-08-16 18:57:57 +00001208 }
1209}
1210
Jim Grosbach36e60e92015-06-04 22:24:41 +00001211void ELFObjectWriter::writeObject(MCAssembler &Asm,
Rafael Espindola1557fd62011-03-20 18:44:20 +00001212 const MCAsmLayout &Layout) {
Rafael Espindolabda19802015-04-30 14:21:49 +00001213 MCContext &Ctx = Asm.getContext();
Rafael Espindola5960cee2015-05-22 23:58:30 +00001214 MCSectionELF *StrtabSection =
1215 Ctx.getELFSection(".strtab", ELF::SHT_STRTAB, 0);
1216 StringTableIndex = addToSectionTable(StrtabSection);
Rafael Espindolabda19802015-04-30 14:21:49 +00001217
Rafael Espindola1557fd62011-03-20 18:44:20 +00001218 RevGroupMapTy RevGroupMap;
1219 SectionIndexMapTy SectionIndexMap;
1220
Rafael Espindolabda19802015-04-30 14:21:49 +00001221 std::map<const MCSymbol *, std::vector<const MCSectionELF *>> GroupMembers;
Rafael Espindola1557fd62011-03-20 18:44:20 +00001222
Rafael Espindola1557fd62011-03-20 18:44:20 +00001223 // Write out the ELF header ...
Rafael Espindola8c7829b2015-04-29 20:39:37 +00001224 writeHeader(Asm);
Rafael Espindola1557fd62011-03-20 18:44:20 +00001225
Rafael Espindola7230f802015-04-08 11:41:24 +00001226 // ... then the sections ...
Rafael Espindolabda19802015-04-30 14:21:49 +00001227 SectionOffsetsTy SectionOffsets;
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001228 std::vector<MCSectionELF *> Groups;
1229 std::vector<MCSectionELF *> Relocations;
Rafael Espindolae15b1b72015-05-27 13:30:50 +00001230 for (MCSection &Sec : Asm) {
1231 MCSectionELF &Section = static_cast<MCSectionELF &>(Sec);
Rafael Espindolabda19802015-04-30 14:21:49 +00001232
Rafael Espindolab20fbb82015-06-05 18:21:00 +00001233 align(Section.getAlignment());
Rafael Espindola1557fd62011-03-20 18:44:20 +00001234
Rafael Espindola642a2212015-04-14 22:54:16 +00001235 // Remember the offset into the file for this section.
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001236 uint64_t SecStart = getStream().tell();
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001237
Rafael Espindola0ccf9b72015-06-02 21:30:13 +00001238 const MCSymbolELF *SignatureSymbol = Section.getGroup();
Rafael Espindolae15b1b72015-05-27 13:30:50 +00001239 writeSectionData(Asm, Section, Layout);
Rafael Espindolab8cbb262015-04-30 00:30:40 +00001240
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001241 uint64_t SecEnd = getStream().tell();
Rafael Espindolabda19802015-04-30 14:21:49 +00001242 SectionOffsets[&Section] = std::make_pair(SecStart, SecEnd);
1243
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001244 MCSectionELF *RelSection = createRelocationSection(Ctx, Section);
Rafael Espindolabda19802015-04-30 14:21:49 +00001245
1246 if (SignatureSymbol) {
Rafael Espindolab5d316b2015-05-29 20:21:02 +00001247 Asm.registerSymbol(*SignatureSymbol);
Rafael Espindolabda19802015-04-30 14:21:49 +00001248 unsigned &GroupIdx = RevGroupMap[SignatureSymbol];
1249 if (!GroupIdx) {
Rafael Espindola0709a7b2015-05-21 19:20:38 +00001250 MCSectionELF *Group = Ctx.createELFGroupSection(SignatureSymbol);
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001251 GroupIdx = addToSectionTable(Group);
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001252 Group->setAlignment(4);
1253 Groups.push_back(Group);
Rafael Espindolabda19802015-04-30 14:21:49 +00001254 }
Rafael Espindola7830fc82015-06-05 17:54:25 +00001255 std::vector<const MCSectionELF *> &Members =
1256 GroupMembers[SignatureSymbol];
1257 Members.push_back(&Section);
Rafael Espindolabda19802015-04-30 14:21:49 +00001258 if (RelSection)
Rafael Espindola7830fc82015-06-05 17:54:25 +00001259 Members.push_back(RelSection);
Rafael Espindolabda19802015-04-30 14:21:49 +00001260 }
1261
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001262 SectionIndexMap[&Section] = addToSectionTable(&Section);
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001263 if (RelSection) {
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001264 SectionIndexMap[RelSection] = addToSectionTable(RelSection);
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001265 Relocations.push_back(RelSection);
1266 }
Rafael Espindolabda19802015-04-30 14:21:49 +00001267 }
1268
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001269 for (MCSectionELF *Group : Groups) {
Rafael Espindolab20fbb82015-06-05 18:21:00 +00001270 align(Group->getAlignment());
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001271
1272 // Remember the offset into the file for this section.
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001273 uint64_t SecStart = getStream().tell();
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001274
1275 const MCSymbol *SignatureSymbol = Group->getGroup();
1276 assert(SignatureSymbol);
1277 write(uint32_t(ELF::GRP_COMDAT));
1278 for (const MCSectionELF *Member : GroupMembers[SignatureSymbol]) {
1279 uint32_t SecIndex = SectionIndexMap.lookup(Member);
1280 write(SecIndex);
1281 }
1282
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001283 uint64_t SecEnd = getStream().tell();
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001284 SectionOffsets[Group] = std::make_pair(SecStart, SecEnd);
1285 }
1286
1287 // Compute symbol table information.
Rafael Espindolaaa486e92015-05-28 17:54:01 +00001288 computeSymbolTable(Asm, Layout, SectionIndexMap, RevGroupMap, SectionOffsets);
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001289
1290 for (MCSectionELF *RelSection : Relocations) {
Rafael Espindolab20fbb82015-06-05 18:21:00 +00001291 align(RelSection->getAlignment());
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001292
1293 // Remember the offset into the file for this section.
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001294 uint64_t SecStart = getStream().tell();
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001295
Evgeniy Stepanov43dcf4d2017-03-14 19:28:51 +00001296 writeRelocations(Asm,
1297 cast<MCSectionELF>(*RelSection->getAssociatedSection()));
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001298
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001299 uint64_t SecEnd = getStream().tell();
Rafael Espindola0a82ad72015-05-21 20:43:13 +00001300 SectionOffsets[RelSection] = std::make_pair(SecStart, SecEnd);
Rafael Espindola642a2212015-04-14 22:54:16 +00001301 }
1302
Rafael Espindola88d1f632015-04-29 20:25:24 +00001303 {
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001304 uint64_t SecStart = getStream().tell();
Rafael Espindolab1863912015-04-30 21:10:06 +00001305 const MCSectionELF *Sec = createStringTable(Ctx);
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001306 uint64_t SecEnd = getStream().tell();
Rafael Espindolabda19802015-04-30 14:21:49 +00001307 SectionOffsets[Sec] = std::make_pair(SecStart, SecEnd);
Rafael Espindola88abc392015-04-29 20:34:31 +00001308 }
1309
Rafael Espindola642a2212015-04-14 22:54:16 +00001310 uint64_t NaturalAlignment = is64Bit() ? 8 : 4;
Rafael Espindolab20fbb82015-06-05 18:21:00 +00001311 align(NaturalAlignment);
Rafael Espindola1557fd62011-03-20 18:44:20 +00001312
Rafael Espindola5568c832016-01-19 15:19:08 +00001313 const uint64_t SectionHeaderOffset = getStream().tell();
Rafael Espindola642a2212015-04-14 22:54:16 +00001314
Rafael Espindola1557fd62011-03-20 18:44:20 +00001315 // ... then the section header table ...
Rafael Espindola57c80832015-06-04 20:55:49 +00001316 writeSectionHeader(Layout, SectionIndexMap, SectionOffsets);
Rafael Espindola642a2212015-04-14 22:54:16 +00001317
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001318 uint16_t NumSections = (SectionTable.size() + 1 >= ELF::SHN_LORESERVE)
Aaron Ballman9cab7322015-04-30 14:03:12 +00001319 ? (uint16_t)ELF::SHN_UNDEF
Rafael Espindola03d7abb2015-04-30 20:53:27 +00001320 : SectionTable.size() + 1;
Rafael Espindola8c7829b2015-04-29 20:39:37 +00001321 if (sys::IsLittleEndianHost != IsLittleEndian)
1322 sys::swapByteOrder(NumSections);
1323 unsigned NumSectionsOffset;
1324
Rafael Espindola642a2212015-04-14 22:54:16 +00001325 if (is64Bit()) {
1326 uint64_t Val = SectionHeaderOffset;
1327 if (sys::IsLittleEndianHost != IsLittleEndian)
1328 sys::swapByteOrder(Val);
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001329 getStream().pwrite(reinterpret_cast<char *>(&Val), sizeof(Val),
1330 offsetof(ELF::Elf64_Ehdr, e_shoff));
Rafael Espindola8c7829b2015-04-29 20:39:37 +00001331 NumSectionsOffset = offsetof(ELF::Elf64_Ehdr, e_shnum);
Rafael Espindola642a2212015-04-14 22:54:16 +00001332 } else {
1333 uint32_t Val = SectionHeaderOffset;
1334 if (sys::IsLittleEndianHost != IsLittleEndian)
1335 sys::swapByteOrder(Val);
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001336 getStream().pwrite(reinterpret_cast<char *>(&Val), sizeof(Val),
1337 offsetof(ELF::Elf32_Ehdr, e_shoff));
Rafael Espindola8c7829b2015-04-29 20:39:37 +00001338 NumSectionsOffset = offsetof(ELF::Elf32_Ehdr, e_shnum);
Rafael Espindola642a2212015-04-14 22:54:16 +00001339 }
David Majnemerabdb2d2a2015-09-01 16:19:03 +00001340 getStream().pwrite(reinterpret_cast<char *>(&NumSections),
1341 sizeof(NumSections), NumSectionsOffset);
Rafael Espindola1557fd62011-03-20 18:44:20 +00001342}
1343
Jim Grosbach36e60e92015-06-04 22:24:41 +00001344bool ELFObjectWriter::isSymbolRefDifferenceFullyResolvedImpl(
Rafael Espindola95fb9b92015-06-02 20:38:46 +00001345 const MCAssembler &Asm, const MCSymbol &SA, const MCFragment &FB,
Rafael Espindola35d61892015-04-17 21:15:17 +00001346 bool InSet, bool IsPCRel) const {
Rafael Espindola95fb9b92015-06-02 20:38:46 +00001347 const auto &SymA = cast<MCSymbolELF>(SA);
Rafael Espindola35d61892015-04-17 21:15:17 +00001348 if (IsPCRel) {
1349 assert(!InSet);
Peter Collingbourne36003052017-04-08 23:35:49 +00001350 if (isWeak(SymA))
Rafael Espindola35d61892015-04-17 21:15:17 +00001351 return false;
1352 }
Jim Grosbach36e60e92015-06-04 22:24:41 +00001353 return MCObjectWriter::isSymbolRefDifferenceFullyResolvedImpl(Asm, SymA, FB,
Rafael Espindola35d61892015-04-17 21:15:17 +00001354 InSet, IsPCRel);
Eli Friedmand92d17b2011-03-03 07:24:36 +00001355}
1356
Lang Hames60fbc7c2017-10-10 16:28:07 +00001357std::unique_ptr<MCObjectWriter>
Lang Hamesdcb312b2017-10-09 23:53:15 +00001358llvm::createELFObjectWriter(std::unique_ptr<MCELFObjectTargetWriter> MOTW,
1359 raw_pwrite_stream &OS, bool IsLittleEndian) {
Lang Hames60fbc7c2017-10-10 16:28:07 +00001360 return llvm::make_unique<ELFObjectWriter>(std::move(MOTW), OS,
1361 IsLittleEndian);
Rafael Espindolab264d332011-12-21 17:30:17 +00001362}