diff options
Diffstat (limited to 'src/headers')
| -rw-r--r-- | src/headers/pe_get_image_data_dirs_addr.c | 6 | ||||
| -rw-r--r-- | src/headers/pe_get_image_special_hdr_addr.c | 2 |
2 files changed, 4 insertions, 4 deletions
diff --git a/src/headers/pe_get_image_data_dirs_addr.c b/src/headers/pe_get_image_data_dirs_addr.c index 1edbdd3..54bf6f3 100644 --- a/src/headers/pe_get_image_data_dirs_addr.c +++ b/src/headers/pe_get_image_data_dirs_addr.c @@ -10,7 +10,7 @@ #include <pemagine/pemagine.h> pe_api -struct pe_data_dirs * pe_get_image_data_dirs_addr(const void * base) +struct pe_raw_data_dirs * pe_get_image_data_dirs_addr(const void * base) { uint16_t * magic; union pe_raw_opt_hdr * hdr; @@ -22,10 +22,10 @@ struct pe_data_dirs * pe_get_image_data_dirs_addr(const void * base) switch (*magic) { case PE_MAGIC_PE32: - return (struct pe_data_dirs *)hdr->opt_hdr_32.rva_and_sizes; + return (struct pe_raw_data_dirs *)hdr->opt_hdr_32.rva_and_sizes; case PE_MAGIC_PE32_PLUS: - return (struct pe_data_dirs *)hdr->opt_hdr_64.rva_and_sizes; + return (struct pe_raw_data_dirs *)hdr->opt_hdr_64.rva_and_sizes; default: return 0; diff --git a/src/headers/pe_get_image_special_hdr_addr.c b/src/headers/pe_get_image_special_hdr_addr.c index 276ec9f..9788099 100644 --- a/src/headers/pe_get_image_special_hdr_addr.c +++ b/src/headers/pe_get_image_special_hdr_addr.c @@ -13,7 +13,7 @@ pe_api void * pe_get_image_special_hdr_addr(const void * base, uint32_t ordinal, uint32_t * sec_size) { - struct pe_data_dirs * dirs; + struct pe_raw_data_dirs*dirs; struct pe_block * dir; uint32_t * count; |