git.oblomov.eu Git - wine/blob - dlls/msi/helpers.c

   1 /*
   2  * Implementation of the Microsoft Installer (msi.dll)
   3  *
   4  * Copyright 2005 Aric Stewart for CodeWeavers
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 /*
  22  * Here are helper functions formally in action.c that are used by a variaty of
  23  * actions and functions.
  24  */
  25
  26 #include <stdarg.h>
  27
  28 #include "windef.h"
  29 #include "winbase.h"
  30 #include "winerror.h"
  31 #include "wine/debug.h"
  32 #include "msipriv.h"
  33 #include "winuser.h"
  34 #include "wine/unicode.h"
  35 #include "action.h"
  36
  37 WINE_DEFAULT_DEBUG_CHANNEL(msi);
  38
  39 static const WCHAR cszTargetDir[] = {'T','A','R','G','E','T','D','I','R',0};
  40 static const WCHAR cszDatabase[]={'D','A','T','A','B','A','S','E',0};
  41
  42 const WCHAR cszSourceDir[] = {'S','o','u','r','c','e','D','i','r',0};
  43 const WCHAR cszRootDrive[] = {'R','O','O','T','D','R','I','V','E',0};
  44 const WCHAR cszbs[]={'\\',0};
  45
  46 LPWSTR build_icon_path(MSIPACKAGE *package, LPCWSTR icon_name )
  47 {
  48     LPWSTR SystemFolder, dest, FilePath;
  49
  50     static const WCHAR szInstaller[] =
  51         {'M','i','c','r','o','s','o','f','t','\\',
  52          'I','n','s','t','a','l','l','e','r','\\',0};
  53     static const WCHAR szFolder[] =
  54         {'A','p','p','D','a','t','a','F','o','l','d','e','r',0};
  55
  56     SystemFolder = msi_dup_property( package, szFolder );
  57
  58     dest = build_directory_name(3, SystemFolder, szInstaller, package->ProductCode);
  59
  60     create_full_pathW(dest);
  61
  62     FilePath = build_directory_name(2, dest, icon_name);
  63
  64     msi_free(SystemFolder);
  65     msi_free(dest);
  66     return FilePath;
  67 }
  68
  69 LPWSTR msi_dup_record_field( MSIRECORD *row, INT index )
  70 {
  71     return strdupW( MSI_RecordGetString(row,index) );
  72 }
  73
  74 LPWSTR msi_dup_property(MSIPACKAGE *package, LPCWSTR prop)
  75 {
  76     DWORD sz = 0;
  77     LPWSTR str;
  78     UINT r;
  79
  80     r = MSI_GetPropertyW(package, prop, NULL, &sz);
  81     if (r != ERROR_SUCCESS && r != ERROR_MORE_DATA)
  82         return NULL;
  83
  84     sz++;
  85     str = msi_alloc(sz*sizeof(WCHAR));
  86     r = MSI_GetPropertyW(package, prop, str, &sz);
  87     if (r != ERROR_SUCCESS)
  88     {
  89         msi_free(str);
  90         str = NULL;
  91     }
  92     return str;
  93 }
  94
  95 int msi_get_property_int( MSIPACKAGE *package, LPCWSTR prop, int def )
  96 {
  97     LPWSTR str = msi_dup_property( package, prop );
  98     int val = str ? atoiW( str ) : def;
  99     msi_free( str );
 100     return val;
 101 }
 102
 103 MSICOMPONENT* get_loaded_component( MSIPACKAGE* package, LPCWSTR Component )
 104 {
 105     MSICOMPONENT *comp;
 106
 107     LIST_FOR_EACH_ENTRY( comp, &package->components, MSICOMPONENT, entry )
 108     {
 109         if (lstrcmpW(Component,comp->Component)==0)
 110             return comp;
 111     }
 112     return NULL;
 113 }
 114
 115 MSIFEATURE* get_loaded_feature(MSIPACKAGE* package, LPCWSTR Feature )
 116 {
 117     MSIFEATURE *feature;
 118
 119     LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
 120     {
 121         if (lstrcmpW( Feature, feature->Feature )==0)
 122             return feature;
 123     }
 124     return NULL;
 125 }
 126
 127 MSIFILE* get_loaded_file( MSIPACKAGE* package, LPCWSTR key )
 128 {
 129     MSIFILE *file;
 130
 131     LIST_FOR_EACH_ENTRY( file, &package->files, MSIFILE, entry )
 132     {
 133         if (lstrcmpW( key, file->File )==0)
 134             return file;
 135     }
 136     return NULL;
 137 }
 138
 139 int track_tempfile( MSIPACKAGE *package, LPCWSTR name, LPCWSTR path )
 140 {
 141     MSITEMPFILE *temp;
 142
 143     LIST_FOR_EACH_ENTRY( temp, &package->tempfiles, MSITEMPFILE, entry )
 144     {
 145         if (lstrcmpW( name, temp->File )==0)
 146         {
 147             TRACE("tempfile %s already exists with path %s\n",
 148                 debugstr_w(temp->File), debugstr_w(temp->Path));
 149             return -1;
 150         }
 151     }
 152
 153     temp = msi_alloc_zero( sizeof (MSITEMPFILE) );
 154     if (!temp)
 155         return -1;
 156
 157     list_add_head( &package->tempfiles, &temp->entry );
 158
 159     temp->File = strdupW( name );
 160     temp->Path = strdupW( path );
 161
 162     TRACE("adding tempfile %s with path %s\n",
 163            debugstr_w(temp->File), debugstr_w(temp->Path));
 164
 165     return 0;
 166 }
 167
 168 MSIFOLDER *get_loaded_folder( MSIPACKAGE *package, LPCWSTR dir )
 169 {
 170     MSIFOLDER *folder;
 171
 172     LIST_FOR_EACH_ENTRY( folder, &package->folders, MSIFOLDER, entry )
 173     {
 174         if (lstrcmpW( dir, folder->Directory )==0)
 175             return folder;
 176     }
 177     return NULL;
 178 }
 179
 180 static LPWSTR get_source_root( MSIPACKAGE *package )
 181 {
 182     LPWSTR path, p;
 183
 184     path = msi_dup_property( package, cszSourceDir );
 185     if (path)
 186         return path;
 187
 188     path = msi_dup_property( package, cszDatabase );
 189     if (path)
 190     {
 191         p = strrchrW(path,'\\');
 192         if (p)
 193             *(p+1) = 0;
 194     }
 195     return path;
 196 }
 197
 198 /*
 199  * clean_spaces_from_path()
 200  *
 201  * removes spaces from the beginning and end of path segments
 202  * removes multiple \\ characters
 203  */
 204 static void clean_spaces_from_path( LPWSTR p )
 205 {
 206     LPWSTR q = p;
 207     int n, len = 0;
 208
 209     while (1)
 210     {
 211         /* copy until the end of the string or a space */
 212         while (*p != ' ' && (*q = *p))
 213         {
 214             p++, len++;
 215             /* reduce many backslashes to one */
 216             if (*p != '\\' || *q != '\\')
 217                 q++;
 218         }
 219
 220         /* quit at the end of the string */
 221         if (!*p)
 222             break;
 223
 224         /* count the number of spaces */
 225         n = 0;
 226         while (p[n] == ' ')
 227             n++;
 228
 229         /* if it's leading or trailing space, skip it */
 230         if ( len == 0 || p[-1] == '\\' || p[n] == '\\' )
 231             p += n;
 232         else  /* copy n spaces */
 233             while (n && (*q++ = *p++)) n--;
 234     }
 235 }
 236
 237 LPWSTR resolve_folder(MSIPACKAGE *package, LPCWSTR name, BOOL source,
 238                       BOOL set_prop, MSIFOLDER **folder)
 239 {
 240     MSIFOLDER *f;
 241     LPWSTR p, path = NULL;
 242
 243     TRACE("Working to resolve %s\n",debugstr_w(name));
 244
 245     if (!name)
 246         return NULL;
 247
 248     /* special resolving for Target and Source root dir */
 249     if (strcmpW(name,cszTargetDir)==0 || strcmpW(name,cszSourceDir)==0)
 250     {
 251         if (!source)
 252         {
 253             LPWSTR check_path;
 254             check_path = msi_dup_property( package, cszTargetDir );
 255             if (!check_path)
 256             {
 257                 check_path = msi_dup_property( package, cszRootDrive );
 258                 if (set_prop)
 259                     MSI_SetPropertyW(package,cszTargetDir,check_path);
 260             }
 261
 262             /* correct misbuilt target dir */
 263             path = build_directory_name(2, check_path, NULL);
 264             clean_spaces_from_path( path );
 265             if (strcmpiW(path,check_path)!=0)
 266                 MSI_SetPropertyW(package,cszTargetDir,path);
 267             msi_free(check_path);
 268         }
 269         else
 270             path = get_source_root( package );
 271         if (folder)
 272             *folder = get_loaded_folder( package, name );
 273         return path;
 274     }
 275
 276     f = get_loaded_folder( package, name );
 277     if (!f)
 278         return NULL;
 279
 280     if (folder)
 281         *folder = f;
 282
 283     if (!source && f->ResolvedTarget)
 284     {
 285         path = strdupW( f->ResolvedTarget );
 286         TRACE("   already resolved to %s\n",debugstr_w(path));
 287         return path;
 288     }
 289     else if (source && f->ResolvedSource)
 290     {
 291         path = strdupW( f->ResolvedSource );
 292         TRACE("   (source)already resolved to %s\n",debugstr_w(path));
 293         return path;
 294     }
 295     else if (!source && f->Property)
 296     {
 297         path = build_directory_name( 2, f->Property, NULL );
 298
 299         TRACE("   internally set to %s\n",debugstr_w(path));
 300         if (set_prop)
 301             MSI_SetPropertyW( package, name, path );
 302         return path;
 303     }
 304
 305     if (f->Parent)
 306     {
 307         LPWSTR parent = f->Parent->Directory;
 308
 309         TRACE(" ! Parent is %s\n", debugstr_w(parent));
 310
 311         p = resolve_folder(package, parent, source, set_prop, NULL);
 312         if (!source)
 313         {
 314             TRACE("   TargetDefault = %s\n", debugstr_w(f->TargetDefault));
 315
 316             path = build_directory_name( 3, p, f->TargetDefault, NULL );
 317             clean_spaces_from_path( path );
 318             f->ResolvedTarget = strdupW( path );
 319             TRACE("target -> %s\n", debugstr_w(path));
 320             if (set_prop)
 321                 MSI_SetPropertyW(package,name,path);
 322         }
 323         else
 324         {
 325             /* source may be in a few different places ... check each of them */
 326             path = NULL;
 327
 328             /* try the long path directory */
 329             if (f->SourceLongPath)
 330             {
 331                 path = build_directory_name( 3, p, f->SourceLongPath, NULL );
 332                 if (INVALID_FILE_ATTRIBUTES == GetFileAttributesW( path ))
 333                 {
 334                     msi_free( path );
 335                     path = NULL;
 336                 }
 337             }
 338
 339             /* try the short path directory */
 340             if (!path && f->SourceShortPath)
 341             {
 342                 path = build_directory_name( 3, p, f->SourceShortPath, NULL );
 343                 if (INVALID_FILE_ATTRIBUTES == GetFileAttributesW( path ))
 344                 {
 345                     msi_free( path );
 346                     path = NULL;
 347                 }
 348             }
 349
 350             /* try the root of the install */
 351             if (!path)
 352                 path = get_source_root( package );
 353
 354             TRACE("source -> %s\n", debugstr_w(path));
 355             f->ResolvedSource = strdupW( path );
 356         }
 357         msi_free(p);
 358     }
 359     return path;
 360 }
 361
 362 /* wrapper to resist a need for a full rewrite right now */
 363 DWORD deformat_string(MSIPACKAGE *package, LPCWSTR ptr, WCHAR** data )
 364 {
 365     if (ptr)
 366     {
 367         MSIRECORD *rec = MSI_CreateRecord(1);
 368         DWORD size = 0;
 369
 370         MSI_RecordSetStringW(rec,0,ptr);
 371         MSI_FormatRecordW(package,rec,NULL,&size);
 372         if (size >= 0)
 373         {
 374             size++;
 375             *data = msi_alloc(size*sizeof(WCHAR));
 376             if (size > 1)
 377                 MSI_FormatRecordW(package,rec,*data,&size);
 378             else
 379                 *data[0] = 0;
 380             msiobj_release( &rec->hdr );
 381             return sizeof(WCHAR)*size;
 382         }
 383         msiobj_release( &rec->hdr );
 384     }
 385
 386     *data = NULL;
 387     return 0;
 388 }
 389
 390 UINT schedule_action(MSIPACKAGE *package, UINT script, LPCWSTR action)
 391 {
 392     UINT count;
 393     LPWSTR *newbuf = NULL;
 394     if (script >= TOTAL_SCRIPTS)
 395     {
 396         FIXME("Unknown script requested %i\n",script);
 397         return ERROR_FUNCTION_FAILED;
 398     }
 399     TRACE("Scheduling Action %s in script %i\n",debugstr_w(action), script);
 400
 401     count = package->script->ActionCount[script];
 402     package->script->ActionCount[script]++;
 403     if (count != 0)
 404         newbuf = msi_realloc( package->script->Actions[script],
 405                         package->script->ActionCount[script]* sizeof(LPWSTR));
 406     else
 407         newbuf = msi_alloc( sizeof(LPWSTR));
 408
 409     newbuf[count] = strdupW(action);
 410     package->script->Actions[script] = newbuf;
 411
 412    return ERROR_SUCCESS;
 413 }
 414
 415 static void remove_tracked_tempfiles(MSIPACKAGE* package)
 416 {
 417     struct list *item, *cursor;
 418
 419     LIST_FOR_EACH_SAFE( item, cursor, &package->tempfiles )
 420     {
 421         MSITEMPFILE *temp = LIST_ENTRY( item, MSITEMPFILE, entry );
 422
 423         list_remove( &temp->entry );
 424         TRACE("deleting temp file %s\n", debugstr_w( temp->Path ));
 425         DeleteFileW( temp->Path );
 426         msi_free( temp->File );
 427         msi_free( temp->Path );
 428         msi_free( temp );
 429     }
 430 }
 431
 432 static void free_feature( MSIFEATURE *feature )
 433 {
 434     struct list *item, *cursor;
 435
 436     LIST_FOR_EACH_SAFE( item, cursor, &feature->Components )
 437     {
 438         ComponentList *cl = LIST_ENTRY( item, ComponentList, entry );
 439         list_remove( &cl->entry );
 440         msi_free( cl );
 441     }
 442     msi_free( feature->Feature );
 443     msi_free( feature->Feature_Parent );
 444     msi_free( feature->Directory );
 445     msi_free( feature->Description );
 446     msi_free( feature->Title );
 447     msi_free( feature );
 448 }
 449
 450 void free_extension( MSIEXTENSION *ext )
 451 {
 452     struct list *item, *cursor;
 453
 454     LIST_FOR_EACH_SAFE( item, cursor, &ext->verbs )
 455     {
 456         MSIVERB *verb = LIST_ENTRY( item, MSIVERB, entry );
 457
 458         list_remove( &verb->entry );
 459         msi_free( verb->Verb );
 460         msi_free( verb->Command );
 461         msi_free( verb->Argument );
 462         msi_free( verb );
 463     }
 464
 465     msi_free( ext->Extension );
 466     msi_free( ext->ProgIDText );
 467     msi_free( ext );
 468 }
 469
 470 /* Called when the package is being closed */
 471 void ACTION_free_package_structures( MSIPACKAGE* package)
 472 {
 473     INT i;
 474     struct list *item, *cursor;
 475
 476     TRACE("Freeing package action data\n");
 477
 478     remove_tracked_tempfiles(package);
 479
 480     LIST_FOR_EACH_SAFE( item, cursor, &package->features )
 481     {
 482         MSIFEATURE *feature = LIST_ENTRY( item, MSIFEATURE, entry );
 483         list_remove( &feature->entry );
 484         free_feature( feature );
 485     }
 486
 487     LIST_FOR_EACH_SAFE( item, cursor, &package->folders )
 488     {
 489         MSIFOLDER *folder = LIST_ENTRY( item, MSIFOLDER, entry );
 490
 491         list_remove( &folder->entry );
 492         msi_free( folder->Directory );
 493         msi_free( folder->TargetDefault );
 494         msi_free( folder->SourceLongPath );
 495         msi_free( folder->SourceShortPath );
 496         msi_free( folder->ResolvedTarget );
 497         msi_free( folder->ResolvedSource );
 498         msi_free( folder->Property );
 499         msi_free( folder );
 500     }
 501
 502     LIST_FOR_EACH_SAFE( item, cursor, &package->components )
 503     {
 504         MSICOMPONENT *comp = LIST_ENTRY( item, MSICOMPONENT, entry );
 505
 506         list_remove( &comp->entry );
 507         msi_free( comp->Component );
 508         msi_free( comp->ComponentId );
 509         msi_free( comp->Directory );
 510         msi_free( comp->Condition );
 511         msi_free( comp->KeyPath );
 512         msi_free( comp->FullKeypath );
 513         msi_free( comp );
 514     }
 515
 516     LIST_FOR_EACH_SAFE( item, cursor, &package->files )
 517     {
 518         MSIFILE *file = LIST_ENTRY( item, MSIFILE, entry );
 519
 520         list_remove( &file->entry );
 521         msi_free( file->File );
 522         msi_free( file->FileName );
 523         msi_free( file->ShortName );
 524         msi_free( file->LongName );
 525         msi_free( file->Version );
 526         msi_free( file->Language );
 527         msi_free( file->SourcePath );
 528         msi_free( file->TargetPath );
 529         msi_free( file );
 530     }
 531
 532     /* clean up extension, progid, class and verb structures */
 533     LIST_FOR_EACH_SAFE( item, cursor, &package->classes )
 534     {
 535         MSICLASS *cls = LIST_ENTRY( item, MSICLASS, entry );
 536
 537         list_remove( &cls->entry );
 538         msi_free( cls->clsid );
 539         msi_free( cls->Context );
 540         msi_free( cls->Description );
 541         msi_free( cls->FileTypeMask );
 542         msi_free( cls->IconPath );
 543         msi_free( cls->DefInprocHandler );
 544         msi_free( cls->DefInprocHandler32 );
 545         msi_free( cls->Argument );
 546         msi_free( cls->ProgIDText );
 547         msi_free( cls );
 548     }
 549
 550     LIST_FOR_EACH_SAFE( item, cursor, &package->extensions )
 551     {
 552         MSIEXTENSION *ext = LIST_ENTRY( item, MSIEXTENSION, entry );
 553
 554         list_remove( &ext->entry );
 555         free_extension( ext );
 556     }
 557
 558     LIST_FOR_EACH_SAFE( item, cursor, &package->progids )
 559     {
 560         MSIPROGID *progid = LIST_ENTRY( item, MSIPROGID, entry );
 561
 562         list_remove( &progid->entry );
 563         msi_free( progid->ProgID );
 564         msi_free( progid->Description );
 565         msi_free( progid->IconPath );
 566         msi_free( progid );
 567     }
 568
 569     LIST_FOR_EACH_SAFE( item, cursor, &package->mimes )
 570     {
 571         MSIMIME *mt = LIST_ENTRY( item, MSIMIME, entry );
 572
 573         list_remove( &mt->entry );
 574         msi_free( mt->clsid );
 575         msi_free( mt->ContentType );
 576         msi_free( mt );
 577     }
 578
 579     LIST_FOR_EACH_SAFE( item, cursor, &package->appids )
 580     {
 581         MSIAPPID *appid = LIST_ENTRY( item, MSIAPPID, entry );
 582
 583         list_remove( &appid->entry );
 584         msi_free( appid->AppID );
 585         msi_free( appid->RemoteServerName );
 586         msi_free( appid->LocalServer );
 587         msi_free( appid->ServiceParameters );
 588         msi_free( appid->DllSurrogate );
 589         msi_free( appid );
 590     }
 591
 592     if (package->script)
 593     {
 594         for (i = 0; i < TOTAL_SCRIPTS; i++)
 595         {
 596             int j;
 597             for (j = 0; j < package->script->ActionCount[i]; j++)
 598                 msi_free(package->script->Actions[i][j]);
 599
 600             msi_free(package->script->Actions[i]);
 601         }
 602
 603         for (i = 0; i < package->script->UniqueActionsCount; i++)
 604             msi_free(package->script->UniqueActions[i]);
 605
 606         msi_free(package->script->UniqueActions);
 607         msi_free(package->script);
 608     }
 609
 610     msi_free(package->PackagePath);
 611     msi_free(package->ProductCode);
 612     msi_free(package->ActionFormat);
 613     msi_free(package->LastAction);
 614
 615     /* cleanup control event subscriptions */
 616     ControlEvent_CleanupSubscriptions(package);
 617 }
 618
 619 /*
 620  *  build_directory_name()
 621  *
 622  *  This function is to save messing round with directory names
 623  *  It handles adding backslashes between path segments,
 624  *   and can add \ at the end of the directory name if told to.
 625  *
 626  *  It takes a variable number of arguments.
 627  *  It always allocates a new string for the result, so make sure
 628  *   to free the return value when finished with it.
 629  *
 630  *  The first arg is the number of path segments that follow.
 631  *  The arguments following count are a list of path segments.
 632  *  A path segment may be NULL.
 633  *
 634  *  Path segments will be added with a \ separating them.
 635  *  A \ will not be added after the last segment, however if the
 636  *    last segment is NULL, then the last character will be a \
 637  *
 638  */
 639 LPWSTR build_directory_name(DWORD count, ...)
 640 {
 641     DWORD sz = 1, i;
 642     LPWSTR dir;
 643     va_list va;
 644
 645     va_start(va,count);
 646     for(i=0; i<count; i++)
 647     {
 648         LPCWSTR str = va_arg(va,LPCWSTR);
 649         if (str)
 650             sz += strlenW(str) + 1;
 651     }
 652     va_end(va);
 653
 654     dir = msi_alloc(sz*sizeof(WCHAR));
 655     dir[0]=0;
 656
 657     va_start(va,count);
 658     for(i=0; i<count; i++)
 659     {
 660         LPCWSTR str = va_arg(va,LPCWSTR);
 661         if (!str)
 662             continue;
 663         strcatW(dir, str);
 664         if( ((i+1)!=count) && dir[strlenW(dir)-1]!='\\')
 665             strcatW(dir, cszbs);
 666     }
 667     return dir;
 668 }
 669
 670 /***********************************************************************
 671  *            create_full_pathW
 672  *
 673  * Recursively create all directories in the path.
 674  *
 675  * shamelessly stolen from setupapi/queue.c
 676  */
 677 BOOL create_full_pathW(const WCHAR *path)
 678 {
 679     BOOL ret = TRUE;
 680     int len;
 681     WCHAR *new_path;
 682
 683     new_path = msi_alloc( (strlenW(path) + 1) * sizeof(WCHAR));
 684
 685     strcpyW(new_path, path);
 686
 687     while((len = strlenW(new_path)) && new_path[len - 1] == '\\')
 688     new_path[len - 1] = 0;
 689
 690     while(!CreateDirectoryW(new_path, NULL))
 691     {
 692         WCHAR *slash;
 693         DWORD last_error = GetLastError();
 694         if(last_error == ERROR_ALREADY_EXISTS)
 695             break;
 696
 697         if(last_error != ERROR_PATH_NOT_FOUND)
 698         {
 699             ret = FALSE;
 700             break;
 701         }
 702
 703         if(!(slash = strrchrW(new_path, '\\')))
 704         {
 705             ret = FALSE;
 706             break;
 707         }
 708
 709         len = slash - new_path;
 710         new_path[len] = 0;
 711         if(!create_full_pathW(new_path))
 712         {
 713             ret = FALSE;
 714             break;
 715         }
 716         new_path[len] = '\\';
 717     }
 718
 719     msi_free(new_path);
 720     return ret;
 721 }
 722
 723 void ui_progress(MSIPACKAGE *package, int a, int b, int c, int d )
 724 {
 725     MSIRECORD * row;
 726
 727     row = MSI_CreateRecord(4);
 728     MSI_RecordSetInteger(row,1,a);
 729     MSI_RecordSetInteger(row,2,b);
 730     MSI_RecordSetInteger(row,3,c);
 731     MSI_RecordSetInteger(row,4,d);
 732     MSI_ProcessMessage(package, INSTALLMESSAGE_PROGRESS, row);
 733     msiobj_release(&row->hdr);
 734
 735     msi_dialog_check_messages(NULL);
 736 }
 737
 738 void ui_actiondata(MSIPACKAGE *package, LPCWSTR action, MSIRECORD * record)
 739 {
 740     static const WCHAR Query_t[] =
 741         {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
 742          '`','A','c','t','i','o', 'n','T','e','x','t','`',' ',
 743          'W','H','E','R','E',' ', '`','A','c','t','i','o','n','`',' ','=',
 744          ' ','\'','%','s','\'',0};
 745     WCHAR message[1024];
 746     MSIRECORD * row = 0;
 747     DWORD size;
 748
 749     if (!package->LastAction || strcmpW(package->LastAction,action))
 750     {
 751         row = MSI_QueryGetRecord(package->db, Query_t, action);
 752         if (!row)
 753             return;
 754
 755         if (MSI_RecordIsNull(row,3))
 756         {
 757             msiobj_release(&row->hdr);
 758             return;
 759         }
 760
 761         /* update the cached actionformat */
 762         msi_free(package->ActionFormat);
 763         package->ActionFormat = msi_dup_record_field(row,3);
 764
 765         msi_free(package->LastAction);
 766         package->LastAction = strdupW(action);
 767
 768         msiobj_release(&row->hdr);
 769     }
 770
 771     MSI_RecordSetStringW(record,0,package->ActionFormat);
 772     size = 1024;
 773     MSI_FormatRecordW(package,record,message,&size);
 774
 775     row = MSI_CreateRecord(1);
 776     MSI_RecordSetStringW(row,1,message);
 777
 778     MSI_ProcessMessage(package, INSTALLMESSAGE_ACTIONDATA, row);
 779
 780     msiobj_release(&row->hdr);
 781 }
 782
 783 BOOL ACTION_VerifyComponentForAction( MSICOMPONENT* comp, INSTALLSTATE check )
 784 {
 785     if (!comp)
 786         return FALSE;
 787
 788     if (comp->Installed == check)
 789         return FALSE;
 790
 791     if (comp->ActionRequest == check)
 792         return TRUE;
 793     else
 794         return FALSE;
 795 }
 796
 797 BOOL ACTION_VerifyFeatureForAction( MSIFEATURE* feature, INSTALLSTATE check )
 798 {
 799     if (!feature)
 800         return FALSE;
 801
 802     if (feature->Installed == check)
 803         return FALSE;
 804
 805     if (feature->ActionRequest == check)
 806         return TRUE;
 807     else
 808         return FALSE;
 809 }
 810
 811 void reduce_to_longfilename(WCHAR* filename)
 812 {
 813     LPWSTR p = strchrW(filename,'|');
 814     if (p)
 815         memmove(filename, p+1, (strlenW(p+1)+1)*sizeof(WCHAR));
 816 }
 817
 818 void reduce_to_shortfilename(WCHAR* filename)
 819 {
 820     LPWSTR p = strchrW(filename,'|');
 821     if (p)
 822         *p = 0;
 823 }
 824
 825 LPWSTR create_component_advertise_string(MSIPACKAGE* package,
 826                 MSICOMPONENT* component, LPCWSTR feature)
 827 {
 828     static const WCHAR fmt[] = {'%','s','%','s','%','c','%','s',0};
 829     WCHAR productid_85[21], component_85[21];
 830     LPWSTR output = NULL;
 831     DWORD sz = 0;
 832     GUID clsid;
 833
 834     /* > is used if there is a component GUID and < if not.  */
 835
 836     productid_85[0] = 0;
 837     component_85[0] = 0;
 838
 839     CLSIDFromString(package->ProductCode, &clsid);
 840     encode_base85_guid(&clsid, productid_85);
 841
 842     if (component)
 843     {
 844         CLSIDFromString(component->ComponentId, &clsid);
 845         encode_base85_guid(&clsid, component_85);
 846     }
 847
 848     TRACE("prod=%s feat=%s comp=%s\n", debugstr_w(productid_85),
 849           debugstr_w(feature), debugstr_w(component_85));
 850
 851     sz = 20 + lstrlenW(feature) + 20 + 3;
 852
 853     output = msi_alloc_zero(sz*sizeof(WCHAR));
 854
 855     sprintfW(output, fmt, productid_85, feature,
 856              component?'>':'<', component_85);
 857
 858     return output;
 859 }
 860
 861 /* update compoennt state based on a feature change */
 862 void ACTION_UpdateComponentStates(MSIPACKAGE *package, LPCWSTR szFeature)
 863 {
 864     INSTALLSTATE newstate;
 865     MSIFEATURE *feature;
 866     ComponentList *cl;
 867
 868     feature = get_loaded_feature(package,szFeature);
 869     if (!feature)
 870         return;
 871
 872     newstate = feature->ActionRequest;
 873
 874     LIST_FOR_EACH_ENTRY( cl, &feature->Components, ComponentList, entry )
 875     {
 876         MSICOMPONENT* component = cl->component;
 877
 878         TRACE("MODIFYING(%i): Component %s (Installed %i, Action %i, Request %i)\n",
 879             newstate, debugstr_w(component->Component), component->Installed,
 880             component->Action, component->ActionRequest);
 881
 882         if (!component->Enabled)
 883             continue;
 884
 885         if (newstate == INSTALLSTATE_LOCAL)
 886         {
 887             component->ActionRequest = INSTALLSTATE_LOCAL;
 888             component->Action = INSTALLSTATE_LOCAL;
 889         }
 890         else
 891         {
 892             ComponentList *clist;
 893             MSIFEATURE *f;
 894
 895             component->ActionRequest = newstate;
 896             component->Action = newstate;
 897
 898             /*if any other feature wants is local we need to set it local*/
 899             LIST_FOR_EACH_ENTRY( f, &package->features, MSIFEATURE, entry )
 900             {
 901                 if ( component->ActionRequest != INSTALLSTATE_LOCAL )
 902                     break;
 903
 904                 LIST_FOR_EACH_ENTRY( clist, &f->Components, ComponentList, entry )
 905                 {
 906                     if ( clist->component == component )
 907                     {
 908                         if (f->ActionRequest == INSTALLSTATE_LOCAL)
 909                         {
 910                             TRACE("Saved by %s\n", debugstr_w(f->Feature));
 911                             component->ActionRequest = INSTALLSTATE_LOCAL;
 912                             component->Action = INSTALLSTATE_LOCAL;
 913                         }
 914                         break;
 915                     }
 916                 }
 917             }
 918         }
 919         TRACE("Result (%i): Component %s (Installed %i, Action %i, Request %i)\n",
 920             newstate, debugstr_w(component->Component), component->Installed,
 921             component->Action, component->ActionRequest);
 922     }
 923 }
 924
 925 UINT register_unique_action(MSIPACKAGE *package, LPCWSTR action)
 926 {
 927     UINT count;
 928     LPWSTR *newbuf = NULL;
 929
 930     if (!package->script)
 931         return FALSE;
 932
 933     TRACE("Registering Action %s as having fun\n",debugstr_w(action));
 934
 935     count = package->script->UniqueActionsCount;
 936     package->script->UniqueActionsCount++;
 937     if (count != 0)
 938         newbuf = msi_realloc( package->script->UniqueActions,
 939                         package->script->UniqueActionsCount* sizeof(LPWSTR));
 940     else
 941         newbuf = msi_alloc( sizeof(LPWSTR));
 942
 943     newbuf[count] = strdupW(action);
 944     package->script->UniqueActions = newbuf;
 945
 946     return ERROR_SUCCESS;
 947 }
 948
 949 BOOL check_unique_action(MSIPACKAGE *package, LPCWSTR action)
 950 {
 951     INT i;
 952
 953     if (!package->script)
 954         return FALSE;
 955
 956     for (i = 0; i < package->script->UniqueActionsCount; i++)
 957         if (!strcmpW(package->script->UniqueActions[i],action))
 958             return TRUE;
 959
 960     return FALSE;
 961 }
 962
 963 WCHAR* generate_error_string(MSIPACKAGE *package, UINT error, DWORD count, ... )
 964 {
 965     static const WCHAR query[] = {'S','E','L','E','C','T',' ','`','M','e','s','s','a','g','e','`',' ','F','R','O','M',' ','`','E','r','r','o','r','`',' ','W','H','E','R','E',' ','`','E','r','r','o','r','`',' ','=',' ','%','i',0};
 966
 967     MSIRECORD *rec;
 968     MSIRECORD *row;
 969     DWORD size = 0;
 970     DWORD i;
 971     va_list va;
 972     LPCWSTR str;
 973     LPWSTR data;
 974
 975     row = MSI_QueryGetRecord(package->db, query, error);
 976     if (!row)
 977         return 0;
 978
 979     rec = MSI_CreateRecord(count+2);
 980
 981     str = MSI_RecordGetString(row,1);
 982     MSI_RecordSetStringW(rec,0,str);
 983     msiobj_release( &row->hdr );
 984     MSI_RecordSetInteger(rec,1,error);
 985
 986     va_start(va,count);
 987     for (i = 0; i < count; i++)
 988     {
 989         str = va_arg(va,LPCWSTR);
 990         MSI_RecordSetStringW(rec,(i+2),str);
 991     }
 992     va_end(va);
 993
 994     MSI_FormatRecordW(package,rec,NULL,&size);
 995     if (size >= 0)
 996     {
 997         size++;
 998         data = msi_alloc(size*sizeof(WCHAR));
 999         if (size > 1)
1000             MSI_FormatRecordW(package,rec,data,&size);
1001         else
1002             data[0] = 0;
1003         msiobj_release( &rec->hdr );
1004         return data;
1005     }
1006
1007     msiobj_release( &rec->hdr );
1008     data = NULL;
1009     return data;
1010 }