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