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