metahost.c 7.21 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
/*
 * Copyright 2010 Vincent Povirk
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
 */

#define COBJMACROS

#include <stdarg.h>

#include "windef.h"
#include "ole2.h"

26
#include "corerror.h"
27
#include "ocidl.h"
28 29 30 31
#include "initguid.h"
#include "metahost.h"
#include "wine/test.h"

32 33 34 35 36 37
#if !defined(__i386__) && !defined(__x86_64__)
static int has_mono = 0;
#else
static int has_mono = 1;
#endif

38 39 40 41 42 43
static HMODULE hmscoree;

static HRESULT (WINAPI *pCLRCreateInstance)(REFCLSID clsid, REFIID riid, LPVOID *ppInterface);

static ICLRMetaHost *metahost;

44 45 46 47
static const WCHAR v4_0[] = {'v','4','.','0','.','3','0','3','1','9',0};

static DWORD expect_runtime_tid;

48
static BOOL init_pointers(void)
49 50 51 52 53 54 55 56 57 58 59 60 61
{
    HRESULT hr = E_FAIL;

    hmscoree = LoadLibraryA("mscoree.dll");

    if (hmscoree)
        pCLRCreateInstance = (void *)GetProcAddress(hmscoree, "CLRCreateInstance");

    if (pCLRCreateInstance)
        hr = pCLRCreateInstance(&CLSID_CLRMetaHost, &IID_ICLRMetaHost, (void**)&metahost);

    if (FAILED(hr))
    {
62
        win_skip(".NET 4 is not installed\n");
63 64 65 66 67 68 69
        FreeLibrary(hmscoree);
        return FALSE;
    }

    return TRUE;
}

70
static void cleanup(void)
71 72 73 74 75 76
{
    ICLRMetaHost_Release(metahost);

    FreeLibrary(hmscoree);
}

77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
static WCHAR *strrchrW( WCHAR *str, WCHAR ch )
{
    WCHAR *ret = NULL;
    do { if (*str == ch) ret = str; } while (*str++);
    return ret;
}

static void test_getruntime(WCHAR *version)
{
    static const WCHAR dotzero[] = {'.','0',0};
    WCHAR *dot;
    HRESULT hr;
    ICLRRuntimeInfo *info;
    DWORD count;
    WCHAR buf[MAX_PATH];

    hr = ICLRMetaHost_GetRuntime(metahost, NULL, &IID_ICLRRuntimeInfo, (void**)&info);
    ok(hr == E_POINTER, "GetVersion failed, hr=%x\n", hr);

    hr = ICLRMetaHost_GetRuntime(metahost, version, &IID_ICLRRuntimeInfo, (void**)&info);
    ok(hr == S_OK, "GetVersion failed, hr=%x\n", hr);
    if (hr != S_OK) return;

    count = MAX_PATH;
    hr = ICLRRuntimeInfo_GetVersionString(info, buf, &count);
    ok(hr == S_OK, "GetVersionString returned %x\n", hr);
    ok(count == lstrlenW(buf)+1, "GetVersionString returned count %u but string of length %u\n", count, lstrlenW(buf)+1);
    ok(lstrcmpW(buf, version) == 0, "got unexpected version %s\n", wine_dbgstr_w(buf));

    ICLRRuntimeInfo_Release(info);

    /* Versions must match exactly. */
    dot = strrchrW(version, '.');
    lstrcpyW(dot, dotzero);
    hr = ICLRMetaHost_GetRuntime(metahost, version, &IID_ICLRRuntimeInfo, (void**)&info);
    ok(hr == CLR_E_SHIM_RUNTIME, "GetVersion failed, hr=%x\n", hr);
}

115
static void test_enumruntimes(void)
116 117 118 119 120 121 122 123 124
{
    IEnumUnknown *runtime_enum;
    IUnknown *unk;
    DWORD count;
    ICLRRuntimeInfo *runtime_info;
    HRESULT hr;
    WCHAR buf[MAX_PATH];

    hr = ICLRMetaHost_EnumerateInstalledRuntimes(metahost, &runtime_enum);
125
    ok(hr == S_OK, "EnumerateInstalledRuntimes returned %x\n", hr);
126 127 128 129 130 131 132 133 134
    if (FAILED(hr)) return;

    while ((hr = IEnumUnknown_Next(runtime_enum, 1, &unk, &count)) == S_OK)
    {
        hr = IUnknown_QueryInterface(unk, &IID_ICLRRuntimeInfo, (void**)&runtime_info);
        ok(hr == S_OK, "QueryInterface returned %x\n", hr);

        count = 1;
        hr = ICLRRuntimeInfo_GetVersionString(runtime_info, buf, &count);
135
        ok(hr == E_NOT_SUFFICIENT_BUFFER, "GetVersionString returned %x\n", hr);
136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151
        ok(count > 1, "GetVersionString returned count %u\n", count);

        count = 0xdeadbeef;
        hr = ICLRRuntimeInfo_GetVersionString(runtime_info, NULL, &count);
        ok(hr == S_OK, "GetVersionString returned %x\n", hr);
        ok(count > 1 && count != 0xdeadbeef, "GetVersionString returned count %u\n", count);

        count = MAX_PATH;
        hr = ICLRRuntimeInfo_GetVersionString(runtime_info, buf, &count);
        ok(hr == S_OK, "GetVersionString returned %x\n", hr);
        ok(count > 1, "GetVersionString returned count %u\n", count);

        trace("runtime found: %s\n", wine_dbgstr_w(buf));

        ICLRRuntimeInfo_Release(runtime_info);
        IUnknown_Release(unk);
152 153

        test_getruntime(buf);
154 155 156 157 158 159 160
    }

    ok(hr == S_FALSE, "IEnumUnknown_Next returned %x\n", hr);

    IEnumUnknown_Release(runtime_enum);
}

161 162 163 164 165 166
static void WINAPI notification_dummy_callback(ICLRRuntimeInfo *pRuntimeInfo, CallbackThreadSetFnPtr pfnCallbackThreadSet,
    CallbackThreadUnsetFnPtr pfnCallbackThreadUnset)
{
    ok(0, "unexpected call\n");
}

167 168 169
static void WINAPI notification_callback(ICLRRuntimeInfo *pRuntimeInfo, CallbackThreadSetFnPtr pfnCallbackThreadSet,
    CallbackThreadUnsetFnPtr pfnCallbackThreadUnset)
{
170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191
    HRESULT hr;
    WCHAR buf[20];
    DWORD buf_size = 20;

    ok(expect_runtime_tid != 0, "unexpected call\n");

    if (expect_runtime_tid != 0)
    {
        ok(GetCurrentThreadId() == expect_runtime_tid,
            "expected call on thread %04x, got thread %04x\n", expect_runtime_tid, GetCurrentThreadId());
        expect_runtime_tid = 0;
    }

    hr = ICLRRuntimeInfo_GetVersionString(pRuntimeInfo, buf, &buf_size);
    ok(hr == S_OK, "GetVersion returned %x\n", hr);
    ok(lstrcmpW(buf, v4_0) == 0, "GetVersion returned %s\n", wine_dbgstr_w(buf));

    hr = pfnCallbackThreadSet();
    ok(hr == S_OK, "pfnCallbackThreadSet returned %x\n", hr);

    hr = pfnCallbackThreadUnset();
    ok(hr == S_OK, "pfnCallbackThreadUnset returned %x\n", hr);
192 193 194 195 196 197 198
}

static void test_notification(void)
{
    HRESULT hr;

    hr = ICLRMetaHost_RequestRuntimeLoadedNotification(metahost, NULL);
199
    ok(hr == E_POINTER, "RequestRuntimeLoadedNotification returned %x\n", hr);
200 201

    hr = ICLRMetaHost_RequestRuntimeLoadedNotification(metahost,notification_callback);
202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219
    ok(hr == S_OK, "RequestRuntimeLoadedNotification failed, hr=%x\n", hr);

    hr = ICLRMetaHost_RequestRuntimeLoadedNotification(metahost,notification_dummy_callback);
    ok(hr == HOST_E_INVALIDOPERATION, "RequestRuntimeLoadedNotification returned %x\n", hr);
}

static void test_notification_cb(void)
{
    HRESULT hr;
    ICLRRuntimeInfo *info;
    ICLRRuntimeHost *host;

    hr = ICLRMetaHost_GetRuntime(metahost, v4_0, &IID_ICLRRuntimeInfo, (void**)&info);
    ok(hr == S_OK, "GetRuntime returned %x\n", hr);

    expect_runtime_tid = GetCurrentThreadId();
    hr = ICLRRuntimeInfo_GetInterface(info, &CLSID_CLRRuntimeHost, &IID_ICLRRuntimeHost, (void**)&host);

220 221 222 223 224
    todo_wine_if(!has_mono) ok(hr == S_OK, "GetInterface returned %x\n", hr);
    todo_wine if(!has_mono) ok(expect_runtime_tid == 0, "notification_callback was not called\n");

    if(has_mono)
        ICLRRuntimeHost_Release(host);
225 226

    ICLRRuntimeInfo_Release(info);
227 228
}

229 230 231 232 233
START_TEST(metahost)
{
    if (!init_pointers())
        return;

234
    test_notification();
235 236
    test_enumruntimes();
    test_notification_cb();
237

238 239
    cleanup();
}