From 29f7e00586412163a20e298fbf84505f8a917425 Mon Sep 17 00:00:00 2001
From: Sean Hall <r.sean.hall@gmail.com>
Date: Tue, 3 May 2022 15:30:50 -0500
Subject: Only block shutdown during Apply.

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

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

diff --git a/src/burn/engine/core.cpp b/src/burn/engine/core.cpp
index ec996c4c..19b739f9 100644
--- a/src/burn/engine/core.cpp
+++ b/src/burn/engine/core.cpp
@@ -681,6 +681,8 @@ extern "C" HRESULT CoreApply(
     hr = ApplyLock(FALSE, &hLock);
     ExitOnFailure(hr, "Another per-user setup is already executing.");
 
+    pEngineState->plan.fApplying = TRUE;
+
     // Initialize only after getting a lock.
     fApplyInitialize = TRUE;
     ApplyInitialize();
@@ -814,6 +816,8 @@ LExit:
         ApplyUninitialize();
     }
 
+    pEngineState->plan.fApplying = FALSE;
+
     if (hLock)
     {
         ::ReleaseMutex(hLock);
-- 
cgit v1.2.3-55-g6feb