1 files changed, 46 insertions, 0 deletions

diff --git a/src/api/burn/bextutil/inc/BextBaseBootstrapperExtensionProc.h b/src/api/burn/bextutil/inc/BextBaseBootstrapperExtensionProc.h
new file mode 100644
index 00000000..4f96399c
--- /dev/null
+++ b/src/api/burn/bextutil/inc/BextBaseBootstrapperExtensionProc.h
@@ -0,0 +1,46 @@
+#pragma once
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+
+#include <windows.h>
+
+#include <IBootstrapperExtensionEngine.h>
+#include <IBootstrapperExtension.h>
+
+static HRESULT BextBaseBEProcSearch(
+    __in IBootstrapperExtension* pBE,
+    __in BOOTSTRAPPER_EXTENSION_SEARCH_ARGS* pArgs,
+    __inout BOOTSTRAPPER_EXTENSION_SEARCH_RESULTS* /*pResults*/
+    )
+{
+    return pBE->Search(pArgs->wzId, pArgs->wzVariable);
+}
+
+/*******************************************************************
+BextBaseBootstrapperExtensionProc - requires pvContext to be of type IBootstrapperExtension.
+                              Provides a default mapping between the message based
+                              BootstrapperExtension interface and the COM-based BootstrapperExtension interface.
+
+*******************************************************************/
+static HRESULT WINAPI BextBaseBootstrapperExtensionProc(
+    __in BOOTSTRAPPER_EXTENSION_MESSAGE message,
+    __in const LPVOID pvArgs,
+    __inout LPVOID pvResults,
+    __in_opt LPVOID pvContext
+    )
+{
+    IBootstrapperExtension* pBE = reinterpret_cast<IBootstrapperExtension*>(pvContext);
+    HRESULT hr = pBE->BootstrapperExtensionProc(message, pvArgs, pvResults, pvContext);
+
+    if (E_NOTIMPL == hr)
+    {
+        switch (message)
+        {
+        case BOOTSTRAPPER_EXTENSION_MESSAGE_SEARCH:
+            hr = BextBaseBEProcSearch(pBE, reinterpret_cast<BOOTSTRAPPER_EXTENSION_SEARCH_ARGS*>(pvArgs), reinterpret_cast<BOOTSTRAPPER_EXTENSION_SEARCH_RESULTS*>(pvResults));
+            break;
+        }
+    }
+
+    return hr;
+}