Fixed RtlSetEnvironmentVariable to deal properly with Unicode strings
[wine] / windows / defdlg.c
1 /*
2  * Default dialog procedure
3  *
4  * Copyright 1993, 1996 Alexandre Julliard
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20
21 #include <stdarg.h>
22
23 #include "windef.h"
24 #include "winbase.h"
25 #include "wingdi.h"
26 #include "wine/winuser16.h"
27 #include "controls.h"
28 #include "win.h"
29 #include "winproc.h"
30 #include "wine/debug.h"
31
32 WINE_DEFAULT_DEBUG_CHANNEL(dialog);
33
34
35 /***********************************************************************
36  *           DEFDLG_GetDlgProc
37  */
38 static WNDPROC DEFDLG_GetDlgProc( HWND hwnd )
39 {
40     WNDPROC ret;
41     WND *wndPtr = WIN_GetPtr( hwnd );
42
43     if (!wndPtr) return 0;
44     if (wndPtr == WND_OTHER_PROCESS)
45     {
46         ERR( "cannot get dlg proc %p from other process\n", hwnd );
47         return 0;
48     }
49     ret = *(WNDPROC *)((char *)wndPtr->wExtra + DWL_DLGPROC);
50     WIN_ReleasePtr( wndPtr );
51     return ret;
52 }
53
54 /***********************************************************************
55  *           DEFDLG_SetFocus
56  *
57  * Set the focus to a control of the dialog, selecting the text if
58  * the control is an edit dialog.
59  */
60 static void DEFDLG_SetFocus( HWND hwndDlg, HWND hwndCtrl )
61 {
62     HWND hwndPrev = GetFocus();
63
64     if (IsChild( hwndDlg, hwndPrev ))
65     {
66         if (SendMessageW( hwndPrev, WM_GETDLGCODE, 0, 0 ) & DLGC_HASSETSEL)
67             SendMessageW( hwndPrev, EM_SETSEL, -1, 0 );
68     }
69     if (SendMessageW( hwndCtrl, WM_GETDLGCODE, 0, 0 ) & DLGC_HASSETSEL)
70         SendMessageW( hwndCtrl, EM_SETSEL, 0, -1 );
71     SetFocus( hwndCtrl );
72 }
73
74
75 /***********************************************************************
76  *           DEFDLG_SaveFocus
77  */
78 static void DEFDLG_SaveFocus( HWND hwnd )
79 {
80     DIALOGINFO *infoPtr;
81     HWND hwndFocus = GetFocus();
82
83     if (!hwndFocus || !IsChild( hwnd, hwndFocus )) return;
84     if (!(infoPtr = DIALOG_get_info( hwnd ))) return;
85     infoPtr->hwndFocus = hwndFocus;
86     /* Remove default button */
87 }
88
89
90 /***********************************************************************
91  *           DEFDLG_RestoreFocus
92  */
93 static void DEFDLG_RestoreFocus( HWND hwnd )
94 {
95     DIALOGINFO *infoPtr;
96
97     if (IsIconic( hwnd )) return;
98     if (!(infoPtr = DIALOG_get_info( hwnd ))) return;
99     if (!IsWindow( infoPtr->hwndFocus )) return;
100     /* Don't set the focus back to controls if EndDialog is already called.*/
101     if (!(infoPtr->flags & DF_END))
102     {
103         DEFDLG_SetFocus( hwnd, infoPtr->hwndFocus );
104         return;
105     }
106     /* This used to set infoPtr->hwndFocus to NULL for no apparent reason,
107        sometimes losing focus when receiving WM_SETFOCUS messages. */
108 }
109
110
111 /***********************************************************************
112  *           DEFDLG_FindDefButton
113  *
114  * Find the current default push-button.
115  */
116 static HWND DEFDLG_FindDefButton( HWND hwndDlg )
117 {
118     HWND hwndChild = GetWindow( hwndDlg, GW_CHILD );
119     while (hwndChild)
120     {
121         if (SendMessageW( hwndChild, WM_GETDLGCODE, 0, 0 ) & DLGC_DEFPUSHBUTTON)
122             break;
123         hwndChild = GetWindow( hwndChild, GW_HWNDNEXT );
124     }
125     return hwndChild;
126 }
127
128
129 /***********************************************************************
130  *           DEFDLG_SetDefButton
131  *
132  * Set the new default button to be hwndNew.
133  */
134 static BOOL DEFDLG_SetDefButton( HWND hwndDlg, DIALOGINFO *dlgInfo, WPARAM wParam )
135 {
136     DWORD dlgcode=0; /* initialize just to avoid a warning */
137     HWND hwndNew = GetDlgItem(hwndDlg, wParam);
138
139     dlgInfo->idResult = wParam;
140     if (hwndNew &&
141         !((dlgcode=SendMessageW(hwndNew, WM_GETDLGCODE, 0, 0 ))
142             & (DLGC_UNDEFPUSHBUTTON | DLGC_BUTTON)))
143         return FALSE;  /* Destination is not a push button */
144
145     if (dlgInfo->idResult)  /* There's already a default pushbutton */
146     {
147         HWND hwndOld = GetDlgItem( hwndDlg, dlgInfo->idResult );
148         if (hwndOld && (SendMessageA( hwndOld, WM_GETDLGCODE, 0, 0) & DLGC_DEFPUSHBUTTON))
149             SendMessageA( hwndOld, BM_SETSTYLE, BS_PUSHBUTTON, TRUE );
150     }
151     if (hwndNew)
152     {
153         if(dlgcode==DLGC_UNDEFPUSHBUTTON)
154             SendMessageA( hwndNew, BM_SETSTYLE, BS_DEFPUSHBUTTON, TRUE );
155     }
156     return TRUE;
157 }
158
159
160 /***********************************************************************
161  *           DEFDLG_Proc
162  *
163  * Implementation of DefDlgProc(). Only handle messages that need special
164  * handling for dialogs.
165  */
166 static LRESULT DEFDLG_Proc( HWND hwnd, UINT msg, WPARAM wParam,
167                             LPARAM lParam, DIALOGINFO *dlgInfo )
168 {
169     switch(msg)
170     {
171         case WM_ERASEBKGND:
172         {
173             HBRUSH brush = (HBRUSH)SendMessageW( hwnd, WM_CTLCOLORDLG, wParam, (LPARAM)hwnd );
174             if (!brush) brush = (HBRUSH)DefWindowProcW( hwnd, WM_CTLCOLORDLG, wParam, (LPARAM)hwnd );
175             if (brush)
176             {
177                 RECT rect;
178                 HDC hdc = (HDC)wParam;
179                 GetClientRect( hwnd, &rect );
180                 DPtoLP( hdc, (LPPOINT)&rect, 2 );
181                 FillRect( hdc, &rect, brush );
182             }
183             return 1;
184         }
185         case WM_NCDESTROY:
186             if ((dlgInfo = (DIALOGINFO *)SetWindowLongW( hwnd, DWL_WINE_DIALOGINFO, 0 )))
187             {
188                 /* Free dialog heap (if created) */
189                 if (dlgInfo->hDialogHeap)
190                 {
191                     GlobalUnlock16(dlgInfo->hDialogHeap);
192                     GlobalFree16(dlgInfo->hDialogHeap);
193                 }
194                 if (dlgInfo->hUserFont) DeleteObject( dlgInfo->hUserFont );
195                 if (dlgInfo->hMenu) DestroyMenu( dlgInfo->hMenu );
196                 WINPROC_FreeProc( DEFDLG_GetDlgProc( hwnd ), WIN_PROC_WINDOW );
197                 HeapFree( GetProcessHeap(), 0, dlgInfo );
198             }
199               /* Window clean-up */
200             return DefWindowProcA( hwnd, msg, wParam, lParam );
201
202         case WM_SHOWWINDOW:
203             if (!wParam) DEFDLG_SaveFocus( hwnd );
204             return DefWindowProcA( hwnd, msg, wParam, lParam );
205
206         case WM_ACTIVATE:
207             if (wParam) DEFDLG_RestoreFocus( hwnd );
208             else DEFDLG_SaveFocus( hwnd );
209             return 0;
210
211         case WM_SETFOCUS:
212             DEFDLG_RestoreFocus( hwnd );
213             return 0;
214
215         case DM_SETDEFID:
216             if (dlgInfo && !(dlgInfo->flags & DF_END))
217                 DEFDLG_SetDefButton( hwnd, dlgInfo, wParam );
218             return 1;
219
220         case DM_GETDEFID:
221             if (dlgInfo && !(dlgInfo->flags & DF_END))
222             {
223                 HWND hwndDefId;
224                 if (dlgInfo->idResult)
225                     return MAKELONG( dlgInfo->idResult, DC_HASDEFID );
226                 if ((hwndDefId = DEFDLG_FindDefButton( hwnd )))
227                     return MAKELONG( GetDlgCtrlID( hwndDefId ), DC_HASDEFID);
228             }
229             return 0;
230
231         case WM_NEXTDLGCTL:
232             if (dlgInfo)
233             {
234                 HWND hwndDest = (HWND)wParam;
235                 if (!lParam)
236                     hwndDest = GetNextDlgTabItem(hwnd, GetFocus(), wParam);
237                 if (hwndDest) DEFDLG_SetFocus( hwnd, hwndDest );
238                 DEFDLG_SetDefButton( hwnd, dlgInfo, GetDlgCtrlID(hwndDest) );
239             }
240             return 0;
241
242         case WM_ENTERMENULOOP:
243         case WM_LBUTTONDOWN:
244         case WM_NCLBUTTONDOWN:
245             {
246                 HWND hwndFocus = GetFocus();
247                 if (hwndFocus)
248                 {
249                     /* always make combo box hide its listbox control */
250                     if (!SendMessageA( hwndFocus, CB_SHOWDROPDOWN, FALSE, 0 ))
251                         SendMessageA( GetParent(hwndFocus), CB_SHOWDROPDOWN, FALSE, 0 );
252                 }
253             }
254             return DefWindowProcA( hwnd, msg, wParam, lParam );
255
256         case WM_GETFONT:
257             return dlgInfo ? (LRESULT)dlgInfo->hUserFont : 0;
258
259         case WM_CLOSE:
260             PostMessageA( hwnd, WM_COMMAND, MAKEWPARAM(IDCANCEL, BN_CLICKED),
261                             (LPARAM)GetDlgItem( hwnd, IDCANCEL ) );
262             return 0;
263
264         case WM_NOTIFYFORMAT:
265             return DefWindowProcA( hwnd, msg, wParam, lParam );
266     }
267     return 0;
268 }
269
270 /***********************************************************************
271  *           DEFDLG_Epilog
272  */
273 static LRESULT DEFDLG_Epilog(HWND hwnd, UINT msg, BOOL fResult)
274 {
275     /* see SDK 3.1 */
276
277     if ((msg >= WM_CTLCOLORMSGBOX && msg <= WM_CTLCOLORSTATIC) ||
278          msg == WM_CTLCOLOR || msg == WM_COMPAREITEM ||
279          msg == WM_VKEYTOITEM || msg == WM_CHARTOITEM ||
280          msg == WM_QUERYDRAGICON || msg == WM_INITDIALOG)
281         return fResult;
282
283     return GetWindowLongA( hwnd, DWL_MSGRESULT );
284 }
285
286 /***********************************************************************
287 *               DEFDLG_InitDlgInfo
288 *
289 * Allocate memory for DIALOGINFO structure and store in DWL_DIALOGINFO
290 * structure. Also flag the window as a dialog type.
291 */
292 static DIALOGINFO* DEFDLG_InitDlgInfo(HWND hwnd)
293 {
294     WND* wndPtr;
295     DIALOGINFO* dlgInfo = DIALOG_get_info( hwnd );
296     if(!dlgInfo)
297     {
298         if (!(dlgInfo = HeapAlloc( GetProcessHeap(), 0, sizeof(*dlgInfo) ))) return NULL;
299         dlgInfo->hwndFocus   = 0;
300         dlgInfo->hUserFont   = 0;
301         dlgInfo->hMenu       = 0;
302         dlgInfo->xBaseUnit   = 0;
303         dlgInfo->yBaseUnit   = 0;
304         dlgInfo->idResult    = 0;
305         dlgInfo->flags       = 0;
306         dlgInfo->hDialogHeap = 0;
307         wndPtr = WIN_GetPtr( hwnd );
308         if (wndPtr && wndPtr != WND_OTHER_PROCESS)
309         {
310             wndPtr->flags |= WIN_ISDIALOG;
311             WIN_ReleasePtr( wndPtr );
312             SetWindowLongW( hwnd, DWL_WINE_DIALOGINFO, (LONG)dlgInfo );
313         }
314         else
315         {
316             HeapFree( GetProcessHeap(), 0, dlgInfo );
317             return NULL;
318         }
319     }
320     return dlgInfo;
321 }
322
323 /***********************************************************************
324  *              DefDlgProc (USER.308)
325  */
326 LRESULT WINAPI DefDlgProc16( HWND16 hwnd, UINT16 msg, WPARAM16 wParam,
327                              LPARAM lParam )
328 {
329     DIALOGINFO *dlgInfo;
330     WNDPROC16 dlgproc;
331     HWND hwnd32 = WIN_Handle32( hwnd );
332     BOOL result = FALSE;
333
334     /* Perform DIALOGINFO intialization if not done */
335     if(!(dlgInfo = DEFDLG_InitDlgInfo(hwnd32))) return -1;
336
337     SetWindowLongW( hwnd32, DWL_MSGRESULT, 0 );
338
339     if ((dlgproc = (WNDPROC16)DEFDLG_GetDlgProc( hwnd32 )))
340     {
341         /* Call dialog procedure */
342         result = CallWindowProc16( dlgproc, hwnd, msg, wParam, lParam );
343         /* 16 bit dlg procs only return BOOL16 */
344         if( WINPROC_GetProcType( (WNDPROC)dlgproc ) == WIN_PROC_16 )
345             result = LOWORD(result);
346     }
347
348     if (!result && IsWindow(hwnd32))
349     {
350         /* callback didn't process this message */
351
352         switch(msg)
353         {
354             case WM_ERASEBKGND:
355             case WM_SHOWWINDOW:
356             case WM_ACTIVATE:
357             case WM_SETFOCUS:
358             case DM_SETDEFID:
359             case DM_GETDEFID:
360             case WM_NEXTDLGCTL:
361             case WM_GETFONT:
362             case WM_CLOSE:
363             case WM_NCDESTROY:
364             case WM_ENTERMENULOOP:
365             case WM_LBUTTONDOWN:
366             case WM_NCLBUTTONDOWN:
367                 return DEFDLG_Proc( hwnd32, msg, (WPARAM)wParam, lParam, dlgInfo );
368             case WM_INITDIALOG:
369             case WM_VKEYTOITEM:
370             case WM_COMPAREITEM:
371             case WM_CHARTOITEM:
372                 break;
373
374             default:
375                 return DefWindowProc16( hwnd, msg, wParam, lParam );
376         }
377     }
378     return DEFDLG_Epilog( hwnd32, msg, result);
379 }
380
381
382 /***********************************************************************
383  *              DefDlgProcA (USER32.@)
384  */
385 LRESULT WINAPI DefDlgProcA( HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam )
386 {
387     DIALOGINFO *dlgInfo;
388     WNDPROC dlgproc;
389     BOOL result = FALSE;
390
391     /* Perform DIALOGINFO initialization if not done */
392     if(!(dlgInfo = DEFDLG_InitDlgInfo(hwnd))) return -1;
393
394     SetWindowLongW( hwnd, DWL_MSGRESULT, 0 );
395
396     if ((dlgproc = DEFDLG_GetDlgProc( hwnd )))
397     {
398         /* Call dialog procedure */
399         result = CallWindowProcA( dlgproc, hwnd, msg, wParam, lParam );
400         /* 16 bit dlg procs only return BOOL16 */
401         if( WINPROC_GetProcType( dlgproc ) == WIN_PROC_16 )
402             result = LOWORD(result);
403     }
404
405     if (!result && IsWindow(hwnd))
406     {
407         /* callback didn't process this message */
408
409         switch(msg)
410         {
411             case WM_ERASEBKGND:
412             case WM_SHOWWINDOW:
413             case WM_ACTIVATE:
414             case WM_SETFOCUS:
415             case DM_SETDEFID:
416             case DM_GETDEFID:
417             case WM_NEXTDLGCTL:
418             case WM_GETFONT:
419             case WM_CLOSE:
420             case WM_NCDESTROY:
421             case WM_ENTERMENULOOP:
422             case WM_LBUTTONDOWN:
423             case WM_NCLBUTTONDOWN:
424                  return DEFDLG_Proc( hwnd, msg, wParam, lParam, dlgInfo );
425             case WM_INITDIALOG:
426             case WM_VKEYTOITEM:
427             case WM_COMPAREITEM:
428             case WM_CHARTOITEM:
429                  break;
430
431             default:
432                  return DefWindowProcA( hwnd, msg, wParam, lParam );
433         }
434     }
435     return DEFDLG_Epilog(hwnd, msg, result);
436 }
437
438
439 /***********************************************************************
440  *              DefDlgProcW (USER32.@)
441  */
442 LRESULT WINAPI DefDlgProcW( HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam )
443 {
444     DIALOGINFO *dlgInfo;
445     BOOL result = FALSE;
446     WNDPROC dlgproc;
447
448     /* Perform DIALOGINFO intialization if not done */
449     if(!(dlgInfo = DEFDLG_InitDlgInfo(hwnd))) return -1;
450
451     SetWindowLongW( hwnd, DWL_MSGRESULT, 0 );
452
453     if ((dlgproc = DEFDLG_GetDlgProc( hwnd )))
454     {
455         /* Call dialog procedure */
456         result = CallWindowProcW( dlgproc, hwnd, msg, wParam, lParam );
457         /* 16 bit dlg procs only return BOOL16 */
458         if( WINPROC_GetProcType( dlgproc ) == WIN_PROC_16 )
459             result = LOWORD(result);
460     }
461
462     if (!result && IsWindow(hwnd))
463     {
464         /* callback didn't process this message */
465
466         switch(msg)
467         {
468             case WM_ERASEBKGND:
469             case WM_SHOWWINDOW:
470             case WM_ACTIVATE:
471             case WM_SETFOCUS:
472             case DM_SETDEFID:
473             case DM_GETDEFID:
474             case WM_NEXTDLGCTL:
475             case WM_GETFONT:
476             case WM_CLOSE:
477             case WM_NCDESTROY:
478             case WM_ENTERMENULOOP:
479             case WM_LBUTTONDOWN:
480             case WM_NCLBUTTONDOWN:
481                  return DEFDLG_Proc( hwnd, msg, wParam, lParam, dlgInfo );
482             case WM_INITDIALOG:
483             case WM_VKEYTOITEM:
484             case WM_COMPAREITEM:
485             case WM_CHARTOITEM:
486                  break;
487
488             default:
489                  return DefWindowProcW( hwnd, msg, wParam, lParam );
490         }
491     }
492     return DEFDLG_Epilog(hwnd, msg, result);
493 }