wininet: Improved non-blocking mode in secure NETCON_recv.
[wine.git] / dlls / fusion / asmcache.c
blob91256555a82827108c1721f3b1587aff5d335c13
1 /*
2 * IAssemblyCache implementation
4 * Copyright 2008 James Hawkins
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>
22 #include <stdio.h>
24 #define COBJMACROS
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"
38 #include "fusionpriv.h"
39 #include "wine/debug.h"
40 #include "wine/unicode.h"
42 WINE_DEFAULT_DEBUG_CHANNEL(fusion);
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};
47 static BOOL create_full_path(LPCWSTR path)
49 LPWSTR new_path;
50 BOOL ret = TRUE;
51 int len;
53 new_path = HeapAlloc(GetProcessHeap(), 0, (strlenW(path) + 1) * sizeof(WCHAR));
54 if (!new_path)
55 return FALSE;
57 strcpyW(new_path, path);
59 while ((len = strlenW(new_path)) && new_path[len - 1] == '\\')
60 new_path[len - 1] = 0;
62 while (!CreateDirectoryW(new_path, NULL))
64 LPWSTR slash;
65 DWORD last_error = GetLastError();
67 if(last_error == ERROR_ALREADY_EXISTS)
68 break;
70 if(last_error != ERROR_PATH_NOT_FOUND)
72 ret = FALSE;
73 break;
76 if(!(slash = strrchrW(new_path, '\\')))
78 ret = FALSE;
79 break;
82 len = slash - new_path;
83 new_path[len] = 0;
84 if(!create_full_path(new_path))
86 ret = FALSE;
87 break;
90 new_path[len] = '\\';
93 HeapFree(GetProcessHeap(), 0, new_path);
94 return ret;
97 static BOOL get_assembly_directory(LPWSTR dir, DWORD size, const char *version, PEKIND architecture)
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);
106 if (!strcmp(version, "v4.0.30319"))
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;
113 else
115 strcpyW(dir + len, gac);
116 len += sizeof(gac)/sizeof(WCHAR) - 1;
118 switch (architecture)
120 case peNone:
121 break;
123 case peMSIL:
124 strcpyW(dir + len, msil);
125 break;
127 case peI386:
128 strcpyW(dir + len, x86);
129 break;
131 case peAMD64:
132 strcpyW(dir + len, amd64);
133 break;
135 default:
136 WARN("unhandled architecture %u\n", architecture);
137 return FALSE;
139 return TRUE;
142 /* IAssemblyCache */
144 typedef struct {
145 IAssemblyCache IAssemblyCache_iface;
147 LONG ref;
148 HANDLE lock;
149 } IAssemblyCacheImpl;
151 static inline IAssemblyCacheImpl *impl_from_IAssemblyCache(IAssemblyCache *iface)
153 return CONTAINING_RECORD(iface, IAssemblyCacheImpl, IAssemblyCache_iface);
156 static HRESULT WINAPI IAssemblyCacheImpl_QueryInterface(IAssemblyCache *iface,
157 REFIID riid, LPVOID *ppobj)
159 IAssemblyCacheImpl *This = impl_from_IAssemblyCache(iface);
161 TRACE("(%p, %s, %p)\n", This, debugstr_guid(riid), ppobj);
163 *ppobj = NULL;
165 if (IsEqualIID(riid, &IID_IUnknown) ||
166 IsEqualIID(riid, &IID_IAssemblyCache))
168 IAssemblyCache_AddRef(iface);
169 *ppobj = This;
170 return S_OK;
173 WARN("(%p, %s, %p): not found\n", This, debugstr_guid(riid), ppobj);
174 return E_NOINTERFACE;
177 static ULONG WINAPI IAssemblyCacheImpl_AddRef(IAssemblyCache *iface)
179 IAssemblyCacheImpl *This = impl_from_IAssemblyCache(iface);
180 ULONG refCount = InterlockedIncrement(&This->ref);
182 TRACE("(%p)->(ref before = %u)\n", This, refCount - 1);
184 return refCount;
187 static ULONG WINAPI IAssemblyCacheImpl_Release(IAssemblyCache *iface)
189 IAssemblyCacheImpl *cache = impl_from_IAssemblyCache(iface);
190 ULONG refCount = InterlockedDecrement( &cache->ref );
192 TRACE("(%p)->(ref before = %u)\n", cache, refCount + 1);
194 if (!refCount)
196 CloseHandle( cache->lock );
197 HeapFree( GetProcessHeap(), 0, cache );
199 return refCount;
202 static void cache_lock( IAssemblyCacheImpl *cache )
204 WaitForSingleObject( cache->lock, INFINITE );
207 static void cache_unlock( IAssemblyCacheImpl *cache )
209 ReleaseMutex( cache->lock );
212 static HRESULT WINAPI IAssemblyCacheImpl_UninstallAssembly(IAssemblyCache *iface,
213 DWORD dwFlags,
214 LPCWSTR pszAssemblyName,
215 LPCFUSION_INSTALL_REFERENCE pRefData,
216 ULONG *pulDisposition)
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;
226 TRACE("(%p, 0%08x, %s, %p, %p)\n", iface, dwFlags,
227 debugstr_w(pszAssemblyName), pRefData, pulDisposition);
229 if (pRefData)
231 FIXME("application reference not supported\n");
232 return E_NOTIMPL;
234 hr = CreateAssemblyNameObject( &asmname, pszAssemblyName, CANOF_PARSE_DISPLAY_NAME, NULL );
235 if (FAILED( hr ))
236 return hr;
238 cache_lock( cache );
240 hr = CreateAssemblyEnum( &asmenum, NULL, asmname, ASM_CACHE_GAC, NULL );
241 if (FAILED( hr ))
242 goto done;
244 hr = IAssemblyEnum_GetNextAssembly( asmenum, NULL, &next, 0 );
245 if (hr == S_FALSE)
247 if (pulDisposition)
248 *pulDisposition = IASSEMBLYCACHE_UNINSTALL_DISPOSITION_ALREADY_UNINSTALLED;
249 goto done;
251 hr = IAssemblyName_GetPath( next, NULL, &len );
252 if (hr != HRESULT_FROM_WIN32( ERROR_INSUFFICIENT_BUFFER ))
253 goto done;
255 if (!(path = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) )))
257 hr = E_OUTOFMEMORY;
258 goto done;
260 hr = IAssemblyName_GetPath( next, path, &len );
261 if (FAILED( hr ))
262 goto done;
264 if (DeleteFileW( path ))
266 if ((p = strrchrW( path, '\\' )))
268 *p = 0;
269 RemoveDirectoryW( path );
270 if ((p = strrchrW( path, '\\' )))
272 *p = 0;
273 RemoveDirectoryW( path );
276 disp = IASSEMBLYCACHE_UNINSTALL_DISPOSITION_UNINSTALLED;
277 hr = S_OK;
279 else
281 disp = IASSEMBLYCACHE_UNINSTALL_DISPOSITION_ALREADY_UNINSTALLED;
282 hr = S_FALSE;
284 if (pulDisposition) *pulDisposition = disp;
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;
295 static HRESULT WINAPI IAssemblyCacheImpl_QueryAssemblyInfo(IAssemblyCache *iface,
296 DWORD dwFlags,
297 LPCWSTR pszAssemblyName,
298 ASSEMBLY_INFO *pAsmInfo)
300 IAssemblyCacheImpl *cache = impl_from_IAssemblyCache(iface);
301 IAssemblyName *asmname, *next = NULL;
302 IAssemblyEnum *asmenum = NULL;
303 HRESULT hr;
305 TRACE("(%p, %d, %s, %p)\n", iface, dwFlags,
306 debugstr_w(pszAssemblyName), pAsmInfo);
308 if (pAsmInfo)
310 if (pAsmInfo->cbAssemblyInfo == 0)
311 pAsmInfo->cbAssemblyInfo = sizeof(ASSEMBLY_INFO);
312 else if (pAsmInfo->cbAssemblyInfo != sizeof(ASSEMBLY_INFO))
313 return E_INVALIDARG;
316 hr = CreateAssemblyNameObject(&asmname, pszAssemblyName,
317 CANOF_PARSE_DISPLAY_NAME, NULL);
318 if (FAILED(hr))
319 return hr;
321 cache_lock( cache );
323 hr = CreateAssemblyEnum(&asmenum, NULL, asmname, ASM_CACHE_GAC, NULL);
324 if (FAILED(hr))
325 goto done;
327 for (;;)
329 hr = IAssemblyEnum_GetNextAssembly(asmenum, NULL, &next, 0);
330 if (hr != S_OK)
332 hr = HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
333 goto done;
335 hr = IAssemblyName_IsEqual(asmname, next, ASM_CMPF_IL_ALL);
336 if (hr == S_OK) break;
339 if (!pAsmInfo)
340 goto done;
342 hr = IAssemblyName_GetPath(next, pAsmInfo->pszCurrentAssemblyPathBuf, &pAsmInfo->cchBuf);
344 pAsmInfo->dwAssemblyFlags = ASSEMBLYINFO_FLAG_INSTALLED;
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;
354 static HRESULT WINAPI IAssemblyCacheImpl_CreateAssemblyCacheItem(IAssemblyCache *iface,
355 DWORD dwFlags,
356 PVOID pvReserved,
357 IAssemblyCacheItem **ppAsmItem,
358 LPCWSTR pszAssemblyName)
360 FIXME("(%p, %d, %p, %p, %s) stub!\n", iface, dwFlags, pvReserved,
361 ppAsmItem, debugstr_w(pszAssemblyName));
363 return E_NOTIMPL;
366 static HRESULT WINAPI IAssemblyCacheImpl_CreateAssemblyScavenger(IAssemblyCache *iface,
367 IUnknown **ppUnkReserved)
369 FIXME("(%p, %p) stub!\n", iface, ppUnkReserved);
370 return E_NOTIMPL;
373 static HRESULT copy_file( const WCHAR *src_dir, DWORD src_len, const WCHAR *dst_dir, DWORD dst_len,
374 const WCHAR *filename )
376 WCHAR *src_file, *dst_file;
377 DWORD len = strlenW( filename );
378 HRESULT hr = S_OK;
380 if (!(src_file = HeapAlloc( GetProcessHeap(), 0, (src_len + len + 1) * sizeof(WCHAR) )))
381 return E_OUTOFMEMORY;
382 memcpy( src_file, src_dir, src_len * sizeof(WCHAR) );
383 strcpyW( src_file + src_len, filename );
385 if (!(dst_file = HeapAlloc( GetProcessHeap(), 0, (dst_len + len + 1) * sizeof(WCHAR) )))
387 HeapFree( GetProcessHeap(), 0, src_file );
388 return E_OUTOFMEMORY;
390 memcpy( dst_file, dst_dir, dst_len * sizeof(WCHAR) );
391 strcpyW( dst_file + dst_len, filename );
393 if (!CopyFileW( src_file, dst_file, FALSE )) hr = HRESULT_FROM_WIN32( GetLastError() );
394 HeapFree( GetProcessHeap(), 0, src_file );
395 HeapFree( GetProcessHeap(), 0, dst_file );
396 return hr;
399 static HRESULT WINAPI IAssemblyCacheImpl_InstallAssembly(IAssemblyCache *iface,
400 DWORD dwFlags,
401 LPCWSTR pszManifestFilePath,
402 LPCFUSION_INSTALL_REFERENCE pRefData)
404 static const WCHAR format[] =
405 {'%','s','\\','%','s','\\','%','s','_','_','%','s','\\',0};
406 static const WCHAR format_v40[] =
407 {'%','s','\\','%','s','\\','v','4','.','0','_','%','s','_','_','%','s','\\',0};
408 static const WCHAR ext_exe[] = {'.','e','x','e',0};
409 static const WCHAR ext_dll[] = {'.','d','l','l',0};
410 IAssemblyCacheImpl *cache = impl_from_IAssemblyCache(iface);
411 ASSEMBLY *assembly;
412 const WCHAR *extension, *filename, *src_dir;
413 WCHAR *name = NULL, *token = NULL, *version = NULL, *asmpath = NULL;
414 WCHAR asmdir[MAX_PATH], *p, **external_files = NULL, *dst_dir = NULL;
415 PEKIND architecture;
416 char *clr_version;
417 DWORD i, count = 0, src_len, dst_len = sizeof(format_v40)/sizeof(format_v40[0]);
418 HRESULT hr;
420 TRACE("(%p, %d, %s, %p)\n", iface, dwFlags,
421 debugstr_w(pszManifestFilePath), pRefData);
423 if (!pszManifestFilePath || !*pszManifestFilePath)
424 return E_INVALIDARG;
426 if (!(extension = strrchrW(pszManifestFilePath, '.')))
427 return HRESULT_FROM_WIN32(ERROR_INVALID_NAME);
429 if (lstrcmpiW(extension, ext_exe) && lstrcmpiW(extension, ext_dll))
430 return HRESULT_FROM_WIN32(ERROR_INVALID_NAME);
432 if (GetFileAttributesW(pszManifestFilePath) == INVALID_FILE_ATTRIBUTES)
433 return HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
435 hr = assembly_create(&assembly, pszManifestFilePath);
436 if (FAILED(hr))
438 hr = COR_E_ASSEMBLYEXPECTED;
439 goto done;
442 hr = assembly_get_name(assembly, &name);
443 if (FAILED(hr))
444 goto done;
446 hr = assembly_get_pubkey_token(assembly, &token);
447 if (FAILED(hr))
448 goto done;
450 hr = assembly_get_version(assembly, &version);
451 if (FAILED(hr))
452 goto done;
454 hr = assembly_get_runtime_version(assembly, &clr_version);
455 if (FAILED(hr))
456 goto done;
458 hr = assembly_get_external_files(assembly, &external_files, &count);
459 if (FAILED(hr))
460 goto done;
462 cache_lock( cache );
464 architecture = assembly_get_architecture(assembly);
465 get_assembly_directory(asmdir, MAX_PATH, clr_version, architecture);
467 dst_len += strlenW(asmdir) + strlenW(name) + strlenW(version) + strlenW(token);
468 if (!(dst_dir = HeapAlloc(GetProcessHeap(), 0, dst_len * sizeof(WCHAR))))
470 hr = E_OUTOFMEMORY;
471 goto done;
473 if (!strcmp(clr_version, "v4.0.30319"))
474 dst_len = sprintfW(dst_dir, format_v40, asmdir, name, version, token);
475 else
476 dst_len = sprintfW(dst_dir, format, asmdir, name, version, token);
478 create_full_path(dst_dir);
480 hr = assembly_get_path(assembly, &asmpath);
481 if (FAILED(hr))
482 goto done;
484 if ((p = strrchrW(asmpath, '\\')))
486 filename = p + 1;
487 src_dir = asmpath;
488 src_len = filename - asmpath;
490 else
492 filename = asmpath;
493 src_dir = NULL;
494 src_len = 0;
496 hr = copy_file(src_dir, src_len, dst_dir, dst_len, filename);
497 if (FAILED(hr))
498 goto done;
500 for (i = 0; i < count; i++)
502 hr = copy_file(src_dir, src_len, dst_dir, dst_len, external_files[i]);
503 if (FAILED(hr))
504 break;
507 done:
508 HeapFree(GetProcessHeap(), 0, name);
509 HeapFree(GetProcessHeap(), 0, token);
510 HeapFree(GetProcessHeap(), 0, version);
511 HeapFree(GetProcessHeap(), 0, asmpath);
512 HeapFree(GetProcessHeap(), 0, dst_dir);
513 for (i = 0; i < count; i++) HeapFree(GetProcessHeap(), 0, external_files[i]);
514 HeapFree(GetProcessHeap(), 0, external_files);
515 assembly_release(assembly);
516 cache_unlock( cache );
517 return hr;
520 static const IAssemblyCacheVtbl AssemblyCacheVtbl = {
521 IAssemblyCacheImpl_QueryInterface,
522 IAssemblyCacheImpl_AddRef,
523 IAssemblyCacheImpl_Release,
524 IAssemblyCacheImpl_UninstallAssembly,
525 IAssemblyCacheImpl_QueryAssemblyInfo,
526 IAssemblyCacheImpl_CreateAssemblyCacheItem,
527 IAssemblyCacheImpl_CreateAssemblyScavenger,
528 IAssemblyCacheImpl_InstallAssembly
531 /******************************************************************
532 * CreateAssemblyCache (FUSION.@)
534 HRESULT WINAPI CreateAssemblyCache(IAssemblyCache **ppAsmCache, DWORD dwReserved)
536 IAssemblyCacheImpl *cache;
538 TRACE("(%p, %d)\n", ppAsmCache, dwReserved);
540 if (!ppAsmCache)
541 return E_INVALIDARG;
543 *ppAsmCache = NULL;
545 cache = HeapAlloc(GetProcessHeap(), 0, sizeof(IAssemblyCacheImpl));
546 if (!cache)
547 return E_OUTOFMEMORY;
549 cache->IAssemblyCache_iface.lpVtbl = &AssemblyCacheVtbl;
550 cache->ref = 1;
551 cache->lock = CreateMutexW( NULL, FALSE, cache_mutex_nameW );
552 if (!cache->lock)
554 HeapFree( GetProcessHeap(), 0, cache );
555 return HRESULT_FROM_WIN32( GetLastError() );
557 *ppAsmCache = &cache->IAssemblyCache_iface;
558 return S_OK;
561 /* IAssemblyCacheItem */
563 typedef struct {
564 IAssemblyCacheItem IAssemblyCacheItem_iface;
566 LONG ref;
567 } IAssemblyCacheItemImpl;
569 static inline IAssemblyCacheItemImpl *impl_from_IAssemblyCacheItem(IAssemblyCacheItem *iface)
571 return CONTAINING_RECORD(iface, IAssemblyCacheItemImpl, IAssemblyCacheItem_iface);
574 static HRESULT WINAPI IAssemblyCacheItemImpl_QueryInterface(IAssemblyCacheItem *iface,
575 REFIID riid, LPVOID *ppobj)
577 IAssemblyCacheItemImpl *This = impl_from_IAssemblyCacheItem(iface);
579 TRACE("(%p, %s, %p)\n", This, debugstr_guid(riid), ppobj);
581 *ppobj = NULL;
583 if (IsEqualIID(riid, &IID_IUnknown) ||
584 IsEqualIID(riid, &IID_IAssemblyCacheItem))
586 IAssemblyCacheItem_AddRef(iface);
587 *ppobj = This;
588 return S_OK;
591 WARN("(%p, %s, %p): not found\n", This, debugstr_guid(riid), ppobj);
592 return E_NOINTERFACE;
595 static ULONG WINAPI IAssemblyCacheItemImpl_AddRef(IAssemblyCacheItem *iface)
597 IAssemblyCacheItemImpl *This = impl_from_IAssemblyCacheItem(iface);
598 ULONG refCount = InterlockedIncrement(&This->ref);
600 TRACE("(%p)->(ref before = %u)\n", This, refCount - 1);
602 return refCount;
605 static ULONG WINAPI IAssemblyCacheItemImpl_Release(IAssemblyCacheItem *iface)
607 IAssemblyCacheItemImpl *This = impl_from_IAssemblyCacheItem(iface);
608 ULONG refCount = InterlockedDecrement(&This->ref);
610 TRACE("(%p)->(ref before = %u)\n", This, refCount + 1);
612 if (!refCount)
613 HeapFree(GetProcessHeap(), 0, This);
615 return refCount;
618 static HRESULT WINAPI IAssemblyCacheItemImpl_CreateStream(IAssemblyCacheItem *iface,
619 DWORD dwFlags,
620 LPCWSTR pszStreamName,
621 DWORD dwFormat,
622 DWORD dwFormatFlags,
623 IStream **ppIStream,
624 ULARGE_INTEGER *puliMaxSize)
626 FIXME("(%p, %d, %s, %d, %d, %p, %p) stub!\n", iface, dwFlags,
627 debugstr_w(pszStreamName), dwFormat, dwFormatFlags, ppIStream, puliMaxSize);
629 return E_NOTIMPL;
632 static HRESULT WINAPI IAssemblyCacheItemImpl_Commit(IAssemblyCacheItem *iface,
633 DWORD dwFlags,
634 ULONG *pulDisposition)
636 FIXME("(%p, %d, %p) stub!\n", iface, dwFlags, pulDisposition);
637 return E_NOTIMPL;
640 static HRESULT WINAPI IAssemblyCacheItemImpl_AbortItem(IAssemblyCacheItem *iface)
642 FIXME("(%p) stub!\n", iface);
643 return E_NOTIMPL;
646 static const IAssemblyCacheItemVtbl AssemblyCacheItemVtbl = {
647 IAssemblyCacheItemImpl_QueryInterface,
648 IAssemblyCacheItemImpl_AddRef,
649 IAssemblyCacheItemImpl_Release,
650 IAssemblyCacheItemImpl_CreateStream,
651 IAssemblyCacheItemImpl_Commit,
652 IAssemblyCacheItemImpl_AbortItem