Branch data Line data Source code
1 : : /* Create new ELF header. 2 : : Copyright (C) 1998, 1999, 2000, 2002, 2015 Red Hat, Inc. 3 : : This file is part of elfutils. 4 : : Written by Ulrich Drepper <drepper@redhat.com>, 1998. 5 : : 6 : : This file is free software; you can redistribute it and/or modify 7 : : it under the terms of either 8 : : 9 : : * the GNU Lesser General Public License as published by the Free 10 : : Software Foundation; either version 3 of the License, or (at 11 : : your option) any later version 12 : : 13 : : or 14 : : 15 : : * the GNU General Public License as published by the Free 16 : : Software Foundation; either version 2 of the License, or (at 17 : : your option) any later version 18 : : 19 : : or both in parallel, as here. 20 : : 21 : : elfutils is distributed in the hope that it will be useful, but 22 : : WITHOUT ANY WARRANTY; without even the implied warranty of 23 : : MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 24 : : General Public License for more details. 25 : : 26 : : You should have received copies of the GNU General Public License and 27 : : the GNU Lesser General Public License along with this program. If 28 : : not, see <http://www.gnu.org/licenses/>. */ 29 : : 30 : : #ifdef HAVE_CONFIG_H 31 : : # include <config.h> 32 : : #endif 33 : : 34 : : #include <stdlib.h> 35 : : #include <string.h> 36 : : 37 : : #include "libelfP.h" 38 : : 39 : : #ifndef LIBELFBITS 40 : : # define LIBELFBITS 32 41 : : #endif 42 : : 43 : : 44 : : ElfW2(LIBELFBITS,Ehdr) * 45 : 1138 : elfw2(LIBELFBITS,newehdr) (Elf *elf) 46 : : { 47 : 1138 : ElfW2(LIBELFBITS,Ehdr) *result; 48 : : 49 [ - + ]: 1138 : if (elf == NULL) 50 : : return NULL; 51 : : 52 [ - + ]: 1138 : if (unlikely (elf->kind != ELF_K_ELF)) 53 : : { 54 : 0 : __libelf_seterrno (ELF_E_INVALID_HANDLE); 55 : 0 : return NULL; 56 : : } 57 : : 58 : 1138 : rwlock_wrlock (elf->lock); 59 : : 60 [ + + ]: 1138 : if (elf->class == 0) 61 : 1050 : elf->class = ELFW(ELFCLASS,LIBELFBITS); 62 [ - + ]: 88 : else if (unlikely (elf->class != ELFW(ELFCLASS,LIBELFBITS))) 63 : : { 64 : 0 : __libelf_seterrno (ELF_E_INVALID_CLASS); 65 : 0 : result = NULL; 66 : 0 : goto out; 67 : : } 68 : : 69 : : /* Don't create an ELF header if one already exists. */ 70 [ + + ]: 1138 : if (elf->state.ELFW(elf,LIBELFBITS).ehdr == NULL) 71 : : { 72 : : /* We use the memory in the ELF descriptor. */ 73 : 1136 : elf->state.ELFW(elf,LIBELFBITS).ehdr = 74 : 1136 : &elf->state.ELFW(elf,LIBELFBITS).ehdr_mem; 75 : : 76 : : /* We clear this memory. */ 77 : 1136 : memset (elf->state.ELFW(elf,LIBELFBITS).ehdr, '\0', 78 : : sizeof (ElfW2(LIBELFBITS,Ehdr))); 79 : : 80 : : /* Mark the ELF header has modified. */ 81 : 1136 : elf->state.ELFW(elf,LIBELFBITS).ehdr_flags |= ELF_F_DIRTY; 82 : : } 83 : : 84 : 1138 : result = elf->state.ELFW(elf,LIBELFBITS).ehdr; 85 : : 86 : : out: 87 : : rwlock_unlock (elf->lock); 88 : : 89 : : return result; 90 : : } 91 : : INTDEF(elfw2(LIBELFBITS,newehdr))