blob: 72776e0d6dc39909c9a78b46fe04d1ed2f28f20f [file] [log] [blame]
Rafael Espindola5805c4f2015-09-21 21:38:08 +00001//===- OutputSections.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
10#include "OutputSections.h"
11#include "Config.h"
Rafael Espindola5805c4f2015-09-21 21:38:08 +000012#include "SymbolTable.h"
Rafael Espindola01205f72015-09-22 18:19:46 +000013#include "Target.h"
Rafael Espindola5805c4f2015-09-21 21:38:08 +000014
Rafael Espindola5805c4f2015-09-21 21:38:08 +000015using namespace llvm;
16using namespace llvm::object;
17using namespace llvm::ELF;
18
19using namespace lld;
20using namespace lld::elf2;
21
22template <bool Is64Bits>
23OutputSectionBase<Is64Bits>::OutputSectionBase(StringRef Name, uint32_t sh_type,
24 uintX_t sh_flags)
25 : Name(Name) {
26 memset(&Header, 0, sizeof(HeaderT));
27 Header.sh_type = sh_type;
28 Header.sh_flags = sh_flags;
29}
30
Rafael Espindola35c6af32015-09-25 17:19:10 +000031template <class ELFT>
32GotSection<ELFT>::GotSection()
33 : OutputSectionBase<ELFT::Is64Bits>(".got", llvm::ELF::SHT_PROGBITS,
34 llvm::ELF::SHF_ALLOC |
35 llvm::ELF::SHF_WRITE) {
36 this->Header.sh_addralign = this->getAddrSize();
37}
38
Rafael Espindola5805c4f2015-09-21 21:38:08 +000039template <class ELFT> void GotSection<ELFT>::addEntry(SymbolBody *Sym) {
40 Sym->setGotIndex(Entries.size());
41 Entries.push_back(Sym);
42}
43
44template <class ELFT>
45typename GotSection<ELFT>::uintX_t
46GotSection<ELFT>::getEntryAddr(const SymbolBody &B) const {
47 return this->getVA() + B.getGotIndex() * this->getAddrSize();
48}
49
Rafael Espindola35c6af32015-09-25 17:19:10 +000050template <class ELFT>
51PltSection<ELFT>::PltSection(const GotSection<ELFT> &GotSec)
52 : OutputSectionBase<ELFT::Is64Bits>(".plt", llvm::ELF::SHT_PROGBITS,
53 llvm::ELF::SHF_ALLOC |
54 llvm::ELF::SHF_EXECINSTR),
55 GotSec(GotSec) {
56 this->Header.sh_addralign = 16;
57}
58
Rafael Espindola5805c4f2015-09-21 21:38:08 +000059template <class ELFT> void PltSection<ELFT>::writeTo(uint8_t *Buf) {
60 uintptr_t Start = reinterpret_cast<uintptr_t>(Buf);
Rafael Espindola5805c4f2015-09-21 21:38:08 +000061 for (const SymbolBody *E : Entries) {
Rafael Espindola01205f72015-09-22 18:19:46 +000062 uint64_t GotEntryAddr = GotSec.getEntryAddr(*E);
Rafael Espindola5805c4f2015-09-21 21:38:08 +000063 uintptr_t InstPos = reinterpret_cast<uintptr_t>(Buf);
Rafael Espindola01205f72015-09-22 18:19:46 +000064 uint64_t PltEntryAddr = (InstPos - Start) + this->getVA();
65 Target->writePltEntry(Buf, GotEntryAddr, PltEntryAddr);
66 Buf += 8;
Rafael Espindola5805c4f2015-09-21 21:38:08 +000067 }
68}
69
70template <class ELFT> void PltSection<ELFT>::addEntry(SymbolBody *Sym) {
71 Sym->setPltIndex(Entries.size());
72 Entries.push_back(Sym);
73}
74
75template <class ELFT>
76typename PltSection<ELFT>::uintX_t
77PltSection<ELFT>::getEntryAddr(const SymbolBody &B) const {
78 return this->getVA() + B.getPltIndex() * EntrySize;
79}
80
Rafael Espindola35c6af32015-09-25 17:19:10 +000081template <class ELFT>
82RelocationSection<ELFT>::RelocationSection(SymbolTableSection<ELFT> &DynSymSec,
83 const GotSection<ELFT> &GotSec,
Rafael Espindola9c3e4d22015-10-05 21:23:08 +000084 const OutputSection<ELFT> &BssSec,
Rafael Espindola35c6af32015-09-25 17:19:10 +000085 bool IsRela)
86 : OutputSectionBase<ELFT::Is64Bits>(IsRela ? ".rela.dyn" : ".rel.dyn",
87 IsRela ? llvm::ELF::SHT_RELA
88 : llvm::ELF::SHT_REL,
89 llvm::ELF::SHF_ALLOC),
Rafael Espindola9c3e4d22015-10-05 21:23:08 +000090 DynSymSec(DynSymSec), GotSec(GotSec), BssSec(BssSec), IsRela(IsRela) {
Rafael Espindola35c6af32015-09-25 17:19:10 +000091 this->Header.sh_entsize = IsRela ? sizeof(Elf_Rela) : sizeof(Elf_Rel);
92 this->Header.sh_addralign = ELFT::Is64Bits ? 8 : 4;
93}
94
Rafael Espindola03ab3362015-10-06 12:39:58 +000095static bool isLocalDefinition(const SymbolBody *Body) {
96 if (!Body)
97 return true;
98 return !Body->isShared() && !Body->isUndefined();
99}
100
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000101template <class ELFT> void RelocationSection<ELFT>::writeTo(uint8_t *Buf) {
Rafael Espindola50534c22015-09-22 17:49:38 +0000102 const unsigned EntrySize = IsRela ? sizeof(Elf_Rela) : sizeof(Elf_Rel);
Rafael Espindolae1901cc2015-09-24 15:11:50 +0000103 bool IsMips64EL = Relocs[0].C.getFile()->getObj().isMips64EL();
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000104 for (const DynamicReloc<ELFT> &Rel : Relocs) {
Rafael Espindola50534c22015-09-22 17:49:38 +0000105 auto *P = reinterpret_cast<Elf_Rel *>(Buf);
106 Buf += EntrySize;
107
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000108 const InputSection<ELFT> &C = Rel.C;
109 const Elf_Rel &RI = Rel.RI;
110 OutputSection<ELFT> *Out = C.getOutputSection();
111 uint32_t SymIndex = RI.getSymbol(IsMips64EL);
Rafael Espindola41127ad2015-10-05 22:49:16 +0000112 const ObjectFile<ELFT> &File = *C.getFile();
113 const SymbolBody *Body = File.getSymbolBody(SymIndex);
114 const ELFFile<ELFT> &Obj = File.getObj();
115
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000116 uint32_t Type = RI.getType(IsMips64EL);
Rafael Espindolaae244002015-10-05 19:30:12 +0000117 if (Body && Target->relocNeedsGot(Type, *Body)) {
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000118 P->r_offset = GotSec.getEntryAddr(*Body);
Rafael Espindola7f074422015-09-22 21:35:51 +0000119 P->setSymbolAndType(Body->getDynamicSymbolTableIndex(),
120 Target->getGotReloc(), IsMips64EL);
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000121 } else {
122 P->r_offset = RI.r_offset + C.getOutputSectionOff() + Out->getVA();
Rafael Espindola3c83e2b2015-10-05 21:09:37 +0000123 uintX_t Addent = 0;
124 if (IsRela)
125 Addent = static_cast<const Elf_Rela &>(RI).r_addend;
126
Rafael Espindola03ab3362015-10-06 12:39:58 +0000127 if (!isLocalDefinition(Body)) {
Rafael Espindolaae244002015-10-05 19:30:12 +0000128 P->setSymbolAndType(Body->getDynamicSymbolTableIndex(), Type,
129 IsMips64EL);
Rafael Espindolaae244002015-10-05 19:30:12 +0000130 } else {
131 P->setSymbolAndType(0, Target->getRelativeReloc(), IsMips64EL);
Rafael Espindola3c83e2b2015-10-05 21:09:37 +0000132 if (IsRela) {
Rafael Espindola3c83e2b2015-10-05 21:09:37 +0000133 if (Body)
Rafael Espindola9c3e4d22015-10-05 21:23:08 +0000134 Addent += getSymVA(cast<ELFSymbolBody<ELFT>>(*Body), BssSec);
135 else
Rafael Espindola41127ad2015-10-05 22:49:16 +0000136 Addent += getLocalSymVA(
137 Obj.getRelocationSymbol(&RI, File.getSymbolTable()), File);
Rafael Espindola3c83e2b2015-10-05 21:09:37 +0000138 }
Rafael Espindolaae244002015-10-05 19:30:12 +0000139 }
Rafael Espindola3c83e2b2015-10-05 21:09:37 +0000140 if (IsRela)
141 static_cast<Elf_Rela *>(P)->r_addend = Addent;
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000142 }
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000143 }
144}
145
146template <class ELFT> void RelocationSection<ELFT>::finalize() {
147 this->Header.sh_link = DynSymSec.getSectionIndex();
148 this->Header.sh_size = Relocs.size() * this->Header.sh_entsize;
149}
150
151template <bool Is64Bits>
152InterpSection<Is64Bits>::InterpSection()
153 : OutputSectionBase<Is64Bits>(".interp", llvm::ELF::SHT_PROGBITS,
154 llvm::ELF::SHF_ALLOC) {
155 this->Header.sh_size = Config->DynamicLinker.size() + 1;
156 this->Header.sh_addralign = 1;
157}
158
159template <bool Is64Bits>
160template <endianness E>
161void OutputSectionBase<Is64Bits>::writeHeaderTo(
162 typename ELFFile<ELFType<E, Is64Bits>>::Elf_Shdr *SHdr) {
163 SHdr->sh_name = Header.sh_name;
164 SHdr->sh_type = Header.sh_type;
165 SHdr->sh_flags = Header.sh_flags;
166 SHdr->sh_addr = Header.sh_addr;
167 SHdr->sh_offset = Header.sh_offset;
168 SHdr->sh_size = Header.sh_size;
169 SHdr->sh_link = Header.sh_link;
170 SHdr->sh_info = Header.sh_info;
171 SHdr->sh_addralign = Header.sh_addralign;
172 SHdr->sh_entsize = Header.sh_entsize;
173}
174
175template <bool Is64Bits> void InterpSection<Is64Bits>::writeTo(uint8_t *Buf) {
176 memcpy(Buf, Config->DynamicLinker.data(), Config->DynamicLinker.size());
177}
178
Rafael Espindola35c6af32015-09-25 17:19:10 +0000179template <class ELFT>
180HashTableSection<ELFT>::HashTableSection(SymbolTableSection<ELFT> &DynSymSec)
181 : OutputSectionBase<ELFT::Is64Bits>(".hash", llvm::ELF::SHT_HASH,
182 llvm::ELF::SHF_ALLOC),
183 DynSymSec(DynSymSec) {
184 this->Header.sh_entsize = sizeof(Elf_Word);
185 this->Header.sh_addralign = sizeof(Elf_Word);
186}
187
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000188template <class ELFT> void HashTableSection<ELFT>::addSymbol(SymbolBody *S) {
189 StringRef Name = S->getName();
190 DynSymSec.addSymbol(Name);
191 Hashes.push_back(hash(Name));
192 S->setDynamicSymbolTableIndex(Hashes.size());
193}
194
Rafael Espindola35c6af32015-09-25 17:19:10 +0000195template <class ELFT>
196DynamicSection<ELFT>::DynamicSection(SymbolTable &SymTab,
197 HashTableSection<ELFT> &HashSec,
Igor Kudrinb1f2b512015-10-05 10:29:46 +0000198 RelocationSection<ELFT> &RelaDynSec,
199 const OutputSection<ELFT> &BssSec)
Rafael Espindola35c6af32015-09-25 17:19:10 +0000200 : OutputSectionBase<ELFT::Is64Bits>(".dynamic", llvm::ELF::SHT_DYNAMIC,
201 llvm::ELF::SHF_ALLOC |
202 llvm::ELF::SHF_WRITE),
203 HashSec(HashSec), DynSymSec(HashSec.getDynSymSec()),
204 DynStrSec(DynSymSec.getStrTabSec()), RelaDynSec(RelaDynSec),
Igor Kudrinb1f2b512015-10-05 10:29:46 +0000205 BssSec(BssSec), SymTab(SymTab) {
Rafael Espindola35c6af32015-09-25 17:19:10 +0000206 typename Base::HeaderT &Header = this->Header;
207 Header.sh_addralign = ELFT::Is64Bits ? 8 : 4;
208 Header.sh_entsize = ELFT::Is64Bits ? 16 : 8;
209}
210
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000211template <class ELFT> void DynamicSection<ELFT>::finalize() {
Michael J. Spencer52bf0eb2015-10-01 21:15:02 +0000212 if (this->Header.sh_size)
213 return; // Already finalized.
214
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000215 typename Base::HeaderT &Header = this->Header;
216 Header.sh_link = DynStrSec.getSectionIndex();
217
218 unsigned NumEntries = 0;
219 if (RelaDynSec.hasRelocs()) {
220 ++NumEntries; // DT_RELA / DT_REL
Rui Ueyama2dfd74f2015-09-30 21:57:53 +0000221 ++NumEntries; // DT_RELASZ / DT_RELSZ
222 ++NumEntries; // DT_RELAENT / DT_RELENT
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000223 }
224 ++NumEntries; // DT_SYMTAB
Rui Ueyama2dfd74f2015-09-30 21:57:53 +0000225 ++NumEntries; // DT_SYMENT
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000226 ++NumEntries; // DT_STRTAB
227 ++NumEntries; // DT_STRSZ
228 ++NumEntries; // DT_HASH
229
Rui Ueyama7de3f372015-10-01 19:36:04 +0000230 if (!Config->RPath.empty()) {
Davide Italianoc39c75d2015-10-06 16:20:00 +0000231 ++NumEntries; // DT_RUNPATH / DT_RPATH
Rui Ueyama7de3f372015-10-01 19:36:04 +0000232 DynStrSec.add(Config->RPath);
233 }
234
235 if (!Config->SoName.empty()) {
236 ++NumEntries; // DT_SONAME
237 DynStrSec.add(Config->SoName);
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000238 }
239
Rafael Espindola77572242015-10-02 19:37:55 +0000240 if (PreInitArraySec)
241 NumEntries += 2;
242 if (InitArraySec)
243 NumEntries += 2;
244 if (FiniArraySec)
245 NumEntries += 2;
246
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000247 const std::vector<std::unique_ptr<SharedFileBase>> &SharedFiles =
248 SymTab.getSharedFiles();
249 for (const std::unique_ptr<SharedFileBase> &File : SharedFiles)
Rafael Espindolac8b15812015-10-01 15:47:50 +0000250 DynStrSec.add(File->getSoName());
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000251 NumEntries += SharedFiles.size();
252
Igor Kudrinb1f2b512015-10-05 10:29:46 +0000253 if (Symbol *S = SymTab.getSymbols().lookup(Config->Init))
254 InitSym = dyn_cast<ELFSymbolBody<ELFT>>(S->Body);
255 if (Symbol *S = SymTab.getSymbols().lookup(Config->Fini))
256 FiniSym = dyn_cast<ELFSymbolBody<ELFT>>(S->Body);
Igor Kudrinb1f2b512015-10-05 10:29:46 +0000257 if (InitSym)
258 ++NumEntries; // DT_INIT
259 if (FiniSym)
260 ++NumEntries; // DT_FINI
261
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000262 ++NumEntries; // DT_NULL
263
264 Header.sh_size = NumEntries * Header.sh_entsize;
265}
266
267template <class ELFT> void DynamicSection<ELFT>::writeTo(uint8_t *Buf) {
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000268 auto *P = reinterpret_cast<Elf_Dyn *>(Buf);
269
Rui Ueyama8c205d52015-10-02 01:33:31 +0000270 auto WritePtr = [&](int32_t Tag, uint64_t Val) {
271 P->d_tag = Tag;
272 P->d_un.d_ptr = Val;
273 ++P;
274 };
275
276 auto WriteVal = [&](int32_t Tag, uint32_t Val) {
277 P->d_tag = Tag;
278 P->d_un.d_val = Val;
279 ++P;
280 };
281
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000282 if (RelaDynSec.hasRelocs()) {
283 bool IsRela = RelaDynSec.isRela();
Rui Ueyama8c205d52015-10-02 01:33:31 +0000284 WritePtr(IsRela ? DT_RELA : DT_REL, RelaDynSec.getVA());
285 WriteVal(IsRela ? DT_RELASZ : DT_RELSZ, RelaDynSec.getSize());
286 WriteVal(IsRela ? DT_RELAENT : DT_RELENT,
287 IsRela ? sizeof(Elf_Rela) : sizeof(Elf_Rel));
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000288 }
289
Rui Ueyama8c205d52015-10-02 01:33:31 +0000290 WritePtr(DT_SYMTAB, DynSymSec.getVA());
291 WritePtr(DT_SYMENT, sizeof(Elf_Sym));
292 WritePtr(DT_STRTAB, DynStrSec.getVA());
293 WriteVal(DT_STRSZ, DynStrSec.data().size());
294 WritePtr(DT_HASH, HashSec.getVA());
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000295
Rui Ueyama8c205d52015-10-02 01:33:31 +0000296 if (!Config->RPath.empty())
Davide Italianoc39c75d2015-10-06 16:20:00 +0000297
298 // If --enable-new-dtags is set lld emits DT_RUNPATH
299 // instead of DT_RPATH. The two tags are functionally
300 // equivalent except for the following:
301 // - DT_RUNPATH is searched after LD_LIBRARY_PATH, while
302 // DT_RPATH is searched before.
303 // - DT_RUNPATH is used only to search for direct
304 // dependencies of the object it's contained in, while
305 // DT_RPATH is used for indirect dependencies as well.
306 WriteVal(Config->EnableNewDtags ? DT_RUNPATH : DT_RPATH,
307 DynStrSec.getFileOff(Config->RPath));
Rui Ueyama2dfd74f2015-09-30 21:57:53 +0000308
Rui Ueyama8c205d52015-10-02 01:33:31 +0000309 if (!Config->SoName.empty())
310 WriteVal(DT_SONAME, DynStrSec.getFileOff(Config->SoName));
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000311
Rafael Espindola77572242015-10-02 19:37:55 +0000312 auto WriteArray = [&](int32_t T1, int32_t T2,
313 const OutputSection<ELFT> *Sec) {
314 if (!Sec)
315 return;
316 WritePtr(T1, Sec->getVA());
317 WriteVal(T2, Sec->getSize());
318 };
319 WriteArray(DT_PREINIT_ARRAY, DT_PREINIT_ARRAYSZ, PreInitArraySec);
320 WriteArray(DT_INIT_ARRAY, DT_INIT_ARRAYSZ, InitArraySec);
321 WriteArray(DT_FINI_ARRAY, DT_FINI_ARRAYSZ, FiniArraySec);
322
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000323 const std::vector<std::unique_ptr<SharedFileBase>> &SharedFiles =
324 SymTab.getSharedFiles();
Rui Ueyama8c205d52015-10-02 01:33:31 +0000325 for (const std::unique_ptr<SharedFileBase> &File : SharedFiles)
326 WriteVal(DT_NEEDED, DynStrSec.getFileOff(File->getSoName()));
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000327
Igor Kudrinb1f2b512015-10-05 10:29:46 +0000328 if (InitSym)
329 WritePtr(DT_INIT, getSymVA(*InitSym, BssSec));
330 if (FiniSym)
331 WritePtr(DT_FINI, getSymVA(*FiniSym, BssSec));
332
Rui Ueyama8c205d52015-10-02 01:33:31 +0000333 WriteVal(DT_NULL, 0);
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000334}
335
336template <class ELFT>
Rafael Espindola35c6af32015-09-25 17:19:10 +0000337OutputSection<ELFT>::OutputSection(const PltSection<ELFT> &PltSec,
338 const GotSection<ELFT> &GotSec,
339 const OutputSection<ELFT> &BssSec,
340 StringRef Name, uint32_t sh_type,
341 uintX_t sh_flags)
342 : OutputSectionBase<ELFT::Is64Bits>(Name, sh_type, sh_flags),
343 PltSec(PltSec), GotSec(GotSec), BssSec(BssSec) {}
344
345template <class ELFT>
Rafael Espindola71675852015-09-22 00:16:19 +0000346void OutputSection<ELFT>::addSection(InputSection<ELFT> *C) {
347 Sections.push_back(C);
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000348 C->setOutputSection(this);
349 uint32_t Align = C->getAlign();
350 if (Align > this->Header.sh_addralign)
351 this->Header.sh_addralign = Align;
352
353 uintX_t Off = this->Header.sh_size;
354 Off = RoundUpToAlignment(Off, Align);
355 C->setOutputSectionOff(Off);
356 Off += C->getSize();
357 this->Header.sh_size = Off;
358}
359
360template <class ELFT>
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000361typename ELFFile<ELFT>::uintX_t
Rafael Espindola8614c562015-10-06 14:33:58 +0000362lld::elf2::getSymVA(const SymbolBody &S, const OutputSection<ELFT> &BssSec) {
Rafael Espindolacd076f02015-09-25 18:19:03 +0000363 switch (S.kind()) {
Rafael Espindola8614c562015-10-06 14:33:58 +0000364 case SymbolBody::DefinedSyntheticKind: {
365 auto &D = cast<DefinedSynthetic<ELFT>>(S);
366 return D.Section.getVA() + D.Sym.st_value;
367 }
Rafael Espindolacd076f02015-09-25 18:19:03 +0000368 case SymbolBody::DefinedAbsoluteKind:
Rafael Espindola8614c562015-10-06 14:33:58 +0000369 return cast<DefinedAbsolute<ELFT>>(S).Sym.st_value;
Rafael Espindolacd076f02015-09-25 18:19:03 +0000370 case SymbolBody::DefinedRegularKind: {
371 const auto &DR = cast<DefinedRegular<ELFT>>(S);
372 const InputSection<ELFT> *SC = &DR.Section;
373 OutputSection<ELFT> *OS = SC->getOutputSection();
374 return OS->getVA() + SC->getOutputSectionOff() + DR.Sym.st_value;
375 }
376 case SymbolBody::DefinedCommonKind:
377 return BssSec.getVA() + cast<DefinedCommon<ELFT>>(S).OffsetInBSS;
378 case SymbolBody::SharedKind:
379 case SymbolBody::UndefinedKind:
380 return 0;
381 case SymbolBody::LazyKind:
Rafael Espindola8614c562015-10-06 14:33:58 +0000382 assert(S.isUsedInRegularObj() && "Lazy symbol reached writer");
383 return 0;
Rafael Espindolacd076f02015-09-25 18:19:03 +0000384 }
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000385}
386
387template <class ELFT>
388typename ELFFile<ELFT>::uintX_t
389lld::elf2::getLocalSymVA(const typename ELFFile<ELFT>::Elf_Sym *Sym,
390 const ObjectFile<ELFT> &File) {
391 uint32_t SecIndex = Sym->st_shndx;
392
393 if (SecIndex == SHN_XINDEX)
Rafael Espindolae1901cc2015-09-24 15:11:50 +0000394 SecIndex = File.getObj().getExtendedSymbolTableIndex(
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000395 Sym, File.getSymbolTable(), File.getSymbolTableShndx());
Rafael Espindola71675852015-09-22 00:16:19 +0000396 ArrayRef<InputSection<ELFT> *> Sections = File.getSections();
397 InputSection<ELFT> *Section = Sections[SecIndex];
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000398 OutputSection<ELFT> *Out = Section->getOutputSection();
399 return Out->getVA() + Section->getOutputSectionOff() + Sym->st_value;
400}
401
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000402template <class ELFT> void OutputSection<ELFT>::writeTo(uint8_t *Buf) {
Rafael Espindola71675852015-09-22 00:16:19 +0000403 for (InputSection<ELFT> *C : Sections)
Rafael Espindolac2d21192015-09-23 18:25:05 +0000404 C->writeTo(Buf, BssSec, PltSec, GotSec);
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000405}
406
407template <bool Is64Bits>
Rafael Espindola35c6af32015-09-25 17:19:10 +0000408StringTableSection<Is64Bits>::StringTableSection(bool Dynamic)
409 : OutputSectionBase<Is64Bits>(Dynamic ? ".dynstr" : ".strtab",
410 llvm::ELF::SHT_STRTAB,
411 Dynamic ? (uintX_t)llvm::ELF::SHF_ALLOC : 0),
412 Dynamic(Dynamic) {
413 this->Header.sh_addralign = 1;
414}
415
416template <bool Is64Bits>
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000417void StringTableSection<Is64Bits>::writeTo(uint8_t *Buf) {
418 StringRef Data = StrTabBuilder.data();
419 memcpy(Buf, Data.data(), Data.size());
420}
421
Rafael Espindola4f674ed2015-10-05 15:24:04 +0000422template <class ELFT> bool lld::elf2::includeInSymtab(const SymbolBody &B) {
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000423 if (!B.isUsedInRegularObj())
424 return false;
Rafael Espindola4f674ed2015-10-05 15:24:04 +0000425
426 // Don't include synthetic symbols like __init_array_start in every output.
427 if (auto *U = dyn_cast<DefinedAbsolute<ELFT>>(&B))
428 if (&U->Sym == &DefinedAbsolute<ELFT>::IgnoreUndef)
429 return false;
430
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000431 return true;
432}
433
Rafael Espindola05a3dd22015-09-22 23:38:23 +0000434bool lld::elf2::includeInDynamicSymtab(const SymbolBody &B) {
Rafael Espindola4f674ed2015-10-05 15:24:04 +0000435 uint8_t V = B.getMostConstrainingVisibility();
436 if (V != STV_DEFAULT && V != STV_PROTECTED)
437 return false;
438
Rafael Espindola05a3dd22015-09-22 23:38:23 +0000439 if (Config->ExportDynamic || Config->Shared)
440 return true;
441 return B.isUsedInDynamicReloc();
442}
443
Rafael Espindolad1cf4212015-10-05 16:25:43 +0000444template <class ELFT>
445bool lld::elf2::shouldKeepInSymtab(StringRef SymName,
446 const typename ELFFile<ELFT>::Elf_Sym &Sym) {
447 if (Sym.getType() == STT_SECTION)
448 return false;
449
Davide Italiano6993ba42015-09-26 00:47:56 +0000450 if (Config->DiscardNone)
451 return true;
452
453 // ELF defines dynamic locals as symbols which name starts with ".L".
454 return !(Config->DiscardLocals && SymName.startswith(".L"));
455}
456
Rafael Espindola35c6af32015-09-25 17:19:10 +0000457template <class ELFT>
458SymbolTableSection<ELFT>::SymbolTableSection(
459 SymbolTable &Table, StringTableSection<ELFT::Is64Bits> &StrTabSec,
460 const OutputSection<ELFT> &BssSec)
461 : OutputSectionBase<ELFT::Is64Bits>(
462 StrTabSec.isDynamic() ? ".dynsym" : ".symtab",
463 StrTabSec.isDynamic() ? llvm::ELF::SHT_DYNSYM : llvm::ELF::SHT_SYMTAB,
464 StrTabSec.isDynamic() ? (uintX_t)llvm::ELF::SHF_ALLOC : 0),
465 Table(Table), StrTabSec(StrTabSec), BssSec(BssSec) {
466 typedef OutputSectionBase<ELFT::Is64Bits> Base;
467 typename Base::HeaderT &Header = this->Header;
468
469 Header.sh_entsize = sizeof(Elf_Sym);
470 Header.sh_addralign = ELFT::Is64Bits ? 8 : 4;
471}
472
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000473template <class ELFT> void SymbolTableSection<ELFT>::writeTo(uint8_t *Buf) {
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000474 Buf += sizeof(Elf_Sym);
475
476 // All symbols with STB_LOCAL binding precede the weak and global symbols.
477 // .dynsym only contains global symbols.
Rui Ueyama8ddfa812015-09-30 00:32:10 +0000478 if (!Config->DiscardAll && !StrTabSec.isDynamic())
479 writeLocalSymbols(Buf);
480
481 writeGlobalSymbols(Buf);
482}
483
484template <class ELFT>
485void SymbolTableSection<ELFT>::writeLocalSymbols(uint8_t *&Buf) {
486 // Iterate over all input object files to copy their local symbols
487 // to the output symbol table pointed by Buf.
488 for (const std::unique_ptr<ObjectFileBase> &FileB : Table.getObjectFiles()) {
489 auto &File = cast<ObjectFile<ELFT>>(*FileB);
490 Elf_Sym_Range Syms = File.getLocalSymbols();
491 for (const Elf_Sym &Sym : Syms) {
Rui Ueyama8ddfa812015-09-30 00:32:10 +0000492 ErrorOr<StringRef> SymName = Sym.getName(File.getStringTable());
Rafael Espindolad1cf4212015-10-05 16:25:43 +0000493 if (SymName && !shouldKeepInSymtab<ELFT>(*SymName, Sym))
Rui Ueyama8ddfa812015-09-30 00:32:10 +0000494 continue;
Rui Ueyamac55733e2015-09-30 00:54:29 +0000495 auto *ESym = reinterpret_cast<Elf_Sym *>(Buf);
496 Buf += sizeof(*ESym);
Rui Ueyama8ddfa812015-09-30 00:32:10 +0000497 ESym->st_name = (SymName) ? StrTabSec.getFileOff(*SymName) : 0;
498 ESym->st_size = Sym.st_size;
499 ESym->setBindingAndType(Sym.getBinding(), Sym.getType());
500 uint32_t SecIndex = Sym.st_shndx;
501 uintX_t VA = Sym.st_value;
502 if (SecIndex == SHN_ABS) {
503 ESym->st_shndx = SHN_ABS;
504 } else {
505 if (SecIndex == SHN_XINDEX)
506 SecIndex = File.getObj().getExtendedSymbolTableIndex(
507 &Sym, File.getSymbolTable(), File.getSymbolTableShndx());
508 ArrayRef<InputSection<ELFT> *> Sections = File.getSections();
509 const InputSection<ELFT> *Section = Sections[SecIndex];
510 const OutputSection<ELFT> *Out = Section->getOutputSection();
511 ESym->st_shndx = Out->getSectionIndex();
512 VA += Out->getVA() + Section->getOutputSectionOff();
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000513 }
Rui Ueyama8ddfa812015-09-30 00:32:10 +0000514 ESym->st_value = VA;
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000515 }
516 }
Rui Ueyama8ddfa812015-09-30 00:32:10 +0000517}
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000518
Rui Ueyama8ddfa812015-09-30 00:32:10 +0000519template <class ELFT>
520void SymbolTableSection<ELFT>::writeGlobalSymbols(uint8_t *&Buf) {
521 // Write the internal symbol table contents to the output symbol table
522 // pointed by Buf.
Rafael Espindola4f674ed2015-10-05 15:24:04 +0000523 uint8_t *Start = Buf;
Rui Ueyama8ddfa812015-09-30 00:32:10 +0000524 for (const std::pair<StringRef, Symbol *> &P : Table.getSymbols()) {
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000525 StringRef Name = P.first;
526 Symbol *Sym = P.second;
527 SymbolBody *Body = Sym->Body;
Rafael Espindola4f674ed2015-10-05 15:24:04 +0000528 if (!includeInSymtab<ELFT>(*Body))
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000529 continue;
Rafael Espindola05a3dd22015-09-22 23:38:23 +0000530 if (StrTabSec.isDynamic() && !includeInDynamicSymtab(*Body))
531 continue;
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000532
533 auto *ESym = reinterpret_cast<Elf_Sym *>(Buf);
Rui Ueyamac55733e2015-09-30 00:54:29 +0000534 Buf += sizeof(*ESym);
Rafael Espindola8614c562015-10-06 14:33:58 +0000535
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000536 ESym->st_name = StrTabSec.getFileOff(Name);
537
Rafael Espindola25b0acb2015-09-25 17:32:37 +0000538 const OutputSection<ELFT> *Out = nullptr;
539 const InputSection<ELFT> *Section = nullptr;
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000540
Rafael Espindola8614c562015-10-06 14:33:58 +0000541 switch (Body->kind()) {
Rafael Espindola0e604f92015-09-25 18:56:53 +0000542 case SymbolBody::DefinedSyntheticKind:
543 Out = &cast<DefinedSynthetic<ELFT>>(Body)->Section;
544 break;
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000545 case SymbolBody::DefinedRegularKind:
Rafael Espindola8614c562015-10-06 14:33:58 +0000546 Section = &cast<DefinedRegular<ELFT>>(Body)->Section;
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000547 break;
548 case SymbolBody::DefinedCommonKind:
Rafael Espindolac2d21192015-09-23 18:25:05 +0000549 Out = &BssSec;
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000550 break;
551 case SymbolBody::UndefinedKind:
552 case SymbolBody::DefinedAbsoluteKind:
553 case SymbolBody::SharedKind:
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000554 case SymbolBody::LazyKind:
Rafael Espindola8614c562015-10-06 14:33:58 +0000555 break;
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000556 }
557
Rafael Espindola8614c562015-10-06 14:33:58 +0000558 unsigned char Binding = Body->isWeak() ? STB_WEAK : STB_GLOBAL;
559 unsigned char Type = STT_NOTYPE;
560 uintX_t Size = 0;
561 if (const auto *EBody = dyn_cast<ELFSymbolBody<ELFT>>(Body)) {
562 const Elf_Sym &InputSym = EBody->Sym;
563 Binding = InputSym.getBinding();
564 Type = InputSym.getType();
565 Size = InputSym.st_size;
566 }
567
568 unsigned char Visibility = Body->getMostConstrainingVisibility();
Rafael Espindola4f674ed2015-10-05 15:24:04 +0000569 if (Visibility != STV_DEFAULT && Visibility != STV_PROTECTED)
570 Binding = STB_LOCAL;
571
Rafael Espindola8614c562015-10-06 14:33:58 +0000572 ESym->setBindingAndType(Binding, Type);
573 ESym->st_size = Size;
Rafael Espindola4f674ed2015-10-05 15:24:04 +0000574 ESym->setVisibility(Visibility);
Rafael Espindola8614c562015-10-06 14:33:58 +0000575 ESym->st_value = getSymVA(*Body, BssSec);
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000576
577 if (Section)
578 Out = Section->getOutputSection();
579
Rafael Espindola8614c562015-10-06 14:33:58 +0000580 if (isa<DefinedAbsolute<ELFT>>(Body))
Rafael Espindola6f4bd532015-10-06 14:17:53 +0000581 ESym->st_shndx = SHN_ABS;
582 else if (Out)
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000583 ESym->st_shndx = Out->getSectionIndex();
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000584 }
Rafael Espindola4f674ed2015-10-05 15:24:04 +0000585 if (!StrTabSec.isDynamic())
586 std::stable_sort(
587 reinterpret_cast<Elf_Sym *>(Start), reinterpret_cast<Elf_Sym *>(Buf),
588 [](const Elf_Sym &A, const Elf_Sym &B) -> bool {
589 return A.getBinding() == STB_LOCAL && B.getBinding() != STB_LOCAL;
590 });
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000591}
592
593namespace lld {
594namespace elf2 {
595template class OutputSectionBase<false>;
596template class OutputSectionBase<true>;
597
598template void OutputSectionBase<false>::writeHeaderTo<support::little>(
Rafael Espindolaf68b7072015-09-21 22:21:46 +0000599 ELFFile<ELFType<support::little, false>>::Elf_Shdr *SHdr);
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000600template void OutputSectionBase<true>::writeHeaderTo<support::little>(
Rafael Espindolaf68b7072015-09-21 22:21:46 +0000601 ELFFile<ELFType<support::little, true>>::Elf_Shdr *SHdr);
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000602template void OutputSectionBase<false>::writeHeaderTo<support::big>(
Rafael Espindolaf68b7072015-09-21 22:21:46 +0000603 ELFFile<ELFType<support::big, false>>::Elf_Shdr *SHdr);
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000604template void OutputSectionBase<true>::writeHeaderTo<support::big>(
Rafael Espindolaf68b7072015-09-21 22:21:46 +0000605 ELFFile<ELFType<support::big, true>>::Elf_Shdr *SHdr);
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000606
607template class GotSection<ELF32LE>;
608template class GotSection<ELF32BE>;
609template class GotSection<ELF64LE>;
610template class GotSection<ELF64BE>;
611
612template class PltSection<ELF32LE>;
613template class PltSection<ELF32BE>;
614template class PltSection<ELF64LE>;
615template class PltSection<ELF64BE>;
616
617template class RelocationSection<ELF32LE>;
618template class RelocationSection<ELF32BE>;
619template class RelocationSection<ELF64LE>;
620template class RelocationSection<ELF64BE>;
621
622template class InterpSection<false>;
623template class InterpSection<true>;
624
625template class HashTableSection<ELF32LE>;
626template class HashTableSection<ELF32BE>;
627template class HashTableSection<ELF64LE>;
628template class HashTableSection<ELF64BE>;
629
630template class DynamicSection<ELF32LE>;
631template class DynamicSection<ELF32BE>;
632template class DynamicSection<ELF64LE>;
633template class DynamicSection<ELF64BE>;
634
635template class OutputSection<ELF32LE>;
636template class OutputSection<ELF32BE>;
637template class OutputSection<ELF64LE>;
638template class OutputSection<ELF64BE>;
639
640template class StringTableSection<false>;
641template class StringTableSection<true>;
642
643template class SymbolTableSection<ELF32LE>;
644template class SymbolTableSection<ELF32BE>;
645template class SymbolTableSection<ELF64LE>;
646template class SymbolTableSection<ELF64BE>;
647
Rafael Espindola8614c562015-10-06 14:33:58 +0000648template ELFFile<ELF32LE>::uintX_t getSymVA(const SymbolBody &,
649 const OutputSection<ELF32LE> &);
650template ELFFile<ELF32BE>::uintX_t getSymVA(const SymbolBody &,
651 const OutputSection<ELF32BE> &);
652template ELFFile<ELF64LE>::uintX_t getSymVA(const SymbolBody &,
653 const OutputSection<ELF64LE> &);
654template ELFFile<ELF64BE>::uintX_t getSymVA(const SymbolBody &,
655 const OutputSection<ELF64BE> &);
Rafael Espindola4ea00212015-09-21 22:01:00 +0000656
Rafael Espindola56f965f2015-09-21 22:48:12 +0000657template ELFFile<ELF32LE>::uintX_t
Rui Ueyamab189b5c2015-09-30 00:43:22 +0000658getLocalSymVA(const ELFFile<ELF32LE>::Elf_Sym *, const ObjectFile<ELF32LE> &);
Rafael Espindola4ea00212015-09-21 22:01:00 +0000659
Rafael Espindola56f965f2015-09-21 22:48:12 +0000660template ELFFile<ELF32BE>::uintX_t
Rui Ueyamab189b5c2015-09-30 00:43:22 +0000661getLocalSymVA(const ELFFile<ELF32BE>::Elf_Sym *, const ObjectFile<ELF32BE> &);
Rafael Espindola4ea00212015-09-21 22:01:00 +0000662
Rafael Espindola56f965f2015-09-21 22:48:12 +0000663template ELFFile<ELF64LE>::uintX_t
Rui Ueyamab189b5c2015-09-30 00:43:22 +0000664getLocalSymVA(const ELFFile<ELF64LE>::Elf_Sym *, const ObjectFile<ELF64LE> &);
Rafael Espindola4ea00212015-09-21 22:01:00 +0000665
Rafael Espindola56f965f2015-09-21 22:48:12 +0000666template ELFFile<ELF64BE>::uintX_t
Rui Ueyamab189b5c2015-09-30 00:43:22 +0000667getLocalSymVA(const ELFFile<ELF64BE>::Elf_Sym *, const ObjectFile<ELF64BE> &);
Rafael Espindola4f674ed2015-10-05 15:24:04 +0000668
669template bool includeInSymtab<ELF32LE>(const SymbolBody &);
670template bool includeInSymtab<ELF32BE>(const SymbolBody &);
671template bool includeInSymtab<ELF64LE>(const SymbolBody &);
672template bool includeInSymtab<ELF64BE>(const SymbolBody &);
Rafael Espindolad1cf4212015-10-05 16:25:43 +0000673
674template bool shouldKeepInSymtab<ELF32LE>(StringRef,
675 const ELFFile<ELF32LE>::Elf_Sym &);
676template bool shouldKeepInSymtab<ELF32BE>(StringRef,
677 const ELFFile<ELF32BE>::Elf_Sym &);
678template bool shouldKeepInSymtab<ELF64LE>(StringRef,
679 const ELFFile<ELF64LE>::Elf_Sym &);
680template bool shouldKeepInSymtab<ELF64BE>(StringRef,
681 const ELFFile<ELF64BE>::Elf_Sym &);
Rafael Espindola5805c4f2015-09-21 21:38:08 +0000682}
683}