Merge pull request VirusTotal#178 from wxsBSD/IMAGE_FILE_MACHINE_AMD64

Use IMAGE_FILE_MACHINE_AMD64 appropriately.
1sn0m4d · Sep 18, 2014 · 6a1c56a · 6a1c56a
2 parents 86ec8c7 + f4b2c31
commit 6a1c56a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libyara/modules/pe.c b/libyara/modules/pe.c
@@ -138,7 +138,7 @@ PIMAGE_DATA_DIRECTORY pe_get_directory_entry(
 {
   PIMAGE_DATA_DIRECTORY result;
 
-  if (pe->header->FileHeader.Machine == 0x8664)  // is a 64-bit PE ?
+  if (pe->header->FileHeader.Machine == IMAGE_FILE_MACHINE_AMD64)
     result = &((PIMAGE_NT_HEADERS64) pe->header)->
         OptionalHeader.DataDirectory[entry];
   else
@@ -414,7 +414,7 @@ void pe_parse(
   char section_name[IMAGE_SIZEOF_SHORT_NAME + 1];
 
 #define OptionalHeader(field) \
-  (pe->header->FileHeader.Machine == 0x8664 ? \
+  (pe->header->FileHeader.Machine == IMAGE_FILE_MACHINE_AMD64 ? \
    ((PIMAGE_NT_HEADERS64) pe->header)->OptionalHeader.field : \
      pe->header->OptionalHeader.field)