OrcCommand: WolfExecution: fix possible nullptr deref on error path

DFIR-ORC · Dec 1, 2020 · 916f079 · 916f079
1 parent 1b963b4
commit 916f079
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/OrcCommand/WolfExecution_Execute.cpp b/src/OrcCommand/WolfExecution_Execute.cpp
@@ -741,8 +741,16 @@ HRESULT WolfExecution::TerminateAllAndComplete()
 HRESULT WolfExecution::CompleteArchive(UploadMessage::ITarget* pUploadMessageQueue)
 {
     HRESULT hr = E_FAIL;
-    m_ProcessStatisticsWriter->Close();
-    m_JobStatisticsWriter->Close();
+
+    if (m_ProcessStatisticsWriter)
+    {
+        m_ProcessStatisticsWriter->Close();
+    }
+
+    if (m_JobStatisticsWriter)
+    {
+        m_JobStatisticsWriter->Close();
+    }
 
     if (VerifyFileExists(m_ProcessStatisticsOutput.Path.c_str()) == S_OK)
     {