From d32f770ca05748df9e356444c7e617d5eeedb60c Mon Sep 17 00:00:00 2001
From: Rob Mensching <rob@firegiant.com>
Date: Tue, 13 Apr 2021 18:16:55 -0700
Subject: Allow BA to update feed source

Fixes wixtoolset/issues#5568
---
 src/engine/EngineForApplication.cpp | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'src/engine/EngineForApplication.cpp')

diff --git a/src/engine/EngineForApplication.cpp b/src/engine/EngineForApplication.cpp
index 361e0f4e..83d88ba1 100644
--- a/src/engine/EngineForApplication.cpp
+++ b/src/engine/EngineForApplication.cpp
@@ -411,6 +411,22 @@ LExit:
     return hr;
 }
 
+static HRESULT BAEngineSetUpdateSource(
+    __in BOOTSTRAPPER_ENGINE_CONTEXT* pContext,
+    __in const LPVOID pvArgs,
+    __inout LPVOID pvResults
+    )
+{
+    HRESULT hr = S_OK;
+    ValidateMessageArgs(hr, pvArgs, BAENGINE_SETUPDATESOURCE_ARGS, pArgs);
+    ValidateMessageResults(hr, pvResults, BAENGINE_SETUPDATESOURCE_RESULTS, pResults);
+
+    hr = ExternalEngineSetUpdateSource(pContext->pEngineState, pArgs->wzUrl);
+
+LExit:
+    return hr;
+}
+
 HRESULT WINAPI EngineForApplicationProc(
     __in BOOTSTRAPPER_ENGINE_MESSAGE message,
     __in const LPVOID pvArgs,
@@ -497,6 +513,9 @@ HRESULT WINAPI EngineForApplicationProc(
     case BOOTSTRAPPER_ENGINE_MESSAGE_LAUNCHAPPROVEDEXE:
         hr = BAEngineLaunchApprovedExe(pContext, pvArgs, pvResults);
         break;
+    case BOOTSTRAPPER_ENGINE_MESSAGE_SETUPDATESOURCE:
+        hr = BAEngineSetUpdateSource(pContext, pvArgs, pvResults);
+        break;
     case BOOTSTRAPPER_ENGINE_MESSAGE_COMPAREVERSIONS:
         hr = BAEngineCompareVersions(pContext, pvArgs, pvResults);
         break;
-- 
cgit v1.2.3-55-g6feb