msi: Prevent array underflow in MsiFormat when measuring with zero-length buffer.
[wine] / dlls / msi / dialog.c
1 /*
2  * Implementation of the Microsoft Installer (msi.dll)
3  *
4  * Copyright 2005 Mike McCormack for CodeWeavers
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 #define COBJMACROS
22 #define NONAMELESSUNION
23 #define NONAMELESSSTRUCT
24
25 #include <stdarg.h>
26
27 #include "windef.h"
28 #include "winbase.h"
29 #include "winuser.h"
30 #include "winnls.h"
31 #include "wingdi.h"
32 #include "msi.h"
33 #include "msipriv.h"
34 #include "msidefs.h"
35 #include "ocidl.h"
36 #include "olectl.h"
37 #include "richedit.h"
38 #include "commctrl.h"
39
40 #include "wine/debug.h"
41 #include "wine/unicode.h"
42
43 #include "action.h"
44
45 WINE_DEFAULT_DEBUG_CHANNEL(msi);
46
47
48 struct msi_control_tag;
49 typedef struct msi_control_tag msi_control;
50 typedef UINT (*msi_handler)( msi_dialog *, msi_control *, WPARAM );
51
52 struct msi_control_tag
53 {
54     struct list entry;
55     HWND hwnd;
56     msi_handler handler;
57     LPWSTR property;
58     LPWSTR value;
59     HBITMAP hBitmap;
60     HICON hIcon;
61     LPWSTR tabnext;
62     HMODULE hDll;
63     WCHAR name[1];
64 };
65
66 typedef struct msi_font_tag
67 {
68     struct msi_font_tag *next;
69     HFONT hfont;
70     WCHAR name[1];
71 } msi_font;
72
73 struct msi_dialog_tag
74 {
75     MSIPACKAGE *package;
76     msi_dialog_event_handler event_handler;
77     BOOL finished;
78     INT scale;
79     DWORD attributes;
80     HWND hwnd;
81     LPWSTR default_font;
82     msi_font *font_list;
83     struct list controls;
84     HWND hWndFocus;
85     LPWSTR control_default;
86     LPWSTR control_cancel;
87     WCHAR name[1];
88 };
89
90 typedef UINT (*msi_dialog_control_func)( msi_dialog *dialog, MSIRECORD *rec );
91 struct control_handler 
92 {
93     LPCWSTR control_type;
94     msi_dialog_control_func func;
95 };
96
97 typedef struct
98 {
99     msi_dialog* dialog;
100     msi_control *parent;
101     DWORD       attributes;
102 } radio_button_group_descr;
103
104 const WCHAR szMsiDialogClass[] = {
105     'M','s','i','D','i','a','l','o','g','C','l','o','s','e','C','l','a','s','s',0
106 };
107 const WCHAR szMsiHiddenWindow[] = {
108     'M','s','i','H','i','d','d','e','n','W','i','n','d','o','w',0 };
109 static const WCHAR szStatic[] = { 'S','t','a','t','i','c',0 };
110 static const WCHAR szButton[] = { 'B','U','T','T','O','N', 0 };
111 static const WCHAR szButtonData[] = { 'M','S','I','D','A','T','A',0 };
112 static const WCHAR szText[] = { 'T','e','x','t',0 };
113 static const WCHAR szPushButton[] = { 'P','u','s','h','B','u','t','t','o','n',0 };
114 static const WCHAR szLine[] = { 'L','i','n','e',0 };
115 static const WCHAR szBitmap[] = { 'B','i','t','m','a','p',0 };
116 static const WCHAR szCheckBox[] = { 'C','h','e','c','k','B','o','x',0 };
117 static const WCHAR szScrollableText[] = {
118     'S','c','r','o','l','l','a','b','l','e','T','e','x','t',0 };
119 static const WCHAR szComboBox[] = { 'C','o','m','b','o','B','o','x',0 };
120 static const WCHAR szEdit[] = { 'E','d','i','t',0 };
121 static const WCHAR szMaskedEdit[] = { 'M','a','s','k','e','d','E','d','i','t',0 };
122 static const WCHAR szPathEdit[] = { 'P','a','t','h','E','d','i','t',0 };
123 static const WCHAR szProgressBar[] = {
124      'P','r','o','g','r','e','s','s','B','a','r',0 };
125 static const WCHAR szRadioButtonGroup[] = { 
126     'R','a','d','i','o','B','u','t','t','o','n','G','r','o','u','p',0 };
127 static const WCHAR szIcon[] = { 'I','c','o','n',0 };
128 static const WCHAR szSelectionTree[] = {
129     'S','e','l','e','c','t','i','o','n','T','r','e','e',0 };
130
131 static UINT msi_dialog_checkbox_handler( msi_dialog *, msi_control *, WPARAM );
132 static void msi_dialog_checkbox_sync_state( msi_dialog *, msi_control * );
133 static UINT msi_dialog_button_handler( msi_dialog *, msi_control *, WPARAM );
134 static UINT msi_dialog_edit_handler( msi_dialog *, msi_control *, WPARAM );
135 static UINT msi_dialog_radiogroup_handler( msi_dialog *, msi_control *, WPARAM param );
136 static UINT msi_dialog_evaluate_control_conditions( msi_dialog *dialog );
137 static LRESULT WINAPI MSIRadioGroup_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam);
138
139
140 /* dialog sequencing */
141
142 #define WM_MSI_DIALOG_CREATE  (WM_USER+0x100)
143 #define WM_MSI_DIALOG_DESTROY (WM_USER+0x101)
144
145 static DWORD uiThreadId;
146 static HWND hMsiHiddenWindow;
147
148 static INT msi_dialog_scale_unit( msi_dialog *dialog, INT val )
149 {
150     return (dialog->scale * val + 5) / 10;
151 }
152
153 static msi_control *msi_dialog_find_control( msi_dialog *dialog, LPCWSTR name )
154 {
155     msi_control *control;
156
157     if( !name )
158         return NULL;
159     LIST_FOR_EACH_ENTRY( control, &dialog->controls, msi_control, entry )
160         if( !strcmpW( control->name, name ) ) /* FIXME: case sensitive? */
161             return control;
162     return NULL;
163 }
164
165 static msi_control *msi_dialog_find_control_by_hwnd( msi_dialog *dialog, HWND hwnd )
166 {
167     msi_control *control;
168
169     LIST_FOR_EACH_ENTRY( control, &dialog->controls, msi_control, entry )
170         if( hwnd == control->hwnd )
171             return control;
172     return NULL;
173 }
174
175 static LPWSTR msi_get_deformatted_field( MSIPACKAGE *package, MSIRECORD *rec, int field )
176 {
177     LPCWSTR str = MSI_RecordGetString( rec, field );
178     LPWSTR ret = NULL;
179
180     if (str)
181         deformat_string( package, str, &ret );
182     return ret;
183 }
184
185 /*
186  * msi_dialog_get_style
187  *
188  * Extract the {\style} string from the front of the text to display and
189  *  update the pointer.
190  */
191 static LPWSTR msi_dialog_get_style( LPCWSTR p, LPCWSTR *rest )
192 {
193     LPWSTR ret = NULL;
194     LPCWSTR q, i;
195     DWORD len;
196
197     *rest = p;
198     if( !p )
199         return ret;
200     if( *p++ != '{' )
201         return ret;
202     q = strchrW( p, '}' );
203     if( !q )
204         return ret;
205     if( *p == '\\' || *p == '&' )
206         p++;
207
208     /* little bit of sanity checking to stop us getting confused with RTF */
209     for( i=p; i<q; i++ )
210         if( *i == '}' || *i == '\\' )
211             return ret;
212     
213     *rest = ++q;
214     len = q - p;
215
216     ret = msi_alloc( len*sizeof(WCHAR) );
217     if( !ret )
218         return ret;
219     memcpy( ret, p, len*sizeof(WCHAR) );
220     ret[len-1] = 0;
221     return ret;
222 }
223
224 static UINT msi_dialog_add_font( MSIRECORD *rec, LPVOID param )
225 {
226     msi_dialog *dialog = param;
227     msi_font *font;
228     LPCWSTR face, name;
229     LOGFONTW lf;
230     INT style;
231     HDC hdc;
232
233     /* create a font and add it to the list */
234     name = MSI_RecordGetString( rec, 1 );
235     font = msi_alloc( sizeof *font + strlenW( name )*sizeof (WCHAR) );
236     strcpyW( font->name, name );
237     font->next = dialog->font_list;
238     dialog->font_list = font;
239
240     memset( &lf, 0, sizeof lf );
241     face = MSI_RecordGetString( rec, 2 );
242     lf.lfHeight = MSI_RecordGetInteger( rec, 3 );
243     style = MSI_RecordGetInteger( rec, 5 );
244     if( style & msidbTextStyleStyleBitsBold )
245         lf.lfWeight = FW_BOLD;
246     if( style & msidbTextStyleStyleBitsItalic )
247         lf.lfItalic = TRUE;
248     if( style & msidbTextStyleStyleBitsUnderline )
249         lf.lfUnderline = TRUE;
250     if( style & msidbTextStyleStyleBitsStrike )
251         lf.lfStrikeOut = TRUE;
252     lstrcpynW( lf.lfFaceName, face, LF_FACESIZE );
253
254     /* adjust the height */
255     hdc = GetDC( dialog->hwnd );
256     if (hdc)
257     {
258         lf.lfHeight = -MulDiv(lf.lfHeight, GetDeviceCaps(hdc, LOGPIXELSY), 72);
259         ReleaseDC( dialog->hwnd, hdc );
260     }
261
262     font->hfont = CreateFontIndirectW( &lf );
263
264     TRACE("Adding font style %s\n", debugstr_w(font->name) );
265
266     return ERROR_SUCCESS;
267 }
268
269 static msi_font *msi_dialog_find_font( msi_dialog *dialog, LPCWSTR name )
270 {
271     msi_font *font;
272
273     for( font = dialog->font_list; font; font = font->next )
274         if( !strcmpW( font->name, name ) )  /* FIXME: case sensitive? */
275             break;
276
277     return font;
278 }
279
280 static UINT msi_dialog_set_font( msi_dialog *dialog, HWND hwnd, LPCWSTR name )
281 {
282     msi_font *font;
283
284     font = msi_dialog_find_font( dialog, name );
285     if( font )
286         SendMessageW( hwnd, WM_SETFONT, (WPARAM) font->hfont, TRUE );
287     else
288         ERR("No font entry for %s\n", debugstr_w(name));
289     return ERROR_SUCCESS;
290 }
291
292 static UINT msi_dialog_build_font_list( msi_dialog *dialog )
293 {
294     static const WCHAR query[] = {
295       'S','E','L','E','C','T',' ','*',' ',
296       'F','R','O','M',' ','`','T','e','x','t','S','t','y','l','e','`',' ',0
297     };
298     UINT r;
299     MSIQUERY *view = NULL;
300
301     TRACE("dialog %p\n", dialog );
302
303     r = MSI_OpenQuery( dialog->package->db, &view, query );
304     if( r != ERROR_SUCCESS )
305         return r;
306
307     r = MSI_IterateRecords( view, NULL, msi_dialog_add_font, dialog );
308     msiobj_release( &view->hdr );
309
310     return r;
311 }
312
313 static msi_control *msi_dialog_create_window( msi_dialog *dialog,
314                 MSIRECORD *rec, LPCWSTR szCls, LPCWSTR name, LPCWSTR text,
315                 DWORD style, HWND parent )
316 {
317     DWORD x, y, width, height;
318     LPWSTR font = NULL, title_font = NULL;
319     LPCWSTR title = NULL;
320     msi_control *control;
321
322     style |= WS_CHILD;
323
324     control = msi_alloc( sizeof *control + strlenW(name)*sizeof(WCHAR) );
325     strcpyW( control->name, name );
326     list_add_head( &dialog->controls, &control->entry );
327     control->handler = NULL;
328     control->property = NULL;
329     control->value = NULL;
330     control->hBitmap = NULL;
331     control->hIcon = NULL;
332     control->hDll = NULL;
333     control->tabnext = strdupW( MSI_RecordGetString( rec, 11) );
334
335     x = MSI_RecordGetInteger( rec, 4 );
336     y = MSI_RecordGetInteger( rec, 5 );
337     width = MSI_RecordGetInteger( rec, 6 );
338     height = MSI_RecordGetInteger( rec, 7 );
339
340     x = msi_dialog_scale_unit( dialog, x );
341     y = msi_dialog_scale_unit( dialog, y );
342     width = msi_dialog_scale_unit( dialog, width );
343     height = msi_dialog_scale_unit( dialog, height );
344
345     if( text )
346     {
347         deformat_string( dialog->package, text, &title_font );
348         font = msi_dialog_get_style( title_font, &title );
349     }
350
351     control->hwnd = CreateWindowW( szCls, title, style,
352                           x, y, width, height, parent, NULL, NULL, NULL );
353
354     TRACE("Dialog %s control %s hwnd %p\n",
355            debugstr_w(dialog->name), debugstr_w(text), control->hwnd );
356
357     msi_dialog_set_font( dialog, control->hwnd,
358                          font ? font : dialog->default_font );
359
360     msi_free( title_font );
361     msi_free( font );
362
363     return control;
364 }
365
366 static MSIRECORD *msi_get_binary_record( MSIDATABASE *db, LPCWSTR name )
367 {
368     static const WCHAR query[] = {
369         's','e','l','e','c','t',' ','*',' ',
370         'f','r','o','m',' ','B','i','n','a','r','y',' ',
371         'w','h','e','r','e',' ',
372             '`','N','a','m','e','`',' ','=',' ','\'','%','s','\'',0
373     };
374
375     return MSI_QueryGetRecord( db, query, name );
376 }
377
378 static LPWSTR msi_create_tmp_path(void)
379 {
380     WCHAR tmp[MAX_PATH];
381     LPWSTR path = NULL;
382     static const WCHAR prefix[] = { 'm','s','i',0 };
383     DWORD len, r;
384
385     r = GetTempPathW( MAX_PATH, tmp );
386     if( !r )
387         return path;
388     len = lstrlenW( tmp ) + 20;
389     path = msi_alloc( len * sizeof (WCHAR) );
390     if( path )
391     {
392         r = GetTempFileNameW( tmp, prefix, 0, path );
393         if (!r)
394         {
395             msi_free( path );
396             path = NULL;
397         }
398     }
399     return path;
400 }
401
402
403 static HANDLE msi_load_image( MSIDATABASE *db, LPCWSTR name, UINT type,
404                               UINT cx, UINT cy, UINT flags )
405 {
406     MSIRECORD *rec = NULL;
407     HANDLE himage = NULL;
408     LPWSTR tmp;
409     UINT r;
410
411     TRACE("%p %s %u %u %08x\n", db, debugstr_w(name), cx, cy, flags);
412
413     tmp = msi_create_tmp_path();
414     if( !tmp )
415         return himage;
416
417     rec = msi_get_binary_record( db, name );
418     if( rec )
419     {
420         r = MSI_RecordStreamToFile( rec, 2, tmp );
421         if( r == ERROR_SUCCESS )
422         {
423             himage = LoadImageW( 0, tmp, type, cx, cy, flags );
424             DeleteFileW( tmp );
425         }
426         msiobj_release( &rec->hdr );
427     }
428
429     msi_free( tmp );
430     return himage;
431 }
432
433 static HICON msi_load_icon( MSIDATABASE *db, LPCWSTR text, UINT attributes )
434 {
435     DWORD cx = 0, cy = 0, flags;
436
437     flags = LR_LOADFROMFILE | LR_DEFAULTSIZE;
438     if( attributes & msidbControlAttributesFixedSize )
439     {
440         flags &= ~LR_DEFAULTSIZE;
441         if( attributes & msidbControlAttributesIconSize16 )
442         {
443             cx += 16;
444             cy += 16;
445         }
446         if( attributes & msidbControlAttributesIconSize32 )
447         {
448             cx += 32;
449             cy += 32;
450         }
451         /* msidbControlAttributesIconSize48 handled by above logic */
452     }
453     return msi_load_image( db, text, IMAGE_ICON, cx, cy, flags );
454 }
455
456
457 /* called from the Control Event subscription code */
458 void msi_dialog_handle_event( msi_dialog* dialog, LPCWSTR control, 
459                               LPCWSTR attribute, MSIRECORD *rec )
460 {
461     msi_control* ctrl;
462     LPCWSTR text;
463
464     ctrl = msi_dialog_find_control( dialog, control );
465     if (!ctrl)
466         return;
467     if( lstrcmpW(attribute, szText) )
468     {
469         ERR("Attribute %s\n", debugstr_w(attribute));
470         return;
471     }
472     text = MSI_RecordGetString( rec , 1 );
473     SetWindowTextW( ctrl->hwnd, text );
474     msi_dialog_check_messages( NULL );
475 }
476
477 static void msi_dialog_map_events(msi_dialog* dialog, LPCWSTR control)
478 {
479     static WCHAR Query[] = {
480         'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
481          '`','E','v','e','n','t','M','a','p','p','i','n','g','`',' ',
482         'W','H','E','R','E',' ',
483          '`','D','i','a','l','o','g','_','`',' ','=',' ','\'','%','s','\'',' ',
484         'A','N','D',' ',
485          '`','C','o','n','t','r','o','l','_','`',' ','=',' ','\'','%','s','\'',0
486     };
487     MSIRECORD *row;
488     LPCWSTR event, attribute;
489
490     row = MSI_QueryGetRecord( dialog->package->db, Query, dialog->name, control );
491     if (!row)
492         return;
493
494     event = MSI_RecordGetString( row, 3 );
495     attribute = MSI_RecordGetString( row, 4 );
496     ControlEvent_SubscribeToEvent( dialog->package, event, control, attribute );
497     msiobj_release( &row->hdr );
498 }
499
500 /* everything except radio buttons */
501 static msi_control *msi_dialog_add_control( msi_dialog *dialog,
502                 MSIRECORD *rec, LPCWSTR szCls, DWORD style )
503 {
504     DWORD attributes;
505     LPCWSTR text, name;
506
507     name = MSI_RecordGetString( rec, 2 );
508     attributes = MSI_RecordGetInteger( rec, 8 );
509     text = MSI_RecordGetString( rec, 10 );
510     if( attributes & msidbControlAttributesVisible )
511         style |= WS_VISIBLE;
512     if( ~attributes & msidbControlAttributesEnabled )
513         style |= WS_DISABLED;
514
515     msi_dialog_map_events(dialog, name);
516
517     return msi_dialog_create_window( dialog, rec, szCls, name, text,
518                                      style, dialog->hwnd );
519 }
520
521 struct msi_text_info
522 {
523     WNDPROC oldproc;
524     DWORD attributes;
525 };
526
527 /*
528  * we don't erase our own background,
529  * so we have to make sure that the parent window redraws first
530  */
531 static void msi_text_on_settext( HWND hWnd )
532 {
533     HWND hParent;
534     RECT rc;
535
536     hParent = GetParent( hWnd );
537     GetWindowRect( hWnd, &rc );
538     MapWindowPoints( NULL, hParent, (LPPOINT) &rc, 2 );
539     InvalidateRect( hParent, &rc, TRUE );
540 }
541
542 static LRESULT WINAPI
543 MSIText_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
544 {
545     struct msi_text_info *info;
546     LRESULT r = 0;
547
548     TRACE("%p %04x %08x %08lx\n", hWnd, msg, wParam, lParam);
549
550     info = GetPropW(hWnd, szButtonData);
551
552     if( msg == WM_CTLCOLORSTATIC &&
553        ( info->attributes & msidbControlAttributesTransparent ) )
554     {
555         SetBkMode( (HDC)wParam, TRANSPARENT );
556         return (LRESULT) GetStockObject(NULL_BRUSH);
557     }
558
559     r = CallWindowProcW(info->oldproc, hWnd, msg, wParam, lParam);
560
561     switch( msg )
562     {
563     case WM_SETTEXT:
564         msi_text_on_settext( hWnd );
565         break;
566     case WM_NCDESTROY:
567         msi_free( info );
568         RemovePropW( hWnd, szButtonData );
569         break;
570     }
571
572     return r;
573 }
574
575 static UINT msi_dialog_text_control( msi_dialog *dialog, MSIRECORD *rec )
576 {
577     msi_control *control;
578     struct msi_text_info *info;
579
580     TRACE("%p %p\n", dialog, rec);
581
582     control = msi_dialog_add_control( dialog, rec, szStatic, SS_LEFT | WS_GROUP );
583     if( !control )
584         return ERROR_FUNCTION_FAILED;
585
586     info = msi_alloc( sizeof *info );
587     if( !info )
588         return ERROR_SUCCESS;
589
590     info->attributes = MSI_RecordGetInteger( rec, 8 );
591     if( info->attributes & msidbControlAttributesTransparent )
592         SetWindowLongPtrW( control->hwnd, GWL_EXSTYLE, WS_EX_TRANSPARENT );
593
594     info->oldproc = (WNDPROC) SetWindowLongPtrW( control->hwnd, GWLP_WNDPROC,
595                                           (LONG_PTR)MSIText_WndProc );
596     SetPropW( control->hwnd, szButtonData, info );
597
598     return ERROR_SUCCESS;
599 }
600
601 static UINT msi_dialog_button_control( msi_dialog *dialog, MSIRECORD *rec )
602 {
603     msi_control *control;
604     UINT attributes, style;
605     LPWSTR text;
606
607     TRACE("%p %p\n", dialog, rec);
608
609     style = WS_TABSTOP;
610     attributes = MSI_RecordGetInteger( rec, 8 );
611     if( attributes & msidbControlAttributesIcon )
612         style |= BS_ICON;
613
614     control = msi_dialog_add_control( dialog, rec, szButton, style );
615     if( !control )
616         return ERROR_FUNCTION_FAILED;
617
618     control->handler = msi_dialog_button_handler;
619
620     /* set the icon */
621     text = msi_get_deformatted_field( dialog->package, rec, 10 );
622     control->hIcon = msi_load_icon( dialog->package->db, text, attributes );
623     if( attributes & msidbControlAttributesIcon )
624         SendMessageW( control->hwnd, BM_SETIMAGE, IMAGE_ICON, (LPARAM) control->hIcon );
625     msi_free( text );
626
627     return ERROR_SUCCESS;
628 }
629
630 static LPWSTR msi_get_checkbox_value( msi_dialog *dialog, LPCWSTR prop )
631 {
632     static const WCHAR query[] = {
633         'S','E','L','E','C','T',' ','*',' ',
634         'F','R','O','M',' ','`','C','h','e','c','k','B','o','x',' ','`',
635         'W','H','E','R','E',' ',
636         '`','P','r','o','p','e','r','t','y','`',' ','=',' ',
637         '\'','%','s','\'',0
638     };
639     MSIRECORD *rec = NULL;
640     LPWSTR ret = NULL;
641
642     /* find if there is a value associated with the checkbox */
643     rec = MSI_QueryGetRecord( dialog->package->db, query, prop );
644     if (!rec)
645         return ret;
646
647     ret = msi_get_deformatted_field( dialog->package, rec, 2 );
648     if( ret && !ret[0] )
649     {
650         msi_free( ret );
651         ret = NULL;
652     }
653     msiobj_release( &rec->hdr );
654     if (ret)
655         return ret;
656
657     ret = msi_dup_property( dialog->package, prop );
658     if( ret && !ret[0] )
659     {
660         msi_free( ret );
661         ret = NULL;
662     }
663
664     return ret;
665 }
666
667 static UINT msi_dialog_checkbox_control( msi_dialog *dialog, MSIRECORD *rec )
668 {
669     msi_control *control;
670     LPCWSTR prop;
671
672     TRACE("%p %p\n", dialog, rec);
673
674     control = msi_dialog_add_control( dialog, rec, szButton,
675                                 BS_CHECKBOX | BS_MULTILINE | WS_TABSTOP );
676     control->handler = msi_dialog_checkbox_handler;
677     prop = MSI_RecordGetString( rec, 9 );
678     if( prop )
679     {
680         control->property = strdupW( prop );
681         control->value = msi_get_checkbox_value( dialog, prop );
682         TRACE("control %s value %s\n", debugstr_w(control->property),
683               debugstr_w(control->value));
684     }
685     msi_dialog_checkbox_sync_state( dialog, control );
686
687     return ERROR_SUCCESS;
688 }
689
690 static UINT msi_dialog_line_control( msi_dialog *dialog, MSIRECORD *rec )
691 {
692     TRACE("%p %p\n", dialog, rec);
693
694     msi_dialog_add_control( dialog, rec, szStatic, SS_ETCHEDHORZ | SS_SUNKEN );
695     return ERROR_SUCCESS;
696 }
697
698 struct msi_streamin_info
699 {
700     LPSTR string;
701     DWORD offset;
702     DWORD length;
703 };
704
705 static DWORD CALLBACK
706 msi_richedit_stream_in( DWORD_PTR arg, LPBYTE buffer, LONG count, LONG *pcb )
707 {
708     struct msi_streamin_info *info = (struct msi_streamin_info*) arg;
709
710     if( (count + info->offset) > info->length )
711         count = info->length - info->offset;
712     memcpy( buffer, &info->string[ info->offset ], count );
713     *pcb = count;
714     info->offset += count;
715
716     TRACE("%ld/%ld\n", info->offset, info->length);
717
718     return 0;
719 }
720
721 static UINT msi_dialog_scrolltext_control( msi_dialog *dialog, MSIRECORD *rec )
722 {
723     static const WCHAR szRichEdit20W[] = {
724         'R','i','c','h','E','d','i','t','2','0','W',0
725     };
726     struct msi_streamin_info info;
727     msi_control *control;
728     LPCWSTR text;
729     EDITSTREAM es;
730     DWORD style;
731     HMODULE hRichedit;
732
733     hRichedit = LoadLibraryA("riched20");
734
735     style = WS_BORDER | ES_MULTILINE | WS_VSCROLL |
736             ES_READONLY | ES_AUTOVSCROLL | WS_TABSTOP;
737     control = msi_dialog_add_control( dialog, rec, szRichEdit20W, style );
738     if (!control)
739         return ERROR_FUNCTION_FAILED;
740
741     control->hDll = hRichedit;
742
743     text = MSI_RecordGetString( rec, 10 );
744     info.string = strdupWtoA( text );
745     info.offset = 0;
746     info.length = lstrlenA( info.string ) + 1;
747
748     es.dwCookie = (DWORD_PTR) &info;
749     es.dwError = 0;
750     es.pfnCallback = msi_richedit_stream_in;
751
752     SendMessageW( control->hwnd, EM_STREAMIN, SF_RTF, (LPARAM) &es );
753
754     msi_free( info.string );
755
756     return ERROR_SUCCESS;
757 }
758
759 static HBITMAP msi_load_picture( MSIDATABASE *db, LPCWSTR name,
760                                  INT cx, INT cy, DWORD flags )
761 {
762     HBITMAP hOleBitmap = 0, hBitmap = 0, hOldSrcBitmap, hOldDestBitmap;
763     MSIRECORD *rec = NULL;
764     IStream *stm = NULL;
765     IPicture *pic = NULL;
766     HDC srcdc, destdc;
767     BITMAP bm;
768     UINT r;
769
770     rec = msi_get_binary_record( db, name );
771     if( !rec )
772         goto end;
773
774     r = MSI_RecordGetIStream( rec, 2, &stm );
775     msiobj_release( &rec->hdr );
776     if( r != ERROR_SUCCESS )
777         goto end;
778
779     r = OleLoadPicture( stm, 0, TRUE, &IID_IPicture, (LPVOID*) &pic );
780     IStream_Release( stm );
781     if( FAILED( r ) )
782     {
783         ERR("failed to load picture\n");
784         goto end;
785     }
786
787     r = IPicture_get_Handle( pic, (OLE_HANDLE*) &hOleBitmap );
788     if( FAILED( r ) )
789     {
790         ERR("failed to get bitmap handle\n");
791         goto end;
792     }
793  
794     /* make the bitmap the desired size */
795     r = GetObjectW( hOleBitmap, sizeof bm, &bm );
796     if (r != sizeof bm )
797     {
798         ERR("failed to get bitmap size\n");
799         goto end;
800     }
801
802     if (flags & LR_DEFAULTSIZE)
803     {
804         cx = bm.bmWidth;
805         cy = bm.bmHeight;
806     }
807
808     srcdc = CreateCompatibleDC( NULL );
809     hOldSrcBitmap = SelectObject( srcdc, hOleBitmap );
810     destdc = CreateCompatibleDC( NULL );
811     hBitmap = CreateCompatibleBitmap( srcdc, cx, cy );
812     hOldDestBitmap = SelectObject( destdc, hBitmap );
813     StretchBlt( destdc, 0, 0, cx, cy,
814                 srcdc, 0, 0, bm.bmWidth, bm.bmHeight, SRCCOPY);
815     SelectObject( srcdc, hOldSrcBitmap );
816     SelectObject( destdc, hOldDestBitmap );
817     DeleteDC( srcdc );
818     DeleteDC( destdc );
819
820 end:
821     if ( pic )
822         IPicture_Release( pic );
823     return hBitmap;
824 }
825
826 static UINT msi_dialog_bitmap_control( msi_dialog *dialog, MSIRECORD *rec )
827 {
828     UINT cx, cy, flags, style, attributes;
829     msi_control *control;
830     LPWSTR text;
831
832     flags = LR_LOADFROMFILE;
833     style = SS_BITMAP | SS_LEFT | WS_GROUP;
834
835     attributes = MSI_RecordGetInteger( rec, 8 );
836     if( attributes & msidbControlAttributesFixedSize )
837     {
838         flags |= LR_DEFAULTSIZE;
839         style |= SS_CENTERIMAGE;
840     }
841
842     control = msi_dialog_add_control( dialog, rec, szStatic, style );
843     cx = MSI_RecordGetInteger( rec, 6 );
844     cy = MSI_RecordGetInteger( rec, 7 );
845     cx = msi_dialog_scale_unit( dialog, cx );
846     cy = msi_dialog_scale_unit( dialog, cy );
847
848     text = msi_get_deformatted_field( dialog->package, rec, 10 );
849     control->hBitmap = msi_load_picture( dialog->package->db, text, cx, cy, flags );
850     if( control->hBitmap )
851         SendMessageW( control->hwnd, STM_SETIMAGE,
852                       IMAGE_BITMAP, (LPARAM) control->hBitmap );
853     else
854         ERR("Failed to load bitmap %s\n", debugstr_w(text));
855
856     msi_free( text );
857     
858     return ERROR_SUCCESS;
859 }
860
861 static UINT msi_dialog_icon_control( msi_dialog *dialog, MSIRECORD *rec )
862 {
863     msi_control *control;
864     DWORD attributes;
865     LPWSTR text;
866
867     TRACE("\n");
868
869     control = msi_dialog_add_control( dialog, rec, szStatic,
870                             SS_ICON | SS_CENTERIMAGE | WS_GROUP );
871             
872     attributes = MSI_RecordGetInteger( rec, 8 );
873     text = msi_get_deformatted_field( dialog->package, rec, 10 );
874     control->hIcon = msi_load_icon( dialog->package->db, text, attributes );
875     if( control->hIcon )
876         SendMessageW( control->hwnd, STM_SETICON, (WPARAM) control->hIcon, 0 );
877     else
878         ERR("Failed to load bitmap %s\n", debugstr_w(text));
879     msi_free( text );
880     return ERROR_SUCCESS;
881 }
882
883 static UINT msi_dialog_combo_control( msi_dialog *dialog, MSIRECORD *rec )
884 {
885     static const WCHAR szCombo[] = { 'C','O','M','B','O','B','O','X',0 };
886
887     msi_dialog_add_control( dialog, rec, szCombo,
888                             SS_BITMAP | SS_LEFT | SS_CENTERIMAGE );
889     return ERROR_SUCCESS;
890 }
891
892 static UINT msi_dialog_edit_control( msi_dialog *dialog, MSIRECORD *rec )
893 {
894     msi_control *control;
895     LPCWSTR prop;
896     LPWSTR val;
897
898     control = msi_dialog_add_control( dialog, rec, szEdit,
899                                       WS_BORDER | WS_TABSTOP );
900     control->handler = msi_dialog_edit_handler;
901     prop = MSI_RecordGetString( rec, 9 );
902     if( prop )
903         control->property = strdupW( prop );
904     val = msi_dup_property( dialog->package, control->property );
905     SetWindowTextW( control->hwnd, val );
906     msi_free( val );
907     return ERROR_SUCCESS;
908 }
909
910 /******************** Masked Edit ********************************************/
911
912 #define MASK_MAX_GROUPS 10
913
914 struct msi_mask_group
915 {
916     UINT len;
917     UINT ofs;
918     WCHAR type;
919     HWND hwnd;
920 };
921
922 struct msi_maskedit_info
923 {
924     msi_dialog *dialog;
925     WNDPROC oldproc;
926     HWND hwnd;
927     LPWSTR prop;
928     UINT num_chars;
929     UINT num_groups;
930     struct msi_mask_group group[MASK_MAX_GROUPS];
931 };
932
933 static BOOL msi_mask_editable( WCHAR type )
934 {
935     switch (type)
936     {
937     case '%':
938     case '#':
939     case '&':
940     case '`':
941     case '?':
942     case '^':
943         return TRUE;
944     }
945     return FALSE;
946 }
947
948 static void msi_mask_control_change( struct msi_maskedit_info *info )
949 {
950     LPWSTR val;
951     UINT i, n, r;
952
953     val = msi_alloc( (info->num_chars+1)*sizeof(WCHAR) );
954     for( i=0, n=0; i<info->num_groups; i++ )
955     {
956         if( (info->group[i].len + n) > info->num_chars )
957         {
958             ERR("can't fit control %d text into template\n",i);
959             break;
960         }
961         if (!msi_mask_editable(info->group[i].type))
962         {
963             for(r=0; r<info->group[i].len; r++)
964                 val[n+r] = info->group[i].type;
965             val[n+r] = 0;
966         }
967         else
968         {
969             r = GetWindowTextW( info->group[i].hwnd, &val[n], info->group[i].len+1 );
970             if( r != info->group[i].len )
971                 break;
972         }
973         n += r;
974     }
975
976     TRACE("%d/%d controls were good\n", i, info->num_groups);
977
978     if( i == info->num_groups )
979     {
980         TRACE("Set property %s to %s\n",
981               debugstr_w(info->prop), debugstr_w(val) );
982         CharUpperBuffW( val, info->num_chars );
983         MSI_SetPropertyW( info->dialog->package, info->prop, val );
984         msi_dialog_evaluate_control_conditions( info->dialog );
985     }
986     msi_free( val );
987 }
988
989 /* now move to the next control if necessary */
990 static VOID msi_mask_next_control( struct msi_maskedit_info *info, HWND hWnd )
991 {
992     HWND hWndNext;
993     UINT len, i;
994
995     for( i=0; i<info->num_groups; i++ )
996         if( info->group[i].hwnd == hWnd )
997             break;
998
999     /* don't move from the last control */
1000     if( i >= (info->num_groups-1) )
1001         return;
1002
1003     len = SendMessageW( hWnd, WM_GETTEXTLENGTH, 0, 0 );
1004     if( len < info->group[i].len )
1005         return;
1006
1007     hWndNext = GetNextDlgTabItem( GetParent( hWnd ), hWnd, FALSE );
1008     SetFocus( hWndNext );
1009 }
1010
1011 static LRESULT WINAPI
1012 MSIMaskedEdit_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
1013 {
1014     struct msi_maskedit_info *info;
1015     HRESULT r;
1016
1017     TRACE("%p %04x %08x %08lx\n", hWnd, msg, wParam, lParam);
1018
1019     info = GetPropW(hWnd, szButtonData);
1020
1021     r = CallWindowProcW(info->oldproc, hWnd, msg, wParam, lParam);
1022
1023     switch( msg )
1024     {
1025     case WM_COMMAND:
1026         if (HIWORD(wParam) == EN_CHANGE)
1027         {
1028             msi_mask_control_change( info );
1029             msi_mask_next_control( info, (HWND) lParam );
1030         }
1031         break;
1032     case WM_NCDESTROY:
1033         msi_free( info->prop );
1034         msi_free( info );
1035         RemovePropW( hWnd, szButtonData );
1036         break;
1037     }
1038
1039     return r;
1040 }
1041
1042 /* fish the various bits of the property out and put them in the control */
1043 static void
1044 msi_maskedit_set_text( struct msi_maskedit_info *info, LPCWSTR text )
1045 {
1046     LPCWSTR p;
1047     UINT i;
1048
1049     p = text;
1050     for( i = 0; i < info->num_groups; i++ )
1051     {
1052         if( info->group[i].len < lstrlenW( p ) )
1053         {
1054             LPWSTR chunk = strdupW( p );
1055             chunk[ info->group[i].len ] = 0;
1056             SetWindowTextW( info->group[i].hwnd, chunk );
1057             msi_free( chunk );
1058         }
1059         else
1060         {
1061             SetWindowTextW( info->group[i].hwnd, p );
1062             break;
1063         }
1064         p += info->group[i].len;
1065     }
1066 }
1067
1068 static struct msi_maskedit_info * msi_dialog_parse_groups( LPCWSTR mask )
1069 {
1070     struct msi_maskedit_info * info = NULL;
1071     int i = 0, n = 0, total = 0;
1072     LPCWSTR p;
1073
1074     TRACE("masked control, template %s\n", debugstr_w(mask));
1075
1076     if( !mask )
1077         return info;
1078
1079     p = strchrW(mask, '<');
1080     if( !p )
1081         return info;
1082
1083     info = msi_alloc_zero( sizeof *info );
1084     if( !info )
1085         return info;
1086
1087     p++;
1088     for( i=0; i<MASK_MAX_GROUPS; i++ )
1089     {
1090         /* stop at the end of the string */
1091         if( p[0] == 0 || p[0] == '>' )
1092             break;
1093
1094         /* count the number of the same identifier */
1095         for( n=0; p[n] == p[0]; n++ )
1096             ;
1097         info->group[i].ofs = total;
1098         info->group[i].type = p[0];
1099         if( p[n] == '=' )
1100         {
1101             n++;
1102             total++; /* an extra not part of the group */
1103         }
1104         info->group[i].len = n;
1105         total += n;
1106         p += n;
1107     }
1108
1109     TRACE("%d characters in %d groups\n", total, i );
1110     if( i == MASK_MAX_GROUPS )
1111         ERR("too many groups in PIDTemplate %s\n", debugstr_w(mask));
1112
1113     info->num_chars = total;
1114     info->num_groups = i;
1115
1116     return info;
1117 }
1118
1119 static void
1120 msi_maskedit_create_children( struct msi_maskedit_info *info, LPCWSTR font )
1121 {
1122     DWORD width, height, style, wx, ww;
1123     RECT rect;
1124     HWND hwnd;
1125     UINT i;
1126
1127     style = WS_CHILD | WS_BORDER | WS_VISIBLE | WS_TABSTOP;
1128
1129     GetClientRect( info->hwnd, &rect );
1130
1131     width = rect.right - rect.left;
1132     height = rect.bottom - rect.top;
1133
1134     for( i = 0; i < info->num_groups; i++ )
1135     {
1136         if (!msi_mask_editable( info->group[i].type ))
1137             continue;
1138         wx = (info->group[i].ofs * width) / info->num_chars;
1139         ww = (info->group[i].len * width) / info->num_chars;
1140
1141         hwnd = CreateWindowW( szEdit, NULL, style, wx, 0, ww, height,
1142                               info->hwnd, NULL, NULL, NULL );
1143         if( !hwnd )
1144         {
1145             ERR("failed to create mask edit sub window\n");
1146             break;
1147         }
1148
1149         SendMessageW( hwnd, EM_LIMITTEXT, info->group[i].len, 0 );
1150
1151         msi_dialog_set_font( info->dialog, hwnd,
1152                              font?font:info->dialog->default_font );
1153         info->group[i].hwnd = hwnd;
1154     }
1155 }
1156
1157 /*
1158  * office 2003 uses "73931<````=````=````=````=`````>@@@@@"
1159  * delphi 7 uses "<????-??????-??????-????>" and "<???-???>"
1160  * filemaker pro 7 uses "<^^^^=^^^^=^^^^=^^^^=^^^^=^^^^=^^^^^>"
1161  */
1162 static UINT msi_dialog_maskedit_control( msi_dialog *dialog, MSIRECORD *rec )
1163 {
1164     LPWSTR font_mask, val = NULL, font;
1165     struct msi_maskedit_info *info = NULL;
1166     UINT ret = ERROR_SUCCESS;
1167     msi_control *control;
1168     LPCWSTR prop, mask;
1169
1170     TRACE("\n");
1171
1172     font_mask = msi_get_deformatted_field( dialog->package, rec, 10 );
1173     font = msi_dialog_get_style( font_mask, &mask );
1174     if( !mask )
1175     {
1176         ERR("mask template is empty\n");
1177         goto end;
1178     }
1179
1180     info = msi_dialog_parse_groups( mask );
1181     if( !info )
1182     {
1183         ERR("template %s is invalid\n", debugstr_w(mask));
1184         goto end;
1185     }
1186
1187     info->dialog = dialog;
1188
1189     control = msi_dialog_add_control( dialog, rec, szStatic,
1190                    SS_OWNERDRAW | WS_GROUP | WS_VISIBLE );
1191     if( !control )
1192     {
1193         ERR("Failed to create maskedit container\n");
1194         ret = ERROR_FUNCTION_FAILED;
1195         goto end;
1196     }
1197     SetWindowLongPtrW( control->hwnd, GWL_EXSTYLE, WS_EX_CONTROLPARENT );
1198
1199     info->hwnd = control->hwnd;
1200
1201     /* subclass the static control */
1202     info->oldproc = (WNDPROC) SetWindowLongPtrW( info->hwnd, GWLP_WNDPROC,
1203                                           (LONG_PTR)MSIMaskedEdit_WndProc );
1204     SetPropW( control->hwnd, szButtonData, info );
1205
1206     prop = MSI_RecordGetString( rec, 9 );
1207     if( prop )
1208         info->prop = strdupW( prop );
1209
1210     msi_maskedit_create_children( info, font );
1211
1212     if( prop )
1213     {
1214         val = msi_dup_property( dialog->package, prop );
1215         if( val )
1216         {
1217             msi_maskedit_set_text( info, val );
1218             msi_free( val );
1219         }
1220     }
1221
1222 end:
1223     if( ret != ERROR_SUCCESS )
1224         msi_free( info );
1225     msi_free( font_mask );
1226     msi_free( font );
1227     return ret;
1228 }
1229
1230 /******************** Progress Bar *****************************************/
1231
1232 static UINT msi_dialog_progress_bar( msi_dialog *dialog, MSIRECORD *rec )
1233 {
1234     msi_dialog_add_control( dialog, rec, PROGRESS_CLASSW, WS_VISIBLE );
1235     return ERROR_SUCCESS;
1236 }
1237
1238 /******************** Path Edit ********************************************/
1239
1240 static UINT msi_dialog_pathedit_control( msi_dialog *dialog, MSIRECORD *rec )
1241 {
1242     FIXME("not implemented properly\n");
1243     return msi_dialog_edit_control( dialog, rec );
1244 }
1245
1246 /* radio buttons are a bit different from normal controls */
1247 static UINT msi_dialog_create_radiobutton( MSIRECORD *rec, LPVOID param )
1248 {
1249     radio_button_group_descr *group = (radio_button_group_descr *)param;
1250     msi_dialog *dialog = group->dialog;
1251     msi_control *control;
1252     LPCWSTR prop, text, name;
1253     DWORD style, attributes = group->attributes;
1254
1255     style = WS_CHILD | BS_AUTORADIOBUTTON | BS_MULTILINE | WS_TABSTOP;
1256     name = MSI_RecordGetString( rec, 3 );
1257     text = MSI_RecordGetString( rec, 8 );
1258     if( attributes & 1 )
1259         style |= WS_VISIBLE;
1260     if( ~attributes & 2 )
1261         style |= WS_DISABLED;
1262
1263     control = msi_dialog_create_window( dialog, rec, szButton, name, text,
1264                                         style, group->parent->hwnd );
1265     if (!control)
1266         return ERROR_FUNCTION_FAILED;
1267     control->handler = msi_dialog_radiogroup_handler;
1268
1269     prop = MSI_RecordGetString( rec, 1 );
1270     if( prop )
1271         control->property = strdupW( prop );
1272
1273     return ERROR_SUCCESS;
1274 }
1275
1276 static UINT msi_dialog_radiogroup_control( msi_dialog *dialog, MSIRECORD *rec )
1277 {
1278     static const WCHAR query[] = {
1279         'S','E','L','E','C','T',' ','*',' ',
1280         'F','R','O','M',' ','R','a','d','i','o','B','u','t','t','o','n',' ',
1281         'W','H','E','R','E',' ',
1282            '`','P','r','o','p','e','r','t','y','`',' ','=',' ','\'','%','s','\'',0};
1283     UINT r;
1284     LPCWSTR prop;
1285     msi_control *control;
1286     MSIQUERY *view = NULL;
1287     radio_button_group_descr group;
1288     MSIPACKAGE *package = dialog->package;
1289     WNDPROC oldproc;
1290
1291     prop = MSI_RecordGetString( rec, 9 );
1292
1293     TRACE("%p %p %s\n", dialog, rec, debugstr_w( prop ));
1294
1295     /* Create parent group box to hold radio buttons */
1296     control = msi_dialog_add_control( dialog, rec, szButton, BS_OWNERDRAW|WS_GROUP );
1297     if( !control )
1298         return ERROR_FUNCTION_FAILED;
1299
1300     oldproc = (WNDPROC) SetWindowLongPtrW( control->hwnd, GWLP_WNDPROC,
1301                                            (LONG_PTR)MSIRadioGroup_WndProc );
1302     SetPropW(control->hwnd, szButtonData, oldproc);
1303     SetWindowLongPtrW( control->hwnd, GWL_EXSTYLE, WS_EX_CONTROLPARENT );
1304
1305     if( prop )
1306         control->property = strdupW( prop );
1307
1308     /* query the Radio Button table for all control in this group */
1309     r = MSI_OpenQuery( package->db, &view, query, prop );
1310     if( r != ERROR_SUCCESS )
1311     {
1312         ERR("query failed for dialog %s radio group %s\n", 
1313             debugstr_w(dialog->name), debugstr_w(prop));
1314         return ERROR_INVALID_PARAMETER;
1315     }
1316
1317     group.dialog = dialog;
1318     group.parent = control;
1319     group.attributes = MSI_RecordGetInteger( rec, 8 );
1320
1321     r = MSI_IterateRecords( view, 0, msi_dialog_create_radiobutton, &group );
1322     msiobj_release( &view->hdr );
1323
1324     return r;
1325 }
1326
1327 /******************** Selection Tree ***************************************/
1328
1329 static void
1330 msi_dialog_tv_add_child_features( MSIPACKAGE *package, HWND hwnd,
1331                                   LPCWSTR parent, HTREEITEM hParent )
1332 {
1333     MSIFEATURE *feature;
1334     TVINSERTSTRUCTW tvis;
1335     HTREEITEM hitem;
1336
1337     LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
1338     {
1339         if ( lstrcmpW( parent, feature->Feature_Parent ) )
1340             continue;
1341
1342         if ( !feature->Title )
1343             continue;
1344
1345         memset( &tvis, 0, sizeof tvis );
1346         tvis.hParent = hParent;
1347         tvis.hInsertAfter = TVI_SORT;
1348         if (feature->Title)
1349         {
1350             tvis.u.item.mask = TVIF_TEXT;
1351             tvis.u.item.pszText = feature->Title;
1352         }
1353         tvis.u.item.lParam = (LPARAM) feature;
1354         hitem = (HTREEITEM) SendMessageW( hwnd, TVM_INSERTITEMW, 0, (LPARAM) &tvis );
1355         if (!hitem)
1356             continue;
1357
1358         msi_dialog_tv_add_child_features( package, hwnd,
1359                                           feature->Feature, hitem );
1360     }
1361 }
1362
1363 static UINT msi_dialog_selection_tree( msi_dialog *dialog, MSIRECORD *rec )
1364 {
1365     msi_control *control;
1366     LPCWSTR prop;
1367     LPWSTR val;
1368     MSIPACKAGE *package = dialog->package;
1369
1370     prop = MSI_RecordGetString( rec, 9 );
1371     val = msi_dup_property( package, prop );
1372     control = msi_dialog_add_control( dialog, rec, WC_TREEVIEWW,
1373                                       TVS_HASBUTTONS | WS_GROUP | WS_VSCROLL );
1374     if (!control)
1375         return ERROR_FUNCTION_FAILED;
1376
1377     msi_dialog_tv_add_child_features( package, control->hwnd, NULL, NULL );
1378
1379     msi_free( val );
1380
1381     return ERROR_SUCCESS;
1382 }
1383
1384 struct control_handler msi_dialog_handler[] =
1385 {
1386     { szText, msi_dialog_text_control },
1387     { szPushButton, msi_dialog_button_control },
1388     { szLine, msi_dialog_line_control },
1389     { szBitmap, msi_dialog_bitmap_control },
1390     { szCheckBox, msi_dialog_checkbox_control },
1391     { szScrollableText, msi_dialog_scrolltext_control },
1392     { szComboBox, msi_dialog_combo_control },
1393     { szEdit, msi_dialog_edit_control },
1394     { szMaskedEdit, msi_dialog_maskedit_control },
1395     { szPathEdit, msi_dialog_pathedit_control },
1396     { szProgressBar, msi_dialog_progress_bar },
1397     { szRadioButtonGroup, msi_dialog_radiogroup_control },
1398     { szIcon, msi_dialog_icon_control },
1399     { szSelectionTree, msi_dialog_selection_tree },
1400 };
1401
1402 #define NUM_CONTROL_TYPES (sizeof msi_dialog_handler/sizeof msi_dialog_handler[0])
1403
1404 static UINT msi_dialog_create_controls( MSIRECORD *rec, LPVOID param )
1405 {
1406     msi_dialog *dialog = param;
1407     LPCWSTR control_type;
1408     UINT i;
1409
1410     /* find and call the function that can create this type of control */
1411     control_type = MSI_RecordGetString( rec, 3 );
1412     for( i=0; i<NUM_CONTROL_TYPES; i++ )
1413         if (!strcmpiW( msi_dialog_handler[i].control_type, control_type ))
1414             break;
1415     if( i != NUM_CONTROL_TYPES )
1416         msi_dialog_handler[i].func( dialog, rec );
1417     else
1418         ERR("no handler for element type %s\n", debugstr_w(control_type));
1419
1420     return ERROR_SUCCESS;
1421 }
1422
1423 static UINT msi_dialog_fill_controls( msi_dialog *dialog )
1424 {
1425     static const WCHAR query[] = {
1426         'S','E','L','E','C','T',' ','*',' ',
1427         'F','R','O','M',' ','C','o','n','t','r','o','l',' ',
1428         'W','H','E','R','E',' ',
1429            '`','D','i','a','l','o','g','_','`',' ','=',' ','\'','%','s','\'',0};
1430     UINT r;
1431     MSIQUERY *view = NULL;
1432     MSIPACKAGE *package = dialog->package;
1433
1434     TRACE("%p %s\n", dialog, debugstr_w(dialog->name) );
1435
1436     /* query the Control table for all the elements of the control */
1437     r = MSI_OpenQuery( package->db, &view, query, dialog->name );
1438     if( r != ERROR_SUCCESS )
1439     {
1440         ERR("query failed for dialog %s\n", debugstr_w(dialog->name));
1441         return ERROR_INVALID_PARAMETER;
1442     }
1443
1444     r = MSI_IterateRecords( view, 0, msi_dialog_create_controls, dialog );
1445     msiobj_release( &view->hdr );
1446
1447     return r;
1448 }
1449
1450 static UINT msi_dialog_set_control_condition( MSIRECORD *rec, LPVOID param )
1451 {
1452     static const WCHAR szHide[] = { 'H','i','d','e',0 };
1453     static const WCHAR szShow[] = { 'S','h','o','w',0 };
1454     static const WCHAR szDisable[] = { 'D','i','s','a','b','l','e',0 };
1455     static const WCHAR szEnable[] = { 'E','n','a','b','l','e',0 };
1456     msi_dialog *dialog = param;
1457     msi_control *control;
1458     LPCWSTR name, action, condition;
1459     UINT r;
1460
1461     name = MSI_RecordGetString( rec, 2 );
1462     action = MSI_RecordGetString( rec, 3 );
1463     condition = MSI_RecordGetString( rec, 4 );
1464     r = MSI_EvaluateConditionW( dialog->package, condition );
1465     control = msi_dialog_find_control( dialog, name );
1466     if( r == MSICONDITION_TRUE && control )
1467     {
1468         TRACE("%s control %s\n", debugstr_w(action), debugstr_w(name));
1469
1470         /* FIXME: case sensitive? */
1471         if(!lstrcmpW(action, szHide))
1472             ShowWindow(control->hwnd, SW_HIDE);
1473         else if(!strcmpW(action, szShow))
1474             ShowWindow(control->hwnd, SW_SHOW);
1475         else if(!strcmpW(action, szDisable))
1476             EnableWindow(control->hwnd, FALSE);
1477         else if(!strcmpW(action, szEnable))
1478             EnableWindow(control->hwnd, TRUE);
1479         else
1480             FIXME("Unhandled action %s\n", debugstr_w(action));
1481     }
1482
1483     return ERROR_SUCCESS;
1484 }
1485
1486 static UINT msi_dialog_evaluate_control_conditions( msi_dialog *dialog )
1487 {
1488     static const WCHAR query[] = {
1489       'S','E','L','E','C','T',' ','*',' ',
1490       'F','R','O','M',' ',
1491         'C','o','n','t','r','o','l','C','o','n','d','i','t','i','o','n',' ',
1492       'W','H','E','R','E',' ',
1493         '`','D','i','a','l','o','g','_','`',' ','=',' ','\'','%','s','\'',0
1494     };
1495     UINT r;
1496     MSIQUERY *view = NULL;
1497     MSIPACKAGE *package = dialog->package;
1498
1499     TRACE("%p %s\n", dialog, debugstr_w(dialog->name) );
1500
1501     /* query the Control table for all the elements of the control */
1502     r = MSI_OpenQuery( package->db, &view, query, dialog->name );
1503     if( r != ERROR_SUCCESS )
1504     {
1505         ERR("query failed for dialog %s\n", debugstr_w(dialog->name));
1506         return ERROR_INVALID_PARAMETER;
1507     }
1508
1509     r = MSI_IterateRecords( view, 0, msi_dialog_set_control_condition, dialog );
1510     msiobj_release( &view->hdr );
1511
1512     return r;
1513 }
1514
1515 /* figure out the height of 10 point MS Sans Serif */
1516 static INT msi_dialog_get_sans_serif_height( HWND hwnd )
1517 {
1518     static const WCHAR szSansSerif[] = {
1519         'M','S',' ','S','a','n','s',' ','S','e','r','i','f',0 };
1520     LOGFONTW lf;
1521     TEXTMETRICW tm;
1522     BOOL r;
1523     LONG height = 0;
1524     HFONT hFont, hOldFont;
1525     HDC hdc;
1526
1527     hdc = GetDC( hwnd );
1528     if (hdc)
1529     {
1530         memset( &lf, 0, sizeof lf );
1531         lf.lfHeight = MulDiv(10, GetDeviceCaps(hdc, LOGPIXELSY), 72);
1532         strcpyW( lf.lfFaceName, szSansSerif );
1533         hFont = CreateFontIndirectW(&lf);
1534         if (hFont)
1535         {
1536             hOldFont = SelectObject( hdc, hFont );
1537             r = GetTextMetricsW( hdc, &tm );
1538             if (r)
1539                 height = tm.tmHeight;
1540             SelectObject( hdc, hOldFont );
1541             DeleteObject( hFont );
1542         }
1543         ReleaseDC( hwnd, hdc );
1544     }
1545     return height;
1546 }
1547
1548 /* fetch the associated record from the Dialog table */
1549 static MSIRECORD *msi_get_dialog_record( msi_dialog *dialog )
1550 {
1551     static const WCHAR query[] = {
1552         'S','E','L','E','C','T',' ','*',' ',
1553         'F','R','O','M',' ','D','i','a','l','o','g',' ',
1554         'W','H','E','R','E',' ',
1555            '`','D','i','a','l','o','g','`',' ','=',' ','\'','%','s','\'',0};
1556     MSIPACKAGE *package = dialog->package;
1557     MSIRECORD *rec = NULL;
1558
1559     TRACE("%p %s\n", dialog, debugstr_w(dialog->name) );
1560
1561     rec = MSI_QueryGetRecord( package->db, query, dialog->name );
1562     if( !rec )
1563         ERR("query failed for dialog %s\n", debugstr_w(dialog->name));
1564
1565     return rec;
1566 }
1567
1568 static void msi_dialog_adjust_dialog_size( msi_dialog *dialog, LPSIZE sz )
1569 {
1570     RECT rect;
1571     LONG style;
1572
1573     /* turn the client size into the window rectangle */
1574     rect.left = 0;
1575     rect.top = 0;
1576     rect.right = msi_dialog_scale_unit( dialog, sz->cx );
1577     rect.bottom = msi_dialog_scale_unit( dialog, sz->cy );
1578     style = GetWindowLongPtrW( dialog->hwnd, GWL_STYLE );
1579     AdjustWindowRect( &rect, style, FALSE );
1580     sz->cx = rect.right - rect.left;
1581     sz->cy = rect.bottom - rect.top;
1582 }
1583
1584 static BOOL msi_control_set_next( msi_control *control, msi_control *next )
1585 {
1586     return SetWindowPos( next->hwnd, control->hwnd, 0, 0, 0, 0,
1587                          SWP_NOMOVE | SWP_NOOWNERZORDER | SWP_NOREDRAW |
1588                          SWP_NOREPOSITION | SWP_NOSENDCHANGING | SWP_NOSIZE );
1589 }
1590
1591 static UINT msi_dialog_set_tab_order( msi_dialog *dialog )
1592 {
1593     msi_control *control, *tab_next;
1594
1595     LIST_FOR_EACH_ENTRY( control, &dialog->controls, msi_control, entry )
1596     {
1597         tab_next = msi_dialog_find_control( dialog, control->tabnext );
1598         if( !tab_next )
1599             continue;
1600         msi_control_set_next( control, tab_next );
1601     }
1602
1603     return ERROR_SUCCESS;
1604 }
1605
1606 static void msi_dialog_set_first_control( msi_dialog* dialog, LPCWSTR name )
1607 {
1608     msi_control *control;
1609
1610     control = msi_dialog_find_control( dialog, name );
1611     if( control )
1612         dialog->hWndFocus = control->hwnd;
1613     else
1614         dialog->hWndFocus = NULL;
1615 }
1616
1617 static LRESULT msi_dialog_oncreate( HWND hwnd, LPCREATESTRUCTW cs )
1618 {
1619     static const WCHAR df[] = {
1620         'D','e','f','a','u','l','t','U','I','F','o','n','t',0 };
1621     static const WCHAR dfv[] = {
1622         'M','S',' ','S','h','e','l','l',' ','D','l','g',0 };
1623     msi_dialog *dialog = (msi_dialog*) cs->lpCreateParams;
1624     MSIRECORD *rec = NULL;
1625     LPWSTR title = NULL;
1626     SIZE size;
1627
1628     TRACE("%p %p\n", dialog, dialog->package);
1629
1630     dialog->hwnd = hwnd;
1631     SetWindowLongPtrW( hwnd, GWLP_USERDATA, (LONG_PTR) dialog );
1632
1633     rec = msi_get_dialog_record( dialog );
1634     if( !rec )
1635     {
1636         TRACE("No record found for dialog %s\n", debugstr_w(dialog->name));
1637         return -1;
1638     }
1639
1640     dialog->scale = msi_dialog_get_sans_serif_height(dialog->hwnd);
1641
1642     size.cx = MSI_RecordGetInteger( rec, 4 );
1643     size.cy = MSI_RecordGetInteger( rec, 5 );
1644     msi_dialog_adjust_dialog_size( dialog, &size );
1645
1646     dialog->attributes = MSI_RecordGetInteger( rec, 6 );
1647
1648     dialog->default_font = msi_dup_property( dialog->package, df );
1649     if (!dialog->default_font)
1650     {
1651         dialog->default_font = strdupW(dfv);
1652         if (!dialog->default_font) return -1;
1653     }
1654
1655     title = msi_get_deformatted_field( dialog->package, rec, 7 );
1656     SetWindowTextW( hwnd, title );
1657     msi_free( title );
1658
1659     SetWindowPos( hwnd, 0, 0, 0, size.cx, size.cy,
1660                   SWP_NOMOVE | SWP_NOACTIVATE | SWP_NOZORDER | SWP_NOREDRAW );
1661
1662
1663     msi_dialog_build_font_list( dialog );
1664     msi_dialog_fill_controls( dialog );
1665     msi_dialog_evaluate_control_conditions( dialog );
1666     msi_dialog_set_tab_order( dialog );
1667     msi_dialog_set_first_control( dialog, MSI_RecordGetString( rec, 8 ) );
1668     msiobj_release( &rec->hdr );
1669
1670     return 0;
1671 }
1672
1673 static UINT msi_dialog_send_event( msi_dialog *dialog, LPCWSTR event, LPCWSTR arg )
1674 {
1675     LPWSTR event_fmt = NULL, arg_fmt = NULL;
1676
1677     TRACE("Sending control event %s %s\n", debugstr_w(event), debugstr_w(arg));
1678
1679     deformat_string( dialog->package, event, &event_fmt );
1680     deformat_string( dialog->package, arg, &arg_fmt );
1681
1682     dialog->event_handler( dialog->package, event_fmt, arg_fmt, dialog );
1683
1684     msi_free( event_fmt );
1685     msi_free( arg_fmt );
1686
1687     return ERROR_SUCCESS;
1688 }
1689
1690 static UINT msi_dialog_set_property( msi_dialog *dialog, LPCWSTR event, LPCWSTR arg )
1691 {
1692     static const WCHAR szNullArg[] = { '{','}',0 };
1693     LPWSTR p, prop, arg_fmt = NULL;
1694     UINT len;
1695
1696     len = strlenW(event);
1697     prop = msi_alloc( len*sizeof(WCHAR));
1698     strcpyW( prop, &event[1] );
1699     p = strchrW( prop, ']' );
1700     if( p && p[1] == 0 )
1701     {
1702         *p = 0;
1703         if( strcmpW( szNullArg, arg ) )
1704             deformat_string( dialog->package, arg, &arg_fmt );
1705         MSI_SetPropertyW( dialog->package, prop, arg_fmt );
1706         msi_free( arg_fmt );
1707     }
1708     else
1709         ERR("Badly formatted property string - what happens?\n");
1710     msi_free( prop );
1711     return ERROR_SUCCESS;
1712 }
1713
1714 static UINT msi_dialog_control_event( MSIRECORD *rec, LPVOID param )
1715 {
1716     msi_dialog *dialog = param;
1717     LPCWSTR condition, event, arg;
1718     UINT r;
1719
1720     condition = MSI_RecordGetString( rec, 5 );
1721     r = MSI_EvaluateConditionW( dialog->package, condition );
1722     if( r == MSICONDITION_TRUE )
1723     {
1724         event = MSI_RecordGetString( rec, 3 );
1725         arg = MSI_RecordGetString( rec, 4 );
1726         if( event[0] == '[' )
1727             msi_dialog_set_property( dialog, event, arg );
1728         else
1729             msi_dialog_send_event( dialog, event, arg );
1730     }
1731
1732     return ERROR_SUCCESS;
1733 }
1734
1735 static UINT msi_dialog_button_handler( msi_dialog *dialog,
1736                                        msi_control *control, WPARAM param )
1737 {
1738     static const WCHAR query[] = {
1739       'S','E','L','E','C','T',' ','*',' ',
1740       'F','R','O','M',' ','C','o','n','t','r','o','l','E','v','e','n','t',' ',
1741       'W','H','E','R','E',' ',
1742          '`','D','i','a','l','o','g','_','`',' ','=',' ','\'','%','s','\'',' ',
1743       'A','N','D',' ',
1744          '`','C','o','n','t','r','o','l','_','`',' ','=',' ','\'','%','s','\'',' ',
1745       'O','R','D','E','R',' ','B','Y',' ','`','O','r','d','e','r','i','n','g','`',0
1746     };
1747     MSIQUERY *view = NULL;
1748     UINT r;
1749
1750     if( HIWORD(param) != BN_CLICKED )
1751         return ERROR_SUCCESS;
1752
1753     r = MSI_OpenQuery( dialog->package->db, &view, query,
1754                        dialog->name, control->name );
1755     if( r != ERROR_SUCCESS )
1756     {
1757         ERR("query failed\n");
1758         return 0;
1759     }
1760
1761     r = MSI_IterateRecords( view, 0, msi_dialog_control_event, dialog );
1762     msiobj_release( &view->hdr );
1763
1764     return r;
1765 }
1766
1767 static UINT msi_dialog_get_checkbox_state( msi_dialog *dialog,
1768                 msi_control *control )
1769 {
1770     WCHAR state[2] = { 0 };
1771     DWORD sz = 2;
1772
1773     MSI_GetPropertyW( dialog->package, control->property, state, &sz );
1774     return state[0] ? 1 : 0;
1775 }
1776
1777 static void msi_dialog_set_checkbox_state( msi_dialog *dialog,
1778                 msi_control *control, UINT state )
1779 {
1780     static const WCHAR szState[] = { '1', 0 };
1781     LPCWSTR val;
1782
1783     /* if uncheck then the property is set to NULL */
1784     if (!state)
1785     {
1786         MSI_SetPropertyW( dialog->package, control->property, NULL );
1787         return;
1788     }
1789
1790     /* check for a custom state */
1791     if (control->value && control->value[0])
1792         val = control->value;
1793     else
1794         val = szState;
1795
1796     MSI_SetPropertyW( dialog->package, control->property, val );
1797 }
1798
1799 static void msi_dialog_checkbox_sync_state( msi_dialog *dialog,
1800                 msi_control *control )
1801 {
1802     UINT state;
1803
1804     state = msi_dialog_get_checkbox_state( dialog, control );
1805     SendMessageW( control->hwnd, BM_SETCHECK,
1806                   state ? BST_CHECKED : BST_UNCHECKED, 0 );
1807 }
1808
1809 static UINT msi_dialog_checkbox_handler( msi_dialog *dialog,
1810                 msi_control *control, WPARAM param )
1811 {
1812     UINT state;
1813
1814     if( HIWORD(param) != BN_CLICKED )
1815         return ERROR_SUCCESS;
1816
1817     TRACE("clicked checkbox %s, set %s\n", debugstr_w(control->name),
1818           debugstr_w(control->property));
1819
1820     state = msi_dialog_get_checkbox_state( dialog, control );
1821     state = state ? 0 : 1;
1822     msi_dialog_set_checkbox_state( dialog, control, state );
1823     msi_dialog_checkbox_sync_state( dialog, control );
1824
1825     return msi_dialog_button_handler( dialog, control, param );
1826 }
1827
1828 static UINT msi_dialog_edit_handler( msi_dialog *dialog,
1829                 msi_control *control, WPARAM param )
1830 {
1831     UINT sz, r;
1832     LPWSTR buf;
1833
1834     if( HIWORD(param) != EN_CHANGE )
1835         return ERROR_SUCCESS;
1836
1837     TRACE("edit %s contents changed, set %s\n", debugstr_w(control->name),
1838           debugstr_w(control->property));
1839
1840     sz = 0x20;
1841     buf = msi_alloc( sz*sizeof(WCHAR) );
1842     while( buf )
1843     {
1844         r = GetWindowTextW( control->hwnd, buf, sz );
1845         if( r < (sz-1) )
1846             break;
1847         sz *= 2;
1848         buf = msi_realloc( buf, sz*sizeof(WCHAR) );
1849     }
1850
1851     MSI_SetPropertyW( dialog->package, control->property, buf );
1852
1853     msi_free( buf );
1854
1855     return ERROR_SUCCESS;
1856 }
1857
1858 static UINT msi_dialog_radiogroup_handler( msi_dialog *dialog,
1859                 msi_control *control, WPARAM param )
1860 {
1861     if( HIWORD(param) != BN_CLICKED )
1862         return ERROR_SUCCESS;
1863
1864     TRACE("clicked radio button %s, set %s\n", debugstr_w(control->name),
1865           debugstr_w(control->property));
1866
1867     MSI_SetPropertyW( dialog->package, control->property, control->name );
1868
1869     return msi_dialog_button_handler( dialog, control, param );
1870 }
1871
1872 static LRESULT msi_dialog_oncommand( msi_dialog *dialog, WPARAM param, HWND hwnd )
1873 {
1874     msi_control *control = NULL;
1875
1876     TRACE("%p %p %08x\n", dialog, hwnd, param);
1877
1878     switch (param)
1879     {
1880     case 1: /* enter */
1881         control = msi_dialog_find_control( dialog, dialog->control_default );
1882         break;
1883     case 2: /* escape */
1884         control = msi_dialog_find_control( dialog, dialog->control_cancel );
1885         break;
1886     default: 
1887         control = msi_dialog_find_control_by_hwnd( dialog, hwnd );
1888     }
1889
1890     if( control )
1891     {
1892         if( control->handler )
1893         {
1894             control->handler( dialog, control, param );
1895             msi_dialog_evaluate_control_conditions( dialog );
1896         }
1897     }
1898     else
1899         ERR("button click from nowhere %p %d %p\n", dialog, param, hwnd);
1900     return 0;
1901 }
1902
1903 static void msi_dialog_setfocus( msi_dialog *dialog )
1904 {
1905     HWND hwnd = dialog->hWndFocus;
1906
1907     hwnd = GetNextDlgTabItem( dialog->hwnd, hwnd, TRUE);
1908     hwnd = GetNextDlgTabItem( dialog->hwnd, hwnd, FALSE);
1909     SetFocus( hwnd );
1910     dialog->hWndFocus = hwnd;
1911 }
1912
1913 static LRESULT WINAPI MSIDialog_WndProc( HWND hwnd, UINT msg,
1914                 WPARAM wParam, LPARAM lParam )
1915 {
1916     msi_dialog *dialog = (LPVOID) GetWindowLongPtrW( hwnd, GWLP_USERDATA );
1917
1918     TRACE("0x%04x\n", msg);
1919
1920     switch (msg)
1921     {
1922     case WM_CREATE:
1923         return msi_dialog_oncreate( hwnd, (LPCREATESTRUCTW)lParam );
1924
1925     case WM_COMMAND:
1926         return msi_dialog_oncommand( dialog, wParam, (HWND)lParam );
1927
1928     case WM_ACTIVATE:
1929         if( LOWORD(wParam) == WA_INACTIVE )
1930             dialog->hWndFocus = GetFocus();
1931         else
1932             msi_dialog_setfocus( dialog );
1933         return 0;
1934
1935     case WM_SETFOCUS:
1936         msi_dialog_setfocus( dialog );
1937         return 0;
1938
1939     /* bounce back to our subclassed static control */
1940     case WM_CTLCOLORSTATIC:
1941         return SendMessageW( (HWND) lParam, WM_CTLCOLORSTATIC, wParam, lParam );
1942
1943     case WM_DESTROY:
1944         dialog->hwnd = NULL;
1945         return 0;
1946     }
1947     return DefWindowProcW(hwnd, msg, wParam, lParam);
1948 }
1949
1950 static LRESULT WINAPI MSIRadioGroup_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
1951 {
1952     WNDPROC oldproc = (WNDPROC) GetPropW(hWnd, szButtonData);
1953
1954     TRACE("hWnd %p msg %04x wParam 0x%08x lParam 0x%08lx\n", hWnd, msg, wParam, lParam);
1955
1956     if (msg == WM_COMMAND) /* Forward notifications to dialog */
1957         SendMessageW(GetParent(hWnd), msg, wParam, lParam);
1958
1959     return CallWindowProcW(oldproc, hWnd, msg, wParam, lParam);
1960 }
1961
1962 static LRESULT WINAPI MSIHiddenWindowProc( HWND hwnd, UINT msg,
1963                 WPARAM wParam, LPARAM lParam )
1964 {
1965     msi_dialog *dialog = (msi_dialog*) lParam;
1966
1967     TRACE("%d %p\n", msg, dialog);
1968
1969     switch (msg)
1970     {
1971     case WM_MSI_DIALOG_CREATE:
1972         return msi_dialog_run_message_loop( dialog );
1973     case WM_MSI_DIALOG_DESTROY:
1974         msi_dialog_destroy( dialog );
1975         return 0;
1976     }
1977     return DefWindowProcW( hwnd, msg, wParam, lParam );
1978 }
1979
1980 /* functions that interface to other modules within MSI */
1981
1982 msi_dialog *msi_dialog_create( MSIPACKAGE* package, LPCWSTR szDialogName,
1983                                 msi_dialog_event_handler event_handler )
1984 {
1985     MSIRECORD *rec = NULL;
1986     msi_dialog *dialog;
1987
1988     TRACE("%p %s\n", package, debugstr_w(szDialogName));
1989
1990     /* allocate the structure for the dialog to use */
1991     dialog = msi_alloc_zero( sizeof *dialog + sizeof(WCHAR)*strlenW(szDialogName) );
1992     if( !dialog )
1993         return NULL;
1994     strcpyW( dialog->name, szDialogName );
1995     msiobj_addref( &package->hdr );
1996     dialog->package = package;
1997     dialog->event_handler = event_handler;
1998     dialog->finished = 0;
1999     list_init( &dialog->controls );
2000
2001     /* verify that the dialog exists */
2002     rec = msi_get_dialog_record( dialog );
2003     if( !rec )
2004     {
2005         msiobj_release( &package->hdr );
2006         msi_free( dialog );
2007         return NULL;
2008     }
2009     dialog->attributes = MSI_RecordGetInteger( rec, 6 );
2010     dialog->control_default = strdupW( MSI_RecordGetString( rec, 9 ) );
2011     dialog->control_cancel = strdupW( MSI_RecordGetString( rec, 10 ) );
2012     msiobj_release( &rec->hdr );
2013
2014     return dialog;
2015 }
2016
2017 static void msi_process_pending_messages( HWND hdlg )
2018 {
2019     MSG msg;
2020
2021     while( PeekMessageW( &msg, 0, 0, 0, PM_REMOVE ) )
2022     {
2023         if( hdlg && IsDialogMessageW( hdlg, &msg ))
2024             continue;
2025         TranslateMessage( &msg );
2026         DispatchMessageW( &msg );
2027     }
2028 }
2029
2030 void msi_dialog_end_dialog( msi_dialog *dialog )
2031 {
2032     TRACE("%p\n", dialog);
2033     dialog->finished = 1;
2034     PostMessageW(dialog->hwnd, WM_NULL, 0, 0);
2035 }
2036
2037 void msi_dialog_check_messages( HANDLE handle )
2038 {
2039     DWORD r;
2040
2041     /* in threads other than the UI thread, block */
2042     if( uiThreadId != GetCurrentThreadId() )
2043     {
2044         if( handle )
2045             WaitForSingleObject( handle, INFINITE );
2046         return;
2047     }
2048
2049     /* there's two choices for the UI thread */
2050     while (1)
2051     {
2052         msi_process_pending_messages( NULL );
2053
2054         if( !handle )
2055             break;
2056
2057         /*
2058          * block here until somebody creates a new dialog or
2059          * the handle we're waiting on becomes ready
2060          */
2061         r = MsgWaitForMultipleObjects( 1, &handle, 0, INFINITE, QS_ALLINPUT );
2062         if( r == WAIT_OBJECT_0 )
2063             break;
2064     }
2065 }
2066
2067 UINT msi_dialog_run_message_loop( msi_dialog *dialog )
2068 {
2069     HWND hwnd;
2070
2071     if( !(dialog->attributes & msidbDialogAttributesVisible) )
2072         return ERROR_SUCCESS;
2073
2074     if( uiThreadId != GetCurrentThreadId() )
2075         return SendMessageW( hMsiHiddenWindow, WM_MSI_DIALOG_CREATE, 0, (LPARAM) dialog );
2076
2077     /* create the dialog window, don't show it yet */
2078     hwnd = CreateWindowW( szMsiDialogClass, dialog->name, WS_OVERLAPPEDWINDOW,
2079                      CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
2080                      NULL, NULL, NULL, dialog );
2081     if( !hwnd )
2082     {
2083         ERR("Failed to create dialog %s\n", debugstr_w( dialog->name ));
2084         return ERROR_FUNCTION_FAILED;
2085     }
2086
2087     ShowWindow( hwnd, SW_SHOW );
2088     /* UpdateWindow( hwnd ); - and causes the transparent static controls not to paint */
2089
2090     if( dialog->attributes & msidbDialogAttributesModal )
2091     {
2092         while( !dialog->finished )
2093         {
2094             MsgWaitForMultipleObjects( 0, NULL, 0, INFINITE, QS_ALLEVENTS );
2095             msi_process_pending_messages( dialog->hwnd );
2096         }
2097     }
2098     else
2099         return ERROR_IO_PENDING;
2100
2101     return ERROR_SUCCESS;
2102 }
2103
2104 void msi_dialog_do_preview( msi_dialog *dialog )
2105 {
2106     TRACE("\n");
2107     dialog->attributes |= msidbDialogAttributesVisible;
2108     dialog->attributes &= ~msidbDialogAttributesModal;
2109     msi_dialog_run_message_loop( dialog );
2110 }
2111
2112 void msi_dialog_destroy( msi_dialog *dialog )
2113 {
2114     if( uiThreadId != GetCurrentThreadId() )
2115     {
2116         SendMessageW( hMsiHiddenWindow, WM_MSI_DIALOG_DESTROY, 0, (LPARAM) dialog );
2117         return;
2118     }
2119
2120     if( dialog->hwnd )
2121         ShowWindow( dialog->hwnd, SW_HIDE );
2122     
2123     if( dialog->hwnd )
2124         DestroyWindow( dialog->hwnd );
2125
2126     /* destroy the list of controls */
2127     while( !list_empty( &dialog->controls ) )
2128     {
2129         msi_control *t = LIST_ENTRY( list_head( &dialog->controls ),
2130                                      msi_control, entry );
2131         list_remove( &t->entry );
2132         /* leave dialog->hwnd - destroying parent destroys child windows */
2133         msi_free( t->property );
2134         msi_free( t->value );
2135         if( t->hBitmap )
2136             DeleteObject( t->hBitmap );
2137         if( t->hIcon )
2138             DestroyIcon( t->hIcon );
2139         msi_free( t->tabnext );
2140         msi_free( t );
2141         if (t->hDll)
2142             FreeLibrary( t->hDll );
2143     }
2144
2145     /* destroy the list of fonts */
2146     while( dialog->font_list )
2147     {
2148         msi_font *t = dialog->font_list;
2149         dialog->font_list = t->next;
2150         DeleteObject( t->hfont );
2151         msi_free( t );
2152     }
2153     msi_free( dialog->default_font );
2154
2155     msi_free( dialog->control_default );
2156     msi_free( dialog->control_cancel );
2157     msiobj_release( &dialog->package->hdr );
2158     dialog->package = NULL;
2159     msi_free( dialog );
2160 }
2161
2162 BOOL msi_dialog_register_class( void )
2163 {
2164     WNDCLASSW cls;
2165
2166     ZeroMemory( &cls, sizeof cls );
2167     cls.lpfnWndProc   = MSIDialog_WndProc;
2168     cls.hInstance     = NULL;
2169     cls.hIcon         = LoadIconW(0, (LPWSTR)IDI_APPLICATION);
2170     cls.hCursor       = LoadCursorW(0, (LPWSTR)IDC_ARROW);
2171     cls.hbrBackground = (HBRUSH)(COLOR_3DFACE + 1);
2172     cls.lpszMenuName  = NULL;
2173     cls.lpszClassName = szMsiDialogClass;
2174
2175     if( !RegisterClassW( &cls ) )
2176         return FALSE;
2177
2178     cls.lpfnWndProc   = MSIHiddenWindowProc;
2179     cls.lpszClassName = szMsiHiddenWindow;
2180
2181     if( !RegisterClassW( &cls ) )
2182         return FALSE;
2183
2184     uiThreadId = GetCurrentThreadId();
2185
2186     hMsiHiddenWindow = CreateWindowW( szMsiHiddenWindow, NULL, WS_OVERLAPPED,
2187                                    0, 0, 100, 100, NULL, NULL, NULL, NULL );
2188     if( !hMsiHiddenWindow )
2189         return FALSE;
2190
2191     return TRUE;
2192 }
2193
2194 void msi_dialog_unregister_class( void )
2195 {
2196     DestroyWindow( hMsiHiddenWindow );
2197     hMsiHiddenWindow = NULL;
2198     UnregisterClassW( szMsiDialogClass, NULL );
2199     UnregisterClassW( szMsiHiddenWindow, NULL );
2200     uiThreadId = 0;
2201 }