From bbc36178172e4198351f2f84a91faf56c65e1475 Mon Sep 17 00:00:00 2001
From: Sean Hall <r.sean.hall@gmail.com>
Date: Fri, 19 Aug 2022 13:48:52 -0500
Subject: Create log for elevated process.

---
 src/burn/engine/engine.cpp | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'src/burn/engine/engine.cpp')

diff --git a/src/burn/engine/engine.cpp b/src/burn/engine/engine.cpp
index 2d2da1d5..323d2c3a 100644
--- a/src/burn/engine/engine.cpp
+++ b/src/burn/engine/engine.cpp
@@ -106,6 +106,7 @@ extern "C" HRESULT EngineRun(
     LPWSTR sczExePath = NULL;
     BOOL fRunUntrusted = FALSE;
     BOOL fRunNormal = FALSE;
+    BOOL fRunElevated = FALSE;
     BOOL fRunRunOnce = FALSE;
     BOOL fRestart = FALSE;
 
@@ -220,6 +221,8 @@ extern "C" HRESULT EngineRun(
         break;
 
     case BURN_MODE_ELEVATED:
+        fRunElevated = TRUE;
+
         hr = RunElevated(hInstance, wzCommandLine, &engineState);
         ExitOnFailure(hr, "Failed to run per-machine mode.");
         break;
@@ -320,6 +323,11 @@ LExit:
     {
         LogId(REPORT_STANDARD, MSG_EXITING_RUN_ONCE, FAILED(hr) ? (int)hr : *pdwExitCode);
     }
+    else if (fRunElevated)
+    {
+        LogId(REPORT_STANDARD, MSG_EXITING_ELEVATED, FAILED(hr) ? (int)hr : *pdwExitCode);
+    }
+
 
     if (fLogInitialized)
     {
@@ -639,6 +647,10 @@ static HRESULT RunElevated(
     HRESULT hr = S_OK;
     HANDLE hLock = NULL;
 
+    // Initialize logging.
+    hr = LoggingOpen(&pEngineState->log, &pEngineState->internalCommand, &pEngineState->command, &pEngineState->variables, pEngineState->registration.sczDisplayName);
+    ExitOnFailure(hr, "Failed to open elevated log.");
+
     // connect to per-user process
     hr = PipeChildConnect(&pEngineState->companionConnection, TRUE);
     ExitOnFailure(hr, "Failed to connect to unelevated process.");
-- 
cgit v1.2.3-55-g6feb