Changeset 3759 for trunk/src

Timestamp:

Jun 26, 2000, 3:21:34 PM (25 years ago)

Author:

sandervl

Message:

* empty log message *

File:

• trunk/src/kernel32/pefile.cpp (modified) (3 diffs)

trunk/src/kernel32/pefile.cpp

@@ -1 @@
-/* $Id: pefile.cpp,v 1.8 2000-05-28 16:45:12 sandervl Exp $ */
+/* $Id: pefile.cpp,v 1.9 2000-06-26 13:21:34 sandervl Exp $ */
 
 /*
@@ -59 +59 @@
 //******************************************************************************
 //******************************************************************************
-LPVOID ImageDirectoryOffset (LPVOID lpFile, DWORD dwIMAGE_DIRECTORY)
+LPVOID ImageDirectoryOffset(LPVOID lpFile, DWORD dwIMAGE_DIRECTORY)
 {
     PIMAGE_OPTIONAL_HEADER   poh = (PIMAGE_OPTIONAL_HEADER)OPTHEADEROFF (lpFile);
@@ -73 +73 @@
 
     return (LPVOID)((ULONG)lpFile + poh->DataDirectory[dwIMAGE_DIRECTORY].VirtualAddress); 
+}
+//******************************************************************************
+//******************************************************************************
+DWORD ImageDirectorySize(LPVOID lpFile, DWORD dwIMAGE_DIRECTORY)
+{
+    PIMAGE_OPTIONAL_HEADER   poh = (PIMAGE_OPTIONAL_HEADER)OPTHEADEROFF (lpFile);
+    IMAGE_SECTION_HEADER     sh;
+
+    if (dwIMAGE_DIRECTORY >= poh->NumberOfRvaAndSizes)
+        return 0;
+
+    if(GetSectionHdrByRVA(lpFile, &sh, poh->DataDirectory[dwIMAGE_DIRECTORY].VirtualAddress) == FALSE)
+    {
+        return 0;
+    }
+
+    return poh->DataDirectory[dwIMAGE_DIRECTORY].Size;
 }
 //******************************************************************************