imagehlp: Correctly handle files without NT headers in MapAndLoad.

4950d3de · Eric Pouech · Alexandre Julliard · 5dbb2517 · 4950d3de
Commit 4950d3de authored Jan 03, 2007 by Eric Pouech Committed by Alexandre Julliard Jan 04, 2007
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

access.c dlls/imagehlp/access.c +6 -1

No files found.
--- a/dlls/imagehlp/access.c
+++ b/dlls/imagehlp/access.c
@@ -187,7 +187,12 @@ BOOL WINAPI MapAndLoad(LPSTR pszImageName, LPSTR pszDllPath, PLOADED_IMAGE pLoad
        goto Error;
    }

-    pNtHeader = RtlImageNtHeader(mapping);
+    if (!(pNtHeader = RtlImageNtHeader(mapping)))
+    {
+        WARN("Not an NT header\n");
+        UnmapViewOfFile(mapping);
+        goto Error;
+    }

    pLoadedImage->ModuleName       = HeapAlloc(GetProcessHeap(), 0,
                                               strlen(szFileName) + 1);