Sync to Wine-0_9_3:
[reactos.git] / reactos / lib / msi / dialog.c
index 69e2b1e..87e14e8 100644 (file)
@@ -19,6 +19,8 @@
  */
 
 #define COBJMACROS
+#define NONAMELESSUNION
+#define NONAMELESSSTRUCT
 
 #include <stdarg.h>
 
@@ -33,6 +35,7 @@
 #include "ocidl.h"
 #include "olectl.h"
 #include "richedit.h"
+#include "commctrl.h"
 
 #include "wine/debug.h"
 #include "wine/unicode.h"
@@ -48,13 +51,15 @@ typedef UINT (*msi_handler)( msi_dialog *, msi_control *, WPARAM );
 
 struct msi_control_tag
 {
-    struct msi_control_tag *next;
+    struct list entry;
     HWND hwnd;
     msi_handler handler;
     LPWSTR property;
     LPWSTR value;
-    IPicture *pic;
+    HBITMAP hBitmap;
     HICON hIcon;
+    LPWSTR tabnext;
+    HMODULE hDll;
     WCHAR name[1];
 };
 
@@ -75,7 +80,10 @@ struct msi_dialog_tag
     HWND hwnd;
     LPWSTR default_font;
     msi_font *font_list;
-    msi_control *control_list;
+    struct list controls;
+    HWND hWndFocus;
+    LPWSTR control_default;
+    LPWSTR control_cancel;
     WCHAR name[1];
 };
 
@@ -98,9 +106,9 @@ const WCHAR szMsiDialogClass[] = {
 };
 const WCHAR szMsiHiddenWindow[] = {
     'M','s','i','H','i','d','d','e','n','W','i','n','d','o','w',0 };
-const static WCHAR szStatic[] = { 'S','t','a','t','i','c',0 };
-const static WCHAR szButton[] = { 'B','U','T','T','O','N', 0 };
-const static WCHAR szButtonData[] = { 'M','S','I','D','A','T','A',0 };
+static const WCHAR szStatic[] = { 'S','t','a','t','i','c',0 };
+static const WCHAR szButton[] = { 'B','U','T','T','O','N', 0 };
+static const WCHAR szButtonData[] = { 'M','S','I','D','A','T','A',0 };
 static const WCHAR szText[] = { 'T','e','x','t',0 };
 static const WCHAR szPushButton[] = { 'P','u','s','h','B','u','t','t','o','n',0 };
 static const WCHAR szLine[] = { 'L','i','n','e',0 };
@@ -112,9 +120,13 @@ static const WCHAR szComboBox[] = { 'C','o','m','b','o','B','o','x',0 };
 static const WCHAR szEdit[] = { 'E','d','i','t',0 };
 static const WCHAR szMaskedEdit[] = { 'M','a','s','k','e','d','E','d','i','t',0 };
 static const WCHAR szPathEdit[] = { 'P','a','t','h','E','d','i','t',0 };
+static const WCHAR szProgressBar[] = {
+     'P','r','o','g','r','e','s','s','B','a','r',0 };
 static const WCHAR szRadioButtonGroup[] = { 
     'R','a','d','i','o','B','u','t','t','o','n','G','r','o','u','p',0 };
 static const WCHAR szIcon[] = { 'I','c','o','n',0 };
+static const WCHAR szSelectionTree[] = {
+    'S','e','l','e','c','t','i','o','n','T','r','e','e',0 };
 
 static UINT msi_dialog_checkbox_handler( msi_dialog *, msi_control *, WPARAM );
 static void msi_dialog_checkbox_sync_state( msi_dialog *, msi_control * );
@@ -132,7 +144,6 @@ static LRESULT WINAPI MSIRadioGroup_WndProc(HWND hWnd, UINT msg, WPARAM wParam,
 
 static DWORD uiThreadId;
 static HWND hMsiHiddenWindow;
-static HMODULE hRichedit;
 
 static INT msi_dialog_scale_unit( msi_dialog *dialog, INT val )
 {
@@ -143,20 +154,32 @@ static msi_control *msi_dialog_find_control( msi_dialog *dialog, LPCWSTR name )
 {
     msi_control *control;
 
-    for( control = dialog->control_list; control; control = control->next )
+    if( !name )
+        return NULL;
+    LIST_FOR_EACH_ENTRY( control, &dialog->controls, msi_control, entry )
         if( !strcmpW( control->name, name ) ) /* FIXME: case sensitive? */
-            break;
-    return control;
+            return control;
+    return NULL;
 }
 
 static msi_control *msi_dialog_find_control_by_hwnd( msi_dialog *dialog, HWND hwnd )
 {
     msi_control *control;
 
-    for( control = dialog->control_list; control; control = control->next )
+    LIST_FOR_EACH_ENTRY( control, &dialog->controls, msi_control, entry )
         if( hwnd == control->hwnd )
-            break;
-    return control;
+            return control;
+    return NULL;
+}
+
+static LPWSTR msi_get_deformatted_field( MSIPACKAGE *package, MSIRECORD *rec, int field )
+{
+    LPCWSTR str = MSI_RecordGetString( rec, field );
+    LPWSTR ret = NULL;
+
+    if (str)
+        deformat_string( package, str, &ret );
+    return ret;
 }
 
 /*
@@ -165,23 +188,32 @@ static msi_control *msi_dialog_find_control_by_hwnd( msi_dialog *dialog, HWND hw
  * Extract the {\style} string from the front of the text to display and
  *  update the pointer.
  */
-static LPWSTR msi_dialog_get_style( LPCWSTR *text )
+static LPWSTR msi_dialog_get_style( LPCWSTR p, LPCWSTR *rest )
 {
     LPWSTR ret = NULL;
-    LPCWSTR p = *text, q;
+    LPCWSTR q, i;
     DWORD len;
 
+    *rest = p;
+    if( !p )
+        return ret;
     if( *p++ != '{' )
         return ret;
     q = strchrW( p, '}' );
     if( !q )
         return ret;
-    *text = ++q;
-    if( *p++ != '\\' )
-        return ret;
-    len = q - p;
+    if( *p == '\\' || *p == '&' )
+        p++;
+
+    /* little bit of sanity checking to stop us getting confused with RTF */
+    for( i=p; i<q; i++ )
+        if( *i == '}' || *i == '\\' )
+            return ret;
     
-    ret = HeapAlloc( GetProcessHeap(), 0, len*sizeof(WCHAR) );
+    *rest = ++q;
+    len = q - p;
+
+    ret = msi_alloc( len*sizeof(WCHAR) );
     if( !ret )
         return ret;
     memcpy( ret, p, len*sizeof(WCHAR) );
@@ -200,8 +232,7 @@ static UINT msi_dialog_add_font( MSIRECORD *rec, LPVOID param )
 
     /* create a font and add it to the list */
     name = MSI_RecordGetString( rec, 1 );
-    font = HeapAlloc( GetProcessHeap(), 0,
-                      sizeof *font + strlenW( name )*sizeof (WCHAR) );
+    font = msi_alloc( sizeof *font + strlenW( name )*sizeof (WCHAR) );
     strcpyW( font->name, name );
     font->next = dialog->font_list;
     dialog->font_list = font;
@@ -284,21 +315,22 @@ static msi_control *msi_dialog_create_window( msi_dialog *dialog,
                 DWORD style, HWND parent )
 {
     DWORD x, y, width, height;
-    LPWSTR font = NULL, title = NULL;
+    LPWSTR font = NULL, title_font = NULL;
+    LPCWSTR title = NULL;
     msi_control *control;
 
     style |= WS_CHILD;
 
-    control = HeapAlloc( GetProcessHeap(), 0,
-                         sizeof *control + strlenW(name)*sizeof(WCHAR) );
+    control = msi_alloc( sizeof *control + strlenW(name)*sizeof(WCHAR) );
     strcpyW( control->name, name );
-    control->next = dialog->control_list;
-    dialog->control_list = control;
+    list_add_head( &dialog->controls, &control->entry );
     control->handler = NULL;
     control->property = NULL;
     control->value = NULL;
-    control->pic = NULL;
+    control->hBitmap = NULL;
     control->hIcon = NULL;
+    control->hDll = NULL;
+    control->tabnext = strdupW( MSI_RecordGetString( rec, 11) );
 
     x = MSI_RecordGetInteger( rec, 4 );
     y = MSI_RecordGetInteger( rec, 5 );
@@ -312,8 +344,8 @@ static msi_control *msi_dialog_create_window( msi_dialog *dialog,
 
     if( text )
     {
-        font = msi_dialog_get_style( &text );
-        deformat_string( dialog->package, text, &title );
+        deformat_string( dialog->package, text, &title_font );
+        font = msi_dialog_get_style( title_font, &title );
     }
 
     control->hwnd = CreateWindowW( szCls, title, style,
@@ -325,12 +357,103 @@ static msi_control *msi_dialog_create_window( msi_dialog *dialog,
     msi_dialog_set_font( dialog, control->hwnd,
                          font ? font : dialog->default_font );
 
-    HeapFree( GetProcessHeap(), 0, font );
-    HeapFree( GetProcessHeap(), 0, title );
+    msi_free( title_font );
+    msi_free( font );
 
     return control;
 }
 
+static MSIRECORD *msi_get_binary_record( MSIDATABASE *db, LPCWSTR name )
+{
+    static const WCHAR query[] = {
+        's','e','l','e','c','t',' ','*',' ',
+        'f','r','o','m',' ','B','i','n','a','r','y',' ',
+        'w','h','e','r','e',' ',
+            '`','N','a','m','e','`',' ','=',' ','\'','%','s','\'',0
+    };
+
+    return MSI_QueryGetRecord( db, query, name );
+}
+
+static LPWSTR msi_create_tmp_path(void)
+{
+    WCHAR tmp[MAX_PATH];
+    LPWSTR path = NULL;
+    static const WCHAR prefix[] = { 'm','s','i',0 };
+    DWORD len, r;
+
+    r = GetTempPathW( MAX_PATH, tmp );
+    if( !r )
+        return path;
+    len = lstrlenW( tmp ) + 20;
+    path = msi_alloc( len * sizeof (WCHAR) );
+    if( path )
+    {
+        r = GetTempFileNameW( tmp, prefix, 0, path );
+        if (!r)
+        {
+            msi_free( path );
+            path = NULL;
+        }
+    }
+    return path;
+}
+
+
+static HANDLE msi_load_image( MSIDATABASE *db, LPCWSTR name, UINT type,
+                              UINT cx, UINT cy, UINT flags )
+{
+    MSIRECORD *rec = NULL;
+    HANDLE himage = NULL;
+    LPWSTR tmp;
+    UINT r;
+
+    TRACE("%p %s %u %u %08x\n", db, debugstr_w(name), cx, cy, flags);
+
+    tmp = msi_create_tmp_path();
+    if( !tmp )
+        return himage;
+
+    rec = msi_get_binary_record( db, name );
+    if( rec )
+    {
+        r = MSI_RecordStreamToFile( rec, 2, tmp );
+        if( r == ERROR_SUCCESS )
+        {
+            himage = LoadImageW( 0, tmp, type, cx, cy, flags );
+            DeleteFileW( tmp );
+        }
+        msiobj_release( &rec->hdr );
+    }
+
+    msi_free( tmp );
+    return himage;
+}
+
+static HICON msi_load_icon( MSIDATABASE *db, LPCWSTR text, UINT attributes )
+{
+    DWORD cx = 0, cy = 0, flags;
+
+    flags = LR_LOADFROMFILE | LR_DEFAULTSIZE;
+    if( attributes & msidbControlAttributesFixedSize )
+    {
+        flags &= ~LR_DEFAULTSIZE;
+        if( attributes & msidbControlAttributesIconSize16 )
+        {
+            cx += 16;
+            cy += 16;
+        }
+        if( attributes & msidbControlAttributesIconSize32 )
+        {
+            cx += 32;
+            cy += 32;
+        }
+        /* msidbControlAttributesIconSize48 handled by above logic */
+    }
+    return msi_load_image( db, text, IMAGE_ICON, cx, cy, flags );
+}
+
+
 /* called from the Control Event subscription code */
 void msi_dialog_handle_event( msi_dialog* dialog, LPCWSTR control, 
                               LPCWSTR attribute, MSIRECORD *rec )
@@ -342,7 +465,10 @@ void msi_dialog_handle_event( msi_dialog* dialog, LPCWSTR control,
     if (!ctrl)
         return;
     if( lstrcmpW(attribute, szText) )
+    {
+        ERR("Attribute %s\n", debugstr_w(attribute));
         return;
+    }
     text = MSI_RecordGetString( rec , 1 );
     SetWindowTextW( ctrl->hwnd, text );
     msi_dialog_check_messages( NULL );
@@ -438,7 +564,7 @@ MSIText_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
         msi_text_on_settext( hWnd );
         break;
     case WM_NCDESTROY:
-        HeapFree( GetProcessHeap(), 0, info );
+        msi_free( info );
         RemovePropW( hWnd, szButtonData );
         break;
     }
@@ -457,7 +583,7 @@ static UINT msi_dialog_text_control( msi_dialog *dialog, MSIRECORD *rec )
     if( !control )
         return ERROR_FUNCTION_FAILED;
 
-    info = HeapAlloc( GetProcessHeap(), 0, sizeof *info );
+    info = msi_alloc( sizeof *info );
     if( !info )
         return ERROR_SUCCESS;
 
@@ -475,18 +601,35 @@ static UINT msi_dialog_text_control( msi_dialog *dialog, MSIRECORD *rec )
 static UINT msi_dialog_button_control( msi_dialog *dialog, MSIRECORD *rec )
 {
     msi_control *control;
+    UINT attributes, style;
+    LPWSTR text;
 
     TRACE("%p %p\n", dialog, rec);
 
-    control = msi_dialog_add_control( dialog, rec, szButton, WS_TABSTOP );
+    style = WS_TABSTOP;
+    attributes = MSI_RecordGetInteger( rec, 8 );
+    if( attributes & msidbControlAttributesIcon )
+        style |= BS_ICON;
+
+    control = msi_dialog_add_control( dialog, rec, szButton, style );
+    if( !control )
+        return ERROR_FUNCTION_FAILED;
+
     control->handler = msi_dialog_button_handler;
 
+    /* set the icon */
+    text = msi_get_deformatted_field( dialog->package, rec, 10 );
+    control->hIcon = msi_load_icon( dialog->package->db, text, attributes );
+    if( attributes & msidbControlAttributesIcon )
+        SendMessageW( control->hwnd, BM_SETIMAGE, IMAGE_ICON, (LPARAM) control->hIcon );
+    msi_free( text );
+
     return ERROR_SUCCESS;
 }
 
 static LPWSTR msi_get_checkbox_value( msi_dialog *dialog, LPCWSTR prop )
 {
-    const static WCHAR query[] = {
+    static const WCHAR query[] = {
         'S','E','L','E','C','T',' ','*',' ',
         'F','R','O','M',' ','`','C','h','e','c','k','B','o','x',' ','`',
         'W','H','E','R','E',' ',
@@ -494,7 +637,6 @@ static LPWSTR msi_get_checkbox_value( msi_dialog *dialog, LPCWSTR prop )
         '\'','%','s','\'',0
     };
     MSIRECORD *rec = NULL;
-    LPCWSTR val = NULL;
     LPWSTR ret = NULL;
 
     /* find if there is a value associated with the checkbox */
@@ -502,24 +644,20 @@ static LPWSTR msi_get_checkbox_value( msi_dialog *dialog, LPCWSTR prop )
     if (!rec)
         return ret;
 
-    val = MSI_RecordGetString( rec, 2 );
-    if (val)
+    ret = msi_get_deformatted_field( dialog->package, rec, 2 );
+    if( ret && !ret[0] )
     {
-        deformat_string( dialog->package, val, &ret );
-        if( ret && !ret[0] )
-        {
-            HeapFree( GetProcessHeap(), 0, ret );
-            ret = NULL;
-        }
+        msi_free( ret );
+        ret = NULL;
     }
     msiobj_release( &rec->hdr );
     if (ret)
         return ret;
 
-    ret = load_dynamic_property(dialog->package, prop, NULL);
+    ret = msi_dup_property( dialog->package, prop );
     if( ret && !ret[0] )
     {
-        HeapFree( GetProcessHeap(), 0, ret );
+        msi_free( ret );
         ret = NULL;
     }
 
@@ -582,7 +720,7 @@ msi_richedit_stream_in( DWORD_PTR arg, LPBYTE buffer, LONG count, LONG *pcb )
 
 static UINT msi_dialog_scrolltext_control( msi_dialog *dialog, MSIRECORD *rec )
 {
-    const static WCHAR szRichEdit20W[] = {
+    static const WCHAR szRichEdit20W[] = {
        'R','i','c','h','E','d','i','t','2','0','W',0
     };
     struct msi_streamin_info info;
@@ -590,10 +728,17 @@ static UINT msi_dialog_scrolltext_control( msi_dialog *dialog, MSIRECORD *rec )
     LPCWSTR text;
     EDITSTREAM es;
     DWORD style;
+    HMODULE hRichedit;
+
+    hRichedit = LoadLibraryA("riched20");
 
     style = WS_BORDER | ES_MULTILINE | WS_VSCROLL |
             ES_READONLY | ES_AUTOVSCROLL | WS_TABSTOP;
     control = msi_dialog_add_control( dialog, rec, szRichEdit20W, style );
+    if (!control)
+        return ERROR_FUNCTION_FAILED;
+
+    control->hDll = hRichedit;
 
     text = MSI_RecordGetString( rec, 10 );
     info.string = strdupWtoA( text );
@@ -606,168 +751,132 @@ static UINT msi_dialog_scrolltext_control( msi_dialog *dialog, MSIRECORD *rec )
 
     SendMessageW( control->hwnd, EM_STREAMIN, SF_RTF, (LPARAM) &es );
 
-    HeapFree( GetProcessHeap(), 0, info.string );
+    msi_free( info.string );
 
     return ERROR_SUCCESS;
 }
 
-static MSIRECORD *msi_get_binary_record( MSIDATABASE *db, LPCWSTR name )
-{
-    const static WCHAR query[] = {
-        's','e','l','e','c','t',' ','*',' ',
-        'f','r','o','m',' ','B','i','n','a','r','y',' ',
-        'w','h','e','r','e',' ',
-            '`','N','a','m','e','`',' ','=',' ','\'','%','s','\'',0
-    };
-
-    return MSI_QueryGetRecord( db, query, name );
-}
-
-static UINT msi_load_bitmap( MSIDATABASE *db, LPCWSTR name, IPicture **pic )
+static HBITMAP msi_load_picture( MSIDATABASE *db, LPCWSTR name,
+                                 INT cx, INT cy, DWORD flags )
 {
+    HBITMAP hOleBitmap = 0, hBitmap = 0, hOldSrcBitmap, hOldDestBitmap;
     MSIRECORD *rec = NULL;
     IStream *stm = NULL;
+    IPicture *pic = NULL;
+    HDC srcdc, destdc;
+    BITMAP bm;
     UINT r;
 
     rec = msi_get_binary_record( db, name );
     if( !rec )
-        return ERROR_FUNCTION_FAILED;
+        goto end;
 
     r = MSI_RecordGetIStream( rec, 2, &stm );
     msiobj_release( &rec->hdr );
     if( r != ERROR_SUCCESS )
-        return r;
+        goto end;
 
-    r = OleLoadPicture( stm, 0, TRUE, &IID_IPicture, (LPVOID*) pic );
+    r = OleLoadPicture( stm, 0, TRUE, &IID_IPicture, (LPVOID*) &pic );
     IStream_Release( stm );
     if( FAILED( r ) )
-        return ERROR_FUNCTION_FAILED;
-
-    return ERROR_SUCCESS;
-}
-
-static UINT msi_dialog_bitmap_control( msi_dialog *dialog, MSIRECORD *rec )
-{
-    IPicture *pic = NULL;
-    msi_control *control;
-    OLE_HANDLE hBitmap = 0;
-    LPCWSTR text;
-    UINT r;
-
-    control = msi_dialog_add_control( dialog, rec, szStatic,
-                            SS_BITMAP | SS_LEFT | SS_CENTERIMAGE );
-    text = MSI_RecordGetString( rec, 10 );
-    r = msi_load_bitmap( dialog->package->db, text, &pic );
-    if( r == ERROR_SUCCESS )
     {
-        r = IPicture_get_Handle( pic, &hBitmap );
-        if( SUCCEEDED( r ) )
-            SendMessageW( control->hwnd, STM_SETIMAGE, IMAGE_BITMAP, hBitmap );
-        control->pic = pic;
+        ERR("failed to load picture\n");
+        goto end;
     }
-    
-    return ERROR_SUCCESS;
-}
 
-static LPWSTR msi_create_tmp_path(void)
-{
-    WCHAR tmp[MAX_PATH];
-    LPWSTR path = NULL;
-    static const WCHAR prefix[] = { 'm','s','i',0 };
-    DWORD len, r;
+    r = IPicture_get_Handle( pic, (OLE_HANDLE*) &hOleBitmap );
+    if( FAILED( r ) )
+    {
+        ERR("failed to get bitmap handle\n");
+        goto end;
+    }
+    /* make the bitmap the desired size */
+    r = GetObjectW( hOleBitmap, sizeof bm, &bm );
+    if (r != sizeof bm )
+    {
+        ERR("failed to get bitmap size\n");
+        goto end;
+    }
 
-    r = GetTempPathW( MAX_PATH, tmp );
-    if( !r )
-        return path;
-    len = lstrlenW( tmp ) + 20;
-    path = HeapAlloc( GetProcessHeap(), 0, len * sizeof (WCHAR) );
-    if( path )
+    if (flags & LR_DEFAULTSIZE)
     {
-        r = GetTempFileNameW( tmp, prefix, 0, path );
-        if (!r)
-        {
-            HeapFree( GetProcessHeap(), 0, path );
-            path = NULL;
-        }
+        cx = bm.bmWidth;
+        cy = bm.bmHeight;
     }
-    return path;
+
+    srcdc = CreateCompatibleDC( NULL );
+    hOldSrcBitmap = SelectObject( srcdc, hOleBitmap );
+    destdc = CreateCompatibleDC( NULL );
+    hBitmap = CreateCompatibleBitmap( srcdc, cx, cy );
+    hOldDestBitmap = SelectObject( destdc, hBitmap );
+    StretchBlt( destdc, 0, 0, cx, cy,
+                srcdc, 0, 0, bm.bmWidth, bm.bmHeight, SRCCOPY);
+    SelectObject( srcdc, hOldSrcBitmap );
+    SelectObject( destdc, hOldDestBitmap );
+    DeleteDC( srcdc );
+    DeleteDC( destdc );
+
+end:
+    if ( pic )
+        IPicture_Release( pic );
+    return hBitmap;
 }
 
-static UINT
-msi_load_icon( MSIDATABASE *db, LPCWSTR name, DWORD attributes, HICON *picon )
+static UINT msi_dialog_bitmap_control( msi_dialog *dialog, MSIRECORD *rec )
 {
-    UINT r = ERROR_FUNCTION_FAILED;
-    LPWSTR tmp;
-    MSIRECORD *rec;
-    HICON hicon = 0;
-
-    TRACE("loading %s\n", debugstr_w( name ) );
+    UINT cx, cy, flags, style, attributes;
+    msi_control *control;
+    LPWSTR text;
 
-    tmp = msi_create_tmp_path();
-    if( !tmp )
-        return r;
+    flags = LR_LOADFROMFILE;
+    style = SS_BITMAP | SS_LEFT | WS_GROUP;
 
-    rec = msi_get_binary_record( db, name );
-    if( rec )
+    attributes = MSI_RecordGetInteger( rec, 8 );
+    if( attributes & msidbControlAttributesFixedSize )
     {
-        r = MSI_RecordStreamToFile( rec, 2, tmp );
-        if( r == ERROR_SUCCESS )
-        {
-            DWORD cx = 0, cy = 0, flags = LR_LOADFROMFILE | LR_DEFAULTSIZE;
-            
-            if( attributes & msidbControlAttributesFixedSize )
-            {
-                flags &= ~LR_DEFAULTSIZE;
-                if( attributes & msidbControlAttributesIconSize16 )
-                {
-                    cx += 16;
-                    cy += 16;
-                }
-                if( attributes & msidbControlAttributesIconSize32 )
-                {
-                    cx += 32;
-                    cy += 32;
-                }
-                /* msidbControlAttributesIconSize48 handled by above logic */
-            }
-            
-            hicon = LoadImageW( 0, tmp, IMAGE_ICON, cx, cy, flags );
-            if( hicon )
-                *picon = hicon;
-            else
-                ERR("failed to load icon from %s\n", debugstr_w( tmp ));
-            DeleteFileW( tmp );
-        }
-        msiobj_release( &rec->hdr );
+        flags |= LR_DEFAULTSIZE;
+        style |= SS_CENTERIMAGE;
     }
 
-    HeapFree( GetProcessHeap(), 0, tmp );
+    control = msi_dialog_add_control( dialog, rec, szStatic, style );
+    cx = MSI_RecordGetInteger( rec, 6 );
+    cy = MSI_RecordGetInteger( rec, 7 );
+    cx = msi_dialog_scale_unit( dialog, cx );
+    cy = msi_dialog_scale_unit( dialog, cy );
+
+    text = msi_get_deformatted_field( dialog->package, rec, 10 );
+    control->hBitmap = msi_load_picture( dialog->package->db, text, cx, cy, flags );
+    if( control->hBitmap )
+        SendMessageW( control->hwnd, STM_SETIMAGE,
+                      IMAGE_BITMAP, (LPARAM) control->hBitmap );
+    else
+        ERR("Failed to load bitmap %s\n", debugstr_w(text));
 
-    return r;
+    msi_free( text );
+    
+    return ERROR_SUCCESS;
 }
 
 static UINT msi_dialog_icon_control( msi_dialog *dialog, MSIRECORD *rec )
 {
     msi_control *control;
     DWORD attributes;
-    HICON hIcon = 0;
-    LPCWSTR text;
-    UINT r;
+    LPWSTR text;
 
     TRACE("\n");
 
     control = msi_dialog_add_control( dialog, rec, szStatic,
                             SS_ICON | SS_CENTERIMAGE | WS_GROUP );
-    text = MSI_RecordGetString( rec, 10 );
+            
     attributes = MSI_RecordGetInteger( rec, 8 );
-    r = msi_load_icon( dialog->package->db, text, attributes, &hIcon );
-    if( r == ERROR_SUCCESS )
-    {
-        r = SendMessageW( control->hwnd, STM_SETICON, (WPARAM) hIcon, 0 );
-        control->hIcon = hIcon;
-    }
+    text = msi_get_deformatted_field( dialog->package, rec, 10 );
+    control->hIcon = msi_load_icon( dialog->package->db, text, attributes );
+    if( control->hIcon )
+        SendMessageW( control->hwnd, STM_SETICON, (WPARAM) control->hIcon, 0 );
     else
         ERR("Failed to load bitmap %s\n", debugstr_w(text));
+    msi_free( text );
     return ERROR_SUCCESS;
 }
 
@@ -792,9 +901,9 @@ static UINT msi_dialog_edit_control( msi_dialog *dialog, MSIRECORD *rec )
     prop = MSI_RecordGetString( rec, 9 );
     if( prop )
         control->property = strdupW( prop );
-    val = load_dynamic_property( dialog->package, control->property, NULL );
+    val = msi_dup_property( dialog->package, control->property );
     SetWindowTextW( control->hwnd, val );
-    HeapFree( GetProcessHeap(), 0, val );
+    msi_free( val );
     return ERROR_SUCCESS;
 }
 
@@ -821,12 +930,27 @@ struct msi_maskedit_info
     struct msi_mask_group group[MASK_MAX_GROUPS];
 };
 
+static BOOL msi_mask_editable( WCHAR type )
+{
+    switch (type)
+    {
+    case '%':
+    case '#':
+    case '&':
+    case '`':
+    case '?':
+    case '^':
+        return TRUE;
+    }
+    return FALSE;
+}
+
 static void msi_mask_control_change( struct msi_maskedit_info *info )
 {
     LPWSTR val;
     UINT i, n, r;
 
-    val = HeapAlloc( GetProcessHeap(), 0, (info->num_chars+1)*sizeof(WCHAR) );
+    val = msi_alloc( (info->num_chars+1)*sizeof(WCHAR) );
     for( i=0, n=0; i<info->num_groups; i++ )
     {
         if( (info->group[i].len + n) > info->num_chars )
@@ -834,9 +958,18 @@ static void msi_mask_control_change( struct msi_maskedit_info *info )
             ERR("can't fit control %d text into template\n",i);
             break;
         }
-        r = GetWindowTextW( info->group[i].hwnd, &val[n], info->group[i].len+1 );
-        if( r != info->group[i].len )
-            break;
+        if (!msi_mask_editable(info->group[i].type))
+        {
+            for(r=0; r<info->group[i].len; r++)
+                val[n+r] = info->group[i].type;
+            val[n+r] = 0;
+        }
+        else
+        {
+            r = GetWindowTextW( info->group[i].hwnd, &val[n], info->group[i].len+1 );
+            if( r != info->group[i].len )
+                break;
+        }
         n += r;
     }
 
@@ -850,7 +983,29 @@ static void msi_mask_control_change( struct msi_maskedit_info *info )
         MSI_SetPropertyW( info->dialog->package, info->prop, val );
         msi_dialog_evaluate_control_conditions( info->dialog );
     }
-    HeapFree( GetProcessHeap(), 0, val );
+    msi_free( val );
+}
+
+/* now move to the next control if necessary */
+static VOID msi_mask_next_control( struct msi_maskedit_info *info, HWND hWnd )
+{
+    HWND hWndNext;
+    UINT len, i;
+
+    for( i=0; i<info->num_groups; i++ )
+        if( info->group[i].hwnd == hWnd )
+            break;
+
+    /* don't move from the last control */
+    if( i >= (info->num_groups-1) )
+        return;
+
+    len = SendMessageW( hWnd, WM_GETTEXTLENGTH, 0, 0 );
+    if( len < info->group[i].len )
+        return;
+
+    hWndNext = GetNextDlgTabItem( GetParent( hWnd ), hWnd, FALSE );
+    SetFocus( hWndNext );
 }
 
 static LRESULT WINAPI
@@ -869,11 +1024,14 @@ MSIMaskedEdit_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
     {
     case WM_COMMAND:
         if (HIWORD(wParam) == EN_CHANGE)
+        {
             msi_mask_control_change( info );
+            msi_mask_next_control( info, (HWND) lParam );
+        }
         break;
     case WM_NCDESTROY:
-        HeapFree( GetProcessHeap(), 0, info->prop );
-        HeapFree( GetProcessHeap(), 0, info );
+        msi_free( info->prop );
+        msi_free( info );
         RemovePropW( hWnd, szButtonData );
         break;
     }
@@ -896,7 +1054,7 @@ msi_maskedit_set_text( struct msi_maskedit_info *info, LPCWSTR text )
             LPWSTR chunk = strdupW( p );
             chunk[ info->group[i].len ] = 0;
             SetWindowTextW( info->group[i].hwnd, chunk );
-            HeapFree( GetProcessHeap(), 0, chunk );
+            msi_free( chunk );
         }
         else
         {
@@ -922,7 +1080,7 @@ static struct msi_maskedit_info * msi_dialog_parse_groups( LPCWSTR mask )
     if( !p )
         return info;
 
-    info = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof *info );
+    info = msi_alloc_zero( sizeof *info );
     if( !info )
         return info;
 
@@ -948,7 +1106,7 @@ static struct msi_maskedit_info * msi_dialog_parse_groups( LPCWSTR mask )
         p += n;
     }
 
-    TRACE("%d characters in %d groups\n", total, info->num_groups );
+    TRACE("%d characters in %d groups\n", total, i );
     if( i == MASK_MAX_GROUPS )
         ERR("too many groups in PIDTemplate %s\n", debugstr_w(mask));
 
@@ -959,10 +1117,9 @@ static struct msi_maskedit_info * msi_dialog_parse_groups( LPCWSTR mask )
 }
 
 static void
-msi_maskedit_create_children( struct msi_maskedit_info *info )
+msi_maskedit_create_children( struct msi_maskedit_info *info, LPCWSTR font )
 {
     DWORD width, height, style, wx, ww;
-    LPCWSTR text, font = NULL;
     RECT rect;
     HWND hwnd;
     UINT i;
@@ -974,11 +1131,10 @@ msi_maskedit_create_children( struct msi_maskedit_info *info )
     width = rect.right - rect.left;
     height = rect.bottom - rect.top;
 
-    if( text )
-        font = msi_dialog_get_style( &text );
-
     for( i = 0; i < info->num_groups; i++ )
     {
+        if (!msi_mask_editable( info->group[i].type ))
+            continue;
         wx = (info->group[i].ofs * width) / info->num_chars;
         ww = (info->group[i].len * width) / info->num_chars;
 
@@ -993,25 +1149,31 @@ msi_maskedit_create_children( struct msi_maskedit_info *info )
         SendMessageW( hwnd, EM_LIMITTEXT, info->group[i].len, 0 );
 
         msi_dialog_set_font( info->dialog, hwnd,
-                       font ? font : info->dialog->default_font );
+                             font?font:info->dialog->default_font );
         info->group[i].hwnd = hwnd;
     }
 }
 
-/* office 2003 uses "73931<````=````=````=````=`````>@@@@@" */
+/*
+ * office 2003 uses "73931<````=````=````=````=`````>@@@@@"
+ * delphi 7 uses "<????-??????-??????-????>" and "<???-???>"
+ * filemaker pro 7 uses "<^^^^=^^^^=^^^^=^^^^=^^^^=^^^^=^^^^^>"
+ */
 static UINT msi_dialog_maskedit_control( msi_dialog *dialog, MSIRECORD *rec )
 {
-    const static WCHAR pidt[] = {'P','I','D','T','e','m','p','l','a','t','e',0};
-    LPWSTR mask = NULL, title = NULL, val = NULL;
+    LPWSTR font_mask, val = NULL, font;
     struct msi_maskedit_info *info = NULL;
     UINT ret = ERROR_SUCCESS;
     msi_control *control;
-    LPCWSTR prop;
+    LPCWSTR prop, mask;
 
-    mask = load_dynamic_property( dialog->package, pidt, NULL );
+    TRACE("\n");
+
+    font_mask = msi_get_deformatted_field( dialog->package, rec, 10 );
+    font = msi_dialog_get_style( font_mask, &mask );
     if( !mask )
     {
-        ERR("PIDTemplate is empty\n");
+        ERR("mask template is empty\n");
         goto end;
     }
 
@@ -1045,26 +1207,34 @@ static UINT msi_dialog_maskedit_control( msi_dialog *dialog, MSIRECORD *rec )
     if( prop )
         info->prop = strdupW( prop );
 
-    msi_maskedit_create_children( info );
+    msi_maskedit_create_children( info, font );
 
     if( prop )
     {
-        val = load_dynamic_property( dialog->package, prop, NULL );
+        val = msi_dup_property( dialog->package, prop );
         if( val )
         {
             msi_maskedit_set_text( info, val );
-            HeapFree( GetProcessHeap(), 0, val );
+            msi_free( val );
         }
     }
 
 end:
     if( ret != ERROR_SUCCESS )
-        HeapFree( GetProcessHeap(), 0, info );
-    HeapFree( GetProcessHeap(), 0, title );
-    HeapFree( GetProcessHeap(), 0, mask );
+        msi_free( info );
+    msi_free( font_mask );
+    msi_free( font );
     return ret;
 }
 
+/******************** Progress Bar *****************************************/
+
+static UINT msi_dialog_progress_bar( msi_dialog *dialog, MSIRECORD *rec )
+{
+    msi_dialog_add_control( dialog, rec, PROGRESS_CLASSW, WS_VISIBLE );
+    return ERROR_SUCCESS;
+}
+
 /******************** Path Edit ********************************************/
 
 static UINT msi_dialog_pathedit_control( msi_dialog *dialog, MSIRECORD *rec )
@@ -1080,8 +1250,7 @@ static UINT msi_dialog_create_radiobutton( MSIRECORD *rec, LPVOID param )
     msi_dialog *dialog = group->dialog;
     msi_control *control;
     LPCWSTR prop, text, name;
-    DWORD style;
-    DWORD attributes = group->attributes;
+    DWORD style, attributes = group->attributes;
 
     style = WS_CHILD | BS_AUTORADIOBUTTON | BS_MULTILINE | WS_TABSTOP;
     name = MSI_RecordGetString( rec, 3 );
@@ -1093,6 +1262,8 @@ static UINT msi_dialog_create_radiobutton( MSIRECORD *rec, LPVOID param )
 
     control = msi_dialog_create_window( dialog, rec, szButton, name, text,
                                         style, group->parent->hwnd );
+    if (!control)
+        return ERROR_FUNCTION_FAILED;
     control->handler = msi_dialog_radiogroup_handler;
 
     prop = MSI_RecordGetString( rec, 1 );
@@ -1153,6 +1324,63 @@ static UINT msi_dialog_radiogroup_control( msi_dialog *dialog, MSIRECORD *rec )
     return r;
 }
 
+/******************** Selection Tree ***************************************/
+
+static void
+msi_dialog_tv_add_child_features( MSIPACKAGE *package, HWND hwnd,
+                                  LPCWSTR parent, HTREEITEM hParent )
+{
+    MSIFEATURE *feature;
+    TVINSERTSTRUCTW tvis;
+    HTREEITEM hitem;
+
+    LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
+    {
+        if ( lstrcmpW( parent, feature->Feature_Parent ) )
+            continue;
+
+        if ( !feature->Title )
+            continue;
+
+        memset( &tvis, 0, sizeof tvis );
+        tvis.hParent = hParent;
+        tvis.hInsertAfter = TVI_SORT;
+        if (feature->Title)
+        {
+            tvis.u.item.mask = TVIF_TEXT;
+            tvis.u.item.pszText = feature->Title;
+        }
+        tvis.u.item.lParam = (LPARAM) feature;
+        hitem = (HTREEITEM) SendMessageW( hwnd, TVM_INSERTITEMW, 0, (LPARAM) &tvis );
+        if (!hitem)
+            continue;
+
+        msi_dialog_tv_add_child_features( package, hwnd,
+                                          feature->Feature, hitem );
+    }
+}
+
+static UINT msi_dialog_selection_tree( msi_dialog *dialog, MSIRECORD *rec )
+{
+    msi_control *control;
+    LPCWSTR prop;
+    LPWSTR val;
+    MSIPACKAGE *package = dialog->package;
+
+    prop = MSI_RecordGetString( rec, 9 );
+    val = msi_dup_property( package, prop );
+    control = msi_dialog_add_control( dialog, rec, WC_TREEVIEWW,
+                                      TVS_HASBUTTONS | WS_GROUP | WS_VSCROLL );
+    if (!control)
+        return ERROR_FUNCTION_FAILED;
+
+    msi_dialog_tv_add_child_features( package, control->hwnd, NULL, NULL );
+
+    msi_free( val );
+
+    return ERROR_SUCCESS;
+}
+
 struct control_handler msi_dialog_handler[] =
 {
     { szText, msi_dialog_text_control },
@@ -1165,8 +1393,10 @@ struct control_handler msi_dialog_handler[] =
     { szEdit, msi_dialog_edit_control },
     { szMaskedEdit, msi_dialog_maskedit_control },
     { szPathEdit, msi_dialog_pathedit_control },
+    { szProgressBar, msi_dialog_progress_bar },
     { szRadioButtonGroup, msi_dialog_radiogroup_control },
     { szIcon, msi_dialog_icon_control },
+    { szSelectionTree, msi_dialog_selection_tree },
 };
 
 #define NUM_CONTROL_TYPES (sizeof msi_dialog_handler/sizeof msi_dialog_handler[0])
@@ -1233,7 +1463,7 @@ static UINT msi_dialog_set_control_condition( MSIRECORD *rec, LPVOID param )
     condition = MSI_RecordGetString( rec, 4 );
     r = MSI_EvaluateConditionW( dialog->package, condition );
     control = msi_dialog_find_control( dialog, name );
-    if( r && control )
+    if( r == MSICONDITION_TRUE && control )
     {
         TRACE("%s control %s\n", debugstr_w(action), debugstr_w(name));
 
@@ -1351,13 +1581,45 @@ static void msi_dialog_adjust_dialog_size( msi_dialog *dialog, LPSIZE sz )
     sz->cy = rect.bottom - rect.top;
 }
 
+static BOOL msi_control_set_next( msi_control *control, msi_control *next )
+{
+    return SetWindowPos( next->hwnd, control->hwnd, 0, 0, 0, 0,
+                         SWP_NOMOVE | SWP_NOOWNERZORDER | SWP_NOREDRAW |
+                         SWP_NOREPOSITION | SWP_NOSENDCHANGING | SWP_NOSIZE );
+}
+
+static UINT msi_dialog_set_tab_order( msi_dialog *dialog )
+{
+    msi_control *control, *tab_next;
+
+    LIST_FOR_EACH_ENTRY( control, &dialog->controls, msi_control, entry )
+    {
+        tab_next = msi_dialog_find_control( dialog, control->tabnext );
+        if( !tab_next )
+            continue;
+        msi_control_set_next( control, tab_next );
+    }
+
+    return ERROR_SUCCESS;
+}
+
+static void msi_dialog_set_first_control( msi_dialog* dialog, LPCWSTR name )
+{
+    msi_control *control;
+
+    control = msi_dialog_find_control( dialog, name );
+    if( control )
+        dialog->hWndFocus = control->hwnd;
+    else
+        dialog->hWndFocus = NULL;
+}
+
 static LRESULT msi_dialog_oncreate( HWND hwnd, LPCREATESTRUCTW cs )
 {
     static const WCHAR df[] = {
         'D','e','f','a','u','l','t','U','I','F','o','n','t',0 };
     msi_dialog *dialog = (msi_dialog*) cs->lpCreateParams;
     MSIRECORD *rec = NULL;
-    LPCWSTR text;
     LPWSTR title = NULL;
     SIZE size;
 
@@ -1380,21 +1642,23 @@ static LRESULT msi_dialog_oncreate( HWND hwnd, LPCREATESTRUCTW cs )
     msi_dialog_adjust_dialog_size( dialog, &size );
 
     dialog->attributes = MSI_RecordGetInteger( rec, 6 );
-    text = MSI_RecordGetString( rec, 7 );
 
-    dialog->default_font = load_dynamic_property( dialog->package, df, NULL );
+    dialog->default_font = msi_dup_property( dialog->package, df );
 
-    deformat_string( dialog->package, text, &title );
+    title = msi_get_deformatted_field( dialog->package, rec, 7 );
     SetWindowTextW( hwnd, title );
+    msi_free( title );
+
     SetWindowPos( hwnd, 0, 0, 0, size.cx, size.cy,
                   SWP_NOMOVE | SWP_NOACTIVATE | SWP_NOZORDER | SWP_NOREDRAW );
 
-    HeapFree( GetProcessHeap(), 0, title );
-    msiobj_release( &rec->hdr );
 
     msi_dialog_build_font_list( dialog );
     msi_dialog_fill_controls( dialog );
     msi_dialog_evaluate_control_conditions( dialog );
+    msi_dialog_set_tab_order( dialog );
+    msi_dialog_set_first_control( dialog, MSI_RecordGetString( rec, 8 ) );
+    msiobj_release( &rec->hdr );
 
     return 0;
 }
@@ -1410,8 +1674,8 @@ static UINT msi_dialog_send_event( msi_dialog *dialog, LPCWSTR event, LPCWSTR ar
 
     dialog->event_handler( dialog->package, event_fmt, arg_fmt, dialog );
 
-    HeapFree( GetProcessHeap(), 0, event_fmt );
-    HeapFree( GetProcessHeap(), 0, arg_fmt );
+    msi_free( event_fmt );
+    msi_free( arg_fmt );
 
     return ERROR_SUCCESS;
 }
@@ -1423,7 +1687,7 @@ static UINT msi_dialog_set_property( msi_dialog *dialog, LPCWSTR event, LPCWSTR
     UINT len;
 
     len = strlenW(event);
-    prop = HeapAlloc( GetProcessHeap(), 0, len*sizeof(WCHAR));
+    prop = msi_alloc( len*sizeof(WCHAR));
     strcpyW( prop, &event[1] );
     p = strchrW( prop, ']' );
     if( p && p[1] == 0 )
@@ -1432,10 +1696,11 @@ static UINT msi_dialog_set_property( msi_dialog *dialog, LPCWSTR event, LPCWSTR
         if( strcmpW( szNullArg, arg ) )
             deformat_string( dialog->package, arg, &arg_fmt );
         MSI_SetPropertyW( dialog->package, prop, arg_fmt );
+        msi_free( arg_fmt );
     }
     else
         ERR("Badly formatted property string - what happens?\n");
-    HeapFree( GetProcessHeap(), 0, prop );
+    msi_free( prop );
     return ERROR_SUCCESS;
 }
 
@@ -1447,7 +1712,7 @@ static UINT msi_dialog_control_event( MSIRECORD *rec, LPVOID param )
 
     condition = MSI_RecordGetString( rec, 5 );
     r = MSI_EvaluateConditionW( dialog->package, condition );
-    if( r )
+    if( r == MSICONDITION_TRUE )
     {
         event = MSI_RecordGetString( rec, 3 );
         arg = MSI_RecordGetString( rec, 4 );
@@ -1566,19 +1831,19 @@ static UINT msi_dialog_edit_handler( msi_dialog *dialog,
           debugstr_w(control->property));
 
     sz = 0x20;
-    buf = HeapAlloc( GetProcessHeap(), 0, sz*sizeof(WCHAR) );
+    buf = msi_alloc( sz*sizeof(WCHAR) );
     while( buf )
     {
         r = GetWindowTextW( control->hwnd, buf, sz );
         if( r < (sz-1) )
             break;
-            sz *= 2;
-        buf = HeapReAlloc( GetProcessHeap(), 0, buf, sz*sizeof(WCHAR) );
+        sz *= 2;
+        buf = msi_realloc( buf, sz*sizeof(WCHAR) );
     }
 
     MSI_SetPropertyW( dialog->package, control->property, buf );
 
-    HeapFree( GetProcessHeap(), 0, buf );
+    msi_free( buf );
 
     return ERROR_SUCCESS;
 }
@@ -1599,11 +1864,22 @@ static UINT msi_dialog_radiogroup_handler( msi_dialog *dialog,
 
 static LRESULT msi_dialog_oncommand( msi_dialog *dialog, WPARAM param, HWND hwnd )
 {
-    msi_control *control;
+    msi_control *control = NULL;
 
     TRACE("%p %p %08x\n", dialog, hwnd, param);
 
-    control = msi_dialog_find_control_by_hwnd( dialog, hwnd );
+    switch (param)
+    {
+    case 1: /* enter */
+        control = msi_dialog_find_control( dialog, dialog->control_default );
+        break;
+    case 2: /* escape */
+        control = msi_dialog_find_control( dialog, dialog->control_cancel );
+        break;
+    default: 
+        control = msi_dialog_find_control_by_hwnd( dialog, hwnd );
+    }
+
     if( control )
     {
         if( control->handler )
@@ -1613,10 +1889,20 @@ static LRESULT msi_dialog_oncommand( msi_dialog *dialog, WPARAM param, HWND hwnd
         }
     }
     else
-        ERR("button click from nowhere\n");
+        ERR("button click from nowhere %p %d %p\n", dialog, param, hwnd);
     return 0;
 }
 
+static void msi_dialog_setfocus( msi_dialog *dialog )
+{
+    HWND hwnd = dialog->hWndFocus;
+
+    hwnd = GetNextDlgTabItem( dialog->hwnd, hwnd, TRUE);
+    hwnd = GetNextDlgTabItem( dialog->hwnd, hwnd, FALSE);
+    SetFocus( hwnd );
+    dialog->hWndFocus = hwnd;
+}
+
 static LRESULT WINAPI MSIDialog_WndProc( HWND hwnd, UINT msg,
                 WPARAM wParam, LPARAM lParam )
 {
@@ -1632,6 +1918,17 @@ static LRESULT WINAPI MSIDialog_WndProc( HWND hwnd, UINT msg,
     case WM_COMMAND:
         return msi_dialog_oncommand( dialog, wParam, (HWND)lParam );
 
+    case WM_ACTIVATE:
+        if( LOWORD(wParam) == WA_INACTIVE )
+            dialog->hWndFocus = GetFocus();
+        else
+            msi_dialog_setfocus( dialog );
+        return 0;
+
+    case WM_SETFOCUS:
+        msi_dialog_setfocus( dialog );
+        return 0;
+
     /* bounce back to our subclassed static control */
     case WM_CTLCOLORSTATIC:
         return SendMessageW( (HWND) lParam, WM_CTLCOLORSTATIC, wParam, lParam );
@@ -1684,8 +1981,7 @@ msi_dialog *msi_dialog_create( MSIPACKAGE* package, LPCWSTR szDialogName,
     TRACE("%p %s\n", package, debugstr_w(szDialogName));
 
     /* allocate the structure for the dialog to use */
-    dialog = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY,
-                        sizeof *dialog + sizeof(WCHAR)*strlenW(szDialogName) );
+    dialog = msi_alloc_zero( sizeof *dialog + sizeof(WCHAR)*strlenW(szDialogName) );
     if( !dialog )
         return NULL;
     strcpyW( dialog->name, szDialogName );
@@ -1693,15 +1989,19 @@ msi_dialog *msi_dialog_create( MSIPACKAGE* package, LPCWSTR szDialogName,
     dialog->package = package;
     dialog->event_handler = event_handler;
     dialog->finished = 0;
+    list_init( &dialog->controls );
 
     /* verify that the dialog exists */
     rec = msi_get_dialog_record( dialog );
     if( !rec )
     {
-        HeapFree( GetProcessHeap(), 0, dialog );
+        msiobj_release( &package->hdr );
+        msi_free( dialog );
         return NULL;
     }
     dialog->attributes = MSI_RecordGetInteger( rec, 6 );
+    dialog->control_default = strdupW( MSI_RecordGetString( rec, 9 ) );
+    dialog->control_cancel = strdupW( MSI_RecordGetString( rec, 10 ) );
     msiobj_release( &rec->hdr );
 
     return dialog;
@@ -1778,7 +2078,7 @@ UINT msi_dialog_run_message_loop( msi_dialog *dialog )
     }
 
     ShowWindow( hwnd, SW_SHOW );
-    UpdateWindow( hwnd );
+    /* UpdateWindow( hwnd ); - and causes the transparent static controls not to paint */
 
     if( dialog->attributes & msidbDialogAttributesModal )
     {
@@ -1813,19 +2113,26 @@ void msi_dialog_destroy( msi_dialog *dialog )
     if( dialog->hwnd )
         ShowWindow( dialog->hwnd, SW_HIDE );
     
+    if( dialog->hwnd )
+        DestroyWindow( dialog->hwnd );
+
     /* destroy the list of controls */
-    while( dialog->control_list )
+    while( !list_empty( &dialog->controls ) )
     {
-        msi_control *t = dialog->control_list;
-        dialog->control_list = t->next;
+        msi_control *t = LIST_ENTRY( list_head( &dialog->controls ),
+                                     msi_control, entry );
+        list_remove( &t->entry );
         /* leave dialog->hwnd - destroying parent destroys child windows */
-        HeapFree( GetProcessHeap(), 0, t->property );
-        HeapFree( GetProcessHeap(), 0, t->value );
-        if( t->pic )
-            IPicture_Release( t->pic );
+        msi_free( t->property );
+        msi_free( t->value );
+        if( t->hBitmap )
+            DeleteObject( t->hBitmap );
         if( t->hIcon )
             DestroyIcon( t->hIcon );
-        HeapFree( GetProcessHeap(), 0, t );
+        msi_free( t->tabnext );
+        msi_free( t );
+        if (t->hDll)
+            FreeLibrary( t->hDll );
     }
 
     /* destroy the list of fonts */
@@ -1834,16 +2141,15 @@ void msi_dialog_destroy( msi_dialog *dialog )
         msi_font *t = dialog->font_list;
         dialog->font_list = t->next;
         DeleteObject( t->hfont );
-        HeapFree( GetProcessHeap(), 0, t );
+        msi_free( t );
     }
-    HeapFree( GetProcessHeap(), 0, dialog->default_font );
-
-    if( dialog->hwnd )
-        DestroyWindow( dialog->hwnd );
+    msi_free( dialog->default_font );
 
+    msi_free( dialog->control_default );
+    msi_free( dialog->control_cancel );
     msiobj_release( &dialog->package->hdr );
     dialog->package = NULL;
-    HeapFree( GetProcessHeap(), 0, dialog );
+    msi_free( dialog );
 }
 
 BOOL msi_dialog_register_class( void )
@@ -1855,7 +2161,7 @@ BOOL msi_dialog_register_class( void )
     cls.hInstance     = NULL;
     cls.hIcon         = LoadIconW(0, (LPWSTR)IDI_APPLICATION);
     cls.hCursor       = LoadCursorW(0, (LPWSTR)IDC_ARROW);
-    cls.hbrBackground = (HBRUSH)(COLOR_WINDOW);
+    cls.hbrBackground = (HBRUSH)(COLOR_3DFACE + 1);
     cls.lpszMenuName  = NULL;
     cls.lpszClassName = szMsiDialogClass;
 
@@ -1875,15 +2181,14 @@ BOOL msi_dialog_register_class( void )
     if( !hMsiHiddenWindow )
         return FALSE;
 
-    hRichedit = LoadLibraryA("riched20");
-
     return TRUE;
 }
 
 void msi_dialog_unregister_class( void )
 {
     DestroyWindow( hMsiHiddenWindow );
+    hMsiHiddenWindow = NULL;
     UnregisterClassW( szMsiDialogClass, NULL );
+    UnregisterClassW( szMsiHiddenWindow, NULL );
     uiThreadId = 0;
-    FreeLibrary( hRichedit );
 }