metahost.c

Go to the documentation of this file.

/*
 * 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"
 
#include "corerror.h"
#include "ocidl.h"
#include "initguid.h"
#include "metahost.h"
#include "wine/test.h"
 
#if !defined(__i386__) && !defined(__x86_64__)
static int has_mono = 0;
#else
static int has_mono = 1;
#endif
 
static HMODULE hmscoree;
 
static HRESULT (WINAPI *pCLRCreateInstance)(REFCLSID clsid, REFIID riid, LPVOID *ppInterface);
 
static ICLRMetaHost *metahost;
 
static const WCHAR v4_0[] = {'v','4','.','0','.','3','0','3','1','9',0};
 
static DWORD expect_runtime_tid;
 
static BOOL init_pointers(void)
{
    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))
    {
        win_skip(".NET 4 is not installed\n");
        FreeLibrary(hmscoree);
        return FALSE;
    }
 
    return TRUE;
}
 
static void cleanup(void)
{
    ICLRMetaHost_Release(metahost);
 
    FreeLibrary(hmscoree);
}
 
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=%lx\n", hr);
 
    hr = ICLRMetaHost_GetRuntime(metahost, version, &IID_ICLRRuntimeInfo, (void**)&info);
    ok(hr == S_OK, "GetVersion failed, hr=%lx\n", hr);
    if (hr != S_OK) return;
 
    count = MAX_PATH;
    hr = ICLRRuntimeInfo_GetVersionString(info, buf, &count);
    ok(hr == S_OK, "GetVersionString returned %lx\n", hr);
    ok(count == lstrlenW(buf)+1, "GetVersionString returned count %lu 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 = wcsrchr(version, '.');
    lstrcpyW(dot, dotzero);
    hr = ICLRMetaHost_GetRuntime(metahost, version, &IID_ICLRRuntimeInfo, (void**)&info);
    ok(hr == CLR_E_SHIM_RUNTIME, "GetVersion failed, hr=%lx\n", hr);
}
 
static void test_enumruntimes(void)
{
    IEnumUnknown *runtime_enum;
    IUnknown *unk;
    DWORD count;
    ICLRRuntimeInfo *runtime_info;
    HRESULT hr;
    WCHAR buf[MAX_PATH];
 
    hr = ICLRMetaHost_EnumerateInstalledRuntimes(metahost, &runtime_enum);
    ok(hr == S_OK, "EnumerateInstalledRuntimes returned %lx\n", hr);
    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 %lx\n", hr);
 
        count = 1;
        hr = ICLRRuntimeInfo_GetVersionString(runtime_info, buf, &count);
        ok(hr == E_NOT_SUFFICIENT_BUFFER, "GetVersionString returned %lx\n", hr);
        ok(count > 1, "GetVersionString returned count %lu\n", count);
 
        count = 0xdeadbeef;
        hr = ICLRRuntimeInfo_GetVersionString(runtime_info, NULL, &count);
        ok(hr == S_OK, "GetVersionString returned %lx\n", hr);
        ok(count > 1 && count != 0xdeadbeef, "GetVersionString returned count %lu\n", count);
 
        count = MAX_PATH;
        hr = ICLRRuntimeInfo_GetVersionString(runtime_info, buf, &count);
        ok(hr == S_OK, "GetVersionString returned %lx\n", hr);
        ok(count > 1, "GetVersionString returned count %lu\n", count);
 
        trace("runtime found: %s\n", wine_dbgstr_w(buf));
 
        ICLRRuntimeInfo_Release(runtime_info);
        IUnknown_Release(unk);
 
        test_getruntime(buf);
    }
 
    ok(hr == S_FALSE, "IEnumUnknown_Next returned %lx\n", hr);
 
    IEnumUnknown_Release(runtime_enum);
}
 
static void WINAPI notification_dummy_callback(ICLRRuntimeInfo *pRuntimeInfo, CallbackThreadSetFnPtr pfnCallbackThreadSet,
    CallbackThreadUnsetFnPtr pfnCallbackThreadUnset)
{
    ok(0, "unexpected call\n");
}
 
static void WINAPI notification_callback(ICLRRuntimeInfo *pRuntimeInfo, CallbackThreadSetFnPtr pfnCallbackThreadSet,
    CallbackThreadUnsetFnPtr pfnCallbackThreadUnset)
{
    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 %04lx, got thread %04lx\n", expect_runtime_tid, GetCurrentThreadId());
        expect_runtime_tid = 0;
    }
 
    hr = ICLRRuntimeInfo_GetVersionString(pRuntimeInfo, buf, &buf_size);
    ok(hr == S_OK, "GetVersion returned %lx\n", hr);
    ok(lstrcmpW(buf, v4_0) == 0, "GetVersion returned %s\n", wine_dbgstr_w(buf));
 
    hr = pfnCallbackThreadSet();
    ok(hr == S_OK, "pfnCallbackThreadSet returned %lx\n", hr);
 
    hr = pfnCallbackThreadUnset();
    ok(hr == S_OK, "pfnCallbackThreadUnset returned %lx\n", hr);
}
 
static void test_notification(void)
{
    HRESULT hr;
 
    hr = ICLRMetaHost_RequestRuntimeLoadedNotification(metahost, NULL);
    ok(hr == E_POINTER, "RequestRuntimeLoadedNotification returned %lx\n", hr);
 
    hr = ICLRMetaHost_RequestRuntimeLoadedNotification(metahost,notification_callback);
    ok(hr == S_OK, "RequestRuntimeLoadedNotification failed, hr=%lx\n", hr);
 
    hr = ICLRMetaHost_RequestRuntimeLoadedNotification(metahost,notification_dummy_callback);
    ok(hr == HOST_E_INVALIDOPERATION, "RequestRuntimeLoadedNotification returned %lx\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 %lx\n", hr);
 
    expect_runtime_tid = GetCurrentThreadId();
    hr = ICLRRuntimeInfo_GetInterface(info, &CLSID_CLRRuntimeHost, &IID_ICLRRuntimeHost, (void**)&host);
#ifdef __REACTOS__
    if (hr != S_OK)
    {
        // skip test if mono is not installed
        win_skip("mono runtime is not installed\n");
        return;
    }
#endif
 
    todo_wine_if(!has_mono) ok(hr == S_OK, "GetInterface returned %lx\n", hr);
    todo_wine if(!has_mono) ok(expect_runtime_tid == 0, "notification_callback was not called\n");
 
    if(has_mono)
        ICLRRuntimeHost_Release(host);
 
    ICLRRuntimeInfo_Release(info);
}
 
START_TEST(metahost)
{
    if (!init_pointers())
        return;
 
    test_notification();
    test_enumruntimes();
    test_notification_cb();
 
    cleanup();
}