diff --git a/reactos/base/applications/screenshot/screenshot.c b/reactos/base/applications/screenshot/screenshot.c index 9700fa121bf..d6057d716b8 100644 --- a/reactos/base/applications/screenshot/screenshot.c +++ b/reactos/base/applications/screenshot/screenshot.c @@ -347,6 +347,8 @@ int WINAPI WinMain(HINSTANCE hInstance, pScrSht = HeapAlloc(GetProcessHeap(), 0, sizeof(SCREENSHOT)); + if (pScrSht == NULL) + return -1; if (bFullScreen) { @@ -358,7 +360,13 @@ int WINAPI WinMain(HINSTANCE hInstance, } if (pScrSht->hSelf == NULL) + { + HeapFree(GetProcessHeap(), + 0, + pScrSht); + return -1; + } if (CaptureScreen(pScrSht)) {