Update buffer.cpp (#200)

set error to PEERR_OPEN if CreateFileA or GetFileSize fails
trailofbits · Apr 11, 2024 · 015345d · 015345d
1 parent b43ff37
commit 015345d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pe-parser-library/src/buffer.cpp b/pe-parser-library/src/buffer.cpp
@@ -202,13 +202,15 @@ bounded_buffer *readFileToFileBuffer(const char *filePath) {
                          FILE_ATTRIBUTE_NORMAL,
                          nullptr);
   if (h == INVALID_HANDLE_VALUE) {
+    PE_ERR(PEERR_OPEN);
     return nullptr;
   }
 
   DWORD fileSize = GetFileSize(h, nullptr);
 
   if (fileSize == INVALID_FILE_SIZE) {
     CloseHandle(h);
+    PE_ERR(PEERR_OPEN);
     return nullptr;
   }