Use IMAGE_NUMBEROF_DIRECTORY_ENTRIES

Is defined in winnt.h, and pefile set it but was not using it.
erocarrera · Dec 13, 2024 · a65fad0 · a65fad0
1 parent 4b3b1e2
commit a65fad0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pefile.py b/pefile.py
@@ -3306,7 +3306,9 @@ def __parse__(self, fname, data, fast_load):
             # if we go beyond that, we assume the number of directories
             # is wrong and stop processing
             if offset >= (
-                optional_header_offset + self.OPTIONAL_HEADER.sizeof() + 8 * 16
+                optional_header_offset
+                + self.OPTIONAL_HEADER.sizeof()
+                + IMAGE_NUMBEROF_DIRECTORY_ENTRIES * 8
             ):
                 break