dlls/shell32/systray.c

   1 /*
   2  * Systray handling
   3  *
   4  * Copyright 1999 Kai Morich    <kai.morich@bigfoot.de>
   5  * Copyright 2004 Mike Hearn, for CodeWeavers
   6  * Copyright 2005 Robert Shearman
   7  *
   8  * This library is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * This library is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with this library; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  21  */
  22
  23 #define NONAMELESSUNION
  24 #define NONAMELESSSTRUCT
  25
  26 #include <stdarg.h>
  27
  28 #include "windef.h"
  29 #include "winbase.h"
  30 #include "wingdi.h"
  31 #include "winnls.h"
  32 #include "winuser.h"
  33 #include "shellapi.h"
  34
  35 #include "wine/debug.h"
  36
  37 WINE_DEFAULT_DEBUG_CHANNEL(systray);
  38
  39 static const WCHAR classname[] = /* Shell_TrayWnd */ {'S','h','e','l','l','_','T','r','a','y','W','n','d','\0'};
  40
  41 /*************************************************************************
  42  * Shell_NotifyIcon                     [SHELL32.296]
  43  * Shell_NotifyIconA                    [SHELL32.297]
  44  */
  45 BOOL WINAPI Shell_NotifyIconA(DWORD dwMessage, PNOTIFYICONDATAA pnid)
  46 {
  47     NOTIFYICONDATAW nidW;
  48     INT cbSize;
  49
  50     /* Validate the cbSize as Windows XP does */
  51     if (pnid->cbSize != NOTIFYICONDATAA_V1_SIZE &&
  52         pnid->cbSize != NOTIFYICONDATAA_V2_SIZE &&
  53         pnid->cbSize != NOTIFYICONDATAA_V3_SIZE &&
  54         pnid->cbSize != sizeof(NOTIFYICONDATAA))
  55     {
  56         WARN("Invalid cbSize (%d) - using only Win95 fields (size=%d)\n",
  57             pnid->cbSize, NOTIFYICONDATAA_V1_SIZE);
  58         cbSize = NOTIFYICONDATAA_V1_SIZE;
  59     }
  60     else
  61         cbSize = pnid->cbSize;
  62
  63     ZeroMemory(&nidW, sizeof(nidW));
  64     nidW.cbSize = sizeof(nidW);
  65     nidW.hWnd   = pnid->hWnd;
  66     nidW.uID    = pnid->uID;
  67     nidW.uFlags = pnid->uFlags;
  68     nidW.uCallbackMessage = pnid->uCallbackMessage;
  69     nidW.hIcon  = pnid->hIcon;
  70
  71     /* szTip */
  72     if (pnid->uFlags & NIF_TIP)
  73         MultiByteToWideChar(CP_ACP, 0, pnid->szTip, -1, nidW.szTip, sizeof(nidW.szTip)/sizeof(WCHAR));
  74
  75     if (cbSize >= NOTIFYICONDATAA_V2_SIZE)
  76     {
  77         nidW.dwState      = pnid->dwState;
  78         nidW.dwStateMask  = pnid->dwStateMask;
  79
  80         /* szInfo, szInfoTitle */
  81         if (pnid->uFlags & NIF_INFO)
  82         {
  83             MultiByteToWideChar(CP_ACP, 0, pnid->szInfo, -1,  nidW.szInfo, sizeof(nidW.szInfo)/sizeof(WCHAR));
  84             MultiByteToWideChar(CP_ACP, 0, pnid->szInfoTitle, -1, nidW.szInfoTitle, sizeof(nidW.szInfoTitle)/sizeof(WCHAR));
  85         }
  86
  87         nidW.u.uTimeout = pnid->u.uTimeout;
  88         nidW.dwInfoFlags = pnid->dwInfoFlags;
  89     }
  90
  91     if (cbSize >= NOTIFYICONDATAA_V3_SIZE)
  92         nidW.guidItem = pnid->guidItem;
  93
  94     if (cbSize >= sizeof(NOTIFYICONDATAA))
  95         nidW.hBalloonIcon = pnid->hBalloonIcon;
  96     return Shell_NotifyIconW(dwMessage, &nidW);
  97 }
  98
  99 /*************************************************************************
 100  * Shell_NotifyIconW                    [SHELL32.298]
 101  */
 102 BOOL WINAPI Shell_NotifyIconW(DWORD dwMessage, PNOTIFYICONDATAW nid)
 103 {
 104     HWND tray;
 105     COPYDATASTRUCT cds;
 106     char *buffer = NULL;
 107     BOOL ret;
 108
 109     TRACE("dwMessage = %d, nid->cbSize=%d\n", dwMessage, nid->cbSize);
 110
 111     /* Validate the cbSize so that WM_COPYDATA doesn't crash the application */
 112     if (nid->cbSize != NOTIFYICONDATAW_V1_SIZE &&
 113         nid->cbSize != NOTIFYICONDATAW_V2_SIZE &&
 114         nid->cbSize != NOTIFYICONDATAW_V3_SIZE &&
 115         nid->cbSize != sizeof(NOTIFYICONDATAW))
 116     {
 117         NOTIFYICONDATAW newNid;
 118
 119         WARN("Invalid cbSize (%d) - using only Win95 fields (size=%d)\n",
 120             nid->cbSize, NOTIFYICONDATAW_V1_SIZE);
 121         CopyMemory(&newNid, nid, NOTIFYICONDATAW_V1_SIZE);
 122         newNid.cbSize = NOTIFYICONDATAW_V1_SIZE;
 123         return Shell_NotifyIconW(dwMessage, &newNid);
 124     }
 125
 126     tray = FindWindowExW(0, NULL, classname, NULL);
 127     if (!tray) return FALSE;
 128
 129     cds.dwData = dwMessage;
 130
 131     /* FIXME: if statement only needed because we don't support interprocess
 132      * icon handles */
 133     if (nid->uFlags & NIF_ICON)
 134     {
 135         ICONINFO iconinfo;
 136         BITMAP bmMask;
 137         BITMAP bmColour;
 138         LONG cbMaskBits;
 139         LONG cbColourBits;
 140
 141         if (!GetIconInfo(nid->hIcon, &iconinfo))
 142             goto noicon;
 143
 144         if (!GetObjectW(iconinfo.hbmMask, sizeof(bmMask), &bmMask) ||
 145             !GetObjectW(iconinfo.hbmColor, sizeof(bmColour), &bmColour))
 146         {
 147             DeleteObject(iconinfo.hbmMask);
 148             DeleteObject(iconinfo.hbmColor);
 149             goto noicon;
 150         }
 151
 152         cbMaskBits = (bmMask.bmPlanes * bmMask.bmWidth * bmMask.bmHeight * bmMask.bmBitsPixel) / 8;
 153         cbColourBits = (bmColour.bmPlanes * bmColour.bmWidth * bmColour.bmHeight * bmColour.bmBitsPixel) / 8;
 154         cds.cbData = nid->cbSize + 2*sizeof(BITMAP) + cbMaskBits + cbColourBits;
 155         buffer = HeapAlloc(GetProcessHeap(), 0, cds.cbData);
 156         if (!buffer)
 157         {
 158             DeleteObject(iconinfo.hbmMask);
 159             DeleteObject(iconinfo.hbmColor);
 160             return FALSE;
 161         }
 162         cds.lpData = buffer;
 163
 164         memcpy(buffer, nid, nid->cbSize);
 165         buffer += nid->cbSize;
 166         memcpy(buffer, &bmMask, sizeof(bmMask));
 167         buffer += sizeof(bmMask);
 168         memcpy(buffer, &bmColour, sizeof(bmColour));
 169         buffer += sizeof(bmColour);
 170         GetBitmapBits(iconinfo.hbmMask, cbMaskBits, buffer);
 171         buffer += cbMaskBits;
 172         GetBitmapBits(iconinfo.hbmColor, cbColourBits, buffer);
 173
 174         /* Reset pointer to allocated block so it can be freed later.
 175          * Note that cds.lpData cannot be passed to HeapFree since it
 176          * points to nid when no icon info is found. */
 177         buffer = cds.lpData;
 178
 179         DeleteObject(iconinfo.hbmMask);
 180         DeleteObject(iconinfo.hbmColor);
 181     }
 182     else
 183     {
 184 noicon:
 185         cds.cbData = nid->cbSize;
 186         cds.lpData = nid;
 187     }
 188
 189     ret = SendMessageW(tray, WM_COPYDATA, (WPARAM)nid->hWnd, (LPARAM)&cds);
 190
 191     /* FIXME: if statement only needed because we don't support interprocess
 192      * icon handles */
 193     HeapFree(GetProcessHeap(), 0, buffer);
 194
 195     return ret;
 196 }