00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024 #include <private.h>
00025
00026 #ifndef lint
00027 static const char rcsid[] = "@(#) $Id: 32.newehdr.c,v 1.1.1.1 2005/10/21 19:00:00 marcel Exp $";
00028 #endif
00029
00030 static char*
00031 _elf_newehdr(Elf *elf, unsigned cls) {
00032 size_t size;
00033
00034 if (!elf) {
00035 return NULL;
00036 }
00037 elf_assert(elf->e_magic == ELF_MAGIC);
00038 if (elf->e_readable) {
00039 return _elf_getehdr(elf, cls);
00040 }
00041 else if (!elf->e_ehdr) {
00042 size = _msize(cls, _elf_version, ELF_T_EHDR);
00043 elf_assert(size);
00044 if ((elf->e_ehdr = (char*)malloc(size))) {
00045 memset(elf->e_ehdr, 0, size);
00046 elf->e_free_ehdr = 1;
00047 elf->e_ehdr_flags |= ELF_F_DIRTY;
00048 elf->e_kind = ELF_K_ELF;
00049 elf->e_class = cls;
00050 return elf->e_ehdr;
00051 }
00052 seterr(ERROR_MEM_EHDR);
00053 }
00054 else if (elf->e_class != cls) {
00055 seterr(ERROR_CLASSMISMATCH);
00056 }
00057 else {
00058 elf_assert(elf->e_kind == ELF_K_ELF);
00059 return elf->e_ehdr;
00060 }
00061 return NULL;
00062 }
00063
00064 Elf32_Ehdr*
00065 elf32_newehdr(Elf *elf) {
00066 return (Elf32_Ehdr*)_elf_newehdr(elf, ELFCLASS32);
00067 }
00068
00069 #if __LIBELF64
00070
00071 Elf64_Ehdr*
00072 elf64_newehdr(Elf *elf) {
00073 return (Elf64_Ehdr*)_elf_newehdr(elf, ELFCLASS64);
00074 }
00075
00076 unsigned long
00077 gelf_newehdr(Elf *elf, int cls) {
00078 if (!valid_class(cls) || !_msize(cls, _elf_version, ELF_T_EHDR)) {
00079 seterr(ERROR_UNKNOWN_CLASS);
00080 return 0;
00081 }
00082 if (!_elf_newehdr(elf, cls)) {
00083 return 0;
00084 }
00085 return 1;
00086 }
00087
00088 #endif