6 files changed, 345 insertions, 0 deletions

diff --git a/src/bextutil/inc/BextBaseBundleExtension.h b/src/bextutil/inc/BextBaseBundleExtension.h
new file mode 100644
index 00000000..69c338e4
--- /dev/null
+++ b/src/bextutil/inc/BextBaseBundleExtension.h
@@ -0,0 +1,120 @@
+// 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 "BundleExtensionEngine.h"
+#include "BundleExtension.h"
+#include "IBundleExtensionEngine.h"
+#include "IBundleExtension.h"
+
+#include "bextutil.h"
+
+class CBextBaseBundleExtension : public IBundleExtension
+{
+public: // IUnknown
+    virtual STDMETHODIMP QueryInterface(
+        __in REFIID riid,
+        __out LPVOID *ppvObject
+        )
+    {
+        if (!ppvObject)
+        {
+            return E_INVALIDARG;
+        }
+
+        *ppvObject = NULL;
+
+        if (::IsEqualIID(__uuidof(IBundleExtension), riid))
+        {
+            *ppvObject = static_cast<IBundleExtension*>(this);
+        }
+        else if (::IsEqualIID(IID_IUnknown, riid))
+        {
+            *ppvObject = static_cast<IUnknown*>(this);
+        }
+        else // no interface for requested iid
+        {
+            return E_NOINTERFACE;
+        }
+
+        AddRef();
+        return S_OK;
+    }
+
+    virtual STDMETHODIMP_(ULONG) AddRef()
+    {
+        return ::InterlockedIncrement(&this->m_cReferences);
+    }
+
+    virtual STDMETHODIMP_(ULONG) Release()
+    {
+        long l = ::InterlockedDecrement(&this->m_cReferences);
+        if (0 < l)
+        {
+            return l;
+        }
+
+        delete this;
+        return 0;
+    }
+
+public: // IBundleExtension
+    virtual STDMETHODIMP Search(
+        __in LPCWSTR /*wzId*/,
+        __in LPCWSTR /*wzVariable*/
+        )
+    {
+        return E_NOTIMPL;
+    }
+
+    virtual STDMETHODIMP BundleExtensionProc(
+        __in BUNDLE_EXTENSION_MESSAGE /*message*/,
+        __in const LPVOID /*pvArgs*/,
+        __inout LPVOID /*pvResults*/,
+        __in_opt LPVOID /*pvContext*/
+        )
+    {
+        return E_NOTIMPL;
+    }
+
+public: //CBextBaseBundleExtension
+    virtual STDMETHODIMP Initialize(
+        __in const BUNDLE_EXTENSION_CREATE_ARGS* pCreateArgs
+    )
+    {
+        HRESULT hr = S_OK;
+
+        hr = StrAllocString(&m_sczBundleExtensionDataPath, pCreateArgs->wzBundleExtensionDataPath, 0);
+        ExitOnFailure(hr, "Failed to copy BundleExtensionDataPath.");
+
+    LExit:
+        return hr;
+    }
+
+protected:
+
+    CBextBaseBundleExtension(
+        __in IBundleExtensionEngine* pEngine
+        )
+    {
+        m_cReferences = 1;
+
+        pEngine->AddRef();
+        m_pEngine = pEngine;
+
+        m_sczBundleExtensionDataPath = NULL;
+    }
+
+    virtual ~CBextBaseBundleExtension()
+    {
+        ReleaseNullObject(m_pEngine);
+        ReleaseStr(m_sczBundleExtensionDataPath);
+    }
+
+protected:
+    IBundleExtensionEngine* m_pEngine;
+    LPWSTR m_sczBundleExtensionDataPath;
+
+private:
+    long m_cReferences;
+};
diff --git a/src/bextutil/inc/BextBaseBundleExtensionProc.h b/src/bextutil/inc/BextBaseBundleExtensionProc.h
new file mode 100644
index 00000000..f71e3b92
--- /dev/null
+++ b/src/bextutil/inc/BextBaseBundleExtensionProc.h
@@ -0,0 +1,48 @@
+#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 "BundleExtensionEngine.h"
+#include "BundleExtension.h"
+#include "IBundleExtensionEngine.h"
+#include "IBundleExtension.h"
+
+static HRESULT BextBaseBEProcSearch(
+    __in IBundleExtension* pBE,
+    __in BUNDLE_EXTENSION_SEARCH_ARGS* pArgs,
+    __inout BUNDLE_EXTENSION_SEARCH_RESULTS* /*pResults*/
+    )
+{
+    return pBE->Search(pArgs->wzId, pArgs->wzVariable);
+}
+
+/*******************************************************************
+BextBaseBundleExtensionProc - requires pvContext to be of type IBundleExtension.
+                              Provides a default mapping between the message based
+                              BundleExtension interface and the COM-based BundleExtension interface.
+
+*******************************************************************/
+static HRESULT WINAPI BextBaseBundleExtensionProc(
+    __in BUNDLE_EXTENSION_MESSAGE message,
+    __in const LPVOID pvArgs,
+    __inout LPVOID pvResults,
+    __in_opt LPVOID pvContext
+    )
+{
+    IBundleExtension* pBE = reinterpret_cast<IBundleExtension*>(pvContext);
+    HRESULT hr = pBE->BundleExtensionProc(message, pvArgs, pvResults, pvContext);
+    
+    if (E_NOTIMPL == hr)
+    {
+        switch (message)
+        {
+        case BUNDLE_EXTENSION_MESSAGE_SEARCH:
+            hr = BextBaseBEProcSearch(pBE, reinterpret_cast<BUNDLE_EXTENSION_SEARCH_ARGS*>(pvArgs), reinterpret_cast<BUNDLE_EXTENSION_SEARCH_RESULTS*>(pvResults));
+            break;
+        }
+    }
+
+    return hr;
+}
diff --git a/src/bextutil/inc/BextBundleExtensionEngine.h b/src/bextutil/inc/BextBundleExtensionEngine.h
new file mode 100644
index 00000000..9fdcb700
--- /dev/null
+++ b/src/bextutil/inc/BextBundleExtensionEngine.h
@@ -0,0 +1,17 @@
+// 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.
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+// function declarations
+
+HRESULT BextBundleExtensionEngineCreate(
+    __in PFN_BUNDLE_EXTENSION_ENGINE_PROC pfnBundleExtensionEngineProc,
+    __in_opt LPVOID pvBundleExtensionEngineProcContext,
+    __out IBundleExtensionEngine** ppEngineForExtension
+    );
+
+#ifdef __cplusplus
+}
+#endif
diff --git a/src/bextutil/inc/IBundleExtension.h b/src/bextutil/inc/IBundleExtension.h
new file mode 100644
index 00000000..7516c11b
--- /dev/null
+++ b/src/bextutil/inc/IBundleExtension.h
@@ -0,0 +1,20 @@
+#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.
+
+
+DECLARE_INTERFACE_IID_(IBundleExtension, IUnknown, "93123C9D-796B-4FCD-A507-6EDEF9A925FD")
+{
+    STDMETHOD(Search)(
+        __in LPCWSTR wzId,
+        __in LPCWSTR wzVariable
+        ) = 0;
+
+    // BundleExtensionProc - The PFN_BUNDLE_EXTENSION_PROC can call this method to give the BundleExtension raw access to the callback from the engine.
+    //                       This might be used to help the BundleExtension support more than one version of the engine.
+    STDMETHOD(BundleExtensionProc)(
+        __in BUNDLE_EXTENSION_MESSAGE message,
+        __in const LPVOID pvArgs,
+        __inout LPVOID pvResults,
+        __in_opt LPVOID pvContext
+        ) = 0;
+};
diff --git a/src/bextutil/inc/IBundleExtensionEngine.h b/src/bextutil/inc/IBundleExtensionEngine.h
new file mode 100644
index 00000000..869c6695
--- /dev/null
+++ b/src/bextutil/inc/IBundleExtensionEngine.h
@@ -0,0 +1,64 @@
+#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.
+
+
+DECLARE_INTERFACE_IID_(IBundleExtensionEngine, IUnknown, "9D027A39-F6B6-42CC-9737-C185089EB263")
+{
+    STDMETHOD(EscapeString)(
+        __in_z LPCWSTR wzIn,
+        __out_ecount_opt(*pcchOut) LPWSTR wzOut,
+        __inout DWORD * pcchOut
+        ) = 0;
+
+    STDMETHOD(EvaluateCondition)(
+        __in_z LPCWSTR wzCondition,
+        __out BOOL * pf
+        ) = 0;
+
+    STDMETHOD(FormatString)(
+        __in_z LPCWSTR wzIn,
+        __out_ecount_opt(*pcchOut) LPWSTR wzOut,
+        __inout DWORD * pcchOut
+        ) = 0;
+
+    STDMETHOD(GetVariableNumeric)(
+        __in_z LPCWSTR wzVariable,
+        __out LONGLONG* pllValue
+        ) = 0;
+
+    STDMETHOD(GetVariableString)(
+        __in_z LPCWSTR wzVariable,
+        __out_ecount_opt(*pcchValue) LPWSTR wzValue,
+        __inout DWORD* pcchValue
+        ) = 0;
+
+    STDMETHOD(GetVariableVersion)(
+        __in_z LPCWSTR wzVariable,
+        __out DWORD64* pqwValue
+        ) = 0;
+
+    STDMETHOD(Log)(
+        __in BUNDLE_EXTENSION_LOG_LEVEL level,
+        __in_z LPCWSTR wzMessage
+        ) = 0;
+
+    STDMETHOD(SetVariableLiteralString)(
+        __in_z LPCWSTR wzVariable,
+        __in_z_opt LPCWSTR wzValue
+        ) = 0;
+
+    STDMETHOD(SetVariableNumeric)(
+        __in_z LPCWSTR wzVariable,
+        __in LONGLONG llValue
+        ) = 0;
+
+    STDMETHOD(SetVariableString)(
+        __in_z LPCWSTR wzVariable,
+        __in_z_opt LPCWSTR wzValue
+        ) = 0;
+
+    STDMETHOD(SetVariableVersion)(
+        __in_z LPCWSTR wzVariable,
+        __in DWORD64 qwValue
+        ) = 0;
+};
diff --git a/src/bextutil/inc/bextutil.h b/src/bextutil/inc/bextutil.h
new file mode 100644
index 00000000..0472f854
--- /dev/null
+++ b/src/bextutil/inc/bextutil.h
@@ -0,0 +1,76 @@
+#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 "dutil.h"
+
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#define BextExitOnFailure(x, f, ...) if (FAILED(x)) { BextLogError(x, f, __VA_ARGS__); ExitTrace(x, f, __VA_ARGS__); goto LExit; }
+#define BextExitOnRootFailure(x, f, ...) if (FAILED(x)) { BextLogError(x, f, __VA_ARGS__); Dutil_RootFailure(__FILE__, __LINE__, x); ExitTrace(x, f, __VA_ARGS__); goto LExit; }
+#define BextExitOnNullWithLastError(p, x, f, ...) if (NULL == p) { DWORD Dutil_er = ::GetLastError(); x = HRESULT_FROM_WIN32(Dutil_er); if (!FAILED(x)) { x = E_FAIL; } BextLogError(x, f, __VA_ARGS__); ExitTrace(x, f, __VA_ARGS__); goto LExit; }
+
+const LPCWSTR BUNDLE_EXTENSION_MANIFEST_FILENAME = L"BundleExtensionData.xml";
+
+
+/*******************************************************************
+ BextInitialize - remembers the engine interface to enable logging and
+                  other functions.
+
+********************************************************************/
+DAPI_(void) BextInitialize(
+    __in IBundleExtensionEngine* pEngine
+    );
+
+/*******************************************************************
+ BextInitializeFromCreateArgs - convenience function to call BextBundleExtensionEngineCreate
+                                then pass it along to BextInitialize.
+
+********************************************************************/
+DAPI_(HRESULT) BextInitializeFromCreateArgs(
+    __in const BUNDLE_EXTENSION_CREATE_ARGS* pArgs,
+    __out IBundleExtensionEngine** ppEngine
+    );
+
+/*******************************************************************
+ BextUninitialize - cleans up utility layer internals.
+
+********************************************************************/
+DAPI_(void) BextUninitialize();
+
+/*******************************************************************
+ BextGetBundleExtensionDataNode - gets the requested BundleExtension node.
+
+********************************************************************/
+DAPI_(HRESULT) BextGetBundleExtensionDataNode(
+    __in IXMLDOMDocument* pixdManifest,
+    __in LPCWSTR wzExtensionId,
+    __out IXMLDOMNode** ppixnBundleExtension
+    );
+
+/*******************************************************************
+ BextLog - logs a message with the engine.
+
+********************************************************************/
+DAPIV_(HRESULT) BextLog(
+    __in BUNDLE_EXTENSION_LOG_LEVEL level,
+    __in_z __format_string LPCSTR szFormat,
+    ...
+    );
+
+/*******************************************************************
+ BextLogError - logs an error message with the engine.
+
+********************************************************************/
+DAPIV_(HRESULT) BextLogError(
+    __in HRESULT hr,
+    __in_z __format_string LPCSTR szFormat,
+    ...
+    );
+
+#ifdef __cplusplus
+}
+#endif