16a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom/* 26a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom * Copyright (C) 2012 The Android Open Source Project 36a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom * 46a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom * Licensed under the Apache License, Version 2.0 (the "License"); 56a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom * you may not use this file except in compliance with the License. 66a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom * You may obtain a copy of the License at 76a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom * 86a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom * http://www.apache.org/licenses/LICENSE-2.0 96a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom * 106a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom * Unless required by applicable law or agreed to in writing, software 116a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom * distributed under the License is distributed on an "AS IS" BASIS, 126a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 136a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom * See the License for the specific language governing permissions and 146a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom * limitations under the License. 156a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom */ 166a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 176a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom#include "elf_stripper.h" 186a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 196a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom#include <vector> 206a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 216a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom#include <llvm/Support/ELF.h> 226a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 236a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom#include "UniquePtr.h" 246a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom#include "base/logging.h" 256a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom#include "elf_file.h" 266a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom#include "utils.h" 276a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 286a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstromnamespace art { 296a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 306a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrombool ElfStripper::Strip(File* file) { 316a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom UniquePtr<ElfFile> elf_file(ElfFile::Open(file, true, false)); 326a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom CHECK(elf_file.get() != NULL); 336a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 346a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // ELF files produced by MCLinker look roughly like this 356a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // 366a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 376a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Ehdr | contains number of Elf32_Shdr and offset to first 386a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 396a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Phdr | program headers 406a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Phdr | 416a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | ... | 426a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Phdr | 436a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 446a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | section | mixture of needed and unneeded sections 456a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 466a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | section | 476a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 486a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | ... | 496a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 506a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | section | 516a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 526a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Shdr | section headers 536a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Shdr | 546a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | ... | contains offset to section start 556a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Shdr | 566a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 576a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // 586a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // To strip: 596a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // - leave the Elf32_Ehdr and Elf32_Phdr values in place. 606a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // - walk the sections making a new set of Elf32_Shdr section headers for what we want to keep 616a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // - move the sections are keeping up to fill in gaps of sections we want to strip 626a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // - write new Elf32_Shdr section headers to end of file, updating Elf32_Ehdr 636a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // - truncate rest of file 646a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // 656a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 666a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom std::vector<llvm::ELF::Elf32_Shdr> section_headers; 676a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom std::vector<llvm::ELF::Elf32_Word> section_headers_original_indexes; 686a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom section_headers.reserve(elf_file->GetSectionHeaderNum()); 696a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 706a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 716a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom llvm::ELF::Elf32_Shdr& string_section = elf_file->GetSectionNameStringSection(); 726a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom for (llvm::ELF::Elf32_Word i = 0; i < elf_file->GetSectionHeaderNum(); i++) { 736a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom llvm::ELF::Elf32_Shdr& sh = elf_file->GetSectionHeader(i); 746a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom const char* name = elf_file->GetString(string_section, sh.sh_name); 756a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom if (name == NULL) { 766a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom CHECK_EQ(0U, i); 776a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom section_headers.push_back(sh); 786a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom section_headers_original_indexes.push_back(0); 796a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom continue; 806a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 816a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom if (StartsWith(name, ".debug") 826a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom || (strcmp(name, ".strtab") == 0) 836a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom || (strcmp(name, ".symtab") == 0)) { 846a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom continue; 856a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 866a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom section_headers.push_back(sh); 876a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom section_headers_original_indexes.push_back(i); 886a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 896a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom CHECK_NE(0U, section_headers.size()); 906a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom CHECK_EQ(section_headers.size(), section_headers_original_indexes.size()); 916a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 926a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // section 0 is the NULL section, sections start at offset of first section 936a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom llvm::ELF::Elf32_Off offset = elf_file->GetSectionHeader(1).sh_offset; 946a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom for (size_t i = 1; i < section_headers.size(); i++) { 956a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom llvm::ELF::Elf32_Shdr& new_sh = section_headers[i]; 966a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom llvm::ELF::Elf32_Shdr& old_sh = elf_file->GetSectionHeader(section_headers_original_indexes[i]); 976a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom CHECK_EQ(new_sh.sh_name, old_sh.sh_name); 986a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom if (old_sh.sh_addralign > 1) { 996a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom offset = RoundUp(offset, old_sh.sh_addralign); 1006a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 1016a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom if (old_sh.sh_offset == offset) { 1026a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // already in place 1036a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom offset += old_sh.sh_size; 1046a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom continue; 1056a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 1066a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // shift section earlier 1076a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom memmove(elf_file->Begin() + offset, 1086a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom elf_file->Begin() + old_sh.sh_offset, 1096a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom old_sh.sh_size); 1106a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom new_sh.sh_offset = offset; 1116a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom offset += old_sh.sh_size; 1126a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 1136a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 1146a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom llvm::ELF::Elf32_Off shoff = offset; 1156a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom size_t section_headers_size_in_bytes = section_headers.size() * sizeof(llvm::ELF::Elf32_Shdr); 1166a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom memcpy(elf_file->Begin() + offset, §ion_headers[0], section_headers_size_in_bytes); 1176a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom offset += section_headers_size_in_bytes; 1186a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 1196a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom elf_file->GetHeader().e_shnum = section_headers.size(); 1206a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom elf_file->GetHeader().e_shoff = shoff; 1216a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom int result = ftruncate(file->Fd(), offset); 1226a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom if (result != 0) { 1236a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom PLOG(ERROR) << "Failed to truncate while stripping ELF file: " << file->GetPath(); 1246a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom return false; 1256a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 1266a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom return true; 1276a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom} 1286a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 1296a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom} // namespace art 130