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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, 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 "wingdi.h"
  30 #include "winuser.h"
  31 #include "winnls.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 #include "winreg.h"
  40 #include "shlwapi.h"
  41
  42 #include "wine/debug.h"
  43 #include "wine/unicode.h"
  44
  45 WINE_DEFAULT_DEBUG_CHANNEL(msi);
  46
  47
  48 extern HINSTANCE msi_hInstance;
  49
  50 struct msi_control_tag;
  51 typedef struct msi_control_tag msi_control;
  52 typedef UINT (*msi_handler)( msi_dialog *, msi_control *, WPARAM );
  53
  54 struct msi_control_tag
  55 {
  56     struct list entry;
  57     HWND hwnd;
  58     msi_handler handler;
  59     LPWSTR property;
  60     LPWSTR value;
  61     HBITMAP hBitmap;
  62     HICON hIcon;
  63     LPWSTR tabnext;
  64     LPWSTR type;
  65     HMODULE hDll;
  66     float progress_current;
  67     float progress_max;
  68     DWORD attributes;
  69     WCHAR name[1];
  70 };
  71
  72 typedef struct msi_font_tag
  73 {
  74     struct msi_font_tag *next;
  75     HFONT hfont;
  76     COLORREF color;
  77     WCHAR name[1];
  78 } msi_font;
  79
  80 struct msi_dialog_tag
  81 {
  82     MSIPACKAGE *package;
  83     msi_dialog *parent;
  84     msi_dialog_event_handler event_handler;
  85     BOOL finished;
  86     INT scale;
  87     DWORD attributes;
  88     SIZE size;
  89     HWND hwnd;
  90     LPWSTR default_font;
  91     msi_font *font_list;
  92     struct list controls;
  93     HWND hWndFocus;
  94     LPWSTR control_default;
  95     LPWSTR control_cancel;
  96     WCHAR name[1];
  97 };
  98
  99 typedef UINT (*msi_dialog_control_func)( msi_dialog *dialog, MSIRECORD *rec );
 100 struct control_handler
 101 {
 102     LPCWSTR control_type;
 103     msi_dialog_control_func func;
 104 };
 105
 106 typedef struct
 107 {
 108     msi_dialog* dialog;
 109     msi_control *parent;
 110     DWORD       attributes;
 111     LPWSTR      propval;
 112 } radio_button_group_descr;
 113
 114 static const WCHAR szMsiDialogClass[] = {
 115     'M','s','i','D','i','a','l','o','g','C','l','o','s','e','C','l','a','s','s',0
 116 };
 117 static const WCHAR szMsiHiddenWindow[] = {
 118     'M','s','i','H','i','d','d','e','n','W','i','n','d','o','w',0 };
 119 static const WCHAR szStatic[] = { 'S','t','a','t','i','c',0 };
 120 static const WCHAR szButton[] = { 'B','U','T','T','O','N', 0 };
 121 static const WCHAR szButtonData[] = { 'M','S','I','D','A','T','A',0 };
 122 static const WCHAR szProgress[] = { 'P','r','o','g','r','e','s','s',0 };
 123 static const WCHAR szText[] = { 'T','e','x','t',0 };
 124 static const WCHAR szPushButton[] = { 'P','u','s','h','B','u','t','t','o','n',0 };
 125 static const WCHAR szLine[] = { 'L','i','n','e',0 };
 126 static const WCHAR szBitmap[] = { 'B','i','t','m','a','p',0 };
 127 static const WCHAR szCheckBox[] = { 'C','h','e','c','k','B','o','x',0 };
 128 static const WCHAR szScrollableText[] = {
 129     'S','c','r','o','l','l','a','b','l','e','T','e','x','t',0 };
 130 static const WCHAR szComboBox[] = { 'C','o','m','b','o','B','o','x',0 };
 131 static const WCHAR szEdit[] = { 'E','d','i','t',0 };
 132 static const WCHAR szMaskedEdit[] = { 'M','a','s','k','e','d','E','d','i','t',0 };
 133 static const WCHAR szPathEdit[] = { 'P','a','t','h','E','d','i','t',0 };
 134 static const WCHAR szProgressBar[] = {
 135      'P','r','o','g','r','e','s','s','B','a','r',0 };
 136 static const WCHAR szRadioButtonGroup[] = {
 137     'R','a','d','i','o','B','u','t','t','o','n','G','r','o','u','p',0 };
 138 static const WCHAR szIcon[] = { 'I','c','o','n',0 };
 139 static const WCHAR szSelectionTree[] = {
 140     'S','e','l','e','c','t','i','o','n','T','r','e','e',0 };
 141 static const WCHAR szGroupBox[] = { 'G','r','o','u','p','B','o','x',0 };
 142 static const WCHAR szListBox[] = { 'L','i','s','t','B','o','x',0 };
 143 static const WCHAR szDirectoryCombo[] = { 'D','i','r','e','c','t','o','r','y','C','o','m','b','o',0 };
 144 static const WCHAR szDirectoryList[] = { 'D','i','r','e','c','t','o','r','y','L','i','s','t',0 };
 145 static const WCHAR szVolumeCostList[] = { 'V','o','l','u','m','e','C','o','s','t','L','i','s','t',0 };
 146 static const WCHAR szVolumeSelectCombo[] = { 'V','o','l','u','m','e','S','e','l','e','c','t','C','o','m','b','o',0 };
 147 static const WCHAR szSelectionDescription[] = {'S','e','l','e','c','t','i','o','n','D','e','s','c','r','i','p','t','i','o','n',0};
 148 static const WCHAR szSelectionPath[] = {'S','e','l','e','c','t','i','o','n','P','a','t','h',0};
 149 static const WCHAR szProperty[] = {'P','r','o','p','e','r','t','y',0};
 150
 151 static UINT msi_dialog_checkbox_handler( msi_dialog *, msi_control *, WPARAM );
 152 static void msi_dialog_checkbox_sync_state( msi_dialog *, msi_control * );
 153 static UINT msi_dialog_button_handler( msi_dialog *, msi_control *, WPARAM );
 154 static UINT msi_dialog_edit_handler( msi_dialog *, msi_control *, WPARAM );
 155 static UINT msi_dialog_radiogroup_handler( msi_dialog *, msi_control *, WPARAM );
 156 static UINT msi_dialog_evaluate_control_conditions( msi_dialog *dialog );
 157 static LRESULT WINAPI MSIRadioGroup_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam);
 158 static MSIFEATURE *msi_seltree_get_selected_feature( msi_control *control );
 159
 160 /* dialog sequencing */
 161
 162 #define WM_MSI_DIALOG_CREATE  (WM_USER+0x100)
 163 #define WM_MSI_DIALOG_DESTROY (WM_USER+0x101)
 164
 165 static DWORD uiThreadId;
 166 static HWND hMsiHiddenWindow;
 167
 168 static INT msi_dialog_scale_unit( msi_dialog *dialog, INT val )
 169 {
 170     return (dialog->scale * val + 5) / 10;
 171 }
 172
 173 static msi_control *msi_dialog_find_control( msi_dialog *dialog, LPCWSTR name )
 174 {
 175     msi_control *control;
 176
 177     if( !name )
 178         return NULL;
 179     if( !dialog->hwnd )
 180         return NULL;
 181     LIST_FOR_EACH_ENTRY( control, &dialog->controls, msi_control, entry )
 182         if( !strcmpW( control->name, name ) ) /* FIXME: case sensitive? */
 183             return control;
 184     return NULL;
 185 }
 186
 187 static msi_control *msi_dialog_find_control_by_type( msi_dialog *dialog, LPCWSTR type )
 188 {
 189     msi_control *control;
 190
 191     if( !type )
 192         return NULL;
 193     if( !dialog->hwnd )
 194         return NULL;
 195     LIST_FOR_EACH_ENTRY( control, &dialog->controls, msi_control, entry )
 196         if( !strcmpW( control->type, type ) ) /* FIXME: case sensitive? */
 197             return control;
 198     return NULL;
 199 }
 200
 201 static msi_control *msi_dialog_find_control_by_hwnd( msi_dialog *dialog, HWND hwnd )
 202 {
 203     msi_control *control;
 204
 205     if( !dialog->hwnd )
 206         return NULL;
 207     LIST_FOR_EACH_ENTRY( control, &dialog->controls, msi_control, entry )
 208         if( hwnd == control->hwnd )
 209             return control;
 210     return NULL;
 211 }
 212
 213 static LPWSTR msi_get_deformatted_field( MSIPACKAGE *package, MSIRECORD *rec, int field )
 214 {
 215     LPCWSTR str = MSI_RecordGetString( rec, field );
 216     LPWSTR ret = NULL;
 217
 218     if (str)
 219         deformat_string( package, str, &ret );
 220     return ret;
 221 }
 222
 223 static LPWSTR msi_dialog_dup_property( msi_dialog *dialog, LPCWSTR property, BOOL indirect )
 224 {
 225     LPWSTR prop = NULL;
 226
 227     if (!property)
 228         return NULL;
 229
 230     if (indirect)
 231         prop = msi_dup_property( dialog->package, property );
 232
 233     if (!prop)
 234         prop = strdupW( property );
 235
 236     return prop;
 237 }
 238
 239 msi_dialog *msi_dialog_get_parent( msi_dialog *dialog )
 240 {
 241     return dialog->parent;
 242 }
 243
 244 LPWSTR msi_dialog_get_name( msi_dialog *dialog )
 245 {
 246     return dialog->name;
 247 }
 248
 249 /*
 250  * msi_dialog_get_style
 251  *
 252  * Extract the {\style} string from the front of the text to display and
 253  * update the pointer.  Only the last style in a list is applied.
 254  */
 255 static LPWSTR msi_dialog_get_style( LPCWSTR p, LPCWSTR *rest )
 256 {
 257     LPWSTR ret;
 258     LPCWSTR q, i, first;
 259     DWORD len;
 260
 261     q = NULL;
 262     *rest = p;
 263     if( !p )
 264         return NULL;
 265
 266     while ((first = strchrW( p, '{' )) && (q = strchrW( first + 1, '}' )))
 267     {
 268         p = first + 1;
 269         if( *p != '\\' && *p != '&' )
 270             return NULL;
 271
 272         /* little bit of sanity checking to stop us getting confused with RTF */
 273         for( i=++p; i<q; i++ )
 274             if( *i == '}' || *i == '\\' )
 275                 return NULL;
 276     }
 277
 278     if (!p || !q)
 279         return NULL;
 280
 281     *rest = ++q;
 282     len = q - p;
 283
 284     ret = msi_alloc( len*sizeof(WCHAR) );
 285     if( !ret )
 286         return ret;
 287     memcpy( ret, p, len*sizeof(WCHAR) );
 288     ret[len-1] = 0;
 289     return ret;
 290 }
 291
 292 static UINT msi_dialog_add_font( MSIRECORD *rec, LPVOID param )
 293 {
 294     msi_dialog *dialog = param;
 295     msi_font *font;
 296     LPCWSTR face, name;
 297     LOGFONTW lf;
 298     INT style;
 299     HDC hdc;
 300
 301     /* create a font and add it to the list */
 302     name = MSI_RecordGetString( rec, 1 );
 303     font = msi_alloc( sizeof *font + strlenW( name )*sizeof (WCHAR) );
 304     strcpyW( font->name, name );
 305     font->next = dialog->font_list;
 306     dialog->font_list = font;
 307
 308     font->color = MSI_RecordGetInteger( rec, 4 );
 309
 310     memset( &lf, 0, sizeof lf );
 311     face = MSI_RecordGetString( rec, 2 );
 312     lf.lfHeight = MSI_RecordGetInteger( rec, 3 );
 313     style = MSI_RecordGetInteger( rec, 5 );
 314     if( style & msidbTextStyleStyleBitsBold )
 315         lf.lfWeight = FW_BOLD;
 316     if( style & msidbTextStyleStyleBitsItalic )
 317         lf.lfItalic = TRUE;
 318     if( style & msidbTextStyleStyleBitsUnderline )
 319         lf.lfUnderline = TRUE;
 320     if( style & msidbTextStyleStyleBitsStrike )
 321         lf.lfStrikeOut = TRUE;
 322     lstrcpynW( lf.lfFaceName, face, LF_FACESIZE );
 323
 324     /* adjust the height */
 325     hdc = GetDC( dialog->hwnd );
 326     if (hdc)
 327     {
 328         lf.lfHeight = -MulDiv(lf.lfHeight, GetDeviceCaps(hdc, LOGPIXELSY), 72);
 329         ReleaseDC( dialog->hwnd, hdc );
 330     }
 331
 332     font->hfont = CreateFontIndirectW( &lf );
 333
 334     TRACE("Adding font style %s\n", debugstr_w(font->name) );
 335
 336     return ERROR_SUCCESS;
 337 }
 338
 339 static msi_font *msi_dialog_find_font( msi_dialog *dialog, LPCWSTR name )
 340 {
 341     msi_font *font;
 342
 343     for( font = dialog->font_list; font; font = font->next )
 344         if( !strcmpW( font->name, name ) )  /* FIXME: case sensitive? */
 345             break;
 346
 347     return font;
 348 }
 349
 350 static UINT msi_dialog_set_font( msi_dialog *dialog, HWND hwnd, LPCWSTR name )
 351 {
 352     msi_font *font;
 353
 354     font = msi_dialog_find_font( dialog, name );
 355     if( font )
 356         SendMessageW( hwnd, WM_SETFONT, (WPARAM) font->hfont, TRUE );
 357     else
 358         ERR("No font entry for %s\n", debugstr_w(name));
 359     return ERROR_SUCCESS;
 360 }
 361
 362 static UINT msi_dialog_build_font_list( msi_dialog *dialog )
 363 {
 364     static const WCHAR query[] = {
 365       'S','E','L','E','C','T',' ','*',' ',
 366       'F','R','O','M',' ','`','T','e','x','t','S','t','y','l','e','`',' ',0
 367     };
 368     UINT r;
 369     MSIQUERY *view = NULL;
 370
 371     TRACE("dialog %p\n", dialog );
 372
 373     r = MSI_OpenQuery( dialog->package->db, &view, query );
 374     if( r != ERROR_SUCCESS )
 375         return r;
 376
 377     r = MSI_IterateRecords( view, NULL, msi_dialog_add_font, dialog );
 378     msiobj_release( &view->hdr );
 379
 380     return r;
 381 }
 382
 383 static void msi_destroy_control( msi_control *t )
 384 {
 385     list_remove( &t->entry );
 386     /* leave dialog->hwnd - destroying parent destroys child windows */
 387     msi_free( t->property );
 388     msi_free( t->value );
 389     if( t->hBitmap )
 390         DeleteObject( t->hBitmap );
 391     if( t->hIcon )
 392         DestroyIcon( t->hIcon );
 393     msi_free( t->tabnext );
 394     msi_free( t->type );
 395     if (t->hDll)
 396         FreeLibrary( t->hDll );
 397     msi_free( t );
 398 }
 399
 400 static msi_control *msi_dialog_create_window( msi_dialog *dialog,
 401                 MSIRECORD *rec, DWORD exstyle, LPCWSTR szCls, LPCWSTR name, LPCWSTR text,
 402                 DWORD style, HWND parent )
 403 {
 404     DWORD x, y, width, height;
 405     LPWSTR font = NULL, title_font = NULL;
 406     LPCWSTR title = NULL;
 407     msi_control *control;
 408
 409     style |= WS_CHILD;
 410
 411     control = msi_alloc( sizeof *control + strlenW(name)*sizeof(WCHAR) );
 412     if (!control)
 413         return NULL;
 414
 415     strcpyW( control->name, name );
 416     list_add_head( &dialog->controls, &control->entry );
 417     control->handler = NULL;
 418     control->property = NULL;
 419     control->value = NULL;
 420     control->hBitmap = NULL;
 421     control->hIcon = NULL;
 422     control->hDll = NULL;
 423     control->tabnext = strdupW( MSI_RecordGetString( rec, 11) );
 424     control->type = strdupW( MSI_RecordGetString( rec, 3 ) );
 425     control->progress_current = 0;
 426     control->progress_max = 100;
 427
 428     x = MSI_RecordGetInteger( rec, 4 );
 429     y = MSI_RecordGetInteger( rec, 5 );
 430     width = MSI_RecordGetInteger( rec, 6 );
 431     height = MSI_RecordGetInteger( rec, 7 );
 432
 433     x = msi_dialog_scale_unit( dialog, x );
 434     y = msi_dialog_scale_unit( dialog, y );
 435     width = msi_dialog_scale_unit( dialog, width );
 436     height = msi_dialog_scale_unit( dialog, height );
 437
 438     if( text )
 439     {
 440         deformat_string( dialog->package, text, &title_font );
 441         font = msi_dialog_get_style( title_font, &title );
 442     }
 443
 444     control->hwnd = CreateWindowExW( exstyle, szCls, title, style,
 445                           x, y, width, height, parent, NULL, NULL, NULL );
 446
 447     TRACE("Dialog %s control %s hwnd %p\n",
 448            debugstr_w(dialog->name), debugstr_w(text), control->hwnd );
 449
 450     msi_dialog_set_font( dialog, control->hwnd,
 451                          font ? font : dialog->default_font );
 452
 453     msi_free( title_font );
 454     msi_free( font );
 455
 456     return control;
 457 }
 458
 459 static LPWSTR msi_dialog_get_uitext( msi_dialog *dialog, LPCWSTR key )
 460 {
 461     MSIRECORD *rec;
 462     LPWSTR text;
 463
 464     static const WCHAR query[] = {
 465         's','e','l','e','c','t',' ','*',' ',
 466         'f','r','o','m',' ','`','U','I','T','e','x','t','`',' ',
 467         'w','h','e','r','e',' ','`','K','e','y','`',' ','=',' ','\'','%','s','\'',0
 468     };
 469
 470     rec = MSI_QueryGetRecord( dialog->package->db, query, key );
 471     if (!rec) return NULL;
 472     text = strdupW( MSI_RecordGetString( rec, 2 ) );
 473     msiobj_release( &rec->hdr );
 474     return text;
 475 }
 476
 477 static MSIRECORD *msi_get_binary_record( MSIDATABASE *db, LPCWSTR name )
 478 {
 479     static const WCHAR query[] = {
 480         's','e','l','e','c','t',' ','*',' ',
 481         'f','r','o','m',' ','B','i','n','a','r','y',' ',
 482         'w','h','e','r','e',' ',
 483             '`','N','a','m','e','`',' ','=',' ','\'','%','s','\'',0
 484     };
 485
 486     return MSI_QueryGetRecord( db, query, name );
 487 }
 488
 489 static LPWSTR msi_create_tmp_path(void)
 490 {
 491     WCHAR tmp[MAX_PATH];
 492     LPWSTR path = NULL;
 493     static const WCHAR prefix[] = { 'm','s','i',0 };
 494     DWORD len, r;
 495
 496     r = GetTempPathW( MAX_PATH, tmp );
 497     if( !r )
 498         return path;
 499     len = lstrlenW( tmp ) + 20;
 500     path = msi_alloc( len * sizeof (WCHAR) );
 501     if( path )
 502     {
 503         r = GetTempFileNameW( tmp, prefix, 0, path );
 504         if (!r)
 505         {
 506             msi_free( path );
 507             path = NULL;
 508         }
 509     }
 510     return path;
 511 }
 512
 513
 514 static HANDLE msi_load_image( MSIDATABASE *db, LPCWSTR name, UINT type,
 515                               UINT cx, UINT cy, UINT flags )
 516 {
 517     MSIRECORD *rec = NULL;
 518     HANDLE himage = NULL;
 519     LPWSTR tmp;
 520     UINT r;
 521
 522     TRACE("%p %s %u %u %08x\n", db, debugstr_w(name), cx, cy, flags);
 523
 524     tmp = msi_create_tmp_path();
 525     if( !tmp )
 526         return himage;
 527
 528     rec = msi_get_binary_record( db, name );
 529     if( rec )
 530     {
 531         r = MSI_RecordStreamToFile( rec, 2, tmp );
 532         if( r == ERROR_SUCCESS )
 533         {
 534             himage = LoadImageW( 0, tmp, type, cx, cy, flags );
 535         }
 536         msiobj_release( &rec->hdr );
 537     }
 538     DeleteFileW( tmp );
 539
 540     msi_free( tmp );
 541     return himage;
 542 }
 543
 544 static HICON msi_load_icon( MSIDATABASE *db, LPCWSTR text, UINT attributes )
 545 {
 546     DWORD cx = 0, cy = 0, flags;
 547
 548     flags = LR_LOADFROMFILE | LR_DEFAULTSIZE;
 549     if( attributes & msidbControlAttributesFixedSize )
 550     {
 551         flags &= ~LR_DEFAULTSIZE;
 552         if( attributes & msidbControlAttributesIconSize16 )
 553         {
 554             cx += 16;
 555             cy += 16;
 556         }
 557         if( attributes & msidbControlAttributesIconSize32 )
 558         {
 559             cx += 32;
 560             cy += 32;
 561         }
 562         /* msidbControlAttributesIconSize48 handled by above logic */
 563     }
 564     return msi_load_image( db, text, IMAGE_ICON, cx, cy, flags );
 565 }
 566
 567
 568 /* called from the Control Event subscription code */
 569 void msi_dialog_handle_event( msi_dialog* dialog, LPCWSTR control,
 570                               LPCWSTR attribute, MSIRECORD *rec )
 571 {
 572     msi_control* ctrl;
 573     LPCWSTR font_text, text = NULL;
 574     LPWSTR font;
 575
 576     static const WCHAR empty[] = {0};
 577
 578     ctrl = msi_dialog_find_control( dialog, control );
 579     if (!ctrl)
 580         return;
 581     if( !lstrcmpW(attribute, szText) )
 582     {
 583         font_text = MSI_RecordGetString( rec , 1 );
 584         font = msi_dialog_get_style( font_text, &text );
 585         if (!text) text = empty;
 586         SetWindowTextW( ctrl->hwnd, text );
 587         msi_free( font );
 588         msi_dialog_check_messages( NULL );
 589     }
 590     else if( !lstrcmpW(attribute, szProgress) )
 591     {
 592         DWORD func, val;
 593
 594         func = MSI_RecordGetInteger( rec , 1 );
 595         val = MSI_RecordGetInteger( rec , 2 );
 596
 597         switch (func)
 598         {
 599         case 0: /* init */
 600             ctrl->progress_max = val;
 601             ctrl->progress_current = 0;
 602             SendMessageW(ctrl->hwnd, PBM_SETRANGE, 0, MAKELPARAM(0,100));
 603             SendMessageW(ctrl->hwnd, PBM_SETPOS, 0, 0);
 604             break;
 605         case 1: /* FIXME: not sure what this is supposed to do */
 606             break;
 607         case 2: /* move */
 608             ctrl->progress_current += val;
 609             SendMessageW(ctrl->hwnd, PBM_SETPOS, 100*(ctrl->progress_current/ctrl->progress_max), 0);
 610             break;
 611         default:
 612             ERR("Unknown progress message %d\n", func);
 613             break;
 614         }
 615     }
 616     else if ( !lstrcmpW(attribute, szProperty) )
 617     {
 618         MSIFEATURE *feature = msi_seltree_get_selected_feature( ctrl );
 619         MSI_SetPropertyW( dialog->package, ctrl->property, feature->Directory );
 620     }
 621     else if ( !lstrcmpW(attribute, szSelectionPath) )
 622     {
 623         LPWSTR prop = msi_dialog_dup_property( dialog, ctrl->property, TRUE );
 624         LPWSTR path;
 625         if (!prop) return;
 626         path = msi_dup_property( dialog->package, prop );
 627         SetWindowTextW( ctrl->hwnd, path );
 628         msi_free(prop);
 629         msi_free(path);
 630     }
 631     else
 632     {
 633         FIXME("Attribute %s not being set\n", debugstr_w(attribute));
 634         return;
 635     }
 636 }
 637
 638 static void msi_dialog_map_events(msi_dialog* dialog, LPCWSTR control)
 639 {
 640     static const WCHAR Query[] = {
 641         'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
 642          '`','E','v','e','n','t','M','a','p','p','i','n','g','`',' ',
 643         'W','H','E','R','E',' ',
 644          '`','D','i','a','l','o','g','_','`',' ','=',' ','\'','%','s','\'',' ',
 645         'A','N','D',' ',
 646          '`','C','o','n','t','r','o','l','_','`',' ','=',' ','\'','%','s','\'',0
 647     };
 648     MSIRECORD *row;
 649     LPCWSTR event, attribute;
 650
 651     row = MSI_QueryGetRecord( dialog->package->db, Query, dialog->name, control );
 652     if (!row)
 653         return;
 654
 655     event = MSI_RecordGetString( row, 3 );
 656     attribute = MSI_RecordGetString( row, 4 );
 657     ControlEvent_SubscribeToEvent( dialog->package, dialog, event, control, attribute );
 658     msiobj_release( &row->hdr );
 659 }
 660
 661 /* everything except radio buttons */
 662 static msi_control *msi_dialog_add_control( msi_dialog *dialog,
 663                 MSIRECORD *rec, LPCWSTR szCls, DWORD style )
 664 {
 665     DWORD attributes;
 666     LPCWSTR text, name;
 667     DWORD exstyle = 0;
 668
 669     name = MSI_RecordGetString( rec, 2 );
 670     attributes = MSI_RecordGetInteger( rec, 8 );
 671     text = MSI_RecordGetString( rec, 10 );
 672     if( attributes & msidbControlAttributesVisible )
 673         style |= WS_VISIBLE;
 674     if( ~attributes & msidbControlAttributesEnabled )
 675         style |= WS_DISABLED;
 676     if( attributes & msidbControlAttributesSunken )
 677         exstyle |= WS_EX_CLIENTEDGE;
 678
 679     msi_dialog_map_events(dialog, name);
 680
 681     return msi_dialog_create_window( dialog, rec, exstyle, szCls, name,
 682                                      text, style, dialog->hwnd );
 683 }
 684
 685 struct msi_text_info
 686 {
 687     msi_font *font;
 688     WNDPROC oldproc;
 689     DWORD attributes;
 690 };
 691
 692 /*
 693  * we don't erase our own background,
 694  * so we have to make sure that the parent window redraws first
 695  */
 696 static void msi_text_on_settext( HWND hWnd )
 697 {
 698     HWND hParent;
 699     RECT rc;
 700
 701     hParent = GetParent( hWnd );
 702     GetWindowRect( hWnd, &rc );
 703     MapWindowPoints( NULL, hParent, (LPPOINT) &rc, 2 );
 704     InvalidateRect( hParent, &rc, TRUE );
 705 }
 706
 707 static LRESULT WINAPI
 708 MSIText_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
 709 {
 710     struct msi_text_info *info;
 711     LRESULT r = 0;
 712
 713     TRACE("%p %04x %08lx %08lx\n", hWnd, msg, wParam, lParam);
 714
 715     info = GetPropW(hWnd, szButtonData);
 716
 717     if ( info->font )
 718         SetTextColor( (HDC)wParam, info->font->color );
 719
 720     if( msg == WM_CTLCOLORSTATIC &&
 721        ( info->attributes & msidbControlAttributesTransparent ) )
 722     {
 723         SetBkMode( (HDC)wParam, TRANSPARENT );
 724         return (LRESULT) GetStockObject(NULL_BRUSH);
 725     }
 726
 727     r = CallWindowProcW(info->oldproc, hWnd, msg, wParam, lParam);
 728
 729     switch( msg )
 730     {
 731     case WM_SETTEXT:
 732         msi_text_on_settext( hWnd );
 733         break;
 734     case WM_NCDESTROY:
 735         msi_free( info );
 736         RemovePropW( hWnd, szButtonData );
 737         break;
 738     }
 739
 740     return r;
 741 }
 742
 743 static UINT msi_dialog_text_control( msi_dialog *dialog, MSIRECORD *rec )
 744 {
 745     msi_control *control;
 746     struct msi_text_info *info;
 747     LPCWSTR text, ptr, prop;
 748     LPWSTR font_name;
 749
 750     TRACE("%p %p\n", dialog, rec);
 751
 752     control = msi_dialog_add_control( dialog, rec, szStatic, SS_LEFT | WS_GROUP );
 753     if( !control )
 754         return ERROR_FUNCTION_FAILED;
 755
 756     info = msi_alloc( sizeof *info );
 757     if( !info )
 758         return ERROR_SUCCESS;
 759
 760     control->attributes = MSI_RecordGetInteger( rec, 8 );
 761     prop = MSI_RecordGetString( rec, 9 );
 762     control->property = msi_dialog_dup_property( dialog, prop, FALSE );
 763
 764     text = MSI_RecordGetString( rec, 10 );
 765     font_name = msi_dialog_get_style( text, &ptr );
 766     info->font = ( font_name ) ? msi_dialog_find_font( dialog, font_name ) : NULL;
 767     msi_free( font_name );
 768
 769     info->attributes = MSI_RecordGetInteger( rec, 8 );
 770     if( info->attributes & msidbControlAttributesTransparent )
 771         SetWindowLongPtrW( control->hwnd, GWL_EXSTYLE, WS_EX_TRANSPARENT );
 772
 773     info->oldproc = (WNDPROC) SetWindowLongPtrW( control->hwnd, GWLP_WNDPROC,
 774                                           (LONG_PTR)MSIText_WndProc );
 775     SetPropW( control->hwnd, szButtonData, info );
 776
 777     ControlEvent_SubscribeToEvent( dialog->package, dialog,
 778                                    szSelectionPath, control->name, szSelectionPath );
 779
 780     return ERROR_SUCCESS;
 781 }
 782
 783 static UINT msi_dialog_button_control( msi_dialog *dialog, MSIRECORD *rec )
 784 {
 785     msi_control *control;
 786     UINT attributes, style;
 787     LPWSTR text;
 788
 789     TRACE("%p %p\n", dialog, rec);
 790
 791     style = WS_TABSTOP;
 792     attributes = MSI_RecordGetInteger( rec, 8 );
 793     if( attributes & msidbControlAttributesIcon )
 794         style |= BS_ICON;
 795
 796     control = msi_dialog_add_control( dialog, rec, szButton, style );
 797     if( !control )
 798         return ERROR_FUNCTION_FAILED;
 799
 800     control->handler = msi_dialog_button_handler;
 801
 802     /* set the icon */
 803     text = msi_get_deformatted_field( dialog->package, rec, 10 );
 804     control->hIcon = msi_load_icon( dialog->package->db, text, attributes );
 805     if( attributes & msidbControlAttributesIcon )
 806         SendMessageW( control->hwnd, BM_SETIMAGE, IMAGE_ICON, (LPARAM) control->hIcon );
 807     msi_free( text );
 808
 809     return ERROR_SUCCESS;
 810 }
 811
 812 static LPWSTR msi_get_checkbox_value( msi_dialog *dialog, LPCWSTR prop )
 813 {
 814     static const WCHAR query[] = {
 815         'S','E','L','E','C','T',' ','*',' ',
 816         'F','R','O','M',' ','`','C','h','e','c','k','B','o','x','`',' ',
 817         'W','H','E','R','E',' ',
 818         '`','P','r','o','p','e','r','t','y','`',' ','=',' ',
 819         '\'','%','s','\'',0
 820     };
 821     MSIRECORD *rec = NULL;
 822     LPWSTR ret = NULL;
 823
 824     /* find if there is a value associated with the checkbox */
 825     rec = MSI_QueryGetRecord( dialog->package->db, query, prop );
 826     if (!rec)
 827         return ret;
 828
 829     ret = msi_get_deformatted_field( dialog->package, rec, 2 );
 830     if( ret && !ret[0] )
 831     {
 832         msi_free( ret );
 833         ret = NULL;
 834     }
 835     msiobj_release( &rec->hdr );
 836     if (ret)
 837         return ret;
 838
 839     ret = msi_dup_property( dialog->package, prop );
 840     if( ret && !ret[0] )
 841     {
 842         msi_free( ret );
 843         ret = NULL;
 844     }
 845
 846     return ret;
 847 }
 848
 849 static UINT msi_dialog_checkbox_control( msi_dialog *dialog, MSIRECORD *rec )
 850 {
 851     msi_control *control;
 852     LPCWSTR prop;
 853
 854     TRACE("%p %p\n", dialog, rec);
 855
 856     control = msi_dialog_add_control( dialog, rec, szButton,
 857                                 BS_CHECKBOX | BS_MULTILINE | WS_TABSTOP );
 858     control->handler = msi_dialog_checkbox_handler;
 859     prop = MSI_RecordGetString( rec, 9 );
 860     if( prop )
 861     {
 862         control->property = strdupW( prop );
 863         control->value = msi_get_checkbox_value( dialog, prop );
 864         TRACE("control %s value %s\n", debugstr_w(control->property),
 865               debugstr_w(control->value));
 866     }
 867     msi_dialog_checkbox_sync_state( dialog, control );
 868
 869     return ERROR_SUCCESS;
 870 }
 871
 872 static UINT msi_dialog_line_control( msi_dialog *dialog, MSIRECORD *rec )
 873 {
 874     TRACE("%p %p\n", dialog, rec);
 875
 876     /* line is exactly 2 units in height */
 877     MSI_RecordSetInteger( rec, 7, 2 );
 878
 879     msi_dialog_add_control( dialog, rec, szStatic, SS_ETCHEDHORZ | SS_SUNKEN );
 880     return ERROR_SUCCESS;
 881 }
 882
 883 /******************** Scroll Text ********************************************/
 884
 885 struct msi_scrolltext_info
 886 {
 887     msi_dialog *dialog;
 888     msi_control *control;
 889     WNDPROC oldproc;
 890 };
 891
 892 static LRESULT WINAPI
 893 MSIScrollText_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
 894 {
 895     struct msi_scrolltext_info *info;
 896     HRESULT r;
 897
 898     TRACE("%p %04x %08lx %08lx\n", hWnd, msg, wParam, lParam);
 899
 900     info = GetPropW( hWnd, szButtonData );
 901
 902     r = CallWindowProcW( info->oldproc, hWnd, msg, wParam, lParam );
 903
 904     switch( msg )
 905     {
 906     case WM_NCDESTROY:
 907         msi_free( info );
 908         RemovePropW( hWnd, szButtonData );
 909         break;
 910     case WM_PAINT:
 911         /* native MSI sets a wait cursor here */
 912         msi_dialog_button_handler( info->dialog, info->control, BN_CLICKED );
 913         break;
 914     }
 915     return r;
 916 }
 917
 918 struct msi_streamin_info
 919 {
 920     LPSTR string;
 921     DWORD offset;
 922     DWORD length;
 923 };
 924
 925 static DWORD CALLBACK
 926 msi_richedit_stream_in( DWORD_PTR arg, LPBYTE buffer, LONG count, LONG *pcb )
 927 {
 928     struct msi_streamin_info *info = (struct msi_streamin_info*) arg;
 929
 930     if( (count + info->offset) > info->length )
 931         count = info->length - info->offset;
 932     memcpy( buffer, &info->string[ info->offset ], count );
 933     *pcb = count;
 934     info->offset += count;
 935
 936     TRACE("%d/%d\n", info->offset, info->length);
 937
 938     return 0;
 939 }
 940
 941 static void msi_scrolltext_add_text( msi_control *control, LPCWSTR text )
 942 {
 943     struct msi_streamin_info info;
 944     EDITSTREAM es;
 945
 946     info.string = strdupWtoA( text );
 947     info.offset = 0;
 948     info.length = lstrlenA( info.string ) + 1;
 949
 950     es.dwCookie = (DWORD_PTR) &info;
 951     es.dwError = 0;
 952     es.pfnCallback = msi_richedit_stream_in;
 953
 954     SendMessageW( control->hwnd, EM_STREAMIN, SF_RTF, (LPARAM) &es );
 955
 956     msi_free( info.string );
 957 }
 958
 959 static UINT msi_dialog_scrolltext_control( msi_dialog *dialog, MSIRECORD *rec )
 960 {
 961     static const WCHAR szRichEdit20W[] = {
 962         'R','i','c','h','E','d','i','t','2','0','W',0
 963     };
 964     struct msi_scrolltext_info *info;
 965     msi_control *control;
 966     HMODULE hRichedit;
 967     LPCWSTR text;
 968     DWORD style;
 969
 970     info = msi_alloc( sizeof *info );
 971     if (!info)
 972         return ERROR_FUNCTION_FAILED;
 973
 974     hRichedit = LoadLibraryA("riched20");
 975
 976     style = WS_BORDER | ES_MULTILINE | WS_VSCROLL |
 977             ES_READONLY | ES_AUTOVSCROLL | WS_TABSTOP;
 978     control = msi_dialog_add_control( dialog, rec, szRichEdit20W, style );
 979     if (!control)
 980     {
 981         FreeLibrary( hRichedit );
 982         msi_free( info );
 983         return ERROR_FUNCTION_FAILED;
 984     }
 985
 986     control->hDll = hRichedit;
 987
 988     info->dialog = dialog;
 989     info->control = control;
 990
 991     /* subclass the static control */
 992     info->oldproc = (WNDPROC) SetWindowLongPtrW( control->hwnd, GWLP_WNDPROC,
 993                                           (LONG_PTR)MSIScrollText_WndProc );
 994     SetPropW( control->hwnd, szButtonData, info );
 995
 996     /* add the text into the richedit */
 997     text = MSI_RecordGetString( rec, 10 );
 998     if (text)
 999         msi_scrolltext_add_text( control, text );
1000
1001     return ERROR_SUCCESS;
1002 }
1003
1004 static HBITMAP msi_load_picture( MSIDATABASE *db, LPCWSTR name,
1005                                  INT cx, INT cy, DWORD flags )
1006 {
1007     HBITMAP hOleBitmap = 0, hBitmap = 0, hOldSrcBitmap, hOldDestBitmap;
1008     MSIRECORD *rec = NULL;
1009     IStream *stm = NULL;
1010     IPicture *pic = NULL;
1011     HDC srcdc, destdc;
1012     BITMAP bm;
1013     UINT r;
1014
1015     rec = msi_get_binary_record( db, name );
1016     if( !rec )
1017         goto end;
1018
1019     r = MSI_RecordGetIStream( rec, 2, &stm );
1020     msiobj_release( &rec->hdr );
1021     if( r != ERROR_SUCCESS )
1022         goto end;
1023
1024     r = OleLoadPicture( stm, 0, TRUE, &IID_IPicture, (LPVOID*) &pic );
1025     IStream_Release( stm );
1026     if( FAILED( r ) )
1027     {
1028         ERR("failed to load picture\n");
1029         goto end;
1030     }
1031
1032     r = IPicture_get_Handle( pic, (OLE_HANDLE*) &hOleBitmap );
1033     if( FAILED( r ) )
1034     {
1035         ERR("failed to get bitmap handle\n");
1036         goto end;
1037     }
1038
1039     /* make the bitmap the desired size */
1040     r = GetObjectW( hOleBitmap, sizeof bm, &bm );
1041     if (r != sizeof bm )
1042     {
1043         ERR("failed to get bitmap size\n");
1044         goto end;
1045     }
1046
1047     if (flags & LR_DEFAULTSIZE)
1048     {
1049         cx = bm.bmWidth;
1050         cy = bm.bmHeight;
1051     }
1052
1053     srcdc = CreateCompatibleDC( NULL );
1054     hOldSrcBitmap = SelectObject( srcdc, hOleBitmap );
1055     destdc = CreateCompatibleDC( NULL );
1056     hBitmap = CreateCompatibleBitmap( srcdc, cx, cy );
1057     hOldDestBitmap = SelectObject( destdc, hBitmap );
1058     StretchBlt( destdc, 0, 0, cx, cy,
1059                 srcdc, 0, 0, bm.bmWidth, bm.bmHeight, SRCCOPY);
1060     SelectObject( srcdc, hOldSrcBitmap );
1061     SelectObject( destdc, hOldDestBitmap );
1062     DeleteDC( srcdc );
1063     DeleteDC( destdc );
1064
1065 end:
1066     if ( pic )
1067         IPicture_Release( pic );
1068     return hBitmap;
1069 }
1070
1071 static UINT msi_dialog_bitmap_control( msi_dialog *dialog, MSIRECORD *rec )
1072 {
1073     UINT cx, cy, flags, style, attributes;
1074     msi_control *control;
1075     LPWSTR text;
1076
1077     flags = LR_LOADFROMFILE;
1078     style = SS_BITMAP | SS_LEFT | WS_GROUP;
1079
1080     attributes = MSI_RecordGetInteger( rec, 8 );
1081     if( attributes & msidbControlAttributesFixedSize )
1082     {
1083         flags |= LR_DEFAULTSIZE;
1084         style |= SS_CENTERIMAGE;
1085     }
1086
1087     control = msi_dialog_add_control( dialog, rec, szStatic, style );
1088     cx = MSI_RecordGetInteger( rec, 6 );
1089     cy = MSI_RecordGetInteger( rec, 7 );
1090     cx = msi_dialog_scale_unit( dialog, cx );
1091     cy = msi_dialog_scale_unit( dialog, cy );
1092
1093     text = msi_get_deformatted_field( dialog->package, rec, 10 );
1094     control->hBitmap = msi_load_picture( dialog->package->db, text, cx, cy, flags );
1095     if( control->hBitmap )
1096         SendMessageW( control->hwnd, STM_SETIMAGE,
1097                       IMAGE_BITMAP, (LPARAM) control->hBitmap );
1098     else
1099         ERR("Failed to load bitmap %s\n", debugstr_w(text));
1100
1101     msi_free( text );
1102
1103     return ERROR_SUCCESS;
1104 }
1105
1106 static UINT msi_dialog_icon_control( msi_dialog *dialog, MSIRECORD *rec )
1107 {
1108     msi_control *control;
1109     DWORD attributes;
1110     LPWSTR text;
1111
1112     TRACE("\n");
1113
1114     control = msi_dialog_add_control( dialog, rec, szStatic,
1115                             SS_ICON | SS_CENTERIMAGE | WS_GROUP );
1116
1117     attributes = MSI_RecordGetInteger( rec, 8 );
1118     text = msi_get_deformatted_field( dialog->package, rec, 10 );
1119     control->hIcon = msi_load_icon( dialog->package->db, text, attributes );
1120     if( control->hIcon )
1121         SendMessageW( control->hwnd, STM_SETICON, (WPARAM) control->hIcon, 0 );
1122     else
1123         ERR("Failed to load bitmap %s\n", debugstr_w(text));
1124     msi_free( text );
1125     return ERROR_SUCCESS;
1126 }
1127
1128 static UINT msi_dialog_combo_control( msi_dialog *dialog, MSIRECORD *rec )
1129 {
1130     static const WCHAR szCombo[] = { 'C','O','M','B','O','B','O','X',0 };
1131     DWORD attributes, style;
1132
1133     style = CBS_AUTOHSCROLL | WS_TABSTOP | WS_GROUP | WS_CHILD;
1134     attributes = MSI_RecordGetInteger( rec, 8 );
1135     if ( ~attributes & msidbControlAttributesSorted)
1136         style |= CBS_SORT;
1137     if ( attributes & msidbControlAttributesComboList)
1138         style |= CBS_DROPDOWNLIST;
1139     else
1140         style |= CBS_DROPDOWN;
1141
1142     msi_dialog_add_control( dialog, rec, szCombo, style );
1143     return ERROR_SUCCESS;
1144 }
1145
1146 static UINT msi_dialog_edit_control( msi_dialog *dialog, MSIRECORD *rec )
1147 {
1148     msi_control *control;
1149     LPCWSTR prop, text;
1150     LPWSTR val, begin, end;
1151     WCHAR num[10];
1152     DWORD limit;
1153
1154     control = msi_dialog_add_control( dialog, rec, szEdit,
1155                                       WS_BORDER | WS_TABSTOP | ES_AUTOHSCROLL );
1156     control->handler = msi_dialog_edit_handler;
1157
1158     text = MSI_RecordGetString( rec, 10 );
1159     if ( text )
1160     {
1161         begin = strchrW( text, '{' );
1162         end = strchrW( text, '}' );
1163
1164         if ( begin && end && end > begin )
1165         {
1166             lstrcpynW( num, begin + 1, end - begin );
1167             limit = atolW( num );
1168
1169             SendMessageW( control->hwnd, EM_SETLIMITTEXT, limit, 0 );
1170         }
1171     }
1172
1173     prop = MSI_RecordGetString( rec, 9 );
1174     if( prop )
1175         control->property = strdupW( prop );
1176
1177     val = msi_dup_property( dialog->package, control->property );
1178     SetWindowTextW( control->hwnd, val );
1179     msi_free( val );
1180     return ERROR_SUCCESS;
1181 }
1182
1183 /******************** Masked Edit ********************************************/
1184
1185 #define MASK_MAX_GROUPS 20
1186
1187 struct msi_mask_group
1188 {
1189     UINT len;
1190     UINT ofs;
1191     WCHAR type;
1192     HWND hwnd;
1193 };
1194
1195 struct msi_maskedit_info
1196 {
1197     msi_dialog *dialog;
1198     WNDPROC oldproc;
1199     HWND hwnd;
1200     LPWSTR prop;
1201     UINT num_chars;
1202     UINT num_groups;
1203     struct msi_mask_group group[MASK_MAX_GROUPS];
1204 };
1205
1206 static BOOL msi_mask_editable( WCHAR type )
1207 {
1208     switch (type)
1209     {
1210     case '%':
1211     case '#':
1212     case '&':
1213     case '`':
1214     case '?':
1215     case '^':
1216         return TRUE;
1217     }
1218     return FALSE;
1219 }
1220
1221 static void msi_mask_control_change( struct msi_maskedit_info *info )
1222 {
1223     LPWSTR val;
1224     UINT i, n, r;
1225
1226     val = msi_alloc( (info->num_chars+1)*sizeof(WCHAR) );
1227     for( i=0, n=0; i<info->num_groups; i++ )
1228     {
1229         if( (info->group[i].len + n) > info->num_chars )
1230         {
1231             ERR("can't fit control %d text into template\n",i);
1232             break;
1233         }
1234         if (!msi_mask_editable(info->group[i].type))
1235         {
1236             for(r=0; r<info->group[i].len; r++)
1237                 val[n+r] = info->group[i].type;
1238             val[n+r] = 0;
1239         }
1240         else
1241         {
1242             r = GetWindowTextW( info->group[i].hwnd, &val[n], info->group[i].len+1 );
1243             if( r != info->group[i].len )
1244                 break;
1245         }
1246         n += r;
1247     }
1248
1249     TRACE("%d/%d controls were good\n", i, info->num_groups);
1250
1251     if( i == info->num_groups )
1252     {
1253         TRACE("Set property %s to %s\n",
1254               debugstr_w(info->prop), debugstr_w(val) );
1255         CharUpperBuffW( val, info->num_chars );
1256         MSI_SetPropertyW( info->dialog->package, info->prop, val );
1257         msi_dialog_evaluate_control_conditions( info->dialog );
1258     }
1259     msi_free( val );
1260 }
1261
1262 /* now move to the next control if necessary */
1263 static VOID msi_mask_next_control( struct msi_maskedit_info *info, HWND hWnd )
1264 {
1265     HWND hWndNext;
1266     UINT len, i;
1267
1268     for( i=0; i<info->num_groups; i++ )
1269         if( info->group[i].hwnd == hWnd )
1270             break;
1271
1272     /* don't move from the last control */
1273     if( i >= (info->num_groups-1) )
1274         return;
1275
1276     len = SendMessageW( hWnd, WM_GETTEXTLENGTH, 0, 0 );
1277     if( len < info->group[i].len )
1278         return;
1279
1280     hWndNext = GetNextDlgTabItem( GetParent( hWnd ), hWnd, FALSE );
1281     SetFocus( hWndNext );
1282 }
1283
1284 static LRESULT WINAPI
1285 MSIMaskedEdit_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
1286 {
1287     struct msi_maskedit_info *info;
1288     HRESULT r;
1289
1290     TRACE("%p %04x %08lx %08lx\n", hWnd, msg, wParam, lParam);
1291
1292     info = GetPropW(hWnd, szButtonData);
1293
1294     r = CallWindowProcW(info->oldproc, hWnd, msg, wParam, lParam);
1295
1296     switch( msg )
1297     {
1298     case WM_COMMAND:
1299         if (HIWORD(wParam) == EN_CHANGE)
1300         {
1301             msi_mask_control_change( info );
1302             msi_mask_next_control( info, (HWND) lParam );
1303         }
1304         break;
1305     case WM_NCDESTROY:
1306         msi_free( info->prop );
1307         msi_free( info );
1308         RemovePropW( hWnd, szButtonData );
1309         break;
1310     }
1311
1312     return r;
1313 }
1314
1315 /* fish the various bits of the property out and put them in the control */
1316 static void
1317 msi_maskedit_set_text( struct msi_maskedit_info *info, LPCWSTR text )
1318 {
1319     LPCWSTR p;
1320     UINT i;
1321
1322     p = text;
1323     for( i = 0; i < info->num_groups; i++ )
1324     {
1325         if( info->group[i].len < lstrlenW( p ) )
1326         {
1327             LPWSTR chunk = strdupW( p );
1328             chunk[ info->group[i].len ] = 0;
1329             SetWindowTextW( info->group[i].hwnd, chunk );
1330             msi_free( chunk );
1331         }
1332         else
1333         {
1334             SetWindowTextW( info->group[i].hwnd, p );
1335             break;
1336         }
1337         p += info->group[i].len;
1338     }
1339 }
1340
1341 static struct msi_maskedit_info * msi_dialog_parse_groups( LPCWSTR mask )
1342 {
1343     struct msi_maskedit_info * info = NULL;
1344     int i = 0, n = 0, total = 0;
1345     LPCWSTR p;
1346
1347     TRACE("masked control, template %s\n", debugstr_w(mask));
1348
1349     if( !mask )
1350         return info;
1351
1352     info = msi_alloc_zero( sizeof *info );
1353     if( !info )
1354         return info;
1355
1356     p = strchrW(mask, '<');
1357     if( p )
1358         p++;
1359     else
1360         p = mask;
1361
1362     for( i=0; i<MASK_MAX_GROUPS; i++ )
1363     {
1364         /* stop at the end of the string */
1365         if( p[0] == 0 || p[0] == '>' )
1366             break;
1367
1368         /* count the number of the same identifier */
1369         for( n=0; p[n] == p[0]; n++ )
1370             ;
1371         info->group[i].ofs = total;
1372         info->group[i].type = p[0];
1373         if( p[n] == '=' )
1374         {
1375             n++;
1376             total++; /* an extra not part of the group */
1377         }
1378         info->group[i].len = n;
1379         total += n;
1380         p += n;
1381     }
1382
1383     TRACE("%d characters in %d groups\n", total, i );
1384     if( i == MASK_MAX_GROUPS )
1385         ERR("too many groups in PIDTemplate %s\n", debugstr_w(mask));
1386
1387     info->num_chars = total;
1388     info->num_groups = i;
1389
1390     return info;
1391 }
1392
1393 static void
1394 msi_maskedit_create_children( struct msi_maskedit_info *info, LPCWSTR font )
1395 {
1396     DWORD width, height, style, wx, ww;
1397     RECT rect;
1398     HWND hwnd;
1399     UINT i;
1400
1401     style = WS_CHILD | WS_BORDER | WS_VISIBLE | WS_TABSTOP | ES_AUTOHSCROLL;
1402
1403     GetClientRect( info->hwnd, &rect );
1404
1405     width = rect.right - rect.left;
1406     height = rect.bottom - rect.top;
1407
1408     for( i = 0; i < info->num_groups; i++ )
1409     {
1410         if (!msi_mask_editable( info->group[i].type ))
1411             continue;
1412         wx = (info->group[i].ofs * width) / info->num_chars;
1413         ww = (info->group[i].len * width) / info->num_chars;
1414
1415         hwnd = CreateWindowW( szEdit, NULL, style, wx, 0, ww, height,
1416                               info->hwnd, NULL, NULL, NULL );
1417         if( !hwnd )
1418         {
1419             ERR("failed to create mask edit sub window\n");
1420             break;
1421         }
1422
1423         SendMessageW( hwnd, EM_LIMITTEXT, info->group[i].len, 0 );
1424
1425         msi_dialog_set_font( info->dialog, hwnd,
1426                              font?font:info->dialog->default_font );
1427         info->group[i].hwnd = hwnd;
1428     }
1429 }
1430
1431 /*
1432  * office 2003 uses "73931<````=````=````=````=`````>@@@@@"
1433  * delphi 7 uses "<????-??????-??????-????>" and "<???-???>"
1434  * filemaker pro 7 uses "<^^^^=^^^^=^^^^=^^^^=^^^^=^^^^=^^^^^>"
1435  */
1436 static UINT msi_dialog_maskedit_control( msi_dialog *dialog, MSIRECORD *rec )
1437 {
1438     LPWSTR font_mask, val = NULL, font;
1439     struct msi_maskedit_info *info = NULL;
1440     UINT ret = ERROR_SUCCESS;
1441     msi_control *control;
1442     LPCWSTR prop, mask;
1443
1444     TRACE("\n");
1445
1446     font_mask = msi_get_deformatted_field( dialog->package, rec, 10 );
1447     font = msi_dialog_get_style( font_mask, &mask );
1448     if( !mask )
1449     {
1450         ERR("mask template is empty\n");
1451         goto end;
1452     }
1453
1454     info = msi_dialog_parse_groups( mask );
1455     if( !info )
1456     {
1457         ERR("template %s is invalid\n", debugstr_w(mask));
1458         goto end;
1459     }
1460
1461     info->dialog = dialog;
1462
1463     control = msi_dialog_add_control( dialog, rec, szStatic,
1464                    SS_OWNERDRAW | WS_GROUP | WS_VISIBLE );
1465     if( !control )
1466     {
1467         ERR("Failed to create maskedit container\n");
1468         ret = ERROR_FUNCTION_FAILED;
1469         goto end;
1470     }
1471     SetWindowLongPtrW( control->hwnd, GWL_EXSTYLE, WS_EX_CONTROLPARENT );
1472
1473     info->hwnd = control->hwnd;
1474
1475     /* subclass the static control */
1476     info->oldproc = (WNDPROC) SetWindowLongPtrW( info->hwnd, GWLP_WNDPROC,
1477                                           (LONG_PTR)MSIMaskedEdit_WndProc );
1478     SetPropW( control->hwnd, szButtonData, info );
1479
1480     prop = MSI_RecordGetString( rec, 9 );
1481     if( prop )
1482         info->prop = strdupW( prop );
1483
1484     msi_maskedit_create_children( info, font );
1485
1486     if( prop )
1487     {
1488         val = msi_dup_property( dialog->package, prop );
1489         if( val )
1490         {
1491             msi_maskedit_set_text( info, val );
1492             msi_free( val );
1493         }
1494     }
1495
1496 end:
1497     if( ret != ERROR_SUCCESS )
1498         msi_free( info );
1499     msi_free( font_mask );
1500     msi_free( font );
1501     return ret;
1502 }
1503
1504 /******************** Progress Bar *****************************************/
1505
1506 static UINT msi_dialog_progress_bar( msi_dialog *dialog, MSIRECORD *rec )
1507 {
1508     msi_dialog_add_control( dialog, rec, PROGRESS_CLASSW, WS_VISIBLE );
1509     return ERROR_SUCCESS;
1510 }
1511
1512 /******************** Path Edit ********************************************/
1513
1514 struct msi_pathedit_info
1515 {
1516     msi_dialog *dialog;
1517     msi_control *control;
1518     WNDPROC oldproc;
1519 };
1520
1521 static LPWSTR msi_get_window_text( HWND hwnd )
1522 {
1523     UINT sz, r;
1524     LPWSTR buf;
1525
1526     sz = 0x20;
1527     buf = msi_alloc( sz*sizeof(WCHAR) );
1528     while ( buf )
1529     {
1530         r = GetWindowTextW( hwnd, buf, sz );
1531         if ( r < (sz - 1) )
1532             break;
1533         sz *= 2;
1534         buf = msi_realloc( buf, sz*sizeof(WCHAR) );
1535     }
1536
1537     return buf;
1538 }
1539
1540 static void msi_dialog_update_pathedit( msi_dialog *dialog, msi_control *control )
1541 {
1542     LPWSTR prop, path;
1543     BOOL indirect;
1544
1545     if (!control && !(control = msi_dialog_find_control_by_type( dialog, szPathEdit )))
1546        return;
1547
1548     indirect = control->attributes & msidbControlAttributesIndirect;
1549     prop = msi_dialog_dup_property( dialog, control->property, indirect );
1550     path = msi_dialog_dup_property( dialog, prop, TRUE );
1551
1552     SetWindowTextW( control->hwnd, path );
1553     SendMessageW( control->hwnd, EM_SETSEL, 0, -1 );
1554
1555     msi_free( path );
1556     msi_free( prop );
1557 }
1558
1559 /* FIXME: test when this should fail */
1560 static BOOL msi_dialog_verify_path( LPWSTR path )
1561 {
1562     if ( !lstrlenW( path ) )
1563         return FALSE;
1564
1565     if ( PathIsRelativeW( path ) )
1566         return FALSE;
1567
1568     return TRUE;
1569 }
1570
1571 /* returns TRUE if the path is valid, FALSE otherwise */
1572 static BOOL msi_dialog_onkillfocus( msi_dialog *dialog, msi_control *control )
1573 {
1574     LPWSTR buf, prop;
1575     BOOL indirect;
1576     BOOL valid;
1577
1578     indirect = control->attributes & msidbControlAttributesIndirect;
1579     prop = msi_dialog_dup_property( dialog, control->property, indirect );
1580
1581     buf = msi_get_window_text( control->hwnd );
1582
1583     if ( !msi_dialog_verify_path( buf ) )
1584     {
1585         /* FIXME: display an error message box */
1586         ERR("Invalid path %s\n", debugstr_w( buf ));
1587         valid = FALSE;
1588         SetFocus( control->hwnd );
1589     }
1590     else
1591     {
1592         valid = TRUE;
1593         MSI_SetPropertyW( dialog->package, prop, buf );
1594     }
1595
1596     msi_dialog_update_pathedit( dialog, control );
1597
1598     TRACE("edit %s contents changed, set %s\n", debugstr_w(control->name),
1599           debugstr_w(prop));
1600
1601     msi_free( buf );
1602     msi_free( prop );
1603
1604     return valid;
1605 }
1606
1607 static LRESULT WINAPI MSIPathEdit_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
1608 {
1609     struct msi_pathedit_info *info = GetPropW(hWnd, szButtonData);
1610     LRESULT r = 0;
1611
1612     TRACE("%p %04x %08lx %08lx\n", hWnd, msg, wParam, lParam);
1613
1614     if ( msg == WM_KILLFOCUS )
1615     {
1616         /* if the path is invalid, don't handle this message */
1617         if ( !msi_dialog_onkillfocus( info->dialog, info->control ) )
1618             return 0;
1619     }
1620
1621     r = CallWindowProcW(info->oldproc, hWnd, msg, wParam, lParam);
1622
1623     if ( msg == WM_NCDESTROY )
1624     {
1625         msi_free( info );
1626         RemovePropW( hWnd, szButtonData );
1627     }
1628
1629     return r;
1630 }
1631
1632 static UINT msi_dialog_pathedit_control( msi_dialog *dialog, MSIRECORD *rec )
1633 {
1634     struct msi_pathedit_info *info;
1635     msi_control *control;
1636     LPCWSTR prop;
1637
1638     info = msi_alloc( sizeof *info );
1639     if (!info)
1640         return ERROR_FUNCTION_FAILED;
1641
1642     control = msi_dialog_add_control( dialog, rec, szEdit,
1643                                       WS_BORDER | WS_TABSTOP );
1644     control->attributes = MSI_RecordGetInteger( rec, 8 );
1645     prop = MSI_RecordGetString( rec, 9 );
1646     control->property = msi_dialog_dup_property( dialog, prop, FALSE );
1647
1648     info->dialog = dialog;
1649     info->control = control;
1650     info->oldproc = (WNDPROC) SetWindowLongPtrW( control->hwnd, GWLP_WNDPROC,
1651                                                  (LONG_PTR)MSIPathEdit_WndProc );
1652     SetPropW( control->hwnd, szButtonData, info );
1653
1654     msi_dialog_update_pathedit( dialog, control );
1655
1656     return ERROR_SUCCESS;
1657 }
1658
1659 /* radio buttons are a bit different from normal controls */
1660 static UINT msi_dialog_create_radiobutton( MSIRECORD *rec, LPVOID param )
1661 {
1662     radio_button_group_descr *group = (radio_button_group_descr *)param;
1663     msi_dialog *dialog = group->dialog;
1664     msi_control *control;
1665     LPCWSTR prop, text, name;
1666     DWORD style, attributes = group->attributes;
1667
1668     style = WS_CHILD | BS_AUTORADIOBUTTON | BS_MULTILINE | WS_TABSTOP;
1669     name = MSI_RecordGetString( rec, 3 );
1670     text = MSI_RecordGetString( rec, 8 );
1671     if( attributes & msidbControlAttributesVisible )
1672         style |= WS_VISIBLE;
1673     if( ~attributes & msidbControlAttributesEnabled )
1674         style |= WS_DISABLED;
1675
1676     control = msi_dialog_create_window( dialog, rec, 0, szButton, name, text,
1677                                         style, group->parent->hwnd );
1678     if (!control)
1679         return ERROR_FUNCTION_FAILED;
1680     control->handler = msi_dialog_radiogroup_handler;
1681
1682     if (!lstrcmpW(control->name, group->propval))
1683         SendMessageW(control->hwnd, BM_SETCHECK, BST_CHECKED, 0);
1684
1685     prop = MSI_RecordGetString( rec, 1 );
1686     if( prop )
1687         control->property = strdupW( prop );
1688
1689     return ERROR_SUCCESS;
1690 }
1691
1692 static UINT msi_dialog_radiogroup_control( msi_dialog *dialog, MSIRECORD *rec )
1693 {
1694     static const WCHAR query[] = {
1695         'S','E','L','E','C','T',' ','*',' ',
1696         'F','R','O','M',' ','R','a','d','i','o','B','u','t','t','o','n',' ',
1697         'W','H','E','R','E',' ',
1698            '`','P','r','o','p','e','r','t','y','`',' ','=',' ','\'','%','s','\'',0};
1699     UINT r;
1700     LPCWSTR prop;
1701     msi_control *control;
1702     MSIQUERY *view = NULL;
1703     radio_button_group_descr group;
1704     MSIPACKAGE *package = dialog->package;
1705     WNDPROC oldproc;
1706
1707     prop = MSI_RecordGetString( rec, 9 );
1708
1709     TRACE("%p %p %s\n", dialog, rec, debugstr_w( prop ));
1710
1711     /* Create parent group box to hold radio buttons */
1712     control = msi_dialog_add_control( dialog, rec, szButton, BS_GROUPBOX|WS_GROUP );
1713     if( !control )
1714         return ERROR_FUNCTION_FAILED;
1715
1716     oldproc = (WNDPROC) SetWindowLongPtrW( control->hwnd, GWLP_WNDPROC,
1717                                            (LONG_PTR)MSIRadioGroup_WndProc );
1718     SetPropW(control->hwnd, szButtonData, oldproc);
1719     SetWindowLongPtrW( control->hwnd, GWL_EXSTYLE, WS_EX_CONTROLPARENT );
1720
1721     if( prop )
1722         control->property = strdupW( prop );
1723
1724     /* query the Radio Button table for all control in this group */
1725     r = MSI_OpenQuery( package->db, &view, query, prop );
1726     if( r != ERROR_SUCCESS )
1727     {
1728         ERR("query failed for dialog %s radio group %s\n",
1729             debugstr_w(dialog->name), debugstr_w(prop));
1730         return ERROR_INVALID_PARAMETER;
1731     }
1732
1733     group.dialog = dialog;
1734     group.parent = control;
1735     group.attributes = MSI_RecordGetInteger( rec, 8 );
1736     group.propval = msi_dup_property( dialog->package, control->property );
1737
1738     r = MSI_IterateRecords( view, 0, msi_dialog_create_radiobutton, &group );
1739     msiobj_release( &view->hdr );
1740     msi_free( group.propval );
1741
1742     return r;
1743 }
1744
1745 /******************** Selection Tree ***************************************/
1746
1747 struct msi_selection_tree_info
1748 {
1749     msi_dialog *dialog;
1750     HWND hwnd;
1751     WNDPROC oldproc;
1752     HTREEITEM selected;
1753 };
1754
1755 static void
1756 msi_seltree_sync_item_state( HWND hwnd, MSIFEATURE *feature, HTREEITEM hItem )
1757 {
1758     TVITEMW tvi;
1759     DWORD index = feature->Action;
1760
1761     TRACE("Feature %s -> %d %d %d\n", debugstr_w(feature->Title),
1762         feature->Installed, feature->Action, feature->ActionRequest);
1763
1764     if (index == INSTALLSTATE_UNKNOWN)
1765         index = INSTALLSTATE_ABSENT;
1766
1767     tvi.mask = TVIF_STATE;
1768     tvi.hItem = hItem;
1769     tvi.state = INDEXTOSTATEIMAGEMASK( index );
1770     tvi.stateMask = TVIS_STATEIMAGEMASK;
1771
1772     SendMessageW( hwnd, TVM_SETITEMW, 0, (LPARAM) &tvi );
1773 }
1774
1775 static UINT
1776 msi_seltree_popup_menu( HWND hwnd, INT x, INT y )
1777 {
1778     HMENU hMenu;
1779     INT r;
1780
1781     /* create a menu to display */
1782     hMenu = CreatePopupMenu();
1783
1784     /* FIXME: load strings from resources */
1785     AppendMenuA( hMenu, MF_ENABLED, INSTALLSTATE_LOCAL, "Install feature locally");
1786     AppendMenuA( hMenu, MF_GRAYED, 0x1000, "Install entire feature");
1787     AppendMenuA( hMenu, MF_ENABLED, INSTALLSTATE_ADVERTISED, "Install on demand");
1788     AppendMenuA( hMenu, MF_ENABLED, INSTALLSTATE_ABSENT, "Don't install");
1789     r = TrackPopupMenu( hMenu, TPM_LEFTALIGN | TPM_TOPALIGN | TPM_RETURNCMD,
1790                         x, y, 0, hwnd, NULL );
1791     DestroyMenu( hMenu );
1792     return r;
1793 }
1794
1795 static MSIFEATURE *
1796 msi_seltree_feature_from_item( HWND hwnd, HTREEITEM hItem )
1797 {
1798     TVITEMW tvi;
1799
1800     /* get the feature from the item */
1801     memset( &tvi, 0, sizeof tvi );
1802     tvi.hItem = hItem;
1803     tvi.mask = TVIF_PARAM | TVIF_HANDLE;
1804     SendMessageW( hwnd, TVM_GETITEMW, 0, (LPARAM) &tvi );
1805
1806     return (MSIFEATURE*) tvi.lParam;
1807 }
1808
1809 static LRESULT
1810 msi_seltree_menu( HWND hwnd, HTREEITEM hItem )
1811 {
1812     struct msi_selection_tree_info *info;
1813     MSIFEATURE *feature;
1814     MSIPACKAGE *package;
1815     union {
1816         RECT rc;
1817         POINT pt[2];
1818         HTREEITEM hItem;
1819     } u;
1820     UINT r;
1821
1822     info = GetPropW(hwnd, szButtonData);
1823     package = info->dialog->package;
1824
1825     feature = msi_seltree_feature_from_item( hwnd, hItem );
1826     if (!feature)
1827     {
1828         ERR("item %p feature was NULL\n", hItem);
1829         return 0;
1830     }
1831
1832     /* get the item's rectangle to put the menu just below it */
1833     u.hItem = hItem;
1834     SendMessageW( hwnd, TVM_GETITEMRECT, 0, (LPARAM) &u.rc );
1835     MapWindowPoints( hwnd, NULL, u.pt, 2 );
1836
1837     r = msi_seltree_popup_menu( hwnd, u.rc.left, u.rc.top );
1838
1839     switch (r)
1840     {
1841     case INSTALLSTATE_LOCAL:
1842     case INSTALLSTATE_ADVERTISED:
1843     case INSTALLSTATE_ABSENT:
1844         msi_feature_set_state( feature, r );
1845         break;
1846     default:
1847         FIXME("select feature and all children\n");
1848     }
1849
1850     /* update */
1851     msi_seltree_sync_item_state( hwnd, feature, hItem );
1852     ACTION_UpdateComponentStates( package, feature->Feature );
1853
1854     return 0;
1855 }
1856
1857 static MSIFEATURE *msi_seltree_get_selected_feature( msi_control *control )
1858 {
1859     struct msi_selection_tree_info *info = GetPropW(control->hwnd, szButtonData);
1860     return msi_seltree_feature_from_item( control->hwnd, info->selected );
1861 }
1862
1863 static LRESULT WINAPI
1864 MSISelectionTree_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
1865 {
1866     struct msi_selection_tree_info *info;
1867     TVHITTESTINFO tvhti;
1868     HRESULT r;
1869
1870     TRACE("%p %04x %08lx %08lx\n", hWnd, msg, wParam, lParam);
1871
1872     info = GetPropW(hWnd, szButtonData);
1873
1874     switch( msg )
1875     {
1876     case WM_LBUTTONDOWN:
1877         tvhti.pt.x = (short)LOWORD( lParam );
1878         tvhti.pt.y = (short)HIWORD( lParam );
1879         tvhti.flags = 0;
1880         tvhti.hItem = 0;
1881         r = CallWindowProcW(info->oldproc, hWnd, TVM_HITTEST, 0, (LPARAM) &tvhti );
1882         if (tvhti.flags & TVHT_ONITEMSTATEICON)
1883             return msi_seltree_menu( hWnd, tvhti.hItem );
1884         break;
1885     }
1886
1887     r = CallWindowProcW(info->oldproc, hWnd, msg, wParam, lParam);
1888
1889     switch( msg )
1890     {
1891     case WM_NCDESTROY:
1892         msi_free( info );
1893         RemovePropW( hWnd, szButtonData );
1894         break;
1895     }
1896     return r;
1897 }
1898
1899 static void
1900 msi_seltree_add_child_features( MSIPACKAGE *package, HWND hwnd,
1901                                 LPCWSTR parent, HTREEITEM hParent )
1902 {
1903     struct msi_selection_tree_info *info = GetPropW( hwnd, szButtonData );
1904     MSIFEATURE *feature;
1905     TVINSERTSTRUCTW tvis;
1906     HTREEITEM hitem, hfirst = NULL;
1907
1908     LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
1909     {
1910         if ( lstrcmpW( parent, feature->Feature_Parent ) )
1911             continue;
1912
1913         if ( !feature->Title )
1914             continue;
1915
1916         if ( !feature->Display )
1917             continue;
1918
1919         memset( &tvis, 0, sizeof tvis );
1920         tvis.hParent = hParent;
1921         tvis.hInsertAfter = TVI_LAST;
1922         tvis.u.item.mask = TVIF_TEXT | TVIF_PARAM;
1923         tvis.u.item.pszText = feature->Title;
1924         tvis.u.item.lParam = (LPARAM) feature;
1925
1926         hitem = (HTREEITEM) SendMessageW( hwnd, TVM_INSERTITEMW, 0, (LPARAM) &tvis );
1927         if (!hitem)
1928             continue;
1929
1930         if (!hfirst)
1931             hfirst = hitem;
1932
1933         msi_seltree_sync_item_state( hwnd, feature, hitem );
1934         msi_seltree_add_child_features( package, hwnd,
1935                                         feature->Feature, hitem );
1936
1937         /* the node is expanded if Display is odd */
1938         if ( feature->Display % 2 != 0 )
1939             SendMessageW( hwnd, TVM_EXPAND, TVE_EXPAND, (LPARAM) hitem );
1940     }
1941
1942     /* select the first item */
1943     SendMessageW( hwnd, TVM_SELECTITEM, TVGN_CARET | TVGN_DROPHILITE, (LPARAM) hfirst );
1944     info->selected = hfirst;
1945 }
1946
1947 static void msi_seltree_create_imagelist( HWND hwnd )
1948 {
1949     const int bm_width = 32, bm_height = 16, bm_count = 3;
1950     const int bm_resource = 0x1001;
1951     HIMAGELIST himl;
1952     int i;
1953     HBITMAP hbmp;
1954
1955     himl = ImageList_Create( bm_width, bm_height, FALSE, 4, 0 );
1956     if (!himl)
1957     {
1958         ERR("failed to create image list\n");
1959         return;
1960     }
1961
1962     for (i=0; i<bm_count; i++)
1963     {
1964         hbmp = LoadBitmapW( msi_hInstance, MAKEINTRESOURCEW(i+bm_resource) );
1965         if (!hbmp)
1966         {
1967             ERR("failed to load bitmap %d\n", i);
1968             break;
1969         }
1970
1971         /*
1972          * Add a dummy bitmap at offset zero because the treeview
1973          * can't use it as a state mask (zero means no user state).
1974          */
1975         if (!i)
1976             ImageList_Add( himl, hbmp, NULL );
1977
1978         ImageList_Add( himl, hbmp, NULL );
1979     }
1980
1981     SendMessageW( hwnd, TVM_SETIMAGELIST, TVSIL_STATE, (LPARAM)himl );
1982 }
1983
1984 static UINT msi_dialog_seltree_handler( msi_dialog *dialog,
1985                                         msi_control *control, WPARAM param )
1986 {
1987     struct msi_selection_tree_info *info = GetPropW( control->hwnd, szButtonData );
1988     LPNMTREEVIEWW tv = (LPNMTREEVIEWW)param;
1989     MSIRECORD *row, *rec;
1990     MSIFOLDER *folder;
1991     LPCWSTR dir;
1992     UINT r = ERROR_SUCCESS;
1993
1994     static const WCHAR select[] = {
1995         'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
1996         '`','F','e','a','t','u','r','e','`',' ','W','H','E','R','E',' ',
1997         '`','T','i','t','l','e','`',' ','=',' ','\'','%','s','\'',0
1998     };
1999
2000     if (tv->hdr.code != TVN_SELCHANGINGW)
2001         return ERROR_SUCCESS;
2002
2003     info->selected = tv->itemNew.hItem;
2004
2005     row = MSI_QueryGetRecord( dialog->package->db, select, tv->itemNew.pszText );
2006     if (!row)
2007         return ERROR_FUNCTION_FAILED;
2008
2009     rec = MSI_CreateRecord( 1 );
2010
2011     MSI_RecordSetStringW( rec, 1, MSI_RecordGetString( row, 4 ) );
2012     ControlEvent_FireSubscribedEvent( dialog->package, szSelectionDescription, rec );
2013
2014     dir = MSI_RecordGetString( row, 7 );
2015     folder = get_loaded_folder( dialog->package, dir );
2016     if (!folder)
2017     {
2018         r = ERROR_FUNCTION_FAILED;
2019         goto done;
2020     }
2021
2022     MSI_RecordSetStringW( rec, 1, folder->ResolvedTarget );
2023     ControlEvent_FireSubscribedEvent( dialog->package, szSelectionPath, rec );
2024
2025 done:
2026     msiobj_release(&row->hdr);
2027     msiobj_release(&rec->hdr);
2028
2029     return r;
2030 }
2031
2032 static UINT msi_dialog_selection_tree( msi_dialog *dialog, MSIRECORD *rec )
2033 {
2034     msi_control *control;
2035     LPCWSTR prop;
2036     MSIPACKAGE *package = dialog->package;
2037     DWORD style;
2038     struct msi_selection_tree_info *info;
2039
2040     info = msi_alloc( sizeof *info );
2041     if (!info)
2042         return ERROR_FUNCTION_FAILED;
2043
2044     /* create the treeview control */
2045     style = TVS_HASLINES | TVS_HASBUTTONS | TVS_LINESATROOT;
2046     style |= WS_GROUP | WS_VSCROLL;
2047     control = msi_dialog_add_control( dialog, rec, WC_TREEVIEWW, style );
2048     if (!control)
2049     {
2050         msi_free(info);
2051         return ERROR_FUNCTION_FAILED;
2052     }
2053
2054     control->handler = msi_dialog_seltree_handler;
2055     control->attributes = MSI_RecordGetInteger( rec, 8 );
2056     prop = MSI_RecordGetString( rec, 9 );
2057     control->property = msi_dialog_dup_property( dialog, prop, FALSE );
2058
2059     /* subclass */
2060     info->dialog = dialog;
2061     info->hwnd = control->hwnd;
2062     info->oldproc = (WNDPROC) SetWindowLongPtrW( control->hwnd, GWLP_WNDPROC,
2063                                           (LONG_PTR)MSISelectionTree_WndProc );
2064     SetPropW( control->hwnd, szButtonData, info );
2065
2066     ControlEvent_SubscribeToEvent( dialog->package, dialog,
2067                                    szSelectionPath, control->name, szProperty );
2068
2069     /* initialize it */
2070     msi_seltree_create_imagelist( control->hwnd );
2071     msi_seltree_add_child_features( package, control->hwnd, NULL, NULL );
2072
2073     return ERROR_SUCCESS;
2074 }
2075
2076 /******************** Group Box ***************************************/
2077
2078 static UINT msi_dialog_group_box( msi_dialog *dialog, MSIRECORD *rec )
2079 {
2080     msi_control *control;
2081     DWORD style;
2082
2083     style = BS_GROUPBOX | WS_CHILD | WS_GROUP;
2084     control = msi_dialog_add_control( dialog, rec, WC_BUTTONW, style );
2085     if (!control)
2086         return ERROR_FUNCTION_FAILED;
2087
2088     return ERROR_SUCCESS;
2089 }
2090
2091 /******************** List Box ***************************************/
2092
2093 struct msi_listbox_info
2094 {
2095     msi_dialog *dialog;
2096     HWND hwnd;
2097     WNDPROC oldproc;
2098     DWORD num_items;
2099     DWORD addpos_items;
2100     LPWSTR *items;
2101 };
2102
2103 static LRESULT WINAPI MSIListBox_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
2104 {
2105     struct msi_listbox_info *info;
2106     LRESULT r;
2107     DWORD j;
2108
2109     TRACE("%p %04x %08lx %08lx\n", hWnd, msg, wParam, lParam);
2110
2111     info = GetPropW( hWnd, szButtonData );
2112     if (!info)
2113         return 0;
2114
2115     r = CallWindowProcW( info->oldproc, hWnd, msg, wParam, lParam );
2116
2117     switch( msg )
2118     {
2119     case WM_NCDESTROY:
2120         for (j = 0; j < info->num_items; j++)
2121             msi_free( info->items[j] );
2122         msi_free( info->items );
2123         msi_free( info );
2124         RemovePropW( hWnd, szButtonData );
2125         break;
2126     }
2127
2128     return r;
2129 }
2130
2131 static UINT msi_listbox_add_item( MSIRECORD *rec, LPVOID param )
2132 {
2133     struct msi_listbox_info *info = param;
2134     LPCWSTR value, text;
2135     int pos;
2136
2137     value = MSI_RecordGetString( rec, 3 );
2138     text = MSI_RecordGetString( rec, 4 );
2139
2140     info->items[info->addpos_items] = strdupW( value );
2141
2142     pos = SendMessageW( info->hwnd, LB_ADDSTRING, 0, (LPARAM)text );
2143     SendMessageW( info->hwnd, LB_SETITEMDATA, pos, (LPARAM)info->items[info->addpos_items] );
2144     info->addpos_items++;
2145     return ERROR_SUCCESS;
2146 }
2147
2148 static UINT msi_listbox_add_items( struct msi_listbox_info *info, LPCWSTR property )
2149 {
2150     UINT r;
2151     MSIQUERY *view = NULL;
2152     DWORD count;
2153
2154     static const WCHAR query[] = {
2155         'S','E','L','E','C','T',' ','*',' ',
2156         'F','R','O','M',' ','`','L','i','s','t','B','o','x','`',' ',
2157         'W','H','E','R','E',' ',
2158         '`','P','r','o','p','e','r','t','y','`',' ','=',' ','\'','%','s','\'',' ',
2159         'O','R','D','E','R',' ','B','Y',' ','`','O','r','d','e','r','`',0
2160     };
2161
2162     r = MSI_OpenQuery( info->dialog->package->db, &view, query, property );
2163     if ( r != ERROR_SUCCESS )
2164         return r;
2165
2166     /* just get the number of records */
2167     r = MSI_IterateRecords( view, &count, NULL, NULL );
2168
2169     info->num_items = count;
2170     info->items = msi_alloc( sizeof(*info->items) * count );
2171
2172     r = MSI_IterateRecords( view, NULL, msi_listbox_add_item, info );
2173     msiobj_release( &view->hdr );
2174
2175     return r;
2176 }
2177
2178 static UINT msi_dialog_listbox_handler( msi_dialog *dialog,
2179                                         msi_control *control, WPARAM param )
2180 {
2181     struct msi_listbox_info *info;
2182     int index;
2183     LPCWSTR value;
2184
2185     if( HIWORD(param) != LBN_SELCHANGE )
2186         return ERROR_SUCCESS;
2187
2188     info = GetPropW( control->hwnd, szButtonData );
2189     index = SendMessageW( control->hwnd, LB_GETCURSEL, 0, 0 );
2190     value = (LPCWSTR) SendMessageW( control->hwnd, LB_GETITEMDATA, index, 0 );
2191
2192     MSI_SetPropertyW( info->dialog->package,
2193                       control->property, value );
2194     msi_dialog_evaluate_control_conditions( info->dialog );
2195
2196     return ERROR_SUCCESS;
2197 }
2198
2199 static UINT msi_dialog_list_box( msi_dialog *dialog, MSIRECORD *rec )
2200 {
2201     struct msi_listbox_info *info;
2202     msi_control *control;
2203     DWORD attributes, style;
2204     LPCWSTR prop;
2205
2206     info = msi_alloc( sizeof *info );
2207     if (!info)
2208         return ERROR_FUNCTION_FAILED;
2209
2210     style = WS_TABSTOP | WS_GROUP | WS_CHILD | LBS_NOTIFY | WS_VSCROLL | WS_BORDER;
2211     attributes = MSI_RecordGetInteger( rec, 8 );
2212     if (~attributes & msidbControlAttributesSorted)
2213         style |= LBS_SORT;
2214
2215     control = msi_dialog_add_control( dialog, rec, WC_LISTBOXW, style );
2216     if (!control)
2217         return ERROR_FUNCTION_FAILED;
2218
2219     control->handler = msi_dialog_listbox_handler;
2220
2221     prop = MSI_RecordGetString( rec, 9 );
2222     control->property = msi_dialog_dup_property( dialog, prop, FALSE );
2223
2224     /* subclass */
2225     info->dialog = dialog;
2226     info->hwnd = control->hwnd;
2227     info->items = NULL;
2228     info->addpos_items = 0;
2229     info->oldproc = (WNDPROC)SetWindowLongPtrW( control->hwnd, GWLP_WNDPROC,
2230                                                 (LONG_PTR)MSIListBox_WndProc );
2231     SetPropW( control->hwnd, szButtonData, info );
2232
2233     if ( control->property )
2234         msi_listbox_add_items( info, control->property );
2235
2236     return ERROR_SUCCESS;
2237 }
2238
2239 /******************** Directory Combo ***************************************/
2240
2241 static void msi_dialog_update_directory_combo( msi_dialog *dialog, msi_control *control )
2242 {
2243     LPWSTR prop, path;
2244     BOOL indirect;
2245
2246     if (!control && !(control = msi_dialog_find_control_by_type( dialog, szDirectoryCombo )))
2247         return;
2248
2249     indirect = control->attributes & msidbControlAttributesIndirect;
2250     prop = msi_dialog_dup_property( dialog, control->property, indirect );
2251     path = msi_dialog_dup_property( dialog, prop, TRUE );
2252
2253     PathStripPathW( path );
2254     PathRemoveBackslashW( path );
2255
2256     SendMessageW( control->hwnd, CB_INSERTSTRING, 0, (LPARAM)path );
2257     SendMessageW( control->hwnd, CB_SETCURSEL, 0, 0 );
2258
2259     msi_free( path );
2260     msi_free( prop );
2261 }
2262
2263 static UINT msi_dialog_directory_combo( msi_dialog *dialog, MSIRECORD *rec )
2264 {
2265     msi_control *control;
2266     LPCWSTR prop;
2267     DWORD style;
2268
2269     /* FIXME: use CBS_OWNERDRAWFIXED and add owner draw code */
2270     style = CBS_DROPDOWNLIST | CBS_HASSTRINGS | WS_CHILD |
2271             WS_GROUP | WS_TABSTOP | WS_VSCROLL;
2272     control = msi_dialog_add_control( dialog, rec, WC_COMBOBOXW, style );
2273     if (!control)
2274         return ERROR_FUNCTION_FAILED;
2275
2276     control->attributes = MSI_RecordGetInteger( rec, 8 );
2277     prop = MSI_RecordGetString( rec, 9 );
2278     control->property = msi_dialog_dup_property( dialog, prop, FALSE );
2279
2280     msi_dialog_update_directory_combo( dialog, control );
2281
2282     return ERROR_SUCCESS;
2283 }
2284
2285 /******************** Directory List ***************************************/
2286
2287 static void msi_dialog_update_directory_list( msi_dialog *dialog, msi_control *control )
2288 {
2289     WCHAR dir_spec[MAX_PATH];
2290     WIN32_FIND_DATAW wfd;
2291     LPWSTR prop, path;
2292     BOOL indirect;
2293     LVITEMW item;
2294     HANDLE file;
2295
2296     static const WCHAR asterisk[] = {'*',0};
2297     static const WCHAR dot[] = {'.',0};
2298     static const WCHAR dotdot[] = {'.','.',0};
2299
2300     if (!control && !(control = msi_dialog_find_control_by_type( dialog, szDirectoryList )))
2301         return;
2302
2303     /* clear the list-view */
2304     SendMessageW( control->hwnd, LVM_DELETEALLITEMS, 0, 0 );
2305
2306     indirect = control->attributes & msidbControlAttributesIndirect;
2307     prop = msi_dialog_dup_property( dialog, control->property, indirect );
2308     path = msi_dialog_dup_property( dialog, prop, TRUE );
2309
2310     lstrcpyW( dir_spec, path );
2311     lstrcatW( dir_spec, asterisk );
2312
2313     file = FindFirstFileW( dir_spec, &wfd );
2314     if ( file == INVALID_HANDLE_VALUE )
2315         return;
2316
2317     do
2318     {
2319         if ( wfd.dwFileAttributes != FILE_ATTRIBUTE_DIRECTORY )
2320             continue;
2321
2322         if ( !lstrcmpW( wfd.cFileName, dot ) || !lstrcmpW( wfd.cFileName, dotdot ) )
2323             continue;
2324
2325         item.mask = LVIF_TEXT;
2326         item.cchTextMax = MAX_PATH;
2327         item.iItem = 0;
2328         item.iSubItem = 0;
2329         item.pszText = wfd.cFileName;
2330
2331         SendMessageW( control->hwnd, LVM_INSERTITEMW, 0, (LPARAM)&item );
2332     } while ( FindNextFileW( file, &wfd ) );
2333
2334     msi_free( prop );
2335     msi_free( path );
2336     FindClose( file );
2337 }
2338
2339 UINT msi_dialog_directorylist_up( msi_dialog *dialog )
2340 {
2341     msi_control *control;
2342     LPWSTR prop, path, ptr;
2343     BOOL indirect;
2344
2345     control = msi_dialog_find_control_by_type( dialog, szDirectoryList );
2346     indirect = control->attributes & msidbControlAttributesIndirect;
2347     prop = msi_dialog_dup_property( dialog, control->property, indirect );
2348     path = msi_dialog_dup_property( dialog, prop, TRUE );
2349
2350     /* strip off the last directory */
2351     ptr = PathFindFileNameW( path );
2352     if (ptr != path) *(ptr - 1) = '\0';
2353     PathAddBackslashW( path );
2354
2355     MSI_SetPropertyW( dialog->package, prop, path );
2356
2357     msi_dialog_update_directory_list( dialog, NULL );
2358     msi_dialog_update_directory_combo( dialog, NULL );
2359     msi_dialog_update_pathedit( dialog, NULL );
2360
2361     msi_free( path );
2362     msi_free( prop );
2363
2364     return ERROR_SUCCESS;
2365 }
2366
2367 static UINT msi_dialog_dirlist_handler( msi_dialog *dialog,
2368                                         msi_control *control, WPARAM param )
2369 {
2370     LPNMHDR nmhdr = (LPNMHDR)param;
2371     WCHAR new_path[MAX_PATH];
2372     WCHAR text[MAX_PATH];
2373     LPWSTR path, prop;
2374     BOOL indirect;
2375     LVITEMW item;
2376     int index;
2377
2378     static const WCHAR backslash[] = {'\\',0};
2379
2380     if (nmhdr->code != LVN_ITEMACTIVATE)
2381         return ERROR_SUCCESS;
2382
2383     index = SendMessageW( control->hwnd, LVM_GETNEXTITEM, -1, LVNI_SELECTED );
2384     if ( index < 0 )
2385     {
2386         ERR("No list-view item selected!\n");
2387         return ERROR_FUNCTION_FAILED;
2388     }
2389
2390     item.iSubItem = 0;
2391     item.pszText = text;
2392     item.cchTextMax = MAX_PATH;
2393     SendMessageW( control->hwnd, LVM_GETITEMTEXTW, index, (LPARAM)&item );
2394
2395     indirect = control->attributes & msidbControlAttributesIndirect;
2396     prop = msi_dialog_dup_property( dialog, control->property, indirect );
2397     path = msi_dialog_dup_property( dialog, prop, TRUE );
2398
2399     lstrcpyW( new_path, path );
2400     lstrcatW( new_path, text );
2401     lstrcatW( new_path, backslash );
2402
2403     MSI_SetPropertyW( dialog->package, prop, new_path );
2404
2405     msi_dialog_update_directory_list( dialog, NULL );
2406     msi_dialog_update_directory_combo( dialog, NULL );
2407     msi_dialog_update_pathedit( dialog, NULL );
2408
2409     msi_free( prop );
2410     msi_free( path );
2411     return ERROR_SUCCESS;
2412 }
2413
2414 static UINT msi_dialog_directory_list( msi_dialog *dialog, MSIRECORD *rec )
2415 {
2416     msi_control *control;
2417     LPCWSTR prop;
2418     DWORD style;
2419
2420     style = LVS_LIST | WS_VSCROLL | LVS_SHAREIMAGELISTS |
2421             LVS_AUTOARRANGE | LVS_SINGLESEL | WS_BORDER |
2422             LVS_SORTASCENDING | WS_CHILD | WS_GROUP | WS_TABSTOP;
2423     control = msi_dialog_add_control( dialog, rec, WC_LISTVIEWW, style );
2424     if (!control)
2425         return ERROR_FUNCTION_FAILED;
2426
2427     control->attributes = MSI_RecordGetInteger( rec, 8 );
2428     control->handler = msi_dialog_dirlist_handler;
2429     prop = MSI_RecordGetString( rec, 9 );
2430     control->property = msi_dialog_dup_property( dialog, prop, FALSE );
2431
2432     /* double click to activate an item in the list */
2433     SendMessageW( control->hwnd, LVM_SETEXTENDEDLISTVIEWSTYLE,
2434                   0, LVS_EX_TWOCLICKACTIVATE );
2435
2436     msi_dialog_update_directory_list( dialog, control );
2437
2438     return ERROR_SUCCESS;
2439 }
2440
2441 /******************** VolumeCost List ***************************************/
2442
2443 static BOOL str_is_number( LPCWSTR str )
2444 {
2445     int i;
2446
2447     for (i = 0; i < lstrlenW( str ); i++)
2448         if (!isdigitW(str[i]))
2449             return FALSE;
2450
2451     return TRUE;
2452 }
2453
2454 static const WCHAR column_keys[][80] =
2455 {
2456     {'V','o','l','u','m','e','C','o','s','t','V','o','l','u','m','e',0},
2457     {'V','o','l','u','m','e','C','o','s','t','S','i','z','e',0},
2458     {'V','o','l','u','m','e','C','o','s','t','A','v','a','i','l','a','b','l','e',0},
2459     {'V','o','l','u','m','e','C','o','s','t','R','e','q','u','i','r','e','d',0},
2460     {'V','o','l','u','m','e','C','o','s','t','D','i','f','f','e','r','e','n','c','e',0}
2461 };
2462
2463 static void msi_dialog_vcl_add_columns( msi_dialog *dialog, msi_control *control, MSIRECORD *rec )
2464 {
2465     LPCWSTR text = MSI_RecordGetString( rec, 10 );
2466     LPCWSTR begin = text, end;
2467     WCHAR num[10];
2468     LVCOLUMNW lvc;
2469     DWORD count = 0;
2470
2471     static const WCHAR zero[] = {'0',0};
2472     static const WCHAR negative[] = {'-',0};
2473
2474     if (!text) return;
2475
2476     while ((begin = strchrW( begin, '{' )) && count < 5)
2477     {
2478         if (!(end = strchrW( begin, '}' )))
2479             return;
2480
2481         lstrcpynW( num, begin + 1, end - begin );
2482         begin += end - begin + 1;
2483
2484         /* empty braces or '0' hides the column */
2485         if ( !num[0] || !lstrcmpW( num, zero ) )
2486         {
2487             count++;
2488             continue;
2489         }
2490
2491         /* the width must be a positive number
2492          * if a width is invalid, all remaining columns are hidden
2493          */
2494         if ( !strncmpW( num, negative, 1 ) || !str_is_number( num ) )
2495             return;
2496
2497         ZeroMemory( &lvc, sizeof(lvc) );
2498         lvc.mask = LVCF_TEXT | LVCF_WIDTH | LVCF_SUBITEM;
2499         lvc.cx = atolW( num );
2500         lvc.pszText = msi_dialog_get_uitext( dialog, column_keys[count] );
2501
2502         SendMessageW( control->hwnd,  LVM_INSERTCOLUMNW, count++, (LPARAM)&lvc );
2503         msi_free( lvc.pszText );
2504     }
2505 }
2506
2507 static void msi_dialog_vcl_add_drives( msi_dialog *dialog, msi_control *control )
2508 {
2509     ULARGE_INTEGER total, free;
2510     WCHAR size_text[MAX_PATH];
2511     LPWSTR drives, ptr;
2512     LVITEMW lvitem;
2513     DWORD size;
2514     int i = 0;
2515
2516     size = GetLogicalDriveStringsW( 0, NULL );
2517     if ( !size ) return;
2518
2519     drives = msi_alloc( (size + 1) * sizeof(WCHAR) );
2520     if ( !drives ) return;
2521
2522     GetLogicalDriveStringsW( size, drives );
2523
2524     ptr = drives;
2525     while (*ptr)
2526     {
2527         lvitem.mask = LVIF_TEXT;
2528         lvitem.iItem = i;
2529         lvitem.iSubItem = 0;
2530         lvitem.pszText = ptr;
2531         lvitem.cchTextMax = lstrlenW(ptr) + 1;
2532         SendMessageW( control->hwnd, LVM_INSERTITEMW, 0, (LPARAM)&lvitem );
2533
2534         GetDiskFreeSpaceExW(ptr, &free, &total, NULL);
2535
2536         StrFormatByteSizeW(total.QuadPart, size_text, MAX_PATH);
2537         lvitem.iSubItem = 1;
2538         lvitem.pszText = size_text;
2539         lvitem.cchTextMax = lstrlenW(size_text) + 1;
2540         SendMessageW( control->hwnd, LVM_SETITEMW, 0, (LPARAM)&lvitem );
2541
2542         StrFormatByteSizeW(free.QuadPart, size_text, MAX_PATH);
2543         lvitem.iSubItem = 2;
2544         lvitem.pszText = size_text;
2545         lvitem.cchTextMax = lstrlenW(size_text) + 1;
2546         SendMessageW( control->hwnd, LVM_SETITEMW, 0, (LPARAM)&lvitem );
2547
2548         ptr += lstrlenW(ptr) + 1;
2549         i++;
2550     }
2551
2552     msi_free( drives );
2553 }
2554
2555 static UINT msi_dialog_volumecost_list( msi_dialog *dialog, MSIRECORD *rec )
2556 {
2557     msi_control *control;
2558     DWORD style;
2559
2560     style = LVS_REPORT | WS_VSCROLL | WS_HSCROLL | LVS_SHAREIMAGELISTS |
2561             LVS_AUTOARRANGE | LVS_SINGLESEL | WS_BORDER |
2562             WS_CHILD | WS_TABSTOP | WS_GROUP;
2563     control = msi_dialog_add_control( dialog, rec, WC_LISTVIEWW, style );
2564     if (!control)
2565         return ERROR_FUNCTION_FAILED;
2566
2567     msi_dialog_vcl_add_columns( dialog, control, rec );
2568     msi_dialog_vcl_add_drives( dialog, control );
2569
2570     return ERROR_SUCCESS;
2571 }
2572
2573 /******************** VolumeSelect Combo ***************************************/
2574
2575 static UINT msi_dialog_volsel_handler( msi_dialog *dialog,
2576                                        msi_control *control, WPARAM param )
2577 {
2578     WCHAR text[MAX_PATH];
2579     LPWSTR prop;
2580     BOOL indirect;
2581     int index;
2582
2583     if (HIWORD(param) != CBN_SELCHANGE)
2584         return ERROR_SUCCESS;
2585
2586     index = SendMessageW( control->hwnd, CB_GETCURSEL, 0, 0 );
2587     if ( index == CB_ERR )
2588     {
2589         ERR("No ComboBox item selected!\n");
2590         return ERROR_FUNCTION_FAILED;
2591     }
2592
2593     SendMessageW( control->hwnd, CB_GETLBTEXT, index, (LPARAM)text );
2594
2595     indirect = control->attributes & msidbControlAttributesIndirect;
2596     prop = msi_dialog_dup_property( dialog, control->property, indirect );
2597
2598     MSI_SetPropertyW( dialog->package, prop, text );
2599
2600     msi_free( prop );
2601     return ERROR_SUCCESS;
2602 }
2603
2604 static void msi_dialog_vsc_add_drives( msi_dialog *dialog, msi_control *control )
2605 {
2606     LPWSTR drives, ptr;
2607     DWORD size;
2608
2609     size = GetLogicalDriveStringsW( 0, NULL );
2610     if ( !size ) return;
2611
2612     drives = msi_alloc( (size + 1) * sizeof(WCHAR) );
2613     if ( !drives ) return;
2614
2615     GetLogicalDriveStringsW( size, drives );
2616
2617     ptr = drives;
2618     while (*ptr)
2619     {
2620         SendMessageW( control->hwnd, CB_ADDSTRING, 0, (LPARAM)ptr );
2621         ptr += lstrlenW(ptr) + 1;
2622     }
2623
2624     msi_free( drives );
2625 }
2626
2627 static UINT msi_dialog_volumeselect_combo( msi_dialog *dialog, MSIRECORD *rec )
2628 {
2629     msi_control *control;
2630     LPCWSTR prop;
2631     DWORD style;
2632
2633     /* FIXME: CBS_OWNERDRAWFIXED */
2634     style = WS_CHILD | WS_VISIBLE | WS_GROUP | WS_TABSTOP |
2635             CBS_DROPDOWNLIST | CBS_SORT | CBS_HASSTRINGS |
2636             WS_EX_LEFT | WS_EX_LTRREADING | WS_EX_RIGHTSCROLLBAR;
2637     control = msi_dialog_add_control( dialog, rec, WC_COMBOBOXW, style );
2638     if (!control)
2639         return ERROR_FUNCTION_FAILED;
2640
2641     control->attributes = MSI_RecordGetInteger( rec, 8 );
2642     control->handler = msi_dialog_volsel_handler;
2643     prop = MSI_RecordGetString( rec, 9 );
2644     control->property = msi_dialog_dup_property( dialog, prop, FALSE );
2645
2646     msi_dialog_vsc_add_drives( dialog, control );
2647
2648     return ERROR_SUCCESS;
2649 }
2650
2651 static const struct control_handler msi_dialog_handler[] =
2652 {
2653     { szText, msi_dialog_text_control },
2654     { szPushButton, msi_dialog_button_control },
2655     { szLine, msi_dialog_line_control },
2656     { szBitmap, msi_dialog_bitmap_control },
2657     { szCheckBox, msi_dialog_checkbox_control },
2658     { szScrollableText, msi_dialog_scrolltext_control },
2659     { szComboBox, msi_dialog_combo_control },
2660     { szEdit, msi_dialog_edit_control },
2661     { szMaskedEdit, msi_dialog_maskedit_control },
2662     { szPathEdit, msi_dialog_pathedit_control },
2663     { szProgressBar, msi_dialog_progress_bar },
2664     { szRadioButtonGroup, msi_dialog_radiogroup_control },
2665     { szIcon, msi_dialog_icon_control },
2666     { szSelectionTree, msi_dialog_selection_tree },
2667     { szGroupBox, msi_dialog_group_box },
2668     { szListBox, msi_dialog_list_box },
2669     { szDirectoryCombo, msi_dialog_directory_combo },
2670     { szDirectoryList, msi_dialog_directory_list },
2671     { szVolumeCostList, msi_dialog_volumecost_list },
2672     { szVolumeSelectCombo, msi_dialog_volumeselect_combo },
2673 };
2674
2675 #define NUM_CONTROL_TYPES (sizeof msi_dialog_handler/sizeof msi_dialog_handler[0])
2676
2677 static UINT msi_dialog_create_controls( MSIRECORD *rec, LPVOID param )
2678 {
2679     msi_dialog *dialog = param;
2680     LPCWSTR control_type;
2681     UINT i;
2682
2683     /* find and call the function that can create this type of control */
2684     control_type = MSI_RecordGetString( rec, 3 );
2685     for( i=0; i<NUM_CONTROL_TYPES; i++ )
2686         if (!strcmpiW( msi_dialog_handler[i].control_type, control_type ))
2687             break;
2688     if( i != NUM_CONTROL_TYPES )
2689         msi_dialog_handler[i].func( dialog, rec );
2690     else
2691         ERR("no handler for element type %s\n", debugstr_w(control_type));
2692
2693     return ERROR_SUCCESS;
2694 }
2695
2696 static UINT msi_dialog_fill_controls( msi_dialog *dialog )
2697 {
2698     static const WCHAR query[] = {
2699         'S','E','L','E','C','T',' ','*',' ',
2700         'F','R','O','M',' ','C','o','n','t','r','o','l',' ',
2701         'W','H','E','R','E',' ',
2702            '`','D','i','a','l','o','g','_','`',' ','=',' ','\'','%','s','\'',0};
2703     UINT r;
2704     MSIQUERY *view = NULL;
2705     MSIPACKAGE *package = dialog->package;
2706
2707     TRACE("%p %s\n", dialog, debugstr_w(dialog->name) );
2708
2709     /* query the Control table for all the elements of the control */
2710     r = MSI_OpenQuery( package->db, &view, query, dialog->name );
2711     if( r != ERROR_SUCCESS )
2712     {
2713         ERR("query failed for dialog %s\n", debugstr_w(dialog->name));
2714         return ERROR_INVALID_PARAMETER;
2715     }
2716
2717     r = MSI_IterateRecords( view, 0, msi_dialog_create_controls, dialog );
2718     msiobj_release( &view->hdr );
2719
2720     return r;
2721 }
2722
2723 static UINT msi_dialog_set_control_condition( MSIRECORD *rec, LPVOID param )
2724 {
2725     static const WCHAR szHide[] = { 'H','i','d','e',0 };
2726     static const WCHAR szShow[] = { 'S','h','o','w',0 };
2727     static const WCHAR szDisable[] = { 'D','i','s','a','b','l','e',0 };
2728     static const WCHAR szEnable[] = { 'E','n','a','b','l','e',0 };
2729     static const WCHAR szDefault[] = { 'D','e','f','a','u','l','t',0 };
2730     msi_dialog *dialog = param;
2731     msi_control *control;
2732     LPCWSTR name, action, condition;
2733     UINT r;
2734
2735     name = MSI_RecordGetString( rec, 2 );
2736     action = MSI_RecordGetString( rec, 3 );
2737     condition = MSI_RecordGetString( rec, 4 );
2738     r = MSI_EvaluateConditionW( dialog->package, condition );
2739     control = msi_dialog_find_control( dialog, name );
2740     if( r == MSICONDITION_TRUE && control )
2741     {
2742         TRACE("%s control %s\n", debugstr_w(action), debugstr_w(name));
2743
2744         /* FIXME: case sensitive? */
2745         if(!lstrcmpW(action, szHide))
2746             ShowWindow(control->hwnd, SW_HIDE);
2747         else if(!strcmpW(action, szShow))
2748             ShowWindow(control->hwnd, SW_SHOW);
2749         else if(!strcmpW(action, szDisable))
2750             EnableWindow(control->hwnd, FALSE);
2751         else if(!strcmpW(action, szEnable))
2752             EnableWindow(control->hwnd, TRUE);
2753         else if(!strcmpW(action, szDefault))
2754             SetFocus(control->hwnd);
2755         else
2756             FIXME("Unhandled action %s\n", debugstr_w(action));
2757     }
2758
2759     return ERROR_SUCCESS;
2760 }
2761
2762 static UINT msi_dialog_evaluate_control_conditions( msi_dialog *dialog )
2763 {
2764     static const WCHAR query[] = {
2765       'S','E','L','E','C','T',' ','*',' ',
2766       'F','R','O','M',' ',
2767         'C','o','n','t','r','o','l','C','o','n','d','i','t','i','o','n',' ',
2768       'W','H','E','R','E',' ',
2769         '`','D','i','a','l','o','g','_','`',' ','=',' ','\'','%','s','\'',0
2770     };
2771     UINT r;
2772     MSIQUERY *view = NULL;
2773     MSIPACKAGE *package = dialog->package;
2774
2775     TRACE("%p %s\n", dialog, debugstr_w(dialog->name) );
2776
2777     /* query the Control table for all the elements of the control */
2778     r = MSI_OpenQuery( package->db, &view, query, dialog->name );
2779     if( r != ERROR_SUCCESS )
2780         return ERROR_SUCCESS;
2781
2782     r = MSI_IterateRecords( view, 0, msi_dialog_set_control_condition, dialog );
2783     msiobj_release( &view->hdr );
2784
2785     return r;
2786 }
2787
2788 UINT msi_dialog_reset( msi_dialog *dialog )
2789 {
2790     /* FIXME: should restore the original values of any properties we changed */
2791     return msi_dialog_evaluate_control_conditions( dialog );
2792 }
2793
2794 /* figure out the height of 10 point MS Sans Serif */
2795 static INT msi_dialog_get_sans_serif_height( HWND hwnd )
2796 {
2797     static const WCHAR szSansSerif[] = {
2798         'M','S',' ','S','a','n','s',' ','S','e','r','i','f',0 };
2799     LOGFONTW lf;
2800     TEXTMETRICW tm;
2801     BOOL r;
2802     LONG height = 0;
2803     HFONT hFont, hOldFont;
2804     HDC hdc;
2805
2806     hdc = GetDC( hwnd );
2807     if (hdc)
2808     {
2809         memset( &lf, 0, sizeof lf );
2810         lf.lfHeight = MulDiv(10, GetDeviceCaps(hdc, LOGPIXELSY), 72);
2811         strcpyW( lf.lfFaceName, szSansSerif );
2812         hFont = CreateFontIndirectW(&lf);
2813         if (hFont)
2814         {
2815             hOldFont = SelectObject( hdc, hFont );
2816             r = GetTextMetricsW( hdc, &tm );
2817             if (r)
2818                 height = tm.tmHeight;
2819             SelectObject( hdc, hOldFont );
2820             DeleteObject( hFont );
2821         }
2822         ReleaseDC( hwnd, hdc );
2823     }
2824     return height;
2825 }
2826
2827 /* fetch the associated record from the Dialog table */
2828 static MSIRECORD *msi_get_dialog_record( msi_dialog *dialog )
2829 {
2830     static const WCHAR query[] = {
2831         'S','E','L','E','C','T',' ','*',' ',
2832         'F','R','O','M',' ','D','i','a','l','o','g',' ',
2833         'W','H','E','R','E',' ',
2834            '`','D','i','a','l','o','g','`',' ','=',' ','\'','%','s','\'',0};
2835     MSIPACKAGE *package = dialog->package;
2836     MSIRECORD *rec = NULL;
2837
2838     TRACE("%p %s\n", dialog, debugstr_w(dialog->name) );
2839
2840     rec = MSI_QueryGetRecord( package->db, query, dialog->name );
2841     if( !rec )
2842         ERR("query failed for dialog %s\n", debugstr_w(dialog->name));
2843
2844     return rec;
2845 }
2846
2847 static void msi_dialog_adjust_dialog_pos( msi_dialog *dialog, MSIRECORD *rec, LPRECT pos )
2848 {
2849     static const WCHAR szScreenX[] = {'S','c','r','e','e','n','X',0};
2850     static const WCHAR szScreenY[] = {'S','c','r','e','e','n','Y',0};
2851
2852     UINT xres, yres;
2853     POINT center;
2854     SIZE sz;
2855     LONG style;
2856
2857     center.x = MSI_RecordGetInteger( rec, 2 );
2858     center.y = MSI_RecordGetInteger( rec, 3 );
2859
2860     sz.cx = MSI_RecordGetInteger( rec, 4 );
2861     sz.cy = MSI_RecordGetInteger( rec, 5 );
2862
2863     sz.cx = msi_dialog_scale_unit( dialog, sz.cx );
2864     sz.cy = msi_dialog_scale_unit( dialog, sz.cy );
2865
2866     xres = msi_get_property_int( dialog->package, szScreenX, 0 );
2867     yres = msi_get_property_int( dialog->package, szScreenY, 0 );
2868
2869     center.x = MulDiv( center.x, xres, 100 );
2870     center.y = MulDiv( center.y, yres, 100 );
2871
2872     /* turn the client pos into the window rectangle */
2873     if (dialog->package->center_x && dialog->package->center_y)
2874     {
2875         pos->left = dialog->package->center_x - sz.cx / 2.0;
2876         pos->right = pos->left + sz.cx;
2877         pos->top = dialog->package->center_y - sz.cy / 2.0;
2878         pos->bottom = pos->top + sz.cy;
2879     }
2880     else
2881     {
2882         pos->left = center.x - sz.cx/2;
2883         pos->right = pos->left + sz.cx;
2884         pos->top = center.y - sz.cy/2;
2885         pos->bottom = pos->top + sz.cy;
2886
2887         /* save the center */
2888         dialog->package->center_x = center.x;
2889         dialog->package->center_y = center.y;
2890     }
2891
2892     dialog->size.cx = sz.cx;
2893     dialog->size.cy = sz.cy;
2894
2895     TRACE("%u %u %u %u\n", pos->left, pos->top, pos->right, pos->bottom);
2896
2897     style = GetWindowLongPtrW( dialog->hwnd, GWL_STYLE );
2898     AdjustWindowRect( pos, style, FALSE );
2899 }
2900
2901 static BOOL msi_control_set_next( msi_control *control, msi_control *next )
2902 {
2903     return SetWindowPos( next->hwnd, control->hwnd, 0, 0, 0, 0,
2904                          SWP_NOMOVE | SWP_NOOWNERZORDER | SWP_NOREDRAW |
2905                          SWP_NOREPOSITION | SWP_NOSENDCHANGING | SWP_NOSIZE );
2906 }
2907
2908 static UINT msi_dialog_set_tab_order( msi_dialog *dialog )
2909 {
2910     msi_control *control, *tab_next;
2911
2912     LIST_FOR_EACH_ENTRY( control, &dialog->controls, msi_control, entry )
2913     {
2914         tab_next = msi_dialog_find_control( dialog, control->tabnext );
2915         if( !tab_next )
2916             continue;
2917         msi_control_set_next( control, tab_next );
2918     }
2919
2920     return ERROR_SUCCESS;
2921 }
2922
2923 static void msi_dialog_set_first_control( msi_dialog* dialog, LPCWSTR name )
2924 {
2925     msi_control *control;
2926
2927     control = msi_dialog_find_control( dialog, name );
2928     if( control )
2929         dialog->hWndFocus = control->hwnd;
2930     else
2931         dialog->hWndFocus = NULL;
2932 }
2933
2934 static LRESULT msi_dialog_oncreate( HWND hwnd, LPCREATESTRUCTW cs )
2935 {
2936     static const WCHAR df[] = {
2937         'D','e','f','a','u','l','t','U','I','F','o','n','t',0 };
2938     static const WCHAR dfv[] = {
2939         'M','S',' ','S','h','e','l','l',' ','D','l','g',0 };
2940     msi_dialog *dialog = (msi_dialog*) cs->lpCreateParams;
2941     MSIRECORD *rec = NULL;
2942     LPWSTR title = NULL;
2943     RECT pos;
2944
2945     TRACE("%p %p\n", dialog, dialog->package);
2946
2947     dialog->hwnd = hwnd;
2948     SetWindowLongPtrW( hwnd, GWLP_USERDATA, (LONG_PTR) dialog );
2949
2950     rec = msi_get_dialog_record( dialog );
2951     if( !rec )
2952     {
2953         TRACE("No record found for dialog %s\n", debugstr_w(dialog->name));
2954         return -1;
2955     }
2956
2957     dialog->scale = msi_dialog_get_sans_serif_height(dialog->hwnd);
2958
2959     msi_dialog_adjust_dialog_pos( dialog, rec, &pos );
2960
2961     dialog->attributes = MSI_RecordGetInteger( rec, 6 );
2962
2963     dialog->default_font = msi_dup_property( dialog->package, df );
2964     if (!dialog->default_font)
2965     {
2966         dialog->default_font = strdupW(dfv);
2967         if (!dialog->default_font) return -1;
2968     }
2969
2970     title = msi_get_deformatted_field( dialog->package, rec, 7 );
2971     SetWindowTextW( hwnd, title );
2972     msi_free( title );
2973
2974     SetWindowPos( hwnd, 0, pos.left, pos.top,
2975                   pos.right - pos.left, pos.bottom - pos.top,
2976                   SWP_NOACTIVATE | SWP_NOZORDER | SWP_NOREDRAW );
2977
2978     msi_dialog_build_font_list( dialog );
2979     msi_dialog_fill_controls( dialog );
2980     msi_dialog_evaluate_control_conditions( dialog );
2981     msi_dialog_set_tab_order( dialog );
2982     msi_dialog_set_first_control( dialog, MSI_RecordGetString( rec, 8 ) );
2983     msiobj_release( &rec->hdr );
2984
2985     return 0;
2986 }
2987
2988 static UINT msi_dialog_send_event( msi_dialog *dialog, LPCWSTR event, LPCWSTR arg )
2989 {
2990     LPWSTR event_fmt = NULL, arg_fmt = NULL;
2991
2992     TRACE("Sending control event %s %s\n", debugstr_w(event), debugstr_w(arg));
2993
2994     deformat_string( dialog->package, event, &event_fmt );
2995     deformat_string( dialog->package, arg, &arg_fmt );
2996
2997     dialog->event_handler( dialog->package, event_fmt, arg_fmt, dialog );
2998
2999     msi_free( event_fmt );
3000     msi_free( arg_fmt );
3001
3002     return ERROR_SUCCESS;
3003 }
3004
3005 static UINT msi_dialog_set_property( msi_dialog *dialog, LPCWSTR event, LPCWSTR arg )
3006 {
3007     static const WCHAR szNullArg[] = { '{','}',0 };
3008     LPWSTR p, prop, arg_fmt = NULL;
3009     UINT len;
3010
3011     len = strlenW(event);
3012     prop = msi_alloc( len*sizeof(WCHAR));
3013     strcpyW( prop, &event[1] );
3014     p = strchrW( prop, ']' );
3015     if( p && p[1] == 0 )
3016     {
3017         *p = 0;
3018         if( strcmpW( szNullArg, arg ) )
3019             deformat_string( dialog->package, arg, &arg_fmt );
3020         MSI_SetPropertyW( dialog->package, prop, arg_fmt );
3021         msi_free( arg_fmt );
3022     }
3023     else
3024         ERR("Badly formatted property string - what happens?\n");
3025     msi_free( prop );
3026     return ERROR_SUCCESS;
3027 }
3028
3029 static UINT msi_dialog_control_event( MSIRECORD *rec, LPVOID param )
3030 {
3031     msi_dialog *dialog = param;
3032     LPCWSTR condition, event, arg;
3033     UINT r;
3034
3035     condition = MSI_RecordGetString( rec, 5 );
3036     r = MSI_EvaluateConditionW( dialog->package, condition );
3037     if( r == MSICONDITION_TRUE || r == MSICONDITION_NONE )
3038     {
3039         event = MSI_RecordGetString( rec, 3 );
3040         arg = MSI_RecordGetString( rec, 4 );
3041         if( event[0] == '[' )
3042             msi_dialog_set_property( dialog, event, arg );
3043         else
3044             msi_dialog_send_event( dialog, event, arg );
3045     }
3046
3047     return ERROR_SUCCESS;
3048 }
3049
3050 struct rec_list
3051 {
3052     struct list entry;
3053     MSIRECORD *rec;
3054 };
3055
3056 static UINT add_rec_to_list( MSIRECORD *rec, LPVOID param )
3057 {
3058     struct rec_list *add_rec;
3059     struct list *records = (struct list *)param;
3060
3061     msiobj_addref( &rec->hdr );
3062
3063     add_rec = msi_alloc( sizeof( *add_rec ) );
3064     if (!add_rec)
3065     {
3066         msiobj_release( &rec->hdr );
3067         return ERROR_OUTOFMEMORY;
3068     }
3069
3070     add_rec->rec = rec;
3071     list_add_tail( records, &add_rec->entry );
3072     return ERROR_SUCCESS;
3073 }
3074
3075 static inline void remove_rec_from_list( struct rec_list *rec_entry )
3076 {
3077     msiobj_release( &rec_entry->rec->hdr );
3078     list_remove( &rec_entry->entry );
3079     msi_free( rec_entry );
3080 }
3081
3082 static UINT msi_dialog_button_handler( msi_dialog *dialog,
3083                                        msi_control *control, WPARAM param )
3084 {
3085     static const WCHAR query[] = {
3086       'S','E','L','E','C','T',' ','*',' ',
3087       'F','R','O','M',' ','C','o','n','t','r','o','l','E','v','e','n','t',' ',
3088       'W','H','E','R','E',' ',
3089          '`','D','i','a','l','o','g','_','`',' ','=',' ','\'','%','s','\'',' ',
3090       'A','N','D',' ',
3091          '`','C','o','n','t','r','o','l','_','`',' ','=',' ','\'','%','s','\'',' ',
3092       'O','R','D','E','R',' ','B','Y',' ','`','O','r','d','e','r','i','n','g','`',0
3093     };
3094     MSIQUERY *view = NULL;
3095     struct rec_list *rec_entry, *next;
3096     struct list events;
3097     UINT r;
3098
3099     if( HIWORD(param) != BN_CLICKED )
3100         return ERROR_SUCCESS;
3101
3102     list_init( &events );
3103
3104     r = MSI_OpenQuery( dialog->package->db, &view, query,
3105                        dialog->name, control->name );
3106     if( r != ERROR_SUCCESS )
3107     {
3108         ERR("query failed\n");
3109         return 0;
3110     }
3111
3112     r = MSI_IterateRecords( view, 0, add_rec_to_list, &events );
3113     msiobj_release( &view->hdr );
3114     if (r != ERROR_SUCCESS)
3115         goto done;
3116
3117     /* handle all SetProperty events first */
3118     LIST_FOR_EACH_ENTRY_SAFE( rec_entry, next, &events, struct rec_list, entry )
3119     {
3120         LPCWSTR event = MSI_RecordGetString( rec_entry->rec, 3 );
3121
3122         if ( event[0] != '[' )
3123             continue;
3124
3125         r = msi_dialog_control_event( rec_entry->rec, dialog );
3126         remove_rec_from_list( rec_entry );
3127
3128         if ( r != ERROR_SUCCESS )
3129             goto done;
3130     }
3131
3132     /* handle all other events */
3133     LIST_FOR_EACH_ENTRY_SAFE( rec_entry, next, &events, struct rec_list, entry )
3134     {
3135         r = msi_dialog_control_event( rec_entry->rec, dialog );
3136         remove_rec_from_list( rec_entry );
3137
3138         if ( r != ERROR_SUCCESS )
3139             goto done;
3140     }
3141
3142 done:
3143     LIST_FOR_EACH_ENTRY_SAFE( rec_entry, next, &events, struct rec_list, entry )
3144     {
3145         remove_rec_from_list( rec_entry );
3146     }
3147
3148     return r;
3149 }
3150
3151 static UINT msi_dialog_get_checkbox_state( msi_dialog *dialog,
3152                 msi_control *control )
3153 {
3154     WCHAR state[2] = { 0 };
3155     DWORD sz = 2;
3156
3157     MSI_GetPropertyW( dialog->package, control->property, state, &sz );
3158     return state[0] ? 1 : 0;
3159 }
3160
3161 static void msi_dialog_set_checkbox_state( msi_dialog *dialog,
3162                 msi_control *control, UINT state )
3163 {
3164     static const WCHAR szState[] = { '1', 0 };
3165     LPCWSTR val;
3166
3167     /* if uncheck then the property is set to NULL */
3168     if (!state)
3169     {
3170         MSI_SetPropertyW( dialog->package, control->property, NULL );
3171         return;
3172     }
3173
3174     /* check for a custom state */
3175     if (control->value && control->value[0])
3176         val = control->value;
3177     else
3178         val = szState;
3179
3180     MSI_SetPropertyW( dialog->package, control->property, val );
3181 }
3182
3183 static void msi_dialog_checkbox_sync_state( msi_dialog *dialog,
3184                 msi_control *control )
3185 {
3186     UINT state;
3187
3188     state = msi_dialog_get_checkbox_state( dialog, control );
3189     SendMessageW( control->hwnd, BM_SETCHECK,
3190                   state ? BST_CHECKED : BST_UNCHECKED, 0 );
3191 }
3192
3193 static UINT msi_dialog_checkbox_handler( msi_dialog *dialog,
3194                 msi_control *control, WPARAM param )
3195 {
3196     UINT state;
3197
3198     if( HIWORD(param) != BN_CLICKED )
3199         return ERROR_SUCCESS;
3200
3201     TRACE("clicked checkbox %s, set %s\n", debugstr_w(control->name),
3202           debugstr_w(control->property));
3203
3204     state = msi_dialog_get_checkbox_state( dialog, control );
3205     state = state ? 0 : 1;
3206     msi_dialog_set_checkbox_state( dialog, control, state );
3207     msi_dialog_checkbox_sync_state( dialog, control );
3208
3209     return msi_dialog_button_handler( dialog, control, param );
3210 }
3211
3212 static UINT msi_dialog_edit_handler( msi_dialog *dialog,
3213                 msi_control *control, WPARAM param )
3214 {
3215     LPWSTR buf;
3216
3217     if( HIWORD(param) != EN_CHANGE )
3218         return ERROR_SUCCESS;
3219
3220     TRACE("edit %s contents changed, set %s\n", debugstr_w(control->name),
3221           debugstr_w(control->property));
3222
3223     buf = msi_get_window_text( control->hwnd );
3224     MSI_SetPropertyW( dialog->package, control->property, buf );
3225
3226     msi_free( buf );
3227
3228     return ERROR_SUCCESS;
3229 }
3230
3231 static UINT msi_dialog_radiogroup_handler( msi_dialog *dialog,
3232                 msi_control *control, WPARAM param )
3233 {
3234     if( HIWORD(param) != BN_CLICKED )
3235         return ERROR_SUCCESS;
3236
3237     TRACE("clicked radio button %s, set %s\n", debugstr_w(control->name),
3238           debugstr_w(control->property));
3239
3240     MSI_SetPropertyW( dialog->package, control->property, control->name );
3241
3242     return msi_dialog_button_handler( dialog, control, param );
3243 }
3244
3245 static LRESULT msi_dialog_oncommand( msi_dialog *dialog, WPARAM param, HWND hwnd )
3246 {
3247     msi_control *control = NULL;
3248
3249     TRACE("%p %p %08lx\n", dialog, hwnd, param);
3250
3251     switch (param)
3252     {
3253     case 1: /* enter */
3254         control = msi_dialog_find_control( dialog, dialog->control_default );
3255         break;
3256     case 2: /* escape */
3257         control = msi_dialog_find_control( dialog, dialog->control_cancel );
3258         break;
3259     default:
3260         control = msi_dialog_find_control_by_hwnd( dialog, hwnd );
3261     }
3262
3263     if( control )
3264     {
3265         if( control->handler )
3266         {
3267             control->handler( dialog, control, param );
3268             msi_dialog_evaluate_control_conditions( dialog );
3269         }
3270     }
3271
3272     return 0;
3273 }
3274
3275 static LRESULT msi_dialog_onnotify( msi_dialog *dialog, LPARAM param )
3276 {
3277     LPNMHDR nmhdr = (LPNMHDR) param;
3278     msi_control *control = msi_dialog_find_control_by_hwnd( dialog, nmhdr->hwndFrom );
3279
3280     TRACE("%p %p\n", dialog, nmhdr->hwndFrom);
3281
3282     if ( control && control->handler )
3283         control->handler( dialog, control, param );
3284
3285     return 0;
3286 }
3287
3288 static void msi_dialog_setfocus( msi_dialog *dialog )
3289 {
3290     HWND hwnd = dialog->hWndFocus;
3291
3292     hwnd = GetNextDlgTabItem( dialog->hwnd, hwnd, TRUE);
3293     hwnd = GetNextDlgTabItem( dialog->hwnd, hwnd, FALSE);
3294     SetFocus( hwnd );
3295     dialog->hWndFocus = hwnd;
3296 }
3297
3298 static LRESULT WINAPI MSIDialog_WndProc( HWND hwnd, UINT msg,
3299                 WPARAM wParam, LPARAM lParam )
3300 {
3301     msi_dialog *dialog = (LPVOID) GetWindowLongPtrW( hwnd, GWLP_USERDATA );
3302
3303     TRACE("0x%04x\n", msg);
3304
3305     switch (msg)
3306     {
3307     case WM_MOVE:
3308         dialog->package->center_x = LOWORD(lParam) + dialog->size.cx / 2.0;
3309         dialog->package->center_y = HIWORD(lParam) + dialog->size.cy / 2.0;
3310         break;
3311
3312     case WM_CREATE:
3313         return msi_dialog_oncreate( hwnd, (LPCREATESTRUCTW)lParam );
3314
3315     case WM_COMMAND:
3316         return msi_dialog_oncommand( dialog, wParam, (HWND)lParam );
3317
3318     case WM_ACTIVATE:
3319         if( LOWORD(wParam) == WA_INACTIVE )
3320             dialog->hWndFocus = GetFocus();
3321         else
3322             msi_dialog_setfocus( dialog );
3323         return 0;
3324
3325     case WM_SETFOCUS:
3326         msi_dialog_setfocus( dialog );
3327         return 0;
3328
3329     /* bounce back to our subclassed static control */
3330     case WM_CTLCOLORSTATIC:
3331         return SendMessageW( (HWND) lParam, WM_CTLCOLORSTATIC, wParam, lParam );
3332
3333     case WM_DESTROY:
3334         dialog->hwnd = NULL;
3335         return 0;
3336     case WM_NOTIFY:
3337         return msi_dialog_onnotify( dialog, lParam );
3338     }
3339     return DefWindowProcW(hwnd, msg, wParam, lParam);
3340 }
3341
3342 static BOOL CALLBACK msi_radioground_child_enum( HWND hWnd, LPARAM lParam )
3343 {
3344     EnableWindow( hWnd, lParam );
3345     return TRUE;
3346 }
3347
3348 static LRESULT WINAPI MSIRadioGroup_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
3349 {
3350     WNDPROC oldproc = (WNDPROC) GetPropW(hWnd, szButtonData);
3351     LRESULT r;
3352
3353     TRACE("hWnd %p msg %04x wParam 0x%08lx lParam 0x%08lx\n", hWnd, msg, wParam, lParam);
3354
3355     if (msg == WM_COMMAND) /* Forward notifications to dialog */
3356         SendMessageW(GetParent(hWnd), msg, wParam, lParam);
3357
3358     r = CallWindowProcW(oldproc, hWnd, msg, wParam, lParam);
3359
3360     /* make sure the radio buttons show as disabled if the parent is disabled */
3361     if (msg == WM_ENABLE)
3362         EnumChildWindows( hWnd, msi_radioground_child_enum, wParam );
3363
3364     return r;
3365 }
3366
3367 static LRESULT WINAPI MSIHiddenWindowProc( HWND hwnd, UINT msg,
3368                 WPARAM wParam, LPARAM lParam )
3369 {
3370     msi_dialog *dialog = (msi_dialog*) lParam;
3371
3372     TRACE("%d %p\n", msg, dialog);
3373
3374     switch (msg)
3375     {
3376     case WM_MSI_DIALOG_CREATE:
3377         return msi_dialog_run_message_loop( dialog );
3378     case WM_MSI_DIALOG_DESTROY:
3379         msi_dialog_destroy( dialog );
3380         return 0;
3381     }
3382     return DefWindowProcW( hwnd, msg, wParam, lParam );
3383 }
3384
3385 /* functions that interface to other modules within MSI */
3386
3387 msi_dialog *msi_dialog_create( MSIPACKAGE* package,
3388                                LPCWSTR szDialogName, msi_dialog *parent,
3389                                msi_dialog_event_handler event_handler )
3390 {
3391     MSIRECORD *rec = NULL;
3392     msi_dialog *dialog;
3393
3394     TRACE("%p %s\n", package, debugstr_w(szDialogName));
3395
3396     /* allocate the structure for the dialog to use */
3397     dialog = msi_alloc_zero( sizeof *dialog + sizeof(WCHAR)*strlenW(szDialogName) );
3398     if( !dialog )
3399         return NULL;
3400     strcpyW( dialog->name, szDialogName );
3401     dialog->parent = parent;
3402     msiobj_addref( &package->hdr );
3403     dialog->package = package;
3404     dialog->event_handler = event_handler;
3405     dialog->finished = 0;
3406     list_init( &dialog->controls );
3407
3408     /* verify that the dialog exists */
3409     rec = msi_get_dialog_record( dialog );
3410     if( !rec )
3411     {
3412         msiobj_release( &package->hdr );
3413         msi_free( dialog );
3414         return NULL;
3415     }
3416     dialog->attributes = MSI_RecordGetInteger( rec, 6 );
3417     dialog->control_default = strdupW( MSI_RecordGetString( rec, 9 ) );
3418     dialog->control_cancel = strdupW( MSI_RecordGetString( rec, 10 ) );
3419     msiobj_release( &rec->hdr );
3420
3421     return dialog;
3422 }
3423
3424 static void msi_process_pending_messages( HWND hdlg )
3425 {
3426     MSG msg;
3427
3428     while( PeekMessageW( &msg, 0, 0, 0, PM_REMOVE ) )
3429     {
3430         if( hdlg && IsDialogMessageW( hdlg, &msg ))
3431             continue;
3432         TranslateMessage( &msg );
3433         DispatchMessageW( &msg );
3434     }
3435 }
3436
3437 void msi_dialog_end_dialog( msi_dialog *dialog )
3438 {
3439     TRACE("%p\n", dialog);
3440     dialog->finished = 1;
3441     PostMessageW(dialog->hwnd, WM_NULL, 0, 0);
3442 }
3443
3444 void msi_dialog_check_messages( HANDLE handle )
3445 {
3446     DWORD r;
3447
3448     /* in threads other than the UI thread, block */
3449     if( uiThreadId != GetCurrentThreadId() )
3450     {
3451         if( handle )
3452             WaitForSingleObject( handle, INFINITE );
3453         return;
3454     }
3455
3456     /* there's two choices for the UI thread */
3457     while (1)
3458     {
3459         msi_process_pending_messages( NULL );
3460
3461         if( !handle )
3462             break;
3463
3464         /*
3465          * block here until somebody creates a new dialog or
3466          * the handle we're waiting on becomes ready
3467          */
3468         r = MsgWaitForMultipleObjects( 1, &handle, 0, INFINITE, QS_ALLINPUT );
3469         if( r == WAIT_OBJECT_0 )
3470             break;
3471     }
3472 }
3473
3474 UINT msi_dialog_run_message_loop( msi_dialog *dialog )
3475 {
3476     DWORD style;
3477     HWND hwnd;
3478
3479     if( uiThreadId != GetCurrentThreadId() )
3480         return SendMessageW( hMsiHiddenWindow, WM_MSI_DIALOG_CREATE, 0, (LPARAM) dialog );
3481
3482     /* create the dialog window, don't show it yet */
3483     style = WS_OVERLAPPED;
3484     if( dialog->attributes & msidbDialogAttributesVisible )
3485         style |= WS_VISIBLE;
3486
3487     hwnd = CreateWindowW( szMsiDialogClass, dialog->name, style,
3488                      CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
3489                      NULL, NULL, NULL, dialog );
3490     if( !hwnd )
3491     {
3492         ERR("Failed to create dialog %s\n", debugstr_w( dialog->name ));
3493         return ERROR_FUNCTION_FAILED;
3494     }
3495
3496     ShowWindow( hwnd, SW_SHOW );
3497     /* UpdateWindow( hwnd ); - and causes the transparent static controls not to paint */
3498
3499     if( dialog->attributes & msidbDialogAttributesModal )
3500     {
3501         while( !dialog->finished )
3502         {
3503             MsgWaitForMultipleObjects( 0, NULL, 0, INFINITE, QS_ALLINPUT );
3504             msi_process_pending_messages( dialog->hwnd );
3505         }
3506     }
3507     else
3508         return ERROR_IO_PENDING;
3509
3510     return ERROR_SUCCESS;
3511 }
3512
3513 void msi_dialog_do_preview( msi_dialog *dialog )
3514 {
3515     TRACE("\n");
3516     dialog->attributes |= msidbDialogAttributesVisible;
3517     dialog->attributes &= ~msidbDialogAttributesModal;
3518     msi_dialog_run_message_loop( dialog );
3519 }
3520
3521 void msi_dialog_destroy( msi_dialog *dialog )
3522 {
3523     if( uiThreadId != GetCurrentThreadId() )
3524     {
3525         SendMessageW( hMsiHiddenWindow, WM_MSI_DIALOG_DESTROY, 0, (LPARAM) dialog );
3526         return;
3527     }
3528
3529     if( dialog->hwnd )
3530         ShowWindow( dialog->hwnd, SW_HIDE );
3531
3532     if( dialog->hwnd )
3533         DestroyWindow( dialog->hwnd );
3534
3535     /* unsubscribe events */
3536     ControlEvent_CleanupDialogSubscriptions(dialog->package, dialog->name);
3537
3538     /* destroy the list of controls */
3539     while( !list_empty( &dialog->controls ) )
3540     {
3541         msi_control *t;
3542
3543         t = LIST_ENTRY( list_head( &dialog->controls ),
3544                         msi_control, entry );
3545         msi_destroy_control( t );
3546     }
3547
3548     /* destroy the list of fonts */
3549     while( dialog->font_list )
3550     {
3551         msi_font *t = dialog->font_list;
3552         dialog->font_list = t->next;
3553         DeleteObject( t->hfont );
3554         msi_free( t );
3555     }
3556     msi_free( dialog->default_font );
3557
3558     msi_free( dialog->control_default );
3559     msi_free( dialog->control_cancel );
3560     msiobj_release( &dialog->package->hdr );
3561     dialog->package = NULL;
3562     msi_free( dialog );
3563 }
3564
3565 BOOL msi_dialog_register_class( void )
3566 {
3567     WNDCLASSW cls;
3568
3569     ZeroMemory( &cls, sizeof cls );
3570     cls.lpfnWndProc   = MSIDialog_WndProc;
3571     cls.hInstance     = NULL;
3572     cls.hIcon         = LoadIconW(0, (LPWSTR)IDI_APPLICATION);
3573     cls.hCursor       = LoadCursorW(0, (LPWSTR)IDC_ARROW);
3574     cls.hbrBackground = (HBRUSH)(COLOR_3DFACE + 1);
3575     cls.lpszMenuName  = NULL;
3576     cls.lpszClassName = szMsiDialogClass;
3577
3578     if( !RegisterClassW( &cls ) )
3579         return FALSE;
3580
3581     cls.lpfnWndProc   = MSIHiddenWindowProc;
3582     cls.lpszClassName = szMsiHiddenWindow;
3583
3584     if( !RegisterClassW( &cls ) )
3585         return FALSE;
3586
3587     uiThreadId = GetCurrentThreadId();
3588
3589     hMsiHiddenWindow = CreateWindowW( szMsiHiddenWindow, NULL, WS_OVERLAPPED,
3590                                    0, 0, 100, 100, NULL, NULL, NULL, NULL );
3591     if( !hMsiHiddenWindow )
3592         return FALSE;
3593
3594     return TRUE;
3595 }
3596
3597 void msi_dialog_unregister_class( void )
3598 {
3599     DestroyWindow( hMsiHiddenWindow );
3600     hMsiHiddenWindow = NULL;
3601     UnregisterClassW( szMsiDialogClass, NULL );
3602     UnregisterClassW( szMsiHiddenWindow, NULL );
3603     uiThreadId = 0;
3604 }
3605
3606 static UINT error_dialog_handler(MSIPACKAGE *package, LPCWSTR event,
3607                                  LPCWSTR argument, msi_dialog* dialog)
3608 {
3609     static const WCHAR end_dialog[] = {'E','n','d','D','i','a','l','o','g',0};
3610     static const WCHAR error_abort[] = {'E','r','r','o','r','A','b','o','r','t',0};
3611     static const WCHAR error_cancel[] = {'E','r','r','o','r','C','a','n','c','e','l',0};
3612     static const WCHAR error_no[] = {'E','r','r','o','r','N','o',0};
3613     static const WCHAR result_prop[] = {
3614         'M','S','I','E','r','r','o','r','D','i','a','l','o','g','R','e','s','u','l','t',0
3615     };
3616
3617     if ( lstrcmpW( event, end_dialog ) )
3618         return ERROR_SUCCESS;
3619
3620     if ( !lstrcmpW( argument, error_abort ) || !lstrcmpW( argument, error_cancel ) ||
3621          !lstrcmpW( argument, error_no ) )
3622     {
3623          MSI_SetPropertyW( package, result_prop, error_abort );
3624     }
3625
3626     ControlEvent_CleanupSubscriptions(package);
3627     msi_dialog_end_dialog( dialog );
3628
3629     return ERROR_SUCCESS;
3630 }
3631
3632 static UINT msi_error_dialog_set_error( MSIPACKAGE *package, LPWSTR error_dialog, LPWSTR error )
3633 {
3634     MSIRECORD * row;
3635
3636     static const WCHAR update[] =
3637         {'U','P','D','A','T','E',' ','`','C','o','n','t','r','o','l','`',' ',
3638          'S','E','T',' ','`','T','e','x','t','`',' ','=',' ','\'','%','s','\'',' ',
3639          'W','H','E','R','E', ' ','`','D','i','a','l','o','g','_','`',' ','=',' ','\'','%','s','\'',' ',
3640          'A','N','D',' ','`','C','o','n','t','r','o','l','`',' ','=',' ',
3641          '\'','E','r','r','o','r','T','e','x','t','\'',0};
3642
3643     row = MSI_QueryGetRecord( package->db, update, error, error_dialog );
3644     if (!row)
3645         return ERROR_FUNCTION_FAILED;
3646
3647     msiobj_release(&row->hdr);
3648     return ERROR_SUCCESS;
3649 }
3650
3651 UINT msi_spawn_error_dialog( MSIPACKAGE *package, LPWSTR error_dialog, LPWSTR error )
3652 {
3653     msi_dialog *dialog;
3654     WCHAR result[MAX_PATH];
3655     UINT r = ERROR_SUCCESS;
3656     DWORD size = MAX_PATH;
3657     int res;
3658
3659     static const WCHAR szUILevel[] = {'U','I','L','e','v','e','l',0};
3660     static const WCHAR pn_prop[] = {'P','r','o','d','u','c','t','N','a','m','e',0};
3661     static const WCHAR title_fmt[] = {'%','s',' ','W','a','r','n','i','n','g',0};
3662     static const WCHAR error_abort[] = {'E','r','r','o','r','A','b','o','r','t',0};
3663     static const WCHAR result_prop[] = {
3664         'M','S','I','E','r','r','o','r','D','i','a','l','o','g','R','e','s','u','l','t',0
3665     };
3666
3667     if ( (msi_get_property_int(package, szUILevel, 0) & INSTALLUILEVEL_MASK) == INSTALLUILEVEL_NONE )
3668         return ERROR_SUCCESS;
3669
3670     if ( !error_dialog )
3671     {
3672         LPWSTR product_name = msi_dup_property( package, pn_prop );
3673         WCHAR title[MAX_PATH];
3674
3675         sprintfW( title, title_fmt, product_name );
3676         res = MessageBoxW( NULL, error, title, MB_OKCANCEL | MB_ICONWARNING );
3677
3678         msi_free( product_name );
3679
3680         if ( res == IDOK )
3681             return ERROR_SUCCESS;
3682         else
3683             return ERROR_FUNCTION_FAILED;
3684     }
3685
3686     r = msi_error_dialog_set_error( package, error_dialog, error );
3687     if ( r != ERROR_SUCCESS )
3688         return r;
3689
3690     dialog = msi_dialog_create( package, error_dialog, package->dialog,
3691                                 error_dialog_handler );
3692     if ( !dialog )
3693         return ERROR_FUNCTION_FAILED;
3694
3695     dialog->finished = FALSE;
3696     r = msi_dialog_run_message_loop( dialog );
3697     if ( r != ERROR_SUCCESS )
3698         goto done;
3699
3700     r = MSI_GetPropertyW( package, result_prop, result, &size );
3701     if ( r != ERROR_SUCCESS)
3702         r = ERROR_SUCCESS;
3703
3704     if ( !lstrcmpW( result, error_abort ) )
3705         r = ERROR_FUNCTION_FAILED;
3706
3707 done:
3708     msi_dialog_destroy( dialog );
3709
3710     return r;
3711 }