2 * Component Object Tests
4 * Copyright 2005 Robert Shearman
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
30 #include "urlmon.h" /* for CLSID_FileProtocol */
32 #include "wine/test.h"
34 /* functions that are not present on all versions of Windows */
35 HRESULT (WINAPI
* pCoInitializeEx
)(LPVOID lpReserved
, DWORD dwCoInit
);
36 HRESULT (WINAPI
* pCoGetObjectContext
)(REFIID riid
, LPVOID
*ppv
);
38 #define ok_ole_success(hr, func) ok(hr == S_OK, func " failed with error 0x%08x\n", hr)
39 #define ok_more_than_one_lock() ok(cLocks > 0, "Number of locks should be > 0, but actually is %d\n", cLocks)
40 #define ok_no_locks() ok(cLocks == 0, "Number of locks should be 0, but actually is %d\n", cLocks)
42 static const CLSID CLSID_non_existent
= { 0x12345678, 0x1234, 0x1234, { 0x12, 0x34, 0x56, 0x78, 0x9a, 0xbc, 0xde, 0xf0 } };
43 static const CLSID CLSID_StdFont
= { 0x0be35203, 0x8f91, 0x11ce, { 0x9d, 0xe3, 0x00, 0xaa, 0x00, 0x4b, 0xb8, 0x51 } };
44 static WCHAR stdfont
[] = {'S','t','d','F','o','n','t',0};
45 static const WCHAR wszNonExistent
[] = {'N','o','n','E','x','i','s','t','e','n','t',0};
46 static WCHAR wszCLSID_StdFont
[] =
48 '{','0','b','e','3','5','2','0','3','-','8','f','9','1','-','1','1','c','e','-',
49 '9','d','e','3','-','0','0','a','a','0','0','4','b','b','8','5','1','}',0
52 static const IID IID_IWineTest
=
57 {0xa1, 0xa2, 0x5d, 0x5a, 0x36, 0x54, 0xd3, 0xbd}
58 }; /* 5201163f-8164-4fd0-a1a2-5d5a3654d3bd */
59 static const CLSID CLSID_WineOOPTest
= {
63 {0xa1, 0xa2, 0x5d, 0x5a, 0x36, 0x54, 0xd3, 0xbd}
64 }; /* 5201163f-8164-4fd0-a1a2-5d5a3654d3bd */
68 static void LockModule(void)
70 InterlockedIncrement(&cLocks
);
73 static void UnlockModule(void)
75 InterlockedDecrement(&cLocks
);
78 static HRESULT WINAPI
Test_IClassFactory_QueryInterface(
83 if (ppvObj
== NULL
) return E_POINTER
;
85 if (IsEqualGUID(riid
, &IID_IUnknown
) ||
86 IsEqualGUID(riid
, &IID_IClassFactory
))
88 *ppvObj
= (LPVOID
)iface
;
89 IClassFactory_AddRef(iface
);
97 static ULONG WINAPI
Test_IClassFactory_AddRef(LPCLASSFACTORY iface
)
100 return 2; /* non-heap-based object */
103 static ULONG WINAPI
Test_IClassFactory_Release(LPCLASSFACTORY iface
)
106 return 1; /* non-heap-based object */
109 static HRESULT WINAPI
Test_IClassFactory_CreateInstance(
110 LPCLASSFACTORY iface
,
116 if (pUnkOuter
) return CLASS_E_NOAGGREGATION
;
117 return E_NOINTERFACE
;
120 static HRESULT WINAPI
Test_IClassFactory_LockServer(
121 LPCLASSFACTORY iface
,
127 static const IClassFactoryVtbl TestClassFactory_Vtbl
=
129 Test_IClassFactory_QueryInterface
,
130 Test_IClassFactory_AddRef
,
131 Test_IClassFactory_Release
,
132 Test_IClassFactory_CreateInstance
,
133 Test_IClassFactory_LockServer
136 static IClassFactory Test_ClassFactory
= { &TestClassFactory_Vtbl
};
138 static void test_ProgIDFromCLSID(void)
141 HRESULT hr
= ProgIDFromCLSID(&CLSID_StdFont
, &progid
);
142 ok(hr
== S_OK
, "ProgIDFromCLSID failed with error 0x%08x\n", hr
);
145 ok(!lstrcmpiW(progid
, stdfont
), "Didn't get expected prog ID\n");
146 CoTaskMemFree(progid
);
149 progid
= (LPWSTR
)0xdeadbeef;
150 hr
= ProgIDFromCLSID(&CLSID_non_existent
, &progid
);
151 ok(hr
== REGDB_E_CLASSNOTREG
, "ProgIDFromCLSID returned %08x\n", hr
);
152 ok(progid
== NULL
, "ProgIDFromCLSID returns with progid %p\n", progid
);
154 hr
= ProgIDFromCLSID(&CLSID_StdFont
, NULL
);
155 ok(hr
== E_INVALIDARG
, "ProgIDFromCLSID should return E_INVALIDARG instead of 0x%08x\n", hr
);
158 static void test_CLSIDFromProgID(void)
161 HRESULT hr
= CLSIDFromProgID(stdfont
, &clsid
);
162 ok(hr
== S_OK
, "CLSIDFromProgID failed with error 0x%08x\n", hr
);
163 ok(IsEqualCLSID(&clsid
, &CLSID_StdFont
), "clsid wasn't equal to CLSID_StdFont\n");
165 hr
= CLSIDFromString(stdfont
, &clsid
);
166 ok_ole_success(hr
, "CLSIDFromString");
167 ok(IsEqualCLSID(&clsid
, &CLSID_StdFont
), "clsid wasn't equal to CLSID_StdFont\n");
169 /* test some failure cases */
171 hr
= CLSIDFromProgID(wszNonExistent
, NULL
);
172 ok(hr
== E_INVALIDARG
, "CLSIDFromProgID should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
174 hr
= CLSIDFromProgID(NULL
, &clsid
);
175 ok(hr
== E_INVALIDARG
, "CLSIDFromProgID should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
177 memset(&clsid
, 0xcc, sizeof(clsid
));
178 hr
= CLSIDFromProgID(wszNonExistent
, &clsid
);
179 ok(hr
== CO_E_CLASSSTRING
, "CLSIDFromProgID on nonexistent ProgID should have returned CO_E_CLASSSTRING instead of 0x%08x\n", hr
);
180 ok(IsEqualCLSID(&clsid
, &CLSID_NULL
), "CLSIDFromProgID should have set clsid to all-zeros on failure\n");
183 static void test_CLSIDFromString(void)
186 HRESULT hr
= CLSIDFromString(wszCLSID_StdFont
, &clsid
);
187 ok_ole_success(hr
, "CLSIDFromString");
188 ok(IsEqualCLSID(&clsid
, &CLSID_StdFont
), "clsid wasn't equal to CLSID_StdFont\n");
190 hr
= CLSIDFromString(NULL
, &clsid
);
191 ok_ole_success(hr
, "CLSIDFromString");
192 ok(IsEqualCLSID(&clsid
, &CLSID_NULL
), "clsid wasn't equal to CLSID_NULL\n");
195 static void test_StringFromGUID2(void)
199 /* Test corner cases for buffer size */
200 len
= StringFromGUID2(&CLSID_StdFont
,str
,50);
201 ok(len
== 39, "len: %d (expected 39)\n", len
);
202 ok(!lstrcmpiW(str
, wszCLSID_StdFont
),"string wasn't equal for CLSID_StdFont\n");
204 memset(str
,0,sizeof str
);
205 len
= StringFromGUID2(&CLSID_StdFont
,str
,39);
206 ok(len
== 39, "len: %d (expected 39)\n", len
);
207 ok(!lstrcmpiW(str
, wszCLSID_StdFont
),"string wasn't equal for CLSID_StdFont\n");
209 len
= StringFromGUID2(&CLSID_StdFont
,str
,38);
210 ok(len
== 0, "len: %d (expected 0)\n", len
);
212 len
= StringFromGUID2(&CLSID_StdFont
,str
,30);
213 ok(len
== 0, "len: %d (expected 0)\n", len
);
216 static void test_CoCreateInstance(void)
218 REFCLSID rclsid
= &CLSID_MyComputer
;
219 IUnknown
*pUnk
= (IUnknown
*)0xdeadbeef;
220 HRESULT hr
= CoCreateInstance(rclsid
, NULL
, CLSCTX_INPROC_SERVER
, &IID_IUnknown
, (void **)&pUnk
);
221 ok(hr
== CO_E_NOTINITIALIZED
, "CoCreateInstance should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr
);
222 ok(pUnk
== NULL
, "CoCreateInstance should have changed the passed in pointer to NULL, instead of %p\n", pUnk
);
225 hr
= CoCreateInstance(rclsid
, NULL
, CLSCTX_INPROC_SERVER
, &IID_IUnknown
, (void **)&pUnk
);
226 ok_ole_success(hr
, "CoCreateInstance");
227 if(pUnk
) IUnknown_Release(pUnk
);
230 hr
= CoCreateInstance(rclsid
, NULL
, CLSCTX_INPROC_SERVER
, &IID_IUnknown
, (void **)&pUnk
);
231 ok(hr
== CO_E_NOTINITIALIZED
, "CoCreateInstance should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr
);
234 static void test_CoGetClassObject(void)
236 IUnknown
*pUnk
= (IUnknown
*)0xdeadbeef;
237 HRESULT hr
= CoGetClassObject(&CLSID_MyComputer
, CLSCTX_INPROC_SERVER
, NULL
, &IID_IUnknown
, (void **)&pUnk
);
238 ok(hr
== CO_E_NOTINITIALIZED
, "CoGetClassObject should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr
);
239 ok(pUnk
== NULL
, "CoGetClassObject should have changed the passed in pointer to NULL, instead of %p\n", pUnk
);
241 hr
= CoGetClassObject(&CLSID_MyComputer
, CLSCTX_INPROC_SERVER
, NULL
, &IID_IUnknown
, NULL
);
242 ok(hr
== E_INVALIDARG
||
243 broken(hr
== CO_E_NOTINITIALIZED
), /* win9x */
244 "CoGetClassObject should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
247 static ATOM
register_dummy_class(void)
255 GetModuleHandle(NULL
),
257 LoadCursor(NULL
, IDC_ARROW
),
258 (HBRUSH
)(COLOR_BTNFACE
+1),
260 TEXT("WineOleTestClass"),
263 return RegisterClass(&wc
);
266 static void test_ole_menu(void)
271 hwndFrame
= CreateWindow(MAKEINTATOM(register_dummy_class()), "Test", 0, CW_USEDEFAULT
, CW_USEDEFAULT
, CW_USEDEFAULT
, CW_USEDEFAULT
, NULL
, NULL
, NULL
, NULL
);
272 hr
= OleSetMenuDescriptor(NULL
, hwndFrame
, NULL
, NULL
, NULL
);
273 todo_wine
ok_ole_success(hr
, "OleSetMenuDescriptor");
275 DestroyWindow(hwndFrame
);
279 static HRESULT WINAPI
MessageFilter_QueryInterface(IMessageFilter
*iface
, REFIID riid
, void ** ppvObj
)
281 if (ppvObj
== NULL
) return E_POINTER
;
283 if (IsEqualGUID(riid
, &IID_IUnknown
) ||
284 IsEqualGUID(riid
, &IID_IClassFactory
))
286 *ppvObj
= (LPVOID
)iface
;
287 IMessageFilter_AddRef(iface
);
291 return E_NOINTERFACE
;
294 static ULONG WINAPI
MessageFilter_AddRef(IMessageFilter
*iface
)
296 return 2; /* non-heap object */
299 static ULONG WINAPI
MessageFilter_Release(IMessageFilter
*iface
)
301 return 1; /* non-heap object */
304 static DWORD WINAPI
MessageFilter_HandleInComingCall(
305 IMessageFilter
*iface
,
307 HTASK threadIDCaller
,
309 LPINTERFACEINFO lpInterfaceInfo
)
311 trace("HandleInComingCall\n");
312 return SERVERCALL_ISHANDLED
;
315 static DWORD WINAPI
MessageFilter_RetryRejectedCall(
316 IMessageFilter
*iface
,
317 HTASK threadIDCallee
,
321 trace("RetryRejectedCall\n");
325 static DWORD WINAPI
MessageFilter_MessagePending(
326 IMessageFilter
*iface
,
327 HTASK threadIDCallee
,
331 trace("MessagePending\n");
332 return PENDINGMSG_WAITNOPROCESS
;
335 static const IMessageFilterVtbl MessageFilter_Vtbl
=
337 MessageFilter_QueryInterface
,
338 MessageFilter_AddRef
,
339 MessageFilter_Release
,
340 MessageFilter_HandleInComingCall
,
341 MessageFilter_RetryRejectedCall
,
342 MessageFilter_MessagePending
345 static IMessageFilter MessageFilter
= { &MessageFilter_Vtbl
};
347 static void test_CoRegisterMessageFilter(void)
350 IMessageFilter
*prev_filter
;
352 hr
= CoRegisterMessageFilter(&MessageFilter
, &prev_filter
);
353 ok(hr
== CO_E_NOT_SUPPORTED
,
354 "CoRegisterMessageFilter should have failed with CO_E_NOT_SUPPORTED instead of 0x%08x\n",
357 pCoInitializeEx(NULL
, COINIT_MULTITHREADED
);
358 prev_filter
= (IMessageFilter
*)0xdeadbeef;
359 hr
= CoRegisterMessageFilter(&MessageFilter
, &prev_filter
);
360 ok(hr
== CO_E_NOT_SUPPORTED
,
361 "CoRegisterMessageFilter should have failed with CO_E_NOT_SUPPORTED instead of 0x%08x\n",
363 ok(prev_filter
== (IMessageFilter
*)0xdeadbeef,
364 "prev_filter should have been set to %p\n", prev_filter
);
367 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
369 hr
= CoRegisterMessageFilter(NULL
, NULL
);
370 ok_ole_success(hr
, "CoRegisterMessageFilter");
372 prev_filter
= (IMessageFilter
*)0xdeadbeef;
373 hr
= CoRegisterMessageFilter(NULL
, &prev_filter
);
374 ok_ole_success(hr
, "CoRegisterMessageFilter");
375 ok(prev_filter
== NULL
, "prev_filter should have been set to NULL instead of %p\n", prev_filter
);
377 hr
= CoRegisterMessageFilter(&MessageFilter
, &prev_filter
);
378 ok_ole_success(hr
, "CoRegisterMessageFilter");
379 ok(prev_filter
== NULL
, "prev_filter should have been set to NULL instead of %p\n", prev_filter
);
381 hr
= CoRegisterMessageFilter(NULL
, NULL
);
382 ok_ole_success(hr
, "CoRegisterMessageFilter");
387 static HRESULT WINAPI
Test_IUnknown_QueryInterface(
392 if (ppvObj
== NULL
) return E_POINTER
;
394 if (IsEqualIID(riid
, &IID_IUnknown
) ||
395 IsEqualIID(riid
, &IID_IWineTest
))
397 *ppvObj
= (LPVOID
)iface
;
398 IUnknown_AddRef(iface
);
403 return E_NOINTERFACE
;
406 static ULONG WINAPI
Test_IUnknown_AddRef(LPUNKNOWN iface
)
408 return 2; /* non-heap-based object */
411 static ULONG WINAPI
Test_IUnknown_Release(LPUNKNOWN iface
)
413 return 1; /* non-heap-based object */
416 static const IUnknownVtbl TestUnknown_Vtbl
=
418 Test_IUnknown_QueryInterface
,
419 Test_IUnknown_AddRef
,
420 Test_IUnknown_Release
,
423 static IUnknown Test_Unknown
= { &TestUnknown_Vtbl
};
425 static HRESULT WINAPI
PSFactoryBuffer_QueryInterface(
426 IPSFactoryBuffer
* This
,
427 /* [in] */ REFIID riid
,
428 /* [iid_is][out] */ void **ppvObject
)
430 if (IsEqualIID(riid
, &IID_IUnknown
) ||
431 IsEqualIID(riid
, &IID_IPSFactoryBuffer
))
434 IPSFactoryBuffer_AddRef(This
);
437 return E_NOINTERFACE
;
440 static ULONG WINAPI
PSFactoryBuffer_AddRef(
441 IPSFactoryBuffer
* This
)
446 static ULONG WINAPI
PSFactoryBuffer_Release(
447 IPSFactoryBuffer
* This
)
452 static HRESULT WINAPI
PSFactoryBuffer_CreateProxy(
453 IPSFactoryBuffer
* This
,
454 /* [in] */ IUnknown
*pUnkOuter
,
455 /* [in] */ REFIID riid
,
456 /* [out] */ IRpcProxyBuffer
**ppProxy
,
457 /* [out] */ void **ppv
)
462 static HRESULT WINAPI
PSFactoryBuffer_CreateStub(
463 IPSFactoryBuffer
* This
,
464 /* [in] */ REFIID riid
,
465 /* [unique][in] */ IUnknown
*pUnkServer
,
466 /* [out] */ IRpcStubBuffer
**ppStub
)
471 static IPSFactoryBufferVtbl PSFactoryBufferVtbl
=
473 PSFactoryBuffer_QueryInterface
,
474 PSFactoryBuffer_AddRef
,
475 PSFactoryBuffer_Release
,
476 PSFactoryBuffer_CreateProxy
,
477 PSFactoryBuffer_CreateStub
480 static IPSFactoryBuffer PSFactoryBuffer
= { &PSFactoryBufferVtbl
};
482 static const CLSID CLSID_WineTestPSFactoryBuffer
=
487 {0xa1, 0xa2, 0x5d, 0x5a, 0x36, 0x54, 0xd3, 0xbd}
488 }; /* 52011640-8164-4fd0-a1a2-5d5a3654d3bd */
490 static void test_CoRegisterPSClsid(void)
493 DWORD dwRegistrationKey
;
497 hr
= CoRegisterPSClsid(&IID_IWineTest
, &CLSID_WineTestPSFactoryBuffer
);
498 ok(hr
== CO_E_NOTINITIALIZED
, "CoRegisterPSClsid should have returened CO_E_NOTINITIALIZED instead of 0x%08x\n", hr
);
500 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
502 hr
= CoRegisterClassObject(&CLSID_WineTestPSFactoryBuffer
, (IUnknown
*)&PSFactoryBuffer
,
503 CLSCTX_INPROC_SERVER
, REGCLS_MULTIPLEUSE
, &dwRegistrationKey
);
504 ok_ole_success(hr
, "CoRegisterClassObject");
506 hr
= CoRegisterPSClsid(&IID_IWineTest
, &CLSID_WineTestPSFactoryBuffer
);
507 ok_ole_success(hr
, "CoRegisterPSClsid");
509 hr
= CreateStreamOnHGlobal(NULL
, TRUE
, &stream
);
510 ok_ole_success(hr
, "CreateStreamOnHGlobal");
512 hr
= CoMarshalInterface(stream
, &IID_IWineTest
, (IUnknown
*)&Test_Unknown
, MSHCTX_INPROC
, NULL
, MSHLFLAGS_NORMAL
);
513 ok(hr
== E_NOTIMPL
, "CoMarshalInterface should have returned E_NOTIMPL instead of 0x%08x\n", hr
);
514 IStream_Release(stream
);
516 hr
= CoRevokeClassObject(dwRegistrationKey
);
517 ok_ole_success(hr
, "CoRevokeClassObject");
521 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
523 hr
= CoGetPSClsid(&IID_IWineTest
, &clsid
);
524 ok(hr
== REGDB_E_IIDNOTREG
, "CoGetPSClsid should have returned REGDB_E_IIDNOTREG instead of 0x%08x\n", hr
);
529 static void test_CoGetPSClsid(void)
534 hr
= CoGetPSClsid(&IID_IClassFactory
, &clsid
);
535 ok(hr
== CO_E_NOTINITIALIZED
,
536 "CoGetPSClsid should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n",
539 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
541 hr
= CoGetPSClsid(&IID_IClassFactory
, &clsid
);
542 ok_ole_success(hr
, "CoGetPSClsid");
544 hr
= CoGetPSClsid(&IID_IWineTest
, &clsid
);
545 ok(hr
== REGDB_E_IIDNOTREG
,
546 "CoGetPSClsid for random IID returned 0x%08x instead of REGDB_E_IIDNOTREG\n",
549 hr
= CoGetPSClsid(&IID_IClassFactory
, NULL
);
550 ok(hr
== E_INVALIDARG
,
551 "CoGetPSClsid for null clsid returned 0x%08x instead of E_INVALIDARG\n",
557 /* basic test, mainly for invalid arguments. see marshal.c for more */
558 static void test_CoUnmarshalInterface(void)
564 hr
= CoUnmarshalInterface(NULL
, &IID_IUnknown
, (void **)&pProxy
);
565 ok(hr
== E_INVALIDARG
, "CoUnmarshalInterface should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
567 hr
= CreateStreamOnHGlobal(NULL
, TRUE
, &pStream
);
568 ok_ole_success(hr
, "CreateStreamOnHGlobal");
570 hr
= CoUnmarshalInterface(pStream
, &IID_IUnknown
, (void **)&pProxy
);
572 ok(hr
== CO_E_NOTINITIALIZED
, "CoUnmarshalInterface should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr
);
574 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
576 hr
= CoUnmarshalInterface(pStream
, &IID_IUnknown
, (void **)&pProxy
);
577 ok(hr
== STG_E_READFAULT
, "CoUnmarshalInterface should have returned STG_E_READFAULT instead of 0x%08x\n", hr
);
581 hr
= CoUnmarshalInterface(pStream
, &IID_IUnknown
, NULL
);
582 ok(hr
== E_INVALIDARG
, "CoUnmarshalInterface should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
584 IStream_Release(pStream
);
587 static void test_CoGetInterfaceAndReleaseStream(void)
592 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
594 hr
= CoGetInterfaceAndReleaseStream(NULL
, &IID_IUnknown
, (void**)&pUnk
);
595 ok(hr
== E_INVALIDARG
, "hr %08x\n", hr
);
600 /* basic test, mainly for invalid arguments. see marshal.c for more */
601 static void test_CoMarshalInterface(void)
605 static const LARGE_INTEGER llZero
;
607 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
609 hr
= CreateStreamOnHGlobal(NULL
, TRUE
, &pStream
);
610 ok_ole_success(hr
, "CreateStreamOnHGlobal");
612 hr
= CoMarshalInterface(pStream
, &IID_IUnknown
, NULL
, MSHCTX_INPROC
, NULL
, MSHLFLAGS_NORMAL
);
613 ok(hr
== E_INVALIDARG
, "CoMarshalInterface should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
615 hr
= CoMarshalInterface(NULL
, &IID_IUnknown
, (IUnknown
*)&Test_ClassFactory
, MSHCTX_INPROC
, NULL
, MSHLFLAGS_NORMAL
);
616 ok(hr
== E_INVALIDARG
, "CoMarshalInterface should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
618 hr
= CoMarshalInterface(pStream
, &IID_IUnknown
, (IUnknown
*)&Test_ClassFactory
, MSHCTX_INPROC
, NULL
, MSHLFLAGS_NORMAL
);
619 ok_ole_success(hr
, "CoMarshalInterface");
621 /* stream not rewound */
622 hr
= CoReleaseMarshalData(pStream
);
623 ok(hr
== STG_E_READFAULT
, "CoReleaseMarshalData should have returned STG_E_READFAULT instead of 0x%08x\n", hr
);
625 hr
= IStream_Seek(pStream
, llZero
, STREAM_SEEK_SET
, NULL
);
626 ok_ole_success(hr
, "IStream_Seek");
628 hr
= CoReleaseMarshalData(pStream
);
629 ok_ole_success(hr
, "CoReleaseMarshalData");
631 IStream_Release(pStream
);
636 static void test_CoMarshalInterThreadInterfaceInStream(void)
640 IClassFactory
*pProxy
;
642 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
646 hr
= CoMarshalInterThreadInterfaceInStream(&IID_IUnknown
, (IUnknown
*)&Test_ClassFactory
, NULL
);
647 ok(hr
== E_INVALIDARG
, "CoMarshalInterThreadInterfaceInStream should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
649 hr
= CoMarshalInterThreadInterfaceInStream(&IID_IUnknown
, NULL
, &pStream
);
650 ok(hr
== E_INVALIDARG
, "CoMarshalInterThreadInterfaceInStream should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
654 hr
= CoMarshalInterThreadInterfaceInStream(&IID_IUnknown
, (IUnknown
*)&Test_ClassFactory
, &pStream
);
655 ok_ole_success(hr
, "CoMarshalInterThreadInterfaceInStream");
657 ok_more_than_one_lock();
659 hr
= CoUnmarshalInterface(pStream
, &IID_IClassFactory
, (void **)&pProxy
);
660 ok_ole_success(hr
, "CoUnmarshalInterface");
662 IClassFactory_Release(pProxy
);
663 IStream_Release(pStream
);
670 static void test_CoRegisterClassObject(void)
676 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
678 /* CLSCTX_INPROC_SERVER */
679 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
680 CLSCTX_INPROC_SERVER
, REGCLS_SINGLEUSE
, &cookie
);
681 ok_ole_success(hr
, "CoRegisterClassObject");
682 hr
= CoRevokeClassObject(cookie
);
683 ok_ole_success(hr
, "CoRevokeClassObject");
685 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
686 CLSCTX_INPROC_SERVER
, REGCLS_MULTIPLEUSE
, &cookie
);
687 ok_ole_success(hr
, "CoRegisterClassObject");
688 hr
= CoRevokeClassObject(cookie
);
689 ok_ole_success(hr
, "CoRevokeClassObject");
691 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
692 CLSCTX_INPROC_SERVER
, REGCLS_MULTI_SEPARATE
, &cookie
);
693 ok_ole_success(hr
, "CoRegisterClassObject");
694 hr
= CoRevokeClassObject(cookie
);
695 ok_ole_success(hr
, "CoRevokeClassObject");
697 /* CLSCTX_LOCAL_SERVER */
698 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
699 CLSCTX_LOCAL_SERVER
, REGCLS_SINGLEUSE
, &cookie
);
700 ok_ole_success(hr
, "CoRegisterClassObject");
701 hr
= CoRevokeClassObject(cookie
);
702 ok_ole_success(hr
, "CoRevokeClassObject");
704 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
705 CLSCTX_LOCAL_SERVER
, REGCLS_MULTIPLEUSE
, &cookie
);
706 ok_ole_success(hr
, "CoRegisterClassObject");
707 hr
= CoRevokeClassObject(cookie
);
708 ok_ole_success(hr
, "CoRevokeClassObject");
710 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
711 CLSCTX_LOCAL_SERVER
, REGCLS_MULTI_SEPARATE
, &cookie
);
712 ok_ole_success(hr
, "CoRegisterClassObject");
713 hr
= CoRevokeClassObject(cookie
);
714 ok_ole_success(hr
, "CoRevokeClassObject");
716 /* CLSCTX_INPROC_SERVER|CLSCTX_LOCAL_SERVER */
717 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
718 CLSCTX_INPROC_SERVER
|CLSCTX_LOCAL_SERVER
, REGCLS_SINGLEUSE
, &cookie
);
719 ok_ole_success(hr
, "CoRegisterClassObject");
720 hr
= CoRevokeClassObject(cookie
);
721 ok_ole_success(hr
, "CoRevokeClassObject");
723 /* test whether registered class becomes invalid when apartment is destroyed */
724 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
725 CLSCTX_INPROC_SERVER
, REGCLS_SINGLEUSE
, &cookie
);
726 ok_ole_success(hr
, "CoRegisterClassObject");
729 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
731 hr
= CoGetClassObject(&CLSID_WineOOPTest
, CLSCTX_INPROC_SERVER
, NULL
,
732 &IID_IClassFactory
, (void **)&pcf
);
733 ok(hr
== REGDB_E_CLASSNOTREG
, "object registered in an apartment shouldn't accessible after it is destroyed\n");
735 /* crashes with at least win9x DCOM! */
737 hr
= CoRevokeClassObject(cookie
);
742 static HRESULT
get_class_object(CLSCTX clsctx
)
747 hr
= CoGetClassObject(&CLSID_WineOOPTest
, clsctx
, NULL
, &IID_IClassFactory
,
751 IClassFactory_Release(pcf
);
756 static DWORD CALLBACK
get_class_object_thread(LPVOID pv
)
758 CLSCTX clsctx
= (CLSCTX
)(DWORD_PTR
)pv
;
761 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
763 hr
= get_class_object(clsctx
);
770 static DWORD CALLBACK
get_class_object_proxy_thread(LPVOID pv
)
772 CLSCTX clsctx
= (CLSCTX
)(DWORD_PTR
)pv
;
777 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
779 hr
= CoGetClassObject(&CLSID_WineOOPTest
, clsctx
, NULL
, &IID_IClassFactory
,
784 hr
= IClassFactory_QueryInterface(pcf
, &IID_IMultiQI
, (void **)&pMQI
);
786 IMultiQI_Release(pMQI
);
787 IClassFactory_Release(pcf
);
795 static DWORD CALLBACK
register_class_object_thread(LPVOID pv
)
800 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
802 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
803 CLSCTX_INPROC_SERVER
, REGCLS_SINGLEUSE
, &cookie
);
810 static DWORD CALLBACK
revoke_class_object_thread(LPVOID pv
)
812 DWORD cookie
= (DWORD_PTR
)pv
;
815 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
817 hr
= CoRevokeClassObject(cookie
);
824 static void test_registered_object_thread_affinity(void)
832 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
834 /* CLSCTX_INPROC_SERVER */
836 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
837 CLSCTX_INPROC_SERVER
, REGCLS_SINGLEUSE
, &cookie
);
838 ok_ole_success(hr
, "CoRegisterClassObject");
840 thread
= CreateThread(NULL
, 0, get_class_object_thread
, (LPVOID
)CLSCTX_INPROC_SERVER
, 0, &tid
);
841 ok(thread
!= NULL
, "CreateThread failed with error %d\n", GetLastError());
842 WaitForSingleObject(thread
, INFINITE
);
843 GetExitCodeThread(thread
, &exitcode
);
845 ok(hr
== REGDB_E_CLASSNOTREG
, "CoGetClassObject on inproc object "
846 "registered in different thread should return REGDB_E_CLASSNOTREG "
847 "instead of 0x%08x\n", hr
);
849 hr
= get_class_object(CLSCTX_INPROC_SERVER
);
850 ok(hr
== S_OK
, "CoGetClassObject on inproc object registered in same "
851 "thread should return S_OK instead of 0x%08x\n", hr
);
853 thread
= CreateThread(NULL
, 0, register_class_object_thread
, NULL
, 0, &tid
);
854 ok(thread
!= NULL
, "CreateThread failed with error %d\n", GetLastError());
855 WaitForSingleObject(thread
, INFINITE
);
856 GetExitCodeThread(thread
, &exitcode
);
858 ok(hr
== S_OK
, "CoRegisterClassObject with same CLSID but in different thread should return S_OK instead of 0x%08x\n", hr
);
860 hr
= CoRevokeClassObject(cookie
);
861 ok_ole_success(hr
, "CoRevokeClassObject");
863 /* CLSCTX_LOCAL_SERVER */
865 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
866 CLSCTX_LOCAL_SERVER
, REGCLS_MULTIPLEUSE
, &cookie
);
867 ok_ole_success(hr
, "CoRegisterClassObject");
869 thread
= CreateThread(NULL
, 0, get_class_object_proxy_thread
, (LPVOID
)CLSCTX_LOCAL_SERVER
, 0, &tid
);
870 ok(thread
!= NULL
, "CreateThread failed with error %d\n", GetLastError());
871 while (MsgWaitForMultipleObjects(1, &thread
, FALSE
, INFINITE
, QS_ALLINPUT
) == WAIT_OBJECT_0
+ 1)
874 while (PeekMessageA(&msg
, NULL
, 0, 0, PM_REMOVE
))
876 TranslateMessage(&msg
);
877 DispatchMessageA(&msg
);
880 GetExitCodeThread(thread
, &exitcode
);
882 ok(hr
== S_OK
, "CoGetClassObject on local server object "
883 "registered in different thread should return S_OK "
884 "instead of 0x%08x\n", hr
);
886 hr
= get_class_object(CLSCTX_LOCAL_SERVER
);
887 ok(hr
== S_OK
, "CoGetClassObject on local server object registered in same "
888 "thread should return S_OK instead of 0x%08x\n", hr
);
890 thread
= CreateThread(NULL
, 0, revoke_class_object_thread
, (LPVOID
)(DWORD_PTR
)cookie
, 0, &tid
);
891 ok(thread
!= NULL
, "CreateThread failed with error %d\n", GetLastError());
892 WaitForSingleObject(thread
, INFINITE
);
893 GetExitCodeThread(thread
, &exitcode
);
895 ok(hr
== RPC_E_WRONG_THREAD
, "CoRevokeClassObject called from different "
896 "thread to where registered should return RPC_E_WRONG_THREAD instead of 0x%08x\n", hr
);
898 thread
= CreateThread(NULL
, 0, register_class_object_thread
, NULL
, 0, &tid
);
899 ok(thread
!= NULL
, "CreateThread failed with error %d\n", GetLastError());
900 WaitForSingleObject(thread
, INFINITE
);
901 GetExitCodeThread(thread
, &exitcode
);
903 ok(hr
== S_OK
, "CoRegisterClassObject with same CLSID but in different "
904 "thread should return S_OK instead of 0x%08x\n", hr
);
906 hr
= CoRevokeClassObject(cookie
);
907 ok_ole_success(hr
, "CoRevokeClassObject");
912 static DWORD CALLBACK
free_libraries_thread(LPVOID p
)
914 CoFreeUnusedLibraries();
918 static inline BOOL
is_module_loaded(const char *module
)
920 return GetModuleHandle(module
) ? TRUE
: FALSE
;
923 static void test_CoFreeUnusedLibraries(void)
930 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
932 ok(!is_module_loaded("urlmon.dll"), "urlmon.dll shouldn't be loaded\n");
934 hr
= CoCreateInstance(&CLSID_FileProtocol
, NULL
, CLSCTX_INPROC_SERVER
, &IID_IInternetProtocol
, (void **)&pUnk
);
935 if (hr
== REGDB_E_CLASSNOTREG
)
937 trace("IE not installed so can't run CoFreeUnusedLibraries test\n");
940 ok_ole_success(hr
, "CoCreateInstance");
942 ok(is_module_loaded("urlmon.dll"), "urlmon.dll should be loaded\n");
945 broken(pUnk
== NULL
), /* win9x */
946 "Expected a valid pointer\n");
948 IUnknown_Release(pUnk
);
950 ok(is_module_loaded("urlmon.dll"), "urlmon.dll should be loaded\n");
952 thread
= CreateThread(NULL
, 0, free_libraries_thread
, NULL
, 0, &tid
);
953 WaitForSingleObject(thread
, INFINITE
);
956 ok(is_module_loaded("urlmon.dll"), "urlmon.dll should be loaded\n");
958 CoFreeUnusedLibraries();
960 ok(!is_module_loaded("urlmon.dll"), "urlmon.dll shouldn't be loaded\n");
965 static void test_CoGetObjectContext(void)
969 IComThreadingInfo
*pComThreadingInfo
;
973 if (!pCoGetObjectContext
)
975 skip("CoGetObjectContext not present\n");
979 hr
= pCoGetObjectContext(&IID_IComThreadingInfo
, (void **)&pComThreadingInfo
);
980 ok(hr
== CO_E_NOTINITIALIZED
, "CoGetObjectContext should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr
);
981 ok(pComThreadingInfo
== NULL
, "pComThreadingInfo should have been set to NULL\n");
983 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
985 hr
= pCoGetObjectContext(&IID_IComThreadingInfo
, (void **)&pComThreadingInfo
);
986 ok_ole_success(hr
, "CoGetObjectContext");
988 hr
= IComThreadingInfo_GetCurrentApartmentType(pComThreadingInfo
, &apttype
);
989 ok_ole_success(hr
, "IComThreadingInfo_GetCurrentApartmentType");
990 ok(apttype
== APTTYPE_MAINSTA
, "apartment type should be APTTYPE_MAINSTA instead of %d\n", apttype
);
992 hr
= IComThreadingInfo_GetCurrentThreadType(pComThreadingInfo
, &thdtype
);
993 ok_ole_success(hr
, "IComThreadingInfo_GetCurrentThreadType");
994 ok(thdtype
== THDTYPE_PROCESSMESSAGES
, "thread type should be THDTYPE_PROCESSMESSAGES instead of %d\n", thdtype
);
996 refs
= IComThreadingInfo_Release(pComThreadingInfo
);
997 ok(refs
== 0, "pComThreadingInfo should have 0 refs instead of %d refs\n", refs
);
1001 pCoInitializeEx(NULL
, COINIT_MULTITHREADED
);
1003 hr
= pCoGetObjectContext(&IID_IComThreadingInfo
, (void **)&pComThreadingInfo
);
1004 ok_ole_success(hr
, "CoGetObjectContext");
1006 hr
= IComThreadingInfo_GetCurrentApartmentType(pComThreadingInfo
, &apttype
);
1007 ok_ole_success(hr
, "IComThreadingInfo_GetCurrentApartmentType");
1008 ok(apttype
== APTTYPE_MTA
, "apartment type should be APTTYPE_MTA instead of %d\n", apttype
);
1010 hr
= IComThreadingInfo_GetCurrentThreadType(pComThreadingInfo
, &thdtype
);
1011 ok_ole_success(hr
, "IComThreadingInfo_GetCurrentThreadType");
1012 ok(thdtype
== THDTYPE_BLOCKMESSAGES
, "thread type should be THDTYPE_BLOCKMESSAGES instead of %d\n", thdtype
);
1014 refs
= IComThreadingInfo_Release(pComThreadingInfo
);
1015 ok(refs
== 0, "pComThreadingInfo should have 0 refs instead of %d refs\n", refs
);
1022 HMODULE hOle32
= GetModuleHandle("ole32");
1023 pCoGetObjectContext
= (void*)GetProcAddress(hOle32
, "CoGetObjectContext");
1024 if (!(pCoInitializeEx
= (void*)GetProcAddress(hOle32
, "CoInitializeEx")))
1026 trace("You need DCOM95 installed to run this test\n");
1030 test_ProgIDFromCLSID();
1031 test_CLSIDFromProgID();
1032 test_CLSIDFromString();
1033 test_StringFromGUID2();
1034 test_CoCreateInstance();
1036 test_CoGetClassObject();
1037 test_CoRegisterMessageFilter();
1038 test_CoRegisterPSClsid();
1039 test_CoGetPSClsid();
1040 test_CoUnmarshalInterface();
1041 test_CoGetInterfaceAndReleaseStream();
1042 test_CoMarshalInterface();
1043 test_CoMarshalInterThreadInterfaceInStream();
1044 test_CoRegisterClassObject();
1045 test_registered_object_thread_affinity();
1046 test_CoFreeUnusedLibraries();
1047 test_CoGetObjectContext();