#pragma ident "@(#)newphdr.c 1.18 08/05/31 SMI"
#include <stdlib.h>
#include <memory.h>
#include <errno.h>
#include "decl.h"
#include "msg.h"
#include <string.h>
#if defined(_ELF64)
#define ELFCLASS ELFCLASS64
#define elf_newphdr elf64_newphdr
#define elf_getehdr elf64_getehdr
#define _elf_msize _elf64_msize
#define elf_fsize elf64_fsize
#else
#define ELFCLASS ELFCLASS32
#define elf_newphdr elf32_newphdr
#define elf_getehdr elf32_getehdr
#define _elf_msize _elf32_msize
#define elf_fsize elf32_fsize
#endif
Phdr *
elf_newphdr(Elf * elf, size_t count)
{
Elf_Void * ph;
size_t sz;
Phdr * rc;
unsigned work;
if (elf == 0)
return (0);
ELFRLOCK(elf)
if (elf->ed_class != ELFCLASS) {
_elf_seterr(EREQ_CLASS, 0);
ELFUNLOCK(elf)
return (0);
}
ELFUNLOCK(elf)
if (elf_getehdr(elf) == 0) {
_elf_seterr(ESEQ_EHDR, 0);
return (0);
}
ELFWLOCK(elf)
if (elf->ed_myflags & EDF_PHALLOC) {
elf->ed_myflags &= ~EDF_PHALLOC;
rc = elf->ed_phdr;
free(rc);
}
ELFACCESSDATA(work, _elf_work)
if ((sz = count * _elf_msize(ELF_T_PHDR, work)) == 0) {
elf->ed_phflags &= ~ELF_F_DIRTY;
elf->ed_phdr = 0;
((Ehdr*)elf->ed_ehdr)->e_phnum = 0;
((Ehdr*)elf->ed_ehdr)->e_phentsize = 0;
elf->ed_phdrsz = 0;
ELFUNLOCK(elf)
return (0);
}
if ((ph = malloc(sz)) == 0) {
_elf_seterr(EMEM_PHDR, errno);
elf->ed_phflags &= ~ELF_F_DIRTY;
elf->ed_phdr = 0;
((Ehdr*)elf->ed_ehdr)->e_phnum = 0;
((Ehdr*)elf->ed_ehdr)->e_phentsize = 0;
elf->ed_phdrsz = 0;
ELFUNLOCK(elf)
return (0);
}
elf->ed_myflags |= EDF_PHALLOC;
(void) memset(ph, 0, sz);
elf->ed_phflags |= ELF_F_DIRTY;
((Ehdr*)elf->ed_ehdr)->e_phnum = (Half)count;
((Ehdr*)elf->ed_ehdr)->e_phentsize
= (Half)elf_fsize(ELF_T_PHDR, 1, work);
elf->ed_phdrsz = sz;
elf->ed_phdr = rc = ph;
ELFUNLOCK(elf)
return (rc);
}