diff options
Diffstat (limited to 'src/elfops/section-in-segment.c')
-rw-r--r-- | src/elfops/section-in-segment.c | 85 |
1 files changed, 85 insertions, 0 deletions
diff --git a/src/elfops/section-in-segment.c b/src/elfops/section-in-segment.c new file mode 100644 index 0000000..e12720f --- /dev/null +++ b/src/elfops/section-in-segment.c @@ -0,0 +1,85 @@ +#include <stdio.h> +#include <stdlib.h> + +#include <libelf.h> +#include <gelf.h> + +#include <libelfu/libelfu.h> + + +/* + * Returns the section that starts at the same point in the file as + * the segment AND is wholly contained in the memory image. + * + * If no section fits, NULL is returned. + */ +Elf_Scn* elfu_firstSectionInSegment(Elf *e, GElf_Phdr *phdr) +{ + Elf_Scn *scn; + + scn = elf_getscn(e, 1); + while (scn) { + GElf_Shdr shdr; + + if (gelf_getshdr(scn, &shdr) != &shdr) { + return NULL; + } + + if (shdr.sh_offset == phdr->p_offset + && elfu_segmentContainsSection(phdr, &shdr)) { + return scn; + } + + scn = elf_nextscn(e, scn); + } + + return NULL; +} + + + +/* + * Returns the first section that is contained in the segment and + * ends as close to its memory image of as possible (the "last" + * section in the segment). + * + * If no section fits, NULL is returned. + */ +Elf_Scn* elfu_lastSectionInSegment(Elf *e, GElf_Phdr *phdr) +{ + Elf_Scn *last = NULL; + Elf_Scn *scn; + + + scn = elf_getscn(e, 1); + while (scn) { + GElf_Shdr shdr; + + if (gelf_getshdr(scn, &shdr) != &shdr) { + fprintf(stderr, "gelf_getshdr() failed: %s\n", elf_errmsg(-1)); + continue; + } + + if (elfu_segmentContainsSection(phdr, &shdr)) { + if (!last) { + last = scn; + } else { + GElf_Shdr shdrOld; + + if (gelf_getshdr(last, &shdrOld) != &shdrOld) { + continue; + } + + if (shdr.sh_offset + shdr.sh_size + > shdrOld.sh_offset + shdrOld.sh_size) { + // TODO: Check (leftover space in memory image) < (p_align) + last = scn; + } + } + } + + scn = elf_nextscn(e, scn); + } + + return last; +} |