fix(windows): allow TaskbarCreated message through UIPI for admin processes

Author: Slinetrac

src/platform_impl/windows/mod.rs

@@ -17,14 +17,15 @@ use windows_sys::{
                 NIM_ADD, NIM_DELETE, NIM_MODIFY, NOTIFYICONDATAW, NOTIFYICONIDENTIFIER,
             },
             WindowsAndMessaging::{
-                CreateWindowExW, DefWindowProcW, DestroyWindow, FindWindowW, GetCursorPos,
-                KillTimer, RegisterClassW, RegisterWindowMessageA, SendMessageW,
-                SetForegroundWindow, SetTimer, TrackPopupMenu, CREATESTRUCTW, CW_USEDEFAULT,
-                GWL_USERDATA, HICON, HMENU, TPM_BOTTOMALIGN, TPM_LEFTALIGN, WM_CREATE, WM_DESTROY,
-                WM_LBUTTONDBLCLK, WM_LBUTTONDOWN, WM_LBUTTONUP, WM_MBUTTONDBLCLK, WM_MBUTTONDOWN,
-                WM_MBUTTONUP, WM_MOUSEMOVE, WM_NCCREATE, WM_RBUTTONDBLCLK, WM_RBUTTONDOWN,
-                WM_RBUTTONUP, WM_TIMER, WNDCLASSW, WS_EX_LAYERED, WS_EX_NOACTIVATE,
-                WS_EX_TOOLWINDOW, WS_EX_TRANSPARENT, WS_OVERLAPPED,
+                ChangeWindowMessageFilterEx, CreateWindowExW, DefWindowProcW, DestroyWindow,
+                FindWindowW, GetCursorPos, KillTimer, RegisterClassW, RegisterWindowMessageA,
+                SendMessageW, SetForegroundWindow, SetTimer, TrackPopupMenu, CREATESTRUCTW,
+                CW_USEDEFAULT, GWL_USERDATA, HICON, HMENU, MSGFLT_ALLOW, TPM_BOTTOMALIGN,
+                TPM_LEFTALIGN, WM_CREATE, WM_DESTROY, WM_LBUTTONDBLCLK, WM_LBUTTONDOWN,
+                WM_LBUTTONUP, WM_MBUTTONDBLCLK, WM_MBUTTONDOWN, WM_MBUTTONUP, WM_MOUSEMOVE,
+                WM_NCCREATE, WM_RBUTTONDBLCLK, WM_RBUTTONDOWN, WM_RBUTTONUP, WM_TIMER, WNDCLASSW,
+                WS_EX_LAYERED, WS_EX_NOACTIVATE, WS_EX_TOOLWINDOW, WS_EX_TRANSPARENT,
+                WS_OVERLAPPED,
             },
         },
     },
@@ -127,6 +128,8 @@ impl TrayIcon {
                 return Err(crate::Error::OsError(std::io::Error::last_os_error()));
             }
 
+            allow_taskbar_restart_message(hwnd);
+
             let hicon = attrs.icon.as_ref().map(|i| i.inner.as_raw_handle());
 
             let registered = register_tray_icon(hwnd, internal_id, &hicon, &attrs.tooltip);
@@ -546,6 +549,12 @@ fn taskbar_window_exists() -> bool {
     !unsafe { FindWindowW(class_name.as_ptr(), ptr::null()) }.is_null()
 }
 
+#[inline]
+unsafe fn allow_taskbar_restart_message(hwnd: HWND) {
+    // Allow TaskbarCreated to cross UIPI for elevated windows.
+    let _ = ChangeWindowMessageFilterEx(hwnd, *S_U_TASKBAR_RESTART, MSGFLT_ALLOW, ptr::null_mut());
+}
+
 #[inline]
 unsafe fn register_tray_icon(
     hwnd: HWND,