diff options
author | norly <ny-git@enpas.org> | 2013-02-23 15:59:12 +0000 |
---|---|---|
committer | norly <ny-git@enpas.org> | 2013-02-23 15:59:12 +0000 |
commit | 6e4cbafe0876ff14ff03a14ab80951167ef67e06 (patch) | |
tree | 6e8ef066e67b841ef5c18ece44d7543958db4e02 /src/elfops/phdr-fixup-selfref.c | |
parent | c4d9a1aaf7c25e45adcaaebc970429f2c61bf6e9 (diff) |
Clean up ELF-related files and functions
Diffstat (limited to 'src/elfops/phdr-fixup-selfref.c')
-rw-r--r-- | src/elfops/phdr-fixup-selfref.c | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/src/elfops/phdr-fixup-selfref.c b/src/elfops/phdr-fixup-selfref.c new file mode 100644 index 0000000..e79f584 --- /dev/null +++ b/src/elfops/phdr-fixup-selfref.c @@ -0,0 +1,41 @@ +#include <stdio.h> + +#include <libelf.h> +#include <gelf.h> + +void elfu_ePhdrFixupSelfRef(Elf *e) +{ + GElf_Ehdr ehdr; + size_t i, n; + + if (!gelf_getehdr(e, &ehdr)) { + fprintf(stderr, "gelf_getehdr() failed: %s.", elf_errmsg(-1)); + return; + } + + if (elf_getphdrnum(e, &n)) { + fprintf(stderr, "elf_getphdrnum() failed: %s\n", elf_errmsg(-1)); + } + + for (i = 0; i < n; i++) { + GElf_Phdr phdr; + + if (gelf_getphdr(e, i, &phdr) != &phdr) { + fprintf(stderr, "gelf_getphdr() failed for #%d: %s\n", i, elf_errmsg(-1)); + continue; + } + + if (phdr.p_type == PT_PHDR) { + phdr.p_offset = ehdr.e_phoff; + phdr.p_filesz = elf32_fsize(ELF_T_PHDR, n, EV_CURRENT); + phdr.p_memsz = phdr.p_filesz; + + if (!gelf_update_phdr (e, i, &phdr)) { + fprintf(stderr, "gelf_update_ehdr() failed: %s\n", elf_errmsg(-1)); + } + } + } + + /* Tell libelf that phdrs have changed */ + elf_flagphdr(e, ELF_C_SET, ELF_F_DIRTY); +} |