#include "windef.h"
#include "winbase.h"
#define USE_COM_CONTEXT_DEF
-#ifndef __REACTOS__
-#include "initguid.h"
-#endif
#include "objbase.h"
#include "shlguid.h"
#include "urlmon.h" /* for CLSID_FileProtocol */
#include "ctxtcall.h"
#include "wine/test.h"
-
-#ifdef __REACTOS__
-#include <initguid.h>
-#endif
+#include "initguid.h"
#define DEFINE_EXPECT(func) \
static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
static const GUID IID_Testiface6 = { 0x72222222, 0x1234, 0x1234, { 0x12, 0x34, 0x56, 0x78, 0x9a, 0xbc, 0xde, 0xf0 } };
static const GUID IID_TestPS = { 0x66666666, 0x8888, 0x7777, { 0x66, 0x66, 0x55, 0x55, 0x55, 0x55, 0x55, 0x55 } };
-DEFINE_GUID(CLSID_InProcFreeMarshaler, 0x0000033a,0x0000,0x0000,0xc0,0x00,0x00,0x00,0x00,0x00,0x00,0x46);
DEFINE_GUID(CLSID_testclsid, 0xacd014c7,0x9535,0x4fac,0x8b,0x53,0xa4,0x8c,0xa7,0xf4,0xd7,0x26);
-DEFINE_GUID(CLSID_GlobalOptions, 0x0000034b,0x0000,0x0000,0xc0,0x00,0x00,0x00,0x00,0x00,0x00,0x46);
static const 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};
WCHAR path[MAX_PATH];
MultiByteToWideChar( CP_ACP, 0, filename, -1, path, MAX_PATH );
- GetFullPathNameW(path, sizeof(manifest_path)/sizeof(WCHAR), manifest_path, NULL);
+ GetFullPathNameW(path, ARRAY_SIZE(manifest_path), manifest_path, NULL);
manifest_len = strlen(manifest);
file = CreateFileW(path, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
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;
-}
-
-#define test_apt_type(t, q, t_t, t_q) _test_apt_type(t, q, t_t, t_q, __LINE__)
-static void _test_apt_type(APTTYPE expected_type, APTTYPEQUALIFIER expected_qualifier, BOOL todo_type,
- BOOL todo_qualifier, int line)
+#define test_apt_type(t, q) _test_apt_type(t, q, __LINE__)
+static void _test_apt_type(APTTYPE expected_type, APTTYPEQUALIFIER expected_qualifier, int line)
{
APTTYPEQUALIFIER qualifier = ~0u;
APTTYPE type = ~0u;
hr = pCoGetApartmentType(&type, &qualifier);
ok_(__FILE__, line)(hr == S_OK || hr == CO_E_NOTINITIALIZED, "Unexpected return code: 0x%08x\n", hr);
-todo_wine_if(todo_type)
ok_(__FILE__, line)(type == expected_type, "Wrong apartment type %d, expected %d\n", type, expected_type);
-todo_wine_if(todo_qualifier)
ok_(__FILE__, line)(qualifier == expected_qualifier, "Wrong apartment qualifier %d, expected %d\n", qualifier,
expected_qualifier);
}
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);
- /* show that COM doesn't have to be initialized for multi-threaded apartments if another
- thread has already done so */
-
- test_apt_type(APTTYPE_CURRENT, APTTYPEQUALIFIER_NONE, FALSE, FALSE);
-
- info.wait = CreateEventA(NULL, TRUE, FALSE, NULL);
- ok(info.wait != NULL, "CreateEvent failed with error %d\n", GetLastError());
-
- info.stop = CreateEventA(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" );
-
- test_apt_type(APTTYPE_MTA, APTTYPEQUALIFIER_IMPLICIT_MTA, TRUE, TRUE);
-
- 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);
-
- test_apt_type(APTTYPE_CURRENT, APTTYPEQUALIFIER_NONE, FALSE, FALSE);
+ test_apt_type(APTTYPE_CURRENT, APTTYPEQUALIFIER_NONE);
}
static void test_CoGetClassObject(void)
{
HRESULT hr;
- HANDLE thread, handle;
- DWORD tid, exitcode;
+ HANDLE handle;
ULONG_PTR cookie;
IUnknown *pUnk;
- struct info info;
REFCLSID rclsid = &CLSID_InternetZoneManager;
HKEY hkey;
LONG res;
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 */
-
- test_apt_type(APTTYPE_CURRENT, APTTYPEQUALIFIER_NONE, FALSE, FALSE);
-
- info.wait = CreateEventA(NULL, TRUE, FALSE, NULL);
- ok(info.wait != NULL, "CreateEvent failed with error %d\n", GetLastError());
-
- info.stop = CreateEventA(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" );
-
- test_apt_type(APTTYPE_MTA, APTTYPEQUALIFIER_IMPLICIT_MTA, TRUE, TRUE);
-
- 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);
-
- test_apt_type(APTTYPE_CURRENT, APTTYPEQUALIFIER_NONE, FALSE, FALSE);
+ test_apt_type(APTTYPE_CURRENT, APTTYPEQUALIFIER_NONE);
if (!pRegOverridePredefKey)
{
IObjContext *pObjContext;
APTTYPE apttype;
THDTYPE thdtype;
- struct info info;
- HANDLE thread;
- DWORD tid, exitcode;
GUID id, id2;
if (!pCoGetObjectContext)
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 */
-
- test_apt_type(APTTYPE_CURRENT, APTTYPEQUALIFIER_NONE, FALSE, FALSE);
-
- info.wait = CreateEventA(NULL, TRUE, FALSE, NULL);
- ok(info.wait != NULL, "CreateEvent failed with error %d\n", GetLastError());
-
- info.stop = CreateEventA(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" );
+ pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
- test_apt_type(APTTYPE_MTA, APTTYPEQUALIFIER_IMPLICIT_MTA, TRUE, TRUE);
+ test_apt_type(APTTYPE_MAINSTA, APTTYPEQUALIFIER_NONE);
- pComThreadingInfo = NULL;
hr = pCoGetObjectContext(&IID_IComThreadingInfo, (void **)&pComThreadingInfo);
- ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", hr);
+ ok_ole_success(hr, "CoGetObjectContext");
threadinginfo2 = NULL;
hr = pCoGetObjectContext(&IID_IComThreadingInfo, (void **)&threadinginfo2);
hr = CoGetCurrentLogicalThreadId(&id2);
ok(IsEqualGUID(&id, &id2), "got %s, expected %s\n", wine_dbgstr_guid(&id), wine_dbgstr_guid(&id2));
- 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);
-
- test_apt_type(APTTYPE_CURRENT, APTTYPEQUALIFIER_NONE, FALSE, FALSE);
-
- pCoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
-
- test_apt_type(APTTYPE_MAINSTA, APTTYPEQUALIFIER_NONE, FALSE, FALSE);
-
- 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);
ULONG refs;
ULONG_PTR token, token2;
IObjContext *ctx;
- struct info info;
- HANDLE thread;
- DWORD tid, exitcode;
if (!pCoGetContextToken)
{
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 */
-
- test_apt_type(APTTYPE_CURRENT, APTTYPEQUALIFIER_NONE, FALSE, FALSE);
-
- info.wait = CreateEventA(NULL, TRUE, FALSE, NULL);
- ok(info.wait != NULL, "CreateEvent failed with error %d\n", GetLastError());
-
- info.stop = CreateEventA(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" );
-
- test_apt_type(APTTYPE_MTA, APTTYPEQUALIFIER_IMPLICIT_MTA, TRUE, TRUE);
-
- token = 0;
- hr = pCoGetContextToken(&token);
- ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", hr);
-
- token2 = 0;
- hr = pCoGetContextToken(&token2);
- ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", hr);
- ok(token == token2, "got different token\n");
-
- 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);
-
- test_apt_type(APTTYPE_CURRENT, APTTYPEQUALIFIER_NONE, FALSE, FALSE);
+ test_apt_type(APTTYPE_CURRENT, APTTYPEQUALIFIER_NONE);
CoInitialize(NULL);
- test_apt_type(APTTYPE_MAINSTA, APTTYPEQUALIFIER_NONE, FALSE, FALSE);
+ test_apt_type(APTTYPE_MAINSTA, APTTYPEQUALIFIER_NONE);
hr = pCoGetContextToken(NULL);
ok(hr == E_POINTER, "Expected E_POINTER, got 0x%08x\n", hr);
}
/* test using registered CLSID */
- StringFromGUID2(&CLSID_non_existent, clsidW, sizeof(clsidW)/sizeof(clsidW[0]));
+ StringFromGUID2(&CLSID_non_existent, clsidW, ARRAY_SIZE(clsidW));
ret = RegCreateKeyExW(HKEY_CLASSES_ROOT, clsidkeyW, 0, NULL, 0, KEY_ALL_ACCESS, NULL, &clsidhkey, &disposition);
if (!ret)
static LRESULT CALLBACK cowait_window_proc(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
{
- if(cowait_msgs_last < sizeof(cowait_msgs)/sizeof(*cowait_msgs))
+ if(cowait_msgs_last < ARRAY_SIZE(cowait_msgs))
cowait_msgs[cowait_msgs_last++] = msg;
if(msg == WM_DDE_FIRST)
return 6;
ok(index == WAIT_OBJECT_0, "WaitForSingleObject failed\n");
CloseHandle(thread);
+ CoUninitialize();
+
+ /* If COM was not initialized, messages are neither pumped nor peeked at */
+ PostMessageA(hWnd, WM_DDE_FIRST, 0, 0);
+ hr = CoWaitForMultipleHandles(0, 100, 2, handles, &index);
+ ok(hr == RPC_S_CALLPENDING, "got %#x\n", hr);
+ success = MsgWaitForMultipleObjectsEx(0, NULL, 2, QS_ALLPOSTMESSAGE, MWMO_ALERTABLE);
+ ok(success == 0, "MsgWaitForMultipleObjects returned %x\n", success);
+ success = PeekMessageA(&msg, hWnd, WM_DDE_FIRST, WM_DDE_FIRST, PM_REMOVE);
+ ok(success, "PeekMessage failed: %u\n", GetLastError());
+
+ /* same in an MTA */
+ CoInitializeEx(NULL, COINIT_MULTITHREADED);
+
+ PostMessageA(hWnd, WM_DDE_FIRST, 0, 0);
+ hr = CoWaitForMultipleHandles(0, 100, 2, handles, &index);
+ ok(hr == RPC_S_CALLPENDING, "got %#x\n", hr);
+ success = MsgWaitForMultipleObjectsEx(0, NULL, 2, QS_ALLPOSTMESSAGE, MWMO_ALERTABLE);
+ ok(success == 0, "MsgWaitForMultipleObjects returned %x\n", success);
+ success = PeekMessageA(&msg, hWnd, WM_DDE_FIRST, WM_DDE_FIRST, PM_REMOVE);
+ ok(success, "PeekMessage failed: %u\n", GetLastError());
+
+ CoUninitialize();
+
CloseHandle(handles[0]);
CloseHandle(handles[1]);
DestroyWindow(hWnd);
success = UnregisterClassA(cls_name, GetModuleHandleA(0));
ok(success, "UnregisterClass failed %u\n", GetLastError());
-
- CoUninitialize();
}
static void test_CoGetMalloc(void)
pReleaseActCtx = (void*)GetProcAddress(hkernel32, "ReleaseActCtx");
}
+static DWORD CALLBACK implicit_mta_proc(void *param)
+{
+ IComThreadingInfo *threading_info;
+ ULONG_PTR token;
+ IUnknown *unk;
+ DWORD cookie;
+ CLSID clsid;
+ HRESULT hr;
+
+ test_apt_type(APTTYPE_MTA, APTTYPEQUALIFIER_IMPLICIT_MTA);
+
+ hr = CoCreateInstance(&CLSID_InternetZoneManager, NULL, CLSCTX_INPROC_SERVER, &IID_IUnknown, (void **)&unk);
+ ok_ole_success(hr, "CoCreateInstance");
+ IUnknown_Release(unk);
+
+ hr = CoGetClassObject(&CLSID_InternetZoneManager, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void **)&unk);
+ ok_ole_success(hr, "CoGetClassObject");
+ IUnknown_Release(unk);
+
+ hr = CoGetObjectContext(&IID_IComThreadingInfo, (void **)&threading_info);
+ ok_ole_success(hr, "CoGetObjectContext");
+ IComThreadingInfo_Release(threading_info);
+
+ hr = CoGetContextToken(&token);
+ ok_ole_success(hr, "CoGetContextToken");
+
+ hr = CoRegisterPSClsid(&IID_IWineTest, &CLSID_WineTestPSFactoryBuffer);
+ ok_ole_success(hr, "CoRegisterPSClsid");
+
+ hr = CoGetPSClsid(&IID_IClassFactory, &clsid);
+ ok_ole_success(hr, "CoGetPSClsid");
+
+ 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 = CoRegisterMessageFilter(NULL, NULL);
+ ok(hr == CO_E_NOT_SUPPORTED, "got %#x\n", hr);
+
+ hr = CoLockObjectExternal((IUnknown *)&Test_Unknown, TRUE, TRUE);
+ ok_ole_success(hr, "CoLockObjectExternal");
+
+ hr = CoDisconnectObject((IUnknown *)&Test_Unknown, 0);
+ ok_ole_success(hr, "CoDisconnectObject");
+
+ return 0;
+}
+
+/* Some COM functions (perhaps even all of them?) can make use of an "implicit"
+ * multi-threaded apartment created by another thread in the same process. */
+static void test_implicit_mta(void)
+{
+ HANDLE thread;
+
+ CoInitializeEx(NULL, COINIT_MULTITHREADED);
+
+ thread = CreateThread(NULL, 0, implicit_mta_proc, NULL, 0, NULL);
+ ok(!WaitForSingleObject(thread, 1000), "wait failed\n");
+
+ CoUninitialize();
+}
+
START_TEST(compobj)
{
init_funcs();
test_IInitializeSpy();
test_CoGetInstanceFromFile();
test_GlobalOptions();
+ test_implicit_mta();
}