xref: /aosp_15_r20/system/unwinding/libunwindstack/ElfInterface.cpp (revision eb293b8f56ee8303637c5595cfcdeef8039e85c6)
1*eb293b8fSAndroid Build Coastguard Worker /*
2*eb293b8fSAndroid Build Coastguard Worker  * Copyright (C) 2017 The Android Open Source Project
3*eb293b8fSAndroid Build Coastguard Worker  *
4*eb293b8fSAndroid Build Coastguard Worker  * Licensed under the Apache License, Version 2.0 (the "License");
5*eb293b8fSAndroid Build Coastguard Worker  * you may not use this file except in compliance with the License.
6*eb293b8fSAndroid Build Coastguard Worker  * You may obtain a copy of the License at
7*eb293b8fSAndroid Build Coastguard Worker  *
8*eb293b8fSAndroid Build Coastguard Worker  *      http://www.apache.org/licenses/LICENSE-2.0
9*eb293b8fSAndroid Build Coastguard Worker  *
10*eb293b8fSAndroid Build Coastguard Worker  * Unless required by applicable law or agreed to in writing, software
11*eb293b8fSAndroid Build Coastguard Worker  * distributed under the License is distributed on an "AS IS" BASIS,
12*eb293b8fSAndroid Build Coastguard Worker  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*eb293b8fSAndroid Build Coastguard Worker  * See the License for the specific language governing permissions and
14*eb293b8fSAndroid Build Coastguard Worker  * limitations under the License.
15*eb293b8fSAndroid Build Coastguard Worker  */
16*eb293b8fSAndroid Build Coastguard Worker 
17*eb293b8fSAndroid Build Coastguard Worker #include <elf.h>
18*eb293b8fSAndroid Build Coastguard Worker #include <stdint.h>
19*eb293b8fSAndroid Build Coastguard Worker 
20*eb293b8fSAndroid Build Coastguard Worker #include <memory>
21*eb293b8fSAndroid Build Coastguard Worker #include <string>
22*eb293b8fSAndroid Build Coastguard Worker #include <utility>
23*eb293b8fSAndroid Build Coastguard Worker 
24*eb293b8fSAndroid Build Coastguard Worker #include <zlib.h>
25*eb293b8fSAndroid Build Coastguard Worker #include <zstd.h>
26*eb293b8fSAndroid Build Coastguard Worker 
27*eb293b8fSAndroid Build Coastguard Worker #include <unwindstack/DwarfError.h>
28*eb293b8fSAndroid Build Coastguard Worker #include <unwindstack/DwarfSection.h>
29*eb293b8fSAndroid Build Coastguard Worker #include <unwindstack/ElfInterface.h>
30*eb293b8fSAndroid Build Coastguard Worker #include <unwindstack/Log.h>
31*eb293b8fSAndroid Build Coastguard Worker #include <unwindstack/Memory.h>
32*eb293b8fSAndroid Build Coastguard Worker #include <unwindstack/Regs.h>
33*eb293b8fSAndroid Build Coastguard Worker 
34*eb293b8fSAndroid Build Coastguard Worker #include "DwarfDebugFrame.h"
35*eb293b8fSAndroid Build Coastguard Worker #include "DwarfEhFrame.h"
36*eb293b8fSAndroid Build Coastguard Worker #include "DwarfEhFrameWithHdr.h"
37*eb293b8fSAndroid Build Coastguard Worker #include "MemoryBuffer.h"
38*eb293b8fSAndroid Build Coastguard Worker #include "MemoryXz.h"
39*eb293b8fSAndroid Build Coastguard Worker #include "Symbols.h"
40*eb293b8fSAndroid Build Coastguard Worker 
41*eb293b8fSAndroid Build Coastguard Worker namespace unwindstack {
42*eb293b8fSAndroid Build Coastguard Worker 
~ElfInterface()43*eb293b8fSAndroid Build Coastguard Worker ElfInterface::~ElfInterface() {
44*eb293b8fSAndroid Build Coastguard Worker   for (auto symbol : symbols_) {
45*eb293b8fSAndroid Build Coastguard Worker     delete symbol;
46*eb293b8fSAndroid Build Coastguard Worker   }
47*eb293b8fSAndroid Build Coastguard Worker }
48*eb293b8fSAndroid Build Coastguard Worker 
IsValidPc(uint64_t pc)49*eb293b8fSAndroid Build Coastguard Worker bool ElfInterface::IsValidPc(uint64_t pc) {
50*eb293b8fSAndroid Build Coastguard Worker   if (!pt_loads_.empty()) {
51*eb293b8fSAndroid Build Coastguard Worker     for (auto& entry : pt_loads_) {
52*eb293b8fSAndroid Build Coastguard Worker       uint64_t start = entry.second.table_offset;
53*eb293b8fSAndroid Build Coastguard Worker       uint64_t end = start + entry.second.table_size;
54*eb293b8fSAndroid Build Coastguard Worker       if (pc >= start && pc < end) {
55*eb293b8fSAndroid Build Coastguard Worker         return true;
56*eb293b8fSAndroid Build Coastguard Worker       }
57*eb293b8fSAndroid Build Coastguard Worker     }
58*eb293b8fSAndroid Build Coastguard Worker     return false;
59*eb293b8fSAndroid Build Coastguard Worker   }
60*eb293b8fSAndroid Build Coastguard Worker 
61*eb293b8fSAndroid Build Coastguard Worker   // No PT_LOAD data, look for a fde for this pc in the section data.
62*eb293b8fSAndroid Build Coastguard Worker   if (debug_frame_ != nullptr && debug_frame_->GetFdeFromPc(pc) != nullptr) {
63*eb293b8fSAndroid Build Coastguard Worker     return true;
64*eb293b8fSAndroid Build Coastguard Worker   }
65*eb293b8fSAndroid Build Coastguard Worker 
66*eb293b8fSAndroid Build Coastguard Worker   if (eh_frame_ != nullptr && eh_frame_->GetFdeFromPc(pc) != nullptr) {
67*eb293b8fSAndroid Build Coastguard Worker     return true;
68*eb293b8fSAndroid Build Coastguard Worker   }
69*eb293b8fSAndroid Build Coastguard Worker 
70*eb293b8fSAndroid Build Coastguard Worker   return false;
71*eb293b8fSAndroid Build Coastguard Worker }
72*eb293b8fSAndroid Build Coastguard Worker 
GetTextRange(uint64_t * addr,uint64_t * size)73*eb293b8fSAndroid Build Coastguard Worker bool ElfInterface::GetTextRange(uint64_t* addr, uint64_t* size) {
74*eb293b8fSAndroid Build Coastguard Worker   if (text_size_ != 0) {
75*eb293b8fSAndroid Build Coastguard Worker     *addr = text_addr_;
76*eb293b8fSAndroid Build Coastguard Worker     *size = text_size_;
77*eb293b8fSAndroid Build Coastguard Worker     return true;
78*eb293b8fSAndroid Build Coastguard Worker   }
79*eb293b8fSAndroid Build Coastguard Worker   return false;
80*eb293b8fSAndroid Build Coastguard Worker }
81*eb293b8fSAndroid Build Coastguard Worker 
CreateGnuDebugdataMemory()82*eb293b8fSAndroid Build Coastguard Worker std::shared_ptr<Memory> ElfInterface::CreateGnuDebugdataMemory() {
83*eb293b8fSAndroid Build Coastguard Worker   if (gnu_debugdata_offset_ == 0 || gnu_debugdata_size_ == 0) {
84*eb293b8fSAndroid Build Coastguard Worker     return nullptr;
85*eb293b8fSAndroid Build Coastguard Worker   }
86*eb293b8fSAndroid Build Coastguard Worker 
87*eb293b8fSAndroid Build Coastguard Worker   auto decompressed = std::make_shared<MemoryXz>(memory_.get(), gnu_debugdata_offset_,
88*eb293b8fSAndroid Build Coastguard Worker                                                  gnu_debugdata_size_, GetSoname());
89*eb293b8fSAndroid Build Coastguard Worker   if (!decompressed || !decompressed->Init()) {
90*eb293b8fSAndroid Build Coastguard Worker     gnu_debugdata_offset_ = 0;
91*eb293b8fSAndroid Build Coastguard Worker     gnu_debugdata_size_ = 0;
92*eb293b8fSAndroid Build Coastguard Worker     return nullptr;
93*eb293b8fSAndroid Build Coastguard Worker   }
94*eb293b8fSAndroid Build Coastguard Worker   return decompressed;
95*eb293b8fSAndroid Build Coastguard Worker }
96*eb293b8fSAndroid Build Coastguard Worker 
ZlibDecompress(uint8_t * compressed_data,size_t compressed_size,MemoryBuffer * memory)97*eb293b8fSAndroid Build Coastguard Worker static bool ZlibDecompress(uint8_t* compressed_data, size_t compressed_size, MemoryBuffer* memory) {
98*eb293b8fSAndroid Build Coastguard Worker   z_stream stream;
99*eb293b8fSAndroid Build Coastguard Worker   stream.zalloc = Z_NULL;
100*eb293b8fSAndroid Build Coastguard Worker   stream.zfree = Z_NULL;
101*eb293b8fSAndroid Build Coastguard Worker   stream.opaque = Z_NULL;
102*eb293b8fSAndroid Build Coastguard Worker   if (inflateInit(&stream) != Z_OK) {
103*eb293b8fSAndroid Build Coastguard Worker     return false;
104*eb293b8fSAndroid Build Coastguard Worker   }
105*eb293b8fSAndroid Build Coastguard Worker   stream.next_in = compressed_data;
106*eb293b8fSAndroid Build Coastguard Worker   stream.avail_in = compressed_size;
107*eb293b8fSAndroid Build Coastguard Worker   stream.next_out = memory->Data();
108*eb293b8fSAndroid Build Coastguard Worker   stream.avail_out = memory->Size();
109*eb293b8fSAndroid Build Coastguard Worker   int ret = inflate(&stream, Z_FINISH);
110*eb293b8fSAndroid Build Coastguard Worker   if (inflateEnd(&stream) != Z_OK) {
111*eb293b8fSAndroid Build Coastguard Worker     return false;
112*eb293b8fSAndroid Build Coastguard Worker   }
113*eb293b8fSAndroid Build Coastguard Worker   return ret == Z_STREAM_END;
114*eb293b8fSAndroid Build Coastguard Worker }
115*eb293b8fSAndroid Build Coastguard Worker 
ZstdDecompress(uint8_t * compressed_data,size_t compressed_size,MemoryBuffer * memory)116*eb293b8fSAndroid Build Coastguard Worker static bool ZstdDecompress(uint8_t* compressed_data, size_t compressed_size, MemoryBuffer* memory) {
117*eb293b8fSAndroid Build Coastguard Worker   size_t decompress_size =
118*eb293b8fSAndroid Build Coastguard Worker       ZSTD_decompress(memory->Data(), memory->Size(), compressed_data, compressed_size);
119*eb293b8fSAndroid Build Coastguard Worker   return memory->Size() == decompress_size;
120*eb293b8fSAndroid Build Coastguard Worker }
121*eb293b8fSAndroid Build Coastguard Worker 
122*eb293b8fSAndroid Build Coastguard Worker template <typename ChdrType>
CreateMemoryFromCompressedSection(SectionInfo & info,std::shared_ptr<Memory> & elf_memory)123*eb293b8fSAndroid Build Coastguard Worker std::shared_ptr<Memory> CreateMemoryFromCompressedSection(SectionInfo& info,
124*eb293b8fSAndroid Build Coastguard Worker                                                           std::shared_ptr<Memory>& elf_memory) {
125*eb293b8fSAndroid Build Coastguard Worker   if (info.size < sizeof(ChdrType)) {
126*eb293b8fSAndroid Build Coastguard Worker     return nullptr;
127*eb293b8fSAndroid Build Coastguard Worker   }
128*eb293b8fSAndroid Build Coastguard Worker 
129*eb293b8fSAndroid Build Coastguard Worker   uint8_t* compressed_data = elf_memory->GetPtr(info.offset);
130*eb293b8fSAndroid Build Coastguard Worker   std::vector<uint8_t> compressed;
131*eb293b8fSAndroid Build Coastguard Worker   if (compressed_data == nullptr || elf_memory->GetPtr(info.offset + info.size - 1) == nullptr) {
132*eb293b8fSAndroid Build Coastguard Worker     compressed.resize(info.size);
133*eb293b8fSAndroid Build Coastguard Worker     if (!elf_memory->ReadFully(info.offset, compressed.data(), info.size)) {
134*eb293b8fSAndroid Build Coastguard Worker       return nullptr;
135*eb293b8fSAndroid Build Coastguard Worker     }
136*eb293b8fSAndroid Build Coastguard Worker     compressed_data = compressed.data();
137*eb293b8fSAndroid Build Coastguard Worker   }
138*eb293b8fSAndroid Build Coastguard Worker 
139*eb293b8fSAndroid Build Coastguard Worker   ChdrType* chdr = reinterpret_cast<ChdrType*>(compressed_data);
140*eb293b8fSAndroid Build Coastguard Worker   std::shared_ptr<MemoryBuffer> memory(new MemoryBuffer(chdr->ch_size, info.offset));
141*eb293b8fSAndroid Build Coastguard Worker 
142*eb293b8fSAndroid Build Coastguard Worker   bool ret = false;
143*eb293b8fSAndroid Build Coastguard Worker   if (chdr->ch_type == ELFCOMPRESS_ZLIB) {
144*eb293b8fSAndroid Build Coastguard Worker     ret = ZlibDecompress(&compressed_data[sizeof(ChdrType)], info.size - sizeof(ChdrType),
145*eb293b8fSAndroid Build Coastguard Worker                          memory.get());
146*eb293b8fSAndroid Build Coastguard Worker   } else if (chdr->ch_type == ELFCOMPRESS_ZSTD) {
147*eb293b8fSAndroid Build Coastguard Worker     ret = ZstdDecompress(&compressed_data[sizeof(ChdrType)], info.size - sizeof(ChdrType),
148*eb293b8fSAndroid Build Coastguard Worker                          memory.get());
149*eb293b8fSAndroid Build Coastguard Worker   }
150*eb293b8fSAndroid Build Coastguard Worker   if (!ret) {
151*eb293b8fSAndroid Build Coastguard Worker     return nullptr;
152*eb293b8fSAndroid Build Coastguard Worker   }
153*eb293b8fSAndroid Build Coastguard Worker   // Set the section info to match the uncompressed section data.
154*eb293b8fSAndroid Build Coastguard Worker   info.size = chdr->ch_size;
155*eb293b8fSAndroid Build Coastguard Worker   info.flags &= ~SHF_COMPRESSED;
156*eb293b8fSAndroid Build Coastguard Worker   return memory;
157*eb293b8fSAndroid Build Coastguard Worker }
158*eb293b8fSAndroid Build Coastguard Worker 
159*eb293b8fSAndroid Build Coastguard Worker template <typename ElfTypes>
InitHeaders()160*eb293b8fSAndroid Build Coastguard Worker void ElfInterfaceImpl<ElfTypes>::InitHeaders() {
161*eb293b8fSAndroid Build Coastguard Worker   if (eh_frame_hdr_info_.offset != 0) {
162*eb293b8fSAndroid Build Coastguard Worker     DwarfEhFrameWithHdr<AddressType>* eh_frame_hdr = new DwarfEhFrameWithHdr<AddressType>(memory_);
163*eb293b8fSAndroid Build Coastguard Worker     eh_frame_.reset(eh_frame_hdr);
164*eb293b8fSAndroid Build Coastguard Worker     if (!eh_frame_hdr->EhFrameInit(eh_frame_info_) || !eh_frame_->Init(eh_frame_hdr_info_)) {
165*eb293b8fSAndroid Build Coastguard Worker       eh_frame_hdr_info_ = {};
166*eb293b8fSAndroid Build Coastguard Worker       eh_frame_.reset(nullptr);
167*eb293b8fSAndroid Build Coastguard Worker     }
168*eb293b8fSAndroid Build Coastguard Worker   }
169*eb293b8fSAndroid Build Coastguard Worker 
170*eb293b8fSAndroid Build Coastguard Worker   if (eh_frame_.get() == nullptr && eh_frame_info_.offset != 0) {
171*eb293b8fSAndroid Build Coastguard Worker     // If there is an eh_frame section without an eh_frame_hdr section,
172*eb293b8fSAndroid Build Coastguard Worker     // or using the frame hdr object failed to init.
173*eb293b8fSAndroid Build Coastguard Worker     eh_frame_.reset(new DwarfEhFrame<AddressType>(memory_));
174*eb293b8fSAndroid Build Coastguard Worker     if (!eh_frame_->Init(eh_frame_info_)) {
175*eb293b8fSAndroid Build Coastguard Worker       eh_frame_info_ = {};
176*eb293b8fSAndroid Build Coastguard Worker       eh_frame_.reset(nullptr);
177*eb293b8fSAndroid Build Coastguard Worker     }
178*eb293b8fSAndroid Build Coastguard Worker   }
179*eb293b8fSAndroid Build Coastguard Worker 
180*eb293b8fSAndroid Build Coastguard Worker   if (debug_frame_info_.offset != 0) {
181*eb293b8fSAndroid Build Coastguard Worker     std::shared_ptr<Memory> debug_memory = memory_;
182*eb293b8fSAndroid Build Coastguard Worker     if (debug_frame_info_.flags & SHF_COMPRESSED) {
183*eb293b8fSAndroid Build Coastguard Worker       debug_memory = CreateMemoryFromCompressedSection<ChdrType>(debug_frame_info_, memory_);
184*eb293b8fSAndroid Build Coastguard Worker     }
185*eb293b8fSAndroid Build Coastguard Worker     debug_frame_.reset(new DwarfDebugFrame<AddressType>(debug_memory));
186*eb293b8fSAndroid Build Coastguard Worker     if (!debug_frame_->Init(debug_frame_info_)) {
187*eb293b8fSAndroid Build Coastguard Worker       debug_frame_.reset(nullptr);
188*eb293b8fSAndroid Build Coastguard Worker       debug_frame_info_ = {};
189*eb293b8fSAndroid Build Coastguard Worker     }
190*eb293b8fSAndroid Build Coastguard Worker   }
191*eb293b8fSAndroid Build Coastguard Worker }
192*eb293b8fSAndroid Build Coastguard Worker 
193*eb293b8fSAndroid Build Coastguard Worker template <typename ElfTypes>
ReadAllHeaders(int64_t * load_bias)194*eb293b8fSAndroid Build Coastguard Worker bool ElfInterfaceImpl<ElfTypes>::ReadAllHeaders(int64_t* load_bias) {
195*eb293b8fSAndroid Build Coastguard Worker   EhdrType ehdr;
196*eb293b8fSAndroid Build Coastguard Worker   if (!memory_->ReadFully(0, &ehdr, sizeof(ehdr))) {
197*eb293b8fSAndroid Build Coastguard Worker     last_error_.code = ERROR_MEMORY_INVALID;
198*eb293b8fSAndroid Build Coastguard Worker     last_error_.address = 0;
199*eb293b8fSAndroid Build Coastguard Worker     return false;
200*eb293b8fSAndroid Build Coastguard Worker   }
201*eb293b8fSAndroid Build Coastguard Worker 
202*eb293b8fSAndroid Build Coastguard Worker   // If we have enough information that this is an elf file, then allow
203*eb293b8fSAndroid Build Coastguard Worker   // malformed program and section headers.
204*eb293b8fSAndroid Build Coastguard Worker   ReadProgramHeaders(ehdr, load_bias);
205*eb293b8fSAndroid Build Coastguard Worker   ReadSectionHeaders(ehdr);
206*eb293b8fSAndroid Build Coastguard Worker   return true;
207*eb293b8fSAndroid Build Coastguard Worker }
208*eb293b8fSAndroid Build Coastguard Worker 
209*eb293b8fSAndroid Build Coastguard Worker template <typename EhdrType, typename PhdrType>
GetLoadBias(Memory * memory)210*eb293b8fSAndroid Build Coastguard Worker int64_t ElfInterface::GetLoadBias(Memory* memory) {
211*eb293b8fSAndroid Build Coastguard Worker   EhdrType ehdr;
212*eb293b8fSAndroid Build Coastguard Worker   if (!memory->ReadFully(0, &ehdr, sizeof(ehdr))) {
213*eb293b8fSAndroid Build Coastguard Worker     return false;
214*eb293b8fSAndroid Build Coastguard Worker   }
215*eb293b8fSAndroid Build Coastguard Worker 
216*eb293b8fSAndroid Build Coastguard Worker   uint64_t offset = ehdr.e_phoff;
217*eb293b8fSAndroid Build Coastguard Worker   for (size_t i = 0; i < ehdr.e_phnum; i++, offset += ehdr.e_phentsize) {
218*eb293b8fSAndroid Build Coastguard Worker     PhdrType phdr;
219*eb293b8fSAndroid Build Coastguard Worker     if (!memory->ReadFully(offset, &phdr, sizeof(phdr))) {
220*eb293b8fSAndroid Build Coastguard Worker       return 0;
221*eb293b8fSAndroid Build Coastguard Worker     }
222*eb293b8fSAndroid Build Coastguard Worker 
223*eb293b8fSAndroid Build Coastguard Worker     // Find the first executable load when looking for the load bias.
224*eb293b8fSAndroid Build Coastguard Worker     if (phdr.p_type == PT_LOAD && (phdr.p_flags & PF_X)) {
225*eb293b8fSAndroid Build Coastguard Worker       return static_cast<uint64_t>(phdr.p_vaddr) - phdr.p_offset;
226*eb293b8fSAndroid Build Coastguard Worker     }
227*eb293b8fSAndroid Build Coastguard Worker   }
228*eb293b8fSAndroid Build Coastguard Worker   return 0;
229*eb293b8fSAndroid Build Coastguard Worker }
230*eb293b8fSAndroid Build Coastguard Worker 
231*eb293b8fSAndroid Build Coastguard Worker template <typename ElfTypes>
ReadProgramHeaders(const EhdrType & ehdr,int64_t * load_bias)232*eb293b8fSAndroid Build Coastguard Worker void ElfInterfaceImpl<ElfTypes>::ReadProgramHeaders(const EhdrType& ehdr, int64_t* load_bias) {
233*eb293b8fSAndroid Build Coastguard Worker   uint64_t offset = ehdr.e_phoff;
234*eb293b8fSAndroid Build Coastguard Worker   bool first_exec_load_header = true;
235*eb293b8fSAndroid Build Coastguard Worker   for (size_t i = 0; i < ehdr.e_phnum; i++, offset += ehdr.e_phentsize) {
236*eb293b8fSAndroid Build Coastguard Worker     PhdrType phdr;
237*eb293b8fSAndroid Build Coastguard Worker     if (!memory_->ReadFully(offset, &phdr, sizeof(phdr))) {
238*eb293b8fSAndroid Build Coastguard Worker       return;
239*eb293b8fSAndroid Build Coastguard Worker     }
240*eb293b8fSAndroid Build Coastguard Worker 
241*eb293b8fSAndroid Build Coastguard Worker     switch (phdr.p_type) {
242*eb293b8fSAndroid Build Coastguard Worker     case PT_LOAD:
243*eb293b8fSAndroid Build Coastguard Worker     {
244*eb293b8fSAndroid Build Coastguard Worker       if ((phdr.p_flags & PF_X) == 0) {
245*eb293b8fSAndroid Build Coastguard Worker         continue;
246*eb293b8fSAndroid Build Coastguard Worker       }
247*eb293b8fSAndroid Build Coastguard Worker 
248*eb293b8fSAndroid Build Coastguard Worker       pt_loads_[phdr.p_offset] = LoadInfo{phdr.p_offset, phdr.p_vaddr,
249*eb293b8fSAndroid Build Coastguard Worker                                           static_cast<size_t>(phdr.p_memsz)};
250*eb293b8fSAndroid Build Coastguard Worker       // Only set the load bias from the first executable load header.
251*eb293b8fSAndroid Build Coastguard Worker       if (first_exec_load_header) {
252*eb293b8fSAndroid Build Coastguard Worker         *load_bias = static_cast<uint64_t>(phdr.p_vaddr) - phdr.p_offset;
253*eb293b8fSAndroid Build Coastguard Worker       }
254*eb293b8fSAndroid Build Coastguard Worker       first_exec_load_header = false;
255*eb293b8fSAndroid Build Coastguard Worker       break;
256*eb293b8fSAndroid Build Coastguard Worker     }
257*eb293b8fSAndroid Build Coastguard Worker 
258*eb293b8fSAndroid Build Coastguard Worker     case PT_GNU_EH_FRAME:
259*eb293b8fSAndroid Build Coastguard Worker       // This is really the pointer to the .eh_frame_hdr section.
260*eb293b8fSAndroid Build Coastguard Worker       eh_frame_hdr_info_ = {
261*eb293b8fSAndroid Build Coastguard Worker           .offset = phdr.p_offset,
262*eb293b8fSAndroid Build Coastguard Worker           .size = phdr.p_memsz,
263*eb293b8fSAndroid Build Coastguard Worker           .flags = phdr.p_flags,
264*eb293b8fSAndroid Build Coastguard Worker           .bias = static_cast<int64_t>(static_cast<uint64_t>(phdr.p_vaddr) - phdr.p_offset)};
265*eb293b8fSAndroid Build Coastguard Worker       break;
266*eb293b8fSAndroid Build Coastguard Worker 
267*eb293b8fSAndroid Build Coastguard Worker     case PT_DYNAMIC:
268*eb293b8fSAndroid Build Coastguard Worker       dynamic_offset_ = phdr.p_offset;
269*eb293b8fSAndroid Build Coastguard Worker       dynamic_vaddr_start_ = phdr.p_vaddr;
270*eb293b8fSAndroid Build Coastguard Worker       if (__builtin_add_overflow(dynamic_vaddr_start_, phdr.p_memsz, &dynamic_vaddr_end_)) {
271*eb293b8fSAndroid Build Coastguard Worker         dynamic_offset_ = 0;
272*eb293b8fSAndroid Build Coastguard Worker         dynamic_vaddr_start_ = 0;
273*eb293b8fSAndroid Build Coastguard Worker         dynamic_vaddr_end_ = 0;
274*eb293b8fSAndroid Build Coastguard Worker       }
275*eb293b8fSAndroid Build Coastguard Worker       break;
276*eb293b8fSAndroid Build Coastguard Worker 
277*eb293b8fSAndroid Build Coastguard Worker     default:
278*eb293b8fSAndroid Build Coastguard Worker       HandleUnknownType(phdr.p_type, phdr.p_offset, phdr.p_filesz);
279*eb293b8fSAndroid Build Coastguard Worker       break;
280*eb293b8fSAndroid Build Coastguard Worker     }
281*eb293b8fSAndroid Build Coastguard Worker   }
282*eb293b8fSAndroid Build Coastguard Worker }
283*eb293b8fSAndroid Build Coastguard Worker 
284*eb293b8fSAndroid Build Coastguard Worker template <typename ElfTypes>
ReadBuildID()285*eb293b8fSAndroid Build Coastguard Worker std::string ElfInterfaceImpl<ElfTypes>::ReadBuildID() {
286*eb293b8fSAndroid Build Coastguard Worker   // Ensure there is no overflow in any of the calulations below.
287*eb293b8fSAndroid Build Coastguard Worker   uint64_t tmp;
288*eb293b8fSAndroid Build Coastguard Worker   if (__builtin_add_overflow(gnu_build_id_offset_, gnu_build_id_size_, &tmp)) {
289*eb293b8fSAndroid Build Coastguard Worker     return "";
290*eb293b8fSAndroid Build Coastguard Worker   }
291*eb293b8fSAndroid Build Coastguard Worker 
292*eb293b8fSAndroid Build Coastguard Worker   uint64_t offset = 0;
293*eb293b8fSAndroid Build Coastguard Worker   while (offset < gnu_build_id_size_) {
294*eb293b8fSAndroid Build Coastguard Worker     if (gnu_build_id_size_ - offset < sizeof(NhdrType)) {
295*eb293b8fSAndroid Build Coastguard Worker       return "";
296*eb293b8fSAndroid Build Coastguard Worker     }
297*eb293b8fSAndroid Build Coastguard Worker     NhdrType hdr;
298*eb293b8fSAndroid Build Coastguard Worker     if (!memory_->ReadFully(gnu_build_id_offset_ + offset, &hdr, sizeof(hdr))) {
299*eb293b8fSAndroid Build Coastguard Worker       return "";
300*eb293b8fSAndroid Build Coastguard Worker     }
301*eb293b8fSAndroid Build Coastguard Worker     offset += sizeof(hdr);
302*eb293b8fSAndroid Build Coastguard Worker 
303*eb293b8fSAndroid Build Coastguard Worker     if (gnu_build_id_size_ - offset < hdr.n_namesz) {
304*eb293b8fSAndroid Build Coastguard Worker       return "";
305*eb293b8fSAndroid Build Coastguard Worker     }
306*eb293b8fSAndroid Build Coastguard Worker     if (hdr.n_namesz > 0) {
307*eb293b8fSAndroid Build Coastguard Worker       std::string name(hdr.n_namesz, '\0');
308*eb293b8fSAndroid Build Coastguard Worker       if (!memory_->ReadFully(gnu_build_id_offset_ + offset, &(name[0]), hdr.n_namesz)) {
309*eb293b8fSAndroid Build Coastguard Worker         return "";
310*eb293b8fSAndroid Build Coastguard Worker       }
311*eb293b8fSAndroid Build Coastguard Worker 
312*eb293b8fSAndroid Build Coastguard Worker       // Trim trailing \0 as GNU is stored as a C string in the ELF file.
313*eb293b8fSAndroid Build Coastguard Worker       if (name.back() == '\0')
314*eb293b8fSAndroid Build Coastguard Worker         name.resize(name.size() - 1);
315*eb293b8fSAndroid Build Coastguard Worker 
316*eb293b8fSAndroid Build Coastguard Worker       // Align hdr.n_namesz to next power multiple of 4. See man 5 elf.
317*eb293b8fSAndroid Build Coastguard Worker       offset += (hdr.n_namesz + 3) & ~3;
318*eb293b8fSAndroid Build Coastguard Worker 
319*eb293b8fSAndroid Build Coastguard Worker       if (name == "GNU" && hdr.n_type == NT_GNU_BUILD_ID) {
320*eb293b8fSAndroid Build Coastguard Worker         if (gnu_build_id_size_ - offset < hdr.n_descsz || hdr.n_descsz == 0) {
321*eb293b8fSAndroid Build Coastguard Worker           return "";
322*eb293b8fSAndroid Build Coastguard Worker         }
323*eb293b8fSAndroid Build Coastguard Worker         std::string build_id(hdr.n_descsz, '\0');
324*eb293b8fSAndroid Build Coastguard Worker         if (memory_->ReadFully(gnu_build_id_offset_ + offset, &build_id[0], hdr.n_descsz)) {
325*eb293b8fSAndroid Build Coastguard Worker           return build_id;
326*eb293b8fSAndroid Build Coastguard Worker         }
327*eb293b8fSAndroid Build Coastguard Worker         return "";
328*eb293b8fSAndroid Build Coastguard Worker       }
329*eb293b8fSAndroid Build Coastguard Worker     }
330*eb293b8fSAndroid Build Coastguard Worker     // Align hdr.n_descsz to next power multiple of 4. See man 5 elf.
331*eb293b8fSAndroid Build Coastguard Worker     offset += (hdr.n_descsz + 3) & ~3;
332*eb293b8fSAndroid Build Coastguard Worker   }
333*eb293b8fSAndroid Build Coastguard Worker   return "";
334*eb293b8fSAndroid Build Coastguard Worker }
335*eb293b8fSAndroid Build Coastguard Worker template <typename ElfTypes>
ReadSectionHeaders(const EhdrType & ehdr)336*eb293b8fSAndroid Build Coastguard Worker void ElfInterfaceImpl<ElfTypes>::ReadSectionHeaders(const EhdrType& ehdr) {
337*eb293b8fSAndroid Build Coastguard Worker   uint64_t offset = ehdr.e_shoff;
338*eb293b8fSAndroid Build Coastguard Worker   uint64_t sec_offset = 0;
339*eb293b8fSAndroid Build Coastguard Worker   uint64_t sec_size = 0;
340*eb293b8fSAndroid Build Coastguard Worker 
341*eb293b8fSAndroid Build Coastguard Worker   // Get the location of the section header names.
342*eb293b8fSAndroid Build Coastguard Worker   // If something is malformed in the header table data, we aren't going
343*eb293b8fSAndroid Build Coastguard Worker   // to terminate, we'll simply ignore this part.
344*eb293b8fSAndroid Build Coastguard Worker   ShdrType shdr;
345*eb293b8fSAndroid Build Coastguard Worker   if (ehdr.e_shstrndx < ehdr.e_shnum) {
346*eb293b8fSAndroid Build Coastguard Worker     uint64_t sh_offset = offset + ehdr.e_shstrndx * ehdr.e_shentsize;
347*eb293b8fSAndroid Build Coastguard Worker     if (memory_->ReadFully(sh_offset, &shdr, sizeof(shdr))) {
348*eb293b8fSAndroid Build Coastguard Worker       sec_offset = shdr.sh_offset;
349*eb293b8fSAndroid Build Coastguard Worker       sec_size = shdr.sh_size;
350*eb293b8fSAndroid Build Coastguard Worker     }
351*eb293b8fSAndroid Build Coastguard Worker   }
352*eb293b8fSAndroid Build Coastguard Worker 
353*eb293b8fSAndroid Build Coastguard Worker   // Skip the first header, it's always going to be NULL.
354*eb293b8fSAndroid Build Coastguard Worker   offset += ehdr.e_shentsize;
355*eb293b8fSAndroid Build Coastguard Worker   for (size_t i = 1; i < ehdr.e_shnum; i++, offset += ehdr.e_shentsize) {
356*eb293b8fSAndroid Build Coastguard Worker     if (!memory_->ReadFully(offset, &shdr, sizeof(shdr))) {
357*eb293b8fSAndroid Build Coastguard Worker       return;
358*eb293b8fSAndroid Build Coastguard Worker     }
359*eb293b8fSAndroid Build Coastguard Worker 
360*eb293b8fSAndroid Build Coastguard Worker     if (shdr.sh_type == SHT_SYMTAB || shdr.sh_type == SHT_DYNSYM) {
361*eb293b8fSAndroid Build Coastguard Worker       // Need to go get the information about the section that contains
362*eb293b8fSAndroid Build Coastguard Worker       // the string terminated names.
363*eb293b8fSAndroid Build Coastguard Worker       ShdrType str_shdr;
364*eb293b8fSAndroid Build Coastguard Worker       if (shdr.sh_link >= ehdr.e_shnum) {
365*eb293b8fSAndroid Build Coastguard Worker         continue;
366*eb293b8fSAndroid Build Coastguard Worker       }
367*eb293b8fSAndroid Build Coastguard Worker       uint64_t str_offset = ehdr.e_shoff + shdr.sh_link * ehdr.e_shentsize;
368*eb293b8fSAndroid Build Coastguard Worker       if (!memory_->ReadFully(str_offset, &str_shdr, sizeof(str_shdr))) {
369*eb293b8fSAndroid Build Coastguard Worker         continue;
370*eb293b8fSAndroid Build Coastguard Worker       }
371*eb293b8fSAndroid Build Coastguard Worker       if (str_shdr.sh_type != SHT_STRTAB) {
372*eb293b8fSAndroid Build Coastguard Worker         continue;
373*eb293b8fSAndroid Build Coastguard Worker       }
374*eb293b8fSAndroid Build Coastguard Worker       symbols_.push_back(new Symbols(shdr.sh_offset, shdr.sh_size, shdr.sh_entsize,
375*eb293b8fSAndroid Build Coastguard Worker                                      str_shdr.sh_offset, str_shdr.sh_size));
376*eb293b8fSAndroid Build Coastguard Worker     } else if ((shdr.sh_type == SHT_PROGBITS || shdr.sh_type == SHT_NOBITS) && sec_size != 0) {
377*eb293b8fSAndroid Build Coastguard Worker       // Look for the .debug_frame and .gnu_debugdata.
378*eb293b8fSAndroid Build Coastguard Worker       if (shdr.sh_name < sec_size) {
379*eb293b8fSAndroid Build Coastguard Worker         std::string name;
380*eb293b8fSAndroid Build Coastguard Worker         if (memory_->ReadString(sec_offset + shdr.sh_name, &name, sec_size - shdr.sh_name)) {
381*eb293b8fSAndroid Build Coastguard Worker           if (name == ".debug_frame") {
382*eb293b8fSAndroid Build Coastguard Worker             debug_frame_info_ = {
383*eb293b8fSAndroid Build Coastguard Worker                 .offset = shdr.sh_offset,
384*eb293b8fSAndroid Build Coastguard Worker                 .size = shdr.sh_size,
385*eb293b8fSAndroid Build Coastguard Worker                 .flags = shdr.sh_flags,
386*eb293b8fSAndroid Build Coastguard Worker                 .bias = static_cast<int64_t>(static_cast<uint64_t>(shdr.sh_addr) - shdr.sh_offset)};
387*eb293b8fSAndroid Build Coastguard Worker           } else if (name == ".gnu_debugdata") {
388*eb293b8fSAndroid Build Coastguard Worker             gnu_debugdata_offset_ = shdr.sh_offset;
389*eb293b8fSAndroid Build Coastguard Worker             gnu_debugdata_size_ = shdr.sh_size;
390*eb293b8fSAndroid Build Coastguard Worker           } else if (name == ".eh_frame") {
391*eb293b8fSAndroid Build Coastguard Worker             eh_frame_info_ = {
392*eb293b8fSAndroid Build Coastguard Worker                 .offset = shdr.sh_offset,
393*eb293b8fSAndroid Build Coastguard Worker                 .size = shdr.sh_size,
394*eb293b8fSAndroid Build Coastguard Worker                 .flags = shdr.sh_flags,
395*eb293b8fSAndroid Build Coastguard Worker                 .bias = static_cast<int64_t>(static_cast<uint64_t>(shdr.sh_addr) - shdr.sh_offset)};
396*eb293b8fSAndroid Build Coastguard Worker           } else if (eh_frame_hdr_info_.offset == 0 && name == ".eh_frame_hdr") {
397*eb293b8fSAndroid Build Coastguard Worker             eh_frame_hdr_info_ = {
398*eb293b8fSAndroid Build Coastguard Worker                 .offset = shdr.sh_offset,
399*eb293b8fSAndroid Build Coastguard Worker                 .size = shdr.sh_size,
400*eb293b8fSAndroid Build Coastguard Worker                 .flags = shdr.sh_flags,
401*eb293b8fSAndroid Build Coastguard Worker                 .bias = static_cast<int64_t>(static_cast<uint64_t>(shdr.sh_addr) - shdr.sh_offset)};
402*eb293b8fSAndroid Build Coastguard Worker           } else if (name == ".data") {
403*eb293b8fSAndroid Build Coastguard Worker             data_offset_ = shdr.sh_offset;
404*eb293b8fSAndroid Build Coastguard Worker             data_vaddr_start_ = shdr.sh_addr;
405*eb293b8fSAndroid Build Coastguard Worker             if (__builtin_add_overflow(data_vaddr_start_, shdr.sh_size, &data_vaddr_end_)) {
406*eb293b8fSAndroid Build Coastguard Worker               data_offset_ = 0;
407*eb293b8fSAndroid Build Coastguard Worker               data_vaddr_start_ = 0;
408*eb293b8fSAndroid Build Coastguard Worker               data_vaddr_end_ = 0;
409*eb293b8fSAndroid Build Coastguard Worker             }
410*eb293b8fSAndroid Build Coastguard Worker           } else if (name == ".text") {
411*eb293b8fSAndroid Build Coastguard Worker             text_addr_ = shdr.sh_addr;
412*eb293b8fSAndroid Build Coastguard Worker             text_size_ = shdr.sh_size;
413*eb293b8fSAndroid Build Coastguard Worker           }
414*eb293b8fSAndroid Build Coastguard Worker         }
415*eb293b8fSAndroid Build Coastguard Worker       }
416*eb293b8fSAndroid Build Coastguard Worker     } else if (shdr.sh_type == SHT_STRTAB) {
417*eb293b8fSAndroid Build Coastguard Worker       // In order to read soname, keep track of address to offset mapping.
418*eb293b8fSAndroid Build Coastguard Worker       strtabs_.push_back(std::make_pair<uint64_t, uint64_t>(static_cast<uint64_t>(shdr.sh_addr),
419*eb293b8fSAndroid Build Coastguard Worker                                                             static_cast<uint64_t>(shdr.sh_offset)));
420*eb293b8fSAndroid Build Coastguard Worker     } else if (shdr.sh_type == SHT_NOTE) {
421*eb293b8fSAndroid Build Coastguard Worker       if (shdr.sh_name < sec_size) {
422*eb293b8fSAndroid Build Coastguard Worker         std::string name;
423*eb293b8fSAndroid Build Coastguard Worker         if (memory_->ReadString(sec_offset + shdr.sh_name, &name, sec_size - shdr.sh_name) &&
424*eb293b8fSAndroid Build Coastguard Worker             name == ".note.gnu.build-id") {
425*eb293b8fSAndroid Build Coastguard Worker           gnu_build_id_offset_ = shdr.sh_offset;
426*eb293b8fSAndroid Build Coastguard Worker           gnu_build_id_size_ = shdr.sh_size;
427*eb293b8fSAndroid Build Coastguard Worker         }
428*eb293b8fSAndroid Build Coastguard Worker       }
429*eb293b8fSAndroid Build Coastguard Worker     }
430*eb293b8fSAndroid Build Coastguard Worker   }
431*eb293b8fSAndroid Build Coastguard Worker }
432*eb293b8fSAndroid Build Coastguard Worker 
433*eb293b8fSAndroid Build Coastguard Worker template <typename ElfTypes>
GetSoname()434*eb293b8fSAndroid Build Coastguard Worker std::string ElfInterfaceImpl<ElfTypes>::GetSoname() {
435*eb293b8fSAndroid Build Coastguard Worker   if (soname_type_ == SONAME_INVALID) {
436*eb293b8fSAndroid Build Coastguard Worker     return "";
437*eb293b8fSAndroid Build Coastguard Worker   }
438*eb293b8fSAndroid Build Coastguard Worker   if (soname_type_ == SONAME_VALID) {
439*eb293b8fSAndroid Build Coastguard Worker     return soname_;
440*eb293b8fSAndroid Build Coastguard Worker   }
441*eb293b8fSAndroid Build Coastguard Worker 
442*eb293b8fSAndroid Build Coastguard Worker   soname_type_ = SONAME_INVALID;
443*eb293b8fSAndroid Build Coastguard Worker 
444*eb293b8fSAndroid Build Coastguard Worker   uint64_t soname_offset = 0;
445*eb293b8fSAndroid Build Coastguard Worker   uint64_t strtab_addr = 0;
446*eb293b8fSAndroid Build Coastguard Worker   uint64_t strtab_size = 0;
447*eb293b8fSAndroid Build Coastguard Worker 
448*eb293b8fSAndroid Build Coastguard Worker   // Find the soname location from the dynamic headers section.
449*eb293b8fSAndroid Build Coastguard Worker   DynType dyn;
450*eb293b8fSAndroid Build Coastguard Worker   uint64_t offset = dynamic_offset_;
451*eb293b8fSAndroid Build Coastguard Worker   uint64_t max_offset = offset + dynamic_vaddr_end_ - dynamic_vaddr_start_;
452*eb293b8fSAndroid Build Coastguard Worker   for (uint64_t offset = dynamic_offset_; offset < max_offset; offset += sizeof(DynType)) {
453*eb293b8fSAndroid Build Coastguard Worker     if (!memory_->ReadFully(offset, &dyn, sizeof(dyn))) {
454*eb293b8fSAndroid Build Coastguard Worker       last_error_.code = ERROR_MEMORY_INVALID;
455*eb293b8fSAndroid Build Coastguard Worker       last_error_.address = offset;
456*eb293b8fSAndroid Build Coastguard Worker       return "";
457*eb293b8fSAndroid Build Coastguard Worker     }
458*eb293b8fSAndroid Build Coastguard Worker 
459*eb293b8fSAndroid Build Coastguard Worker     if (dyn.d_tag == DT_STRTAB) {
460*eb293b8fSAndroid Build Coastguard Worker       strtab_addr = dyn.d_un.d_ptr;
461*eb293b8fSAndroid Build Coastguard Worker     } else if (dyn.d_tag == DT_STRSZ) {
462*eb293b8fSAndroid Build Coastguard Worker       strtab_size = dyn.d_un.d_val;
463*eb293b8fSAndroid Build Coastguard Worker     } else if (dyn.d_tag == DT_SONAME) {
464*eb293b8fSAndroid Build Coastguard Worker       soname_offset = dyn.d_un.d_val;
465*eb293b8fSAndroid Build Coastguard Worker     } else if (dyn.d_tag == DT_NULL) {
466*eb293b8fSAndroid Build Coastguard Worker       break;
467*eb293b8fSAndroid Build Coastguard Worker     }
468*eb293b8fSAndroid Build Coastguard Worker   }
469*eb293b8fSAndroid Build Coastguard Worker 
470*eb293b8fSAndroid Build Coastguard Worker   // Need to map the strtab address to the real offset.
471*eb293b8fSAndroid Build Coastguard Worker   for (const auto& entry : strtabs_) {
472*eb293b8fSAndroid Build Coastguard Worker     if (entry.first == strtab_addr) {
473*eb293b8fSAndroid Build Coastguard Worker       soname_offset = entry.second + soname_offset;
474*eb293b8fSAndroid Build Coastguard Worker       uint64_t soname_max = entry.second + strtab_size;
475*eb293b8fSAndroid Build Coastguard Worker       if (soname_offset >= soname_max) {
476*eb293b8fSAndroid Build Coastguard Worker         return "";
477*eb293b8fSAndroid Build Coastguard Worker       }
478*eb293b8fSAndroid Build Coastguard Worker       if (!memory_->ReadString(soname_offset, &soname_, soname_max - soname_offset)) {
479*eb293b8fSAndroid Build Coastguard Worker         return "";
480*eb293b8fSAndroid Build Coastguard Worker       }
481*eb293b8fSAndroid Build Coastguard Worker       soname_type_ = SONAME_VALID;
482*eb293b8fSAndroid Build Coastguard Worker       return soname_;
483*eb293b8fSAndroid Build Coastguard Worker     }
484*eb293b8fSAndroid Build Coastguard Worker   }
485*eb293b8fSAndroid Build Coastguard Worker   return "";
486*eb293b8fSAndroid Build Coastguard Worker }
487*eb293b8fSAndroid Build Coastguard Worker 
488*eb293b8fSAndroid Build Coastguard Worker template <typename ElfTypes>
GetFunctionName(uint64_t addr,SharedString * name,uint64_t * func_offset)489*eb293b8fSAndroid Build Coastguard Worker bool ElfInterfaceImpl<ElfTypes>::GetFunctionName(uint64_t addr, SharedString* name,
490*eb293b8fSAndroid Build Coastguard Worker                                                  uint64_t* func_offset) {
491*eb293b8fSAndroid Build Coastguard Worker   if (symbols_.empty()) {
492*eb293b8fSAndroid Build Coastguard Worker     return false;
493*eb293b8fSAndroid Build Coastguard Worker   }
494*eb293b8fSAndroid Build Coastguard Worker 
495*eb293b8fSAndroid Build Coastguard Worker   for (const auto symbol : symbols_) {
496*eb293b8fSAndroid Build Coastguard Worker     if (symbol->template GetName<SymType>(addr, memory_.get(), name, func_offset)) {
497*eb293b8fSAndroid Build Coastguard Worker       return true;
498*eb293b8fSAndroid Build Coastguard Worker     }
499*eb293b8fSAndroid Build Coastguard Worker   }
500*eb293b8fSAndroid Build Coastguard Worker   return false;
501*eb293b8fSAndroid Build Coastguard Worker }
502*eb293b8fSAndroid Build Coastguard Worker 
503*eb293b8fSAndroid Build Coastguard Worker template <typename ElfTypes>
GetGlobalVariable(const std::string & name,uint64_t * memory_address)504*eb293b8fSAndroid Build Coastguard Worker bool ElfInterfaceImpl<ElfTypes>::GetGlobalVariable(const std::string& name,
505*eb293b8fSAndroid Build Coastguard Worker                                                    uint64_t* memory_address) {
506*eb293b8fSAndroid Build Coastguard Worker   if (symbols_.empty()) {
507*eb293b8fSAndroid Build Coastguard Worker     return false;
508*eb293b8fSAndroid Build Coastguard Worker   }
509*eb293b8fSAndroid Build Coastguard Worker 
510*eb293b8fSAndroid Build Coastguard Worker   for (const auto symbol : symbols_) {
511*eb293b8fSAndroid Build Coastguard Worker     if (symbol->template GetGlobal<SymType>(memory_.get(), name, memory_address)) {
512*eb293b8fSAndroid Build Coastguard Worker       return true;
513*eb293b8fSAndroid Build Coastguard Worker     }
514*eb293b8fSAndroid Build Coastguard Worker   }
515*eb293b8fSAndroid Build Coastguard Worker   return false;
516*eb293b8fSAndroid Build Coastguard Worker }
517*eb293b8fSAndroid Build Coastguard Worker 
Step(uint64_t pc,Regs * regs,Memory * process_memory,bool * finished,bool * is_signal_frame)518*eb293b8fSAndroid Build Coastguard Worker bool ElfInterface::Step(uint64_t pc, Regs* regs, Memory* process_memory, bool* finished,
519*eb293b8fSAndroid Build Coastguard Worker                         bool* is_signal_frame) {
520*eb293b8fSAndroid Build Coastguard Worker   last_error_.code = ERROR_NONE;
521*eb293b8fSAndroid Build Coastguard Worker   last_error_.address = 0;
522*eb293b8fSAndroid Build Coastguard Worker 
523*eb293b8fSAndroid Build Coastguard Worker   // Try the debug_frame first since it contains the most specific unwind
524*eb293b8fSAndroid Build Coastguard Worker   // information.
525*eb293b8fSAndroid Build Coastguard Worker   DwarfSection* debug_frame = debug_frame_.get();
526*eb293b8fSAndroid Build Coastguard Worker   if (debug_frame != nullptr &&
527*eb293b8fSAndroid Build Coastguard Worker       debug_frame->Step(pc, regs, process_memory, finished, is_signal_frame)) {
528*eb293b8fSAndroid Build Coastguard Worker     return true;
529*eb293b8fSAndroid Build Coastguard Worker   }
530*eb293b8fSAndroid Build Coastguard Worker 
531*eb293b8fSAndroid Build Coastguard Worker   // Try the eh_frame next.
532*eb293b8fSAndroid Build Coastguard Worker   DwarfSection* eh_frame = eh_frame_.get();
533*eb293b8fSAndroid Build Coastguard Worker   if (eh_frame != nullptr && eh_frame->Step(pc, regs, process_memory, finished, is_signal_frame)) {
534*eb293b8fSAndroid Build Coastguard Worker     return true;
535*eb293b8fSAndroid Build Coastguard Worker   }
536*eb293b8fSAndroid Build Coastguard Worker 
537*eb293b8fSAndroid Build Coastguard Worker   if (gnu_debugdata_interface_ != nullptr &&
538*eb293b8fSAndroid Build Coastguard Worker       gnu_debugdata_interface_->Step(pc, regs, process_memory, finished, is_signal_frame)) {
539*eb293b8fSAndroid Build Coastguard Worker     return true;
540*eb293b8fSAndroid Build Coastguard Worker   }
541*eb293b8fSAndroid Build Coastguard Worker 
542*eb293b8fSAndroid Build Coastguard Worker   // Set the error code based on the first error encountered.
543*eb293b8fSAndroid Build Coastguard Worker   DwarfSection* section = nullptr;
544*eb293b8fSAndroid Build Coastguard Worker   if (debug_frame_ != nullptr) {
545*eb293b8fSAndroid Build Coastguard Worker     section = debug_frame_.get();
546*eb293b8fSAndroid Build Coastguard Worker   } else if (eh_frame_ != nullptr) {
547*eb293b8fSAndroid Build Coastguard Worker     section = eh_frame_.get();
548*eb293b8fSAndroid Build Coastguard Worker   } else if (gnu_debugdata_interface_ != nullptr) {
549*eb293b8fSAndroid Build Coastguard Worker     last_error_ = gnu_debugdata_interface_->last_error();
550*eb293b8fSAndroid Build Coastguard Worker     return false;
551*eb293b8fSAndroid Build Coastguard Worker   } else {
552*eb293b8fSAndroid Build Coastguard Worker     return false;
553*eb293b8fSAndroid Build Coastguard Worker   }
554*eb293b8fSAndroid Build Coastguard Worker 
555*eb293b8fSAndroid Build Coastguard Worker   // Convert the DWARF ERROR to an external error.
556*eb293b8fSAndroid Build Coastguard Worker   DwarfErrorCode code = section->LastErrorCode();
557*eb293b8fSAndroid Build Coastguard Worker   switch (code) {
558*eb293b8fSAndroid Build Coastguard Worker     case DWARF_ERROR_NONE:
559*eb293b8fSAndroid Build Coastguard Worker       last_error_.code = ERROR_NONE;
560*eb293b8fSAndroid Build Coastguard Worker       break;
561*eb293b8fSAndroid Build Coastguard Worker 
562*eb293b8fSAndroid Build Coastguard Worker     case DWARF_ERROR_MEMORY_INVALID:
563*eb293b8fSAndroid Build Coastguard Worker       last_error_.code = ERROR_MEMORY_INVALID;
564*eb293b8fSAndroid Build Coastguard Worker       last_error_.address = section->LastErrorAddress();
565*eb293b8fSAndroid Build Coastguard Worker       break;
566*eb293b8fSAndroid Build Coastguard Worker 
567*eb293b8fSAndroid Build Coastguard Worker     case DWARF_ERROR_ILLEGAL_VALUE:
568*eb293b8fSAndroid Build Coastguard Worker     case DWARF_ERROR_ILLEGAL_STATE:
569*eb293b8fSAndroid Build Coastguard Worker     case DWARF_ERROR_STACK_INDEX_NOT_VALID:
570*eb293b8fSAndroid Build Coastguard Worker     case DWARF_ERROR_TOO_MANY_ITERATIONS:
571*eb293b8fSAndroid Build Coastguard Worker     case DWARF_ERROR_CFA_NOT_DEFINED:
572*eb293b8fSAndroid Build Coastguard Worker     case DWARF_ERROR_NO_FDES:
573*eb293b8fSAndroid Build Coastguard Worker       last_error_.code = ERROR_UNWIND_INFO;
574*eb293b8fSAndroid Build Coastguard Worker       break;
575*eb293b8fSAndroid Build Coastguard Worker 
576*eb293b8fSAndroid Build Coastguard Worker     case DWARF_ERROR_NOT_IMPLEMENTED:
577*eb293b8fSAndroid Build Coastguard Worker     case DWARF_ERROR_UNSUPPORTED_VERSION:
578*eb293b8fSAndroid Build Coastguard Worker       last_error_.code = ERROR_UNSUPPORTED;
579*eb293b8fSAndroid Build Coastguard Worker       break;
580*eb293b8fSAndroid Build Coastguard Worker   }
581*eb293b8fSAndroid Build Coastguard Worker   return false;
582*eb293b8fSAndroid Build Coastguard Worker }
583*eb293b8fSAndroid Build Coastguard Worker 
584*eb293b8fSAndroid Build Coastguard Worker // This is an estimation of the size of the elf file using the location
585*eb293b8fSAndroid Build Coastguard Worker // of the section headers and size. This assumes that the section headers
586*eb293b8fSAndroid Build Coastguard Worker // are at the end of the elf file. If the elf has a load bias, the size
587*eb293b8fSAndroid Build Coastguard Worker // will be too large, but this is acceptable.
588*eb293b8fSAndroid Build Coastguard Worker template <typename ElfTypes>
GetMaxSize(Memory * memory,uint64_t * size)589*eb293b8fSAndroid Build Coastguard Worker void ElfInterfaceImpl<ElfTypes>::GetMaxSize(Memory* memory, uint64_t* size) {
590*eb293b8fSAndroid Build Coastguard Worker   EhdrType ehdr;
591*eb293b8fSAndroid Build Coastguard Worker   if (!memory->ReadFully(0, &ehdr, sizeof(ehdr))) {
592*eb293b8fSAndroid Build Coastguard Worker     *size = 0;
593*eb293b8fSAndroid Build Coastguard Worker     return;
594*eb293b8fSAndroid Build Coastguard Worker   }
595*eb293b8fSAndroid Build Coastguard Worker 
596*eb293b8fSAndroid Build Coastguard Worker   // If this winds up as zero, the PT_LOAD reading will get a better value.
597*eb293b8fSAndroid Build Coastguard Worker   uint64_t elf_size = ehdr.e_shoff + ehdr.e_shentsize * ehdr.e_shnum;
598*eb293b8fSAndroid Build Coastguard Worker 
599*eb293b8fSAndroid Build Coastguard Worker   // Search through the PT_LOAD values and if any result in a larger elf
600*eb293b8fSAndroid Build Coastguard Worker   // size, use that.
601*eb293b8fSAndroid Build Coastguard Worker   uint64_t offset = ehdr.e_phoff;
602*eb293b8fSAndroid Build Coastguard Worker   for (size_t i = 0; i < ehdr.e_phnum; i++, offset += ehdr.e_phentsize) {
603*eb293b8fSAndroid Build Coastguard Worker     PhdrType phdr;
604*eb293b8fSAndroid Build Coastguard Worker     if (!memory->ReadFully(offset, &phdr, sizeof(phdr))) {
605*eb293b8fSAndroid Build Coastguard Worker       break;
606*eb293b8fSAndroid Build Coastguard Worker     }
607*eb293b8fSAndroid Build Coastguard Worker     if (phdr.p_type == PT_LOAD) {
608*eb293b8fSAndroid Build Coastguard Worker       uint64_t end_offset;
609*eb293b8fSAndroid Build Coastguard Worker       if (__builtin_add_overflow(phdr.p_offset, phdr.p_memsz, &end_offset)) {
610*eb293b8fSAndroid Build Coastguard Worker         continue;
611*eb293b8fSAndroid Build Coastguard Worker       }
612*eb293b8fSAndroid Build Coastguard Worker       if (end_offset > elf_size) {
613*eb293b8fSAndroid Build Coastguard Worker         elf_size = end_offset;
614*eb293b8fSAndroid Build Coastguard Worker       }
615*eb293b8fSAndroid Build Coastguard Worker     }
616*eb293b8fSAndroid Build Coastguard Worker   }
617*eb293b8fSAndroid Build Coastguard Worker 
618*eb293b8fSAndroid Build Coastguard Worker   *size = elf_size;
619*eb293b8fSAndroid Build Coastguard Worker }
620*eb293b8fSAndroid Build Coastguard Worker 
621*eb293b8fSAndroid Build Coastguard Worker template <typename EhdrType, typename ShdrType>
GetBuildIDInfo(Memory * memory,uint64_t * build_id_offset,uint64_t * build_id_size)622*eb293b8fSAndroid Build Coastguard Worker bool GetBuildIDInfo(Memory* memory, uint64_t* build_id_offset, uint64_t* build_id_size) {
623*eb293b8fSAndroid Build Coastguard Worker   EhdrType ehdr;
624*eb293b8fSAndroid Build Coastguard Worker   if (!memory->ReadFully(0, &ehdr, sizeof(ehdr))) {
625*eb293b8fSAndroid Build Coastguard Worker     return false;
626*eb293b8fSAndroid Build Coastguard Worker   }
627*eb293b8fSAndroid Build Coastguard Worker 
628*eb293b8fSAndroid Build Coastguard Worker   uint64_t offset = ehdr.e_shoff;
629*eb293b8fSAndroid Build Coastguard Worker   uint64_t sec_offset;
630*eb293b8fSAndroid Build Coastguard Worker   uint64_t sec_size;
631*eb293b8fSAndroid Build Coastguard Worker   ShdrType shdr;
632*eb293b8fSAndroid Build Coastguard Worker   if (ehdr.e_shstrndx >= ehdr.e_shnum) {
633*eb293b8fSAndroid Build Coastguard Worker     return false;
634*eb293b8fSAndroid Build Coastguard Worker   }
635*eb293b8fSAndroid Build Coastguard Worker 
636*eb293b8fSAndroid Build Coastguard Worker   uint64_t sh_offset = offset + ehdr.e_shstrndx * ehdr.e_shentsize;
637*eb293b8fSAndroid Build Coastguard Worker   if (!memory->ReadFully(sh_offset, &shdr, sizeof(shdr))) {
638*eb293b8fSAndroid Build Coastguard Worker     return false;
639*eb293b8fSAndroid Build Coastguard Worker   }
640*eb293b8fSAndroid Build Coastguard Worker   sec_offset = shdr.sh_offset;
641*eb293b8fSAndroid Build Coastguard Worker   sec_size = shdr.sh_size;
642*eb293b8fSAndroid Build Coastguard Worker 
643*eb293b8fSAndroid Build Coastguard Worker   // Skip the first header, it's always going to be NULL.
644*eb293b8fSAndroid Build Coastguard Worker   offset += ehdr.e_shentsize;
645*eb293b8fSAndroid Build Coastguard Worker   for (size_t i = 1; i < ehdr.e_shnum; i++, offset += ehdr.e_shentsize) {
646*eb293b8fSAndroid Build Coastguard Worker     if (!memory->ReadFully(offset, &shdr, sizeof(shdr))) {
647*eb293b8fSAndroid Build Coastguard Worker       return false;
648*eb293b8fSAndroid Build Coastguard Worker     }
649*eb293b8fSAndroid Build Coastguard Worker     std::string name;
650*eb293b8fSAndroid Build Coastguard Worker     if (shdr.sh_type == SHT_NOTE && shdr.sh_name < sec_size &&
651*eb293b8fSAndroid Build Coastguard Worker         memory->ReadString(sec_offset + shdr.sh_name, &name, sec_size - shdr.sh_name) &&
652*eb293b8fSAndroid Build Coastguard Worker         name == ".note.gnu.build-id") {
653*eb293b8fSAndroid Build Coastguard Worker       *build_id_offset = shdr.sh_offset;
654*eb293b8fSAndroid Build Coastguard Worker       *build_id_size = shdr.sh_size;
655*eb293b8fSAndroid Build Coastguard Worker       return true;
656*eb293b8fSAndroid Build Coastguard Worker     }
657*eb293b8fSAndroid Build Coastguard Worker   }
658*eb293b8fSAndroid Build Coastguard Worker 
659*eb293b8fSAndroid Build Coastguard Worker   return false;
660*eb293b8fSAndroid Build Coastguard Worker }
661*eb293b8fSAndroid Build Coastguard Worker 
662*eb293b8fSAndroid Build Coastguard Worker template <typename EhdrType, typename ShdrType, typename NhdrType>
ReadBuildIDFromMemory(Memory * memory)663*eb293b8fSAndroid Build Coastguard Worker std::string ElfInterface::ReadBuildIDFromMemory(Memory* memory) {
664*eb293b8fSAndroid Build Coastguard Worker   uint64_t note_offset;
665*eb293b8fSAndroid Build Coastguard Worker   uint64_t note_size;
666*eb293b8fSAndroid Build Coastguard Worker   if (!GetBuildIDInfo<EhdrType, ShdrType>(memory, &note_offset, &note_size)) {
667*eb293b8fSAndroid Build Coastguard Worker     return "";
668*eb293b8fSAndroid Build Coastguard Worker   }
669*eb293b8fSAndroid Build Coastguard Worker 
670*eb293b8fSAndroid Build Coastguard Worker   // Ensure there is no overflow in any of the calculations below.
671*eb293b8fSAndroid Build Coastguard Worker   uint64_t tmp;
672*eb293b8fSAndroid Build Coastguard Worker   if (__builtin_add_overflow(note_offset, note_size, &tmp)) {
673*eb293b8fSAndroid Build Coastguard Worker     return "";
674*eb293b8fSAndroid Build Coastguard Worker   }
675*eb293b8fSAndroid Build Coastguard Worker 
676*eb293b8fSAndroid Build Coastguard Worker   uint64_t offset = 0;
677*eb293b8fSAndroid Build Coastguard Worker   while (offset < note_size) {
678*eb293b8fSAndroid Build Coastguard Worker     if (note_size - offset < sizeof(NhdrType)) {
679*eb293b8fSAndroid Build Coastguard Worker       return "";
680*eb293b8fSAndroid Build Coastguard Worker     }
681*eb293b8fSAndroid Build Coastguard Worker     NhdrType hdr;
682*eb293b8fSAndroid Build Coastguard Worker     if (!memory->ReadFully(note_offset + offset, &hdr, sizeof(hdr))) {
683*eb293b8fSAndroid Build Coastguard Worker       return "";
684*eb293b8fSAndroid Build Coastguard Worker     }
685*eb293b8fSAndroid Build Coastguard Worker     offset += sizeof(hdr);
686*eb293b8fSAndroid Build Coastguard Worker 
687*eb293b8fSAndroid Build Coastguard Worker     if (note_size - offset < hdr.n_namesz) {
688*eb293b8fSAndroid Build Coastguard Worker       return "";
689*eb293b8fSAndroid Build Coastguard Worker     }
690*eb293b8fSAndroid Build Coastguard Worker     if (hdr.n_namesz > 0) {
691*eb293b8fSAndroid Build Coastguard Worker       std::string name(hdr.n_namesz, '\0');
692*eb293b8fSAndroid Build Coastguard Worker       if (!memory->ReadFully(note_offset + offset, &(name[0]), hdr.n_namesz)) {
693*eb293b8fSAndroid Build Coastguard Worker         return "";
694*eb293b8fSAndroid Build Coastguard Worker       }
695*eb293b8fSAndroid Build Coastguard Worker 
696*eb293b8fSAndroid Build Coastguard Worker       // Trim trailing \0 as GNU is stored as a C string in the ELF file.
697*eb293b8fSAndroid Build Coastguard Worker       if (name.back() == '\0') name.resize(name.size() - 1);
698*eb293b8fSAndroid Build Coastguard Worker 
699*eb293b8fSAndroid Build Coastguard Worker       // Align hdr.n_namesz to next power multiple of 4. See man 5 elf.
700*eb293b8fSAndroid Build Coastguard Worker       offset += (hdr.n_namesz + 3) & ~3;
701*eb293b8fSAndroid Build Coastguard Worker 
702*eb293b8fSAndroid Build Coastguard Worker       if (name == "GNU" && hdr.n_type == NT_GNU_BUILD_ID) {
703*eb293b8fSAndroid Build Coastguard Worker         if (note_size - offset < hdr.n_descsz || hdr.n_descsz == 0) {
704*eb293b8fSAndroid Build Coastguard Worker           return "";
705*eb293b8fSAndroid Build Coastguard Worker         }
706*eb293b8fSAndroid Build Coastguard Worker         std::string build_id(hdr.n_descsz, '\0');
707*eb293b8fSAndroid Build Coastguard Worker         if (memory->ReadFully(note_offset + offset, &build_id[0], hdr.n_descsz)) {
708*eb293b8fSAndroid Build Coastguard Worker           return build_id;
709*eb293b8fSAndroid Build Coastguard Worker         }
710*eb293b8fSAndroid Build Coastguard Worker         return "";
711*eb293b8fSAndroid Build Coastguard Worker       }
712*eb293b8fSAndroid Build Coastguard Worker     }
713*eb293b8fSAndroid Build Coastguard Worker     // Align hdr.n_descsz to next power multiple of 4. See man 5 elf.
714*eb293b8fSAndroid Build Coastguard Worker     offset += (hdr.n_descsz + 3) & ~3;
715*eb293b8fSAndroid Build Coastguard Worker   }
716*eb293b8fSAndroid Build Coastguard Worker   return "";
717*eb293b8fSAndroid Build Coastguard Worker }
718*eb293b8fSAndroid Build Coastguard Worker 
719*eb293b8fSAndroid Build Coastguard Worker // Instantiate all of the needed template functions.
720*eb293b8fSAndroid Build Coastguard Worker template class ElfInterfaceImpl<ElfTypes32>;
721*eb293b8fSAndroid Build Coastguard Worker template class ElfInterfaceImpl<ElfTypes64>;
722*eb293b8fSAndroid Build Coastguard Worker 
723*eb293b8fSAndroid Build Coastguard Worker template int64_t ElfInterface::GetLoadBias<Elf32_Ehdr, Elf32_Phdr>(Memory*);
724*eb293b8fSAndroid Build Coastguard Worker template int64_t ElfInterface::GetLoadBias<Elf64_Ehdr, Elf64_Phdr>(Memory*);
725*eb293b8fSAndroid Build Coastguard Worker 
726*eb293b8fSAndroid Build Coastguard Worker template std::string ElfInterface::ReadBuildIDFromMemory<Elf32_Ehdr, Elf32_Shdr, Elf32_Nhdr>(
727*eb293b8fSAndroid Build Coastguard Worker     Memory*);
728*eb293b8fSAndroid Build Coastguard Worker template std::string ElfInterface::ReadBuildIDFromMemory<Elf64_Ehdr, Elf64_Shdr, Elf64_Nhdr>(
729*eb293b8fSAndroid Build Coastguard Worker     Memory*);
730*eb293b8fSAndroid Build Coastguard Worker 
731*eb293b8fSAndroid Build Coastguard Worker }  // namespace unwindstack
732