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 (at) 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 elfw2(LIBELFBITS,newehdr) (Elf *elf) 46 { 47 ElfW2(LIBELFBITS,Ehdr) *result; 48 49 if (elf == NULL) 50 return NULL; 51 52 if (unlikely (elf->kind != ELF_K_ELF)) 53 { 54 __libelf_seterrno (ELF_E_INVALID_HANDLE); 55 return NULL; 56 } 57 58 rwlock_wrlock (elf->lock); 59 60 if (elf->class == 0) 61 elf->class = ELFW(ELFCLASS,LIBELFBITS); 62 else if (unlikely (elf->class != ELFW(ELFCLASS,LIBELFBITS))) 63 { 64 __libelf_seterrno (ELF_E_INVALID_CLASS); 65 result = NULL; 66 goto out; 67 } 68 69 /* Don't create an ELF header if one already exists. */ 70 if (elf->state.ELFW(elf,LIBELFBITS).ehdr == NULL) 71 { 72 /* We use the memory in the ELF descriptor. */ 73 elf->state.ELFW(elf,LIBELFBITS).ehdr = 74 &elf->state.ELFW(elf,LIBELFBITS).ehdr_mem; 75 76 /* We clear this memory. */ 77 memset (elf->state.ELFW(elf,LIBELFBITS).ehdr, '\0', 78 sizeof (ElfW2(LIBELFBITS,Ehdr))); 79 80 /* Mark the ELF header has modified. */ 81 elf->state.ELFW(elf,LIBELFBITS).ehdr_flags |= ELF_F_DIRTY; 82 } 83 84 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)) 92
