1 /*
2 * Copyright 2010 Vincent Povirk
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 #define COBJMACROS
20
21 #include <stdarg.h>
22
23 #include "windef.h"
24 #include "ole2.h"
25
26 #include "corerror.h"
27 #include "ocidl.h"
28 #include "initguid.h"
29 #include "metahost.h"
30 #include "wine/test.h"
31
32 static HMODULE hmscoree;
33
34 static HRESULT (WINAPI *pCLRCreateInstance)(REFCLSID clsid, REFIID riid, LPVOID *ppInterface);
35
36 static ICLRMetaHost *metahost;
37
init_pointers(void)38 static BOOL init_pointers(void)
39 {
40 HRESULT hr = E_FAIL;
41
42 hmscoree = LoadLibraryA("mscoree.dll");
43
44 if (hmscoree)
45 pCLRCreateInstance = (void *)GetProcAddress(hmscoree, "CLRCreateInstance");
46
47 if (pCLRCreateInstance)
48 hr = pCLRCreateInstance(&CLSID_CLRMetaHost, &IID_ICLRMetaHost, (void**)&metahost);
49
50 if (FAILED(hr))
51 {
52 win_skip(".NET 4 is not installed\n");
53 FreeLibrary(hmscoree);
54 return FALSE;
55 }
56
57 return TRUE;
58 }
59
cleanup(void)60 static void cleanup(void)
61 {
62 ICLRMetaHost_Release(metahost);
63
64 FreeLibrary(hmscoree);
65 }
66
test_enumruntimes(void)67 static void test_enumruntimes(void)
68 {
69 IEnumUnknown *runtime_enum;
70 IUnknown *unk;
71 DWORD count;
72 ICLRRuntimeInfo *runtime_info;
73 HRESULT hr;
74 WCHAR buf[MAX_PATH];
75
76 hr = ICLRMetaHost_EnumerateInstalledRuntimes(metahost, &runtime_enum);
77 ok(hr == S_OK, "EnumerateInstalledRuntimes returned %x\n", hr);
78 if (FAILED(hr)) return;
79
80 while ((hr = IEnumUnknown_Next(runtime_enum, 1, &unk, &count)) == S_OK)
81 {
82 hr = IUnknown_QueryInterface(unk, &IID_ICLRRuntimeInfo, (void**)&runtime_info);
83 ok(hr == S_OK, "QueryInterface returned %x\n", hr);
84
85 count = 1;
86 hr = ICLRRuntimeInfo_GetVersionString(runtime_info, buf, &count);
87 ok(hr == HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER), "GetVersionString returned %x\n", hr);
88 ok(count > 1, "GetVersionString returned count %u\n", count);
89
90 count = 0xdeadbeef;
91 hr = ICLRRuntimeInfo_GetVersionString(runtime_info, NULL, &count);
92 ok(hr == S_OK, "GetVersionString returned %x\n", hr);
93 ok(count > 1 && count != 0xdeadbeef, "GetVersionString returned count %u\n", count);
94
95 count = MAX_PATH;
96 hr = ICLRRuntimeInfo_GetVersionString(runtime_info, buf, &count);
97 ok(hr == S_OK, "GetVersionString returned %x\n", hr);
98 ok(count > 1, "GetVersionString returned count %u\n", count);
99
100 trace("runtime found: %s\n", wine_dbgstr_w(buf));
101
102 ICLRRuntimeInfo_Release(runtime_info);
103
104 IUnknown_Release(unk);
105 }
106
107 ok(hr == S_FALSE, "IEnumUnknown_Next returned %x\n", hr);
108
109 IEnumUnknown_Release(runtime_enum);
110 }
111
test_getruntime(void)112 static void test_getruntime(void)
113 {
114 static const WCHAR twodotzero[] = {'v','2','.','0','.','5','0','7','2','7',0};
115 static const WCHAR twodotzerodotzero[] = {'v','2','.','0','.','0',0};
116 HRESULT hr;
117 ICLRRuntimeInfo *info;
118 DWORD count;
119 WCHAR buf[MAX_PATH];
120
121 hr = ICLRMetaHost_GetRuntime(metahost, NULL, &IID_ICLRRuntimeInfo, (void**)&info);
122 ok(hr == E_POINTER, "GetVersion failed, hr=%x\n", hr);
123
124 hr = ICLRMetaHost_GetRuntime(metahost, twodotzero, &IID_ICLRRuntimeInfo, (void**)&info);
125 if (hr == CLR_E_SHIM_RUNTIME)
126 /* FIXME: Get Mono properly packaged so we can fail here. */
127 todo_wine ok(hr == S_OK, "GetVersion failed, hr=%x\n", hr);
128 else
129 ok(hr == S_OK, "GetVersion failed, hr=%x\n", hr);
130 if (hr != S_OK) return;
131
132 count = MAX_PATH;
133 hr = ICLRRuntimeInfo_GetVersionString(info, buf, &count);
134 ok(hr == S_OK, "GetVersionString returned %x\n", hr);
135 ok(count == lstrlenW(buf)+1, "GetVersionString returned count %u but string of length %u\n", count, lstrlenW(buf)+1);
136 ok(lstrcmpW(buf, twodotzero) == 0, "got unexpected version %s\n", wine_dbgstr_w(buf));
137
138 ICLRRuntimeInfo_Release(info);
139
140 /* Versions must match exactly. */
141 hr = ICLRMetaHost_GetRuntime(metahost, twodotzerodotzero, &IID_ICLRRuntimeInfo, (void**)&info);
142 ok(hr == CLR_E_SHIM_RUNTIME, "GetVersion failed, hr=%x\n", hr);
143 }
144
START_TEST(metahost)145 START_TEST(metahost)
146 {
147 if (!init_pointers())
148 return;
149
150 test_enumruntimes();
151
152 test_getruntime();
153
154 cleanup();
155 }
156