reactos/dll/win32/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 "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 extern const WCHAR szInstallFiles[];
  51 extern const WCHAR szDuplicateFiles[];
  52 extern const WCHAR szMoveFiles[];
  53 extern const WCHAR szPatchFiles[];
  54 extern const WCHAR szRemoveDuplicateFiles[];
  55 extern const WCHAR szRemoveFiles[];
  56
  57 static void msi_file_update_ui( MSIPACKAGE *package, MSIFILE *f, const WCHAR *action )
  58 {
  59     MSIRECORD *uirow;
  60     LPWSTR uipath, p;
  61
  62     /* the UI chunk */
  63     uirow = MSI_CreateRecord( 9 );
  64     MSI_RecordSetStringW( uirow, 1, f->FileName );
  65     uipath = strdupW( f->TargetPath );
  66     p = strrchrW(uipath,'\\');
  67     if (p)
  68         p[1]=0;
  69     MSI_RecordSetStringW( uirow, 9, uipath);
  70     MSI_RecordSetInteger( uirow, 6, f->FileSize );
  71     ui_actiondata( package, action, uirow);
  72     msiobj_release( &uirow->hdr );
  73     msi_free( uipath );
  74     ui_progress( package, 2, f->FileSize, 0, 0);
  75 }
  76
  77 /* compares the version of a file read from the filesystem and
  78  * the version specified in the File table
  79  */
  80 static int msi_compare_file_version(MSIFILE *file)
  81 {
  82     WCHAR version[MAX_PATH];
  83     DWORD size;
  84     UINT r;
  85
  86     size = MAX_PATH;
  87     version[0] = '\0';
  88     r = MsiGetFileVersionW(file->TargetPath, version, &size, NULL, NULL);
  89     if (r != ERROR_SUCCESS)
  90         return 0;
  91
  92     return lstrcmpW(version, file->Version);
  93 }
  94
  95 static UINT get_file_target(MSIPACKAGE *package, LPCWSTR file_key,
  96                             MSIFILE** file)
  97 {
  98     LIST_FOR_EACH_ENTRY( *file, &package->files, MSIFILE, entry )
  99     {
 100         if (lstrcmpW( file_key, (*file)->File )==0)
 101         {
 102             if ((*file)->state >= msifs_overwrite)
 103                 return ERROR_SUCCESS;
 104             else
 105                 return ERROR_FILE_NOT_FOUND;
 106         }
 107     }
 108
 109     return ERROR_FUNCTION_FAILED;
 110 }
 111
 112 static void schedule_install_files(MSIPACKAGE *package)
 113 {
 114     MSIFILE *file;
 115
 116     LIST_FOR_EACH_ENTRY(file, &package->files, MSIFILE, entry)
 117     {
 118         if (!ACTION_VerifyComponentForAction(file->Component, INSTALLSTATE_LOCAL))
 119         {
 120             TRACE("File %s is not scheduled for install\n", debugstr_w(file->File));
 121
 122             ui_progress(package,2,file->FileSize,0,0);
 123             file->state = msifs_skipped;
 124         }
 125     }
 126 }
 127
 128 static UINT copy_file(MSIFILE *file, LPWSTR source)
 129 {
 130     BOOL ret;
 131
 132     ret = CopyFileW(source, file->TargetPath, FALSE);
 133     if (!ret)
 134         return GetLastError();
 135
 136     SetFileAttributesW(file->TargetPath, FILE_ATTRIBUTE_NORMAL);
 137
 138     file->state = msifs_installed;
 139     return ERROR_SUCCESS;
 140 }
 141
 142 static UINT copy_install_file(MSIFILE *file, LPWSTR source)
 143 {
 144     UINT gle;
 145
 146     TRACE("Copying %s to %s\n", debugstr_w(source),
 147           debugstr_w(file->TargetPath));
 148
 149     gle = copy_file(file, source);
 150     if (gle == ERROR_SUCCESS)
 151         return gle;
 152
 153     if (gle == ERROR_ALREADY_EXISTS && file->state == msifs_overwrite)
 154     {
 155         TRACE("overwriting existing file\n");
 156         gle = ERROR_SUCCESS;
 157     }
 158     else if (gle == ERROR_ACCESS_DENIED)
 159     {
 160         SetFileAttributesW(file->TargetPath, FILE_ATTRIBUTE_NORMAL);
 161
 162         gle = copy_file(file, source);
 163         TRACE("Overwriting existing file: %d\n", gle);
 164     }
 165
 166     return gle;
 167 }
 168
 169 static BOOL check_dest_hash_matches(MSIFILE *file)
 170 {
 171     MSIFILEHASHINFO hash;
 172     UINT r;
 173
 174     if (!file->hash.dwFileHashInfoSize)
 175         return FALSE;
 176
 177     hash.dwFileHashInfoSize = sizeof(MSIFILEHASHINFO);
 178     r = MsiGetFileHashW(file->TargetPath, 0, &hash);
 179     if (r != ERROR_SUCCESS)
 180         return FALSE;
 181
 182     return !memcmp(&hash, &file->hash, sizeof(MSIFILEHASHINFO));
 183 }
 184
 185 static BOOL installfiles_cb(MSIPACKAGE *package, LPCWSTR file, DWORD action,
 186                             LPWSTR *path, DWORD *attrs, PVOID user)
 187 {
 188     static MSIFILE *f = NULL;
 189
 190     if (action == MSICABEXTRACT_BEGINEXTRACT)
 191     {
 192         f = get_loaded_file(package, file);
 193         if (!f)
 194         {
 195             WARN("unknown file in cabinet (%s)\n", debugstr_w(file));
 196             return FALSE;
 197         }
 198
 199         if (f->state != msifs_missing && f->state != msifs_overwrite)
 200         {
 201             TRACE("Skipping extraction of %s\n", debugstr_w(file));
 202             return FALSE;
 203         }
 204
 205         msi_file_update_ui(package, f, szInstallFiles);
 206
 207         *path = strdupW(f->TargetPath);
 208         *attrs = f->Attributes;
 209     }
 210     else if (action == MSICABEXTRACT_FILEEXTRACTED)
 211     {
 212         f->state = msifs_installed;
 213         f = NULL;
 214     }
 215
 216     return TRUE;
 217 }
 218
 219 /*
 220  * ACTION_InstallFiles()
 221  *
 222  * For efficiency, this is done in two passes:
 223  * 1) Correct all the TargetPaths and determine what files are to be installed.
 224  * 2) Extract Cabinets and copy files.
 225  */
 226 UINT ACTION_InstallFiles(MSIPACKAGE *package)
 227 {
 228     MSIMEDIAINFO *mi;
 229     UINT rc = ERROR_SUCCESS;
 230     MSIFILE *file;
 231
 232     /* increment progress bar each time action data is sent */
 233     ui_progress(package,1,1,0,0);
 234
 235     schedule_install_files(package);
 236
 237     /*
 238      * Despite MSDN specifying that the CreateFolders action
 239      * should be called before InstallFiles, some installers don't
 240      * do that, and they seem to work correctly.  We need to create
 241      * directories here to make sure that the files can be copied.
 242      */
 243     msi_create_component_directories( package );
 244
 245     mi = msi_alloc_zero( sizeof(MSIMEDIAINFO) );
 246
 247     LIST_FOR_EACH_ENTRY( file, &package->files, MSIFILE, entry )
 248     {
 249         if (file->state != msifs_missing && !mi->is_continuous && file->state != msifs_overwrite)
 250             continue;
 251
 252         if (check_dest_hash_matches(file))
 253         {
 254             TRACE("File hashes match, not overwriting\n");
 255             continue;
 256         }
 257
 258         if (MsiGetFileVersionW(file->TargetPath, NULL, NULL, NULL, NULL) == ERROR_SUCCESS &&
 259             msi_compare_file_version(file) >= 0)
 260         {
 261             TRACE("Destination file version greater, not overwriting\n");
 262             continue;
 263         }
 264
 265         if (file->Sequence > mi->last_sequence || mi->is_continuous ||
 266             (file->IsCompressed && !mi->is_extracted))
 267         {
 268             MSICABDATA data;
 269
 270             rc = ready_media(package, file, mi);
 271             if (rc != ERROR_SUCCESS)
 272             {
 273                 ERR("Failed to ready media\n");
 274                 break;
 275             }
 276
 277             data.mi = mi;
 278             data.package = package;
 279             data.cb = installfiles_cb;
 280             data.user = NULL;
 281
 282             if (file->IsCompressed &&
 283                 !msi_cabextract(package, mi, &data))
 284             {
 285                 ERR("Failed to extract cabinet: %s\n", debugstr_w(mi->cabinet));
 286                 rc = ERROR_FUNCTION_FAILED;
 287                 break;
 288             }
 289         }
 290
 291         if (!file->IsCompressed)
 292         {
 293             LPWSTR source = resolve_file_source(package, file);
 294
 295             TRACE("file paths %s to %s\n", debugstr_w(source),
 296                   debugstr_w(file->TargetPath));
 297
 298             msi_file_update_ui(package, file, szInstallFiles);
 299             rc = copy_install_file(file, source);
 300             if (rc != ERROR_SUCCESS)
 301             {
 302                 ERR("Failed to copy %s to %s (%d)\n", debugstr_w(source),
 303                     debugstr_w(file->TargetPath), rc);
 304                 rc = ERROR_INSTALL_FAILURE;
 305                 msi_free(source);
 306                 break;
 307             }
 308
 309             msi_free(source);
 310         }
 311         else if (file->state != msifs_installed)
 312         {
 313             ERR("compressed file wasn't extracted (%s)\n",
 314                 debugstr_w(file->TargetPath));
 315             rc = ERROR_INSTALL_FAILURE;
 316             break;
 317         }
 318     }
 319
 320     msi_free_media_info(mi);
 321     return rc;
 322 }
 323
 324 static UINT ITERATE_DuplicateFiles(MSIRECORD *row, LPVOID param)
 325 {
 326     MSIPACKAGE *package = (MSIPACKAGE*)param;
 327     WCHAR dest_name[0x100];
 328     LPWSTR dest_path, dest;
 329     LPCWSTR file_key, component;
 330     DWORD sz;
 331     DWORD rc;
 332     MSICOMPONENT *comp;
 333     MSIFILE *file;
 334
 335     component = MSI_RecordGetString(row,2);
 336     comp = get_loaded_component(package,component);
 337
 338     if (!ACTION_VerifyComponentForAction( comp, INSTALLSTATE_LOCAL ))
 339     {
 340         TRACE("Skipping copy due to disabled component %s\n",
 341                         debugstr_w(component));
 342
 343         /* the action taken was the same as the current install state */
 344         comp->Action = comp->Installed;
 345
 346         return ERROR_SUCCESS;
 347     }
 348
 349     comp->Action = INSTALLSTATE_LOCAL;
 350
 351     file_key = MSI_RecordGetString(row,3);
 352     if (!file_key)
 353     {
 354         ERR("Unable to get file key\n");
 355         return ERROR_FUNCTION_FAILED;
 356     }
 357
 358     rc = get_file_target(package,file_key,&file);
 359
 360     if (rc != ERROR_SUCCESS)
 361     {
 362         ERR("Original file unknown %s\n",debugstr_w(file_key));
 363         return ERROR_SUCCESS;
 364     }
 365
 366     if (MSI_RecordIsNull(row,4))
 367         strcpyW(dest_name,strrchrW(file->TargetPath,'\\')+1);
 368     else
 369     {
 370         sz=0x100;
 371         MSI_RecordGetStringW(row,4,dest_name,&sz);
 372         reduce_to_longfilename(dest_name);
 373     }
 374
 375     if (MSI_RecordIsNull(row,5))
 376     {
 377         LPWSTR p;
 378         dest_path = strdupW(file->TargetPath);
 379         p = strrchrW(dest_path,'\\');
 380         if (p)
 381             *p=0;
 382     }
 383     else
 384     {
 385         LPCWSTR destkey;
 386         destkey = MSI_RecordGetString(row,5);
 387         dest_path = resolve_folder(package, destkey, FALSE, FALSE, TRUE, NULL);
 388         if (!dest_path)
 389         {
 390             /* try a Property */
 391             dest_path = msi_dup_property( package, destkey );
 392             if (!dest_path)
 393             {
 394                 FIXME("Unable to get destination folder, try AppSearch properties\n");
 395                 return ERROR_SUCCESS;
 396             }
 397         }
 398     }
 399
 400     dest = build_directory_name(2, dest_path, dest_name);
 401     create_full_pathW(dest_path);
 402
 403     TRACE("Duplicating file %s to %s\n",debugstr_w(file->TargetPath),
 404                     debugstr_w(dest));
 405
 406     if (strcmpW(file->TargetPath,dest))
 407         rc = !CopyFileW(file->TargetPath,dest,TRUE);
 408     else
 409         rc = ERROR_SUCCESS;
 410
 411     if (rc != ERROR_SUCCESS)
 412         ERR("Failed to copy file %s -> %s, last error %d\n",
 413             debugstr_w(file->TargetPath), debugstr_w(dest_path), GetLastError());
 414
 415     FIXME("We should track these duplicate files as well\n");
 416
 417     msi_free(dest_path);
 418     msi_free(dest);
 419
 420     msi_file_update_ui(package, file, szDuplicateFiles);
 421
 422     return ERROR_SUCCESS;
 423 }
 424
 425 UINT ACTION_DuplicateFiles(MSIPACKAGE *package)
 426 {
 427     UINT rc;
 428     MSIQUERY * view;
 429     static const WCHAR ExecSeqQuery[] =
 430         {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
 431          '`','D','u','p','l','i','c','a','t','e','F','i','l','e','`',0};
 432
 433     rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
 434     if (rc != ERROR_SUCCESS)
 435         return ERROR_SUCCESS;
 436
 437     rc = MSI_IterateRecords(view, NULL, ITERATE_DuplicateFiles, package);
 438     msiobj_release(&view->hdr);
 439
 440     return rc;
 441 }
 442
 443 static BOOL verify_comp_for_removal(MSICOMPONENT *comp, UINT install_mode)
 444 {
 445     INSTALLSTATE request = comp->ActionRequest;
 446
 447     if (request == INSTALLSTATE_UNKNOWN)
 448         return FALSE;
 449
 450     if (install_mode == msidbRemoveFileInstallModeOnInstall &&
 451         (request == INSTALLSTATE_LOCAL || request == INSTALLSTATE_SOURCE))
 452         return TRUE;
 453
 454     if (request == INSTALLSTATE_ABSENT)
 455     {
 456         if (!comp->ComponentId)
 457             return FALSE;
 458
 459         if (install_mode == msidbRemoveFileInstallModeOnRemove)
 460             return TRUE;
 461     }
 462
 463     if (install_mode == msidbRemoveFileInstallModeOnBoth)
 464         return TRUE;
 465
 466     return FALSE;
 467 }
 468
 469 static UINT ITERATE_RemoveFiles(MSIRECORD *row, LPVOID param)
 470 {
 471     MSIPACKAGE *package = (MSIPACKAGE*)param;
 472     MSICOMPONENT *comp;
 473     LPCWSTR component, filename, dirprop;
 474     UINT install_mode;
 475     LPWSTR dir = NULL, path = NULL;
 476     DWORD size;
 477     UINT r;
 478
 479     component = MSI_RecordGetString(row, 2);
 480     filename = MSI_RecordGetString(row, 3);
 481     dirprop = MSI_RecordGetString(row, 4);
 482     install_mode = MSI_RecordGetInteger(row, 5);
 483
 484     comp = get_loaded_component(package, component);
 485     if (!comp)
 486     {
 487         ERR("Invalid component: %s\n", debugstr_w(component));
 488         return ERROR_FUNCTION_FAILED;
 489     }
 490
 491     if (!verify_comp_for_removal(comp, install_mode))
 492     {
 493         TRACE("Skipping removal due to missing conditions\n");
 494         comp->Action = comp->Installed;
 495         return ERROR_SUCCESS;
 496     }
 497
 498     dir = msi_dup_property(package, dirprop);
 499     if (!dir)
 500         return ERROR_OUTOFMEMORY;
 501
 502     size = (filename != NULL) ? lstrlenW(filename) : 0;
 503     size += lstrlenW(dir) + 2;
 504     path = msi_alloc(size * sizeof(WCHAR));
 505     if (!path)
 506     {
 507         r = ERROR_OUTOFMEMORY;
 508         goto done;
 509     }
 510
 511     lstrcpyW(path, dir);
 512     PathAddBackslashW(path);
 513
 514     if (filename)
 515     {
 516         lstrcatW(path, filename);
 517
 518         TRACE("Deleting misc file: %s\n", debugstr_w(path));
 519         DeleteFileW(path);
 520     }
 521     else
 522     {
 523         TRACE("Removing misc directory: %s\n", debugstr_w(path));
 524         RemoveDirectoryW(path);
 525     }
 526
 527 done:
 528     msi_free(path);
 529     msi_free(dir);
 530     return ERROR_SUCCESS;
 531 }
 532
 533 UINT ACTION_RemoveFiles( MSIPACKAGE *package )
 534 {
 535     MSIQUERY *view;
 536     MSIFILE *file;
 537     UINT r;
 538
 539     static const WCHAR query[] = {
 540         'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
 541         '`','R','e','m','o','v','e','F','i','l','e','`',0};
 542
 543     r = MSI_DatabaseOpenViewW(package->db, query, &view);
 544     if (r == ERROR_SUCCESS)
 545     {
 546         MSI_IterateRecords(view, NULL, ITERATE_RemoveFiles, package);
 547         msiobj_release(&view->hdr);
 548     }
 549
 550     LIST_FOR_EACH_ENTRY( file, &package->files, MSIFILE, entry )
 551     {
 552         MSIRECORD *uirow;
 553         LPWSTR uipath, p;
 554
 555         if ( file->state == msifs_installed )
 556             ERR("removing installed file %s\n", debugstr_w(file->TargetPath));
 557
 558         if ( file->Component->ActionRequest != INSTALLSTATE_ABSENT ||
 559              file->Component->Installed == INSTALLSTATE_SOURCE )
 560             continue;
 561
 562         /* don't remove a file if the old file
 563          * is strictly newer than the version to be installed
 564          */
 565         if ( msi_compare_file_version( file ) < 0 )
 566             continue;
 567
 568         TRACE("removing %s\n", debugstr_w(file->File) );
 569         if ( !DeleteFileW( file->TargetPath ) )
 570             TRACE("failed to delete %s\n",  debugstr_w(file->TargetPath));
 571         file->state = msifs_missing;
 572
 573         /* the UI chunk */
 574         uirow = MSI_CreateRecord( 9 );
 575         MSI_RecordSetStringW( uirow, 1, file->FileName );
 576         uipath = strdupW( file->TargetPath );
 577         p = strrchrW(uipath,'\\');
 578         if (p)
 579             p[1]=0;
 580         MSI_RecordSetStringW( uirow, 9, uipath);
 581         ui_actiondata( package, szRemoveFiles, uirow);
 582         msiobj_release( &uirow->hdr );
 583         msi_free( uipath );
 584         /* FIXME: call ui_progress here? */
 585     }
 586
 587     return ERROR_SUCCESS;
 588 }