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