From 6c740c1070475a8a894393a186cdcdd8a8a81112 Mon Sep 17 00:00:00 2001
From: Rob Mensching <rob@firegiant.com>
Date: Wed, 14 Apr 2021 12:18:39 -0700
Subject: Allow BA to update feed source

Fixes wixtoolset/issues#5568
---
 src/balutil/BalBootstrapperEngine.cpp | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'src/balutil/BalBootstrapperEngine.cpp')

diff --git a/src/balutil/BalBootstrapperEngine.cpp b/src/balutil/BalBootstrapperEngine.cpp
index 8e40f7ae..dda98cb9 100644
--- a/src/balutil/BalBootstrapperEngine.cpp
+++ b/src/balutil/BalBootstrapperEngine.cpp
@@ -535,6 +535,21 @@ public: // IBootstrapperEngine
         return m_pfnBAEngineProc(BOOTSTRAPPER_ENGINE_MESSAGE_LAUNCHAPPROVEDEXE, &args, &results, m_pvBAEngineProcContext);
     }
 
+    virtual STDMETHODIMP SetUpdateSource(
+        __in_z LPCWSTR wzUrl
+        )
+    {
+        BAENGINE_SETUPDATESOURCE_ARGS args = { };
+        BAENGINE_SETUPDATESOURCE_RESULTS results = { };
+
+        args.cbSize = sizeof(args);
+        args.wzUrl = wzUrl;
+
+        results.cbSize = sizeof(results);
+
+        return m_pfnBAEngineProc(BOOTSTRAPPER_ENGINE_MESSAGE_SETUPDATESOURCE, &args, &results, m_pvBAEngineProcContext);
+    }
+
     virtual STDMETHODIMP CompareVersions(
         __in_z LPCWSTR wzVersion1,
         __in_z LPCWSTR wzVersion2,
-- 
cgit v1.2.3-55-g6feb