1 files changed, 27 insertions, 8 deletions

diff --git a/src/api/burn/WixToolset.Mba.Core/IBootstrapperApplication.cs b/src/api/burn/WixToolset.Mba.Core/IBootstrapperApplication.cs
index 530fb1a9..e6e03906 100644
--- a/src/api/burn/WixToolset.Mba.Core/IBootstrapperApplication.cs
+++ b/src/api/burn/WixToolset.Mba.Core/IBootstrapperApplication.cs
@@ -500,12 +500,12 @@ namespace WixToolset.Mba.Core
         /// <summary>
         /// See <see cref="IDefaultBootstrapperApplication.RegisterBegin"/>.
         /// </summary>
-        /// <param name="fCancel"></param>
-        /// <returns></returns>
         [PreserveSig]
         [return: MarshalAs(UnmanagedType.I4)]
         int OnRegisterBegin(
-            [MarshalAs(UnmanagedType.Bool)] ref bool fCancel
+            [MarshalAs(UnmanagedType.I4)] RegistrationType recommendedRegistrationType,
+            [MarshalAs(UnmanagedType.Bool)] ref bool fCancel,
+            [MarshalAs(UnmanagedType.I4)] ref RegistrationType pRegistrationType
             );
 
         /// <summary>
@@ -820,14 +820,11 @@ namespace WixToolset.Mba.Core
         /// <summary>
         /// See <see cref="IDefaultBootstrapperApplication.UnregisterBegin"/>.
         /// </summary>
-        /// <param name="fKeepRegistration"></param>
-        /// <param name="fForceKeepRegistration"></param>
-        /// <returns></returns>
         [PreserveSig]
         [return: MarshalAs(UnmanagedType.I4)]
         int OnUnregisterBegin(
-            [MarshalAs(UnmanagedType.Bool)] bool fKeepRegistration,
-            [MarshalAs(UnmanagedType.Bool)] ref bool fForceKeepRegistration
+            [MarshalAs(UnmanagedType.I4)] RegistrationType recommendedRegistrationType,
+            [MarshalAs(UnmanagedType.I4)] ref RegistrationType pRegistrationType
             );
 
         /// <summary>
@@ -1260,6 +1257,28 @@ namespace WixToolset.Mba.Core
     }
 
     /// <summary>
+    /// The display name to use when registering in Add/Remove Programs.
+    /// </summary>
+    public enum RegistrationType
+    {
+        /// <summary>
+        /// No registration.
+        /// The engine will ignore None if it recommended InProgress or Full.
+        /// </summary>
+        None,
+
+        /// <summary>
+        /// The in-progress display name.
+        /// </summary>
+        InProgress,
+
+        /// <summary>
+        /// The default display name.
+        /// </summary>
+        Full,
+    }
+
+    /// <summary>
     /// Result codes (based on Dialog Box Command IDs from WinUser.h).
     /// </summary>
     public enum Result