mshtml: Fixed Exec_ShellDocView_84 tests.
[wine/multimedia.git] / dlls / sxs / cache.c
blob46686e541fea50edc5f2a000f69ee6a1c077ea16
1 /*
2 * IAssemblyCache implementation
4 * Copyright 2010 Hans Leidekker for CodeWeavers
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.
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.
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
21 #include <stdarg.h>
23 #define COBJMACROS
24 #define INITGUID
26 #include "windef.h"
27 #include "winbase.h"
28 #include "ole2.h"
29 #include "winsxs.h"
30 #include "msxml2.h"
32 #include "wine/debug.h"
33 #include "wine/list.h"
34 #include "wine/unicode.h"
35 #include "sxs_private.h"
37 WINE_DEFAULT_DEBUG_CHANNEL(sxs);
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};
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};
46 struct cache
48 IAssemblyCache IAssemblyCache_iface;
49 LONG refs;
50 HANDLE lock;
53 static inline struct cache *impl_from_IAssemblyCache(IAssemblyCache *iface)
55 return CONTAINING_RECORD(iface, struct cache, IAssemblyCache_iface);
58 static HRESULT WINAPI cache_QueryInterface(
59 IAssemblyCache *iface,
60 REFIID riid,
61 void **obj )
63 struct cache *cache = impl_from_IAssemblyCache(iface);
65 TRACE("%p, %s, %p\n", cache, debugstr_guid(riid), obj);
67 *obj = NULL;
69 if (IsEqualIID(riid, &IID_IUnknown) ||
70 IsEqualIID(riid, &IID_IAssemblyCache))
72 IUnknown_AddRef( iface );
73 *obj = cache;
74 return S_OK;
77 return E_NOINTERFACE;
80 static ULONG WINAPI cache_AddRef( IAssemblyCache *iface )
82 struct cache *cache = impl_from_IAssemblyCache(iface);
83 return InterlockedIncrement( &cache->refs );
86 static ULONG WINAPI cache_Release( IAssemblyCache *iface )
88 struct cache *cache = impl_from_IAssemblyCache(iface);
89 ULONG refs = InterlockedDecrement( &cache->refs );
91 if (!refs)
93 TRACE("destroying %p\n", cache);
94 CloseHandle( cache->lock );
95 HeapFree( GetProcessHeap(), 0, cache );
97 return refs;
100 static unsigned int build_sxs_path( WCHAR *path )
102 static const WCHAR winsxsW[] = {'\\','w','i','n','s','x','s','\\',0};
103 unsigned int len = GetWindowsDirectoryW( path, MAX_PATH );
105 memcpy( path + len, winsxsW, sizeof(winsxsW) );
106 return len + sizeof(winsxsW) / sizeof(winsxsW[0]) - 1;
109 static WCHAR *build_assembly_name( const WCHAR *arch, const WCHAR *name, const WCHAR *token,
110 const WCHAR *version, unsigned int *len )
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;
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;
127 static WCHAR *build_manifest_path( const WCHAR *arch, const WCHAR *name, const WCHAR *token,
128 const WCHAR *version )
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;
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) )))
140 HeapFree( GetProcessHeap(), 0, path );
141 return NULL;
143 sprintfW( ret, fmtW, sxsdir, path );
144 HeapFree( GetProcessHeap(), 0, path );
145 return ret;
148 static WCHAR *build_policy_name( const WCHAR *arch, const WCHAR *name, const WCHAR *token,
149 unsigned int *len )
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;
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;
165 static WCHAR *build_policy_path( const WCHAR *arch, const WCHAR *name, const WCHAR *token,
166 const WCHAR *version )
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;
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) )))
179 HeapFree( GetProcessHeap(), 0, path );
180 return NULL;
182 sprintfW( ret, fmtW, sxsdir, path, version );
183 HeapFree( GetProcessHeap(), 0, path );
184 return ret;
187 static void cache_lock( struct cache *cache )
189 WaitForSingleObject( cache->lock, INFINITE );
192 static void cache_unlock( struct cache *cache )
194 ReleaseMutex( cache->lock );
197 #define ASSEMBLYINFO_FLAG_INSTALLED 1
199 static HRESULT WINAPI cache_QueryAssemblyInfo(
200 IAssemblyCache *iface,
201 DWORD flags,
202 LPCWSTR assembly_name,
203 ASSEMBLY_INFO *info )
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;
212 TRACE("%p, 0x%08x, %s, %p\n", iface, flags, debugstr_w(assembly_name), info);
214 if (flags || (info && info->cbAssemblyInfo != sizeof(*info)))
215 return E_INVALIDARG;
217 hr = CreateAssemblyNameObject( &name_obj, assembly_name, CANOF_PARSE_DISPLAY_NAME, 0 );
218 if (FAILED( hr ))
219 return hr;
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)
228 IAssemblyName_Release( name_obj );
229 return HRESULT_FROM_WIN32( ERROR_SXS_MISSING_ASSEMBLY_IDENTITY_ATTRIBUTE );
231 if (!info)
233 IAssemblyName_Release( name_obj );
234 return S_OK;
236 cache_lock( cache );
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
242 hr = HRESULT_FROM_WIN32( ERROR_SXS_INVALID_IDENTITY_ATTRIBUTE_VALUE );
243 goto done;
245 if (!path)
247 hr = E_OUTOFMEMORY;
248 goto done;
250 hr = S_OK;
251 if (GetFileAttributesW( path ) != INVALID_FILE_ATTRIBUTES) /* FIXME: better check */
253 info->dwAssemblyFlags = ASSEMBLYINFO_FLAG_INSTALLED;
254 TRACE("assembly is installed\n");
256 if ((p = strrchrW( path, '\\' ))) *p = 0;
257 len = strlenW( path ) + 1;
258 if (info->pszCurrentAssemblyPathBuf)
260 if (info->cchBuf < len)
262 info->cchBuf = len;
263 hr = HRESULT_FROM_WIN32( ERROR_INSUFFICIENT_BUFFER );
265 else strcpyW( info->pszCurrentAssemblyPathBuf, path );
268 done:
269 HeapFree( GetProcessHeap(), 0, path );
270 IAssemblyName_Release( name_obj );
271 cache_unlock( cache );
272 return hr;
275 static HRESULT WINAPI cache_CreateAssemblyCacheItem(
276 IAssemblyCache *iface,
277 DWORD flags,
278 PVOID reserved,
279 IAssemblyCacheItem **item,
280 LPCWSTR name )
282 FIXME("%p, 0x%08x, %p, %p, %s\n", iface, flags, reserved, item, debugstr_w(name));
283 return E_NOTIMPL;
286 static HRESULT WINAPI cache_Reserved(
287 IAssemblyCache *iface,
288 IUnknown **reserved)
290 FIXME("%p\n", reserved);
291 return E_NOTIMPL;
294 static BSTR get_attribute_value( IXMLDOMNamedNodeMap *map, const WCHAR *value_name )
296 HRESULT hr;
297 IXMLDOMNode *attr;
298 VARIANT var;
299 BSTR str;
301 str = SysAllocString( value_name );
302 hr = IXMLDOMNamedNodeMap_getNamedItem( map, str, &attr );
303 SysFreeString( str );
304 if (hr != S_OK) return NULL;
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)
311 VariantClear( &var );
312 return NULL;
314 TRACE("%s=%s\n", debugstr_w(value_name), debugstr_w(V_BSTR( &var )));
315 return V_BSTR( &var );
318 struct file
320 struct list entry;
321 BSTR name;
324 struct assembly
326 BSTR type;
327 BSTR name;
328 BSTR version;
329 BSTR arch;
330 BSTR token;
331 struct list files;
334 static void free_assembly( struct assembly *assembly )
336 struct list *item, *cursor;
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 )
346 struct file *file = LIST_ENTRY( item, struct file, entry );
347 list_remove( &file->entry );
348 SysFreeString( file->name );
349 HeapFree( GetProcessHeap(), 0, file );
351 HeapFree( GetProcessHeap(), 0, assembly );
354 static HRESULT parse_files( IXMLDOMDocument *doc, struct assembly *assembly )
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;
366 str = SysAllocString( fileW );
367 hr = IXMLDOMDocument_getElementsByTagName( doc, str, &list );
368 SysFreeString( str );
369 if (hr != S_OK) return hr;
371 hr = IXMLDOMNodeList_get_length( list, &len );
372 if (hr != S_OK) goto done;
373 TRACE("found %d files\n", len);
374 if (!len)
376 hr = ERROR_SXS_MANIFEST_FORMAT_ERROR;
377 goto done;
380 for (;;)
382 hr = IXMLDOMNodeList_nextNode( list, &node );
383 if (hr != S_OK || !node)
385 hr = S_OK;
386 break;
389 /* FIXME: validate node type */
391 hr = IXMLDOMNode_get_attributes( node, &attrs );
392 IXMLDOMNode_Release( node );
393 if (hr != S_OK)
394 goto done;
396 if (!(f = HeapAlloc( GetProcessHeap(), 0, sizeof(struct file) )))
398 IXMLDOMNamedNodeMap_Release( attrs );
399 hr = E_OUTOFMEMORY;
400 goto done;
403 f->name = get_attribute_value( attrs, nameW );
404 IXMLDOMNamedNodeMap_Release( attrs );
405 if (!f->name)
407 HeapFree( GetProcessHeap(), 0, f );
408 hr = ERROR_SXS_MANIFEST_FORMAT_ERROR;
409 goto done;
411 list_add_tail( &assembly->files, &f->entry );
414 if (list_empty( &assembly->files ))
416 WARN("no files found\n");
417 hr = ERROR_SXS_MANIFEST_FORMAT_ERROR;
420 done:
421 IXMLDOMNodeList_Release( list );
422 return hr;
425 static HRESULT parse_assembly( IXMLDOMDocument *doc, struct assembly **assembly )
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;
441 str = SysAllocString( identityW );
442 hr = IXMLDOMDocument_getElementsByTagName( doc, str, &list );
443 SysFreeString( str );
444 if (hr != S_OK) goto done;
446 hr = IXMLDOMNodeList_get_length( list, &len );
447 if (hr != S_OK) goto done;
448 if (!len)
450 hr = ERROR_SXS_MANIFEST_FORMAT_ERROR;
451 goto done;
453 hr = IXMLDOMNodeList_nextNode( list, &node );
454 if (hr != S_OK) goto done;
455 if (!node)
457 hr = ERROR_SXS_MANIFEST_FORMAT_ERROR;
458 goto done;
460 if (!(a = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(struct assembly) )))
462 hr = E_OUTOFMEMORY;
463 goto done;
465 list_init( &a->files );
467 hr = IXMLDOMNode_get_attributes( node, &attrs );
468 if (hr != S_OK) goto done;
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 );
476 if (!a->type || (strcmpW( a->type, win32W ) && strcmpW( a->type, win32_policyW )) ||
477 !a->name || !a->version || !a->arch || !a->token)
479 WARN("invalid win32 assembly\n");
480 hr = ERROR_SXS_MANIFEST_FORMAT_ERROR;
481 goto done;
483 if (!strcmpW( a->type, win32W )) hr = parse_files( doc, a );
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;
494 static WCHAR *build_policy_filename( const WCHAR *arch, const WCHAR *name, const WCHAR *token,
495 const WCHAR *version )
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;
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) )))
509 HeapFree( GetProcessHeap(), 0, fullname );
510 return NULL;
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 );
521 HeapFree( GetProcessHeap(), 0, fullname );
522 return ret;
525 static HRESULT install_policy( const WCHAR *manifest, struct assembly *assembly )
527 WCHAR *dst;
528 BOOL ret;
530 /* FIXME: handle catalog file */
532 dst = build_policy_filename( assembly->arch, assembly->name, assembly->token, assembly->version );
533 if (!dst) return E_OUTOFMEMORY;
535 ret = CopyFileW( manifest, dst, FALSE );
536 HeapFree( GetProcessHeap(), 0, dst );
537 if (!ret)
539 HRESULT hr = HRESULT_FROM_WIN32( GetLastError() );
540 WARN("failed to copy policy manifest file 0x%08x\n", hr);
541 return hr;
543 return S_OK;
546 static WCHAR *build_source_filename( const WCHAR *manifest, struct file *file )
548 WCHAR *src;
549 const WCHAR *p;
550 int len;
552 p = strrchrW( manifest, '\\' );
553 if (!p) p = strrchrW( manifest, '/' );
554 if (!p) return strdupW( manifest );
556 len = p - manifest + 1;
557 if (!(src = HeapAlloc( GetProcessHeap(), 0, (len + strlenW( file->name ) + 1) * sizeof(WCHAR) )))
558 return NULL;
560 memcpy( src, manifest, len * sizeof(WCHAR) );
561 strcpyW( src + len, file->name );
562 return src;
565 static WCHAR *build_manifest_filename( const WCHAR *arch, const WCHAR *name, const WCHAR *token,
566 const WCHAR *version )
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;
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) )))
579 HeapFree( GetProcessHeap(), 0, fullname );
580 return NULL;
582 strcpyW( ret, sxsdir );
583 strcatW( ret, manifestsW );
584 strcatW( ret, fullname );
585 strcatW( ret, suffixW );
587 HeapFree( GetProcessHeap(), 0, fullname );
588 return ret;
591 static HRESULT load_manifest( IXMLDOMDocument *doc, const WCHAR *filename )
593 HRESULT hr;
594 VARIANT var;
595 VARIANT_BOOL b;
596 BSTR str;
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)
607 WARN("failed to load manifest\n");
608 return S_FALSE;
610 return S_OK;
613 static HRESULT install_assembly( const WCHAR *manifest, struct assembly *assembly )
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;
621 dst = build_manifest_filename( assembly->arch, assembly->name, assembly->token, assembly->version );
622 if (!dst) return E_OUTOFMEMORY;
624 ret = CopyFileW( manifest, dst, FALSE );
625 HeapFree( GetProcessHeap(), 0, dst );
626 if (!ret)
628 hr = HRESULT_FROM_WIN32( GetLastError() );
629 WARN("failed to copy manifest file 0x%08x\n", hr);
630 return hr;
633 name = build_assembly_name( assembly->arch, assembly->name, assembly->token, assembly->version,
634 &len_name );
635 if (!name) return E_OUTOFMEMORY;
637 /* FIXME: this should be a transaction */
638 LIST_FOR_EACH_ENTRY( file, &assembly->files, struct file, entry )
640 if (!(src = build_source_filename( manifest, file ))) goto done;
642 len = len_sxsdir + len_name + strlenW( file->name );
643 if (!(dst = HeapAlloc( GetProcessHeap(), 0, (len + 2) * sizeof(WCHAR) )))
645 HeapFree( GetProcessHeap(), 0, src );
646 goto done;
648 strcpyW( dst, sxsdir );
649 strcatW( dst, name );
650 CreateDirectoryW( dst, NULL );
652 strcatW( dst, backslashW );
653 strcatW( dst, file->name );
654 for (p = dst; *p; p++) *p = tolowerW( *p );
656 ret = CopyFileW( src, dst, FALSE );
657 HeapFree( GetProcessHeap(), 0, src );
658 HeapFree( GetProcessHeap(), 0, dst );
659 if (!ret)
661 hr = HRESULT_FROM_WIN32( GetLastError() );
662 WARN("failed to copy file 0x%08x\n", hr);
663 goto done;
666 hr = S_OK;
668 done:
669 HeapFree( GetProcessHeap(), 0, name );
670 return hr;
673 static HRESULT WINAPI cache_InstallAssembly(
674 IAssemblyCache *iface,
675 DWORD flags,
676 LPCWSTR path,
677 LPCFUSION_INSTALL_REFERENCE ref )
679 struct cache *cache = impl_from_IAssemblyCache( iface );
680 HRESULT hr, init;
681 IXMLDOMDocument *doc = NULL;
682 struct assembly *assembly = NULL;
684 TRACE("%p, 0x%08x, %s, %p\n", iface, flags, debugstr_w(path), ref);
686 cache_lock( cache );
687 init = CoInitialize( NULL );
689 hr = CoCreateInstance( &CLSID_DOMDocument, NULL, CLSCTX_INPROC_SERVER, &IID_IXMLDOMDocument, (void **)&doc );
690 if (hr != S_OK)
691 goto done;
693 if ((hr = load_manifest( doc, path )) != S_OK) goto done;
694 if ((hr = parse_assembly( doc, &assembly )) != S_OK) goto done;
696 /* FIXME: verify name attributes */
698 if (!strcmpW( assembly->type, win32_policyW ))
699 hr = install_policy( path, assembly );
700 else
701 hr = install_assembly( path, assembly );
703 done:
704 free_assembly( assembly );
705 if (doc) IXMLDOMDocument_Release( doc );
706 if (SUCCEEDED(init)) CoUninitialize();
707 cache_unlock( cache );
708 return hr;
711 static HRESULT uninstall_assembly( struct assembly *assembly )
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;
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 );
726 LIST_FOR_EACH_ENTRY( file, &assembly->files, struct file, entry )
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 );
734 if (!DeleteFileW( filename )) WARN( "failed to delete file %u\n", GetLastError() );
735 HeapFree( GetProcessHeap(), 0, filename );
737 RemoveDirectoryW( dirname );
738 hr = S_OK;
740 done:
741 HeapFree( GetProcessHeap(), 0, dirname );
742 HeapFree( GetProcessHeap(), 0, name );
743 return hr;
746 static HRESULT WINAPI cache_UninstallAssembly(
747 IAssemblyCache *iface,
748 DWORD flags,
749 LPCWSTR assembly_name,
750 LPCFUSION_INSTALL_REFERENCE ref,
751 ULONG *disp )
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;
761 TRACE("%p, 0x%08x, %s, %p, %p\n", iface, flags, debugstr_w(assembly_name), ref, disp);
763 if (ref)
765 FIXME("application reference not supported\n");
766 return E_NOTIMPL;
768 cache_lock( cache );
769 init = CoInitialize( NULL );
771 hr = CreateAssemblyNameObject( &name_obj, assembly_name, CANOF_PARSE_DISPLAY_NAME, NULL );
772 if (FAILED( hr ))
773 goto done;
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)
782 hr = E_INVALIDARG;
783 goto done;
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
789 hr = E_INVALIDARG;
790 goto done;
793 hr = CoCreateInstance( &CLSID_DOMDocument, NULL, CLSCTX_INPROC_SERVER, &IID_IXMLDOMDocument, (void **)&doc );
794 if (hr != S_OK)
795 goto done;
797 if ((hr = load_manifest( doc, path )) != S_OK) goto done;
798 if ((hr = parse_assembly( doc, &assembly )) != S_OK) goto done;
800 if (!DeleteFileW( path )) WARN( "unable to remove manifest file %u\n", GetLastError() );
801 else if ((p = strrchrW( path, '\\' )))
803 *p = 0;
804 RemoveDirectoryW( path );
806 if (!strcmpW( assembly->type, win32W )) hr = uninstall_assembly( assembly );
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;
818 static const IAssemblyCacheVtbl cache_vtbl =
820 cache_QueryInterface,
821 cache_AddRef,
822 cache_Release,
823 cache_UninstallAssembly,
824 cache_QueryAssemblyInfo,
825 cache_CreateAssemblyCacheItem,
826 cache_Reserved,
827 cache_InstallAssembly
830 /******************************************************************
831 * CreateAssemblyCache (SXS.@)
833 HRESULT WINAPI CreateAssemblyCache( IAssemblyCache **obj, DWORD reserved )
835 struct cache *cache;
837 TRACE("%p, %u\n", obj, reserved);
839 if (!obj)
840 return E_INVALIDARG;
842 *obj = NULL;
844 cache = HeapAlloc( GetProcessHeap(), 0, sizeof(struct cache) );
845 if (!cache)
846 return E_OUTOFMEMORY;
848 cache->IAssemblyCache_iface.lpVtbl = &cache_vtbl;
849 cache->refs = 1;
850 cache->lock = CreateMutexW( NULL, FALSE, cache_mutex_nameW );
851 if (!cache->lock)
853 HeapFree( GetProcessHeap(), 0, cache );
854 return HRESULT_FROM_WIN32( GetLastError() );
856 *obj = &cache->IAssemblyCache_iface;
857 return S_OK;