2 files changed, 21 insertions, 3 deletions

diff --git a/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h b/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h
index aa1ca56f..c3223bee 100644
--- a/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h
+++ b/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h
@@ -451,8 +451,8 @@ public: // IBootstrapperApplication
         __in BOOTSTRAPPER_ERROR_TYPE errorType,
         __in_z LPCWSTR wzPackageId,
         __in DWORD dwCode,
-        __in_z LPCWSTR /*wzError*/,
-        __in DWORD /*dwUIHint*/,
+        __in_z LPCWSTR wzError,
+        __in DWORD dwUIHint,
         __in DWORD /*cData*/,
         __in_ecount_z_opt(cData) LPCWSTR* /*rgwzData*/,
         __in int /*nRecommendation*/,
@@ -461,7 +461,15 @@ public: // IBootstrapperApplication
     {
         BalRetryErrorOccurred(wzPackageId, dwCode);
 
-        if (CheckCanceled())
+        if (BOOTSTRAPPER_DISPLAY_EMBEDDED == m_display)
+        {
+            HRESULT hr = m_pEngine->SendEmbeddedError(dwCode, wzError, dwUIHint, pResult);
+            if (FAILED(hr))
+            {
+                *pResult = IDERROR;
+            }
+        }
+        else if (CheckCanceled())
         {
             *pResult = IDCANCEL;
         }
diff --git a/src/api/burn/balutil/inc/balinfo.h b/src/api/burn/balutil/inc/balinfo.h
index cd61f34d..818ff5ef 100644
--- a/src/api/burn/balutil/inc/balinfo.h
+++ b/src/api/burn/balutil/inc/balinfo.h
@@ -20,6 +20,15 @@ typedef enum BAL_INFO_PACKAGE_TYPE
     BAL_INFO_PACKAGE_TYPE_BUNDLE_CHAIN,
 } BAL_INFO_PACKAGE_TYPE;
 
+typedef enum _BAL_INFO_PRIMARY_PACKAGE_TYPE
+{
+    BAL_INFO_PRIMARY_PACKAGE_TYPE_NONE,
+    BAL_INFO_PRIMARY_PACKAGE_TYPE_DEFAULT,
+    BAL_INFO_PRIMARY_PACKAGE_TYPE_X86,
+    BAL_INFO_PRIMARY_PACKAGE_TYPE_X64,
+    BAL_INFO_PRIMARY_PACKAGE_TYPE_ARM64,
+} BAL_INFO_PRIMARY_PACKAGE_TYPE;
+
 typedef enum _BAL_INFO_RESTART
 {
     BAL_INFO_RESTART_UNKNOWN,
@@ -54,6 +63,7 @@ typedef struct _BAL_INFO_PACKAGE
     BOOL fPrereqPackage;
     LPWSTR sczPrereqLicenseFile;
     LPWSTR sczPrereqLicenseUrl;
+    BAL_INFO_PRIMARY_PACKAGE_TYPE primaryPackageType;
     LPVOID pvCustomData;
 } BAL_INFO_PACKAGE;