Launch a thread to log premature termination of companion elevated process

Author: nirbar

README.md

@@ -22,4 +22,5 @@ This repository contains the PanelSwWix4: A custom WiX Toolset codebase
 - [7778](https://github.com/wixtoolset/issues/issues/7778): Bundle/@Wix3DependencyMode=yes/no. If yes, then default MSI/MSP provider key uses the ProductCode/PatchCode. Use only if a WiX4 bundle chains packages of a WiX3 bundle.
 - [4889](https://github.com/wixtoolset/issues/issues/4889): Support custom container compressions using bundle extensions
 - Not overwriting log files when retrying to execute a package
-- Support sending custom messages on embedded pipe
+- Support sending custom messages on embedded pipe
+- Best effort to log premature termination of companion process

src/burn/engine/core.cpp

@@ -86,6 +86,9 @@ static void LogRelatedBundles(
 static void LogRollbackBoundary(
     __in const BURN_ROLLBACK_BOUNDARY* pRollbackBoundary
     );
+static DWORD WINAPI MonitorCompanionProcessThreadProc(
+    __in LPVOID lpParameter
+    );
 
 
 // function definitions
@@ -622,12 +625,52 @@ extern "C" HRESULT CoreElevate(
 
         pEngineState->hUnelevatedLoggingThread = ::CreateThread(NULL, 0, LoggingThreadProc, pEngineState, 0, NULL);
         ExitOnNullWithLastError(pEngineState->hUnelevatedLoggingThread, hr, "Failed to create unelevated logging thread.");
+
+        // Best effort to log premature termination of the elevated process
+        pEngineState->companionConnection.hProcess = ::OpenProcess(SYNCHRONIZE, FALSE, pEngineState->companionConnection.dwProcessId);
+        if (pEngineState->companionConnection.hProcess)
+        {
+            pEngineState->companionConnection.hQuitRequested = ::CreateEvent(NULL, TRUE, FALSE, NULL);
+            if (pEngineState->companionConnection.hQuitRequested)
+            {
+                pEngineState->companionConnection.hQuitMonitorThread = ::CreateThread(NULL, 0, MonitorCompanionProcessThreadProc, &pEngineState->companionConnection, 0, NULL);
+            }
+        }
     }
 
 LExit:
     return hr;
 }
 
+static DWORD WINAPI MonitorCompanionProcessThreadProc(
+    __in LPVOID lpParameter
+    )
+{
+    BURN_PIPE_CONNECTION* pConnection = (BURN_PIPE_CONNECTION*)lpParameter;
+    HANDLE rghHandles[2] = { pConnection->hQuitRequested, pConnection->hProcess };
+    DWORD dwRes = ERROR_SUCCESS;
+    HRESULT hr = S_OK;
+
+    dwRes = ::WaitForMultipleObjects(ARRAYSIZE(rghHandles), rghHandles, FALSE, INFINITE);
+    switch (dwRes)
+    {
+        case WAIT_OBJECT_0:
+            hr = S_OK;
+            break;
+        case WAIT_OBJECT_0 + 1:
+            LogRedirect(NULL, NULL);
+            hr = E_SUSPECTED_AV_INTERFERENCE;
+            ExitOnFailure(hr, "Companion elevated process has been terminated prematurely.");
+            break;
+        default:
+            ExitWithLastError(hr, "Failed to monitor proper termination of companion process.");
+            break;
+    }
+
+LExit:
+    return HRESULT_CODE(hr);
+}
+
 extern "C" HRESULT CoreApply(
     __in BURN_ENGINE_STATE* pEngineState,
     __in_opt HWND hwndParent

src/burn/engine/pipe.cpp

@@ -63,6 +63,8 @@ void PipeConnectionUninitialize(
     ReleaseFileHandle(pConnection->hLoggingPipe);
     ReleaseFileHandle(pConnection->hCachePipe);
     ReleaseFileHandle(pConnection->hPipe);
+    ReleaseHandle(pConnection->hQuitRequested);
+    ReleaseHandle(pConnection->hQuitMonitorThread);
     ReleaseHandle(pConnection->hProcess);
     ReleaseStr(pConnection->sczSecret);
     ReleaseStr(pConnection->sczName);
@@ -462,6 +464,11 @@ extern "C" HRESULT PipeTerminateChildProcess(
     SIZE_T cbData = 0;
     BOOL fTimedOut = FALSE;
 
+    if (pConnection->hQuitRequested)
+    {
+        ::SetEvent(pConnection->hQuitRequested);
+    }
+
     // Prepare the exit message.
     hr = BuffWriteNumber(&pbData, &cbData, dwParentExitCode);
     ExitOnFailure(hr, "Failed to write exit code to message buffer.");
@@ -523,7 +530,6 @@ extern "C" HRESULT PipeChildConnect(
 {
     Assert(pConnection->sczName);
     Assert(pConnection->sczSecret);
-    Assert(!pConnection->hProcess);
     Assert(INVALID_HANDLE_VALUE == pConnection->hPipe);
     Assert(INVALID_HANDLE_VALUE == pConnection->hCachePipe);
     Assert(INVALID_HANDLE_VALUE == pConnection->hLoggingPipe);
@@ -591,8 +597,11 @@ extern "C" HRESULT PipeChildConnect(
         ExitOnFailure(hr, "Failed to verify parent logging pipe: %ls", sczPipeName);
     }
 
-    pConnection->hProcess = ::OpenProcess(SYNCHRONIZE, FALSE, pConnection->dwProcessId);
-    ExitOnNullWithLastError(pConnection->hProcess, hr, "Failed to open companion process with PID: %u", pConnection->dwProcessId);
+    if (!pConnection->hProcess)
+    {
+        pConnection->hProcess = ::OpenProcess(SYNCHRONIZE, FALSE, pConnection->dwProcessId);
+        ExitOnNullWithLastError(pConnection->hProcess, hr, "Failed to open companion process with PID: %u", pConnection->dwProcessId);
+    }
 
 LExit:
     ReleaseStr(sczPipeName);

src/burn/engine/pipe.h

@@ -27,6 +27,8 @@ typedef struct _BURN_PIPE_CONNECTION
     HANDLE hPipe;
     HANDLE hCachePipe;
     HANDLE hLoggingPipe;
+    HANDLE hQuitRequested;
+    HANDLE hQuitMonitorThread;
 } BURN_PIPE_CONNECTION;
 
 typedef enum _BURN_PIPE_MESSAGE_TYPE : DWORD