1 files changed, 15 insertions, 0 deletions

diff --git a/src/burn/engine/core.cpp b/src/burn/engine/core.cpp
index 25124c8a..3c1ed117 100644
--- a/src/burn/engine/core.cpp
+++ b/src/burn/engine/core.cpp
@@ -1963,6 +1963,21 @@ LExit:
     return hr;
 }
 
+extern "C" void CoreUpdateRestartState(
+    __in BURN_ENGINE_STATE* pEngineState,
+    __in BURN_RESTART_STATE restartState
+    )
+{
+    ::EnterCriticalSection(&pEngineState->csRestartState);
+
+    if (pEngineState->fRestarting && restartState > pEngineState->restartState)
+    {
+        pEngineState->restartState = restartState;
+    }
+
+    ::LeaveCriticalSection(&pEngineState->csRestartState);
+}
+
 extern "C" void CoreFunctionOverride(
     __in_opt PFN_CREATEPROCESSW pfnCreateProcessW,
     __in_opt PFN_PROCWAITFORCOMPLETION pfnProcWaitForCompletion