blob: eb0c1416638ab664ac11f4a75c0c8910a2188da4 [file] [log] [blame]
Michael J. Spencer84487f12015-07-24 21:03:07 +00001//===- Writer.cpp ---------------------------------------------------------===//
2//
3// The LLVM Linker
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
Michael J. Spencerf8325412015-09-04 22:48:30 +000010#include "Writer.h"
Rui Ueyamacb8474ed2015-08-05 23:51:50 +000011#include "Config.h"
Rafael Espindola5805c4f2015-09-21 21:38:08 +000012#include "OutputSections.h"
Rui Ueyamaafff74e22015-08-05 23:24:46 +000013#include "SymbolTable.h"
Rafael Espindola01205f72015-09-22 18:19:46 +000014#include "Target.h"
Rafael Espindola6b83b902015-08-12 00:00:24 +000015
Denis Protivensky8e3b38a2015-11-12 09:52:08 +000016#include "llvm/ADT/StringMap.h"
Hal Finkel3bae2d82015-10-12 20:51:48 +000017#include "llvm/ADT/StringSwitch.h"
Rui Ueyamaafff74e22015-08-05 23:24:46 +000018#include "llvm/Support/FileOutputBuffer.h"
George Rimara5fbebc2015-12-10 09:12:18 +000019#include "llvm/Support/raw_ostream.h"
Rui Ueyamad9189ce2015-10-15 17:11:03 +000020#include "llvm/Support/StringSaver.h"
Michael J. Spencer84487f12015-07-24 21:03:07 +000021
22using namespace llvm;
23using namespace llvm::ELF;
24using namespace llvm::object;
25
26using namespace lld;
27using namespace lld::elf2;
28
Rui Ueyamaafff74e22015-08-05 23:24:46 +000029namespace {
30// The writer writes a SymbolTable result to a file.
31template <class ELFT> class Writer {
32public:
Rafael Espindola18608a02015-09-08 21:57:31 +000033 typedef typename ELFFile<ELFT>::uintX_t uintX_t;
34 typedef typename ELFFile<ELFT>::Elf_Shdr Elf_Shdr;
35 typedef typename ELFFile<ELFT>::Elf_Ehdr Elf_Ehdr;
36 typedef typename ELFFile<ELFT>::Elf_Phdr Elf_Phdr;
37 typedef typename ELFFile<ELFT>::Elf_Sym Elf_Sym;
Davide Italiano6d328d32015-09-16 20:45:57 +000038 typedef typename ELFFile<ELFT>::Elf_Sym_Range Elf_Sym_Range;
Rafael Espindola19e38892015-09-16 15:54:15 +000039 typedef typename ELFFile<ELFT>::Elf_Rela Elf_Rela;
Rui Ueyama3ce825e2015-10-09 21:07:25 +000040 Writer(SymbolTable<ELFT> &S) : Symtab(S) {}
Rui Ueyamaafff74e22015-08-05 23:24:46 +000041 void run();
42
43private:
Rui Ueyama5a9640b2015-10-08 23:49:30 +000044 void copyLocalSymbols();
Rui Ueyamaafff74e22015-08-05 23:24:46 +000045 void createSections();
Rui Ueyama2df0fd82015-12-25 07:38:58 +000046
47 OutputSectionBase<ELFT> *createOutputSection(InputSectionBase<ELFT> *C,
48 StringRef Name, uintX_t Type,
49 uintX_t Flags);
50
Rafael Espindola67a5da62015-09-17 14:02:10 +000051 template <bool isRela>
Rafael Espindola0c6a4f12015-11-11 19:54:14 +000052 void scanRelocs(InputSectionBase<ELFT> &C,
Rafael Espindola67a5da62015-09-17 14:02:10 +000053 iterator_range<const Elf_Rel_Impl<ELFT, isRela> *> Rels);
Rui Ueyama2df0fd82015-12-25 07:38:58 +000054
Rafael Espindolaa0fa8482015-11-11 15:29:50 +000055 void scanRelocs(InputSection<ELFT> &C);
Rafael Espindola0c6a4f12015-11-11 19:54:14 +000056 void scanRelocs(InputSectionBase<ELFT> &S, const Elf_Shdr &RelSec);
Rafael Espindola97cbe3e32015-12-23 15:20:38 +000057 void updateRelro(Elf_Phdr *Cur, Elf_Phdr *GnuRelroPhdr, uintX_t VA);
Rui Ueyamaafff74e22015-08-05 23:24:46 +000058 void assignAddresses();
Denis Protivensky8e3b38a2015-11-12 09:52:08 +000059 void buildSectionMap();
Rui Ueyamaafff74e22015-08-05 23:24:46 +000060 void openFile(StringRef OutputPath);
61 void writeHeader();
62 void writeSections();
Denis Protivensky8e3b38a2015-11-12 09:52:08 +000063 bool isDiscarded(InputSectionBase<ELFT> *IS) const;
64 StringRef getOutputSectionName(StringRef S) const;
Rafael Espindola70107762015-09-11 18:49:42 +000065 bool needsInterpSection() const {
Rui Ueyama0d0bcf72015-10-07 21:25:39 +000066 return !Symtab.getSharedFiles().empty() && !Config->DynamicLinker.empty();
Rafael Espindola70107762015-09-11 18:49:42 +000067 }
Michael J. Spencerf32446f2015-10-06 20:39:09 +000068 bool isOutputDynamic() const {
Rui Ueyama0d0bcf72015-10-07 21:25:39 +000069 return !Symtab.getSharedFiles().empty() || Config->Shared;
Rafael Espindola4340aad2015-09-11 22:42:45 +000070 }
Rui Ueyama803195e2015-10-23 21:45:59 +000071 int getPhdrsNum() const;
Rui Ueyamaafff74e22015-08-05 23:24:46 +000072
Rafael Espindola443f50a2015-11-03 21:35:14 +000073 OutputSection<ELFT> *getBSS();
Rafael Espindola11191912015-12-24 16:23:37 +000074 void addCommonSymbols(std::vector<DefinedCommon *> &Syms);
Rafael Espindola443f50a2015-11-03 21:35:14 +000075 void addSharedCopySymbols(std::vector<SharedSymbol<ELFT> *> &Syms);
76
Rui Ueyamaafff74e22015-08-05 23:24:46 +000077 std::unique_ptr<llvm::FileOutputBuffer> Buffer;
Michael J. Spencer2f008242015-09-17 19:58:07 +000078
Rui Ueyamad9189ce2015-10-15 17:11:03 +000079 BumpPtrAllocator Alloc;
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +000080 std::vector<OutputSectionBase<ELFT> *> OutputSections;
Rui Ueyamad4ea7dd2015-12-26 07:01:26 +000081 std::vector<std::unique_ptr<OutputSectionBase<ELFT>>> OwningSections;
Rafael Espindola5f553872015-09-08 17:39:39 +000082 unsigned getNumSections() const { return OutputSections.size() + 1; }
Rui Ueyamaafff74e22015-08-05 23:24:46 +000083
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +000084 void addStartStopSymbols(OutputSectionBase<ELFT> *Sec);
Rui Ueyama536c9992015-10-11 19:45:58 +000085 void setPhdr(Elf_Phdr *PH, uint32_t Type, uint32_t Flags, uintX_t FileOff,
Rui Ueyama803195e2015-10-23 21:45:59 +000086 uintX_t VA, uintX_t Size, uintX_t Align);
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +000087 void copyPhdr(Elf_Phdr *PH, OutputSectionBase<ELFT> *From);
Rui Ueyama2f1b79f2015-10-10 22:34:30 +000088
George Rimare3336c02015-11-24 10:15:50 +000089 bool HasRelro = false;
Rui Ueyama3ce825e2015-10-09 21:07:25 +000090 SymbolTable<ELFT> &Symtab;
Rui Ueyama953c2c42015-10-10 23:59:57 +000091 std::vector<Elf_Phdr> Phdrs;
Michael J. Spencer2f008242015-09-17 19:58:07 +000092
Rafael Espindola98f6bd02015-08-11 23:14:13 +000093 uintX_t FileSize;
Rui Ueyamaafff74e22015-08-05 23:24:46 +000094 uintX_t SectionHeaderOff;
Denis Protivensky8e3b38a2015-11-12 09:52:08 +000095
96 llvm::StringMap<llvm::StringRef> InputToOutputSection;
Rui Ueyamaafff74e22015-08-05 23:24:46 +000097};
98} // anonymous namespace
99
Rui Ueyama6192c122015-12-16 23:33:56 +0000100template <class ELFT> static bool shouldUseRela() {
101 ELFKind K = cast<ELFFileBase<ELFT>>(Config->FirstElf)->getELFKind();
102 return K == ELF64LEKind || K == ELF64BEKind;
103}
104
Rui Ueyama3ce825e2015-10-09 21:07:25 +0000105template <class ELFT> void lld::elf2::writeResult(SymbolTable<ELFT> *Symtab) {
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000106 // Initialize output sections that are handled by Writer specially.
107 // Don't reorder because the order of initialization matters.
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +0000108 InterpSection<ELFT> Interp;
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000109 Out<ELFT>::Interp = &Interp;
George Rimar0f5ac9f2015-10-20 17:21:35 +0000110 StringTableSection<ELFT> ShStrTab(".shstrtab", false);
111 Out<ELFT>::ShStrTab = &ShStrTab;
112 StringTableSection<ELFT> StrTab(".strtab", false);
George Rimar5dad7c12015-10-24 08:52:46 +0000113 if (!Config->StripAll)
114 Out<ELFT>::StrTab = &StrTab;
George Rimar0f5ac9f2015-10-20 17:21:35 +0000115 StringTableSection<ELFT> DynStrTab(".dynstr", true);
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000116 Out<ELFT>::DynStrTab = &DynStrTab;
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000117 GotSection<ELFT> Got;
118 Out<ELFT>::Got = &Got;
George Rimar648a2c32015-10-20 08:54:27 +0000119 GotPltSection<ELFT> GotPlt;
120 if (Target->supportsLazyRelocations())
121 Out<ELFT>::GotPlt = &GotPlt;
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000122 PltSection<ELFT> Plt;
123 Out<ELFT>::Plt = &Plt;
George Rimar5dad7c12015-10-24 08:52:46 +0000124 std::unique_ptr<SymbolTableSection<ELFT>> SymTab;
125 if (!Config->StripAll) {
126 SymTab.reset(new SymbolTableSection<ELFT>(*Symtab, *Out<ELFT>::StrTab));
127 Out<ELFT>::SymTab = SymTab.get();
128 }
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000129 SymbolTableSection<ELFT> DynSymTab(*Symtab, *Out<ELFT>::DynStrTab);
130 Out<ELFT>::DynSymTab = &DynSymTab;
131 HashTableSection<ELFT> HashTab;
Igor Kudrin1b0d7062015-10-22 08:21:35 +0000132 if (Config->SysvHash)
133 Out<ELFT>::HashTab = &HashTab;
134 GnuHashTableSection<ELFT> GnuHashTab;
135 if (Config->GnuHash)
136 Out<ELFT>::GnuHashTab = &GnuHashTab;
Rui Ueyama6192c122015-12-16 23:33:56 +0000137 bool IsRela = shouldUseRela<ELFT>();
George Rimar648a2c32015-10-20 08:54:27 +0000138 RelocationSection<ELFT> RelaDyn(IsRela ? ".rela.dyn" : ".rel.dyn", IsRela);
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000139 Out<ELFT>::RelaDyn = &RelaDyn;
George Rimar648a2c32015-10-20 08:54:27 +0000140 RelocationSection<ELFT> RelaPlt(IsRela ? ".rela.plt" : ".rel.plt", IsRela);
141 if (Target->supportsLazyRelocations())
142 Out<ELFT>::RelaPlt = &RelaPlt;
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000143 DynamicSection<ELFT> Dynamic(*Symtab);
144 Out<ELFT>::Dynamic = &Dynamic;
145
Rui Ueyama0d0bcf72015-10-07 21:25:39 +0000146 Writer<ELFT>(*Symtab).run();
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000147}
148
Michael J. Spencer84487f12015-07-24 21:03:07 +0000149// The main function of the writer.
Rui Ueyamaafff74e22015-08-05 23:24:46 +0000150template <class ELFT> void Writer<ELFT>::run() {
Denis Protivensky8e3b38a2015-11-12 09:52:08 +0000151 buildSectionMap();
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000152 if (!Config->DiscardAll)
153 copyLocalSymbols();
Michael J. Spencer84487f12015-07-24 21:03:07 +0000154 createSections();
155 assignAddresses();
Rui Ueyamacb8474ed2015-08-05 23:51:50 +0000156 openFile(Config->OutputFile);
Michael J. Spencer84487f12015-07-24 21:03:07 +0000157 writeHeader();
158 writeSections();
159 error(Buffer->commit());
160}
161
Rafael Espindolaa7471792015-08-13 17:04:50 +0000162namespace {
163template <bool Is64Bits> struct SectionKey {
164 typedef typename std::conditional<Is64Bits, uint64_t, uint32_t>::type uintX_t;
165 StringRef Name;
Rui Ueyama0fb1ee02015-10-02 21:13:19 +0000166 uint32_t Type;
167 uintX_t Flags;
Rafael Espindolac159c962015-10-19 21:00:02 +0000168 uintX_t EntSize;
Rafael Espindolaa7471792015-08-13 17:04:50 +0000169};
170}
171namespace llvm {
172template <bool Is64Bits> struct DenseMapInfo<SectionKey<Is64Bits>> {
173 static SectionKey<Is64Bits> getEmptyKey() {
Rafael Espindolac159c962015-10-19 21:00:02 +0000174 return SectionKey<Is64Bits>{DenseMapInfo<StringRef>::getEmptyKey(), 0, 0,
175 0};
Rafael Espindolaa7471792015-08-13 17:04:50 +0000176 }
177 static SectionKey<Is64Bits> getTombstoneKey() {
178 return SectionKey<Is64Bits>{DenseMapInfo<StringRef>::getTombstoneKey(), 0,
Rafael Espindolac159c962015-10-19 21:00:02 +0000179 0, 0};
Rafael Espindolaa7471792015-08-13 17:04:50 +0000180 }
181 static unsigned getHashValue(const SectionKey<Is64Bits> &Val) {
Rafael Espindolac159c962015-10-19 21:00:02 +0000182 return hash_combine(Val.Name, Val.Type, Val.Flags, Val.EntSize);
Rafael Espindolaa7471792015-08-13 17:04:50 +0000183 }
184 static bool isEqual(const SectionKey<Is64Bits> &LHS,
185 const SectionKey<Is64Bits> &RHS) {
186 return DenseMapInfo<StringRef>::isEqual(LHS.Name, RHS.Name) &&
Rafael Espindolac159c962015-10-19 21:00:02 +0000187 LHS.Type == RHS.Type && LHS.Flags == RHS.Flags &&
188 LHS.EntSize == RHS.EntSize;
Rafael Espindolaa7471792015-08-13 17:04:50 +0000189 }
190};
191}
192
Rafael Espindola19e38892015-09-16 15:54:15 +0000193// The reason we have to do this early scan is as follows
194// * To mmap the output file, we need to know the size
195// * For that, we need to know how many dynamic relocs we will have.
196// It might be possible to avoid this by outputting the file with write:
197// * Write the allocated output sections, computing addresses.
198// * Apply relocations, recording which ones require a dynamic reloc.
199// * Write the dynamic relocations.
200// * Write the rest of the file.
201template <class ELFT>
Rafael Espindola67a5da62015-09-17 14:02:10 +0000202template <bool isRela>
203void Writer<ELFT>::scanRelocs(
Rafael Espindola0c6a4f12015-11-11 19:54:14 +0000204 InputSectionBase<ELFT> &C,
Rafael Espindola67a5da62015-09-17 14:02:10 +0000205 iterator_range<const Elf_Rel_Impl<ELFT, isRela> *> Rels) {
206 typedef Elf_Rel_Impl<ELFT, isRela> RelType;
207 const ObjectFile<ELFT> &File = *C.getFile();
Rafael Espindola67a5da62015-09-17 14:02:10 +0000208 for (const RelType &RI : Rels) {
Rui Ueyama64558522015-10-16 22:51:43 +0000209 uint32_t SymIndex = RI.getSymbol(Config->Mips64EL);
Rafael Espindola5c2310c2015-09-18 14:40:19 +0000210 SymbolBody *Body = File.getSymbolBody(SymIndex);
Rui Ueyama64558522015-10-16 22:51:43 +0000211 uint32_t Type = RI.getType(Config->Mips64EL);
Rui Ueyama35da9b62015-10-11 20:59:12 +0000212
George Rimarbfb7bf72015-12-21 10:00:12 +0000213 if (Target->isGotRelative(Type))
214 HasGotOffRel = true;
215
Michael J. Spencerecd7f372015-11-13 00:32:58 +0000216 if (Target->isTlsLocalDynamicReloc(Type)) {
George Rimar6713cf82015-11-25 21:46:05 +0000217 if (Target->isTlsOptimized(Type, nullptr))
218 continue;
George Rimar95c1a582015-12-07 08:02:20 +0000219 if (Out<ELFT>::Got->addCurrentModuleTlsIndex())
Michael J. Spencer627ae702015-11-13 00:28:34 +0000220 Out<ELFT>::RelaDyn->addReloc({&C, &RI});
Michael J. Spencer1e225612015-11-11 01:00:24 +0000221 continue;
222 }
223
Rui Ueyama35da9b62015-10-11 20:59:12 +0000224 // Set "used" bit for --as-needed.
225 if (Body && Body->isUndefined() && !Body->isWeak())
226 if (auto *S = dyn_cast<SharedSymbol<ELFT>>(Body->repl()))
227 S->File->IsUsed = true;
228
229 if (Body)
230 Body = Body->repl();
Michael J. Spencer627ae702015-11-13 00:28:34 +0000231
Rui Ueyama62d0e322015-12-17 00:04:18 +0000232 if (Body && Body->isTls() && Target->isTlsGlobalDynamicReloc(Type)) {
George Rimar25411f252015-12-04 11:20:13 +0000233 bool Opt = Target->isTlsOptimized(Type, Body);
234 if (!Opt && Out<ELFT>::Got->addDynTlsEntry(Body)) {
George Rimar90cd0a82015-12-01 19:20:26 +0000235 Out<ELFT>::RelaDyn->addReloc({&C, &RI});
236 Out<ELFT>::RelaDyn->addReloc({nullptr, nullptr});
237 Body->setUsedInDynamicReloc();
George Rimar25411f252015-12-04 11:20:13 +0000238 continue;
George Rimar90cd0a82015-12-01 19:20:26 +0000239 }
George Rimar25411f252015-12-04 11:20:13 +0000240 if (!canBePreempted(Body, true))
241 continue;
Michael J. Spencer627ae702015-11-13 00:28:34 +0000242 }
243
George Rimar6f17e092015-12-17 09:32:21 +0000244 if (Body && Body->isTls() && !Target->isTlsDynReloc(Type, *Body))
George Rimar687138c2015-11-13 16:28:53 +0000245 continue;
246
George Rimar6f17e092015-12-17 09:32:21 +0000247 if (Target->relocNeedsDynRelative(Type)) {
248 RelType *Rel = new (Alloc) RelType;
249 Rel->setSymbolAndType(0, Target->getRelativeReloc(), Config->Mips64EL);
250 Rel->r_offset = RI.r_offset;
251 Out<ELFT>::RelaDyn->addReloc({&C, Rel});
252 }
253
Rafael Espindolacc6ebb82015-10-14 18:42:16 +0000254 bool NeedsGot = false;
George Rimar648a2c32015-10-20 08:54:27 +0000255 bool NeedsPlt = false;
Rafael Espindolaae244002015-10-05 19:30:12 +0000256 if (Body) {
George Rimarbc590fe2015-10-28 16:48:58 +0000257 if (auto *E = dyn_cast<SharedSymbol<ELFT>>(Body)) {
Rui Ueyamabb936062015-12-17 01:14:23 +0000258 if (E->NeedsCopy)
George Rimarbc590fe2015-10-28 16:48:58 +0000259 continue;
Rui Ueyama02dfd492015-12-17 01:18:40 +0000260 if (Target->needsCopyRel(Type, *Body))
Rui Ueyamabb936062015-12-17 01:14:23 +0000261 E->NeedsCopy = true;
George Rimarbc590fe2015-10-28 16:48:58 +0000262 }
George Rimar648a2c32015-10-20 08:54:27 +0000263 NeedsPlt = Target->relocNeedsPlt(Type, *Body);
264 if (NeedsPlt) {
Rafael Espindolaae244002015-10-05 19:30:12 +0000265 if (Body->isInPlt())
266 continue;
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000267 Out<ELFT>::Plt->addEntry(Body);
Rui Ueyamac58656c2015-10-13 16:59:30 +0000268 }
Rafael Espindolacc6ebb82015-10-14 18:42:16 +0000269 NeedsGot = Target->relocNeedsGot(Type, *Body);
270 if (NeedsGot) {
George Rimar648a2c32015-10-20 08:54:27 +0000271 if (NeedsPlt && Target->supportsLazyRelocations()) {
272 Out<ELFT>::GotPlt->addEntry(Body);
273 } else {
274 if (Body->isInGot())
275 continue;
276 Out<ELFT>::Got->addEntry(Body);
277 }
Rafael Espindolaae244002015-10-05 19:30:12 +0000278 }
Rafael Espindolaeb792732015-09-21 15:11:29 +0000279 }
Rui Ueyama35da9b62015-10-11 20:59:12 +0000280
George Rimara07ff662015-12-21 10:12:06 +0000281 // An STT_GNU_IFUNC symbol always uses a PLT entry, and all references
282 // to the symbol go through the PLT. This is true even for a local
283 // symbol, although local symbols normally do not require PLT entries.
284 if (Body && isGnuIFunc<ELFT>(*Body)) {
285 Body->setUsedInDynamicReloc();
286 Out<ELFT>::RelaPlt->addReloc({&C, &RI});
287 continue;
288 }
289
Igor Kudrin15cd9ff2015-11-06 07:43:03 +0000290 if (Config->EMachine == EM_MIPS && NeedsGot) {
291 // MIPS ABI has special rules to process GOT entries
292 // and doesn't require relocation entries for them.
293 // See "Global Offset Table" in Chapter 5 in the following document
294 // for detailed description:
295 // ftp://www.linux-mips.org/pub/linux/mips/doc/ABI/mipsabi.pdf
296 Body->setUsedInDynamicReloc();
297 continue;
298 }
Rafael Espindolacc6ebb82015-10-14 18:42:16 +0000299 bool CBP = canBePreempted(Body, NeedsGot);
Rui Ueyama35da9b62015-10-11 20:59:12 +0000300 if (!CBP && (!Config->Shared || Target->isRelRelative(Type)))
301 continue;
302 if (CBP)
Rafael Espindolaa6627382015-10-06 23:56:53 +0000303 Body->setUsedInDynamicReloc();
George Rimar648a2c32015-10-20 08:54:27 +0000304 if (NeedsPlt && Target->supportsLazyRelocations())
Michael J. Spencer627ae702015-11-13 00:28:34 +0000305 Out<ELFT>::RelaPlt->addReloc({&C, &RI});
George Rimar648a2c32015-10-20 08:54:27 +0000306 else
Michael J. Spencer627ae702015-11-13 00:28:34 +0000307 Out<ELFT>::RelaDyn->addReloc({&C, &RI});
Rafael Espindola67a5da62015-09-17 14:02:10 +0000308 }
309}
310
Rafael Espindolaa0fa8482015-11-11 15:29:50 +0000311template <class ELFT> void Writer<ELFT>::scanRelocs(InputSection<ELFT> &C) {
Rafael Espindola19e38892015-09-16 15:54:15 +0000312 if (!(C.getSectionHdr()->sh_flags & SHF_ALLOC))
313 return;
314
Rafael Espindola0c6a4f12015-11-11 19:54:14 +0000315 for (const Elf_Shdr *RelSec : C.RelocSections)
316 scanRelocs(C, *RelSec);
317}
318
319template <class ELFT>
320void Writer<ELFT>::scanRelocs(InputSectionBase<ELFT> &S,
321 const Elf_Shdr &RelSec) {
322 ELFFile<ELFT> &EObj = S.getFile()->getObj();
323 if (RelSec.sh_type == SHT_RELA)
324 scanRelocs(S, EObj.relas(&RelSec));
325 else
326 scanRelocs(S, EObj.rels(&RelSec));
Rafael Espindola19e38892015-09-16 15:54:15 +0000327}
328
Rafael Espindola6173f842015-09-23 14:37:01 +0000329template <class ELFT>
Rui Ueyama3ce825e2015-10-09 21:07:25 +0000330static void reportUndefined(const SymbolTable<ELFT> &S, const SymbolBody &Sym) {
George Rimar57e40de2015-10-01 20:14:45 +0000331 if (Config->Shared && !Config->NoUndefined)
George Rimaree058282015-10-01 17:24:24 +0000332 return;
333
Rafael Espindola8a9f90e2015-12-21 19:09:19 +0000334 ELFFileBase<ELFT> *SymFile = findFile<ELFT>(S.getObjectFiles(), &Sym);
Rafael Espindola551dfd82015-09-25 19:24:57 +0000335 std::string Message = "undefined symbol: " + Sym.getName().str();
Rafael Espindola6173f842015-09-23 14:37:01 +0000336 if (SymFile)
Rafael Espindola551dfd82015-09-25 19:24:57 +0000337 Message += " in " + SymFile->getName().str();
338 if (Config->NoInhibitExec)
339 warning(Message);
Rafael Espindola6173f842015-09-23 14:37:01 +0000340 else
Rafael Espindola551dfd82015-09-25 19:24:57 +0000341 error(Message);
Rafael Espindola6173f842015-09-23 14:37:01 +0000342}
343
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000344// Local symbols are not in the linker's symbol table. This function scans
345// each object file's symbol table to copy local symbols to the output.
346template <class ELFT> void Writer<ELFT>::copyLocalSymbols() {
Rui Ueyama3ce825e2015-10-09 21:07:25 +0000347 for (const std::unique_ptr<ObjectFile<ELFT>> &F : Symtab.getObjectFiles()) {
348 for (const Elf_Sym &Sym : F->getLocalSymbols()) {
349 ErrorOr<StringRef> SymNameOrErr = Sym.getName(F->getStringTable());
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000350 error(SymNameOrErr);
351 StringRef SymName = *SymNameOrErr;
Rui Ueyama3ce825e2015-10-09 21:07:25 +0000352 if (!shouldKeepInSymtab<ELFT>(*F, SymName, Sym))
Rafael Espindola444576d2015-10-09 19:25:07 +0000353 continue;
George Rimar5dad7c12015-10-24 08:52:46 +0000354 if (Out<ELFT>::SymTab)
355 Out<ELFT>::SymTab->addLocalSymbol(SymName);
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000356 }
357 }
358}
359
Hal Finkel3bae2d82015-10-12 20:51:48 +0000360// PPC64 has a number of special SHT_PROGBITS+SHF_ALLOC+SHF_WRITE sections that
361// we would like to make sure appear is a specific order to maximize their
362// coverage by a single signed 16-bit offset from the TOC base pointer.
363// Conversely, the special .tocbss section should be first among all SHT_NOBITS
364// sections. This will put it next to the loaded special PPC64 sections (and,
365// thus, within reach of the TOC base pointer).
366static int getPPC64SectionRank(StringRef SectionName) {
367 return StringSwitch<int>(SectionName)
368 .Case(".tocbss", 0)
369 .Case(".branch_lt", 2)
370 .Case(".toc", 3)
371 .Case(".toc1", 4)
372 .Case(".opd", 5)
373 .Default(1);
374}
375
George Rimare3336c02015-11-24 10:15:50 +0000376template <class ELFT> static bool isRelroSection(OutputSectionBase<ELFT> *Sec) {
377 typename OutputSectionBase<ELFT>::uintX_t Flags = Sec->getFlags();
378 if (!(Flags & SHF_ALLOC) || !(Flags & SHF_WRITE))
379 return false;
Rui Ueyamaccfc3262015-12-10 19:13:08 +0000380 if (Flags & SHF_TLS)
381 return true;
George Rimare3336c02015-11-24 10:15:50 +0000382 uint32_t Type = Sec->getType();
Rui Ueyamaccfc3262015-12-10 19:13:08 +0000383 if (Type == SHT_INIT_ARRAY || Type == SHT_FINI_ARRAY ||
384 Type == SHT_PREINIT_ARRAY)
George Rimare3336c02015-11-24 10:15:50 +0000385 return true;
386 if (Sec == Out<ELFT>::GotPlt)
387 return Config->ZNow;
388 if (Sec == Out<ELFT>::Dynamic || Sec == Out<ELFT>::Got)
389 return true;
Rui Ueyama01faef02015-12-10 19:19:04 +0000390 StringRef S = Sec->getName();
391 return S == ".data.rel.ro" || S == ".ctors" || S == ".dtors" || S == ".jcr" ||
392 S == ".eh_frame";
George Rimare3336c02015-11-24 10:15:50 +0000393}
394
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000395// Output section ordering is determined by this function.
396template <class ELFT>
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +0000397static bool compareOutputSections(OutputSectionBase<ELFT> *A,
398 OutputSectionBase<ELFT> *B) {
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000399 typedef typename ELFFile<ELFT>::uintX_t uintX_t;
400
401 uintX_t AFlags = A->getFlags();
402 uintX_t BFlags = B->getFlags();
403
404 // Allocatable sections go first to reduce the total PT_LOAD size and
405 // so debug info doesn't change addresses in actual code.
406 bool AIsAlloc = AFlags & SHF_ALLOC;
407 bool BIsAlloc = BFlags & SHF_ALLOC;
408 if (AIsAlloc != BIsAlloc)
409 return AIsAlloc;
410
411 // We don't have any special requirements for the relative order of
412 // two non allocatable sections.
413 if (!AIsAlloc)
414 return false;
415
416 // We want the read only sections first so that they go in the PT_LOAD
417 // covering the program headers at the start of the file.
418 bool AIsWritable = AFlags & SHF_WRITE;
419 bool BIsWritable = BFlags & SHF_WRITE;
420 if (AIsWritable != BIsWritable)
421 return BIsWritable;
422
423 // For a corresponding reason, put non exec sections first (the program
424 // header PT_LOAD is not executable).
425 bool AIsExec = AFlags & SHF_EXECINSTR;
426 bool BIsExec = BFlags & SHF_EXECINSTR;
427 if (AIsExec != BIsExec)
428 return BIsExec;
429
Hal Finkel0d7e83b2015-10-13 17:57:46 +0000430 // If we got here we know that both A and B are in the same PT_LOAD.
Michael J. Spencer1bf73002015-10-16 23:11:07 +0000431
432 // The TLS initialization block needs to be a single contiguous block in a R/W
433 // PT_LOAD, so stick TLS sections directly before R/W sections. The TLS NOBITS
434 // sections are placed here as they don't take up virtual address space in the
435 // PT_LOAD.
Rui Ueyama61805ec2015-12-17 00:12:03 +0000436 bool AIsTls = AFlags & SHF_TLS;
437 bool BIsTls = BFlags & SHF_TLS;
438 if (AIsTls != BIsTls)
439 return AIsTls;
Michael J. Spencer1bf73002015-10-16 23:11:07 +0000440
Hal Finkel08be6142015-10-13 18:55:01 +0000441 // The next requirement we have is to put nobits sections last. The
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000442 // reason is that the only thing the dynamic linker will see about
443 // them is a p_memsz that is larger than p_filesz. Seeing that it
444 // zeros the end of the PT_LOAD, so that has to correspond to the
445 // nobits sections.
Hal Finkel600ff142015-10-13 19:27:12 +0000446 bool AIsNoBits = A->getType() == SHT_NOBITS;
447 bool BIsNoBits = B->getType() == SHT_NOBITS;
448 if (AIsNoBits != BIsNoBits)
449 return BIsNoBits;
Hal Finkel3bae2d82015-10-12 20:51:48 +0000450
George Rimare3336c02015-11-24 10:15:50 +0000451 // We place RelRo section before plain r/w ones.
452 bool AIsRelRo = isRelroSection(A);
453 bool BIsRelRo = isRelroSection(B);
454 if (AIsRelRo != BIsRelRo)
455 return AIsRelRo;
456
Hal Finkel9abc2a52015-10-13 19:07:29 +0000457 // Some architectures have additional ordering restrictions for sections
458 // within the same PT_LOAD.
459 if (Config->EMachine == EM_PPC64)
460 return getPPC64SectionRank(A->getName()) <
461 getPPC64SectionRank(B->getName());
462
463 return false;
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000464}
465
Rafael Espindola443f50a2015-11-03 21:35:14 +0000466template <class ELFT> OutputSection<ELFT> *Writer<ELFT>::getBSS() {
467 if (!Out<ELFT>::Bss) {
Rui Ueyamad4ea7dd2015-12-26 07:01:26 +0000468 Out<ELFT>::Bss =
469 new OutputSection<ELFT>(".bss", SHT_NOBITS, SHF_ALLOC | SHF_WRITE);
470 OwningSections.emplace_back(Out<ELFT>::Bss);
Rafael Espindola443f50a2015-11-03 21:35:14 +0000471 OutputSections.push_back(Out<ELFT>::Bss);
472 }
473 return Out<ELFT>::Bss;
474}
475
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000476// Until this function is called, common symbols do not belong to any section.
477// This function adds them to end of BSS section.
478template <class ELFT>
Rafael Espindola11191912015-12-24 16:23:37 +0000479void Writer<ELFT>::addCommonSymbols(std::vector<DefinedCommon *> &Syms) {
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000480 typedef typename ELFFile<ELFT>::uintX_t uintX_t;
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000481
Rafael Espindola443f50a2015-11-03 21:35:14 +0000482 if (Syms.empty())
483 return;
484
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000485 // Sort the common symbols by alignment as an heuristic to pack them better.
Rafael Espindola11191912015-12-24 16:23:37 +0000486 std::stable_sort(Syms.begin(), Syms.end(),
487 [](const DefinedCommon *A, const DefinedCommon *B) {
488 return A->MaxAlignment > B->MaxAlignment;
489 });
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000490
Rafael Espindola443f50a2015-11-03 21:35:14 +0000491 uintX_t Off = getBSS()->getSize();
Rafael Espindola11191912015-12-24 16:23:37 +0000492 for (DefinedCommon *C : Syms) {
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000493 uintX_t Align = C->MaxAlignment;
494 Off = RoundUpToAlignment(Off, Align);
495 C->OffsetInBSS = Off;
Rafael Espindola11191912015-12-24 16:23:37 +0000496 Off += C->Size;
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000497 }
498
499 Out<ELFT>::Bss->setSize(Off);
500}
501
George Rimarbc590fe2015-10-28 16:48:58 +0000502template <class ELFT>
Rafael Espindola443f50a2015-11-03 21:35:14 +0000503void Writer<ELFT>::addSharedCopySymbols(
504 std::vector<SharedSymbol<ELFT> *> &Syms) {
George Rimarbc590fe2015-10-28 16:48:58 +0000505 typedef typename ELFFile<ELFT>::uintX_t uintX_t;
506 typedef typename ELFFile<ELFT>::Elf_Sym Elf_Sym;
Rafael Espindola115f0f32015-11-03 14:13:40 +0000507 typedef typename ELFFile<ELFT>::Elf_Shdr Elf_Shdr;
George Rimarbc590fe2015-10-28 16:48:58 +0000508
Rafael Espindola443f50a2015-11-03 21:35:14 +0000509 if (Syms.empty())
510 return;
511
512 uintX_t Off = getBSS()->getSize();
George Rimarbc590fe2015-10-28 16:48:58 +0000513 for (SharedSymbol<ELFT> *C : Syms) {
514 const Elf_Sym &Sym = C->Sym;
Rafael Espindola115f0f32015-11-03 14:13:40 +0000515 const Elf_Shdr *Sec = C->File->getSection(Sym);
516 uintX_t SecAlign = Sec->sh_addralign;
Rafael Espindola69d1ca6a2015-12-10 22:53:24 +0000517 unsigned TrailingZeros =
518 std::min(countTrailingZeros(SecAlign),
519 countTrailingZeros((uintX_t)Sym.st_value));
520 uintX_t Align = 1 << TrailingZeros;
Rafael Espindola115f0f32015-11-03 14:13:40 +0000521 Out<ELFT>::Bss->updateAlign(Align);
522 Off = RoundUpToAlignment(Off, Align);
George Rimarbc590fe2015-10-28 16:48:58 +0000523 C->OffsetInBSS = Off;
524 Off += Sym.st_size;
525 }
526 Out<ELFT>::Bss->setSize(Off);
527}
528
Denis Protivensky8e3b38a2015-11-12 09:52:08 +0000529template <class ELFT>
530StringRef Writer<ELFT>::getOutputSectionName(StringRef S) const {
531 auto It = InputToOutputSection.find(S);
532 if (It != std::end(InputToOutputSection))
533 return It->second;
534
Rui Ueyama4fcadaf2015-10-14 19:21:25 +0000535 if (S.startswith(".text."))
536 return ".text";
537 if (S.startswith(".rodata."))
538 return ".rodata";
George Rimar51e37d92015-11-13 07:56:27 +0000539 if (S.startswith(".data.rel.ro"))
540 return ".data.rel.ro";
Rui Ueyama4fcadaf2015-10-14 19:21:25 +0000541 if (S.startswith(".data."))
542 return ".data";
543 if (S.startswith(".bss."))
544 return ".bss";
545 return S;
546}
547
Denis Protivensky8e3b38a2015-11-12 09:52:08 +0000548template <class ELFT>
George Rimara5fbebc2015-12-10 09:12:18 +0000549void reportDiscarded(InputSectionBase<ELFT> *IS,
550 const std::unique_ptr<ObjectFile<ELFT>> &File) {
551 if (!Config->PrintGcSections || !IS || IS->isLive())
552 return;
553 llvm::errs() << "removing unused section from '" << IS->getSectionName()
554 << "' in file '" << File->getName() << "'\n";
555}
556
557template <class ELFT>
Denis Protivensky8e3b38a2015-11-12 09:52:08 +0000558bool Writer<ELFT>::isDiscarded(InputSectionBase<ELFT> *IS) const {
559 if (!IS || !IS->isLive() || IS == &InputSection<ELFT>::Discarded)
560 return true;
561 return InputToOutputSection.lookup(IS->getSectionName()) == "/DISCARD/";
562}
563
564template <class ELFT>
565static bool compareSections(OutputSectionBase<ELFT> *A,
566 OutputSectionBase<ELFT> *B) {
567 auto ItA = Config->OutputSections.find(A->getName());
568 auto ItEnd = std::end(Config->OutputSections);
569 if (ItA == ItEnd)
570 return compareOutputSections(A, B);
571 auto ItB = Config->OutputSections.find(B->getName());
572 if (ItB == ItEnd)
573 return compareOutputSections(A, B);
574
575 return std::distance(ItA, ItB) > 0;
576}
577
George Rimara07ff662015-12-21 10:12:06 +0000578// A statically linked executable will have rel[a].plt section
579// to hold R_[*]_IRELATIVE relocations.
580// The multi-arch libc will use these symbols to locate
581// these relocations at program startup time.
582// If RelaPlt is empty then there is no reason to create this symbols.
583template <class ELFT>
584static void addIRelocMarkers(SymbolTable<ELFT> &Symtab, bool IsDynamic) {
585 if (IsDynamic || !Out<ELFT>::RelaPlt || !Out<ELFT>::RelaPlt->hasRelocs())
586 return;
587 bool IsRela = shouldUseRela<ELFT>();
588 auto AddMarker = [&](StringRef Name, typename Writer<ELFT>::Elf_Sym &Sym) {
589 if (SymbolBody *B = Symtab.find(Name))
590 if (B->isUndefined())
591 Symtab.addAbsolute(Name, Sym);
592 };
593 AddMarker(IsRela ? "__rela_iplt_start" : "__rel_iplt_start",
Rui Ueyamaa246e0942015-12-25 06:12:18 +0000594 ElfSym<ELFT>::RelaIpltStart);
George Rimara07ff662015-12-21 10:12:06 +0000595 AddMarker(IsRela ? "__rela_iplt_end" : "__rel_iplt_end",
Rui Ueyamaa246e0942015-12-25 06:12:18 +0000596 ElfSym<ELFT>::RelaIpltEnd);
George Rimara07ff662015-12-21 10:12:06 +0000597}
598
Rafael Espindolaae533242015-12-23 20:37:51 +0000599template <class ELFT> static bool includeInSymtab(const SymbolBody &B) {
600 if (!B.isUsedInRegularObj())
601 return false;
602
603 // Don't include synthetic symbols like __init_array_start in every output.
Rafael Espindola02ce26a2015-12-24 14:22:24 +0000604 if (auto *U = dyn_cast<DefinedRegular<ELFT>>(&B))
Rui Ueyamaa246e0942015-12-25 06:12:18 +0000605 if (&U->Sym == &ElfSym<ELFT>::IgnoreUndef)
Rafael Espindolaae533242015-12-23 20:37:51 +0000606 return false;
607
608 return true;
609}
610
Rui Ueyama89f4ec72015-12-25 07:01:09 +0000611static bool includeInDynamicSymtab(const SymbolBody &B) {
612 uint8_t V = B.getVisibility();
613 if (V != STV_DEFAULT && V != STV_PROTECTED)
614 return false;
615 if (Config->ExportDynamic || Config->Shared)
616 return true;
617 return B.isUsedInDynamicReloc();
618}
619
Rui Ueyama2df0fd82015-12-25 07:38:58 +0000620template <class ELFT>
621OutputSectionBase<ELFT> *
622Writer<ELFT>::createOutputSection(InputSectionBase<ELFT> *C, StringRef Name,
623 uintX_t Type, uintX_t Flags) {
624 switch (C->SectionKind) {
625 case InputSectionBase<ELFT>::Regular:
Rui Ueyamad4ea7dd2015-12-26 07:01:26 +0000626 return new OutputSection<ELFT>(Name, Type, Flags);
Rui Ueyama2df0fd82015-12-25 07:38:58 +0000627 case InputSectionBase<ELFT>::EHFrame:
Rui Ueyamad4ea7dd2015-12-26 07:01:26 +0000628 return new EHOutputSection<ELFT>(Name, Type, Flags);
Rui Ueyama2df0fd82015-12-25 07:38:58 +0000629 case InputSectionBase<ELFT>::Merge:
Rui Ueyamad4ea7dd2015-12-26 07:01:26 +0000630 return new MergeOutputSection<ELFT>(Name, Type, Flags);
Rui Ueyama2df0fd82015-12-25 07:38:58 +0000631 case InputSectionBase<ELFT>::MipsReginfo:
Rui Ueyamad4ea7dd2015-12-26 07:01:26 +0000632 return new MipsReginfoOutputSection<ELFT>();
Rui Ueyama2df0fd82015-12-25 07:38:58 +0000633 }
George Rimara7afaca2015-12-25 11:15:26 +0000634 llvm_unreachable("Unknown output section type");
Rui Ueyama2df0fd82015-12-25 07:38:58 +0000635}
636
Michael J. Spencer84487f12015-07-24 21:03:07 +0000637// Create output section objects and add them to OutputSections.
638template <class ELFT> void Writer<ELFT>::createSections() {
Rui Ueyama69960ba2015-10-10 23:25:39 +0000639 // .interp needs to be on the first page in the output file.
640 if (needsInterpSection())
641 OutputSections.push_back(Out<ELFT>::Interp);
642
Rafael Espindolac159c962015-10-19 21:00:02 +0000643 SmallDenseMap<SectionKey<ELFT::Is64Bits>, OutputSectionBase<ELFT> *> Map;
Rafael Espindolac2d21192015-09-23 18:25:05 +0000644
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +0000645 std::vector<OutputSectionBase<ELFT> *> RegularSections;
Rui Ueyamad9189ce2015-10-15 17:11:03 +0000646
Rui Ueyama3ce825e2015-10-09 21:07:25 +0000647 for (const std::unique_ptr<ObjectFile<ELFT>> &F : Symtab.getObjectFiles()) {
Rafael Espindolac159c962015-10-19 21:00:02 +0000648 for (InputSectionBase<ELFT> *C : F->getSections()) {
George Rimara5fbebc2015-12-10 09:12:18 +0000649 if (isDiscarded(C)) {
650 reportDiscarded(C, F);
Rafael Espindola19e38892015-09-16 15:54:15 +0000651 continue;
George Rimara5fbebc2015-12-10 09:12:18 +0000652 }
Rafael Espindola19e38892015-09-16 15:54:15 +0000653 const Elf_Shdr *H = C->getSectionHdr();
Rafael Espindola444576d2015-10-09 19:25:07 +0000654 uintX_t OutFlags = H->sh_flags & ~SHF_GROUP;
Rafael Espindolac159c962015-10-19 21:00:02 +0000655 // For SHF_MERGE we create different output sections for each sh_entsize.
656 // This makes each output section simple and keeps a single level
657 // mapping from input to output.
Rui Ueyama2df0fd82015-12-25 07:38:58 +0000658 uintX_t EntSize = isa<MergeInputSection<ELFT>>(C) ? H->sh_entsize : 0;
Rafael Espindolaa016b1e2015-11-11 17:33:22 +0000659 uint32_t OutType = H->sh_type;
660 if (OutType == SHT_PROGBITS && C->getSectionName() == ".eh_frame" &&
661 Config->EMachine == EM_X86_64)
662 OutType = SHT_X86_64_UNWIND;
Denis Protivensky8e3b38a2015-11-12 09:52:08 +0000663 SectionKey<ELFT::Is64Bits> Key{getOutputSectionName(C->getSectionName()),
Rafael Espindolaa016b1e2015-11-11 17:33:22 +0000664 OutType, OutFlags, EntSize};
Rafael Espindolac159c962015-10-19 21:00:02 +0000665 OutputSectionBase<ELFT> *&Sec = Map[Key];
Rafael Espindolad13d9602015-09-25 15:08:44 +0000666 if (!Sec) {
Rui Ueyama2df0fd82015-12-25 07:38:58 +0000667 Sec = createOutputSection(C, Key.Name, Key.Type, Key.Flags);
Rui Ueyamad4ea7dd2015-12-26 07:01:26 +0000668 OwningSections.emplace_back(Sec);
Rafael Espindolad13d9602015-09-25 15:08:44 +0000669 OutputSections.push_back(Sec);
Rui Ueyamad9189ce2015-10-15 17:11:03 +0000670 RegularSections.push_back(Sec);
Rafael Espindolad13d9602015-09-25 15:08:44 +0000671 }
Rui Ueyama40845e62015-12-26 05:51:07 +0000672 Sec->addSection(C);
Rafael Espindola19e38892015-09-16 15:54:15 +0000673 }
674 }
675
Rafael Espindola443f50a2015-11-03 21:35:14 +0000676 Out<ELFT>::Bss = static_cast<OutputSection<ELFT> *>(
677 Map[{".bss", SHT_NOBITS, SHF_ALLOC | SHF_WRITE, 0}]);
678
Rafael Espindolac159c962015-10-19 21:00:02 +0000679 Out<ELFT>::Dynamic->PreInitArraySec = Map.lookup(
680 {".preinit_array", SHT_PREINIT_ARRAY, SHF_WRITE | SHF_ALLOC, 0});
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000681 Out<ELFT>::Dynamic->InitArraySec =
Rafael Espindolac159c962015-10-19 21:00:02 +0000682 Map.lookup({".init_array", SHT_INIT_ARRAY, SHF_WRITE | SHF_ALLOC, 0});
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000683 Out<ELFT>::Dynamic->FiniArraySec =
Rafael Espindolac159c962015-10-19 21:00:02 +0000684 Map.lookup({".fini_array", SHT_FINI_ARRAY, SHF_WRITE | SHF_ALLOC, 0});
Rafael Espindola77572242015-10-02 19:37:55 +0000685
Rui Ueyama0d0bcf72015-10-07 21:25:39 +0000686 auto AddStartEnd = [&](StringRef Start, StringRef End,
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +0000687 OutputSectionBase<ELFT> *OS) {
Michael J. Spencer95538ca2015-10-06 01:16:17 +0000688 if (OS) {
Rui Ueyamadd7d9982015-12-16 22:31:14 +0000689 Symtab.addSynthetic(Start, *OS, 0);
690 Symtab.addSynthetic(End, *OS, OS->getSize());
Rafael Espindola334c3e12015-10-19 15:21:42 +0000691 } else {
Rui Ueyamadd7d9982015-12-16 22:31:14 +0000692 Symtab.addIgnored(Start);
693 Symtab.addIgnored(End);
Michael J. Spencer95538ca2015-10-06 01:16:17 +0000694 }
695 };
696
Michael J. Spencerfcacad22015-10-06 19:57:05 +0000697 AddStartEnd("__preinit_array_start", "__preinit_array_end",
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000698 Out<ELFT>::Dynamic->PreInitArraySec);
Michael J. Spencerfcacad22015-10-06 19:57:05 +0000699 AddStartEnd("__init_array_start", "__init_array_end",
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000700 Out<ELFT>::Dynamic->InitArraySec);
Michael J. Spencerfcacad22015-10-06 19:57:05 +0000701 AddStartEnd("__fini_array_start", "__fini_array_end",
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000702 Out<ELFT>::Dynamic->FiniArraySec);
Rafael Espindola0e604f92015-09-25 18:56:53 +0000703
Rafael Espindola334c3e12015-10-19 15:21:42 +0000704 for (OutputSectionBase<ELFT> *Sec : RegularSections)
705 addStartStopSymbols(Sec);
706
707 // __tls_get_addr is defined by the dynamic linker for dynamic ELFs. For
708 // static linking the linker is required to optimize away any references to
709 // __tls_get_addr, so it's not defined anywhere. Create a hidden definition
710 // to avoid the undefined symbol error.
711 if (!isOutputDynamic())
Rui Ueyamadd7d9982015-12-16 22:31:14 +0000712 Symtab.addIgnored("__tls_get_addr");
Rafael Espindola334c3e12015-10-19 15:21:42 +0000713
Igor Kudrinb044af52015-11-20 02:32:35 +0000714 // If the "_end" symbol is referenced, it is expected to point to the address
715 // right after the data segment. Usually, this symbol points to the end
716 // of .bss section or to the end of .data section if .bss section is absent.
717 // The order of the sections can be affected by linker script,
718 // so it is hard to predict which section will be the last one.
719 // So, if this symbol is referenced, we just add the placeholder here
720 // and update its value later.
721 if (Symtab.find("_end"))
Rui Ueyamaa246e0942015-12-25 06:12:18 +0000722 Symtab.addAbsolute("_end", ElfSym<ELFT>::End);
Igor Kudrinb044af52015-11-20 02:32:35 +0000723
724 // If there is an undefined symbol "end", we should initialize it
725 // with the same value as "_end". In any other case it should stay intact,
726 // because it is an allowable name for a user symbol.
727 if (SymbolBody *B = Symtab.find("end"))
728 if (B->isUndefined())
Rui Ueyamaa246e0942015-12-25 06:12:18 +0000729 Symtab.addAbsolute("end", ElfSym<ELFT>::End);
Igor Kudrinb044af52015-11-20 02:32:35 +0000730
Rafael Espindola334c3e12015-10-19 15:21:42 +0000731 // Scan relocations. This must be done after every symbol is declared so that
732 // we can correctly decide if a dynamic relocation is needed.
Rafael Espindola0c6a4f12015-11-11 19:54:14 +0000733 for (const std::unique_ptr<ObjectFile<ELFT>> &F : Symtab.getObjectFiles()) {
Denis Protivensky8e3b38a2015-11-12 09:52:08 +0000734 for (InputSectionBase<ELFT> *C : F->getSections()) {
735 if (isDiscarded(C))
736 continue;
737 if (auto *S = dyn_cast<InputSection<ELFT>>(C))
738 scanRelocs(*S);
739 else if (auto *S = dyn_cast<EHInputSection<ELFT>>(C))
Rafael Espindola0c6a4f12015-11-11 19:54:14 +0000740 if (S->RelocSection)
741 scanRelocs(*S, *S->RelocSection);
Rafael Espindola0c6a4f12015-11-11 19:54:14 +0000742 }
743 }
Rafael Espindola334c3e12015-10-19 15:21:42 +0000744
George Rimara07ff662015-12-21 10:12:06 +0000745 addIRelocMarkers<ELFT>(Symtab, isOutputDynamic());
746
Rafael Espindola11191912015-12-24 16:23:37 +0000747 std::vector<DefinedCommon *> CommonSymbols;
George Rimarbc590fe2015-10-28 16:48:58 +0000748 std::vector<SharedSymbol<ELFT> *> SharedCopySymbols;
Rafael Espindola05a3dd22015-09-22 23:38:23 +0000749 for (auto &P : Symtab.getSymbols()) {
Rafael Espindola05a3dd22015-09-22 23:38:23 +0000750 SymbolBody *Body = P.second->Body;
Rafael Espindola5d7593b2015-12-22 23:00:50 +0000751 if (auto *U = dyn_cast<Undefined>(Body))
Denis Protivensky22220d52015-10-05 09:43:57 +0000752 if (!U->isWeak() && !U->canKeepUndefined())
753 reportUndefined<ELFT>(Symtab, *Body);
Rafael Espindola05a3dd22015-09-22 23:38:23 +0000754
Rafael Espindola11191912015-12-24 16:23:37 +0000755 if (auto *C = dyn_cast<DefinedCommon>(Body))
Rafael Espindola05a3dd22015-09-22 23:38:23 +0000756 CommonSymbols.push_back(C);
George Rimarbc590fe2015-10-28 16:48:58 +0000757 if (auto *SC = dyn_cast<SharedSymbol<ELFT>>(Body))
Rui Ueyamabb936062015-12-17 01:14:23 +0000758 if (SC->NeedsCopy)
George Rimarbc590fe2015-10-28 16:48:58 +0000759 SharedCopySymbols.push_back(SC);
760
Rafael Espindola4f674ed2015-10-05 15:24:04 +0000761 if (!includeInSymtab<ELFT>(*Body))
Rafael Espindola05a3dd22015-09-22 23:38:23 +0000762 continue;
George Rimar5dad7c12015-10-24 08:52:46 +0000763 if (Out<ELFT>::SymTab)
764 Out<ELFT>::SymTab->addSymbol(Body);
Rafael Espindola05a3dd22015-09-22 23:38:23 +0000765
Michael J. Spencer350e5b52015-10-12 23:39:23 +0000766 if (isOutputDynamic() && includeInDynamicSymtab(*Body))
Igor Kudrinab665fc2015-10-20 21:47:58 +0000767 Out<ELFT>::DynSymTab->addSymbol(Body);
Rafael Espindola05a3dd22015-09-22 23:38:23 +0000768 }
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000769 addCommonSymbols(CommonSymbols);
George Rimarbc590fe2015-10-28 16:48:58 +0000770 addSharedCopySymbols(SharedCopySymbols);
Michael J. Spencer84487f12015-07-24 21:03:07 +0000771
Rui Ueyamac6ef3f22015-10-15 21:50:30 +0000772 // This order is not the same as the final output order
773 // because we sort the sections using their attributes below.
George Rimar5dad7c12015-10-24 08:52:46 +0000774 if (Out<ELFT>::SymTab)
775 OutputSections.push_back(Out<ELFT>::SymTab);
George Rimar0f5ac9f2015-10-20 17:21:35 +0000776 OutputSections.push_back(Out<ELFT>::ShStrTab);
George Rimar5dad7c12015-10-24 08:52:46 +0000777 if (Out<ELFT>::StrTab)
778 OutputSections.push_back(Out<ELFT>::StrTab);
Michael J. Spencer350e5b52015-10-12 23:39:23 +0000779 if (isOutputDynamic()) {
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000780 OutputSections.push_back(Out<ELFT>::DynSymTab);
Igor Kudrin1b0d7062015-10-22 08:21:35 +0000781 if (Out<ELFT>::GnuHashTab)
782 OutputSections.push_back(Out<ELFT>::GnuHashTab);
783 if (Out<ELFT>::HashTab)
784 OutputSections.push_back(Out<ELFT>::HashTab);
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000785 OutputSections.push_back(Out<ELFT>::Dynamic);
786 OutputSections.push_back(Out<ELFT>::DynStrTab);
787 if (Out<ELFT>::RelaDyn->hasRelocs())
788 OutputSections.push_back(Out<ELFT>::RelaDyn);
Igor Kudrin304860a2015-11-12 04:39:49 +0000789 // This is a MIPS specific section to hold a space within the data segment
790 // of executable file which is pointed to by the DT_MIPS_RLD_MAP entry.
791 // See "Dynamic section" in Chapter 5 in the following document:
792 // ftp://www.linux-mips.org/pub/linux/mips/doc/ABI/mipsabi.pdf
793 if (Config->EMachine == EM_MIPS && !Config->Shared) {
Rui Ueyamad4ea7dd2015-12-26 07:01:26 +0000794 Out<ELFT>::MipsRldMap = new OutputSection<ELFT>(".rld_map", SHT_PROGBITS,
795 SHF_ALLOC | SHF_WRITE);
Igor Kudrin304860a2015-11-12 04:39:49 +0000796 Out<ELFT>::MipsRldMap->setSize(ELFT::Is64Bits ? 8 : 4);
797 Out<ELFT>::MipsRldMap->updateAlign(ELFT::Is64Bits ? 8 : 4);
Rui Ueyamad4ea7dd2015-12-26 07:01:26 +0000798 OwningSections.emplace_back(Out<ELFT>::MipsRldMap);
Igor Kudrin304860a2015-11-12 04:39:49 +0000799 OutputSections.push_back(Out<ELFT>::MipsRldMap);
800 }
Rafael Espindola3f4228f2015-09-09 15:33:08 +0000801 }
Igor Kudrin304860a2015-11-12 04:39:49 +0000802
George Rimara07ff662015-12-21 10:12:06 +0000803 // We always need to add rel[a].plt to output if it has entries.
804 // Even during static linking it can contain R_[*]_IRELATIVE relocations.
805 if (Out<ELFT>::RelaPlt && Out<ELFT>::RelaPlt->hasRelocs()) {
806 OutputSections.push_back(Out<ELFT>::RelaPlt);
807 Out<ELFT>::RelaPlt->Static = !isOutputDynamic();
808 }
809
George Rimarbfb7bf72015-12-21 10:00:12 +0000810 bool needsGot = !Out<ELFT>::Got->empty();
Igor Kudrin304860a2015-11-12 04:39:49 +0000811 // We add the .got section to the result for dynamic MIPS target because
812 // its address and properties are mentioned in the .dynamic section.
George Rimarbfb7bf72015-12-21 10:00:12 +0000813 if (Config->EMachine == EM_MIPS)
814 needsGot |= isOutputDynamic();
815 // If we have a relocation that is relative to GOT (such as GOTOFFREL),
816 // we need to emit a GOT even if it's empty.
817 if (HasGotOffRel)
818 needsGot = true;
819
820 if (needsGot)
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000821 OutputSections.push_back(Out<ELFT>::Got);
George Rimar648a2c32015-10-20 08:54:27 +0000822 if (Out<ELFT>::GotPlt && !Out<ELFT>::GotPlt->empty())
823 OutputSections.push_back(Out<ELFT>::GotPlt);
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000824 if (!Out<ELFT>::Plt->empty())
825 OutputSections.push_back(Out<ELFT>::Plt);
Rafael Espindola740fafe2015-09-08 19:43:27 +0000826
Rui Ueyama5a9640b2015-10-08 23:49:30 +0000827 std::stable_sort(OutputSections.begin(), OutputSections.end(),
Denis Protivensky8e3b38a2015-11-12 09:52:08 +0000828 compareSections<ELFT>);
Rui Ueyama3f4ec662015-09-25 03:48:25 +0000829
George Rimare3336c02015-11-24 10:15:50 +0000830 for (unsigned I = 0, N = OutputSections.size(); I < N; ++I) {
Rui Ueyama2317d0d2015-10-15 20:55:22 +0000831 OutputSections[I]->SectionIndex = I + 1;
George Rimare3336c02015-11-24 10:15:50 +0000832 HasRelro |= (Config->ZRelro && isRelroSection(OutputSections[I]));
833 }
Michael J. Spencer52bf0eb2015-10-01 21:15:02 +0000834
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +0000835 for (OutputSectionBase<ELFT> *Sec : OutputSections)
George Rimar0f5ac9f2015-10-20 17:21:35 +0000836 Out<ELFT>::ShStrTab->add(Sec->getName());
Rui Ueyamac6ef3f22015-10-15 21:50:30 +0000837
Igor Kudrin2169b1b2015-11-02 10:46:14 +0000838 // Finalizers fix each section's size.
839 // .dynamic section's finalizer may add strings to .dynstr,
840 // so finalize that early.
841 // Likewise, .dynsym is finalized early since that may fill up .gnu.hash.
Rui Ueyama15ef5e12015-10-07 19:18:16 +0000842 Out<ELFT>::Dynamic->finalize();
Igor Kudrin2169b1b2015-11-02 10:46:14 +0000843 if (isOutputDynamic())
844 Out<ELFT>::DynSymTab->finalize();
Rui Ueyamaaf311c12015-10-09 16:03:53 +0000845
Rui Ueyamac6ef3f22015-10-15 21:50:30 +0000846 // Fill other section headers.
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +0000847 for (OutputSectionBase<ELFT> *Sec : OutputSections)
Rui Ueyamaaf311c12015-10-09 16:03:53 +0000848 Sec->finalize();
Hal Finkeldaedc122015-10-12 23:16:53 +0000849
850 // If we have a .opd section (used under PPC64 for function descriptors),
851 // store a pointer to it here so that we can use it later when processing
852 // relocations.
Rafael Espindolac159c962015-10-19 21:00:02 +0000853 Out<ELFT>::Opd = Map.lookup({".opd", SHT_PROGBITS, SHF_WRITE | SHF_ALLOC, 0});
Rafael Espindolaabad6182015-08-13 15:23:46 +0000854}
855
Rui Ueyamad9189ce2015-10-15 17:11:03 +0000856static bool isAlpha(char C) {
857 return ('a' <= C && C <= 'z') || ('A' <= C && C <= 'Z') || C == '_';
858}
859
860static bool isAlnum(char C) { return isAlpha(C) || ('0' <= C && C <= '9'); }
861
862// Returns true if S is valid as a C language identifier.
863static bool isValidCIdentifier(StringRef S) {
864 if (S.empty() || !isAlpha(S[0]))
865 return false;
866 return std::all_of(S.begin() + 1, S.end(), isAlnum);
867}
868
869// If a section name is valid as a C identifier (which is rare because of
870// the leading '.'), linkers are expected to define __start_<secname> and
871// __stop_<secname> symbols. They are at beginning and end of the section,
872// respectively. This is not requested by the ELF standard, but GNU ld and
873// gold provide the feature, and used by many programs.
874template <class ELFT>
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +0000875void Writer<ELFT>::addStartStopSymbols(OutputSectionBase<ELFT> *Sec) {
Rui Ueyamad9189ce2015-10-15 17:11:03 +0000876 StringRef S = Sec->getName();
877 if (!isValidCIdentifier(S))
878 return;
879 StringSaver Saver(Alloc);
880 StringRef Start = Saver.save("__start_" + S);
881 StringRef Stop = Saver.save("__stop_" + S);
882 if (Symtab.isUndefined(Start))
Rui Ueyamadd7d9982015-12-16 22:31:14 +0000883 Symtab.addSynthetic(Start, *Sec, 0);
Rui Ueyamad9189ce2015-10-15 17:11:03 +0000884 if (Symtab.isUndefined(Stop))
Rui Ueyamadd7d9982015-12-16 22:31:14 +0000885 Symtab.addSynthetic(Stop, *Sec, Sec->getSize());
Rui Ueyamad9189ce2015-10-15 17:11:03 +0000886}
887
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +0000888template <class ELFT> static bool needsPhdr(OutputSectionBase<ELFT> *Sec) {
Michael J. Spencer2f008242015-09-17 19:58:07 +0000889 return Sec->getFlags() & SHF_ALLOC;
Michael J. Spencer1d299a82015-09-09 20:48:09 +0000890}
891
Rui Ueyama9eb7ed02015-10-24 18:22:59 +0000892static uint32_t toPhdrFlags(uint64_t Flags) {
893 uint32_t Ret = PF_R;
894 if (Flags & SHF_WRITE)
895 Ret |= PF_W;
896 if (Flags & SHF_EXECINSTR)
897 Ret |= PF_X;
898 return Ret;
899}
900
George Rimare3336c02015-11-24 10:15:50 +0000901template <class ELFT>
902void Writer<ELFT>::updateRelro(Elf_Phdr *Cur, Elf_Phdr *GnuRelroPhdr,
Rafael Espindola97cbe3e32015-12-23 15:20:38 +0000903 uintX_t VA) {
George Rimare3336c02015-11-24 10:15:50 +0000904 if (!GnuRelroPhdr->p_type)
905 setPhdr(GnuRelroPhdr, PT_GNU_RELRO, PF_R, Cur->p_offset, Cur->p_vaddr,
906 VA - Cur->p_vaddr, 1 /*p_align*/);
907 GnuRelroPhdr->p_filesz = VA - Cur->p_vaddr;
908 GnuRelroPhdr->p_memsz = VA - Cur->p_vaddr;
909}
910
Rui Ueyama803195e2015-10-23 21:45:59 +0000911// Visits all sections to create PHDRs and to assign incremental,
912// non-overlapping addresses to output sections.
Michael J. Spencer84487f12015-07-24 21:03:07 +0000913template <class ELFT> void Writer<ELFT>::assignAddresses() {
Igor Kudrinf6f45472015-11-10 08:39:27 +0000914 uintX_t VA = Target->getVAStart() + sizeof(Elf_Ehdr);
Rui Ueyama2f1b79f2015-10-10 22:34:30 +0000915 uintX_t FileOff = sizeof(Elf_Ehdr);
Rafael Espindola60252d82015-09-09 22:53:55 +0000916
Rui Ueyama803195e2015-10-23 21:45:59 +0000917 // Calculate and reserve the space for the program header first so that
918 // the first section can start right after the program header.
919 Phdrs.resize(getPhdrsNum());
920 size_t PhdrSize = sizeof(Elf_Phdr) * Phdrs.size();
Rui Ueyama3486fe52015-10-11 17:44:22 +0000921
Rui Ueyama803195e2015-10-23 21:45:59 +0000922 // The first phdr entry is PT_PHDR which describes the program header itself.
923 setPhdr(&Phdrs[0], PT_PHDR, PF_R, FileOff, VA, PhdrSize, /*Align=*/8);
924 FileOff += PhdrSize;
925 VA += PhdrSize;
Rui Ueyama953c2c42015-10-10 23:59:57 +0000926
Rui Ueyama803195e2015-10-23 21:45:59 +0000927 // PT_INTERP must be the second entry if exists.
928 int PhdrIdx = 0;
Rui Ueyama953c2c42015-10-10 23:59:57 +0000929 Elf_Phdr *Interp = nullptr;
Rui Ueyama803195e2015-10-23 21:45:59 +0000930 if (needsInterpSection())
931 Interp = &Phdrs[++PhdrIdx];
Rafael Espindola70107762015-09-11 18:49:42 +0000932
Rui Ueyama803195e2015-10-23 21:45:59 +0000933 // Add the first PT_LOAD segment for regular output sections.
Igor Kudrinf6f45472015-11-10 08:39:27 +0000934 setPhdr(&Phdrs[++PhdrIdx], PT_LOAD, PF_R, 0, Target->getVAStart(), FileOff,
Rui Ueyama803195e2015-10-23 21:45:59 +0000935 Target->getPageSize());
Rafael Espindola0a2e2112015-09-10 15:41:34 +0000936
George Rimare3336c02015-11-24 10:15:50 +0000937 Elf_Phdr GnuRelroPhdr = {};
Michael J. Spencer2c6f2132015-11-03 22:43:11 +0000938 Elf_Phdr TlsPhdr{};
Rafael Espindola97cbe3e32015-12-23 15:20:38 +0000939 bool RelroAligned = false;
Michael J. Spencer78aa1de2015-11-03 00:34:39 +0000940 uintX_t ThreadBSSOffset = 0;
Rui Ueyama803195e2015-10-23 21:45:59 +0000941 // Create phdrs as we assign VAs and file offsets to all output sections.
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +0000942 for (OutputSectionBase<ELFT> *Sec : OutputSections) {
Rafael Espindola97cbe3e32015-12-23 15:20:38 +0000943 Elf_Phdr *PH = &Phdrs[PhdrIdx];
Rafael Espindola5fbe0082015-11-03 23:19:42 +0000944 if (needsPhdr<ELFT>(Sec)) {
945 uintX_t Flags = toPhdrFlags(Sec->getFlags());
Rafael Espindola97cbe3e32015-12-23 15:20:38 +0000946 bool InRelRo = Config->ZRelro && (Flags & PF_W) && isRelroSection(Sec);
947 bool FirstNonRelRo = GnuRelroPhdr.p_type && !InRelRo && !RelroAligned;
948 if (FirstNonRelRo || PH->p_flags != Flags) {
Rafael Espindola61362ce2015-11-03 22:08:08 +0000949 VA = RoundUpToAlignment(VA, Target->getPageSize());
950 FileOff = RoundUpToAlignment(FileOff, Target->getPageSize());
Rafael Espindola97cbe3e32015-12-23 15:20:38 +0000951 if (FirstNonRelRo)
952 RelroAligned = true;
953 }
954
955 if (PH->p_flags != Flags) {
956 // Flags changed. Create a new PT_LOAD.
957 PH = &Phdrs[++PhdrIdx];
Rafael Espindola61362ce2015-11-03 22:08:08 +0000958 setPhdr(PH, PT_LOAD, Flags, FileOff, VA, 0, Target->getPageSize());
Michael J. Spencer2f008242015-09-17 19:58:07 +0000959 }
Michael J. Spencer1d299a82015-09-09 20:48:09 +0000960
Rafael Espindola5fbe0082015-11-03 23:19:42 +0000961 if (Sec->getFlags() & SHF_TLS) {
Rafael Espindolaea7a1e902015-11-06 22:14:44 +0000962 if (!TlsPhdr.p_vaddr)
Rafael Espindola5fbe0082015-11-03 23:19:42 +0000963 setPhdr(&TlsPhdr, PT_TLS, PF_R, FileOff, VA, 0, Sec->getAlign());
Rafael Espindola5fbe0082015-11-03 23:19:42 +0000964 if (Sec->getType() != SHT_NOBITS)
965 VA = RoundUpToAlignment(VA, Sec->getAlign());
966 uintX_t TVA = RoundUpToAlignment(VA + ThreadBSSOffset, Sec->getAlign());
967 Sec->setVA(TVA);
968 TlsPhdr.p_memsz += Sec->getSize();
969 if (Sec->getType() == SHT_NOBITS) {
970 ThreadBSSOffset = TVA - VA + Sec->getSize();
971 } else {
972 TlsPhdr.p_filesz += Sec->getSize();
973 VA += Sec->getSize();
974 }
975 TlsPhdr.p_align = std::max<uintX_t>(TlsPhdr.p_align, Sec->getAlign());
Rafael Espindola9eb3e572015-11-03 15:46:15 +0000976 } else {
Rafael Espindola5fbe0082015-11-03 23:19:42 +0000977 VA = RoundUpToAlignment(VA, Sec->getAlign());
978 Sec->setVA(VA);
Michael J. Spencer78aa1de2015-11-03 00:34:39 +0000979 VA += Sec->getSize();
Rafael Espindola97cbe3e32015-12-23 15:20:38 +0000980 if (InRelRo)
981 updateRelro(PH, &GnuRelroPhdr, VA);
Michael J. Spencer78aa1de2015-11-03 00:34:39 +0000982 }
Rafael Espindolaef1ac012015-08-13 15:31:17 +0000983 }
Michael J. Spencer78aa1de2015-11-03 00:34:39 +0000984
Rui Ueyama803195e2015-10-23 21:45:59 +0000985 FileOff = RoundUpToAlignment(FileOff, Sec->getAlign());
Michael J. Spencer84487f12015-07-24 21:03:07 +0000986 Sec->setFileOffset(FileOff);
Rafael Espindola058f3432015-08-31 20:23:57 +0000987 if (Sec->getType() != SHT_NOBITS)
Rui Ueyama803195e2015-10-23 21:45:59 +0000988 FileOff += Sec->getSize();
Rafael Espindola5fbe0082015-11-03 23:19:42 +0000989 if (needsPhdr<ELFT>(Sec)) {
Rafael Espindola97cbe3e32015-12-23 15:20:38 +0000990 PH->p_filesz = FileOff - PH->p_offset;
991 PH->p_memsz = VA - PH->p_vaddr;
Rafael Espindola5fbe0082015-11-03 23:19:42 +0000992 }
Michael J. Spencer84487f12015-07-24 21:03:07 +0000993 }
Rafael Espindola6b83b902015-08-12 00:00:24 +0000994
Michael J. Spencer6c34eff2015-11-05 02:00:35 +0000995 if (TlsPhdr.p_vaddr) {
Rafael Espindolaea7a1e902015-11-06 22:14:44 +0000996 // The TLS pointer goes after PT_TLS. At least glibc will align it,
997 // so round up the size to make sure the offsets are correct.
998 TlsPhdr.p_memsz = RoundUpToAlignment(TlsPhdr.p_memsz, TlsPhdr.p_align);
Michael J. Spencer78aa1de2015-11-03 00:34:39 +0000999 Phdrs[++PhdrIdx] = TlsPhdr;
Rafael Espindolaea7a1e902015-11-06 22:14:44 +00001000 Out<ELFT>::TlsPhdr = &Phdrs[PhdrIdx];
Michael J. Spencer6c34eff2015-11-05 02:00:35 +00001001 }
Michael J. Spencer78aa1de2015-11-03 00:34:39 +00001002
Rui Ueyama803195e2015-10-23 21:45:59 +00001003 // Add an entry for .dynamic.
Michael J. Spencer350e5b52015-10-12 23:39:23 +00001004 if (isOutputDynamic()) {
Rui Ueyama803195e2015-10-23 21:45:59 +00001005 Elf_Phdr *PH = &Phdrs[++PhdrIdx];
Rui Ueyama953c2c42015-10-10 23:59:57 +00001006 PH->p_type = PT_DYNAMIC;
Rui Ueyama536c9992015-10-11 19:45:58 +00001007 copyPhdr(PH, Out<ELFT>::Dynamic);
Rui Ueyama2f1b79f2015-10-10 22:34:30 +00001008 }
Rafael Espindola91009b32015-08-12 01:45:28 +00001009
George Rimare3336c02015-11-24 10:15:50 +00001010 if (HasRelro) {
1011 Elf_Phdr *PH = &Phdrs[++PhdrIdx];
1012 *PH = GnuRelroPhdr;
1013 }
1014
Rui Ueyamae79b09a2015-11-21 22:19:32 +00001015 // PT_GNU_STACK is a special section to tell the loader to make the
1016 // pages for the stack non-executable.
Rui Ueyamaa46566f2015-11-24 23:42:33 +00001017 if (!Config->ZExecStack) {
1018 Elf_Phdr *PH = &Phdrs[++PhdrIdx];
1019 PH->p_type = PT_GNU_STACK;
Rui Ueyamae79b09a2015-11-21 22:19:32 +00001020 PH->p_flags = PF_R | PF_W;
Rui Ueyamaa46566f2015-11-24 23:42:33 +00001021 }
George Rimar03220302015-11-14 20:56:08 +00001022
Rui Ueyama803195e2015-10-23 21:45:59 +00001023 // Fix up PT_INTERP as we now know the address of .interp section.
1024 if (Interp) {
1025 Interp->p_type = PT_INTERP;
1026 copyPhdr(Interp, Out<ELFT>::Interp);
Rui Ueyama69960ba2015-10-10 23:25:39 +00001027 }
1028
Michael J. Spencer8039dae22015-07-29 00:30:10 +00001029 // Add space for section headers.
Rui Ueyama803195e2015-10-23 21:45:59 +00001030 SectionHeaderOff = RoundUpToAlignment(FileOff, ELFT::Is64Bits ? 8 : 4);
1031 FileSize = SectionHeaderOff + getNumSections() * sizeof(Elf_Shdr);
Igor Kudrin15cd9ff2015-11-06 07:43:03 +00001032
Igor Kudrinb044af52015-11-20 02:32:35 +00001033 // Update "_end" and "end" symbols so that they
1034 // point to the end of the data segment.
Rui Ueyamaa246e0942015-12-25 06:12:18 +00001035 ElfSym<ELFT>::End.st_value = VA;
Igor Kudrinb044af52015-11-20 02:32:35 +00001036
George Rimara07ff662015-12-21 10:12:06 +00001037 // Update __rel_iplt_start/__rel_iplt_end to wrap the
1038 // rela.plt section.
1039 if (Out<ELFT>::RelaPlt) {
1040 uintX_t Start = Out<ELFT>::RelaPlt->getVA();
Rui Ueyamaa246e0942015-12-25 06:12:18 +00001041 ElfSym<ELFT>::RelaIpltStart.st_value = Start;
1042 ElfSym<ELFT>::RelaIpltEnd.st_value = Start + Out<ELFT>::RelaPlt->getSize();
George Rimara07ff662015-12-21 10:12:06 +00001043 }
1044
Igor Kudrin15cd9ff2015-11-06 07:43:03 +00001045 // Update MIPS _gp absolute symbol so that it points to the static data.
1046 if (Config->EMachine == EM_MIPS)
Rui Ueyamaa246e0942015-12-25 06:12:18 +00001047 ElfSym<ELFT>::MipsGp.st_value = getMipsGpAddr<ELFT>();
Rui Ueyama803195e2015-10-23 21:45:59 +00001048}
1049
1050// Returns the number of PHDR entries.
1051template <class ELFT> int Writer<ELFT>::getPhdrsNum() const {
Michael J. Spencer78aa1de2015-11-03 00:34:39 +00001052 bool Tls = false;
Rui Ueyamaa46566f2015-11-24 23:42:33 +00001053 int I = 2; // 2 for PT_PHDR and first PT_LOAD
Rui Ueyama803195e2015-10-23 21:45:59 +00001054 if (needsInterpSection())
1055 ++I;
1056 if (isOutputDynamic())
1057 ++I;
Rui Ueyamaa46566f2015-11-24 23:42:33 +00001058 if (!Config->ZExecStack)
1059 ++I;
Rui Ueyama803195e2015-10-23 21:45:59 +00001060 uintX_t Last = PF_R;
1061 for (OutputSectionBase<ELFT> *Sec : OutputSections) {
Rafael Espindola61362ce2015-11-03 22:08:08 +00001062 if (!needsPhdr<ELFT>(Sec))
Rui Ueyama803195e2015-10-23 21:45:59 +00001063 continue;
Michael J. Spencer78aa1de2015-11-03 00:34:39 +00001064 if (Sec->getFlags() & SHF_TLS)
1065 Tls = true;
Rui Ueyama803195e2015-10-23 21:45:59 +00001066 uintX_t Flags = toPhdrFlags(Sec->getFlags());
1067 if (Last != Flags) {
1068 Last = Flags;
1069 ++I;
1070 }
1071 }
Michael J. Spencer78aa1de2015-11-03 00:34:39 +00001072 if (Tls)
1073 ++I;
George Rimare3336c02015-11-24 10:15:50 +00001074 if (HasRelro)
1075 ++I;
Rui Ueyama803195e2015-10-23 21:45:59 +00001076 return I;
Michael J. Spencer84487f12015-07-24 21:03:07 +00001077}
1078
Simon Atanasyan034c4cd2015-12-19 05:51:49 +00001079static uint32_t getELFFlags() {
1080 if (Config->EMachine != EM_MIPS)
1081 return 0;
1082 // FIXME: In fact ELF flags depends on ELF flags of input object files
1083 // and selected emulation. For now just use hadr coded values.
1084 uint32_t V = EF_MIPS_ABI_O32 | EF_MIPS_CPIC | EF_MIPS_ARCH_32R2;
1085 if (Config->Shared)
1086 V |= EF_MIPS_PIC;
1087 return V;
1088}
1089
Rui Ueyama3bfaba92015-12-24 08:37:34 +00001090template <class ELFT>
1091static typename ELFFile<ELFT>::uintX_t getEntryAddr() {
1092 if (Config->EntrySym) {
1093 if (SymbolBody *E = Config->EntrySym->repl())
1094 return getSymVA<ELFT>(*E);
1095 return 0;
1096 }
1097 if (Config->EntryAddr != uint64_t(-1))
1098 return Config->EntryAddr;
1099 return 0;
1100}
1101
Michael J. Spencer84487f12015-07-24 21:03:07 +00001102template <class ELFT> void Writer<ELFT>::writeHeader() {
1103 uint8_t *Buf = Buffer->getBufferStart();
Rui Ueyamae08cd672015-10-23 22:44:39 +00001104 memcpy(Buf, "\177ELF", 4);
1105
Rui Ueyama6621d8e2015-10-24 17:57:40 +00001106 // Write the ELF header.
Rafael Espindola18608a02015-09-08 21:57:31 +00001107 auto *EHdr = reinterpret_cast<Elf_Ehdr *>(Buf);
Rafael Espindola4b7c2fc2015-08-05 15:08:40 +00001108 EHdr->e_ident[EI_CLASS] = ELFT::Is64Bits ? ELFCLASS64 : ELFCLASS32;
1109 EHdr->e_ident[EI_DATA] = ELFT::TargetEndianness == llvm::support::little
1110 ? ELFDATA2LSB
1111 : ELFDATA2MSB;
Michael J. Spencer84487f12015-07-24 21:03:07 +00001112 EHdr->e_ident[EI_VERSION] = EV_CURRENT;
Davide Italianoaa7c5332015-09-25 01:59:13 +00001113
Rafael Espindolab9ca7bb2015-10-12 11:52:31 +00001114 auto &FirstObj = cast<ELFFileBase<ELFT>>(*Config->FirstElf);
Davide Italianoaa7c5332015-09-25 01:59:13 +00001115 EHdr->e_ident[EI_OSABI] = FirstObj.getOSABI();
Michael J. Spencer84487f12015-07-24 21:03:07 +00001116
Rafael Espindola4340aad2015-09-11 22:42:45 +00001117 EHdr->e_type = Config->Shared ? ET_DYN : ET_EXEC;
Rafael Espindolaf98d6d82015-09-03 20:03:54 +00001118 EHdr->e_machine = FirstObj.getEMachine();
Michael J. Spencer84487f12015-07-24 21:03:07 +00001119 EHdr->e_version = EV_CURRENT;
Rui Ueyama3bfaba92015-12-24 08:37:34 +00001120 EHdr->e_entry = getEntryAddr<ELFT>();
Rui Ueyama5bfd7d42015-10-10 22:36:36 +00001121 EHdr->e_phoff = sizeof(Elf_Ehdr);
Michael J. Spencer8039dae22015-07-29 00:30:10 +00001122 EHdr->e_shoff = SectionHeaderOff;
Simon Atanasyan034c4cd2015-12-19 05:51:49 +00001123 EHdr->e_flags = getELFFlags();
Rafael Espindola18608a02015-09-08 21:57:31 +00001124 EHdr->e_ehsize = sizeof(Elf_Ehdr);
1125 EHdr->e_phentsize = sizeof(Elf_Phdr);
Rui Ueyama2f1b79f2015-10-10 22:34:30 +00001126 EHdr->e_phnum = Phdrs.size();
Rafael Espindola18608a02015-09-08 21:57:31 +00001127 EHdr->e_shentsize = sizeof(Elf_Shdr);
Rafael Espindola5f553872015-09-08 17:39:39 +00001128 EHdr->e_shnum = getNumSections();
George Rimar0f5ac9f2015-10-20 17:21:35 +00001129 EHdr->e_shstrndx = Out<ELFT>::ShStrTab->SectionIndex;
Rui Ueyama6621d8e2015-10-24 17:57:40 +00001130
1131 // Write the program header table.
Rui Ueyamabdca0b12015-10-11 00:10:36 +00001132 memcpy(Buf + EHdr->e_phoff, &Phdrs[0], Phdrs.size() * sizeof(Phdrs[0]));
Rafael Espindolae438e072015-09-08 22:55:28 +00001133
Rui Ueyama6621d8e2015-10-24 17:57:40 +00001134 // Write the section header table. Note that the first table entry is null.
Rafael Espindola18608a02015-09-08 21:57:31 +00001135 auto SHdrs = reinterpret_cast<Elf_Shdr *>(Buf + EHdr->e_shoff);
Rui Ueyama157c4332015-10-24 17:57:39 +00001136 for (OutputSectionBase<ELFT> *Sec : OutputSections)
Rui Ueyama6621d8e2015-10-24 17:57:40 +00001137 Sec->writeHeaderTo(++SHdrs);
Michael J. Spencer84487f12015-07-24 21:03:07 +00001138}
1139
1140template <class ELFT> void Writer<ELFT>::openFile(StringRef Path) {
Rafael Espindolabdc8f2f2015-08-13 00:31:46 +00001141 ErrorOr<std::unique_ptr<FileOutputBuffer>> BufferOrErr =
1142 FileOutputBuffer::create(Path, FileSize, FileOutputBuffer::F_executable);
Rui Ueyama3d312a62015-12-24 08:41:55 +00001143 error(BufferOrErr, "failed to open " + Path);
Rafael Espindolabdc8f2f2015-08-13 00:31:46 +00001144 Buffer = std::move(*BufferOrErr);
Michael J. Spencer84487f12015-07-24 21:03:07 +00001145}
1146
1147// Write section contents to a mmap'ed file.
1148template <class ELFT> void Writer<ELFT>::writeSections() {
1149 uint8_t *Buf = Buffer->getBufferStart();
Hal Finkeldaedc122015-10-12 23:16:53 +00001150
1151 // PPC64 needs to process relocations in the .opd section before processing
1152 // relocations in code-containing sections.
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +00001153 if (OutputSectionBase<ELFT> *Sec = Out<ELFT>::Opd) {
Rafael Espindola7a513052015-10-13 14:45:51 +00001154 Out<ELFT>::OpdBuf = Buf + Sec->getFileOff();
1155 Sec->writeTo(Buf + Sec->getFileOff());
1156 }
Hal Finkeldaedc122015-10-12 23:16:53 +00001157
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +00001158 for (OutputSectionBase<ELFT> *Sec : OutputSections)
Rafael Espindola7a513052015-10-13 14:45:51 +00001159 if (Sec != Out<ELFT>::Opd)
Hal Finkeldaedc122015-10-12 23:16:53 +00001160 Sec->writeTo(Buf + Sec->getFileOff());
Michael J. Spencer84487f12015-07-24 21:03:07 +00001161}
Rui Ueyama3ce825e2015-10-09 21:07:25 +00001162
Rui Ueyama2f1b79f2015-10-10 22:34:30 +00001163template <class ELFT>
Rui Ueyama536c9992015-10-11 19:45:58 +00001164void Writer<ELFT>::setPhdr(Elf_Phdr *PH, uint32_t Type, uint32_t Flags,
Rui Ueyama803195e2015-10-23 21:45:59 +00001165 uintX_t FileOff, uintX_t VA, uintX_t Size,
1166 uintX_t Align) {
Rui Ueyama2f1b79f2015-10-10 22:34:30 +00001167 PH->p_type = Type;
1168 PH->p_flags = Flags;
1169 PH->p_offset = FileOff;
1170 PH->p_vaddr = VA;
1171 PH->p_paddr = VA;
Rui Ueyama803195e2015-10-23 21:45:59 +00001172 PH->p_filesz = Size;
1173 PH->p_memsz = Size;
Rui Ueyama2f1b79f2015-10-10 22:34:30 +00001174 PH->p_align = Align;
1175}
1176
1177template <class ELFT>
Rui Ueyamac7cc6ec2015-10-15 22:27:29 +00001178void Writer<ELFT>::copyPhdr(Elf_Phdr *PH, OutputSectionBase<ELFT> *From) {
Rui Ueyama2f1b79f2015-10-10 22:34:30 +00001179 PH->p_flags = toPhdrFlags(From->getFlags());
1180 PH->p_offset = From->getFileOff();
1181 PH->p_vaddr = From->getVA();
1182 PH->p_paddr = From->getVA();
1183 PH->p_filesz = From->getSize();
1184 PH->p_memsz = From->getSize();
1185 PH->p_align = From->getAlign();
1186}
1187
Denis Protivensky8e3b38a2015-11-12 09:52:08 +00001188template <class ELFT> void Writer<ELFT>::buildSectionMap() {
1189 for (const std::pair<StringRef, std::vector<StringRef>> &OutSec :
1190 Config->OutputSections)
1191 for (StringRef Name : OutSec.second)
1192 InputToOutputSection[Name] = OutSec.first;
1193}
1194
Rui Ueyama3ce825e2015-10-09 21:07:25 +00001195template void lld::elf2::writeResult<ELF32LE>(SymbolTable<ELF32LE> *Symtab);
1196template void lld::elf2::writeResult<ELF32BE>(SymbolTable<ELF32BE> *Symtab);
1197template void lld::elf2::writeResult<ELF64LE>(SymbolTable<ELF64LE> *Symtab);
1198template void lld::elf2::writeResult<ELF64BE>(SymbolTable<ELF64BE> *Symtab);