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