| /* |
| * Copyright (C) 2004 Mike Hearn, for CodeWeavers |
| * Copyright (C) 2005 Robert Shearman |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA |
| */ |
| |
| /* There are two types of window involved here. The first is the |
| * listener window. This is like the taskbar in Windows. It doesn't |
| * ever appear on-screen in our implementation, instead we create |
| * individual mini "adaptor" windows which are docked by the native |
| * systray host. |
| * |
| * In future for those who don't have a systray we could make the |
| * listener window more clever so it can draw itself like the Windows |
| * tray area does (with a clock and stuff). |
| */ |
| |
| #include <assert.h> |
| |
| #define UNICODE |
| #define _WIN32_IE 0x500 |
| #include <windows.h> |
| |
| #include <wine/debug.h> |
| #include <wine/list.h> |
| |
| #include "explorer_private.h" |
| |
| WINE_DEFAULT_DEBUG_CHANNEL(systray); |
| |
| #define IS_OPTION_FALSE(ch) \ |
| ((ch) == 'n' || (ch) == 'N' || (ch) == 'f' || (ch) == 'F' || (ch) == '0') |
| |
| const static WCHAR adaptor_classname[] = /* Adaptor */ {'A','d','a','p','t','o','r',0}; |
| |
| /* tray state */ |
| struct tray |
| { |
| HWND window; |
| struct list icons; |
| }; |
| |
| /* an individual systray icon, unpacked from the NOTIFYICONDATA and always in unicode */ |
| struct icon |
| { |
| struct list entry; |
| HICON image; /* the image to render */ |
| HWND owner; /* the HWND passed in to the Shell_NotifyIcon call */ |
| HWND window; /* the adaptor window */ |
| UINT id; /* the unique id given by the app */ |
| UINT callback_message; |
| }; |
| |
| static struct tray tray; |
| static BOOL hide_systray; |
| |
| /* adaptor code */ |
| |
| #define ICON_SIZE GetSystemMetrics(SM_CXSMICON) |
| /* space around icon (forces icon to center of KDE systray area) */ |
| #define ICON_BORDER 4 |
| |
| static LRESULT WINAPI adaptor_wndproc(HWND window, UINT msg, |
| WPARAM wparam, LPARAM lparam) |
| { |
| struct icon *icon = NULL; |
| BOOL ret; |
| |
| WINE_TRACE("hwnd=%p, msg=0x%x\n", window, msg); |
| |
| /* set the icon data for the window from the data passed into CreateWindow */ |
| if (msg == WM_NCCREATE) |
| SetWindowLongPtrW(window, GWLP_USERDATA, (LPARAM)((const CREATESTRUCT *)lparam)->lpCreateParams); |
| |
| icon = (struct icon *) GetWindowLongPtr(window, GWLP_USERDATA); |
| |
| switch (msg) |
| { |
| case WM_PAINT: |
| { |
| RECT rc; |
| int top; |
| PAINTSTRUCT ps; |
| HDC hdc; |
| |
| WINE_TRACE("painting\n"); |
| |
| hdc = BeginPaint(window, &ps); |
| GetClientRect(window, &rc); |
| |
| /* calculate top so we can deal with arbitrary sized trays */ |
| top = ((rc.bottom-rc.top)/2) - ((ICON_SIZE)/2); |
| |
| DrawIconEx(hdc, (ICON_BORDER/2), top, icon->image, |
| ICON_SIZE, ICON_SIZE, 0, 0, DI_DEFAULTSIZE|DI_NORMAL); |
| |
| EndPaint(window, &ps); |
| break; |
| } |
| |
| case WM_MOUSEMOVE: |
| case WM_LBUTTONDOWN: |
| case WM_LBUTTONUP: |
| case WM_RBUTTONDOWN: |
| case WM_RBUTTONUP: |
| case WM_MBUTTONDOWN: |
| case WM_MBUTTONUP: |
| case WM_LBUTTONDBLCLK: |
| case WM_RBUTTONDBLCLK: |
| case WM_MBUTTONDBLCLK: |
| { |
| /* notify the owner hwnd of the message */ |
| WINE_TRACE("relaying 0x%x\n", msg); |
| ret = PostMessage(icon->owner, icon->callback_message, (WPARAM) icon->id, (LPARAM) msg); |
| if (!ret && (GetLastError() == ERROR_INVALID_HANDLE)) |
| { |
| WINE_WARN("application window was destroyed without removing " |
| "notification icon, removing automatically\n"); |
| DestroyWindow(window); |
| } |
| return 0; |
| } |
| |
| case WM_NCDESTROY: |
| SetWindowLongPtr(window, GWLP_USERDATA, 0); |
| |
| list_remove(&icon->entry); |
| DestroyIcon(icon->image); |
| HeapFree(GetProcessHeap(), 0, icon); |
| break; |
| } |
| |
| return DefWindowProc(window, msg, wparam, lparam); |
| } |
| |
| |
| /* listener code */ |
| |
| static struct icon *get_icon(HWND owner, UINT id) |
| { |
| struct icon *this; |
| |
| /* search for the icon */ |
| LIST_FOR_EACH_ENTRY( this, &tray.icons, struct icon, entry ) |
| if ((this->id == id) && (this->owner == owner)) return this; |
| |
| return NULL; |
| } |
| |
| static void modify_icon(const NOTIFYICONDATAW *nid) |
| { |
| struct icon *icon; |
| |
| WINE_TRACE("id=0x%x, hwnd=%p\n", nid->uID, nid->hWnd); |
| |
| /* demarshal the request from the NID */ |
| icon = get_icon(nid->hWnd, nid->uID); |
| if (!icon) |
| { |
| WINE_WARN("Invalid icon ID (0x%x) for HWND %p\n", nid->uID, nid->hWnd); |
| return; |
| } |
| |
| if (nid->uFlags & NIF_ICON) |
| { |
| if (icon->image) DestroyIcon(icon->image); |
| icon->image = CopyIcon(nid->hIcon); |
| |
| RedrawWindow(icon->window, NULL, NULL, RDW_ERASE | RDW_INVALIDATE | RDW_UPDATENOW); |
| } |
| |
| if (nid->uFlags & NIF_MESSAGE) |
| { |
| icon->callback_message = nid->uCallbackMessage; |
| } |
| } |
| |
| static void add_icon(const NOTIFYICONDATAW *nid) |
| { |
| RECT rect; |
| struct icon *icon; |
| const static WCHAR adaptor_windowname[] = /* Wine System Tray Adaptor */ {'W','i','n','e',' ','S','y','s','t','e','m',' ','T','r','a','y',' ','A','d','a','p','t','o','r',0}; |
| |
| WINE_TRACE("id=0x%x, hwnd=%p\n", nid->uID, nid->hWnd); |
| |
| if ((icon = get_icon(nid->hWnd, nid->uID))) |
| { |
| WINE_WARN("duplicate tray icon add, buggy app?\n"); |
| return; |
| } |
| |
| if (!(icon = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*icon)))) |
| { |
| WINE_ERR("out of memory\n"); |
| return; |
| } |
| |
| icon->id = nid->uID; |
| icon->owner = nid->hWnd; |
| icon->image = NULL; |
| |
| rect.left = 0; |
| rect.top = 0; |
| rect.right = GetSystemMetrics(SM_CXSMICON) + ICON_BORDER; |
| rect.bottom = GetSystemMetrics(SM_CYSMICON) + ICON_BORDER; |
| AdjustWindowRect(&rect, WS_CLIPSIBLINGS | WS_CAPTION, FALSE); |
| |
| /* create the adaptor window */ |
| icon->window = CreateWindowEx(WS_EX_TRAYWINDOW, adaptor_classname, |
| adaptor_windowname, |
| WS_CLIPSIBLINGS | WS_CAPTION, |
| CW_USEDEFAULT, CW_USEDEFAULT, |
| rect.right - rect.left, |
| rect.bottom - rect.top, |
| NULL, NULL, NULL, icon); |
| |
| if (!hide_systray) |
| ShowWindow(icon->window, SW_SHOWNA); |
| |
| list_add_tail(&tray.icons, &icon->entry); |
| |
| modify_icon(nid); |
| } |
| |
| static void delete_icon(const NOTIFYICONDATAW *nid) |
| { |
| struct icon *icon = get_icon(nid->hWnd, nid->uID); |
| |
| WINE_TRACE("id=0x%x, hwnd=%p\n", nid->uID, nid->hWnd); |
| |
| if (!icon) |
| { |
| WINE_ERR("invalid tray icon ID specified: %ud\n", nid->uID); |
| return; |
| } |
| |
| DestroyWindow(icon->window); |
| } |
| |
| static void handle_incoming(HWND hwndSource, COPYDATASTRUCT *cds) |
| { |
| NOTIFYICONDATAW nid; |
| |
| if (cds->cbData < sizeof(nid)) return; |
| memcpy(&nid, cds->lpData, sizeof(nid)); |
| |
| /* FIXME: if statement only needed because we don't support interprocess |
| * icon handles */ |
| if ((nid.uFlags & NIF_ICON) && (cds->cbData >= sizeof(nid) + 2 * sizeof(BITMAP))) |
| { |
| LONG cbMaskBits; |
| LONG cbColourBits; |
| BITMAP bmMask; |
| BITMAP bmColour; |
| const char *buffer = cds->lpData; |
| |
| buffer += sizeof(nid); |
| |
| memcpy(&bmMask, buffer, sizeof(bmMask)); |
| buffer += sizeof(bmMask); |
| memcpy(&bmColour, buffer, sizeof(bmColour)); |
| buffer += sizeof(bmColour); |
| |
| cbMaskBits = (bmMask.bmPlanes * bmMask.bmWidth * bmMask.bmHeight * bmMask.bmBitsPixel) / 8; |
| cbColourBits = (bmColour.bmPlanes * bmColour.bmWidth * bmColour.bmHeight * bmColour.bmBitsPixel) / 8; |
| |
| if (cds->cbData < sizeof(nid) + 2 * sizeof(BITMAP) + cbMaskBits + cbColourBits) |
| { |
| WINE_ERR("buffer underflow\n"); |
| return; |
| } |
| |
| /* sanity check */ |
| if ((bmColour.bmWidth != bmMask.bmWidth) || (bmColour.bmHeight != bmMask.bmHeight)) |
| { |
| WINE_ERR("colour and mask bitmaps aren't consistent\n"); |
| return; |
| } |
| |
| nid.hIcon = CreateIcon(NULL, bmColour.bmWidth, bmColour.bmHeight, |
| bmColour.bmPlanes, bmColour.bmBitsPixel, |
| buffer, buffer + cbMaskBits); |
| } |
| |
| switch (cds->dwData) |
| { |
| case NIM_ADD: |
| add_icon(&nid); |
| break; |
| case NIM_DELETE: |
| delete_icon(&nid); |
| break; |
| case NIM_MODIFY: |
| modify_icon(&nid); |
| break; |
| default: |
| WINE_FIXME("unhandled tray message: %ld\n", cds->dwData); |
| break; |
| } |
| |
| /* FIXME: if statement only needed because we don't support interprocess |
| * icon handles */ |
| if (nid.uFlags & NIF_ICON) |
| DestroyIcon(nid.hIcon); |
| } |
| |
| static LRESULT WINAPI listener_wndproc(HWND window, UINT msg, |
| WPARAM wparam, LPARAM lparam) |
| { |
| if (msg == WM_COPYDATA) |
| handle_incoming((HWND)wparam, (COPYDATASTRUCT *)lparam); |
| |
| return DefWindowProc(window, msg, wparam, lparam); |
| } |
| |
| |
| static BOOL is_systray_hidden(void) |
| { |
| const WCHAR show_systray_keyname[] = {'S','o','f','t','w','a','r','e','\\','W','i','n','e','\\', |
| 'X','1','1',' ','D','r','i','v','e','r',0}; |
| const WCHAR show_systray_valuename[] = {'S','h','o','w','S','y','s','t','r','a','y',0}; |
| HKEY hkey; |
| BOOL ret = FALSE; |
| |
| /* @@ Wine registry key: HKCU\Software\Wine\X11 Driver */ |
| if (RegOpenKeyW(HKEY_CURRENT_USER, show_systray_keyname, &hkey) == ERROR_SUCCESS) |
| { |
| WCHAR value[10]; |
| DWORD type, size = sizeof(value); |
| if (RegQueryValueExW(hkey, show_systray_valuename, 0, &type, (LPBYTE)&value, &size) == ERROR_SUCCESS) |
| { |
| ret = IS_OPTION_FALSE(value[0]); |
| } |
| RegCloseKey(hkey); |
| } |
| return ret; |
| } |
| |
| /* this function creates the the listener window */ |
| void initialize_systray(void) |
| { |
| WNDCLASSEX class; |
| static const WCHAR classname[] = /* Shell_TrayWnd */ {'S','h','e','l','l','_','T','r','a','y','W','n','d',0}; |
| static const WCHAR winname[] = /* Wine Systray Listener */ |
| {'W','i','n','e',' ','S','y','s','t','r','a','y',' ','L','i','s','t','e','n','e','r',0}; |
| |
| WINE_TRACE("initiaizing\n"); |
| |
| hide_systray = is_systray_hidden(); |
| |
| list_init(&tray.icons); |
| |
| /* register the systray listener window class */ |
| ZeroMemory(&class, sizeof(class)); |
| class.cbSize = sizeof(class); |
| class.lpfnWndProc = &listener_wndproc; |
| class.hInstance = NULL; |
| class.hIcon = LoadIcon(0, IDI_WINLOGO); |
| class.hCursor = LoadCursor(0, IDC_ARROW); |
| class.hbrBackground = (HBRUSH) COLOR_WINDOW; |
| class.lpszClassName = (WCHAR *) &classname; |
| |
| if (!RegisterClassEx(&class)) |
| { |
| WINE_ERR("Could not register SysTray window class\n"); |
| return; |
| } |
| |
| /* now register the adaptor window class */ |
| ZeroMemory(&class, sizeof(class)); |
| class.cbSize = sizeof(class); |
| class.lpfnWndProc = adaptor_wndproc; |
| class.hInstance = NULL; |
| class.hIcon = LoadIcon(0, IDI_WINLOGO); |
| class.hCursor = LoadCursor(0, IDC_ARROW); |
| class.hbrBackground = (HBRUSH) COLOR_WINDOW; |
| class.lpszClassName = adaptor_classname; |
| class.style = CS_SAVEBITS | CS_DBLCLKS; |
| |
| if (!RegisterClassEx(&class)) |
| { |
| WINE_ERR("Could not register adaptor class\n"); |
| return; |
| } |
| |
| tray.window = CreateWindow(classname, winname, WS_OVERLAPPED, |
| CW_USEDEFAULT, CW_USEDEFAULT, |
| 0, 0, 0, 0, 0, 0); |
| |
| if (!tray.window) |
| { |
| WINE_ERR("Could not create tray window\n"); |
| return; |
| } |
| } |