Only block shutdown during Apply.

1 files changed, 4 insertions, 0 deletions

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);