diff --git a/src/headers/pe_get_image_dos_hdr_addr.c b/src/headers/pe_get_image_dos_hdr_addr.c index a18f1df..0b68d71 100644 --- a/src/headers/pe_get_image_dos_hdr_addr.c +++ b/src/headers/pe_get_image_dos_hdr_addr.c @@ -10,15 +10,13 @@ #include #include -pe_api + struct pe_raw_image_dos_hdr * pe_get_image_dos_hdr_addr(const void * base) { struct pe_raw_image_dos_hdr * dos; dos = (struct pe_raw_image_dos_hdr *)base; - if ((dos->dos_magic[0] == 'M') && (dos->dos_magic[1] == 'Z')) - return dos; - else - return 0; + return ((dos->dos_magic[0] == 'M') && (dos->dos_magic[1] == 'Z')) + ? dos : 0; }