From b7ba5f6a9f7dcfecfce52fea5c3af6e9700a24d8 Mon Sep 17 00:00:00 2001 From: midipix Date: Sat, 19 Nov 2016 18:54:18 -0500 Subject: pe_get_image_opt_hdr_addr(): code maintenance. --- src/headers/pe_get_image_opt_hdr_addr.c | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'src') diff --git a/src/headers/pe_get_image_opt_hdr_addr.c b/src/headers/pe_get_image_opt_hdr_addr.c index d784aed..7fcd0ad 100644 --- a/src/headers/pe_get_image_opt_hdr_addr.c +++ b/src/headers/pe_get_image_opt_hdr_addr.c @@ -11,10 +11,12 @@ pe_api union pe_raw_opt_hdr * pe_get_image_opt_hdr_addr(const void * base) { - struct pe_raw_coff_file_hdr * coff; + struct pe_raw_coff_file_hdr * coff; + void * addr; if (!(coff = pe_get_image_coff_hdr_addr(base))) return 0; - else - return (union pe_raw_opt_hdr *)pe_va_from_rva(coff, sizeof(*coff)); + + addr = pe_va_from_rva(coff,sizeof(*coff)); + return (union pe_raw_opt_hdr *)addr; } -- cgit v1.2.3