[PATCH 3/5] comctl32/taskdialog: Add support for loading standard icons.

[Zhiyi Zhang]

Signed-off-by: Zhiyi Zhang <zzhang at codeweavers.com>
---
 dlls/comctl32/taskdialog.c | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/dlls/comctl32/taskdialog.c b/dlls/comctl32/taskdialog.c
index 14c6ccb6e4..59a915977c 100644
--- a/dlls/comctl32/taskdialog.c
+++ b/dlls/comctl32/taskdialog.c
@@ -247,6 +247,20 @@ static void taskdialog_get_label_size(struct taskdialog_info *dialog_info, HWND
     ReleaseDC(hwnd, hdc);
 }
 
+static ULONG_PTR taskdialog_get_standard_icon(LPCWSTR icon)
+{
+    if (icon == TD_WARNING_ICON)
+        return IDI_WARNING;
+    else if (icon == TD_ERROR_ICON)
+        return IDI_ERROR;
+    else if (icon == TD_INFORMATION_ICON)
+        return IDI_INFORMATION;
+    else if (icon == TD_SHIELD_ICON)
+        return IDI_SHIELD;
+    else
+        return (ULONG_PTR)icon;
+}
+
 static void taskdialog_set_icon(struct taskdialog_info *dialog_info, INT element, HICON icon)
 {
     DWORD flags = dialog_info->taskconfig->dwFlags;
@@ -259,6 +273,9 @@ static void taskdialog_set_icon(struct taskdialog_info *dialog_info, INT element
     else
     {
         hicon = LoadImageW(dialog_info->taskconfig->hInstance, (LPCWSTR)icon, IMAGE_ICON, 0, 0, LR_SHARED | LR_DEFAULTSIZE);
+        if (!hicon)
+            hicon = LoadImageW(NULL, (LPCWSTR)taskdialog_get_standard_icon((LPCWSTR)icon), IMAGE_ICON, 0, 0,
+                               LR_SHARED | LR_DEFAULTSIZE);
     }
 
     if (!hicon) return;
-- 
2.17.1