119 lines
3.0 KiB
C
119 lines
3.0 KiB
C
|
/*
|
||
|
* 32.newphdr.c - implementation of the elf{32,64}_newphdr(3) functions.
|
||
|
* Copyright (C) 1995 - 2006 Michael Riepe
|
||
|
*
|
||
|
* This library is free software; you can redistribute it and/or
|
||
|
* modify it under the terms of the GNU Library General Public
|
||
|
* License as published by the Free Software Foundation; either
|
||
|
* version 2 of the License, or (at your option) any later version.
|
||
|
*
|
||
|
* This library is distributed in the hope that it will be useful,
|
||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
|
* Library General Public License for more details.
|
||
|
*
|
||
|
* You should have received a copy of the GNU Library General Public
|
||
|
* License along with this library; if not, write to the Free Software
|
||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
|
||
|
*/
|
||
|
|
||
|
#include <private.h>
|
||
|
|
||
|
#ifndef lint
|
||
|
static const char rcsid[] = "@(#) $Id: 32.newphdr.c,v 1.16 2008/05/23 08:15:34 michael Exp $";
|
||
|
#endif /* lint */
|
||
|
|
||
|
static char*
|
||
|
_elf_newphdr(Elf *elf, size_t count, unsigned cls) {
|
||
|
size_t extcount = 0;
|
||
|
Elf_Scn *scn = NULL;
|
||
|
char *phdr = NULL;
|
||
|
size_t size;
|
||
|
|
||
|
if (!elf) {
|
||
|
return NULL;
|
||
|
}
|
||
|
elf_assert(elf->e_magic == ELF_MAGIC);
|
||
|
if (!elf->e_ehdr && !elf->e_readable) {
|
||
|
seterr(ERROR_NOEHDR);
|
||
|
}
|
||
|
else if (elf->e_kind != ELF_K_ELF) {
|
||
|
seterr(ERROR_NOTELF);
|
||
|
}
|
||
|
else if (elf->e_class != cls) {
|
||
|
seterr(ERROR_CLASSMISMATCH);
|
||
|
}
|
||
|
else if (elf->e_ehdr || _elf_cook(elf)) {
|
||
|
size = _msize(cls, _elf_version, ELF_T_PHDR);
|
||
|
elf_assert(size);
|
||
|
if (!(scn = _elf_first_scn(elf))) {
|
||
|
return NULL;
|
||
|
}
|
||
|
if (count) {
|
||
|
if (!(phdr = (char*)malloc(count * size))) {
|
||
|
seterr(ERROR_MEM_PHDR);
|
||
|
return NULL;
|
||
|
}
|
||
|
memset(phdr, 0, count * size);
|
||
|
}
|
||
|
elf_assert(elf->e_ehdr);
|
||
|
elf->e_phnum = count;
|
||
|
if (count >= PN_XNUM) {
|
||
|
/*
|
||
|
* get NULL section (create it if necessary)
|
||
|
*/
|
||
|
extcount = count;
|
||
|
count = PN_XNUM;
|
||
|
}
|
||
|
if (cls == ELFCLASS32) {
|
||
|
((Elf32_Ehdr*)elf->e_ehdr)->e_phnum = count;
|
||
|
scn->s_shdr32.sh_info = extcount;
|
||
|
}
|
||
|
#if __LIBELF64
|
||
|
else if (cls == ELFCLASS64) {
|
||
|
((Elf64_Ehdr*)elf->e_ehdr)->e_phnum = count;
|
||
|
scn->s_shdr64.sh_info = extcount;
|
||
|
}
|
||
|
#endif /* __LIBELF64 */
|
||
|
else {
|
||
|
seterr(ERROR_UNIMPLEMENTED);
|
||
|
if (phdr) {
|
||
|
free(phdr);
|
||
|
}
|
||
|
return NULL;
|
||
|
}
|
||
|
if (elf->e_phdr) {
|
||
|
free(elf->e_phdr);
|
||
|
}
|
||
|
elf->e_phdr = phdr;
|
||
|
elf->e_phdr_flags |= ELF_F_DIRTY;
|
||
|
elf->e_ehdr_flags |= ELF_F_DIRTY;
|
||
|
scn->s_scn_flags |= ELF_F_DIRTY;
|
||
|
return phdr;
|
||
|
}
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
Elf32_Phdr*
|
||
|
elf32_newphdr(Elf *elf, size_t count) {
|
||
|
return (Elf32_Phdr*)_elf_newphdr(elf, count, ELFCLASS32);
|
||
|
}
|
||
|
|
||
|
#if __LIBELF64
|
||
|
|
||
|
Elf64_Phdr*
|
||
|
elf64_newphdr(Elf *elf, size_t count) {
|
||
|
return (Elf64_Phdr*)_elf_newphdr(elf, count, ELFCLASS64);
|
||
|
}
|
||
|
|
||
|
unsigned long
|
||
|
gelf_newphdr(Elf *elf, size_t phnum) {
|
||
|
if (!valid_class(elf->e_class)) {
|
||
|
seterr(ERROR_UNKNOWN_CLASS);
|
||
|
return 0;
|
||
|
}
|
||
|
return (unsigned long)_elf_newphdr(elf, phnum, elf->e_class);
|
||
|
}
|
||
|
|
||
|
#endif /* __LIBELF64 */
|