OLD | NEW |
(Empty) | |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 // Implementation notes: |
| 6 // |
| 7 // We need to remove a piece from the ELF shared library. However, we also |
| 8 // want to ensure that code and data loads at the same addresses as before |
| 9 // packing, so that tools like breakpad can still match up addresses found |
| 10 // in any crash dumps with data extracted from the pre-packed version of |
| 11 // the shared library. |
| 12 // |
| 13 // Arranging this means that we have to split one of the LOAD segments into |
| 14 // two. Unfortunately, the program headers are located at the very start |
| 15 // of the shared library file, so expanding the program header section |
| 16 // would cause a lot of consequent changes to files offsets that we don't |
| 17 // really want to have to handle. |
| 18 // |
| 19 // Luckily, though, there is a segment that is always present and always |
| 20 // unused on Android; the GNU_STACK segment. What we do is to steal that |
| 21 // and repurpose it to be one of the split LOAD segments. We then have to |
| 22 // sort LOAD segments by offset to keep the crazy linker happy. |
| 23 // |
| 24 // All of this takes place in SplitProgramHeadersForHole(), used on packing, |
| 25 // and is unraveled on unpacking in CoalesceProgramHeadersForHole(). See |
| 26 // commentary on those functions for an example of this segment stealing |
| 27 // in action. |
| 28 |
| 29 #include "elf_file.h" |
| 30 |
| 31 #include <stdlib.h> |
| 32 #include <sys/types.h> |
| 33 #include <unistd.h> |
| 34 #include <algorithm> |
| 35 #include <string> |
| 36 #include <vector> |
| 37 |
| 38 #include "debug.h" |
| 39 #include "elf_traits.h" |
| 40 #include "libelf.h" |
| 41 #include "packer.h" |
| 42 |
| 43 namespace relocation_packer { |
| 44 |
| 45 // Stub identifier written to 'null out' packed data, "NULL". |
| 46 static const uint32_t kStubIdentifier = 0x4c4c554eu; |
| 47 |
| 48 // Out-of-band dynamic tags used to indicate the offset and size of the |
| 49 // android packed relocations section. |
| 50 static const ELF::Sword DT_ANDROID_REL_OFFSET = DT_LOOS; |
| 51 static const ELF::Sword DT_ANDROID_REL_SIZE = DT_LOOS + 1; |
| 52 |
| 53 // Alignment to preserve, in bytes. This must be at least as large as the |
| 54 // largest d_align and sh_addralign values found in the loaded file. |
| 55 // Out of caution for RELRO page alignment, we preserve to a complete target |
| 56 // page. See http://www.airs.com/blog/archives/189. |
| 57 static const size_t kPreserveAlignment = 4096; |
| 58 |
| 59 namespace { |
| 60 |
| 61 // Get section data. Checks that the section has exactly one data entry, |
| 62 // so that the section size and the data size are the same. True in |
| 63 // practice for all sections we resize when packing or unpacking. Done |
| 64 // by ensuring that a call to elf_getdata(section, data) returns NULL as |
| 65 // the next data entry. |
| 66 Elf_Data* GetSectionData(Elf_Scn* section) { |
| 67 Elf_Data* data = elf_getdata(section, NULL); |
| 68 CHECK(data && elf_getdata(section, data) == NULL); |
| 69 return data; |
| 70 } |
| 71 |
| 72 // Rewrite section data. Allocates new data and makes it the data element's |
| 73 // buffer. Relies on program exit to free allocated data. |
| 74 void RewriteSectionData(Elf_Data* data, |
| 75 const void* section_data, |
| 76 size_t size) { |
| 77 CHECK(size == data->d_size); |
| 78 uint8_t* area = new uint8_t[size]; |
| 79 memcpy(area, section_data, size); |
| 80 data->d_buf = area; |
| 81 } |
| 82 |
| 83 // Verbose ELF header logging. |
| 84 void VerboseLogElfHeader(const ELF::Ehdr* elf_header) { |
| 85 VLOG(1) << "e_phoff = " << elf_header->e_phoff; |
| 86 VLOG(1) << "e_shoff = " << elf_header->e_shoff; |
| 87 VLOG(1) << "e_ehsize = " << elf_header->e_ehsize; |
| 88 VLOG(1) << "e_phentsize = " << elf_header->e_phentsize; |
| 89 VLOG(1) << "e_phnum = " << elf_header->e_phnum; |
| 90 VLOG(1) << "e_shnum = " << elf_header->e_shnum; |
| 91 VLOG(1) << "e_shstrndx = " << elf_header->e_shstrndx; |
| 92 } |
| 93 |
| 94 // Verbose ELF program header logging. |
| 95 void VerboseLogProgramHeader(size_t program_header_index, |
| 96 const ELF::Phdr* program_header) { |
| 97 std::string type; |
| 98 switch (program_header->p_type) { |
| 99 case PT_NULL: type = "NULL"; break; |
| 100 case PT_LOAD: type = "LOAD"; break; |
| 101 case PT_DYNAMIC: type = "DYNAMIC"; break; |
| 102 case PT_INTERP: type = "INTERP"; break; |
| 103 case PT_PHDR: type = "PHDR"; break; |
| 104 case PT_GNU_RELRO: type = "GNU_RELRO"; break; |
| 105 case PT_GNU_STACK: type = "GNU_STACK"; break; |
| 106 case PT_ARM_EXIDX: type = "EXIDX"; break; |
| 107 default: type = "(OTHER)"; break; |
| 108 } |
| 109 VLOG(1) << "phdr[" << program_header_index << "] : " << type; |
| 110 VLOG(1) << " p_offset = " << program_header->p_offset; |
| 111 VLOG(1) << " p_vaddr = " << program_header->p_vaddr; |
| 112 VLOG(1) << " p_paddr = " << program_header->p_paddr; |
| 113 VLOG(1) << " p_filesz = " << program_header->p_filesz; |
| 114 VLOG(1) << " p_memsz = " << program_header->p_memsz; |
| 115 VLOG(1) << " p_flags = " << program_header->p_flags; |
| 116 VLOG(1) << " p_align = " << program_header->p_align; |
| 117 } |
| 118 |
| 119 // Verbose ELF section header logging. |
| 120 void VerboseLogSectionHeader(const std::string& section_name, |
| 121 const ELF::Shdr* section_header) { |
| 122 VLOG(1) << "section " << section_name; |
| 123 VLOG(1) << " sh_addr = " << section_header->sh_addr; |
| 124 VLOG(1) << " sh_offset = " << section_header->sh_offset; |
| 125 VLOG(1) << " sh_size = " << section_header->sh_size; |
| 126 VLOG(1) << " sh_addralign = " << section_header->sh_addralign; |
| 127 } |
| 128 |
| 129 // Verbose ELF section data logging. |
| 130 void VerboseLogSectionData(const Elf_Data* data) { |
| 131 VLOG(1) << " data"; |
| 132 VLOG(1) << " d_buf = " << data->d_buf; |
| 133 VLOG(1) << " d_off = " << data->d_off; |
| 134 VLOG(1) << " d_size = " << data->d_size; |
| 135 VLOG(1) << " d_align = " << data->d_align; |
| 136 } |
| 137 |
| 138 } // namespace |
| 139 |
| 140 // Load the complete ELF file into a memory image in libelf, and identify |
| 141 // the .rel.dyn or .rela.dyn, .dynamic, and .android.rel.dyn or |
| 142 // .android.rela.dyn sections. No-op if the ELF file has already been loaded. |
| 143 bool ElfFile::Load() { |
| 144 if (elf_) |
| 145 return true; |
| 146 |
| 147 Elf* elf = elf_begin(fd_, ELF_C_RDWR, NULL); |
| 148 CHECK(elf); |
| 149 |
| 150 if (elf_kind(elf) != ELF_K_ELF) { |
| 151 LOG(ERROR) << "File not in ELF format"; |
| 152 return false; |
| 153 } |
| 154 |
| 155 ELF::Ehdr* elf_header = ELF::getehdr(elf); |
| 156 if (!elf_header) { |
| 157 LOG(ERROR) << "Failed to load ELF header: " << elf_errmsg(elf_errno()); |
| 158 return false; |
| 159 } |
| 160 if (elf_header->e_machine != ELF::kMachine) { |
| 161 LOG(ERROR) << "ELF file architecture is not " << ELF::Machine(); |
| 162 return false; |
| 163 } |
| 164 if (elf_header->e_type != ET_DYN) { |
| 165 LOG(ERROR) << "ELF file is not a shared object"; |
| 166 return false; |
| 167 } |
| 168 |
| 169 // Require that our endianness matches that of the target, and that both |
| 170 // are little-endian. Safe for all current build/target combinations. |
| 171 const int endian = elf_header->e_ident[EI_DATA]; |
| 172 CHECK(endian == ELFDATA2LSB); |
| 173 CHECK(__BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__); |
| 174 |
| 175 // Also require that the file class is as expected. |
| 176 const int file_class = elf_header->e_ident[EI_CLASS]; |
| 177 CHECK(file_class == ELF::kFileClass); |
| 178 |
| 179 VLOG(1) << "endian = " << endian << ", file class = " << file_class; |
| 180 VerboseLogElfHeader(elf_header); |
| 181 |
| 182 const ELF::Phdr* elf_program_header = ELF::getphdr(elf); |
| 183 CHECK(elf_program_header); |
| 184 |
| 185 const ELF::Phdr* dynamic_program_header = NULL; |
| 186 for (size_t i = 0; i < elf_header->e_phnum; ++i) { |
| 187 const ELF::Phdr* program_header = &elf_program_header[i]; |
| 188 VerboseLogProgramHeader(i, program_header); |
| 189 |
| 190 if (program_header->p_type == PT_DYNAMIC) { |
| 191 CHECK(dynamic_program_header == NULL); |
| 192 dynamic_program_header = program_header; |
| 193 } |
| 194 } |
| 195 CHECK(dynamic_program_header != NULL); |
| 196 |
| 197 size_t string_index; |
| 198 elf_getshdrstrndx(elf, &string_index); |
| 199 |
| 200 // Notes of the dynamic relocations, packed relocations, and .dynamic |
| 201 // sections. Found while iterating sections, and later stored in class |
| 202 // attributes. |
| 203 Elf_Scn* found_relocations_section = NULL; |
| 204 Elf_Scn* found_android_relocations_section = NULL; |
| 205 Elf_Scn* found_dynamic_section = NULL; |
| 206 |
| 207 // Notes of relocation section types seen. We require one or the other of |
| 208 // these; both is unsupported. |
| 209 bool has_rel_relocations = false; |
| 210 bool has_rela_relocations = false; |
| 211 |
| 212 Elf_Scn* section = NULL; |
| 213 while ((section = elf_nextscn(elf, section)) != NULL) { |
| 214 const ELF::Shdr* section_header = ELF::getshdr(section); |
| 215 std::string name = elf_strptr(elf, string_index, section_header->sh_name); |
| 216 VerboseLogSectionHeader(name, section_header); |
| 217 |
| 218 // Note relocation section types. |
| 219 if (section_header->sh_type == SHT_REL) { |
| 220 has_rel_relocations = true; |
| 221 } |
| 222 if (section_header->sh_type == SHT_RELA) { |
| 223 has_rela_relocations = true; |
| 224 } |
| 225 |
| 226 // Note special sections as we encounter them. |
| 227 if ((name == ".rel.dyn" || name == ".rela.dyn") && |
| 228 section_header->sh_size > 0) { |
| 229 found_relocations_section = section; |
| 230 } |
| 231 if ((name == ".android.rel.dyn" || name == ".android.rela.dyn") && |
| 232 section_header->sh_size > 0) { |
| 233 found_android_relocations_section = section; |
| 234 } |
| 235 if (section_header->sh_offset == dynamic_program_header->p_offset) { |
| 236 found_dynamic_section = section; |
| 237 } |
| 238 |
| 239 // Ensure we preserve alignment, repeated later for the data block(s). |
| 240 CHECK(section_header->sh_addralign <= kPreserveAlignment); |
| 241 |
| 242 Elf_Data* data = NULL; |
| 243 while ((data = elf_getdata(section, data)) != NULL) { |
| 244 CHECK(data->d_align <= kPreserveAlignment); |
| 245 VerboseLogSectionData(data); |
| 246 } |
| 247 } |
| 248 |
| 249 // Loading failed if we did not find the required special sections. |
| 250 if (!found_relocations_section) { |
| 251 LOG(ERROR) << "Missing or empty .rel.dyn or .rela.dyn section"; |
| 252 return false; |
| 253 } |
| 254 if (!found_android_relocations_section) { |
| 255 LOG(ERROR) << "Missing or empty .android.rel.dyn or .android.rela.dyn " |
| 256 << "section (to fix, run with --help and follow the " |
| 257 << "pre-packing instructions)"; |
| 258 return false; |
| 259 } |
| 260 if (!found_dynamic_section) { |
| 261 LOG(ERROR) << "Missing .dynamic section"; |
| 262 return false; |
| 263 } |
| 264 |
| 265 // Loading failed if we could not identify the relocations type. |
| 266 if (!has_rel_relocations && !has_rela_relocations) { |
| 267 LOG(ERROR) << "No relocations sections found"; |
| 268 return false; |
| 269 } |
| 270 if (has_rel_relocations && has_rela_relocations) { |
| 271 LOG(ERROR) << "Multiple relocations sections with different types found, " |
| 272 << "not currently supported"; |
| 273 return false; |
| 274 } |
| 275 |
| 276 elf_ = elf; |
| 277 relocations_section_ = found_relocations_section; |
| 278 dynamic_section_ = found_dynamic_section; |
| 279 android_relocations_section_ = found_android_relocations_section; |
| 280 relocations_type_ = has_rel_relocations ? REL : RELA; |
| 281 return true; |
| 282 } |
| 283 |
| 284 namespace { |
| 285 |
| 286 // Helper for ResizeSection(). Adjust the main ELF header for the hole. |
| 287 void AdjustElfHeaderForHole(ELF::Ehdr* elf_header, |
| 288 ELF::Off hole_start, |
| 289 ssize_t hole_size) { |
| 290 if (elf_header->e_phoff > hole_start) { |
| 291 elf_header->e_phoff += hole_size; |
| 292 VLOG(1) << "e_phoff adjusted to " << elf_header->e_phoff; |
| 293 } |
| 294 if (elf_header->e_shoff > hole_start) { |
| 295 elf_header->e_shoff += hole_size; |
| 296 VLOG(1) << "e_shoff adjusted to " << elf_header->e_shoff; |
| 297 } |
| 298 } |
| 299 |
| 300 // Helper for ResizeSection(). Adjust all section headers for the hole. |
| 301 void AdjustSectionHeadersForHole(Elf* elf, |
| 302 ELF::Off hole_start, |
| 303 ssize_t hole_size) { |
| 304 size_t string_index; |
| 305 elf_getshdrstrndx(elf, &string_index); |
| 306 |
| 307 Elf_Scn* section = NULL; |
| 308 while ((section = elf_nextscn(elf, section)) != NULL) { |
| 309 ELF::Shdr* section_header = ELF::getshdr(section); |
| 310 std::string name = elf_strptr(elf, string_index, section_header->sh_name); |
| 311 |
| 312 if (section_header->sh_offset > hole_start) { |
| 313 section_header->sh_offset += hole_size; |
| 314 VLOG(1) << "section " << name |
| 315 << " sh_offset adjusted to " << section_header->sh_offset; |
| 316 } |
| 317 } |
| 318 } |
| 319 |
| 320 // Helper for ResizeSection(). Adjust the offsets of any program headers |
| 321 // that have offsets currently beyond the hole start. |
| 322 void AdjustProgramHeaderOffsets(ELF::Phdr* program_headers, |
| 323 size_t count, |
| 324 ELF::Phdr* ignored_1, |
| 325 ELF::Phdr* ignored_2, |
| 326 ELF::Off hole_start, |
| 327 ssize_t hole_size) { |
| 328 for (size_t i = 0; i < count; ++i) { |
| 329 ELF::Phdr* program_header = &program_headers[i]; |
| 330 |
| 331 if (program_header == ignored_1 || program_header == ignored_2) |
| 332 continue; |
| 333 |
| 334 if (program_header->p_offset > hole_start) { |
| 335 // The hole start is past this segment, so adjust offset. |
| 336 program_header->p_offset += hole_size; |
| 337 VLOG(1) << "phdr[" << i |
| 338 << "] p_offset adjusted to "<< program_header->p_offset; |
| 339 } |
| 340 } |
| 341 } |
| 342 |
| 343 // Helper for ResizeSection(). Find the first loadable segment in the |
| 344 // file. We expect it to map from file offset zero. |
| 345 ELF::Phdr* FindFirstLoadSegment(ELF::Phdr* program_headers, |
| 346 size_t count) { |
| 347 ELF::Phdr* first_loadable_segment = NULL; |
| 348 |
| 349 for (size_t i = 0; i < count; ++i) { |
| 350 ELF::Phdr* program_header = &program_headers[i]; |
| 351 |
| 352 if (program_header->p_type == PT_LOAD && |
| 353 program_header->p_offset == 0 && |
| 354 program_header->p_vaddr == 0 && |
| 355 program_header->p_paddr == 0) { |
| 356 first_loadable_segment = program_header; |
| 357 } |
| 358 } |
| 359 LOG_IF(FATAL, !first_loadable_segment) |
| 360 << "Cannot locate a LOAD segment with address and offset zero"; |
| 361 |
| 362 return first_loadable_segment; |
| 363 } |
| 364 |
| 365 // Helper for ResizeSection(). Find the PT_GNU_STACK segment, and check |
| 366 // that it contains what we expect so we can restore it on unpack if needed. |
| 367 ELF::Phdr* FindUnusedGnuStackSegment(ELF::Phdr* program_headers, |
| 368 size_t count) { |
| 369 ELF::Phdr* unused_segment = NULL; |
| 370 |
| 371 for (size_t i = 0; i < count; ++i) { |
| 372 ELF::Phdr* program_header = &program_headers[i]; |
| 373 |
| 374 if (program_header->p_type == PT_GNU_STACK && |
| 375 program_header->p_offset == 0 && |
| 376 program_header->p_vaddr == 0 && |
| 377 program_header->p_paddr == 0 && |
| 378 program_header->p_filesz == 0 && |
| 379 program_header->p_memsz == 0 && |
| 380 program_header->p_flags == (PF_R | PF_W) && |
| 381 program_header->p_align == ELF::kGnuStackSegmentAlignment) { |
| 382 unused_segment = program_header; |
| 383 } |
| 384 } |
| 385 LOG_IF(FATAL, !unused_segment) |
| 386 << "Cannot locate the expected GNU_STACK segment"; |
| 387 |
| 388 return unused_segment; |
| 389 } |
| 390 |
| 391 // Helper for ResizeSection(). Find the segment that was the first loadable |
| 392 // one before we split it into two. This is the one into which we coalesce |
| 393 // the split segments on unpacking. |
| 394 ELF::Phdr* FindOriginalFirstLoadSegment(ELF::Phdr* program_headers, |
| 395 size_t count) { |
| 396 const ELF::Phdr* first_loadable_segment = |
| 397 FindFirstLoadSegment(program_headers, count); |
| 398 |
| 399 ELF::Phdr* original_first_loadable_segment = NULL; |
| 400 |
| 401 for (size_t i = 0; i < count; ++i) { |
| 402 ELF::Phdr* program_header = &program_headers[i]; |
| 403 |
| 404 // The original first loadable segment is the one that follows on from |
| 405 // the one we wrote on split to be the current first loadable segment. |
| 406 if (program_header->p_type == PT_LOAD && |
| 407 program_header->p_offset == first_loadable_segment->p_filesz) { |
| 408 original_first_loadable_segment = program_header; |
| 409 } |
| 410 } |
| 411 LOG_IF(FATAL, !original_first_loadable_segment) |
| 412 << "Cannot locate the LOAD segment that follows a LOAD at offset zero"; |
| 413 |
| 414 return original_first_loadable_segment; |
| 415 } |
| 416 |
| 417 // Helper for ResizeSection(). Find the segment that contains the hole. |
| 418 Elf_Scn* FindSectionContainingHole(Elf* elf, |
| 419 ELF::Off hole_start, |
| 420 ssize_t hole_size) { |
| 421 Elf_Scn* section = NULL; |
| 422 Elf_Scn* last_unholed_section = NULL; |
| 423 |
| 424 while ((section = elf_nextscn(elf, section)) != NULL) { |
| 425 const ELF::Shdr* section_header = ELF::getshdr(section); |
| 426 |
| 427 // Because we get here after section headers have been adjusted for the |
| 428 // hole, we need to 'undo' that adjustment to give a view of the original |
| 429 // sections layout. |
| 430 ELF::Off offset = section_header->sh_offset; |
| 431 if (section_header->sh_offset >= hole_start) { |
| 432 offset -= hole_size; |
| 433 } |
| 434 |
| 435 if (offset <= hole_start) { |
| 436 last_unholed_section = section; |
| 437 } |
| 438 } |
| 439 LOG_IF(FATAL, !last_unholed_section) |
| 440 << "Cannot identify the section before the one containing the hole"; |
| 441 |
| 442 // The section containing the hole is the one after the last one found |
| 443 // by the loop above. |
| 444 Elf_Scn* holed_section = elf_nextscn(elf, last_unholed_section); |
| 445 LOG_IF(FATAL, !holed_section) |
| 446 << "Cannot identify the section containing the hole"; |
| 447 |
| 448 return holed_section; |
| 449 } |
| 450 |
| 451 // Helper for ResizeSection(). Find the last section contained in a segment. |
| 452 Elf_Scn* FindLastSectionInSegment(Elf* elf, |
| 453 ELF::Phdr* program_header, |
| 454 ELF::Off hole_start, |
| 455 ssize_t hole_size) { |
| 456 const ELF::Off segment_end = |
| 457 program_header->p_offset + program_header->p_filesz; |
| 458 |
| 459 Elf_Scn* section = NULL; |
| 460 Elf_Scn* last_section = NULL; |
| 461 |
| 462 while ((section = elf_nextscn(elf, section)) != NULL) { |
| 463 const ELF::Shdr* section_header = ELF::getshdr(section); |
| 464 |
| 465 // As above, 'undo' any section offset adjustment to give a view of the |
| 466 // original sections layout. |
| 467 ELF::Off offset = section_header->sh_offset; |
| 468 if (section_header->sh_offset >= hole_start) { |
| 469 offset -= hole_size; |
| 470 } |
| 471 |
| 472 if (offset < segment_end) { |
| 473 last_section = section; |
| 474 } |
| 475 } |
| 476 LOG_IF(FATAL, !last_section) |
| 477 << "Cannot identify the last section in the given segment"; |
| 478 |
| 479 return last_section; |
| 480 } |
| 481 |
| 482 // Helper for ResizeSection(). Order loadable segments by their offsets. |
| 483 // The crazy linker contains assumptions about loadable segment ordering, |
| 484 // and it is better if we do not break them. |
| 485 void SortOrderSensitiveProgramHeaders(ELF::Phdr* program_headers, |
| 486 size_t count) { |
| 487 std::vector<ELF::Phdr*> orderable; |
| 488 |
| 489 // Collect together orderable program headers. These are all the LOAD |
| 490 // segments, and any GNU_STACK that may be present (removed on packing, |
| 491 // but replaced on unpacking). |
| 492 for (size_t i = 0; i < count; ++i) { |
| 493 ELF::Phdr* program_header = &program_headers[i]; |
| 494 |
| 495 if (program_header->p_type == PT_LOAD || |
| 496 program_header->p_type == PT_GNU_STACK) { |
| 497 orderable.push_back(program_header); |
| 498 } |
| 499 } |
| 500 |
| 501 // Order these program headers so that any PT_GNU_STACK is last, and |
| 502 // the LOAD segments that precede it appear in offset order. Uses |
| 503 // insertion sort. |
| 504 for (size_t i = 1; i < orderable.size(); ++i) { |
| 505 for (size_t j = i; j > 0; --j) { |
| 506 ELF::Phdr* first = orderable[j - 1]; |
| 507 ELF::Phdr* second = orderable[j]; |
| 508 |
| 509 if (!(first->p_type == PT_GNU_STACK || |
| 510 first->p_offset > second->p_offset)) { |
| 511 break; |
| 512 } |
| 513 std::swap(*first, *second); |
| 514 } |
| 515 } |
| 516 } |
| 517 |
| 518 // Helper for ResizeSection(). The GNU_STACK program header is unused in |
| 519 // Android, so we can repurpose it here. Before packing, the program header |
| 520 // table contains something like: |
| 521 // |
| 522 // Type Offset VirtAddr PhysAddr FileSiz MemSiz Flg Align |
| 523 // LOAD 0x000000 0x00000000 0x00000000 0x1efc818 0x1efc818 R E 0x1000 |
| 524 // LOAD 0x1efd008 0x01efe008 0x01efe008 0x17ec3c 0x1a0324 RW 0x1000 |
| 525 // DYNAMIC 0x205ec50 0x0205fc50 0x0205fc50 0x00108 0x00108 RW 0x4 |
| 526 // GNU_STACK 0x000000 0x00000000 0x00000000 0x00000 0x00000 RW 0 |
| 527 // |
| 528 // The hole in the file is in the first of these. In order to preserve all |
| 529 // load addresses, what we do is to turn the GNU_STACK into a new LOAD entry |
| 530 // that maps segments up to where we created the hole, adjust the first LOAD |
| 531 // entry so that it maps segments after that, adjust any other program |
| 532 // headers whose offset is after the hole start, and finally order the LOAD |
| 533 // segments by offset, to give: |
| 534 // |
| 535 // Type Offset VirtAddr PhysAddr FileSiz MemSiz Flg Align |
| 536 // LOAD 0x000000 0x00000000 0x00000000 0x14ea4 0x14ea4 R E 0x1000 |
| 537 // LOAD 0x014ea4 0x00212ea4 0x00212ea4 0x1cea164 0x1cea164 R E 0x1000 |
| 538 // DYNAMIC 0x1e60c50 0x0205fc50 0x0205fc50 0x00108 0x00108 RW 0x4 |
| 539 // LOAD 0x1cff008 0x01efe008 0x01efe008 0x17ec3c 0x1a0324 RW 0x1000 |
| 540 // |
| 541 // We work out the split points by finding the .rel.dyn or .rela.dyn section |
| 542 // that contains the hole, and by finding the last section in a given segment. |
| 543 // |
| 544 // To unpack, we reverse the above to leave the file as it was originally. |
| 545 void SplitProgramHeadersForHole(Elf* elf, |
| 546 ELF::Off hole_start, |
| 547 ssize_t hole_size) { |
| 548 CHECK(hole_size < 0); |
| 549 const ELF::Ehdr* elf_header = ELF::getehdr(elf); |
| 550 CHECK(elf_header); |
| 551 |
| 552 ELF::Phdr* elf_program_header = ELF::getphdr(elf); |
| 553 CHECK(elf_program_header); |
| 554 |
| 555 const size_t program_header_count = elf_header->e_phnum; |
| 556 |
| 557 // Locate the segment that we can overwrite to form the new LOAD entry, |
| 558 // and the segment that we are going to split into two parts. |
| 559 ELF::Phdr* spliced_header = |
| 560 FindUnusedGnuStackSegment(elf_program_header, program_header_count); |
| 561 ELF::Phdr* split_header = |
| 562 FindFirstLoadSegment(elf_program_header, program_header_count); |
| 563 |
| 564 VLOG(1) << "phdr[" << split_header - elf_program_header << "] split"; |
| 565 VLOG(1) << "phdr[" << spliced_header - elf_program_header << "] new LOAD"; |
| 566 |
| 567 // Find the section that contains the hole. We split on the section that |
| 568 // follows it. |
| 569 Elf_Scn* holed_section = |
| 570 FindSectionContainingHole(elf, hole_start, hole_size); |
| 571 |
| 572 size_t string_index; |
| 573 elf_getshdrstrndx(elf, &string_index); |
| 574 |
| 575 ELF::Shdr* section_header = ELF::getshdr(holed_section); |
| 576 std::string name = elf_strptr(elf, string_index, section_header->sh_name); |
| 577 VLOG(1) << "section " << name << " split after"; |
| 578 |
| 579 // Find the last section in the segment we are splitting. |
| 580 Elf_Scn* last_section = |
| 581 FindLastSectionInSegment(elf, split_header, hole_start, hole_size); |
| 582 |
| 583 section_header = ELF::getshdr(last_section); |
| 584 name = elf_strptr(elf, string_index, section_header->sh_name); |
| 585 VLOG(1) << "section " << name << " split end"; |
| 586 |
| 587 // Split on the section following the holed one, and up to (but not |
| 588 // including) the section following the last one in the split segment. |
| 589 Elf_Scn* split_section = elf_nextscn(elf, holed_section); |
| 590 LOG_IF(FATAL, !split_section) |
| 591 << "No section follows the section that contains the hole"; |
| 592 Elf_Scn* end_section = elf_nextscn(elf, last_section); |
| 593 LOG_IF(FATAL, !end_section) |
| 594 << "No section follows the last section in the segment being split"; |
| 595 |
| 596 // Split the first portion of split_header into spliced_header. |
| 597 const ELF::Shdr* split_section_header = ELF::getshdr(split_section); |
| 598 spliced_header->p_type = split_header->p_type; |
| 599 spliced_header->p_offset = split_header->p_offset; |
| 600 spliced_header->p_vaddr = split_header->p_vaddr; |
| 601 spliced_header->p_paddr = split_header->p_paddr; |
| 602 CHECK(split_header->p_filesz == split_header->p_memsz); |
| 603 spliced_header->p_filesz = split_section_header->sh_offset; |
| 604 spliced_header->p_memsz = split_section_header->sh_offset; |
| 605 spliced_header->p_flags = split_header->p_flags; |
| 606 spliced_header->p_align = split_header->p_align; |
| 607 |
| 608 // Now rewrite split_header to remove the part we spliced from it. |
| 609 const ELF::Shdr* end_section_header = ELF::getshdr(end_section); |
| 610 split_header->p_offset = spliced_header->p_filesz; |
| 611 CHECK(split_header->p_vaddr == split_header->p_paddr); |
| 612 split_header->p_vaddr = split_section_header->sh_addr; |
| 613 split_header->p_paddr = split_section_header->sh_addr; |
| 614 CHECK(split_header->p_filesz == split_header->p_memsz); |
| 615 split_header->p_filesz = |
| 616 end_section_header->sh_offset - spliced_header->p_filesz; |
| 617 split_header->p_memsz = |
| 618 end_section_header->sh_offset - spliced_header->p_filesz; |
| 619 |
| 620 // Adjust the offsets of all program headers that are not one of the pair |
| 621 // we just created by splitting. |
| 622 AdjustProgramHeaderOffsets(elf_program_header, |
| 623 program_header_count, |
| 624 spliced_header, |
| 625 split_header, |
| 626 hole_start, |
| 627 hole_size); |
| 628 |
| 629 // Finally, order loadable segments by offset/address. The crazy linker |
| 630 // contains assumptions about loadable segment ordering. |
| 631 SortOrderSensitiveProgramHeaders(elf_program_header, |
| 632 program_header_count); |
| 633 } |
| 634 |
| 635 // Helper for ResizeSection(). Undo the work of SplitProgramHeadersForHole(). |
| 636 void CoalesceProgramHeadersForHole(Elf* elf, |
| 637 ELF::Off hole_start, |
| 638 ssize_t hole_size) { |
| 639 CHECK(hole_size > 0); |
| 640 const ELF::Ehdr* elf_header = ELF::getehdr(elf); |
| 641 CHECK(elf_header); |
| 642 |
| 643 ELF::Phdr* elf_program_header = ELF::getphdr(elf); |
| 644 CHECK(elf_program_header); |
| 645 |
| 646 const size_t program_header_count = elf_header->e_phnum; |
| 647 |
| 648 // Locate the segment that we overwrote to form the new LOAD entry, and |
| 649 // the segment that we split into two parts on packing. |
| 650 ELF::Phdr* spliced_header = |
| 651 FindFirstLoadSegment(elf_program_header, program_header_count); |
| 652 ELF::Phdr* split_header = |
| 653 FindOriginalFirstLoadSegment(elf_program_header, program_header_count); |
| 654 |
| 655 VLOG(1) << "phdr[" << spliced_header - elf_program_header << "] stack"; |
| 656 VLOG(1) << "phdr[" << split_header - elf_program_header << "] coalesce"; |
| 657 |
| 658 // Find the last section in the second segment we are coalescing. |
| 659 Elf_Scn* last_section = |
| 660 FindLastSectionInSegment(elf, split_header, hole_start, hole_size); |
| 661 |
| 662 size_t string_index; |
| 663 elf_getshdrstrndx(elf, &string_index); |
| 664 |
| 665 const ELF::Shdr* section_header = ELF::getshdr(last_section); |
| 666 std::string name = elf_strptr(elf, string_index, section_header->sh_name); |
| 667 VLOG(1) << "section " << name << " coalesced"; |
| 668 |
| 669 // Rewrite the coalesced segment into split_header. |
| 670 const ELF::Shdr* last_section_header = ELF::getshdr(last_section); |
| 671 split_header->p_offset = spliced_header->p_offset; |
| 672 CHECK(split_header->p_vaddr == split_header->p_paddr); |
| 673 split_header->p_vaddr = spliced_header->p_vaddr; |
| 674 split_header->p_paddr = spliced_header->p_vaddr; |
| 675 CHECK(split_header->p_filesz == split_header->p_memsz); |
| 676 split_header->p_filesz = |
| 677 last_section_header->sh_offset + last_section_header->sh_size; |
| 678 split_header->p_memsz = |
| 679 last_section_header->sh_offset + last_section_header->sh_size; |
| 680 |
| 681 // Reconstruct the original GNU_STACK segment into spliced_header. |
| 682 spliced_header->p_type = PT_GNU_STACK; |
| 683 spliced_header->p_offset = 0; |
| 684 spliced_header->p_vaddr = 0; |
| 685 spliced_header->p_paddr = 0; |
| 686 spliced_header->p_filesz = 0; |
| 687 spliced_header->p_memsz = 0; |
| 688 spliced_header->p_flags = PF_R | PF_W; |
| 689 spliced_header->p_align = ELF::kGnuStackSegmentAlignment; |
| 690 |
| 691 // Adjust the offsets of all program headers that are not one of the pair |
| 692 // we just coalesced. |
| 693 AdjustProgramHeaderOffsets(elf_program_header, |
| 694 program_header_count, |
| 695 spliced_header, |
| 696 split_header, |
| 697 hole_start, |
| 698 hole_size); |
| 699 |
| 700 // Finally, order loadable segments by offset/address. The crazy linker |
| 701 // contains assumptions about loadable segment ordering. |
| 702 SortOrderSensitiveProgramHeaders(elf_program_header, |
| 703 program_header_count); |
| 704 } |
| 705 |
| 706 // Helper for ResizeSection(). Rewrite program headers. |
| 707 void RewriteProgramHeadersForHole(Elf* elf, |
| 708 ELF::Off hole_start, |
| 709 ssize_t hole_size) { |
| 710 // If hole_size is negative then we are removing a piece of the file, and |
| 711 // we want to split program headers so that we keep the same addresses |
| 712 // for text and data. If positive, then we are putting that piece of the |
| 713 // file back in, so we coalesce the previously split program headers. |
| 714 if (hole_size < 0) |
| 715 SplitProgramHeadersForHole(elf, hole_start, hole_size); |
| 716 else if (hole_size > 0) |
| 717 CoalesceProgramHeadersForHole(elf, hole_start, hole_size); |
| 718 } |
| 719 |
| 720 // Helper for ResizeSection(). Locate and return the dynamic section. |
| 721 Elf_Scn* GetDynamicSection(Elf* elf) { |
| 722 const ELF::Ehdr* elf_header = ELF::getehdr(elf); |
| 723 CHECK(elf_header); |
| 724 |
| 725 const ELF::Phdr* elf_program_header = ELF::getphdr(elf); |
| 726 CHECK(elf_program_header); |
| 727 |
| 728 // Find the program header that describes the dynamic section. |
| 729 const ELF::Phdr* dynamic_program_header = NULL; |
| 730 for (size_t i = 0; i < elf_header->e_phnum; ++i) { |
| 731 const ELF::Phdr* program_header = &elf_program_header[i]; |
| 732 |
| 733 if (program_header->p_type == PT_DYNAMIC) { |
| 734 dynamic_program_header = program_header; |
| 735 } |
| 736 } |
| 737 CHECK(dynamic_program_header); |
| 738 |
| 739 // Now find the section with the same offset as this program header. |
| 740 Elf_Scn* dynamic_section = NULL; |
| 741 Elf_Scn* section = NULL; |
| 742 while ((section = elf_nextscn(elf, section)) != NULL) { |
| 743 ELF::Shdr* section_header = ELF::getshdr(section); |
| 744 |
| 745 if (section_header->sh_offset == dynamic_program_header->p_offset) { |
| 746 dynamic_section = section; |
| 747 } |
| 748 } |
| 749 CHECK(dynamic_section != NULL); |
| 750 |
| 751 return dynamic_section; |
| 752 } |
| 753 |
| 754 // Helper for ResizeSection(). Adjust the .dynamic section for the hole. |
| 755 template <typename Rel> |
| 756 void AdjustDynamicSectionForHole(Elf_Scn* dynamic_section, |
| 757 ELF::Off hole_start, |
| 758 ssize_t hole_size) { |
| 759 Elf_Data* data = GetSectionData(dynamic_section); |
| 760 |
| 761 const ELF::Dyn* dynamic_base = reinterpret_cast<ELF::Dyn*>(data->d_buf); |
| 762 std::vector<ELF::Dyn> dynamics( |
| 763 dynamic_base, |
| 764 dynamic_base + data->d_size / sizeof(dynamics[0])); |
| 765 |
| 766 for (size_t i = 0; i < dynamics.size(); ++i) { |
| 767 ELF::Dyn* dynamic = &dynamics[i]; |
| 768 const ELF::Sword tag = dynamic->d_tag; |
| 769 |
| 770 // DT_RELSZ or DT_RELASZ indicate the overall size of relocations. |
| 771 // Only one will be present. Adjust by hole size. |
| 772 if (tag == DT_RELSZ || tag == DT_RELASZ) { |
| 773 dynamic->d_un.d_val += hole_size; |
| 774 VLOG(1) << "dynamic[" << i << "] " << dynamic->d_tag |
| 775 << " d_val adjusted to " << dynamic->d_un.d_val; |
| 776 } |
| 777 |
| 778 // DT_RELCOUNT or DT_RELACOUNT hold the count of relative relocations. |
| 779 // Only one will be present. Packing reduces it to the alignment |
| 780 // padding, if any; unpacking restores it to its former value. The |
| 781 // crazy linker does not use it, but we update it anyway. |
| 782 if (tag == DT_RELCOUNT || tag == DT_RELACOUNT) { |
| 783 // Cast sizeof to a signed type to avoid the division result being |
| 784 // promoted into an unsigned size_t. |
| 785 const ssize_t sizeof_rel = static_cast<ssize_t>(sizeof(Rel)); |
| 786 dynamic->d_un.d_val += hole_size / sizeof_rel; |
| 787 VLOG(1) << "dynamic[" << i << "] " << dynamic->d_tag |
| 788 << " d_val adjusted to " << dynamic->d_un.d_val; |
| 789 } |
| 790 |
| 791 // DT_RELENT and DT_RELAENT do not change, but make sure they are what |
| 792 // we expect. Only one will be present. |
| 793 if (tag == DT_RELENT || tag == DT_RELAENT) { |
| 794 CHECK(dynamic->d_un.d_val == sizeof(Rel)); |
| 795 } |
| 796 } |
| 797 |
| 798 void* section_data = &dynamics[0]; |
| 799 size_t bytes = dynamics.size() * sizeof(dynamics[0]); |
| 800 RewriteSectionData(data, section_data, bytes); |
| 801 } |
| 802 |
| 803 // Resize a section. If the new size is larger than the current size, open |
| 804 // up a hole by increasing file offsets that come after the hole. If smaller |
| 805 // than the current size, remove the hole by decreasing those offsets. |
| 806 template <typename Rel> |
| 807 void ResizeSection(Elf* elf, Elf_Scn* section, size_t new_size) { |
| 808 ELF::Shdr* section_header = ELF::getshdr(section); |
| 809 if (section_header->sh_size == new_size) |
| 810 return; |
| 811 |
| 812 // Note if we are resizing the real dyn relocations. |
| 813 size_t string_index; |
| 814 elf_getshdrstrndx(elf, &string_index); |
| 815 const std::string section_name = |
| 816 elf_strptr(elf, string_index, section_header->sh_name); |
| 817 const bool is_relocations_resize = |
| 818 (section_name == ".rel.dyn" || section_name == ".rela.dyn"); |
| 819 |
| 820 // Require that the section size and the data size are the same. True |
| 821 // in practice for all sections we resize when packing or unpacking. |
| 822 Elf_Data* data = GetSectionData(section); |
| 823 CHECK(data->d_off == 0 && data->d_size == section_header->sh_size); |
| 824 |
| 825 // Require that the section is not zero-length (that is, has allocated |
| 826 // data that we can validly expand). |
| 827 CHECK(data->d_size && data->d_buf); |
| 828 |
| 829 const ELF::Off hole_start = section_header->sh_offset; |
| 830 const ssize_t hole_size = new_size - data->d_size; |
| 831 |
| 832 VLOG_IF(1, (hole_size > 0)) << "expand section size = " << data->d_size; |
| 833 VLOG_IF(1, (hole_size < 0)) << "shrink section size = " << data->d_size; |
| 834 |
| 835 // Resize the data and the section header. |
| 836 data->d_size += hole_size; |
| 837 section_header->sh_size += hole_size; |
| 838 |
| 839 // Add the hole size to all offsets in the ELF file that are after the |
| 840 // start of the hole. If the hole size is positive we are expanding the |
| 841 // section to create a new hole; if negative, we are closing up a hole. |
| 842 |
| 843 // Start with the main ELF header. |
| 844 ELF::Ehdr* elf_header = ELF::getehdr(elf); |
| 845 AdjustElfHeaderForHole(elf_header, hole_start, hole_size); |
| 846 |
| 847 // Adjust all section headers. |
| 848 AdjustSectionHeadersForHole(elf, hole_start, hole_size); |
| 849 |
| 850 // If resizing the dynamic relocations, rewrite the program headers to |
| 851 // either split or coalesce segments, and adjust dynamic entries to match. |
| 852 if (is_relocations_resize) { |
| 853 RewriteProgramHeadersForHole(elf, hole_start, hole_size); |
| 854 |
| 855 Elf_Scn* dynamic_section = GetDynamicSection(elf); |
| 856 AdjustDynamicSectionForHole<Rel>(dynamic_section, hole_start, hole_size); |
| 857 } |
| 858 } |
| 859 |
| 860 // Find the first slot in a dynamics array with the given tag. The array |
| 861 // always ends with a free (unused) element, and which we exclude from the |
| 862 // search. Returns dynamics->size() if not found. |
| 863 size_t FindDynamicEntry(ELF::Sword tag, |
| 864 std::vector<ELF::Dyn>* dynamics) { |
| 865 // Loop until the penultimate entry. We exclude the end sentinel. |
| 866 for (size_t i = 0; i < dynamics->size() - 1; ++i) { |
| 867 if (dynamics->at(i).d_tag == tag) |
| 868 return i; |
| 869 } |
| 870 |
| 871 // The tag was not found. |
| 872 return dynamics->size(); |
| 873 } |
| 874 |
| 875 // Replace the first free (unused) slot in a dynamics vector with the given |
| 876 // value. The vector always ends with a free (unused) element, so the slot |
| 877 // found cannot be the last one in the vector. |
| 878 void AddDynamicEntry(const ELF::Dyn& dyn, |
| 879 std::vector<ELF::Dyn>* dynamics) { |
| 880 const size_t slot = FindDynamicEntry(DT_NULL, dynamics); |
| 881 if (slot == dynamics->size()) { |
| 882 LOG(FATAL) << "No spare dynamic array slots found " |
| 883 << "(to fix, increase gold's --spare-dynamic-tags value)"; |
| 884 } |
| 885 |
| 886 // Replace this entry with the one supplied. |
| 887 dynamics->at(slot) = dyn; |
| 888 VLOG(1) << "dynamic[" << slot << "] overwritten with " << dyn.d_tag; |
| 889 } |
| 890 |
| 891 // Remove the element in the dynamics vector that matches the given tag with |
| 892 // unused slot data. Shuffle the following elements up, and ensure that the |
| 893 // last is the null sentinel. |
| 894 void RemoveDynamicEntry(ELF::Sword tag, |
| 895 std::vector<ELF::Dyn>* dynamics) { |
| 896 const size_t slot = FindDynamicEntry(tag, dynamics); |
| 897 CHECK(slot != dynamics->size()); |
| 898 |
| 899 // Remove this entry by shuffling up everything that follows. |
| 900 for (size_t i = slot; i < dynamics->size() - 1; ++i) { |
| 901 dynamics->at(i) = dynamics->at(i + 1); |
| 902 VLOG(1) << "dynamic[" << i |
| 903 << "] overwritten with dynamic[" << i + 1 << "]"; |
| 904 } |
| 905 |
| 906 // Ensure that the end sentinel is still present. |
| 907 CHECK(dynamics->at(dynamics->size() - 1).d_tag == DT_NULL); |
| 908 } |
| 909 |
| 910 // Construct a null relocation without addend. |
| 911 void NullRelocation(ELF::Rel* relocation) { |
| 912 relocation->r_offset = 0; |
| 913 relocation->r_info = ELF_R_INFO(0, ELF::kNoRelocationCode); |
| 914 } |
| 915 |
| 916 // Construct a null relocation with addend. |
| 917 void NullRelocation(ELF::Rela* relocation) { |
| 918 relocation->r_offset = 0; |
| 919 relocation->r_info = ELF_R_INFO(0, ELF::kNoRelocationCode); |
| 920 relocation->r_addend = 0; |
| 921 } |
| 922 |
| 923 // Pad relocations with the given number of null entries. Generates its |
| 924 // null entry with the appropriate NullRelocation() invocation. |
| 925 template <typename Rel> |
| 926 void PadRelocations(size_t count, std::vector<Rel>* relocations) { |
| 927 Rel null_relocation; |
| 928 NullRelocation(&null_relocation); |
| 929 std::vector<Rel> padding(count, null_relocation); |
| 930 relocations->insert(relocations->end(), padding.begin(), padding.end()); |
| 931 } |
| 932 |
| 933 } // namespace |
| 934 |
| 935 // Remove relative entries from dynamic relocations and write as packed |
| 936 // data into android packed relocations. |
| 937 bool ElfFile::PackRelocations() { |
| 938 // Load the ELF file into libelf. |
| 939 if (!Load()) { |
| 940 LOG(ERROR) << "Failed to load as ELF"; |
| 941 return false; |
| 942 } |
| 943 |
| 944 // Retrieve the current dynamic relocations section data. |
| 945 Elf_Data* data = GetSectionData(relocations_section_); |
| 946 |
| 947 if (relocations_type_ == REL) { |
| 948 // Convert data to a vector of relocations. |
| 949 const ELF::Rel* relocations_base = reinterpret_cast<ELF::Rel*>(data->d_buf); |
| 950 std::vector<ELF::Rel> relocations( |
| 951 relocations_base, |
| 952 relocations_base + data->d_size / sizeof(relocations[0])); |
| 953 |
| 954 LOG(INFO) << "Relocations : REL"; |
| 955 return PackTypedRelocations<ELF::Rel>(relocations, data); |
| 956 } |
| 957 |
| 958 if (relocations_type_ == RELA) { |
| 959 // Convert data to a vector of relocations with addends. |
| 960 const ELF::Rela* relocations_base = |
| 961 reinterpret_cast<ELF::Rela*>(data->d_buf); |
| 962 std::vector<ELF::Rela> relocations( |
| 963 relocations_base, |
| 964 relocations_base + data->d_size / sizeof(relocations[0])); |
| 965 |
| 966 LOG(INFO) << "Relocations : RELA"; |
| 967 return PackTypedRelocations<ELF::Rela>(relocations, data); |
| 968 } |
| 969 |
| 970 NOTREACHED(); |
| 971 return false; |
| 972 } |
| 973 |
| 974 // Helper for PackRelocations(). Rel type is one of ELF::Rel or ELF::Rela. |
| 975 template <typename Rel> |
| 976 bool ElfFile::PackTypedRelocations(const std::vector<Rel>& relocations, |
| 977 Elf_Data* data) { |
| 978 // Filter relocations into those that are relative and others. |
| 979 std::vector<Rel> relative_relocations; |
| 980 std::vector<Rel> other_relocations; |
| 981 |
| 982 for (size_t i = 0; i < relocations.size(); ++i) { |
| 983 const Rel& relocation = relocations[i]; |
| 984 if (ELF_R_TYPE(relocation.r_info) == ELF::kRelativeRelocationCode) { |
| 985 CHECK(ELF_R_SYM(relocation.r_info) == 0); |
| 986 relative_relocations.push_back(relocation); |
| 987 } else { |
| 988 other_relocations.push_back(relocation); |
| 989 } |
| 990 } |
| 991 LOG(INFO) << "Relative : " << relative_relocations.size() << " entries"; |
| 992 LOG(INFO) << "Other : " << other_relocations.size() << " entries"; |
| 993 LOG(INFO) << "Total : " << relocations.size() << " entries"; |
| 994 |
| 995 // If no relative relocations then we have nothing packable. Perhaps |
| 996 // the shared object has already been packed? |
| 997 if (relative_relocations.empty()) { |
| 998 LOG(ERROR) << "No relative relocations found (already packed?)"; |
| 999 return false; |
| 1000 } |
| 1001 |
| 1002 // If not padding fully, apply only enough padding to preserve alignment. |
| 1003 // Otherwise, pad so that we do not shrink the relocations section at all. |
| 1004 if (!is_padding_relocations_) { |
| 1005 // Calculate the size of the hole we will close up when we rewrite |
| 1006 // dynamic relocations. |
| 1007 ssize_t hole_size = |
| 1008 relative_relocations.size() * sizeof(relative_relocations[0]); |
| 1009 const ssize_t unaligned_hole_size = hole_size; |
| 1010 |
| 1011 // Adjust the actual hole size to preserve alignment. We always adjust |
| 1012 // by a whole number of NONE-type relocations. |
| 1013 while (hole_size % kPreserveAlignment) |
| 1014 hole_size -= sizeof(relative_relocations[0]); |
| 1015 LOG(INFO) << "Compaction : " << hole_size << " bytes"; |
| 1016 |
| 1017 // Adjusting for alignment may have removed any packing benefit. |
| 1018 if (hole_size == 0) { |
| 1019 LOG(INFO) << "Too few relative relocations to pack after alignment"; |
| 1020 return false; |
| 1021 } |
| 1022 |
| 1023 // Find the padding needed in other_relocations to preserve alignment. |
| 1024 // Ensure that we never completely empty the real relocations section. |
| 1025 size_t padding_bytes = unaligned_hole_size - hole_size; |
| 1026 if (padding_bytes == 0 && other_relocations.size() == 0) { |
| 1027 do { |
| 1028 padding_bytes += sizeof(relative_relocations[0]); |
| 1029 } while (padding_bytes % kPreserveAlignment); |
| 1030 } |
| 1031 CHECK(padding_bytes % sizeof(other_relocations[0]) == 0); |
| 1032 const size_t padding = padding_bytes / sizeof(other_relocations[0]); |
| 1033 |
| 1034 // Padding may have removed any packing benefit. |
| 1035 if (padding >= relative_relocations.size()) { |
| 1036 LOG(INFO) << "Too few relative relocations to pack after padding"; |
| 1037 return false; |
| 1038 } |
| 1039 |
| 1040 // Add null relocations to other_relocations to preserve alignment. |
| 1041 PadRelocations<Rel>(padding, &other_relocations); |
| 1042 LOG(INFO) << "Alignment pad : " << padding << " relocations"; |
| 1043 } else { |
| 1044 // If padding, add NONE-type relocations to other_relocations to make it |
| 1045 // the same size as the the original relocations we read in. This makes |
| 1046 // the ResizeSection() below a no-op. |
| 1047 const size_t padding = relocations.size() - other_relocations.size(); |
| 1048 PadRelocations<Rel>(padding, &other_relocations); |
| 1049 } |
| 1050 |
| 1051 // Pack relative relocations. |
| 1052 const size_t initial_bytes = |
| 1053 relative_relocations.size() * sizeof(relative_relocations[0]); |
| 1054 LOG(INFO) << "Unpacked relative: " << initial_bytes << " bytes"; |
| 1055 std::vector<uint8_t> packed; |
| 1056 RelocationPacker packer; |
| 1057 packer.PackRelativeRelocations(relative_relocations, &packed); |
| 1058 const void* packed_data = &packed[0]; |
| 1059 const size_t packed_bytes = packed.size() * sizeof(packed[0]); |
| 1060 LOG(INFO) << "Packed relative: " << packed_bytes << " bytes"; |
| 1061 |
| 1062 // If we have insufficient relative relocations to form a run then |
| 1063 // packing fails. |
| 1064 if (packed.empty()) { |
| 1065 LOG(INFO) << "Too few relative relocations to pack"; |
| 1066 return false; |
| 1067 } |
| 1068 |
| 1069 // Run a loopback self-test as a check that packing is lossless. |
| 1070 std::vector<Rel> unpacked; |
| 1071 packer.UnpackRelativeRelocations(packed, &unpacked); |
| 1072 CHECK(unpacked.size() == relative_relocations.size()); |
| 1073 CHECK(!memcmp(&unpacked[0], |
| 1074 &relative_relocations[0], |
| 1075 unpacked.size() * sizeof(unpacked[0]))); |
| 1076 |
| 1077 // Make sure packing saved some space. |
| 1078 if (packed_bytes >= initial_bytes) { |
| 1079 LOG(INFO) << "Packing relative relocations saves no space"; |
| 1080 return false; |
| 1081 } |
| 1082 |
| 1083 // Rewrite the current dynamic relocations section to be only the ARM |
| 1084 // non-relative relocations, then shrink it to size. |
| 1085 const void* section_data = &other_relocations[0]; |
| 1086 const size_t bytes = other_relocations.size() * sizeof(other_relocations[0]); |
| 1087 ResizeSection<Rel>(elf_, relocations_section_, bytes); |
| 1088 RewriteSectionData(data, section_data, bytes); |
| 1089 |
| 1090 // Rewrite the current packed android relocations section to hold the packed |
| 1091 // relative relocations. |
| 1092 data = GetSectionData(android_relocations_section_); |
| 1093 ResizeSection<Rel>(elf_, android_relocations_section_, packed_bytes); |
| 1094 RewriteSectionData(data, packed_data, packed_bytes); |
| 1095 |
| 1096 // Rewrite .dynamic to include two new tags describing the packed android |
| 1097 // relocations. |
| 1098 data = GetSectionData(dynamic_section_); |
| 1099 const ELF::Dyn* dynamic_base = reinterpret_cast<ELF::Dyn*>(data->d_buf); |
| 1100 std::vector<ELF::Dyn> dynamics( |
| 1101 dynamic_base, |
| 1102 dynamic_base + data->d_size / sizeof(dynamics[0])); |
| 1103 // Use two of the spare slots to describe the packed section. |
| 1104 ELF::Shdr* section_header = ELF::getshdr(android_relocations_section_); |
| 1105 { |
| 1106 ELF::Dyn dyn; |
| 1107 dyn.d_tag = DT_ANDROID_REL_OFFSET; |
| 1108 dyn.d_un.d_ptr = section_header->sh_offset; |
| 1109 AddDynamicEntry(dyn, &dynamics); |
| 1110 } |
| 1111 { |
| 1112 ELF::Dyn dyn; |
| 1113 dyn.d_tag = DT_ANDROID_REL_SIZE; |
| 1114 dyn.d_un.d_val = section_header->sh_size; |
| 1115 AddDynamicEntry(dyn, &dynamics); |
| 1116 } |
| 1117 const void* dynamics_data = &dynamics[0]; |
| 1118 const size_t dynamics_bytes = dynamics.size() * sizeof(dynamics[0]); |
| 1119 RewriteSectionData(data, dynamics_data, dynamics_bytes); |
| 1120 |
| 1121 Flush(); |
| 1122 return true; |
| 1123 } |
| 1124 |
| 1125 // Find packed relative relocations in the packed android relocations |
| 1126 // section, unpack them, and rewrite the dynamic relocations section to |
| 1127 // contain unpacked data. |
| 1128 bool ElfFile::UnpackRelocations() { |
| 1129 // Load the ELF file into libelf. |
| 1130 if (!Load()) { |
| 1131 LOG(ERROR) << "Failed to load as ELF"; |
| 1132 return false; |
| 1133 } |
| 1134 |
| 1135 // Retrieve the current packed android relocations section data. |
| 1136 Elf_Data* data = GetSectionData(android_relocations_section_); |
| 1137 |
| 1138 // Convert data to a vector of bytes. |
| 1139 const uint8_t* packed_base = reinterpret_cast<uint8_t*>(data->d_buf); |
| 1140 std::vector<uint8_t> packed( |
| 1141 packed_base, |
| 1142 packed_base + data->d_size / sizeof(packed[0])); |
| 1143 |
| 1144 if (packed.size() > 3 && |
| 1145 packed[0] == 'A' && |
| 1146 packed[1] == 'P' && |
| 1147 packed[2] == 'R' && |
| 1148 packed[3] == '1') { |
| 1149 // Signature is APR1, unpack relocations. |
| 1150 CHECK(relocations_type_ == REL); |
| 1151 LOG(INFO) << "Relocations : REL"; |
| 1152 return UnpackTypedRelocations<ELF::Rel>(packed, data); |
| 1153 } |
| 1154 |
| 1155 if (packed.size() > 3 && |
| 1156 packed[0] == 'A' && |
| 1157 packed[1] == 'P' && |
| 1158 packed[2] == 'A' && |
| 1159 packed[3] == '1') { |
| 1160 // Signature is APA1, unpack relocations with addends. |
| 1161 CHECK(relocations_type_ == RELA); |
| 1162 LOG(INFO) << "Relocations : RELA"; |
| 1163 return UnpackTypedRelocations<ELF::Rela>(packed, data); |
| 1164 } |
| 1165 |
| 1166 LOG(ERROR) << "Packed relative relocations not found (not packed?)"; |
| 1167 return false; |
| 1168 } |
| 1169 |
| 1170 // Helper for UnpackRelocations(). Rel type is one of ELF::Rel or ELF::Rela. |
| 1171 template <typename Rel> |
| 1172 bool ElfFile::UnpackTypedRelocations(const std::vector<uint8_t>& packed, |
| 1173 Elf_Data* data) { |
| 1174 // Unpack the data to re-materialize the relative relocations. |
| 1175 const size_t packed_bytes = packed.size() * sizeof(packed[0]); |
| 1176 LOG(INFO) << "Packed relative: " << packed_bytes << " bytes"; |
| 1177 std::vector<Rel> relative_relocations; |
| 1178 RelocationPacker packer; |
| 1179 packer.UnpackRelativeRelocations(packed, &relative_relocations); |
| 1180 const size_t unpacked_bytes = |
| 1181 relative_relocations.size() * sizeof(relative_relocations[0]); |
| 1182 LOG(INFO) << "Unpacked relative: " << unpacked_bytes << " bytes"; |
| 1183 |
| 1184 // Retrieve the current dynamic relocations section data. |
| 1185 data = GetSectionData(relocations_section_); |
| 1186 |
| 1187 // Interpret data as relocations. |
| 1188 const Rel* relocations_base = reinterpret_cast<Rel*>(data->d_buf); |
| 1189 std::vector<Rel> relocations( |
| 1190 relocations_base, |
| 1191 relocations_base + data->d_size / sizeof(relocations[0])); |
| 1192 |
| 1193 std::vector<Rel> other_relocations; |
| 1194 size_t padding = 0; |
| 1195 |
| 1196 // Filter relocations to locate any that are NONE-type. These will occur |
| 1197 // if padding was turned on for packing. |
| 1198 for (size_t i = 0; i < relocations.size(); ++i) { |
| 1199 const Rel& relocation = relocations[i]; |
| 1200 if (ELF_R_TYPE(relocation.r_info) != ELF::kNoRelocationCode) { |
| 1201 other_relocations.push_back(relocation); |
| 1202 } else { |
| 1203 ++padding; |
| 1204 } |
| 1205 } |
| 1206 LOG(INFO) << "Relative : " << relative_relocations.size() << " entries"; |
| 1207 LOG(INFO) << "Other : " << other_relocations.size() << " entries"; |
| 1208 |
| 1209 // If we found the same number of null relocation entries in the dynamic |
| 1210 // relocations section as we hold as unpacked relative relocations, then |
| 1211 // this is a padded file. |
| 1212 const bool is_padded = padding == relative_relocations.size(); |
| 1213 |
| 1214 // Unless padded, report by how much we expand the file. |
| 1215 if (!is_padded) { |
| 1216 // Calculate the size of the hole we will open up when we rewrite |
| 1217 // dynamic relocations. |
| 1218 ssize_t hole_size = |
| 1219 relative_relocations.size() * sizeof(relative_relocations[0]); |
| 1220 |
| 1221 // Adjust the hole size for the padding added to preserve alignment. |
| 1222 hole_size -= padding * sizeof(other_relocations[0]); |
| 1223 LOG(INFO) << "Expansion : " << hole_size << " bytes"; |
| 1224 } |
| 1225 |
| 1226 // Rewrite the current dynamic relocations section to be the relative |
| 1227 // relocations followed by other relocations. This is the usual order in |
| 1228 // which we find them after linking, so this action will normally put the |
| 1229 // entire dynamic relocations section back to its pre-split-and-packed state. |
| 1230 relocations.assign(relative_relocations.begin(), relative_relocations.end()); |
| 1231 relocations.insert(relocations.end(), |
| 1232 other_relocations.begin(), other_relocations.end()); |
| 1233 const void* section_data = &relocations[0]; |
| 1234 const size_t bytes = relocations.size() * sizeof(relocations[0]); |
| 1235 LOG(INFO) << "Total : " << relocations.size() << " entries"; |
| 1236 ResizeSection<Rel>(elf_, relocations_section_, bytes); |
| 1237 RewriteSectionData(data, section_data, bytes); |
| 1238 |
| 1239 // Nearly empty the current packed android relocations section. Leaves a |
| 1240 // four-byte stub so that some data remains allocated to the section. |
| 1241 // This is a convenience which allows us to re-pack this file again without |
| 1242 // having to remove the section and then add a new small one with objcopy. |
| 1243 // The way we resize sections relies on there being some data in a section. |
| 1244 data = GetSectionData(android_relocations_section_); |
| 1245 ResizeSection<Rel>( |
| 1246 elf_, android_relocations_section_, sizeof(kStubIdentifier)); |
| 1247 RewriteSectionData(data, &kStubIdentifier, sizeof(kStubIdentifier)); |
| 1248 |
| 1249 // Rewrite .dynamic to remove two tags describing packed android relocations. |
| 1250 data = GetSectionData(dynamic_section_); |
| 1251 const ELF::Dyn* dynamic_base = reinterpret_cast<ELF::Dyn*>(data->d_buf); |
| 1252 std::vector<ELF::Dyn> dynamics( |
| 1253 dynamic_base, |
| 1254 dynamic_base + data->d_size / sizeof(dynamics[0])); |
| 1255 RemoveDynamicEntry(DT_ANDROID_REL_OFFSET, &dynamics); |
| 1256 RemoveDynamicEntry(DT_ANDROID_REL_SIZE, &dynamics); |
| 1257 const void* dynamics_data = &dynamics[0]; |
| 1258 const size_t dynamics_bytes = dynamics.size() * sizeof(dynamics[0]); |
| 1259 RewriteSectionData(data, dynamics_data, dynamics_bytes); |
| 1260 |
| 1261 Flush(); |
| 1262 return true; |
| 1263 } |
| 1264 |
| 1265 // Flush rewritten shared object file data. |
| 1266 void ElfFile::Flush() { |
| 1267 // Flag all ELF data held in memory as needing to be written back to the |
| 1268 // file, and tell libelf that we have controlled the file layout. |
| 1269 elf_flagelf(elf_, ELF_C_SET, ELF_F_DIRTY); |
| 1270 elf_flagelf(elf_, ELF_C_SET, ELF_F_LAYOUT); |
| 1271 |
| 1272 // Write ELF data back to disk. |
| 1273 const off_t file_bytes = elf_update(elf_, ELF_C_WRITE); |
| 1274 CHECK(file_bytes > 0); |
| 1275 VLOG(1) << "elf_update returned: " << file_bytes; |
| 1276 |
| 1277 // Clean up libelf, and truncate the output file to the number of bytes |
| 1278 // written by elf_update(). |
| 1279 elf_end(elf_); |
| 1280 elf_ = NULL; |
| 1281 const int truncate = ftruncate(fd_, file_bytes); |
| 1282 CHECK(truncate == 0); |
| 1283 } |
| 1284 |
| 1285 } // namespace relocation_packer |
OLD | NEW |