elf_stripper.cc revision e84bfb5a809310f29b97e18203e99a2126f2f084
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 19e84bfb5a809310f29b97e18203e99a2126f2f084Nicolas Geoffray#include <unistd.h> 20e84bfb5a809310f29b97e18203e99a2126f2f084Nicolas Geoffray#include <sys/types.h> 216a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom#include <vector> 226a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 236a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom#include "UniquePtr.h" 246a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom#include "base/logging.h" 256a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom#include "elf_file.h" 2650cfe74daaece80853cb3b45d4338329b7d0345bNicolas Geoffray#include "elf_utils.h" 276a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom#include "utils.h" 286a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 296a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstromnamespace art { 306a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 318d31bbd3d6536de12bc20e3d29cfe03fe848f9daIan Rogersbool ElfStripper::Strip(File* file, std::string* error_msg) { 328d31bbd3d6536de12bc20e3d29cfe03fe848f9daIan Rogers UniquePtr<ElfFile> elf_file(ElfFile::Open(file, true, false, error_msg)); 338d31bbd3d6536de12bc20e3d29cfe03fe848f9daIan Rogers if (elf_file.get() == nullptr) { 348d31bbd3d6536de12bc20e3d29cfe03fe848f9daIan Rogers return false; 358d31bbd3d6536de12bc20e3d29cfe03fe848f9daIan Rogers } 366a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 376a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // ELF files produced by MCLinker look roughly like this 386a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // 396a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 406a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Ehdr | contains number of Elf32_Shdr and offset to first 416a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 426a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Phdr | program headers 436a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Phdr | 446a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | ... | 456a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Phdr | 466a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 476a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | section | mixture of needed and unneeded sections 486a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 496a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | section | 506a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 516a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | ... | 526a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 536a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | section | 546a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 556a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Shdr | section headers 566a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Shdr | 576a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | ... | contains offset to section start 586a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // | Elf32_Shdr | 596a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // +------------+ 606a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // 616a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // To strip: 626a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // - leave the Elf32_Ehdr and Elf32_Phdr values in place. 636a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // - walk the sections making a new set of Elf32_Shdr section headers for what we want to keep 646a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // - move the sections are keeping up to fill in gaps of sections we want to strip 656a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // - write new Elf32_Shdr section headers to end of file, updating Elf32_Ehdr 666a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // - truncate rest of file 676a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // 686a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 6950cfe74daaece80853cb3b45d4338329b7d0345bNicolas Geoffray std::vector<Elf32_Shdr> section_headers; 7050cfe74daaece80853cb3b45d4338329b7d0345bNicolas Geoffray std::vector<Elf32_Word> section_headers_original_indexes; 716a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom section_headers.reserve(elf_file->GetSectionHeaderNum()); 726a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 736a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 7450cfe74daaece80853cb3b45d4338329b7d0345bNicolas Geoffray Elf32_Shdr& string_section = elf_file->GetSectionNameStringSection(); 7550cfe74daaece80853cb3b45d4338329b7d0345bNicolas Geoffray for (Elf32_Word i = 0; i < elf_file->GetSectionHeaderNum(); i++) { 7650cfe74daaece80853cb3b45d4338329b7d0345bNicolas Geoffray Elf32_Shdr& sh = elf_file->GetSectionHeader(i); 776a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom const char* name = elf_file->GetString(string_section, sh.sh_name); 786a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom if (name == NULL) { 796a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom CHECK_EQ(0U, i); 806a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom section_headers.push_back(sh); 816a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom section_headers_original_indexes.push_back(0); 826a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom continue; 836a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 846a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom if (StartsWith(name, ".debug") 856a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom || (strcmp(name, ".strtab") == 0) 866a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom || (strcmp(name, ".symtab") == 0)) { 876a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom continue; 886a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 896a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom section_headers.push_back(sh); 906a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom section_headers_original_indexes.push_back(i); 916a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 926a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom CHECK_NE(0U, section_headers.size()); 936a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom CHECK_EQ(section_headers.size(), section_headers_original_indexes.size()); 946a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 956a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // section 0 is the NULL section, sections start at offset of first section 9650cfe74daaece80853cb3b45d4338329b7d0345bNicolas Geoffray Elf32_Off offset = elf_file->GetSectionHeader(1).sh_offset; 976a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom for (size_t i = 1; i < section_headers.size(); i++) { 9850cfe74daaece80853cb3b45d4338329b7d0345bNicolas Geoffray Elf32_Shdr& new_sh = section_headers[i]; 9950cfe74daaece80853cb3b45d4338329b7d0345bNicolas Geoffray Elf32_Shdr& old_sh = elf_file->GetSectionHeader(section_headers_original_indexes[i]); 1006a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom CHECK_EQ(new_sh.sh_name, old_sh.sh_name); 1016a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom if (old_sh.sh_addralign > 1) { 1026a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom offset = RoundUp(offset, old_sh.sh_addralign); 1036a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 1046a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom if (old_sh.sh_offset == offset) { 1056a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // already in place 1066a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom offset += old_sh.sh_size; 1076a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom continue; 1086a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 1096a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom // shift section earlier 1106a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom memmove(elf_file->Begin() + offset, 1116a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom elf_file->Begin() + old_sh.sh_offset, 1126a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom old_sh.sh_size); 1136a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom new_sh.sh_offset = offset; 1146a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom offset += old_sh.sh_size; 1156a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 1166a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 11750cfe74daaece80853cb3b45d4338329b7d0345bNicolas Geoffray Elf32_Off shoff = offset; 11850cfe74daaece80853cb3b45d4338329b7d0345bNicolas Geoffray size_t section_headers_size_in_bytes = section_headers.size() * sizeof(Elf32_Shdr); 1196a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom memcpy(elf_file->Begin() + offset, §ion_headers[0], section_headers_size_in_bytes); 1206a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom offset += section_headers_size_in_bytes; 1216a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 1226a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom elf_file->GetHeader().e_shnum = section_headers.size(); 1236a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom elf_file->GetHeader().e_shoff = shoff; 1246a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom int result = ftruncate(file->Fd(), offset); 1256a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom if (result != 0) { 1268d31bbd3d6536de12bc20e3d29cfe03fe848f9daIan Rogers *error_msg = StringPrintf("Failed to truncate while stripping ELF file: '%s': %s", 1278d31bbd3d6536de12bc20e3d29cfe03fe848f9daIan Rogers file->GetPath().c_str(), strerror(errno)); 1286a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom return false; 1296a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom } 1306a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom return true; 1316a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom} 1326a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom 1336a47b9dc850b903aabefcfab4adb132cb68bba2eBrian Carlstrom} // namespace art 134