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 */
35 #include "wine/test.h"
37 /* functions that are not present on all versions of Windows */
38 HRESULT (WINAPI
* pCoInitializeEx
)(LPVOID lpReserved
, DWORD dwCoInit
);
39 HRESULT (WINAPI
* pCoGetObjectContext
)(REFIID riid
, LPVOID
*ppv
);
40 HRESULT (WINAPI
* pCoSwitchCallContext
)(IUnknown
*pObject
, IUnknown
**ppOldObject
);
42 #define ok_ole_success(hr, func) ok(hr == S_OK, func " failed with error 0x%08x\n", hr)
43 #define ok_more_than_one_lock() ok(cLocks > 0, "Number of locks should be > 0, but actually is %d\n", cLocks)
44 #define ok_no_locks() ok(cLocks == 0, "Number of locks should be 0, but actually is %d\n", cLocks)
46 static const CLSID CLSID_non_existent
= { 0x12345678, 0x1234, 0x1234, { 0x12, 0x34, 0x56, 0x78, 0x9a, 0xbc, 0xde, 0xf0 } };
47 static const CLSID CLSID_StdFont
= { 0x0be35203, 0x8f91, 0x11ce, { 0x9d, 0xe3, 0x00, 0xaa, 0x00, 0x4b, 0xb8, 0x51 } };
48 static WCHAR stdfont
[] = {'S','t','d','F','o','n','t',0};
49 static const WCHAR wszNonExistent
[] = {'N','o','n','E','x','i','s','t','e','n','t',0};
50 static WCHAR wszCLSID_StdFont
[] =
52 '{','0','b','e','3','5','2','0','3','-','8','f','9','1','-','1','1','c','e','-',
53 '9','d','e','3','-','0','0','a','a','0','0','4','b','b','8','5','1','}',0
56 static const IID IID_IWineTest
=
61 {0xa1, 0xa2, 0x5d, 0x5a, 0x36, 0x54, 0xd3, 0xbd}
62 }; /* 5201163f-8164-4fd0-a1a2-5d5a3654d3bd */
63 static const CLSID CLSID_WineOOPTest
= {
67 {0xa1, 0xa2, 0x5d, 0x5a, 0x36, 0x54, 0xd3, 0xbd}
68 }; /* 5201163f-8164-4fd0-a1a2-5d5a3654d3bd */
72 static void LockModule(void)
74 InterlockedIncrement(&cLocks
);
77 static void UnlockModule(void)
79 InterlockedDecrement(&cLocks
);
82 static HRESULT WINAPI
Test_IClassFactory_QueryInterface(
87 if (ppvObj
== NULL
) return E_POINTER
;
89 if (IsEqualGUID(riid
, &IID_IUnknown
) ||
90 IsEqualGUID(riid
, &IID_IClassFactory
))
93 IClassFactory_AddRef(iface
);
101 static ULONG WINAPI
Test_IClassFactory_AddRef(LPCLASSFACTORY iface
)
104 return 2; /* non-heap-based object */
107 static ULONG WINAPI
Test_IClassFactory_Release(LPCLASSFACTORY iface
)
110 return 1; /* non-heap-based object */
113 static HRESULT WINAPI
Test_IClassFactory_CreateInstance(
114 LPCLASSFACTORY iface
,
120 if (pUnkOuter
) return CLASS_E_NOAGGREGATION
;
121 return E_NOINTERFACE
;
124 static HRESULT WINAPI
Test_IClassFactory_LockServer(
125 LPCLASSFACTORY iface
,
131 static const IClassFactoryVtbl TestClassFactory_Vtbl
=
133 Test_IClassFactory_QueryInterface
,
134 Test_IClassFactory_AddRef
,
135 Test_IClassFactory_Release
,
136 Test_IClassFactory_CreateInstance
,
137 Test_IClassFactory_LockServer
140 static IClassFactory Test_ClassFactory
= { &TestClassFactory_Vtbl
};
142 static void test_ProgIDFromCLSID(void)
145 HRESULT hr
= ProgIDFromCLSID(&CLSID_StdFont
, &progid
);
146 ok(hr
== S_OK
, "ProgIDFromCLSID failed with error 0x%08x\n", hr
);
149 ok(!lstrcmpiW(progid
, stdfont
), "Didn't get expected prog ID\n");
150 CoTaskMemFree(progid
);
153 progid
= (LPWSTR
)0xdeadbeef;
154 hr
= ProgIDFromCLSID(&CLSID_non_existent
, &progid
);
155 ok(hr
== REGDB_E_CLASSNOTREG
, "ProgIDFromCLSID returned %08x\n", hr
);
156 ok(progid
== NULL
, "ProgIDFromCLSID returns with progid %p\n", progid
);
158 hr
= ProgIDFromCLSID(&CLSID_StdFont
, NULL
);
159 ok(hr
== E_INVALIDARG
, "ProgIDFromCLSID should return E_INVALIDARG instead of 0x%08x\n", hr
);
162 static void test_CLSIDFromProgID(void)
165 HRESULT hr
= CLSIDFromProgID(stdfont
, &clsid
);
166 ok(hr
== S_OK
, "CLSIDFromProgID failed with error 0x%08x\n", hr
);
167 ok(IsEqualCLSID(&clsid
, &CLSID_StdFont
), "clsid wasn't equal to CLSID_StdFont\n");
169 hr
= CLSIDFromString(stdfont
, &clsid
);
170 ok_ole_success(hr
, "CLSIDFromString");
171 ok(IsEqualCLSID(&clsid
, &CLSID_StdFont
), "clsid wasn't equal to CLSID_StdFont\n");
173 /* test some failure cases */
175 hr
= CLSIDFromProgID(wszNonExistent
, NULL
);
176 ok(hr
== E_INVALIDARG
, "CLSIDFromProgID should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
178 hr
= CLSIDFromProgID(NULL
, &clsid
);
179 ok(hr
== E_INVALIDARG
, "CLSIDFromProgID should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
181 memset(&clsid
, 0xcc, sizeof(clsid
));
182 hr
= CLSIDFromProgID(wszNonExistent
, &clsid
);
183 ok(hr
== CO_E_CLASSSTRING
, "CLSIDFromProgID on nonexistent ProgID should have returned CO_E_CLASSSTRING instead of 0x%08x\n", hr
);
184 ok(IsEqualCLSID(&clsid
, &CLSID_NULL
), "CLSIDFromProgID should have set clsid to all-zeros on failure\n");
187 static void test_CLSIDFromString(void)
190 HRESULT hr
= CLSIDFromString(wszCLSID_StdFont
, &clsid
);
191 ok_ole_success(hr
, "CLSIDFromString");
192 ok(IsEqualCLSID(&clsid
, &CLSID_StdFont
), "clsid wasn't equal to CLSID_StdFont\n");
194 hr
= CLSIDFromString(NULL
, &clsid
);
195 ok_ole_success(hr
, "CLSIDFromString");
196 ok(IsEqualCLSID(&clsid
, &CLSID_NULL
), "clsid wasn't equal to CLSID_NULL\n");
199 static void test_StringFromGUID2(void)
203 /* Test corner cases for buffer size */
204 len
= StringFromGUID2(&CLSID_StdFont
,str
,50);
205 ok(len
== 39, "len: %d (expected 39)\n", len
);
206 ok(!lstrcmpiW(str
, wszCLSID_StdFont
),"string wasn't equal for CLSID_StdFont\n");
208 memset(str
,0,sizeof str
);
209 len
= StringFromGUID2(&CLSID_StdFont
,str
,39);
210 ok(len
== 39, "len: %d (expected 39)\n", len
);
211 ok(!lstrcmpiW(str
, wszCLSID_StdFont
),"string wasn't equal for CLSID_StdFont\n");
213 len
= StringFromGUID2(&CLSID_StdFont
,str
,38);
214 ok(len
== 0, "len: %d (expected 0)\n", len
);
216 len
= StringFromGUID2(&CLSID_StdFont
,str
,30);
217 ok(len
== 0, "len: %d (expected 0)\n", len
);
220 static void test_CoCreateInstance(void)
222 REFCLSID rclsid
= &CLSID_MyComputer
;
223 IUnknown
*pUnk
= (IUnknown
*)0xdeadbeef;
224 HRESULT hr
= CoCreateInstance(rclsid
, NULL
, CLSCTX_INPROC_SERVER
, &IID_IUnknown
, (void **)&pUnk
);
225 ok(hr
== CO_E_NOTINITIALIZED
, "CoCreateInstance should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr
);
226 ok(pUnk
== NULL
, "CoCreateInstance should have changed the passed in pointer to NULL, instead of %p\n", pUnk
);
229 hr
= CoCreateInstance(rclsid
, NULL
, CLSCTX_INPROC_SERVER
, &IID_IUnknown
, (void **)&pUnk
);
230 ok_ole_success(hr
, "CoCreateInstance");
231 if(pUnk
) IUnknown_Release(pUnk
);
234 hr
= CoCreateInstance(rclsid
, NULL
, CLSCTX_INPROC_SERVER
, &IID_IUnknown
, (void **)&pUnk
);
235 ok(hr
== CO_E_NOTINITIALIZED
, "CoCreateInstance should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr
);
238 static void test_CoGetClassObject(void)
240 IUnknown
*pUnk
= (IUnknown
*)0xdeadbeef;
241 HRESULT hr
= CoGetClassObject(&CLSID_MyComputer
, CLSCTX_INPROC_SERVER
, NULL
, &IID_IUnknown
, (void **)&pUnk
);
242 ok(hr
== CO_E_NOTINITIALIZED
, "CoGetClassObject should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr
);
243 ok(pUnk
== NULL
, "CoGetClassObject should have changed the passed in pointer to NULL, instead of %p\n", pUnk
);
245 hr
= CoGetClassObject(&CLSID_MyComputer
, CLSCTX_INPROC_SERVER
, NULL
, &IID_IUnknown
, NULL
);
246 ok(hr
== E_INVALIDARG
||
247 broken(hr
== CO_E_NOTINITIALIZED
), /* win9x */
248 "CoGetClassObject should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
251 static ATOM
register_dummy_class(void)
259 GetModuleHandle(NULL
),
261 LoadCursor(NULL
, IDC_ARROW
),
262 (HBRUSH
)(COLOR_BTNFACE
+1),
264 TEXT("WineOleTestClass"),
267 return RegisterClass(&wc
);
270 static void test_ole_menu(void)
275 hwndFrame
= CreateWindow(MAKEINTATOM(register_dummy_class()), "Test", 0, CW_USEDEFAULT
, CW_USEDEFAULT
, CW_USEDEFAULT
, CW_USEDEFAULT
, NULL
, NULL
, NULL
, NULL
);
276 hr
= OleSetMenuDescriptor(NULL
, hwndFrame
, NULL
, NULL
, NULL
);
277 todo_wine
ok_ole_success(hr
, "OleSetMenuDescriptor");
279 DestroyWindow(hwndFrame
);
283 static HRESULT WINAPI
MessageFilter_QueryInterface(IMessageFilter
*iface
, REFIID riid
, void ** ppvObj
)
285 if (ppvObj
== NULL
) return E_POINTER
;
287 if (IsEqualGUID(riid
, &IID_IUnknown
) ||
288 IsEqualGUID(riid
, &IID_IClassFactory
))
291 IMessageFilter_AddRef(iface
);
295 return E_NOINTERFACE
;
298 static ULONG WINAPI
MessageFilter_AddRef(IMessageFilter
*iface
)
300 return 2; /* non-heap object */
303 static ULONG WINAPI
MessageFilter_Release(IMessageFilter
*iface
)
305 return 1; /* non-heap object */
308 static DWORD WINAPI
MessageFilter_HandleInComingCall(
309 IMessageFilter
*iface
,
311 HTASK threadIDCaller
,
313 LPINTERFACEINFO lpInterfaceInfo
)
315 trace("HandleInComingCall\n");
316 return SERVERCALL_ISHANDLED
;
319 static DWORD WINAPI
MessageFilter_RetryRejectedCall(
320 IMessageFilter
*iface
,
321 HTASK threadIDCallee
,
325 trace("RetryRejectedCall\n");
329 static DWORD WINAPI
MessageFilter_MessagePending(
330 IMessageFilter
*iface
,
331 HTASK threadIDCallee
,
335 trace("MessagePending\n");
336 return PENDINGMSG_WAITNOPROCESS
;
339 static const IMessageFilterVtbl MessageFilter_Vtbl
=
341 MessageFilter_QueryInterface
,
342 MessageFilter_AddRef
,
343 MessageFilter_Release
,
344 MessageFilter_HandleInComingCall
,
345 MessageFilter_RetryRejectedCall
,
346 MessageFilter_MessagePending
349 static IMessageFilter MessageFilter
= { &MessageFilter_Vtbl
};
351 static void test_CoRegisterMessageFilter(void)
354 IMessageFilter
*prev_filter
;
356 hr
= CoRegisterMessageFilter(&MessageFilter
, &prev_filter
);
357 ok(hr
== CO_E_NOT_SUPPORTED
,
358 "CoRegisterMessageFilter should have failed with CO_E_NOT_SUPPORTED instead of 0x%08x\n",
361 pCoInitializeEx(NULL
, COINIT_MULTITHREADED
);
362 prev_filter
= (IMessageFilter
*)0xdeadbeef;
363 hr
= CoRegisterMessageFilter(&MessageFilter
, &prev_filter
);
364 ok(hr
== CO_E_NOT_SUPPORTED
,
365 "CoRegisterMessageFilter should have failed with CO_E_NOT_SUPPORTED instead of 0x%08x\n",
367 ok(prev_filter
== (IMessageFilter
*)0xdeadbeef,
368 "prev_filter should have been set to %p\n", prev_filter
);
371 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
373 hr
= CoRegisterMessageFilter(NULL
, NULL
);
374 ok_ole_success(hr
, "CoRegisterMessageFilter");
376 prev_filter
= (IMessageFilter
*)0xdeadbeef;
377 hr
= CoRegisterMessageFilter(NULL
, &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(&MessageFilter
, &prev_filter
);
382 ok_ole_success(hr
, "CoRegisterMessageFilter");
383 ok(prev_filter
== NULL
, "prev_filter should have been set to NULL instead of %p\n", prev_filter
);
385 hr
= CoRegisterMessageFilter(NULL
, NULL
);
386 ok_ole_success(hr
, "CoRegisterMessageFilter");
391 static HRESULT WINAPI
Test_IUnknown_QueryInterface(
396 if (ppvObj
== NULL
) return E_POINTER
;
398 if (IsEqualIID(riid
, &IID_IUnknown
) ||
399 IsEqualIID(riid
, &IID_IWineTest
))
402 IUnknown_AddRef(iface
);
407 return E_NOINTERFACE
;
410 static ULONG WINAPI
Test_IUnknown_AddRef(LPUNKNOWN iface
)
412 return 2; /* non-heap-based object */
415 static ULONG WINAPI
Test_IUnknown_Release(LPUNKNOWN iface
)
417 return 1; /* non-heap-based object */
420 static const IUnknownVtbl TestUnknown_Vtbl
=
422 Test_IUnknown_QueryInterface
,
423 Test_IUnknown_AddRef
,
424 Test_IUnknown_Release
,
427 static IUnknown Test_Unknown
= { &TestUnknown_Vtbl
};
429 static HRESULT WINAPI
PSFactoryBuffer_QueryInterface(
430 IPSFactoryBuffer
* This
,
431 /* [in] */ REFIID riid
,
432 /* [iid_is][out] */ void **ppvObject
)
434 if (IsEqualIID(riid
, &IID_IUnknown
) ||
435 IsEqualIID(riid
, &IID_IPSFactoryBuffer
))
438 IPSFactoryBuffer_AddRef(This
);
441 return E_NOINTERFACE
;
444 static ULONG WINAPI
PSFactoryBuffer_AddRef(
445 IPSFactoryBuffer
* This
)
450 static ULONG WINAPI
PSFactoryBuffer_Release(
451 IPSFactoryBuffer
* This
)
456 static HRESULT WINAPI
PSFactoryBuffer_CreateProxy(
457 IPSFactoryBuffer
* This
,
458 /* [in] */ IUnknown
*pUnkOuter
,
459 /* [in] */ REFIID riid
,
460 /* [out] */ IRpcProxyBuffer
**ppProxy
,
461 /* [out] */ void **ppv
)
466 static HRESULT WINAPI
PSFactoryBuffer_CreateStub(
467 IPSFactoryBuffer
* This
,
468 /* [in] */ REFIID riid
,
469 /* [unique][in] */ IUnknown
*pUnkServer
,
470 /* [out] */ IRpcStubBuffer
**ppStub
)
475 static IPSFactoryBufferVtbl PSFactoryBufferVtbl
=
477 PSFactoryBuffer_QueryInterface
,
478 PSFactoryBuffer_AddRef
,
479 PSFactoryBuffer_Release
,
480 PSFactoryBuffer_CreateProxy
,
481 PSFactoryBuffer_CreateStub
484 static IPSFactoryBuffer PSFactoryBuffer
= { &PSFactoryBufferVtbl
};
486 static const CLSID CLSID_WineTestPSFactoryBuffer
=
491 {0xa1, 0xa2, 0x5d, 0x5a, 0x36, 0x54, 0xd3, 0xbd}
492 }; /* 52011640-8164-4fd0-a1a2-5d5a3654d3bd */
494 static void test_CoRegisterPSClsid(void)
497 DWORD dwRegistrationKey
;
501 hr
= CoRegisterPSClsid(&IID_IWineTest
, &CLSID_WineTestPSFactoryBuffer
);
502 ok(hr
== CO_E_NOTINITIALIZED
, "CoRegisterPSClsid should have returened CO_E_NOTINITIALIZED instead of 0x%08x\n", hr
);
504 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
506 hr
= CoRegisterClassObject(&CLSID_WineTestPSFactoryBuffer
, (IUnknown
*)&PSFactoryBuffer
,
507 CLSCTX_INPROC_SERVER
, REGCLS_MULTIPLEUSE
, &dwRegistrationKey
);
508 ok_ole_success(hr
, "CoRegisterClassObject");
510 hr
= CoRegisterPSClsid(&IID_IWineTest
, &CLSID_WineTestPSFactoryBuffer
);
511 ok_ole_success(hr
, "CoRegisterPSClsid");
513 hr
= CreateStreamOnHGlobal(NULL
, TRUE
, &stream
);
514 ok_ole_success(hr
, "CreateStreamOnHGlobal");
516 hr
= CoMarshalInterface(stream
, &IID_IWineTest
, &Test_Unknown
, MSHCTX_INPROC
, NULL
, MSHLFLAGS_NORMAL
);
517 ok(hr
== E_NOTIMPL
, "CoMarshalInterface should have returned E_NOTIMPL instead of 0x%08x\n", hr
);
518 IStream_Release(stream
);
520 hr
= CoRevokeClassObject(dwRegistrationKey
);
521 ok_ole_success(hr
, "CoRevokeClassObject");
525 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
527 hr
= CoGetPSClsid(&IID_IWineTest
, &clsid
);
528 ok(hr
== REGDB_E_IIDNOTREG
, "CoGetPSClsid should have returned REGDB_E_IIDNOTREG instead of 0x%08x\n", hr
);
533 static void test_CoGetPSClsid(void)
538 hr
= CoGetPSClsid(&IID_IClassFactory
, &clsid
);
539 ok(hr
== CO_E_NOTINITIALIZED
,
540 "CoGetPSClsid should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n",
543 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
545 hr
= CoGetPSClsid(&IID_IClassFactory
, &clsid
);
546 ok_ole_success(hr
, "CoGetPSClsid");
548 hr
= CoGetPSClsid(&IID_IWineTest
, &clsid
);
549 ok(hr
== REGDB_E_IIDNOTREG
,
550 "CoGetPSClsid for random IID returned 0x%08x instead of REGDB_E_IIDNOTREG\n",
553 hr
= CoGetPSClsid(&IID_IClassFactory
, NULL
);
554 ok(hr
== E_INVALIDARG
,
555 "CoGetPSClsid for null clsid returned 0x%08x instead of E_INVALIDARG\n",
561 /* basic test, mainly for invalid arguments. see marshal.c for more */
562 static void test_CoUnmarshalInterface(void)
568 hr
= CoUnmarshalInterface(NULL
, &IID_IUnknown
, (void **)&pProxy
);
569 ok(hr
== E_INVALIDARG
, "CoUnmarshalInterface should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
571 hr
= CreateStreamOnHGlobal(NULL
, TRUE
, &pStream
);
572 ok_ole_success(hr
, "CreateStreamOnHGlobal");
574 hr
= CoUnmarshalInterface(pStream
, &IID_IUnknown
, (void **)&pProxy
);
576 ok(hr
== CO_E_NOTINITIALIZED
, "CoUnmarshalInterface should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr
);
578 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
580 hr
= CoUnmarshalInterface(pStream
, &IID_IUnknown
, (void **)&pProxy
);
581 ok(hr
== STG_E_READFAULT
, "CoUnmarshalInterface should have returned STG_E_READFAULT instead of 0x%08x\n", hr
);
585 hr
= CoUnmarshalInterface(pStream
, &IID_IUnknown
, NULL
);
586 ok(hr
== E_INVALIDARG
, "CoUnmarshalInterface should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
588 IStream_Release(pStream
);
591 static void test_CoGetInterfaceAndReleaseStream(void)
596 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
598 hr
= CoGetInterfaceAndReleaseStream(NULL
, &IID_IUnknown
, (void**)&pUnk
);
599 ok(hr
== E_INVALIDARG
, "hr %08x\n", hr
);
604 /* basic test, mainly for invalid arguments. see marshal.c for more */
605 static void test_CoMarshalInterface(void)
609 static const LARGE_INTEGER llZero
;
611 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
613 hr
= CreateStreamOnHGlobal(NULL
, TRUE
, &pStream
);
614 ok_ole_success(hr
, "CreateStreamOnHGlobal");
616 hr
= CoMarshalInterface(pStream
, &IID_IUnknown
, NULL
, MSHCTX_INPROC
, NULL
, MSHLFLAGS_NORMAL
);
617 ok(hr
== E_INVALIDARG
, "CoMarshalInterface should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
619 hr
= CoMarshalInterface(NULL
, &IID_IUnknown
, (IUnknown
*)&Test_ClassFactory
, MSHCTX_INPROC
, NULL
, MSHLFLAGS_NORMAL
);
620 ok(hr
== E_INVALIDARG
, "CoMarshalInterface should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
622 hr
= CoMarshalInterface(pStream
, &IID_IUnknown
, (IUnknown
*)&Test_ClassFactory
, MSHCTX_INPROC
, NULL
, MSHLFLAGS_NORMAL
);
623 ok_ole_success(hr
, "CoMarshalInterface");
625 /* stream not rewound */
626 hr
= CoReleaseMarshalData(pStream
);
627 ok(hr
== STG_E_READFAULT
, "CoReleaseMarshalData should have returned STG_E_READFAULT instead of 0x%08x\n", hr
);
629 hr
= IStream_Seek(pStream
, llZero
, STREAM_SEEK_SET
, NULL
);
630 ok_ole_success(hr
, "IStream_Seek");
632 hr
= CoReleaseMarshalData(pStream
);
633 ok_ole_success(hr
, "CoReleaseMarshalData");
635 IStream_Release(pStream
);
640 static void test_CoMarshalInterThreadInterfaceInStream(void)
644 IClassFactory
*pProxy
;
646 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
650 hr
= CoMarshalInterThreadInterfaceInStream(&IID_IUnknown
, (IUnknown
*)&Test_ClassFactory
, NULL
);
651 ok(hr
== E_INVALIDARG
, "CoMarshalInterThreadInterfaceInStream should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
653 hr
= CoMarshalInterThreadInterfaceInStream(&IID_IUnknown
, NULL
, &pStream
);
654 ok(hr
== E_INVALIDARG
, "CoMarshalInterThreadInterfaceInStream should have returned E_INVALIDARG instead of 0x%08x\n", hr
);
658 hr
= CoMarshalInterThreadInterfaceInStream(&IID_IUnknown
, (IUnknown
*)&Test_ClassFactory
, &pStream
);
659 ok_ole_success(hr
, "CoMarshalInterThreadInterfaceInStream");
661 ok_more_than_one_lock();
663 hr
= CoUnmarshalInterface(pStream
, &IID_IClassFactory
, (void **)&pProxy
);
664 ok_ole_success(hr
, "CoUnmarshalInterface");
666 IClassFactory_Release(pProxy
);
667 IStream_Release(pStream
);
674 static void test_CoRegisterClassObject(void)
680 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
682 /* CLSCTX_INPROC_SERVER */
683 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
684 CLSCTX_INPROC_SERVER
, REGCLS_SINGLEUSE
, &cookie
);
685 ok_ole_success(hr
, "CoRegisterClassObject");
686 hr
= CoRevokeClassObject(cookie
);
687 ok_ole_success(hr
, "CoRevokeClassObject");
689 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
690 CLSCTX_INPROC_SERVER
, REGCLS_MULTIPLEUSE
, &cookie
);
691 ok_ole_success(hr
, "CoRegisterClassObject");
692 hr
= CoRevokeClassObject(cookie
);
693 ok_ole_success(hr
, "CoRevokeClassObject");
695 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
696 CLSCTX_INPROC_SERVER
, REGCLS_MULTI_SEPARATE
, &cookie
);
697 ok_ole_success(hr
, "CoRegisterClassObject");
698 hr
= CoRevokeClassObject(cookie
);
699 ok_ole_success(hr
, "CoRevokeClassObject");
701 /* CLSCTX_LOCAL_SERVER */
702 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
703 CLSCTX_LOCAL_SERVER
, REGCLS_SINGLEUSE
, &cookie
);
704 ok_ole_success(hr
, "CoRegisterClassObject");
705 hr
= CoRevokeClassObject(cookie
);
706 ok_ole_success(hr
, "CoRevokeClassObject");
708 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
709 CLSCTX_LOCAL_SERVER
, REGCLS_MULTIPLEUSE
, &cookie
);
710 ok_ole_success(hr
, "CoRegisterClassObject");
711 hr
= CoRevokeClassObject(cookie
);
712 ok_ole_success(hr
, "CoRevokeClassObject");
714 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
715 CLSCTX_LOCAL_SERVER
, REGCLS_MULTI_SEPARATE
, &cookie
);
716 ok_ole_success(hr
, "CoRegisterClassObject");
717 hr
= CoRevokeClassObject(cookie
);
718 ok_ole_success(hr
, "CoRevokeClassObject");
720 /* CLSCTX_INPROC_SERVER|CLSCTX_LOCAL_SERVER */
721 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
722 CLSCTX_INPROC_SERVER
|CLSCTX_LOCAL_SERVER
, REGCLS_SINGLEUSE
, &cookie
);
723 ok_ole_success(hr
, "CoRegisterClassObject");
724 hr
= CoRevokeClassObject(cookie
);
725 ok_ole_success(hr
, "CoRevokeClassObject");
727 /* test whether registered class becomes invalid when apartment is destroyed */
728 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
729 CLSCTX_INPROC_SERVER
, REGCLS_SINGLEUSE
, &cookie
);
730 ok_ole_success(hr
, "CoRegisterClassObject");
733 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
735 hr
= CoGetClassObject(&CLSID_WineOOPTest
, CLSCTX_INPROC_SERVER
, NULL
,
736 &IID_IClassFactory
, (void **)&pcf
);
737 ok(hr
== REGDB_E_CLASSNOTREG
, "object registered in an apartment shouldn't accessible after it is destroyed\n");
739 /* crashes with at least win9x DCOM! */
741 hr
= CoRevokeClassObject(cookie
);
746 static HRESULT
get_class_object(CLSCTX clsctx
)
751 hr
= CoGetClassObject(&CLSID_WineOOPTest
, clsctx
, NULL
, &IID_IClassFactory
,
755 IClassFactory_Release(pcf
);
760 static DWORD CALLBACK
get_class_object_thread(LPVOID pv
)
762 CLSCTX clsctx
= (CLSCTX
)(DWORD_PTR
)pv
;
765 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
767 hr
= get_class_object(clsctx
);
774 static DWORD CALLBACK
get_class_object_proxy_thread(LPVOID pv
)
776 CLSCTX clsctx
= (CLSCTX
)(DWORD_PTR
)pv
;
781 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
783 hr
= CoGetClassObject(&CLSID_WineOOPTest
, clsctx
, NULL
, &IID_IClassFactory
,
788 hr
= IClassFactory_QueryInterface(pcf
, &IID_IMultiQI
, (void **)&pMQI
);
790 IMultiQI_Release(pMQI
);
791 IClassFactory_Release(pcf
);
799 static DWORD CALLBACK
register_class_object_thread(LPVOID pv
)
804 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
806 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
807 CLSCTX_INPROC_SERVER
, REGCLS_SINGLEUSE
, &cookie
);
814 static DWORD CALLBACK
revoke_class_object_thread(LPVOID pv
)
816 DWORD cookie
= (DWORD_PTR
)pv
;
819 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
821 hr
= CoRevokeClassObject(cookie
);
828 static void test_registered_object_thread_affinity(void)
836 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
838 /* CLSCTX_INPROC_SERVER */
840 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
841 CLSCTX_INPROC_SERVER
, REGCLS_SINGLEUSE
, &cookie
);
842 ok_ole_success(hr
, "CoRegisterClassObject");
844 thread
= CreateThread(NULL
, 0, get_class_object_thread
, (LPVOID
)CLSCTX_INPROC_SERVER
, 0, &tid
);
845 ok(thread
!= NULL
, "CreateThread failed with error %d\n", GetLastError());
846 WaitForSingleObject(thread
, INFINITE
);
847 GetExitCodeThread(thread
, &exitcode
);
849 ok(hr
== REGDB_E_CLASSNOTREG
, "CoGetClassObject on inproc object "
850 "registered in different thread should return REGDB_E_CLASSNOTREG "
851 "instead of 0x%08x\n", hr
);
853 hr
= get_class_object(CLSCTX_INPROC_SERVER
);
854 ok(hr
== S_OK
, "CoGetClassObject on inproc object registered in same "
855 "thread should return S_OK instead of 0x%08x\n", hr
);
857 thread
= CreateThread(NULL
, 0, register_class_object_thread
, NULL
, 0, &tid
);
858 ok(thread
!= NULL
, "CreateThread failed with error %d\n", GetLastError());
859 WaitForSingleObject(thread
, INFINITE
);
860 GetExitCodeThread(thread
, &exitcode
);
862 ok(hr
== S_OK
, "CoRegisterClassObject with same CLSID but in different thread should return S_OK instead of 0x%08x\n", hr
);
864 hr
= CoRevokeClassObject(cookie
);
865 ok_ole_success(hr
, "CoRevokeClassObject");
867 /* CLSCTX_LOCAL_SERVER */
869 hr
= CoRegisterClassObject(&CLSID_WineOOPTest
, (IUnknown
*)&Test_ClassFactory
,
870 CLSCTX_LOCAL_SERVER
, REGCLS_MULTIPLEUSE
, &cookie
);
871 ok_ole_success(hr
, "CoRegisterClassObject");
873 thread
= CreateThread(NULL
, 0, get_class_object_proxy_thread
, (LPVOID
)CLSCTX_LOCAL_SERVER
, 0, &tid
);
874 ok(thread
!= NULL
, "CreateThread failed with error %d\n", GetLastError());
875 while (MsgWaitForMultipleObjects(1, &thread
, FALSE
, INFINITE
, QS_ALLINPUT
) == WAIT_OBJECT_0
+ 1)
878 while (PeekMessageA(&msg
, NULL
, 0, 0, PM_REMOVE
))
880 TranslateMessage(&msg
);
881 DispatchMessageA(&msg
);
884 GetExitCodeThread(thread
, &exitcode
);
886 ok(hr
== S_OK
, "CoGetClassObject on local server object "
887 "registered in different thread should return S_OK "
888 "instead of 0x%08x\n", hr
);
890 hr
= get_class_object(CLSCTX_LOCAL_SERVER
);
891 ok(hr
== S_OK
, "CoGetClassObject on local server object registered in same "
892 "thread should return S_OK instead of 0x%08x\n", hr
);
894 thread
= CreateThread(NULL
, 0, revoke_class_object_thread
, (LPVOID
)(DWORD_PTR
)cookie
, 0, &tid
);
895 ok(thread
!= NULL
, "CreateThread failed with error %d\n", GetLastError());
896 WaitForSingleObject(thread
, INFINITE
);
897 GetExitCodeThread(thread
, &exitcode
);
899 ok(hr
== RPC_E_WRONG_THREAD
, "CoRevokeClassObject called from different "
900 "thread to where registered should return RPC_E_WRONG_THREAD instead of 0x%08x\n", hr
);
902 thread
= CreateThread(NULL
, 0, register_class_object_thread
, NULL
, 0, &tid
);
903 ok(thread
!= NULL
, "CreateThread failed with error %d\n", GetLastError());
904 WaitForSingleObject(thread
, INFINITE
);
905 GetExitCodeThread(thread
, &exitcode
);
907 ok(hr
== S_OK
, "CoRegisterClassObject with same CLSID but in different "
908 "thread should return S_OK instead of 0x%08x\n", hr
);
910 hr
= CoRevokeClassObject(cookie
);
911 ok_ole_success(hr
, "CoRevokeClassObject");
916 static DWORD CALLBACK
free_libraries_thread(LPVOID p
)
918 CoFreeUnusedLibraries();
922 static inline BOOL
is_module_loaded(const char *module
)
924 return GetModuleHandle(module
) ? TRUE
: FALSE
;
927 static void test_CoFreeUnusedLibraries(void)
934 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
936 ok(!is_module_loaded("urlmon.dll"), "urlmon.dll shouldn't be loaded\n");
938 hr
= CoCreateInstance(&CLSID_FileProtocol
, NULL
, CLSCTX_INPROC_SERVER
, &IID_IInternetProtocol
, (void **)&pUnk
);
939 if (hr
== REGDB_E_CLASSNOTREG
)
941 trace("IE not installed so can't run CoFreeUnusedLibraries test\n");
945 ok_ole_success(hr
, "CoCreateInstance");
947 ok(is_module_loaded("urlmon.dll"), "urlmon.dll should be loaded\n");
950 broken(pUnk
== NULL
), /* win9x */
951 "Expected a valid pointer\n");
953 IUnknown_Release(pUnk
);
955 ok(is_module_loaded("urlmon.dll"), "urlmon.dll should be loaded\n");
957 thread
= CreateThread(NULL
, 0, free_libraries_thread
, NULL
, 0, &tid
);
958 WaitForSingleObject(thread
, INFINITE
);
961 ok(is_module_loaded("urlmon.dll"), "urlmon.dll should be loaded\n");
963 CoFreeUnusedLibraries();
965 ok(!is_module_loaded("urlmon.dll"), "urlmon.dll shouldn't be loaded\n");
970 static void test_CoGetObjectContext(void)
974 IComThreadingInfo
*pComThreadingInfo
;
975 IContextCallback
*pContextCallback
;
979 if (!pCoGetObjectContext
)
981 skip("CoGetObjectContext not present\n");
985 hr
= pCoGetObjectContext(&IID_IComThreadingInfo
, (void **)&pComThreadingInfo
);
986 ok(hr
== CO_E_NOTINITIALIZED
, "CoGetObjectContext should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr
);
987 ok(pComThreadingInfo
== NULL
, "pComThreadingInfo should have been set to NULL\n");
989 pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
991 hr
= pCoGetObjectContext(&IID_IComThreadingInfo
, (void **)&pComThreadingInfo
);
992 ok_ole_success(hr
, "CoGetObjectContext");
994 hr
= IComThreadingInfo_GetCurrentApartmentType(pComThreadingInfo
, &apttype
);
995 ok_ole_success(hr
, "IComThreadingInfo_GetCurrentApartmentType");
996 ok(apttype
== APTTYPE_MAINSTA
, "apartment type should be APTTYPE_MAINSTA instead of %d\n", apttype
);
998 hr
= IComThreadingInfo_GetCurrentThreadType(pComThreadingInfo
, &thdtype
);
999 ok_ole_success(hr
, "IComThreadingInfo_GetCurrentThreadType");
1000 ok(thdtype
== THDTYPE_PROCESSMESSAGES
, "thread type should be THDTYPE_PROCESSMESSAGES instead of %d\n", thdtype
);
1002 refs
= IComThreadingInfo_Release(pComThreadingInfo
);
1003 ok(refs
== 0, "pComThreadingInfo should have 0 refs instead of %d refs\n", refs
);
1005 hr
= pCoGetObjectContext(&IID_IContextCallback
, (void **)&pContextCallback
);
1006 ok_ole_success(hr
, "CoGetObjectContext(ContextCallback)");
1010 refs
= IContextCallback_Release(pContextCallback
);
1011 ok(refs
== 0, "pContextCallback should have 0 refs instead of %d refs\n", refs
);
1016 pCoInitializeEx(NULL
, COINIT_MULTITHREADED
);
1018 hr
= pCoGetObjectContext(&IID_IComThreadingInfo
, (void **)&pComThreadingInfo
);
1019 ok_ole_success(hr
, "CoGetObjectContext");
1021 hr
= IComThreadingInfo_GetCurrentApartmentType(pComThreadingInfo
, &apttype
);
1022 ok_ole_success(hr
, "IComThreadingInfo_GetCurrentApartmentType");
1023 ok(apttype
== APTTYPE_MTA
, "apartment type should be APTTYPE_MTA instead of %d\n", apttype
);
1025 hr
= IComThreadingInfo_GetCurrentThreadType(pComThreadingInfo
, &thdtype
);
1026 ok_ole_success(hr
, "IComThreadingInfo_GetCurrentThreadType");
1027 ok(thdtype
== THDTYPE_BLOCKMESSAGES
, "thread type should be THDTYPE_BLOCKMESSAGES instead of %d\n", thdtype
);
1029 refs
= IComThreadingInfo_Release(pComThreadingInfo
);
1030 ok(refs
== 0, "pComThreadingInfo should have 0 refs instead of %d refs\n", refs
);
1032 hr
= pCoGetObjectContext(&IID_IContextCallback
, (void **)&pContextCallback
);
1033 ok_ole_success(hr
, "CoGetObjectContext(ContextCallback)");
1037 refs
= IContextCallback_Release(pContextCallback
);
1038 ok(refs
== 0, "pContextCallback should have 0 refs instead of %d refs\n", refs
);
1045 const IUnknownVtbl
*lpVtbl
;
1049 static HRESULT WINAPI
Test_CallContext_QueryInterface(
1054 if (ppvObj
== NULL
) return E_POINTER
;
1056 if (IsEqualGUID(riid
, &IID_IUnknown
))
1059 IUnknown_AddRef(iface
);
1064 return E_NOINTERFACE
;
1067 static ULONG WINAPI
Test_CallContext_AddRef(IUnknown
*iface
)
1069 Test_CallContext
*This
= (Test_CallContext
*)iface
;
1070 return InterlockedIncrement(&This
->refs
);
1073 static ULONG WINAPI
Test_CallContext_Release(IUnknown
*iface
)
1075 Test_CallContext
*This
= (Test_CallContext
*)iface
;
1076 ULONG refs
= InterlockedDecrement(&This
->refs
);
1078 HeapFree(GetProcessHeap(), 0, This
);
1082 static const IUnknownVtbl TestCallContext_Vtbl
=
1084 Test_CallContext_QueryInterface
,
1085 Test_CallContext_AddRef
,
1086 Test_CallContext_Release
1089 static void test_CoGetCallContext(void)
1094 IUnknown
*test_object
;
1096 if (!pCoSwitchCallContext
)
1098 skip("CoSwitchCallContext not present\n");
1104 test_object
= HeapAlloc(GetProcessHeap(), 0, sizeof(Test_CallContext
));
1105 ((Test_CallContext
*)test_object
)->lpVtbl
= &TestCallContext_Vtbl
;
1106 ((Test_CallContext
*)test_object
)->refs
= 1;
1108 hr
= CoGetCallContext(&IID_IUnknown
, (void**)&pUnk
);
1109 ok(hr
== RPC_E_CALL_COMPLETE
, "Expected RPC_E_CALL_COMPLETE, got 0x%08x\n", hr
);
1111 pUnk
= (IUnknown
*)0xdeadbeef;
1112 hr
= pCoSwitchCallContext(test_object
, &pUnk
);
1113 ok_ole_success(hr
, "CoSwitchCallContext");
1114 ok(pUnk
== NULL
, "expected NULL, got %p\n", pUnk
);
1115 refs
= IUnknown_AddRef(test_object
);
1116 ok(refs
== 2, "Expected refcount 2, got %d\n", refs
);
1117 IUnknown_Release(test_object
);
1119 pUnk
= (IUnknown
*)0xdeadbeef;
1120 hr
= CoGetCallContext(&IID_IUnknown
, (void**)&pUnk
);
1121 ok_ole_success(hr
, "CoGetCallContext");
1122 ok(pUnk
== test_object
, "expected %p, got %p\n", test_object
, pUnk
);
1123 refs
= IUnknown_AddRef(test_object
);
1124 ok(refs
== 3, "Expected refcount 3, got %d\n", refs
);
1125 IUnknown_Release(test_object
);
1126 IUnknown_Release(pUnk
);
1128 pUnk
= (IUnknown
*)0xdeadbeef;
1129 hr
= pCoSwitchCallContext(NULL
, &pUnk
);
1130 ok_ole_success(hr
, "CoSwitchCallContext");
1131 ok(pUnk
== test_object
, "expected %p, got %p\n", test_object
, pUnk
);
1132 refs
= IUnknown_AddRef(test_object
);
1133 ok(refs
== 2, "Expected refcount 2, got %d\n", refs
);
1134 IUnknown_Release(test_object
);
1136 hr
= CoGetCallContext(&IID_IUnknown
, (void**)&pUnk
);
1137 ok(hr
== RPC_E_CALL_COMPLETE
, "Expected RPC_E_CALL_COMPLETE, got 0x%08x\n", hr
);
1139 IUnknown_Release(test_object
);
1144 static void test_CoInitializeEx(void)
1148 hr
= pCoInitializeEx(NULL
, COINIT_APARTMENTTHREADED
);
1149 ok(hr
== S_OK
, "CoInitializeEx failed with error 0x%08x\n", hr
);
1151 /* Calling OleInitialize for the first time should yield S_OK even with
1152 * apartment already initialized by previous CoInitialize(Ex) calls. */
1153 hr
= OleInitialize(NULL
);
1154 todo_wine
ok(hr
== S_OK
, "OleInitialize failed with error 0x%08x\n", hr
);
1156 /* Subsequent calls to OleInitialize should return S_FALSE */
1157 hr
= OleInitialize(NULL
);
1158 ok(hr
== S_FALSE
, "Expected S_FALSE, hr = 0x%08x\n", hr
);
1167 HMODULE hOle32
= GetModuleHandle("ole32");
1168 pCoGetObjectContext
= (void*)GetProcAddress(hOle32
, "CoGetObjectContext");
1169 pCoSwitchCallContext
= (void*)GetProcAddress(hOle32
, "CoSwitchCallContext");
1170 if (!(pCoInitializeEx
= (void*)GetProcAddress(hOle32
, "CoInitializeEx")))
1172 trace("You need DCOM95 installed to run this test\n");
1176 test_ProgIDFromCLSID();
1177 test_CLSIDFromProgID();
1178 test_CLSIDFromString();
1179 test_StringFromGUID2();
1180 test_CoCreateInstance();
1182 test_CoGetClassObject();
1183 test_CoRegisterMessageFilter();
1184 test_CoRegisterPSClsid();
1185 test_CoGetPSClsid();
1186 test_CoUnmarshalInterface();
1187 test_CoGetInterfaceAndReleaseStream();
1188 test_CoMarshalInterface();
1189 test_CoMarshalInterThreadInterfaceInStream();
1190 test_CoRegisterClassObject();
1191 test_registered_object_thread_affinity();
1192 test_CoFreeUnusedLibraries();
1193 test_CoGetObjectContext();
1194 test_CoGetCallContext();
1195 test_CoInitializeEx();