dlls/fusion/asmcache.c

   1 /*
   2  * IAssemblyCache implementation
   3  *
   4  * Copyright 2008 James Hawkins
   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 #include <stdarg.h>
  22 #include <stdio.h>
  23
  24 #define COBJMACROS
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winuser.h"
  29 #include "winver.h"
  30 #include "wincrypt.h"
  31 #include "winreg.h"
  32 #include "shlwapi.h"
  33 #include "dbghelp.h"
  34 #include "ole2.h"
  35 #include "fusion.h"
  36 #include "corerror.h"
  37
  38 #include "fusionpriv.h"
  39 #include "wine/debug.h"
  40 #include "wine/unicode.h"
  41
  42 WINE_DEFAULT_DEBUG_CHANNEL(fusion);
  43
  44 static const WCHAR cache_mutex_nameW[] =
  45     {'_','_','W','I','N','E','_','F','U','S','I','O','N','_','C','A','C','H','E','_','M','U','T','E','X','_','_',0};
  46
  47 static BOOL create_full_path(LPCWSTR path)
  48 {
  49     LPWSTR new_path;
  50     BOOL ret = TRUE;
  51     int len;
  52
  53     new_path = HeapAlloc(GetProcessHeap(), 0, (strlenW(path) + 1) * sizeof(WCHAR));
  54     if (!new_path)
  55         return FALSE;
  56
  57     strcpyW(new_path, path);
  58
  59     while ((len = strlenW(new_path)) && new_path[len - 1] == '\\')
  60         new_path[len - 1] = 0;
  61
  62     while (!CreateDirectoryW(new_path, NULL))
  63     {
  64         LPWSTR slash;
  65         DWORD last_error = GetLastError();
  66
  67         if(last_error == ERROR_ALREADY_EXISTS)
  68             break;
  69
  70         if(last_error != ERROR_PATH_NOT_FOUND)
  71         {
  72             ret = FALSE;
  73             break;
  74         }
  75
  76         if(!(slash = strrchrW(new_path, '\\')))
  77         {
  78             ret = FALSE;
  79             break;
  80         }
  81
  82         len = slash - new_path;
  83         new_path[len] = 0;
  84         if(!create_full_path(new_path))
  85         {
  86             ret = FALSE;
  87             break;
  88         }
  89
  90         new_path[len] = '\\';
  91     }
  92
  93     HeapFree(GetProcessHeap(), 0, new_path);
  94     return ret;
  95 }
  96
  97 static BOOL get_assembly_directory(LPWSTR dir, DWORD size, const char *version, PEKIND architecture)
  98 {
  99     static const WCHAR dotnet[] = {'\\','M','i','c','r','o','s','o','f','t','.','N','E','T','\\',0};
 100     static const WCHAR gac[] = {'\\','a','s','s','e','m','b','l','y','\\','G','A','C',0};
 101     static const WCHAR msil[] = {'_','M','S','I','L',0};
 102     static const WCHAR x86[] = {'_','3','2',0};
 103     static const WCHAR amd64[] = {'_','6','4',0};
 104     DWORD len = GetWindowsDirectoryW(dir, size);
 105
 106     if (!strcmp(version, "v4.0.30319"))
 107     {
 108         strcpyW(dir + len, dotnet);
 109         len += sizeof(dotnet)/sizeof(WCHAR) -1;
 110         strcpyW(dir + len, gac + 1);
 111         len += sizeof(gac)/sizeof(WCHAR) - 2;
 112     }
 113     else
 114     {
 115         strcpyW(dir + len, gac);
 116         len += sizeof(gac)/sizeof(WCHAR) - 1;
 117     }
 118     switch (architecture)
 119     {
 120         case peNone:
 121             break;
 122
 123         case peMSIL:
 124             strcpyW(dir + len, msil);
 125             break;
 126
 127         case peI386:
 128             strcpyW(dir + len, x86);
 129             break;
 130
 131         case peAMD64:
 132             strcpyW(dir + len, amd64);
 133             break;
 134
 135         default:
 136             WARN("unhandled architecture %u\n", architecture);
 137             return FALSE;
 138     }
 139     return TRUE;
 140 }
 141
 142 /* IAssemblyCache */
 143
 144 typedef struct {
 145     IAssemblyCache IAssemblyCache_iface;
 146
 147     LONG ref;
 148     HANDLE lock;
 149 } IAssemblyCacheImpl;
 150
 151 static inline IAssemblyCacheImpl *impl_from_IAssemblyCache(IAssemblyCache *iface)
 152 {
 153     return CONTAINING_RECORD(iface, IAssemblyCacheImpl, IAssemblyCache_iface);
 154 }
 155
 156 static HRESULT WINAPI IAssemblyCacheImpl_QueryInterface(IAssemblyCache *iface,
 157                                                         REFIID riid, LPVOID *ppobj)
 158 {
 159     IAssemblyCacheImpl *This = impl_from_IAssemblyCache(iface);
 160
 161     TRACE("(%p, %s, %p)\n", This, debugstr_guid(riid), ppobj);
 162
 163     *ppobj = NULL;
 164
 165     if (IsEqualIID(riid, &IID_IUnknown) ||
 166         IsEqualIID(riid, &IID_IAssemblyCache))
 167     {
 168         IUnknown_AddRef(iface);
 169         *ppobj = This;
 170         return S_OK;
 171     }
 172
 173     WARN("(%p, %s, %p): not found\n", This, debugstr_guid(riid), ppobj);
 174     return E_NOINTERFACE;
 175 }
 176
 177 static ULONG WINAPI IAssemblyCacheImpl_AddRef(IAssemblyCache *iface)
 178 {
 179     IAssemblyCacheImpl *This = impl_from_IAssemblyCache(iface);
 180     ULONG refCount = InterlockedIncrement(&This->ref);
 181
 182     TRACE("(%p)->(ref before = %u)\n", This, refCount - 1);
 183
 184     return refCount;
 185 }
 186
 187 static ULONG WINAPI IAssemblyCacheImpl_Release(IAssemblyCache *iface)
 188 {
 189     IAssemblyCacheImpl *cache = impl_from_IAssemblyCache(iface);
 190     ULONG refCount = InterlockedDecrement( &cache->ref );
 191
 192     TRACE("(%p)->(ref before = %u)\n", cache, refCount + 1);
 193
 194     if (!refCount)
 195     {
 196         CloseHandle( cache->lock );
 197         HeapFree( GetProcessHeap(), 0, cache );
 198     }
 199     return refCount;
 200 }
 201
 202 static void cache_lock( IAssemblyCacheImpl *cache )
 203 {
 204     WaitForSingleObject( cache->lock, INFINITE );
 205 }
 206
 207 static void cache_unlock( IAssemblyCacheImpl *cache )
 208 {
 209     ReleaseMutex( cache->lock );
 210 }
 211
 212 static HRESULT WINAPI IAssemblyCacheImpl_UninstallAssembly(IAssemblyCache *iface,
 213                                                            DWORD dwFlags,
 214                                                            LPCWSTR pszAssemblyName,
 215                                                            LPCFUSION_INSTALL_REFERENCE pRefData,
 216                                                            ULONG *pulDisposition)
 217 {
 218     HRESULT hr;
 219     IAssemblyCacheImpl *cache = impl_from_IAssemblyCache(iface);
 220     IAssemblyName *asmname, *next = NULL;
 221     IAssemblyEnum *asmenum = NULL;
 222     WCHAR *p, *path = NULL;
 223     ULONG disp;
 224     DWORD len;
 225
 226     TRACE("(%p, 0%08x, %s, %p, %p)\n", iface, dwFlags,
 227           debugstr_w(pszAssemblyName), pRefData, pulDisposition);
 228
 229     if (pRefData)
 230     {
 231         FIXME("application reference not supported\n");
 232         return E_NOTIMPL;
 233     }
 234     hr = CreateAssemblyNameObject( &asmname, pszAssemblyName, CANOF_PARSE_DISPLAY_NAME, NULL );
 235     if (FAILED( hr ))
 236         return hr;
 237
 238     cache_lock( cache );
 239
 240     hr = CreateAssemblyEnum( &asmenum, NULL, asmname, ASM_CACHE_GAC, NULL );
 241     if (FAILED( hr ))
 242         goto done;
 243
 244     hr = IAssemblyEnum_GetNextAssembly( asmenum, NULL, &next, 0 );
 245     if (hr == S_FALSE)
 246     {
 247         if (pulDisposition)
 248             *pulDisposition = IASSEMBLYCACHE_UNINSTALL_DISPOSITION_ALREADY_UNINSTALLED;
 249         goto done;
 250     }
 251     hr = IAssemblyName_GetPath( next, NULL, &len );
 252     if (hr != HRESULT_FROM_WIN32( ERROR_INSUFFICIENT_BUFFER ))
 253         goto done;
 254
 255     if (!(path = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) )))
 256     {
 257         hr = E_OUTOFMEMORY;
 258         goto done;
 259     }
 260     hr = IAssemblyName_GetPath( next, path, &len );
 261     if (FAILED( hr ))
 262         goto done;
 263
 264     if (DeleteFileW( path ))
 265     {
 266         if ((p = strrchrW( path, '\\' )))
 267         {
 268             *p = 0;
 269             RemoveDirectoryW( path );
 270             if ((p = strrchrW( path, '\\' )))
 271             {
 272                 *p = 0;
 273                 RemoveDirectoryW( path );
 274             }
 275         }
 276         disp = IASSEMBLYCACHE_UNINSTALL_DISPOSITION_UNINSTALLED;
 277         hr = S_OK;
 278     }
 279     else
 280     {
 281         disp = IASSEMBLYCACHE_UNINSTALL_DISPOSITION_ALREADY_UNINSTALLED;
 282         hr = S_FALSE;
 283     }
 284     if (pulDisposition) *pulDisposition = disp;
 285
 286 done:
 287     IAssemblyName_Release( asmname );
 288     if (next) IAssemblyName_Release( next );
 289     if (asmenum) IAssemblyEnum_Release( asmenum );
 290     HeapFree( GetProcessHeap(), 0, path );
 291     cache_unlock( cache );
 292     return hr;
 293 }
 294
 295 static HRESULT WINAPI IAssemblyCacheImpl_QueryAssemblyInfo(IAssemblyCache *iface,
 296                                                            DWORD dwFlags,
 297                                                            LPCWSTR pszAssemblyName,
 298                                                            ASSEMBLY_INFO *pAsmInfo)
 299 {
 300     IAssemblyCacheImpl *cache = impl_from_IAssemblyCache(iface);
 301     IAssemblyName *asmname, *next = NULL;
 302     IAssemblyEnum *asmenum = NULL;
 303     HRESULT hr;
 304
 305     TRACE("(%p, %d, %s, %p)\n", iface, dwFlags,
 306           debugstr_w(pszAssemblyName), pAsmInfo);
 307
 308     if (pAsmInfo)
 309     {
 310         if (pAsmInfo->cbAssemblyInfo == 0)
 311             pAsmInfo->cbAssemblyInfo = sizeof(ASSEMBLY_INFO);
 312         else if (pAsmInfo->cbAssemblyInfo != sizeof(ASSEMBLY_INFO))
 313             return E_INVALIDARG;
 314     }
 315
 316     hr = CreateAssemblyNameObject(&asmname, pszAssemblyName,
 317                                   CANOF_PARSE_DISPLAY_NAME, NULL);
 318     if (FAILED(hr))
 319         return hr;
 320
 321     cache_lock( cache );
 322
 323     hr = CreateAssemblyEnum(&asmenum, NULL, asmname, ASM_CACHE_GAC, NULL);
 324     if (FAILED(hr))
 325         goto done;
 326
 327     for (;;)
 328     {
 329         hr = IAssemblyEnum_GetNextAssembly(asmenum, NULL, &next, 0);
 330         if (hr != S_OK)
 331         {
 332             hr = HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
 333             goto done;
 334         }
 335         hr = IAssemblyName_IsEqual(asmname, next, ASM_CMPF_IL_ALL);
 336         if (hr == S_OK) break;
 337     }
 338
 339     if (!pAsmInfo)
 340         goto done;
 341
 342     hr = IAssemblyName_GetPath(next, pAsmInfo->pszCurrentAssemblyPathBuf, &pAsmInfo->cchBuf);
 343
 344     pAsmInfo->dwAssemblyFlags = ASSEMBLYINFO_FLAG_INSTALLED;
 345
 346 done:
 347     IAssemblyName_Release(asmname);
 348     if (next) IAssemblyName_Release(next);
 349     if (asmenum) IAssemblyEnum_Release(asmenum);
 350     cache_unlock( cache );
 351     return hr;
 352 }
 353
 354 static HRESULT WINAPI IAssemblyCacheImpl_CreateAssemblyCacheItem(IAssemblyCache *iface,
 355                                                                  DWORD dwFlags,
 356                                                                  PVOID pvReserved,
 357                                                                  IAssemblyCacheItem **ppAsmItem,
 358                                                                  LPCWSTR pszAssemblyName)
 359 {
 360     FIXME("(%p, %d, %p, %p, %s) stub!\n", iface, dwFlags, pvReserved,
 361           ppAsmItem, debugstr_w(pszAssemblyName));
 362
 363     return E_NOTIMPL;
 364 }
 365
 366 static HRESULT WINAPI IAssemblyCacheImpl_CreateAssemblyScavenger(IAssemblyCache *iface,
 367                                                                  IUnknown **ppUnkReserved)
 368 {
 369     FIXME("(%p, %p) stub!\n", iface, ppUnkReserved);
 370     return E_NOTIMPL;
 371 }
 372
 373 static HRESULT WINAPI IAssemblyCacheImpl_InstallAssembly(IAssemblyCache *iface,
 374                                                          DWORD dwFlags,
 375                                                          LPCWSTR pszManifestFilePath,
 376                                                          LPCFUSION_INSTALL_REFERENCE pRefData)
 377 {
 378     static const WCHAR format[] =
 379         {'%','s','\\','%','s','\\','%','s','_','_','%','s','\\',0};
 380     static const WCHAR format_v40[] =
 381         {'%','s','\\','%','s','\\','v','4','.','0','_','%','s','_','_','%','s','\\',0};
 382     static const WCHAR ext_exe[] = {'.','e','x','e',0};
 383     static const WCHAR ext_dll[] = {'.','d','l','l',0};
 384     IAssemblyCacheImpl *cache = impl_from_IAssemblyCache(iface);
 385     ASSEMBLY *assembly;
 386     WCHAR *filename, *ext;
 387     WCHAR *name = NULL, *token = NULL, *version = NULL, *asmpath = NULL;
 388     WCHAR path[MAX_PATH], asmdir[MAX_PATH];
 389     PEKIND architecture;
 390     char *clr_version;
 391     HRESULT hr;
 392
 393     TRACE("(%p, %d, %s, %p)\n", iface, dwFlags,
 394           debugstr_w(pszManifestFilePath), pRefData);
 395
 396     if (!pszManifestFilePath || !*pszManifestFilePath)
 397         return E_INVALIDARG;
 398
 399     if (!(ext = strrchrW(pszManifestFilePath, '.')))
 400         return HRESULT_FROM_WIN32(ERROR_INVALID_NAME);
 401
 402     if (lstrcmpiW(ext, ext_exe) && lstrcmpiW(ext, ext_dll))
 403         return HRESULT_FROM_WIN32(ERROR_INVALID_NAME);
 404
 405     if (GetFileAttributesW(pszManifestFilePath) == INVALID_FILE_ATTRIBUTES)
 406         return HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
 407
 408     hr = assembly_create(&assembly, pszManifestFilePath);
 409     if (FAILED(hr))
 410     {
 411         hr = COR_E_ASSEMBLYEXPECTED;
 412         goto done;
 413     }
 414
 415     hr = assembly_get_name(assembly, &name);
 416     if (FAILED(hr))
 417         goto done;
 418
 419     hr = assembly_get_pubkey_token(assembly, &token);
 420     if (FAILED(hr))
 421         goto done;
 422
 423     hr = assembly_get_version(assembly, &version);
 424     if (FAILED(hr))
 425         goto done;
 426
 427     hr = assembly_get_runtime_version(assembly, &clr_version);
 428     if (FAILED(hr))
 429         goto done;
 430
 431     cache_lock( cache );
 432
 433     architecture = assembly_get_architecture(assembly);
 434     get_assembly_directory(asmdir, MAX_PATH, clr_version, architecture);
 435
 436     if (!strcmp(clr_version, "v4.0.30319"))
 437         sprintfW(path, format_v40, asmdir, name, version, token);
 438     else
 439         sprintfW(path, format, asmdir, name, version, token);
 440
 441     create_full_path(path);
 442
 443     hr = assembly_get_path(assembly, &asmpath);
 444     if (FAILED(hr))
 445         goto done;
 446
 447     filename = PathFindFileNameW(asmpath);
 448
 449     strcatW(path, filename);
 450     if (!CopyFileW(asmpath, path, FALSE))
 451         hr = HRESULT_FROM_WIN32(GetLastError());
 452
 453 done:
 454     HeapFree(GetProcessHeap(), 0, name);
 455     HeapFree(GetProcessHeap(), 0, token);
 456     HeapFree(GetProcessHeap(), 0, version);
 457     HeapFree(GetProcessHeap(), 0, asmpath);
 458     assembly_release(assembly);
 459     cache_unlock( cache );
 460     return hr;
 461 }
 462
 463 static const IAssemblyCacheVtbl AssemblyCacheVtbl = {
 464     IAssemblyCacheImpl_QueryInterface,
 465     IAssemblyCacheImpl_AddRef,
 466     IAssemblyCacheImpl_Release,
 467     IAssemblyCacheImpl_UninstallAssembly,
 468     IAssemblyCacheImpl_QueryAssemblyInfo,
 469     IAssemblyCacheImpl_CreateAssemblyCacheItem,
 470     IAssemblyCacheImpl_CreateAssemblyScavenger,
 471     IAssemblyCacheImpl_InstallAssembly
 472 };
 473
 474 /******************************************************************
 475  *  CreateAssemblyCache   (FUSION.@)
 476  */
 477 HRESULT WINAPI CreateAssemblyCache(IAssemblyCache **ppAsmCache, DWORD dwReserved)
 478 {
 479     IAssemblyCacheImpl *cache;
 480
 481     TRACE("(%p, %d)\n", ppAsmCache, dwReserved);
 482
 483     if (!ppAsmCache)
 484         return E_INVALIDARG;
 485
 486     *ppAsmCache = NULL;
 487
 488     cache = HeapAlloc(GetProcessHeap(), 0, sizeof(IAssemblyCacheImpl));
 489     if (!cache)
 490         return E_OUTOFMEMORY;
 491
 492     cache->IAssemblyCache_iface.lpVtbl = &AssemblyCacheVtbl;
 493     cache->ref = 1;
 494     cache->lock = CreateMutexW( NULL, FALSE, cache_mutex_nameW );
 495     if (!cache->lock)
 496     {
 497         HeapFree( GetProcessHeap(), 0, cache );
 498         return HRESULT_FROM_WIN32( GetLastError() );
 499     }
 500     *ppAsmCache = &cache->IAssemblyCache_iface;
 501     return S_OK;
 502 }
 503
 504 /* IAssemblyCacheItem */
 505
 506 typedef struct {
 507     IAssemblyCacheItem IAssemblyCacheItem_iface;
 508
 509     LONG ref;
 510 } IAssemblyCacheItemImpl;
 511
 512 static inline IAssemblyCacheItemImpl *impl_from_IAssemblyCacheItem(IAssemblyCacheItem *iface)
 513 {
 514     return CONTAINING_RECORD(iface, IAssemblyCacheItemImpl, IAssemblyCacheItem_iface);
 515 }
 516
 517 static HRESULT WINAPI IAssemblyCacheItemImpl_QueryInterface(IAssemblyCacheItem *iface,
 518                                                             REFIID riid, LPVOID *ppobj)
 519 {
 520     IAssemblyCacheItemImpl *This = impl_from_IAssemblyCacheItem(iface);
 521
 522     TRACE("(%p, %s, %p)\n", This, debugstr_guid(riid), ppobj);
 523
 524     *ppobj = NULL;
 525
 526     if (IsEqualIID(riid, &IID_IUnknown) ||
 527         IsEqualIID(riid, &IID_IAssemblyCacheItem))
 528     {
 529         IUnknown_AddRef(iface);
 530         *ppobj = This;
 531         return S_OK;
 532     }
 533
 534     WARN("(%p, %s, %p): not found\n", This, debugstr_guid(riid), ppobj);
 535     return E_NOINTERFACE;
 536 }
 537
 538 static ULONG WINAPI IAssemblyCacheItemImpl_AddRef(IAssemblyCacheItem *iface)
 539 {
 540     IAssemblyCacheItemImpl *This = impl_from_IAssemblyCacheItem(iface);
 541     ULONG refCount = InterlockedIncrement(&This->ref);
 542
 543     TRACE("(%p)->(ref before = %u)\n", This, refCount - 1);
 544
 545     return refCount;
 546 }
 547
 548 static ULONG WINAPI IAssemblyCacheItemImpl_Release(IAssemblyCacheItem *iface)
 549 {
 550     IAssemblyCacheItemImpl *This = impl_from_IAssemblyCacheItem(iface);
 551     ULONG refCount = InterlockedDecrement(&This->ref);
 552
 553     TRACE("(%p)->(ref before = %u)\n", This, refCount + 1);
 554
 555     if (!refCount)
 556         HeapFree(GetProcessHeap(), 0, This);
 557
 558     return refCount;
 559 }
 560
 561 static HRESULT WINAPI IAssemblyCacheItemImpl_CreateStream(IAssemblyCacheItem *iface,
 562                                                         DWORD dwFlags,
 563                                                         LPCWSTR pszStreamName,
 564                                                         DWORD dwFormat,
 565                                                         DWORD dwFormatFlags,
 566                                                         IStream **ppIStream,
 567                                                         ULARGE_INTEGER *puliMaxSize)
 568 {
 569     FIXME("(%p, %d, %s, %d, %d, %p, %p) stub!\n", iface, dwFlags,
 570           debugstr_w(pszStreamName), dwFormat, dwFormatFlags, ppIStream, puliMaxSize);
 571
 572     return E_NOTIMPL;
 573 }
 574
 575 static HRESULT WINAPI IAssemblyCacheItemImpl_Commit(IAssemblyCacheItem *iface,
 576                                                   DWORD dwFlags,
 577                                                   ULONG *pulDisposition)
 578 {
 579     FIXME("(%p, %d, %p) stub!\n", iface, dwFlags, pulDisposition);
 580     return E_NOTIMPL;
 581 }
 582
 583 static HRESULT WINAPI IAssemblyCacheItemImpl_AbortItem(IAssemblyCacheItem *iface)
 584 {
 585     FIXME("(%p) stub!\n", iface);
 586     return E_NOTIMPL;
 587 }
 588
 589 static const IAssemblyCacheItemVtbl AssemblyCacheItemVtbl = {
 590     IAssemblyCacheItemImpl_QueryInterface,
 591     IAssemblyCacheItemImpl_AddRef,
 592     IAssemblyCacheItemImpl_Release,
 593     IAssemblyCacheItemImpl_CreateStream,
 594     IAssemblyCacheItemImpl_Commit,
 595     IAssemblyCacheItemImpl_AbortItem
 596 };