• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (C) 2012 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #include "elf_file.h"
18 
19 #include <inttypes.h>
20 #include <sys/mman.h>  // For the PROT_* and MAP_* constants.
21 #include <sys/types.h>
22 #include <unistd.h>
23 
24 #include "android-base/stringprintf.h"
25 #include "android-base/strings.h"
26 
27 #include "arch/instruction_set.h"
28 #include "base/leb128.h"
29 #include "base/stl_util.h"
30 #include "base/unix_file/fd_file.h"
31 #include "base/utils.h"
32 #include "elf/elf_utils.h"
33 #include "elf_file_impl.h"
34 
35 namespace art {
36 
37 using android::base::StringPrintf;
38 
39 template <typename ElfTypes>
ElfFileImpl(File * file,bool writable,bool program_header_only)40 ElfFileImpl<ElfTypes>::ElfFileImpl(File* file, bool writable, bool program_header_only)
41   : writable_(writable),
42     program_header_only_(program_header_only),
43     header_(nullptr),
44     base_address_(nullptr),
45     program_headers_start_(nullptr),
46     section_headers_start_(nullptr),
47     dynamic_program_header_(nullptr),
48     dynamic_section_start_(nullptr),
49     symtab_section_start_(nullptr),
50     dynsym_section_start_(nullptr),
51     strtab_section_start_(nullptr),
52     dynstr_section_start_(nullptr),
53     hash_section_start_(nullptr),
54     symtab_symbol_table_(nullptr),
55     dynsym_symbol_table_(nullptr) {
56   CHECK(file != nullptr);
57 }
58 
59 template <typename ElfTypes>
Open(File * file,bool writable,bool program_header_only,bool low_4gb,std::string * error_msg)60 ElfFileImpl<ElfTypes>* ElfFileImpl<ElfTypes>::Open(File* file,
61                                                    bool writable,
62                                                    bool program_header_only,
63                                                    bool low_4gb,
64                                                    std::string* error_msg) {
65   std::unique_ptr<ElfFileImpl<ElfTypes>> elf_file(
66       new ElfFileImpl<ElfTypes>(file, writable, program_header_only));
67   int prot;
68   int flags;
69   if (writable) {
70     prot = PROT_READ | PROT_WRITE;
71     flags = MAP_SHARED;
72   } else {
73     prot = PROT_READ;
74     flags = MAP_PRIVATE;
75   }
76   if (!elf_file->Setup(file, prot, flags, low_4gb, error_msg)) {
77     return nullptr;
78   }
79   return elf_file.release();
80 }
81 
82 template <typename ElfTypes>
Open(File * file,int prot,int flags,bool low_4gb,std::string * error_msg)83 ElfFileImpl<ElfTypes>* ElfFileImpl<ElfTypes>::Open(File* file,
84                                                    int prot,
85                                                    int flags,
86                                                    bool low_4gb,
87                                                    std::string* error_msg) {
88   std::unique_ptr<ElfFileImpl<ElfTypes>> elf_file(
89       new ElfFileImpl<ElfTypes>(file, (prot & PROT_WRITE) != 0, /* program_header_only= */ false));
90   if (!elf_file->Setup(file, prot, flags, low_4gb, error_msg)) {
91     return nullptr;
92   }
93   return elf_file.release();
94 }
95 
96 template <typename ElfTypes>
Setup(File * file,int prot,int flags,bool low_4gb,std::string * error_msg)97 bool ElfFileImpl<ElfTypes>::Setup(File* file,
98                                   int prot,
99                                   int flags,
100                                   bool low_4gb,
101                                   std::string* error_msg) {
102   int64_t temp_file_length = file->GetLength();
103   if (temp_file_length < 0) {
104     errno = -temp_file_length;
105     *error_msg = StringPrintf("Failed to get length of file: '%s' fd=%d: %s",
106                               file->GetPath().c_str(), file->Fd(), strerror(errno));
107     return false;
108   }
109   size_t file_length = static_cast<size_t>(temp_file_length);
110   if (file_length < sizeof(Elf_Ehdr)) {
111     *error_msg = StringPrintf("File size of %zd bytes not large enough to contain ELF header of "
112                               "%zd bytes: '%s'", file_length, sizeof(Elf_Ehdr),
113                               file->GetPath().c_str());
114     return false;
115   }
116 
117   if (program_header_only_) {
118     // first just map ELF header to get program header size information
119     size_t elf_header_size = sizeof(Elf_Ehdr);
120     if (!SetMap(file,
121                 MemMap::MapFile(elf_header_size,
122                                 prot,
123                                 flags,
124                                 file->Fd(),
125                                 0,
126                                 low_4gb,
127                                 file->GetPath().c_str(),
128                                 error_msg),
129                 error_msg)) {
130       return false;
131     }
132     // then remap to cover program header
133     size_t program_header_size = header_->e_phoff + (header_->e_phentsize * header_->e_phnum);
134     if (file_length < program_header_size) {
135       *error_msg = StringPrintf("File size of %zd bytes not large enough to contain ELF program "
136                                 "header of %zd bytes: '%s'", file_length,
137                                 sizeof(Elf_Ehdr), file->GetPath().c_str());
138       return false;
139     }
140     if (!SetMap(file,
141                 MemMap::MapFile(program_header_size,
142                                 prot,
143                                 flags,
144                                 file->Fd(),
145                                 0,
146                                 low_4gb,
147                                 file->GetPath().c_str(),
148                                 error_msg),
149                 error_msg)) {
150       *error_msg = StringPrintf("Failed to map ELF program headers: %s", error_msg->c_str());
151       return false;
152     }
153   } else {
154     // otherwise map entire file
155     if (!SetMap(file,
156                 MemMap::MapFile(file->GetLength(),
157                                 prot,
158                                 flags,
159                                 file->Fd(),
160                                 0,
161                                 low_4gb,
162                                 file->GetPath().c_str(),
163                                 error_msg),
164                 error_msg)) {
165       *error_msg = StringPrintf("Failed to map ELF file: %s", error_msg->c_str());
166       return false;
167     }
168   }
169 
170   if (program_header_only_) {
171     program_headers_start_ = Begin() + GetHeader().e_phoff;
172   } else {
173     if (!CheckAndSet(GetHeader().e_phoff, "program headers", &program_headers_start_, error_msg)) {
174       return false;
175     }
176 
177     // Setup section headers.
178     if (!CheckAndSet(GetHeader().e_shoff, "section headers", &section_headers_start_, error_msg)) {
179       return false;
180     }
181 
182     // Find shstrtab.
183     Elf_Shdr* shstrtab_section_header = GetSectionNameStringSection();
184     if (shstrtab_section_header == nullptr) {
185       *error_msg = StringPrintf("Failed to find shstrtab section header in ELF file: '%s'",
186                                 file->GetPath().c_str());
187       return false;
188     }
189 
190     // Find .dynamic section info from program header
191     dynamic_program_header_ = FindProgamHeaderByType(PT_DYNAMIC);
192     if (dynamic_program_header_ == nullptr) {
193       *error_msg = StringPrintf("Failed to find PT_DYNAMIC program header in ELF file: '%s'",
194                                 file->GetPath().c_str());
195       return false;
196     }
197 
198     if (!CheckAndSet(GetDynamicProgramHeader().p_offset, "dynamic section",
199                      reinterpret_cast<uint8_t**>(&dynamic_section_start_), error_msg)) {
200       return false;
201     }
202 
203     // Find other sections from section headers
204     for (Elf_Word i = 0; i < GetSectionHeaderNum(); i++) {
205       Elf_Shdr* section_header = GetSectionHeader(i);
206       if (section_header == nullptr) {
207         *error_msg = StringPrintf("Failed to find section header for section %d in ELF file: '%s'",
208                                   i, file->GetPath().c_str());
209         return false;
210       }
211       switch (section_header->sh_type) {
212         case SHT_SYMTAB: {
213           if (!CheckAndSet(section_header->sh_offset, "symtab",
214                            reinterpret_cast<uint8_t**>(&symtab_section_start_), error_msg)) {
215             return false;
216           }
217           break;
218         }
219         case SHT_DYNSYM: {
220           if (!CheckAndSet(section_header->sh_offset, "dynsym",
221                            reinterpret_cast<uint8_t**>(&dynsym_section_start_), error_msg)) {
222             return false;
223           }
224           break;
225         }
226         case SHT_STRTAB: {
227           // TODO: base these off of sh_link from .symtab and .dynsym above
228           if ((section_header->sh_flags & SHF_ALLOC) != 0) {
229             // Check that this is named ".dynstr" and ignore otherwise.
230             const char* header_name = GetString(*shstrtab_section_header, section_header->sh_name);
231             if (strncmp(".dynstr", header_name, 8) == 0) {
232               if (!CheckAndSet(section_header->sh_offset, "dynstr",
233                                reinterpret_cast<uint8_t**>(&dynstr_section_start_), error_msg)) {
234                 return false;
235               }
236             }
237           } else {
238             // Check that this is named ".strtab" and ignore otherwise.
239             const char* header_name = GetString(*shstrtab_section_header, section_header->sh_name);
240             if (strncmp(".strtab", header_name, 8) == 0) {
241               if (!CheckAndSet(section_header->sh_offset, "strtab",
242                                reinterpret_cast<uint8_t**>(&strtab_section_start_), error_msg)) {
243                 return false;
244               }
245             }
246           }
247           break;
248         }
249         case SHT_DYNAMIC: {
250           if (reinterpret_cast<uint8_t*>(dynamic_section_start_) !=
251               Begin() + section_header->sh_offset) {
252             LOG(WARNING) << "Failed to find matching SHT_DYNAMIC for PT_DYNAMIC in "
253                          << file->GetPath() << ": " << std::hex
254                          << reinterpret_cast<void*>(dynamic_section_start_)
255                          << " != " << reinterpret_cast<void*>(Begin() + section_header->sh_offset);
256             return false;
257           }
258           break;
259         }
260         case SHT_HASH: {
261           if (!CheckAndSet(section_header->sh_offset, "hash section",
262                            reinterpret_cast<uint8_t**>(&hash_section_start_), error_msg)) {
263             return false;
264           }
265           break;
266         }
267       }
268     }
269 
270     // Check for the existence of some sections.
271     if (!CheckSectionsExist(file, error_msg)) {
272       return false;
273     }
274   }
275 
276   return true;
277 }
278 
279 template <typename ElfTypes>
~ElfFileImpl()280 ElfFileImpl<ElfTypes>::~ElfFileImpl() {
281   delete symtab_symbol_table_;
282   delete dynsym_symbol_table_;
283 }
284 
285 template <typename ElfTypes>
CheckAndSet(Elf32_Off offset,const char * label,uint8_t ** target,std::string * error_msg)286 bool ElfFileImpl<ElfTypes>::CheckAndSet(Elf32_Off offset, const char* label,
287                                         uint8_t** target, std::string* error_msg) {
288   if (Begin() + offset >= End()) {
289     *error_msg = StringPrintf("Offset %d is out of range for %s in ELF file: '%s'", offset, label,
290                               file_path_.c_str());
291     return false;
292   }
293   *target = Begin() + offset;
294   return true;
295 }
296 
297 template <typename ElfTypes>
CheckSectionsLinked(const uint8_t * source,const uint8_t * target) const298 bool ElfFileImpl<ElfTypes>::CheckSectionsLinked(const uint8_t* source,
299                                                 const uint8_t* target) const {
300   // Only works in whole-program mode, as we need to iterate over the sections.
301   // Note that we normally can't search by type, as duplicates are allowed for most section types.
302   if (program_header_only_) {
303     return true;
304   }
305 
306   Elf_Shdr* source_section = nullptr;
307   Elf_Word target_index = 0;
308   bool target_found = false;
309   for (Elf_Word i = 0; i < GetSectionHeaderNum(); i++) {
310     Elf_Shdr* section_header = GetSectionHeader(i);
311 
312     if (Begin() + section_header->sh_offset == source) {
313       // Found the source.
314       source_section = section_header;
315       if (target_index) {
316         break;
317       }
318     } else if (Begin() + section_header->sh_offset == target) {
319       target_index = i;
320       target_found = true;
321       if (source_section != nullptr) {
322         break;
323       }
324     }
325   }
326 
327   return target_found && source_section != nullptr && source_section->sh_link == target_index;
328 }
329 
330 template <typename ElfTypes>
CheckSectionsExist(File * file,std::string * error_msg) const331   bool ElfFileImpl<ElfTypes>::CheckSectionsExist(File* file, std::string* error_msg) const {
332   if (!program_header_only_) {
333     // If in full mode, need section headers.
334     if (section_headers_start_ == nullptr) {
335       *error_msg = StringPrintf("No section headers in ELF file: '%s'", file->GetPath().c_str());
336       return false;
337     }
338   }
339 
340   // This is redundant, but defensive.
341   if (dynamic_program_header_ == nullptr) {
342     *error_msg = StringPrintf("Failed to find PT_DYNAMIC program header in ELF file: '%s'",
343                               file->GetPath().c_str());
344     return false;
345   }
346 
347   // Need a dynamic section. This is redundant, but defensive.
348   if (dynamic_section_start_ == nullptr) {
349     *error_msg = StringPrintf("Failed to find dynamic section in ELF file: '%s'",
350                               file->GetPath().c_str());
351     return false;
352   }
353 
354   // Symtab validation. These is not really a hard failure, as we are currently not using the
355   // symtab internally, but it's nice to be defensive.
356   if (symtab_section_start_ != nullptr) {
357     // When there's a symtab, there should be a strtab.
358     if (strtab_section_start_ == nullptr) {
359       *error_msg = StringPrintf("No strtab for symtab in ELF file: '%s'", file->GetPath().c_str());
360       return false;
361     }
362 
363     // The symtab should link to the strtab.
364     if (!CheckSectionsLinked(reinterpret_cast<const uint8_t*>(symtab_section_start_),
365                              reinterpret_cast<const uint8_t*>(strtab_section_start_))) {
366       *error_msg = StringPrintf("Symtab is not linked to the strtab in ELF file: '%s'",
367                                 file->GetPath().c_str());
368       return false;
369     }
370   }
371 
372   // We always need a dynstr & dynsym.
373   if (dynstr_section_start_ == nullptr) {
374     *error_msg = StringPrintf("No dynstr in ELF file: '%s'", file->GetPath().c_str());
375     return false;
376   }
377   if (dynsym_section_start_ == nullptr) {
378     *error_msg = StringPrintf("No dynsym in ELF file: '%s'", file->GetPath().c_str());
379     return false;
380   }
381 
382   // Need a hash section for dynamic symbol lookup.
383   if (hash_section_start_ == nullptr) {
384     *error_msg = StringPrintf("Failed to find hash section in ELF file: '%s'",
385                               file->GetPath().c_str());
386     return false;
387   }
388 
389   // And the hash section should be linking to the dynsym.
390   if (!CheckSectionsLinked(reinterpret_cast<const uint8_t*>(hash_section_start_),
391                            reinterpret_cast<const uint8_t*>(dynsym_section_start_))) {
392     *error_msg = StringPrintf("Hash section is not linked to the dynstr in ELF file: '%s'",
393                               file->GetPath().c_str());
394     return false;
395   }
396 
397   // We'd also like to confirm a shstrtab in program_header_only_ mode (else Open() does this for
398   // us). This is usually the last in an oat file, and a good indicator of whether writing was
399   // successful (or the process crashed and left garbage).
400   if (program_header_only_) {
401     // It might not be mapped, but we can compare against the file size.
402     int64_t offset = static_cast<int64_t>(GetHeader().e_shoff +
403                                           (GetHeader().e_shstrndx * GetHeader().e_shentsize));
404     if (offset >= file->GetLength()) {
405       *error_msg = StringPrintf("Shstrtab is not in the mapped ELF file: '%s'",
406                                 file->GetPath().c_str());
407       return false;
408     }
409   }
410 
411   return true;
412 }
413 
414 template <typename ElfTypes>
SetMap(File * file,MemMap && map,std::string * error_msg)415 bool ElfFileImpl<ElfTypes>::SetMap(File* file, MemMap&& map, std::string* error_msg) {
416   if (!map.IsValid()) {
417     // MemMap::Open should have already set an error.
418     DCHECK(!error_msg->empty());
419     return false;
420   }
421   map_ = std::move(map);
422   CHECK(map_.IsValid()) << file->GetPath();
423   CHECK(map_.Begin() != nullptr) << file->GetPath();
424 
425   header_ = reinterpret_cast<Elf_Ehdr*>(map_.Begin());
426   if ((ELFMAG0 != header_->e_ident[EI_MAG0])
427       || (ELFMAG1 != header_->e_ident[EI_MAG1])
428       || (ELFMAG2 != header_->e_ident[EI_MAG2])
429       || (ELFMAG3 != header_->e_ident[EI_MAG3])) {
430     *error_msg = StringPrintf("Failed to find ELF magic value %d %d %d %d in %s, found %d %d %d %d",
431                               ELFMAG0, ELFMAG1, ELFMAG2, ELFMAG3,
432                               file->GetPath().c_str(),
433                               header_->e_ident[EI_MAG0],
434                               header_->e_ident[EI_MAG1],
435                               header_->e_ident[EI_MAG2],
436                               header_->e_ident[EI_MAG3]);
437     return false;
438   }
439   uint8_t elf_class = (sizeof(Elf_Addr) == sizeof(Elf64_Addr)) ? ELFCLASS64 : ELFCLASS32;
440   if (elf_class != header_->e_ident[EI_CLASS]) {
441     *error_msg = StringPrintf("Failed to find expected EI_CLASS value %d in %s, found %d",
442                               elf_class,
443                               file->GetPath().c_str(),
444                               header_->e_ident[EI_CLASS]);
445     return false;
446   }
447   if (ELFDATA2LSB != header_->e_ident[EI_DATA]) {
448     *error_msg = StringPrintf("Failed to find expected EI_DATA value %d in %s, found %d",
449                               ELFDATA2LSB,
450                               file->GetPath().c_str(),
451                               header_->e_ident[EI_CLASS]);
452     return false;
453   }
454   if (EV_CURRENT != header_->e_ident[EI_VERSION]) {
455     *error_msg = StringPrintf("Failed to find expected EI_VERSION value %d in %s, found %d",
456                               EV_CURRENT,
457                               file->GetPath().c_str(),
458                               header_->e_ident[EI_CLASS]);
459     return false;
460   }
461   if (ET_DYN != header_->e_type) {
462     *error_msg = StringPrintf("Failed to find expected e_type value %d in %s, found %d",
463                               ET_DYN,
464                               file->GetPath().c_str(),
465                               header_->e_type);
466     return false;
467   }
468   if (EV_CURRENT != header_->e_version) {
469     *error_msg = StringPrintf("Failed to find expected e_version value %d in %s, found %d",
470                               EV_CURRENT,
471                               file->GetPath().c_str(),
472                               header_->e_version);
473     return false;
474   }
475   if (0 != header_->e_entry) {
476     *error_msg = StringPrintf("Failed to find expected e_entry value %d in %s, found %d",
477                               0,
478                               file->GetPath().c_str(),
479                               static_cast<int32_t>(header_->e_entry));
480     return false;
481   }
482   if (0 == header_->e_phoff) {
483     *error_msg = StringPrintf("Failed to find non-zero e_phoff value in %s",
484                               file->GetPath().c_str());
485     return false;
486   }
487   if (0 == header_->e_shoff) {
488     *error_msg = StringPrintf("Failed to find non-zero e_shoff value in %s",
489                               file->GetPath().c_str());
490     return false;
491   }
492   if (0 == header_->e_ehsize) {
493     *error_msg = StringPrintf("Failed to find non-zero e_ehsize value in %s",
494                               file->GetPath().c_str());
495     return false;
496   }
497   if (0 == header_->e_phentsize) {
498     *error_msg = StringPrintf("Failed to find non-zero e_phentsize value in %s",
499                               file->GetPath().c_str());
500     return false;
501   }
502   if (0 == header_->e_phnum) {
503     *error_msg = StringPrintf("Failed to find non-zero e_phnum value in %s",
504                               file->GetPath().c_str());
505     return false;
506   }
507   if (0 == header_->e_shentsize) {
508     *error_msg = StringPrintf("Failed to find non-zero e_shentsize value in %s",
509                               file->GetPath().c_str());
510     return false;
511   }
512   if (0 == header_->e_shnum) {
513     *error_msg = StringPrintf("Failed to find non-zero e_shnum value in %s",
514                               file->GetPath().c_str());
515     return false;
516   }
517   if (0 == header_->e_shstrndx) {
518     *error_msg = StringPrintf("Failed to find non-zero e_shstrndx value in %s",
519                               file->GetPath().c_str());
520     return false;
521   }
522   if (header_->e_shstrndx >= header_->e_shnum) {
523     *error_msg = StringPrintf("Failed to find e_shnum value %d less than %d in %s",
524                               header_->e_shstrndx,
525                               header_->e_shnum,
526                               file->GetPath().c_str());
527     return false;
528   }
529 
530   if (!program_header_only_) {
531     if (header_->e_phoff >= Size()) {
532       *error_msg = StringPrintf("Failed to find e_phoff value %" PRIu64 " less than %zd in %s",
533                                 static_cast<uint64_t>(header_->e_phoff),
534                                 Size(),
535                                 file->GetPath().c_str());
536       return false;
537     }
538     if (header_->e_shoff >= Size()) {
539       *error_msg = StringPrintf("Failed to find e_shoff value %" PRIu64 " less than %zd in %s",
540                                 static_cast<uint64_t>(header_->e_shoff),
541                                 Size(),
542                                 file->GetPath().c_str());
543       return false;
544     }
545   }
546   return true;
547 }
548 
549 template <typename ElfTypes>
GetHeader() const550 typename ElfTypes::Ehdr& ElfFileImpl<ElfTypes>::GetHeader() const {
551   CHECK(header_ != nullptr);  // Header has been checked in SetMap. This is a sanity check.
552   return *header_;
553 }
554 
555 template <typename ElfTypes>
GetProgramHeadersStart() const556 uint8_t* ElfFileImpl<ElfTypes>::GetProgramHeadersStart() const {
557   CHECK(program_headers_start_ != nullptr);  // Header has been set in Setup. This is a sanity
558                                              // check.
559   return program_headers_start_;
560 }
561 
562 template <typename ElfTypes>
GetSectionHeadersStart() const563 uint8_t* ElfFileImpl<ElfTypes>::GetSectionHeadersStart() const {
564   CHECK(!program_header_only_);              // Only used in "full" mode.
565   CHECK(section_headers_start_ != nullptr);  // Is checked in CheckSectionsExist. Sanity check.
566   return section_headers_start_;
567 }
568 
569 template <typename ElfTypes>
GetDynamicProgramHeader() const570 typename ElfTypes::Phdr& ElfFileImpl<ElfTypes>::GetDynamicProgramHeader() const {
571   CHECK(dynamic_program_header_ != nullptr);  // Is checked in CheckSectionsExist. Sanity check.
572   return *dynamic_program_header_;
573 }
574 
575 template <typename ElfTypes>
GetDynamicSectionStart() const576 typename ElfTypes::Dyn* ElfFileImpl<ElfTypes>::GetDynamicSectionStart() const {
577   CHECK(dynamic_section_start_ != nullptr);  // Is checked in CheckSectionsExist. Sanity check.
578   return dynamic_section_start_;
579 }
580 
581 template <typename ElfTypes>
GetSymbolSectionStart(Elf_Word section_type) const582 typename ElfTypes::Sym* ElfFileImpl<ElfTypes>::GetSymbolSectionStart(
583     Elf_Word section_type) const {
584   CHECK(IsSymbolSectionType(section_type)) << file_path_ << " " << section_type;
585   switch (section_type) {
586     case SHT_SYMTAB: {
587       return symtab_section_start_;
588       break;
589     }
590     case SHT_DYNSYM: {
591       return dynsym_section_start_;
592       break;
593     }
594     default: {
595       LOG(FATAL) << section_type;
596       return nullptr;
597     }
598   }
599 }
600 
601 template <typename ElfTypes>
GetStringSectionStart(Elf_Word section_type) const602 const char* ElfFileImpl<ElfTypes>::GetStringSectionStart(
603     Elf_Word section_type) const {
604   CHECK(IsSymbolSectionType(section_type)) << file_path_ << " " << section_type;
605   switch (section_type) {
606     case SHT_SYMTAB: {
607       return strtab_section_start_;
608     }
609     case SHT_DYNSYM: {
610       return dynstr_section_start_;
611     }
612     default: {
613       LOG(FATAL) << section_type;
614       return nullptr;
615     }
616   }
617 }
618 
619 template <typename ElfTypes>
GetString(Elf_Word section_type,Elf_Word i) const620 const char* ElfFileImpl<ElfTypes>::GetString(Elf_Word section_type,
621                                              Elf_Word i) const {
622   CHECK(IsSymbolSectionType(section_type)) << file_path_ << " " << section_type;
623   if (i == 0) {
624     return nullptr;
625   }
626   const char* string_section_start = GetStringSectionStart(section_type);
627   if (string_section_start == nullptr) {
628     return nullptr;
629   }
630   return string_section_start + i;
631 }
632 
633 // WARNING: The following methods do not check for an error condition (non-existent hash section).
634 //          It is the caller's job to do this.
635 
636 template <typename ElfTypes>
GetHashSectionStart() const637 typename ElfTypes::Word* ElfFileImpl<ElfTypes>::GetHashSectionStart() const {
638   return hash_section_start_;
639 }
640 
641 template <typename ElfTypes>
GetHashBucketNum() const642 typename ElfTypes::Word ElfFileImpl<ElfTypes>::GetHashBucketNum() const {
643   return GetHashSectionStart()[0];
644 }
645 
646 template <typename ElfTypes>
GetHashChainNum() const647 typename ElfTypes::Word ElfFileImpl<ElfTypes>::GetHashChainNum() const {
648   return GetHashSectionStart()[1];
649 }
650 
651 template <typename ElfTypes>
GetHashBucket(size_t i,bool * ok) const652 typename ElfTypes::Word ElfFileImpl<ElfTypes>::GetHashBucket(size_t i, bool* ok) const {
653   if (i >= GetHashBucketNum()) {
654     *ok = false;
655     return 0;
656   }
657   *ok = true;
658   // 0 is nbucket, 1 is nchain
659   return GetHashSectionStart()[2 + i];
660 }
661 
662 template <typename ElfTypes>
GetHashChain(size_t i,bool * ok) const663 typename ElfTypes::Word ElfFileImpl<ElfTypes>::GetHashChain(size_t i, bool* ok) const {
664   if (i >= GetHashChainNum()) {
665     *ok = false;
666     return 0;
667   }
668   *ok = true;
669   // 0 is nbucket, 1 is nchain, & chains are after buckets
670   return GetHashSectionStart()[2 + GetHashBucketNum() + i];
671 }
672 
673 template <typename ElfTypes>
GetProgramHeaderNum() const674 typename ElfTypes::Word ElfFileImpl<ElfTypes>::GetProgramHeaderNum() const {
675   return GetHeader().e_phnum;
676 }
677 
678 template <typename ElfTypes>
GetProgramHeader(Elf_Word i) const679 typename ElfTypes::Phdr* ElfFileImpl<ElfTypes>::GetProgramHeader(Elf_Word i) const {
680   CHECK_LT(i, GetProgramHeaderNum()) << file_path_;  // Sanity check for caller.
681   uint8_t* program_header = GetProgramHeadersStart() + (i * GetHeader().e_phentsize);
682   CHECK_LT(program_header, End());
683   return reinterpret_cast<Elf_Phdr*>(program_header);
684 }
685 
686 template <typename ElfTypes>
FindProgamHeaderByType(Elf_Word type) const687 typename ElfTypes::Phdr* ElfFileImpl<ElfTypes>::FindProgamHeaderByType(Elf_Word type) const {
688   for (Elf_Word i = 0; i < GetProgramHeaderNum(); i++) {
689     Elf_Phdr* program_header = GetProgramHeader(i);
690     if (program_header->p_type == type) {
691       return program_header;
692     }
693   }
694   return nullptr;
695 }
696 
697 template <typename ElfTypes>
GetSectionHeaderNum() const698 typename ElfTypes::Word ElfFileImpl<ElfTypes>::GetSectionHeaderNum() const {
699   return GetHeader().e_shnum;
700 }
701 
702 template <typename ElfTypes>
GetSectionHeader(Elf_Word i) const703 typename ElfTypes::Shdr* ElfFileImpl<ElfTypes>::GetSectionHeader(Elf_Word i) const {
704   // Can only access arbitrary sections when we have the whole file, not just program header.
705   // Even if we Load(), it doesn't bring in all the sections.
706   CHECK(!program_header_only_) << file_path_;
707   if (i >= GetSectionHeaderNum()) {
708     return nullptr;  // Failure condition.
709   }
710   uint8_t* section_header = GetSectionHeadersStart() + (i * GetHeader().e_shentsize);
711   if (section_header >= End()) {
712     return nullptr;  // Failure condition.
713   }
714   return reinterpret_cast<Elf_Shdr*>(section_header);
715 }
716 
717 template <typename ElfTypes>
FindSectionByType(Elf_Word type) const718 typename ElfTypes::Shdr* ElfFileImpl<ElfTypes>::FindSectionByType(Elf_Word type) const {
719   // Can only access arbitrary sections when we have the whole file, not just program header.
720   // We could change this to switch on known types if they were detected during loading.
721   CHECK(!program_header_only_) << file_path_;
722   for (Elf_Word i = 0; i < GetSectionHeaderNum(); i++) {
723     Elf_Shdr* section_header = GetSectionHeader(i);
724     if (section_header->sh_type == type) {
725       return section_header;
726     }
727   }
728   return nullptr;
729 }
730 
731 // from bionic
elfhash(const char * _name)732 static unsigned elfhash(const char *_name) {
733   const unsigned char *name = (const unsigned char *) _name;
734   unsigned h = 0, g;
735 
736   while (*name) {
737     h = (h << 4) + *name++;
738     g = h & 0xf0000000;
739     h ^= g;
740     h ^= g >> 24;
741   }
742   return h;
743 }
744 
745 template <typename ElfTypes>
GetSectionNameStringSection() const746 typename ElfTypes::Shdr* ElfFileImpl<ElfTypes>::GetSectionNameStringSection() const {
747   return GetSectionHeader(GetHeader().e_shstrndx);
748 }
749 
750 template <typename ElfTypes>
FindDynamicSymbolAddress(const std::string & symbol_name) const751 const uint8_t* ElfFileImpl<ElfTypes>::FindDynamicSymbolAddress(
752     const std::string& symbol_name) const {
753   // Check that we have a hash section.
754   if (GetHashSectionStart() == nullptr) {
755     return nullptr;  // Failure condition.
756   }
757   const Elf_Sym* sym = FindDynamicSymbol(symbol_name);
758   if (sym != nullptr) {
759     // TODO: we need to change this to calculate base_address_ in ::Open,
760     // otherwise it will be wrongly 0 if ::Load has not yet been called.
761     return base_address_ + sym->st_value;
762   } else {
763     return nullptr;
764   }
765 }
766 
767 // WARNING: Only called from FindDynamicSymbolAddress. Elides check for hash section.
768 template <typename ElfTypes>
FindDynamicSymbol(const std::string & symbol_name) const769 const typename ElfTypes::Sym* ElfFileImpl<ElfTypes>::FindDynamicSymbol(
770     const std::string& symbol_name) const {
771   if (GetHashBucketNum() == 0) {
772     // No dynamic symbols at all.
773     return nullptr;
774   }
775   Elf_Word hash = elfhash(symbol_name.c_str());
776   Elf_Word bucket_index = hash % GetHashBucketNum();
777   bool ok;
778   Elf_Word symbol_and_chain_index = GetHashBucket(bucket_index, &ok);
779   if (!ok) {
780     return nullptr;
781   }
782   while (symbol_and_chain_index != 0 /* STN_UNDEF */) {
783     Elf_Sym* symbol = GetSymbol(SHT_DYNSYM, symbol_and_chain_index);
784     if (symbol == nullptr) {
785       return nullptr;  // Failure condition.
786     }
787     const char* name = GetString(SHT_DYNSYM, symbol->st_name);
788     if (symbol_name == name) {
789       return symbol;
790     }
791     symbol_and_chain_index = GetHashChain(symbol_and_chain_index, &ok);
792     if (!ok) {
793       return nullptr;
794     }
795   }
796   return nullptr;
797 }
798 
799 template <typename ElfTypes>
IsSymbolSectionType(Elf_Word section_type)800 bool ElfFileImpl<ElfTypes>::IsSymbolSectionType(Elf_Word section_type) {
801   return ((section_type == SHT_SYMTAB) || (section_type == SHT_DYNSYM));
802 }
803 
804 template <typename ElfTypes>
GetSymbolNum(Elf_Shdr & section_header) const805 typename ElfTypes::Word ElfFileImpl<ElfTypes>::GetSymbolNum(Elf_Shdr& section_header) const {
806   CHECK(IsSymbolSectionType(section_header.sh_type))
807       << file_path_ << " " << section_header.sh_type;
808   CHECK_NE(0U, section_header.sh_entsize) << file_path_;
809   return section_header.sh_size / section_header.sh_entsize;
810 }
811 
812 template <typename ElfTypes>
GetSymbol(Elf_Word section_type,Elf_Word i) const813 typename ElfTypes::Sym* ElfFileImpl<ElfTypes>::GetSymbol(Elf_Word section_type, Elf_Word i) const {
814   Elf_Sym* sym_start = GetSymbolSectionStart(section_type);
815   if (sym_start == nullptr) {
816     return nullptr;
817   }
818   return sym_start + i;
819 }
820 
821 template <typename ElfTypes>
822 typename ElfFileImpl<ElfTypes>::SymbolTable**
GetSymbolTable(Elf_Word section_type)823 ElfFileImpl<ElfTypes>::GetSymbolTable(Elf_Word section_type) {
824   CHECK(IsSymbolSectionType(section_type)) << file_path_ << " " << section_type;
825   switch (section_type) {
826     case SHT_SYMTAB: {
827       return &symtab_symbol_table_;
828     }
829     case SHT_DYNSYM: {
830       return &dynsym_symbol_table_;
831     }
832     default: {
833       LOG(FATAL) << section_type;
834       return nullptr;
835     }
836   }
837 }
838 
839 template <typename ElfTypes>
FindSymbolByName(Elf_Word section_type,const std::string & symbol_name,bool build_map)840 typename ElfTypes::Sym* ElfFileImpl<ElfTypes>::FindSymbolByName(
841     Elf_Word section_type, const std::string& symbol_name, bool build_map) {
842   CHECK(!program_header_only_) << file_path_;
843   CHECK(IsSymbolSectionType(section_type)) << file_path_ << " " << section_type;
844 
845   SymbolTable** symbol_table = GetSymbolTable(section_type);
846   if (*symbol_table != nullptr || build_map) {
847     if (*symbol_table == nullptr) {
848       DCHECK(build_map);
849       *symbol_table = new SymbolTable;
850       Elf_Shdr* symbol_section = FindSectionByType(section_type);
851       if (symbol_section == nullptr) {
852         return nullptr;  // Failure condition.
853       }
854       Elf_Shdr* string_section = GetSectionHeader(symbol_section->sh_link);
855       if (string_section == nullptr) {
856         return nullptr;  // Failure condition.
857       }
858       for (uint32_t i = 0; i < GetSymbolNum(*symbol_section); i++) {
859         Elf_Sym* symbol = GetSymbol(section_type, i);
860         if (symbol == nullptr) {
861           return nullptr;  // Failure condition.
862         }
863         unsigned char type = (sizeof(Elf_Addr) == sizeof(Elf64_Addr))
864                              ? ELF64_ST_TYPE(symbol->st_info)
865                              : ELF32_ST_TYPE(symbol->st_info);
866         if (type == STT_NOTYPE) {
867           continue;
868         }
869         const char* name = GetString(*string_section, symbol->st_name);
870         if (name == nullptr) {
871           continue;
872         }
873         std::pair<typename SymbolTable::iterator, bool> result =
874             (*symbol_table)->insert(std::make_pair(name, symbol));
875         if (!result.second) {
876           // If a duplicate, make sure it has the same logical value. Seen on x86.
877           if ((symbol->st_value != result.first->second->st_value) ||
878               (symbol->st_size != result.first->second->st_size) ||
879               (symbol->st_info != result.first->second->st_info) ||
880               (symbol->st_other != result.first->second->st_other) ||
881               (symbol->st_shndx != result.first->second->st_shndx)) {
882             return nullptr;  // Failure condition.
883           }
884         }
885       }
886     }
887     CHECK(*symbol_table != nullptr);
888     typename SymbolTable::const_iterator it = (*symbol_table)->find(symbol_name);
889     if (it == (*symbol_table)->end()) {
890       return nullptr;
891     }
892     return it->second;
893   }
894 
895   // Fall back to linear search
896   Elf_Shdr* symbol_section = FindSectionByType(section_type);
897   if (symbol_section == nullptr) {
898     return nullptr;
899   }
900   Elf_Shdr* string_section = GetSectionHeader(symbol_section->sh_link);
901   if (string_section == nullptr) {
902     return nullptr;
903   }
904   for (uint32_t i = 0; i < GetSymbolNum(*symbol_section); i++) {
905     Elf_Sym* symbol = GetSymbol(section_type, i);
906     if (symbol == nullptr) {
907       return nullptr;  // Failure condition.
908     }
909     const char* name = GetString(*string_section, symbol->st_name);
910     if (name == nullptr) {
911       continue;
912     }
913     if (symbol_name == name) {
914       return symbol;
915     }
916   }
917   return nullptr;
918 }
919 
920 template <typename ElfTypes>
FindSymbolAddress(Elf_Word section_type,const std::string & symbol_name,bool build_map)921 typename ElfTypes::Addr ElfFileImpl<ElfTypes>::FindSymbolAddress(
922     Elf_Word section_type, const std::string& symbol_name, bool build_map) {
923   Elf_Sym* symbol = FindSymbolByName(section_type, symbol_name, build_map);
924   if (symbol == nullptr) {
925     return 0;
926   }
927   return symbol->st_value;
928 }
929 
930 template <typename ElfTypes>
GetString(Elf_Shdr & string_section,Elf_Word i) const931 const char* ElfFileImpl<ElfTypes>::GetString(Elf_Shdr& string_section,
932                                              Elf_Word i) const {
933   CHECK(!program_header_only_) << file_path_;
934   // TODO: remove this static_cast from enum when using -std=gnu++0x
935   if (static_cast<Elf_Word>(SHT_STRTAB) != string_section.sh_type) {
936     return nullptr;  // Failure condition.
937   }
938   if (i >= string_section.sh_size) {
939     return nullptr;
940   }
941   if (i == 0) {
942     return nullptr;
943   }
944   uint8_t* strings = Begin() + string_section.sh_offset;
945   uint8_t* string = strings + i;
946   if (string >= End()) {
947     return nullptr;
948   }
949   return reinterpret_cast<const char*>(string);
950 }
951 
952 template <typename ElfTypes>
GetDynamicNum() const953 typename ElfTypes::Word ElfFileImpl<ElfTypes>::GetDynamicNum() const {
954   return GetDynamicProgramHeader().p_filesz / sizeof(Elf_Dyn);
955 }
956 
957 template <typename ElfTypes>
GetDynamic(Elf_Word i) const958 typename ElfTypes::Dyn& ElfFileImpl<ElfTypes>::GetDynamic(Elf_Word i) const {
959   CHECK_LT(i, GetDynamicNum()) << file_path_;
960   return *(GetDynamicSectionStart() + i);
961 }
962 
963 template <typename ElfTypes>
FindDynamicByType(Elf_Sword type) const964 typename ElfTypes::Dyn* ElfFileImpl<ElfTypes>::FindDynamicByType(Elf_Sword type) const {
965   for (Elf_Word i = 0; i < GetDynamicNum(); i++) {
966     Elf_Dyn* dyn = &GetDynamic(i);
967     if (dyn->d_tag == type) {
968       return dyn;
969     }
970   }
971   return nullptr;
972 }
973 
974 template <typename ElfTypes>
FindDynamicValueByType(Elf_Sword type) const975 typename ElfTypes::Word ElfFileImpl<ElfTypes>::FindDynamicValueByType(Elf_Sword type) const {
976   Elf_Dyn* dyn = FindDynamicByType(type);
977   if (dyn == nullptr) {
978     return 0;
979   } else {
980     return dyn->d_un.d_val;
981   }
982 }
983 
984 template <typename ElfTypes>
GetRelSectionStart(Elf_Shdr & section_header) const985 typename ElfTypes::Rel* ElfFileImpl<ElfTypes>::GetRelSectionStart(Elf_Shdr& section_header) const {
986   CHECK(SHT_REL == section_header.sh_type) << file_path_ << " " << section_header.sh_type;
987   return reinterpret_cast<Elf_Rel*>(Begin() + section_header.sh_offset);
988 }
989 
990 template <typename ElfTypes>
GetRelNum(Elf_Shdr & section_header) const991 typename ElfTypes::Word ElfFileImpl<ElfTypes>::GetRelNum(Elf_Shdr& section_header) const {
992   CHECK(SHT_REL == section_header.sh_type) << file_path_ << " " << section_header.sh_type;
993   CHECK_NE(0U, section_header.sh_entsize) << file_path_;
994   return section_header.sh_size / section_header.sh_entsize;
995 }
996 
997 template <typename ElfTypes>
GetRel(Elf_Shdr & section_header,Elf_Word i) const998 typename ElfTypes::Rel& ElfFileImpl<ElfTypes>::GetRel(Elf_Shdr& section_header, Elf_Word i) const {
999   CHECK(SHT_REL == section_header.sh_type) << file_path_ << " " << section_header.sh_type;
1000   CHECK_LT(i, GetRelNum(section_header)) << file_path_;
1001   return *(GetRelSectionStart(section_header) + i);
1002 }
1003 
1004 template <typename ElfTypes>
GetRelaSectionStart(Elf_Shdr & section_header) const1005 typename ElfTypes::Rela* ElfFileImpl<ElfTypes>::GetRelaSectionStart(Elf_Shdr& section_header) const {
1006   CHECK(SHT_RELA == section_header.sh_type) << file_path_ << " " << section_header.sh_type;
1007   return reinterpret_cast<Elf_Rela*>(Begin() + section_header.sh_offset);
1008 }
1009 
1010 template <typename ElfTypes>
GetRelaNum(Elf_Shdr & section_header) const1011 typename ElfTypes::Word ElfFileImpl<ElfTypes>::GetRelaNum(Elf_Shdr& section_header) const {
1012   CHECK(SHT_RELA == section_header.sh_type) << file_path_ << " " << section_header.sh_type;
1013   return section_header.sh_size / section_header.sh_entsize;
1014 }
1015 
1016 template <typename ElfTypes>
GetRela(Elf_Shdr & section_header,Elf_Word i) const1017 typename ElfTypes::Rela& ElfFileImpl<ElfTypes>::GetRela(Elf_Shdr& section_header, Elf_Word i) const {
1018   CHECK(SHT_RELA == section_header.sh_type) << file_path_ << " " << section_header.sh_type;
1019   CHECK_LT(i, GetRelaNum(section_header)) << file_path_;
1020   return *(GetRelaSectionStart(section_header) + i);
1021 }
1022 
1023 template <typename ElfTypes>
GetLoadedSize(size_t * size,std::string * error_msg) const1024 bool ElfFileImpl<ElfTypes>::GetLoadedSize(size_t* size, std::string* error_msg) const {
1025   uint8_t* vaddr_begin;
1026   return GetLoadedAddressRange(&vaddr_begin, size, error_msg);
1027 }
1028 
1029 // Base on bionic phdr_table_get_load_size
1030 template <typename ElfTypes>
GetLoadedAddressRange(uint8_t ** vaddr_begin,size_t * vaddr_size,std::string * error_msg) const1031 bool ElfFileImpl<ElfTypes>::GetLoadedAddressRange(/*out*/uint8_t** vaddr_begin,
1032                                                   /*out*/size_t* vaddr_size,
1033                                                   /*out*/std::string* error_msg) const {
1034   Elf_Addr min_vaddr = static_cast<Elf_Addr>(-1);
1035   Elf_Addr max_vaddr = 0u;
1036   for (Elf_Word i = 0; i < GetProgramHeaderNum(); i++) {
1037     Elf_Phdr* program_header = GetProgramHeader(i);
1038     if (program_header->p_type != PT_LOAD) {
1039       continue;
1040     }
1041     Elf_Addr begin_vaddr = program_header->p_vaddr;
1042     if (begin_vaddr < min_vaddr) {
1043        min_vaddr = begin_vaddr;
1044     }
1045     Elf_Addr end_vaddr = program_header->p_vaddr + program_header->p_memsz;
1046     if (UNLIKELY(begin_vaddr > end_vaddr)) {
1047       std::ostringstream oss;
1048       oss << "Program header #" << i << " has overflow in p_vaddr+p_memsz: 0x" << std::hex
1049           << program_header->p_vaddr << "+0x" << program_header->p_memsz << "=0x" << end_vaddr
1050           << " in ELF file \"" << file_path_ << "\"";
1051       *error_msg = oss.str();
1052       *vaddr_begin = nullptr;
1053       *vaddr_size = static_cast<size_t>(-1);
1054       return false;
1055     }
1056     if (end_vaddr > max_vaddr) {
1057       max_vaddr = end_vaddr;
1058     }
1059   }
1060   min_vaddr = RoundDown(min_vaddr, kPageSize);
1061   max_vaddr = RoundUp(max_vaddr, kPageSize);
1062   CHECK_LT(min_vaddr, max_vaddr) << file_path_;
1063   // Check that the range fits into the runtime address space.
1064   if (UNLIKELY(max_vaddr - 1u > std::numeric_limits<size_t>::max())) {
1065     std::ostringstream oss;
1066     oss << "Loaded range is 0x" << std::hex << min_vaddr << "-0x" << max_vaddr
1067         << " but maximum size_t is 0x" << std::numeric_limits<size_t>::max()
1068         << " for ELF file \"" << file_path_ << "\"";
1069     *error_msg = oss.str();
1070     *vaddr_begin = nullptr;
1071     *vaddr_size = static_cast<size_t>(-1);
1072     return false;
1073   }
1074   *vaddr_begin = reinterpret_cast<uint8_t*>(min_vaddr);
1075   *vaddr_size = dchecked_integral_cast<size_t>(max_vaddr - min_vaddr);
1076   return true;
1077 }
1078 
GetInstructionSetFromELF(uint16_t e_machine,uint32_t e_flags)1079 static InstructionSet GetInstructionSetFromELF(uint16_t e_machine, uint32_t e_flags) {
1080   switch (e_machine) {
1081     case EM_ARM:
1082       return InstructionSet::kArm;
1083     case EM_AARCH64:
1084       return InstructionSet::kArm64;
1085     case EM_386:
1086       return InstructionSet::kX86;
1087     case EM_X86_64:
1088       return InstructionSet::kX86_64;
1089     case EM_MIPS: {
1090       if ((e_flags & EF_MIPS_ARCH) == EF_MIPS_ARCH_32R2 ||
1091           (e_flags & EF_MIPS_ARCH) == EF_MIPS_ARCH_32R6) {
1092         return InstructionSet::kMips;
1093       } else if ((e_flags & EF_MIPS_ARCH) == EF_MIPS_ARCH_64R6) {
1094         return InstructionSet::kMips64;
1095       }
1096       break;
1097     }
1098   }
1099   return InstructionSet::kNone;
1100 }
1101 
1102 template <typename ElfTypes>
Load(File * file,bool executable,bool low_4gb,MemMap * reservation,std::string * error_msg)1103 bool ElfFileImpl<ElfTypes>::Load(File* file,
1104                                  bool executable,
1105                                  bool low_4gb,
1106                                  /*inout*/MemMap* reservation,
1107                                  /*out*/std::string* error_msg) {
1108   CHECK(program_header_only_) << file->GetPath();
1109 
1110   if (executable) {
1111     InstructionSet elf_ISA = GetInstructionSetFromELF(GetHeader().e_machine, GetHeader().e_flags);
1112     if (elf_ISA != kRuntimeISA) {
1113       std::ostringstream oss;
1114       oss << "Expected ISA " << kRuntimeISA << " but found " << elf_ISA;
1115       *error_msg = oss.str();
1116       return false;
1117     }
1118   }
1119 
1120   bool reserved = false;
1121   for (Elf_Word i = 0; i < GetProgramHeaderNum(); i++) {
1122     Elf_Phdr* program_header = GetProgramHeader(i);
1123 
1124     // Record .dynamic header information for later use
1125     if (program_header->p_type == PT_DYNAMIC) {
1126       dynamic_program_header_ = program_header;
1127       continue;
1128     }
1129 
1130     // Not something to load, move on.
1131     if (program_header->p_type != PT_LOAD) {
1132       continue;
1133     }
1134 
1135     // Found something to load.
1136 
1137     // Before load the actual segments, reserve a contiguous chunk
1138     // of required size and address for all segments, but with no
1139     // permissions. We'll then carve that up with the proper
1140     // permissions as we load the actual segments. If p_vaddr is
1141     // non-zero, the segments require the specific address specified,
1142     // which either was specified in the file because we already set
1143     // base_address_ after the first zero segment).
1144     int64_t temp_file_length = file->GetLength();
1145     if (temp_file_length < 0) {
1146       errno = -temp_file_length;
1147       *error_msg = StringPrintf("Failed to get length of file: '%s' fd=%d: %s",
1148                                 file->GetPath().c_str(), file->Fd(), strerror(errno));
1149       return false;
1150     }
1151     size_t file_length = static_cast<size_t>(temp_file_length);
1152     if (!reserved) {
1153       uint8_t* vaddr_begin;
1154       size_t vaddr_size;
1155       if (!GetLoadedAddressRange(&vaddr_begin, &vaddr_size, error_msg)) {
1156         DCHECK(!error_msg->empty());
1157         return false;
1158       }
1159       std::string reservation_name = "ElfFile reservation for " + file->GetPath();
1160       MemMap local_reservation = MemMap::MapAnonymous(
1161           reservation_name.c_str(),
1162           (reservation != nullptr) ? reservation->Begin() : nullptr,
1163           vaddr_size,
1164           PROT_NONE,
1165           low_4gb,
1166           /* reuse= */ false,
1167           reservation,
1168           error_msg);
1169       if (!local_reservation.IsValid()) {
1170         *error_msg = StringPrintf("Failed to allocate %s: %s",
1171                                   reservation_name.c_str(),
1172                                   error_msg->c_str());
1173         return false;
1174       }
1175       reserved = true;
1176 
1177       // Base address is the difference of actual mapped location and the vaddr_begin.
1178       base_address_ = reinterpret_cast<uint8_t*>(
1179           static_cast<uintptr_t>(local_reservation.Begin() - vaddr_begin));
1180       // By adding the p_vaddr of a section/symbol to base_address_ we will always get the
1181       // dynamic memory address of where that object is actually mapped
1182       //
1183       // TODO: base_address_ needs to be calculated in ::Open, otherwise
1184       // FindDynamicSymbolAddress returns the wrong values until Load is called.
1185       segments_.push_back(std::move(local_reservation));
1186     }
1187     // empty segment, nothing to map
1188     if (program_header->p_memsz == 0) {
1189       continue;
1190     }
1191     uint8_t* p_vaddr = base_address_ + program_header->p_vaddr;
1192     int prot = 0;
1193     if (executable && ((program_header->p_flags & PF_X) != 0)) {
1194       prot |= PROT_EXEC;
1195     }
1196     if ((program_header->p_flags & PF_W) != 0) {
1197       prot |= PROT_WRITE;
1198     }
1199     if ((program_header->p_flags & PF_R) != 0) {
1200       prot |= PROT_READ;
1201     }
1202     int flags = 0;
1203     if (writable_) {
1204       prot |= PROT_WRITE;
1205       flags |= MAP_SHARED;
1206     } else {
1207       flags |= MAP_PRIVATE;
1208     }
1209     if (program_header->p_filesz > program_header->p_memsz) {
1210       *error_msg = StringPrintf("Invalid p_filesz > p_memsz (%" PRIu64 " > %" PRIu64 "): %s",
1211                                 static_cast<uint64_t>(program_header->p_filesz),
1212                                 static_cast<uint64_t>(program_header->p_memsz),
1213                                 file->GetPath().c_str());
1214       return false;
1215     }
1216     if (program_header->p_filesz < program_header->p_memsz &&
1217         !IsAligned<kPageSize>(program_header->p_filesz)) {
1218       *error_msg = StringPrintf("Unsupported unaligned p_filesz < p_memsz (%" PRIu64
1219                                 " < %" PRIu64 "): %s",
1220                                 static_cast<uint64_t>(program_header->p_filesz),
1221                                 static_cast<uint64_t>(program_header->p_memsz),
1222                                 file->GetPath().c_str());
1223       return false;
1224     }
1225     if (file_length < (program_header->p_offset + program_header->p_filesz)) {
1226       *error_msg = StringPrintf("File size of %zd bytes not large enough to contain ELF segment "
1227                                 "%d of %" PRIu64 " bytes: '%s'", file_length, i,
1228                                 static_cast<uint64_t>(program_header->p_offset + program_header->p_filesz),
1229                                 file->GetPath().c_str());
1230       return false;
1231     }
1232     if (program_header->p_filesz != 0u) {
1233       MemMap segment =
1234           MemMap::MapFileAtAddress(p_vaddr,
1235                                    program_header->p_filesz,
1236                                    prot,
1237                                    flags,
1238                                    file->Fd(),
1239                                    program_header->p_offset,
1240                                    /* low_4gb= */ false,
1241                                    file->GetPath().c_str(),
1242                                    /* reuse= */ true,  // implies MAP_FIXED
1243                                    /* reservation= */ nullptr,
1244                                    error_msg);
1245       if (!segment.IsValid()) {
1246         *error_msg = StringPrintf("Failed to map ELF file segment %d from %s: %s",
1247                                   i, file->GetPath().c_str(), error_msg->c_str());
1248         return false;
1249       }
1250       if (segment.Begin() != p_vaddr) {
1251         *error_msg = StringPrintf("Failed to map ELF file segment %d from %s at expected address %p, "
1252                                   "instead mapped to %p",
1253                                   i, file->GetPath().c_str(), p_vaddr, segment.Begin());
1254         return false;
1255       }
1256       segments_.push_back(std::move(segment));
1257     }
1258     if (program_header->p_filesz < program_header->p_memsz) {
1259       std::string name = StringPrintf("Zero-initialized segment %" PRIu64 " of ELF file %s",
1260                                       static_cast<uint64_t>(i), file->GetPath().c_str());
1261       MemMap segment = MemMap::MapAnonymous(name.c_str(),
1262                                             p_vaddr + program_header->p_filesz,
1263                                             program_header->p_memsz - program_header->p_filesz,
1264                                             prot,
1265                                             /* low_4gb= */ false,
1266                                             /* reuse= */ true,
1267                                             /* reservation= */ nullptr,
1268                                             error_msg);
1269       if (!segment.IsValid()) {
1270         *error_msg = StringPrintf("Failed to map zero-initialized ELF file segment %d from %s: %s",
1271                                   i, file->GetPath().c_str(), error_msg->c_str());
1272         return false;
1273       }
1274       if (segment.Begin() != p_vaddr) {
1275         *error_msg = StringPrintf("Failed to map zero-initialized ELF file segment %d from %s "
1276                                   "at expected address %p, instead mapped to %p",
1277                                   i, file->GetPath().c_str(), p_vaddr, segment.Begin());
1278         return false;
1279       }
1280       segments_.push_back(std::move(segment));
1281     }
1282   }
1283 
1284   // Now that we are done loading, .dynamic should be in memory to find .dynstr, .dynsym, .hash
1285   uint8_t* dsptr = base_address_ + GetDynamicProgramHeader().p_vaddr;
1286   if ((dsptr < Begin() || dsptr >= End()) && !ValidPointer(dsptr)) {
1287     *error_msg = StringPrintf("dynamic section address invalid in ELF file %s",
1288                               file->GetPath().c_str());
1289     return false;
1290   }
1291   dynamic_section_start_ = reinterpret_cast<Elf_Dyn*>(dsptr);
1292 
1293   for (Elf_Word i = 0; i < GetDynamicNum(); i++) {
1294     Elf_Dyn& elf_dyn = GetDynamic(i);
1295     uint8_t* d_ptr = base_address_ + elf_dyn.d_un.d_ptr;
1296     switch (elf_dyn.d_tag) {
1297       case DT_HASH: {
1298         if (!ValidPointer(d_ptr)) {
1299           *error_msg = StringPrintf("DT_HASH value %p does not refer to a loaded ELF segment of %s",
1300                                     d_ptr, file->GetPath().c_str());
1301           return false;
1302         }
1303         hash_section_start_ = reinterpret_cast<Elf_Word*>(d_ptr);
1304         break;
1305       }
1306       case DT_STRTAB: {
1307         if (!ValidPointer(d_ptr)) {
1308           *error_msg = StringPrintf("DT_HASH value %p does not refer to a loaded ELF segment of %s",
1309                                     d_ptr, file->GetPath().c_str());
1310           return false;
1311         }
1312         dynstr_section_start_ = reinterpret_cast<char*>(d_ptr);
1313         break;
1314       }
1315       case DT_SYMTAB: {
1316         if (!ValidPointer(d_ptr)) {
1317           *error_msg = StringPrintf("DT_HASH value %p does not refer to a loaded ELF segment of %s",
1318                                     d_ptr, file->GetPath().c_str());
1319           return false;
1320         }
1321         dynsym_section_start_ = reinterpret_cast<Elf_Sym*>(d_ptr);
1322         break;
1323       }
1324       case DT_NULL: {
1325         if (GetDynamicNum() != i+1) {
1326           *error_msg = StringPrintf("DT_NULL found after %d .dynamic entries, "
1327                                     "expected %d as implied by size of PT_DYNAMIC segment in %s",
1328                                     i + 1, GetDynamicNum(), file->GetPath().c_str());
1329           return false;
1330         }
1331         break;
1332       }
1333     }
1334   }
1335 
1336   // Check for the existence of some sections.
1337   if (!CheckSectionsExist(file, error_msg)) {
1338     return false;
1339   }
1340 
1341   return true;
1342 }
1343 
1344 template <typename ElfTypes>
ValidPointer(const uint8_t * start) const1345 bool ElfFileImpl<ElfTypes>::ValidPointer(const uint8_t* start) const {
1346   for (const MemMap& segment : segments_) {
1347     if (segment.Begin() <= start && start < segment.End()) {
1348       return true;
1349     }
1350   }
1351   return false;
1352 }
1353 
1354 
1355 template <typename ElfTypes>
FindSectionByName(const std::string & name) const1356 typename ElfTypes::Shdr* ElfFileImpl<ElfTypes>::FindSectionByName(
1357     const std::string& name) const {
1358   CHECK(!program_header_only_);
1359   Elf_Shdr* shstrtab_sec = GetSectionNameStringSection();
1360   if (shstrtab_sec == nullptr) {
1361     return nullptr;
1362   }
1363   for (uint32_t i = 0; i < GetSectionHeaderNum(); i++) {
1364     Elf_Shdr* shdr = GetSectionHeader(i);
1365     if (shdr == nullptr) {
1366       return nullptr;
1367     }
1368     const char* sec_name = GetString(*shstrtab_sec, shdr->sh_name);
1369     if (sec_name == nullptr) {
1370       continue;
1371     }
1372     if (name == sec_name) {
1373       return shdr;
1374     }
1375   }
1376   return nullptr;
1377 }
1378 
1379 template <typename ElfTypes>
FixupDebugSections(Elf_Addr base_address_delta)1380 bool ElfFileImpl<ElfTypes>::FixupDebugSections(Elf_Addr base_address_delta) {
1381   if (base_address_delta == 0) {
1382     return true;
1383   }
1384   return ApplyOatPatchesTo(".debug_frame", base_address_delta) &&
1385          ApplyOatPatchesTo(".debug_info", base_address_delta) &&
1386          ApplyOatPatchesTo(".debug_line", base_address_delta);
1387 }
1388 
1389 template <typename ElfTypes>
ApplyOatPatchesTo(const char * target_section_name,Elf_Addr delta)1390 bool ElfFileImpl<ElfTypes>::ApplyOatPatchesTo(
1391     const char* target_section_name, Elf_Addr delta) {
1392   auto target_section = FindSectionByName(target_section_name);
1393   if (target_section == nullptr) {
1394     return true;
1395   }
1396   std::string patches_name = target_section_name + std::string(".oat_patches");
1397   auto patches_section = FindSectionByName(patches_name.c_str());
1398   if (patches_section == nullptr) {
1399     LOG(ERROR) << patches_name << " section not found.";
1400     return false;
1401   }
1402   if (patches_section->sh_type != SHT_OAT_PATCH) {
1403     LOG(ERROR) << "Unexpected type of " << patches_name;
1404     return false;
1405   }
1406   ApplyOatPatches(
1407       Begin() + patches_section->sh_offset,
1408       Begin() + patches_section->sh_offset + patches_section->sh_size,
1409       delta,
1410       Begin() + target_section->sh_offset,
1411       Begin() + target_section->sh_offset + target_section->sh_size);
1412   return true;
1413 }
1414 
1415 // Apply LEB128 encoded patches to given section.
1416 template <typename ElfTypes>
ApplyOatPatches(const uint8_t * patches,const uint8_t * patches_end,Elf_Addr delta,uint8_t * to_patch,const uint8_t * to_patch_end)1417 void ElfFileImpl<ElfTypes>::ApplyOatPatches(
1418     const uint8_t* patches, const uint8_t* patches_end, Elf_Addr delta,
1419     uint8_t* to_patch, const uint8_t* to_patch_end) {
1420   using UnalignedAddress __attribute__((__aligned__(1))) = Elf_Addr;
1421   while (patches < patches_end) {
1422     to_patch += DecodeUnsignedLeb128(&patches);
1423     DCHECK_LE(patches, patches_end) << "Unexpected end of patch list.";
1424     DCHECK_LT(to_patch, to_patch_end) << "Patch past the end of section.";
1425     *reinterpret_cast<UnalignedAddress*>(to_patch) += delta;
1426   }
1427 }
1428 
1429 template <typename ElfTypes>
Strip(File * file,std::string * error_msg)1430 bool ElfFileImpl<ElfTypes>::Strip(File* file, std::string* error_msg) {
1431   // ELF files produced by MCLinker look roughly like this
1432   //
1433   // +------------+
1434   // | Elf_Ehdr   | contains number of Elf_Shdr and offset to first
1435   // +------------+
1436   // | Elf_Phdr   | program headers
1437   // | Elf_Phdr   |
1438   // | ...        |
1439   // | Elf_Phdr   |
1440   // +------------+
1441   // | section    | mixture of needed and unneeded sections
1442   // +------------+
1443   // | section    |
1444   // +------------+
1445   // | ...        |
1446   // +------------+
1447   // | section    |
1448   // +------------+
1449   // | Elf_Shdr   | section headers
1450   // | Elf_Shdr   |
1451   // | ...        | contains offset to section start
1452   // | Elf_Shdr   |
1453   // +------------+
1454   //
1455   // To strip:
1456   // - leave the Elf_Ehdr and Elf_Phdr values in place.
1457   // - walk the sections making a new set of Elf_Shdr section headers for what we want to keep
1458   // - move the sections are keeping up to fill in gaps of sections we want to strip
1459   // - write new Elf_Shdr section headers to end of file, updating Elf_Ehdr
1460   // - truncate rest of file
1461   //
1462 
1463   std::vector<Elf_Shdr> section_headers;
1464   std::vector<Elf_Word> section_headers_original_indexes;
1465   section_headers.reserve(GetSectionHeaderNum());
1466 
1467 
1468   Elf_Shdr* string_section = GetSectionNameStringSection();
1469   CHECK(string_section != nullptr);
1470   for (Elf_Word i = 0; i < GetSectionHeaderNum(); i++) {
1471     Elf_Shdr* sh = GetSectionHeader(i);
1472     CHECK(sh != nullptr);
1473     const char* name = GetString(*string_section, sh->sh_name);
1474     if (name == nullptr) {
1475       CHECK_EQ(0U, i);
1476       section_headers.push_back(*sh);
1477       section_headers_original_indexes.push_back(0);
1478       continue;
1479     }
1480     if (android::base::StartsWith(name, ".debug")
1481         || (strcmp(name, ".strtab") == 0)
1482         || (strcmp(name, ".symtab") == 0)) {
1483       continue;
1484     }
1485     section_headers.push_back(*sh);
1486     section_headers_original_indexes.push_back(i);
1487   }
1488   CHECK_NE(0U, section_headers.size());
1489   CHECK_EQ(section_headers.size(), section_headers_original_indexes.size());
1490 
1491   // section 0 is the null section, sections start at offset of first section
1492   CHECK(GetSectionHeader(1) != nullptr);
1493   Elf_Off offset = GetSectionHeader(1)->sh_offset;
1494   for (size_t i = 1; i < section_headers.size(); i++) {
1495     Elf_Shdr& new_sh = section_headers[i];
1496     Elf_Shdr* old_sh = GetSectionHeader(section_headers_original_indexes[i]);
1497     CHECK(old_sh != nullptr);
1498     CHECK_EQ(new_sh.sh_name, old_sh->sh_name);
1499     if (old_sh->sh_addralign > 1) {
1500       offset = RoundUp(offset, old_sh->sh_addralign);
1501     }
1502     if (old_sh->sh_offset == offset) {
1503       // already in place
1504       offset += old_sh->sh_size;
1505       continue;
1506     }
1507     // shift section earlier
1508     memmove(Begin() + offset,
1509             Begin() + old_sh->sh_offset,
1510             old_sh->sh_size);
1511     new_sh.sh_offset = offset;
1512     offset += old_sh->sh_size;
1513   }
1514 
1515   Elf_Off shoff = offset;
1516   size_t section_headers_size_in_bytes = section_headers.size() * sizeof(Elf_Shdr);
1517   memcpy(Begin() + offset, &section_headers[0], section_headers_size_in_bytes);
1518   offset += section_headers_size_in_bytes;
1519 
1520   GetHeader().e_shnum = section_headers.size();
1521   GetHeader().e_shoff = shoff;
1522   int result = ftruncate(file->Fd(), offset);
1523   if (result != 0) {
1524     *error_msg = StringPrintf("Failed to truncate while stripping ELF file: '%s': %s",
1525                               file->GetPath().c_str(), strerror(errno));
1526     return false;
1527   }
1528   return true;
1529 }
1530 
1531 static const bool DEBUG_FIXUP = false;
1532 
1533 template <typename ElfTypes>
Fixup(Elf_Addr base_address)1534 bool ElfFileImpl<ElfTypes>::Fixup(Elf_Addr base_address) {
1535   if (!FixupDynamic(base_address)) {
1536     LOG(WARNING) << "Failed to fixup .dynamic in " << file_path_;
1537     return false;
1538   }
1539   if (!FixupSectionHeaders(base_address)) {
1540     LOG(WARNING) << "Failed to fixup section headers in " << file_path_;
1541     return false;
1542   }
1543   if (!FixupProgramHeaders(base_address)) {
1544     LOG(WARNING) << "Failed to fixup program headers in " << file_path_;
1545     return false;
1546   }
1547   if (!FixupSymbols(base_address, true)) {
1548     LOG(WARNING) << "Failed to fixup .dynsym in " << file_path_;
1549     return false;
1550   }
1551   if (!FixupSymbols(base_address, false)) {
1552     LOG(WARNING) << "Failed to fixup .symtab in " << file_path_;
1553     return false;
1554   }
1555   if (!FixupRelocations(base_address)) {
1556     LOG(WARNING) << "Failed to fixup .rel.dyn in " << file_path_;
1557     return false;
1558   }
1559   static_assert(sizeof(Elf_Off) >= sizeof(base_address), "Potentially losing precision.");
1560   if (!FixupDebugSections(static_cast<Elf_Off>(base_address))) {
1561     LOG(WARNING) << "Failed to fixup debug sections in " << file_path_;
1562     return false;
1563   }
1564   return true;
1565 }
1566 
1567 template <typename ElfTypes>
FixupDynamic(Elf_Addr base_address)1568 bool ElfFileImpl<ElfTypes>::FixupDynamic(Elf_Addr base_address) {
1569   for (Elf_Word i = 0; i < GetDynamicNum(); i++) {
1570     Elf_Dyn& elf_dyn = GetDynamic(i);
1571     Elf_Word d_tag = elf_dyn.d_tag;
1572     if (IsDynamicSectionPointer(d_tag, GetHeader().e_machine)) {
1573       Elf_Addr d_ptr = elf_dyn.d_un.d_ptr;
1574       if (DEBUG_FIXUP) {
1575         LOG(INFO) << StringPrintf("In %s moving Elf_Dyn[%d] from 0x%" PRIx64 " to 0x%" PRIx64,
1576                                   file_path_.c_str(), i,
1577                                   static_cast<uint64_t>(d_ptr),
1578                                   static_cast<uint64_t>(d_ptr + base_address));
1579       }
1580       d_ptr += base_address;
1581       elf_dyn.d_un.d_ptr = d_ptr;
1582     }
1583   }
1584   return true;
1585 }
1586 
1587 template <typename ElfTypes>
FixupSectionHeaders(Elf_Addr base_address)1588 bool ElfFileImpl<ElfTypes>::FixupSectionHeaders(Elf_Addr base_address) {
1589   for (Elf_Word i = 0; i < GetSectionHeaderNum(); i++) {
1590     Elf_Shdr* sh = GetSectionHeader(i);
1591     CHECK(sh != nullptr);
1592     // 0 implies that the section will not exist in the memory of the process
1593     if (sh->sh_addr == 0) {
1594       continue;
1595     }
1596     if (DEBUG_FIXUP) {
1597       LOG(INFO) << StringPrintf("In %s moving Elf_Shdr[%d] from 0x%" PRIx64 " to 0x%" PRIx64,
1598                                 file_path_.c_str(), i,
1599                                 static_cast<uint64_t>(sh->sh_addr),
1600                                 static_cast<uint64_t>(sh->sh_addr + base_address));
1601     }
1602     sh->sh_addr += base_address;
1603   }
1604   return true;
1605 }
1606 
1607 template <typename ElfTypes>
FixupProgramHeaders(Elf_Addr base_address)1608 bool ElfFileImpl<ElfTypes>::FixupProgramHeaders(Elf_Addr base_address) {
1609   // TODO: ELFObjectFile doesn't have give to Elf_Phdr, so we do that ourselves for now.
1610   for (Elf_Word i = 0; i < GetProgramHeaderNum(); i++) {
1611     Elf_Phdr* ph = GetProgramHeader(i);
1612     CHECK(ph != nullptr);
1613     CHECK_EQ(ph->p_vaddr, ph->p_paddr) << file_path_ << " i=" << i;
1614     CHECK((ph->p_align == 0) || (0 == ((ph->p_vaddr - ph->p_offset) & (ph->p_align - 1))))
1615         << file_path_ << " i=" << i;
1616     if (DEBUG_FIXUP) {
1617       LOG(INFO) << StringPrintf("In %s moving Elf_Phdr[%d] from 0x%" PRIx64 " to 0x%" PRIx64,
1618                                 file_path_.c_str(), i,
1619                                 static_cast<uint64_t>(ph->p_vaddr),
1620                                 static_cast<uint64_t>(ph->p_vaddr + base_address));
1621     }
1622     ph->p_vaddr += base_address;
1623     ph->p_paddr += base_address;
1624     CHECK((ph->p_align == 0) || (0 == ((ph->p_vaddr - ph->p_offset) & (ph->p_align - 1))))
1625         << file_path_ << " i=" << i;
1626   }
1627   return true;
1628 }
1629 
1630 template <typename ElfTypes>
FixupSymbols(Elf_Addr base_address,bool dynamic)1631 bool ElfFileImpl<ElfTypes>::FixupSymbols(Elf_Addr base_address, bool dynamic) {
1632   Elf_Word section_type = dynamic ? SHT_DYNSYM : SHT_SYMTAB;
1633   // TODO: Unfortunate ELFObjectFile has protected symbol access, so use ElfFile
1634   Elf_Shdr* symbol_section = FindSectionByType(section_type);
1635   if (symbol_section == nullptr) {
1636     // file is missing optional .symtab
1637     CHECK(!dynamic) << file_path_;
1638     return true;
1639   }
1640   for (uint32_t i = 0; i < GetSymbolNum(*symbol_section); i++) {
1641     Elf_Sym* symbol = GetSymbol(section_type, i);
1642     CHECK(symbol != nullptr);
1643     if (symbol->st_value != 0) {
1644       if (DEBUG_FIXUP) {
1645         LOG(INFO) << StringPrintf("In %s moving Elf_Sym[%d] from 0x%" PRIx64 " to 0x%" PRIx64,
1646                                   file_path_.c_str(), i,
1647                                   static_cast<uint64_t>(symbol->st_value),
1648                                   static_cast<uint64_t>(symbol->st_value + base_address));
1649       }
1650       symbol->st_value += base_address;
1651     }
1652   }
1653   return true;
1654 }
1655 
1656 template <typename ElfTypes>
FixupRelocations(Elf_Addr base_address)1657 bool ElfFileImpl<ElfTypes>::FixupRelocations(Elf_Addr base_address) {
1658   for (Elf_Word i = 0; i < GetSectionHeaderNum(); i++) {
1659     Elf_Shdr* sh = GetSectionHeader(i);
1660     CHECK(sh != nullptr);
1661     if (sh->sh_type == SHT_REL) {
1662       for (uint32_t j = 0; j < GetRelNum(*sh); j++) {
1663         Elf_Rel& rel = GetRel(*sh, j);
1664         if (DEBUG_FIXUP) {
1665           LOG(INFO) << StringPrintf("In %s moving Elf_Rel[%d] from 0x%" PRIx64 " to 0x%" PRIx64,
1666                                     file_path_.c_str(), j,
1667                                     static_cast<uint64_t>(rel.r_offset),
1668                                     static_cast<uint64_t>(rel.r_offset + base_address));
1669         }
1670         rel.r_offset += base_address;
1671       }
1672     } else if (sh->sh_type == SHT_RELA) {
1673       for (uint32_t j = 0; j < GetRelaNum(*sh); j++) {
1674         Elf_Rela& rela = GetRela(*sh, j);
1675         if (DEBUG_FIXUP) {
1676           LOG(INFO) << StringPrintf("In %s moving Elf_Rela[%d] from 0x%" PRIx64 " to 0x%" PRIx64,
1677                                     file_path_.c_str(), j,
1678                                     static_cast<uint64_t>(rela.r_offset),
1679                                     static_cast<uint64_t>(rela.r_offset + base_address));
1680         }
1681         rela.r_offset += base_address;
1682       }
1683     }
1684   }
1685   return true;
1686 }
1687 
1688 // Explicit instantiations
1689 template class ElfFileImpl<ElfTypes32>;
1690 template class ElfFileImpl<ElfTypes64>;
1691 
ElfFile(ElfFileImpl32 * elf32)1692 ElfFile::ElfFile(ElfFileImpl32* elf32) : elf32_(elf32), elf64_(nullptr) {
1693 }
1694 
ElfFile(ElfFileImpl64 * elf64)1695 ElfFile::ElfFile(ElfFileImpl64* elf64) : elf32_(nullptr), elf64_(elf64) {
1696 }
1697 
~ElfFile()1698 ElfFile::~ElfFile() {
1699   // Should never have 32 and 64-bit impls.
1700   CHECK_NE(elf32_.get() == nullptr, elf64_.get() == nullptr);
1701 }
1702 
Open(File * file,bool writable,bool program_header_only,bool low_4gb,std::string * error_msg)1703 ElfFile* ElfFile::Open(File* file,
1704                        bool writable,
1705                        bool program_header_only,
1706                        bool low_4gb,
1707                        /*out*/std::string* error_msg) {
1708   if (file->GetLength() < EI_NIDENT) {
1709     *error_msg = StringPrintf("File %s is too short to be a valid ELF file",
1710                               file->GetPath().c_str());
1711     return nullptr;
1712   }
1713   MemMap map = MemMap::MapFile(EI_NIDENT,
1714                                PROT_READ,
1715                                MAP_PRIVATE,
1716                                file->Fd(),
1717                                0,
1718                                low_4gb,
1719                                file->GetPath().c_str(),
1720                                error_msg);
1721   if (!map.IsValid() || map.Size() != EI_NIDENT) {
1722     return nullptr;
1723   }
1724   uint8_t* header = map.Begin();
1725   if (header[EI_CLASS] == ELFCLASS64) {
1726     ElfFileImpl64* elf_file_impl = ElfFileImpl64::Open(file,
1727                                                        writable,
1728                                                        program_header_only,
1729                                                        low_4gb,
1730                                                        error_msg);
1731     if (elf_file_impl == nullptr) {
1732       return nullptr;
1733     }
1734     return new ElfFile(elf_file_impl);
1735   } else if (header[EI_CLASS] == ELFCLASS32) {
1736     ElfFileImpl32* elf_file_impl = ElfFileImpl32::Open(file,
1737                                                        writable,
1738                                                        program_header_only,
1739                                                        low_4gb,
1740                                                        error_msg);
1741     if (elf_file_impl == nullptr) {
1742       return nullptr;
1743     }
1744     return new ElfFile(elf_file_impl);
1745   } else {
1746     *error_msg = StringPrintf("Failed to find expected EI_CLASS value %d or %d in %s, found %d",
1747                               ELFCLASS32, ELFCLASS64,
1748                               file->GetPath().c_str(),
1749                               header[EI_CLASS]);
1750     return nullptr;
1751   }
1752 }
1753 
Open(File * file,int mmap_prot,int mmap_flags,std::string * error_msg)1754 ElfFile* ElfFile::Open(File* file, int mmap_prot, int mmap_flags, /*out*/std::string* error_msg) {
1755   // low_4gb support not required for this path.
1756   constexpr bool low_4gb = false;
1757   if (file->GetLength() < EI_NIDENT) {
1758     *error_msg = StringPrintf("File %s is too short to be a valid ELF file",
1759                               file->GetPath().c_str());
1760     return nullptr;
1761   }
1762   MemMap map = MemMap::MapFile(EI_NIDENT,
1763                                PROT_READ,
1764                                MAP_PRIVATE,
1765                                file->Fd(),
1766                                /* start= */ 0,
1767                                low_4gb,
1768                                file->GetPath().c_str(),
1769                                error_msg);
1770   if (!map.IsValid() || map.Size() != EI_NIDENT) {
1771     return nullptr;
1772   }
1773   uint8_t* header = map.Begin();
1774   if (header[EI_CLASS] == ELFCLASS64) {
1775     ElfFileImpl64* elf_file_impl = ElfFileImpl64::Open(file,
1776                                                        mmap_prot,
1777                                                        mmap_flags,
1778                                                        low_4gb,
1779                                                        error_msg);
1780     if (elf_file_impl == nullptr) {
1781       return nullptr;
1782     }
1783     return new ElfFile(elf_file_impl);
1784   } else if (header[EI_CLASS] == ELFCLASS32) {
1785     ElfFileImpl32* elf_file_impl = ElfFileImpl32::Open(file,
1786                                                        mmap_prot,
1787                                                        mmap_flags,
1788                                                        low_4gb,
1789                                                        error_msg);
1790     if (elf_file_impl == nullptr) {
1791       return nullptr;
1792     }
1793     return new ElfFile(elf_file_impl);
1794   } else {
1795     *error_msg = StringPrintf("Failed to find expected EI_CLASS value %d or %d in %s, found %d",
1796                               ELFCLASS32, ELFCLASS64,
1797                               file->GetPath().c_str(),
1798                               header[EI_CLASS]);
1799     return nullptr;
1800   }
1801 }
1802 
1803 #define DELEGATE_TO_IMPL(func, ...) \
1804   if (elf64_.get() != nullptr) { \
1805     return elf64_->func(__VA_ARGS__); \
1806   } else { \
1807     DCHECK(elf32_.get() != nullptr); \
1808     return elf32_->func(__VA_ARGS__); \
1809   }
1810 
Load(File * file,bool executable,bool low_4gb,MemMap * reservation,std::string * error_msg)1811 bool ElfFile::Load(File* file,
1812                    bool executable,
1813                    bool low_4gb,
1814                    /*inout*/MemMap* reservation,
1815                    /*out*/std::string* error_msg) {
1816   DELEGATE_TO_IMPL(Load, file, executable, low_4gb, reservation, error_msg);
1817 }
1818 
FindDynamicSymbolAddress(const std::string & symbol_name) const1819 const uint8_t* ElfFile::FindDynamicSymbolAddress(const std::string& symbol_name) const {
1820   DELEGATE_TO_IMPL(FindDynamicSymbolAddress, symbol_name);
1821 }
1822 
Size() const1823 size_t ElfFile::Size() const {
1824   DELEGATE_TO_IMPL(Size);
1825 }
1826 
Begin() const1827 uint8_t* ElfFile::Begin() const {
1828   DELEGATE_TO_IMPL(Begin);
1829 }
1830 
End() const1831 uint8_t* ElfFile::End() const {
1832   DELEGATE_TO_IMPL(End);
1833 }
1834 
GetFilePath() const1835 const std::string& ElfFile::GetFilePath() const {
1836   DELEGATE_TO_IMPL(GetFilePath);
1837 }
1838 
GetSectionOffsetAndSize(const char * section_name,uint64_t * offset,uint64_t * size) const1839 bool ElfFile::GetSectionOffsetAndSize(const char* section_name, uint64_t* offset,
1840                                       uint64_t* size) const {
1841   if (elf32_.get() == nullptr) {
1842     CHECK(elf64_.get() != nullptr);
1843 
1844     Elf64_Shdr *shdr = elf64_->FindSectionByName(section_name);
1845     if (shdr == nullptr) {
1846       return false;
1847     }
1848     if (offset != nullptr) {
1849       *offset = shdr->sh_offset;
1850     }
1851     if (size != nullptr) {
1852       *size = shdr->sh_size;
1853     }
1854     return true;
1855   } else {
1856     Elf32_Shdr *shdr = elf32_->FindSectionByName(section_name);
1857     if (shdr == nullptr) {
1858       return false;
1859     }
1860     if (offset != nullptr) {
1861       *offset = shdr->sh_offset;
1862     }
1863     if (size != nullptr) {
1864       *size = shdr->sh_size;
1865     }
1866     return true;
1867   }
1868 }
1869 
HasSection(const std::string & name) const1870 bool ElfFile::HasSection(const std::string& name) const {
1871   if (elf64_.get() != nullptr) {
1872     return elf64_->FindSectionByName(name) != nullptr;
1873   } else {
1874     return elf32_->FindSectionByName(name) != nullptr;
1875   }
1876 }
1877 
FindSymbolAddress(unsigned section_type,const std::string & symbol_name,bool build_map)1878 uint64_t ElfFile::FindSymbolAddress(unsigned section_type,
1879                                     const std::string& symbol_name,
1880                                     bool build_map) {
1881   DELEGATE_TO_IMPL(FindSymbolAddress, section_type, symbol_name, build_map);
1882 }
1883 
GetLoadedSize(size_t * size,std::string * error_msg) const1884 bool ElfFile::GetLoadedSize(size_t* size, std::string* error_msg) const {
1885   DELEGATE_TO_IMPL(GetLoadedSize, size, error_msg);
1886 }
1887 
Strip(File * file,std::string * error_msg)1888 bool ElfFile::Strip(File* file, std::string* error_msg) {
1889   std::unique_ptr<ElfFile> elf_file(ElfFile::Open(file, true, false, /*low_4gb=*/false, error_msg));
1890   if (elf_file.get() == nullptr) {
1891     return false;
1892   }
1893 
1894   if (elf_file->elf64_.get() != nullptr) {
1895     return elf_file->elf64_->Strip(file, error_msg);
1896   } else {
1897     return elf_file->elf32_->Strip(file, error_msg);
1898   }
1899 }
1900 
Fixup(uint64_t base_address)1901 bool ElfFile::Fixup(uint64_t base_address) {
1902   if (elf64_.get() != nullptr) {
1903     return elf64_->Fixup(static_cast<Elf64_Addr>(base_address));
1904   } else {
1905     DCHECK(elf32_.get() != nullptr);
1906     CHECK(IsUint<32>(base_address)) << std::hex << base_address;
1907     return elf32_->Fixup(static_cast<Elf32_Addr>(base_address));
1908   }
1909   DELEGATE_TO_IMPL(Fixup, base_address);
1910 }
1911 
1912 }  // namespace art
1913