Add a manual test bundle for testing BAFunctions and thmutil.

29 files changed, 785 insertions, 0 deletions

diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/BafThmUtilTesting.cpp b/src/test/burn/TestData/Manual/BafThmutilTesting/BafThmUtilTesting.cpp
new file mode 100644
index 00000000..8b49cab6
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/BafThmUtilTesting.cpp
@@ -0,0 +1,432 @@
+// 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 "BalBaseBAFunctions.h"
+#include "BalBaseBAFunctionsProc.h"
+
+static const LPCWSTR BAFTHMUTILTESTING_WINDOW_CLASS = L"BafThmUtilTesting";
+
+enum BAFTHMUTILTESTING_CONTROL
+{
+    BAFTHMUTILTESTING_CONTROL_LISTVIEW_TOP_LEFT = THEME_FIRST_ASSIGN_CONTROL_ID,
+    BAFTHMUTILTESTING_CONTROL_LISTVIEW_TOP_RIGHT,
+    BAFTHMUTILTESTING_CONTROL_LISTVIEW_BOTTOM_LEFT,
+    BAFTHMUTILTESTING_CONTROL_LISTVIEW_BOTTOM_RIGHT,
+    BAFTHMUTILTESTING_CONTROL_PROGRESSBAR_STANDARD,
+    BAFTHMUTILTESTING_CONTROL_PROGRESSBAR_IMAGE,
+};
+
+static THEME_ASSIGN_CONTROL_ID vrgInitControls[] = {
+    { BAFTHMUTILTESTING_CONTROL_LISTVIEW_TOP_LEFT, L"ListViewTopLeft" },
+    { BAFTHMUTILTESTING_CONTROL_LISTVIEW_TOP_RIGHT, L"ListViewTopRight" },
+    { BAFTHMUTILTESTING_CONTROL_LISTVIEW_BOTTOM_LEFT, L"ListViewBottomLeft" },
+    { BAFTHMUTILTESTING_CONTROL_LISTVIEW_BOTTOM_RIGHT, L"ListViewBottomRight" },
+    { BAFTHMUTILTESTING_CONTROL_PROGRESSBAR_IMAGE, L"ImageProgressBar" },
+    { BAFTHMUTILTESTING_CONTROL_PROGRESSBAR_STANDARD, L"StandardProgressBar" },
+};
+
+static void CALLBACK BafThmUtilTestingTraceError(
+    __in_z LPCSTR szFile,
+    __in int iLine,
+    __in REPORT_LEVEL rl,
+    __in UINT source,
+    __in HRESULT hrError,
+    __in_z __format_string LPCSTR szFormat,
+    __in va_list args
+    );
+
+class CBafThmUtilTesting : public CBalBaseBAFunctions
+{
+public: // IBAFunctions
+    /*virtual STDMETHODIMP OnThemeLoaded(
+        THEME* pTheme,
+        WIX_LOCALIZATION* pWixLoc
+        )
+    {
+        HRESULT hr = S_OK;
+
+        hr = __super::OnThemeLoaded(pTheme, pWixLoc);
+
+        return hr;
+    }*/
+
+    virtual STDMETHODIMP WndProc(
+        __in THEME* pTheme,
+        __in HWND hWnd,
+        __in UINT uMsg,
+        __in WPARAM wParam,
+        __in LPARAM lParam,
+        __inout LRESULT* plRes
+        )
+    {
+        HRESULT hr = S_OK;
+
+        __super::WndProc(pTheme, hWnd, uMsg, wParam, lParam, plRes);
+
+        // Show our window when any button is clicked.
+        switch (uMsg)
+        {
+        case WM_COMMAND:
+            switch (HIWORD(wParam))
+            {
+            case BN_CLICKED:
+                OnShowTheme();
+                break;
+            }
+            break;
+        }
+
+        return hr;
+    }
+
+private:
+    HRESULT OnShowTheme()
+    {
+        HRESULT hr = S_OK;
+        BOOL fRet = FALSE;
+        MSG msg = { };
+
+        hr = ThemeLoadFromResource(m_hModule, MAKEINTRESOURCEA(1), &m_pBafTheme);
+        BalExitOnFailure(hr, "Failed to load BafThmUtilTesting theme.");
+
+        hr = CreateTestingWindow();
+        BalExitOnFailure(hr, "Failed to create BafThmUtilTesting window.");
+
+        ::EnableWindow(m_pTheme->hwndParent, FALSE);
+
+        // message pump
+        while (0 != (fRet = ::GetMessageW(&msg, NULL, 0, 0)))
+        {
+            if (-1 == fRet)
+            {
+                hr = E_UNEXPECTED;
+                BalExitOnFailure(hr, "Unexpected return value from message pump.");
+            }
+            else if (!ThemeHandleKeyboardMessage(m_pBafTheme, msg.hwnd, &msg))
+            {
+                ::TranslateMessage(&msg);
+                ::DispatchMessageW(&msg);
+            }
+        }
+
+    LExit:
+        ::EnableWindow(m_pTheme->hwndParent, TRUE);
+
+        DestroyTestingWindow();
+
+        ReleaseTheme(m_pBafTheme);
+
+        return hr;
+    }
+
+    HRESULT CreateTestingWindow()
+    {
+        HRESULT hr = S_OK;
+        HICON hIcon = reinterpret_cast<HICON>(m_pTheme->hIcon);
+        WNDCLASSW wc = { };
+        int x = CW_USEDEFAULT;
+        int y = CW_USEDEFAULT;
+        POINT ptCursor = { };
+
+        // If the theme did not provide an icon, try using the icon from the bundle engine.
+        if (!hIcon)
+        {
+            HMODULE hBootstrapperEngine = ::GetModuleHandleW(NULL);
+            if (hBootstrapperEngine)
+            {
+                hIcon = ::LoadIconW(hBootstrapperEngine, MAKEINTRESOURCEW(1));
+            }
+        }
+
+        // Register the window class and create the window.
+        wc.lpfnWndProc = CBafThmUtilTesting::TestingWndProc;
+        wc.hInstance = m_hModule;
+        wc.hIcon = hIcon;
+        wc.hCursor = ::LoadCursorW(NULL, (LPCWSTR)IDC_ARROW);
+        wc.hbrBackground = m_pTheme->rgFonts[m_pBafTheme->dwFontId].hBackground;
+        wc.lpszMenuName = NULL;
+        wc.lpszClassName = BAFTHMUTILTESTING_WINDOW_CLASS;
+        if (!::RegisterClassW(&wc))
+        {
+            ExitWithLastError(hr, "Failed to register window.");
+        }
+
+        m_fRegistered = TRUE;
+
+        // Center the window on the monitor with the mouse.
+        if (::GetCursorPos(&ptCursor))
+        {
+            x = ptCursor.x;
+            y = ptCursor.y;
+        }
+
+        hr = ThemeCreateParentWindow(m_pBafTheme, 0, wc.lpszClassName, m_pBafTheme->sczCaption, m_pBafTheme->dwStyle, x, y, m_pTheme->hwndParent, m_hModule, this, THEME_WINDOW_INITIAL_POSITION_CENTER_MONITOR_FROM_COORDINATES, &m_hWndBaf);
+        ExitOnFailure(hr, "Failed to create window.");
+
+        hr = S_OK;
+
+    LExit:
+        return hr;
+    }
+
+    void DestroyTestingWindow()
+    {
+        if (::IsWindow(m_hWndBaf))
+        {
+            ::DestroyWindow(m_hWndBaf);
+            m_hWndBaf = NULL;
+        }
+
+        if (m_fRegistered)
+        {
+            ::UnregisterClassW(BAFTHMUTILTESTING_WINDOW_CLASS, m_hModule);
+            m_fRegistered = FALSE;
+        }
+    }
+
+    static LRESULT CALLBACK TestingWndProc(
+        __in HWND hWnd,
+        __in UINT uMsg,
+        __in WPARAM wParam,
+        __in LPARAM lParam
+        )
+    {
+#pragma warning(suppress:4312)
+        CBafThmUtilTesting* pBaf = reinterpret_cast<CBafThmUtilTesting*>(::GetWindowLongPtrW(hWnd, GWLP_USERDATA));
+
+        switch (uMsg)
+        {
+        case WM_NCCREATE:
+        {
+            LPCREATESTRUCT lpcs = reinterpret_cast<LPCREATESTRUCT>(lParam);
+            pBaf = reinterpret_cast<CBafThmUtilTesting*>(lpcs->lpCreateParams);
+#pragma warning(suppress:4244)
+            ::SetWindowLongPtrW(hWnd, GWLP_USERDATA, reinterpret_cast<LONG_PTR>(pBaf));
+            break;
+        }
+
+        case WM_CLOSE:
+            if (pBaf)
+            {
+                ::EnableWindow(pBaf->m_pTheme->hwndParent, TRUE);
+            }
+
+            break;
+
+        case WM_NCDESTROY:
+        {
+            LRESULT lres = ThemeDefWindowProc(pBaf ? pBaf->m_pBafTheme : NULL, hWnd, uMsg, wParam, lParam);
+            ::SetWindowLongPtrW(hWnd, GWLP_USERDATA, 0);
+
+            ::PostQuitMessage(0);
+            return lres;
+        }
+
+        case WM_CREATE:
+            if (!pBaf->OnCreate(hWnd))
+            {
+                return -1;
+            }
+            break;
+
+        case WM_TIMER:
+            if (!lParam && pBaf)
+            {
+                pBaf->UpdateProgressBarProgress();
+
+                return 0;
+            }
+            break;
+        }
+
+        return ThemeDefWindowProc(pBaf ? pBaf->m_pBafTheme : NULL, hWnd, uMsg, wParam, lParam);
+    }
+
+    BOOL OnCreate(
+        __in HWND hWnd
+        )
+    {
+        HRESULT hr = S_OK;
+        LVITEMW lvitem = { };
+        LVGROUP lvgroup = { };
+        static UINT puColumns[] = { 0, 1, 2 };
+        HWND hwndTopLeft = NULL;
+        HWND hwndTopRight = NULL;
+        HWND hwndBottomLeft = NULL;
+        HWND hwndBottomRight = NULL;
+
+        hr = ThemeLoadControls(m_pBafTheme, vrgInitControls, countof(vrgInitControls));
+        BalExitOnFailure(hr, "Failed to load theme controls.");
+
+        hwndTopLeft = ::GetDlgItem(m_pBafTheme->hwndParent, BAFTHMUTILTESTING_CONTROL_LISTVIEW_TOP_LEFT);
+        BalExitOnNull(hwndTopLeft, hr, E_INVALIDSTATE, "Failed to get top left list view hWnd.");
+
+        hwndTopRight = ::GetDlgItem(m_pBafTheme->hwndParent, BAFTHMUTILTESTING_CONTROL_LISTVIEW_TOP_RIGHT);
+        BalExitOnNull(hwndTopRight, hr, E_INVALIDSTATE, "Failed to get top right list view hWnd.");
+
+        hwndBottomLeft = ::GetDlgItem(m_pBafTheme->hwndParent, BAFTHMUTILTESTING_CONTROL_LISTVIEW_BOTTOM_LEFT);
+        BalExitOnNull(hwndBottomLeft, hr, E_INVALIDSTATE, "Failed to get bottom left list view hWnd.");
+
+        hwndBottomRight = ::GetDlgItem(m_pBafTheme->hwndParent, BAFTHMUTILTESTING_CONTROL_LISTVIEW_BOTTOM_RIGHT);
+        BalExitOnNull(hwndBottomRight, hr, E_INVALIDSTATE, "Failed to get bottom right list view hWnd.");
+
+        lvgroup.cbSize = sizeof(LVGROUP);
+        lvgroup.mask = LVGF_GROUPID | LVGF_TITLEIMAGE | LVGF_DESCRIPTIONTOP | LVGF_HEADER;
+
+        for (int i = 0; i < 3; ++i)
+        {
+            lvgroup.iGroupId = i;
+            lvgroup.iTitleImage = i;
+
+            hr = StrAllocFormatted(&lvgroup.pszDescriptionTop, L"DescriptionTop_%d", i);
+            BalExitOnFailure(hr, "Failed to alloc list view group description.");
+
+            hr = StrAllocFormatted(&lvgroup.pszHeader, L"Header_%d", i);
+            BalExitOnFailure(hr, "Failed to alloc list view group header.");
+
+            ListView_InsertGroup(hwndTopLeft, -1, &lvgroup);
+            ListView_InsertGroup(hwndTopRight, -1, &lvgroup);
+            ListView_InsertGroup(hwndBottomLeft, -1, &lvgroup);
+            ListView_InsertGroup(hwndBottomRight, -1, &lvgroup);
+
+            lvitem.mask = LVIF_COLUMNS | LVIF_GROUPID | LVIF_IMAGE | LVIF_TEXT;
+            lvitem.iItem = i;
+            lvitem.iSubItem = 0;
+
+            hr = StrAllocFormatted(&lvitem.pszText, L"ListViewItem_%d", i);
+            BalExitOnFailure(hr, "Failed to alloc list view item text.");
+
+            lvitem.iImage = i;
+            lvitem.iGroupId = i;
+            lvitem.cColumns = countof(puColumns);
+            lvitem.puColumns = puColumns;
+
+            ListView_InsertItem(hwndTopLeft, &lvitem);
+            ListView_InsertItem(hwndTopRight, &lvitem);
+            ListView_InsertItem(hwndBottomLeft, &lvitem);
+            ListView_InsertItem(hwndBottomRight, &lvitem);
+
+            for (int j = 0; j < 3; ++j)
+            {
+                lvitem.mask = LVIF_TEXT;
+                lvitem.iSubItem = j + 1;
+
+                hr = StrAllocFormatted(&lvitem.pszText, L"%d_%d", j, i);
+                BalExitOnFailure(hr, "Failed to alloc list view subitem text.");
+
+                ListView_InsertItem(hwndTopLeft, &lvitem);
+                ListView_InsertItem(hwndTopRight, &lvitem);
+                ListView_InsertItem(hwndBottomLeft, &lvitem);
+                ListView_InsertItem(hwndBottomRight, &lvitem);
+            }
+        }
+
+        ListView_EnableGroupView(hwndTopRight, TRUE);
+
+        ::SetTimer(hWnd, BAFTHMUTILTESTING_CONTROL_PROGRESSBAR_IMAGE, 500, NULL);
+
+    LExit:
+        ReleaseStr(lvgroup.pszDescriptionTop);
+        ReleaseStr(lvgroup.pszHeader);
+        ReleaseStr(lvitem.pszText);
+
+        return SUCCEEDED(hr);
+    }
+
+    void UpdateProgressBarProgress()
+    {
+        static DWORD dwProgress = 0;
+        DWORD dwCurrent = dwProgress < 100 ? dwProgress : 200 - dwProgress;
+
+        if (0 == dwProgress || 100 == dwProgress)
+        {
+            ThemeSetProgressControlColor(m_pBafTheme, BAFTHMUTILTESTING_CONTROL_PROGRESSBAR_IMAGE, 100 == dwProgress ? 1 : 0);
+        }
+
+        dwProgress = (dwProgress + 10) % 200;
+
+        ThemeSetProgressControl(m_pBafTheme, BAFTHMUTILTESTING_CONTROL_PROGRESSBAR_IMAGE, dwCurrent);
+        ThemeSetProgressControl(m_pBafTheme, BAFTHMUTILTESTING_CONTROL_PROGRESSBAR_STANDARD, dwCurrent);
+    }
+
+public:
+    //
+    // Constructor - initialize member variables.
+    //
+    CBafThmUtilTesting(
+        __in HMODULE hModule,
+        __in IBootstrapperEngine* pEngine,
+        __in const BA_FUNCTIONS_CREATE_ARGS* pArgs
+        ) : CBalBaseBAFunctions(hModule, pEngine, pArgs)
+    {
+        m_pBafTheme = NULL;
+        m_fRegistered = FALSE;
+        m_hWndBaf = NULL;
+
+        ThemeInitialize(hModule);
+    }
+
+    //
+    // Destructor - release member variables.
+    //
+    ~CBafThmUtilTesting()
+    {
+        Assert(!::IsWindow(m_hWndBaf));
+        Assert(!m_pBafTheme);
+
+        ThemeUninitialize();
+    }
+
+private:
+    THEME* m_pBafTheme;
+    BOOL m_fRegistered;
+    HWND m_hWndBaf;
+};
+
+
+HRESULT WINAPI CreateBAFunctions(
+    __in HMODULE hModule,
+    __in const BA_FUNCTIONS_CREATE_ARGS* pArgs,
+    __inout BA_FUNCTIONS_CREATE_RESULTS* pResults
+    )
+{
+    HRESULT hr = S_OK;
+    CBafThmUtilTesting* pBAFunctions = NULL;
+    IBootstrapperEngine* pEngine = NULL;
+
+    DutilInitialize(&BafThmUtilTestingTraceError);
+
+    hr = BalInitializeFromCreateArgs(pArgs->pBootstrapperCreateArgs, &pEngine);
+    ExitOnFailure(hr, "Failed to initialize Bal.");
+
+    pBAFunctions = new CBafThmUtilTesting(hModule, pEngine, pArgs);
+    ExitOnNull(pBAFunctions, hr, E_OUTOFMEMORY, "Failed to create new CBafThmUtilTesting object.");
+
+    pResults->pfnBAFunctionsProc = BalBaseBAFunctionsProc;
+    pResults->pvBAFunctionsProcContext = pBAFunctions;
+    pBAFunctions = NULL;
+
+LExit:
+    ReleaseObject(pBAFunctions);
+    ReleaseObject(pEngine);
+
+    return hr;
+}
+
+static void CALLBACK BafThmUtilTestingTraceError(
+    __in_z LPCSTR /*szFile*/,
+    __in int /*iLine*/,
+    __in REPORT_LEVEL /*rl*/,
+    __in UINT source,
+    __in HRESULT hrError,
+    __in_z __format_string LPCSTR szFormat,
+    __in va_list args
+    )
+{
+    // BalLogError currently uses the Exit... macros,
+    // so if expanding the scope need to ensure this doesn't get called recursively.
+    if (DUTIL_SOURCE_THMUTIL == source)
+    {
+        BalLogErrorArgs(hrError, szFormat, args);
+    }
+}
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/BafThmUtilTesting.def b/src/test/burn/TestData/Manual/BafThmutilTesting/BafThmUtilTesting.def
new file mode 100644
index 00000000..6e016dad
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/BafThmUtilTesting.def
@@ -0,0 +1,6 @@
+; 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.
+
+
+EXPORTS
+    BAFunctionsCreate
+    BAFunctionsDestroy
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/BafThmUtilTesting.vcxproj b/src/test/burn/TestData/Manual/BafThmutilTesting/BafThmUtilTesting.vcxproj
new file mode 100644
index 00000000..8e7e9bd9
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/BafThmUtilTesting.vcxproj
@@ -0,0 +1,64 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!-- 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. -->
+<Project DefaultTargets="Build" Toolsxmlns="http://schemas.microsoft.com/developer/msbuild/2003">
+  <ItemGroup Label="ProjectConfigurations">
+    <ProjectConfiguration Include="Debug|ARM64">
+      <Configuration>Debug</Configuration>
+      <Platform>ARM64</Platform>
+    </ProjectConfiguration>
+    <ProjectConfiguration Include="Release|ARM64">
+      <Configuration>Release</Configuration>
+      <Platform>ARM64</Platform>
+    </ProjectConfiguration>
+    <ProjectConfiguration Include="Debug|Win32">
+      <Configuration>Debug</Configuration>
+      <Platform>Win32</Platform>
+    </ProjectConfiguration>
+    <ProjectConfiguration Include="Release|Win32">
+      <Configuration>Release</Configuration>
+      <Platform>Win32</Platform>
+    </ProjectConfiguration>
+    <ProjectConfiguration Include="Debug|x64">
+      <Configuration>Debug</Configuration>
+      <Platform>x64</Platform>
+    </ProjectConfiguration>
+    <ProjectConfiguration Include="Release|x64">
+      <Configuration>Release</Configuration>
+      <Platform>x64</Platform>
+    </ProjectConfiguration>
+  </ItemGroup>
+  <PropertyGroup Label="Globals">
+    <ProjectGuid>{565E99AE-975F-4F26-8A6E-852603386A80}</ProjectGuid>
+    <ConfigurationType>DynamicLibrary</ConfigurationType>
+    <PlatformToolset>v142</PlatformToolset>
+    <CharacterSet>Unicode</CharacterSet>
+    <TargetName>BafThmUtilTesting</TargetName>
+    <ProjectModuleDefinitionFile>BafThmUtilTesting.def</ProjectModuleDefinitionFile>
+  </PropertyGroup>
+  <Import Project="$(VCTargetsPath)\Microsoft.Cpp.Default.props" />
+  <Import Project="$(VCTargetsPath)\Microsoft.Cpp.props" />
+  <PropertyGroup>
+    <ProjectAdditionalLinkLibraries>comctl32.lib;gdiplus.lib;msimg32.lib;shlwapi.lib;wininet.lib</ProjectAdditionalLinkLibraries>
+  </PropertyGroup>
+  <ItemGroup>
+    <ClCompile Include="BafThmUtilTesting.cpp" />
+    <ClCompile Include="precomp.cpp">
+      <PrecompiledHeader>Create</PrecompiledHeader>
+    </ClCompile>
+  </ItemGroup>
+  <ItemGroup>
+    <ClInclude Include="precomp.h" />
+  </ItemGroup>
+  <ItemGroup>
+    <ResourceCompile Include="theme\theme.rc" Link="theme.rc" />
+  </ItemGroup>
+  <ItemGroup>
+    <None Include="BafThmUtilTesting.def" />
+  </ItemGroup>
+  <ItemGroup>
+    <PackageReference Include="WixToolset.Dutil" />
+    <PackageReference Include="WixToolset.BalUtil" />
+    <PackageReference Include="WixToolset.BootstrapperCore.Native" />
+  </ItemGroup>
+  <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
+</Project>
\ No newline at end of file
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/precomp.cpp b/src/test/burn/TestData/Manual/BafThmutilTesting/precomp.cpp
new file mode 100644
index 00000000..b20f4230
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/precomp.cpp
@@ -0,0 +1,46 @@
+// 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"
+
+static HINSTANCE vhInstance = NULL;
+
+extern "C" BOOL WINAPI DllMain(
+    IN HINSTANCE hInstance,
+    IN DWORD dwReason,
+    IN LPVOID /* pvReserved */
+    )
+{
+    switch (dwReason)
+    {
+    case DLL_PROCESS_ATTACH:
+        ::DisableThreadLibraryCalls(hInstance);
+        vhInstance = hInstance;
+        break;
+
+    case DLL_PROCESS_DETACH:
+        vhInstance = NULL;
+        break;
+    }
+
+    return TRUE;
+}
+
+extern "C" HRESULT WINAPI BAFunctionsCreate(
+    __in const BA_FUNCTIONS_CREATE_ARGS* pArgs,
+    __inout BA_FUNCTIONS_CREATE_RESULTS* pResults
+    )
+{
+    HRESULT hr = S_OK;
+    
+    hr = CreateBAFunctions(vhInstance, pArgs, pResults);
+    BalExitOnFailure(hr, "Failed to create BAFunctions interface.");
+
+LExit:
+    return hr;
+}
+
+extern "C" void WINAPI BAFunctionsDestroy(
+    )
+{
+    BalUninitialize();
+}
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/precomp.h b/src/test/burn/TestData/Manual/BafThmutilTesting/precomp.h
new file mode 100644
index 00000000..e7cd39dc
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/precomp.h
@@ -0,0 +1,40 @@
+#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>
+
+#pragma warning(push)
+#pragma warning(disable:4458) // declaration of 'xxx' hides class member
+#include <gdiplus.h>
+#pragma warning(pop)
+
+#include <msiquery.h>
+#include <objbase.h>
+#include <shlobj.h>
+#include <shlwapi.h>
+#include <stdlib.h>
+#include <strsafe.h>
+#include <CommCtrl.h>
+
+#include "dutil.h"
+#include "dictutil.h"
+#include "fileutil.h"
+#include "locutil.h"
+#include "pathutil.h"
+#include "strutil.h"
+#include "thmutil.h"
+#include "regutil.h"
+#include "xmlutil.h"
+
+#include "BalBaseBootstrapperApplication.h"
+#include "balutil.h"
+
+#include "BAFunctions.h"
+#include "IBAFunctions.h"
+
+HRESULT WINAPI CreateBAFunctions(
+    __in HMODULE hModule,
+    __in const BA_FUNCTIONS_CREATE_ARGS* pArgs,
+    __inout BA_FUNCTIONS_CREATE_RESULTS* pResults
+    );
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/128x128.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/128x128.png
new file mode 100644
index 00000000..4dfceeba
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/128x128.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/128x128_focus.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/128x128_focus.png
new file mode 100644
index 00000000..2caffb30
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/128x128_focus.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/128x128_hover.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/128x128_hover.png
new file mode 100644
index 00000000..8242b0d1
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/128x128_hover.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/128x128_selected.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/128x128_selected.png
new file mode 100644
index 00000000..7355fc06
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/128x128_selected.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/16x16.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/16x16.png
new file mode 100644
index 00000000..52975ac0
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/16x16.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/256x256.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/256x256.png
new file mode 100644
index 00000000..6924ec48
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/256x256.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/256x256_focus.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/256x256_focus.png
new file mode 100644
index 00000000..8cc141b2
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/256x256_focus.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/256x256_hover.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/256x256_hover.png
new file mode 100644
index 00000000..a1cff210
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/256x256_hover.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/256x256_selected.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/256x256_selected.png
new file mode 100644
index 00000000..184c1f78
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/256x256_selected.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/32x32.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/32x32.png
new file mode 100644
index 00000000..01d621b3
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/32x32.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/64x64.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/64x64.png
new file mode 100644
index 00000000..697274f9
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/64x64.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/64x64_focus.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/64x64_focus.png
new file mode 100644
index 00000000..7bd5b2ca
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/64x64_focus.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/64x64_hover.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/64x64_hover.png
new file mode 100644
index 00000000..88b2e8c2
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/64x64_hover.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/64x64_selected.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/64x64_selected.png
new file mode 100644
index 00000000..39e75799
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/64x64_selected.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/BafThmUtilTestingTheme.xml b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/BafThmUtilTestingTheme.xml
new file mode 100644
index 00000000..0d0dabf5
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/BafThmUtilTestingTheme.xml
@@ -0,0 +1,137 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!-- 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. -->
+
+
+<Theme xmlns="http://wixtoolset.org/schemas/v4/thmutil">
+    <Font Id="Default" Height="-12" Weight="500" Foreground="windowtext" Background="window">Segoe UI</Font>
+    <Image Id="star_transparent.bmp" ImageResource="2" />
+    <Image Id="star_opaque.bmp" ImageResource="3" />
+    <Image Id="star_transparent.png" ImageResource="4" />
+    <Image Id="square_default" ImageResource="5">
+        <AlternateResolution ImageResource="6" />
+        <AlternateResolution ImageResource="7" />
+        <AlternateResolution ImageResource="8" />
+        <AlternateResolution ImageResource="9" />
+    </Image>
+    <Image Id="square_focus" ImageResource="13">
+        <AlternateResolution ImageResource="14" />
+        <AlternateResolution ImageResource="15" />
+    </Image>
+    <Image Id="square_hover" ImageResource="19">
+        <AlternateResolution ImageResource="20" />
+        <AlternateResolution ImageResource="21" />
+    </Image>
+    <Image Id="square_selected" ImageResource="25">
+        <AlternateResolution ImageResource="26" />
+        <AlternateResolution ImageResource="27" />
+    </Image>
+    <Image Id="progressbar" ImageResource="30" />
+    <Image Id="progressbar_reverse" ImageResource="31" />
+
+    <Window Width="600" Height="450" FontId="Default" Caption="BafThmUtilTestingTheme" HexStyle="10cf0000" AutoResize="yes">
+        <ImageList Name="Stars">
+            <ImageListItem ImageResource="2" />
+            <ImageListItem ImageResource="3" />
+            <ImageListItem ImageResource="4" />
+        </ImageList>
+        <Page Name="Transparency">
+            <Label X="6" Y="6" Width="-6" Height="94" FontId="Default">
+                This page has three versions of an image. The top image is a bitmap with a transparent background, the yellow star should be visible but its black background should not (this is currently broken). The middle image is the same bitmap except the black background is fully opaque so the yellow star should be visible on a black background. The bottom image in a PNG version of the top image and should look exactly the same.
+            </Label>
+            <ImageControl X="6" Y="106" Width="64" Height="64" ImageId="star_transparent.bmp" />
+            <ImageControl X="6" Y="176" Width="64" Height="64" ImageId="star_opaque.bmp" />
+            <ImageControl X="6" Y="246" Width="64" Height="64" ImageId="star_transparent.png" />
+            <Button Name="ListViewButton" X="-183" Y="-11" Width="75" Height="23" TabStop="yes" FontId="Default">
+                <Text>Next</Text>
+                <ChangePageAction Page="ListView" />
+            </Button>
+        </Page>
+        <Page Name="ListView">
+            <Label X="6" Y="6" Width="-6" Height="18" FontId="Default">
+                This page is for testing ListView.
+            </Label>
+            <ListView Name="ListViewTopLeft" X="6" Y="30" Width="288" Height="176" FontId="Default" HexStyle="00" ImageList="Stars">
+                <Column Width="100">TL One</Column>
+                <Column Width="100">TL Two</Column>
+                <Column Width="88">TL Three</Column>
+            </ListView>
+            <ListView Name="ListViewTopRight" X="300" Y="30" Width="-11" Height="176" FontId="Default" HexStyle="01" ImageListGroupHeader="Stars">
+                <Column Width="100">TR One</Column>
+                <Column Width="100">TR Two</Column>
+                <Column Width="88" Expands="yes">TR Exp</Column>
+            </ListView>
+            <ListView Name="ListViewBottomLeft" X="6" Y="212" Width="288" Height="176" FontId="Default" HexStyle="02" ImageListSmall="Stars">
+                <Column Width="100">BL One</Column>
+                <Column Width="100">BL Two</Column>
+                <Column Width="88">BL Three</Column>
+            </ListView>
+            <ListView Name="ListViewBottomRight" X="300" Y="212" Width="-11" Height="176" FontId="Default" HexStyle="03">
+                <Column Width="100">BR One</Column>
+                <Column Width="100">BR Two</Column>
+                <Column Width="88" Expands="yes">BR Exp</Column>
+            </ListView>
+            <Button Name="ListViewBackButton" X="-269" Y="-11" Width="75" Height="23" TabStop="yes" FontId="Default">
+                <Text>Back</Text>
+                <ChangePageAction Page="Transparency" />
+            </Button>
+            <Button Name="ButtonButton" X="-183" Y="-11" Width="75" Height="23" TabStop="yes" FontId="Default">
+                <Text>Next</Text>
+                <ChangePageAction Page="Button" />
+            </Button>
+        </Page>
+        <Page Name="Button">
+            <Label X="6" Y="6" Width="-6" Height="132" FontId="Default">
+                This page is for testing graphical buttons. The left image is 16x16 at 100% scaling, the middle image is 32x32 at 100% scaling. The images for the buttons display their dimensions, so the left should be 16x16 at 100% but 32x32 at 200%. The right button starts at 32x32 at 100%, but grows and shrinks with the window and its image should also change.
+
+There are currently four states for a button: default, focus, hover, and selected (in that priority order). There should be a different image for each state, although the left button should have a focus rectangle over the default image since it didn't specify an image for the focused state.
+            </Label>
+            <Button Name="Square16Button" X="6" Y="140" Width="16" Height="16" FontId="Default" TabStop="yes">
+                <ButtonImage ImageId="square_default" />
+                <ButtonHoverImage ImageId="square_hover" />
+                <ButtonSelectedImage ImageId="square_selected" />
+            </Button>
+            <Button Name="Square32Button" X="54" Y="140" Width="32" Height="32" FontId="Default" TabStop="yes">
+                <ButtonImage ImageId="square_default" />
+                <ButtonFocusImage ImageId="square_focus" />
+                <ButtonHoverImage ImageId="square_hover" />
+                <ButtonSelectedImage ImageId="square_selected" />
+            </Button>
+            <Button Name="SquareExpandButton" X="92" Y="140" Width="-476" Height="-278" FontId="Default" TabStop="yes">
+                <ButtonImage ImageId="square_default" />
+                <ButtonFocusImage ImageId="square_focus" />
+                <ButtonHoverImage ImageId="square_hover" />
+                <ButtonSelectedImage ImageId="square_selected" />
+            </Button>
+            <Button Name="ButtonBackButton" X="-269" Y="-11" Width="75" Height="23" TabStop="yes" FontId="Default">
+                <Text>Back</Text>
+                <ChangePageAction Page="ListView" />
+            </Button>
+            <Button Name="ProgressbarButton" X="-183" Y="-11" Width="75" Height="23" TabStop="yes" FontId="Default">
+                <Text>Next</Text>
+                <ChangePageAction Page="Progressbar" />
+            </Button>
+        </Page>
+        <Page Name="Progressbar">
+            <Label X="6" Y="6" Width="-6" Height="94" FontId="Default">
+                This page has two progress bars. The top progress bar is a standard control, the bottom one is custom drawn from an image. The bottom one should have its left side one pixel wide as black, the right side one pixel wide as grey, the progress as green, and the remaining as blue. The colors should be reversed when the progress goes backward (this is a way to test different colors, not a standard feature of thmutil).
+            </Label>
+            <Progressbar Name="StandardProgressBar" X="6" Y="106" Width="-6" Height="23" />
+            <Progressbar Name="ImageProgressBar" X="6" Y="141" Width="-6" Height="23">
+                <ProgressbarImage ImageId="progressbar" />
+                <ProgressbarImage ImageId="progressbar_reverse" />
+            </Progressbar>
+            <Button Name="ProgressbarBackButton" X="-269" Y="-11" Width="75" Height="23" TabStop="yes" FontId="Default">
+                <Text>Back</Text>
+                <ChangePageAction Page="Button" />
+            </Button>
+        </Page>
+        <Button Name="StartButton" X="-97" Y="-11" Width="75" Height="23" TabStop="yes" FontId="Default" Visible="yes">
+            <Text>Start</Text>
+            <ChangePageAction Page="Transparency" />
+        </Button>
+        <Button Name="CloseButton" X="-11" Y="-11" Width="75" Height="23" TabStop="yes" FontId="Default" Visible="yes">
+            <Text>ThemeClose</Text>
+            <CloseWindowAction />
+        </Button>
+    </Window>
+</Theme>
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/progressbar.bmp b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/progressbar.bmp
new file mode 100644
index 00000000..2bab3944
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/progressbar.bmp
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/progressbar_reverse.bmp b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/progressbar_reverse.bmp
new file mode 100644
index 00000000..6fedd05e
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/progressbar_reverse.bmp
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/star_opaque.bmp b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/star_opaque.bmp
new file mode 100644
index 00000000..230d958c
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/star_opaque.bmp
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/star_transparent.bmp b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/star_transparent.bmp
new file mode 100644
index 00000000..6596bcff
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/star_transparent.bmp
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/star_transparent.png b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/star_transparent.png
new file mode 100644
index 00000000..2f018763
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/star_transparent.png
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/theme/theme.rc b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/theme.rc
new file mode 100644
index 00000000..808d29d5
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/theme/theme.rc
@@ -0,0 +1,20 @@
+1 RCDATA "BafThmutilTestingTheme.xml"
+2 RCDATA "star_transparent.bmp"
+3 RCDATA "star_opaque.bmp"
+4 RCDATA "star_transparent.png"
+5 RCDATA "16x16.png"
+6 RCDATA "32x32.png"
+7 RCDATA "64x64.png"
+8 RCDATA "128x128.png"
+9 RCDATA "256x256.png"
+13 RCDATA "64x64_focus.png"
+14 RCDATA "128x128_focus.png"
+15 RCDATA "256x256_focus.png"
+19 RCDATA "64x64_hover.png"
+20 RCDATA "128x128_hover.png"
+21 RCDATA "256x256_hover.png"
+25 RCDATA "64x64_selected.png"
+26 RCDATA "128x128_selected.png"
+27 RCDATA "256x256_selected.png"
+30 RCDATA "progressbar.bmp"
+31 RCDATA "progressbar_reverse.bmp"
\ No newline at end of file
diff --git a/src/test/burn/TestData/Manual/BundleA/BundleA.wixproj b/src/test/burn/TestData/Manual/BundleA/BundleA.wixproj
new file mode 100644
index 00000000..dcfd3b7e
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BundleA/BundleA.wixproj
@@ -0,0 +1,18 @@
+<!-- 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. -->
+<Project Sdk="WixToolset.Sdk">
+  <PropertyGroup>
+    <OutputType>Bundle</OutputType>
+    <BA>hyperlinkLicense</BA>
+    <UpgradeCode>{98ACBCF6-B54A-46AF-8990-DFB8795B965B}</UpgradeCode>
+  </PropertyGroup>
+  <ItemGroup>
+    <Compile Include="..\..\Templates\Bundle.wxs" Link="Bundle.wxs" />
+  </ItemGroup>
+  <ItemGroup>
+    <ProjectReference Include="..\BafThmUtilTesting\BafThmUtilTesting.vcxproj" Properties="Platform=x86" ReferenceOutputAssembly="false" />
+    <ProjectReference Include="..\PackageA\PackageA.wixproj" />
+  </ItemGroup>
+  <ItemGroup>
+    <PackageReference Include="WixToolset.Bal.wixext" />
+  </ItemGroup>
+</Project>
\ No newline at end of file
diff --git a/src/test/burn/TestData/Manual/BundleA/BundleA.wxs b/src/test/burn/TestData/Manual/BundleA/BundleA.wxs
new file mode 100644
index 00000000..1706f4e8
--- /dev/null
+++ b/src/test/burn/TestData/Manual/BundleA/BundleA.wxs
@@ -0,0 +1,13 @@
+<!-- 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. -->
+
+
+<Wix xmlns="http://wixtoolset.org/schemas/v4/wxs" xmlns:bal="http://wixtoolset.org/schemas/v4/wxs/bal">
+  <Fragment>
+    <BootstrapperApplication>
+      <Payload SourceFile="$(var.BafThmUtilTesting.TargetPath)" bal:BAFunctions="yes" />
+    </BootstrapperApplication>
+    <PackageGroup Id="BundlePackages">
+      <MsiPackage Id="PackageA" SourceFile="$(var.PackageA.TargetPath)" />
+    </PackageGroup>
+  </Fragment>
+</Wix>
diff --git a/src/test/burn/TestData/Manual/PackageA/PackageA.wixproj b/src/test/burn/TestData/Manual/PackageA/PackageA.wixproj
new file mode 100644
index 00000000..c9c81c3a
--- /dev/null
+++ b/src/test/burn/TestData/Manual/PackageA/PackageA.wixproj
@@ -0,0 +1,9 @@
+<!-- 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. -->
+<Project Sdk="WixToolset.Sdk">
+  <PropertyGroup>
+    <UpgradeCode>{0D803A6E-8090-4174-8DAC-810ECC2B1BBF}</UpgradeCode>
+  </PropertyGroup>
+  <ItemGroup>
+    <Compile Include="..\..\Templates\Package.wxs" Link="Package.wxs" />
+  </ItemGroup>
+</Project>
\ No newline at end of file