1 files changed, 50 insertions, 0 deletions

diff --git a/src/ext/Bal/stdbas/stdbas.cpp b/src/ext/Bal/stdbas/stdbas.cpp
new file mode 100644
index 00000000..909bdc23
--- /dev/null
+++ b/src/ext/Bal/stdbas/stdbas.cpp
@@ -0,0 +1,50 @@
+// 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 const LPCWSTR WIX_BUNDLE_ICON_FILENAME = L"WixBundle.ico";
+
+
+//
+// LoadBundleIcon - loads the icon that was (optionally) authored in the bundle otherwise use the one embedded in the bootstrapper application.
+//
+HRESULT LoadBundleIcon(
+    __in HMODULE hModule,
+    __out HICON* phIcon,
+    __out HICON* phSmallIcon
+)
+{
+    HRESULT hr = S_OK;
+    LPWSTR sczIconPath = NULL;
+    int nIconWidth = ::GetSystemMetrics(SM_CXICON);
+    int nIconHeight = ::GetSystemMetrics(SM_CYICON);
+    int nSmallIconWidth = ::GetSystemMetrics(SM_CXSMICON);
+    int nSmallIconHeight = ::GetSystemMetrics(SM_CYSMICON);
+    HICON hIcon = NULL;
+    HICON hSmallIcon = NULL;
+
+    // First look for the optional authored bundle icon.
+    hr = PathRelativeToModule(&sczIconPath, WIX_BUNDLE_ICON_FILENAME, hModule);
+    ExitOnFailure(hr, "Failed to get path to bundle icon: %ls", WIX_BUNDLE_ICON_FILENAME);
+
+    if (FileExistsEx(sczIconPath, NULL))
+    {
+        hIcon = reinterpret_cast<HICON>(::LoadImageW(NULL, sczIconPath, IMAGE_ICON, nIconWidth, nIconHeight, LR_LOADFROMFILE));
+
+        hSmallIcon = reinterpret_cast<HICON>(::LoadImageW(NULL, sczIconPath, IMAGE_ICON, nSmallIconWidth, nSmallIconHeight, LR_LOADFROMFILE));
+    }
+    else // fallback to the first icon resource in the bootstrapper application.
+    {
+        hIcon = reinterpret_cast<HICON>(::LoadImageW(hModule, MAKEINTRESOURCEW(1), IMAGE_ICON, nIconWidth, nIconHeight, LR_DEFAULTCOLOR));
+
+        hSmallIcon = reinterpret_cast<HICON>(::LoadImageW(hModule, MAKEINTRESOURCEW(1), IMAGE_ICON, nSmallIconWidth, nSmallIconHeight, LR_DEFAULTCOLOR));
+    }
+
+    *phIcon = hIcon;
+    *phSmallIcon = hSmallIcon;
+
+LExit:
+    ReleaseStr(sczIconPath);
+
+    return hr;
+}