diff options
| -rw-r--r-- | include/perk/perk.h | 4 | ||||
| -rw-r--r-- | src/logic/pe_get_image_meta.c | 43 |
2 files changed, 47 insertions, 0 deletions
diff --git a/include/perk/perk.h b/include/perk/perk.h index 988e0f6..3918d24 100644 --- a/include/perk/perk.h +++ b/include/perk/perk.h @@ -216,6 +216,10 @@ perk_api void pe_free_image_meta (struct pe_image_meta *); perk_api int pe_get_named_section_index(const struct pe_image_meta *, const char * name); perk_api int pe_get_block_section_index(const struct pe_image_meta *, const struct pe_block *); + +perk_api int pe_get_roffset_from_rva (const struct pe_image_meta *, uint32_t rva, uint32_t * roffset); +perk_api int pe_get_rva_from_roffset (const struct pe_image_meta *, uint32_t roffset, uint32_t * rva); + perk_api int pe_get_expsym_by_name (const struct pe_image_meta *, const char * name, struct pe_expsym * optional); perk_api int pe_get_expsym_by_index (const struct pe_image_meta *, unsigned index, struct pe_expsym * optional); diff --git a/src/logic/pe_get_image_meta.c b/src/logic/pe_get_image_meta.c index ee2b632..5f31a9a 100644 --- a/src/logic/pe_get_image_meta.c +++ b/src/logic/pe_get_image_meta.c @@ -59,6 +59,49 @@ int pe_get_block_section_index(const struct pe_image_meta * m, const struct pe_b return -1; } +int pe_get_roffset_from_rva(const struct pe_image_meta * m, uint32_t rva, uint32_t * roffset) +{ + int i; + uint32_t low,high; + + for (i=0; i<m->coff.num_of_sections; i++) { + low = m->sectbl[i].virtual_addr; + high = low + m->sectbl[i].virtual_size; + + if ((rva >= low) && (rva < high)) { + *roffset = (rva - low) + m->sectbl[i].ptr_to_raw_data; + return 0; + } + } + + return -1; +} + +int pe_get_rva_from_roffset(const struct pe_image_meta * m, uint32_t roffset, uint32_t * rva) +{ + int i; + uint32_t low,high,ref; + + for (i=0, ref=-1; i<m->coff.num_of_sections; i++) { + low = m->sectbl[i].ptr_to_raw_data; + high = low + m->sectbl[i].virtual_size; + + if ((roffset >= low) && (roffset < high)) { + *rva = (roffset - low) + m->sectbl[i].virtual_addr; + return 0; + } else if (ref > low) { + ref = low; + } + } + + if (roffset < ref) { + *rva = roffset; + return 0; + } + + return -1; +} + int pe_get_expsym_by_name( const struct pe_image_meta * m, const char * name, |