Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 1 | //===- 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. Spencer | f832541 | 2015-09-04 22:48:30 +0000 | [diff] [blame] | 10 | #include "Writer.h" |
Rui Ueyama | cb8474ed | 2015-08-05 23:51:50 +0000 | [diff] [blame] | 11 | #include "Config.h" |
Rafael Espindola | 5805c4f | 2015-09-21 21:38:08 +0000 | [diff] [blame] | 12 | #include "OutputSections.h" |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 13 | #include "SymbolTable.h" |
Rafael Espindola | 01205f7 | 2015-09-22 18:19:46 +0000 | [diff] [blame] | 14 | #include "Target.h" |
Rafael Espindola | 6b83b90 | 2015-08-12 00:00:24 +0000 | [diff] [blame] | 15 | |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 16 | #include "llvm/Support/FileOutputBuffer.h" |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 17 | |
| 18 | using namespace llvm; |
| 19 | using namespace llvm::ELF; |
| 20 | using namespace llvm::object; |
| 21 | |
| 22 | using namespace lld; |
| 23 | using namespace lld::elf2; |
| 24 | |
| 25 | static const int PageSize = 4096; |
| 26 | |
Rafael Espindola | d67bb26 | 2015-09-11 13:20:07 +0000 | [diff] [blame] | 27 | // On freebsd x86_64 the first page cannot be mmaped. |
| 28 | // On linux that is controled by vm.mmap_min_addr. At least on some x86_64 |
| 29 | // installs that is 65536, so the first 15 pages cannot be used. |
| 30 | // Given that, the smallest value that can be used in here is 0x10000. |
Rafael Espindola | 0a2e211 | 2015-09-10 15:41:34 +0000 | [diff] [blame] | 31 | // If using 2MB pages, the smallest page aligned address that works is |
| 32 | // 0x200000, but it looks like every OS uses 4k pages for executables. |
| 33 | // FIXME: This is architecture and OS dependent. |
| 34 | static const int VAStart = 0x10000; |
| 35 | |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 36 | namespace { |
Rafael Espindola | 740fafe | 2015-09-08 19:43:27 +0000 | [diff] [blame] | 37 | |
Rui Ueyama | 04dae55 | 2015-10-02 21:23:17 +0000 | [diff] [blame] | 38 | static uint32_t toPHDRFlags(uint64_t Flags) { |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 39 | uint32_t Ret = PF_R; |
| 40 | if (Flags & SHF_WRITE) |
| 41 | Ret |= PF_W; |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 42 | if (Flags & SHF_EXECINSTR) |
| 43 | Ret |= PF_X; |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 44 | return Ret; |
| 45 | } |
| 46 | |
Rui Ueyama | 5713b8d | 2015-10-02 22:17:07 +0000 | [diff] [blame] | 47 | template <class ELFT> struct ProgramHeader { |
| 48 | typedef typename ELFFile<ELFT>::uintX_t uintX_t; |
| 49 | typedef typename ELFFile<ELFT>::Elf_Phdr Elf_Phdr; |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 50 | |
Rui Ueyama | 04dae55 | 2015-10-02 21:23:17 +0000 | [diff] [blame] | 51 | ProgramHeader(uintX_t Type, uintX_t Flags, uintX_t FileOff, uintX_t VA) { |
Rui Ueyama | 5713b8d | 2015-10-02 22:17:07 +0000 | [diff] [blame] | 52 | std::memset(&Header, 0, sizeof(Elf_Phdr)); |
Rui Ueyama | 0fb1ee0 | 2015-10-02 21:13:19 +0000 | [diff] [blame] | 53 | Header.p_type = Type; |
| 54 | Header.p_flags = Flags; |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 55 | Header.p_align = PageSize; |
Rui Ueyama | 04dae55 | 2015-10-02 21:23:17 +0000 | [diff] [blame] | 56 | Header.p_offset = FileOff; |
| 57 | Header.p_vaddr = VA; |
| 58 | Header.p_paddr = VA; |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 59 | } |
| 60 | |
Rui Ueyama | 5713b8d | 2015-10-02 22:17:07 +0000 | [diff] [blame] | 61 | void setValuesFromSection(OutputSectionBase<ELFT::Is64Bits> &Sec) { |
Rui Ueyama | 04dae55 | 2015-10-02 21:23:17 +0000 | [diff] [blame] | 62 | Header.p_flags = toPHDRFlags(Sec.getFlags()); |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 63 | Header.p_offset = Sec.getFileOff(); |
| 64 | Header.p_vaddr = Sec.getVA(); |
| 65 | Header.p_paddr = Header.p_vaddr; |
| 66 | Header.p_filesz = Sec.getSize(); |
| 67 | Header.p_memsz = Header.p_filesz; |
| 68 | Header.p_align = Sec.getAlign(); |
| 69 | } |
| 70 | |
Rui Ueyama | 5713b8d | 2015-10-02 22:17:07 +0000 | [diff] [blame] | 71 | Elf_Phdr Header; |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 72 | bool Closed = false; |
| 73 | }; |
| 74 | |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 75 | // The writer writes a SymbolTable result to a file. |
| 76 | template <class ELFT> class Writer { |
| 77 | public: |
Rafael Espindola | 18608a0 | 2015-09-08 21:57:31 +0000 | [diff] [blame] | 78 | typedef typename ELFFile<ELFT>::uintX_t uintX_t; |
| 79 | typedef typename ELFFile<ELFT>::Elf_Shdr Elf_Shdr; |
| 80 | typedef typename ELFFile<ELFT>::Elf_Ehdr Elf_Ehdr; |
| 81 | typedef typename ELFFile<ELFT>::Elf_Phdr Elf_Phdr; |
| 82 | typedef typename ELFFile<ELFT>::Elf_Sym Elf_Sym; |
Davide Italiano | 6d328d3 | 2015-09-16 20:45:57 +0000 | [diff] [blame] | 83 | typedef typename ELFFile<ELFT>::Elf_Sym_Range Elf_Sym_Range; |
Rafael Espindola | 19e3889 | 2015-09-16 15:54:15 +0000 | [diff] [blame] | 84 | typedef typename ELFFile<ELFT>::Elf_Rela Elf_Rela; |
Rafael Espindola | 3f4228f | 2015-09-09 15:33:08 +0000 | [diff] [blame] | 85 | Writer(SymbolTable *T) |
Rafael Espindola | c2d2119 | 2015-09-23 18:25:05 +0000 | [diff] [blame] | 86 | : SymTabSec(*T, StrTabSec, BssSec), DynSymSec(*T, DynStrSec, BssSec), |
Rafael Espindola | eb79273 | 2015-09-21 15:11:29 +0000 | [diff] [blame] | 87 | RelaDynSec(DynSymSec, GotSec, T->shouldUseRela()), PltSec(GotSec), |
Igor Kudrin | b1f2b51 | 2015-10-05 10:29:46 +0000 | [diff] [blame] | 88 | HashSec(DynSymSec), DynamicSec(*T, HashSec, RelaDynSec, BssSec), |
Rafael Espindola | c2d2119 | 2015-09-23 18:25:05 +0000 | [diff] [blame] | 89 | BssSec(PltSec, GotSec, BssSec, ".bss", SHT_NOBITS, |
| 90 | SHF_ALLOC | SHF_WRITE) {} |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 91 | void run(); |
| 92 | |
| 93 | private: |
| 94 | void createSections(); |
Rafael Espindola | 67a5da6 | 2015-09-17 14:02:10 +0000 | [diff] [blame] | 95 | template <bool isRela> |
Rafael Espindola | 53d5cea | 2015-09-21 17:47:00 +0000 | [diff] [blame] | 96 | void scanRelocs(const InputSection<ELFT> &C, |
Rafael Espindola | 67a5da6 | 2015-09-17 14:02:10 +0000 | [diff] [blame] | 97 | iterator_range<const Elf_Rel_Impl<ELFT, isRela> *> Rels); |
Rafael Espindola | 53d5cea | 2015-09-21 17:47:00 +0000 | [diff] [blame] | 98 | void scanRelocs(const InputSection<ELFT> &C); |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 99 | void assignAddresses(); |
| 100 | void openFile(StringRef OutputPath); |
| 101 | void writeHeader(); |
| 102 | void writeSections(); |
Rafael Espindola | 7010776 | 2015-09-11 18:49:42 +0000 | [diff] [blame] | 103 | bool needsInterpSection() const { |
| 104 | return !SymTabSec.getSymTable().getSharedFiles().empty() && |
| 105 | !Config->DynamicLinker.empty(); |
| 106 | } |
Rafael Espindola | 4340aad | 2015-09-11 22:42:45 +0000 | [diff] [blame] | 107 | bool needsDynamicSections() const { |
| 108 | return !SymTabSec.getSymTable().getSharedFiles().empty() || Config->Shared; |
| 109 | } |
| 110 | unsigned getVAStart() const { return Config->Shared ? 0 : VAStart; } |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 111 | |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 112 | std::unique_ptr<llvm::FileOutputBuffer> Buffer; |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 113 | |
Rafael Espindola | 83b0dc6 | 2015-08-13 22:21:37 +0000 | [diff] [blame] | 114 | llvm::SpecificBumpPtrAllocator<OutputSection<ELFT>> CAlloc; |
Rafael Espindola | ebd2108 | 2015-08-13 22:14:37 +0000 | [diff] [blame] | 115 | std::vector<OutputSectionBase<ELFT::Is64Bits> *> OutputSections; |
Rafael Espindola | 5f55387 | 2015-09-08 17:39:39 +0000 | [diff] [blame] | 116 | unsigned getNumSections() const { return OutputSections.size() + 1; } |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 117 | |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 118 | llvm::BumpPtrAllocator PAlloc; |
Rui Ueyama | 5713b8d | 2015-10-02 22:17:07 +0000 | [diff] [blame] | 119 | std::vector<ProgramHeader<ELFT> *> PHDRs; |
| 120 | ProgramHeader<ELFT> FileHeaderPHDR{PT_LOAD, PF_R, 0, 0}; |
| 121 | ProgramHeader<ELFT> InterpPHDR{PT_INTERP, 0, 0, 0}; |
| 122 | ProgramHeader<ELFT> DynamicPHDR{PT_DYNAMIC, 0, 0, 0}; |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 123 | |
Rafael Espindola | 98f6bd0 | 2015-08-11 23:14:13 +0000 | [diff] [blame] | 124 | uintX_t FileSize; |
Michael J. Spencer | 1d299a8 | 2015-09-09 20:48:09 +0000 | [diff] [blame] | 125 | uintX_t ProgramHeaderOff; |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 126 | uintX_t SectionHeaderOff; |
Rafael Espindola | ebd2108 | 2015-08-13 22:14:37 +0000 | [diff] [blame] | 127 | |
Rui Ueyama | 9078f73 | 2015-09-14 20:32:41 +0000 | [diff] [blame] | 128 | StringTableSection<ELFT::Is64Bits> StrTabSec = { /*dynamic=*/false }; |
| 129 | StringTableSection<ELFT::Is64Bits> DynStrSec = { /*dynamic=*/true }; |
Rafael Espindola | ebd2108 | 2015-08-13 22:14:37 +0000 | [diff] [blame] | 130 | |
Rafael Espindola | 5805c4f | 2015-09-21 21:38:08 +0000 | [diff] [blame] | 131 | lld::elf2::SymbolTableSection<ELFT> SymTabSec; |
| 132 | lld::elf2::SymbolTableSection<ELFT> DynSymSec; |
Rafael Espindola | 740fafe | 2015-09-08 19:43:27 +0000 | [diff] [blame] | 133 | |
Rafael Espindola | 19e3889 | 2015-09-16 15:54:15 +0000 | [diff] [blame] | 134 | RelocationSection<ELFT> RelaDynSec; |
| 135 | |
Rafael Espindola | 5c2310c | 2015-09-18 14:40:19 +0000 | [diff] [blame] | 136 | GotSection<ELFT> GotSec; |
Rafael Espindola | eb79273 | 2015-09-21 15:11:29 +0000 | [diff] [blame] | 137 | PltSection<ELFT> PltSec; |
Rafael Espindola | 5c2310c | 2015-09-18 14:40:19 +0000 | [diff] [blame] | 138 | |
Rafael Espindola | 601771e | 2015-09-14 20:20:34 +0000 | [diff] [blame] | 139 | HashTableSection<ELFT> HashSec; |
| 140 | |
Rafael Espindola | 80faee8 | 2015-09-14 22:08:55 +0000 | [diff] [blame] | 141 | DynamicSection<ELFT> DynamicSec; |
| 142 | |
Rafael Espindola | 7010776 | 2015-09-11 18:49:42 +0000 | [diff] [blame] | 143 | InterpSection<ELFT::Is64Bits> InterpSec; |
| 144 | |
Rafael Espindola | c2d2119 | 2015-09-23 18:25:05 +0000 | [diff] [blame] | 145 | OutputSection<ELFT> BssSec; |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 146 | }; |
| 147 | } // anonymous namespace |
| 148 | |
| 149 | namespace lld { |
| 150 | namespace elf2 { |
| 151 | |
Rafael Espindola | 2ffdd4d | 2015-08-04 14:29:01 +0000 | [diff] [blame] | 152 | template <class ELFT> |
Rui Ueyama | cb8474ed | 2015-08-05 23:51:50 +0000 | [diff] [blame] | 153 | void writeResult(SymbolTable *Symtab) { Writer<ELFT>(Symtab).run(); } |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 154 | |
Rui Ueyama | cb8474ed | 2015-08-05 23:51:50 +0000 | [diff] [blame] | 155 | template void writeResult<ELF32LE>(SymbolTable *); |
| 156 | template void writeResult<ELF32BE>(SymbolTable *); |
| 157 | template void writeResult<ELF64LE>(SymbolTable *); |
| 158 | template void writeResult<ELF64BE>(SymbolTable *); |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 159 | |
| 160 | } // namespace elf2 |
| 161 | } // namespace lld |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 162 | |
| 163 | // The main function of the writer. |
Rui Ueyama | afff74e2 | 2015-08-05 23:24:46 +0000 | [diff] [blame] | 164 | template <class ELFT> void Writer<ELFT>::run() { |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 165 | createSections(); |
| 166 | assignAddresses(); |
Rui Ueyama | cb8474ed | 2015-08-05 23:51:50 +0000 | [diff] [blame] | 167 | openFile(Config->OutputFile); |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 168 | writeHeader(); |
| 169 | writeSections(); |
| 170 | error(Buffer->commit()); |
| 171 | } |
| 172 | |
Rafael Espindola | a747179 | 2015-08-13 17:04:50 +0000 | [diff] [blame] | 173 | namespace { |
| 174 | template <bool Is64Bits> struct SectionKey { |
| 175 | typedef typename std::conditional<Is64Bits, uint64_t, uint32_t>::type uintX_t; |
| 176 | StringRef Name; |
Rui Ueyama | 0fb1ee0 | 2015-10-02 21:13:19 +0000 | [diff] [blame] | 177 | uint32_t Type; |
| 178 | uintX_t Flags; |
Rafael Espindola | a747179 | 2015-08-13 17:04:50 +0000 | [diff] [blame] | 179 | }; |
| 180 | } |
| 181 | namespace llvm { |
| 182 | template <bool Is64Bits> struct DenseMapInfo<SectionKey<Is64Bits>> { |
| 183 | static SectionKey<Is64Bits> getEmptyKey() { |
| 184 | return SectionKey<Is64Bits>{DenseMapInfo<StringRef>::getEmptyKey(), 0, 0}; |
| 185 | } |
| 186 | static SectionKey<Is64Bits> getTombstoneKey() { |
| 187 | return SectionKey<Is64Bits>{DenseMapInfo<StringRef>::getTombstoneKey(), 0, |
| 188 | 0}; |
| 189 | } |
| 190 | static unsigned getHashValue(const SectionKey<Is64Bits> &Val) { |
Rui Ueyama | 0fb1ee0 | 2015-10-02 21:13:19 +0000 | [diff] [blame] | 191 | return hash_combine(Val.Name, Val.Type, Val.Flags); |
Rafael Espindola | a747179 | 2015-08-13 17:04:50 +0000 | [diff] [blame] | 192 | } |
| 193 | static bool isEqual(const SectionKey<Is64Bits> &LHS, |
| 194 | const SectionKey<Is64Bits> &RHS) { |
| 195 | return DenseMapInfo<StringRef>::isEqual(LHS.Name, RHS.Name) && |
Rui Ueyama | 0fb1ee0 | 2015-10-02 21:13:19 +0000 | [diff] [blame] | 196 | LHS.Type == RHS.Type && LHS.Flags == RHS.Flags; |
Rafael Espindola | a747179 | 2015-08-13 17:04:50 +0000 | [diff] [blame] | 197 | } |
| 198 | }; |
| 199 | } |
| 200 | |
Rafael Espindola | 19e3889 | 2015-09-16 15:54:15 +0000 | [diff] [blame] | 201 | // The reason we have to do this early scan is as follows |
| 202 | // * To mmap the output file, we need to know the size |
| 203 | // * For that, we need to know how many dynamic relocs we will have. |
| 204 | // It might be possible to avoid this by outputting the file with write: |
| 205 | // * Write the allocated output sections, computing addresses. |
| 206 | // * Apply relocations, recording which ones require a dynamic reloc. |
| 207 | // * Write the dynamic relocations. |
| 208 | // * Write the rest of the file. |
| 209 | template <class ELFT> |
Rafael Espindola | 67a5da6 | 2015-09-17 14:02:10 +0000 | [diff] [blame] | 210 | template <bool isRela> |
| 211 | void Writer<ELFT>::scanRelocs( |
Rafael Espindola | 53d5cea | 2015-09-21 17:47:00 +0000 | [diff] [blame] | 212 | const InputSection<ELFT> &C, |
Rafael Espindola | 67a5da6 | 2015-09-17 14:02:10 +0000 | [diff] [blame] | 213 | iterator_range<const Elf_Rel_Impl<ELFT, isRela> *> Rels) { |
| 214 | typedef Elf_Rel_Impl<ELFT, isRela> RelType; |
| 215 | const ObjectFile<ELFT> &File = *C.getFile(); |
Rafael Espindola | e1901cc | 2015-09-24 15:11:50 +0000 | [diff] [blame] | 216 | bool IsMips64EL = File.getObj().isMips64EL(); |
Rafael Espindola | 67a5da6 | 2015-09-17 14:02:10 +0000 | [diff] [blame] | 217 | for (const RelType &RI : Rels) { |
| 218 | uint32_t SymIndex = RI.getSymbol(IsMips64EL); |
Rafael Espindola | 5c2310c | 2015-09-18 14:40:19 +0000 | [diff] [blame] | 219 | SymbolBody *Body = File.getSymbolBody(SymIndex); |
Rafael Espindola | eb79273 | 2015-09-21 15:11:29 +0000 | [diff] [blame] | 220 | uint32_t Type = RI.getType(IsMips64EL); |
Rafael Espindola | ae24400 | 2015-10-05 19:30:12 +0000 | [diff] [blame^] | 221 | if (Body) { |
| 222 | if (Target->relocNeedsPlt(Type, *Body)) { |
| 223 | if (Body->isInPlt()) |
| 224 | continue; |
| 225 | PltSec.addEntry(Body); |
| 226 | } |
| 227 | if (Target->relocNeedsGot(Type, *Body)) { |
| 228 | if (Body->isInGot()) |
| 229 | continue; |
| 230 | GotSec.addEntry(Body); |
| 231 | Body->setUsedInDynamicReloc(); |
| 232 | RelaDynSec.addReloc({C, RI}); |
Rafael Espindola | eb79273 | 2015-09-21 15:11:29 +0000 | [diff] [blame] | 233 | continue; |
Rafael Espindola | ae24400 | 2015-10-05 19:30:12 +0000 | [diff] [blame^] | 234 | } |
| 235 | if (Body->isShared()) { |
| 236 | Body->setUsedInDynamicReloc(); |
| 237 | RelaDynSec.addReloc({C, RI}); |
| 238 | continue; |
| 239 | } |
Rafael Espindola | eb79273 | 2015-09-21 15:11:29 +0000 | [diff] [blame] | 240 | } |
Rafael Espindola | ae24400 | 2015-10-05 19:30:12 +0000 | [diff] [blame^] | 241 | if (Config->Shared && !Target->isRelRelative(Type)) |
| 242 | RelaDynSec.addReloc({C, RI}); |
Rafael Espindola | 67a5da6 | 2015-09-17 14:02:10 +0000 | [diff] [blame] | 243 | } |
| 244 | } |
| 245 | |
| 246 | template <class ELFT> |
Rafael Espindola | 53d5cea | 2015-09-21 17:47:00 +0000 | [diff] [blame] | 247 | void Writer<ELFT>::scanRelocs(const InputSection<ELFT> &C) { |
Rafael Espindola | e1901cc | 2015-09-24 15:11:50 +0000 | [diff] [blame] | 248 | ObjectFile<ELFT> *File = C.getFile(); |
| 249 | ELFFile<ELFT> &EObj = File->getObj(); |
Rafael Espindola | 19e3889 | 2015-09-16 15:54:15 +0000 | [diff] [blame] | 250 | |
| 251 | if (!(C.getSectionHdr()->sh_flags & SHF_ALLOC)) |
| 252 | return; |
| 253 | |
| 254 | for (const Elf_Shdr *RelSec : C.RelocSections) { |
Rafael Espindola | 67a5da6 | 2015-09-17 14:02:10 +0000 | [diff] [blame] | 255 | if (RelSec->sh_type == SHT_RELA) |
Rafael Espindola | e1901cc | 2015-09-24 15:11:50 +0000 | [diff] [blame] | 256 | scanRelocs(C, EObj.relas(RelSec)); |
Rafael Espindola | 67a5da6 | 2015-09-17 14:02:10 +0000 | [diff] [blame] | 257 | else |
Rafael Espindola | e1901cc | 2015-09-24 15:11:50 +0000 | [diff] [blame] | 258 | scanRelocs(C, EObj.rels(RelSec)); |
Rafael Espindola | 19e3889 | 2015-09-16 15:54:15 +0000 | [diff] [blame] | 259 | } |
| 260 | } |
| 261 | |
Rafael Espindola | 6173f84 | 2015-09-23 14:37:01 +0000 | [diff] [blame] | 262 | template <class ELFT> |
Rui Ueyama | 44f5d91 | 2015-10-01 14:46:54 +0000 | [diff] [blame] | 263 | static void reportUndefined(const SymbolTable &S, const SymbolBody &Sym) { |
Rafael Espindola | 6173f84 | 2015-09-23 14:37:01 +0000 | [diff] [blame] | 264 | typedef typename ELFFile<ELFT>::Elf_Sym Elf_Sym; |
| 265 | typedef typename ELFFile<ELFT>::Elf_Sym_Range Elf_Sym_Range; |
| 266 | |
George Rimar | 57e40de | 2015-10-01 20:14:45 +0000 | [diff] [blame] | 267 | if (Config->Shared && !Config->NoUndefined) |
George Rimar | ee05828 | 2015-10-01 17:24:24 +0000 | [diff] [blame] | 268 | return; |
| 269 | |
Rafael Espindola | 6173f84 | 2015-09-23 14:37:01 +0000 | [diff] [blame] | 270 | const Elf_Sym &SymE = cast<ELFSymbolBody<ELFT>>(Sym).Sym; |
| 271 | ELFFileBase *SymFile = nullptr; |
| 272 | |
| 273 | for (const std::unique_ptr<ObjectFileBase> &F : S.getObjectFiles()) { |
| 274 | const auto &File = cast<ObjectFile<ELFT>>(*F); |
Rafael Espindola | e1901cc | 2015-09-24 15:11:50 +0000 | [diff] [blame] | 275 | Elf_Sym_Range Syms = File.getObj().symbols(File.getSymbolTable()); |
Rafael Espindola | 6173f84 | 2015-09-23 14:37:01 +0000 | [diff] [blame] | 276 | if (&SymE > Syms.begin() && &SymE < Syms.end()) |
| 277 | SymFile = F.get(); |
| 278 | } |
Rafael Espindola | 551dfd8 | 2015-09-25 19:24:57 +0000 | [diff] [blame] | 279 | |
| 280 | std::string Message = "undefined symbol: " + Sym.getName().str(); |
Rafael Espindola | 6173f84 | 2015-09-23 14:37:01 +0000 | [diff] [blame] | 281 | if (SymFile) |
Rafael Espindola | 551dfd8 | 2015-09-25 19:24:57 +0000 | [diff] [blame] | 282 | Message += " in " + SymFile->getName().str(); |
| 283 | if (Config->NoInhibitExec) |
| 284 | warning(Message); |
Rafael Espindola | 6173f84 | 2015-09-23 14:37:01 +0000 | [diff] [blame] | 285 | else |
Rafael Espindola | 551dfd8 | 2015-09-25 19:24:57 +0000 | [diff] [blame] | 286 | error(Message); |
Rafael Espindola | 6173f84 | 2015-09-23 14:37:01 +0000 | [diff] [blame] | 287 | } |
| 288 | |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 289 | // Create output section objects and add them to OutputSections. |
| 290 | template <class ELFT> void Writer<ELFT>::createSections() { |
Rafael Espindola | 83b0dc6 | 2015-08-13 22:21:37 +0000 | [diff] [blame] | 291 | SmallDenseMap<SectionKey<ELFT::Is64Bits>, OutputSection<ELFT> *> Map; |
Rafael Espindola | c2d2119 | 2015-09-23 18:25:05 +0000 | [diff] [blame] | 292 | |
| 293 | OutputSections.push_back(&BssSec); |
| 294 | Map[{BssSec.getName(), BssSec.getType(), BssSec.getFlags()}] = &BssSec; |
| 295 | |
Rafael Espindola | 0e604f9 | 2015-09-25 18:56:53 +0000 | [diff] [blame] | 296 | SymbolTable &Symtab = SymTabSec.getSymTable(); |
Rafael Espindola | 19e3889 | 2015-09-16 15:54:15 +0000 | [diff] [blame] | 297 | for (const std::unique_ptr<ObjectFileBase> &FileB : Symtab.getObjectFiles()) { |
| 298 | auto &File = cast<ObjectFile<ELFT>>(*FileB); |
Davide Italiano | 6d328d3 | 2015-09-16 20:45:57 +0000 | [diff] [blame] | 299 | if (!Config->DiscardAll) { |
| 300 | Elf_Sym_Range Syms = File.getLocalSymbols(); |
| 301 | for (const Elf_Sym &Sym : Syms) { |
| 302 | ErrorOr<StringRef> SymName = Sym.getName(File.getStringTable()); |
Rafael Espindola | d1cf421 | 2015-10-05 16:25:43 +0000 | [diff] [blame] | 303 | if (SymName && shouldKeepInSymtab<ELFT>(*SymName, Sym)) |
Davide Italiano | 6d328d3 | 2015-09-16 20:45:57 +0000 | [diff] [blame] | 304 | SymTabSec.addSymbol(*SymName, true); |
| 305 | } |
| 306 | } |
Rafael Espindola | 7167585 | 2015-09-22 00:16:19 +0000 | [diff] [blame] | 307 | for (InputSection<ELFT> *C : File.getSections()) { |
Rafael Espindola | 19e3889 | 2015-09-16 15:54:15 +0000 | [diff] [blame] | 308 | if (!C) |
| 309 | continue; |
| 310 | const Elf_Shdr *H = C->getSectionHdr(); |
Rafael Espindola | d13d960 | 2015-09-25 15:08:44 +0000 | [diff] [blame] | 311 | SectionKey<ELFT::Is64Bits> Key{C->getSectionName(), H->sh_type, |
| 312 | H->sh_flags}; |
| 313 | OutputSection<ELFT> *&Sec = Map[Key]; |
| 314 | if (!Sec) { |
| 315 | Sec = new (CAlloc.Allocate()) OutputSection<ELFT>( |
Rui Ueyama | 0fb1ee0 | 2015-10-02 21:13:19 +0000 | [diff] [blame] | 316 | PltSec, GotSec, BssSec, Key.Name, Key.Type, Key.Flags); |
Rafael Espindola | d13d960 | 2015-09-25 15:08:44 +0000 | [diff] [blame] | 317 | OutputSections.push_back(Sec); |
| 318 | } |
Rafael Espindola | 7167585 | 2015-09-22 00:16:19 +0000 | [diff] [blame] | 319 | Sec->addSection(C); |
Rafael Espindola | 19e3889 | 2015-09-16 15:54:15 +0000 | [diff] [blame] | 320 | scanRelocs(*C); |
| 321 | } |
| 322 | } |
| 323 | |
Rafael Espindola | 7757224 | 2015-10-02 19:37:55 +0000 | [diff] [blame] | 324 | DynamicSec.PreInitArraySec = |
| 325 | Map.lookup({".preinit_array", SHT_PREINIT_ARRAY, SHF_WRITE | SHF_ALLOC}); |
| 326 | DynamicSec.InitArraySec = |
| 327 | Map.lookup({".init_array", SHT_INIT_ARRAY, SHF_WRITE | SHF_ALLOC}); |
| 328 | DynamicSec.FiniArraySec = |
| 329 | Map.lookup({".fini_array", SHT_FINI_ARRAY, SHF_WRITE | SHF_ALLOC}); |
| 330 | |
| 331 | if (OutputSection<ELFT> *OS = DynamicSec.InitArraySec) { |
Rafael Espindola | 0e604f9 | 2015-09-25 18:56:53 +0000 | [diff] [blame] | 332 | Symtab.addSyntheticSym<ELFT>("__init_array_start", *OS, 0); |
| 333 | Symtab.addSyntheticSym<ELFT>("__init_array_end", *OS, OS->getSize()); |
Rafael Espindola | 5d41326 | 2015-10-01 21:22:26 +0000 | [diff] [blame] | 334 | } else { |
| 335 | Symtab.addIgnoredSym<ELFT>("__init_array_start"); |
| 336 | Symtab.addIgnoredSym<ELFT>("__init_array_end"); |
Rafael Espindola | 0e604f9 | 2015-09-25 18:56:53 +0000 | [diff] [blame] | 337 | } |
| 338 | |
Rafael Espindola | 05a3dd2 | 2015-09-22 23:38:23 +0000 | [diff] [blame] | 339 | // FIXME: Try to avoid the extra walk over all global symbols. |
| 340 | std::vector<DefinedCommon<ELFT> *> CommonSymbols; |
| 341 | for (auto &P : Symtab.getSymbols()) { |
| 342 | StringRef Name = P.first; |
| 343 | SymbolBody *Body = P.second->Body; |
Denis Protivensky | 22220d5 | 2015-10-05 09:43:57 +0000 | [diff] [blame] | 344 | if (auto *U = dyn_cast<Undefined<ELFT>>(Body)) { |
| 345 | if (!U->isWeak() && !U->canKeepUndefined()) |
| 346 | reportUndefined<ELFT>(Symtab, *Body); |
| 347 | } |
Rafael Espindola | 05a3dd2 | 2015-09-22 23:38:23 +0000 | [diff] [blame] | 348 | |
| 349 | if (auto *C = dyn_cast<DefinedCommon<ELFT>>(Body)) |
| 350 | CommonSymbols.push_back(C); |
Rafael Espindola | 4f674ed | 2015-10-05 15:24:04 +0000 | [diff] [blame] | 351 | if (!includeInSymtab<ELFT>(*Body)) |
Rafael Espindola | 05a3dd2 | 2015-09-22 23:38:23 +0000 | [diff] [blame] | 352 | continue; |
| 353 | SymTabSec.addSymbol(Name); |
| 354 | |
| 355 | if (needsDynamicSections() && includeInDynamicSymtab(*Body)) |
| 356 | HashSec.addSymbol(Body); |
| 357 | } |
| 358 | |
Rafael Espindola | b56cb94 | 2015-09-01 00:16:38 +0000 | [diff] [blame] | 359 | // Sort the common symbols by alignment as an heuristic to pack them better. |
Rui Ueyama | 3f4ec66 | 2015-09-25 03:48:25 +0000 | [diff] [blame] | 360 | std::stable_sort( |
| 361 | CommonSymbols.begin(), CommonSymbols.end(), |
| 362 | [](const DefinedCommon<ELFT> *A, const DefinedCommon<ELFT> *B) { |
| 363 | return A->MaxAlignment > B->MaxAlignment; |
| 364 | }); |
| 365 | |
Rafael Espindola | c2d2119 | 2015-09-23 18:25:05 +0000 | [diff] [blame] | 366 | uintX_t Off = BssSec.getSize(); |
Rafael Espindola | b56cb94 | 2015-09-01 00:16:38 +0000 | [diff] [blame] | 367 | for (DefinedCommon<ELFT> *C : CommonSymbols) { |
Rafael Espindola | 0518574 | 2015-08-31 22:07:18 +0000 | [diff] [blame] | 368 | const Elf_Sym &Sym = C->Sym; |
Rafael Espindola | f31f961 | 2015-09-01 01:19:12 +0000 | [diff] [blame] | 369 | uintX_t Align = C->MaxAlignment; |
Rafael Espindola | 0518574 | 2015-08-31 22:07:18 +0000 | [diff] [blame] | 370 | Off = RoundUpToAlignment(Off, Align); |
Rafael Espindola | ce8c9c0 | 2015-08-31 22:55:21 +0000 | [diff] [blame] | 371 | C->OffsetInBSS = Off; |
Rafael Espindola | 0518574 | 2015-08-31 22:07:18 +0000 | [diff] [blame] | 372 | Off += Sym.st_size; |
| 373 | } |
Rafael Espindola | b56cb94 | 2015-09-01 00:16:38 +0000 | [diff] [blame] | 374 | |
Rafael Espindola | c2d2119 | 2015-09-23 18:25:05 +0000 | [diff] [blame] | 375 | BssSec.setSize(Off); |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 376 | |
Rafael Espindola | fb81528 | 2015-09-11 00:30:13 +0000 | [diff] [blame] | 377 | OutputSections.push_back(&SymTabSec); |
Rafael Espindola | b01b574 | 2015-09-08 18:08:57 +0000 | [diff] [blame] | 378 | |
Rafael Espindola | 4340aad | 2015-09-11 22:42:45 +0000 | [diff] [blame] | 379 | if (needsDynamicSections()) { |
Rafael Espindola | 7010776 | 2015-09-11 18:49:42 +0000 | [diff] [blame] | 380 | if (needsInterpSection()) |
| 381 | OutputSections.push_back(&InterpSec); |
Rafael Espindola | fb81528 | 2015-09-11 00:30:13 +0000 | [diff] [blame] | 382 | OutputSections.push_back(&DynSymSec); |
Rafael Espindola | 601771e | 2015-09-14 20:20:34 +0000 | [diff] [blame] | 383 | OutputSections.push_back(&HashSec); |
Rafael Espindola | 740fafe | 2015-09-08 19:43:27 +0000 | [diff] [blame] | 384 | OutputSections.push_back(&DynamicSec); |
Rafael Espindola | 3f4228f | 2015-09-09 15:33:08 +0000 | [diff] [blame] | 385 | OutputSections.push_back(&DynStrSec); |
Denis Protivensky | 18add76 | 2015-09-17 09:54:29 +0000 | [diff] [blame] | 386 | if (RelaDynSec.hasRelocs()) |
Rafael Espindola | 19e3889 | 2015-09-16 15:54:15 +0000 | [diff] [blame] | 387 | OutputSections.push_back(&RelaDynSec); |
Rafael Espindola | 3f4228f | 2015-09-09 15:33:08 +0000 | [diff] [blame] | 388 | } |
Rafael Espindola | cdfecff | 2015-09-23 20:08:25 +0000 | [diff] [blame] | 389 | if (!GotSec.empty()) |
| 390 | OutputSections.push_back(&GotSec); |
| 391 | if (!PltSec.empty()) |
| 392 | OutputSections.push_back(&PltSec); |
Rafael Espindola | 740fafe | 2015-09-08 19:43:27 +0000 | [diff] [blame] | 393 | |
Rui Ueyama | 3f4ec66 | 2015-09-25 03:48:25 +0000 | [diff] [blame] | 394 | std::stable_sort( |
| 395 | OutputSections.begin(), OutputSections.end(), |
| 396 | [](OutputSectionBase<ELFT::Is64Bits> *A, |
| 397 | OutputSectionBase<ELFT::Is64Bits> *B) { |
Michael J. Spencer | 52bf0eb | 2015-10-01 21:15:02 +0000 | [diff] [blame] | 398 | uintX_t AFlags = A->getFlags(); |
| 399 | uintX_t BFlags = B->getFlags(); |
| 400 | |
| 401 | // Allocatable sections go first to reduce the total PT_LOAD size and |
| 402 | // so debug info doesn't change addresses in actual code. |
| 403 | bool AIsAlloc = AFlags & SHF_ALLOC; |
| 404 | bool BIsAlloc = BFlags & SHF_ALLOC; |
| 405 | if (AIsAlloc != BIsAlloc) |
| 406 | return AIsAlloc; |
| 407 | |
| 408 | // We don't have any special requirements for the relative order of |
| 409 | // two non allocatable sections. |
| 410 | if (!AIsAlloc) |
| 411 | return false; |
| 412 | |
| 413 | // We want the read only sections first so that they go in the PT_LOAD |
| 414 | // covering the program headers at the start of the file. |
| 415 | bool AIsWritable = AFlags & SHF_WRITE; |
| 416 | bool BIsWritable = BFlags & SHF_WRITE; |
| 417 | if (AIsWritable != BIsWritable) |
| 418 | return BIsWritable; |
| 419 | |
| 420 | // For a corresponding reason, put non exec sections first (the program |
| 421 | // header PT_LOAD is not executable). |
| 422 | bool AIsExec = AFlags & SHF_EXECINSTR; |
| 423 | bool BIsExec = BFlags & SHF_EXECINSTR; |
| 424 | if (AIsExec != BIsExec) |
| 425 | return BIsExec; |
| 426 | |
| 427 | // If we got here we know that both A and B and in the same PT_LOAD. |
| 428 | // The last requirement we have is to put nobits section last. The |
| 429 | // reason is that the only thing the dynamic linker will see about |
| 430 | // them is a p_memsz that is larger than p_filesz. Seeing that it |
| 431 | // zeros the end of the PT_LOAD, so that has to correspond to the |
| 432 | // nobits sections. |
| 433 | return A->getType() != SHT_NOBITS && B->getType() == SHT_NOBITS; |
Rui Ueyama | 3f4ec66 | 2015-09-25 03:48:25 +0000 | [diff] [blame] | 434 | }); |
| 435 | |
Michael J. Spencer | 52bf0eb | 2015-10-01 21:15:02 +0000 | [diff] [blame] | 436 | // Always put StrTabSec last so that no section names are added to it after |
| 437 | // it's finalized. |
| 438 | OutputSections.push_back(&StrTabSec); |
| 439 | |
Rafael Espindola | b01b574 | 2015-09-08 18:08:57 +0000 | [diff] [blame] | 440 | for (unsigned I = 0, N = OutputSections.size(); I < N; ++I) |
| 441 | OutputSections[I]->setSectionIndex(I + 1); |
Michael J. Spencer | 52bf0eb | 2015-10-01 21:15:02 +0000 | [diff] [blame] | 442 | |
| 443 | // Fill the DynStrSec early. |
| 444 | DynamicSec.finalize(); |
Rafael Espindola | abad618 | 2015-08-13 15:23:46 +0000 | [diff] [blame] | 445 | } |
| 446 | |
Michael J. Spencer | 1d299a8 | 2015-09-09 20:48:09 +0000 | [diff] [blame] | 447 | template <class ELFT> |
Rui Ueyama | 04dae55 | 2015-10-02 21:23:17 +0000 | [diff] [blame] | 448 | static bool needsPHDR(OutputSectionBase<ELFT::Is64Bits> *Sec) { |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 449 | return Sec->getFlags() & SHF_ALLOC; |
Michael J. Spencer | 1d299a8 | 2015-09-09 20:48:09 +0000 | [diff] [blame] | 450 | } |
| 451 | |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 452 | // Visits all sections to assign incremental, non-overlapping RVAs and |
| 453 | // file offsets. |
| 454 | template <class ELFT> void Writer<ELFT>::assignAddresses() { |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 455 | assert(!OutputSections.empty() && "No output sections to layout!"); |
Rafael Espindola | 4340aad | 2015-09-11 22:42:45 +0000 | [diff] [blame] | 456 | uintX_t VA = getVAStart(); |
Rafael Espindola | 0a2e211 | 2015-09-10 15:41:34 +0000 | [diff] [blame] | 457 | uintX_t FileOff = 0; |
Rafael Espindola | 60252d8 | 2015-09-09 22:53:55 +0000 | [diff] [blame] | 458 | |
Rafael Espindola | 0a2e211 | 2015-09-10 15:41:34 +0000 | [diff] [blame] | 459 | FileOff += sizeof(Elf_Ehdr); |
| 460 | VA += sizeof(Elf_Ehdr); |
Rafael Espindola | abad618 | 2015-08-13 15:23:46 +0000 | [diff] [blame] | 461 | |
Michael J. Spencer | 1d299a8 | 2015-09-09 20:48:09 +0000 | [diff] [blame] | 462 | // Reserve space for PHDRs. |
| 463 | ProgramHeaderOff = FileOff; |
| 464 | FileOff = RoundUpToAlignment(FileOff, PageSize); |
Rafael Espindola | 0a2e211 | 2015-09-10 15:41:34 +0000 | [diff] [blame] | 465 | VA = RoundUpToAlignment(VA, PageSize); |
Michael J. Spencer | 1d299a8 | 2015-09-09 20:48:09 +0000 | [diff] [blame] | 466 | |
Rafael Espindola | 7010776 | 2015-09-11 18:49:42 +0000 | [diff] [blame] | 467 | if (needsInterpSection()) |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 468 | PHDRs.push_back(&InterpPHDR); |
Rafael Espindola | 7010776 | 2015-09-11 18:49:42 +0000 | [diff] [blame] | 469 | |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 470 | // Create a PHDR for the file header. |
| 471 | PHDRs.push_back(&FileHeaderPHDR); |
| 472 | FileHeaderPHDR.Header.p_vaddr = getVAStart(); |
| 473 | FileHeaderPHDR.Header.p_paddr = getVAStart(); |
| 474 | FileHeaderPHDR.Header.p_align = PageSize; |
Rafael Espindola | 0a2e211 | 2015-09-10 15:41:34 +0000 | [diff] [blame] | 475 | |
Rafael Espindola | ebd2108 | 2015-08-13 22:14:37 +0000 | [diff] [blame] | 476 | for (OutputSectionBase<ELFT::Is64Bits> *Sec : OutputSections) { |
Rafael Espindola | 3f4228f | 2015-09-09 15:33:08 +0000 | [diff] [blame] | 477 | StrTabSec.add(Sec->getName()); |
Rafael Espindola | ebd2108 | 2015-08-13 22:14:37 +0000 | [diff] [blame] | 478 | Sec->finalize(); |
| 479 | |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 480 | if (Sec->getSize()) { |
Rui Ueyama | 04dae55 | 2015-10-02 21:23:17 +0000 | [diff] [blame] | 481 | uintX_t Flags = toPHDRFlags(Sec->getFlags()); |
Rui Ueyama | 5713b8d | 2015-10-02 22:17:07 +0000 | [diff] [blame] | 482 | ProgramHeader<ELFT> *Last = PHDRs.back(); |
Rui Ueyama | 04dae55 | 2015-10-02 21:23:17 +0000 | [diff] [blame] | 483 | if (Last->Header.p_flags != Flags || !needsPHDR<ELFT>(Sec)) { |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 484 | // Flags changed. End current PHDR and potentially create a new one. |
Rui Ueyama | 04dae55 | 2015-10-02 21:23:17 +0000 | [diff] [blame] | 485 | if (!Last->Closed) { |
| 486 | Last->Header.p_filesz = FileOff - Last->Header.p_offset; |
| 487 | Last->Header.p_memsz = VA - Last->Header.p_vaddr; |
| 488 | Last->Closed = true; |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 489 | } |
| 490 | |
Rui Ueyama | 04dae55 | 2015-10-02 21:23:17 +0000 | [diff] [blame] | 491 | if (needsPHDR<ELFT>(Sec)) { |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 492 | VA = RoundUpToAlignment(VA, PageSize); |
| 493 | FileOff = RoundUpToAlignment(FileOff, PageSize); |
Rui Ueyama | 5713b8d | 2015-10-02 22:17:07 +0000 | [diff] [blame] | 494 | PHDRs.push_back(new (PAlloc) |
| 495 | ProgramHeader<ELFT>(PT_LOAD, Flags, FileOff, VA)); |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 496 | } |
| 497 | } |
Michael J. Spencer | 1d299a8 | 2015-09-09 20:48:09 +0000 | [diff] [blame] | 498 | } |
| 499 | |
Rafael Espindola | 2db634d | 2015-08-13 20:24:18 +0000 | [diff] [blame] | 500 | uintX_t Align = Sec->getAlign(); |
| 501 | uintX_t Size = Sec->getSize(); |
Rafael Espindola | ef1ac01 | 2015-08-13 15:31:17 +0000 | [diff] [blame] | 502 | if (Sec->getFlags() & SHF_ALLOC) { |
Rafael Espindola | bfcdfb3 | 2015-09-14 19:00:35 +0000 | [diff] [blame] | 503 | VA = RoundUpToAlignment(VA, Align); |
Rafael Espindola | ef1ac01 | 2015-08-13 15:31:17 +0000 | [diff] [blame] | 504 | Sec->setVA(VA); |
Rafael Espindola | bfcdfb3 | 2015-09-14 19:00:35 +0000 | [diff] [blame] | 505 | VA += Size; |
Rafael Espindola | ef1ac01 | 2015-08-13 15:31:17 +0000 | [diff] [blame] | 506 | } |
Rafael Espindola | bfcdfb3 | 2015-09-14 19:00:35 +0000 | [diff] [blame] | 507 | FileOff = RoundUpToAlignment(FileOff, Align); |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 508 | Sec->setFileOffset(FileOff); |
Rafael Espindola | 058f343 | 2015-08-31 20:23:57 +0000 | [diff] [blame] | 509 | if (Sec->getType() != SHT_NOBITS) |
Rafael Espindola | bfcdfb3 | 2015-09-14 19:00:35 +0000 | [diff] [blame] | 510 | FileOff += Size; |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 511 | } |
Rafael Espindola | 6b83b90 | 2015-08-12 00:00:24 +0000 | [diff] [blame] | 512 | |
Michael J. Spencer | 1d299a8 | 2015-09-09 20:48:09 +0000 | [diff] [blame] | 513 | // Add a PHDR for the dynamic table. |
Rafael Espindola | 4340aad | 2015-09-11 22:42:45 +0000 | [diff] [blame] | 514 | if (needsDynamicSections()) |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 515 | PHDRs.push_back(&DynamicPHDR); |
Michael J. Spencer | 1d299a8 | 2015-09-09 20:48:09 +0000 | [diff] [blame] | 516 | |
Rafael Espindola | 91009b3 | 2015-08-12 01:45:28 +0000 | [diff] [blame] | 517 | FileOff += OffsetToAlignment(FileOff, ELFT::Is64Bits ? 8 : 4); |
| 518 | |
Michael J. Spencer | 8039dae2 | 2015-07-29 00:30:10 +0000 | [diff] [blame] | 519 | // Add space for section headers. |
| 520 | SectionHeaderOff = FileOff; |
Rafael Espindola | 18608a0 | 2015-09-08 21:57:31 +0000 | [diff] [blame] | 521 | FileOff += getNumSections() * sizeof(Elf_Shdr); |
Michael J. Spencer | 1d299a8 | 2015-09-09 20:48:09 +0000 | [diff] [blame] | 522 | FileSize = FileOff; |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 523 | } |
| 524 | |
| 525 | template <class ELFT> void Writer<ELFT>::writeHeader() { |
| 526 | uint8_t *Buf = Buffer->getBufferStart(); |
Rafael Espindola | 18608a0 | 2015-09-08 21:57:31 +0000 | [diff] [blame] | 527 | auto *EHdr = reinterpret_cast<Elf_Ehdr *>(Buf); |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 528 | EHdr->e_ident[EI_MAG0] = 0x7F; |
| 529 | EHdr->e_ident[EI_MAG1] = 0x45; |
| 530 | EHdr->e_ident[EI_MAG2] = 0x4C; |
| 531 | EHdr->e_ident[EI_MAG3] = 0x46; |
Rafael Espindola | 4b7c2fc | 2015-08-05 15:08:40 +0000 | [diff] [blame] | 532 | EHdr->e_ident[EI_CLASS] = ELFT::Is64Bits ? ELFCLASS64 : ELFCLASS32; |
| 533 | EHdr->e_ident[EI_DATA] = ELFT::TargetEndianness == llvm::support::little |
| 534 | ? ELFDATA2LSB |
| 535 | : ELFDATA2MSB; |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 536 | EHdr->e_ident[EI_VERSION] = EV_CURRENT; |
Davide Italiano | aa7c533 | 2015-09-25 01:59:13 +0000 | [diff] [blame] | 537 | |
| 538 | const SymbolTable &Symtab = SymTabSec.getSymTable(); |
| 539 | auto &FirstObj = cast<ObjectFile<ELFT>>(*Symtab.getFirstELF()); |
| 540 | EHdr->e_ident[EI_OSABI] = FirstObj.getOSABI(); |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 541 | |
Rafael Espindola | e438e07 | 2015-09-08 22:55:28 +0000 | [diff] [blame] | 542 | // FIXME: Generalize the segment construction similar to how we create |
| 543 | // output sections. |
Rafael Espindola | e438e07 | 2015-09-08 22:55:28 +0000 | [diff] [blame] | 544 | |
Rafael Espindola | 4340aad | 2015-09-11 22:42:45 +0000 | [diff] [blame] | 545 | EHdr->e_type = Config->Shared ? ET_DYN : ET_EXEC; |
Rafael Espindola | f98d6d8 | 2015-09-03 20:03:54 +0000 | [diff] [blame] | 546 | EHdr->e_machine = FirstObj.getEMachine(); |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 547 | EHdr->e_version = EV_CURRENT; |
Rafael Espindola | 4340aad | 2015-09-11 22:42:45 +0000 | [diff] [blame] | 548 | SymbolBody *Entry = Symtab.getEntrySym(); |
Rafael Espindola | cd076f0 | 2015-09-25 18:19:03 +0000 | [diff] [blame] | 549 | EHdr->e_entry = |
| 550 | Entry ? getSymVA(cast<ELFSymbolBody<ELFT>>(*Entry), BssSec) : 0; |
Michael J. Spencer | 1d299a8 | 2015-09-09 20:48:09 +0000 | [diff] [blame] | 551 | EHdr->e_phoff = ProgramHeaderOff; |
Michael J. Spencer | 8039dae2 | 2015-07-29 00:30:10 +0000 | [diff] [blame] | 552 | EHdr->e_shoff = SectionHeaderOff; |
Rafael Espindola | 18608a0 | 2015-09-08 21:57:31 +0000 | [diff] [blame] | 553 | EHdr->e_ehsize = sizeof(Elf_Ehdr); |
| 554 | EHdr->e_phentsize = sizeof(Elf_Phdr); |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 555 | EHdr->e_phnum = PHDRs.size(); |
Rafael Espindola | 18608a0 | 2015-09-08 21:57:31 +0000 | [diff] [blame] | 556 | EHdr->e_shentsize = sizeof(Elf_Shdr); |
Rafael Espindola | 5f55387 | 2015-09-08 17:39:39 +0000 | [diff] [blame] | 557 | EHdr->e_shnum = getNumSections(); |
Rafael Espindola | 3f4228f | 2015-09-09 15:33:08 +0000 | [diff] [blame] | 558 | EHdr->e_shstrndx = StrTabSec.getSectionIndex(); |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 559 | |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 560 | // If nothing was merged into the file header PT_LOAD, set the size correctly. |
Rui Ueyama | 5713b8d | 2015-10-02 22:17:07 +0000 | [diff] [blame] | 561 | if (FileHeaderPHDR.Header.p_filesz == PageSize) { |
| 562 | uint64_t Size = sizeof(Elf_Ehdr) + sizeof(Elf_Phdr) * PHDRs.size(); |
| 563 | FileHeaderPHDR.Header.p_filesz = Size; |
| 564 | FileHeaderPHDR.Header.p_memsz = Size; |
| 565 | } |
Michael J. Spencer | 2f00824 | 2015-09-17 19:58:07 +0000 | [diff] [blame] | 566 | |
| 567 | if (needsInterpSection()) |
| 568 | InterpPHDR.setValuesFromSection(InterpSec); |
| 569 | if (needsDynamicSections()) |
| 570 | DynamicPHDR.setValuesFromSection(DynamicSec); |
| 571 | |
Rafael Espindola | 18608a0 | 2015-09-08 21:57:31 +0000 | [diff] [blame] | 572 | auto PHdrs = reinterpret_cast<Elf_Phdr *>(Buf + EHdr->e_phoff); |
Rui Ueyama | 5713b8d | 2015-10-02 22:17:07 +0000 | [diff] [blame] | 573 | for (ProgramHeader<ELFT> *PHDR : PHDRs) |
| 574 | *PHdrs++ = PHDR->Header; |
Rafael Espindola | e438e07 | 2015-09-08 22:55:28 +0000 | [diff] [blame] | 575 | |
Rafael Espindola | 18608a0 | 2015-09-08 21:57:31 +0000 | [diff] [blame] | 576 | auto SHdrs = reinterpret_cast<Elf_Shdr *>(Buf + EHdr->e_shoff); |
Michael J. Spencer | 8039dae2 | 2015-07-29 00:30:10 +0000 | [diff] [blame] | 577 | // First entry is null. |
| 578 | ++SHdrs; |
Rafael Espindola | ebd2108 | 2015-08-13 22:14:37 +0000 | [diff] [blame] | 579 | for (OutputSectionBase<ELFT::Is64Bits> *Sec : OutputSections) { |
Rafael Espindola | 3f4228f | 2015-09-09 15:33:08 +0000 | [diff] [blame] | 580 | Sec->setNameOffset(StrTabSec.getFileOff(Sec->getName())); |
Rafael Espindola | a175eb6 | 2015-08-13 18:37:23 +0000 | [diff] [blame] | 581 | Sec->template writeHeaderTo<ELFT::TargetEndianness>(SHdrs++); |
Rafael Espindola | 6b83b90 | 2015-08-12 00:00:24 +0000 | [diff] [blame] | 582 | } |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 583 | } |
| 584 | |
| 585 | template <class ELFT> void Writer<ELFT>::openFile(StringRef Path) { |
Rafael Espindola | bdc8f2f | 2015-08-13 00:31:46 +0000 | [diff] [blame] | 586 | ErrorOr<std::unique_ptr<FileOutputBuffer>> BufferOrErr = |
| 587 | FileOutputBuffer::create(Path, FileSize, FileOutputBuffer::F_executable); |
| 588 | error(BufferOrErr, Twine("failed to open ") + Path); |
| 589 | Buffer = std::move(*BufferOrErr); |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 590 | } |
| 591 | |
| 592 | // Write section contents to a mmap'ed file. |
| 593 | template <class ELFT> void Writer<ELFT>::writeSections() { |
| 594 | uint8_t *Buf = Buffer->getBufferStart(); |
Rafael Espindola | ebd2108 | 2015-08-13 22:14:37 +0000 | [diff] [blame] | 595 | for (OutputSectionBase<ELFT::Is64Bits> *Sec : OutputSections) |
Rui Ueyama | 8050d32 | 2015-08-14 05:17:30 +0000 | [diff] [blame] | 596 | Sec->writeTo(Buf + Sec->getFileOff()); |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 597 | } |