10b57cec5SDimitry Andric //===- ELFObjectFile.cpp - ELF object file implementation -----------------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric //
90b57cec5SDimitry Andric // Part of the ELFObjectFile class implementation.
100b57cec5SDimitry Andric //
110b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
120b57cec5SDimitry Andric 
130b57cec5SDimitry Andric #include "llvm/Object/ELFObjectFile.h"
140b57cec5SDimitry Andric #include "llvm/BinaryFormat/ELF.h"
150b57cec5SDimitry Andric #include "llvm/MC/MCInstrAnalysis.h"
16349cc55cSDimitry Andric #include "llvm/MC/TargetRegistry.h"
170b57cec5SDimitry Andric #include "llvm/Object/ELF.h"
180b57cec5SDimitry Andric #include "llvm/Object/ELFTypes.h"
190b57cec5SDimitry Andric #include "llvm/Object/Error.h"
200b57cec5SDimitry Andric #include "llvm/Support/ARMAttributeParser.h"
210b57cec5SDimitry Andric #include "llvm/Support/ARMBuildAttributes.h"
220b57cec5SDimitry Andric #include "llvm/Support/ErrorHandling.h"
230b57cec5SDimitry Andric #include "llvm/Support/MathExtras.h"
245ffd83dbSDimitry Andric #include "llvm/Support/RISCVAttributeParser.h"
255ffd83dbSDimitry Andric #include "llvm/Support/RISCVAttributes.h"
26bdd1243dSDimitry Andric #include "llvm/Support/RISCVISAInfo.h"
2706c3fb27SDimitry Andric #include "llvm/TargetParser/SubtargetFeature.h"
2806c3fb27SDimitry Andric #include "llvm/TargetParser/Triple.h"
290b57cec5SDimitry Andric #include <algorithm>
300b57cec5SDimitry Andric #include <cstddef>
310b57cec5SDimitry Andric #include <cstdint>
320b57cec5SDimitry Andric #include <memory>
33bdd1243dSDimitry Andric #include <optional>
340b57cec5SDimitry Andric #include <string>
350b57cec5SDimitry Andric #include <utility>
360b57cec5SDimitry Andric 
370b57cec5SDimitry Andric using namespace llvm;
380b57cec5SDimitry Andric using namespace object;
390b57cec5SDimitry Andric 
400b57cec5SDimitry Andric const EnumEntry<unsigned> llvm::object::ElfSymbolTypes[NumElfSymbolTypes] = {
410b57cec5SDimitry Andric     {"None", "NOTYPE", ELF::STT_NOTYPE},
420b57cec5SDimitry Andric     {"Object", "OBJECT", ELF::STT_OBJECT},
430b57cec5SDimitry Andric     {"Function", "FUNC", ELF::STT_FUNC},
440b57cec5SDimitry Andric     {"Section", "SECTION", ELF::STT_SECTION},
450b57cec5SDimitry Andric     {"File", "FILE", ELF::STT_FILE},
460b57cec5SDimitry Andric     {"Common", "COMMON", ELF::STT_COMMON},
470b57cec5SDimitry Andric     {"TLS", "TLS", ELF::STT_TLS},
488bcb0991SDimitry Andric     {"Unknown", "<unknown>: 7", 7},
498bcb0991SDimitry Andric     {"Unknown", "<unknown>: 8", 8},
508bcb0991SDimitry Andric     {"Unknown", "<unknown>: 9", 9},
518bcb0991SDimitry Andric     {"GNU_IFunc", "IFUNC", ELF::STT_GNU_IFUNC},
528bcb0991SDimitry Andric     {"OS Specific", "<OS specific>: 11", 11},
538bcb0991SDimitry Andric     {"OS Specific", "<OS specific>: 12", 12},
548bcb0991SDimitry Andric     {"Proc Specific", "<processor specific>: 13", 13},
558bcb0991SDimitry Andric     {"Proc Specific", "<processor specific>: 14", 14},
568bcb0991SDimitry Andric     {"Proc Specific", "<processor specific>: 15", 15}
578bcb0991SDimitry Andric };
580b57cec5SDimitry Andric 
ELFObjectFileBase(unsigned int Type,MemoryBufferRef Source)590b57cec5SDimitry Andric ELFObjectFileBase::ELFObjectFileBase(unsigned int Type, MemoryBufferRef Source)
600b57cec5SDimitry Andric     : ObjectFile(Type, Source) {}
610b57cec5SDimitry Andric 
620b57cec5SDimitry Andric template <class ELFT>
630b57cec5SDimitry Andric static Expected<std::unique_ptr<ELFObjectFile<ELFT>>>
createPtr(MemoryBufferRef Object,bool InitContent)64e8d8bef9SDimitry Andric createPtr(MemoryBufferRef Object, bool InitContent) {
65e8d8bef9SDimitry Andric   auto Ret = ELFObjectFile<ELFT>::create(Object, InitContent);
660b57cec5SDimitry Andric   if (Error E = Ret.takeError())
670b57cec5SDimitry Andric     return std::move(E);
688bcb0991SDimitry Andric   return std::make_unique<ELFObjectFile<ELFT>>(std::move(*Ret));
690b57cec5SDimitry Andric }
700b57cec5SDimitry Andric 
710b57cec5SDimitry Andric Expected<std::unique_ptr<ObjectFile>>
createELFObjectFile(MemoryBufferRef Obj,bool InitContent)72e8d8bef9SDimitry Andric ObjectFile::createELFObjectFile(MemoryBufferRef Obj, bool InitContent) {
730b57cec5SDimitry Andric   std::pair<unsigned char, unsigned char> Ident =
740b57cec5SDimitry Andric       getElfArchType(Obj.getBuffer());
750b57cec5SDimitry Andric   std::size_t MaxAlignment =
7606c3fb27SDimitry Andric       1ULL << llvm::countr_zero(
77e8d8bef9SDimitry Andric           reinterpret_cast<uintptr_t>(Obj.getBufferStart()));
780b57cec5SDimitry Andric 
790b57cec5SDimitry Andric   if (MaxAlignment < 2)
800b57cec5SDimitry Andric     return createError("Insufficient alignment");
810b57cec5SDimitry Andric 
820b57cec5SDimitry Andric   if (Ident.first == ELF::ELFCLASS32) {
830b57cec5SDimitry Andric     if (Ident.second == ELF::ELFDATA2LSB)
84e8d8bef9SDimitry Andric       return createPtr<ELF32LE>(Obj, InitContent);
850b57cec5SDimitry Andric     else if (Ident.second == ELF::ELFDATA2MSB)
86e8d8bef9SDimitry Andric       return createPtr<ELF32BE>(Obj, InitContent);
870b57cec5SDimitry Andric     else
880b57cec5SDimitry Andric       return createError("Invalid ELF data");
890b57cec5SDimitry Andric   } else if (Ident.first == ELF::ELFCLASS64) {
900b57cec5SDimitry Andric     if (Ident.second == ELF::ELFDATA2LSB)
91e8d8bef9SDimitry Andric       return createPtr<ELF64LE>(Obj, InitContent);
920b57cec5SDimitry Andric     else if (Ident.second == ELF::ELFDATA2MSB)
93e8d8bef9SDimitry Andric       return createPtr<ELF64BE>(Obj, InitContent);
940b57cec5SDimitry Andric     else
950b57cec5SDimitry Andric       return createError("Invalid ELF data");
960b57cec5SDimitry Andric   }
970b57cec5SDimitry Andric   return createError("Invalid ELF class");
980b57cec5SDimitry Andric }
990b57cec5SDimitry Andric 
getMIPSFeatures() const1000b57cec5SDimitry Andric SubtargetFeatures ELFObjectFileBase::getMIPSFeatures() const {
1010b57cec5SDimitry Andric   SubtargetFeatures Features;
1020b57cec5SDimitry Andric   unsigned PlatformFlags = getPlatformFlags();
1030b57cec5SDimitry Andric 
1040b57cec5SDimitry Andric   switch (PlatformFlags & ELF::EF_MIPS_ARCH) {
1050b57cec5SDimitry Andric   case ELF::EF_MIPS_ARCH_1:
1060b57cec5SDimitry Andric     break;
1070b57cec5SDimitry Andric   case ELF::EF_MIPS_ARCH_2:
1080b57cec5SDimitry Andric     Features.AddFeature("mips2");
1090b57cec5SDimitry Andric     break;
1100b57cec5SDimitry Andric   case ELF::EF_MIPS_ARCH_3:
1110b57cec5SDimitry Andric     Features.AddFeature("mips3");
1120b57cec5SDimitry Andric     break;
1130b57cec5SDimitry Andric   case ELF::EF_MIPS_ARCH_4:
1140b57cec5SDimitry Andric     Features.AddFeature("mips4");
1150b57cec5SDimitry Andric     break;
1160b57cec5SDimitry Andric   case ELF::EF_MIPS_ARCH_5:
1170b57cec5SDimitry Andric     Features.AddFeature("mips5");
1180b57cec5SDimitry Andric     break;
1190b57cec5SDimitry Andric   case ELF::EF_MIPS_ARCH_32:
1200b57cec5SDimitry Andric     Features.AddFeature("mips32");
1210b57cec5SDimitry Andric     break;
1220b57cec5SDimitry Andric   case ELF::EF_MIPS_ARCH_64:
1230b57cec5SDimitry Andric     Features.AddFeature("mips64");
1240b57cec5SDimitry Andric     break;
1250b57cec5SDimitry Andric   case ELF::EF_MIPS_ARCH_32R2:
1260b57cec5SDimitry Andric     Features.AddFeature("mips32r2");
1270b57cec5SDimitry Andric     break;
1280b57cec5SDimitry Andric   case ELF::EF_MIPS_ARCH_64R2:
1290b57cec5SDimitry Andric     Features.AddFeature("mips64r2");
1300b57cec5SDimitry Andric     break;
1310b57cec5SDimitry Andric   case ELF::EF_MIPS_ARCH_32R6:
1320b57cec5SDimitry Andric     Features.AddFeature("mips32r6");
1330b57cec5SDimitry Andric     break;
1340b57cec5SDimitry Andric   case ELF::EF_MIPS_ARCH_64R6:
1350b57cec5SDimitry Andric     Features.AddFeature("mips64r6");
1360b57cec5SDimitry Andric     break;
1370b57cec5SDimitry Andric   default:
1380b57cec5SDimitry Andric     llvm_unreachable("Unknown EF_MIPS_ARCH value");
1390b57cec5SDimitry Andric   }
1400b57cec5SDimitry Andric 
1410b57cec5SDimitry Andric   switch (PlatformFlags & ELF::EF_MIPS_MACH) {
1420b57cec5SDimitry Andric   case ELF::EF_MIPS_MACH_NONE:
1430b57cec5SDimitry Andric     // No feature associated with this value.
1440b57cec5SDimitry Andric     break;
1450b57cec5SDimitry Andric   case ELF::EF_MIPS_MACH_OCTEON:
1460b57cec5SDimitry Andric     Features.AddFeature("cnmips");
1470b57cec5SDimitry Andric     break;
1480b57cec5SDimitry Andric   default:
1490b57cec5SDimitry Andric     llvm_unreachable("Unknown EF_MIPS_ARCH value");
1500b57cec5SDimitry Andric   }
1510b57cec5SDimitry Andric 
1520b57cec5SDimitry Andric   if (PlatformFlags & ELF::EF_MIPS_ARCH_ASE_M16)
1530b57cec5SDimitry Andric     Features.AddFeature("mips16");
1540b57cec5SDimitry Andric   if (PlatformFlags & ELF::EF_MIPS_MICROMIPS)
1550b57cec5SDimitry Andric     Features.AddFeature("micromips");
1560b57cec5SDimitry Andric 
1570b57cec5SDimitry Andric   return Features;
1580b57cec5SDimitry Andric }
1590b57cec5SDimitry Andric 
getARMFeatures() const1600b57cec5SDimitry Andric SubtargetFeatures ELFObjectFileBase::getARMFeatures() const {
1610b57cec5SDimitry Andric   SubtargetFeatures Features;
1620b57cec5SDimitry Andric   ARMAttributeParser Attributes;
1635ffd83dbSDimitry Andric   if (Error E = getBuildAttributes(Attributes)) {
1645ffd83dbSDimitry Andric     consumeError(std::move(E));
1650b57cec5SDimitry Andric     return SubtargetFeatures();
1665ffd83dbSDimitry Andric   }
1670b57cec5SDimitry Andric 
1680b57cec5SDimitry Andric   // both ARMv7-M and R have to support thumb hardware div
1690b57cec5SDimitry Andric   bool isV7 = false;
170bdd1243dSDimitry Andric   std::optional<unsigned> Attr =
1715ffd83dbSDimitry Andric       Attributes.getAttributeValue(ARMBuildAttrs::CPU_arch);
17281ad6265SDimitry Andric   if (Attr)
173bdd1243dSDimitry Andric     isV7 = *Attr == ARMBuildAttrs::v7;
1740b57cec5SDimitry Andric 
1755ffd83dbSDimitry Andric   Attr = Attributes.getAttributeValue(ARMBuildAttrs::CPU_arch_profile);
17681ad6265SDimitry Andric   if (Attr) {
177bdd1243dSDimitry Andric     switch (*Attr) {
1780b57cec5SDimitry Andric     case ARMBuildAttrs::ApplicationProfile:
1790b57cec5SDimitry Andric       Features.AddFeature("aclass");
1800b57cec5SDimitry Andric       break;
1810b57cec5SDimitry Andric     case ARMBuildAttrs::RealTimeProfile:
1820b57cec5SDimitry Andric       Features.AddFeature("rclass");
1830b57cec5SDimitry Andric       if (isV7)
1840b57cec5SDimitry Andric         Features.AddFeature("hwdiv");
1850b57cec5SDimitry Andric       break;
1860b57cec5SDimitry Andric     case ARMBuildAttrs::MicroControllerProfile:
1870b57cec5SDimitry Andric       Features.AddFeature("mclass");
1880b57cec5SDimitry Andric       if (isV7)
1890b57cec5SDimitry Andric         Features.AddFeature("hwdiv");
1900b57cec5SDimitry Andric       break;
1910b57cec5SDimitry Andric     }
1920b57cec5SDimitry Andric   }
1930b57cec5SDimitry Andric 
1945ffd83dbSDimitry Andric   Attr = Attributes.getAttributeValue(ARMBuildAttrs::THUMB_ISA_use);
19581ad6265SDimitry Andric   if (Attr) {
196bdd1243dSDimitry Andric     switch (*Attr) {
1970b57cec5SDimitry Andric     default:
1980b57cec5SDimitry Andric       break;
1990b57cec5SDimitry Andric     case ARMBuildAttrs::Not_Allowed:
2000b57cec5SDimitry Andric       Features.AddFeature("thumb", false);
2010b57cec5SDimitry Andric       Features.AddFeature("thumb2", false);
2020b57cec5SDimitry Andric       break;
2030b57cec5SDimitry Andric     case ARMBuildAttrs::AllowThumb32:
2040b57cec5SDimitry Andric       Features.AddFeature("thumb2");
2050b57cec5SDimitry Andric       break;
2060b57cec5SDimitry Andric     }
2070b57cec5SDimitry Andric   }
2080b57cec5SDimitry Andric 
2095ffd83dbSDimitry Andric   Attr = Attributes.getAttributeValue(ARMBuildAttrs::FP_arch);
21081ad6265SDimitry Andric   if (Attr) {
211bdd1243dSDimitry Andric     switch (*Attr) {
2120b57cec5SDimitry Andric     default:
2130b57cec5SDimitry Andric       break;
2140b57cec5SDimitry Andric     case ARMBuildAttrs::Not_Allowed:
2158bcb0991SDimitry Andric       Features.AddFeature("vfp2sp", false);
2160b57cec5SDimitry Andric       Features.AddFeature("vfp3d16sp", false);
2170b57cec5SDimitry Andric       Features.AddFeature("vfp4d16sp", false);
2180b57cec5SDimitry Andric       break;
2190b57cec5SDimitry Andric     case ARMBuildAttrs::AllowFPv2:
2200b57cec5SDimitry Andric       Features.AddFeature("vfp2");
2210b57cec5SDimitry Andric       break;
2220b57cec5SDimitry Andric     case ARMBuildAttrs::AllowFPv3A:
2230b57cec5SDimitry Andric     case ARMBuildAttrs::AllowFPv3B:
2240b57cec5SDimitry Andric       Features.AddFeature("vfp3");
2250b57cec5SDimitry Andric       break;
2260b57cec5SDimitry Andric     case ARMBuildAttrs::AllowFPv4A:
2270b57cec5SDimitry Andric     case ARMBuildAttrs::AllowFPv4B:
2280b57cec5SDimitry Andric       Features.AddFeature("vfp4");
2290b57cec5SDimitry Andric       break;
2300b57cec5SDimitry Andric     }
2310b57cec5SDimitry Andric   }
2320b57cec5SDimitry Andric 
2335ffd83dbSDimitry Andric   Attr = Attributes.getAttributeValue(ARMBuildAttrs::Advanced_SIMD_arch);
23481ad6265SDimitry Andric   if (Attr) {
235bdd1243dSDimitry Andric     switch (*Attr) {
2360b57cec5SDimitry Andric     default:
2370b57cec5SDimitry Andric       break;
2380b57cec5SDimitry Andric     case ARMBuildAttrs::Not_Allowed:
2390b57cec5SDimitry Andric       Features.AddFeature("neon", false);
2400b57cec5SDimitry Andric       Features.AddFeature("fp16", false);
2410b57cec5SDimitry Andric       break;
2420b57cec5SDimitry Andric     case ARMBuildAttrs::AllowNeon:
2430b57cec5SDimitry Andric       Features.AddFeature("neon");
2440b57cec5SDimitry Andric       break;
2450b57cec5SDimitry Andric     case ARMBuildAttrs::AllowNeon2:
2460b57cec5SDimitry Andric       Features.AddFeature("neon");
2470b57cec5SDimitry Andric       Features.AddFeature("fp16");
2480b57cec5SDimitry Andric       break;
2490b57cec5SDimitry Andric     }
2500b57cec5SDimitry Andric   }
2510b57cec5SDimitry Andric 
2525ffd83dbSDimitry Andric   Attr = Attributes.getAttributeValue(ARMBuildAttrs::MVE_arch);
25381ad6265SDimitry Andric   if (Attr) {
254bdd1243dSDimitry Andric     switch (*Attr) {
2550b57cec5SDimitry Andric     default:
2560b57cec5SDimitry Andric       break;
2570b57cec5SDimitry Andric     case ARMBuildAttrs::Not_Allowed:
2580b57cec5SDimitry Andric       Features.AddFeature("mve", false);
2590b57cec5SDimitry Andric       Features.AddFeature("mve.fp", false);
2600b57cec5SDimitry Andric       break;
2610b57cec5SDimitry Andric     case ARMBuildAttrs::AllowMVEInteger:
2620b57cec5SDimitry Andric       Features.AddFeature("mve.fp", false);
2630b57cec5SDimitry Andric       Features.AddFeature("mve");
2640b57cec5SDimitry Andric       break;
2650b57cec5SDimitry Andric     case ARMBuildAttrs::AllowMVEIntegerAndFloat:
2660b57cec5SDimitry Andric       Features.AddFeature("mve.fp");
2670b57cec5SDimitry Andric       break;
2680b57cec5SDimitry Andric     }
2690b57cec5SDimitry Andric   }
2700b57cec5SDimitry Andric 
2715ffd83dbSDimitry Andric   Attr = Attributes.getAttributeValue(ARMBuildAttrs::DIV_use);
27281ad6265SDimitry Andric   if (Attr) {
273bdd1243dSDimitry Andric     switch (*Attr) {
2740b57cec5SDimitry Andric     default:
2750b57cec5SDimitry Andric       break;
2760b57cec5SDimitry Andric     case ARMBuildAttrs::DisallowDIV:
2770b57cec5SDimitry Andric       Features.AddFeature("hwdiv", false);
2780b57cec5SDimitry Andric       Features.AddFeature("hwdiv-arm", false);
2790b57cec5SDimitry Andric       break;
2800b57cec5SDimitry Andric     case ARMBuildAttrs::AllowDIVExt:
2810b57cec5SDimitry Andric       Features.AddFeature("hwdiv");
2820b57cec5SDimitry Andric       Features.AddFeature("hwdiv-arm");
2830b57cec5SDimitry Andric       break;
2840b57cec5SDimitry Andric     }
2850b57cec5SDimitry Andric   }
2860b57cec5SDimitry Andric 
2870b57cec5SDimitry Andric   return Features;
2880b57cec5SDimitry Andric }
2890b57cec5SDimitry Andric 
getRISCVFeatures() const290bdd1243dSDimitry Andric Expected<SubtargetFeatures> ELFObjectFileBase::getRISCVFeatures() const {
2910b57cec5SDimitry Andric   SubtargetFeatures Features;
2920b57cec5SDimitry Andric   unsigned PlatformFlags = getPlatformFlags();
2930b57cec5SDimitry Andric 
2940b57cec5SDimitry Andric   if (PlatformFlags & ELF::EF_RISCV_RVC) {
2950b57cec5SDimitry Andric     Features.AddFeature("c");
2960b57cec5SDimitry Andric   }
2970b57cec5SDimitry Andric 
2985ffd83dbSDimitry Andric   RISCVAttributeParser Attributes;
2995ffd83dbSDimitry Andric   if (Error E = getBuildAttributes(Attributes)) {
300bdd1243dSDimitry Andric     return std::move(E);
3015ffd83dbSDimitry Andric   }
3025ffd83dbSDimitry Andric 
303bdd1243dSDimitry Andric   std::optional<StringRef> Attr =
304bdd1243dSDimitry Andric       Attributes.getAttributeString(RISCVAttrs::ARCH);
30581ad6265SDimitry Andric   if (Attr) {
3061ac55f4cSDimitry Andric     auto ParseResult = RISCVISAInfo::parseNormalizedArchString(*Attr);
307bdd1243dSDimitry Andric     if (!ParseResult)
308bdd1243dSDimitry Andric       return ParseResult.takeError();
309bdd1243dSDimitry Andric     auto &ISAInfo = *ParseResult;
310bdd1243dSDimitry Andric 
311bdd1243dSDimitry Andric     if (ISAInfo->getXLen() == 32)
3125ffd83dbSDimitry Andric       Features.AddFeature("64bit", false);
313bdd1243dSDimitry Andric     else if (ISAInfo->getXLen() == 64)
3145ffd83dbSDimitry Andric       Features.AddFeature("64bit");
315bdd1243dSDimitry Andric     else
316bdd1243dSDimitry Andric       llvm_unreachable("XLEN should be 32 or 64.");
3175ffd83dbSDimitry Andric 
3181db9f3b2SDimitry Andric     Features.addFeaturesVector(ISAInfo->toFeatures());
3195ffd83dbSDimitry Andric   }
3205ffd83dbSDimitry Andric 
3210b57cec5SDimitry Andric   return Features;
3220b57cec5SDimitry Andric }
3230b57cec5SDimitry Andric 
getLoongArchFeatures() const324bdd1243dSDimitry Andric SubtargetFeatures ELFObjectFileBase::getLoongArchFeatures() const {
325bdd1243dSDimitry Andric   SubtargetFeatures Features;
326bdd1243dSDimitry Andric 
327bdd1243dSDimitry Andric   switch (getPlatformFlags() & ELF::EF_LOONGARCH_ABI_MODIFIER_MASK) {
328bdd1243dSDimitry Andric   case ELF::EF_LOONGARCH_ABI_SOFT_FLOAT:
329bdd1243dSDimitry Andric     break;
330bdd1243dSDimitry Andric   case ELF::EF_LOONGARCH_ABI_DOUBLE_FLOAT:
331bdd1243dSDimitry Andric     Features.AddFeature("d");
332bdd1243dSDimitry Andric     // D implies F according to LoongArch ISA spec.
333bdd1243dSDimitry Andric     [[fallthrough]];
334bdd1243dSDimitry Andric   case ELF::EF_LOONGARCH_ABI_SINGLE_FLOAT:
335bdd1243dSDimitry Andric     Features.AddFeature("f");
336bdd1243dSDimitry Andric     break;
337bdd1243dSDimitry Andric   }
338bdd1243dSDimitry Andric 
339bdd1243dSDimitry Andric   return Features;
340bdd1243dSDimitry Andric }
341bdd1243dSDimitry Andric 
getFeatures() const342bdd1243dSDimitry Andric Expected<SubtargetFeatures> ELFObjectFileBase::getFeatures() const {
3430b57cec5SDimitry Andric   switch (getEMachine()) {
3440b57cec5SDimitry Andric   case ELF::EM_MIPS:
3450b57cec5SDimitry Andric     return getMIPSFeatures();
3460b57cec5SDimitry Andric   case ELF::EM_ARM:
3470b57cec5SDimitry Andric     return getARMFeatures();
3480b57cec5SDimitry Andric   case ELF::EM_RISCV:
3490b57cec5SDimitry Andric     return getRISCVFeatures();
350bdd1243dSDimitry Andric   case ELF::EM_LOONGARCH:
351bdd1243dSDimitry Andric     return getLoongArchFeatures();
3520b57cec5SDimitry Andric   default:
3530b57cec5SDimitry Andric     return SubtargetFeatures();
3540b57cec5SDimitry Andric   }
3550b57cec5SDimitry Andric }
3560b57cec5SDimitry Andric 
tryGetCPUName() const357bdd1243dSDimitry Andric std::optional<StringRef> ELFObjectFileBase::tryGetCPUName() const {
358e8d8bef9SDimitry Andric   switch (getEMachine()) {
359e8d8bef9SDimitry Andric   case ELF::EM_AMDGPU:
360e8d8bef9SDimitry Andric     return getAMDGPUCPUName();
361cb14a3feSDimitry Andric   case ELF::EM_CUDA:
362cb14a3feSDimitry Andric     return getNVPTXCPUName();
36306c3fb27SDimitry Andric   case ELF::EM_PPC:
36481ad6265SDimitry Andric   case ELF::EM_PPC64:
36581ad6265SDimitry Andric     return StringRef("future");
366e8d8bef9SDimitry Andric   default:
367bdd1243dSDimitry Andric     return std::nullopt;
368e8d8bef9SDimitry Andric   }
369e8d8bef9SDimitry Andric }
370e8d8bef9SDimitry Andric 
getAMDGPUCPUName() const371e8d8bef9SDimitry Andric StringRef ELFObjectFileBase::getAMDGPUCPUName() const {
372e8d8bef9SDimitry Andric   assert(getEMachine() == ELF::EM_AMDGPU);
373e8d8bef9SDimitry Andric   unsigned CPU = getPlatformFlags() & ELF::EF_AMDGPU_MACH;
374e8d8bef9SDimitry Andric 
375e8d8bef9SDimitry Andric   switch (CPU) {
376e8d8bef9SDimitry Andric   // Radeon HD 2000/3000 Series (R600).
377e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_R600:
378e8d8bef9SDimitry Andric     return "r600";
379e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_R630:
380e8d8bef9SDimitry Andric     return "r630";
381e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_RS880:
382e8d8bef9SDimitry Andric     return "rs880";
383e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_RV670:
384e8d8bef9SDimitry Andric     return "rv670";
385e8d8bef9SDimitry Andric 
386e8d8bef9SDimitry Andric   // Radeon HD 4000 Series (R700).
387e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_RV710:
388e8d8bef9SDimitry Andric     return "rv710";
389e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_RV730:
390e8d8bef9SDimitry Andric     return "rv730";
391e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_RV770:
392e8d8bef9SDimitry Andric     return "rv770";
393e8d8bef9SDimitry Andric 
394e8d8bef9SDimitry Andric   // Radeon HD 5000 Series (Evergreen).
395e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_CEDAR:
396e8d8bef9SDimitry Andric     return "cedar";
397e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_CYPRESS:
398e8d8bef9SDimitry Andric     return "cypress";
399e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_JUNIPER:
400e8d8bef9SDimitry Andric     return "juniper";
401e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_REDWOOD:
402e8d8bef9SDimitry Andric     return "redwood";
403e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_SUMO:
404e8d8bef9SDimitry Andric     return "sumo";
405e8d8bef9SDimitry Andric 
406e8d8bef9SDimitry Andric   // Radeon HD 6000 Series (Northern Islands).
407e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_BARTS:
408e8d8bef9SDimitry Andric     return "barts";
409e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_CAICOS:
410e8d8bef9SDimitry Andric     return "caicos";
411e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_CAYMAN:
412e8d8bef9SDimitry Andric     return "cayman";
413e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_R600_TURKS:
414e8d8bef9SDimitry Andric     return "turks";
415e8d8bef9SDimitry Andric 
416e8d8bef9SDimitry Andric   // AMDGCN GFX6.
417e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX600:
418e8d8bef9SDimitry Andric     return "gfx600";
419e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX601:
420e8d8bef9SDimitry Andric     return "gfx601";
421e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX602:
422e8d8bef9SDimitry Andric     return "gfx602";
423e8d8bef9SDimitry Andric 
424e8d8bef9SDimitry Andric   // AMDGCN GFX7.
425e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX700:
426e8d8bef9SDimitry Andric     return "gfx700";
427e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX701:
428e8d8bef9SDimitry Andric     return "gfx701";
429e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX702:
430e8d8bef9SDimitry Andric     return "gfx702";
431e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX703:
432e8d8bef9SDimitry Andric     return "gfx703";
433e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX704:
434e8d8bef9SDimitry Andric     return "gfx704";
435e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX705:
436e8d8bef9SDimitry Andric     return "gfx705";
437e8d8bef9SDimitry Andric 
438e8d8bef9SDimitry Andric   // AMDGCN GFX8.
439e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX801:
440e8d8bef9SDimitry Andric     return "gfx801";
441e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX802:
442e8d8bef9SDimitry Andric     return "gfx802";
443e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX803:
444e8d8bef9SDimitry Andric     return "gfx803";
445e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX805:
446e8d8bef9SDimitry Andric     return "gfx805";
447e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX810:
448e8d8bef9SDimitry Andric     return "gfx810";
449e8d8bef9SDimitry Andric 
450e8d8bef9SDimitry Andric   // AMDGCN GFX9.
451e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX900:
452e8d8bef9SDimitry Andric     return "gfx900";
453e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX902:
454e8d8bef9SDimitry Andric     return "gfx902";
455e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX904:
456e8d8bef9SDimitry Andric     return "gfx904";
457e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX906:
458e8d8bef9SDimitry Andric     return "gfx906";
459e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX908:
460e8d8bef9SDimitry Andric     return "gfx908";
461e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX909:
462e8d8bef9SDimitry Andric     return "gfx909";
463fe6060f1SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX90A:
464fe6060f1SDimitry Andric     return "gfx90a";
465e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX90C:
466e8d8bef9SDimitry Andric     return "gfx90c";
46781ad6265SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX940:
46881ad6265SDimitry Andric     return "gfx940";
46906c3fb27SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX941:
47006c3fb27SDimitry Andric     return "gfx941";
47106c3fb27SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX942:
47206c3fb27SDimitry Andric     return "gfx942";
473e8d8bef9SDimitry Andric 
474e8d8bef9SDimitry Andric   // AMDGCN GFX10.
475e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1010:
476e8d8bef9SDimitry Andric     return "gfx1010";
477e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1011:
478e8d8bef9SDimitry Andric     return "gfx1011";
479e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1012:
480e8d8bef9SDimitry Andric     return "gfx1012";
481fe6060f1SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1013:
482fe6060f1SDimitry Andric     return "gfx1013";
483e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1030:
484e8d8bef9SDimitry Andric     return "gfx1030";
485e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1031:
486e8d8bef9SDimitry Andric     return "gfx1031";
487e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1032:
488e8d8bef9SDimitry Andric     return "gfx1032";
489e8d8bef9SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1033:
490e8d8bef9SDimitry Andric     return "gfx1033";
491fe6060f1SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1034:
492fe6060f1SDimitry Andric     return "gfx1034";
493fe6060f1SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1035:
494fe6060f1SDimitry Andric     return "gfx1035";
49581ad6265SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1036:
49681ad6265SDimitry Andric     return "gfx1036";
49781ad6265SDimitry Andric 
49881ad6265SDimitry Andric   // AMDGCN GFX11.
49981ad6265SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1100:
50081ad6265SDimitry Andric     return "gfx1100";
50181ad6265SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1101:
50281ad6265SDimitry Andric     return "gfx1101";
50381ad6265SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1102:
50481ad6265SDimitry Andric     return "gfx1102";
50581ad6265SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1103:
50681ad6265SDimitry Andric     return "gfx1103";
50706c3fb27SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1150:
50806c3fb27SDimitry Andric     return "gfx1150";
50906c3fb27SDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1151:
51006c3fb27SDimitry Andric     return "gfx1151";
5115f757f3fSDimitry Andric 
5125f757f3fSDimitry Andric   // AMDGCN GFX12.
5135f757f3fSDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1200:
5145f757f3fSDimitry Andric     return "gfx1200";
5155f757f3fSDimitry Andric   case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1201:
5165f757f3fSDimitry Andric     return "gfx1201";
517e8d8bef9SDimitry Andric   default:
518e8d8bef9SDimitry Andric     llvm_unreachable("Unknown EF_AMDGPU_MACH value");
519e8d8bef9SDimitry Andric   }
520e8d8bef9SDimitry Andric }
521e8d8bef9SDimitry Andric 
getNVPTXCPUName() const522cb14a3feSDimitry Andric StringRef ELFObjectFileBase::getNVPTXCPUName() const {
523cb14a3feSDimitry Andric   assert(getEMachine() == ELF::EM_CUDA);
524cb14a3feSDimitry Andric   unsigned SM = getPlatformFlags() & ELF::EF_CUDA_SM;
525cb14a3feSDimitry Andric 
526cb14a3feSDimitry Andric   switch (SM) {
527cb14a3feSDimitry Andric   // Fermi architecture.
528cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM20:
529cb14a3feSDimitry Andric     return "sm_20";
530cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM21:
531cb14a3feSDimitry Andric     return "sm_21";
532cb14a3feSDimitry Andric 
533cb14a3feSDimitry Andric   // Kepler architecture.
534cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM30:
535cb14a3feSDimitry Andric     return "sm_30";
536cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM32:
537cb14a3feSDimitry Andric     return "sm_32";
538cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM35:
539cb14a3feSDimitry Andric     return "sm_35";
540cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM37:
541cb14a3feSDimitry Andric     return "sm_37";
542cb14a3feSDimitry Andric 
543cb14a3feSDimitry Andric   // Maxwell architecture.
544cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM50:
545cb14a3feSDimitry Andric     return "sm_50";
546cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM52:
547cb14a3feSDimitry Andric     return "sm_52";
548cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM53:
549cb14a3feSDimitry Andric     return "sm_53";
550cb14a3feSDimitry Andric 
551cb14a3feSDimitry Andric   // Pascal architecture.
552cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM60:
553cb14a3feSDimitry Andric     return "sm_60";
554cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM61:
555cb14a3feSDimitry Andric     return "sm_61";
556cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM62:
557cb14a3feSDimitry Andric     return "sm_62";
558cb14a3feSDimitry Andric 
559cb14a3feSDimitry Andric   // Volta architecture.
560cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM70:
561cb14a3feSDimitry Andric     return "sm_70";
562cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM72:
563cb14a3feSDimitry Andric     return "sm_72";
564cb14a3feSDimitry Andric 
565cb14a3feSDimitry Andric   // Turing architecture.
566cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM75:
567cb14a3feSDimitry Andric     return "sm_75";
568cb14a3feSDimitry Andric 
569cb14a3feSDimitry Andric   // Ampere architecture.
570cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM80:
571cb14a3feSDimitry Andric     return "sm_80";
572cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM86:
573cb14a3feSDimitry Andric     return "sm_86";
574cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM87:
575cb14a3feSDimitry Andric     return "sm_87";
576cb14a3feSDimitry Andric 
577cb14a3feSDimitry Andric   // Ada architecture.
578cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM89:
579cb14a3feSDimitry Andric     return "sm_89";
580cb14a3feSDimitry Andric 
581cb14a3feSDimitry Andric   // Hopper architecture.
582cb14a3feSDimitry Andric   case ELF::EF_CUDA_SM90:
583cb14a3feSDimitry Andric     return getPlatformFlags() & ELF::EF_CUDA_ACCELERATORS ? "sm_90a" : "sm_90";
584cb14a3feSDimitry Andric   default:
585cb14a3feSDimitry Andric     llvm_unreachable("Unknown EF_CUDA_SM value");
586cb14a3feSDimitry Andric   }
587cb14a3feSDimitry Andric }
588cb14a3feSDimitry Andric 
5890b57cec5SDimitry Andric // FIXME Encode from a tablegen description or target parser.
setARMSubArch(Triple & TheTriple) const5900b57cec5SDimitry Andric void ELFObjectFileBase::setARMSubArch(Triple &TheTriple) const {
5910b57cec5SDimitry Andric   if (TheTriple.getSubArch() != Triple::NoSubArch)
5920b57cec5SDimitry Andric     return;
5930b57cec5SDimitry Andric 
5940b57cec5SDimitry Andric   ARMAttributeParser Attributes;
5955ffd83dbSDimitry Andric   if (Error E = getBuildAttributes(Attributes)) {
5965ffd83dbSDimitry Andric     // TODO Propagate Error.
5975ffd83dbSDimitry Andric     consumeError(std::move(E));
5980b57cec5SDimitry Andric     return;
5995ffd83dbSDimitry Andric   }
6000b57cec5SDimitry Andric 
6010b57cec5SDimitry Andric   std::string Triple;
6020b57cec5SDimitry Andric   // Default to ARM, but use the triple if it's been set.
6030b57cec5SDimitry Andric   if (TheTriple.isThumb())
6040b57cec5SDimitry Andric     Triple = "thumb";
6050b57cec5SDimitry Andric   else
6060b57cec5SDimitry Andric     Triple = "arm";
6070b57cec5SDimitry Andric 
608bdd1243dSDimitry Andric   std::optional<unsigned> Attr =
6095ffd83dbSDimitry Andric       Attributes.getAttributeValue(ARMBuildAttrs::CPU_arch);
61081ad6265SDimitry Andric   if (Attr) {
611bdd1243dSDimitry Andric     switch (*Attr) {
6120b57cec5SDimitry Andric     case ARMBuildAttrs::v4:
6130b57cec5SDimitry Andric       Triple += "v4";
6140b57cec5SDimitry Andric       break;
6150b57cec5SDimitry Andric     case ARMBuildAttrs::v4T:
6160b57cec5SDimitry Andric       Triple += "v4t";
6170b57cec5SDimitry Andric       break;
6180b57cec5SDimitry Andric     case ARMBuildAttrs::v5T:
6190b57cec5SDimitry Andric       Triple += "v5t";
6200b57cec5SDimitry Andric       break;
6210b57cec5SDimitry Andric     case ARMBuildAttrs::v5TE:
6220b57cec5SDimitry Andric       Triple += "v5te";
6230b57cec5SDimitry Andric       break;
6240b57cec5SDimitry Andric     case ARMBuildAttrs::v5TEJ:
6250b57cec5SDimitry Andric       Triple += "v5tej";
6260b57cec5SDimitry Andric       break;
6270b57cec5SDimitry Andric     case ARMBuildAttrs::v6:
6280b57cec5SDimitry Andric       Triple += "v6";
6290b57cec5SDimitry Andric       break;
6300b57cec5SDimitry Andric     case ARMBuildAttrs::v6KZ:
6310b57cec5SDimitry Andric       Triple += "v6kz";
6320b57cec5SDimitry Andric       break;
6330b57cec5SDimitry Andric     case ARMBuildAttrs::v6T2:
6340b57cec5SDimitry Andric       Triple += "v6t2";
6350b57cec5SDimitry Andric       break;
6360b57cec5SDimitry Andric     case ARMBuildAttrs::v6K:
6370b57cec5SDimitry Andric       Triple += "v6k";
6380b57cec5SDimitry Andric       break;
639349cc55cSDimitry Andric     case ARMBuildAttrs::v7: {
640bdd1243dSDimitry Andric       std::optional<unsigned> ArchProfileAttr =
641349cc55cSDimitry Andric           Attributes.getAttributeValue(ARMBuildAttrs::CPU_arch_profile);
64281ad6265SDimitry Andric       if (ArchProfileAttr &&
643bdd1243dSDimitry Andric           *ArchProfileAttr == ARMBuildAttrs::MicroControllerProfile)
644349cc55cSDimitry Andric         Triple += "v7m";
645349cc55cSDimitry Andric       else
6460b57cec5SDimitry Andric         Triple += "v7";
6470b57cec5SDimitry Andric       break;
648349cc55cSDimitry Andric     }
6490b57cec5SDimitry Andric     case ARMBuildAttrs::v6_M:
6500b57cec5SDimitry Andric       Triple += "v6m";
6510b57cec5SDimitry Andric       break;
6520b57cec5SDimitry Andric     case ARMBuildAttrs::v6S_M:
6530b57cec5SDimitry Andric       Triple += "v6sm";
6540b57cec5SDimitry Andric       break;
6550b57cec5SDimitry Andric     case ARMBuildAttrs::v7E_M:
6560b57cec5SDimitry Andric       Triple += "v7em";
6570b57cec5SDimitry Andric       break;
6588bcb0991SDimitry Andric     case ARMBuildAttrs::v8_A:
6598bcb0991SDimitry Andric       Triple += "v8a";
6608bcb0991SDimitry Andric       break;
6618bcb0991SDimitry Andric     case ARMBuildAttrs::v8_R:
6628bcb0991SDimitry Andric       Triple += "v8r";
6638bcb0991SDimitry Andric       break;
6648bcb0991SDimitry Andric     case ARMBuildAttrs::v8_M_Base:
6658bcb0991SDimitry Andric       Triple += "v8m.base";
6668bcb0991SDimitry Andric       break;
6678bcb0991SDimitry Andric     case ARMBuildAttrs::v8_M_Main:
6688bcb0991SDimitry Andric       Triple += "v8m.main";
6698bcb0991SDimitry Andric       break;
6708bcb0991SDimitry Andric     case ARMBuildAttrs::v8_1_M_Main:
6718bcb0991SDimitry Andric       Triple += "v8.1m.main";
6728bcb0991SDimitry Andric       break;
67381ad6265SDimitry Andric     case ARMBuildAttrs::v9_A:
67481ad6265SDimitry Andric       Triple += "v9a";
67581ad6265SDimitry Andric       break;
6760b57cec5SDimitry Andric     }
6770b57cec5SDimitry Andric   }
6780b57cec5SDimitry Andric   if (!isLittleEndian())
6790b57cec5SDimitry Andric     Triple += "eb";
6800b57cec5SDimitry Andric 
6810b57cec5SDimitry Andric   TheTriple.setArchName(Triple);
6820b57cec5SDimitry Andric }
6830b57cec5SDimitry Andric 
getPltEntries() const68406c3fb27SDimitry Andric std::vector<ELFPltEntry> ELFObjectFileBase::getPltEntries() const {
6850b57cec5SDimitry Andric   std::string Err;
6860b57cec5SDimitry Andric   const auto Triple = makeTriple();
6870b57cec5SDimitry Andric   const auto *T = TargetRegistry::lookupTarget(Triple.str(), Err);
6880b57cec5SDimitry Andric   if (!T)
6890b57cec5SDimitry Andric     return {};
69006c3fb27SDimitry Andric   uint32_t JumpSlotReloc = 0, GlobDatReloc = 0;
6910b57cec5SDimitry Andric   switch (Triple.getArch()) {
6920b57cec5SDimitry Andric     case Triple::x86:
6930b57cec5SDimitry Andric       JumpSlotReloc = ELF::R_386_JUMP_SLOT;
69406c3fb27SDimitry Andric       GlobDatReloc = ELF::R_386_GLOB_DAT;
6950b57cec5SDimitry Andric       break;
6960b57cec5SDimitry Andric     case Triple::x86_64:
6970b57cec5SDimitry Andric       JumpSlotReloc = ELF::R_X86_64_JUMP_SLOT;
69806c3fb27SDimitry Andric       GlobDatReloc = ELF::R_X86_64_GLOB_DAT;
6990b57cec5SDimitry Andric       break;
7000b57cec5SDimitry Andric     case Triple::aarch64:
701fe6060f1SDimitry Andric     case Triple::aarch64_be:
7020b57cec5SDimitry Andric       JumpSlotReloc = ELF::R_AARCH64_JUMP_SLOT;
7030b57cec5SDimitry Andric       break;
7040b57cec5SDimitry Andric     default:
7050b57cec5SDimitry Andric       return {};
7060b57cec5SDimitry Andric   }
7070b57cec5SDimitry Andric   std::unique_ptr<const MCInstrInfo> MII(T->createMCInstrInfo());
7080b57cec5SDimitry Andric   std::unique_ptr<const MCInstrAnalysis> MIA(
7090b57cec5SDimitry Andric       T->createMCInstrAnalysis(MII.get()));
7100b57cec5SDimitry Andric   if (!MIA)
7110b57cec5SDimitry Andric     return {};
71206c3fb27SDimitry Andric   std::vector<std::pair<uint64_t, uint64_t>> PltEntries;
71306c3fb27SDimitry Andric   std::optional<SectionRef> RelaPlt, RelaDyn;
71406c3fb27SDimitry Andric   uint64_t GotBaseVA = 0;
7150b57cec5SDimitry Andric   for (const SectionRef &Section : sections()) {
7168bcb0991SDimitry Andric     Expected<StringRef> NameOrErr = Section.getName();
7178bcb0991SDimitry Andric     if (!NameOrErr) {
7188bcb0991SDimitry Andric       consumeError(NameOrErr.takeError());
7190b57cec5SDimitry Andric       continue;
7208bcb0991SDimitry Andric     }
7218bcb0991SDimitry Andric     StringRef Name = *NameOrErr;
7228bcb0991SDimitry Andric 
72306c3fb27SDimitry Andric     if (Name == ".rela.plt" || Name == ".rel.plt") {
7240b57cec5SDimitry Andric       RelaPlt = Section;
72506c3fb27SDimitry Andric     } else if (Name == ".rela.dyn" || Name == ".rel.dyn") {
72606c3fb27SDimitry Andric       RelaDyn = Section;
72706c3fb27SDimitry Andric     } else if (Name == ".got.plt") {
72806c3fb27SDimitry Andric       GotBaseVA = Section.getAddress();
72906c3fb27SDimitry Andric     } else if (Name == ".plt" || Name == ".plt.got") {
73006c3fb27SDimitry Andric       Expected<StringRef> PltContents = Section.getContents();
7310b57cec5SDimitry Andric       if (!PltContents) {
7320b57cec5SDimitry Andric         consumeError(PltContents.takeError());
7330b57cec5SDimitry Andric         return {};
7340b57cec5SDimitry Andric       }
73506c3fb27SDimitry Andric       llvm::append_range(
73606c3fb27SDimitry Andric           PltEntries,
73706c3fb27SDimitry Andric           MIA->findPltEntries(Section.getAddress(),
73806c3fb27SDimitry Andric                               arrayRefFromStringRef(*PltContents), Triple));
73906c3fb27SDimitry Andric     }
74006c3fb27SDimitry Andric   }
74106c3fb27SDimitry Andric 
7420b57cec5SDimitry Andric   // Build a map from GOT entry virtual address to PLT entry virtual address.
7430b57cec5SDimitry Andric   DenseMap<uint64_t, uint64_t> GotToPlt;
74406c3fb27SDimitry Andric   for (auto [Plt, GotPlt] : PltEntries) {
74506c3fb27SDimitry Andric     uint64_t GotPltEntry = GotPlt;
74606c3fb27SDimitry Andric     // An x86-32 PIC PLT uses jmp DWORD PTR [ebx-offset]. Add
74706c3fb27SDimitry Andric     // _GLOBAL_OFFSET_TABLE_ (EBX) to get the .got.plt (or .got) entry address.
74806c3fb27SDimitry Andric     // See X86MCTargetDesc.cpp:findPltEntries for the 1 << 32 bit.
74906c3fb27SDimitry Andric     if (GotPltEntry & (uint64_t(1) << 32) && getEMachine() == ELF::EM_386)
75006c3fb27SDimitry Andric       GotPltEntry = static_cast<int32_t>(GotPltEntry) + GotBaseVA;
75106c3fb27SDimitry Andric     GotToPlt.insert(std::make_pair(GotPltEntry, Plt));
75206c3fb27SDimitry Andric   }
75306c3fb27SDimitry Andric 
7540b57cec5SDimitry Andric   // Find the relocations in the dynamic relocation table that point to
7550b57cec5SDimitry Andric   // locations in the GOT for which we know the corresponding PLT entry.
75606c3fb27SDimitry Andric   std::vector<ELFPltEntry> Result;
75706c3fb27SDimitry Andric   auto handleRels = [&](iterator_range<relocation_iterator> Rels,
75806c3fb27SDimitry Andric                         uint32_t RelType, StringRef PltSec) {
75906c3fb27SDimitry Andric     for (const auto &R : Rels) {
76006c3fb27SDimitry Andric       if (R.getType() != RelType)
7610b57cec5SDimitry Andric         continue;
76206c3fb27SDimitry Andric       auto PltEntryIter = GotToPlt.find(R.getOffset());
763e8d8bef9SDimitry Andric       if (PltEntryIter != GotToPlt.end()) {
76406c3fb27SDimitry Andric         symbol_iterator Sym = R.getSymbol();
765e8d8bef9SDimitry Andric         if (Sym == symbol_end())
76606c3fb27SDimitry Andric           Result.push_back(
76706c3fb27SDimitry Andric               ELFPltEntry{PltSec, std::nullopt, PltEntryIter->second});
768e8d8bef9SDimitry Andric         else
76906c3fb27SDimitry Andric           Result.push_back(ELFPltEntry{PltSec, Sym->getRawDataRefImpl(),
77006c3fb27SDimitry Andric                                        PltEntryIter->second});
771e8d8bef9SDimitry Andric       }
7720b57cec5SDimitry Andric     }
77306c3fb27SDimitry Andric   };
77406c3fb27SDimitry Andric 
77506c3fb27SDimitry Andric   if (RelaPlt)
77606c3fb27SDimitry Andric     handleRels(RelaPlt->relocations(), JumpSlotReloc, ".plt");
77706c3fb27SDimitry Andric 
77806c3fb27SDimitry Andric   // If a symbol needing a PLT entry also needs a GLOB_DAT relocation, GNU ld's
77906c3fb27SDimitry Andric   // x86 port places the PLT entry in the .plt.got section.
78006c3fb27SDimitry Andric   if (RelaDyn)
78106c3fb27SDimitry Andric     handleRels(RelaDyn->relocations(), GlobDatReloc, ".plt.got");
78206c3fb27SDimitry Andric 
7830b57cec5SDimitry Andric   return Result;
7840b57cec5SDimitry Andric }
785349cc55cSDimitry Andric 
786349cc55cSDimitry Andric template <class ELFT>
readBBAddrMapImpl(const ELFFile<ELFT> & EF,std::optional<unsigned> TextSectionIndex,std::vector<PGOAnalysisMap> * PGOAnalyses)787bdd1243dSDimitry Andric Expected<std::vector<BBAddrMap>> static readBBAddrMapImpl(
7885f757f3fSDimitry Andric     const ELFFile<ELFT> &EF, std::optional<unsigned> TextSectionIndex,
7895f757f3fSDimitry Andric     std::vector<PGOAnalysisMap> *PGOAnalyses) {
79081ad6265SDimitry Andric   using Elf_Shdr = typename ELFT::Shdr;
79106c3fb27SDimitry Andric   bool IsRelocatable = EF.getHeader().e_type == ELF::ET_REL;
79281ad6265SDimitry Andric   std::vector<BBAddrMap> BBAddrMaps;
7935f757f3fSDimitry Andric   if (PGOAnalyses)
7945f757f3fSDimitry Andric     PGOAnalyses->clear();
79506c3fb27SDimitry Andric 
79681ad6265SDimitry Andric   const auto &Sections = cantFail(EF.sections());
79706c3fb27SDimitry Andric   auto IsMatch = [&](const Elf_Shdr &Sec) -> Expected<bool> {
79881ad6265SDimitry Andric     if (Sec.sh_type != ELF::SHT_LLVM_BB_ADDR_MAP &&
79981ad6265SDimitry Andric         Sec.sh_type != ELF::SHT_LLVM_BB_ADDR_MAP_V0)
80006c3fb27SDimitry Andric       return false;
80106c3fb27SDimitry Andric     if (!TextSectionIndex)
80206c3fb27SDimitry Andric       return true;
80381ad6265SDimitry Andric     Expected<const Elf_Shdr *> TextSecOrErr = EF.getSection(Sec.sh_link);
80481ad6265SDimitry Andric     if (!TextSecOrErr)
80581ad6265SDimitry Andric       return createError("unable to get the linked-to section for " +
80681ad6265SDimitry Andric                          describe(EF, Sec) + ": " +
80781ad6265SDimitry Andric                          toString(TextSecOrErr.takeError()));
80881ad6265SDimitry Andric     if (*TextSectionIndex != std::distance(Sections.begin(), *TextSecOrErr))
80906c3fb27SDimitry Andric       return false;
81006c3fb27SDimitry Andric     return true;
81106c3fb27SDimitry Andric   };
81206c3fb27SDimitry Andric 
81306c3fb27SDimitry Andric   Expected<MapVector<const Elf_Shdr *, const Elf_Shdr *>> SectionRelocMapOrErr =
81406c3fb27SDimitry Andric       EF.getSectionAndRelocations(IsMatch);
81506c3fb27SDimitry Andric   if (!SectionRelocMapOrErr)
81606c3fb27SDimitry Andric     return SectionRelocMapOrErr.takeError();
81706c3fb27SDimitry Andric 
81806c3fb27SDimitry Andric   for (auto const &[Sec, RelocSec] : *SectionRelocMapOrErr) {
81906c3fb27SDimitry Andric     if (IsRelocatable && !RelocSec)
82006c3fb27SDimitry Andric       return createError("unable to get relocation section for " +
82106c3fb27SDimitry Andric                          describe(EF, *Sec));
82206c3fb27SDimitry Andric     Expected<std::vector<BBAddrMap>> BBAddrMapOrErr =
8235f757f3fSDimitry Andric         EF.decodeBBAddrMap(*Sec, RelocSec, PGOAnalyses);
8245f757f3fSDimitry Andric     if (!BBAddrMapOrErr) {
8255f757f3fSDimitry Andric       if (PGOAnalyses)
8265f757f3fSDimitry Andric         PGOAnalyses->clear();
82706c3fb27SDimitry Andric       return createError("unable to read " + describe(EF, *Sec) + ": " +
82881ad6265SDimitry Andric                          toString(BBAddrMapOrErr.takeError()));
8295f757f3fSDimitry Andric     }
83081ad6265SDimitry Andric     std::move(BBAddrMapOrErr->begin(), BBAddrMapOrErr->end(),
83181ad6265SDimitry Andric               std::back_inserter(BBAddrMaps));
83281ad6265SDimitry Andric   }
8337a6dacacSDimitry Andric   if (PGOAnalyses)
8347a6dacacSDimitry Andric     assert(PGOAnalyses->size() == BBAddrMaps.size() &&
8357a6dacacSDimitry Andric            "The same number of BBAddrMaps and PGOAnalysisMaps should be "
8367a6dacacSDimitry Andric            "returned when PGO information is requested");
83781ad6265SDimitry Andric   return BBAddrMaps;
83881ad6265SDimitry Andric }
83981ad6265SDimitry Andric 
84081ad6265SDimitry Andric template <class ELFT>
841349cc55cSDimitry Andric static Expected<std::vector<VersionEntry>>
readDynsymVersionsImpl(const ELFFile<ELFT> & EF,ELFObjectFileBase::elf_symbol_iterator_range Symbols)842349cc55cSDimitry Andric readDynsymVersionsImpl(const ELFFile<ELFT> &EF,
843349cc55cSDimitry Andric                        ELFObjectFileBase::elf_symbol_iterator_range Symbols) {
844349cc55cSDimitry Andric   using Elf_Shdr = typename ELFT::Shdr;
845349cc55cSDimitry Andric   const Elf_Shdr *VerSec = nullptr;
846349cc55cSDimitry Andric   const Elf_Shdr *VerNeedSec = nullptr;
847349cc55cSDimitry Andric   const Elf_Shdr *VerDefSec = nullptr;
848349cc55cSDimitry Andric   // The user should ensure sections() can't fail here.
849349cc55cSDimitry Andric   for (const Elf_Shdr &Sec : cantFail(EF.sections())) {
850349cc55cSDimitry Andric     if (Sec.sh_type == ELF::SHT_GNU_versym)
851349cc55cSDimitry Andric       VerSec = &Sec;
852349cc55cSDimitry Andric     else if (Sec.sh_type == ELF::SHT_GNU_verdef)
853349cc55cSDimitry Andric       VerDefSec = &Sec;
854349cc55cSDimitry Andric     else if (Sec.sh_type == ELF::SHT_GNU_verneed)
855349cc55cSDimitry Andric       VerNeedSec = &Sec;
856349cc55cSDimitry Andric   }
857349cc55cSDimitry Andric   if (!VerSec)
858349cc55cSDimitry Andric     return std::vector<VersionEntry>();
859349cc55cSDimitry Andric 
860bdd1243dSDimitry Andric   Expected<SmallVector<std::optional<VersionEntry>, 0>> MapOrErr =
861349cc55cSDimitry Andric       EF.loadVersionMap(VerNeedSec, VerDefSec);
862349cc55cSDimitry Andric   if (!MapOrErr)
863349cc55cSDimitry Andric     return MapOrErr.takeError();
864349cc55cSDimitry Andric 
865349cc55cSDimitry Andric   std::vector<VersionEntry> Ret;
866349cc55cSDimitry Andric   size_t I = 0;
8674824e7fdSDimitry Andric   for (const ELFSymbolRef &Sym : Symbols) {
868349cc55cSDimitry Andric     ++I;
869349cc55cSDimitry Andric     Expected<const typename ELFT::Versym *> VerEntryOrErr =
870349cc55cSDimitry Andric         EF.template getEntry<typename ELFT::Versym>(*VerSec, I);
871349cc55cSDimitry Andric     if (!VerEntryOrErr)
872349cc55cSDimitry Andric       return createError("unable to read an entry with index " + Twine(I) +
873349cc55cSDimitry Andric                          " from " + describe(EF, *VerSec) + ": " +
874349cc55cSDimitry Andric                          toString(VerEntryOrErr.takeError()));
875349cc55cSDimitry Andric 
8764824e7fdSDimitry Andric     Expected<uint32_t> FlagsOrErr = Sym.getFlags();
877349cc55cSDimitry Andric     if (!FlagsOrErr)
878349cc55cSDimitry Andric       return createError("unable to read flags for symbol with index " +
879349cc55cSDimitry Andric                          Twine(I) + ": " + toString(FlagsOrErr.takeError()));
880349cc55cSDimitry Andric 
881349cc55cSDimitry Andric     bool IsDefault;
882349cc55cSDimitry Andric     Expected<StringRef> VerOrErr = EF.getSymbolVersionByIndex(
883349cc55cSDimitry Andric         (*VerEntryOrErr)->vs_index, IsDefault, *MapOrErr,
884349cc55cSDimitry Andric         (*FlagsOrErr) & SymbolRef::SF_Undefined);
885349cc55cSDimitry Andric     if (!VerOrErr)
886349cc55cSDimitry Andric       return createError("unable to get a version for entry " + Twine(I) +
887349cc55cSDimitry Andric                          " of " + describe(EF, *VerSec) + ": " +
888349cc55cSDimitry Andric                          toString(VerOrErr.takeError()));
889349cc55cSDimitry Andric 
890349cc55cSDimitry Andric     Ret.push_back({(*VerOrErr).str(), IsDefault});
891349cc55cSDimitry Andric   }
892349cc55cSDimitry Andric 
893349cc55cSDimitry Andric   return Ret;
894349cc55cSDimitry Andric }
895349cc55cSDimitry Andric 
896349cc55cSDimitry Andric Expected<std::vector<VersionEntry>>
readDynsymVersions() const897349cc55cSDimitry Andric ELFObjectFileBase::readDynsymVersions() const {
898349cc55cSDimitry Andric   elf_symbol_iterator_range Symbols = getDynamicSymbolIterators();
899349cc55cSDimitry Andric   if (const auto *Obj = dyn_cast<ELF32LEObjectFile>(this))
900349cc55cSDimitry Andric     return readDynsymVersionsImpl(Obj->getELFFile(), Symbols);
901349cc55cSDimitry Andric   if (const auto *Obj = dyn_cast<ELF32BEObjectFile>(this))
902349cc55cSDimitry Andric     return readDynsymVersionsImpl(Obj->getELFFile(), Symbols);
903349cc55cSDimitry Andric   if (const auto *Obj = dyn_cast<ELF64LEObjectFile>(this))
904349cc55cSDimitry Andric     return readDynsymVersionsImpl(Obj->getELFFile(), Symbols);
905349cc55cSDimitry Andric   return readDynsymVersionsImpl(cast<ELF64BEObjectFile>(this)->getELFFile(),
906349cc55cSDimitry Andric                                 Symbols);
907349cc55cSDimitry Andric }
90881ad6265SDimitry Andric 
readBBAddrMap(std::optional<unsigned> TextSectionIndex,std::vector<PGOAnalysisMap> * PGOAnalyses) const909bdd1243dSDimitry Andric Expected<std::vector<BBAddrMap>> ELFObjectFileBase::readBBAddrMap(
9105f757f3fSDimitry Andric     std::optional<unsigned> TextSectionIndex,
9115f757f3fSDimitry Andric     std::vector<PGOAnalysisMap> *PGOAnalyses) const {
91281ad6265SDimitry Andric   if (const auto *Obj = dyn_cast<ELF32LEObjectFile>(this))
9135f757f3fSDimitry Andric     return readBBAddrMapImpl(Obj->getELFFile(), TextSectionIndex, PGOAnalyses);
91481ad6265SDimitry Andric   if (const auto *Obj = dyn_cast<ELF64LEObjectFile>(this))
9155f757f3fSDimitry Andric     return readBBAddrMapImpl(Obj->getELFFile(), TextSectionIndex, PGOAnalyses);
91681ad6265SDimitry Andric   if (const auto *Obj = dyn_cast<ELF32BEObjectFile>(this))
9175f757f3fSDimitry Andric     return readBBAddrMapImpl(Obj->getELFFile(), TextSectionIndex, PGOAnalyses);
91806c3fb27SDimitry Andric   return readBBAddrMapImpl(cast<ELF64BEObjectFile>(this)->getELFFile(),
9195f757f3fSDimitry Andric                            TextSectionIndex, PGOAnalyses);
92081ad6265SDimitry Andric }
921