systray.c 7.8 KB
Newer Older
1
/*
2
 * Systray handling
3
 *
4 5 6
 * Copyright 1999 Kai Morich	<kai.morich@bigfoot.de>
 * Copyright 2004 Mike Hearn, for CodeWeavers
 * Copyright 2005 Robert Shearman
7
 *
8 9 10 11 12 13 14 15 16 17 18 19
 * 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
20
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 22
 */

23 24
#define NONAMELESSUNION
#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 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
struct notify_data  /* platform-independent format for NOTIFYICONDATA */
{
    LONG  hWnd;
    UINT  uID;
    UINT  uFlags;
    UINT  uCallbackMessage;
    WCHAR szTip[128];
    DWORD dwState;
    DWORD dwStateMask;
    WCHAR szInfo[256];
    union {
        UINT uTimeout;
        UINT uVersion;
    } u;
    WCHAR szInfoTitle[64];
    DWORD dwInfoFlags;
    GUID  guidItem;
    /* data for the icon bitmap */
    UINT width;
    UINT height;
    UINT planes;
    UINT bpp;
};

65 66 67 68 69
/*************************************************************************
 * Shell_NotifyIcon			[SHELL32.296]
 * Shell_NotifyIconA			[SHELL32.297]
 */
BOOL WINAPI Shell_NotifyIconA(DWORD dwMessage, PNOTIFYICONDATAA pnid)
70
{
71
    NOTIFYICONDATAW nidW;
72 73 74 75 76 77 78 79 80 81 82 83 84 85
    INT cbSize;

    /* Validate the cbSize as Windows XP does */
    if (pnid->cbSize != NOTIFYICONDATAA_V1_SIZE &&
        pnid->cbSize != NOTIFYICONDATAA_V2_SIZE &&
        pnid->cbSize != NOTIFYICONDATAA_V3_SIZE &&
        pnid->cbSize != sizeof(NOTIFYICONDATAA))
    {
        WARN("Invalid cbSize (%d) - using only Win95 fields (size=%d)\n",
            pnid->cbSize, NOTIFYICONDATAA_V1_SIZE);
        cbSize = NOTIFYICONDATAA_V1_SIZE;
    }
    else
        cbSize = pnid->cbSize;
86 87

    ZeroMemory(&nidW, sizeof(nidW));
88 89 90 91 92 93
    nidW.cbSize = sizeof(nidW);
    nidW.hWnd   = pnid->hWnd;
    nidW.uID    = pnid->uID;
    nidW.uFlags = pnid->uFlags;
    nidW.uCallbackMessage = pnid->uCallbackMessage;
    nidW.hIcon  = pnid->hIcon;
94

95
    /* szTip */
96 97
    if (pnid->uFlags & NIF_TIP)
        MultiByteToWideChar(CP_ACP, 0, pnid->szTip, -1, nidW.szTip, sizeof(nidW.szTip)/sizeof(WCHAR));
98

99
    if (cbSize >= NOTIFYICONDATAA_V2_SIZE)
100 101 102
    {
        nidW.dwState      = pnid->dwState;
        nidW.dwStateMask  = pnid->dwStateMask;
103

104 105 106 107 108 109
        /* szInfo, szInfoTitle */
        if (pnid->uFlags & NIF_INFO)
        {
            MultiByteToWideChar(CP_ACP, 0, pnid->szInfo, -1,  nidW.szInfo, sizeof(nidW.szInfo)/sizeof(WCHAR));
            MultiByteToWideChar(CP_ACP, 0, pnid->szInfoTitle, -1, nidW.szInfoTitle, sizeof(nidW.szInfoTitle)/sizeof(WCHAR));
        }
110

111 112 113
        nidW.u.uTimeout = pnid->u.uTimeout;
        nidW.dwInfoFlags = pnid->dwInfoFlags;
    }
114
    
115
    if (cbSize >= NOTIFYICONDATAA_V3_SIZE)
116
        nidW.guidItem = pnid->guidItem;
117

118
    if (cbSize >= sizeof(NOTIFYICONDATAA))
119
        nidW.hBalloonIcon = pnid->hBalloonIcon;
120
    return Shell_NotifyIconW(dwMessage, &nidW);
121 122
}

123 124 125 126
/*************************************************************************
 * Shell_NotifyIconW			[SHELL32.298]
 */
BOOL WINAPI Shell_NotifyIconW(DWORD dwMessage, PNOTIFYICONDATAW nid)
127
{
128 129
    HWND tray;
    COPYDATASTRUCT cds;
130 131
    struct notify_data data_buffer;
    struct notify_data *data = &data_buffer;
132
    BOOL ret;
133

134
    TRACE("dwMessage = %d, nid->cbSize=%d\n", dwMessage, nid->cbSize);
135

136 137 138 139 140 141 142 143 144 145 146 147 148 149 150
    /* Validate the cbSize so that WM_COPYDATA doesn't crash the application */
    if (nid->cbSize != NOTIFYICONDATAW_V1_SIZE &&
        nid->cbSize != NOTIFYICONDATAW_V2_SIZE &&
        nid->cbSize != NOTIFYICONDATAW_V3_SIZE &&
        nid->cbSize != sizeof(NOTIFYICONDATAW))
    {
        NOTIFYICONDATAW newNid;

        WARN("Invalid cbSize (%d) - using only Win95 fields (size=%d)\n",
            nid->cbSize, NOTIFYICONDATAW_V1_SIZE);
        CopyMemory(&newNid, nid, NOTIFYICONDATAW_V1_SIZE);
        newNid.cbSize = NOTIFYICONDATAW_V1_SIZE;
        return Shell_NotifyIconW(dwMessage, &newNid);
    }

151 152 153 154
    tray = FindWindowExW(0, NULL, classname, NULL);
    if (!tray) return FALSE;

    cds.dwData = dwMessage;
155 156
    cds.cbData = sizeof(*data);
    memset( data, 0, sizeof(*data) );
157 158 159 160 161 162 163 164 165

    /* FIXME: if statement only needed because we don't support interprocess
     * icon handles */
    if (nid->uFlags & NIF_ICON)
    {
        ICONINFO iconinfo;
        BITMAP bmMask;
        BITMAP bmColour;
        LONG cbMaskBits;
166
        LONG cbColourBits = 0;
167
        char *buffer;
168 169

        if (!GetIconInfo(nid->hIcon, &iconinfo))
170
            goto noicon;
171 172

        if (!GetObjectW(iconinfo.hbmMask, sizeof(bmMask), &bmMask) ||
173
            (iconinfo.hbmColor && !GetObjectW(iconinfo.hbmColor, sizeof(bmColour), &bmColour)))
174 175
        {
            DeleteObject(iconinfo.hbmMask);
176
            if (iconinfo.hbmColor) DeleteObject(iconinfo.hbmColor);
177
            goto noicon;
178
        }
179

180
        cbMaskBits = (bmMask.bmPlanes * bmMask.bmWidth * bmMask.bmHeight * bmMask.bmBitsPixel + 15) / 16 * 2;
181 182
        if (iconinfo.hbmColor)
            cbColourBits = (bmColour.bmPlanes * bmColour.bmWidth * bmColour.bmHeight * bmColour.bmBitsPixel + 15) / 16 * 2;
183
        cds.cbData = sizeof(*data) + cbMaskBits + cbColourBits;
184
        buffer = HeapAlloc(GetProcessHeap(), 0, cds.cbData);
185 186 187
        if (!buffer)
        {
            DeleteObject(iconinfo.hbmMask);
188
            if (iconinfo.hbmColor) DeleteObject(iconinfo.hbmColor);
189 190
            return FALSE;
        }
191 192 193 194

        data = (struct notify_data *)buffer;
        memset( data, 0, sizeof(*data) );
        buffer += sizeof(*data);
195
        GetBitmapBits(iconinfo.hbmMask, cbMaskBits, buffer);
196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212
        if (!iconinfo.hbmColor)
        {
            data->width  = bmMask.bmWidth;
            data->height = bmMask.bmHeight / 2;
            data->planes = 1;
            data->bpp    = 1;
        }
        else
        {
            data->width  = bmColour.bmWidth;
            data->height = bmColour.bmHeight;
            data->planes = bmColour.bmPlanes;
            data->bpp    = bmColour.bmBitsPixel;
            buffer += cbMaskBits;
            GetBitmapBits(iconinfo.hbmColor, cbColourBits, buffer);
            DeleteObject(iconinfo.hbmColor);
        }
213 214
        DeleteObject(iconinfo.hbmMask);
    }
215

216
noicon:
217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234
    data->hWnd   = HandleToLong( nid->hWnd );
    data->uID    = nid->uID;
    data->uFlags = nid->uFlags;
    if (data->uFlags & NIF_MESSAGE)
        data->uCallbackMessage = nid->uCallbackMessage;
    if (data->uFlags & NIF_TIP)
        lstrcpynW( data->szTip, nid->szTip, sizeof(data->szTip)/sizeof(WCHAR) );
    if (data->uFlags & NIF_STATE)
    {
        data->dwState     = nid->dwState;
        data->dwStateMask = nid->dwStateMask;
    }
    if (data->uFlags & NIF_INFO)
    {
        lstrcpynW( data->szInfo, nid->szInfo, sizeof(data->szInfo)/sizeof(WCHAR) );
        lstrcpynW( data->szInfoTitle, nid->szInfoTitle, sizeof(data->szInfoTitle)/sizeof(WCHAR) );
        data->u.uTimeout  = nid->u.uTimeout;
        data->dwInfoFlags = nid->dwInfoFlags;
235
    }
236 237 238
    if (data->uFlags & NIF_GUID)
        data->guidItem = nid->guidItem;
    /* FIXME: balloon icon */
Juergen Schmied's avatar
Juergen Schmied committed
239

240
    cds.lpData = data;
241
    ret = SendMessageW(tray, WM_COPYDATA, (WPARAM)nid->hWnd, (LPARAM)&cds);
242
    if (data != &data_buffer) HeapFree( GetProcessHeap(), 0, data );
243
    return ret;
Juergen Schmied's avatar
Juergen Schmied committed
244
}