Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1 | //===- ELFObjectFile.h - ELF object file implementation ---------*- C++ -*-===// |
| 2 | // |
Andrew Walbran | 16937d0 | 2019-10-22 13:54:20 +0100 | [diff] [blame] | 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 4 | // See https://llvm.org/LICENSE.txt for license information. |
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 6 | // |
| 7 | //===----------------------------------------------------------------------===// |
| 8 | // |
| 9 | // This file declares the ELFObjectFile template class. |
| 10 | // |
| 11 | //===----------------------------------------------------------------------===// |
| 12 | |
| 13 | #ifndef LLVM_OBJECT_ELFOBJECTFILE_H |
| 14 | #define LLVM_OBJECT_ELFOBJECTFILE_H |
| 15 | |
| 16 | #include "llvm/ADT/ArrayRef.h" |
Andrew Scull | cdfcccc | 2018-10-05 20:58:37 +0100 | [diff] [blame] | 17 | #include "llvm/ADT/STLExtras.h" |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 18 | #include "llvm/ADT/SmallVector.h" |
| 19 | #include "llvm/ADT/StringRef.h" |
| 20 | #include "llvm/ADT/Triple.h" |
| 21 | #include "llvm/ADT/iterator_range.h" |
| 22 | #include "llvm/BinaryFormat/ELF.h" |
| 23 | #include "llvm/MC/SubtargetFeature.h" |
| 24 | #include "llvm/Object/Binary.h" |
| 25 | #include "llvm/Object/ELF.h" |
| 26 | #include "llvm/Object/ELFTypes.h" |
| 27 | #include "llvm/Object/Error.h" |
| 28 | #include "llvm/Object/ObjectFile.h" |
| 29 | #include "llvm/Object/SymbolicFile.h" |
| 30 | #include "llvm/Support/ARMAttributeParser.h" |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 31 | #include "llvm/Support/Casting.h" |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 32 | #include "llvm/Support/ELFAttributes.h" |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 33 | #include "llvm/Support/Endian.h" |
| 34 | #include "llvm/Support/Error.h" |
| 35 | #include "llvm/Support/ErrorHandling.h" |
| 36 | #include "llvm/Support/MemoryBuffer.h" |
| 37 | #include <cassert> |
| 38 | #include <cstdint> |
| 39 | #include <system_error> |
| 40 | |
| 41 | namespace llvm { |
| 42 | namespace object { |
| 43 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 44 | constexpr int NumElfSymbolTypes = 16; |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 45 | extern const llvm::EnumEntry<unsigned> ElfSymbolTypes[NumElfSymbolTypes]; |
| 46 | |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 47 | class elf_symbol_iterator; |
| 48 | |
| 49 | class ELFObjectFileBase : public ObjectFile { |
| 50 | friend class ELFRelocationRef; |
| 51 | friend class ELFSectionRef; |
| 52 | friend class ELFSymbolRef; |
| 53 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 54 | SubtargetFeatures getMIPSFeatures() const; |
| 55 | SubtargetFeatures getARMFeatures() const; |
| 56 | SubtargetFeatures getRISCVFeatures() const; |
| 57 | |
| 58 | StringRef getAMDGPUCPUName() const; |
| 59 | |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 60 | protected: |
| 61 | ELFObjectFileBase(unsigned int Type, MemoryBufferRef Source); |
| 62 | |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 63 | virtual uint64_t getSymbolSize(DataRefImpl Symb) const = 0; |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 64 | virtual uint8_t getSymbolBinding(DataRefImpl Symb) const = 0; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 65 | virtual uint8_t getSymbolOther(DataRefImpl Symb) const = 0; |
| 66 | virtual uint8_t getSymbolELFType(DataRefImpl Symb) const = 0; |
| 67 | |
| 68 | virtual uint32_t getSectionType(DataRefImpl Sec) const = 0; |
| 69 | virtual uint64_t getSectionFlags(DataRefImpl Sec) const = 0; |
| 70 | virtual uint64_t getSectionOffset(DataRefImpl Sec) const = 0; |
| 71 | |
| 72 | virtual Expected<int64_t> getRelocationAddend(DataRefImpl Rel) const = 0; |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 73 | virtual Error getBuildAttributes(ELFAttributeParser &Attributes) const = 0; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 74 | |
| 75 | public: |
| 76 | using elf_symbol_iterator_range = iterator_range<elf_symbol_iterator>; |
| 77 | |
| 78 | virtual elf_symbol_iterator_range getDynamicSymbolIterators() const = 0; |
| 79 | |
| 80 | /// Returns platform-specific object flags, if any. |
| 81 | virtual unsigned getPlatformFlags() const = 0; |
| 82 | |
| 83 | elf_symbol_iterator_range symbols() const; |
| 84 | |
| 85 | static bool classof(const Binary *v) { return v->isELF(); } |
| 86 | |
| 87 | SubtargetFeatures getFeatures() const override; |
| 88 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 89 | Optional<StringRef> tryGetCPUName() const override; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 90 | |
| 91 | void setARMSubArch(Triple &TheTriple) const override; |
Andrew Scull | cdfcccc | 2018-10-05 20:58:37 +0100 | [diff] [blame] | 92 | |
| 93 | virtual uint16_t getEType() const = 0; |
| 94 | |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 95 | virtual uint16_t getEMachine() const = 0; |
| 96 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 97 | std::vector<std::pair<Optional<DataRefImpl>, uint64_t>> |
| 98 | getPltAddresses() const; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 99 | }; |
| 100 | |
| 101 | class ELFSectionRef : public SectionRef { |
| 102 | public: |
| 103 | ELFSectionRef(const SectionRef &B) : SectionRef(B) { |
| 104 | assert(isa<ELFObjectFileBase>(SectionRef::getObject())); |
| 105 | } |
| 106 | |
| 107 | const ELFObjectFileBase *getObject() const { |
| 108 | return cast<ELFObjectFileBase>(SectionRef::getObject()); |
| 109 | } |
| 110 | |
| 111 | uint32_t getType() const { |
| 112 | return getObject()->getSectionType(getRawDataRefImpl()); |
| 113 | } |
| 114 | |
| 115 | uint64_t getFlags() const { |
| 116 | return getObject()->getSectionFlags(getRawDataRefImpl()); |
| 117 | } |
| 118 | |
| 119 | uint64_t getOffset() const { |
| 120 | return getObject()->getSectionOffset(getRawDataRefImpl()); |
| 121 | } |
| 122 | }; |
| 123 | |
| 124 | class elf_section_iterator : public section_iterator { |
| 125 | public: |
| 126 | elf_section_iterator(const section_iterator &B) : section_iterator(B) { |
| 127 | assert(isa<ELFObjectFileBase>(B->getObject())); |
| 128 | } |
| 129 | |
| 130 | const ELFSectionRef *operator->() const { |
| 131 | return static_cast<const ELFSectionRef *>(section_iterator::operator->()); |
| 132 | } |
| 133 | |
| 134 | const ELFSectionRef &operator*() const { |
| 135 | return static_cast<const ELFSectionRef &>(section_iterator::operator*()); |
| 136 | } |
| 137 | }; |
| 138 | |
| 139 | class ELFSymbolRef : public SymbolRef { |
| 140 | public: |
| 141 | ELFSymbolRef(const SymbolRef &B) : SymbolRef(B) { |
| 142 | assert(isa<ELFObjectFileBase>(SymbolRef::getObject())); |
| 143 | } |
| 144 | |
| 145 | const ELFObjectFileBase *getObject() const { |
| 146 | return cast<ELFObjectFileBase>(BasicSymbolRef::getObject()); |
| 147 | } |
| 148 | |
| 149 | uint64_t getSize() const { |
| 150 | return getObject()->getSymbolSize(getRawDataRefImpl()); |
| 151 | } |
| 152 | |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 153 | uint8_t getBinding() const { |
| 154 | return getObject()->getSymbolBinding(getRawDataRefImpl()); |
| 155 | } |
| 156 | |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 157 | uint8_t getOther() const { |
| 158 | return getObject()->getSymbolOther(getRawDataRefImpl()); |
| 159 | } |
| 160 | |
| 161 | uint8_t getELFType() const { |
| 162 | return getObject()->getSymbolELFType(getRawDataRefImpl()); |
| 163 | } |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 164 | |
| 165 | StringRef getELFTypeName() const { |
| 166 | uint8_t Type = getELFType(); |
| 167 | for (auto &EE : ElfSymbolTypes) { |
| 168 | if (EE.Value == Type) { |
| 169 | return EE.AltName; |
| 170 | } |
| 171 | } |
| 172 | return ""; |
| 173 | } |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 174 | }; |
| 175 | |
| 176 | class elf_symbol_iterator : public symbol_iterator { |
| 177 | public: |
| 178 | elf_symbol_iterator(const basic_symbol_iterator &B) |
| 179 | : symbol_iterator(SymbolRef(B->getRawDataRefImpl(), |
| 180 | cast<ELFObjectFileBase>(B->getObject()))) {} |
| 181 | |
| 182 | const ELFSymbolRef *operator->() const { |
| 183 | return static_cast<const ELFSymbolRef *>(symbol_iterator::operator->()); |
| 184 | } |
| 185 | |
| 186 | const ELFSymbolRef &operator*() const { |
| 187 | return static_cast<const ELFSymbolRef &>(symbol_iterator::operator*()); |
| 188 | } |
| 189 | }; |
| 190 | |
| 191 | class ELFRelocationRef : public RelocationRef { |
| 192 | public: |
| 193 | ELFRelocationRef(const RelocationRef &B) : RelocationRef(B) { |
| 194 | assert(isa<ELFObjectFileBase>(RelocationRef::getObject())); |
| 195 | } |
| 196 | |
| 197 | const ELFObjectFileBase *getObject() const { |
| 198 | return cast<ELFObjectFileBase>(RelocationRef::getObject()); |
| 199 | } |
| 200 | |
| 201 | Expected<int64_t> getAddend() const { |
| 202 | return getObject()->getRelocationAddend(getRawDataRefImpl()); |
| 203 | } |
| 204 | }; |
| 205 | |
| 206 | class elf_relocation_iterator : public relocation_iterator { |
| 207 | public: |
| 208 | elf_relocation_iterator(const relocation_iterator &B) |
| 209 | : relocation_iterator(RelocationRef( |
| 210 | B->getRawDataRefImpl(), cast<ELFObjectFileBase>(B->getObject()))) {} |
| 211 | |
| 212 | const ELFRelocationRef *operator->() const { |
| 213 | return static_cast<const ELFRelocationRef *>( |
| 214 | relocation_iterator::operator->()); |
| 215 | } |
| 216 | |
| 217 | const ELFRelocationRef &operator*() const { |
| 218 | return static_cast<const ELFRelocationRef &>( |
| 219 | relocation_iterator::operator*()); |
| 220 | } |
| 221 | }; |
| 222 | |
| 223 | inline ELFObjectFileBase::elf_symbol_iterator_range |
| 224 | ELFObjectFileBase::symbols() const { |
| 225 | return elf_symbol_iterator_range(symbol_begin(), symbol_end()); |
| 226 | } |
| 227 | |
| 228 | template <class ELFT> class ELFObjectFile : public ELFObjectFileBase { |
| 229 | uint16_t getEMachine() const override; |
Andrew Scull | cdfcccc | 2018-10-05 20:58:37 +0100 | [diff] [blame] | 230 | uint16_t getEType() const override; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 231 | uint64_t getSymbolSize(DataRefImpl Sym) const override; |
| 232 | |
| 233 | public: |
| 234 | LLVM_ELF_IMPORT_TYPES_ELFT(ELFT) |
| 235 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 236 | SectionRef toSectionRef(const Elf_Shdr *Sec) const { |
| 237 | return SectionRef(toDRI(Sec), this); |
| 238 | } |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 239 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 240 | ELFSymbolRef toSymbolRef(const Elf_Shdr *SymTable, unsigned SymbolNum) const { |
| 241 | return ELFSymbolRef({toDRI(SymTable, SymbolNum), this}); |
| 242 | } |
| 243 | |
| 244 | bool IsContentValid() const { return ContentValid; } |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 245 | |
| 246 | private: |
| 247 | ELFObjectFile(MemoryBufferRef Object, ELFFile<ELFT> EF, |
| 248 | const Elf_Shdr *DotDynSymSec, const Elf_Shdr *DotSymtabSec, |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 249 | const Elf_Shdr *DotSymtabShndxSec); |
| 250 | |
| 251 | bool ContentValid = false; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 252 | |
| 253 | protected: |
| 254 | ELFFile<ELFT> EF; |
| 255 | |
| 256 | const Elf_Shdr *DotDynSymSec = nullptr; // Dynamic symbol table section. |
| 257 | const Elf_Shdr *DotSymtabSec = nullptr; // Symbol table section. |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 258 | const Elf_Shdr *DotSymtabShndxSec = nullptr; // SHT_SYMTAB_SHNDX section. |
| 259 | |
| 260 | Error initContent() override; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 261 | |
| 262 | void moveSymbolNext(DataRefImpl &Symb) const override; |
| 263 | Expected<StringRef> getSymbolName(DataRefImpl Symb) const override; |
| 264 | Expected<uint64_t> getSymbolAddress(DataRefImpl Symb) const override; |
| 265 | uint64_t getSymbolValueImpl(DataRefImpl Symb) const override; |
| 266 | uint32_t getSymbolAlignment(DataRefImpl Symb) const override; |
| 267 | uint64_t getCommonSymbolSizeImpl(DataRefImpl Symb) const override; |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 268 | Expected<uint32_t> getSymbolFlags(DataRefImpl Symb) const override; |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 269 | uint8_t getSymbolBinding(DataRefImpl Symb) const override; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 270 | uint8_t getSymbolOther(DataRefImpl Symb) const override; |
| 271 | uint8_t getSymbolELFType(DataRefImpl Symb) const override; |
| 272 | Expected<SymbolRef::Type> getSymbolType(DataRefImpl Symb) const override; |
| 273 | Expected<section_iterator> getSymbolSection(const Elf_Sym *Symb, |
| 274 | const Elf_Shdr *SymTab) const; |
| 275 | Expected<section_iterator> getSymbolSection(DataRefImpl Symb) const override; |
| 276 | |
| 277 | void moveSectionNext(DataRefImpl &Sec) const override; |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 278 | Expected<StringRef> getSectionName(DataRefImpl Sec) const override; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 279 | uint64_t getSectionAddress(DataRefImpl Sec) const override; |
| 280 | uint64_t getSectionIndex(DataRefImpl Sec) const override; |
| 281 | uint64_t getSectionSize(DataRefImpl Sec) const override; |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 282 | Expected<ArrayRef<uint8_t>> |
| 283 | getSectionContents(DataRefImpl Sec) const override; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 284 | uint64_t getSectionAlignment(DataRefImpl Sec) const override; |
| 285 | bool isSectionCompressed(DataRefImpl Sec) const override; |
| 286 | bool isSectionText(DataRefImpl Sec) const override; |
| 287 | bool isSectionData(DataRefImpl Sec) const override; |
| 288 | bool isSectionBSS(DataRefImpl Sec) const override; |
| 289 | bool isSectionVirtual(DataRefImpl Sec) const override; |
Andrew Walbran | 16937d0 | 2019-10-22 13:54:20 +0100 | [diff] [blame] | 290 | bool isBerkeleyText(DataRefImpl Sec) const override; |
| 291 | bool isBerkeleyData(DataRefImpl Sec) const override; |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 292 | bool isDebugSection(StringRef SectionName) const override; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 293 | relocation_iterator section_rel_begin(DataRefImpl Sec) const override; |
| 294 | relocation_iterator section_rel_end(DataRefImpl Sec) const override; |
Andrew Scull | cdfcccc | 2018-10-05 20:58:37 +0100 | [diff] [blame] | 295 | std::vector<SectionRef> dynamic_relocation_sections() const override; |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 296 | Expected<section_iterator> |
| 297 | getRelocatedSection(DataRefImpl Sec) const override; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 298 | |
| 299 | void moveRelocationNext(DataRefImpl &Rel) const override; |
| 300 | uint64_t getRelocationOffset(DataRefImpl Rel) const override; |
| 301 | symbol_iterator getRelocationSymbol(DataRefImpl Rel) const override; |
| 302 | uint64_t getRelocationType(DataRefImpl Rel) const override; |
| 303 | void getRelocationTypeName(DataRefImpl Rel, |
| 304 | SmallVectorImpl<char> &Result) const override; |
| 305 | |
| 306 | uint32_t getSectionType(DataRefImpl Sec) const override; |
| 307 | uint64_t getSectionFlags(DataRefImpl Sec) const override; |
| 308 | uint64_t getSectionOffset(DataRefImpl Sec) const override; |
| 309 | StringRef getRelocationTypeName(uint32_t Type) const; |
| 310 | |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 311 | DataRefImpl toDRI(const Elf_Shdr *SymTable, unsigned SymbolNum) const { |
| 312 | DataRefImpl DRI; |
| 313 | if (!SymTable) { |
| 314 | DRI.d.a = 0; |
| 315 | DRI.d.b = 0; |
| 316 | return DRI; |
| 317 | } |
| 318 | assert(SymTable->sh_type == ELF::SHT_SYMTAB || |
| 319 | SymTable->sh_type == ELF::SHT_DYNSYM); |
| 320 | |
| 321 | auto SectionsOrErr = EF.sections(); |
| 322 | if (!SectionsOrErr) { |
| 323 | DRI.d.a = 0; |
| 324 | DRI.d.b = 0; |
| 325 | return DRI; |
| 326 | } |
| 327 | uintptr_t SHT = reinterpret_cast<uintptr_t>((*SectionsOrErr).begin()); |
| 328 | unsigned SymTableIndex = |
| 329 | (reinterpret_cast<uintptr_t>(SymTable) - SHT) / sizeof(Elf_Shdr); |
| 330 | |
| 331 | DRI.d.a = SymTableIndex; |
| 332 | DRI.d.b = SymbolNum; |
| 333 | return DRI; |
| 334 | } |
| 335 | |
| 336 | const Elf_Shdr *toELFShdrIter(DataRefImpl Sec) const { |
| 337 | return reinterpret_cast<const Elf_Shdr *>(Sec.p); |
| 338 | } |
| 339 | |
| 340 | DataRefImpl toDRI(const Elf_Shdr *Sec) const { |
| 341 | DataRefImpl DRI; |
| 342 | DRI.p = reinterpret_cast<uintptr_t>(Sec); |
| 343 | return DRI; |
| 344 | } |
| 345 | |
| 346 | DataRefImpl toDRI(const Elf_Dyn *Dyn) const { |
| 347 | DataRefImpl DRI; |
| 348 | DRI.p = reinterpret_cast<uintptr_t>(Dyn); |
| 349 | return DRI; |
| 350 | } |
| 351 | |
| 352 | bool isExportedToOtherDSO(const Elf_Sym *ESym) const { |
| 353 | unsigned char Binding = ESym->getBinding(); |
| 354 | unsigned char Visibility = ESym->getVisibility(); |
| 355 | |
| 356 | // A symbol is exported if its binding is either GLOBAL or WEAK, and its |
| 357 | // visibility is either DEFAULT or PROTECTED. All other symbols are not |
| 358 | // exported. |
Andrew Walbran | 16937d0 | 2019-10-22 13:54:20 +0100 | [diff] [blame] | 359 | return ( |
| 360 | (Binding == ELF::STB_GLOBAL || Binding == ELF::STB_WEAK || |
| 361 | Binding == ELF::STB_GNU_UNIQUE) && |
| 362 | (Visibility == ELF::STV_DEFAULT || Visibility == ELF::STV_PROTECTED)); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 363 | } |
| 364 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 365 | Error getBuildAttributes(ELFAttributeParser &Attributes) const override { |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 366 | auto SectionsOrErr = EF.sections(); |
| 367 | if (!SectionsOrErr) |
| 368 | return SectionsOrErr.takeError(); |
| 369 | |
| 370 | for (const Elf_Shdr &Sec : *SectionsOrErr) { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 371 | if (Sec.sh_type == ELF::SHT_ARM_ATTRIBUTES || |
| 372 | Sec.sh_type == ELF::SHT_RISCV_ATTRIBUTES) { |
| 373 | auto ErrorOrContents = EF.getSectionContents(Sec); |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 374 | if (!ErrorOrContents) |
| 375 | return ErrorOrContents.takeError(); |
| 376 | |
| 377 | auto Contents = ErrorOrContents.get(); |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 378 | if (Contents[0] != ELFAttrs::Format_Version || Contents.size() == 1) |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 379 | return Error::success(); |
| 380 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 381 | if (Error E = Attributes.parse(Contents, ELFT::TargetEndianness)) |
| 382 | return E; |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 383 | break; |
| 384 | } |
| 385 | } |
| 386 | return Error::success(); |
| 387 | } |
| 388 | |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 389 | // This flag is used for classof, to distinguish ELFObjectFile from |
| 390 | // its subclass. If more subclasses will be created, this flag will |
| 391 | // have to become an enum. |
| 392 | bool isDyldELFObject; |
| 393 | |
| 394 | public: |
| 395 | ELFObjectFile(ELFObjectFile<ELFT> &&Other); |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 396 | static Expected<ELFObjectFile<ELFT>> create(MemoryBufferRef Object, |
| 397 | bool InitContent = true); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 398 | |
| 399 | const Elf_Rel *getRel(DataRefImpl Rel) const; |
| 400 | const Elf_Rela *getRela(DataRefImpl Rela) const; |
| 401 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 402 | Expected<const Elf_Sym *> getSymbol(DataRefImpl Sym) const { |
| 403 | return EF.template getEntry<Elf_Sym>(Sym.d.a, Sym.d.b); |
| 404 | } |
| 405 | |
| 406 | /// Get the relocation section that contains \a Rel. |
| 407 | const Elf_Shdr *getRelSection(DataRefImpl Rel) const { |
| 408 | auto RelSecOrErr = EF.getSection(Rel.d.a); |
| 409 | if (!RelSecOrErr) |
| 410 | report_fatal_error(errorToErrorCode(RelSecOrErr.takeError()).message()); |
| 411 | return *RelSecOrErr; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 412 | } |
| 413 | |
| 414 | const Elf_Shdr *getSection(DataRefImpl Sec) const { |
| 415 | return reinterpret_cast<const Elf_Shdr *>(Sec.p); |
| 416 | } |
| 417 | |
| 418 | basic_symbol_iterator symbol_begin() const override; |
| 419 | basic_symbol_iterator symbol_end() const override; |
| 420 | |
| 421 | elf_symbol_iterator dynamic_symbol_begin() const; |
| 422 | elf_symbol_iterator dynamic_symbol_end() const; |
| 423 | |
| 424 | section_iterator section_begin() const override; |
| 425 | section_iterator section_end() const override; |
| 426 | |
| 427 | Expected<int64_t> getRelocationAddend(DataRefImpl Rel) const override; |
| 428 | |
| 429 | uint8_t getBytesInAddress() const override; |
| 430 | StringRef getFileFormatName() const override; |
| 431 | Triple::ArchType getArch() const override; |
Andrew Scull | cdfcccc | 2018-10-05 20:58:37 +0100 | [diff] [blame] | 432 | Expected<uint64_t> getStartAddress() const override; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 433 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 434 | unsigned getPlatformFlags() const override { return EF.getHeader().e_flags; } |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 435 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 436 | const ELFFile<ELFT> &getELFFile() const { return EF; } |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 437 | |
| 438 | bool isDyldType() const { return isDyldELFObject; } |
| 439 | static bool classof(const Binary *v) { |
| 440 | return v->getType() == getELFType(ELFT::TargetEndianness == support::little, |
| 441 | ELFT::Is64Bits); |
| 442 | } |
| 443 | |
| 444 | elf_symbol_iterator_range getDynamicSymbolIterators() const override; |
| 445 | |
| 446 | bool isRelocatableObject() const override; |
| 447 | }; |
| 448 | |
| 449 | using ELF32LEObjectFile = ELFObjectFile<ELF32LE>; |
| 450 | using ELF64LEObjectFile = ELFObjectFile<ELF64LE>; |
| 451 | using ELF32BEObjectFile = ELFObjectFile<ELF32BE>; |
| 452 | using ELF64BEObjectFile = ELFObjectFile<ELF64BE>; |
| 453 | |
| 454 | template <class ELFT> |
| 455 | void ELFObjectFile<ELFT>::moveSymbolNext(DataRefImpl &Sym) const { |
| 456 | ++Sym.d.b; |
| 457 | } |
| 458 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 459 | template <class ELFT> Error ELFObjectFile<ELFT>::initContent() { |
| 460 | auto SectionsOrErr = EF.sections(); |
| 461 | if (!SectionsOrErr) |
| 462 | return SectionsOrErr.takeError(); |
| 463 | |
| 464 | for (const Elf_Shdr &Sec : *SectionsOrErr) { |
| 465 | switch (Sec.sh_type) { |
| 466 | case ELF::SHT_DYNSYM: { |
| 467 | if (!DotDynSymSec) |
| 468 | DotDynSymSec = &Sec; |
| 469 | break; |
| 470 | } |
| 471 | case ELF::SHT_SYMTAB: { |
| 472 | if (!DotSymtabSec) |
| 473 | DotSymtabSec = &Sec; |
| 474 | break; |
| 475 | } |
| 476 | case ELF::SHT_SYMTAB_SHNDX: { |
| 477 | if (!DotSymtabShndxSec) |
| 478 | DotSymtabShndxSec = &Sec; |
| 479 | break; |
| 480 | } |
| 481 | } |
| 482 | } |
| 483 | |
| 484 | ContentValid = true; |
| 485 | return Error::success(); |
| 486 | } |
| 487 | |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 488 | template <class ELFT> |
| 489 | Expected<StringRef> ELFObjectFile<ELFT>::getSymbolName(DataRefImpl Sym) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 490 | Expected<const Elf_Sym *> SymOrErr = getSymbol(Sym); |
| 491 | if (!SymOrErr) |
| 492 | return SymOrErr.takeError(); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 493 | auto SymTabOrErr = EF.getSection(Sym.d.a); |
| 494 | if (!SymTabOrErr) |
| 495 | return SymTabOrErr.takeError(); |
| 496 | const Elf_Shdr *SymTableSec = *SymTabOrErr; |
| 497 | auto StrTabOrErr = EF.getSection(SymTableSec->sh_link); |
| 498 | if (!StrTabOrErr) |
| 499 | return StrTabOrErr.takeError(); |
| 500 | const Elf_Shdr *StringTableSec = *StrTabOrErr; |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 501 | auto SymStrTabOrErr = EF.getStringTable(*StringTableSec); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 502 | if (!SymStrTabOrErr) |
| 503 | return SymStrTabOrErr.takeError(); |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 504 | Expected<StringRef> Name = (*SymOrErr)->getName(*SymStrTabOrErr); |
| 505 | if (Name && !Name->empty()) |
| 506 | return Name; |
Andrew Walbran | 16937d0 | 2019-10-22 13:54:20 +0100 | [diff] [blame] | 507 | |
| 508 | // If the symbol name is empty use the section name. |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 509 | if ((*SymOrErr)->getType() == ELF::STT_SECTION) { |
| 510 | if (Expected<section_iterator> SecOrErr = getSymbolSection(Sym)) { |
| 511 | consumeError(Name.takeError()); |
| 512 | return (*SecOrErr)->getName(); |
| 513 | } |
Andrew Walbran | 16937d0 | 2019-10-22 13:54:20 +0100 | [diff] [blame] | 514 | } |
| 515 | return Name; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 516 | } |
| 517 | |
| 518 | template <class ELFT> |
| 519 | uint64_t ELFObjectFile<ELFT>::getSectionFlags(DataRefImpl Sec) const { |
| 520 | return getSection(Sec)->sh_flags; |
| 521 | } |
| 522 | |
| 523 | template <class ELFT> |
| 524 | uint32_t ELFObjectFile<ELFT>::getSectionType(DataRefImpl Sec) const { |
| 525 | return getSection(Sec)->sh_type; |
| 526 | } |
| 527 | |
| 528 | template <class ELFT> |
| 529 | uint64_t ELFObjectFile<ELFT>::getSectionOffset(DataRefImpl Sec) const { |
| 530 | return getSection(Sec)->sh_offset; |
| 531 | } |
| 532 | |
| 533 | template <class ELFT> |
| 534 | uint64_t ELFObjectFile<ELFT>::getSymbolValueImpl(DataRefImpl Symb) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 535 | Expected<const Elf_Sym *> SymOrErr = getSymbol(Symb); |
| 536 | if (!SymOrErr) |
| 537 | report_fatal_error(SymOrErr.takeError()); |
| 538 | |
| 539 | uint64_t Ret = (*SymOrErr)->st_value; |
| 540 | if ((*SymOrErr)->st_shndx == ELF::SHN_ABS) |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 541 | return Ret; |
| 542 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 543 | const Elf_Ehdr &Header = EF.getHeader(); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 544 | // Clear the ARM/Thumb or microMIPS indicator flag. |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 545 | if ((Header.e_machine == ELF::EM_ARM || Header.e_machine == ELF::EM_MIPS) && |
| 546 | (*SymOrErr)->getType() == ELF::STT_FUNC) |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 547 | Ret &= ~1; |
| 548 | |
| 549 | return Ret; |
| 550 | } |
| 551 | |
| 552 | template <class ELFT> |
| 553 | Expected<uint64_t> |
| 554 | ELFObjectFile<ELFT>::getSymbolAddress(DataRefImpl Symb) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 555 | Expected<uint64_t> SymbolValueOrErr = getSymbolValue(Symb); |
| 556 | if (!SymbolValueOrErr) |
| 557 | // TODO: Test this error. |
| 558 | return SymbolValueOrErr.takeError(); |
| 559 | |
| 560 | uint64_t Result = *SymbolValueOrErr; |
| 561 | Expected<const Elf_Sym *> SymOrErr = getSymbol(Symb); |
| 562 | if (!SymOrErr) |
| 563 | return SymOrErr.takeError(); |
| 564 | |
| 565 | switch ((*SymOrErr)->st_shndx) { |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 566 | case ELF::SHN_COMMON: |
| 567 | case ELF::SHN_UNDEF: |
| 568 | case ELF::SHN_ABS: |
| 569 | return Result; |
| 570 | } |
| 571 | |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 572 | auto SymTabOrErr = EF.getSection(Symb.d.a); |
| 573 | if (!SymTabOrErr) |
| 574 | return SymTabOrErr.takeError(); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 575 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 576 | if (EF.getHeader().e_type == ELF::ET_REL) { |
| 577 | ArrayRef<Elf_Word> ShndxTable; |
| 578 | if (DotSymtabShndxSec) { |
| 579 | // TODO: Test this error. |
| 580 | if (Expected<ArrayRef<Elf_Word>> ShndxTableOrErr = |
| 581 | EF.getSHNDXTable(*DotSymtabShndxSec)) |
| 582 | ShndxTable = *ShndxTableOrErr; |
| 583 | else |
| 584 | return ShndxTableOrErr.takeError(); |
| 585 | } |
| 586 | |
| 587 | Expected<const Elf_Shdr *> SectionOrErr = |
| 588 | EF.getSection(**SymOrErr, *SymTabOrErr, ShndxTable); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 589 | if (!SectionOrErr) |
| 590 | return SectionOrErr.takeError(); |
| 591 | const Elf_Shdr *Section = *SectionOrErr; |
| 592 | if (Section) |
| 593 | Result += Section->sh_addr; |
| 594 | } |
| 595 | |
| 596 | return Result; |
| 597 | } |
| 598 | |
| 599 | template <class ELFT> |
| 600 | uint32_t ELFObjectFile<ELFT>::getSymbolAlignment(DataRefImpl Symb) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 601 | Expected<const Elf_Sym *> SymOrErr = getSymbol(Symb); |
| 602 | if (!SymOrErr) |
| 603 | report_fatal_error(SymOrErr.takeError()); |
| 604 | if ((*SymOrErr)->st_shndx == ELF::SHN_COMMON) |
| 605 | return (*SymOrErr)->st_value; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 606 | return 0; |
| 607 | } |
| 608 | |
| 609 | template <class ELFT> |
| 610 | uint16_t ELFObjectFile<ELFT>::getEMachine() const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 611 | return EF.getHeader().e_machine; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 612 | } |
| 613 | |
Andrew Scull | cdfcccc | 2018-10-05 20:58:37 +0100 | [diff] [blame] | 614 | template <class ELFT> uint16_t ELFObjectFile<ELFT>::getEType() const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 615 | return EF.getHeader().e_type; |
Andrew Scull | cdfcccc | 2018-10-05 20:58:37 +0100 | [diff] [blame] | 616 | } |
| 617 | |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 618 | template <class ELFT> |
| 619 | uint64_t ELFObjectFile<ELFT>::getSymbolSize(DataRefImpl Sym) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 620 | Expected<const Elf_Sym *> SymOrErr = getSymbol(Sym); |
| 621 | if (!SymOrErr) |
| 622 | report_fatal_error(SymOrErr.takeError()); |
| 623 | return (*SymOrErr)->st_size; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 624 | } |
| 625 | |
| 626 | template <class ELFT> |
| 627 | uint64_t ELFObjectFile<ELFT>::getCommonSymbolSizeImpl(DataRefImpl Symb) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 628 | return getSymbolSize(Symb); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 629 | } |
| 630 | |
| 631 | template <class ELFT> |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 632 | uint8_t ELFObjectFile<ELFT>::getSymbolBinding(DataRefImpl Symb) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 633 | Expected<const Elf_Sym *> SymOrErr = getSymbol(Symb); |
| 634 | if (!SymOrErr) |
| 635 | report_fatal_error(SymOrErr.takeError()); |
| 636 | return (*SymOrErr)->getBinding(); |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 637 | } |
| 638 | |
| 639 | template <class ELFT> |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 640 | uint8_t ELFObjectFile<ELFT>::getSymbolOther(DataRefImpl Symb) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 641 | Expected<const Elf_Sym *> SymOrErr = getSymbol(Symb); |
| 642 | if (!SymOrErr) |
| 643 | report_fatal_error(SymOrErr.takeError()); |
| 644 | return (*SymOrErr)->st_other; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 645 | } |
| 646 | |
| 647 | template <class ELFT> |
| 648 | uint8_t ELFObjectFile<ELFT>::getSymbolELFType(DataRefImpl Symb) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 649 | Expected<const Elf_Sym *> SymOrErr = getSymbol(Symb); |
| 650 | if (!SymOrErr) |
| 651 | report_fatal_error(SymOrErr.takeError()); |
| 652 | return (*SymOrErr)->getType(); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 653 | } |
| 654 | |
| 655 | template <class ELFT> |
| 656 | Expected<SymbolRef::Type> |
| 657 | ELFObjectFile<ELFT>::getSymbolType(DataRefImpl Symb) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 658 | Expected<const Elf_Sym *> SymOrErr = getSymbol(Symb); |
| 659 | if (!SymOrErr) |
| 660 | return SymOrErr.takeError(); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 661 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 662 | switch ((*SymOrErr)->getType()) { |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 663 | case ELF::STT_NOTYPE: |
| 664 | return SymbolRef::ST_Unknown; |
| 665 | case ELF::STT_SECTION: |
| 666 | return SymbolRef::ST_Debug; |
| 667 | case ELF::STT_FILE: |
| 668 | return SymbolRef::ST_File; |
| 669 | case ELF::STT_FUNC: |
| 670 | return SymbolRef::ST_Function; |
| 671 | case ELF::STT_OBJECT: |
| 672 | case ELF::STT_COMMON: |
| 673 | case ELF::STT_TLS: |
| 674 | return SymbolRef::ST_Data; |
| 675 | default: |
| 676 | return SymbolRef::ST_Other; |
| 677 | } |
| 678 | } |
| 679 | |
| 680 | template <class ELFT> |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 681 | Expected<uint32_t> ELFObjectFile<ELFT>::getSymbolFlags(DataRefImpl Sym) const { |
| 682 | Expected<const Elf_Sym *> SymOrErr = getSymbol(Sym); |
| 683 | if (!SymOrErr) |
| 684 | return SymOrErr.takeError(); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 685 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 686 | const Elf_Sym *ESym = *SymOrErr; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 687 | uint32_t Result = SymbolRef::SF_None; |
| 688 | |
| 689 | if (ESym->getBinding() != ELF::STB_LOCAL) |
| 690 | Result |= SymbolRef::SF_Global; |
| 691 | |
| 692 | if (ESym->getBinding() == ELF::STB_WEAK) |
| 693 | Result |= SymbolRef::SF_Weak; |
| 694 | |
| 695 | if (ESym->st_shndx == ELF::SHN_ABS) |
| 696 | Result |= SymbolRef::SF_Absolute; |
| 697 | |
| 698 | if (ESym->getType() == ELF::STT_FILE || ESym->getType() == ELF::STT_SECTION) |
| 699 | Result |= SymbolRef::SF_FormatSpecific; |
| 700 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 701 | if (Expected<typename ELFT::SymRange> SymbolsOrErr = |
| 702 | EF.symbols(DotSymtabSec)) { |
| 703 | // Set the SF_FormatSpecific flag for the 0-index null symbol. |
| 704 | if (ESym == SymbolsOrErr->begin()) |
| 705 | Result |= SymbolRef::SF_FormatSpecific; |
| 706 | } else |
| 707 | // TODO: Test this error. |
| 708 | return SymbolsOrErr.takeError(); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 709 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 710 | if (Expected<typename ELFT::SymRange> SymbolsOrErr = |
| 711 | EF.symbols(DotDynSymSec)) { |
| 712 | // Set the SF_FormatSpecific flag for the 0-index null symbol. |
| 713 | if (ESym == SymbolsOrErr->begin()) |
| 714 | Result |= SymbolRef::SF_FormatSpecific; |
| 715 | } else |
| 716 | // TODO: Test this error. |
| 717 | return SymbolsOrErr.takeError(); |
| 718 | |
| 719 | if (EF.getHeader().e_machine == ELF::EM_ARM) { |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 720 | if (Expected<StringRef> NameOrErr = getSymbolName(Sym)) { |
| 721 | StringRef Name = *NameOrErr; |
| 722 | if (Name.startswith("$d") || Name.startswith("$t") || |
| 723 | Name.startswith("$a")) |
| 724 | Result |= SymbolRef::SF_FormatSpecific; |
| 725 | } else { |
| 726 | // TODO: Actually report errors helpfully. |
| 727 | consumeError(NameOrErr.takeError()); |
| 728 | } |
| 729 | if (ESym->getType() == ELF::STT_FUNC && (ESym->st_value & 1) == 1) |
| 730 | Result |= SymbolRef::SF_Thumb; |
| 731 | } |
| 732 | |
| 733 | if (ESym->st_shndx == ELF::SHN_UNDEF) |
| 734 | Result |= SymbolRef::SF_Undefined; |
| 735 | |
| 736 | if (ESym->getType() == ELF::STT_COMMON || ESym->st_shndx == ELF::SHN_COMMON) |
| 737 | Result |= SymbolRef::SF_Common; |
| 738 | |
| 739 | if (isExportedToOtherDSO(ESym)) |
| 740 | Result |= SymbolRef::SF_Exported; |
| 741 | |
| 742 | if (ESym->getVisibility() == ELF::STV_HIDDEN) |
| 743 | Result |= SymbolRef::SF_Hidden; |
| 744 | |
| 745 | return Result; |
| 746 | } |
| 747 | |
| 748 | template <class ELFT> |
| 749 | Expected<section_iterator> |
| 750 | ELFObjectFile<ELFT>::getSymbolSection(const Elf_Sym *ESym, |
| 751 | const Elf_Shdr *SymTab) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 752 | ArrayRef<Elf_Word> ShndxTable; |
| 753 | if (DotSymtabShndxSec) { |
| 754 | // TODO: Test this error. |
| 755 | Expected<ArrayRef<Elf_Word>> ShndxTableOrErr = |
| 756 | EF.getSHNDXTable(*DotSymtabShndxSec); |
| 757 | if (!ShndxTableOrErr) |
| 758 | return ShndxTableOrErr.takeError(); |
| 759 | ShndxTable = *ShndxTableOrErr; |
| 760 | } |
| 761 | |
| 762 | auto ESecOrErr = EF.getSection(*ESym, SymTab, ShndxTable); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 763 | if (!ESecOrErr) |
| 764 | return ESecOrErr.takeError(); |
| 765 | |
| 766 | const Elf_Shdr *ESec = *ESecOrErr; |
| 767 | if (!ESec) |
| 768 | return section_end(); |
| 769 | |
| 770 | DataRefImpl Sec; |
| 771 | Sec.p = reinterpret_cast<intptr_t>(ESec); |
| 772 | return section_iterator(SectionRef(Sec, this)); |
| 773 | } |
| 774 | |
| 775 | template <class ELFT> |
| 776 | Expected<section_iterator> |
| 777 | ELFObjectFile<ELFT>::getSymbolSection(DataRefImpl Symb) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 778 | Expected<const Elf_Sym *> SymOrErr = getSymbol(Symb); |
| 779 | if (!SymOrErr) |
| 780 | return SymOrErr.takeError(); |
| 781 | |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 782 | auto SymTabOrErr = EF.getSection(Symb.d.a); |
| 783 | if (!SymTabOrErr) |
| 784 | return SymTabOrErr.takeError(); |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 785 | return getSymbolSection(*SymOrErr, *SymTabOrErr); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 786 | } |
| 787 | |
| 788 | template <class ELFT> |
| 789 | void ELFObjectFile<ELFT>::moveSectionNext(DataRefImpl &Sec) const { |
| 790 | const Elf_Shdr *ESec = getSection(Sec); |
| 791 | Sec = toDRI(++ESec); |
| 792 | } |
| 793 | |
| 794 | template <class ELFT> |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 795 | Expected<StringRef> ELFObjectFile<ELFT>::getSectionName(DataRefImpl Sec) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 796 | return EF.getSectionName(*getSection(Sec)); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 797 | } |
| 798 | |
| 799 | template <class ELFT> |
| 800 | uint64_t ELFObjectFile<ELFT>::getSectionAddress(DataRefImpl Sec) const { |
| 801 | return getSection(Sec)->sh_addr; |
| 802 | } |
| 803 | |
| 804 | template <class ELFT> |
| 805 | uint64_t ELFObjectFile<ELFT>::getSectionIndex(DataRefImpl Sec) const { |
| 806 | auto SectionsOrErr = EF.sections(); |
| 807 | handleAllErrors(std::move(SectionsOrErr.takeError()), |
| 808 | [](const ErrorInfoBase &) { |
| 809 | llvm_unreachable("unable to get section index"); |
| 810 | }); |
| 811 | const Elf_Shdr *First = SectionsOrErr->begin(); |
| 812 | return getSection(Sec) - First; |
| 813 | } |
| 814 | |
| 815 | template <class ELFT> |
| 816 | uint64_t ELFObjectFile<ELFT>::getSectionSize(DataRefImpl Sec) const { |
| 817 | return getSection(Sec)->sh_size; |
| 818 | } |
| 819 | |
| 820 | template <class ELFT> |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 821 | Expected<ArrayRef<uint8_t>> |
| 822 | ELFObjectFile<ELFT>::getSectionContents(DataRefImpl Sec) const { |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 823 | const Elf_Shdr *EShdr = getSection(Sec); |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 824 | if (EShdr->sh_type == ELF::SHT_NOBITS) |
| 825 | return makeArrayRef((const uint8_t *)base(), 0); |
| 826 | if (Error E = |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 827 | checkOffset(getMemoryBufferRef(), |
| 828 | (uintptr_t)base() + EShdr->sh_offset, EShdr->sh_size)) |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 829 | return std::move(E); |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 830 | return makeArrayRef((const uint8_t *)base() + EShdr->sh_offset, |
| 831 | EShdr->sh_size); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 832 | } |
| 833 | |
| 834 | template <class ELFT> |
| 835 | uint64_t ELFObjectFile<ELFT>::getSectionAlignment(DataRefImpl Sec) const { |
| 836 | return getSection(Sec)->sh_addralign; |
| 837 | } |
| 838 | |
| 839 | template <class ELFT> |
| 840 | bool ELFObjectFile<ELFT>::isSectionCompressed(DataRefImpl Sec) const { |
| 841 | return getSection(Sec)->sh_flags & ELF::SHF_COMPRESSED; |
| 842 | } |
| 843 | |
| 844 | template <class ELFT> |
| 845 | bool ELFObjectFile<ELFT>::isSectionText(DataRefImpl Sec) const { |
| 846 | return getSection(Sec)->sh_flags & ELF::SHF_EXECINSTR; |
| 847 | } |
| 848 | |
| 849 | template <class ELFT> |
| 850 | bool ELFObjectFile<ELFT>::isSectionData(DataRefImpl Sec) const { |
| 851 | const Elf_Shdr *EShdr = getSection(Sec); |
Andrew Scull | cdfcccc | 2018-10-05 20:58:37 +0100 | [diff] [blame] | 852 | return EShdr->sh_type == ELF::SHT_PROGBITS && |
| 853 | EShdr->sh_flags & ELF::SHF_ALLOC && |
| 854 | !(EShdr->sh_flags & ELF::SHF_EXECINSTR); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 855 | } |
| 856 | |
| 857 | template <class ELFT> |
| 858 | bool ELFObjectFile<ELFT>::isSectionBSS(DataRefImpl Sec) const { |
| 859 | const Elf_Shdr *EShdr = getSection(Sec); |
| 860 | return EShdr->sh_flags & (ELF::SHF_ALLOC | ELF::SHF_WRITE) && |
| 861 | EShdr->sh_type == ELF::SHT_NOBITS; |
| 862 | } |
| 863 | |
| 864 | template <class ELFT> |
Andrew Scull | cdfcccc | 2018-10-05 20:58:37 +0100 | [diff] [blame] | 865 | std::vector<SectionRef> |
| 866 | ELFObjectFile<ELFT>::dynamic_relocation_sections() const { |
| 867 | std::vector<SectionRef> Res; |
| 868 | std::vector<uintptr_t> Offsets; |
| 869 | |
| 870 | auto SectionsOrErr = EF.sections(); |
| 871 | if (!SectionsOrErr) |
| 872 | return Res; |
| 873 | |
| 874 | for (const Elf_Shdr &Sec : *SectionsOrErr) { |
| 875 | if (Sec.sh_type != ELF::SHT_DYNAMIC) |
| 876 | continue; |
| 877 | Elf_Dyn *Dynamic = |
| 878 | reinterpret_cast<Elf_Dyn *>((uintptr_t)base() + Sec.sh_offset); |
| 879 | for (; Dynamic->d_tag != ELF::DT_NULL; Dynamic++) { |
| 880 | if (Dynamic->d_tag == ELF::DT_REL || Dynamic->d_tag == ELF::DT_RELA || |
| 881 | Dynamic->d_tag == ELF::DT_JMPREL) { |
| 882 | Offsets.push_back(Dynamic->d_un.d_val); |
| 883 | } |
| 884 | } |
| 885 | } |
| 886 | for (const Elf_Shdr &Sec : *SectionsOrErr) { |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 887 | if (is_contained(Offsets, Sec.sh_addr)) |
Andrew Scull | cdfcccc | 2018-10-05 20:58:37 +0100 | [diff] [blame] | 888 | Res.emplace_back(toDRI(&Sec), this); |
| 889 | } |
| 890 | return Res; |
| 891 | } |
| 892 | |
| 893 | template <class ELFT> |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 894 | bool ELFObjectFile<ELFT>::isSectionVirtual(DataRefImpl Sec) const { |
| 895 | return getSection(Sec)->sh_type == ELF::SHT_NOBITS; |
| 896 | } |
| 897 | |
| 898 | template <class ELFT> |
Andrew Walbran | 16937d0 | 2019-10-22 13:54:20 +0100 | [diff] [blame] | 899 | bool ELFObjectFile<ELFT>::isBerkeleyText(DataRefImpl Sec) const { |
| 900 | return getSection(Sec)->sh_flags & ELF::SHF_ALLOC && |
| 901 | (getSection(Sec)->sh_flags & ELF::SHF_EXECINSTR || |
| 902 | !(getSection(Sec)->sh_flags & ELF::SHF_WRITE)); |
| 903 | } |
| 904 | |
| 905 | template <class ELFT> |
| 906 | bool ELFObjectFile<ELFT>::isBerkeleyData(DataRefImpl Sec) const { |
| 907 | const Elf_Shdr *EShdr = getSection(Sec); |
| 908 | return !isBerkeleyText(Sec) && EShdr->sh_type != ELF::SHT_NOBITS && |
| 909 | EShdr->sh_flags & ELF::SHF_ALLOC; |
| 910 | } |
| 911 | |
| 912 | template <class ELFT> |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 913 | bool ELFObjectFile<ELFT>::isDebugSection(StringRef SectionName) const { |
| 914 | return SectionName.startswith(".debug") || |
| 915 | SectionName.startswith(".zdebug") || SectionName == ".gdb_index"; |
| 916 | } |
| 917 | |
| 918 | template <class ELFT> |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 919 | relocation_iterator |
| 920 | ELFObjectFile<ELFT>::section_rel_begin(DataRefImpl Sec) const { |
| 921 | DataRefImpl RelData; |
| 922 | auto SectionsOrErr = EF.sections(); |
| 923 | if (!SectionsOrErr) |
| 924 | return relocation_iterator(RelocationRef()); |
| 925 | uintptr_t SHT = reinterpret_cast<uintptr_t>((*SectionsOrErr).begin()); |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 926 | RelData.d.a = (Sec.p - SHT) / EF.getHeader().e_shentsize; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 927 | RelData.d.b = 0; |
| 928 | return relocation_iterator(RelocationRef(RelData, this)); |
| 929 | } |
| 930 | |
| 931 | template <class ELFT> |
| 932 | relocation_iterator |
| 933 | ELFObjectFile<ELFT>::section_rel_end(DataRefImpl Sec) const { |
| 934 | const Elf_Shdr *S = reinterpret_cast<const Elf_Shdr *>(Sec.p); |
| 935 | relocation_iterator Begin = section_rel_begin(Sec); |
| 936 | if (S->sh_type != ELF::SHT_RELA && S->sh_type != ELF::SHT_REL) |
| 937 | return Begin; |
| 938 | DataRefImpl RelData = Begin->getRawDataRefImpl(); |
| 939 | const Elf_Shdr *RelSec = getRelSection(RelData); |
| 940 | |
| 941 | // Error check sh_link here so that getRelocationSymbol can just use it. |
| 942 | auto SymSecOrErr = EF.getSection(RelSec->sh_link); |
| 943 | if (!SymSecOrErr) |
| 944 | report_fatal_error(errorToErrorCode(SymSecOrErr.takeError()).message()); |
| 945 | |
| 946 | RelData.d.b += S->sh_size / S->sh_entsize; |
| 947 | return relocation_iterator(RelocationRef(RelData, this)); |
| 948 | } |
| 949 | |
| 950 | template <class ELFT> |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 951 | Expected<section_iterator> |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 952 | ELFObjectFile<ELFT>::getRelocatedSection(DataRefImpl Sec) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 953 | if (EF.getHeader().e_type != ELF::ET_REL) |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 954 | return section_end(); |
| 955 | |
| 956 | const Elf_Shdr *EShdr = getSection(Sec); |
| 957 | uintX_t Type = EShdr->sh_type; |
| 958 | if (Type != ELF::SHT_REL && Type != ELF::SHT_RELA) |
| 959 | return section_end(); |
| 960 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 961 | Expected<const Elf_Shdr *> SecOrErr = EF.getSection(EShdr->sh_info); |
| 962 | if (!SecOrErr) |
| 963 | return SecOrErr.takeError(); |
| 964 | return section_iterator(SectionRef(toDRI(*SecOrErr), this)); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 965 | } |
| 966 | |
| 967 | // Relocations |
| 968 | template <class ELFT> |
| 969 | void ELFObjectFile<ELFT>::moveRelocationNext(DataRefImpl &Rel) const { |
| 970 | ++Rel.d.b; |
| 971 | } |
| 972 | |
| 973 | template <class ELFT> |
| 974 | symbol_iterator |
| 975 | ELFObjectFile<ELFT>::getRelocationSymbol(DataRefImpl Rel) const { |
| 976 | uint32_t symbolIdx; |
| 977 | const Elf_Shdr *sec = getRelSection(Rel); |
| 978 | if (sec->sh_type == ELF::SHT_REL) |
| 979 | symbolIdx = getRel(Rel)->getSymbol(EF.isMips64EL()); |
| 980 | else |
| 981 | symbolIdx = getRela(Rel)->getSymbol(EF.isMips64EL()); |
| 982 | if (!symbolIdx) |
| 983 | return symbol_end(); |
| 984 | |
| 985 | // FIXME: error check symbolIdx |
| 986 | DataRefImpl SymbolData; |
| 987 | SymbolData.d.a = sec->sh_link; |
| 988 | SymbolData.d.b = symbolIdx; |
| 989 | return symbol_iterator(SymbolRef(SymbolData, this)); |
| 990 | } |
| 991 | |
| 992 | template <class ELFT> |
| 993 | uint64_t ELFObjectFile<ELFT>::getRelocationOffset(DataRefImpl Rel) const { |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 994 | const Elf_Shdr *sec = getRelSection(Rel); |
| 995 | if (sec->sh_type == ELF::SHT_REL) |
| 996 | return getRel(Rel)->r_offset; |
| 997 | |
| 998 | return getRela(Rel)->r_offset; |
| 999 | } |
| 1000 | |
| 1001 | template <class ELFT> |
| 1002 | uint64_t ELFObjectFile<ELFT>::getRelocationType(DataRefImpl Rel) const { |
| 1003 | const Elf_Shdr *sec = getRelSection(Rel); |
| 1004 | if (sec->sh_type == ELF::SHT_REL) |
| 1005 | return getRel(Rel)->getType(EF.isMips64EL()); |
| 1006 | else |
| 1007 | return getRela(Rel)->getType(EF.isMips64EL()); |
| 1008 | } |
| 1009 | |
| 1010 | template <class ELFT> |
| 1011 | StringRef ELFObjectFile<ELFT>::getRelocationTypeName(uint32_t Type) const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1012 | return getELFRelocationTypeName(EF.getHeader().e_machine, Type); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1013 | } |
| 1014 | |
| 1015 | template <class ELFT> |
| 1016 | void ELFObjectFile<ELFT>::getRelocationTypeName( |
| 1017 | DataRefImpl Rel, SmallVectorImpl<char> &Result) const { |
| 1018 | uint32_t type = getRelocationType(Rel); |
| 1019 | EF.getRelocationTypeName(type, Result); |
| 1020 | } |
| 1021 | |
| 1022 | template <class ELFT> |
| 1023 | Expected<int64_t> |
| 1024 | ELFObjectFile<ELFT>::getRelocationAddend(DataRefImpl Rel) const { |
| 1025 | if (getRelSection(Rel)->sh_type != ELF::SHT_RELA) |
| 1026 | return createError("Section is not SHT_RELA"); |
| 1027 | return (int64_t)getRela(Rel)->r_addend; |
| 1028 | } |
| 1029 | |
| 1030 | template <class ELFT> |
| 1031 | const typename ELFObjectFile<ELFT>::Elf_Rel * |
| 1032 | ELFObjectFile<ELFT>::getRel(DataRefImpl Rel) const { |
| 1033 | assert(getRelSection(Rel)->sh_type == ELF::SHT_REL); |
| 1034 | auto Ret = EF.template getEntry<Elf_Rel>(Rel.d.a, Rel.d.b); |
| 1035 | if (!Ret) |
| 1036 | report_fatal_error(errorToErrorCode(Ret.takeError()).message()); |
| 1037 | return *Ret; |
| 1038 | } |
| 1039 | |
| 1040 | template <class ELFT> |
| 1041 | const typename ELFObjectFile<ELFT>::Elf_Rela * |
| 1042 | ELFObjectFile<ELFT>::getRela(DataRefImpl Rela) const { |
| 1043 | assert(getRelSection(Rela)->sh_type == ELF::SHT_RELA); |
| 1044 | auto Ret = EF.template getEntry<Elf_Rela>(Rela.d.a, Rela.d.b); |
| 1045 | if (!Ret) |
| 1046 | report_fatal_error(errorToErrorCode(Ret.takeError()).message()); |
| 1047 | return *Ret; |
| 1048 | } |
| 1049 | |
| 1050 | template <class ELFT> |
| 1051 | Expected<ELFObjectFile<ELFT>> |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1052 | ELFObjectFile<ELFT>::create(MemoryBufferRef Object, bool InitContent) { |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1053 | auto EFOrErr = ELFFile<ELFT>::create(Object.getBuffer()); |
| 1054 | if (Error E = EFOrErr.takeError()) |
| 1055 | return std::move(E); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1056 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1057 | ELFObjectFile<ELFT> Obj = {Object, std::move(*EFOrErr), nullptr, nullptr, |
| 1058 | nullptr}; |
| 1059 | if (InitContent) |
| 1060 | if (Error E = Obj.initContent()) |
| 1061 | return std::move(E); |
| 1062 | return std::move(Obj); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1063 | } |
| 1064 | |
| 1065 | template <class ELFT> |
| 1066 | ELFObjectFile<ELFT>::ELFObjectFile(MemoryBufferRef Object, ELFFile<ELFT> EF, |
| 1067 | const Elf_Shdr *DotDynSymSec, |
| 1068 | const Elf_Shdr *DotSymtabSec, |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1069 | const Elf_Shdr *DotSymtabShndx) |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1070 | : ELFObjectFileBase( |
| 1071 | getELFType(ELFT::TargetEndianness == support::little, ELFT::Is64Bits), |
| 1072 | Object), |
| 1073 | EF(EF), DotDynSymSec(DotDynSymSec), DotSymtabSec(DotSymtabSec), |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1074 | DotSymtabShndxSec(DotSymtabShndx) {} |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1075 | |
| 1076 | template <class ELFT> |
| 1077 | ELFObjectFile<ELFT>::ELFObjectFile(ELFObjectFile<ELFT> &&Other) |
| 1078 | : ELFObjectFile(Other.Data, Other.EF, Other.DotDynSymSec, |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1079 | Other.DotSymtabSec, Other.DotSymtabShndxSec) {} |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1080 | |
| 1081 | template <class ELFT> |
| 1082 | basic_symbol_iterator ELFObjectFile<ELFT>::symbol_begin() const { |
Andrew Walbran | 3d2c197 | 2020-04-07 12:24:26 +0100 | [diff] [blame] | 1083 | DataRefImpl Sym = |
| 1084 | toDRI(DotSymtabSec, |
| 1085 | DotSymtabSec && DotSymtabSec->sh_size >= sizeof(Elf_Sym) ? 1 : 0); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1086 | return basic_symbol_iterator(SymbolRef(Sym, this)); |
| 1087 | } |
| 1088 | |
| 1089 | template <class ELFT> |
| 1090 | basic_symbol_iterator ELFObjectFile<ELFT>::symbol_end() const { |
| 1091 | const Elf_Shdr *SymTab = DotSymtabSec; |
| 1092 | if (!SymTab) |
| 1093 | return symbol_begin(); |
| 1094 | DataRefImpl Sym = toDRI(SymTab, SymTab->sh_size / sizeof(Elf_Sym)); |
| 1095 | return basic_symbol_iterator(SymbolRef(Sym, this)); |
| 1096 | } |
| 1097 | |
| 1098 | template <class ELFT> |
| 1099 | elf_symbol_iterator ELFObjectFile<ELFT>::dynamic_symbol_begin() const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1100 | if (!DotDynSymSec || DotDynSymSec->sh_size < sizeof(Elf_Sym)) |
| 1101 | // Ignore errors here where the dynsym is empty or sh_size less than the |
| 1102 | // size of one symbol. These should be handled elsewhere. |
| 1103 | return symbol_iterator(SymbolRef(toDRI(DotDynSymSec, 0), this)); |
| 1104 | // Skip 0-index NULL symbol. |
| 1105 | return symbol_iterator(SymbolRef(toDRI(DotDynSymSec, 1), this)); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1106 | } |
| 1107 | |
| 1108 | template <class ELFT> |
| 1109 | elf_symbol_iterator ELFObjectFile<ELFT>::dynamic_symbol_end() const { |
| 1110 | const Elf_Shdr *SymTab = DotDynSymSec; |
| 1111 | if (!SymTab) |
| 1112 | return dynamic_symbol_begin(); |
| 1113 | DataRefImpl Sym = toDRI(SymTab, SymTab->sh_size / sizeof(Elf_Sym)); |
| 1114 | return basic_symbol_iterator(SymbolRef(Sym, this)); |
| 1115 | } |
| 1116 | |
| 1117 | template <class ELFT> |
| 1118 | section_iterator ELFObjectFile<ELFT>::section_begin() const { |
| 1119 | auto SectionsOrErr = EF.sections(); |
| 1120 | if (!SectionsOrErr) |
| 1121 | return section_iterator(SectionRef()); |
| 1122 | return section_iterator(SectionRef(toDRI((*SectionsOrErr).begin()), this)); |
| 1123 | } |
| 1124 | |
| 1125 | template <class ELFT> |
| 1126 | section_iterator ELFObjectFile<ELFT>::section_end() const { |
| 1127 | auto SectionsOrErr = EF.sections(); |
| 1128 | if (!SectionsOrErr) |
| 1129 | return section_iterator(SectionRef()); |
| 1130 | return section_iterator(SectionRef(toDRI((*SectionsOrErr).end()), this)); |
| 1131 | } |
| 1132 | |
| 1133 | template <class ELFT> |
| 1134 | uint8_t ELFObjectFile<ELFT>::getBytesInAddress() const { |
| 1135 | return ELFT::Is64Bits ? 8 : 4; |
| 1136 | } |
| 1137 | |
| 1138 | template <class ELFT> |
| 1139 | StringRef ELFObjectFile<ELFT>::getFileFormatName() const { |
| 1140 | bool IsLittleEndian = ELFT::TargetEndianness == support::little; |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1141 | switch (EF.getHeader().e_ident[ELF::EI_CLASS]) { |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1142 | case ELF::ELFCLASS32: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1143 | switch (EF.getHeader().e_machine) { |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1144 | case ELF::EM_386: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1145 | return "elf32-i386"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1146 | case ELF::EM_IAMCU: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1147 | return "elf32-iamcu"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1148 | case ELF::EM_X86_64: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1149 | return "elf32-x86-64"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1150 | case ELF::EM_ARM: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1151 | return (IsLittleEndian ? "elf32-littlearm" : "elf32-bigarm"); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1152 | case ELF::EM_AVR: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1153 | return "elf32-avr"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1154 | case ELF::EM_HEXAGON: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1155 | return "elf32-hexagon"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1156 | case ELF::EM_LANAI: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1157 | return "elf32-lanai"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1158 | case ELF::EM_MIPS: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1159 | return "elf32-mips"; |
Andrew Walbran | 16937d0 | 2019-10-22 13:54:20 +0100 | [diff] [blame] | 1160 | case ELF::EM_MSP430: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1161 | return "elf32-msp430"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1162 | case ELF::EM_PPC: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1163 | return (IsLittleEndian ? "elf32-powerpcle" : "elf32-powerpc"); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1164 | case ELF::EM_RISCV: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1165 | return "elf32-littleriscv"; |
| 1166 | case ELF::EM_CSKY: |
| 1167 | return "elf32-csky"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1168 | case ELF::EM_SPARC: |
| 1169 | case ELF::EM_SPARC32PLUS: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1170 | return "elf32-sparc"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1171 | case ELF::EM_AMDGPU: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1172 | return "elf32-amdgpu"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1173 | default: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1174 | return "elf32-unknown"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1175 | } |
| 1176 | case ELF::ELFCLASS64: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1177 | switch (EF.getHeader().e_machine) { |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1178 | case ELF::EM_386: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1179 | return "elf64-i386"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1180 | case ELF::EM_X86_64: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1181 | return "elf64-x86-64"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1182 | case ELF::EM_AARCH64: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1183 | return (IsLittleEndian ? "elf64-littleaarch64" : "elf64-bigaarch64"); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1184 | case ELF::EM_PPC64: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1185 | return (IsLittleEndian ? "elf64-powerpcle" : "elf64-powerpc"); |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1186 | case ELF::EM_RISCV: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1187 | return "elf64-littleriscv"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1188 | case ELF::EM_S390: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1189 | return "elf64-s390"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1190 | case ELF::EM_SPARCV9: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1191 | return "elf64-sparc"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1192 | case ELF::EM_MIPS: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1193 | return "elf64-mips"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1194 | case ELF::EM_AMDGPU: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1195 | return "elf64-amdgpu"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1196 | case ELF::EM_BPF: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1197 | return "elf64-bpf"; |
| 1198 | case ELF::EM_VE: |
| 1199 | return "elf64-ve"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1200 | default: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1201 | return "elf64-unknown"; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1202 | } |
| 1203 | default: |
| 1204 | // FIXME: Proper error handling. |
| 1205 | report_fatal_error("Invalid ELFCLASS!"); |
| 1206 | } |
| 1207 | } |
| 1208 | |
| 1209 | template <class ELFT> Triple::ArchType ELFObjectFile<ELFT>::getArch() const { |
| 1210 | bool IsLittleEndian = ELFT::TargetEndianness == support::little; |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1211 | switch (EF.getHeader().e_machine) { |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1212 | case ELF::EM_386: |
| 1213 | case ELF::EM_IAMCU: |
| 1214 | return Triple::x86; |
| 1215 | case ELF::EM_X86_64: |
| 1216 | return Triple::x86_64; |
| 1217 | case ELF::EM_AARCH64: |
| 1218 | return IsLittleEndian ? Triple::aarch64 : Triple::aarch64_be; |
| 1219 | case ELF::EM_ARM: |
| 1220 | return Triple::arm; |
| 1221 | case ELF::EM_AVR: |
| 1222 | return Triple::avr; |
| 1223 | case ELF::EM_HEXAGON: |
| 1224 | return Triple::hexagon; |
| 1225 | case ELF::EM_LANAI: |
| 1226 | return Triple::lanai; |
| 1227 | case ELF::EM_MIPS: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1228 | switch (EF.getHeader().e_ident[ELF::EI_CLASS]) { |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1229 | case ELF::ELFCLASS32: |
| 1230 | return IsLittleEndian ? Triple::mipsel : Triple::mips; |
| 1231 | case ELF::ELFCLASS64: |
| 1232 | return IsLittleEndian ? Triple::mips64el : Triple::mips64; |
| 1233 | default: |
| 1234 | report_fatal_error("Invalid ELFCLASS!"); |
| 1235 | } |
Andrew Walbran | 16937d0 | 2019-10-22 13:54:20 +0100 | [diff] [blame] | 1236 | case ELF::EM_MSP430: |
| 1237 | return Triple::msp430; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1238 | case ELF::EM_PPC: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1239 | return IsLittleEndian ? Triple::ppcle : Triple::ppc; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1240 | case ELF::EM_PPC64: |
| 1241 | return IsLittleEndian ? Triple::ppc64le : Triple::ppc64; |
| 1242 | case ELF::EM_RISCV: |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1243 | switch (EF.getHeader().e_ident[ELF::EI_CLASS]) { |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1244 | case ELF::ELFCLASS32: |
| 1245 | return Triple::riscv32; |
| 1246 | case ELF::ELFCLASS64: |
| 1247 | return Triple::riscv64; |
| 1248 | default: |
| 1249 | report_fatal_error("Invalid ELFCLASS!"); |
| 1250 | } |
| 1251 | case ELF::EM_S390: |
| 1252 | return Triple::systemz; |
| 1253 | |
| 1254 | case ELF::EM_SPARC: |
| 1255 | case ELF::EM_SPARC32PLUS: |
| 1256 | return IsLittleEndian ? Triple::sparcel : Triple::sparc; |
| 1257 | case ELF::EM_SPARCV9: |
| 1258 | return Triple::sparcv9; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1259 | |
| 1260 | case ELF::EM_AMDGPU: { |
| 1261 | if (!IsLittleEndian) |
| 1262 | return Triple::UnknownArch; |
| 1263 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1264 | unsigned MACH = EF.getHeader().e_flags & ELF::EF_AMDGPU_MACH; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1265 | if (MACH >= ELF::EF_AMDGPU_MACH_R600_FIRST && |
| 1266 | MACH <= ELF::EF_AMDGPU_MACH_R600_LAST) |
| 1267 | return Triple::r600; |
| 1268 | if (MACH >= ELF::EF_AMDGPU_MACH_AMDGCN_FIRST && |
| 1269 | MACH <= ELF::EF_AMDGPU_MACH_AMDGCN_LAST) |
| 1270 | return Triple::amdgcn; |
| 1271 | |
| 1272 | return Triple::UnknownArch; |
| 1273 | } |
| 1274 | |
| 1275 | case ELF::EM_BPF: |
| 1276 | return IsLittleEndian ? Triple::bpfel : Triple::bpfeb; |
| 1277 | |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1278 | case ELF::EM_VE: |
| 1279 | return Triple::ve; |
| 1280 | case ELF::EM_CSKY: |
| 1281 | return Triple::csky; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1282 | default: |
| 1283 | return Triple::UnknownArch; |
| 1284 | } |
| 1285 | } |
| 1286 | |
| 1287 | template <class ELFT> |
Andrew Scull | cdfcccc | 2018-10-05 20:58:37 +0100 | [diff] [blame] | 1288 | Expected<uint64_t> ELFObjectFile<ELFT>::getStartAddress() const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1289 | return EF.getHeader().e_entry; |
Andrew Scull | cdfcccc | 2018-10-05 20:58:37 +0100 | [diff] [blame] | 1290 | } |
| 1291 | |
| 1292 | template <class ELFT> |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1293 | ELFObjectFileBase::elf_symbol_iterator_range |
| 1294 | ELFObjectFile<ELFT>::getDynamicSymbolIterators() const { |
| 1295 | return make_range(dynamic_symbol_begin(), dynamic_symbol_end()); |
| 1296 | } |
| 1297 | |
| 1298 | template <class ELFT> bool ELFObjectFile<ELFT>::isRelocatableObject() const { |
Olivier Deprez | f4ef2d0 | 2021-04-20 13:36:24 +0200 | [diff] [blame] | 1299 | return EF.getHeader().e_type == ELF::ET_REL; |
Andrew Scull | 5e1ddfa | 2018-08-14 10:06:54 +0100 | [diff] [blame] | 1300 | } |
| 1301 | |
| 1302 | } // end namespace object |
| 1303 | } // end namespace llvm |
| 1304 | |
| 1305 | #endif // LLVM_OBJECT_ELFOBJECTFILE_H |