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
  28  * PatchFiles (TODO)
  29  * RemoveDuplicateFiles
  30  * RemoveFiles
  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 "msipriv.h"
  43 #include "winuser.h"
  44 #include "winreg.h"
  45 #include "shlwapi.h"
  46 #include "wine/unicode.h"
  47
  48 WINE_DEFAULT_DEBUG_CHANNEL(msi);
  49
  50 static void msi_file_update_ui( MSIPACKAGE *package, MSIFILE *f, const WCHAR *action )
  51 {
  52     MSIRECORD *uirow;
  53     LPWSTR uipath, p;
  54
  55     /* the UI chunk */
  56     uirow = MSI_CreateRecord( 9 );
  57     MSI_RecordSetStringW( uirow, 1, f->FileName );
  58     uipath = strdupW( f->TargetPath );
  59     p = strrchrW(uipath,'\\');
  60     if (p)
  61         p[1]=0;
  62     MSI_RecordSetStringW( uirow, 9, uipath);
  63     MSI_RecordSetInteger( uirow, 6, f->FileSize );
  64     ui_actiondata( package, action, uirow);
  65     msiobj_release( &uirow->hdr );
  66     msi_free( uipath );
  67     ui_progress( package, 2, f->FileSize, 0, 0);
  68 }
  69
  70 /* compares the version of a file read from the filesystem and
  71  * the version specified in the File table
  72  */
  73 static int msi_compare_file_version(MSIFILE *file)
  74 {
  75     WCHAR version[MAX_PATH];
  76     DWORD size;
  77     UINT r;
  78
  79     size = MAX_PATH;
  80     version[0] = '\0';
  81     r = MsiGetFileVersionW(file->TargetPath, version, &size, NULL, NULL);
  82     if (r != ERROR_SUCCESS)
  83         return 0;
  84
  85     return lstrcmpW(version, file->Version);
  86 }
  87
  88 static void schedule_install_files(MSIPACKAGE *package)
  89 {
  90     MSIFILE *file;
  91
  92     LIST_FOR_EACH_ENTRY(file, &package->files, MSIFILE, entry)
  93     {
  94         if (file->Component->ActionRequest != INSTALLSTATE_LOCAL)
  95         {
  96             TRACE("File %s is not scheduled for install\n", debugstr_w(file->File));
  97
  98             ui_progress(package,2,file->FileSize,0,0);
  99             file->state = msifs_skipped;
 100         }
 101     }
 102 }
 103
 104 static UINT copy_file(MSIFILE *file, LPWSTR source)
 105 {
 106     BOOL ret;
 107
 108     ret = CopyFileW(source, file->TargetPath, FALSE);
 109     if (!ret)
 110         return GetLastError();
 111
 112     SetFileAttributesW(file->TargetPath, FILE_ATTRIBUTE_NORMAL);
 113
 114     file->state = msifs_installed;
 115     return ERROR_SUCCESS;
 116 }
 117
 118 static UINT copy_install_file(MSIPACKAGE *package, MSIFILE *file, LPWSTR source)
 119 {
 120     UINT gle;
 121
 122     TRACE("Copying %s to %s\n", debugstr_w(source),
 123           debugstr_w(file->TargetPath));
 124
 125     gle = copy_file(file, source);
 126     if (gle == ERROR_SUCCESS)
 127         return gle;
 128
 129     if (gle == ERROR_ALREADY_EXISTS && file->state == msifs_overwrite)
 130     {
 131         TRACE("overwriting existing file\n");
 132         return ERROR_SUCCESS;
 133     }
 134     else if (gle == ERROR_ACCESS_DENIED)
 135     {
 136         SetFileAttributesW(file->TargetPath, FILE_ATTRIBUTE_NORMAL);
 137
 138         gle = copy_file(file, source);
 139         TRACE("Overwriting existing file: %d\n", gle);
 140     }
 141     if (gle == ERROR_SHARING_VIOLATION || gle == ERROR_USER_MAPPED_FILE)
 142     {
 143         WCHAR tmpfileW[MAX_PATH], *pathW, *p;
 144         DWORD len;
 145
 146         TRACE("file in use, scheduling rename operation\n");
 147
 148         GetTempFileNameW(szBackSlash, szMsi, 0, tmpfileW);
 149         len = strlenW(file->TargetPath) + strlenW(tmpfileW) + 1;
 150         if (!(pathW = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR))))
 151             return ERROR_OUTOFMEMORY;
 152
 153         strcpyW(pathW, file->TargetPath);
 154         if ((p = strrchrW(pathW, '\\'))) *p = 0;
 155         strcatW(pathW, tmpfileW);
 156
 157         if (CopyFileW(source, pathW, FALSE) &&
 158             MoveFileExW(file->TargetPath, NULL, MOVEFILE_DELAY_UNTIL_REBOOT) &&
 159             MoveFileExW(pathW, file->TargetPath, MOVEFILE_DELAY_UNTIL_REBOOT))
 160         {
 161             file->state = msifs_installed;
 162             package->need_reboot = 1;
 163             gle = ERROR_SUCCESS;
 164         }
 165         else
 166         {
 167             gle = GetLastError();
 168             WARN("failed to schedule rename operation: %d)\n", gle);
 169         }
 170         HeapFree(GetProcessHeap(), 0, pathW);
 171     }
 172
 173     return gle;
 174 }
 175
 176 static BOOL check_dest_hash_matches(MSIFILE *file)
 177 {
 178     MSIFILEHASHINFO hash;
 179     UINT r;
 180
 181     if (!file->hash.dwFileHashInfoSize)
 182         return FALSE;
 183
 184     hash.dwFileHashInfoSize = sizeof(MSIFILEHASHINFO);
 185     r = MsiGetFileHashW(file->TargetPath, 0, &hash);
 186     if (r != ERROR_SUCCESS)
 187         return FALSE;
 188
 189     return !memcmp(&hash, &file->hash, sizeof(MSIFILEHASHINFO));
 190 }
 191
 192 static BOOL installfiles_cb(MSIPACKAGE *package, LPCWSTR file, DWORD action,
 193                             LPWSTR *path, DWORD *attrs, PVOID user)
 194 {
 195     static MSIFILE *f = NULL;
 196
 197     if (action == MSICABEXTRACT_BEGINEXTRACT)
 198     {
 199         f = get_loaded_file(package, file);
 200         if (!f)
 201         {
 202             WARN("unknown file in cabinet (%s)\n", debugstr_w(file));
 203             return FALSE;
 204         }
 205
 206         if (f->state != msifs_missing && f->state != msifs_overwrite)
 207         {
 208             TRACE("Skipping extraction of %s\n", debugstr_w(file));
 209             return FALSE;
 210         }
 211
 212         msi_file_update_ui(package, f, szInstallFiles);
 213
 214         *path = strdupW(f->TargetPath);
 215         *attrs = f->Attributes;
 216     }
 217     else if (action == MSICABEXTRACT_FILEEXTRACTED)
 218     {
 219         f->state = msifs_installed;
 220         f = NULL;
 221     }
 222
 223     return TRUE;
 224 }
 225
 226 /*
 227  * ACTION_InstallFiles()
 228  *
 229  * For efficiency, this is done in two passes:
 230  * 1) Correct all the TargetPaths and determine what files are to be installed.
 231  * 2) Extract Cabinets and copy files.
 232  */
 233 UINT ACTION_InstallFiles(MSIPACKAGE *package)
 234 {
 235     MSIMEDIAINFO *mi;
 236     UINT rc = ERROR_SUCCESS;
 237     MSIFILE *file;
 238
 239     /* increment progress bar each time action data is sent */
 240     ui_progress(package,1,1,0,0);
 241
 242     schedule_install_files(package);
 243
 244     /*
 245      * Despite MSDN specifying that the CreateFolders action
 246      * should be called before InstallFiles, some installers don't
 247      * do that, and they seem to work correctly.  We need to create
 248      * directories here to make sure that the files can be copied.
 249      */
 250     msi_create_component_directories( package );
 251
 252     mi = msi_alloc_zero( sizeof(MSIMEDIAINFO) );
 253
 254     LIST_FOR_EACH_ENTRY( file, &package->files, MSIFILE, entry )
 255     {
 256         if (file->state != msifs_missing && !mi->is_continuous && file->state != msifs_overwrite)
 257             continue;
 258
 259         if (check_dest_hash_matches(file))
 260         {
 261             TRACE("File hashes match, not overwriting\n");
 262             continue;
 263         }
 264
 265         if (MsiGetFileVersionW(file->TargetPath, NULL, NULL, NULL, NULL) == ERROR_SUCCESS &&
 266             msi_compare_file_version(file) >= 0)
 267         {
 268             TRACE("Destination file version greater, not overwriting\n");
 269             continue;
 270         }
 271
 272         if (file->Sequence > mi->last_sequence || mi->is_continuous ||
 273             (file->IsCompressed && !mi->is_extracted))
 274         {
 275             MSICABDATA data;
 276
 277             rc = ready_media(package, file, mi);
 278             if (rc != ERROR_SUCCESS)
 279             {
 280                 ERR("Failed to ready media\n");
 281                 break;
 282             }
 283
 284             data.mi = mi;
 285             data.package = package;
 286             data.cb = installfiles_cb;
 287             data.user = NULL;
 288
 289             if (file->IsCompressed &&
 290                 !msi_cabextract(package, mi, &data))
 291             {
 292                 ERR("Failed to extract cabinet: %s\n", debugstr_w(mi->cabinet));
 293                 rc = ERROR_FUNCTION_FAILED;
 294                 break;
 295             }
 296         }
 297
 298         if (!file->IsCompressed)
 299         {
 300             LPWSTR source = resolve_file_source(package, file);
 301
 302             TRACE("file paths %s to %s\n", debugstr_w(source),
 303                   debugstr_w(file->TargetPath));
 304
 305             msi_file_update_ui(package, file, szInstallFiles);
 306             rc = copy_install_file(package, file, source);
 307             if (rc != ERROR_SUCCESS)
 308             {
 309                 ERR("Failed to copy %s to %s (%d)\n", debugstr_w(source),
 310                     debugstr_w(file->TargetPath), rc);
 311                 rc = ERROR_INSTALL_FAILURE;
 312                 msi_free(source);
 313                 break;
 314             }
 315
 316             msi_free(source);
 317         }
 318         else if (file->state != msifs_installed)
 319         {
 320             ERR("compressed file wasn't extracted (%s)\n",
 321                 debugstr_w(file->TargetPath));
 322             rc = ERROR_INSTALL_FAILURE;
 323             break;
 324         }
 325     }
 326
 327     msi_free_media_info(mi);
 328     return rc;
 329 }
 330
 331 #define is_dot_dir(x) ((x[0] == '.') && ((x[1] == 0) || ((x[1] == '.') && (x[2] == 0))))
 332
 333 typedef struct
 334 {
 335     struct list entry;
 336     LPWSTR sourcename;
 337     LPWSTR destname;
 338     LPWSTR source;
 339     LPWSTR dest;
 340 } FILE_LIST;
 341
 342 static BOOL msi_move_file(LPCWSTR source, LPCWSTR dest, int options)
 343 {
 344     BOOL ret;
 345
 346     if (GetFileAttributesW(source) == FILE_ATTRIBUTE_DIRECTORY ||
 347         GetFileAttributesW(dest) == FILE_ATTRIBUTE_DIRECTORY)
 348     {
 349         WARN("Source or dest is directory, not moving\n");
 350         return FALSE;
 351     }
 352
 353     if (options == msidbMoveFileOptionsMove)
 354     {
 355         TRACE("moving %s -> %s\n", debugstr_w(source), debugstr_w(dest));
 356         ret = MoveFileExW(source, dest, MOVEFILE_REPLACE_EXISTING);
 357         if (!ret)
 358         {
 359             WARN("MoveFile failed: %d\n", GetLastError());
 360             return FALSE;
 361         }
 362     }
 363     else
 364     {
 365         TRACE("copying %s -> %s\n", debugstr_w(source), debugstr_w(dest));
 366         ret = CopyFileW(source, dest, FALSE);
 367         if (!ret)
 368         {
 369             WARN("CopyFile failed: %d\n", GetLastError());
 370             return FALSE;
 371         }
 372     }
 373
 374     return TRUE;
 375 }
 376
 377 static LPWSTR wildcard_to_file(LPWSTR wildcard, LPWSTR filename)
 378 {
 379     LPWSTR path, ptr;
 380     DWORD dirlen, pathlen;
 381
 382     ptr = strrchrW(wildcard, '\\');
 383     dirlen = ptr - wildcard + 1;
 384
 385     pathlen = dirlen + lstrlenW(filename) + 1;
 386     path = msi_alloc(pathlen * sizeof(WCHAR));
 387
 388     lstrcpynW(path, wildcard, dirlen + 1);
 389     lstrcatW(path, filename);
 390
 391     return path;
 392 }
 393
 394 static void free_file_entry(FILE_LIST *file)
 395 {
 396     msi_free(file->source);
 397     msi_free(file->dest);
 398     msi_free(file);
 399 }
 400
 401 static void free_list(FILE_LIST *list)
 402 {
 403     while (!list_empty(&list->entry))
 404     {
 405         FILE_LIST *file = LIST_ENTRY(list_head(&list->entry), FILE_LIST, entry);
 406
 407         list_remove(&file->entry);
 408         free_file_entry(file);
 409     }
 410 }
 411
 412 static BOOL add_wildcard(FILE_LIST *files, LPWSTR source, LPWSTR dest)
 413 {
 414     FILE_LIST *new, *file;
 415     LPWSTR ptr, filename;
 416     DWORD size;
 417
 418     new = msi_alloc_zero(sizeof(FILE_LIST));
 419     if (!new)
 420         return FALSE;
 421
 422     new->source = strdupW(source);
 423     ptr = strrchrW(dest, '\\') + 1;
 424     filename = strrchrW(new->source, '\\') + 1;
 425
 426     new->sourcename = filename;
 427
 428     if (*ptr)
 429         new->destname = ptr;
 430     else
 431         new->destname = new->sourcename;
 432
 433     size = (ptr - dest) + lstrlenW(filename) + 1;
 434     new->dest = msi_alloc(size * sizeof(WCHAR));
 435     if (!new->dest)
 436     {
 437         free_file_entry(new);
 438         return FALSE;
 439     }
 440
 441     lstrcpynW(new->dest, dest, ptr - dest + 1);
 442     lstrcatW(new->dest, filename);
 443
 444     if (list_empty(&files->entry))
 445     {
 446         list_add_head(&files->entry, &new->entry);
 447         return TRUE;
 448     }
 449
 450     LIST_FOR_EACH_ENTRY(file, &files->entry, FILE_LIST, entry)
 451     {
 452         if (lstrcmpW(source, file->source) < 0)
 453         {
 454             list_add_before(&file->entry, &new->entry);
 455             return TRUE;
 456         }
 457     }
 458
 459     list_add_after(&file->entry, &new->entry);
 460     return TRUE;
 461 }
 462
 463 static BOOL move_files_wildcard(LPWSTR source, LPWSTR dest, int options)
 464 {
 465     WIN32_FIND_DATAW wfd;
 466     HANDLE hfile;
 467     LPWSTR path;
 468     BOOL res;
 469     FILE_LIST files, *file;
 470     DWORD size;
 471
 472     hfile = FindFirstFileW(source, &wfd);
 473     if (hfile == INVALID_HANDLE_VALUE) return FALSE;
 474
 475     list_init(&files.entry);
 476
 477     for (res = TRUE; res; res = FindNextFileW(hfile, &wfd))
 478     {
 479         if (is_dot_dir(wfd.cFileName)) continue;
 480
 481         path = wildcard_to_file(source, wfd.cFileName);
 482         if (!path)
 483         {
 484             res = FALSE;
 485             goto done;
 486         }
 487
 488         add_wildcard(&files, path, dest);
 489         msi_free(path);
 490     }
 491
 492     /* no files match the wildcard */
 493     if (list_empty(&files.entry))
 494         goto done;
 495
 496     /* only the first wildcard match gets renamed to dest */
 497     file = LIST_ENTRY(list_head(&files.entry), FILE_LIST, entry);
 498     size = (strrchrW(file->dest, '\\') - file->dest) + lstrlenW(file->destname) + 2;
 499     file->dest = msi_realloc(file->dest, size * sizeof(WCHAR));
 500     if (!file->dest)
 501     {
 502         res = FALSE;
 503         goto done;
 504     }
 505
 506     /* file->dest may be shorter after the reallocation, so add a NULL
 507      * terminator.  This is needed for the call to strrchrW, as there will no
 508      * longer be a NULL terminator within the bounds of the allocation in this case.
 509      */
 510     file->dest[size - 1] = '\0';
 511     lstrcpyW(strrchrW(file->dest, '\\') + 1, file->destname);
 512
 513     while (!list_empty(&files.entry))
 514     {
 515         file = LIST_ENTRY(list_head(&files.entry), FILE_LIST, entry);
 516
 517         msi_move_file(file->source, file->dest, options);
 518
 519         list_remove(&file->entry);
 520         free_file_entry(file);
 521     }
 522
 523     res = TRUE;
 524
 525 done:
 526     free_list(&files);
 527     FindClose(hfile);
 528     return res;
 529 }
 530
 531 static UINT ITERATE_MoveFiles( MSIRECORD *rec, LPVOID param )
 532 {
 533     MSIPACKAGE *package = param;
 534     MSICOMPONENT *comp;
 535     LPCWSTR sourcename, component;
 536     LPWSTR destname = NULL;
 537     LPWSTR sourcedir = NULL, destdir = NULL;
 538     LPWSTR source = NULL, dest = NULL;
 539     int options;
 540     DWORD size;
 541     BOOL ret, wildcards;
 542
 543     component = MSI_RecordGetString(rec, 2);
 544     comp = get_loaded_component(package, component);
 545     if (!comp)
 546         return ERROR_SUCCESS;
 547
 548     if (comp->ActionRequest != INSTALLSTATE_LOCAL && comp->ActionRequest != INSTALLSTATE_SOURCE)
 549     {
 550         TRACE("Component not scheduled for installation: %s\n", debugstr_w(component));
 551         comp->Action = comp->Installed;
 552         return ERROR_SUCCESS;
 553     }
 554     comp->Action = comp->ActionRequest;
 555
 556     sourcename = MSI_RecordGetString(rec, 3);
 557     options = MSI_RecordGetInteger(rec, 7);
 558
 559     sourcedir = msi_dup_property(package, MSI_RecordGetString(rec, 5));
 560     if (!sourcedir)
 561         goto done;
 562
 563     destdir = msi_dup_property(package, MSI_RecordGetString(rec, 6));
 564     if (!destdir)
 565         goto done;
 566
 567     if (!sourcename)
 568     {
 569         if (GetFileAttributesW(sourcedir) == INVALID_FILE_ATTRIBUTES)
 570             goto done;
 571
 572         source = strdupW(sourcedir);
 573         if (!source)
 574             goto done;
 575     }
 576     else
 577     {
 578         size = lstrlenW(sourcedir) + lstrlenW(sourcename) + 2;
 579         source = msi_alloc(size * sizeof(WCHAR));
 580         if (!source)
 581             goto done;
 582
 583         lstrcpyW(source, sourcedir);
 584         if (source[lstrlenW(source) - 1] != '\\')
 585             lstrcatW(source, szBackSlash);
 586         lstrcatW(source, sourcename);
 587     }
 588
 589     wildcards = strchrW(source, '*') || strchrW(source, '?');
 590
 591     if (MSI_RecordIsNull(rec, 4))
 592     {
 593         if (!wildcards)
 594         {
 595             destname = strdupW(sourcename);
 596             if (!destname)
 597                 goto done;
 598         }
 599     }
 600     else
 601     {
 602         destname = strdupW(MSI_RecordGetString(rec, 4));
 603         if (destname)
 604             reduce_to_longfilename(destname);
 605     }
 606
 607     size = 0;
 608     if (destname)
 609         size = lstrlenW(destname);
 610
 611     size += lstrlenW(destdir) + 2;
 612     dest = msi_alloc(size * sizeof(WCHAR));
 613     if (!dest)
 614         goto done;
 615
 616     lstrcpyW(dest, destdir);
 617     if (dest[lstrlenW(dest) - 1] != '\\')
 618         lstrcatW(dest, szBackSlash);
 619
 620     if (destname)
 621         lstrcatW(dest, destname);
 622
 623     if (GetFileAttributesW(destdir) == INVALID_FILE_ATTRIBUTES)
 624     {
 625         ret = CreateDirectoryW(destdir, NULL);
 626         if (!ret)
 627         {
 628             WARN("CreateDirectory failed: %d\n", GetLastError());
 629             return ERROR_SUCCESS;
 630         }
 631     }
 632
 633     if (!wildcards)
 634         msi_move_file(source, dest, options);
 635     else
 636         move_files_wildcard(source, dest, options);
 637
 638 done:
 639     msi_free(sourcedir);
 640     msi_free(destdir);
 641     msi_free(destname);
 642     msi_free(source);
 643     msi_free(dest);
 644
 645     return ERROR_SUCCESS;
 646 }
 647
 648 UINT ACTION_MoveFiles( MSIPACKAGE *package )
 649 {
 650     UINT rc;
 651     MSIQUERY *view;
 652
 653     static const WCHAR ExecSeqQuery[] =
 654         {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
 655          '`','M','o','v','e','F','i','l','e','`',0};
 656
 657     rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
 658     if (rc != ERROR_SUCCESS)
 659         return ERROR_SUCCESS;
 660
 661     rc = MSI_IterateRecords(view, NULL, ITERATE_MoveFiles, package);
 662     msiobj_release(&view->hdr);
 663
 664     return rc;
 665 }
 666
 667 static WCHAR *get_duplicate_filename( MSIPACKAGE *package, MSIRECORD *row, const WCHAR *file_key, const WCHAR *src )
 668 {
 669     DWORD len;
 670     WCHAR *dst_name, *dst_path, *dst;
 671
 672     if (MSI_RecordIsNull( row, 4 ))
 673     {
 674         len = strlenW( src ) + 1;
 675         if (!(dst_name = msi_alloc( len * sizeof(WCHAR)))) return NULL;
 676         strcpyW( dst_name, strrchrW( src, '\\' ) + 1 );
 677     }
 678     else
 679     {
 680         MSI_RecordGetStringW( row, 4, NULL, &len );
 681         if (!(dst_name = msi_alloc( ++len * sizeof(WCHAR) ))) return NULL;
 682         MSI_RecordGetStringW( row, 4, dst_name, &len );
 683         reduce_to_longfilename( dst_name );
 684     }
 685
 686     if (MSI_RecordIsNull( row, 5 ))
 687     {
 688         WCHAR *p;
 689         dst_path = strdupW( src );
 690         p = strrchrW( dst_path, '\\' );
 691         if (p) *p = 0;
 692     }
 693     else
 694     {
 695         const WCHAR *dst_key = MSI_RecordGetString( row, 5 );
 696
 697         dst_path = resolve_folder( package, dst_key, FALSE, FALSE, TRUE, NULL );
 698         if (!dst_path)
 699         {
 700             /* try a property */
 701             dst_path = msi_dup_property( package, dst_key );
 702             if (!dst_path)
 703             {
 704                 FIXME("Unable to get destination folder, try AppSearch properties\n");
 705                 msi_free( dst_name );
 706                 return NULL;
 707             }
 708         }
 709     }
 710
 711     dst = build_directory_name( 2, dst_path, dst_name );
 712     create_full_pathW( dst_path );
 713
 714     msi_free( dst_name );
 715     msi_free( dst_path );
 716     return dst;
 717 }
 718
 719 static UINT ITERATE_DuplicateFiles(MSIRECORD *row, LPVOID param)
 720 {
 721     MSIPACKAGE *package = param;
 722     LPWSTR dest;
 723     LPCWSTR file_key, component;
 724     MSICOMPONENT *comp;
 725     MSIRECORD *uirow;
 726     MSIFILE *file;
 727
 728     component = MSI_RecordGetString(row,2);
 729     comp = get_loaded_component(package,component);
 730     if (!comp)
 731         return ERROR_SUCCESS;
 732
 733     if (comp->ActionRequest != INSTALLSTATE_LOCAL)
 734     {
 735         TRACE("Component not scheduled for installation %s\n", debugstr_w(component));
 736         comp->Action = comp->Installed;
 737         return ERROR_SUCCESS;
 738     }
 739     comp->Action = INSTALLSTATE_LOCAL;
 740
 741     file_key = MSI_RecordGetString(row,3);
 742     if (!file_key)
 743     {
 744         ERR("Unable to get file key\n");
 745         return ERROR_FUNCTION_FAILED;
 746     }
 747
 748     file = get_loaded_file( package, file_key );
 749     if (!file)
 750     {
 751         ERR("Original file unknown %s\n", debugstr_w(file_key));
 752         return ERROR_SUCCESS;
 753     }
 754
 755     dest = get_duplicate_filename( package, row, file_key, file->TargetPath );
 756     if (!dest)
 757     {
 758         WARN("Unable to get duplicate filename\n");
 759         return ERROR_SUCCESS;
 760     }
 761
 762     TRACE("Duplicating file %s to %s\n", debugstr_w(file->TargetPath), debugstr_w(dest));
 763
 764     if (!CopyFileW( file->TargetPath, dest, TRUE ))
 765     {
 766         WARN("Failed to copy file %s -> %s (%u)\n",
 767              debugstr_w(file->TargetPath), debugstr_w(dest), GetLastError());
 768     }
 769
 770     FIXME("We should track these duplicate files as well\n");
 771
 772     uirow = MSI_CreateRecord( 9 );
 773     MSI_RecordSetStringW( uirow, 1, MSI_RecordGetString( row, 1 ) );
 774     MSI_RecordSetInteger( uirow, 6, file->FileSize );
 775     MSI_RecordSetStringW( uirow, 9, MSI_RecordGetString( row, 5 ) );
 776     ui_actiondata( package, szDuplicateFiles, uirow );
 777     msiobj_release( &uirow->hdr );
 778
 779     msi_free(dest);
 780     return ERROR_SUCCESS;
 781 }
 782
 783 UINT ACTION_DuplicateFiles(MSIPACKAGE *package)
 784 {
 785     UINT rc;
 786     MSIQUERY * view;
 787     static const WCHAR ExecSeqQuery[] =
 788         {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
 789          '`','D','u','p','l','i','c','a','t','e','F','i','l','e','`',0};
 790
 791     rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
 792     if (rc != ERROR_SUCCESS)
 793         return ERROR_SUCCESS;
 794
 795     rc = MSI_IterateRecords(view, NULL, ITERATE_DuplicateFiles, package);
 796     msiobj_release(&view->hdr);
 797
 798     return rc;
 799 }
 800
 801 static UINT ITERATE_RemoveDuplicateFiles( MSIRECORD *row, LPVOID param )
 802 {
 803     MSIPACKAGE *package = param;
 804     LPWSTR dest;
 805     LPCWSTR file_key, component;
 806     MSICOMPONENT *comp;
 807     MSIRECORD *uirow;
 808     MSIFILE *file;
 809
 810     component = MSI_RecordGetString( row, 2 );
 811     comp = get_loaded_component( package, component );
 812     if (!comp)
 813         return ERROR_SUCCESS;
 814
 815     if (comp->ActionRequest != INSTALLSTATE_ABSENT)
 816     {
 817         TRACE("Component not scheduled for removal %s\n", debugstr_w(component));
 818         comp->Action = comp->Installed;
 819         return ERROR_SUCCESS;
 820     }
 821     comp->Action = INSTALLSTATE_ABSENT;
 822
 823     file_key = MSI_RecordGetString( row, 3 );
 824     if (!file_key)
 825     {
 826         ERR("Unable to get file key\n");
 827         return ERROR_FUNCTION_FAILED;
 828     }
 829
 830     file = get_loaded_file( package, file_key );
 831     if (!file)
 832     {
 833         ERR("Original file unknown %s\n", debugstr_w(file_key));
 834         return ERROR_SUCCESS;
 835     }
 836
 837     dest = get_duplicate_filename( package, row, file_key, file->TargetPath );
 838     if (!dest)
 839     {
 840         WARN("Unable to get duplicate filename\n");
 841         return ERROR_SUCCESS;
 842     }
 843
 844     TRACE("Removing duplicate %s of %s\n", debugstr_w(dest), debugstr_w(file->TargetPath));
 845
 846     if (!DeleteFileW( dest ))
 847     {
 848         WARN("Failed to delete duplicate file %s (%u)\n", debugstr_w(dest), GetLastError());
 849     }
 850
 851     uirow = MSI_CreateRecord( 9 );
 852     MSI_RecordSetStringW( uirow, 1, MSI_RecordGetString( row, 1 ) );
 853     MSI_RecordSetStringW( uirow, 9, MSI_RecordGetString( row, 5 ) );
 854     ui_actiondata( package, szRemoveDuplicateFiles, uirow );
 855     msiobj_release( &uirow->hdr );
 856
 857     msi_free(dest);
 858     return ERROR_SUCCESS;
 859 }
 860
 861 UINT ACTION_RemoveDuplicateFiles( MSIPACKAGE *package )
 862 {
 863     UINT rc;
 864     MSIQUERY *view;
 865     static const WCHAR query[] =
 866         {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
 867          '`','D','u','p','l','i','c','a','t','e','F','i','l','e','`',0};
 868
 869     rc = MSI_DatabaseOpenViewW( package->db, query, &view );
 870     if (rc != ERROR_SUCCESS)
 871         return ERROR_SUCCESS;
 872
 873     rc = MSI_IterateRecords( view, NULL, ITERATE_RemoveDuplicateFiles, package );
 874     msiobj_release( &view->hdr );
 875
 876     return rc;
 877 }
 878
 879 static BOOL verify_comp_for_removal(MSICOMPONENT *comp, UINT install_mode)
 880 {
 881     INSTALLSTATE request = comp->ActionRequest;
 882
 883     if (request == INSTALLSTATE_UNKNOWN)
 884         return FALSE;
 885
 886     if (install_mode == msidbRemoveFileInstallModeOnInstall &&
 887         (request == INSTALLSTATE_LOCAL || request == INSTALLSTATE_SOURCE))
 888         return TRUE;
 889
 890     if (request == INSTALLSTATE_ABSENT)
 891     {
 892         if (!comp->ComponentId)
 893             return FALSE;
 894
 895         if (install_mode == msidbRemoveFileInstallModeOnRemove)
 896             return TRUE;
 897     }
 898
 899     if (install_mode == msidbRemoveFileInstallModeOnBoth)
 900         return TRUE;
 901
 902     return FALSE;
 903 }
 904
 905 static UINT ITERATE_RemoveFiles(MSIRECORD *row, LPVOID param)
 906 {
 907     MSIPACKAGE *package = param;
 908     MSICOMPONENT *comp;
 909     LPCWSTR component, filename, dirprop;
 910     UINT install_mode;
 911     LPWSTR dir = NULL, path = NULL;
 912     DWORD size;
 913     UINT r;
 914
 915     component = MSI_RecordGetString(row, 2);
 916     filename = MSI_RecordGetString(row, 3);
 917     dirprop = MSI_RecordGetString(row, 4);
 918     install_mode = MSI_RecordGetInteger(row, 5);
 919
 920     comp = get_loaded_component(package, component);
 921     if (!comp)
 922     {
 923         ERR("Invalid component: %s\n", debugstr_w(component));
 924         return ERROR_FUNCTION_FAILED;
 925     }
 926
 927     if (!verify_comp_for_removal(comp, install_mode))
 928     {
 929         TRACE("Skipping removal due to missing conditions\n");
 930         comp->Action = comp->Installed;
 931         return ERROR_SUCCESS;
 932     }
 933
 934     dir = msi_dup_property(package, dirprop);
 935     if (!dir)
 936         return ERROR_OUTOFMEMORY;
 937
 938     size = (filename != NULL) ? lstrlenW(filename) : 0;
 939     size += lstrlenW(dir) + 2;
 940     path = msi_alloc(size * sizeof(WCHAR));
 941     if (!path)
 942     {
 943         r = ERROR_OUTOFMEMORY;
 944         goto done;
 945     }
 946
 947     if (filename)
 948     {
 949         lstrcpyW(path, dir);
 950         PathAddBackslashW(path);
 951         lstrcatW(path, filename);
 952
 953         TRACE("Deleting misc file: %s\n", debugstr_w(path));
 954         DeleteFileW(path);
 955     }
 956     else
 957     {
 958         TRACE("Removing misc directory: %s\n", debugstr_w(dir));
 959         RemoveDirectoryW(dir);
 960     }
 961
 962 done:
 963     msi_free(path);
 964     msi_free(dir);
 965     return ERROR_SUCCESS;
 966 }
 967
 968 UINT ACTION_RemoveFiles( MSIPACKAGE *package )
 969 {
 970     MSIQUERY *view;
 971     MSIFILE *file;
 972     UINT r;
 973
 974     static const WCHAR query[] = {
 975         'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
 976         '`','R','e','m','o','v','e','F','i','l','e','`',0};
 977     static const WCHAR folder_query[] = {
 978         'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
 979         '`','C','r','e','a','t','e','F','o','l','d','e','r','`',0};
 980
 981     r = MSI_DatabaseOpenViewW(package->db, query, &view);
 982     if (r == ERROR_SUCCESS)
 983     {
 984         MSI_IterateRecords(view, NULL, ITERATE_RemoveFiles, package);
 985         msiobj_release(&view->hdr);
 986     }
 987
 988     r = MSI_DatabaseOpenViewW(package->db, folder_query, &view);
 989     if (r == ERROR_SUCCESS)
 990         msiobj_release(&view->hdr);
 991
 992     LIST_FOR_EACH_ENTRY( file, &package->files, MSIFILE, entry )
 993     {
 994         MSIRECORD *uirow;
 995         LPWSTR dir, uipath, p;
 996
 997         if ( file->state == msifs_installed )
 998             ERR("removing installed file %s\n", debugstr_w(file->TargetPath));
 999
1000         if ( file->Component->ActionRequest != INSTALLSTATE_ABSENT ||
1001              file->Component->Installed == INSTALLSTATE_SOURCE )
1002             continue;
1003
1004         /* don't remove a file if the old file
1005          * is strictly newer than the version to be installed
1006          */
1007         if ( msi_compare_file_version( file ) < 0 )
1008             continue;
1009
1010         TRACE("removing %s\n", debugstr_w(file->File) );
1011         if (!DeleteFileW( file->TargetPath ))
1012         {
1013             WARN("failed to delete %s\n",  debugstr_w(file->TargetPath));
1014         }
1015         /* FIXME: check persistence for each directory */
1016         else if (r && (dir = strdupW( file->TargetPath )))
1017         {
1018             if ((p = strrchrW( dir, '\\' ))) *p = 0;
1019             RemoveDirectoryW( dir );
1020             msi_free( dir );
1021         }
1022         file->state = msifs_missing;
1023
1024         /* the UI chunk */
1025         uirow = MSI_CreateRecord( 9 );
1026         MSI_RecordSetStringW( uirow, 1, file->FileName );
1027         uipath = strdupW( file->TargetPath );
1028         p = strrchrW(uipath,'\\');
1029         if (p)
1030             p[1]=0;
1031         MSI_RecordSetStringW( uirow, 9, uipath);
1032         ui_actiondata( package, szRemoveFiles, uirow);
1033         msiobj_release( &uirow->hdr );
1034         msi_free( uipath );
1035         /* FIXME: call ui_progress here? */
1036     }
1037
1038     return ERROR_SUCCESS;
1039 }