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