dlls/msi/dialog.c

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