diff options
author | midipix <writeonce@midipix.org> | 2015-05-08 23:22:07 -0400 |
---|---|---|
committer | midipix <writeonce@midipix.org> | 2015-05-08 23:22:07 -0400 |
commit | feffc7263bb2fd33ae467de2dd51f1ddbbb1b895 (patch) | |
tree | 983daec02a2d1833796ad8bd04d43d9b3ec42765 /src/headers/pe_get_image_data_dirs_addr.c | |
parent | 23329916dde5e0ffa056f74a81aeda1bfb7e54cc (diff) | |
download | pemagine-feffc7263bb2fd33ae467de2dd51f1ddbbb1b895.tar.bz2 pemagine-feffc7263bb2fd33ae467de2dd51f1ddbbb1b895.tar.xz |
initial commit.
Diffstat (limited to 'src/headers/pe_get_image_data_dirs_addr.c')
-rw-r--r-- | src/headers/pe_get_image_data_dirs_addr.c | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/src/headers/pe_get_image_data_dirs_addr.c b/src/headers/pe_get_image_data_dirs_addr.c new file mode 100644 index 0000000..d0167c0 --- /dev/null +++ b/src/headers/pe_get_image_data_dirs_addr.c @@ -0,0 +1,33 @@ +/*****************************************************************************/ +/* pemagination: a (virtual) tour into portable bits and executable bytes */ +/* Copyright (C) 2013,2014,2015 Z. Gilboa */ +/* Released under GPLv2 and GPLv3; see COPYING.PEMAGINE. */ +/*****************************************************************************/ + +#include <psxtypes/psxtypes.h> +#include <pemagine/pe_consts.h> +#include <pemagine/pe_structs.h> +#include <pemagine/pemagine.h> + +pe_api +struct pe_data_dirs * pe_get_image_data_dirs_addr(const void * base) +{ + uint16_t * magic; + union pe_opt_hdr * hdr; + + if (!(hdr = pe_get_image_opt_hdr_addr(base))) + return 0; + + magic = (uint16_t *)hdr; + + switch (*magic) { + case PE_MAGIC_PE32: + return (struct pe_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; + + default: + return 0; + } +} |