From b7ba5f6a9f7dcfecfce52fea5c3af6e9700a24d8 Mon Sep 17 00:00:00 2001 From: midipix Date: Nov 19 2016 23:54:18 +0000 Subject: pe_get_image_opt_hdr_addr(): code maintenance. --- 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; }