dlls/fusion/tests/fusion.c

   1 /*
   2  * Copyright 2008 James Hawkins
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include <windows.h>
  20 #include <fusion.h>
  21
  22 #include "wine/test.h"
  23
  24 static HMODULE hmscoree;
  25
  26 static HRESULT (WINAPI *pGetCachePath)(ASM_CACHE_FLAGS dwCacheFlags,
  27                                        LPWSTR pwzCachePath, PDWORD pcchPath);
  28 static HRESULT (WINAPI *pLoadLibraryShim)(LPCWSTR szDllName, LPCWSTR szVersion,
  29                                           LPVOID pvReserved, HMODULE *phModDll);
  30 static HRESULT (WINAPI *pGetCORVersion)(LPWSTR pbuffer, DWORD cchBuffer,
  31                                         DWORD *dwLength);
  32
  33 static BOOL init_functionpointers(void)
  34 {
  35     HRESULT hr;
  36     HMODULE hfusion;
  37
  38     static const WCHAR szFusion[] = {'f','u','s','i','o','n','.','d','l','l',0};
  39
  40     hmscoree = LoadLibraryA("mscoree.dll");
  41     if (!hmscoree)
  42     {
  43         win_skip("mscoree.dll not available\n");
  44         return FALSE;
  45     }
  46
  47     pLoadLibraryShim = (void *)GetProcAddress(hmscoree, "LoadLibraryShim");
  48     if (!pLoadLibraryShim)
  49     {
  50         win_skip("LoadLibraryShim not available\n");
  51         FreeLibrary(hmscoree);
  52         return FALSE;
  53     }
  54
  55     pGetCORVersion = (void *)GetProcAddress(hmscoree, "GetCORVersion");
  56
  57     hr = pLoadLibraryShim(szFusion, NULL, NULL, &hfusion);
  58     if (FAILED(hr))
  59     {
  60         win_skip("fusion.dll not available\n");
  61         FreeLibrary(hmscoree);
  62         return FALSE;
  63     }
  64
  65     pGetCachePath = (void *)GetProcAddress(hfusion, "GetCachePath");
  66     return TRUE;
  67 }
  68
  69 static void test_GetCachePath(void)
  70 {
  71     CHAR windirA[MAX_PATH];
  72     WCHAR windir[MAX_PATH];
  73     WCHAR cachepath[MAX_PATH];
  74     WCHAR version[MAX_PATH];
  75     WCHAR path[MAX_PATH];
  76     DWORD size;
  77     HRESULT hr;
  78
  79     static const WCHAR backslash[] = {'\\',0};
  80     static const WCHAR nochange[] = {'n','o','c','h','a','n','g','e',0};
  81     static const WCHAR assembly[] = {'a','s','s','e','m','b','l','y',0};
  82     static const WCHAR gac[] = {'G','A','C',0};
  83
  84     if (!pGetCachePath)
  85     {
  86         win_skip("GetCachePath not implemented\n");
  87         return;
  88     }
  89
  90     GetWindowsDirectoryA(windirA, MAX_PATH);
  91     MultiByteToWideChar(CP_ACP, 0, windirA, -1, windir, MAX_PATH);
  92     lstrcpyW(cachepath, windir);
  93     lstrcatW(cachepath, backslash);
  94     lstrcatW(cachepath, assembly);
  95     lstrcatW(cachepath, backslash);
  96     lstrcatW(cachepath, gac);
  97
  98     /* NULL pwzCachePath, pcchPath is 0 */
  99     size = 0;
 100     hr = pGetCachePath(ASM_CACHE_GAC, NULL, &size);
 101     ok(hr == E_NOT_SUFFICIENT_BUFFER, "Expected E_NOT_SUFFICIENT_BUFFER, got %08x\n", hr);
 102     ok(size == lstrlenW(cachepath) + 1,
 103        "Expected %d, got %d\n", lstrlenW(cachepath) + 1, size);
 104
 105     /* NULL pwszCachePath, pcchPath is MAX_PATH */
 106     size = MAX_PATH;
 107     hr = pGetCachePath(ASM_CACHE_GAC, NULL, &size);
 108     ok(hr == E_NOT_SUFFICIENT_BUFFER, "Expected E_NOT_SUFFICIENT_BUFFER, got %08x\n", hr);
 109     ok(size == lstrlenW(cachepath) + 1,
 110        "Expected %d, got %d\n", lstrlenW(cachepath) + 1, size);
 111
 112     /* both pwszCachePath and pcchPath NULL */
 113     hr = pGetCachePath(ASM_CACHE_GAC, NULL, NULL);
 114     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
 115
 116     /* NULL pcchPath */
 117     lstrcpyW(path, nochange);
 118     hr = pGetCachePath(ASM_CACHE_GAC, path, NULL);
 119     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
 120     ok( !lstrcmpW( nochange, path ), "Expected %s,  got %s\n", wine_dbgstr_w(nochange), wine_dbgstr_w(path));
 121
 122     /* get the cache path */
 123     lstrcpyW(path, nochange);
 124     size = MAX_PATH;
 125     hr = pGetCachePath(ASM_CACHE_GAC, path, &size);
 126     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 127     ok( !lstrcmpW( cachepath, path ), "Expected %s,  got %s\n", wine_dbgstr_w(cachepath), wine_dbgstr_w(path));
 128
 129     /* pcchPath has no room for NULL terminator */
 130     lstrcpyW(path, nochange);
 131     size = lstrlenW(cachepath);
 132     hr = pGetCachePath(ASM_CACHE_GAC, path, &size);
 133     ok(hr == E_NOT_SUFFICIENT_BUFFER, "Expected E_NOT_SUFFICIENT_BUFFER, got %08x\n", hr);
 134     ok( !lstrcmpW( nochange, path ), "Expected %s,  got %s\n", wine_dbgstr_w(nochange), wine_dbgstr_w(path));
 135
 136     lstrcpyW(cachepath, windir);
 137     lstrcatW(cachepath, backslash);
 138     lstrcatW(cachepath, assembly);
 139
 140     /* ASM_CACHE_ROOT */
 141     lstrcpyW(path, nochange);
 142     size = MAX_PATH;
 143     hr = pGetCachePath(ASM_CACHE_ROOT, path, &size);
 144     ok(hr == S_OK ||
 145        broken(hr == E_INVALIDARG), /* .NET 1.1 */
 146        "Expected S_OK, got %08x\n", hr);
 147     if (hr == S_OK)
 148         ok( !lstrcmpW( cachepath, path ), "Expected %s,  got %s\n", wine_dbgstr_w(cachepath), wine_dbgstr_w(path));
 149
 150     if (pGetCORVersion)
 151     {
 152         CHAR versionA[MAX_PATH];
 153         CHAR cachepathA[MAX_PATH];
 154         CHAR nativeimgA[MAX_PATH];
 155         CHAR zapfmtA[MAX_PATH];
 156
 157         if (hr == S_OK)
 158         {
 159             lstrcpyA(nativeimgA, "NativeImages_");
 160 #ifdef _WIN64
 161             lstrcpyA(zapfmtA, "%s\\%s\\%s%s_64");
 162 #else
 163             lstrcpyA(zapfmtA, "%s\\%s\\%s%s_32");
 164 #endif
 165         }
 166         else
 167         {
 168             lstrcpyA(nativeimgA, "NativeImages1_");
 169             lstrcpyA(zapfmtA, "%s\\%s\\%s%s");
 170         }
 171
 172         pGetCORVersion(version, MAX_PATH, &size);
 173         WideCharToMultiByte(CP_ACP, 0, version, -1, versionA, MAX_PATH, 0, 0);
 174
 175         wsprintfA(cachepathA, zapfmtA, windirA, "assembly", nativeimgA, versionA);
 176         MultiByteToWideChar(CP_ACP, 0, cachepathA, -1, cachepath, MAX_PATH);
 177
 178         /* ASM_CACHE_ZAP */
 179         lstrcpyW(path, nochange);
 180         size = MAX_PATH;
 181         hr = pGetCachePath(ASM_CACHE_ZAP, path, &size);
 182         ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 183         ok( !lstrcmpW( cachepath, path ), "Expected %s,  got %s\n", wine_dbgstr_w(cachepath), wine_dbgstr_w(path));
 184     }
 185
 186     /* two flags at once */
 187     lstrcpyW(path, nochange);
 188     size = MAX_PATH;
 189     hr = pGetCachePath(ASM_CACHE_GAC | ASM_CACHE_ROOT, path, &size);
 190     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
 191     ok( !lstrcmpW( nochange, path ), "Expected %s,  got %s\n", wine_dbgstr_w(nochange), wine_dbgstr_w(path));
 192 }
 193
 194 START_TEST(fusion)
 195 {
 196     if (!init_functionpointers())
 197         return;
 198
 199     test_GetCachePath();
 200
 201     FreeLibrary(hmscoree);
 202 }