2 * Copyright (c) 2006 Joseph Koshy
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 * Retrieve section #0, allocating a new section if needed.
36 _libelf_getscn0(Elf
*e
)
40 if ((s
= STAILQ_FIRST(&e
->e_u
.e_elf
.e_scn
)) != NULL
)
43 return (_libelf_allocate_scn(e
, (size_t) SHN_UNDEF
));
47 _libelf_setshnum(Elf
*e
, void *eh
, int ec
, size_t shnum
)
51 if (shnum
>= SHN_LORESERVE
) {
52 if ((scn
= _libelf_getscn0(e
)) == NULL
)
55 assert(scn
->s_ndx
== SHN_UNDEF
);
58 scn
->s_shdr
.s_shdr32
.sh_size
= shnum
;
60 scn
->s_shdr
.s_shdr64
.sh_size
= shnum
;
62 (void) elf_flagshdr(scn
, ELF_C_SET
, ELF_F_DIRTY
);
68 ((Elf32_Ehdr
*) eh
)->e_shnum
= shnum
;
70 ((Elf64_Ehdr
*) eh
)->e_shnum
= shnum
;
77 _libelf_setshstrndx(Elf
*e
, void *eh
, int ec
, size_t shstrndx
)
81 if (shstrndx
>= SHN_LORESERVE
) {
82 if ((scn
= _libelf_getscn0(e
)) == NULL
)
85 assert(scn
->s_ndx
== SHN_UNDEF
);
88 scn
->s_shdr
.s_shdr32
.sh_link
= shstrndx
;
90 scn
->s_shdr
.s_shdr64
.sh_link
= shstrndx
;
92 (void) elf_flagshdr(scn
, ELF_C_SET
, ELF_F_DIRTY
);
94 shstrndx
= SHN_XINDEX
;
98 ((Elf32_Ehdr
*) eh
)->e_shstrndx
= shstrndx
;
100 ((Elf64_Ehdr
*) eh
)->e_shstrndx
= shstrndx
;
106 _libelf_setphnum(Elf
*e
, void *eh
, int ec
, size_t phnum
)
110 if (phnum
>= PN_XNUM
) {
111 if ((scn
= _libelf_getscn0(e
)) == NULL
)
114 assert(scn
->s_ndx
== SHN_UNDEF
);
116 if (ec
== ELFCLASS32
)
117 scn
->s_shdr
.s_shdr32
.sh_info
= phnum
;
119 scn
->s_shdr
.s_shdr64
.sh_info
= phnum
;
121 (void) elf_flagshdr(scn
, ELF_C_SET
, ELF_F_DIRTY
);
126 if (ec
== ELFCLASS32
)
127 ((Elf32_Ehdr
*) eh
)->e_phnum
= phnum
;
129 ((Elf64_Ehdr
*) eh
)->e_phnum
= phnum
;