1 files changed, 102 insertions, 0 deletions

diff --git a/src/ext/NetFx/be/NetfxBootstrapperExtension.cpp b/src/ext/NetFx/be/NetfxBootstrapperExtension.cpp
new file mode 100644
index 00000000..ab30e378
--- /dev/null
+++ b/src/ext/NetFx/be/NetfxBootstrapperExtension.cpp
@@ -0,0 +1,102 @@
+// 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 "precomp.h"
+#include "BextBaseBootstrapperExtension.h"
+
+class CWixNetfxBootstrapperExtension : public CBextBaseBootstrapperExtension
+{
+public: // IBootstrapperExtension
+    virtual STDMETHODIMP Search(
+        __in LPCWSTR wzId,
+        __in LPCWSTR wzVariable
+        )
+    {
+        HRESULT hr = S_OK;
+
+        hr = NetfxSearchExecute(&m_searches, wzId, wzVariable, m_pEngine, m_sczBaseDirectory);
+
+        return hr;
+    }
+
+public: //CBextBaseBootstrapperExtension
+    virtual STDMETHODIMP Initialize(
+        __in const BOOTSTRAPPER_EXTENSION_CREATE_ARGS* pCreateArgs
+        )
+    {
+        HRESULT hr = S_OK;
+        LPWSTR sczModulePath = NULL;
+        IXMLDOMDocument* pixdManifest = NULL;
+        IXMLDOMNode* pixnBootstrapperExtension = NULL;
+
+        hr = __super::Initialize(pCreateArgs);
+        BextExitOnFailure(hr, "CBextBaseBootstrapperExtension initialization failed.");
+
+        hr = PathForCurrentProcess(&sczModulePath, m_hInstance);
+        BextExitOnFailure(hr, "Failed to get bundle extension path.");
+
+        hr = PathGetDirectory(sczModulePath, &m_sczBaseDirectory);
+        BextExitOnFailure(hr, "Failed to get bundle extension base directory.");
+
+        hr = XmlLoadDocumentFromFile(m_sczBootstrapperExtensionDataPath, &pixdManifest);
+        BextExitOnFailure(hr, "Failed to load bundle extension manifest from path: %ls", m_sczBootstrapperExtensionDataPath);
+
+        hr = BextGetBootstrapperExtensionDataNode(pixdManifest, NETFX_BOOTSTRAPPER_EXTENSION_ID, &pixnBootstrapperExtension);
+        BextExitOnFailure(hr, "Failed to get BootstrapperExtension '%ls'", NETFX_BOOTSTRAPPER_EXTENSION_ID);
+
+        hr = NetfxSearchParseFromXml(&m_searches, pixnBootstrapperExtension);
+        BextExitOnFailure(hr, "Failed to parse searches from bundle extension manifest.");
+
+    LExit:
+        ReleaseObject(pixnBootstrapperExtension);
+        ReleaseObject(pixdManifest);
+        ReleaseStr(sczModulePath);
+
+        return hr;
+    }
+
+public:
+    CWixNetfxBootstrapperExtension(
+        __in HINSTANCE hInstance,
+        __in IBootstrapperExtensionEngine* pEngine
+        ) : CBextBaseBootstrapperExtension(pEngine)
+    {
+        m_searches = { };
+        m_hInstance = hInstance;
+        m_sczBaseDirectory = NULL;
+    }
+
+    ~CWixNetfxBootstrapperExtension()
+    {
+        NetfxSearchUninitialize(&m_searches);
+        ReleaseStr(m_sczBaseDirectory);
+    }
+
+private:
+    NETFX_SEARCHES m_searches;
+    HINSTANCE m_hInstance;
+    LPWSTR m_sczBaseDirectory;
+};
+
+HRESULT NetfxBootstrapperExtensionCreate(
+    __in HINSTANCE hInstance,
+    __in IBootstrapperExtensionEngine* pEngine,
+    __in const BOOTSTRAPPER_EXTENSION_CREATE_ARGS* pArgs,
+    __out IBootstrapperExtension** ppBootstrapperExtension
+    )
+{
+    HRESULT hr = S_OK;
+    CWixNetfxBootstrapperExtension* pExtension = NULL;
+
+    pExtension = new CWixNetfxBootstrapperExtension(hInstance, pEngine);
+    BextExitOnNull(pExtension, hr, E_OUTOFMEMORY, "Failed to create new CWixNetfxBootstrapperExtension.");
+
+    hr = pExtension->Initialize(pArgs);
+    BextExitOnFailure(hr, "CWixNetfxBootstrapperExtension initialization failed.");
+
+    *ppBootstrapperExtension = pExtension;
+    pExtension = NULL;
+
+LExit:
+    ReleaseObject(pExtension);
+    return hr;
+}