dlls/sxs/cache.c

   1 /*
   2  * IAssemblyCache implementation
   3  *
   4  * Copyright 2010 Hans Leidekker 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 #include <stdarg.h>
  22
  23 #define COBJMACROS
  24 #define INITGUID
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "ole2.h"
  29 #include "winsxs.h"
  30 #include "msxml2.h"
  31
  32 #include "wine/debug.h"
  33 #include "wine/list.h"
  34 #include "wine/unicode.h"
  35 #include "sxs_private.h"
  36
  37 WINE_DEFAULT_DEBUG_CHANNEL(sxs);
  38
  39 static const WCHAR cache_mutex_nameW[] =
  40     {'_','_','W','I','N','E','_','S','X','S','_','C','A','C','H','E','_','M','U','T','E','X','_','_',0};
  41
  42 static const WCHAR win32W[] = {'w','i','n','3','2',0};
  43 static const WCHAR win32_policyW[] = {'w','i','n','3','2','-','p','o','l','i','c','y',0};
  44 static const WCHAR backslashW[] = {'\\',0};
  45
  46 struct cache
  47 {
  48     IAssemblyCache IAssemblyCache_iface;
  49     LONG refs;
  50     HANDLE lock;
  51 };
  52
  53 static inline struct cache *impl_from_IAssemblyCache(IAssemblyCache *iface)
  54 {
  55     return CONTAINING_RECORD(iface, struct cache, IAssemblyCache_iface);
  56 }
  57
  58 static HRESULT WINAPI cache_QueryInterface(
  59     IAssemblyCache *iface,
  60     REFIID riid,
  61     void **obj )
  62 {
  63     struct cache *cache = impl_from_IAssemblyCache(iface);
  64
  65     TRACE("%p, %s, %p\n", cache, debugstr_guid(riid), obj);
  66
  67     *obj = NULL;
  68
  69     if (IsEqualIID(riid, &IID_IUnknown) ||
  70         IsEqualIID(riid, &IID_IAssemblyCache))
  71     {
  72         IAssemblyCache_AddRef( iface );
  73         *obj = cache;
  74         return S_OK;
  75     }
  76
  77     return E_NOINTERFACE;
  78 }
  79
  80 static ULONG WINAPI cache_AddRef( IAssemblyCache *iface )
  81 {
  82     struct cache *cache = impl_from_IAssemblyCache(iface);
  83     return InterlockedIncrement( &cache->refs );
  84 }
  85
  86 static ULONG WINAPI cache_Release( IAssemblyCache *iface )
  87 {
  88     struct cache *cache = impl_from_IAssemblyCache(iface);
  89     ULONG refs = InterlockedDecrement( &cache->refs );
  90
  91     if (!refs)
  92     {
  93         TRACE("destroying %p\n", cache);
  94         CloseHandle( cache->lock );
  95         HeapFree( GetProcessHeap(), 0, cache );
  96     }
  97     return refs;
  98 }
  99
 100 static unsigned int build_sxs_path( WCHAR *path )
 101 {
 102     static const WCHAR winsxsW[] = {'\\','w','i','n','s','x','s','\\',0};
 103     unsigned int len = GetWindowsDirectoryW( path, MAX_PATH );
 104
 105     memcpy( path + len, winsxsW, sizeof(winsxsW) );
 106     return len + sizeof(winsxsW) / sizeof(winsxsW[0]) - 1;
 107 }
 108
 109 static WCHAR *build_assembly_name( const WCHAR *arch, const WCHAR *name, const WCHAR *token,
 110                                    const WCHAR *version, unsigned int *len )
 111 {
 112     static const WCHAR fmtW[] =
 113         {'%','s','_','%','s','_','%','s','_','%','s','_','n','o','n','e','_','d','e','a','d','b','e','e','f',0};
 114     unsigned int buflen = sizeof(fmtW) / sizeof(fmtW[0]);
 115     WCHAR *ret, *p;
 116
 117     buflen += strlenW( arch );
 118     buflen += strlenW( name );
 119     buflen += strlenW( token );
 120     buflen += strlenW( version );
 121     if (!(ret = HeapAlloc( GetProcessHeap(), 0, buflen * sizeof(WCHAR) ))) return NULL;
 122     *len = sprintfW( ret, fmtW, arch, name, token, version );
 123     for (p = ret; *p; p++) *p = tolowerW( *p );
 124     return ret;
 125 }
 126
 127 static WCHAR *build_manifest_path( const WCHAR *arch, const WCHAR *name, const WCHAR *token,
 128                                    const WCHAR *version )
 129 {
 130     static const WCHAR fmtW[] =
 131         {'%','s','m','a','n','i','f','e','s','t','s','\\','%','s','.','m','a','n','i','f','e','s','t',0};
 132     WCHAR *path = NULL, *ret, sxsdir[MAX_PATH];
 133     unsigned int len;
 134
 135     if (!(path = build_assembly_name( arch, name, token, version, &len ))) return NULL;
 136     len += sizeof(fmtW) / sizeof(fmtW[0]);
 137     len += build_sxs_path( sxsdir );
 138     if (!(ret = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) )))
 139     {
 140         HeapFree( GetProcessHeap(), 0, path );
 141         return NULL;
 142     }
 143     sprintfW( ret, fmtW, sxsdir, path );
 144     HeapFree( GetProcessHeap(), 0, path );
 145     return ret;
 146 }
 147
 148 static WCHAR *build_policy_name( const WCHAR *arch, const WCHAR *name, const WCHAR *token,
 149                                  unsigned int *len )
 150 {
 151     static const WCHAR fmtW[] =
 152         {'%','s','_','%','s','_','%','s','_','n','o','n','e','_','d','e','a','d','b','e','e','f',0};
 153     unsigned int buflen = sizeof(fmtW) / sizeof(fmtW[0]);
 154     WCHAR *ret, *p;
 155
 156     buflen += strlenW( arch );
 157     buflen += strlenW( name );
 158     buflen += strlenW( token );
 159     if (!(ret = HeapAlloc( GetProcessHeap(), 0, buflen * sizeof(WCHAR) ))) return NULL;
 160     *len = sprintfW( ret, fmtW, arch, name, token );
 161     for (p = ret; *p; p++) *p = tolowerW( *p );
 162     return ret;
 163 }
 164
 165 static WCHAR *build_policy_path( const WCHAR *arch, const WCHAR *name, const WCHAR *token,
 166                                  const WCHAR *version )
 167 {
 168     static const WCHAR fmtW[] =
 169         {'%','s','p','o','l','i','c','i','e','s','\\','%','s','\\','%','s','.','p','o','l','i','c','y',0};
 170     WCHAR *path = NULL, *ret, sxsdir[MAX_PATH];
 171     unsigned int len;
 172
 173     if (!(path = build_policy_name( arch, name, token, &len ))) return NULL;
 174     len += sizeof(fmtW) / sizeof(fmtW[0]);
 175     len += build_sxs_path( sxsdir );
 176     len += strlenW( version );
 177     if (!(ret = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) )))
 178     {
 179         HeapFree( GetProcessHeap(), 0, path );
 180         return NULL;
 181     }
 182     sprintfW( ret, fmtW, sxsdir, path, version );
 183     HeapFree( GetProcessHeap(), 0, path );
 184     return ret;
 185 }
 186
 187 static void cache_lock( struct cache *cache )
 188 {
 189     WaitForSingleObject( cache->lock, INFINITE );
 190 }
 191
 192 static void cache_unlock( struct cache *cache )
 193 {
 194     ReleaseMutex( cache->lock );
 195 }
 196
 197 #define ASSEMBLYINFO_FLAG_INSTALLED 1
 198
 199 static HRESULT WINAPI cache_QueryAssemblyInfo(
 200     IAssemblyCache *iface,
 201     DWORD flags,
 202     LPCWSTR assembly_name,
 203     ASSEMBLY_INFO *info )
 204 {
 205     struct cache *cache = impl_from_IAssemblyCache( iface );
 206     IAssemblyName *name_obj;
 207     const WCHAR *arch, *name, *token, *type, *version;
 208     WCHAR *p, *path = NULL;
 209     unsigned int len;
 210     HRESULT hr;
 211
 212     TRACE("%p, 0x%08x, %s, %p\n", iface, flags, debugstr_w(assembly_name), info);
 213
 214     if (flags || (info && info->cbAssemblyInfo != sizeof(*info)))
 215         return E_INVALIDARG;
 216
 217     hr = CreateAssemblyNameObject( &name_obj, assembly_name, CANOF_PARSE_DISPLAY_NAME, 0 );
 218     if (FAILED( hr ))
 219         return hr;
 220
 221     arch = get_name_attribute( name_obj, NAME_ATTR_ID_ARCH );
 222     name = get_name_attribute( name_obj, NAME_ATTR_ID_NAME );
 223     token = get_name_attribute( name_obj, NAME_ATTR_ID_TOKEN );
 224     type = get_name_attribute( name_obj, NAME_ATTR_ID_TYPE );
 225     version = get_name_attribute( name_obj, NAME_ATTR_ID_VERSION );
 226     if (!arch || !name || !token || !type || !version)
 227     {
 228         IAssemblyName_Release( name_obj );
 229         return HRESULT_FROM_WIN32( ERROR_SXS_MISSING_ASSEMBLY_IDENTITY_ATTRIBUTE );
 230     }
 231     if (!info)
 232     {
 233         IAssemblyName_Release( name_obj );
 234         return S_OK;
 235     }
 236     cache_lock( cache );
 237
 238     if (!strcmpW( type, win32W )) path = build_manifest_path( arch, name, token, version );
 239     else if (!strcmpW( type, win32_policyW )) path = build_policy_path( arch, name, token, version );
 240     else
 241     {
 242         hr = HRESULT_FROM_WIN32( ERROR_SXS_INVALID_IDENTITY_ATTRIBUTE_VALUE );
 243         goto done;
 244     }
 245     if (!path)
 246     {
 247         hr = E_OUTOFMEMORY;
 248         goto done;
 249     }
 250     hr = S_OK;
 251     if (GetFileAttributesW( path ) != INVALID_FILE_ATTRIBUTES) /* FIXME: better check */
 252     {
 253         info->dwAssemblyFlags = ASSEMBLYINFO_FLAG_INSTALLED;
 254         TRACE("assembly is installed\n");
 255     }
 256     if ((p = strrchrW( path, '\\' ))) *p = 0;
 257     len = strlenW( path ) + 1;
 258     if (info->pszCurrentAssemblyPathBuf)
 259     {
 260         if (info->cchBuf < len)
 261         {
 262             info->cchBuf = len;
 263             hr = HRESULT_FROM_WIN32( ERROR_INSUFFICIENT_BUFFER );
 264         }
 265         else strcpyW( info->pszCurrentAssemblyPathBuf, path );
 266     }
 267
 268 done:
 269     HeapFree( GetProcessHeap(), 0, path );
 270     IAssemblyName_Release( name_obj );
 271     cache_unlock( cache );
 272     return hr;
 273 }
 274
 275 static HRESULT WINAPI cache_CreateAssemblyCacheItem(
 276     IAssemblyCache *iface,
 277     DWORD flags,
 278     PVOID reserved,
 279     IAssemblyCacheItem **item,
 280     LPCWSTR name )
 281 {
 282     FIXME("%p, 0x%08x, %p, %p, %s\n", iface, flags, reserved, item, debugstr_w(name));
 283     return E_NOTIMPL;
 284 }
 285
 286 static HRESULT WINAPI cache_Reserved(
 287     IAssemblyCache *iface,
 288     IUnknown **reserved)
 289 {
 290     FIXME("%p\n", reserved);
 291     return E_NOTIMPL;
 292 }
 293
 294 static BSTR get_attribute_value( IXMLDOMNamedNodeMap *map, const WCHAR *value_name )
 295 {
 296     HRESULT hr;
 297     IXMLDOMNode *attr;
 298     VARIANT var;
 299     BSTR str;
 300
 301     str = SysAllocString( value_name );
 302     hr = IXMLDOMNamedNodeMap_getNamedItem( map, str, &attr );
 303     SysFreeString( str );
 304     if (hr != S_OK) return NULL;
 305
 306     hr = IXMLDOMNode_get_nodeValue( attr, &var );
 307     IXMLDOMNode_Release( attr );
 308     if (hr != S_OK) return NULL;
 309     if (V_VT(&var) != VT_BSTR)
 310     {
 311         VariantClear( &var );
 312         return NULL;
 313     }
 314     TRACE("%s=%s\n", debugstr_w(value_name), debugstr_w(V_BSTR( &var )));
 315     return V_BSTR( &var );
 316 }
 317
 318 struct file
 319 {
 320     struct list entry;
 321     BSTR name;
 322 };
 323
 324 struct assembly
 325 {
 326     BSTR type;
 327     BSTR name;
 328     BSTR version;
 329     BSTR arch;
 330     BSTR token;
 331     struct list files;
 332 };
 333
 334 static void free_assembly( struct assembly *assembly )
 335 {
 336     struct list *item, *cursor;
 337
 338     if (!assembly) return;
 339     SysFreeString( assembly->type );
 340     SysFreeString( assembly->name );
 341     SysFreeString( assembly->version );
 342     SysFreeString( assembly->arch );
 343     SysFreeString( assembly->token );
 344     LIST_FOR_EACH_SAFE( item, cursor, &assembly->files )
 345     {
 346         struct file *file = LIST_ENTRY( item, struct file, entry );
 347         list_remove( &file->entry );
 348         SysFreeString( file->name );
 349         HeapFree( GetProcessHeap(), 0, file );
 350     }
 351     HeapFree( GetProcessHeap(), 0, assembly );
 352 }
 353
 354 static HRESULT parse_files( IXMLDOMDocument *doc, struct assembly *assembly )
 355 {
 356     static const WCHAR fileW[] = {'f','i','l','e',0};
 357     static const WCHAR nameW[] = {'n','a','m','e',0};
 358     IXMLDOMNamedNodeMap *attrs;
 359     IXMLDOMNodeList *list;
 360     IXMLDOMNode *node;
 361     struct file *f;
 362     BSTR str;
 363     HRESULT hr;
 364     LONG len;
 365
 366     str = SysAllocString( fileW );
 367     hr = IXMLDOMDocument_getElementsByTagName( doc, str, &list );
 368     SysFreeString( str );
 369     if (hr != S_OK) return hr;
 370
 371     hr = IXMLDOMNodeList_get_length( list, &len );
 372     if (hr != S_OK) goto done;
 373     TRACE("found %d files\n", len);
 374     if (!len)
 375     {
 376         hr = ERROR_SXS_MANIFEST_FORMAT_ERROR;
 377         goto done;
 378     }
 379
 380     for (;;)
 381     {
 382         hr = IXMLDOMNodeList_nextNode( list, &node );
 383         if (hr != S_OK || !node)
 384         {
 385             hr = S_OK;
 386             break;
 387         }
 388
 389         /* FIXME: validate node type */
 390
 391         hr = IXMLDOMNode_get_attributes( node, &attrs );
 392         IXMLDOMNode_Release( node );
 393         if (hr != S_OK)
 394             goto done;
 395
 396         if (!(f = HeapAlloc( GetProcessHeap(), 0, sizeof(struct file) )))
 397         {
 398             IXMLDOMNamedNodeMap_Release( attrs );
 399             hr = E_OUTOFMEMORY;
 400             goto done;
 401         }
 402
 403         f->name = get_attribute_value( attrs, nameW );
 404         IXMLDOMNamedNodeMap_Release( attrs );
 405         if (!f->name)
 406         {
 407             HeapFree( GetProcessHeap(), 0, f );
 408             hr = ERROR_SXS_MANIFEST_FORMAT_ERROR;
 409             goto done;
 410         }
 411         list_add_tail( &assembly->files, &f->entry );
 412     }
 413
 414     if (list_empty( &assembly->files ))
 415     {
 416         WARN("no files found\n");
 417         hr = ERROR_SXS_MANIFEST_FORMAT_ERROR;
 418     }
 419
 420 done:
 421     IXMLDOMNodeList_Release( list );
 422     return hr;
 423 }
 424
 425 static HRESULT parse_assembly( IXMLDOMDocument *doc, struct assembly **assembly )
 426 {
 427     static const WCHAR identityW[] = {'a','s','s','e','m','b','l','y','I','d','e','n','t','i','t','y',0};
 428     static const WCHAR typeW[] = {'t','y','p','e',0};
 429     static const WCHAR nameW[] = {'n','a','m','e',0};
 430     static const WCHAR versionW[] = {'v','e','r','s','i','o','n',0};
 431     static const WCHAR architectureW[] = {'p','r','o','c','e','s','s','o','r','A','r','c','h','i','t','e','c','t','u','r','e',0};
 432     static const WCHAR tokenW[] = {'p','u','b','l','i','c','K','e','y','T','o','k','e','n',0};
 433     IXMLDOMNodeList *list = NULL;
 434     IXMLDOMNode *node = NULL;
 435     IXMLDOMNamedNodeMap *attrs = NULL;
 436     struct assembly *a = NULL;
 437     BSTR str;
 438     HRESULT hr;
 439     LONG len;
 440
 441     str = SysAllocString( identityW );
 442     hr = IXMLDOMDocument_getElementsByTagName( doc, str, &list );
 443     SysFreeString( str );
 444     if (hr != S_OK) goto done;
 445
 446     hr = IXMLDOMNodeList_get_length( list, &len );
 447     if (hr != S_OK) goto done;
 448     if (!len)
 449     {
 450         hr = ERROR_SXS_MANIFEST_FORMAT_ERROR;
 451         goto done;
 452     }
 453     hr = IXMLDOMNodeList_nextNode( list, &node );
 454     if (hr != S_OK) goto done;
 455     if (!node)
 456     {
 457         hr = ERROR_SXS_MANIFEST_FORMAT_ERROR;
 458         goto done;
 459     }
 460     if (!(a = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(struct assembly) )))
 461     {
 462         hr = E_OUTOFMEMORY;
 463         goto done;
 464     }
 465     list_init( &a->files );
 466
 467     hr = IXMLDOMNode_get_attributes( node, &attrs );
 468     if (hr != S_OK) goto done;
 469
 470     a->type    = get_attribute_value( attrs, typeW );
 471     a->name    = get_attribute_value( attrs, nameW );
 472     a->version = get_attribute_value( attrs, versionW );
 473     a->arch    = get_attribute_value( attrs, architectureW );
 474     a->token   = get_attribute_value( attrs, tokenW );
 475
 476     if (!a->type || (strcmpW( a->type, win32W ) && strcmpW( a->type, win32_policyW )) ||
 477         !a->name || !a->version || !a->arch || !a->token)
 478     {
 479         WARN("invalid win32 assembly\n");
 480         hr = ERROR_SXS_MANIFEST_FORMAT_ERROR;
 481         goto done;
 482     }
 483     if (!strcmpW( a->type, win32W )) hr = parse_files( doc, a );
 484
 485 done:
 486     if (attrs) IXMLDOMNamedNodeMap_Release( attrs );
 487     if (node) IXMLDOMNode_Release( node );
 488     if (list) IXMLDOMNodeList_Release( list );
 489     if (hr == S_OK) *assembly = a;
 490     else free_assembly( a );
 491     return hr;
 492 }
 493
 494 static WCHAR *build_policy_filename( const WCHAR *arch, const WCHAR *name, const WCHAR *token,
 495                                      const WCHAR *version )
 496 {
 497     static const WCHAR policiesW[] = {'p','o','l','i','c','i','e','s','\\',0};
 498     static const WCHAR suffixW[] = {'.','p','o','l','i','c','y',0};
 499     WCHAR sxsdir[MAX_PATH], *ret, *fullname;
 500     unsigned int len;
 501
 502     if (!(fullname = build_policy_name( arch, name, token, &len ))) return NULL;
 503     len += build_sxs_path( sxsdir );
 504     len += sizeof(policiesW) / sizeof(policiesW[0]) - 1;
 505     len += strlenW( version );
 506     len += sizeof(suffixW) / sizeof(suffixW[0]) - 1;
 507     if (!(ret = HeapAlloc( GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR) )))
 508     {
 509         HeapFree( GetProcessHeap(), 0, fullname );
 510         return NULL;
 511     }
 512     strcpyW( ret, sxsdir );
 513     strcatW( ret, policiesW );
 514     CreateDirectoryW( ret, NULL );
 515     strcatW( ret, name );
 516     CreateDirectoryW( ret, NULL );
 517     strcatW( ret, backslashW );
 518     strcatW( ret, version );
 519     strcatW( ret, suffixW );
 520
 521     HeapFree( GetProcessHeap(), 0, fullname );
 522     return ret;
 523 }
 524
 525 static HRESULT install_policy( const WCHAR *manifest, struct assembly *assembly )
 526 {
 527     WCHAR *dst;
 528     BOOL ret;
 529
 530     /* FIXME: handle catalog file */
 531
 532     dst = build_policy_filename( assembly->arch, assembly->name, assembly->token, assembly->version );
 533     if (!dst) return E_OUTOFMEMORY;
 534
 535     ret = CopyFileW( manifest, dst, FALSE );
 536     HeapFree( GetProcessHeap(), 0, dst );
 537     if (!ret)
 538     {
 539         HRESULT hr = HRESULT_FROM_WIN32( GetLastError() );
 540         WARN("failed to copy policy manifest file 0x%08x\n", hr);
 541         return hr;
 542     }
 543     return S_OK;
 544 }
 545
 546 static WCHAR *build_source_filename( const WCHAR *manifest, struct file *file )
 547 {
 548     WCHAR *src;
 549     const WCHAR *p;
 550     int len;
 551
 552     p = strrchrW( manifest, '\\' );
 553     if (!p) p = strrchrW( manifest, '/' );
 554     if (!p) return strdupW( manifest );
 555
 556     len = p - manifest + 1;
 557     if (!(src = HeapAlloc( GetProcessHeap(), 0, (len + strlenW( file->name ) + 1) * sizeof(WCHAR) )))
 558         return NULL;
 559
 560     memcpy( src, manifest, len * sizeof(WCHAR) );
 561     strcpyW( src + len, file->name );
 562     return src;
 563 }
 564
 565 static WCHAR *build_manifest_filename( const WCHAR *arch, const WCHAR *name, const WCHAR *token,
 566                                        const WCHAR *version )
 567 {
 568     static const WCHAR manifestsW[] = {'m','a','n','i','f','e','s','t','s','\\',0};
 569     static const WCHAR suffixW[] = {'.','m','a','n','i','f','e','s','t',0};
 570     WCHAR sxsdir[MAX_PATH], *ret, *fullname;
 571     unsigned int len;
 572
 573     if (!(fullname = build_assembly_name( arch, name, token, version, &len ))) return NULL;
 574     len += build_sxs_path( sxsdir );
 575     len += sizeof(manifestsW) / sizeof(manifestsW[0]) - 1;
 576     len += sizeof(suffixW) / sizeof(suffixW[0]) - 1;
 577     if (!(ret = HeapAlloc( GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR) )))
 578     {
 579         HeapFree( GetProcessHeap(), 0, fullname );
 580         return NULL;
 581     }
 582     strcpyW( ret, sxsdir );
 583     strcatW( ret, manifestsW );
 584     strcatW( ret, fullname );
 585     strcatW( ret, suffixW );
 586
 587     HeapFree( GetProcessHeap(), 0, fullname );
 588     return ret;
 589 }
 590
 591 static HRESULT load_manifest( IXMLDOMDocument *doc, const WCHAR *filename )
 592 {
 593     HRESULT hr;
 594     VARIANT var;
 595     VARIANT_BOOL b;
 596     BSTR str;
 597
 598     str = SysAllocString( filename );
 599     VariantInit( &var );
 600     V_VT( &var ) = VT_BSTR;
 601     V_BSTR( &var ) = str;
 602     hr = IXMLDOMDocument_load( doc, var, &b );
 603     SysFreeString( str );
 604     if (hr != S_OK) return hr;
 605     if (!b)
 606     {
 607         WARN("failed to load manifest\n");
 608         return S_FALSE;
 609     }
 610     return S_OK;
 611 }
 612
 613 static HRESULT install_assembly( const WCHAR *manifest, struct assembly *assembly )
 614 {
 615     WCHAR sxsdir[MAX_PATH], *p, *name, *dst, *src;
 616     unsigned int len, len_name, len_sxsdir = build_sxs_path( sxsdir );
 617     struct file *file;
 618     HRESULT hr = E_OUTOFMEMORY;
 619     BOOL ret;
 620
 621     dst = build_manifest_filename( assembly->arch, assembly->name, assembly->token, assembly->version );
 622     if (!dst) return E_OUTOFMEMORY;
 623
 624     ret = CopyFileW( manifest, dst, FALSE );
 625     HeapFree( GetProcessHeap(), 0, dst );
 626     if (!ret)
 627     {
 628         hr = HRESULT_FROM_WIN32( GetLastError() );
 629         WARN("failed to copy manifest file 0x%08x\n", hr);
 630         return hr;
 631     }
 632
 633     name = build_assembly_name( assembly->arch, assembly->name, assembly->token, assembly->version,
 634                                 &len_name );
 635     if (!name) return E_OUTOFMEMORY;
 636
 637     /* FIXME: this should be a transaction */
 638     LIST_FOR_EACH_ENTRY( file, &assembly->files, struct file, entry )
 639     {
 640         if (!(src = build_source_filename( manifest, file ))) goto done;
 641
 642         len = len_sxsdir + len_name + strlenW( file->name );
 643         if (!(dst = HeapAlloc( GetProcessHeap(), 0, (len + 2) * sizeof(WCHAR) )))
 644         {
 645             HeapFree( GetProcessHeap(), 0, src );
 646             goto done;
 647         }
 648         strcpyW( dst, sxsdir );
 649         strcatW( dst, name );
 650         CreateDirectoryW( dst, NULL );
 651
 652         strcatW( dst, backslashW );
 653         strcatW( dst, file->name );
 654         for (p = dst; *p; p++) *p = tolowerW( *p );
 655
 656         ret = CopyFileW( src, dst, FALSE );
 657         HeapFree( GetProcessHeap(), 0, src );
 658         HeapFree( GetProcessHeap(), 0, dst );
 659         if (!ret)
 660         {
 661             hr = HRESULT_FROM_WIN32( GetLastError() );
 662             WARN("failed to copy file 0x%08x\n", hr);
 663             goto done;
 664         }
 665     }
 666     hr = S_OK;
 667
 668 done:
 669     HeapFree( GetProcessHeap(), 0, name );
 670     return hr;
 671 }
 672
 673 static HRESULT WINAPI cache_InstallAssembly(
 674     IAssemblyCache *iface,
 675     DWORD flags,
 676     LPCWSTR path,
 677     LPCFUSION_INSTALL_REFERENCE ref )
 678 {
 679     struct cache *cache = impl_from_IAssemblyCache( iface );
 680     HRESULT hr, init;
 681     IXMLDOMDocument *doc = NULL;
 682     struct assembly *assembly = NULL;
 683
 684     TRACE("%p, 0x%08x, %s, %p\n", iface, flags, debugstr_w(path), ref);
 685
 686     cache_lock( cache );
 687     init = CoInitialize( NULL );
 688
 689     hr = CoCreateInstance( &CLSID_DOMDocument, NULL, CLSCTX_INPROC_SERVER, &IID_IXMLDOMDocument, (void **)&doc );
 690     if (hr != S_OK)
 691         goto done;
 692
 693     if ((hr = load_manifest( doc, path )) != S_OK) goto done;
 694     if ((hr = parse_assembly( doc, &assembly )) != S_OK) goto done;
 695
 696     /* FIXME: verify name attributes */
 697
 698     if (!strcmpW( assembly->type, win32_policyW ))
 699         hr = install_policy( path, assembly );
 700     else
 701         hr = install_assembly( path, assembly );
 702
 703 done:
 704     free_assembly( assembly );
 705     if (doc) IXMLDOMDocument_Release( doc );
 706     if (SUCCEEDED(init)) CoUninitialize();
 707     cache_unlock( cache );
 708     return hr;
 709 }
 710
 711 static HRESULT uninstall_assembly( struct assembly *assembly )
 712 {
 713     WCHAR sxsdir[MAX_PATH], *name, *dirname, *filename;
 714     unsigned int len, len_name, len_sxsdir = build_sxs_path( sxsdir );
 715     HRESULT hr = E_OUTOFMEMORY;
 716     struct file *file;
 717
 718     name = build_assembly_name( assembly->arch, assembly->name, assembly->token, assembly->version,
 719                                 &len_name );
 720     if (!name) return E_OUTOFMEMORY;
 721     if (!(dirname = HeapAlloc( GetProcessHeap(), 0, (len_sxsdir + len_name + 1) * sizeof(WCHAR) )))
 722         goto done;
 723     strcpyW( dirname, sxsdir );
 724     strcpyW( dirname + len_sxsdir, name );
 725
 726     LIST_FOR_EACH_ENTRY( file, &assembly->files, struct file, entry )
 727     {
 728         len = len_sxsdir + len_name + 1 + strlenW( file->name );
 729         if (!(filename = HeapAlloc( GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR) ))) goto done;
 730         strcpyW( filename, dirname );
 731         strcatW( filename, backslashW );
 732         strcatW( filename, file->name );
 733
 734         if (!DeleteFileW( filename )) WARN( "failed to delete file %u\n", GetLastError() );
 735         HeapFree( GetProcessHeap(), 0, filename );
 736     }
 737     RemoveDirectoryW( dirname );
 738     hr = S_OK;
 739
 740 done:
 741     HeapFree( GetProcessHeap(), 0, dirname );
 742     HeapFree( GetProcessHeap(), 0, name );
 743     return hr;
 744 }
 745
 746 static HRESULT WINAPI cache_UninstallAssembly(
 747     IAssemblyCache *iface,
 748     DWORD flags,
 749     LPCWSTR assembly_name,
 750     LPCFUSION_INSTALL_REFERENCE ref,
 751     ULONG *disp )
 752 {
 753     struct cache *cache = impl_from_IAssemblyCache( iface );
 754     HRESULT hr, init;
 755     IXMLDOMDocument *doc = NULL;
 756     struct assembly *assembly = NULL;
 757     IAssemblyName *name_obj = NULL;
 758     const WCHAR *arch, *name, *token, *type, *version;
 759     WCHAR *p, *path = NULL;
 760
 761     TRACE("%p, 0x%08x, %s, %p, %p\n", iface, flags, debugstr_w(assembly_name), ref, disp);
 762
 763     if (ref)
 764     {
 765         FIXME("application reference not supported\n");
 766         return E_NOTIMPL;
 767     }
 768     cache_lock( cache );
 769     init = CoInitialize( NULL );
 770
 771     hr = CreateAssemblyNameObject( &name_obj, assembly_name, CANOF_PARSE_DISPLAY_NAME, NULL );
 772     if (FAILED( hr ))
 773         goto done;
 774
 775     arch = get_name_attribute( name_obj, NAME_ATTR_ID_ARCH );
 776     name = get_name_attribute( name_obj, NAME_ATTR_ID_NAME );
 777     token = get_name_attribute( name_obj, NAME_ATTR_ID_TOKEN );
 778     type = get_name_attribute( name_obj, NAME_ATTR_ID_TYPE );
 779     version = get_name_attribute( name_obj, NAME_ATTR_ID_VERSION );
 780     if (!arch || !name || !token || !type || !version)
 781     {
 782         hr = E_INVALIDARG;
 783         goto done;
 784     }
 785     if (!strcmpW( type, win32W )) path = build_manifest_filename( arch, name, token, version );
 786     else if (!strcmpW( type, win32_policyW )) path = build_policy_filename( arch, name, token, version );
 787     else
 788     {
 789         hr = E_INVALIDARG;
 790         goto done;
 791     }
 792
 793     hr = CoCreateInstance( &CLSID_DOMDocument, NULL, CLSCTX_INPROC_SERVER, &IID_IXMLDOMDocument, (void **)&doc );
 794     if (hr != S_OK)
 795         goto done;
 796
 797     if ((hr = load_manifest( doc, path )) != S_OK) goto done;
 798     if ((hr = parse_assembly( doc, &assembly )) != S_OK) goto done;
 799
 800     if (!DeleteFileW( path )) WARN( "unable to remove manifest file %u\n", GetLastError() );
 801     else if ((p = strrchrW( path, '\\' )))
 802     {
 803         *p = 0;
 804         RemoveDirectoryW( path );
 805     }
 806     if (!strcmpW( assembly->type, win32W )) hr = uninstall_assembly( assembly );
 807
 808 done:
 809     if (name_obj) IAssemblyName_Release( name_obj );
 810     HeapFree( GetProcessHeap(), 0, path );
 811     free_assembly( assembly );
 812     if (doc) IXMLDOMDocument_Release( doc );
 813     if (SUCCEEDED(init)) CoUninitialize();
 814     cache_unlock( cache );
 815     return hr;
 816 }
 817
 818 static const IAssemblyCacheVtbl cache_vtbl =
 819 {
 820     cache_QueryInterface,
 821     cache_AddRef,
 822     cache_Release,
 823     cache_UninstallAssembly,
 824     cache_QueryAssemblyInfo,
 825     cache_CreateAssemblyCacheItem,
 826     cache_Reserved,
 827     cache_InstallAssembly
 828 };
 829
 830 /******************************************************************
 831  *  CreateAssemblyCache   (SXS.@)
 832  */
 833 HRESULT WINAPI CreateAssemblyCache( IAssemblyCache **obj, DWORD reserved )
 834 {
 835     struct cache *cache;
 836
 837     TRACE("%p, %u\n", obj, reserved);
 838
 839     if (!obj)
 840         return E_INVALIDARG;
 841
 842     *obj = NULL;
 843
 844     cache = HeapAlloc( GetProcessHeap(), 0, sizeof(struct cache) );
 845     if (!cache)
 846         return E_OUTOFMEMORY;
 847
 848     cache->IAssemblyCache_iface.lpVtbl = &cache_vtbl;
 849     cache->refs = 1;
 850     cache->lock = CreateMutexW( NULL, FALSE, cache_mutex_nameW );
 851     if (!cache->lock)
 852     {
 853         HeapFree( GetProcessHeap(), 0, cache );
 854         return HRESULT_FROM_WIN32( GetLastError() );
 855     }
 856     *obj = &cache->IAssemblyCache_iface;
 857     return S_OK;
 858 }