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 : 471 : elfw2(LIBELFBITS,newehdr) (Elf *elf)
46 : : {
47 : 471 : ElfW2(LIBELFBITS,Ehdr) *result;
48 : :
49 [ + - ]: 471 : if (elf == NULL)
50 : : return NULL;
51 : :
52 [ - + ]: 471 : if (unlikely (elf->kind != ELF_K_ELF))
53 : : {
54 : 0 : __libelf_seterrno (ELF_E_INVALID_HANDLE);
55 : 0 : return NULL;
56 : : }
57 : :
58 : 471 : rwlock_wrlock (elf->lock);
59 : :
60 [ + + ]: 471 : if (elf->class == 0)
61 : 438 : elf->class = ELFW(ELFCLASS,LIBELFBITS);
62 [ - + ]: 33 : 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 [ + + ]: 471 : if (elf->state.ELFW(elf,LIBELFBITS).ehdr == NULL)
71 : : {
72 : : /* We use the memory in the ELF descriptor. */
73 : 940 : elf->state.ELFW(elf,LIBELFBITS).ehdr =
74 : 470 : &elf->state.ELFW(elf,LIBELFBITS).ehdr_mem;
75 : :
76 : : /* We clear this memory. */
77 : 470 : memset (elf->state.ELFW(elf,LIBELFBITS).ehdr, '\0',
78 : : sizeof (ElfW2(LIBELFBITS,Ehdr)));
79 : :
80 : : /* Mark the ELF header has modified. */
81 : 470 : elf->state.ELFW(elf,LIBELFBITS).ehdr_flags |= ELF_F_DIRTY;
82 : : }
83 : :
84 : 471 : 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))
|