125b3c049e70834cf33790a28643ab058b507b35cBen Cheng/* Update ELF header. 225b3c049e70834cf33790a28643ab058b507b35cBen Cheng Copyright (C) 2000, 2001, 2002, 2010 Red Hat, Inc. 303333823c75a1c1887e923828113a1b0fd12020cElliott Hughes This file is part of elfutils. 425b3c049e70834cf33790a28643ab058b507b35cBen Cheng Written by Ulrich Drepper <drepper@redhat.com>, 2000. 525b3c049e70834cf33790a28643ab058b507b35cBen Cheng 603333823c75a1c1887e923828113a1b0fd12020cElliott Hughes This file is free software; you can redistribute it and/or modify 703333823c75a1c1887e923828113a1b0fd12020cElliott Hughes it under the terms of either 825b3c049e70834cf33790a28643ab058b507b35cBen Cheng 903333823c75a1c1887e923828113a1b0fd12020cElliott Hughes * the GNU Lesser General Public License as published by the Free 1003333823c75a1c1887e923828113a1b0fd12020cElliott Hughes Software Foundation; either version 3 of the License, or (at 1103333823c75a1c1887e923828113a1b0fd12020cElliott Hughes your option) any later version 1203333823c75a1c1887e923828113a1b0fd12020cElliott Hughes 1303333823c75a1c1887e923828113a1b0fd12020cElliott Hughes or 1403333823c75a1c1887e923828113a1b0fd12020cElliott Hughes 1503333823c75a1c1887e923828113a1b0fd12020cElliott Hughes * the GNU General Public License as published by the Free 1603333823c75a1c1887e923828113a1b0fd12020cElliott Hughes Software Foundation; either version 2 of the License, or (at 1703333823c75a1c1887e923828113a1b0fd12020cElliott Hughes your option) any later version 1803333823c75a1c1887e923828113a1b0fd12020cElliott Hughes 1903333823c75a1c1887e923828113a1b0fd12020cElliott Hughes or both in parallel, as here. 2003333823c75a1c1887e923828113a1b0fd12020cElliott Hughes 2103333823c75a1c1887e923828113a1b0fd12020cElliott Hughes elfutils is distributed in the hope that it will be useful, but 2225b3c049e70834cf33790a28643ab058b507b35cBen Cheng WITHOUT ANY WARRANTY; without even the implied warranty of 2325b3c049e70834cf33790a28643ab058b507b35cBen Cheng MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 2425b3c049e70834cf33790a28643ab058b507b35cBen Cheng General Public License for more details. 2525b3c049e70834cf33790a28643ab058b507b35cBen Cheng 2603333823c75a1c1887e923828113a1b0fd12020cElliott Hughes You should have received copies of the GNU General Public License and 2703333823c75a1c1887e923828113a1b0fd12020cElliott Hughes the GNU Lesser General Public License along with this program. If 2803333823c75a1c1887e923828113a1b0fd12020cElliott Hughes not, see <http://www.gnu.org/licenses/>. */ 2925b3c049e70834cf33790a28643ab058b507b35cBen Cheng 3025b3c049e70834cf33790a28643ab058b507b35cBen Cheng#ifdef HAVE_CONFIG_H 3125b3c049e70834cf33790a28643ab058b507b35cBen Cheng# include <config.h> 3225b3c049e70834cf33790a28643ab058b507b35cBen Cheng#endif 3325b3c049e70834cf33790a28643ab058b507b35cBen Cheng 3425b3c049e70834cf33790a28643ab058b507b35cBen Cheng#include <gelf.h> 3525b3c049e70834cf33790a28643ab058b507b35cBen Cheng#include <string.h> 3625b3c049e70834cf33790a28643ab058b507b35cBen Cheng 3725b3c049e70834cf33790a28643ab058b507b35cBen Cheng#include "libelfP.h" 3825b3c049e70834cf33790a28643ab058b507b35cBen Cheng 3925b3c049e70834cf33790a28643ab058b507b35cBen Cheng 4025b3c049e70834cf33790a28643ab058b507b35cBen Chengint 4125b3c049e70834cf33790a28643ab058b507b35cBen Chenggelf_update_ehdr (Elf *elf, GElf_Ehdr *src) 4225b3c049e70834cf33790a28643ab058b507b35cBen Cheng{ 4325b3c049e70834cf33790a28643ab058b507b35cBen Cheng int result = 0; 4425b3c049e70834cf33790a28643ab058b507b35cBen Cheng 4525b3c049e70834cf33790a28643ab058b507b35cBen Cheng if (elf == NULL) 4625b3c049e70834cf33790a28643ab058b507b35cBen Cheng return 0; 4725b3c049e70834cf33790a28643ab058b507b35cBen Cheng 4825b3c049e70834cf33790a28643ab058b507b35cBen Cheng if (unlikely (elf->kind != ELF_K_ELF)) 4925b3c049e70834cf33790a28643ab058b507b35cBen Cheng { 5025b3c049e70834cf33790a28643ab058b507b35cBen Cheng __libelf_seterrno (ELF_E_INVALID_HANDLE); 5125b3c049e70834cf33790a28643ab058b507b35cBen Cheng return 0; 5225b3c049e70834cf33790a28643ab058b507b35cBen Cheng } 5325b3c049e70834cf33790a28643ab058b507b35cBen Cheng 5425b3c049e70834cf33790a28643ab058b507b35cBen Cheng rwlock_wrlock (elf->lock); 5525b3c049e70834cf33790a28643ab058b507b35cBen Cheng 5625b3c049e70834cf33790a28643ab058b507b35cBen Cheng if (elf->class == ELFCLASS32) 5725b3c049e70834cf33790a28643ab058b507b35cBen Cheng { 5825b3c049e70834cf33790a28643ab058b507b35cBen Cheng Elf32_Ehdr *ehdr = elf->state.elf32.ehdr; 5925b3c049e70834cf33790a28643ab058b507b35cBen Cheng 6025b3c049e70834cf33790a28643ab058b507b35cBen Cheng if (ehdr == NULL) 6125b3c049e70834cf33790a28643ab058b507b35cBen Cheng { 6225b3c049e70834cf33790a28643ab058b507b35cBen Cheng __libelf_seterrno (ELF_E_WRONG_ORDER_EHDR); 6325b3c049e70834cf33790a28643ab058b507b35cBen Cheng goto out; 6425b3c049e70834cf33790a28643ab058b507b35cBen Cheng } 6525b3c049e70834cf33790a28643ab058b507b35cBen Cheng 6625b3c049e70834cf33790a28643ab058b507b35cBen Cheng /* We have to convert the data to the 32 bit format. This might 6725b3c049e70834cf33790a28643ab058b507b35cBen Cheng overflow some fields so we have to test for this case before 6825b3c049e70834cf33790a28643ab058b507b35cBen Cheng copying. */ 6925b3c049e70834cf33790a28643ab058b507b35cBen Cheng if (unlikely (src->e_entry > 0xffffffffull) 7025b3c049e70834cf33790a28643ab058b507b35cBen Cheng || unlikely (src->e_phoff > 0xffffffffull) 7125b3c049e70834cf33790a28643ab058b507b35cBen Cheng || unlikely (src->e_shoff > 0xffffffffull)) 7225b3c049e70834cf33790a28643ab058b507b35cBen Cheng { 7325b3c049e70834cf33790a28643ab058b507b35cBen Cheng __libelf_seterrno (ELF_E_INVALID_DATA); 7425b3c049e70834cf33790a28643ab058b507b35cBen Cheng goto out; 7525b3c049e70834cf33790a28643ab058b507b35cBen Cheng } 7625b3c049e70834cf33790a28643ab058b507b35cBen Cheng 7725b3c049e70834cf33790a28643ab058b507b35cBen Cheng /* Copy the data. */ 7825b3c049e70834cf33790a28643ab058b507b35cBen Cheng memcpy (ehdr->e_ident, src->e_ident, EI_NIDENT); 7925b3c049e70834cf33790a28643ab058b507b35cBen Cheng#define COPY(name) \ 8025b3c049e70834cf33790a28643ab058b507b35cBen Cheng ehdr->name = src->name 8125b3c049e70834cf33790a28643ab058b507b35cBen Cheng COPY (e_type); 8225b3c049e70834cf33790a28643ab058b507b35cBen Cheng COPY (e_machine); 8325b3c049e70834cf33790a28643ab058b507b35cBen Cheng COPY (e_version); 8425b3c049e70834cf33790a28643ab058b507b35cBen Cheng COPY (e_entry); 8525b3c049e70834cf33790a28643ab058b507b35cBen Cheng COPY (e_phoff); 8625b3c049e70834cf33790a28643ab058b507b35cBen Cheng COPY (e_shoff); 8725b3c049e70834cf33790a28643ab058b507b35cBen Cheng COPY (e_flags); 8825b3c049e70834cf33790a28643ab058b507b35cBen Cheng COPY (e_ehsize); 8925b3c049e70834cf33790a28643ab058b507b35cBen Cheng COPY (e_phentsize); 9025b3c049e70834cf33790a28643ab058b507b35cBen Cheng COPY (e_phnum); 9125b3c049e70834cf33790a28643ab058b507b35cBen Cheng COPY (e_shentsize); 9225b3c049e70834cf33790a28643ab058b507b35cBen Cheng COPY (e_shnum); 9325b3c049e70834cf33790a28643ab058b507b35cBen Cheng COPY (e_shstrndx); 9425b3c049e70834cf33790a28643ab058b507b35cBen Cheng } 9525b3c049e70834cf33790a28643ab058b507b35cBen Cheng else 9625b3c049e70834cf33790a28643ab058b507b35cBen Cheng { 9725b3c049e70834cf33790a28643ab058b507b35cBen Cheng Elf64_Ehdr *ehdr = elf->state.elf64.ehdr; 9825b3c049e70834cf33790a28643ab058b507b35cBen Cheng 9925b3c049e70834cf33790a28643ab058b507b35cBen Cheng if (ehdr == NULL) 10025b3c049e70834cf33790a28643ab058b507b35cBen Cheng { 10125b3c049e70834cf33790a28643ab058b507b35cBen Cheng __libelf_seterrno (ELF_E_WRONG_ORDER_EHDR); 10225b3c049e70834cf33790a28643ab058b507b35cBen Cheng goto out; 10325b3c049e70834cf33790a28643ab058b507b35cBen Cheng } 10425b3c049e70834cf33790a28643ab058b507b35cBen Cheng 10525b3c049e70834cf33790a28643ab058b507b35cBen Cheng /* Just copy the data. */ 10625b3c049e70834cf33790a28643ab058b507b35cBen Cheng memcpy (ehdr, src, sizeof (Elf64_Ehdr)); 10725b3c049e70834cf33790a28643ab058b507b35cBen Cheng } 10825b3c049e70834cf33790a28643ab058b507b35cBen Cheng 10925b3c049e70834cf33790a28643ab058b507b35cBen Cheng /* Mark the ELF header as modified. */ 11025b3c049e70834cf33790a28643ab058b507b35cBen Cheng elf->state.elf.ehdr_flags |= ELF_F_DIRTY; 11125b3c049e70834cf33790a28643ab058b507b35cBen Cheng 11225b3c049e70834cf33790a28643ab058b507b35cBen Cheng result = 1; 11325b3c049e70834cf33790a28643ab058b507b35cBen Cheng 11425b3c049e70834cf33790a28643ab058b507b35cBen Cheng out: 11525b3c049e70834cf33790a28643ab058b507b35cBen Cheng rwlock_unlock (elf->lock); 11625b3c049e70834cf33790a28643ab058b507b35cBen Cheng 11725b3c049e70834cf33790a28643ab058b507b35cBen Cheng return result; 11825b3c049e70834cf33790a28643ab058b507b35cBen Cheng} 119