2 files changed, 19 insertions, 0 deletions

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,
diff --git a/src/balutil/inc/IBootstrapperEngine.h b/src/balutil/inc/IBootstrapperEngine.h
index cd89a9f0..af6379f4 100644
--- a/src/balutil/inc/IBootstrapperEngine.h
+++ b/src/balutil/inc/IBootstrapperEngine.h
@@ -128,6 +128,10 @@ DECLARE_INTERFACE_IID_(IBootstrapperEngine, IUnknown, "6480D616-27A0-44D7-905B-8
         __in DWORD dwWaitForInputIdleTimeout
         ) = 0;
 
+    STDMETHOD(SetUpdateSource)(
+        __in_z LPCWSTR wzUrl
+        ) = 0;
+
     STDMETHOD(CompareVersions)(
         __in_z LPCWSTR wzVersion1,
         __in_z LPCWSTR wzVersion2,