/* * Component Object Tests * * Copyright 2005 Robert Shearman * * 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 #define CONST_VTABLE #include #include "windef.h" #include "winbase.h" #define USE_COM_CONTEXT_DEF #include "initguid.h" #include "objbase.h" #include "shlguid.h" #include "urlmon.h" /* for CLSID_FileProtocol */ #include "ctxtcall.h" #include "wine/test.h" /* functions that are not present on all versions of Windows */ static HRESULT (WINAPI * pCoInitializeEx)(LPVOID lpReserved, DWORD dwCoInit); static HRESULT (WINAPI * pCoGetObjectContext)(REFIID riid, LPVOID *ppv); static HRESULT (WINAPI * pCoSwitchCallContext)(IUnknown *pObject, IUnknown **ppOldObject); static HRESULT (WINAPI * pCoGetTreatAsClass)(REFCLSID clsidOld, LPCLSID pClsidNew); static HRESULT (WINAPI * pCoGetContextToken)(ULONG_PTR *token); #define ok_ole_success(hr, func) ok(hr == S_OK, func " failed with error 0x%08x\n", hr) #define ok_more_than_one_lock() ok(cLocks > 0, "Number of locks should be > 0, but actually is %d\n", cLocks) #define ok_no_locks() ok(cLocks == 0, "Number of locks should be 0, but actually is %d\n", cLocks) static const CLSID CLSID_non_existent = { 0x12345678, 0x1234, 0x1234, { 0x12, 0x34, 0x56, 0x78, 0x9a, 0xbc, 0xde, 0xf0 } }; static const CLSID CLSID_StdFont = { 0x0be35203, 0x8f91, 0x11ce, { 0x9d, 0xe3, 0x00, 0xaa, 0x00, 0x4b, 0xb8, 0x51 } }; static WCHAR stdfont[] = {'S','t','d','F','o','n','t',0}; static const WCHAR wszNonExistent[] = {'N','o','n','E','x','i','s','t','e','n','t',0}; static WCHAR wszCLSID_StdFont[] = { '{','0','b','e','3','5','2','0','3','-','8','f','9','1','-','1','1','c','e','-', '9','d','e','3','-','0','0','a','a','0','0','4','b','b','8','5','1','}',0 }; static const IID IID_IWineTest = { 0x5201163f, 0x8164, 0x4fd0, {0xa1, 0xa2, 0x5d, 0x5a, 0x36, 0x54, 0xd3, 0xbd} }; /* 5201163f-8164-4fd0-a1a2-5d5a3654d3bd */ static const CLSID CLSID_WineOOPTest = { 0x5201163f, 0x8164, 0x4fd0, {0xa1, 0xa2, 0x5d, 0x5a, 0x36, 0x54, 0xd3, 0xbd} }; /* 5201163f-8164-4fd0-a1a2-5d5a3654d3bd */ static LONG cLocks; static void LockModule(void) { InterlockedIncrement(&cLocks); } static void UnlockModule(void) { InterlockedDecrement(&cLocks); } static HRESULT WINAPI Test_IClassFactory_QueryInterface( LPCLASSFACTORY iface, REFIID riid, LPVOID *ppvObj) { if (ppvObj == NULL) return E_POINTER; if (IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_IClassFactory)) { *ppvObj = iface; IClassFactory_AddRef(iface); return S_OK; } *ppvObj = NULL; return E_NOINTERFACE; } static ULONG WINAPI Test_IClassFactory_AddRef(LPCLASSFACTORY iface) { LockModule(); return 2; /* non-heap-based object */ } static ULONG WINAPI Test_IClassFactory_Release(LPCLASSFACTORY iface) { UnlockModule(); return 1; /* non-heap-based object */ } static HRESULT WINAPI Test_IClassFactory_CreateInstance( LPCLASSFACTORY iface, IUnknown *pUnkOuter, REFIID riid, LPVOID *ppvObj) { *ppvObj = NULL; if (pUnkOuter) return CLASS_E_NOAGGREGATION; return E_NOINTERFACE; } static HRESULT WINAPI Test_IClassFactory_LockServer( LPCLASSFACTORY iface, BOOL fLock) { return S_OK; } static const IClassFactoryVtbl TestClassFactory_Vtbl = { Test_IClassFactory_QueryInterface, Test_IClassFactory_AddRef, Test_IClassFactory_Release, Test_IClassFactory_CreateInstance, Test_IClassFactory_LockServer }; static IClassFactory Test_ClassFactory = { &TestClassFactory_Vtbl }; static void test_ProgIDFromCLSID(void) { LPWSTR progid; HRESULT hr = ProgIDFromCLSID(&CLSID_StdFont, &progid); ok(hr == S_OK, "ProgIDFromCLSID failed with error 0x%08x\n", hr); if (hr == S_OK) { ok(!lstrcmpiW(progid, stdfont), "Didn't get expected prog ID\n"); CoTaskMemFree(progid); } progid = (LPWSTR)0xdeadbeef; hr = ProgIDFromCLSID(&CLSID_non_existent, &progid); ok(hr == REGDB_E_CLASSNOTREG, "ProgIDFromCLSID returned %08x\n", hr); ok(progid == NULL, "ProgIDFromCLSID returns with progid %p\n", progid); hr = ProgIDFromCLSID(&CLSID_StdFont, NULL); ok(hr == E_INVALIDARG, "ProgIDFromCLSID should return E_INVALIDARG instead of 0x%08x\n", hr); } static void test_CLSIDFromProgID(void) { CLSID clsid; HRESULT hr = CLSIDFromProgID(stdfont, &clsid); ok(hr == S_OK, "CLSIDFromProgID failed with error 0x%08x\n", hr); ok(IsEqualCLSID(&clsid, &CLSID_StdFont), "clsid wasn't equal to CLSID_StdFont\n"); hr = CLSIDFromString(stdfont, &clsid); ok_ole_success(hr, "CLSIDFromString"); ok(IsEqualCLSID(&clsid, &CLSID_StdFont), "clsid wasn't equal to CLSID_StdFont\n"); /* test some failure cases */ hr = CLSIDFromProgID(wszNonExistent, NULL); ok(hr == E_INVALIDARG, "CLSIDFromProgID should have returned E_INVALIDARG instead of 0x%08x\n", hr); hr = CLSIDFromProgID(NULL, &clsid); ok(hr == E_INVALIDARG, "CLSIDFromProgID should have returned E_INVALIDARG instead of 0x%08x\n", hr); memset(&clsid, 0xcc, sizeof(clsid)); hr = CLSIDFromProgID(wszNonExistent, &clsid); ok(hr == CO_E_CLASSSTRING, "CLSIDFromProgID on nonexistent ProgID should have returned CO_E_CLASSSTRING instead of 0x%08x\n", hr); ok(IsEqualCLSID(&clsid, &CLSID_NULL), "CLSIDFromProgID should have set clsid to all-zeros on failure\n"); } static void test_CLSIDFromString(void) { CLSID clsid; WCHAR wszCLSID_Broken[50]; UINT i; HRESULT hr = CLSIDFromString(wszCLSID_StdFont, &clsid); ok_ole_success(hr, "CLSIDFromString"); ok(IsEqualCLSID(&clsid, &CLSID_StdFont), "clsid wasn't equal to CLSID_StdFont\n"); hr = CLSIDFromString(NULL, &clsid); ok_ole_success(hr, "CLSIDFromString"); ok(IsEqualCLSID(&clsid, &CLSID_NULL), "clsid wasn't equal to CLSID_NULL\n"); lstrcpyW(wszCLSID_Broken, wszCLSID_StdFont); for(i = lstrlenW(wszCLSID_StdFont); i < 49; i++) wszCLSID_Broken[i] = 'A'; wszCLSID_Broken[i] = '\0'; memset(&clsid, 0, sizeof(CLSID)); hr = CLSIDFromString(wszCLSID_Broken, &clsid); ok(hr == CO_E_CLASSSTRING, "Got %08x\n", hr); ok(IsEqualCLSID(&clsid, &CLSID_StdFont), "clsid wasn't equal to CLSID_StdFont\n"); wszCLSID_Broken[lstrlenW(wszCLSID_StdFont)-1] = 'A'; memset(&clsid, 0, sizeof(CLSID)); hr = CLSIDFromString(wszCLSID_Broken, &clsid); ok(hr == CO_E_CLASSSTRING, "Got %08x\n", hr); ok(IsEqualCLSID(&clsid, &CLSID_StdFont), "clsid wasn't equal to CLSID_StdFont\n"); wszCLSID_Broken[lstrlenW(wszCLSID_StdFont)] = '\0'; memset(&clsid, 0, sizeof(CLSID)); hr = CLSIDFromString(wszCLSID_Broken, &clsid); ok(hr == CO_E_CLASSSTRING, "Got %08x\n", hr); ok(IsEqualCLSID(&clsid, &CLSID_StdFont), "clsid wasn't equal to CLSID_StdFont\n"); wszCLSID_Broken[lstrlenW(wszCLSID_StdFont)-1] = '\0'; memset(&clsid, 0, sizeof(CLSID)); hr = CLSIDFromString(wszCLSID_Broken, &clsid); ok(hr == CO_E_CLASSSTRING, "Got %08x\n", hr); ok(IsEqualCLSID(&clsid, &CLSID_StdFont), "clsid wasn't equal to CLSID_StdFont\n"); memset(&clsid, 0xcc, sizeof(CLSID)); hr = CLSIDFromString(wszCLSID_Broken+1, &clsid); ok(hr == CO_E_CLASSSTRING, "Got %08x\n", hr); ok(IsEqualCLSID(&clsid, &CLSID_NULL), "clsid wasn't equal to CLSID_NULL\n"); wszCLSID_Broken[9] = '*'; memset(&clsid, 0xcc, sizeof(CLSID)); hr = CLSIDFromString(wszCLSID_Broken, &clsid); ok(hr == CO_E_CLASSSTRING, "Got %08x\n", hr); ok(clsid.Data1 == CLSID_StdFont.Data1, "Got %08x\n", clsid.Data1); ok(clsid.Data2 == 0xcccc, "Got %04x\n", clsid.Data2); wszCLSID_Broken[3] = '*'; memset(&clsid, 0xcc, sizeof(CLSID)); hr = CLSIDFromString(wszCLSID_Broken, &clsid); ok(hr == CO_E_CLASSSTRING, "Got %08x\n", hr); ok(clsid.Data1 == 0xb, "Got %08x\n", clsid.Data1); ok(clsid.Data2 == 0xcccc, "Got %04x\n", clsid.Data2); wszCLSID_Broken[3] = '\0'; memset(&clsid, 0xcc, sizeof(CLSID)); hr = CLSIDFromString(wszCLSID_Broken, &clsid); ok(hr == CO_E_CLASSSTRING, "Got %08x\n", hr); ok(clsid.Data1 == 0xb, "Got %08x\n", clsid.Data1); ok(clsid.Data2 == 0xcccc, "Got %04x\n", clsid.Data2); } static void test_StringFromGUID2(void) { WCHAR str[50]; int len; /* invalid pointer */ SetLastError(0xdeadbeef); len = StringFromGUID2(NULL,str,50); ok(len == 0, "len: %d (expected 0)\n", len); ok(GetLastError() == 0xdeadbeef, "Expected 0xdeadbeef, got %x\n", GetLastError()); /* Test corner cases for buffer size */ len = StringFromGUID2(&CLSID_StdFont,str,50); ok(len == 39, "len: %d (expected 39)\n", len); ok(!lstrcmpiW(str, wszCLSID_StdFont),"string wasn't equal for CLSID_StdFont\n"); memset(str,0,sizeof str); len = StringFromGUID2(&CLSID_StdFont,str,39); ok(len == 39, "len: %d (expected 39)\n", len); ok(!lstrcmpiW(str, wszCLSID_StdFont),"string wasn't equal for CLSID_StdFont\n"); len = StringFromGUID2(&CLSID_StdFont,str,38); ok(len == 0, "len: %d (expected 0)\n", len); len = StringFromGUID2(&CLSID_StdFont,str,30); ok(len == 0, "len: %d (expected 0)\n", len); } struct info { HANDLE wait, stop; }; static DWORD CALLBACK ole_initialize_thread(LPVOID pv) { HRESULT hr; struct info *info = pv; hr = pCoInitializeEx(NULL, COINIT_MULTITHREADED); SetEvent(info->wait); WaitForSingleObject(info->stop, 10000); CoUninitialize(); return hr; } static void test_CoCreateInstance(void) { HRESULT hr; HANDLE thread; DWORD tid, exitcode; IUnknown *pUnk; struct info info; REFCLSID rclsid = &CLSID_InternetZoneManager; pUnk = (IUnknown *)0xdeadbeef; hr = CoCreateInstance(rclsid, NULL, CLSCTX_INPROC_SERVER, &IID_IUnknown, (void **)&pUnk); ok(hr == CO_E_NOTINITIALIZED, "CoCreateInstance should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr); ok(pUnk == NULL, "CoCreateInstance should have changed the passed in pointer to NULL, instead of %p\n", pUnk); OleInitialize(NULL); /* test errors returned for non-registered clsids */ hr = CoCreateInstance(&CLSID_non_existent, NULL, CLSCTX_INPROC_SERVER, &IID_IUnknown, (void **)&pUnk); ok(hr == REGDB_E_CLASSNOTREG, "CoCreateInstance for non-registered inproc server should have returned REGDB_E_CLASSNOTREG instead of 0x%08x\n", hr); hr = CoCreateInstance(&CLSID_non_existent, NULL, CLSCTX_INPROC_HANDLER, &IID_IUnknown, (void **)&pUnk); ok(hr == REGDB_E_CLASSNOTREG, "CoCreateInstance for non-registered inproc handler should have returned REGDB_E_CLASSNOTREG instead of 0x%08x\n", hr); hr = CoCreateInstance(&CLSID_non_existent, NULL, CLSCTX_LOCAL_SERVER, &IID_IUnknown, (void **)&pUnk); ok(hr == REGDB_E_CLASSNOTREG, "CoCreateInstance for non-registered local server should have returned REGDB_E_CLASSNOTREG instead of 0x%08x\n", hr); hr = CoCreateInstance(&CLSID_non_existent, NULL, CLSCTX_REMOTE_SERVER, &IID_IUnknown, (void **)&pUnk); ok(hr == REGDB_E_CLASSNOTREG, "CoCreateInstance for non-registered remote server should have returned REGDB_E_CLASSNOTREG instead of 0x%08x\n", hr); hr = CoCreateInstance(rclsid, NULL, CLSCTX_INPROC_SERVER, &IID_IUnknown, (void **)&pUnk); if(hr == REGDB_E_CLASSNOTREG) { skip("IE not installed so can't test CoCreateInstance\n"); OleUninitialize(); return; } ok_ole_success(hr, "CoCreateInstance"); if(pUnk) IUnknown_Release(pUnk); OleUninitialize(); hr = CoCreateInstance(rclsid, NULL, CLSCTX_INPROC_SERVER, &IID_IUnknown, (void **)&pUnk); ok(hr == CO_E_NOTINITIALIZED, "CoCreateInstance should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr); /* show that COM doesn't have to be initialized for multi-threaded apartments if another thread has already done so */ info.wait = CreateEvent(NULL, TRUE, FALSE, NULL); ok(info.wait != NULL, "CreateEvent failed with error %d\n", GetLastError()); info.stop = CreateEvent(NULL, TRUE, FALSE, NULL); ok(info.stop != NULL, "CreateEvent failed with error %d\n", GetLastError()); thread = CreateThread(NULL, 0, ole_initialize_thread, &info, 0, &tid); ok(thread != NULL, "CreateThread failed with error %d\n", GetLastError()); ok( !WaitForSingleObject(info.wait, 10000 ), "wait timed out\n" ); pUnk = (IUnknown *)0xdeadbeef; hr = CoCreateInstance(rclsid, NULL, CLSCTX_INPROC_SERVER, &IID_IUnknown, (void **)&pUnk); ok(hr == S_OK, "CoCreateInstance should have returned S_OK instead of 0x%08x\n", hr); if (pUnk) IUnknown_Release(pUnk); SetEvent(info.stop); ok( !WaitForSingleObject(thread, 10000), "wait timed out\n" ); GetExitCodeThread(thread, &exitcode); hr = exitcode; ok(hr == S_OK, "thread should have returned S_OK instead of 0x%08x\n", hr); CloseHandle(thread); CloseHandle(info.wait); CloseHandle(info.stop); } static void test_CoGetClassObject(void) { HRESULT hr; HANDLE thread; DWORD tid, exitcode; IUnknown *pUnk; struct info info; REFCLSID rclsid = &CLSID_InternetZoneManager; hr = CoGetClassObject(rclsid, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void **)&pUnk); ok(hr == CO_E_NOTINITIALIZED, "CoGetClassObject should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr); ok(pUnk == NULL, "CoGetClassObject should have changed the passed in pointer to NULL, instead of %p\n", pUnk); hr = CoGetClassObject(rclsid, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, NULL); ok(hr == E_INVALIDARG || broken(hr == CO_E_NOTINITIALIZED), /* win9x */ "CoGetClassObject should have returned E_INVALIDARG instead of 0x%08x\n", hr); /* show that COM doesn't have to be initialized for multi-threaded apartments if another thread has already done so */ info.wait = CreateEvent(NULL, TRUE, FALSE, NULL); ok(info.wait != NULL, "CreateEvent failed with error %d\n", GetLastError()); info.stop = CreateEvent(NULL, TRUE, FALSE, NULL); ok(info.stop != NULL, "CreateEvent failed with error %d\n", GetLastError()); thread = CreateThread(NULL, 0, ole_initialize_thread, &info, 0, &tid); ok(thread != NULL, "CreateThread failed with error %d\n", GetLastError()); ok( !WaitForSingleObject(info.wait, 10000), "wait timed out\n" ); pUnk = (IUnknown *)0xdeadbeef; hr = CoGetClassObject(rclsid, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void **)&pUnk); if(hr == REGDB_E_CLASSNOTREG) skip("IE not installed so can't test CoGetClassObject\n"); else { ok(hr == S_OK, "CoGetClassObject should have returned S_OK instead of 0x%08x\n", hr); if (pUnk) IUnknown_Release(pUnk); } SetEvent(info.stop); ok( !WaitForSingleObject(thread, 10000), "wait timed out\n" ); GetExitCodeThread(thread, &exitcode); hr = exitcode; ok(hr == S_OK, "thread should have returned S_OK instead of 0x%08x\n", hr); CloseHandle(thread); CloseHandle(info.wait); CloseHandle(info.stop); } static ATOM register_dummy_class(void) { WNDCLASS wc = { 0, DefWindowProc, 0, 0, GetModuleHandle(NULL), NULL, LoadCursor(NULL, IDC_ARROW), (HBRUSH)(COLOR_BTNFACE+1), NULL, TEXT("WineOleTestClass"), }; return RegisterClass(&wc); } static void test_ole_menu(void) { HWND hwndFrame; HRESULT hr; hwndFrame = CreateWindow(MAKEINTATOM(register_dummy_class()), "Test", 0, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, NULL, NULL, NULL, NULL); hr = OleSetMenuDescriptor(NULL, hwndFrame, NULL, NULL, NULL); todo_wine ok_ole_success(hr, "OleSetMenuDescriptor"); DestroyWindow(hwndFrame); } static HRESULT WINAPI MessageFilter_QueryInterface(IMessageFilter *iface, REFIID riid, void ** ppvObj) { if (ppvObj == NULL) return E_POINTER; if (IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_IClassFactory)) { *ppvObj = iface; IMessageFilter_AddRef(iface); return S_OK; } return E_NOINTERFACE; } static ULONG WINAPI MessageFilter_AddRef(IMessageFilter *iface) { return 2; /* non-heap object */ } static ULONG WINAPI MessageFilter_Release(IMessageFilter *iface) { return 1; /* non-heap object */ } static DWORD WINAPI MessageFilter_HandleInComingCall( IMessageFilter *iface, DWORD dwCallType, HTASK threadIDCaller, DWORD dwTickCount, LPINTERFACEINFO lpInterfaceInfo) { trace("HandleInComingCall\n"); return SERVERCALL_ISHANDLED; } static DWORD WINAPI MessageFilter_RetryRejectedCall( IMessageFilter *iface, HTASK threadIDCallee, DWORD dwTickCount, DWORD dwRejectType) { trace("RetryRejectedCall\n"); return 0; } static DWORD WINAPI MessageFilter_MessagePending( IMessageFilter *iface, HTASK threadIDCallee, DWORD dwTickCount, DWORD dwPendingType) { trace("MessagePending\n"); return PENDINGMSG_WAITNOPROCESS; } static const IMessageFilterVtbl MessageFilter_Vtbl = { MessageFilter_QueryInterface, MessageFilter_AddRef, MessageFilter_Release, MessageFilter_HandleInComingCall, MessageFilter_RetryRejectedCall, MessageFilter_MessagePending }; static IMessageFilter MessageFilter = { &MessageFilter_Vtbl }; static void test_CoRegisterMessageFilter(void) { HRESULT hr; IMessageFilter *prev_filter; hr = CoRegisterMessageFilter(&MessageFilter, &prev_filter); ok(hr == CO_E_NOT_SUPPORTED, "CoRegisterMessageFilter should have failed with CO_E_NOT_SUPPORTED instead of 0x%08x\n", hr); pCoInitializeEx(NULL, COINIT_MULTITHREADED); prev_filter = (IMessageFilter *)0xdeadbeef; hr = CoRegisterMessageFilter(&MessageFilter, &prev_filter); ok(hr == CO_E_NOT_SUPPORTED, "CoRegisterMessageFilter should have failed with CO_E_NOT_SUPPORTED instead of 0x%08x\n", hr); ok(prev_filter == (IMessageFilter *)0xdeadbeef, "prev_filter should have been set to %p\n", prev_filter); CoUninitialize(); pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); hr = CoRegisterMessageFilter(NULL, NULL); ok_ole_success(hr, "CoRegisterMessageFilter"); prev_filter = (IMessageFilter *)0xdeadbeef; hr = CoRegisterMessageFilter(NULL, &prev_filter); ok_ole_success(hr, "CoRegisterMessageFilter"); ok(prev_filter == NULL, "prev_filter should have been set to NULL instead of %p\n", prev_filter); hr = CoRegisterMessageFilter(&MessageFilter, &prev_filter); ok_ole_success(hr, "CoRegisterMessageFilter"); ok(prev_filter == NULL, "prev_filter should have been set to NULL instead of %p\n", prev_filter); hr = CoRegisterMessageFilter(NULL, NULL); ok_ole_success(hr, "CoRegisterMessageFilter"); CoUninitialize(); } static HRESULT WINAPI Test_IUnknown_QueryInterface( IUnknown *iface, REFIID riid, LPVOID *ppvObj) { if (ppvObj == NULL) return E_POINTER; if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IWineTest)) { *ppvObj = iface; IUnknown_AddRef(iface); return S_OK; } *ppvObj = NULL; return E_NOINTERFACE; } static ULONG WINAPI Test_IUnknown_AddRef(IUnknown *iface) { return 2; /* non-heap-based object */ } static ULONG WINAPI Test_IUnknown_Release(IUnknown *iface) { return 1; /* non-heap-based object */ } static const IUnknownVtbl TestUnknown_Vtbl = { Test_IUnknown_QueryInterface, Test_IUnknown_AddRef, Test_IUnknown_Release, }; static IUnknown Test_Unknown = { &TestUnknown_Vtbl }; static HRESULT WINAPI PSFactoryBuffer_QueryInterface( IPSFactoryBuffer * This, /* [in] */ REFIID riid, /* [iid_is][out] */ void **ppvObject) { if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IPSFactoryBuffer)) { *ppvObject = This; IPSFactoryBuffer_AddRef(This); return S_OK; } return E_NOINTERFACE; } static ULONG WINAPI PSFactoryBuffer_AddRef( IPSFactoryBuffer * This) { return 2; } static ULONG WINAPI PSFactoryBuffer_Release( IPSFactoryBuffer * This) { return 1; } static HRESULT WINAPI PSFactoryBuffer_CreateProxy( IPSFactoryBuffer * This, /* [in] */ IUnknown *pUnkOuter, /* [in] */ REFIID riid, /* [out] */ IRpcProxyBuffer **ppProxy, /* [out] */ void **ppv) { return E_NOTIMPL; } static HRESULT WINAPI PSFactoryBuffer_CreateStub( IPSFactoryBuffer * This, /* [in] */ REFIID riid, /* [unique][in] */ IUnknown *pUnkServer, /* [out] */ IRpcStubBuffer **ppStub) { return E_NOTIMPL; } static IPSFactoryBufferVtbl PSFactoryBufferVtbl = { PSFactoryBuffer_QueryInterface, PSFactoryBuffer_AddRef, PSFactoryBuffer_Release, PSFactoryBuffer_CreateProxy, PSFactoryBuffer_CreateStub }; static IPSFactoryBuffer PSFactoryBuffer = { &PSFactoryBufferVtbl }; static const CLSID CLSID_WineTestPSFactoryBuffer = { 0x52011640, 0x8164, 0x4fd0, {0xa1, 0xa2, 0x5d, 0x5a, 0x36, 0x54, 0xd3, 0xbd} }; /* 52011640-8164-4fd0-a1a2-5d5a3654d3bd */ static void test_CoRegisterPSClsid(void) { HRESULT hr; DWORD dwRegistrationKey; IStream *stream; CLSID clsid; hr = CoRegisterPSClsid(&IID_IWineTest, &CLSID_WineTestPSFactoryBuffer); ok(hr == CO_E_NOTINITIALIZED, "CoRegisterPSClsid should have returened CO_E_NOTINITIALIZED instead of 0x%08x\n", hr); pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); hr = CoRegisterClassObject(&CLSID_WineTestPSFactoryBuffer, (IUnknown *)&PSFactoryBuffer, CLSCTX_INPROC_SERVER, REGCLS_MULTIPLEUSE, &dwRegistrationKey); ok_ole_success(hr, "CoRegisterClassObject"); hr = CoRegisterPSClsid(&IID_IWineTest, &CLSID_WineTestPSFactoryBuffer); ok_ole_success(hr, "CoRegisterPSClsid"); hr = CreateStreamOnHGlobal(NULL, TRUE, &stream); ok_ole_success(hr, "CreateStreamOnHGlobal"); hr = CoMarshalInterface(stream, &IID_IWineTest, &Test_Unknown, MSHCTX_INPROC, NULL, MSHLFLAGS_NORMAL); ok(hr == E_NOTIMPL, "CoMarshalInterface should have returned E_NOTIMPL instead of 0x%08x\n", hr); IStream_Release(stream); hr = CoRevokeClassObject(dwRegistrationKey); ok_ole_success(hr, "CoRevokeClassObject"); CoUninitialize(); pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); hr = CoGetPSClsid(&IID_IWineTest, &clsid); ok(hr == REGDB_E_IIDNOTREG, "CoGetPSClsid should have returned REGDB_E_IIDNOTREG instead of 0x%08x\n", hr); CoUninitialize(); } static void test_CoGetPSClsid(void) { HRESULT hr; CLSID clsid; hr = CoGetPSClsid(&IID_IClassFactory, &clsid); ok(hr == CO_E_NOTINITIALIZED, "CoGetPSClsid should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr); pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); hr = CoGetPSClsid(&IID_IClassFactory, &clsid); ok_ole_success(hr, "CoGetPSClsid"); hr = CoGetPSClsid(&IID_IWineTest, &clsid); ok(hr == REGDB_E_IIDNOTREG, "CoGetPSClsid for random IID returned 0x%08x instead of REGDB_E_IIDNOTREG\n", hr); hr = CoGetPSClsid(&IID_IClassFactory, NULL); ok(hr == E_INVALIDARG, "CoGetPSClsid for null clsid returned 0x%08x instead of E_INVALIDARG\n", hr); CoUninitialize(); } /* basic test, mainly for invalid arguments. see marshal.c for more */ static void test_CoUnmarshalInterface(void) { IUnknown *pProxy; IStream *pStream; HRESULT hr; hr = CoUnmarshalInterface(NULL, &IID_IUnknown, (void **)&pProxy); ok(hr == E_INVALIDARG, "CoUnmarshalInterface should have returned E_INVALIDARG instead of 0x%08x\n", hr); hr = CreateStreamOnHGlobal(NULL, TRUE, &pStream); ok_ole_success(hr, "CreateStreamOnHGlobal"); hr = CoUnmarshalInterface(pStream, &IID_IUnknown, (void **)&pProxy); todo_wine ok(hr == CO_E_NOTINITIALIZED, "CoUnmarshalInterface should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr); pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); hr = CoUnmarshalInterface(pStream, &IID_IUnknown, (void **)&pProxy); ok(hr == STG_E_READFAULT, "CoUnmarshalInterface should have returned STG_E_READFAULT instead of 0x%08x\n", hr); CoUninitialize(); hr = CoUnmarshalInterface(pStream, &IID_IUnknown, NULL); ok(hr == E_INVALIDARG, "CoUnmarshalInterface should have returned E_INVALIDARG instead of 0x%08x\n", hr); IStream_Release(pStream); } static void test_CoGetInterfaceAndReleaseStream(void) { HRESULT hr; IUnknown *pUnk; pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); hr = CoGetInterfaceAndReleaseStream(NULL, &IID_IUnknown, (void**)&pUnk); ok(hr == E_INVALIDARG, "hr %08x\n", hr); CoUninitialize(); } /* basic test, mainly for invalid arguments. see marshal.c for more */ static void test_CoMarshalInterface(void) { IStream *pStream; HRESULT hr; static const LARGE_INTEGER llZero; pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); hr = CreateStreamOnHGlobal(NULL, TRUE, &pStream); ok_ole_success(hr, "CreateStreamOnHGlobal"); hr = CoMarshalInterface(pStream, &IID_IUnknown, NULL, MSHCTX_INPROC, NULL, MSHLFLAGS_NORMAL); ok(hr == E_INVALIDARG, "CoMarshalInterface should have returned E_INVALIDARG instead of 0x%08x\n", hr); hr = CoMarshalInterface(NULL, &IID_IUnknown, (IUnknown *)&Test_ClassFactory, MSHCTX_INPROC, NULL, MSHLFLAGS_NORMAL); ok(hr == E_INVALIDARG, "CoMarshalInterface should have returned E_INVALIDARG instead of 0x%08x\n", hr); hr = CoMarshalInterface(pStream, &IID_IUnknown, (IUnknown *)&Test_ClassFactory, MSHCTX_INPROC, NULL, MSHLFLAGS_NORMAL); ok_ole_success(hr, "CoMarshalInterface"); /* stream not rewound */ hr = CoReleaseMarshalData(pStream); ok(hr == STG_E_READFAULT, "CoReleaseMarshalData should have returned STG_E_READFAULT instead of 0x%08x\n", hr); hr = IStream_Seek(pStream, llZero, STREAM_SEEK_SET, NULL); ok_ole_success(hr, "IStream_Seek"); hr = CoReleaseMarshalData(pStream); ok_ole_success(hr, "CoReleaseMarshalData"); IStream_Release(pStream); CoUninitialize(); } static void test_CoMarshalInterThreadInterfaceInStream(void) { IStream *pStream; HRESULT hr; IClassFactory *pProxy; pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); cLocks = 0; hr = CoMarshalInterThreadInterfaceInStream(&IID_IUnknown, (IUnknown *)&Test_ClassFactory, NULL); ok(hr == E_INVALIDARG, "CoMarshalInterThreadInterfaceInStream should have returned E_INVALIDARG instead of 0x%08x\n", hr); hr = CoMarshalInterThreadInterfaceInStream(&IID_IUnknown, NULL, &pStream); ok(hr == E_INVALIDARG, "CoMarshalInterThreadInterfaceInStream should have returned E_INVALIDARG instead of 0x%08x\n", hr); ok_no_locks(); hr = CoMarshalInterThreadInterfaceInStream(&IID_IUnknown, (IUnknown *)&Test_ClassFactory, &pStream); ok_ole_success(hr, "CoMarshalInterThreadInterfaceInStream"); ok_more_than_one_lock(); hr = CoUnmarshalInterface(pStream, &IID_IClassFactory, (void **)&pProxy); ok_ole_success(hr, "CoUnmarshalInterface"); IClassFactory_Release(pProxy); IStream_Release(pStream); ok_no_locks(); CoUninitialize(); } static void test_CoRegisterClassObject(void) { DWORD cookie; HRESULT hr; IClassFactory *pcf; pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); /* CLSCTX_INPROC_SERVER */ hr = CoRegisterClassObject(&CLSID_WineOOPTest, (IUnknown *)&Test_ClassFactory, CLSCTX_INPROC_SERVER, REGCLS_SINGLEUSE, &cookie); ok_ole_success(hr, "CoRegisterClassObject"); hr = CoRevokeClassObject(cookie); ok_ole_success(hr, "CoRevokeClassObject"); hr = CoRegisterClassObject(&CLSID_WineOOPTest, (IUnknown *)&Test_ClassFactory, CLSCTX_INPROC_SERVER, REGCLS_MULTIPLEUSE, &cookie); ok_ole_success(hr, "CoRegisterClassObject"); hr = CoRevokeClassObject(cookie); ok_ole_success(hr, "CoRevokeClassObject"); hr = CoRegisterClassObject(&CLSID_WineOOPTest, (IUnknown *)&Test_ClassFactory, CLSCTX_INPROC_SERVER, REGCLS_MULTI_SEPARATE, &cookie); ok_ole_success(hr, "CoRegisterClassObject"); hr = CoRevokeClassObject(cookie); ok_ole_success(hr, "CoRevokeClassObject"); /* CLSCTX_LOCAL_SERVER */ hr = CoRegisterClassObject(&CLSID_WineOOPTest, (IUnknown *)&Test_ClassFactory, CLSCTX_LOCAL_SERVER, REGCLS_SINGLEUSE, &cookie); ok_ole_success(hr, "CoRegisterClassObject"); hr = CoRevokeClassObject(cookie); ok_ole_success(hr, "CoRevokeClassObject"); hr = CoRegisterClassObject(&CLSID_WineOOPTest, (IUnknown *)&Test_ClassFactory, CLSCTX_LOCAL_SERVER, REGCLS_MULTIPLEUSE, &cookie); ok_ole_success(hr, "CoRegisterClassObject"); hr = CoRevokeClassObject(cookie); ok_ole_success(hr, "CoRevokeClassObject"); hr = CoRegisterClassObject(&CLSID_WineOOPTest, (IUnknown *)&Test_ClassFactory, CLSCTX_LOCAL_SERVER, REGCLS_MULTI_SEPARATE, &cookie); ok_ole_success(hr, "CoRegisterClassObject"); hr = CoRevokeClassObject(cookie); ok_ole_success(hr, "CoRevokeClassObject"); /* CLSCTX_INPROC_SERVER|CLSCTX_LOCAL_SERVER */ hr = CoRegisterClassObject(&CLSID_WineOOPTest, (IUnknown *)&Test_ClassFactory, CLSCTX_INPROC_SERVER|CLSCTX_LOCAL_SERVER, REGCLS_SINGLEUSE, &cookie); ok_ole_success(hr, "CoRegisterClassObject"); hr = CoRevokeClassObject(cookie); ok_ole_success(hr, "CoRevokeClassObject"); /* test whether an object that doesn't support IClassFactory can be * registered for CLSCTX_LOCAL_SERVER */ hr = CoRegisterClassObject(&CLSID_WineOOPTest, &Test_Unknown, CLSCTX_LOCAL_SERVER, REGCLS_SINGLEUSE, &cookie); ok_ole_success(hr, "CoRegisterClassObject"); hr = CoRevokeClassObject(cookie); ok_ole_success(hr, "CoRevokeClassObject"); /* test whether registered class becomes invalid when apartment is destroyed */ hr = CoRegisterClassObject(&CLSID_WineOOPTest, (IUnknown *)&Test_ClassFactory, CLSCTX_INPROC_SERVER, REGCLS_SINGLEUSE, &cookie); ok_ole_success(hr, "CoRegisterClassObject"); CoUninitialize(); pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); hr = CoGetClassObject(&CLSID_WineOOPTest, CLSCTX_INPROC_SERVER, NULL, &IID_IClassFactory, (void **)&pcf); ok(hr == REGDB_E_CLASSNOTREG, "object registered in an apartment shouldn't accessible after it is destroyed\n"); /* crashes with at least win9x DCOM! */ if (0) CoRevokeClassObject(cookie); CoUninitialize(); } static HRESULT get_class_object(CLSCTX clsctx) { HRESULT hr; IClassFactory *pcf; hr = CoGetClassObject(&CLSID_WineOOPTest, clsctx, NULL, &IID_IClassFactory, (void **)&pcf); if (SUCCEEDED(hr)) IClassFactory_Release(pcf); return hr; } static DWORD CALLBACK get_class_object_thread(LPVOID pv) { CLSCTX clsctx = (CLSCTX)(DWORD_PTR)pv; HRESULT hr; pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); hr = get_class_object(clsctx); CoUninitialize(); return hr; } static DWORD CALLBACK get_class_object_proxy_thread(LPVOID pv) { CLSCTX clsctx = (CLSCTX)(DWORD_PTR)pv; HRESULT hr; IClassFactory *pcf; IMultiQI *pMQI; pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); hr = CoGetClassObject(&CLSID_WineOOPTest, clsctx, NULL, &IID_IClassFactory, (void **)&pcf); if (SUCCEEDED(hr)) { hr = IClassFactory_QueryInterface(pcf, &IID_IMultiQI, (void **)&pMQI); if (SUCCEEDED(hr)) IMultiQI_Release(pMQI); IClassFactory_Release(pcf); } CoUninitialize(); return hr; } static DWORD CALLBACK register_class_object_thread(LPVOID pv) { HRESULT hr; DWORD cookie; pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); hr = CoRegisterClassObject(&CLSID_WineOOPTest, (IUnknown *)&Test_ClassFactory, CLSCTX_INPROC_SERVER, REGCLS_SINGLEUSE, &cookie); CoUninitialize(); return hr; } static DWORD CALLBACK revoke_class_object_thread(LPVOID pv) { DWORD cookie = (DWORD_PTR)pv; HRESULT hr; pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); hr = CoRevokeClassObject(cookie); CoUninitialize(); return hr; } static void test_registered_object_thread_affinity(void) { HRESULT hr; DWORD cookie; HANDLE thread; DWORD tid; DWORD exitcode; pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); /* CLSCTX_INPROC_SERVER */ hr = CoRegisterClassObject(&CLSID_WineOOPTest, (IUnknown *)&Test_ClassFactory, CLSCTX_INPROC_SERVER, REGCLS_SINGLEUSE, &cookie); ok_ole_success(hr, "CoRegisterClassObject"); thread = CreateThread(NULL, 0, get_class_object_thread, (LPVOID)CLSCTX_INPROC_SERVER, 0, &tid); ok(thread != NULL, "CreateThread failed with error %d\n", GetLastError()); ok( !WaitForSingleObject(thread, 10000), "wait timed out\n" ); GetExitCodeThread(thread, &exitcode); hr = exitcode; ok(hr == REGDB_E_CLASSNOTREG, "CoGetClassObject on inproc object " "registered in different thread should return REGDB_E_CLASSNOTREG " "instead of 0x%08x\n", hr); hr = get_class_object(CLSCTX_INPROC_SERVER); ok(hr == S_OK, "CoGetClassObject on inproc object registered in same " "thread should return S_OK instead of 0x%08x\n", hr); thread = CreateThread(NULL, 0, register_class_object_thread, NULL, 0, &tid); ok(thread != NULL, "CreateThread failed with error %d\n", GetLastError()); ok ( !WaitForSingleObject(thread, 10000), "wait timed out\n" ); GetExitCodeThread(thread, &exitcode); hr = exitcode; ok(hr == S_OK, "CoRegisterClassObject with same CLSID but in different thread should return S_OK instead of 0x%08x\n", hr); hr = CoRevokeClassObject(cookie); ok_ole_success(hr, "CoRevokeClassObject"); /* CLSCTX_LOCAL_SERVER */ hr = CoRegisterClassObject(&CLSID_WineOOPTest, (IUnknown *)&Test_ClassFactory, CLSCTX_LOCAL_SERVER, REGCLS_MULTIPLEUSE, &cookie); ok_ole_success(hr, "CoRegisterClassObject"); thread = CreateThread(NULL, 0, get_class_object_proxy_thread, (LPVOID)CLSCTX_LOCAL_SERVER, 0, &tid); ok(thread != NULL, "CreateThread failed with error %d\n", GetLastError()); while (MsgWaitForMultipleObjects(1, &thread, FALSE, 10000, QS_ALLINPUT) == WAIT_OBJECT_0 + 1) { MSG msg; while (PeekMessageA(&msg, NULL, 0, 0, PM_REMOVE)) { TranslateMessage(&msg); DispatchMessageA(&msg); } } GetExitCodeThread(thread, &exitcode); hr = exitcode; ok(hr == S_OK, "CoGetClassObject on local server object " "registered in different thread should return S_OK " "instead of 0x%08x\n", hr); hr = get_class_object(CLSCTX_LOCAL_SERVER); ok(hr == S_OK, "CoGetClassObject on local server object registered in same " "thread should return S_OK instead of 0x%08x\n", hr); thread = CreateThread(NULL, 0, revoke_class_object_thread, (LPVOID)(DWORD_PTR)cookie, 0, &tid); ok(thread != NULL, "CreateThread failed with error %d\n", GetLastError()); ok( !WaitForSingleObject(thread, 10000), "wait timed out\n" ); GetExitCodeThread(thread, &exitcode); hr = exitcode; ok(hr == RPC_E_WRONG_THREAD, "CoRevokeClassObject called from different " "thread to where registered should return RPC_E_WRONG_THREAD instead of 0x%08x\n", hr); thread = CreateThread(NULL, 0, register_class_object_thread, NULL, 0, &tid); ok(thread != NULL, "CreateThread failed with error %d\n", GetLastError()); ok( !WaitForSingleObject(thread, 10000), "wait timed out\n" ); GetExitCodeThread(thread, &exitcode); hr = exitcode; ok(hr == S_OK, "CoRegisterClassObject with same CLSID but in different " "thread should return S_OK instead of 0x%08x\n", hr); hr = CoRevokeClassObject(cookie); ok_ole_success(hr, "CoRevokeClassObject"); CoUninitialize(); } static DWORD CALLBACK free_libraries_thread(LPVOID p) { CoFreeUnusedLibraries(); return 0; } static inline BOOL is_module_loaded(const char *module) { return GetModuleHandle(module) != 0; } static void test_CoFreeUnusedLibraries(void) { HRESULT hr; IUnknown *pUnk; DWORD tid; HANDLE thread; pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); ok(!is_module_loaded("urlmon.dll"), "urlmon.dll shouldn't be loaded\n"); hr = CoCreateInstance(&CLSID_FileProtocol, NULL, CLSCTX_INPROC_SERVER, &IID_IInternetProtocol, (void **)&pUnk); if (hr == REGDB_E_CLASSNOTREG) { skip("IE not installed so can't run CoFreeUnusedLibraries test\n"); CoUninitialize(); return; } ok_ole_success(hr, "CoCreateInstance"); ok(is_module_loaded("urlmon.dll"), "urlmon.dll should be loaded\n"); ok(pUnk != NULL || broken(pUnk == NULL), /* win9x */ "Expected a valid pointer\n"); if (pUnk) IUnknown_Release(pUnk); ok(is_module_loaded("urlmon.dll"), "urlmon.dll should be loaded\n"); thread = CreateThread(NULL, 0, free_libraries_thread, NULL, 0, &tid); ok( !WaitForSingleObject(thread, 10000), "wait timed out\n" ); CloseHandle(thread); ok(is_module_loaded("urlmon.dll"), "urlmon.dll should be loaded\n"); CoFreeUnusedLibraries(); ok(!is_module_loaded("urlmon.dll"), "urlmon.dll shouldn't be loaded\n"); CoUninitialize(); } static void test_CoGetObjectContext(void) { HRESULT hr; ULONG refs; IComThreadingInfo *pComThreadingInfo; IContextCallback *pContextCallback; IObjContext *pObjContext; APTTYPE apttype; THDTYPE thdtype; struct info info; HANDLE thread; DWORD tid, exitcode; if (!pCoGetObjectContext) { skip("CoGetObjectContext not present\n"); return; } hr = pCoGetObjectContext(&IID_IComThreadingInfo, (void **)&pComThreadingInfo); ok(hr == CO_E_NOTINITIALIZED, "CoGetObjectContext should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr); ok(pComThreadingInfo == NULL, "pComThreadingInfo should have been set to NULL\n"); /* show that COM doesn't have to be initialized for multi-threaded apartments if another thread has already done so */ info.wait = CreateEvent(NULL, TRUE, FALSE, NULL); ok(info.wait != NULL, "CreateEvent failed with error %d\n", GetLastError()); info.stop = CreateEvent(NULL, TRUE, FALSE, NULL); ok(info.stop != NULL, "CreateEvent failed with error %d\n", GetLastError()); thread = CreateThread(NULL, 0, ole_initialize_thread, &info, 0, &tid); ok(thread != NULL, "CreateThread failed with error %d\n", GetLastError()); ok( !WaitForSingleObject(info.wait, 10000), "wait timed out\n" ); pComThreadingInfo = NULL; hr = pCoGetObjectContext(&IID_IComThreadingInfo, (void **)&pComThreadingInfo); ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", hr); IComThreadingInfo_Release(pComThreadingInfo); SetEvent(info.stop); ok( !WaitForSingleObject(thread, 10000), "wait timed out\n" ); GetExitCodeThread(thread, &exitcode); hr = exitcode; ok(hr == S_OK, "thread should have returned S_OK instead of 0x%08x\n", hr); CloseHandle(thread); CloseHandle(info.wait); CloseHandle(info.stop); pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); hr = pCoGetObjectContext(&IID_IComThreadingInfo, (void **)&pComThreadingInfo); ok_ole_success(hr, "CoGetObjectContext"); hr = IComThreadingInfo_GetCurrentApartmentType(pComThreadingInfo, &apttype); ok_ole_success(hr, "IComThreadingInfo_GetCurrentApartmentType"); ok(apttype == APTTYPE_MAINSTA, "apartment type should be APTTYPE_MAINSTA instead of %d\n", apttype); hr = IComThreadingInfo_GetCurrentThreadType(pComThreadingInfo, &thdtype); ok_ole_success(hr, "IComThreadingInfo_GetCurrentThreadType"); ok(thdtype == THDTYPE_PROCESSMESSAGES, "thread type should be THDTYPE_PROCESSMESSAGES instead of %d\n", thdtype); refs = IComThreadingInfo_Release(pComThreadingInfo); ok(refs == 0, "pComThreadingInfo should have 0 refs instead of %d refs\n", refs); hr = pCoGetObjectContext(&IID_IContextCallback, (void **)&pContextCallback); ok_ole_success(hr, "CoGetObjectContext(ContextCallback)"); if (hr == S_OK) { refs = IContextCallback_Release(pContextCallback); ok(refs == 0, "pContextCallback should have 0 refs instead of %d refs\n", refs); } CoUninitialize(); pCoInitializeEx(NULL, COINIT_MULTITHREADED); hr = pCoGetObjectContext(&IID_IComThreadingInfo, (void **)&pComThreadingInfo); ok_ole_success(hr, "CoGetObjectContext"); hr = IComThreadingInfo_GetCurrentApartmentType(pComThreadingInfo, &apttype); ok_ole_success(hr, "IComThreadingInfo_GetCurrentApartmentType"); ok(apttype == APTTYPE_MTA, "apartment type should be APTTYPE_MTA instead of %d\n", apttype); hr = IComThreadingInfo_GetCurrentThreadType(pComThreadingInfo, &thdtype); ok_ole_success(hr, "IComThreadingInfo_GetCurrentThreadType"); ok(thdtype == THDTYPE_BLOCKMESSAGES, "thread type should be THDTYPE_BLOCKMESSAGES instead of %d\n", thdtype); refs = IComThreadingInfo_Release(pComThreadingInfo); ok(refs == 0, "pComThreadingInfo should have 0 refs instead of %d refs\n", refs); hr = pCoGetObjectContext(&IID_IContextCallback, (void **)&pContextCallback); ok_ole_success(hr, "CoGetObjectContext(ContextCallback)"); if (hr == S_OK) { refs = IContextCallback_Release(pContextCallback); ok(refs == 0, "pContextCallback should have 0 refs instead of %d refs\n", refs); } hr = pCoGetObjectContext(&IID_IObjContext, (void **)&pObjContext); ok_ole_success(hr, "CoGetObjectContext"); refs = IObjContext_Release(pObjContext); ok(refs == 0, "pObjContext should have 0 refs instead of %d refs\n", refs); CoUninitialize(); } typedef struct { IUnknown IUnknown_iface; LONG refs; } Test_CallContext; static inline Test_CallContext *impl_from_IUnknown(IUnknown *iface) { return CONTAINING_RECORD(iface, Test_CallContext, IUnknown_iface); } static HRESULT WINAPI Test_CallContext_QueryInterface( IUnknown *iface, REFIID riid, LPVOID *ppvObj) { if (ppvObj == NULL) return E_POINTER; if (IsEqualGUID(riid, &IID_IUnknown)) { *ppvObj = iface; IUnknown_AddRef(iface); return S_OK; } *ppvObj = NULL; return E_NOINTERFACE; } static ULONG WINAPI Test_CallContext_AddRef(IUnknown *iface) { Test_CallContext *This = impl_from_IUnknown(iface); return InterlockedIncrement(&This->refs); } static ULONG WINAPI Test_CallContext_Release(IUnknown *iface) { Test_CallContext *This = impl_from_IUnknown(iface); ULONG refs = InterlockedDecrement(&This->refs); if (!refs) HeapFree(GetProcessHeap(), 0, This); return refs; } static const IUnknownVtbl TestCallContext_Vtbl = { Test_CallContext_QueryInterface, Test_CallContext_AddRef, Test_CallContext_Release }; static void test_CoGetCallContext(void) { HRESULT hr; ULONG refs; IUnknown *pUnk; Test_CallContext *test_object; if (!pCoSwitchCallContext) { skip("CoSwitchCallContext not present\n"); return; } CoInitialize(NULL); test_object = HeapAlloc(GetProcessHeap(), 0, sizeof(Test_CallContext)); test_object->IUnknown_iface.lpVtbl = &TestCallContext_Vtbl; test_object->refs = 1; hr = CoGetCallContext(&IID_IUnknown, (void**)&pUnk); ok(hr == RPC_E_CALL_COMPLETE, "Expected RPC_E_CALL_COMPLETE, got 0x%08x\n", hr); pUnk = (IUnknown*)0xdeadbeef; hr = pCoSwitchCallContext(&test_object->IUnknown_iface, &pUnk); ok_ole_success(hr, "CoSwitchCallContext"); ok(pUnk == NULL, "expected NULL, got %p\n", pUnk); refs = IUnknown_AddRef(&test_object->IUnknown_iface); ok(refs == 2, "Expected refcount 2, got %d\n", refs); IUnknown_Release(&test_object->IUnknown_iface); pUnk = (IUnknown*)0xdeadbeef; hr = CoGetCallContext(&IID_IUnknown, (void**)&pUnk); ok_ole_success(hr, "CoGetCallContext"); ok(pUnk == &test_object->IUnknown_iface, "expected %p, got %p\n", &test_object->IUnknown_iface, pUnk); refs = IUnknown_AddRef(&test_object->IUnknown_iface); ok(refs == 3, "Expected refcount 3, got %d\n", refs); IUnknown_Release(&test_object->IUnknown_iface); IUnknown_Release(pUnk); pUnk = (IUnknown*)0xdeadbeef; hr = pCoSwitchCallContext(NULL, &pUnk); ok_ole_success(hr, "CoSwitchCallContext"); ok(pUnk == &test_object->IUnknown_iface, "expected %p, got %p\n", &test_object->IUnknown_iface, pUnk); refs = IUnknown_AddRef(&test_object->IUnknown_iface); ok(refs == 2, "Expected refcount 2, got %d\n", refs); IUnknown_Release(&test_object->IUnknown_iface); hr = CoGetCallContext(&IID_IUnknown, (void**)&pUnk); ok(hr == RPC_E_CALL_COMPLETE, "Expected RPC_E_CALL_COMPLETE, got 0x%08x\n", hr); IUnknown_Release(&test_object->IUnknown_iface); CoUninitialize(); } static void test_CoGetContextToken(void) { HRESULT hr; ULONG refs; ULONG_PTR token; IObjContext *ctx; struct info info; HANDLE thread; DWORD tid, exitcode; if (!pCoGetContextToken) { win_skip("CoGetContextToken not present\n"); return; } token = 0xdeadbeef; hr = pCoGetContextToken(&token); ok(hr == CO_E_NOTINITIALIZED, "Expected CO_E_NOTINITIALIZED, got 0x%08x\n", hr); ok(token == 0xdeadbeef, "Expected 0, got 0x%lx\n", token); /* show that COM doesn't have to be initialized for multi-threaded apartments if another thread has already done so */ info.wait = CreateEvent(NULL, TRUE, FALSE, NULL); ok(info.wait != NULL, "CreateEvent failed with error %d\n", GetLastError()); info.stop = CreateEvent(NULL, TRUE, FALSE, NULL); ok(info.stop != NULL, "CreateEvent failed with error %d\n", GetLastError()); thread = CreateThread(NULL, 0, ole_initialize_thread, &info, 0, &tid); ok(thread != NULL, "CreateThread failed with error %d\n", GetLastError()); ok( !WaitForSingleObject(info.wait, 10000), "wait timed out\n" ); token = 0; hr = pCoGetContextToken(&token); ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", hr); SetEvent(info.stop); ok( !WaitForSingleObject(thread, 10000), "wait timed out\n" ); GetExitCodeThread(thread, &exitcode); hr = exitcode; ok(hr == S_OK, "thread should have returned S_OK instead of 0x%08x\n", hr); CloseHandle(thread); CloseHandle(info.wait); CloseHandle(info.stop); CoInitialize(NULL); hr = pCoGetContextToken(NULL); ok(hr == E_POINTER, "Expected E_POINTER, got 0x%08x\n", hr); token = 0; hr = pCoGetContextToken(&token); ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", hr); ok(token, "Expected token != 0\n"); refs = IUnknown_AddRef((IUnknown *)token); todo_wine ok(refs == 1, "Expected 1, got %u\n", refs); hr = pCoGetObjectContext(&IID_IObjContext, (void **)&ctx); ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", hr); todo_wine ok(ctx == (IObjContext *)token, "Expected interface pointers to be the same\n"); refs = IUnknown_AddRef((IUnknown *)ctx); todo_wine ok(refs == 3, "Expected 3, got %u\n", refs); refs = IUnknown_Release((IUnknown *)ctx); todo_wine ok(refs == 2, "Expected 2, got %u\n", refs); refs = IUnknown_Release((IUnknown *)token); ok(refs == 1, "Expected 1, got %u\n", refs); /* CoGetContextToken does not add a reference */ token = 0; hr = pCoGetContextToken(&token); ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", hr); ok(token, "Expected token != 0\n"); todo_wine ok(ctx == (IObjContext *)token, "Expected interface pointers to be the same\n"); refs = IUnknown_AddRef((IUnknown *)ctx); ok(refs == 2, "Expected 1, got %u\n", refs); refs = IUnknown_Release((IUnknown *)ctx); ok(refs == 1, "Expected 0, got %u\n", refs); refs = IUnknown_Release((IUnknown *)ctx); ok(refs == 0, "Expected 0, got %u\n", refs); CoUninitialize(); } static void test_CoGetTreatAsClass(void) { HRESULT hr; CLSID out; static GUID deadbeef = {0xdeadbeef,0xdead,0xbeef,{0xde,0xad,0xbe,0xef,0xde,0xad,0xbe,0xef}}; if (!pCoGetTreatAsClass) { win_skip("CoGetTreatAsClass not present\n"); return; } hr = pCoGetTreatAsClass(&deadbeef,&out); ok (hr == S_FALSE, "expected S_FALSE got %x\n",hr); ok (IsEqualGUID(&out,&deadbeef), "expected to get same clsid back\n"); } static void test_CoInitializeEx(void) { HRESULT hr; hr = pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED); ok(hr == S_OK, "CoInitializeEx failed with error 0x%08x\n", hr); /* Calling OleInitialize for the first time should yield S_OK even with * apartment already initialized by previous CoInitialize(Ex) calls. */ hr = OleInitialize(NULL); ok(hr == S_OK, "OleInitialize failed with error 0x%08x\n", hr); /* Subsequent calls to OleInitialize should return S_FALSE */ hr = OleInitialize(NULL); ok(hr == S_FALSE, "Expected S_FALSE, hr = 0x%08x\n", hr); /* Cleanup */ CoUninitialize(); OleUninitialize(); } START_TEST(compobj) { HMODULE hOle32 = GetModuleHandle("ole32"); pCoGetObjectContext = (void*)GetProcAddress(hOle32, "CoGetObjectContext"); pCoSwitchCallContext = (void*)GetProcAddress(hOle32, "CoSwitchCallContext"); pCoGetTreatAsClass = (void*)GetProcAddress(hOle32,"CoGetTreatAsClass"); pCoGetContextToken = (void*)GetProcAddress(hOle32, "CoGetContextToken"); if (!(pCoInitializeEx = (void*)GetProcAddress(hOle32, "CoInitializeEx"))) { trace("You need DCOM95 installed to run this test\n"); return; } test_ProgIDFromCLSID(); test_CLSIDFromProgID(); test_CLSIDFromString(); test_StringFromGUID2(); test_CoCreateInstance(); test_ole_menu(); test_CoGetClassObject(); test_CoRegisterMessageFilter(); test_CoRegisterPSClsid(); test_CoGetPSClsid(); test_CoUnmarshalInterface(); test_CoGetInterfaceAndReleaseStream(); test_CoMarshalInterface(); test_CoMarshalInterThreadInterfaceInStream(); test_CoRegisterClassObject(); test_registered_object_thread_affinity(); test_CoFreeUnusedLibraries(); test_CoGetObjectContext(); test_CoGetCallContext(); test_CoGetContextToken(); test_CoGetTreatAsClass(); test_CoInitializeEx(); }