Add checkpoint so Exe and Msu packages rollback after being cancelled.

Fixes 5950

1 files changed, 18 insertions, 0 deletions

diff --git a/src/burn/engine/elevation.cpp b/src/burn/engine/elevation.cpp
index 3c2872f1..ea56e242 100644
--- a/src/burn/engine/elevation.cpp
+++ b/src/burn/engine/elevation.cpp
@@ -44,6 +44,8 @@ typedef enum _BURN_ELEVATION_MESSAGE_TYPE
     BURN_ELEVATION_MESSAGE_TYPE_BURN_CACHE_SUCCESS,
     BURN_ELEVATION_MESSAGE_TYPE_EXECUTE_PROGRESS,
     BURN_ELEVATION_MESSAGE_TYPE_EXECUTE_PROCESS_CANCEL,
+    BURN_ELEVATION_MESSAGE_TYPE_EXECUTE_PROCESS_STARTED,
+    BURN_ELEVATION_MESSAGE_TYPE_EXECUTE_PROCESS_COMPLETED,
     BURN_ELEVATION_MESSAGE_TYPE_EXECUTE_ERROR,
     BURN_ELEVATION_MESSAGE_TYPE_EXECUTE_MSI_MESSAGE,
     BURN_ELEVATION_MESSAGE_TYPE_EXECUTE_MSI_FILES_IN_USE,
@@ -1823,6 +1825,14 @@ static HRESULT ProcessGenericExecuteMessages(
         ExitOnFailure(hr, "Failed to read processId.");
         break;
 
+    case BURN_ELEVATION_MESSAGE_TYPE_EXECUTE_PROCESS_STARTED:
+        message.type = GENERIC_EXECUTE_MESSAGE_PROCESS_STARTED;
+        break;
+
+    case BURN_ELEVATION_MESSAGE_TYPE_EXECUTE_PROCESS_COMPLETED:
+        message.type = GENERIC_EXECUTE_MESSAGE_PROCESS_COMPLETED;
+        break;
+
     case BURN_ELEVATION_MESSAGE_TYPE_EXECUTE_ERROR:
         message.type = GENERIC_EXECUTE_MESSAGE_ERROR;
 
@@ -3465,6 +3475,14 @@ static int GenericExecuteMessageHandler(
         dwMessage = BURN_ELEVATION_MESSAGE_TYPE_EXECUTE_PROCESS_CANCEL;
         break;
 
+    case GENERIC_EXECUTE_MESSAGE_PROCESS_STARTED:
+        dwMessage = BURN_ELEVATION_MESSAGE_TYPE_EXECUTE_PROCESS_STARTED;
+        break;
+
+    case GENERIC_EXECUTE_MESSAGE_PROCESS_COMPLETED:
+        dwMessage = BURN_ELEVATION_MESSAGE_TYPE_EXECUTE_PROCESS_COMPLETED;
+        break;
+
     case GENERIC_EXECUTE_MESSAGE_ERROR:
         // serialize message data
         hr = BuffWriteNumber(&pbData, &cbData, pMessage->error.dwErrorCode);