crypt32: Introduce function to encode an array of items as a set.
[wine] / dlls / user32 / msgbox.c
1 /*
2  * Message boxes
3  *
4  * Copyright 1995 Bernd Schmidt
5  * Copyright 2004 Ivan Leo Puoti, Juan Lang
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20  */
21
22 #include <stdarg.h>
23 #include <string.h>
24
25 #include "windef.h"
26 #include "winbase.h"
27 #include "wingdi.h"
28 #include "wine/winbase16.h"
29 #include "wine/winuser16.h"
30 #include "winternl.h"
31 #include "dlgs.h"
32 #include "user_private.h"
33 #include "wine/debug.h"
34
35 WINE_DEFAULT_DEBUG_CHANNEL(dialog);
36 WINE_DECLARE_DEBUG_CHANNEL(msgbox);
37
38 #define MSGBOX_IDICON 1088
39 #define MSGBOX_IDTEXT 100
40 #define IDS_ERROR     2
41
42 struct ThreadWindows
43 {
44     UINT numHandles;
45     UINT numAllocs;
46     HWND *handles;
47 };
48
49 static BOOL CALLBACK MSGBOX_EnumProc(HWND hwnd, LPARAM lParam)
50 {
51     struct ThreadWindows *threadWindows = (struct ThreadWindows *)lParam;
52
53     if (!EnableWindow(hwnd, FALSE))
54     {
55         if(threadWindows->numHandles >= threadWindows->numAllocs)
56         {
57             threadWindows->handles = HeapReAlloc(GetProcessHeap(), 0, threadWindows->handles,
58                                                  (threadWindows->numAllocs*2)*sizeof(HWND));
59             threadWindows->numAllocs *= 2;
60         }
61         threadWindows->handles[threadWindows->numHandles++]=hwnd;
62     }
63    return TRUE;
64 }
65
66 static HFONT MSGBOX_OnInit(HWND hwnd, LPMSGBOXPARAMSW lpmb)
67 {
68     HFONT hFont = 0, hPrevFont = 0;
69     RECT rect;
70     HWND hItem;
71     HDC hdc;
72     int i, buttons;
73     int bspace, bw, bh, theight, tleft, wwidth, wheight, wleft, wtop, bpos;
74     int borheight, borwidth, iheight, ileft, iwidth, twidth, tiheight;
75     NONCLIENTMETRICSW nclm;
76     HMONITOR monitor = 0;
77     MONITORINFO mon_info;
78     LPCWSTR lpszText;
79     WCHAR buf[256];
80
81     nclm.cbSize = sizeof(nclm);
82     SystemParametersInfoW (SPI_GETNONCLIENTMETRICS, 0, &nclm, 0);
83     hFont = CreateFontIndirectW (&nclm.lfMessageFont);
84     /* set button font */
85     for (i=1; i < 8; i++)
86         SendDlgItemMessageW (hwnd, i, WM_SETFONT, (WPARAM)hFont, 0);
87     /* set text font */
88     SendDlgItemMessageW (hwnd, MSGBOX_IDTEXT, WM_SETFONT, (WPARAM)hFont, 0);
89
90     if (HIWORD(lpmb->lpszCaption)) {
91        SetWindowTextW(hwnd, lpmb->lpszCaption);
92     } else {
93        UINT res_id = LOWORD(lpmb->lpszCaption);
94        if (res_id)
95        {
96            if (LoadStringW(lpmb->hInstance, res_id, buf, 256))
97                SetWindowTextW(hwnd, buf);
98        }
99        else
100        {
101            if (LoadStringW(user32_module, IDS_ERROR, buf, 256))
102                SetWindowTextW(hwnd, buf);
103        }
104     }
105     if (HIWORD(lpmb->lpszText)) {
106        lpszText = lpmb->lpszText;
107     } else {
108        lpszText = buf;
109        if (!LoadStringW(lpmb->hInstance, LOWORD(lpmb->lpszText), buf, 256))
110           *buf = 0;     /* FIXME ?? */
111     }
112     
113     TRACE_(msgbox)("%s\n", debugstr_w(lpszText));
114     SetWindowTextW(GetDlgItem(hwnd, MSGBOX_IDTEXT), lpszText);
115
116     /* Hide not selected buttons */
117     switch(lpmb->dwStyle & MB_TYPEMASK) {
118     case MB_OK:
119         ShowWindow(GetDlgItem(hwnd, IDCANCEL), SW_HIDE);
120         /* fall through */
121     case MB_OKCANCEL:
122         ShowWindow(GetDlgItem(hwnd, IDABORT), SW_HIDE);
123         ShowWindow(GetDlgItem(hwnd, IDRETRY), SW_HIDE);
124         ShowWindow(GetDlgItem(hwnd, IDIGNORE), SW_HIDE);
125         ShowWindow(GetDlgItem(hwnd, IDYES), SW_HIDE);
126         ShowWindow(GetDlgItem(hwnd, IDNO), SW_HIDE);
127         break;
128     case MB_ABORTRETRYIGNORE:
129         ShowWindow(GetDlgItem(hwnd, IDOK), SW_HIDE);
130         ShowWindow(GetDlgItem(hwnd, IDCANCEL), SW_HIDE);
131         ShowWindow(GetDlgItem(hwnd, IDYES), SW_HIDE);
132         ShowWindow(GetDlgItem(hwnd, IDNO), SW_HIDE);
133         break;
134     case MB_YESNO:
135         ShowWindow(GetDlgItem(hwnd, IDCANCEL), SW_HIDE);
136         /* fall through */
137     case MB_YESNOCANCEL:
138         ShowWindow(GetDlgItem(hwnd, IDOK), SW_HIDE);
139         ShowWindow(GetDlgItem(hwnd, IDABORT), SW_HIDE);
140         ShowWindow(GetDlgItem(hwnd, IDRETRY), SW_HIDE);
141         ShowWindow(GetDlgItem(hwnd, IDIGNORE), SW_HIDE);
142         break;
143     case MB_RETRYCANCEL:
144         ShowWindow(GetDlgItem(hwnd, IDOK), SW_HIDE);
145         ShowWindow(GetDlgItem(hwnd, IDABORT), SW_HIDE);
146         ShowWindow(GetDlgItem(hwnd, IDIGNORE), SW_HIDE);
147         ShowWindow(GetDlgItem(hwnd, IDYES), SW_HIDE);
148         ShowWindow(GetDlgItem(hwnd, IDNO), SW_HIDE);
149         break;
150     }
151     /* Set the icon */
152     switch(lpmb->dwStyle & MB_ICONMASK) {
153     case MB_ICONEXCLAMATION:
154         SendDlgItemMessageW(hwnd, stc1, STM_SETICON,
155                             (WPARAM)LoadIconW(0, (LPWSTR)IDI_EXCLAMATION), 0);
156         break;
157     case MB_ICONQUESTION:
158         SendDlgItemMessageW(hwnd, stc1, STM_SETICON,
159                             (WPARAM)LoadIconW(0, (LPWSTR)IDI_QUESTION), 0);
160         break;
161     case MB_ICONASTERISK:
162         SendDlgItemMessageW(hwnd, stc1, STM_SETICON,
163                             (WPARAM)LoadIconW(0, (LPWSTR)IDI_ASTERISK), 0);
164         break;
165     case MB_ICONHAND:
166       SendDlgItemMessageW(hwnd, stc1, STM_SETICON,
167                             (WPARAM)LoadIconW(0, (LPWSTR)IDI_HAND), 0);
168       break;
169     case MB_USERICON:
170       SendDlgItemMessageW(hwnd, stc1, STM_SETICON,
171                           (WPARAM)LoadIconW(lpmb->hInstance, lpmb->lpszIcon), 0);
172       break;
173     default:
174         /* By default, Windows 95/98/NT do not associate an icon to message boxes.
175          * So wine should do the same.
176          */
177         break;
178     }
179
180     /* Position everything */
181     GetWindowRect(hwnd, &rect);
182     borheight = rect.bottom - rect.top;
183     borwidth  = rect.right - rect.left;
184     GetClientRect(hwnd, &rect);
185     borheight -= rect.bottom - rect.top;
186     borwidth  -= rect.right - rect.left;
187
188     /* Get the icon height */
189     GetWindowRect(GetDlgItem(hwnd, MSGBOX_IDICON), &rect);
190     MapWindowPoints(0, hwnd, (LPPOINT)&rect, 2);
191     if (!(lpmb->dwStyle & MB_ICONMASK))
192     {
193         rect.bottom = rect.top;
194         rect.right = rect.left;
195     }
196     iheight = rect.bottom - rect.top;
197     ileft = rect.left;
198     iwidth = rect.right - ileft;
199
200     hdc = GetDC(hwnd);
201     if (hFont)
202         hPrevFont = SelectObject(hdc, hFont);
203
204     /* Get the number of visible buttons and their size */
205     bh = bw = 1; /* Minimum button sizes */
206     for (buttons = 0, i = 1; i < 8; i++)
207     {
208         hItem = GetDlgItem(hwnd, i);
209         if (GetWindowLongW(hItem, GWL_STYLE) & WS_VISIBLE)
210         {
211             WCHAR buttonText[1024];
212             int w, h;
213             buttons++;
214             if (GetWindowTextW(hItem, buttonText, 1024))
215             {
216                 DrawTextW( hdc, buttonText, -1, &rect, DT_LEFT | DT_EXPANDTABS | DT_CALCRECT);
217                 h = rect.bottom - rect.top;
218                 w = rect.right - rect.left;
219                 if (h > bh) bh = h;
220                 if (w > bw)  bw = w ;
221             }
222         }
223     }
224     bw = max(bw, bh * 2);
225     /* Button white space */
226     bh = bh * 2;
227     bw = bw * 2;
228     bspace = bw/3; /* Space between buttons */
229
230     /* Get the text size */
231     GetClientRect(GetDlgItem(hwnd, MSGBOX_IDTEXT), &rect);
232     rect.top = rect.left = rect.bottom = 0;
233     DrawTextW( hdc, lpszText, -1, &rect,
234                DT_LEFT | DT_EXPANDTABS | DT_WORDBREAK | DT_CALCRECT);
235     /* Min text width corresponds to space for the buttons */
236     tleft = ileft;
237     if (iwidth) tleft += ileft + iwidth;
238     twidth = max((bw + bspace) * buttons + bspace - tleft, rect.right);
239     theight = rect.bottom;
240
241     if (hFont)
242         SelectObject(hdc, hPrevFont);
243     ReleaseDC(hwnd, hdc);
244
245     tiheight = 16 + max(iheight, theight);
246     wwidth  = tleft + twidth + ileft + borwidth;
247     wheight = 8 + tiheight + bh + borheight;
248
249     /* Message boxes are always desktop centered, so query desktop size and center window */
250     monitor = MonitorFromWindow(lpmb->hwndOwner ? lpmb->hwndOwner : GetActiveWindow(), MONITOR_DEFAULTTOPRIMARY);
251     GetMonitorInfoW(monitor, &mon_info);
252     wleft = (mon_info.rcWork.left + mon_info.rcWork.right - wwidth) / 2;
253     wtop = (mon_info.rcWork.top + mon_info.rcWork.bottom - wheight) / 2;
254
255     /* Resize and center the window */
256     SetWindowPos(hwnd, 0, wleft, wtop, wwidth, wheight,
257                  SWP_NOZORDER | SWP_NOACTIVATE | SWP_NOREDRAW);
258
259     /* Position the icon */
260     SetWindowPos(GetDlgItem(hwnd, MSGBOX_IDICON), 0, ileft, (tiheight - iheight) / 2, 0, 0,
261                  SWP_NOSIZE | SWP_NOZORDER | SWP_NOACTIVATE | SWP_NOREDRAW);
262
263     /* Position the text */
264     SetWindowPos(GetDlgItem(hwnd, MSGBOX_IDTEXT), 0, tleft, (tiheight - theight) / 2, twidth, theight,
265                  SWP_NOZORDER | SWP_NOACTIVATE | SWP_NOREDRAW);
266
267     /* Position the buttons */
268     bpos = (wwidth - (bw + bspace) * buttons + bspace) / 2;
269     for (buttons = i = 0; i < 7; i++) {
270         /* some arithmetic to get the right order for YesNoCancel windows */
271         hItem = GetDlgItem(hwnd, (i + 5) % 7 + 1);
272         if (GetWindowLongW(hItem, GWL_STYLE) & WS_VISIBLE) {
273             if (buttons++ == ((lpmb->dwStyle & MB_DEFMASK) >> 8)) {
274                 SetFocus(hItem);
275                 SendMessageW( hItem, BM_SETSTYLE, BS_DEFPUSHBUTTON, TRUE );
276             }
277             SetWindowPos(hItem, 0, bpos, tiheight, bw, bh,
278                          SWP_NOZORDER|SWP_NOACTIVATE|SWP_NOREDRAW);
279             bpos += bw + bspace;
280         }
281     }
282
283     /*handle modal message boxes*/
284     if (((lpmb->dwStyle & MB_TASKMODAL) && (lpmb->hwndOwner==NULL)) || (lpmb->dwStyle & MB_SYSTEMMODAL))
285         SetWindowPos(hwnd, HWND_TOP, 0, 0, 0, 0, SWP_NOSIZE | SWP_NOMOVE);
286
287     return hFont;
288 }
289
290
291 /**************************************************************************
292  *           MSGBOX_DlgProc
293  *
294  * Dialog procedure for message boxes.
295  */
296 static INT_PTR CALLBACK MSGBOX_DlgProc( HWND hwnd, UINT message,
297                                         WPARAM wParam, LPARAM lParam )
298 {
299   HFONT hFont;
300
301   switch(message) {
302    case WM_INITDIALOG:
303    {
304        LPMSGBOXPARAMSW mbp = (LPMSGBOXPARAMSW)lParam;
305        SetWindowContextHelpId(hwnd, mbp->dwContextHelpId);
306        hFont = MSGBOX_OnInit(hwnd, mbp);
307        SetPropA(hwnd, "WINE_MSGBOX_HFONT", (HANDLE)hFont);
308        SetPropA(hwnd, "WINE_MSGBOX_HELPCALLBACK", (HANDLE)mbp->lpfnMsgBoxCallback);
309        break;
310    }
311
312    case WM_COMMAND:
313     switch (LOWORD(wParam))
314     {
315      case IDOK:
316      case IDCANCEL:
317      case IDABORT:
318      case IDRETRY:
319      case IDIGNORE:
320      case IDYES:
321      case IDNO:
322       hFont = GetPropA(hwnd, "WINE_MSGBOX_HFONT");
323       EndDialog(hwnd, wParam);
324       if (hFont)
325           DeleteObject(hFont);
326       break;
327     }
328     break;
329
330     case WM_HELP:
331     {
332         MSGBOXCALLBACK callback = (MSGBOXCALLBACK)GetPropA(hwnd, "WINE_MSGBOX_HELPCALLBACK");
333         HELPINFO hi;
334
335         memcpy(&hi, (void *)lParam, sizeof(hi));
336         hi.dwContextId = GetWindowContextHelpId(hwnd);
337
338         if (callback)
339             callback(&hi);
340         else
341             SendMessageW(GetWindow(hwnd, GW_OWNER), WM_HELP, 0, (LPARAM)&hi);
342         break;
343    }
344
345    default:
346      /* Ok. Ignore all the other messages */
347      TRACE("Message number 0x%04x is being ignored.\n", message);
348     break;
349   }
350   return 0;
351 }
352
353
354 /**************************************************************************
355  *              MessageBoxA (USER32.@)
356  */
357 INT WINAPI MessageBoxA(HWND hWnd, LPCSTR text, LPCSTR title, UINT type)
358 {
359     return MessageBoxExA(hWnd, text, title, type, LANG_NEUTRAL);
360 }
361
362
363 /**************************************************************************
364  *              MessageBoxW (USER32.@)
365  */
366 INT WINAPI MessageBoxW( HWND hwnd, LPCWSTR text, LPCWSTR title, UINT type )
367 {
368     return MessageBoxExW(hwnd, text, title, type, LANG_NEUTRAL);
369 }
370
371
372 /**************************************************************************
373  *              MessageBoxExA (USER32.@)
374  */
375 INT WINAPI MessageBoxExA( HWND hWnd, LPCSTR text, LPCSTR title,
376                               UINT type, WORD langid )
377 {
378     MSGBOXPARAMSA msgbox;
379
380     msgbox.cbSize = sizeof(msgbox);
381     msgbox.hwndOwner = hWnd;
382     msgbox.hInstance = 0;
383     msgbox.lpszText = text;
384     msgbox.lpszCaption = title;
385     msgbox.dwStyle = type;
386     msgbox.lpszIcon = NULL;
387     msgbox.dwContextHelpId = 0;
388     msgbox.lpfnMsgBoxCallback = NULL;
389     msgbox.dwLanguageId = langid;
390
391     return MessageBoxIndirectA(&msgbox);
392 }
393
394 /**************************************************************************
395  *              MessageBoxExW (USER32.@)
396  */
397 INT WINAPI MessageBoxExW( HWND hWnd, LPCWSTR text, LPCWSTR title,
398                               UINT type, WORD langid )
399 {
400     MSGBOXPARAMSW msgbox;
401
402     msgbox.cbSize = sizeof(msgbox);
403     msgbox.hwndOwner = hWnd;
404     msgbox.hInstance = 0;
405     msgbox.lpszText = text;
406     msgbox.lpszCaption = title;
407     msgbox.dwStyle = type;
408     msgbox.lpszIcon = NULL;
409     msgbox.dwContextHelpId = 0;
410     msgbox.lpfnMsgBoxCallback = NULL;
411     msgbox.dwLanguageId = langid;
412
413     return MessageBoxIndirectW(&msgbox);
414 }
415
416 /**************************************************************************
417  *              MessageBoxIndirectA (USER32.@)
418  */
419 INT WINAPI MessageBoxIndirectA( LPMSGBOXPARAMSA msgbox )
420 {
421     MSGBOXPARAMSW msgboxW;
422     UNICODE_STRING textW, captionW, iconW;
423     int ret;
424
425     if (HIWORD(msgbox->lpszText))
426         RtlCreateUnicodeStringFromAsciiz(&textW, msgbox->lpszText);
427     else
428         textW.Buffer = (LPWSTR)msgbox->lpszText;
429     if (HIWORD(msgbox->lpszCaption))
430         RtlCreateUnicodeStringFromAsciiz(&captionW, msgbox->lpszCaption);
431     else
432         captionW.Buffer = (LPWSTR)msgbox->lpszCaption;
433
434     if (msgbox->dwStyle & MB_USERICON)
435     {
436         if (HIWORD(msgbox->lpszIcon))
437             RtlCreateUnicodeStringFromAsciiz(&iconW, msgbox->lpszIcon);
438         else
439             iconW.Buffer = (LPWSTR)msgbox->lpszIcon;
440     }
441     else
442         iconW.Buffer = NULL;
443
444     msgboxW.cbSize = sizeof(msgboxW);
445     msgboxW.hwndOwner = msgbox->hwndOwner;
446     msgboxW.hInstance = msgbox->hInstance;
447     msgboxW.lpszText = textW.Buffer;
448     msgboxW.lpszCaption = captionW.Buffer;
449     msgboxW.dwStyle = msgbox->dwStyle;
450     msgboxW.lpszIcon = iconW.Buffer;
451     msgboxW.dwContextHelpId = msgbox->dwContextHelpId;
452     msgboxW.lpfnMsgBoxCallback = msgbox->lpfnMsgBoxCallback;
453     msgboxW.dwLanguageId = msgbox->dwLanguageId;
454
455     ret = MessageBoxIndirectW(&msgboxW);
456
457     if (HIWORD(textW.Buffer)) RtlFreeUnicodeString(&textW);
458     if (HIWORD(captionW.Buffer)) RtlFreeUnicodeString(&captionW);
459     if (HIWORD(iconW.Buffer)) RtlFreeUnicodeString(&iconW);
460     return ret;
461 }
462
463 /**************************************************************************
464  *              MessageBoxIndirectW (USER32.@)
465  */
466 INT WINAPI MessageBoxIndirectW( LPMSGBOXPARAMSW msgbox )
467 {
468     LPVOID tmplate;
469     HRSRC hRes;
470     int ret;
471     UINT i;
472     struct ThreadWindows threadWindows;
473     static const WCHAR msg_box_res_nameW[] = { 'M','S','G','B','O','X',0 };
474
475     if (!(hRes = FindResourceExW(user32_module, (LPWSTR)RT_DIALOG,
476                                  msg_box_res_nameW, msgbox->dwLanguageId)))
477         return 0;
478     if (!(tmplate = (LPVOID)LoadResource(user32_module, hRes)))
479         return 0;
480
481     if ((msgbox->dwStyle & MB_TASKMODAL) && (msgbox->hwndOwner==NULL))
482     {
483         threadWindows.numHandles = 0;
484         threadWindows.numAllocs = 10;
485         threadWindows.handles = HeapAlloc(GetProcessHeap(), 0, 10*sizeof(HWND));
486         EnumThreadWindows(GetCurrentThreadId(), MSGBOX_EnumProc, (LPARAM)&threadWindows);
487     }
488
489     ret=DialogBoxIndirectParamW(msgbox->hInstance, tmplate,
490                                 msgbox->hwndOwner, MSGBOX_DlgProc, (LPARAM)msgbox);
491
492     if ((msgbox->dwStyle & MB_TASKMODAL) && (msgbox->hwndOwner==NULL))
493     {
494         for (i = 0; i < threadWindows.numHandles; i++)
495             EnableWindow(threadWindows.handles[i], TRUE);
496         HeapFree(GetProcessHeap(), 0, threadWindows.handles);
497     }
498     return ret;
499 }