xref: /llvm-project-15.0.7/llvm/lib/Object/ELF.cpp (revision 1d2ce4da)
11df42facSEugene Zelenko //===- ELF.cpp - ELF object file implementation ---------------------------===//
2126973baSMichael J. Spencer //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6126973baSMichael J. Spencer //
7126973baSMichael J. Spencer //===----------------------------------------------------------------------===//
8126973baSMichael J. Spencer 
9126973baSMichael J. Spencer #include "llvm/Object/ELF.h"
10264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELF.h"
113ca502a7SRahman Lavaee #include "llvm/Support/DataExtractor.h"
12126973baSMichael J. Spencer 
131df42facSEugene Zelenko using namespace llvm;
141df42facSEugene Zelenko using namespace object;
15126973baSMichael J. Spencer 
163ba25737SRafael Espindola #define STRINGIFY_ENUM_CASE(ns, name)                                          \
173ba25737SRafael Espindola   case ns::name:                                                               \
183ba25737SRafael Espindola     return #name;
193ba25737SRafael Espindola 
203ba25737SRafael Espindola #define ELF_RELOC(name, value) STRINGIFY_ENUM_CASE(ELF, name)
21126973baSMichael J. Spencer 
getELFRelocationTypeName(uint32_t Machine,uint32_t Type)221df42facSEugene Zelenko StringRef llvm::object::getELFRelocationTypeName(uint32_t Machine,
231df42facSEugene Zelenko                                                  uint32_t Type) {
24126973baSMichael J. Spencer   switch (Machine) {
258dddc152SMin-Yih Hsu   case ELF::EM_68K:
268dddc152SMin-Yih Hsu     switch (Type) {
278dddc152SMin-Yih Hsu #include "llvm/BinaryFormat/ELFRelocs/M68k.def"
288dddc152SMin-Yih Hsu     default:
298dddc152SMin-Yih Hsu       break;
308dddc152SMin-Yih Hsu     }
318dddc152SMin-Yih Hsu     break;
32126973baSMichael J. Spencer   case ELF::EM_X86_64:
33126973baSMichael J. Spencer     switch (Type) {
34264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/x86_64.def"
35126973baSMichael J. Spencer     default:
36126973baSMichael J. Spencer       break;
37126973baSMichael J. Spencer     }
38126973baSMichael J. Spencer     break;
39126973baSMichael J. Spencer   case ELF::EM_386:
40a3b79dd7SMichael Kuperstein   case ELF::EM_IAMCU:
41126973baSMichael J. Spencer     switch (Type) {
42264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/i386.def"
43126973baSMichael J. Spencer     default:
44126973baSMichael J. Spencer       break;
45126973baSMichael J. Spencer     }
46126973baSMichael J. Spencer     break;
47126973baSMichael J. Spencer   case ELF::EM_MIPS:
48126973baSMichael J. Spencer     switch (Type) {
49264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/Mips.def"
50126973baSMichael J. Spencer     default:
51126973baSMichael J. Spencer       break;
52126973baSMichael J. Spencer     }
53126973baSMichael J. Spencer     break;
54126973baSMichael J. Spencer   case ELF::EM_AARCH64:
55126973baSMichael J. Spencer     switch (Type) {
56264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/AArch64.def"
57126973baSMichael J. Spencer     default:
58126973baSMichael J. Spencer       break;
59126973baSMichael J. Spencer     }
60126973baSMichael J. Spencer     break;
61126973baSMichael J. Spencer   case ELF::EM_ARM:
62126973baSMichael J. Spencer     switch (Type) {
63264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/ARM.def"
64126973baSMichael J. Spencer     default:
65126973baSMichael J. Spencer       break;
66126973baSMichael J. Spencer     }
67126973baSMichael J. Spencer     break;
6849277d1fSLeslie Zhai   case ELF::EM_ARC_COMPACT:
6949277d1fSLeslie Zhai   case ELF::EM_ARC_COMPACT2:
7049277d1fSLeslie Zhai     switch (Type) {
7149277d1fSLeslie Zhai #include "llvm/BinaryFormat/ELFRelocs/ARC.def"
7249277d1fSLeslie Zhai     default:
7349277d1fSLeslie Zhai       break;
7449277d1fSLeslie Zhai     }
7549277d1fSLeslie Zhai     break;
7635047ed7SDylan McKay   case ELF::EM_AVR:
7735047ed7SDylan McKay     switch (Type) {
78264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/AVR.def"
7935047ed7SDylan McKay     default:
8035047ed7SDylan McKay       break;
8135047ed7SDylan McKay     }
8235047ed7SDylan McKay     break;
83126973baSMichael J. Spencer   case ELF::EM_HEXAGON:
84126973baSMichael J. Spencer     switch (Type) {
85264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/Hexagon.def"
86126973baSMichael J. Spencer     default:
87126973baSMichael J. Spencer       break;
88126973baSMichael J. Spencer     }
89126973baSMichael J. Spencer     break;
90ea9f25a7SJacques Pienaar   case ELF::EM_LANAI:
91ea9f25a7SJacques Pienaar     switch (Type) {
92264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/Lanai.def"
93ea9f25a7SJacques Pienaar     default:
94ea9f25a7SJacques Pienaar       break;
95ea9f25a7SJacques Pienaar     }
96ea9f25a7SJacques Pienaar     break;
97126973baSMichael J. Spencer   case ELF::EM_PPC:
98126973baSMichael J. Spencer     switch (Type) {
99264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/PowerPC.def"
100126973baSMichael J. Spencer     default:
101126973baSMichael J. Spencer       break;
102126973baSMichael J. Spencer     }
103126973baSMichael J. Spencer     break;
104126973baSMichael J. Spencer   case ELF::EM_PPC64:
105126973baSMichael J. Spencer     switch (Type) {
106264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/PowerPC64.def"
107126973baSMichael J. Spencer     default:
108126973baSMichael J. Spencer       break;
109126973baSMichael J. Spencer     }
110126973baSMichael J. Spencer     break;
1111524f62bSAlex Bradbury   case ELF::EM_RISCV:
1121524f62bSAlex Bradbury     switch (Type) {
113264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/RISCV.def"
1141524f62bSAlex Bradbury     default:
1151524f62bSAlex Bradbury       break;
1161524f62bSAlex Bradbury     }
1171524f62bSAlex Bradbury     break;
118126973baSMichael J. Spencer   case ELF::EM_S390:
119126973baSMichael J. Spencer     switch (Type) {
120264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/SystemZ.def"
121126973baSMichael J. Spencer     default:
122126973baSMichael J. Spencer       break;
123126973baSMichael J. Spencer     }
124126973baSMichael J. Spencer     break;
125cdee0edfSVenkatraman Govindaraju   case ELF::EM_SPARC:
126cdee0edfSVenkatraman Govindaraju   case ELF::EM_SPARC32PLUS:
127cdee0edfSVenkatraman Govindaraju   case ELF::EM_SPARCV9:
128cdee0edfSVenkatraman Govindaraju     switch (Type) {
129264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/Sparc.def"
130cdee0edfSVenkatraman Govindaraju     default:
131cdee0edfSVenkatraman Govindaraju       break;
132cdee0edfSVenkatraman Govindaraju     }
133cdee0edfSVenkatraman Govindaraju     break;
134f8db61c5STom Stellard   case ELF::EM_AMDGPU:
135f8db61c5STom Stellard     switch (Type) {
136264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/AMDGPU.def"
137f8db61c5STom Stellard     default:
138f8db61c5STom Stellard       break;
139f8db61c5STom Stellard     }
1400e6694d1SAdrian Prantl     break;
141cfb51f54SAlexei Starovoitov   case ELF::EM_BPF:
142cfb51f54SAlexei Starovoitov     switch (Type) {
143264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELFRelocs/BPF.def"
144cfb51f54SAlexei Starovoitov     default:
145cfb51f54SAlexei Starovoitov       break;
146cfb51f54SAlexei Starovoitov     }
147f8db61c5STom Stellard     break;
14849045c6aSAnton Korobeynikov   case ELF::EM_MSP430:
14949045c6aSAnton Korobeynikov     switch (Type) {
15049045c6aSAnton Korobeynikov #include "llvm/BinaryFormat/ELFRelocs/MSP430.def"
15149045c6aSAnton Korobeynikov     default:
15249045c6aSAnton Korobeynikov       break;
15349045c6aSAnton Korobeynikov     }
15449045c6aSAnton Korobeynikov     break;
1555921782fSKazushi (Jam) Marukawa   case ELF::EM_VE:
1565921782fSKazushi (Jam) Marukawa     switch (Type) {
1575921782fSKazushi (Jam) Marukawa #include "llvm/BinaryFormat/ELFRelocs/VE.def"
1585921782fSKazushi (Jam) Marukawa     default:
1595921782fSKazushi (Jam) Marukawa       break;
1605921782fSKazushi (Jam) Marukawa     }
1615921782fSKazushi (Jam) Marukawa     break;
16269f2c79fSZi Xuan Wu   case ELF::EM_CSKY:
16369f2c79fSZi Xuan Wu     switch (Type) {
16469f2c79fSZi Xuan Wu #include "llvm/BinaryFormat/ELFRelocs/CSKY.def"
16569f2c79fSZi Xuan Wu     default:
16669f2c79fSZi Xuan Wu       break;
16769f2c79fSZi Xuan Wu     }
16869f2c79fSZi Xuan Wu     break;
169e53e6ec6SLu Weining   case ELF::EM_LOONGARCH:
170e53e6ec6SLu Weining     switch (Type) {
171e53e6ec6SLu Weining #include "llvm/BinaryFormat/ELFRelocs/LoongArch.def"
172e53e6ec6SLu Weining     default:
173e53e6ec6SLu Weining       break;
174e53e6ec6SLu Weining     }
175e53e6ec6SLu Weining     break;
176126973baSMichael J. Spencer   default:
177126973baSMichael J. Spencer     break;
178126973baSMichael J. Spencer   }
179126973baSMichael J. Spencer   return "Unknown";
180126973baSMichael J. Spencer }
181126973baSMichael J. Spencer 
182242785c0STim Northover #undef ELF_RELOC
1833ba25737SRafael Espindola 
getELFRelativeRelocationType(uint32_t Machine)184d2ed5be8SFangrui Song uint32_t llvm::object::getELFRelativeRelocationType(uint32_t Machine) {
1850f440d83SJake Ehrlich   switch (Machine) {
1860f440d83SJake Ehrlich   case ELF::EM_X86_64:
1870f440d83SJake Ehrlich     return ELF::R_X86_64_RELATIVE;
1880f440d83SJake Ehrlich   case ELF::EM_386:
1890f440d83SJake Ehrlich   case ELF::EM_IAMCU:
1900f440d83SJake Ehrlich     return ELF::R_386_RELATIVE;
1910f440d83SJake Ehrlich   case ELF::EM_MIPS:
1920f440d83SJake Ehrlich     break;
1930f440d83SJake Ehrlich   case ELF::EM_AARCH64:
1940f440d83SJake Ehrlich     return ELF::R_AARCH64_RELATIVE;
1950f440d83SJake Ehrlich   case ELF::EM_ARM:
1960f440d83SJake Ehrlich     return ELF::R_ARM_RELATIVE;
1970f440d83SJake Ehrlich   case ELF::EM_ARC_COMPACT:
1980f440d83SJake Ehrlich   case ELF::EM_ARC_COMPACT2:
1990f440d83SJake Ehrlich     return ELF::R_ARC_RELATIVE;
2000f440d83SJake Ehrlich   case ELF::EM_AVR:
2010f440d83SJake Ehrlich     break;
2020f440d83SJake Ehrlich   case ELF::EM_HEXAGON:
2030f440d83SJake Ehrlich     return ELF::R_HEX_RELATIVE;
2040f440d83SJake Ehrlich   case ELF::EM_LANAI:
2050f440d83SJake Ehrlich     break;
2060f440d83SJake Ehrlich   case ELF::EM_PPC:
2070f440d83SJake Ehrlich     break;
2080f440d83SJake Ehrlich   case ELF::EM_PPC64:
2090f440d83SJake Ehrlich     return ELF::R_PPC64_RELATIVE;
2100f440d83SJake Ehrlich   case ELF::EM_RISCV:
2110f440d83SJake Ehrlich     return ELF::R_RISCV_RELATIVE;
2120f440d83SJake Ehrlich   case ELF::EM_S390:
2130f440d83SJake Ehrlich     return ELF::R_390_RELATIVE;
2140f440d83SJake Ehrlich   case ELF::EM_SPARC:
2150f440d83SJake Ehrlich   case ELF::EM_SPARC32PLUS:
2160f440d83SJake Ehrlich   case ELF::EM_SPARCV9:
2170f440d83SJake Ehrlich     return ELF::R_SPARC_RELATIVE;
21869f2c79fSZi Xuan Wu   case ELF::EM_CSKY:
21969f2c79fSZi Xuan Wu     return ELF::R_CKCORE_RELATIVE;
220d1057f96SKazushi (Jam) Marukawa   case ELF::EM_VE:
221d1057f96SKazushi (Jam) Marukawa     return ELF::R_VE_RELATIVE;
2220f440d83SJake Ehrlich   case ELF::EM_AMDGPU:
2230f440d83SJake Ehrlich     break;
2240f440d83SJake Ehrlich   case ELF::EM_BPF:
2250f440d83SJake Ehrlich     break;
2260f440d83SJake Ehrlich   default:
2270f440d83SJake Ehrlich     break;
2280f440d83SJake Ehrlich   }
2290f440d83SJake Ehrlich   return 0;
2300f440d83SJake Ehrlich }
2310f440d83SJake Ehrlich 
getELFSectionTypeName(uint32_t Machine,unsigned Type)2323ba25737SRafael Espindola StringRef llvm::object::getELFSectionTypeName(uint32_t Machine, unsigned Type) {
2333ba25737SRafael Espindola   switch (Machine) {
2343ba25737SRafael Espindola   case ELF::EM_ARM:
2353ba25737SRafael Espindola     switch (Type) {
2363ba25737SRafael Espindola       STRINGIFY_ENUM_CASE(ELF, SHT_ARM_EXIDX);
2373ba25737SRafael Espindola       STRINGIFY_ENUM_CASE(ELF, SHT_ARM_PREEMPTMAP);
2383ba25737SRafael Espindola       STRINGIFY_ENUM_CASE(ELF, SHT_ARM_ATTRIBUTES);
2393ba25737SRafael Espindola       STRINGIFY_ENUM_CASE(ELF, SHT_ARM_DEBUGOVERLAY);
2403ba25737SRafael Espindola       STRINGIFY_ENUM_CASE(ELF, SHT_ARM_OVERLAYSECTION);
2413ba25737SRafael Espindola     }
2423ba25737SRafael Espindola     break;
2433ba25737SRafael Espindola   case ELF::EM_HEXAGON:
2443ba25737SRafael Espindola     switch (Type) { STRINGIFY_ENUM_CASE(ELF, SHT_HEX_ORDERED); }
2453ba25737SRafael Espindola     break;
2463ba25737SRafael Espindola   case ELF::EM_X86_64:
2473ba25737SRafael Espindola     switch (Type) { STRINGIFY_ENUM_CASE(ELF, SHT_X86_64_UNWIND); }
2483ba25737SRafael Espindola     break;
2493ba25737SRafael Espindola   case ELF::EM_MIPS:
2503ba25737SRafael Espindola   case ELF::EM_MIPS_RS3_LE:
2513ba25737SRafael Espindola     switch (Type) {
2523ba25737SRafael Espindola       STRINGIFY_ENUM_CASE(ELF, SHT_MIPS_REGINFO);
2533ba25737SRafael Espindola       STRINGIFY_ENUM_CASE(ELF, SHT_MIPS_OPTIONS);
2543ba25737SRafael Espindola       STRINGIFY_ENUM_CASE(ELF, SHT_MIPS_DWARF);
25561cd368cSFangrui Song       STRINGIFY_ENUM_CASE(ELF, SHT_MIPS_ABIFLAGS);
2563ba25737SRafael Espindola     }
2573ba25737SRafael Espindola     break;
2586cfb4d46SJozef Lawrynowicz   case ELF::EM_MSP430:
2596cfb4d46SJozef Lawrynowicz     switch (Type) { STRINGIFY_ENUM_CASE(ELF, SHT_MSP430_ATTRIBUTES); }
2606cfb4d46SJozef Lawrynowicz     break;
261581ba352SKai Wang   case ELF::EM_RISCV:
262581ba352SKai Wang     switch (Type) { STRINGIFY_ENUM_CASE(ELF, SHT_RISCV_ATTRIBUTES); }
263581ba352SKai Wang     break;
2643ba25737SRafael Espindola   default:
2653ba25737SRafael Espindola     break;
2663ba25737SRafael Espindola   }
2673ba25737SRafael Espindola 
2683ba25737SRafael Espindola   switch (Type) {
2693ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_NULL);
2703ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_PROGBITS);
2713ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_SYMTAB);
2723ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_STRTAB);
2733ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_RELA);
2743ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_HASH);
2753ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_DYNAMIC);
2763ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_NOTE);
2773ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_NOBITS);
2783ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_REL);
2793ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_SHLIB);
2803ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_DYNSYM);
2813ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_INIT_ARRAY);
2823ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_FINI_ARRAY);
2833ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_PREINIT_ARRAY);
2843ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_GROUP);
2853ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_SYMTAB_SHNDX);
2860f440d83SJake Ehrlich     STRINGIFY_ENUM_CASE(ELF, SHT_RELR);
2875c54f15cSPeter Collingbourne     STRINGIFY_ENUM_CASE(ELF, SHT_ANDROID_REL);
2885c54f15cSPeter Collingbourne     STRINGIFY_ENUM_CASE(ELF, SHT_ANDROID_RELA);
2890f440d83SJake Ehrlich     STRINGIFY_ENUM_CASE(ELF, SHT_ANDROID_RELR);
290f0e26e72SPeter Collingbourne     STRINGIFY_ENUM_CASE(ELF, SHT_LLVM_ODRTAB);
291b36fbbc3SSaleem Abdulrasool     STRINGIFY_ENUM_CASE(ELF, SHT_LLVM_LINKER_OPTIONS);
292ae6eeaeaSMichael J. Spencer     STRINGIFY_ENUM_CASE(ELF, SHT_LLVM_CALL_GRAPH_PROFILE);
2933e227336SPeter Collingbourne     STRINGIFY_ENUM_CASE(ELF, SHT_LLVM_ADDRSIG);
2941d16515fSBen Dunbobbin     STRINGIFY_ENUM_CASE(ELF, SHT_LLVM_DEPENDENT_LIBRARIES);
29531fda09bSPeter Collingbourne     STRINGIFY_ENUM_CASE(ELF, SHT_LLVM_SYMPART);
2969d2504b6SFangrui Song     STRINGIFY_ENUM_CASE(ELF, SHT_LLVM_PART_EHDR);
2979d2504b6SFangrui Song     STRINGIFY_ENUM_CASE(ELF, SHT_LLVM_PART_PHDR);
2980aa6df65SRahman Lavaee     STRINGIFY_ENUM_CASE(ELF, SHT_LLVM_BB_ADDR_MAP_V0);
2992b0c5d76SRahman Lavaee     STRINGIFY_ENUM_CASE(ELF, SHT_LLVM_BB_ADDR_MAP);
300*1d2ce4daSJoseph Huber     STRINGIFY_ENUM_CASE(ELF, SHT_LLVM_OFFLOADING);
3013ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_GNU_ATTRIBUTES);
3023ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_GNU_HASH);
3033ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_GNU_verdef);
3043ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_GNU_verneed);
3053ba25737SRafael Espindola     STRINGIFY_ENUM_CASE(ELF, SHT_GNU_versym);
3063ba25737SRafael Espindola   default:
3073ba25737SRafael Espindola     return "Unknown";
3083ba25737SRafael Espindola   }
3093ba25737SRafael Espindola }
310689e6c05SPeter Collingbourne 
311689e6c05SPeter Collingbourne template <class ELFT>
3126ae5b9e4SGeorgii Rymar std::vector<typename ELFT::Rel>
decode_relrs(Elf_Relr_Range relrs) const3130f440d83SJake Ehrlich ELFFile<ELFT>::decode_relrs(Elf_Relr_Range relrs) const {
3140f440d83SJake Ehrlich   // This function decodes the contents of an SHT_RELR packed relocation
3150f440d83SJake Ehrlich   // section.
3160f440d83SJake Ehrlich   //
3170f440d83SJake Ehrlich   // Proposal for adding SHT_RELR sections to generic-abi is here:
3180f440d83SJake Ehrlich   //   https://groups.google.com/forum/#!topic/generic-abi/bX460iggiKg
3190f440d83SJake Ehrlich   //
3200f440d83SJake Ehrlich   // The encoded sequence of Elf64_Relr entries in a SHT_RELR section looks
3210f440d83SJake Ehrlich   // like [ AAAAAAAA BBBBBBB1 BBBBBBB1 ... AAAAAAAA BBBBBB1 ... ]
3220f440d83SJake Ehrlich   //
3230f440d83SJake Ehrlich   // i.e. start with an address, followed by any number of bitmaps. The address
3240f440d83SJake Ehrlich   // entry encodes 1 relocation. The subsequent bitmap entries encode up to 63
3250f440d83SJake Ehrlich   // relocations each, at subsequent offsets following the last address entry.
3260f440d83SJake Ehrlich   //
3270f440d83SJake Ehrlich   // The bitmap entries must have 1 in the least significant bit. The assumption
3280f440d83SJake Ehrlich   // here is that an address cannot have 1 in lsb. Odd addresses are not
3290f440d83SJake Ehrlich   // supported.
3300f440d83SJake Ehrlich   //
3310f440d83SJake Ehrlich   // Excluding the least significant bit in the bitmap, each non-zero bit in
3320f440d83SJake Ehrlich   // the bitmap represents a relocation to be applied to a corresponding machine
3330f440d83SJake Ehrlich   // word that follows the base address word. The second least significant bit
3340f440d83SJake Ehrlich   // represents the machine word immediately following the initial address, and
3350f440d83SJake Ehrlich   // each bit that follows represents the next word, in linear order. As such,
3360f440d83SJake Ehrlich   // a single bitmap can encode up to 31 relocations in a 32-bit object, and
3370f440d83SJake Ehrlich   // 63 relocations in a 64-bit object.
3380f440d83SJake Ehrlich   //
3390f440d83SJake Ehrlich   // This encoding has a couple of interesting properties:
3400f440d83SJake Ehrlich   // 1. Looking at any entry, it is clear whether it's an address or a bitmap:
3410f440d83SJake Ehrlich   //    even means address, odd means bitmap.
3420f440d83SJake Ehrlich   // 2. Just a simple list of addresses is a valid encoding.
3430f440d83SJake Ehrlich 
3442a4df6a3SGeorgii Rymar   Elf_Rel Rel;
3452a4df6a3SGeorgii Rymar   Rel.r_info = 0;
3462a4df6a3SGeorgii Rymar   Rel.setType(getRelativeRelocationType(), false);
3472a4df6a3SGeorgii Rymar   std::vector<Elf_Rel> Relocs;
3480f440d83SJake Ehrlich 
3490f440d83SJake Ehrlich   // Word type: uint32_t for Elf32, and uint64_t for Elf64.
3508e1d5327SFangrui Song   using Addr = typename ELFT::uint;
3510f440d83SJake Ehrlich 
3528e1d5327SFangrui Song   Addr Base = 0;
3538e1d5327SFangrui Song   for (Elf_Relr R : relrs) {
3548e1d5327SFangrui Song     typename ELFT::uint Entry = R;
3550f440d83SJake Ehrlich     if ((Entry & 1) == 0) {
3560f440d83SJake Ehrlich       // Even entry: encodes the offset for next relocation.
3572a4df6a3SGeorgii Rymar       Rel.r_offset = Entry;
3582a4df6a3SGeorgii Rymar       Relocs.push_back(Rel);
3590f440d83SJake Ehrlich       // Set base offset for subsequent bitmap entries.
3608e1d5327SFangrui Song       Base = Entry + sizeof(Addr);
3618e1d5327SFangrui Song     } else {
3620f440d83SJake Ehrlich       // Odd entry: encodes bitmap for relocations starting at base.
3638e1d5327SFangrui Song       for (Addr Offset = Base; (Entry >>= 1) != 0; Offset += sizeof(Addr))
3640f440d83SJake Ehrlich         if ((Entry & 1) != 0) {
3652a4df6a3SGeorgii Rymar           Rel.r_offset = Offset;
3662a4df6a3SGeorgii Rymar           Relocs.push_back(Rel);
3670f440d83SJake Ehrlich         }
3688e1d5327SFangrui Song       Base += (CHAR_BIT * sizeof(Entry) - 1) * sizeof(Addr);
3690f440d83SJake Ehrlich     }
3700f440d83SJake Ehrlich   }
3710f440d83SJake Ehrlich 
3720f440d83SJake Ehrlich   return Relocs;
3730f440d83SJake Ehrlich }
3740f440d83SJake Ehrlich 
3750f440d83SJake Ehrlich template <class ELFT>
3760f440d83SJake Ehrlich Expected<std::vector<typename ELFT::Rela>>
android_relas(const Elf_Shdr & Sec) const3774845531fSGeorgii Rymar ELFFile<ELFT>::android_relas(const Elf_Shdr &Sec) const {
378689e6c05SPeter Collingbourne   // This function reads relocations in Android's packed relocation format,
379689e6c05SPeter Collingbourne   // which is based on SLEB128 and delta encoding.
380689e6c05SPeter Collingbourne   Expected<ArrayRef<uint8_t>> ContentsOrErr = getSectionContents(Sec);
381689e6c05SPeter Collingbourne   if (!ContentsOrErr)
382689e6c05SPeter Collingbourne     return ContentsOrErr.takeError();
3833ca502a7SRahman Lavaee   ArrayRef<uint8_t> Content = *ContentsOrErr;
3843ca502a7SRahman Lavaee   if (Content.size() < 4 || Content[0] != 'A' || Content[1] != 'P' ||
3853ca502a7SRahman Lavaee       Content[2] != 'S' || Content[3] != '2')
386689e6c05SPeter Collingbourne     return createError("invalid packed relocation header");
3873ca502a7SRahman Lavaee   DataExtractor Data(Content, isLE(), ELFT::Is64Bits ? 8 : 4);
3883ca502a7SRahman Lavaee   DataExtractor::Cursor Cur(/*Offset=*/4);
389689e6c05SPeter Collingbourne 
3903ca502a7SRahman Lavaee   uint64_t NumRelocs = Data.getSLEB128(Cur);
3913ca502a7SRahman Lavaee   uint64_t Offset = Data.getSLEB128(Cur);
392689e6c05SPeter Collingbourne   uint64_t Addend = 0;
393689e6c05SPeter Collingbourne 
3943ca502a7SRahman Lavaee   if (!Cur)
3953ca502a7SRahman Lavaee     return std::move(Cur.takeError());
396689e6c05SPeter Collingbourne 
397689e6c05SPeter Collingbourne   std::vector<Elf_Rela> Relocs;
398689e6c05SPeter Collingbourne   Relocs.reserve(NumRelocs);
399689e6c05SPeter Collingbourne   while (NumRelocs) {
4003ca502a7SRahman Lavaee     uint64_t NumRelocsInGroup = Data.getSLEB128(Cur);
4013ca502a7SRahman Lavaee     if (!Cur)
4023ca502a7SRahman Lavaee       return std::move(Cur.takeError());
403689e6c05SPeter Collingbourne     if (NumRelocsInGroup > NumRelocs)
404689e6c05SPeter Collingbourne       return createError("relocation group unexpectedly large");
405689e6c05SPeter Collingbourne     NumRelocs -= NumRelocsInGroup;
406689e6c05SPeter Collingbourne 
4073ca502a7SRahman Lavaee     uint64_t GroupFlags = Data.getSLEB128(Cur);
408689e6c05SPeter Collingbourne     bool GroupedByInfo = GroupFlags & ELF::RELOCATION_GROUPED_BY_INFO_FLAG;
409689e6c05SPeter Collingbourne     bool GroupedByOffsetDelta = GroupFlags & ELF::RELOCATION_GROUPED_BY_OFFSET_DELTA_FLAG;
410689e6c05SPeter Collingbourne     bool GroupedByAddend = GroupFlags & ELF::RELOCATION_GROUPED_BY_ADDEND_FLAG;
411689e6c05SPeter Collingbourne     bool GroupHasAddend = GroupFlags & ELF::RELOCATION_GROUP_HAS_ADDEND_FLAG;
412689e6c05SPeter Collingbourne 
413689e6c05SPeter Collingbourne     uint64_t GroupOffsetDelta;
414689e6c05SPeter Collingbourne     if (GroupedByOffsetDelta)
4153ca502a7SRahman Lavaee       GroupOffsetDelta = Data.getSLEB128(Cur);
416689e6c05SPeter Collingbourne 
417689e6c05SPeter Collingbourne     uint64_t GroupRInfo;
418689e6c05SPeter Collingbourne     if (GroupedByInfo)
4193ca502a7SRahman Lavaee       GroupRInfo = Data.getSLEB128(Cur);
420689e6c05SPeter Collingbourne 
421689e6c05SPeter Collingbourne     if (GroupedByAddend && GroupHasAddend)
4223ca502a7SRahman Lavaee       Addend += Data.getSLEB128(Cur);
423689e6c05SPeter Collingbourne 
42462e4fc48SPeter Collingbourne     if (!GroupHasAddend)
42562e4fc48SPeter Collingbourne       Addend = 0;
42662e4fc48SPeter Collingbourne 
4273ca502a7SRahman Lavaee     for (uint64_t I = 0; Cur && I != NumRelocsInGroup; ++I) {
428689e6c05SPeter Collingbourne       Elf_Rela R;
4293ca502a7SRahman Lavaee       Offset += GroupedByOffsetDelta ? GroupOffsetDelta : Data.getSLEB128(Cur);
430689e6c05SPeter Collingbourne       R.r_offset = Offset;
4313ca502a7SRahman Lavaee       R.r_info = GroupedByInfo ? GroupRInfo : Data.getSLEB128(Cur);
43262e4fc48SPeter Collingbourne       if (GroupHasAddend && !GroupedByAddend)
4333ca502a7SRahman Lavaee         Addend += Data.getSLEB128(Cur);
434689e6c05SPeter Collingbourne       R.r_addend = Addend;
435689e6c05SPeter Collingbourne       Relocs.push_back(R);
436689e6c05SPeter Collingbourne     }
4373ca502a7SRahman Lavaee     if (!Cur)
4383ca502a7SRahman Lavaee       return std::move(Cur.takeError());
439689e6c05SPeter Collingbourne   }
440689e6c05SPeter Collingbourne 
441689e6c05SPeter Collingbourne   return Relocs;
442689e6c05SPeter Collingbourne }
443689e6c05SPeter Collingbourne 
4440913dcd7SPaul Semel template <class ELFT>
getDynamicTagAsString(unsigned Arch,uint64_t Type) const445b2858789SXing GUO std::string ELFFile<ELFT>::getDynamicTagAsString(unsigned Arch,
4460913dcd7SPaul Semel                                                  uint64_t Type) const {
4470913dcd7SPaul Semel #define DYNAMIC_STRINGIFY_ENUM(tag, value)                                     \
4480913dcd7SPaul Semel   case value:                                                                  \
4490913dcd7SPaul Semel     return #tag;
4500913dcd7SPaul Semel 
4510913dcd7SPaul Semel #define DYNAMIC_TAG(n, v)
4520913dcd7SPaul Semel   switch (Arch) {
45349d7221fSPeter Smith   case ELF::EM_AARCH64:
45449d7221fSPeter Smith     switch (Type) {
45549d7221fSPeter Smith #define AARCH64_DYNAMIC_TAG(name, value) DYNAMIC_STRINGIFY_ENUM(name, value)
45649d7221fSPeter Smith #include "llvm/BinaryFormat/DynamicTags.def"
45749d7221fSPeter Smith #undef AARCH64_DYNAMIC_TAG
45849d7221fSPeter Smith     }
45949d7221fSPeter Smith     break;
46049d7221fSPeter Smith 
4610913dcd7SPaul Semel   case ELF::EM_HEXAGON:
4620913dcd7SPaul Semel     switch (Type) {
4630913dcd7SPaul Semel #define HEXAGON_DYNAMIC_TAG(name, value) DYNAMIC_STRINGIFY_ENUM(name, value)
4640913dcd7SPaul Semel #include "llvm/BinaryFormat/DynamicTags.def"
4650913dcd7SPaul Semel #undef HEXAGON_DYNAMIC_TAG
4660913dcd7SPaul Semel     }
467c64d73ccSJonas Hahnfeld     break;
4680913dcd7SPaul Semel 
4690913dcd7SPaul Semel   case ELF::EM_MIPS:
4700913dcd7SPaul Semel     switch (Type) {
4710913dcd7SPaul Semel #define MIPS_DYNAMIC_TAG(name, value) DYNAMIC_STRINGIFY_ENUM(name, value)
4720913dcd7SPaul Semel #include "llvm/BinaryFormat/DynamicTags.def"
4730913dcd7SPaul Semel #undef MIPS_DYNAMIC_TAG
4740913dcd7SPaul Semel     }
475c64d73ccSJonas Hahnfeld     break;
4760913dcd7SPaul Semel 
47767d4d7cfSFangrui Song   case ELF::EM_PPC:
47867d4d7cfSFangrui Song     switch (Type) {
47967d4d7cfSFangrui Song #define PPC_DYNAMIC_TAG(name, value) DYNAMIC_STRINGIFY_ENUM(name, value)
48067d4d7cfSFangrui Song #include "llvm/BinaryFormat/DynamicTags.def"
48167d4d7cfSFangrui Song #undef PPC_DYNAMIC_TAG
48267d4d7cfSFangrui Song     }
48367d4d7cfSFangrui Song     break;
48467d4d7cfSFangrui Song 
4850913dcd7SPaul Semel   case ELF::EM_PPC64:
4860913dcd7SPaul Semel     switch (Type) {
4870913dcd7SPaul Semel #define PPC64_DYNAMIC_TAG(name, value) DYNAMIC_STRINGIFY_ENUM(name, value)
4880913dcd7SPaul Semel #include "llvm/BinaryFormat/DynamicTags.def"
4890913dcd7SPaul Semel #undef PPC64_DYNAMIC_TAG
4900913dcd7SPaul Semel     }
491c64d73ccSJonas Hahnfeld     break;
4928971b99cSFangrui Song 
4938971b99cSFangrui Song   case ELF::EM_RISCV:
4948971b99cSFangrui Song     switch (Type) {
4958971b99cSFangrui Song #define RISCV_DYNAMIC_TAG(name, value) DYNAMIC_STRINGIFY_ENUM(name, value)
4968971b99cSFangrui Song #include "llvm/BinaryFormat/DynamicTags.def"
4978971b99cSFangrui Song #undef RISCV_DYNAMIC_TAG
4988971b99cSFangrui Song     }
4998971b99cSFangrui Song     break;
5000913dcd7SPaul Semel   }
5010913dcd7SPaul Semel #undef DYNAMIC_TAG
5020913dcd7SPaul Semel   switch (Type) {
5030913dcd7SPaul Semel // Now handle all dynamic tags except the architecture specific ones
50449d7221fSPeter Smith #define AARCH64_DYNAMIC_TAG(name, value)
5050913dcd7SPaul Semel #define MIPS_DYNAMIC_TAG(name, value)
5060913dcd7SPaul Semel #define HEXAGON_DYNAMIC_TAG(name, value)
50767d4d7cfSFangrui Song #define PPC_DYNAMIC_TAG(name, value)
5080913dcd7SPaul Semel #define PPC64_DYNAMIC_TAG(name, value)
5098971b99cSFangrui Song #define RISCV_DYNAMIC_TAG(name, value)
5100913dcd7SPaul Semel // Also ignore marker tags such as DT_HIOS (maps to DT_VERNEEDNUM), etc.
5110913dcd7SPaul Semel #define DYNAMIC_TAG_MARKER(name, value)
512301cb914SGeorgii Rymar #define DYNAMIC_TAG(name, value) case value: return #name;
5130913dcd7SPaul Semel #include "llvm/BinaryFormat/DynamicTags.def"
5140913dcd7SPaul Semel #undef DYNAMIC_TAG
51549d7221fSPeter Smith #undef AARCH64_DYNAMIC_TAG
5160913dcd7SPaul Semel #undef MIPS_DYNAMIC_TAG
5170913dcd7SPaul Semel #undef HEXAGON_DYNAMIC_TAG
51867d4d7cfSFangrui Song #undef PPC_DYNAMIC_TAG
5190913dcd7SPaul Semel #undef PPC64_DYNAMIC_TAG
5208971b99cSFangrui Song #undef RISCV_DYNAMIC_TAG
5210913dcd7SPaul Semel #undef DYNAMIC_TAG_MARKER
5220913dcd7SPaul Semel #undef DYNAMIC_STRINGIFY_ENUM
5230913dcd7SPaul Semel   default:
524b2858789SXing GUO     return "<unknown:>0x" + utohexstr(Type, true);
5250913dcd7SPaul Semel   }
5260913dcd7SPaul Semel }
5270913dcd7SPaul Semel 
5280913dcd7SPaul Semel template <class ELFT>
getDynamicTagAsString(uint64_t Type) const529b2858789SXing GUO std::string ELFFile<ELFT>::getDynamicTagAsString(uint64_t Type) const {
5304845531fSGeorgii Rymar   return getDynamicTagAsString(getHeader().e_machine, Type);
5310913dcd7SPaul Semel }
5320913dcd7SPaul Semel 
5330913dcd7SPaul Semel template <class ELFT>
dynamicEntries() const5340913dcd7SPaul Semel Expected<typename ELFT::DynRange> ELFFile<ELFT>::dynamicEntries() const {
5350913dcd7SPaul Semel   ArrayRef<Elf_Dyn> Dyn;
5360913dcd7SPaul Semel 
5370913dcd7SPaul Semel   auto ProgramHeadersOrError = program_headers();
5380913dcd7SPaul Semel   if (!ProgramHeadersOrError)
5390913dcd7SPaul Semel     return ProgramHeadersOrError.takeError();
5400913dcd7SPaul Semel 
5410913dcd7SPaul Semel   for (const Elf_Phdr &Phdr : *ProgramHeadersOrError) {
5420913dcd7SPaul Semel     if (Phdr.p_type == ELF::PT_DYNAMIC) {
5430913dcd7SPaul Semel       Dyn = makeArrayRef(
5440913dcd7SPaul Semel           reinterpret_cast<const Elf_Dyn *>(base() + Phdr.p_offset),
5450913dcd7SPaul Semel           Phdr.p_filesz / sizeof(Elf_Dyn));
5460913dcd7SPaul Semel       break;
5470913dcd7SPaul Semel     }
5480913dcd7SPaul Semel   }
5490913dcd7SPaul Semel 
5500913dcd7SPaul Semel   // If we can't find the dynamic section in the program headers, we just fall
5510913dcd7SPaul Semel   // back on the sections.
5520913dcd7SPaul Semel   if (Dyn.empty()) {
5530913dcd7SPaul Semel     auto SectionsOrError = sections();
5540913dcd7SPaul Semel     if (!SectionsOrError)
5550913dcd7SPaul Semel       return SectionsOrError.takeError();
5560913dcd7SPaul Semel 
5570913dcd7SPaul Semel     for (const Elf_Shdr &Sec : *SectionsOrError) {
5580913dcd7SPaul Semel       if (Sec.sh_type == ELF::SHT_DYNAMIC) {
5590913dcd7SPaul Semel         Expected<ArrayRef<Elf_Dyn>> DynOrError =
5604845531fSGeorgii Rymar             getSectionContentsAsArray<Elf_Dyn>(Sec);
5610913dcd7SPaul Semel         if (!DynOrError)
5620913dcd7SPaul Semel           return DynOrError.takeError();
5630913dcd7SPaul Semel         Dyn = *DynOrError;
5640913dcd7SPaul Semel         break;
5650913dcd7SPaul Semel       }
5660913dcd7SPaul Semel     }
5670913dcd7SPaul Semel 
5680913dcd7SPaul Semel     if (!Dyn.data())
5690913dcd7SPaul Semel       return ArrayRef<Elf_Dyn>();
5700913dcd7SPaul Semel   }
5710913dcd7SPaul Semel 
5720913dcd7SPaul Semel   if (Dyn.empty())
5730913dcd7SPaul Semel     return createError("invalid empty dynamic section");
5740913dcd7SPaul Semel 
5750913dcd7SPaul Semel   if (Dyn.back().d_tag != ELF::DT_NULL)
5760913dcd7SPaul Semel     return createError("dynamic sections must be DT_NULL terminated");
5770913dcd7SPaul Semel 
5780913dcd7SPaul Semel   return Dyn;
5790913dcd7SPaul Semel }
5800913dcd7SPaul Semel 
5810913dcd7SPaul Semel template <class ELFT>
58278aea983SGeorgii Rymar Expected<const uint8_t *>
toMappedAddr(uint64_t VAddr,WarningHandler WarnHandler) const58378aea983SGeorgii Rymar ELFFile<ELFT>::toMappedAddr(uint64_t VAddr, WarningHandler WarnHandler) const {
5840913dcd7SPaul Semel   auto ProgramHeadersOrError = program_headers();
5850913dcd7SPaul Semel   if (!ProgramHeadersOrError)
5860913dcd7SPaul Semel     return ProgramHeadersOrError.takeError();
5870913dcd7SPaul Semel 
5880913dcd7SPaul Semel   llvm::SmallVector<Elf_Phdr *, 4> LoadSegments;
5890913dcd7SPaul Semel 
5900913dcd7SPaul Semel   for (const Elf_Phdr &Phdr : *ProgramHeadersOrError)
5910913dcd7SPaul Semel     if (Phdr.p_type == ELF::PT_LOAD)
5920913dcd7SPaul Semel       LoadSegments.push_back(const_cast<Elf_Phdr *>(&Phdr));
5930913dcd7SPaul Semel 
59478aea983SGeorgii Rymar   auto SortPred = [](const Elf_Phdr_Impl<ELFT> *A,
59578aea983SGeorgii Rymar                      const Elf_Phdr_Impl<ELFT> *B) {
59678aea983SGeorgii Rymar     return A->p_vaddr < B->p_vaddr;
59778aea983SGeorgii Rymar   };
59878aea983SGeorgii Rymar   if (!llvm::is_sorted(LoadSegments, SortPred)) {
59978aea983SGeorgii Rymar     if (Error E =
60078aea983SGeorgii Rymar             WarnHandler("loadable segments are unsorted by virtual address"))
60178aea983SGeorgii Rymar       return std::move(E);
60278aea983SGeorgii Rymar     llvm::stable_sort(LoadSegments, SortPred);
60378aea983SGeorgii Rymar   }
60478aea983SGeorgii Rymar 
605cd088ba7SKazu Hirata   const Elf_Phdr *const *I = llvm::upper_bound(
606cd088ba7SKazu Hirata       LoadSegments, VAddr, [](uint64_t VAddr, const Elf_Phdr_Impl<ELFT> *Phdr) {
6070913dcd7SPaul Semel         return VAddr < Phdr->p_vaddr;
6080913dcd7SPaul Semel       });
6090913dcd7SPaul Semel 
6100913dcd7SPaul Semel   if (I == LoadSegments.begin())
611d0921a46SGeorge Rimar     return createError("virtual address is not in any segment: 0x" +
612d0921a46SGeorge Rimar                        Twine::utohexstr(VAddr));
6130913dcd7SPaul Semel   --I;
6140913dcd7SPaul Semel   const Elf_Phdr &Phdr = **I;
6150913dcd7SPaul Semel   uint64_t Delta = VAddr - Phdr.p_vaddr;
6160913dcd7SPaul Semel   if (Delta >= Phdr.p_filesz)
617d0921a46SGeorge Rimar     return createError("virtual address is not in any segment: 0x" +
618d0921a46SGeorge Rimar                        Twine::utohexstr(VAddr));
61930c1f9a5SGeorgii Rymar 
62030c1f9a5SGeorgii Rymar   uint64_t Offset = Phdr.p_offset + Delta;
62130c1f9a5SGeorgii Rymar   if (Offset >= getBufSize())
62230c1f9a5SGeorgii Rymar     return createError("can't map virtual address 0x" +
62330c1f9a5SGeorgii Rymar                        Twine::utohexstr(VAddr) + " to the segment with index " +
62430c1f9a5SGeorgii Rymar                        Twine(&Phdr - (*ProgramHeadersOrError).data() + 1) +
62530c1f9a5SGeorgii Rymar                        ": the segment ends at 0x" +
62630c1f9a5SGeorgii Rymar                        Twine::utohexstr(Phdr.p_offset + Phdr.p_filesz) +
62730c1f9a5SGeorgii Rymar                        ", which is greater than the file size (0x" +
62830c1f9a5SGeorgii Rymar                        Twine::utohexstr(getBufSize()) + ")");
62930c1f9a5SGeorgii Rymar 
63030c1f9a5SGeorgii Rymar   return base() + Offset;
6310913dcd7SPaul Semel }
6320913dcd7SPaul Semel 
633c245c21cSRahman Lavaee template <class ELFT>
634f533ec37SRahman Lavaee Expected<std::vector<BBAddrMap>>
decodeBBAddrMap(const Elf_Shdr & Sec) const635c245c21cSRahman Lavaee ELFFile<ELFT>::decodeBBAddrMap(const Elf_Shdr &Sec) const {
636c245c21cSRahman Lavaee   Expected<ArrayRef<uint8_t>> ContentsOrErr = getSectionContents(Sec);
637c245c21cSRahman Lavaee   if (!ContentsOrErr)
638c245c21cSRahman Lavaee     return ContentsOrErr.takeError();
639c245c21cSRahman Lavaee   ArrayRef<uint8_t> Content = *ContentsOrErr;
640c245c21cSRahman Lavaee   DataExtractor Data(Content, isLE(), ELFT::Is64Bits ? 8 : 4);
641f533ec37SRahman Lavaee   std::vector<BBAddrMap> FunctionEntries;
642c245c21cSRahman Lavaee 
643c245c21cSRahman Lavaee   DataExtractor::Cursor Cur(0);
644c245c21cSRahman Lavaee   Error ULEBSizeErr = Error::success();
645c245c21cSRahman Lavaee   // Helper to extract and decode the next ULEB128 value as uint32_t.
646c245c21cSRahman Lavaee   // Returns zero and sets ULEBSizeErr if the ULEB128 value exceeds the uint32_t
647c245c21cSRahman Lavaee   // limit.
648c245c21cSRahman Lavaee   // Also returns zero if ULEBSizeErr is already in an error state.
649c245c21cSRahman Lavaee   auto ReadULEB128AsUInt32 = [&Data, &Cur, &ULEBSizeErr]() -> uint32_t {
650c245c21cSRahman Lavaee     // Bail out and do not extract data if ULEBSizeErr is already set.
651c245c21cSRahman Lavaee     if (ULEBSizeErr)
652c245c21cSRahman Lavaee       return 0;
653c245c21cSRahman Lavaee     uint64_t Offset = Cur.tell();
654c245c21cSRahman Lavaee     uint64_t Value = Data.getULEB128(Cur);
655c245c21cSRahman Lavaee     if (Value > UINT32_MAX) {
656c245c21cSRahman Lavaee       ULEBSizeErr = createError(
657c245c21cSRahman Lavaee           "ULEB128 value at offset 0x" + Twine::utohexstr(Offset) +
658c245c21cSRahman Lavaee           " exceeds UINT32_MAX (0x" + Twine::utohexstr(Value) + ")");
659c245c21cSRahman Lavaee       return 0;
660c245c21cSRahman Lavaee     }
661c245c21cSRahman Lavaee     return static_cast<uint32_t>(Value);
662c245c21cSRahman Lavaee   };
663c245c21cSRahman Lavaee 
6640aa6df65SRahman Lavaee   uint8_t Version = 0;
665c245c21cSRahman Lavaee   while (!ULEBSizeErr && Cur && Cur.tell() < Content.size()) {
6660aa6df65SRahman Lavaee     if (Sec.sh_type == ELF::SHT_LLVM_BB_ADDR_MAP) {
6670aa6df65SRahman Lavaee       Version = Data.getU8(Cur);
6680aa6df65SRahman Lavaee       if (!Cur)
6690aa6df65SRahman Lavaee         break;
6700aa6df65SRahman Lavaee       if (Version > 1)
6710aa6df65SRahman Lavaee         return createError("unsupported SHT_LLVM_BB_ADDR_MAP version: " +
6720aa6df65SRahman Lavaee                            Twine(static_cast<int>(Version)));
6730aa6df65SRahman Lavaee       Data.getU8(Cur); // Feature byte
6740aa6df65SRahman Lavaee     }
675c245c21cSRahman Lavaee     uintX_t Address = static_cast<uintX_t>(Data.getAddress(Cur));
676c245c21cSRahman Lavaee     uint32_t NumBlocks = ReadULEB128AsUInt32();
677f533ec37SRahman Lavaee     std::vector<BBAddrMap::BBEntry> BBEntries;
6780aa6df65SRahman Lavaee     uint32_t PrevBBEndOffset = 0;
679c245c21cSRahman Lavaee     for (uint32_t BlockID = 0; !ULEBSizeErr && Cur && (BlockID < NumBlocks);
680c245c21cSRahman Lavaee          ++BlockID) {
681c245c21cSRahman Lavaee       uint32_t Offset = ReadULEB128AsUInt32();
682c245c21cSRahman Lavaee       uint32_t Size = ReadULEB128AsUInt32();
683c245c21cSRahman Lavaee       uint32_t Metadata = ReadULEB128AsUInt32();
6840aa6df65SRahman Lavaee       if (Version >= 1) {
6850aa6df65SRahman Lavaee         // Offset is calculated relative to the end of the previous BB.
6860aa6df65SRahman Lavaee         Offset += PrevBBEndOffset;
6870aa6df65SRahman Lavaee         PrevBBEndOffset = Offset + Size;
6880aa6df65SRahman Lavaee       }
689c245c21cSRahman Lavaee       BBEntries.push_back({Offset, Size, Metadata});
690c245c21cSRahman Lavaee     }
6910aa6df65SRahman Lavaee     FunctionEntries.push_back({Address, std::move(BBEntries)});
692c245c21cSRahman Lavaee   }
693c245c21cSRahman Lavaee   // Either Cur is in the error state, or ULEBSizeError is set (not both), but
694c245c21cSRahman Lavaee   // we join the two errors here to be safe.
695c245c21cSRahman Lavaee   if (!Cur || ULEBSizeErr)
696c245c21cSRahman Lavaee     return joinErrors(Cur.takeError(), std::move(ULEBSizeErr));
697c245c21cSRahman Lavaee   return FunctionEntries;
698c245c21cSRahman Lavaee }
699c245c21cSRahman Lavaee 
700689e6c05SPeter Collingbourne template class llvm::object::ELFFile<ELF32LE>;
701689e6c05SPeter Collingbourne template class llvm::object::ELFFile<ELF32BE>;
702689e6c05SPeter Collingbourne template class llvm::object::ELFFile<ELF64LE>;
703689e6c05SPeter Collingbourne template class llvm::object::ELFFile<ELF64BE>;
704