inject.cpp

// Injector: asks XAML Diagnostics to load our TAP DLL into the process that owns
// the taskbar. Windows does the loading; we never touch remote memory ourselves.

#include <windows.h>
#include <xamlom.h>
#include <cstdio>

// Must match tap.cpp
// {6D5F7B21-3C8A-44E1-9E2B-7A1F4C905E10}
static const CLSID CLSID_TaskbarTap =
    { 0x6d5f7b21, 0x3c8a, 0x44e1, { 0x9e, 0x2b, 0x7a, 0x1f, 0x4c, 0x90, 0x5e, 0x10 } };

typedef HRESULT (WINAPI *PFN_InitializeXamlDiagnosticsEx)(
    LPCWSTR endPointName, DWORD pid, LPCWSTR wszDllXamlDiagnostics,
    LPCWSTR wszTAPDllName, CLSID tapClsid, LPCWSTR wszInitializationData);

int wmain(int argc, wchar_t** argv)
{
    if (argc < 2) {
        wprintf(L"usage: inject.exe <full path to tap.dll>\n");
        return 1;
    }
    const wchar_t* tapPath = argv[1];

    HWND tray = FindWindowW(L"Shell_TrayWnd", nullptr);
    if (!tray) { wprintf(L"Shell_TrayWnd not found\n"); return 2; }
    DWORD pid = 0;
    GetWindowThreadProcessId(tray, &pid);
    wprintf(L"taskbar pid = %lu\n", pid);

    HMODULE xaml = LoadLibraryW(L"Windows.UI.Xaml.dll");
    if (!xaml) { wprintf(L"LoadLibrary Windows.UI.Xaml.dll failed %lu\n", GetLastError()); return 3; }

    auto fn = (PFN_InitializeXamlDiagnosticsEx)GetProcAddress(xaml, "InitializeXamlDiagnosticsEx");
    if (!fn) { wprintf(L"InitializeXamlDiagnosticsEx not exported\n"); return 4; }

    HRESULT hr = fn(L"VisualDiagConnection1", pid, nullptr, tapPath, CLSID_TaskbarTap, nullptr);
    wprintf(L"InitializeXamlDiagnosticsEx hr = 0x%08X\n", hr);
    return SUCCEEDED(hr) ? 0 : 5;
}