private SafeNotifyIconData GetTrayItemIconData(TrayItem trayItem) { SafeNotifyIconData nid = new SafeNotifyIconData(); nid.hWnd = trayItem.hWnd; nid.uID = trayItem.uID; nid.uCallbackMessage = trayItem.uCallbackMessage; nid.szTip = trayItem.szIconText; nid.hIcon = trayItem.hIcon; nid.uVersion = trayItem.uVersion; nid.guidItem = trayItem.guidItem; nid.dwState = 0; nid.uFlags = NIF.GUID | NIF.MESSAGE | NIF.TIP; if (nid.hIcon != IntPtr.Zero) { nid.uFlags |= NIF.ICON; } else { CairoLogger.Instance.Warning($"ExplorerTrayService: Unable to use {trayItem.szIconText} icon handle for NOTIFYICONDATA struct"); } return(nid); }
private void GetTrayItems() { IntPtr toolbarHwnd = FindExplorerTrayToolbarHwnd(); if (toolbarHwnd == IntPtr.Zero) { return; } int count = GetNumTrayIcons(toolbarHwnd); if (count < 1) { return; } GetWindowThreadProcessId(toolbarHwnd, out var processId); IntPtr hProcess = OpenProcess(ProcessAccessFlags.All, false, (int)processId); IntPtr hBuffer = VirtualAllocEx(hProcess, IntPtr.Zero, (uint)Marshal.SizeOf(new TBBUTTON()), AllocationType.Commit, MemoryProtection.ReadWrite); for (int i = 0; i < count; i++) { TrayItem trayItem = GetTrayItem(i, hBuffer, hProcess, toolbarHwnd); if (trayItem.hWnd == IntPtr.Zero || !IsWindow(trayItem.hWnd)) { CairoLogger.Instance.Debug($"ExplorerTrayService: Ignored notify icon {trayItem.szIconText} due to invalid handle"); continue; } SafeNotifyIconData nid = GetTrayItemIconData(trayItem); if (trayDelegate != null) { if (!trayDelegate((uint)NIM.NIM_ADD, nid)) { CairoLogger.Instance.Debug("ExplorerTrayService: Ignored notify icon message"); } } else { CairoLogger.Instance.Debug("ExplorerTrayService: trayDelegate is null"); } } VirtualFreeEx(hProcess, hBuffer, 0, AllocationType.Release); CloseHandle((int)hProcess); }
private bool SysTrayCallback(uint message, SafeNotifyIconData nicData) { if (nicData.hWnd == IntPtr.Zero) { return(false); } NotifyIcon trayIcon = new NotifyIcon(nicData.hWnd); trayIcon.UID = nicData.uID; lock (_lockObject) { if ((NIM)message == NIM.NIM_ADD || (NIM)message == NIM.NIM_MODIFY) { try { bool exists = false; // hide icons while we are shell which require UWP support & we have a separate implementation for if (nicData.guidItem == new Guid(VOLUME_GUID) && ((Shell.IsCairoRunningAsShell && Shell.IsWindows10OrBetter) || GroupPolicyManager.Instance.HideScaVolume)) { return(false); } foreach (NotifyIcon ti in TrayIcons) { if (ti.Equals(nicData)) { exists = true; trayIcon = ti; break; } } if ((NIF.STATE & nicData.uFlags) != 0) { trayIcon.IsHidden = nicData.dwState == 1; } if ((NIF.TIP & nicData.uFlags) != 0 && !string.IsNullOrEmpty(nicData.szTip)) { trayIcon.Title = nicData.szTip; } if ((NIF.ICON & nicData.uFlags) != 0) { if (nicData.hIcon != IntPtr.Zero) { try { System.Windows.Media.Imaging.BitmapSource bs = System.Windows.Interop.Imaging.CreateBitmapSourceFromHIcon( nicData.hIcon, Int32Rect.Empty, System.Windows.Media.Imaging.BitmapSizeOptions.FromEmptyOptions()); DestroyIcon(nicData.hIcon); if (bs != null) { bs.Freeze(); trayIcon.Icon = bs; } } catch { if (trayIcon.Icon == null) { trayIcon.Icon = Common.IconImageConverter.GetDefaultIcon(); } } } else { trayIcon.Icon = null; } } trayIcon.HWnd = nicData.hWnd; trayIcon.UID = nicData.uID; if ((NIF.GUID & nicData.uFlags) != 0) { trayIcon.GUID = nicData.guidItem; } // guess version in case we are receiving icons that aren't sending NIM_SETVERSION to new explorers if ((NIF.VISTA_MASK & nicData.uFlags) != 0) { trayIcon.Version = 4; } else if ((NIF.XP_MASK & nicData.uFlags) != 0) { trayIcon.Version = 3; } if (nicData.uVersion > 0 && nicData.uVersion <= 4) { trayIcon.Version = nicData.uVersion; } if ((NIF.MESSAGE & nicData.uFlags) != 0) { trayIcon.CallbackMessage = nicData.uCallbackMessage; } if (!exists) { // default placement to a menu bar-like rect trayIcon.Placement = defaultPlacement; // set properties used for pinning trayIcon.Path = Shell.GetPathForHandle(trayIcon.HWnd); trayIcon.SetPinValues(); if (trayIcon.Icon == null) { trayIcon.Icon = Common.IconImageConverter.GetDefaultIcon(); } TrayIcons.Add(trayIcon); CairoLogger.Instance.Debug($"NotificationArea: Added: {trayIcon.Title} Path: {trayIcon.Path} Hidden: {trayIcon.IsHidden} GUID: {trayIcon.GUID} UID: {trayIcon.UID} Version: {trayIcon.Version}"); if ((NIM)message == NIM.NIM_MODIFY) { // return an error to the notifyicon as we received a modify for an icon we did not yet have return(false); } } else { CairoLogger.Instance.Debug($"NotificationArea: Modified: {trayIcon.Title}"); } } catch (Exception ex) { CairoLogger.Instance.Error("NotificationArea: Unable to modify the icon in the collection.", ex); } } else if ((NIM)message == NIM.NIM_DELETE) { try { if (!TrayIcons.Contains(trayIcon)) { // Nothing to remove. return(false); } TrayIcons.Remove(trayIcon); CairoLogger.Instance.Debug($"NotificationArea: Removed: {nicData.szTip}"); } catch (Exception ex) { CairoLogger.Instance.Error("NotificationArea: Unable to remove the icon from the collection.", ex); } } else if ((NIM)message == NIM.NIM_SETVERSION) { foreach (NotifyIcon ti in TrayIcons) { if (ti.Equals(nicData)) { ti.Version = nicData.uVersion; CairoLogger.Instance.Debug($"NotificationArea: Modified version to {ti.Version} on: {ti.Title}"); break; } } } } return(true); }
public bool Equals(SafeNotifyIconData other) { return((HWnd.Equals(other.hWnd) && UID.Equals(other.uID)) || (other.guidItem != Guid.Empty && GUID.Equals(other.guidItem))); }