1 files changed, 11 insertions, 2 deletions

diff --git a/src/api/burn/WixToolset.BootstrapperCore.Native/inc/BootstrapperApplication.h b/src/api/burn/WixToolset.BootstrapperCore.Native/inc/BootstrapperApplication.h
index 2a6d5c8a..56f6b361 100644
--- a/src/api/burn/WixToolset.BootstrapperCore.Native/inc/BootstrapperApplication.h
+++ b/src/api/burn/WixToolset.BootstrapperCore.Native/inc/BootstrapperApplication.h
@@ -20,6 +20,13 @@ enum BOOTSTRAPPER_RESTART
     BOOTSTRAPPER_RESTART_ALWAYS,
 };
 
+enum BOOTSTRAPPER_REGISTRATION_TYPE
+{
+    BOOTSTRAPPER_REGISTRATION_TYPE_NONE, // The engine will ignore NONE if it recommended INPROGRESS or FULL.
+    BOOTSTRAPPER_REGISTRATION_TYPE_INPROGRESS,
+    BOOTSTRAPPER_REGISTRATION_TYPE_FULL,
+};
+
 enum BOOTSTRAPPER_RESUME_TYPE
 {
     BOOTSTRAPPER_RESUME_TYPE_NONE,
@@ -1163,12 +1170,14 @@ struct BA_ONPROGRESS_RESULTS
 struct BA_ONREGISTERBEGIN_ARGS
 {
     DWORD cbSize;
+    BOOTSTRAPPER_REGISTRATION_TYPE recommendedRegistrationType;
 };
 
 struct BA_ONREGISTERBEGIN_RESULTS
 {
     DWORD cbSize;
     BOOL fCancel;
+    BOOTSTRAPPER_REGISTRATION_TYPE registrationType;
 };
 
 struct BA_ONREGISTERCOMPLETE_ARGS
@@ -1262,13 +1271,13 @@ struct BA_ONSYSTEMSHUTDOWN_RESULTS
 struct BA_ONUNREGISTERBEGIN_ARGS
 {
     DWORD cbSize;
-    BOOL fKeepRegistration;
+    BOOTSTRAPPER_REGISTRATION_TYPE recommendedRegistrationType;
 };
 
 struct BA_ONUNREGISTERBEGIN_RESULTS
 {
     DWORD cbSize;
-    BOOL fForceKeepRegistration;
+    BOOTSTRAPPER_REGISTRATION_TYPE registrationType;
 };
 
 struct BA_ONUNREGISTERCOMPLETE_ARGS