dmloader: Simplify the module refcount handling.
[wine] / dlls / oleaut32 / tests / tmarshal.c
1 /*
2  * Copyright (C) 2005-2006 Robert Shearman for CodeWeavers
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  *
18  */
19
20 #define COBJMACROS
21 #define CONST_VTABLE
22
23 #include <windows.h>
24 #include <ocidl.h>
25 #include <stdio.h>
26
27 #include "wine/test.h"
28
29 #include "tmarshal.h"
30 #include "tmarshal_dispids.h"
31
32 #define ok_ole_success(hr, func) ok(hr == S_OK, #func " failed with error 0x%08lx\n", (unsigned long int)hr)
33
34 /* ULL suffix is not portable */
35 #define ULL_CONST(dw1, dw2) ((((ULONGLONG)dw1) << 32) | (ULONGLONG)dw2)
36
37 const MYSTRUCT MYSTRUCT_BYVAL = {0x12345678, ULL_CONST(0xdeadbeef, 0x98765432)};
38 const MYSTRUCT MYSTRUCT_BYPTR = {0x91827364, ULL_CONST(0x88776655, 0x44332211)};
39 const MYSTRUCT MYSTRUCT_ARRAY[5] = {
40     {0x1a1b1c1d, ULL_CONST(0x1e1f1011, 0x12131415)},
41     {0x2a2b2c2d, ULL_CONST(0x2e2f2021, 0x22232425)},
42     {0x3a3b3c3d, ULL_CONST(0x3e3f3031, 0x32333435)},
43     {0x4a4b4c4d, ULL_CONST(0x4e4f4041, 0x42434445)},
44     {0x5a5b5c5d, ULL_CONST(0x5e5f5051, 0x52535455)},
45 };
46
47
48 #define RELEASEMARSHALDATA WM_USER
49
50 struct host_object_data
51 {
52     IStream *stream;
53     IID iid;
54     IUnknown *object;
55     MSHLFLAGS marshal_flags;
56     HANDLE marshal_event;
57     IMessageFilter *filter;
58 };
59
60 static DWORD CALLBACK host_object_proc(LPVOID p)
61 {
62     struct host_object_data *data = p;
63     HRESULT hr;
64     MSG msg;
65
66     CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
67
68     if (data->filter)
69     {
70         IMessageFilter * prev_filter = NULL;
71         hr = CoRegisterMessageFilter(data->filter, &prev_filter);
72         if (prev_filter) IMessageFilter_Release(prev_filter);
73         ok_ole_success(hr, CoRegisterMessageFilter);
74     }
75
76     hr = CoMarshalInterface(data->stream, &data->iid, data->object, MSHCTX_INPROC, NULL, data->marshal_flags);
77     ok_ole_success(hr, CoMarshalInterface);
78
79     /* force the message queue to be created before signaling parent thread */
80     PeekMessage(&msg, NULL, WM_USER, WM_USER, PM_NOREMOVE);
81
82     SetEvent(data->marshal_event);
83
84     while (GetMessage(&msg, NULL, 0, 0))
85     {
86         if (msg.hwnd == NULL && msg.message == RELEASEMARSHALDATA)
87         {
88             trace("releasing marshal data\n");
89             CoReleaseMarshalData(data->stream);
90             SetEvent((HANDLE)msg.lParam);
91         }
92         else
93             DispatchMessage(&msg);
94     }
95
96     HeapFree(GetProcessHeap(), 0, data);
97
98     CoUninitialize();
99
100     return hr;
101 }
102
103 static DWORD start_host_object2(IStream *stream, REFIID riid, IUnknown *object, MSHLFLAGS marshal_flags, IMessageFilter *filter, HANDLE *thread)
104 {
105     DWORD tid = 0;
106     HANDLE marshal_event = CreateEvent(NULL, FALSE, FALSE, NULL);
107     struct host_object_data *data = HeapAlloc(GetProcessHeap(), 0, sizeof(*data));
108
109     data->stream = stream;
110     data->iid = *riid;
111     data->object = object;
112     data->marshal_flags = marshal_flags;
113     data->marshal_event = marshal_event;
114     data->filter = filter;
115
116     *thread = CreateThread(NULL, 0, host_object_proc, data, 0, &tid);
117
118     /* wait for marshaling to complete before returning */
119     WaitForSingleObject(marshal_event, INFINITE);
120     CloseHandle(marshal_event);
121
122     return tid;
123 }
124
125 static DWORD start_host_object(IStream *stream, REFIID riid, IUnknown *object, MSHLFLAGS marshal_flags, HANDLE *thread)
126 {
127     return start_host_object2(stream, riid, object, marshal_flags, NULL, thread);
128 }
129
130 #if 0 /* not used */
131 /* asks thread to release the marshal data because it has to be done by the
132  * same thread that marshaled the interface in the first place. */
133 static void release_host_object(DWORD tid)
134 {
135     HANDLE event = CreateEvent(NULL, FALSE, FALSE, NULL);
136     PostThreadMessage(tid, RELEASEMARSHALDATA, 0, (LPARAM)event);
137     WaitForSingleObject(event, INFINITE);
138     CloseHandle(event);
139 }
140 #endif
141
142 static void end_host_object(DWORD tid, HANDLE thread)
143 {
144     BOOL ret = PostThreadMessage(tid, WM_QUIT, 0, 0);
145     ok(ret, "PostThreadMessage failed with error %d\n", GetLastError());
146     /* be careful of races - don't return until hosting thread has terminated */
147     WaitForSingleObject(thread, INFINITE);
148     CloseHandle(thread);
149 }
150
151 static ItestDual TestDual, TestDualDisp;
152
153 static HRESULT WINAPI TestDual_QueryInterface(ItestDual *iface, REFIID riid, void **ppvObject)
154 {
155     if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IDispatch)) {
156         *ppvObject = &TestDualDisp;
157         return S_OK;
158     }else if(IsEqualGUID(riid, &IID_ItestDual)) {
159         *ppvObject = &TestDual;
160         return S_OK;
161     }
162
163     *ppvObject = NULL;
164     return E_NOINTERFACE;
165 }
166
167 static ULONG WINAPI TestDual_AddRef(ItestDual *iface)
168 {
169     return 2;
170 }
171
172 static ULONG WINAPI TestDual_Release(ItestDual *iface)
173 {
174     return 1;
175 }
176
177 static HRESULT WINAPI TestDual_GetTypeInfoCount(ItestDual *iface, UINT *pctinfo)
178 {
179     ok(0, "unexpected call\n");
180     return E_NOTIMPL;
181 }
182
183 static HRESULT WINAPI TestDual_GetTypeInfo(ItestDual *iface, UINT iTInfo, LCID lcid, ITypeInfo **ppTInfo)
184 {
185     ok(0, "unexpected call\n");
186     return E_NOTIMPL;
187 }
188
189 static HRESULT WINAPI TestDual_GetIDsOfNames(ItestDual *iface, REFIID riid, LPOLESTR *rgszNames,
190         UINT cNames, LCID lcid, DISPID *rgDispId)
191 {
192     ok(0, "unexpected call\n");
193     return E_NOTIMPL;
194 }
195
196 static HRESULT WINAPI TestDual_Invoke(ItestDual *iface, DISPID dispIdMember, REFIID riid, LCID lcid,
197         WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult, EXCEPINFO *pExcepInfo,
198         UINT *puArgErr)
199 {
200     ok(0, "unexpected call\n");
201     return E_NOTIMPL;
202 }
203
204 static ItestDualVtbl TestDualVtbl = {
205     TestDual_QueryInterface,
206     TestDual_AddRef,
207     TestDual_Release,
208     TestDual_GetTypeInfoCount,
209     TestDual_GetTypeInfo,
210     TestDual_GetIDsOfNames,
211     TestDual_Invoke
212 };
213
214 static ItestDual TestDual = { &TestDualVtbl };
215 static ItestDual TestDualDisp = { &TestDualVtbl };
216
217 typedef struct Widget
218 {
219     IWidget IWidget_iface;
220     LONG refs;
221     IUnknown *pDispatchUnknown;
222 } Widget;
223
224 static inline Widget *impl_from_IWidget(IWidget *iface)
225 {
226     return CONTAINING_RECORD(iface, Widget, IWidget_iface);
227 }
228
229 static HRESULT WINAPI Widget_QueryInterface(
230     IWidget *iface,
231     /* [in] */ REFIID riid,
232     /* [iid_is][out] */ void __RPC_FAR *__RPC_FAR *ppvObject)
233 {
234     if (IsEqualIID(riid, &IID_IWidget) || IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IDispatch))
235     {
236         IWidget_AddRef(iface);
237         *ppvObject = iface;
238         return S_OK;
239     }
240     else
241     {
242         *ppvObject = NULL;
243         return E_NOINTERFACE;
244     }
245 }
246
247 static ULONG WINAPI Widget_AddRef(
248     IWidget *iface)
249 {
250     Widget *This = impl_from_IWidget(iface);
251
252     return InterlockedIncrement(&This->refs);
253 }
254
255 static ULONG WINAPI Widget_Release(
256     IWidget *iface)
257 {
258     Widget *This = impl_from_IWidget(iface);
259     ULONG refs = InterlockedDecrement(&This->refs);
260     if (!refs)
261     {
262         IUnknown_Release(This->pDispatchUnknown);
263         memset(This, 0xcc, sizeof(*This));
264         HeapFree(GetProcessHeap(), 0, This);
265         trace("Widget destroyed!\n");
266     }
267
268     return refs;
269 }
270
271 static HRESULT WINAPI Widget_GetTypeInfoCount(
272     IWidget *iface,
273     /* [out] */ UINT __RPC_FAR *pctinfo)
274 {
275     Widget *This = impl_from_IWidget(iface);
276     IDispatch *pDispatch;
277     HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
278     if (SUCCEEDED(hr))
279     {
280         hr = IDispatch_GetTypeInfoCount(pDispatch, pctinfo);
281         IDispatch_Release(pDispatch);
282     }
283     return hr;
284 }
285
286 static HRESULT WINAPI Widget_GetTypeInfo(
287     IWidget __RPC_FAR * iface,
288     /* [in] */ UINT iTInfo,
289     /* [in] */ LCID lcid,
290     /* [out] */ ITypeInfo __RPC_FAR *__RPC_FAR *ppTInfo)
291 {
292     Widget *This = impl_from_IWidget(iface);
293     IDispatch *pDispatch;
294     HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
295     if (SUCCEEDED(hr))
296     {
297         hr = IDispatch_GetTypeInfo(pDispatch, iTInfo, lcid, ppTInfo);
298         IDispatch_Release(pDispatch);
299     }
300     return hr;
301 }
302
303 static HRESULT WINAPI Widget_GetIDsOfNames(
304     IWidget __RPC_FAR * iface,
305     /* [in] */ REFIID riid,
306     /* [size_is][in] */ LPOLESTR __RPC_FAR *rgszNames,
307     /* [in] */ UINT cNames,
308     /* [in] */ LCID lcid,
309     /* [size_is][out] */ DISPID __RPC_FAR *rgDispId)
310 {
311     Widget *This = impl_from_IWidget(iface);
312     IDispatch *pDispatch;
313     HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
314     if (SUCCEEDED(hr))
315     {
316         hr = IDispatch_GetIDsOfNames(pDispatch, riid, rgszNames, cNames, lcid, rgDispId);
317         IDispatch_Release(pDispatch);
318     }
319     return hr;
320 }
321
322 static HRESULT WINAPI Widget_Invoke(
323     IWidget __RPC_FAR * iface,
324     /* [in] */ DISPID dispIdMember,
325     /* [in] */ REFIID riid,
326     /* [in] */ LCID lcid,
327     /* [in] */ WORD wFlags,
328     /* [out][in] */ DISPPARAMS __RPC_FAR *pDispParams,
329     /* [out] */ VARIANT __RPC_FAR *pVarResult,
330     /* [out] */ EXCEPINFO __RPC_FAR *pExcepInfo,
331     /* [out] */ UINT __RPC_FAR *puArgErr)
332 {
333     Widget *This = impl_from_IWidget(iface);
334     IDispatch *pDispatch;
335     HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
336     if (SUCCEEDED(hr))
337     {
338         hr = IDispatch_Invoke(pDispatch, dispIdMember, riid, lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
339         IDispatch_Release(pDispatch);
340     }
341     return hr;
342 }
343
344 static HRESULT WINAPI Widget_put_Name(
345     IWidget __RPC_FAR * iface,
346     /* [in] */ BSTR name)
347 {
348     trace("put_Name(%s)\n", wine_dbgstr_w(name));
349     return S_OK;
350 }
351
352 static HRESULT WINAPI Widget_get_Name(
353     IWidget __RPC_FAR * iface,
354     /* [out] */ BSTR __RPC_FAR *name)
355 {
356     static const WCHAR szCat[] = { 'C','a','t',0 };
357     trace("get_Name()\n");
358     *name = SysAllocString(szCat);
359     return S_OK;
360 }
361
362 static HRESULT WINAPI Widget_DoSomething(
363     IWidget __RPC_FAR * iface,
364     /* [in] */ double number,
365     /* [out] */ BSTR *str1,
366     /* [defaultvalue][in] */ BSTR str2,
367     /* [optional][in] */ VARIANT __RPC_FAR *opt)
368 {
369     static const WCHAR szString[] = { 'S','t','r','i','n','g',0 };
370     trace("DoSomething()\n");
371
372     ok(number == 3.141, "number(%f) != 3.141\n", number);
373     ok(*str2 == '\0', "str2(%s) != \"\"\n", wine_dbgstr_w(str2));
374     ok(V_VT(opt) == VT_ERROR, "V_VT(opt) should be VT_ERROR instead of 0x%x\n", V_VT(opt));
375     ok(V_ERROR(opt) == DISP_E_PARAMNOTFOUND, "V_ERROR(opt) should be DISP_E_PARAMNOTFOUND instead of 0x%08x\n", V_ERROR(opt));
376     *str1 = SysAllocString(szString);
377
378     return S_FALSE;
379 }
380
381 static HRESULT WINAPI Widget_get_State(
382     IWidget __RPC_FAR * iface,
383     /* [retval][out] */ STATE __RPC_FAR *state)
384 {
385     trace("get_State() = STATE_WIDGETIFIED\n");
386     *state = STATE_WIDGETIFIED;
387     return S_OK;
388 }
389
390 static HRESULT WINAPI Widget_put_State(
391     IWidget __RPC_FAR * iface,
392     /* [in] */ STATE state)
393 {
394     trace("put_State(%d)\n", state);
395     return S_OK;
396 }
397
398 static HRESULT WINAPI Widget_Map(
399     IWidget * iface,
400     BSTR bstrId,
401     BSTR *sValue)
402 {
403     trace("Map(%s, %p)\n", wine_dbgstr_w(bstrId), sValue);
404     *sValue = SysAllocString(bstrId);
405     return S_OK;
406 }
407
408 static HRESULT WINAPI Widget_SetOleColor(
409     IWidget * iface,
410     OLE_COLOR val)
411 {
412     trace("SetOleColor(0x%x)\n", val);
413     return S_OK;
414 }
415
416 static HRESULT WINAPI Widget_GetOleColor(
417     IWidget * iface,
418     OLE_COLOR *pVal)
419 {
420     trace("GetOleColor() = 0x8000000f\n");
421     *pVal = 0x8000000f;
422     return S_FALSE;
423 }
424
425 static HRESULT WINAPI Widget_Clone(
426     IWidget *iface,
427     IWidget **ppVal)
428 {
429     trace("Clone()\n");
430     return Widget_QueryInterface(iface, &IID_IWidget, (void **)ppVal);
431 }
432
433 static HRESULT WINAPI Widget_CloneDispatch(
434     IWidget *iface,
435     IDispatch **ppVal)
436 {
437     trace("CloneDispatch()\n");
438     return Widget_QueryInterface(iface, &IID_IWidget, (void **)ppVal);
439 }
440
441 static HRESULT WINAPI Widget_CloneCoclass(
442     IWidget *iface,
443     ApplicationObject2 **ppVal)
444 {
445     trace("CloneCoclass()\n");
446     return S_OK;
447 }
448
449 static HRESULT WINAPI Widget_Value(
450     IWidget __RPC_FAR * iface,
451     VARIANT *value,
452     VARIANT *retval)
453 {
454     trace("Value(%p, %p)\n", value, retval);
455     ok(V_VT(value) == VT_I2, "V_VT(value) was %d instead of VT_I2\n", V_VT(value));
456     ok(V_I2(value) == 1, "V_I2(value) was %d instead of 1\n", V_I2(value));
457     V_VT(retval) = VT_I2;
458     V_I2(retval) = 1234;
459     return S_OK;
460 }
461
462 static HRESULT WINAPI Widget_Array(
463     IWidget * iface,
464     SAFEARRAY * values)
465 {
466     trace("Array(%p)\n", values);
467     return S_OK;
468 }
469
470 static HRESULT WINAPI Widget_VariantArrayPtr(
471     IWidget * iface,
472     SAFEARRAY ** values)
473 {
474     trace("VariantArrayPtr(%p)\n", values);
475     return S_OK;
476 }
477
478 static HRESULT WINAPI Widget_Variant(
479     IWidget __RPC_FAR * iface,
480     VARIANT var)
481 {
482     trace("Variant()\n");
483     ok(V_VT(&var) == VT_CY, "V_VT(&var) was %d\n", V_VT(&var));
484     ok(S(V_CY(&var)).Hi == 0xdababe, "V_CY(&var).Hi was 0x%x\n", S(V_CY(&var)).Hi);
485     ok(S(V_CY(&var)).Lo == 0xdeadbeef, "V_CY(&var).Lo was 0x%x\n", S(V_CY(&var)).Lo);
486     return S_OK;
487 }
488
489 static HRESULT WINAPI Widget_VarArg(
490     IWidget * iface,
491     int numexpect,
492     SAFEARRAY * values)
493 {
494     LONG lbound, ubound, i;
495     VARIANT * data;
496     HRESULT hr;
497
498     trace("VarArg(%p)\n", values);
499
500     hr = SafeArrayGetLBound(values, 1, &lbound);
501     ok(hr == S_OK, "SafeArrayGetLBound failed with %x\n", hr);
502     ok(lbound == 0, "SafeArrayGetLBound returned %d\n", lbound);
503
504     hr = SafeArrayGetUBound(values, 1, &ubound);
505     ok(hr == S_OK, "SafeArrayGetUBound failed with %x\n", hr);
506     ok(ubound == numexpect-1, "SafeArrayGetUBound returned %d, but expected %d\n", ubound, numexpect-1);
507
508     hr = SafeArrayAccessData(values, (LPVOID)&data);
509     ok(hr == S_OK, "SafeArrayAccessData failed with %x\n", hr);
510
511     for (i=0; i<=ubound-lbound; i++)
512     {
513         ok(V_VT(&data[i]) == VT_I4, "V_VT(&data[%d]) was %d\n", i, V_VT(&data[i]));
514         ok(V_I4(&data[i]) == i, "V_I4(&data[%d]) was %d\n", i, V_I4(&data[i]));
515     }
516
517     hr = SafeArrayUnaccessData(values);
518     ok(hr == S_OK, "SafeArrayUnaccessData failed with %x\n", hr);
519
520     return S_OK;
521 }
522
523 static HRESULT WINAPI Widget_StructArgs(
524     IWidget * iface,
525     MYSTRUCT byval,
526     MYSTRUCT *byptr,
527     MYSTRUCT arr[5])
528 {
529     int i, diff = 0;
530     ok(byval.field1 == MYSTRUCT_BYVAL.field1 &&
531        byval.field2 == MYSTRUCT_BYVAL.field2,
532        "Struct parameter passed by value corrupted\n");
533     ok(byptr->field1 == MYSTRUCT_BYPTR.field1 &&
534        byptr->field2 == MYSTRUCT_BYPTR.field2,
535        "Struct parameter passed by pointer corrupted\n");
536     for (i = 0; i < 5; i++)
537         if (arr[i].field1 != MYSTRUCT_ARRAY[i].field1 ||
538             arr[i].field2 != MYSTRUCT_ARRAY[i].field2)
539             diff++;
540     ok(diff == 0, "Array of structs corrupted\n");
541     return S_OK;
542 }
543
544
545 static HRESULT WINAPI Widget_Error(
546     IWidget __RPC_FAR * iface)
547 {
548     trace("Error()\n");
549     return E_NOTIMPL;
550 }
551
552 static HRESULT WINAPI Widget_CloneInterface(
553     IWidget __RPC_FAR * iface,
554     ISomethingFromDispatch **ppVal)
555 {
556     trace("CloneInterface()\n");
557     *ppVal = 0;
558     return S_OK;
559 }
560
561 static HRESULT WINAPI Widget_put_prop_with_lcid(
562     IWidget* iface, LONG lcid, INT i)
563 {
564     trace("put_prop_with_lcid(%08x, %x)\n", lcid, i);
565     ok(lcid == MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_US), "got lcid %08x\n", lcid);
566     ok(i == 0xcafe, "got %08x\n", i);
567     return S_OK;
568 }
569
570 static HRESULT WINAPI Widget_get_prop_with_lcid(
571     IWidget* iface, LONG lcid, INT *i)
572 {
573     trace("get_prop_with_lcid(%08x, %p)\n", lcid, i);
574     ok(lcid == MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_US), "got lcid %08x\n", lcid);
575     *i = lcid;
576     return S_OK;
577 }
578
579 static HRESULT WINAPI Widget_get_prop_int(
580     IWidget* iface, INT *i)
581 {
582     trace("get_prop_int(%p)\n", i);
583     *i = -13;
584     return S_OK;
585 }
586
587 static HRESULT WINAPI Widget_get_prop_uint(
588     IWidget* iface, UINT *i)
589 {
590     trace("get_prop_uint(%p)\n", i);
591     *i = 42;
592     return S_OK;
593 }
594
595 static HRESULT WINAPI Widget_ByRefUInt(
596     IWidget* iface, UINT *i)
597 {
598     *i = 42;
599     return S_OK;
600 }
601
602 static HRESULT WINAPI Widget_put_prop_opt_arg(
603     IWidget* iface, INT opt, INT i)
604 {
605     trace("put_prop_opt_arg(%08x, %08x)\n", opt, i);
606     todo_wine ok(opt == 0, "got opt=%08x\n", opt);
607     ok(i == 0xcafe, "got i=%08x\n", i);
608     return S_OK;
609 }
610
611 static HRESULT WINAPI Widget_put_prop_req_arg(
612     IWidget* iface, INT req, INT i)
613 {
614     trace("put_prop_req_arg(%08x, %08x)\n", req, i);
615     ok(req == 0x5678, "got req=%08x\n", req);
616     ok(i == 0x1234, "got i=%08x\n", i);
617     return S_OK;
618 }
619
620 static const struct IWidgetVtbl Widget_VTable =
621 {
622     Widget_QueryInterface,
623     Widget_AddRef,
624     Widget_Release,
625     Widget_GetTypeInfoCount,
626     Widget_GetTypeInfo,
627     Widget_GetIDsOfNames,
628     Widget_Invoke,
629     Widget_put_Name,
630     Widget_get_Name,
631     Widget_DoSomething,
632     Widget_get_State,
633     Widget_put_State,
634     Widget_Map,
635     Widget_SetOleColor,
636     Widget_GetOleColor,
637     Widget_Clone,
638     Widget_CloneDispatch,
639     Widget_CloneCoclass,
640     Widget_Value,
641     Widget_Array,
642     Widget_VariantArrayPtr,
643     Widget_Variant,
644     Widget_VarArg,
645     Widget_StructArgs,
646     Widget_Error,
647     Widget_CloneInterface,
648     Widget_put_prop_with_lcid,
649     Widget_get_prop_with_lcid,
650     Widget_get_prop_int,
651     Widget_get_prop_uint,
652     Widget_ByRefUInt,
653     Widget_put_prop_opt_arg,
654     Widget_put_prop_req_arg,
655 };
656
657 static HRESULT WINAPI StaticWidget_QueryInterface(IStaticWidget *iface, REFIID riid, void **ppvObject)
658 {
659     if (IsEqualIID(riid, &IID_IStaticWidget) || IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IDispatch))
660     {
661         IStaticWidget_AddRef(iface);
662         *ppvObject = iface;
663         return S_OK;
664     }
665
666     *ppvObject = NULL;
667     return E_NOINTERFACE;
668 }
669
670 static ULONG WINAPI StaticWidget_AddRef(IStaticWidget *iface)
671 {
672     return 2;
673 }
674
675 static ULONG WINAPI StaticWidget_Release(IStaticWidget *iface)
676 {
677     return 1;
678 }
679
680 static HRESULT WINAPI StaticWidget_GetTypeInfoCount(IStaticWidget *iface, UINT *pctinfo)
681 {
682     ok(0, "unexpected call\n");
683     return E_NOTIMPL;
684 }
685
686 static HRESULT WINAPI StaticWidget_GetTypeInfo(IStaticWidget *iface, UINT iTInfo, LCID lcid,
687         ITypeInfo **ppTInfo)
688 {
689     ok(0, "unexpected call\n");
690     return E_NOTIMPL;
691 }
692
693 static HRESULT WINAPI StaticWidget_GetIDsOfNames(IStaticWidget *iface, REFIID riid, LPOLESTR *rgszNames,
694         UINT cNames, LCID lcid, DISPID *rgDispId)
695 {
696     ok(0, "unexpected call\n");
697     return E_NOTIMPL;
698 }
699
700 static HRESULT WINAPI StaticWidget_Invoke(IStaticWidget *iface, DISPID dispIdMember, REFIID riid,
701         LCID lcid, WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult, EXCEPINFO *pExcepInfo,
702          UINT *puArgErr)
703 {
704     ok(0, "unexpected call\n");
705     return E_NOTIMPL;
706 }
707
708 static HRESULT WINAPI StaticWidget_TestDual(IStaticWidget *iface, ItestDual *p)
709 {
710     trace("TestDual()\n");
711     todo_wine
712     ok(p == &TestDual, "wrong ItestDual\n");
713     return S_OK;
714 }
715
716 static const IStaticWidgetVtbl StaticWidgetVtbl = {
717     StaticWidget_QueryInterface,
718     StaticWidget_AddRef,
719     StaticWidget_Release,
720     StaticWidget_GetTypeInfoCount,
721     StaticWidget_GetTypeInfo,
722     StaticWidget_GetIDsOfNames,
723     StaticWidget_Invoke,
724     StaticWidget_TestDual
725 };
726
727 static IStaticWidget StaticWidget = { &StaticWidgetVtbl };
728
729 typedef struct KindaEnum
730 {
731     IKindaEnumWidget IKindaEnumWidget_iface;
732     LONG refs;
733 } KindaEnum;
734
735 static inline KindaEnum *impl_from_IKindaEnumWidget(IKindaEnumWidget *iface)
736 {
737     return CONTAINING_RECORD(iface, KindaEnum, IKindaEnumWidget_iface);
738 }
739
740 static HRESULT register_current_module_typelib(void)
741 {
742     WCHAR path[MAX_PATH];
743     CHAR pathA[MAX_PATH];
744     HRESULT hr;
745     ITypeLib *typelib;
746
747     GetModuleFileNameA(NULL, pathA, MAX_PATH);
748     MultiByteToWideChar(CP_ACP, 0, pathA, -1, path, MAX_PATH);
749
750     hr = LoadTypeLib(path, &typelib);
751     if (SUCCEEDED(hr))
752     {
753         hr = RegisterTypeLib(typelib, path, NULL);
754         ITypeLib_Release(typelib);
755     }
756     return hr;
757 }
758
759 static ITypeInfo *get_type_info(REFIID riid)
760 {
761     ITypeInfo *pTypeInfo;
762     ITypeLib *pTypeLib;
763     HRESULT hr;
764
765     hr = LoadRegTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL, &pTypeLib);
766     ok_ole_success(hr, LoadRegTypeLib);
767     if (FAILED(hr))
768         return NULL;
769
770     hr = ITypeLib_GetTypeInfoOfGuid(pTypeLib, riid, &pTypeInfo);
771     ITypeLib_Release(pTypeLib);
772     ok_ole_success(hr, ITypeLib_GetTypeInfoOfGuid);
773     if (FAILED(hr))
774         return NULL;
775
776     return pTypeInfo;
777 }
778
779 static IWidget *Widget_Create(void)
780 {
781     Widget *This;
782     ITypeInfo *pTypeInfo;
783     HRESULT hr = E_FAIL;
784
785     pTypeInfo = get_type_info(&IID_IWidget);
786     if(!pTypeInfo)
787         return NULL;
788
789     This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
790     This->IWidget_iface.lpVtbl = &Widget_VTable;
791     This->refs = 1;
792     This->pDispatchUnknown = NULL;
793
794     hr = CreateStdDispatch((IUnknown *)&This->IWidget_iface, This, pTypeInfo,
795                            &This->pDispatchUnknown);
796     ok_ole_success(hr, CreateStdDispatch);
797     ITypeInfo_Release(pTypeInfo);
798
799     if (SUCCEEDED(hr))
800         return &This->IWidget_iface;
801     else
802     {
803         HeapFree(GetProcessHeap(), 0, This);
804         return NULL;
805     }
806 }
807
808 static HRESULT WINAPI KindaEnum_QueryInterface(
809     IKindaEnumWidget *iface,
810     /* [in] */ REFIID riid,
811     /* [iid_is][out] */ void __RPC_FAR *__RPC_FAR *ppvObject)
812 {
813     if (IsEqualIID(riid, &IID_IKindaEnumWidget) || IsEqualIID(riid, &IID_IUnknown))
814     {
815         IKindaEnumWidget_AddRef(iface);
816         *ppvObject = iface;
817         return S_OK;
818     }
819     else
820     {
821         *ppvObject = NULL;
822         return E_NOINTERFACE;
823     }
824 }
825
826 static ULONG WINAPI KindaEnum_AddRef(
827     IKindaEnumWidget *iface)
828 {
829     KindaEnum *This = impl_from_IKindaEnumWidget(iface);
830
831     return InterlockedIncrement(&This->refs);
832 }
833
834 static ULONG WINAPI KindaEnum_Release(
835     IKindaEnumWidget *iface)
836 {
837     KindaEnum *This = impl_from_IKindaEnumWidget(iface);
838     ULONG refs = InterlockedDecrement(&This->refs);
839     if (!refs)
840     {
841         memset(This, 0xcc, sizeof(*This));
842         HeapFree(GetProcessHeap(), 0, This);
843         trace("KindaEnumWidget destroyed!\n");
844     }
845
846     return refs;
847 }
848
849 static HRESULT WINAPI KindaEnum_Next(
850     IKindaEnumWidget *iface,
851     /* [out] */ IWidget __RPC_FAR *__RPC_FAR *widget)
852 {
853     *widget = Widget_Create();
854     if (*widget)
855         return S_OK;
856     else
857         return E_OUTOFMEMORY;
858 }
859
860 static HRESULT WINAPI KindaEnum_Count(
861     IKindaEnumWidget *iface,
862     /* [out] */ ULONG __RPC_FAR *count)
863 {
864     return E_NOTIMPL;
865 }
866
867 static HRESULT WINAPI KindaEnum_Reset(
868     IKindaEnumWidget *iface)
869 {
870     return E_NOTIMPL;
871 }
872
873 static HRESULT WINAPI KindaEnum_Clone(
874     IKindaEnumWidget *iface,
875     /* [out] */ IKindaEnumWidget __RPC_FAR *__RPC_FAR *ppenum)
876 {
877     return E_NOTIMPL;
878 }
879
880 static const IKindaEnumWidgetVtbl KindaEnumWidget_VTable =
881 {
882     KindaEnum_QueryInterface,
883     KindaEnum_AddRef,
884     KindaEnum_Release,
885     KindaEnum_Next,
886     KindaEnum_Count,
887     KindaEnum_Reset,
888     KindaEnum_Clone
889 };
890
891 static IKindaEnumWidget *KindaEnumWidget_Create(void)
892 {
893     KindaEnum *This;
894
895     This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
896     if (!This) return NULL;
897     This->IKindaEnumWidget_iface.lpVtbl = &KindaEnumWidget_VTable;
898     This->refs = 1;
899     return &This->IKindaEnumWidget_iface;
900 }
901
902 static HRESULT WINAPI NonOleAutomation_QueryInterface(INonOleAutomation *iface, REFIID riid, void **ppv)
903 {
904     if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_INonOleAutomation))
905     {
906         *(INonOleAutomation **)ppv = iface;
907         return S_OK;
908     }
909     *ppv = NULL;
910     return E_NOINTERFACE;
911 }
912
913 static ULONG WINAPI NonOleAutomation_AddRef(INonOleAutomation *iface)
914 {
915     return 2;
916 }
917
918 static ULONG WINAPI NonOleAutomation_Release(INonOleAutomation *iface)
919 {
920     return 1;
921 }
922
923 static BSTR WINAPI NonOleAutomation_BstrRet(INonOleAutomation *iface)
924 {
925     static const WCHAR wszTestString[] = {'T','h','i','s',' ','i','s',' ','a',' ','t','e','s','t',' ','s','t','r','i','n','g',0};
926     return SysAllocString(wszTestString);
927 }
928
929 static HRESULT WINAPI NonOleAutomation_Error(INonOleAutomation *iface)
930 {
931     return E_NOTIMPL;
932 }
933
934 static INonOleAutomationVtbl NonOleAutomation_VTable =
935 {
936     NonOleAutomation_QueryInterface,
937     NonOleAutomation_AddRef,
938     NonOleAutomation_Release,
939     NonOleAutomation_BstrRet,
940     NonOleAutomation_Error
941 };
942
943 static INonOleAutomation NonOleAutomation = { &NonOleAutomation_VTable };
944
945 static ITypeInfo *NonOleAutomation_GetTypeInfo(void)
946 {
947     ITypeLib *pTypeLib;
948     HRESULT hr = LoadRegTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL, &pTypeLib);
949     ok_ole_success(hr, LoadRegTypeLib);
950     if (SUCCEEDED(hr))
951     {
952         ITypeInfo *pTypeInfo;
953         hr = ITypeLib_GetTypeInfoOfGuid(pTypeLib, &IID_INonOleAutomation, &pTypeInfo);
954         ok_ole_success(hr, ITypeLib_GetTypeInfoOfGuid);
955         ITypeLib_Release(pTypeLib);
956         return pTypeInfo;
957     }
958     return NULL;
959 }
960
961 static void test_typelibmarshal(void)
962 {
963     static const WCHAR szCat[] = { 'C','a','t',0 };
964     static const WCHAR szTestTest[] = { 'T','e','s','t','T','e','s','t',0 };
965     static WCHAR szSuperman[] = { 'S','u','p','e','r','m','a','n',0 };
966     HRESULT hr;
967     IKindaEnumWidget *pKEW = KindaEnumWidget_Create();
968     IWidget *pWidget;
969     IStream *pStream;
970     IDispatch *pDispatch;
971     static const LARGE_INTEGER ullZero;
972     EXCEPINFO excepinfo;
973     VARIANT varresult;
974     DISPID dispidNamed = DISPID_PROPERTYPUT;
975     DISPPARAMS dispparams;
976     VARIANTARG vararg[4];
977     STATE the_state;
978     HANDLE thread;
979     DWORD tid;
980     BSTR bstr;
981     ITypeInfo *pTypeInfo;
982     MYSTRUCT mystruct;
983     MYSTRUCT mystructArray[5];
984     UINT uval;
985
986     ok(pKEW != NULL, "Widget creation failed\n");
987
988     hr = CreateStreamOnHGlobal(NULL, TRUE, &pStream);
989     ok_ole_success(hr, CreateStreamOnHGlobal);
990     tid = start_host_object(pStream, &IID_IKindaEnumWidget, (IUnknown *)pKEW, MSHLFLAGS_NORMAL, &thread);
991     IKindaEnumWidget_Release(pKEW);
992
993     IStream_Seek(pStream, ullZero, STREAM_SEEK_SET, NULL);
994     hr = CoUnmarshalInterface(pStream, &IID_IKindaEnumWidget, (void **)&pKEW);
995     ok_ole_success(hr, CoUnmarshalInterface);
996     IStream_Release(pStream);
997
998     if (hr == S_OK)
999     {
1000         hr = IKindaEnumWidget_Next(pKEW, &pWidget);
1001         ok_ole_success(hr, IKindaEnumWidget_Next);
1002
1003         IKindaEnumWidget_Release(pKEW);
1004     }
1005
1006     hr = IWidget_QueryInterface(pWidget, &IID_IDispatch, (void **)&pDispatch);
1007     ok_ole_success(hr, IWidget_QueryInterface);
1008
1009     /* call put_Name */
1010     VariantInit(&vararg[0]);
1011     dispparams.cNamedArgs = 1;
1012     dispparams.rgdispidNamedArgs = &dispidNamed;
1013     dispparams.cArgs = 1;
1014     dispparams.rgvarg = vararg;
1015     VariantInit(&varresult);
1016     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1017     ok_ole_success(hr, IDispatch_Invoke);
1018     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1019         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1020         excepinfo.wCode, excepinfo.scode);
1021     VariantClear(&varresult);
1022
1023     /* call put_Name (direct) */
1024     bstr = SysAllocString(szSuperman);
1025     hr = IWidget_put_Name(pWidget, bstr);
1026     ok_ole_success(hr, IWidget_put_Name);
1027     SysFreeString(bstr);
1028
1029     /* call get_Name */
1030     dispparams.cNamedArgs = 0;
1031     dispparams.rgdispidNamedArgs = NULL;
1032     dispparams.cArgs = 0;
1033     dispparams.rgvarg = NULL;
1034     VariantInit(&varresult);
1035     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1036     ok_ole_success(hr, IDispatch_Invoke);
1037     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1038         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1039         excepinfo.wCode, excepinfo.scode);
1040     trace("Name = %s\n", wine_dbgstr_w(V_BSTR(&varresult)));
1041     VariantClear(&varresult);
1042
1043     /* call get_Name (direct) */
1044     bstr = NULL;
1045     hr = IWidget_get_Name(pWidget, &bstr);
1046     ok_ole_success(hr, IWidget_get_Name);
1047     ok(!lstrcmpW(bstr, szCat), "IWidget_get_Name should have returned string \"Cat\" instead of %s\n", wine_dbgstr_w(bstr));
1048     SysFreeString(bstr);
1049
1050     /* call DoSomething */
1051     VariantInit(&vararg[0]);
1052     VariantInit(&vararg[1]);
1053     V_VT(&vararg[1]) = VT_R8;
1054     V_R8(&vararg[1]) = 3.141;
1055     dispparams.cNamedArgs = 0;
1056     dispparams.cArgs = 2;
1057     dispparams.rgdispidNamedArgs = NULL;
1058     dispparams.rgvarg = vararg;
1059     VariantInit(&varresult);
1060     hr = IDispatch_Invoke(pDispatch, DISPID_TM_DOSOMETHING, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1061     ok_ole_success(hr, IDispatch_Invoke);
1062     ok(V_VT(&varresult) == VT_EMPTY, "varresult should be VT_EMPTY\n");
1063     VariantClear(&varresult);
1064
1065     /* call get_State */
1066     dispparams.cNamedArgs = 0;
1067     dispparams.cArgs = 0;
1068     dispparams.rgdispidNamedArgs = NULL;
1069     dispparams.rgvarg = NULL;
1070     hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1071     ok_ole_success(hr, IDispatch_Invoke);
1072     ok((V_VT(&varresult) == VT_I4) && (V_I4(&varresult) == STATE_WIDGETIFIED), "Return val mismatch\n");
1073
1074     /* call get_State (direct) */
1075     hr = IWidget_get_State(pWidget, &the_state);
1076     ok_ole_success(hr, IWidget_get_state);
1077     ok(the_state == STATE_WIDGETIFIED, "should have returned WIDGET_WIDGETIFIED instead of %d\n", the_state);
1078
1079     /* call put_State */
1080     the_state = STATE_WIDGETIFIED;
1081     VariantInit(&vararg[0]);
1082     V_VT(&vararg[0]) = VT_BYREF|VT_I4;
1083     V_I4REF(&vararg[0]) = (int *)&the_state;
1084     dispparams.cNamedArgs = 1;
1085     dispparams.cArgs = 1;
1086     dispparams.rgdispidNamedArgs = &dispidNamed;
1087     dispparams.rgvarg = vararg;
1088     hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1089     ok_ole_success(hr, IDispatch_Invoke);
1090
1091     /* call Map */
1092     bstr = SysAllocString(szTestTest);
1093     VariantInit(&vararg[0]);
1094     V_VT(&vararg[0]) = VT_BYREF|VT_BSTR;
1095     V_BSTRREF(&vararg[0]) = &bstr;
1096     dispparams.cNamedArgs = 0;
1097     dispparams.cArgs = 1;
1098     dispparams.rgdispidNamedArgs = NULL;
1099     dispparams.rgvarg = vararg;
1100     VariantInit(&varresult);
1101     hr = IDispatch_Invoke(pDispatch, DISPID_TM_MAP, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1102     ok_ole_success(hr, IDispatch_Invoke);
1103     ok(V_VT(&varresult) == VT_BSTR, "Return value should be of type BSTR instead of %d\n", V_VT(&varresult));
1104     ok(!lstrcmpW(V_BSTR(&varresult), szTestTest), "Return value should have been \"TestTest\" instead of %s\n", wine_dbgstr_w(V_BSTR(&varresult)));
1105     VariantClear(&varresult);
1106     SysFreeString(bstr);
1107
1108     /* call SetOleColor with large negative VT_I4 param */
1109     VariantInit(&vararg[0]);
1110     V_VT(&vararg[0]) = VT_I4;
1111     V_I4(&vararg[0]) = 0x80000005;
1112     dispparams.cNamedArgs = 0;
1113     dispparams.cArgs = 1;
1114     dispparams.rgdispidNamedArgs = NULL;
1115     dispparams.rgvarg = vararg;
1116     hr = IDispatch_Invoke(pDispatch, DISPID_TM_SETOLECOLOR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, &excepinfo, NULL);
1117     ok_ole_success(hr, IDispatch_Invoke);
1118
1119     /* call GetOleColor */
1120     dispparams.cNamedArgs = 0;
1121     dispparams.cArgs = 0;
1122     dispparams.rgdispidNamedArgs = NULL;
1123     dispparams.rgvarg = NULL;
1124     VariantInit(&varresult);
1125     hr = IDispatch_Invoke(pDispatch, DISPID_TM_GETOLECOLOR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1126     ok_ole_success(hr, IDispatch_Invoke);
1127     VariantClear(&varresult);
1128
1129     /* call StructArgs (direct) */
1130     mystruct = MYSTRUCT_BYPTR;
1131     memcpy(mystructArray, MYSTRUCT_ARRAY, sizeof(mystructArray));
1132     hr = IWidget_StructArgs(pWidget, MYSTRUCT_BYVAL, &mystruct, mystructArray);
1133     ok_ole_success(hr, IWidget_StructArgs);
1134
1135     /* call Clone */
1136     dispparams.cNamedArgs = 0;
1137     dispparams.cArgs = 0;
1138     dispparams.rgdispidNamedArgs = NULL;
1139     dispparams.rgvarg = NULL;
1140     VariantInit(&varresult);
1141     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1142     ok_ole_success(hr, IDispatch_Invoke);
1143     ok(V_VT(&varresult) == VT_DISPATCH, "vt %x\n", V_VT(&varresult));
1144     VariantClear(&varresult);
1145
1146     /* call CloneInterface */
1147     dispparams.cNamedArgs = 0;
1148     dispparams.cArgs = 0;
1149     dispparams.rgdispidNamedArgs = NULL;
1150     dispparams.rgvarg = NULL;
1151     VariantInit(&varresult);
1152     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONEINTERFACE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1153     ok_ole_success(hr, IDispatch_Invoke);
1154     ok(V_VT(&varresult) == VT_DISPATCH, "vt %x\n", V_VT(&varresult));
1155     VariantClear(&varresult);
1156
1157     /* call CloneDispatch with automatic value getting */
1158     V_VT(&vararg[0]) = VT_I2;
1159     V_I2(&vararg[0]) = 1;
1160     dispparams.cNamedArgs = 0;
1161     dispparams.rgdispidNamedArgs = NULL;
1162     dispparams.cArgs = 1;
1163     dispparams.rgvarg = vararg;
1164     VariantInit(&varresult);
1165     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONEDISPATCH, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1166     ok_ole_success(hr, IDispatch_Invoke);
1167
1168     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1169         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1170         excepinfo.wCode, excepinfo.scode);
1171
1172     ok(V_VT(&varresult) == VT_I2, "V_VT(&varresult) was %d instead of VT_I2\n", V_VT(&varresult));
1173     ok(V_I2(&varresult) == 1234, "V_I2(&varresult) was %d instead of 1234\n", V_I2(&varresult));
1174     VariantClear(&varresult);
1175
1176     /* call CloneCoclass */
1177     dispparams.cNamedArgs = 0;
1178     dispparams.cArgs = 0;
1179     dispparams.rgdispidNamedArgs = NULL;
1180     dispparams.rgvarg = NULL;
1181     VariantInit(&varresult);
1182     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONECOCLASS, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1183     ok_ole_success(hr, IDispatch_Invoke);
1184
1185     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1186        "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1187        excepinfo.wCode, excepinfo.scode);
1188
1189     ok(V_VT(&varresult) == VT_DISPATCH, "V_VT(&varresult) was %d instead of VT_DISPATCH\n", V_VT(&varresult));
1190     ok(!V_DISPATCH(&varresult), "V_DISPATCH(&varresult) should be NULL instead of %p\n", V_DISPATCH(&varresult));
1191     VariantClear(&varresult);
1192
1193     /* call Value with a VT_VARIANT|VT_BYREF type */
1194     V_VT(&vararg[0]) = VT_VARIANT|VT_BYREF;
1195     V_VARIANTREF(&vararg[0]) = &vararg[1];
1196     V_VT(&vararg[1]) = VT_I2;
1197     V_I2(&vararg[1]) = 1;
1198     dispparams.cNamedArgs = 0;
1199     dispparams.rgdispidNamedArgs = NULL;
1200     dispparams.cArgs = 1;
1201     dispparams.rgvarg = vararg;
1202     VariantInit(&varresult);
1203     hr = IDispatch_Invoke(pDispatch, DISPID_VALUE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1204     ok_ole_success(hr, IDispatch_Invoke);
1205
1206     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1207         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1208         excepinfo.wCode, excepinfo.scode);
1209
1210     ok(V_VT(&varresult) == VT_I2, "V_VT(&varresult) was %d instead of VT_I2\n", V_VT(&varresult));
1211     ok(V_I2(&varresult) == 1234, "V_I2(&varresult) was %d instead of 1234\n", V_I2(&varresult));
1212     VariantClear(&varresult);
1213
1214     /* call Variant - exercises variant copying in ITypeInfo::Invoke and
1215      * handling of void return types */
1216     /* use a big type to ensure that the variant was properly copied into the
1217      * destination function's args */
1218     V_VT(&vararg[0]) = VT_CY;
1219     S(V_CY(&vararg[0])).Hi = 0xdababe;
1220     S(V_CY(&vararg[0])).Lo = 0xdeadbeef;
1221     dispparams.cNamedArgs = 0;
1222     dispparams.cArgs = 1;
1223     dispparams.rgdispidNamedArgs = NULL;
1224     dispparams.rgvarg = vararg;
1225     VariantInit(&varresult);
1226     hr = IDispatch_Invoke(pDispatch, DISPID_TM_VARIANT, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, NULL, NULL);
1227     ok_ole_success(hr, IDispatch_Invoke);
1228     VariantClear(&varresult);
1229
1230     /* call VarArg */
1231     VariantInit(&vararg[3]);
1232     V_VT(&vararg[3]) = VT_I4;
1233     V_I4(&vararg[3]) = 3;
1234     VariantInit(&vararg[2]);
1235     V_VT(&vararg[2]) = VT_I4;
1236     V_I4(&vararg[2]) = 0;
1237     VariantInit(&vararg[1]);
1238     V_VT(&vararg[1]) = VT_I4;
1239     V_I4(&vararg[1]) = 1;
1240     VariantInit(&vararg[0]);
1241     V_VT(&vararg[0]) = VT_I4;
1242     V_I4(&vararg[0]) = 2;
1243     dispparams.cNamedArgs = 0;
1244     dispparams.cArgs = 4;
1245     dispparams.rgdispidNamedArgs = NULL;
1246     dispparams.rgvarg = vararg;
1247     hr = IDispatch_Invoke(pDispatch, DISPID_TM_VARARG, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, NULL, NULL);
1248     ok_ole_success(hr, ITypeInfo_Invoke);
1249
1250     /* call VarArg, even one (non-optional, non-safearray) named argument is not allowed */
1251     dispidNamed = 0;
1252     dispparams.cNamedArgs = 1;
1253     dispparams.rgdispidNamedArgs = &dispidNamed;
1254     hr = IDispatch_Invoke(pDispatch, DISPID_TM_VARARG, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, NULL, NULL);
1255     ok(hr == DISP_E_NONAMEDARGS, "IDispatch_Invoke should have returned DISP_E_NONAMEDARGS instead of 0x%08x\n", hr);
1256     dispidNamed = DISPID_PROPERTYPUT;
1257
1258     /* call Error */
1259     dispparams.cNamedArgs = 0;
1260     dispparams.cArgs = 0;
1261     dispparams.rgdispidNamedArgs = NULL;
1262     dispparams.rgvarg = NULL;
1263     VariantInit(&varresult);
1264     hr = IDispatch_Invoke(pDispatch, DISPID_TM_ERROR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, &excepinfo, NULL);
1265     ok(hr == DISP_E_EXCEPTION, "IDispatch_Invoke should have returned DISP_E_EXCEPTION instead of 0x%08x\n", hr);
1266     ok(excepinfo.wCode == 0x0 && excepinfo.scode == E_NOTIMPL,
1267         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1268         excepinfo.wCode, excepinfo.scode);
1269     VariantClear(&varresult);
1270
1271     /* call BstrRet */
1272     pTypeInfo = NonOleAutomation_GetTypeInfo();
1273     dispparams.cNamedArgs = 0;
1274     dispparams.cArgs = 0;
1275     dispparams.rgdispidNamedArgs = NULL;
1276     dispparams.rgvarg = NULL;
1277     VariantInit(&varresult);
1278     hr = ITypeInfo_Invoke(pTypeInfo, &NonOleAutomation, DISPID_NOA_BSTRRET, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1279     ok_ole_success(hr, ITypeInfo_Invoke);
1280     ok(V_VT(&varresult) == VT_BSTR, "V_VT(&varresult) should be VT_BSTR instead of %d\n", V_VT(&varresult));
1281     ok(V_BSTR(&varresult) != NULL, "V_BSTR(&varresult) should not be NULL\n");
1282
1283     VariantClear(&varresult);
1284
1285     dispparams.cNamedArgs = 0;
1286     dispparams.cArgs = 0;
1287     dispparams.rgdispidNamedArgs = NULL;
1288     dispparams.rgvarg = NULL;
1289     hr = ITypeInfo_Invoke(pTypeInfo, &NonOleAutomation, DISPID_NOA_ERROR, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1290     ok(hr == DISP_E_EXCEPTION, "ITypeInfo_Invoke should have returned DISP_E_EXCEPTION instead of 0x%08x\n", hr);
1291     ok(V_VT(&varresult) == VT_EMPTY, "V_VT(&varresult) should be VT_EMPTY instead of %d\n", V_VT(&varresult));
1292     ok(excepinfo.wCode == 0x0 && excepinfo.scode == E_NOTIMPL,
1293         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1294         excepinfo.wCode, excepinfo.scode);
1295     VariantClear(&varresult);
1296
1297     ITypeInfo_Release(pTypeInfo);
1298
1299     /* tests call put_Name without named arg */
1300     VariantInit(&vararg[0]);
1301     dispparams.cNamedArgs = 0;
1302     dispparams.rgdispidNamedArgs = NULL;
1303     dispparams.cArgs = 1;
1304     dispparams.rgvarg = vararg;
1305     VariantInit(&varresult);
1306     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1307     ok(hr == DISP_E_PARAMNOTFOUND, "IDispatch_Invoke should have returned DISP_E_PARAMNOTFOUND instead of 0x%08x\n", hr);
1308     VariantClear(&varresult);
1309
1310     /* tests param type that cannot be coerced */
1311     VariantInit(&vararg[0]);
1312     V_VT(&vararg[0]) = VT_UNKNOWN;
1313     V_UNKNOWN(&vararg[0]) = NULL;
1314     dispparams.cNamedArgs = 1;
1315     dispparams.rgdispidNamedArgs = &dispidNamed;
1316     dispparams.cArgs = 1;
1317     dispparams.rgvarg = vararg;
1318     VariantInit(&varresult);
1319     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1320     ok(hr == DISP_E_TYPEMISMATCH, "IDispatch_Invoke should have returned DISP_E_TYPEMISMATCH instead of 0x%08x\n", hr);
1321     VariantClear(&varresult);
1322
1323     /* tests bad param type */
1324     VariantInit(&vararg[0]);
1325     V_VT(&vararg[0]) = VT_CLSID;
1326     V_BYREF(&vararg[0]) = NULL;
1327     dispparams.cNamedArgs = 1;
1328     dispparams.rgdispidNamedArgs = &dispidNamed;
1329     dispparams.cArgs = 1;
1330     dispparams.rgvarg = vararg;
1331     VariantInit(&varresult);
1332     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1333     ok(hr == DISP_E_BADVARTYPE, "IDispatch_Invoke should have returned DISP_E_BADVARTYPE instead of 0x%08x\n", hr);
1334     VariantClear(&varresult);
1335
1336     /* tests too small param count */
1337     dispparams.cNamedArgs = 0;
1338     dispparams.rgdispidNamedArgs = NULL;
1339     dispparams.cArgs = 0;
1340     dispparams.rgvarg = NULL;
1341     VariantInit(&varresult);
1342     hr = IDispatch_Invoke(pDispatch, DISPID_TM_DOSOMETHING, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1343     ok(hr == DISP_E_BADPARAMCOUNT, "IDispatch_Invoke should have returned DISP_E_BADPARAMCOUNT instead of 0x%08x\n", hr);
1344     VariantClear(&varresult);
1345
1346     /* tests propget function with large param count */
1347     VariantInit(&vararg[0]);
1348     V_VT(&vararg[0]) = VT_BSTR;
1349     V_BSTR(&vararg[0]) = NULL;
1350     V_VT(&vararg[1]) = VT_I4;
1351     V_I4(&vararg[1]) = 1;
1352     dispparams.cNamedArgs = 0;
1353     dispparams.cArgs = 2;
1354     dispparams.rgdispidNamedArgs = NULL;
1355     dispparams.rgvarg = vararg;
1356     hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1357     ok(hr == DISP_E_NOTACOLLECTION, "IDispatch_Invoke should have returned DISP_E_NOTACOLLECTION instead of 0x%08x\n", hr);
1358
1359     /* test propput with lcid */
1360
1361     /* the lcid passed to the function is the first lcid in the typelib header.
1362        Since we don't explicitly set an lcid in the idl, it'll default to US English. */
1363     VariantInit(&vararg[0]);
1364     V_VT(&vararg[0]) = VT_I4;
1365     V_I4(&vararg[0]) = 0xcafe;
1366     dispparams.cNamedArgs = 1;
1367     dispparams.rgdispidNamedArgs = &dispidNamed;
1368     dispparams.cArgs = 1;
1369     dispparams.rgvarg = vararg;
1370     VariantInit(&varresult);
1371     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_WITH_LCID, &IID_NULL, 0x40c, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1372     ok_ole_success(hr, ITypeInfo_Invoke);
1373     VariantClear(&varresult);
1374
1375     /* test propget with lcid */
1376     dispparams.cNamedArgs = 0;
1377     dispparams.cArgs = 0;
1378     dispparams.rgvarg = NULL;
1379     dispparams.rgdispidNamedArgs = NULL;
1380     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_WITH_LCID, &IID_NULL, 0x40c, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1381     ok_ole_success(hr, ITypeInfo_Invoke);
1382     ok(V_VT(&varresult) == VT_I4, "got %x\n", V_VT(&varresult));
1383     ok(V_I4(&varresult) == 0x409, "got %x\n", V_I4(&varresult));
1384     VariantClear(&varresult);
1385
1386     /* test propget of INT value */
1387     dispparams.cNamedArgs = 0;
1388     dispparams.cArgs = 0;
1389     dispparams.rgvarg = NULL;
1390     dispparams.rgdispidNamedArgs = NULL;
1391     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_INT, &IID_NULL, 0x40c, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1392     ok_ole_success(hr, ITypeInfo_Invoke);
1393     ok(V_VT(&varresult) == VT_I4, "got %x\n", V_VT(&varresult));
1394     ok(V_I4(&varresult) == -13, "got %x\n", V_I4(&varresult));
1395     VariantClear(&varresult);
1396
1397     /* test propget of INT value */
1398     dispparams.cNamedArgs = 0;
1399     dispparams.cArgs = 0;
1400     dispparams.rgvarg = NULL;
1401     dispparams.rgdispidNamedArgs = NULL;
1402     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_UINT, &IID_NULL, 0x40c, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1403     ok_ole_success(hr, ITypeInfo_Invoke);
1404     ok(V_VT(&varresult) == VT_UI4, "got %x\n", V_VT(&varresult));
1405     ok(V_UI4(&varresult) == 42, "got %x\n", V_UI4(&varresult));
1406     VariantClear(&varresult);
1407
1408     /* test byref marshalling */
1409     uval = 666;
1410     VariantInit(&vararg[0]);
1411     V_VT(&vararg[0]) = VT_UI4|VT_BYREF;
1412     V_UI4REF(&vararg[0]) = &uval;
1413     dispparams.cNamedArgs = 0;
1414     dispparams.cArgs = 1;
1415     dispparams.rgvarg = vararg;
1416     dispparams.rgdispidNamedArgs = NULL;
1417     hr = IDispatch_Invoke(pDispatch, DISPID_TM_BYREF_UINT, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1418     ok_ole_success(hr, ITypeInfo_Invoke);
1419     ok(V_VT(&varresult) == VT_EMPTY, "varresult should be VT_EMPTY\n");
1420     ok(V_VT(&vararg[0]) == (VT_UI4|VT_BYREF), "arg VT not unmarshalled correctly: %x\n", V_VT(&vararg[0]));
1421     ok(V_UI4REF(&vararg[0]) == &uval, "Byref pointer not preserved: %p/%p\n", &uval, V_UI4REF(&vararg[0]));
1422     ok(*V_UI4REF(&vararg[0]) == 42, "Expected 42 to be returned instead of %u\n", *V_UI4REF(&vararg[0]));
1423     VariantClear(&varresult);
1424     VariantClear(&vararg[0]);
1425
1426     /* test propput with optional argument. */
1427     VariantInit(&vararg[0]);
1428     V_VT(&vararg[0]) = VT_I4;
1429     V_I4(&vararg[0]) = 0xcafe;
1430     dispparams.cNamedArgs = 1;
1431     dispparams.rgdispidNamedArgs = &dispidNamed;
1432     dispparams.cArgs = 1;
1433     dispparams.rgvarg = vararg;
1434     VariantInit(&varresult);
1435     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_OPT_ARG, &IID_NULL, 0x40c, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1436     ok_ole_success(hr, ITypeInfo_Invoke);
1437     VariantClear(&varresult);
1438
1439     /* test propput with required argument. */
1440     VariantInit(&vararg[0]);
1441     VariantInit(&vararg[1]);
1442     V_VT(&vararg[0]) = VT_I4;
1443     V_I4(&vararg[0]) = 0x1234;
1444     V_VT(&vararg[1]) = VT_I4;
1445     V_I4(&vararg[1]) = 0x5678;
1446     dispparams.cNamedArgs = 1;
1447     dispparams.rgdispidNamedArgs = &dispidNamed;
1448     dispparams.cArgs = 2;
1449     dispparams.rgvarg = vararg;
1450     VariantInit(&varresult);
1451     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_REQ_ARG, &IID_NULL, 0x40c, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1452     ok_ole_success(hr, ITypeInfo_Invoke);
1453     VariantClear(&varresult);
1454
1455     IDispatch_Release(pDispatch);
1456     IWidget_Release(pWidget);
1457
1458     trace("calling end_host_object\n");
1459     end_host_object(tid, thread);
1460 }
1461
1462 static void test_DispCallFunc(void)
1463 {
1464     static const WCHAR szEmpty[] = { 0 };
1465     VARTYPE rgvt[] = { VT_R8, VT_BSTR, VT_BSTR, VT_VARIANT|VT_BYREF };
1466     VARIANTARG vararg[4];
1467     VARIANTARG varref;
1468     VARIANTARG *rgpvarg[4] = { &vararg[0], &vararg[1], &vararg[2], &vararg[3] };
1469     VARIANTARG varresult;
1470     HRESULT hr;
1471     IWidget *pWidget = Widget_Create();
1472     V_VT(&vararg[0]) = VT_R8;
1473     V_R8(&vararg[0]) = 3.141;
1474     V_VT(&vararg[1]) = VT_BSTR;
1475     V_BSTRREF(&vararg[1]) = CoTaskMemAlloc(sizeof(BSTR));
1476     V_VT(&vararg[2]) = VT_BSTR;
1477     V_BSTR(&vararg[2]) = SysAllocString(szEmpty);
1478     V_VT(&vararg[3]) = VT_VARIANT|VT_BYREF;
1479     V_VARIANTREF(&vararg[3]) = &varref;
1480     V_VT(&varref) = VT_ERROR;
1481     V_ERROR(&varref) = DISP_E_PARAMNOTFOUND;
1482     VariantInit(&varresult);
1483     hr = DispCallFunc(pWidget, 9*sizeof(void*), CC_STDCALL, VT_UI4, 4, rgvt, rgpvarg, &varresult);
1484     ok_ole_success(hr, DispCallFunc);
1485     VariantClear(&varresult);
1486     SysFreeString(*V_BSTRREF(&vararg[1]));
1487     CoTaskMemFree(V_BSTRREF(&vararg[1]));
1488     VariantClear(&vararg[2]);
1489     IWidget_Release(pWidget);
1490 }
1491
1492 static void test_StaticWidget(void)
1493 {
1494     ITypeInfo *type_info;
1495     DISPPARAMS dispparams;
1496     VARIANTARG vararg[4];
1497     EXCEPINFO excepinfo;
1498     VARIANT varresult;
1499     HRESULT hr;
1500
1501     type_info = get_type_info(&IID_IStaticWidget);
1502
1503     /* call TestDual */
1504     dispparams.cNamedArgs = 0;
1505     dispparams.cArgs = 1;
1506     dispparams.rgdispidNamedArgs = NULL;
1507     dispparams.rgvarg = vararg;
1508     V_VT(vararg) = VT_DISPATCH;
1509     V_DISPATCH(vararg) = (IDispatch*)&TestDualDisp;
1510     VariantInit(&varresult);
1511     hr = ITypeInfo_Invoke(type_info, &StaticWidget, DISPID_TM_TESTDUAL, DISPATCH_METHOD,
1512             &dispparams, &varresult, &excepinfo, NULL);
1513     ok_ole_success(hr, IDispatch_Invoke);
1514     ok(V_VT(&varresult) == VT_EMPTY, "vt %x\n", V_VT(&varresult));
1515     VariantClear(&varresult);
1516
1517     ITypeInfo_Release(type_info);
1518 }
1519
1520 static void test_libattr(void)
1521 {
1522     ITypeLib *pTypeLib;
1523     HRESULT hr;
1524     TLIBATTR *pattr;
1525
1526     hr = LoadRegTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL, &pTypeLib);
1527     ok_ole_success(hr, LoadRegTypeLib);
1528     if (FAILED(hr))
1529         return;
1530
1531     hr = ITypeLib_GetLibAttr(pTypeLib, &pattr);
1532     ok_ole_success(hr, GetLibAttr);
1533     if (SUCCEEDED(hr))
1534     {
1535         ok(pattr->lcid == MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL), "lcid %x\n", pattr->lcid);
1536
1537         ITypeLib_ReleaseTLibAttr(pTypeLib, pattr);
1538     }
1539
1540     ITypeLib_Release(pTypeLib);
1541 }
1542
1543 START_TEST(tmarshal)
1544 {
1545     HRESULT hr;
1546
1547     CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
1548
1549     hr = register_current_module_typelib();
1550     ok_ole_success(hr, register_current_module_typelib);
1551
1552     test_typelibmarshal();
1553     test_DispCallFunc();
1554     test_StaticWidget();
1555     test_libattr();
1556
1557     hr = UnRegisterTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL,
1558                            sizeof(void*) == 8 ? SYS_WIN64 : SYS_WIN32);
1559     ok_ole_success(hr, UnRegisterTypeLib);
1560
1561     CoUninitialize();
1562 }