Supply hashes to BA if present in update feed.

Fixes https://github.com/wixtoolset/issues/issues/6353.

9 files changed, 31 insertions, 6 deletions

diff --git a/src/api/burn/WixToolset.BootstrapperCore.Native/inc/BootstrapperApplication.h b/src/api/burn/WixToolset.BootstrapperCore.Native/inc/BootstrapperApplication.h
index ad3ef8a3..c65ca86b 100644
--- a/src/api/burn/WixToolset.BootstrapperCore.Native/inc/BootstrapperApplication.h
+++ b/src/api/burn/WixToolset.BootstrapperCore.Native/inc/BootstrapperApplication.h
@@ -870,6 +870,8 @@ struct BA_ONDETECTUPDATE_ARGS
     DWORD cbSize;
     LPCWSTR wzUpdateLocation;
     DWORD64 dw64Size;
+    LPCWSTR wzHash;
+    BOOTSTRAPPER_UPDATE_HASH_TYPE hashAlgorithm;
     LPCWSTR wzVersion;
     LPCWSTR wzTitle;
     LPCWSTR wzSummary;
diff --git a/src/api/burn/WixToolset.Mba.Core/BootstrapperApplication.cs b/src/api/burn/WixToolset.Mba.Core/BootstrapperApplication.cs
index fe9322ce..ecc99069 100644
--- a/src/api/burn/WixToolset.Mba.Core/BootstrapperApplication.cs
+++ b/src/api/burn/WixToolset.Mba.Core/BootstrapperApplication.cs
@@ -1451,9 +1451,9 @@ namespace WixToolset.Mba.Core
             return args.HResult;
         }
 
-        int IBootstrapperApplication.OnDetectUpdate(string wzUpdateLocation, long dw64Size, string wzVersion, string wzTitle, string wzSummary, string wzContentType, string wzContent, ref bool fCancel, ref bool fStopProcessingUpdates)
+        int IBootstrapperApplication.OnDetectUpdate(string wzUpdateLocation, long dw64Size, string wzHash, UpdateHashType hashAlgorithm, string wzVersion, string wzTitle, string wzSummary, string wzContentType, string wzContent, ref bool fCancel, ref bool fStopProcessingUpdates)
         {
-            DetectUpdateEventArgs args = new DetectUpdateEventArgs(wzUpdateLocation, dw64Size, wzVersion, wzTitle, wzSummary, wzContentType, wzContent, fCancel, fStopProcessingUpdates);
+            DetectUpdateEventArgs args = new DetectUpdateEventArgs(wzUpdateLocation, dw64Size, wzHash, hashAlgorithm, wzVersion, wzTitle, wzSummary, wzContentType, wzContent, fCancel, fStopProcessingUpdates);
             this.OnDetectUpdate(args);
 
             fCancel = args.Cancel;
diff --git a/src/api/burn/WixToolset.Mba.Core/EventArgs.cs b/src/api/burn/WixToolset.Mba.Core/EventArgs.cs
index 2b414d91..48850fd7 100644
--- a/src/api/burn/WixToolset.Mba.Core/EventArgs.cs
+++ b/src/api/burn/WixToolset.Mba.Core/EventArgs.cs
@@ -5,6 +5,7 @@ namespace WixToolset.Mba.Core
     using System;
     using System.Collections.Generic;
     using System.Collections.ObjectModel;
+    using System.Runtime.InteropServices;
 
     /// <summary>
     /// Base class for BA <see cref="EventArgs"/> classes.
@@ -328,11 +329,13 @@ namespace WixToolset.Mba.Core
     public class DetectUpdateEventArgs : CancellableHResultEventArgs
     {
         /// <summary />
-        public DetectUpdateEventArgs(string updateLocation, long size, string version, string title, string summary, string contentType, string content, bool cancelRecommendation, bool stopRecommendation)
+        public DetectUpdateEventArgs(string updateLocation, long size, string hash, UpdateHashType hashAlgorithm, string version, string title, string summary, string contentType, string content, bool cancelRecommendation, bool stopRecommendation)
             : base(cancelRecommendation)
         {
             this.UpdateLocation = updateLocation;
             this.Size = size;
+            this.Hash = hash;
+            this.HashAlgorithm = hashAlgorithm;
             this.Version = version;
             this.Title = title;
             this.Summary = summary;
@@ -352,6 +355,16 @@ namespace WixToolset.Mba.Core
         public long Size { get; private set; }
 
         /// <summary>
+        /// File hash of the updated bundle.
+        /// </summary>
+        public string Hash { get; }
+
+        /// <summary>
+        /// The algorithm of the updated bundle's hash.
+        /// </summary>
+        public UpdateHashType HashAlgorithm { get; }
+
+        /// <summary>
         /// Gets the version of the updated bundle.
         /// </summary>
         public string Version { get; private set; }
diff --git a/src/api/burn/WixToolset.Mba.Core/IBootstrapperApplication.cs b/src/api/burn/WixToolset.Mba.Core/IBootstrapperApplication.cs
index 87da2191..daa95e17 100644
--- a/src/api/burn/WixToolset.Mba.Core/IBootstrapperApplication.cs
+++ b/src/api/burn/WixToolset.Mba.Core/IBootstrapperApplication.cs
@@ -106,6 +106,8 @@ namespace WixToolset.Mba.Core
         /// <param name="wzUpdateLocation"></param>
         /// <param name="dw64Size"></param>
         /// <param name="wzVersion"></param>
+        /// <param name="wzHash"></param>
+        /// <param name="hashAlgorithm"></param>
         /// <param name="wzTitle"></param>
         /// <param name="wzSummary"></param>
         /// <param name="wzContentType"></param>
@@ -118,6 +120,8 @@ namespace WixToolset.Mba.Core
         int OnDetectUpdate(
             [MarshalAs(UnmanagedType.LPWStr)] string wzUpdateLocation,
             [MarshalAs(UnmanagedType.U8)] long dw64Size,
+            [MarshalAs(UnmanagedType.LPWStr)] string wzHash,
+            [MarshalAs(UnmanagedType.U4)] UpdateHashType hashAlgorithm,
             [MarshalAs(UnmanagedType.LPWStr)] string wzVersion,
             [MarshalAs(UnmanagedType.LPWStr)] string wzTitle,
             [MarshalAs(UnmanagedType.LPWStr)] string wzSummary,
diff --git a/src/api/burn/WixToolset.Mba.Core/IBootstrapperEngine.cs b/src/api/burn/WixToolset.Mba.Core/IBootstrapperEngine.cs
index ebea6c4f..fdc9c3f6 100644
--- a/src/api/burn/WixToolset.Mba.Core/IBootstrapperEngine.cs
+++ b/src/api/burn/WixToolset.Mba.Core/IBootstrapperEngine.cs
@@ -421,9 +421,9 @@ namespace WixToolset.Mba.Core
         None,
 
         /// <summary>
-        /// SHA-1 based hash provided.
+        /// SHA-512 based hash provided.
         /// </summary>
-        Sha1,
+        Sha512,
     }
 
     /// <summary>
diff --git a/src/api/burn/balutil/inc/BalBaseBAFunctions.h b/src/api/burn/balutil/inc/BalBaseBAFunctions.h
index 3f99673d..9a8ac87e 100644
--- a/src/api/burn/balutil/inc/BalBaseBAFunctions.h
+++ b/src/api/burn/balutil/inc/BalBaseBAFunctions.h
@@ -133,6 +133,8 @@ public: // IBootstrapperApplication
     virtual STDMETHODIMP OnDetectUpdate(
         __in_z LPCWSTR /*wzUpdateLocation*/,
         __in DWORD64 /*dw64Size*/,
+        __in_z_opt LPCWSTR /*wzHash*/,
+        __in BOOTSTRAPPER_UPDATE_HASH_TYPE /*hashAlgorithm*/,
         __in LPCWSTR /*wzVersion*/,
         __in_z LPCWSTR /*wzTitle*/,
         __in_z LPCWSTR /*wzSummary*/,
diff --git a/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h b/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h
index c3223bee..58cc0673 100644
--- a/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h
+++ b/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h
@@ -132,6 +132,8 @@ public: // IBootstrapperApplication
     virtual STDMETHODIMP OnDetectUpdate(
         __in_z LPCWSTR /*wzUpdateLocation*/,
         __in DWORD64 /*dw64Size*/,
+        __in_z_opt LPCWSTR /*wzHash*/,
+        __in BOOTSTRAPPER_UPDATE_HASH_TYPE /*hashAlgorithm*/,
         __in LPCWSTR /*wzVersion*/,
         __in_z LPCWSTR /*wzTitle*/,
         __in_z LPCWSTR /*wzSummary*/,
diff --git a/src/api/burn/balutil/inc/BalBaseBootstrapperApplicationProc.h b/src/api/burn/balutil/inc/BalBaseBootstrapperApplicationProc.h
index 3054731f..62cb85bc 100644
--- a/src/api/burn/balutil/inc/BalBaseBootstrapperApplicationProc.h
+++ b/src/api/burn/balutil/inc/BalBaseBootstrapperApplicationProc.h
@@ -87,7 +87,7 @@ static HRESULT BalBaseBAProcOnDetectUpdate(
     __inout BA_ONDETECTUPDATE_RESULTS* pResults
     )
 {
-    return pBA->OnDetectUpdate(pArgs->wzUpdateLocation, pArgs->dw64Size, pArgs->wzVersion, pArgs->wzTitle, pArgs->wzSummary, pArgs->wzContentType, pArgs->wzContent, &pResults->fCancel, &pResults->fStopProcessingUpdates);
+    return pBA->OnDetectUpdate(pArgs->wzUpdateLocation, pArgs->dw64Size, pArgs->wzHash, pArgs->hashAlgorithm, pArgs->wzVersion, pArgs->wzTitle, pArgs->wzSummary, pArgs->wzContentType, pArgs->wzContent, &pResults->fCancel, &pResults->fStopProcessingUpdates);
 }
 
 static HRESULT BalBaseBAProcOnDetectUpdateComplete(
diff --git a/src/api/burn/balutil/inc/IBootstrapperApplication.h b/src/api/burn/balutil/inc/IBootstrapperApplication.h
index 0362e171..fba919ea 100644
--- a/src/api/burn/balutil/inc/IBootstrapperApplication.h
+++ b/src/api/burn/balutil/inc/IBootstrapperApplication.h
@@ -63,6 +63,8 @@ DECLARE_INTERFACE_IID_(IBootstrapperApplication, IUnknown, "53C31D56-49C0-426B-A
     STDMETHOD(OnDetectUpdate)(
         __in_z_opt LPCWSTR wzUpdateLocation,
         __in DWORD64 dw64Size,
+        __in_z_opt LPCWSTR wzHash,
+        __in BOOTSTRAPPER_UPDATE_HASH_TYPE hashAlgorithm,
         __in_z LPCWSTR wzVersion,
         __in_z_opt LPCWSTR wzTitle,
         __in_z_opt LPCWSTR wzSummary,