123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790 |
- //===- ELFObjectFile.cpp - ELF object file implementation -----------------===//
- //
- // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
- // See https://llvm.org/LICENSE.txt for license information.
- // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
- //
- //===----------------------------------------------------------------------===//
- //
- // Part of the ELFObjectFile class implementation.
- //
- //===----------------------------------------------------------------------===//
- #include "llvm/Object/ELFObjectFile.h"
- #include "llvm/ADT/Triple.h"
- #include "llvm/BinaryFormat/ELF.h"
- #include "llvm/MC/MCInstrAnalysis.h"
- #include "llvm/MC/SubtargetFeature.h"
- #include "llvm/MC/TargetRegistry.h"
- #include "llvm/Object/ELF.h"
- #include "llvm/Object/ELFTypes.h"
- #include "llvm/Object/Error.h"
- #include "llvm/Support/ARMAttributeParser.h"
- #include "llvm/Support/ARMBuildAttributes.h"
- #include "llvm/Support/ErrorHandling.h"
- #include "llvm/Support/MathExtras.h"
- #include "llvm/Support/RISCVAttributeParser.h"
- #include "llvm/Support/RISCVAttributes.h"
- #include "llvm/Support/RISCVISAInfo.h"
- #include <algorithm>
- #include <cstddef>
- #include <cstdint>
- #include <memory>
- #include <optional>
- #include <string>
- #include <utility>
- using namespace llvm;
- using namespace object;
- const EnumEntry<unsigned> llvm::object::ElfSymbolTypes[NumElfSymbolTypes] = {
- {"None", "NOTYPE", ELF::STT_NOTYPE},
- {"Object", "OBJECT", ELF::STT_OBJECT},
- {"Function", "FUNC", ELF::STT_FUNC},
- {"Section", "SECTION", ELF::STT_SECTION},
- {"File", "FILE", ELF::STT_FILE},
- {"Common", "COMMON", ELF::STT_COMMON},
- {"TLS", "TLS", ELF::STT_TLS},
- {"Unknown", "<unknown>: 7", 7},
- {"Unknown", "<unknown>: 8", 8},
- {"Unknown", "<unknown>: 9", 9},
- {"GNU_IFunc", "IFUNC", ELF::STT_GNU_IFUNC},
- {"OS Specific", "<OS specific>: 11", 11},
- {"OS Specific", "<OS specific>: 12", 12},
- {"Proc Specific", "<processor specific>: 13", 13},
- {"Proc Specific", "<processor specific>: 14", 14},
- {"Proc Specific", "<processor specific>: 15", 15}
- };
- ELFObjectFileBase::ELFObjectFileBase(unsigned int Type, MemoryBufferRef Source)
- : ObjectFile(Type, Source) {}
- template <class ELFT>
- static Expected<std::unique_ptr<ELFObjectFile<ELFT>>>
- createPtr(MemoryBufferRef Object, bool InitContent) {
- auto Ret = ELFObjectFile<ELFT>::create(Object, InitContent);
- if (Error E = Ret.takeError())
- return std::move(E);
- return std::make_unique<ELFObjectFile<ELFT>>(std::move(*Ret));
- }
- Expected<std::unique_ptr<ObjectFile>>
- ObjectFile::createELFObjectFile(MemoryBufferRef Obj, bool InitContent) {
- std::pair<unsigned char, unsigned char> Ident =
- getElfArchType(Obj.getBuffer());
- std::size_t MaxAlignment =
- 1ULL << countTrailingZeros(
- reinterpret_cast<uintptr_t>(Obj.getBufferStart()));
- if (MaxAlignment < 2)
- return createError("Insufficient alignment");
- if (Ident.first == ELF::ELFCLASS32) {
- if (Ident.second == ELF::ELFDATA2LSB)
- return createPtr<ELF32LE>(Obj, InitContent);
- else if (Ident.second == ELF::ELFDATA2MSB)
- return createPtr<ELF32BE>(Obj, InitContent);
- else
- return createError("Invalid ELF data");
- } else if (Ident.first == ELF::ELFCLASS64) {
- if (Ident.second == ELF::ELFDATA2LSB)
- return createPtr<ELF64LE>(Obj, InitContent);
- else if (Ident.second == ELF::ELFDATA2MSB)
- return createPtr<ELF64BE>(Obj, InitContent);
- else
- return createError("Invalid ELF data");
- }
- return createError("Invalid ELF class");
- }
- SubtargetFeatures ELFObjectFileBase::getMIPSFeatures() const {
- SubtargetFeatures Features;
- unsigned PlatformFlags = getPlatformFlags();
- switch (PlatformFlags & ELF::EF_MIPS_ARCH) {
- case ELF::EF_MIPS_ARCH_1:
- break;
- case ELF::EF_MIPS_ARCH_2:
- Features.AddFeature("mips2");
- break;
- case ELF::EF_MIPS_ARCH_3:
- Features.AddFeature("mips3");
- break;
- case ELF::EF_MIPS_ARCH_4:
- Features.AddFeature("mips4");
- break;
- case ELF::EF_MIPS_ARCH_5:
- Features.AddFeature("mips5");
- break;
- case ELF::EF_MIPS_ARCH_32:
- Features.AddFeature("mips32");
- break;
- case ELF::EF_MIPS_ARCH_64:
- Features.AddFeature("mips64");
- break;
- case ELF::EF_MIPS_ARCH_32R2:
- Features.AddFeature("mips32r2");
- break;
- case ELF::EF_MIPS_ARCH_64R2:
- Features.AddFeature("mips64r2");
- break;
- case ELF::EF_MIPS_ARCH_32R6:
- Features.AddFeature("mips32r6");
- break;
- case ELF::EF_MIPS_ARCH_64R6:
- Features.AddFeature("mips64r6");
- break;
- default:
- llvm_unreachable("Unknown EF_MIPS_ARCH value");
- }
- switch (PlatformFlags & ELF::EF_MIPS_MACH) {
- case ELF::EF_MIPS_MACH_NONE:
- // No feature associated with this value.
- break;
- case ELF::EF_MIPS_MACH_OCTEON:
- Features.AddFeature("cnmips");
- break;
- default:
- llvm_unreachable("Unknown EF_MIPS_ARCH value");
- }
- if (PlatformFlags & ELF::EF_MIPS_ARCH_ASE_M16)
- Features.AddFeature("mips16");
- if (PlatformFlags & ELF::EF_MIPS_MICROMIPS)
- Features.AddFeature("micromips");
- return Features;
- }
- SubtargetFeatures ELFObjectFileBase::getARMFeatures() const {
- SubtargetFeatures Features;
- ARMAttributeParser Attributes;
- if (Error E = getBuildAttributes(Attributes)) {
- consumeError(std::move(E));
- return SubtargetFeatures();
- }
- // both ARMv7-M and R have to support thumb hardware div
- bool isV7 = false;
- std::optional<unsigned> Attr =
- Attributes.getAttributeValue(ARMBuildAttrs::CPU_arch);
- if (Attr)
- isV7 = *Attr == ARMBuildAttrs::v7;
- Attr = Attributes.getAttributeValue(ARMBuildAttrs::CPU_arch_profile);
- if (Attr) {
- switch (*Attr) {
- case ARMBuildAttrs::ApplicationProfile:
- Features.AddFeature("aclass");
- break;
- case ARMBuildAttrs::RealTimeProfile:
- Features.AddFeature("rclass");
- if (isV7)
- Features.AddFeature("hwdiv");
- break;
- case ARMBuildAttrs::MicroControllerProfile:
- Features.AddFeature("mclass");
- if (isV7)
- Features.AddFeature("hwdiv");
- break;
- }
- }
- Attr = Attributes.getAttributeValue(ARMBuildAttrs::THUMB_ISA_use);
- if (Attr) {
- switch (*Attr) {
- default:
- break;
- case ARMBuildAttrs::Not_Allowed:
- Features.AddFeature("thumb", false);
- Features.AddFeature("thumb2", false);
- break;
- case ARMBuildAttrs::AllowThumb32:
- Features.AddFeature("thumb2");
- break;
- }
- }
- Attr = Attributes.getAttributeValue(ARMBuildAttrs::FP_arch);
- if (Attr) {
- switch (*Attr) {
- default:
- break;
- case ARMBuildAttrs::Not_Allowed:
- Features.AddFeature("vfp2sp", false);
- Features.AddFeature("vfp3d16sp", false);
- Features.AddFeature("vfp4d16sp", false);
- break;
- case ARMBuildAttrs::AllowFPv2:
- Features.AddFeature("vfp2");
- break;
- case ARMBuildAttrs::AllowFPv3A:
- case ARMBuildAttrs::AllowFPv3B:
- Features.AddFeature("vfp3");
- break;
- case ARMBuildAttrs::AllowFPv4A:
- case ARMBuildAttrs::AllowFPv4B:
- Features.AddFeature("vfp4");
- break;
- }
- }
- Attr = Attributes.getAttributeValue(ARMBuildAttrs::Advanced_SIMD_arch);
- if (Attr) {
- switch (*Attr) {
- default:
- break;
- case ARMBuildAttrs::Not_Allowed:
- Features.AddFeature("neon", false);
- Features.AddFeature("fp16", false);
- break;
- case ARMBuildAttrs::AllowNeon:
- Features.AddFeature("neon");
- break;
- case ARMBuildAttrs::AllowNeon2:
- Features.AddFeature("neon");
- Features.AddFeature("fp16");
- break;
- }
- }
- Attr = Attributes.getAttributeValue(ARMBuildAttrs::MVE_arch);
- if (Attr) {
- switch (*Attr) {
- default:
- break;
- case ARMBuildAttrs::Not_Allowed:
- Features.AddFeature("mve", false);
- Features.AddFeature("mve.fp", false);
- break;
- case ARMBuildAttrs::AllowMVEInteger:
- Features.AddFeature("mve.fp", false);
- Features.AddFeature("mve");
- break;
- case ARMBuildAttrs::AllowMVEIntegerAndFloat:
- Features.AddFeature("mve.fp");
- break;
- }
- }
- Attr = Attributes.getAttributeValue(ARMBuildAttrs::DIV_use);
- if (Attr) {
- switch (*Attr) {
- default:
- break;
- case ARMBuildAttrs::DisallowDIV:
- Features.AddFeature("hwdiv", false);
- Features.AddFeature("hwdiv-arm", false);
- break;
- case ARMBuildAttrs::AllowDIVExt:
- Features.AddFeature("hwdiv");
- Features.AddFeature("hwdiv-arm");
- break;
- }
- }
- return Features;
- }
- Expected<SubtargetFeatures> ELFObjectFileBase::getRISCVFeatures() const {
- SubtargetFeatures Features;
- unsigned PlatformFlags = getPlatformFlags();
- if (PlatformFlags & ELF::EF_RISCV_RVC) {
- Features.AddFeature("c");
- }
- RISCVAttributeParser Attributes;
- if (Error E = getBuildAttributes(Attributes)) {
- return std::move(E);
- }
- std::optional<StringRef> Attr =
- Attributes.getAttributeString(RISCVAttrs::ARCH);
- if (Attr) {
- // Suppress version checking for experimental extensions to prevent erroring
- // when getting any unknown version of experimental extension.
- auto ParseResult = RISCVISAInfo::parseArchString(
- *Attr, /*EnableExperimentalExtension=*/true,
- /*ExperimentalExtensionVersionCheck=*/false,
- /*IgnoreUnknown=*/true);
- if (!ParseResult)
- return ParseResult.takeError();
- auto &ISAInfo = *ParseResult;
- if (ISAInfo->getXLen() == 32)
- Features.AddFeature("64bit", false);
- else if (ISAInfo->getXLen() == 64)
- Features.AddFeature("64bit");
- else
- llvm_unreachable("XLEN should be 32 or 64.");
- Features.addFeaturesVector(ISAInfo->toFeatureVector());
- }
- return Features;
- }
- SubtargetFeatures ELFObjectFileBase::getLoongArchFeatures() const {
- SubtargetFeatures Features;
- switch (getPlatformFlags() & ELF::EF_LOONGARCH_ABI_MODIFIER_MASK) {
- case ELF::EF_LOONGARCH_ABI_SOFT_FLOAT:
- break;
- case ELF::EF_LOONGARCH_ABI_DOUBLE_FLOAT:
- Features.AddFeature("d");
- // D implies F according to LoongArch ISA spec.
- [[fallthrough]];
- case ELF::EF_LOONGARCH_ABI_SINGLE_FLOAT:
- Features.AddFeature("f");
- break;
- }
- return Features;
- }
- Expected<SubtargetFeatures> ELFObjectFileBase::getFeatures() const {
- switch (getEMachine()) {
- case ELF::EM_MIPS:
- return getMIPSFeatures();
- case ELF::EM_ARM:
- return getARMFeatures();
- case ELF::EM_RISCV:
- return getRISCVFeatures();
- case ELF::EM_LOONGARCH:
- return getLoongArchFeatures();
- default:
- return SubtargetFeatures();
- }
- }
- std::optional<StringRef> ELFObjectFileBase::tryGetCPUName() const {
- switch (getEMachine()) {
- case ELF::EM_AMDGPU:
- return getAMDGPUCPUName();
- case ELF::EM_PPC64:
- return StringRef("future");
- default:
- return std::nullopt;
- }
- }
- StringRef ELFObjectFileBase::getAMDGPUCPUName() const {
- assert(getEMachine() == ELF::EM_AMDGPU);
- unsigned CPU = getPlatformFlags() & ELF::EF_AMDGPU_MACH;
- switch (CPU) {
- // Radeon HD 2000/3000 Series (R600).
- case ELF::EF_AMDGPU_MACH_R600_R600:
- return "r600";
- case ELF::EF_AMDGPU_MACH_R600_R630:
- return "r630";
- case ELF::EF_AMDGPU_MACH_R600_RS880:
- return "rs880";
- case ELF::EF_AMDGPU_MACH_R600_RV670:
- return "rv670";
- // Radeon HD 4000 Series (R700).
- case ELF::EF_AMDGPU_MACH_R600_RV710:
- return "rv710";
- case ELF::EF_AMDGPU_MACH_R600_RV730:
- return "rv730";
- case ELF::EF_AMDGPU_MACH_R600_RV770:
- return "rv770";
- // Radeon HD 5000 Series (Evergreen).
- case ELF::EF_AMDGPU_MACH_R600_CEDAR:
- return "cedar";
- case ELF::EF_AMDGPU_MACH_R600_CYPRESS:
- return "cypress";
- case ELF::EF_AMDGPU_MACH_R600_JUNIPER:
- return "juniper";
- case ELF::EF_AMDGPU_MACH_R600_REDWOOD:
- return "redwood";
- case ELF::EF_AMDGPU_MACH_R600_SUMO:
- return "sumo";
- // Radeon HD 6000 Series (Northern Islands).
- case ELF::EF_AMDGPU_MACH_R600_BARTS:
- return "barts";
- case ELF::EF_AMDGPU_MACH_R600_CAICOS:
- return "caicos";
- case ELF::EF_AMDGPU_MACH_R600_CAYMAN:
- return "cayman";
- case ELF::EF_AMDGPU_MACH_R600_TURKS:
- return "turks";
- // AMDGCN GFX6.
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX600:
- return "gfx600";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX601:
- return "gfx601";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX602:
- return "gfx602";
- // AMDGCN GFX7.
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX700:
- return "gfx700";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX701:
- return "gfx701";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX702:
- return "gfx702";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX703:
- return "gfx703";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX704:
- return "gfx704";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX705:
- return "gfx705";
- // AMDGCN GFX8.
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX801:
- return "gfx801";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX802:
- return "gfx802";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX803:
- return "gfx803";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX805:
- return "gfx805";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX810:
- return "gfx810";
- // AMDGCN GFX9.
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX900:
- return "gfx900";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX902:
- return "gfx902";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX904:
- return "gfx904";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX906:
- return "gfx906";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX908:
- return "gfx908";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX909:
- return "gfx909";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX90A:
- return "gfx90a";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX90C:
- return "gfx90c";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX940:
- return "gfx940";
- // AMDGCN GFX10.
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1010:
- return "gfx1010";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1011:
- return "gfx1011";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1012:
- return "gfx1012";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1013:
- return "gfx1013";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1030:
- return "gfx1030";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1031:
- return "gfx1031";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1032:
- return "gfx1032";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1033:
- return "gfx1033";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1034:
- return "gfx1034";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1035:
- return "gfx1035";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1036:
- return "gfx1036";
- // AMDGCN GFX11.
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1100:
- return "gfx1100";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1101:
- return "gfx1101";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1102:
- return "gfx1102";
- case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1103:
- return "gfx1103";
- default:
- llvm_unreachable("Unknown EF_AMDGPU_MACH value");
- }
- }
- // FIXME Encode from a tablegen description or target parser.
- void ELFObjectFileBase::setARMSubArch(Triple &TheTriple) const {
- if (TheTriple.getSubArch() != Triple::NoSubArch)
- return;
- ARMAttributeParser Attributes;
- if (Error E = getBuildAttributes(Attributes)) {
- // TODO Propagate Error.
- consumeError(std::move(E));
- return;
- }
- std::string Triple;
- // Default to ARM, but use the triple if it's been set.
- if (TheTriple.isThumb())
- Triple = "thumb";
- else
- Triple = "arm";
- std::optional<unsigned> Attr =
- Attributes.getAttributeValue(ARMBuildAttrs::CPU_arch);
- if (Attr) {
- switch (*Attr) {
- case ARMBuildAttrs::v4:
- Triple += "v4";
- break;
- case ARMBuildAttrs::v4T:
- Triple += "v4t";
- break;
- case ARMBuildAttrs::v5T:
- Triple += "v5t";
- break;
- case ARMBuildAttrs::v5TE:
- Triple += "v5te";
- break;
- case ARMBuildAttrs::v5TEJ:
- Triple += "v5tej";
- break;
- case ARMBuildAttrs::v6:
- Triple += "v6";
- break;
- case ARMBuildAttrs::v6KZ:
- Triple += "v6kz";
- break;
- case ARMBuildAttrs::v6T2:
- Triple += "v6t2";
- break;
- case ARMBuildAttrs::v6K:
- Triple += "v6k";
- break;
- case ARMBuildAttrs::v7: {
- std::optional<unsigned> ArchProfileAttr =
- Attributes.getAttributeValue(ARMBuildAttrs::CPU_arch_profile);
- if (ArchProfileAttr &&
- *ArchProfileAttr == ARMBuildAttrs::MicroControllerProfile)
- Triple += "v7m";
- else
- Triple += "v7";
- break;
- }
- case ARMBuildAttrs::v6_M:
- Triple += "v6m";
- break;
- case ARMBuildAttrs::v6S_M:
- Triple += "v6sm";
- break;
- case ARMBuildAttrs::v7E_M:
- Triple += "v7em";
- break;
- case ARMBuildAttrs::v8_A:
- Triple += "v8a";
- break;
- case ARMBuildAttrs::v8_R:
- Triple += "v8r";
- break;
- case ARMBuildAttrs::v8_M_Base:
- Triple += "v8m.base";
- break;
- case ARMBuildAttrs::v8_M_Main:
- Triple += "v8m.main";
- break;
- case ARMBuildAttrs::v8_1_M_Main:
- Triple += "v8.1m.main";
- break;
- case ARMBuildAttrs::v9_A:
- Triple += "v9a";
- break;
- }
- }
- if (!isLittleEndian())
- Triple += "eb";
- TheTriple.setArchName(Triple);
- }
- std::vector<std::pair<std::optional<DataRefImpl>, uint64_t>>
- ELFObjectFileBase::getPltAddresses() const {
- std::string Err;
- const auto Triple = makeTriple();
- const auto *T = TargetRegistry::lookupTarget(Triple.str(), Err);
- if (!T)
- return {};
- uint64_t JumpSlotReloc = 0;
- switch (Triple.getArch()) {
- case Triple::x86:
- JumpSlotReloc = ELF::R_386_JUMP_SLOT;
- break;
- case Triple::x86_64:
- JumpSlotReloc = ELF::R_X86_64_JUMP_SLOT;
- break;
- case Triple::aarch64:
- case Triple::aarch64_be:
- JumpSlotReloc = ELF::R_AARCH64_JUMP_SLOT;
- break;
- default:
- return {};
- }
- std::unique_ptr<const MCInstrInfo> MII(T->createMCInstrInfo());
- std::unique_ptr<const MCInstrAnalysis> MIA(
- T->createMCInstrAnalysis(MII.get()));
- if (!MIA)
- return {};
- std::optional<SectionRef> Plt, RelaPlt, GotPlt;
- for (const SectionRef &Section : sections()) {
- Expected<StringRef> NameOrErr = Section.getName();
- if (!NameOrErr) {
- consumeError(NameOrErr.takeError());
- continue;
- }
- StringRef Name = *NameOrErr;
- if (Name == ".plt")
- Plt = Section;
- else if (Name == ".rela.plt" || Name == ".rel.plt")
- RelaPlt = Section;
- else if (Name == ".got.plt")
- GotPlt = Section;
- }
- if (!Plt || !RelaPlt || !GotPlt)
- return {};
- Expected<StringRef> PltContents = Plt->getContents();
- if (!PltContents) {
- consumeError(PltContents.takeError());
- return {};
- }
- auto PltEntries = MIA->findPltEntries(Plt->getAddress(),
- arrayRefFromStringRef(*PltContents),
- GotPlt->getAddress(), Triple);
- // Build a map from GOT entry virtual address to PLT entry virtual address.
- DenseMap<uint64_t, uint64_t> GotToPlt;
- for (const auto &Entry : PltEntries)
- GotToPlt.insert(std::make_pair(Entry.second, Entry.first));
- // Find the relocations in the dynamic relocation table that point to
- // locations in the GOT for which we know the corresponding PLT entry.
- std::vector<std::pair<std::optional<DataRefImpl>, uint64_t>> Result;
- for (const auto &Relocation : RelaPlt->relocations()) {
- if (Relocation.getType() != JumpSlotReloc)
- continue;
- auto PltEntryIter = GotToPlt.find(Relocation.getOffset());
- if (PltEntryIter != GotToPlt.end()) {
- symbol_iterator Sym = Relocation.getSymbol();
- if (Sym == symbol_end())
- Result.emplace_back(std::nullopt, PltEntryIter->second);
- else
- Result.emplace_back(Sym->getRawDataRefImpl(), PltEntryIter->second);
- }
- }
- return Result;
- }
- template <class ELFT>
- Expected<std::vector<BBAddrMap>> static readBBAddrMapImpl(
- const ELFFile<ELFT> &EF, std::optional<unsigned> TextSectionIndex) {
- using Elf_Shdr = typename ELFT::Shdr;
- std::vector<BBAddrMap> BBAddrMaps;
- const auto &Sections = cantFail(EF.sections());
- for (const Elf_Shdr &Sec : Sections) {
- if (Sec.sh_type != ELF::SHT_LLVM_BB_ADDR_MAP &&
- Sec.sh_type != ELF::SHT_LLVM_BB_ADDR_MAP_V0)
- continue;
- if (TextSectionIndex) {
- Expected<const Elf_Shdr *> TextSecOrErr = EF.getSection(Sec.sh_link);
- if (!TextSecOrErr)
- return createError("unable to get the linked-to section for " +
- describe(EF, Sec) + ": " +
- toString(TextSecOrErr.takeError()));
- if (*TextSectionIndex != std::distance(Sections.begin(), *TextSecOrErr))
- continue;
- }
- Expected<std::vector<BBAddrMap>> BBAddrMapOrErr = EF.decodeBBAddrMap(Sec);
- if (!BBAddrMapOrErr)
- return createError("unable to read " + describe(EF, Sec) + ": " +
- toString(BBAddrMapOrErr.takeError()));
- std::move(BBAddrMapOrErr->begin(), BBAddrMapOrErr->end(),
- std::back_inserter(BBAddrMaps));
- }
- return BBAddrMaps;
- }
- template <class ELFT>
- static Expected<std::vector<VersionEntry>>
- readDynsymVersionsImpl(const ELFFile<ELFT> &EF,
- ELFObjectFileBase::elf_symbol_iterator_range Symbols) {
- using Elf_Shdr = typename ELFT::Shdr;
- const Elf_Shdr *VerSec = nullptr;
- const Elf_Shdr *VerNeedSec = nullptr;
- const Elf_Shdr *VerDefSec = nullptr;
- // The user should ensure sections() can't fail here.
- for (const Elf_Shdr &Sec : cantFail(EF.sections())) {
- if (Sec.sh_type == ELF::SHT_GNU_versym)
- VerSec = &Sec;
- else if (Sec.sh_type == ELF::SHT_GNU_verdef)
- VerDefSec = &Sec;
- else if (Sec.sh_type == ELF::SHT_GNU_verneed)
- VerNeedSec = &Sec;
- }
- if (!VerSec)
- return std::vector<VersionEntry>();
- Expected<SmallVector<std::optional<VersionEntry>, 0>> MapOrErr =
- EF.loadVersionMap(VerNeedSec, VerDefSec);
- if (!MapOrErr)
- return MapOrErr.takeError();
- std::vector<VersionEntry> Ret;
- size_t I = 0;
- for (const ELFSymbolRef &Sym : Symbols) {
- ++I;
- Expected<const typename ELFT::Versym *> VerEntryOrErr =
- EF.template getEntry<typename ELFT::Versym>(*VerSec, I);
- if (!VerEntryOrErr)
- return createError("unable to read an entry with index " + Twine(I) +
- " from " + describe(EF, *VerSec) + ": " +
- toString(VerEntryOrErr.takeError()));
- Expected<uint32_t> FlagsOrErr = Sym.getFlags();
- if (!FlagsOrErr)
- return createError("unable to read flags for symbol with index " +
- Twine(I) + ": " + toString(FlagsOrErr.takeError()));
- bool IsDefault;
- Expected<StringRef> VerOrErr = EF.getSymbolVersionByIndex(
- (*VerEntryOrErr)->vs_index, IsDefault, *MapOrErr,
- (*FlagsOrErr) & SymbolRef::SF_Undefined);
- if (!VerOrErr)
- return createError("unable to get a version for entry " + Twine(I) +
- " of " + describe(EF, *VerSec) + ": " +
- toString(VerOrErr.takeError()));
- Ret.push_back({(*VerOrErr).str(), IsDefault});
- }
- return Ret;
- }
- Expected<std::vector<VersionEntry>>
- ELFObjectFileBase::readDynsymVersions() const {
- elf_symbol_iterator_range Symbols = getDynamicSymbolIterators();
- if (const auto *Obj = dyn_cast<ELF32LEObjectFile>(this))
- return readDynsymVersionsImpl(Obj->getELFFile(), Symbols);
- if (const auto *Obj = dyn_cast<ELF32BEObjectFile>(this))
- return readDynsymVersionsImpl(Obj->getELFFile(), Symbols);
- if (const auto *Obj = dyn_cast<ELF64LEObjectFile>(this))
- return readDynsymVersionsImpl(Obj->getELFFile(), Symbols);
- return readDynsymVersionsImpl(cast<ELF64BEObjectFile>(this)->getELFFile(),
- Symbols);
- }
- Expected<std::vector<BBAddrMap>> ELFObjectFileBase::readBBAddrMap(
- std::optional<unsigned> TextSectionIndex) const {
- if (const auto *Obj = dyn_cast<ELF32LEObjectFile>(this))
- return readBBAddrMapImpl(Obj->getELFFile(), TextSectionIndex);
- if (const auto *Obj = dyn_cast<ELF64LEObjectFile>(this))
- return readBBAddrMapImpl(Obj->getELFFile(), TextSectionIndex);
- if (const auto *Obj = dyn_cast<ELF32BEObjectFile>(this))
- return readBBAddrMapImpl(Obj->getELFFile(), TextSectionIndex);
- if (const auto *Obj = cast<ELF64BEObjectFile>(this))
- return readBBAddrMapImpl(Obj->getELFFile(), TextSectionIndex);
- else
- llvm_unreachable("Unsupported binary format");
- }
|