1 files changed, 18 insertions, 0 deletions

diff --git a/src/burn/engine/core.h b/src/burn/engine/core.h
index 7e594b52..28b5ba5d 100644
--- a/src/burn/engine/core.h
+++ b/src/burn/engine/core.h
@@ -82,6 +82,16 @@ enum BURN_AU_PAUSE_ACTION
     BURN_AU_PAUSE_ACTION_IFELEVATED_NORESUME,
 };
 
+enum BURN_RESTART_STATE
+{
+    BURN_RESTART_STATE_NONE,
+    BURN_RESTART_STATE_REQUESTING,
+    BURN_RESTART_STATE_REQUESTED,
+    BURN_RESTART_STATE_INITIATING,
+    BURN_RESTART_STATE_INITIATED,
+    BURN_RESTART_STATE_BLOCKED,
+};
+
 
 // structs
 
@@ -160,6 +170,10 @@ typedef struct _BURN_ENGINE_STATE
     BURN_PIPE_CONNECTION companionConnection;
     BURN_PIPE_CONNECTION embeddedConnection;
 
+    CRITICAL_SECTION csRestartState;
+    BOOL fRestarting;
+    BURN_RESTART_STATE restartState;
+
     BOOL fCriticalShutdownInitiated;
     BURN_RESUME_MODE resumeMode;
     LPCWSTR wzRestartInitiatedPackageId;
@@ -300,6 +314,10 @@ HRESULT CoreParseCommandLine(
     __inout HANDLE* phSectionFile,
     __inout HANDLE* phSourceEngineFile
     );
+void CoreUpdateRestartState(
+    __in BURN_ENGINE_STATE* pEngineState,
+    __in BURN_RESTART_STATE restartState
+    );
 void CoreFunctionOverride(
     __in_opt PFN_CREATEPROCESSW pfnCreateProcessW,
     __in_opt PFN_PROCWAITFORCOMPLETION pfnProcWaitForCompletion