git.oblomov.eu Git - wine/blob - dlls/msi/files.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 /*
  23  * Actions dealing with files These are
  24  *
  25  * InstallFiles
  26  * DuplicateFiles
  27  * MoveFiles (TODO)
  28  * PatchFiles (TODO)
  29  * RemoveDuplicateFiles(TODO)
  30  * RemoveFiles(TODO)
  31  */
  32
  33 #include <stdarg.h>
  34
  35 #include "windef.h"
  36 #include "winbase.h"
  37 #include "winerror.h"
  38 #include "wine/debug.h"
  39 #include "fdi.h"
  40 #include "msi.h"
  41 #include "msidefs.h"
  42 #include "msvcrt/fcntl.h"
  43 #include "msipriv.h"
  44 #include "winuser.h"
  45 #include "winreg.h"
  46 #include "shlwapi.h"
  47 #include "wine/unicode.h"
  48
  49 WINE_DEFAULT_DEBUG_CHANNEL(msi);
  50
  51 extern const WCHAR szInstallFiles[];
  52 extern const WCHAR szDuplicateFiles[];
  53 extern const WCHAR szMoveFiles[];
  54 extern const WCHAR szPatchFiles[];
  55 extern const WCHAR szRemoveDuplicateFiles[];
  56 extern const WCHAR szRemoveFiles[];
  57
  58 static const WCHAR cszTempFolder[]= {'T','e','m','p','F','o','l','d','e','r',0};
  59
  60 struct media_info {
  61     UINT disk_id;
  62     UINT last_sequence;
  63     LPWSTR disk_prompt;
  64     LPWSTR cabinet;
  65     LPWSTR volume_label;
  66     BOOL is_continuous;
  67     BOOL is_extracted;
  68     WCHAR source[MAX_PATH];
  69 };
  70
  71 static UINT msi_change_media( MSIPACKAGE *package, struct media_info *mi )
  72 {
  73     LPSTR msg;
  74     LPWSTR error, error_dialog;
  75     UINT r = ERROR_SUCCESS;
  76
  77     static const WCHAR szUILevel[] = {'U','I','L','e','v','e','l',0};
  78     static const WCHAR error_prop[] = {'E','r','r','o','r','D','i','a','l','o','g',0};
  79
  80     if ( (msi_get_property_int(package, szUILevel, 0) & INSTALLUILEVEL_MASK) == INSTALLUILEVEL_NONE && !gUIHandlerA )
  81         return ERROR_SUCCESS;
  82
  83     error = generate_error_string( package, 1302, 1, mi->disk_prompt );
  84     error_dialog = msi_dup_property( package, error_prop );
  85
  86     while ( r == ERROR_SUCCESS && GetFileAttributesW( mi->source ) == INVALID_FILE_ATTRIBUTES )
  87     {
  88         r = msi_spawn_error_dialog( package, error_dialog, error );
  89
  90         if (gUIHandlerA)
  91         {
  92             msg = strdupWtoA( error );
  93             gUIHandlerA( gUIContext, MB_RETRYCANCEL | INSTALLMESSAGE_ERROR, msg );
  94             msi_free(msg);
  95         }
  96     }
  97
  98     msi_free( error );
  99     msi_free( error_dialog );
 100
 101     return r;
 102 }
 103
 104 /*
 105  * This is a helper function for handling embedded cabinet media
 106  */
 107 static UINT writeout_cabinet_stream(MSIPACKAGE *package, LPCWSTR stream_name,
 108                                     WCHAR* source)
 109 {
 110     UINT rc;
 111     USHORT* data;
 112     UINT    size;
 113     DWORD   write;
 114     HANDLE  the_file;
 115     WCHAR tmp[MAX_PATH];
 116
 117     rc = read_raw_stream_data(package->db,stream_name,&data,&size);
 118     if (rc != ERROR_SUCCESS)
 119         return rc;
 120
 121     write = MAX_PATH;
 122     if (MSI_GetPropertyW(package, cszTempFolder, tmp, &write))
 123         GetTempPathW(MAX_PATH,tmp);
 124
 125     GetTempFileNameW(tmp,stream_name,0,source);
 126
 127     track_tempfile(package, source);
 128     the_file = CreateFileW(source, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
 129                            FILE_ATTRIBUTE_NORMAL, NULL);
 130
 131     if (the_file == INVALID_HANDLE_VALUE)
 132     {
 133         ERR("Unable to create file %s\n",debugstr_w(source));
 134         rc = ERROR_FUNCTION_FAILED;
 135         goto end;
 136     }
 137
 138     WriteFile(the_file,data,size,&write,NULL);
 139     CloseHandle(the_file);
 140     TRACE("wrote %i bytes to %s\n",write,debugstr_w(source));
 141 end:
 142     msi_free(data);
 143     return rc;
 144 }
 145
 146
 147 /* Support functions for FDI functions */
 148 typedef struct
 149 {
 150     MSIPACKAGE* package;
 151     struct media_info *mi;
 152 } CabData;
 153
 154 static void * cabinet_alloc(ULONG cb)
 155 {
 156     return msi_alloc(cb);
 157 }
 158
 159 static void cabinet_free(void *pv)
 160 {
 161     msi_free(pv);
 162 }
 163
 164 static INT_PTR cabinet_open(char *pszFile, int oflag, int pmode)
 165 {
 166     HANDLE handle;
 167     DWORD dwAccess = 0;
 168     DWORD dwShareMode = 0;
 169     DWORD dwCreateDisposition = OPEN_EXISTING;
 170     switch (oflag & _O_ACCMODE)
 171     {
 172     case _O_RDONLY:
 173         dwAccess = GENERIC_READ;
 174         dwShareMode = FILE_SHARE_READ | FILE_SHARE_DELETE;
 175         break;
 176     case _O_WRONLY:
 177         dwAccess = GENERIC_WRITE;
 178         dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE;
 179         break;
 180     case _O_RDWR:
 181         dwAccess = GENERIC_READ | GENERIC_WRITE;
 182         dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE;
 183         break;
 184     }
 185     if ((oflag & (_O_CREAT | _O_EXCL)) == (_O_CREAT | _O_EXCL))
 186         dwCreateDisposition = CREATE_NEW;
 187     else if (oflag & _O_CREAT)
 188         dwCreateDisposition = CREATE_ALWAYS;
 189     handle = CreateFileA( pszFile, dwAccess, dwShareMode, NULL,
 190                           dwCreateDisposition, 0, NULL );
 191     if (handle == INVALID_HANDLE_VALUE)
 192         return 0;
 193     return (INT_PTR) handle;
 194 }
 195
 196 static UINT cabinet_read(INT_PTR hf, void *pv, UINT cb)
 197 {
 198     HANDLE handle = (HANDLE) hf;
 199     DWORD dwRead;
 200     if (ReadFile(handle, pv, cb, &dwRead, NULL))
 201         return dwRead;
 202     return 0;
 203 }
 204
 205 static UINT cabinet_write(INT_PTR hf, void *pv, UINT cb)
 206 {
 207     HANDLE handle = (HANDLE) hf;
 208     DWORD dwWritten;
 209     if (WriteFile(handle, pv, cb, &dwWritten, NULL))
 210         return dwWritten;
 211     return 0;
 212 }
 213
 214 static int cabinet_close(INT_PTR hf)
 215 {
 216     HANDLE handle = (HANDLE) hf;
 217     return CloseHandle(handle) ? 0 : -1;
 218 }
 219
 220 static long cabinet_seek(INT_PTR hf, long dist, int seektype)
 221 {
 222     HANDLE handle = (HANDLE) hf;
 223     /* flags are compatible and so are passed straight through */
 224     return SetFilePointer(handle, dist, NULL, seektype);
 225 }
 226
 227 static void msi_file_update_ui( MSIPACKAGE *package, MSIFILE *f, const WCHAR *action )
 228 {
 229     MSIRECORD *uirow;
 230     LPWSTR uipath, p;
 231
 232     /* the UI chunk */
 233     uirow = MSI_CreateRecord( 9 );
 234     MSI_RecordSetStringW( uirow, 1, f->FileName );
 235     uipath = strdupW( f->TargetPath );
 236     p = strrchrW(uipath,'\\');
 237     if (p)
 238         p[1]=0;
 239     MSI_RecordSetStringW( uirow, 9, uipath);
 240     MSI_RecordSetInteger( uirow, 6, f->FileSize );
 241     ui_actiondata( package, action, uirow);
 242     msiobj_release( &uirow->hdr );
 243     msi_free( uipath );
 244     ui_progress( package, 2, f->FileSize, 0, 0);
 245 }
 246
 247 static UINT msi_media_get_disk_info( MSIPACKAGE *package, struct media_info *mi )
 248 {
 249     MSIRECORD *row;
 250     LPWSTR ptr;
 251
 252     static const WCHAR query[] =
 253         {'S','E','L','E','C','T',' ','*',' ', 'F','R','O','M',' ',
 254          '`','M','e','d','i','a','`',' ','W','H','E','R','E',' ',
 255          '`','D','i','s','k','I','d','`',' ','=',' ','%','i',0};
 256
 257     row = MSI_QueryGetRecord(package->db, query, mi->disk_id);
 258     if (!row)
 259     {
 260         TRACE("Unable to query row\n");
 261         return ERROR_FUNCTION_FAILED;
 262     }
 263
 264     mi->disk_prompt = strdupW(MSI_RecordGetString(row, 3));
 265     mi->cabinet = strdupW(MSI_RecordGetString(row, 4));
 266
 267     ptr = strrchrW(mi->source, '\\') + 1;
 268     lstrcpyW(ptr, mi->cabinet);
 269     msiobj_release(&row->hdr);
 270
 271     return ERROR_SUCCESS;
 272 }
 273
 274 static INT_PTR cabinet_notify(FDINOTIFICATIONTYPE fdint, PFDINOTIFICATION pfdin)
 275 {
 276     TRACE("(%d)\n", fdint);
 277
 278     switch (fdint)
 279     {
 280     case fdintPARTIAL_FILE:
 281     {
 282         CabData *data = (CabData *)pfdin->pv;
 283         data->mi->is_continuous = FALSE;
 284         return 0;
 285     }
 286     case fdintNEXT_CABINET:
 287     {
 288         CabData *data = (CabData *)pfdin->pv;
 289         struct media_info *mi = data->mi;
 290         LPWSTR cab = strdupAtoW(pfdin->psz1);
 291         UINT rc;
 292
 293         msi_free(mi->disk_prompt);
 294
 295         mi->disk_id++;
 296         mi->is_continuous = TRUE;
 297
 298         rc = msi_media_get_disk_info(data->package, mi);
 299         if (rc != ERROR_SUCCESS)
 300         {
 301             ERR("Failed to get next cabinet information: %d\n", rc);
 302             return -1;
 303         }
 304
 305         if (lstrcmpiW(mi->cabinet, cab))
 306         {
 307             msi_free(cab);
 308             ERR("Continuous cabinet does not match the next cabinet in the Media table\n");
 309             return -1;
 310         }
 311
 312         msi_free(cab);
 313
 314         TRACE("Searching for %s\n", debugstr_w(mi->source));
 315
 316         if (GetFileAttributesW(mi->source) == INVALID_FILE_ATTRIBUTES)
 317             rc = msi_change_media(data->package, mi);
 318
 319         if (rc != ERROR_SUCCESS)
 320             return -1;
 321
 322         return 0;
 323     }
 324     case fdintCOPY_FILE:
 325     {
 326         CabData *data = (CabData*) pfdin->pv;
 327         HANDLE handle;
 328         LPWSTR file;
 329         MSIFILE *f;
 330         DWORD attrs;
 331
 332         file = strdupAtoW(pfdin->psz1);
 333         f = get_loaded_file(data->package, file);
 334         msi_free(file);
 335
 336         if (!f)
 337         {
 338             WARN("unknown file in cabinet (%s)\n",debugstr_a(pfdin->psz1));
 339             return 0;
 340         }
 341
 342         if (f->state != msifs_missing && f->state != msifs_overwrite)
 343         {
 344             TRACE("Skipping extraction of %s\n",debugstr_a(pfdin->psz1));
 345             return 0;
 346         }
 347
 348         msi_file_update_ui( data->package, f, szInstallFiles );
 349
 350         TRACE("extracting %s\n", debugstr_w(f->TargetPath) );
 351
 352         attrs = f->Attributes & (FILE_ATTRIBUTE_READONLY|FILE_ATTRIBUTE_HIDDEN|FILE_ATTRIBUTE_SYSTEM);
 353         if (!attrs) attrs = FILE_ATTRIBUTE_NORMAL;
 354
 355         handle = CreateFileW( f->TargetPath, GENERIC_READ | GENERIC_WRITE, 0,
 356                               NULL, CREATE_ALWAYS, attrs, NULL );
 357         if ( handle == INVALID_HANDLE_VALUE )
 358         {
 359             if ( GetFileAttributesW( f->TargetPath ) != INVALID_FILE_ATTRIBUTES )
 360                 f->state = msifs_installed;
 361             else
 362                 ERR("failed to create %s (error %d)\n",
 363                     debugstr_w( f->TargetPath ), GetLastError() );
 364
 365             return 0;
 366         }
 367
 368         f->state = msifs_installed;
 369         return (INT_PTR) handle;
 370     }
 371     case fdintCLOSE_FILE_INFO:
 372     {
 373         FILETIME ft;
 374         FILETIME ftLocal;
 375         HANDLE handle = (HANDLE) pfdin->hf;
 376
 377         if (!DosDateTimeToFileTime(pfdin->date, pfdin->time, &ft))
 378             return -1;
 379         if (!LocalFileTimeToFileTime(&ft, &ftLocal))
 380             return -1;
 381         if (!SetFileTime(handle, &ftLocal, 0, &ftLocal))
 382             return -1;
 383         CloseHandle(handle);
 384         return 1;
 385     }
 386     default:
 387         return 0;
 388     }
 389 }
 390
 391 /***********************************************************************
 392  *            extract_cabinet_file
 393  *
 394  * Extract files from a cab file.
 395  */
 396 static BOOL extract_cabinet_file(MSIPACKAGE* package, struct media_info *mi)
 397 {
 398     LPSTR cabinet, cab_path = NULL;
 399     LPWSTR ptr;
 400     HFDI hfdi;
 401     ERF erf;
 402     BOOL ret = FALSE;
 403     CabData data;
 404
 405     TRACE("Extracting %s\n", debugstr_w(mi->source));
 406
 407     hfdi = FDICreate(cabinet_alloc, cabinet_free, cabinet_open, cabinet_read,
 408                      cabinet_write, cabinet_close, cabinet_seek, 0, &erf);
 409     if (!hfdi)
 410     {
 411         ERR("FDICreate failed\n");
 412         return FALSE;
 413     }
 414
 415     ptr = strrchrW(mi->source, '\\') + 1;
 416     cabinet = strdupWtoA(ptr);
 417     if (!cabinet)
 418         goto done;
 419
 420     cab_path = strdupWtoA(mi->source);
 421     if (!cab_path)
 422         goto done;
 423
 424     cab_path[ptr - mi->source] = '\0';
 425
 426     data.package = package;
 427     data.mi = mi;
 428
 429     ret = FDICopy(hfdi, cabinet, cab_path, 0, cabinet_notify, NULL, &data);
 430     if (!ret)
 431         ERR("FDICopy failed\n");
 432
 433 done:
 434     FDIDestroy(hfdi);
 435     msi_free(cabinet);
 436     msi_free(cab_path);
 437
 438     if (ret)
 439         mi->is_extracted = TRUE;
 440
 441     return ret;
 442 }
 443
 444 static VOID set_file_source(MSIPACKAGE* package, MSIFILE* file, LPCWSTR path)
 445 {
 446     if (!file->IsCompressed)
 447     {
 448         LPWSTR p, path;
 449         p = resolve_folder(package, file->Component->Directory, TRUE, FALSE, NULL);
 450         path = build_directory_name(2, p, file->ShortName);
 451         if (INVALID_FILE_ATTRIBUTES == GetFileAttributesW( path ))
 452         {
 453             msi_free(path);
 454             path = build_directory_name(2, p, file->LongName);
 455         }
 456         file->SourcePath = path;
 457         msi_free(p);
 458     }
 459     else
 460         file->SourcePath = build_directory_name(2, path, file->File);
 461 }
 462
 463 static void free_media_info( struct media_info *mi )
 464 {
 465     msi_free( mi->disk_prompt );
 466     msi_free( mi->cabinet );
 467     msi_free( mi->volume_label );
 468     msi_free( mi );
 469 }
 470
 471 static UINT download_remote_cabinet(MSIPACKAGE *package, struct media_info *mi)
 472 {
 473     WCHAR temppath[MAX_PATH];
 474     LPWSTR src, ptr;
 475     LPCWSTR cab;
 476
 477     src = strdupW(package->BaseURL);
 478     if (!src)
 479         return ERROR_OUTOFMEMORY;
 480
 481     ptr = strrchrW(src, '/');
 482     if (!ptr)
 483     {
 484         msi_free(src);
 485         return ERROR_FUNCTION_FAILED;
 486     }
 487
 488     *(ptr + 1) = '\0';
 489     ptr = strrchrW(mi->source, '\\');
 490     if (!ptr)
 491         ptr = mi->source;
 492
 493     src = msi_realloc(src, (lstrlenW(src) + lstrlenW(ptr)) * sizeof(WCHAR));
 494     if (!src)
 495         return ERROR_OUTOFMEMORY;
 496
 497     lstrcatW(src, ptr + 1);
 498
 499     temppath[0] = '\0';
 500     cab = msi_download_file(src, temppath);
 501     lstrcpyW(mi->source, cab);
 502
 503     msi_free(src);
 504     return ERROR_SUCCESS;
 505 }
 506
 507 static UINT load_media_info(MSIPACKAGE *package, MSIFILE *file, struct media_info *mi)
 508 {
 509     MSIRECORD *row;
 510     LPWSTR source_dir;
 511     UINT r;
 512
 513     static const WCHAR query[] = {
 514         'S','E','L','E','C','T',' ','*',' ', 'F','R','O','M',' ',
 515         '`','M','e','d','i','a','`',' ','W','H','E','R','E',' ',
 516         '`','L','a','s','t','S','e','q','u','e','n','c','e','`',' ','>','=',
 517         ' ','%','i',' ','A','N','D',' ','`','D','i','s','k','I','d','`',' ','>','=',
 518         ' ','%','i',' ','O','R','D','E','R',' ','B','Y',' ',
 519         '`','D','i','s','k','I','d','`',0
 520     };
 521
 522     row = MSI_QueryGetRecord(package->db, query, file->Sequence, mi->disk_id);
 523     if (!row)
 524     {
 525         TRACE("Unable to query row\n");
 526         return ERROR_FUNCTION_FAILED;
 527     }
 528
 529     mi->disk_id = MSI_RecordGetInteger(row, 1);
 530     mi->last_sequence = MSI_RecordGetInteger(row, 2);
 531     msi_free(mi->disk_prompt);
 532     mi->disk_prompt = strdupW(MSI_RecordGetString(row, 3));
 533     msi_free(mi->cabinet);
 534     mi->cabinet = strdupW(MSI_RecordGetString(row, 4));
 535     msi_free(mi->volume_label);
 536     mi->volume_label = strdupW(MSI_RecordGetString(row, 5));
 537     msiobj_release(&row->hdr);
 538
 539     source_dir = msi_dup_property(package, cszSourceDir);
 540
 541     if (mi->cabinet && mi->cabinet[0] == '#')
 542     {
 543         r = writeout_cabinet_stream(package, &mi->cabinet[1], mi->source);
 544         if (r != ERROR_SUCCESS)
 545         {
 546             ERR("Failed to extract cabinet stream\n");
 547             return ERROR_FUNCTION_FAILED;
 548         }
 549     }
 550     else
 551     {
 552         lstrcpyW(mi->source, source_dir);
 553
 554
 555         if (mi->cabinet)
 556             lstrcatW(mi->source, mi->cabinet);
 557     }
 558
 559     MsiSourceListAddMediaDiskW(package->ProductCode, NULL,
 560         MSIINSTALLCONTEXT_USERMANAGED, MSICODE_PRODUCT,
 561         mi->disk_id, mi->volume_label, mi->disk_prompt);
 562
 563     MsiSourceListSetInfoW(package->ProductCode, NULL,
 564         MSIINSTALLCONTEXT_USERMANAGED,
 565         MSICODE_PRODUCT | MSISOURCETYPE_MEDIA,
 566         INSTALLPROPERTY_LASTUSEDSOURCEW, mi->source);
 567
 568     msi_free(source_dir);
 569     return ERROR_SUCCESS;
 570 }
 571
 572 static UINT ready_media(MSIPACKAGE *package, MSIFILE *file, struct media_info *mi)
 573 {
 574     UINT rc = ERROR_SUCCESS;
 575     BOOL found = FALSE;
 576
 577     /* media info for continuous cabinet is already loaded */
 578     if (mi->is_continuous)
 579         return ERROR_SUCCESS;
 580
 581     rc = load_media_info(package, file, mi);
 582     if (rc != ERROR_SUCCESS)
 583     {
 584         ERR("Unable to load media info\n");
 585         return ERROR_FUNCTION_FAILED;
 586     }
 587
 588     if (file->IsCompressed &&
 589         GetFileAttributesW(mi->source) == INVALID_FILE_ATTRIBUTES)
 590     {
 591         if (package->BaseURL && UrlIsW(package->BaseURL, URLIS_URL))
 592         {
 593             rc = download_remote_cabinet(package, mi);
 594             if (rc == ERROR_SUCCESS &&
 595                 GetFileAttributesW(mi->source) != INVALID_FILE_ATTRIBUTES)
 596             {
 597                 found = TRUE;
 598             }
 599         }
 600
 601         if (!found)
 602             rc = msi_change_media(package, mi);
 603     }
 604
 605     return rc;
 606 }
 607
 608 static UINT get_file_target(MSIPACKAGE *package, LPCWSTR file_key,
 609                             MSIFILE** file)
 610 {
 611     LIST_FOR_EACH_ENTRY( *file, &package->files, MSIFILE, entry )
 612     {
 613         if (lstrcmpW( file_key, (*file)->File )==0)
 614         {
 615             if ((*file)->state >= msifs_overwrite)
 616                 return ERROR_SUCCESS;
 617             else
 618                 return ERROR_FILE_NOT_FOUND;
 619         }
 620     }
 621
 622     return ERROR_FUNCTION_FAILED;
 623 }
 624
 625 static void schedule_install_files(MSIPACKAGE *package)
 626 {
 627     MSIFILE *file;
 628
 629     LIST_FOR_EACH_ENTRY(file, &package->files, MSIFILE, entry)
 630     {
 631         if (!ACTION_VerifyComponentForAction(file->Component, INSTALLSTATE_LOCAL))
 632         {
 633             TRACE("File %s is not scheduled for install\n", debugstr_w(file->File));
 634
 635             ui_progress(package,2,file->FileSize,0,0);
 636             file->state = msifs_skipped;
 637         }
 638     }
 639 }
 640
 641 static UINT copy_install_file(MSIFILE *file)
 642 {
 643     BOOL ret;
 644     UINT gle;
 645
 646     TRACE("Copying %s to %s\n", debugstr_w(file->SourcePath),
 647           debugstr_w(file->TargetPath));
 648
 649     ret = CopyFileW(file->SourcePath, file->TargetPath, FALSE);
 650     if (ret)
 651     {
 652         file->state = msifs_installed;
 653         return ERROR_SUCCESS;
 654     }
 655
 656     gle = GetLastError();
 657     if (gle == ERROR_ALREADY_EXISTS && file->state == msifs_overwrite)
 658     {
 659         TRACE("overwriting existing file\n");
 660         gle = ERROR_SUCCESS;
 661     }
 662     else if (gle == ERROR_FILE_NOT_FOUND)
 663     {
 664         /* FIXME: this needs to be tested, I'm pretty sure it fails */
 665         TRACE("Source file not found\n");
 666         gle = ERROR_SUCCESS;
 667     }
 668     else if (!(file->Attributes & msidbFileAttributesVital))
 669     {
 670         TRACE("Ignoring error for nonvital\n");
 671         gle = ERROR_SUCCESS;
 672     }
 673
 674     return gle;
 675 }
 676
 677 /*
 678  * ACTION_InstallFiles()
 679  *
 680  * For efficiency, this is done in two passes:
 681  * 1) Correct all the TargetPaths and determine what files are to be installed.
 682  * 2) Extract Cabinets and copy files.
 683  */
 684 UINT ACTION_InstallFiles(MSIPACKAGE *package)
 685 {
 686     struct media_info *mi;
 687     UINT rc = ERROR_SUCCESS;
 688     LPWSTR ptr;
 689     MSIFILE *file;
 690
 691     /* increment progress bar each time action data is sent */
 692     ui_progress(package,1,1,0,0);
 693
 694     /* handle the keys for the SourceList */
 695     ptr = strrchrW(package->PackagePath,'\\');
 696     if (ptr)
 697     {
 698         ptr++;
 699         MsiSourceListSetInfoW(package->ProductCode, NULL,
 700                 MSIINSTALLCONTEXT_USERMANAGED,
 701                 MSICODE_PRODUCT,
 702                 INSTALLPROPERTY_PACKAGENAMEW, ptr);
 703     }
 704
 705     schedule_install_files(package);
 706
 707     /*
 708      * Despite MSDN specifying that the CreateFolders action
 709      * should be called before InstallFiles, some installers don't
 710      * do that, and they seem to work correctly.  We need to create
 711      * directories here to make sure that the files can be copied.
 712      */
 713     msi_create_component_directories( package );
 714
 715     mi = msi_alloc_zero( sizeof(struct media_info) );
 716
 717     LIST_FOR_EACH_ENTRY( file, &package->files, MSIFILE, entry )
 718     {
 719         if (file->state != msifs_missing && !mi->is_continuous && file->state != msifs_overwrite)
 720             continue;
 721
 722         if (file->Sequence > mi->last_sequence || mi->is_continuous ||
 723             (file->IsCompressed && !mi->is_extracted))
 724         {
 725             rc = ready_media(package, file, mi);
 726             if (rc != ERROR_SUCCESS)
 727             {
 728                 ERR("Failed to ready media\n");
 729                 rc = ERROR_FUNCTION_FAILED;
 730                 break;
 731             }
 732
 733             if (file->IsCompressed && !extract_cabinet_file(package, mi))
 734             {
 735                 ERR("Failed to extract cabinet: %s\n", debugstr_w(mi->cabinet));
 736                 rc = ERROR_FUNCTION_FAILED;
 737                 break;
 738             }
 739         }
 740
 741         set_file_source(package, file, mi->source);
 742
 743         TRACE("file paths %s to %s\n",debugstr_w(file->SourcePath),
 744               debugstr_w(file->TargetPath));
 745
 746         if (!file->IsCompressed)
 747         {
 748             rc = copy_install_file(file);
 749             if (rc != ERROR_SUCCESS)
 750             {
 751                 ERR("Failed to copy %s to %s (%d)\n", debugstr_w(file->SourcePath),
 752                     debugstr_w(file->TargetPath), rc);
 753                 rc = ERROR_INSTALL_FAILURE;
 754                 break;
 755             }
 756         }
 757         else if (file->state != msifs_installed)
 758         {
 759             ERR("compressed file wasn't extracted (%s)\n", debugstr_w(file->TargetPath));
 760             rc = ERROR_INSTALL_FAILURE;
 761             break;
 762         }
 763     }
 764
 765     free_media_info( mi );
 766     return rc;
 767 }
 768
 769 static UINT ITERATE_DuplicateFiles(MSIRECORD *row, LPVOID param)
 770 {
 771     MSIPACKAGE *package = (MSIPACKAGE*)param;
 772     WCHAR dest_name[0x100];
 773     LPWSTR dest_path, dest;
 774     LPCWSTR file_key, component;
 775     DWORD sz;
 776     DWORD rc;
 777     MSICOMPONENT *comp;
 778     MSIFILE *file;
 779
 780     component = MSI_RecordGetString(row,2);
 781     comp = get_loaded_component(package,component);
 782
 783     if (!ACTION_VerifyComponentForAction( comp, INSTALLSTATE_LOCAL ))
 784     {
 785         TRACE("Skipping copy due to disabled component %s\n",
 786                         debugstr_w(component));
 787
 788         /* the action taken was the same as the current install state */
 789         comp->Action = comp->Installed;
 790
 791         return ERROR_SUCCESS;
 792     }
 793
 794     comp->Action = INSTALLSTATE_LOCAL;
 795
 796     file_key = MSI_RecordGetString(row,3);
 797     if (!file_key)
 798     {
 799         ERR("Unable to get file key\n");
 800         return ERROR_FUNCTION_FAILED;
 801     }
 802
 803     rc = get_file_target(package,file_key,&file);
 804
 805     if (rc != ERROR_SUCCESS)
 806     {
 807         ERR("Original file unknown %s\n",debugstr_w(file_key));
 808         return ERROR_SUCCESS;
 809     }
 810
 811     if (MSI_RecordIsNull(row,4))
 812         strcpyW(dest_name,strrchrW(file->TargetPath,'\\')+1);
 813     else
 814     {
 815         sz=0x100;
 816         MSI_RecordGetStringW(row,4,dest_name,&sz);
 817         reduce_to_longfilename(dest_name);
 818     }
 819
 820     if (MSI_RecordIsNull(row,5))
 821     {
 822         LPWSTR p;
 823         dest_path = strdupW(file->TargetPath);
 824         p = strrchrW(dest_path,'\\');
 825         if (p)
 826             *p=0;
 827     }
 828     else
 829     {
 830         LPCWSTR destkey;
 831         destkey = MSI_RecordGetString(row,5);
 832         dest_path = resolve_folder(package, destkey, FALSE,FALSE,NULL);
 833         if (!dest_path)
 834         {
 835             /* try a Property */
 836             dest_path = msi_dup_property( package, destkey );
 837             if (!dest_path)
 838             {
 839                 FIXME("Unable to get destination folder, try AppSearch properties\n");
 840                 return ERROR_SUCCESS;
 841             }
 842         }
 843     }
 844
 845     dest = build_directory_name(2, dest_path, dest_name);
 846
 847     TRACE("Duplicating file %s to %s\n",debugstr_w(file->TargetPath),
 848                     debugstr_w(dest));
 849
 850     if (strcmpW(file->TargetPath,dest))
 851         rc = !CopyFileW(file->TargetPath,dest,TRUE);
 852     else
 853         rc = ERROR_SUCCESS;
 854
 855     if (rc != ERROR_SUCCESS)
 856         ERR("Failed to copy file %s -> %s, last error %d\n",
 857             debugstr_w(file->TargetPath), debugstr_w(dest_path), GetLastError());
 858
 859     FIXME("We should track these duplicate files as well\n");
 860
 861     msi_free(dest_path);
 862     msi_free(dest);
 863
 864     msi_file_update_ui(package, file, szDuplicateFiles);
 865
 866     return ERROR_SUCCESS;
 867 }
 868
 869 UINT ACTION_DuplicateFiles(MSIPACKAGE *package)
 870 {
 871     UINT rc;
 872     MSIQUERY * view;
 873     static const WCHAR ExecSeqQuery[] =
 874         {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
 875          '`','D','u','p','l','i','c','a','t','e','F','i','l','e','`',0};
 876
 877     rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
 878     if (rc != ERROR_SUCCESS)
 879         return ERROR_SUCCESS;
 880
 881     rc = MSI_IterateRecords(view, NULL, ITERATE_DuplicateFiles, package);
 882     msiobj_release(&view->hdr);
 883
 884     return rc;
 885 }
 886
 887 /* compares the version of a file read from the filesystem and
 888  * the version specified in the File table
 889  */
 890 static int msi_compare_file_version( MSIFILE *file )
 891 {
 892     WCHAR version[MAX_PATH];
 893     DWORD size;
 894     UINT r;
 895
 896     size = MAX_PATH;
 897     version[0] = '\0';
 898     r = MsiGetFileVersionW( file->TargetPath, version, &size, NULL, NULL );
 899     if ( r != ERROR_SUCCESS )
 900         return 0;
 901
 902     return lstrcmpW( version, file->Version );
 903 }
 904
 905 UINT ACTION_RemoveFiles( MSIPACKAGE *package )
 906 {
 907     MSIFILE *file;
 908
 909     LIST_FOR_EACH_ENTRY( file, &package->files, MSIFILE, entry )
 910     {
 911         MSIRECORD *uirow;
 912         LPWSTR uipath, p;
 913
 914         if ( !file->Component )
 915             continue;
 916         if ( file->Component->Installed == INSTALLSTATE_LOCAL )
 917             continue;
 918
 919         if ( file->state == msifs_installed )
 920             ERR("removing installed file %s\n", debugstr_w(file->TargetPath));
 921
 922         if ( file->state != msifs_present )
 923             continue;
 924
 925         /* only remove a file if the version to be installed
 926          * is strictly newer than the old file
 927          */
 928         if ( msi_compare_file_version( file ) >= 0 )
 929             continue;
 930
 931         TRACE("removing %s\n", debugstr_w(file->File) );
 932         if ( !DeleteFileW( file->TargetPath ) )
 933             ERR("failed to delete %s\n",  debugstr_w(file->TargetPath) );
 934         file->state = msifs_missing;
 935
 936         /* the UI chunk */
 937         uirow = MSI_CreateRecord( 9 );
 938         MSI_RecordSetStringW( uirow, 1, file->FileName );
 939         uipath = strdupW( file->TargetPath );
 940         p = strrchrW(uipath,'\\');
 941         if (p)
 942             p[1]=0;
 943         MSI_RecordSetStringW( uirow, 9, uipath);
 944         ui_actiondata( package, szRemoveFiles, uirow);
 945         msiobj_release( &uirow->hdr );
 946         msi_free( uipath );
 947         /* FIXME: call ui_progress here? */
 948     }
 949
 950     return ERROR_SUCCESS;
 951 }