From b7ba5f6a9f7dcfecfce52fea5c3af6e9700a24d8 Mon Sep 17 00:00:00 2001
From: midipix <writeonce@midipix.org>
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/headers')

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